From 3c35ae83603dd7548598ab14d6ab34a1d1328a0a Mon Sep 17 00:00:00 2001 From: iabdalkader Date: Thu, 11 Feb 2021 20:04:20 +0200 Subject: [PATCH] Update test TCP client. --- tools/client.py | 34 +++++++++++++++++++++++++--------- 1 file changed, 25 insertions(+), 9 deletions(-) diff --git a/tools/client.py b/tools/client.py index b57fa31a5..9c8032135 100755 --- a/tools/client.py +++ b/tools/client.py @@ -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 @@ -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()