Added cleaning up the allocated memory. not for shellcodeinject

dllinject.py

@@ -203,7 +203,7 @@ def inject(self, dllpath): #inject dllpath into our process
                                         len(loaddll),
                                         None
                                         )
-        self.kernel32.CreateRemoteThread(
+        thread = self.kernel32.CreateRemoteThread(
                                         self.handle,
                                         None,
                                         0,
@@ -212,6 +212,16 @@ def inject(self, dllpath): #inject dllpath into our process
                                         0,
                                         None
                                         )
+        self.kernel32.WaitForSingleObject(
+                                        thread,
+                                        -1
+                                        )
+        self.kernel32.VirtualFreeEx(
+                                    self.handle,
+                                    shellcodeaddress,
+                                    len(loaddll),
+                                    0x8000
+                                    )
 
     def injectshellcode(self, shellcode):
         """This function merely executes what it is given"""
@@ -229,7 +239,7 @@ def injectshellcode(self, shellcode):
                                         len(shellcode),
                                         None
                                         )
-        self.kernel32.CreateRemoteThread(
+        thread = self.kernel32.CreateRemoteThread(
                                         self.handle,
                                         None,
                                         0,
@@ -240,6 +250,7 @@ def injectshellcode(self, shellcode):
                                         )
 
     def injectshellcodefromfile(self, file):
+        """This function merely executes what it is given as a raw file"""
         fh=open(file,'rb')
         shellcode=fh.read()
         fh.close()