use shlex for script path to args

Author: b1llow

scripts/run_ida_script.py

@@ -5,14 +5,10 @@
 author: Willi Ballenthin
 email: [email protected]
 """
-import re
-import sys
-import os.path
-import logging
-
 import argparse
 import logging
 import os.path
+import shlex
 import sys
 
 import idb
@@ -21,17 +17,6 @@
 logger = logging.getLogger(__name__)
 
 
-def script_path_to_args(script_path):
-    # Split based on spaces but preserve words inside double and single quotes
-    regex_extract_quoted = "[^\s\"']+|\"([^\"]*)\"|'([^']*)'"
-    matches = re.finditer(regex_extract_quoted, script_path)
-    # Strip quotes away using groups
-    return [
-        m.group(2) if m.group(2) else m.group(1) if m.group(1) else m.group(0)
-        for m in matches
-    ]
-
-
 def main(argv=None):
     # TODO: do version check for 3.x
 
@@ -44,9 +29,10 @@ def main(argv=None):
     parser.add_argument(
         "script_path",
         type=str,
-        help='Path to script file. \
-                              Command line arguments can be passed using quotes: \
-                              "myscrypt.py arg1 arg2 \\"arg3 arg3\\""',
+        help="""Path to script file.
+                Command line arguments can be passed using quotes:
+                "myscrypt.py arg1 arg2 "arg3 arg3""
+        """,
     )
     parser.add_argument("idbpath", type=str, help="Path to input idb file")
     parser.add_argument(
@@ -83,7 +69,7 @@ def main(argv=None):
 
         hooks = idb.shim.install(db, ScreenEA=screenea)
 
-        script_args = script_path_to_args(args.script_path)
+        script_args = shlex.split(args.script_path)
         # update sys.path to point to directory containing script.
         # so scripts can import .py files in the same directory.
         script_dir = os.path.dirname(script_args[0])