Update _auditwall.py

Author: KRRT7

codeflash/verification/_auditwall.py

@@ -43,7 +43,6 @@ def check_open(event: str, args: tuple) -> None:
 
 
 def check_msvcrt_open(event: str, args: tuple) -> None:
-    print(args)
     (handle, flags) = args
     if flags & _BLOCKED_OPEN_FLAGS:
         msg = f"codeflash has detected: {event}({', '.join(map(repr, args))})."
@@ -68,12 +67,21 @@ def check_subprocess(event: str, args: tuple) -> None:
     if not inside_module(modules_with_allowed_popen()):
         reject(event, args)
 
+
 def handle_os_remove(event: str, args: tuple) -> None:
-    print("os.remove", args)
-    if any(arg in args_allow_list for arg in args):
-        accept(event, args)
-    else:
-        reject(event, args)
+    filename = args[0]  # first argument is the filename
+    print(f"DEBUG: handle_os_remove called with filename: {filename}")
+    print(f"DEBUG: args_allow_list: {args_allow_list}")
+    print("DEBUG: checking if any pattern in allow list matches filename")
+
+    for pattern in args_allow_list:
+        if pattern in str(filename):
+            print(f"DEBUG: Found matching pattern '{pattern}' in filename")
+            accept(event, args)
+            return
+
+    print("DEBUG: No matching patterns found - rejecting removal")
+    reject(event, args)
 
 
 def check_sqlite_connect(event: str, args: tuple) -> None: