Fix: flt.lttng-utils.debug-info: extra colon in `bin` member
[babeltrace.git] / plugins / ctf / fs-src / query.c
CommitLineData
04c0ba87
JG
1/*
2 * query.c
3 *
4 * Babeltrace CTF file system Reader Component queries
5 *
6 * Copyright 2017 Jérémie Galarneau <jeremie.galarneau@efficios.com>
7 *
8 * Permission is hereby granted, free of charge, to any person obtaining a copy
9 * of this software and associated documentation files (the "Software"), to deal
10 * in the Software without restriction, including without limitation the rights
11 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
12 * copies of the Software, and to permit persons to whom the Software is
13 * furnished to do so, subject to the following conditions:
14 *
15 * The above copyright notice and this permission notice shall be included in
16 * all copies or substantial portions of the Software.
17 *
18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
19 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
20 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
21 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
22 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
23 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
24 * SOFTWARE.
25 */
26
27#include "query.h"
28#include <stdbool.h>
f6ccaed9 29#include <babeltrace/assert-internal.h>
04c0ba87
JG
30#include "metadata.h"
31#include "../common/metadata/decoder.h"
55314f2a 32#include <babeltrace/common-internal.h>
97ade20b 33#include <babeltrace/babeltrace-internal.h>
9d408fca 34#include <babeltrace/babeltrace.h>
97ade20b 35#include "fs.h"
55314f2a
JG
36
37#define BT_LOG_TAG "PLUGIN-CTF-FS-QUERY-SRC"
38#include "logging.h"
04c0ba87
JG
39
40#define METADATA_TEXT_SIG "/* CTF 1.8"
41
97ade20b
JG
42struct range {
43 int64_t begin_ns;
44 int64_t end_ns;
45 bool set;
46};
47
04c0ba87 48BT_HIDDEN
4cdfc5e8 49bt_query_status metadata_info_query(
b19ff26f
PP
50 bt_self_component_class_source *comp_class,
51 const bt_value *params,
52 const bt_value **user_result)
04c0ba87 53{
4cdfc5e8 54 bt_query_status status = BT_QUERY_STATUS_OK;
b19ff26f
PP
55 bt_value *result = NULL;
56 const bt_value *path_value = NULL;
04c0ba87
JG
57 char *metadata_text = NULL;
58 FILE *metadata_fp = NULL;
59 GString *g_metadata_text = NULL;
60 int ret;
61 int bo;
62 const char *path;
63 bool is_packetized;
64
05e21286 65 result = bt_value_map_create();
da91b29a 66 if (!result) {
d94d92ac 67 status = BT_QUERY_STATUS_NOMEM;
04c0ba87
JG
68 goto error;
69 }
70
f6ccaed9
PP
71 BT_ASSERT(params);
72
04c0ba87 73 if (!bt_value_is_map(params)) {
f7d0e29c 74 BT_LOGE_STR("Query parameters is not a map value object.");
d94d92ac 75 status = BT_QUERY_STATUS_INVALID_PARAMS;
04c0ba87
JG
76 goto error;
77 }
78
05e21286 79 path_value = bt_value_map_borrow_entry_value_const(params, "path");
601b0d3c 80 path = bt_value_string_get(path_value);
04c0ba87 81
f6ccaed9 82 BT_ASSERT(path);
04c0ba87
JG
83 metadata_fp = ctf_fs_metadata_open_file(path);
84 if (!metadata_fp) {
f7d0e29c 85 BT_LOGE("Cannot open trace metadata: path=\"%s\".", path);
04c0ba87
JG
86 goto error;
87 }
88
89 is_packetized = ctf_metadata_decoder_is_packetized(metadata_fp,
90 &bo);
91
92 if (is_packetized) {
93 ret = ctf_metadata_decoder_packetized_file_stream_to_buf(
94 metadata_fp, &metadata_text, bo);
95 if (ret) {
f7d0e29c
JG
96 BT_LOGE("Cannot decode packetized metadata file: path=\"%s\"",
97 path);
04c0ba87
JG
98 goto error;
99 }
100 } else {
101 long filesize;
102
677e1a21
MD
103 ret = fseek(metadata_fp, 0, SEEK_END);
104 if (ret) {
f7d0e29c
JG
105 BT_LOGE_ERRNO("Failed to seek to the end of the metadata file",
106 ": path=\"%s\"", path);
677e1a21
MD
107 goto error;
108 }
04c0ba87 109 filesize = ftell(metadata_fp);
677e1a21 110 if (filesize < 0) {
f7d0e29c
JG
111 BT_LOGE_ERRNO("Failed to get the current position in the metadata file",
112 ": path=\"%s\"", path);
677e1a21
MD
113 goto error;
114 }
04c0ba87
JG
115 rewind(metadata_fp);
116 metadata_text = malloc(filesize + 1);
117 if (!metadata_text) {
f7d0e29c 118 BT_LOGE_STR("Cannot allocate buffer for metadata text.");
04c0ba87
JG
119 goto error;
120 }
121
122 if (fread(metadata_text, filesize, 1, metadata_fp) != 1) {
f7d0e29c
JG
123 BT_LOGE_ERRNO("Cannot read metadata file", ": path=\"%s\"",
124 path);
04c0ba87
JG
125 goto error;
126 }
127
128 metadata_text[filesize] = '\0';
129 }
130
131 g_metadata_text = g_string_new(NULL);
132 if (!g_metadata_text) {
133 goto error;
134 }
135
136 if (strncmp(metadata_text, METADATA_TEXT_SIG,
137 sizeof(METADATA_TEXT_SIG) - 1) != 0) {
138 g_string_assign(g_metadata_text, METADATA_TEXT_SIG);
139 g_string_append(g_metadata_text, " */\n\n");
140 }
141
142 g_string_append(g_metadata_text, metadata_text);
143
05e21286 144 ret = bt_value_map_insert_string_entry(result, "text",
04c0ba87
JG
145 g_metadata_text->str);
146 if (ret) {
f7d0e29c 147 BT_LOGE_STR("Cannot insert metadata text into query result.");
04c0ba87
JG
148 goto error;
149 }
150
05e21286 151 ret = bt_value_map_insert_bool_entry(result, "is-packetized",
04c0ba87
JG
152 is_packetized);
153 if (ret) {
f7d0e29c 154 BT_LOGE_STR("Cannot insert \"is-packetized\" attribute into query result.");
04c0ba87
JG
155 goto error;
156 }
157
158 goto end;
159
160error:
c5b9b441 161 BT_VALUE_PUT_REF_AND_RESET(result);
d94d92ac 162 result = NULL;
c7eee084 163
d94d92ac
PP
164 if (status >= 0) {
165 status = BT_QUERY_STATUS_ERROR;
c7eee084 166 }
04c0ba87
JG
167
168end:
04c0ba87
JG
169 free(metadata_text);
170
171 if (g_metadata_text) {
172 g_string_free(g_metadata_text, TRUE);
173 }
174
175 if (metadata_fp) {
176 fclose(metadata_fp);
177 }
c7eee084 178
05e21286 179 *user_result = result;
d94d92ac 180 return status;
97ade20b 181}
9ec238a8 182
97ade20b 183static
b19ff26f 184int add_range(bt_value *info, struct range *range,
97ade20b
JG
185 const char *range_name)
186{
187 int ret = 0;
4cdfc5e8 188 bt_value_status status;
b19ff26f 189 bt_value *range_map = NULL;
97ade20b
JG
190
191 if (!range->set) {
192 /* Not an error. */
193 goto end;
194 }
195
05e21286 196 range_map = bt_value_map_create();
97ade20b
JG
197 if (!range_map) {
198 ret = -1;
199 goto end;
200 }
201
05e21286 202 status = bt_value_map_insert_integer_entry(range_map, "begin",
97ade20b
JG
203 range->begin_ns);
204 if (status != BT_VALUE_STATUS_OK) {
205 ret = -1;
206 goto end;
207 }
208
05e21286 209 status = bt_value_map_insert_integer_entry(range_map, "end",
97ade20b
JG
210 range->end_ns);
211 if (status != BT_VALUE_STATUS_OK) {
212 ret = -1;
213 goto end;
214 }
215
05e21286
PP
216 status = bt_value_map_insert_entry(info, range_name,
217 range_map);
97ade20b
JG
218 if (status != BT_VALUE_STATUS_OK) {
219 ret = -1;
220 goto end;
221 }
da91b29a 222
97ade20b 223end:
c5b9b441 224 bt_value_put_ref(range_map);
97ade20b
JG
225 return ret;
226}
227
228static
60363f2f 229int add_stream_ids(bt_value *info, struct ctf_fs_ds_file_group *ds_file_group)
97ade20b
JG
230{
231 int ret = 0;
4cdfc5e8 232 bt_value_status status;
97ade20b 233
60363f2f 234 if (ds_file_group->stream_id != UINT64_C(-1)) {
05e21286 235 status = bt_value_map_insert_integer_entry(info, "id",
60363f2f 236 (int64_t) ds_file_group->stream_id);
97ade20b
JG
237 if (status != BT_VALUE_STATUS_OK) {
238 ret = -1;
239 goto end;
240 }
241 }
242
60363f2f
PP
243 status = bt_value_map_insert_integer_entry(info, "class-id",
244 (int64_t) ds_file_group->sc->id);
97ade20b
JG
245 if (status != BT_VALUE_STATUS_OK) {
246 ret = -1;
247 goto end;
248 }
0b29603d 249
97ade20b 250end:
97ade20b
JG
251 return ret;
252}
253
254static
255int populate_stream_info(struct ctf_fs_ds_file_group *group,
b19ff26f 256 bt_value *group_info, struct range *stream_range)
97ade20b
JG
257{
258 int ret = 0;
259 size_t file_idx;
4cdfc5e8 260 bt_value_status status;
b19ff26f 261 bt_value *file_paths;
97ade20b
JG
262
263 stream_range->begin_ns = INT64_MAX;
264 stream_range->end_ns = 0;
265
05e21286 266 file_paths = bt_value_array_create();
97ade20b
JG
267 if (!file_paths) {
268 ret = -1;
269 goto end;
270 }
271
272 for (file_idx = 0; file_idx < group->ds_file_infos->len; file_idx++) {
273 int64_t file_begin_epoch, file_end_epoch;
274 struct ctf_fs_ds_file_info *info =
60363f2f
PP
275 g_ptr_array_index(group->ds_file_infos,
276 file_idx);
97ade20b
JG
277
278 if (!info->index || info->index->entries->len == 0) {
279 BT_LOGW("Cannot determine range of unindexed stream file \'%s\'",
280 info->path->str);
281 ret = -1;
282 goto end;
283 }
284
05e21286 285 status = bt_value_array_append_string_element(file_paths,
97ade20b
JG
286 info->path->str);
287 if (status != BT_VALUE_STATUS_OK) {
288 ret = -1;
289 goto end;
290 }
291
292 /*
293 * file range is from timestamp_begin of the first entry to the
294 * timestamp_end of the last entry.
295 */
296 file_begin_epoch = ((struct ctf_fs_ds_index_entry *) &g_array_index(info->index->entries,
297 struct ctf_fs_ds_index_entry, 0))->timestamp_begin_ns;
298 file_end_epoch = ((struct ctf_fs_ds_index_entry *) &g_array_index(info->index->entries,
299 struct ctf_fs_ds_index_entry, info->index->entries->len - 1))->timestamp_end_ns;
300
301 stream_range->begin_ns = min(stream_range->begin_ns, file_begin_epoch);
302 stream_range->end_ns = max(stream_range->end_ns, file_end_epoch);
303 stream_range->set = true;
304 }
305
306 if (stream_range->set) {
307 ret = add_range(group_info, stream_range, "range-ns");
308 if (ret) {
309 goto end;
310 }
311 }
312
05e21286
PP
313 status = bt_value_map_insert_entry(group_info, "paths",
314 file_paths);
97ade20b
JG
315 if (status != BT_VALUE_STATUS_OK) {
316 ret = -1;
317 goto end;
318 }
319
60363f2f 320 ret = add_stream_ids(group_info, group);
97ade20b
JG
321 if (ret) {
322 goto end;
323 }
324end:
c5b9b441 325 bt_value_put_ref(file_paths);
97ade20b
JG
326 return ret;
327}
328
329static
f280892e 330int populate_trace_info(const struct ctf_fs_trace *trace, bt_value *trace_info)
97ade20b
JG
331{
332 int ret = 0;
333 size_t group_idx;
4cdfc5e8 334 bt_value_status status;
f280892e 335 bt_value *file_groups = NULL;
97ade20b
JG
336 struct range trace_range = {
337 .begin_ns = INT64_MAX,
338 .end_ns = 0,
339 .set = false,
340 };
341 struct range trace_intersection = {
342 .begin_ns = 0,
343 .end_ns = INT64_MAX,
344 .set = false,
345 };
346
f280892e
SM
347 BT_ASSERT(trace->ds_file_groups);
348 /* Add trace range info only if it contains streams. */
349 if (trace->ds_file_groups->len == 0) {
350 ret = -1;
351 goto end;
352 }
353
05e21286 354 file_groups = bt_value_array_create();
97ade20b
JG
355 if (!file_groups) {
356 goto end;
357 }
358
05e21286 359 status = bt_value_map_insert_string_entry(trace_info, "name",
f280892e 360 trace->name->str);
97ade20b
JG
361 if (status != BT_VALUE_STATUS_OK) {
362 ret = -1;
363 goto end;
364 }
05e21286 365 status = bt_value_map_insert_string_entry(trace_info, "path",
f280892e 366 trace->path->str);
97ade20b
JG
367 if (status != BT_VALUE_STATUS_OK) {
368 ret = -1;
369 goto end;
370 }
371
97ade20b
JG
372 /* Find range of all stream groups, and of the trace. */
373 for (group_idx = 0; group_idx < trace->ds_file_groups->len;
374 group_idx++) {
b19ff26f 375 bt_value *group_info;
97ade20b
JG
376 struct range group_range = { .set = false };
377 struct ctf_fs_ds_file_group *group = g_ptr_array_index(
378 trace->ds_file_groups, group_idx);
379
05e21286 380 group_info = bt_value_map_create();
97ade20b
JG
381 if (!group_info) {
382 ret = -1;
383 goto end;
384 }
385
386 ret = populate_stream_info(group, group_info, &group_range);
387 if (ret) {
c5b9b441 388 bt_value_put_ref(group_info);
97ade20b
JG
389 goto end;
390 }
391
392 if (group_range.set) {
393 trace_range.begin_ns = min(trace_range.begin_ns,
394 group_range.begin_ns);
395 trace_range.end_ns = max(trace_range.end_ns,
396 group_range.end_ns);
397 trace_range.set = true;
398
399 trace_intersection.begin_ns = max(trace_intersection.begin_ns,
400 group_range.begin_ns);
401 trace_intersection.end_ns = min(trace_intersection.end_ns,
402 group_range.end_ns);
403 trace_intersection.set = true;
05e21286 404 status = bt_value_array_append_element(
da91b29a 405 file_groups,
05e21286 406 group_info);
c5b9b441 407 bt_value_put_ref(group_info);
97ade20b
JG
408 if (status != BT_VALUE_STATUS_OK) {
409 goto end;
410 }
411 }
412 }
413
414 ret = add_range(trace_info, &trace_range, "range-ns");
415 if (ret) {
416 goto end;
417 }
c6daf8b2
PP
418
419 if (trace_intersection.begin_ns < trace_intersection.end_ns) {
420 ret = add_range(trace_info, &trace_intersection,
421 "intersection-range-ns");
422 if (ret) {
423 goto end;
424 }
97ade20b
JG
425 }
426
05e21286
PP
427 status = bt_value_map_insert_entry(trace_info, "streams",
428 file_groups);
c5b9b441 429 BT_VALUE_PUT_REF_AND_RESET(file_groups);
97ade20b
JG
430 if (status != BT_VALUE_STATUS_OK) {
431 ret = -1;
432 goto end;
433 }
434
435end:
c5b9b441 436 bt_value_put_ref(file_groups);
97ade20b
JG
437 return ret;
438}
439
440BT_HIDDEN
4cdfc5e8 441bt_query_status trace_info_query(
b19ff26f
PP
442 bt_self_component_class_source *comp_class,
443 const bt_value *params,
444 const bt_value **user_result)
97ade20b 445{
f280892e 446 struct ctf_fs_component *ctf_fs = NULL;
4cdfc5e8 447 bt_query_status status = BT_QUERY_STATUS_OK;
b19ff26f 448 bt_value *result = NULL;
f280892e 449 const bt_value *paths_value = NULL;
97ade20b 450 int ret = 0;
f280892e 451 guint i;
97ade20b 452
f6ccaed9
PP
453 BT_ASSERT(params);
454
97ade20b
JG
455 if (!bt_value_is_map(params)) {
456 BT_LOGE("Query parameters is not a map value object.");
d94d92ac 457 status = BT_QUERY_STATUS_INVALID_PARAMS;
97ade20b
JG
458 goto error;
459 }
460
f280892e
SM
461 paths_value = bt_value_map_borrow_entry_value_const(params, "paths");
462 if (!validate_paths_parameter(paths_value)) {
97ade20b
JG
463 goto error;
464 }
97ade20b 465
f280892e
SM
466 ctf_fs = ctf_fs_component_create();
467 if (!ctf_fs) {
97ade20b
JG
468 goto error;
469 }
470
f280892e 471 if (ctf_fs_component_create_ctf_fs_traces(NULL, ctf_fs, paths_value)) {
97ade20b
JG
472 goto error;
473 }
474
05e21286 475 result = bt_value_array_create();
da91b29a 476 if (!result) {
d94d92ac 477 status = BT_QUERY_STATUS_NOMEM;
97ade20b
JG
478 goto error;
479 }
480
f280892e
SM
481 for (i = 0; i < ctf_fs->traces->len; i++) {
482 struct ctf_fs_trace *trace;
b19ff26f 483 bt_value *trace_info;
f280892e
SM
484 bt_value_status status;
485
486 trace = g_ptr_array_index(ctf_fs->traces, i);
487 BT_ASSERT(trace);
97ade20b 488
05e21286 489 trace_info = bt_value_map_create();
97ade20b
JG
490 if (!trace_info) {
491 BT_LOGE("Failed to create trace info map.");
492 goto error;
493 }
494
f280892e 495 ret = populate_trace_info(trace, trace_info);
97ade20b 496 if (ret) {
c5b9b441 497 bt_value_put_ref(trace_info);
97ade20b
JG
498 goto error;
499 }
500
05e21286 501 status = bt_value_array_append_element(result, trace_info);
c5b9b441 502 bt_value_put_ref(trace_info);
97ade20b
JG
503 if (status != BT_VALUE_STATUS_OK) {
504 goto error;
505 }
506 }
507
508 goto end;
509
510error:
c5b9b441 511 BT_VALUE_PUT_REF_AND_RESET(result);
d94d92ac 512 result = NULL;
c7eee084 513
d94d92ac
PP
514 if (status >= 0) {
515 status = BT_QUERY_STATUS_ERROR;
c7eee084
PP
516 }
517
97ade20b 518end:
f280892e
SM
519 if (ctf_fs) {
520 ctf_fs_destroy(ctf_fs);
521 ctf_fs = NULL;
97ade20b 522 }
d94d92ac 523
05e21286 524 *user_result = result;
d94d92ac 525 return status;
04c0ba87 526}
This page took 0.056508 seconds and 4 git commands to generate.