dirty-ghidra / Dockerfile
ejschwartz's picture
Update model to release/0.2
04ad17c
# From https://huggingface.co/spaces/sayakpaul/demo-docker-gradio/blob/main/Dockerfile
FROM blacktop/ghidra:latest
WORKDIR /code
COPY ./requirements.txt /code/requirements.txt
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 libprotobuf-dev nano sudo unzip
RUN --mount=type=cache,target=/root/.cache pip install --break-system-packages --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 --mount=type=cache,target=/root/.cache pip install --break-system-packages -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 -b development https://github.com/edmcman/DIRTY-ghidra /DIRTY # 11/25/24
RUN --mount=type=cache,target=/root/.cache pip install --break-system-packages --upgrade -r /DIRTY/requirements.txt
# Download the DIRTY model
RUN --mount=type=cache,target=/root/.cache pip install --break-system-packages huggingface_hub[cli]
RUN --mount=type=cache,target=/tmp/model-dl huggingface-cli download --revision release/0.2 --repo-type model ejschwartz/dirty-ghidra --local-dir /tmp/model-dl && cp -R /tmp/model-dl/data1 /DIRTY/dirty/data1
# Set up a new user named "user"
RUN useradd -m -o -u 1000 user
# Switch to the "user" user
USER user
# Set home to the user's home directory
ENV HOME=/home/user \
PATH=/home/user/.local/bin:$PATH
# Set the working directory to the user's home directory
WORKDIR $HOME/app
# Copy the current directory contents into the container at $HOME/app setting the owner to the user
COPY --chown=user . $HOME/app
CMD ["python", "main.py"]