Add a basic clickhouse_driver sink

Author: bitterpanda63

aikido_zen/__init__.py

@@ -51,6 +51,8 @@ def protect(mode="daemon"):
     import aikido_zen.sinks.psycopg2
     import aikido_zen.sinks.psycopg
     import aikido_zen.sinks.asyncpg
+    import aikido_zen.sinks.clickhouse_driver
+
     import aikido_zen.sinks.builtins
     import aikido_zen.sinks.os
     import aikido_zen.sinks.shutil

aikido_zen/sinks/clickhouse_driver.py

@@ -0,0 +1,20 @@
+from aikido_zen.helpers.get_argument import get_argument
+from aikido_zen.sinks import before, on_import, patch_function
+from aikido_zen.vulnerabilities import run_vulnerability_scan
+
+
+@before
+def _execute(func, instance, args, kwargs):
+    kind = "sql_injection"
+    op = "clickhouse_driver.Client.execute"
+    query = get_argument(args, kwargs, 0, "query")
+    run_vulnerability_scan(kind, op, args=(query, "clickhouse"))
+
+
+@on_import("clickhouse_driver", package="clickhouse_driver")
+def patch(m):
+    """
+    patching module clickhouse_driver
+    - patches clickhouse_driver.Client.execute
+    """
+    patch_function(m, "Client.execute", _execute)