Fang Bao / HTTPClient

HTTPClient

Fork of HTTPClient by Donatien Garnier

HTTPClient.cpp@4:c071b05ac026, 2012-05-24 (annotated)

Committer:
donatien
Date:
Thu May 24 16:27:32 2012 +0000
Revision:
4:c071b05ac026
Parent:
3:5aef7584de4b
Child:
5:791fc3dcb6c4
FIX

Who changed what in which revision?

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