Commit | Line | Data |
---|---|---|
e0dfa761 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 | ||
3fadfbc0 | 23 | #include <babeltrace2/babeltrace.h> |
91d81473 | 24 | #include "common/macros.h" |
578e048b | 25 | #include "common/assert.h" |
c2b71c92 JG |
26 | #include "dummy.h" |
27 | ||
5badd463 PP |
28 | static |
29 | const char * const in_port_name = "in"; | |
30 | ||
c2b71c92 JG |
31 | void destroy_private_dummy_data(struct dummy *dummy) |
32 | { | |
d6e69534 | 33 | bt_self_component_port_input_message_iterator_put_ref(dummy->msg_iter); |
c2b71c92 | 34 | g_free(dummy); |
d94d92ac | 35 | |
c2b71c92 JG |
36 | } |
37 | ||
d94d92ac | 38 | BT_HIDDEN |
b19ff26f | 39 | void dummy_finalize(bt_self_component_sink *comp) |
c2b71c92 JG |
40 | { |
41 | struct dummy *dummy; | |
42 | ||
d94d92ac PP |
43 | BT_ASSERT(comp); |
44 | dummy = bt_self_component_get_data( | |
707b7d35 | 45 | bt_self_component_sink_as_self_component(comp)); |
f6ccaed9 | 46 | BT_ASSERT(dummy); |
c2b71c92 JG |
47 | destroy_private_dummy_data(dummy); |
48 | } | |
49 | ||
d94d92ac | 50 | BT_HIDDEN |
d24d5663 | 51 | bt_component_class_init_method_status dummy_init( |
b19ff26f PP |
52 | bt_self_component_sink *component, |
53 | const bt_value *params, | |
c88dd1cb | 54 | __attribute__((unused)) void *init_method_data) |
c2b71c92 | 55 | { |
d24d5663 PP |
56 | bt_component_class_init_method_status status = |
57 | BT_COMPONENT_CLASS_INIT_METHOD_STATUS_OK; | |
58 | bt_self_component_add_port_status add_port_status; | |
c2b71c92 JG |
59 | struct dummy *dummy = g_new0(struct dummy, 1); |
60 | ||
61 | if (!dummy) { | |
d24d5663 | 62 | status = BT_COMPONENT_CLASS_INIT_METHOD_STATUS_MEMORY_ERROR; |
c2b71c92 JG |
63 | goto end; |
64 | } | |
65 | ||
d24d5663 | 66 | add_port_status = bt_self_component_sink_add_input_port(component, |
147337a3 | 67 | "in", NULL, NULL); |
d24d5663 PP |
68 | switch (add_port_status) { |
69 | case BT_SELF_COMPONENT_ADD_PORT_STATUS_ERROR: | |
70 | status = BT_COMPONENT_CLASS_INIT_METHOD_STATUS_ERROR; | |
c2b71c92 | 71 | goto error; |
d24d5663 PP |
72 | case BT_SELF_COMPONENT_ADD_PORT_STATUS_MEMORY_ERROR: |
73 | status = BT_COMPONENT_CLASS_INIT_METHOD_STATUS_MEMORY_ERROR; | |
74 | goto error; | |
75 | default: | |
76 | break; | |
c2b71c92 | 77 | } |
d19348b6 | 78 | |
d94d92ac | 79 | bt_self_component_set_data( |
707b7d35 | 80 | bt_self_component_sink_as_self_component(component), dummy); |
d19348b6 PP |
81 | goto end; |
82 | ||
c2b71c92 JG |
83 | error: |
84 | destroy_private_dummy_data(dummy); | |
d19348b6 PP |
85 | |
86 | end: | |
d24d5663 | 87 | return status; |
c2b71c92 JG |
88 | } |
89 | ||
d94d92ac | 90 | BT_HIDDEN |
d24d5663 | 91 | bt_component_class_sink_graph_is_configured_method_status dummy_graph_is_configured( |
5badd463 | 92 | bt_self_component_sink *comp) |
c2b71c92 | 93 | { |
e803df70 SM |
94 | bt_component_class_sink_graph_is_configured_method_status status; |
95 | bt_self_component_port_input_message_iterator_create_from_sink_component_status | |
96 | msg_iter_status; | |
c2b71c92 | 97 | struct dummy *dummy; |
d6e69534 | 98 | bt_self_component_port_input_message_iterator *iterator; |
c2b71c92 | 99 | |
d94d92ac | 100 | dummy = bt_self_component_get_data( |
707b7d35 | 101 | bt_self_component_sink_as_self_component(comp)); |
f6ccaed9 | 102 | BT_ASSERT(dummy); |
e803df70 | 103 | msg_iter_status = bt_self_component_port_input_message_iterator_create_from_sink_component( |
ca02df0a | 104 | comp, bt_self_component_sink_borrow_input_port_by_name(comp, |
e803df70 SM |
105 | in_port_name), &iterator); |
106 | if (msg_iter_status != BT_SELF_COMPONENT_PORT_INPUT_MESSAGE_ITERATOR_CREATE_FROM_SINK_COMPONENT_STATUS_OK) { | |
107 | status = (int) msg_iter_status; | |
c2b71c92 JG |
108 | goto end; |
109 | } | |
110 | ||
d6e69534 PP |
111 | BT_SELF_COMPONENT_PORT_INPUT_MESSAGE_ITERATOR_MOVE_REF( |
112 | dummy->msg_iter, iterator); | |
0d8b4d8e | 113 | |
e803df70 SM |
114 | status = BT_COMPONENT_CLASS_SINK_GRAPH_IS_CONFIGURED_METHOD_STATUS_OK; |
115 | ||
c2b71c92 | 116 | end: |
bf55043c | 117 | return status; |
c2b71c92 | 118 | } |
e0dfa761 | 119 | |
d94d92ac | 120 | BT_HIDDEN |
d24d5663 | 121 | bt_component_class_sink_consume_method_status dummy_consume( |
b19ff26f | 122 | bt_self_component_sink *component) |
e0dfa761 | 123 | { |
d24d5663 PP |
124 | bt_component_class_sink_consume_method_status status = |
125 | BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_OK; | |
d6e69534 | 126 | bt_message_array_const msgs; |
d4393e08 | 127 | uint64_t count; |
c2b71c92 | 128 | struct dummy *dummy; |
d24d5663 | 129 | bt_message_iterator_next_status next_status; |
d4393e08 | 130 | uint64_t i; |
e0dfa761 | 131 | |
d94d92ac | 132 | dummy = bt_self_component_get_data( |
707b7d35 | 133 | bt_self_component_sink_as_self_component(component)); |
f6ccaed9 | 134 | BT_ASSERT(dummy); |
e0dfa761 | 135 | |
91d81473 | 136 | if (G_UNLIKELY(!dummy->msg_iter)) { |
d24d5663 | 137 | status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_END; |
d19348b6 | 138 | goto end; |
e0dfa761 PP |
139 | } |
140 | ||
d6e69534 | 141 | /* Consume one message */ |
d24d5663 | 142 | next_status = bt_self_component_port_input_message_iterator_next( |
d6e69534 | 143 | dummy->msg_iter, &msgs, &count); |
d24d5663 PP |
144 | switch (next_status) { |
145 | case BT_MESSAGE_ITERATOR_NEXT_STATUS_OK: | |
146 | status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_OK; | |
d4393e08 PP |
147 | |
148 | for (i = 0; i < count; i++) { | |
d6e69534 | 149 | bt_message_put_ref(msgs[i]); |
d4393e08 PP |
150 | } |
151 | ||
152 | break; | |
d24d5663 PP |
153 | case BT_MESSAGE_ITERATOR_NEXT_STATUS_AGAIN: |
154 | status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_AGAIN; | |
d19348b6 | 155 | goto end; |
d24d5663 PP |
156 | case BT_MESSAGE_ITERATOR_NEXT_STATUS_END: |
157 | status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_END; | |
d19348b6 | 158 | goto end; |
d24d5663 PP |
159 | case BT_MESSAGE_ITERATOR_NEXT_STATUS_ERROR: |
160 | status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_ERROR; | |
d94d92ac | 161 | goto end; |
d24d5663 PP |
162 | case BT_MESSAGE_ITERATOR_NEXT_STATUS_MEMORY_ERROR: |
163 | status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_MEMORY_ERROR; | |
d4393e08 | 164 | goto end; |
d19348b6 PP |
165 | default: |
166 | break; | |
e0dfa761 | 167 | } |
d19348b6 | 168 | |
e0dfa761 | 169 | end: |
d24d5663 | 170 | return status; |
e0dfa761 | 171 | } |