Update test TCP client.

This commit is contained in:
iabdalkader 2021-02-11 20:04:20 +02:00
parent 6a04514887
commit 3c35ae8360

View File

@ -1,4 +1,4 @@
#!/usr/bin/env python2
#!/usr/bin/env python
# This file is part of the OpenMV project.
#
# Copyright (c) 2013-2019 Ibrahim Abdelkader <iabdalkader@openmv.io>
@ -6,19 +6,35 @@
#
# This work is licensed under the MIT license, see the file LICENSE for details.
#
# An example sockets client.
# An example TCP client for testing WiFi modules.
import time
import select
import socket
ADDR=('192.168.1.101', 8000)
UPLOAD_LEN = 5*1024
DOWNLOAD_LEN = 10*1024
ADDR=('192.168.1.103', 8080)
def recvall(sock, n):
# Helper function to recv n bytes or return None if EOF is hit
data = bytearray()
while len(data) < n:
packet = sock.recv(n - len(data))
if not packet:
return None
data.extend(packet)
return data
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(ADDR)
time.sleep(0.500)
s.send("HelloWorld")
time.sleep(0.500)
print (s.recv(10))
time.sleep(3)
print ("closing")
upload = 0
download = 0
while (True):
s.sendall(b'0' * UPLOAD_LEN)
buf = recvall(s, DOWNLOAD_LEN)
upload += UPLOAD_LEN
download += DOWNLOAD_LEN
print("Upload: %.3f MBytes Download: %.3f MBytes" %(upload/(1024*1024), download/(1024*1024)))
s.close()