Commit | Line | Data |
---|---|---|
de713ce0 JG |
1 | /* |
2 | * component.c | |
3 | * | |
4 | * Babeltrace Plugin Component | |
5 | * | |
6 | * Copyright 2015 Jérémie Galarneau <jeremie.galarneau@efficios.com> | |
7 | * | |
8 | * Author: Jérémie Galarneau <jeremie.galarneau@efficios.com> | |
9 | * | |
10 | * Permission is hereby granted, free of charge, to any person obtaining a copy | |
11 | * of this software and associated documentation files (the "Software"), to deal | |
12 | * in the Software without restriction, including without limitation the rights | |
13 | * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |
14 | * copies of the Software, and to permit persons to whom the Software is | |
15 | * furnished to do so, subject to the following conditions: | |
16 | * | |
17 | * The above copyright notice and this permission notice shall be included in | |
18 | * all copies or substantial portions of the Software. | |
19 | * | |
20 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
21 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
22 | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |
23 | * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |
24 | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
25 | * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |
26 | * SOFTWARE. | |
27 | */ | |
28 | ||
29 | #include <babeltrace/plugin/component-internal.h> | |
30 | #include <babeltrace/babeltrace-internal.h> | |
31 | #include <babeltrace/compiler.h> | |
32 | ||
33 | static void bt_component_destroy(struct bt_ctf_ref *ref); | |
34 | ||
35 | const char *bt_component_get_name(struct bt_component *component) | |
36 | { | |
37 | const char *ret = NULL; | |
38 | ||
39 | if (!component) { | |
40 | goto end; | |
41 | } | |
42 | ||
43 | ret = component->name->str; | |
44 | end: | |
45 | return ret; | |
46 | } | |
47 | ||
48 | enum bt_component_status bt_component_set_name(struct bt_component *component, | |
49 | const char *name) | |
50 | { | |
51 | enum bt_component_status ret = BT_COMPONENT_STATUS_OK; | |
52 | ||
53 | if (!component || !name || name[0] == '\0') { | |
54 | ret = BT_COMPONENT_STATUS_INVAL; | |
55 | goto end; | |
56 | } | |
57 | ||
58 | g_string_assign(component->name, name); | |
59 | end: | |
60 | return ret; | |
61 | } | |
62 | ||
63 | enum bt_component_type bt_component_get_type(struct bt_component *component) | |
64 | { | |
65 | enum bt_component_type type = BT_COMPONENT_TYPE_UNKNOWN; | |
66 | ||
67 | if (!component) { | |
68 | goto end; | |
69 | } | |
70 | ||
71 | type = component->type; | |
72 | end: | |
73 | return type; | |
74 | } | |
75 | ||
76 | enum bt_component_status bt_component_set_error_stream( | |
77 | struct bt_component *component, FILE *stream) | |
78 | { | |
79 | enum bt_component_status ret = BT_COMPONENT_STATUS_OK; | |
80 | ||
81 | if (!component) { | |
82 | ret = BT_COMPONENT_STATUS_INVAL; | |
83 | goto end; | |
84 | } | |
85 | ||
86 | component->error_stream = stream; | |
87 | end: | |
88 | return ret; | |
89 | } | |
90 | ||
91 | void bt_component_get(struct bt_component *component) | |
92 | { | |
93 | if (!component) { | |
94 | return; | |
95 | } | |
96 | ||
97 | bt_ctf_ref_get(&component->ref_count); | |
98 | } | |
99 | ||
100 | void bt_component_put(struct bt_component *component) | |
101 | { | |
102 | if (!component) { | |
103 | return; | |
104 | } | |
105 | ||
106 | bt_ctf_ref_put(&component->ref_count, bt_component_destroy); | |
107 | } | |
108 | ||
109 | BT_HIDDEN | |
110 | enum bt_component_status bt_component_init(struct bt_component *component, const char *name, | |
111 | void *user_data, bt_component_destroy_cb user_destroy_func, | |
112 | enum bt_component_type component_type, | |
113 | bt_component_destroy_cb component_destroy) | |
114 | { | |
115 | enum bt_component_status ret = BT_COMPONENT_STATUS_OK; | |
116 | ||
117 | if (!component || !name || name[0] == '\0' || | |
118 | !user_destroy_func || !user_data || !component_destroy) { | |
119 | ret = BT_COMPONENT_STATUS_INVAL; | |
120 | goto end; | |
121 | } | |
122 | ||
123 | bt_ctf_ref_init(&component->ref_count); | |
124 | component->type = component_type; | |
125 | component->user_data = user_data; | |
126 | component->user_data_destroy = user_destroy_func; | |
127 | component->destroy = component_destroy; | |
128 | ||
129 | component->name = g_string_new(name); | |
130 | if (!component->name) { | |
131 | ret = BT_COMPONENT_STATUS_NOMEM; | |
132 | goto end; | |
133 | } | |
134 | end: | |
135 | return ret; | |
136 | } | |
137 | ||
138 | void *bt_component_get_private_data(struct bt_component *component) | |
139 | { | |
140 | void *ret = NULL; | |
141 | ||
142 | if (!component) { | |
143 | goto end; | |
144 | } | |
145 | ||
146 | ret = component->user_data; | |
147 | end: | |
148 | return ret; | |
149 | } | |
150 | ||
151 | enum bt_component_status | |
152 | bt_component_set_private_data(struct bt_component *component, | |
153 | void *data) | |
154 | { | |
155 | enum bt_component_status ret = BT_COMPONENT_STATUS_OK; | |
156 | ||
157 | if (!component) { | |
158 | ret = BT_COMPONENT_STATUS_INVAL; | |
159 | goto end; | |
160 | } | |
161 | ||
162 | component->user_data = data; | |
163 | end: | |
164 | return ret; | |
165 | } | |
166 | ||
167 | static | |
168 | void bt_component_destroy(struct bt_ctf_ref *ref) | |
169 | { | |
170 | struct bt_component *component = NULL; | |
171 | ||
172 | if (!ref) { | |
173 | return; | |
174 | } | |
175 | ||
176 | component = container_of(ref, struct bt_component, ref_count); | |
177 | ||
178 | /** | |
179 | * User data is destroyed first, followed by the concrete component | |
180 | * instance. | |
181 | */ | |
182 | assert(!component->user_data || component->user_data_destroy); | |
183 | component->user_data_destroy(component->user_data); | |
184 | ||
185 | g_string_free(component->name, TRUE); | |
186 | ||
187 | assert(component->destroy); | |
188 | component->destroy(component); | |
189 | } |