admin/SuperMega
1
0
Fork 0
mirror of https://github.com/dobin/SuperMega synced 2026-06-02 17:27:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: cleanup temp files

Browse Source
This commit is contained in:
Dobin
2024-02-18 13:55:55 +00:00
parent a36b22db84
commit 224d252e78
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_derbackdoorer.py
+3 -2
View File
@@ -23,11 +23,11 @@ from derbackdoorer.derbackdoorer import PeBackdoor
class DerBackdoorerTest(unittest.TestCase): class DerBackdoorerTest(unittest.TestCase):
def test_backdoor_ep(self): def test_backdoor_ep(self):
# Write example shellcode # Write example shellcode
shellcode_path = "exes/shellcode.test"
shellcode = b"\x90" * 200 shellcode = b"\x90" * 200
with open("exes/shellcode.test", "wb") as f: with open(shellcode_path, "wb") as f:
f.write(shellcode) f.write(shellcode)
shellcode_path = "exes/shellcode.test"
exe_path = "exes/iattest-full.exe" exe_path = "exes/iattest-full.exe"
exe_out_path = "exes/iattest-full-test.exe" exe_out_path = "exes/iattest-full-test.exe"
@@ -48,6 +48,7 @@ class DerBackdoorerTest(unittest.TestCase):
self.assertEqual(shellcode, extracted_code) self.assertEqual(shellcode, extracted_code)
os.remove(exe_out_path) os.remove(exe_out_path)
os.remove(shellcode_path)
def test_backdoor_hijack(self): def test_backdoor_hijack(self):