fix: some dll bugs

pe/derbackdoorer.py

@@ -51,11 +51,14 @@ class FunctionBackdoorer:
             logger.warn("Text section too small?")
 
         # write
+        #logger.info("Trampoline: {}".format(compiled_trampoline))
+        #asm_disasm(compiled_trampoline, offset=function_addr)
         self.superpe.pe.set_bytes_at_rva(addr, bytes(compiled_trampoline))
 
         # Show Result
         logger.info("--[ Patched result of function: ".format())
-        data = self.pe_data[function_addr:addr+len(compiled_trampoline)]
+        #data = self.pe_data[function_addr:addr+len(compiled_trampoline)]
+        data = self.superpe.pe.get_data(function_addr, addr+len(compiled_trampoline)-function_addr)
         asm_disasm(data, offset=function_addr)
 
 

pe/superpe.py

@@ -140,6 +140,8 @@ class SuperPe():
 
 
     def patch_subsystem(self):
+        if self.is_dll():
+            return
         if self.pe.OPTIONAL_HEADER.Subsystem != pefile.SUBSYSTEM_TYPE['IMAGE_SUBSYSTEM_WINDOWS_GUI']:
             logger.info("PE is not a GUI application. Patching subsystem to GUI")
             self.pe.OPTIONAL_HEADER.Subsystem = pefile.SUBSYSTEM_TYPE['IMAGE_SUBSYSTEM_WINDOWS_GUI']
@@ -228,7 +230,7 @@ class SuperPe():
     def getExportEntryPoint(self, exportName: str):
         dec = lambda x: '???' if x is None else x.decode() 
         d = [pefile.DIRECTORY_ENTRY["IMAGE_DIRECTORY_ENTRY_EXPORT"]]
-        self.pe.parse_data_directories(directories=d)
+        #self.pe.parse_data_directories(directories=d)
 
         if self.pe.DIRECTORY_ENTRY_EXPORT.symbols == 0:
             raise Exception('No DLL exports found!')
@@ -319,11 +321,12 @@ class SuperPe():
         for reloc in self.get_base_relocs():
             reloc_addr = reloc.rva
             if reloc_addr >= section.virt_addr and reloc_addr < section.virt_addr + section.virt_size:
+                #logger.info("ADDR: 0x{:X}".format(reloc_addr))
                 ret.append(reloc)
         return ret
     
 
-    ## IAT related
+    ## IAT
 
     def get_vaddr_of_iatentry(self, func_name: str) -> int:
         iat = self.get_iat_entries()
@@ -336,6 +339,7 @@ class SuperPe():
 
     def get_iat_entries(self) -> Dict[str, IatEntry]:
         return self.iat_entries
+    
 
     def make_iat_entries(self) -> Dict[str, IatEntry]:
         iat = {}