refactor: better InvokeStyle name

This commit is contained in:
Dobin
2024-06-05 19:43:35 +01:00
parent 700b1d9e44
commit 5455a1122d
3 changed files with 3 additions and 16 deletions
-11
View File
@@ -127,17 +127,6 @@ def file_readall_binary(filepath) -> bytes:
return data
def carrier_invoke_style_str(carrier_invoke_style):
carrier_invoke_style = str(carrier_invoke_style)
if carrier_invoke_style == "1":
return "change address of entrypoint"
elif carrier_invoke_style == "2":
return "hijack branching instruction in entrypoint"
else:
return "Invalid: {}".format(carrier_invoke_style)
def file_to_lf(filename):
with open(filename, 'rb') as f:
data = f.read()
+2 -2
View File
@@ -32,8 +32,8 @@ class PayloadLocation(Enum):
class CarrierInvokeStyle(Enum):
ChangeEntryPoint = "change AddressOfEntryPoint"
BackdoorCallInstr = "hijack branching instruction in entrypoint"
ChangeEntryPoint = "change EntryPoint"
BackdoorCallInstr = "hijack Main"
class FunctionInvokeStyle(Enum):
+1 -3
View File
@@ -67,9 +67,7 @@ def main():
elif args.carrier_invoke == "backdoor":
settings.carrier_invoke_style = CarrierInvokeStyle.BackdoorCallInstr
else:
logging.error("Invalid carrier_invoke, use one of:")
for i in ["eop", "backdoor"]:
logging.error(" {} {}".format(i, carrier_invoke_style_str(i)))
logging.error("Invalid carrier_invoke, use: eop, backdoor")
return
if not args.shellcode or not args.inject: