Fang Bao / HTTPClient

HTTPClient

Fork of HTTPClient by Donatien Garnier

HTTPClient.cpp@13:be61104f4e91, 2012-08-05 (annotated)

Committer:
donatien
Date:
Sun Aug 05 16:12:10 2012 +0000
Revision:
13:be61104f4e91
Parent:
12:89d09a6db00a
Child:
14:2744e0c0e527
Fixed blocking mode

Who changed what in which revision?

UserRevisionLine numberNew contents of line
donatien 0:2ccb9960a044 1 /* HTTPClient.cpp */
donatien 10:e1351de84c16 2 /* Copyright (C) 2012 mbed.org, MIT License
donatien 10:e1351de84c16 3 *
donatien 10:e1351de84c16 4 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software
donatien 10:e1351de84c16 5 * and associated documentation files (the "Software"), to deal in the Software without restriction,
donatien 10:e1351de84c16 6 * including without limitation the rights to use, copy, modify, merge, publish, distribute,
donatien 10:e1351de84c16 7 * sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
donatien 10:e1351de84c16 8 * furnished to do so, subject to the following conditions:
donatien 10:e1351de84c16 9 *
donatien 10:e1351de84c16 10 * The above copyright notice and this permission notice shall be included in all copies or
donatien 10:e1351de84c16 11 * substantial portions of the Software.
donatien 10:e1351de84c16 12 *
donatien 10:e1351de84c16 13 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
donatien 10:e1351de84c16 14 * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
donatien 10:e1351de84c16 15 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
donatien 10:e1351de84c16 16 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
donatien 10:e1351de84c16 17 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
donatien 10:e1351de84c16 18 */
donatien 0:2ccb9960a044 19
donatien 7:4e39864f7b15 20 //Debug is disabled by default
donatien 13:be61104f4e91 21 #if 1
donatien 12:89d09a6db00a 22 //Enable debug
donatien 11:390362de8c3f 23 #include <cstdio>
donatien 13:be61104f4e91 24 //#define DBG(x, ...) std::printf("[HTTPClient : DBG]"x"\r\n", ##__VA_ARGS__);
donatien 13:be61104f4e91 25 #define DBG(x, ...)
donatien 11:390362de8c3f 26 #define WARN(x, ...) std::printf("[HTTPClient : WARN]"x"\r\n", ##__VA_ARGS__);
donatien 11:390362de8c3f 27 #define ERR(x, ...) std::printf("[HTTPClient : ERR]"x"\r\n", ##__VA_ARGS__);
donatien 12:89d09a6db00a 28
donatien 12:89d09a6db00a 29 #else
donatien 12:89d09a6db00a 30 //Disable debug
donatien 12:89d09a6db00a 31 #define DBG(x, ...)
donatien 12:89d09a6db00a 32 #define WARN(x, ...)
donatien 12:89d09a6db00a 33 #define ERR(x, ...)
donatien 12:89d09a6db00a 34
donatien 7:4e39864f7b15 35 #endif
donatien 0:2ccb9960a044 36
donatien 0:2ccb9960a044 37 #define HTTP_PORT 80
donatien 0:2ccb9960a044 38
donatien 11:390362de8c3f 39 #define OK 0
donatien 11:390362de8c3f 40
donatien 11:390362de8c3f 41 #define MIN(x,y) (((x)<(y))?(x):(y))
donatien 11:390362de8c3f 42 #define MAX(x,y) (((x)>(y))?(x):(y))
donatien 11:390362de8c3f 43
donatien 0:2ccb9960a044 44 #define CHUNK_SIZE 256
donatien 0:2ccb9960a044 45
donatien 0:2ccb9960a044 46 #include <cstring>
donatien 0:2ccb9960a044 47
donatien 11:390362de8c3f 48 #include "HTTPClient.h"
donatien 11:390362de8c3f 49
donatien 0:2ccb9960a044 50 HTTPClient::HTTPClient() :
donatien 7:4e39864f7b15 51 m_sock(), m_basicAuthUser(NULL), m_basicAuthPassword(NULL), m_httpResponseCode(0)
donatien 0:2ccb9960a044 52 {
donatien 0:2ccb9960a044 53
donatien 0:2ccb9960a044 54 }
donatien 0:2ccb9960a044 55
donatien 0:2ccb9960a044 56 HTTPClient::~HTTPClient()
donatien 0:2ccb9960a044 57 {
donatien 0:2ccb9960a044 58
donatien 0:2ccb9960a044 59 }
donatien 0:2ccb9960a044 60
donatien 0:2ccb9960a044 61 #if 0
donatien 0:2ccb9960a044 62 void HTTPClient::basicAuth(const char* user, const char* password) //Basic Authentification
donatien 0:2ccb9960a044 63 {
donatien 0:2ccb9960a044 64 m_basicAuthUser = user;
donatien 0:2ccb9960a044 65 m_basicAuthPassword = password;
donatien 0:2ccb9960a044 66 }
donatien 0:2ccb9960a044 67 #endif
donatien 0:2ccb9960a044 68
donatien 12:89d09a6db00a 69 HTTPResult HTTPClient::get(const char* url, IHTTPDataIn* pDataIn, int timeout /*= HTTP_CLIENT_DEFAULT_TIMEOUT*/) //Blocking
donatien 0:2ccb9960a044 70 {
donatien 0:2ccb9960a044 71 return connect(url, HTTP_GET, NULL, pDataIn, timeout);
donatien 0:2ccb9960a044 72 }
donatien 0:2ccb9960a044 73
donatien 12:89d09a6db00a 74 HTTPResult HTTPClient::get(const char* url, char* result, size_t maxResultLen, int timeout /*= HTTP_CLIENT_DEFAULT_TIMEOUT*/) //Blocking
donatien 0:2ccb9960a044 75 {
donatien 0:2ccb9960a044 76 HTTPText str(result, maxResultLen);
donatien 0:2ccb9960a044 77 return get(url, &str, timeout);
donatien 0:2ccb9960a044 78 }
donatien 0:2ccb9960a044 79
donatien 12:89d09a6db00a 80 HTTPResult HTTPClient::post(const char* url, const IHTTPDataOut& dataOut, IHTTPDataIn* pDataIn, int timeout /*= HTTP_CLIENT_DEFAULT_TIMEOUT*/) //Blocking
donatien 0:2ccb9960a044 81 {
donatien 0:2ccb9960a044 82 return connect(url, HTTP_POST, (IHTTPDataOut*)&dataOut, pDataIn, timeout);
donatien 0:2ccb9960a044 83 }
donatien 0:2ccb9960a044 84
donatien 0:2ccb9960a044 85 int HTTPClient::getHTTPResponseCode()
donatien 0:2ccb9960a044 86 {
donatien 0:2ccb9960a044 87 return m_httpResponseCode;
donatien 0:2ccb9960a044 88 }
donatien 0:2ccb9960a044 89
donatien 5:791fc3dcb6c4 90 #define CHECK_CONN_ERR(ret) \
donatien 5:791fc3dcb6c4 91 do{ \
donatien 7:4e39864f7b15 92 if(ret) { \
donatien 7:4e39864f7b15 93 m_sock.close(); \
donatien 5:791fc3dcb6c4 94 ERR("Connection error (%d)", ret); \
donatien 11:390362de8c3f 95 return HTTP_CONN; \
donatien 5:791fc3dcb6c4 96 } \
donatien 5:791fc3dcb6c4 97 } while(0)
donatien 5:791fc3dcb6c4 98
donatien 5:791fc3dcb6c4 99 #define PRTCL_ERR() \
donatien 5:791fc3dcb6c4 100 do{ \
donatien 7:4e39864f7b15 101 m_sock.close(); \
donatien 5:791fc3dcb6c4 102 ERR("Protocol error"); \
donatien 11:390362de8c3f 103 return HTTP_PRTCL; \
donatien 5:791fc3dcb6c4 104 } while(0)
donatien 0:2ccb9960a044 105
donatien 12:89d09a6db00a 106 HTTPResult HTTPClient::connect(const char* url, HTTP_METH method, IHTTPDataOut* pDataOut, IHTTPDataIn* pDataIn, int timeout) //Execute request
donatien 12:89d09a6db00a 107 {
donatien 0:2ccb9960a044 108 m_httpResponseCode = 0; //Invalidate code
donatien 0:2ccb9960a044 109 m_timeout = timeout;
donatien 0:2ccb9960a044 110
donatien 0:2ccb9960a044 111 char scheme[8];
donatien 0:2ccb9960a044 112 uint16_t port;
donatien 0:2ccb9960a044 113 char host[32];
donatien 0:2ccb9960a044 114 char path[64];
donatien 0:2ccb9960a044 115 //First we need to parse the url (http[s]://host[:port][/[path]]) -- HTTPS not supported (yet?)
donatien 11:390362de8c3f 116 HTTPResult res = parseURL(url, scheme, sizeof(scheme), host, sizeof(host), &port, path, sizeof(path));
donatien 11:390362de8c3f 117 if(res != HTTP_OK)
donatien 0:2ccb9960a044 118 {
donatien 11:390362de8c3f 119 ERR("parseURL returned %d", res);
donatien 11:390362de8c3f 120 return res;
donatien 0:2ccb9960a044 121 }
donatien 0:2ccb9960a044 122
donatien 0:2ccb9960a044 123 if(port == 0) //TODO do handle HTTPS->443
donatien 0:2ccb9960a044 124 {
donatien 0:2ccb9960a044 125 port = 80;
donatien 0:2ccb9960a044 126 }
donatien 0:2ccb9960a044 127
donatien 0:2ccb9960a044 128 DBG("Scheme: %s", scheme);
donatien 0:2ccb9960a044 129 DBG("Host: %s", host);
donatien 0:2ccb9960a044 130 DBG("Port: %d", port);
donatien 0:2ccb9960a044 131 DBG("Path: %s", path);
donatien 0:2ccb9960a044 132
donatien 7:4e39864f7b15 133 //Connect
donatien 7:4e39864f7b15 134 DBG("Connecting socket to server");
donatien 11:390362de8c3f 135 int ret = m_sock.connect(host, port);
donatien 0:2ccb9960a044 136 if (ret < 0)
donatien 0:2ccb9960a044 137 {
donatien 7:4e39864f7b15 138 m_sock.close();
donatien 0:2ccb9960a044 139 ERR("Could not connect");
donatien 11:390362de8c3f 140 return HTTP_CONN;
donatien 0:2ccb9960a044 141 }
donatien 0:2ccb9960a044 142
donatien 0:2ccb9960a044 143 //Send request
donatien 0:2ccb9960a044 144 DBG("Sending request");
donatien 9:ff30cc189191 145 char buf[CHUNK_SIZE];
donatien 0:2ccb9960a044 146 const char* meth = (method==HTTP_GET)?"GET":(method==HTTP_POST)?"POST":"";
donatien 9:ff30cc189191 147 snprintf(buf, sizeof(buf), "%s %s HTTP/1.1\r\nHost: %s\r\n", meth, path, host); //Write request
donatien 9:ff30cc189191 148 ret = send(buf);
donatien 0:2ccb9960a044 149 if(ret)
donatien 0:2ccb9960a044 150 {
donatien 7:4e39864f7b15 151 m_sock.close();
donatien 0:2ccb9960a044 152 ERR("Could not write request");
donatien 11:390362de8c3f 153 return HTTP_CONN;
donatien 0:2ccb9960a044 154 }
donatien 0:2ccb9960a044 155
donatien 0:2ccb9960a044 156 //Send all headers
donatien 0:2ccb9960a044 157
donatien 0:2ccb9960a044 158 //Send default headers
donatien 0:2ccb9960a044 159 DBG("Sending headers");
donatien 0:2ccb9960a044 160 if( (method == HTTP_POST) && (pDataOut != NULL) )
donatien 0:2ccb9960a044 161 {
donatien 0:2ccb9960a044 162 if( pDataOut->getIsChunked() )
donatien 0:2ccb9960a044 163 {
donatien 0:2ccb9960a044 164 ret = send("Transfer-Encoding: chunked\r\n");
donatien 5:791fc3dcb6c4 165 CHECK_CONN_ERR(ret);
donatien 0:2ccb9960a044 166 }
donatien 0:2ccb9960a044 167 else
donatien 0:2ccb9960a044 168 {
donatien 9:ff30cc189191 169 snprintf(buf, sizeof(buf), "Content-Length: %d\r\n", pDataOut->getDataLen());
donatien 9:ff30cc189191 170 ret = send(buf);
donatien 5:791fc3dcb6c4 171 CHECK_CONN_ERR(ret);
donatien 0:2ccb9960a044 172 }
donatien 0:2ccb9960a044 173 char type[48];
donatien 11:390362de8c3f 174 if( pDataOut->getDataType(type, 48) == HTTP_OK )
donatien 0:2ccb9960a044 175 {
donatien 9:ff30cc189191 176 snprintf(buf, sizeof(buf), "Content-Type: %s\r\n", type);
donatien 9:ff30cc189191 177 ret = send(buf);
donatien 5:791fc3dcb6c4 178 CHECK_CONN_ERR(ret);
donatien 0:2ccb9960a044 179 }
donatien 0:2ccb9960a044 180 }
donatien 9:ff30cc189191 181
donatien 0:2ccb9960a044 182 //Close headers
donatien 0:2ccb9960a044 183 DBG("Headers sent");
donatien 0:2ccb9960a044 184 ret = send("\r\n");
donatien 5:791fc3dcb6c4 185 CHECK_CONN_ERR(ret);
donatien 0:2ccb9960a044 186
donatien 4:c071b05ac026 187 size_t trfLen;
donatien 9:ff30cc189191 188
donatien 0:2ccb9960a044 189 //Send data (if POST)
donatien 0:2ccb9960a044 190 if( (method == HTTP_POST) && (pDataOut != NULL) )
donatien 0:2ccb9960a044 191 {
donatien 0:2ccb9960a044 192 DBG("Sending data");
donatien 0:2ccb9960a044 193 while(true)
donatien 0:2ccb9960a044 194 {
donatien 0:2ccb9960a044 195 size_t writtenLen = 0;
donatien 0:2ccb9960a044 196 pDataOut->read(buf, CHUNK_SIZE, &trfLen);
donatien 0:2ccb9960a044 197 if( pDataOut->getIsChunked() )
donatien 0:2ccb9960a044 198 {
donatien 0:2ccb9960a044 199 //Write chunk header
donatien 9:ff30cc189191 200 char chunkHeader[16];
donatien 9:ff30cc189191 201 snprintf(chunkHeader, sizeof(chunkHeader), "%X\r\n", trfLen); //In hex encoding
donatien 9:ff30cc189191 202 ret = send(chunkHeader);
donatien 5:791fc3dcb6c4 203 CHECK_CONN_ERR(ret);
donatien 0:2ccb9960a044 204 }
donatien 0:2ccb9960a044 205 else if( trfLen == 0 )
donatien 0:2ccb9960a044 206 {
donatien 0:2ccb9960a044 207 break;
donatien 0:2ccb9960a044 208 }
donatien 0:2ccb9960a044 209 if( trfLen != 0 )
donatien 0:2ccb9960a044 210 {
donatien 0:2ccb9960a044 211 ret = send(buf, trfLen);
donatien 5:791fc3dcb6c4 212 CHECK_CONN_ERR(ret);
donatien 0:2ccb9960a044 213 }
donatien 0:2ccb9960a044 214
donatien 0:2ccb9960a044 215 if( pDataOut->getIsChunked() )
donatien 0:2ccb9960a044 216 {
donatien 0:2ccb9960a044 217 ret = send("\r\n"); //Chunk-terminating CRLF
donatien 5:791fc3dcb6c4 218 CHECK_CONN_ERR(ret);
donatien 0:2ccb9960a044 219 }
donatien 0:2ccb9960a044 220 else
donatien 0:2ccb9960a044 221 {
donatien 0:2ccb9960a044 222 writtenLen += trfLen;
donatien 0:2ccb9960a044 223 if( writtenLen >= pDataOut->getDataLen() )
donatien 0:2ccb9960a044 224 {
donatien 0:2ccb9960a044 225 break;
donatien 0:2ccb9960a044 226 }
donatien 0:2ccb9960a044 227 }
donatien 0:2ccb9960a044 228
donatien 0:2ccb9960a044 229 if( trfLen == 0 )
donatien 0:2ccb9960a044 230 {
donatien 0:2ccb9960a044 231 break;
donatien 0:2ccb9960a044 232 }
donatien 0:2ccb9960a044 233 }
donatien 0:2ccb9960a044 234
donatien 0:2ccb9960a044 235 }
donatien 9:ff30cc189191 236
donatien 0:2ccb9960a044 237 //Receive response
donatien 0:2ccb9960a044 238 DBG("Receiving response");
donatien 4:c071b05ac026 239 ret = recv(buf, CHUNK_SIZE - 1, CHUNK_SIZE - 1, &trfLen); //Read n bytes
donatien 5:791fc3dcb6c4 240 CHECK_CONN_ERR(ret);
donatien 0:2ccb9960a044 241
donatien 0:2ccb9960a044 242 buf[trfLen] = '\0';
donatien 0:2ccb9960a044 243
donatien 0:2ccb9960a044 244 char* crlfPtr = strstr(buf, "\r\n");
donatien 0:2ccb9960a044 245 if(crlfPtr == NULL)
donatien 0:2ccb9960a044 246 {
donatien 5:791fc3dcb6c4 247 PRTCL_ERR();
donatien 0:2ccb9960a044 248 }
donatien 0:2ccb9960a044 249
donatien 4:c071b05ac026 250 int crlfPos = crlfPtr - buf;
donatien 0:2ccb9960a044 251 buf[crlfPos] = '\0';
donatien 0:2ccb9960a044 252
donatien 0:2ccb9960a044 253 //Parse HTTP response
donatien 0:2ccb9960a044 254 if( sscanf(buf, "HTTP/%*d.%*d %d %*[^\r\n]", &m_httpResponseCode) != 1 )
donatien 0:2ccb9960a044 255 {
donatien 0:2ccb9960a044 256 //Cannot match string, error
donatien 0:2ccb9960a044 257 ERR("Not a correct HTTP answer : %s\n", buf);
donatien 5:791fc3dcb6c4 258 PRTCL_ERR();
donatien 0:2ccb9960a044 259 }
donatien 0:2ccb9960a044 260
donatien 0:2ccb9960a044 261 if(m_httpResponseCode != 200)
donatien 0:2ccb9960a044 262 {
donatien 0:2ccb9960a044 263 //Cannot match string, error
donatien 0:2ccb9960a044 264 WARN("Response code %d", m_httpResponseCode);
donatien 5:791fc3dcb6c4 265 PRTCL_ERR();
donatien 0:2ccb9960a044 266 }
donatien 0:2ccb9960a044 267
donatien 0:2ccb9960a044 268 DBG("Reading headers");
donatien 0:2ccb9960a044 269
donatien 4:c071b05ac026 270 memmove(buf, &buf[crlfPos+2], trfLen - (crlfPos + 2) + 1); //Be sure to move NULL-terminating char as well
donatien 0:2ccb9960a044 271 trfLen -= (crlfPos + 2);
donatien 0:2ccb9960a044 272
donatien 4:c071b05ac026 273 size_t recvContentLength = 0;
donatien 4:c071b05ac026 274 bool recvChunked = false;
donatien 0:2ccb9960a044 275 //Now get headers
donatien 0:2ccb9960a044 276 while( true )
donatien 0:2ccb9960a044 277 {
donatien 0:2ccb9960a044 278 crlfPtr = strstr(buf, "\r\n");
donatien 0:2ccb9960a044 279 if(crlfPtr == NULL)
donatien 0:2ccb9960a044 280 {
donatien 4:c071b05ac026 281 if( trfLen < CHUNK_SIZE - 1 )
donatien 0:2ccb9960a044 282 {
donatien 0:2ccb9960a044 283 size_t newTrfLen;
donatien 0:2ccb9960a044 284 ret = recv(buf + trfLen, 1, CHUNK_SIZE - trfLen - 1, &newTrfLen);
donatien 0:2ccb9960a044 285 trfLen += newTrfLen;
donatien 0:2ccb9960a044 286 buf[trfLen] = '\0';
donatien 4:c071b05ac026 287 DBG("Read %d chars; In buf: [%s]", newTrfLen, buf);
donatien 5:791fc3dcb6c4 288 CHECK_CONN_ERR(ret);
donatien 0:2ccb9960a044 289 continue;
donatien 0:2ccb9960a044 290 }
donatien 0:2ccb9960a044 291 else
donatien 0:2ccb9960a044 292 {
donatien 5:791fc3dcb6c4 293 PRTCL_ERR();
donatien 0:2ccb9960a044 294 }
donatien 0:2ccb9960a044 295 }
donatien 0:2ccb9960a044 296
donatien 0:2ccb9960a044 297 crlfPos = crlfPtr - buf;
donatien 0:2ccb9960a044 298
donatien 0:2ccb9960a044 299 if(crlfPos == 0) //End of headers
donatien 0:2ccb9960a044 300 {
donatien 0:2ccb9960a044 301 DBG("Headers read");
donatien 4:c071b05ac026 302 memmove(buf, &buf[2], trfLen - 2 + 1); //Be sure to move NULL-terminating char as well
donatien 0:2ccb9960a044 303 trfLen -= 2;
donatien 0:2ccb9960a044 304 break;
donatien 0:2ccb9960a044 305 }
donatien 0:2ccb9960a044 306
donatien 0:2ccb9960a044 307 buf[crlfPos] = '\0';
donatien 0:2ccb9960a044 308
donatien 4:c071b05ac026 309 char key[32];
donatien 4:c071b05ac026 310 char value[32];
donatien 0:2ccb9960a044 311
donatien 4:c071b05ac026 312 key[31] = '\0';
donatien 4:c071b05ac026 313 value[31] = '\0';
donatien 4:c071b05ac026 314
donatien 4:c071b05ac026 315 int n = sscanf(buf, "%31[^:]: %31[^\r\n]", key, value);
donatien 0:2ccb9960a044 316 if ( n == 2 )
donatien 0:2ccb9960a044 317 {
donatien 0:2ccb9960a044 318 DBG("Read header : %s: %s\n", key, value);
donatien 0:2ccb9960a044 319 if( !strcmp(key, "Content-Length") )
donatien 0:2ccb9960a044 320 {
donatien 0:2ccb9960a044 321 sscanf(value, "%d", &recvContentLength);
donatien 0:2ccb9960a044 322 pDataIn->setDataLen(recvContentLength);
donatien 0:2ccb9960a044 323 }
donatien 0:2ccb9960a044 324 else if( !strcmp(key, "Transfer-Encoding") )
donatien 0:2ccb9960a044 325 {
donatien 0:2ccb9960a044 326 if( !strcmp(value, "Chunked") || !strcmp(value, "chunked") )
donatien 0:2ccb9960a044 327 {
donatien 0:2ccb9960a044 328 recvChunked = true;
donatien 0:2ccb9960a044 329 pDataIn->setIsChunked(true);
donatien 0:2ccb9960a044 330 }
donatien 0:2ccb9960a044 331 }
donatien 0:2ccb9960a044 332 else if( !strcmp(key, "Content-Type") )
donatien 0:2ccb9960a044 333 {
donatien 0:2ccb9960a044 334 pDataIn->setDataType(value);
donatien 0:2ccb9960a044 335 }
donatien 0:2ccb9960a044 336
donatien 4:c071b05ac026 337 memmove(buf, &buf[crlfPos+2], trfLen - (crlfPos + 2) + 1); //Be sure to move NULL-terminating char as well
donatien 0:2ccb9960a044 338 trfLen -= (crlfPos + 2);
donatien 0:2ccb9960a044 339
donatien 0:2ccb9960a044 340 }
donatien 0:2ccb9960a044 341 else
donatien 0:2ccb9960a044 342 {
donatien 0:2ccb9960a044 343 ERR("Could not parse header");
donatien 5:791fc3dcb6c4 344 PRTCL_ERR();
donatien 0:2ccb9960a044 345 }
donatien 0:2ccb9960a044 346
donatien 0:2ccb9960a044 347 }
donatien 0:2ccb9960a044 348
donatien 0:2ccb9960a044 349 //Receive data
donatien 0:2ccb9960a044 350 DBG("Receiving data");
donatien 0:2ccb9960a044 351 while(true)
donatien 0:2ccb9960a044 352 {
donatien 0:2ccb9960a044 353 size_t readLen = 0;
donatien 0:2ccb9960a044 354
donatien 0:2ccb9960a044 355 if( recvChunked )
donatien 0:2ccb9960a044 356 {
donatien 0:2ccb9960a044 357 //Read chunk header
donatien 0:2ccb9960a044 358 crlfPos=0;
donatien 0:2ccb9960a044 359 for(crlfPos++; crlfPos < trfLen - 2; crlfPos++)
donatien 0:2ccb9960a044 360 {
donatien 0:2ccb9960a044 361 if( buf[crlfPos] == '\r' && buf[crlfPos + 1] == '\n' )
donatien 0:2ccb9960a044 362 {
donatien 0:2ccb9960a044 363 break;
donatien 0:2ccb9960a044 364 }
donatien 0:2ccb9960a044 365 }
donatien 0:2ccb9960a044 366 if(crlfPos >= trfLen - 2) //Try to read more
donatien 0:2ccb9960a044 367 {
donatien 0:2ccb9960a044 368 if( trfLen < CHUNK_SIZE )
donatien 0:2ccb9960a044 369 {
donatien 0:2ccb9960a044 370 size_t newTrfLen;
donatien 0:2ccb9960a044 371 ret = recv(buf + trfLen, 0, CHUNK_SIZE - trfLen - 1, &newTrfLen);
donatien 0:2ccb9960a044 372 trfLen += newTrfLen;
donatien 5:791fc3dcb6c4 373 CHECK_CONN_ERR(ret);
donatien 0:2ccb9960a044 374 continue;
donatien 0:2ccb9960a044 375 }
donatien 0:2ccb9960a044 376 else
donatien 0:2ccb9960a044 377 {
donatien 5:791fc3dcb6c4 378 PRTCL_ERR();
donatien 0:2ccb9960a044 379 }
donatien 0:2ccb9960a044 380 }
donatien 0:2ccb9960a044 381 buf[crlfPos] = '\0';
donatien 0:2ccb9960a044 382 int n = sscanf(buf, "%x", &readLen);
donatien 0:2ccb9960a044 383 if(n!=1)
donatien 0:2ccb9960a044 384 {
donatien 0:2ccb9960a044 385 ERR("Could not read chunk length");
donatien 5:791fc3dcb6c4 386 PRTCL_ERR();
donatien 0:2ccb9960a044 387 }
donatien 0:2ccb9960a044 388
donatien 4:c071b05ac026 389 memmove(buf, &buf[crlfPos+2], trfLen - (crlfPos + 2)); //Not need to move NULL-terminating char any more
donatien 0:2ccb9960a044 390 trfLen -= (crlfPos + 2);
donatien 0:2ccb9960a044 391
donatien 0:2ccb9960a044 392 if( readLen == 0 )
donatien 0:2ccb9960a044 393 {
donatien 0:2ccb9960a044 394 //Last chunk
donatien 0:2ccb9960a044 395 break;
donatien 0:2ccb9960a044 396 }
donatien 0:2ccb9960a044 397 }
donatien 0:2ccb9960a044 398 else
donatien 0:2ccb9960a044 399 {
donatien 0:2ccb9960a044 400 readLen = recvContentLength;
donatien 0:2ccb9960a044 401 }
donatien 0:2ccb9960a044 402
donatien 0:2ccb9960a044 403 DBG("Retrieving %d bytes", readLen);
donatien 0:2ccb9960a044 404
donatien 0:2ccb9960a044 405 do
donatien 0:2ccb9960a044 406 {
donatien 0:2ccb9960a044 407 pDataIn->write(buf, MIN(trfLen, readLen));
donatien 0:2ccb9960a044 408 if( trfLen > readLen )
donatien 0:2ccb9960a044 409 {
donatien 0:2ccb9960a044 410 memmove(buf, &buf[readLen], trfLen - readLen);
donatien 0:2ccb9960a044 411 trfLen -= readLen;
donatien 0:2ccb9960a044 412 readLen = 0;
donatien 0:2ccb9960a044 413 }
donatien 0:2ccb9960a044 414 else
donatien 0:2ccb9960a044 415 {
donatien 0:2ccb9960a044 416 readLen -= trfLen;
donatien 0:2ccb9960a044 417 }
donatien 0:2ccb9960a044 418
donatien 0:2ccb9960a044 419 if(readLen)
donatien 0:2ccb9960a044 420 {
donatien 0:2ccb9960a044 421 ret = recv(buf, 1, CHUNK_SIZE - trfLen - 1, &trfLen);
donatien 5:791fc3dcb6c4 422 CHECK_CONN_ERR(ret);
donatien 0:2ccb9960a044 423 }
donatien 0:2ccb9960a044 424 } while(readLen);
donatien 0:2ccb9960a044 425
donatien 0:2ccb9960a044 426 if( recvChunked )
donatien 0:2ccb9960a044 427 {
donatien 0:2ccb9960a044 428 if(trfLen < 2)
donatien 0:2ccb9960a044 429 {
donatien 0:2ccb9960a044 430 size_t newTrfLen;
donatien 0:2ccb9960a044 431 //Read missing chars to find end of chunk
donatien 0:2ccb9960a044 432 ret = recv(buf, 2 - trfLen, CHUNK_SIZE, &newTrfLen);
donatien 5:791fc3dcb6c4 433 CHECK_CONN_ERR(ret);
donatien 0:2ccb9960a044 434 trfLen += newTrfLen;
donatien 0:2ccb9960a044 435 }
donatien 0:2ccb9960a044 436 if( (buf[0] != '\r') || (buf[1] != '\n') )
donatien 0:2ccb9960a044 437 {
donatien 0:2ccb9960a044 438 ERR("Format error");
donatien 5:791fc3dcb6c4 439 PRTCL_ERR();
donatien 0:2ccb9960a044 440 }
donatien 0:2ccb9960a044 441 memmove(buf, &buf[2], trfLen - 2);
donatien 0:2ccb9960a044 442 trfLen -= 2;
donatien 0:2ccb9960a044 443 }
donatien 0:2ccb9960a044 444 else
donatien 0:2ccb9960a044 445 {
donatien 0:2ccb9960a044 446 break;
donatien 0:2ccb9960a044 447 }
donatien 0:2ccb9960a044 448
donatien 0:2ccb9960a044 449 }
donatien 0:2ccb9960a044 450
donatien 7:4e39864f7b15 451 m_sock.close();
donatien 0:2ccb9960a044 452 DBG("Completed HTTP transaction");
donatien 0:2ccb9960a044 453
donatien 11:390362de8c3f 454 return HTTP_OK;
donatien 0:2ccb9960a044 455 }
donatien 0:2ccb9960a044 456
donatien 11:390362de8c3f 457 HTTPResult HTTPClient::recv(char* buf, size_t minLen, size_t maxLen, size_t* pReadLen) //0 on success, err code on failure
donatien 0:2ccb9960a044 458 {
donatien 0:2ccb9960a044 459 DBG("Trying to read between %d and %d bytes", minLen, maxLen);
donatien 0:2ccb9960a044 460 size_t readLen = 0;
donatien 12:89d09a6db00a 461
donatien 12:89d09a6db00a 462 if(!m_sock.is_connected())
donatien 12:89d09a6db00a 463 {
donatien 12:89d09a6db00a 464 WARN("Connection was closed by server");
donatien 12:89d09a6db00a 465 return HTTP_CLOSED; //Connection was closed by server
donatien 12:89d09a6db00a 466 }
donatien 12:89d09a6db00a 467
donatien 7:4e39864f7b15 468 int ret;
donatien 7:4e39864f7b15 469 while(readLen < maxLen)
donatien 0:2ccb9960a044 470 {
donatien 7:4e39864f7b15 471 if(readLen < minLen)
donatien 0:2ccb9960a044 472 {
donatien 12:89d09a6db00a 473 DBG("Trying to read at most %d bytes [Blocking]", minLen - readLen);
donatien 13:be61104f4e91 474 m_sock.set_blocking(false, m_timeout);
donatien 12:89d09a6db00a 475 ret = m_sock.receive_all(buf + readLen, minLen - readLen);
donatien 0:2ccb9960a044 476 }
donatien 7:4e39864f7b15 477 else
donatien 7:4e39864f7b15 478 {
donatien 12:89d09a6db00a 479 DBG("Trying to read at most %d bytes [Not blocking]", maxLen - readLen);
donatien 13:be61104f4e91 480 m_sock.set_blocking(false, 0);
donatien 12:89d09a6db00a 481 ret = m_sock.receive(buf + readLen, maxLen - readLen);
donatien 7:4e39864f7b15 482 }
donatien 7:4e39864f7b15 483
donatien 0:2ccb9960a044 484 if( ret > 0)
donatien 0:2ccb9960a044 485 {
donatien 0:2ccb9960a044 486 readLen += ret;
donatien 0:2ccb9960a044 487 }
donatien 0:2ccb9960a044 488 else if( ret == 0 )
donatien 0:2ccb9960a044 489 {
donatien 7:4e39864f7b15 490 break;
donatien 7:4e39864f7b15 491 }
donatien 7:4e39864f7b15 492 else
donatien 7:4e39864f7b15 493 {
donatien 12:89d09a6db00a 494 if(!m_sock.is_connected())
donatien 12:89d09a6db00a 495 {
donatien 12:89d09a6db00a 496 ERR("Connection error (recv returned %d)", ret);
donatien 12:89d09a6db00a 497 *pReadLen = readLen;
donatien 12:89d09a6db00a 498 return HTTP_CONN;
donatien 12:89d09a6db00a 499 }
donatien 12:89d09a6db00a 500 else
donatien 12:89d09a6db00a 501 {
donatien 12:89d09a6db00a 502 break;
donatien 12:89d09a6db00a 503 }
donatien 12:89d09a6db00a 504 }
donatien 12:89d09a6db00a 505
donatien 12:89d09a6db00a 506 if(!m_sock.is_connected())
donatien 12:89d09a6db00a 507 {
donatien 12:89d09a6db00a 508 break;
donatien 7:4e39864f7b15 509 }
donatien 7:4e39864f7b15 510 }
donatien 7:4e39864f7b15 511 DBG("Read %d bytes", readLen);
donatien 7:4e39864f7b15 512 *pReadLen = readLen;
donatien 11:390362de8c3f 513 return HTTP_OK;
donatien 7:4e39864f7b15 514 }
donatien 7:4e39864f7b15 515
donatien 11:390362de8c3f 516 HTTPResult HTTPClient::send(char* buf, size_t len) //0 on success, err code on failure
donatien 7:4e39864f7b15 517 {
donatien 7:4e39864f7b15 518 if(len == 0)
donatien 7:4e39864f7b15 519 {
donatien 7:4e39864f7b15 520 len = strlen(buf);
donatien 7:4e39864f7b15 521 }
donatien 7:4e39864f7b15 522 DBG("Trying to write %d bytes", len);
donatien 7:4e39864f7b15 523 size_t writtenLen = 0;
donatien 12:89d09a6db00a 524
donatien 12:89d09a6db00a 525 if(!m_sock.is_connected())
donatien 12:89d09a6db00a 526 {
donatien 12:89d09a6db00a 527 WARN("Connection was closed by server");
donatien 12:89d09a6db00a 528 return HTTP_CLOSED; //Connection was closed by server
donatien 12:89d09a6db00a 529 }
donatien 7:4e39864f7b15 530
donatien 13:be61104f4e91 531 m_sock.set_blocking(false, m_timeout);
donatien 12:89d09a6db00a 532 int ret = m_sock.send_all(buf, len);
donatien 12:89d09a6db00a 533 if(ret > 0)
donatien 7:4e39864f7b15 534 {
donatien 12:89d09a6db00a 535 writtenLen += ret;
donatien 12:89d09a6db00a 536 }
donatien 12:89d09a6db00a 537 else if( ret == 0 )
donatien 12:89d09a6db00a 538 {
donatien 12:89d09a6db00a 539 WARN("Connection was closed by server");
donatien 12:89d09a6db00a 540 return HTTP_CLOSED; //Connection was closed by server
donatien 12:89d09a6db00a 541 }
donatien 12:89d09a6db00a 542 else
donatien 12:89d09a6db00a 543 {
donatien 12:89d09a6db00a 544 ERR("Connection error (send returned %d)", ret);
donatien 12:89d09a6db00a 545 return HTTP_CONN;
donatien 12:89d09a6db00a 546 }
donatien 7:4e39864f7b15 547
donatien 0:2ccb9960a044 548 DBG("Written %d bytes", writtenLen);
donatien 11:390362de8c3f 549 return HTTP_OK;
donatien 0:2ccb9960a044 550 }
donatien 0:2ccb9960a044 551
donatien 11:390362de8c3f 552 HTTPResult HTTPClient::parseURL(const char* url, char* scheme, size_t maxSchemeLen, char* host, size_t maxHostLen, uint16_t* port, char* path, size_t maxPathLen) //Parse URL
donatien 0:2ccb9960a044 553 {
donatien 0:2ccb9960a044 554 char* schemePtr = (char*) url;
donatien 0:2ccb9960a044 555 char* hostPtr = (char*) strstr(url, "://");
donatien 0:2ccb9960a044 556 if(hostPtr == NULL)
donatien 0:2ccb9960a044 557 {
donatien 0:2ccb9960a044 558 WARN("Could not find host");
donatien 11:390362de8c3f 559 return HTTP_PARSE; //URL is invalid
donatien 0:2ccb9960a044 560 }
donatien 0:2ccb9960a044 561
donatien 0:2ccb9960a044 562 if( maxSchemeLen < hostPtr - schemePtr + 1 ) //including NULL-terminating char
donatien 0:2ccb9960a044 563 {
donatien 0:2ccb9960a044 564 WARN("Scheme str is too small (%d >= %d)", maxSchemeLen, hostPtr - schemePtr + 1);
donatien 11:390362de8c3f 565 return HTTP_PARSE;
donatien 0:2ccb9960a044 566 }
donatien 0:2ccb9960a044 567 memcpy(scheme, schemePtr, hostPtr - schemePtr);
donatien 0:2ccb9960a044 568 scheme[hostPtr - schemePtr] = '\0';
donatien 0:2ccb9960a044 569
donatien 0:2ccb9960a044 570 hostPtr+=3;
donatien 0:2ccb9960a044 571
donatien 0:2ccb9960a044 572 size_t hostLen = 0;
donatien 0:2ccb9960a044 573
donatien 0:2ccb9960a044 574 char* portPtr = strchr(hostPtr, ':');
donatien 0:2ccb9960a044 575 if( portPtr != NULL )
donatien 0:2ccb9960a044 576 {
donatien 0:2ccb9960a044 577 hostLen = portPtr - hostPtr;
donatien 0:2ccb9960a044 578 portPtr++;
donatien 8:45c8da29a1cf 579 if( sscanf(portPtr, "%hu", port) != 1)
donatien 0:2ccb9960a044 580 {
donatien 0:2ccb9960a044 581 WARN("Could not find port");
donatien 11:390362de8c3f 582 return HTTP_PARSE;
donatien 0:2ccb9960a044 583 }
donatien 0:2ccb9960a044 584 }
donatien 0:2ccb9960a044 585 else
donatien 0:2ccb9960a044 586 {
donatien 0:2ccb9960a044 587 *port=0;
donatien 0:2ccb9960a044 588 }
donatien 0:2ccb9960a044 589 char* pathPtr = strchr(hostPtr, '/');
donatien 0:2ccb9960a044 590 if( hostLen == 0 )
donatien 0:2ccb9960a044 591 {
donatien 0:2ccb9960a044 592 hostLen = pathPtr - hostPtr;
donatien 0:2ccb9960a044 593 }
donatien 0:2ccb9960a044 594
donatien 0:2ccb9960a044 595 if( maxHostLen < hostLen + 1 ) //including NULL-terminating char
donatien 0:2ccb9960a044 596 {
donatien 0:2ccb9960a044 597 WARN("Host str is too small (%d >= %d)", maxHostLen, hostLen + 1);
donatien 11:390362de8c3f 598 return HTTP_PARSE;
donatien 0:2ccb9960a044 599 }
donatien 0:2ccb9960a044 600 memcpy(host, hostPtr, hostLen);
donatien 0:2ccb9960a044 601 host[hostLen] = '\0';
donatien 0:2ccb9960a044 602
donatien 0:2ccb9960a044 603 size_t pathLen;
donatien 0:2ccb9960a044 604 char* fragmentPtr = strchr(hostPtr, '#');
donatien 0:2ccb9960a044 605 if(fragmentPtr != NULL)
donatien 0:2ccb9960a044 606 {
donatien 0:2ccb9960a044 607 pathLen = fragmentPtr - pathPtr;
donatien 0:2ccb9960a044 608 }
donatien 0:2ccb9960a044 609 else
donatien 0:2ccb9960a044 610 {
donatien 0:2ccb9960a044 611 pathLen = strlen(pathPtr);
donatien 0:2ccb9960a044 612 }
donatien 0:2ccb9960a044 613
donatien 0:2ccb9960a044 614 if( maxPathLen < pathLen + 1 ) //including NULL-terminating char
donatien 0:2ccb9960a044 615 {
donatien 0:2ccb9960a044 616 WARN("Path str is too small (%d >= %d)", maxPathLen, pathLen + 1);
donatien 11:390362de8c3f 617 return HTTP_PARSE;
donatien 0:2ccb9960a044 618 }
donatien 0:2ccb9960a044 619 memcpy(path, pathPtr, pathLen);
donatien 0:2ccb9960a044 620 path[pathLen] = '\0';
donatien 0:2ccb9960a044 621
donatien 11:390362de8c3f 622 return HTTP_OK;
donatien 0:2ccb9960a044 623 }