| 25 | } |
| 26 | |
| 27 | void request_handler::handle_request(const request& req, reply& rep) |
| 28 | { |
| 29 | // Decode url to path. |
| 30 | std::string request_path; |
| 31 | if (!url_decode(req.uri, request_path)) |
| 32 | { |
| 33 | rep = reply::stock_reply(reply::bad_request); |
| 34 | return; |
| 35 | } |
| 36 | |
| 37 | // Request path must be absolute and not contain "..". |
| 38 | if (request_path.empty() || request_path[0] != '/' |
| 39 | || request_path.find("..") != std::string::npos) |
| 40 | { |
| 41 | rep = reply::stock_reply(reply::bad_request); |
| 42 | return; |
| 43 | } |
| 44 | |
| 45 | // If path ends in slash (i.e. is a directory) then add "index.html". |
| 46 | if (request_path[request_path.size() - 1] == '/') |
| 47 | { |
| 48 | request_path += "index.html"; |
| 49 | } |
| 50 | |
| 51 | // Determine the file extension. |
| 52 | std::size_t last_slash_pos = request_path.find_last_of("/"); |
| 53 | std::size_t last_dot_pos = request_path.find_last_of("."); |
| 54 | std::string extension; |
| 55 | if (last_dot_pos != std::string::npos && last_dot_pos > last_slash_pos) |
| 56 | { |
| 57 | extension = request_path.substr(last_dot_pos + 1); |
| 58 | } |
| 59 | |
| 60 | // Open the file to send back. |
| 61 | std::string full_path = doc_root_ + request_path; |
| 62 | std::ifstream is(full_path.c_str(), std::ios::in | std::ios::binary); |
| 63 | if (!is) |
| 64 | { |
| 65 | rep = reply::stock_reply(reply::not_found); |
| 66 | return; |
| 67 | } |
| 68 | |
| 69 | // Fill out the reply to be sent to the client. |
| 70 | rep.status = reply::ok; |
| 71 | char buf[512]; |
| 72 | while (is.read(buf, sizeof(buf)).gcount() > 0) |
| 73 | rep.content.append(buf, is.gcount()); |
| 74 | rep.headers.resize(2); |
| 75 | rep.headers[0].name = "Content-Length"; |
| 76 | rep.headers[0].value = std::to_string(rep.content.size()); |
| 77 | rep.headers[1].name = "Content-Type"; |
| 78 | rep.headers[1].value = mime_types::extension_to_type(extension); |
| 79 | } |
| 80 | |
| 81 | bool request_handler::url_decode(const std::string& in, std::string& out) |
| 82 | { |