File: | lwan.c |
Warning: | line 311, column 17 Potential memory leak |
Press '?' to see keyboard shortcuts
Keyboard shortcuts:
1 | /* | |||
2 | * lwan - simple web server | |||
3 | * Copyright (c) 2012, 2013 Leandro A. F. Pereira <leandro@hardinfo.org> | |||
4 | * | |||
5 | * This program is free software; you can redistribute it and/or | |||
6 | * modify it under the terms of the GNU General Public License | |||
7 | * as published by the Free Software Foundation; either version 2 | |||
8 | * of the License, or any later version. | |||
9 | * | |||
10 | * This program is distributed in the hope that it will be useful, | |||
11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |||
12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |||
13 | * GNU General Public License for more details. | |||
14 | * | |||
15 | * You should have received a copy of the GNU General Public License | |||
16 | * along with this program; if not, write to the Free Software | |||
17 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, | |||
18 | * USA. | |||
19 | */ | |||
20 | ||||
21 | #define _GNU_SOURCE | |||
22 | #include <assert.h> | |||
23 | #include <ctype.h> | |||
24 | #include <dlfcn.h> | |||
25 | #include <errno(*__errno_location ()).h> | |||
26 | #include <fcntl.h> | |||
27 | #include <libproc.h> | |||
28 | #include <limits.h> | |||
29 | #include <signal.h> | |||
30 | #include <stdlib.h> | |||
31 | #include <string.h> | |||
32 | #include <sys/resource.h> | |||
33 | #include <sys/socket.h> | |||
34 | #include <sys/types.h> | |||
35 | #include <unistd.h> | |||
36 | ||||
37 | #include "lwan-private.h" | |||
38 | ||||
39 | #include "lwan-config.h" | |||
40 | #include "lwan-http-authorize.h" | |||
41 | ||||
42 | #if defined(HAVE_LUA) | |||
43 | #include "lwan-lua.h" | |||
44 | #endif | |||
45 | ||||
46 | /* See detect_fastest_monotonic_clock() */ | |||
47 | clockid_t monotonic_clock_id = CLOCK_MONOTONIC1; | |||
48 | ||||
49 | static const struct lwan_config default_config = { | |||
50 | .listener = "localhost:8080", | |||
51 | .keep_alive_timeout = 15, | |||
52 | .quiet = false0, | |||
53 | .reuse_port = false0, | |||
54 | .proxy_protocol = false0, | |||
55 | .allow_cors = false0, | |||
56 | .expires = 1 * ONE_WEEK(((60 * 60) * 24) * 7), | |||
57 | .n_threads = 0, | |||
58 | .max_post_data_size = 10 * DEFAULT_BUFFER_SIZE4096, | |||
59 | .allow_post_temp_file = false0, | |||
60 | }; | |||
61 | ||||
62 | LWAN_HANDLER(brew_coffee)static enum lwan_http_status lwan_handler_brew_coffee( struct lwan_request *, struct lwan_response *, void *); static const struct lwan_handler_info __attribute__((used, section("lwan_handler" ))) lwan_handler_info_brew_coffee = {.name = "brew_coffee", . handler = lwan_handler_brew_coffee}; static enum lwan_http_status lwan_handler_brew_coffee( struct lwan_request *request __attribute__ ((unused)), struct lwan_response *response __attribute__((unused )), void *data __attribute__((unused))) | |||
63 | { | |||
64 | /* Placeholder handler so that __start_lwan_handler and __stop_lwan_handler | |||
65 | * symbols will get defined. | |||
66 | */ | |||
67 | return HTTP_I_AM_A_TEAPOT; | |||
68 | } | |||
69 | ||||
70 | __attribute__((no_sanitize_address)) | |||
71 | static void *find_handler(const char *name) | |||
72 | { | |||
73 | const struct lwan_handler_info *handler; | |||
74 | ||||
75 | LWAN_SECTION_FOREACH(lwan_handler, handler)for (handler = ({ extern const typeof(*handler) __start_lwan_handler []; __start_lwan_handler; }); handler < ({ extern const typeof (*handler) __stop_lwan_handler[]; __stop_lwan_handler; }); (handler )++) { | |||
76 | if (streq(handler->name, name)) | |||
77 | return handler->handler; | |||
78 | } | |||
79 | ||||
80 | return NULL((void*)0); | |||
81 | } | |||
82 | ||||
83 | __attribute__((no_sanitize_address)) | |||
84 | static const struct lwan_module *find_module(const char *name) | |||
85 | { | |||
86 | const struct lwan_module_info *module; | |||
87 | ||||
88 | LWAN_SECTION_FOREACH(lwan_module, module)for (module = ({ extern const typeof(*module) __start_lwan_module []; __start_lwan_module; }); module < ({ extern const typeof (*module) __stop_lwan_module[]; __stop_lwan_module; }); (module )++) { | |||
89 | if (streq(module->name, name)) | |||
90 | return module->module; | |||
91 | } | |||
92 | ||||
93 | return NULL((void*)0); | |||
94 | } | |||
95 | ||||
96 | static void destroy_urlmap(void *data) | |||
97 | { | |||
98 | struct lwan_url_map *url_map = data; | |||
99 | ||||
100 | if (url_map->module) { | |||
101 | const struct lwan_module *module = url_map->module; | |||
102 | ||||
103 | if (module->destroy) | |||
104 | module->destroy(url_map->data); | |||
105 | } else if (url_map->data && url_map->flags & HANDLER_DATA_IS_HASH_TABLE) { | |||
106 | hash_free(url_map->data); | |||
107 | } | |||
108 | ||||
109 | free(url_map->authorization.realm); | |||
110 | free(url_map->authorization.password_file); | |||
111 | free((char *)url_map->prefix); | |||
112 | free(url_map); | |||
113 | } | |||
114 | ||||
115 | static struct lwan_url_map *add_url_map(struct lwan_trie *t, const char *prefix, | |||
116 | const struct lwan_url_map *map) | |||
117 | { | |||
118 | struct lwan_url_map *copy = malloc(sizeof(*copy)); | |||
119 | ||||
120 | if (!copy) | |||
121 | lwan_status_critical_perror("Could not copy URL map")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 121, __FUNCTION__, "Could not copy URL map"); | |||
122 | ||||
123 | memcpy(copy, map, sizeof(*copy)); | |||
124 | ||||
125 | copy->prefix = strdup(prefix ? prefix : copy->prefix); | |||
126 | if (!copy->prefix) | |||
127 | lwan_status_critical_perror("Could not copy URL prefix")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 127, __FUNCTION__, "Could not copy URL prefix"); | |||
128 | ||||
129 | copy->prefix_len = strlen(copy->prefix); | |||
130 | lwan_trie_add(t, copy->prefix, copy); | |||
131 | ||||
132 | return copy; | |||
133 | } | |||
134 | ||||
135 | static bool_Bool can_override_header(const char *name) | |||
136 | { | |||
137 | /* NOTE: Update lwan_prepare_response_header_full() in lwan-response.c | |||
138 | * if new headers are added here. */ | |||
139 | ||||
140 | if (!strcasecmp(name, "Date")) | |||
141 | return false0; | |||
142 | if (!strcasecmp(name, "Expires")) | |||
143 | return false0; | |||
144 | if (!strcasecmp(name, "WWW-Authenticate")) | |||
145 | return false0; | |||
146 | if (!strcasecmp(name, "Connection")) | |||
147 | return false0; | |||
148 | if (!strcasecmp(name, "Content-Type")) | |||
149 | return false0; | |||
150 | if (!strcasecmp(name, "Transfer-Encoding")) | |||
151 | return false0; | |||
152 | if (!strncasecmp(name, "Access-Control-Allow-", | |||
153 | sizeof("Access-Control-Allow-") - 1)) | |||
154 | return false0; | |||
155 | ||||
156 | return true1; | |||
157 | } | |||
158 | ||||
159 | static void build_response_headers(struct lwan *l, | |||
160 | const struct lwan_key_value *kv) | |||
161 | { | |||
162 | bool_Bool set_server = false0; | |||
163 | ||||
164 | assert(l)((void) sizeof ((l) ? 1 : 0), __extension__ ({ if (l) ; else __assert_fail ("l", "/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 164, __extension__ __PRETTY_FUNCTION__); })); | |||
165 | ||||
166 | lwan_strbuf_init(&l->headers); | |||
167 | ||||
168 | for (; kv && kv->key; kv++) { | |||
169 | if (!can_override_header(kv->key)) { | |||
170 | lwan_status_warning("Cannot override header '%s'", kv->key)lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 170, __FUNCTION__, "Cannot override header '%s'", kv->key ); | |||
171 | } else { | |||
172 | if (!strcasecmp(kv->key, "Server")) | |||
173 | set_server = true1; | |||
174 | ||||
175 | lwan_strbuf_append_printf(&l->headers, "\r\n%s: %s", kv->key, | |||
176 | kv->value); | |||
177 | } | |||
178 | } | |||
179 | ||||
180 | if (!set_server) | |||
181 | lwan_strbuf_append_strz(&l->headers, "\r\nServer: lwan"); | |||
182 | ||||
183 | lwan_strbuf_append_strz(&l->headers, "\r\n\r\n"); | |||
184 | } | |||
185 | ||||
186 | static void parse_global_headers(struct config *c, | |||
187 | struct lwan *lwan) | |||
188 | { | |||
189 | struct lwan_key_value_array hdrs; | |||
190 | const struct config_line *l; | |||
191 | struct lwan_key_value *kv; | |||
192 | ||||
193 | lwan_key_value_array_init(&hdrs); | |||
194 | ||||
195 | while ((l = config_read_line(c))) { | |||
196 | switch (l->type) { | |||
197 | case CONFIG_LINE_TYPE_SECTION: | |||
198 | config_error( | |||
199 | c, "No sections are supported under the 'headers' section"); | |||
200 | goto cleanup; | |||
201 | ||||
202 | case CONFIG_LINE_TYPE_LINE: | |||
203 | kv = lwan_key_value_array_append(&hdrs); | |||
204 | if (!kv) { | |||
205 | lwan_status_critical_perror(lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 206, __FUNCTION__, "Could not allocate memory for custom response header" ) | |||
206 | "Could not allocate memory for custom response header")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 206, __FUNCTION__, "Could not allocate memory for custom response header" ); | |||
207 | } | |||
208 | ||||
209 | kv->key = strdup(l->key); | |||
210 | if (!kv->key) { | |||
211 | lwan_status_critical_perror(lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 212, __FUNCTION__, "Could not allocate memory for custom response header" ) | |||
212 | "Could not allocate memory for custom response header")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 212, __FUNCTION__, "Could not allocate memory for custom response header" ); | |||
213 | } | |||
214 | ||||
215 | kv->value = strdup(l->value); | |||
216 | if (!kv->value) { | |||
217 | lwan_status_critical_perror(lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 218, __FUNCTION__, "Could not allocate memory for custom response header" ) | |||
218 | "Could not allocate memory for custom response header")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 218, __FUNCTION__, "Could not allocate memory for custom response header" ); | |||
219 | } | |||
220 | break; | |||
221 | ||||
222 | case CONFIG_LINE_TYPE_SECTION_END: | |||
223 | kv = lwan_key_value_array_append(&hdrs); | |||
224 | if (!kv) { | |||
225 | lwan_status_critical_perror(lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 226, __FUNCTION__, "Could not allocate memory for custom response header" ) | |||
226 | "Could not allocate memory for custom response header")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 226, __FUNCTION__, "Could not allocate memory for custom response header" ); | |||
227 | } | |||
228 | ||||
229 | kv->key = NULL((void*)0); | |||
230 | kv->value = NULL((void*)0); | |||
231 | ||||
232 | build_response_headers(lwan, lwan_key_value_array_get_array(&hdrs)); | |||
233 | goto cleanup; | |||
234 | } | |||
235 | } | |||
236 | ||||
237 | config_error(c, "EOF while looking for end of 'headers' section"); | |||
238 | ||||
239 | cleanup: | |||
240 | LWAN_ARRAY_FOREACH (&hdrs, kv)for (kv = (&hdrs)->base.base; kv < ((typeof(kv))(& hdrs)->base.base + (&hdrs)->base.elements); kv++) { | |||
241 | free(kv->key); | |||
242 | free(kv->value); | |||
243 | } | |||
244 | lwan_key_value_array_reset(&hdrs); | |||
245 | } | |||
246 | ||||
247 | static void parse_listener_prefix_authorization(struct config *c, | |||
248 | const struct config_line *l, | |||
249 | struct lwan_url_map *url_map) | |||
250 | { | |||
251 | if (!streq(l->value, "basic")) { | |||
252 | config_error(c, "Only basic authorization supported"); | |||
253 | return; | |||
254 | } | |||
255 | ||||
256 | memset(&url_map->authorization, 0, sizeof(url_map->authorization)); | |||
257 | ||||
258 | while ((l = config_read_line(c))) { | |||
259 | switch (l->type) { | |||
260 | case CONFIG_LINE_TYPE_LINE: | |||
261 | if (streq(l->key, "realm")) { | |||
262 | free(url_map->authorization.realm); | |||
263 | url_map->authorization.realm = strdup(l->value); | |||
264 | } else if (streq(l->key, "password_file")) { | |||
265 | free(url_map->authorization.password_file); | |||
266 | url_map->authorization.password_file = realpath(l->value, NULL((void*)0)); | |||
267 | if (!url_map->authorization.password_file) | |||
268 | config_error(c, "Could not determine full path for password file: %s", l->value); | |||
269 | } | |||
270 | break; | |||
271 | ||||
272 | case CONFIG_LINE_TYPE_SECTION: | |||
273 | config_error(c, "Unexpected section: %s", l->key); | |||
274 | goto error; | |||
275 | ||||
276 | case CONFIG_LINE_TYPE_SECTION_END: | |||
277 | if (!url_map->authorization.realm) | |||
278 | url_map->authorization.realm = strdup("Lwan"); | |||
279 | if (!url_map->authorization.password_file) | |||
280 | url_map->authorization.password_file = strdup("htpasswd"); | |||
281 | ||||
282 | url_map->flags |= HANDLER_MUST_AUTHORIZE; | |||
283 | return; | |||
284 | } | |||
285 | } | |||
286 | ||||
287 | config_error(c, "Could not find end of authorization section"); | |||
288 | ||||
289 | error: | |||
290 | free(url_map->authorization.realm); | |||
291 | free(url_map->authorization.password_file); | |||
292 | } | |||
293 | ||||
294 | static void parse_listener_prefix(struct config *c, | |||
295 | const struct config_line *l, | |||
296 | struct lwan *lwan, | |||
297 | const struct lwan_module *module, | |||
298 | void *handler) | |||
299 | { | |||
300 | struct lwan_url_map url_map = {}; | |||
301 | struct hash *hash = hash_str_new(free, free); | |||
302 | char *prefix = strdupa(l->value)(__extension__ ({ const char *__old = (l->value); size_t __len = strlen (__old) + 1; char *__new = (char *) __builtin_alloca (__len); (char *) memcpy (__new, __old, __len); })); | |||
303 | struct config *isolated; | |||
304 | ||||
305 | isolated = config_isolate_section(c, l); | |||
306 | if (!isolated) { | |||
307 | config_error(c, "Could not isolate configuration file"); | |||
308 | goto out; | |||
309 | } | |||
310 | ||||
311 | while ((l = config_read_line(c))) { | |||
| ||||
312 | switch (l->type) { | |||
313 | case CONFIG_LINE_TYPE_LINE: | |||
314 | hash_add(hash, strdup(l->key), strdup(l->value)); | |||
315 | break; | |||
316 | ||||
317 | case CONFIG_LINE_TYPE_SECTION: | |||
318 | if (streq(l->key, "authorization")) { | |||
319 | parse_listener_prefix_authorization(c, l, &url_map); | |||
320 | } else if (!config_skip_section(c, l)) { | |||
321 | config_error(c, "Could not skip section"); | |||
322 | goto out; | |||
323 | } | |||
324 | break; | |||
325 | ||||
326 | case CONFIG_LINE_TYPE_SECTION_END: | |||
327 | goto add_map; | |||
328 | } | |||
329 | } | |||
330 | ||||
331 | config_error(c, "Expecting section end while parsing prefix"); | |||
332 | goto out; | |||
333 | ||||
334 | add_map: | |||
335 | assert((handler && !module) || (!handler && module))((void) sizeof (((handler && !module) || (!handler && module)) ? 1 : 0), __extension__ ({ if ((handler && ! module) || (!handler && module)) ; else __assert_fail ("(handler && !module) || (!handler && module)" , "/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 335, __extension__ __PRETTY_FUNCTION__); })); | |||
336 | ||||
337 | if (handler) { | |||
338 | url_map.handler = handler; | |||
339 | url_map.flags |= HANDLER_PARSE_MASK | HANDLER_DATA_IS_HASH_TABLE; | |||
340 | url_map.data = hash; | |||
341 | url_map.module = NULL((void*)0); | |||
342 | ||||
343 | hash = NULL((void*)0); | |||
344 | } else if (module->create_from_hash && module->handle_request) { | |||
345 | url_map.data = module->create_from_hash(prefix, hash); | |||
346 | if (!url_map.data) { | |||
347 | config_error(c, "Could not create module instance"); | |||
348 | goto out; | |||
349 | } | |||
350 | ||||
351 | if (module->parse_conf && !module->parse_conf(url_map.data, isolated)) { | |||
352 | const char *msg = config_last_error(isolated); | |||
353 | ||||
354 | config_error(c, "Error from module: %s", msg ? msg : "Unknown"); | |||
355 | goto out; | |||
356 | } | |||
357 | ||||
358 | url_map.handler = module->handle_request; | |||
359 | url_map.flags |= module->flags; | |||
360 | url_map.module = module; | |||
361 | } else if (UNLIKELY(!module->create_from_hash)__builtin_expect(((!module->create_from_hash)), (0))) { | |||
362 | config_error(c, "Module isn't prepared to load settings from a file; " | |||
363 | "create_from_hash() method isn't present"); | |||
364 | goto out; | |||
365 | } else if (UNLIKELY(!module->handle_request)__builtin_expect(((!module->handle_request)), (0))) { | |||
366 | config_error(c, "Module does not have handle_request() method"); | |||
367 | goto out; | |||
368 | } | |||
369 | ||||
370 | add_url_map(&lwan->url_map_trie, prefix, &url_map); | |||
371 | ||||
372 | out: | |||
373 | hash_free(hash); | |||
374 | config_close(isolated); | |||
375 | } | |||
376 | ||||
377 | void lwan_set_url_map(struct lwan *l, const struct lwan_url_map *map) | |||
378 | { | |||
379 | lwan_trie_destroy(&l->url_map_trie); | |||
380 | if (UNLIKELY(!lwan_trie_init(&l->url_map_trie, destroy_urlmap))__builtin_expect(((!lwan_trie_init(&l->url_map_trie, destroy_urlmap ))), (0))) | |||
381 | lwan_status_critical_perror("Could not initialize trie")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 381, __FUNCTION__, "Could not initialize trie"); | |||
382 | ||||
383 | for (; map->prefix; map++) { | |||
384 | struct lwan_url_map *copy = add_url_map(&l->url_map_trie, NULL((void*)0), map); | |||
385 | ||||
386 | if (copy->module && copy->module->create) { | |||
387 | copy->data = copy->module->create (map->prefix, copy->args); | |||
388 | copy->flags = copy->module->flags; | |||
389 | copy->handler = copy->module->handle_request; | |||
390 | } else { | |||
391 | copy->flags = HANDLER_PARSE_MASK; | |||
392 | } | |||
393 | } | |||
394 | } | |||
395 | ||||
396 | static void parse_listener(struct config *c, | |||
397 | const struct config_line *l, | |||
398 | struct lwan *lwan) | |||
399 | { | |||
400 | free(lwan->config.listener); | |||
401 | lwan->config.listener = strdup(l->value); | |||
402 | ||||
403 | while ((l = config_read_line(c))) { | |||
404 | switch (l->type) { | |||
405 | case CONFIG_LINE_TYPE_LINE: | |||
406 | config_error(c, "Expecting prefix section"); | |||
407 | return; | |||
408 | case CONFIG_LINE_TYPE_SECTION: | |||
409 | if (l->key[0] == '&') { | |||
410 | void *handler = find_handler(l->key + 1); | |||
411 | if (handler) { | |||
412 | parse_listener_prefix(c, l, lwan, NULL((void*)0), handler); | |||
413 | continue; | |||
414 | } | |||
415 | ||||
416 | config_error(c, "Could not find handler name: %s", l->key + 1); | |||
417 | return; | |||
418 | } | |||
419 | ||||
420 | const struct lwan_module *module = find_module(l->key); | |||
421 | if (module) { | |||
422 | parse_listener_prefix(c, l, lwan, module, NULL((void*)0)); | |||
423 | continue; | |||
424 | } | |||
425 | ||||
426 | config_error(c, "Invalid section or module not found: %s", l->key); | |||
427 | return; | |||
428 | case CONFIG_LINE_TYPE_SECTION_END: | |||
429 | return; | |||
430 | } | |||
431 | } | |||
432 | ||||
433 | config_error(c, "Expecting section end while parsing listener"); | |||
434 | } | |||
435 | ||||
436 | const char *lwan_get_config_path(char *path_buf, size_t path_buf_len) | |||
437 | { | |||
438 | char buffer[PATH_MAX4096]; | |||
439 | ||||
440 | if (proc_pidpath(getpid(), buffer, sizeof(buffer)) < 0) | |||
441 | goto out; | |||
442 | ||||
443 | char *path = strrchr(buffer, '/'); | |||
444 | if (!path) | |||
445 | goto out; | |||
446 | int ret = snprintf(path_buf, path_buf_len, "%s.conf", path + 1); | |||
447 | if (ret < 0 || ret >= (int)path_buf_len) | |||
448 | goto out; | |||
449 | ||||
450 | return path_buf; | |||
451 | ||||
452 | out: | |||
453 | return "lwan.conf"; | |||
454 | } | |||
455 | ||||
456 | static bool_Bool setup_from_config(struct lwan *lwan, const char *path) | |||
457 | { | |||
458 | const struct config_line *line; | |||
459 | struct config *conf; | |||
460 | bool_Bool has_listener = false0; | |||
461 | char path_buf[PATH_MAX4096]; | |||
462 | ||||
463 | if (!path
| |||
464 | path = lwan_get_config_path(path_buf, sizeof(path_buf)); | |||
465 | lwan_status_info("Loading configuration file: %s", path)lwan_status_info_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 465, __FUNCTION__, "Loading configuration file: %s", path); | |||
466 | ||||
467 | conf = config_open(path); | |||
468 | if (!conf) | |||
469 | return false0; | |||
470 | ||||
471 | if (!lwan_trie_init(&lwan->url_map_trie, destroy_urlmap)) | |||
472 | return false0; | |||
473 | ||||
474 | while ((line = config_read_line(conf))) { | |||
475 | switch (line->type) { | |||
476 | case CONFIG_LINE_TYPE_LINE: | |||
477 | if (streq(line->key, "keep_alive_timeout")) { | |||
478 | lwan->config.keep_alive_timeout = (unsigned int)parse_long( | |||
479 | line->value, default_config.keep_alive_timeout); | |||
480 | } else if (streq(line->key, "quiet")) { | |||
481 | lwan->config.quiet = | |||
482 | parse_bool(line->value, default_config.quiet); | |||
483 | } else if (streq(line->key, "reuse_port")) { | |||
484 | lwan->config.reuse_port = | |||
485 | parse_bool(line->value, default_config.reuse_port); | |||
486 | } else if (streq(line->key, "proxy_protocol")) { | |||
487 | lwan->config.proxy_protocol = | |||
488 | parse_bool(line->value, default_config.proxy_protocol); | |||
489 | } else if (streq(line->key, "allow_cors")) { | |||
490 | lwan->config.allow_cors = | |||
491 | parse_bool(line->value, default_config.allow_cors); | |||
492 | } else if (streq(line->key, "expires")) { | |||
493 | lwan->config.expires = | |||
494 | parse_time_period(line->value, default_config.expires); | |||
495 | } else if (streq(line->key, "error_template")) { | |||
496 | free(lwan->config.error_template); | |||
497 | lwan->config.error_template = strdup(line->value); | |||
498 | } else if (streq(line->key, "threads")) { | |||
499 | long n_threads = | |||
500 | parse_long(line->value, default_config.n_threads); | |||
501 | if (n_threads < 0) | |||
502 | config_error(conf, "Invalid number of threads: %ld", | |||
503 | n_threads); | |||
504 | lwan->config.n_threads = (unsigned int)n_threads; | |||
505 | } else if (streq(line->key, "max_post_data_size")) { | |||
506 | long max_post_data_size = parse_long( | |||
507 | line->value, (long)default_config.max_post_data_size); | |||
508 | if (max_post_data_size < 0) | |||
509 | config_error(conf, "Negative maximum post data size"); | |||
510 | else if (max_post_data_size > 128 * (1 << 20)) | |||
511 | config_error(conf, | |||
512 | "Maximum post data can't be over 128MiB"); | |||
513 | lwan->config.max_post_data_size = (size_t)max_post_data_size; | |||
514 | } else if (streq(line->key, "allow_temp_files")) { | |||
515 | lwan->config.allow_post_temp_file = | |||
516 | !!strstr(line->value, "post"); | |||
517 | } else { | |||
518 | config_error(conf, "Unknown config key: %s", line->key); | |||
519 | } | |||
520 | break; | |||
521 | case CONFIG_LINE_TYPE_SECTION: | |||
522 | if (streq(line->key, "listener")) { | |||
523 | if (!has_listener
| |||
524 | parse_listener(conf, line, lwan); | |||
525 | has_listener = true1; | |||
526 | } else { | |||
527 | config_error(conf, "Only one listener supported"); | |||
528 | } | |||
529 | } else if (streq(line->key, "straitjacket")) { | |||
530 | lwan_straitjacket_enforce_from_config(conf); | |||
531 | } else if (streq(line->key, "headers")) { | |||
532 | parse_global_headers(conf, lwan); | |||
533 | } else { | |||
534 | config_error(conf, "Unknown section type: %s", line->key); | |||
535 | } | |||
536 | break; | |||
537 | case CONFIG_LINE_TYPE_SECTION_END: | |||
538 | config_error(conf, "Unexpected section end"); | |||
539 | } | |||
540 | } | |||
541 | ||||
542 | if (config_last_error(conf)) { | |||
543 | lwan_status_critical("Error on config file \"%s\", line %d: %s", path,lwan_status_critical_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 544, __FUNCTION__, "Error on config file \"%s\", line %d: %s" , path, config_cur_line(conf), config_last_error(conf)) | |||
544 | config_cur_line(conf), config_last_error(conf))lwan_status_critical_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 544, __FUNCTION__, "Error on config file \"%s\", line %d: %s" , path, config_cur_line(conf), config_last_error(conf)); | |||
545 | } | |||
546 | ||||
547 | config_close(conf); | |||
548 | ||||
549 | return true1; | |||
550 | } | |||
551 | ||||
552 | static void try_setup_from_config(struct lwan *l, | |||
553 | const struct lwan_config *config) | |||
554 | { | |||
555 | if (!setup_from_config(l, config->config_file_path)) { | |||
556 | if (config->config_file_path) { | |||
557 | lwan_status_critical("Could not read config file: %s",lwan_status_critical_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 558, __FUNCTION__, "Could not read config file: %s", config ->config_file_path) | |||
558 | config->config_file_path)lwan_status_critical_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 558, __FUNCTION__, "Could not read config file: %s", config ->config_file_path); | |||
559 | } | |||
560 | } | |||
561 | ||||
562 | lwan_status_init(l); /* `quiet` key might have changed value. */ | |||
563 | ||||
564 | l->config.request_flags = | |||
565 | (l->config.proxy_protocol ? REQUEST_ALLOW_PROXY_REQS : 0) | | |||
566 | (l->config.allow_cors ? REQUEST_ALLOW_CORS : 0); | |||
567 | } | |||
568 | ||||
569 | static rlim_t setup_open_file_count_limits(void) | |||
570 | { | |||
571 | struct rlimit r; | |||
572 | ||||
573 | if (getrlimit(RLIMIT_NOFILERLIMIT_NOFILE, &r) < 0) { | |||
574 | lwan_status_perror("Could not obtain maximum number of file "lwan_status_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 576, __FUNCTION__, "Could not obtain maximum number of file " "descriptors. Assuming %d", 256) | |||
575 | "descriptors. Assuming %d",lwan_status_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 576, __FUNCTION__, "Could not obtain maximum number of file " "descriptors. Assuming %d", 256) | |||
576 | OPEN_MAX)lwan_status_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 576, __FUNCTION__, "Could not obtain maximum number of file " "descriptors. Assuming %d", 256); | |||
577 | return OPEN_MAX256; | |||
578 | } | |||
579 | ||||
580 | if (r.rlim_max != r.rlim_cur) { | |||
581 | const rlim_t current = r.rlim_cur; | |||
582 | ||||
583 | if (r.rlim_max == RLIM_INFINITY0xffffffffffffffffuLL && r.rlim_cur < OPEN_MAX256) { | |||
584 | r.rlim_cur = OPEN_MAX256; | |||
585 | } else if (r.rlim_cur < r.rlim_max) { | |||
586 | r.rlim_cur = r.rlim_max; | |||
587 | } else { | |||
588 | /* Shouldn't happen, so just return the current value. */ | |||
589 | goto out; | |||
590 | } | |||
591 | ||||
592 | if (setrlimit(RLIMIT_NOFILERLIMIT_NOFILE, &r) < 0) { | |||
593 | lwan_status_perror("Could not raise maximum number of file "lwan_status_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 595, __FUNCTION__, "Could not raise maximum number of file " "descriptors to %" "l" "u" ". Leaving at " "%" "l" "u", r.rlim_max , current) | |||
594 | "descriptors to %" PRIu64 ". Leaving at "lwan_status_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 595, __FUNCTION__, "Could not raise maximum number of file " "descriptors to %" "l" "u" ". Leaving at " "%" "l" "u", r.rlim_max , current) | |||
595 | "%" PRIu64, r.rlim_max, current)lwan_status_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 595, __FUNCTION__, "Could not raise maximum number of file " "descriptors to %" "l" "u" ". Leaving at " "%" "l" "u", r.rlim_max , current); | |||
596 | r.rlim_cur = current; | |||
597 | } | |||
598 | } | |||
599 | ||||
600 | out: | |||
601 | return r.rlim_cur; | |||
602 | } | |||
603 | ||||
604 | static void allocate_connections(struct lwan *l, size_t max_open_files) | |||
605 | { | |||
606 | const size_t sz = max_open_files * sizeof(struct lwan_connection); | |||
607 | ||||
608 | l->conns = lwan_aligned_alloc(sz, 64); | |||
609 | if (UNLIKELY(!l->conns)__builtin_expect(((!l->conns)), (0))) | |||
610 | lwan_status_critical_perror("lwan_alloc_aligned")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 610, __FUNCTION__, "lwan_alloc_aligned"); | |||
611 | ||||
612 | memset(l->conns, 0, sz); | |||
613 | } | |||
614 | ||||
615 | static unsigned int get_number_of_cpus(void) | |||
616 | { | |||
617 | long n_online_cpus = sysconf(_SC_NPROCESSORS_ONLN_SC_NPROCESSORS_ONLN); | |||
618 | ||||
619 | if (UNLIKELY(n_online_cpus < 0)__builtin_expect(((n_online_cpus < 0)), (0))) { | |||
620 | lwan_status_warning(lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 621, __FUNCTION__, "Could not get number of online CPUs, assuming 1 CPU" ) | |||
621 | "Could not get number of online CPUs, assuming 1 CPU")lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 621, __FUNCTION__, "Could not get number of online CPUs, assuming 1 CPU" ); | |||
622 | return 1; | |||
623 | } | |||
624 | ||||
625 | return (unsigned int)n_online_cpus; | |||
626 | } | |||
627 | ||||
628 | void lwan_init(struct lwan *l) { lwan_init_with_config(l, &default_config); } | |||
| ||||
629 | ||||
630 | const struct lwan_config *lwan_get_default_config(void) | |||
631 | { | |||
632 | return &default_config; | |||
633 | } | |||
634 | ||||
635 | static char *dup_or_null(const char *s) | |||
636 | { | |||
637 | return s ? strdup(s) : NULL((void*)0); | |||
638 | } | |||
639 | ||||
640 | void lwan_init_with_config(struct lwan *l, const struct lwan_config *config) | |||
641 | { | |||
642 | /* Load defaults */ | |||
643 | memset(l, 0, sizeof(*l)); | |||
644 | memcpy(&l->config, config, sizeof(*config)); | |||
645 | l->config.listener = dup_or_null(l->config.listener); | |||
646 | l->config.config_file_path = dup_or_null(l->config.config_file_path); | |||
647 | ||||
648 | /* Initialize status first, as it is used by other things during | |||
649 | * their initialization. */ | |||
650 | lwan_status_init(l); | |||
651 | ||||
652 | /* These will only print debugging messages. Debug messages are always | |||
653 | * printed if we're on a debug build, so the quiet setting will be | |||
654 | * respected. */ | |||
655 | lwan_job_thread_init(); | |||
656 | lwan_tables_init(); | |||
657 | ||||
658 | try_setup_from_config(l, config); | |||
659 | ||||
660 | if (!lwan_strbuf_get_length(&l->headers)) | |||
661 | build_response_headers(l, config->global_headers); | |||
662 | ||||
663 | lwan_response_init(l); | |||
664 | ||||
665 | /* Continue initialization as normal. */ | |||
666 | lwan_status_debug("Initializing lwan web server")lwan_status_debug_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 666, __FUNCTION__, "Initializing lwan web server"); | |||
667 | ||||
668 | l->n_cpus = get_number_of_cpus(); | |||
669 | if (!l->config.n_threads) { | |||
670 | l->thread.count = l->n_cpus; | |||
671 | if (l->thread.count == 1) | |||
672 | l->thread.count = 2; | |||
673 | } else if (l->config.n_threads > 3 * l->n_cpus) { | |||
674 | l->thread.count = l->n_cpus * 3; | |||
675 | ||||
676 | lwan_status_warning("%d threads requested, but only %d online CPUs; "lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 678, __FUNCTION__, "%d threads requested, but only %d online CPUs; " "capping to %d threads", l->config.n_threads, l->n_cpus , 3 * l->n_cpus) | |||
677 | "capping to %d threads",lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 678, __FUNCTION__, "%d threads requested, but only %d online CPUs; " "capping to %d threads", l->config.n_threads, l->n_cpus , 3 * l->n_cpus) | |||
678 | l->config.n_threads, l->n_cpus, 3 * l->n_cpus)lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 678, __FUNCTION__, "%d threads requested, but only %d online CPUs; " "capping to %d threads", l->config.n_threads, l->n_cpus , 3 * l->n_cpus); | |||
679 | } else if (l->config.n_threads > 255) { | |||
680 | l->thread.count = 256; | |||
681 | ||||
682 | lwan_status_warning("%d threads requested, but max 256 supported",lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 683, __FUNCTION__, "%d threads requested, but max 256 supported" , l->config.n_threads) | |||
683 | l->config.n_threads)lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 683, __FUNCTION__, "%d threads requested, but max 256 supported" , l->config.n_threads); | |||
684 | } else { | |||
685 | l->thread.count = l->config.n_threads; | |||
686 | } | |||
687 | ||||
688 | rlim_t max_open_files = setup_open_file_count_limits(); | |||
689 | allocate_connections(l, (size_t)max_open_files); | |||
690 | ||||
691 | l->thread.max_fd = (unsigned)max_open_files / (unsigned)l->thread.count; | |||
692 | lwan_status_info("Using %d threads, maximum %d sockets per thread",lwan_status_info_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 693, __FUNCTION__, "Using %d threads, maximum %d sockets per thread" , l->thread.count, l->thread.max_fd) | |||
693 | l->thread.count, l->thread.max_fd)lwan_status_info_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 693, __FUNCTION__, "Using %d threads, maximum %d sockets per thread" , l->thread.count, l->thread.max_fd); | |||
694 | ||||
695 | signal(SIGPIPE13, SIG_IGN((__sighandler_t) 1)); | |||
696 | ||||
697 | lwan_readahead_init(); | |||
698 | lwan_thread_init(l); | |||
699 | lwan_socket_init(l); | |||
700 | lwan_http_authorize_init(); | |||
701 | } | |||
702 | ||||
703 | void lwan_shutdown(struct lwan *l) | |||
704 | { | |||
705 | lwan_status_info("Shutting down")lwan_status_info_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 705, __FUNCTION__, "Shutting down"); | |||
706 | ||||
707 | free(l->config.listener); | |||
708 | free(l->config.error_template); | |||
709 | free(l->config.config_file_path); | |||
710 | ||||
711 | lwan_job_thread_shutdown(); | |||
712 | lwan_thread_shutdown(l); | |||
713 | ||||
714 | lwan_status_debug("Shutting down URL handlers")lwan_status_debug_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 714, __FUNCTION__, "Shutting down URL handlers"); | |||
715 | lwan_trie_destroy(&l->url_map_trie); | |||
716 | ||||
717 | lwan_strbuf_free(&l->headers); | |||
718 | free(l->conns); | |||
719 | ||||
720 | lwan_response_shutdown(l); | |||
721 | lwan_tables_shutdown(); | |||
722 | lwan_status_shutdown(l); | |||
723 | lwan_http_authorize_shutdown(); | |||
724 | lwan_readahead_shutdown(); | |||
725 | } | |||
726 | ||||
727 | static ALWAYS_INLINEinline __attribute__((always_inline)) int schedule_client(struct lwan *l, int fd) | |||
728 | { | |||
729 | struct lwan_thread *thread = l->conns[fd].thread; | |||
730 | ||||
731 | lwan_thread_add_client(thread, fd); | |||
732 | ||||
733 | return (int)(thread - l->thread.threads); | |||
734 | } | |||
735 | ||||
736 | static volatile sig_atomic_t main_socket = -1; | |||
737 | ||||
738 | static_assert(sizeof(main_socket) >= sizeof(int),extern int (*__Static_assert_function (void)) [!!sizeof (struct { int __error_if_negative: (sizeof(main_socket) >= sizeof (int)) ? 2 : -1; })] | |||
739 | "size of sig_atomic_t > size of int")extern int (*__Static_assert_function (void)) [!!sizeof (struct { int __error_if_negative: (sizeof(main_socket) >= sizeof (int)) ? 2 : -1; })]; | |||
740 | ||||
741 | static void sigint_handler(int signal_number __attribute__((unused))) | |||
742 | { | |||
743 | if (main_socket < 0) | |||
744 | return; | |||
745 | ||||
746 | shutdown((int)main_socket, SHUT_RDWRSHUT_RDWR); | |||
747 | close((int)main_socket); | |||
748 | ||||
749 | main_socket = -1; | |||
750 | } | |||
751 | ||||
752 | enum herd_accept { HERD_MORE = 0, HERD_GONE = -1, HERD_SHUTDOWN = 1 }; | |||
753 | ||||
754 | struct core_bitmap { | |||
755 | uint64_t bitmap[4]; | |||
756 | }; | |||
757 | ||||
758 | static ALWAYS_INLINEinline __attribute__((always_inline)) enum herd_accept | |||
759 | accept_one(struct lwan *l, struct core_bitmap *cores) | |||
760 | { | |||
761 | int fd = accept4((int)main_socket, NULL((void*)0), NULL((void*)0), SOCK_NONBLOCKSOCK_NONBLOCK | SOCK_CLOEXECSOCK_CLOEXEC); | |||
762 | ||||
763 | if (LIKELY(fd >= 0)__builtin_expect((!!(fd >= 0)), (1))) { | |||
764 | int core = schedule_client(l, fd); | |||
765 | ||||
766 | cores->bitmap[core / 64] |= UINT64_C(1)1UL<<(core % 64); | |||
767 | ||||
768 | return HERD_MORE; | |||
769 | } | |||
770 | ||||
771 | switch (errno(*__errno_location ())) { | |||
772 | case EAGAIN11: | |||
773 | return HERD_GONE; | |||
774 | ||||
775 | case EBADF9: | |||
776 | case ECONNABORTED103: | |||
777 | case EINVAL22: | |||
778 | if (main_socket < 0) { | |||
779 | lwan_status_info("Signal 2 (Interrupt) received")lwan_status_info_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 779, __FUNCTION__, "Signal 2 (Interrupt) received"); | |||
780 | } else { | |||
781 | lwan_status_info("Main socket closed for unknown reasons")lwan_status_info_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 781, __FUNCTION__, "Main socket closed for unknown reasons" ); | |||
782 | } | |||
783 | return HERD_SHUTDOWN; | |||
784 | ||||
785 | default: | |||
786 | lwan_status_perror("accept")lwan_status_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 786, __FUNCTION__, "accept"); | |||
787 | return HERD_MORE; | |||
788 | } | |||
789 | } | |||
790 | ||||
791 | void lwan_main_loop(struct lwan *l) | |||
792 | { | |||
793 | struct core_bitmap cores = {}; | |||
794 | ||||
795 | assert(main_socket == -1)((void) sizeof ((main_socket == -1) ? 1 : 0), __extension__ ( { if (main_socket == -1) ; else __assert_fail ("main_socket == -1" , "/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 795, __extension__ __PRETTY_FUNCTION__); })); | |||
796 | main_socket = l->main_socket; | |||
797 | ||||
798 | if (signal(SIGINT2, sigint_handler) == SIG_ERR((__sighandler_t) -1)) | |||
799 | lwan_status_critical("Could not set signal handler")lwan_status_critical_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 799, __FUNCTION__, "Could not set signal handler"); | |||
800 | ||||
801 | lwan_status_info("Ready to serve")lwan_status_info_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan.c" , 801, __FUNCTION__, "Ready to serve"); | |||
802 | ||||
803 | while (true1) { | |||
804 | enum herd_accept ha; | |||
805 | ||||
806 | fcntl(l->main_socket, F_SETFL4, 0); | |||
807 | ha = accept_one(l, &cores); | |||
808 | if (ha == HERD_MORE) { | |||
809 | fcntl(l->main_socket, F_SETFL4, O_NONBLOCK04000); | |||
810 | ||||
811 | do { | |||
812 | ha = accept_one(l, &cores); | |||
813 | } while (ha == HERD_MORE); | |||
814 | } | |||
815 | ||||
816 | if (UNLIKELY(ha > HERD_MORE)__builtin_expect(((ha > HERD_MORE)), (0))) | |||
817 | break; | |||
818 | ||||
819 | for (size_t i = 0; i < N_ELEMENTS(cores.bitmap)((!sizeof(char[1 - 2 * __builtin_types_compatible_p( __typeof__ (cores.bitmap), __typeof__(&(cores.bitmap)[0]))])) | sizeof (cores.bitmap) / sizeof(cores.bitmap[0])); i++) { | |||
820 | for (uint64_t c = cores.bitmap[i]; c; c ^= c & -c) { | |||
821 | size_t core = (size_t)__builtin_ctzl(c); | |||
822 | lwan_thread_nudge(&l->thread.threads[i * 64 + core]); | |||
823 | } | |||
824 | } | |||
825 | cores = (struct core_bitmap){}; | |||
826 | } | |||
827 | } | |||
828 | ||||
829 | #ifdef CLOCK_MONOTONIC_COARSE6 | |||
830 | __attribute__((constructor)) static void detect_fastest_monotonic_clock(void) | |||
831 | { | |||
832 | struct timespec ts; | |||
833 | ||||
834 | if (!clock_gettime(CLOCK_MONOTONIC_COARSE6, &ts)) | |||
835 | monotonic_clock_id = CLOCK_MONOTONIC_COARSE6; | |||
836 | } | |||
837 | #endif | |||
838 | ||||
839 | void lwan_set_thread_name(const char *name) | |||
840 | { | |||
841 | char thread_name[16]; | |||
842 | char process_name[PATH_MAX4096]; | |||
843 | char *tmp; | |||
844 | int ret; | |||
845 | ||||
846 | if (proc_pidpath(getpid(), process_name, sizeof(process_name)) < 0) | |||
847 | return; | |||
848 | ||||
849 | tmp = strrchr(process_name, '/'); | |||
850 | if (!tmp) | |||
851 | return; | |||
852 | ||||
853 | ret = snprintf(thread_name, sizeof(thread_name), "%s %s", tmp + 1, name); | |||
854 | if (ret < 0) | |||
855 | return; | |||
856 | ||||
857 | pthread_set_name_np(pthread_self(), thread_name); | |||
858 | } |