fix(fuzzer,runner): add quotes around inserted values in DB queries

rayanht · May 2, 2022 · e3e9728 · e3e9728
1 parent 7b4efcd
commit e3e9728
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 14 deletions.
diff --git a/amber_server.py b/amber_server.py
@@ -69,18 +69,18 @@ def insert_BQ_entry(
     insert_query = f"""
     INSERT INTO `spirvsmith.spirv.shader_data`
     VALUES (
-        {shader_id},
+        "{shader_id}",
         {original_entry.shader_rev},
-        {original_entry.generator_id},
-        {original_entry.generator_version},
+        "{original_entry.generator_id}",
+        "{original_entry.generator_version}",
         {original_entry.n_buffers},
-        {buffer_dump},
-        {platform_os},
-        {hardware_type},
-        {hardware_vendor},
-        {hardware_model},
-        {hardware_driver_version},
-        {backend},
+        "{buffer_dump}",
+        "{platform_os}",
+        "{hardware_type}",
+        "{hardware_vendor}",
+        "{hardware_model}",
+        "{hardware_driver_version}",
+        "{backend}",
     )
     """
     delete_query = f"""

diff --git a/src/fuzzing_server.py b/src/fuzzing_server.py
@@ -192,20 +192,21 @@ def broadcast_shader_data(generator_id: ULID, shader: SPIRVShader, monitor: Moni
     insert_query = f"""
         INSERT INTO `spirvsmith.spirv.shader_data`
         VALUES (
-            {shader.id},
+            "{shader.id}",
             1,
-            {generator_id},
-            {shader.context.config.misc.version},
+            "{generator_id}",
+            "{shader.context.config.misc.version}",
             {len(shader.context.get_storage_buffers())},
             NULL,
             NULL,
             NULL,
             NULL,
             NULL,
             NULL,
-            NULL,
+            NULL
         )
     """
+    print(insert_query)
     BQ_client.query(insert_query).result()