add some structure

Dockerfile

@@ -5,9 +5,9 @@ WORKDIR /root
 ENV DEBIAN_FRONTEND=noninteractive
 RUN apt-get update && \
     apt-get install -y build-essential git python3 python3-pip wget
-RUN pip3 install torch torchvision torchaudio
+RUN pip3 install torch torchvision torchaudio lief
 
-RUN mkdir dataset && cd dataset && \
-    wget --no-verbose --no-parent --recursive --reject html,signature --no-check-certificate https://security.ece.cmu.edu/byteweight/elf_64/
+# RUN mkdir dataset && cd dataset && \
+#     wget --no-verbose --no-parent --recursive --reject html,signature --no-check-certificate https://security.ece.cmu.edu/byteweight/elf_64/
 
 ENTRYPOINT ["/bin/bash"]

README.md

@@ -0,0 +1,11 @@
+# Identifying Function Boundaries using CNN's
+
+## Quickstart
+```sh
+git clone [url]
+cd bsidescam23
+./scripts/build.sh
+./scripts/run.sh
+cd app
+python3 src/driver.py
+```

scripts/build.sh

@@ -0,0 +1,3 @@
+#!/usr/bin/env sh
+
+docker build -t mov0xdecafe/fi .

scripts/run.sh

@@ -0,0 +1,3 @@
+#!/usr/bin/env sh
+
+docker run -it -v $PWD/src:/root/app/src mov0xdecafe/fi

src/dataset.py

@@ -0,0 +1,13 @@
+class DatasetGenerator():
+    """Generates the dataset for processing by the CNN"""
+
+    def __init__(self):
+        pass
+
+    def _get_code_elf(self, elf_bin=None):
+        if elf_bin is None:
+            return None
+
+class FunctionIdentificationDataset(torch.utils.data.Dataset):
+    def __init__(self):
+        pass

src/driver.py

@@ -0,0 +1,4 @@
+#/usr/bin/env python3
+
+if __name__ == "__main__":
+    print("Function Boundary Identification")