mirror of https://github.com/ARMmbed/mbed-os.git
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
import socket
|
|
import string, random
|
|
from time import time
|
|
|
|
from private_settings import SERVER_ADDRESS
|
|
|
|
ECHO_PORT = 7
|
|
|
|
LEN_PACKET = 127
|
|
N_PACKETS = 5000
|
|
TOT_BITS = float(LEN_PACKET * N_PACKETS * 8) * 2
|
|
MEGA = float(1024 * 1024)
|
|
UPDATE_STEP = (N_PACKETS/10)
|
|
|
|
class TCP_EchoClient:
|
|
def __init__(self, host):
|
|
self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
self.s.connect((host, ECHO_PORT))
|
|
self.packet = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(LEN_PACKET))
|
|
|
|
def __packet(self):
|
|
# Comment out the checks when measuring the throughput
|
|
# self.packet = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(LEN_PACKET))
|
|
self.s.send(self.packet)
|
|
data = self.s.recv(LEN_PACKET)
|
|
# assert self.packet == data, "packet error:\n%s\n%s\n" % (self.packet, data)
|
|
|
|
def test(self):
|
|
start = time()
|
|
for i in range(N_PACKETS):
|
|
if (i % UPDATE_STEP) == 0: print '%.2f%%' % ((float(i)/float(N_PACKETS)) * 100.)
|
|
self.__packet()
|
|
t = time() - start
|
|
print 'Throughput: (%.2f)Mbits/s' % ((TOT_BITS / t)/MEGA)
|
|
|
|
def __del__(self):
|
|
self.s.close()
|
|
|
|
while True:
|
|
e = TCP_EchoClient(SERVER_ADDRESS)
|
|
e.test()
|