Commit | Line | Data |
---|---|---|
a58c490f | 1 | /* |
ab5be9fa | 2 | * Copyright (C) 2017 Jérémie Galarneau <jeremie.galarneau@efficios.com> |
a58c490f | 3 | * |
ab5be9fa | 4 | * SPDX-License-Identifier: LGPL-2.1-only |
a58c490f | 5 | * |
a58c490f JG |
6 | */ |
7 | ||
8 | #include <lttng/trigger/trigger-internal.h> | |
9 | #include <lttng/condition/condition-internal.h> | |
10 | #include <lttng/action/action-internal.h> | |
3da864a9 | 11 | #include <common/credentials.h> |
9e620ea7 JG |
12 | #include <common/payload.h> |
13 | #include <common/payload-view.h> | |
a58c490f | 14 | #include <common/error.h> |
3da864a9 | 15 | #include <common/optional.h> |
a58c490f JG |
16 | #include <assert.h> |
17 | ||
18 | LTTNG_HIDDEN | |
19 | bool lttng_trigger_validate(struct lttng_trigger *trigger) | |
20 | { | |
21 | bool valid; | |
22 | ||
23 | if (!trigger) { | |
24 | valid = false; | |
25 | goto end; | |
26 | } | |
27 | ||
28 | valid = lttng_condition_validate(trigger->condition) && | |
29 | lttng_action_validate(trigger->action); | |
30 | end: | |
31 | return valid; | |
32 | } | |
33 | ||
34 | struct lttng_trigger *lttng_trigger_create( | |
35 | struct lttng_condition *condition, | |
36 | struct lttng_action *action) | |
37 | { | |
38 | struct lttng_trigger *trigger = NULL; | |
39 | ||
40 | if (!condition || !action) { | |
41 | goto end; | |
42 | } | |
43 | ||
44 | trigger = zmalloc(sizeof(struct lttng_trigger)); | |
45 | if (!trigger) { | |
46 | goto end; | |
47 | } | |
48 | ||
49 | trigger->condition = condition; | |
50 | trigger->action = action; | |
3da864a9 | 51 | |
a58c490f JG |
52 | end: |
53 | return trigger; | |
54 | } | |
55 | ||
56 | struct lttng_condition *lttng_trigger_get_condition( | |
57 | struct lttng_trigger *trigger) | |
58 | { | |
59 | return trigger ? trigger->condition : NULL; | |
60 | } | |
61 | ||
9b63a4aa JG |
62 | LTTNG_HIDDEN |
63 | const struct lttng_condition *lttng_trigger_get_const_condition( | |
64 | const struct lttng_trigger *trigger) | |
65 | { | |
66 | return trigger->condition; | |
67 | } | |
68 | ||
e2ba1c78 | 69 | struct lttng_action *lttng_trigger_get_action( |
a58c490f JG |
70 | struct lttng_trigger *trigger) |
71 | { | |
72 | return trigger ? trigger->action : NULL; | |
73 | } | |
74 | ||
9b63a4aa JG |
75 | LTTNG_HIDDEN |
76 | const struct lttng_action *lttng_trigger_get_const_action( | |
77 | const struct lttng_trigger *trigger) | |
78 | { | |
79 | return trigger->action; | |
80 | } | |
81 | ||
a58c490f JG |
82 | void lttng_trigger_destroy(struct lttng_trigger *trigger) |
83 | { | |
84 | if (!trigger) { | |
85 | return; | |
86 | } | |
87 | ||
88 | free(trigger); | |
89 | } | |
90 | ||
91 | LTTNG_HIDDEN | |
c0a66c84 JG |
92 | ssize_t lttng_trigger_create_from_payload( |
93 | struct lttng_payload_view *src_view, | |
a58c490f JG |
94 | struct lttng_trigger **trigger) |
95 | { | |
96 | ssize_t ret, offset = 0, condition_size, action_size; | |
97 | struct lttng_condition *condition = NULL; | |
98 | struct lttng_action *action = NULL; | |
99 | const struct lttng_trigger_comm *trigger_comm; | |
a58c490f JG |
100 | |
101 | if (!src_view || !trigger) { | |
102 | ret = -1; | |
103 | goto end; | |
104 | } | |
105 | ||
106 | /* lttng_trigger_comm header */ | |
c0a66c84 | 107 | trigger_comm = (typeof(trigger_comm)) src_view->buffer.data; |
a58c490f | 108 | offset += sizeof(*trigger_comm); |
c0a66c84 JG |
109 | { |
110 | /* struct lttng_condition */ | |
111 | struct lttng_payload_view condition_view = | |
112 | lttng_payload_view_from_view( | |
113 | src_view, offset, -1); | |
114 | ||
115 | condition_size = lttng_condition_create_from_payload(&condition_view, | |
116 | &condition); | |
117 | } | |
a58c490f | 118 | |
a58c490f JG |
119 | if (condition_size < 0) { |
120 | ret = condition_size; | |
121 | goto end; | |
122 | } | |
c0a66c84 | 123 | |
a58c490f | 124 | offset += condition_size; |
c0a66c84 JG |
125 | { |
126 | /* struct lttng_action */ | |
127 | struct lttng_payload_view action_view = | |
128 | lttng_payload_view_from_view( | |
129 | src_view, offset, -1); | |
130 | ||
131 | action_size = lttng_action_create_from_payload(&action_view, &action); | |
132 | } | |
a58c490f | 133 | |
a58c490f JG |
134 | if (action_size < 0) { |
135 | ret = action_size; | |
136 | goto end; | |
137 | } | |
138 | offset += action_size; | |
139 | ||
140 | /* Unexpected size of inner-elements; the buffer is corrupted. */ | |
141 | if ((ssize_t) trigger_comm->length != condition_size + action_size) { | |
142 | ret = -1; | |
143 | goto error; | |
144 | } | |
145 | ||
146 | *trigger = lttng_trigger_create(condition, action); | |
147 | if (!*trigger) { | |
148 | ret = -1; | |
149 | goto error; | |
150 | } | |
c0a66c84 | 151 | |
a58c490f JG |
152 | ret = offset; |
153 | end: | |
154 | return ret; | |
155 | error: | |
156 | lttng_condition_destroy(condition); | |
157 | lttng_action_destroy(action); | |
158 | return ret; | |
159 | } | |
160 | ||
161 | /* | |
a58c490f JG |
162 | * Both elements are stored contiguously, see their "*_comm" structure |
163 | * for the detailed format. | |
164 | */ | |
165 | LTTNG_HIDDEN | |
3647288f | 166 | int lttng_trigger_serialize(struct lttng_trigger *trigger, |
c0a66c84 | 167 | struct lttng_payload *payload) |
a58c490f | 168 | { |
3647288f JG |
169 | int ret; |
170 | size_t header_offset, size_before_payload; | |
c0a66c84 | 171 | struct lttng_trigger_comm trigger_comm = {}; |
3647288f | 172 | struct lttng_trigger_comm *header; |
a58c490f | 173 | |
c0a66c84 JG |
174 | header_offset = payload->buffer.size; |
175 | ret = lttng_dynamic_buffer_append(&payload->buffer, &trigger_comm, | |
3647288f JG |
176 | sizeof(trigger_comm)); |
177 | if (ret) { | |
a58c490f JG |
178 | goto end; |
179 | } | |
180 | ||
c0a66c84 JG |
181 | size_before_payload = payload->buffer.size; |
182 | ret = lttng_condition_serialize(trigger->condition, payload); | |
3647288f | 183 | if (ret) { |
a58c490f JG |
184 | goto end; |
185 | } | |
a58c490f | 186 | |
c0a66c84 | 187 | ret = lttng_action_serialize(trigger->action, payload); |
3647288f | 188 | if (ret) { |
a58c490f JG |
189 | goto end; |
190 | } | |
a58c490f | 191 | |
3647288f | 192 | /* Update payload size. */ |
c0a66c84 JG |
193 | header = (typeof(header)) (payload->buffer.data + header_offset); |
194 | header->length = payload->buffer.size - size_before_payload; | |
a58c490f JG |
195 | end: |
196 | return ret; | |
197 | } | |
3da864a9 JR |
198 | |
199 | LTTNG_HIDDEN | |
200 | const struct lttng_credentials *lttng_trigger_get_credentials( | |
201 | const struct lttng_trigger *trigger) | |
202 | { | |
203 | return LTTNG_OPTIONAL_GET_PTR(trigger->creds); | |
204 | } | |
205 | ||
206 | LTTNG_HIDDEN | |
207 | void lttng_trigger_set_credentials( | |
208 | struct lttng_trigger *trigger, | |
209 | const struct lttng_credentials *creds) | |
210 | { | |
211 | assert(creds); | |
212 | LTTNG_OPTIONAL_SET(&trigger->creds, *creds); | |
213 | } |