mirror of https://github.com/ARMmbed/mbed-os.git
101 lines
3.3 KiB
Python
101 lines
3.3 KiB
Python
"""
|
|
mbed SDK
|
|
Copyright (c) 2011-2013 ARM Limited
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
"""
|
|
|
|
import re
|
|
import sys
|
|
import uuid
|
|
import socket
|
|
from sys import stdout
|
|
from time import time
|
|
from host_test import DefaultTest
|
|
|
|
class TCPEchoServerTest(DefaultTest):
|
|
ECHO_SERVER_ADDRESS = ""
|
|
ECHO_PORT = 0
|
|
ECHO_LOOPs = 100
|
|
s = None # Socket
|
|
|
|
PATTERN_SERVER_IP = "Server IP Address is (\d+).(\d+).(\d+).(\d+):(\d+)"
|
|
re_detect_server_ip = re.compile(PATTERN_SERVER_IP)
|
|
|
|
def run(self):
|
|
result = False
|
|
c = self.mbed.serial_readline()
|
|
if c is None:
|
|
self.print_result("ioerr_serial")
|
|
return
|
|
print c
|
|
stdout.flush()
|
|
|
|
m = self.re_detect_server_ip.search(c)
|
|
if m and len(m.groups()):
|
|
self.ECHO_SERVER_ADDRESS = ".".join(m.groups()[:4])
|
|
self.ECHO_PORT = int(m.groups()[4]) # must be integer for socket.connect method
|
|
print "HOST: TCP Server found at: " + self.ECHO_SERVER_ADDRESS + ":" + str(self.ECHO_PORT)
|
|
stdout.flush()
|
|
|
|
# We assume this test fails so can't send 'error' message to server
|
|
try:
|
|
self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
self.s.connect((self.ECHO_SERVER_ADDRESS, self.ECHO_PORT))
|
|
except Exception, e:
|
|
print "HOST: Error: %s" % e
|
|
self.print_result('error')
|
|
exit(-1)
|
|
|
|
print 'HOST: Sending %d echo strings...'% self.ECHO_LOOPs,
|
|
for i in range(0, self.ECHO_LOOPs):
|
|
TEST_STRING = str(uuid.uuid4())
|
|
self.s.sendall(TEST_STRING)
|
|
data = self.s.recv(128)
|
|
received_str = repr(data)[1:-1]
|
|
if TEST_STRING == received_str: # We need to cut not needed single quotes from the string
|
|
sys.stdout.write('.')
|
|
stdout.flush()
|
|
result = True
|
|
else:
|
|
print "Expected: "
|
|
print "'%s'"% TEST_STRING
|
|
print "received: "
|
|
print "'%s'"% received_str
|
|
result = False
|
|
break
|
|
self.s.close()
|
|
else:
|
|
print "HOST: TCP Server not found"
|
|
result = False
|
|
|
|
if result:
|
|
self.print_result('success')
|
|
else:
|
|
self.print_result('failure')
|
|
|
|
# Receiving
|
|
try:
|
|
while True:
|
|
c = self.mbed.serial_read(512)
|
|
if c is None:
|
|
self.print_result("ioerr_serial")
|
|
break
|
|
stdout.write(c)
|
|
stdout.flush()
|
|
except KeyboardInterrupt, _:
|
|
print "\n[CTRL+c] exit"
|
|
|
|
if __name__ == '__main__':
|
|
TCPEchoServerTest().run()
|