Emulator-Archive/ppsspp
1
0
Fork 0
mirror of https://github.com/hrydgard/ppsspp.git synced 2024-12-01 01:11:46 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

http: Make it possible to override 404 handling.

Browse Source
This commit is contained in:
Unknown W. Brackets 2016-05-25 22:28:28 -07:00
parent 1cc49f3be4
commit 8fc610e2d1
2 changed files with 28 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
ext/native/net/http_server.cpp
View File

@ -92,12 +92,17 @@ void Request::Close() {
Server::Server(threading::Executor *executor)
: port_(0), executor_(executor) {
RegisterHandler("/", std::bind(&Server::HandleListing, this, placeholder::_1));
SetFallbackHandler(std::bind(&Server::Handle404, this, placeholder::_1));
}
void Server::RegisterHandler(const char *url_path, UrlHandlerFunc handler) {
handlers_[std::string(url_path)] = handler;
}
void Server::SetFallbackHandler(UrlHandlerFunc handler) {
fallback_ = handler;
}
bool Server::Run(int port) {
ILOG("HTTP server started on port %i", port);
port_ = port;
@ -153,28 +158,32 @@ void Server::HandleConnection(int conn_fd) {
}
void Server::HandleRequest(const Request &request) {
HandleRequestDefault(request);
HandleRequestDefault(request);
}
void Server::HandleRequestDefault(const Request &request) {
// First, look through all handlers. If we got one, use it.
for (auto iter = handlers_.begin(); iter != handlers_.end(); ++iter) {
if (iter->first == request.resource()) {
(iter->second)(request);
return;
}
}
ILOG("No handler for '%s', falling back to 404.", request.resource());
const char *payload = "<html><body>404 not found</body></html>\r\n";
request.WriteHttpResponseHeader(404, (int)strlen(payload));
request.Out()->Push(payload);
// First, look through all handlers. If we got one, use it.
auto handler = handlers_.find(request.resource());
if (handler != handlers_.end()) {
(handler->second)(request);
} else {
// Let's hit the 404 handler instead.
fallback_(request);
}
}
void Server::Handle404(const Request &request) {
ILOG("No handler for '%s', falling back to 404.", request.resource());
const char *payload = "<html><body>404 not found</body></html>\r\n";
request.WriteHttpResponseHeader(404, (int)strlen(payload));
request.Out()->Push(payload);
}
void Server::HandleListing(const Request &request) {
request.WriteHttpResponseHeader(200, -1);
for (auto iter = handlers_.begin(); iter != handlers_.end(); ++iter) {
request.Out()->Printf("%s\n", iter->first.c_str());
}
request.WriteHttpResponseHeader(200, -1);
for (auto iter = handlers_.begin(); iter != handlers_.end(); ++iter) {
request.Out()->Printf("%s\n", iter->first.c_str());
}
}
} // namespace http

3
ext/native/net/http_server.h
View File

@ -68,6 +68,7 @@ class Server {
bool Run(int port);
void RegisterHandler(const char *url_path, UrlHandlerFunc handler);
void SetFallbackHandler(UrlHandlerFunc handler);
// If you want to customize things at a lower level than just a simple path handler,
// then inherit and override this. Implementations should forward to HandleRequestDefault
@ -84,10 +85,12 @@ class Server {
// Neat built-in handlers that are tied to the server.
void HandleListing(const Request &request);
void Handle404(const Request &request);
int port_;
UrlHandlerMap handlers_;
UrlHandlerFunc fallback_;
threading::Executor *executor_;
};