48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
#!/usr/bin/python
|
|
|
|
import socket
|
|
import sys
|
|
|
|
# Check server ip address set
|
|
if len(sys.argv) < 2:
|
|
raise RuntimeError('pass IP address of the server')
|
|
|
|
# Set the server address here like 1.2.3.4
|
|
SERVER_ADDR = sys.argv[1]
|
|
|
|
# These constants should match the server
|
|
BUF_SIZE = 2048
|
|
SERVER_PORT = 4242
|
|
TEST_ITERATIONS = 10
|
|
|
|
# Open socket to the server
|
|
sock = socket.socket()
|
|
addr = (SERVER_ADDR, SERVER_PORT)
|
|
sock.connect(addr)
|
|
|
|
# Repeat test for a number of iterations
|
|
for test_iteration in range(TEST_ITERATIONS):
|
|
|
|
# Read BUF_SIZE bytes from the server
|
|
total_size = BUF_SIZE
|
|
read_buf = b''
|
|
while total_size > 0:
|
|
buf = sock.recv(BUF_SIZE)
|
|
print('read %d bytes from server' % len(buf))
|
|
total_size -= len(buf)
|
|
read_buf += buf
|
|
|
|
# Check size of data received
|
|
if len(read_buf) != BUF_SIZE:
|
|
raise RuntimeError('wrong amount of data read %d', len(read_buf))
|
|
|
|
# Send the data back to the server
|
|
write_len = sock.send(read_buf)
|
|
print('written %d bytes to server' % write_len)
|
|
if write_len != BUF_SIZE:
|
|
raise RuntimeError('wrong amount of data written')
|
|
|
|
# All done
|
|
sock.close()
|
|
print("test completed")
|