DuProcess
101 lines
3.4 KiB
Bash
Executable File
101 lines
3.4 KiB
Bash
Executable File
#!/usr/bin/env sh
|
|
set -eu
|
|
|
|
cargo build --release
|
|
|
|
workdir="$(mktemp -d)"
|
|
cleanup() {
|
|
if [ -n "${container_id:-}" ]; then
|
|
docker rm -f "$container_id" >/dev/null 2>&1 || true
|
|
fi
|
|
rm -rf "$workdir"
|
|
}
|
|
trap cleanup EXIT INT TERM
|
|
|
|
ssh-keygen -t ed25519 -N "" -f "$workdir/id_ed25519" >/dev/null
|
|
cp target/release/dosh-server "$workdir/dosh-server"
|
|
cp target/release/dosh-auth "$workdir/dosh-auth"
|
|
cp "$workdir/id_ed25519.pub" "$workdir/bench_authorized_keys"
|
|
|
|
docker build -q -t dosh-ssh-bench -f packaging/docker/ssh-bench.Dockerfile "$workdir" >/dev/null
|
|
mosh_port="${DOSH_BENCH_MOSH_PORT:-60000}"
|
|
container_id="$(docker run -d -p 127.0.0.1:0:22/tcp -p 127.0.0.1:0:50000/udp -p 127.0.0.1:${mosh_port}:60000/udp dosh-ssh-bench)"
|
|
ssh_port="$(docker port "$container_id" 22/tcp | sed 's/.*://')"
|
|
dosh_port="$(docker port "$container_id" 50000/udp | sed 's/.*://')"
|
|
|
|
for _ in 1 2 3 4 5; do
|
|
if ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -i "$workdir/id_ed25519" -p "$ssh_port" bench@127.0.0.1 true >/dev/null 2>&1; then
|
|
break
|
|
fi
|
|
sleep 1
|
|
done
|
|
|
|
ssh-keyscan -p "$ssh_port" 127.0.0.1 > "$workdir/known_hosts"
|
|
mkdir -p "$workdir/home" "$workdir/home-controlmaster" "$workdir/home-cached" "$workdir/home-mosh"
|
|
|
|
if ! HOME="$workdir/home" target/release/dosh-bench \
|
|
--server bench@127.0.0.1 \
|
|
--ssh-port "$ssh_port" \
|
|
--dosh-port "$dosh_port" \
|
|
--ssh-key "$workdir/id_ed25519" \
|
|
--ssh-known-hosts "$workdir/known_hosts" \
|
|
--ssh-auth-command /usr/local/bin/dosh-auth \
|
|
--iterations 3 \
|
|
--no-cache \
|
|
--assert-ssh-plus-ms 500; then
|
|
docker logs "$container_id" || true
|
|
docker exec "$container_id" sh -lc 'cat /tmp/dosh-server.log 2>/dev/null || true' || true
|
|
exit 1
|
|
fi
|
|
|
|
if ! HOME="$workdir/home-controlmaster" target/release/dosh-bench \
|
|
--server bench@127.0.0.1 \
|
|
--ssh-port "$ssh_port" \
|
|
--dosh-port "$dosh_port" \
|
|
--ssh-key "$workdir/id_ed25519" \
|
|
--ssh-known-hosts "$workdir/known_hosts" \
|
|
--ssh-auth-command /usr/local/bin/dosh-auth \
|
|
--iterations 3 \
|
|
--no-cache \
|
|
--controlmaster \
|
|
--assert-ssh-plus-ms 500; then
|
|
docker logs "$container_id" || true
|
|
docker exec "$container_id" sh -lc 'cat /tmp/dosh-server.log 2>/dev/null || true' || true
|
|
exit 1
|
|
fi
|
|
|
|
if ! HOME="$workdir/home-cached" target/release/dosh-bench \
|
|
--server bench@127.0.0.1 \
|
|
--ssh-port "$ssh_port" \
|
|
--dosh-port "$dosh_port" \
|
|
--ssh-key "$workdir/id_ed25519" \
|
|
--ssh-known-hosts "$workdir/known_hosts" \
|
|
--ssh-auth-command /usr/local/bin/dosh-auth \
|
|
--iterations 10 \
|
|
--warm-cache \
|
|
--skip-ssh-baseline \
|
|
--assert-dosh-max-ms "${DOSH_BENCH_CACHED_MAX_MS:-25}"; then
|
|
docker logs "$container_id" || true
|
|
docker exec "$container_id" sh -lc 'cat /tmp/dosh-server.log 2>/dev/null || true' || true
|
|
exit 1
|
|
fi
|
|
|
|
if [ "${DOSH_BENCH_INCLUDE_MOSH:-0}" = "1" ]; then
|
|
if ! HOME="$workdir/home-mosh" target/release/dosh-bench \
|
|
--server bench@127.0.0.1 \
|
|
--ssh-port "$ssh_port" \
|
|
--dosh-port "$dosh_port" \
|
|
--ssh-key "$workdir/id_ed25519" \
|
|
--ssh-known-hosts "$workdir/known_hosts" \
|
|
--ssh-auth-command /usr/local/bin/dosh-auth \
|
|
--iterations 3 \
|
|
--no-cache \
|
|
--include-mosh \
|
|
--mosh-server-command /usr/bin/mosh-server \
|
|
--mosh-port "$mosh_port"; then
|
|
docker logs "$container_id" || true
|
|
docker exec "$container_id" sh -lc 'cat /tmp/dosh-server.log 2>/dev/null || true' || true
|
|
exit 1
|
|
fi
|
|
fi
|