ACKme / WiConnect

Host library for controlling a WiConnect enabled Wi-Fi module.

Dependents:   wiconnect-ota_example wiconnect-web_setup_example wiconnect-test-console wiconnect-tcp_server_example ... more

TcpServer.cpp@42:8ffb253b09e7, 2015-09-28 (annotated)

Committer:
aymangrais
Date:
Mon Sep 28 03:38:43 2015 +0000
Revision:
42:8ffb253b09e7
Parent:
29:b6af04b77a56
Child:
40:4b4306f3d829
increase ota timeout to be 5 seconds (instead of 1.5 sec)

Who changed what in which revision?

UserRevisionLine numberNew contents of line
dan_ackme 29:b6af04b77a56 1 /**
dan_ackme 29:b6af04b77a56 2 * ACKme WiConnect Host Library is licensed under the BSD licence:
dan_ackme 29:b6af04b77a56 3 *
dan_ackme 29:b6af04b77a56 4 * Copyright (c)2014 ACKme Networks.
dan_ackme 29:b6af04b77a56 5 * All rights reserved.
dan_ackme 29:b6af04b77a56 6 *
dan_ackme 29:b6af04b77a56 7 * Redistribution and use in source and binary forms, with or without modification,
dan_ackme 29:b6af04b77a56 8 * are permitted provided that the following conditions are met:
dan_ackme 29:b6af04b77a56 9 *
dan_ackme 29:b6af04b77a56 10 * 1. Redistributions of source code must retain the above copyright notice,
dan_ackme 29:b6af04b77a56 11 * this list of conditions and the following disclaimer.
dan_ackme 29:b6af04b77a56 12 * 2. Redistributions in binary form must reproduce the above copyright notice,
dan_ackme 29:b6af04b77a56 13 * this list of conditions and the following disclaimer in the documentation
dan_ackme 29:b6af04b77a56 14 * and/or other materials provided with the distribution.
dan_ackme 29:b6af04b77a56 15 * 3. The name of the author may not be used to endorse or promote products
dan_ackme 29:b6af04b77a56 16 * derived from this software without specific prior written permission.
dan_ackme 29:b6af04b77a56 17 *
dan_ackme 29:b6af04b77a56 18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS AND ANY EXPRESS OR IMPLIED
dan_ackme 29:b6af04b77a56 19 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
dan_ackme 29:b6af04b77a56 20 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT
dan_ackme 29:b6af04b77a56 21 * SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
dan_ackme 29:b6af04b77a56 22 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
dan_ackme 29:b6af04b77a56 23 * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
dan_ackme 29:b6af04b77a56 24 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
dan_ackme 29:b6af04b77a56 25 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
dan_ackme 29:b6af04b77a56 26 * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
dan_ackme 29:b6af04b77a56 27 * OF SUCH DAMAGE.
dan_ackme 29:b6af04b77a56 28 */
dan_ackme 29:b6af04b77a56 29
dan_ackme 29:b6af04b77a56 30 #include "Wiconnect.h"
dan_ackme 29:b6af04b77a56 31 #include "internal/common.h"
dan_ackme 29:b6af04b77a56 32 #include "api/StringUtil.h"
dan_ackme 29:b6af04b77a56 33
dan_ackme 29:b6af04b77a56 34 #ifdef WICONNECT_GPIO_IRQ_ENABLED
dan_ackme 29:b6af04b77a56 35 #include "api/types/SocketIrqHandlerMap.h"
dan_ackme 29:b6af04b77a56 36 #endif
dan_ackme 29:b6af04b77a56 37
dan_ackme 29:b6af04b77a56 38 #define TCP_SERVER_MONITOR_PERIOD 250 //ms
dan_ackme 29:b6af04b77a56 39
dan_ackme 29:b6af04b77a56 40
dan_ackme 29:b6af04b77a56 41
dan_ackme 29:b6af04b77a56 42 static WiconnectResult parseIpPortStr(char *str, uint32_t *ipAddress, uint16_t *port);
dan_ackme 29:b6af04b77a56 43
dan_ackme 29:b6af04b77a56 44
dan_ackme 29:b6af04b77a56 45
dan_ackme 29:b6af04b77a56 46
dan_ackme 29:b6af04b77a56 47 /*************************************************************************************************/
dan_ackme 29:b6af04b77a56 48 WiconnectResult SocketInterface::tcpListen(uint16_t listeningPort, int maxClients GPIO_IRQ_ARG)
dan_ackme 29:b6af04b77a56 49 {
dan_ackme 29:b6af04b77a56 50 WiconnectResult result = WICONNECT_ERROR;
dan_ackme 29:b6af04b77a56 51
dan_ackme 29:b6af04b77a56 52 enum
dan_ackme 29:b6af04b77a56 53 {
dan_ackme 29:b6af04b77a56 54 FS_SET_MAX_CLIENTS,
dan_ackme 29:b6af04b77a56 55 FS_SET_DATA_GPIO,
dan_ackme 29:b6af04b77a56 56 FS_START_SERVER,
dan_ackme 29:b6af04b77a56 57 };
dan_ackme 29:b6af04b77a56 58
dan_ackme 29:b6af04b77a56 59 CHECK_OTHER_COMMAND_EXECUTING();
dan_ackme 29:b6af04b77a56 60
dan_ackme 29:b6af04b77a56 61 if(wiconnect->internalProcessingState == FS_SET_MAX_CLIENTS)
dan_ackme 29:b6af04b77a56 62 {
dan_ackme 29:b6af04b77a56 63 if(WICONNECT_SUCCEEDED(result, wiconnect->sendCommand("set tcp.server.max_clients", maxClients)))
dan_ackme 29:b6af04b77a56 64 {
dan_ackme 29:b6af04b77a56 65 wiconnect->internalProcessingState = FS_SET_DATA_GPIO;
dan_ackme 29:b6af04b77a56 66 }
dan_ackme 29:b6af04b77a56 67 else if(result == WICONNECT_CMD_RESPONSE_ERROR)
dan_ackme 29:b6af04b77a56 68 {
dan_ackme 29:b6af04b77a56 69 // if there was a module error, then the wiconnect version probably doesn't support this option
dan_ackme 29:b6af04b77a56 70 // just continue to the next state
dan_ackme 29:b6af04b77a56 71 wiconnect->internalProcessingState = FS_SET_DATA_GPIO;
dan_ackme 29:b6af04b77a56 72 }
dan_ackme 29:b6af04b77a56 73 }
dan_ackme 29:b6af04b77a56 74
dan_ackme 29:b6af04b77a56 75 if(wiconnect->internalProcessingState == FS_SET_DATA_GPIO)
dan_ackme 29:b6af04b77a56 76 {
dan_ackme 29:b6af04b77a56 77 #ifdef WICONNECT_GPIO_IRQ_ENABLED
dan_ackme 29:b6af04b77a56 78 if(irqPin == PIN_NC)
dan_ackme 29:b6af04b77a56 79 {
dan_ackme 29:b6af04b77a56 80 wiconnect->internalProcessingState = FS_START_SERVER;
dan_ackme 29:b6af04b77a56 81 }
dan_ackme 29:b6af04b77a56 82 else
dan_ackme 29:b6af04b77a56 83 {
dan_ackme 29:b6af04b77a56 84 PinToGpioMapper mapper = wiconnect->pinToGpioMapper;
dan_ackme 29:b6af04b77a56 85 if(mapper == NULL)
dan_ackme 29:b6af04b77a56 86 {
dan_ackme 29:b6af04b77a56 87 return WICONNECT_PINNAME_TO_GPIO_MAPPER_NULL;
dan_ackme 29:b6af04b77a56 88 }
dan_ackme 29:b6af04b77a56 89 int8_t gpio = mapper(irqPin);
dan_ackme 29:b6af04b77a56 90 if(gpio == -1)
dan_ackme 29:b6af04b77a56 91 {
dan_ackme 29:b6af04b77a56 92 return WICONNECT_PINNAME_TO_GPIO_NO_MAPPING;
dan_ackme 29:b6af04b77a56 93 }
dan_ackme 29:b6af04b77a56 94 else if(!irqHandlers.pinIsRegistered(irqPin))
dan_ackme 29:b6af04b77a56 95 {
dan_ackme 29:b6af04b77a56 96 return WICONNECT_NOT_FOUND;
dan_ackme 29:b6af04b77a56 97 }
dan_ackme 29:b6af04b77a56 98 else if(WICONNECT_SUCCEEDED(result, wiconnect->sendCommand("set tcp.server.data_gpio %d", gpio)))
dan_ackme 29:b6af04b77a56 99 {
dan_ackme 29:b6af04b77a56 100 wiconnect->internalProcessingState = FS_START_SERVER;
dan_ackme 29:b6af04b77a56 101 }
dan_ackme 29:b6af04b77a56 102 }
dan_ackme 29:b6af04b77a56 103 #else
dan_ackme 29:b6af04b77a56 104 wiconnect->internalProcessingState = FS_START_SERVER;
dan_ackme 29:b6af04b77a56 105 #endif
dan_ackme 29:b6af04b77a56 106 }
dan_ackme 29:b6af04b77a56 107
dan_ackme 29:b6af04b77a56 108 if(wiconnect->internalProcessingState == FS_START_SERVER)
dan_ackme 29:b6af04b77a56 109 {
dan_ackme 29:b6af04b77a56 110 if(WICONNECT_SUCCEEDED(result, wiconnect->sendCommand("tcps start %d", listeningPort)))
dan_ackme 29:b6af04b77a56 111 {
dan_ackme 29:b6af04b77a56 112 //#ifdef WICONNECT_ASYNC_TIMER_ENABLED
dan_ackme 29:b6af04b77a56 113 // if(clientConnectedCallback.isValid() && !wiconnect->nonBlocking)
dan_ackme 29:b6af04b77a56 114 // {
dan_ackme 29:b6af04b77a56 115 // serverClientConnectedCallback = clientConnectedCallback;
dan_ackme 29:b6af04b77a56 116 // serverMonitorTimer.start(this, &SocketInterface::serverClientMonitor, TCP_SERVER_MONITOR_PERIOD);
dan_ackme 29:b6af04b77a56 117 // }
dan_ackme 29:b6af04b77a56 118 //#endif
dan_ackme 29:b6af04b77a56 119 }
dan_ackme 29:b6af04b77a56 120 }
dan_ackme 29:b6af04b77a56 121
dan_ackme 29:b6af04b77a56 122 CHECK_CLEANUP_COMMAND();
dan_ackme 29:b6af04b77a56 123
dan_ackme 29:b6af04b77a56 124 return result;
dan_ackme 29:b6af04b77a56 125 }
dan_ackme 29:b6af04b77a56 126
dan_ackme 29:b6af04b77a56 127 /*************************************************************************************************/
dan_ackme 29:b6af04b77a56 128 WiconnectResult SocketInterface::tcpAccept(WiconnectSocket &socket, uint32_t timeoutMs)
dan_ackme 29:b6af04b77a56 129 {
dan_ackme 29:b6af04b77a56 130 TimeoutTimer timer;
dan_ackme 29:b6af04b77a56 131
dan_ackme 29:b6af04b77a56 132 do
dan_ackme 29:b6af04b77a56 133 {
dan_ackme 29:b6af04b77a56 134 uint8_t handle;
dan_ackme 29:b6af04b77a56 135 uint16_t local, remote;
dan_ackme 29:b6af04b77a56 136 uint32_t ipAddress;
dan_ackme 29:b6af04b77a56 137 WiconnectResult result;
dan_ackme 29:b6af04b77a56 138
dan_ackme 29:b6af04b77a56 139 if(WICONNECT_SUCCEEDED(result, pollForServerClient(&handle, &local, &remote, &ipAddress)))
dan_ackme 29:b6af04b77a56 140 {
dan_ackme 29:b6af04b77a56 141 char ipBuffer[17];
dan_ackme 29:b6af04b77a56 142 if(WICONNECT_FAILED(result, socket.init(handle, SOCKET_TYPE_TCP, Wiconnect::ipToStr(ipAddress, ipBuffer), remote, local)))
dan_ackme 29:b6af04b77a56 143 {
dan_ackme 29:b6af04b77a56 144 return result;
dan_ackme 29:b6af04b77a56 145 }
dan_ackme 29:b6af04b77a56 146 serverConnectedClientList |= (1 << handle);
dan_ackme 29:b6af04b77a56 147 return WICONNECT_SUCCESS;
dan_ackme 29:b6af04b77a56 148 }
dan_ackme 29:b6af04b77a56 149 else if(!(result == WICONNECT_PROCESSING || result == WICONNECT_NOT_FOUND))
dan_ackme 29:b6af04b77a56 150 {
dan_ackme 29:b6af04b77a56 151 return result;
dan_ackme 29:b6af04b77a56 152 }
dan_ackme 29:b6af04b77a56 153
dan_ackme 29:b6af04b77a56 154 } while(timeoutMs == WICONNECT_WAIT_FOREVER || !timer.timedOut(timeoutMs));
dan_ackme 29:b6af04b77a56 155
dan_ackme 29:b6af04b77a56 156 return WICONNECT_TIMEOUT;
dan_ackme 29:b6af04b77a56 157 }
dan_ackme 29:b6af04b77a56 158
dan_ackme 29:b6af04b77a56 159 /*************************************************************************************************/
dan_ackme 29:b6af04b77a56 160 WiconnectResult SocketInterface::tcpServerStop(void)
dan_ackme 29:b6af04b77a56 161 {
dan_ackme 29:b6af04b77a56 162 WiconnectResult result = WICONNECT_ERROR;
dan_ackme 29:b6af04b77a56 163
dan_ackme 29:b6af04b77a56 164 CHECK_OTHER_COMMAND_EXECUTING();
dan_ackme 29:b6af04b77a56 165
dan_ackme 29:b6af04b77a56 166 result = wiconnect->sendCommand("tcps stop");
dan_ackme 29:b6af04b77a56 167
dan_ackme 29:b6af04b77a56 168 CHECK_CLEANUP_COMMAND();
dan_ackme 29:b6af04b77a56 169
dan_ackme 29:b6af04b77a56 170 return result;
dan_ackme 29:b6af04b77a56 171 }
dan_ackme 29:b6af04b77a56 172
dan_ackme 29:b6af04b77a56 173 /*************************************************************************************************/
dan_ackme 29:b6af04b77a56 174 WiconnectResult SocketInterface::pollForServerClient(uint8_t *handlePtr, uint16_t *localPort, uint16_t *remotePort, uint32_t *ipAddress)
dan_ackme 29:b6af04b77a56 175 {
dan_ackme 29:b6af04b77a56 176 WiconnectResult result;
dan_ackme 29:b6af04b77a56 177
dan_ackme 29:b6af04b77a56 178 CHECK_OTHER_COMMAND_EXECUTING();
dan_ackme 29:b6af04b77a56 179
dan_ackme 29:b6af04b77a56 180 if(WICONNECT_SUCCEEDED(result, wiconnect->sendCommand("list")))
dan_ackme 29:b6af04b77a56 181 {
dan_ackme 29:b6af04b77a56 182 uint32_t connectedClients = 0;
dan_ackme 29:b6af04b77a56 183 char *line, *savedLine;
dan_ackme 29:b6af04b77a56 184 result = WICONNECT_NOT_FOUND;
dan_ackme 29:b6af04b77a56 185
dan_ackme 29:b6af04b77a56 186 for(savedLine = wiconnect->internalBuffer; (line = StringUtil::strtok_r(savedLine, "\r\n", &savedLine)) != NULL;)
dan_ackme 29:b6af04b77a56 187 {
dan_ackme 29:b6af04b77a56 188 char *toks[4], *savedTok;
dan_ackme 29:b6af04b77a56 189
dan_ackme 29:b6af04b77a56 190 if(*line != '#')
dan_ackme 29:b6af04b77a56 191 {
dan_ackme 29:b6af04b77a56 192 continue;
dan_ackme 29:b6af04b77a56 193 }
dan_ackme 29:b6af04b77a56 194 savedTok = line + 2;
dan_ackme 29:b6af04b77a56 195
dan_ackme 29:b6af04b77a56 196 for(int i = 0; i < 4 && (toks[i] = StringUtil::strtok_r(savedTok, " ", &savedTok)) != NULL; ++i)
dan_ackme 29:b6af04b77a56 197 {
dan_ackme 29:b6af04b77a56 198 if(toks[i] == NULL)
dan_ackme 29:b6af04b77a56 199 {
dan_ackme 29:b6af04b77a56 200 result = WICONNECT_RESPONSE_PARSE_ERROR;
dan_ackme 29:b6af04b77a56 201 goto exit;
dan_ackme 29:b6af04b77a56 202 }
dan_ackme 29:b6af04b77a56 203 }
dan_ackme 29:b6af04b77a56 204
dan_ackme 29:b6af04b77a56 205 if(strcmp(toks[1], "TCPS") != 0)
dan_ackme 29:b6af04b77a56 206 {
dan_ackme 29:b6af04b77a56 207 continue;
dan_ackme 29:b6af04b77a56 208 }
dan_ackme 29:b6af04b77a56 209
dan_ackme 29:b6af04b77a56 210 uint8_t handle = (uint8_t)(*toks[0] - '0');
dan_ackme 29:b6af04b77a56 211 if(handle >= WICONNECT_MAX_SOCKETS)
dan_ackme 29:b6af04b77a56 212 {
dan_ackme 29:b6af04b77a56 213 result = WICONNECT_RESPONSE_PARSE_ERROR;
dan_ackme 29:b6af04b77a56 214 goto exit;
dan_ackme 29:b6af04b77a56 215 }
dan_ackme 29:b6af04b77a56 216 const uint32_t handleMask = (1 << handle);
dan_ackme 29:b6af04b77a56 217 connectedClients |= handleMask;
dan_ackme 29:b6af04b77a56 218
dan_ackme 29:b6af04b77a56 219 if(result == WICONNECT_SUCCESS)
dan_ackme 29:b6af04b77a56 220 {
dan_ackme 29:b6af04b77a56 221 continue;
dan_ackme 29:b6af04b77a56 222 }
dan_ackme 29:b6af04b77a56 223 else if(serverConnectedClientList & handleMask)
dan_ackme 29:b6af04b77a56 224 {
dan_ackme 29:b6af04b77a56 225 continue;
dan_ackme 29:b6af04b77a56 226 }
dan_ackme 29:b6af04b77a56 227
dan_ackme 29:b6af04b77a56 228 result = WICONNECT_SUCCESS;
dan_ackme 29:b6af04b77a56 229
dan_ackme 29:b6af04b77a56 230 if(handlePtr != NULL)
dan_ackme 29:b6af04b77a56 231 {
dan_ackme 29:b6af04b77a56 232 *handlePtr = handle;
dan_ackme 29:b6af04b77a56 233 parseIpPortStr(toks[2], NULL, localPort);
dan_ackme 29:b6af04b77a56 234 parseIpPortStr(toks[3], ipAddress, remotePort);
dan_ackme 29:b6af04b77a56 235 }
dan_ackme 29:b6af04b77a56 236 }
dan_ackme 29:b6af04b77a56 237
dan_ackme 29:b6af04b77a56 238 uint32_t mask = 1;
dan_ackme 29:b6af04b77a56 239 for(int i = 0; i < WICONNECT_MAX_SOCKETS; ++i, mask <<= 1)
dan_ackme 29:b6af04b77a56 240 {
dan_ackme 29:b6af04b77a56 241 if(!(connectedClients & mask))
dan_ackme 29:b6af04b77a56 242 {
dan_ackme 29:b6af04b77a56 243 serverConnectedClientList &= ~mask;
dan_ackme 29:b6af04b77a56 244 }
dan_ackme 29:b6af04b77a56 245 }
dan_ackme 29:b6af04b77a56 246 }
dan_ackme 29:b6af04b77a56 247
dan_ackme 29:b6af04b77a56 248
dan_ackme 29:b6af04b77a56 249 exit:
dan_ackme 29:b6af04b77a56 250 CHECK_CLEANUP_COMMAND();
dan_ackme 29:b6af04b77a56 251
dan_ackme 29:b6af04b77a56 252 return result;
dan_ackme 29:b6af04b77a56 253 }
dan_ackme 29:b6af04b77a56 254
dan_ackme 29:b6af04b77a56 255 /*************************************************************************************************/
dan_ackme 29:b6af04b77a56 256 static WiconnectResult parseIpPortStr(char *str, uint32_t *ipAddress, uint16_t *port)
dan_ackme 29:b6af04b77a56 257 {
dan_ackme 29:b6af04b77a56 258 char *colon = strchr(str, ':');
dan_ackme 29:b6af04b77a56 259 if(colon == NULL)
dan_ackme 29:b6af04b77a56 260 {
dan_ackme 29:b6af04b77a56 261 return WICONNECT_RESPONSE_PARSE_ERROR;
dan_ackme 29:b6af04b77a56 262 }
dan_ackme 29:b6af04b77a56 263 *colon++ = 0;
dan_ackme 29:b6af04b77a56 264
dan_ackme 29:b6af04b77a56 265 if(ipAddress != NULL && !Wiconnect::strToIp(str, ipAddress))
dan_ackme 29:b6af04b77a56 266 {
dan_ackme 29:b6af04b77a56 267 return WICONNECT_RESPONSE_PARSE_ERROR;
dan_ackme 29:b6af04b77a56 268 }
dan_ackme 29:b6af04b77a56 269 else if(!StringUtil::strToUint16(colon, port))
dan_ackme 29:b6af04b77a56 270 {
dan_ackme 29:b6af04b77a56 271 return WICONNECT_RESPONSE_PARSE_ERROR;
dan_ackme 29:b6af04b77a56 272 }
dan_ackme 29:b6af04b77a56 273
dan_ackme 29:b6af04b77a56 274 return WICONNECT_SUCCESS;
dan_ackme 29:b6af04b77a56 275 }