private fork

Fork of GSwifiInterface by gs fan

Committer:
gsfan
Date:
Fri Nov 22 14:03:07 2013 +0000
Revision:
12:057089026a20
Child:
15:086d1a33a197
add more function (smtp, http, httpd, websocket); fix some function;

Who changed what in which revision?

UserRevisionLine numberNew contents of line
gsfan 12:057089026a20 1 /* Copyright (C) 2013 gsfan, MIT License
gsfan 12:057089026a20 2 *
gsfan 12:057089026a20 3 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software
gsfan 12:057089026a20 4 * and associated documentation files (the "Software"), to deal in the Software without restriction,
gsfan 12:057089026a20 5 * including without limitation the rights to use, copy, modify, merge, publish, distribute,
gsfan 12:057089026a20 6 * sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
gsfan 12:057089026a20 7 * furnished to do so, subject to the following conditions:
gsfan 12:057089026a20 8 *
gsfan 12:057089026a20 9 * The above copyright notice and this permission notice shall be included in all copies or
gsfan 12:057089026a20 10 * substantial portions of the Software.
gsfan 12:057089026a20 11 *
gsfan 12:057089026a20 12 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
gsfan 12:057089026a20 13 * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
gsfan 12:057089026a20 14 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
gsfan 12:057089026a20 15 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
gsfan 12:057089026a20 16 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
gsfan 12:057089026a20 17 */
gsfan 12:057089026a20 18
gsfan 12:057089026a20 19 #include "GSwifi.h"
gsfan 12:057089026a20 20
gsfan 12:057089026a20 21 #ifdef CFG_ENABLE_HTTPD
gsfan 12:057089026a20 22
gsfan 12:057089026a20 23 int GSwifi::httpd (int port) {
gsfan 12:057089026a20 24 int i;
gsfan 12:057089026a20 25
gsfan 12:057089026a20 26 if (!isAssociated() || _state.status != STAT_READY) return -1;
gsfan 12:057089026a20 27
gsfan 12:057089026a20 28 memset(&_httpd, 0, sizeof(_httpd));
gsfan 12:057089026a20 29 for (i = 0; i < 16; i ++) {
gsfan 12:057089026a20 30 _httpd[i].mode = HTTPDMODE_REQUEST;
gsfan 12:057089026a20 31 }
gsfan 12:057089026a20 32 _handler_count = 0;
gsfan 12:057089026a20 33
gsfan 12:057089026a20 34 _httpd_cid = listen(PROTO_TCP, port);
gsfan 12:057089026a20 35 if (_httpd_cid < 0) return -1;
gsfan 12:057089026a20 36
gsfan 12:057089026a20 37 _con[_httpd_cid].protocol = PROTO_HTTPD;
gsfan 12:057089026a20 38 return _httpd_cid;
gsfan 12:057089026a20 39 }
gsfan 12:057089026a20 40
gsfan 12:057089026a20 41 void GSwifi::httpdRecvData (int cid, char c) {
gsfan 12:057089026a20 42
gsfan 12:057089026a20 43 switch (_httpd[cid].mode) {
gsfan 12:057089026a20 44 case HTTPDMODE_REQUEST:
gsfan 12:057089026a20 45 if (_con[cid].buf == NULL)
gsfan 12:057089026a20 46 _con[cid].buf = new CircBuffer<char>(CFG_DATA_SIZE);
gsfan 12:057089026a20 47 _httpd[cid].buf = _con[cid].buf;
gsfan 12:057089026a20 48 _httpd[cid].buf->flush();
gsfan 12:057089026a20 49 if ((c >= 'A' && c <= 'Z') || (c >= 'a' && c <= 'z')) {
gsfan 12:057089026a20 50 _httpd[cid].buf->queue(c);
gsfan 12:057089026a20 51 _httpd[cid].mode = HTTPDMODE_REQUEST_STR;
gsfan 12:057089026a20 52 }
gsfan 12:057089026a20 53 break;
gsfan 12:057089026a20 54 case HTTPDMODE_REQUEST_STR:
gsfan 12:057089026a20 55 switch (c) {
gsfan 12:057089026a20 56 case 0:
gsfan 12:057089026a20 57 case 0x0d: // CR
gsfan 12:057089026a20 58 break;
gsfan 12:057089026a20 59 case 0x0a: // LF
gsfan 12:057089026a20 60 if (httpdParseRequest(cid)) {
gsfan 12:057089026a20 61 _httpd[cid].mode = HTTPDMODE_REQUEST_STR;
gsfan 12:057089026a20 62 } else {
gsfan 12:057089026a20 63 _httpd[cid].mode = HTTPDMODE_HEADER;
gsfan 12:057089026a20 64 }
gsfan 12:057089026a20 65 _httpd[cid].enter = 0;
gsfan 12:057089026a20 66 break;
gsfan 12:057089026a20 67 default:
gsfan 12:057089026a20 68 _httpd[cid].buf->queue(c);
gsfan 12:057089026a20 69 break;
gsfan 12:057089026a20 70 }
gsfan 12:057089026a20 71 break;
gsfan 12:057089026a20 72 case HTTPDMODE_HEADER:
gsfan 12:057089026a20 73 switch (c) {
gsfan 12:057089026a20 74 case 0:
gsfan 12:057089026a20 75 case 0x0d: // CR
gsfan 12:057089026a20 76 break;
gsfan 12:057089026a20 77 case 0x0a: // LF
gsfan 12:057089026a20 78 if (_httpd[cid].buf->available() == 0) {
gsfan 12:057089026a20 79 // if ((_httpd[cid].enter == 0x0d && c == 0x0a) || (_httpd[cid].enter == 0x0a && c == 0x0a)) {
gsfan 12:057089026a20 80 if (_httpd[cid].enter == 0x0a && c == 0x0a) {
gsfan 12:057089026a20 81 _httpd[cid].buf->flush();
gsfan 12:057089026a20 82 #ifdef CFG_ENABLE_WEBSOCKET
gsfan 12:057089026a20 83 if (_httpd[cid].websocket) {
gsfan 12:057089026a20 84 INFO("MODE_WEBSOCKET");
gsfan 12:057089026a20 85 wsAccept(cid);
gsfan 12:057089026a20 86 _httpd[cid].mode = HTTPDMODE_WEBSOCKET;
gsfan 12:057089026a20 87 } else
gsfan 12:057089026a20 88 #endif
gsfan 12:057089026a20 89 if (_httpd[cid].length) {
gsfan 12:057089026a20 90 INFO("MODE_BODY");
gsfan 12:057089026a20 91 _httpd[cid].mode = HTTPDMODE_BODY;
gsfan 12:057089026a20 92 } else {
gsfan 12:057089026a20 93 INFO("MODE_ENTER");
gsfan 12:057089026a20 94 _httpd[cid].mode = HTTPDMODE_ENTER;
gsfan 12:057089026a20 95 _con[cid].received = true;
gsfan 12:057089026a20 96 }
gsfan 12:057089026a20 97 }
gsfan 12:057089026a20 98 _httpd[cid].buf->flush();
gsfan 12:057089026a20 99 _httpd[cid].enter = c;
gsfan 12:057089026a20 100 break;
gsfan 12:057089026a20 101 }
gsfan 12:057089026a20 102
gsfan 12:057089026a20 103 httpdParseHeader(cid);
gsfan 12:057089026a20 104 _httpd[cid].enter = c;
gsfan 12:057089026a20 105 break;
gsfan 12:057089026a20 106 default:
gsfan 12:057089026a20 107 _httpd[cid].buf->queue(c);
gsfan 12:057089026a20 108 _httpd[cid].enter = 0;
gsfan 12:057089026a20 109 break;
gsfan 12:057089026a20 110 }
gsfan 12:057089026a20 111 break;
gsfan 12:057089026a20 112 case HTTPDMODE_BODY:
gsfan 12:057089026a20 113 _httpd[cid].buf->queue(c);
gsfan 12:057089026a20 114 if (_httpd[cid].buf->available() >= _httpd[cid].length) {
gsfan 12:057089026a20 115 _httpd[cid].mode = HTTPDMODE_ENTER;
gsfan 12:057089026a20 116 _con[cid].received = true;
gsfan 12:057089026a20 117 }
gsfan 12:057089026a20 118 break;
gsfan 12:057089026a20 119 #ifdef CFG_ENABLE_WEBSOCKET
gsfan 12:057089026a20 120 case HTTPDMODE_WEBSOCKET:
gsfan 12:057089026a20 121 case HTTPDMODE_WEBSOCKET_MASK:
gsfan 12:057089026a20 122 case HTTPDMODE_WEBSOCKET_BODY:
gsfan 12:057089026a20 123 wsRecvData(cid, c);
gsfan 12:057089026a20 124 break;
gsfan 12:057089026a20 125 #endif
gsfan 12:057089026a20 126 }
gsfan 12:057089026a20 127 }
gsfan 12:057089026a20 128
gsfan 12:057089026a20 129 void GSwifi::httpdPoll () {
gsfan 12:057089026a20 130 for (int cid = 0; cid < 16; cid ++) {
gsfan 12:057089026a20 131 if (isConnected(cid) && _con[cid].protocol == PROTO_HTTPD) {
gsfan 12:057089026a20 132
gsfan 12:057089026a20 133 if (_httpd[cid].mode == HTTPDMODE_ENTER) {
gsfan 12:057089026a20 134 int i = httpdGetHandler(_httpd[cid].uri);
gsfan 12:057089026a20 135 if (i >= 0) {
gsfan 12:057089026a20 136 if (_httpd_handler[i].dir) {
gsfan 12:057089026a20 137 // file
gsfan 12:057089026a20 138 httpdFile(cid, _httpd_handler[i].dir);
gsfan 12:057089026a20 139 } else
gsfan 12:057089026a20 140 if (_httpd_handler[i].func) {
gsfan 12:057089026a20 141 // cgi
gsfan 12:057089026a20 142 _httpd_handler[i].func(cid);
gsfan 12:057089026a20 143 // _httpd[cid].keepalive = 0;
gsfan 12:057089026a20 144 } else {
gsfan 12:057089026a20 145 httpdError(cid, 403);
gsfan 12:057089026a20 146 }
gsfan 12:057089026a20 147 } else {
gsfan 12:057089026a20 148 httpdError(cid, 404);
gsfan 12:057089026a20 149 }
gsfan 12:057089026a20 150
gsfan 12:057089026a20 151 if (_httpd[cid].keepalive) {
gsfan 12:057089026a20 152 DBG("keepalive %d", _httpd[cid].keepalive);
gsfan 12:057089026a20 153 _httpd[cid].keepalive --;
gsfan 12:057089026a20 154 } else {
gsfan 12:057089026a20 155 close(cid);
gsfan 12:057089026a20 156 }
gsfan 12:057089026a20 157 _httpd[cid].mode = HTTPDMODE_REQUEST;
gsfan 12:057089026a20 158 #ifdef CFG_ENABLE_WEBSOCKET
gsfan 12:057089026a20 159 } else
gsfan 12:057089026a20 160 if (_httpd[cid].mode == HTTPDMODE_WEBSOCKET_ENTER) {
gsfan 12:057089026a20 161 wsParseRequest(cid);
gsfan 12:057089026a20 162 _httpd[cid].mode = HTTPDMODE_WEBSOCKET;
gsfan 12:057089026a20 163 #endif
gsfan 12:057089026a20 164 }
gsfan 12:057089026a20 165
gsfan 12:057089026a20 166 }
gsfan 12:057089026a20 167 }
gsfan 12:057089026a20 168 }
gsfan 12:057089026a20 169
gsfan 12:057089026a20 170 int GSwifi::httpdParseRequest (int cid) {
gsfan 12:057089026a20 171 int i, j, len;
gsfan 12:057089026a20 172 char buf[CFG_CMD_SIZE];
gsfan 12:057089026a20 173
gsfan 12:057089026a20 174 for (len = 0; len < sizeof(buf); len++) {
gsfan 12:057089026a20 175 if (_httpd[cid].buf->dequeue(&buf[len]) == false) break;
gsfan 12:057089026a20 176 }
gsfan 12:057089026a20 177 buf[len] = 0;
gsfan 12:057089026a20 178
gsfan 12:057089026a20 179 if (strnicmp(buf, "GET ", 4) == 0) {
gsfan 12:057089026a20 180 _httpd[cid].req = REQ_HTTPGET;
gsfan 12:057089026a20 181 j = 4;
gsfan 12:057089026a20 182 } else
gsfan 12:057089026a20 183 if (strnicmp(buf, "POST ", 5) == 0) {
gsfan 12:057089026a20 184 _httpd[cid].req = REQ_HTTPPOST;
gsfan 12:057089026a20 185 j = 5;
gsfan 12:057089026a20 186 } else {
gsfan 12:057089026a20 187 return -1;
gsfan 12:057089026a20 188 }
gsfan 12:057089026a20 189
gsfan 12:057089026a20 190 if (_httpd[cid].uri == NULL)
gsfan 12:057089026a20 191 _httpd[cid].uri = (char*)malloc(CFG_CMD_SIZE);
gsfan 12:057089026a20 192
gsfan 12:057089026a20 193 for (i = 0; i < len - j; i ++) {
gsfan 12:057089026a20 194 _httpd[cid].uri[i] = buf[i + j];
gsfan 12:057089026a20 195 if (buf[i + j] == ' ' || i >= CFG_CMD_SIZE - 1) {
gsfan 12:057089026a20 196 _httpd[cid].uri[i] = 0;
gsfan 12:057089026a20 197 INFO("URI %d '%s'", _httpd[cid].req, _httpd[cid].uri);
gsfan 12:057089026a20 198 _httpd[cid].mode = HTTPDMODE_HEADER;
gsfan 12:057089026a20 199 _httpd[cid].buf->flush();
gsfan 12:057089026a20 200 _httpd[cid].length = 0;
gsfan 12:057089026a20 201 _httpd[cid].n = 0;
gsfan 12:057089026a20 202 _httpd[cid].filename = NULL;
gsfan 12:057089026a20 203 _httpd[cid].querystring = NULL;
gsfan 12:057089026a20 204 #ifdef CFG_ENABLE_WEBSOCKET
gsfan 12:057089026a20 205 _httpd[cid].websocket = 0;
gsfan 12:057089026a20 206 #endif
gsfan 12:057089026a20 207 break;
gsfan 12:057089026a20 208 }
gsfan 12:057089026a20 209 }
gsfan 12:057089026a20 210
gsfan 12:057089026a20 211 i = httpdGetHandler(_httpd[cid].uri);
gsfan 12:057089026a20 212 if (i >= 0) {
gsfan 12:057089026a20 213 _httpd[cid].filename = &_httpd[cid].uri[strlen(_httpd_handler[i].uri)];
gsfan 12:057089026a20 214 for (i = 0; i < strlen(_httpd[cid].filename); i ++) {
gsfan 12:057089026a20 215 if (_httpd[cid].filename[i] == '?') {
gsfan 12:057089026a20 216 _httpd[cid].filename[i] = 0;
gsfan 12:057089026a20 217 _httpd[cid].querystring = _httpd[cid].filename + i + 1;
gsfan 12:057089026a20 218 break;
gsfan 12:057089026a20 219 }
gsfan 12:057089026a20 220 }
gsfan 12:057089026a20 221 INFO("FILE '%s' QUERY '%s'", _httpd[cid].filename, _httpd[cid].querystring);
gsfan 12:057089026a20 222 }
gsfan 12:057089026a20 223 return 0;
gsfan 12:057089026a20 224 }
gsfan 12:057089026a20 225
gsfan 12:057089026a20 226 #define HEADER_TABLE_NUM 5
gsfan 12:057089026a20 227 int GSwifi::httpdParseHeader (int cid) {
gsfan 12:057089026a20 228 int i;
gsfan 12:057089026a20 229 char buf[CFG_CMD_SIZE];
gsfan 12:057089026a20 230 static const struct HEADER_TABLE {
gsfan 12:057089026a20 231 const char header[24];
gsfan 12:057089026a20 232 void (GSwifi::*func)(int id, const char*);
gsfan 12:057089026a20 233 } header_table[HEADER_TABLE_NUM] = {
gsfan 12:057089026a20 234 {"Content-Length:", &GSwifi::reqContentLength},
gsfan 12:057089026a20 235 {"Connection:", &GSwifi::reqConnection},
gsfan 12:057089026a20 236 {"Upgrade: websocket", &GSwifi::reqUpgrade},
gsfan 12:057089026a20 237 {"Sec-WebSocket-Version:", &GSwifi::reqWebSocketVersion},
gsfan 12:057089026a20 238 {"Sec-WebSocket-Key:", &GSwifi::reqWebSocketKey},
gsfan 12:057089026a20 239 };
gsfan 12:057089026a20 240 for (i = 0; i < sizeof(buf); i++) {
gsfan 12:057089026a20 241 if (_httpd[cid].buf->dequeue(&buf[i]) == false) break;
gsfan 12:057089026a20 242 }
gsfan 12:057089026a20 243 buf[i] = 0;
gsfan 12:057089026a20 244
gsfan 12:057089026a20 245 for (i = 0; i < HEADER_TABLE_NUM; i ++) {
gsfan 12:057089026a20 246 if (strnicmp(buf, header_table[i].header, strlen(header_table[i].header)) == 0) {
gsfan 12:057089026a20 247 DBG("parse header %d '%s'\r\n", i, buf);
gsfan 12:057089026a20 248 if (header_table[i].func != NULL) {
gsfan 12:057089026a20 249 (this->*(header_table[i].func))(cid, buf);
gsfan 12:057089026a20 250 }
gsfan 12:057089026a20 251 return 0;
gsfan 12:057089026a20 252 }
gsfan 12:057089026a20 253 }
gsfan 12:057089026a20 254
gsfan 12:057089026a20 255 return -1;
gsfan 12:057089026a20 256 }
gsfan 12:057089026a20 257
gsfan 12:057089026a20 258 void GSwifi::reqContentLength (int cid, const char *buf) {
gsfan 12:057089026a20 259 _httpd[cid].length = atoi(&buf[16]);
gsfan 12:057089026a20 260 }
gsfan 12:057089026a20 261
gsfan 12:057089026a20 262 void GSwifi::reqConnection (int cid, const char *buf) {
gsfan 12:057089026a20 263 if (strnicmp(&buf[12], "Keep-Alive", 10) == 0 && _httpd[cid].keepalive == 0) {
gsfan 12:057089026a20 264 _httpd[cid].keepalive = CFG_HTTPD_KEEPALIVE;
gsfan 12:057089026a20 265 } else {
gsfan 12:057089026a20 266 _httpd[cid].keepalive = 0;
gsfan 12:057089026a20 267 }
gsfan 12:057089026a20 268 }
gsfan 12:057089026a20 269
gsfan 12:057089026a20 270 #ifdef CFG_ENABLE_WEBSOCKET
gsfan 12:057089026a20 271 void GSwifi::reqUpgrade (int cid, const char *buf) {
gsfan 12:057089026a20 272 if (! _httpd[cid].websocket) _httpd[cid].websocket = 1;
gsfan 12:057089026a20 273 }
gsfan 12:057089026a20 274
gsfan 12:057089026a20 275 void GSwifi::reqWebSocketVersion (int cid, const char *buf) {
gsfan 12:057089026a20 276 _httpd[cid].websocket = atoi(&buf[23]);
gsfan 12:057089026a20 277 }
gsfan 12:057089026a20 278
gsfan 12:057089026a20 279 void GSwifi::reqWebSocketKey (int cid, const char *buf) {
gsfan 12:057089026a20 280 if (_httpd[cid].websocket_key == NULL) {
gsfan 12:057089026a20 281 _httpd[cid].websocket_key = (char*)malloc(30);
gsfan 12:057089026a20 282 }
gsfan 12:057089026a20 283 strncpy(_httpd[cid].websocket_key, &buf[19], 30);
gsfan 12:057089026a20 284 }
gsfan 12:057089026a20 285 #else
gsfan 12:057089026a20 286 void GSwifi::reqUpgrade (int cid, const char *buf) {
gsfan 12:057089026a20 287 }
gsfan 12:057089026a20 288 void GSwifi::reqWebSocketVersion (int cid, const char *buf) {
gsfan 12:057089026a20 289 }
gsfan 12:057089026a20 290 void GSwifi::reqWebSocketKey (int cid, const char *buf) {
gsfan 12:057089026a20 291 }
gsfan 12:057089026a20 292 #endif
gsfan 12:057089026a20 293
gsfan 12:057089026a20 294 #endif