update

Dockerfile

@@ -6,16 +6,32 @@ WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt
 
-RUN apt-get -y update && apt-get -y install -y python3-pip python-is-python3 git pkg-config libsentencepiece-dev
+RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
+  --mount=type=cache,target=/var/lib/apt,sharing=locked \
+  apt-get -y update && apt-get -y install -y python3-pip python-is-python3 \
+  git pkg-config libsentencepiece-dev nano sudo unzip
 
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+# RUN --mount=type=cache,target=/root/.npm
+RUN --mount=type=cache,target=/root/.cache pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Install Ghidrathon
+
+WORKDIR /tmp/
+
+RUN wget https://github.com/mandiant/Ghidrathon/releases/download/v4.0.0/Ghidrathon-v4.0.0.zip
+RUN unzip Ghidrathon-v4.0.0.zip -d ghidrathon
+RUN python -m pip install -r ghidrathon/requirements.txt
+RUN python ghidrathon/ghidrathon_configure.py /ghidra
+RUN unzip ghidrathon/Ghidrathon-v4.0.0.zip -d /ghidra/Ghidra/Extensions
 
 # Install DIRTY Ghidra
 
 WORKDIR /
 
-RUN git clone https://github.com/edmcman/DIRTY
-RUN pip install --no-cache-dir --upgrade -r /DIRTY/requirements.txt
+#RUN git clone -b main https://github.com/edmcman/DIRTY
+ADD ./DIRTY /DIRTY
+
+RUN --mount=type=cache,target=/root/.cache pip install --no-cache-dir --upgrade -r /DIRTY/requirements.txt
 
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user

main.py

@@ -9,13 +9,12 @@ def new_binary(file):
 
     with tempfile.TemporaryDirectory() as TEMP_DIR, tempfile.TemporaryDirectory() as OUTPUT_DIR:
         shutil.copy2(file.name, TEMP_DIR)
+        subprocess.run(f"ls -lR {TEMP_DIR}", shell=True)
 
         # python3 generate.py --ghidra PATH_TO_GHIDRA_ANALYZEHEADLESS -t NUM_THREADS -n [NUM_FILES|None] -b BINARIES_DIR -o OUTPUT_DIR
         print("Running DIRTY-Ghidra...", file=sys.stderr)
         subprocess.run(f"python /DIRTY/dataset-gen-ghidra/generate.py --verbose --ghidra /ghidra/support/analyzeHeadless -t 1 -b {TEMP_DIR} -o {OUTPUT_DIR}", shell=True)
-        subprocess.run(f"ls -lR {TEMP_DIR}", shell=True)
         subprocess.run(f"ls -lR {OUTPUT_DIR}", shell=True)
-        subprocess.run(f"/ghidra/support/analyzeHeadless -h", shell=True)
 
 
 with gr.Blocks() as demo:
@@ -42,7 +41,6 @@ with gr.Blocks() as demo:
         else:
 
             #fun_data = {42: 2, 43: 3}
-            print("hi")
             new_binary(file)
             progress(0, desc="Decompiling binary...")
             #fun_data = get_all_dis(file.name)