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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
#include "EventManager.hpp"
#include <event2/buffer.h>
#include <event2/event.h>
#include <event2/http.h>
#include <event2/keyvalq_struct.h>
#include <event2/listener.h>
#include <event2/util.h>
#include <signal.h>
extern "C" {
static void GenerateInternalErrorPage(struct evhttp_request * const req,
std::string text)
{
evhttp_add_header(evhttp_request_get_output_headers(req), "Content-Type",
"text/html");
struct evbuffer *const output = evbuffer_new();
if (output != nullptr) {
evbuffer_add(output, text.c_str(), text.size());
evhttp_send_reply(req, 500, "Internal Error", output);
evbuffer_free(output);
}
}
static void EvGenericInterceptor(struct evhttp_request *const req,
void *ev_c_callback) {
if (ev_c_callback != nullptr) {
struct ev_callback *const evcb = (struct ev_callback *)ev_c_callback;
evcb->cb(req, evcb->ud);
} else {
GenerateInternalErrorPage(req, "EvGenericInterceptor: ev_c_callback == nullptr");
}
}
static void EvContentManagerInterceptor(struct evhttp_request * const req,
void * ev_c_callback) {
if (ev_c_callback != nullptr) {
Content * const cntnt = (Content *)ev_c_callback;
std::string out;
if (cntnt->Render(out) == false) {
std::string text;
text = "ContentModule(\"" + cntnt->GetBasePath() + "\")->Render() failed.\n";
GenerateInternalErrorPage(req, text);
} else {
evhttp_add_header(evhttp_request_get_output_headers(req), "Content-Type",
"text/html");
struct evbuffer *const output = evbuffer_new();
if (output != nullptr) {
evbuffer_add(output, out.c_str(), out.size());
evhttp_send_reply(req, 200, "OK", output);
evbuffer_free(output);
}
}
} else {
GenerateInternalErrorPage(req, "EvContentManagerInterceptor: ev_c_callback == nullptr");
}
}
static void do_term(int sig, short events, void *arg) {
(void)events;
struct event_base *base = (struct event_base *)arg;
event_base_loopbreak(base);
fprintf(stderr, "Got %i, Terminating\n", sig);
}
}
EventManager::~EventManager() {
if (m_EvConfig != nullptr)
event_config_free(m_EvConfig);
if (m_EvHttp != nullptr)
evhttp_free(m_EvHttp);
if (m_EvTermEvent != nullptr)
event_free(m_EvTermEvent);
if (m_EvBase != nullptr)
event_base_free(m_EvBase);
}
bool EventManager::Init(std::string host, uint16_t port) {
struct event_config *cfg = nullptr;
if (signal(SIGPIPE, SIG_IGN) == SIG_ERR) {
return false;
}
event_enable_debug_logging(EVENT_DBG_ALL);
cfg = event_config_new();
event_config_set_flag(cfg, EVENT_BASE_FLAG_STARTUP_IOCP);
m_EvBase = event_base_new_with_config(cfg);
if (m_EvBase == nullptr) {
fprintf(stderr, "Couldn't create an event_base: exiting\n");
return false;
}
event_config_free(cfg);
cfg = nullptr;
/* Create a new evhttp object to handle requests. */
m_EvHttp = evhttp_new(m_EvBase);
if (m_EvHttp == nullptr) {
fprintf(stderr, "couldn't create evhttp. Exiting.\n");
return false;
}
for (auto & uc : m_UrlCallbacks) {
struct ev_callback * const evcb = &std::get<1>(uc);
if (evhttp_set_cb(m_EvHttp, std::get<0>(uc).c_str(), EvGenericInterceptor,
evcb) != 0) {
return false;
}
}
for (auto & cm : m_ContentModules) {
std::shared_ptr<Content> content = cm.second;
if (evhttp_set_cb(m_EvHttp, cm.first.c_str(), EvContentManagerInterceptor,
content.get()) != 0) {
return false;
}
}
evhttp_set_gencb(m_EvHttp, EvGenericInterceptor, &m_DefaultCallback);
m_EvSocket = evhttp_bind_socket_with_handle(m_EvHttp, host.c_str(), port);
if (m_EvSocket == nullptr) {
fprintf(stderr, "couldn't bind to %s:%d. Exiting.\n", host.c_str(), port);
return false;
}
m_EvTermEvent = evsignal_new(m_EvBase, SIGINT, do_term, m_EvBase);
if (m_EvTermEvent == nullptr) {
return false;
}
if (event_add(m_EvTermEvent, NULL) != 0) {
return false;
}
event_base_dispatch(m_EvBase);
return true;
}
void EventManager::SetDefaultCallback(EvFunction fn, EvUserData dat) {
m_DefaultCallback.cb = fn;
m_DefaultCallback.ud = dat;
}
void EventManager::AddCallback(std::string url, EvFunction fn, EvUserData dat) {
m_UrlCallbacks.push_back(EvUrlCallback(url, {fn, dat}));
}
void EventManager::AddContentManager(ContentManager const & cmgr)
{
ContentModules new_mods = m_ContentModules;
new_mods.insert(cmgr.GetAllModules().begin(), cmgr.GetAllModules().end());
m_ContentModules = new_mods;
}
|