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