Writer: introduce try_set_structure_integer_field
[babeltrace.git] / formats / 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
JG
42#include <babeltrace/ctf-writer/functor-internal.h>
43#include <babeltrace/compiler.h>
44#include <babeltrace/align.h>
d246b111 45#include <babeltrace/ctf/ctf-index.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) !=
69 BT_CTF_TYPE_ID_INTEGER) {
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) !=
9a19a512 115 BT_CTF_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) !=
9a19a512 166 BT_CTF_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) !=
9a19a512 183 BT_CTF_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) !=
9a19a512 239 BT_CTF_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
337 ctf_init_pos(&stream->pos, NULL, fd, O_RDWR);
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) !=
9a19a512 844 BT_CTF_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
273b65be
JG
871int bt_ctf_stream_flush(struct bt_ctf_stream *stream)
872{
873 int ret = 0;
874 size_t i;
12c8a1a3 875 uint64_t timestamp_begin, timestamp_end, events_discarded;
273b65be
JG
876 struct bt_ctf_field *integer = NULL;
877 struct ctf_stream_pos packet_context_pos;
98edd02c 878 bool empty_packet;
0686ef94 879 uint64_t packet_size_bits;
273b65be 880
bc37ae52
JG
881 if (!stream || stream->pos.fd < 0) {
882 /*
883 * Stream does not have an associated fd. It is,
884 * therefore, not a stream being used to write events.
885 */
273b65be
JG
886 ret = -1;
887 goto end;
888 }
889
664f50c8
JG
890 if (!stream->packet_context && stream->flushed_packet_count > 0) {
891 /*
892 * A stream without a packet context, and thus without
893 * content and packet size members, can't have more than
894 * one packet.
895 */
896 ret = -1;
897 goto end;
898 }
899
98edd02c 900 empty_packet = (stream->events->len == 0);
d246b111 901
28362f2b
JG
902 /* mmap the next packet */
903 ctf_packet_seek(&stream->pos.parent, 0, SEEK_CUR);
d246b111
JG
904
905 ret = bt_ctf_field_serialize(stream->packet_header, &stream->pos);
906 if (ret) {
907 goto end;
273b65be
JG
908 }
909
98edd02c
JG
910 if (stream->packet_context) {
911 /* Set the default context attributes if present and unset. */
912 if (!empty_packet && !get_event_header_timestamp(
913 ((struct bt_ctf_event *) g_ptr_array_index(
914 stream->events, 0))->event_header, &timestamp_begin)) {
af9296f3
JG
915 ret = try_set_structure_field_integer(
916 stream->packet_context,
98edd02c 917 "timestamp_begin", timestamp_begin);
af9296f3 918 if (ret < 0) {
98edd02c
JG
919 goto end;
920 }
921 }
922
923 if (!empty_packet && !get_event_header_timestamp(
924 ((struct bt_ctf_event *) g_ptr_array_index(
925 stream->events, stream->events->len - 1))->event_header,
926 &timestamp_end)) {
927
af9296f3
JG
928 ret = try_set_structure_field_integer(
929 stream->packet_context,
98edd02c 930 "timestamp_end", timestamp_end);
af9296f3 931 if (ret < 0) {
98edd02c
JG
932 goto end;
933 }
934 }
af9296f3 935 ret = try_set_structure_field_integer(stream->packet_context,
98edd02c 936 "content_size", UINT64_MAX);
af9296f3 937 if (ret < 0) {
9a220c32
JG
938 goto end;
939 }
273b65be 940
af9296f3 941 ret = try_set_structure_field_integer(stream->packet_context,
98edd02c 942 "packet_size", UINT64_MAX);
af9296f3 943 if (ret < 0) {
98edd02c
JG
944 goto end;
945 }
12c8a1a3 946
10cb7b41
JG
947 /* Write packet context */
948 memcpy(&packet_context_pos, &stream->pos,
949 sizeof(struct ctf_stream_pos));
950 ret = bt_ctf_field_serialize(stream->packet_context,
951 &stream->pos);
98edd02c
JG
952 if (ret) {
953 goto end;
954 }
12c8a1a3
JG
955 }
956
273b65be
JG
957 for (i = 0; i < stream->events->len; i++) {
958 struct bt_ctf_event *event = g_ptr_array_index(
959 stream->events, i);
273b65be 960
662e778c 961 ret = bt_ctf_field_reset(event->event_header);
12c8a1a3
JG
962 if (ret) {
963 goto end;
964 }
965
273b65be 966 /* Write event header */
662e778c 967 ret = bt_ctf_field_serialize(event->event_header,
273b65be
JG
968 &stream->pos);
969 if (ret) {
970 goto end;
971 }
972
8bfa3f9c 973 /* Write stream event context */
5fd2e9fd 974 if (event->stream_event_context) {
8bfa3f9c 975 ret = bt_ctf_field_serialize(
5fd2e9fd 976 event->stream_event_context, &stream->pos);
8bfa3f9c
JG
977 if (ret) {
978 goto end;
979 }
980 }
981
273b65be
JG
982 /* Write event content */
983 ret = bt_ctf_event_serialize(event, &stream->pos);
984 if (ret) {
985 goto end;
986 }
987 }
988
0686ef94
JG
989 /* Rounded-up in case content_size is not byte-aligned. */
990 packet_size_bits = (stream->pos.offset + (CHAR_BIT - 1)) &
991 ~(CHAR_BIT - 1);
992 stream->pos.packet_size = packet_size_bits;
993
98edd02c
JG
994 if (stream->packet_context) {
995 /*
996 * Update the packet total size and content size and overwrite
997 * the packet context.
998 * Copy base_mma as the packet may have been remapped (e.g. when
999 * a packet is resized).
1000 */
1001 packet_context_pos.base_mma = stream->pos.base_mma;
1002 ret = set_structure_field_integer(stream->packet_context,
1003 "content_size", stream->pos.offset);
1004 if (ret) {
1005 goto end;
1006 }
273b65be 1007
98edd02c 1008 ret = set_structure_field_integer(stream->packet_context,
0686ef94 1009 "packet_size", packet_size_bits);
98edd02c
JG
1010 if (ret) {
1011 goto end;
1012 }
273b65be 1013
98edd02c
JG
1014 ret = bt_ctf_field_serialize(stream->packet_context,
1015 &packet_context_pos);
1016 if (ret) {
1017 goto end;
1018 }
273b65be
JG
1019 }
1020
1021 g_ptr_array_set_size(stream->events, 0);
1022 stream->flushed_packet_count++;
0686ef94 1023 stream->size += packet_size_bits / CHAR_BIT;
273b65be 1024end:
83509119 1025 bt_put(integer);
273b65be
JG
1026 return ret;
1027}
1028
1029void bt_ctf_stream_get(struct bt_ctf_stream *stream)
1030{
83509119 1031 bt_get(stream);
273b65be
JG
1032}
1033
1034void bt_ctf_stream_put(struct bt_ctf_stream *stream)
1035{
83509119 1036 bt_put(stream);
273b65be
JG
1037}
1038
1039static
83509119 1040void bt_ctf_stream_destroy(struct bt_object *obj)
273b65be
JG
1041{
1042 struct bt_ctf_stream *stream;
273b65be 1043
83509119 1044 stream = container_of(obj, struct bt_ctf_stream, base);
273b65be 1045 ctf_fini_pos(&stream->pos);
0686ef94
JG
1046 if (stream->pos.fd >= 0) {
1047 int ret;
1048
1049 /*
1050 * Truncate the file's size to the minimum required to fit the
1051 * last packet as we might have grown it too much on the last
1052 * mmap.
1053 */
1054 do {
1055 ret = ftruncate(stream->pos.fd, stream->size);
1056 } while (ret == -1 && errno == EINTR);
1057 if (ret) {
1058 perror("ftruncate");
1059 }
1060
1061 if (close(stream->pos.fd)) {
1062 perror("close");
1063 }
9f56e450 1064 }
12c8a1a3 1065
12c8a1a3
JG
1066 if (stream->events) {
1067 g_ptr_array_free(stream->events, TRUE);
1068 }
b71d7298
PP
1069
1070 if (stream->name) {
1071 g_string_free(stream->name, TRUE);
1072 }
41ac640a 1073
83509119
JG
1074 bt_put(stream->packet_header);
1075 bt_put(stream->packet_context);
273b65be
JG
1076 g_free(stream);
1077}
1078
273b65be 1079static
af9296f3
JG
1080int _set_structure_field_integer(struct bt_ctf_field *structure, char *name,
1081 uint64_t value, bool force)
273b65be
JG
1082{
1083 int ret = 0;
d7b1ea66 1084 struct bt_ctf_field_type *field_type = NULL;
273b65be
JG
1085 struct bt_ctf_field *integer =
1086 bt_ctf_field_structure_get_field(structure, name);
12c8a1a3
JG
1087
1088 if (!structure || !name) {
273b65be
JG
1089 ret = -1;
1090 goto end;
1091 }
1092
12c8a1a3
JG
1093 if (!integer) {
1094 /* Field not found, not an error. */
1095 goto end;
1096 }
1097
1098 /* Make sure the payload has not already been set. */
af9296f3 1099 if (!force && bt_ctf_field_is_set(integer)) {
12c8a1a3
JG
1100 /* Payload already set, not an error */
1101 goto end;
1102 }
1103
d7b1ea66 1104 field_type = bt_ctf_field_get_type(integer);
d7b1ea66 1105 assert(field_type);
9a19a512 1106 if (bt_ctf_field_type_get_type_id(field_type) != BT_CTF_TYPE_ID_INTEGER) {
d7b1ea66
JG
1107 /*
1108 * The user most likely meant for us to populate this field
1109 * automatically. However, we can only do this if the field
1110 * is an integer. Return an error.
1111 */
1112 ret = -1;
1113 goto end;
1114 }
1115
1116 if (bt_ctf_field_type_integer_get_signed(field_type)) {
1117 ret = bt_ctf_field_signed_integer_set_value(integer,
1118 (int64_t) value);
1119 } else {
1120 ret = bt_ctf_field_unsigned_integer_set_value(integer, value);
1121 }
af9296f3 1122 ret = !ret ? 1 : ret;
273b65be 1123end:
83509119
JG
1124 bt_put(integer);
1125 bt_put(field_type);
273b65be
JG
1126 return ret;
1127}
b71d7298 1128
af9296f3
JG
1129static
1130int set_structure_field_integer(struct bt_ctf_field *structure, char *name,
1131 uint64_t value)
1132{
1133 return _set_structure_field_integer(structure, name, value, true);
1134}
1135
1136/*
1137 * Returns the following codes:
1138 * 1 if the field was found and set,
1139 * 0 if nothing was done (field not found, or was already set),
1140 * <0 if an error was encoutered
1141 */
1142static
1143int try_set_structure_field_integer(struct bt_ctf_field *structure, char *name,
1144 uint64_t value)
1145{
1146 return _set_structure_field_integer(structure, name, value, false);
1147}
1148
b71d7298
PP
1149const char *bt_ctf_stream_get_name(struct bt_ctf_stream *stream)
1150{
1151 const char *name = NULL;
1152
1153 if (!stream) {
1154 goto end;
1155 }
1156
1157 name = stream->name ? stream->name->str : NULL;
1158
1159end:
1160 return name;
1161}
98a4cbef
PP
1162
1163int bt_ctf_stream_is_writer(struct bt_ctf_stream *stream)
1164{
1165 int ret = -1;
1166
1167 if (!stream) {
1168 goto end;
1169 }
1170
1171 ret = (stream->pos.fd >= 0);
1172
1173end:
1174 return ret;
1175}
This page took 0.086985 seconds and 4 git commands to generate.