Fix loading bin files without an absolute path

radareorg · Sep 10, 2024 · 4843d1a · 4843d1a
1 parent a7ed802
commit 4843d1a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -14,7 +14,7 @@ def is_valid_file(parser, arg):
 
 def main():
     parser = argparse.ArgumentParser()
-    parser.add_argument("bin", nargs="?", type=str)
+    parser.add_argument("bin", nargs="?", type=lambda x: os.path.join(os.environ["R2AI_DIR"], x))
     parser.add_argument("-w", "--webserver", action="store_true",
         help="Start the r2ai webserver. Same as r2ai -c=-w")
     parser.add_argument("-c", "--command", action="append",

diff --git a/r2ai.sh b/r2ai.sh
@@ -2,6 +2,7 @@
 unset DYLD_LIBRARY_PATH
 unset LD_LIBRARY_PATH
 PYTHON=python3
+R2AI_DIR="$(pwd)"
 if [ -h "$0" ]; then
 	F=`readlink $0`
 else
@@ -17,4 +18,4 @@ else
 	. venv/bin/activate
 fi
 # export PYTHONPATH=$PWD
-$PYTHON $D/main.py $@
+env R2AI_DIR="$R2AI_DIR" $PYTHON $D/main.py $@