feat: add comprehensive forward handler e2e tests

Add TCP/UDP forward handler e2e tests covering basic forwarding, alias (forward/tcp), sniffing, raw pipe, idleTimeout, multi-node protocol filtering, sniffing bypass (403), stateful UDP, and stateless UDP. New files: - forward_test.go: ForwardSuite with 9 test methods - testdata/forward/: server configs (tcp, udp, stateless, sniffing, bypass, multi-node, idleTimeout) - scripts/tcp_idle_timeout.py, udp_forward_test.py: Python harnesses for in-container idle timeout and UDP echo verification
2026-06-27 13:59:22 +08:00
parent a0427be086
commit 0bb9109c71
10 changed files with 471 additions and 0 deletions
@@ -0,0 +1,53 @@
+import socket
+import sys
+import time
+
+
+def main():
+    host = sys.argv[1] if len(sys.argv) > 1 else "127.0.0.1"
+    port = int(sys.argv[2]) if len(sys.argv) > 2 else 8000
+    idle_timeout = int(sys.argv[3]) if len(sys.argv) > 3 else 3
+
+    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    s.settimeout(10)
+    s.connect((host, port))
+
+    # Send an HTTP GET request — the forward handler pipes us to tcp-echo:5678
+    req = b"GET / HTTP/1.0\r\nHost: tcp-echo\r\n\r\n"
+    s.sendall(req)
+
+    # Read response — should contain "hello-gost" (echo server reply)
+    resp = b""
+    while b"hello-gost" not in resp:
+        chunk = s.recv(4096)
+        if not chunk:
+            break
+        resp += chunk
+
+    if b"hello-gost" not in resp:
+        print(f"FAIL: expected hello-gost in response, got {resp.decode(errors='replace')}")
+        sys.exit(1)
+    print(f"PASS: first request through forward pipe succeeded")
+
+    # Wait longer than idleTimeout
+    wait = idle_timeout + 2
+    print(f"Waiting {wait}s for idle timeout...")
+    time.sleep(wait)
+
+    # Try to send more data — idle timeout should have closed the pipe
+    try:
+        s.sendall(b"GET / HTTP/1.0\r\nHost: tcp-echo\r\n\r\n")
+        data = s.recv(4096)
+        if not data:
+            print("PASS: connection closed after idle timeout (empty recv)")
+            sys.exit(0)
+        # Got data means the pipe is still alive
+        print(f"FAIL: connection still alive after idle timeout, got {data!r}")
+        sys.exit(1)
+    except (socket.timeout, ConnectionResetError, BrokenPipeError, OSError) as e:
+        print(f"PASS: connection closed after idle timeout: {e}")
+        sys.exit(0)
+
+
+if __name__ == "__main__":
+    main()
@@ -0,0 +1,25 @@
+import socket
+import sys
+
+
+def main():
+    host = sys.argv[1] if len(sys.argv) > 1 else "127.0.0.1"
+    port = int(sys.argv[2]) if len(sys.argv) > 2 else 9000
+
+    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    sock.settimeout(5)
+
+    payload = b"hello-gost-udp"
+    sock.sendto(payload, (host, port))
+
+    data, addr = sock.recvfrom(2048)
+    if data == payload:
+        print("PASS: received echo")
+        sys.exit(0)
+    else:
+        print(f"FAIL: expected {payload!r}, got {data!r}")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()