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(struct 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_object_put_ref(di_trace->writer_trace);
  82         g_free(di_trace);
  83 }
  84
  85 static
  86 void debug_info_iterator_destroy(struct 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_object_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 struct bt_notification *handle_notification(FILE *err,
 109                 struct debug_info_iterator *debug_it,
 110                 struct bt_notification *notification)
 111 {
 112         struct bt_notification *new_notification = NULL;
 113
 114         switch (bt_notification_get_type(notification)) {
 115         case BT_NOTIFICATION_TYPE_PACKET_BEGIN:
 116         {
 117                 const struct bt_packet *packet =
 118                         bt_notification_packet_begin_get_packet(notification);
 119                 const struct 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_begin_create(
 128                                 writer_packet);
 129                 BT_ASSERT(new_notification);
 130                 bt_object_put_ref(packet);
 131                 bt_object_put_ref(writer_packet);
 132                 break;
 133         }
 134         case BT_NOTIFICATION_TYPE_PACKET_END:
 135         {
 136                 const struct bt_packet *packet =
 137                         bt_notification_packet_end_get_packet(notification);
 138                 const struct 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_object_put_ref(packet);
 150                 bt_object_put_ref(writer_packet);
 151                 break;
 152         }
 153         case BT_NOTIFICATION_TYPE_EVENT:
 154         {
 155                 const struct bt_event *event = bt_notification_event_get_event(
 156                                 notification);
 157                 const struct bt_event *writer_event;
 158                 struct 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_BEGIN:
 176         {
 177                 const struct bt_stream *stream =
 178                         bt_notification_stream_begin_get_stream(notification);
 179                 const struct 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_begin_create(
 188                                 writer_stream);
 189                 BT_ASSERT(new_notification);
 190                 bt_object_put_ref(stream);
 191                 bt_object_put_ref(writer_stream);
 192                 break;
 193         }
 194         case BT_NOTIFICATION_TYPE_STREAM_END:
 195         {
 196                 const struct bt_stream *stream =
 197                         bt_notification_stream_end_get_stream(notification);
 198                 const struct 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_object_put_ref(stream);
 210                 bt_object_put_ref(writer_stream);
 211                 break;
 212         }
 213         default:
 214                 new_notification = bt_object_get_ref(notification);
 215                 break;
 216         }
 217
 218 end:
 219         return new_notification;
 220 }
 221
 222 static
 223 struct bt_notification_iterator_next_method_return debug_info_iterator_next(
 224                 struct bt_self_notification_iterator *iterator)
 225 {
 226         struct debug_info_iterator *debug_it = NULL;
 227         struct bt_self_component *component = NULL;
 228         struct debug_info_component *debug_info = NULL;
 229         struct bt_notification_iterator *source_it = NULL;
 230         struct bt_notification *notification;
 231         struct 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_object_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                 struct 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         struct 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                 struct bt_value *params)
 330 {
 331         struct 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_object_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_object_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_object_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_object_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         struct bt_self_component *component, struct 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);