Code cave size

helviojunior · Apr 12, 2023 · 63bb72b · 63bb72b
1 parent a6cd303
commit 63bb72b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/shellcodetester/args.py b/shellcodetester/args.py
@@ -70,6 +70,14 @@ def _add_custom_args(self, custom):
                             type=str,
                             help=Color.s('List of bad chars to highlight (ex: {G}\\x00\\x0a{W}, default: {G}\\0x00{W})'))
 
+        custom.add_argument('--cave-size',
+                            action='store',
+                            dest='cave_size',
+                            metavar='[size]',
+                            type=int,
+                            default=1024,
+                            help=Color.s('Code cave size (default: {G}1024{W})'))
+
         custom.add_argument('--fill-with-nop',
                             action='store_true',
                             default=False,

diff --git a/shellcodetester/config.py b/shellcodetester/config.py
@@ -119,10 +119,12 @@ def load_from_arguments():
         if args.out_file:
             Configuration.out_file = args.out_file
 
+        Configuration.cave_size = int(args.cave_size)
         Configuration.breakpoint = args.breakpoint
         Configuration.fill = args.fill
         Configuration.transform_format = Transform.parse_format(args.transform_format)
 
+        Logger.pl('     {C}code cave size:{O} %s{W}' % Configuration.cave_size)
         Logger.pl('     {C}transform format:{O} %s{W}' % (Transform.get_name(Configuration.transform_format)))
 
         if len(Configuration.bad_chars) > 0: