From 4792733ce32f7f99521ebd91aa022e5dd6925cff Mon Sep 17 00:00:00 2001
From: Nathan R <nathan@0xdecafe.com>
Date: Thu, 24 Aug 2023 09:11:07 +0100
Subject: [PATCH] add some structure

---
 Dockerfile       |  6 +++---
 README.md        | 11 +++++++++++
 scripts/build.sh |  3 +++
 scripts/run.sh   |  3 +++
 src/dataset.py   | 13 +++++++++++++
 src/driver.py    |  4 ++++
 6 files changed, 37 insertions(+), 3 deletions(-)
 create mode 100644 README.md
 create mode 100755 scripts/build.sh
 create mode 100755 scripts/run.sh
 create mode 100644 src/dataset.py
 create mode 100755 src/driver.py

diff --git a/Dockerfile b/Dockerfile
index 0b1fd8f..4877eed 100644
--- a/Dockerfile
+++ b/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"]
\ No newline at end of file
diff --git a/README.md b/README.md
new file mode 100644
index 0000000..115412a
--- /dev/null
+++ b/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
+```
\ No newline at end of file
diff --git a/scripts/build.sh b/scripts/build.sh
new file mode 100755
index 0000000..fcca30d
--- /dev/null
+++ b/scripts/build.sh
@@ -0,0 +1,3 @@
+#!/usr/bin/env sh
+
+docker build -t mov0xdecafe/fi .
\ No newline at end of file
diff --git a/scripts/run.sh b/scripts/run.sh
new file mode 100755
index 0000000..1d202f7
--- /dev/null
+++ b/scripts/run.sh
@@ -0,0 +1,3 @@
+#!/usr/bin/env sh
+
+docker run -it -v $PWD/src:/root/app/src mov0xdecafe/fi
\ No newline at end of file
diff --git a/src/dataset.py b/src/dataset.py
new file mode 100644
index 0000000..0b88adb
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/src/driver.py b/src/driver.py
new file mode 100755
index 0000000..d92992d
--- /dev/null
+++ b/src/driver.py
@@ -0,0 +1,4 @@
+#/usr/bin/env python3
+
+if __name__ == "__main__":
+    print("Function Boundary Identification")
\ No newline at end of file