if __name__ == "__main__":
SIZE = 15 * 1000 * 1000
msg = bytearray(SIZE)
sock_type = argv[1][0:3]
argv[1] = argv[1][6:]
HOST, PORT = argv[1].split(':')
addr = (HOST, int(PORT))
N = int(argv[2])
if sock_type == "udt":
sock = UDTSocket()
sock.UDP_SNDBUF = 512 * 1024
sock.UDP_RCVBUF = 2 * 1024 * 1024
sock.UDT_SNDBUF = (15 * 1000 + 1000) * 1000
sock.UDT_RCVBUF = (15 * 1000 + 1000) * 1000
elif sock_type == "tcp":
sock = socket.socket()
else:
raise Exception("Socket type \"%s\" is not supported" % sock_type)
if argv[1].startswith("0.0.0.0"):
sock.bind(addr)
sock.listen(1)
peer, _ = sock.accept()
start = time.time()
all = 0
if sock_type == "udt":
while all < N * SIZE:
all += peer.recv(msg)
else:
while all < N * SIZE:
from udt4py import UDTSocket, UDTException
if __name__ == "__main__":
SIZE = 15 * 1000 * 1000
msg = bytearray(SIZE)
sock_type = argv[1][0:3]
argv[1] = argv[1][6:]
HOST, PORT = argv[1].split(':')
addr = (HOST, int(PORT))
N = int(argv[2])
if sock_type == "udt":
sock = UDTSocket()
sock.UDP_SNDBUF = 512 * 1024
sock.UDP_RCVBUF = 2 * 1024 * 1024
sock.UDT_SNDBUF = (15 * 1000 + 1000) * 1000
sock.UDT_RCVBUF = (15 * 1000 + 1000) * 1000
elif sock_type == "tcp":
sock = socket.socket()
else:
raise Exception("Socket type \"%s\" is not supported" % sock_type)
if argv[1].startswith("0.0.0.0"):
sock.bind(addr)
sock.listen(1)
peer, _ = sock.accept()
start = time.time()
all = 0
if sock_type == "udt":
while all < N * SIZE:
all += peer.recv(msg)
else:
while all < N * SIZE:
