espilon-source/tools/c2/test_udp.py

55 lines
1.6 KiB
Python

#!/usr/bin/env python3
"""Simple UDP test server to debug camera streaming."""
import socket
import sys
HOST = "0.0.0.0"
PORT = 5000
TOKEN = b"Sup3rS3cretT0k3n"
def main():
port = int(sys.argv[1]) if len(sys.argv) > 1 else PORT
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
sock.bind((HOST, port))
print(f"[UDP] Listening on {HOST}:{port}")
print(f"[UDP] Token: {TOKEN.decode()}")
print("[UDP] Waiting for packets...\n")
packet_count = 0
frame_count = 0
try:
while True:
data, addr = sock.recvfrom(65535)
packet_count += 1
# Check token
if data.startswith(TOKEN):
payload = data[len(TOKEN):]
if payload == b"START":
print(f"[{addr[0]}:{addr[1]}] START (new frame)")
elif payload == b"END":
frame_count += 1
print(f"[{addr[0]}:{addr[1]}] END (frame #{frame_count} complete)")
else:
print(f"[{addr[0]}:{addr[1]}] CHUNK: {len(payload)} bytes")
else:
print(f"[{addr[0]}:{addr[1]}] INVALID TOKEN: {data[:20]}...")
# Stats every 100 packets
if packet_count % 100 == 0:
print(f"\n--- Stats: {packet_count} packets, {frame_count} frames ---\n")
except KeyboardInterrupt:
print(f"\n[UDP] Stopped. Total: {packet_count} packets, {frame_count} frames")
finally:
sock.close()
if __name__ == "__main__":
main()