Junichi Katsu / MultiIoTBoardLib

MultiIoTBoard library

Dependencies:   BME280 SB1602E

Dependents:   MITB_Sample

ESP8266InterfaceTiny/Socket/TCPSocketConnection.cpp@0:bad9495b4215, 2017-07-28 (annotated)

Committer:
jksoft
Date:
Fri Jul 28 02:23:25 2017 +0000
Revision:
0:bad9495b4215
First edition

Who changed what in which revision?

UserRevisionLine numberNew contents of line
jksoft 0:bad9495b4215 1 /* Copyright (C) 2012 mbed.org, MIT License
jksoft 0:bad9495b4215 2 *
jksoft 0:bad9495b4215 3 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software
jksoft 0:bad9495b4215 4 * and associated documentation files (the "Software"), to deal in the Software without restriction,
jksoft 0:bad9495b4215 5 * including without limitation the rights to use, copy, modify, merge, publish, distribute,
jksoft 0:bad9495b4215 6 * sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
jksoft 0:bad9495b4215 7 * furnished to do so, subject to the following conditions:
jksoft 0:bad9495b4215 8 *
jksoft 0:bad9495b4215 9 * The above copyright notice and this permission notice shall be included in all copies or
jksoft 0:bad9495b4215 10 * substantial portions of the Software.
jksoft 0:bad9495b4215 11 *
jksoft 0:bad9495b4215 12 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
jksoft 0:bad9495b4215 13 * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
jksoft 0:bad9495b4215 14 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
jksoft 0:bad9495b4215 15 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
jksoft 0:bad9495b4215 16 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
jksoft 0:bad9495b4215 17 */
jksoft 0:bad9495b4215 18 #include "TCPSocketConnection.h"
jksoft 0:bad9495b4215 19 #include <cstring>
jksoft 0:bad9495b4215 20 #include <algorithm>
jksoft 0:bad9495b4215 21
jksoft 0:bad9495b4215 22 using std::memset;
jksoft 0:bad9495b4215 23 using std::memcpy;
jksoft 0:bad9495b4215 24
jksoft 0:bad9495b4215 25 //Debug is disabled by default
jksoft 0:bad9495b4215 26 #if 1
jksoft 0:bad9495b4215 27 #define DBG(x, ...) printf("[TCPConnection : DBG]"x" \t[%s,%d]\r\n", ##__VA_ARGS__,__FILE__,__LINE__);
jksoft 0:bad9495b4215 28 #define WARN(x, ...) printf("[TCPConnection: WARN]"x" \t[%s,%d]\r\n", ##__VA_ARGS__,__FILE__,__LINE__);
jksoft 0:bad9495b4215 29 #define ERR(x, ...) printf("[TCPConnection : ERR]"x" \t[%s,%d]\r\n", ##__VA_ARGS__,__FILE__,__LINE__);
jksoft 0:bad9495b4215 30 #else
jksoft 0:bad9495b4215 31 #define DBG(x, ...)
jksoft 0:bad9495b4215 32 #define WARN(x, ...)
jksoft 0:bad9495b4215 33 #define ERR(x, ...)
jksoft 0:bad9495b4215 34 #endif
jksoft 0:bad9495b4215 35
jksoft 0:bad9495b4215 36 TCPSocketConnection::TCPSocketConnection() :
jksoft 0:bad9495b4215 37 _is_connected(false)
jksoft 0:bad9495b4215 38 {
jksoft 0:bad9495b4215 39 }
jksoft 0:bad9495b4215 40
jksoft 0:bad9495b4215 41 int TCPSocketConnection::connect(const char* host, const int port)
jksoft 0:bad9495b4215 42 {
jksoft 0:bad9495b4215 43 // if (init_socket(SOCK_STREAM) < 0)
jksoft 0:bad9495b4215 44 // return -1;
jksoft 0:bad9495b4215 45 //
jksoft 0:bad9495b4215 46 if (set_address(host, port) != 0)
jksoft 0:bad9495b4215 47 return -1;
jksoft 0:bad9495b4215 48 //
jksoft 0:bad9495b4215 49 // if (lwip_connect(_sock_fd, (const struct sockaddr *) &_remoteHost, sizeof(_remoteHost)) < 0) {
jksoft 0:bad9495b4215 50 // close();
jksoft 0:bad9495b4215 51 // return -1;
jksoft 0:bad9495b4215 52 // }
jksoft 0:bad9495b4215 53 // _is_connected = true;
jksoft 0:bad9495b4215 54
jksoft 0:bad9495b4215 55 _is_connected = ESP8266->start(ESP_TCP_TYPE,_ipAddress,_port);
jksoft 0:bad9495b4215 56 if(_is_connected) { //success
jksoft 0:bad9495b4215 57 return 0;
jksoft 0:bad9495b4215 58 } else { // fail
jksoft 0:bad9495b4215 59 return -1;
jksoft 0:bad9495b4215 60 }
jksoft 0:bad9495b4215 61 }
jksoft 0:bad9495b4215 62
jksoft 0:bad9495b4215 63 bool TCPSocketConnection::is_connected(void)
jksoft 0:bad9495b4215 64 {
jksoft 0:bad9495b4215 65 return _is_connected;
jksoft 0:bad9495b4215 66 }
jksoft 0:bad9495b4215 67
jksoft 0:bad9495b4215 68 int TCPSocketConnection::send(char* data, int length)
jksoft 0:bad9495b4215 69 {
jksoft 0:bad9495b4215 70 if (!_is_connected) {
jksoft 0:bad9495b4215 71 ERR("TCPSocketConnection::receive() - _is_connected is false : you cant receive data until you connect to a socket!");
jksoft 0:bad9495b4215 72 return -1;
jksoft 0:bad9495b4215 73 }
jksoft 0:bad9495b4215 74 Timer tmr;
jksoft 0:bad9495b4215 75 int idx = 0;
jksoft 0:bad9495b4215 76 tmr.start();
jksoft 0:bad9495b4215 77 while ((tmr.read_ms() < _timeout) || _blocking) {
jksoft 0:bad9495b4215 78
jksoft 0:bad9495b4215 79 idx += wifi->send(data, length);
jksoft 0:bad9495b4215 80
jksoft 0:bad9495b4215 81 if (idx == length)
jksoft 0:bad9495b4215 82 return idx;
jksoft 0:bad9495b4215 83 }
jksoft 0:bad9495b4215 84 return (idx == 0) ? -1 : idx;
jksoft 0:bad9495b4215 85
jksoft 0:bad9495b4215 86 //return wifi->send(data,length);
jksoft 0:bad9495b4215 87 //
jksoft 0:bad9495b4215 88 // if (!_blocking) {
jksoft 0:bad9495b4215 89 // TimeInterval timeout(_timeout);
jksoft 0:bad9495b4215 90 // if (wait_writable(timeout) != 0)
jksoft 0:bad9495b4215 91 // return -1;
jksoft 0:bad9495b4215 92 // }
jksoft 0:bad9495b4215 93 //
jksoft 0:bad9495b4215 94 // int n = lwip_send(_sock_fd, data, length, 0);
jksoft 0:bad9495b4215 95 // _is_connected = (n != 0);
jksoft 0:bad9495b4215 96 //
jksoft 0:bad9495b4215 97 // return n;
jksoft 0:bad9495b4215 98
jksoft 0:bad9495b4215 99 }
jksoft 0:bad9495b4215 100
jksoft 0:bad9495b4215 101 // -1 if unsuccessful, else number of bytes written
jksoft 0:bad9495b4215 102 int TCPSocketConnection::send_all(char* data, int length)
jksoft 0:bad9495b4215 103 {
jksoft 0:bad9495b4215 104 // if ((_sock_fd < 0) || !_is_connected)
jksoft 0:bad9495b4215 105 // return -1;
jksoft 0:bad9495b4215 106 //
jksoft 0:bad9495b4215 107 // int writtenLen = 0;
jksoft 0:bad9495b4215 108 // TimeInterval timeout(_timeout);
jksoft 0:bad9495b4215 109 // while (writtenLen < length) {
jksoft 0:bad9495b4215 110 // if (!_blocking) {
jksoft 0:bad9495b4215 111 // // Wait for socket to be writeable
jksoft 0:bad9495b4215 112 // if (wait_writable(timeout) != 0)
jksoft 0:bad9495b4215 113 // return writtenLen;
jksoft 0:bad9495b4215 114 // }
jksoft 0:bad9495b4215 115 //
jksoft 0:bad9495b4215 116 // int ret = lwip_send(_sock_fd, data + writtenLen, length - writtenLen, 0);
jksoft 0:bad9495b4215 117 // if (ret > 0) {
jksoft 0:bad9495b4215 118 // writtenLen += ret;
jksoft 0:bad9495b4215 119 // continue;
jksoft 0:bad9495b4215 120 // } else if (ret == 0) {
jksoft 0:bad9495b4215 121 // _is_connected = false;
jksoft 0:bad9495b4215 122 // return writtenLen;
jksoft 0:bad9495b4215 123 // } else {
jksoft 0:bad9495b4215 124 // return -1; //Connnection error
jksoft 0:bad9495b4215 125 // }
jksoft 0:bad9495b4215 126 // }
jksoft 0:bad9495b4215 127 // return writtenLen;
jksoft 0:bad9495b4215 128 return send(data,length); // just remap to send
jksoft 0:bad9495b4215 129 }
jksoft 0:bad9495b4215 130
jksoft 0:bad9495b4215 131 int TCPSocketConnection::receive(char* buffer, int length)
jksoft 0:bad9495b4215 132 {
jksoft 0:bad9495b4215 133 if (!_is_connected) {
jksoft 0:bad9495b4215 134 ERR("TCPSocketConnection::receive() - _is_connected is false : you cant receive data until you connect to a socket!");
jksoft 0:bad9495b4215 135 return -1;
jksoft 0:bad9495b4215 136 }
jksoft 0:bad9495b4215 137 Timer tmr;
jksoft 0:bad9495b4215 138 int idx = 0;
jksoft 0:bad9495b4215 139 int nb_available = 0;
jksoft 0:bad9495b4215 140 int time = -1;
jksoft 0:bad9495b4215 141
jksoft 0:bad9495b4215 142 //make this the non-blocking case and return if <= 0
jksoft 0:bad9495b4215 143 // remember to change the config to blocking
jksoft 0:bad9495b4215 144 // if ( ! _blocking) {
jksoft 0:bad9495b4215 145 // if ( wifi.readable <= 0 ) {
jksoft 0:bad9495b4215 146 // return (wifi.readable);
jksoft 0:bad9495b4215 147 // }
jksoft 0:bad9495b4215 148 // }
jksoft 0:bad9495b4215 149 //---
jksoft 0:bad9495b4215 150 tmr.start();
jksoft 0:bad9495b4215 151 if (_blocking) {
jksoft 0:bad9495b4215 152 while (1) {
jksoft 0:bad9495b4215 153 nb_available = wifi->readable();
jksoft 0:bad9495b4215 154 if (nb_available != 0) {
jksoft 0:bad9495b4215 155 break;
jksoft 0:bad9495b4215 156 }
jksoft 0:bad9495b4215 157 }
jksoft 0:bad9495b4215 158 }
jksoft 0:bad9495b4215 159 //---
jksoft 0:bad9495b4215 160 // blocking case
jksoft 0:bad9495b4215 161 else {
jksoft 0:bad9495b4215 162 tmr.reset();
jksoft 0:bad9495b4215 163
jksoft 0:bad9495b4215 164 while (time < _timeout) {
jksoft 0:bad9495b4215 165 nb_available = wifi->readable();
jksoft 0:bad9495b4215 166 if (nb_available < 0) return nb_available;
jksoft 0:bad9495b4215 167 if (nb_available > 0) break ;
jksoft 0:bad9495b4215 168 time = tmr.read_ms();
jksoft 0:bad9495b4215 169 }
jksoft 0:bad9495b4215 170
jksoft 0:bad9495b4215 171 if (nb_available == 0) return nb_available;
jksoft 0:bad9495b4215 172 }
jksoft 0:bad9495b4215 173
jksoft 0:bad9495b4215 174 // change this to < 20 mS timeout per byte to detect end of packet gap
jksoft 0:bad9495b4215 175 // this may not work due to buffering at the UART interface
jksoft 0:bad9495b4215 176 tmr.reset();
jksoft 0:bad9495b4215 177 // while ( tmr.read_ms() < 20 ) {
jksoft 0:bad9495b4215 178 // if ( wifi.readable() && (idx < length) ) {
jksoft 0:bad9495b4215 179 // buffer[idx++] = wifi->getc();
jksoft 0:bad9495b4215 180 // tmr.reset();
jksoft 0:bad9495b4215 181 // }
jksoft 0:bad9495b4215 182 // if ( idx == length ) {
jksoft 0:bad9495b4215 183 // break;
jksoft 0:bad9495b4215 184 // }
jksoft 0:bad9495b4215 185 // }
jksoft 0:bad9495b4215 186 //---
jksoft 0:bad9495b4215 187 while (time < _timeout) {
jksoft 0:bad9495b4215 188
jksoft 0:bad9495b4215 189 nb_available = wifi->readable();
jksoft 0:bad9495b4215 190 //for (int i = 0; i < min(nb_available, length); i++) {
jksoft 0:bad9495b4215 191 for (int i = 0; i < min(nb_available, (length-idx)); i++) {
jksoft 0:bad9495b4215 192 buffer[idx] = wifi->getc();
jksoft 0:bad9495b4215 193 idx++;
jksoft 0:bad9495b4215 194 }
jksoft 0:bad9495b4215 195 if (idx == length) {
jksoft 0:bad9495b4215 196 break;
jksoft 0:bad9495b4215 197 }
jksoft 0:bad9495b4215 198 time = tmr.read_ms();
jksoft 0:bad9495b4215 199 }
jksoft 0:bad9495b4215 200 //---
jksoft 0:bad9495b4215 201 return (idx == 0) ? -1 : idx;
jksoft 0:bad9495b4215 202
jksoft 0:bad9495b4215 203 //************************ original code below
jksoft 0:bad9495b4215 204 //
jksoft 0:bad9495b4215 205 // if (!_blocking) {
jksoft 0:bad9495b4215 206 // TimeInterval timeout(_timeout);
jksoft 0:bad9495b4215 207 // if (wait_readable(timeout) != 0)
jksoft 0:bad9495b4215 208 // return -1;
jksoft 0:bad9495b4215 209 // }
jksoft 0:bad9495b4215 210 //
jksoft 0:bad9495b4215 211 // int n = lwip_recv(_sock_fd, data, length, 0);
jksoft 0:bad9495b4215 212 // _is_connected = (n != 0);
jksoft 0:bad9495b4215 213 //
jksoft 0:bad9495b4215 214 // return n;
jksoft 0:bad9495b4215 215
jksoft 0:bad9495b4215 216 }
jksoft 0:bad9495b4215 217
jksoft 0:bad9495b4215 218 // -1 if unsuccessful, else number of bytes received
jksoft 0:bad9495b4215 219 int TCPSocketConnection::receive_all(char* data, int length)
jksoft 0:bad9495b4215 220 {
jksoft 0:bad9495b4215 221 //ERR("receive_all() not yet implimented");
jksoft 0:bad9495b4215 222 // if ((_sock_fd < 0) || !_is_connected)
jksoft 0:bad9495b4215 223 // return -1;
jksoft 0:bad9495b4215 224 //
jksoft 0:bad9495b4215 225 // int readLen = 0;
jksoft 0:bad9495b4215 226 // TimeInterval timeout(_timeout);
jksoft 0:bad9495b4215 227 // while (readLen < length) {
jksoft 0:bad9495b4215 228 // if (!_blocking) {
jksoft 0:bad9495b4215 229 // //Wait for socket to be readable
jksoft 0:bad9495b4215 230 // if (wait_readable(timeout) != 0)
jksoft 0:bad9495b4215 231 // return readLen;
jksoft 0:bad9495b4215 232 // }
jksoft 0:bad9495b4215 233 //
jksoft 0:bad9495b4215 234 // int ret = lwip_recv(_sock_fd, data + readLen, length - readLen, 0);
jksoft 0:bad9495b4215 235 // if (ret > 0) {
jksoft 0:bad9495b4215 236 // readLen += ret;
jksoft 0:bad9495b4215 237 // } else if (ret == 0) {
jksoft 0:bad9495b4215 238 // _is_connected = false;
jksoft 0:bad9495b4215 239 // return readLen;
jksoft 0:bad9495b4215 240 // } else {
jksoft 0:bad9495b4215 241 // return -1; //Connnection error
jksoft 0:bad9495b4215 242 // }
jksoft 0:bad9495b4215 243 // }
jksoft 0:bad9495b4215 244 // return readLen;
jksoft 0:bad9495b4215 245 receive(data,length);
jksoft 0:bad9495b4215 246 }