Junichi Katsu / Milkcocoa

データの保存、更新、取得ができるWebサービス「milkcocoa」に接続し、データのプッシュ、送信、取得ができるライブラリです。 https://mlkcca.com/

Dependencies:   MQTT

Dependents:   MilkcocoaSample MilkcocoaSampleESP8266_LED MilkcocoaSampleESP8266 MilkcocoaSample_3G ... more

ESP8266InterfaceTiny/Socket/TCPSocketConnection.cpp@0:23e533c4b1ec, 2015-12-15 (annotated)

Committer:
jksoft
Date:
Tue Dec 15 09:56:32 2015 +0000
Revision:
0:23e533c4b1ec
??; Subscribe?1?????; ?Milkcocoa::on???????????????????????

Who changed what in which revision?

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