Add trace_creation_datetime to the metadata env field
[deliverable/lttng-modules.git] / lttng-abi.c
1 /* SPDX-License-Identifier: (GPL-2.0 or LGPL-2.1)
2 *
3 * lttng-abi.c
4 *
5 * LTTng ABI
6 *
7 * Copyright (C) 2010-2012 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
8 *
9 * Mimic system calls for:
10 * - session creation, returns a file descriptor or failure.
11 * - channel creation, returns a file descriptor or failure.
12 * - Operates on a session file descriptor
13 * - Takes all channel options as parameters.
14 * - stream get, returns a file descriptor or failure.
15 * - Operates on a channel file descriptor.
16 * - stream notifier get, returns a file descriptor or failure.
17 * - Operates on a channel file descriptor.
18 * - event creation, returns a file descriptor or failure.
19 * - Operates on a channel file descriptor
20 * - Takes an event name as parameter
21 * - Takes an instrumentation source as parameter
22 * - e.g. tracepoints, dynamic_probes...
23 * - Takes instrumentation source specific arguments.
24 */
25
26 #include <linux/module.h>
27 #include <linux/proc_fs.h>
28 #include <linux/anon_inodes.h>
29 #include <linux/file.h>
30 #include <linux/uaccess.h>
31 #include <linux/slab.h>
32 #include <linux/err.h>
33 #include <wrapper/vmalloc.h> /* for wrapper_vmalloc_sync_all() */
34 #include <wrapper/ringbuffer/vfs.h>
35 #include <wrapper/ringbuffer/backend.h>
36 #include <wrapper/ringbuffer/frontend.h>
37 #include <wrapper/poll.h>
38 #include <wrapper/file.h>
39 #include <wrapper/kref.h>
40 #include <lttng-string-utils.h>
41 #include <lttng-abi.h>
42 #include <lttng-abi-old.h>
43 #include <lttng-events.h>
44 #include <lttng-tracer.h>
45 #include <lttng-tp-mempool.h>
46 #include <lib/ringbuffer/frontend_types.h>
47
48 /*
49 * This is LTTng's own personal way to create a system call as an external
50 * module. We use ioctl() on /proc/lttng.
51 */
52
53 static struct proc_dir_entry *lttng_proc_dentry;
54 static const struct file_operations lttng_fops;
55 static const struct file_operations lttng_session_fops;
56 static const struct file_operations lttng_channel_fops;
57 static const struct file_operations lttng_metadata_fops;
58 static const struct file_operations lttng_event_fops;
59 static struct file_operations lttng_stream_ring_buffer_file_operations;
60
61 static int put_u64(uint64_t val, unsigned long arg);
62
63 /*
64 * Teardown management: opened file descriptors keep a refcount on the module,
65 * so it can only exit when all file descriptors are closed.
66 */
67
68 static
69 int lttng_abi_create_session(void)
70 {
71 struct lttng_session *session;
72 struct file *session_file;
73 int session_fd, ret;
74
75 session = lttng_session_create();
76 if (!session)
77 return -ENOMEM;
78 session_fd = lttng_get_unused_fd();
79 if (session_fd < 0) {
80 ret = session_fd;
81 goto fd_error;
82 }
83 session_file = anon_inode_getfile("[lttng_session]",
84 &lttng_session_fops,
85 session, O_RDWR);
86 if (IS_ERR(session_file)) {
87 ret = PTR_ERR(session_file);
88 goto file_error;
89 }
90 session->file = session_file;
91 fd_install(session_fd, session_file);
92 return session_fd;
93
94 file_error:
95 put_unused_fd(session_fd);
96 fd_error:
97 lttng_session_destroy(session);
98 return ret;
99 }
100
101 static
102 int lttng_abi_tracepoint_list(void)
103 {
104 struct file *tracepoint_list_file;
105 int file_fd, ret;
106
107 file_fd = lttng_get_unused_fd();
108 if (file_fd < 0) {
109 ret = file_fd;
110 goto fd_error;
111 }
112
113 tracepoint_list_file = anon_inode_getfile("[lttng_tracepoint_list]",
114 &lttng_tracepoint_list_fops,
115 NULL, O_RDWR);
116 if (IS_ERR(tracepoint_list_file)) {
117 ret = PTR_ERR(tracepoint_list_file);
118 goto file_error;
119 }
120 ret = lttng_tracepoint_list_fops.open(NULL, tracepoint_list_file);
121 if (ret < 0)
122 goto open_error;
123 fd_install(file_fd, tracepoint_list_file);
124 return file_fd;
125
126 open_error:
127 fput(tracepoint_list_file);
128 file_error:
129 put_unused_fd(file_fd);
130 fd_error:
131 return ret;
132 }
133
134 #ifndef CONFIG_HAVE_SYSCALL_TRACEPOINTS
135 static inline
136 int lttng_abi_syscall_list(void)
137 {
138 return -ENOSYS;
139 }
140 #else
141 static
142 int lttng_abi_syscall_list(void)
143 {
144 struct file *syscall_list_file;
145 int file_fd, ret;
146
147 file_fd = lttng_get_unused_fd();
148 if (file_fd < 0) {
149 ret = file_fd;
150 goto fd_error;
151 }
152
153 syscall_list_file = anon_inode_getfile("[lttng_syscall_list]",
154 &lttng_syscall_list_fops,
155 NULL, O_RDWR);
156 if (IS_ERR(syscall_list_file)) {
157 ret = PTR_ERR(syscall_list_file);
158 goto file_error;
159 }
160 ret = lttng_syscall_list_fops.open(NULL, syscall_list_file);
161 if (ret < 0)
162 goto open_error;
163 fd_install(file_fd, syscall_list_file);
164 return file_fd;
165
166 open_error:
167 fput(syscall_list_file);
168 file_error:
169 put_unused_fd(file_fd);
170 fd_error:
171 return ret;
172 }
173 #endif
174
175 static
176 void lttng_abi_tracer_version(struct lttng_kernel_tracer_version *v)
177 {
178 v->major = LTTNG_MODULES_MAJOR_VERSION;
179 v->minor = LTTNG_MODULES_MINOR_VERSION;
180 v->patchlevel = LTTNG_MODULES_PATCHLEVEL_VERSION;
181 }
182
183 static
184 void lttng_abi_tracer_abi_version(struct lttng_kernel_tracer_abi_version *v)
185 {
186 v->major = LTTNG_MODULES_ABI_MAJOR_VERSION;
187 v->minor = LTTNG_MODULES_ABI_MINOR_VERSION;
188 }
189
190 static
191 long lttng_abi_add_context(struct file *file,
192 struct lttng_kernel_context *context_param,
193 struct lttng_ctx **ctx, struct lttng_session *session)
194 {
195
196 if (session->been_active)
197 return -EPERM;
198
199 switch (context_param->ctx) {
200 case LTTNG_KERNEL_CONTEXT_PID:
201 return lttng_add_pid_to_ctx(ctx);
202 case LTTNG_KERNEL_CONTEXT_PRIO:
203 return lttng_add_prio_to_ctx(ctx);
204 case LTTNG_KERNEL_CONTEXT_NICE:
205 return lttng_add_nice_to_ctx(ctx);
206 case LTTNG_KERNEL_CONTEXT_VPID:
207 return lttng_add_vpid_to_ctx(ctx);
208 case LTTNG_KERNEL_CONTEXT_TID:
209 return lttng_add_tid_to_ctx(ctx);
210 case LTTNG_KERNEL_CONTEXT_VTID:
211 return lttng_add_vtid_to_ctx(ctx);
212 case LTTNG_KERNEL_CONTEXT_PPID:
213 return lttng_add_ppid_to_ctx(ctx);
214 case LTTNG_KERNEL_CONTEXT_VPPID:
215 return lttng_add_vppid_to_ctx(ctx);
216 case LTTNG_KERNEL_CONTEXT_PERF_COUNTER:
217 context_param->u.perf_counter.name[LTTNG_KERNEL_SYM_NAME_LEN - 1] = '\0';
218 return lttng_add_perf_counter_to_ctx(context_param->u.perf_counter.type,
219 context_param->u.perf_counter.config,
220 context_param->u.perf_counter.name,
221 ctx);
222 case LTTNG_KERNEL_CONTEXT_PROCNAME:
223 return lttng_add_procname_to_ctx(ctx);
224 case LTTNG_KERNEL_CONTEXT_HOSTNAME:
225 return lttng_add_hostname_to_ctx(ctx);
226 case LTTNG_KERNEL_CONTEXT_CPU_ID:
227 return lttng_add_cpu_id_to_ctx(ctx);
228 case LTTNG_KERNEL_CONTEXT_INTERRUPTIBLE:
229 return lttng_add_interruptible_to_ctx(ctx);
230 case LTTNG_KERNEL_CONTEXT_NEED_RESCHEDULE:
231 return lttng_add_need_reschedule_to_ctx(ctx);
232 case LTTNG_KERNEL_CONTEXT_PREEMPTIBLE:
233 return lttng_add_preemptible_to_ctx(ctx);
234 case LTTNG_KERNEL_CONTEXT_MIGRATABLE:
235 return lttng_add_migratable_to_ctx(ctx);
236 case LTTNG_KERNEL_CONTEXT_CALLSTACK_KERNEL:
237 case LTTNG_KERNEL_CONTEXT_CALLSTACK_USER:
238 return lttng_add_callstack_to_ctx(ctx, context_param->ctx);
239 case LTTNG_KERNEL_CONTEXT_CGROUP_NS:
240 return lttng_add_cgroup_ns_to_ctx(ctx);
241 case LTTNG_KERNEL_CONTEXT_IPC_NS:
242 return lttng_add_ipc_ns_to_ctx(ctx);
243 case LTTNG_KERNEL_CONTEXT_MNT_NS:
244 return lttng_add_mnt_ns_to_ctx(ctx);
245 case LTTNG_KERNEL_CONTEXT_NET_NS:
246 return lttng_add_net_ns_to_ctx(ctx);
247 case LTTNG_KERNEL_CONTEXT_PID_NS:
248 return lttng_add_pid_ns_to_ctx(ctx);
249 case LTTNG_KERNEL_CONTEXT_USER_NS:
250 return lttng_add_user_ns_to_ctx(ctx);
251 case LTTNG_KERNEL_CONTEXT_UTS_NS:
252 return lttng_add_uts_ns_to_ctx(ctx);
253 case LTTNG_KERNEL_CONTEXT_UID:
254 return lttng_add_uid_to_ctx(ctx);
255 case LTTNG_KERNEL_CONTEXT_EUID:
256 return lttng_add_euid_to_ctx(ctx);
257 case LTTNG_KERNEL_CONTEXT_SUID:
258 return lttng_add_suid_to_ctx(ctx);
259 case LTTNG_KERNEL_CONTEXT_GID:
260 return lttng_add_gid_to_ctx(ctx);
261 case LTTNG_KERNEL_CONTEXT_EGID:
262 return lttng_add_egid_to_ctx(ctx);
263 case LTTNG_KERNEL_CONTEXT_SGID:
264 return lttng_add_sgid_to_ctx(ctx);
265 case LTTNG_KERNEL_CONTEXT_VUID:
266 return lttng_add_vuid_to_ctx(ctx);
267 case LTTNG_KERNEL_CONTEXT_VEUID:
268 return lttng_add_veuid_to_ctx(ctx);
269 case LTTNG_KERNEL_CONTEXT_VSUID:
270 return lttng_add_vsuid_to_ctx(ctx);
271 case LTTNG_KERNEL_CONTEXT_VGID:
272 return lttng_add_vgid_to_ctx(ctx);
273 case LTTNG_KERNEL_CONTEXT_VEGID:
274 return lttng_add_vegid_to_ctx(ctx);
275 case LTTNG_KERNEL_CONTEXT_VSGID:
276 return lttng_add_vsgid_to_ctx(ctx);
277 default:
278 return -EINVAL;
279 }
280 }
281
282 /**
283 * lttng_ioctl - lttng syscall through ioctl
284 *
285 * @file: the file
286 * @cmd: the command
287 * @arg: command arg
288 *
289 * This ioctl implements lttng commands:
290 * LTTNG_KERNEL_SESSION
291 * Returns a LTTng trace session file descriptor
292 * LTTNG_KERNEL_TRACER_VERSION
293 * Returns the LTTng kernel tracer version
294 * LTTNG_KERNEL_TRACEPOINT_LIST
295 * Returns a file descriptor listing available tracepoints
296 * LTTNG_KERNEL_WAIT_QUIESCENT
297 * Returns after all previously running probes have completed
298 * LTTNG_KERNEL_TRACER_ABI_VERSION
299 * Returns the LTTng kernel tracer ABI version
300 *
301 * The returned session will be deleted when its file descriptor is closed.
302 */
303 static
304 long lttng_ioctl(struct file *file, unsigned int cmd, unsigned long arg)
305 {
306 switch (cmd) {
307 case LTTNG_KERNEL_OLD_SESSION:
308 case LTTNG_KERNEL_SESSION:
309 return lttng_abi_create_session();
310 case LTTNG_KERNEL_OLD_TRACER_VERSION:
311 {
312 struct lttng_kernel_tracer_version v;
313 struct lttng_kernel_old_tracer_version oldv;
314 struct lttng_kernel_old_tracer_version *uversion =
315 (struct lttng_kernel_old_tracer_version __user *) arg;
316
317 lttng_abi_tracer_version(&v);
318 oldv.major = v.major;
319 oldv.minor = v.minor;
320 oldv.patchlevel = v.patchlevel;
321
322 if (copy_to_user(uversion, &oldv, sizeof(oldv)))
323 return -EFAULT;
324 return 0;
325 }
326 case LTTNG_KERNEL_TRACER_VERSION:
327 {
328 struct lttng_kernel_tracer_version version;
329 struct lttng_kernel_tracer_version *uversion =
330 (struct lttng_kernel_tracer_version __user *) arg;
331
332 lttng_abi_tracer_version(&version);
333
334 if (copy_to_user(uversion, &version, sizeof(version)))
335 return -EFAULT;
336 return 0;
337 }
338 case LTTNG_KERNEL_TRACER_ABI_VERSION:
339 {
340 struct lttng_kernel_tracer_abi_version version;
341 struct lttng_kernel_tracer_abi_version *uversion =
342 (struct lttng_kernel_tracer_abi_version __user *) arg;
343
344 lttng_abi_tracer_abi_version(&version);
345
346 if (copy_to_user(uversion, &version, sizeof(version)))
347 return -EFAULT;
348 return 0;
349 }
350 case LTTNG_KERNEL_OLD_TRACEPOINT_LIST:
351 case LTTNG_KERNEL_TRACEPOINT_LIST:
352 return lttng_abi_tracepoint_list();
353 case LTTNG_KERNEL_SYSCALL_LIST:
354 return lttng_abi_syscall_list();
355 case LTTNG_KERNEL_OLD_WAIT_QUIESCENT:
356 case LTTNG_KERNEL_WAIT_QUIESCENT:
357 synchronize_trace();
358 return 0;
359 case LTTNG_KERNEL_OLD_CALIBRATE:
360 {
361 struct lttng_kernel_old_calibrate __user *ucalibrate =
362 (struct lttng_kernel_old_calibrate __user *) arg;
363 struct lttng_kernel_old_calibrate old_calibrate;
364 struct lttng_kernel_calibrate calibrate;
365 int ret;
366
367 if (copy_from_user(&old_calibrate, ucalibrate, sizeof(old_calibrate)))
368 return -EFAULT;
369 calibrate.type = old_calibrate.type;
370 ret = lttng_calibrate(&calibrate);
371 if (copy_to_user(ucalibrate, &old_calibrate, sizeof(old_calibrate)))
372 return -EFAULT;
373 return ret;
374 }
375 case LTTNG_KERNEL_CALIBRATE:
376 {
377 struct lttng_kernel_calibrate __user *ucalibrate =
378 (struct lttng_kernel_calibrate __user *) arg;
379 struct lttng_kernel_calibrate calibrate;
380 int ret;
381
382 if (copy_from_user(&calibrate, ucalibrate, sizeof(calibrate)))
383 return -EFAULT;
384 ret = lttng_calibrate(&calibrate);
385 if (copy_to_user(ucalibrate, &calibrate, sizeof(calibrate)))
386 return -EFAULT;
387 return ret;
388 }
389 default:
390 return -ENOIOCTLCMD;
391 }
392 }
393
394 static const struct file_operations lttng_fops = {
395 .owner = THIS_MODULE,
396 .unlocked_ioctl = lttng_ioctl,
397 #ifdef CONFIG_COMPAT
398 .compat_ioctl = lttng_ioctl,
399 #endif
400 };
401
402 static
403 int lttng_abi_create_channel(struct file *session_file,
404 struct lttng_kernel_channel *chan_param,
405 enum channel_type channel_type)
406 {
407 struct lttng_session *session = session_file->private_data;
408 const struct file_operations *fops = NULL;
409 const char *transport_name;
410 struct lttng_channel *chan;
411 struct file *chan_file;
412 int chan_fd;
413 int ret = 0;
414
415 chan_fd = lttng_get_unused_fd();
416 if (chan_fd < 0) {
417 ret = chan_fd;
418 goto fd_error;
419 }
420 switch (channel_type) {
421 case PER_CPU_CHANNEL:
422 fops = &lttng_channel_fops;
423 break;
424 case METADATA_CHANNEL:
425 fops = &lttng_metadata_fops;
426 break;
427 }
428
429 chan_file = anon_inode_getfile("[lttng_channel]",
430 fops,
431 NULL, O_RDWR);
432 if (IS_ERR(chan_file)) {
433 ret = PTR_ERR(chan_file);
434 goto file_error;
435 }
436 switch (channel_type) {
437 case PER_CPU_CHANNEL:
438 if (chan_param->output == LTTNG_KERNEL_SPLICE) {
439 transport_name = chan_param->overwrite ?
440 "relay-overwrite" : "relay-discard";
441 } else if (chan_param->output == LTTNG_KERNEL_MMAP) {
442 transport_name = chan_param->overwrite ?
443 "relay-overwrite-mmap" : "relay-discard-mmap";
444 } else {
445 return -EINVAL;
446 }
447 break;
448 case METADATA_CHANNEL:
449 if (chan_param->output == LTTNG_KERNEL_SPLICE)
450 transport_name = "relay-metadata";
451 else if (chan_param->output == LTTNG_KERNEL_MMAP)
452 transport_name = "relay-metadata-mmap";
453 else
454 return -EINVAL;
455 break;
456 default:
457 transport_name = "<unknown>";
458 break;
459 }
460 if (!atomic_long_add_unless(&session_file->f_count, 1, LONG_MAX)) {
461 ret = -EOVERFLOW;
462 goto refcount_error;
463 }
464 /*
465 * We tolerate no failure path after channel creation. It will stay
466 * invariant for the rest of the session.
467 */
468 chan = lttng_channel_create(session, transport_name, NULL,
469 chan_param->subbuf_size,
470 chan_param->num_subbuf,
471 chan_param->switch_timer_interval,
472 chan_param->read_timer_interval,
473 channel_type);
474 if (!chan) {
475 ret = -EINVAL;
476 goto chan_error;
477 }
478 chan->file = chan_file;
479 chan_file->private_data = chan;
480 fd_install(chan_fd, chan_file);
481
482 return chan_fd;
483
484 chan_error:
485 atomic_long_dec(&session_file->f_count);
486 refcount_error:
487 fput(chan_file);
488 file_error:
489 put_unused_fd(chan_fd);
490 fd_error:
491 return ret;
492 }
493
494 static
495 int lttng_abi_session_set_name(struct lttng_session *session,
496 struct lttng_kernel_session_name *name)
497 {
498 size_t len;
499
500 len = strnlen(name->name, LTTNG_KERNEL_SESSION_NAME_LEN);
501
502 if (len == LTTNG_KERNEL_SESSION_NAME_LEN) {
503 /* Name is too long/malformed */
504 return -EINVAL;
505 }
506
507 strcpy(session->name, name->name);
508 return 0;
509 }
510
511 static
512 int lttng_abi_session_set_creation_time(struct lttng_session *session,
513 struct lttng_kernel_session_creation_time *time)
514 {
515 size_t len;
516
517 len = strnlen(time->iso8601, LTTNG_KERNEL_SESSION_CREATION_TIME_ISO8601_LEN);
518
519 if (len == LTTNG_KERNEL_SESSION_CREATION_TIME_ISO8601_LEN) {
520 /* Time is too long/malformed */
521 return -EINVAL;
522 }
523
524 strcpy(session->creation_time, time->iso8601);
525 return 0;
526 }
527
528 /**
529 * lttng_session_ioctl - lttng session fd ioctl
530 *
531 * @file: the file
532 * @cmd: the command
533 * @arg: command arg
534 *
535 * This ioctl implements lttng commands:
536 * LTTNG_KERNEL_CHANNEL
537 * Returns a LTTng channel file descriptor
538 * LTTNG_KERNEL_ENABLE
539 * Enables tracing for a session (weak enable)
540 * LTTNG_KERNEL_DISABLE
541 * Disables tracing for a session (strong disable)
542 * LTTNG_KERNEL_METADATA
543 * Returns a LTTng metadata file descriptor
544 * LTTNG_KERNEL_SESSION_TRACK_PID
545 * Add PID to session tracker
546 * LTTNG_KERNEL_SESSION_UNTRACK_PID
547 * Remove PID from session tracker
548 *
549 * The returned channel will be deleted when its file descriptor is closed.
550 */
551 static
552 long lttng_session_ioctl(struct file *file, unsigned int cmd, unsigned long arg)
553 {
554 struct lttng_session *session = file->private_data;
555
556 switch (cmd) {
557 case LTTNG_KERNEL_OLD_CHANNEL:
558 {
559 struct lttng_kernel_channel chan_param;
560 struct lttng_kernel_old_channel old_chan_param;
561
562 if (copy_from_user(&old_chan_param,
563 (struct lttng_kernel_old_channel __user *) arg,
564 sizeof(struct lttng_kernel_old_channel)))
565 return -EFAULT;
566 chan_param.overwrite = old_chan_param.overwrite;
567 chan_param.subbuf_size = old_chan_param.subbuf_size;
568 chan_param.num_subbuf = old_chan_param.num_subbuf;
569 chan_param.switch_timer_interval = old_chan_param.switch_timer_interval;
570 chan_param.read_timer_interval = old_chan_param.read_timer_interval;
571 chan_param.output = old_chan_param.output;
572
573 return lttng_abi_create_channel(file, &chan_param,
574 PER_CPU_CHANNEL);
575 }
576 case LTTNG_KERNEL_CHANNEL:
577 {
578 struct lttng_kernel_channel chan_param;
579
580 if (copy_from_user(&chan_param,
581 (struct lttng_kernel_channel __user *) arg,
582 sizeof(struct lttng_kernel_channel)))
583 return -EFAULT;
584 return lttng_abi_create_channel(file, &chan_param,
585 PER_CPU_CHANNEL);
586 }
587 case LTTNG_KERNEL_OLD_SESSION_START:
588 case LTTNG_KERNEL_OLD_ENABLE:
589 case LTTNG_KERNEL_SESSION_START:
590 case LTTNG_KERNEL_ENABLE:
591 return lttng_session_enable(session);
592 case LTTNG_KERNEL_OLD_SESSION_STOP:
593 case LTTNG_KERNEL_OLD_DISABLE:
594 case LTTNG_KERNEL_SESSION_STOP:
595 case LTTNG_KERNEL_DISABLE:
596 return lttng_session_disable(session);
597 case LTTNG_KERNEL_OLD_METADATA:
598 {
599 struct lttng_kernel_channel chan_param;
600 struct lttng_kernel_old_channel old_chan_param;
601
602 if (copy_from_user(&old_chan_param,
603 (struct lttng_kernel_old_channel __user *) arg,
604 sizeof(struct lttng_kernel_old_channel)))
605 return -EFAULT;
606 chan_param.overwrite = old_chan_param.overwrite;
607 chan_param.subbuf_size = old_chan_param.subbuf_size;
608 chan_param.num_subbuf = old_chan_param.num_subbuf;
609 chan_param.switch_timer_interval = old_chan_param.switch_timer_interval;
610 chan_param.read_timer_interval = old_chan_param.read_timer_interval;
611 chan_param.output = old_chan_param.output;
612
613 return lttng_abi_create_channel(file, &chan_param,
614 METADATA_CHANNEL);
615 }
616 case LTTNG_KERNEL_METADATA:
617 {
618 struct lttng_kernel_channel chan_param;
619
620 if (copy_from_user(&chan_param,
621 (struct lttng_kernel_channel __user *) arg,
622 sizeof(struct lttng_kernel_channel)))
623 return -EFAULT;
624 return lttng_abi_create_channel(file, &chan_param,
625 METADATA_CHANNEL);
626 }
627 case LTTNG_KERNEL_SESSION_TRACK_PID:
628 return lttng_session_track_pid(session, (int) arg);
629 case LTTNG_KERNEL_SESSION_UNTRACK_PID:
630 return lttng_session_untrack_pid(session, (int) arg);
631 case LTTNG_KERNEL_SESSION_LIST_TRACKER_PIDS:
632 return lttng_session_list_tracker_pids(session);
633 case LTTNG_KERNEL_SESSION_METADATA_REGEN:
634 return lttng_session_metadata_regenerate(session);
635 case LTTNG_KERNEL_SESSION_STATEDUMP:
636 return lttng_session_statedump(session);
637 case LTTNG_KERNEL_SESSION_SET_NAME:
638 {
639 struct lttng_kernel_session_name name;
640
641 if (copy_from_user(&name,
642 (struct lttng_kernel_session_name __user *) arg,
643 sizeof(struct lttng_kernel_session_name)))
644 return -EFAULT;
645 return lttng_abi_session_set_name(session, &name);
646 }
647 case LTTNG_KERNEL_SESSION_SET_CREATION_TIME:
648 {
649 struct lttng_kernel_session_creation_time time;
650
651 if (copy_from_user(&time,
652 (struct lttng_kernel_session_creation_time __user *) arg,
653 sizeof(struct lttng_kernel_session_creation_time)))
654 return -EFAULT;
655 return lttng_abi_session_set_creation_time(session, &time);
656 }
657 default:
658 return -ENOIOCTLCMD;
659 }
660 }
661
662 /*
663 * Called when the last file reference is dropped.
664 *
665 * Big fat note: channels and events are invariant for the whole session after
666 * their creation. So this session destruction also destroys all channel and
667 * event structures specific to this session (they are not destroyed when their
668 * individual file is released).
669 */
670 static
671 int lttng_session_release(struct inode *inode, struct file *file)
672 {
673 struct lttng_session *session = file->private_data;
674
675 if (session)
676 lttng_session_destroy(session);
677 return 0;
678 }
679
680 static const struct file_operations lttng_session_fops = {
681 .owner = THIS_MODULE,
682 .release = lttng_session_release,
683 .unlocked_ioctl = lttng_session_ioctl,
684 #ifdef CONFIG_COMPAT
685 .compat_ioctl = lttng_session_ioctl,
686 #endif
687 };
688
689 /**
690 * lttng_metadata_ring_buffer_poll - LTTng ring buffer poll file operation
691 * @filp: the file
692 * @wait: poll table
693 *
694 * Handles the poll operations for the metadata channels.
695 */
696 static
697 unsigned int lttng_metadata_ring_buffer_poll(struct file *filp,
698 poll_table *wait)
699 {
700 struct lttng_metadata_stream *stream = filp->private_data;
701 struct lib_ring_buffer *buf = stream->priv;
702 int finalized;
703 unsigned int mask = 0;
704
705 if (filp->f_mode & FMODE_READ) {
706 poll_wait_set_exclusive(wait);
707 poll_wait(filp, &stream->read_wait, wait);
708
709 finalized = stream->finalized;
710
711 /*
712 * lib_ring_buffer_is_finalized() contains a smp_rmb()
713 * ordering finalized load before offsets loads.
714 */
715 WARN_ON(atomic_long_read(&buf->active_readers) != 1);
716
717 if (finalized)
718 mask |= POLLHUP;
719
720 mutex_lock(&stream->metadata_cache->lock);
721 if (stream->metadata_cache->metadata_written >
722 stream->metadata_out)
723 mask |= POLLIN;
724 mutex_unlock(&stream->metadata_cache->lock);
725 }
726
727 return mask;
728 }
729
730 static
731 void lttng_metadata_ring_buffer_ioctl_put_next_subbuf(struct file *filp,
732 unsigned int cmd, unsigned long arg)
733 {
734 struct lttng_metadata_stream *stream = filp->private_data;
735
736 stream->metadata_out = stream->metadata_in;
737 }
738
739 /*
740 * Reset the counter of how much metadata has been consumed to 0. That way,
741 * the consumer receives the content of the metadata cache unchanged. This is
742 * different from the metadata_regenerate where the offset from epoch is
743 * resampled, here we want the exact same content as the last time the metadata
744 * was generated. This command is only possible if all the metadata written
745 * in the cache has been output to the metadata stream to avoid corrupting the
746 * metadata file.
747 *
748 * Return 0 on success, a negative value on error.
749 */
750 static
751 int lttng_metadata_cache_dump(struct lttng_metadata_stream *stream)
752 {
753 int ret;
754 struct lttng_metadata_cache *cache = stream->metadata_cache;
755
756 mutex_lock(&cache->lock);
757 if (stream->metadata_out != cache->metadata_written) {
758 ret = -EBUSY;
759 goto end;
760 }
761 stream->metadata_out = 0;
762 stream->metadata_in = 0;
763 wake_up_interruptible(&stream->read_wait);
764 ret = 0;
765
766 end:
767 mutex_unlock(&cache->lock);
768 return ret;
769 }
770
771 static
772 long lttng_metadata_ring_buffer_ioctl(struct file *filp,
773 unsigned int cmd, unsigned long arg)
774 {
775 int ret;
776 struct lttng_metadata_stream *stream = filp->private_data;
777 struct lib_ring_buffer *buf = stream->priv;
778
779 switch (cmd) {
780 case RING_BUFFER_GET_NEXT_SUBBUF:
781 {
782 struct lttng_metadata_stream *stream = filp->private_data;
783 struct lib_ring_buffer *buf = stream->priv;
784 struct channel *chan = buf->backend.chan;
785
786 ret = lttng_metadata_output_channel(stream, chan);
787 if (ret > 0) {
788 lib_ring_buffer_switch_slow(buf, SWITCH_ACTIVE);
789 ret = 0;
790 } else if (ret < 0)
791 goto err;
792 break;
793 }
794 case RING_BUFFER_GET_SUBBUF:
795 {
796 /*
797 * Random access is not allowed for metadata channel.
798 */
799 return -ENOSYS;
800 }
801 case RING_BUFFER_FLUSH_EMPTY: /* Fall-through. */
802 case RING_BUFFER_FLUSH:
803 {
804 struct lttng_metadata_stream *stream = filp->private_data;
805 struct lib_ring_buffer *buf = stream->priv;
806 struct channel *chan = buf->backend.chan;
807
808 /*
809 * Before doing the actual ring buffer flush, write up to one
810 * packet of metadata in the ring buffer.
811 */
812 ret = lttng_metadata_output_channel(stream, chan);
813 if (ret < 0)
814 goto err;
815 break;
816 }
817 case RING_BUFFER_GET_METADATA_VERSION:
818 {
819 struct lttng_metadata_stream *stream = filp->private_data;
820
821 return put_u64(stream->version, arg);
822 }
823 case RING_BUFFER_METADATA_CACHE_DUMP:
824 {
825 struct lttng_metadata_stream *stream = filp->private_data;
826
827 return lttng_metadata_cache_dump(stream);
828 }
829 default:
830 break;
831 }
832 /* PUT_SUBBUF is the one from lib ring buffer, unmodified. */
833
834 /* Performing lib ring buffer ioctl after our own. */
835 ret = lib_ring_buffer_ioctl(filp, cmd, arg, buf);
836 if (ret < 0)
837 goto err;
838
839 switch (cmd) {
840 case RING_BUFFER_PUT_NEXT_SUBBUF:
841 {
842 lttng_metadata_ring_buffer_ioctl_put_next_subbuf(filp,
843 cmd, arg);
844 break;
845 }
846 default:
847 break;
848 }
849 err:
850 return ret;
851 }
852
853 #ifdef CONFIG_COMPAT
854 static
855 long lttng_metadata_ring_buffer_compat_ioctl(struct file *filp,
856 unsigned int cmd, unsigned long arg)
857 {
858 int ret;
859 struct lttng_metadata_stream *stream = filp->private_data;
860 struct lib_ring_buffer *buf = stream->priv;
861
862 switch (cmd) {
863 case RING_BUFFER_GET_NEXT_SUBBUF:
864 {
865 struct lttng_metadata_stream *stream = filp->private_data;
866 struct lib_ring_buffer *buf = stream->priv;
867 struct channel *chan = buf->backend.chan;
868
869 ret = lttng_metadata_output_channel(stream, chan);
870 if (ret > 0) {
871 lib_ring_buffer_switch_slow(buf, SWITCH_ACTIVE);
872 ret = 0;
873 } else if (ret < 0)
874 goto err;
875 break;
876 }
877 case RING_BUFFER_GET_SUBBUF:
878 {
879 /*
880 * Random access is not allowed for metadata channel.
881 */
882 return -ENOSYS;
883 }
884 case RING_BUFFER_FLUSH_EMPTY: /* Fall-through. */
885 case RING_BUFFER_FLUSH:
886 {
887 struct lttng_metadata_stream *stream = filp->private_data;
888 struct lib_ring_buffer *buf = stream->priv;
889 struct channel *chan = buf->backend.chan;
890
891 /*
892 * Before doing the actual ring buffer flush, write up to one
893 * packet of metadata in the ring buffer.
894 */
895 ret = lttng_metadata_output_channel(stream, chan);
896 if (ret < 0)
897 goto err;
898 break;
899 }
900 case RING_BUFFER_GET_METADATA_VERSION:
901 {
902 struct lttng_metadata_stream *stream = filp->private_data;
903
904 return put_u64(stream->version, arg);
905 }
906 case RING_BUFFER_METADATA_CACHE_DUMP:
907 {
908 struct lttng_metadata_stream *stream = filp->private_data;
909
910 return lttng_metadata_cache_dump(stream);
911 }
912 default:
913 break;
914 }
915 /* PUT_SUBBUF is the one from lib ring buffer, unmodified. */
916
917 /* Performing lib ring buffer ioctl after our own. */
918 ret = lib_ring_buffer_compat_ioctl(filp, cmd, arg, buf);
919 if (ret < 0)
920 goto err;
921
922 switch (cmd) {
923 case RING_BUFFER_PUT_NEXT_SUBBUF:
924 {
925 lttng_metadata_ring_buffer_ioctl_put_next_subbuf(filp,
926 cmd, arg);
927 break;
928 }
929 default:
930 break;
931 }
932 err:
933 return ret;
934 }
935 #endif
936
937 /*
938 * This is not used by anonymous file descriptors. This code is left
939 * there if we ever want to implement an inode with open() operation.
940 */
941 static
942 int lttng_metadata_ring_buffer_open(struct inode *inode, struct file *file)
943 {
944 struct lttng_metadata_stream *stream = inode->i_private;
945 struct lib_ring_buffer *buf = stream->priv;
946
947 file->private_data = buf;
948 /*
949 * Since life-time of metadata cache differs from that of
950 * session, we need to keep our own reference on the transport.
951 */
952 if (!try_module_get(stream->transport->owner)) {
953 printk(KERN_WARNING "LTT : Can't lock transport module.\n");
954 return -EBUSY;
955 }
956 return lib_ring_buffer_open(inode, file, buf);
957 }
958
959 static
960 int lttng_metadata_ring_buffer_release(struct inode *inode, struct file *file)
961 {
962 struct lttng_metadata_stream *stream = file->private_data;
963 struct lib_ring_buffer *buf = stream->priv;
964
965 kref_put(&stream->metadata_cache->refcount, metadata_cache_destroy);
966 module_put(stream->transport->owner);
967 return lib_ring_buffer_release(inode, file, buf);
968 }
969
970 static
971 ssize_t lttng_metadata_ring_buffer_splice_read(struct file *in, loff_t *ppos,
972 struct pipe_inode_info *pipe, size_t len,
973 unsigned int flags)
974 {
975 struct lttng_metadata_stream *stream = in->private_data;
976 struct lib_ring_buffer *buf = stream->priv;
977
978 return lib_ring_buffer_splice_read(in, ppos, pipe, len,
979 flags, buf);
980 }
981
982 static
983 int lttng_metadata_ring_buffer_mmap(struct file *filp,
984 struct vm_area_struct *vma)
985 {
986 struct lttng_metadata_stream *stream = filp->private_data;
987 struct lib_ring_buffer *buf = stream->priv;
988
989 return lib_ring_buffer_mmap(filp, vma, buf);
990 }
991
992 static
993 const struct file_operations lttng_metadata_ring_buffer_file_operations = {
994 .owner = THIS_MODULE,
995 .open = lttng_metadata_ring_buffer_open,
996 .release = lttng_metadata_ring_buffer_release,
997 .poll = lttng_metadata_ring_buffer_poll,
998 .splice_read = lttng_metadata_ring_buffer_splice_read,
999 .mmap = lttng_metadata_ring_buffer_mmap,
1000 .unlocked_ioctl = lttng_metadata_ring_buffer_ioctl,
1001 .llseek = vfs_lib_ring_buffer_no_llseek,
1002 #ifdef CONFIG_COMPAT
1003 .compat_ioctl = lttng_metadata_ring_buffer_compat_ioctl,
1004 #endif
1005 };
1006
1007 static
1008 int lttng_abi_create_stream_fd(struct file *channel_file, void *stream_priv,
1009 const struct file_operations *fops)
1010 {
1011 int stream_fd, ret;
1012 struct file *stream_file;
1013
1014 stream_fd = lttng_get_unused_fd();
1015 if (stream_fd < 0) {
1016 ret = stream_fd;
1017 goto fd_error;
1018 }
1019 stream_file = anon_inode_getfile("[lttng_stream]", fops,
1020 stream_priv, O_RDWR);
1021 if (IS_ERR(stream_file)) {
1022 ret = PTR_ERR(stream_file);
1023 goto file_error;
1024 }
1025 /*
1026 * OPEN_FMODE, called within anon_inode_getfile/alloc_file, don't honor
1027 * FMODE_LSEEK, FMODE_PREAD nor FMODE_PWRITE. We need to read from this
1028 * file descriptor, so we set FMODE_PREAD here.
1029 */
1030 stream_file->f_mode |= FMODE_PREAD;
1031 fd_install(stream_fd, stream_file);
1032 /*
1033 * The stream holds a reference to the channel within the generic ring
1034 * buffer library, so no need to hold a refcount on the channel and
1035 * session files here.
1036 */
1037 return stream_fd;
1038
1039 file_error:
1040 put_unused_fd(stream_fd);
1041 fd_error:
1042 return ret;
1043 }
1044
1045 static
1046 int lttng_abi_open_stream(struct file *channel_file)
1047 {
1048 struct lttng_channel *channel = channel_file->private_data;
1049 struct lib_ring_buffer *buf;
1050 int ret;
1051 void *stream_priv;
1052
1053 buf = channel->ops->buffer_read_open(channel->chan);
1054 if (!buf)
1055 return -ENOENT;
1056
1057 stream_priv = buf;
1058 ret = lttng_abi_create_stream_fd(channel_file, stream_priv,
1059 &lttng_stream_ring_buffer_file_operations);
1060 if (ret < 0)
1061 goto fd_error;
1062
1063 return ret;
1064
1065 fd_error:
1066 channel->ops->buffer_read_close(buf);
1067 return ret;
1068 }
1069
1070 static
1071 int lttng_abi_open_metadata_stream(struct file *channel_file)
1072 {
1073 struct lttng_channel *channel = channel_file->private_data;
1074 struct lttng_session *session = channel->session;
1075 struct lib_ring_buffer *buf;
1076 int ret;
1077 struct lttng_metadata_stream *metadata_stream;
1078 void *stream_priv;
1079
1080 buf = channel->ops->buffer_read_open(channel->chan);
1081 if (!buf)
1082 return -ENOENT;
1083
1084 metadata_stream = kzalloc(sizeof(struct lttng_metadata_stream),
1085 GFP_KERNEL);
1086 if (!metadata_stream) {
1087 ret = -ENOMEM;
1088 goto nomem;
1089 }
1090 metadata_stream->metadata_cache = session->metadata_cache;
1091 init_waitqueue_head(&metadata_stream->read_wait);
1092 metadata_stream->priv = buf;
1093 stream_priv = metadata_stream;
1094 metadata_stream->transport = channel->transport;
1095
1096 /*
1097 * Since life-time of metadata cache differs from that of
1098 * session, we need to keep our own reference on the transport.
1099 */
1100 if (!try_module_get(metadata_stream->transport->owner)) {
1101 printk(KERN_WARNING "LTT : Can't lock transport module.\n");
1102 ret = -EINVAL;
1103 goto notransport;
1104 }
1105
1106 if (!lttng_kref_get(&session->metadata_cache->refcount)) {
1107 ret = -EOVERFLOW;
1108 goto kref_error;
1109 }
1110
1111 ret = lttng_abi_create_stream_fd(channel_file, stream_priv,
1112 &lttng_metadata_ring_buffer_file_operations);
1113 if (ret < 0)
1114 goto fd_error;
1115
1116 list_add(&metadata_stream->list,
1117 &session->metadata_cache->metadata_stream);
1118 return ret;
1119
1120 fd_error:
1121 kref_put(&session->metadata_cache->refcount, metadata_cache_destroy);
1122 kref_error:
1123 module_put(metadata_stream->transport->owner);
1124 notransport:
1125 kfree(metadata_stream);
1126 nomem:
1127 channel->ops->buffer_read_close(buf);
1128 return ret;
1129 }
1130
1131 static
1132 int lttng_abi_create_event(struct file *channel_file,
1133 struct lttng_kernel_event *event_param)
1134 {
1135 struct lttng_channel *channel = channel_file->private_data;
1136 int event_fd, ret;
1137 struct file *event_file;
1138 void *priv;
1139
1140 event_param->name[LTTNG_KERNEL_SYM_NAME_LEN - 1] = '\0';
1141 switch (event_param->instrumentation) {
1142 case LTTNG_KERNEL_KRETPROBE:
1143 event_param->u.kretprobe.symbol_name[LTTNG_KERNEL_SYM_NAME_LEN - 1] = '\0';
1144 break;
1145 case LTTNG_KERNEL_KPROBE:
1146 event_param->u.kprobe.symbol_name[LTTNG_KERNEL_SYM_NAME_LEN - 1] = '\0';
1147 break;
1148 case LTTNG_KERNEL_FUNCTION:
1149 event_param->u.ftrace.symbol_name[LTTNG_KERNEL_SYM_NAME_LEN - 1] = '\0';
1150 break;
1151 default:
1152 break;
1153 }
1154 event_fd = lttng_get_unused_fd();
1155 if (event_fd < 0) {
1156 ret = event_fd;
1157 goto fd_error;
1158 }
1159 event_file = anon_inode_getfile("[lttng_event]",
1160 &lttng_event_fops,
1161 NULL, O_RDWR);
1162 if (IS_ERR(event_file)) {
1163 ret = PTR_ERR(event_file);
1164 goto file_error;
1165 }
1166 /* The event holds a reference on the channel */
1167 if (!atomic_long_add_unless(&channel_file->f_count, 1, LONG_MAX)) {
1168 ret = -EOVERFLOW;
1169 goto refcount_error;
1170 }
1171 if (event_param->instrumentation == LTTNG_KERNEL_TRACEPOINT
1172 || event_param->instrumentation == LTTNG_KERNEL_SYSCALL) {
1173 struct lttng_enabler *enabler;
1174
1175 if (strutils_is_star_glob_pattern(event_param->name)) {
1176 /*
1177 * If the event name is a star globbing pattern,
1178 * we create the special star globbing enabler.
1179 */
1180 enabler = lttng_enabler_create(LTTNG_ENABLER_STAR_GLOB,
1181 event_param, channel);
1182 } else {
1183 enabler = lttng_enabler_create(LTTNG_ENABLER_NAME,
1184 event_param, channel);
1185 }
1186 priv = enabler;
1187 } else {
1188 struct lttng_event *event;
1189
1190 /*
1191 * We tolerate no failure path after event creation. It
1192 * will stay invariant for the rest of the session.
1193 */
1194 event = lttng_event_create(channel, event_param,
1195 NULL, NULL,
1196 event_param->instrumentation);
1197 WARN_ON_ONCE(!event);
1198 if (IS_ERR(event)) {
1199 ret = PTR_ERR(event);
1200 goto event_error;
1201 }
1202 priv = event;
1203 }
1204 event_file->private_data = priv;
1205 fd_install(event_fd, event_file);
1206 return event_fd;
1207
1208 event_error:
1209 atomic_long_dec(&channel_file->f_count);
1210 refcount_error:
1211 fput(event_file);
1212 file_error:
1213 put_unused_fd(event_fd);
1214 fd_error:
1215 return ret;
1216 }
1217
1218 /**
1219 * lttng_channel_ioctl - lttng syscall through ioctl
1220 *
1221 * @file: the file
1222 * @cmd: the command
1223 * @arg: command arg
1224 *
1225 * This ioctl implements lttng commands:
1226 * LTTNG_KERNEL_STREAM
1227 * Returns an event stream file descriptor or failure.
1228 * (typically, one event stream records events from one CPU)
1229 * LTTNG_KERNEL_EVENT
1230 * Returns an event file descriptor or failure.
1231 * LTTNG_KERNEL_CONTEXT
1232 * Prepend a context field to each event in the channel
1233 * LTTNG_KERNEL_ENABLE
1234 * Enable recording for events in this channel (weak enable)
1235 * LTTNG_KERNEL_DISABLE
1236 * Disable recording for events in this channel (strong disable)
1237 *
1238 * Channel and event file descriptors also hold a reference on the session.
1239 */
1240 static
1241 long lttng_channel_ioctl(struct file *file, unsigned int cmd, unsigned long arg)
1242 {
1243 struct lttng_channel *channel = file->private_data;
1244
1245 switch (cmd) {
1246 case LTTNG_KERNEL_OLD_STREAM:
1247 case LTTNG_KERNEL_STREAM:
1248 return lttng_abi_open_stream(file);
1249 case LTTNG_KERNEL_OLD_EVENT:
1250 {
1251 struct lttng_kernel_event *uevent_param;
1252 struct lttng_kernel_old_event *old_uevent_param;
1253 int ret;
1254
1255 uevent_param = kmalloc(sizeof(struct lttng_kernel_event),
1256 GFP_KERNEL);
1257 if (!uevent_param) {
1258 ret = -ENOMEM;
1259 goto old_event_end;
1260 }
1261 old_uevent_param = kmalloc(
1262 sizeof(struct lttng_kernel_old_event),
1263 GFP_KERNEL);
1264 if (!old_uevent_param) {
1265 ret = -ENOMEM;
1266 goto old_event_error_free_param;
1267 }
1268 if (copy_from_user(old_uevent_param,
1269 (struct lttng_kernel_old_event __user *) arg,
1270 sizeof(struct lttng_kernel_old_event))) {
1271 ret = -EFAULT;
1272 goto old_event_error_free_old_param;
1273 }
1274
1275 memcpy(uevent_param->name, old_uevent_param->name,
1276 sizeof(uevent_param->name));
1277 uevent_param->instrumentation =
1278 old_uevent_param->instrumentation;
1279
1280 switch (old_uevent_param->instrumentation) {
1281 case LTTNG_KERNEL_KPROBE:
1282 uevent_param->u.kprobe.addr =
1283 old_uevent_param->u.kprobe.addr;
1284 uevent_param->u.kprobe.offset =
1285 old_uevent_param->u.kprobe.offset;
1286 memcpy(uevent_param->u.kprobe.symbol_name,
1287 old_uevent_param->u.kprobe.symbol_name,
1288 sizeof(uevent_param->u.kprobe.symbol_name));
1289 break;
1290 case LTTNG_KERNEL_KRETPROBE:
1291 uevent_param->u.kretprobe.addr =
1292 old_uevent_param->u.kretprobe.addr;
1293 uevent_param->u.kretprobe.offset =
1294 old_uevent_param->u.kretprobe.offset;
1295 memcpy(uevent_param->u.kretprobe.symbol_name,
1296 old_uevent_param->u.kretprobe.symbol_name,
1297 sizeof(uevent_param->u.kretprobe.symbol_name));
1298 break;
1299 case LTTNG_KERNEL_FUNCTION:
1300 memcpy(uevent_param->u.ftrace.symbol_name,
1301 old_uevent_param->u.ftrace.symbol_name,
1302 sizeof(uevent_param->u.ftrace.symbol_name));
1303 break;
1304 default:
1305 break;
1306 }
1307 ret = lttng_abi_create_event(file, uevent_param);
1308
1309 old_event_error_free_old_param:
1310 kfree(old_uevent_param);
1311 old_event_error_free_param:
1312 kfree(uevent_param);
1313 old_event_end:
1314 return ret;
1315 }
1316 case LTTNG_KERNEL_EVENT:
1317 {
1318 struct lttng_kernel_event uevent_param;
1319
1320 if (copy_from_user(&uevent_param,
1321 (struct lttng_kernel_event __user *) arg,
1322 sizeof(uevent_param)))
1323 return -EFAULT;
1324 return lttng_abi_create_event(file, &uevent_param);
1325 }
1326 case LTTNG_KERNEL_OLD_CONTEXT:
1327 {
1328 struct lttng_kernel_context *ucontext_param;
1329 struct lttng_kernel_old_context *old_ucontext_param;
1330 int ret;
1331
1332 ucontext_param = kmalloc(sizeof(struct lttng_kernel_context),
1333 GFP_KERNEL);
1334 if (!ucontext_param) {
1335 ret = -ENOMEM;
1336 goto old_ctx_end;
1337 }
1338 old_ucontext_param = kmalloc(sizeof(struct lttng_kernel_old_context),
1339 GFP_KERNEL);
1340 if (!old_ucontext_param) {
1341 ret = -ENOMEM;
1342 goto old_ctx_error_free_param;
1343 }
1344
1345 if (copy_from_user(old_ucontext_param,
1346 (struct lttng_kernel_old_context __user *) arg,
1347 sizeof(struct lttng_kernel_old_context))) {
1348 ret = -EFAULT;
1349 goto old_ctx_error_free_old_param;
1350 }
1351 ucontext_param->ctx = old_ucontext_param->ctx;
1352 memcpy(ucontext_param->padding, old_ucontext_param->padding,
1353 sizeof(ucontext_param->padding));
1354 /* only type that uses the union */
1355 if (old_ucontext_param->ctx == LTTNG_KERNEL_CONTEXT_PERF_COUNTER) {
1356 ucontext_param->u.perf_counter.type =
1357 old_ucontext_param->u.perf_counter.type;
1358 ucontext_param->u.perf_counter.config =
1359 old_ucontext_param->u.perf_counter.config;
1360 memcpy(ucontext_param->u.perf_counter.name,
1361 old_ucontext_param->u.perf_counter.name,
1362 sizeof(ucontext_param->u.perf_counter.name));
1363 }
1364
1365 ret = lttng_abi_add_context(file,
1366 ucontext_param,
1367 &channel->ctx, channel->session);
1368
1369 old_ctx_error_free_old_param:
1370 kfree(old_ucontext_param);
1371 old_ctx_error_free_param:
1372 kfree(ucontext_param);
1373 old_ctx_end:
1374 return ret;
1375 }
1376 case LTTNG_KERNEL_CONTEXT:
1377 {
1378 struct lttng_kernel_context ucontext_param;
1379
1380 if (copy_from_user(&ucontext_param,
1381 (struct lttng_kernel_context __user *) arg,
1382 sizeof(ucontext_param)))
1383 return -EFAULT;
1384 return lttng_abi_add_context(file,
1385 &ucontext_param,
1386 &channel->ctx, channel->session);
1387 }
1388 case LTTNG_KERNEL_OLD_ENABLE:
1389 case LTTNG_KERNEL_ENABLE:
1390 return lttng_channel_enable(channel);
1391 case LTTNG_KERNEL_OLD_DISABLE:
1392 case LTTNG_KERNEL_DISABLE:
1393 return lttng_channel_disable(channel);
1394 case LTTNG_KERNEL_SYSCALL_MASK:
1395 return lttng_channel_syscall_mask(channel,
1396 (struct lttng_kernel_syscall_mask __user *) arg);
1397 default:
1398 return -ENOIOCTLCMD;
1399 }
1400 }
1401
1402 /**
1403 * lttng_metadata_ioctl - lttng syscall through ioctl
1404 *
1405 * @file: the file
1406 * @cmd: the command
1407 * @arg: command arg
1408 *
1409 * This ioctl implements lttng commands:
1410 * LTTNG_KERNEL_STREAM
1411 * Returns an event stream file descriptor or failure.
1412 *
1413 * Channel and event file descriptors also hold a reference on the session.
1414 */
1415 static
1416 long lttng_metadata_ioctl(struct file *file, unsigned int cmd, unsigned long arg)
1417 {
1418 switch (cmd) {
1419 case LTTNG_KERNEL_OLD_STREAM:
1420 case LTTNG_KERNEL_STREAM:
1421 return lttng_abi_open_metadata_stream(file);
1422 default:
1423 return -ENOIOCTLCMD;
1424 }
1425 }
1426
1427 /**
1428 * lttng_channel_poll - lttng stream addition/removal monitoring
1429 *
1430 * @file: the file
1431 * @wait: poll table
1432 */
1433 unsigned int lttng_channel_poll(struct file *file, poll_table *wait)
1434 {
1435 struct lttng_channel *channel = file->private_data;
1436 unsigned int mask = 0;
1437
1438 if (file->f_mode & FMODE_READ) {
1439 poll_wait_set_exclusive(wait);
1440 poll_wait(file, channel->ops->get_hp_wait_queue(channel->chan),
1441 wait);
1442
1443 if (channel->ops->is_disabled(channel->chan))
1444 return POLLERR;
1445 if (channel->ops->is_finalized(channel->chan))
1446 return POLLHUP;
1447 if (channel->ops->buffer_has_read_closed_stream(channel->chan))
1448 return POLLIN | POLLRDNORM;
1449 return 0;
1450 }
1451 return mask;
1452
1453 }
1454
1455 static
1456 int lttng_channel_release(struct inode *inode, struct file *file)
1457 {
1458 struct lttng_channel *channel = file->private_data;
1459
1460 if (channel)
1461 fput(channel->session->file);
1462 return 0;
1463 }
1464
1465 static
1466 int lttng_metadata_channel_release(struct inode *inode, struct file *file)
1467 {
1468 struct lttng_channel *channel = file->private_data;
1469
1470 if (channel) {
1471 fput(channel->session->file);
1472 lttng_metadata_channel_destroy(channel);
1473 }
1474
1475 return 0;
1476 }
1477
1478 static const struct file_operations lttng_channel_fops = {
1479 .owner = THIS_MODULE,
1480 .release = lttng_channel_release,
1481 .poll = lttng_channel_poll,
1482 .unlocked_ioctl = lttng_channel_ioctl,
1483 #ifdef CONFIG_COMPAT
1484 .compat_ioctl = lttng_channel_ioctl,
1485 #endif
1486 };
1487
1488 static const struct file_operations lttng_metadata_fops = {
1489 .owner = THIS_MODULE,
1490 .release = lttng_metadata_channel_release,
1491 .unlocked_ioctl = lttng_metadata_ioctl,
1492 #ifdef CONFIG_COMPAT
1493 .compat_ioctl = lttng_metadata_ioctl,
1494 #endif
1495 };
1496
1497 /**
1498 * lttng_event_ioctl - lttng syscall through ioctl
1499 *
1500 * @file: the file
1501 * @cmd: the command
1502 * @arg: command arg
1503 *
1504 * This ioctl implements lttng commands:
1505 * LTTNG_KERNEL_CONTEXT
1506 * Prepend a context field to each record of this event
1507 * LTTNG_KERNEL_ENABLE
1508 * Enable recording for this event (weak enable)
1509 * LTTNG_KERNEL_DISABLE
1510 * Disable recording for this event (strong disable)
1511 */
1512 static
1513 long lttng_event_ioctl(struct file *file, unsigned int cmd, unsigned long arg)
1514 {
1515 struct lttng_event *event;
1516 struct lttng_enabler *enabler;
1517 enum lttng_event_type *evtype = file->private_data;
1518
1519 switch (cmd) {
1520 case LTTNG_KERNEL_OLD_CONTEXT:
1521 {
1522 /* Not implemented */
1523 return -ENOSYS;
1524 }
1525 case LTTNG_KERNEL_CONTEXT:
1526 {
1527 /* Not implemented */
1528 return -ENOSYS;
1529 }
1530 case LTTNG_KERNEL_OLD_ENABLE:
1531 case LTTNG_KERNEL_ENABLE:
1532 switch (*evtype) {
1533 case LTTNG_TYPE_EVENT:
1534 event = file->private_data;
1535 return lttng_event_enable(event);
1536 case LTTNG_TYPE_ENABLER:
1537 enabler = file->private_data;
1538 return lttng_enabler_enable(enabler);
1539 default:
1540 WARN_ON_ONCE(1);
1541 return -ENOSYS;
1542 }
1543 case LTTNG_KERNEL_OLD_DISABLE:
1544 case LTTNG_KERNEL_DISABLE:
1545 switch (*evtype) {
1546 case LTTNG_TYPE_EVENT:
1547 event = file->private_data;
1548 return lttng_event_disable(event);
1549 case LTTNG_TYPE_ENABLER:
1550 enabler = file->private_data;
1551 return lttng_enabler_disable(enabler);
1552 default:
1553 WARN_ON_ONCE(1);
1554 return -ENOSYS;
1555 }
1556 case LTTNG_KERNEL_FILTER:
1557 switch (*evtype) {
1558 case LTTNG_TYPE_EVENT:
1559 return -EINVAL;
1560 case LTTNG_TYPE_ENABLER:
1561 {
1562 enabler = file->private_data;
1563 return lttng_enabler_attach_bytecode(enabler,
1564 (struct lttng_kernel_filter_bytecode __user *) arg);
1565 }
1566 default:
1567 WARN_ON_ONCE(1);
1568 return -ENOSYS;
1569 }
1570 case LTTNG_KERNEL_ADD_CALLSITE:
1571 switch (*evtype) {
1572 case LTTNG_TYPE_EVENT:
1573 event = file->private_data;
1574 return lttng_event_add_callsite(event,
1575 (struct lttng_kernel_event_callsite __user *) arg);
1576 case LTTNG_TYPE_ENABLER:
1577 return -EINVAL;
1578 }
1579 default:
1580 return -ENOIOCTLCMD;
1581 }
1582 }
1583
1584 static
1585 int lttng_event_release(struct inode *inode, struct file *file)
1586 {
1587 struct lttng_event *event;
1588 struct lttng_enabler *enabler;
1589 enum lttng_event_type *evtype = file->private_data;
1590
1591 if (!evtype)
1592 return 0;
1593
1594 switch (*evtype) {
1595 case LTTNG_TYPE_EVENT:
1596 event = file->private_data;
1597 if (event)
1598 fput(event->chan->file);
1599 break;
1600 case LTTNG_TYPE_ENABLER:
1601 enabler = file->private_data;
1602 if (enabler)
1603 fput(enabler->chan->file);
1604 break;
1605 default:
1606 WARN_ON_ONCE(1);
1607 break;
1608 }
1609
1610 return 0;
1611 }
1612
1613 /* TODO: filter control ioctl */
1614 static const struct file_operations lttng_event_fops = {
1615 .owner = THIS_MODULE,
1616 .release = lttng_event_release,
1617 .unlocked_ioctl = lttng_event_ioctl,
1618 #ifdef CONFIG_COMPAT
1619 .compat_ioctl = lttng_event_ioctl,
1620 #endif
1621 };
1622
1623 static int put_u64(uint64_t val, unsigned long arg)
1624 {
1625 return put_user(val, (uint64_t __user *) arg);
1626 }
1627
1628 static long lttng_stream_ring_buffer_ioctl(struct file *filp,
1629 unsigned int cmd, unsigned long arg)
1630 {
1631 struct lib_ring_buffer *buf = filp->private_data;
1632 struct channel *chan = buf->backend.chan;
1633 const struct lib_ring_buffer_config *config = &chan->backend.config;
1634 const struct lttng_channel_ops *ops = chan->backend.priv_ops;
1635 int ret;
1636
1637 if (atomic_read(&chan->record_disabled))
1638 return -EIO;
1639
1640 switch (cmd) {
1641 case LTTNG_RING_BUFFER_GET_TIMESTAMP_BEGIN:
1642 {
1643 uint64_t ts;
1644
1645 ret = ops->timestamp_begin(config, buf, &ts);
1646 if (ret < 0)
1647 goto error;
1648 return put_u64(ts, arg);
1649 }
1650 case LTTNG_RING_BUFFER_GET_TIMESTAMP_END:
1651 {
1652 uint64_t ts;
1653
1654 ret = ops->timestamp_end(config, buf, &ts);
1655 if (ret < 0)
1656 goto error;
1657 return put_u64(ts, arg);
1658 }
1659 case LTTNG_RING_BUFFER_GET_EVENTS_DISCARDED:
1660 {
1661 uint64_t ed;
1662
1663 ret = ops->events_discarded(config, buf, &ed);
1664 if (ret < 0)
1665 goto error;
1666 return put_u64(ed, arg);
1667 }
1668 case LTTNG_RING_BUFFER_GET_CONTENT_SIZE:
1669 {
1670 uint64_t cs;
1671
1672 ret = ops->content_size(config, buf, &cs);
1673 if (ret < 0)
1674 goto error;
1675 return put_u64(cs, arg);
1676 }
1677 case LTTNG_RING_BUFFER_GET_PACKET_SIZE:
1678 {
1679 uint64_t ps;
1680
1681 ret = ops->packet_size(config, buf, &ps);
1682 if (ret < 0)
1683 goto error;
1684 return put_u64(ps, arg);
1685 }
1686 case LTTNG_RING_BUFFER_GET_STREAM_ID:
1687 {
1688 uint64_t si;
1689
1690 ret = ops->stream_id(config, buf, &si);
1691 if (ret < 0)
1692 goto error;
1693 return put_u64(si, arg);
1694 }
1695 case LTTNG_RING_BUFFER_GET_CURRENT_TIMESTAMP:
1696 {
1697 uint64_t ts;
1698
1699 ret = ops->current_timestamp(config, buf, &ts);
1700 if (ret < 0)
1701 goto error;
1702 return put_u64(ts, arg);
1703 }
1704 case LTTNG_RING_BUFFER_GET_SEQ_NUM:
1705 {
1706 uint64_t seq;
1707
1708 ret = ops->sequence_number(config, buf, &seq);
1709 if (ret < 0)
1710 goto error;
1711 return put_u64(seq, arg);
1712 }
1713 case LTTNG_RING_BUFFER_INSTANCE_ID:
1714 {
1715 uint64_t id;
1716
1717 ret = ops->instance_id(config, buf, &id);
1718 if (ret < 0)
1719 goto error;
1720 return put_u64(id, arg);
1721 }
1722 default:
1723 return lib_ring_buffer_file_operations.unlocked_ioctl(filp,
1724 cmd, arg);
1725 }
1726
1727 error:
1728 return -ENOSYS;
1729 }
1730
1731 #ifdef CONFIG_COMPAT
1732 static long lttng_stream_ring_buffer_compat_ioctl(struct file *filp,
1733 unsigned int cmd, unsigned long arg)
1734 {
1735 struct lib_ring_buffer *buf = filp->private_data;
1736 struct channel *chan = buf->backend.chan;
1737 const struct lib_ring_buffer_config *config = &chan->backend.config;
1738 const struct lttng_channel_ops *ops = chan->backend.priv_ops;
1739 int ret;
1740
1741 if (atomic_read(&chan->record_disabled))
1742 return -EIO;
1743
1744 switch (cmd) {
1745 case LTTNG_RING_BUFFER_COMPAT_GET_TIMESTAMP_BEGIN:
1746 {
1747 uint64_t ts;
1748
1749 ret = ops->timestamp_begin(config, buf, &ts);
1750 if (ret < 0)
1751 goto error;
1752 return put_u64(ts, arg);
1753 }
1754 case LTTNG_RING_BUFFER_COMPAT_GET_TIMESTAMP_END:
1755 {
1756 uint64_t ts;
1757
1758 ret = ops->timestamp_end(config, buf, &ts);
1759 if (ret < 0)
1760 goto error;
1761 return put_u64(ts, arg);
1762 }
1763 case LTTNG_RING_BUFFER_COMPAT_GET_EVENTS_DISCARDED:
1764 {
1765 uint64_t ed;
1766
1767 ret = ops->events_discarded(config, buf, &ed);
1768 if (ret < 0)
1769 goto error;
1770 return put_u64(ed, arg);
1771 }
1772 case LTTNG_RING_BUFFER_COMPAT_GET_CONTENT_SIZE:
1773 {
1774 uint64_t cs;
1775
1776 ret = ops->content_size(config, buf, &cs);
1777 if (ret < 0)
1778 goto error;
1779 return put_u64(cs, arg);
1780 }
1781 case LTTNG_RING_BUFFER_COMPAT_GET_PACKET_SIZE:
1782 {
1783 uint64_t ps;
1784
1785 ret = ops->packet_size(config, buf, &ps);
1786 if (ret < 0)
1787 goto error;
1788 return put_u64(ps, arg);
1789 }
1790 case LTTNG_RING_BUFFER_COMPAT_GET_STREAM_ID:
1791 {
1792 uint64_t si;
1793
1794 ret = ops->stream_id(config, buf, &si);
1795 if (ret < 0)
1796 goto error;
1797 return put_u64(si, arg);
1798 }
1799 case LTTNG_RING_BUFFER_GET_CURRENT_TIMESTAMP:
1800 {
1801 uint64_t ts;
1802
1803 ret = ops->current_timestamp(config, buf, &ts);
1804 if (ret < 0)
1805 goto error;
1806 return put_u64(ts, arg);
1807 }
1808 case LTTNG_RING_BUFFER_COMPAT_GET_SEQ_NUM:
1809 {
1810 uint64_t seq;
1811
1812 ret = ops->sequence_number(config, buf, &seq);
1813 if (ret < 0)
1814 goto error;
1815 return put_u64(seq, arg);
1816 }
1817 case LTTNG_RING_BUFFER_COMPAT_INSTANCE_ID:
1818 {
1819 uint64_t id;
1820
1821 ret = ops->instance_id(config, buf, &id);
1822 if (ret < 0)
1823 goto error;
1824 return put_u64(id, arg);
1825 }
1826 default:
1827 return lib_ring_buffer_file_operations.compat_ioctl(filp,
1828 cmd, arg);
1829 }
1830
1831 error:
1832 return -ENOSYS;
1833 }
1834 #endif /* CONFIG_COMPAT */
1835
1836 static void lttng_stream_override_ring_buffer_fops(void)
1837 {
1838 lttng_stream_ring_buffer_file_operations.owner = THIS_MODULE;
1839 lttng_stream_ring_buffer_file_operations.open =
1840 lib_ring_buffer_file_operations.open;
1841 lttng_stream_ring_buffer_file_operations.release =
1842 lib_ring_buffer_file_operations.release;
1843 lttng_stream_ring_buffer_file_operations.poll =
1844 lib_ring_buffer_file_operations.poll;
1845 lttng_stream_ring_buffer_file_operations.splice_read =
1846 lib_ring_buffer_file_operations.splice_read;
1847 lttng_stream_ring_buffer_file_operations.mmap =
1848 lib_ring_buffer_file_operations.mmap;
1849 lttng_stream_ring_buffer_file_operations.unlocked_ioctl =
1850 lttng_stream_ring_buffer_ioctl;
1851 lttng_stream_ring_buffer_file_operations.llseek =
1852 lib_ring_buffer_file_operations.llseek;
1853 #ifdef CONFIG_COMPAT
1854 lttng_stream_ring_buffer_file_operations.compat_ioctl =
1855 lttng_stream_ring_buffer_compat_ioctl;
1856 #endif
1857 }
1858
1859 int __init lttng_abi_init(void)
1860 {
1861 int ret = 0;
1862
1863 wrapper_vmalloc_sync_all();
1864 lttng_clock_ref();
1865
1866 ret = lttng_tp_mempool_init();
1867 if (ret) {
1868 goto error;
1869 }
1870
1871 lttng_proc_dentry = proc_create_data("lttng", S_IRUSR | S_IWUSR, NULL,
1872 &lttng_fops, NULL);
1873
1874 if (!lttng_proc_dentry) {
1875 printk(KERN_ERR "Error creating LTTng control file\n");
1876 ret = -ENOMEM;
1877 goto error;
1878 }
1879 lttng_stream_override_ring_buffer_fops();
1880 return 0;
1881
1882 error:
1883 lttng_tp_mempool_destroy();
1884 lttng_clock_unref();
1885 return ret;
1886 }
1887
1888 /* No __exit annotation because used by init error path too. */
1889 void lttng_abi_exit(void)
1890 {
1891 lttng_tp_mempool_destroy();
1892 lttng_clock_unref();
1893 if (lttng_proc_dentry)
1894 remove_proc_entry("lttng", NULL);
1895 }
This page took 0.12311 seconds and 5 git commands to generate.