Replace assert() -> BT_ASSERT() and some preconditions with BT_ASSERT_PRE()
[babeltrace.git] / plugins / utils / dummy / dummy.c
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
23 #include <babeltrace/babeltrace.h>
24 #include <babeltrace/babeltrace-internal.h>
25 #include <plugins-common.h>
26 #include <babeltrace/assert-internal.h>
27 #include "dummy.h"
28
29 void destroy_private_dummy_data(struct dummy *dummy)
30 {
31 bt_put(dummy->notif_iter);
32 g_free(dummy);
33 }
34
35 void dummy_finalize(struct bt_private_component *component)
36 {
37 struct dummy *dummy;
38
39 BT_ASSERT(component);
40 dummy = bt_private_component_get_user_data(component);
41 BT_ASSERT(dummy);
42 destroy_private_dummy_data(dummy);
43 }
44
45 enum bt_component_status dummy_init(struct bt_private_component *component,
46 struct bt_value *params, UNUSED_VAR void *init_method_data)
47 {
48 enum bt_component_status ret;
49 struct dummy *dummy = g_new0(struct dummy, 1);
50
51 if (!dummy) {
52 ret = BT_COMPONENT_STATUS_NOMEM;
53 goto end;
54 }
55
56 ret = bt_private_component_sink_add_input_private_port(component,
57 "in", NULL, NULL);
58 if (ret != BT_COMPONENT_STATUS_OK) {
59 goto end;
60 }
61
62 ret = bt_private_component_set_user_data(component, dummy);
63 if (ret != BT_COMPONENT_STATUS_OK) {
64 goto error;
65 }
66
67 goto end;
68
69 error:
70 destroy_private_dummy_data(dummy);
71
72 end:
73 return ret;
74 }
75
76 void dummy_port_connected(
77 struct bt_private_component *component,
78 struct bt_private_port *self_port,
79 struct bt_port *other_port)
80 {
81 struct dummy *dummy;
82 struct bt_notification_iterator *iterator;
83 struct bt_private_connection *connection;
84 enum bt_connection_status conn_status;
85
86 dummy = bt_private_component_get_user_data(component);
87 BT_ASSERT(dummy);
88 connection = bt_private_port_get_private_connection(self_port);
89 BT_ASSERT(connection);
90 conn_status = bt_private_connection_create_notification_iterator(
91 connection, NULL, &iterator);
92 if (conn_status != BT_CONNECTION_STATUS_OK) {
93 dummy->error = true;
94 goto end;
95 }
96
97 BT_MOVE(dummy->notif_iter, iterator);
98
99 end:
100 bt_put(connection);
101 }
102
103 enum bt_component_status dummy_consume(struct bt_private_component *component)
104 {
105 enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
106 struct bt_notification *notif = NULL;
107 struct dummy *dummy;
108 enum bt_notification_iterator_status it_ret;
109
110 dummy = bt_private_component_get_user_data(component);
111 BT_ASSERT(dummy);
112
113 if (unlikely(dummy->error)) {
114 ret = BT_COMPONENT_STATUS_ERROR;
115 goto end;
116 }
117
118 if (unlikely(!dummy->notif_iter)) {
119 ret = BT_COMPONENT_STATUS_END;
120 goto end;
121 }
122
123 /* Consume one notification */
124 it_ret = bt_notification_iterator_next(dummy->notif_iter);
125 switch (it_ret) {
126 case BT_NOTIFICATION_ITERATOR_STATUS_ERROR:
127 ret = BT_COMPONENT_STATUS_ERROR;
128 goto end;
129 case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN:
130 ret = BT_COMPONENT_STATUS_AGAIN;
131 goto end;
132 case BT_NOTIFICATION_ITERATOR_STATUS_END:
133 ret = BT_COMPONENT_STATUS_END;
134 goto end;
135 default:
136 break;
137 }
138
139 end:
140 bt_put(notif);
141 return ret;
142 }
This page took 0.032691 seconds and 4 git commands to generate.