Demo/sockets/throughput.py

   1 #! /usr/bin/env python
   2
   3 # Test network throughput.
   4 #
   5 # Usage:
   6 # 1) on host_A: throughput -s [port]                    # start a server
   7 # 2) on host_B: throughput -c  count host_A [port]      # start a client
   8 #
   9 # The server will service multiple clients until it is killed.
  10 #
  11 # The client performs one transfer of count*BUFSIZE bytes and
  12 # measures the time it takes (roundtrip!).
  13
  14
  15 import sys, time
  16 from socket import *
  17
  18 MY_PORT = 50000 + 42
  19
  20 BUFSIZE = 1024
  21
  22
  23 def main():
  24     if len(sys.argv) < 2:
  25         usage()
  26     if sys.argv[1] == '-s':
  27         server()
  28     elif sys.argv[1] == '-c':
  29         client()
  30     else:
  31         usage()
  32
  33
  34 def usage():
  35     sys.stdout = sys.stderr
  36     print 'Usage:    (on host_A) throughput -s [port]'
  37     print 'and then: (on host_B) throughput -c count host_A [port]'
  38     sys.exit(2)
  39
  40
  41 def server():
  42     if len(sys.argv) > 2:
  43         port = eval(sys.argv[2])
  44     else:
  45         port = MY_PORT
  46     s = socket(AF_INET, SOCK_STREAM)
  47     s.bind(('', port))
  48     s.listen(1)
  49     print 'Server ready...'
  50     while 1:
  51         conn, (host, remoteport) = s.accept()
  52         while 1:
  53             data = conn.recv(BUFSIZE)
  54             if not data:
  55                 break
  56             del data
  57         conn.send('OK\n')
  58         conn.close()
  59         print 'Done with', host, 'port', remoteport
  60
  61
  62 def client():
  63     if len(sys.argv) < 4:
  64         usage()
  65     count = int(eval(sys.argv[2]))
  66     host = sys.argv[3]
  67     if len(sys.argv) > 4:
  68         port = eval(sys.argv[4])
  69     else:
  70         port = MY_PORT
  71     testdata = 'x' * (BUFSIZE-1) + '\n'
  72     t1 = time.time()
  73     s = socket(AF_INET, SOCK_STREAM)
  74     t2 = time.time()
  75     s.connect((host, port))
  76     t3 = time.time()
  77     i = 0
  78     while i < count:
  79         i = i+1
  80         s.send(testdata)
  81     s.shutdown(1) # Send EOF
  82     t4 = time.time()
  83     data = s.recv(BUFSIZE)
  84     t5 = time.time()
  85     print data
  86     print 'Raw timers:', t1, t2, t3, t4, t5
  87     print 'Intervals:', t2-t1, t3-t2, t4-t3, t5-t4
  88     print 'Total:', t5-t1
  89     print 'Throughput:', round((BUFSIZE*count*0.001) / (t5-t1), 3),
  90     print 'K/sec.'
  91
  92
  93 main()