on ca33633 help sign in
vibebox/src/ssh.sh
179 lines 4.7 kB history raw
1#!/bin/sh
2set -eu
4SSH_USER="__SSH_USER__"
5PROJECT_NAME="__PROJECT_NAME__"
6KEY_PATH="__KEY_PATH__"
8diag() { echo "[vibebox][diag] $*" >&2; }
10# Extract default route facts (no hardcode)
11default_route_line() { ip -4 route show default 2>/dev/null | head -n 1 || true; }
12default_dev() {
13 default_route_line | awk '{for(i=1;i<=NF;i++) if($i=="dev"){print $(i+1); exit}}'
15default_gw() {
16 default_route_line | awk '{for(i=1;i<=NF;i++) if($i=="via"){print $(i+1); exit}}'
19dump_diag() {
20 diag "=== default route ==="
21 default_route_line >&2 || true
23 diag "=== systemctl status ssh ==="
24 systemctl status ssh --no-pager >&2 || true
26 diag "=== journalctl -u ssh (tail) ==="
27 journalctl -u ssh -n 120 --no-pager >&2 || true
29 diag "=== sshd -t ==="
30 sshd -t >&2 || true
32 diag "=== sshd -T (listen/addressfamily) ==="
33 sshd -T 2>/dev/null | egrep -i '^(listenaddress|addressfamily|port|permitrootlogin|allowusers)\b' >&2 || true
35 diag "=== listeners on :22 ==="
36 ss -lntp 2>/dev/null | awk 'NR==1 || $4 ~ /:22$/' >&2 || true
38 diag "=== ip -br addr ==="
39 ip -br addr >&2 || true
41 diag "=== ip route ==="
42 ip route >&2 || true
44 gw="$(default_gw || true)"
45 if [ -n "$gw" ]; then
46 diag "=== ping default gateway ($gw) ==="
47 ping -c1 -W1 "$gw" >/dev/null 2>&1 && diag "ping gw OK" || diag "ping gw FAIL"
48 fi
51# 1) tmpfs mount
52TARGET="/root/${PROJECT_NAME}/.vibebox"
53if [ -d "$TARGET" ] && ! mountpoint -q "$TARGET"; then
54 mount -t tmpfs tmpfs "$TARGET"
55fi
57# 2) user + authorized_keys
58if ! id -u "$SSH_USER" >/dev/null 2>&1; then
59 useradd -m -s /bin/bash -U "$SSH_USER"
60 usermod -aG sudo "$SSH_USER" || true
61fi
63install -d -m 700 -o "$SSH_USER" -g "$SSH_USER" "/home/${SSH_USER}/.ssh"
64install -m 600 -o "$SSH_USER" -g "$SSH_USER" "$KEY_PATH" "/home/${SSH_USER}/.ssh/authorized_keys"
66# Ensure codex/claude are visible in the user's HOME
67USER_HOME="$(getent passwd "$SSH_USER" | cut -d: -f6 2>/dev/null || true)"
68if [ -z "$USER_HOME" ]; then
69 USER_HOME="/home/${SSH_USER}"
70fi
71install -d -m 755 /usr/local/codex /usr/local/claude
72if [ ! -e "${USER_HOME}/.codex" ]; then
73 ln -s /usr/local/codex "${USER_HOME}/.codex"
74fi
75if [ ! -e "${USER_HOME}/.claude" ]; then
76 ln -s /usr/local/claude "${USER_HOME}/.claude"
77fi
78chown -h "${SSH_USER}:${SSH_USER}" "${USER_HOME}/.codex" "${USER_HOME}/.claude" 2>/dev/null || true
80# Install Mise
81MISE_BIN="${USER_HOME}/.local/bin/mise"
82if [ ! -x "$MISE_BIN" ] && ! command -v mise >/dev/null 2>&1; then
83 curl https://mise.run | HOME="$USER_HOME" sh
84fi
85echo 'eval "$(~/.local/bin/mise activate bash)"' >> "${USER_HOME}/.bashrc"
87export PATH="${USER_HOME}/.local/bin:/usr/local/bin:$PATH"
89mkdir -p "${USER_HOME}/.config/mise"
91cat > "${USER_HOME}/.config/mise/config.toml" <<MISE
92 [settings]
93 # Always use the venv created by uv, if available in directory
94 python.uv_venv_auto = true
95 experimental = true
96 idiomatic_version_file_enable_tools = ["rust"]
98 [tools]
99 uv = "0.9.25"
100 node = "24.13.0"
101 "npm:@openai/codex" = "latest"
102 "npm:@anthropic-ai/claude-code" = "latest"
103MISE
105touch "${USER_HOME}/.config/mise/mise.lock"
106if [ -x "$MISE_BIN" ]; then
107 HOME="$USER_HOME" "$MISE_BIN" install
108else
109 HOME="$USER_HOME" mise install
110fi
112# 3) start ssh (don't swallow failures)
113# If ssh is already active, don't force start/restart.
114if ! systemctl is-active --quiet ssh; then
115 if ! systemctl start ssh; then
116 diag "systemctl start ssh failed"
117 dump_diag
118 exit 1
119 fi
120fi
122# 4) obtain stable IPv4 on the default-route interface (wait up to ~30s)
123ip_on_dev() {
124 dev="$1"
125 ip -4 -o addr show dev "$dev" scope global 2>/dev/null \
126 | awk '{print $4}' | cut -d/ -f1 | head -n 1 || true
129ip=""
130dev=""
131gw=""
132t=0
133while [ "$t" -lt 60 ]; do
134 dev="$(default_dev || true)"
135 gw="$(default_gw || true)"
136 if [ -n "$dev" ]; then
137 ip="$(ip_on_dev "$dev")"
138 else
139 ip=""
140 fi
142 if [ -n "$dev" ] && [ -n "$ip" ]; then
143 # optional: if a gateway exists, require it to answer to avoid "ip exists but link dead"
144 if [ -z "$gw" ] || ping -c1 -W1 "$gw" >/dev/null 2>&1; then
145 break
146 fi
147 fi
149 t=$((t+1))
150 sleep 0.5
151done
153if [ -z "$dev" ] || [ -z "$ip" ]; then
154 diag "no stable IPv4 on default route interface"
155 dump_diag
156 exit 1
157fi
159# 5) strong verify: ssh must listen externally (0.0.0.0:22 or $ip:22 or [::]:22)
160listens_ok() {
161 ss -lnt 2>/dev/null \
162 | awk 'NR>1 {print $4}' \
163 | grep -Eq "^(0\.0\.0\.0:22|\\[::\\]:22|${ip}:22)$"
166i=0
167while ! listens_ok && [ "$i" -lt 80 ]; do # ~8s
168 i=$((i+1))
169 sleep 0.1
170done
172if ! listens_ok; then
173 diag "sshd not listening on 0.0.0.0:22 / ${ip}:22"
174 dump_diag
175 exit 1
176fi
178echo VIBEBOX_SSH_READY
179echo "VIBEBOX_IPV4=$ip"