Skip to content

Commit 1412c3a

Browse files
committed
Update _auditwall.py
1 parent 8b1f067 commit 1412c3a

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

codeflash/verification/_auditwall.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -85,10 +85,14 @@ def handle_os_remove(event: str, args: tuple) -> None:
8585

8686

8787
def check_sqlite_connect(event: str, args: tuple) -> None:
88-
if any(pattern in str(arg) for pattern in args_allow_list for arg in args):
88+
if event == "sqlite3.connect":
89+
if any(pattern in str(args[0]) for pattern in args_allow_list):
90+
accept(event, args)
91+
return
92+
elif event == "sqlite3.connect/handle":
8993
accept(event, args)
90-
else:
91-
reject(event, args)
94+
return
95+
reject(event, args)
9296

9397

9498
_SPECIAL_HANDLERS = {

0 commit comments

Comments
 (0)