Remove notification-schema.h: not needed for 2.0
[babeltrace.git] / lib / ctf-ir / stream.c
CommitLineData
273b65be
JG
1/*
2 * stream.c
3 *
d2dc44b6 4 * Babeltrace CTF IR - Stream
273b65be 5 *
de9dd397 6 * Copyright 2013, 2014 Jérémie Galarneau <jeremie.galarneau@efficios.com>
273b65be
JG
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
ac0c6bdd
PP
29#include <babeltrace/ctf-ir/clock-class.h>
30#include <babeltrace/ctf-writer/clock.h>
31#include <babeltrace/ctf-writer/clock-internal.h>
273b65be 32#include <babeltrace/ctf-writer/event.h>
adc315b8 33#include <babeltrace/ctf-ir/event-internal.h>
2e33ac5a
PP
34#include <babeltrace/ctf-ir/field-types-internal.h>
35#include <babeltrace/ctf-ir/fields-internal.h>
3f043b05
JG
36#include <babeltrace/ctf-ir/stream.h>
37#include <babeltrace/ctf-ir/stream-internal.h>
adc315b8 38#include <babeltrace/ctf-ir/stream-class-internal.h>
319fd969
PP
39#include <babeltrace/ctf-ir/trace-internal.h>
40#include <babeltrace/ctf-writer/writer-internal.h>
83509119 41#include <babeltrace/ref.h>
273b65be 42#include <babeltrace/ctf-writer/functor-internal.h>
3d9990ac
PP
43#include <babeltrace/compiler-internal.h>
44#include <babeltrace/align-internal.h>
dc3fffef 45#include <inttypes.h>
273b65be
JG
46
47static
83509119 48void bt_ctf_stream_destroy(struct bt_object *obj);
273b65be 49static
af9296f3
JG
50int try_set_structure_field_integer(struct bt_ctf_field *, char *, uint64_t);
51static
273b65be
JG
52int set_structure_field_integer(struct bt_ctf_field *, char *, uint64_t);
53
ac0c6bdd
PP
54static
55int set_integer_field_value(struct bt_ctf_field* field, uint64_t value)
56{
57 int ret = 0;
58 struct bt_ctf_field_type *field_type = NULL;
59
60 if (!field) {
61 ret = -1;
62 goto end;
63 }
64
65 field_type = bt_ctf_field_get_type(field);
66 assert(field_type);
67
68 if (bt_ctf_field_type_get_type_id(field_type) !=
1487a16a 69 BT_CTF_FIELD_TYPE_ID_INTEGER) {
ac0c6bdd
PP
70 /* Not an integer and the value is unset, error. */
71 ret = -1;
72 goto end;
73 }
74
75 if (bt_ctf_field_type_integer_get_signed(field_type)) {
76 ret = bt_ctf_field_signed_integer_set_value(field, (int64_t) value);
77 if (ret) {
78 /* Value is out of range, error. */
79 goto end;
80 }
81 } else {
82 ret = bt_ctf_field_unsigned_integer_set_value(field, value);
83 if (ret) {
84 /* Value is out of range, error. */
85 goto end;
86 }
87 }
88end:
89 bt_put(field_type);
90 return ret;
91}
92
d246b111
JG
93static
94int set_packet_header_magic(struct bt_ctf_stream *stream)
95{
96 int ret = 0;
97 struct bt_ctf_field_type *magic_field_type = NULL;
98 struct bt_ctf_field *magic_field = bt_ctf_field_structure_get_field(
99 stream->packet_header, "magic");
100
101 if (!magic_field) {
102 /* No magic field found. Not an error, skip. */
103 goto end;
104 }
105
fc47320a 106 if (bt_ctf_field_is_set(magic_field)) {
d246b111
JG
107 /* Value already set. Not an error, skip. */
108 goto end;
109 }
110
111 magic_field_type = bt_ctf_field_get_type(magic_field);
112 assert(magic_field_type);
113
114 if (bt_ctf_field_type_get_type_id(magic_field_type) !=
1487a16a 115 BT_CTF_FIELD_TYPE_ID_INTEGER) {
d246b111
JG
116 /* Magic field is not an integer. Not an error, skip. */
117 goto end;
118 }
119
120 if (bt_ctf_field_type_integer_get_size(magic_field_type) != 32) {
121 /*
122 * Magic field is not of the expected size.
123 * Not an error, skip.
124 */
125 goto end;
126 }
127
128 ret = bt_ctf_field_type_integer_get_signed(magic_field_type);
129 assert(ret >= 0);
130 if (ret) {
131 ret = bt_ctf_field_signed_integer_set_value(magic_field,
132 (int64_t) 0xC1FC1FC1);
133 } else {
134 ret = bt_ctf_field_unsigned_integer_set_value(magic_field,
135 (uint64_t) 0xC1FC1FC1);
136 }
137end:
83509119
JG
138 bt_put(magic_field);
139 bt_put(magic_field_type);
d246b111
JG
140 return ret;
141}
142
143static
144int set_packet_header_uuid(struct bt_ctf_stream *stream)
145{
146 int i, ret = 0;
e6a8e8e4 147 struct bt_ctf_trace *trace = NULL;
d246b111
JG
148 struct bt_ctf_field_type *uuid_field_type = NULL;
149 struct bt_ctf_field_type *element_field_type = NULL;
150 struct bt_ctf_field *uuid_field = bt_ctf_field_structure_get_field(
151 stream->packet_header, "uuid");
152
153 if (!uuid_field) {
154 /* No uuid field found. Not an error, skip. */
155 goto end;
156 }
157
fc47320a 158 if (bt_ctf_field_is_set(uuid_field)) {
d246b111
JG
159 /* Value already set. Not an error, skip. */
160 goto end;
161 }
162
163 uuid_field_type = bt_ctf_field_get_type(uuid_field);
164 assert(uuid_field_type);
165 if (bt_ctf_field_type_get_type_id(uuid_field_type) !=
1487a16a 166 BT_CTF_FIELD_TYPE_ID_ARRAY) {
d246b111
JG
167 /* UUID field is not an array. Not an error, skip. */
168 goto end;
169 }
170
171 if (bt_ctf_field_type_array_get_length(uuid_field_type) != 16) {
172 /*
173 * UUID field is not of the expected size.
174 * Not an error, skip.
175 */
176 goto end;
177 }
178
179 element_field_type = bt_ctf_field_type_array_get_element_type(
180 uuid_field_type);
181 assert(element_field_type);
182 if (bt_ctf_field_type_get_type_id(element_field_type) !=
1487a16a 183 BT_CTF_FIELD_TYPE_ID_INTEGER) {
d246b111
JG
184 /* UUID array elements are not integers. Not an error, skip */
185 goto end;
186 }
187
e6a8e8e4 188 trace = (struct bt_ctf_trace *) bt_object_get_parent(stream);
d246b111
JG
189 for (i = 0; i < 16; i++) {
190 struct bt_ctf_field *uuid_element =
191 bt_ctf_field_array_get_field(uuid_field, i);
192
193 ret = bt_ctf_field_type_integer_get_signed(element_field_type);
194 assert(ret >= 0);
195
196 if (ret) {
197 ret = bt_ctf_field_signed_integer_set_value(
e6a8e8e4 198 uuid_element, (int64_t) trace->uuid[i]);
d246b111
JG
199 } else {
200 ret = bt_ctf_field_unsigned_integer_set_value(
201 uuid_element,
e6a8e8e4 202 (uint64_t) trace->uuid[i]);
d246b111 203 }
83509119 204 bt_put(uuid_element);
d246b111
JG
205 if (ret) {
206 goto end;
207 }
208 }
209
210end:
83509119
JG
211 bt_put(uuid_field);
212 bt_put(uuid_field_type);
213 bt_put(element_field_type);
e6a8e8e4 214 BT_PUT(trace);
d246b111
JG
215 return ret;
216}
217static
218int set_packet_header_stream_id(struct bt_ctf_stream *stream)
219{
220 int ret = 0;
221 uint32_t stream_id;
222 struct bt_ctf_field_type *stream_id_field_type = NULL;
223 struct bt_ctf_field *stream_id_field = bt_ctf_field_structure_get_field(
224 stream->packet_header, "stream_id");
225
226 if (!stream_id_field) {
227 /* No stream_id field found. Not an error, skip. */
228 goto end;
229 }
230
fc47320a 231 if (bt_ctf_field_is_set(stream_id_field)) {
d246b111
JG
232 /* Value already set. Not an error, skip. */
233 goto end;
234 }
235
236 stream_id_field_type = bt_ctf_field_get_type(stream_id_field);
237 assert(stream_id_field_type);
238 if (bt_ctf_field_type_get_type_id(stream_id_field_type) !=
1487a16a 239 BT_CTF_FIELD_TYPE_ID_INTEGER) {
d246b111
JG
240 /* stream_id field is not an integer. Not an error, skip. */
241 goto end;
242 }
243
244 stream_id = stream->stream_class->id;
245 ret = bt_ctf_field_type_integer_get_signed(stream_id_field_type);
246 assert(ret >= 0);
247 if (ret) {
248 ret = bt_ctf_field_signed_integer_set_value(stream_id_field,
249 (int64_t) stream_id);
250 } else {
251 ret = bt_ctf_field_unsigned_integer_set_value(stream_id_field,
252 (uint64_t) stream_id);
253 }
254end:
83509119
JG
255 bt_put(stream_id_field);
256 bt_put(stream_id_field_type);
d246b111
JG
257 return ret;
258}
259
260static
261int set_packet_header(struct bt_ctf_stream *stream)
262{
263 int ret;
264
265 ret = set_packet_header_magic(stream);
266 if (ret) {
267 goto end;
268 }
269
270 ret = set_packet_header_uuid(stream);
271 if (ret) {
272 goto end;
273 }
274
275 ret = set_packet_header_stream_id(stream);
276 if (ret) {
277 goto end;
278 }
279end:
280 return ret;
281}
282
123fbdec 283static
e6a8e8e4 284void release_event(struct bt_ctf_event *event)
123fbdec 285{
e6a8e8e4
JG
286 if (bt_object_get_ref_count(event)) {
287 /*
288 * The event is being orphaned, but it must guarantee the
289 * existence of its event class for the duration of its
290 * lifetime.
291 */
292 bt_get(event->event_class);
293 BT_PUT(event->base.parent);
294 } else {
295 bt_object_release(event);
296 }
123fbdec
JG
297}
298
319fd969
PP
299static
300int create_stream_file(struct bt_ctf_writer *writer,
301 struct bt_ctf_stream *stream)
302{
303 int fd;
304 GString *filename = g_string_new(stream->stream_class->name->str);
305
306 if (stream->stream_class->name->len == 0) {
307 int64_t ret;
308
309 ret = bt_ctf_stream_class_get_id(stream->stream_class);
310 if (ret < 0) {
311 fd = -1;
312 goto error;
313 }
314
315 g_string_printf(filename, "stream_%" PRId64, ret);
316 }
317
318 g_string_append_printf(filename, "_%" PRIu32, stream->id);
319 fd = openat(writer->trace_dir_fd, filename->str,
320 O_RDWR | O_CREAT | O_TRUNC,
321 S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP);
322error:
323 g_string_free(filename, TRUE);
324 return fd;
325}
326
327static
328int set_stream_fd(struct bt_ctf_stream *stream, int fd)
329{
330 int ret = 0;
331
332 if (stream->pos.fd != -1) {
333 ret = -1;
334 goto end;
335 }
336
dc3fffef 337 (void) bt_ctf_stream_pos_init(&stream->pos, fd, O_RDWR);
319fd969
PP
338 stream->pos.fd = fd;
339end:
340 return ret;
341}
342
273b65be 343struct bt_ctf_stream *bt_ctf_stream_create(
b71d7298
PP
344 struct bt_ctf_stream_class *stream_class,
345 const char *name)
273b65be 346{
12c8a1a3 347 int ret;
273b65be 348 struct bt_ctf_stream *stream = NULL;
319fd969
PP
349 struct bt_ctf_trace *trace = NULL;
350 struct bt_ctf_writer *writer = NULL;
273b65be 351
319fd969 352 if (!stream_class) {
b71d7298 353 goto error;
319fd969
PP
354 }
355
356 trace = bt_ctf_stream_class_get_trace(stream_class);
357 if (!trace) {
b71d7298 358 goto error;
273b65be
JG
359 }
360
361 stream = g_new0(struct bt_ctf_stream, 1);
362 if (!stream) {
b71d7298 363 goto error;
273b65be
JG
364 }
365
83509119 366 bt_object_init(stream, bt_ctf_stream_destroy);
e6a8e8e4
JG
367 /*
368 * Acquire reference to parent since stream will become publicly
369 * reachable; it needs its parent to remain valid.
370 */
371 bt_object_set_parent(stream, trace);
273b65be
JG
372 stream->id = stream_class->next_stream_id++;
373 stream->stream_class = stream_class;
319fd969 374 stream->pos.fd = -1;
d246b111 375
b71d7298
PP
376 if (name) {
377 stream->name = g_string_new(name);
378 if (!stream->name) {
379 goto error;
380 }
381 }
382
319fd969
PP
383 if (trace->is_created_by_writer) {
384 int fd;
385 writer = (struct bt_ctf_writer *)
386 bt_object_get_parent(trace);
387
388 assert(writer);
98edd02c
JG
389 if (stream_class->packet_context_type) {
390 stream->packet_context = bt_ctf_field_create(
391 stream_class->packet_context_type);
392 if (!stream->packet_context) {
393 goto error;
394 }
319fd969 395
98edd02c 396 /* Initialize events_discarded */
af9296f3
JG
397 ret = try_set_structure_field_integer(
398 stream->packet_context,
98edd02c 399 "events_discarded", 0);
af9296f3 400 if (ret != 1) {
98edd02c
JG
401 goto error;
402 }
319fd969
PP
403 }
404
405 stream->events = g_ptr_array_new_with_free_func(
406 (GDestroyNotify) release_event);
407 if (!stream->events) {
408 goto error;
409 }
410
411 /* A trace is not allowed to have a NULL packet header */
412 assert(trace->packet_header_type);
413 stream->packet_header =
414 bt_ctf_field_create(trace->packet_header_type);
415 if (!stream->packet_header) {
416 goto error;
417 }
418
419 /*
420 * Attempt to populate the default trace packet header fields
421 * (magic, uuid and stream_id). This will _not_ fail shall the
422 * fields not be found or be of an incompatible type; they will
423 * simply not be populated automatically. The user will have to
424 * make sure to set the trace packet header fields himself
425 * before flushing.
426 */
427 ret = set_packet_header(stream);
428 if (ret) {
429 goto error;
430 }
431
432 /* Create file associated with this stream */
433 fd = create_stream_file(writer, stream);
434 if (fd < 0) {
435 goto error;
436 }
437
438 ret = set_stream_fd(stream, fd);
439 if (ret) {
440 goto error;
441 }
442
443 /* Freeze the writer */
444 bt_ctf_writer_freeze(writer);
445 } else {
446 /* Non-writer stream indicated by a negative FD */
447 ret = set_stream_fd(stream, -1);
448 if (ret) {
449 goto error;
450 }
d246b111 451 }
319fd969
PP
452
453 /* Add this stream to the trace's streams */
454 g_ptr_array_add(trace->streams, stream);
455
319fd969
PP
456 BT_PUT(trace);
457 BT_PUT(writer);
273b65be 458 return stream;
83509119
JG
459error:
460 BT_PUT(stream);
319fd969
PP
461 BT_PUT(trace);
462 BT_PUT(writer);
83509119 463 return stream;
273b65be
JG
464}
465
3baf0856
JG
466struct bt_ctf_stream_class *bt_ctf_stream_get_class(
467 struct bt_ctf_stream *stream)
468{
469 struct bt_ctf_stream_class *stream_class = NULL;
470
471 if (!stream) {
472 goto end;
473 }
474
475 stream_class = stream->stream_class;
83509119 476 bt_get(stream_class);
3baf0856
JG
477end:
478 return stream_class;
479}
480
a78a2e25
JG
481int bt_ctf_stream_get_discarded_events_count(
482 struct bt_ctf_stream *stream, uint64_t *count)
483{
484 int64_t ret = 0;
12c8a1a3
JG
485 int field_signed;
486 struct bt_ctf_field *events_discarded_field = NULL;
487 struct bt_ctf_field_type *events_discarded_field_type = NULL;
a78a2e25 488
1579cde5
PP
489 if (!stream || !count || !stream->packet_context ||
490 stream->pos.fd < 0) {
a78a2e25
JG
491 ret = -1;
492 goto end;
493 }
494
12c8a1a3
JG
495 events_discarded_field = bt_ctf_field_structure_get_field(
496 stream->packet_context, "events_discarded");
497 if (!events_discarded_field) {
498 ret = -1;
499 goto end;
500 }
501
502 events_discarded_field_type = bt_ctf_field_get_type(
503 events_discarded_field);
504 if (!events_discarded_field_type) {
505 ret = -1;
506 goto end;
507 }
508
509 field_signed = bt_ctf_field_type_integer_get_signed(
510 events_discarded_field_type);
511 if (field_signed < 0) {
512 ret = field_signed;
513 goto end;
514 }
515
516 if (field_signed) {
517 int64_t signed_count;
518
519 ret = bt_ctf_field_signed_integer_get_value(
520 events_discarded_field, &signed_count);
521 if (ret) {
522 goto end;
523 }
524 if (signed_count < 0) {
525 /* Invalid value */
526 ret = -1;
527 goto end;
528 }
529 *count = (uint64_t) signed_count;
530 } else {
531 ret = bt_ctf_field_unsigned_integer_get_value(
532 events_discarded_field, count);
533 if (ret) {
534 goto end;
535 }
536 }
a78a2e25 537end:
83509119
JG
538 bt_put(events_discarded_field);
539 bt_put(events_discarded_field_type);
a78a2e25
JG
540 return ret;
541}
542
273b65be
JG
543void bt_ctf_stream_append_discarded_events(struct bt_ctf_stream *stream,
544 uint64_t event_count)
545{
12c8a1a3
JG
546 int ret;
547 int field_signed;
548 uint64_t previous_count;
549 uint64_t new_count;
550 struct bt_ctf_field *events_discarded_field = NULL;
551 struct bt_ctf_field_type *events_discarded_field_type = NULL;
552
1579cde5 553 if (!stream || !stream->packet_context || stream->pos.fd < 0) {
12c8a1a3
JG
554 goto end;
555 }
556
557 ret = bt_ctf_stream_get_discarded_events_count(stream,
558 &previous_count);
559 if (ret) {
560 goto end;
561 }
562
563 events_discarded_field = bt_ctf_field_structure_get_field(
564 stream->packet_context, "events_discarded");
565 if (!events_discarded_field) {
566 goto end;
567 }
568
569 events_discarded_field_type = bt_ctf_field_get_type(
570 events_discarded_field);
571 if (!events_discarded_field_type) {
572 goto end;
573 }
574
575 field_signed = bt_ctf_field_type_integer_get_signed(
576 events_discarded_field_type);
577 if (field_signed < 0) {
578 goto end;
273b65be
JG
579 }
580
12c8a1a3
JG
581 new_count = previous_count + event_count;
582 if (field_signed) {
583 ret = bt_ctf_field_signed_integer_set_value(
584 events_discarded_field, (int64_t) new_count);
585 if (ret) {
586 goto end;
587 }
588 } else {
589 ret = bt_ctf_field_unsigned_integer_set_value(
590 events_discarded_field, new_count);
591 if (ret) {
592 goto end;
593 }
594 }
595
596end:
83509119
JG
597 bt_put(events_discarded_field);
598 bt_put(events_discarded_field_type);
273b65be
JG
599}
600
ac0c6bdd
PP
601static int auto_populate_event_header(struct bt_ctf_stream *stream,
602 struct bt_ctf_event *event)
603{
604 int ret = 0;
605 struct bt_ctf_field *id_field = NULL, *timestamp_field = NULL;
606 struct bt_ctf_clock_class *mapped_clock_class = NULL;
607
608 if (!event || event->frozen) {
609 ret = -1;
610 goto end;
611 }
612
613 /*
614 * The condition to automatically set the ID are:
615 *
616 * 1. The event header field "id" exists and is an integer
617 * field.
618 * 2. The event header field "id" is NOT set.
619 */
620 id_field = bt_ctf_field_structure_get_field(event->event_header, "id");
621 if (id_field && !bt_ctf_field_is_set(id_field)) {
622 ret = set_integer_field_value(id_field,
623 (uint64_t) bt_ctf_event_class_get_id(
624 event->event_class));
625 if (ret) {
626 goto end;
627 }
628 }
629
630 /*
631 * The conditions to automatically set the timestamp are:
632 *
633 * 1. The event header field "timestamp" exists and is an
634 * integer field.
635 * 2. This stream's class has a registered clock (set with
636 * bt_ctf_stream_class_set_clock()).
637 * 3. The event header field "timestamp" has its type mapped to
638 * a clock class which is also the clock class of this
639 * stream's class's registered clock.
640 * 4. The event header field "timestamp" is NOT set.
641 */
642 timestamp_field = bt_ctf_field_structure_get_field(event->event_header,
643 "timestamp");
644 if (timestamp_field && !bt_ctf_field_is_set(timestamp_field) &&
645 stream->stream_class->clock) {
646 struct bt_ctf_clock_class *stream_class_clock_class =
647 stream->stream_class->clock->clock_class;
648 struct bt_ctf_field_type *timestamp_field_type =
649 bt_ctf_field_get_type(timestamp_field);
650
651 assert(timestamp_field_type);
652 mapped_clock_class =
653 bt_ctf_field_type_integer_get_mapped_clock_class(
654 timestamp_field_type);
655 BT_PUT(timestamp_field_type);
656 if (mapped_clock_class == stream_class_clock_class) {
657 uint64_t timestamp;
658
659 ret = bt_ctf_clock_get_value(
660 stream->stream_class->clock,
661 &timestamp);
662 if (ret) {
663 goto end;
664 }
665
666 ret = set_integer_field_value(timestamp_field,
667 timestamp);
668 if (ret) {
669 goto end;
670 }
671 }
672 }
673
674end:
675 bt_put(id_field);
676 bt_put(timestamp_field);
677 bt_put(mapped_clock_class);
678 return ret;
679}
680
273b65be
JG
681int bt_ctf_stream_append_event(struct bt_ctf_stream *stream,
682 struct bt_ctf_event *event)
683{
684 int ret = 0;
273b65be 685
1579cde5 686 if (!stream || !event || stream->pos.fd < 0) {
273b65be
JG
687 ret = -1;
688 goto end;
689 }
690
fa29ba83
PP
691 /*
692 * The event is not supposed to have a parent stream at this
693 * point. The only other way an event can have a parent stream
694 * is if it was assigned when setting a packet to the event,
695 * in which case the packet's stream is not a writer stream,
696 * and thus the user is trying to append an event which belongs
697 * to another stream.
698 */
699 if (event->base.parent) {
700 ret = -1;
701 goto end;
702 }
703
e6a8e8e4 704 bt_object_set_parent(event, stream);
ac0c6bdd 705 ret = auto_populate_event_header(stream, event);
662e778c 706 if (ret) {
37f30168 707 goto error;
662e778c
JG
708 }
709
5fd2e9fd 710 /* Make sure the various scopes of the event are set */
273b65be
JG
711 ret = bt_ctf_event_validate(event);
712 if (ret) {
37f30168 713 goto error;
273b65be
JG
714 }
715
0d688c15
PP
716 /* Save the new event and freeze it */
717 bt_ctf_event_freeze(event);
273b65be 718 g_ptr_array_add(stream->events, event);
5fd2e9fd 719
e6a8e8e4
JG
720 /*
721 * Event had to hold a reference to its event class as long as it wasn't
722 * part of the same trace hierarchy. From now on, the event and its
723 * class share the same lifetime guarantees and the reference is no
724 * longer needed.
725 */
726 bt_put(event->event_class);
37f30168 727
273b65be 728end:
37f30168
PP
729 return ret;
730
731error:
732 /*
733 * Orphan the event; we were not successful in associating it to
734 * a stream.
735 */
736 bt_object_set_parent(event, NULL);
737
273b65be
JG
738 return ret;
739}
740
12c8a1a3
JG
741struct bt_ctf_field *bt_ctf_stream_get_packet_context(
742 struct bt_ctf_stream *stream)
743{
744 struct bt_ctf_field *packet_context = NULL;
745
1579cde5 746 if (!stream || stream->pos.fd < 0) {
12c8a1a3
JG
747 goto end;
748 }
749
750 packet_context = stream->packet_context;
12c8a1a3 751 if (packet_context) {
83509119 752 bt_get(packet_context);
12c8a1a3 753 }
34629a55 754end:
12c8a1a3
JG
755 return packet_context;
756}
757
758int bt_ctf_stream_set_packet_context(struct bt_ctf_stream *stream,
759 struct bt_ctf_field *field)
760{
761 int ret = 0;
762 struct bt_ctf_field_type *field_type;
763
835b2d10 764 if (!stream || stream->pos.fd < 0) {
12c8a1a3
JG
765 ret = -1;
766 goto end;
767 }
768
769 field_type = bt_ctf_field_get_type(field);
09840de5
PP
770 if (bt_ctf_field_type_compare(field_type,
771 stream->stream_class->packet_context_type)) {
12c8a1a3
JG
772 ret = -1;
773 goto end;
774 }
775
83509119 776 bt_put(field_type);
83509119 777 bt_put(stream->packet_context);
835b2d10 778 stream->packet_context = bt_get(field);
12c8a1a3
JG
779end:
780 return ret;
781}
782
263a7df5
JG
783struct bt_ctf_field *bt_ctf_stream_get_packet_header(
784 struct bt_ctf_stream *stream)
785{
786 struct bt_ctf_field *packet_header = NULL;
787
1579cde5 788 if (!stream || stream->pos.fd < 0) {
263a7df5
JG
789 goto end;
790 }
791
792 packet_header = stream->packet_header;
793 if (packet_header) {
83509119 794 bt_get(packet_header);
263a7df5
JG
795 }
796end:
797 return packet_header;
798}
799
800int bt_ctf_stream_set_packet_header(struct bt_ctf_stream *stream,
801 struct bt_ctf_field *field)
802{
803 int ret = 0;
e6a8e8e4 804 struct bt_ctf_trace *trace = NULL;
263a7df5
JG
805 struct bt_ctf_field_type *field_type = NULL;
806
835b2d10 807 if (!stream || stream->pos.fd < 0) {
263a7df5
JG
808 ret = -1;
809 goto end;
810 }
811
e6a8e8e4 812 trace = (struct bt_ctf_trace *) bt_object_get_parent(stream);
263a7df5 813 field_type = bt_ctf_field_get_type(field);
09840de5 814 if (bt_ctf_field_type_compare(field_type, trace->packet_header_type)) {
263a7df5
JG
815 ret = -1;
816 goto end;
817 }
818
83509119 819 bt_put(stream->packet_header);
835b2d10 820 stream->packet_header = bt_get(field);
263a7df5 821end:
e6a8e8e4 822 BT_PUT(trace);
83509119 823 bt_put(field_type);
263a7df5
JG
824 return ret;
825}
826
9a220c32
JG
827static
828int get_event_header_timestamp(struct bt_ctf_field *event_header, uint64_t *timestamp)
829{
830 int ret = 0;
831 struct bt_ctf_field *timestamp_field = NULL;
832 struct bt_ctf_field_type *timestamp_field_type = NULL;
833
834 timestamp_field = bt_ctf_field_structure_get_field(event_header,
835 "timestamp");
836 if (!timestamp_field) {
837 ret = -1;
838 goto end;
839 }
840
841 timestamp_field_type = bt_ctf_field_get_type(timestamp_field);
842 assert(timestamp_field_type);
843 if (bt_ctf_field_type_get_type_id(timestamp_field_type) !=
1487a16a 844 BT_CTF_FIELD_TYPE_ID_INTEGER) {
9a220c32
JG
845 ret = -1;
846 goto end;
847 }
848
849 if (bt_ctf_field_type_integer_get_signed(timestamp_field_type)) {
850 int64_t val;
851
852 ret = bt_ctf_field_signed_integer_get_value(timestamp_field,
853 &val);
854 if (ret) {
855 goto end;
856 }
857 *timestamp = (uint64_t) val;
858 } else {
859 ret = bt_ctf_field_unsigned_integer_get_value(timestamp_field,
860 timestamp);
861 if (ret) {
862 goto end;
863 }
864 }
865end:
83509119
JG
866 bt_put(timestamp_field);
867 bt_put(timestamp_field_type);
9a220c32
JG
868 return ret;
869}
870
c9af50d1
JG
871static
872void reset_structure_field(struct bt_ctf_field *structure, const char *name)
873{
874 struct bt_ctf_field *member;
875
876 member = bt_ctf_field_structure_get_field(structure, name);
877 assert(member);
878 (void) bt_ctf_field_reset(member);
e027ff8c 879 bt_put(member);
c9af50d1
JG
880}
881
273b65be
JG
882int bt_ctf_stream_flush(struct bt_ctf_stream *stream)
883{
884 int ret = 0;
885 size_t i;
c9af50d1 886 uint64_t timestamp_begin, timestamp_end;
273b65be 887 struct bt_ctf_field *integer = NULL;
dc3fffef
PP
888 struct bt_ctf_stream_pos packet_context_pos;
889 struct bt_ctf_trace *trace;
890 enum bt_ctf_byte_order native_byte_order;
98edd02c 891 bool empty_packet;
0686ef94 892 uint64_t packet_size_bits;
c9af50d1
JG
893 struct {
894 bool timestamp_begin;
895 bool timestamp_end;
896 bool content_size;
897 bool packet_size;
898 } auto_set_fields = { 0 };
273b65be 899
bc37ae52
JG
900 if (!stream || stream->pos.fd < 0) {
901 /*
902 * Stream does not have an associated fd. It is,
903 * therefore, not a stream being used to write events.
904 */
273b65be
JG
905 ret = -1;
906 goto end;
907 }
908
664f50c8
JG
909 if (!stream->packet_context && stream->flushed_packet_count > 0) {
910 /*
911 * A stream without a packet context, and thus without
912 * content and packet size members, can't have more than
913 * one packet.
914 */
915 ret = -1;
916 goto end;
917 }
918
dc3fffef
PP
919 trace = bt_ctf_stream_class_borrow_trace(stream->stream_class);
920 assert(trace);
921 native_byte_order = bt_ctf_trace_get_byte_order(trace);
98edd02c 922 empty_packet = (stream->events->len == 0);
d246b111 923
28362f2b 924 /* mmap the next packet */
dc3fffef
PP
925 bt_ctf_stream_pos_packet_seek(&stream->pos, 0, SEEK_CUR);
926 ret = bt_ctf_field_serialize(stream->packet_header, &stream->pos,
927 native_byte_order);
d246b111
JG
928 if (ret) {
929 goto end;
273b65be
JG
930 }
931
98edd02c
JG
932 if (stream->packet_context) {
933 /* Set the default context attributes if present and unset. */
934 if (!empty_packet && !get_event_header_timestamp(
935 ((struct bt_ctf_event *) g_ptr_array_index(
936 stream->events, 0))->event_header, &timestamp_begin)) {
af9296f3
JG
937 ret = try_set_structure_field_integer(
938 stream->packet_context,
98edd02c 939 "timestamp_begin", timestamp_begin);
af9296f3 940 if (ret < 0) {
98edd02c
JG
941 goto end;
942 }
c9af50d1 943 auto_set_fields.timestamp_begin = ret == 1;
98edd02c
JG
944 }
945
946 if (!empty_packet && !get_event_header_timestamp(
947 ((struct bt_ctf_event *) g_ptr_array_index(
948 stream->events, stream->events->len - 1))->event_header,
949 &timestamp_end)) {
950
af9296f3
JG
951 ret = try_set_structure_field_integer(
952 stream->packet_context,
98edd02c 953 "timestamp_end", timestamp_end);
af9296f3 954 if (ret < 0) {
98edd02c
JG
955 goto end;
956 }
c9af50d1 957 auto_set_fields.timestamp_end = ret == 1;
98edd02c 958 }
af9296f3 959 ret = try_set_structure_field_integer(stream->packet_context,
98edd02c 960 "content_size", UINT64_MAX);
af9296f3 961 if (ret < 0) {
9a220c32
JG
962 goto end;
963 }
c9af50d1 964 auto_set_fields.content_size = ret == 1;
273b65be 965
af9296f3 966 ret = try_set_structure_field_integer(stream->packet_context,
98edd02c 967 "packet_size", UINT64_MAX);
af9296f3 968 if (ret < 0) {
98edd02c
JG
969 goto end;
970 }
c9af50d1 971 auto_set_fields.packet_size = ret == 1;
12c8a1a3 972
10cb7b41
JG
973 /* Write packet context */
974 memcpy(&packet_context_pos, &stream->pos,
dc3fffef 975 sizeof(packet_context_pos));
10cb7b41 976 ret = bt_ctf_field_serialize(stream->packet_context,
dc3fffef 977 &stream->pos, native_byte_order);
98edd02c
JG
978 if (ret) {
979 goto end;
980 }
12c8a1a3
JG
981 }
982
273b65be
JG
983 for (i = 0; i < stream->events->len; i++) {
984 struct bt_ctf_event *event = g_ptr_array_index(
985 stream->events, i);
273b65be 986
273b65be 987 /* Write event header */
662e778c 988 ret = bt_ctf_field_serialize(event->event_header,
dc3fffef 989 &stream->pos, native_byte_order);
273b65be
JG
990 if (ret) {
991 goto end;
992 }
993
8bfa3f9c 994 /* Write stream event context */
5fd2e9fd 995 if (event->stream_event_context) {
8bfa3f9c 996 ret = bt_ctf_field_serialize(
dc3fffef
PP
997 event->stream_event_context, &stream->pos,
998 native_byte_order);
8bfa3f9c
JG
999 if (ret) {
1000 goto end;
1001 }
1002 }
1003
273b65be 1004 /* Write event content */
dc3fffef
PP
1005 ret = bt_ctf_event_serialize(event, &stream->pos,
1006 native_byte_order);
273b65be
JG
1007 if (ret) {
1008 goto end;
1009 }
1010 }
1011
0686ef94
JG
1012 /* Rounded-up in case content_size is not byte-aligned. */
1013 packet_size_bits = (stream->pos.offset + (CHAR_BIT - 1)) &
1014 ~(CHAR_BIT - 1);
1015 stream->pos.packet_size = packet_size_bits;
1016
98edd02c
JG
1017 if (stream->packet_context) {
1018 /*
1019 * Update the packet total size and content size and overwrite
1020 * the packet context.
1021 * Copy base_mma as the packet may have been remapped (e.g. when
1022 * a packet is resized).
1023 */
1024 packet_context_pos.base_mma = stream->pos.base_mma;
c9af50d1
JG
1025 if (auto_set_fields.content_size) {
1026 ret = set_structure_field_integer(
1027 stream->packet_context,
1028 "content_size", stream->pos.offset);
1029 if (ret < 0) {
1030 goto end;
1031 }
98edd02c 1032 }
273b65be 1033
c9af50d1
JG
1034 if (auto_set_fields.packet_size) {
1035 ret = set_structure_field_integer(stream->packet_context,
1036 "packet_size", packet_size_bits);
1037 if (ret < 0) {
1038 goto end;
1039 }
98edd02c 1040 }
273b65be 1041
98edd02c 1042 ret = bt_ctf_field_serialize(stream->packet_context,
dc3fffef 1043 &packet_context_pos, native_byte_order);
98edd02c
JG
1044 if (ret) {
1045 goto end;
1046 }
273b65be
JG
1047 }
1048
1049 g_ptr_array_set_size(stream->events, 0);
1050 stream->flushed_packet_count++;
0686ef94 1051 stream->size += packet_size_bits / CHAR_BIT;
273b65be 1052end:
c9af50d1
JG
1053 /* Reset automatically-set fields. */
1054 if (auto_set_fields.timestamp_begin) {
1055 reset_structure_field(stream->packet_context,
1056 "timestamp_begin");
1057 }
1058 if (auto_set_fields.timestamp_end) {
1059 reset_structure_field(stream->packet_context,
1060 "timestamp_end");
1061 }
1062 if (auto_set_fields.packet_size) {
1063 reset_structure_field(stream->packet_context,
1064 "packet_size");
1065 }
1066 if (auto_set_fields.content_size) {
1067 reset_structure_field(stream->packet_context,
1068 "content_size");
1069 }
83509119 1070 bt_put(integer);
cd7d8fb7
JG
1071
1072 if (ret < 0) {
1073 /*
1074 * We failed to write the packet. Its size is therefore set to 0
1075 * to ensure the next mapping is done in the same place rather
1076 * than advancing by "stream->pos.packet_size", which would
1077 * leave a corrupted packet in the trace.
1078 */
1079 stream->pos.packet_size = 0;
1080 }
273b65be
JG
1081 return ret;
1082}
1083
1084void bt_ctf_stream_get(struct bt_ctf_stream *stream)
1085{
83509119 1086 bt_get(stream);
273b65be
JG
1087}
1088
1089void bt_ctf_stream_put(struct bt_ctf_stream *stream)
1090{
83509119 1091 bt_put(stream);
273b65be
JG
1092}
1093
1094static
83509119 1095void bt_ctf_stream_destroy(struct bt_object *obj)
273b65be
JG
1096{
1097 struct bt_ctf_stream *stream;
273b65be 1098
83509119 1099 stream = container_of(obj, struct bt_ctf_stream, base);
dc3fffef 1100 (void) bt_ctf_stream_pos_fini(&stream->pos);
0686ef94
JG
1101 if (stream->pos.fd >= 0) {
1102 int ret;
1103
1104 /*
1105 * Truncate the file's size to the minimum required to fit the
1106 * last packet as we might have grown it too much on the last
1107 * mmap.
1108 */
1109 do {
1110 ret = ftruncate(stream->pos.fd, stream->size);
1111 } while (ret == -1 && errno == EINTR);
1112 if (ret) {
1113 perror("ftruncate");
1114 }
1115
1116 if (close(stream->pos.fd)) {
1117 perror("close");
1118 }
9f56e450 1119 }
12c8a1a3 1120
12c8a1a3
JG
1121 if (stream->events) {
1122 g_ptr_array_free(stream->events, TRUE);
1123 }
b71d7298
PP
1124
1125 if (stream->name) {
1126 g_string_free(stream->name, TRUE);
1127 }
41ac640a 1128
83509119
JG
1129 bt_put(stream->packet_header);
1130 bt_put(stream->packet_context);
273b65be
JG
1131 g_free(stream);
1132}
1133
273b65be 1134static
af9296f3
JG
1135int _set_structure_field_integer(struct bt_ctf_field *structure, char *name,
1136 uint64_t value, bool force)
273b65be
JG
1137{
1138 int ret = 0;
d7b1ea66 1139 struct bt_ctf_field_type *field_type = NULL;
273b65be
JG
1140 struct bt_ctf_field *integer =
1141 bt_ctf_field_structure_get_field(structure, name);
12c8a1a3
JG
1142
1143 if (!structure || !name) {
273b65be
JG
1144 ret = -1;
1145 goto end;
1146 }
1147
12c8a1a3
JG
1148 if (!integer) {
1149 /* Field not found, not an error. */
1150 goto end;
1151 }
1152
1153 /* Make sure the payload has not already been set. */
af9296f3 1154 if (!force && bt_ctf_field_is_set(integer)) {
12c8a1a3
JG
1155 /* Payload already set, not an error */
1156 goto end;
1157 }
1158
d7b1ea66 1159 field_type = bt_ctf_field_get_type(integer);
d7b1ea66 1160 assert(field_type);
1487a16a 1161 if (bt_ctf_field_type_get_type_id(field_type) != BT_CTF_FIELD_TYPE_ID_INTEGER) {
d7b1ea66
JG
1162 /*
1163 * The user most likely meant for us to populate this field
1164 * automatically. However, we can only do this if the field
1165 * is an integer. Return an error.
1166 */
1167 ret = -1;
1168 goto end;
1169 }
1170
1171 if (bt_ctf_field_type_integer_get_signed(field_type)) {
1172 ret = bt_ctf_field_signed_integer_set_value(integer,
1173 (int64_t) value);
1174 } else {
1175 ret = bt_ctf_field_unsigned_integer_set_value(integer, value);
1176 }
af9296f3 1177 ret = !ret ? 1 : ret;
273b65be 1178end:
83509119
JG
1179 bt_put(integer);
1180 bt_put(field_type);
273b65be
JG
1181 return ret;
1182}
b71d7298 1183
af9296f3
JG
1184static
1185int set_structure_field_integer(struct bt_ctf_field *structure, char *name,
1186 uint64_t value)
1187{
1188 return _set_structure_field_integer(structure, name, value, true);
1189}
1190
1191/*
1192 * Returns the following codes:
1193 * 1 if the field was found and set,
1194 * 0 if nothing was done (field not found, or was already set),
1195 * <0 if an error was encoutered
1196 */
1197static
1198int try_set_structure_field_integer(struct bt_ctf_field *structure, char *name,
1199 uint64_t value)
1200{
1201 return _set_structure_field_integer(structure, name, value, false);
1202}
1203
b71d7298
PP
1204const char *bt_ctf_stream_get_name(struct bt_ctf_stream *stream)
1205{
1206 const char *name = NULL;
1207
1208 if (!stream) {
1209 goto end;
1210 }
1211
1212 name = stream->name ? stream->name->str : NULL;
1213
1214end:
1215 return name;
1216}
98a4cbef
PP
1217
1218int bt_ctf_stream_is_writer(struct bt_ctf_stream *stream)
1219{
1220 int ret = -1;
1221
1222 if (!stream) {
1223 goto end;
1224 }
1225
1226 ret = (stream->pos.fd >= 0);
1227
1228end:
1229 return ret;
1230}
This page took 0.094018 seconds and 4 git commands to generate.