Example self-announcing webserver which controls a servo through a smallHTML userinterface.

Dependencies:   mbed

Committer:
dirkx
Date:
Sat Aug 14 15:56:01 2010 +0000
Revision:
0:a259777c45a3

        

Who changed what in which revision?

UserRevisionLine numberNew contents of line
dirkx 0:a259777c45a3 1
dirkx 0:a259777c45a3 2 /*
dirkx 0:a259777c45a3 3 Copyright (c) 2010 Donatien Garnier (donatiengar [at] gmail [dot] com)
dirkx 0:a259777c45a3 4
dirkx 0:a259777c45a3 5 Permission is hereby granted, free of charge, to any person obtaining a copy
dirkx 0:a259777c45a3 6 of this software and associated documentation files (the "Software"), to deal
dirkx 0:a259777c45a3 7 in the Software without restriction, including without limitation the rights
dirkx 0:a259777c45a3 8 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
dirkx 0:a259777c45a3 9 copies of the Software, and to permit persons to whom the Software is
dirkx 0:a259777c45a3 10 furnished to do so, subject to the following conditions:
dirkx 0:a259777c45a3 11
dirkx 0:a259777c45a3 12 The above copyright notice and this permission notice shall be included in
dirkx 0:a259777c45a3 13 all copies or substantial portions of the Software.
dirkx 0:a259777c45a3 14
dirkx 0:a259777c45a3 15 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
dirkx 0:a259777c45a3 16 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
dirkx 0:a259777c45a3 17 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
dirkx 0:a259777c45a3 18 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
dirkx 0:a259777c45a3 19 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
dirkx 0:a259777c45a3 20 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
dirkx 0:a259777c45a3 21 THE SOFTWARE.
dirkx 0:a259777c45a3 22 */
dirkx 0:a259777c45a3 23
dirkx 0:a259777c45a3 24 #include "HTTPRequestHandler.h"
dirkx 0:a259777c45a3 25
dirkx 0:a259777c45a3 26 #include <string.h>
dirkx 0:a259777c45a3 27
dirkx 0:a259777c45a3 28 //#define __DEBUG
dirkx 0:a259777c45a3 29 #include "dbg/dbg.h"
dirkx 0:a259777c45a3 30
dirkx 0:a259777c45a3 31 #define HTTP_REQUEST_TIMEOUT 5000
dirkx 0:a259777c45a3 32
dirkx 0:a259777c45a3 33 HTTPRequestHandler::HTTPRequestHandler(const char* rootPath, const char* path, TCPSocket* pTCPSocket) : NetService(),
dirkx 0:a259777c45a3 34 m_pTCPSocket(pTCPSocket), m_reqHeaders(), m_respHeaders(),
dirkx 0:a259777c45a3 35 m_rootPath(rootPath), m_path(path), m_errc(200),
dirkx 0:a259777c45a3 36 m_watchdog(), m_timeout(0), m_closed(false), m_headersSent(false) //OK
dirkx 0:a259777c45a3 37 {
dirkx 0:a259777c45a3 38 //Read & parse headers
dirkx 0:a259777c45a3 39 readHeaders();
dirkx 0:a259777c45a3 40 m_pTCPSocket->setOnEvent(this, &HTTPRequestHandler::onTCPSocketEvent);
dirkx 0:a259777c45a3 41 setTimeout(HTTP_REQUEST_TIMEOUT);
dirkx 0:a259777c45a3 42 }
dirkx 0:a259777c45a3 43
dirkx 0:a259777c45a3 44 HTTPRequestHandler::~HTTPRequestHandler()
dirkx 0:a259777c45a3 45 {
dirkx 0:a259777c45a3 46 close();
dirkx 0:a259777c45a3 47 }
dirkx 0:a259777c45a3 48
dirkx 0:a259777c45a3 49 void HTTPRequestHandler::onTimeout() //Connection has timed out
dirkx 0:a259777c45a3 50 {
dirkx 0:a259777c45a3 51 close();
dirkx 0:a259777c45a3 52 }
dirkx 0:a259777c45a3 53
dirkx 0:a259777c45a3 54 void HTTPRequestHandler::close() //Close socket and destroy data
dirkx 0:a259777c45a3 55 {
dirkx 0:a259777c45a3 56 if(m_closed)
dirkx 0:a259777c45a3 57 return;
dirkx 0:a259777c45a3 58 m_closed = true; //Prevent recursive calling or calling on an object being destructed by someone else
dirkx 0:a259777c45a3 59 m_watchdog.detach();
dirkx 0:a259777c45a3 60 onClose();
dirkx 0:a259777c45a3 61 m_pTCPSocket->resetOnEvent();
dirkx 0:a259777c45a3 62 m_pTCPSocket->close();
dirkx 0:a259777c45a3 63 delete m_pTCPSocket; //Can safely destroy socket
dirkx 0:a259777c45a3 64 NetService::close();
dirkx 0:a259777c45a3 65 }
dirkx 0:a259777c45a3 66
dirkx 0:a259777c45a3 67 map<string, string>& HTTPRequestHandler::reqHeaders() //const
dirkx 0:a259777c45a3 68 {
dirkx 0:a259777c45a3 69 return m_reqHeaders;
dirkx 0:a259777c45a3 70 }
dirkx 0:a259777c45a3 71
dirkx 0:a259777c45a3 72 string& HTTPRequestHandler::path() //const
dirkx 0:a259777c45a3 73 {
dirkx 0:a259777c45a3 74 return m_path;
dirkx 0:a259777c45a3 75 }
dirkx 0:a259777c45a3 76
dirkx 0:a259777c45a3 77 int HTTPRequestHandler::dataLen() const
dirkx 0:a259777c45a3 78 {
dirkx 0:a259777c45a3 79 map<string,string>::iterator it;
dirkx 0:a259777c45a3 80 it = m_reqHeaders.find("Content-Length");
dirkx 0:a259777c45a3 81 if( it == m_reqHeaders.end() )
dirkx 0:a259777c45a3 82 {
dirkx 0:a259777c45a3 83 return 0;
dirkx 0:a259777c45a3 84 }
dirkx 0:a259777c45a3 85 return atoi((*it).second.c_str()); //return 0 if parse fails, so that's fine
dirkx 0:a259777c45a3 86 }
dirkx 0:a259777c45a3 87
dirkx 0:a259777c45a3 88 int HTTPRequestHandler::readData(char* buf, int len)
dirkx 0:a259777c45a3 89 {
dirkx 0:a259777c45a3 90 return m_pTCPSocket->recv(buf, len);
dirkx 0:a259777c45a3 91 }
dirkx 0:a259777c45a3 92
dirkx 0:a259777c45a3 93 string& HTTPRequestHandler::rootPath() //const
dirkx 0:a259777c45a3 94 {
dirkx 0:a259777c45a3 95 return m_rootPath;
dirkx 0:a259777c45a3 96 }
dirkx 0:a259777c45a3 97
dirkx 0:a259777c45a3 98 void HTTPRequestHandler::setErrCode(int errc)
dirkx 0:a259777c45a3 99 {
dirkx 0:a259777c45a3 100 m_errc = errc;
dirkx 0:a259777c45a3 101 }
dirkx 0:a259777c45a3 102
dirkx 0:a259777c45a3 103 void HTTPRequestHandler::setContentLen(int len)
dirkx 0:a259777c45a3 104 {
dirkx 0:a259777c45a3 105 char len_str[6] = {0};
dirkx 0:a259777c45a3 106 sprintf(len_str, "%d", len);
dirkx 0:a259777c45a3 107 respHeaders()["Content-Length"] = len_str;
dirkx 0:a259777c45a3 108 }
dirkx 0:a259777c45a3 109
dirkx 0:a259777c45a3 110 map<string, string>& HTTPRequestHandler::respHeaders()
dirkx 0:a259777c45a3 111 {
dirkx 0:a259777c45a3 112 return m_respHeaders;
dirkx 0:a259777c45a3 113 }
dirkx 0:a259777c45a3 114
dirkx 0:a259777c45a3 115 int HTTPRequestHandler::writeData(const char* buf, int len)
dirkx 0:a259777c45a3 116 {
dirkx 0:a259777c45a3 117 if(!m_headersSent)
dirkx 0:a259777c45a3 118 {
dirkx 0:a259777c45a3 119 m_headersSent = true;
dirkx 0:a259777c45a3 120 writeHeaders();
dirkx 0:a259777c45a3 121 }
dirkx 0:a259777c45a3 122
dirkx 0:a259777c45a3 123 return m_pTCPSocket->send(buf, len);
dirkx 0:a259777c45a3 124 }
dirkx 0:a259777c45a3 125
dirkx 0:a259777c45a3 126 void HTTPRequestHandler::setTimeout(int ms)
dirkx 0:a259777c45a3 127 {
dirkx 0:a259777c45a3 128 m_timeout = 1000*ms;
dirkx 0:a259777c45a3 129 resetTimeout();
dirkx 0:a259777c45a3 130 }
dirkx 0:a259777c45a3 131
dirkx 0:a259777c45a3 132 void HTTPRequestHandler::resetTimeout()
dirkx 0:a259777c45a3 133 {
dirkx 0:a259777c45a3 134 m_watchdog.detach();
dirkx 0:a259777c45a3 135 m_watchdog.attach_us<HTTPRequestHandler>(this, &HTTPRequestHandler::onTimeout, m_timeout);
dirkx 0:a259777c45a3 136 }
dirkx 0:a259777c45a3 137
dirkx 0:a259777c45a3 138
dirkx 0:a259777c45a3 139 void HTTPRequestHandler::readHeaders()
dirkx 0:a259777c45a3 140 {
dirkx 0:a259777c45a3 141 static char line[128];
dirkx 0:a259777c45a3 142 static char key[128];
dirkx 0:a259777c45a3 143 static char value[128];
dirkx 0:a259777c45a3 144 while( readLine(line, 128) > 0) //if == 0, it is an empty line = end of headers
dirkx 0:a259777c45a3 145 {
dirkx 0:a259777c45a3 146 int n = sscanf(line, "%[^:]: %[^\n]", key, value);
dirkx 0:a259777c45a3 147 if ( n == 2 )
dirkx 0:a259777c45a3 148 {
dirkx 0:a259777c45a3 149 DBG("\r\nRead header : %s : %s\r\n", key, value);
dirkx 0:a259777c45a3 150 m_reqHeaders[key] = value;
dirkx 0:a259777c45a3 151 }
dirkx 0:a259777c45a3 152 //TODO: Impl n==1 case (part 2 of previous header)
dirkx 0:a259777c45a3 153 }
dirkx 0:a259777c45a3 154 }
dirkx 0:a259777c45a3 155
dirkx 0:a259777c45a3 156 void HTTPRequestHandler::writeHeaders() //Called at the first writeData call
dirkx 0:a259777c45a3 157 {
dirkx 0:a259777c45a3 158 static char line[128];
dirkx 0:a259777c45a3 159
dirkx 0:a259777c45a3 160 //Response line
dirkx 0:a259777c45a3 161 sprintf(line, "HTTP/1.1 %d MbedInfo\r\n", m_errc); //Not a violation of the standard not to include the descriptive text
dirkx 0:a259777c45a3 162 m_pTCPSocket->send(line, strlen(line));
dirkx 0:a259777c45a3 163
dirkx 0:a259777c45a3 164 map<string,string>::iterator it;
dirkx 0:a259777c45a3 165 while( !m_respHeaders.empty() )
dirkx 0:a259777c45a3 166 {
dirkx 0:a259777c45a3 167 it = m_respHeaders.begin();
dirkx 0:a259777c45a3 168 sprintf(line, "%s: %s\r\n", (*it).first.c_str(), (*it).second.c_str() );
dirkx 0:a259777c45a3 169 DBG("\r\n%s", line);
dirkx 0:a259777c45a3 170 m_pTCPSocket->send(line, strlen(line));
dirkx 0:a259777c45a3 171 m_respHeaders.erase(it);
dirkx 0:a259777c45a3 172 }
dirkx 0:a259777c45a3 173 m_pTCPSocket->send("\r\n",2); //End of head
dirkx 0:a259777c45a3 174 }
dirkx 0:a259777c45a3 175
dirkx 0:a259777c45a3 176 int HTTPRequestHandler::readLine(char* str, int maxLen)
dirkx 0:a259777c45a3 177 {
dirkx 0:a259777c45a3 178 int ret;
dirkx 0:a259777c45a3 179 int len = 0;
dirkx 0:a259777c45a3 180 for(int i = 0; i < maxLen - 1; i++)
dirkx 0:a259777c45a3 181 {
dirkx 0:a259777c45a3 182 ret = m_pTCPSocket->recv(str, 1);
dirkx 0:a259777c45a3 183 if(!ret)
dirkx 0:a259777c45a3 184 {
dirkx 0:a259777c45a3 185 break;
dirkx 0:a259777c45a3 186 }
dirkx 0:a259777c45a3 187 if( (len > 1) && *(str-1)=='\r' && *str=='\n' )
dirkx 0:a259777c45a3 188 {
dirkx 0:a259777c45a3 189 str--;
dirkx 0:a259777c45a3 190 len-=2;
dirkx 0:a259777c45a3 191 break;
dirkx 0:a259777c45a3 192 }
dirkx 0:a259777c45a3 193 else if( *str=='\n' )
dirkx 0:a259777c45a3 194 {
dirkx 0:a259777c45a3 195 len--;
dirkx 0:a259777c45a3 196 break;
dirkx 0:a259777c45a3 197 }
dirkx 0:a259777c45a3 198 str++;
dirkx 0:a259777c45a3 199 len++;
dirkx 0:a259777c45a3 200 }
dirkx 0:a259777c45a3 201 *str = 0;
dirkx 0:a259777c45a3 202 return len;
dirkx 0:a259777c45a3 203 }
dirkx 0:a259777c45a3 204
dirkx 0:a259777c45a3 205 void HTTPRequestHandler::onTCPSocketEvent(TCPSocketEvent e)
dirkx 0:a259777c45a3 206 {
dirkx 0:a259777c45a3 207
dirkx 0:a259777c45a3 208 DBG("\r\nEvent %d in HTTPRequestHandler\r\n", e);
dirkx 0:a259777c45a3 209
dirkx 0:a259777c45a3 210 if(m_closed)
dirkx 0:a259777c45a3 211 {
dirkx 0:a259777c45a3 212 DBG("\r\nWARN: Discarded\r\n");
dirkx 0:a259777c45a3 213 return;
dirkx 0:a259777c45a3 214 }
dirkx 0:a259777c45a3 215
dirkx 0:a259777c45a3 216 switch(e)
dirkx 0:a259777c45a3 217 {
dirkx 0:a259777c45a3 218 case TCPSOCKET_READABLE:
dirkx 0:a259777c45a3 219 resetTimeout();
dirkx 0:a259777c45a3 220 onReadable();
dirkx 0:a259777c45a3 221 break;
dirkx 0:a259777c45a3 222 case TCPSOCKET_WRITEABLE:
dirkx 0:a259777c45a3 223 resetTimeout();
dirkx 0:a259777c45a3 224 onWriteable();
dirkx 0:a259777c45a3 225 break;
dirkx 0:a259777c45a3 226 case TCPSOCKET_CONTIMEOUT:
dirkx 0:a259777c45a3 227 case TCPSOCKET_CONRST:
dirkx 0:a259777c45a3 228 case TCPSOCKET_CONABRT:
dirkx 0:a259777c45a3 229 case TCPSOCKET_ERROR:
dirkx 0:a259777c45a3 230 case TCPSOCKET_DISCONNECTED:
dirkx 0:a259777c45a3 231 DBG("\r\nConnection error in handler\r\n");
dirkx 0:a259777c45a3 232 close();
dirkx 0:a259777c45a3 233 break;
dirkx 0:a259777c45a3 234 }
dirkx 0:a259777c45a3 235
dirkx 0:a259777c45a3 236 }