From 9a0c74b701700d9eee9c25f0b045de29d1d06f7f Mon Sep 17 00:00:00 2001
From: Pin <wf6DJd8a3xSSCZbn@protonmail.com>
Date: Tue, 20 Dec 2022 22:15:44 -0500
Subject: [PATCH] initial commit

---
 .gitignore                     |  3 +++
 client/Dockerfile              | 11 +++++++++++
 client/scripts/setup.sh        | 25 +++++++++++++++++++++++++
 controller/Dockerfile          |  7 +++++++
 controller/scripts/gen_keys.sh | 10 ++++++++++
 controller/scripts/init.sh     | 27 +++++++++++++++++++++++++++
 docker-compose.yml             | 21 +++++++++++++++++++++
 7 files changed, 104 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 client/Dockerfile
 create mode 100755 client/scripts/setup.sh
 create mode 100644 controller/Dockerfile
 create mode 100755 controller/scripts/gen_keys.sh
 create mode 100755 controller/scripts/init.sh
 create mode 100644 docker-compose.yml

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..eddde25
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,3 @@
+.env
+
+keys/
diff --git a/client/Dockerfile b/client/Dockerfile
new file mode 100644
index 0000000..c130430
--- /dev/null
+++ b/client/Dockerfile
@@ -0,0 +1,11 @@
+FROM alpine:3.16
+
+RUN apk --no-cache add openrc openssh bash && \
+    ssh-keygen -A && \
+    mkdir -p /run/openrc && \
+    touch /run/openrc/softlevel
+
+COPY scripts /opt/scripts
+RUN chmod 700 /opt/scripts
+
+ENTRYPOINT ["/opt/scripts/setup.sh"]
diff --git a/client/scripts/setup.sh b/client/scripts/setup.sh
new file mode 100755
index 0000000..22da73d
--- /dev/null
+++ b/client/scripts/setup.sh
@@ -0,0 +1,25 @@
+#!/bin/bash
+
+TEAM_NUM=${TEAM_NUM:=10}
+
+for (( i=1; i<=TEAM_NUM; i++ )); do
+  echo "Creating Team ${i}"
+  adduser -D "team${i}"
+  chmod 750 "/home/team${i}"
+  PASSWORD="TEAM${i}_PASSWORD"
+  if [[ -z "${!PASSWORD}" ]]; then
+    PASSWORD=$(head -c10 </dev/urandom | base64)
+  else
+    PASSWORD=${!PASSWORD}
+  fi
+  echo -e "Password: ${PASSWORD}\n"
+  echo -e "${PASSWORD}\n${PASSWORD}" | passwd "team${i}"
+  unset PASSWORD
+done
+
+if [[ -n "${SSHD_CHALLENGE_DIR}" ]]; then
+  sed -i "s|^AuthorizedKeysFile.*|AuthorizedKeysFile ${SSHD_CHALLENGE_DIR}|" /etc/ssh/sshd_config
+fi
+
+/usr/sbin/sshd -D
+
diff --git a/controller/Dockerfile b/controller/Dockerfile
new file mode 100644
index 0000000..473151f
--- /dev/null
+++ b/controller/Dockerfile
@@ -0,0 +1,7 @@
+FROM alpine:3.16
+
+RUN apk --no-cache add openssh-client bash
+
+COPY scripts /opt/scripts
+
+ENTRYPOINT ["sh", "-c", "/opt/scripts/init.sh"]
diff --git a/controller/scripts/gen_keys.sh b/controller/scripts/gen_keys.sh
new file mode 100755
index 0000000..a656bb8
--- /dev/null
+++ b/controller/scripts/gen_keys.sh
@@ -0,0 +1,10 @@
+#!/bin/bash
+
+TEAM_NUM=${TEAM_NUM:=10}
+
+for (( i=1; i<=TEAM_NUM; i++ )); do
+  if [[ ! -e "${HOME}/.ssh/team-${i}" ]]; then
+    ssh-keygen -q -t ed25519 -N '' -f "${HOME}/.ssh/team-${i}" -C "team${i}"
+  fi
+done
+
diff --git a/controller/scripts/init.sh b/controller/scripts/init.sh
new file mode 100755
index 0000000..3201770
--- /dev/null
+++ b/controller/scripts/init.sh
@@ -0,0 +1,27 @@
+#!/bin/bash
+
+SCORING_POD=${SCORING_POD:=client}
+TEAM_NUM=${TEAM_NUM:=10}
+
+echo "Generating scoring details"
+
+/opt/scripts/gen_keys.sh
+
+echo "Scoring Details"
+
+cat /root/.ssh/*.pub
+
+while true; do
+
+  echo "Testing Scoring"
+
+  for (( i=1; i<=${TEAM_NUM}; i++ )); do
+    echo "Testing Team ${i}"
+    ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no "team${i}@${SCORING_POD}" -i "/root/.ssh/team-${i}" \
+      'echo "flag" >.flag'
+  done
+
+  sleep 30
+
+done
+
diff --git a/docker-compose.yml b/docker-compose.yml
new file mode 100644
index 0000000..335bd22
--- /dev/null
+++ b/docker-compose.yml
@@ -0,0 +1,21 @@
+---
+version: "3.9"
+
+services:
+  client:
+    image: local/c2games-client:latest
+    build:
+      context: ./client
+    environment:
+      TEAM_NUM: 5
+      SSHD_CHALLENGE_DIR: .ssh/auth_keys
+
+  controller:
+    image: local/c2games-controller:latest
+    build:
+      context: ./controller
+    environment:
+      TEAM_NUM: 5
+    volumes:
+      - "${PWD}/keys:/root/.ssh"
+...