Commit | Line | Data |
---|---|---|
b7cdc182 | 1 | /* SPDX-License-Identifier: (GPL-2.0-only or LGPL-2.1-only) |
9f36eaed | 2 | * |
886d51a3 | 3 | * lttng-context-nice.c |
53f1f0ca MD |
4 | * |
5 | * LTTng nice context. | |
6 | * | |
886d51a3 | 7 | * Copyright (C) 2009-2012 Mathieu Desnoyers <mathieu.desnoyers@efficios.com> |
53f1f0ca MD |
8 | */ |
9 | ||
10 | #include <linux/module.h> | |
11 | #include <linux/slab.h> | |
12 | #include <linux/sched.h> | |
2df37e95 | 13 | #include <lttng/events.h> |
437d5aa5 | 14 | #include <lttng/events-internal.h> |
24591303 | 15 | #include <ringbuffer/frontend_types.h> |
241ae9a8 | 16 | #include <wrapper/vmalloc.h> |
2df37e95 | 17 | #include <lttng/tracer.h> |
53f1f0ca MD |
18 | |
19 | static | |
a92e844e | 20 | size_t nice_get_size(void *priv, struct lttng_kernel_probe_ctx *probe_ctx, size_t offset) |
53f1f0ca MD |
21 | { |
22 | size_t size = 0; | |
23 | ||
a90917c3 | 24 | size += lib_ring_buffer_align(offset, lttng_alignof(int)); |
53f1f0ca MD |
25 | size += sizeof(int); |
26 | return size; | |
27 | } | |
28 | ||
29 | static | |
a92e844e | 30 | void nice_record(void *priv, struct lttng_kernel_probe_ctx *probe_ctx, |
8a57ec02 | 31 | struct lttng_kernel_ring_buffer_ctx *ctx, |
f7d06400 | 32 | struct lttng_kernel_channel_buffer *chan) |
53f1f0ca MD |
33 | { |
34 | int nice; | |
35 | ||
36 | nice = task_nice(current); | |
f5ffbd77 | 37 | chan->ops->event_write(ctx, &nice, sizeof(nice), lttng_alignof(nice)); |
53f1f0ca MD |
38 | } |
39 | ||
f127e61e | 40 | static |
2dc781e0 | 41 | void nice_get_value(void *priv, |
a92e844e | 42 | struct lttng_kernel_probe_ctx *lttng_probe_ctx, |
2dc781e0 | 43 | struct lttng_ctx_value *value) |
f127e61e | 44 | { |
2dc781e0 | 45 | value->u.s64 = task_nice(current); |
f127e61e MD |
46 | } |
47 | ||
437d5aa5 MD |
48 | static const struct lttng_kernel_ctx_field *ctx_field = lttng_kernel_static_ctx_field( |
49 | lttng_kernel_static_event_field("nice", | |
50 | lttng_kernel_static_type_integer_from_type(int, __BYTE_ORDER, 10), | |
51 | false, false, false), | |
52 | nice_get_size, | |
437d5aa5 MD |
53 | nice_record, |
54 | nice_get_value, | |
55 | NULL, NULL); | |
56 | ||
57 | int lttng_add_nice_to_ctx(struct lttng_kernel_ctx **ctx) | |
53f1f0ca | 58 | { |
437d5aa5 | 59 | int ret; |
53f1f0ca | 60 | |
437d5aa5 | 61 | if (lttng_kernel_find_context(*ctx, ctx_field->event_field->name)) |
44252f0f | 62 | return -EEXIST; |
437d5aa5 | 63 | ret = lttng_kernel_context_append(ctx, ctx_field); |
263b6c88 | 64 | wrapper_vmalloc_sync_mappings(); |
437d5aa5 | 65 | return ret; |
53f1f0ca MD |
66 | } |
67 | EXPORT_SYMBOL_GPL(lttng_add_nice_to_ctx); |