refactor: small SuperPe improvements

2026-06-02 17:27:10 +00:00 · 2024-04-13 11:12:43 +01:00
parent f40d527783
commit b9d3a5a97e
3 changed files with 51 additions and 62 deletions
@@ -114,9 +114,10 @@ class PeBackdoor:
        trampoline = ''
        addrOffset = -1

-        registers = ['rax', 'rbx', 'rcx', 'rdx', 'rsi', 'rdi']
-
-        if self.superpe.arch == 'x86':
+        if self.superpe.is_64():
+            registers = ['rax', 'rbx', 'rcx', 'rdx', 'rsi', 'rdi']
+        else:
+            # Not really used
            registers = ['eax', 'ebx', 'ecx', 'edx', 'esi', 'edi'] 

        reg = random.choice(registers).upper()