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

fix: exceptions -> errors

Browse Source
This commit is contained in:
Dobin
2024-03-09 11:56:33 +00:00
parent 90ae7b83ad
commit 7aedc0acbd
1 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
supermega.py
+12 -5
View File
@@ -175,14 +175,18 @@ def start(settings: Settings):
short_call_patching = project.settings.short_call_patching) short_call_patching = project.settings.short_call_patching)
except Exception as e: except Exception as e:
logger.error(f'Error compiling: {e}') logger.error(f'Error compiling: {e}')
return exit(1) return 1
# Assemble: Assemble .asm to .shc (ASM -> SHC) # Assemble: Assemble .asm to .shc (ASM -> SHC)
if settings.generate_shc_from_asm: if settings.generate_shc_from_asm:
phases.assembler.asm_to_shellcode( try:
asm_in = main_asm_file, phases.assembler.asm_to_shellcode(
build_exe = main_exe_file, asm_in = main_asm_file,
shellcode_out = main_shc_file) build_exe = main_exe_file,
shellcode_out = main_shc_file)
except Exception as e:
logger.error("Error: Assembling failed")
return 2
# Merge: shellcode/loader with payload (SHC + PAYLOAD -> SHC) # Merge: shellcode/loader with payload (SHC + PAYLOAD -> SHC)
if settings.dataref_style == DataRefStyle.APPEND: if settings.dataref_style == DataRefStyle.APPEND:
@@ -207,6 +211,9 @@ def start(settings: Settings):
except PermissionError as e: except PermissionError as e:
logger.error(f'Error writing file: {e}') logger.error(f'Error writing file: {e}')
return exit(2) return exit(2)
except Exception as e:
logger.error(f'Error injecting: {e}')
return exit(3)
observer.add_code("exe_final", extract_code_from_exe_file_ep(settings.inject_exe_out, 300)) observer.add_code("exe_final", extract_code_from_exe_file_ep(settings.inject_exe_out, 300))