From 49f46cd528429c354be0bf85bedf3413c9aee806 Mon Sep 17 00:00:00 2001
From: nom3ad <19239479+nom3ad@users.noreply.github.com>
Date: Mon, 1 Jan 2024 15:51:22 +0530
Subject: [PATCH] Add containers based testbed setup
---
hack/Containerfile | 15 ++++++++++++++
hack/compose.yml | 29 +++++++++++++++++++++++++++
hack/exec-iperf | 15 ++++++++++++++
hack/exec-sshuttle | 29 +++++++++++++++++++++++++++
hack/setup.service | 49 ++++++++++++++++++++++++++++++++++++++++++++++
hack/test-bed-up | 9 +++++++++
6 files changed, 146 insertions(+)
create mode 100644 hack/Containerfile
create mode 100644 hack/compose.yml
create mode 100755 hack/exec-iperf
create mode 100755 hack/exec-sshuttle
create mode 100755 hack/setup.service
create mode 100755 hack/test-bed-up
diff --git a/hack/Containerfile b/hack/Containerfile
new file mode 100644
index 0000000..29e6305
--- /dev/null
+++ b/hack/Containerfile
@@ -0,0 +1,15 @@
+FROM docker.io/linuxserver/openssh-server:latest
+# https://hub.docker.com/r/linuxserver/openssh-server/
+
+RUN apk add --no-cache bash python3 nginx iperf3
+
+# suppress linuxserver.io logo printing
+RUN sed -i '1 a exec &>/dev/null' /etc/s6-overlay/s6-rc.d/init-adduser/run
+
+ENV PUID=1000
+ENV PGID=1000
+ENV PASSWORD_ACCESS=true
+ENV USER_NAME=test
+ENV USER_PASSWORD=test
+ENV LOG_STDOUT=true
+COPY ./setup.service /etc/services.d/setup.service/run
\ No newline at end of file
diff --git a/hack/compose.yml b/hack/compose.yml
new file mode 100644
index 0000000..59fe7d5
--- /dev/null
+++ b/hack/compose.yml
@@ -0,0 +1,29 @@
+name: sshuttle-testbed
+
+services:
+ node-1:
+ image: ghcr.io/sshuttle/sshuttle-testbed
+ container_name: sshuttle-testbed-node-1
+ hostname: node-1
+ ports:
+ - 22001:2222
+ cap_add:
+ - "NET_ADMIN"
+ environment:
+ - IP_ADDRESSES=10.55.1.77/24
+ node-2:
+ image: ghcr.io/sshuttle/sshuttle-testbed
+ container_name: sshuttle-testbed-node-2
+ hostname: node-2
+ ports:
+ - 22002:2222
+ cap_add:
+ - "NET_ADMIN"
+ environment:
+ - IP_ADDRESSES=10.55.2.77/32
+
+networks:
+ default:
+ driver: bridge
+ enable_ipv6: true
+ internal: true
\ No newline at end of file
diff --git a/hack/exec-iperf b/hack/exec-iperf
new file mode 100755
index 0000000..14b47d7
--- /dev/null
+++ b/hack/exec-iperf
@@ -0,0 +1,15 @@
+#!/usr/bin/env bash
+set -e
+
+
+node=$1
+
+if [[ ! $node =~ [1-9]+ ]]; then
+ echo "node argument missing. should be '1' , '2' etc"
+ exit 2
+fi
+shift
+
+ip="10.55.$node.77"
+
+exec iperf3 --client "$ip" --port 5001
diff --git a/hack/exec-sshuttle b/hack/exec-sshuttle
new file mode 100755
index 0000000..5d5f692
--- /dev/null
+++ b/hack/exec-sshuttle
@@ -0,0 +1,29 @@
+#!/usr/bin/env bash
+set -e
+
+node=$1
+
+if [[ ! $node =~ [1-9]+ ]]; then
+ echo "node argument missing. should be '1' , '2' etc"
+ exit 2
+fi
+shift
+
+port="2200$node"
+subnet_args="-N"
+host=localhost
+user="test:test"
+
+if ! command -v sshpass >/dev/null; then
+ echo "sshpass is not found. You have to manually enter ssh password: 'test'" >&2
+ user="test"
+fi
+ssh_cmd='ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null'
+
+if [[ "$*" =~ -v ]]; then
+ ssh_cmd+='-v'
+fi
+
+SSHUTTLE_BIN=${SSHUTTLE_BIN:-"$(dirname "$0")/../run"}
+set -x
+exec "${SSHUTTLE_BIN}" -r "$user@$host:$port" --ssh-cmd "$ssh_cmd" "$@" $subnet_args
diff --git a/hack/setup.service b/hack/setup.service
new file mode 100755
index 0000000..95e55c5
--- /dev/null
+++ b/hack/setup.service
@@ -0,0 +1,49 @@
+#!/usr/bin/with-contenv bash
+# shellcheck shell=bash
+
+set -e
+
+echo -e ">>> Setting up $(hostname) | id: $(id) | $(python --version) \nip: $(ip a)\n route: $(ip r)"
+
+iface="$(ip route | awk '/default/ { print $5 }')"
+default_gw="$(ip route | awk '/default/ { print $3 }')"
+for addr in ${IP_ADDRESSES//,/ }; do
+ echo ">>> Adding $addr to interface $iface"
+ net_addr=$(ipcalc -n "$addr" | awk -F= '{print $2}')
+ (
+ set -ex
+ ip addr add "$addr" dev "$iface"
+ ip route add "$net_addr" via "$default_gw" dev "$iface" # so that sshuttle -N can discover routes
+ )
+done
+
+echo ">>> Starting iperf3 server"
+iperf3 --server --port 5001 &
+
+mkdir -p /www
+echo -e "Hello from $(hostname)
+
+ip address
+$(ip address)
+ip route
+$(ip route)
+
" >/www/index.html
+echo "
+daemon off;
+worker_processes 1;
+error_log /dev/stdout info;
+events {
+ worker_connections 1024;
+}
+http {
+ include /etc/nginx/mime.types;
+ server {
+ access_log /dev/stdout;
+ listen 8080 default_server;
+ listen [::]:8080 default_server;
+ root /www;
+ }
+}" >/etc/nginx/nginx.conf
+
+echo ">>> Starting nginx"
+exec nginx
diff --git a/hack/test-bed-up b/hack/test-bed-up
new file mode 100755
index 0000000..d23ade4
--- /dev/null
+++ b/hack/test-bed-up
@@ -0,0 +1,9 @@
+#!/usr/bin/env bash
+set -e
+cd "$(dirname "$0")"
+
+# podman build -t ghcr.io/sshuttle/sshuttle-testbed .
+# podman-compose up
+
+docker build -t ghcr.io/sshuttle/sshuttle-testbed -f Containerfile .
+docker compose up
\ No newline at end of file