From dfffb18de123b2eaa25dfedbdc69a04094043e6f Mon Sep 17 00:00:00 2001
From: Pierre-Olivier Mercier <nemunaire@nemunai.re>
Date: Sat, 19 Jan 2019 08:10:23 +0100
Subject: [PATCH] Add udev rule and scripts used to flash the USB sticks

Original work by Alexis Daviot <alexis.daviot@epita.fr>
---
 udev/1-usb.rules   |   1 +
 udev/README.txt    |   8 ++++
 udev/run.sh        |   2 +
 udev/setup_keys.sh | 109 +++++++++++++++++++++++++++++++++++++++++++++
 4 files changed, 120 insertions(+)
 create mode 100644 udev/1-usb.rules
 create mode 100644 udev/README.txt
 create mode 100755 udev/run.sh
 create mode 100755 udev/setup_keys.sh

diff --git a/udev/1-usb.rules b/udev/1-usb.rules
new file mode 100644
index 00000000..4c3c7cf7
--- /dev/null
+++ b/udev/1-usb.rules
@@ -0,0 +1 @@
+ACTION=="add",SUBSYSTEMS=="usb",ATTR{partition}=="1",ATTRS{idVendor}=="14cd", ATTRS{idProduct}=="2536",SYMLINK+="fickey",RUN+="/root/udev-fic/run.sh"
diff --git a/udev/README.txt b/udev/README.txt
new file mode 100644
index 00000000..55399ded
--- /dev/null
+++ b/udev/README.txt
@@ -0,0 +1,8 @@
+Bienvenue au challenge forensic 2019 de l'EPITA !
+
+Commencez par vous connecter au réseau filaire afin d'obtenir une IP. Vous
+n'aurez pas besoin d'être connecté au WiFi en parallèle.
+
+Rendez-vous ensuite sur https://fic.srs.epita.fr/ pour commencer le challenge.
+
+Bon courage !
diff --git a/udev/run.sh b/udev/run.sh
new file mode 100755
index 00000000..4350c28d
--- /dev/null
+++ b/udev/run.sh
@@ -0,0 +1,2 @@
+#!/bin/sh
+/bin/sh /root/udev-fic/setup_keys.sh 2>&1 $DEVNAME >> /root/udev-fic/log
diff --git a/udev/setup_keys.sh b/udev/setup_keys.sh
new file mode 100755
index 00000000..ef03bcf9
--- /dev/null
+++ b/udev/setup_keys.sh
@@ -0,0 +1,109 @@
+#!/bin/sh
+
+export PATH="/bin:/usr/bin:/sbin:/usr/sbin"
+
+cd $(dirname $0)
+
+DEVICE_PATH=$1
+KEY_BY_TEAM=3
+MOUNT_DIR="/mnt"
+COUNT1_FILE="count1"
+COUNT2_FILE="count2"
+BASE_URL="http://192.168.23.1:8081/api/certs/"
+FILES_TO_COPY="README.txt"
+
+which curl > /dev/null || { echo 'curl required!'; exit 1; }
+which jq > /dev/null || { echo 'jq required!'; exit 1; }
+which fatlabel > /dev/null || { echo 'dosfstools required!'; exit 1; }
+which mkfs.vfat > /dev/null || { echo 'dosfstools required!'; exit 1; }
+
+echo -n "[+] Starting at "
+date
+
+test -e "${COUNT1_FILE}" || echo -n '1' > "${COUNT1_FILE}"
+test -e "${COUNT2_FILE}" || echo -n '1' > "${COUNT2_FILE}"
+
+COUNT=$(cat "${COUNT1_FILE}")
+COUNT2=$(cat "${COUNT2_FILE}")
+
+echo ">>> Doing operations for TEAM ${COUNT}"
+
+ORIG_LABEL=$(fatlabel "${DEVICE_PATH}")
+
+echo "[+] Dumping ${DEVICE_PATH}"
+
+if [ $(echo -n "${ORIG_LABEL}" | cut -c 1-3) = "FIC" ]
+then
+    echo "[-] WARNING, this key has already FIC label: ${ORIG_LABEL}!!"
+    echo "Exiting...."
+    exit 42
+fi
+
+echo -n "Getting cert id: "
+CERT_ID=$(curl -q "${BASE_URL}" | jq -r .[].id | head -n "${COUNT}" | tail -1)
+echo $CERT_ID
+
+LABEL="FIC_"$(echo -n ${CERT_ID})
+echo "format USB key with label: ${LABEL}"
+mkfs.vfat -n "${LABEL}" "${DEVICE_PATH}" > /dev/null
+
+if [ $? -ne 0 ]
+then
+    echo "[-] FORMAT ERROR! Aborting..."
+    exit 42
+fi
+
+echo "${DEVICE_PATH} to ${MOUNT_DIR}"
+mount "${DEVICE_PATH}" "${MOUNT_DIR}"
+if [ $? -ne 0 ]
+then
+    echo "[-] MOUNT ERROR! Aborting..."
+    exit 42
+fi
+
+echo "Copy files:"
+wget -O "/tmp/team-${CERT_ID}.p12" "${BASE_URL}${CERT_ID}"
+FILES_TO_COPY="${FILES_TO_COPY} /tmp/team-${CERT_ID}.p12"
+
+for i in $FILES_TO_COPY
+do
+    cp -v "${i}" "${MOUNT_DIR}"
+done
+sync
+
+echo "Done!"
+echo "Umounting"
+umount "${MOUNT_DIR}"
+
+echo "[+]Verify..."
+mount "${DEVICE_PATH}" "${MOUNT_DIR}"
+
+for i in $FILES_TO_COPY
+do
+    SHA_1=$(sha512sum "${i}"| awk '{ print $1 }')
+    j=$(echo -n "${i}" |sed 's/.*\///g')
+    SHA_2=$(sha512sum "${MOUNT_DIR}/${j}"| awk '{ print $1 }')
+    if [ "${SHA_1}" != "" ] && [ "${SHA_1}" = "${SHA_2}" ]
+    then
+        echo "File \"${j}\" ok!"
+    else
+        echo "[-] -------------------- File \"${j}\" KO!"
+        echo "We CANNOT continue, umouting & exiting :("
+        umount "${MOUNT_DIR}"
+        exit 42
+    fi
+done
+
+echo "[+]Exiting..."
+umount "${MOUNT_DIR}"
+if [ $COUNT2 -ge "${KEY_BY_TEAM}" ]
+then
+    rm $COUNT2_FILE
+    echo -n $(( $COUNT + 1 )) > "${COUNT1_FILE}"
+else
+    echo "${COUNT2}/${KEY_BY_TEAM} keys done"
+    echo -n $(( $COUNT2 + 1 )) > "${COUNT2_FILE}"
+fi
+echo "Done, bye (:"
+echo ""
+echo ""