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