Diff: tests/blocking/socket/UdpEchoTest.cpp

Revision:

0:836c9a6383e0

Child:

1:5137ec8f4c45

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/blocking/socket/UdpEchoTest.cpp	Mon Aug 11 11:31:32 2014 +0000
@@ -0,0 +1,153 @@
+/*
+ * Copyright 2014, ACKme Networks
+ * All Rights Reserved.
+ *
+ * This is UNPUBLISHED PROPRIETARY SOURCE CODE of ACKme Networks;
+ * the contents of this file may not be disclosed to third parties, copied
+ * or duplicated in any form, in whole or in part, without the prior
+ * written permission of ACKme Networks.
+ */
+
+
+#include "tests/Tests.h"
+#include "Wiconnect.h"
+
+
+
+
+static WiconnectResult processData(Socket &socket, uint32_t count, uint32_t delay, uint32_t size);
+
+
+
+/*************************************************************************************************/
+WiconnectResult socketUdpClientEchoCommand(int argc, char **argv)
+{
+    WiconnectResult result;
+    Wiconnect *wiconnect = Wiconnect::getInstance();
+    Socket socket(sizeof(testBuffer)/2, testBuffer, sizeof(testBuffer)/2, testBuffer + sizeof(testBuffer)/2);
+    const char *host;
+    uint32_t port;
+    uint32_t delay = 5;
+    uint32_t count = 1000;
+    uint32_t size = 256;
+
+    if(argc < 2)
+    {
+        LOG_ERROR("Must specify host and port");
+        return WICONNECT_BAD_ARG;
+    }
+    host = argv[0];
+
+    if(!StringUtil::strToUint32((const char*)argv[1], &port))
+    {
+        LOG_ERROR("Invalid port");
+        return WICONNECT_BAD_ARG;
+    }
+    if(argc > 2 && !StringUtil::strToUint32((const char*)argv[2], &count))
+    {
+        LOG_ERROR("Invalid packet count");
+        return WICONNECT_BAD_ARG;
+    }
+    if(argc > 3 && !StringUtil::strToUint32((const char*)argv[3], &delay))
+    {
+        LOG_ERROR("Invalid packet delay");
+        return WICONNECT_BAD_ARG;
+    }
+    if(argc > 4 && (!StringUtil::strToUint32((const char*)argv[4], &size) || size > sizeof(testBuffer)/2))
+    {
+        LOG_ERROR("Invalid packet size");
+        return WICONNECT_BAD_ARG;
+    }
+
+    if(WICONNECT_FAILED(result, wiconnect->udpConnect(socket, host, port)))
+    {
+        LOG_ERROR("Failed to connect to server");
+    }
+    else
+    {
+        result = processData(socket, count, delay, size);
+    }
+
+    return result;
+}
+
+
+/*************************************************************************************************/
+static WiconnectResult processData(Socket &socket, uint32_t count, uint32_t delay, uint32_t size)
+{
+    WiconnectResult result;
+    uint8_t *txBuffer = socket.getTxBuffer();
+    const int txBufferLen = socket.getTxBufferSize();
+    uint8_t writeCounter = 32;
+    uint8_t readCounter = 32;
+    uint32_t bytesSent = 0;
+    uint32_t bytesReceived = 0;
+
+    LOG_INFO("\r\n------------------\r\n"
+             "Starting UDP Echo Test:\r\n"
+            "Packet count: %d\r\n"
+            "Packet size: %d\r\n"
+            "Packet delay: %d\r\n"
+            "(Press any key to terminate)",
+            count, size, delay);
+
+    for(int i = 0; i < count || bytesReceived < bytesSent; ++i)
+    {
+        if(i < count)
+        {
+            uint8_t *ptr = txBuffer;
+            int l = size;
+            while(l--)
+            {
+                *ptr++ = writeCounter++;
+                if(writeCounter == 128)
+                    writeCounter = 32;
+            }
+
+            if(WICONNECT_FAILED(result, socket.write(size)))
+            {
+                break;
+            }
+            bytesSent += size;
+            consoleSerial.putc('.');
+        }
+
+        if(consoleSerial.readable())
+        {
+            int c = consoleSerial.getc();
+            break;
+        }
+
+        delayMs(delay);
+
+        uint8_t *readPtr;
+        uint16_t readLen;
+        if(WICONNECT_FAILED(result, socket.read(&readPtr, &readLen)))
+        {
+            break;
+        }
+        bytesReceived += readLen;
+
+        while(readLen--)
+        {
+            if(*readPtr != readCounter)
+            {
+                LOG_ERROR("Invalid: %02X != %02X", *readPtr, readCounter);
+            }
+            ++readPtr;
+            ++readCounter;
+            if(readCounter == 128)
+                readCounter = 32;
+        }
+    }
+
+    consoleSerial.write("\r\n");
+    LOG_INFO("------------------\r\n"
+             "UDP Echo Test Complete:\r\n"
+            "Bytes sent: %d\r\n"
+            "Bytes received: %d\r\n",
+            bytesSent, bytesReceived);
+
+    return result;
+}
+