Added request parser based on Boost.Spirit.

1 files changed, 58 insertions, 3 deletions

diff --git a/http_connection.cpp b/http_connection.cpp
index 36c0f00..fb45aac 100644
--- a/http_connection.cpp
+++ b/http_connection.cpp
@@ -4,9 +4,10 @@
 #include "music.h"
 
 #include <boost/bind.hpp>
-#include <boost/algorithm/string/split.hpp>
-#include <boost/algorithm/string/classification.hpp>
-#include <boost/algorithm/string/predicate.hpp>
+#include <boost/spirit/include/qi.hpp>
+#include <boost/fusion/include/std_pair.hpp>
+
+namespace qi = boost::spirit::qi;
 
 HTTP::Connection::Connection(boost::asio::io_service& io_service) : socket(io_service) {
 }
@@ -15,7 +16,30 @@ void HTTP::Connection::handle_write(const boost::system::error_code& error, size
 }
 
 void HTTP::Connection::handle_read(const boost::system::error_code& error, size_t bytes_transferred) {
+	if(!parse_request(buf)) {
+		// Request parse error.
+		return;
+	}
+	
+	std::cout << "Path: " << std::endl;
+	for(std::vector<std::string>::iterator it = path.begin(); it != path.end(); it++) {
+		std::cout << "  " << *it << std::endl;
+	}
+	
+	std::cout << "Args: " << std::endl;
+	for(std::map<std::string, std::string>::iterator it = args.begin(); it != args.end(); it++) {
+		std::cout << "  " << it->first << " = " << it->second << std::endl;
+	}
+	
+	std::cout << "Headers: " << std::endl;
+	for(std::map<std::string, std::string>::iterator it = headers.begin(); it != headers.end(); it++) {
+		std::cout << "  " << it->first << " = " << it->second << std::endl;
+	}
+	
+	return;
+	
 	std::istream is(&buf);
+	
 	HTTPRequest req(is);
 
 	HTTPResponse res(socket);
@@ -32,6 +56,37 @@ void HTTP::Connection::handle_read(const boost::system::error_code& error, size_
 	}
 }
 
+void print(char c) {
+	std::cout << "Char: " << int(c) << std::endl;
+}
+
+bool HTTP::Connection::parse_request(boost::asio::streambuf& buf) {
+	typedef boost::asio::buffers_iterator<boost::asio::streambuf::const_buffers_type> Iterator;
+	
+	Iterator begin = Iterator::begin(buf.data());
+	Iterator end = Iterator::end(buf.data());
+	
+	qi::rule<Iterator, std::string()> word_p = +(qi::char_ - ' ' - '\r');
+	
+	qi::rule<Iterator, std::vector<std::string>()> path_p = '/' >> +(qi::char_ - '/' - '?' - ' ') % '/';
+	
+	qi::rule<Iterator, std::pair<std::string, std::string>()> pair_p = +(qi::char_ - '=') >> '=' >> +(qi::char_ - '&' - ' ');
+	qi::rule<Iterator, std::map<std::string, std::string>()> args_p = '?' >> pair_p >> *('&' >> pair_p);
+	
+	qi::rule<Iterator, std::pair<std::string, std::string>()> header_p = +(qi::char_ - ':') >> ": " >> +(qi::char_ - '\r');
+	qi::rule<Iterator, std::map<std::string, std::string>()> headers_p = *(header_p >> "\r\n");
+	
+	return qi::parse(begin, end,
+		// Method, path, args, version:
+		word_p >> ' ' >> path_p >> -args_p >> ' ' >> word_p >> "\r\n" >>
+		// Headers:
+		headers_p >>
+		// End of headers:
+		"\r\n",
+		// Store into:
+		method, path, args, version, headers);
+}
+
 void HTTP::Connection::start() {
 	boost::asio::async_read_until(socket, buf, "\r\n\r\n", boost::bind(&Connection::handle_read, shared_from_this(),
 				boost::asio::placeholders::error, boost::asio::placeholders::bytes_transferred));