summaryrefslogtreecommitdiff
path: root/tests/multi_net/udp_data.py
blob: fd3e00918f5322d949054b55446f486a4b0c3c80 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
# Simple test of a UDP server and client transferring data

import socket

NUM_NEW_SOCKETS = 4
NUM_TRANSFERS = 4
PORT = 8000

# Server
def instance0():
    multitest.globals(IP=multitest.get_network_ip())
    multitest.next()
    for i in range(NUM_NEW_SOCKETS):
        s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        s.bind(socket.getaddrinfo("0.0.0.0", PORT)[0][-1])
        multitest.broadcast("server ready")
        for j in range(NUM_TRANSFERS):
            data, addr = s.recvfrom(1000)
            print(data)
            s.sendto(b"server to client %d %d" % (i, j), addr)
        s.close()


# Client
def instance1():
    multitest.next()
    ai = socket.getaddrinfo(IP, PORT)[0][-1]
    for i in range(NUM_NEW_SOCKETS):
        s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        multitest.wait("server ready")
        for j in range(NUM_TRANSFERS):
            s.sendto(b"client to server %d %d" % (i, j), ai)
            data, addr = s.recvfrom(1000)
            print(data)
        s.close()