diff options
-rw-r--r-- | http_connection.cpp | 21 | ||||
-rw-r--r-- | http_connection.h | 2 | ||||
-rw-r--r-- | httpd.cpp | 2 | ||||
-rw-r--r-- | main.cpp | 2 |
4 files changed, 21 insertions, 6 deletions
diff --git a/http_connection.cpp b/http_connection.cpp index ff7e795..a16d237 100644 --- a/http_connection.cpp +++ b/http_connection.cpp @@ -5,8 +5,23 @@ #include <boost/bind.hpp> #include <boost/format.hpp> -void HTTP::Connection::send_error(int code, std::string name) { - boost::asio::write(socket, boost::asio::buffer(boost::str(boost::format("HTTP/1.1 %1$d %2$s\r\n\r\n<h1>%1$d %2$s</h1>") % code % name))); +namespace response_map_init { + typedef std::pair<int, std::string> P; + const P m[] = { + P(200, "OK"), + P(400, "Bad Request"), + P(404, "Not Found"), + P(500, "Internal Server Error"), + P(501, "Not Implemented") + }; + const P* begin = m; + const P* end = m + sizeof(m) / sizeof(P); +} + +const std::map<int, std::string> response_map(response_map_init::begin, response_map_init::end); + +void HTTP::Connection::send_error(int code) { + boost::asio::write(socket, boost::asio::buffer(boost::str(boost::format("HTTP/1.1 %1$d %2$s\r\n\r\n<h1>%1$d %2$s</h1>") % code % response_map.find(code)->second))); } void HTTP::Connection::add_header(std::string key, std::string value) { @@ -47,7 +62,7 @@ void HTTP::Connection::handle_read(const boost::system::error_code& error, size_ if(!parse_request(buf)) { // Request parse error. std::cout << "Request parse error." << std::endl; - send_error(400, "Bad Request"); + send_error(400); return; } diff --git a/http_connection.h b/http_connection.h index e3a3721..d810254 100644 --- a/http_connection.h +++ b/http_connection.h @@ -44,7 +44,7 @@ namespace HTTP { std::map<std::string, std::string> headers; //! Send error. - void send_error(int code, std::string name); + void send_error(int code); //! Add response header. void add_header(std::string key, std::string value); @@ -29,6 +29,6 @@ void HTTP::Server::handle_request(Connection::p connection) { // Call handler. handlers[handler](connection); } else { - connection->send_error(404, "Not Found"); + connection->send_error(404); } } @@ -17,7 +17,7 @@ void foo_handler(HTTP::Connection::p connection) { if(ml) { ml->render(connection); } else { - connection->send_error(404, "Not Found"); + connection->send_error(404); } } |