mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-11-22 23:50:19 +00:00
268 lines
8.5 KiB
C++
268 lines
8.5 KiB
C++
/*
|
|
* Copyright (c) 2020, Andreas Kling <kling@serenityos.org>
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright notice, this
|
|
* list of conditions and the following disclaimer.
|
|
*
|
|
* 2. Redistributions in binary form must reproduce the above copyright notice,
|
|
* this list of conditions and the following disclaimer in the documentation
|
|
* and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
|
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
|
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
|
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "Client.h"
|
|
#include <AK/Base64.h>
|
|
#include <AK/LexicalPath.h>
|
|
#include <AK/MappedFile.h>
|
|
#include <AK/StringBuilder.h>
|
|
#include <AK/URLParser.h>
|
|
#include <LibCore/DateTime.h>
|
|
#include <LibCore/DirIterator.h>
|
|
#include <LibCore/File.h>
|
|
#include <LibCore/MimeData.h>
|
|
#include <LibHTTP/HttpRequest.h>
|
|
#include <stdio.h>
|
|
#include <sys/stat.h>
|
|
#include <time.h>
|
|
#include <unistd.h>
|
|
|
|
namespace WebServer {
|
|
|
|
Client::Client(NonnullRefPtr<Core::TCPSocket> socket, const String& root, Core::Object* parent)
|
|
: Core::Object(parent)
|
|
, m_socket(socket)
|
|
, m_root_path(root)
|
|
{
|
|
}
|
|
|
|
void Client::die()
|
|
{
|
|
remove_from_parent();
|
|
}
|
|
|
|
void Client::start()
|
|
{
|
|
m_socket->on_ready_to_read = [this] {
|
|
auto raw_request = m_socket->read_all();
|
|
if (raw_request.is_null()) {
|
|
die();
|
|
return;
|
|
}
|
|
|
|
dbg() << "Got raw request: '" << String::copy(raw_request) << "'";
|
|
|
|
handle_request(raw_request.bytes());
|
|
die();
|
|
};
|
|
}
|
|
|
|
void Client::handle_request(ReadonlyBytes raw_request)
|
|
{
|
|
auto request_or_error = HTTP::HttpRequest::from_raw_request(raw_request);
|
|
if (!request_or_error.has_value())
|
|
return;
|
|
auto& request = request_or_error.value();
|
|
|
|
dbg() << "Got HTTP request: " << request.method_name() << " " << request.resource();
|
|
for (auto& header : request.headers()) {
|
|
dbg() << " " << header.name << " => " << header.value;
|
|
}
|
|
|
|
if (request.method() != HTTP::HttpRequest::Method::GET) {
|
|
send_error_response(403, "Forbidden!", request);
|
|
return;
|
|
}
|
|
|
|
auto requested_path = LexicalPath::canonicalized_path(request.resource());
|
|
dbg() << "Canonical requested path: '" << requested_path << "'";
|
|
|
|
StringBuilder path_builder;
|
|
path_builder.append(m_root_path);
|
|
path_builder.append('/');
|
|
path_builder.append(requested_path);
|
|
auto real_path = path_builder.to_string();
|
|
|
|
if (Core::File::is_directory(real_path)) {
|
|
|
|
if (!request.resource().ends_with("/")) {
|
|
StringBuilder red;
|
|
|
|
red.append(requested_path);
|
|
red.append("/");
|
|
|
|
send_redirect(red.to_string(), request);
|
|
return;
|
|
}
|
|
|
|
StringBuilder index_html_path_builder;
|
|
index_html_path_builder.append(real_path);
|
|
index_html_path_builder.append("/index.html");
|
|
auto index_html_path = index_html_path_builder.to_string();
|
|
if (!Core::File::exists(index_html_path)) {
|
|
handle_directory_listing(requested_path, real_path, request);
|
|
return;
|
|
}
|
|
real_path = index_html_path;
|
|
}
|
|
|
|
auto file = Core::File::construct(real_path);
|
|
if (!file->open(Core::File::ReadOnly)) {
|
|
send_error_response(404, "Not found!", request);
|
|
return;
|
|
}
|
|
|
|
send_response(file->read_all(), request, Core::guess_mime_type_based_on_filename(real_path));
|
|
}
|
|
|
|
void Client::send_response(StringView response, const HTTP::HttpRequest& request, const String& content_type)
|
|
{
|
|
StringBuilder builder;
|
|
builder.append("HTTP/1.0 200 OK\r\n");
|
|
builder.append("Server: WebServer (SerenityOS)\r\n");
|
|
builder.append("Content-Type: ");
|
|
builder.append(content_type);
|
|
builder.append("\r\n");
|
|
builder.append("\r\n");
|
|
|
|
m_socket->write(builder.to_string());
|
|
m_socket->write(response);
|
|
|
|
log_response(200, request);
|
|
}
|
|
|
|
void Client::send_redirect(StringView redirect_path, const HTTP::HttpRequest& request)
|
|
{
|
|
StringBuilder builder;
|
|
builder.append("HTTP/1.0 301 Moved Permanently\r\n");
|
|
builder.append("Location: ");
|
|
builder.append(redirect_path);
|
|
builder.append("\r\n");
|
|
builder.append("\r\n");
|
|
|
|
m_socket->write(builder.to_string());
|
|
|
|
log_response(301, request);
|
|
}
|
|
|
|
static String folder_image_data()
|
|
{
|
|
static String cache;
|
|
if (cache.is_empty()) {
|
|
MappedFile image("/res/icons/16x16/filetype-folder.png");
|
|
cache = encode_base64({ image.data(), image.size() });
|
|
}
|
|
return cache;
|
|
}
|
|
|
|
static String file_image_data()
|
|
{
|
|
static String cache;
|
|
if (cache.is_empty()) {
|
|
MappedFile image("/res/icons/16x16/filetype-unknown.png");
|
|
cache = encode_base64({ image.data(), image.size() });
|
|
}
|
|
return cache;
|
|
}
|
|
|
|
void Client::handle_directory_listing(const String& requested_path, const String& real_path, const HTTP::HttpRequest& request)
|
|
{
|
|
StringBuilder builder;
|
|
|
|
builder.append("<!DOCTYPE html>\n");
|
|
builder.append("<html>\n");
|
|
builder.append("<head><title>Index of ");
|
|
builder.append(escape_html_entities(requested_path));
|
|
builder.append("</title><style>\n");
|
|
builder.append(".folder { width: 16px; height: 16px; background-image: url('data:image/png;base64,");
|
|
builder.append(folder_image_data());
|
|
builder.append("'); }\n");
|
|
builder.append(".file { width: 16px; height: 16px; background-image: url('data:image/png;base64,");
|
|
builder.append(file_image_data());
|
|
builder.append("'); }\n");
|
|
builder.append("</style></head><body>\n");
|
|
builder.append("<h1>Index of ");
|
|
builder.append(escape_html_entities(requested_path));
|
|
builder.append("</h1>\n");
|
|
builder.append("<hr>\n");
|
|
builder.append("<code><table>\n");
|
|
|
|
Core::DirIterator dt(real_path);
|
|
while (dt.has_next()) {
|
|
auto name = dt.next_path();
|
|
|
|
StringBuilder path_builder;
|
|
path_builder.append(real_path);
|
|
path_builder.append('/');
|
|
path_builder.append(name);
|
|
struct stat st;
|
|
memset(&st, 0, sizeof(st));
|
|
int rc = stat(path_builder.to_string().characters(), &st);
|
|
if (rc < 0) {
|
|
perror("stat");
|
|
}
|
|
|
|
bool is_directory = S_ISDIR(st.st_mode) || name.is_one_of(".", "..");
|
|
|
|
builder.append("<tr>");
|
|
builder.appendf("<td><div class=\"%s\"></div></td>", is_directory ? "folder" : "file");
|
|
builder.append("<td><a href=\"");
|
|
builder.append(urlencode(name));
|
|
builder.append("\">");
|
|
builder.append(escape_html_entities(name));
|
|
builder.append("</a></td><td> </td>");
|
|
|
|
builder.appendf("<td>%10d</td><td> </td>", st.st_size);
|
|
builder.append("<td>");
|
|
builder.append(Core::DateTime::from_timestamp(st.st_mtime).to_string());
|
|
builder.append("</td>");
|
|
builder.append("</tr>\n");
|
|
}
|
|
|
|
builder.append("</table></code>\n");
|
|
builder.append("<hr>\n");
|
|
builder.append("<i>Generated by WebServer (SerenityOS)</i>\n");
|
|
builder.append("</body>\n");
|
|
builder.append("</html>\n");
|
|
|
|
send_response(builder.to_string(), request, "text/html");
|
|
}
|
|
|
|
void Client::send_error_response(unsigned code, const StringView& message, const HTTP::HttpRequest& request)
|
|
{
|
|
StringBuilder builder;
|
|
builder.appendf("HTTP/1.0 %u ", code);
|
|
builder.append(message);
|
|
builder.append("\r\n\r\n");
|
|
builder.append("<!DOCTYPE html><html><body><h1>");
|
|
builder.appendf("%u ", code);
|
|
builder.append(message);
|
|
builder.append("</h1></body></html>");
|
|
m_socket->write(builder.to_string());
|
|
|
|
log_response(code, request);
|
|
}
|
|
|
|
void Client::log_response(unsigned code, const HTTP::HttpRequest& request)
|
|
{
|
|
printf("%s :: %03u :: %s %s\n",
|
|
Core::DateTime::now().to_string().characters(),
|
|
code,
|
|
request.method_name().characters(),
|
|
request.resource().characters());
|
|
}
|
|
|
|
}
|