lib: add aliases for Babeltrace structure types
[babeltrace.git] / plugins / lttng-utils / plugin.c
1 /*
2 * plugin.c
3 *
4 * Babeltrace Debug Info Plug-in
5 *
6 * Copyright 2016 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 #define BT_LOG_TAG "PLUGIN-CTF-LTTNG-UTILS-DEBUG-INFO-FLT"
30 #include "logging.h"
31
32 #include <babeltrace/babeltrace.h>
33 #include <plugins-common.h>
34 #include <babeltrace/assert-internal.h>
35 #include "debug-info.h"
36 #include "copy.h"
37
38 static
39 gboolean empty_trace_map(gpointer key, gpointer value, gpointer user_data)
40 {
41 struct debug_info_trace *di_trace = value;
42
43 di_trace->trace_static = 1;
44 debug_info_close_trace(di_trace->debug_it, di_trace);
45
46 return TRUE;
47 }
48
49 static
50 void destroy_debug_info_data(struct debug_info_component *debug_info)
51 {
52 free(debug_info->arg_debug_info_field_name);
53 g_free(debug_info);
54 }
55
56 static
57 void destroy_debug_info_component(bt_self_component *component)
58 {
59 void *data = bt_self_component_get_user_data(component);
60 destroy_debug_info_data(data);
61 }
62
63 static
64 struct debug_info_component *create_debug_info_component_data(void)
65 {
66 struct debug_info_component *debug_info;
67
68 debug_info = g_new0(struct debug_info_component, 1);
69 if (!debug_info) {
70 goto end;
71 }
72 debug_info->err = stderr;
73
74 end:
75 return debug_info;
76 }
77
78 static
79 void unref_trace(struct debug_info_trace *di_trace)
80 {
81 bt_trace_put_ref(di_trace->writer_trace);
82 g_free(di_trace);
83 }
84
85 static
86 void debug_info_iterator_destroy(bt_self_notification_iterator *it)
87 {
88 struct debug_info_iterator *it_data;
89
90 it_data = bt_self_notification_iterator_get_user_data(it);
91 BT_ASSERT(it_data);
92
93 if (it_data->input_iterator_group) {
94 g_ptr_array_free(it_data->input_iterator_group, TRUE);
95 }
96
97 g_hash_table_foreach_remove(it_data->trace_map,
98 empty_trace_map, it_data);
99 g_hash_table_destroy(it_data->trace_map);
100
101 bt_notification_put_ref(it_data->current_notification);
102 bt_object_put_ref(it_data->input_iterator);
103
104 g_free(it_data);
105 }
106
107 static
108 const bt_notification *handle_notification(FILE *err,
109 struct debug_info_iterator *debug_it,
110 const bt_notification *notification)
111 {
112 const bt_notification *new_notification = NULL;
113
114 switch (bt_notification_get_type(notification)) {
115 case BT_NOTIFICATION_TYPE_PACKET_BEGINNING:
116 {
117 const bt_packet *packet =
118 bt_notification_packet_beginning_get_packet(notification);
119 const bt_packet *writer_packet;
120
121 if (!packet) {
122 goto end;
123 }
124
125 writer_packet = debug_info_new_packet(debug_it, packet);
126 BT_ASSERT(writer_packet);
127 new_notification = bt_notification_packet_beginning_create(
128 writer_packet);
129 BT_ASSERT(new_notification);
130 bt_packet_put_ref(packet);
131 bt_packet_put_ref(writer_packet);
132 break;
133 }
134 case BT_NOTIFICATION_TYPE_PACKET_END:
135 {
136 const bt_packet *packet =
137 bt_notification_packet_end_get_packet(notification);
138 const bt_packet *writer_packet;
139
140 if (!packet) {
141 goto end;
142 }
143
144 writer_packet = debug_info_close_packet(debug_it, packet);
145 BT_ASSERT(writer_packet);
146 new_notification = bt_notification_packet_end_create(
147 writer_packet);
148 BT_ASSERT(new_notification);
149 bt_packet_put_ref(packet);
150 bt_packet_put_ref(writer_packet);
151 break;
152 }
153 case BT_NOTIFICATION_TYPE_EVENT:
154 {
155 const bt_event *event = bt_notification_event_get_event(
156 notification);
157 const bt_event *writer_event;
158 bt_clock_class_priority_map *cc_prio_map =
159 bt_notification_event_get_clock_class_priority_map(
160 notification);
161
162 if (!event) {
163 goto end;
164 }
165 writer_event = debug_info_output_event(debug_it, event);
166 BT_ASSERT(writer_event);
167 new_notification = bt_notification_event_create(writer_event,
168 cc_prio_map);
169 bt_object_put_ref(cc_prio_map);
170 BT_ASSERT(new_notification);
171 bt_object_put_ref(event);
172 bt_object_put_ref(writer_event);
173 break;
174 }
175 case BT_NOTIFICATION_TYPE_STREAM_BEGINNING:
176 {
177 const bt_stream *stream =
178 bt_notification_stream_beginning_get_stream(notification);
179 const bt_stream *writer_stream;
180
181 if (!stream) {
182 goto end;
183 }
184
185 writer_stream = debug_info_stream_begin(debug_it, stream);
186 BT_ASSERT(writer_stream);
187 new_notification = bt_notification_stream_beginning_create(
188 writer_stream);
189 BT_ASSERT(new_notification);
190 bt_stream_put_ref(stream);
191 bt_stream_put_ref(writer_stream);
192 break;
193 }
194 case BT_NOTIFICATION_TYPE_STREAM_END:
195 {
196 const bt_stream *stream =
197 bt_notification_stream_end_get_stream(notification);
198 const bt_stream *writer_stream;
199
200 if (!stream) {
201 goto end;
202 }
203
204 writer_stream = debug_info_stream_end(debug_it, stream);
205 BT_ASSERT(writer_stream);
206 new_notification = bt_notification_stream_end_create(
207 writer_stream);
208 BT_ASSERT(new_notification);
209 bt_stream_put_ref(stream);
210 bt_stream_put_ref(writer_stream);
211 break;
212 }
213 default:
214 new_notification = bt_notification_get_ref(notification);
215 break;
216 }
217
218 end:
219 return new_notification;
220 }
221
222 static
223 bt_notification_iterator_next_method_return debug_info_iterator_next(
224 bt_self_notification_iterator *iterator)
225 {
226 struct debug_info_iterator *debug_it = NULL;
227 bt_self_component *component = NULL;
228 struct debug_info_component *debug_info = NULL;
229 bt_notification_iterator *source_it = NULL;
230 const bt_notification *notification;
231 bt_notification_iterator_next_method_return ret = {
232 .status = BT_NOTIFICATION_ITERATOR_STATUS_OK,
233 .notification = NULL,
234 };
235
236 debug_it = bt_self_notification_iterator_get_user_data(iterator);
237 BT_ASSERT(debug_it);
238
239 component = bt_self_notification_iterator_get_private_component(iterator);
240 BT_ASSERT(component);
241 debug_info = bt_self_component_get_user_data(component);
242 BT_ASSERT(debug_info);
243
244 source_it = debug_it->input_iterator;
245
246 ret.status = bt_notification_iterator_next(source_it);
247 if (ret.status != BT_NOTIFICATION_ITERATOR_STATUS_OK) {
248 goto end;
249 }
250
251 notification = bt_notification_iterator_get_notification(
252 source_it);
253 if (!notification) {
254 ret.status = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
255 goto end;
256 }
257
258 ret.notification = handle_notification(debug_info->err, debug_it,
259 notification);
260 BT_ASSERT(ret.notification);
261 bt_notification_put_ref(notification);
262
263 end:
264 bt_object_put_ref(component);
265 return ret;
266 }
267
268 static
269 enum bt_notification_iterator_status debug_info_iterator_init(
270 bt_self_notification_iterator *iterator,
271 struct bt_private_port *port)
272 {
273 enum bt_notification_iterator_status ret =
274 BT_NOTIFICATION_ITERATOR_STATUS_OK;
275 enum bt_notification_iterator_status it_ret;
276 enum bt_connection_status conn_status;
277 struct bt_private_connection *connection = NULL;
278 bt_self_component *component =
279 bt_self_notification_iterator_get_private_component(iterator);
280 struct debug_info_iterator *it_data = g_new0(struct debug_info_iterator, 1);
281 struct bt_private_port *input_port;
282
283 if (!it_data) {
284 ret = BT_NOTIFICATION_ITERATOR_STATUS_NOMEM;
285 goto end;
286 }
287
288 input_port = bt_self_component_filter_get_input_port_by_name(
289 component, "in");
290 if (!input_port) {
291 ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
292 goto end;
293 }
294
295 connection = bt_private_port_get_connection(input_port);
296 bt_object_put_ref(input_port);
297 if (!connection) {
298 ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
299 goto end;
300 }
301
302 conn_status = bt_private_connection_create_notification_iterator(
303 connection, &it_data->input_iterator);
304 if (conn_status != BT_CONNECTION_STATUS_OK) {
305 ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
306 goto end;
307 }
308
309 it_data->debug_info_component = (struct debug_info_component *)
310 bt_self_component_get_user_data(component);
311 it_data->err = it_data->debug_info_component->err;
312 it_data->trace_map = g_hash_table_new_full(g_direct_hash,
313 g_direct_equal, NULL, (GDestroyNotify) unref_trace);
314
315 it_ret = bt_self_notification_iterator_set_user_data(iterator, it_data);
316 if (it_ret) {
317 goto end;
318 }
319
320 end:
321 bt_object_put_ref(connection);
322 bt_object_put_ref(component);
323 return ret;
324 }
325
326 static
327 enum bt_component_status init_from_params(
328 struct debug_info_component *debug_info_component,
329 bt_value *params)
330 {
331 bt_value *value = NULL;
332 enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
333
334 BT_ASSERT(params);
335
336 value = bt_value_map_get(params, "debug-info-field-name");
337 if (value) {
338 enum bt_value_status value_ret;
339 const char *tmp;
340
341 tmp = bt_value_string_get(value);
342 strcpy(debug_info_component->arg_debug_info_field_name, tmp);
343 bt_value_put_ref(value);
344 } else {
345 debug_info_component->arg_debug_info_field_name =
346 malloc(strlen("debug_info") + 1);
347 if (!debug_info_component->arg_debug_info_field_name) {
348 ret = BT_COMPONENT_STATUS_NOMEM;
349 BT_LOGE_STR("Missing field name.");
350 goto end;
351 }
352 sprintf(debug_info_component->arg_debug_info_field_name,
353 "debug_info");
354 }
355 if (ret != BT_COMPONENT_STATUS_OK) {
356 goto end;
357 }
358
359 value = bt_value_map_get(params, "debug-info-dir");
360 if (value) {
361 enum bt_value_status value_ret;
362
363 debug_info_component->arg_debug_dir = bt_value_string_get(value);
364 }
365 bt_value_put_ref(value);
366 if (ret != BT_COMPONENT_STATUS_OK) {
367 goto end;
368 }
369
370 value = bt_value_map_get(params, "target-prefix");
371 if (value) {
372 enum bt_value_status value_ret;
373
374 debug_info_component->arg_target_prefix = bt_value_string_get(value);
375 }
376 bt_value_put_ref(value);
377 if (ret != BT_COMPONENT_STATUS_OK) {
378 goto end;
379 }
380
381 value = bt_value_map_get(params, "full-path");
382 if (value) {
383 enum bt_value_status value_ret;
384 bt_bool bool_val;
385
386 bool_val = bt_value_bool_get(value);
387
388 debug_info_component->arg_full_path = bool_val;
389 }
390 bt_value_put_ref(value);
391 if (ret != BT_COMPONENT_STATUS_OK) {
392 goto end;
393 }
394
395 end:
396 return ret;
397 }
398
399 enum bt_component_status debug_info_component_init(
400 bt_self_component *component, bt_value *params,
401 UNUSED_VAR void *init_method_data)
402 {
403 enum bt_component_status ret;
404 struct debug_info_component *debug_info = create_debug_info_component_data();
405
406 if (!debug_info) {
407 ret = BT_COMPONENT_STATUS_NOMEM;
408 goto end;
409 }
410
411 ret = bt_self_component_set_user_data(component, debug_info);
412 if (ret != BT_COMPONENT_STATUS_OK) {
413 goto error;
414 }
415
416 ret = bt_self_component_filter_add_input_port(
417 component, "in", NULL, NULL);
418 if (ret != BT_COMPONENT_STATUS_OK) {
419 goto end;
420 }
421
422 ret = bt_self_component_filter_add_output_port(
423 component, "out", NULL, NULL);
424 if (ret != BT_COMPONENT_STATUS_OK) {
425 goto end;
426 }
427
428 ret = init_from_params(debug_info, params);
429 end:
430 return ret;
431 error:
432 destroy_debug_info_data(debug_info);
433 return ret;
434 }
435
436 #ifndef BT_BUILT_IN_PLUGINS
437 BT_PLUGIN_MODULE();
438 #endif
439
440 /* Initialize plug-in entry points. */
441 BT_PLUGIN_WITH_ID(lttng_utils, "lttng-utils");
442 BT_PLUGIN_DESCRIPTION_WITH_ID(lttng_utils, "LTTng utilities");
443 BT_PLUGIN_AUTHOR_WITH_ID(lttng_utils, "Julien Desfossez");
444 BT_PLUGIN_LICENSE_WITH_ID(lttng_utils, "MIT");
445
446 BT_PLUGIN_FILTER_COMPONENT_CLASS_WITH_ID(lttng_utils, debug_info, "debug-info",
447 debug_info_iterator_next);
448 BT_PLUGIN_FILTER_COMPONENT_CLASS_DESCRIPTION_WITH_ID(lttng_utils, debug_info,
449 "Augment compatible events with debugging information.");
450 BT_PLUGIN_FILTER_COMPONENT_CLASS_INIT_METHOD_WITH_ID(lttng_utils,
451 debug_info, debug_info_component_init);
452 BT_PLUGIN_FILTER_COMPONENT_CLASS_FINALIZE_METHOD_WITH_ID(lttng_utils,
453 debug_info, destroy_debug_info_component);
454 BT_PLUGIN_FILTER_COMPONENT_CLASS_NOTIFICATION_ITERATOR_INIT_METHOD_WITH_ID(
455 lttng_utils, debug_info, debug_info_iterator_init);
456 BT_PLUGIN_FILTER_COMPONENT_CLASS_NOTIFICATION_ITERATOR_FINALIZE_METHOD_WITH_ID(
457 lttng_utils, debug_info, debug_info_iterator_destroy);
This page took 0.0398849999999999 seconds and 4 git commands to generate.