LRSD stephane / Mbed 2 deprecated WEBserver0-lrsd

Dependencies:   mbed

Committer:
geiineuville
Date:
Fri Sep 02 08:36:24 2011 +0000
Revision:
0:441400ffd086
V0

Who changed what in which revision?

UserRevisionLine numberNew contents of line
geiineuville 0:441400ffd086 1 #include "HTTPClient.h"
geiineuville 0:441400ffd086 2 #include "NetServer.h"
geiineuville 0:441400ffd086 3 #include "iputil.h"
geiineuville 0:441400ffd086 4
geiineuville 0:441400ffd086 5 using namespace mbed;
geiineuville 0:441400ffd086 6 using namespace std;
geiineuville 0:441400ffd086 7
geiineuville 0:441400ffd086 8 #define POST 0x1
geiineuville 0:441400ffd086 9 #define FDATA 0x2
geiineuville 0:441400ffd086 10 #define FRES 0x4
geiineuville 0:441400ffd086 11 #define GET 0x0
geiineuville 0:441400ffd086 12 #define CDATA 0x0
geiineuville 0:441400ffd086 13 #define CRES 0x0
geiineuville 0:441400ffd086 14
geiineuville 0:441400ffd086 15 long fleft(FILE *fd) {
geiineuville 0:441400ffd086 16 long len, cur;
geiineuville 0:441400ffd086 17 cur = ftell(fd);
geiineuville 0:441400ffd086 18 fseek(fd, 0, SEEK_END);
geiineuville 0:441400ffd086 19 len = ftell(fd);
geiineuville 0:441400ffd086 20 fseek(fd, cur, SEEK_SET);
geiineuville 0:441400ffd086 21 return len;
geiineuville 0:441400ffd086 22 }
geiineuville 0:441400ffd086 23
geiineuville 0:441400ffd086 24
geiineuville 0:441400ffd086 25 HTTPClient::HTTPClient(const char *hostname, struct ip_addr ip, struct ip_addr nm, struct ip_addr gw, struct ip_addr dns)
geiineuville 0:441400ffd086 26 : TCPConnection(), _auth(NULL), _timeout(0), _data(NULL), _headerfields(NULL), _timeout_max(60000) {
geiineuville 0:441400ffd086 27 NetServer *net = NULL;
geiineuville 0:441400ffd086 28 if(ip.addr != ip_addr_any.addr && nm.addr != ip_addr_any.addr && gw.addr != ip_addr_any.addr) {
geiineuville 0:441400ffd086 29 net = NetServer::create(ip, nm, gw);
geiineuville 0:441400ffd086 30 if(dns.addr != ip_addr_any.addr) {
geiineuville 0:441400ffd086 31 net->setDNS1(dns);
geiineuville 0:441400ffd086 32 }
geiineuville 0:441400ffd086 33 } else if(hostname) {
geiineuville 0:441400ffd086 34 net = NetServer::create();
geiineuville 0:441400ffd086 35 }
geiineuville 0:441400ffd086 36 if(hostname) {
geiineuville 0:441400ffd086 37 net->setHostname(hostname);
geiineuville 0:441400ffd086 38 }
geiineuville 0:441400ffd086 39 }
geiineuville 0:441400ffd086 40
geiineuville 0:441400ffd086 41 void HTTPClient::timeout(int value) {
geiineuville 0:441400ffd086 42 _timeout_max = value;
geiineuville 0:441400ffd086 43 }
geiineuville 0:441400ffd086 44
geiineuville 0:441400ffd086 45 void HTTPClient::err(err_t err) {
geiineuville 0:441400ffd086 46 release_callbacks();
geiineuville 0:441400ffd086 47 }
geiineuville 0:441400ffd086 48
geiineuville 0:441400ffd086 49 err_t HTTPClient::poll() {
geiineuville 0:441400ffd086 50 if(NetServer::time() - _timeout > _timeout_max) {
geiineuville 0:441400ffd086 51 release_callbacks();
geiineuville 0:441400ffd086 52 close();
geiineuville 0:441400ffd086 53 printf("TIMEOUT\n");
geiineuville 0:441400ffd086 54 _ready = true;
geiineuville 0:441400ffd086 55 _state = 99;
geiineuville 0:441400ffd086 56 }
geiineuville 0:441400ffd086 57 return ERR_OK;
geiineuville 0:441400ffd086 58 }
geiineuville 0:441400ffd086 59
geiineuville 0:441400ffd086 60 void HTTPClient::dnsreply(const char *hostname, struct ip_addr *ipaddr) {
geiineuville 0:441400ffd086 61 _ready = true;
geiineuville 0:441400ffd086 62 _ipaddr = *ipaddr;
geiineuville 0:441400ffd086 63 _state = (ipaddr==NULL)? 99 : 0;
geiineuville 0:441400ffd086 64 _timeout = NetServer::time();
geiineuville 0:441400ffd086 65 }
geiineuville 0:441400ffd086 66
geiineuville 0:441400ffd086 67 err_t HTTPClient::connected(err_t err) {
geiineuville 0:441400ffd086 68 TCPConnection::connected(err);
geiineuville 0:441400ffd086 69 _ready = false;
geiineuville 0:441400ffd086 70 _state = 0;
geiineuville 0:441400ffd086 71 _resultoff = 0;
geiineuville 0:441400ffd086 72 if(_mode&POST) {
geiineuville 0:441400ffd086 73 write((void *)"POST ", 5, TCP_WRITE_FLAG_COPY | TCP_WRITE_FLAG_MORE);
geiineuville 0:441400ffd086 74 } else {
geiineuville 0:441400ffd086 75 write((void *)"GET ", 4, TCP_WRITE_FLAG_COPY | TCP_WRITE_FLAG_MORE);
geiineuville 0:441400ffd086 76 }
geiineuville 0:441400ffd086 77 if(strlen(_path) == 0) {
geiineuville 0:441400ffd086 78 write((void *)"/", 1, TCP_WRITE_FLAG_COPY | TCP_WRITE_FLAG_MORE);
geiineuville 0:441400ffd086 79 } else {
geiineuville 0:441400ffd086 80 write((void *)_path, strlen(_path), TCP_WRITE_FLAG_COPY | TCP_WRITE_FLAG_MORE);
geiineuville 0:441400ffd086 81 }
geiineuville 0:441400ffd086 82 write((void *)" HTTP/1.1\r\nHost: ", 17, TCP_WRITE_FLAG_COPY | TCP_WRITE_FLAG_MORE);
geiineuville 0:441400ffd086 83 write((void *)_host, _hostlen, TCP_WRITE_FLAG_COPY | TCP_WRITE_FLAG_MORE);
geiineuville 0:441400ffd086 84 if(_auth&&(*_auth!='\0')) {
geiineuville 0:441400ffd086 85 write((void *)"\r\nAuthorization: Basic ", 23, TCP_WRITE_FLAG_COPY | TCP_WRITE_FLAG_MORE);
geiineuville 0:441400ffd086 86 write((void *)_auth, strlen(_auth), TCP_WRITE_FLAG_COPY |TCP_WRITE_FLAG_MORE);
geiineuville 0:441400ffd086 87 }
geiineuville 0:441400ffd086 88 if(_headerfields&&(*_headerfields!='\0')) {
geiineuville 0:441400ffd086 89 write((void *)"\r\n", 2, TCP_WRITE_FLAG_COPY | TCP_WRITE_FLAG_MORE);
geiineuville 0:441400ffd086 90 write((void *)_headerfields, strlen(_headerfields), TCP_WRITE_FLAG_COPY |TCP_WRITE_FLAG_MORE);
geiineuville 0:441400ffd086 91 }
geiineuville 0:441400ffd086 92 write((void *)"\r\nConnection: Close\r\n", 21, TCP_WRITE_FLAG_COPY | TCP_WRITE_FLAG_COPY);
geiineuville 0:441400ffd086 93 if(_data) {
geiineuville 0:441400ffd086 94 char clen[256];
geiineuville 0:441400ffd086 95 int len, blen;
geiineuville 0:441400ffd086 96 if(_mode&FDATA) {
geiineuville 0:441400ffd086 97 //printf("Send file\n");
geiineuville 0:441400ffd086 98 len = fleft((FILE *)_data);
geiineuville 0:441400ffd086 99 sprintf(clen, "Content-Length: %d\r\n\r\n\0", len);
geiineuville 0:441400ffd086 100 write((void *)clen, strlen(clen), TCP_WRITE_FLAG_COPY | TCP_WRITE_FLAG_COPY);
geiineuville 0:441400ffd086 101 if(len) {
geiineuville 0:441400ffd086 102 do {
geiineuville 0:441400ffd086 103 int sb = sndbuf();
geiineuville 0:441400ffd086 104 blen = fread(clen, sizeof(char), (int)min(sb, 100), (FILE *)_data);
geiineuville 0:441400ffd086 105 write(clen, blen, (TCP_WRITE_FLAG_COPY | TCP_WRITE_FLAG_MORE));
geiineuville 0:441400ffd086 106 len -= blen;
geiineuville 0:441400ffd086 107 } while(len > 1 && blen);
geiineuville 0:441400ffd086 108 }
geiineuville 0:441400ffd086 109 } else {
geiineuville 0:441400ffd086 110 len = strlen((const char *)_data);
geiineuville 0:441400ffd086 111 sprintf(clen, "Content-Length: %d\r\n\r\n\0", len);
geiineuville 0:441400ffd086 112 write((void *)clen, strlen(clen), TCP_WRITE_FLAG_COPY | TCP_WRITE_FLAG_COPY);
geiineuville 0:441400ffd086 113 write((void *)_data, len, TCP_WRITE_FLAG_COPY | TCP_WRITE_FLAG_MORE);
geiineuville 0:441400ffd086 114 }
geiineuville 0:441400ffd086 115 }
geiineuville 0:441400ffd086 116 write((void *)"\r\n", 2, TCP_WRITE_FLAG_COPY);
geiineuville 0:441400ffd086 117 _timeout = NetServer::time();
geiineuville 0:441400ffd086 118 return ERR_OK;
geiineuville 0:441400ffd086 119 }
geiineuville 0:441400ffd086 120
geiineuville 0:441400ffd086 121 #define _min(x, y) (((x)<(y))?(x):(y))
geiineuville 0:441400ffd086 122 err_t HTTPClient::recv(struct pbuf *p, err_t err) {
geiineuville 0:441400ffd086 123 if(err == ERR_OK && p != NULL&&_state<10) {
geiineuville 0:441400ffd086 124 _timeout = NetServer::time();
geiineuville 0:441400ffd086 125 struct pbuf *q = p;
geiineuville 0:441400ffd086 126 char *d = (char *)q->payload;
geiineuville 0:441400ffd086 127 recved(p->tot_len);
geiineuville 0:441400ffd086 128
geiineuville 0:441400ffd086 129 while(q&&_state<10) {
geiineuville 0:441400ffd086 130 unsigned int end = ((unsigned int)(q->payload)+(unsigned int)(q->len));
geiineuville 0:441400ffd086 131 switch(_state) {
geiineuville 0:441400ffd086 132 case 0: {
geiineuville 0:441400ffd086 133 for(; _state==0 && ((unsigned int)d < end); d++) {
geiineuville 0:441400ffd086 134 if((*((char *)(d-0))=='\n')&&(*((char *)(d-1))=='\r')&&
geiineuville 0:441400ffd086 135 (*((char *)(d-2))=='\n')&&(*((char *)(d-3))=='\r')) {
geiineuville 0:441400ffd086 136 _state = 1;
geiineuville 0:441400ffd086 137 d += 1;
geiineuville 0:441400ffd086 138 break;
geiineuville 0:441400ffd086 139 }
geiineuville 0:441400ffd086 140 }
geiineuville 0:441400ffd086 141 } break;
geiineuville 0:441400ffd086 142 case 1: {
geiineuville 0:441400ffd086 143 if(_result) {
geiineuville 0:441400ffd086 144 if(_mode&FRES) {
geiineuville 0:441400ffd086 145 fwrite(d, sizeof(char), end - (unsigned int)d, (FILE *)_result);
geiineuville 0:441400ffd086 146 _resultoff += (end - (unsigned int)d);
geiineuville 0:441400ffd086 147 d = (char *)end;
geiineuville 0:441400ffd086 148 } else {
geiineuville 0:441400ffd086 149 unsigned int len = _min(_resultleft, (end-(unsigned int)d));
geiineuville 0:441400ffd086 150 memcpy((char *)_result + _resultoff, d, len);
geiineuville 0:441400ffd086 151 _resultleft -= len;
geiineuville 0:441400ffd086 152 _resultoff += len;
geiineuville 0:441400ffd086 153 d += len;
geiineuville 0:441400ffd086 154
geiineuville 0:441400ffd086 155 if(!_resultleft) {
geiineuville 0:441400ffd086 156 _state = 10;
geiineuville 0:441400ffd086 157 }
geiineuville 0:441400ffd086 158 }
geiineuville 0:441400ffd086 159 } else {
geiineuville 0:441400ffd086 160 _state = 10;
geiineuville 0:441400ffd086 161 }
geiineuville 0:441400ffd086 162 } break;
geiineuville 0:441400ffd086 163 default: {
geiineuville 0:441400ffd086 164 break;
geiineuville 0:441400ffd086 165 }
geiineuville 0:441400ffd086 166 }
geiineuville 0:441400ffd086 167 if(_state<10&&(unsigned int)d==end) {
geiineuville 0:441400ffd086 168 q = q->next;
geiineuville 0:441400ffd086 169 if(q) {
geiineuville 0:441400ffd086 170 d = static_cast<char *>(q->payload);
geiineuville 0:441400ffd086 171 }
geiineuville 0:441400ffd086 172 }
geiineuville 0:441400ffd086 173 }
geiineuville 0:441400ffd086 174 }
geiineuville 0:441400ffd086 175
geiineuville 0:441400ffd086 176 if(p!=NULL) {
geiineuville 0:441400ffd086 177 pbuf_free(p);
geiineuville 0:441400ffd086 178 }
geiineuville 0:441400ffd086 179
geiineuville 0:441400ffd086 180 if(_state>10||p==NULL||err!=ERR_OK) {
geiineuville 0:441400ffd086 181 release_callbacks();
geiineuville 0:441400ffd086 182 close();
geiineuville 0:441400ffd086 183 _ready = true;
geiineuville 0:441400ffd086 184 }
geiineuville 0:441400ffd086 185 return ERR_OK;
geiineuville 0:441400ffd086 186 }
geiineuville 0:441400ffd086 187
geiineuville 0:441400ffd086 188 unsigned int HTTPClient::make(const char *request) {
geiineuville 0:441400ffd086 189 _request = request;
geiineuville 0:441400ffd086 190 _resultoff = 0;
geiineuville 0:441400ffd086 191 _ready = false;
geiineuville 0:441400ffd086 192 _state = 0;
geiineuville 0:441400ffd086 193 _hostlen = 0;
geiineuville 0:441400ffd086 194 _port = 0;
geiineuville 0:441400ffd086 195 _timeout = NetServer::time();
geiineuville 0:441400ffd086 196 NetServer::ready();
geiineuville 0:441400ffd086 197
geiineuville 0:441400ffd086 198 int hostlen = 0;
geiineuville 0:441400ffd086 199 if(strlen(_request)<10||strncmp("http://", _request, 7)!=0) {
geiineuville 0:441400ffd086 200 printf("Only http requests are allowed\n");
geiineuville 0:441400ffd086 201 return 0;
geiineuville 0:441400ffd086 202 }
geiineuville 0:441400ffd086 203 _path = _host = _request + 6;
geiineuville 0:441400ffd086 204 _host++;
geiineuville 0:441400ffd086 205 while(!_state == 1) {
geiineuville 0:441400ffd086 206 switch(*(++_path)) {
geiineuville 0:441400ffd086 207 case ':':
geiineuville 0:441400ffd086 208 _port = atoi(_path+1);
geiineuville 0:441400ffd086 209 break;
geiineuville 0:441400ffd086 210 case '/':
geiineuville 0:441400ffd086 211 case '\0':
geiineuville 0:441400ffd086 212 _port = (_port)?_port:80;
geiineuville 0:441400ffd086 213 _state = 1;
geiineuville 0:441400ffd086 214 break;
geiineuville 0:441400ffd086 215 default:
geiineuville 0:441400ffd086 216 break;
geiineuville 0:441400ffd086 217 }
geiineuville 0:441400ffd086 218 if(!_port) {
geiineuville 0:441400ffd086 219 hostlen++;
geiineuville 0:441400ffd086 220 }
geiineuville 0:441400ffd086 221 if(!_state == 1) {
geiineuville 0:441400ffd086 222 _hostlen++;
geiineuville 0:441400ffd086 223 }
geiineuville 0:441400ffd086 224 }
geiineuville 0:441400ffd086 225 _state = 0;
geiineuville 0:441400ffd086 226
geiineuville 0:441400ffd086 227 if(hostlen>256) {
geiineuville 0:441400ffd086 228 printf("Hostname longer than allowed\n");
geiineuville 0:441400ffd086 229 return 0;
geiineuville 0:441400ffd086 230 }
geiineuville 0:441400ffd086 231
geiineuville 0:441400ffd086 232 char host[257];
geiineuville 0:441400ffd086 233 memcpy(host, _host, hostlen);
geiineuville 0:441400ffd086 234 host[hostlen] = 0;
geiineuville 0:441400ffd086 235 _timeout = NetServer::time();
geiineuville 0:441400ffd086 236 struct in_addr in;
geiineuville 0:441400ffd086 237 if(!inet_aton(host, &in)) {
geiineuville 0:441400ffd086 238 _ready = false;
geiineuville 0:441400ffd086 239 if(dnsrequest(host, &_ipaddr)==ERR_INPROGRESS) {
geiineuville 0:441400ffd086 240 while(!_ready) {
geiineuville 0:441400ffd086 241 NetServer::poll();
geiineuville 0:441400ffd086 242 wait_ms(10);
geiineuville 0:441400ffd086 243 }
geiineuville 0:441400ffd086 244 if(_state==99) {
geiineuville 0:441400ffd086 245 printf("Server not found\n");
geiineuville 0:441400ffd086 246 return 0;
geiineuville 0:441400ffd086 247 }
geiineuville 0:441400ffd086 248 }
geiineuville 0:441400ffd086 249 } else {
geiineuville 0:441400ffd086 250 _ipaddr.addr = in.s_addr;
geiineuville 0:441400ffd086 251 }
geiineuville 0:441400ffd086 252
geiineuville 0:441400ffd086 253 _ready = false;
geiineuville 0:441400ffd086 254 _timeout = NetServer::time();
geiineuville 0:441400ffd086 255 connect();
geiineuville 0:441400ffd086 256 set_poll_interval(10);
geiineuville 0:441400ffd086 257 tcp_setprio(_pcb, TCP_PRIO_MIN);
geiineuville 0:441400ffd086 258 while(!_ready) {
geiineuville 0:441400ffd086 259 NetServer::poll();
geiineuville 0:441400ffd086 260 wait_ms(10);
geiineuville 0:441400ffd086 261 }
geiineuville 0:441400ffd086 262 //release_callbacks();
geiineuville 0:441400ffd086 263 close();
geiineuville 0:441400ffd086 264 if(_state==99) {
geiineuville 0:441400ffd086 265 printf("Connection error\n");
geiineuville 0:441400ffd086 266 return 0;
geiineuville 0:441400ffd086 267 }
geiineuville 0:441400ffd086 268
geiineuville 0:441400ffd086 269 if(!_mode&FRES&&_result) {
geiineuville 0:441400ffd086 270 ((char *)_result)[_resultoff+1] = '\0';
geiineuville 0:441400ffd086 271 }
geiineuville 0:441400ffd086 272 return ((!_mode&FRES)&&!_result)?1:_resultoff;
geiineuville 0:441400ffd086 273
geiineuville 0:441400ffd086 274 }
geiineuville 0:441400ffd086 275
geiineuville 0:441400ffd086 276 void HTTPClient::auth(const char *user, const char *password) {
geiineuville 0:441400ffd086 277 if(user) {
geiineuville 0:441400ffd086 278 char up[256];
geiineuville 0:441400ffd086 279 sprintf(up, "%s:%s", user, password);
geiineuville 0:441400ffd086 280 _auth = new char[base64enc_len(up)+1];
geiineuville 0:441400ffd086 281 base64enc(up, strlen(up), _auth);
geiineuville 0:441400ffd086 282 } else if(_auth) {
geiineuville 0:441400ffd086 283 delete _auth;
geiineuville 0:441400ffd086 284 _auth = NULL;
geiineuville 0:441400ffd086 285 }
geiineuville 0:441400ffd086 286 }
geiineuville 0:441400ffd086 287
geiineuville 0:441400ffd086 288 void HTTPClient::headers(const char *fields) {
geiineuville 0:441400ffd086 289 _headerfields = fields;
geiineuville 0:441400ffd086 290 }
geiineuville 0:441400ffd086 291
geiineuville 0:441400ffd086 292 unsigned int HTTPClient::get(const char *url, char *result, int rsize) {
geiineuville 0:441400ffd086 293 _mode = GET | CDATA | CRES;
geiineuville 0:441400ffd086 294 _data = (void *)NULL;
geiineuville 0:441400ffd086 295 _result = (void *)result;
geiineuville 0:441400ffd086 296 _resultleft = rsize -1;
geiineuville 0:441400ffd086 297
geiineuville 0:441400ffd086 298 return make(url);
geiineuville 0:441400ffd086 299 }
geiineuville 0:441400ffd086 300
geiineuville 0:441400ffd086 301 unsigned int HTTPClient::get(const char *url, FILE *result) {
geiineuville 0:441400ffd086 302 _mode = GET | CDATA | FRES;
geiineuville 0:441400ffd086 303 _data = (void *)NULL;
geiineuville 0:441400ffd086 304 _result = (void *)result;
geiineuville 0:441400ffd086 305 _resultleft = 0;
geiineuville 0:441400ffd086 306
geiineuville 0:441400ffd086 307 return make(url);
geiineuville 0:441400ffd086 308 }
geiineuville 0:441400ffd086 309
geiineuville 0:441400ffd086 310 unsigned int HTTPClient::post(const char *url, const char *data, char *result, int rsize) {
geiineuville 0:441400ffd086 311 _mode = POST | CDATA | CRES;
geiineuville 0:441400ffd086 312 _data = (void *)data;
geiineuville 0:441400ffd086 313 _result = (void *)result;
geiineuville 0:441400ffd086 314 _resultleft = rsize -1;
geiineuville 0:441400ffd086 315
geiineuville 0:441400ffd086 316 return make(url);
geiineuville 0:441400ffd086 317 }
geiineuville 0:441400ffd086 318
geiineuville 0:441400ffd086 319 unsigned int HTTPClient::post(const char *url, const char *data, FILE *result) {
geiineuville 0:441400ffd086 320 _mode = POST | CDATA | FRES;
geiineuville 0:441400ffd086 321 _data = (void *)data;
geiineuville 0:441400ffd086 322 _result = (void *)result;
geiineuville 0:441400ffd086 323 _resultleft = 0;
geiineuville 0:441400ffd086 324
geiineuville 0:441400ffd086 325 return make(url);
geiineuville 0:441400ffd086 326 }
geiineuville 0:441400ffd086 327
geiineuville 0:441400ffd086 328 unsigned int HTTPClient::post(const char *url, FILE *data, FILE *result) {
geiineuville 0:441400ffd086 329 _mode = POST | FDATA | FRES;
geiineuville 0:441400ffd086 330 _data = (void *)data;
geiineuville 0:441400ffd086 331 _result = (void *)result;
geiineuville 0:441400ffd086 332 _resultleft = 0;
geiineuville 0:441400ffd086 333
geiineuville 0:441400ffd086 334 return make(url);
geiineuville 0:441400ffd086 335 }
geiineuville 0:441400ffd086 336
geiineuville 0:441400ffd086 337 unsigned int HTTPClient::post(const char *url, FILE *data, char *result, int rsize) {
geiineuville 0:441400ffd086 338 _mode = POST | FDATA | CRES;
geiineuville 0:441400ffd086 339 _data = (void *)data;
geiineuville 0:441400ffd086 340 _result = (void *)result;
geiineuville 0:441400ffd086 341 _resultleft = rsize -1;
geiineuville 0:441400ffd086 342
geiineuville 0:441400ffd086 343 return make(url);
geiineuville 0:441400ffd086 344 }
geiineuville 0:441400ffd086 345