File: | lwan-thread.c |
Warning: | line 824, column 31 Array subscript is undefined |
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 <errno(*__errno_location ()).h> | |||
24 | #include <fcntl.h> | |||
25 | #include <pthread.h> | |||
26 | #include <sched.h> | |||
27 | #include <stdlib.h> | |||
28 | #include <string.h> | |||
29 | #include <sys/epoll.h> | |||
30 | #include <sys/ioctl.h> | |||
31 | #include <sys/socket.h> | |||
32 | #include <unistd.h> | |||
33 | ||||
34 | #if defined(HAVE_SO_ATTACH_REUSEPORT_CBPF) | |||
35 | #include <linux1/filter.h> | |||
36 | #endif | |||
37 | ||||
38 | #include "lwan-private.h" | |||
39 | #include "lwan-tq.h" | |||
40 | #include "list.h" | |||
41 | ||||
42 | static void lwan_strbuf_free_defer(void *data) | |||
43 | { | |||
44 | lwan_strbuf_free((struct lwan_strbuf *)data); | |||
45 | } | |||
46 | ||||
47 | static void graceful_close(struct lwan *l, | |||
48 | struct lwan_connection *conn, | |||
49 | char buffer[static DEFAULT_BUFFER_SIZE4096]) | |||
50 | { | |||
51 | int fd = lwan_connection_get_fd(l, conn); | |||
52 | ||||
53 | while (TIOCOUTQ0x5411) { | |||
54 | /* This ioctl isn't probably doing what it says on the tin; the details | |||
55 | * are subtle, but it seems to do the trick to allow gracefully closing | |||
56 | * the connection in some cases with minimal system calls. */ | |||
57 | int bytes_waiting; | |||
58 | int r = ioctl(fd, TIOCOUTQ0x5411, &bytes_waiting); | |||
59 | ||||
60 | if (!r && !bytes_waiting) /* See note about close(2) below. */ | |||
61 | return; | |||
62 | if (r < 0 && errno(*__errno_location ()) == EINTR4) | |||
63 | continue; | |||
64 | ||||
65 | break; | |||
66 | } | |||
67 | ||||
68 | if (UNLIKELY(shutdown(fd, SHUT_WR) < 0)__builtin_expect(((shutdown(fd, SHUT_WR) < 0)), (0))) { | |||
69 | if (UNLIKELY(errno == ENOTCONN)__builtin_expect((((*__errno_location ()) == 107)), (0))) | |||
70 | return; | |||
71 | } | |||
72 | ||||
73 | for (int tries = 0; tries < 20; tries++) { | |||
74 | ssize_t r = read(fd, buffer, DEFAULT_BUFFER_SIZE4096); | |||
75 | ||||
76 | if (!r) | |||
77 | break; | |||
78 | ||||
79 | if (r < 0) { | |||
80 | switch (errno(*__errno_location ())) { | |||
81 | case EAGAIN11: | |||
82 | break; | |||
83 | case EINTR4: | |||
84 | continue; | |||
85 | default: | |||
86 | return; | |||
87 | } | |||
88 | } | |||
89 | ||||
90 | coro_yield(conn->coro, CONN_CORO_WANT_READ); | |||
91 | } | |||
92 | ||||
93 | /* close(2) will be called when the coroutine yields with CONN_CORO_ABORT */ | |||
94 | } | |||
95 | ||||
96 | __attribute__((noreturn)) static int process_request_coro(struct coro *coro, | |||
97 | void *data) | |||
98 | { | |||
99 | /* NOTE: This function should not return; coro_yield should be used | |||
100 | * instead. This ensures the storage for `strbuf` is alive when the | |||
101 | * coroutine ends and lwan_strbuf_free() is called. */ | |||
102 | struct lwan_connection *conn = data; | |||
103 | struct lwan *lwan = conn->thread->lwan; | |||
104 | int fd = lwan_connection_get_fd(lwan, conn); | |||
105 | enum lwan_request_flags flags = lwan->config.request_flags; | |||
106 | struct lwan_strbuf strbuf = LWAN_STRBUF_STATIC_INIT(struct lwan_strbuf) { .buffer = "" }; | |||
107 | char request_buffer[DEFAULT_BUFFER_SIZE4096]; | |||
108 | struct lwan_value buffer = {.value = request_buffer, .len = 0}; | |||
109 | char *next_request = NULL((void*)0); | |||
110 | char *header_start[N_HEADER_START64]; | |||
111 | struct lwan_proxy proxy; | |||
112 | const int error_when_n_packets = lwan_calculate_n_packets(DEFAULT_BUFFER_SIZE4096); | |||
113 | ||||
114 | coro_defer(coro, lwan_strbuf_free_defer, &strbuf); | |||
115 | ||||
116 | const size_t init_gen = 1; /* 1 call to coro_defer() */ | |||
117 | assert(init_gen == coro_deferred_get_generation(coro))((void) sizeof ((init_gen == coro_deferred_get_generation(coro )) ? 1 : 0), __extension__ ({ if (init_gen == coro_deferred_get_generation (coro)) ; else __assert_fail ("init_gen == coro_deferred_get_generation(coro)" , "/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 117, __extension__ __PRETTY_FUNCTION__); })); | |||
118 | ||||
119 | while (true1) { | |||
120 | struct lwan_request_parser_helper helper = { | |||
121 | .buffer = &buffer, | |||
122 | .next_request = next_request, | |||
123 | .error_when_n_packets = error_when_n_packets, | |||
124 | .header_start = header_start, | |||
125 | }; | |||
126 | struct lwan_request request = {.conn = conn, | |||
127 | .global_response_headers = &lwan->headers, | |||
128 | .fd = fd, | |||
129 | .response = {.buffer = &strbuf}, | |||
130 | .flags = flags, | |||
131 | .proxy = &proxy, | |||
132 | .helper = &helper}; | |||
133 | ||||
134 | lwan_process_request(lwan, &request); | |||
135 | ||||
136 | /* Run the deferred instructions now (except those used to initialize | |||
137 | * the coroutine), so that if the connection is gracefully closed, | |||
138 | * the storage for ``helper'' is still there. */ | |||
139 | coro_deferred_run(coro, init_gen); | |||
140 | ||||
141 | if (UNLIKELY(!(conn->flags & CONN_IS_KEEP_ALIVE))__builtin_expect(((!(conn->flags & CONN_IS_KEEP_ALIVE) )), (0))) { | |||
142 | graceful_close(lwan, conn, request_buffer); | |||
143 | break; | |||
144 | } | |||
145 | ||||
146 | if (next_request && *next_request) { | |||
147 | conn->flags |= CONN_CORK; | |||
148 | ||||
149 | if (!(conn->flags & CONN_EVENTS_WRITE)) | |||
150 | coro_yield(coro, CONN_CORO_WANT_WRITE); | |||
151 | } else { | |||
152 | conn->flags &= ~CONN_CORK; | |||
153 | coro_yield(coro, CONN_CORO_WANT_READ); | |||
154 | } | |||
155 | ||||
156 | /* Ensure string buffer is reset between requests, and that the backing | |||
157 | * store isn't over 2KB. */ | |||
158 | lwan_strbuf_reset_trim(&strbuf, 2048); | |||
159 | ||||
160 | /* Only allow flags from config. */ | |||
161 | flags = request.flags & (REQUEST_PROXIED | REQUEST_ALLOW_CORS); | |||
162 | next_request = helper.next_request; | |||
163 | } | |||
164 | ||||
165 | coro_yield(coro, CONN_CORO_ABORT); | |||
166 | __builtin_unreachable(); | |||
167 | } | |||
168 | ||||
169 | static ALWAYS_INLINEinline __attribute__((always_inline)) uint32_t | |||
170 | conn_flags_to_epoll_events(enum lwan_connection_flags flags) | |||
171 | { | |||
172 | static const uint32_t map[CONN_EVENTS_MASK + 1] = { | |||
173 | [0 /* Suspended (timer or await) */] = EPOLLRDHUPEPOLLRDHUP, | |||
174 | [CONN_EVENTS_WRITE] = EPOLLOUTEPOLLOUT | EPOLLRDHUPEPOLLRDHUP, | |||
175 | [CONN_EVENTS_READ] = EPOLLINEPOLLIN | EPOLLRDHUPEPOLLRDHUP, | |||
176 | [CONN_EVENTS_READ_WRITE] = EPOLLINEPOLLIN | EPOLLOUTEPOLLOUT | EPOLLRDHUPEPOLLRDHUP, | |||
177 | }; | |||
178 | ||||
179 | return map[flags & CONN_EVENTS_MASK]; | |||
180 | } | |||
181 | ||||
182 | static void update_epoll_flags(int fd, | |||
183 | struct lwan_connection *conn, | |||
184 | int epoll_fd, | |||
185 | enum lwan_connection_coro_yield yield_result) | |||
186 | { | |||
187 | static const enum lwan_connection_flags or_mask[CONN_CORO_MAX] = { | |||
188 | [CONN_CORO_YIELD] = 0, | |||
189 | ||||
190 | [CONN_CORO_WANT_READ_WRITE] = CONN_EVENTS_READ_WRITE, | |||
191 | [CONN_CORO_WANT_READ] = CONN_EVENTS_READ, | |||
192 | [CONN_CORO_WANT_WRITE] = CONN_EVENTS_WRITE, | |||
193 | ||||
194 | /* While the coro is suspended, we're not interested in either EPOLLIN | |||
195 | * or EPOLLOUT events. We still want to track this fd in epoll, though, | |||
196 | * so unset both so that only EPOLLRDHUP (plus the implicitly-set ones) | |||
197 | * are set. */ | |||
198 | [CONN_CORO_SUSPEND] = CONN_SUSPENDED, | |||
199 | ||||
200 | /* Ideally, when suspending a coroutine, the current flags&CONN_EVENTS_MASK | |||
201 | * would have to be stored and restored -- however, resuming as if the | |||
202 | * client coroutine is interested in a write event always guarantees that | |||
203 | * they'll be resumed as they're TCP sockets. There's a good chance that | |||
204 | * trying to read from a socket after resuming a coroutine will succeed, | |||
205 | * but if it doesn't because read() returns -EAGAIN, the I/O wrappers will | |||
206 | * yield with CONN_CORO_WANT_READ anyway. */ | |||
207 | [CONN_CORO_RESUME] = CONN_EVENTS_WRITE, | |||
208 | }; | |||
209 | static const enum lwan_connection_flags and_mask[CONN_CORO_MAX] = { | |||
210 | [CONN_CORO_YIELD] = ~0, | |||
211 | ||||
212 | [CONN_CORO_WANT_READ_WRITE] = ~0, | |||
213 | [CONN_CORO_WANT_READ] = ~CONN_EVENTS_WRITE, | |||
214 | [CONN_CORO_WANT_WRITE] = ~CONN_EVENTS_READ, | |||
215 | ||||
216 | [CONN_CORO_SUSPEND] = ~CONN_EVENTS_READ_WRITE, | |||
217 | [CONN_CORO_RESUME] = ~CONN_SUSPENDED, | |||
218 | }; | |||
219 | enum lwan_connection_flags prev_flags = conn->flags; | |||
220 | ||||
221 | conn->flags |= or_mask[yield_result]; | |||
222 | conn->flags &= and_mask[yield_result]; | |||
223 | ||||
224 | if (conn->flags == prev_flags) | |||
225 | return; | |||
226 | ||||
227 | struct epoll_event event = { | |||
228 | .events = conn_flags_to_epoll_events(conn->flags), | |||
229 | .data.ptr = conn, | |||
230 | }; | |||
231 | ||||
232 | if (UNLIKELY(epoll_ctl(epoll_fd, EPOLL_CTL_MOD, fd, &event) < 0)__builtin_expect(((epoll_ctl(epoll_fd, 3, fd, &event) < 0)), (0))) | |||
233 | lwan_status_perror("epoll_ctl")lwan_status_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 233, __FUNCTION__, "epoll_ctl"); | |||
234 | } | |||
235 | ||||
236 | static void clear_async_await_flag(void *data) | |||
237 | { | |||
238 | struct lwan_connection *async_fd_conn = data; | |||
239 | ||||
240 | async_fd_conn->flags &= ~CONN_ASYNC_AWAIT; | |||
241 | } | |||
242 | ||||
243 | static enum lwan_connection_coro_yield | |||
244 | resume_async(struct timeout_queue *tq, | |||
245 | enum lwan_connection_coro_yield yield_result, | |||
246 | int64_t from_coro, | |||
247 | struct lwan_connection *conn, | |||
248 | int epoll_fd) | |||
249 | { | |||
250 | static const enum lwan_connection_flags to_connection_flags[] = { | |||
251 | [CONN_CORO_ASYNC_AWAIT_READ] = CONN_EVENTS_READ, | |||
252 | [CONN_CORO_ASYNC_AWAIT_WRITE] = CONN_EVENTS_WRITE, | |||
253 | [CONN_CORO_ASYNC_AWAIT_READ_WRITE] = CONN_EVENTS_READ_WRITE, | |||
254 | }; | |||
255 | int await_fd = (int)((uint64_t)from_coro >> 32); | |||
256 | enum lwan_connection_flags flags; | |||
257 | int op; | |||
258 | ||||
259 | assert(await_fd >= 0)((void) sizeof ((await_fd >= 0) ? 1 : 0), __extension__ ({ if (await_fd >= 0) ; else __assert_fail ("await_fd >= 0" , "/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 259, __extension__ __PRETTY_FUNCTION__); })); | |||
260 | assert(yield_result >= CONN_CORO_ASYNC_AWAIT_READ &&((void) sizeof ((yield_result >= CONN_CORO_ASYNC_AWAIT_READ && yield_result <= CONN_CORO_ASYNC_AWAIT_READ_WRITE ) ? 1 : 0), __extension__ ({ if (yield_result >= CONN_CORO_ASYNC_AWAIT_READ && yield_result <= CONN_CORO_ASYNC_AWAIT_READ_WRITE ) ; else __assert_fail ("yield_result >= CONN_CORO_ASYNC_AWAIT_READ && yield_result <= CONN_CORO_ASYNC_AWAIT_READ_WRITE" , "/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 261, __extension__ __PRETTY_FUNCTION__); })) | |||
261 | yield_result <= CONN_CORO_ASYNC_AWAIT_READ_WRITE)((void) sizeof ((yield_result >= CONN_CORO_ASYNC_AWAIT_READ && yield_result <= CONN_CORO_ASYNC_AWAIT_READ_WRITE ) ? 1 : 0), __extension__ ({ if (yield_result >= CONN_CORO_ASYNC_AWAIT_READ && yield_result <= CONN_CORO_ASYNC_AWAIT_READ_WRITE ) ; else __assert_fail ("yield_result >= CONN_CORO_ASYNC_AWAIT_READ && yield_result <= CONN_CORO_ASYNC_AWAIT_READ_WRITE" , "/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 261, __extension__ __PRETTY_FUNCTION__); })); | |||
262 | ||||
263 | flags = to_connection_flags[yield_result]; | |||
264 | ||||
265 | struct lwan_connection *await_fd_conn = &tq->lwan->conns[await_fd]; | |||
266 | if (LIKELY(await_fd_conn->flags & CONN_ASYNC_AWAIT)__builtin_expect((!!(await_fd_conn->flags & CONN_ASYNC_AWAIT )), (1))) { | |||
267 | if (LIKELY((await_fd_conn->flags & CONN_EVENTS_MASK) == flags)__builtin_expect((!!((await_fd_conn->flags & CONN_EVENTS_MASK ) == flags)), (1))) | |||
268 | return CONN_CORO_SUSPEND; | |||
269 | ||||
270 | op = EPOLL_CTL_MOD3; | |||
271 | } else { | |||
272 | op = EPOLL_CTL_ADD1; | |||
273 | flags |= CONN_ASYNC_AWAIT; | |||
274 | coro_defer(conn->coro, clear_async_await_flag, await_fd_conn); | |||
275 | } | |||
276 | ||||
277 | struct epoll_event event = {.events = conn_flags_to_epoll_events(flags), | |||
278 | .data.ptr = conn}; | |||
279 | if (LIKELY(!epoll_ctl(epoll_fd, op, await_fd, &event))__builtin_expect((!!(!epoll_ctl(epoll_fd, op, await_fd, & event))), (1))) { | |||
280 | await_fd_conn->flags &= ~CONN_EVENTS_MASK; | |||
281 | await_fd_conn->flags |= flags; | |||
282 | return CONN_CORO_SUSPEND; | |||
283 | } | |||
284 | ||||
285 | return CONN_CORO_ABORT; | |||
286 | } | |||
287 | ||||
288 | static ALWAYS_INLINEinline __attribute__((always_inline)) void resume_coro(struct timeout_queue *tq, | |||
289 | struct lwan_connection *conn, | |||
290 | int epoll_fd) | |||
291 | { | |||
292 | assert(conn->coro)((void) sizeof ((conn->coro) ? 1 : 0), __extension__ ({ if (conn->coro) ; else __assert_fail ("conn->coro", "/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 292, __extension__ __PRETTY_FUNCTION__); })); | |||
293 | ||||
294 | int64_t from_coro = coro_resume(conn->coro); | |||
295 | enum lwan_connection_coro_yield yield_result = from_coro & 0xffffffff; | |||
296 | ||||
297 | if (UNLIKELY(yield_result >= CONN_CORO_ASYNC)__builtin_expect(((yield_result >= CONN_CORO_ASYNC)), (0))) | |||
298 | yield_result = resume_async(tq, yield_result, from_coro, conn, epoll_fd); | |||
299 | ||||
300 | if (UNLIKELY(yield_result == CONN_CORO_ABORT)__builtin_expect(((yield_result == CONN_CORO_ABORT)), (0))) | |||
301 | return timeout_queue_expire(tq, conn); | |||
302 | ||||
303 | return update_epoll_flags(lwan_connection_get_fd(tq->lwan, conn), conn, | |||
304 | epoll_fd, yield_result); | |||
305 | } | |||
306 | ||||
307 | static void update_date_cache(struct lwan_thread *thread) | |||
308 | { | |||
309 | time_t now = time(NULL((void*)0)); | |||
310 | ||||
311 | lwan_format_rfc_time(now, thread->date.date); | |||
312 | lwan_format_rfc_time(now + (time_t)thread->lwan->config.expires, | |||
313 | thread->date.expires); | |||
314 | } | |||
315 | ||||
316 | static ALWAYS_INLINEinline __attribute__((always_inline)) void spawn_coro(struct lwan_connection *conn, | |||
317 | struct coro_switcher *switcher, | |||
318 | struct timeout_queue *tq) | |||
319 | { | |||
320 | struct lwan_thread *t = conn->thread; | |||
321 | ||||
322 | assert(!conn->coro)((void) sizeof ((!conn->coro) ? 1 : 0), __extension__ ({ if (!conn->coro) ; else __assert_fail ("!conn->coro", "/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 322, __extension__ __PRETTY_FUNCTION__); })); | |||
323 | assert(!(conn->flags & CONN_ASYNC_AWAIT))((void) sizeof ((!(conn->flags & CONN_ASYNC_AWAIT)) ? 1 : 0), __extension__ ({ if (!(conn->flags & CONN_ASYNC_AWAIT )) ; else __assert_fail ("!(conn->flags & CONN_ASYNC_AWAIT)" , "/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 323, __extension__ __PRETTY_FUNCTION__); })); | |||
324 | assert(t)((void) sizeof ((t) ? 1 : 0), __extension__ ({ if (t) ; else __assert_fail ("t", "/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 324, __extension__ __PRETTY_FUNCTION__); })); | |||
325 | assert((uintptr_t)t >= (uintptr_t)tq->lwan->thread.threads)((void) sizeof (((uintptr_t)t >= (uintptr_t)tq->lwan-> thread.threads) ? 1 : 0), __extension__ ({ if ((uintptr_t)t >= (uintptr_t)tq->lwan->thread.threads) ; else __assert_fail ("(uintptr_t)t >= (uintptr_t)tq->lwan->thread.threads" , "/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 325, __extension__ __PRETTY_FUNCTION__); })); | |||
326 | assert((uintptr_t)t <((void) sizeof (((uintptr_t)t < (uintptr_t)(tq->lwan-> thread.threads + tq->lwan->thread.count)) ? 1 : 0), __extension__ ({ if ((uintptr_t)t < (uintptr_t)(tq->lwan->thread. threads + tq->lwan->thread.count)) ; else __assert_fail ("(uintptr_t)t < (uintptr_t)(tq->lwan->thread.threads + tq->lwan->thread.count)" , "/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 327, __extension__ __PRETTY_FUNCTION__); })) | |||
327 | (uintptr_t)(tq->lwan->thread.threads + tq->lwan->thread.count))((void) sizeof (((uintptr_t)t < (uintptr_t)(tq->lwan-> thread.threads + tq->lwan->thread.count)) ? 1 : 0), __extension__ ({ if ((uintptr_t)t < (uintptr_t)(tq->lwan->thread. threads + tq->lwan->thread.count)) ; else __assert_fail ("(uintptr_t)t < (uintptr_t)(tq->lwan->thread.threads + tq->lwan->thread.count)" , "/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 327, __extension__ __PRETTY_FUNCTION__); })); | |||
328 | ||||
329 | *conn = (struct lwan_connection) { | |||
330 | .coro = coro_new(switcher, process_request_coro, conn), | |||
331 | .flags = CONN_EVENTS_READ, | |||
332 | .time_to_expire = tq->current_time + tq->move_to_last_bump, | |||
333 | .thread = t, | |||
334 | }; | |||
335 | if (LIKELY(conn->coro)__builtin_expect((!!(conn->coro)), (1))) { | |||
336 | timeout_queue_insert(tq, conn); | |||
337 | return; | |||
338 | } | |||
339 | ||||
340 | /* FIXME: send a "busy" response to this client? we don't have a coroutine | |||
341 | * at this point, can't use lwan_send() here */ | |||
342 | lwan_status_error("Could not create coroutine, dropping connection")lwan_status_error_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 342, __FUNCTION__, "Could not create coroutine, dropping connection" ); | |||
343 | ||||
344 | conn->flags = 0; | |||
345 | ||||
346 | int fd = lwan_connection_get_fd(tq->lwan, conn); | |||
347 | shutdown(fd, SHUT_RDWRSHUT_RDWR); | |||
348 | close(fd); | |||
349 | } | |||
350 | ||||
351 | static bool_Bool process_pending_timers(struct timeout_queue *tq, | |||
352 | struct lwan_thread *t, | |||
353 | int epoll_fd) | |||
354 | { | |||
355 | struct timeout *timeout; | |||
356 | bool_Bool should_expire_timers = false0; | |||
357 | ||||
358 | while ((timeout = timeouts_get(t->wheel))) { | |||
359 | struct lwan_request *request; | |||
360 | ||||
361 | if (timeout == &tq->timeout) { | |||
362 | should_expire_timers = true1; | |||
363 | continue; | |||
364 | } | |||
365 | ||||
366 | request = container_of(timeout, struct lwan_request, timeout)((struct lwan_request *) ((char *)(timeout) - __builtin_offsetof (struct lwan_request, timeout)) + ((typeof(*(timeout)) *)0 != (typeof(((struct lwan_request *)0)->timeout) *)0)); | |||
367 | ||||
368 | update_epoll_flags(request->fd, request->conn, epoll_fd, | |||
369 | CONN_CORO_RESUME); | |||
370 | } | |||
371 | ||||
372 | if (should_expire_timers) { | |||
373 | timeout_queue_expire_waiting(tq); | |||
374 | ||||
375 | /* tq timeout expires every 1000ms if there are connections, so | |||
376 | * update the date cache at this point as well. */ | |||
377 | update_date_cache(t); | |||
378 | ||||
379 | if (!timeout_queue_empty(tq)) { | |||
380 | timeouts_add(t->wheel, &tq->timeout, 1000); | |||
381 | return true1; | |||
382 | } | |||
383 | ||||
384 | timeouts_del(t->wheel, &tq->timeout); | |||
385 | } | |||
386 | ||||
387 | return false0; | |||
388 | } | |||
389 | ||||
390 | static int | |||
391 | turn_timer_wheel(struct timeout_queue *tq, struct lwan_thread *t, int epoll_fd) | |||
392 | { | |||
393 | const int infinite_timeout = -1; | |||
394 | timeout_t wheel_timeout; | |||
395 | struct timespec now; | |||
396 | ||||
397 | if (UNLIKELY(clock_gettime(monotonic_clock_id, &now) < 0)__builtin_expect(((clock_gettime(monotonic_clock_id, &now ) < 0)), (0))) | |||
398 | lwan_status_critical("Could not get monotonic time")lwan_status_critical_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 398, __FUNCTION__, "Could not get monotonic time"); | |||
399 | ||||
400 | timeouts_update(t->wheel, | |||
401 | (timeout_t)(now.tv_sec * 1000 + now.tv_nsec / 1000000)); | |||
402 | ||||
403 | /* Check if there's an expired timer. */ | |||
404 | wheel_timeout = timeouts_timeout(t->wheel); | |||
405 | if (wheel_timeout > 0) { | |||
406 | return (int)wheel_timeout; /* No, but will soon. Wake us up in | |||
407 | wheel_timeout ms. */ | |||
408 | } | |||
409 | ||||
410 | if (UNLIKELY((int64_t)wheel_timeout < 0)__builtin_expect((((int64_t)wheel_timeout < 0)), (0))) | |||
411 | return infinite_timeout; /* None found. */ | |||
412 | ||||
413 | if (!process_pending_timers(tq, t, epoll_fd)) | |||
414 | return infinite_timeout; /* No more timers to process. */ | |||
415 | ||||
416 | /* After processing pending timers, determine when to wake up. */ | |||
417 | return (int)timeouts_timeout(t->wheel); | |||
418 | } | |||
419 | ||||
420 | static bool_Bool accept_waiting_clients(const struct lwan_thread *t) | |||
421 | { | |||
422 | const struct lwan_connection *conns = t->lwan->conns; | |||
423 | ||||
424 | while (true1) { | |||
425 | int fd = | |||
426 | accept4(t->listen_fd, NULL((void*)0), NULL((void*)0), SOCK_NONBLOCKSOCK_NONBLOCK | SOCK_CLOEXECSOCK_CLOEXEC); | |||
427 | ||||
428 | if (LIKELY(fd >= 0)__builtin_expect((!!(fd >= 0)), (1))) { | |||
429 | const struct lwan_connection *conn = &conns[fd]; | |||
430 | struct epoll_event ev = { | |||
431 | .data.ptr = (void *)conn, | |||
432 | .events = conn_flags_to_epoll_events(CONN_EVENTS_READ), | |||
433 | }; | |||
434 | int r = epoll_ctl(conn->thread->epoll_fd, EPOLL_CTL_ADD1, fd, &ev); | |||
435 | ||||
436 | if (UNLIKELY(r < 0)__builtin_expect(((r < 0)), (0))) { | |||
437 | /* FIXME: send a "busy" response here? No coroutine has been | |||
438 | * created at this point to use the usual stuff, though. */ | |||
439 | lwan_status_perror("Could not add file descriptor %d to epoll "lwan_status_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 441, __FUNCTION__, "Could not add file descriptor %d to epoll " "set %d. Dropping connection", fd, conn->thread->epoll_fd ) | |||
440 | "set %d. Dropping connection",lwan_status_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 441, __FUNCTION__, "Could not add file descriptor %d to epoll " "set %d. Dropping connection", fd, conn->thread->epoll_fd ) | |||
441 | fd, conn->thread->epoll_fd)lwan_status_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 441, __FUNCTION__, "Could not add file descriptor %d to epoll " "set %d. Dropping connection", fd, conn->thread->epoll_fd ); | |||
442 | shutdown(fd, SHUT_RDWRSHUT_RDWR); | |||
443 | close(fd); | |||
444 | } | |||
445 | ||||
446 | #if defined(HAVE_SO_INCOMING_CPU) && defined(__x86_64__1) | |||
447 | /* Ignore errors here, as this is just a hint */ | |||
448 | (void)setsockopt(fd, SOL_SOCKET1, SO_INCOMING_CPU49, &t->cpu, sizeof(t->cpu)); | |||
449 | #endif | |||
450 | ||||
451 | continue; | |||
452 | } | |||
453 | ||||
454 | switch (errno(*__errno_location ())) { | |||
455 | default: | |||
456 | lwan_status_perror("Unexpected error while accepting connections")lwan_status_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 456, __FUNCTION__, "Unexpected error while accepting connections" ); | |||
457 | /* fallthrough */ | |||
458 | ||||
459 | case EAGAIN11: | |||
460 | return true1; | |||
461 | ||||
462 | case EBADF9: | |||
463 | case ECONNABORTED103: | |||
464 | case EINVAL22: | |||
465 | lwan_status_info("Listening socket closed")lwan_status_info_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 465, __FUNCTION__, "Listening socket closed"); | |||
466 | return false0; | |||
467 | } | |||
468 | } | |||
469 | ||||
470 | __builtin_unreachable(); | |||
471 | } | |||
472 | ||||
473 | static int create_listen_socket(struct lwan_thread *t, | |||
474 | unsigned int num, | |||
475 | unsigned int num_sockets) | |||
476 | { | |||
477 | int listen_fd; | |||
478 | ||||
479 | listen_fd = lwan_create_listen_socket(t->lwan, num == 0); | |||
480 | if (listen_fd < 0) | |||
481 | lwan_status_critical("Could not create listen_fd")lwan_status_critical_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 481, __FUNCTION__, "Could not create listen_fd"); | |||
482 | ||||
483 | /* Ignore errors here, as this is just a hint */ | |||
484 | #if defined(HAVE_SO_ATTACH_REUSEPORT_CBPF) | |||
485 | /* From socket(7): "These options may be set repeatedly at any time on | |||
486 | * any socket in the group to replace the current BPF program used by | |||
487 | * all sockets in the group." */ | |||
488 | if (num == 0) { | |||
489 | /* From socket(7): "The BPF program must return an index between 0 and | |||
490 | * N-1 representing the socket which should receive the packet (where N | |||
491 | * is the number of sockets in the group)." */ | |||
492 | const uint32_t cpu_ad_off = (uint32_t)SKF_AD_OFF(-0x1000) + SKF_AD_CPU36; | |||
493 | struct sock_filter filter[] = { | |||
494 | {BPF_LD0x00 | BPF_W0x00 | BPF_ABS0x20, 0, 0, cpu_ad_off}, /* A = curr_cpu_index */ | |||
495 | {BPF_ALU0x04 | BPF_MOD0x90 | BPF_K0x00, 0, 0, num_sockets}, /* A %= num_sockets */ | |||
496 | {BPF_RET0x06 | BPF_A0x10, 0, 0, 0}, /* return A */ | |||
497 | }; | |||
498 | struct sock_fprog fprog = {.filter = filter, .len = N_ELEMENTS(filter)((!sizeof(char[1 - 2 * __builtin_types_compatible_p( __typeof__ (filter), __typeof__(&(filter)[0]))])) | sizeof(filter) / sizeof(filter[0]))}; | |||
499 | ||||
500 | if (!(num_sockets & (num_sockets - 1))) { | |||
501 | /* FIXME: Is this strength reduction already made by the kernel? */ | |||
502 | filter[1].code &= (uint16_t)~BPF_MOD0x90; | |||
503 | filter[1].code |= BPF_AND0x50; | |||
504 | filter[1].k = num_sockets - 1; | |||
505 | } | |||
506 | ||||
507 | (void)setsockopt(listen_fd, SOL_SOCKET1, SO_ATTACH_REUSEPORT_CBPF51, | |||
508 | &fprog, sizeof(fprog)); | |||
509 | (void)setsockopt(listen_fd, SOL_SOCKET1, SO_LOCK_FILTER44, | |||
510 | (int[]){1}, sizeof(int)); | |||
511 | } | |||
512 | #elif defined(HAVE_SO_INCOMING_CPU) && defined(__x86_64__1) | |||
513 | (void)setsockopt(listen_fd, SOL_SOCKET1, SO_INCOMING_CPU49, &t->cpu, | |||
514 | sizeof(t->cpu)); | |||
515 | #endif | |||
516 | ||||
517 | struct epoll_event event = { | |||
518 | .events = EPOLLINEPOLLIN | EPOLLETEPOLLET | EPOLLERREPOLLERR, | |||
519 | .data.ptr = NULL((void*)0), | |||
520 | }; | |||
521 | if (epoll_ctl(t->epoll_fd, EPOLL_CTL_ADD1, listen_fd, &event) < 0) | |||
522 | lwan_status_critical_perror("Could not add socket to epoll")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 522, __FUNCTION__, "Could not add socket to epoll"); | |||
523 | ||||
524 | return listen_fd; | |||
525 | } | |||
526 | ||||
527 | static void *thread_io_loop(void *data) | |||
528 | { | |||
529 | struct lwan_thread *t = data; | |||
530 | int epoll_fd = t->epoll_fd; | |||
531 | const int max_events = LWAN_MIN((int)t->lwan->thread.max_fd, 1024)({ const __typeof__(((int)t->lwan->thread.max_fd) + 0) lwan_tmp_id8 = ((int)t->lwan->thread.max_fd); const __typeof__((1024 ) + 0) lwan_tmp_id9 = (1024); lwan_tmp_id8 > lwan_tmp_id9 ? lwan_tmp_id9 : lwan_tmp_id8; }); | |||
532 | struct lwan *lwan = t->lwan; | |||
533 | struct epoll_event *events; | |||
534 | struct coro_switcher switcher; | |||
535 | struct timeout_queue tq; | |||
536 | ||||
537 | lwan_status_debug("Worker thread #%zd starting",lwan_status_debug_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 538, __FUNCTION__, "Worker thread #%zd starting", t - t-> lwan->thread.threads + 1) | |||
538 | t - t->lwan->thread.threads + 1)lwan_status_debug_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 538, __FUNCTION__, "Worker thread #%zd starting", t - t-> lwan->thread.threads + 1); | |||
539 | lwan_set_thread_name("worker"); | |||
540 | ||||
541 | events = calloc((size_t)max_events, sizeof(*events)); | |||
542 | if (UNLIKELY(!events)__builtin_expect(((!events)), (0))) | |||
543 | lwan_status_critical("Could not allocate memory for events")lwan_status_critical_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 543, __FUNCTION__, "Could not allocate memory for events"); | |||
544 | ||||
545 | update_date_cache(t); | |||
546 | ||||
547 | timeout_queue_init(&tq, lwan); | |||
548 | ||||
549 | pthread_barrier_wait(&lwan->thread.barrier); | |||
550 | ||||
551 | for (;;) { | |||
552 | int timeout = turn_timer_wheel(&tq, t, epoll_fd); | |||
553 | int n_fds = epoll_wait(epoll_fd, events, max_events, timeout); | |||
554 | bool_Bool accepted_connections = false0; | |||
555 | ||||
556 | if (UNLIKELY(n_fds < 0)__builtin_expect(((n_fds < 0)), (0))) { | |||
557 | if (errno(*__errno_location ()) == EBADF9 || errno(*__errno_location ()) == EINVAL22) | |||
558 | break; | |||
559 | continue; | |||
560 | } | |||
561 | ||||
562 | for (struct epoll_event *event = events; n_fds--; event++) { | |||
563 | struct lwan_connection *conn; | |||
564 | ||||
565 | if (!event->data.ptr) { | |||
566 | if (LIKELY(accept_waiting_clients(t))__builtin_expect((!!(accept_waiting_clients(t))), (1))) { | |||
567 | accepted_connections = true1; | |||
568 | continue; | |||
569 | } | |||
570 | close(epoll_fd); | |||
571 | epoll_fd = -1; | |||
572 | break; | |||
573 | } | |||
574 | ||||
575 | conn = event->data.ptr; | |||
576 | ||||
577 | if (UNLIKELY(event->events & (EPOLLRDHUP | EPOLLHUP))__builtin_expect(((event->events & (EPOLLRDHUP | EPOLLHUP ))), (0))) { | |||
578 | timeout_queue_expire(&tq, conn); | |||
579 | continue; | |||
580 | } | |||
581 | ||||
582 | if (!conn->coro) { | |||
583 | spawn_coro(conn, &switcher, &tq); | |||
584 | continue; | |||
585 | } | |||
586 | ||||
587 | resume_coro(&tq, conn, epoll_fd); | |||
588 | timeout_queue_move_to_last(&tq, conn); | |||
589 | } | |||
590 | ||||
591 | if (accepted_connections) { | |||
592 | timeouts_add(t->wheel, &tq.timeout, 1000); | |||
593 | accepted_connections = false0; | |||
594 | } | |||
595 | } | |||
596 | ||||
597 | pthread_barrier_wait(&lwan->thread.barrier); | |||
598 | ||||
599 | timeout_queue_expire_all(&tq); | |||
600 | free(events); | |||
601 | ||||
602 | return NULL((void*)0); | |||
603 | } | |||
604 | ||||
605 | static void create_thread(struct lwan *l, struct lwan_thread *thread) | |||
606 | { | |||
607 | int ignore; | |||
608 | pthread_attr_t attr; | |||
609 | ||||
610 | thread->lwan = l; | |||
611 | ||||
612 | thread->wheel = timeouts_open(&ignore); | |||
613 | if (!thread->wheel) | |||
614 | lwan_status_critical("Could not create timer wheel")lwan_status_critical_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 614, __FUNCTION__, "Could not create timer wheel"); | |||
615 | ||||
616 | if ((thread->epoll_fd = epoll_create1(EPOLL_CLOEXECEPOLL_CLOEXEC)) < 0) | |||
617 | lwan_status_critical_perror("epoll_create")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 617, __FUNCTION__, "epoll_create"); | |||
618 | ||||
619 | if (pthread_attr_init(&attr)) | |||
620 | lwan_status_critical_perror("pthread_attr_init")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 620, __FUNCTION__, "pthread_attr_init"); | |||
621 | ||||
622 | if (pthread_attr_setscope(&attr, PTHREAD_SCOPE_SYSTEMPTHREAD_SCOPE_SYSTEM)) | |||
623 | lwan_status_critical_perror("pthread_attr_setscope")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 623, __FUNCTION__, "pthread_attr_setscope"); | |||
624 | ||||
625 | if (pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLEPTHREAD_CREATE_JOINABLE)) | |||
626 | lwan_status_critical_perror("pthread_attr_setdetachstate")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 626, __FUNCTION__, "pthread_attr_setdetachstate"); | |||
627 | ||||
628 | if (pthread_create(&thread->self, &attr, thread_io_loop, thread)) | |||
629 | lwan_status_critical_perror("pthread_create")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 629, __FUNCTION__, "pthread_create"); | |||
630 | ||||
631 | if (pthread_attr_destroy(&attr)) | |||
632 | lwan_status_critical_perror("pthread_attr_destroy")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 632, __FUNCTION__, "pthread_attr_destroy"); | |||
633 | } | |||
634 | ||||
635 | #if defined(__linux__1) && defined(__x86_64__1) | |||
636 | static bool_Bool read_cpu_topology(struct lwan *l, uint32_t siblings[]) | |||
637 | { | |||
638 | char path[PATH_MAX4096]; | |||
639 | ||||
640 | for (uint32_t i = 0; i < l->available_cpus; i++) | |||
641 | siblings[i] = 0xbebacafe; | |||
642 | ||||
643 | for (unsigned int i = 0; i < l->available_cpus; i++) { | |||
644 | FILE *sib; | |||
645 | uint32_t id, sibling; | |||
646 | char separator; | |||
647 | ||||
648 | snprintf(path, sizeof(path), | |||
649 | "/sys/devices/system/cpu/cpu%d/topology/thread_siblings_list", | |||
650 | i); | |||
651 | ||||
652 | sib = fopen(path, "re"); | |||
653 | if (!sib) { | |||
654 | lwan_status_warning("Could not open `%s` to determine CPU topology",lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 655, __FUNCTION__, "Could not open `%s` to determine CPU topology" , path) | |||
655 | path)lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 655, __FUNCTION__, "Could not open `%s` to determine CPU topology" , path); | |||
656 | return false0; | |||
657 | } | |||
658 | ||||
659 | switch (fscanf(sib, "%u%c%u", &id, &separator, &sibling)) { | |||
660 | case 2: /* No SMT */ | |||
661 | siblings[i] = id; | |||
662 | break; | |||
663 | case 3: /* SMT */ | |||
664 | if (!(separator == ',' || separator == '-')) { | |||
665 | lwan_status_critical("Expecting either ',' or '-' for sibling separator")lwan_status_critical_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 665, __FUNCTION__, "Expecting either ',' or '-' for sibling separator" ); | |||
666 | __builtin_unreachable(); | |||
667 | } | |||
668 | ||||
669 | siblings[i] = sibling; | |||
670 | break; | |||
671 | default: | |||
672 | lwan_status_critical("%s has invalid format", path)lwan_status_critical_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 672, __FUNCTION__, "%s has invalid format", path); | |||
673 | __builtin_unreachable(); | |||
674 | } | |||
675 | ||||
676 | fclose(sib); | |||
677 | } | |||
678 | ||||
679 | /* Perform a sanity check here, as some systems seem to filter out the | |||
680 | * result of sysconf() to obtain the number of configured and online | |||
681 | * CPUs but don't bother changing what's available through sysfs as far | |||
682 | * as the CPU topology information goes. It's better to fall back to a | |||
683 | * possibly non-optimal setup than just crash during startup while | |||
684 | * trying to perform an out-of-bounds array access. */ | |||
685 | for (unsigned int i = 0; i < l->available_cpus; i++) { | |||
686 | if (siblings[i] == 0xbebacafe) { | |||
687 | lwan_status_warning("Could not determine sibling for CPU %d", i)lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 687, __FUNCTION__, "Could not determine sibling for CPU %d" , i); | |||
688 | return false0; | |||
689 | } | |||
690 | ||||
691 | if (siblings[i] >= l->available_cpus) { | |||
692 | lwan_status_warning("CPU information topology says CPU %d exists, "lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 695, __FUNCTION__, "CPU information topology says CPU %d exists, " "but max available CPUs is %d (online CPUs: %d). " "Is Lwan running in a (broken) container?" , siblings[i], l->available_cpus, l->online_cpus) | |||
693 | "but max available CPUs is %d (online CPUs: %d). "lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 695, __FUNCTION__, "CPU information topology says CPU %d exists, " "but max available CPUs is %d (online CPUs: %d). " "Is Lwan running in a (broken) container?" , siblings[i], l->available_cpus, l->online_cpus) | |||
694 | "Is Lwan running in a (broken) container?",lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 695, __FUNCTION__, "CPU information topology says CPU %d exists, " "but max available CPUs is %d (online CPUs: %d). " "Is Lwan running in a (broken) container?" , siblings[i], l->available_cpus, l->online_cpus) | |||
695 | siblings[i], l->available_cpus, l->online_cpus)lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 695, __FUNCTION__, "CPU information topology says CPU %d exists, " "but max available CPUs is %d (online CPUs: %d). " "Is Lwan running in a (broken) container?" , siblings[i], l->available_cpus, l->online_cpus); | |||
696 | return false0; | |||
697 | } | |||
698 | } | |||
699 | ||||
700 | return true1; | |||
701 | } | |||
702 | ||||
703 | static void | |||
704 | siblings_to_schedtbl(struct lwan *l, uint32_t siblings[], uint32_t schedtbl[]) | |||
705 | { | |||
706 | int *seen = alloca(l->available_cpus * sizeof(int))__builtin_alloca (l->available_cpus * sizeof(int)); | |||
707 | unsigned int n_schedtbl = 0; | |||
708 | ||||
709 | for (uint32_t i = 0; i < l->available_cpus; i++) | |||
710 | seen[i] = -1; | |||
711 | ||||
712 | for (uint32_t i = 0; i < l->available_cpus; i++) { | |||
713 | if (seen[siblings[i]] < 0) { | |||
714 | seen[siblings[i]] = (int)i; | |||
715 | } else { | |||
716 | schedtbl[n_schedtbl++] = (uint32_t)seen[siblings[i]]; | |||
717 | schedtbl[n_schedtbl++] = i; | |||
718 | } | |||
719 | } | |||
720 | ||||
721 | if (n_schedtbl != l->available_cpus) | |||
722 | memcpy(schedtbl, seen, l->available_cpus * sizeof(int)); | |||
723 | } | |||
724 | ||||
725 | static bool_Bool | |||
726 | topology_to_schedtbl(struct lwan *l, uint32_t schedtbl[], uint32_t n_threads) | |||
727 | { | |||
728 | uint32_t *siblings = alloca(l->available_cpus * sizeof(uint32_t))__builtin_alloca (l->available_cpus * sizeof(uint32_t)); | |||
729 | ||||
730 | if (read_cpu_topology(l, siblings)) { | |||
731 | uint32_t *affinity = alloca(l->available_cpus * sizeof(uint32_t))__builtin_alloca (l->available_cpus * sizeof(uint32_t)); | |||
732 | ||||
733 | siblings_to_schedtbl(l, siblings, affinity); | |||
734 | ||||
735 | for (uint32_t i = 0; i < n_threads; i++) | |||
736 | schedtbl[i] = affinity[i % l->available_cpus]; | |||
737 | return true1; | |||
738 | } | |||
739 | ||||
740 | for (uint32_t i = 0; i < n_threads; i++) | |||
741 | schedtbl[i] = (i / 2) % l->thread.count; | |||
742 | return false0; | |||
743 | } | |||
744 | ||||
745 | static void | |||
746 | adjust_threads_affinity(struct lwan *l, uint32_t *schedtbl, uint32_t mask) | |||
747 | { | |||
748 | for (uint32_t i = 0; i < l->thread.count; i++) { | |||
749 | cpu_set_t set; | |||
750 | ||||
751 | CPU_ZERO(&set)do __builtin_memset (&set, '\0', sizeof (cpu_set_t)); while (0); | |||
752 | CPU_SET(schedtbl[i & mask], &set)(__extension__ ({ size_t __cpu = (schedtbl[i & mask]); __cpu / 8 < (sizeof (cpu_set_t)) ? (((__cpu_mask *) ((&set) ->__bits))[((__cpu) / (8 * sizeof (__cpu_mask)))] |= ((__cpu_mask ) 1 << ((__cpu) % (8 * sizeof (__cpu_mask))))) : 0; })); | |||
753 | ||||
754 | if (pthread_setaffinity_np(l->thread.threads[i].self, sizeof(set), | |||
755 | &set)) | |||
756 | lwan_status_warning("Could not set affinity for thread %d", i)lwan_status_warning_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 756, __FUNCTION__, "Could not set affinity for thread %d", i ); | |||
757 | } | |||
758 | } | |||
759 | #elif defined(__x86_64__1) | |||
760 | static bool_Bool | |||
761 | topology_to_schedtbl(struct lwan *l, uint32_t schedtbl[], uint32_t n_threads) | |||
762 | { | |||
763 | for (uint32_t i = 0; i < n_threads; i++) | |||
764 | schedtbl[i] = (i / 2) % l->thread.count; | |||
765 | return false0; | |||
766 | } | |||
767 | ||||
768 | static void | |||
769 | adjust_threads_affinity(struct lwan *l, uint32_t *schedtbl, uint32_t n) | |||
770 | { | |||
771 | } | |||
772 | #endif | |||
773 | ||||
774 | void lwan_thread_init(struct lwan *l) | |||
775 | { | |||
776 | const unsigned int total_conns = l->thread.max_fd * l->thread.count; | |||
777 | ||||
778 | if (pthread_barrier_init(&l->thread.barrier, NULL((void*)0), | |||
| ||||
779 | (unsigned)l->thread.count + 1)) | |||
780 | lwan_status_critical("Could not create barrier")lwan_status_critical_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 780, __FUNCTION__, "Could not create barrier"); | |||
781 | ||||
782 | lwan_status_debug("Initializing threads")lwan_status_debug_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 782, __FUNCTION__, "Initializing threads"); | |||
783 | ||||
784 | l->thread.threads = | |||
785 | calloc((size_t)l->thread.count, sizeof(struct lwan_thread)); | |||
786 | if (!l->thread.threads) | |||
787 | lwan_status_critical("Could not allocate memory for threads")lwan_status_critical_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 787, __FUNCTION__, "Could not allocate memory for threads"); | |||
788 | ||||
789 | #ifdef __x86_64__1 | |||
790 | static_assert(sizeof(struct lwan_connection) == 32,extern int (*__Static_assert_function (void)) [!!sizeof (struct { int __error_if_negative: (sizeof(struct lwan_connection) == 32) ? 2 : -1; })] | |||
791 | "Two connections per cache line")extern int (*__Static_assert_function (void)) [!!sizeof (struct { int __error_if_negative: (sizeof(struct lwan_connection) == 32) ? 2 : -1; })]; | |||
792 | #ifdef _SC_LEVEL1_DCACHE_LINESIZE_SC_LEVEL1_DCACHE_LINESIZE | |||
793 | assert(sysconf(_SC_LEVEL1_DCACHE_LINESIZE) == 64)((void) sizeof ((sysconf(_SC_LEVEL1_DCACHE_LINESIZE) == 64) ? 1 : 0), __extension__ ({ if (sysconf(_SC_LEVEL1_DCACHE_LINESIZE ) == 64) ; else __assert_fail ("sysconf(_SC_LEVEL1_DCACHE_LINESIZE) == 64" , "/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 793, __extension__ __PRETTY_FUNCTION__); })); | |||
794 | #endif | |||
795 | ||||
796 | lwan_status_debug("%d CPUs of %d are online. "lwan_status_debug_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 798, __FUNCTION__, "%d CPUs of %d are online. " "Reading topology to pre-schedule clients" , l->online_cpus, l->available_cpus) | |||
797 | "Reading topology to pre-schedule clients",lwan_status_debug_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 798, __FUNCTION__, "%d CPUs of %d are online. " "Reading topology to pre-schedule clients" , l->online_cpus, l->available_cpus) | |||
798 | l->online_cpus, l->available_cpus)lwan_status_debug_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 798, __FUNCTION__, "%d CPUs of %d are online. " "Reading topology to pre-schedule clients" , l->online_cpus, l->available_cpus); | |||
799 | ||||
800 | /* | |||
801 | * Pre-schedule each file descriptor, to reduce some operations in the | |||
802 | * fast path. | |||
803 | * | |||
804 | * Since struct lwan_connection is guaranteed to be 32-byte long, two of | |||
805 | * them can fill up a cache line. Assume siblings share cache lines and | |||
806 | * use the CPU topology to group two connections per cache line in such | |||
807 | * a way that false sharing is avoided. | |||
808 | */ | |||
809 | uint32_t n_threads = (uint32_t)lwan_nextpow2((size_t)((l->thread.count - 1) * 2)); | |||
810 | uint32_t *schedtbl = alloca(n_threads * sizeof(uint32_t))__builtin_alloca (n_threads * sizeof(uint32_t)); | |||
811 | ||||
812 | bool_Bool adj_affinity = topology_to_schedtbl(l, schedtbl, n_threads); | |||
813 | ||||
814 | n_threads--; /* Transform count into mask for AND below */ | |||
815 | ||||
816 | if (adj_affinity
| |||
817 | /* Save which CPU this tread will be pinned at so we can use | |||
818 | * SO_INCOMING_CPU later. */ | |||
819 | for (unsigned int i = 0; i < l->thread.count; i++) | |||
820 | l->thread.threads[i].cpu = schedtbl[i & n_threads]; | |||
821 | } | |||
822 | ||||
823 | for (unsigned int i = 0; i < total_conns; i++) | |||
824 | l->conns[i].thread = &l->thread.threads[schedtbl[i & n_threads]]; | |||
| ||||
825 | #else | |||
826 | for (unsigned int i = 0; i < l->thread.count; i++) | |||
827 | l->thread.threads[i].cpu = i % l->online_cpus; | |||
828 | for (unsigned int i = 0; i < total_conns; i++) | |||
829 | l->conns[i].thread = &l->thread.threads[i % l->thread.count]; | |||
830 | #endif | |||
831 | ||||
832 | for (unsigned int i = 0; i < l->thread.count; i++) { | |||
833 | struct lwan_thread *thread = &l->thread.threads[i]; | |||
834 | ||||
835 | create_thread(l, thread); | |||
836 | ||||
837 | if ((thread->listen_fd = create_listen_socket(thread, i, l->thread.count)) < 0) | |||
838 | lwan_status_critical_perror("Could not create listening socket")lwan_status_critical_perror_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 838, __FUNCTION__, "Could not create listening socket"); | |||
839 | } | |||
840 | ||||
841 | #ifdef __x86_64__1 | |||
842 | if (adj_affinity) | |||
843 | adjust_threads_affinity(l, schedtbl, n_threads); | |||
844 | #endif | |||
845 | ||||
846 | pthread_barrier_wait(&l->thread.barrier); | |||
847 | ||||
848 | lwan_status_debug("Worker threads created and ready to serve")lwan_status_debug_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 848, __FUNCTION__, "Worker threads created and ready to serve" ); | |||
849 | } | |||
850 | ||||
851 | void lwan_thread_shutdown(struct lwan *l) | |||
852 | { | |||
853 | lwan_status_debug("Shutting down threads")lwan_status_debug_debug("/home/buildbot/lwan-worker/clang-analyze/build/src/lib/lwan-thread.c" , 853, __FUNCTION__, "Shutting down threads"); | |||
854 | ||||
855 | for (unsigned int i = 0; i < l->thread.count; i++) { | |||
856 | struct lwan_thread *t = &l->thread.threads[i]; | |||
857 | int epoll_fd = t->epoll_fd; | |||
858 | int listen_fd = t->listen_fd; | |||
859 | ||||
860 | t->listen_fd = -1; | |||
861 | t->epoll_fd = -1; | |||
862 | close(epoll_fd); | |||
863 | close(listen_fd); | |||
864 | } | |||
865 | ||||
866 | pthread_barrier_wait(&l->thread.barrier); | |||
867 | pthread_barrier_destroy(&l->thread.barrier); | |||
868 | ||||
869 | for (unsigned int i = 0; i < l->thread.count; i++) { | |||
870 | struct lwan_thread *t = &l->thread.threads[i]; | |||
871 | ||||
872 | pthread_join(l->thread.threads[i].self, NULL((void*)0)); | |||
873 | timeouts_close(t->wheel); | |||
874 | } | |||
875 | ||||
876 | free(l->thread.threads); | |||
877 | } |