Week4 split dockerfiles

.gitignore

@@ -3,5 +3,5 @@ logs/
 __pycache__/
 **/__pycache__/
 
-Dockerfile
-**/Dockerfile
+
+.env

build.py

@@ -0,0 +1,19 @@
+import os
+import sys
+import subprocess
+
+def build_sqlancer():
+    subprocess.run(["docker", "build", "-t", "sqlancer:latest", "./sqlancer"], check=True)
+
+def build_db(dbms, version):
+    tag = f"{dbms}:{version}"
+    subprocess.run(["docker", "pull", tag], check=True)
+
+if __name__ == "__main__":
+    cmd = sys.argv[1]
+    if cmd == "sqlancer":
+        build_sqlancer()
+    elif cmd == "db":
+        dbms = sys.argv[2]
+        version = sys.argv[4]
+        build_db(dbms, version)

config.json

@@ -1,3 +1,10 @@
 {
-  "dbms_list": ["mysql", "postgres"]
+  "dbms_list": ["mysql", "postgres"],
+  "dbms": "mysql",
+  "version": "custom",
+  "username": "root",
+  "password": "12345678",
+  "num_threads": 4,
+  "timeout_seconds": 60,
+  "oracle": "FUZZER"
 }

mysql/config.json

@@ -1,9 +1,15 @@
 {
-    "version": "8.0",
-    "username": "root",
-    "password": "12345678",
-    "num_threads": 4,
-    "timeout_seconds": 60,
-    "oracle": "FUZZER"
-  }
-
+  "image": "mysql:8.0",
+  "container_name": "mysql-test",
+  "port": 3306,
+  "env": {
+    "MYSQL_ROOT_PASSWORD": "12345678"
+  },
+  "username": "root",
+  "password": "12345678",
+  "oracle": "FUZZER",
+  "num_threads": 4,
+  "timeout_seconds": 60,
+  "init_sql": "CREATE DATABASE test;",
+  "init_sql_command": "docker exec {container_name} mysql -u{username} -p{password} -e \"{sql}\""
+}

postgres/config.json

@@ -1,9 +1,17 @@
 {
-    "version": "12",
+    "image": "postgres:13",
+    "container_name": "postgres-test",
+    "port": 5432,
+    "env": {
+      "POSTGRES_USER": "postgres",
+      "POSTGRES_PASSWORD": "12345678"
+    },
     "username": "postgres",
     "password": "12345678",
-    "num_threads": 2,
-    "timeout_seconds": 120,
-    "oracle": "QUERY_PARTITIONING"
-}
+    "oracle": "NOREC",
+    "num_threads": 4,
+    "timeout_seconds": 60,
+    "init_sql": "CREATE DATABASE test;",
+    "init_sql_command_template": "psql -U {username} -c \"{sql}\""
+  }