diff options
| -rw-r--r-- | http_connection.cpp | 1 | ||||
| -rw-r--r-- | http_static.cpp | 33 | ||||
| -rw-r--r-- | http_static.h | 26 | ||||
| -rw-r--r-- | main.cpp | 6 |
4 files changed, 66 insertions, 0 deletions
diff --git a/http_connection.cpp b/http_connection.cpp index f04dc28..041cd05 100644 --- a/http_connection.cpp +++ b/http_connection.cpp @@ -10,6 +10,7 @@ namespace response_map_init { const P m[] = { P(200, "OK"), P(400, "Bad Request"), + P(403, "Forbidden"), P(404, "Not Found"), P(500, "Internal Server Error"), P(501, "Not Implemented") diff --git a/http_static.cpp b/http_static.cpp new file mode 100644 index 0000000..7f56744 --- /dev/null +++ b/http_static.cpp @@ -0,0 +1,33 @@ +#include "http_static.h" + +#include <boost/filesystem/fstream.hpp> +#include <boost/format.hpp> + +HTTP::Static::Static(fs::path directory, std::string index) : directory_(directory), index_(index) { + +} + +void HTTP::Static::operator()(Connection::p connection) { + fs::path path = directory_; + + if(connection->path.size()) { + for(Connection::PathList::iterator it = connection->path.begin(); it != connection->path.end(); it++) { + path /= *it; + } + } else { + path /= index_; + } + + if(fs::is_regular_file(path)) { + fs::ifstream is(path, std::ios::in | std::ios::binary); + is.seekg(0, std::ios::end); + connection->add_header("content-length", boost::str(boost::format("%d") % is.tellg())); + is.seekg(0, std::ios::beg); + + connection->send_data(is); + } else if(fs::exists(path)) { + connection->send_error(403); + } else { + connection->send_error(404); + } +} diff --git a/http_static.h b/http_static.h new file mode 100644 index 0000000..f8d2bec --- /dev/null +++ b/http_static.h @@ -0,0 +1,26 @@ +#ifndef HTTP_STATIC_H +#define HTTP_STATIC_H + +#include "http_connection.h" + +#include <boost/filesystem.hpp> + +namespace fs = boost::filesystem; + +namespace HTTP { + //! HTTP Handler serving static files from a directory. + class Static { + private: + fs::path directory_; + std::string index_; + + public: + //! Construct Static-handler. + Static(fs::path directory, std::string index = "index.html"); + + //! Handle request. + void operator()(Connection::p connection); + }; +}; + +#endif @@ -5,6 +5,8 @@ #include "httpd.h" #include "telnetd.h" +#include "http_static.h" + #include <iostream> #include <vector> @@ -34,6 +36,10 @@ int main(int argc, char **argv) { httpd.add_handler("files", &foo_handler); + HTTP::Static static_files("static"); + httpd.add_handler("files", static_files); + httpd.add_handler("", static_files); + telnet::Server telnetd(io_service, tcp::endpoint(tcp::v6(), config::vm["audist.telnetd_port"].as<int>())); std::size_t num_threads = config::vm["audist.threads"].as<std::size_t>(); |