Fix: flt.lttng-utils.debug-info: memory leak bt_dwarf_die struct
[babeltrace.git] / plugins / utils / muxer / muxer.c
CommitLineData
958f7d11
PP
1/*
2 * Copyright 2017 Philippe Proulx <pproulx@efficios.com>
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a copy
5 * of this software and associated documentation files (the "Software"), to deal
6 * in the Software without restriction, including without limitation the rights
7 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8 * copies of the Software, and to permit persons to whom the Software is
9 * furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice shall be included in
12 * all copies or substantial portions of the Software.
13 *
14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
20 * SOFTWARE.
21 */
22
fed72692
PP
23#define BT_LOG_TAG "PLUGIN-UTILS-MUXER-FLT"
24#include "logging.h"
25
958f7d11 26#include <babeltrace/babeltrace-internal.h>
282c8cd0 27#include <babeltrace/compat/uuid-internal.h>
9d408fca 28#include <babeltrace/babeltrace.h>
c6bd8523 29#include <babeltrace/value-internal.h>
fed72692 30#include <babeltrace/graph/component-internal.h>
d6e69534 31#include <babeltrace/graph/message-iterator-internal.h>
fed72692 32#include <babeltrace/graph/connection-internal.h>
958f7d11
PP
33#include <plugins-common.h>
34#include <glib.h>
c55a9f58 35#include <stdbool.h>
fed72692 36#include <inttypes.h>
f6ccaed9 37#include <babeltrace/assert-internal.h>
da91b29a 38#include <babeltrace/common-internal.h>
0fbb9a9f 39#include <stdlib.h>
282c8cd0 40#include <string.h>
958f7d11 41
fdf0b89e
FD
42#include "muxer.h"
43
c3acd5f3 44#define ASSUME_ABSOLUTE_CLOCK_CLASSES_PARAM_NAME "assume-absolute-clock-classes"
65ee897d 45
958f7d11 46struct muxer_comp {
958f7d11 47 /* Weak ref */
b19ff26f 48 bt_self_component_filter *self_comp;
d94d92ac 49
958f7d11
PP
50 unsigned int next_port_num;
51 size_t available_input_ports;
d6e69534 52 bool initializing_muxer_msg_iter;
282c8cd0 53 bool assume_absolute_clock_classes;
958f7d11
PP
54};
55
d6e69534 56struct muxer_upstream_msg_iter {
ab11110e 57 /* Owned by this, NULL if ended */
d6e69534 58 bt_self_component_port_input_message_iterator *msg_iter;
958f7d11 59
d6e69534
PP
60 /* Contains `const bt_message *`, owned by this */
61 GQueue *msgs;
958f7d11
PP
62};
63
d6e69534
PP
64enum muxer_msg_iter_clock_class_expectation {
65 MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_ANY = 0,
60f3d027 66 MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_NONE,
d6e69534
PP
67 MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_ABSOLUTE,
68 MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_NOT_ABS_SPEC_UUID,
69 MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_NOT_ABS_NO_UUID,
282c8cd0
PP
70};
71
d6e69534 72struct muxer_msg_iter {
ab11110e 73 /*
d6e69534 74 * Array of struct muxer_upstream_msg_iter * (owned by this).
ab11110e
PP
75 *
76 * NOTE: This array is searched in linearly to find the youngest
d6e69534 77 * current message. Keep this until benchmarks confirm that
ab11110e
PP
78 * another data structure is faster than this for our typical
79 * use cases.
80 */
54bdc1f7
PP
81 GPtrArray *active_muxer_upstream_msg_iters;
82
83 /*
84 * Array of struct muxer_upstream_msg_iter * (owned by this).
85 *
86 * We move ended message iterators from
87 * `active_muxer_upstream_msg_iters` to this array so as to be
88 * able to restore them when seeking.
89 */
90 GPtrArray *ended_muxer_upstream_msg_iters;
958f7d11 91
d6e69534 92 /* Last time returned in a message */
958f7d11 93 int64_t last_returned_ts_ns;
282c8cd0
PP
94
95 /* Clock class expectation state */
d6e69534 96 enum muxer_msg_iter_clock_class_expectation clock_class_expectation;
282c8cd0
PP
97
98 /*
99 * Expected clock class UUID, only valid when
100 * clock_class_expectation is
d6e69534 101 * MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_NOT_ABS_SPEC_UUID.
282c8cd0
PP
102 */
103 unsigned char expected_clock_class_uuid[BABELTRACE_UUID_LEN];
958f7d11
PP
104};
105
54bdc1f7
PP
106static
107void empty_message_queue(struct muxer_upstream_msg_iter *upstream_msg_iter)
108{
109 const bt_message *msg;
110
111 while ((msg = g_queue_pop_head(upstream_msg_iter->msgs))) {
112 bt_message_put_ref(msg);
113 }
114}
115
ab11110e 116static
d6e69534
PP
117void destroy_muxer_upstream_msg_iter(
118 struct muxer_upstream_msg_iter *muxer_upstream_msg_iter)
ab11110e 119{
d6e69534 120 if (!muxer_upstream_msg_iter) {
ab11110e
PP
121 return;
122 }
123
d6e69534
PP
124 BT_LOGD("Destroying muxer's upstream message iterator wrapper: "
125 "addr=%p, msg-iter-addr=%p, queue-len=%u",
126 muxer_upstream_msg_iter,
127 muxer_upstream_msg_iter->msg_iter,
128 muxer_upstream_msg_iter->msgs->length);
54bdc1f7
PP
129 bt_self_component_port_input_message_iterator_put_ref(
130 muxer_upstream_msg_iter->msg_iter);
d4393e08 131
d6e69534 132 if (muxer_upstream_msg_iter->msgs) {
54bdc1f7 133 empty_message_queue(muxer_upstream_msg_iter);
d6e69534 134 g_queue_free(muxer_upstream_msg_iter->msgs);
d4393e08
PP
135 }
136
d6e69534 137 g_free(muxer_upstream_msg_iter);
ab11110e
PP
138}
139
958f7d11 140static
c61018b9 141int muxer_msg_iter_add_upstream_msg_iter(struct muxer_msg_iter *muxer_msg_iter,
d6e69534 142 bt_self_component_port_input_message_iterator *self_msg_iter)
958f7d11 143{
c61018b9 144 int ret = 0;
d6e69534
PP
145 struct muxer_upstream_msg_iter *muxer_upstream_msg_iter =
146 g_new0(struct muxer_upstream_msg_iter, 1);
958f7d11 147
d6e69534
PP
148 if (!muxer_upstream_msg_iter) {
149 BT_LOGE_STR("Failed to allocate one muxer's upstream message iterator wrapper.");
6c20f4a0 150 goto error;
958f7d11
PP
151 }
152
d6e69534
PP
153 muxer_upstream_msg_iter->msg_iter = self_msg_iter;
154 bt_self_component_port_input_message_iterator_get_ref(muxer_upstream_msg_iter->msg_iter);
155 muxer_upstream_msg_iter->msgs = g_queue_new();
156 if (!muxer_upstream_msg_iter->msgs) {
d4393e08 157 BT_LOGE_STR("Failed to allocate a GQueue.");
6c20f4a0 158 goto error;
d4393e08
PP
159 }
160
54bdc1f7 161 g_ptr_array_add(muxer_msg_iter->active_muxer_upstream_msg_iters,
d6e69534
PP
162 muxer_upstream_msg_iter);
163 BT_LOGD("Added muxer's upstream message iterator wrapper: "
164 "addr=%p, muxer-msg-iter-addr=%p, msg-iter-addr=%p",
165 muxer_upstream_msg_iter, muxer_msg_iter,
166 self_msg_iter);
958f7d11 167
6c20f4a0
FD
168 goto end;
169
170error:
171 g_free(muxer_upstream_msg_iter);
c61018b9 172 ret = -1;
6c20f4a0 173
958f7d11 174end:
c61018b9 175 return ret;
958f7d11
PP
176}
177
958f7d11 178static
5badd463 179bt_self_component_status add_available_input_port(
b19ff26f 180 bt_self_component_filter *self_comp)
958f7d11 181{
d94d92ac 182 struct muxer_comp *muxer_comp = bt_self_component_get_data(
707b7d35 183 bt_self_component_filter_as_self_component(self_comp));
4cdfc5e8 184 bt_self_component_status status = BT_SELF_COMPONENT_STATUS_OK;
958f7d11 185 GString *port_name = NULL;
958f7d11 186
f6ccaed9 187 BT_ASSERT(muxer_comp);
958f7d11
PP
188 port_name = g_string_new("in");
189 if (!port_name) {
fed72692 190 BT_LOGE_STR("Failed to allocate a GString.");
d94d92ac 191 status = BT_SELF_COMPONENT_STATUS_NOMEM;
958f7d11
PP
192 goto end;
193 }
194
195 g_string_append_printf(port_name, "%u", muxer_comp->next_port_num);
d94d92ac
PP
196 status = bt_self_component_filter_add_input_port(
197 self_comp, port_name->str, NULL, NULL);
198 if (status != BT_SELF_COMPONENT_STATUS_OK) {
fed72692
PP
199 BT_LOGE("Cannot add input port to muxer component: "
200 "port-name=\"%s\", comp-addr=%p, status=%s",
d94d92ac
PP
201 port_name->str, self_comp,
202 bt_self_component_status_string(status));
958f7d11
PP
203 goto end;
204 }
205
206 muxer_comp->available_input_ports++;
207 muxer_comp->next_port_num++;
fed72692
PP
208 BT_LOGD("Added one input port to muxer component: "
209 "port-name=\"%s\", comp-addr=%p",
d94d92ac 210 port_name->str, self_comp);
5badd463 211
958f7d11
PP
212end:
213 if (port_name) {
214 g_string_free(port_name, TRUE);
215 }
216
147337a3 217 return status;
958f7d11
PP
218}
219
958f7d11 220static
4cdfc5e8 221bt_self_component_status create_output_port(
b19ff26f 222 bt_self_component_filter *self_comp)
958f7d11 223{
d94d92ac
PP
224 return bt_self_component_filter_add_output_port(
225 self_comp, "out", NULL, NULL);
958f7d11
PP
226}
227
228static
229void destroy_muxer_comp(struct muxer_comp *muxer_comp)
230{
231 if (!muxer_comp) {
232 return;
233 }
234
958f7d11
PP
235 g_free(muxer_comp);
236}
237
65ee897d 238static
b19ff26f 239bt_value *get_default_params(void)
65ee897d 240{
b19ff26f 241 bt_value *params;
65ee897d
PP
242 int ret;
243
05e21286 244 params = bt_value_map_create();
65ee897d 245 if (!params) {
fed72692 246 BT_LOGE_STR("Cannot create a map value object.");
65ee897d
PP
247 goto error;
248 }
249
05e21286 250 ret = bt_value_map_insert_bool_entry(params,
c3acd5f3 251 ASSUME_ABSOLUTE_CLOCK_CLASSES_PARAM_NAME, false);
65ee897d 252 if (ret) {
fed72692 253 BT_LOGE_STR("Cannot add boolean value to map value object.");
65ee897d
PP
254 goto error;
255 }
256
257 goto end;
258
259error:
c5b9b441 260 BT_VALUE_PUT_REF_AND_RESET(params);
65ee897d
PP
261
262end:
263 return params;
264}
265
266static
05e21286 267int configure_muxer_comp(struct muxer_comp *muxer_comp,
b19ff26f 268 const bt_value *params)
65ee897d 269{
b19ff26f
PP
270 bt_value *default_params = NULL;
271 bt_value *real_params = NULL;
272 const bt_value *assume_absolute_clock_classes = NULL;
65ee897d 273 int ret = 0;
c55a9f58 274 bt_bool bool_val;
65ee897d
PP
275
276 default_params = get_default_params();
277 if (!default_params) {
fed72692
PP
278 BT_LOGE("Cannot get default parameters: "
279 "muxer-comp-addr=%p", muxer_comp);
65ee897d
PP
280 goto error;
281 }
282
35294e83 283 ret = bt_value_map_extend(default_params, params, &real_params);
601b0d3c 284 if (ret) {
fed72692
PP
285 BT_LOGE("Cannot extend default parameters map value: "
286 "muxer-comp-addr=%p, def-params-addr=%p, "
287 "params-addr=%p", muxer_comp, default_params,
288 params);
65ee897d
PP
289 goto error;
290 }
291
05e21286
PP
292 assume_absolute_clock_classes = bt_value_map_borrow_entry_value(real_params,
293 ASSUME_ABSOLUTE_CLOCK_CLASSES_PARAM_NAME);
f6ccaed9
PP
294 if (assume_absolute_clock_classes &&
295 !bt_value_is_bool(assume_absolute_clock_classes)) {
fed72692
PP
296 BT_LOGE("Expecting a boolean value for the `%s` parameter: "
297 "muxer-comp-addr=%p, value-type=%s",
298 ASSUME_ABSOLUTE_CLOCK_CLASSES_PARAM_NAME, muxer_comp,
da91b29a 299 bt_common_value_type_string(
fed72692 300 bt_value_get_type(assume_absolute_clock_classes)));
65ee897d
PP
301 goto error;
302 }
303
601b0d3c 304 bool_val = bt_value_bool_get(assume_absolute_clock_classes);
282c8cd0 305 muxer_comp->assume_absolute_clock_classes = (bool) bool_val;
fed72692
PP
306 BT_LOGD("Configured muxer component: muxer-comp-addr=%p, "
307 "assume-absolute-clock-classes=%d",
308 muxer_comp, muxer_comp->assume_absolute_clock_classes);
65ee897d
PP
309 goto end;
310
311error:
312 ret = -1;
313
314end:
c5b9b441
PP
315 bt_value_put_ref(default_params);
316 bt_value_put_ref(real_params);
65ee897d
PP
317 return ret;
318}
319
958f7d11 320BT_HIDDEN
4cdfc5e8 321bt_self_component_status muxer_init(
b19ff26f 322 bt_self_component_filter *self_comp,
fdf0b89e 323 const bt_value *params, void *init_data)
958f7d11
PP
324{
325 int ret;
4cdfc5e8 326 bt_self_component_status status = BT_SELF_COMPONENT_STATUS_OK;
958f7d11
PP
327 struct muxer_comp *muxer_comp = g_new0(struct muxer_comp, 1);
328
fed72692 329 BT_LOGD("Initializing muxer component: "
d94d92ac 330 "comp-addr=%p, params-addr=%p", self_comp, params);
fed72692 331
958f7d11 332 if (!muxer_comp) {
fed72692 333 BT_LOGE_STR("Failed to allocate one muxer component.");
958f7d11
PP
334 goto error;
335 }
336
65ee897d
PP
337 ret = configure_muxer_comp(muxer_comp, params);
338 if (ret) {
fed72692
PP
339 BT_LOGE("Cannot configure muxer component: "
340 "muxer-comp-addr=%p, params-addr=%p",
341 muxer_comp, params);
65ee897d
PP
342 goto error;
343 }
344
d94d92ac
PP
345 muxer_comp->self_comp = self_comp;
346 bt_self_component_set_data(
707b7d35 347 bt_self_component_filter_as_self_component(self_comp),
d94d92ac 348 muxer_comp);
5badd463 349 status = add_available_input_port(self_comp);
d94d92ac 350 if (status != BT_SELF_COMPONENT_STATUS_OK) {
fed72692
PP
351 BT_LOGE("Cannot ensure that at least one muxer component's input port is available: "
352 "muxer-comp-addr=%p, status=%s",
353 muxer_comp,
d94d92ac 354 bt_self_component_status_string(status));
958f7d11
PP
355 goto error;
356 }
357
d94d92ac 358 status = create_output_port(self_comp);
fed72692
PP
359 if (status) {
360 BT_LOGE("Cannot create muxer component's output port: "
361 "muxer-comp-addr=%p, status=%s",
362 muxer_comp,
d94d92ac 363 bt_self_component_status_string(status));
958f7d11
PP
364 goto error;
365 }
366
fed72692
PP
367 BT_LOGD("Initialized muxer component: "
368 "comp-addr=%p, params-addr=%p, muxer-comp-addr=%p",
d94d92ac 369 self_comp, params, muxer_comp);
fed72692 370
958f7d11
PP
371 goto end;
372
373error:
374 destroy_muxer_comp(muxer_comp);
d94d92ac 375 bt_self_component_set_data(
707b7d35 376 bt_self_component_filter_as_self_component(self_comp),
d94d92ac 377 NULL);
147337a3 378
d94d92ac
PP
379 if (status == BT_SELF_COMPONENT_STATUS_OK) {
380 status = BT_SELF_COMPONENT_STATUS_ERROR;
147337a3 381 }
958f7d11
PP
382
383end:
384 return status;
385}
386
387BT_HIDDEN
b19ff26f 388void muxer_finalize(bt_self_component_filter *self_comp)
958f7d11 389{
d94d92ac 390 struct muxer_comp *muxer_comp = bt_self_component_get_data(
707b7d35 391 bt_self_component_filter_as_self_component(self_comp));
958f7d11 392
fed72692 393 BT_LOGD("Finalizing muxer component: comp-addr=%p",
d94d92ac 394 self_comp);
958f7d11
PP
395 destroy_muxer_comp(muxer_comp);
396}
397
398static
d6e69534 399bt_self_component_port_input_message_iterator *
5badd463 400create_msg_iter_on_input_port(bt_self_component_port_input *self_port)
958f7d11 401{
b19ff26f 402 const bt_port *port = bt_self_component_port_as_port(
707b7d35 403 bt_self_component_port_input_as_self_component_port(
d94d92ac 404 self_port));
d6e69534 405 bt_self_component_port_input_message_iterator *msg_iter =
d94d92ac 406 NULL;
958f7d11 407
f6ccaed9
PP
408 BT_ASSERT(port);
409 BT_ASSERT(bt_port_is_connected(port));
958f7d11 410
ab11110e 411 // TODO: Advance the iterator to >= the time of the latest
d6e69534 412 // returned message by the muxer message
ab11110e 413 // iterator which creates it.
d6e69534 414 msg_iter = bt_self_component_port_input_message_iterator_create(
d94d92ac 415 self_port);
d6e69534
PP
416 if (!msg_iter) {
417 BT_LOGE("Cannot create upstream message iterator on input port: "
d94d92ac
PP
418 "port-addr=%p, port-name=\"%s\"",
419 port, bt_port_get_name(port));
958f7d11
PP
420 goto end;
421 }
422
d6e69534
PP
423 BT_LOGD("Created upstream message iterator on input port: "
424 "port-addr=%p, port-name=\"%s\", msg-iter-addr=%p",
425 port, bt_port_get_name(port), msg_iter);
fed72692 426
958f7d11 427end:
d6e69534 428 return msg_iter;
958f7d11
PP
429}
430
ab11110e 431static
fdf0b89e 432bt_self_message_iterator_status muxer_upstream_msg_iter_next(
54bdc1f7
PP
433 struct muxer_upstream_msg_iter *muxer_upstream_msg_iter,
434 bool *is_ended)
ab11110e 435{
fdf0b89e
FD
436 bt_self_message_iterator_status status;
437 bt_message_iterator_status input_port_iter_status;
d6e69534 438 bt_message_array_const msgs;
d4393e08
PP
439 uint64_t i;
440 uint64_t count;
ab11110e 441
d6e69534
PP
442 BT_LOGV("Calling upstream message iterator's \"next\" method: "
443 "muxer-upstream-msg-iter-wrap-addr=%p, msg-iter-addr=%p",
444 muxer_upstream_msg_iter,
445 muxer_upstream_msg_iter->msg_iter);
fdf0b89e 446 input_port_iter_status = bt_self_component_port_input_message_iterator_next(
d6e69534
PP
447 muxer_upstream_msg_iter->msg_iter, &msgs, &count);
448 BT_LOGV("Upstream message iterator's \"next\" method returned: "
fdf0b89e 449 "status=%s", bt_message_iterator_status_string(input_port_iter_status));
ab11110e 450
fdf0b89e 451 switch (input_port_iter_status) {
d6e69534 452 case BT_MESSAGE_ITERATOR_STATUS_OK:
089717de 453 /*
d6e69534 454 * Message iterator's current message is
d4393e08 455 * valid: it must be considered for muxing operations.
089717de 456 */
d6e69534 457 BT_LOGV_STR("Validated upstream message iterator wrapper.");
d4393e08
PP
458 BT_ASSERT(count > 0);
459
d6e69534 460 /* Move messages to our queue */
d4393e08
PP
461 for (i = 0; i < count; i++) {
462 /*
463 * Push to tail in order; other side
d6e69534 464 * (muxer_msg_iter_do_next_one()) consumes
d4393e08
PP
465 * from the head first.
466 */
d6e69534
PP
467 g_queue_push_tail(muxer_upstream_msg_iter->msgs,
468 (void *) msgs[i]);
d4393e08 469 }
fdf0b89e 470 status = BT_SELF_MESSAGE_ITERATOR_STATUS_OK;
ab11110e 471 break;
d6e69534 472 case BT_MESSAGE_ITERATOR_STATUS_AGAIN:
089717de 473 /*
d6e69534 474 * Message iterator's current message is not
089717de 475 * valid anymore. Return
d6e69534 476 * BT_MESSAGE_ITERATOR_STATUS_AGAIN immediately.
089717de 477 */
fdf0b89e 478 status = BT_SELF_MESSAGE_ITERATOR_STATUS_AGAIN;
ab11110e 479 break;
d6e69534 480 case BT_MESSAGE_ITERATOR_STATUS_END: /* Fall-through. */
ab11110e 481 /*
d6e69534 482 * Message iterator reached the end: release it. It
ab11110e 483 * won't be considered again to find the youngest
d6e69534 484 * message.
ab11110e 485 */
54bdc1f7 486 *is_ended = true;
fdf0b89e 487 status = BT_SELF_MESSAGE_ITERATOR_STATUS_OK;
089717de 488 break;
ab11110e
PP
489 default:
490 /* Error or unsupported status code */
fed72692 491 BT_LOGE("Error or unsupported status code: "
fdf0b89e
FD
492 "status-code=%d", input_port_iter_status);
493 status = BT_SELF_MESSAGE_ITERATOR_STATUS_ERROR;
089717de 494 break;
ab11110e
PP
495 }
496
089717de 497 return status;
ab11110e
PP
498}
499
958f7d11 500static
d6e69534
PP
501int get_msg_ts_ns(struct muxer_comp *muxer_comp,
502 struct muxer_msg_iter *muxer_msg_iter,
503 const bt_message *msg, int64_t last_returned_ts_ns,
958f7d11
PP
504 int64_t *ts_ns)
505{
605e1019 506 const bt_clock_snapshot *clock_snapshot = NULL;
958f7d11 507 int ret = 0;
5366eb53 508 bt_message_stream_activity_clock_snapshot_state sa_cs_state;
958f7d11 509
d6e69534 510 BT_ASSERT(msg);
f6ccaed9 511 BT_ASSERT(ts_ns);
d6e69534
PP
512 BT_LOGV("Getting message's timestamp: "
513 "muxer-msg-iter-addr=%p, msg-addr=%p, "
fed72692 514 "last-returned-ts=%" PRId64,
d6e69534 515 muxer_msg_iter, msg, last_returned_ts_ns);
fed72692 516
60f3d027
PP
517 if (unlikely(muxer_msg_iter->clock_class_expectation ==
518 MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_NONE)) {
519 *ts_ns = last_returned_ts_ns;
520 goto end;
521 }
522
d6e69534
PP
523 switch (bt_message_get_type(msg)) {
524 case BT_MESSAGE_TYPE_EVENT:
60f3d027
PP
525 BT_ASSERT(bt_message_event_borrow_stream_class_default_clock_class_const(
526 msg));
0cbc2c33
PP
527 clock_snapshot = bt_message_event_borrow_default_clock_snapshot_const(
528 msg);
958f7d11 529 break;
5366eb53 530 case BT_MESSAGE_TYPE_PACKET_BEGINNING:
60f3d027
PP
531 BT_ASSERT(bt_message_packet_beginning_borrow_stream_class_default_clock_class_const(
532 msg));
0cbc2c33
PP
533 clock_snapshot = bt_message_packet_beginning_borrow_default_clock_snapshot_const(
534 msg);
5366eb53
PP
535 break;
536 case BT_MESSAGE_TYPE_PACKET_END:
60f3d027
PP
537 BT_ASSERT(bt_message_packet_end_borrow_stream_class_default_clock_class_const(
538 msg));
0cbc2c33
PP
539 clock_snapshot = bt_message_packet_end_borrow_default_clock_snapshot_const(
540 msg);
5366eb53
PP
541 break;
542 case BT_MESSAGE_TYPE_DISCARDED_EVENTS:
60f3d027
PP
543 BT_ASSERT(bt_message_discarded_events_borrow_stream_class_default_clock_class_const(
544 msg));
0cbc2c33
PP
545 clock_snapshot = bt_message_discarded_events_borrow_default_beginning_clock_snapshot_const(
546 msg);
5366eb53
PP
547 break;
548 case BT_MESSAGE_TYPE_DISCARDED_PACKETS:
60f3d027
PP
549 BT_ASSERT(bt_message_discarded_packets_borrow_stream_class_default_clock_class_const(
550 msg));
0cbc2c33
PP
551 clock_snapshot = bt_message_discarded_packets_borrow_default_beginning_clock_snapshot_const(
552 msg);
5366eb53
PP
553 break;
554 case BT_MESSAGE_TYPE_STREAM_ACTIVITY_BEGINNING:
60f3d027
PP
555 BT_ASSERT(bt_message_stream_activity_beginning_borrow_stream_class_default_clock_class_const(
556 msg));
5366eb53
PP
557 sa_cs_state = bt_message_stream_activity_beginning_borrow_default_clock_snapshot_const(
558 msg, &clock_snapshot);
559 if (sa_cs_state != BT_MESSAGE_STREAM_ACTIVITY_CLOCK_SNAPSHOT_STATE_KNOWN) {
4c0f1c8c 560 goto no_clock_snapshot;
5366eb53 561 }
958f7d11 562
5366eb53
PP
563 break;
564 case BT_MESSAGE_TYPE_STREAM_ACTIVITY_END:
60f3d027
PP
565 BT_ASSERT(bt_message_stream_activity_end_borrow_stream_class_default_clock_class_const(
566 msg));
5366eb53
PP
567 sa_cs_state = bt_message_stream_activity_end_borrow_default_clock_snapshot_const(
568 msg, &clock_snapshot);
569 if (sa_cs_state != BT_MESSAGE_STREAM_ACTIVITY_CLOCK_SNAPSHOT_STATE_KNOWN) {
4c0f1c8c 570 goto no_clock_snapshot;
5366eb53
PP
571 }
572
573 break;
b9fd9cbb 574 case BT_MESSAGE_TYPE_MESSAGE_ITERATOR_INACTIVITY:
0cbc2c33
PP
575 clock_snapshot = bt_message_message_iterator_inactivity_borrow_default_clock_snapshot_const(
576 msg);
958f7d11
PP
577 break;
578 default:
d6e69534
PP
579 /* All the other messages have a higher priority */
580 BT_LOGV_STR("Message has no timestamp: using the last returned timestamp.");
958f7d11
PP
581 *ts_ns = last_returned_ts_ns;
582 goto end;
583 }
584
60f3d027
PP
585 ret = bt_clock_snapshot_get_ns_from_origin(clock_snapshot, ts_ns);
586 if (ret) {
587 BT_LOGE("Cannot get nanoseconds from Epoch of clock snapshot: "
588 "clock-snapshot-addr=%p", clock_snapshot);
589 goto error;
590 }
591
592 goto end;
593
594no_clock_snapshot:
595 BT_LOGV_STR("Message's default clock snapshot is missing: "
596 "using the last returned timestamp.");
597 *ts_ns = last_returned_ts_ns;
598 goto end;
599
600error:
601 ret = -1;
602
603end:
604 if (ret == 0) {
605 BT_LOGV("Found message's timestamp: "
606 "muxer-msg-iter-addr=%p, msg-addr=%p, "
607 "last-returned-ts=%" PRId64 ", ts=%" PRId64,
608 muxer_msg_iter, msg, last_returned_ts_ns,
609 *ts_ns);
44c440bc
PP
610 }
611
60f3d027
PP
612 return ret;
613}
614
615static inline
616int validate_clock_class(struct muxer_msg_iter *muxer_msg_iter,
617 struct muxer_comp *muxer_comp,
618 const bt_clock_class *clock_class)
619{
620 int ret = 0;
621 const unsigned char *cc_uuid;
622 const char *cc_name;
623
e22b45d0 624 BT_ASSERT(clock_class);
50842bdc
PP
625 cc_uuid = bt_clock_class_get_uuid(clock_class);
626 cc_name = bt_clock_class_get_name(clock_class);
282c8cd0 627
d6e69534
PP
628 if (muxer_msg_iter->clock_class_expectation ==
629 MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_ANY) {
282c8cd0
PP
630 /*
631 * This is the first clock class that this muxer
d6e69534 632 * message iterator encounters. Its properties
282c8cd0
PP
633 * determine what to expect for the whole lifetime of
634 * the iterator without a true
635 * `assume-absolute-clock-classes` parameter.
636 */
5552377a 637 if (bt_clock_class_origin_is_unix_epoch(clock_class)) {
282c8cd0 638 /* Expect absolute clock classes */
d6e69534
PP
639 muxer_msg_iter->clock_class_expectation =
640 MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_ABSOLUTE;
282c8cd0
PP
641 } else {
642 if (cc_uuid) {
643 /*
644 * Expect non-absolute clock classes
645 * with a specific UUID.
646 */
d6e69534
PP
647 muxer_msg_iter->clock_class_expectation =
648 MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_NOT_ABS_SPEC_UUID;
649 memcpy(muxer_msg_iter->expected_clock_class_uuid,
282c8cd0
PP
650 cc_uuid, BABELTRACE_UUID_LEN);
651 } else {
652 /*
653 * Expect non-absolute clock classes
654 * with no UUID.
655 */
d6e69534
PP
656 muxer_msg_iter->clock_class_expectation =
657 MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_NOT_ABS_NO_UUID;
282c8cd0
PP
658 }
659 }
660 }
661
662 if (!muxer_comp->assume_absolute_clock_classes) {
d6e69534
PP
663 switch (muxer_msg_iter->clock_class_expectation) {
664 case MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_ABSOLUTE:
5552377a 665 if (!bt_clock_class_origin_is_unix_epoch(clock_class)) {
fed72692
PP
666 BT_LOGE("Expecting an absolute clock class, "
667 "but got a non-absolute one: "
668 "clock-class-addr=%p, clock-class-name=\"%s\"",
669 clock_class, cc_name);
282c8cd0
PP
670 goto error;
671 }
672 break;
d6e69534 673 case MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_NOT_ABS_NO_UUID:
5552377a 674 if (bt_clock_class_origin_is_unix_epoch(clock_class)) {
fed72692
PP
675 BT_LOGE("Expecting a non-absolute clock class with no UUID, "
676 "but got an absolute one: "
677 "clock-class-addr=%p, clock-class-name=\"%s\"",
678 clock_class, cc_name);
282c8cd0
PP
679 goto error;
680 }
681
682 if (cc_uuid) {
fed72692
PP
683 BT_LOGE("Expecting a non-absolute clock class with no UUID, "
684 "but got one with a UUID: "
685 "clock-class-addr=%p, clock-class-name=\"%s\", "
686 "uuid=\"%02x%02x%02x%02x-%02x%02x-%02x%02x-%02x%02x-%02x%02x%02x%02x%02x%02x\"",
687 clock_class, cc_name,
688 (unsigned int) cc_uuid[0],
689 (unsigned int) cc_uuid[1],
690 (unsigned int) cc_uuid[2],
691 (unsigned int) cc_uuid[3],
692 (unsigned int) cc_uuid[4],
693 (unsigned int) cc_uuid[5],
694 (unsigned int) cc_uuid[6],
695 (unsigned int) cc_uuid[7],
696 (unsigned int) cc_uuid[8],
697 (unsigned int) cc_uuid[9],
698 (unsigned int) cc_uuid[10],
699 (unsigned int) cc_uuid[11],
700 (unsigned int) cc_uuid[12],
701 (unsigned int) cc_uuid[13],
702 (unsigned int) cc_uuid[14],
703 (unsigned int) cc_uuid[15]);
282c8cd0
PP
704 goto error;
705 }
706 break;
d6e69534 707 case MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_NOT_ABS_SPEC_UUID:
5552377a 708 if (bt_clock_class_origin_is_unix_epoch(clock_class)) {
fed72692
PP
709 BT_LOGE("Expecting a non-absolute clock class with a specific UUID, "
710 "but got an absolute one: "
711 "clock-class-addr=%p, clock-class-name=\"%s\"",
712 clock_class, cc_name);
282c8cd0
PP
713 goto error;
714 }
715
716 if (!cc_uuid) {
fed72692
PP
717 BT_LOGE("Expecting a non-absolute clock class with a specific UUID, "
718 "but got one with no UUID: "
719 "clock-class-addr=%p, clock-class-name=\"%s\"",
720 clock_class, cc_name);
282c8cd0
PP
721 goto error;
722 }
723
d6e69534 724 if (memcmp(muxer_msg_iter->expected_clock_class_uuid,
282c8cd0 725 cc_uuid, BABELTRACE_UUID_LEN) != 0) {
fed72692
PP
726 BT_LOGE("Expecting a non-absolute clock class with a specific UUID, "
727 "but got one with different UUID: "
728 "clock-class-addr=%p, clock-class-name=\"%s\", "
729 "expected-uuid=\"%02x%02x%02x%02x-%02x%02x-%02x%02x-%02x%02x-%02x%02x%02x%02x%02x%02x\", "
730 "uuid=\"%02x%02x%02x%02x-%02x%02x-%02x%02x-%02x%02x-%02x%02x%02x%02x%02x%02x\"",
731 clock_class, cc_name,
d6e69534
PP
732 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[0],
733 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[1],
734 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[2],
735 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[3],
736 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[4],
737 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[5],
738 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[6],
739 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[7],
740 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[8],
741 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[9],
742 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[10],
743 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[11],
744 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[12],
745 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[13],
746 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[14],
747 (unsigned int) muxer_msg_iter->expected_clock_class_uuid[15],
fed72692
PP
748 (unsigned int) cc_uuid[0],
749 (unsigned int) cc_uuid[1],
750 (unsigned int) cc_uuid[2],
751 (unsigned int) cc_uuid[3],
752 (unsigned int) cc_uuid[4],
753 (unsigned int) cc_uuid[5],
754 (unsigned int) cc_uuid[6],
755 (unsigned int) cc_uuid[7],
756 (unsigned int) cc_uuid[8],
757 (unsigned int) cc_uuid[9],
758 (unsigned int) cc_uuid[10],
759 (unsigned int) cc_uuid[11],
760 (unsigned int) cc_uuid[12],
761 (unsigned int) cc_uuid[13],
762 (unsigned int) cc_uuid[14],
763 (unsigned int) cc_uuid[15]);
282c8cd0
PP
764 goto error;
765 }
766 break;
60f3d027
PP
767 case MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_NONE:
768 BT_LOGE("Expecting no clock class, but got one: "
769 "clock-class-addr=%p, clock-class-name=\"%s\"",
770 clock_class, cc_name);
771 goto error;
282c8cd0
PP
772 default:
773 /* Unexpected */
fed72692
PP
774 BT_LOGF("Unexpected clock class expectation: "
775 "expectation-code=%d",
d6e69534 776 muxer_msg_iter->clock_class_expectation);
282c8cd0
PP
777 abort();
778 }
958f7d11
PP
779 }
780
4c0f1c8c
PP
781 goto end;
782
958f7d11
PP
783error:
784 ret = -1;
785
786end:
60f3d027
PP
787 return ret;
788}
789
790static inline
791int validate_new_stream_clock_class(struct muxer_msg_iter *muxer_msg_iter,
792 struct muxer_comp *muxer_comp, const bt_stream *stream)
793{
794 int ret = 0;
795 const bt_stream_class *stream_class =
796 bt_stream_borrow_class_const(stream);
797 const bt_clock_class *clock_class =
798 bt_stream_class_borrow_default_clock_class_const(stream_class);
799
800 if (!clock_class) {
801 if (muxer_msg_iter->clock_class_expectation ==
802 MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_ANY) {
803 /* Expect no clock class */
804 muxer_msg_iter->clock_class_expectation =
805 MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_NONE;
806 } else {
807 BT_LOGE("Expecting stream class with a default clock class: "
808 "stream-class-addr=%p, stream-class-name=\"%s\", "
809 "stream-class-id=%" PRIu64,
810 stream_class, bt_stream_class_get_name(stream_class),
811 bt_stream_class_get_id(stream_class));
812 ret = -1;
813 }
814
815 goto end;
fed72692
PP
816 }
817
60f3d027
PP
818 ret = validate_clock_class(muxer_msg_iter, muxer_comp, clock_class);
819
820end:
958f7d11
PP
821 return ret;
822}
823
ab11110e 824/*
d6e69534
PP
825 * This function finds the youngest available message amongst the
826 * non-ended upstream message iterators and returns the upstream
827 * message iterator which has it, or
828 * BT_MESSAGE_ITERATOR_STATUS_END if there's no available
829 * message.
ab11110e
PP
830 *
831 * This function does NOT:
832 *
d6e69534 833 * * Update any upstream message iterator.
d6e69534 834 * * Check the upstream message iterators to retry.
ab11110e 835 *
d6e69534
PP
836 * On sucess, this function sets *muxer_upstream_msg_iter to the
837 * upstream message iterator of which the current message is
ab11110e
PP
838 * the youngest, and sets *ts_ns to its time.
839 */
958f7d11 840static
fdf0b89e 841bt_self_message_iterator_status
d6e69534 842muxer_msg_iter_youngest_upstream_msg_iter(
958f7d11 843 struct muxer_comp *muxer_comp,
d6e69534
PP
844 struct muxer_msg_iter *muxer_msg_iter,
845 struct muxer_upstream_msg_iter **muxer_upstream_msg_iter,
958f7d11
PP
846 int64_t *ts_ns)
847{
848 size_t i;
849 int ret;
850 int64_t youngest_ts_ns = INT64_MAX;
fdf0b89e
FD
851 bt_self_message_iterator_status status =
852 BT_SELF_MESSAGE_ITERATOR_STATUS_OK;
958f7d11 853
f6ccaed9 854 BT_ASSERT(muxer_comp);
d6e69534
PP
855 BT_ASSERT(muxer_msg_iter);
856 BT_ASSERT(muxer_upstream_msg_iter);
857 *muxer_upstream_msg_iter = NULL;
858
54bdc1f7
PP
859 for (i = 0; i < muxer_msg_iter->active_muxer_upstream_msg_iters->len;
860 i++) {
d6e69534
PP
861 const bt_message *msg;
862 struct muxer_upstream_msg_iter *cur_muxer_upstream_msg_iter =
54bdc1f7
PP
863 g_ptr_array_index(
864 muxer_msg_iter->active_muxer_upstream_msg_iters,
865 i);
d6e69534
PP
866 int64_t msg_ts_ns;
867
868 if (!cur_muxer_upstream_msg_iter->msg_iter) {
869 /* This upstream message iterator is ended */
870 BT_LOGV("Skipping ended upstream message iterator: "
871 "muxer-upstream-msg-iter-wrap-addr=%p",
872 cur_muxer_upstream_msg_iter);
958f7d11
PP
873 continue;
874 }
875
d6e69534
PP
876 BT_ASSERT(cur_muxer_upstream_msg_iter->msgs->length > 0);
877 msg = g_queue_peek_head(cur_muxer_upstream_msg_iter->msgs);
878 BT_ASSERT(msg);
60f3d027
PP
879
880 if (unlikely(bt_message_get_type(msg) ==
881 BT_MESSAGE_TYPE_STREAM_BEGINNING)) {
882 ret = validate_new_stream_clock_class(
883 muxer_msg_iter, muxer_comp,
884 bt_message_stream_beginning_borrow_stream_const(
885 msg));
886 if (ret) {
887 /*
888 * validate_new_stream_clock_class() logs
889 * errors.
890 */
891 status = BT_SELF_MESSAGE_ITERATOR_STATUS_ERROR;
892 goto end;
893 }
894 } else if (unlikely(bt_message_get_type(msg) ==
895 BT_MESSAGE_TYPE_MESSAGE_ITERATOR_INACTIVITY)) {
896 const bt_clock_snapshot *cs;
60f3d027 897
0cbc2c33
PP
898 cs = bt_message_message_iterator_inactivity_borrow_default_clock_snapshot_const(
899 msg);
60f3d027
PP
900 ret = validate_clock_class(muxer_msg_iter, muxer_comp,
901 bt_clock_snapshot_borrow_clock_class_const(cs));
902 if (ret) {
903 /* validate_clock_class() logs errors */
904 status = BT_SELF_MESSAGE_ITERATOR_STATUS_ERROR;
905 goto end;
906 }
907 }
908
d6e69534
PP
909 ret = get_msg_ts_ns(muxer_comp, muxer_msg_iter, msg,
910 muxer_msg_iter->last_returned_ts_ns, &msg_ts_ns);
958f7d11 911 if (ret) {
d6e69534
PP
912 /* get_msg_ts_ns() logs errors */
913 *muxer_upstream_msg_iter = NULL;
fdf0b89e 914 status = BT_SELF_MESSAGE_ITERATOR_STATUS_ERROR;
958f7d11
PP
915 goto end;
916 }
917
d6e69534
PP
918 if (msg_ts_ns <= youngest_ts_ns) {
919 *muxer_upstream_msg_iter =
920 cur_muxer_upstream_msg_iter;
921 youngest_ts_ns = msg_ts_ns;
958f7d11
PP
922 *ts_ns = youngest_ts_ns;
923 }
924 }
925
d6e69534 926 if (!*muxer_upstream_msg_iter) {
fdf0b89e 927 status = BT_SELF_MESSAGE_ITERATOR_STATUS_END;
958f7d11
PP
928 *ts_ns = INT64_MIN;
929 }
930
931end:
932 return status;
933}
934
935static
fdf0b89e 936bt_self_message_iterator_status validate_muxer_upstream_msg_iter(
54bdc1f7
PP
937 struct muxer_upstream_msg_iter *muxer_upstream_msg_iter,
938 bool *is_ended)
958f7d11 939{
fdf0b89e
FD
940 bt_self_message_iterator_status status =
941 BT_SELF_MESSAGE_ITERATOR_STATUS_OK;
958f7d11 942
d6e69534
PP
943 BT_LOGV("Validating muxer's upstream message iterator wrapper: "
944 "muxer-upstream-msg-iter-wrap-addr=%p",
945 muxer_upstream_msg_iter);
fed72692 946
d6e69534
PP
947 if (muxer_upstream_msg_iter->msgs->length > 0 ||
948 !muxer_upstream_msg_iter->msg_iter) {
d4393e08 949 BT_LOGV("Already valid or not considered: "
d6e69534
PP
950 "queue-len=%u, upstream-msg-iter-addr=%p",
951 muxer_upstream_msg_iter->msgs->length,
952 muxer_upstream_msg_iter->msg_iter);
ab11110e
PP
953 goto end;
954 }
955
d6e69534 956 /* muxer_upstream_msg_iter_next() logs details/errors */
54bdc1f7
PP
957 status = muxer_upstream_msg_iter_next(muxer_upstream_msg_iter,
958 is_ended);
089717de
PP
959
960end:
961 return status;
962}
963
964static
fdf0b89e 965bt_self_message_iterator_status validate_muxer_upstream_msg_iters(
d6e69534 966 struct muxer_msg_iter *muxer_msg_iter)
089717de 967{
fdf0b89e
FD
968 bt_self_message_iterator_status status =
969 BT_SELF_MESSAGE_ITERATOR_STATUS_OK;
089717de
PP
970 size_t i;
971
d6e69534
PP
972 BT_LOGV("Validating muxer's upstream message iterator wrappers: "
973 "muxer-msg-iter-addr=%p", muxer_msg_iter);
fed72692 974
54bdc1f7
PP
975 for (i = 0; i < muxer_msg_iter->active_muxer_upstream_msg_iters->len;
976 i++) {
6bf2abbd 977 bool is_ended = false;
d6e69534 978 struct muxer_upstream_msg_iter *muxer_upstream_msg_iter =
089717de 979 g_ptr_array_index(
54bdc1f7 980 muxer_msg_iter->active_muxer_upstream_msg_iters,
089717de
PP
981 i);
982
d6e69534 983 status = validate_muxer_upstream_msg_iter(
54bdc1f7 984 muxer_upstream_msg_iter, &is_ended);
fdf0b89e 985 if (status != BT_SELF_MESSAGE_ITERATOR_STATUS_OK) {
fed72692 986 if (status < 0) {
d6e69534
PP
987 BT_LOGE("Cannot validate muxer's upstream message iterator wrapper: "
988 "muxer-msg-iter-addr=%p, "
989 "muxer-upstream-msg-iter-wrap-addr=%p",
990 muxer_msg_iter,
991 muxer_upstream_msg_iter);
fed72692 992 } else {
d6e69534
PP
993 BT_LOGV("Cannot validate muxer's upstream message iterator wrapper: "
994 "muxer-msg-iter-addr=%p, "
995 "muxer-upstream-msg-iter-wrap-addr=%p",
996 muxer_msg_iter,
997 muxer_upstream_msg_iter);
fed72692
PP
998 }
999
089717de
PP
1000 goto end;
1001 }
744ba28b
PP
1002
1003 /*
54bdc1f7
PP
1004 * Move this muxer upstream message iterator to the
1005 * array of ended iterators if it's ended.
744ba28b 1006 */
54bdc1f7
PP
1007 if (unlikely(is_ended)) {
1008 BT_LOGV("Muxer's upstream message iterator wrapper: ended or canceled: "
1009 "muxer-msg-iter-addr=%p, "
1010 "muxer-upstream-msg-iter-wrap-addr=%p",
1011 muxer_msg_iter, muxer_upstream_msg_iter);
1012 g_ptr_array_add(
1013 muxer_msg_iter->ended_muxer_upstream_msg_iters,
1014 muxer_upstream_msg_iter);
1015 muxer_msg_iter->active_muxer_upstream_msg_iters->pdata[i] = NULL;
1016
744ba28b
PP
1017 /*
1018 * Use g_ptr_array_remove_fast() because the
1019 * order of those elements is not important.
1020 */
1021 g_ptr_array_remove_index_fast(
54bdc1f7 1022 muxer_msg_iter->active_muxer_upstream_msg_iters,
744ba28b
PP
1023 i);
1024 i--;
1025 }
089717de
PP
1026 }
1027
1028end:
1029 return status;
1030}
1031
d4393e08 1032static inline
fdf0b89e 1033bt_self_message_iterator_status muxer_msg_iter_do_next_one(
089717de 1034 struct muxer_comp *muxer_comp,
d6e69534
PP
1035 struct muxer_msg_iter *muxer_msg_iter,
1036 const bt_message **msg)
089717de 1037{
5badd463 1038 bt_self_message_iterator_status status;
d6e69534 1039 struct muxer_upstream_msg_iter *muxer_upstream_msg_iter = NULL;
089717de
PP
1040 int64_t next_return_ts;
1041
5badd463
PP
1042 status = validate_muxer_upstream_msg_iters(muxer_msg_iter);
1043 if (status != BT_SELF_MESSAGE_ITERATOR_STATUS_OK) {
1044 /* validate_muxer_upstream_msg_iters() logs details */
1045 goto end;
958f7d11
PP
1046 }
1047
1048 /*
089717de 1049 * At this point we know that all the existing upstream
d6e69534
PP
1050 * message iterators are valid. We can find the one,
1051 * amongst those, of which the current message is the
089717de 1052 * youngest.
958f7d11 1053 */
d6e69534
PP
1054 status = muxer_msg_iter_youngest_upstream_msg_iter(muxer_comp,
1055 muxer_msg_iter, &muxer_upstream_msg_iter,
089717de 1056 &next_return_ts);
fdf0b89e 1057 if (status < 0 || status == BT_SELF_MESSAGE_ITERATOR_STATUS_END) {
d4393e08 1058 if (status < 0) {
d6e69534 1059 BT_LOGE("Cannot find the youngest upstream message iterator wrapper: "
fed72692 1060 "status=%s",
bd1a54fe 1061 bt_common_self_message_iterator_status_string(status));
fed72692 1062 } else {
d6e69534 1063 BT_LOGV("Cannot find the youngest upstream message iterator wrapper: "
fed72692 1064 "status=%s",
bd1a54fe 1065 bt_common_self_message_iterator_status_string(status));
fed72692
PP
1066 }
1067
958f7d11
PP
1068 goto end;
1069 }
1070
d6e69534
PP
1071 if (next_return_ts < muxer_msg_iter->last_returned_ts_ns) {
1072 BT_LOGE("Youngest upstream message iterator wrapper's timestamp is less than muxer's message iterator's last returned timestamp: "
1073 "muxer-msg-iter-addr=%p, ts=%" PRId64 ", "
fed72692 1074 "last-returned-ts=%" PRId64,
d6e69534
PP
1075 muxer_msg_iter, next_return_ts,
1076 muxer_msg_iter->last_returned_ts_ns);
fdf0b89e 1077 status = BT_SELF_MESSAGE_ITERATOR_STATUS_ERROR;
958f7d11
PP
1078 goto end;
1079 }
1080
d6e69534
PP
1081 BT_LOGV("Found youngest upstream message iterator wrapper: "
1082 "muxer-msg-iter-addr=%p, "
1083 "muxer-upstream-msg-iter-wrap-addr=%p, "
fed72692 1084 "ts=%" PRId64,
d6e69534 1085 muxer_msg_iter, muxer_upstream_msg_iter, next_return_ts);
fdf0b89e 1086 BT_ASSERT(status == BT_SELF_MESSAGE_ITERATOR_STATUS_OK);
d6e69534 1087 BT_ASSERT(muxer_upstream_msg_iter);
958f7d11
PP
1088
1089 /*
d4393e08 1090 * Consume from the queue's head: other side
d6e69534 1091 * (muxer_upstream_msg_iter_next()) writes to the tail.
958f7d11 1092 */
d6e69534
PP
1093 *msg = g_queue_pop_head(muxer_upstream_msg_iter->msgs);
1094 BT_ASSERT(*msg);
1095 muxer_msg_iter->last_returned_ts_ns = next_return_ts;
958f7d11
PP
1096
1097end:
d4393e08
PP
1098 return status;
1099}
1100
1101static
fdf0b89e 1102bt_self_message_iterator_status muxer_msg_iter_do_next(
d4393e08 1103 struct muxer_comp *muxer_comp,
d6e69534
PP
1104 struct muxer_msg_iter *muxer_msg_iter,
1105 bt_message_array_const msgs, uint64_t capacity,
d4393e08
PP
1106 uint64_t *count)
1107{
fdf0b89e
FD
1108 bt_self_message_iterator_status status =
1109 BT_SELF_MESSAGE_ITERATOR_STATUS_OK;
d4393e08
PP
1110 uint64_t i = 0;
1111
4725a201 1112 while (i < capacity && status == BT_SELF_MESSAGE_ITERATOR_STATUS_OK) {
d6e69534
PP
1113 status = muxer_msg_iter_do_next_one(muxer_comp,
1114 muxer_msg_iter, &msgs[i]);
4725a201 1115 if (status == BT_SELF_MESSAGE_ITERATOR_STATUS_OK) {
d4393e08
PP
1116 i++;
1117 }
1118 }
1119
1120 if (i > 0) {
1121 /*
d6e69534 1122 * Even if muxer_msg_iter_do_next_one() returned
d4393e08 1123 * something else than
d6e69534
PP
1124 * BT_MESSAGE_ITERATOR_STATUS_OK, we accumulated
1125 * message objects in the output message
d4393e08 1126 * array, so we need to return
d6e69534 1127 * BT_MESSAGE_ITERATOR_STATUS_OK so that they are
d4393e08 1128 * transfered to downstream. This other status occurs
d6e69534 1129 * again the next time muxer_msg_iter_do_next() is
d4393e08 1130 * called, possibly without any accumulated
d6e69534 1131 * message, in which case we'll return it.
d4393e08
PP
1132 */
1133 *count = i;
fdf0b89e 1134 status = BT_SELF_MESSAGE_ITERATOR_STATUS_OK;
d4393e08
PP
1135 }
1136
1137 return status;
958f7d11
PP
1138}
1139
1140static
d6e69534 1141void destroy_muxer_msg_iter(struct muxer_msg_iter *muxer_msg_iter)
ab11110e 1142{
d6e69534 1143 if (!muxer_msg_iter) {
ab11110e
PP
1144 return;
1145 }
1146
d6e69534
PP
1147 BT_LOGD("Destroying muxer component's message iterator: "
1148 "muxer-msg-iter-addr=%p", muxer_msg_iter);
fed72692 1149
54bdc1f7
PP
1150 if (muxer_msg_iter->active_muxer_upstream_msg_iters) {
1151 BT_LOGD_STR("Destroying muxer's active upstream message iterator wrappers.");
1152 g_ptr_array_free(
1153 muxer_msg_iter->active_muxer_upstream_msg_iters, TRUE);
1154 }
1155
1156 if (muxer_msg_iter->ended_muxer_upstream_msg_iters) {
1157 BT_LOGD_STR("Destroying muxer's ended upstream message iterator wrappers.");
ab11110e 1158 g_ptr_array_free(
54bdc1f7 1159 muxer_msg_iter->ended_muxer_upstream_msg_iters, TRUE);
ab11110e
PP
1160 }
1161
d6e69534 1162 g_free(muxer_msg_iter);
ab11110e
PP
1163}
1164
1165static
5badd463 1166int muxer_msg_iter_init_upstream_iterators(struct muxer_comp *muxer_comp,
d6e69534 1167 struct muxer_msg_iter *muxer_msg_iter)
958f7d11 1168{
544d0515
PP
1169 int64_t count;
1170 int64_t i;
ab11110e 1171 int ret = 0;
958f7d11 1172
d94d92ac 1173 count = bt_component_filter_get_input_port_count(
707b7d35 1174 bt_self_component_filter_as_component_filter(
d94d92ac 1175 muxer_comp->self_comp));
544d0515 1176 if (count < 0) {
d6e69534
PP
1177 BT_LOGD("No input port to initialize for muxer component's message iterator: "
1178 "muxer-comp-addr=%p, muxer-msg-iter-addr=%p",
1179 muxer_comp, muxer_msg_iter);
ab11110e
PP
1180 goto end;
1181 }
958f7d11
PP
1182
1183 for (i = 0; i < count; i++) {
5badd463 1184 bt_self_component_port_input_message_iterator *upstream_msg_iter;
b19ff26f 1185 bt_self_component_port_input *self_port =
d94d92ac
PP
1186 bt_self_component_filter_borrow_input_port_by_index(
1187 muxer_comp->self_comp, i);
b19ff26f 1188 const bt_port *port;
958f7d11 1189
d94d92ac 1190 BT_ASSERT(self_port);
707b7d35
PP
1191 port = bt_self_component_port_as_port(
1192 bt_self_component_port_input_as_self_component_port(
d94d92ac 1193 self_port));
f6ccaed9 1194 BT_ASSERT(port);
958f7d11
PP
1195
1196 if (!bt_port_is_connected(port)) {
5badd463 1197 /* Skip non-connected port */
958f7d11
PP
1198 continue;
1199 }
1200
5badd463
PP
1201 upstream_msg_iter = create_msg_iter_on_input_port(self_port);
1202 if (!upstream_msg_iter) {
1203 /* create_msg_iter_on_input_port() logs errors */
1204 BT_ASSERT(!upstream_msg_iter);
ab11110e
PP
1205 ret = -1;
1206 goto end;
958f7d11 1207 }
fed72692 1208
c61018b9
FD
1209 ret = muxer_msg_iter_add_upstream_msg_iter(muxer_msg_iter,
1210 upstream_msg_iter);
5badd463
PP
1211 bt_self_component_port_input_message_iterator_put_ref(
1212 upstream_msg_iter);
c61018b9 1213 if (ret) {
5badd463 1214 /* muxer_msg_iter_add_upstream_msg_iter() logs errors */
5badd463
PP
1215 goto end;
1216 }
958f7d11
PP
1217 }
1218
1219end:
958f7d11
PP
1220 return ret;
1221}
1222
958f7d11 1223BT_HIDDEN
4cdfc5e8 1224bt_self_message_iterator_status muxer_msg_iter_init(
d6e69534 1225 bt_self_message_iterator *self_msg_iter,
b19ff26f
PP
1226 bt_self_component_filter *self_comp,
1227 bt_self_component_port_output *port)
958f7d11
PP
1228{
1229 struct muxer_comp *muxer_comp = NULL;
d6e69534 1230 struct muxer_msg_iter *muxer_msg_iter = NULL;
4cdfc5e8 1231 bt_self_message_iterator_status status =
fdf0b89e 1232 BT_SELF_MESSAGE_ITERATOR_STATUS_OK;
958f7d11
PP
1233 int ret;
1234
d94d92ac 1235 muxer_comp = bt_self_component_get_data(
707b7d35 1236 bt_self_component_filter_as_self_component(self_comp));
f6ccaed9 1237 BT_ASSERT(muxer_comp);
d6e69534
PP
1238 BT_LOGD("Initializing muxer component's message iterator: "
1239 "comp-addr=%p, muxer-comp-addr=%p, msg-iter-addr=%p",
1240 self_comp, muxer_comp, self_msg_iter);
a09c6b95 1241
d6e69534 1242 if (muxer_comp->initializing_muxer_msg_iter) {
a09c6b95 1243 /*
089717de 1244 * Weird, unhandled situation detected: downstream
d6e69534
PP
1245 * creates a muxer message iterator while creating
1246 * another muxer message iterator (same component).
a09c6b95 1247 */
d6e69534
PP
1248 BT_LOGE("Recursive initialization of muxer component's message iterator: "
1249 "comp-addr=%p, muxer-comp-addr=%p, msg-iter-addr=%p",
1250 self_comp, muxer_comp, self_msg_iter);
958f7d11
PP
1251 goto error;
1252 }
1253
d6e69534
PP
1254 muxer_comp->initializing_muxer_msg_iter = true;
1255 muxer_msg_iter = g_new0(struct muxer_msg_iter, 1);
1256 if (!muxer_msg_iter) {
1257 BT_LOGE_STR("Failed to allocate one muxer component's message iterator.");
ab11110e
PP
1258 goto error;
1259 }
1260
d6e69534 1261 muxer_msg_iter->last_returned_ts_ns = INT64_MIN;
54bdc1f7 1262 muxer_msg_iter->active_muxer_upstream_msg_iters =
958f7d11 1263 g_ptr_array_new_with_free_func(
d6e69534 1264 (GDestroyNotify) destroy_muxer_upstream_msg_iter);
54bdc1f7
PP
1265 if (!muxer_msg_iter->active_muxer_upstream_msg_iters) {
1266 BT_LOGE_STR("Failed to allocate a GPtrArray.");
1267 goto error;
1268 }
1269
1270 muxer_msg_iter->ended_muxer_upstream_msg_iters =
1271 g_ptr_array_new_with_free_func(
1272 (GDestroyNotify) destroy_muxer_upstream_msg_iter);
1273 if (!muxer_msg_iter->ended_muxer_upstream_msg_iters) {
fed72692 1274 BT_LOGE_STR("Failed to allocate a GPtrArray.");
958f7d11
PP
1275 goto error;
1276 }
1277
5badd463 1278 ret = muxer_msg_iter_init_upstream_iterators(muxer_comp,
d6e69534 1279 muxer_msg_iter);
a09c6b95 1280 if (ret) {
5badd463 1281 BT_LOGE("Cannot initialize connected input ports for muxer component's message iterator: "
fed72692 1282 "comp-addr=%p, muxer-comp-addr=%p, "
d6e69534
PP
1283 "muxer-msg-iter-addr=%p, msg-iter-addr=%p, ret=%d",
1284 self_comp, muxer_comp, muxer_msg_iter,
1285 self_msg_iter, ret);
a09c6b95
PP
1286 goto error;
1287 }
1288
5badd463 1289 bt_self_message_iterator_set_data(self_msg_iter, muxer_msg_iter);
d6e69534
PP
1290 BT_LOGD("Initialized muxer component's message iterator: "
1291 "comp-addr=%p, muxer-comp-addr=%p, muxer-msg-iter-addr=%p, "
1292 "msg-iter-addr=%p",
1293 self_comp, muxer_comp, muxer_msg_iter, self_msg_iter);
958f7d11
PP
1294 goto end;
1295
1296error:
d6e69534 1297 destroy_muxer_msg_iter(muxer_msg_iter);
5badd463 1298 bt_self_message_iterator_set_data(self_msg_iter, NULL);
fdf0b89e 1299 status = BT_SELF_MESSAGE_ITERATOR_STATUS_ERROR;
958f7d11
PP
1300
1301end:
d6e69534 1302 muxer_comp->initializing_muxer_msg_iter = false;
958f7d11
PP
1303 return status;
1304}
1305
1306BT_HIDDEN
54bdc1f7 1307void muxer_msg_iter_finalize(bt_self_message_iterator *self_msg_iter)
958f7d11 1308{
d6e69534
PP
1309 struct muxer_msg_iter *muxer_msg_iter =
1310 bt_self_message_iterator_get_data(self_msg_iter);
b19ff26f 1311 bt_self_component *self_comp = NULL;
958f7d11
PP
1312 struct muxer_comp *muxer_comp = NULL;
1313
d6e69534
PP
1314 self_comp = bt_self_message_iterator_borrow_component(
1315 self_msg_iter);
d94d92ac
PP
1316 BT_ASSERT(self_comp);
1317 muxer_comp = bt_self_component_get_data(self_comp);
d6e69534
PP
1318 BT_LOGD("Finalizing muxer component's message iterator: "
1319 "comp-addr=%p, muxer-comp-addr=%p, muxer-msg-iter-addr=%p, "
1320 "msg-iter-addr=%p",
1321 self_comp, muxer_comp, muxer_msg_iter, self_msg_iter);
958f7d11 1322
5badd463 1323 if (muxer_msg_iter) {
d6e69534 1324 destroy_muxer_msg_iter(muxer_msg_iter);
958f7d11 1325 }
958f7d11
PP
1326}
1327
1328BT_HIDDEN
fdf0b89e 1329bt_self_message_iterator_status muxer_msg_iter_next(
d6e69534
PP
1330 bt_self_message_iterator *self_msg_iter,
1331 bt_message_array_const msgs, uint64_t capacity,
d4393e08 1332 uint64_t *count)
958f7d11 1333{
fdf0b89e 1334 bt_self_message_iterator_status status;
d6e69534
PP
1335 struct muxer_msg_iter *muxer_msg_iter =
1336 bt_self_message_iterator_get_data(self_msg_iter);
b19ff26f 1337 bt_self_component *self_comp = NULL;
958f7d11 1338 struct muxer_comp *muxer_comp = NULL;
958f7d11 1339
d6e69534
PP
1340 BT_ASSERT(muxer_msg_iter);
1341 self_comp = bt_self_message_iterator_borrow_component(
1342 self_msg_iter);
d94d92ac
PP
1343 BT_ASSERT(self_comp);
1344 muxer_comp = bt_self_component_get_data(self_comp);
f6ccaed9 1345 BT_ASSERT(muxer_comp);
d6e69534
PP
1346 BT_LOGV("Muxer component's message iterator's \"next\" method called: "
1347 "comp-addr=%p, muxer-comp-addr=%p, muxer-msg-iter-addr=%p, "
1348 "msg-iter-addr=%p",
1349 self_comp, muxer_comp, muxer_msg_iter, self_msg_iter);
fed72692 1350
d6e69534
PP
1351 status = muxer_msg_iter_do_next(muxer_comp, muxer_msg_iter,
1352 msgs, capacity, count);
d4393e08 1353 if (status < 0) {
d6e69534
PP
1354 BT_LOGE("Cannot get next message: "
1355 "comp-addr=%p, muxer-comp-addr=%p, muxer-msg-iter-addr=%p, "
1356 "msg-iter-addr=%p, status=%s",
1357 self_comp, muxer_comp, muxer_msg_iter, self_msg_iter,
bd1a54fe 1358 bt_common_self_message_iterator_status_string(status));
fed72692 1359 } else {
d6e69534 1360 BT_LOGV("Returning from muxer component's message iterator's \"next\" method: "
d4393e08 1361 "status=%s",
bd1a54fe 1362 bt_common_self_message_iterator_status_string(status));
fed72692 1363 }
958f7d11 1364
d4393e08 1365 return status;
958f7d11
PP
1366}
1367
1368BT_HIDDEN
4cdfc5e8 1369bt_self_component_status muxer_input_port_connected(
b19ff26f
PP
1370 bt_self_component_filter *self_comp,
1371 bt_self_component_port_input *self_port,
1372 const bt_port_output *other_port)
958f7d11 1373{
5badd463 1374 bt_self_component_status status;
958f7d11 1375
5badd463
PP
1376 status = add_available_input_port(self_comp);
1377 if (status) {
06a2cb0d
PP
1378 /*
1379 * Only way to report an error later since this
1380 * method does not return anything.
1381 */
5badd463
PP
1382 BT_LOGE("Cannot add one muxer component's input port: "
1383 "status=%s",
1384 bt_self_component_status_string(status));
06a2cb0d
PP
1385 goto end;
1386 }
1387
958f7d11 1388end:
bf55043c 1389 return status;
958f7d11 1390}
b5443165 1391
54bdc1f7
PP
1392static inline
1393bt_bool muxer_upstream_msg_iters_can_all_seek_beginning(
1394 GPtrArray *muxer_upstream_msg_iters)
b5443165 1395{
b5443165
PP
1396 uint64_t i;
1397 bt_bool ret = BT_TRUE;
1398
54bdc1f7 1399 for (i = 0; i < muxer_upstream_msg_iters->len; i++) {
b5443165 1400 struct muxer_upstream_msg_iter *upstream_msg_iter =
54bdc1f7 1401 muxer_upstream_msg_iters->pdata[i];
b5443165
PP
1402
1403 if (!bt_self_component_port_input_message_iterator_can_seek_beginning(
1404 upstream_msg_iter->msg_iter)) {
1405 ret = BT_FALSE;
1406 goto end;
1407 }
1408 }
1409
1410end:
1411 return ret;
1412}
1413
54bdc1f7
PP
1414BT_HIDDEN
1415bt_bool muxer_msg_iter_can_seek_beginning(
1416 bt_self_message_iterator *self_msg_iter)
1417{
1418 struct muxer_msg_iter *muxer_msg_iter =
1419 bt_self_message_iterator_get_data(self_msg_iter);
1420 bt_bool ret = BT_TRUE;
1421
1422 if (!muxer_upstream_msg_iters_can_all_seek_beginning(
1423 muxer_msg_iter->active_muxer_upstream_msg_iters)) {
1424 ret = BT_FALSE;
1425 goto end;
1426 }
1427
1428 if (!muxer_upstream_msg_iters_can_all_seek_beginning(
1429 muxer_msg_iter->ended_muxer_upstream_msg_iters)) {
1430 ret = BT_FALSE;
1431 goto end;
1432 }
1433
1434end:
1435 return ret;
1436}
1437
b5443165
PP
1438BT_HIDDEN
1439bt_self_message_iterator_status muxer_msg_iter_seek_beginning(
1440 bt_self_message_iterator *self_msg_iter)
1441{
1442 struct muxer_msg_iter *muxer_msg_iter =
1443 bt_self_message_iterator_get_data(self_msg_iter);
ffb12c4f 1444 bt_message_iterator_status status = BT_MESSAGE_ITERATOR_STATUS_OK;
b5443165
PP
1445 uint64_t i;
1446
54bdc1f7
PP
1447 /* Seek all ended upstream iterators first */
1448 for (i = 0; i < muxer_msg_iter->ended_muxer_upstream_msg_iters->len;
1449 i++) {
b5443165 1450 struct muxer_upstream_msg_iter *upstream_msg_iter =
54bdc1f7 1451 muxer_msg_iter->ended_muxer_upstream_msg_iters->pdata[i];
b5443165
PP
1452
1453 status = bt_self_component_port_input_message_iterator_seek_beginning(
1454 upstream_msg_iter->msg_iter);
1455 if (status != BT_MESSAGE_ITERATOR_STATUS_OK) {
1456 goto end;
1457 }
54bdc1f7
PP
1458
1459 empty_message_queue(upstream_msg_iter);
1460 }
1461
1462 /* Seek all previously active upstream iterators */
1463 for (i = 0; i < muxer_msg_iter->active_muxer_upstream_msg_iters->len;
1464 i++) {
1465 struct muxer_upstream_msg_iter *upstream_msg_iter =
1466 muxer_msg_iter->active_muxer_upstream_msg_iters->pdata[i];
1467
1468 status = bt_self_component_port_input_message_iterator_seek_beginning(
1469 upstream_msg_iter->msg_iter);
1470 if (status != BT_MESSAGE_ITERATOR_STATUS_OK) {
1471 goto end;
1472 }
1473
1474 empty_message_queue(upstream_msg_iter);
1475 }
1476
1477 /* Make them all active */
1478 for (i = 0; i < muxer_msg_iter->ended_muxer_upstream_msg_iters->len;
1479 i++) {
1480 struct muxer_upstream_msg_iter *upstream_msg_iter =
1481 muxer_msg_iter->ended_muxer_upstream_msg_iters->pdata[i];
1482
1483 g_ptr_array_add(muxer_msg_iter->active_muxer_upstream_msg_iters,
1484 upstream_msg_iter);
1485 muxer_msg_iter->ended_muxer_upstream_msg_iters->pdata[i] = NULL;
b5443165
PP
1486 }
1487
54bdc1f7
PP
1488 g_ptr_array_remove_range(muxer_msg_iter->ended_muxer_upstream_msg_iters,
1489 0, muxer_msg_iter->ended_muxer_upstream_msg_iters->len);
b5443165 1490 muxer_msg_iter->last_returned_ts_ns = INT64_MIN;
54bdc1f7
PP
1491 muxer_msg_iter->clock_class_expectation =
1492 MUXER_MSG_ITER_CLOCK_CLASS_EXPECTATION_ANY;
b5443165
PP
1493
1494end:
ffb12c4f 1495 return (bt_self_message_iterator_status) status;
b5443165 1496}
This page took 0.122505 seconds and 4 git commands to generate.