Add bt_ctf_trace_get_stream_count() and bt_ctf_trace_get_stream()
[babeltrace.git] / lib / ctf-ir / trace.c
CommitLineData
bc37ae52
JG
1/*
2 * trace.c
3 *
4 * Babeltrace CTF IR - Trace
5 *
6 * Copyright 2014 Jérémie Galarneau <jeremie.galarneau@efficios.com>
7 *
8 * Author: Jérémie Galarneau <jeremie.galarneau@efficios.com>
9 *
10 * Permission is hereby granted, free of charge, to any person obtaining a copy
11 * of this software and associated documentation files (the "Software"), to deal
12 * in the Software without restriction, including without limitation the rights
13 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
14 * copies of the Software, and to permit persons to whom the Software is
15 * furnished to do so, subject to the following conditions:
16 *
17 * The above copyright notice and this permission notice shall be included in
18 * all copies or substantial portions of the Software.
19 *
20 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
21 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
22 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
23 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
24 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
25 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
26 * SOFTWARE.
27 */
28
29#include <babeltrace/ctf-ir/trace-internal.h>
ac0c6bdd 30#include <babeltrace/ctf-ir/clock-class-internal.h>
bc37ae52
JG
31#include <babeltrace/ctf-ir/stream-internal.h>
32#include <babeltrace/ctf-ir/stream-class-internal.h>
09840de5 33#include <babeltrace/ctf-ir/event-internal.h>
272df73e
PP
34#include <babeltrace/ctf-ir/event-class.h>
35#include <babeltrace/ctf-ir/event-class-internal.h>
bc37ae52 36#include <babeltrace/ctf-writer/functor-internal.h>
ac0c6bdd 37#include <babeltrace/ctf-writer/clock-internal.h>
2e33ac5a 38#include <babeltrace/ctf-ir/field-types-internal.h>
44e0a4f5 39#include <babeltrace/ctf-ir/attributes-internal.h>
09840de5 40#include <babeltrace/ctf-ir/validation-internal.h>
8bf65fbd 41#include <babeltrace/ctf-ir/visitor-internal.h>
654c1444 42#include <babeltrace/ctf-ir/utils.h>
3d9990ac 43#include <babeltrace/compiler-internal.h>
dac5c838 44#include <babeltrace/values.h>
83509119 45#include <babeltrace/ref.h>
3d9990ac 46#include <babeltrace/endian-internal.h>
dc3fffef 47#include <inttypes.h>
bc37ae52
JG
48
49#define DEFAULT_IDENTIFIER_SIZE 128
50#define DEFAULT_METADATA_STRING_SIZE 4096
51
50ad4244
JG
52struct listener_wrapper {
53 bt_ctf_listener_cb listener;
2204c381
JG
54 void *data;
55};
56
bc37ae52 57static
83509119 58void bt_ctf_trace_destroy(struct bt_object *obj);
bc37ae52
JG
59static
60int init_trace_packet_header(struct bt_ctf_trace *trace);
f67f3a6e 61static
09840de5 62void bt_ctf_trace_freeze(struct bt_ctf_trace *trace);
bc37ae52 63
bc37ae52
JG
64static
65const unsigned int field_type_aliases_alignments[] = {
66 [FIELD_TYPE_ALIAS_UINT5_T] = 1,
67 [FIELD_TYPE_ALIAS_UINT8_T ... FIELD_TYPE_ALIAS_UINT16_T] = 8,
68 [FIELD_TYPE_ALIAS_UINT27_T] = 1,
69 [FIELD_TYPE_ALIAS_UINT32_T ... FIELD_TYPE_ALIAS_UINT64_T] = 8,
70};
71
72static
73const unsigned int field_type_aliases_sizes[] = {
74 [FIELD_TYPE_ALIAS_UINT5_T] = 5,
75 [FIELD_TYPE_ALIAS_UINT8_T] = 8,
76 [FIELD_TYPE_ALIAS_UINT16_T] = 16,
77 [FIELD_TYPE_ALIAS_UINT27_T] = 27,
78 [FIELD_TYPE_ALIAS_UINT32_T] = 32,
79 [FIELD_TYPE_ALIAS_UINT64_T] = 64,
80};
81
bc37ae52
JG
82struct bt_ctf_trace *bt_ctf_trace_create(void)
83{
84 struct bt_ctf_trace *trace = NULL;
85
86 trace = g_new0(struct bt_ctf_trace, 1);
87 if (!trace) {
88 goto error;
89 }
90
391c8f0d 91 trace->native_byte_order = BT_CTF_BYTE_ORDER_NATIVE;
83509119 92 bt_object_init(trace, bt_ctf_trace_destroy);
bc37ae52 93 trace->clocks = g_ptr_array_new_with_free_func(
83509119 94 (GDestroyNotify) bt_put);
bc37ae52 95 trace->streams = g_ptr_array_new_with_free_func(
e6a8e8e4 96 (GDestroyNotify) bt_object_release);
bc37ae52 97 trace->stream_classes = g_ptr_array_new_with_free_func(
e6a8e8e4 98 (GDestroyNotify) bt_object_release);
7f800dc7 99 if (!trace->clocks || !trace->stream_classes || !trace->streams) {
83509119 100 goto error;
bc37ae52
JG
101 }
102
103 /* Generate a trace UUID */
104 uuid_generate(trace->uuid);
105 if (init_trace_packet_header(trace)) {
83509119 106 goto error;
bc37ae52
JG
107 }
108
7f800dc7
PP
109 /* Create the environment array object */
110 trace->environment = bt_ctf_attributes_create();
111 if (!trace->environment) {
83509119 112 goto error;
7f800dc7
PP
113 }
114
9b888ff3
JG
115 trace->listeners = g_ptr_array_new_with_free_func(
116 (GDestroyNotify) g_free);
117 if (!trace->listeners) {
118 goto error;
119 }
120
bc37ae52
JG
121 return trace;
122
bc37ae52 123error:
83509119 124 BT_PUT(trace);
bc37ae52
JG
125 return trace;
126}
127
e96045d4
JG
128const char *bt_ctf_trace_get_name(struct bt_ctf_trace *trace)
129{
130 const char *name = NULL;
131
132 if (!trace || !trace->name) {
133 goto end;
134 }
135
136 name = trace->name->str;
137end:
138 return name;
139}
140
141int bt_ctf_trace_set_name(struct bt_ctf_trace *trace, const char *name)
142{
143 int ret = 0;
144
145 if (!trace || !name || trace->frozen) {
146 ret = -1;
147 goto end;
148 }
149
150 trace->name = trace->name ? g_string_assign(trace->name, name) :
151 g_string_new(name);
152 if (!trace->name) {
153 ret = -1;
154 goto end;
155 }
156end:
157 return ret;
158}
159
83509119 160void bt_ctf_trace_destroy(struct bt_object *obj)
bc37ae52
JG
161{
162 struct bt_ctf_trace *trace;
bc37ae52 163
83509119 164 trace = container_of(obj, struct bt_ctf_trace, base);
bc37ae52 165 if (trace->environment) {
7f800dc7 166 bt_ctf_attributes_destroy(trace->environment);
bc37ae52
JG
167 }
168
e96045d4
JG
169 if (trace->name) {
170 g_string_free(trace->name, TRUE);
171 }
172
bc37ae52
JG
173 if (trace->clocks) {
174 g_ptr_array_free(trace->clocks, TRUE);
175 }
176
177 if (trace->streams) {
178 g_ptr_array_free(trace->streams, TRUE);
179 }
180
181 if (trace->stream_classes) {
182 g_ptr_array_free(trace->stream_classes, TRUE);
183 }
184
9b888ff3
JG
185 if (trace->listeners) {
186 g_ptr_array_free(trace->listeners, TRUE);
187 }
188
83509119 189 bt_put(trace->packet_header_type);
bc37ae52
JG
190 g_free(trace);
191}
192
7f800dc7 193int bt_ctf_trace_set_environment_field(struct bt_ctf_trace *trace,
dac5c838 194 const char *name, struct bt_value *value)
bc37ae52 195{
bc37ae52
JG
196 int ret = 0;
197
a0d12916 198 if (!trace || !name || !value ||
7f800dc7 199 bt_ctf_validate_identifier(name) ||
dac5c838 200 !(bt_value_is_integer(value) || bt_value_is_string(value))) {
bc37ae52 201 ret = -1;
7f800dc7 202 goto end;
bc37ae52
JG
203 }
204
205 if (strchr(name, ' ')) {
206 ret = -1;
7f800dc7 207 goto end;
bc37ae52
JG
208 }
209
a0d12916
JG
210 if (trace->frozen) {
211 /*
212 * New environment fields may be added to a frozen trace,
213 * but existing fields may not be changed.
214 *
215 * The object passed is frozen like all other attributes.
216 */
dac5c838 217 struct bt_value *attribute =
a0d12916
JG
218 bt_ctf_attributes_get_field_value_by_name(
219 trace->environment, name);
220
221 if (attribute) {
83509119 222 BT_PUT(attribute);
a0d12916
JG
223 ret = -1;
224 goto end;
225 }
226
dac5c838 227 bt_value_freeze(value);
a0d12916
JG
228 }
229
7f800dc7
PP
230 ret = bt_ctf_attributes_set_field_value(trace->environment, name,
231 value);
bc37ae52 232
7f800dc7
PP
233end:
234 return ret;
235}
bc37ae52 236
7f800dc7
PP
237int bt_ctf_trace_set_environment_field_string(struct bt_ctf_trace *trace,
238 const char *name, const char *value)
239{
240 int ret = 0;
dac5c838 241 struct bt_value *env_value_string_obj = NULL;
7f800dc7
PP
242
243 if (!trace || !name || !value) {
bc37ae52 244 ret = -1;
7f800dc7 245 goto end;
bc37ae52
JG
246 }
247
a0d12916
JG
248 if (trace->frozen) {
249 /*
250 * New environment fields may be added to a frozen trace,
251 * but existing fields may not be changed.
252 */
dac5c838 253 struct bt_value *attribute =
a0d12916
JG
254 bt_ctf_attributes_get_field_value_by_name(
255 trace->environment, name);
256
257 if (attribute) {
83509119 258 BT_PUT(attribute);
a0d12916
JG
259 ret = -1;
260 goto end;
261 }
262 }
263
dac5c838 264 env_value_string_obj = bt_value_string_create_init(value);
bc37ae52 265
7f800dc7
PP
266 if (!env_value_string_obj) {
267 ret = -1;
268 goto end;
bc37ae52
JG
269 }
270
a0d12916 271 if (trace->frozen) {
dac5c838 272 bt_value_freeze(env_value_string_obj);
a0d12916 273 }
7f800dc7
PP
274 ret = bt_ctf_trace_set_environment_field(trace, name,
275 env_value_string_obj);
276
277end:
83509119 278 BT_PUT(env_value_string_obj);
3487c9f3
JG
279 return ret;
280}
281
7f800dc7
PP
282int bt_ctf_trace_set_environment_field_integer(struct bt_ctf_trace *trace,
283 const char *name, int64_t value)
3487c9f3 284{
3487c9f3 285 int ret = 0;
dac5c838 286 struct bt_value *env_value_integer_obj = NULL;
3487c9f3
JG
287
288 if (!trace || !name) {
289 ret = -1;
7f800dc7 290 goto end;
3487c9f3
JG
291 }
292
a0d12916
JG
293 if (trace->frozen) {
294 /*
295 * New environment fields may be added to a frozen trace,
296 * but existing fields may not be changed.
297 */
dac5c838 298 struct bt_value *attribute =
a0d12916
JG
299 bt_ctf_attributes_get_field_value_by_name(
300 trace->environment, name);
301
302 if (attribute) {
83509119 303 BT_PUT(attribute);
a0d12916
JG
304 ret = -1;
305 goto end;
306 }
307 }
308
dac5c838 309 env_value_integer_obj = bt_value_integer_create_init(value);
7f800dc7 310 if (!env_value_integer_obj) {
3487c9f3 311 ret = -1;
7f800dc7 312 goto end;
3487c9f3
JG
313 }
314
7f800dc7
PP
315 ret = bt_ctf_trace_set_environment_field(trace, name,
316 env_value_integer_obj);
a0d12916 317 if (trace->frozen) {
dac5c838 318 bt_value_freeze(env_value_integer_obj);
a0d12916 319 }
7f800dc7 320end:
83509119 321 BT_PUT(env_value_integer_obj);
bc37ae52
JG
322 return ret;
323}
324
e6fa2160
JG
325int bt_ctf_trace_get_environment_field_count(struct bt_ctf_trace *trace)
326{
327 int ret = 0;
328
329 if (!trace) {
330 ret = -1;
331 goto end;
332 }
333
7f800dc7 334 ret = bt_ctf_attributes_get_count(trace->environment);
e6fa2160 335
e6fa2160 336end:
7f800dc7 337 return ret;
e6fa2160
JG
338}
339
340const char *
341bt_ctf_trace_get_environment_field_name(struct bt_ctf_trace *trace,
342 int index)
343{
e6fa2160
JG
344 const char *ret = NULL;
345
7f800dc7 346 if (!trace) {
e6fa2160
JG
347 goto end;
348 }
349
7f800dc7
PP
350 ret = bt_ctf_attributes_get_field_name(trace->environment, index);
351
e6fa2160
JG
352end:
353 return ret;
354}
355
dac5c838 356struct bt_value *bt_ctf_trace_get_environment_field_value(
7f800dc7 357 struct bt_ctf_trace *trace, int index)
e6fa2160 358{
dac5c838 359 struct bt_value *ret = NULL;
e6fa2160 360
7f800dc7 361 if (!trace) {
e6fa2160
JG
362 goto end;
363 }
364
7f800dc7
PP
365 ret = bt_ctf_attributes_get_field_value(trace->environment, index);
366
e6fa2160
JG
367end:
368 return ret;
369}
370
dac5c838 371struct bt_value *bt_ctf_trace_get_environment_field_value_by_name(
7f800dc7 372 struct bt_ctf_trace *trace, const char *name)
e6fa2160 373{
dac5c838 374 struct bt_value *ret = NULL;
e6fa2160 375
7f800dc7 376 if (!trace || !name) {
e6fa2160
JG
377 goto end;
378 }
379
7f800dc7
PP
380 ret = bt_ctf_attributes_get_field_value_by_name(trace->environment,
381 name);
382
e6fa2160
JG
383end:
384 return ret;
385}
386
ac0c6bdd
PP
387int bt_ctf_trace_add_clock_class(struct bt_ctf_trace *trace,
388 struct bt_ctf_clock_class *clock_class)
bc37ae52
JG
389{
390 int ret = 0;
bc37ae52 391
ac0c6bdd 392 if (!trace || !bt_ctf_clock_class_is_valid(clock_class)) {
bc37ae52
JG
393 ret = -1;
394 goto end;
395 }
396
ac0c6bdd 397 /* Check for duplicate clock classes */
c9d90a34 398 if (bt_ctf_trace_has_clock_class(trace, clock_class)) {
bc37ae52
JG
399 ret = -1;
400 goto end;
401 }
402
ac0c6bdd
PP
403 bt_get(clock_class);
404 g_ptr_array_add(trace->clocks, clock_class);
cfeb617e 405
6474e016 406 if (trace->frozen) {
ac0c6bdd 407 bt_ctf_clock_class_freeze(clock_class);
6474e016 408 }
bc37ae52
JG
409end:
410 return ret;
411}
412
ac0c6bdd 413int bt_ctf_trace_get_clock_class_count(struct bt_ctf_trace *trace)
884cd6c3
JG
414{
415 int ret = -1;
416
417 if (!trace) {
418 goto end;
419 }
420
421 ret = trace->clocks->len;
422end:
423 return ret;
424}
425
ac0c6bdd
PP
426struct bt_ctf_clock_class *bt_ctf_trace_get_clock_class(
427 struct bt_ctf_trace *trace, int index)
884cd6c3 428{
ac0c6bdd 429 struct bt_ctf_clock_class *clock_class = NULL;
884cd6c3
JG
430
431 if (!trace || index < 0 || index >= trace->clocks->len) {
432 goto end;
433 }
434
ac0c6bdd
PP
435 clock_class = g_ptr_array_index(trace->clocks, index);
436 bt_get(clock_class);
884cd6c3 437end:
ac0c6bdd 438 return clock_class;
884cd6c3
JG
439}
440
ef0c4a15
JG
441int bt_ctf_trace_add_stream_class(struct bt_ctf_trace *trace,
442 struct bt_ctf_stream_class *stream_class)
443{
444 int ret, i;
445 int64_t stream_id;
09840de5
PP
446 struct bt_ctf_validation_output trace_sc_validation_output = { 0 };
447 struct bt_ctf_validation_output *ec_validation_outputs = NULL;
448 const enum bt_ctf_validation_flag trace_sc_validation_flags =
449 BT_CTF_VALIDATION_FLAG_TRACE |
450 BT_CTF_VALIDATION_FLAG_STREAM;
451 const enum bt_ctf_validation_flag ec_validation_flags =
452 BT_CTF_VALIDATION_FLAG_EVENT;
453 struct bt_ctf_field_type *packet_header_type = NULL;
454 struct bt_ctf_field_type *packet_context_type = NULL;
455 struct bt_ctf_field_type *event_header_type = NULL;
456 struct bt_ctf_field_type *stream_event_ctx_type = NULL;
457 int event_class_count;
2789e5d9 458 struct bt_ctf_trace *current_parent_trace = NULL;
ef0c4a15
JG
459
460 if (!trace || !stream_class) {
461 ret = -1;
462 goto end;
463 }
464
d490fcbe
PP
465 /*
466 * At the end of this function we freeze the trace, so its
467 * native byte order must NOT be BT_CTF_BYTE_ORDER_NATIVE.
468 */
469 if (trace->native_byte_order == BT_CTF_BYTE_ORDER_NATIVE) {
470 ret = -1;
471 goto end;
472 }
473
2789e5d9
JG
474 current_parent_trace = bt_ctf_stream_class_get_trace(stream_class);
475 if (current_parent_trace) {
476 /* Stream class is already associated to a trace, abort. */
477 ret = -1;
478 goto end;
479 }
480
09840de5
PP
481 event_class_count =
482 bt_ctf_stream_class_get_event_class_count(stream_class);
483 assert(event_class_count >= 0);
484
485 /* Check for duplicate stream classes */
ef0c4a15
JG
486 for (i = 0; i < trace->stream_classes->len; i++) {
487 if (trace->stream_classes->pdata[i] == stream_class) {
09840de5 488 /* Stream class already registered to the trace */
ef0c4a15
JG
489 ret = -1;
490 goto end;
491 }
492 }
493
fe13b5c0 494 if (stream_class->clock) {
ac0c6bdd
PP
495 struct bt_ctf_clock_class *stream_clock_class =
496 stream_class->clock->clock_class;
497
498 if (trace->is_created_by_writer) {
499 /*
500 * Make sure this clock was also added to the
501 * trace (potentially through its CTF writer
502 * owner).
503 */
504 size_t i;
505
506 for (i = 0; i < trace->clocks->len; i++) {
507 if (trace->clocks->pdata[i] ==
508 stream_clock_class) {
509 /* Found! */
510 break;
511 }
512 }
513
514 if (i == trace->clocks->len) {
515 /* Not found */
fe13b5c0
PP
516 ret = -1;
517 goto end;
518 }
519 } else {
ac0c6bdd
PP
520 /*
521 * This trace was NOT created by a CTF writer,
522 * thus do not allow the stream class to add to
523 * have a clock at all. Those are two
524 * independent APIs (non-writer and writer
525 * APIs), and isolating them simplifies things.
526 */
527 ret = -1;
528 goto end;
fe13b5c0
PP
529 }
530 }
531
09840de5
PP
532 /*
533 * We're about to freeze both the trace and the stream class.
534 * Also, each event class contained in this stream class are
535 * already frozen.
536 *
537 * This trace, this stream class, and all its event classes
538 * should be valid at this point.
539 *
540 * Validate trace and stream class first, then each event
541 * class of this stream class can be validated individually.
542 */
543 packet_header_type =
544 bt_ctf_trace_get_packet_header_type(trace);
545 packet_context_type =
546 bt_ctf_stream_class_get_packet_context_type(stream_class);
547 event_header_type =
548 bt_ctf_stream_class_get_event_header_type(stream_class);
549 stream_event_ctx_type =
550 bt_ctf_stream_class_get_event_context_type(stream_class);
551 ret = bt_ctf_validate_class_types(trace->environment,
552 packet_header_type, packet_context_type, event_header_type,
553 stream_event_ctx_type, NULL, NULL, trace->valid,
554 stream_class->valid, 1, &trace_sc_validation_output,
555 trace_sc_validation_flags);
556 BT_PUT(packet_header_type);
557 BT_PUT(packet_context_type);
558 BT_PUT(event_header_type);
559 BT_PUT(stream_event_ctx_type);
560
2bd7f864 561 if (ret) {
09840de5
PP
562 /*
563 * This means something went wrong during the validation
564 * process, not that the objects are invalid.
565 */
2bd7f864
JG
566 goto end;
567 }
568
09840de5
PP
569 if ((trace_sc_validation_output.valid_flags &
570 trace_sc_validation_flags) !=
571 trace_sc_validation_flags) {
572 /* Invalid trace/stream class */
573 ret = -1;
574 goto end;
575 }
576
577 if (event_class_count > 0) {
578 ec_validation_outputs = g_new0(struct bt_ctf_validation_output,
579 event_class_count);
580 if (!ec_validation_outputs) {
581 ret = -1;
582 goto end;
583 }
584 }
585
586 /* Validate each event class individually */
6474e016 587 for (i = 0; i < event_class_count; i++) {
09840de5
PP
588 struct bt_ctf_event_class *event_class =
589 bt_ctf_stream_class_get_event_class(stream_class, i);
590 struct bt_ctf_field_type *event_context_type = NULL;
591 struct bt_ctf_field_type *event_payload_type = NULL;
592
593 event_context_type =
594 bt_ctf_event_class_get_context_type(event_class);
595 event_payload_type =
596 bt_ctf_event_class_get_payload_type(event_class);
597
598 /*
599 * It is important to use the field types returned by
600 * the previous trace and stream class validation here
601 * because copies could have been made.
602 */
603 ret = bt_ctf_validate_class_types(trace->environment,
604 trace_sc_validation_output.packet_header_type,
605 trace_sc_validation_output.packet_context_type,
606 trace_sc_validation_output.event_header_type,
607 trace_sc_validation_output.stream_event_ctx_type,
608 event_context_type, event_payload_type,
609 1, 1, event_class->valid, &ec_validation_outputs[i],
610 ec_validation_flags);
611 BT_PUT(event_context_type);
612 BT_PUT(event_payload_type);
613 BT_PUT(event_class);
614
615 if (ret) {
616 goto end;
617 }
618
619 if ((ec_validation_outputs[i].valid_flags &
620 ec_validation_flags) != ec_validation_flags) {
621 /* Invalid event class */
622 ret = -1;
623 goto end;
624 }
625 }
626
ef0c4a15
JG
627 stream_id = bt_ctf_stream_class_get_id(stream_class);
628 if (stream_id < 0) {
629 stream_id = trace->next_stream_id++;
630
631 /* Try to assign a new stream id */
632 for (i = 0; i < trace->stream_classes->len; i++) {
633 if (stream_id == bt_ctf_stream_class_get_id(
634 trace->stream_classes->pdata[i])) {
635 /* Duplicate stream id found */
636 ret = -1;
637 goto end;
638 }
639 }
640
0ddffae5
JG
641 if (bt_ctf_stream_class_set_id_no_check(stream_class,
642 stream_id)) {
ef0c4a15
JG
643 /* TODO Should retry with a different stream id */
644 ret = -1;
645 goto end;
646 }
647 }
648
e6a8e8e4 649 bt_object_set_parent(stream_class, trace);
ef0c4a15
JG
650 g_ptr_array_add(trace->stream_classes, stream_class);
651
652 /*
09840de5
PP
653 * At this point we know that the function will be successful.
654 * Therefore we can replace the trace and stream class field
655 * types with what's in their validation output structure and
656 * mark them as valid. We can also replace the field types of
657 * all the event classes of the stream class and mark them as
658 * valid.
659 */
660 bt_ctf_validation_replace_types(trace, stream_class, NULL,
661 &trace_sc_validation_output, trace_sc_validation_flags);
662 trace->valid = 1;
663 stream_class->valid = 1;
664
665 /*
666 * Put what was not moved in bt_ctf_validation_replace_types().
667 */
668 bt_ctf_validation_output_put_types(&trace_sc_validation_output);
669
6474e016 670 for (i = 0; i < event_class_count; i++) {
09840de5
PP
671 struct bt_ctf_event_class *event_class =
672 bt_ctf_stream_class_get_event_class(stream_class, i);
673
674 bt_ctf_validation_replace_types(NULL, NULL, event_class,
675 &ec_validation_outputs[i], ec_validation_flags);
676 event_class->valid = 1;
677 BT_PUT(event_class);
678
679 /*
680 * Put what was not moved in
681 * bt_ctf_validation_replace_types().
682 */
683 bt_ctf_validation_output_put_types(&ec_validation_outputs[i]);
684 }
685
09840de5
PP
686 /*
687 * Freeze the trace and the stream class.
688 */
f67f3a6e 689 bt_ctf_stream_class_freeze(stream_class);
09840de5
PP
690 bt_ctf_trace_freeze(trace);
691
9b888ff3
JG
692 /* Notifiy listeners of the trace's schema modification. */
693 bt_ctf_stream_class_visit(stream_class,
d9a13d86 694 bt_ctf_trace_object_modification, trace);
ef0c4a15 695end:
d3814b54 696 if (ret) {
e6a8e8e4 697 bt_object_set_parent(stream_class, NULL);
09840de5
PP
698
699 if (ec_validation_outputs) {
6474e016 700 for (i = 0; i < event_class_count; i++) {
09840de5
PP
701 bt_ctf_validation_output_put_types(
702 &ec_validation_outputs[i]);
703 }
704 }
d3814b54 705 }
09840de5
PP
706
707 g_free(ec_validation_outputs);
708 bt_ctf_validation_output_put_types(&trace_sc_validation_output);
2789e5d9 709 bt_put(current_parent_trace);
09840de5
PP
710 assert(!packet_header_type);
711 assert(!packet_context_type);
712 assert(!event_header_type);
713 assert(!stream_event_ctx_type);
714
ef0c4a15
JG
715 return ret;
716}
717
c1e730fe
PP
718int bt_ctf_trace_get_stream_count(struct bt_ctf_trace *trace)
719{
720 int ret;
721
722 if (!trace) {
723 ret = -1;
724 goto end;
725 }
726
727 ret = trace->streams->len;
728
729end:
730 return ret;
731}
732
733struct bt_ctf_stream *bt_ctf_trace_get_stream(struct bt_ctf_trace *trace,
734 int index)
735{
736 struct bt_ctf_stream *stream = NULL;
737
738 if (!trace || index >= trace->streams->len) {
739 goto end;
740 }
741
742 stream = bt_get(g_ptr_array_index(trace->streams, index));
743
744end:
745 return stream;
746}
747
ef0c4a15
JG
748int bt_ctf_trace_get_stream_class_count(struct bt_ctf_trace *trace)
749{
750 int ret;
751
752 if (!trace) {
753 ret = -1;
754 goto end;
755 }
756
757 ret = trace->stream_classes->len;
758end:
759 return ret;
760}
761
762struct bt_ctf_stream_class *bt_ctf_trace_get_stream_class(
763 struct bt_ctf_trace *trace, int index)
764{
765 struct bt_ctf_stream_class *stream_class = NULL;
766
767 if (!trace || index < 0 || index >= trace->stream_classes->len) {
768 goto end;
769 }
770
771 stream_class = g_ptr_array_index(trace->stream_classes, index);
83509119 772 bt_get(stream_class);
ef0c4a15
JG
773end:
774 return stream_class;
775}
776
4841ccc1
PP
777struct bt_ctf_stream_class *bt_ctf_trace_get_stream_class_by_id(
778 struct bt_ctf_trace *trace, uint32_t id)
779{
780 int i;
781 struct bt_ctf_stream_class *stream_class = NULL;
782
783 if (!trace) {
784 goto end;
785 }
786
6474e016 787 for (i = 0; i < trace->stream_classes->len; i++) {
4841ccc1
PP
788 struct bt_ctf_stream_class *stream_class_candidate;
789
790 stream_class_candidate =
791 g_ptr_array_index(trace->stream_classes, i);
792
793 if (bt_ctf_stream_class_get_id(stream_class_candidate) ==
794 (int64_t) id) {
795 stream_class = stream_class_candidate;
83509119 796 bt_get(stream_class);
4841ccc1
PP
797 goto end;
798 }
799 }
800
801end:
802 return stream_class;
803}
804
ac0c6bdd 805struct bt_ctf_clock_class *bt_ctf_trace_get_clock_class_by_name(
7e4347a3
PP
806 struct bt_ctf_trace *trace, const char *name)
807{
808 size_t i;
ac0c6bdd 809 struct bt_ctf_clock_class *clock_class = NULL;
7e4347a3
PP
810
811 if (!trace || !name) {
812 goto end;
813 }
814
6474e016 815 for (i = 0; i < trace->clocks->len; i++) {
ac0c6bdd 816 struct bt_ctf_clock_class *cur_clk =
7e4347a3 817 g_ptr_array_index(trace->clocks, i);
ac0c6bdd 818 const char *cur_clk_name = bt_ctf_clock_class_get_name(cur_clk);
7e4347a3
PP
819
820 if (!cur_clk_name) {
821 goto end;
822 }
823
824 if (!strcmp(cur_clk_name, name)) {
ac0c6bdd
PP
825 clock_class = cur_clk;
826 bt_get(clock_class);
7e4347a3
PP
827 goto end;
828 }
829 }
830
831end:
ac0c6bdd 832 return clock_class;
7e4347a3
PP
833}
834
c9d90a34
PP
835BT_HIDDEN
836bool bt_ctf_trace_has_clock_class(struct bt_ctf_trace *trace,
837 struct bt_ctf_clock_class *clock_class)
838{
839 struct search_query query = { .value = clock_class, .found = 0 };
840
841 assert(trace);
842 assert(clock_class);
843
844 g_ptr_array_foreach(trace->clocks, value_exists, &query);
845 return query.found;
846}
847
bc37ae52 848BT_HIDDEN
dc3fffef 849const char *get_byte_order_string(enum bt_ctf_byte_order byte_order)
bc37ae52
JG
850{
851 const char *string;
852
853 switch (byte_order) {
dc3fffef 854 case BT_CTF_BYTE_ORDER_LITTLE_ENDIAN:
bc37ae52
JG
855 string = "le";
856 break;
dc3fffef 857 case BT_CTF_BYTE_ORDER_BIG_ENDIAN:
bc37ae52
JG
858 string = "be";
859 break;
dc3fffef
PP
860 case BT_CTF_BYTE_ORDER_NATIVE:
861 string = "native";
bc37ae52 862 break;
dc3fffef
PP
863 default:
864 assert(false);
bc37ae52
JG
865 }
866
867 return string;
868}
869
870static
871int append_trace_metadata(struct bt_ctf_trace *trace,
872 struct metadata_context *context)
873{
874 unsigned char *uuid = trace->uuid;
875 int ret;
876
877 g_string_append(context->string, "trace {\n");
878
879 g_string_append(context->string, "\tmajor = 1;\n");
880 g_string_append(context->string, "\tminor = 8;\n");
dc3fffef
PP
881 assert(trace->native_byte_order == BT_CTF_BYTE_ORDER_LITTLE_ENDIAN ||
882 trace->native_byte_order == BT_CTF_BYTE_ORDER_BIG_ENDIAN ||
883 trace->native_byte_order == BT_CTF_BYTE_ORDER_NETWORK);
bc37ae52
JG
884 g_string_append_printf(context->string,
885 "\tuuid = \"%02x%02x%02x%02x-%02x%02x-%02x%02x-%02x%02x-%02x%02x%02x%02x%02x%02x\";\n",
886 uuid[0], uuid[1], uuid[2], uuid[3],
887 uuid[4], uuid[5], uuid[6], uuid[7],
888 uuid[8], uuid[9], uuid[10], uuid[11],
889 uuid[12], uuid[13], uuid[14], uuid[15]);
890 g_string_append_printf(context->string, "\tbyte_order = %s;\n",
dc3fffef 891 get_byte_order_string(trace->native_byte_order));
bc37ae52
JG
892
893 g_string_append(context->string, "\tpacket.header := ");
894 context->current_indentation_level++;
895 g_string_assign(context->field_name, "");
d246b111 896 ret = bt_ctf_field_type_serialize(trace->packet_header_type,
bc37ae52
JG
897 context);
898 if (ret) {
899 goto end;
900 }
901 context->current_indentation_level--;
902
903 g_string_append(context->string, ";\n};\n\n");
904end:
905 return ret;
906}
907
bc37ae52
JG
908static
909void append_env_metadata(struct bt_ctf_trace *trace,
910 struct metadata_context *context)
911{
7f800dc7
PP
912 int i;
913 int env_size;
914
915 env_size = bt_ctf_attributes_get_count(trace->environment);
916
917 if (env_size <= 0) {
bc37ae52
JG
918 return;
919 }
920
921 g_string_append(context->string, "env {\n");
7f800dc7 922
6474e016 923 for (i = 0; i < env_size; i++) {
dac5c838 924 struct bt_value *env_field_value_obj = NULL;
7f800dc7 925 const char *entry_name;
7f800dc7
PP
926
927 entry_name = bt_ctf_attributes_get_field_name(
928 trace->environment, i);
929 env_field_value_obj = bt_ctf_attributes_get_field_value(
930 trace->environment, i);
931
932 if (!entry_name || !env_field_value_obj) {
933 goto loop_next;
934 }
935
dac5c838
PP
936 switch (bt_value_get_type(env_field_value_obj)) {
937 case BT_VALUE_TYPE_INTEGER:
b8248cc0
PP
938 {
939 int ret;
940 int64_t int_value;
941
dac5c838 942 ret = bt_value_integer_get(env_field_value_obj,
7f800dc7
PP
943 &int_value);
944
945 if (ret) {
946 goto loop_next;
947 }
948
949 g_string_append_printf(context->string,
950 "\t%s = %" PRId64 ";\n", entry_name,
951 int_value);
952 break;
b8248cc0 953 }
dac5c838 954 case BT_VALUE_TYPE_STRING:
7f800dc7
PP
955 {
956 int ret;
957 const char *str_value;
958 char *escaped_str = NULL;
959
dac5c838 960 ret = bt_value_string_get(env_field_value_obj,
7f800dc7
PP
961 &str_value);
962
963 if (ret) {
964 goto loop_next;
965 }
966
967 escaped_str = g_strescape(str_value, NULL);
968
969 if (!escaped_str) {
970 goto loop_next;
971 }
972
973 g_string_append_printf(context->string,
974 "\t%s = \"%s\";\n", entry_name, escaped_str);
975 free(escaped_str);
976 break;
977 }
978
979 default:
980 goto loop_next;
981 }
982
983loop_next:
83509119 984 BT_PUT(env_field_value_obj);
7f800dc7
PP
985 }
986
bc37ae52
JG
987 g_string_append(context->string, "};\n\n");
988}
989
990char *bt_ctf_trace_get_metadata_string(struct bt_ctf_trace *trace)
991{
992 char *metadata = NULL;
993 struct metadata_context *context = NULL;
994 int err = 0;
995 size_t i;
996
997 if (!trace) {
998 goto end;
999 }
1000
1001 context = g_new0(struct metadata_context, 1);
1002 if (!context) {
1003 goto end;
1004 }
1005
1006 context->field_name = g_string_sized_new(DEFAULT_IDENTIFIER_SIZE);
1007 context->string = g_string_sized_new(DEFAULT_METADATA_STRING_SIZE);
1008 g_string_append(context->string, "/* CTF 1.8 */\n\n");
1009 if (append_trace_metadata(trace, context)) {
1010 goto error;
1011 }
1012 append_env_metadata(trace, context);
1013 g_ptr_array_foreach(trace->clocks,
ac0c6bdd 1014 (GFunc)bt_ctf_clock_class_serialize, context);
bc37ae52
JG
1015
1016 for (i = 0; i < trace->stream_classes->len; i++) {
1017 err = bt_ctf_stream_class_serialize(
1018 trace->stream_classes->pdata[i], context);
1019 if (err) {
1020 goto error;
1021 }
1022 }
1023
1024 metadata = context->string->str;
1025error:
1026 g_string_free(context->string, err ? TRUE : FALSE);
1027 g_string_free(context->field_name, TRUE);
1028 g_free(context);
1029end:
1030 return metadata;
1031}
1032
4ed90fb3
JG
1033enum bt_ctf_byte_order bt_ctf_trace_get_byte_order(struct bt_ctf_trace *trace)
1034{
1035 enum bt_ctf_byte_order ret = BT_CTF_BYTE_ORDER_UNKNOWN;
1036
1037 if (!trace) {
1038 goto end;
1039 }
1040
dc3fffef
PP
1041 ret = trace->native_byte_order;
1042
4ed90fb3
JG
1043end:
1044 return ret;
1045}
1046
391c8f0d 1047int bt_ctf_trace_set_native_byte_order(struct bt_ctf_trace *trace,
bc37ae52
JG
1048 enum bt_ctf_byte_order byte_order)
1049{
1050 int ret = 0;
bc37ae52
JG
1051
1052 if (!trace || trace->frozen) {
1053 ret = -1;
1054 goto end;
1055 }
1056
dc3fffef
PP
1057 if (byte_order != BT_CTF_BYTE_ORDER_LITTLE_ENDIAN &&
1058 byte_order != BT_CTF_BYTE_ORDER_BIG_ENDIAN &&
1059 byte_order != BT_CTF_BYTE_ORDER_NETWORK) {
bc37ae52
JG
1060 ret = -1;
1061 goto end;
1062 }
1063
dc3fffef
PP
1064 trace->native_byte_order = byte_order;
1065
bc37ae52
JG
1066end:
1067 return ret;
1068}
1069
d246b111
JG
1070struct bt_ctf_field_type *bt_ctf_trace_get_packet_header_type(
1071 struct bt_ctf_trace *trace)
1072{
1073 struct bt_ctf_field_type *field_type = NULL;
1074
1075 if (!trace) {
1076 goto end;
1077 }
1078
83509119 1079 bt_get(trace->packet_header_type);
d246b111
JG
1080 field_type = trace->packet_header_type;
1081end:
1082 return field_type;
1083}
1084
1085int bt_ctf_trace_set_packet_header_type(struct bt_ctf_trace *trace,
1086 struct bt_ctf_field_type *packet_header_type)
1087{
1088 int ret = 0;
1089
835b2d10 1090 if (!trace || trace->frozen) {
d246b111
JG
1091 ret = -1;
1092 goto end;
1093 }
1094
835b2d10
JG
1095 /* packet_header_type must be a structure. */
1096 if (packet_header_type &&
1097 bt_ctf_field_type_get_type_id(packet_header_type) !=
1487a16a 1098 BT_CTF_FIELD_TYPE_ID_STRUCT) {
d246b111
JG
1099 ret = -1;
1100 goto end;
1101 }
1102
83509119 1103 bt_put(trace->packet_header_type);
835b2d10 1104 trace->packet_header_type = bt_get(packet_header_type);
d246b111
JG
1105end:
1106 return ret;
1107}
1108
8bf65fbd
JG
1109static
1110int get_stream_class_count(void *element)
1111{
1112 return bt_ctf_trace_get_stream_class_count(
1113 (struct bt_ctf_trace *) element);
1114}
1115
1116static
1117void *get_stream_class(void *element, int i)
1118{
1119 return bt_ctf_trace_get_stream_class(
1120 (struct bt_ctf_trace *) element, i);
1121}
1122
1123static
d9a13d86 1124int visit_stream_class(void *object, bt_ctf_visitor visitor,void *data)
8bf65fbd 1125{
d9a13d86 1126 return bt_ctf_stream_class_visit(object, visitor, data);
8bf65fbd
JG
1127}
1128
1129int bt_ctf_trace_visit(struct bt_ctf_trace *trace,
d9a13d86 1130 bt_ctf_visitor visitor, void *data)
8bf65fbd
JG
1131{
1132 int ret;
d9a13d86
PP
1133 struct bt_ctf_object obj =
1134 { .object = trace, .type = BT_CTF_OBJECT_TYPE_TRACE };
8bf65fbd
JG
1135
1136 if (!trace || !visitor) {
1137 ret = -1;
1138 goto end;
1139 }
1140
d9a13d86 1141 ret = visitor_helper(&obj, get_stream_class_count,
8bf65fbd
JG
1142 get_stream_class, visit_stream_class, visitor, data);
1143end:
1144 return ret;
1145}
1146
1147static
d9a13d86 1148int invoke_listener(struct bt_ctf_object *object, void *data)
8bf65fbd 1149{
9b888ff3 1150 struct listener_wrapper *listener_wrapper = data;
8bf65fbd 1151
d9a13d86 1152 listener_wrapper->listener(object, listener_wrapper->data);
9b888ff3 1153 return 0;
8bf65fbd
JG
1154}
1155
50ad4244
JG
1156int bt_ctf_trace_add_listener(struct bt_ctf_trace *trace,
1157 bt_ctf_listener_cb listener, void *listener_data)
2204c381
JG
1158{
1159 int ret = 0;
50ad4244
JG
1160 struct listener_wrapper *listener_wrapper =
1161 g_new0(struct listener_wrapper, 1);
2204c381 1162
50ad4244 1163 if (!trace || !listener || !listener_wrapper) {
2204c381
JG
1164 ret = -1;
1165 goto error;
1166 }
1167
50ad4244
JG
1168 listener_wrapper->listener = listener;
1169 listener_wrapper->data = listener_data;
8bf65fbd 1170
50ad4244 1171 /* Visit the current schema. */
9b888ff3 1172 ret = bt_ctf_trace_visit(trace, invoke_listener, listener_wrapper);
8bf65fbd
JG
1173 if (ret) {
1174 goto error;
1175 }
1176
50ad4244
JG
1177 /*
1178 * Add listener to the array of callbacks which will be invoked on
1179 * schema changes.
1180 */
1181 g_ptr_array_add(trace->listeners, listener_wrapper);
2204c381
JG
1182 return ret;
1183error:
50ad4244 1184 g_free(listener_wrapper);
2204c381
JG
1185 return ret;
1186}
1187
bc37ae52 1188BT_HIDDEN
d9a13d86 1189int bt_ctf_trace_object_modification(struct bt_ctf_object *object,
9b888ff3
JG
1190 void *trace_ptr)
1191{
1192 size_t i;
1193 struct bt_ctf_trace *trace = trace_ptr;
1194
1195 assert(trace);
d9a13d86 1196 assert(object);
9b888ff3
JG
1197
1198 if (trace->listeners->len == 0) {
1199 goto end;
1200 }
1201
1202 for (i = 0; i < trace->listeners->len; i++) {
1203 struct listener_wrapper *listener =
1204 g_ptr_array_index(trace->listeners, i);
1205
d9a13d86 1206 listener->listener(object, listener->data);
9b888ff3
JG
1207 }
1208end:
1209 return 0;
1210}
1211
1212BT_HIDDEN
bc37ae52
JG
1213struct bt_ctf_field_type *get_field_type(enum field_type_alias alias)
1214{
a25e0d14 1215 int ret;
bc37ae52 1216 unsigned int alignment, size;
8bdcb82e 1217 struct bt_ctf_field_type *field_type = NULL;
bc37ae52
JG
1218
1219 if (alias >= NR_FIELD_TYPE_ALIAS) {
8bdcb82e 1220 goto end;
bc37ae52
JG
1221 }
1222
1223 alignment = field_type_aliases_alignments[alias];
1224 size = field_type_aliases_sizes[alias];
1225 field_type = bt_ctf_field_type_integer_create(size);
a25e0d14
JG
1226 ret = bt_ctf_field_type_set_alignment(field_type, alignment);
1227 if (ret) {
1228 BT_PUT(field_type);
1229 }
8bdcb82e 1230end:
bc37ae52
JG
1231 return field_type;
1232}
1233
f67f3a6e 1234static
09840de5 1235void bt_ctf_trace_freeze(struct bt_ctf_trace *trace)
f67f3a6e 1236{
6474e016
PP
1237 int i;
1238
09840de5 1239 bt_ctf_field_type_freeze(trace->packet_header_type);
f67f3a6e 1240 bt_ctf_attributes_freeze(trace->environment);
6474e016
PP
1241
1242 for (i = 0; i < trace->clocks->len; i++) {
ac0c6bdd 1243 struct bt_ctf_clock_class *clock_class =
6474e016
PP
1244 g_ptr_array_index(trace->clocks, i);
1245
ac0c6bdd 1246 bt_ctf_clock_class_freeze(clock_class);
6474e016
PP
1247 }
1248
f67f3a6e
JG
1249 trace->frozen = 1;
1250}
1251
bc37ae52
JG
1252static
1253int init_trace_packet_header(struct bt_ctf_trace *trace)
1254{
bc37ae52 1255 int ret = 0;
d246b111 1256 struct bt_ctf_field *magic = NULL, *uuid_array = NULL;
bc37ae52
JG
1257 struct bt_ctf_field_type *_uint32_t =
1258 get_field_type(FIELD_TYPE_ALIAS_UINT32_T);
1259 struct bt_ctf_field_type *_uint8_t =
1260 get_field_type(FIELD_TYPE_ALIAS_UINT8_T);
1261 struct bt_ctf_field_type *trace_packet_header_type =
1262 bt_ctf_field_type_structure_create();
1263 struct bt_ctf_field_type *uuid_array_type =
1264 bt_ctf_field_type_array_create(_uint8_t, 16);
1265
1266 if (!trace_packet_header_type || !uuid_array_type) {
1267 ret = -1;
1268 goto end;
1269 }
1270
bc37ae52
JG
1271 ret = bt_ctf_field_type_structure_add_field(trace_packet_header_type,
1272 _uint32_t, "magic");
1273 if (ret) {
1274 goto end;
1275 }
1276
1277 ret = bt_ctf_field_type_structure_add_field(trace_packet_header_type,
1278 uuid_array_type, "uuid");
1279 if (ret) {
1280 goto end;
1281 }
1282
1283 ret = bt_ctf_field_type_structure_add_field(trace_packet_header_type,
1284 _uint32_t, "stream_id");
1285 if (ret) {
1286 goto end;
1287 }
1288
662e778c
JG
1289 ret = bt_ctf_trace_set_packet_header_type(trace,
1290 trace_packet_header_type);
1291 if (ret) {
1292 goto end;
1293 }
bc37ae52 1294end:
83509119
JG
1295 bt_put(uuid_array_type);
1296 bt_put(_uint32_t);
1297 bt_put(_uint8_t);
1298 bt_put(magic);
1299 bt_put(uuid_array);
1300 bt_put(trace_packet_header_type);
bc37ae52
JG
1301 return ret;
1302}
This page took 0.097916 seconds and 4 git commands to generate.