#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
+#include <stdbool.h>
#include <inttypes.h>
#include <string.h>
#include <assert.h>
#include <babeltrace/graph/component-sink.h>
#include <babeltrace/graph/component-source.h>
#include <babeltrace/graph/component.h>
+#include <babeltrace/graph/connection.h>
#include <babeltrace/graph/graph.h>
#include <babeltrace/graph/notification-event.h>
#include <babeltrace/graph/notification-inactivity.h>
#include <babeltrace/graph/notification-iterator.h>
#include <babeltrace/graph/notification-packet.h>
#include <babeltrace/graph/notification-stream.h>
+#include <babeltrace/graph/output-port-notification-iterator.h>
#include <babeltrace/graph/port.h>
#include <babeltrace/graph/private-component-source.h>
#include <babeltrace/graph/private-component-sink.h>
#include <babeltrace/graph/private-component.h>
#include <babeltrace/graph/private-connection.h>
-#include <babeltrace/graph/private-notification-iterator.h>
+#include <babeltrace/graph/private-connection-private-notification-iterator.h>
#include <babeltrace/graph/private-port.h>
#include <babeltrace/plugin/plugin.h>
#include <babeltrace/ref.h>
#include "tap/tap.h"
-#define NR_TESTS 24
+#define NR_TESTS 31
enum test {
TEST_NO_AUTO_NOTIFS,
TEST_AUTO_PACKET_END_STREAM_END_FROM_END,
TEST_MULTIPLE_AUTO_STREAM_END_FROM_END,
TEST_MULTIPLE_AUTO_PACKET_END_STREAM_END_FROM_END,
+ TEST_OUTPUT_PORT_NOTIFICATION_ITERATOR,
};
enum test_event_type {
struct bt_ctf_packet *packet;
};
-struct source_muxer_sink {
- struct bt_component *source;
- struct bt_component *sink;
-};
-
static bool debug = false;
static enum test current_test;
static GArray *test_events;
assert(empty_struct_ft);
trace = bt_ctf_trace_create();
assert(trace);
- ret = bt_ctf_trace_set_native_byte_order(trace,
- BT_CTF_BYTE_ORDER_LITTLE_ENDIAN);
- assert(ret == 0);
ret = bt_ctf_trace_set_packet_header_type(trace, empty_struct_ft);
assert(ret == 0);
src_empty_cc_prio_map = bt_clock_class_priority_map_create();
static
void src_iter_finalize(
- struct bt_private_notification_iterator *private_notification_iterator)
+ struct bt_private_connection_private_notification_iterator *private_notification_iterator)
{
struct src_iter_user_data *user_data =
- bt_private_notification_iterator_get_user_data(
+ bt_private_connection_private_notification_iterator_get_user_data(
private_notification_iterator);
if (user_data) {
static
enum bt_notification_iterator_status src_iter_init(
- struct bt_private_notification_iterator *priv_notif_iter,
+ struct bt_private_connection_private_notification_iterator *priv_notif_iter,
struct bt_private_port *private_port)
{
struct src_iter_user_data *user_data =
int ret;
assert(user_data);
- ret = bt_private_notification_iterator_set_user_data(priv_notif_iter,
- user_data);
+ ret = bt_private_connection_private_notification_iterator_set_user_data(
+ priv_notif_iter, user_data);
assert(ret == 0);
switch (current_test) {
case TEST_NO_AUTO_NOTIFS:
+ case TEST_OUTPUT_PORT_NOTIFICATION_ITERATOR:
user_data->seq = seq_no_auto_notifs;
break;
case TEST_AUTO_STREAM_BEGIN_FROM_PACKET_BEGIN:
user_data->seq = seq_multiple_auto_packet_end_stream_end_from_end;
break;
default:
- assert(false);
+ abort();
}
return BT_NOTIFICATION_ITERATOR_STATUS_OK;
}
static
-struct bt_notification_iterator_next_return src_iter_next_seq(
+struct bt_notification_iterator_next_method_return src_iter_next_seq(
struct src_iter_user_data *user_data)
{
- struct bt_notification_iterator_next_return next_return = {
+ struct bt_notification_iterator_next_method_return next_return = {
.status = BT_NOTIFICATION_ITERATOR_STATUS_OK,
};
int64_t cur_ts_ns;
event_packet = src_stream2_packet2;
break;
default:
- assert(false);
+ abort();
}
if (event_packet) {
}
static
-struct bt_notification_iterator_next_return src_iter_next(
- struct bt_private_notification_iterator *priv_iterator)
+struct bt_notification_iterator_next_method_return src_iter_next(
+ struct bt_private_connection_private_notification_iterator *priv_iterator)
{
- struct bt_notification_iterator_next_return next_return = {
+ struct bt_notification_iterator_next_method_return next_return = {
.status = BT_NOTIFICATION_ITERATOR_STATUS_OK,
.notification = NULL,
};
struct src_iter_user_data *user_data =
- bt_private_notification_iterator_get_user_data(priv_iterator);
+ bt_private_connection_private_notification_iterator_get_user_data(priv_iterator);
assert(user_data);
next_return = src_iter_next_seq(user_data);
struct bt_private_component *private_component,
struct bt_value *params, void *init_method_data)
{
- void *priv_port;
+ int ret;
- priv_port = bt_private_component_source_add_output_private_port(
- private_component, "out", NULL);
- assert(priv_port);
- bt_put(priv_port);
+ ret = bt_private_component_source_add_output_private_port(
+ private_component, "out", NULL, NULL);
+ assert(ret == 0);
return BT_COMPONENT_STATUS_OK;
}
}
static
-enum bt_component_status sink_consume(
- struct bt_private_component *priv_component)
+enum bt_notification_iterator_status common_consume(
+ struct bt_notification_iterator *notif_iter)
{
- enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
+ enum bt_notification_iterator_status ret;
struct bt_notification *notification = NULL;
- struct sink_user_data *user_data =
- bt_private_component_get_user_data(priv_component);
- enum bt_notification_iterator_status it_ret;
struct test_event test_event = { 0 };
bool do_append_test_event = true;
+ assert(notif_iter);
- assert(user_data && user_data->notif_iter);
- it_ret = bt_notification_iterator_next(user_data->notif_iter);
-
- if (it_ret < 0) {
- ret = BT_COMPONENT_STATUS_ERROR;
+ ret = bt_notification_iterator_next(notif_iter);
+ if (ret < 0) {
do_append_test_event = false;
goto end;
}
- switch (it_ret) {
+ switch (ret) {
case BT_NOTIFICATION_ITERATOR_STATUS_END:
test_event.type = TEST_EV_TYPE_END;
- ret = BT_COMPONENT_STATUS_END;
- BT_PUT(user_data->notif_iter);
goto end;
case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN:
- assert(false);
+ abort();
default:
break;
}
notification = bt_notification_iterator_get_notification(
- user_data->notif_iter);
+ notif_iter);
assert(notification);
switch (bt_notification_get_type(notification)) {
return ret;
}
+static
+enum bt_component_status sink_consume(
+ struct bt_private_component *priv_component)
+{
+ enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
+ struct sink_user_data *user_data =
+ bt_private_component_get_user_data(priv_component);
+ enum bt_notification_iterator_status it_ret;
+
+ assert(user_data && user_data->notif_iter);
+ it_ret = common_consume(user_data->notif_iter);
+
+ if (it_ret < 0) {
+ ret = BT_COMPONENT_STATUS_ERROR;
+ goto end;
+ }
+
+ switch (it_ret) {
+ case BT_NOTIFICATION_ITERATOR_STATUS_END:
+ ret = BT_COMPONENT_STATUS_END;
+ BT_PUT(user_data->notif_iter);
+ goto end;
+ case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN:
+ abort();
+ default:
+ break;
+ }
+
+end:
+ return ret;
+}
+
static
void sink_port_connected(struct bt_private_component *private_component,
struct bt_private_port *self_private_port,
bt_private_port_get_private_connection(self_private_port);
struct sink_user_data *user_data = bt_private_component_get_user_data(
private_component);
+ enum bt_connection_status conn_status;
assert(user_data);
assert(priv_conn);
- user_data->notif_iter =
- bt_private_connection_create_notification_iterator(priv_conn);
- assert(user_data->notif_iter);
+ conn_status = bt_private_connection_create_notification_iterator(
+ priv_conn, NULL, &user_data->notif_iter);
+ assert(conn_status == 0);
bt_put(priv_conn);
}
{
struct sink_user_data *user_data = g_new0(struct sink_user_data, 1);
int ret;
- void *priv_port;
assert(user_data);
ret = bt_private_component_set_user_data(private_component,
user_data);
assert(ret == 0);
- priv_port = bt_private_component_sink_add_input_private_port(
- private_component, "in", NULL);
- assert(priv_port);
- bt_put(priv_port);
+ ret = bt_private_component_sink_add_input_private_port(
+ private_component, "in", NULL, NULL);
+ assert(ret == 0);
return BT_COMPONENT_STATUS_OK;
}
}
static
-void create_source_sink(struct bt_component **source,
+void create_source_sink(struct bt_graph *graph, struct bt_component **source,
struct bt_component **sink)
{
struct bt_component_class *src_comp_class;
int ret;
/* Create source component */
- src_comp_class = bt_component_class_source_create("src", src_iter_next);
- assert(src_comp_class);
- ret = bt_component_class_set_init_method(src_comp_class, src_init);
- assert(ret == 0);
- ret = bt_component_class_set_finalize_method(src_comp_class,
- src_finalize);
- assert(ret == 0);
- ret = bt_component_class_source_set_notification_iterator_init_method(
- src_comp_class, src_iter_init);
- assert(ret == 0);
- ret = bt_component_class_source_set_notification_iterator_finalize_method(
- src_comp_class, src_iter_finalize);
- assert(ret == 0);
- *source = bt_component_create(src_comp_class, "source", NULL);
- assert(*source);
+ if (source) {
+ src_comp_class = bt_component_class_source_create("src",
+ src_iter_next);
+ assert(src_comp_class);
+ ret = bt_component_class_set_init_method(src_comp_class,
+ src_init);
+ assert(ret == 0);
+ ret = bt_component_class_set_finalize_method(src_comp_class,
+ src_finalize);
+ assert(ret == 0);
+ ret = bt_component_class_source_set_notification_iterator_init_method(
+ src_comp_class, src_iter_init);
+ assert(ret == 0);
+ ret = bt_component_class_source_set_notification_iterator_finalize_method(
+ src_comp_class, src_iter_finalize);
+ assert(ret == 0);
+ ret = bt_graph_add_component(graph, src_comp_class, "source",
+ NULL, source);
+ assert(ret == 0);
+ bt_put(src_comp_class);
+ }
/* Create sink component */
- sink_comp_class = bt_component_class_sink_create("sink", sink_consume);
- assert(sink_comp_class);
- ret = bt_component_class_set_init_method(sink_comp_class, sink_init);
- assert(ret == 0);
- ret = bt_component_class_set_finalize_method(sink_comp_class,
- sink_finalize);
- ret = bt_component_class_set_port_connected_method(sink_comp_class,
- sink_port_connected);
- assert(ret == 0);
- *sink = bt_component_create(sink_comp_class, "sink", NULL);
-
- bt_put(src_comp_class);
- bt_put(sink_comp_class);
+ if (sink) {
+ sink_comp_class = bt_component_class_sink_create("sink",
+ sink_consume);
+ assert(sink_comp_class);
+ ret = bt_component_class_set_init_method(sink_comp_class,
+ sink_init);
+ assert(ret == 0);
+ ret = bt_component_class_set_finalize_method(sink_comp_class,
+ sink_finalize);
+ ret = bt_component_class_set_port_connected_method(
+ sink_comp_class, sink_port_connected);
+ assert(ret == 0);
+ ret = bt_graph_add_component(graph, sink_comp_class, "sink",
+ NULL, sink);
+ assert(ret == 0);
+ bt_put(sink_comp_class);
+ }
}
static
struct bt_port *upstream_port;
struct bt_port *downstream_port;
struct bt_graph *graph;
- void *conn;
enum bt_graph_status graph_status = BT_GRAPH_STATUS_OK;
clear_test_events();
current_test = test;
diag("test: %s", name);
- create_source_sink(&src_comp, &sink_comp);
graph = bt_graph_create();
assert(graph);
+ create_source_sink(graph, &src_comp, &sink_comp);
/* Connect source to sink */
upstream_port = bt_component_source_get_output_port_by_name(src_comp, "out");
assert(upstream_port);
downstream_port = bt_component_sink_get_input_port_by_name(sink_comp, "in");
assert(downstream_port);
- conn = bt_graph_connect_ports(graph, upstream_port, downstream_port);
- assert(conn);
- bt_put(conn);
+ graph_status = bt_graph_connect_ports(graph, upstream_port,
+ downstream_port, NULL);
bt_put(upstream_port);
bt_put(downstream_port);
"the produced sequence of test events is the expected one");
}
+static
+void test_output_port_notification_iterator(void)
+{
+ const struct test_event expected_test_events[] = {
+ { .type = TEST_EV_TYPE_NOTIF_STREAM_BEGIN, .stream = src_stream1, .packet = NULL, },
+ { .type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN, .stream = src_stream1, .packet = src_stream1_packet1, },
+ { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, },
+ { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, },
+ { .type = TEST_EV_TYPE_NOTIF_STREAM_BEGIN, .stream = src_stream2, .packet = NULL, },
+ { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, },
+ { .type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN, .stream = src_stream2, .packet = src_stream2_packet2, },
+ { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream2, .packet = src_stream2_packet2, },
+ { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, },
+ { .type = TEST_EV_TYPE_NOTIF_PACKET_END, .stream = src_stream1, .packet = src_stream1_packet1, },
+ { .type = TEST_EV_TYPE_NOTIF_PACKET_END, .stream = src_stream2, .packet = src_stream2_packet2, },
+ { .type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN, .stream = src_stream1, .packet = src_stream1_packet2, },
+ { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet2, },
+ { .type = TEST_EV_TYPE_NOTIF_STREAM_END, .stream = src_stream2, .packet = NULL, },
+ { .type = TEST_EV_TYPE_NOTIF_PACKET_END, .stream = src_stream1, .packet = src_stream1_packet2, },
+ { .type = TEST_EV_TYPE_NOTIF_STREAM_END, .stream = src_stream1, .packet = NULL, },
+ { .type = TEST_EV_TYPE_END, },
+ { .type = TEST_EV_TYPE_SENTINEL, },
+ };
+ struct bt_component *src_comp;
+ struct bt_notification_iterator *notif_iter;
+ enum bt_notification_iterator_status iter_status =
+ BT_NOTIFICATION_ITERATOR_STATUS_OK;
+ struct bt_port *upstream_port;
+ struct bt_graph *graph;
+
+ clear_test_events();
+ current_test = TEST_OUTPUT_PORT_NOTIFICATION_ITERATOR;
+ diag("test: output port notification iterator");
+ graph = bt_graph_create();
+ assert(graph);
+ create_source_sink(graph, &src_comp, NULL);
+
+ /* Create notification iterator on source's output port */
+ upstream_port = bt_component_source_get_output_port_by_name(src_comp, "out");
+ notif_iter = bt_output_port_notification_iterator_create(upstream_port,
+ NULL, NULL);
+ ok(notif_iter, "bt_output_port_notification_iterator_create() succeeds");
+ bt_put(upstream_port);
+
+ /* Consume the notification iterator */
+ while (iter_status == BT_NOTIFICATION_ITERATOR_STATUS_OK) {
+ iter_status = common_consume(notif_iter);
+ }
+
+ ok(iter_status == BT_NOTIFICATION_ITERATOR_STATUS_END,
+ "output port notification iterator finishes without any error");
+
+ /* Compare the resulting test events */
+ ok(compare_test_events(expected_test_events),
+ "the produced sequence of test events is the expected one");
+
+ bt_put(src_comp);
+ bt_put(graph);
+ bt_put(notif_iter);
+}
+
+static
+void test_output_port_notification_iterator_subscribe_events(void)
+{
+ const struct test_event expected_test_events[] = {
+ { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, },
+ { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, },
+ { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, },
+ { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream2, .packet = src_stream2_packet2, },
+ { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, },
+ { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet2, },
+ { .type = TEST_EV_TYPE_END, },
+ { .type = TEST_EV_TYPE_SENTINEL, },
+ };
+ const enum bt_notification_type notification_types[] = {
+ BT_NOTIFICATION_TYPE_EVENT,
+ BT_NOTIFICATION_TYPE_SENTINEL,
+ };
+ struct bt_component *src_comp;
+ struct bt_notification_iterator *notif_iter;
+ enum bt_notification_iterator_status iter_status =
+ BT_NOTIFICATION_ITERATOR_STATUS_OK;
+ struct bt_port *upstream_port;
+ struct bt_graph *graph;
+
+ clear_test_events();
+ current_test = TEST_OUTPUT_PORT_NOTIFICATION_ITERATOR;
+ diag("test: output port notification iterator with event subscription");
+ graph = bt_graph_create();
+ assert(graph);
+ create_source_sink(graph, &src_comp, NULL);
+
+ /* Create notification iterator on source's output port */
+ upstream_port = bt_component_source_get_output_port_by_name(src_comp, "out");
+ notif_iter = bt_output_port_notification_iterator_create(upstream_port,
+ NULL, notification_types);
+ ok(notif_iter, "bt_output_port_notification_iterator_create() succeeds");
+ bt_put(upstream_port);
+
+ /* Consume the notification iterator */
+ while (iter_status == BT_NOTIFICATION_ITERATOR_STATUS_OK) {
+ iter_status = common_consume(notif_iter);
+ }
+
+ ok(iter_status == BT_NOTIFICATION_ITERATOR_STATUS_END,
+ "output port notification iterator finishes without any error");
+
+ /* Compare the resulting test events */
+ ok(compare_test_events(expected_test_events),
+ "the produced sequence of test events is the expected one");
+
+ bt_put(src_comp);
+ bt_put(graph);
+ bt_put(notif_iter);
+}
+
+static
+void test_output_port_notification_iterator_cannot_consume(void)
+{
+ struct bt_component *src_comp;
+ struct bt_notification_iterator *notif_iter;
+ struct bt_port *upstream_port;
+ struct bt_graph *graph;
+
+ clear_test_events();
+ current_test = TEST_OUTPUT_PORT_NOTIFICATION_ITERATOR;
+ diag("test: cannot consume graph with existing output port notification iterator");
+ graph = bt_graph_create();
+ assert(graph);
+ create_source_sink(graph, &src_comp, NULL);
+
+ /* Create notification iterator on source's output port */
+ upstream_port = bt_component_source_get_output_port_by_name(src_comp, "out");
+ notif_iter = bt_output_port_notification_iterator_create(upstream_port,
+ NULL, NULL);
+ assert(notif_iter);
+ bt_put(upstream_port);
+
+ /*
+ * This should fail because the graph is now managed by the
+ * notification iterator.
+ */
+ ok(bt_graph_run(graph) == BT_GRAPH_STATUS_CANNOT_CONSUME,
+ "bt_graph_run() returns BT_GRAPH_STATUS_CANNOT_CONSUME when there's an output port notification iterator");
+
+ bt_put(src_comp);
+ bt_put(graph);
+ bt_put(notif_iter);
+}
+
#define DEBUG_ENV_VAR "TEST_BT_NOTIFICATION_ITERATOR_DEBUG"
int main(int argc, char **argv)
test_auto_packet_end_stream_end_from_end();
test_multiple_auto_stream_end_from_end();
test_multiple_auto_packet_end_stream_end_from_end();
+ test_output_port_notification_iterator();
+ test_output_port_notification_iterator_subscribe_events();
+ test_output_port_notification_iterator_cannot_consume();
fini_static_data();
return exit_status();
}
projects / babeltrace.git / blobdiff