Ubuntu injector fix; tighten test condition

gruuya · Mar 23, 2021 · 53c2973 · 53c2973
1 parent e323efd
commit 53c2973
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/django_sql_sniffer/injector.py b/django_sql_sniffer/injector.py
@@ -20,9 +20,9 @@ def inject(pid, code_to_inject, verbose=True):
             command = ["lldb", "-p", pid, "-b"] + args
         else:
             args = [
-                "-eval-command='call PyGILState_Ensure()'",
-                f"-eval-command='call PyRun_SimpleString(\"exec(open(\\\"{temp_file.name}\\\").read())\")'",
-                "-eval-command='call PyGILState_Release($1)'"
+                "-eval-command='call (char *) PyGILState_Ensure()'",
+                f"-eval-command='call (void) PyRun_SimpleString(\"exec(open(\\\"{temp_file.name}\\\").read())\")'",
+                "-eval-command='call (void) PyGILState_Release($1)'"
             ]
             command = ["gdb", "-p", pid, "-batch"] + args
 

diff --git a/tests/functional.py b/tests/functional.py
@@ -50,8 +50,7 @@ def dummy_query_executor():
     tp.kill()
 
     # check all queries captured
-    for query in queries:
-        assert query in lt.analyzer._executed_queries
+    assert set(queries) == set(lt.analyzer._executed_queries.keys())
 
     # check that listener shuts down once it detects the target process is not alive
     time.sleep(4)