diff --git a/iat.py b/iat.py index e265da5..d5fa9cb 100644 --- a/iat.py +++ b/iat.py @@ -59,14 +59,14 @@ def link_func_to_dll(func_list): return res def main(): - parser = argparse.ArgumentParser(prog="iat.py", description="Create a patched PE from a binary dump and a traceCFG file.") + parser = argparse.ArgumentParser(prog="iat.py", description="Create a patched PE from a binary dump and a traceCFG file.", formatter_class=argparse.ArgumentDefaultsHelpFormatter) # Input arguments - parser.add_argument("dump", type=str, help="The path to the wave dump file") - parser.add_argument("trace", type=str, help="The path to the traceCFG file") + parser.add_argument("dump", type=str, help="The path to the wave dump file (usually ends with .dump)") + parser.add_argument("trace", type=str, help="The path to the traceCFG file (.json)") # Additional arguments - parser.add_argument("-o", "--output", type=str, help="Specify an output filepath for the patched PE.") + parser.add_argument("-o", "--output", type=str, default="patched.exe", help="Specify an output filepath for the patched PE.") parser.add_argument("-w", "--wave", type=int, help="Specify the wave number for the binary dump (if it can't be inferred from the filename)") parser.add_argument("-v", '--verbose', action='store_true', help="Output additional debug info")