diff options
Diffstat (limited to 'thirdparty/cpr/test/httpServer.cpp')
| -rw-r--r-- | thirdparty/cpr/test/httpServer.cpp | 927 |
1 files changed, 0 insertions, 927 deletions
diff --git a/thirdparty/cpr/test/httpServer.cpp b/thirdparty/cpr/test/httpServer.cpp deleted file mode 100644 index 9514de51f..000000000 --- a/thirdparty/cpr/test/httpServer.cpp +++ /dev/null @@ -1,927 +0,0 @@ -#include "httpServer.hpp" -#include <chrono> -#include <string> -#include <system_error> -#include <thread> - -namespace cpr { - -std::string HttpServer::GetBaseUrl() { - return "http://127.0.0.1:" + std::to_string(GetPort()); -} - -uint16_t HttpServer::GetPort() { - // Unassigned port number in the ephemeral range - return 61936; -} - -mg_connection* HttpServer::initServer(mg_mgr* mgr, mg_event_handler_t event_handler) { - // Based on: https://mongoose.ws/tutorials/http-server/ - mg_mgr_init(mgr); - std::string port = std::to_string(GetPort()); - mg_connection* c = mg_http_listen(mgr, GetBaseUrl().c_str(), event_handler, this); - if (!c) { - throw std::system_error(errno, std::system_category(), "Failed to listen at port " + port); - } - return c; -} - -void HttpServer::acceptConnection(mg_connection* /* conn */) {} - -void HttpServer::OnRequestHello(mg_connection* conn, mg_http_message* msg) { - if (std::string{msg->method.ptr, msg->method.len} == std::string{"OPTIONS"}) { - OnRequestOptions(conn, msg); - } else { - std::string response{"Hello world!"}; - std::string headers = "Content-Type: text/html\r\n"; - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); - } -} - -void HttpServer::OnRequestRoot(mg_connection* conn, mg_http_message* msg) { - if (std::string{msg->method.ptr, msg->method.len} == std::string{"OPTIONS"}) { - OnRequestOptions(conn, msg); - } else { - std::string errorMessage{"Method Not Allowed"}; - SendError(conn, 405, errorMessage); - } -} - -void HttpServer::OnRequestNotFound(mg_connection* conn, mg_http_message* msg) { - if (std::string{msg->method.ptr, msg->method.len} == std::string{"OPTIONS"}) { - OnRequestOptions(conn, msg); - } else { - std::string errorMessage{"Not Found"}; - SendError(conn, 404, errorMessage); - } -} - -void HttpServer::OnRequestOptions(mg_connection* conn, mg_http_message* /*msg*/) { - std::string headers = - "Content-Type: text/plain\r\n" - "Access-Control-Allow-Origin: *\r\n" - "Access-Control-Allow-Credentials: true\r\n" - "Access-Control-Allow-Methods: GET, POST, PUT, DELETE, PATCH, OPTIONS\r\n" - "Access-Control-Max-Age: 3600\r\n"; - - std::string response; - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); -} - -void HttpServer::OnRequestTimeout(mg_connection* conn, mg_http_message* msg) { - std::this_thread::sleep_for(std::chrono::milliseconds(100)); - OnRequestHello(conn, msg); -} - -void HttpServer::OnRequestLongTimeout(mg_connection* conn, mg_http_message* msg) { - std::this_thread::sleep_for(std::chrono::seconds(2)); - OnRequestHello(conn, msg); -} - -// Send the header, then send "Hello world!" every 100ms -// For this, we use a mongoose timer -void HttpServer::OnRequestLowSpeedTimeout(mg_connection* conn, mg_http_message* /* msg */, TimerArg* timer_arg) { - std::string response{"Hello world!"}; - mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\nContent-Length: %d\r\n\r\n", response.length() * 20); - timer_arg->connection_id = conn->id; - mg_timer_init( - &timer_arg->mgr->timers, &timer_arg->timer, 100, MG_TIMER_REPEAT, - // The following lambda function gets executed each time the timer is called. - // It sends "Hello world!" to the client each 100ms at most 20 times. - [](void* arg) { - TimerArg* timer_arg = static_cast<TimerArg*>(arg); - if (timer_arg->counter < 20 && IsConnectionActive(timer_arg->mgr, timer_arg->connection) && timer_arg->connection->id == timer_arg->connection_id) { - std::string response{"Hello world!"}; - mg_send(timer_arg->connection, response.c_str(), response.length()); - ++timer_arg->counter; - } else { - timer_arg->counter = 20; // Make sure that this timer is never called again - } - }, - timer_arg); -} - -// Before and after calling an endpoint that calls this method, the test needs to wait until all previous connections are closed -// The nested call to mg_mgr_poll can lead to problems otherwise -void HttpServer::OnRequestLowSpeed(mg_connection* conn, mg_http_message* /*msg*/, mg_mgr* mgr) { - std::string response{"Hello world!"}; - mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\nContent-Length: %d\r\n\r\n", response.length()); - mg_timer_add( - mgr, 2000, MG_TIMER_ONCE, - [](void* connection) { - std::string response{"Hello world!"}; - mg_send(static_cast<mg_connection*>(connection), response.c_str(), response.length()); - }, - conn); -} - -// Before and after calling an endpoint that calls this method, the test needs to wait until all previous connections are closed -// The nested call to mg_mgr_poll can lead to problems otherwise -void HttpServer::OnRequestLowSpeedBytes(mg_connection* conn, mg_http_message* /*msg*/, TimerArg* timer_arg) { - std::string response{'a'}; - mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\nContent-Length: %d\r\n\r\n", response.length() * 20); - - mg_timer_init( - &timer_arg->mgr->timers, &timer_arg->timer, 100, MG_TIMER_REPEAT, - // The following lambda function gets executed each time the timer is called. - // It first waits for 2 seconds, then sends "a" to the client each 100ms at most 20 times. - [](void* arg) { - TimerArg* timer_arg = static_cast<TimerArg*>(arg); - if (timer_arg->counter == 0) { - std::this_thread::sleep_for(std::chrono::seconds(2)); - } - if (timer_arg->counter < 20 && IsConnectionActive(timer_arg->mgr, timer_arg->connection) && timer_arg->connection->id == timer_arg->connection_id) { - std::string response{'a'}; - mg_send(timer_arg->connection, response.c_str(), response.length()); - ++timer_arg->counter; - } else { - timer_arg->counter = 20; // Make sure that this timer is never called again - } - }, - timer_arg); -} - -void HttpServer::OnRequestBasicCookies(mg_connection* conn, mg_http_message* /*msg*/) { - time_t expires_time = 3905119080; // Expires=Wed, 30 Sep 2093 03:18:00 GMT - std::array<char, EXPIRES_STRING_SIZE> expires_string; - std::strftime(expires_string.data(), expires_string.size(), "%a, %d %b %Y %H:%M:%S GMT", gmtime(&expires_time)); - - std::string cookie1{"SID=31d4d96e407aad42; Expires=" + std::string(expires_string.data()) + "; Secure"}; - std::string cookie2{"lang=en-US; Expires=" + std::string(expires_string.data()) + "; Secure"}; - std::string headers = - "Content-Type: text/html\r\n" - "Set-Cookie: " + - cookie1 + - "\r\n" - "Set-Cookie: " + - cookie2 + "\r\n"; - std::string response{"Basic Cookies"}; - - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); -} - -void HttpServer::OnRequestEmptyCookies(mg_connection* conn, mg_http_message* /*msg*/) { - time_t expires_time = 3905119080; // Expires=Wed, 30 Sep 2093 03:18:00 GMT - std::array<char, EXPIRES_STRING_SIZE> expires_string; - std::strftime(expires_string.data(), sizeof(expires_string), "%a, %d %b %Y %H:%M:%S GMT", gmtime(&expires_time)); - - std::string cookie1{"SID=; Expires=" + std::string(expires_string.data()) + "; Secure"}; - std::string cookie2{"lang=; Expires=" + std::string(expires_string.data()) + "; Secure"}; - std::string headers = - "Content-Type: text/html\r\n" - "Set-Cookie: " + - cookie1 + - "\r\n" - "Set-Cookie: " + - cookie2 + "\r\n"; - std::string response{"Empty Cookies"}; - - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); -} - -void HttpServer::OnRequestCookiesReflect(mg_connection* conn, mg_http_message* msg) { - mg_str* request_cookies; - if ((request_cookies = mg_http_get_header(msg, "Cookie")) == nullptr) { - std::string errorMessage{"Cookie not found"}; - SendError(conn, 400, errorMessage); - return; - } - std::string cookie_str{request_cookies->ptr, request_cookies->len}; - std::string headers = "Content-Type: text/html\r\n"; - mg_http_reply(conn, 200, headers.c_str(), cookie_str.c_str()); -} - -void HttpServer::OnRequestRedirectionWithChangingCookies(mg_connection* conn, mg_http_message* msg) { - time_t expires_time = 3905119080; // Expires=Wed, 30 Sep 2093 03:18:00 GMT - std::array<char, EXPIRES_STRING_SIZE> expires_string; - std::strftime(expires_string.data(), sizeof(expires_string), "%a, %d %b %Y %H:%M:%S GMT", gmtime(&expires_time)); - - mg_str* request_cookies; - std::string cookie_str; - if ((request_cookies = mg_http_get_header(msg, "Cookie")) != nullptr) { - cookie_str = std::string{request_cookies->ptr, request_cookies->len}; - } - - if (cookie_str.find("SID=31d4d96e407aad42") == std::string::npos) { - std::string cookie1{"SID=31d4d96e407aad42; Expires=" + std::string(expires_string.data()) + "; Secure"}; - std::string cookie2{"lang=en-US; Expires=" + std::string(expires_string.data()) + "; Secure"}; - std::string headers = - "Content-Type: text/html\r\n" - "Location: http://127.0.0.1:61936/redirection_with_changing_cookies.html\r\n" - "Set-Cookie: " + - cookie1 + - "\r\n" - "Set-Cookie: " + - cookie2 + "\r\n"; - - mg_http_reply(conn, 302, headers.c_str(), ""); - } else { - cookie_str = "Received cookies are: " + cookie_str; - std::string headers = "Content-Type: text/html\r\n"; - mg_http_reply(conn, 200, headers.c_str(), cookie_str.c_str()); - } -} - -void HttpServer::OnRequestBasicAuth(mg_connection* conn, mg_http_message* msg) { - mg_str* requested_auth; - std::string auth{"Basic"}; - if ((requested_auth = mg_http_get_header(msg, "Authorization")) == nullptr || mg_ncasecmp(requested_auth->ptr, auth.c_str(), auth.length()) != 0) { - std::string errorMessage{"Unauthorized"}; - SendError(conn, 401, errorMessage); - return; - } - std::string auth_string{requested_auth->ptr, requested_auth->len}; - size_t basic_token = auth_string.find(' ') + 1; - auth_string = auth_string.substr(basic_token, auth_string.length() - basic_token); - auth_string = Base64Decode(auth_string); - size_t colon = auth_string.find(':'); - std::string username = auth_string.substr(0, colon); - std::string password = auth_string.substr(colon + 1, auth_string.length() - colon - 1); - if (username == "user" && password == "password") { - OnRequestHeaderReflect(conn, msg); - } else { - std::string errorMessage{"Unauthorized"}; - SendError(conn, 401, errorMessage); - } -} - -void HttpServer::OnRequestBearerAuth(mg_connection* conn, mg_http_message* msg) { - mg_str* requested_auth; - std::string auth{"Bearer"}; - if ((requested_auth = mg_http_get_header(msg, "Authorization")) == nullptr || mg_ncasecmp(requested_auth->ptr, auth.c_str(), auth.length()) != 0) { - std::string errorMessage{"Unauthorized"}; - SendError(conn, 401, errorMessage); - return; - } - std::string auth_string{requested_auth->ptr, requested_auth->len}; - size_t basic_token = auth_string.find(' ') + 1; - auth_string = auth_string.substr(basic_token, auth_string.length() - basic_token); - if (auth_string == "the_token") { - OnRequestHeaderReflect(conn, msg); - } else { - std::string errorMessage{"Unauthorized"}; - SendError(conn, 401, errorMessage); - } -} - -void HttpServer::OnRequestBasicJson(mg_connection* conn, mg_http_message* /*msg*/) { - std::string response = - "[\n" - " {\n" - " \"first_key\": \"first_value\",\n" - " \"second_key\": \"second_value\"\n" - " }\n" - "]"; - std::string headers = "Content-Type: application/json\r\n"; - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); -} - -void HttpServer::OnRequestHeaderReflect(mg_connection* conn, mg_http_message* msg) { - std::string response = "Header reflect " + std::string{msg->method.ptr, msg->method.len}; - std::string headers; - bool hasContentTypeHeader = false; - for (const mg_http_header& header : msg->headers) { - if (!header.name.ptr) { - continue; - } - - std::string name = std::string(header.name.ptr, header.name.len); - if (std::string{"Content-Type"} == name) { - hasContentTypeHeader = true; - } - - if (std::string{"Host"} != name && std::string{"Accept"} != name) { - if (header.value.ptr) { - headers.append(name + ": " + std::string(header.value.ptr, header.value.len) + "\r\n"); - } - } - } - - if (!hasContentTypeHeader) { - headers.append("Content-Type: text/html\r\n"); - } - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); -} - -void HttpServer::OnRequestTempRedirect(mg_connection* conn, mg_http_message* msg) { - // Get the requested target location: - std::string location; - for (mg_http_header& header : msg->headers) { - if (!header.name.ptr) { - continue; - } - - std::string name = std::string(header.name.ptr, header.name.len); - if (std::string{"RedirectLocation"} == name) { - location = std::string(header.value.ptr, header.value.len); - break; - } - } - - // Check if the request contains a valid location, else default to 'hello.html': - if (location.empty()) { - location = "hello.html"; - } - std::string headers = "Location: " + location + "\r\n"; - std::string response = "Moved Temporarily"; - mg_http_reply(conn, 302, headers.c_str(), response.c_str()); -} - -void HttpServer::OnRequestPermRedirect(mg_connection* conn, mg_http_message* msg) { - // Get the requested target location: - std::string location; - for (mg_http_header& header : msg->headers) { - if (!header.name.ptr) { - continue; - } - - std::string name = std::string(header.name.ptr, header.name.len); - if (std::string{"RedirectLocation"} == name) { - location = std::string(header.value.ptr, header.value.len); - break; - } - } - - // Check if the request contains a valid location, else default to 'hello.html': - if (location.empty()) { - location = "hello.html"; - } - std::string headers = "Location: " + location + "\r\n"; - std::string response = "Moved Permanently"; - - mg_http_reply(conn, 301, headers.c_str(), response.c_str()); -} - -void HttpServer::OnRequestResolvePermRedirect(mg_connection* conn, mg_http_message* msg) { - // Get the requested target location: - std::string location; - for (mg_http_header& header : msg->headers) { - if (!header.name.ptr) { - continue; - } - - std::string name = std::string(header.name.ptr, header.name.len); - if (std::string{"RedirectLocation"} == name) { - location = std::string(header.value.ptr, header.value.len); - break; - } - } - - if (location.empty()) { - std::string errorMessage{"Redirect location missing"}; - SendError(conn, 405, errorMessage); - return; - } - - std::string headers = "Location: " + location + "\r\n"; - std::string response = "Moved Permanently"; - - mg_http_reply(conn, 301, headers.c_str(), response.c_str()); -} - -void HttpServer::OnRequestTwoRedirects(mg_connection* conn, mg_http_message* /*msg*/) { - std::string response = "Moved Permanently"; - std::string headers = "Location: permanent_redirect.html\r\n"; - mg_http_reply(conn, 301, headers.c_str(), response.c_str()); -} - -void HttpServer::OnRequestUrlPost(mg_connection* conn, mg_http_message* msg) { - if (std::string{msg->method.ptr, msg->method.len} != std::string{"POST"}) { - std::string errorMessage{"Method Not Allowed"}; - SendError(conn, 405, errorMessage); - return; - } - - std::string headers = "Content-Type: application/json\r\n"; - - char x[100]; - char y[100]; - mg_http_get_var(&(msg->body), "x", x, sizeof(x)); - mg_http_get_var(&(msg->body), "y", y, sizeof(y)); - std::string x_string{x}; - std::string y_string{y}; - std::string response; - if (y_string.empty()) { - response = std::string{ - "{\n" - " \"x\": " + - x_string + - "\n" - "}"}; - } else { - response = std::string{ - "{\n" - " \"x\": " + - x_string + - ",\n" - " \"y\": " + - y_string + - ",\n" - " \"sum\": " + - std::to_string(atoi(x) + atoi(y)) + - "\n" - "}"}; - } - mg_http_reply(conn, 201, headers.c_str(), response.c_str()); -} - -void HttpServer::OnRequestBodyGet(mg_connection* conn, mg_http_message* msg) { - if (std::string{msg->method.ptr, msg->method.len} != std::string{"GET"}) { - std::string errorMessage{"Method Not Allowed"}; - SendError(conn, 405, errorMessage); - return; - } - std::array<char, 100> message{}; - mg_http_get_var(&(msg->body), "message", message.data(), message.size()); - if (msg->body.len <= 0) { - std::string errorMessage{"No Content"}; - SendError(conn, 405, errorMessage); - return; - } - std::string response = message.data(); - std::string headers = "Content-Type: text/html\r\n"; - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); -} - -void HttpServer::OnRequestJsonPost(mg_connection* conn, mg_http_message* msg) { - mg_str* content_type{nullptr}; - if ((content_type = mg_http_get_header(msg, "Content-Type")) == nullptr || std::string{content_type->ptr, content_type->len} != "application/json") { - std::string errorMessage{"Unsupported Media Type"}; - SendError(conn, 415, errorMessage); - return; - } - - std::string headers = "Content-Type: application/json\r\n"; - mg_http_reply(conn, 201, headers.c_str(), msg->body.ptr); -} - -void HttpServer::OnRequestFormPost(mg_connection* conn, mg_http_message* msg) { - size_t pos{0}; - mg_http_part part{}; - - std::string headers = "Content-Type: application/json\r\n"; - std::string response{}; - response += "{\n"; - while ((pos = mg_http_next_multipart(msg->body, pos, &part)) > 0) { - response += " \"" + std::string(part.name.ptr, part.name.len) + "\": \""; - if (!std::string(part.filename.ptr, part.filename.len).empty()) { - response += std::string(part.filename.ptr, part.filename.len) + "="; - } - response += std::string(part.body.ptr, part.body.len) + "\",\n"; - } - response.erase(response.find_last_not_of(",\n") + 1); - response += "\n}"; - - mg_http_reply(conn, 201, headers.c_str(), response.c_str()); -} - -void HttpServer::OnRequestDelete(mg_connection* conn, mg_http_message* msg) { - bool has_json_header = false; - for (mg_http_header& header : msg->headers) { - if (!header.name.ptr) { - continue; - } - - std::string name = std::string(header.name.ptr, header.name.len); - std::string value = std::string(header.value.ptr, header.value.len); - if (std::string{"Content-Type"} == name && std::string{"application/json"} == value) { - has_json_header = true; - break; - } - } - if (std::string{msg->method.ptr, msg->method.len} == std::string{"DELETE"}) { - std::string headers; - std::string response = "Patch success"; - if (!has_json_header) { - headers = "Content-Type: text/html\r\n"; - response = "Delete success"; - } else { - headers = "Content-Type: application/json\r\n"; - response = std::string{msg->body.ptr, msg->body.len}; - } - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); - } else { - std::string errorMessage{"Method Not Allowed"}; - SendError(conn, 405, errorMessage); - } -} - -void HttpServer::OnRequestDeleteNotAllowed(mg_connection* conn, mg_http_message* msg) { - if (std::string{msg->method.ptr, msg->method.len} == std::string{"DELETE"}) { - std::string errorMessage{"Method Not Allowed"}; - SendError(conn, 405, errorMessage); - } else { - std::string headers = "Content-Type: text/html\r\n"; - std::string response = "Delete success"; - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); - } -} - -void HttpServer::OnRequestPut(mg_connection* conn, mg_http_message* msg) { - if (std::string{msg->method.ptr, msg->method.len} == std::string{"PUT"}) { - char x[100]; - char y[100]; - mg_http_get_var(&(msg->body), "x", x, sizeof(x)); - mg_http_get_var(&(msg->body), "y", y, sizeof(y)); - std::string x_string = std::string{x}; - std::string y_string = std::string{y}; - std::string headers = "Content-Type: application/json\r\n"; - std::string response; - if (y_string.empty()) { - response = std::string{ - "{\n" - " \"x\": " + - x_string + - "\n" - "}"}; - } else { - response = std::string{ - "{\n" - " \"x\": " + - x_string + - ",\n" - " \"y\": " + - y_string + - ",\n" - " \"sum\": " + - std::to_string(atoi(x) + atoi(y)) + - "\n" - "}"}; - } - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); - } else { - std::string errorMessage{"Method Not Allowed"}; - SendError(conn, 405, errorMessage); - } -} - -void HttpServer::OnRequestPostReflect(mg_connection* conn, mg_http_message* msg) { - if (std::string{msg->method.ptr, msg->method.len} != std::string{"POST"}) { - std::string errorMessage{"Method Not Allowed"}; - SendError(conn, 405, errorMessage); - } - - std::string response = std::string{msg->body.ptr, msg->body.len}; - std::string headers; - for (mg_http_header& header : msg->headers) { - if (!header.name.ptr) { - continue; - } - - std::string name{header.name.ptr, header.name.len}; - if (std::string{"Host"} != name && std::string{"Accept"} != name) { - if (header.value.ptr) { - headers.append(name + ": " + std::string(header.value.ptr, header.value.len) + "\r\n"); - } - } - } - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); -} - -void HttpServer::OnRequestPutNotAllowed(mg_connection* conn, mg_http_message* msg) { - if (std::string{msg->method.ptr, msg->method.len} == std::string{"PUT"}) { - std::string errorMessage{"Method Not Allowed"}; - SendError(conn, 405, errorMessage); - } else { - std::string headers = "Content-Type: text/html\r\n"; - std::string response = "Delete success"; - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); - } -} - -void HttpServer::OnRequestPatch(mg_connection* conn, mg_http_message* msg) { - if (std::string{msg->method.ptr, msg->method.len} == std::string{"PATCH"}) { - char x[100]; - char y[100]; - mg_http_get_var(&(msg->body), "x", x, sizeof(x)); - mg_http_get_var(&(msg->body), "y", y, sizeof(y)); - std::string x_string = std::string{x}; - std::string y_string = std::string{y}; - std::string headers = "Content-Type: application/json\r\n"; - std::string response; - if (y_string.empty()) { - response = std::string{ - "{\n" - " \"x\": " + - x_string + - "\n" - "}"}; - } else { - response = std::string{ - "{\n" - " \"x\": " + - x_string + - ",\n" - " \"y\": " + - y_string + - ",\n" - " \"sum\": " + - std::to_string(atoi(x) + atoi(y)) + - "\n" - "}"}; - } - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); - } else { - std::string errorMessage{"Method Not Allowed"}; - SendError(conn, 405, errorMessage); - } -} - -void HttpServer::OnRequestPatchNotAllowed(mg_connection* conn, mg_http_message* msg) { - if (std::string{msg->method.ptr, msg->method.len} == std::string{"PATCH"}) { - std::string errorMessage{"Method Not Allowed"}; - SendError(conn, 405, errorMessage); - } else { - std::string headers = "Content-Type: text/html\r\n"; - std::string response = "Delete success"; - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); - } -} - -void HttpServer::OnRequestDownloadGzip(mg_connection* conn, mg_http_message* msg) { - if (std::string{msg->method.ptr, msg->method.len} == std::string{"DOWNLOAD"}) { - std::string errorMessage{"Method Not Allowed"}; - SendError(conn, 405, errorMessage); - } else { - std::string encoding; - std::string range; - std::vector<std::pair<int64_t, int64_t>> ranges; - - for (mg_http_header& header : msg->headers) { - if (!header.name.ptr) { - continue; - } - - std::string name = std::string(header.name.ptr, header.name.len); - if (std::string{"Accept-Encoding"} == name) { - encoding = std::string(header.value.ptr, header.value.len); - } else if (std::string{"Range"} == name) { - range = std::string(header.value.ptr, header.value.len); - } - } - if (encoding.find("gzip") == std::string::npos) { - std::string errorMessage{"Invalid encoding: " + encoding}; - SendError(conn, 405, errorMessage); - return; - } - if (!range.empty()) { - std::string::size_type eq_pos = range.find('='); - if (eq_pos == std::string::npos) { - std::string errorMessage{"Invalid range header: " + range}; - SendError(conn, 405, errorMessage); - return; - } - - int64_t current_start_index = eq_pos + 1; - int64_t current_end_index; - std::string::size_type range_len = range.length(); - std::string::size_type com_pos; - std::string::size_type sep_pos; - bool more_ranges_exists; - - do { - com_pos = range.find(',', current_start_index); - if (com_pos < range_len) { - current_end_index = com_pos - 1; - } else { - current_end_index = range_len - 1; - } - - std::pair<int64_t, int64_t> current_range{0, -1}; - - sep_pos = range.find('-', current_start_index); - if (sep_pos == std::string::npos) { - std::string errorMessage{"Invalid range format " + range.substr(current_start_index, current_end_index)}; - SendError(conn, 405, errorMessage); - return; - } - if (sep_pos == eq_pos + 1) { - std::string errorMessage{"Suffix ranage not supported: " + range.substr(current_start_index, current_end_index)}; - SendError(conn, 405, errorMessage); - return; - } - - current_range.first = std::strtoll(range.substr(current_start_index, sep_pos - 1).c_str(), nullptr, 10); - if (current_range.first == LLONG_MAX || current_range.first == LLONG_MIN) { - std::string errorMessage{"Start range is invalid number: " + range.substr(current_start_index, current_end_index)}; - SendError(conn, 405, errorMessage); - return; - } - - std::string er_str = range.substr(sep_pos + 1, current_end_index); - if (!er_str.empty()) { - current_range.second = std::strtoll(er_str.c_str(), nullptr, 10); - if (current_range.second == 0 || current_range.second == LLONG_MAX || current_range.second == LLONG_MIN) { - std::string errorMessage{"End range is invalid number: " + range.substr(current_start_index, current_end_index)}; - SendError(conn, 405, errorMessage); - return; - } - } - - ranges.push_back(current_range); - - if (current_end_index >= static_cast<int64_t>(range.length() - 1)) { - more_ranges_exists = false; - } else { - // Multiple ranges are separated by ', ' - more_ranges_exists = true; - current_start_index = current_end_index + 3; - } - } while (more_ranges_exists); - } - - std::string response = "Download!"; - int status_code = 200; - std::string headers; - - if (!ranges.empty()) { - // Create response parts - std::vector<std::string> responses; - for (std::pair<int64_t, int64_t> local_range : ranges) { - if (local_range.first >= 0) { - if (local_range.first >= (int64_t) response.length()) { - responses.push_back(""); - } else if (local_range.second == -1 || local_range.second >= (int64_t) response.length()) { - responses.push_back(response.substr(local_range.first)); - } else { - responses.push_back(response.substr(local_range.first, local_range.second - local_range.first + 1)); - } - } - } - - if (responses.size() > 1) { - // Create mime multipart response - std::string boundary = "3d6b6a416f9b5"; - status_code = 206; - response.clear(); - - for (size_t i{0}; i < responses.size(); ++i) { - response += "--" + boundary + "\n"; - response += "Content-Range: bytes " + std::to_string(ranges.at(i).first) + "-"; - if (ranges.at(i).second > 0) { - response += std::to_string(ranges.at(i).second); - } else { - response += std::to_string(responses.at(i).length()); - } - response += "/" + std::to_string(responses.at(i).length()) + "\n\n"; - response += responses.at(i) + "\n"; - } - response += "--" + boundary + "--"; - } else { - if (ranges.at(0).second == -1 || ranges.at(0).second >= (int64_t) response.length()) { - status_code = ranges.at(0).first > 0 ? 206 : 200; - } else { - status_code = 206; - } - response = responses.at(0); - - if (status_code == 206) { - headers = "Content-Range: bytes " + std::to_string(ranges.at(0).first) + "-"; - if (ranges.at(0).second > 0) { - headers += std::to_string(ranges.at(0).second); - } else { - headers += std::to_string(response.length()); - } - headers += "/" + std::to_string(response.length()); - } - } - } - if (!headers.empty()) { - headers += "\r\n"; - } - - mg_http_reply(conn, status_code, headers.c_str(), response.c_str()); - } -} - -void HttpServer::OnRequestCheckAcceptEncoding(mg_connection* conn, mg_http_message* msg) { - std::string response; - for (mg_http_header& header : msg->headers) { - if (!header.name.ptr) { - continue; - } - std::string name = std::string(header.name.ptr, header.name.len); - if (std::string{"Accept-Encoding"} == name) { - response = std::string(header.value.ptr, header.value.len); - } - } - std::string headers = "Content-Type: text/html\r\n"; - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); -} - -void HttpServer::OnRequestCheckExpect100Continue(mg_connection* conn, mg_http_message* msg) { - std::string response; - for (mg_http_header& header : msg->headers) { - if (!header.name.ptr) { - continue; - } - std::string name = std::string(header.name.ptr, header.name.len); - if (std::string{"Expect"} == name) { - response = std::string(header.value.ptr, header.value.len); - } - } - std::string headers = "Content-Type: text/html\r\n"; - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); -} - -void HttpServer::OnRequestGetDownloadFileLength(mg_connection* conn, mg_http_message* msg) { - auto method = std::string{msg->method.ptr, msg->method.len}; - if (method == std::string{"HEAD"}) { - mg_http_reply(conn, 405, nullptr, ""); - } else { - std::string response("this is a file content."); - std::string headers = "Content-Type: text/plain\r\n"; - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); - } -} - -void HttpServer::OnRequest(mg_connection* conn, mg_http_message* msg) { - std::string uri = std::string(msg->uri.ptr, msg->uri.len); - if (uri == "/") { - OnRequestRoot(conn, msg); - } else if (uri == "/hello.html") { - OnRequestHello(conn, msg); - } else if (uri == "/timeout.html") { - OnRequestTimeout(conn, msg); - } else if (uri == "/long_timeout.html") { - OnRequestLongTimeout(conn, msg); - } else if (uri == "/low_speed_timeout.html") { - timer_args.emplace_back(std::make_unique<TimerArg>(&mgr, conn, mg_timer{})); - OnRequestLowSpeedTimeout(conn, msg, timer_args.back().get()); - } else if (uri == "/low_speed.html") { - OnRequestLowSpeed(conn, msg, &mgr); - } else if (uri == "/low_speed_bytes.html") { - timer_args.emplace_back(std::make_unique<TimerArg>(&mgr, conn, mg_timer{})); - OnRequestLowSpeedBytes(conn, msg, timer_args.back().get()); - } else if (uri == "/basic_cookies.html") { - OnRequestBasicCookies(conn, msg); - } else if (uri == "/empty_cookies.html") { - OnRequestEmptyCookies(conn, msg); - } else if (uri == "/cookies_reflect.html") { - OnRequestCookiesReflect(conn, msg); - } else if (uri == "/redirection_with_changing_cookies.html") { - OnRequestRedirectionWithChangingCookies(conn, msg); - } else if (uri == "/basic_auth.html") { - OnRequestBasicAuth(conn, msg); - } else if (uri == "/bearer_token.html") { - OnRequestBearerAuth(conn, msg); - } else if (uri == "/digest_auth.html") { - OnRequestHeaderReflect(conn, msg); - } else if (uri == "/basic.json") { - OnRequestBasicJson(conn, msg); - } else if (uri == "/header_reflect.html") { - OnRequestHeaderReflect(conn, msg); - } else if (uri == "/temporary_redirect.html") { - OnRequestTempRedirect(conn, msg); - } else if (uri == "/permanent_redirect.html") { - OnRequestPermRedirect(conn, msg); - } else if (uri == "/resolve_permanent_redirect.html") { - OnRequestResolvePermRedirect(conn, msg); - } else if (uri == "/two_redirects.html") { - OnRequestTwoRedirects(conn, msg); - } else if (uri == "/url_post.html") { - OnRequestUrlPost(conn, msg); - } else if (uri == "/body_get.html") { - OnRequestBodyGet(conn, msg); - } else if (uri == "/json_post.html") { - OnRequestJsonPost(conn, msg); - } else if (uri == "/form_post.html") { - OnRequestFormPost(conn, msg); - } else if (uri == "/post_reflect.html") { - OnRequestPostReflect(conn, msg); - } else if (uri == "/delete.html") { - OnRequestDelete(conn, msg); - } else if (uri == "/delete_unallowed.html") { - OnRequestDeleteNotAllowed(conn, msg); - } else if (uri == "/put.html") { - OnRequestPut(conn, msg); - } else if (uri == "/put_unallowed.html") { - OnRequestPutNotAllowed(conn, msg); - } else if (uri == "/patch.html") { - OnRequestPatch(conn, msg); - } else if (uri == "/patch_unallowed.html") { - OnRequestPatchNotAllowed(conn, msg); - } else if (uri == "/download_gzip.html") { - OnRequestDownloadGzip(conn, msg); - } else if (uri == "/local_port.html") { - OnRequestLocalPort(conn, msg); - } else if (uri == "/check_accept_encoding.html") { - OnRequestCheckAcceptEncoding(conn, msg); - } else if (uri == "/check_expect_100_continue.html") { - OnRequestCheckExpect100Continue(conn, msg); - } else if (uri == "/get_download_file_length.html") { - OnRequestGetDownloadFileLength(conn, msg); - } else { - OnRequestNotFound(conn, msg); - } -} - -void HttpServer::OnRequestLocalPort(mg_connection* conn, mg_http_message* /*msg*/) { - // send source port number as response for checking SetLocalPort/SetLocalPortRange - std::string headers = "Content-Type: text/plain\r\n"; - // Convert from big endian to little endian - std::string response = std::to_string(AbstractServer::GetRemotePort(conn)); - mg_http_reply(conn, 200, headers.c_str(), response.c_str()); -} - -} // namespace cpr |