4 * Babeltrace CTF file system Reader Component
6 * Copyright 2016 Jérémie Galarneau <jeremie.galarneau@efficios.com>
8 * Author: Jérémie Galarneau <jeremie.galarneau@efficios.com>
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:
17 * The above copyright notice and this permission notice shall be included in
18 * all copies or substantial portions of the Software.
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
29 #include <babeltrace/ctf-ir/packet.h>
30 #include <babeltrace/ctf-ir/clock-class.h>
31 #include <babeltrace/graph/private-port.h>
32 #include <babeltrace/graph/private-component.h>
33 #include <babeltrace/graph/private-component-source.h>
34 #include <babeltrace/graph/private-notification-iterator.h>
35 #include <babeltrace/graph/component.h>
36 #include <babeltrace/graph/notification-iterator.h>
37 #include <babeltrace/graph/clock-class-priority-map.h>
38 #include <plugins-common.h>
44 #include "data-stream.h"
46 #include "../common/metadata/decoder.h"
48 #define PRINT_ERR_STREAM ctf_fs->error_fp
49 #define PRINT_PREFIX "ctf-fs"
51 #define METADATA_TEXT_SIG "/* CTF 1.8"
56 struct bt_notification_iterator_next_return
ctf_fs_iterator_next(
57 struct bt_private_notification_iterator
*iterator
)
59 struct ctf_fs_stream
*fs_stream
=
60 bt_private_notification_iterator_get_user_data(iterator
);
62 return ctf_fs_stream_next(fs_stream
);
65 void ctf_fs_iterator_finalize(struct bt_private_notification_iterator
*it
)
68 bt_private_notification_iterator_get_user_data(it
);
70 ctf_fs_stream_destroy(ctf_fs_stream
);
73 enum bt_notification_iterator_status
ctf_fs_iterator_init(
74 struct bt_private_notification_iterator
*it
,
75 struct bt_private_port
*port
)
77 struct ctf_fs_stream
*stream
= NULL
;
78 struct ctf_fs_component
*ctf_fs
;
79 struct ctf_fs_port_data
*port_data
;
80 struct bt_private_component
*priv_comp
=
81 bt_private_notification_iterator_get_private_component(it
);
82 enum bt_notification_iterator_status ret
=
83 BT_NOTIFICATION_ITERATOR_STATUS_OK
;
87 ctf_fs
= bt_private_component_get_user_data(priv_comp
);
89 ret
= BT_NOTIFICATION_ITERATOR_STATUS_INVAL
;
93 port_data
= bt_private_port_get_user_data(port
);
95 ret
= BT_NOTIFICATION_ITERATOR_STATUS_INVAL
;
99 stream
= ctf_fs_stream_create(ctf_fs
, port_data
->path
->str
);
104 ret
= bt_private_notification_iterator_set_user_data(it
, stream
);
113 (void) bt_private_notification_iterator_set_user_data(it
, NULL
);
115 if (ret
== BT_NOTIFICATION_ITERATOR_STATUS_OK
) {
116 ret
= BT_NOTIFICATION_ITERATOR_STATUS_ERROR
;
120 ctf_fs_stream_destroy(stream
);
126 void ctf_fs_destroy_data(struct ctf_fs_component
*ctf_fs
)
132 if (ctf_fs
->trace_path
) {
133 g_string_free(ctf_fs
->trace_path
, TRUE
);
136 if (ctf_fs
->port_data
) {
137 g_ptr_array_free(ctf_fs
->port_data
, TRUE
);
140 if (ctf_fs
->metadata
) {
141 ctf_fs_metadata_fini(ctf_fs
->metadata
);
142 g_free(ctf_fs
->metadata
);
145 bt_put(ctf_fs
->cc_prio_map
);
149 void ctf_fs_finalize(struct bt_private_component
*component
)
151 void *data
= bt_private_component_get_user_data(component
);
153 ctf_fs_destroy_data(data
);
157 void port_data_destroy(void *data
) {
158 struct ctf_fs_port_data
*port_data
= data
;
164 if (port_data
->path
) {
165 g_string_free(port_data
->path
, TRUE
);
172 int create_one_port(struct ctf_fs_component
*ctf_fs
,
173 const char *stream_basename
, const char *stream_path
)
176 struct bt_private_port
*port
= NULL
;
177 struct ctf_fs_port_data
*port_data
= NULL
;
178 GString
*port_name
= NULL
;
180 port_name
= g_string_new(NULL
);
185 /* Assign the name for the new output port */
186 g_string_assign(port_name
, "");
187 g_string_printf(port_name
, "trace0-stream-%s", stream_basename
);
188 PDBG("Creating one port named `%s` associated with path `%s`\n",
189 port_name
->str
, stream_path
);
191 /* Create output port for this file */
192 port
= bt_private_component_source_add_output_private_port(
193 ctf_fs
->priv_comp
, port_name
->str
);
198 port_data
= g_new0(struct ctf_fs_port_data
, 1);
203 port_data
->path
= g_string_new(stream_path
);
204 if (!port_data
->path
) {
208 ret
= bt_private_port_set_user_data(port
, port_data
);
213 g_ptr_array_add(ctf_fs
->port_data
, port_data
);
222 g_string_free(port_name
, TRUE
);
226 port_data_destroy(port_data
);
231 int create_ports(struct ctf_fs_component
*ctf_fs
)
234 const char *basename
;
235 GError
*error
= NULL
;
237 struct bt_private_port
*def_port
;
238 struct ctf_fs_file
*file
= NULL
;
240 /* Remove default port if needed */
241 def_port
= bt_private_component_source_get_default_output_private_port(
244 bt_private_port_remove_from_component(def_port
);
248 /* Create one output port for each stream file */
249 dir
= g_dir_open(ctf_fs
->trace_path
->str
, 0, &error
);
251 PERR("Cannot open directory `%s`: %s (code %d)\n",
252 ctf_fs
->trace_path
->str
, error
->message
,
257 while ((basename
= g_dir_read_name(dir
))) {
258 if (!strcmp(basename
, CTF_FS_METADATA_FILENAME
)) {
259 /* Ignore the metadata stream. */
260 PDBG("Ignoring metadata file `%s`\n", basename
);
264 if (basename
[0] == '.') {
265 PDBG("Ignoring hidden file `%s`\n", basename
);
269 /* Create the file. */
270 file
= ctf_fs_file_create(ctf_fs
);
272 PERR("Cannot create stream file object for file `%s`\n",
277 /* Create full path string. */
278 g_string_append_printf(file
->path
, "%s/%s",
279 ctf_fs
->trace_path
->str
, basename
);
280 if (!g_file_test(file
->path
->str
, G_FILE_TEST_IS_REGULAR
)) {
281 PDBG("Ignoring non-regular file `%s`\n", basename
);
282 ctf_fs_file_destroy(file
);
287 ret
= ctf_fs_file_open(ctf_fs
, file
, "rb");
289 PERR("Cannot open stream file `%s`\n", basename
);
293 if (file
->size
== 0) {
294 /* Skip empty stream. */
295 PDBG("Ignoring empty file `%s`\n", basename
);
296 ctf_fs_file_destroy(file
);
301 ret
= create_one_port(ctf_fs
, basename
, file
->path
->str
);
303 PERR("Cannot create output port for file `%s`\n",
308 ctf_fs_file_destroy(file
);
327 ctf_fs_file_destroy(file
);
332 int create_cc_prio_map(struct ctf_fs_component
*ctf_fs
)
339 ctf_fs
->cc_prio_map
= bt_clock_class_priority_map_create();
340 if (!ctf_fs
->cc_prio_map
) {
345 count
= bt_ctf_trace_get_clock_class_count(ctf_fs
->metadata
->trace
);
348 for (i
= 0; i
< count
; i
++) {
349 struct bt_ctf_clock_class
*clock_class
=
350 bt_ctf_trace_get_clock_class(ctf_fs
->metadata
->trace
,
354 ret
= bt_clock_class_priority_map_add_clock_class(
355 ctf_fs
->cc_prio_map
, clock_class
, 0);
368 struct ctf_fs_component
*ctf_fs_create(struct bt_private_component
*priv_comp
,
369 struct bt_value
*params
)
371 struct ctf_fs_component
*ctf_fs
;
372 struct bt_value
*value
= NULL
;
376 ctf_fs
= g_new0(struct ctf_fs_component
, 1);
382 * We don't need to get a new reference here because as long as
383 * our private ctf_fs_component object exists, the containing
384 * private component should also exist.
386 ctf_fs
->priv_comp
= priv_comp
;
388 /* FIXME: should probably look for a source URI */
389 value
= bt_value_map_get(params
, "path");
390 if (!value
|| bt_value_is_null(value
) || !bt_value_is_string(value
)) {
394 ret
= bt_value_string_get(value
, &path
);
399 ctf_fs
->port_data
= g_ptr_array_new_with_free_func(port_data_destroy
);
400 if (!ctf_fs
->port_data
) {
404 ctf_fs
->trace_path
= g_string_new(path
);
405 if (!ctf_fs
->trace_path
) {
411 value
= bt_value_map_get(params
, "offset-s");
415 if (!bt_value_is_integer(value
)) {
417 "offset-s should be an integer\n");
420 ret
= bt_value_integer_get(value
, &offset
);
421 if (ret
!= BT_VALUE_STATUS_OK
) {
423 "Failed to get offset-s value\n");
426 ctf_fs
->options
.clock_offset
= offset
;
430 value
= bt_value_map_get(params
, "offset-ns");
434 if (!bt_value_is_integer(value
)) {
436 "offset-ns should be an integer\n");
439 ret
= bt_value_integer_get(value
, &offset
);
440 if (ret
!= BT_VALUE_STATUS_OK
) {
442 "Failed to get offset-ns value\n");
445 ctf_fs
->options
.clock_offset_ns
= offset
;
449 ctf_fs
->error_fp
= stderr
;
450 ctf_fs
->page_size
= (size_t) getpagesize();
452 // FIXME: check error.
453 ctf_fs
->metadata
= g_new0(struct ctf_fs_metadata
, 1);
454 if (!ctf_fs
->metadata
) {
458 ret
= ctf_fs_metadata_set_trace(ctf_fs
);
463 ret
= create_cc_prio_map(ctf_fs
);
468 ret
= create_ports(ctf_fs
);
476 ctf_fs_destroy_data(ctf_fs
);
483 enum bt_component_status
ctf_fs_init(struct bt_private_component
*priv_comp
,
484 struct bt_value
*params
, UNUSED_VAR
void *init_method_data
)
486 struct ctf_fs_component
*ctf_fs
;
487 enum bt_component_status ret
= BT_COMPONENT_STATUS_OK
;
490 ctf_fs_debug
= g_strcmp0(getenv("CTF_FS_DEBUG"), "1") == 0;
491 ctf_fs
= ctf_fs_create(priv_comp
, params
);
493 ret
= BT_COMPONENT_STATUS_NOMEM
;
497 ret
= bt_private_component_set_user_data(priv_comp
, ctf_fs
);
498 if (ret
!= BT_COMPONENT_STATUS_OK
) {
504 (void) bt_private_component_set_user_data(priv_comp
, NULL
);
505 ctf_fs_destroy_data(ctf_fs
);
510 struct bt_value
*ctf_fs_query(struct bt_component_class
*comp_class
,
511 const char *object
, struct bt_value
*params
)
513 struct bt_value
*results
= NULL
;
514 struct bt_value
*path_value
= NULL
;
515 char *metadata_text
= NULL
;
516 FILE *metadata_fp
= NULL
;
517 GString
*g_metadata_text
= NULL
;
519 if (strcmp(object
, "metadata-info") == 0) {
525 results
= bt_value_map_create();
530 if (!bt_value_is_map(params
)) {
532 "Query parameters is not a map value object\n");
536 path_value
= bt_value_map_get(params
, "path");
537 ret
= bt_value_string_get(path_value
, &path
);
540 "Cannot get `path` string parameter\n");
545 metadata_fp
= ctf_fs_metadata_open_file(path
);
548 "Cannot open trace at path `%s`\n", path
);
552 is_packetized
= ctf_metadata_decoder_is_packetized(metadata_fp
,
556 ret
= ctf_metadata_decoder_packetized_file_stream_to_buf(
557 metadata_fp
, &metadata_text
, bo
);
560 "Cannot decode packetized metadata file\n");
566 fseek(metadata_fp
, 0, SEEK_END
);
567 filesize
= ftell(metadata_fp
);
569 metadata_text
= malloc(filesize
+ 1);
570 if (!metadata_text
) {
572 "Cannot allocate buffer for metadata text\n");
576 if (fread(metadata_text
, filesize
, 1, metadata_fp
) !=
579 "Cannot read metadata file\n");
583 metadata_text
[filesize
] = '\0';
586 g_metadata_text
= g_string_new(NULL
);
587 if (!g_metadata_text
) {
591 if (strncmp(metadata_text
, METADATA_TEXT_SIG
,
592 sizeof(METADATA_TEXT_SIG
) - 1) != 0) {
593 g_string_assign(g_metadata_text
, METADATA_TEXT_SIG
);
594 g_string_append(g_metadata_text
, " */\n\n");
597 g_string_append(g_metadata_text
, metadata_text
);
599 ret
= bt_value_map_insert_string(results
, "text",
600 g_metadata_text
->str
);
602 fprintf(stderr
, "Cannot insert metadata text into results\n");
606 ret
= bt_value_map_insert_bool(results
, "is-packetized",
609 fprintf(stderr
, "Cannot insert is packetized into results\n");
613 fprintf(stderr
, "Unknown query object `%s`\n", object
);
626 if (g_metadata_text
) {
627 g_string_free(g_metadata_text
, TRUE
);