Commit | Line | Data |
---|---|---|
bc37ae52 | 1 | /* |
bc37ae52 JG |
2 | * Copyright 2014 Jérémie Galarneau <jeremie.galarneau@efficios.com> |
3 | * | |
4 | * Author: Jérémie Galarneau <jeremie.galarneau@efficios.com> | |
5 | * | |
6 | * Permission is hereby granted, free of charge, to any person obtaining a copy | |
7 | * of this software and associated documentation files (the "Software"), to deal | |
8 | * in the Software without restriction, including without limitation the rights | |
9 | * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |
10 | * copies of the Software, and to permit persons to whom the Software is | |
11 | * furnished to do so, subject to the following conditions: | |
12 | * | |
13 | * The above copyright notice and this permission notice shall be included in | |
14 | * all copies or substantial portions of the Software. | |
15 | * | |
16 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
17 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
18 | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |
19 | * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |
20 | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
21 | * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |
22 | * SOFTWARE. | |
23 | */ | |
24 | ||
e011d2c1 PP |
25 | #define BT_LOG_TAG "TRACE" |
26 | #include <babeltrace/lib-logging-internal.h> | |
27 | ||
8deee039 | 28 | #include <babeltrace/assert-pre-internal.h> |
9e550e5f | 29 | #include <babeltrace/trace-ir/private-trace.h> |
108b91d0 PP |
30 | #include <babeltrace/trace-ir/trace-internal.h> |
31 | #include <babeltrace/trace-ir/clock-class-internal.h> | |
32 | #include <babeltrace/trace-ir/stream-internal.h> | |
33 | #include <babeltrace/trace-ir/stream-class-internal.h> | |
34 | #include <babeltrace/trace-ir/event-internal.h> | |
35 | #include <babeltrace/trace-ir/event-class.h> | |
36 | #include <babeltrace/trace-ir/event-class-internal.h> | |
bc37ae52 | 37 | #include <babeltrace/ctf-writer/functor-internal.h> |
ac0c6bdd | 38 | #include <babeltrace/ctf-writer/clock-internal.h> |
108b91d0 | 39 | #include <babeltrace/trace-ir/field-wrapper-internal.h> |
939190b3 | 40 | #include <babeltrace/trace-ir/field-classes-internal.h> |
108b91d0 PP |
41 | #include <babeltrace/trace-ir/attributes-internal.h> |
42 | #include <babeltrace/trace-ir/utils-internal.h> | |
43 | #include <babeltrace/trace-ir/resolve-field-path-internal.h> | |
3d9990ac | 44 | #include <babeltrace/compiler-internal.h> |
dac5c838 | 45 | #include <babeltrace/values.h> |
17582c6d | 46 | #include <babeltrace/private-values.h> |
95c09b3a | 47 | #include <babeltrace/values-internal.h> |
8138bfe1 | 48 | #include <babeltrace/object.h> |
c55a9f58 | 49 | #include <babeltrace/types.h> |
3d9990ac | 50 | #include <babeltrace/endian-internal.h> |
8b45963b | 51 | #include <babeltrace/assert-internal.h> |
7b33a0e0 | 52 | #include <babeltrace/compat/glib-internal.h> |
dc3fffef | 53 | #include <inttypes.h> |
544d0515 | 54 | #include <stdint.h> |
4a32fda0 | 55 | #include <string.h> |
0fbb9a9f | 56 | #include <stdlib.h> |
bc37ae52 | 57 | |
839d52a5 | 58 | struct bt_trace_is_static_listener_elem { |
9e550e5f PP |
59 | bt_private_trace_is_static_listener func; |
60 | bt_private_trace_listener_removed removed; | |
3602afb0 PP |
61 | void *data; |
62 | }; | |
63 | ||
7b33a0e0 PP |
64 | #define BT_ASSERT_PRE_TRACE_HOT(_trace) \ |
65 | BT_ASSERT_PRE_HOT((_trace), "Trace", ": %!+t", (_trace)) | |
18acc6f8 | 66 | |
bc37ae52 | 67 | static |
7b33a0e0 | 68 | void destroy_trace(struct bt_object *obj) |
8deee039 PP |
69 | { |
70 | struct bt_trace *trace = (void *) obj; | |
bc37ae52 | 71 | |
7b33a0e0 | 72 | BT_LIB_LOGD("Destroying trace object: %!+t", trace); |
bc37ae52 | 73 | |
8deee039 PP |
74 | /* |
75 | * Call remove listeners first so that everything else still | |
76 | * exists in the trace. | |
77 | */ | |
78 | if (trace->is_static_listeners) { | |
79 | size_t i; | |
bc37ae52 | 80 | |
8deee039 PP |
81 | for (i = 0; i < trace->is_static_listeners->len; i++) { |
82 | struct bt_trace_is_static_listener_elem elem = | |
83 | g_array_index(trace->is_static_listeners, | |
84 | struct bt_trace_is_static_listener_elem, i); | |
bc37ae52 | 85 | |
8deee039 | 86 | if (elem.removed) { |
9e550e5f | 87 | elem.removed((void *) trace, elem.data); |
8deee039 PP |
88 | } |
89 | } | |
90 | ||
91 | g_array_free(trace->is_static_listeners, TRUE); | |
1248f5ea | 92 | trace->is_static_listeners = NULL; |
bc37ae52 JG |
93 | } |
94 | ||
a6918753 | 95 | bt_object_pool_finalize(&trace->packet_header_field_pool); |
8deee039 | 96 | |
7b33a0e0 PP |
97 | if (trace->environment) { |
98 | BT_LOGD_STR("Destroying environment attributes."); | |
99 | bt_attributes_destroy(trace->environment); | |
1248f5ea | 100 | trace->environment = NULL; |
95c09b3a PP |
101 | } |
102 | ||
7b33a0e0 PP |
103 | if (trace->name.str) { |
104 | g_string_free(trace->name.str, TRUE); | |
1248f5ea PP |
105 | trace->name.str = NULL; |
106 | trace->name.value = NULL; | |
95c09b3a PP |
107 | } |
108 | ||
7b33a0e0 PP |
109 | if (trace->streams) { |
110 | BT_LOGD_STR("Destroying streams."); | |
111 | g_ptr_array_free(trace->streams, TRUE); | |
1248f5ea | 112 | trace->streams = NULL; |
bc37ae52 JG |
113 | } |
114 | ||
7b33a0e0 PP |
115 | if (trace->stream_classes) { |
116 | BT_LOGD_STR("Destroying stream classes."); | |
117 | g_ptr_array_free(trace->stream_classes, TRUE); | |
1248f5ea | 118 | trace->stream_classes = NULL; |
7f800dc7 PP |
119 | } |
120 | ||
7b33a0e0 PP |
121 | if (trace->stream_classes_stream_count) { |
122 | g_hash_table_destroy(trace->stream_classes_stream_count); | |
1248f5ea | 123 | trace->stream_classes_stream_count = NULL; |
7b33a0e0 | 124 | } |
8deee039 | 125 | |
939190b3 | 126 | BT_LOGD_STR("Putting packet header field classe."); |
8138bfe1 | 127 | bt_object_put_ref(trace->packet_header_fc); |
1248f5ea | 128 | trace->packet_header_fc = NULL; |
7b33a0e0 | 129 | g_free(trace); |
8deee039 PP |
130 | } |
131 | ||
a6918753 PP |
132 | static |
133 | void free_packet_header_field(struct bt_field_wrapper *field_wrapper, | |
134 | struct bt_trace *trace) | |
135 | { | |
136 | bt_field_wrapper_destroy(field_wrapper); | |
137 | } | |
138 | ||
9e550e5f | 139 | struct bt_private_trace *bt_private_trace_create(void) |
8deee039 PP |
140 | { |
141 | struct bt_trace *trace = NULL; | |
142 | int ret; | |
143 | ||
7b33a0e0 | 144 | BT_LOGD_STR("Creating default trace object."); |
8deee039 PP |
145 | trace = g_new0(struct bt_trace, 1); |
146 | if (!trace) { | |
147 | BT_LOGE_STR("Failed to allocate one trace."); | |
148 | goto error; | |
149 | } | |
150 | ||
7b33a0e0 PP |
151 | bt_object_init_shared_with_parent(&trace->base, destroy_trace); |
152 | trace->streams = g_ptr_array_new_with_free_func( | |
153 | (GDestroyNotify) bt_object_try_spec_release); | |
154 | if (!trace->streams) { | |
155 | BT_LOGE_STR("Failed to allocate one GPtrArray."); | |
8deee039 PP |
156 | goto error; |
157 | } | |
158 | ||
7b33a0e0 PP |
159 | trace->stream_classes = g_ptr_array_new_with_free_func( |
160 | (GDestroyNotify) bt_object_try_spec_release); | |
161 | if (!trace->stream_classes) { | |
95c09b3a | 162 | BT_LOGE_STR("Failed to allocate one GPtrArray."); |
9b888ff3 JG |
163 | goto error; |
164 | } | |
165 | ||
7b33a0e0 PP |
166 | trace->stream_classes_stream_count = g_hash_table_new(g_direct_hash, |
167 | g_direct_equal); | |
168 | if (!trace->stream_classes_stream_count) { | |
169 | BT_LOGE_STR("Failed to allocate one GHashTable."); | |
170 | goto error; | |
171 | } | |
172 | ||
173 | trace->name.str = g_string_new(NULL); | |
174 | if (!trace->name.str) { | |
175 | BT_LOGE_STR("Failed to allocate one GString."); | |
176 | goto error; | |
177 | } | |
178 | ||
179 | trace->environment = bt_attributes_create(); | |
180 | if (!trace->environment) { | |
181 | BT_LOGE_STR("Cannot create empty attributes object."); | |
182 | goto error; | |
183 | } | |
184 | ||
3602afb0 | 185 | trace->is_static_listeners = g_array_new(FALSE, TRUE, |
839d52a5 | 186 | sizeof(struct bt_trace_is_static_listener_elem)); |
3602afb0 PP |
187 | if (!trace->is_static_listeners) { |
188 | BT_LOGE_STR("Failed to allocate one GArray."); | |
189 | goto error; | |
190 | } | |
191 | ||
7b33a0e0 | 192 | trace->assigns_automatic_stream_class_id = true; |
a6918753 PP |
193 | ret = bt_object_pool_initialize(&trace->packet_header_field_pool, |
194 | (bt_object_pool_new_object_func) bt_field_wrapper_new, | |
195 | (bt_object_pool_destroy_object_func) free_packet_header_field, | |
196 | trace); | |
197 | if (ret) { | |
198 | BT_LOGE("Failed to initialize packet header field pool: ret=%d", | |
199 | ret); | |
200 | goto error; | |
201 | } | |
202 | ||
7b33a0e0 PP |
203 | BT_LIB_LOGD("Created trace object: %!+t", trace); |
204 | goto end; | |
bc37ae52 | 205 | |
bc37ae52 | 206 | error: |
8138bfe1 | 207 | BT_OBJECT_PUT_REF_AND_RESET(trace); |
7b33a0e0 PP |
208 | |
209 | end: | |
9e550e5f | 210 | return (void *) trace; |
bc37ae52 JG |
211 | } |
212 | ||
839d52a5 | 213 | const char *bt_trace_get_name(struct bt_trace *trace) |
e96045d4 | 214 | { |
18acc6f8 | 215 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
7b33a0e0 | 216 | return trace->name.value; |
e96045d4 JG |
217 | } |
218 | ||
9e550e5f PP |
219 | int bt_private_trace_set_name(struct bt_private_trace *priv_trace, |
220 | const char *name) | |
e96045d4 | 221 | { |
9e550e5f PP |
222 | struct bt_trace *trace = (void *) priv_trace; |
223 | ||
7b33a0e0 PP |
224 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
225 | BT_ASSERT_PRE_NON_NULL(name, "Name"); | |
226 | BT_ASSERT_PRE_TRACE_HOT(trace); | |
227 | g_string_assign(trace->name.str, name); | |
228 | trace->name.value = trace->name.str->str; | |
229 | BT_LIB_LOGV("Set trace's name: %!+t", trace); | |
230 | return 0; | |
e96045d4 JG |
231 | } |
232 | ||
7b33a0e0 | 233 | bt_uuid bt_trace_get_uuid(struct bt_trace *trace) |
4a32fda0 | 234 | { |
18acc6f8 | 235 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
7b33a0e0 | 236 | return trace->uuid.value; |
4a32fda0 PP |
237 | } |
238 | ||
c8bbf821 PP |
239 | void bt_private_trace_set_uuid(struct bt_private_trace *priv_trace, |
240 | bt_uuid uuid) | |
4a32fda0 | 241 | { |
9e550e5f PP |
242 | struct bt_trace *trace = (void *) priv_trace; |
243 | ||
7b33a0e0 PP |
244 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
245 | BT_ASSERT_PRE_NON_NULL(uuid, "UUID"); | |
246 | BT_ASSERT_PRE_TRACE_HOT(trace); | |
247 | memcpy(trace->uuid.uuid, uuid, BABELTRACE_UUID_LEN); | |
248 | trace->uuid.value = trace->uuid.uuid; | |
249 | BT_LIB_LOGV("Set trace's UUID: %!+t", trace); | |
4a32fda0 PP |
250 | } |
251 | ||
7b33a0e0 PP |
252 | BT_ASSERT_FUNC |
253 | static | |
254 | bool trace_has_environment_entry(struct bt_trace *trace, const char *name) | |
bc37ae52 | 255 | { |
7b33a0e0 | 256 | BT_ASSERT(trace); |
95c09b3a | 257 | |
17582c6d PP |
258 | return bt_attributes_borrow_field_value_by_name( |
259 | trace->environment, name) != NULL; | |
7b33a0e0 | 260 | } |
a0d12916 | 261 | |
7b33a0e0 PP |
262 | static |
263 | int set_environment_entry(struct bt_trace *trace, const char *name, | |
17582c6d | 264 | struct bt_private_value *value) |
7b33a0e0 PP |
265 | { |
266 | int ret; | |
a0d12916 | 267 | |
7b33a0e0 PP |
268 | BT_ASSERT(trace); |
269 | BT_ASSERT(name); | |
270 | BT_ASSERT(value); | |
271 | BT_ASSERT_PRE(!trace->frozen || | |
272 | !trace_has_environment_entry(trace, name), | |
273 | "Trace is frozen: cannot replace environment entry: " | |
274 | "%![trace-]+t, entry-name=\"%s\"", trace, name); | |
839d52a5 | 275 | ret = bt_attributes_set_field_value(trace->environment, name, |
7f800dc7 | 276 | value); |
bb61965b | 277 | bt_value_freeze(bt_private_value_as_value(value)); |
95c09b3a | 278 | if (ret) { |
7b33a0e0 PP |
279 | BT_LIB_LOGE("Cannot set trace's environment entry: " |
280 | "%![trace-]+t, entry-name=\"%s\"", trace, name); | |
95c09b3a | 281 | } else { |
7b33a0e0 PP |
282 | BT_LIB_LOGV("Set trace's environment entry: " |
283 | "%![trace-]+t, entry-name=\"%s\"", trace, name); | |
95c09b3a | 284 | } |
bc37ae52 | 285 | |
7f800dc7 PP |
286 | return ret; |
287 | } | |
bc37ae52 | 288 | |
96854e6a | 289 | int bt_private_trace_set_environment_entry_string( |
9e550e5f | 290 | struct bt_private_trace *priv_trace, |
7f800dc7 PP |
291 | const char *name, const char *value) |
292 | { | |
7b33a0e0 | 293 | int ret; |
17582c6d | 294 | struct bt_private_value *value_obj; |
9e550e5f | 295 | struct bt_trace *trace = (void *) priv_trace; |
bc37ae52 | 296 | |
7b33a0e0 PP |
297 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
298 | BT_ASSERT_PRE_NON_NULL(name, "Name"); | |
299 | BT_ASSERT_PRE_NON_NULL(value, "Value"); | |
17582c6d | 300 | value_obj = bt_private_value_string_create_init(value); |
7b33a0e0 PP |
301 | if (!value_obj) { |
302 | BT_LOGE_STR("Cannot create a string value object."); | |
7f800dc7 PP |
303 | ret = -1; |
304 | goto end; | |
bc37ae52 JG |
305 | } |
306 | ||
7b33a0e0 PP |
307 | /* set_environment_entry() logs errors */ |
308 | ret = set_environment_entry(trace, name, value_obj); | |
7f800dc7 PP |
309 | |
310 | end: | |
8138bfe1 | 311 | bt_object_put_ref(value_obj); |
3487c9f3 JG |
312 | return ret; |
313 | } | |
314 | ||
96854e6a | 315 | int bt_private_trace_set_environment_entry_integer( |
9e550e5f PP |
316 | struct bt_private_trace *priv_trace, |
317 | const char *name, int64_t value) | |
3487c9f3 | 318 | { |
7b33a0e0 | 319 | int ret; |
17582c6d | 320 | struct bt_private_value *value_obj; |
9e550e5f | 321 | struct bt_trace *trace = (void *) priv_trace; |
3487c9f3 | 322 | |
7b33a0e0 PP |
323 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
324 | BT_ASSERT_PRE_NON_NULL(name, "Name"); | |
17582c6d | 325 | value_obj = bt_private_value_integer_create_init(value); |
7b33a0e0 PP |
326 | if (!value_obj) { |
327 | BT_LOGE_STR("Cannot create an integer value object."); | |
3487c9f3 | 328 | ret = -1; |
7f800dc7 | 329 | goto end; |
3487c9f3 JG |
330 | } |
331 | ||
7b33a0e0 PP |
332 | /* set_environment_entry() logs errors */ |
333 | ret = set_environment_entry(trace, name, value_obj); | |
95c09b3a | 334 | |
7f800dc7 | 335 | end: |
8138bfe1 | 336 | bt_object_put_ref(value_obj); |
bc37ae52 JG |
337 | return ret; |
338 | } | |
339 | ||
7b33a0e0 | 340 | uint64_t bt_trace_get_environment_entry_count(struct bt_trace *trace) |
8deee039 | 341 | { |
18acc6f8 PP |
342 | int64_t ret; |
343 | ||
344 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); | |
345 | ret = bt_attributes_get_count(trace->environment); | |
346 | BT_ASSERT(ret >= 0); | |
7b33a0e0 | 347 | return (uint64_t) ret; |
e6fa2160 JG |
348 | } |
349 | ||
7b33a0e0 PP |
350 | void bt_trace_borrow_environment_entry_by_index( |
351 | struct bt_trace *trace, uint64_t index, | |
352 | const char **name, struct bt_value **value) | |
e6fa2160 | 353 | { |
18acc6f8 | 354 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
7b33a0e0 PP |
355 | BT_ASSERT_PRE_NON_NULL(name, "Name"); |
356 | BT_ASSERT_PRE_NON_NULL(value, "Value"); | |
357 | BT_ASSERT_PRE_VALID_INDEX(index, | |
358 | bt_attributes_get_count(trace->environment)); | |
bb61965b | 359 | *value = bt_private_value_as_value( |
17582c6d | 360 | bt_attributes_borrow_field_value(trace->environment, index)); |
7b33a0e0 PP |
361 | BT_ASSERT(*value); |
362 | *name = bt_attributes_get_field_name(trace->environment, index); | |
363 | BT_ASSERT(*name); | |
e6fa2160 JG |
364 | } |
365 | ||
96854e6a | 366 | void bt_private_trace_borrow_environment_entry_by_index( |
9e550e5f PP |
367 | struct bt_private_trace *trace, uint64_t index, |
368 | const char **name, struct bt_private_value **value) | |
369 | { | |
370 | bt_trace_borrow_environment_entry_by_index((void *) trace, | |
371 | index, name, (void *) value); | |
372 | } | |
373 | ||
7b33a0e0 | 374 | struct bt_value *bt_trace_borrow_environment_entry_value_by_name( |
839d52a5 | 375 | struct bt_trace *trace, const char *name) |
e6fa2160 | 376 | { |
18acc6f8 PP |
377 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
378 | BT_ASSERT_PRE_NON_NULL(name, "Name"); | |
bb61965b | 379 | return bt_private_value_as_value( |
17582c6d PP |
380 | bt_attributes_borrow_field_value_by_name(trace->environment, |
381 | name)); | |
e6fa2160 JG |
382 | } |
383 | ||
9e550e5f | 384 | struct bt_private_value * |
96854e6a | 385 | bt_private_trace_borrow_environment_entry_value_by_name( |
9e550e5f PP |
386 | struct bt_private_trace *trace, const char *name) |
387 | { | |
388 | return (void *) bt_trace_borrow_environment_entry_value_by_name( | |
389 | (void *) trace, name); | |
390 | } | |
391 | ||
7b33a0e0 | 392 | uint64_t bt_trace_get_stream_count(struct bt_trace *trace) |
bc37ae52 | 393 | { |
7b33a0e0 PP |
394 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
395 | return (uint64_t) trace->streams->len; | |
396 | } | |
95c09b3a | 397 | |
7b33a0e0 PP |
398 | struct bt_stream *bt_trace_borrow_stream_by_index( |
399 | struct bt_trace *trace, uint64_t index) | |
400 | { | |
401 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); | |
402 | BT_ASSERT_PRE_VALID_INDEX(index, trace->streams->len); | |
403 | return g_ptr_array_index(trace->streams, index); | |
404 | } | |
18acc6f8 | 405 | |
96854e6a | 406 | struct bt_private_stream *bt_private_trace_borrow_stream_by_index( |
9e550e5f PP |
407 | struct bt_private_trace *trace, uint64_t index) |
408 | { | |
409 | return (void *) bt_trace_borrow_stream_by_index((void *) trace, index); | |
410 | } | |
411 | ||
7b33a0e0 PP |
412 | struct bt_stream *bt_trace_borrow_stream_by_id( |
413 | struct bt_trace *trace, uint64_t id) | |
414 | { | |
415 | struct bt_stream *stream = NULL; | |
416 | uint64_t i; | |
bc37ae52 | 417 | |
7b33a0e0 | 418 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
bc37ae52 | 419 | |
7b33a0e0 PP |
420 | for (i = 0; i < trace->streams->len; i++) { |
421 | struct bt_stream *stream_candidate = | |
422 | g_ptr_array_index(trace->streams, i); | |
cfeb617e | 423 | |
7b33a0e0 PP |
424 | if (stream_candidate->id == id) { |
425 | stream = stream_candidate; | |
426 | goto end; | |
427 | } | |
6474e016 | 428 | } |
95c09b3a | 429 | |
bc37ae52 | 430 | end: |
7b33a0e0 | 431 | return stream; |
bc37ae52 JG |
432 | } |
433 | ||
9e550e5f PP |
434 | struct bt_private_stream *bt_private_trace_borrow_private_stream_by_id( |
435 | struct bt_private_trace *trace, uint64_t id) | |
436 | { | |
437 | return (void *) bt_trace_borrow_stream_by_id((void *) trace, id); | |
438 | } | |
439 | ||
7b33a0e0 | 440 | uint64_t bt_trace_get_stream_class_count(struct bt_trace *trace) |
884cd6c3 | 441 | { |
18acc6f8 | 442 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
7b33a0e0 | 443 | return (uint64_t) trace->stream_classes->len; |
884cd6c3 JG |
444 | } |
445 | ||
7b33a0e0 | 446 | struct bt_stream_class *bt_trace_borrow_stream_class_by_index( |
839d52a5 | 447 | struct bt_trace *trace, uint64_t index) |
884cd6c3 | 448 | { |
18acc6f8 | 449 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
7b33a0e0 PP |
450 | BT_ASSERT_PRE_VALID_INDEX(index, trace->stream_classes->len); |
451 | return g_ptr_array_index(trace->stream_classes, index); | |
884cd6c3 JG |
452 | } |
453 | ||
9e550e5f | 454 | struct bt_private_stream_class * |
96854e6a | 455 | bt_private_trace_borrow_stream_class_by_index( |
9e550e5f PP |
456 | struct bt_private_trace *trace, uint64_t index) |
457 | { | |
458 | return (void *) bt_trace_borrow_stream_class_by_index( | |
459 | (void *) trace, index); | |
460 | } | |
461 | ||
7b33a0e0 PP |
462 | struct bt_stream_class *bt_trace_borrow_stream_class_by_id( |
463 | struct bt_trace *trace, uint64_t id) | |
e011d2c1 | 464 | { |
7b33a0e0 PP |
465 | struct bt_stream_class *stream_class = NULL; |
466 | uint64_t i; | |
e011d2c1 | 467 | |
7b33a0e0 | 468 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
e011d2c1 | 469 | |
7b33a0e0 PP |
470 | for (i = 0; i < trace->stream_classes->len; i++) { |
471 | struct bt_stream_class *stream_class_candidate = | |
472 | g_ptr_array_index(trace->stream_classes, i); | |
e011d2c1 | 473 | |
7b33a0e0 PP |
474 | if (stream_class_candidate->id == id) { |
475 | stream_class = stream_class_candidate; | |
476 | goto end; | |
e011d2c1 | 477 | } |
7b33a0e0 | 478 | } |
e011d2c1 | 479 | |
7b33a0e0 PP |
480 | end: |
481 | return stream_class; | |
482 | } | |
e011d2c1 | 483 | |
9e550e5f | 484 | struct bt_private_stream_class * |
96854e6a | 485 | bt_private_trace_borrow_stream_class_by_id( |
9e550e5f PP |
486 | struct bt_private_trace *trace, uint64_t id) |
487 | { | |
488 | return (void *) bt_trace_borrow_stream_class_by_id((void *) trace, id); | |
489 | } | |
490 | ||
939190b3 | 491 | struct bt_field_class *bt_trace_borrow_packet_header_field_class( |
7b33a0e0 PP |
492 | struct bt_trace *trace) |
493 | { | |
494 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); | |
939190b3 | 495 | return trace->packet_header_fc; |
7b33a0e0 | 496 | } |
e011d2c1 | 497 | |
96854e6a | 498 | int bt_private_trace_set_packet_header_field_class( |
9e550e5f PP |
499 | struct bt_private_trace *priv_trace, |
500 | struct bt_private_field_class *priv_field_class) | |
7b33a0e0 PP |
501 | { |
502 | int ret; | |
9e550e5f PP |
503 | struct bt_trace *trace = (void *) priv_trace; |
504 | struct bt_field_class *field_class = (void *) priv_field_class; | |
7b33a0e0 | 505 | struct bt_resolve_field_path_context resolve_ctx = { |
939190b3 | 506 | .packet_header = field_class, |
7b33a0e0 PP |
507 | .packet_context = NULL, |
508 | .event_header = NULL, | |
509 | .event_common_context = NULL, | |
510 | .event_specific_context = NULL, | |
511 | .event_payload = NULL, | |
512 | }; | |
e011d2c1 | 513 | |
7b33a0e0 | 514 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
939190b3 | 515 | BT_ASSERT_PRE_NON_NULL(field_class, "Field class"); |
7b33a0e0 | 516 | BT_ASSERT_PRE_TRACE_HOT(trace); |
af0c18e3 PP |
517 | BT_ASSERT_PRE(bt_field_class_get_type(field_class) == |
518 | BT_FIELD_CLASS_TYPE_STRUCTURE, | |
939190b3 PP |
519 | "Packet header field classe is not a structure field classe: %!+F", |
520 | field_class); | |
521 | ret = bt_resolve_field_paths(field_class, &resolve_ctx); | |
7b33a0e0 PP |
522 | if (ret) { |
523 | goto end; | |
e011d2c1 PP |
524 | } |
525 | ||
939190b3 | 526 | bt_field_class_make_part_of_trace(field_class); |
8138bfe1 PP |
527 | bt_object_put_ref(trace->packet_header_fc); |
528 | trace->packet_header_fc = bt_object_get_ref(field_class); | |
939190b3 PP |
529 | bt_field_class_freeze(field_class); |
530 | BT_LIB_LOGV("Set trace's packet header field classe: %!+t", trace); | |
e011d2c1 | 531 | |
8bf65fbd JG |
532 | end: |
533 | return ret; | |
534 | } | |
535 | ||
839d52a5 | 536 | bt_bool bt_trace_is_static(struct bt_trace *trace) |
5acf2ae6 | 537 | { |
44ea72c5 | 538 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
7b33a0e0 | 539 | return (bt_bool) trace->is_static; |
5acf2ae6 PP |
540 | } |
541 | ||
9e550e5f | 542 | int bt_private_trace_make_static(struct bt_private_trace *priv_trace) |
5acf2ae6 | 543 | { |
9e550e5f | 544 | struct bt_trace *trace = (void *) priv_trace; |
7b33a0e0 | 545 | uint64_t i; |
bae99cf9 | 546 | |
7b33a0e0 PP |
547 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
548 | trace->is_static = true; | |
18acc6f8 | 549 | bt_trace_freeze(trace); |
7b33a0e0 | 550 | BT_LIB_LOGV("Trace is now static: %!+t", trace); |
5acf2ae6 | 551 | |
3602afb0 PP |
552 | /* Call all the "trace is static" listeners */ |
553 | for (i = 0; i < trace->is_static_listeners->len; i++) { | |
839d52a5 | 554 | struct bt_trace_is_static_listener_elem elem = |
3602afb0 | 555 | g_array_index(trace->is_static_listeners, |
839d52a5 | 556 | struct bt_trace_is_static_listener_elem, i); |
3602afb0 PP |
557 | |
558 | if (elem.func) { | |
9e550e5f | 559 | elem.func((void *) trace, elem.data); |
3602afb0 PP |
560 | } |
561 | } | |
562 | ||
7b33a0e0 | 563 | return 0; |
3602afb0 PP |
564 | } |
565 | ||
9e550e5f PP |
566 | int bt_private_trace_add_is_static_listener( |
567 | struct bt_private_trace *priv_trace, | |
568 | bt_private_trace_is_static_listener listener, | |
569 | bt_private_trace_listener_removed listener_removed, void *data, | |
7b33a0e0 | 570 | uint64_t *listener_id) |
3602afb0 | 571 | { |
9e550e5f | 572 | struct bt_trace *trace = (void *) priv_trace; |
7b33a0e0 | 573 | uint64_t i; |
839d52a5 | 574 | struct bt_trace_is_static_listener_elem new_elem = { |
3602afb0 | 575 | .func = listener, |
9962ad4b | 576 | .removed = listener_removed, |
3602afb0 PP |
577 | .data = data, |
578 | }; | |
579 | ||
7b33a0e0 PP |
580 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
581 | BT_ASSERT_PRE_NON_NULL(listener, "Listener"); | |
582 | BT_ASSERT_PRE(!trace->is_static, | |
583 | "Trace is already static: %!+t", trace); | |
584 | BT_ASSERT_PRE(trace->in_remove_listener, | |
585 | "Cannot call this function while executing a " | |
586 | "remove listener: %!+t", trace); | |
9962ad4b | 587 | |
3602afb0 PP |
588 | /* Find the next available spot */ |
589 | for (i = 0; i < trace->is_static_listeners->len; i++) { | |
839d52a5 | 590 | struct bt_trace_is_static_listener_elem elem = |
3602afb0 | 591 | g_array_index(trace->is_static_listeners, |
839d52a5 | 592 | struct bt_trace_is_static_listener_elem, i); |
3602afb0 PP |
593 | |
594 | if (!elem.func) { | |
595 | break; | |
596 | } | |
597 | } | |
598 | ||
599 | if (i == trace->is_static_listeners->len) { | |
600 | g_array_append_val(trace->is_static_listeners, new_elem); | |
601 | } else { | |
602 | g_array_insert_val(trace->is_static_listeners, i, new_elem); | |
603 | } | |
604 | ||
7b33a0e0 PP |
605 | if (listener_id) { |
606 | *listener_id = i; | |
607 | } | |
3602afb0 | 608 | |
7b33a0e0 PP |
609 | BT_LIB_LOGV("Added \"trace is static\" listener: " |
610 | "%![trace-]+t, listener-id=%" PRIu64, trace, i); | |
611 | return 0; | |
612 | } | |
613 | ||
614 | BT_ASSERT_PRE_FUNC | |
615 | static | |
616 | bool has_listener_id(struct bt_trace *trace, uint64_t listener_id) | |
617 | { | |
618 | BT_ASSERT(listener_id < trace->is_static_listeners->len); | |
619 | return (&g_array_index(trace->is_static_listeners, | |
620 | struct bt_trace_is_static_listener_elem, | |
621 | listener_id))->func != NULL; | |
3602afb0 PP |
622 | } |
623 | ||
9e550e5f PP |
624 | int bt_private_trace_remove_is_static_listener( |
625 | struct bt_private_trace *priv_trace, uint64_t listener_id) | |
3602afb0 | 626 | { |
9e550e5f | 627 | struct bt_trace *trace = (void *) priv_trace; |
839d52a5 | 628 | struct bt_trace_is_static_listener_elem *elem; |
3602afb0 | 629 | |
7b33a0e0 PP |
630 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
631 | BT_ASSERT_PRE(!trace->is_static, | |
632 | "Trace is already static: %!+t", trace); | |
633 | BT_ASSERT_PRE(trace->in_remove_listener, | |
634 | "Cannot call this function while executing a " | |
635 | "remove listener: %!+t", trace); | |
636 | BT_ASSERT_PRE(has_listener_id(trace, listener_id), | |
637 | "Trace has no such \"trace is static\" listener ID: " | |
638 | "%![trace-]+t, %" PRIu64, trace, listener_id); | |
3602afb0 | 639 | elem = &g_array_index(trace->is_static_listeners, |
839d52a5 | 640 | struct bt_trace_is_static_listener_elem, |
3602afb0 | 641 | listener_id); |
7b33a0e0 | 642 | BT_ASSERT(elem->func); |
3602afb0 | 643 | |
9962ad4b PP |
644 | if (elem->removed) { |
645 | /* Call remove listener */ | |
7b33a0e0 PP |
646 | BT_LIB_LOGV("Calling remove listener: " |
647 | "%![trace-]+t, listener-id=%" PRIu64, | |
648 | trace, listener_id); | |
649 | trace->in_remove_listener = true; | |
9e550e5f | 650 | elem->removed((void *) trace, elem->data); |
7b33a0e0 | 651 | trace->in_remove_listener = false; |
9962ad4b PP |
652 | } |
653 | ||
3602afb0 | 654 | elem->func = NULL; |
9962ad4b | 655 | elem->removed = NULL; |
3602afb0 | 656 | elem->data = NULL; |
7b33a0e0 PP |
657 | BT_LIB_LOGV("Removed \"trace is static\" listener: " |
658 | "%![trace-]+t, listener-id=%" PRIu64, | |
659 | trace, listener_id); | |
660 | return 0; | |
661 | } | |
3602afb0 | 662 | |
7b33a0e0 PP |
663 | BT_HIDDEN |
664 | void _bt_trace_freeze(struct bt_trace *trace) | |
665 | { | |
939190b3 | 666 | /* The packet header field classe is already frozen */ |
7b33a0e0 PP |
667 | BT_ASSERT(trace); |
668 | BT_LIB_LOGD("Freezing trace: %!+t", trace); | |
669 | trace->frozen = true; | |
5acf2ae6 | 670 | } |
a6918753 | 671 | |
7b33a0e0 | 672 | bt_bool bt_trace_assigns_automatic_stream_class_id(struct bt_trace *trace) |
a6918753 | 673 | { |
7b33a0e0 PP |
674 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
675 | return (bt_bool) trace->assigns_automatic_stream_class_id; | |
676 | } | |
a6918753 | 677 | |
c8bbf821 | 678 | void bt_private_trace_set_assigns_automatic_stream_class_id( |
9e550e5f | 679 | struct bt_private_trace *priv_trace, bt_bool value) |
7b33a0e0 | 680 | { |
9e550e5f PP |
681 | struct bt_trace *trace = (void *) priv_trace; |
682 | ||
a6918753 | 683 | BT_ASSERT_PRE_NON_NULL(trace, "Trace"); |
7b33a0e0 PP |
684 | BT_ASSERT_PRE_TRACE_HOT(trace); |
685 | trace->assigns_automatic_stream_class_id = (bool) value; | |
686 | BT_LIB_LOGV("Set trace's automatic stream class ID " | |
687 | "assignment property: %!+t", trace); | |
7b33a0e0 | 688 | } |
a6918753 | 689 | |
7b33a0e0 PP |
690 | BT_HIDDEN |
691 | void bt_trace_add_stream(struct bt_trace *trace, struct bt_stream *stream) | |
692 | { | |
693 | guint count = 0; | |
694 | ||
695 | bt_object_set_parent(&stream->base, &trace->base); | |
696 | g_ptr_array_add(trace->streams, stream); | |
18acc6f8 | 697 | bt_trace_freeze(trace); |
a6918753 | 698 | |
7b33a0e0 PP |
699 | if (bt_g_hash_table_contains(trace->stream_classes_stream_count, |
700 | stream->class)) { | |
701 | count = GPOINTER_TO_UINT(g_hash_table_lookup( | |
702 | trace->stream_classes_stream_count, stream->class)); | |
a6918753 PP |
703 | } |
704 | ||
7b33a0e0 PP |
705 | g_hash_table_insert(trace->stream_classes_stream_count, |
706 | stream->class, GUINT_TO_POINTER(count + 1)); | |
707 | } | |
708 | ||
709 | BT_HIDDEN | |
710 | uint64_t bt_trace_get_automatic_stream_id(struct bt_trace *trace, | |
711 | struct bt_stream_class *stream_class) | |
712 | { | |
713 | gpointer orig_key; | |
714 | gpointer value; | |
715 | uint64_t id = 0; | |
716 | ||
717 | BT_ASSERT(stream_class); | |
718 | BT_ASSERT(trace); | |
719 | if (g_hash_table_lookup_extended(trace->stream_classes_stream_count, | |
720 | stream_class, &orig_key, &value)) { | |
721 | id = (uint64_t) GPOINTER_TO_UINT(value); | |
722 | } | |
723 | ||
724 | return id; | |
a6918753 | 725 | } |