Brian Daniels / mbed-tools

Fork of mbed-tools by Morpheus

Committer:
screamer
Date:
Wed Mar 30 16:23:44 2016 +0000
Revision:
0:4a2e5f0422d6
Initial revision

Who changed what in which revision?

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