stack-orchestrator/scripts/relay-test-tcp-sport.py

29 lines
707 B
Python
Raw Normal View History

#!/usr/bin/env python3
"""TCP sport 8001 round trip via HTTP HEAD to 1.1.1.1."""
import socket
import sys
PORT = int(sys.argv[1]) if len(sys.argv) > 1 else 8001
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.settimeout(5)
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
s.bind(("0.0.0.0", PORT))
try:
s.connect(("1.1.1.1", 80))
s.sendall(b"HEAD / HTTP/1.0\r\nHost: 1.1.1.1\r\n\r\n")
resp = s.recv(256)
if b"HTTP" in resp:
print("OK HTTP response received")
else:
print(f"OK {len(resp)} bytes (non-HTTP)")
except socket.timeout:
print("TIMEOUT")
sys.exit(1)
except Exception as e:
print(f"ERROR {e}")
sys.exit(1)
finally:
s.close()