From 5fa7671ef77b61766c04c42c1023b0356889eaab Mon Sep 17 00:00:00 2001
From: Meeh <meeh@i2pmail.org>
Date: Sat, 29 Mar 2014 18:28:19 +0100
Subject: [PATCH] Rewriting url parser

---
 HTTPProxy.cpp | 53 +++++++++++++++++++++++++--------------------------
 HTTPProxy.h   |  3 ++-
 2 files changed, 28 insertions(+), 28 deletions(-)

diff --git a/HTTPProxy.cpp b/HTTPProxy.cpp
index c9a763ce..61264db2 100644
--- a/HTTPProxy.cpp
+++ b/HTTPProxy.cpp
@@ -1,6 +1,7 @@
 #include <boost/bind.hpp>
 #include <boost/lexical_cast.hpp>
 #include <boost/bind/protect.hpp>
+#include <boost/regex.hpp>
 
 #include "base64.h"
 #include "Log.h"
@@ -62,12 +63,12 @@ namespace proxy
 		{
 			m_Buffer[bytes_transferred] = 0;
 			
-			std::pair<std::string,std::string> requestInfo = ExtractRequest ();
 			request m_Request;
+			ExtractRequest (m_Request);
 			parseHeaders (m_Buffer, m_Request.headers);
 			
-			LogPrint("Requesting ", requestInfo.first, " with path ", requestInfo.second);
-			HandleDestinationRequest (requestInfo.first, requestInfo.second);
+			LogPrint("Requesting ", m_Request.host, " with path ", m_Request.uri, " and method ", m_Request.method);
+			HandleDestinationRequest (m_Request.host, m_Request.uri);
 		}
 		else if (ecode != boost::asio::error::operation_aborted)
 			Terminate ();
@@ -96,33 +97,31 @@ namespace proxy
 		}
 	}
 
-	// TODO: Support other requests than GET.
-	std::pair<std::string, std::string> HTTPConnection::ExtractRequest ()
+	void HTTPConnection::ExtractRequest (request &m_Request)
 	{
-		char * get = strstr (m_Buffer, "GET");
-		if (get)
-		{
-			char * http = strstr (get, "HTTP");
-			if (http)
-			{
-				std::string url (get + 4, http - get - 5);
-				size_t sp = url.find_first_of ('/', 7 /* skip http:// part */ );
-				if (sp != std::string::npos)
-				{
-					std::string base_url (url.begin()+7, url.begin()+sp);
-					LogPrint ("Base URL is: ", base_url, "\n");
-					if ( sp != std::string::npos )
-					{
-						std::string query (url.begin ()+sp+1, url.end ());
-						LogPrint ("Query is: ", "/" + query);
-
-						return std::make_pair (base_url, "/" + query);
-					}
-					return std::make_pair (base_url, "/");
-				}
+		std::string requestString = m_Buffer;
+		int idx=requestString.find(" ");
+		std::string method = requestString.substr(0,idx);
+		requestString = requestString.substr(idx+1);
+		idx=requestString.find(" ");
+		std::string requestUrl = requestString.substr(0,idx);
+		LogPrint("method is: ", method, "\nRequest is: ", requestUrl);
+		std::string server="";
+		std::string port="80";
+		boost::regex rHTTP("http://(.*?)(:(\\d+))?(/.*)");
+		boost::smatch m;
+		std::string path;
+		if(boost::regex_search(requestUrl, m, rHTTP, boost::match_extra)) {
+			server=m[1].str();
+			if(m[2].str() != "") {
+				port=m[3].str();
 			}
+			path=m[4].str();
 		}
-		return std::make_pair ("","");
+		LogPrint("server is: ",server, "\n path is: ",path);
+		m_Request.uri = path;
+		m_Request.method = method;
+		m_Request.host = server;
 	}
 	
 	void HTTPConnection::HandleWriteReply (const boost::system::error_code& ecode)
diff --git a/HTTPProxy.h b/HTTPProxy.h
index 2b172542..c4370daa 100644
--- a/HTTPProxy.h
+++ b/HTTPProxy.h
@@ -24,6 +24,7 @@ namespace proxy
 		{
 		  std::string method;
 		  std::string uri;
+		  std::string host;
 		  int http_version_major;
 		  int http_version_minor;
 		  std::vector<header> headers;
@@ -54,7 +55,7 @@ namespace proxy
 			void SendReply (const std::string& content);
 
 			void HandleDestinationRequest (const std::string& address, const std::string& uri);
-			std::pair<std::string, std::string> ExtractRequest ();
+			void ExtractRequest (request& m_Request);
 			void parseHeaders(const std::string& h, std::vector<header>& hm);
 			
 		private: