Merge remote-tracking branch 'refs/remotes/origin/main'

Author: devdanzin

fusil/process/prepare.py

@@ -35,7 +35,10 @@ def prepareProcess(process):
     except Exception as e:
         print(e)
     # Change the user and group
-    changeUserGroup(config, options)
+    try:
+        changeUserGroup(config, options)
+    except Exception as e:
+        print(e)
 
     try:
         chdir(directory)
@@ -55,7 +58,8 @@ def prepareProcess(process):
 
     # Make sure that the program is executable by the current user
     program = process.current_arguments[0]
-    if not access(program, X_OK):
+    # if not access(program, X_OK):
+    if 0:
         user = getuid()
         user = getpwuid(user).pw_name
         message = "The user %s is not allowed to execute the file %s" % (user, program)
@@ -66,7 +70,8 @@ def prepareProcess(process):
         raise ChildError(message)
 
     # Limit process resources
-    limitResources(process, config, options)
+    if 0:
+        limitResources(process, config, options)
 
 
 def limitResources(process, config, options):

fusil/python/blacklists.py

@@ -232,6 +232,7 @@
     "Lock",
     "RLock",
     "Semaphore",
+    "AtomicInt64"
 }
 METHOD_BLACKLIST = {
     "__class__",

fusil/python/python_source.py

@@ -67,7 +67,7 @@ def __init__(self, project: Project, options: FusilConfig, source_output_path: s
                 path = pack.__path__[0]
                 print(f"Adding package: {package} ({path})")
                 package_walker = all_modules.discover_modules([path], package + ".")
-                self.modules |= set(mod.name for mod in package_walker)
+                self.modules |= set(name for finder, name, ispgk in package_walker)
 
             if self.options.verbose:
                 print(f"\nKnown modules ({len(self.modules)}): {','.join(sorted(self.modules))}")

fusil/python/write_python_code.py

@@ -550,6 +550,8 @@ def _write_main_fuzzing_logic(self) -> None:
                 if not self.module_classes:
                     break
                 class_name = choice(self.module_classes)
+                if class_name in OBJECT_BLACKLIST:
+                    continue
                 try:
                     class_obj = getattr(self.module, class_name)
                 except AttributeError:
@@ -588,7 +590,11 @@ def _fuzz_one_class(self, class_idx: int, class_name_str: str, class_type: type)
         self.write_print_to_stderr(
             0, f'"[{prefix}] Attempting to instantiate class: {class_name_str}"'
         )
-
+        if class_name_str in OBJECT_BLACKLIST:
+            self.write_print_to_stderr(
+            0, f'"[{prefix}] Skipping blacklisted class: {class_name_str}"'
+            )
+            return
         instance_var_name = (
             f"instance_{prefix}_{class_name_str.lower().replace('.', '_')}"  # Unique name
         )
@@ -793,7 +799,7 @@ def _fuzz_generic_object_methods(
         )
         self.write(
             0,
-            f"if callable({current_prefix}_attr_val): {current_prefix}_methods.append(({current_prefix}_attr_name, {current_prefix}_attr_val))",
+            f"if callable({current_prefix}_attr_val) and not {current_prefix}_attr_val.__name__ == 'wait': {current_prefix}_methods.append(({current_prefix}_attr_name, {current_prefix}_attr_val))",
         )
         self.restoreLevel(self.base_level - 1)  # Exit inner try
         self.write(0, f"except Exception: pass")
@@ -824,7 +830,7 @@ def _fuzz_generic_object_methods(
         self.write(0, f"# Conceptual call to generic method fuzzer")
         self.write(
             0,
-            f"callMethod(f'{current_prefix}_gen{{_i_{current_prefix}}}', {target_obj_expr_str}, {current_prefix}_method_name_to_call)",
+            f"if {current_prefix}_method_name_to_call != 'wait': callMethod(f'{current_prefix}_gen{{_i_{current_prefix}}}', {target_obj_expr_str}, {current_prefix}_method_name_to_call)",
         )  # Example simplified call
         self.restoreLevel(self.base_level - 1)  # Exit for loop
         self.restoreLevel(self.base_level - 1)  # Exit if methods

setup.py

@@ -43,6 +43,8 @@
     "fusil.network",
     "fusil.process",
     "fusil.python",
+    "fusil.python.jit",
+    "fusil.python.samples",
 )
 
 SCRIPTS = glob("fuzzers/fusil-*")