add sqlite support (#10)

Author: CelestaLiu

build.py

@@ -55,5 +55,6 @@ def build_db_image(cfg, use_cache, custom=False, dockerfile_path=""):
 def build_environment(cfg, use_cache, custom=False, dockerfile_path=""):
     build_network()
     build_sqlancer_image(force_rebuild=not use_cache)
-    build_db_image(cfg, use_cache, custom, dockerfile_path)
+    if cfg["embedded"] == "no":
+        build_db_image(cfg, use_cache, custom, dockerfile_path)
 

config.json

@@ -1,5 +1,5 @@
 {
-  "dbms_list": ["mysql", "postgres"],
+  "dbms_list": ["mysql", "postgres", "sqlite"],
   "dbms": "mysql",
   "container_name": "mysql-custom",
   "image": "mysql-custom",

mysql/config.json

@@ -1,4 +1,6 @@
 {
+  "embedded": "no",
+  "dbms": "mysql",
   "image": "mysql:8.0",
   "container_name": "mysql-sqlancer",
   "port": 3306,

postgres/config.json

@@ -1,4 +1,6 @@
 {
+    "embedded": "no",
+    "dbms": "postgres",
     "image": "postgres:13",
     "container_name": "postgres-sqlancer",
     "port": 5432,
@@ -8,7 +10,7 @@
     },
     "username": "postgres",
     "password": "12345678",
-    "oracle": "NOREC",
+    "oracle": "NoREC",
     "num_threads": 4,
     "timeout_seconds": 60,
     "init_sql": "CREATE DATABASE test;",

sqlite/__init__.py

@@ -0,0 +1,11 @@
+{
+    "embedded": "yes",
+    "dbms": "sqlite3",
+    "username": "localhost",
+    "password": "localhost",
+    "container_name": "localhost",
+    "oracle": "NoREC",
+    "num_threads": 4,
+    "timeout_seconds": 60
+}
+  

sqlite/config.json

@@ -47,15 +47,15 @@ def main():
                     print(f"[WARNING] Skipping {dbms}, missing config file.")
                     continue
                 cfg = load_json(config_path)
-                build_environment(args.dbms, cfg, use_cache)
-                test_single(dbms, cfg, use_cache)
+                build_environment(cfg, use_cache)
+                test_single(cfg, use_cache)
 
         elif args.dbms:
             if not args.config:
                 parser.error("Single DBMS test requires --config")
             cfg = load_json(args.config)
-            build_environment(args.dbms, cfg, use_cache)
-            test_single(args.dbms, cfg, use_cache)
+            build_environment(cfg, use_cache)
+            test_single(cfg, use_cache)
 
         else:
             parser.error("Must specify either --dbms or --dockerfile")

start.py

@@ -87,11 +87,12 @@ def start_sqlancer_container(dbms, host_container_name, username, password, orac
     ])
 
 
-def test_single(dbms, cfg, use_cache=False):
+def test_single(cfg, use_cache=False):
+    if cfg["embedded"] == "no":
+        start_db_container(cfg["dbms"], cfg)
 
-    start_db_container(dbms, cfg)
     start_sqlancer_container(
-        dbms=dbms,
+        dbms=cfg["dbms"],
         host_container_name=cfg["container_name"],
         username=cfg["username"],
         password=cfg["password"],
@@ -100,7 +101,8 @@ def test_single(dbms, cfg, use_cache=False):
         timeout=cfg["timeout_seconds"]
     )
 
-    remove_container(cfg["container_name"])
+    if cfg["embedded"] == "no":
+        remove_container(cfg["container_name"])
 
 def remove_container(container_name):
     try: