1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
#include "Client.h"
#include <AK/FileSystemPath.h>
#include <AK/StringBuilder.h>
#include <LibCore/File.h>
#include <LibCore/HttpRequest.h>
#include <stdio.h>
#include <time.h>
namespace WebServer {
Client::Client(NonnullRefPtr<Core::TCPSocket> socket, Core::Object* parent)
: Core::Object(parent)
, m_socket(socket)
{
}
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(move(raw_request));
die();
};
}
void Client::handle_request(ByteBuffer raw_request)
{
auto request_or_error = Core::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() != Core::HttpRequest::Method::GET) {
send_error_response(403, "Forbidden, bro!", request);
return;
}
FileSystemPath canonical_path(request.resource());
dbg() << "Requested canonical path: '" << canonical_path.string() << "'";
StringBuilder path_builder;
path_builder.append("/www/");
path_builder.append(canonical_path.string());
if (Core::File::is_directory(path_builder.to_string()))
path_builder.append("/index.html");
auto file = Core::File::construct(path_builder.to_string());
if (!file->open(Core::File::ReadOnly)) {
send_error_response(404, "Not found, bro!", request);
return;
}
StringBuilder builder;
builder.append("HTTP/1.0 200 OK\r\n");
builder.append("Server: WebServer (SerenityOS)\r\n");
builder.append("Content-Type: text/html\r\n");
builder.append("\r\n");
m_socket->write(builder.to_string());
m_socket->write(file->read_all());
log_response(200, request);
}
void Client::send_error_response(unsigned code, const StringView& message, const Core::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 Core::HttpRequest& request)
{
printf("%lld :: %03u :: %s %s\n", time(nullptr), code, request.method_name().characters(), request.resource().characters());
}
}
|
