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