Brandon Fictorie / Mbed 2 deprecated BF_Websocket

Dependencies:   mbed

Committer:
bfictorie
Date:
Sun Mar 25 17:26:30 2012 +0000
Revision:
0:8cdad1c73e8e

        

Who changed what in which revision?

UserRevisionLine numberNew contents of line
bfictorie 0:8cdad1c73e8e 1
bfictorie 0:8cdad1c73e8e 2 /*
bfictorie 0:8cdad1c73e8e 3 Copyright (c) 2010 Donatien Garnier (donatiengar [at] gmail [dot] com)
bfictorie 0:8cdad1c73e8e 4
bfictorie 0:8cdad1c73e8e 5 Permission is hereby granted, free of charge, to any person obtaining a copy
bfictorie 0:8cdad1c73e8e 6 of this software and associated documentation files (the "Software"), to deal
bfictorie 0:8cdad1c73e8e 7 in the Software without restriction, including without limitation the rights
bfictorie 0:8cdad1c73e8e 8 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
bfictorie 0:8cdad1c73e8e 9 copies of the Software, and to permit persons to whom the Software is
bfictorie 0:8cdad1c73e8e 10 furnished to do so, subject to the following conditions:
bfictorie 0:8cdad1c73e8e 11
bfictorie 0:8cdad1c73e8e 12 The above copyright notice and this permission notice shall be included in
bfictorie 0:8cdad1c73e8e 13 all copies or substantial portions of the Software.
bfictorie 0:8cdad1c73e8e 14
bfictorie 0:8cdad1c73e8e 15 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
bfictorie 0:8cdad1c73e8e 16 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
bfictorie 0:8cdad1c73e8e 17 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
bfictorie 0:8cdad1c73e8e 18 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
bfictorie 0:8cdad1c73e8e 19 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
bfictorie 0:8cdad1c73e8e 20 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
bfictorie 0:8cdad1c73e8e 21 THE SOFTWARE.
bfictorie 0:8cdad1c73e8e 22 */
bfictorie 0:8cdad1c73e8e 23
bfictorie 0:8cdad1c73e8e 24 #include "core/netservice.h"
bfictorie 0:8cdad1c73e8e 25 #include "HTTPRequestDispatcher.h"
bfictorie 0:8cdad1c73e8e 26 #include "HTTPRequestHandler.h"
bfictorie 0:8cdad1c73e8e 27 #include <string.h>
bfictorie 0:8cdad1c73e8e 28
bfictorie 0:8cdad1c73e8e 29 //#define __DEBUG
bfictorie 0:8cdad1c73e8e 30 #include "dbg/dbg.h"
bfictorie 0:8cdad1c73e8e 31
bfictorie 0:8cdad1c73e8e 32 HTTPRequestDispatcher::HTTPRequestDispatcher(HTTPServer* pSvr, TCPSocket* pTCPSocket) : NetService(), m_pSvr(pSvr), m_pTCPSocket(pTCPSocket), m_watchdog(), m_closed(false)
bfictorie 0:8cdad1c73e8e 33 {
bfictorie 0:8cdad1c73e8e 34 m_pTCPSocket->setOnEvent(this, &HTTPRequestDispatcher::onTCPSocketEvent);
bfictorie 0:8cdad1c73e8e 35 m_watchdog.attach_us<HTTPRequestDispatcher>(this, &HTTPRequestDispatcher::onTimeout, HTTP_REQUEST_TIMEOUT * 1000);
bfictorie 0:8cdad1c73e8e 36 }
bfictorie 0:8cdad1c73e8e 37
bfictorie 0:8cdad1c73e8e 38 HTTPRequestDispatcher::~HTTPRequestDispatcher()
bfictorie 0:8cdad1c73e8e 39 {
bfictorie 0:8cdad1c73e8e 40 close();
bfictorie 0:8cdad1c73e8e 41 }
bfictorie 0:8cdad1c73e8e 42
bfictorie 0:8cdad1c73e8e 43 void HTTPRequestDispatcher::dispatchRequest()
bfictorie 0:8cdad1c73e8e 44 {
bfictorie 0:8cdad1c73e8e 45 string path;
bfictorie 0:8cdad1c73e8e 46 string meth;
bfictorie 0:8cdad1c73e8e 47 HTTP_METH methCode;
bfictorie 0:8cdad1c73e8e 48
bfictorie 0:8cdad1c73e8e 49 DBG("Dispatching req\r\n");
bfictorie 0:8cdad1c73e8e 50
bfictorie 0:8cdad1c73e8e 51 if( !getRequest(&path, &meth ) )
bfictorie 0:8cdad1c73e8e 52 {
bfictorie 0:8cdad1c73e8e 53 close();
bfictorie 0:8cdad1c73e8e 54 return; //Invalid request
bfictorie 0:8cdad1c73e8e 55 }
bfictorie 0:8cdad1c73e8e 56
bfictorie 0:8cdad1c73e8e 57 if( !meth.compare("GET") )
bfictorie 0:8cdad1c73e8e 58 {
bfictorie 0:8cdad1c73e8e 59 methCode = HTTP_GET;
bfictorie 0:8cdad1c73e8e 60 }
bfictorie 0:8cdad1c73e8e 61 else if( !meth.compare("POST") )
bfictorie 0:8cdad1c73e8e 62 {
bfictorie 0:8cdad1c73e8e 63 methCode = HTTP_POST;
bfictorie 0:8cdad1c73e8e 64 }
bfictorie 0:8cdad1c73e8e 65 else if( !meth.compare("HEAD") )
bfictorie 0:8cdad1c73e8e 66 {
bfictorie 0:8cdad1c73e8e 67 methCode = HTTP_HEAD;
bfictorie 0:8cdad1c73e8e 68 }
bfictorie 0:8cdad1c73e8e 69 else
bfictorie 0:8cdad1c73e8e 70 {
bfictorie 0:8cdad1c73e8e 71 close(); //Parse error
bfictorie 0:8cdad1c73e8e 72 return;
bfictorie 0:8cdad1c73e8e 73 }
bfictorie 0:8cdad1c73e8e 74
bfictorie 0:8cdad1c73e8e 75 DBG("Looking for a handler\r\n");
bfictorie 0:8cdad1c73e8e 76
bfictorie 0:8cdad1c73e8e 77 map< string, HTTPRequestHandler*(*)(const char*, const char*, TCPSocket*) >::iterator it;
bfictorie 0:8cdad1c73e8e 78 // it = m_pSvr->m_lpHandlers.find(rootPath); //We are friends so we can do that
bfictorie 0:8cdad1c73e8e 79 // NEW CODE START:
bfictorie 0:8cdad1c73e8e 80 int root_len = 0;
bfictorie 0:8cdad1c73e8e 81 for (it = m_pSvr->m_lpHandlers.begin(); it != m_pSvr->m_lpHandlers.end(); it++)
bfictorie 0:8cdad1c73e8e 82 {
bfictorie 0:8cdad1c73e8e 83 DBG("Checking %s...\n", (*it).first.c_str());
bfictorie 0:8cdad1c73e8e 84 root_len = (*it).first.length();
bfictorie 0:8cdad1c73e8e 85 if ( root_len &&
bfictorie 0:8cdad1c73e8e 86 !path.compare( 0, root_len, (*it).first ) &&
bfictorie 0:8cdad1c73e8e 87 (path[root_len] == '/' || path[root_len] == '\0'))
bfictorie 0:8cdad1c73e8e 88 {
bfictorie 0:8cdad1c73e8e 89 DBG("Found (%s)\n", (*it).first.c_str());
bfictorie 0:8cdad1c73e8e 90 // Found!
bfictorie 0:8cdad1c73e8e 91 break; // for
bfictorie 0:8cdad1c73e8e 92 }
bfictorie 0:8cdad1c73e8e 93 }
bfictorie 0:8cdad1c73e8e 94 // NEW CODE END
bfictorie 0:8cdad1c73e8e 95 if((it == m_pSvr->m_lpHandlers.end()) && !(m_pSvr->m_lpHandlers.empty()))
bfictorie 0:8cdad1c73e8e 96 {
bfictorie 0:8cdad1c73e8e 97 DBG("Using default handler\n");
bfictorie 0:8cdad1c73e8e 98 it = m_pSvr->m_lpHandlers.end();
bfictorie 0:8cdad1c73e8e 99 it--; //Get the last element
bfictorie 0:8cdad1c73e8e 100 if( ! (((*it).first.length() == 0) || !(*it).first.compare("/")) ) //This is not the default handler
bfictorie 0:8cdad1c73e8e 101 it = m_pSvr->m_lpHandlers.end();
bfictorie 0:8cdad1c73e8e 102 root_len = 0;
bfictorie 0:8cdad1c73e8e 103 }
bfictorie 0:8cdad1c73e8e 104 if(it == m_pSvr->m_lpHandlers.end())
bfictorie 0:8cdad1c73e8e 105 {
bfictorie 0:8cdad1c73e8e 106 DBG("No handler found\n");
bfictorie 0:8cdad1c73e8e 107 close(); //No handler found
bfictorie 0:8cdad1c73e8e 108 return;
bfictorie 0:8cdad1c73e8e 109 }
bfictorie 0:8cdad1c73e8e 110
bfictorie 0:8cdad1c73e8e 111 DBG("Handler found.\r\n");
bfictorie 0:8cdad1c73e8e 112
bfictorie 0:8cdad1c73e8e 113 //HTTPRequestHandler* pHdlr = (*it).second(rootPath.c_str(), subPath.c_str(), m_pTCPSocket);
bfictorie 0:8cdad1c73e8e 114 //NEW CODE 1 LINE:
bfictorie 0:8cdad1c73e8e 115 HTTPRequestHandler* pHdlr = (*it).second((*it).first.c_str(), path.c_str() + root_len, m_pTCPSocket);
bfictorie 0:8cdad1c73e8e 116 m_pTCPSocket = NULL; //We don't own it anymore
bfictorie 0:8cdad1c73e8e 117
bfictorie 0:8cdad1c73e8e 118 switch(methCode)
bfictorie 0:8cdad1c73e8e 119 {
bfictorie 0:8cdad1c73e8e 120 case HTTP_GET:
bfictorie 0:8cdad1c73e8e 121 pHdlr->doGet();
bfictorie 0:8cdad1c73e8e 122 break;
bfictorie 0:8cdad1c73e8e 123 case HTTP_POST:
bfictorie 0:8cdad1c73e8e 124 pHdlr->doPost();
bfictorie 0:8cdad1c73e8e 125 break;
bfictorie 0:8cdad1c73e8e 126 case HTTP_HEAD:
bfictorie 0:8cdad1c73e8e 127 pHdlr->doHead();
bfictorie 0:8cdad1c73e8e 128 break;
bfictorie 0:8cdad1c73e8e 129 }
bfictorie 0:8cdad1c73e8e 130
bfictorie 0:8cdad1c73e8e 131 DBG("Req handled (or being handled)\r\n");
bfictorie 0:8cdad1c73e8e 132 close();
bfictorie 0:8cdad1c73e8e 133 }
bfictorie 0:8cdad1c73e8e 134
bfictorie 0:8cdad1c73e8e 135 void HTTPRequestDispatcher::close() //Close socket and destroy data
bfictorie 0:8cdad1c73e8e 136 {
bfictorie 0:8cdad1c73e8e 137 if(m_closed)
bfictorie 0:8cdad1c73e8e 138 return;
bfictorie 0:8cdad1c73e8e 139 m_closed = true; //Prevent recursive calling or calling on an object being destructed by someone else
bfictorie 0:8cdad1c73e8e 140 m_watchdog.detach();
bfictorie 0:8cdad1c73e8e 141 if(m_pTCPSocket) //m_pTCPSocket Should only be destroyed if ownership not passed to an handler
bfictorie 0:8cdad1c73e8e 142 {
bfictorie 0:8cdad1c73e8e 143 m_pTCPSocket->resetOnEvent();
bfictorie 0:8cdad1c73e8e 144 m_pTCPSocket->close();
bfictorie 0:8cdad1c73e8e 145 delete m_pTCPSocket; //This fn might have been called by this socket (through an event), so DO NOT DESTROY IT HERE
bfictorie 0:8cdad1c73e8e 146 }
bfictorie 0:8cdad1c73e8e 147 NetService::close();
bfictorie 0:8cdad1c73e8e 148 }
bfictorie 0:8cdad1c73e8e 149
bfictorie 0:8cdad1c73e8e 150
bfictorie 0:8cdad1c73e8e 151 void HTTPRequestDispatcher::onTimeout() //Connection has timed out
bfictorie 0:8cdad1c73e8e 152 {
bfictorie 0:8cdad1c73e8e 153 close();
bfictorie 0:8cdad1c73e8e 154 }
bfictorie 0:8cdad1c73e8e 155
bfictorie 0:8cdad1c73e8e 156 bool HTTPRequestDispatcher::getRequest(string* path, string* meth)
bfictorie 0:8cdad1c73e8e 157 {
bfictorie 0:8cdad1c73e8e 158 char req[128];
bfictorie 0:8cdad1c73e8e 159 char c_path[128];
bfictorie 0:8cdad1c73e8e 160 char c_meth[128];
bfictorie 0:8cdad1c73e8e 161 const int maxLen = 128;
bfictorie 0:8cdad1c73e8e 162 char* p = req;
bfictorie 0:8cdad1c73e8e 163 //Read Line
bfictorie 0:8cdad1c73e8e 164 int ret;
bfictorie 0:8cdad1c73e8e 165 int len = 0;
bfictorie 0:8cdad1c73e8e 166 for(int i = 0; i < maxLen - 1; i++)
bfictorie 0:8cdad1c73e8e 167 {
bfictorie 0:8cdad1c73e8e 168 ret = m_pTCPSocket->recv(p, 1);
bfictorie 0:8cdad1c73e8e 169 if(!ret)
bfictorie 0:8cdad1c73e8e 170 {
bfictorie 0:8cdad1c73e8e 171 break;
bfictorie 0:8cdad1c73e8e 172 }
bfictorie 0:8cdad1c73e8e 173 if( (len > 1) && *(p-1)=='\r' && *p=='\n' )
bfictorie 0:8cdad1c73e8e 174 {
bfictorie 0:8cdad1c73e8e 175 p--;
bfictorie 0:8cdad1c73e8e 176 len-=2;
bfictorie 0:8cdad1c73e8e 177 break;
bfictorie 0:8cdad1c73e8e 178 }
bfictorie 0:8cdad1c73e8e 179 else if( *p=='\n' )
bfictorie 0:8cdad1c73e8e 180 {
bfictorie 0:8cdad1c73e8e 181 len--;
bfictorie 0:8cdad1c73e8e 182 break;
bfictorie 0:8cdad1c73e8e 183 }
bfictorie 0:8cdad1c73e8e 184 p++;
bfictorie 0:8cdad1c73e8e 185 len++;
bfictorie 0:8cdad1c73e8e 186 }
bfictorie 0:8cdad1c73e8e 187 *p = 0;
bfictorie 0:8cdad1c73e8e 188
bfictorie 0:8cdad1c73e8e 189 DBG("Parsing request : %s\r\n", req);
bfictorie 0:8cdad1c73e8e 190
bfictorie 0:8cdad1c73e8e 191 ret = sscanf(req, "%s %s HTTP/%*d.%*d", c_meth, c_path);
bfictorie 0:8cdad1c73e8e 192 if(ret !=2)
bfictorie 0:8cdad1c73e8e 193 return false;
bfictorie 0:8cdad1c73e8e 194
bfictorie 0:8cdad1c73e8e 195 *meth = string(c_meth);
bfictorie 0:8cdad1c73e8e 196 // NEW CODE (old code removed):
bfictorie 0:8cdad1c73e8e 197 *path = string(c_path);
bfictorie 0:8cdad1c73e8e 198 return true;
bfictorie 0:8cdad1c73e8e 199 }
bfictorie 0:8cdad1c73e8e 200
bfictorie 0:8cdad1c73e8e 201
bfictorie 0:8cdad1c73e8e 202
bfictorie 0:8cdad1c73e8e 203 void HTTPRequestDispatcher::onTCPSocketEvent(TCPSocketEvent e)
bfictorie 0:8cdad1c73e8e 204 {
bfictorie 0:8cdad1c73e8e 205
bfictorie 0:8cdad1c73e8e 206 DBG("\r\nEvent %d\r\n", e);
bfictorie 0:8cdad1c73e8e 207
bfictorie 0:8cdad1c73e8e 208 if(m_closed)
bfictorie 0:8cdad1c73e8e 209 {
bfictorie 0:8cdad1c73e8e 210 DBG("\r\nWARN: Discarded\r\n");
bfictorie 0:8cdad1c73e8e 211 return;
bfictorie 0:8cdad1c73e8e 212 }
bfictorie 0:8cdad1c73e8e 213
bfictorie 0:8cdad1c73e8e 214 switch(e)
bfictorie 0:8cdad1c73e8e 215 {
bfictorie 0:8cdad1c73e8e 216 case TCPSOCKET_READABLE:
bfictorie 0:8cdad1c73e8e 217 m_watchdog.detach();
bfictorie 0:8cdad1c73e8e 218 m_pTCPSocket->resetOnEvent();
bfictorie 0:8cdad1c73e8e 219 //Req arrived, dispatch :
bfictorie 0:8cdad1c73e8e 220 dispatchRequest();
bfictorie 0:8cdad1c73e8e 221 break;
bfictorie 0:8cdad1c73e8e 222 case TCPSOCKET_CONTIMEOUT:
bfictorie 0:8cdad1c73e8e 223 case TCPSOCKET_CONRST:
bfictorie 0:8cdad1c73e8e 224 case TCPSOCKET_CONABRT:
bfictorie 0:8cdad1c73e8e 225 case TCPSOCKET_ERROR:
bfictorie 0:8cdad1c73e8e 226 case TCPSOCKET_DISCONNECTED:
bfictorie 0:8cdad1c73e8e 227 close();
bfictorie 0:8cdad1c73e8e 228 break;
bfictorie 0:8cdad1c73e8e 229 }
bfictorie 0:8cdad1c73e8e 230
bfictorie 0:8cdad1c73e8e 231 }