EL4121 Embedded System / mbed-os

mbed-os

Dependents:   cobaLCDJoyMotor_Thread odometry_omni_3roda_v3 odometry_omni_3roda_v1 odometry_omni_3roda_v2 ... more

tools/host_tests/udp_link_layer_auto.py@0:b74591d5ab33, 2017-12-11 (annotated)

Committer:
be_bryan
Date:
Mon Dec 11 17:54:04 2017 +0000
Revision:
0:b74591d5ab33
motor ++

Who changed what in which revision?

UserRevisionLine numberNew contents of line
be_bryan 0:b74591d5ab33 1 """
be_bryan 0:b74591d5ab33 2 mbed SDK
be_bryan 0:b74591d5ab33 3 Copyright (c) 2011-2013 ARM Limited
be_bryan 0:b74591d5ab33 4
be_bryan 0:b74591d5ab33 5 Licensed under the Apache License, Version 2.0 (the "License");
be_bryan 0:b74591d5ab33 6 you may not use this file except in compliance with the License.
be_bryan 0:b74591d5ab33 7 You may obtain a copy of the License at
be_bryan 0:b74591d5ab33 8
be_bryan 0:b74591d5ab33 9 http://www.apache.org/licenses/LICENSE-2.0
be_bryan 0:b74591d5ab33 10
be_bryan 0:b74591d5ab33 11 Unless required by applicable law or agreed to in writing, software
be_bryan 0:b74591d5ab33 12 distributed under the License is distributed on an "AS IS" BASIS,
be_bryan 0:b74591d5ab33 13 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
be_bryan 0:b74591d5ab33 14 See the License for the specific language governing permissions and
be_bryan 0:b74591d5ab33 15 limitations under the License.
be_bryan 0:b74591d5ab33 16 """
be_bryan 0:b74591d5ab33 17
be_bryan 0:b74591d5ab33 18 """
be_bryan 0:b74591d5ab33 19 How to use:
be_bryan 0:b74591d5ab33 20 make.py -m LPC1768 -t ARM -d E:\ -n NET_14
be_bryan 0:b74591d5ab33 21 udp_link_layer_auto.py -p COM20 -d E:\ -t 10
be_bryan 0:b74591d5ab33 22 """
be_bryan 0:b74591d5ab33 23
be_bryan 0:b74591d5ab33 24 import re
be_bryan 0:b74591d5ab33 25 import uuid
be_bryan 0:b74591d5ab33 26 import socket
be_bryan 0:b74591d5ab33 27 import thread
be_bryan 0:b74591d5ab33 28 from sys import stdout
be_bryan 0:b74591d5ab33 29 from time import time, sleep
be_bryan 0:b74591d5ab33 30 from host_test import DefaultTest
be_bryan 0:b74591d5ab33 31 from SocketServer import BaseRequestHandler, UDPServer
be_bryan 0:b74591d5ab33 32
be_bryan 0:b74591d5ab33 33
be_bryan 0:b74591d5ab33 34 # Received datagrams (with time)
be_bryan 0:b74591d5ab33 35 dict_udp_recv_datagrams = dict()
be_bryan 0:b74591d5ab33 36
be_bryan 0:b74591d5ab33 37 # Sent datagrams (with time)
be_bryan 0:b74591d5ab33 38 dict_udp_sent_datagrams = dict()
be_bryan 0:b74591d5ab33 39
be_bryan 0:b74591d5ab33 40
be_bryan 0:b74591d5ab33 41 class UDPEchoClient_Handler(BaseRequestHandler):
be_bryan 0:b74591d5ab33 42 def handle(self):
be_bryan 0:b74591d5ab33 43 """ One handle per connection
be_bryan 0:b74591d5ab33 44 """
be_bryan 0:b74591d5ab33 45 _data, _socket = self.request
be_bryan 0:b74591d5ab33 46 # Process received datagram
be_bryan 0:b74591d5ab33 47 data_str = repr(_data)[1:-1]
be_bryan 0:b74591d5ab33 48 dict_udp_recv_datagrams[data_str] = time()
be_bryan 0:b74591d5ab33 49
be_bryan 0:b74591d5ab33 50
be_bryan 0:b74591d5ab33 51 def udp_packet_recv(threadName, server_ip, server_port):
be_bryan 0:b74591d5ab33 52 """ This function will receive packet stream from mbed device
be_bryan 0:b74591d5ab33 53 """
be_bryan 0:b74591d5ab33 54 server = UDPServer((server_ip, server_port), UDPEchoClient_Handler)
be_bryan 0:b74591d5ab33 55 print "[UDP_COUNTER] Listening for connections... %s:%d"% (server_ip, server_port)
be_bryan 0:b74591d5ab33 56 server.serve_forever()
be_bryan 0:b74591d5ab33 57
be_bryan 0:b74591d5ab33 58
be_bryan 0:b74591d5ab33 59 class UDPEchoServerTest(DefaultTest):
be_bryan 0:b74591d5ab33 60 ECHO_SERVER_ADDRESS = "" # UDP IP of datagram bursts
be_bryan 0:b74591d5ab33 61 ECHO_PORT = 0 # UDP port for datagram bursts
be_bryan 0:b74591d5ab33 62 CONTROL_PORT = 23 # TCP port used to get stats from mbed device, e.g. counters
be_bryan 0:b74591d5ab33 63 s = None # Socket
be_bryan 0:b74591d5ab33 64
be_bryan 0:b74591d5ab33 65 TEST_PACKET_COUNT = 1000 # how many packets should be send
be_bryan 0:b74591d5ab33 66 TEST_STRESS_FACTOR = 0.001 # stress factor: 10 ms
be_bryan 0:b74591d5ab33 67 PACKET_SATURATION_RATIO = 29.9 # Acceptable packet transmission in %
be_bryan 0:b74591d5ab33 68
be_bryan 0:b74591d5ab33 69 PATTERN_SERVER_IP = "Server IP Address is (\d+).(\d+).(\d+).(\d+):(\d+)"
be_bryan 0:b74591d5ab33 70 re_detect_server_ip = re.compile(PATTERN_SERVER_IP)
be_bryan 0:b74591d5ab33 71
be_bryan 0:b74591d5ab33 72 def get_control_data(self, command="stat\n"):
be_bryan 0:b74591d5ab33 73 BUFFER_SIZE = 256
be_bryan 0:b74591d5ab33 74 try:
be_bryan 0:b74591d5ab33 75 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
be_bryan 0:b74591d5ab33 76 s.connect((self.ECHO_SERVER_ADDRESS, self.CONTROL_PORT))
be_bryan 0:b74591d5ab33 77 except Exception, e:
be_bryan 0:b74591d5ab33 78 data = None
be_bryan 0:b74591d5ab33 79 s.send(command)
be_bryan 0:b74591d5ab33 80 data = s.recv(BUFFER_SIZE)
be_bryan 0:b74591d5ab33 81 s.close()
be_bryan 0:b74591d5ab33 82 return data
be_bryan 0:b74591d5ab33 83
be_bryan 0:b74591d5ab33 84 def test(self):
be_bryan 0:b74591d5ab33 85 serial_ip_msg = self.mbed.serial_readline()
be_bryan 0:b74591d5ab33 86 if serial_ip_msg is None:
be_bryan 0:b74591d5ab33 87 return self.RESULT_IO_SERIAL
be_bryan 0:b74591d5ab33 88 stdout.write(serial_ip_msg)
be_bryan 0:b74591d5ab33 89 stdout.flush()
be_bryan 0:b74591d5ab33 90 # Searching for IP address and port prompted by server
be_bryan 0:b74591d5ab33 91 m = self.re_detect_server_ip.search(serial_ip_msg)
be_bryan 0:b74591d5ab33 92 if m and len(m.groups()):
be_bryan 0:b74591d5ab33 93 self.ECHO_SERVER_ADDRESS = ".".join(m.groups()[:4])
be_bryan 0:b74591d5ab33 94 self.ECHO_PORT = int(m.groups()[4]) # must be integer for socket.connect method
be_bryan 0:b74591d5ab33 95 self.notify("HOST: UDP Server found at: " + self.ECHO_SERVER_ADDRESS + ":" + str(self.ECHO_PORT))
be_bryan 0:b74591d5ab33 96
be_bryan 0:b74591d5ab33 97 # Open client socket to burst datagrams to UDP server in mbed
be_bryan 0:b74591d5ab33 98 try:
be_bryan 0:b74591d5ab33 99 self.s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
be_bryan 0:b74591d5ab33 100 except Exception, e:
be_bryan 0:b74591d5ab33 101 self.s = None
be_bryan 0:b74591d5ab33 102 self.notify("HOST: Error: %s"% e)
be_bryan 0:b74591d5ab33 103 return self.RESULT_ERROR
be_bryan 0:b74591d5ab33 104
be_bryan 0:b74591d5ab33 105 # UDP replied receiver works in background to get echoed datagrams
be_bryan 0:b74591d5ab33 106 SERVER_IP = str(socket.gethostbyname(socket.getfqdn()))
be_bryan 0:b74591d5ab33 107 SERVER_PORT = self.ECHO_PORT + 1
be_bryan 0:b74591d5ab33 108 thread.start_new_thread(udp_packet_recv, ("Thread-udp-recv", SERVER_IP, SERVER_PORT))
be_bryan 0:b74591d5ab33 109 sleep(0.5)
be_bryan 0:b74591d5ab33 110
be_bryan 0:b74591d5ab33 111 # Burst part
be_bryan 0:b74591d5ab33 112 for no in range(self.TEST_PACKET_COUNT):
be_bryan 0:b74591d5ab33 113 TEST_STRING = str(uuid.uuid4())
be_bryan 0:b74591d5ab33 114 payload = str(no) + "__" + TEST_STRING
be_bryan 0:b74591d5ab33 115 self.s.sendto(payload, (self.ECHO_SERVER_ADDRESS, self.ECHO_PORT))
be_bryan 0:b74591d5ab33 116 dict_udp_sent_datagrams[payload] = time()
be_bryan 0:b74591d5ab33 117 sleep(self.TEST_STRESS_FACTOR)
be_bryan 0:b74591d5ab33 118
be_bryan 0:b74591d5ab33 119 if self.s is not None:
be_bryan 0:b74591d5ab33 120 self.s.close()
be_bryan 0:b74591d5ab33 121
be_bryan 0:b74591d5ab33 122 # Wait 5 seconds for packets to come
be_bryan 0:b74591d5ab33 123 result = True
be_bryan 0:b74591d5ab33 124 self.notify("HOST: Test Summary:")
be_bryan 0:b74591d5ab33 125 for d in range(5):
be_bryan 0:b74591d5ab33 126 sleep(1.0)
be_bryan 0:b74591d5ab33 127 summary_datagram_success = (float(len(dict_udp_recv_datagrams)) / float(self.TEST_PACKET_COUNT)) * 100.0
be_bryan 0:b74591d5ab33 128 self.notify("HOST: Datagrams received after +%d sec: %.3f%% (%d / %d), stress=%.3f ms"% (d,
be_bryan 0:b74591d5ab33 129 summary_datagram_success,
be_bryan 0:b74591d5ab33 130 len(dict_udp_recv_datagrams),
be_bryan 0:b74591d5ab33 131 self.TEST_PACKET_COUNT,
be_bryan 0:b74591d5ab33 132 self.TEST_STRESS_FACTOR))
be_bryan 0:b74591d5ab33 133 result = result and (summary_datagram_success >= self.PACKET_SATURATION_RATIO)
be_bryan 0:b74591d5ab33 134 stdout.flush()
be_bryan 0:b74591d5ab33 135
be_bryan 0:b74591d5ab33 136 # Getting control data from test
be_bryan 0:b74591d5ab33 137 self.notify("...")
be_bryan 0:b74591d5ab33 138 self.notify("HOST: Mbed Summary:")
be_bryan 0:b74591d5ab33 139 mbed_stats = self.get_control_data()
be_bryan 0:b74591d5ab33 140 self.notify(mbed_stats)
be_bryan 0:b74591d5ab33 141 return self.RESULT_SUCCESS if result else self.RESULT_FAILURE
be_bryan 0:b74591d5ab33 142
be_bryan 0:b74591d5ab33 143
be_bryan 0:b74591d5ab33 144 if __name__ == '__main__':
be_bryan 0:b74591d5ab33 145 UDPEchoServerTest().run()