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/plugin/plugin-system.h>
30 #include <babeltrace/plugin/notification/iterator.h>
36 #include "data-stream.h"
38 static bool ctf_fs_debug
;
41 struct bt_notification
*ctf_fs_iterator_get(
42 struct bt_notification_iterator
*iterator
)
44 struct bt_notification
*notification
= NULL
;
45 struct ctf_fs_component
*ctf_fs
;
46 struct bt_component
*component
= bt_notification_iterator_get_component(
53 ctf_fs
= bt_component_get_private_data(component
);
58 notification
= bt_get(ctf_fs
->current_notification
);
64 enum bt_notification_iterator_status
ctf_fs_iterator_next(
65 struct bt_notification_iterator
*iterator
)
67 enum bt_notification_iterator_status ret
;
68 struct bt_ctf_notif_iter_notif
*notification
;
69 // struct bt_notification *notification = NULL;
70 struct ctf_fs_component
*ctf_fs
;
71 struct bt_component
*component
= bt_notification_iterator_get_component(
75 ret
= BT_NOTIFICATION_ITERATOR_STATUS_ERROR
;
79 ctf_fs
= bt_component_get_private_data(component
);
82 ret
= ctf_fs_data_stream_get_next_notification(ctf_fs
, ¬ification
);
83 if (ret
|| !notification
) {
87 switch (notification
->type
) {
88 case BT_CTF_NOTIF_ITER_NOTIF_NEW_PACKET
:
90 struct bt_ctf_notif_iter_notif_new_packet
*notif
=
91 (struct bt_ctf_notif_iter_notif_new_packet
*) notification
;
95 case BT_CTF_NOTIF_ITER_NOTIF_EVENT
:
97 struct bt_ctf_notif_iter_notif_event
*notif
=
98 (struct bt_ctf_notif_iter_notif_event
*) notification
;
102 case BT_CTF_NOTIF_ITER_NOTIF_END_OF_PACKET
:
104 struct bt_ctf_notif_iter_notif_end_of_packet
*notif
=
105 (struct bt_ctf_notif_iter_notif_end_of_packet
*) notification
;
113 bt_ctf_notif_iter_notif_destroy(notification
);
115 return BT_NOTIFICATION_ITERATOR_STATUS_OK
;
119 void ctf_fs_iterator_destroy_data(struct ctf_fs_iterator
*ctf_it
)
125 void ctf_fs_iterator_destroy(struct bt_notification_iterator
*it
)
127 void *data
= bt_notification_iterator_get_private_data(it
);
129 ctf_fs_iterator_destroy_data(data
);
133 enum bt_component_status
ctf_fs_iterator_init(struct bt_component
*source
,
134 struct bt_notification_iterator
*it
)
136 struct ctf_fs_iterator
*ctf_it
;
137 enum bt_component_status ret
= BT_COMPONENT_STATUS_OK
;
139 assert(source
&& it
);
140 ctf_it
= g_new0(struct ctf_fs_iterator
, 1);
142 ret
= BT_COMPONENT_STATUS_NOMEM
;
146 ret
= bt_notification_iterator_set_get_cb(it
, ctf_fs_iterator_get
);
151 ret
= bt_notification_iterator_set_next_cb(it
, ctf_fs_iterator_next
);
156 ret
= bt_notification_iterator_set_destroy_cb(it
,
157 ctf_fs_iterator_destroy
);
162 ret
= bt_notification_iterator_set_private_data(it
, ctf_it
);
169 (void) bt_notification_iterator_set_private_data(it
, NULL
);
170 ctf_fs_iterator_destroy_data(ctf_it
);
175 void ctf_fs_destroy_data(struct ctf_fs_component
*component
)
177 if (component
->trace_path
) {
178 g_string_free(component
->trace_path
, TRUE
);
181 ctf_fs_metadata_fini(&component
->metadata
);
182 ctf_fs_data_stream_fini(&component
->data_stream
);
183 BT_PUT(component
->current_notification
);
188 void ctf_fs_destroy(struct bt_component
*component
)
190 void *data
= bt_component_get_private_data(component
);
192 ctf_fs_destroy_data(data
);
196 struct ctf_fs_component
*ctf_fs_create(struct bt_value
*params
)
198 struct ctf_fs_component
*ctf_fs
;
199 struct bt_value
*value
= NULL
;
201 enum bt_value_status ret
;
203 ctf_fs
= g_new0(struct ctf_fs_component
, 1);
208 /* FIXME: should probably look for a source URI */
209 value
= bt_value_map_get(params
, "path");
210 if (!value
|| bt_value_is_null(value
) || !bt_value_is_string(value
)) {
214 ret
= bt_value_string_get(value
, &path
);
215 if (ret
!= BT_VALUE_STATUS_OK
) {
219 ctf_fs
->trace_path
= g_string_new(path
);
220 if (!ctf_fs
->trace_path
) {
224 ctf_fs
->error_fp
= stderr
;
225 ctf_fs
->page_size
= (size_t) getpagesize();
226 ctf_fs_data_stream_init(ctf_fs
, &ctf_fs
->data_stream
);
227 ctf_fs_metadata_set_trace(ctf_fs
);
228 ctf_fs_data_stream_open_streams(ctf_fs
);
232 ctf_fs_destroy_data(ctf_fs
);
239 enum bt_component_status
ctf_fs_init(struct bt_component
*source
,
240 struct bt_value
*params
)
242 struct ctf_fs_component
*ctf_fs
;
243 enum bt_component_status ret
= BT_COMPONENT_STATUS_OK
;
246 ctf_fs_debug
= g_strcmp0(getenv("CTF_FS_DEBUG"), "1") == 0;
247 ctf_fs
= ctf_fs_create(params
);
249 ret
= BT_COMPONENT_STATUS_NOMEM
;
253 ret
= bt_component_set_destroy_cb(source
, ctf_fs_destroy
);
254 if (ret
!= BT_COMPONENT_STATUS_OK
) {
258 ret
= bt_component_set_private_data(source
, ctf_fs
);
259 if (ret
!= BT_COMPONENT_STATUS_OK
) {
263 ret
= bt_component_source_set_iterator_init_cb(source
,
264 ctf_fs_iterator_init
);
265 if (ret
!= BT_COMPONENT_STATUS_OK
) {
272 (void) bt_component_set_private_data(source
, NULL
);
273 ctf_fs_destroy_data(ctf_fs
);