← Back to branch summary

~malept/ubuntu/lucid/python2.6/dev-dependency-fix

« back to all changes in this revision

Viewing changes to Demo/sockets/throughput.py

  • Committer: Bazaar Package Importer
  • Author(s): Matthias Klose
  • Date: 2009-02-13 12:51:00 UTC
  • Revision ID: james.westby@ubuntu.com-20090213125100-uufgcb9yeqzujpqw
Tags: upstream-2.6.1
Import upstream version 2.6.1

Show diffs side-by-side

added added

removed removed

Lines of Context:
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()