Commit | Line | Data |
---|---|---|
a8c3ad3e | 1 | /* |
ab5be9fa | 2 | * Copyright (C) 2018 Mathieu Desnoyers <mathieu.desnoyers@efficios.com> |
159b042f | 3 | * Copyright (C) 2020 Jérémie Galarneau <jeremie.galarneau@efficios.com> |
a8c3ad3e | 4 | * |
ab5be9fa | 5 | * SPDX-License-Identifier: GPL-2.0-only |
a8c3ad3e | 6 | * |
a8c3ad3e MD |
7 | */ |
8 | ||
c9e313bc SM |
9 | #include "common/dynamic-array.hpp" |
10 | #include "common/macros.hpp" | |
28ab034a | 11 | #include "lttng/tracker.h" |
a8c3ad3e | 12 | #define _LGPL_SOURCE |
c9e313bc | 13 | #include "tracker.hpp" |
28ab034a | 14 | |
c9e313bc SM |
15 | #include <common/defaults.hpp> |
16 | #include <common/error.hpp> | |
17 | #include <common/hashtable/hashtable.hpp> | |
18 | #include <common/hashtable/utils.hpp> | |
19 | #include <common/tracker.hpp> | |
56047f5a | 20 | #include <common/urcu.hpp> |
28ab034a | 21 | |
a8c3ad3e MD |
22 | #include <lttng/lttng-error.h> |
23 | ||
28ab034a JG |
24 | #include <grp.h> |
25 | #include <pwd.h> | |
26 | #include <sys/types.h> | |
27 | #include <unistd.h> | |
28 | #include <urcu.h> | |
29 | #include <urcu/list.h> | |
30 | #include <urcu/rculfhash.h> | |
31 | ||
f1494934 JG |
32 | struct process_attr_tracker { |
33 | enum lttng_tracking_policy policy; | |
34 | struct cds_lfht *inclusion_set_ht; | |
35 | }; | |
36 | ||
37 | namespace { | |
159b042f JG |
38 | struct process_attr_tracker_value_node { |
39 | struct process_attr_value *value; | |
40 | struct cds_lfht_node inclusion_set_ht_node; | |
41 | struct rcu_head rcu_head; | |
42 | }; | |
f1494934 | 43 | } /* namespace */ |
159b042f JG |
44 | |
45 | static void process_attr_tracker_value_node_rcu_free(struct rcu_head *rcu_head) | |
46 | { | |
28ab034a JG |
47 | struct process_attr_tracker_value_node *node = |
48 | lttng::utils::container_of(rcu_head, &process_attr_tracker_value_node::rcu_head); | |
159b042f JG |
49 | |
50 | free(node); | |
51 | } | |
52 | ||
cd9adb8b | 53 | struct process_attr_tracker *process_attr_tracker_create() |
a8c3ad3e | 54 | { |
159b042f | 55 | struct process_attr_tracker *tracker; |
a8c3ad3e | 56 | |
64803277 | 57 | tracker = zmalloc<process_attr_tracker>(); |
159b042f | 58 | if (!tracker) { |
cd9adb8b | 59 | return nullptr; |
a8c3ad3e | 60 | } |
159b042f | 61 | |
28ab034a | 62 | (void) process_attr_tracker_set_tracking_policy(tracker, LTTNG_TRACKING_POLICY_INCLUDE_ALL); |
159b042f | 63 | |
28ab034a | 64 | tracker->inclusion_set_ht = cds_lfht_new( |
cd9adb8b | 65 | DEFAULT_HT_SIZE, 1, 0, CDS_LFHT_AUTO_RESIZE | CDS_LFHT_ACCOUNTING, nullptr); |
159b042f | 66 | if (!tracker->inclusion_set_ht) { |
a8c3ad3e MD |
67 | goto error; |
68 | } | |
a8c3ad3e | 69 | |
159b042f | 70 | return tracker; |
a8c3ad3e | 71 | error: |
159b042f | 72 | process_attr_tracker_destroy(tracker); |
cd9adb8b | 73 | return nullptr; |
a8c3ad3e MD |
74 | } |
75 | ||
28ab034a JG |
76 | static void |
77 | process_attr_tracker_remove_value_node(struct process_attr_tracker *tracker, | |
78 | struct process_attr_tracker_value_node *value_node) | |
a8c3ad3e | 79 | { |
28ab034a | 80 | cds_lfht_del(tracker->inclusion_set_ht, &value_node->inclusion_set_ht_node); |
159b042f | 81 | process_attr_value_destroy(value_node->value); |
28ab034a | 82 | call_rcu(&value_node->rcu_head, process_attr_tracker_value_node_rcu_free); |
a8c3ad3e MD |
83 | } |
84 | ||
28ab034a | 85 | static void process_attr_tracker_clear_inclusion_set(struct process_attr_tracker *tracker) |
a8c3ad3e | 86 | { |
159b042f JG |
87 | if (!tracker->inclusion_set_ht) { |
88 | return; | |
89 | } | |
a8c3ad3e | 90 | |
706b812c JG |
91 | for (auto *value_node : lttng::urcu::lfht_iteration_adapter< |
92 | process_attr_tracker_value_node, | |
93 | decltype(process_attr_tracker_value_node::inclusion_set_ht_node), | |
94 | &process_attr_tracker_value_node::inclusion_set_ht_node>( | |
95 | *tracker->inclusion_set_ht)) { | |
96 | process_attr_tracker_remove_value_node(tracker, value_node); | |
a8c3ad3e | 97 | } |
56047f5a | 98 | |
706b812c | 99 | const auto ret = cds_lfht_destroy(tracker->inclusion_set_ht, nullptr); |
a0377dfe | 100 | LTTNG_ASSERT(ret == 0); |
cd9adb8b | 101 | tracker->inclusion_set_ht = nullptr; |
a8c3ad3e MD |
102 | } |
103 | ||
28ab034a | 104 | static int process_attr_tracker_create_inclusion_set(struct process_attr_tracker *tracker) |
a8c3ad3e | 105 | { |
a0377dfe | 106 | LTTNG_ASSERT(!tracker->inclusion_set_ht); |
28ab034a | 107 | tracker->inclusion_set_ht = cds_lfht_new( |
cd9adb8b | 108 | DEFAULT_HT_SIZE, 1, 0, CDS_LFHT_AUTO_RESIZE | CDS_LFHT_ACCOUNTING, nullptr); |
159b042f | 109 | return tracker->inclusion_set_ht ? 0 : -1; |
a8c3ad3e MD |
110 | } |
111 | ||
159b042f | 112 | void process_attr_tracker_destroy(struct process_attr_tracker *tracker) |
a8c3ad3e | 113 | { |
159b042f JG |
114 | if (!tracker) { |
115 | return; | |
116 | } | |
a8c3ad3e | 117 | |
159b042f JG |
118 | process_attr_tracker_clear_inclusion_set(tracker); |
119 | free(tracker); | |
a8c3ad3e MD |
120 | } |
121 | ||
28ab034a JG |
122 | enum lttng_tracking_policy |
123 | process_attr_tracker_get_tracking_policy(const struct process_attr_tracker *tracker) | |
a8c3ad3e | 124 | { |
159b042f | 125 | return tracker->policy; |
a8c3ad3e MD |
126 | } |
127 | ||
28ab034a JG |
128 | int process_attr_tracker_set_tracking_policy(struct process_attr_tracker *tracker, |
129 | enum lttng_tracking_policy tracking_policy) | |
a8c3ad3e | 130 | { |
159b042f | 131 | int ret = 0; |
a8c3ad3e | 132 | |
159b042f JG |
133 | if (tracker->policy == tracking_policy) { |
134 | goto end; | |
a8c3ad3e | 135 | } |
2d97a006 | 136 | |
159b042f JG |
137 | process_attr_tracker_clear_inclusion_set(tracker); |
138 | ret = process_attr_tracker_create_inclusion_set(tracker); | |
139 | if (ret) { | |
140 | goto end; | |
a8c3ad3e | 141 | } |
159b042f JG |
142 | tracker->policy = tracking_policy; |
143 | end: | |
a8c3ad3e MD |
144 | return ret; |
145 | } | |
146 | ||
28ab034a | 147 | static int match_inclusion_set_value(struct cds_lfht_node *node, const void *key) |
a8c3ad3e | 148 | { |
7966af57 | 149 | const struct process_attr_value *value_key = (process_attr_value *) key; |
28ab034a JG |
150 | const struct process_attr_tracker_value_node *value_node = caa_container_of( |
151 | node, struct process_attr_tracker_value_node, inclusion_set_ht_node); | |
a8c3ad3e | 152 | |
159b042f JG |
153 | return process_attr_tracker_value_equal(value_node->value, value_key); |
154 | } | |
a8c3ad3e | 155 | |
28ab034a JG |
156 | static struct process_attr_tracker_value_node * |
157 | process_attr_tracker_lookup(const struct process_attr_tracker *tracker, | |
158 | const struct process_attr_value *value) | |
159b042f JG |
159 | { |
160 | struct cds_lfht_iter iter; | |
161 | struct cds_lfht_node *node; | |
a8c3ad3e | 162 | |
a0377dfe | 163 | LTTNG_ASSERT(tracker->policy == LTTNG_TRACKING_POLICY_INCLUDE_SET); |
159b042f | 164 | |
07c4863f | 165 | const lttng::urcu::read_lock_guard read_lock; |
159b042f JG |
166 | cds_lfht_lookup(tracker->inclusion_set_ht, |
167 | process_attr_value_hash(value), | |
28ab034a JG |
168 | match_inclusion_set_value, |
169 | value, | |
170 | &iter); | |
159b042f | 171 | node = cds_lfht_iter_get_node(&iter); |
159b042f | 172 | |
28ab034a JG |
173 | return node ? lttng::utils::container_of( |
174 | node, &process_attr_tracker_value_node::inclusion_set_ht_node) : | |
cd9adb8b | 175 | nullptr; |
a8c3ad3e MD |
176 | } |
177 | ||
159b042f | 178 | /* Protected by session mutex held by caller. */ |
28ab034a JG |
179 | enum process_attr_tracker_status |
180 | process_attr_tracker_inclusion_set_add_value(struct process_attr_tracker *tracker, | |
181 | const struct process_attr_value *value) | |
a8c3ad3e | 182 | { |
28ab034a | 183 | enum process_attr_tracker_status status = PROCESS_ATTR_TRACKER_STATUS_OK; |
cd9adb8b JG |
184 | struct process_attr_value *value_copy = nullptr; |
185 | struct process_attr_tracker_value_node *value_node = nullptr; | |
b7e1aba3 | 186 | |
07c4863f | 187 | const lttng::urcu::read_lock_guard read_lock; |
159b042f JG |
188 | if (tracker->policy != LTTNG_TRACKING_POLICY_INCLUDE_SET) { |
189 | status = PROCESS_ATTR_TRACKER_STATUS_INVALID_TRACKING_POLICY; | |
190 | goto end; | |
a8c3ad3e | 191 | } |
a8c3ad3e | 192 | |
159b042f JG |
193 | if (process_attr_tracker_lookup(tracker, value)) { |
194 | status = PROCESS_ATTR_TRACKER_STATUS_EXISTS; | |
a8c3ad3e MD |
195 | goto end; |
196 | } | |
a8c3ad3e | 197 | |
64803277 | 198 | value_node = zmalloc<process_attr_tracker_value_node>(); |
159b042f JG |
199 | if (!value_node) { |
200 | status = PROCESS_ATTR_TRACKER_STATUS_ERROR; | |
201 | goto end; | |
a8c3ad3e | 202 | } |
159b042f JG |
203 | |
204 | value_copy = process_attr_value_copy(value); | |
205 | if (!value_copy) { | |
206 | status = PROCESS_ATTR_TRACKER_STATUS_ERROR; | |
a8c3ad3e MD |
207 | goto end; |
208 | } | |
159b042f JG |
209 | |
210 | value_node->value = value_copy; | |
211 | cds_lfht_add(tracker->inclusion_set_ht, | |
28ab034a JG |
212 | process_attr_value_hash(value_copy), |
213 | &value_node->inclusion_set_ht_node); | |
cd9adb8b JG |
214 | value_copy = nullptr; |
215 | value_node = nullptr; | |
159b042f JG |
216 | end: |
217 | if (value_copy) { | |
218 | process_attr_value_destroy(value_copy); | |
a8c3ad3e | 219 | } |
159b042f JG |
220 | if (value_node) { |
221 | free(value_node); | |
a8c3ad3e | 222 | } |
159b042f | 223 | return status; |
a8c3ad3e MD |
224 | } |
225 | ||
159b042f JG |
226 | /* Protected by session mutex held by caller. */ |
227 | enum process_attr_tracker_status | |
28ab034a JG |
228 | process_attr_tracker_inclusion_set_remove_value(struct process_attr_tracker *tracker, |
229 | const struct process_attr_value *value) | |
a8c3ad3e | 230 | { |
159b042f | 231 | struct process_attr_tracker_value_node *value_node; |
28ab034a | 232 | enum process_attr_tracker_status status = PROCESS_ATTR_TRACKER_STATUS_OK; |
a8c3ad3e | 233 | |
07c4863f | 234 | const lttng::urcu::read_lock_guard read_lock; |
159b042f JG |
235 | if (tracker->policy != LTTNG_TRACKING_POLICY_INCLUDE_SET) { |
236 | status = PROCESS_ATTR_TRACKER_STATUS_INVALID_TRACKING_POLICY; | |
237 | goto end; | |
238 | } | |
a7a533cd | 239 | |
159b042f JG |
240 | value_node = process_attr_tracker_lookup(tracker, value); |
241 | if (!value_node) { | |
242 | status = PROCESS_ATTR_TRACKER_STATUS_MISSING; | |
243 | goto end; | |
a8c3ad3e | 244 | } |
a7a533cd | 245 | |
159b042f | 246 | process_attr_tracker_remove_value_node(tracker, value_node); |
a8c3ad3e | 247 | end: |
159b042f | 248 | return status; |
a8c3ad3e MD |
249 | } |
250 | ||
28ab034a JG |
251 | enum process_attr_tracker_status |
252 | process_attr_tracker_get_inclusion_set(const struct process_attr_tracker *tracker, | |
253 | struct lttng_process_attr_values **_values) | |
a8c3ad3e | 254 | { |
28ab034a | 255 | enum process_attr_tracker_status status = PROCESS_ATTR_TRACKER_STATUS_OK; |
159b042f | 256 | struct lttng_process_attr_values *values; |
cd9adb8b | 257 | struct process_attr_value *new_value = nullptr; |
159b042f JG |
258 | |
259 | values = lttng_process_attr_values_create(); | |
260 | if (!values) { | |
261 | status = PROCESS_ATTR_TRACKER_STATUS_ERROR; | |
262 | goto error; | |
e283e4a0 | 263 | } |
a7a533cd | 264 | |
159b042f JG |
265 | if (tracker->policy != LTTNG_TRACKING_POLICY_INCLUDE_SET) { |
266 | status = PROCESS_ATTR_TRACKER_STATUS_INVALID_TRACKING_POLICY; | |
267 | goto error; | |
a8c3ad3e | 268 | } |
a7a533cd | 269 | |
706b812c JG |
270 | for (auto *value_node : lttng::urcu::lfht_iteration_adapter< |
271 | process_attr_tracker_value_node, | |
272 | decltype(process_attr_tracker_value_node::inclusion_set_ht_node), | |
273 | &process_attr_tracker_value_node::inclusion_set_ht_node>( | |
274 | *tracker->inclusion_set_ht)) { | |
275 | int ret; | |
276 | ||
277 | new_value = process_attr_value_copy(value_node->value); | |
278 | if (!new_value) { | |
279 | status = PROCESS_ATTR_TRACKER_STATUS_ERROR; | |
280 | goto error_unlock; | |
281 | } | |
159b042f | 282 | |
706b812c JG |
283 | ret = lttng_dynamic_pointer_array_add_pointer(&values->array, new_value); |
284 | if (ret) { | |
285 | status = PROCESS_ATTR_TRACKER_STATUS_ERROR; | |
286 | goto error_unlock; | |
56047f5a | 287 | } |
706b812c JG |
288 | |
289 | new_value = nullptr; | |
a8c3ad3e | 290 | } |
56047f5a | 291 | |
159b042f JG |
292 | *_values = values; |
293 | return status; | |
294 | error_unlock: | |
159b042f JG |
295 | error: |
296 | lttng_process_attr_values_destroy(values); | |
297 | process_attr_value_destroy(new_value); | |
298 | return status; | |
a8c3ad3e | 299 | } |