add TaintedPath in main

garbervetsky · Oct 16, 2020 · 6eafd8b · 6eafd8b
1 parent 960bb16
commit 6eafd8b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/constraintsolving/main.py b/constraintsolving/main.py
@@ -34,10 +34,10 @@ def create_project_list(projectListFile):
 
 parser.add_argument("--project-dir", dest="project_dir", required=True, type=str,
                     help="Directory of the CodeQL database")
-parser.add_argument("--query-type", dest="query_type", required=True, type=str, choices=["Xss", "NoSql", "Sql", "Sel"],
+parser.add_argument("--query-type", dest="query_type", required=True, type=str, choices=["Xss", "NoSql", "Sql", "Sel", "Path"],
                     help="Type of the query to solve")
 parser.add_argument("--query-name", dest="query_name", required=True, type=str,
-                    choices=["NosqlInjectionWorse", "SqlInjectionWorse", "DomBasedXssWorse", "SeldonWorse"],
+                    choices=["NosqlInjectionWorse", "SqlInjectionWorse", "DomBasedXssWorse", "SeldonWorse", "TaintedPathWorse"],
                     help="Name of the query to solve")
 parser.add_argument("--project-list", dest="projectListFile", required=False, type=str, 
                     help="Run all steps on the project passed on this list")