admin/monoxgas-sRDI
1
0
Fork 0
mirror of https://github.com/monoxgas/sRDI synced 2026-06-06 16:14:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add arguments.output_format to ConvertToShellcode.py

Browse Source
This commit is contained in:
hmsec
2021-08-09 19:02:14 +04:00
committed by Nick Landers
parent 7889036a75
commit 5690685aee
1 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Python/ConvertToShellcode.py
+15 -4
View File
@@ -11,13 +11,14 @@ def main():
parser.add_argument('-u', '--user-data', dest='user_data', help='Data to pass to the target function', default='dave')
parser.add_argument('-c', '--clear-header', dest='clear_header', action='store_true', help='Clear the PE header on load')
parser.add_argument('-i', '--obfuscate-imports', dest='obfuscate_imports', action='store_true', help='Randomize import dependency load order', default=False)
parser.add_argument('-d', '--import-delay', dest='import_delay', help='Number of seconds to pause between loading imports', type=int, default=0)
parser.add_argument('-d', '--import-delay', dest='import_delay', help='Number of seconds to pause between loading imports', type=int, default=0)
parser.add_argument('-of', '--output-format', dest='output_format', help='Output format of the shellcode (e.g. raw,string)', type=str, default="raw")
arguments = parser.parse_args()
input_dll = arguments.input_dll
output_bin = input_dll.replace('.dll', '.bin')
print('Creating Shellcode: {}'.format(output_bin))
dll = open(arguments.input_dll, 'rb').read()
flags = 0
@@ -29,9 +30,19 @@ def main():
flags = flags | 0x4 | arguments.import_delay << 16
converted_dll = ConvertToShellcode(dll, HashFunctionName(arguments.function_name), arguments.user_data.encode(), flags)
if arguments.output_format=="raw":
print('Creating Shellcode: {}'.format(output_bin))
with open(output_bin, 'wb') as f:
f.write(converted_dll)
elif arguments.output_format=="string":
output_bin = input_dll.replace('.dll', '.txt')
converted_dll_text ="".join([r"\x{}".format(str(format(c,'02x'))) for c in converted_dll])
with open(output_bin, 'wb') as f:
f.write(converted_dll)
print('Creating Shellcode: {}'.format(output_bin))
with open(output_bin, 'w') as f:
f.write(converted_dll_text)
if __name__ == '__main__':
main()