v0.7.7 - Prevent sql attacks and rate limiting on subscriptions
This commit is contained in:
63
tests/subscription_limits.sh
Executable file
63
tests/subscription_limits.sh
Executable file
@@ -0,0 +1,63 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Simple test script to verify subscription limit enforcement and rate limiting
|
||||
# This script tests that subscription limits are enforced early
|
||||
|
||||
set -e
|
||||
|
||||
RELAY_URL="ws://127.0.0.1:8888"
|
||||
|
||||
echo "=== Subscription Limit Test ==="
|
||||
echo "[INFO] Testing relay at: $RELAY_URL"
|
||||
echo "[INFO] Note: This test assumes default subscription limits (max 25 per client)"
|
||||
echo ""
|
||||
|
||||
# Test basic connectivity first
|
||||
echo "=== Test 1: Basic Connectivity ==="
|
||||
echo "[INFO] Testing basic WebSocket connection..."
|
||||
|
||||
# Send a simple REQ message
|
||||
response=$(echo '["REQ","basic_test",{}]' | timeout 5 websocat -n1 "$RELAY_URL" 2>/dev/null || echo "TIMEOUT")
|
||||
|
||||
if echo "$response" | grep -q "EOSE\|EVENT\|NOTICE"; then
|
||||
echo "[PASS] Basic connectivity works"
|
||||
else
|
||||
echo "[FAIL] Basic connectivity failed. Response: $response"
|
||||
exit 1
|
||||
fi
|
||||
echo ""
|
||||
|
||||
# Test subscription limits
|
||||
echo "=== Test 2: Subscription Limit Enforcement ==="
|
||||
echo "[INFO] Testing subscription limits by creating multiple subscriptions..."
|
||||
|
||||
success_count=0
|
||||
limit_hit=false
|
||||
|
||||
# Create multiple subscriptions in sequence (each in its own connection)
|
||||
for i in {1..30}; do
|
||||
echo "[INFO] Creating subscription $i..."
|
||||
sub_id="limit_test_$i_$(date +%s%N)"
|
||||
response=$(echo "[\"REQ\",\"$sub_id\",{}]" | timeout 5 websocat -n1 "$RELAY_URL" 2>/dev/null || echo "TIMEOUT")
|
||||
|
||||
if echo "$response" | grep -q "CLOSED.*$sub_id.*exceeded"; then
|
||||
echo "[INFO] Hit subscription limit at subscription $i"
|
||||
limit_hit=true
|
||||
break
|
||||
elif echo "$response" | grep -q "EOSE\|EVENT"; then
|
||||
((success_count++))
|
||||
else
|
||||
echo "[WARN] Unexpected response for subscription $i: $response"
|
||||
fi
|
||||
|
||||
sleep 0.1
|
||||
done
|
||||
|
||||
if [ "$limit_hit" = true ]; then
|
||||
echo "[PASS] Subscription limit enforcement working (limit hit after $success_count subscriptions)"
|
||||
else
|
||||
echo "[WARN] Subscription limit not hit after 30 attempts"
|
||||
fi
|
||||
echo ""
|
||||
|
||||
echo "=== Test Complete ==="
|
||||
Reference in New Issue
Block a user