Commit | Line | Data |
---|---|---|
b7cdc182 | 1 | /* SPDX-License-Identifier: (GPL-2.0-only or LGPL-2.1-only) |
9f36eaed | 2 | * |
79150a49 JD |
3 | * lttng-context-need-reschedule.c |
4 | * | |
5 | * LTTng need_reschedule context. | |
6 | * | |
7 | * Copyright (C) 2009-2015 Mathieu Desnoyers <mathieu.desnoyers@efficios.com> | |
79150a49 JD |
8 | */ |
9 | ||
10 | #include <linux/module.h> | |
11 | #include <linux/slab.h> | |
12 | #include <linux/sched.h> | |
13 | #include <linux/irqflags.h> | |
2df37e95 | 14 | #include <lttng/events.h> |
437d5aa5 | 15 | #include <lttng/events-internal.h> |
24591303 | 16 | #include <ringbuffer/frontend_types.h> |
241ae9a8 | 17 | #include <wrapper/vmalloc.h> |
2df37e95 | 18 | #include <lttng/tracer.h> |
79150a49 JD |
19 | |
20 | static | |
21 | size_t need_reschedule_get_size(size_t offset) | |
22 | { | |
23 | size_t size = 0; | |
24 | ||
25 | size += lib_ring_buffer_align(offset, lttng_alignof(uint8_t)); | |
26 | size += sizeof(uint8_t); | |
27 | return size; | |
28 | } | |
29 | ||
30 | static | |
437d5aa5 | 31 | void need_reschedule_record(struct lttng_kernel_ctx_field *field, |
79150a49 JD |
32 | struct lib_ring_buffer_ctx *ctx, |
33 | struct lttng_channel *chan) | |
34 | { | |
35 | uint8_t need_reschedule = test_tsk_need_resched(current); | |
36 | ||
37 | lib_ring_buffer_align_ctx(ctx, lttng_alignof(need_reschedule)); | |
38 | chan->ops->event_write(ctx, &need_reschedule, sizeof(need_reschedule)); | |
39 | } | |
40 | ||
41 | static | |
437d5aa5 | 42 | void need_reschedule_get_value(struct lttng_kernel_ctx_field *field, |
79150a49 JD |
43 | struct lttng_probe_ctx *lttng_probe_ctx, |
44 | union lttng_ctx_value *value) | |
45 | { | |
46 | value->s64 = test_tsk_need_resched(current);; | |
47 | } | |
48 | ||
437d5aa5 MD |
49 | static const struct lttng_kernel_ctx_field *ctx_field = lttng_kernel_static_ctx_field( |
50 | lttng_kernel_static_event_field("need_reschedule", | |
51 | lttng_kernel_static_type_integer_from_type(uint8_t, __BYTE_ORDER, 10), | |
52 | false, false, false), | |
53 | need_reschedule_get_size, | |
54 | NULL, | |
55 | need_reschedule_record, | |
56 | need_reschedule_get_value, | |
57 | NULL, NULL); | |
58 | ||
59 | int lttng_add_need_reschedule_to_ctx(struct lttng_kernel_ctx **ctx) | |
79150a49 | 60 | { |
437d5aa5 | 61 | int ret; |
79150a49 | 62 | |
437d5aa5 | 63 | if (lttng_kernel_find_context(*ctx, ctx_field->event_field->name)) |
79150a49 | 64 | return -EEXIST; |
437d5aa5 | 65 | ret = lttng_kernel_context_append(ctx, ctx_field); |
263b6c88 | 66 | wrapper_vmalloc_sync_mappings(); |
437d5aa5 | 67 | return ret; |
79150a49 JD |
68 | } |
69 | EXPORT_SYMBOL_GPL(lttng_add_need_reschedule_to_ctx); |