src/plugins/text/dmesg/dmesg.c

   1 /*
   2  * Copyright 2010-2011 EfficiOS Inc. and Linux Foundation
   3  * Copyright 2017 Philippe Proulx <pproulx@efficios.com>
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a copy
   6  * of this software and associated documentation files (the "Software"), to deal
   7  * in the Software without restriction, including without limitation the rights
   8  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   9  * copies of the Software, and to permit persons to whom the Software is
  10  * furnished to do so, subject to the following conditions:
  11  *
  12  * The above copyright notice and this permission notice shall be included in
  13  * all copies or substantial portions of the Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21  * SOFTWARE.
  22  */
  23
  24 #define BT_COMP_LOG_SELF_COMP (dmesg_comp->self_comp)
  25 #define BT_LOG_OUTPUT_LEVEL (dmesg_comp->log_level)
  26 #define BT_LOG_TAG "PLUGIN/SRC.TEXT.DMESG"
  27 #include "logging/comp-logging.h"
  28
  29 #include "dmesg.h"
  30
  31 #include <stdbool.h>
  32 #include <string.h>
  33 #include <ctype.h>
  34 #include <stdio.h>
  35 #include "common/common.h"
  36 #include "common/assert.h"
  37 #include <babeltrace2/babeltrace.h>
  38 #include "compat/utc.h"
  39 #include "compat/stdio.h"
  40 #include <glib.h>
  41 #include "plugins/common/param-validation/param-validation.h"
  42
  43 #define NSEC_PER_USEC 1000UL
  44 #define NSEC_PER_MSEC 1000000UL
  45 #define NSEC_PER_SEC 1000000000ULL
  46 #define USEC_PER_SEC 1000000UL
  47
  48 struct dmesg_component;
  49
  50 struct dmesg_msg_iter {
  51         struct dmesg_component *dmesg_comp;
  52
  53         /* Weak */
  54         bt_self_message_iterator *self_msg_iter;
  55
  56         char *linebuf;
  57         size_t linebuf_len;
  58         FILE *fp;
  59         bt_message *tmp_event_msg;
  60         uint64_t last_clock_value;
  61
  62         enum {
  63                 STATE_EMIT_STREAM_BEGINNING,
  64                 STATE_EMIT_EVENT,
  65                 STATE_EMIT_STREAM_END,
  66                 STATE_DONE,
  67         } state;
  68 };
  69
  70 struct dmesg_component {
  71         bt_logging_level log_level;
  72
  73         struct {
  74                 GString *path;
  75                 bt_bool read_from_stdin;
  76                 bt_bool no_timestamp;
  77         } params;
  78
  79         bt_self_component_source *self_comp_src;
  80         bt_self_component *self_comp;
  81         bt_trace_class *trace_class;
  82         bt_stream_class *stream_class;
  83         bt_event_class *event_class;
  84         bt_trace *trace;
  85         bt_stream *stream;
  86         bt_clock_class *clock_class;
  87 };
  88
  89 static
  90 bt_field_class *create_event_payload_fc(struct dmesg_component *dmesg_comp,
  91                 bt_trace_class *trace_class)
  92 {
  93         bt_field_class *root_fc = NULL;
  94         bt_field_class *fc = NULL;
  95         int ret;
  96
  97         root_fc = bt_field_class_structure_create(trace_class);
  98         if (!root_fc) {
  99                 BT_COMP_LOGE_STR("Cannot create an empty structure field class object.");
 100                 goto error;
 101         }
 102
 103         fc = bt_field_class_string_create(trace_class);
 104         if (!fc) {
 105                 BT_COMP_LOGE_STR("Cannot create a string field class object.");
 106                 goto error;
 107         }
 108
 109         ret = bt_field_class_structure_append_member(root_fc,
 110                 "str", fc);
 111         if (ret) {
 112                 BT_COMP_LOGE("Cannot add `str` member to structure field class: "
 113                         "ret=%d", ret);
 114                 goto error;
 115         }
 116
 117         goto end;
 118
 119 error:
 120         BT_FIELD_CLASS_PUT_REF_AND_RESET(root_fc);
 121
 122 end:
 123         bt_field_class_put_ref(fc);
 124         return root_fc;
 125 }
 126
 127 static
 128 int create_meta(struct dmesg_component *dmesg_comp, bool has_ts)
 129 {
 130         bt_field_class *fc = NULL;
 131         int ret = 0;
 132
 133         dmesg_comp->trace_class = bt_trace_class_create(dmesg_comp->self_comp);
 134         if (!dmesg_comp->trace_class) {
 135                 BT_COMP_LOGE_STR("Cannot create an empty trace class object.");
 136                 goto error;
 137         }
 138
 139         dmesg_comp->stream_class = bt_stream_class_create(
 140                 dmesg_comp->trace_class);
 141         if (!dmesg_comp->stream_class) {
 142                 BT_COMP_LOGE_STR("Cannot create a stream class object.");
 143                 goto error;
 144         }
 145
 146         if (has_ts) {
 147                 dmesg_comp->clock_class = bt_clock_class_create(
 148                         dmesg_comp->self_comp);
 149                 if (!dmesg_comp->clock_class) {
 150                         BT_COMP_LOGE_STR("Cannot create clock class.");
 151                         goto error;
 152                 }
 153
 154                 /*
 155                  * The `dmesg` timestamp's origin is not the Unix epoch,
 156                  * it's the boot time.
 157                  */
 158                 bt_clock_class_set_origin_is_unix_epoch(dmesg_comp->clock_class,
 159                         BT_FALSE);
 160
 161                 ret = bt_stream_class_set_default_clock_class(
 162                         dmesg_comp->stream_class, dmesg_comp->clock_class);
 163                 if (ret) {
 164                         BT_COMP_LOGE_STR("Cannot set stream class's default clock class.");
 165                         goto error;
 166                 }
 167         }
 168
 169         dmesg_comp->event_class = bt_event_class_create(
 170                 dmesg_comp->stream_class);
 171         if (!dmesg_comp->event_class) {
 172                 BT_COMP_LOGE_STR("Cannot create an event class object.");
 173                 goto error;
 174         }
 175
 176         ret = bt_event_class_set_name(dmesg_comp->event_class, "string");
 177         if (ret) {
 178                 BT_COMP_LOGE_STR("Cannot set event class's name.");
 179                 goto error;
 180         }
 181
 182         fc = create_event_payload_fc(dmesg_comp, dmesg_comp->trace_class);
 183         if (!fc) {
 184                 BT_COMP_LOGE_STR("Cannot create event payload field class.");
 185                 goto error;
 186         }
 187
 188         ret = bt_event_class_set_payload_field_class(dmesg_comp->event_class, fc);
 189         if (ret) {
 190                 BT_COMP_LOGE_STR("Cannot set event class's event payload field class.");
 191                 goto error;
 192         }
 193
 194         goto end;
 195
 196 error:
 197         ret = -1;
 198
 199 end:
 200         bt_field_class_put_ref(fc);
 201         return ret;
 202 }
 203
 204 struct bt_param_validation_map_value_entry_descr dmesg_params[] = {
 205         { "no-extract-timestamp", BT_PARAM_VALIDATION_MAP_VALUE_ENTRY_OPTIONAL, { .type = BT_VALUE_TYPE_BOOL } },
 206         { "path", BT_PARAM_VALIDATION_MAP_VALUE_ENTRY_OPTIONAL, { .type = BT_VALUE_TYPE_STRING } },
 207         BT_PARAM_VALIDATION_MAP_VALUE_ENTRY_END
 208 };
 209
 210 static
 211 bt_component_class_initialize_method_status handle_params(
 212                 struct dmesg_component *dmesg_comp,
 213                 const bt_value *params)
 214 {
 215         const bt_value *no_timestamp = NULL;
 216         const bt_value *path = NULL;
 217         bt_component_class_initialize_method_status status;
 218         enum bt_param_validation_status validation_status;
 219         gchar *validate_error = NULL;
 220
 221         validation_status = bt_param_validation_validate(params,
 222                 dmesg_params, &validate_error);
 223         if (validation_status == BT_PARAM_VALIDATION_STATUS_MEMORY_ERROR) {
 224                 status = BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_MEMORY_ERROR;
 225                 goto end;
 226         } else if (validation_status == BT_PARAM_VALIDATION_STATUS_VALIDATION_ERROR) {
 227                 status = BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_ERROR;
 228                 BT_COMP_LOGE_APPEND_CAUSE(dmesg_comp->self_comp,
 229                         "%s", validate_error);
 230                 goto end;
 231         }
 232
 233         no_timestamp = bt_value_map_borrow_entry_value_const(params,
 234                 "no-extract-timestamp");
 235         if (no_timestamp) {
 236                 dmesg_comp->params.no_timestamp =
 237                         bt_value_bool_get(no_timestamp);
 238         }
 239
 240         path = bt_value_map_borrow_entry_value_const(params, "path");
 241         if (path) {
 242                 const char *path_str = bt_value_string_get(path);
 243
 244                 g_string_assign(dmesg_comp->params.path, path_str);
 245         } else {
 246                 dmesg_comp->params.read_from_stdin = true;
 247         }
 248
 249         status = BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_OK;
 250 end:
 251         g_free(validate_error);
 252
 253         return status;
 254 }
 255
 256 static
 257 int create_stream_and_trace(struct dmesg_component *dmesg_comp)
 258 {
 259         int ret = 0;
 260         const char *trace_name = NULL;
 261         gchar *basename = NULL;
 262
 263         dmesg_comp->trace = bt_trace_create(dmesg_comp->trace_class);
 264         if (!dmesg_comp->trace) {
 265                 BT_COMP_LOGE_STR("Cannot create trace object.");
 266                 goto error;
 267         }
 268
 269         if (dmesg_comp->params.read_from_stdin) {
 270                 trace_name = "STDIN";
 271         } else {
 272                 basename = g_path_get_basename(dmesg_comp->params.path->str);
 273                 BT_ASSERT(basename);
 274
 275                 if (strcmp(basename, G_DIR_SEPARATOR_S) != 0 &&
 276                                 strcmp(basename, ".") != 0) {
 277                         trace_name = basename;
 278                 }
 279         }
 280
 281         if (trace_name) {
 282                 ret = bt_trace_set_name(dmesg_comp->trace, trace_name);
 283                 if (ret) {
 284                         BT_COMP_LOGE("Cannot set trace's name: name=\"%s\"",
 285                                 trace_name);
 286                         goto error;
 287                 }
 288         }
 289
 290         dmesg_comp->stream = bt_stream_create(dmesg_comp->stream_class,
 291                 dmesg_comp->trace);
 292         if (!dmesg_comp->stream) {
 293                 BT_COMP_LOGE_STR("Cannot create stream object.");
 294                 goto error;
 295         }
 296
 297         goto end;
 298
 299 error:
 300         ret = -1;
 301
 302 end:
 303         g_free(basename);
 304
 305         return ret;
 306 }
 307
 308 static
 309 int try_create_meta_stream(struct dmesg_component *dmesg_comp, bool has_ts)
 310 {
 311         int ret = 0;
 312
 313         if (dmesg_comp->trace) {
 314                 /* Already created */
 315                 goto end;
 316         }
 317
 318         ret = create_meta(dmesg_comp, has_ts);
 319         if (ret) {
 320                 BT_COMP_LOGE("Cannot create metadata objects: dmesg-comp-addr=%p",
 321                         dmesg_comp);
 322                 goto error;
 323         }
 324
 325         ret = create_stream_and_trace(dmesg_comp);
 326         if (ret) {
 327                 BT_COMP_LOGE("Cannot create stream object: "
 328                         "dmesg-comp-addr=%p", dmesg_comp);
 329                 goto error;
 330         }
 331
 332         goto end;
 333
 334 error:
 335         ret = -1;
 336
 337 end:
 338         return ret;
 339 }
 340
 341 static
 342 void destroy_dmesg_component(struct dmesg_component *dmesg_comp)
 343 {
 344         if (!dmesg_comp) {
 345                 return;
 346         }
 347
 348         if (dmesg_comp->params.path) {
 349                 g_string_free(dmesg_comp->params.path, TRUE);
 350         }
 351
 352         bt_trace_put_ref(dmesg_comp->trace);
 353         bt_stream_class_put_ref(dmesg_comp->stream_class);
 354         bt_event_class_put_ref(dmesg_comp->event_class);
 355         bt_stream_put_ref(dmesg_comp->stream);
 356         bt_clock_class_put_ref(dmesg_comp->clock_class);
 357         bt_trace_class_put_ref(dmesg_comp->trace_class);
 358         g_free(dmesg_comp);
 359 }
 360
 361 static
 362 bt_component_class_initialize_method_status create_port(
 363                 bt_self_component_source *self_comp)
 364 {
 365         bt_component_class_initialize_method_status status;
 366         bt_self_component_add_port_status add_port_status;
 367
 368         add_port_status = bt_self_component_source_add_output_port(self_comp,
 369                 "out", NULL, NULL);
 370         switch (add_port_status) {
 371         case BT_SELF_COMPONENT_ADD_PORT_STATUS_OK:
 372                 status = BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_OK;
 373                 break;
 374         case BT_SELF_COMPONENT_ADD_PORT_STATUS_ERROR:
 375                 status = BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_ERROR;
 376                 break;
 377         case BT_SELF_COMPONENT_ADD_PORT_STATUS_MEMORY_ERROR:
 378                 status = BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_MEMORY_ERROR;
 379                 break;
 380         default:
 381                 bt_common_abort();
 382         }
 383
 384         return status;
 385 }
 386
 387 BT_HIDDEN
 388 bt_component_class_initialize_method_status dmesg_init(
 389                 bt_self_component_source *self_comp_src,
 390                 bt_self_component_source_configuration *config,
 391                 const bt_value *params, void *init_method_data)
 392 {
 393         struct dmesg_component *dmesg_comp = g_new0(struct dmesg_component, 1);
 394         bt_component_class_initialize_method_status status;
 395         bt_self_component *self_comp =
 396                 bt_self_component_source_as_self_component(self_comp_src);
 397         const bt_component *comp = bt_self_component_as_component(self_comp);
 398         bt_logging_level log_level = bt_component_get_logging_level(comp);
 399
 400         if (!dmesg_comp) {
 401                 /* Implicit log level is not available here */
 402                 BT_COMP_LOG_CUR_LVL(BT_LOG_ERROR, log_level, self_comp,
 403                         "Failed to allocate one dmesg component structure.");
 404                 status = BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_ERROR;
 405                 goto error;
 406         }
 407
 408         dmesg_comp->log_level = log_level;
 409         dmesg_comp->self_comp = self_comp;
 410         dmesg_comp->self_comp_src = self_comp_src;
 411         dmesg_comp->params.path = g_string_new(NULL);
 412         if (!dmesg_comp->params.path) {
 413                 BT_COMP_LOGE_STR("Failed to allocate a GString.");
 414                 status = BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_ERROR;
 415                 goto error;
 416         }
 417
 418         status = handle_params(dmesg_comp, params);
 419         if (status != BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_OK) {
 420                 BT_COMP_LOGE("Invalid parameters: comp-addr=%p", self_comp);
 421                 status = BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_ERROR;
 422                 goto error;
 423         }
 424
 425         if (!dmesg_comp->params.read_from_stdin &&
 426                         !g_file_test(dmesg_comp->params.path->str,
 427                         G_FILE_TEST_IS_REGULAR)) {
 428                 BT_COMP_LOGE("Input path is not a regular file: "
 429                         "comp-addr=%p, path=\"%s\"", self_comp,
 430                         dmesg_comp->params.path->str);
 431                 status = BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_ERROR;
 432                 goto error;
 433         }
 434
 435         status = create_port(self_comp_src);
 436         if (status != BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_OK) {
 437                 goto error;
 438         }
 439
 440         bt_self_component_set_data(self_comp, dmesg_comp);
 441         BT_COMP_LOGI_STR("Component initialized.");
 442
 443         status = BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_OK;
 444         goto end;
 445
 446 error:
 447         destroy_dmesg_component(dmesg_comp);
 448         bt_self_component_set_data(self_comp, NULL);
 449
 450 end:
 451         return status;
 452 }
 453
 454 BT_HIDDEN
 455 void dmesg_finalize(bt_self_component_source *self_comp)
 456 {
 457         destroy_dmesg_component(bt_self_component_get_data(
 458                 bt_self_component_source_as_self_component(self_comp)));
 459 }
 460
 461 static
 462 bt_message *create_init_event_msg_from_line(
 463                 struct dmesg_msg_iter *msg_iter,
 464                 const char *line, const char **new_start)
 465 {
 466         bt_event *event;
 467         bt_message *msg = NULL;
 468         bool has_timestamp = false;
 469         unsigned long sec, usec, msec;
 470         unsigned int year, mon, mday, hour, min;
 471         uint64_t ts = 0;
 472         int ret = 0;
 473         struct dmesg_component *dmesg_comp = msg_iter->dmesg_comp;
 474
 475         *new_start = line;
 476
 477         if (dmesg_comp->params.no_timestamp) {
 478                 goto skip_ts;
 479         }
 480
 481         /* Extract time from input line */
 482         if (sscanf(line, "[%lu.%lu] ", &sec, &usec) == 2) {
 483                 ts = (uint64_t) sec * USEC_PER_SEC + (uint64_t) usec;
 484
 485                 /*
 486                  * The clock class we use has a 1 GHz frequency: convert
 487                  * from µs to ns.
 488                  */
 489                 ts *= NSEC_PER_USEC;
 490                 has_timestamp = true;
 491         } else if (sscanf(line, "[%u-%u-%u %u:%u:%lu.%lu] ",
 492                         &year, &mon, &mday, &hour, &min,
 493                         &sec, &msec) == 7) {
 494                 time_t ep_sec;
 495                 struct tm ti;
 496
 497                 memset(&ti, 0, sizeof(ti));
 498                 ti.tm_year = year - 1900;       /* From 1900 */
 499                 ti.tm_mon = mon - 1;            /* 0 to 11 */
 500                 ti.tm_mday = mday;
 501                 ti.tm_hour = hour;
 502                 ti.tm_min = min;
 503                 ti.tm_sec = sec;
 504
 505                 ep_sec = bt_timegm(&ti);
 506                 if (ep_sec != (time_t) -1) {
 507                         ts = (uint64_t) ep_sec * NSEC_PER_SEC
 508                                 + (uint64_t) msec * NSEC_PER_MSEC;
 509                 }
 510
 511                 has_timestamp = true;
 512         }
 513
 514         if (has_timestamp) {
 515                 /* Set new start for the message portion of the line */
 516                 *new_start = strchr(line, ']');
 517                 BT_ASSERT_DBG(*new_start);
 518                 (*new_start)++;
 519
 520                 if ((*new_start)[0] == ' ') {
 521                         (*new_start)++;
 522                 }
 523         }
 524
 525 skip_ts:
 526         /*
 527          * At this point, we know if the stream class's event header
 528          * field class should have a timestamp or not, so we can lazily
 529          * create the metadata and stream objects.
 530          */
 531         ret = try_create_meta_stream(dmesg_comp, has_timestamp);
 532         if (ret) {
 533                 /* try_create_meta_stream() logs errors */
 534                 goto error;
 535         }
 536
 537         if (dmesg_comp->clock_class) {
 538                 msg = bt_message_event_create_with_default_clock_snapshot(
 539                         msg_iter->self_msg_iter,
 540                         dmesg_comp->event_class, dmesg_comp->stream, ts);
 541                 msg_iter->last_clock_value = ts;
 542         } else {
 543                 msg = bt_message_event_create(msg_iter->self_msg_iter,
 544                         dmesg_comp->event_class, dmesg_comp->stream);
 545         }
 546
 547         if (!msg) {
 548                 BT_COMP_LOGE_STR("Cannot create event message.");
 549                 goto error;
 550         }
 551
 552         event = bt_message_event_borrow_event(msg);
 553         BT_ASSERT_DBG(event);
 554         goto end;
 555
 556 error:
 557         BT_MESSAGE_PUT_REF_AND_RESET(msg);
 558
 559 end:
 560         return msg;
 561 }
 562
 563 static
 564 int fill_event_payload_from_line(struct dmesg_component *dmesg_comp,
 565                 const char *line, bt_event *event)
 566 {
 567         bt_field *ep_field = NULL;
 568         bt_field *str_field = NULL;
 569         size_t len;
 570         int ret;
 571
 572         ep_field = bt_event_borrow_payload_field(event);
 573         BT_ASSERT_DBG(ep_field);
 574         str_field = bt_field_structure_borrow_member_field_by_index(
 575                 ep_field, 0);
 576         if (!str_field) {
 577                 BT_COMP_LOGE_STR("Cannot borrow `timestamp` field from event payload structure field.");
 578                 goto error;
 579         }
 580
 581         len = strlen(line);
 582         if (line[len - 1] == '\n') {
 583                 /* Do not include the newline character in the payload */
 584                 len--;
 585         }
 586
 587         bt_field_string_clear(str_field);
 588         ret = bt_field_string_append_with_length(str_field, line, len);
 589         if (ret) {
 590                 BT_COMP_LOGE("Cannot append value to string field object: "
 591                         "len=%zu", len);
 592                 goto error;
 593         }
 594
 595         goto end;
 596
 597 error:
 598         ret = -1;
 599
 600 end:
 601         return ret;
 602 }
 603
 604 static
 605 bt_message *create_msg_from_line(
 606                 struct dmesg_msg_iter *dmesg_msg_iter, const char *line)
 607 {
 608         struct dmesg_component *dmesg_comp = dmesg_msg_iter->dmesg_comp;
 609         bt_event *event = NULL;
 610         bt_message *msg = NULL;
 611         const char *new_start;
 612         int ret;
 613
 614         msg = create_init_event_msg_from_line(dmesg_msg_iter,
 615                 line, &new_start);
 616         if (!msg) {
 617                 BT_COMP_LOGE_STR("Cannot create and initialize event message from line.");
 618                 goto error;
 619         }
 620
 621         event = bt_message_event_borrow_event(msg);
 622         BT_ASSERT_DBG(event);
 623         ret = fill_event_payload_from_line(dmesg_comp, new_start, event);
 624         if (ret) {
 625                 BT_COMP_LOGE("Cannot fill event payload field from line: "
 626                         "ret=%d", ret);
 627                 goto error;
 628         }
 629
 630         goto end;
 631
 632 error:
 633         BT_MESSAGE_PUT_REF_AND_RESET(msg);
 634
 635 end:
 636         return msg;
 637 }
 638
 639 static
 640 void destroy_dmesg_msg_iter(struct dmesg_msg_iter *dmesg_msg_iter)
 641 {
 642         struct dmesg_component *dmesg_comp;
 643
 644         if (!dmesg_msg_iter) {
 645                 return;
 646         }
 647
 648         dmesg_comp = dmesg_msg_iter->dmesg_comp;
 649
 650         if (dmesg_msg_iter->fp && dmesg_msg_iter->fp != stdin) {
 651                 if (fclose(dmesg_msg_iter->fp)) {
 652                         BT_COMP_LOGE_ERRNO("Cannot close input file", ".");
 653                 }
 654         }
 655
 656         bt_message_put_ref(dmesg_msg_iter->tmp_event_msg);
 657         free(dmesg_msg_iter->linebuf);
 658         g_free(dmesg_msg_iter);
 659 }
 660
 661
 662
 663 BT_HIDDEN
 664 bt_component_class_message_iterator_initialize_method_status dmesg_msg_iter_init(
 665                 bt_self_message_iterator *self_msg_iter,
 666                 bt_self_message_iterator_configuration *config,
 667                 bt_self_component_source *self_comp,
 668                 bt_self_component_port_output *self_port)
 669 {
 670         struct dmesg_component *dmesg_comp = bt_self_component_get_data(
 671                 bt_self_component_source_as_self_component(self_comp));
 672         struct dmesg_msg_iter *dmesg_msg_iter =
 673                 g_new0(struct dmesg_msg_iter, 1);
 674         bt_component_class_message_iterator_initialize_method_status status =
 675                 BT_COMPONENT_CLASS_MESSAGE_ITERATOR_INITIALIZE_METHOD_STATUS_OK;
 676
 677         if (!dmesg_msg_iter) {
 678                 BT_COMP_LOGE_STR("Failed to allocate on dmesg message iterator structure.");
 679                 goto error;
 680         }
 681
 682         BT_ASSERT(dmesg_comp);
 683         dmesg_msg_iter->dmesg_comp = dmesg_comp;
 684         dmesg_msg_iter->self_msg_iter = self_msg_iter;
 685
 686         if (dmesg_comp->params.read_from_stdin) {
 687                 dmesg_msg_iter->fp = stdin;
 688         } else {
 689                 dmesg_msg_iter->fp = fopen(dmesg_comp->params.path->str, "r");
 690                 if (!dmesg_msg_iter->fp) {
 691                         BT_COMP_LOGE_ERRNO("Cannot open input file in read mode", ": path=\"%s\"",
 692                                 dmesg_comp->params.path->str);
 693                         goto error;
 694                 }
 695         }
 696
 697         bt_self_message_iterator_set_data(self_msg_iter,
 698                 dmesg_msg_iter);
 699         goto end;
 700
 701 error:
 702         destroy_dmesg_msg_iter(dmesg_msg_iter);
 703         bt_self_message_iterator_set_data(self_msg_iter, NULL);
 704         if (status >= 0) {
 705                 status = BT_COMPONENT_CLASS_MESSAGE_ITERATOR_INITIALIZE_METHOD_STATUS_ERROR;
 706         }
 707
 708 end:
 709         return status;
 710 }
 711
 712 BT_HIDDEN
 713 void dmesg_msg_iter_finalize(
 714                 bt_self_message_iterator *priv_msg_iter)
 715 {
 716         destroy_dmesg_msg_iter(bt_self_message_iterator_get_data(
 717                 priv_msg_iter));
 718 }
 719
 720 static
 721 bt_component_class_message_iterator_next_method_status dmesg_msg_iter_next_one(
 722                 struct dmesg_msg_iter *dmesg_msg_iter,
 723                 bt_message **msg)
 724 {
 725         ssize_t len;
 726         struct dmesg_component *dmesg_comp;
 727         bt_component_class_message_iterator_next_method_status status =
 728                 BT_COMPONENT_CLASS_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_OK;
 729
 730         BT_ASSERT_DBG(dmesg_msg_iter);
 731         dmesg_comp = dmesg_msg_iter->dmesg_comp;
 732         BT_ASSERT_DBG(dmesg_comp);
 733
 734         if (dmesg_msg_iter->state == STATE_DONE) {
 735                 status = BT_COMPONENT_CLASS_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_END;
 736                 goto end;
 737         }
 738
 739         if (dmesg_msg_iter->tmp_event_msg ||
 740                         dmesg_msg_iter->state == STATE_EMIT_STREAM_END) {
 741                 goto handle_state;
 742         }
 743
 744         while (true) {
 745                 const char *ch;
 746                 bool only_spaces = true;
 747
 748                 len = bt_getline(&dmesg_msg_iter->linebuf,
 749                         &dmesg_msg_iter->linebuf_len, dmesg_msg_iter->fp);
 750                 if (len < 0) {
 751                         if (errno == EINVAL) {
 752                                 status = BT_COMPONENT_CLASS_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_ERROR;
 753                         } else if (errno == ENOMEM) {
 754                                 status = BT_COMPONENT_CLASS_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_MEMORY_ERROR;
 755                         } else {
 756                                 if (dmesg_msg_iter->state == STATE_EMIT_STREAM_BEGINNING) {
 757                                         /* Stream did not even begin */
 758                                         status = BT_COMPONENT_CLASS_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_END;
 759                                         goto end;
 760                                 } else {
 761                                         /* End stream now */
 762                                         dmesg_msg_iter->state =
 763                                                 STATE_EMIT_STREAM_END;
 764                                         goto handle_state;
 765                                 }
 766                         }
 767
 768                         goto end;
 769                 }
 770
 771                 BT_ASSERT_DBG(dmesg_msg_iter->linebuf);
 772
 773                 /* Ignore empty lines, once trimmed */
 774                 for (ch = dmesg_msg_iter->linebuf; *ch != '\0'; ch++) {
 775                         if (!isspace(*ch)) {
 776                                 only_spaces = false;
 777                                 break;
 778                         }
 779                 }
 780
 781                 if (!only_spaces) {
 782                         break;
 783                 }
 784         }
 785
 786         dmesg_msg_iter->tmp_event_msg = create_msg_from_line(
 787                 dmesg_msg_iter, dmesg_msg_iter->linebuf);
 788         if (!dmesg_msg_iter->tmp_event_msg) {
 789                 BT_COMP_LOGE("Cannot create event message from line: "
 790                         "dmesg-comp-addr=%p, line=\"%s\"", dmesg_comp,
 791                         dmesg_msg_iter->linebuf);
 792                 goto end;
 793         }
 794
 795 handle_state:
 796         BT_ASSERT_DBG(dmesg_comp->trace);
 797
 798         switch (dmesg_msg_iter->state) {
 799         case STATE_EMIT_STREAM_BEGINNING:
 800                 BT_ASSERT_DBG(dmesg_msg_iter->tmp_event_msg);
 801                 *msg = bt_message_stream_beginning_create(
 802                         dmesg_msg_iter->self_msg_iter, dmesg_comp->stream);
 803                 dmesg_msg_iter->state = STATE_EMIT_EVENT;
 804                 break;
 805         case STATE_EMIT_EVENT:
 806                 BT_ASSERT_DBG(dmesg_msg_iter->tmp_event_msg);
 807                 *msg = dmesg_msg_iter->tmp_event_msg;
 808                 dmesg_msg_iter->tmp_event_msg = NULL;
 809                 break;
 810         case STATE_EMIT_STREAM_END:
 811                 *msg = bt_message_stream_end_create(
 812                         dmesg_msg_iter->self_msg_iter, dmesg_comp->stream);
 813                 dmesg_msg_iter->state = STATE_DONE;
 814                 break;
 815         default:
 816                 break;
 817         }
 818
 819         if (!*msg) {
 820                 BT_COMP_LOGE("Cannot create message: dmesg-comp-addr=%p",
 821                         dmesg_comp);
 822                 status = BT_COMPONENT_CLASS_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_ERROR;
 823         }
 824
 825 end:
 826         return status;
 827 }
 828
 829 BT_HIDDEN
 830 bt_component_class_message_iterator_next_method_status dmesg_msg_iter_next(
 831                 bt_self_message_iterator *self_msg_iter,
 832                 bt_message_array_const msgs, uint64_t capacity,
 833                 uint64_t *count)
 834 {
 835         struct dmesg_msg_iter *dmesg_msg_iter =
 836                 bt_self_message_iterator_get_data(
 837                         self_msg_iter);
 838         bt_component_class_message_iterator_next_method_status status =
 839                 BT_COMPONENT_CLASS_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_OK;
 840         uint64_t i = 0;
 841
 842         while (i < capacity &&
 843                         status == BT_COMPONENT_CLASS_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_OK) {
 844                 bt_message *priv_msg = NULL;
 845
 846                 status = dmesg_msg_iter_next_one(dmesg_msg_iter,
 847                         &priv_msg);
 848                 msgs[i] = priv_msg;
 849                 if (status == BT_COMPONENT_CLASS_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_OK) {
 850                         i++;
 851                 }
 852         }
 853
 854         if (i > 0) {
 855                 /*
 856                  * Even if dmesg_msg_iter_next_one() returned
 857                  * something else than
 858                  * BT_SELF_MESSAGE_ITERATOR_STATUS_OK, we
 859                  * accumulated message objects in the output
 860                  * message array, so we need to return
 861                  * BT_SELF_MESSAGE_ITERATOR_STATUS_OK so that they
 862                  * are transfered to downstream. This other status
 863                  * occurs again the next time muxer_msg_iter_do_next()
 864                  * is called, possibly without any accumulated
 865                  * message, in which case we'll return it.
 866                  */
 867                 *count = i;
 868                 status = BT_COMPONENT_CLASS_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_OK;
 869         }
 870
 871         return status;
 872 }
 873
 874 BT_HIDDEN
 875 bt_component_class_message_iterator_can_seek_beginning_method_status
 876 dmesg_msg_iter_can_seek_beginning(
 877                 bt_self_message_iterator *self_msg_iter, bt_bool *can_seek)
 878 {
 879         struct dmesg_msg_iter *dmesg_msg_iter =
 880                 bt_self_message_iterator_get_data(self_msg_iter);
 881
 882         /* Can't seek the beginning of the standard input stream */
 883         *can_seek = !dmesg_msg_iter->dmesg_comp->params.read_from_stdin;
 884
 885         return BT_COMPONENT_CLASS_MESSAGE_ITERATOR_CAN_SEEK_BEGINNING_METHOD_STATUS_OK;
 886 }
 887
 888 BT_HIDDEN
 889 bt_component_class_message_iterator_seek_beginning_method_status
 890 dmesg_msg_iter_seek_beginning(
 891                 bt_self_message_iterator *self_msg_iter)
 892 {
 893         struct dmesg_msg_iter *dmesg_msg_iter =
 894                 bt_self_message_iterator_get_data(self_msg_iter);
 895
 896         BT_ASSERT(!dmesg_msg_iter->dmesg_comp->params.read_from_stdin);
 897
 898         BT_MESSAGE_PUT_REF_AND_RESET(dmesg_msg_iter->tmp_event_msg);
 899         dmesg_msg_iter->last_clock_value = 0;
 900         dmesg_msg_iter->state = STATE_EMIT_STREAM_BEGINNING;
 901         return BT_COMPONENT_CLASS_MESSAGE_ITERATOR_SEEK_BEGINNING_METHOD_STATUS_OK;
 902 }