A modified version of http://mbed.org/users/segundo/libraries/NetServices/ljhqix to add HTTP proxy feature (based on http://mbed.org/users/igorsk/programs/NetServicesSource/ltjpag)

Dependents:   SenseClient

Committer:
mimil
Date:
Tue Sep 06 13:26:45 2011 +0000
Revision:
0:308f83189a3f

        

Who changed what in which revision?

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