2 * Copyright (C) 2013 - Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License, version 2 only,
6 * as published by the Free Software Foundation.
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
13 * You should have received a copy of the GNU General Public License along
14 * with this program; if not, write to the Free Software Foundation, Inc.,
15 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
21 #include <common/hashtable/hashtable.h>
22 #include <common/common.h>
23 #include <common/utils.h>
26 #include "lttng-sessiond.h"
27 #include "health-sessiond.h"
28 #include "testpoint.h"
30 #include "ht-cleanup.h"
32 static int ht_cleanup_quit_pipe
[2] = { -1, -1 };
35 * Check if the ht_cleanup thread quit pipe was triggered.
37 * Return true if it was triggered else false;
39 static bool check_quit_pipe(int fd
, uint32_t events
)
41 return (fd
== ht_cleanup_quit_pipe
[0] && (events
& LPOLLIN
));
44 static int init_pipe(int *pipe_fds
)
50 PERROR("ht_cleanup thread quit pipe");
54 for (i
= 0; i
< 2; i
++) {
55 ret
= fcntl(pipe_fds
[i
], F_SETFD
, FD_CLOEXEC
);
57 PERROR("fcntl ht_cleanup_quit_pipe");
66 * Create a poll set with O_CLOEXEC and add the thread quit pipe to the set.
68 static int set_pollset(struct lttng_poll_event
*events
, size_t size
)
72 ret
= lttng_poll_create(events
, size
, LTTNG_CLOEXEC
);
77 ret
= lttng_poll_add(events
, ht_cleanup_quit_pipe
[0],
83 ret
= lttng_poll_add(events
, ht_cleanup_pipe
[0], LPOLLIN
| LPOLLERR
);
85 DBG("[ht-thread] lttng_poll_add error %d.", ret
);
95 static void cleanup_ht_cleanup_thread(void *data
)
97 utils_close_pipe(ht_cleanup_quit_pipe
);
98 utils_close_pipe(ht_cleanup_pipe
);
101 static void *thread_ht_cleanup(void *data
)
103 int ret
, i
, pollfd
, err
= -1;
105 uint32_t revents
, nb_fd
;
106 struct lttng_poll_event events
;
108 DBG("[ht-thread] startup.");
110 rcu_register_thread();
113 health_register(health_sessiond
, HEALTH_SESSIOND_TYPE_HT_CLEANUP
);
115 if (testpoint(sessiond_thread_ht_cleanup
)) {
116 DBG("[ht-thread] testpoint.");
117 goto error_testpoint
;
120 health_code_update();
122 ret
= set_pollset(&events
, 2);
124 DBG("[ht-thread] sessiond_set_ht_cleanup_thread_pollset error %d.", ret
);
125 goto error_poll_create
;
128 health_code_update();
131 DBG3("[ht-thread] Polling.");
133 ret
= lttng_poll_wait(&events
, -1);
134 DBG3("[ht-thread] Returning from poll on %d fds.",
135 LTTNG_POLL_GETNB(&events
));
139 * Restart interrupted system call.
141 if (errno
== EINTR
) {
148 for (i
= 0; i
< nb_fd
; i
++) {
151 health_code_update();
153 /* Fetch once the poll data */
154 revents
= LTTNG_POLL_GETEV(&events
, i
);
155 pollfd
= LTTNG_POLL_GETFD(&events
, i
);
158 /* No activity for this FD (poll implementation). */
162 if (pollfd
!= ht_cleanup_pipe
[0]) {
166 if (revents
& LPOLLIN
) {
167 /* Get socket from dispatch thread. */
168 size_ret
= lttng_read(ht_cleanup_pipe
[0], &ht
,
170 if (size_ret
< sizeof(ht
)) {
171 PERROR("ht cleanup notify pipe");
174 health_code_update();
176 * The whole point of this thread is to call
177 * lttng_ht_destroy from a context that is NOT:
178 * 1) a read-side RCU lock,
179 * 2) a call_rcu thread.
181 lttng_ht_destroy(ht
);
183 health_code_update();
184 } else if (revents
& (LPOLLERR
| LPOLLHUP
| LPOLLRDHUP
)) {
185 ERR("ht cleanup pipe error");
188 ERR("Unexpected poll events %u for sock %d", revents
, pollfd
);
193 for (i
= 0; i
< nb_fd
; i
++) {
194 health_code_update();
196 /* Fetch once the poll data */
197 revents
= LTTNG_POLL_GETEV(&events
, i
);
198 pollfd
= LTTNG_POLL_GETFD(&events
, i
);
201 /* No activity for this FD (poll implementation). */
205 if (pollfd
== ht_cleanup_pipe
[0]) {
209 /* Thread quit pipe has been closed. Killing thread. */
210 ret
= check_quit_pipe(pollfd
, revents
);
213 DBG("[ht-cleanup] quit.");
221 lttng_poll_clean(&events
);
224 DBG("[ht-cleanup] Thread terminates.");
227 ERR("Health error occurred in %s", __func__
);
229 health_unregister(health_sessiond
);
230 rcu_thread_offline();
231 rcu_unregister_thread();
235 static bool shutdown_ht_cleanup_thread(void *data
)
239 ret
= notify_thread_pipe(ht_cleanup_quit_pipe
[1]);
241 ERR("write error on ht_cleanup quit pipe");
248 struct lttng_thread
*launch_ht_cleanup_thread(void)
251 struct lttng_thread
*thread
;
253 ret
= init_pipe(ht_cleanup_pipe
);
258 ret
= init_pipe(ht_cleanup_quit_pipe
);
263 thread
= lttng_thread_create("HT cleanup",
265 shutdown_ht_cleanup_thread
,
266 cleanup_ht_cleanup_thread
,
274 cleanup_ht_cleanup_thread(NULL
);