Maxim mbed development library
tools/host_tests/udpecho_server_auto.py@0:0e018d759a2a, 2016-11-08 (annotated)
- Committer:
- switches
- Date:
- Tue Nov 08 18:27:11 2016 +0000
- Revision:
- 0:0e018d759a2a
Initial commit
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
switches | 0:0e018d759a2a | 1 | """ |
switches | 0:0e018d759a2a | 2 | mbed SDK |
switches | 0:0e018d759a2a | 3 | Copyright (c) 2011-2013 ARM Limited |
switches | 0:0e018d759a2a | 4 | |
switches | 0:0e018d759a2a | 5 | Licensed under the Apache License, Version 2.0 (the "License"); |
switches | 0:0e018d759a2a | 6 | you may not use this file except in compliance with the License. |
switches | 0:0e018d759a2a | 7 | You may obtain a copy of the License at |
switches | 0:0e018d759a2a | 8 | |
switches | 0:0e018d759a2a | 9 | http://www.apache.org/licenses/LICENSE-2.0 |
switches | 0:0e018d759a2a | 10 | |
switches | 0:0e018d759a2a | 11 | Unless required by applicable law or agreed to in writing, software |
switches | 0:0e018d759a2a | 12 | distributed under the License is distributed on an "AS IS" BASIS, |
switches | 0:0e018d759a2a | 13 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
switches | 0:0e018d759a2a | 14 | See the License for the specific language governing permissions and |
switches | 0:0e018d759a2a | 15 | limitations under the License. |
switches | 0:0e018d759a2a | 16 | """ |
switches | 0:0e018d759a2a | 17 | |
switches | 0:0e018d759a2a | 18 | import re |
switches | 0:0e018d759a2a | 19 | import sys |
switches | 0:0e018d759a2a | 20 | import uuid |
switches | 0:0e018d759a2a | 21 | from sys import stdout |
switches | 0:0e018d759a2a | 22 | from socket import socket, AF_INET, SOCK_DGRAM |
switches | 0:0e018d759a2a | 23 | |
switches | 0:0e018d759a2a | 24 | class UDPEchoServerTest(): |
switches | 0:0e018d759a2a | 25 | ECHO_SERVER_ADDRESS = "" |
switches | 0:0e018d759a2a | 26 | ECHO_PORT = 0 |
switches | 0:0e018d759a2a | 27 | s = None # Socket |
switches | 0:0e018d759a2a | 28 | |
switches | 0:0e018d759a2a | 29 | PATTERN_SERVER_IP = "Server IP Address is (\d+).(\d+).(\d+).(\d+):(\d+)" |
switches | 0:0e018d759a2a | 30 | re_detect_server_ip = re.compile(PATTERN_SERVER_IP) |
switches | 0:0e018d759a2a | 31 | |
switches | 0:0e018d759a2a | 32 | def test(self, selftest): |
switches | 0:0e018d759a2a | 33 | result = True |
switches | 0:0e018d759a2a | 34 | serial_ip_msg = selftest.mbed.serial_readline() |
switches | 0:0e018d759a2a | 35 | if serial_ip_msg is None: |
switches | 0:0e018d759a2a | 36 | return selftest.RESULT_IO_SERIAL |
switches | 0:0e018d759a2a | 37 | selftest.notify(serial_ip_msg) |
switches | 0:0e018d759a2a | 38 | # Searching for IP address and port prompted by server |
switches | 0:0e018d759a2a | 39 | m = self.re_detect_server_ip.search(serial_ip_msg) |
switches | 0:0e018d759a2a | 40 | if m and len(m.groups()): |
switches | 0:0e018d759a2a | 41 | self.ECHO_SERVER_ADDRESS = ".".join(m.groups()[:4]) |
switches | 0:0e018d759a2a | 42 | self.ECHO_PORT = int(m.groups()[4]) # must be integer for socket.connect method |
switches | 0:0e018d759a2a | 43 | selftest.notify("HOST: UDP Server found at: " + self.ECHO_SERVER_ADDRESS + ":" + str(self.ECHO_PORT)) |
switches | 0:0e018d759a2a | 44 | |
switches | 0:0e018d759a2a | 45 | # We assume this test fails so can't send 'error' message to server |
switches | 0:0e018d759a2a | 46 | try: |
switches | 0:0e018d759a2a | 47 | self.s = socket(AF_INET, SOCK_DGRAM) |
switches | 0:0e018d759a2a | 48 | except Exception, e: |
switches | 0:0e018d759a2a | 49 | self.s = None |
switches | 0:0e018d759a2a | 50 | selftest.notify("HOST: Socket error: %s"% e) |
switches | 0:0e018d759a2a | 51 | return selftest.RESULT_ERROR |
switches | 0:0e018d759a2a | 52 | |
switches | 0:0e018d759a2a | 53 | for i in range(0, 100): |
switches | 0:0e018d759a2a | 54 | TEST_STRING = str(uuid.uuid4()) |
switches | 0:0e018d759a2a | 55 | self.s.sendto(TEST_STRING, (self.ECHO_SERVER_ADDRESS, self.ECHO_PORT)) |
switches | 0:0e018d759a2a | 56 | data = self.s.recv(len(TEST_STRING)) |
switches | 0:0e018d759a2a | 57 | received_str = repr(data)[1:-1] |
switches | 0:0e018d759a2a | 58 | if TEST_STRING != received_str: |
switches | 0:0e018d759a2a | 59 | result = False |
switches | 0:0e018d759a2a | 60 | break |
switches | 0:0e018d759a2a | 61 | sys.stdout.write('.') |
switches | 0:0e018d759a2a | 62 | stdout.flush() |
switches | 0:0e018d759a2a | 63 | else: |
switches | 0:0e018d759a2a | 64 | result = False |
switches | 0:0e018d759a2a | 65 | |
switches | 0:0e018d759a2a | 66 | if self.s is not None: |
switches | 0:0e018d759a2a | 67 | self.s.close() |
switches | 0:0e018d759a2a | 68 | return selftest.RESULT_SUCCESS if result else selftest.RESULT_FAILURE |