projects / babeltrace.git / blob
? search:


3713a2782bcd7d54c2d358776428ac7b36a3c7ea
[babeltrace.git] / src / plugins / ctf / common / msg-iter / msg-iter.c
1 /*
2 * Babeltrace - CTF message iterator
3 *
4 * Copyright (c) 2015-2018 EfficiOS Inc. and Linux Foundation
5 * Copyright (c) 2015-2018 Philippe Proulx <pproulx@efficios.com>
6 *
7 * Permission is hereby granted, free of charge, to any person obtaining a copy
8 * of this software and associated documentation files (the "Software"), to deal
9 * in the Software without restriction, including without limitation the rights
10 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11 * copies of the Software, and to permit persons to whom the Software is
12 * furnished to do so, subject to the following conditions:
13 *
14 * The above copyright notice and this permission notice shall be included in
15 * all copies or substantial portions of the Software.
16 *
17 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
20 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
23 * SOFTWARE.
24 */
25
26 #define BT_LOG_TAG "PLUGIN-CTF-MSG-ITER"
27 #include "logging.h"
28
29 #include <stdint.h>
30 #include <inttypes.h>
31 #include <stdio.h>
32 #include <stddef.h>
33 #include <stdbool.h>
34 #include "common/assert.h"
35 #include <string.h>
36 #include <babeltrace2/babeltrace.h>
37 #include "common/common.h"
38 #include <glib.h>
39 #include <stdlib.h>
40
41 #include "msg-iter.h"
42 #include "../bfcr/bfcr.h"
43
44 struct bt_msg_iter;
45
46 /* A visit stack entry */
47 struct stack_entry {
48 /*
49 * Current base field, one of:
50 *
51 * * string
52 * * structure
53 * * array
54 * * sequence
55 * * variant
56 *
57 * Field is borrowed.
58 */
59 bt_field *base;
60
61 /* Index of next field to set */
62 size_t index;
63 };
64
65 /* Visit stack */
66 struct stack {
67 /* Entries (struct stack_entry) */
68 GArray *entries;
69
70 /* Number of active entries */
71 size_t size;
72 };
73
74 /* State */
75 enum state {
76 STATE_INIT,
77 STATE_DSCOPE_TRACE_PACKET_HEADER_BEGIN,
78 STATE_DSCOPE_TRACE_PACKET_HEADER_CONTINUE,
79 STATE_AFTER_TRACE_PACKET_HEADER,
80 STATE_DSCOPE_STREAM_PACKET_CONTEXT_BEGIN,
81 STATE_DSCOPE_STREAM_PACKET_CONTEXT_CONTINUE,
82 STATE_AFTER_STREAM_PACKET_CONTEXT,
83 STATE_CHECK_EMIT_MSG_STREAM_BEGINNING,
84 STATE_EMIT_MSG_STREAM_BEGINNING,
85 STATE_EMIT_MSG_STREAM_ACTIVITY_BEGINNING,
86 STATE_CHECK_EMIT_MSG_DISCARDED_EVENTS,
87 STATE_CHECK_EMIT_MSG_DISCARDED_PACKETS,
88 STATE_EMIT_MSG_DISCARDED_EVENTS,
89 STATE_EMIT_MSG_DISCARDED_PACKETS,
90 STATE_EMIT_MSG_PACKET_BEGINNING,
91 STATE_DSCOPE_EVENT_HEADER_BEGIN,
92 STATE_DSCOPE_EVENT_HEADER_CONTINUE,
93 STATE_AFTER_EVENT_HEADER,
94 STATE_DSCOPE_EVENT_COMMON_CONTEXT_BEGIN,
95 STATE_DSCOPE_EVENT_COMMON_CONTEXT_CONTINUE,
96 STATE_DSCOPE_EVENT_SPEC_CONTEXT_BEGIN,
97 STATE_DSCOPE_EVENT_SPEC_CONTEXT_CONTINUE,
98 STATE_DSCOPE_EVENT_PAYLOAD_BEGIN,
99 STATE_DSCOPE_EVENT_PAYLOAD_CONTINUE,
100 STATE_EMIT_MSG_EVENT,
101 STATE_SKIP_PACKET_PADDING,
102 STATE_EMIT_MSG_PACKET_END_MULTI,
103 STATE_EMIT_MSG_PACKET_END_SINGLE,
104 STATE_CHECK_EMIT_MSG_STREAM_ACTIVITY_END,
105 STATE_EMIT_MSG_STREAM_ACTIVITY_END,
106 STATE_EMIT_MSG_STREAM_END,
107 STATE_DONE,
108 };
109
110 struct end_of_packet_snapshots {
111 uint64_t discarded_events;
112 uint64_t packets;
113 uint64_t beginning_clock;
114 uint64_t end_clock;
115 };
116
117 /* CTF message iterator */
118 struct bt_msg_iter {
119 /* Visit stack */
120 struct stack *stack;
121
122 /* Current message iterator to create messages (weak) */
123 bt_self_message_iterator *msg_iter;
124
125 /*
126 * True to emit stream beginning and stream activity beginning
127 * messages.
128 */
129 bool emit_stream_begin_msg;
130
131 /* True to emit stream end and stream activity end messages */
132 bool emit_stream_end_msg;
133
134 /* True to set the stream */
135 bool set_stream;
136
137 /*
138 * Current dynamic scope field pointer.
139 *
140 * This is set by read_dscope_begin_state() and contains the
141 * value of one of the pointers in `dscopes` below.
142 */
143 bt_field *cur_dscope_field;
144
145 /*
146 * True if we're done filling a string field from a text
147 * array/sequence payload.
148 */
149 bool done_filling_string;
150
151 /* Trace and classes */
152 /* True to set IR fields */
153 bool set_ir_fields;
154
155 struct {
156 struct ctf_trace_class *tc;
157 struct ctf_stream_class *sc;
158 struct ctf_event_class *ec;
159 } meta;
160
161 /* Current packet context field wrapper (NULL if not created yet) */
162 bt_packet_context_field *packet_context_field;
163
164 /* Current packet (NULL if not created yet) */
165 bt_packet *packet;
166
167 /* Current stream (NULL if not set yet) */
168 bt_stream *stream;
169
170 /* Current event (NULL if not created yet) */
171 bt_event *event;
172
173 /* Current event message (NULL if not created yet) */
174 bt_message *event_msg;
175
176 /* Database of current dynamic scopes */
177 struct {
178 bt_field *stream_packet_context;
179 bt_field *event_common_context;
180 bt_field *event_spec_context;
181 bt_field *event_payload;
182 } dscopes;
183
184 /* Current state */
185 enum state state;
186
187 /* Current medium buffer data */
188 struct {
189 /* Last address provided by medium */
190 const uint8_t *addr;
191
192 /* Buffer size provided by medium (bytes) */
193 size_t sz;
194
195 /* Offset within whole packet of addr (bits) */
196 size_t packet_offset;
197
198 /* Current position from addr (bits) */
199 size_t at;
200
201 /* Position of the last event header from addr (bits) */
202 size_t last_eh_at;
203 } buf;
204
205 /* Binary type reader */
206 struct bt_bfcr *bfcr;
207
208 /* Current medium data */
209 struct {
210 struct bt_msg_iter_medium_ops medops;
211 size_t max_request_sz;
212 void *data;
213 } medium;
214
215 /* Current packet size (bits) (-1 if unknown) */
216 int64_t cur_exp_packet_total_size;
217
218 /* Current content size (bits) (-1 if unknown) */
219 int64_t cur_exp_packet_content_size;
220
221 /* Current stream class ID */
222 int64_t cur_stream_class_id;
223
224 /* Current event class ID */
225 int64_t cur_event_class_id;
226
227 /* Current data stream ID */
228 int64_t cur_data_stream_id;
229
230 /*
231 * Offset, in the underlying media, of the current packet's
232 * start (-1 if unknown).
233 */
234 off_t cur_packet_offset;
235
236 /* Default clock's current value */
237 uint64_t default_clock_snapshot;
238
239 /* End of current packet snapshots */
240 struct end_of_packet_snapshots snapshots;
241
242 /* End of previous packet snapshots */
243 struct end_of_packet_snapshots prev_packet_snapshots;
244
245 /* Stored values (for sequence lengths, variant tags) */
246 GArray *stored_values;
247 };
248
249 static inline
250 const char *state_string(enum state state)
251 {
252 switch (state) {
253 case STATE_INIT:
254 return "STATE_INIT";
255 case STATE_DSCOPE_TRACE_PACKET_HEADER_BEGIN:
256 return "STATE_DSCOPE_TRACE_PACKET_HEADER_BEGIN";
257 case STATE_DSCOPE_TRACE_PACKET_HEADER_CONTINUE:
258 return "STATE_DSCOPE_TRACE_PACKET_HEADER_CONTINUE";
259 case STATE_AFTER_TRACE_PACKET_HEADER:
260 return "STATE_AFTER_TRACE_PACKET_HEADER";
261 case STATE_DSCOPE_STREAM_PACKET_CONTEXT_BEGIN:
262 return "STATE_DSCOPE_STREAM_PACKET_CONTEXT_BEGIN";
263 case STATE_DSCOPE_STREAM_PACKET_CONTEXT_CONTINUE:
264 return "STATE_DSCOPE_STREAM_PACKET_CONTEXT_CONTINUE";
265 case STATE_AFTER_STREAM_PACKET_CONTEXT:
266 return "STATE_AFTER_STREAM_PACKET_CONTEXT";
267 case STATE_EMIT_MSG_STREAM_BEGINNING:
268 return "STATE_EMIT_MSG_STREAM_BEGINNING";
269 case STATE_EMIT_MSG_STREAM_ACTIVITY_BEGINNING:
270 return "STATE_EMIT_MSG_STREAM_ACTIVITY_BEGINNING";
271 case STATE_EMIT_MSG_PACKET_BEGINNING:
272 return "STATE_EMIT_MSG_PACKET_BEGINNING";
273 case STATE_EMIT_MSG_DISCARDED_EVENTS:
274 return "STATE_EMIT_MSG_DISCARDED_EVENTS";
275 case STATE_EMIT_MSG_DISCARDED_PACKETS:
276 return "STATE_EMIT_MSG_DISCARDED_PACKETS";
277 case STATE_DSCOPE_EVENT_HEADER_BEGIN:
278 return "STATE_DSCOPE_EVENT_HEADER_BEGIN";
279 case STATE_DSCOPE_EVENT_HEADER_CONTINUE:
280 return "STATE_DSCOPE_EVENT_HEADER_CONTINUE";
281 case STATE_AFTER_EVENT_HEADER:
282 return "STATE_AFTER_EVENT_HEADER";
283 case STATE_DSCOPE_EVENT_COMMON_CONTEXT_BEGIN:
284 return "STATE_DSCOPE_EVENT_COMMON_CONTEXT_BEGIN";
285 case STATE_DSCOPE_EVENT_COMMON_CONTEXT_CONTINUE:
286 return "STATE_DSCOPE_EVENT_COMMON_CONTEXT_CONTINUE";
287 case STATE_DSCOPE_EVENT_SPEC_CONTEXT_BEGIN:
288 return "STATE_DSCOPE_EVENT_SPEC_CONTEXT_BEGIN";
289 case STATE_DSCOPE_EVENT_SPEC_CONTEXT_CONTINUE:
290 return "STATE_DSCOPE_EVENT_SPEC_CONTEXT_CONTINUE";
291 case STATE_DSCOPE_EVENT_PAYLOAD_BEGIN:
292 return "STATE_DSCOPE_EVENT_PAYLOAD_BEGIN";
293 case STATE_DSCOPE_EVENT_PAYLOAD_CONTINUE:
294 return "STATE_DSCOPE_EVENT_PAYLOAD_CONTINUE";
295 case STATE_EMIT_MSG_EVENT:
296 return "STATE_EMIT_MSG_EVENT";
297 case STATE_SKIP_PACKET_PADDING:
298 return "STATE_SKIP_PACKET_PADDING";
299 case STATE_EMIT_MSG_PACKET_END_MULTI:
300 return "STATE_EMIT_MSG_PACKET_END_MULTI";
301 case STATE_EMIT_MSG_PACKET_END_SINGLE:
302 return "STATE_EMIT_MSG_PACKET_END_SINGLE";
303 case STATE_EMIT_MSG_STREAM_ACTIVITY_END:
304 return "STATE_EMIT_MSG_STREAM_ACTIVITY_END";
305 case STATE_EMIT_MSG_STREAM_END:
306 return "STATE_EMIT_MSG_STREAM_END";
307 case STATE_DONE:
308 return "STATE_DONE";
309 default:
310 return "(unknown)";
311 }
312 }
313
314 static
315 int bt_msg_iter_switch_packet(struct bt_msg_iter *notit);
316
317 static
318 struct stack *stack_new(struct bt_msg_iter *notit)
319 {
320 struct stack *stack = NULL;
321
322 stack = g_new0(struct stack, 1);
323 if (!stack) {
324 BT_LOGE_STR("Failed to allocate one stack.");
325 goto error;
326 }
327
328 stack->entries = g_array_new(FALSE, TRUE, sizeof(struct stack_entry));
329 if (!stack->entries) {
330 BT_LOGE_STR("Failed to allocate a GArray.");
331 goto error;
332 }
333
334 BT_LOGD("Created stack: notit-addr=%p, stack-addr=%p", notit, stack);
335 goto end;
336
337 error:
338 g_free(stack);
339 stack = NULL;
340
341 end:
342 return stack;
343 }
344
345 static
346 void stack_destroy(struct stack *stack)
347 {
348 BT_ASSERT(stack);
349 BT_LOGD("Destroying stack: addr=%p", stack);
350
351 if (stack->entries) {
352 g_array_free(stack->entries, TRUE);
353 }
354
355 g_free(stack);
356 }
357
358 static
359 void stack_push(struct stack *stack, bt_field *base)
360 {
361 struct stack_entry *entry;
362
363 BT_ASSERT(stack);
364 BT_ASSERT(base);
365 BT_LOGV("Pushing base field on stack: stack-addr=%p, "
366 "stack-size-before=%zu, stack-size-after=%zu",
367 stack, stack->size, stack->size + 1);
368
369 if (stack->entries->len == stack->size) {
370 g_array_set_size(stack->entries, stack->size + 1);
371 }
372
373 entry = &g_array_index(stack->entries, struct stack_entry, stack->size);
374 entry->base = base;
375 entry->index = 0;
376 stack->size++;
377 }
378
379 static inline
380 unsigned int stack_size(struct stack *stack)
381 {
382 BT_ASSERT(stack);
383 return stack->size;
384 }
385
386 static
387 void stack_pop(struct stack *stack)
388 {
389 BT_ASSERT(stack);
390 BT_ASSERT(stack_size(stack));
391 BT_LOGV("Popping from stack: "
392 "stack-addr=%p, stack-size-before=%zu, stack-size-after=%zu",
393 stack, stack->size, stack->size - 1);
394 stack->size--;
395 }
396
397 static inline
398 struct stack_entry *stack_top(struct stack *stack)
399 {
400 BT_ASSERT(stack);
401 BT_ASSERT(stack_size(stack));
402 return &g_array_index(stack->entries, struct stack_entry,
403 stack->size - 1);
404 }
405
406 static inline
407 bool stack_empty(struct stack *stack)
408 {
409 return stack_size(stack) == 0;
410 }
411
412 static
413 void stack_clear(struct stack *stack)
414 {
415 BT_ASSERT(stack);
416 stack->size = 0;
417 }
418
419 static inline
420 enum bt_msg_iter_status msg_iter_status_from_m_status(
421 enum bt_msg_iter_medium_status m_status)
422 {
423 /* They are the same */
424 return (int) m_status;
425 }
426
427 static inline
428 size_t buf_size_bits(struct bt_msg_iter *notit)
429 {
430 return notit->buf.sz * 8;
431 }
432
433 static inline
434 size_t buf_available_bits(struct bt_msg_iter *notit)
435 {
436 return buf_size_bits(notit) - notit->buf.at;
437 }
438
439 static inline
440 size_t packet_at(struct bt_msg_iter *notit)
441 {
442 return notit->buf.packet_offset + notit->buf.at;
443 }
444
445 static inline
446 void buf_consume_bits(struct bt_msg_iter *notit, size_t incr)
447 {
448 BT_LOGV("Advancing cursor: notit-addr=%p, cur-before=%zu, cur-after=%zu",
449 notit, notit->buf.at, notit->buf.at + incr);
450 notit->buf.at += incr;
451 }
452
453 static
454 enum bt_msg_iter_status request_medium_bytes(
455 struct bt_msg_iter *notit)
456 {
457 uint8_t *buffer_addr = NULL;
458 size_t buffer_sz = 0;
459 enum bt_msg_iter_medium_status m_status;
460
461 BT_LOGD("Calling user function (request bytes): notit-addr=%p, "
462 "request-size=%zu", notit, notit->medium.max_request_sz);
463 m_status = notit->medium.medops.request_bytes(
464 notit->medium.max_request_sz, &buffer_addr,
465 &buffer_sz, notit->medium.data);
466 BT_LOGD("User function returned: status=%s, buf-addr=%p, buf-size=%zu",
467 bt_msg_iter_medium_status_string(m_status),
468 buffer_addr, buffer_sz);
469 if (m_status == BT_MSG_ITER_MEDIUM_STATUS_OK) {
470 BT_ASSERT(buffer_sz != 0);
471
472 /* New packet offset is old one + old size (in bits) */
473 notit->buf.packet_offset += buf_size_bits(notit);
474
475 /* Restart at the beginning of the new medium buffer */
476 notit->buf.at = 0;
477 notit->buf.last_eh_at = SIZE_MAX;
478
479 /* New medium buffer size */
480 notit->buf.sz = buffer_sz;
481
482 /* New medium buffer address */
483 notit->buf.addr = buffer_addr;
484
485 BT_LOGD("User function returned new bytes: "
486 "packet-offset=%zu, cur=%zu, size=%zu, addr=%p",
487 notit->buf.packet_offset, notit->buf.at,
488 notit->buf.sz, notit->buf.addr);
489 BT_LOGD_MEM(buffer_addr, buffer_sz, "Returned bytes at %p:",
490 buffer_addr);
491 } else if (m_status == BT_MSG_ITER_MEDIUM_STATUS_EOF) {
492 /*
493 * User returned end of stream: validate that we're not
494 * in the middle of a packet header, packet context, or
495 * event.
496 */
497 if (notit->cur_exp_packet_total_size >= 0) {
498 if (packet_at(notit) ==
499 notit->cur_exp_packet_total_size) {
500 goto end;
501 }
502 } else {
503 if (packet_at(notit) == 0) {
504 goto end;
505 }
506
507 if (notit->buf.last_eh_at != SIZE_MAX &&
508 notit->buf.at == notit->buf.last_eh_at) {
509 goto end;
510 }
511 }
512
513 /* All other states are invalid */
514 BT_LOGW("User function returned %s, but message iterator is in an unexpected state: "
515 "state=%s, cur-packet-size=%" PRId64 ", cur=%zu, "
516 "packet-cur=%zu, last-eh-at=%zu",
517 bt_msg_iter_medium_status_string(m_status),
518 state_string(notit->state),
519 notit->cur_exp_packet_total_size,
520 notit->buf.at, packet_at(notit),
521 notit->buf.last_eh_at);
522 m_status = BT_MSG_ITER_MEDIUM_STATUS_ERROR;
523 } else if (m_status < 0) {
524 BT_LOGW("User function failed: status=%s",
525 bt_msg_iter_medium_status_string(m_status));
526 }
527
528 end:
529 return msg_iter_status_from_m_status(m_status);
530 }
531
532 static inline
533 enum bt_msg_iter_status buf_ensure_available_bits(
534 struct bt_msg_iter *notit)
535 {
536 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
537
538 if (G_UNLIKELY(buf_available_bits(notit) == 0)) {
539 /*
540 * This _cannot_ return BT_MSG_ITER_STATUS_OK
541 * _and_ no bits.
542 */
543 status = request_medium_bytes(notit);
544 }
545
546 return status;
547 }
548
549 static
550 enum bt_msg_iter_status read_dscope_begin_state(
551 struct bt_msg_iter *notit,
552 struct ctf_field_class *dscope_fc,
553 enum state done_state, enum state continue_state,
554 bt_field *dscope_field)
555 {
556 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
557 enum bt_bfcr_status bfcr_status;
558 size_t consumed_bits;
559
560 notit->cur_dscope_field = dscope_field;
561 BT_LOGV("Starting BFCR: notit-addr=%p, bfcr-addr=%p, fc-addr=%p",
562 notit, notit->bfcr, dscope_fc);
563 consumed_bits = bt_bfcr_start(notit->bfcr, dscope_fc,
564 notit->buf.addr, notit->buf.at, packet_at(notit),
565 notit->buf.sz, &bfcr_status);
566 BT_LOGV("BFCR consumed bits: size=%zu", consumed_bits);
567
568 switch (bfcr_status) {
569 case BT_BFCR_STATUS_OK:
570 /* Field class was read completely */
571 BT_LOGV_STR("Field was completely decoded.");
572 notit->state = done_state;
573 break;
574 case BT_BFCR_STATUS_EOF:
575 BT_LOGV_STR("BFCR needs more data to decode field completely.");
576 notit->state = continue_state;
577 break;
578 default:
579 BT_LOGW("BFCR failed to start: notit-addr=%p, bfcr-addr=%p, "
580 "status=%s", notit, notit->bfcr,
581 bt_bfcr_status_string(bfcr_status));
582 status = BT_MSG_ITER_STATUS_ERROR;
583 goto end;
584 }
585
586 /* Consume bits now since we know we're not in an error state */
587 buf_consume_bits(notit, consumed_bits);
588
589 end:
590 return status;
591 }
592
593 static
594 enum bt_msg_iter_status read_dscope_continue_state(
595 struct bt_msg_iter *notit, enum state done_state)
596 {
597 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
598 enum bt_bfcr_status bfcr_status;
599 size_t consumed_bits;
600
601 BT_LOGV("Continuing BFCR: notit-addr=%p, bfcr-addr=%p",
602 notit, notit->bfcr);
603
604 status = buf_ensure_available_bits(notit);
605 if (status != BT_MSG_ITER_STATUS_OK) {
606 if (status < 0) {
607 BT_LOGW("Cannot ensure that buffer has at least one byte: "
608 "msg-addr=%p, status=%s",
609 notit, bt_msg_iter_status_string(status));
610 } else {
611 BT_LOGV("Cannot ensure that buffer has at least one byte: "
612 "msg-addr=%p, status=%s",
613 notit, bt_msg_iter_status_string(status));
614 }
615
616 goto end;
617 }
618
619 consumed_bits = bt_bfcr_continue(notit->bfcr, notit->buf.addr,
620 notit->buf.sz, &bfcr_status);
621 BT_LOGV("BFCR consumed bits: size=%zu", consumed_bits);
622
623 switch (bfcr_status) {
624 case BT_BFCR_STATUS_OK:
625 /* Type was read completely. */
626 BT_LOGV_STR("Field was completely decoded.");
627 notit->state = done_state;
628 break;
629 case BT_BFCR_STATUS_EOF:
630 /* Stay in this continue state. */
631 BT_LOGV_STR("BFCR needs more data to decode field completely.");
632 break;
633 default:
634 BT_LOGW("BFCR failed to continue: notit-addr=%p, bfcr-addr=%p, "
635 "status=%s", notit, notit->bfcr,
636 bt_bfcr_status_string(bfcr_status));
637 status = BT_MSG_ITER_STATUS_ERROR;
638 goto end;
639 }
640
641 /* Consume bits now since we know we're not in an error state. */
642 buf_consume_bits(notit, consumed_bits);
643 end:
644 return status;
645 }
646
647 static
648 void release_event_dscopes(struct bt_msg_iter *notit)
649 {
650 notit->dscopes.event_common_context = NULL;
651 notit->dscopes.event_spec_context = NULL;
652 notit->dscopes.event_payload = NULL;
653 }
654
655 static
656 void release_all_dscopes(struct bt_msg_iter *notit)
657 {
658 notit->dscopes.stream_packet_context = NULL;
659
660 if (notit->packet_context_field) {
661 bt_packet_context_field_release(notit->packet_context_field);
662 notit->packet_context_field = NULL;
663 }
664
665 release_event_dscopes(notit);
666 }
667
668 static
669 enum bt_msg_iter_status read_packet_header_begin_state(
670 struct bt_msg_iter *notit)
671 {
672 struct ctf_field_class *packet_header_fc = NULL;
673 enum bt_msg_iter_status ret = BT_MSG_ITER_STATUS_OK;
674
675 if (bt_msg_iter_switch_packet(notit)) {
676 BT_LOGW("Cannot switch packet: notit-addr=%p", notit);
677 ret = BT_MSG_ITER_STATUS_ERROR;
678 goto end;
679 }
680
681 /*
682 * Make sure at least one bit is available for this packet. An
683 * empty packet is impossible. If we reach the end of the medium
684 * at this point, then it's considered the end of the stream.
685 */
686 ret = buf_ensure_available_bits(notit);
687 switch (ret) {
688 case BT_MSG_ITER_STATUS_OK:
689 break;
690 case BT_MSG_ITER_STATUS_EOF:
691 ret = BT_MSG_ITER_STATUS_OK;
692 notit->state = STATE_CHECK_EMIT_MSG_STREAM_ACTIVITY_END;
693 goto end;
694 default:
695 goto end;
696 }
697
698 /* Packet header class is common to the whole trace class. */
699 packet_header_fc = notit->meta.tc->packet_header_fc;
700 if (!packet_header_fc) {
701 notit->state = STATE_AFTER_TRACE_PACKET_HEADER;
702 goto end;
703 }
704
705 notit->cur_stream_class_id = -1;
706 notit->cur_event_class_id = -1;
707 notit->cur_data_stream_id = -1;
708 BT_LOGD("Decoding packet header field:"
709 "notit-addr=%p, trace-class-addr=%p, fc-addr=%p",
710 notit, notit->meta.tc, packet_header_fc);
711 ret = read_dscope_begin_state(notit, packet_header_fc,
712 STATE_AFTER_TRACE_PACKET_HEADER,
713 STATE_DSCOPE_TRACE_PACKET_HEADER_CONTINUE, NULL);
714 if (ret < 0) {
715 BT_LOGW("Cannot decode packet header field: "
716 "notit-addr=%p, trace-class-addr=%p, "
717 "fc-addr=%p",
718 notit, notit->meta.tc, packet_header_fc);
719 }
720
721 end:
722 return ret;
723 }
724
725 static
726 enum bt_msg_iter_status read_packet_header_continue_state(
727 struct bt_msg_iter *notit)
728 {
729 return read_dscope_continue_state(notit,
730 STATE_AFTER_TRACE_PACKET_HEADER);
731 }
732
733 static inline
734 enum bt_msg_iter_status set_current_stream_class(struct bt_msg_iter *notit)
735 {
736 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
737 struct ctf_stream_class *new_stream_class = NULL;
738
739 if (notit->cur_stream_class_id == -1) {
740 /*
741 * No current stream class ID field, therefore only one
742 * stream class.
743 */
744 if (notit->meta.tc->stream_classes->len != 1) {
745 BT_LOGW("Need exactly one stream class since there's "
746 "no stream class ID field: "
747 "notit-addr=%p", notit);
748 status = BT_MSG_ITER_STATUS_ERROR;
749 goto end;
750 }
751
752 new_stream_class = notit->meta.tc->stream_classes->pdata[0];
753 notit->cur_stream_class_id = new_stream_class->id;
754 }
755
756 new_stream_class = ctf_trace_class_borrow_stream_class_by_id(
757 notit->meta.tc, notit->cur_stream_class_id);
758 if (!new_stream_class) {
759 BT_LOGW("No stream class with ID of stream class ID to use in trace class: "
760 "notit-addr=%p, stream-class-id=%" PRIu64 ", "
761 "trace-class-addr=%p",
762 notit, notit->cur_stream_class_id, notit->meta.tc);
763 status = BT_MSG_ITER_STATUS_ERROR;
764 goto end;
765 }
766
767 if (notit->meta.sc) {
768 if (new_stream_class != notit->meta.sc) {
769 BT_LOGW("Two packets refer to two different stream classes within the same packet sequence: "
770 "notit-addr=%p, prev-stream-class-addr=%p, "
771 "prev-stream-class-id=%" PRId64 ", "
772 "next-stream-class-addr=%p, "
773 "next-stream-class-id=%" PRId64 ", "
774 "trace-addr=%p",
775 notit, notit->meta.sc,
776 notit->meta.sc->id,
777 new_stream_class,
778 new_stream_class->id,
779 notit->meta.tc);
780 status = BT_MSG_ITER_STATUS_ERROR;
781 goto end;
782 }
783 } else {
784 notit->meta.sc = new_stream_class;
785 }
786
787 BT_LOGD("Set current stream class: "
788 "notit-addr=%p, stream-class-addr=%p, "
789 "stream-class-id=%" PRId64,
790 notit, notit->meta.sc, notit->meta.sc->id);
791
792 end:
793 return status;
794 }
795
796 static inline
797 enum bt_msg_iter_status set_current_stream(struct bt_msg_iter *notit)
798 {
799 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
800 bt_stream *stream = NULL;
801
802 BT_LOGD("Calling user function (get stream): notit-addr=%p, "
803 "stream-class-addr=%p, stream-class-id=%" PRId64,
804 notit, notit->meta.sc,
805 notit->meta.sc->id);
806 stream = notit->medium.medops.borrow_stream(
807 notit->meta.sc->ir_sc, notit->cur_data_stream_id,
808 notit->medium.data);
809 bt_stream_get_ref(stream);
810 BT_LOGD("User function returned: stream-addr=%p", stream);
811 if (!stream) {
812 BT_LOGW_STR("User function failed to return a stream object "
813 "for the given stream class.");
814 status = BT_MSG_ITER_STATUS_ERROR;
815 goto end;
816 }
817
818 if (notit->stream && stream != notit->stream) {
819 BT_LOGW("User function returned a different stream than the "
820 "previous one for the same sequence of packets.");
821 status = BT_MSG_ITER_STATUS_ERROR;
822 goto end;
823 }
824
825 BT_STREAM_MOVE_REF(notit->stream, stream);
826
827 end:
828 bt_stream_put_ref(stream);
829 return status;
830 }
831
832 static inline
833 enum bt_msg_iter_status set_current_packet(struct bt_msg_iter *notit)
834 {
835 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
836 bt_packet *packet = NULL;
837
838 BT_LOGD("Creating packet for packet message: "
839 "notit-addr=%p", notit);
840 BT_LOGD("Creating packet from stream: "
841 "notit-addr=%p, stream-addr=%p, "
842 "stream-class-addr=%p, "
843 "stream-class-id=%" PRId64,
844 notit, notit->stream, notit->meta.sc,
845 notit->meta.sc->id);
846
847 /* Create packet */
848 BT_ASSERT(notit->stream);
849 packet = bt_packet_create(notit->stream);
850 if (!packet) {
851 BT_LOGE("Cannot create packet from stream: "
852 "notit-addr=%p, stream-addr=%p, "
853 "stream-class-addr=%p, "
854 "stream-class-id=%" PRId64,
855 notit, notit->stream, notit->meta.sc,
856 notit->meta.sc->id);
857 goto error;
858 }
859
860 goto end;
861
862 error:
863 BT_PACKET_PUT_REF_AND_RESET(packet);
864 status = BT_MSG_ITER_STATUS_ERROR;
865
866 end:
867 BT_PACKET_MOVE_REF(notit->packet, packet);
868 return status;
869 }
870
871 static
872 enum bt_msg_iter_status after_packet_header_state(
873 struct bt_msg_iter *notit)
874 {
875 enum bt_msg_iter_status status;
876
877 status = set_current_stream_class(notit);
878 if (status != BT_MSG_ITER_STATUS_OK) {
879 goto end;
880 }
881
882 notit->state = STATE_DSCOPE_STREAM_PACKET_CONTEXT_BEGIN;
883
884 end:
885 return status;
886 }
887
888 static
889 enum bt_msg_iter_status read_packet_context_begin_state(
890 struct bt_msg_iter *notit)
891 {
892 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
893 struct ctf_field_class *packet_context_fc;
894
895 BT_ASSERT(notit->meta.sc);
896 packet_context_fc = notit->meta.sc->packet_context_fc;
897 if (!packet_context_fc) {
898 BT_LOGD("No packet packet context field class in stream class: continuing: "
899 "notit-addr=%p, stream-class-addr=%p, "
900 "stream-class-id=%" PRId64,
901 notit, notit->meta.sc,
902 notit->meta.sc->id);
903 notit->state = STATE_AFTER_STREAM_PACKET_CONTEXT;
904 goto end;
905 }
906
907 BT_ASSERT(!notit->packet_context_field);
908
909 if (packet_context_fc->in_ir) {
910 /*
911 * Create free packet context field from stream class.
912 * This field is going to be moved to the packet once we
913 * create it. We cannot create the packet now because a
914 * packet is created from a stream, and this API must be
915 * able to return the packet context properties without
916 * creating a stream
917 * (bt_msg_iter_get_packet_properties()).
918 */
919 notit->packet_context_field =
920 bt_packet_context_field_create(
921 notit->meta.sc->ir_sc);
922 if (!notit->packet_context_field) {
923 BT_LOGE_STR("Cannot create packet context field wrapper from stream class.");
924 status = BT_MSG_ITER_STATUS_ERROR;
925 goto end;
926 }
927
928 notit->dscopes.stream_packet_context =
929 bt_packet_context_field_borrow_field(
930 notit->packet_context_field);
931 BT_ASSERT(notit->dscopes.stream_packet_context);
932 }
933
934 BT_LOGD("Decoding packet context field: "
935 "notit-addr=%p, stream-class-addr=%p, "
936 "stream-class-id=%" PRId64 ", fc-addr=%p",
937 notit, notit->meta.sc,
938 notit->meta.sc->id, packet_context_fc);
939 status = read_dscope_begin_state(notit, packet_context_fc,
940 STATE_AFTER_STREAM_PACKET_CONTEXT,
941 STATE_DSCOPE_STREAM_PACKET_CONTEXT_CONTINUE,
942 notit->dscopes.stream_packet_context);
943 if (status < 0) {
944 BT_LOGW("Cannot decode packet context field: "
945 "notit-addr=%p, stream-class-addr=%p, "
946 "stream-class-id=%" PRId64 ", fc-addr=%p",
947 notit, notit->meta.sc,
948 notit->meta.sc->id,
949 packet_context_fc);
950 }
951
952 end:
953 return status;
954 }
955
956 static
957 enum bt_msg_iter_status read_packet_context_continue_state(
958 struct bt_msg_iter *notit)
959 {
960 return read_dscope_continue_state(notit,
961 STATE_AFTER_STREAM_PACKET_CONTEXT);
962 }
963
964 static
965 enum bt_msg_iter_status set_current_packet_content_sizes(
966 struct bt_msg_iter *notit)
967 {
968 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
969
970 if (notit->cur_exp_packet_total_size == -1) {
971 if (notit->cur_exp_packet_content_size != -1) {
972 notit->cur_exp_packet_total_size =
973 notit->cur_exp_packet_content_size;
974 }
975 } else {
976 if (notit->cur_exp_packet_content_size == -1) {
977 notit->cur_exp_packet_content_size =
978 notit->cur_exp_packet_total_size;
979 }
980 }
981
982 BT_ASSERT((notit->cur_exp_packet_total_size >= 0 &&
983 notit->cur_exp_packet_content_size >= 0) ||
984 (notit->cur_exp_packet_total_size < 0 &&
985 notit->cur_exp_packet_content_size < 0));
986
987 if (notit->cur_exp_packet_content_size >
988 notit->cur_exp_packet_total_size) {
989 BT_LOGW("Invalid packet or content size: "
990 "content size is greater than packet size: "
991 "notit-addr=%p, packet-context-field-addr=%p, "
992 "packet-size=%" PRId64 ", content-size=%" PRId64,
993 notit, notit->dscopes.stream_packet_context,
994 notit->cur_exp_packet_total_size,
995 notit->cur_exp_packet_content_size);
996 status = BT_MSG_ITER_STATUS_ERROR;
997 goto end;
998 }
999
1000 BT_LOGD("Set current packet and content sizes: "
1001 "notit-addr=%p, packet-size=%" PRIu64 ", content-size=%" PRIu64,
1002 notit, notit->cur_exp_packet_total_size,
1003 notit->cur_exp_packet_content_size);
1004
1005 end:
1006 return status;
1007 }
1008
1009 static
1010 enum bt_msg_iter_status after_packet_context_state(struct bt_msg_iter *notit)
1011 {
1012 enum bt_msg_iter_status status;
1013
1014 status = set_current_packet_content_sizes(notit);
1015 if (status != BT_MSG_ITER_STATUS_OK) {
1016 goto end;
1017 }
1018
1019 if (notit->stream) {
1020 /*
1021 * Stream exists, which means we already emitted at
1022 * least one packet beginning message, so the initial
1023 * stream beginning message was also emitted.
1024 */
1025 notit->state = STATE_CHECK_EMIT_MSG_DISCARDED_EVENTS;
1026 } else {
1027 notit->state = STATE_CHECK_EMIT_MSG_STREAM_BEGINNING;
1028 }
1029
1030 end:
1031 return status;
1032 }
1033
1034 static
1035 enum bt_msg_iter_status read_event_header_begin_state(struct bt_msg_iter *notit)
1036 {
1037 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
1038 struct ctf_field_class *event_header_fc = NULL;
1039
1040 /* Reset the position of the last event header */
1041 notit->buf.last_eh_at = notit->buf.at;
1042 notit->cur_event_class_id = -1;
1043
1044 /* Check if we have some content left */
1045 if (notit->cur_exp_packet_content_size >= 0) {
1046 if (G_UNLIKELY(packet_at(notit) ==
1047 notit->cur_exp_packet_content_size)) {
1048 /* No more events! */
1049 BT_LOGD("Reached end of packet: notit-addr=%p, "
1050 "cur=%zu", notit, packet_at(notit));
1051 notit->state = STATE_EMIT_MSG_PACKET_END_MULTI;
1052 goto end;
1053 } else if (G_UNLIKELY(packet_at(notit) >
1054 notit->cur_exp_packet_content_size)) {
1055 /* That's not supposed to happen */
1056 BT_LOGD("Before decoding event header field: cursor is passed the packet's content: "
1057 "notit-addr=%p, content-size=%" PRId64 ", "
1058 "cur=%zu", notit,
1059 notit->cur_exp_packet_content_size,
1060 packet_at(notit));
1061 status = BT_MSG_ITER_STATUS_ERROR;
1062 goto end;
1063 }
1064 } else {
1065 /*
1066 * "Infinite" content: we're done when the medium has
1067 * nothing else for us.
1068 */
1069 status = buf_ensure_available_bits(notit);
1070 switch (status) {
1071 case BT_MSG_ITER_STATUS_OK:
1072 break;
1073 case BT_MSG_ITER_STATUS_EOF:
1074 status = BT_MSG_ITER_STATUS_OK;
1075 notit->state = STATE_EMIT_MSG_PACKET_END_SINGLE;
1076 goto end;
1077 default:
1078 goto end;
1079 }
1080 }
1081
1082 release_event_dscopes(notit);
1083 BT_ASSERT(notit->meta.sc);
1084 event_header_fc = notit->meta.sc->event_header_fc;
1085 if (!event_header_fc) {
1086 notit->state = STATE_AFTER_EVENT_HEADER;
1087 goto end;
1088 }
1089
1090 BT_LOGD("Decoding event header field: "
1091 "notit-addr=%p, stream-class-addr=%p, "
1092 "stream-class-id=%" PRId64 ", "
1093 "fc-addr=%p",
1094 notit, notit->meta.sc,
1095 notit->meta.sc->id,
1096 event_header_fc);
1097 status = read_dscope_begin_state(notit, event_header_fc,
1098 STATE_AFTER_EVENT_HEADER,
1099 STATE_DSCOPE_EVENT_HEADER_CONTINUE, NULL);
1100 if (status < 0) {
1101 BT_LOGW("Cannot decode event header field: "
1102 "notit-addr=%p, stream-class-addr=%p, "
1103 "stream-class-id=%" PRId64 ", fc-addr=%p",
1104 notit, notit->meta.sc,
1105 notit->meta.sc->id,
1106 event_header_fc);
1107 }
1108
1109 end:
1110 return status;
1111 }
1112
1113 static
1114 enum bt_msg_iter_status read_event_header_continue_state(
1115 struct bt_msg_iter *notit)
1116 {
1117 return read_dscope_continue_state(notit,
1118 STATE_AFTER_EVENT_HEADER);
1119 }
1120
1121 static inline
1122 enum bt_msg_iter_status set_current_event_class(struct bt_msg_iter *notit)
1123 {
1124 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
1125
1126 struct ctf_event_class *new_event_class = NULL;
1127
1128 if (notit->cur_event_class_id == -1) {
1129 /*
1130 * No current event class ID field, therefore only one
1131 * event class.
1132 */
1133 if (notit->meta.sc->event_classes->len != 1) {
1134 BT_LOGW("Need exactly one event class since there's "
1135 "no event class ID field: "
1136 "notit-addr=%p", notit);
1137 status = BT_MSG_ITER_STATUS_ERROR;
1138 goto end;
1139 }
1140
1141 new_event_class = notit->meta.sc->event_classes->pdata[0];
1142 notit->cur_event_class_id = new_event_class->id;
1143 }
1144
1145 new_event_class = ctf_stream_class_borrow_event_class_by_id(
1146 notit->meta.sc, notit->cur_event_class_id);
1147 if (!new_event_class) {
1148 BT_LOGW("No event class with ID of event class ID to use in stream class: "
1149 "notit-addr=%p, stream-class-id=%" PRIu64 ", "
1150 "event-class-id=%" PRIu64 ", "
1151 "trace-class-addr=%p",
1152 notit, notit->meta.sc->id, notit->cur_event_class_id,
1153 notit->meta.tc);
1154 status = BT_MSG_ITER_STATUS_ERROR;
1155 goto end;
1156 }
1157
1158 notit->meta.ec = new_event_class;
1159 BT_LOGD("Set current event class: "
1160 "notit-addr=%p, event-class-addr=%p, "
1161 "event-class-id=%" PRId64 ", "
1162 "event-class-name=\"%s\"",
1163 notit, notit->meta.ec, notit->meta.ec->id,
1164 notit->meta.ec->name->str);
1165
1166 end:
1167 return status;
1168 }
1169
1170 static inline
1171 enum bt_msg_iter_status set_current_event_message(
1172 struct bt_msg_iter *notit)
1173 {
1174 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
1175 bt_message *msg = NULL;
1176
1177 BT_ASSERT(notit->meta.ec);
1178 BT_ASSERT(notit->packet);
1179 BT_LOGD("Creating event message from event class and packet: "
1180 "notit-addr=%p, ec-addr=%p, ec-name=\"%s\", packet-addr=%p",
1181 notit, notit->meta.ec,
1182 notit->meta.ec->name->str,
1183 notit->packet);
1184 BT_ASSERT(notit->msg_iter);
1185 BT_ASSERT(notit->meta.sc);
1186
1187 if (bt_stream_class_borrow_default_clock_class(notit->meta.sc->ir_sc)) {
1188 msg = bt_message_event_create_with_default_clock_snapshot(
1189 notit->msg_iter, notit->meta.ec->ir_ec,
1190 notit->packet, notit->default_clock_snapshot);
1191 } else {
1192 msg = bt_message_event_create(notit->msg_iter,
1193 notit->meta.ec->ir_ec, notit->packet);
1194 }
1195
1196 if (!msg) {
1197 BT_LOGE("Cannot create event message: "
1198 "notit-addr=%p, ec-addr=%p, ec-name=\"%s\", "
1199 "packet-addr=%p",
1200 notit, notit->meta.ec,
1201 notit->meta.ec->name->str,
1202 notit->packet);
1203 goto error;
1204 }
1205
1206 goto end;
1207
1208 error:
1209 BT_MESSAGE_PUT_REF_AND_RESET(msg);
1210 status = BT_MSG_ITER_STATUS_ERROR;
1211
1212 end:
1213 BT_MESSAGE_MOVE_REF(notit->event_msg, msg);
1214 return status;
1215 }
1216
1217 static
1218 enum bt_msg_iter_status after_event_header_state(
1219 struct bt_msg_iter *notit)
1220 {
1221 enum bt_msg_iter_status status;
1222
1223 status = set_current_event_class(notit);
1224 if (status != BT_MSG_ITER_STATUS_OK) {
1225 goto end;
1226 }
1227
1228 status = set_current_event_message(notit);
1229 if (status != BT_MSG_ITER_STATUS_OK) {
1230 goto end;
1231 }
1232
1233 notit->event = bt_message_event_borrow_event(
1234 notit->event_msg);
1235 BT_ASSERT(notit->event);
1236 notit->state = STATE_DSCOPE_EVENT_COMMON_CONTEXT_BEGIN;
1237
1238 end:
1239 return status;
1240 }
1241
1242 static
1243 enum bt_msg_iter_status read_event_common_context_begin_state(
1244 struct bt_msg_iter *notit)
1245 {
1246 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
1247 struct ctf_field_class *event_common_context_fc;
1248
1249 event_common_context_fc = notit->meta.sc->event_common_context_fc;
1250 if (!event_common_context_fc) {
1251 notit->state = STATE_DSCOPE_EVENT_SPEC_CONTEXT_BEGIN;
1252 goto end;
1253 }
1254
1255 if (event_common_context_fc->in_ir) {
1256 BT_ASSERT(!notit->dscopes.event_common_context);
1257 notit->dscopes.event_common_context =
1258 bt_event_borrow_common_context_field(
1259 notit->event);
1260 BT_ASSERT(notit->dscopes.event_common_context);
1261 }
1262
1263 BT_LOGV("Decoding event common context field: "
1264 "notit-addr=%p, stream-class-addr=%p, "
1265 "stream-class-id=%" PRId64 ", "
1266 "fc-addr=%p",
1267 notit, notit->meta.sc,
1268 notit->meta.sc->id,
1269 event_common_context_fc);
1270 status = read_dscope_begin_state(notit, event_common_context_fc,
1271 STATE_DSCOPE_EVENT_SPEC_CONTEXT_BEGIN,
1272 STATE_DSCOPE_EVENT_COMMON_CONTEXT_CONTINUE,
1273 notit->dscopes.event_common_context);
1274 if (status < 0) {
1275 BT_LOGW("Cannot decode event common context field: "
1276 "notit-addr=%p, stream-class-addr=%p, "
1277 "stream-class-id=%" PRId64 ", fc-addr=%p",
1278 notit, notit->meta.sc,
1279 notit->meta.sc->id,
1280 event_common_context_fc);
1281 }
1282
1283 end:
1284 return status;
1285 }
1286
1287 static
1288 enum bt_msg_iter_status read_event_common_context_continue_state(
1289 struct bt_msg_iter *notit)
1290 {
1291 return read_dscope_continue_state(notit,
1292 STATE_DSCOPE_EVENT_SPEC_CONTEXT_BEGIN);
1293 }
1294
1295 static
1296 enum bt_msg_iter_status read_event_spec_context_begin_state(
1297 struct bt_msg_iter *notit)
1298 {
1299 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
1300 struct ctf_field_class *event_spec_context_fc;
1301
1302 event_spec_context_fc = notit->meta.ec->spec_context_fc;
1303 if (!event_spec_context_fc) {
1304 notit->state = STATE_DSCOPE_EVENT_PAYLOAD_BEGIN;
1305 goto end;
1306 }
1307
1308 if (event_spec_context_fc->in_ir) {
1309 BT_ASSERT(!notit->dscopes.event_spec_context);
1310 notit->dscopes.event_spec_context =
1311 bt_event_borrow_specific_context_field(
1312 notit->event);
1313 BT_ASSERT(notit->dscopes.event_spec_context);
1314 }
1315
1316 BT_LOGV("Decoding event specific context field: "
1317 "notit-addr=%p, event-class-addr=%p, "
1318 "event-class-name=\"%s\", event-class-id=%" PRId64 ", "
1319 "fc-addr=%p",
1320 notit, notit->meta.ec,
1321 notit->meta.ec->name->str,
1322 notit->meta.ec->id,
1323 event_spec_context_fc);
1324 status = read_dscope_begin_state(notit, event_spec_context_fc,
1325 STATE_DSCOPE_EVENT_PAYLOAD_BEGIN,
1326 STATE_DSCOPE_EVENT_SPEC_CONTEXT_CONTINUE,
1327 notit->dscopes.event_spec_context);
1328 if (status < 0) {
1329 BT_LOGW("Cannot decode event specific context field: "
1330 "notit-addr=%p, event-class-addr=%p, "
1331 "event-class-name=\"%s\", "
1332 "event-class-id=%" PRId64 ", fc-addr=%p",
1333 notit, notit->meta.ec,
1334 notit->meta.ec->name->str,
1335 notit->meta.ec->id,
1336 event_spec_context_fc);
1337 }
1338
1339 end:
1340 return status;
1341 }
1342
1343 static
1344 enum bt_msg_iter_status read_event_spec_context_continue_state(
1345 struct bt_msg_iter *notit)
1346 {
1347 return read_dscope_continue_state(notit,
1348 STATE_DSCOPE_EVENT_PAYLOAD_BEGIN);
1349 }
1350
1351 static
1352 enum bt_msg_iter_status read_event_payload_begin_state(
1353 struct bt_msg_iter *notit)
1354 {
1355 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
1356 struct ctf_field_class *event_payload_fc;
1357
1358 event_payload_fc = notit->meta.ec->payload_fc;
1359 if (!event_payload_fc) {
1360 notit->state = STATE_EMIT_MSG_EVENT;
1361 goto end;
1362 }
1363
1364 if (event_payload_fc->in_ir) {
1365 BT_ASSERT(!notit->dscopes.event_payload);
1366 notit->dscopes.event_payload =
1367 bt_event_borrow_payload_field(
1368 notit->event);
1369 BT_ASSERT(notit->dscopes.event_payload);
1370 }
1371
1372 BT_LOGV("Decoding event payload field: "
1373 "notit-addr=%p, event-class-addr=%p, "
1374 "event-class-name=\"%s\", event-class-id=%" PRId64 ", "
1375 "fc-addr=%p",
1376 notit, notit->meta.ec,
1377 notit->meta.ec->name->str,
1378 notit->meta.ec->id,
1379 event_payload_fc);
1380 status = read_dscope_begin_state(notit, event_payload_fc,
1381 STATE_EMIT_MSG_EVENT,
1382 STATE_DSCOPE_EVENT_PAYLOAD_CONTINUE,
1383 notit->dscopes.event_payload);
1384 if (status < 0) {
1385 BT_LOGW("Cannot decode event payload field: "
1386 "notit-addr=%p, event-class-addr=%p, "
1387 "event-class-name=\"%s\", "
1388 "event-class-id=%" PRId64 ", fc-addr=%p",
1389 notit, notit->meta.ec,
1390 notit->meta.ec->name->str,
1391 notit->meta.ec->id,
1392 event_payload_fc);
1393 }
1394
1395 end:
1396 return status;
1397 }
1398
1399 static
1400 enum bt_msg_iter_status read_event_payload_continue_state(
1401 struct bt_msg_iter *notit)
1402 {
1403 return read_dscope_continue_state(notit, STATE_EMIT_MSG_EVENT);
1404 }
1405
1406 static
1407 enum bt_msg_iter_status skip_packet_padding_state(struct bt_msg_iter *notit)
1408 {
1409 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
1410 size_t bits_to_skip;
1411
1412 BT_ASSERT(notit->cur_exp_packet_total_size > 0);
1413 bits_to_skip = notit->cur_exp_packet_total_size - packet_at(notit);
1414 if (bits_to_skip == 0) {
1415 notit->state = STATE_DSCOPE_TRACE_PACKET_HEADER_BEGIN;
1416 goto end;
1417 } else {
1418 size_t bits_to_consume;
1419
1420 BT_LOGD("Trying to skip %zu bits of padding: notit-addr=%p, size=%zu",
1421 bits_to_skip, notit, bits_to_skip);
1422 status = buf_ensure_available_bits(notit);
1423 if (status != BT_MSG_ITER_STATUS_OK) {
1424 goto end;
1425 }
1426
1427 bits_to_consume = MIN(buf_available_bits(notit), bits_to_skip);
1428 BT_LOGD("Skipping %zu bits of padding: notit-addr=%p, size=%zu",
1429 bits_to_consume, notit, bits_to_consume);
1430 buf_consume_bits(notit, bits_to_consume);
1431 bits_to_skip = notit->cur_exp_packet_total_size -
1432 packet_at(notit);
1433 if (bits_to_skip == 0) {
1434 notit->state = STATE_DSCOPE_TRACE_PACKET_HEADER_BEGIN;
1435 goto end;
1436 }
1437 }
1438
1439 end:
1440 return status;
1441 }
1442
1443 static
1444 enum bt_msg_iter_status check_emit_msg_stream_beginning_state(
1445 struct bt_msg_iter *notit)
1446 {
1447 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
1448
1449 if (notit->set_stream) {
1450 status = set_current_stream(notit);
1451 if (status != BT_MSG_ITER_STATUS_OK) {
1452 goto end;
1453 }
1454 }
1455
1456 if (notit->emit_stream_begin_msg) {
1457 notit->state = STATE_EMIT_MSG_STREAM_BEGINNING;
1458 } else {
1459 /* Stream's first packet */
1460 notit->state = STATE_CHECK_EMIT_MSG_DISCARDED_EVENTS;
1461 }
1462
1463 end:
1464 return status;
1465 }
1466
1467 static
1468 enum bt_msg_iter_status check_emit_msg_discarded_events(
1469 struct bt_msg_iter *notit)
1470 {
1471 notit->state = STATE_EMIT_MSG_DISCARDED_EVENTS;
1472
1473 if (!notit->meta.sc->has_discarded_events) {
1474 notit->state = STATE_CHECK_EMIT_MSG_DISCARDED_PACKETS;
1475 goto end;
1476 }
1477
1478 if (notit->prev_packet_snapshots.discarded_events == UINT64_C(-1)) {
1479 if (notit->snapshots.discarded_events == 0 ||
1480 notit->snapshots.discarded_events == UINT64_C(-1)) {
1481 /*
1482 * Stream's first packet with no discarded
1483 * events or no information about discarded
1484 * events: do not emit.
1485 */
1486 notit->state = STATE_CHECK_EMIT_MSG_DISCARDED_PACKETS;
1487 }
1488 } else {
1489 /*
1490 * If the previous packet has a value for this counter,
1491 * then this counter is defined for the whole stream.
1492 */
1493 BT_ASSERT(notit->snapshots.discarded_events != UINT64_C(-1));
1494
1495 if (notit->snapshots.discarded_events -
1496 notit->prev_packet_snapshots.discarded_events == 0) {
1497 /*
1498 * No discarded events since previous packet: do
1499 * not emit.
1500 */
1501 notit->state = STATE_CHECK_EMIT_MSG_DISCARDED_PACKETS;
1502 }
1503 }
1504
1505 end:
1506 return BT_MSG_ITER_STATUS_OK;
1507 }
1508
1509 static
1510 enum bt_msg_iter_status check_emit_msg_discarded_packets(
1511 struct bt_msg_iter *notit)
1512 {
1513 notit->state = STATE_EMIT_MSG_DISCARDED_PACKETS;
1514
1515 if (!notit->meta.sc->has_discarded_packets) {
1516 notit->state = STATE_EMIT_MSG_PACKET_BEGINNING;
1517 goto end;
1518 }
1519
1520 if (notit->prev_packet_snapshots.packets == UINT64_C(-1)) {
1521 /*
1522 * Stream's first packet or no information about
1523 * discarded packets: do not emit. In other words, if
1524 * this is the first packet and its sequence number is
1525 * not 0, do not consider that packets were previously
1526 * lost: we might be reading a partial stream (LTTng
1527 * snapshot for example).
1528 */
1529 notit->state = STATE_EMIT_MSG_PACKET_BEGINNING;
1530 } else {
1531 /*
1532 * If the previous packet has a value for this counter,
1533 * then this counter is defined for the whole stream.
1534 */
1535 BT_ASSERT(notit->snapshots.packets != UINT64_C(-1));
1536
1537 if (notit->snapshots.packets -
1538 notit->prev_packet_snapshots.packets <= 1) {
1539 /*
1540 * No discarded packets since previous packet:
1541 * do not emit.
1542 */
1543 notit->state = STATE_EMIT_MSG_PACKET_BEGINNING;
1544 }
1545 }
1546
1547 end:
1548 return BT_MSG_ITER_STATUS_OK;
1549 }
1550
1551 static
1552 enum bt_msg_iter_status check_emit_msg_stream_activity_end(
1553 struct bt_msg_iter *notit)
1554 {
1555 if (notit->emit_stream_end_msg) {
1556 notit->state = STATE_EMIT_MSG_STREAM_ACTIVITY_END;
1557 } else {
1558 notit->state = STATE_DONE;
1559 }
1560
1561 return BT_MSG_ITER_STATUS_OK;
1562 }
1563
1564 static inline
1565 enum bt_msg_iter_status handle_state(struct bt_msg_iter *notit)
1566 {
1567 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
1568 const enum state state = notit->state;
1569
1570 BT_LOGV("Handling state: notit-addr=%p, state=%s",
1571 notit, state_string(state));
1572
1573 // TODO: optimalize!
1574 switch (state) {
1575 case STATE_INIT:
1576 notit->state = STATE_DSCOPE_TRACE_PACKET_HEADER_BEGIN;
1577 break;
1578 case STATE_DSCOPE_TRACE_PACKET_HEADER_BEGIN:
1579 status = read_packet_header_begin_state(notit);
1580 break;
1581 case STATE_DSCOPE_TRACE_PACKET_HEADER_CONTINUE:
1582 status = read_packet_header_continue_state(notit);
1583 break;
1584 case STATE_AFTER_TRACE_PACKET_HEADER:
1585 status = after_packet_header_state(notit);
1586 break;
1587 case STATE_DSCOPE_STREAM_PACKET_CONTEXT_BEGIN:
1588 status = read_packet_context_begin_state(notit);
1589 break;
1590 case STATE_DSCOPE_STREAM_PACKET_CONTEXT_CONTINUE:
1591 status = read_packet_context_continue_state(notit);
1592 break;
1593 case STATE_AFTER_STREAM_PACKET_CONTEXT:
1594 status = after_packet_context_state(notit);
1595 break;
1596 case STATE_CHECK_EMIT_MSG_STREAM_BEGINNING:
1597 status = check_emit_msg_stream_beginning_state(notit);
1598 break;
1599 case STATE_EMIT_MSG_STREAM_BEGINNING:
1600 notit->state = STATE_EMIT_MSG_STREAM_ACTIVITY_BEGINNING;
1601 break;
1602 case STATE_EMIT_MSG_STREAM_ACTIVITY_BEGINNING:
1603 notit->state = STATE_CHECK_EMIT_MSG_DISCARDED_EVENTS;
1604 break;
1605 case STATE_CHECK_EMIT_MSG_DISCARDED_EVENTS:
1606 status = check_emit_msg_discarded_events(notit);
1607 break;
1608 case STATE_EMIT_MSG_DISCARDED_EVENTS:
1609 notit->state = STATE_CHECK_EMIT_MSG_DISCARDED_PACKETS;
1610 break;
1611 case STATE_CHECK_EMIT_MSG_DISCARDED_PACKETS:
1612 status = check_emit_msg_discarded_packets(notit);
1613 break;
1614 case STATE_EMIT_MSG_DISCARDED_PACKETS:
1615 notit->state = STATE_EMIT_MSG_PACKET_BEGINNING;
1616 break;
1617 case STATE_EMIT_MSG_PACKET_BEGINNING:
1618 notit->state = STATE_DSCOPE_EVENT_HEADER_BEGIN;
1619 break;
1620 case STATE_DSCOPE_EVENT_HEADER_BEGIN:
1621 status = read_event_header_begin_state(notit);
1622 break;
1623 case STATE_DSCOPE_EVENT_HEADER_CONTINUE:
1624 status = read_event_header_continue_state(notit);
1625 break;
1626 case STATE_AFTER_EVENT_HEADER:
1627 status = after_event_header_state(notit);
1628 break;
1629 case STATE_DSCOPE_EVENT_COMMON_CONTEXT_BEGIN:
1630 status = read_event_common_context_begin_state(notit);
1631 break;
1632 case STATE_DSCOPE_EVENT_COMMON_CONTEXT_CONTINUE:
1633 status = read_event_common_context_continue_state(notit);
1634 break;
1635 case STATE_DSCOPE_EVENT_SPEC_CONTEXT_BEGIN:
1636 status = read_event_spec_context_begin_state(notit);
1637 break;
1638 case STATE_DSCOPE_EVENT_SPEC_CONTEXT_CONTINUE:
1639 status = read_event_spec_context_continue_state(notit);
1640 break;
1641 case STATE_DSCOPE_EVENT_PAYLOAD_BEGIN:
1642 status = read_event_payload_begin_state(notit);
1643 break;
1644 case STATE_DSCOPE_EVENT_PAYLOAD_CONTINUE:
1645 status = read_event_payload_continue_state(notit);
1646 break;
1647 case STATE_EMIT_MSG_EVENT:
1648 notit->state = STATE_DSCOPE_EVENT_HEADER_BEGIN;
1649 break;
1650 case STATE_SKIP_PACKET_PADDING:
1651 status = skip_packet_padding_state(notit);
1652 break;
1653 case STATE_EMIT_MSG_PACKET_END_MULTI:
1654 notit->state = STATE_SKIP_PACKET_PADDING;
1655 break;
1656 case STATE_EMIT_MSG_PACKET_END_SINGLE:
1657 notit->state = STATE_CHECK_EMIT_MSG_STREAM_ACTIVITY_END;
1658 break;
1659 case STATE_CHECK_EMIT_MSG_STREAM_ACTIVITY_END:
1660 status = check_emit_msg_stream_activity_end(notit);
1661 break;
1662 case STATE_EMIT_MSG_STREAM_ACTIVITY_END:
1663 notit->state = STATE_EMIT_MSG_STREAM_END;
1664 break;
1665 case STATE_EMIT_MSG_STREAM_END:
1666 notit->state = STATE_DONE;
1667 break;
1668 case STATE_DONE:
1669 break;
1670 default:
1671 BT_LOGF("Unknown CTF plugin message iterator state: "
1672 "notit-addr=%p, state=%d", notit, notit->state);
1673 abort();
1674 }
1675
1676 BT_LOGV("Handled state: notit-addr=%p, status=%s, "
1677 "prev-state=%s, cur-state=%s",
1678 notit, bt_msg_iter_status_string(status),
1679 state_string(state), state_string(notit->state));
1680 return status;
1681 }
1682
1683 BT_HIDDEN
1684 void bt_msg_iter_reset_for_next_stream_file(struct bt_msg_iter *notit)
1685 {
1686 BT_ASSERT(notit);
1687 BT_LOGD("Resetting message iterator: addr=%p", notit);
1688 stack_clear(notit->stack);
1689 notit->meta.sc = NULL;
1690 notit->meta.ec = NULL;
1691 BT_PACKET_PUT_REF_AND_RESET(notit->packet);
1692 BT_STREAM_PUT_REF_AND_RESET(notit->stream);
1693 BT_MESSAGE_PUT_REF_AND_RESET(notit->event_msg);
1694 release_all_dscopes(notit);
1695 notit->cur_dscope_field = NULL;
1696
1697 if (notit->packet_context_field) {
1698 bt_packet_context_field_release(notit->packet_context_field);
1699 notit->packet_context_field = NULL;
1700 }
1701
1702 notit->buf.addr = NULL;
1703 notit->buf.sz = 0;
1704 notit->buf.at = 0;
1705 notit->buf.last_eh_at = SIZE_MAX;
1706 notit->buf.packet_offset = 0;
1707 notit->state = STATE_INIT;
1708 notit->cur_exp_packet_content_size = -1;
1709 notit->cur_exp_packet_total_size = -1;
1710 notit->cur_packet_offset = -1;
1711 notit->cur_event_class_id = -1;
1712 notit->snapshots.beginning_clock = UINT64_C(-1);
1713 notit->snapshots.end_clock = UINT64_C(-1);
1714 }
1715
1716 /**
1717 * Resets the internal state of a CTF message iterator.
1718 */
1719 BT_HIDDEN
1720 void bt_msg_iter_reset(struct bt_msg_iter *notit)
1721 {
1722 bt_msg_iter_reset_for_next_stream_file(notit);
1723 notit->cur_stream_class_id = -1;
1724 notit->cur_data_stream_id = -1;
1725 notit->emit_stream_begin_msg = true;
1726 notit->emit_stream_end_msg = true;
1727 notit->snapshots.discarded_events = UINT64_C(-1);
1728 notit->snapshots.packets = UINT64_C(-1);
1729 notit->prev_packet_snapshots.discarded_events = UINT64_C(-1);
1730 notit->prev_packet_snapshots.packets = UINT64_C(-1);
1731 notit->prev_packet_snapshots.beginning_clock = UINT64_C(-1);
1732 notit->prev_packet_snapshots.end_clock = UINT64_C(-1);
1733 }
1734
1735 static
1736 int bt_msg_iter_switch_packet(struct bt_msg_iter *notit)
1737 {
1738 int ret = 0;
1739
1740 /*
1741 * We don't put the stream class here because we need to make
1742 * sure that all the packets processed by the same message
1743 * iterator refer to the same stream class (the first one).
1744 */
1745 BT_ASSERT(notit);
1746
1747 if (notit->cur_exp_packet_total_size != -1) {
1748 notit->cur_packet_offset += notit->cur_exp_packet_total_size;
1749 }
1750
1751 BT_LOGD("Switching packet: notit-addr=%p, cur=%zu, "
1752 "packet-offset=%" PRId64, notit, notit->buf.at,
1753 notit->cur_packet_offset);
1754 stack_clear(notit->stack);
1755 notit->meta.ec = NULL;
1756 BT_PACKET_PUT_REF_AND_RESET(notit->packet);
1757 BT_MESSAGE_PUT_REF_AND_RESET(notit->event_msg);
1758 release_all_dscopes(notit);
1759 notit->cur_dscope_field = NULL;
1760
1761 /*
1762 * Adjust current buffer so that addr points to the beginning of the new
1763 * packet.
1764 */
1765 if (notit->buf.addr) {
1766 size_t consumed_bytes = (size_t) (notit->buf.at / CHAR_BIT);
1767
1768 /* Packets are assumed to start on a byte frontier. */
1769 if (notit->buf.at % CHAR_BIT) {
1770 BT_LOGW("Cannot switch packet: current position is not a multiple of 8: "
1771 "notit-addr=%p, cur=%zu", notit, notit->buf.at);
1772 ret = -1;
1773 goto end;
1774 }
1775
1776 notit->buf.addr += consumed_bytes;
1777 notit->buf.sz -= consumed_bytes;
1778 notit->buf.at = 0;
1779 notit->buf.packet_offset = 0;
1780 BT_LOGD("Adjusted buffer: addr=%p, size=%zu",
1781 notit->buf.addr, notit->buf.sz);
1782 }
1783
1784 notit->cur_exp_packet_content_size = -1;
1785 notit->cur_exp_packet_total_size = -1;
1786 notit->cur_stream_class_id = -1;
1787 notit->cur_event_class_id = -1;
1788 notit->cur_data_stream_id = -1;
1789 notit->prev_packet_snapshots = notit->snapshots;
1790 notit->snapshots.discarded_events = UINT64_C(-1);
1791 notit->snapshots.packets = UINT64_C(-1);
1792 notit->snapshots.beginning_clock = UINT64_C(-1);
1793 notit->snapshots.end_clock = UINT64_C(-1);
1794
1795 end:
1796 return ret;
1797 }
1798
1799 static
1800 bt_field *borrow_next_field(struct bt_msg_iter *notit)
1801 {
1802 bt_field *next_field = NULL;
1803 bt_field *base_field;
1804 const bt_field_class *base_fc;
1805 size_t index;
1806
1807 BT_ASSERT(!stack_empty(notit->stack));
1808 index = stack_top(notit->stack)->index;
1809 base_field = stack_top(notit->stack)->base;
1810 BT_ASSERT(base_field);
1811 base_fc = bt_field_borrow_class_const(base_field);
1812 BT_ASSERT(base_fc);
1813
1814 switch (bt_field_class_get_type(base_fc)) {
1815 case BT_FIELD_CLASS_TYPE_STRUCTURE:
1816 {
1817 BT_ASSERT(index <
1818 bt_field_class_structure_get_member_count(
1819 bt_field_borrow_class_const(
1820 base_field)));
1821 next_field =
1822 bt_field_structure_borrow_member_field_by_index(
1823 base_field, index);
1824 break;
1825 }
1826 case BT_FIELD_CLASS_TYPE_STATIC_ARRAY:
1827 case BT_FIELD_CLASS_TYPE_DYNAMIC_ARRAY:
1828 BT_ASSERT(index < bt_field_array_get_length(base_field));
1829 next_field = bt_field_array_borrow_element_field_by_index(
1830 base_field, index);
1831 break;
1832 case BT_FIELD_CLASS_TYPE_VARIANT:
1833 BT_ASSERT(index == 0);
1834 next_field = bt_field_variant_borrow_selected_option_field(
1835 base_field);
1836 break;
1837 default:
1838 abort();
1839 }
1840
1841 BT_ASSERT(next_field);
1842 return next_field;
1843 }
1844
1845 static
1846 void update_default_clock(struct bt_msg_iter *notit, uint64_t new_val,
1847 uint64_t new_val_size)
1848 {
1849 uint64_t new_val_mask;
1850 uint64_t cur_value_masked;
1851
1852 BT_ASSERT(new_val_size > 0);
1853
1854 /*
1855 * Special case for a 64-bit new value, which is the limit
1856 * of a clock value as of this version: overwrite the
1857 * current value directly.
1858 */
1859 if (new_val_size == 64) {
1860 notit->default_clock_snapshot = new_val;
1861 goto end;
1862 }
1863
1864 new_val_mask = (1ULL << new_val_size) - 1;
1865 cur_value_masked = notit->default_clock_snapshot & new_val_mask;
1866
1867 if (new_val < cur_value_masked) {
1868 /*
1869 * It looks like a wrap happened on the number of bits
1870 * of the requested new value. Assume that the clock
1871 * value wrapped only one time.
1872 */
1873 notit->default_clock_snapshot += new_val_mask + 1;
1874 }
1875
1876 /* Clear the low bits of the current clock value. */
1877 notit->default_clock_snapshot &= ~new_val_mask;
1878
1879 /* Set the low bits of the current clock value. */
1880 notit->default_clock_snapshot |= new_val;
1881
1882 end:
1883 BT_LOGV("Updated default clock's value from integer field's value: "
1884 "value=%" PRIu64, notit->default_clock_snapshot);
1885 }
1886
1887 static
1888 enum bt_bfcr_status bfcr_unsigned_int_cb(uint64_t value,
1889 struct ctf_field_class *fc, void *data)
1890 {
1891 struct bt_msg_iter *notit = data;
1892 enum bt_bfcr_status status = BT_BFCR_STATUS_OK;
1893 bt_field *field = NULL;
1894 struct ctf_field_class_int *int_fc = (void *) fc;
1895
1896 BT_LOGV("Unsigned integer function called from BFCR: "
1897 "notit-addr=%p, bfcr-addr=%p, fc-addr=%p, "
1898 "fc-type=%d, fc-in-ir=%d, value=%" PRIu64,
1899 notit, notit->bfcr, fc, fc->type, fc->in_ir, value);
1900
1901 if (G_LIKELY(int_fc->meaning == CTF_FIELD_CLASS_MEANING_NONE)) {
1902 goto update_def_clock;
1903 }
1904
1905 switch (int_fc->meaning) {
1906 case CTF_FIELD_CLASS_MEANING_EVENT_CLASS_ID:
1907 notit->cur_event_class_id = value;
1908 break;
1909 case CTF_FIELD_CLASS_MEANING_DATA_STREAM_ID:
1910 notit->cur_data_stream_id = value;
1911 break;
1912 case CTF_FIELD_CLASS_MEANING_PACKET_BEGINNING_TIME:
1913 notit->snapshots.beginning_clock = value;
1914 break;
1915 case CTF_FIELD_CLASS_MEANING_PACKET_END_TIME:
1916 notit->snapshots.end_clock = value;
1917 break;
1918 case CTF_FIELD_CLASS_MEANING_STREAM_CLASS_ID:
1919 notit->cur_stream_class_id = value;
1920 break;
1921 case CTF_FIELD_CLASS_MEANING_MAGIC:
1922 if (value != 0xc1fc1fc1) {
1923 BT_LOGW("Invalid CTF magic number: notit-addr=%p, "
1924 "magic=%" PRIx64, notit, value);
1925 status = BT_BFCR_STATUS_ERROR;
1926 goto end;
1927 }
1928
1929 break;
1930 case CTF_FIELD_CLASS_MEANING_PACKET_COUNTER_SNAPSHOT:
1931 notit->snapshots.packets = value;
1932 break;
1933 case CTF_FIELD_CLASS_MEANING_DISC_EV_REC_COUNTER_SNAPSHOT:
1934 notit->snapshots.discarded_events = value;
1935 break;
1936 case CTF_FIELD_CLASS_MEANING_EXP_PACKET_TOTAL_SIZE:
1937 notit->cur_exp_packet_total_size = value;
1938 break;
1939 case CTF_FIELD_CLASS_MEANING_EXP_PACKET_CONTENT_SIZE:
1940 notit->cur_exp_packet_content_size = value;
1941 break;
1942 default:
1943 abort();
1944 }
1945
1946 update_def_clock:
1947 if (G_UNLIKELY(int_fc->mapped_clock_class)) {
1948 update_default_clock(notit, value, int_fc->base.size);
1949 }
1950
1951 if (G_UNLIKELY(int_fc->storing_index >= 0)) {
1952 g_array_index(notit->stored_values, uint64_t,
1953 (uint64_t) int_fc->storing_index) = value;
1954 }
1955
1956 if (G_UNLIKELY(!fc->in_ir)) {
1957 goto end;
1958 }
1959
1960 field = borrow_next_field(notit);
1961 BT_ASSERT(field);
1962 BT_ASSERT(bt_field_borrow_class_const(field) == fc->ir_fc);
1963 BT_ASSERT(bt_field_get_class_type(field) ==
1964 BT_FIELD_CLASS_TYPE_UNSIGNED_INTEGER ||
1965 bt_field_get_class_type(field) ==
1966 BT_FIELD_CLASS_TYPE_UNSIGNED_ENUMERATION);
1967 bt_field_unsigned_integer_set_value(field, value);
1968 stack_top(notit->stack)->index++;
1969
1970 end:
1971 return status;
1972 }
1973
1974 static
1975 enum bt_bfcr_status bfcr_unsigned_int_char_cb(uint64_t value,
1976 struct ctf_field_class *fc, void *data)
1977 {
1978 int ret;
1979 struct bt_msg_iter *notit = data;
1980 enum bt_bfcr_status status = BT_BFCR_STATUS_OK;
1981 bt_field *string_field = NULL;
1982 struct ctf_field_class_int *int_fc = (void *) fc;
1983 char str[2] = {'\0', '\0'};
1984
1985 BT_LOGV("Unsigned integer character function called from BFCR: "
1986 "notit-addr=%p, bfcr-addr=%p, fc-addr=%p, "
1987 "fc-type=%d, fc-in-ir=%d, value=%" PRIu64,
1988 notit, notit->bfcr, fc, fc->type, fc->in_ir, value);
1989 BT_ASSERT(int_fc->meaning == CTF_FIELD_CLASS_MEANING_NONE);
1990 BT_ASSERT(!int_fc->mapped_clock_class);
1991 BT_ASSERT(int_fc->storing_index < 0);
1992
1993 if (G_UNLIKELY(!fc->in_ir)) {
1994 goto end;
1995 }
1996
1997 if (notit->done_filling_string) {
1998 goto end;
1999 }
2000
2001 if (value == 0) {
2002 notit->done_filling_string = true;
2003 goto end;
2004 }
2005
2006 string_field = stack_top(notit->stack)->base;
2007 BT_ASSERT(bt_field_get_class_type(string_field) ==
2008 BT_FIELD_CLASS_TYPE_STRING);
2009
2010 /* Append character */
2011 str[0] = (char) value;
2012 ret = bt_field_string_append_with_length(string_field, str, 1);
2013 if (ret) {
2014 BT_LOGE("Cannot append character to string field's value: "
2015 "notit-addr=%p, field-addr=%p, ret=%d",
2016 notit, string_field, ret);
2017 status = BT_BFCR_STATUS_ERROR;
2018 goto end;
2019 }
2020
2021 end:
2022 return status;
2023 }
2024
2025 static
2026 enum bt_bfcr_status bfcr_signed_int_cb(int64_t value,
2027 struct ctf_field_class *fc, void *data)
2028 {
2029 enum bt_bfcr_status status = BT_BFCR_STATUS_OK;
2030 bt_field *field = NULL;
2031 struct bt_msg_iter *notit = data;
2032 struct ctf_field_class_int *int_fc = (void *) fc;
2033
2034 BT_LOGV("Signed integer function called from BFCR: "
2035 "notit-addr=%p, bfcr-addr=%p, fc-addr=%p, "
2036 "fc-type=%d, fc-in-ir=%d, value=%" PRId64,
2037 notit, notit->bfcr, fc, fc->type, fc->in_ir, value);
2038 BT_ASSERT(int_fc->meaning == CTF_FIELD_CLASS_MEANING_NONE);
2039
2040 if (G_UNLIKELY(int_fc->storing_index >= 0)) {
2041 g_array_index(notit->stored_values, uint64_t,
2042 (uint64_t) int_fc->storing_index) = (uint64_t) value;
2043 }
2044
2045 if (G_UNLIKELY(!fc->in_ir)) {
2046 goto end;
2047 }
2048
2049 field = borrow_next_field(notit);
2050 BT_ASSERT(field);
2051 BT_ASSERT(bt_field_borrow_class_const(field) == fc->ir_fc);
2052 BT_ASSERT(bt_field_get_class_type(field) ==
2053 BT_FIELD_CLASS_TYPE_SIGNED_INTEGER ||
2054 bt_field_get_class_type(field) ==
2055 BT_FIELD_CLASS_TYPE_SIGNED_ENUMERATION);
2056 bt_field_signed_integer_set_value(field, value);
2057 stack_top(notit->stack)->index++;
2058
2059 end:
2060 return status;
2061 }
2062
2063 static
2064 enum bt_bfcr_status bfcr_floating_point_cb(double value,
2065 struct ctf_field_class *fc, void *data)
2066 {
2067 enum bt_bfcr_status status = BT_BFCR_STATUS_OK;
2068 bt_field *field = NULL;
2069 struct bt_msg_iter *notit = data;
2070
2071 BT_LOGV("Floating point number function called from BFCR: "
2072 "notit-addr=%p, bfcr-addr=%p, fc-addr=%p, "
2073 "fc-type=%d, fc-in-ir=%d, value=%f",
2074 notit, notit->bfcr, fc, fc->type, fc->in_ir, value);
2075
2076 if (G_UNLIKELY(!fc->in_ir)) {
2077 goto end;
2078 }
2079
2080 field = borrow_next_field(notit);
2081 BT_ASSERT(field);
2082 BT_ASSERT(bt_field_borrow_class_const(field) == fc->ir_fc);
2083 BT_ASSERT(bt_field_get_class_type(field) ==
2084 BT_FIELD_CLASS_TYPE_REAL);
2085 bt_field_real_set_value(field, value);
2086 stack_top(notit->stack)->index++;
2087
2088 end:
2089 return status;
2090 }
2091
2092 static
2093 enum bt_bfcr_status bfcr_string_begin_cb(
2094 struct ctf_field_class *fc, void *data)
2095 {
2096 bt_field *field = NULL;
2097 struct bt_msg_iter *notit = data;
2098 int ret;
2099
2100 BT_LOGV("String (beginning) function called from BFCR: "
2101 "notit-addr=%p, bfcr-addr=%p, fc-addr=%p, "
2102 "fc-type=%d, fc-in-ir=%d",
2103 notit, notit->bfcr, fc, fc->type, fc->in_ir);
2104
2105 if (G_UNLIKELY(!fc->in_ir)) {
2106 goto end;
2107 }
2108
2109 field = borrow_next_field(notit);
2110 BT_ASSERT(field);
2111 BT_ASSERT(bt_field_borrow_class_const(field) == fc->ir_fc);
2112 BT_ASSERT(bt_field_get_class_type(field) ==
2113 BT_FIELD_CLASS_TYPE_STRING);
2114 ret = bt_field_string_clear(field);
2115 BT_ASSERT(ret == 0);
2116
2117 /*
2118 * Push on stack. Not a compound class per se, but we know that
2119 * only bfcr_string_cb() may be called between this call and a
2120 * subsequent call to bfcr_string_end_cb().
2121 */
2122 stack_push(notit->stack, field);
2123
2124 end:
2125 return BT_BFCR_STATUS_OK;
2126 }
2127
2128 static
2129 enum bt_bfcr_status bfcr_string_cb(const char *value,
2130 size_t len, struct ctf_field_class *fc, void *data)
2131 {
2132 enum bt_bfcr_status status = BT_BFCR_STATUS_OK;
2133 bt_field *field = NULL;
2134 struct bt_msg_iter *notit = data;
2135 int ret;
2136
2137 BT_LOGV("String (substring) function called from BFCR: "
2138 "notit-addr=%p, bfcr-addr=%p, fc-addr=%p, "
2139 "fc-type=%d, fc-in-ir=%d, string-length=%zu",
2140 notit, notit->bfcr, fc, fc->type, fc->in_ir,
2141 len);
2142
2143 if (G_UNLIKELY(!fc->in_ir)) {
2144 goto end;
2145 }
2146
2147 field = stack_top(notit->stack)->base;
2148 BT_ASSERT(field);
2149
2150 /* Append current substring */
2151 ret = bt_field_string_append_with_length(field, value, len);
2152 if (ret) {
2153 BT_LOGE("Cannot append substring to string field's value: "
2154 "notit-addr=%p, field-addr=%p, string-length=%zu, "
2155 "ret=%d", notit, field, len, ret);
2156 status = BT_BFCR_STATUS_ERROR;
2157 goto end;
2158 }
2159
2160 end:
2161 return status;
2162 }
2163
2164 static
2165 enum bt_bfcr_status bfcr_string_end_cb(
2166 struct ctf_field_class *fc, void *data)
2167 {
2168 struct bt_msg_iter *notit = data;
2169
2170 BT_LOGV("String (end) function called from BFCR: "
2171 "notit-addr=%p, bfcr-addr=%p, fc-addr=%p, "
2172 "fc-type=%d, fc-in-ir=%d",
2173 notit, notit->bfcr, fc, fc->type, fc->in_ir);
2174
2175 if (G_UNLIKELY(!fc->in_ir)) {
2176 goto end;
2177 }
2178
2179 /* Pop string field */
2180 stack_pop(notit->stack);
2181
2182 /* Go to next field */
2183 stack_top(notit->stack)->index++;
2184
2185 end:
2186 return BT_BFCR_STATUS_OK;
2187 }
2188
2189 enum bt_bfcr_status bfcr_compound_begin_cb(
2190 struct ctf_field_class *fc, void *data)
2191 {
2192 struct bt_msg_iter *notit = data;
2193 bt_field *field;
2194
2195 BT_LOGV("Compound (beginning) function called from BFCR: "
2196 "notit-addr=%p, bfcr-addr=%p, fc-addr=%p, "
2197 "fc-type=%d, fc-in-ir=%d",
2198 notit, notit->bfcr, fc, fc->type, fc->in_ir);
2199
2200 if (!fc->in_ir) {
2201 goto end;
2202 }
2203
2204 /* Borrow field */
2205 if (stack_empty(notit->stack)) {
2206 /* Root: already set by read_dscope_begin_state() */
2207 field = notit->cur_dscope_field;
2208 } else {
2209 field = borrow_next_field(notit);
2210 BT_ASSERT(field);
2211 }
2212
2213 /* Push field */
2214 BT_ASSERT(field);
2215 BT_ASSERT(bt_field_borrow_class_const(field) == fc->ir_fc);
2216 stack_push(notit->stack, field);
2217
2218 /*
2219 * Change BFCR "unsigned int" callback if it's a text
2220 * array/sequence.
2221 */
2222 if (fc->type == CTF_FIELD_CLASS_TYPE_ARRAY ||
2223 fc->type == CTF_FIELD_CLASS_TYPE_SEQUENCE) {
2224 struct ctf_field_class_array_base *array_fc = (void *) fc;
2225
2226 if (array_fc->is_text) {
2227 int ret;
2228
2229 BT_ASSERT(bt_field_get_class_type(field) ==
2230 BT_FIELD_CLASS_TYPE_STRING);
2231 notit->done_filling_string = false;
2232 ret = bt_field_string_clear(field);
2233 BT_ASSERT(ret == 0);
2234 bt_bfcr_set_unsigned_int_cb(notit->bfcr,
2235 bfcr_unsigned_int_char_cb);
2236 }
2237 }
2238
2239 end:
2240 return BT_BFCR_STATUS_OK;
2241 }
2242
2243 enum bt_bfcr_status bfcr_compound_end_cb(
2244 struct ctf_field_class *fc, void *data)
2245 {
2246 struct bt_msg_iter *notit = data;
2247
2248 BT_LOGV("Compound (end) function called from BFCR: "
2249 "notit-addr=%p, bfcr-addr=%p, fc-addr=%p, "
2250 "fc-type=%d, fc-in-ir=%d",
2251 notit, notit->bfcr, fc, fc->type, fc->in_ir);
2252
2253 if (!fc->in_ir) {
2254 goto end;
2255 }
2256
2257 BT_ASSERT(!stack_empty(notit->stack));
2258 BT_ASSERT(bt_field_borrow_class_const(stack_top(notit->stack)->base) ==
2259 fc->ir_fc);
2260
2261 /*
2262 * Reset BFCR "unsigned int" callback if it's a text
2263 * array/sequence.
2264 */
2265 if (fc->type == CTF_FIELD_CLASS_TYPE_ARRAY ||
2266 fc->type == CTF_FIELD_CLASS_TYPE_SEQUENCE) {
2267 struct ctf_field_class_array_base *array_fc = (void *) fc;
2268
2269 if (array_fc->is_text) {
2270 BT_ASSERT(bt_field_get_class_type(
2271 stack_top(notit->stack)->base) ==
2272 BT_FIELD_CLASS_TYPE_STRING);
2273 bt_bfcr_set_unsigned_int_cb(notit->bfcr,
2274 bfcr_unsigned_int_cb);
2275 }
2276 }
2277
2278 /* Pop stack */
2279 stack_pop(notit->stack);
2280
2281 /* If the stack is not empty, increment the base's index */
2282 if (!stack_empty(notit->stack)) {
2283 stack_top(notit->stack)->index++;
2284 }
2285
2286 end:
2287 return BT_BFCR_STATUS_OK;
2288 }
2289
2290 static
2291 int64_t bfcr_get_sequence_length_cb(struct ctf_field_class *fc, void *data)
2292 {
2293 bt_field *seq_field;
2294 struct bt_msg_iter *notit = data;
2295 struct ctf_field_class_sequence *seq_fc = (void *) fc;
2296 int64_t length = -1;
2297 int ret;
2298
2299 length = (uint64_t) g_array_index(notit->stored_values, uint64_t,
2300 seq_fc->stored_length_index);
2301 seq_field = stack_top(notit->stack)->base;
2302 BT_ASSERT(seq_field);
2303
2304 /*
2305 * bfcr_get_sequence_length_cb() also gets called back for a
2306 * text sequence, but the destination field is a string field.
2307 * Only set the field's sequence length if the destination field
2308 * is a sequence field.
2309 */
2310 if (!seq_fc->base.is_text) {
2311 BT_ASSERT(bt_field_get_class_type(seq_field) ==
2312 BT_FIELD_CLASS_TYPE_DYNAMIC_ARRAY);
2313 ret = bt_field_dynamic_array_set_length(seq_field,
2314 (uint64_t) length);
2315 if (ret) {
2316 BT_LOGE("Cannot set dynamic array field's length field: "
2317 "notit-addr=%p, field-addr=%p, "
2318 "length=%" PRIu64, notit, seq_field, length);
2319 }
2320 }
2321
2322 return length;
2323 }
2324
2325 static
2326 struct ctf_field_class *bfcr_borrow_variant_selected_field_class_cb(
2327 struct ctf_field_class *fc, void *data)
2328 {
2329 int ret;
2330 uint64_t i;
2331 int64_t option_index = -1;
2332 struct bt_msg_iter *notit = data;
2333 struct ctf_field_class_variant *var_fc = (void *) fc;
2334 struct ctf_named_field_class *selected_option = NULL;
2335 struct ctf_field_class *ret_fc = NULL;
2336 union {
2337 uint64_t u;
2338 int64_t i;
2339 } tag;
2340
2341 /* Get variant's tag */
2342 tag.u = g_array_index(notit->stored_values, uint64_t,
2343 var_fc->stored_tag_index);
2344
2345 /*
2346 * Check each range to find the selected option's index.
2347 */
2348 if (var_fc->tag_fc->base.is_signed) {
2349 for (i = 0; i < var_fc->ranges->len; i++) {
2350 struct ctf_field_class_variant_range *range =
2351 ctf_field_class_variant_borrow_range_by_index(
2352 var_fc, i);
2353
2354 if (tag.i >= range->range.lower.i &&
2355 tag.i <= range->range.upper.i) {
2356 option_index = (int64_t) range->option_index;
2357 break;
2358 }
2359 }
2360 } else {
2361 for (i = 0; i < var_fc->ranges->len; i++) {
2362 struct ctf_field_class_variant_range *range =
2363 ctf_field_class_variant_borrow_range_by_index(
2364 var_fc, i);
2365
2366 if (tag.u >= range->range.lower.u &&
2367 tag.u <= range->range.upper.u) {
2368 option_index = (int64_t) range->option_index;
2369 break;
2370 }
2371 }
2372 }
2373
2374 if (option_index < 0) {
2375 BT_LOGW("Cannot find variant field class's option: "
2376 "notit-addr=%p, var-fc-addr=%p, u-tag=%" PRIu64 ", "
2377 "i-tag=%" PRId64, notit, var_fc, tag.u, tag.i);
2378 goto end;
2379 }
2380
2381 selected_option = ctf_field_class_variant_borrow_option_by_index(
2382 var_fc, (uint64_t) option_index);
2383
2384 if (selected_option->fc->in_ir) {
2385 bt_field *var_field = stack_top(notit->stack)->base;
2386
2387 ret = bt_field_variant_select_option_field(
2388 var_field, option_index);
2389 if (ret) {
2390 BT_LOGW("Cannot select variant field's option field: "
2391 "notit-addr=%p, var-field-addr=%p, "
2392 "opt-index=%" PRId64, notit, var_field,
2393 option_index);
2394 goto end;
2395 }
2396 }
2397
2398 ret_fc = selected_option->fc;
2399
2400 end:
2401 return ret_fc;
2402 }
2403
2404 static
2405 void create_msg_stream_beginning(struct bt_msg_iter *notit,
2406 bt_message **message)
2407 {
2408 bt_message *ret = NULL;
2409
2410 BT_ASSERT(notit->stream);
2411 BT_ASSERT(notit->msg_iter);
2412 ret = bt_message_stream_beginning_create(notit->msg_iter,
2413 notit->stream);
2414 if (!ret) {
2415 BT_LOGE("Cannot create stream beginning message: "
2416 "notit-addr=%p, stream-addr=%p",
2417 notit, notit->stream);
2418 return;
2419 }
2420
2421 *message = ret;
2422 }
2423
2424 static
2425 void create_msg_stream_activity_beginning(struct bt_msg_iter *notit,
2426 bt_message **message)
2427 {
2428 bt_message *ret = NULL;
2429
2430 BT_ASSERT(notit->stream);
2431 BT_ASSERT(notit->msg_iter);
2432 ret = bt_message_stream_activity_beginning_create(notit->msg_iter,
2433 notit->stream);
2434 if (!ret) {
2435 BT_LOGE("Cannot create stream activity beginning message: "
2436 "notit-addr=%p, stream-addr=%p",
2437 notit, notit->stream);
2438 return;
2439 }
2440
2441 *message = ret;
2442 }
2443
2444 static
2445 void create_msg_stream_activity_end(struct bt_msg_iter *notit,
2446 bt_message **message)
2447 {
2448 bt_message *ret = NULL;
2449
2450 if (!notit->stream) {
2451 BT_LOGE("Cannot create stream for stream message: "
2452 "notit-addr=%p", notit);
2453 return;
2454 }
2455
2456 BT_ASSERT(notit->stream);
2457 BT_ASSERT(notit->msg_iter);
2458 ret = bt_message_stream_activity_end_create(notit->msg_iter,
2459 notit->stream);
2460 if (!ret) {
2461 BT_LOGE("Cannot create stream activity end message: "
2462 "notit-addr=%p, stream-addr=%p",
2463 notit, notit->stream);
2464 return;
2465 }
2466
2467 *message = ret;
2468 }
2469
2470 static
2471 void create_msg_stream_end(struct bt_msg_iter *notit, bt_message **message)
2472 {
2473 bt_message *ret;
2474
2475 if (!notit->stream) {
2476 BT_LOGE("Cannot create stream for stream message: "
2477 "notit-addr=%p", notit);
2478 return;
2479 }
2480
2481 BT_ASSERT(notit->msg_iter);
2482 ret = bt_message_stream_end_create(notit->msg_iter,
2483 notit->stream);
2484 if (!ret) {
2485 BT_LOGE("Cannot create stream end message: "
2486 "notit-addr=%p, stream-addr=%p",
2487 notit, notit->stream);
2488 return;
2489 }
2490
2491 *message = ret;
2492 }
2493
2494 static
2495 void create_msg_packet_beginning(struct bt_msg_iter *notit,
2496 bt_message **message)
2497 {
2498 int ret;
2499 enum bt_msg_iter_status status;
2500 bt_message *msg = NULL;
2501 const bt_stream_class *sc;
2502
2503 status = set_current_packet(notit);
2504 if (status != BT_MSG_ITER_STATUS_OK) {
2505 goto end;
2506 }
2507
2508 BT_ASSERT(notit->packet);
2509 sc = notit->meta.sc->ir_sc;
2510 BT_ASSERT(sc);
2511
2512 if (notit->packet_context_field) {
2513 ret = bt_packet_move_context_field(
2514 notit->packet, notit->packet_context_field);
2515 if (ret) {
2516 goto end;
2517 }
2518
2519 notit->packet_context_field = NULL;
2520
2521 /*
2522 * At this point notit->dscopes.stream_packet_context
2523 * has the same value as the packet context field within
2524 * notit->packet.
2525 */
2526 BT_ASSERT(bt_packet_borrow_context_field(
2527 notit->packet) ==
2528 notit->dscopes.stream_packet_context);
2529 }
2530
2531 BT_ASSERT(notit->msg_iter);
2532
2533 if (notit->meta.sc->packets_have_ts_begin) {
2534 BT_ASSERT(notit->snapshots.beginning_clock != UINT64_C(-1));
2535 msg = bt_message_packet_beginning_create_with_default_clock_snapshot(
2536 notit->msg_iter, notit->packet,
2537 notit->snapshots.beginning_clock);
2538 } else {
2539 msg = bt_message_packet_beginning_create(notit->msg_iter,
2540 notit->packet);
2541 }
2542
2543 if (!msg) {
2544 BT_LOGE("Cannot create packet beginning message: "
2545 "notit-addr=%p, packet-addr=%p",
2546 notit, notit->packet);
2547 goto end;
2548 }
2549
2550 *message = msg;
2551
2552 end:
2553 return;
2554 }
2555
2556 static
2557 void create_msg_packet_end(struct bt_msg_iter *notit, bt_message **message)
2558 {
2559 bt_message *msg;
2560
2561 if (!notit->packet) {
2562 return;
2563 }
2564
2565 /* Update default clock from packet's end time */
2566 if (notit->snapshots.end_clock != UINT64_C(-1)) {
2567 notit->default_clock_snapshot = notit->snapshots.end_clock;
2568 }
2569
2570 BT_ASSERT(notit->msg_iter);
2571
2572 if (notit->meta.sc->packets_have_ts_end) {
2573 BT_ASSERT(notit->snapshots.end_clock != UINT64_C(-1));
2574 msg = bt_message_packet_end_create_with_default_clock_snapshot(
2575 notit->msg_iter, notit->packet,
2576 notit->snapshots.end_clock);
2577 } else {
2578 msg = bt_message_packet_end_create(notit->msg_iter,
2579 notit->packet);
2580 }
2581
2582 if (!msg) {
2583 BT_LOGE("Cannot create packet end message: "
2584 "notit-addr=%p, packet-addr=%p",
2585 notit, notit->packet);
2586 return;
2587
2588 }
2589
2590 BT_PACKET_PUT_REF_AND_RESET(notit->packet);
2591 *message = msg;
2592 }
2593
2594 static
2595 void create_msg_discarded_events(struct bt_msg_iter *notit,
2596 bt_message **message)
2597 {
2598 bt_message *msg;
2599 uint64_t beginning_raw_value = UINT64_C(-1);
2600 uint64_t end_raw_value = UINT64_C(-1);
2601
2602 BT_ASSERT(notit->msg_iter);
2603 BT_ASSERT(notit->stream);
2604 BT_ASSERT(notit->meta.sc->has_discarded_events);
2605
2606 if (notit->meta.sc->discarded_events_have_default_cs) {
2607 if (notit->prev_packet_snapshots.discarded_events == UINT64_C(-1)) {
2608 /*
2609 * We discarded events, but before (and possibly
2610 * including) the current packet: use this packet's time
2611 * range, and do not have a specific count.
2612 */
2613 beginning_raw_value = notit->snapshots.beginning_clock;
2614 end_raw_value = notit->snapshots.end_clock;
2615 } else {
2616 beginning_raw_value = notit->prev_packet_snapshots.end_clock;
2617 end_raw_value = notit->snapshots.end_clock;
2618 }
2619
2620 BT_ASSERT(beginning_raw_value != UINT64_C(-1));
2621 BT_ASSERT(end_raw_value != UINT64_C(-1));
2622 msg = bt_message_discarded_events_create_with_default_clock_snapshots(
2623 notit->msg_iter, notit->stream, beginning_raw_value,
2624 end_raw_value);
2625 } else {
2626 msg = bt_message_discarded_events_create(notit->msg_iter,
2627 notit->stream);
2628 }
2629
2630 if (!msg) {
2631 BT_LOGE("Cannot create discarded events message: "
2632 "notit-addr=%p, stream-addr=%p",
2633 notit, notit->stream);
2634 return;
2635 }
2636
2637 if (notit->prev_packet_snapshots.discarded_events != UINT64_C(-1)) {
2638 bt_message_discarded_events_set_count(msg,
2639 notit->snapshots.discarded_events -
2640 notit->prev_packet_snapshots.discarded_events);
2641 }
2642
2643 *message = msg;
2644 }
2645
2646 static
2647 void create_msg_discarded_packets(struct bt_msg_iter *notit,
2648 bt_message **message)
2649 {
2650 bt_message *msg;
2651
2652 BT_ASSERT(notit->msg_iter);
2653 BT_ASSERT(notit->stream);
2654 BT_ASSERT(notit->meta.sc->has_discarded_packets);
2655 BT_ASSERT(notit->prev_packet_snapshots.packets !=
2656 UINT64_C(-1));
2657
2658 if (notit->meta.sc->discarded_packets_have_default_cs) {
2659 BT_ASSERT(notit->prev_packet_snapshots.end_clock != UINT64_C(-1));
2660 BT_ASSERT(notit->snapshots.beginning_clock != UINT64_C(-1));
2661 msg = bt_message_discarded_packets_create_with_default_clock_snapshots(
2662 notit->msg_iter, notit->stream,
2663 notit->prev_packet_snapshots.end_clock,
2664 notit->snapshots.beginning_clock);
2665 } else {
2666 msg = bt_message_discarded_packets_create(notit->msg_iter,
2667 notit->stream);
2668 }
2669
2670 if (!msg) {
2671 BT_LOGE("Cannot create discarded packets message: "
2672 "notit-addr=%p, stream-addr=%p",
2673 notit, notit->stream);
2674 return;
2675 }
2676
2677 bt_message_discarded_packets_set_count(msg,
2678 notit->snapshots.packets -
2679 notit->prev_packet_snapshots.packets - 1);
2680 *message = msg;
2681 }
2682
2683 BT_HIDDEN
2684 struct bt_msg_iter *bt_msg_iter_create(struct ctf_trace_class *tc,
2685 size_t max_request_sz,
2686 struct bt_msg_iter_medium_ops medops, void *data)
2687 {
2688 struct bt_msg_iter *notit = NULL;
2689 struct bt_bfcr_cbs cbs = {
2690 .classes = {
2691 .signed_int = bfcr_signed_int_cb,
2692 .unsigned_int = bfcr_unsigned_int_cb,
2693 .floating_point = bfcr_floating_point_cb,
2694 .string_begin = bfcr_string_begin_cb,
2695 .string = bfcr_string_cb,
2696 .string_end = bfcr_string_end_cb,
2697 .compound_begin = bfcr_compound_begin_cb,
2698 .compound_end = bfcr_compound_end_cb,
2699 },
2700 .query = {
2701 .get_sequence_length = bfcr_get_sequence_length_cb,
2702 .borrow_variant_selected_field_class = bfcr_borrow_variant_selected_field_class_cb,
2703 },
2704 };
2705
2706 BT_ASSERT(tc);
2707 BT_ASSERT(medops.request_bytes);
2708 BT_ASSERT(medops.borrow_stream);
2709 BT_LOGD("Creating CTF plugin message iterator: "
2710 "trace-addr=%p, max-request-size=%zu, "
2711 "data=%p", tc, max_request_sz, data);
2712 notit = g_new0(struct bt_msg_iter, 1);
2713 if (!notit) {
2714 BT_LOGE_STR("Failed to allocate one CTF plugin message iterator.");
2715 goto end;
2716 }
2717 notit->meta.tc = tc;
2718 notit->medium.medops = medops;
2719 notit->medium.max_request_sz = max_request_sz;
2720 notit->medium.data = data;
2721 notit->stack = stack_new(notit);
2722 notit->stored_values = g_array_new(FALSE, TRUE, sizeof(uint64_t));
2723 g_array_set_size(notit->stored_values, tc->stored_value_count);
2724
2725 if (!notit->stack) {
2726 BT_LOGE_STR("Failed to create field stack.");
2727 goto error;
2728 }
2729
2730 notit->bfcr = bt_bfcr_create(cbs, notit);
2731 if (!notit->bfcr) {
2732 BT_LOGE_STR("Failed to create binary class reader (BFCR).");
2733 goto error;
2734 }
2735
2736 bt_msg_iter_reset(notit);
2737 BT_LOGD("Created CTF plugin message iterator: "
2738 "trace-addr=%p, max-request-size=%zu, "
2739 "data=%p, notit-addr=%p",
2740 tc, max_request_sz, data, notit);
2741 notit->cur_packet_offset = 0;
2742
2743 end:
2744 return notit;
2745
2746 error:
2747 bt_msg_iter_destroy(notit);
2748 notit = NULL;
2749 goto end;
2750 }
2751
2752 void bt_msg_iter_destroy(struct bt_msg_iter *notit)
2753 {
2754 BT_PACKET_PUT_REF_AND_RESET(notit->packet);
2755 BT_STREAM_PUT_REF_AND_RESET(notit->stream);
2756 release_all_dscopes(notit);
2757
2758 BT_LOGD("Destroying CTF plugin message iterator: addr=%p", notit);
2759
2760 if (notit->stack) {
2761 BT_LOGD_STR("Destroying field stack.");
2762 stack_destroy(notit->stack);
2763 }
2764
2765 if (notit->bfcr) {
2766 BT_LOGD("Destroying BFCR: bfcr-addr=%p", notit->bfcr);
2767 bt_bfcr_destroy(notit->bfcr);
2768 }
2769
2770 if (notit->stored_values) {
2771 g_array_free(notit->stored_values, TRUE);
2772 }
2773
2774 g_free(notit);
2775 }
2776
2777 enum bt_msg_iter_status bt_msg_iter_get_next_message(
2778 struct bt_msg_iter *notit,
2779 bt_self_message_iterator *msg_iter, bt_message **message)
2780 {
2781 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
2782
2783 BT_ASSERT(notit);
2784 BT_ASSERT(message);
2785 notit->msg_iter = msg_iter;
2786 notit->set_stream = true;
2787 BT_LOGD("Getting next message: notit-addr=%p", notit);
2788
2789 while (true) {
2790 status = handle_state(notit);
2791 if (G_UNLIKELY(status == BT_MSG_ITER_STATUS_AGAIN)) {
2792 BT_LOGD_STR("Medium returned BT_MSG_ITER_STATUS_AGAIN.");
2793 goto end;
2794 } else if (G_UNLIKELY(status != BT_MSG_ITER_STATUS_OK)) {
2795 BT_LOGW("Cannot handle state: notit-addr=%p, state=%s",
2796 notit, state_string(notit->state));
2797 goto end;
2798 }
2799
2800 switch (notit->state) {
2801 case STATE_EMIT_MSG_EVENT:
2802 BT_ASSERT(notit->event_msg);
2803 *message = notit->event_msg;
2804 notit->event_msg = NULL;
2805 goto end;
2806 case STATE_EMIT_MSG_DISCARDED_EVENTS:
2807 /* create_msg_discared_events() logs errors */
2808 create_msg_discarded_events(notit, message);
2809
2810 if (!*message) {
2811 status = BT_MSG_ITER_STATUS_ERROR;
2812 }
2813
2814 goto end;
2815 case STATE_EMIT_MSG_DISCARDED_PACKETS:
2816 /* create_msg_discared_packets() logs errors */
2817 create_msg_discarded_packets(notit, message);
2818
2819 if (!*message) {
2820 status = BT_MSG_ITER_STATUS_ERROR;
2821 }
2822
2823 goto end;
2824 case STATE_EMIT_MSG_PACKET_BEGINNING:
2825 /* create_msg_packet_beginning() logs errors */
2826 create_msg_packet_beginning(notit, message);
2827
2828 if (!*message) {
2829 status = BT_MSG_ITER_STATUS_ERROR;
2830 }
2831
2832 goto end;
2833 case STATE_EMIT_MSG_PACKET_END_SINGLE:
2834 case STATE_EMIT_MSG_PACKET_END_MULTI:
2835 /* create_msg_packet_end() logs errors */
2836 create_msg_packet_end(notit, message);
2837
2838 if (!*message) {
2839 status = BT_MSG_ITER_STATUS_ERROR;
2840 }
2841
2842 goto end;
2843 case STATE_EMIT_MSG_STREAM_ACTIVITY_BEGINNING:
2844 /* create_msg_stream_activity_beginning() logs errors */
2845 create_msg_stream_activity_beginning(notit, message);
2846
2847 if (!*message) {
2848 status = BT_MSG_ITER_STATUS_ERROR;
2849 }
2850
2851 goto end;
2852 case STATE_EMIT_MSG_STREAM_ACTIVITY_END:
2853 /* create_msg_stream_activity_end() logs errors */
2854 create_msg_stream_activity_end(notit, message);
2855
2856 if (!*message) {
2857 status = BT_MSG_ITER_STATUS_ERROR;
2858 }
2859
2860 goto end;
2861 case STATE_EMIT_MSG_STREAM_BEGINNING:
2862 /* create_msg_stream_beginning() logs errors */
2863 create_msg_stream_beginning(notit, message);
2864
2865 if (!*message) {
2866 status = BT_MSG_ITER_STATUS_ERROR;
2867 }
2868
2869 goto end;
2870 case STATE_EMIT_MSG_STREAM_END:
2871 /* create_msg_stream_end() logs errors */
2872 create_msg_stream_end(notit, message);
2873
2874 if (!*message) {
2875 status = BT_MSG_ITER_STATUS_ERROR;
2876 }
2877
2878 goto end;
2879 case STATE_DONE:
2880 status = BT_MSG_ITER_STATUS_EOF;
2881 goto end;
2882 default:
2883 /* Non-emitting state: continue */
2884 break;
2885 }
2886 }
2887
2888 end:
2889 return status;
2890 }
2891
2892 static
2893 enum bt_msg_iter_status read_packet_header_context_fields(
2894 struct bt_msg_iter *notit)
2895 {
2896 int ret;
2897 enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
2898
2899 BT_ASSERT(notit);
2900 notit->set_stream = false;
2901
2902 if (notit->state == STATE_EMIT_MSG_PACKET_BEGINNING) {
2903 /* We're already there */
2904 goto end;
2905 }
2906
2907 while (true) {
2908 status = handle_state(notit);
2909 if (G_UNLIKELY(status == BT_MSG_ITER_STATUS_AGAIN)) {
2910 BT_LOGD_STR("Medium returned BT_MSG_ITER_STATUS_AGAIN.");
2911 goto end;
2912 } else if (G_UNLIKELY(status != BT_MSG_ITER_STATUS_OK)) {
2913 BT_LOGW("Cannot handle state: notit-addr=%p, state=%s",
2914 notit, state_string(notit->state));
2915 goto end;
2916 }
2917
2918 switch (notit->state) {
2919 case STATE_EMIT_MSG_PACKET_BEGINNING:
2920 /*
2921 * Packet header and context fields are
2922 * potentially decoded (or they don't exist).
2923 */
2924 goto end;
2925 case STATE_INIT:
2926 case STATE_DSCOPE_TRACE_PACKET_HEADER_BEGIN:
2927 case STATE_DSCOPE_TRACE_PACKET_HEADER_CONTINUE:
2928 case STATE_AFTER_TRACE_PACKET_HEADER:
2929 case STATE_DSCOPE_STREAM_PACKET_CONTEXT_BEGIN:
2930 case STATE_DSCOPE_STREAM_PACKET_CONTEXT_CONTINUE:
2931 case STATE_AFTER_STREAM_PACKET_CONTEXT:
2932 case STATE_CHECK_EMIT_MSG_STREAM_BEGINNING:
2933 case STATE_EMIT_MSG_STREAM_BEGINNING:
2934 case STATE_EMIT_MSG_STREAM_ACTIVITY_BEGINNING:
2935 case STATE_CHECK_EMIT_MSG_DISCARDED_EVENTS:
2936 case STATE_EMIT_MSG_DISCARDED_EVENTS:
2937 case STATE_CHECK_EMIT_MSG_DISCARDED_PACKETS:
2938 case STATE_EMIT_MSG_DISCARDED_PACKETS:
2939 /* Non-emitting state: continue */
2940 break;
2941 default:
2942 /*
2943 * We should never get past the
2944 * STATE_EMIT_MSG_PACKET_BEGINNING state.
2945 */
2946 BT_LOGF("Unexpected state: notit-addr=%p, state=%s",
2947 notit, state_string(notit->state));
2948 abort();
2949 }
2950 }
2951
2952 end:
2953 ret = set_current_packet_content_sizes(notit);
2954 if (ret) {
2955 status = BT_MSG_ITER_STATUS_ERROR;
2956 }
2957
2958 return status;
2959 }
2960
2961 BT_HIDDEN
2962 void bt_msg_iter_set_medops_data(struct bt_msg_iter *notit,
2963 void *medops_data)
2964 {
2965 BT_ASSERT(notit);
2966 notit->medium.data = medops_data;
2967 }
2968
2969 BT_HIDDEN
2970 enum bt_msg_iter_status bt_msg_iter_seek(struct bt_msg_iter *notit,
2971 off_t offset)
2972 {
2973 enum bt_msg_iter_status ret = BT_MSG_ITER_STATUS_OK;
2974 enum bt_msg_iter_medium_status medium_status;
2975
2976 BT_ASSERT(notit);
2977 if (offset < 0) {
2978 BT_LOGE("Cannot seek to negative offset: offset=%jd", offset);
2979 ret = BT_MSG_ITER_STATUS_INVAL;
2980 goto end;
2981 }
2982
2983 if (!notit->medium.medops.seek) {
2984 ret = BT_MSG_ITER_STATUS_UNSUPPORTED;
2985 BT_LOGD("Aborting seek as the iterator's underlying media does not implement seek support.");
2986 goto end;
2987 }
2988
2989 medium_status = notit->medium.medops.seek(
2990 BT_MSG_ITER_SEEK_WHENCE_SET, offset, notit->medium.data);
2991 if (medium_status != BT_MSG_ITER_MEDIUM_STATUS_OK) {
2992 if (medium_status == BT_MSG_ITER_MEDIUM_STATUS_EOF) {
2993 ret = BT_MSG_ITER_STATUS_EOF;
2994 } else {
2995 ret = BT_MSG_ITER_STATUS_ERROR;
2996 goto end;
2997 }
2998 }
2999
3000 bt_msg_iter_reset(notit);
3001 notit->cur_packet_offset = offset;
3002
3003 end:
3004 return ret;
3005 }
3006
3007 BT_HIDDEN
3008 enum bt_msg_iter_status bt_msg_iter_get_packet_properties(
3009 struct bt_msg_iter *notit,
3010 struct bt_msg_iter_packet_properties *props)
3011 {
3012 enum bt_msg_iter_status status;
3013
3014 BT_ASSERT(notit);
3015 BT_ASSERT(props);
3016 status = read_packet_header_context_fields(notit);
3017 if (status != BT_MSG_ITER_STATUS_OK) {
3018 goto end;
3019 }
3020
3021 props->exp_packet_total_size = notit->cur_exp_packet_total_size;
3022 props->exp_packet_content_size = notit->cur_exp_packet_content_size;
3023 props->stream_class_id = (uint64_t) notit->cur_stream_class_id;
3024 props->data_stream_id = notit->cur_data_stream_id;
3025 props->snapshots.discarded_events = notit->snapshots.discarded_events;
3026 props->snapshots.packets = notit->snapshots.packets;
3027 props->snapshots.beginning_clock = notit->snapshots.beginning_clock;
3028 props->snapshots.end_clock = notit->snapshots.end_clock;
3029
3030 end:
3031 return status;
3032 }
3033
3034 BT_HIDDEN
3035 void bt_msg_iter_set_emit_stream_beginning_message(struct bt_msg_iter *notit,
3036 bool val)
3037 {
3038 notit->emit_stream_begin_msg = val;
3039 }
3040
3041 BT_HIDDEN
3042 void bt_msg_iter_set_emit_stream_end_message(struct bt_msg_iter *notit,
3043 bool val)
3044 {
3045 notit->emit_stream_end_msg = val;
3046 }
BabelTrace - Trace Format Babel Tower
This page took 0.091789 seconds and 3 git commands to generate.