-
Notifications
You must be signed in to change notification settings - Fork 1
/
server.py
62 lines (56 loc) · 2.38 KB
/
server.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
import socket
import argparse
import time
parser = argparse.ArgumentParser(description='Traffic Generetor Python')
parser.add_argument('--type',dest="type", type=str ,help='udp or tcp',default='udp')
parser.add_argument('--add',dest="add", type=str ,help='address of the server',default="0.0.0.0")
parser.add_argument('--port',dest="port",type=int , help='port on the server',default="10002")
args = parser.parse_args()
byteSum = 0
nPackets = 0
latencySum = 0
if args.type == "udp":
try:
sock = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
sock.bind((args.add,args.port ))
start_time = time.time()
while True:
data, addr = sock.recvfrom(10000) # buffer size is max MTU
byteSum += len(data)
latency = time.time() - float(data.decode("utf-8").strip("0"))
latencySum += latency
nPackets +=1
if time.time() - start_time >= 1:
print(time.time() - start_time )
print ( str(byteSum*8/1000)+" kbps, "+str((latency/nPackets) *1000)+" msec, " +str(nPackets)+" packtes")
start_time = time.time()
byteSum = 0
nPackets =0
latencySum=0
except socket.error as err:
print ("socket creation failed with error %s" %(err) )
if args.type == "tcp":
try:
sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
sock.bind((args.add,args.port ))
sock.listen()
conn, addr = sock.accept()
if conn:
print("connected to ",addr)
start_time = time.time()
while True:
data = conn.recv(10000)
if data:
byteSum += len(data)
nPackets +=1
if len(data.decode("utf-8").strip("0")) <= 17:
latency = time.time() - float(data.decode("utf-8").strip("0"))
latencySum += latency
if time.time() - start_time >= 1:
print ( str(byteSum*8/1000)+" kbps, "+str((latency/nPackets) *1000)+" msec, " +str(nPackets)+" packtes")
byteSum = 0
nPackets =0
latencySum=0
start_time = time.time()
except socket.error as err:
print ("socket creation failed with error %s" %(err) )