Cleanup: Use a switch case instead of conditionals
[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
3f043b05 29#include <babeltrace/ctf-ir/clock.h>
adc315b8 30#include <babeltrace/ctf-ir/clock-internal.h>
273b65be 31#include <babeltrace/ctf-writer/event.h>
adc315b8
JG
32#include <babeltrace/ctf-ir/event-internal.h>
33#include <babeltrace/ctf-ir/event-types-internal.h>
34#include <babeltrace/ctf-ir/event-fields-internal.h>
3f043b05
JG
35#include <babeltrace/ctf-ir/stream.h>
36#include <babeltrace/ctf-ir/stream-internal.h>
adc315b8 37#include <babeltrace/ctf-ir/stream-class-internal.h>
273b65be
JG
38#include <babeltrace/ctf-writer/functor-internal.h>
39#include <babeltrace/compiler.h>
40#include <babeltrace/align.h>
d246b111 41#include <babeltrace/ctf/ctf-index.h>
273b65be
JG
42
43static
44void bt_ctf_stream_destroy(struct bt_ctf_ref *ref);
45static
273b65be
JG
46int set_structure_field_integer(struct bt_ctf_field *, char *, uint64_t);
47
d246b111
JG
48static
49int set_packet_header_magic(struct bt_ctf_stream *stream)
50{
51 int ret = 0;
52 struct bt_ctf_field_type *magic_field_type = NULL;
53 struct bt_ctf_field *magic_field = bt_ctf_field_structure_get_field(
54 stream->packet_header, "magic");
55
56 if (!magic_field) {
57 /* No magic field found. Not an error, skip. */
58 goto end;
59 }
60
61 if (!bt_ctf_field_validate(magic_field)) {
62 /* Value already set. Not an error, skip. */
63 goto end;
64 }
65
66 magic_field_type = bt_ctf_field_get_type(magic_field);
67 assert(magic_field_type);
68
69 if (bt_ctf_field_type_get_type_id(magic_field_type) !=
70 CTF_TYPE_INTEGER) {
71 /* Magic field is not an integer. Not an error, skip. */
72 goto end;
73 }
74
75 if (bt_ctf_field_type_integer_get_size(magic_field_type) != 32) {
76 /*
77 * Magic field is not of the expected size.
78 * Not an error, skip.
79 */
80 goto end;
81 }
82
83 ret = bt_ctf_field_type_integer_get_signed(magic_field_type);
84 assert(ret >= 0);
85 if (ret) {
86 ret = bt_ctf_field_signed_integer_set_value(magic_field,
87 (int64_t) 0xC1FC1FC1);
88 } else {
89 ret = bt_ctf_field_unsigned_integer_set_value(magic_field,
90 (uint64_t) 0xC1FC1FC1);
91 }
92end:
93 if (magic_field) {
94 bt_ctf_field_put(magic_field);
95 }
96 if (magic_field_type) {
97 bt_ctf_field_type_put(magic_field_type);
98 }
99 return ret;
100}
101
102static
103int set_packet_header_uuid(struct bt_ctf_stream *stream)
104{
105 int i, ret = 0;
106 struct bt_ctf_field_type *uuid_field_type = NULL;
107 struct bt_ctf_field_type *element_field_type = NULL;
108 struct bt_ctf_field *uuid_field = bt_ctf_field_structure_get_field(
109 stream->packet_header, "uuid");
110
111 if (!uuid_field) {
112 /* No uuid field found. Not an error, skip. */
113 goto end;
114 }
115
116 if (!bt_ctf_field_validate(uuid_field)) {
117 /* Value already set. Not an error, skip. */
118 goto end;
119 }
120
121 uuid_field_type = bt_ctf_field_get_type(uuid_field);
122 assert(uuid_field_type);
123 if (bt_ctf_field_type_get_type_id(uuid_field_type) !=
124 CTF_TYPE_ARRAY) {
125 /* UUID field is not an array. Not an error, skip. */
126 goto end;
127 }
128
129 if (bt_ctf_field_type_array_get_length(uuid_field_type) != 16) {
130 /*
131 * UUID field is not of the expected size.
132 * Not an error, skip.
133 */
134 goto end;
135 }
136
137 element_field_type = bt_ctf_field_type_array_get_element_type(
138 uuid_field_type);
139 assert(element_field_type);
140 if (bt_ctf_field_type_get_type_id(element_field_type) !=
141 CTF_TYPE_INTEGER) {
142 /* UUID array elements are not integers. Not an error, skip */
143 goto end;
144 }
145
146 for (i = 0; i < 16; i++) {
147 struct bt_ctf_field *uuid_element =
148 bt_ctf_field_array_get_field(uuid_field, i);
149
150 ret = bt_ctf_field_type_integer_get_signed(element_field_type);
151 assert(ret >= 0);
152
153 if (ret) {
154 ret = bt_ctf_field_signed_integer_set_value(
155 uuid_element, (int64_t) stream->trace->uuid[i]);
156 } else {
157 ret = bt_ctf_field_unsigned_integer_set_value(
158 uuid_element,
159 (uint64_t) stream->trace->uuid[i]);
160 }
161 bt_ctf_field_put(uuid_element);
162 if (ret) {
163 goto end;
164 }
165 }
166
167end:
168 if (uuid_field) {
169 bt_ctf_field_put(uuid_field);
170 }
171 if (uuid_field_type) {
172 bt_ctf_field_type_put(uuid_field_type);
173 }
174 if (element_field_type) {
175 bt_ctf_field_type_put(element_field_type);
176 }
177 return ret;
178}
179static
180int set_packet_header_stream_id(struct bt_ctf_stream *stream)
181{
182 int ret = 0;
183 uint32_t stream_id;
184 struct bt_ctf_field_type *stream_id_field_type = NULL;
185 struct bt_ctf_field *stream_id_field = bt_ctf_field_structure_get_field(
186 stream->packet_header, "stream_id");
187
188 if (!stream_id_field) {
189 /* No stream_id field found. Not an error, skip. */
190 goto end;
191 }
192
193 if (!bt_ctf_field_validate(stream_id_field)) {
194 /* Value already set. Not an error, skip. */
195 goto end;
196 }
197
198 stream_id_field_type = bt_ctf_field_get_type(stream_id_field);
199 assert(stream_id_field_type);
200 if (bt_ctf_field_type_get_type_id(stream_id_field_type) !=
201 CTF_TYPE_INTEGER) {
202 /* stream_id field is not an integer. Not an error, skip. */
203 goto end;
204 }
205
206 stream_id = stream->stream_class->id;
207 ret = bt_ctf_field_type_integer_get_signed(stream_id_field_type);
208 assert(ret >= 0);
209 if (ret) {
210 ret = bt_ctf_field_signed_integer_set_value(stream_id_field,
211 (int64_t) stream_id);
212 } else {
213 ret = bt_ctf_field_unsigned_integer_set_value(stream_id_field,
214 (uint64_t) stream_id);
215 }
216end:
217 if (stream_id_field) {
218 bt_ctf_field_put(stream_id_field);
219 }
220 if (stream_id_field_type) {
221 bt_ctf_field_type_put(stream_id_field_type);
222 }
223 return ret;
224}
225
226static
227int set_packet_header(struct bt_ctf_stream *stream)
228{
229 int ret;
230
231 ret = set_packet_header_magic(stream);
232 if (ret) {
233 goto end;
234 }
235
236 ret = set_packet_header_uuid(stream);
237 if (ret) {
238 goto end;
239 }
240
241 ret = set_packet_header_stream_id(stream);
242 if (ret) {
243 goto end;
244 }
245end:
246 return ret;
247}
248
273b65be
JG
249BT_HIDDEN
250struct bt_ctf_stream *bt_ctf_stream_create(
d246b111
JG
251 struct bt_ctf_stream_class *stream_class,
252 struct bt_ctf_trace *trace)
273b65be 253{
12c8a1a3 254 int ret;
273b65be
JG
255 struct bt_ctf_stream *stream = NULL;
256
d246b111 257 if (!stream_class || !trace) {
273b65be
JG
258 goto end;
259 }
260
261 stream = g_new0(struct bt_ctf_stream, 1);
262 if (!stream) {
263 goto end;
264 }
265
d246b111 266 stream->trace = trace;
273b65be 267 bt_ctf_ref_init(&stream->ref_count);
12c8a1a3
JG
268 stream->packet_context = bt_ctf_field_create(
269 stream_class->packet_context_type);
270 if (!stream->packet_context) {
271 goto error_destroy;
272 }
273
af181248
JG
274 /*
275 * A stream class may not have a stream event context defined
276 * in which case this stream will never have a stream_event_context
277 * member since, after a stream's creation, the parent stream class
278 * is "frozen" (immutable).
279 */
280 if (stream_class->event_context_type) {
281 stream->event_context = bt_ctf_field_create(
282 stream_class->event_context_type);
283 if (!stream->packet_context) {
284 goto error_destroy;
285 }
286 }
287
e19cc57d 288 /* Initialize events_discarded */
12c8a1a3
JG
289 ret = set_structure_field_integer(stream->packet_context,
290 "events_discarded", 0);
291 if (ret) {
292 goto error_destroy;
293 }
294
273b65be
JG
295 stream->pos.fd = -1;
296 stream->id = stream_class->next_stream_id++;
297 stream->stream_class = stream_class;
298 bt_ctf_stream_class_get(stream_class);
273b65be 299 stream->events = g_ptr_array_new_with_free_func(
8bfa3f9c
JG
300 (GDestroyNotify) bt_ctf_event_put);
301 if (!stream->events) {
302 goto error_destroy;
303 }
304 if (stream_class->event_context_type) {
305 stream->event_contexts = g_ptr_array_new_with_free_func(
306 (GDestroyNotify) bt_ctf_field_put);
307 if (!stream->event_contexts) {
308 goto error_destroy;
309 }
310 }
d246b111
JG
311
312 /* A trace is not allowed to have a NULL packet header */
313 assert(trace->packet_header_type);
314 stream->packet_header = bt_ctf_field_create(trace->packet_header_type);
315 /*
316 * Attempt to populate the default trace packet header fields
317 * (magic, uuid and stream_id). This will _not_ fail shall the
318 * fields not be found or be of an incompatible type; they will
319 * simply not be populated automatically. The user will have to
320 * make sure to set the trace packet header fields himself before
321 * flushing.
322 */
323 ret = set_packet_header(stream);
324 if (ret) {
325 goto error_destroy;
326 }
273b65be
JG
327end:
328 return stream;
12c8a1a3
JG
329error_destroy:
330 bt_ctf_stream_destroy(&stream->ref_count);
331 return NULL;
273b65be
JG
332}
333
273b65be
JG
334BT_HIDDEN
335int bt_ctf_stream_set_fd(struct bt_ctf_stream *stream, int fd)
336{
337 int ret = 0;
338
339 if (stream->pos.fd != -1) {
340 ret = -1;
341 goto end;
342 }
343
344 ctf_init_pos(&stream->pos, NULL, fd, O_RDWR);
345 stream->pos.fd = fd;
346end:
347 return ret;
348}
349
d246b111
JG
350BT_HIDDEN
351void bt_ctf_stream_set_trace(struct bt_ctf_stream *stream,
352 struct bt_ctf_trace *trace)
353{
354 stream->trace = trace;
355}
356
3baf0856
JG
357struct bt_ctf_stream_class *bt_ctf_stream_get_class(
358 struct bt_ctf_stream *stream)
359{
360 struct bt_ctf_stream_class *stream_class = NULL;
361
362 if (!stream) {
363 goto end;
364 }
365
366 stream_class = stream->stream_class;
367 bt_ctf_stream_class_get(stream_class);
368end:
369 return stream_class;
370}
371
a78a2e25
JG
372int bt_ctf_stream_get_discarded_events_count(
373 struct bt_ctf_stream *stream, uint64_t *count)
374{
375 int64_t ret = 0;
12c8a1a3
JG
376 int field_signed;
377 struct bt_ctf_field *events_discarded_field = NULL;
378 struct bt_ctf_field_type *events_discarded_field_type = NULL;
a78a2e25 379
12c8a1a3 380 if (!stream || !count || !stream->packet_context) {
a78a2e25
JG
381 ret = -1;
382 goto end;
383 }
384
12c8a1a3
JG
385 events_discarded_field = bt_ctf_field_structure_get_field(
386 stream->packet_context, "events_discarded");
387 if (!events_discarded_field) {
388 ret = -1;
389 goto end;
390 }
391
392 events_discarded_field_type = bt_ctf_field_get_type(
393 events_discarded_field);
394 if (!events_discarded_field_type) {
395 ret = -1;
396 goto end;
397 }
398
399 field_signed = bt_ctf_field_type_integer_get_signed(
400 events_discarded_field_type);
401 if (field_signed < 0) {
402 ret = field_signed;
403 goto end;
404 }
405
406 if (field_signed) {
407 int64_t signed_count;
408
409 ret = bt_ctf_field_signed_integer_get_value(
410 events_discarded_field, &signed_count);
411 if (ret) {
412 goto end;
413 }
414 if (signed_count < 0) {
415 /* Invalid value */
416 ret = -1;
417 goto end;
418 }
419 *count = (uint64_t) signed_count;
420 } else {
421 ret = bt_ctf_field_unsigned_integer_get_value(
422 events_discarded_field, count);
423 if (ret) {
424 goto end;
425 }
426 }
a78a2e25 427end:
12c8a1a3
JG
428 if (events_discarded_field) {
429 bt_ctf_field_put(events_discarded_field);
430 }
431 if (events_discarded_field_type) {
432 bt_ctf_field_type_put(events_discarded_field_type);
433 }
a78a2e25
JG
434 return ret;
435}
436
273b65be
JG
437void bt_ctf_stream_append_discarded_events(struct bt_ctf_stream *stream,
438 uint64_t event_count)
439{
12c8a1a3
JG
440 int ret;
441 int field_signed;
442 uint64_t previous_count;
443 uint64_t new_count;
444 struct bt_ctf_field *events_discarded_field = NULL;
445 struct bt_ctf_field_type *events_discarded_field_type = NULL;
446
447 if (!stream || !stream->packet_context) {
448 goto end;
449 }
450
451 ret = bt_ctf_stream_get_discarded_events_count(stream,
452 &previous_count);
453 if (ret) {
454 goto end;
455 }
456
457 events_discarded_field = bt_ctf_field_structure_get_field(
458 stream->packet_context, "events_discarded");
459 if (!events_discarded_field) {
460 goto end;
461 }
462
463 events_discarded_field_type = bt_ctf_field_get_type(
464 events_discarded_field);
465 if (!events_discarded_field_type) {
466 goto end;
467 }
468
469 field_signed = bt_ctf_field_type_integer_get_signed(
470 events_discarded_field_type);
471 if (field_signed < 0) {
472 goto end;
273b65be
JG
473 }
474
12c8a1a3
JG
475 new_count = previous_count + event_count;
476 if (field_signed) {
477 ret = bt_ctf_field_signed_integer_set_value(
478 events_discarded_field, (int64_t) new_count);
479 if (ret) {
480 goto end;
481 }
482 } else {
483 ret = bt_ctf_field_unsigned_integer_set_value(
484 events_discarded_field, new_count);
485 if (ret) {
486 goto end;
487 }
488 }
489
490end:
491 if (events_discarded_field) {
492 bt_ctf_field_put(events_discarded_field);
493 }
494 if (events_discarded_field_type) {
495 bt_ctf_field_type_put(events_discarded_field_type);
496 }
273b65be
JG
497}
498
499int bt_ctf_stream_append_event(struct bt_ctf_stream *stream,
500 struct bt_ctf_event *event)
501{
502 int ret = 0;
503 uint64_t timestamp;
af181248 504 struct bt_ctf_field *event_context_copy = NULL;
273b65be
JG
505
506 if (!stream || !event) {
507 ret = -1;
508 goto end;
509 }
510
785b5390 511 /* Make sure the event's payload is set */
273b65be
JG
512 ret = bt_ctf_event_validate(event);
513 if (ret) {
514 goto end;
515 }
516
8bfa3f9c
JG
517 /* Sample the current stream event context by copying it */
518 if (stream->event_context) {
519 /* Make sure the event context's payload is set */
520 ret = bt_ctf_field_validate(stream->event_context);
521 if (ret) {
522 goto end;
523 }
524
525 event_context_copy = bt_ctf_field_copy(stream->event_context);
526 if (!event_context_copy) {
527 ret = -1;
528 goto end;
529 }
530 }
531
273b65be
JG
532 timestamp = bt_ctf_clock_get_time(stream->stream_class->clock);
533 ret = bt_ctf_event_set_timestamp(event, timestamp);
534 if (ret) {
535 goto end;
536 }
537
538 bt_ctf_event_get(event);
8bfa3f9c 539 /* Save the new event along with its associated stream event context */
273b65be 540 g_ptr_array_add(stream->events, event);
8bfa3f9c
JG
541 if (event_context_copy) {
542 g_ptr_array_add(stream->event_contexts, event_context_copy);
543 }
273b65be
JG
544end:
545 return ret;
546}
547
12c8a1a3
JG
548struct bt_ctf_field *bt_ctf_stream_get_packet_context(
549 struct bt_ctf_stream *stream)
550{
551 struct bt_ctf_field *packet_context = NULL;
552
553 if (!stream) {
554 goto end;
555 }
556
557 packet_context = stream->packet_context;
12c8a1a3
JG
558 if (packet_context) {
559 bt_ctf_field_get(packet_context);
560 }
34629a55 561end:
12c8a1a3
JG
562 return packet_context;
563}
564
565int bt_ctf_stream_set_packet_context(struct bt_ctf_stream *stream,
566 struct bt_ctf_field *field)
567{
568 int ret = 0;
569 struct bt_ctf_field_type *field_type;
570
571 if (!stream || !field) {
572 ret = -1;
573 goto end;
574 }
575
576 field_type = bt_ctf_field_get_type(field);
577 if (field_type != stream->stream_class->packet_context_type) {
578 ret = -1;
579 goto end;
580 }
581
582 bt_ctf_field_type_put(field_type);
583 bt_ctf_field_get(field);
584 bt_ctf_field_put(stream->packet_context);
585 stream->packet_context = field;
586end:
587 return ret;
588}
589
8bfa3f9c
JG
590struct bt_ctf_field *bt_ctf_stream_get_event_context(
591 struct bt_ctf_stream *stream)
592{
593 struct bt_ctf_field *event_context = NULL;
594
595 if (!stream) {
596 goto end;
597 }
598
599 event_context = stream->event_context;
600 if (event_context) {
601 bt_ctf_field_get(event_context);
602 }
603end:
604 return event_context;
605}
606
607int bt_ctf_stream_set_event_context(struct bt_ctf_stream *stream,
608 struct bt_ctf_field *field)
609{
610 int ret = 0;
611 struct bt_ctf_field_type *field_type = NULL;
612
613 if (!stream || !field) {
614 ret = -1;
615 goto end;
616 }
617
618 field_type = bt_ctf_field_get_type(field);
619 if (field_type != stream->stream_class->event_context_type) {
620 ret = -1;
621 goto end;
622 }
623
624 bt_ctf_field_get(field);
625 bt_ctf_field_put(stream->event_context);
626 stream->event_context = field;
627end:
628 if (field_type) {
629 bt_ctf_field_type_put(field_type);
630 }
631 return ret;
632}
633
263a7df5
JG
634struct bt_ctf_field *bt_ctf_stream_get_packet_header(
635 struct bt_ctf_stream *stream)
636{
637 struct bt_ctf_field *packet_header = NULL;
638
639 if (!stream) {
640 goto end;
641 }
642
643 packet_header = stream->packet_header;
644 if (packet_header) {
645 bt_ctf_field_get(packet_header);
646 }
647end:
648 return packet_header;
649}
650
651int bt_ctf_stream_set_packet_header(struct bt_ctf_stream *stream,
652 struct bt_ctf_field *field)
653{
654 int ret = 0;
655 struct bt_ctf_field_type *field_type = NULL;
656
657 if (!stream || !field) {
658 ret = -1;
659 goto end;
660 }
661
662 field_type = bt_ctf_field_get_type(field);
663 if (field_type != stream->trace->packet_header_type) {
664 ret = -1;
665 goto end;
666 }
667
668 bt_ctf_field_get(field);
669 bt_ctf_field_put(stream->packet_header);
670 stream->packet_header = field;
671end:
672 if (field_type) {
673 bt_ctf_field_type_put(field_type);
674 }
675 return ret;
676}
677
273b65be
JG
678int bt_ctf_stream_flush(struct bt_ctf_stream *stream)
679{
680 int ret = 0;
681 size_t i;
12c8a1a3 682 uint64_t timestamp_begin, timestamp_end, events_discarded;
de876b7f 683 struct bt_ctf_stream_class *stream_class;
273b65be
JG
684 struct bt_ctf_field *integer = NULL;
685 struct ctf_stream_pos packet_context_pos;
686
bc37ae52
JG
687 if (!stream || stream->pos.fd < 0) {
688 /*
689 * Stream does not have an associated fd. It is,
690 * therefore, not a stream being used to write events.
691 */
273b65be
JG
692 ret = -1;
693 goto end;
694 }
695
696 if (!stream->events->len) {
697 goto end;
698 }
699
d246b111
JG
700 ret = bt_ctf_field_validate(stream->packet_header);
701 if (ret) {
702 goto end;
703 }
704
28362f2b
JG
705 /* mmap the next packet */
706 ctf_packet_seek(&stream->pos.parent, 0, SEEK_CUR);
d246b111
JG
707
708 ret = bt_ctf_field_serialize(stream->packet_header, &stream->pos);
709 if (ret) {
710 goto end;
273b65be
JG
711 }
712
de876b7f 713 stream_class = stream->stream_class;
273b65be
JG
714 timestamp_begin = ((struct bt_ctf_event *) g_ptr_array_index(
715 stream->events, 0))->timestamp;
716 timestamp_end = ((struct bt_ctf_event *) g_ptr_array_index(
717 stream->events, stream->events->len - 1))->timestamp;
12c8a1a3
JG
718
719 /* Set the default context attributes if present and unset. */
720 ret = set_structure_field_integer(stream->packet_context,
273b65be
JG
721 "timestamp_begin", timestamp_begin);
722 if (ret) {
723 goto end;
724 }
725
12c8a1a3 726 ret = set_structure_field_integer(stream->packet_context,
273b65be
JG
727 "timestamp_end", timestamp_end);
728 if (ret) {
729 goto end;
730 }
731
12c8a1a3 732 ret = set_structure_field_integer(stream->packet_context,
273b65be
JG
733 "content_size", UINT64_MAX);
734 if (ret) {
735 goto end;
736 }
737
12c8a1a3 738 ret = set_structure_field_integer(stream->packet_context,
273b65be
JG
739 "packet_size", UINT64_MAX);
740 if (ret) {
741 goto end;
742 }
743
744 /* Write packet context */
745 memcpy(&packet_context_pos, &stream->pos,
746 sizeof(struct ctf_stream_pos));
12c8a1a3 747 ret = bt_ctf_field_serialize(stream->packet_context,
273b65be
JG
748 &stream->pos);
749 if (ret) {
750 goto end;
751 }
752
12c8a1a3
JG
753 ret = bt_ctf_stream_get_discarded_events_count(stream,
754 &events_discarded);
755 if (ret) {
756 goto end;
757 }
758
759 /* Unset the packet context's fields. */
760 ret = bt_ctf_field_reset(stream->packet_context);
761 if (ret) {
762 goto end;
763 }
764
765 /* Set the previous number of discarded events. */
766 ret = set_structure_field_integer(stream->packet_context,
767 "events_discarded", events_discarded);
768 if (ret) {
769 goto end;
770 }
771
273b65be
JG
772 for (i = 0; i < stream->events->len; i++) {
773 struct bt_ctf_event *event = g_ptr_array_index(
774 stream->events, i);
775 uint32_t event_id = bt_ctf_event_class_get_id(
776 event->event_class);
777 uint64_t timestamp = bt_ctf_event_get_timestamp(event);
778
de876b7f 779 ret = bt_ctf_field_reset(stream_class->event_header);
12c8a1a3
JG
780 if (ret) {
781 goto end;
782 }
783
de876b7f 784 ret = set_structure_field_integer(stream_class->event_header,
273b65be
JG
785 "id", event_id);
786 if (ret) {
787 goto end;
788 }
de876b7f 789 ret = set_structure_field_integer(stream_class->event_header,
273b65be
JG
790 "timestamp", timestamp);
791 if (ret) {
792 goto end;
793 }
794
795 /* Write event header */
de876b7f 796 ret = bt_ctf_field_serialize(stream_class->event_header,
273b65be
JG
797 &stream->pos);
798 if (ret) {
799 goto end;
800 }
801
8bfa3f9c
JG
802 /* Write stream event context */
803 if (stream->event_contexts) {
804 ret = bt_ctf_field_serialize(
805 g_ptr_array_index(stream->event_contexts, i),
806 &stream->pos);
807 if (ret) {
808 goto end;
809 }
810 }
811
273b65be
JG
812 /* Write event content */
813 ret = bt_ctf_event_serialize(event, &stream->pos);
814 if (ret) {
815 goto end;
816 }
817 }
818
819 /*
820 * Update the packet total size and content size and overwrite the
821 * packet context.
3a092c05
JG
822 * Copy base_mma as the packet may have been remapped (e.g. when a
823 * packet is resized).
273b65be 824 */
3a092c05 825 packet_context_pos.base_mma = stream->pos.base_mma;
12c8a1a3 826 ret = set_structure_field_integer(stream->packet_context,
273b65be
JG
827 "content_size", stream->pos.offset);
828 if (ret) {
829 goto end;
830 }
831
12c8a1a3 832 ret = set_structure_field_integer(stream->packet_context,
273b65be
JG
833 "packet_size", stream->pos.packet_size);
834 if (ret) {
835 goto end;
836 }
837
12c8a1a3 838 ret = bt_ctf_field_serialize(stream->packet_context,
273b65be
JG
839 &packet_context_pos);
840 if (ret) {
841 goto end;
842 }
843
844 g_ptr_array_set_size(stream->events, 0);
8bfa3f9c
JG
845 if (stream->event_contexts) {
846 g_ptr_array_set_size(stream->event_contexts, 0);
847 }
273b65be
JG
848 stream->flushed_packet_count++;
849end:
850 bt_ctf_field_put(integer);
851 return ret;
852}
853
854void bt_ctf_stream_get(struct bt_ctf_stream *stream)
855{
856 if (!stream) {
857 return;
858 }
859
860 bt_ctf_ref_get(&stream->ref_count);
861}
862
863void bt_ctf_stream_put(struct bt_ctf_stream *stream)
864{
865 if (!stream) {
866 return;
867 }
868
869 bt_ctf_ref_put(&stream->ref_count, bt_ctf_stream_destroy);
870}
871
872static
873void bt_ctf_stream_destroy(struct bt_ctf_ref *ref)
874{
875 struct bt_ctf_stream *stream;
876
877 if (!ref) {
878 return;
879 }
880
881 stream = container_of(ref, struct bt_ctf_stream, ref_count);
882 ctf_fini_pos(&stream->pos);
9f56e450
JG
883 if (close(stream->pos.fd)) {
884 perror("close");
885 }
12c8a1a3
JG
886
887 if (stream->stream_class) {
888 bt_ctf_stream_class_put(stream->stream_class);
889 }
890 if (stream->events) {
891 g_ptr_array_free(stream->events, TRUE);
892 }
8bfa3f9c
JG
893 if (stream->event_contexts) {
894 g_ptr_array_free(stream->event_contexts, TRUE);
895 }
d246b111
JG
896 if (stream->packet_header) {
897 bt_ctf_field_put(stream->packet_header);
898 }
12c8a1a3
JG
899 if (stream->packet_context) {
900 bt_ctf_field_put(stream->packet_context);
901 }
8bfa3f9c
JG
902 if (stream->event_context) {
903 bt_ctf_field_put(stream->event_context);
904 }
273b65be
JG
905 g_free(stream);
906}
907
273b65be
JG
908static
909int set_structure_field_integer(struct bt_ctf_field *structure, char *name,
910 uint64_t value)
911{
912 int ret = 0;
d7b1ea66 913 struct bt_ctf_field_type *field_type = NULL;
273b65be
JG
914 struct bt_ctf_field *integer =
915 bt_ctf_field_structure_get_field(structure, name);
12c8a1a3
JG
916
917 if (!structure || !name) {
273b65be
JG
918 ret = -1;
919 goto end;
920 }
921
12c8a1a3
JG
922 if (!integer) {
923 /* Field not found, not an error. */
924 goto end;
925 }
926
927 /* Make sure the payload has not already been set. */
928 if (!bt_ctf_field_validate(integer)) {
929 /* Payload already set, not an error */
930 goto end;
931 }
932
d7b1ea66
JG
933 field_type = bt_ctf_field_get_type(integer);
934 /* Something is serioulsly wrong */
935 assert(field_type);
936 if (bt_ctf_field_type_get_type_id(field_type) != CTF_TYPE_INTEGER) {
937 /*
938 * The user most likely meant for us to populate this field
939 * automatically. However, we can only do this if the field
940 * is an integer. Return an error.
941 */
942 ret = -1;
943 goto end;
944 }
945
946 if (bt_ctf_field_type_integer_get_signed(field_type)) {
947 ret = bt_ctf_field_signed_integer_set_value(integer,
948 (int64_t) value);
949 } else {
950 ret = bt_ctf_field_unsigned_integer_set_value(integer, value);
951 }
273b65be 952end:
d7b1ea66
JG
953 if (integer) {
954 bt_ctf_field_put(integer);
955 }
956 if (field_type) {
957 bt_ctf_field_type_put(field_type);
958 }
273b65be
JG
959 return ret;
960}
This page took 0.063627 seconds and 4 git commands to generate.