Merge pull request sqlancer#1275 from Mohab-Sobhy/feat/mariadb-add-delete-generator

Add DeleteGenerator for MariaDB

Author: mrigger

src/sqlancer/mariadb/MariaDBProvider.java

@@ -19,6 +19,7 @@
 import sqlancer.common.DBMSCommon;
 import sqlancer.common.query.SQLQueryAdapter;
 import sqlancer.mariadb.MariaDBProvider.MariaDBGlobalState;
+import sqlancer.mariadb.gen.MariaDBDeleteGenerator;
 import sqlancer.mariadb.gen.MariaDBIndexGenerator;
 import sqlancer.mariadb.gen.MariaDBInsertGenerator;
 import sqlancer.mariadb.gen.MariaDBSetGenerator;
@@ -47,6 +48,7 @@ enum Action {
         SET, //
         TRUNCATE, //
         UPDATE, //
+        DELETE,
     }
 
     @Override
@@ -77,6 +79,9 @@ public void generateDatabase(MariaDBGlobalState globalState) throws Exception {
             case CREATE_INDEX:
                 nrPerformed = globalState.getRandomly().getInteger(0, 2);
                 break;
+            case DELETE:
+                nrPerformed = globalState.getRandomly().getInteger(0, 2);
+                break;
             case SET:
                 nrPerformed = 20;
                 break;
@@ -140,6 +145,9 @@ public void generateDatabase(MariaDBGlobalState globalState) throws Exception {
                 case SET:
                     query = MariaDBSetGenerator.set(globalState.getRandomly(), options);
                     break;
+                case DELETE:
+                    query = MariaDBDeleteGenerator.delete(globalState.getSchema(), globalState.getRandomly());
+                    break;
                 default:
                     throw new AssertionError(nextAction);
                 }

src/sqlancer/mariadb/gen/MariaDBDeleteGenerator.java

@@ -0,0 +1,93 @@
+package sqlancer.mariadb.gen;
+
+import java.util.Collections;
+
+import sqlancer.Randomly;
+import sqlancer.common.query.ExpectedErrors;
+import sqlancer.common.query.SQLQueryAdapter;
+import sqlancer.common.schema.AbstractTables;
+import sqlancer.mariadb.MariaDBSchema;
+import sqlancer.mariadb.MariaDBSchema.MariaDBColumn;
+import sqlancer.mariadb.MariaDBSchema.MariaDBTable;
+import sqlancer.mariadb.ast.MariaDBVisitor;
+
+public final class MariaDBDeleteGenerator {
+
+    private MariaDBDeleteGenerator() {
+    }
+
+    public static SQLQueryAdapter delete(MariaDBSchema schema, Randomly r) {
+        MariaDBTable table = schema.getRandomTable();
+
+        MariaDBExpressionGenerator expressionGenerator = new MariaDBExpressionGenerator(r);
+
+        AbstractTables<MariaDBTable, MariaDBColumn> tablesAndColumns = new AbstractTables<>(
+                Collections.singletonList(table));
+        expressionGenerator.setTablesAndColumns(tablesAndColumns);
+
+        ExpectedErrors errors = new ExpectedErrors();
+
+        errors.add("foreign key constraint fails");
+        errors.add("cannot delete or update a parent row");
+        errors.add("Data truncated");
+        errors.add("Division by 0");
+        errors.add("Incorrect value");
+
+        StringBuilder sb = new StringBuilder("DELETE");
+
+        if (Randomly.getBooleanWithRatherLowProbability()) {
+            sb.append(" LOW_PRIORITY");
+        }
+        if (Randomly.getBooleanWithRatherLowProbability()) {
+            sb.append(" QUICK");
+        }
+        if (Randomly.getBooleanWithRatherLowProbability()) {
+            sb.append(" IGNORE");
+        }
+
+        sb.append(" FROM ");
+        sb.append(table.getName());
+
+        if (Randomly.getBoolean()) {
+            sb.append(" WHERE ");
+            if (Randomly.getBooleanWithRatherLowProbability()) {
+                sb.append(MariaDBVisitor.asString(MariaDBExpressionGenerator.getRandomConstant(r)));
+            } else {
+                sb.append(MariaDBVisitor.asString(expressionGenerator.getRandomExpression()));
+            }
+        }
+
+        // ORDER BY + LIMIT
+        if (Randomly.getBooleanWithRatherLowProbability() && !table.getColumns().isEmpty()) {
+            sb.append(" ORDER BY ");
+            sb.append(Randomly.fromList(table.getColumns()).getName());
+            if (Randomly.getBoolean()) {
+                sb.append(Randomly.getBoolean() ? " ASC" : " DESC");
+            }
+        }
+
+        if (Randomly.getBooleanWithRatherLowProbability()) {
+            sb.append(" LIMIT ");
+            sb.append(Randomly.getNotCachedInteger(1, 10));
+        }
+
+        // RETURNING clause (MariaDB >= 10.5)
+        if (Randomly.getBooleanWithRatherLowProbability()) {
+            sb.append(" RETURNING ");
+            if (Randomly.getBooleanWithRatherLowProbability()) {
+                sb.append(MariaDBVisitor.asString(MariaDBExpressionGenerator.getRandomConstant(r)));
+            } else {
+                sb.append(MariaDBVisitor.asString(expressionGenerator.getRandomExpression()));
+            }
+        }
+
+        String query = sb.toString();
+        if (query.contains("RLIKE") || query.contains("REGEXP")) {
+            errors.add("Regex error");
+            errors.add("quantifier does not follow a repeatable item");
+            errors.add("Got error");
+        }
+
+        return new SQLQueryAdapter(query, errors);
+    }
+}