1#!/usr/bin/env python3
 2#
 3# SPDX-License-Identifier: GPL-2.0
 4#
 5
 6import sys, os, os.path, getopt
 7import socket, time
 8import subprocess
 9import select
10
11def read(sock, n):
12    buf = b''
13    while len(buf) < n:
14        rem = n - len(buf)
15        try: s = sock.recv(rem)
16        except (socket.error) as e: return b''
17        buf += s
18    return buf
19
20def send(sock, s):
21    total = len(s)
22    count = 0
23    while count < total:
24        try: n = sock.send(s)
25        except (socket.error) as e: n = 0
26        if n == 0:
27            return count;
28        count += n
29    return count
30
31
32SERVER_PORT = 12877
33MAX_PORTS = 2
34
35serverPort = SERVER_PORT
36serverSocket = None
37
38# create passive socket
39serverSocket = socket.socket(socket.AF_INET6, socket.SOCK_STREAM)
40
41try: serverSocket.bind(('localhost', 0))
42except socket.error as msg:
43    print('bind fails: ' + str(msg))
44
45sn = serverSocket.getsockname()
46serverPort = sn[1]
47
48cmdStr = ("./tcp_client.py %d &") % (serverPort)
49os.system(cmdStr)
50
51buf = b''
52n = 0
53while n < 500:
54    buf += b'.'
55    n += 1
56
57serverSocket.listen(MAX_PORTS)
58readList = [serverSocket]
59
60while True:
61    readyRead, readyWrite, inError = \
62        select.select(readList, [], [], 2)
63
64    if len(readyRead) > 0:
65        waitCount = 0
66        for sock in readyRead:
67            if sock == serverSocket:
68                (clientSocket, address) = serverSocket.accept()
69                address = str(address[0])
70                readList.append(clientSocket)
71            else:
72                sock.settimeout(1);
73                s = read(sock, 1000)
74                n = send(sock, buf)
75                sock.close()
76                serverSocket.close()
77                sys.exit(0)
78    else:
79        print('Select timeout!')
80        sys.exit(1)