Fix: per-uid flush and ust registry locking
[lttng-tools.git] / src / bin / lttng-sessiond / ht-cleanup.c
1 /*
2 * Copyright (C) 2013 - Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
3 *
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.
7 *
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.
12 *
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.
16 */
17
18 #define _GNU_SOURCE
19 #define _LGPL_SOURCE
20 #include <assert.h>
21
22 #include <common/hashtable/hashtable.h>
23 #include <common/common.h>
24 #include <common/utils.h>
25
26 #include "lttng-sessiond.h"
27 #include "health-sessiond.h"
28 #include "testpoint.h"
29
30 void *thread_ht_cleanup(void *data)
31 {
32 int ret, i, pollfd, err = -1;
33 ssize_t size_ret;
34 uint32_t revents, nb_fd;
35 struct lttng_poll_event events;
36
37 DBG("[ht-thread] startup.");
38
39 rcu_register_thread();
40 rcu_thread_online();
41
42 health_register(health_sessiond, HEALTH_SESSIOND_TYPE_HT_CLEANUP);
43
44 if (testpoint(sessiond_thread_ht_cleanup)) {
45 DBG("[ht-thread] testpoint.");
46 goto error_testpoint;
47 }
48
49 health_code_update();
50
51 ret = sessiond_set_ht_cleanup_thread_pollset(&events, 2);
52 if (ret < 0) {
53 DBG("[ht-thread] sessiond_set_ht_cleanup_thread_pollset error %d.", ret);
54 goto error_poll_create;
55 }
56
57 /* Add pipe to the pollset. */
58 ret = lttng_poll_add(&events, ht_cleanup_pipe[0], LPOLLIN | LPOLLERR);
59 if (ret < 0) {
60 DBG("[ht-thread] lttng_poll_add error %d.", ret);
61 goto error;
62 }
63
64 health_code_update();
65
66 while (1) {
67 DBG3("[ht-thread] Polling.");
68
69 /* Inifinite blocking call, waiting for transmission */
70 restart:
71 health_poll_entry();
72 ret = lttng_poll_wait(&events, -1);
73 DBG3("[ht-thread] Returning from poll on %d fds.",
74 LTTNG_POLL_GETNB(&events));
75 health_poll_exit();
76 if (ret < 0) {
77 /*
78 * Restart interrupted system call.
79 */
80 if (errno == EINTR) {
81 goto restart;
82 }
83 goto error;
84 }
85
86 nb_fd = ret;
87
88 for (i = 0; i < nb_fd; i++) {
89 struct lttng_ht *ht;
90
91 health_code_update();
92
93 /* Fetch once the poll data */
94 revents = LTTNG_POLL_GETEV(&events, i);
95 pollfd = LTTNG_POLL_GETFD(&events, i);
96
97 if (!revents) {
98 /* No activity for this FD (poll implementation). */
99 continue;
100 }
101
102 if (pollfd != ht_cleanup_pipe[0]) {
103 continue;
104 }
105
106 if (revents & (LPOLLERR | LPOLLHUP | LPOLLRDHUP)) {
107 ERR("ht cleanup pipe error");
108 goto error;
109 } else if (!(revents & LPOLLIN)) {
110 /* No POLLIN and not a catched error, stop the thread. */
111 ERR("ht cleanup failed. revent: %u", revents);
112 goto error;
113 }
114
115 /* Get socket from dispatch thread. */
116 size_ret = lttng_read(ht_cleanup_pipe[0], &ht,
117 sizeof(ht));
118 if (size_ret < sizeof(ht)) {
119 PERROR("ht cleanup notify pipe");
120 goto error;
121 }
122 health_code_update();
123 /*
124 * The whole point of this thread is to call
125 * lttng_ht_destroy from a context that is NOT:
126 * 1) a read-side RCU lock,
127 * 2) a call_rcu thread.
128 */
129 lttng_ht_destroy(ht);
130
131 health_code_update();
132 }
133
134 for (i = 0; i < nb_fd; i++) {
135 health_code_update();
136
137 /* Fetch once the poll data */
138 revents = LTTNG_POLL_GETEV(&events, i);
139 pollfd = LTTNG_POLL_GETFD(&events, i);
140
141 if (!revents) {
142 /* No activity for this FD (poll implementation). */
143 continue;
144 }
145
146 if (pollfd == ht_cleanup_pipe[0]) {
147 continue;
148 }
149
150 /* Thread quit pipe has been closed. Killing thread. */
151 ret = sessiond_check_ht_cleanup_quit(pollfd, revents);
152 if (ret) {
153 err = 0;
154 DBG("[ht-cleanup] quit.");
155 goto exit;
156 }
157 }
158 }
159
160 exit:
161 error:
162 lttng_poll_clean(&events);
163 error_poll_create:
164 error_testpoint:
165 DBG("[ht-cleanup] Thread terminates.");
166 if (err) {
167 health_error();
168 ERR("Health error occurred in %s", __func__);
169 }
170 health_unregister(health_sessiond);
171 rcu_thread_offline();
172 rcu_unregister_thread();
173 return NULL;
174 }
This page took 0.033669 seconds and 4 git commands to generate.