Anders Blomdell / mbed-sdk-tools

Clone of official tools

host_tests/tcpecho_server_auto.py@47:21ae3e5a7128, 2021-02-04 (annotated)

Committer:
Anders Blomdell
Date:
Thu Feb 04 17:17:13 2021 +0100
Revision:
47:21ae3e5a7128
Parent:
43:2a7da56ebd24 
Add a few normpath calls

Who changed what in which revision?

UserRevisionLine numberNew contents of line
screamer 0:66f3b5499f7f 1 """
screamer 0:66f3b5499f7f 2 mbed SDK
screamer 0:66f3b5499f7f 3 Copyright (c) 2011-2013 ARM Limited
screamer 0:66f3b5499f7f 4
screamer 0:66f3b5499f7f 5 Licensed under the Apache License, Version 2.0 (the "License");
screamer 0:66f3b5499f7f 6 you may not use this file except in compliance with the License.
screamer 0:66f3b5499f7f 7 You may obtain a copy of the License at
screamer 0:66f3b5499f7f 8
screamer 0:66f3b5499f7f 9 http://www.apache.org/licenses/LICENSE-2.0
screamer 0:66f3b5499f7f 10
screamer 0:66f3b5499f7f 11 Unless required by applicable law or agreed to in writing, software
screamer 0:66f3b5499f7f 12 distributed under the License is distributed on an "AS IS" BASIS,
screamer 0:66f3b5499f7f 13 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
screamer 0:66f3b5499f7f 14 See the License for the specific language governing permissions and
screamer 0:66f3b5499f7f 15 limitations under the License.
screamer 0:66f3b5499f7f 16 """
theotherjimmy 43:2a7da56ebd24 17 from __future__ import print_function
screamer 0:66f3b5499f7f 18
screamer 0:66f3b5499f7f 19 import re
screamer 0:66f3b5499f7f 20 import sys
screamer 0:66f3b5499f7f 21 import uuid
screamer 0:66f3b5499f7f 22 import socket
screamer 0:66f3b5499f7f 23 from sys import stdout
screamer 0:66f3b5499f7f 24
screamer 0:66f3b5499f7f 25 class TCPEchoServerTest():
screamer 0:66f3b5499f7f 26 ECHO_SERVER_ADDRESS = ""
screamer 0:66f3b5499f7f 27 ECHO_PORT = 0
screamer 0:66f3b5499f7f 28 ECHO_LOOPs = 100
screamer 0:66f3b5499f7f 29 s = None # Socket
screamer 0:66f3b5499f7f 30
screamer 0:66f3b5499f7f 31 PATTERN_SERVER_IP = "Server IP Address is (\d+).(\d+).(\d+).(\d+):(\d+)"
screamer 0:66f3b5499f7f 32 re_detect_server_ip = re.compile(PATTERN_SERVER_IP)
screamer 0:66f3b5499f7f 33
screamer 0:66f3b5499f7f 34 def test(self, selftest):
screamer 0:66f3b5499f7f 35 result = False
screamer 0:66f3b5499f7f 36 c = selftest.mbed.serial_readline()
screamer 0:66f3b5499f7f 37 if c is None:
screamer 0:66f3b5499f7f 38 return selftest.RESULT_IO_SERIAL
screamer 0:66f3b5499f7f 39 selftest.notify(c)
screamer 0:66f3b5499f7f 40
screamer 0:66f3b5499f7f 41 m = self.re_detect_server_ip.search(c)
screamer 0:66f3b5499f7f 42 if m and len(m.groups()):
screamer 0:66f3b5499f7f 43 self.ECHO_SERVER_ADDRESS = ".".join(m.groups()[:4])
screamer 0:66f3b5499f7f 44 self.ECHO_PORT = int(m.groups()[4]) # must be integer for socket.connect method
screamer 0:66f3b5499f7f 45 selftest.notify("HOST: TCP Server found at: " + self.ECHO_SERVER_ADDRESS + ":" + str(self.ECHO_PORT))
screamer 0:66f3b5499f7f 46
screamer 0:66f3b5499f7f 47 # We assume this test fails so can't send 'error' message to server
screamer 0:66f3b5499f7f 48 try:
screamer 0:66f3b5499f7f 49 self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
screamer 0:66f3b5499f7f 50 self.s.connect((self.ECHO_SERVER_ADDRESS, self.ECHO_PORT))
theotherjimmy 43:2a7da56ebd24 51 except Exception as e:
screamer 0:66f3b5499f7f 52 self.s = None
screamer 0:66f3b5499f7f 53 selftest.notify("HOST: Socket error: %s"% e)
screamer 0:66f3b5499f7f 54 return selftest.RESULT_ERROR
screamer 0:66f3b5499f7f 55
theotherjimmy 43:2a7da56ebd24 56 print('HOST: Sending %d echo strings...'% self.ECHO_LOOPs,)
screamer 0:66f3b5499f7f 57 for i in range(0, self.ECHO_LOOPs):
screamer 0:66f3b5499f7f 58 TEST_STRING = str(uuid.uuid4())
screamer 0:66f3b5499f7f 59 try:
screamer 0:66f3b5499f7f 60 self.s.sendall(TEST_STRING)
screamer 0:66f3b5499f7f 61 data = self.s.recv(128)
theotherjimmy 43:2a7da56ebd24 62 except Exception as e:
screamer 0:66f3b5499f7f 63 self.s = None
screamer 0:66f3b5499f7f 64 selftest.notify("HOST: Socket error: %s"% e)
screamer 0:66f3b5499f7f 65 return selftest.RESULT_ERROR
screamer 0:66f3b5499f7f 66
screamer 0:66f3b5499f7f 67 received_str = repr(data)[1:-1]
screamer 0:66f3b5499f7f 68 if TEST_STRING == received_str: # We need to cut not needed single quotes from the string
screamer 0:66f3b5499f7f 69 sys.stdout.write('.')
screamer 0:66f3b5499f7f 70 stdout.flush()
screamer 0:66f3b5499f7f 71 result = True
screamer 0:66f3b5499f7f 72 else:
theotherjimmy 43:2a7da56ebd24 73 print("Expected: ")
theotherjimmy 43:2a7da56ebd24 74 print("'%s'"% TEST_STRING)
theotherjimmy 43:2a7da56ebd24 75 print("received: ")
theotherjimmy 43:2a7da56ebd24 76 print("'%s'"% received_str)
screamer 0:66f3b5499f7f 77 result = False
screamer 0:66f3b5499f7f 78 break
screamer 0:66f3b5499f7f 79
screamer 0:66f3b5499f7f 80 if self.s is not None:
screamer 0:66f3b5499f7f 81 self.s.close()
screamer 0:66f3b5499f7f 82 else:
screamer 0:66f3b5499f7f 83 selftest.notify("HOST: TCP Server not found")
screamer 0:66f3b5499f7f 84 result = False
screamer 0:66f3b5499f7f 85 return selftest.RESULT_SUCCESS if result else selftest.RESULT_FAILURE