http://mbed.org/users/okini3939/notebook/comet_websocket/
Dependencies: EthernetNetIf mbed RingBuffer MbedJSONValue
CometClient.cpp
- Committer:
- okini3939
- Date:
- 2011-11-19
- Revision:
- 0:632cb8c03ca3
File content as of revision 0:632cb8c03ca3:
/* * mbed pseudo Comet HTTP Client * Copyright (c) 2011 Hiroshi Suga * Released under the MIT License: http://mbed.org/license/mit */ /** @file * @brief pseudo Comet HTTP Client */ #include "mbed.h" #include "EthernetNetIf.h" #include "TCPSocket.h" #include "DNSRequest.h" #include "RingBuffer.h" #include "CometClient.h" #include <ctype.h> #define DEBUG #include "dbg.h" static TCPSocket *http; static volatile int tcp_ready = 0, tcp_readable = 0, tcp_writable = 0; static volatile int tcp_busy = 0, mode = 0; static volatile int dns_status = 0; static Host remote; static RingBuffer fifo(1500); static char remoteuri[100]; // Copyright (c) 2010 Donatien Garnier (donatiengar [at] gmail [dot] com) int base64enc(const char *input, unsigned int length, char *output, int len) { static const char base64[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"; unsigned int c, c1, c2, c3; if (len < ((((length-1)/3)+1)<<2)) return -1; for(unsigned int i = 0, j = 0; i<length; i+=3,j+=4) { c1 = ((((unsigned char)*((unsigned char *)&input[i])))); c2 = (length>i+1)?((((unsigned char)*((unsigned char *)&input[i+1])))):0; c3 = (length>i+2)?((((unsigned char)*((unsigned char *)&input[i+2])))):0; c = ((c1 & 0xFC) >> 2); output[j+0] = base64[c]; c = ((c1 & 0x03) << 4) | ((c2 & 0xF0) >> 4); output[j+1] = base64[c]; c = ((c2 & 0x0F) << 2) | ((c3 & 0xC0) >> 6); output[j+2] = (length>i+1)?base64[c]:'='; c = (c3 & 0x3F); output[j+3] = (length>i+2)?base64[c]:'='; } output[(((length-1)/3)+1)<<2] = '\0'; return 0; } // Copyright (c) 2010 Donatien Garnier (donatiengar [at] gmail [dot] com) int urlencode(char *str, char *buf, int len) { static const char to_hex[] = "0123456789ABCDEF"; // char *pstr = str, *buf = (char*)malloc(strlen(str) * 3 + 1), *pbuf = buf; char *pstr = str, *pbuf = buf; if (len < (strlen(str) * 3 + 1)) return -1; while (*pstr) { if (isalnum(*pstr) || *pstr == '-' || *pstr == '_' || *pstr == '.' || *pstr == '~') { *pbuf++ = *pstr; } else if (*pstr == ' ') { *pbuf++ = '+'; } else { *pbuf++ = '%'; *pbuf++ = to_hex[(*pstr >> 4) & 0x0f]; *pbuf++ = to_hex[*pstr & 0x0f]; } pstr++; } *pbuf = '\0'; return 0; } void isr_http (TCPSocketEvent e) { DBG("http(%d)\r\n", e); switch(e) { case TCPSOCKET_CONNECTED: tcp_ready = 1; tcp_busy = 0; mode = 0; break; case TCPSOCKET_READABLE: //Incoming data { char buf[1500]; int i = 0, len; len = http->recv(buf, sizeof(buf)); for (i = 0; i < len; i ++) { switch (mode) { case 0: if (strncmp(buf, "HTTP/", 5) == 0 && atoi(&buf[9]) == 200) { mode = 1; } break; case 1: case 2: if (buf[i] == '\r') { mode ++; } else if (buf[i] == '\n') { } else { mode = 1; } break; case 3: if (buf[i] == 0x7f) { mode = 4; } break; case 4: fifo.put(buf[i]); break; } } } // Incoming data tcp_readable = 1; break; case TCPSOCKET_WRITEABLE: //We can send data tcp_writable = 1; break; case TCPSOCKET_CONTIMEOUT: case TCPSOCKET_CONRST: case TCPSOCKET_CONABRT: case TCPSOCKET_ERROR: case TCPSOCKET_DISCONNECTED: http->close(); tcp_ready = 0; tcp_readable = 0; tcp_writable = 0; tcp_busy = 0; break; } } void createauth (char *user, char *pwd, char *buf, int len) { char tmp[80]; strncpy(buf, "Authorization: Basic ", len); snprintf(tmp, sizeof(tmp), "%s:%s", user, pwd); base64enc(tmp, strlen(tmp), &buf[strlen(buf)], len - strlen(buf)); strncat(buf, "\r\n", len - strlen(buf)); } void isr_dns (DNSReply r) { DBG("dns(%d)\r\n", r); if (DNS_FOUND) { dns_status = 1; } else { dns_status = -1; } } void pollComet () { Net::poll(); if (! tcp_ready && ! tcp_busy) { tcp_busy = 1; openComet(NULL, NULL, NULL, NULL); } } int recvComet (char *buf, int size) { int i, len; len = fifo.use(); if (len == 0) return 0; DBG("readComet %d\r\n", len); if (len > size) len = size; for (i = 0; i < len; i ++) { if (fifo.get(&buf[i])) break; } tcp_readable = 0; return len; } int sendComet (char *buf) { while (tcp_busy) { Net::poll(); } tcp_busy = 1; if (tcp_ready) { DBG("sendComet close\r\n"); http->close(); } DBG("sendComet open\r\n"); openComet(NULL, NULL, NULL, buf); return 0; } int openComet (Host *host, char *uri, char *head, char *body) { TCPSocketErr err; Timer timeout; char buf[1500]; tcp_busy = 1; if (host) { http = new TCPSocket; http->setOnEvent(isr_http); // connect if (host->getIp().isNull()) { // resolv DNSRequest dns; dns_status = 0; dns.setOnReply(isr_dns); if (dns.resolve(host) != DNS_OK) goto exit; timeout.reset(); timeout.start(); while (timeout.read_ms() < HTTP_TIMEOUT) { if (dns_status) break; Net::poll(); } timeout.stop(); if (dns_status <= 0) goto exit; DBG("%s [%d.%d.%d.%d]\r\n", host->getName(), (unsigned char)host->getIp()[0], (unsigned char)host->getIp()[1], (unsigned char)host->getIp()[2], (unsigned char)host->getIp()[3]); } if (! host->getPort()) { host->setPort(HTTP_PORT); } remote = *host; strncpy(remoteuri, uri, sizeof(remoteuri)); } // if (host) err = http->connect(remote); if (err != TCPSOCKET_OK) goto exit; // wait connect timeout.reset(); timeout.start(); while (timeout.read_ms() < HTTP_TIMEOUT) { if (tcp_ready) break; Net::poll(); } timeout.stop(); if (! tcp_ready) goto exit; // send request http->send("POST ", 5); http->send(remoteuri, strlen(remoteuri)); http->send(" HTTP/1.1\r\nHost: ", 17); http->send(remote.getName(), strlen(remote.getName())); http->send("\r\n", 2); http->send("Connection: close\r\n", 19); http->send("Content-type: application/x-www-form-urlencoded\r\n", 49); if (head) { http->send(head, strlen(head)); } sprintf(buf, "Content-Length: %d\r\n", strlen(body)); http->send(buf, strlen(buf)); http->send("\r\n", 2); // post method if (body) { http->send(body, strlen(body)); } Net::poll(); return 0; exit: http->resetOnEvent(); http->close(); delete http; return -1; }