mirror of
https://github.com/m1ngsama/TNT.git
synced 2026-02-08 08:54:05 +00:00
56 lines
1.1 KiB
Bash
Executable file
56 lines
1.1 KiB
Bash
Executable file
#!/bin/sh
|
|
# Stress test for TNT server
|
|
# Usage: ./test_stress.sh [num_clients]
|
|
|
|
PORT=${PORT:-2222}
|
|
CLIENTS=${1:-10}
|
|
DURATION=${2:-30}
|
|
BIN="../tnt"
|
|
|
|
if [ ! -f "$BIN" ]; then
|
|
echo "Error: Binary $BIN not found."
|
|
exit 1
|
|
fi
|
|
|
|
# Detect timeout command
|
|
TIMEOUT_CMD="timeout"
|
|
if command -v gtimeout >/dev/null 2>&1; then
|
|
TIMEOUT_CMD="gtimeout"
|
|
fi
|
|
|
|
echo "Starting TNT server on port $PORT..."
|
|
$BIN -p $PORT &
|
|
SERVER_PID=$!
|
|
sleep 2
|
|
|
|
if ! kill -0 $SERVER_PID 2>/dev/null; then
|
|
echo "Server failed to start"
|
|
exit 1
|
|
fi
|
|
|
|
echo "Spawning $CLIENTS clients for ${DURATION}s..."
|
|
|
|
for i in $(seq 1 $CLIENTS); do
|
|
(
|
|
sleep $((i % 5))
|
|
echo "test user $i" | $TIMEOUT_CMD $DURATION ssh -o StrictHostKeyChecking=no \
|
|
-o UserKnownHostsFile=/dev/null -p $PORT localhost \
|
|
>/dev/null 2>&1
|
|
) &
|
|
done
|
|
|
|
echo "Running stress test..."
|
|
sleep $DURATION
|
|
|
|
echo "Cleaning up..."
|
|
kill $SERVER_PID 2>/dev/null
|
|
wait
|
|
|
|
echo "Stress test complete"
|
|
if ps aux | grep tnt | grep -v grep > /dev/null; then
|
|
echo "WARNING: tnt process still running"
|
|
else
|
|
echo "Server shutdown confirmed."
|
|
fi
|
|
|
|
exit 0
|