Jim Flynn / Mbed OS wifi_Example

wifi test

Dependencies:   X_NUCLEO_IKS01A2 mbed-http

easy-connect/wizfi310-driver/TESTS/net/host_tests/udp_echo.py@0:24d3eb812fd4, 2018-09-05 (annotated)

Committer:
JMF
Date:
Wed Sep 05 14:28:24 2018 +0000
Revision:
0:24d3eb812fd4
Initial commit

Who changed what in which revision?

UserRevisionLine numberNew contents of line
JMF 0:24d3eb812fd4 1 """
JMF 0:24d3eb812fd4 2 mbed SDK
JMF 0:24d3eb812fd4 3 Copyright (c) 2011-2013 ARM Limited
JMF 0:24d3eb812fd4 4
JMF 0:24d3eb812fd4 5 Licensed under the Apache License, Version 2.0 (the "License");
JMF 0:24d3eb812fd4 6 you may not use this file except in compliance with the License.
JMF 0:24d3eb812fd4 7 You may obtain a copy of the License at
JMF 0:24d3eb812fd4 8
JMF 0:24d3eb812fd4 9 http://www.apache.org/licenses/LICENSE-2.0
JMF 0:24d3eb812fd4 10
JMF 0:24d3eb812fd4 11 Unless required by applicable law or agreed to in writing, software
JMF 0:24d3eb812fd4 12 distributed under the License is distributed on an "AS IS" BASIS,
JMF 0:24d3eb812fd4 13 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
JMF 0:24d3eb812fd4 14 See the License for the specific language governing permissions and
JMF 0:24d3eb812fd4 15 limitations under the License.
JMF 0:24d3eb812fd4 16 """
JMF 0:24d3eb812fd4 17
JMF 0:24d3eb812fd4 18 import sys
JMF 0:24d3eb812fd4 19 import socket
JMF 0:24d3eb812fd4 20 from sys import stdout
JMF 0:24d3eb812fd4 21 from threading import Thread
JMF 0:24d3eb812fd4 22 from SocketServer import BaseRequestHandler, UDPServer
JMF 0:24d3eb812fd4 23 from mbed_host_tests import BaseHostTest, event_callback
JMF 0:24d3eb812fd4 24
JMF 0:24d3eb812fd4 25
JMF 0:24d3eb812fd4 26 class UDPEchoClientHandler(BaseRequestHandler):
JMF 0:24d3eb812fd4 27 def handle(self):
JMF 0:24d3eb812fd4 28 """ UDP packet handler. Echoes data back to sender's address.
JMF 0:24d3eb812fd4 29 """
JMF 0:24d3eb812fd4 30 data, sock = self.request
JMF 0:24d3eb812fd4 31 sock.sendto(data, self.client_address)
JMF 0:24d3eb812fd4 32
JMF 0:24d3eb812fd4 33
JMF 0:24d3eb812fd4 34 class UDPEchoClientTest(BaseHostTest):
JMF 0:24d3eb812fd4 35
JMF 0:24d3eb812fd4 36 def __init__(self):
JMF 0:24d3eb812fd4 37 """
JMF 0:24d3eb812fd4 38 Initialise test parameters.
JMF 0:24d3eb812fd4 39
JMF 0:24d3eb812fd4 40 :return:
JMF 0:24d3eb812fd4 41 """
JMF 0:24d3eb812fd4 42 BaseHostTest.__init__(self)
JMF 0:24d3eb812fd4 43 self.SERVER_IP = None # Will be determined after knowing the target IP
JMF 0:24d3eb812fd4 44 self.SERVER_PORT = 0 # Let TCPServer choose an arbitrary port
JMF 0:24d3eb812fd4 45 self.server = None
JMF 0:24d3eb812fd4 46 self.server_thread = None
JMF 0:24d3eb812fd4 47 self.target_ip = None
JMF 0:24d3eb812fd4 48
JMF 0:24d3eb812fd4 49 @staticmethod
JMF 0:24d3eb812fd4 50 def find_interface_to_target_addr(target_ip):
JMF 0:24d3eb812fd4 51 """
JMF 0:24d3eb812fd4 52 Finds IP address of the interface through which it is connected to the target.
JMF 0:24d3eb812fd4 53
JMF 0:24d3eb812fd4 54 :return:
JMF 0:24d3eb812fd4 55 """
JMF 0:24d3eb812fd4 56 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
JMF 0:24d3eb812fd4 57 try:
JMF 0:24d3eb812fd4 58 s.connect((target_ip, 0)) # Target IP, any port
JMF 0:24d3eb812fd4 59 except socket.error:
JMF 0:24d3eb812fd4 60 s.connect((target_ip, 8000)) # Target IP, 'random' port
JMF 0:24d3eb812fd4 61 ip = s.getsockname()[0]
JMF 0:24d3eb812fd4 62 s.close()
JMF 0:24d3eb812fd4 63 return ip
JMF 0:24d3eb812fd4 64
JMF 0:24d3eb812fd4 65 def setup_udp_server(self):
JMF 0:24d3eb812fd4 66 """
JMF 0:24d3eb812fd4 67 sets up a UDP server for target to connect and send test data.
JMF 0:24d3eb812fd4 68
JMF 0:24d3eb812fd4 69 :return:
JMF 0:24d3eb812fd4 70 """
JMF 0:24d3eb812fd4 71 # !NOTE: There should mechanism to assert in the host test
JMF 0:24d3eb812fd4 72 if self.SERVER_IP is None:
JMF 0:24d3eb812fd4 73 self.log("setup_udp_server() called before determining server IP!")
JMF 0:24d3eb812fd4 74 self.notify_complete(False)
JMF 0:24d3eb812fd4 75
JMF 0:24d3eb812fd4 76 # Returning none will suppress host test from printing success code
JMF 0:24d3eb812fd4 77 self.server = UDPServer((self.SERVER_IP, self.SERVER_PORT), UDPEchoClientHandler)
JMF 0:24d3eb812fd4 78 ip, port = self.server.server_address
JMF 0:24d3eb812fd4 79 self.SERVER_PORT = port
JMF 0:24d3eb812fd4 80 self.server.allow_reuse_address = True
JMF 0:24d3eb812fd4 81 self.log("HOST: Listening for UDP packets: " + self.SERVER_IP + ":" + str(self.SERVER_PORT))
JMF 0:24d3eb812fd4 82 self.server_thread = Thread(target=UDPEchoClientTest.server_thread_func, args=(self,))
JMF 0:24d3eb812fd4 83 self.server_thread.start()
JMF 0:24d3eb812fd4 84
JMF 0:24d3eb812fd4 85 @staticmethod
JMF 0:24d3eb812fd4 86 def server_thread_func(this):
JMF 0:24d3eb812fd4 87 """
JMF 0:24d3eb812fd4 88 Thread function to run TCP server forever.
JMF 0:24d3eb812fd4 89
JMF 0:24d3eb812fd4 90 :param this:
JMF 0:24d3eb812fd4 91 :return:
JMF 0:24d3eb812fd4 92 """
JMF 0:24d3eb812fd4 93 this.server.serve_forever()
JMF 0:24d3eb812fd4 94
JMF 0:24d3eb812fd4 95 @event_callback("target_ip")
JMF 0:24d3eb812fd4 96 def _callback_target_ip(self, key, value, timestamp):
JMF 0:24d3eb812fd4 97 """
JMF 0:24d3eb812fd4 98 Callback to handle reception of target's IP address.
JMF 0:24d3eb812fd4 99
JMF 0:24d3eb812fd4 100 :param key:
JMF 0:24d3eb812fd4 101 :param value:
JMF 0:24d3eb812fd4 102 :param timestamp:
JMF 0:24d3eb812fd4 103 :return:
JMF 0:24d3eb812fd4 104 """
JMF 0:24d3eb812fd4 105 self.target_ip = value
JMF 0:24d3eb812fd4 106 self.SERVER_IP = self.find_interface_to_target_addr(self.target_ip)
JMF 0:24d3eb812fd4 107 self.setup_udp_server()
JMF 0:24d3eb812fd4 108
JMF 0:24d3eb812fd4 109 @event_callback("host_ip")
JMF 0:24d3eb812fd4 110 def _callback_host_ip(self, key, value, timestamp):
JMF 0:24d3eb812fd4 111 """
JMF 0:24d3eb812fd4 112 Callback for request for host IP Addr
JMF 0:24d3eb812fd4 113
JMF 0:24d3eb812fd4 114 """
JMF 0:24d3eb812fd4 115 self.send_kv("host_ip", self.SERVER_IP)
JMF 0:24d3eb812fd4 116
JMF 0:24d3eb812fd4 117 @event_callback("host_port")
JMF 0:24d3eb812fd4 118 def _callback_host_port(self, key, value, timestamp):
JMF 0:24d3eb812fd4 119 """
JMF 0:24d3eb812fd4 120 Callback for request for host port
JMF 0:24d3eb812fd4 121 """
JMF 0:24d3eb812fd4 122 self.send_kv("host_port", self.SERVER_PORT)
JMF 0:24d3eb812fd4 123
JMF 0:24d3eb812fd4 124 def teardown(self):
JMF 0:24d3eb812fd4 125 if self.server:
JMF 0:24d3eb812fd4 126 self.server.shutdown()
JMF 0:24d3eb812fd4 127 self.server_thread.join()