projects / lttng-modules.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Refactoring: uprobes: combine common code between recorder and notifier
[lttng-modules.git] / src / lttng-events.c
CommitLineData
b7cdc182 1/* SPDX-License-Identifier: (GPL-2.0-only or LGPL-2.1-only)
9f36eaed 2 *
a90917c3 3 * lttng-events.c
4e3c1b9b 4 *
4e3c1b9b 5 * Holds LTTng per-session event registry.
17baffe2 6 *
886d51a3 7 * Copyright (C) 2010-2012 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
4e3c1b9b
MD		 8 */
9
389d7070
MD		 10/*
11 * This page_alloc.h wrapper needs to be included before gfpflags.h because it
12 * overrides a function with a define.
13 */
14#include "wrapper/page_alloc.h"
15
4e3c1b9b 16#include <linux/module.h>
c0e31d2e
MD		 17#include <linux/mutex.h>
18#include <linux/sched.h>
11b5a3c2 19#include <linux/slab.h>
c099397a 20#include <linux/jiffies.h>
99dc9597 21#include <linux/utsname.h>
abc0446a 22#include <linux/err.h>
7e6f9ef6
MD		 23#include <linux/seq_file.h>
24#include <linux/file.h>
25#include <linux/anon_inodes.h>
241ae9a8 26#include <wrapper/file.h>
8c6e7f13 27#include <linux/uaccess.h>
a606b6e8 28#include <linux/vmalloc.h>
c88f762e 29#include <linux/dmi.h>
3c997079 30
c190d76e 31#include <wrapper/compiler_attributes.h>
a7008254 32#include <wrapper/uuid.h>
263b6c88 33#include <wrapper/vmalloc.h> /* for wrapper_vmalloc_sync_mappings() */
241ae9a8
MD		 34#include <wrapper/random.h>
35#include <wrapper/tracepoint.h>
36#include <wrapper/list.h>
cbc19040 37#include <wrapper/types.h>
2df37e95
MD		 38#include <lttng/kernel-version.h>
39#include <lttng/events.h>
92bc1e23 40#include <lttng/events-internal.h>
0b365677 41#include <lttng/lttng-bytecode.h>
2df37e95 42#include <lttng/tracer.h>
21f58fb7 43#include <lttng/event-notifier-notification.h>
2df37e95
MD		 44#include <lttng/abi-old.h>
45#include <lttng/endian.h>
46#include <lttng/string-utils.h>
4a11ef1c 47#include <lttng/utils.h>
24591303
MD		 48#include <ringbuffer/backend.h>
49#include <ringbuffer/frontend.h>
3636085f 50#include <wrapper/time.h>
4e3c1b9b 51
d83004aa
JD		 52#define METADATA_CACHE_DEFAULT_SIZE 4096
53
4e3c1b9b 54static LIST_HEAD(sessions);
750b05f2 55static LIST_HEAD(event_notifier_groups);
a90917c3 56static LIST_HEAD(lttng_transport_list);
a101fa10 57static LIST_HEAD(lttng_counter_transport_list);
d83004aa
JD		 58/*
59 * Protect the sessions and metadata caches.
60 */
4e3c1b9b 61static DEFINE_MUTEX(sessions_mutex);
a67ba386
MD		 62static struct kmem_cache *event_recorder_cache;
63static struct kmem_cache *event_recorder_private_cache;
dffef45d 64static struct kmem_cache *event_notifier_cache;
a67ba386 65static struct kmem_cache *event_notifier_private_cache;
4e3c1b9b 66
8cdc1a81
MD		 67static void lttng_session_lazy_sync_event_enablers(struct lttng_kernel_session *session);
68static void lttng_session_sync_event_enablers(struct lttng_kernel_session *session);
b01155ba 69static void lttng_event_notifier_group_sync_enablers(struct lttng_event_notifier_group *event_notifier_group);
7b25ab0e 70static void lttng_event_enabler_sync(struct lttng_event_enabler_common *event_enabler);
3c997079 71
0648898e 72static void _lttng_event_destroy(struct lttng_kernel_event_common *event);
f7d06400 73static void _lttng_channel_destroy(struct lttng_kernel_channel_buffer *chan);
96c59af1 74static int _lttng_event_recorder_unregister(struct lttng_kernel_event_recorder *event);
a67ba386 75static int _lttng_event_notifier_unregister(struct lttng_kernel_event_notifier *event_notifier);
c099397a 76static
8cdc1a81 77int _lttng_event_metadata_statedump(struct lttng_kernel_session *session,
f7d06400 78 struct lttng_kernel_channel_buffer *chan,
a67ba386 79 struct lttng_kernel_event_recorder *event);
c099397a 80static
8cdc1a81 81int _lttng_session_metadata_statedump(struct lttng_kernel_session *session);
d83004aa
JD		 82static
83void _lttng_metadata_channel_hangup(struct lttng_metadata_stream *stream);
f513b2bf 84static
8cdc1a81 85int _lttng_type_statedump(struct lttng_kernel_session *session,
437d5aa5
MD		 86 const struct lttng_kernel_type_common *type,
87 enum lttng_kernel_string_encoding parent_encoding,
ceabb767
MD		 88 size_t nesting);
89static
8cdc1a81 90int _lttng_field_statedump(struct lttng_kernel_session *session,
437d5aa5 91 const struct lttng_kernel_event_field *field,
51ef4536 92 size_t nesting, const char **prev_field_name_p);
c099397a 93
c099397a 94void synchronize_trace(void)
abcca994 95{
05355f0b
MJ		 96#if (LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(5,1,0) || \
97 LTTNG_RHEL_KERNEL_RANGE(4,18,0,193,0,0, 4,19,0,0,0,0))
92da05ce
MJ		 98 synchronize_rcu();
99#else
abcca994 100 synchronize_sched();
92da05ce
MJ		 101#endif
102
5f4c791e 103#if (LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(3,4,0))
b8590f40
MD		 104#ifdef CONFIG_PREEMPT_RT_FULL
105 synchronize_rcu();
106#endif
5f4c791e 107#else /* (LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(3,4,0)) */
abcca994
MD		 108#ifdef CONFIG_PREEMPT_RT
109 synchronize_rcu();
110#endif
5f4c791e 111#endif /* (LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(3,4,0)) */
abcca994
MD		 112}
113
3c997079
MD		 114void lttng_lock_sessions(void)
115{
116 mutex_lock(&sessions_mutex);
117}
118
119void lttng_unlock_sessions(void)
120{
121 mutex_unlock(&sessions_mutex);
122}
123
9fe593d2
FD		 124static struct lttng_transport *lttng_transport_find(const char *name)
125{
126 struct lttng_transport *transport;
127
128 list_for_each_entry(transport, &lttng_transport_list, node) {
129 if (!strcmp(transport->name, name))
130 return transport;
131 }
132 return NULL;
133}
134
3c997079
MD		 135/*
136 * Called with sessions lock held.
137 */
138int lttng_session_active(void)
139{
8cdc1a81 140 struct lttng_kernel_session_private *iter;
3c997079
MD		 141
142 list_for_each_entry(iter, &sessions, list) {
8cdc1a81 143 if (iter->pub->active)
3c997079
MD		 144 return 1;
145 }
146 return 0;
147}
148
8cdc1a81 149struct lttng_kernel_session *lttng_session_create(void)
4e3c1b9b 150{
8cdc1a81
MD		 151 struct lttng_kernel_session *session;
152 struct lttng_kernel_session_private *session_priv;
d83004aa 153 struct lttng_metadata_cache *metadata_cache;
3c997079 154 int i;
4e3c1b9b
MD		 155
156 mutex_lock(&sessions_mutex);
8cdc1a81 157 session = lttng_kvzalloc(sizeof(*session), GFP_KERNEL);
4e3c1b9b 158 if (!session)
d83004aa 159 goto err;
8cdc1a81
MD		 160 session_priv = lttng_kvzalloc(sizeof(*session_priv), GFP_KERNEL);
161 if (!session_priv)
162 goto err_free_session;
163 session->priv = session_priv;
164 session_priv->pub = session;
165
166 INIT_LIST_HEAD(&session_priv->chan);
167 INIT_LIST_HEAD(&session_priv->events);
168 lttng_guid_gen(&session_priv->uuid);
d83004aa
JD		 169
170 metadata_cache = kzalloc(sizeof(struct lttng_metadata_cache),
171 GFP_KERNEL);
172 if (!metadata_cache)
8cdc1a81 173 goto err_free_session_private;
0ca42eb7 174 metadata_cache->data = vzalloc(METADATA_CACHE_DEFAULT_SIZE);
d83004aa
JD		 175 if (!metadata_cache->data)
176 goto err_free_cache;
177 metadata_cache->cache_alloc = METADATA_CACHE_DEFAULT_SIZE;
178 kref_init(&metadata_cache->refcount);
92d9f5e6 179 mutex_init(&metadata_cache->lock);
8cdc1a81 180 session_priv->metadata_cache = metadata_cache;
d83004aa 181 INIT_LIST_HEAD(&metadata_cache->metadata_stream);
8cdc1a81 182 memcpy(&metadata_cache->uuid, &session_priv->uuid,
a36580d5 183 sizeof(metadata_cache->uuid));
8cdc1a81 184 INIT_LIST_HEAD(&session_priv->enablers_head);
3c997079 185 for (i = 0; i < LTTNG_EVENT_HT_SIZE; i++)
8cdc1a81
MD		 186 INIT_HLIST_HEAD(&session_priv->events_ht.table[i]);
187 list_add(&session_priv->list, &sessions);
188
8c0393c3
MD		 189 if (lttng_id_tracker_init(&session->pid_tracker, session, TRACKER_PID))
190 goto tracker_alloc_error;
191 if (lttng_id_tracker_init(&session->vpid_tracker, session, TRACKER_VPID))
192 goto tracker_alloc_error;
193 if (lttng_id_tracker_init(&session->uid_tracker, session, TRACKER_UID))
194 goto tracker_alloc_error;
195 if (lttng_id_tracker_init(&session->vuid_tracker, session, TRACKER_VUID))
196 goto tracker_alloc_error;
197 if (lttng_id_tracker_init(&session->gid_tracker, session, TRACKER_GID))
198 goto tracker_alloc_error;
199 if (lttng_id_tracker_init(&session->vgid_tracker, session, TRACKER_VGID))
200 goto tracker_alloc_error;
201
4e3c1b9b 202 mutex_unlock(&sessions_mutex);
8cdc1a81 203
4e3c1b9b 204 return session;
d83004aa 205
8c0393c3
MD		 206tracker_alloc_error:
207 lttng_id_tracker_fini(&session->pid_tracker);
208 lttng_id_tracker_fini(&session->vpid_tracker);
209 lttng_id_tracker_fini(&session->uid_tracker);
210 lttng_id_tracker_fini(&session->vuid_tracker);
211 lttng_id_tracker_fini(&session->gid_tracker);
212 lttng_id_tracker_fini(&session->vgid_tracker);
d83004aa
JD		 213err_free_cache:
214 kfree(metadata_cache);
8cdc1a81
MD		 215err_free_session_private:
216 lttng_kvfree(session_priv);
d83004aa 217err_free_session:
48f5e0b5 218 lttng_kvfree(session);
d83004aa
JD		 219err:
220 mutex_unlock(&sessions_mutex);
221 return NULL;
222}
223
99f52fcc
FD		 224static
225struct lttng_counter_transport *lttng_counter_transport_find(const char *name)
226{
227 struct lttng_counter_transport *transport;
228
229 list_for_each_entry(transport, &lttng_counter_transport_list, node) {
230 if (!strcmp(transport->name, name))
231 return transport;
232 }
233 return NULL;
234}
235
236struct lttng_counter *lttng_kernel_counter_create(
237 const char *counter_transport_name,
238 size_t number_dimensions, const size_t *dimensions_sizes)
239{
240 struct lttng_counter *counter = NULL;
241 struct lttng_counter_transport *counter_transport = NULL;
242
243 counter_transport = lttng_counter_transport_find(counter_transport_name);
244 if (!counter_transport) {
245 printk(KERN_WARNING "LTTng: counter transport %s not found.\n",
246 counter_transport_name);
247 goto notransport;
248 }
249 if (!try_module_get(counter_transport->owner)) {
250 printk(KERN_WARNING "LTTng: Can't lock counter transport module.\n");
251 goto notransport;
252 }
253
254 counter = lttng_kvzalloc(sizeof(struct lttng_counter), GFP_KERNEL);
255 if (!counter)
256 goto nomem;
257
258 /* Create event notifier error counter. */
259 counter->ops = &counter_transport->ops;
260 counter->transport = counter_transport;
261
262 counter->counter = counter->ops->counter_create(
263 number_dimensions, dimensions_sizes, 0);
264 if (!counter->counter) {
265 goto create_error;
266 }
267
268 return counter;
269
270create_error:
271 lttng_kvfree(counter);
272nomem:
273 if (counter_transport)
274 module_put(counter_transport->owner);
275notransport:
276 return NULL;
277}
278
750b05f2
FD		 279struct lttng_event_notifier_group *lttng_event_notifier_group_create(void)
280{
281 struct lttng_transport *transport = NULL;
282 struct lttng_event_notifier_group *event_notifier_group;
283 const char *transport_name = "relay-event-notifier";
284 size_t subbuf_size = 4096; //TODO
285 size_t num_subbuf = 16; //TODO
286 unsigned int switch_timer_interval = 0;
287 unsigned int read_timer_interval = 0;
dffef45d 288 int i;
750b05f2
FD		 289
290 mutex_lock(&sessions_mutex);
291
292 transport = lttng_transport_find(transport_name);
293 if (!transport) {
294 printk(KERN_WARNING "LTTng: transport %s not found\n",
295 transport_name);
296 goto notransport;
297 }
298 if (!try_module_get(transport->owner)) {
299 printk(KERN_WARNING "LTTng: Can't lock transport %s module.\n",
300 transport_name);
301 goto notransport;
302 }
303
304 event_notifier_group = lttng_kvzalloc(sizeof(struct lttng_event_notifier_group),
305 GFP_KERNEL);
306 if (!event_notifier_group)
307 goto nomem;
308
309 /*
310 * Initialize the ring buffer used to store event notifier
311 * notifications.
312 */
313 event_notifier_group->ops = &transport->ops;
4a399b76 314 event_notifier_group->chan = transport->ops.priv->channel_create(
750b05f2
FD		 315 transport_name, event_notifier_group, NULL,
316 subbuf_size, num_subbuf, switch_timer_interval,
317 read_timer_interval);
318 if (!event_notifier_group->chan)
319 goto create_error;
320
321 event_notifier_group->transport = transport;
dffef45d
FD		 322
323 INIT_LIST_HEAD(&event_notifier_group->enablers_head);
324 INIT_LIST_HEAD(&event_notifier_group->event_notifiers_head);
325 for (i = 0; i < LTTNG_EVENT_NOTIFIER_HT_SIZE; i++)
326 INIT_HLIST_HEAD(&event_notifier_group->event_notifiers_ht.table[i]);
327
750b05f2
FD		 328 list_add(&event_notifier_group->node, &event_notifier_groups);
329
330 mutex_unlock(&sessions_mutex);
331
332 return event_notifier_group;
333
334create_error:
335 lttng_kvfree(event_notifier_group);
336nomem:
337 if (transport)
338 module_put(transport->owner);
339notransport:
340 mutex_unlock(&sessions_mutex);
341 return NULL;
342}
343
d83004aa
JD		 344void metadata_cache_destroy(struct kref *kref)
345{
346 struct lttng_metadata_cache *cache =
347 container_of(kref, struct lttng_metadata_cache, refcount);
a606b6e8 348 vfree(cache->data);
d83004aa 349 kfree(cache);
4e3c1b9b
MD		 350}
351
8cdc1a81 352void lttng_session_destroy(struct lttng_kernel_session *session)
4e3c1b9b 353{
f7d06400 354 struct lttng_kernel_channel_buffer_private *chan_priv, *tmpchan_priv;
a67ba386 355 struct lttng_kernel_event_recorder_private *event_recorder_priv, *tmpevent_recorder_priv;
d83004aa 356 struct lttng_metadata_stream *metadata_stream;
44b1abdc 357 struct lttng_event_recorder_enabler *event_recorder_enabler, *tmp_event_recorder_enabler;
dda6a249 358 int ret;
4e3c1b9b
MD		 359
360 mutex_lock(&sessions_mutex);
a8f2d0c7 361 WRITE_ONCE(session->active, 0);
f7d06400 362 list_for_each_entry(chan_priv, &session->priv->chan, node) {
6053e75e 363 ret = lttng_syscalls_unregister_syscall_table(&chan_priv->parent.syscall_table);
1ec65de1
MD		 364 WARN_ON(ret);
365 }
8cdc1a81 366 list_for_each_entry(event_recorder_priv, &session->priv->events, node) {
96c59af1 367 ret = _lttng_event_recorder_unregister(event_recorder_priv->pub);
dda6a249
MD		 368 WARN_ON(ret);
369 }
abcca994 370 synchronize_trace(); /* Wait for in-flight events to complete */
f7d06400 371 list_for_each_entry(chan_priv, &session->priv->chan, node) {
6053e75e 372 ret = lttng_syscalls_destroy_syscall_table(&chan_priv->parent.syscall_table);
badfe9f5
MD		 373 WARN_ON(ret);
374 }
44b1abdc 375 list_for_each_entry_safe(event_recorder_enabler, tmp_event_recorder_enabler,
8cdc1a81 376 &session->priv->enablers_head, node)
44b1abdc 377 lttng_event_enabler_destroy(&event_recorder_enabler->parent);
8cdc1a81 378 list_for_each_entry_safe(event_recorder_priv, tmpevent_recorder_priv, &session->priv->events, node)
0648898e 379 _lttng_event_destroy(&event_recorder_priv->pub->parent);
f7d06400
MD		 380 list_for_each_entry_safe(chan_priv, tmpchan_priv, &session->priv->chan, node) {
381 BUG_ON(chan_priv->channel_type == METADATA_CHANNEL);
382 _lttng_channel_destroy(chan_priv->pub);
d83004aa 383 }
8cdc1a81
MD		 384 mutex_lock(&session->priv->metadata_cache->lock);
385 list_for_each_entry(metadata_stream, &session->priv->metadata_cache->metadata_stream, list)
d83004aa 386 _lttng_metadata_channel_hangup(metadata_stream);
8cdc1a81 387 mutex_unlock(&session->priv->metadata_cache->lock);
8c0393c3
MD		 388 lttng_id_tracker_fini(&session->pid_tracker);
389 lttng_id_tracker_fini(&session->vpid_tracker);
390 lttng_id_tracker_fini(&session->uid_tracker);
391 lttng_id_tracker_fini(&session->vuid_tracker);
392 lttng_id_tracker_fini(&session->gid_tracker);
393 lttng_id_tracker_fini(&session->vgid_tracker);
8cdc1a81
MD		 394 kref_put(&session->priv->metadata_cache->refcount, metadata_cache_destroy);
395 list_del(&session->priv->list);
4e3c1b9b 396 mutex_unlock(&sessions_mutex);
8cdc1a81 397 lttng_kvfree(session->priv);
48f5e0b5 398 lttng_kvfree(session);
4e3c1b9b
MD		 399}
400
dffef45d
FD		 401void lttng_event_notifier_group_destroy(
402 struct lttng_event_notifier_group *event_notifier_group)
750b05f2 403{
dffef45d 404 struct lttng_event_notifier_enabler *event_notifier_enabler, *tmp_event_notifier_enabler;
a67ba386 405 struct lttng_kernel_event_notifier_private *event_notifier_priv, *tmpevent_notifier_priv;
dffef45d
FD		 406 int ret;
407
750b05f2
FD		 408 if (!event_notifier_group)
409 return;
410
411 mutex_lock(&sessions_mutex);
dffef45d 412
6053e75e 413 ret = lttng_syscalls_unregister_syscall_table(&event_notifier_group->syscall_table);
8a8ac9a8
FD		 414 WARN_ON(ret);
415
a67ba386
MD		 416 list_for_each_entry_safe(event_notifier_priv, tmpevent_notifier_priv,
417 &event_notifier_group->event_notifiers_head, node) {
418 ret = _lttng_event_notifier_unregister(event_notifier_priv->pub);
dffef45d
FD		 419 WARN_ON(ret);
420 }
421
b01155ba
FD		 422 /* Wait for in-flight event notifier to complete */
423 synchronize_trace();
424
21f58fb7
FD		 425 irq_work_sync(&event_notifier_group->wakeup_pending);
426
6053e75e
MD		 427 ret = lttng_syscalls_destroy_syscall_table(&event_notifier_group->syscall_table);
428 WARN_ON(ret);
8a8ac9a8 429
dffef45d
FD		 430 list_for_each_entry_safe(event_notifier_enabler, tmp_event_notifier_enabler,
431 &event_notifier_group->enablers_head, node)
44b1abdc 432 lttng_event_enabler_destroy(&event_notifier_enabler->parent);
dffef45d 433
a67ba386
MD		 434 list_for_each_entry_safe(event_notifier_priv, tmpevent_notifier_priv,
435 &event_notifier_group->event_notifiers_head, node)
0648898e 436 _lttng_event_destroy(&event_notifier_priv->pub->parent);
dffef45d 437
99f52fcc
FD		 438 if (event_notifier_group->error_counter) {
439 struct lttng_counter *error_counter = event_notifier_group->error_counter;
ab04d7b1 440
99f52fcc
FD		 441 error_counter->ops->counter_destroy(error_counter->counter);
442 module_put(error_counter->transport->owner);
443 lttng_kvfree(error_counter);
444 event_notifier_group->error_counter = NULL;
445 }
446
4a399b76 447 event_notifier_group->ops->priv->channel_destroy(event_notifier_group->chan);
750b05f2
FD		 448 module_put(event_notifier_group->transport->owner);
449 list_del(&event_notifier_group->node);
dffef45d 450
750b05f2
FD		 451 mutex_unlock(&sessions_mutex);
452 lttng_kvfree(event_notifier_group);
453}
454
8cdc1a81 455int lttng_session_statedump(struct lttng_kernel_session *session)
601252cf
MD		 456{
457 int ret;
458
459 mutex_lock(&sessions_mutex);
460 ret = lttng_statedump_start(session);
461 mutex_unlock(&sessions_mutex);
462 return ret;
463}
464
8cdc1a81 465int lttng_session_enable(struct lttng_kernel_session *session)
c0e31d2e
MD		 466{
467 int ret = 0;
f7d06400 468 struct lttng_kernel_channel_buffer_private *chan_priv;
c0e31d2e
MD		 469
470 mutex_lock(&sessions_mutex);
471 if (session->active) {
472 ret = -EBUSY;
473 goto end;
474 }
c099397a 475
3c997079 476 /* Set transient enabler state to "enabled" */
8cdc1a81 477 session->priv->tstate = 1;
3c997079 478
4ef81e4a 479 /* We need to sync enablers with session before activation. */
b2bc0bc8 480 lttng_session_sync_event_enablers(session);
4ef81e4a 481
c099397a
MD		 482 /*
483 * Snapshot the number of events per channel to know the type of header
484 * we need to use.
485 */
f7d06400
MD		 486 list_for_each_entry(chan_priv, &session->priv->chan, node) {
487 if (chan_priv->header_type)
488 continue; /* don't change it if session stop/restart */
489 if (chan_priv->free_event_id < 31)
490 chan_priv->header_type = 1; /* compact */
c099397a 491 else
f7d06400 492 chan_priv->header_type = 2; /* large */
c099397a
MD		 493 }
494
64af2437 495 /* Clear each stream's quiescent state. */
f7d06400
MD		 496 list_for_each_entry(chan_priv, &session->priv->chan, node) {
497 if (chan_priv->channel_type != METADATA_CHANNEL)
498 lib_ring_buffer_clear_quiescent_channel(chan_priv->rb_chan);
09b93db9 499 }
64af2437 500
a8f2d0c7 501 WRITE_ONCE(session->active, 1);
8cdc1a81 502 WRITE_ONCE(session->priv->been_active, 1);
a90917c3 503 ret = _lttng_session_metadata_statedump(session);
c337ddc2 504 if (ret) {
a8f2d0c7 505 WRITE_ONCE(session->active, 0);
c337ddc2
MD		 506 goto end;
507 }
508 ret = lttng_statedump_start(session);
360f38ea 509 if (ret)
a8f2d0c7 510 WRITE_ONCE(session->active, 0);
c0e31d2e
MD		 511end:
512 mutex_unlock(&sessions_mutex);
11b5a3c2 513 return ret;
c0e31d2e
MD		 514}
515
8cdc1a81 516int lttng_session_disable(struct lttng_kernel_session *session)
c0e31d2e
MD		 517{
518 int ret = 0;
f7d06400 519 struct lttng_kernel_channel_buffer_private *chan_priv;
c0e31d2e
MD		 520
521 mutex_lock(&sessions_mutex);
522 if (!session->active) {
523 ret = -EBUSY;
524 goto end;
525 }
a8f2d0c7 526 WRITE_ONCE(session->active, 0);
3c997079
MD		 527
528 /* Set transient enabler state to "disabled" */
8cdc1a81 529 session->priv->tstate = 0;
b2bc0bc8 530 lttng_session_sync_event_enablers(session);
64af2437
MD		 531
532 /* Set each stream's quiescent state. */
f7d06400
MD		 533 list_for_each_entry(chan_priv, &session->priv->chan, node) {
534 if (chan_priv->channel_type != METADATA_CHANNEL)
535 lib_ring_buffer_set_quiescent_channel(chan_priv->rb_chan);
09b93db9 536 }
c0e31d2e
MD		 537end:
538 mutex_unlock(&sessions_mutex);
11b5a3c2 539 return ret;
c0e31d2e
MD		 540}
541
8cdc1a81 542int lttng_session_metadata_regenerate(struct lttng_kernel_session *session)
9616f0bf
JD		 543{
544 int ret = 0;
f7d06400 545 struct lttng_kernel_channel_buffer_private *chan_priv;
a67ba386 546 struct lttng_kernel_event_recorder_private *event_recorder_priv;
8cdc1a81 547 struct lttng_metadata_cache *cache = session->priv->metadata_cache;
9616f0bf
JD		 548 struct lttng_metadata_stream *stream;
549
550 mutex_lock(&sessions_mutex);
551 if (!session->active) {
552 ret = -EBUSY;
553 goto end;
554 }
555
556 mutex_lock(&cache->lock);
557 memset(cache->data, 0, cache->cache_alloc);
558 cache->metadata_written = 0;
559 cache->version++;
8cdc1a81 560 list_for_each_entry(stream, &session->priv->metadata_cache->metadata_stream, list) {
9616f0bf
JD		 561 stream->metadata_out = 0;
562 stream->metadata_in = 0;
563 }
564 mutex_unlock(&cache->lock);
565
8cdc1a81 566 session->priv->metadata_dumped = 0;
f7d06400
MD		 567 list_for_each_entry(chan_priv, &session->priv->chan, node) {
568 chan_priv->metadata_dumped = 0;
9616f0bf
JD		 569 }
570
8cdc1a81 571 list_for_each_entry(event_recorder_priv, &session->priv->events, node) {
a67ba386 572 event_recorder_priv->metadata_dumped = 0;
9616f0bf
JD		 573 }
574
575 ret = _lttng_session_metadata_statedump(session);
576
577end:
578 mutex_unlock(&sessions_mutex);
579 return ret;
580}
581
6b757027
MD		 582static
583bool is_channel_buffer_metadata(struct lttng_kernel_channel_common *channel)
584{
585 struct lttng_kernel_channel_buffer *chan_buf;
586
587 if (channel->type != LTTNG_KERNEL_CHANNEL_TYPE_BUFFER)
588 return false;
589 chan_buf = container_of(channel, struct lttng_kernel_channel_buffer, parent);
590 if (chan_buf->priv->channel_type == METADATA_CHANNEL)
591 return true;
592 return false;
593}
594
595int lttng_channel_enable(struct lttng_kernel_channel_common *channel)
e64957da 596{
3c997079 597 int ret = 0;
e64957da 598
3c997079 599 mutex_lock(&sessions_mutex);
6b757027 600 if (is_channel_buffer_metadata(channel)) {
3c997079
MD		 601 ret = -EPERM;
602 goto end;
603 }
6b757027 604 if (channel->enabled) {
3c997079
MD		 605 ret = -EEXIST;
606 goto end;
607 }
608 /* Set transient enabler state to "enabled" */
6b757027
MD		 609 channel->priv->tstate = 1;
610 lttng_session_sync_event_enablers(channel->session);
3c997079 611 /* Set atomically the state to "enabled" */
6b757027 612 WRITE_ONCE(channel->enabled, 1);
3c997079
MD		 613end:
614 mutex_unlock(&sessions_mutex);
615 return ret;
e64957da
MD		 616}
617
6b757027 618int lttng_channel_disable(struct lttng_kernel_channel_common *channel)
e64957da 619{
3c997079 620 int ret = 0;
e64957da 621
3c997079 622 mutex_lock(&sessions_mutex);
6b757027 623 if (is_channel_buffer_metadata(channel)) {
3c997079
MD		 624 ret = -EPERM;
625 goto end;
626 }
6b757027 627 if (!channel->enabled) {
3c997079
MD		 628 ret = -EEXIST;
629 goto end;
630 }
631 /* Set atomically the state to "disabled" */
6b757027 632 WRITE_ONCE(channel->enabled, 0);
3c997079 633 /* Set transient enabler state to "enabled" */
6b757027
MD		 634 channel->priv->tstate = 0;
635 lttng_session_sync_event_enablers(channel->session);
3c997079
MD		 636end:
637 mutex_unlock(&sessions_mutex);
638 return ret;
e64957da
MD		 639}
640
9d993668 641int lttng_event_enable(struct lttng_kernel_event_common *event)
e64957da 642{
3c997079 643 int ret = 0;
e64957da 644
3c997079 645 mutex_lock(&sessions_mutex);
9d993668
MD		 646 switch (event->type) {
647 case LTTNG_KERNEL_EVENT_TYPE_RECORDER:
648 {
649 struct lttng_kernel_event_recorder *event_recorder =
650 container_of(event, struct lttng_kernel_event_recorder, parent);
651
f7d06400 652 if (event_recorder->chan->priv->channel_type == METADATA_CHANNEL) {
9d993668
MD		 653 ret = -EPERM;
654 goto end;
655 }
a0493bef 656 break;
9d993668
MD		 657 }
658 case LTTNG_KERNEL_EVENT_TYPE_NOTIFIER:
659 switch (event->priv->instrumentation) {
660 case LTTNG_KERNEL_ABI_KRETPROBE:
661 ret = -EINVAL;
662 goto end;
663 default:
664 break;
665 }
a0493bef
MD		 666 break;
667 default:
9d993668 668 break;
a0493bef 669 }
e64957da 670
9d993668 671 if (event->enabled) {
3c997079
MD		 672 ret = -EEXIST;
673 goto end;
674 }
a67ba386 675 switch (event->priv->instrumentation) {
c190d76e
MJ		 676 case LTTNG_KERNEL_ABI_TRACEPOINT:
677 lttng_fallthrough;
606828e4 678 case LTTNG_KERNEL_ABI_SYSCALL:
a0493bef
MD		 679 ret = -EINVAL;
680 break;
b4a75708 681
c190d76e
MJ		 682 case LTTNG_KERNEL_ABI_KPROBE:
683 lttng_fallthrough;
606828e4 684 case LTTNG_KERNEL_ABI_UPROBE:
9d993668 685 WRITE_ONCE(event->enabled, 1);
a0493bef 686 break;
b4a75708 687
606828e4 688 case LTTNG_KERNEL_ABI_KRETPROBE:
9d993668 689 ret = lttng_kretprobes_event_enable_state(event, 1);
a0493bef 690 break;
b4a75708 691
c190d76e
MJ		 692 case LTTNG_KERNEL_ABI_FUNCTION:
693 lttng_fallthrough;
694 case LTTNG_KERNEL_ABI_NOOP:
695 lttng_fallthrough;
a0493bef
MD		 696 default:
697 WARN_ON_ONCE(1);
698 ret = -EINVAL;
699 }
3c997079
MD		 700end:
701 mutex_unlock(&sessions_mutex);
702 return ret;
e64957da
MD		 703}
704
9d993668 705int lttng_event_disable(struct lttng_kernel_event_common *event)
dffef45d
FD		 706{
707 int ret = 0;
708
709 mutex_lock(&sessions_mutex);
9d993668
MD		 710 switch (event->type) {
711 case LTTNG_KERNEL_EVENT_TYPE_RECORDER:
712 {
713 struct lttng_kernel_event_recorder *event_recorder =
714 container_of(event, struct lttng_kernel_event_recorder, parent);
715
f7d06400 716 if (event_recorder->chan->priv->channel_type == METADATA_CHANNEL) {
9d993668
MD		 717 ret = -EPERM;
718 goto end;
719 }
b01155ba 720 break;
9d993668
MD		 721 }
722 case LTTNG_KERNEL_EVENT_TYPE_NOTIFIER:
723 switch (event->priv->instrumentation) {
724 case LTTNG_KERNEL_ABI_KRETPROBE:
725 ret = -EINVAL;
726 goto end;
727 default:
728 break;
729 }
2b16f0c9 730 break;
dffef45d 731 default:
9d993668 732 break;
dffef45d 733 }
dffef45d 734
a67ba386 735 if (!event->enabled) {
dffef45d
FD		 736 ret = -EEXIST;
737 goto end;
738 }
a67ba386 739 switch (event->priv->instrumentation) {
c190d76e
MJ		 740 case LTTNG_KERNEL_ABI_TRACEPOINT:
741 lttng_fallthrough;
606828e4 742 case LTTNG_KERNEL_ABI_SYSCALL:
b01155ba
FD		 743 ret = -EINVAL;
744 break;
b4a75708 745
c190d76e
MJ		 746 case LTTNG_KERNEL_ABI_KPROBE:
747 lttng_fallthrough;
606828e4 748 case LTTNG_KERNEL_ABI_UPROBE:
a67ba386 749 WRITE_ONCE(event->enabled, 0);
2b16f0c9 750 break;
b4a75708 751
9d993668
MD		 752 case LTTNG_KERNEL_ABI_KRETPROBE:
753 ret = lttng_kretprobes_event_enable_state(event, 0);
754 break;
755
c190d76e
MJ		 756 case LTTNG_KERNEL_ABI_FUNCTION:
757 lttng_fallthrough;
758 case LTTNG_KERNEL_ABI_NOOP:
759 lttng_fallthrough;
dffef45d
FD		 760 default:
761 WARN_ON_ONCE(1);
762 ret = -EINVAL;
763 }
764end:
765 mutex_unlock(&sessions_mutex);
766 return ret;
767}
768
4e252d1e 769struct lttng_kernel_channel_buffer *lttng_channel_buffer_create(struct lttng_kernel_session *session,
5dbbdb43
MD		 770 const char *transport_name,
771 void *buf_addr,
4e3c1b9b
MD		 772 size_t subbuf_size, size_t num_subbuf,
773 unsigned int switch_timer_interval,
d83004aa
JD		 774 unsigned int read_timer_interval,
775 enum channel_type channel_type)
4e3c1b9b 776{
f7d06400
MD		 777 struct lttng_kernel_channel_buffer *chan;
778 struct lttng_kernel_channel_buffer_private *chan_priv;
a90917c3 779 struct lttng_transport *transport = NULL;
4e3c1b9b
MD		 780
781 mutex_lock(&sessions_mutex);
8cdc1a81 782 if (session->priv->been_active && channel_type != METADATA_CHANNEL)
e5382b6d 783 goto active; /* Refuse to add channel to active session */
a90917c3 784 transport = lttng_transport_find(transport_name);
f3d01b96 785 if (!transport) {
5a15f70c 786 printk(KERN_WARNING "LTTng: transport %s not found\n",
f3d01b96 787 transport_name);
c0e31d2e 788 goto notransport;
f3d01b96 789 }
a33c9927 790 if (!try_module_get(transport->owner)) {
5a15f70c 791 printk(KERN_WARNING "LTTng: Can't lock transport module.\n");
a33c9927
MD		 792 goto notransport;
793 }
f7d06400 794 chan = kzalloc(sizeof(struct lttng_kernel_channel_buffer), GFP_KERNEL);
4e3c1b9b 795 if (!chan)
c0e31d2e 796 goto nomem;
f7d06400
MD		 797 chan_priv = kzalloc(sizeof(struct lttng_kernel_channel_buffer_private), GFP_KERNEL);
798 if (!chan_priv)
799 goto nomem_priv;
800 chan->priv = chan_priv;
801 chan_priv->pub = chan;
802 chan->parent.type = LTTNG_KERNEL_CHANNEL_TYPE_BUFFER;
803 chan->parent.session = session;
804 chan->priv->id = session->priv->free_chan_id++;
3b731ab1 805 chan->ops = &transport->ops;
125b4df4
MD		 806 /*
807 * Note: the channel creation op already writes into the packet
808 * headers. Therefore the "chan" information used as input
809 * should be already accessible.
810 */
f7d06400 811 chan->priv->rb_chan = transport->ops.priv->channel_create(transport_name,
5f5ddf01
MD		 812 chan, buf_addr, subbuf_size, num_subbuf,
813 switch_timer_interval, read_timer_interval);
f7d06400 814 if (!chan->priv->rb_chan)
f3d01b96 815 goto create_error;
f7d06400
MD		 816 chan->priv->parent.tstate = 1;
817 chan->parent.enabled = 1;
818 chan->priv->transport = transport;
819 chan->priv->channel_type = channel_type;
820 list_add(&chan->priv->node, &session->priv->chan);
4e3c1b9b
MD		 821 mutex_unlock(&sessions_mutex);
822 return chan;
823
f3d01b96 824create_error:
f7d06400
MD		 825 kfree(chan_priv);
826nomem_priv:
f3d01b96 827 kfree(chan);
c0e31d2e 828nomem:
a33c9927
MD		 829 if (transport)
830 module_put(transport->owner);
c0e31d2e 831notransport:
e5382b6d 832active:
4e3c1b9b
MD		 833 mutex_unlock(&sessions_mutex);
834 return NULL;
835}
836
837/*
d83004aa
JD		 838 * Only used internally at session destruction for per-cpu channels, and
839 * when metadata channel is released.
840 * Needs to be called with sessions mutex held.
4e3c1b9b 841 */
aa7c23a9 842static
f7d06400 843void _lttng_channel_destroy(struct lttng_kernel_channel_buffer *chan)
4e3c1b9b 844{
f7d06400
MD		 845 chan->ops->priv->channel_destroy(chan->priv->rb_chan);
846 module_put(chan->priv->transport->owner);
847 list_del(&chan->priv->node);
848 lttng_kernel_destroy_context(chan->priv->ctx);
849 kfree(chan->priv);
4e3c1b9b
MD		 850 kfree(chan);
851}
852
f7d06400 853void lttng_metadata_channel_destroy(struct lttng_kernel_channel_buffer *chan)
d83004aa 854{
f7d06400 855 BUG_ON(chan->priv->channel_type != METADATA_CHANNEL);
d83004aa
JD		 856
857 /* Protect the metadata cache with the sessions_mutex. */
858 mutex_lock(&sessions_mutex);
859 _lttng_channel_destroy(chan);
860 mutex_unlock(&sessions_mutex);
861}
862EXPORT_SYMBOL_GPL(lttng_metadata_channel_destroy);
863
864static
865void _lttng_metadata_channel_hangup(struct lttng_metadata_stream *stream)
866{
867 stream->finalized = 1;
868 wake_up_interruptible(&stream->read_wait);
869}
870
a101fa10 871
e5382b6d
MD		 872/*
873 * Supports event creation while tracing session is active.
3c997079 874 * Needs to be called with sessions mutex held.
e5382b6d 875 */
1ae083ba 876struct lttng_kernel_event_recorder *_lttng_kernel_event_recorder_create(struct lttng_event_recorder_enabler *event_enabler,
def1e304 877 const struct lttng_kernel_event_desc *event_desc)
4e3c1b9b 878{
def1e304 879 struct lttng_kernel_channel_buffer *chan = event_enabler->chan;
5b9817ae 880 struct lttng_kernel_abi_event *event_param = &event_enabler->parent.event_param;
def1e304 881 enum lttng_kernel_abi_instrumentation itype = event_param->instrumentation;
f7d06400 882 struct lttng_kernel_session *session = chan->parent.session;
a67ba386
MD		 883 struct lttng_kernel_event_recorder *event_recorder;
884 struct lttng_kernel_event_recorder_private *event_recorder_priv;
3c997079
MD		 885 const char *event_name;
886 struct hlist_head *head;
3d084699 887 int ret;
4e3c1b9b 888
f7d06400 889 if (chan->priv->free_event_id == -1U) {
4cf0bf51 890 ret = -EMFILE;
e5382b6d 891 goto full;
abc0446a 892 }
3c997079
MD		 893
894 switch (itype) {
606828e4 895 case LTTNG_KERNEL_ABI_TRACEPOINT:
437d5aa5 896 event_name = event_desc->event_name;
3c997079 897 break;
b4a75708 898
c190d76e
MJ		 899 case LTTNG_KERNEL_ABI_KPROBE:
900 lttng_fallthrough;
901 case LTTNG_KERNEL_ABI_UPROBE:
902 lttng_fallthrough;
903 case LTTNG_KERNEL_ABI_KRETPROBE:
904 lttng_fallthrough;
606828e4 905 case LTTNG_KERNEL_ABI_SYSCALL:
3c997079
MD		 906 event_name = event_param->name;
907 break;
b4a75708 908
c190d76e
MJ		 909 case LTTNG_KERNEL_ABI_FUNCTION:
910 lttng_fallthrough;
911 case LTTNG_KERNEL_ABI_NOOP:
912 lttng_fallthrough;
3c997079
MD		 913 default:
914 WARN_ON_ONCE(1);
915 ret = -EINVAL;
916 goto type_error;
917 }
4a11ef1c 918
8cdc1a81 919 head = utils_borrow_hash_table_bucket(session->priv->events_ht.table,
4a11ef1c 920 LTTNG_EVENT_HT_SIZE, event_name);
a67ba386
MD		 921 lttng_hlist_for_each_entry(event_recorder_priv, head, hlist) {
922 WARN_ON_ONCE(!event_recorder_priv->parent.desc);
923 if (!strncmp(event_recorder_priv->parent.desc->event_name, event_name,
606828e4 924 LTTNG_KERNEL_ABI_SYM_NAME_LEN - 1)
a67ba386 925 && chan == event_recorder_priv->pub->chan) {
3c997079
MD		 926 ret = -EEXIST;
927 goto exist;
abc0446a
MD		 928 }
929 }
3c997079 930
a67ba386
MD		 931 event_recorder = kmem_cache_zalloc(event_recorder_cache, GFP_KERNEL);
932 if (!event_recorder) {
4cf0bf51 933 ret = -ENOMEM;
4e3c1b9b 934 goto cache_error;
abc0446a 935 }
a67ba386
MD		 936 event_recorder_priv = kmem_cache_zalloc(event_recorder_private_cache, GFP_KERNEL);
937 if (!event_recorder_priv) {
938 ret = -ENOMEM;
939 goto cache_private_error;
940 }
941 event_recorder_priv->pub = event_recorder;
942 event_recorder_priv->parent.pub = &event_recorder->parent;
943 event_recorder->priv = event_recorder_priv;
944 event_recorder->parent.priv = &event_recorder_priv->parent;
945 event_recorder->parent.type = LTTNG_KERNEL_EVENT_TYPE_RECORDER;
946
8a445457 947 event_recorder->parent.run_filter = lttng_kernel_interpret_event_filter;
a67ba386 948 event_recorder->chan = chan;
f7d06400 949 event_recorder->priv->id = chan->priv->free_event_id++;
a67ba386
MD		 950 event_recorder->priv->parent.instrumentation = itype;
951 INIT_LIST_HEAD(&event_recorder->priv->parent.filter_bytecode_runtime_head);
952 INIT_LIST_HEAD(&event_recorder->priv->parent.enablers_ref_head);
3c997079
MD		 953
954 switch (itype) {
606828e4 955 case LTTNG_KERNEL_ABI_TRACEPOINT:
3c997079 956 /* Event will be enabled by enabler sync. */
a67ba386
MD		 957 event_recorder->parent.enabled = 0;
958 event_recorder->priv->parent.registered = 0;
959 event_recorder->priv->parent.desc = lttng_event_desc_get(event_name);
960 if (!event_recorder->priv->parent.desc) {
4cf0bf51 961 ret = -ENOENT;
d3dbe23c 962 goto register_error;
abc0446a 963 }
3c997079
MD		 964 /* Populate lttng_event structure before event registration. */
965 smp_wmb();
baf20995 966 break;
b4a75708 967
606828e4 968 case LTTNG_KERNEL_ABI_KPROBE:
a0493bef
MD		 969 /*
970 * Needs to be explicitly enabled after creation, since
971 * we may want to apply filters.
972 */
a67ba386
MD		 973 event_recorder->parent.enabled = 0;
974 event_recorder->priv->parent.registered = 1;
3c997079
MD		 975 /*
976 * Populate lttng_event structure before event
977 * registration.
978 */
979 smp_wmb();
8bf17deb 980 ret = lttng_kprobes_register_event(event_name,
f17701fb
MD		 981 event_param->u.kprobe.symbol_name,
982 event_param->u.kprobe.offset,
983 event_param->u.kprobe.addr,
a67ba386 984 event_recorder);
abc0446a 985 if (ret) {
4cf0bf51 986 ret = -EINVAL;
d6d808f3 987 goto register_error;
abc0446a 988 }
a67ba386 989 ret = try_module_get(event_recorder->priv->parent.desc->owner);
edeb3137 990 WARN_ON_ONCE(!ret);
d6d808f3 991 break;
b4a75708 992
606828e4 993 case LTTNG_KERNEL_ABI_KRETPROBE:
7371f44c 994 {
a67ba386
MD		 995 struct lttng_kernel_event_recorder *event_recorder_return;
996 struct lttng_kernel_event_recorder_private *event_recorder_return_priv;
7371f44c
MD		 997
998 /* kretprobe defines 2 events */
a0493bef
MD		 999 /*
1000 * Needs to be explicitly enabled after creation, since
1001 * we may want to apply filters.
1002 */
a67ba386
MD		 1003 event_recorder->parent.enabled = 0;
1004 event_recorder->priv->parent.registered = 1;
1005
1006 event_recorder_return = kmem_cache_zalloc(event_recorder_cache, GFP_KERNEL);
1007 if (!event_recorder_return) {
1008 ret = -ENOMEM;
1009 goto register_error;
1010 }
1011 event_recorder_return_priv = kmem_cache_zalloc(event_recorder_private_cache, GFP_KERNEL);
1012 if (!event_recorder_return_priv) {
1013 kmem_cache_free(event_recorder_cache, event_recorder_return);
4cf0bf51 1014 ret = -ENOMEM;
7371f44c 1015 goto register_error;
abc0446a 1016 }
a67ba386
MD		 1017 event_recorder_return_priv->pub = event_recorder_return;
1018 event_recorder_return_priv->parent.pub = &event_recorder_return->parent;
1019 event_recorder_return->priv = event_recorder_return_priv;
1020 event_recorder_return->parent.priv = &event_recorder_return_priv->parent;
1021 event_recorder_return->parent.type = LTTNG_KERNEL_EVENT_TYPE_RECORDER;
1022
8a445457 1023 event_recorder_return->parent.run_filter = lttng_kernel_interpret_event_filter;
a67ba386 1024 event_recorder_return->chan = chan;
f7d06400 1025 event_recorder_return->priv->id = chan->priv->free_event_id++;
a67ba386
MD		 1026 event_recorder_return->priv->parent.instrumentation = itype;
1027 event_recorder_return->parent.enabled = 0;
1028 event_recorder_return->priv->parent.registered = 1;
1029 INIT_LIST_HEAD(&event_recorder_return->priv->parent.filter_bytecode_runtime_head);
1030 INIT_LIST_HEAD(&event_recorder_return->priv->parent.enablers_ref_head);
7371f44c 1031 /*
a90917c3 1032 * Populate lttng_event structure before kretprobe registration.
7371f44c
MD		 1033 */
1034 smp_wmb();
3c997079 1035 ret = lttng_kretprobes_register(event_name,
7371f44c
MD		 1036 event_param->u.kretprobe.symbol_name,
1037 event_param->u.kretprobe.offset,
1038 event_param->u.kretprobe.addr,
a67ba386 1039 event_recorder, event_recorder_return);
7371f44c 1040 if (ret) {
a67ba386
MD		 1041 kmem_cache_free(event_recorder_private_cache, event_recorder_return_priv);
1042 kmem_cache_free(event_recorder_cache, event_recorder_return);
4cf0bf51 1043 ret = -EINVAL;
7371f44c
MD		 1044 goto register_error;
1045 }
1046 /* Take 2 refs on the module: one per event. */
a67ba386 1047 ret = try_module_get(event_recorder->priv->parent.desc->owner);
7371f44c 1048 WARN_ON_ONCE(!ret);
a67ba386 1049 ret = try_module_get(event_recorder_return->priv->parent.desc->owner);
7371f44c 1050 WARN_ON_ONCE(!ret);
f7d06400 1051 ret = _lttng_event_metadata_statedump(chan->parent.session, chan,
a67ba386 1052 event_recorder_return);
abc0446a 1053 WARN_ON_ONCE(ret > 0);
7371f44c 1054 if (ret) {
a67ba386
MD		 1055 kmem_cache_free(event_recorder_private_cache, event_recorder_return_priv);
1056 kmem_cache_free(event_recorder_cache, event_recorder_return);
1057 module_put(event_recorder_return->priv->parent.desc->owner);
1058 module_put(event_recorder->priv->parent.desc->owner);
7371f44c
MD		 1059 goto statedump_error;
1060 }
f7d06400 1061 list_add(&event_recorder_return->priv->node, &chan->parent.session->priv->events);
7371f44c
MD		 1062 break;
1063 }
b4a75708 1064
606828e4 1065 case LTTNG_KERNEL_ABI_SYSCALL:
a0493bef
MD		 1066 /*
1067 * Needs to be explicitly enabled after creation, since
1068 * we may want to apply filters.
1069 */
a67ba386
MD		 1070 event_recorder->parent.enabled = 0;
1071 event_recorder->priv->parent.registered = 0;
1072 event_recorder->priv->parent.desc = event_desc;
badfe9f5 1073 switch (event_param->u.syscall.entryexit) {
606828e4 1074 case LTTNG_KERNEL_ABI_SYSCALL_ENTRYEXIT:
badfe9f5
MD		 1075 ret = -EINVAL;
1076 goto register_error;
606828e4 1077 case LTTNG_KERNEL_ABI_SYSCALL_ENTRY:
a67ba386 1078 event_recorder->priv->parent.u.syscall.entryexit = LTTNG_SYSCALL_ENTRY;
badfe9f5 1079 break;
606828e4 1080 case LTTNG_KERNEL_ABI_SYSCALL_EXIT:
a67ba386 1081 event_recorder->priv->parent.u.syscall.entryexit = LTTNG_SYSCALL_EXIT;
badfe9f5
MD		 1082 break;
1083 }
1084 switch (event_param->u.syscall.abi) {
606828e4 1085 case LTTNG_KERNEL_ABI_SYSCALL_ABI_ALL:
badfe9f5
MD		 1086 ret = -EINVAL;
1087 goto register_error;
606828e4 1088 case LTTNG_KERNEL_ABI_SYSCALL_ABI_NATIVE:
a67ba386 1089 event_recorder->priv->parent.u.syscall.abi = LTTNG_SYSCALL_ABI_NATIVE;
badfe9f5 1090 break;
606828e4 1091 case LTTNG_KERNEL_ABI_SYSCALL_ABI_COMPAT:
a67ba386 1092 event_recorder->priv->parent.u.syscall.abi = LTTNG_SYSCALL_ABI_COMPAT;
badfe9f5
MD		 1093 break;
1094 }
a67ba386 1095 if (!event_recorder->priv->parent.desc) {
4cf0bf51 1096 ret = -EINVAL;
259b6cb3 1097 goto register_error;
abc0446a 1098 }
1ec65de1 1099 break;
b4a75708 1100
606828e4 1101 case LTTNG_KERNEL_ABI_UPROBE:
a344f64b
FD		 1102 /*
1103 * Needs to be explicitly enabled after creation, since
1104 * we may want to apply filters.
1105 */
a67ba386
MD		 1106 event_recorder->parent.enabled = 0;
1107 event_recorder->priv->parent.registered = 1;
3aed4dca 1108
a344f64b
FD		 1109 /*
1110 * Populate lttng_event structure before event
1111 * registration.
1112 */
1113 smp_wmb();
149b9a9d 1114
83b802dc 1115 ret = lttng_uprobes_register_event(event_param->name,
56377c91 1116 event_param->u.uprobe.fd,
17806c46 1117 &event_recorder->parent);
149b9a9d
YB		 1118 if (ret)
1119 goto register_error;
a67ba386 1120 ret = try_module_get(event_recorder->priv->parent.desc->owner);
149b9a9d
YB		 1121 WARN_ON_ONCE(!ret);
1122 break;
b4a75708 1123
c190d76e
MJ		 1124 case LTTNG_KERNEL_ABI_FUNCTION:
1125 lttng_fallthrough;
1126 case LTTNG_KERNEL_ABI_NOOP:
1127 lttng_fallthrough;
baf20995
MD		 1128 default:
1129 WARN_ON_ONCE(1);
4cf0bf51 1130 ret = -EINVAL;
df07930b 1131 goto register_error;
baf20995 1132 }
f7d06400 1133 ret = _lttng_event_metadata_statedump(chan->parent.session, chan, event_recorder);
abc0446a
MD		 1134 WARN_ON_ONCE(ret > 0);
1135 if (ret) {
c099397a 1136 goto statedump_error;
abc0446a 1137 }
a67ba386 1138 hlist_add_head(&event_recorder->priv->hlist, head);
f7d06400 1139 list_add(&event_recorder->priv->node, &chan->parent.session->priv->events);
a67ba386 1140 return event_recorder;
4e3c1b9b 1141
c099397a 1142statedump_error:
259b6cb3 1143 /* If a statedump error occurs, events will not be readable. */
11b5a3c2 1144register_error:
a67ba386
MD		 1145 kmem_cache_free(event_recorder_private_cache, event_recorder_priv);
1146cache_private_error:
1147 kmem_cache_free(event_recorder_cache, event_recorder);
4e3c1b9b
MD		 1148cache_error:
1149exist:
3c997079 1150type_error:
e5382b6d 1151full:
4cf0bf51 1152 return ERR_PTR(ret);
4e3c1b9b
MD		 1153}
1154
a67ba386 1155struct lttng_kernel_event_notifier *_lttng_event_notifier_create(
437d5aa5 1156 const struct lttng_kernel_event_desc *event_desc,
99f52fcc
FD		 1157 uint64_t token, uint64_t error_counter_index,
1158 struct lttng_event_notifier_group *event_notifier_group,
606828e4 1159 struct lttng_kernel_abi_event_notifier *event_notifier_param,
a67ba386 1160 enum lttng_kernel_abi_instrumentation itype)
dffef45d 1161{
a67ba386
MD		 1162 struct lttng_kernel_event_notifier *event_notifier;
1163 struct lttng_kernel_event_notifier_private *event_notifier_priv;
ab04d7b1 1164 struct lttng_counter *error_counter;
dffef45d
FD		 1165 const char *event_name;
1166 struct hlist_head *head;
1167 int ret;
1168
1169 switch (itype) {
606828e4 1170 case LTTNG_KERNEL_ABI_TRACEPOINT:
437d5aa5 1171 event_name = event_desc->event_name;
b01155ba 1172 break;
b4a75708 1173
c190d76e
MJ		 1174 case LTTNG_KERNEL_ABI_KPROBE:
1175 lttng_fallthrough;
1176 case LTTNG_KERNEL_ABI_UPROBE:
1177 lttng_fallthrough;
606828e4 1178 case LTTNG_KERNEL_ABI_SYSCALL:
2b16f0c9
FD		 1179 event_name = event_notifier_param->event.name;
1180 break;
b4a75708 1181
c190d76e
MJ		 1182 case LTTNG_KERNEL_ABI_KRETPROBE:
1183 lttng_fallthrough;
1184 case LTTNG_KERNEL_ABI_FUNCTION:
1185 lttng_fallthrough;
1186 case LTTNG_KERNEL_ABI_NOOP:
1187 lttng_fallthrough;
dffef45d
FD		 1188 default:
1189 WARN_ON_ONCE(1);
1190 ret = -EINVAL;
1191 goto type_error;
1192 }
1193
1194 head = utils_borrow_hash_table_bucket(event_notifier_group->event_notifiers_ht.table,
1195 LTTNG_EVENT_NOTIFIER_HT_SIZE, event_name);
a67ba386
MD		 1196 lttng_hlist_for_each_entry(event_notifier_priv, head, hlist) {
1197 WARN_ON_ONCE(!event_notifier_priv->parent.desc);
1198 if (!strncmp(event_notifier_priv->parent.desc->event_name, event_name,
606828e4 1199 LTTNG_KERNEL_ABI_SYM_NAME_LEN - 1)
a67ba386
MD		 1200 && event_notifier_group == event_notifier_priv->group
1201 && token == event_notifier_priv->parent.user_token) {
dffef45d
FD		 1202 ret = -EEXIST;
1203 goto exist;
1204 }
1205 }
1206
1207 event_notifier = kmem_cache_zalloc(event_notifier_cache, GFP_KERNEL);
1208 if (!event_notifier) {
1209 ret = -ENOMEM;
1210 goto cache_error;
1211 }
a67ba386
MD		 1212 event_notifier_priv = kmem_cache_zalloc(event_notifier_private_cache, GFP_KERNEL);
1213 if (!event_notifier_priv) {
1214 ret = -ENOMEM;
1215 goto cache_private_error;
1216 }
1217 event_notifier_priv->pub = event_notifier;
1218 event_notifier_priv->parent.pub = &event_notifier->parent;
1219 event_notifier->priv = event_notifier_priv;
1220 event_notifier->parent.priv = &event_notifier_priv->parent;
1221 event_notifier->parent.type = LTTNG_KERNEL_EVENT_TYPE_NOTIFIER;
1222
1223 event_notifier->priv->group = event_notifier_group;
1224 event_notifier->priv->parent.user_token = token;
1225 event_notifier->priv->error_counter_index = error_counter_index;
1226 event_notifier->priv->num_captures = 0;
1227 event_notifier->priv->parent.instrumentation = itype;
1228 event_notifier->notification_send = lttng_event_notifier_notification_send;
1229 INIT_LIST_HEAD(&event_notifier->priv->parent.filter_bytecode_runtime_head);
1230 INIT_LIST_HEAD(&event_notifier->priv->parent.enablers_ref_head);
1231 INIT_LIST_HEAD(&event_notifier->priv->capture_bytecode_runtime_head);
8a445457 1232 event_notifier->parent.run_filter = lttng_kernel_interpret_event_filter;
dffef45d
FD		 1233
1234 switch (itype) {
606828e4 1235 case LTTNG_KERNEL_ABI_TRACEPOINT:
dffef45d 1236 /* Event will be enabled by enabler sync. */
a67ba386
MD		 1237 event_notifier->parent.enabled = 0;
1238 event_notifier->priv->parent.registered = 0;
1239 event_notifier->priv->parent.desc = lttng_event_desc_get(event_name);
1240 if (!event_notifier->priv->parent.desc) {
dffef45d
FD		 1241 ret = -ENOENT;
1242 goto register_error;
1243 }
1244 /* Populate lttng_event_notifier structure before event registration. */
1245 smp_wmb();
1246 break;
b4a75708 1247
606828e4 1248 case LTTNG_KERNEL_ABI_KPROBE:
2b16f0c9
FD		 1249 /*
1250 * Needs to be explicitly enabled after creation, since
1251 * we may want to apply filters.
1252 */
a67ba386
MD		 1253 event_notifier->parent.enabled = 0;
1254 event_notifier->priv->parent.registered = 1;
2b16f0c9
FD		 1255 /*
1256 * Populate lttng_event_notifier structure before event
1257 * registration.
1258 */
1259 smp_wmb();
1260 ret = lttng_kprobes_register_event_notifier(
1261 event_notifier_param->event.u.kprobe.symbol_name,
1262 event_notifier_param->event.u.kprobe.offset,
1263 event_notifier_param->event.u.kprobe.addr,
1264 event_notifier);
1265 if (ret) {
1266 ret = -EINVAL;
1267 goto register_error;
1268 }
a67ba386 1269 ret = try_module_get(event_notifier->priv->parent.desc->owner);
2b16f0c9
FD		 1270 WARN_ON_ONCE(!ret);
1271 break;
b4a75708 1272
606828e4 1273 case LTTNG_KERNEL_ABI_SYSCALL:
8a8ac9a8
FD		 1274 /*
1275 * Needs to be explicitly enabled after creation, since
1276 * we may want to apply filters.
1277 */
a67ba386
MD		 1278 event_notifier->parent.enabled = 0;
1279 event_notifier->priv->parent.registered = 0;
1280 event_notifier->priv->parent.desc = event_desc;
8a8ac9a8 1281 switch (event_notifier_param->event.u.syscall.entryexit) {
606828e4 1282 case LTTNG_KERNEL_ABI_SYSCALL_ENTRYEXIT:
8a8ac9a8
FD		 1283 ret = -EINVAL;
1284 goto register_error;
606828e4 1285 case LTTNG_KERNEL_ABI_SYSCALL_ENTRY:
a67ba386 1286 event_notifier->priv->parent.u.syscall.entryexit = LTTNG_SYSCALL_ENTRY;
8a8ac9a8 1287 break;
606828e4 1288 case LTTNG_KERNEL_ABI_SYSCALL_EXIT:
a67ba386 1289 event_notifier->priv->parent.u.syscall.entryexit = LTTNG_SYSCALL_EXIT;
8a8ac9a8
FD		 1290 break;
1291 }
1292 switch (event_notifier_param->event.u.syscall.abi) {
606828e4 1293 case LTTNG_KERNEL_ABI_SYSCALL_ABI_ALL:
8a8ac9a8
FD		 1294 ret = -EINVAL;
1295 goto register_error;
606828e4 1296 case LTTNG_KERNEL_ABI_SYSCALL_ABI_NATIVE:
a67ba386 1297 event_notifier->priv->parent.u.syscall.abi = LTTNG_SYSCALL_ABI_NATIVE;
8a8ac9a8 1298 break;
606828e4 1299 case LTTNG_KERNEL_ABI_SYSCALL_ABI_COMPAT:
a67ba386 1300 event_notifier->priv->parent.u.syscall.abi = LTTNG_SYSCALL_ABI_COMPAT;
8a8ac9a8
FD		 1301 break;
1302 }
1303
a67ba386 1304 if (!event_notifier->priv->parent.desc) {
8a8ac9a8
FD		 1305 ret = -EINVAL;
1306 goto register_error;
1307 }
1308 break;
b4a75708 1309
606828e4 1310 case LTTNG_KERNEL_ABI_UPROBE:
9de67196
FD		 1311 /*
1312 * Needs to be explicitly enabled after creation, since
1313 * we may want to apply filters.
1314 */
a67ba386
MD		 1315 event_notifier->parent.enabled = 0;
1316 event_notifier->priv->parent.registered = 1;
9de67196
FD		 1317
1318 /*
1319 * Populate lttng_event_notifier structure before
1320 * event_notifier registration.
1321 */
1322 smp_wmb();
1323
17806c46 1324 ret = lttng_uprobes_register_event(event_notifier_param->event.name,
9de67196 1325 event_notifier_param->event.u.uprobe.fd,
17806c46 1326 &event_notifier->parent);
9de67196
FD		 1327 if (ret)
1328 goto register_error;
a67ba386 1329 ret = try_module_get(event_notifier->priv->parent.desc->owner);
9de67196
FD		 1330 WARN_ON_ONCE(!ret);
1331 break;
b4a75708 1332
c190d76e
MJ		 1333 case LTTNG_KERNEL_ABI_KRETPROBE:
1334 lttng_fallthrough;
1335 case LTTNG_KERNEL_ABI_FUNCTION:
1336 lttng_fallthrough;
1337 case LTTNG_KERNEL_ABI_NOOP:
1338 lttng_fallthrough;
dffef45d
FD		 1339 default:
1340 WARN_ON_ONCE(1);
1341 ret = -EINVAL;
1342 goto register_error;
1343 }
1344
a67ba386
MD		 1345 list_add(&event_notifier->priv->node, &event_notifier_group->event_notifiers_head);
1346 hlist_add_head(&event_notifier->priv->hlist, head);
99f52fcc
FD		 1347
1348 /*
1349 * Clear the error counter bucket. The sessiond keeps track of which
ab04d7b1
MD		 1350 * bucket is currently in use. We trust it. The session lock
1351 * synchronizes against concurrent creation of the error
1352 * counter.
99f52fcc 1353 */
ab04d7b1
MD		 1354 error_counter = event_notifier_group->error_counter;
1355 if (error_counter) {
99f52fcc
FD		 1356 size_t dimension_index[1];
1357
1358 /*
1359 * Check that the index is within the boundary of the counter.
1360 */
a67ba386 1361 if (event_notifier->priv->error_counter_index >= event_notifier_group->error_counter_len) {
99f52fcc 1362 printk(KERN_INFO "LTTng: event_notifier: Error counter index out-of-bound: counter-len=%zu, index=%llu\n",
a67ba386 1363 event_notifier_group->error_counter_len, event_notifier->priv->error_counter_index);
99f52fcc
FD		 1364 ret = -EINVAL;
1365 goto register_error;
1366 }
1367
a67ba386 1368 dimension_index[0] = event_notifier->priv->error_counter_index;
ab04d7b1 1369 ret = error_counter->ops->counter_clear(error_counter->counter, dimension_index);
99f52fcc
FD		 1370 if (ret) {
1371 printk(KERN_INFO "LTTng: event_notifier: Unable to clear error counter bucket %llu\n",
a67ba386 1372 event_notifier->priv->error_counter_index);
99f52fcc
FD		 1373 goto register_error;
1374 }
1375 }
1376
dffef45d
FD		 1377 return event_notifier;
1378
1379register_error:
a67ba386
MD		 1380 kmem_cache_free(event_notifier_private_cache, event_notifier_priv);
1381cache_private_error:
dffef45d
FD		 1382 kmem_cache_free(event_notifier_cache, event_notifier);
1383cache_error:
1384exist:
1385type_error:
1386 return ERR_PTR(ret);
1387}
1388
99f52fcc
FD		 1389int lttng_kernel_counter_read(struct lttng_counter *counter,
1390 const size_t *dim_indexes, int32_t cpu,
1391 int64_t *val, bool *overflow, bool *underflow)
1392{
1393 return counter->ops->counter_read(counter->counter, dim_indexes,
1394 cpu, val, overflow, underflow);
1395}
1396
1397int lttng_kernel_counter_aggregate(struct lttng_counter *counter,
1398 const size_t *dim_indexes, int64_t *val,
1399 bool *overflow, bool *underflow)
1400{
1401 return counter->ops->counter_aggregate(counter->counter, dim_indexes,
1402 val, overflow, underflow);
1403}
1404
1405int lttng_kernel_counter_clear(struct lttng_counter *counter,
1406 const size_t *dim_indexes)
1407{
1408 return counter->ops->counter_clear(counter->counter, dim_indexes);
1409}
1410
1ae083ba 1411struct lttng_kernel_event_recorder *lttng_kernel_event_recorder_create(struct lttng_event_recorder_enabler *event_enabler,
def1e304 1412 const struct lttng_kernel_event_desc *event_desc)
3c997079 1413{
a67ba386 1414 struct lttng_kernel_event_recorder *event;
3c997079
MD		 1415
1416 mutex_lock(&sessions_mutex);
def1e304 1417 event = _lttng_kernel_event_recorder_create(event_enabler, event_desc);
3c997079
MD		 1418 mutex_unlock(&sessions_mutex);
1419 return event;
1420}
1421
a67ba386 1422struct lttng_kernel_event_notifier *lttng_event_notifier_create(
437d5aa5 1423 const struct lttng_kernel_event_desc *event_desc,
99f52fcc
FD		 1424 uint64_t id, uint64_t error_counter_index,
1425 struct lttng_event_notifier_group *event_notifier_group,
606828e4 1426 struct lttng_kernel_abi_event_notifier *event_notifier_param,
a67ba386 1427 enum lttng_kernel_abi_instrumentation itype)
dffef45d 1428{
a67ba386 1429 struct lttng_kernel_event_notifier *event_notifier;
dffef45d
FD		 1430
1431 mutex_lock(&sessions_mutex);
1432 event_notifier = _lttng_event_notifier_create(event_desc, id,
99f52fcc 1433 error_counter_index, event_notifier_group,
a67ba386 1434 event_notifier_param, itype);
dffef45d
FD		 1435 mutex_unlock(&sessions_mutex);
1436 return event_notifier;
1437}
1438
3c997079
MD		 1439/* Only used for tracepoints for now. */
1440static
96c59af1 1441void register_event_recorder(struct lttng_kernel_event_recorder *event_recorder)
3c997079 1442{
437d5aa5 1443 const struct lttng_kernel_event_desc *desc;
f127e61e 1444 int ret = -EINVAL;
3c997079 1445
a67ba386 1446 if (event_recorder->priv->parent.registered)
3c997079 1447 return;
4ecb5ad5 1448
a67ba386
MD		 1449 desc = event_recorder->priv->parent.desc;
1450 switch (event_recorder->priv->parent.instrumentation) {
606828e4 1451 case LTTNG_KERNEL_ABI_TRACEPOINT:
437d5aa5 1452 ret = lttng_wrapper_tracepoint_probe_register(desc->event_kname,
6c8c025b 1453 desc->tp_class->probe_callback,
14a15bad 1454 &event_recorder->parent);
4ecb5ad5 1455 break;
b4a75708 1456
606828e4 1457 case LTTNG_KERNEL_ABI_SYSCALL:
9b2f1c54 1458 ret = lttng_syscall_filter_enable_event(&event_recorder->parent);
4ecb5ad5 1459 break;
b4a75708 1460
c190d76e
MJ		 1461 case LTTNG_KERNEL_ABI_KPROBE:
1462 lttng_fallthrough;
1463 case LTTNG_KERNEL_ABI_UPROBE:
1464 lttng_fallthrough;
606828e4 1465 case LTTNG_KERNEL_ABI_KRETPROBE:
4ecb5ad5
MD		 1466 ret = 0;
1467 break;
b4a75708 1468
c190d76e
MJ		 1469 case LTTNG_KERNEL_ABI_FUNCTION:
1470 lttng_fallthrough;
1471 case LTTNG_KERNEL_ABI_NOOP:
1472 lttng_fallthrough;
4ecb5ad5
MD		 1473 default:
1474 WARN_ON_ONCE(1);
1475 }
3c997079 1476 if (!ret)
a67ba386 1477 event_recorder->priv->parent.registered = 1;
3c997079
MD		 1478}
1479
4e3c1b9b
MD		 1480/*
1481 * Only used internally at session destruction.
1482 */
96c59af1 1483int _lttng_event_recorder_unregister(struct lttng_kernel_event_recorder *event_recorder)
4e3c1b9b 1484{
a67ba386 1485 struct lttng_kernel_event_common_private *event_priv = &event_recorder->priv->parent;
437d5aa5 1486 const struct lttng_kernel_event_desc *desc;
11b5a3c2
MD		 1487 int ret = -EINVAL;
1488
a67ba386 1489 if (!event_priv->registered)
3c997079
MD		 1490 return 0;
1491
a67ba386
MD		 1492 desc = event_priv->desc;
1493 switch (event_priv->instrumentation) {
606828e4 1494 case LTTNG_KERNEL_ABI_TRACEPOINT:
a67ba386 1495 ret = lttng_wrapper_tracepoint_probe_unregister(event_priv->desc->event_kname,
6c8c025b 1496 event_priv->desc->tp_class->probe_callback,
14a15bad 1497 &event_recorder->parent);
baf20995 1498 break;
b4a75708 1499
606828e4 1500 case LTTNG_KERNEL_ABI_KPROBE:
a67ba386 1501 lttng_kprobes_unregister_event(event_recorder);
d6d808f3
MD		 1502 ret = 0;
1503 break;
b4a75708 1504
606828e4 1505 case LTTNG_KERNEL_ABI_KRETPROBE:
a67ba386 1506 lttng_kretprobes_unregister(event_recorder);
7371f44c
MD		 1507 ret = 0;
1508 break;
b4a75708 1509
606828e4 1510 case LTTNG_KERNEL_ABI_SYSCALL:
9b2f1c54 1511 ret = lttng_syscall_filter_disable_event(&event_recorder->parent);
4ecb5ad5 1512 break;
b4a75708 1513
606828e4 1514 case LTTNG_KERNEL_ABI_NOOP:
259b6cb3
MD		 1515 ret = 0;
1516 break;
b4a75708 1517
606828e4 1518 case LTTNG_KERNEL_ABI_UPROBE:
17806c46 1519 lttng_uprobes_unregister_event(&event_recorder->parent);
149b9a9d
YB		 1520 ret = 0;
1521 break;
b4a75708 1522
c190d76e
MJ		 1523 case LTTNG_KERNEL_ABI_FUNCTION:
1524 lttng_fallthrough;
baf20995
MD		 1525 default:
1526 WARN_ON_ONCE(1);
1527 }
3c997079 1528 if (!ret)
a67ba386 1529 event_priv->registered = 0;
dda6a249
MD		 1530 return ret;
1531}
1532
dffef45d
FD		 1533/* Only used for tracepoints for now. */
1534static
a67ba386 1535void register_event_notifier(struct lttng_kernel_event_notifier *event_notifier)
dffef45d 1536{
437d5aa5 1537 const struct lttng_kernel_event_desc *desc;
dffef45d
FD		 1538 int ret = -EINVAL;
1539
a67ba386 1540 if (event_notifier->priv->parent.registered)
dffef45d
FD		 1541 return;
1542
a67ba386
MD		 1543 desc = event_notifier->priv->parent.desc;
1544 switch (event_notifier->priv->parent.instrumentation) {
606828e4 1545 case LTTNG_KERNEL_ABI_TRACEPOINT:
437d5aa5 1546 ret = lttng_wrapper_tracepoint_probe_register(desc->event_kname,
6c8c025b 1547 desc->tp_class->probe_callback,
14a15bad 1548 &event_notifier->parent);
b01155ba 1549 break;
b4a75708 1550
606828e4 1551 case LTTNG_KERNEL_ABI_SYSCALL:
9b2f1c54 1552 ret = lttng_syscall_filter_enable_event(&event_notifier->parent);
8a8ac9a8 1553 break;
b4a75708 1554
c190d76e
MJ		 1555 case LTTNG_KERNEL_ABI_KPROBE:
1556 lttng_fallthrough;
606828e4 1557 case LTTNG_KERNEL_ABI_UPROBE:
2b16f0c9
FD		 1558 ret = 0;
1559 break;
b4a75708 1560
c190d76e
MJ		 1561 case LTTNG_KERNEL_ABI_KRETPROBE:
1562 lttng_fallthrough;
1563 case LTTNG_KERNEL_ABI_FUNCTION:
1564 lttng_fallthrough;
1565 case LTTNG_KERNEL_ABI_NOOP:
1566 lttng_fallthrough;
dffef45d
FD		 1567 default:
1568 WARN_ON_ONCE(1);
1569 }
1570 if (!ret)
a67ba386 1571 event_notifier->priv->parent.registered = 1;
dffef45d
FD		 1572}
1573
1574static
1575int _lttng_event_notifier_unregister(
a67ba386 1576 struct lttng_kernel_event_notifier *event_notifier)
dffef45d 1577{
437d5aa5 1578 const struct lttng_kernel_event_desc *desc;
dffef45d
FD		 1579 int ret = -EINVAL;
1580
a67ba386 1581 if (!event_notifier->priv->parent.registered)
dffef45d
FD		 1582 return 0;
1583
a67ba386
MD		 1584 desc = event_notifier->priv->parent.desc;
1585 switch (event_notifier->priv->parent.instrumentation) {
606828e4 1586 case LTTNG_KERNEL_ABI_TRACEPOINT:
a67ba386 1587 ret = lttng_wrapper_tracepoint_probe_unregister(event_notifier->priv->parent.desc->event_kname,
6c8c025b 1588 event_notifier->priv->parent.desc->tp_class->probe_callback,
14a15bad 1589 &event_notifier->parent);
b01155ba 1590 break;
b4a75708 1591
606828e4 1592 case LTTNG_KERNEL_ABI_KPROBE:
2b16f0c9
FD		 1593 lttng_kprobes_unregister_event_notifier(event_notifier);
1594 ret = 0;
1595 break;
b4a75708 1596
606828e4 1597 case LTTNG_KERNEL_ABI_UPROBE:
17806c46 1598 lttng_uprobes_unregister_event(&event_notifier->parent);
9de67196
FD		 1599 ret = 0;
1600 break;
b4a75708 1601
606828e4 1602 case LTTNG_KERNEL_ABI_SYSCALL:
9b2f1c54 1603 ret = lttng_syscall_filter_disable_event(&event_notifier->parent);
8a8ac9a8 1604 break;
b4a75708 1605
c190d76e
MJ		 1606 case LTTNG_KERNEL_ABI_KRETPROBE:
1607 lttng_fallthrough;
1608 case LTTNG_KERNEL_ABI_FUNCTION:
1609 lttng_fallthrough;
1610 case LTTNG_KERNEL_ABI_NOOP:
1611 lttng_fallthrough;
dffef45d
FD		 1612 default:
1613 WARN_ON_ONCE(1);
1614 }
1615 if (!ret)
a67ba386 1616 event_notifier->priv->parent.registered = 0;
dffef45d
FD		 1617 return ret;
1618}
1619
dda6a249
MD		 1620/*
1621 * Only used internally at session destruction.
1622 */
be066e6c 1623static
0648898e 1624void _lttng_event_destroy(struct lttng_kernel_event_common *event)
dda6a249 1625{
0648898e 1626 struct lttng_kernel_event_common_private *event_priv = event->priv;
966ad253
MD		 1627 struct lttng_enabler_ref *enabler_ref, *tmp_enabler_ref;
1628
0648898e
MD		 1629 lttng_free_event_filter_runtime(event);
1630 /* Free event enabler refs */
1631 list_for_each_entry_safe(enabler_ref, tmp_enabler_ref,
1632 &event_priv->enablers_ref_head, node)
1633 kfree(enabler_ref);
b4a75708 1634
0648898e
MD		 1635 switch (event->type) {
1636 case LTTNG_KERNEL_EVENT_TYPE_RECORDER:
1637 {
1638 struct lttng_kernel_event_recorder *event_recorder =
1639 container_of(event, struct lttng_kernel_event_recorder, parent);
b4a75708 1640
0648898e
MD		 1641 switch (event_priv->instrumentation) {
1642 case LTTNG_KERNEL_ABI_TRACEPOINT:
1643 lttng_event_desc_put(event_priv->desc);
1644 break;
b4a75708 1645
0648898e
MD		 1646 case LTTNG_KERNEL_ABI_KPROBE:
1647 module_put(event_priv->desc->owner);
1648 lttng_kprobes_destroy_event_private(event_recorder);
1649 break;
b4a75708 1650
0648898e
MD		 1651 case LTTNG_KERNEL_ABI_KRETPROBE:
1652 module_put(event_priv->desc->owner);
1653 lttng_kretprobes_destroy_private(event_recorder);
1654 break;
b4a75708 1655
0648898e
MD		 1656 case LTTNG_KERNEL_ABI_SYSCALL:
1657 break;
4e3c1b9b 1658
0648898e
MD		 1659 case LTTNG_KERNEL_ABI_UPROBE:
1660 module_put(event_priv->desc->owner);
17806c46 1661 lttng_uprobes_destroy_event_private(&event_recorder->parent);
0648898e 1662 break;
3ef5cc03 1663
c190d76e
MJ		 1664 case LTTNG_KERNEL_ABI_FUNCTION:
1665 lttng_fallthrough;
1666 case LTTNG_KERNEL_ABI_NOOP:
1667 lttng_fallthrough;
0648898e
MD		 1668 default:
1669 WARN_ON_ONCE(1);
1670 }
1671 list_del(&event_recorder->priv->node);
1672 kmem_cache_free(event_recorder_private_cache, event_recorder->priv);
1673 kmem_cache_free(event_recorder_cache, event_recorder);
b01155ba 1674 break;
0648898e
MD		 1675 }
1676 case LTTNG_KERNEL_EVENT_TYPE_NOTIFIER:
1677 {
1678 struct lttng_kernel_event_notifier *event_notifier =
1679 container_of(event, struct lttng_kernel_event_notifier, parent);
b4a75708 1680
0648898e
MD		 1681 switch (event_notifier->priv->parent.instrumentation) {
1682 case LTTNG_KERNEL_ABI_TRACEPOINT:
1683 lttng_event_desc_put(event_notifier->priv->parent.desc);
1684 break;
b4a75708 1685
0648898e
MD		 1686 case LTTNG_KERNEL_ABI_KPROBE:
1687 module_put(event_notifier->priv->parent.desc->owner);
1688 lttng_kprobes_destroy_event_notifier_private(event_notifier);
1689 break;
b4a75708 1690
0648898e
MD		 1691 case LTTNG_KERNEL_ABI_SYSCALL:
1692 break;
b4a75708 1693
0648898e
MD		 1694 case LTTNG_KERNEL_ABI_UPROBE:
1695 module_put(event_notifier->priv->parent.desc->owner);
17806c46 1696 lttng_uprobes_destroy_event_private(&event_notifier->parent);
0648898e
MD		 1697 break;
1698
c190d76e
MJ		 1699 case LTTNG_KERNEL_ABI_KRETPROBE:
1700 lttng_fallthrough;
1701 case LTTNG_KERNEL_ABI_FUNCTION:
1702 lttng_fallthrough;
1703 case LTTNG_KERNEL_ABI_NOOP:
1704 lttng_fallthrough;
0648898e
MD		 1705 default:
1706 WARN_ON_ONCE(1);
1707 }
1708 list_del(&event_notifier->priv->node);
1709 kmem_cache_free(event_notifier_private_cache, event_notifier->priv);
1710 kmem_cache_free(event_notifier_cache, event_notifier);
1711 break;
1712 }
dffef45d
FD		 1713 default:
1714 WARN_ON_ONCE(1);
1715 }
dffef45d
FD		 1716}
1717
8c0393c3 1718struct lttng_kernel_id_tracker *get_tracker(struct lttng_kernel_session *session,
d1f652f8 1719 enum tracker_type tracker_type)
e0130fab 1720{
d1f652f8
MD		 1721 switch (tracker_type) {
1722 case TRACKER_PID:
1723 return &session->pid_tracker;
1724 case TRACKER_VPID:
1725 return &session->vpid_tracker;
1726 case TRACKER_UID:
1727 return &session->uid_tracker;
1728 case TRACKER_VUID:
1729 return &session->vuid_tracker;
1730 case TRACKER_GID:
1731 return &session->gid_tracker;
1732 case TRACKER_VGID:
1733 return &session->vgid_tracker;
1734 default:
1735 WARN_ON_ONCE(1);
1736 return NULL;
1737 }
1738}
1739
8cdc1a81 1740int lttng_session_track_id(struct lttng_kernel_session *session,
d1f652f8
MD		 1741 enum tracker_type tracker_type, int id)
1742{
8c0393c3 1743 struct lttng_kernel_id_tracker *tracker;
e0130fab
MD		 1744 int ret;
1745
d1f652f8
MD		 1746 tracker = get_tracker(session, tracker_type);
1747 if (!tracker)
1748 return -EINVAL;
1749 if (id < -1)
e0130fab
MD		 1750 return -EINVAL;
1751 mutex_lock(&sessions_mutex);
d1f652f8
MD		 1752 if (id == -1) {
1753 /* track all ids: destroy tracker. */
1754 lttng_id_tracker_destroy(tracker, true);
e0130fab
MD		 1755 ret = 0;
1756 } else {
d1f652f8 1757 ret = lttng_id_tracker_add(tracker, id);
e0130fab 1758 }
e0130fab
MD		 1759 mutex_unlock(&sessions_mutex);
1760 return ret;
1761}
1762
8cdc1a81 1763int lttng_session_untrack_id(struct lttng_kernel_session *session,
d1f652f8 1764 enum tracker_type tracker_type, int id)
e0130fab 1765{
8c0393c3 1766 struct lttng_kernel_id_tracker *tracker;
e0130fab
MD		 1767 int ret;
1768
d1f652f8
MD		 1769 tracker = get_tracker(session, tracker_type);
1770 if (!tracker)
1771 return -EINVAL;
1772 if (id < -1)
e0130fab
MD		 1773 return -EINVAL;
1774 mutex_lock(&sessions_mutex);
d1f652f8
MD		 1775 if (id == -1) {
1776 /* untrack all ids: replace by empty tracker. */
1777 ret = lttng_id_tracker_empty_set(tracker);
e0130fab 1778 } else {
d1f652f8 1779 ret = lttng_id_tracker_del(tracker, id);
e0130fab 1780 }
e0130fab
MD		 1781 mutex_unlock(&sessions_mutex);
1782 return ret;
1783}
1784
7e6f9ef6 1785static
d1f652f8 1786void *id_list_start(struct seq_file *m, loff_t *pos)
7e6f9ef6 1787{
8c0393c3
MD		 1788 struct lttng_kernel_id_tracker *id_tracker = m->private;
1789 struct lttng_kernel_id_tracker_rcu *id_tracker_p = id_tracker->p;
d1f652f8 1790 struct lttng_id_hash_node *e;
7e6f9ef6
MD		 1791 int iter = 0, i;
1792
1793 mutex_lock(&sessions_mutex);
d1f652f8
MD		 1794 if (id_tracker_p) {
1795 for (i = 0; i < LTTNG_ID_TABLE_SIZE; i++) {
1796 struct hlist_head *head = &id_tracker_p->id_hash[i];
7e6f9ef6 1797
f934e302 1798 lttng_hlist_for_each_entry(e, head, hlist) {
7e6f9ef6
MD		 1799 if (iter++ >= *pos)
1800 return e;
1801 }
1802 }
1803 } else {
d1f652f8 1804 /* ID tracker disabled. */
7e6f9ef6 1805 if (iter >= *pos && iter == 0) {
d1f652f8 1806 return id_tracker_p; /* empty tracker */
7e6f9ef6
MD		 1807 }
1808 iter++;
1809 }
1810 /* End of list */
1811 return NULL;
1812}
1813
1814/* Called with sessions_mutex held. */
1815static
d1f652f8 1816void *id_list_next(struct seq_file *m, void *p, loff_t *ppos)
7e6f9ef6 1817{
8c0393c3
MD		 1818 struct lttng_kernel_id_tracker *id_tracker = m->private;
1819 struct lttng_kernel_id_tracker_rcu *id_tracker_p = id_tracker->p;
d1f652f8 1820 struct lttng_id_hash_node *e;
7e6f9ef6
MD		 1821 int iter = 0, i;
1822
1823 (*ppos)++;
d1f652f8
MD		 1824 if (id_tracker_p) {
1825 for (i = 0; i < LTTNG_ID_TABLE_SIZE; i++) {
1826 struct hlist_head *head = &id_tracker_p->id_hash[i];
7e6f9ef6 1827
f934e302 1828 lttng_hlist_for_each_entry(e, head, hlist) {
7e6f9ef6
MD		 1829 if (iter++ >= *ppos)
1830 return e;
1831 }
1832 }
1833 } else {
d1f652f8 1834 /* ID tracker disabled. */
7e6f9ef6 1835 if (iter >= *ppos && iter == 0)
d1f652f8 1836 return p; /* empty tracker */
7e6f9ef6
MD		 1837 iter++;
1838 }
1839
1840 /* End of list */
1841 return NULL;
1842}
1843
1844static
d1f652f8 1845void id_list_stop(struct seq_file *m, void *p)
7e6f9ef6
MD		 1846{
1847 mutex_unlock(&sessions_mutex);
1848}
1849
1850static
d1f652f8 1851int id_list_show(struct seq_file *m, void *p)
7e6f9ef6 1852{
8c0393c3
MD		 1853 struct lttng_kernel_id_tracker *id_tracker = m->private;
1854 struct lttng_kernel_id_tracker_rcu *id_tracker_p = id_tracker->p;
d1f652f8 1855 int id;
7e6f9ef6 1856
d1f652f8 1857 if (p == id_tracker_p) {
7e6f9ef6 1858 /* Tracker disabled. */
d1f652f8 1859 id = -1;
7e6f9ef6 1860 } else {
d1f652f8 1861 const struct lttng_id_hash_node *e = p;
7e6f9ef6 1862
d1f652f8
MD		 1863 id = lttng_id_tracker_get_node_id(e);
1864 }
8c0393c3 1865 switch (id_tracker->priv->tracker_type) {
d1f652f8
MD		 1866 case TRACKER_PID:
1867 seq_printf(m, "process { pid = %d; };\n", id);
1868 break;
1869 case TRACKER_VPID:
1870 seq_printf(m, "process { vpid = %d; };\n", id);
1871 break;
1872 case TRACKER_UID:
1873 seq_printf(m, "user { uid = %d; };\n", id);
1874 break;
1875 case TRACKER_VUID:
1876 seq_printf(m, "user { vuid = %d; };\n", id);
1877 break;
1878 case TRACKER_GID:
1879 seq_printf(m, "group { gid = %d; };\n", id);
1880 break;
1881 case TRACKER_VGID:
1882 seq_printf(m, "group { vgid = %d; };\n", id);
1883 break;
1884 default:
1885 seq_printf(m, "UNKNOWN { field = %d };\n", id);
7e6f9ef6 1886 }
7e6f9ef6
MD		 1887 return 0;
1888}
1889
1890static
d1f652f8
MD		 1891const struct seq_operations lttng_tracker_ids_list_seq_ops = {
1892 .start = id_list_start,
1893 .next = id_list_next,
1894 .stop = id_list_stop,
1895 .show = id_list_show,
7e6f9ef6
MD		 1896};
1897
1898static
d1f652f8 1899int lttng_tracker_ids_list_open(struct inode *inode, struct file *file)
7e6f9ef6 1900{
d1f652f8 1901 return seq_open(file, &lttng_tracker_ids_list_seq_ops);
7e6f9ef6
MD		 1902}
1903
1904static
d1f652f8 1905int lttng_tracker_ids_list_release(struct inode *inode, struct file *file)
7e6f9ef6
MD		 1906{
1907 struct seq_file *m = file->private_data;
8c0393c3 1908 struct lttng_kernel_id_tracker *id_tracker = m->private;
7e6f9ef6
MD		 1909 int ret;
1910
d1f652f8 1911 WARN_ON_ONCE(!id_tracker);
7e6f9ef6 1912 ret = seq_release(inode, file);
d1f652f8 1913 if (!ret)
8c0393c3 1914 fput(id_tracker->priv->session->priv->file);
7e6f9ef6
MD		 1915 return ret;
1916}
1917
d1f652f8 1918const struct file_operations lttng_tracker_ids_list_fops = {
7e6f9ef6 1919 .owner = THIS_MODULE,
d1f652f8 1920 .open = lttng_tracker_ids_list_open,
7e6f9ef6
MD		 1921 .read = seq_read,
1922 .llseek = seq_lseek,
d1f652f8 1923 .release = lttng_tracker_ids_list_release,
7e6f9ef6
MD		 1924};
1925
8cdc1a81 1926int lttng_session_list_tracker_ids(struct lttng_kernel_session *session,
d1f652f8 1927 enum tracker_type tracker_type)
7e6f9ef6 1928{
d1f652f8 1929 struct file *tracker_ids_list_file;
7e6f9ef6
MD		 1930 struct seq_file *m;
1931 int file_fd, ret;
1932
1933 file_fd = lttng_get_unused_fd();
1934 if (file_fd < 0) {
1935 ret = file_fd;
1936 goto fd_error;
1937 }
1938
d1f652f8
MD		 1939 tracker_ids_list_file = anon_inode_getfile("[lttng_tracker_ids_list]",
1940 &lttng_tracker_ids_list_fops,
7e6f9ef6 1941 NULL, O_RDWR);
d1f652f8
MD		 1942 if (IS_ERR(tracker_ids_list_file)) {
1943 ret = PTR_ERR(tracker_ids_list_file);
7e6f9ef6
MD		 1944 goto file_error;
1945 }
8cdc1a81 1946 if (!atomic_long_add_unless(&session->priv->file->f_count, 1, LONG_MAX)) {
98d7281c 1947 ret = -EOVERFLOW;
9c1f4643
MD		 1948 goto refcount_error;
1949 }
d1f652f8 1950 ret = lttng_tracker_ids_list_fops.open(NULL, tracker_ids_list_file);
7e6f9ef6
MD		 1951 if (ret < 0)
1952 goto open_error;
d1f652f8
MD		 1953 m = tracker_ids_list_file->private_data;
1954
1955 m->private = get_tracker(session, tracker_type);
1956 BUG_ON(!m->private);
1957 fd_install(file_fd, tracker_ids_list_file);
7e6f9ef6
MD		 1958
1959 return file_fd;
1960
1961open_error:
8cdc1a81 1962 atomic_long_dec(&session->priv->file->f_count);
9c1f4643 1963refcount_error:
d1f652f8 1964 fput(tracker_ids_list_file);
7e6f9ef6
MD		 1965file_error:
1966 put_unused_fd(file_fd);
1967fd_error:
1968 return ret;
1969}
1970
3c997079
MD		 1971/*
1972 * Enabler management.
1973 */
1974static
4993071a
PP		 1975int lttng_match_enabler_star_glob(const char *desc_name,
1976 const char *pattern)
3c997079 1977{
cbc19040
MD		 1978 if (!strutils_star_glob_match(pattern, LTTNG_SIZE_MAX,
1979 desc_name, LTTNG_SIZE_MAX))
3c997079
MD		 1980 return 0;
1981 return 1;
1982}
1983
1984static
4ecb5ad5
MD		 1985int lttng_match_enabler_name(const char *desc_name,
1986 const char *name)
3c997079 1987{
4ecb5ad5 1988 if (strcmp(desc_name, name))
3c997079
MD		 1989 return 0;
1990 return 1;
1991}
1992
437d5aa5 1993int lttng_desc_match_enabler(const struct lttng_kernel_event_desc *desc,
8d554755 1994 struct lttng_event_enabler_common *enabler)
3c997079 1995{
4ecb5ad5 1996 const char *desc_name, *enabler_name;
badfe9f5 1997 bool compat = false, entry = false;
4ecb5ad5
MD		 1998
1999 enabler_name = enabler->event_param.name;
2000 switch (enabler->event_param.instrumentation) {
606828e4 2001 case LTTNG_KERNEL_ABI_TRACEPOINT:
437d5aa5 2002 desc_name = desc->event_name;
3b861b22
FD		 2003 switch (enabler->format_type) {
2004 case LTTNG_ENABLER_FORMAT_STAR_GLOB:
badfe9f5 2005 return lttng_match_enabler_star_glob(desc_name, enabler_name);
3b861b22 2006 case LTTNG_ENABLER_FORMAT_NAME:
badfe9f5
MD		 2007 return lttng_match_enabler_name(desc_name, enabler_name);
2008 default:
2009 return -EINVAL;
2010 }
4ecb5ad5 2011 break;
b4a75708 2012
606828e4 2013 case LTTNG_KERNEL_ABI_SYSCALL:
437d5aa5 2014 desc_name = desc->event_name;
badfe9f5 2015 if (!strncmp(desc_name, "compat_", strlen("compat_"))) {
4ecb5ad5 2016 desc_name += strlen("compat_");
badfe9f5
MD		 2017 compat = true;
2018 }
4ecb5ad5
MD		 2019 if (!strncmp(desc_name, "syscall_exit_",
2020 strlen("syscall_exit_"))) {
2021 desc_name += strlen("syscall_exit_");
2022 } else if (!strncmp(desc_name, "syscall_entry_",
2023 strlen("syscall_entry_"))) {
2024 desc_name += strlen("syscall_entry_");
badfe9f5 2025 entry = true;
4ecb5ad5
MD		 2026 } else {
2027 WARN_ON_ONCE(1);
2028 return -EINVAL;
2029 }
badfe9f5 2030 switch (enabler->event_param.u.syscall.entryexit) {
606828e4 2031 case LTTNG_KERNEL_ABI_SYSCALL_ENTRYEXIT:
badfe9f5 2032 break;
606828e4 2033 case LTTNG_KERNEL_ABI_SYSCALL_ENTRY:
badfe9f5
MD		 2034 if (!entry)
2035 return 0;
2036 break;
606828e4 2037 case LTTNG_KERNEL_ABI_SYSCALL_EXIT:
badfe9f5
MD		 2038 if (entry)
2039 return 0;
2040 break;
2041 default:
2042 return -EINVAL;
2043 }
2044 switch (enabler->event_param.u.syscall.abi) {
606828e4 2045 case LTTNG_KERNEL_ABI_SYSCALL_ABI_ALL:
badfe9f5 2046 break;
606828e4 2047 case LTTNG_KERNEL_ABI_SYSCALL_ABI_NATIVE:
badfe9f5
MD		 2048 if (compat)
2049 return 0;
2050 break;
606828e4 2051 case LTTNG_KERNEL_ABI_SYSCALL_ABI_COMPAT:
badfe9f5
MD		 2052 if (!compat)
2053 return 0;
2054 break;
2055 default:
2056 return -EINVAL;
2057 }
2058 switch (enabler->event_param.u.syscall.match) {
606828e4 2059 case LTTNG_KERNEL_ABI_SYSCALL_MATCH_NAME:
3b861b22
FD		 2060 switch (enabler->format_type) {
2061 case LTTNG_ENABLER_FORMAT_STAR_GLOB:
badfe9f5 2062 return lttng_match_enabler_star_glob(desc_name, enabler_name);
3b861b22 2063 case LTTNG_ENABLER_FORMAT_NAME:
badfe9f5
MD		 2064 return lttng_match_enabler_name(desc_name, enabler_name);
2065 default:
2066 return -EINVAL;
2067 }
2068 break;
606828e4 2069 case LTTNG_KERNEL_ABI_SYSCALL_MATCH_NR:
badfe9f5
MD		 2070 return -EINVAL; /* Not implemented. */
2071 default:
2072 return -EINVAL;
2073 }
4ecb5ad5 2074 break;
b4a75708 2075
4ecb5ad5
MD		 2076 default:
2077 WARN_ON_ONCE(1);
2078 return -EINVAL;
2079 }
3c997079
MD		 2080}
2081
2082static
1ae083ba 2083int lttng_event_enabler_match_event(struct lttng_event_recorder_enabler *event_enabler,
a67ba386 2084 struct lttng_kernel_event_recorder *event_recorder)
3c997079 2085{
1ae083ba 2086 struct lttng_event_enabler_common *base_enabler = lttng_event_recorder_enabler_as_enabler(
b2bc0bc8
FD		 2087 event_enabler);
2088
a67ba386 2089 if (base_enabler->event_param.instrumentation != event_recorder->priv->parent.instrumentation)
33a39a3c 2090 return 0;
a67ba386
MD		 2091 if (lttng_desc_match_enabler(event_recorder->priv->parent.desc, base_enabler)
2092 && event_recorder->chan == event_enabler->chan)
3c997079
MD		 2093 return 1;
2094 else
2095 return 0;
2096}
2097
b01155ba
FD		 2098static
2099int lttng_event_notifier_enabler_match_event_notifier(struct lttng_event_notifier_enabler *event_notifier_enabler,
a67ba386 2100 struct lttng_kernel_event_notifier *event_notifier)
b01155ba 2101{
8d554755 2102 struct lttng_event_enabler_common *base_enabler = lttng_event_notifier_enabler_as_enabler(
b01155ba
FD		 2103 event_notifier_enabler);
2104
a67ba386 2105 if (base_enabler->event_param.instrumentation != event_notifier->priv->parent.instrumentation)
b01155ba 2106 return 0;
a67ba386
MD		 2107 if (lttng_desc_match_enabler(event_notifier->priv->parent.desc, base_enabler)
2108 && event_notifier->priv->group == event_notifier_enabler->group
5b9817ae 2109 && event_notifier->priv->parent.user_token == event_notifier_enabler->parent.user_token)
b01155ba
FD		 2110 return 1;
2111 else
2112 return 0;
2113}
2114
3c997079 2115static
b2bc0bc8
FD		 2116struct lttng_enabler_ref *lttng_enabler_ref(
2117 struct list_head *enablers_ref_list,
8d554755 2118 struct lttng_event_enabler_common *enabler)
3c997079
MD		 2119{
2120 struct lttng_enabler_ref *enabler_ref;
2121
b2bc0bc8 2122 list_for_each_entry(enabler_ref, enablers_ref_list, node) {
3c997079
MD		 2123 if (enabler_ref->ref == enabler)
2124 return enabler_ref;
2125 }
2126 return NULL;
2127}
2128
3c997079 2129static
1ae083ba 2130void lttng_create_tracepoint_event_if_missing(struct lttng_event_recorder_enabler *event_enabler)
3c997079 2131{
f7d06400 2132 struct lttng_kernel_session *session = event_enabler->chan->parent.session;
437d5aa5
MD		 2133 struct lttng_kernel_probe_desc *probe_desc;
2134 const struct lttng_kernel_event_desc *desc;
3c997079
MD		 2135 int i;
2136 struct list_head *probe_list;
2137
2138 probe_list = lttng_get_probe_list_head();
2139 /*
2140 * For each probe event, if we find that a probe event matches
2141 * our enabler, create an associated lttng_event if not
2142 * already present.
2143 */
2144 list_for_each_entry(probe_desc, probe_list, head) {
2145 for (i = 0; i < probe_desc->nr_events; i++) {
2146 int found = 0;
2147 struct hlist_head *head;
a67ba386
MD		 2148 struct lttng_kernel_event_recorder_private *event_recorder_private;
2149 struct lttng_kernel_event_recorder *event_recorder;
3c997079
MD		 2150
2151 desc = probe_desc->event_desc[i];
b2bc0bc8 2152 if (!lttng_desc_match_enabler(desc,
1ae083ba 2153 lttng_event_recorder_enabler_as_enabler(event_enabler)))
3c997079 2154 continue;
3c997079
MD		 2155
2156 /*
2157 * Check if already created.
2158 */
4a11ef1c 2159 head = utils_borrow_hash_table_bucket(
8cdc1a81 2160 session->priv->events_ht.table, LTTNG_EVENT_HT_SIZE,
437d5aa5 2161 desc->event_name);
a67ba386
MD		 2162 lttng_hlist_for_each_entry(event_recorder_private, head, hlist) {
2163 if (event_recorder_private->parent.desc == desc
2164 && event_recorder_private->pub->chan == event_enabler->chan)
3c997079
MD		 2165 found = 1;
2166 }
2167 if (found)
2168 continue;
2169
2170 /*
2171 * We need to create an event for this
2172 * event probe.
2173 */
def1e304 2174 event_recorder = _lttng_kernel_event_recorder_create(event_enabler, desc);
a67ba386 2175 if (!event_recorder) {
5a15f70c 2176 printk(KERN_INFO "LTTng: Unable to create event %s\n",
437d5aa5 2177 probe_desc->event_desc[i]->event_name);
3c997079
MD		 2178 }
2179 }
2180 }
2181}
2182
b01155ba
FD		 2183static
2184void lttng_create_tracepoint_event_notifier_if_missing(struct lttng_event_notifier_enabler *event_notifier_enabler)
2185{
2186 struct lttng_event_notifier_group *event_notifier_group = event_notifier_enabler->group;
437d5aa5
MD		 2187 struct lttng_kernel_probe_desc *probe_desc;
2188 const struct lttng_kernel_event_desc *desc;
b01155ba
FD		 2189 int i;
2190 struct list_head *probe_list;
2191
2192 probe_list = lttng_get_probe_list_head();
2193 /*
2194 * For each probe event, if we find that a probe event matches
2195 * our enabler, create an associated lttng_event_notifier if not
2196 * already present.
2197 */
2198 list_for_each_entry(probe_desc, probe_list, head) {
2199 for (i = 0; i < probe_desc->nr_events; i++) {
2200 int found = 0;
2201 struct hlist_head *head;
a67ba386
MD		 2202 struct lttng_kernel_event_notifier_private *event_notifier_priv;
2203 struct lttng_kernel_event_notifier *event_notifier;
b01155ba
FD		 2204
2205 desc = probe_desc->event_desc[i];
2206 if (!lttng_desc_match_enabler(desc,
2207 lttng_event_notifier_enabler_as_enabler(event_notifier_enabler)))
2208 continue;
2209
2210 /*
2211 * Check if already created.
2212 */
2213 head = utils_borrow_hash_table_bucket(
2214 event_notifier_group->event_notifiers_ht.table,
437d5aa5 2215 LTTNG_EVENT_NOTIFIER_HT_SIZE, desc->event_name);
a67ba386
MD		 2216 lttng_hlist_for_each_entry(event_notifier_priv, head, hlist) {
2217 if (event_notifier_priv->parent.desc == desc
5b9817ae 2218 && event_notifier_priv->parent.user_token == event_notifier_enabler->parent.user_token)
b01155ba
FD		 2219 found = 1;
2220 }
2221 if (found)
2222 continue;
2223
2224 /*
2225 * We need to create a event_notifier for this event probe.
2226 */
2227 event_notifier = _lttng_event_notifier_create(desc,
5b9817ae 2228 event_notifier_enabler->parent.user_token,
99f52fcc 2229 event_notifier_enabler->error_counter_index,
a67ba386 2230 event_notifier_group, NULL,
606828e4 2231 LTTNG_KERNEL_ABI_TRACEPOINT);
b01155ba
FD		 2232 if (IS_ERR(event_notifier)) {
2233 printk(KERN_INFO "Unable to create event_notifier %s\n",
437d5aa5 2234 probe_desc->event_desc[i]->event_name);
b01155ba
FD		 2235 }
2236 }
2237 }
2238}
2239
33a39a3c 2240static
1ae083ba 2241void lttng_create_syscall_event_if_missing(struct lttng_event_recorder_enabler *event_enabler)
33a39a3c
MD		 2242{
2243 int ret;
2244
d26015f6 2245 ret = lttng_syscalls_register_event(&event_enabler->parent);
33a39a3c
MD		 2246 WARN_ON_ONCE(ret);
2247}
2248
8a8ac9a8
FD		 2249static
2250void lttng_create_syscall_event_notifier_if_missing(struct lttng_event_notifier_enabler *event_notifier_enabler)
2251{
2252 int ret;
2253
d26015f6 2254 ret = lttng_syscalls_register_event(&event_notifier_enabler->parent);
8a8ac9a8 2255 WARN_ON_ONCE(ret);
a67ba386 2256 ret = lttng_syscalls_create_matching_event_notifiers(event_notifier_enabler);
8a8ac9a8
FD		 2257 WARN_ON_ONCE(ret);
2258}
2259
33a39a3c 2260/*
a67ba386 2261 * Create struct lttng_kernel_event_recorder if it is missing and present in the list of
33a39a3c
MD		 2262 * tracepoint probes.
2263 * Should be called with sessions mutex held.
2264 */
2265static
1ae083ba 2266void lttng_create_event_if_missing(struct lttng_event_recorder_enabler *event_enabler)
33a39a3c 2267{
5b9817ae 2268 switch (event_enabler->parent.event_param.instrumentation) {
606828e4 2269 case LTTNG_KERNEL_ABI_TRACEPOINT:
1b917d6b 2270 lttng_create_tracepoint_event_if_missing(event_enabler);
33a39a3c 2271 break;
b4a75708 2272
606828e4 2273 case LTTNG_KERNEL_ABI_SYSCALL:
1b917d6b 2274 lttng_create_syscall_event_if_missing(event_enabler);
33a39a3c 2275 break;
b4a75708 2276
33a39a3c
MD		 2277 default:
2278 WARN_ON_ONCE(1);
2279 break;
2280 }
2281}
2282
3c997079 2283/*
b2bc0bc8 2284 * Create events associated with an event_enabler (if not already present),
3c997079
MD		 2285 * and add backward reference from the event to the enabler.
2286 * Should be called with sessions mutex held.
2287 */
2288static
1ae083ba 2289int lttng_event_enabler_ref_events(struct lttng_event_recorder_enabler *event_enabler)
3c997079 2290{
f7d06400
MD		 2291 struct lttng_kernel_channel_buffer *chan = event_enabler->chan;
2292 struct lttng_kernel_session *session = event_enabler->chan->parent.session;
1ae083ba 2293 struct lttng_event_enabler_common *base_enabler = lttng_event_recorder_enabler_as_enabler(event_enabler);
a67ba386 2294 struct lttng_kernel_event_recorder_private *event_recorder_priv;
3c997079 2295
606828e4
MD		 2296 if (base_enabler->event_param.instrumentation == LTTNG_KERNEL_ABI_SYSCALL &&
2297 base_enabler->event_param.u.syscall.abi == LTTNG_KERNEL_ABI_SYSCALL_ABI_ALL &&
2298 base_enabler->event_param.u.syscall.match == LTTNG_KERNEL_ABI_SYSCALL_MATCH_NAME &&
b2bc0bc8 2299 !strcmp(base_enabler->event_param.name, "*")) {
3b82c4e1 2300 int enabled = base_enabler->enabled;
606828e4 2301 enum lttng_kernel_abi_syscall_entryexit entryexit = base_enabler->event_param.u.syscall.entryexit;
3b82c4e1 2302
606828e4 2303 if (entryexit == LTTNG_KERNEL_ABI_SYSCALL_ENTRY || entryexit == LTTNG_KERNEL_ABI_SYSCALL_ENTRYEXIT)
0bb716a8 2304 WRITE_ONCE(chan->priv->parent.syscall_table.syscall_all_entry, enabled);
3b82c4e1 2305
606828e4 2306 if (entryexit == LTTNG_KERNEL_ABI_SYSCALL_EXIT || entryexit == LTTNG_KERNEL_ABI_SYSCALL_ENTRYEXIT)
0bb716a8 2307 WRITE_ONCE(chan->priv->parent.syscall_table.syscall_all_exit, enabled);
badfe9f5
MD		 2308 }
2309
3c997079 2310 /* First ensure that probe events are created for this enabler. */
b2bc0bc8 2311 lttng_create_event_if_missing(event_enabler);
3c997079 2312
b2bc0bc8 2313 /* For each event matching event_enabler in session event list. */
8cdc1a81 2314 list_for_each_entry(event_recorder_priv, &session->priv->events, node) {
a67ba386 2315 struct lttng_kernel_event_recorder *event_recorder = event_recorder_priv->pub;
3c997079
MD		 2316 struct lttng_enabler_ref *enabler_ref;
2317
a67ba386 2318 if (!lttng_event_enabler_match_event(event_enabler, event_recorder))
3c997079 2319 continue;
a67ba386 2320 enabler_ref = lttng_enabler_ref(&event_recorder_priv->parent.enablers_ref_head,
1ae083ba 2321 lttng_event_recorder_enabler_as_enabler(event_enabler));
3c997079
MD		 2322 if (!enabler_ref) {
2323 /*
2324 * If no backward ref, create it.
b2bc0bc8 2325 * Add backward ref from event to event_enabler.
3c997079
MD		 2326 */
2327 enabler_ref = kzalloc(sizeof(*enabler_ref), GFP_KERNEL);
2328 if (!enabler_ref)
2329 return -ENOMEM;
1ae083ba 2330 enabler_ref->ref = lttng_event_recorder_enabler_as_enabler(event_enabler);
3c997079 2331 list_add(&enabler_ref->node,
a67ba386 2332 &event_recorder_priv->parent.enablers_ref_head);
3c997079
MD		 2333 }
2334
f127e61e
MD		 2335 /*
2336 * Link filter bytecodes if not linked yet.
2337 */
a67ba386 2338 lttng_enabler_link_bytecode(event_recorder_priv->parent.desc,
2dfda770 2339 lttng_static_ctx,
a67ba386 2340 &event_recorder_priv->parent.filter_bytecode_runtime_head,
1ae083ba 2341 &lttng_event_recorder_enabler_as_enabler(event_enabler)->filter_bytecode_head);
3c997079
MD		 2342 }
2343 return 0;
2344}
2345
b01155ba 2346/*
a67ba386 2347 * Create struct lttng_kernel_event_notifier if it is missing and present in the list of
b01155ba
FD		 2348 * tracepoint probes.
2349 * Should be called with sessions mutex held.
2350 */
2351static
2352void lttng_create_event_notifier_if_missing(struct lttng_event_notifier_enabler *event_notifier_enabler)
2353{
5b9817ae 2354 switch (event_notifier_enabler->parent.event_param.instrumentation) {
606828e4 2355 case LTTNG_KERNEL_ABI_TRACEPOINT:
b01155ba
FD		 2356 lttng_create_tracepoint_event_notifier_if_missing(event_notifier_enabler);
2357 break;
b4a75708 2358
606828e4 2359 case LTTNG_KERNEL_ABI_SYSCALL:
8a8ac9a8
FD		 2360 lttng_create_syscall_event_notifier_if_missing(event_notifier_enabler);
2361 break;
b4a75708 2362
b01155ba
FD		 2363 default:
2364 WARN_ON_ONCE(1);
2365 break;
2366 }
2367}
2368
2369/*
2370 * Create event_notifiers associated with a event_notifier enabler (if not already present).
2371 */
2372static
8a8ac9a8
FD		 2373int lttng_event_notifier_enabler_ref_event_notifiers(
2374 struct lttng_event_notifier_enabler *event_notifier_enabler)
b01155ba
FD		 2375{
2376 struct lttng_event_notifier_group *event_notifier_group = event_notifier_enabler->group;
8d554755 2377 struct lttng_event_enabler_common *base_enabler = lttng_event_notifier_enabler_as_enabler(event_notifier_enabler);
a67ba386 2378 struct lttng_kernel_event_notifier_private *event_notifier_priv;
b01155ba 2379
606828e4
MD		 2380 if (base_enabler->event_param.instrumentation == LTTNG_KERNEL_ABI_SYSCALL &&
2381 base_enabler->event_param.u.syscall.abi == LTTNG_KERNEL_ABI_SYSCALL_ABI_ALL &&
2382 base_enabler->event_param.u.syscall.match == LTTNG_KERNEL_ABI_SYSCALL_MATCH_NAME &&
8a8ac9a8
FD		 2383 !strcmp(base_enabler->event_param.name, "*")) {
2384
2385 int enabled = base_enabler->enabled;
606828e4 2386 enum lttng_kernel_abi_syscall_entryexit entryexit = base_enabler->event_param.u.syscall.entryexit;
8a8ac9a8 2387
606828e4 2388 if (entryexit == LTTNG_KERNEL_ABI_SYSCALL_ENTRY || entryexit == LTTNG_KERNEL_ABI_SYSCALL_ENTRYEXIT)
0bb716a8 2389 WRITE_ONCE(event_notifier_group->syscall_table.syscall_all_entry, enabled);
8a8ac9a8 2390
606828e4 2391 if (entryexit == LTTNG_KERNEL_ABI_SYSCALL_EXIT || entryexit == LTTNG_KERNEL_ABI_SYSCALL_ENTRYEXIT)
0bb716a8 2392 WRITE_ONCE(event_notifier_group->syscall_table.syscall_all_exit, enabled);
8a8ac9a8
FD		 2393
2394 }
2395
b01155ba
FD		 2396 /* First ensure that probe event_notifiers are created for this enabler. */
2397 lttng_create_event_notifier_if_missing(event_notifier_enabler);
2398
2399 /* Link the created event_notifier with its associated enabler. */
a67ba386
MD		 2400 list_for_each_entry(event_notifier_priv, &event_notifier_group->event_notifiers_head, node) {
2401 struct lttng_kernel_event_notifier *event_notifier = event_notifier_priv->pub;
b01155ba
FD		 2402 struct lttng_enabler_ref *enabler_ref;
2403
2404 if (!lttng_event_notifier_enabler_match_event_notifier(event_notifier_enabler, event_notifier))
2405 continue;
2406
a67ba386 2407 enabler_ref = lttng_enabler_ref(&event_notifier_priv->parent.enablers_ref_head,
b01155ba
FD		 2408 lttng_event_notifier_enabler_as_enabler(event_notifier_enabler));
2409 if (!enabler_ref) {
2410 /*
2411 * If no backward ref, create it.
2412 * Add backward ref from event_notifier to enabler.
2413 */
2414 enabler_ref = kzalloc(sizeof(*enabler_ref), GFP_KERNEL);
2415 if (!enabler_ref)
2416 return -ENOMEM;
2417
2418 enabler_ref->ref = lttng_event_notifier_enabler_as_enabler(
2419 event_notifier_enabler);
2420 list_add(&enabler_ref->node,
a67ba386 2421 &event_notifier_priv->parent.enablers_ref_head);
b01155ba
FD		 2422 }
2423
2424 /*
2425 * Link filter bytecodes if not linked yet.
2426 */
a67ba386
MD		 2427 lttng_enabler_link_bytecode(event_notifier_priv->parent.desc,
2428 lttng_static_ctx, &event_notifier_priv->parent.filter_bytecode_runtime_head,
60206944 2429 &lttng_event_notifier_enabler_as_enabler(event_notifier_enabler)->filter_bytecode_head);
99d223ad
FD		 2430
2431 /* Link capture bytecodes if not linked yet. */
a67ba386
MD		 2432 lttng_enabler_link_bytecode(event_notifier_priv->parent.desc,
2433 lttng_static_ctx, &event_notifier_priv->capture_bytecode_runtime_head,
99d223ad
FD		 2434 &event_notifier_enabler->capture_bytecode_head);
2435
a67ba386 2436 event_notifier_priv->num_captures = event_notifier_enabler->num_captures;
b01155ba
FD		 2437 }
2438 return 0;
2439}
2440
3c997079
MD		 2441/*
2442 * Called at module load: connect the probe on all enablers matching
2443 * this event.
2444 * Called with sessions lock held.
2445 */
2446int lttng_fix_pending_events(void)
2447{
8cdc1a81 2448 struct lttng_kernel_session_private *session_priv;
3c997079 2449
8cdc1a81
MD		 2450 list_for_each_entry(session_priv, &sessions, list)
2451 lttng_session_lazy_sync_event_enablers(session_priv->pub);
3c997079
MD		 2452 return 0;
2453}
2454
b01155ba
FD		 2455static bool lttng_event_notifier_group_has_active_event_notifiers(
2456 struct lttng_event_notifier_group *event_notifier_group)
2457{
2458 struct lttng_event_notifier_enabler *event_notifier_enabler;
2459
2460 list_for_each_entry(event_notifier_enabler, &event_notifier_group->enablers_head,
2461 node) {
5b9817ae 2462 if (event_notifier_enabler->parent.enabled)
b01155ba
FD		 2463 return true;
2464 }
2465 return false;
2466}
2467
2468bool lttng_event_notifier_active(void)
2469{
2470 struct lttng_event_notifier_group *event_notifier_group;
2471
2472 list_for_each_entry(event_notifier_group, &event_notifier_groups, node) {
2473 if (lttng_event_notifier_group_has_active_event_notifiers(event_notifier_group))
2474 return true;
2475 }
2476 return false;
2477}
2478
2479int lttng_fix_pending_event_notifiers(void)
2480{
2481 struct lttng_event_notifier_group *event_notifier_group;
2482
2483 list_for_each_entry(event_notifier_group, &event_notifier_groups, node)
2484 lttng_event_notifier_group_sync_enablers(event_notifier_group);
2485 return 0;
2486}
2487
1ae083ba 2488struct lttng_event_recorder_enabler *lttng_event_recorder_enabler_create(
b2bc0bc8 2489 enum lttng_enabler_format_type format_type,
606828e4 2490 struct lttng_kernel_abi_event *event_param,
f7d06400 2491 struct lttng_kernel_channel_buffer *chan)
3c997079 2492{
1ae083ba 2493 struct lttng_event_recorder_enabler *event_enabler;
3c997079 2494
b2bc0bc8
FD		 2495 event_enabler = kzalloc(sizeof(*event_enabler), GFP_KERNEL);
2496 if (!event_enabler)
3c997079 2497 return NULL;
0bb716a8 2498 event_enabler->parent.enabler_type = LTTNG_EVENT_ENABLER_TYPE_RECORDER;
5b9817ae
MD		 2499 event_enabler->parent.format_type = format_type;
2500 INIT_LIST_HEAD(&event_enabler->parent.filter_bytecode_head);
2501 memcpy(&event_enabler->parent.event_param, event_param,
2502 sizeof(event_enabler->parent.event_param));
b2bc0bc8 2503 event_enabler->chan = chan;
3c997079 2504 /* ctx left NULL */
5b9817ae 2505 event_enabler->parent.enabled = 0;
def1e304
MD		 2506 return event_enabler;
2507}
2508
2509void lttng_event_enabler_session_add(struct lttng_kernel_session *session,
1ae083ba 2510 struct lttng_event_recorder_enabler *event_enabler)
def1e304 2511{
3c997079 2512 mutex_lock(&sessions_mutex);
def1e304
MD		 2513 list_add(&event_enabler->node, &session->priv->enablers_head);
2514 event_enabler->published = true;
2515 lttng_session_lazy_sync_event_enablers(session);
3c997079 2516 mutex_unlock(&sessions_mutex);
3c997079
MD		 2517}
2518
7b25ab0e 2519int lttng_event_enabler_enable(struct lttng_event_enabler_common *event_enabler)
3c997079
MD		 2520{
2521 mutex_lock(&sessions_mutex);
7b25ab0e
MD		 2522 event_enabler->enabled = 1;
2523 lttng_event_enabler_sync(event_enabler);
3c997079
MD		 2524 mutex_unlock(&sessions_mutex);
2525 return 0;
2526}
2527
7b25ab0e 2528int lttng_event_enabler_disable(struct lttng_event_enabler_common *event_enabler)
3c997079
MD		 2529{
2530 mutex_lock(&sessions_mutex);
7b25ab0e
MD		 2531 event_enabler->enabled = 0;
2532 lttng_event_enabler_sync(event_enabler);
3c997079
MD		 2533 mutex_unlock(&sessions_mutex);
2534 return 0;
2535}
2536
b2bc0bc8 2537static
8d554755 2538int lttng_enabler_attach_filter_bytecode(struct lttng_event_enabler_common *enabler,
606828e4 2539 struct lttng_kernel_abi_filter_bytecode __user *bytecode)
07dfc1d0 2540{
d64ca521 2541 struct lttng_kernel_bytecode_node *bytecode_node;
07dfc1d0
MD		 2542 uint32_t bytecode_len;
2543 int ret;
2544
2545 ret = get_user(bytecode_len, &bytecode->len);
2546 if (ret)
2547 return ret;
52cfcdf3 2548 bytecode_node = lttng_kvzalloc(sizeof(*bytecode_node) + bytecode_len,
07dfc1d0
MD		 2549 GFP_KERNEL);
2550 if (!bytecode_node)
2551 return -ENOMEM;
2552 ret = copy_from_user(&bytecode_node->bc, bytecode,
2553 sizeof(*bytecode) + bytecode_len);
2554 if (ret)
2555 goto error_free;
b2bc0bc8 2556
8a445457 2557 bytecode_node->type = LTTNG_KERNEL_BYTECODE_TYPE_FILTER;
07dfc1d0
MD		 2558 bytecode_node->enabler = enabler;
2559 /* Enforce length based on allocated size */
2560 bytecode_node->bc.len = bytecode_len;
2561 list_add_tail(&bytecode_node->node, &enabler->filter_bytecode_head);
b2bc0bc8 2562
07dfc1d0
MD		 2563 return 0;
2564
2565error_free:
52cfcdf3 2566 lttng_kvfree(bytecode_node);
07dfc1d0
MD		 2567 return ret;
2568}
2569
6048044f 2570int lttng_event_enabler_attach_filter_bytecode(struct lttng_event_enabler_common *event_enabler,
606828e4 2571 struct lttng_kernel_abi_filter_bytecode __user *bytecode)
b2bc0bc8
FD		 2572{
2573 int ret;
6048044f 2574 ret = lttng_enabler_attach_filter_bytecode(event_enabler, bytecode);
b2bc0bc8
FD		 2575 if (ret)
2576 goto error;
7b25ab0e 2577 lttng_event_enabler_sync(event_enabler);
b2bc0bc8
FD		 2578 return 0;
2579
2580error:
2581 return ret;
2582}
2583
e2d5dbc7 2584int lttng_event_add_callsite(struct lttng_kernel_event_common *event,
606828e4 2585 struct lttng_kernel_abi_event_callsite __user *callsite)
3aed4dca
FD		 2586{
2587
e2d5dbc7 2588 switch (event->priv->instrumentation) {
606828e4 2589 case LTTNG_KERNEL_ABI_UPROBE:
e2d5dbc7 2590 return lttng_uprobes_event_add_callsite(event, callsite);
3aed4dca
FD		 2591 default:
2592 return -EINVAL;
2593 }
2594}
2595
3c997079 2596static
8d554755 2597void lttng_enabler_destroy(struct lttng_event_enabler_common *enabler)
3c997079 2598{
d64ca521 2599 struct lttng_kernel_bytecode_node *filter_node, *tmp_filter_node;
07dfc1d0
MD		 2600
2601 /* Destroy filter bytecode */
2602 list_for_each_entry_safe(filter_node, tmp_filter_node,
2603 &enabler->filter_bytecode_head, node) {
52cfcdf3 2604 lttng_kvfree(filter_node);
07dfc1d0 2605 }
b2bc0bc8
FD		 2606}
2607
44b1abdc 2608void lttng_event_enabler_destroy(struct lttng_event_enabler_common *event_enabler)
b2bc0bc8 2609{
44b1abdc
MD		 2610 switch (event_enabler->enabler_type) {
2611 case LTTNG_EVENT_ENABLER_TYPE_RECORDER:
2612 {
2613 struct lttng_event_recorder_enabler *event_recorder_enabler =
2614 container_of(event_enabler, struct lttng_event_recorder_enabler, parent);
2615
2616 lttng_enabler_destroy(event_enabler);
2617 if (event_recorder_enabler->published)
2618 list_del(&event_recorder_enabler->node);
2619 kfree(event_recorder_enabler);
2620 break;
2621 }
2622 case LTTNG_EVENT_ENABLER_TYPE_NOTIFIER:
2623 {
2624 struct lttng_event_notifier_enabler *event_notifier_enabler =
2625 container_of(event_enabler, struct lttng_event_notifier_enabler, parent);
07dfc1d0 2626
44b1abdc
MD		 2627 list_del(&event_notifier_enabler->node);
2628 lttng_enabler_destroy(event_enabler);
2629 kfree(event_notifier_enabler);
2630 break;
2631 }
2632 default:
2633 WARN_ON_ONCE(1);
2634 }
3c997079
MD		 2635}
2636
dffef45d
FD		 2637struct lttng_event_notifier_enabler *lttng_event_notifier_enabler_create(
2638 struct lttng_event_notifier_group *event_notifier_group,
2639 enum lttng_enabler_format_type format_type,
606828e4 2640 struct lttng_kernel_abi_event_notifier *event_notifier_param)
dffef45d
FD		 2641{
2642 struct lttng_event_notifier_enabler *event_notifier_enabler;
2643
2644 event_notifier_enabler = kzalloc(sizeof(*event_notifier_enabler), GFP_KERNEL);
2645 if (!event_notifier_enabler)
2646 return NULL;
2647
0bb716a8 2648 event_notifier_enabler->parent.enabler_type = LTTNG_EVENT_ENABLER_TYPE_NOTIFIER;
5b9817ae
MD		 2649 event_notifier_enabler->parent.format_type = format_type;
2650 INIT_LIST_HEAD(&event_notifier_enabler->parent.filter_bytecode_head);
99d223ad
FD		 2651 INIT_LIST_HEAD(&event_notifier_enabler->capture_bytecode_head);
2652
99f52fcc 2653 event_notifier_enabler->error_counter_index = event_notifier_param->error_counter_index;
99d223ad 2654 event_notifier_enabler->num_captures = 0;
dffef45d 2655
5b9817ae
MD		 2656 memcpy(&event_notifier_enabler->parent.event_param, &event_notifier_param->event,
2657 sizeof(event_notifier_enabler->parent.event_param));
dffef45d 2658
5b9817ae
MD		 2659 event_notifier_enabler->parent.enabled = 0;
2660 event_notifier_enabler->parent.user_token = event_notifier_param->event.token;
dffef45d
FD		 2661 event_notifier_enabler->group = event_notifier_group;
2662
2663 mutex_lock(&sessions_mutex);
2664 list_add(&event_notifier_enabler->node, &event_notifier_enabler->group->enablers_head);
b01155ba 2665 lttng_event_notifier_group_sync_enablers(event_notifier_enabler->group);
dffef45d
FD		 2666
2667 mutex_unlock(&sessions_mutex);
2668
2669 return event_notifier_enabler;
2670}
2671
2672int lttng_event_notifier_enabler_enable(
2673 struct lttng_event_notifier_enabler *event_notifier_enabler)
2674{
2675 mutex_lock(&sessions_mutex);
2676 lttng_event_notifier_enabler_as_enabler(event_notifier_enabler)->enabled = 1;
b01155ba 2677 lttng_event_notifier_group_sync_enablers(event_notifier_enabler->group);
dffef45d
FD		 2678 mutex_unlock(&sessions_mutex);
2679 return 0;
2680}
2681
2682int lttng_event_notifier_enabler_disable(
2683 struct lttng_event_notifier_enabler *event_notifier_enabler)
2684{
2685 mutex_lock(&sessions_mutex);
2686 lttng_event_notifier_enabler_as_enabler(event_notifier_enabler)->enabled = 0;
b01155ba 2687 lttng_event_notifier_group_sync_enablers(event_notifier_enabler->group);
dffef45d
FD		 2688 mutex_unlock(&sessions_mutex);
2689 return 0;
2690}
2691
99d223ad
FD		 2692int lttng_event_notifier_enabler_attach_capture_bytecode(
2693 struct lttng_event_notifier_enabler *event_notifier_enabler,
606828e4 2694 struct lttng_kernel_abi_capture_bytecode __user *bytecode)
99d223ad 2695{
d64ca521 2696 struct lttng_kernel_bytecode_node *bytecode_node;
8d554755 2697 struct lttng_event_enabler_common *enabler =
99d223ad
FD		 2698 lttng_event_notifier_enabler_as_enabler(event_notifier_enabler);
2699 uint32_t bytecode_len;
2700 int ret;
2701
2702 ret = get_user(bytecode_len, &bytecode->len);
2703 if (ret)
2704 return ret;
2705
2706 bytecode_node = lttng_kvzalloc(sizeof(*bytecode_node) + bytecode_len,
2707 GFP_KERNEL);
2708 if (!bytecode_node)
2709 return -ENOMEM;
2710
2711 ret = copy_from_user(&bytecode_node->bc, bytecode,
2712 sizeof(*bytecode) + bytecode_len);
2713 if (ret)
2714 goto error_free;
2715
8a445457 2716 bytecode_node->type = LTTNG_KERNEL_BYTECODE_TYPE_CAPTURE;
99d223ad
FD		 2717 bytecode_node->enabler = enabler;
2718
2719 /* Enforce length based on allocated size */
2720 bytecode_node->bc.len = bytecode_len;
2721 list_add_tail(&bytecode_node->node, &event_notifier_enabler->capture_bytecode_head);
2722
2723 event_notifier_enabler->num_captures++;
2724
2725 lttng_event_notifier_group_sync_enablers(event_notifier_enabler->group);
2726 goto end;
2727
2728error_free:
2729 lttng_kvfree(bytecode_node);
2730end:
2731 return ret;
2732}
2733
3c997079 2734/*
b2bc0bc8 2735 * lttng_session_sync_event_enablers should be called just before starting a
3c997079
MD		 2736 * session.
2737 * Should be called with sessions mutex held.
2738 */
2739static
8cdc1a81 2740void lttng_session_sync_event_enablers(struct lttng_kernel_session *session)
3c997079 2741{
1ae083ba 2742 struct lttng_event_recorder_enabler *event_enabler;
a67ba386 2743 struct lttng_kernel_event_recorder_private *event_recorder_priv;
3c997079 2744
8cdc1a81 2745 list_for_each_entry(event_enabler, &session->priv->enablers_head, node)
b2bc0bc8 2746 lttng_event_enabler_ref_events(event_enabler);
3c997079
MD		 2747 /*
2748 * For each event, if at least one of its enablers is enabled,
2749 * and its channel and session transient states are enabled, we
2750 * enable the event, else we disable it.
2751 */
8cdc1a81 2752 list_for_each_entry(event_recorder_priv, &session->priv->events, node) {
a67ba386 2753 struct lttng_kernel_event_recorder *event_recorder = event_recorder_priv->pub;
3c997079 2754 struct lttng_enabler_ref *enabler_ref;
218585b9 2755 struct lttng_kernel_bytecode_runtime *runtime;
e2d5dbc7
MD		 2756 int enabled = 0, has_enablers_without_filter_bytecode = 0;
2757 int nr_filters = 0;
3c997079 2758
a67ba386 2759 switch (event_recorder_priv->parent.instrumentation) {
c190d76e
MJ		 2760 case LTTNG_KERNEL_ABI_TRACEPOINT:
2761 lttng_fallthrough;
606828e4 2762 case LTTNG_KERNEL_ABI_SYSCALL:
3c997079
MD		 2763 /* Enable events */
2764 list_for_each_entry(enabler_ref,
a67ba386 2765 &event_recorder_priv->parent.enablers_ref_head, node) {
3c997079
MD		 2766 if (enabler_ref->ref->enabled) {
2767 enabled = 1;
2768 break;
2769 }
2770 }
4ecb5ad5 2771 break;
b4a75708 2772
4ecb5ad5 2773 default:
3c997079
MD		 2774 /* Not handled with lazy sync. */
2775 continue;
2776 }
2777 /*
2778 * Enabled state is based on union of enablers, with
2779 * intesection of session and channel transient enable
2780 * states.
2781 */
f7d06400 2782 enabled = enabled && session->priv->tstate && event_recorder->chan->priv->parent.tstate;
3c997079 2783
a67ba386 2784 WRITE_ONCE(event_recorder->parent.enabled, enabled);
3c997079
MD		 2785 /*
2786 * Sync tracepoint registration with event enabled
2787 * state.
2788 */
2789 if (enabled) {
96c59af1 2790 register_event_recorder(event_recorder);
3c997079 2791 } else {
96c59af1 2792 _lttng_event_recorder_unregister(event_recorder);
3c997079 2793 }
07dfc1d0
MD		 2794
2795 /* Check if has enablers without bytecode enabled */
2796 list_for_each_entry(enabler_ref,
a67ba386 2797 &event_recorder_priv->parent.enablers_ref_head, node) {
07dfc1d0
MD		 2798 if (enabler_ref->ref->enabled
2799 && list_empty(&enabler_ref->ref->filter_bytecode_head)) {
e2d5dbc7 2800 has_enablers_without_filter_bytecode = 1;
07dfc1d0
MD		 2801 break;
2802 }
2803 }
a67ba386 2804 event_recorder_priv->parent.has_enablers_without_filter_bytecode =
e2d5dbc7 2805 has_enablers_without_filter_bytecode;
07dfc1d0
MD		 2806
2807 /* Enable filters */
2808 list_for_each_entry(runtime,
e2d5dbc7 2809 &event_recorder_priv->parent.filter_bytecode_runtime_head, node) {
8a445457 2810 lttng_bytecode_sync_state(runtime);
e2d5dbc7
MD		 2811 nr_filters++;
2812 }
2813 WRITE_ONCE(event_recorder_priv->parent.pub->eval_filter,
2814 !(has_enablers_without_filter_bytecode || !nr_filters));
3c997079
MD		 2815 }
2816}
2817
2818/*
2819 * Apply enablers to session events, adding events to session if need
2820 * be. It is required after each modification applied to an active
2821 * session, and right before session "start".
2822 * "lazy" sync means we only sync if required.
2823 * Should be called with sessions mutex held.
2824 */
2825static
8cdc1a81 2826void lttng_session_lazy_sync_event_enablers(struct lttng_kernel_session *session)
3c997079
MD		 2827{
2828 /* We can skip if session is not active */
2829 if (!session->active)
2830 return;
b2bc0bc8 2831 lttng_session_sync_event_enablers(session);
3c997079
MD		 2832}
2833
b01155ba
FD		 2834static
2835void lttng_event_notifier_group_sync_enablers(struct lttng_event_notifier_group *event_notifier_group)
2836{
2837 struct lttng_event_notifier_enabler *event_notifier_enabler;
a67ba386 2838 struct lttng_kernel_event_notifier_private *event_notifier_priv;
b01155ba
FD		 2839
2840 list_for_each_entry(event_notifier_enabler, &event_notifier_group->enablers_head, node)
2841 lttng_event_notifier_enabler_ref_event_notifiers(event_notifier_enabler);
2842
2843 /*
2844 * For each event_notifier, if at least one of its enablers is enabled,
2845 * we enable the event_notifier, else we disable it.
2846 */
a67ba386
MD		 2847 list_for_each_entry(event_notifier_priv, &event_notifier_group->event_notifiers_head, node) {
2848 struct lttng_kernel_event_notifier *event_notifier = event_notifier_priv->pub;
b01155ba 2849 struct lttng_enabler_ref *enabler_ref;
218585b9 2850 struct lttng_kernel_bytecode_runtime *runtime;
e2d5dbc7
MD		 2851 int enabled = 0, has_enablers_without_filter_bytecode = 0;
2852 int nr_filters = 0, nr_captures = 0;
b01155ba 2853
a67ba386 2854 switch (event_notifier_priv->parent.instrumentation) {
c190d76e
MJ		 2855 case LTTNG_KERNEL_ABI_TRACEPOINT:
2856 lttng_fallthrough;
606828e4 2857 case LTTNG_KERNEL_ABI_SYSCALL:
b01155ba
FD		 2858 /* Enable event_notifiers */
2859 list_for_each_entry(enabler_ref,
a67ba386 2860 &event_notifier_priv->parent.enablers_ref_head, node) {
b01155ba
FD		 2861 if (enabler_ref->ref->enabled) {
2862 enabled = 1;
2863 break;
2864 }
2865 }
2866 break;
b4a75708 2867
b01155ba
FD		 2868 default:
2869 /* Not handled with sync. */
2870 continue;
2871 }
2872
a67ba386 2873 WRITE_ONCE(event_notifier->parent.enabled, enabled);
b01155ba
FD		 2874 /*
2875 * Sync tracepoint registration with event_notifier enabled
2876 * state.
2877 */
2878 if (enabled) {
a67ba386 2879 if (!event_notifier_priv->parent.registered)
b01155ba
FD		 2880 register_event_notifier(event_notifier);
2881 } else {
a67ba386 2882 if (event_notifier_priv->parent.registered)
b01155ba
FD		 2883 _lttng_event_notifier_unregister(event_notifier);
2884 }
2885
2886 /* Check if has enablers without bytecode enabled */
2887 list_for_each_entry(enabler_ref,
a67ba386 2888 &event_notifier_priv->parent.enablers_ref_head, node) {
b01155ba
FD		 2889 if (enabler_ref->ref->enabled
2890 && list_empty(&enabler_ref->ref->filter_bytecode_head)) {
e2d5dbc7 2891 has_enablers_without_filter_bytecode = 1;
b01155ba
FD		 2892 break;
2893 }
2894 }
a67ba386 2895 event_notifier_priv->parent.has_enablers_without_filter_bytecode =
e2d5dbc7 2896 has_enablers_without_filter_bytecode;
b01155ba
FD		 2897
2898 /* Enable filters */
2899 list_for_each_entry(runtime,
e2d5dbc7 2900 &event_notifier_priv->parent.filter_bytecode_runtime_head, node) {
8a445457 2901 lttng_bytecode_sync_state(runtime);
e2d5dbc7
MD		 2902 nr_filters++;
2903 }
2904 WRITE_ONCE(event_notifier_priv->parent.pub->eval_filter,
2905 !(has_enablers_without_filter_bytecode || !nr_filters));
99d223ad
FD		 2906
2907 /* Enable captures */
2908 list_for_each_entry(runtime,
e2d5dbc7 2909 &event_notifier_priv->capture_bytecode_runtime_head, node) {
8a445457 2910 lttng_bytecode_sync_state(runtime);
e2d5dbc7
MD		 2911 nr_captures++;
2912 }
2913 WRITE_ONCE(event_notifier->eval_capture, !!nr_captures);
b01155ba
FD		 2914 }
2915}
2916
7b25ab0e
MD		 2917static
2918void lttng_event_enabler_sync(struct lttng_event_enabler_common *event_enabler)
2919{
2920 switch (event_enabler->enabler_type) {
2921 case LTTNG_EVENT_ENABLER_TYPE_RECORDER:
2922 {
2923 struct lttng_event_recorder_enabler *event_recorder_enabler =
2924 container_of(event_enabler, struct lttng_event_recorder_enabler, parent);
2925 lttng_session_lazy_sync_event_enablers(event_recorder_enabler->chan->parent.session);
2926 break;
2927 }
2928 case LTTNG_EVENT_ENABLER_TYPE_NOTIFIER:
2929 {
2930 struct lttng_event_notifier_enabler *event_notifier_enabler =
2931 container_of(event_enabler, struct lttng_event_notifier_enabler, parent);
2932 lttng_event_notifier_group_sync_enablers(event_notifier_enabler->group);
2933 break;
2934 }
2935 default:
2936 WARN_ON_ONCE(1);
2937 }
2938}
2939
1ec3f75a 2940/*
d83004aa
JD		 2941 * Serialize at most one packet worth of metadata into a metadata
2942 * channel.
92d9f5e6
MD		 2943 * We grab the metadata cache mutex to get exclusive access to our metadata
2944 * buffer and to the metadata cache. Exclusive access to the metadata buffer
2945 * allows us to do racy operations such as looking for remaining space left in
2946 * packet and write, since mutual exclusion protects us from concurrent writes.
2947 * Mutual exclusion on the metadata cache allow us to read the cache content
2948 * without racing against reallocation of the cache by updates.
35097f36
JD		 2949 * Returns the number of bytes written in the channel, 0 if no data
2950 * was written and a negative value on error.
1ec3f75a 2951 */
b3b8072b 2952int lttng_metadata_output_channel(struct lttng_metadata_stream *stream,
860c213b 2953 struct lttng_kernel_ring_buffer_channel *chan, bool *coherent)
d83004aa 2954{
8a57ec02 2955 struct lttng_kernel_ring_buffer_ctx ctx;
d83004aa
JD		 2956 int ret = 0;
2957 size_t len, reserve_len;
2958
f613e3e6 2959 /*
92d9f5e6
MD		 2960 * Ensure we support mutiple get_next / put sequences followed by
2961 * put_next. The metadata cache lock protects reading the metadata
2962 * cache. It can indeed be read concurrently by "get_next_subbuf" and
2963 * "flush" operations on the buffer invoked by different processes.
2964 * Moreover, since the metadata cache memory can be reallocated, we
2965 * need to have exclusive access against updates even though we only
2966 * read it.
f613e3e6 2967 */
92d9f5e6 2968 mutex_lock(&stream->metadata_cache->lock);
f613e3e6
MD		 2969 WARN_ON(stream->metadata_in < stream->metadata_out);
2970 if (stream->metadata_in != stream->metadata_out)
de23d59d 2971 goto end;
f613e3e6 2972
9616f0bf
JD		 2973 /* Metadata regenerated, change the version. */
2974 if (stream->metadata_cache->version != stream->version)
2975 stream->version = stream->metadata_cache->version;
2976
d83004aa 2977 len = stream->metadata_cache->metadata_written -
f613e3e6 2978 stream->metadata_in;
9de2c215 2979 if (!len)
de23d59d 2980 goto end;
d83004aa 2981 reserve_len = min_t(size_t,
4a399b76 2982 stream->transport->ops.priv->packet_avail_size(chan),
d83004aa 2983 len);
a775608d 2984 lib_ring_buffer_ctx_init(&ctx, chan, reserve_len,
b1199bd3 2985 sizeof(char), NULL);
d83004aa
JD		 2986 /*
2987 * If reservation failed, return an error to the caller.
2988 */
c2fb9c1c 2989 ret = stream->transport->ops.event_reserve(&ctx);
d83004aa
JD		 2990 if (ret != 0) {
2991 printk(KERN_WARNING "LTTng: Metadata event reservation failed\n");
8b97fd42 2992 stream->coherent = false;
d83004aa
JD		 2993 goto end;
2994 }
b3b8072b 2995 stream->transport->ops.event_write(&ctx,
f613e3e6 2996 stream->metadata_cache->data + stream->metadata_in,
f5ffbd77 2997 reserve_len, 1);
b3b8072b 2998 stream->transport->ops.event_commit(&ctx);
f613e3e6 2999 stream->metadata_in += reserve_len;
3e75e2a7 3000 if (reserve_len < len)
8b97fd42
MD		 3001 stream->coherent = false;
3002 else
3003 stream->coherent = true;
d83004aa
JD		 3004 ret = reserve_len;
3005
3006end:
8b97fd42
MD		 3007 if (coherent)
3008 *coherent = stream->coherent;
92d9f5e6 3009 mutex_unlock(&stream->metadata_cache->lock);
d83004aa
JD		 3010 return ret;
3011}
3012
8b97fd42 3013static
8cdc1a81 3014void lttng_metadata_begin(struct lttng_kernel_session *session)
8b97fd42 3015{
8cdc1a81
MD		 3016 if (atomic_inc_return(&session->priv->metadata_cache->producing) == 1)
3017 mutex_lock(&session->priv->metadata_cache->lock);
8b97fd42
MD		 3018}
3019
3020static
8cdc1a81 3021void lttng_metadata_end(struct lttng_kernel_session *session)
8b97fd42 3022{
8cdc1a81
MD		 3023 WARN_ON_ONCE(!atomic_read(&session->priv->metadata_cache->producing));
3024 if (atomic_dec_return(&session->priv->metadata_cache->producing) == 0) {
3e75e2a7
MD		 3025 struct lttng_metadata_stream *stream;
3026
8cdc1a81 3027 list_for_each_entry(stream, &session->priv->metadata_cache->metadata_stream, list)
3e75e2a7 3028 wake_up_interruptible(&stream->read_wait);
8cdc1a81 3029 mutex_unlock(&session->priv->metadata_cache->lock);
3e75e2a7 3030 }
8b97fd42
MD		 3031}
3032
d83004aa
JD		 3033/*
3034 * Write the metadata to the metadata cache.
3035 * Must be called with sessions_mutex held.
92d9f5e6
MD		 3036 * The metadata cache lock protects us from concurrent read access from
3037 * thread outputting metadata content to ring buffer.
8b97fd42
MD		 3038 * The content of the printf is printed as a single atomic metadata
3039 * transaction.
d83004aa 3040 */
8cdc1a81 3041int lttng_metadata_printf(struct lttng_kernel_session *session,
c099397a
MD		 3042 const char *fmt, ...)
3043{
c099397a 3044 char *str;
d83004aa 3045 size_t len;
c099397a
MD		 3046 va_list ap;
3047
585e5dcc 3048 WARN_ON_ONCE(!LTTNG_READ_ONCE(session->active));
c099397a
MD		 3049
3050 va_start(ap, fmt);
3051 str = kvasprintf(GFP_KERNEL, fmt, ap);
3052 va_end(ap);
3053 if (!str)
3054 return -ENOMEM;
3055
1ec3f75a 3056 len = strlen(str);
8cdc1a81
MD		 3057 WARN_ON_ONCE(!atomic_read(&session->priv->metadata_cache->producing));
3058 if (session->priv->metadata_cache->metadata_written + len >
3059 session->priv->metadata_cache->cache_alloc) {
d83004aa
JD		 3060 char *tmp_cache_realloc;
3061 unsigned int tmp_cache_alloc_size;
3062
3063 tmp_cache_alloc_size = max_t(unsigned int,
8cdc1a81
MD		 3064 session->priv->metadata_cache->cache_alloc + len,
3065 session->priv->metadata_cache->cache_alloc << 1);
0ca42eb7 3066 tmp_cache_realloc = vzalloc(tmp_cache_alloc_size);
d83004aa
JD		 3067 if (!tmp_cache_realloc)
3068 goto err;
8cdc1a81 3069 if (session->priv->metadata_cache->data) {
a606b6e8 3070 memcpy(tmp_cache_realloc,
8cdc1a81
MD		 3071 session->priv->metadata_cache->data,
3072 session->priv->metadata_cache->cache_alloc);
3073 vfree(session->priv->metadata_cache->data);
a606b6e8
ML		 3074 }
3075
8cdc1a81
MD		 3076 session->priv->metadata_cache->cache_alloc = tmp_cache_alloc_size;
3077 session->priv->metadata_cache->data = tmp_cache_realloc;
c099397a 3078 }
8cdc1a81
MD		 3079 memcpy(session->priv->metadata_cache->data +
3080 session->priv->metadata_cache->metadata_written,
d83004aa 3081 str, len);
8cdc1a81 3082 session->priv->metadata_cache->metadata_written += len;
c099397a 3083 kfree(str);
d83004aa 3084
d83004aa
JD		 3085 return 0;
3086
3087err:
3088 kfree(str);
3089 return -ENOMEM;
c099397a
MD		 3090}
3091
f513b2bf 3092static
8cdc1a81 3093int print_tabs(struct lttng_kernel_session *session, size_t nesting)
f513b2bf
MD		 3094{
3095 size_t i;
3096
3097 for (i = 0; i < nesting; i++) {
3098 int ret;
3099
3100 ret = lttng_metadata_printf(session, " ");
3101 if (ret) {
3102 return ret;
3103 }
3104 }
3105 return 0;
3106}
3107
ceabb767 3108static
8cdc1a81 3109int lttng_field_name_statedump(struct lttng_kernel_session *session,
437d5aa5 3110 const struct lttng_kernel_event_field *field,
ceabb767
MD		 3111 size_t nesting)
3112{
3113 return lttng_metadata_printf(session, " _%s;\n", field->name);
3114}
3115
3116static
8cdc1a81 3117int _lttng_integer_type_statedump(struct lttng_kernel_session *session,
437d5aa5
MD		 3118 const struct lttng_kernel_type_integer *type,
3119 enum lttng_kernel_string_encoding parent_encoding,
ceabb767
MD		 3120 size_t nesting)
3121{
3122 int ret;
3123
ceabb767
MD		 3124 ret = print_tabs(session, nesting);
3125 if (ret)
3126 return ret;
3127 ret = lttng_metadata_printf(session,
3128 "integer { size = %u; align = %u; signed = %u; encoding = %s; base = %u;%s }",
437d5aa5
MD		 3129 type->size,
3130 type->alignment,
3131 type->signedness,
3132 (parent_encoding == lttng_kernel_string_encoding_none)
ceabb767 3133 ? "none"
437d5aa5 3134 : (parent_encoding == lttng_kernel_string_encoding_UTF8)
ceabb767
MD		 3135 ? "UTF8"
3136 : "ASCII",
437d5aa5 3137 type->base,
ceabb767 3138#if __BYTE_ORDER == __BIG_ENDIAN
437d5aa5 3139 type->reverse_byte_order ? " byte_order = le;" : ""
ceabb767 3140#else
437d5aa5 3141 type->reverse_byte_order ? " byte_order = be;" : ""
ceabb767
MD		 3142#endif
3143 );
3144 return ret;
3145}
3146
f513b2bf
MD		 3147/*
3148 * Must be called with sessions_mutex held.
3149 */
3150static
8cdc1a81 3151int _lttng_struct_type_statedump(struct lttng_kernel_session *session,
437d5aa5 3152 const struct lttng_kernel_type_struct *type,
f513b2bf
MD		 3153 size_t nesting)
3154{
51ef4536 3155 const char *prev_field_name = NULL;
f513b2bf
MD		 3156 int ret;
3157 uint32_t i, nr_fields;
ceabb767
MD		 3158 unsigned int alignment;
3159
f513b2bf
MD		 3160 ret = print_tabs(session, nesting);
3161 if (ret)
3162 return ret;
3163 ret = lttng_metadata_printf(session,
3164 "struct {\n");
3165 if (ret)
3166 return ret;
437d5aa5 3167 nr_fields = type->nr_fields;
f513b2bf 3168 for (i = 0; i < nr_fields; i++) {
437d5aa5 3169 const struct lttng_kernel_event_field *iter_field;
f513b2bf 3170
437d5aa5 3171 iter_field = type->fields[i];
51ef4536 3172 ret = _lttng_field_statedump(session, iter_field, nesting + 1, &prev_field_name);
f513b2bf
MD		 3173 if (ret)
3174 return ret;
3175 }
3176 ret = print_tabs(session, nesting);
3177 if (ret)
3178 return ret;
437d5aa5 3179 alignment = type->alignment;
ceabb767
MD		 3180 if (alignment) {
3181 ret = lttng_metadata_printf(session,
3182 "} align(%u)",
3183 alignment);
3184 } else {
3185 ret = lttng_metadata_printf(session,
3186 "}");
3187 }
f513b2bf
MD		 3188 return ret;
3189}
3190
3191/*
3192 * Must be called with sessions_mutex held.
3193 */
3194static
8cdc1a81 3195int _lttng_struct_field_statedump(struct lttng_kernel_session *session,
437d5aa5 3196 const struct lttng_kernel_event_field *field,
f513b2bf
MD		 3197 size_t nesting)
3198{
3199 int ret;
3200
3201 ret = _lttng_struct_type_statedump(session,
437d5aa5 3202 lttng_kernel_get_type_struct(field->type), nesting);
f513b2bf
MD		 3203 if (ret)
3204 return ret;
ceabb767 3205 return lttng_field_name_statedump(session, field, nesting);
f513b2bf
MD		 3206}
3207
65c85aa6
MD		 3208/*
3209 * Must be called with sessions_mutex held.
3210 */
3211static
8cdc1a81 3212int _lttng_variant_type_statedump(struct lttng_kernel_session *session,
437d5aa5 3213 const struct lttng_kernel_type_variant *type,
51ef4536
MD		 3214 size_t nesting,
3215 const char *prev_field_name)
65c85aa6 3216{
51ef4536 3217 const char *tag_name;
65c85aa6
MD		 3218 int ret;
3219 uint32_t i, nr_choices;
3220
51ef4536
MD		 3221 tag_name = type->tag_name;
3222 if (!tag_name)
3223 tag_name = prev_field_name;
3224 if (!tag_name)
3225 return -EINVAL;
ceabb767
MD		 3226 /*
3227 * CTF 1.8 does not allow expressing nonzero variant alignment in a nestable way.
3228 */
437d5aa5 3229 if (type->alignment != 0)
ceabb767 3230 return -EINVAL;
65c85aa6
MD		 3231 ret = print_tabs(session, nesting);
3232 if (ret)
3233 return ret;
3234 ret = lttng_metadata_printf(session,
3235 "variant <_%s> {\n",
51ef4536 3236 tag_name);
65c85aa6
MD		 3237 if (ret)
3238 return ret;
437d5aa5 3239 nr_choices = type->nr_choices;
65c85aa6 3240 for (i = 0; i < nr_choices; i++) {
437d5aa5 3241 const struct lttng_kernel_event_field *iter_field;
65c85aa6 3242
437d5aa5 3243 iter_field = type->choices[i];
51ef4536 3244 ret = _lttng_field_statedump(session, iter_field, nesting + 1, NULL);
65c85aa6
MD		 3245 if (ret)
3246 return ret;
3247 }
3248 ret = print_tabs(session, nesting);
3249 if (ret)
3250 return ret;
3251 ret = lttng_metadata_printf(session,
3252 "}");
3253 return ret;
3254}
3255
3256/*
3257 * Must be called with sessions_mutex held.
3258 */
3259static
8cdc1a81 3260int _lttng_variant_field_statedump(struct lttng_kernel_session *session,
437d5aa5 3261 const struct lttng_kernel_event_field *field,
51ef4536
MD		 3262 size_t nesting,
3263 const char *prev_field_name)
65c85aa6
MD		 3264{
3265 int ret;
3266
3267 ret = _lttng_variant_type_statedump(session,
51ef4536
MD		 3268 lttng_kernel_get_type_variant(field->type), nesting,
3269 prev_field_name);
65c85aa6
MD		 3270 if (ret)
3271 return ret;
ceabb767 3272 return lttng_field_name_statedump(session, field, nesting);
65c85aa6
MD		 3273}
3274
f513b2bf
MD		 3275/*
3276 * Must be called with sessions_mutex held.
3277 */
3278static
8cdc1a81 3279int _lttng_array_field_statedump(struct lttng_kernel_session *session,
437d5aa5 3280 const struct lttng_kernel_event_field *field,
f513b2bf
MD		 3281 size_t nesting)
3282{
3283 int ret;
437d5aa5
MD		 3284 const struct lttng_kernel_type_array *array_type;
3285 const struct lttng_kernel_type_common *elem_type;
f513b2bf 3286
437d5aa5
MD		 3287 array_type = lttng_kernel_get_type_array(field->type);
3288 WARN_ON_ONCE(!array_type);
ceabb767 3289
437d5aa5 3290 if (array_type->alignment) {
ceabb767 3291 ret = print_tabs(session, nesting);
f513b2bf
MD		 3292 if (ret)
3293 return ret;
ceabb767
MD		 3294 ret = lttng_metadata_printf(session,
3295 "struct { } align(%u) _%s_padding;\n",
437d5aa5 3296 array_type->alignment * CHAR_BIT,
ceabb767
MD		 3297 field->name);
3298 if (ret)
3299 return ret;
3300 }
3301 /*
3302 * Nested compound types: Only array of structures and variants are
3303 * currently supported.
3304 */
437d5aa5 3305 elem_type = array_type->elem_type;
12bb2edb
MD		 3306 switch (elem_type->type) {
3307 case lttng_kernel_type_integer:
437d5aa5
MD		 3308 case lttng_kernel_type_struct:
3309 case lttng_kernel_type_variant:
3310 ret = _lttng_type_statedump(session, elem_type,
3311 array_type->encoding, nesting);
65c85aa6
MD		 3312 if (ret)
3313 return ret;
3314 break;
ceabb767 3315
f513b2bf
MD		 3316 default:
3317 return -EINVAL;
3318 }
3319 ret = lttng_metadata_printf(session,
3320 " _%s[%u];\n",
3321 field->name,
437d5aa5 3322 array_type->length);
f513b2bf
MD		 3323 return ret;
3324}
3325
3326/*
3327 * Must be called with sessions_mutex held.
3328 */
3329static
8cdc1a81 3330int _lttng_sequence_field_statedump(struct lttng_kernel_session *session,
437d5aa5 3331 const struct lttng_kernel_event_field *field,
51ef4536
MD		 3332 size_t nesting,
3333 const char *prev_field_name)
f513b2bf
MD		 3334{
3335 int ret;
3336 const char *length_name;
437d5aa5
MD		 3337 const struct lttng_kernel_type_sequence *sequence_type;
3338 const struct lttng_kernel_type_common *elem_type;
f513b2bf 3339
437d5aa5
MD		 3340 sequence_type = lttng_kernel_get_type_sequence(field->type);
3341 WARN_ON_ONCE(!sequence_type);
f513b2bf 3342
437d5aa5 3343 length_name = sequence_type->length_name;
51ef4536
MD		 3344 if (!length_name)
3345 length_name = prev_field_name;
3346 if (!length_name)
3347 return -EINVAL;
ceabb767 3348
437d5aa5 3349 if (sequence_type->alignment) {
ceabb767 3350 ret = print_tabs(session, nesting);
f513b2bf
MD		 3351 if (ret)
3352 return ret;
ceabb767
MD		 3353 ret = lttng_metadata_printf(session,
3354 "struct { } align(%u) _%s_padding;\n",
437d5aa5 3355 sequence_type->alignment * CHAR_BIT,
ceabb767
MD		 3356 field->name);
3357 if (ret)
3358 return ret;
3359 }
3360
3361 /*
3362 * Nested compound types: Only array of structures and variants are
3363 * currently supported.
3364 */
437d5aa5 3365 elem_type = sequence_type->elem_type;
12bb2edb
MD		 3366 switch (elem_type->type) {
3367 case lttng_kernel_type_integer:
437d5aa5
MD		 3368 case lttng_kernel_type_struct:
3369 case lttng_kernel_type_variant:
3370 ret = _lttng_type_statedump(session, elem_type,
3371 sequence_type->encoding, nesting);
65c85aa6
MD		 3372 if (ret)
3373 return ret;
3374 break;
ceabb767 3375
f513b2bf
MD		 3376 default:
3377 return -EINVAL;
3378 }
3379 ret = lttng_metadata_printf(session,
3380 " _%s[ _%s ];\n",
3381 field->name,
51ef4536 3382 length_name);
f513b2bf
MD		 3383 return ret;
3384}
3385
141ddf28
MD		 3386/*
3387 * Must be called with sessions_mutex held.
3388 */
3389static
8cdc1a81 3390int _lttng_enum_type_statedump(struct lttng_kernel_session *session,
437d5aa5 3391 const struct lttng_kernel_type_enum *type,
141ddf28
MD		 3392 size_t nesting)
3393{
d96a4a7a 3394 const struct lttng_kernel_enum_desc *enum_desc;
437d5aa5 3395 const struct lttng_kernel_type_common *container_type;
141ddf28
MD		 3396 int ret;
3397 unsigned int i, nr_entries;
3398
437d5aa5 3399 container_type = type->container_type;
12bb2edb 3400 if (container_type->type != lttng_kernel_type_integer) {
ceabb767
MD		 3401 ret = -EINVAL;
3402 goto end;
3403 }
437d5aa5 3404 enum_desc = type->desc;
141ddf28
MD		 3405 nr_entries = enum_desc->nr_entries;
3406
3407 ret = print_tabs(session, nesting);
3408 if (ret)
3409 goto end;
ceabb767
MD		 3410 ret = lttng_metadata_printf(session, "enum : ");
3411 if (ret)
3412 goto end;
437d5aa5
MD		 3413 ret = _lttng_integer_type_statedump(session, lttng_kernel_get_type_integer(container_type),
3414 lttng_kernel_string_encoding_none, 0);
141ddf28 3415 if (ret)
ceabb767
MD		 3416 goto end;
3417 ret = lttng_metadata_printf(session, " {\n");
3418 if (ret)
3419 goto end;
141ddf28
MD		 3420 /* Dump all entries */
3421 for (i = 0; i < nr_entries; i++) {
437d5aa5 3422 const struct lttng_kernel_enum_entry *entry = enum_desc->entries[i];
141ddf28
MD		 3423 int j, len;
3424
3425 ret = print_tabs(session, nesting + 1);
3426 if (ret)
3427 goto end;
3428 ret = lttng_metadata_printf(session,
3429 "\"");
3430 if (ret)
3431 goto end;
3432 len = strlen(entry->string);
3433 /* Escape the character '"' */
3434 for (j = 0; j < len; j++) {
3435 char c = entry->string[j];
3436
3437 switch (c) {
3438 case '"':
3439 ret = lttng_metadata_printf(session,
3440 "\\\"");
3441 break;
3442 case '\\':
3443 ret = lttng_metadata_printf(session,
3444 "\\\\");
3445 break;
3446 default:
3447 ret = lttng_metadata_printf(session,
3448 "%c", c);
3449 break;
3450 }
3451 if (ret)
3452 goto end;
3453 }
08ad1061
PP		 3454 ret = lttng_metadata_printf(session, "\"");
3455 if (ret)
3456 goto end;
3457
3458 if (entry->options.is_auto) {
3459 ret = lttng_metadata_printf(session, ",\n");
3460 if (ret)
3461 goto end;
141ddf28 3462 } else {
7c559490
MD		 3463 ret = lttng_metadata_printf(session,
3464 " = ");
3465 if (ret)
3466 goto end;
3467 if (entry->start.signedness)
141ddf28 3468 ret = lttng_metadata_printf(session,
7c559490
MD		 3469 "%lld", (long long) entry->start.value);
3470 else
3471 ret = lttng_metadata_printf(session,
3472 "%llu", entry->start.value);
3473 if (ret)
3474 goto end;
3475 if (entry->start.signedness == entry->end.signedness &&
3476 entry->start.value
3477 == entry->end.value) {
141ddf28 3478 ret = lttng_metadata_printf(session,
7c559490
MD		 3479 ",\n");
3480 } else {
3481 if (entry->end.signedness) {
3482 ret = lttng_metadata_printf(session,
3483 " ... %lld,\n",
3484 (long long) entry->end.value);
3485 } else {
3486 ret = lttng_metadata_printf(session,
3487 " ... %llu,\n",
3488 entry->end.value);
3489 }
141ddf28 3490 }
7c559490
MD		 3491 if (ret)
3492 goto end;
141ddf28 3493 }
141ddf28
MD		 3494 }
3495 ret = print_tabs(session, nesting);
3496 if (ret)
3497 goto end;
ceabb767 3498 ret = lttng_metadata_printf(session, "}");
141ddf28
MD		 3499end:
3500 return ret;
3501}
3502
d83004aa
JD		 3503/*
3504 * Must be called with sessions_mutex held.
3505 */
c099397a 3506static
8cdc1a81 3507int _lttng_enum_field_statedump(struct lttng_kernel_session *session,
437d5aa5 3508 const struct lttng_kernel_event_field *field,
ceabb767
MD		 3509 size_t nesting)
3510{
3511 int ret;
437d5aa5 3512 const struct lttng_kernel_type_enum *enum_type;
ceabb767 3513
437d5aa5
MD		 3514 enum_type = lttng_kernel_get_type_enum(field->type);
3515 WARN_ON_ONCE(!enum_type);
3516 ret = _lttng_enum_type_statedump(session, enum_type, nesting);
ceabb767
MD		 3517 if (ret)
3518 return ret;
3519 return lttng_field_name_statedump(session, field, nesting);
3520}
3521
3522static
8cdc1a81 3523int _lttng_integer_field_statedump(struct lttng_kernel_session *session,
437d5aa5 3524 const struct lttng_kernel_event_field *field,
ceabb767
MD		 3525 size_t nesting)
3526{
3527 int ret;
3528
437d5aa5
MD		 3529 ret = _lttng_integer_type_statedump(session, lttng_kernel_get_type_integer(field->type),
3530 lttng_kernel_string_encoding_none, nesting);
ceabb767
MD		 3531 if (ret)
3532 return ret;
3533 return lttng_field_name_statedump(session, field, nesting);
3534}
3535
3536static
8cdc1a81 3537int _lttng_string_type_statedump(struct lttng_kernel_session *session,
437d5aa5 3538 const struct lttng_kernel_type_string *type,
ceabb767
MD		 3539 size_t nesting)
3540{
3541 int ret;
3542
ceabb767
MD		 3543 /* Default encoding is UTF8 */
3544 ret = print_tabs(session, nesting);
3545 if (ret)
3546 return ret;
3547 ret = lttng_metadata_printf(session,
3548 "string%s",
437d5aa5 3549 type->encoding == lttng_kernel_string_encoding_ASCII ?
ceabb767
MD		 3550 " { encoding = ASCII; }" : "");
3551 return ret;
3552}
3553
3554static
8cdc1a81 3555int _lttng_string_field_statedump(struct lttng_kernel_session *session,
437d5aa5 3556 const struct lttng_kernel_event_field *field,
f513b2bf 3557 size_t nesting)
ceabb767 3558{
437d5aa5 3559 const struct lttng_kernel_type_string *string_type;
ceabb767
MD		 3560 int ret;
3561
437d5aa5
MD		 3562 string_type = lttng_kernel_get_type_string(field->type);
3563 WARN_ON_ONCE(!string_type);
3564 ret = _lttng_string_type_statedump(session, string_type, nesting);
ceabb767
MD		 3565 if (ret)
3566 return ret;
3567 return lttng_field_name_statedump(session, field, nesting);
3568}
3569
3570/*
3571 * Must be called with sessions_mutex held.
3572 */
3573static
8cdc1a81 3574int _lttng_type_statedump(struct lttng_kernel_session *session,
437d5aa5
MD		 3575 const struct lttng_kernel_type_common *type,
3576 enum lttng_kernel_string_encoding parent_encoding,
ceabb767 3577 size_t nesting)
c099397a 3578{
c099397a 3579 int ret = 0;
c099397a 3580
12bb2edb
MD		 3581 switch (type->type) {
3582 case lttng_kernel_type_integer:
437d5aa5
MD		 3583 ret = _lttng_integer_type_statedump(session,
3584 lttng_kernel_get_type_integer(type),
3585 parent_encoding, nesting);
8070f5c0 3586 break;
437d5aa5
MD		 3587 case lttng_kernel_type_enum:
3588 ret = _lttng_enum_type_statedump(session,
3589 lttng_kernel_get_type_enum(type),
3590 nesting);
8070f5c0 3591 break;
12bb2edb 3592 case lttng_kernel_type_string:
437d5aa5
MD		 3593 ret = _lttng_string_type_statedump(session,
3594 lttng_kernel_get_type_string(type),
3595 nesting);
ceabb767 3596 break;
437d5aa5
MD		 3597 case lttng_kernel_type_struct:
3598 ret = _lttng_struct_type_statedump(session,
3599 lttng_kernel_get_type_struct(type),
3600 nesting);
8070f5c0 3601 break;
437d5aa5
MD		 3602 case lttng_kernel_type_variant:
3603 ret = _lttng_variant_type_statedump(session,
3604 lttng_kernel_get_type_variant(type),
51ef4536 3605 nesting, NULL);
ceabb767
MD		 3606 break;
3607
3608 /* Nested arrays and sequences are not supported yet. */
437d5aa5
MD		 3609 case lttng_kernel_type_array:
3610 case lttng_kernel_type_sequence:
ceabb767
MD		 3611 default:
3612 WARN_ON_ONCE(1);
3613 return -EINVAL;
8070f5c0 3614 }
ceabb767
MD		 3615 return ret;
3616}
8070f5c0 3617
ceabb767
MD		 3618/*
3619 * Must be called with sessions_mutex held.
3620 */
3621static
8cdc1a81 3622int _lttng_field_statedump(struct lttng_kernel_session *session,
437d5aa5 3623 const struct lttng_kernel_event_field *field,
51ef4536
MD		 3624 size_t nesting,
3625 const char **prev_field_name_p)
ceabb767 3626{
51ef4536 3627 const char *prev_field_name = NULL;
ceabb767 3628 int ret = 0;
27d2368e 3629
51ef4536
MD		 3630 if (prev_field_name_p)
3631 prev_field_name = *prev_field_name_p;
437d5aa5 3632 switch (field->type->type) {
12bb2edb 3633 case lttng_kernel_type_integer:
ceabb767
MD		 3634 ret = _lttng_integer_field_statedump(session, field, nesting);
3635 break;
437d5aa5 3636 case lttng_kernel_type_enum:
ceabb767 3637 ret = _lttng_enum_field_statedump(session, field, nesting);
8070f5c0 3638 break;
12bb2edb 3639 case lttng_kernel_type_string:
ceabb767 3640 ret = _lttng_string_field_statedump(session, field, nesting);
8070f5c0 3641 break;
437d5aa5 3642 case lttng_kernel_type_struct:
ceabb767 3643 ret = _lttng_struct_field_statedump(session, field, nesting);
f513b2bf 3644 break;
437d5aa5 3645 case lttng_kernel_type_array:
ceabb767 3646 ret = _lttng_array_field_statedump(session, field, nesting);
f513b2bf 3647 break;
437d5aa5 3648 case lttng_kernel_type_sequence:
51ef4536 3649 ret = _lttng_sequence_field_statedump(session, field, nesting, prev_field_name);
f513b2bf 3650 break;
437d5aa5 3651 case lttng_kernel_type_variant:
51ef4536 3652 ret = _lttng_variant_field_statedump(session, field, nesting, prev_field_name);
65c85aa6 3653 break;
f513b2bf 3654
8070f5c0
MD		 3655 default:
3656 WARN_ON_ONCE(1);
3657 return -EINVAL;
3658 }
51ef4536
MD		 3659 if (prev_field_name_p)
3660 *prev_field_name_p = field->name;
8070f5c0
MD		 3661 return ret;
3662}
3663
3664static
8cdc1a81 3665int _lttng_context_metadata_statedump(struct lttng_kernel_session *session,
437d5aa5 3666 struct lttng_kernel_ctx *ctx)
8070f5c0 3667{
51ef4536 3668 const char *prev_field_name = NULL;
8070f5c0
MD		 3669 int ret = 0;
3670 int i;
3671
3672 if (!ctx)
3673 return 0;
3674 for (i = 0; i < ctx->nr_fields; i++) {
437d5aa5 3675 const struct lttng_kernel_ctx_field *field = &ctx->fields[i];
8070f5c0 3676
51ef4536 3677 ret = _lttng_field_statedump(session, field->event_field, 2, &prev_field_name);
8070f5c0
MD		 3678 if (ret)
3679 return ret;
3680 }
3681 return ret;
3682}
3683
3684static
8cdc1a81 3685int _lttng_fields_metadata_statedump(struct lttng_kernel_session *session,
a67ba386 3686 struct lttng_kernel_event_recorder *event_recorder)
8070f5c0 3687{
51ef4536 3688 const char *prev_field_name = NULL;
a67ba386 3689 const struct lttng_kernel_event_desc *desc = event_recorder->priv->parent.desc;
8070f5c0
MD		 3690 int ret = 0;
3691 int i;
3692
6c8c025b
MD		 3693 for (i = 0; i < desc->tp_class->nr_fields; i++) {
3694 const struct lttng_kernel_event_field *field = desc->tp_class->fields[i];
8070f5c0 3695
51ef4536 3696 ret = _lttng_field_statedump(session, field, 2, &prev_field_name);
8070f5c0
MD		 3697 if (ret)
3698 return ret;
c099397a
MD		 3699 }
3700 return ret;
3701}
3702
d83004aa
JD		 3703/*
3704 * Must be called with sessions_mutex held.
8b97fd42
MD		 3705 * The entire event metadata is printed as a single atomic metadata
3706 * transaction.
d83004aa 3707 */
c099397a 3708static
8cdc1a81 3709int _lttng_event_metadata_statedump(struct lttng_kernel_session *session,
f7d06400 3710 struct lttng_kernel_channel_buffer *chan,
a67ba386 3711 struct lttng_kernel_event_recorder *event_recorder)
c099397a
MD		 3712{
3713 int ret = 0;
3714
a67ba386 3715 if (event_recorder->priv->metadata_dumped || !LTTNG_READ_ONCE(session->active))
c099397a 3716 return 0;
f7d06400 3717 if (chan->priv->channel_type == METADATA_CHANNEL)
c099397a
MD		 3718 return 0;
3719
8b97fd42
MD		 3720 lttng_metadata_begin(session);
3721
c099397a
MD		 3722 ret = lttng_metadata_printf(session,
3723 "event {\n"
ae734547 3724 " name = \"%s\";\n"
c099397a 3725 " id = %u;\n"
b9074a1b 3726 " stream_id = %u;\n",
a67ba386
MD		 3727 event_recorder->priv->parent.desc->event_name,
3728 event_recorder->priv->id,
f7d06400 3729 event_recorder->chan->priv->id);
c099397a
MD		 3730 if (ret)
3731 goto end;
3732
8070f5c0 3733 ret = lttng_metadata_printf(session,
8070f5c0
MD		 3734 " fields := struct {\n"
3735 );
3736 if (ret)
3737 goto end;
3738
a67ba386 3739 ret = _lttng_fields_metadata_statedump(session, event_recorder);
c099397a
MD		 3740 if (ret)
3741 goto end;
3742
3743 /*
3744 * LTTng space reservation can only reserve multiples of the
3745 * byte size.
3746 */
3747 ret = lttng_metadata_printf(session,
9115fbdc
MD		 3748 " };\n"
3749 "};\n\n");
c099397a
MD		 3750 if (ret)
3751 goto end;
3752
a67ba386 3753 event_recorder->priv->metadata_dumped = 1;
c099397a 3754end:
8b97fd42 3755 lttng_metadata_end(session);
c099397a
MD		 3756 return ret;
3757
3758}
3759
d83004aa
JD		 3760/*
3761 * Must be called with sessions_mutex held.
8b97fd42
MD		 3762 * The entire channel metadata is printed as a single atomic metadata
3763 * transaction.
d83004aa 3764 */
c099397a 3765static
8cdc1a81 3766int _lttng_channel_metadata_statedump(struct lttng_kernel_session *session,
f7d06400 3767 struct lttng_kernel_channel_buffer *chan)
c099397a
MD		 3768{
3769 int ret = 0;
3770
f7d06400 3771 if (chan->priv->metadata_dumped || !LTTNG_READ_ONCE(session->active))
c099397a 3772 return 0;
d83004aa 3773
f7d06400 3774 if (chan->priv->channel_type == METADATA_CHANNEL)
c099397a
MD		 3775 return 0;
3776
8b97fd42
MD		 3777 lttng_metadata_begin(session);
3778
f7d06400 3779 WARN_ON_ONCE(!chan->priv->header_type);
c099397a
MD		 3780 ret = lttng_metadata_printf(session,
3781 "stream {\n"
3782 " id = %u;\n"
9115fbdc 3783 " event.header := %s;\n"
b9074a1b 3784 " packet.context := struct packet_context;\n",
f7d06400
MD		 3785 chan->priv->id,
3786 chan->priv->header_type == 1 ? "struct event_header_compact" :
c099397a
MD		 3787 "struct event_header_large");
3788 if (ret)
3789 goto end;
3790
f7d06400 3791 if (chan->priv->ctx) {
b9074a1b
MD		 3792 ret = lttng_metadata_printf(session,
3793 " event.context := struct {\n");
3794 if (ret)
3795 goto end;
3796 }
f7d06400 3797 ret = _lttng_context_metadata_statedump(session, chan->priv->ctx);
8070f5c0
MD		 3798 if (ret)
3799 goto end;
f7d06400 3800 if (chan->priv->ctx) {
b9074a1b
MD		 3801 ret = lttng_metadata_printf(session,
3802 " };\n");
3803 if (ret)
3804 goto end;
3805 }
8070f5c0
MD		 3806
3807 ret = lttng_metadata_printf(session,
b9074a1b 3808 "};\n\n");
8070f5c0 3809
f7d06400 3810 chan->priv->metadata_dumped = 1;
c099397a 3811end:
8b97fd42 3812 lttng_metadata_end(session);
c099397a
MD		 3813 return ret;
3814}
3815
d83004aa
JD		 3816/*
3817 * Must be called with sessions_mutex held.
3818 */
9115fbdc 3819static
8cdc1a81 3820int _lttng_stream_packet_context_declare(struct lttng_kernel_session *session)
9115fbdc
MD		 3821{
3822 return lttng_metadata_printf(session,
3823 "struct packet_context {\n"
a3ccff4f
MD		 3824 " uint64_clock_monotonic_t timestamp_begin;\n"
3825 " uint64_clock_monotonic_t timestamp_end;\n"
576ca06a
MD		 3826 " uint64_t content_size;\n"
3827 " uint64_t packet_size;\n"
5b3cf4f9 3828 " uint64_t packet_seq_num;\n"
a9afe705 3829 " unsigned long events_discarded;\n"
9115fbdc 3830 " uint32_t cpu_id;\n"
c6dfdf6f 3831 "};\n\n"
9115fbdc
MD		 3832 );
3833}
3834
3835/*
3836 * Compact header:
3837 * id: range: 0 - 30.
3838 * id 31 is reserved to indicate an extended header.
3839 *
3840 * Large header:
3841 * id: range: 0 - 65534.
3842 * id 65535 is reserved to indicate an extended header.
d83004aa
JD		 3843 *
3844 * Must be called with sessions_mutex held.
9115fbdc
MD		 3845 */
3846static
8cdc1a81 3847int _lttng_event_header_declare(struct lttng_kernel_session *session)
9115fbdc
MD		 3848{
3849 return lttng_metadata_printf(session,
3850 "struct event_header_compact {\n"
3851 " enum : uint5_t { compact = 0 ... 30, extended = 31 } id;\n"
3852 " variant <id> {\n"
3853 " struct {\n"
a3ccff4f 3854 " uint27_clock_monotonic_t timestamp;\n"
9115fbdc
MD		 3855 " } compact;\n"
3856 " struct {\n"
3857 " uint32_t id;\n"
a3ccff4f 3858 " uint64_clock_monotonic_t timestamp;\n"
9115fbdc
MD		 3859 " } extended;\n"
3860 " } v;\n"
3861 "} align(%u);\n"
3862 "\n"
3863 "struct event_header_large {\n"
3864 " enum : uint16_t { compact = 0 ... 65534, extended = 65535 } id;\n"
3865 " variant <id> {\n"
3866 " struct {\n"
a3ccff4f 3867 " uint32_clock_monotonic_t timestamp;\n"
9115fbdc
MD		 3868 " } compact;\n"
3869 " struct {\n"
3870 " uint32_t id;\n"
a3ccff4f 3871 " uint64_clock_monotonic_t timestamp;\n"
9115fbdc
MD		 3872 " } extended;\n"
3873 " } v;\n"
3874 "} align(%u);\n\n",
a90917c3
MD		 3875 lttng_alignof(uint32_t) * CHAR_BIT,
3876 lttng_alignof(uint16_t) * CHAR_BIT
9115fbdc
MD		 3877 );
3878}
3879
a3ccff4f
MD		 3880 /*
3881 * Approximation of NTP time of day to clock monotonic correlation,
3882 * taken at start of trace.
3883 * Yes, this is only an approximation. Yes, we can (and will) do better
3884 * in future versions.
5283350b
MD		 3885 * This function may return a negative offset. It may happen if the
3886 * system sets the REALTIME clock to 0 after boot.
3636085f
MJ		 3887 *
3888 * Use 64bit timespec on kernels that have it, this makes 32bit arch
3889 * y2038 compliant.
a3ccff4f
MD		 3890 */
3891static
8967b2a3 3892int64_t measure_clock_offset(void)
a3ccff4f 3893{
dcb93448 3894 uint64_t monotonic_avg, monotonic[2], realtime;
2754583e 3895 uint64_t tcf = trace_clock_freq();
dcb93448 3896 int64_t offset;
a3ccff4f 3897 unsigned long flags;
3636085f
MJ		 3898#ifdef LTTNG_KERNEL_HAS_TIMESPEC64
3899 struct timespec64 rts = { 0, 0 };
3900#else
3901 struct timespec rts = { 0, 0 };
3902#endif
a3ccff4f
MD		 3903
3904 /* Disable interrupts to increase correlation precision. */
3905 local_irq_save(flags);
3906 monotonic[0] = trace_clock_read64();
3636085f
MJ		 3907#ifdef LTTNG_KERNEL_HAS_TIMESPEC64
3908 ktime_get_real_ts64(&rts);
3909#else
216b6baa 3910 getnstimeofday(&rts);
3636085f 3911#endif
a3ccff4f
MD		 3912 monotonic[1] = trace_clock_read64();
3913 local_irq_restore(flags);
3914
dcb93448 3915 monotonic_avg = (monotonic[0] + monotonic[1]) >> 1;
2754583e
MD		 3916 realtime = (uint64_t) rts.tv_sec * tcf;
3917 if (tcf == NSEC_PER_SEC) {
3918 realtime += rts.tv_nsec;
3919 } else {
3920 uint64_t n = rts.tv_nsec * tcf;
3921
3922 do_div(n, NSEC_PER_SEC);
3923 realtime += n;
3924 }
dcb93448 3925 offset = (int64_t) realtime - monotonic_avg;
a3ccff4f
MD		 3926 return offset;
3927}
3928
3eb95be4 3929static
8cdc1a81 3930int print_escaped_ctf_string(struct lttng_kernel_session *session, const char *string)
3eb95be4 3931{
0f3eaad4 3932 int ret = 0;
3eb95be4
JR		 3933 size_t i;
3934 char cur;
3935
3936 i = 0;
3937 cur = string[i];
3938 while (cur != '\0') {
3939 switch (cur) {
3940 case '\n':
3941 ret = lttng_metadata_printf(session, "%s", "\\n");
3942 break;
3943 case '\\':
3944 case '"':
3945 ret = lttng_metadata_printf(session, "%c", '\\');
3946 if (ret)
3947 goto error;
3948 /* We still print the current char */
c190d76e 3949 lttng_fallthrough;
3eb95be4
JR		 3950 default:
3951 ret = lttng_metadata_printf(session, "%c", cur);
3952 break;
3953 }
3954
3955 if (ret)
3956 goto error;
3957
3958 cur = string[++i];
3959 }
3960error:
3961 return ret;
3962}
3963
3964static
8cdc1a81 3965int print_metadata_escaped_field(struct lttng_kernel_session *session, const char *field,
1c88f269 3966 const char *field_value)
3eb95be4
JR		 3967{
3968 int ret;
3969
1c88f269 3970 ret = lttng_metadata_printf(session, " %s = \"", field);
3eb95be4
JR		 3971 if (ret)
3972 goto error;
3973
1c88f269 3974 ret = print_escaped_ctf_string(session, field_value);
3eb95be4
JR		 3975 if (ret)
3976 goto error;
3977
3978 ret = lttng_metadata_printf(session, "\";\n");
3979
3980error:
3981 return ret;
3982}
3983
c099397a
MD		 3984/*
3985 * Output metadata into this session's metadata buffers.
d83004aa 3986 * Must be called with sessions_mutex held.
c099397a
MD		 3987 */
3988static
8cdc1a81 3989int _lttng_session_metadata_statedump(struct lttng_kernel_session *session)
c099397a 3990{
8cdc1a81 3991 unsigned char *uuid_c = session->priv->uuid.b;
a82c63f1 3992 unsigned char uuid_s[37], clock_uuid_s[BOOT_ID_LEN];
c88f762e 3993 const char *product_uuid;
f7d06400 3994 struct lttng_kernel_channel_buffer_private *chan_priv;
a67ba386 3995 struct lttng_kernel_event_recorder_private *event_recorder_priv;
c099397a
MD		 3996 int ret = 0;
3997
585e5dcc 3998 if (!LTTNG_READ_ONCE(session->active))
c099397a 3999 return 0;
8b97fd42
MD		 4000
4001 lttng_metadata_begin(session);
4002
8cdc1a81 4003 if (session->priv->metadata_dumped)
c099397a 4004 goto skip_session;
c099397a 4005
d793d5e1 4006 snprintf(uuid_s, sizeof(uuid_s),
30bdb6e4
MD		 4007 "%02x%02x%02x%02x-%02x%02x-%02x%02x-%02x%02x-%02x%02x%02x%02x%02x%02x",
4008 uuid_c[0], uuid_c[1], uuid_c[2], uuid_c[3],
4009 uuid_c[4], uuid_c[5], uuid_c[6], uuid_c[7],
4010 uuid_c[8], uuid_c[9], uuid_c[10], uuid_c[11],
4011 uuid_c[12], uuid_c[13], uuid_c[14], uuid_c[15]);
d793d5e1
MD		 4012
4013 ret = lttng_metadata_printf(session,
9115fbdc
MD		 4014 "typealias integer { size = 8; align = %u; signed = false; } := uint8_t;\n"
4015 "typealias integer { size = 16; align = %u; signed = false; } := uint16_t;\n"
4016 "typealias integer { size = 32; align = %u; signed = false; } := uint32_t;\n"
4017 "typealias integer { size = 64; align = %u; signed = false; } := uint64_t;\n"
a9afe705 4018 "typealias integer { size = %u; align = %u; signed = false; } := unsigned long;\n"
9115fbdc
MD		 4019 "typealias integer { size = 5; align = 1; signed = false; } := uint5_t;\n"
4020 "typealias integer { size = 27; align = 1; signed = false; } := uint27_t;\n"
d793d5e1
MD		 4021 "\n"
4022 "trace {\n"
4023 " major = %u;\n"
4024 " minor = %u;\n"
30bdb6e4 4025 " uuid = \"%s\";\n"
d793d5e1
MD		 4026 " byte_order = %s;\n"
4027 " packet.header := struct {\n"
4028 " uint32_t magic;\n"
1ec3f75a 4029 " uint8_t uuid[16];\n"
d793d5e1 4030 " uint32_t stream_id;\n"
5594698f 4031 " uint64_t stream_instance_id;\n"
0eb25f58 4032 " };\n"
d793d5e1 4033 "};\n\n",
a90917c3
MD		 4034 lttng_alignof(uint8_t) * CHAR_BIT,
4035 lttng_alignof(uint16_t) * CHAR_BIT,
4036 lttng_alignof(uint32_t) * CHAR_BIT,
4037 lttng_alignof(uint64_t) * CHAR_BIT,
a9afe705
MD		 4038 sizeof(unsigned long) * CHAR_BIT,
4039 lttng_alignof(unsigned long) * CHAR_BIT,
c6c9e10f
MD		 4040 CTF_SPEC_MAJOR,
4041 CTF_SPEC_MINOR,
d793d5e1 4042 uuid_s,
5214fa50 4043#if __BYTE_ORDER == __BIG_ENDIAN
d793d5e1
MD		 4044 "be"
4045#else
4046 "le"
4047#endif
4048 );
4049 if (ret)
4050 goto end;
4051
99dc9597
MD		 4052 ret = lttng_metadata_printf(session,
4053 "env {\n"
a6058143 4054 " hostname = \"%s\";\n"
c6c9e10f 4055 " domain = \"kernel\";\n"
99dc9597 4056 " sysname = \"%s\";\n"
c6c9e10f
MD		 4057 " kernel_release = \"%s\";\n"
4058 " kernel_version = \"%s\";\n"
4059 " tracer_name = \"lttng-modules\";\n"
4060 " tracer_major = %d;\n"
4061 " tracer_minor = %d;\n"
4062 " tracer_patchlevel = %d;\n"
3eb95be4 4063 " trace_buffering_scheme = \"global\";\n",
3d0d43db 4064 current->nsproxy->uts_ns->name.nodename,
99dc9597
MD		 4065 utsname()->sysname,
4066 utsname()->release,
c6c9e10f
MD		 4067 utsname()->version,
4068 LTTNG_MODULES_MAJOR_VERSION,
4069 LTTNG_MODULES_MINOR_VERSION,
4070 LTTNG_MODULES_PATCHLEVEL_VERSION
99dc9597
MD		 4071 );
4072 if (ret)
4073 goto end;
4074
8cdc1a81 4075 ret = print_metadata_escaped_field(session, "trace_name", session->priv->name);
1c88f269
JR		 4076 if (ret)
4077 goto end;
4078 ret = print_metadata_escaped_field(session, "trace_creation_datetime",
8cdc1a81 4079 session->priv->creation_time);
3eb95be4
JR		 4080 if (ret)
4081 goto end;
4082
c88f762e
GB		 4083 /* Add the product UUID to the 'env' section */
4084 product_uuid = dmi_get_system_info(DMI_PRODUCT_UUID);
4085 if (product_uuid) {
4086 ret = lttng_metadata_printf(session,
4087 " product_uuid = \"%s\";\n",
4088 product_uuid
4089 );
4090 if (ret)
4091 goto end;
4092 }
4093
4094 /* Close the 'env' section */
3eb95be4
JR		 4095 ret = lttng_metadata_printf(session, "};\n\n");
4096 if (ret)
4097 goto end;
4098
a3ccff4f
MD		 4099 ret = lttng_metadata_printf(session,
4100 "clock {\n"
2754583e
MD		 4101 " name = \"%s\";\n",
4102 trace_clock_name()
a82c63f1
MD		 4103 );
4104 if (ret)
4105 goto end;
4106
4107 if (!trace_clock_uuid(clock_uuid_s)) {
4108 ret = lttng_metadata_printf(session,
7c27cb17 4109 " uuid = \"%s\";\n",
a82c63f1
MD		 4110 clock_uuid_s
4111 );
4112 if (ret)
4113 goto end;
4114 }
4115
4116 ret = lttng_metadata_printf(session,
2754583e 4117 " description = \"%s\";\n"
a3ccff4f
MD		 4118 " freq = %llu; /* Frequency, in Hz */\n"
4119 " /* clock value offset from Epoch is: offset * (1/freq) */\n"
8967b2a3 4120 " offset = %lld;\n"
a3ccff4f 4121 "};\n\n",
2754583e 4122 trace_clock_description(),
a3ccff4f 4123 (unsigned long long) trace_clock_freq(),
8967b2a3 4124 (long long) measure_clock_offset()
a3ccff4f
MD		 4125 );
4126 if (ret)
4127 goto end;
4128
4129 ret = lttng_metadata_printf(session,
4130 "typealias integer {\n"
4131 " size = 27; align = 1; signed = false;\n"
2754583e 4132 " map = clock.%s.value;\n"
a3ccff4f
MD		 4133 "} := uint27_clock_monotonic_t;\n"
4134 "\n"
4135 "typealias integer {\n"
4136 " size = 32; align = %u; signed = false;\n"
2754583e 4137 " map = clock.%s.value;\n"
a3ccff4f
MD		 4138 "} := uint32_clock_monotonic_t;\n"
4139 "\n"
4140 "typealias integer {\n"
4141 " size = 64; align = %u; signed = false;\n"
2754583e 4142 " map = clock.%s.value;\n"
a3ccff4f 4143 "} := uint64_clock_monotonic_t;\n\n",
2754583e 4144 trace_clock_name(),
a3ccff4f 4145 lttng_alignof(uint32_t) * CHAR_BIT,
2754583e
MD		 4146 trace_clock_name(),
4147 lttng_alignof(uint64_t) * CHAR_BIT,
4148 trace_clock_name()
a3ccff4f
MD		 4149 );
4150 if (ret)
4151 goto end;
4152
a90917c3 4153 ret = _lttng_stream_packet_context_declare(session);
9115fbdc
MD		 4154 if (ret)
4155 goto end;
4156
a90917c3 4157 ret = _lttng_event_header_declare(session);
9115fbdc
MD		 4158 if (ret)
4159 goto end;
4160
c099397a 4161skip_session:
f7d06400
MD		 4162 list_for_each_entry(chan_priv, &session->priv->chan, node) {
4163 ret = _lttng_channel_metadata_statedump(session, chan_priv->pub);
c099397a
MD		 4164 if (ret)
4165 goto end;
4166 }
4167
8cdc1a81 4168 list_for_each_entry(event_recorder_priv, &session->priv->events, node) {
a67ba386
MD		 4169 ret = _lttng_event_metadata_statedump(session, event_recorder_priv->pub->chan,
4170 event_recorder_priv->pub);
c099397a
MD		 4171 if (ret)
4172 goto end;
4173 }
8cdc1a81 4174 session->priv->metadata_dumped = 1;
c099397a 4175end:
8b97fd42 4176 lttng_metadata_end(session);
c099397a
MD		 4177 return ret;
4178}
4179
c0e31d2e 4180/**
a90917c3 4181 * lttng_transport_register - LTT transport registration
c0e31d2e
MD		 4182 * @transport: transport structure
4183 *
4184 * Registers a transport which can be used as output to extract the data out of
4185 * LTTng. The module calling this registration function must ensure that no
4186 * trap-inducing code will be executed by the transport functions. E.g.
263b6c88 4187 * vmalloc_sync_mappings() must be called between a vmalloc and the moment the memory
c0e31d2e 4188 * is made visible to the transport function. This registration acts as a
263b6c88 4189 * vmalloc_sync_mappings. Therefore, only if the module allocates virtual memory
c0e31d2e
MD		 4190 * after its registration must it synchronize the TLBs.
4191 */
a90917c3 4192void lttng_transport_register(struct lttng_transport *transport)
c0e31d2e
MD		 4193{
4194 /*
4195 * Make sure no page fault can be triggered by the module about to be
4196 * registered. We deal with this here so we don't have to call
263b6c88 4197 * vmalloc_sync_mappings() in each module's init.
c0e31d2e 4198 */
263b6c88 4199 wrapper_vmalloc_sync_mappings();
c0e31d2e
MD		 4200
4201 mutex_lock(&sessions_mutex);
a90917c3 4202 list_add_tail(&transport->node, &lttng_transport_list);
c0e31d2e
MD		 4203 mutex_unlock(&sessions_mutex);
4204}
a90917c3 4205EXPORT_SYMBOL_GPL(lttng_transport_register);
c0e31d2e
MD		 4206
4207/**
a90917c3 4208 * lttng_transport_unregister - LTT transport unregistration
c0e31d2e
MD		 4209 * @transport: transport structure
4210 */
a90917c3 4211void lttng_transport_unregister(struct lttng_transport *transport)
c0e31d2e
MD		 4212{
4213 mutex_lock(&sessions_mutex);
4214 list_del(&transport->node);
4215 mutex_unlock(&sessions_mutex);
4216}
a90917c3 4217EXPORT_SYMBOL_GPL(lttng_transport_unregister);
c0e31d2e 4218
a101fa10
MD		 4219void lttng_counter_transport_register(struct lttng_counter_transport *transport)
4220{
4221 /*
4222 * Make sure no page fault can be triggered by the module about to be
4223 * registered. We deal with this here so we don't have to call
4224 * vmalloc_sync_mappings() in each module's init.
4225 */
4226 wrapper_vmalloc_sync_mappings();
4227
4228 mutex_lock(&sessions_mutex);
4229 list_add_tail(&transport->node, &lttng_counter_transport_list);
4230 mutex_unlock(&sessions_mutex);
4231}
4232EXPORT_SYMBOL_GPL(lttng_counter_transport_register);
4233
4234void lttng_counter_transport_unregister(struct lttng_counter_transport *transport)
4235{
4236 mutex_lock(&sessions_mutex);
4237 list_del(&transport->node);
4238 mutex_unlock(&sessions_mutex);
4239}
4240EXPORT_SYMBOL_GPL(lttng_counter_transport_unregister);
4241
5f4c791e 4242#if (LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(4,10,0))
1e367326
MD		 4243
4244enum cpuhp_state lttng_hp_prepare;
4245enum cpuhp_state lttng_hp_online;
4246
4247static int lttng_hotplug_prepare(unsigned int cpu, struct hlist_node *node)
4248{
4249 struct lttng_cpuhp_node *lttng_node;
4250
4251 lttng_node = container_of(node, struct lttng_cpuhp_node, node);
4252 switch (lttng_node->component) {
4253 case LTTNG_RING_BUFFER_FRONTEND:
4254 return 0;
4255 case LTTNG_RING_BUFFER_BACKEND:
4256 return lttng_cpuhp_rb_backend_prepare(cpu, lttng_node);
4257 case LTTNG_RING_BUFFER_ITER:
4258 return 0;
4259 case LTTNG_CONTEXT_PERF_COUNTERS:
4260 return 0;
4261 default:
4262 return -EINVAL;
4263 }
4264}
4265
4266static int lttng_hotplug_dead(unsigned int cpu, struct hlist_node *node)
4267{
4268 struct lttng_cpuhp_node *lttng_node;
4269
4270 lttng_node = container_of(node, struct lttng_cpuhp_node, node);
4271 switch (lttng_node->component) {
4272 case LTTNG_RING_BUFFER_FRONTEND:
4273 return lttng_cpuhp_rb_frontend_dead(cpu, lttng_node);
4274 case LTTNG_RING_BUFFER_BACKEND:
4275 return 0;
4276 case LTTNG_RING_BUFFER_ITER:
4277 return 0;
4278 case LTTNG_CONTEXT_PERF_COUNTERS:
4279 return lttng_cpuhp_perf_counter_dead(cpu, lttng_node);
4280 default:
4281 return -EINVAL;
4282 }
4283}
4284
4285static int lttng_hotplug_online(unsigned int cpu, struct hlist_node *node)
4286{
4287 struct lttng_cpuhp_node *lttng_node;
4288
4289 lttng_node = container_of(node, struct lttng_cpuhp_node, node);
4290 switch (lttng_node->component) {
4291 case LTTNG_RING_BUFFER_FRONTEND:
4292 return lttng_cpuhp_rb_frontend_online(cpu, lttng_node);
4293 case LTTNG_RING_BUFFER_BACKEND:
4294 return 0;
4295 case LTTNG_RING_BUFFER_ITER:
4296 return lttng_cpuhp_rb_iter_online(cpu, lttng_node);
4297 case LTTNG_CONTEXT_PERF_COUNTERS:
4298 return lttng_cpuhp_perf_counter_online(cpu, lttng_node);
4299 default:
4300 return -EINVAL;
4301 }
4302}
4303
4304static int lttng_hotplug_offline(unsigned int cpu, struct hlist_node *node)
4305{
4306 struct lttng_cpuhp_node *lttng_node;
4307
4308 lttng_node = container_of(node, struct lttng_cpuhp_node, node);
4309 switch (lttng_node->component) {
4310 case LTTNG_RING_BUFFER_FRONTEND:
4311 return lttng_cpuhp_rb_frontend_offline(cpu, lttng_node);
4312 case LTTNG_RING_BUFFER_BACKEND:
4313 return 0;
4314 case LTTNG_RING_BUFFER_ITER:
4315 return 0;
4316 case LTTNG_CONTEXT_PERF_COUNTERS:
4317 return 0;
4318 default:
4319 return -EINVAL;
4320 }
4321}
4322
4323static int __init lttng_init_cpu_hotplug(void)
4324{
4325 int ret;
4326
4327 ret = cpuhp_setup_state_multi(CPUHP_BP_PREPARE_DYN, "lttng:prepare",
4328 lttng_hotplug_prepare,
4329 lttng_hotplug_dead);
4330 if (ret < 0) {
4331 return ret;
4332 }
4333 lttng_hp_prepare = ret;
4334 lttng_rb_set_hp_prepare(ret);
4335
4336 ret = cpuhp_setup_state_multi(CPUHP_AP_ONLINE_DYN, "lttng:online",
4337 lttng_hotplug_online,
4338 lttng_hotplug_offline);
4339 if (ret < 0) {
4340 cpuhp_remove_multi_state(lttng_hp_prepare);
4341 lttng_hp_prepare = 0;
4342 return ret;
4343 }
4344 lttng_hp_online = ret;
4345 lttng_rb_set_hp_online(ret);
4346
4347 return 0;
4348}
4349
4350static void __exit lttng_exit_cpu_hotplug(void)
4351{
4352 lttng_rb_set_hp_online(0);
4353 cpuhp_remove_multi_state(lttng_hp_online);
4354 lttng_rb_set_hp_prepare(0);
4355 cpuhp_remove_multi_state(lttng_hp_prepare);
4356}
4357
5f4c791e 4358#else /* #if (LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(4,10,0)) */
1e367326
MD		 4359static int lttng_init_cpu_hotplug(void)
4360{
4361 return 0;
4362}
4363static void lttng_exit_cpu_hotplug(void)
4364{
4365}
5f4c791e 4366#endif /* #else #if (LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(4,10,0)) */
1e367326 4367
a90917c3 4368static int __init lttng_events_init(void)
4e3c1b9b 4369{
1c25284c
MD		 4370 int ret;
4371
453b2495
JD		 4372 ret = wrapper_lttng_fixup_sig(THIS_MODULE);
4373 if (ret)
4374 return ret;
389d7070 4375 ret = wrapper_get_pfnblock_flags_mask_init();
36561775
MD		 4376 if (ret)
4377 return ret;
4378 ret = wrapper_get_pageblock_flags_mask_init();
114667d5
MD		 4379 if (ret)
4380 return ret;
4381 ret = lttng_probes_init();
389d7070
MD		 4382 if (ret)
4383 return ret;
07dfc1d0 4384 ret = lttng_context_init();
20591cf7
MD		 4385 if (ret)
4386 return ret;
07dfc1d0
MD		 4387 ret = lttng_tracepoint_init();
4388 if (ret)
4389 goto error_tp;
a67ba386
MD		 4390 event_recorder_cache = KMEM_CACHE(lttng_kernel_event_recorder, 0);
4391 if (!event_recorder_cache) {
20591cf7 4392 ret = -ENOMEM;
a67ba386 4393 goto error_kmem_event_recorder;
dffef45d 4394 }
a67ba386
MD		 4395 event_recorder_private_cache = KMEM_CACHE(lttng_kernel_event_recorder_private, 0);
4396 if (!event_recorder_private_cache) {
4397 ret = -ENOMEM;
4398 goto error_kmem_event_recorder_private;
4399 }
4400 event_notifier_cache = KMEM_CACHE(lttng_kernel_event_notifier, 0);
dffef45d
FD		 4401 if (!event_notifier_cache) {
4402 ret = -ENOMEM;
4403 goto error_kmem_event_notifier;
20591cf7 4404 }
a67ba386
MD		 4405 event_notifier_private_cache = KMEM_CACHE(lttng_kernel_event_notifier_private, 0);
4406 if (!event_notifier_private_cache) {
4407 ret = -ENOMEM;
4408 goto error_kmem_event_notifier_private;
4409 }
80996790 4410 ret = lttng_abi_init();
02119ee5
MD		 4411 if (ret)
4412 goto error_abi;
0c956676
MD		 4413 ret = lttng_logger_init();
4414 if (ret)
4415 goto error_logger;
1e367326
MD		 4416 ret = lttng_init_cpu_hotplug();
4417 if (ret)
4418 goto error_hotplug;
101215b7 4419 printk(KERN_NOTICE "LTTng: Loaded modules v%s.%s.%s%s (%s)%s%s\n",
489de24b
MJ		 4420 __stringify(LTTNG_MODULES_MAJOR_VERSION),
4421 __stringify(LTTNG_MODULES_MINOR_VERSION),
4422 __stringify(LTTNG_MODULES_PATCHLEVEL_VERSION),
4423 LTTNG_MODULES_EXTRAVERSION,
101215b7
MJ		 4424 LTTNG_VERSION_NAME,
4425#ifdef LTTNG_EXTRA_VERSION_GIT
4426 LTTNG_EXTRA_VERSION_GIT[0] == '\0' ? "" : " - " LTTNG_EXTRA_VERSION_GIT,
4427#else
4428 "",
4429#endif
4430#ifdef LTTNG_EXTRA_VERSION_NAME
4431 LTTNG_EXTRA_VERSION_NAME[0] == '\0' ? "" : " - " LTTNG_EXTRA_VERSION_NAME);
4432#else
4433 "");
4434#endif
c1551e57
MJ		 4435#ifdef CONFIG_LTTNG_EXPERIMENTAL_BITWISE_ENUM
4436 printk(KERN_NOTICE "LTTng: Experimental bitwise enum enabled.\n");
4437#endif /* CONFIG_LTTNG_EXPERIMENTAL_BITWISE_ENUM */
4e3c1b9b 4438 return 0;
0c956676 4439
1e367326
MD		 4440error_hotplug:
4441 lttng_logger_exit();
0c956676
MD		 4442error_logger:
4443 lttng_abi_exit();
02119ee5 4444error_abi:
a67ba386
MD		 4445 kmem_cache_destroy(event_notifier_private_cache);
4446error_kmem_event_notifier_private:
dffef45d
FD		 4447 kmem_cache_destroy(event_notifier_cache);
4448error_kmem_event_notifier:
a67ba386
MD		 4449 kmem_cache_destroy(event_recorder_private_cache);
4450error_kmem_event_recorder_private:
4451 kmem_cache_destroy(event_recorder_cache);
4452error_kmem_event_recorder:
20591cf7 4453 lttng_tracepoint_exit();
07dfc1d0
MD		 4454error_tp:
4455 lttng_context_exit();
101215b7 4456 printk(KERN_NOTICE "LTTng: Failed to load modules v%s.%s.%s%s (%s)%s%s\n",
489de24b
MJ		 4457 __stringify(LTTNG_MODULES_MAJOR_VERSION),
4458 __stringify(LTTNG_MODULES_MINOR_VERSION),
4459 __stringify(LTTNG_MODULES_PATCHLEVEL_VERSION),
4460 LTTNG_MODULES_EXTRAVERSION,
101215b7
MJ		 4461 LTTNG_VERSION_NAME,
4462#ifdef LTTNG_EXTRA_VERSION_GIT
4463 LTTNG_EXTRA_VERSION_GIT[0] == '\0' ? "" : " - " LTTNG_EXTRA_VERSION_GIT,
4464#else
4465 "",
4466#endif
4467#ifdef LTTNG_EXTRA_VERSION_NAME
4468 LTTNG_EXTRA_VERSION_NAME[0] == '\0' ? "" : " - " LTTNG_EXTRA_VERSION_NAME);
4469#else
4470 "");
4471#endif
1c25284c 4472 return ret;
4e3c1b9b
MD		 4473}
4474
a90917c3 4475module_init(lttng_events_init);
11b5a3c2 4476
a90917c3 4477static void __exit lttng_events_exit(void)
4e3c1b9b 4478{
8cdc1a81 4479 struct lttng_kernel_session_private *session_priv, *tmpsession_priv;
92e94819 4480
1e367326 4481 lttng_exit_cpu_hotplug();
0c956676 4482 lttng_logger_exit();
80996790 4483 lttng_abi_exit();
8cdc1a81
MD		 4484 list_for_each_entry_safe(session_priv, tmpsession_priv, &sessions, list)
4485 lttng_session_destroy(session_priv->pub);
a67ba386
MD		 4486 kmem_cache_destroy(event_recorder_cache);
4487 kmem_cache_destroy(event_recorder_private_cache);
dffef45d 4488 kmem_cache_destroy(event_notifier_cache);
a67ba386 4489 kmem_cache_destroy(event_notifier_private_cache);
20591cf7 4490 lttng_tracepoint_exit();
07dfc1d0 4491 lttng_context_exit();
101215b7 4492 printk(KERN_NOTICE "LTTng: Unloaded modules v%s.%s.%s%s (%s)%s%s\n",
489de24b
MJ		 4493 __stringify(LTTNG_MODULES_MAJOR_VERSION),
4494 __stringify(LTTNG_MODULES_MINOR_VERSION),
4495 __stringify(LTTNG_MODULES_PATCHLEVEL_VERSION),
4496 LTTNG_MODULES_EXTRAVERSION,
101215b7
MJ		 4497 LTTNG_VERSION_NAME,
4498#ifdef LTTNG_EXTRA_VERSION_GIT
4499 LTTNG_EXTRA_VERSION_GIT[0] == '\0' ? "" : " - " LTTNG_EXTRA_VERSION_GIT,
4500#else
4501 "",
4502#endif
4503#ifdef LTTNG_EXTRA_VERSION_NAME
4504 LTTNG_EXTRA_VERSION_NAME[0] == '\0' ? "" : " - " LTTNG_EXTRA_VERSION_NAME);
4505#else
4506 "");
4507#endif
4e3c1b9b 4508}
92e94819 4509
a90917c3 4510module_exit(lttng_events_exit);
11b5a3c2 4511
7ca11e77 4512#include <generated/patches.h>
101215b7
MJ		 4513#ifdef LTTNG_EXTRA_VERSION_GIT
4514MODULE_INFO(extra_version_git, LTTNG_EXTRA_VERSION_GIT);
4515#endif
4516#ifdef LTTNG_EXTRA_VERSION_NAME
4517MODULE_INFO(extra_version_name, LTTNG_EXTRA_VERSION_NAME);
4518#endif
92e94819
MD		 4519MODULE_LICENSE("GPL and additional rights");
4520MODULE_AUTHOR("Mathieu Desnoyers <mathieu.desnoyers@efficios.com>");
1c124020 4521MODULE_DESCRIPTION("LTTng tracer");
9a9973ef
MD		 4522MODULE_VERSION(__stringify(LTTNG_MODULES_MAJOR_VERSION) "."
4523 __stringify(LTTNG_MODULES_MINOR_VERSION) "."
309370e2
MD		 4524 __stringify(LTTNG_MODULES_PATCHLEVEL_VERSION)
4525 LTTNG_MODULES_EXTRAVERSION);
LTTng 2.0/0.x modules
This page took 0.324149 seconds and 4 git commands to generate.