throughput.py

#! /usr/bin/env python

# Test network throughput.
#
# Usage:
# 1) on host_A: throughput -s [port] # start a server
# 2) on host_B: throughput -c count host_A [port] # start a client
#
# The server will service multiple clients until it is killed.
#
# The client performs one transfer of count*BUFSIZE bytes and
# measures the time it takes (roundtrip!).


importsys, time
fromsocketimport*

MY_PORT=50000+42

BUFSIZE=1024


defmain():
iflen(sys.argv) <2:
usage()
ifsys.argv[1] =='-s':
server()
elifsys.argv[1] =='-c':
client()
else:
usage()


defusage():
sys.stdout=sys.stderr
print'Usage: (on host_A) throughput -s [port]'
print'and then: (on host_B) throughput -c count host_A [port]'
sys.exit(2)


defserver():
iflen(sys.argv) >2:
port=eval(sys.argv[2])
else:
port=MY_PORT
s=socket(AF_INET, SOCK_STREAM)
s.bind(('', port))
s.listen(1)
print'Server ready...'
while1:
conn, (host, remoteport) =s.accept()
while1:
data=conn.recv(BUFSIZE)
ifnotdata:
break
deldata
conn.send('OK\n')
conn.close()
print'Done with', host, 'port', remoteport


defclient():
iflen(sys.argv) <4:
usage()
count=int(eval(sys.argv[2]))
host=sys.argv[3]
iflen(sys.argv) >4:
port=eval(sys.argv[4])
else:
port=MY_PORT
testdata='x'* (BUFSIZE-1) +'\n'
t1=time.time()
s=socket(AF_INET, SOCK_STREAM)
t2=time.time()
s.connect((host, port))
t3=time.time()
i=0
whilei<count:
i=i+1
s.send(testdata)
s.shutdown(1) # Send EOF
t4=time.time()
data=s.recv(BUFSIZE)
t5=time.time()
printdata
print'Raw timers:', t1, t2, t3, t4, t5
print'Intervals:', t2-t1, t3-t2, t4-t3, t5-t4
print'Total:', t5-t1
print'Throughput:', round((BUFSIZE*count*0.001) / (t5-t1), 3),
print'K/sec.'


main()