2 * Copyright 2016-2017 - Philippe Proulx <pproulx@efficios.com>
4 * Permission is hereby granted, free of charge, to any person obtaining a copy
5 * of this software and associated documentation files (the "Software"), to deal
6 * in the Software without restriction, including without limitation the rights
7 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8 * copies of the Software, and to permit persons to whom the Software is
9 * furnished to do so, subject to the following conditions:
11 * The above copyright notice and this permission notice shall be included in
12 * all copies or substantial portions of the Software.
20 #include <babeltrace/compat/uuid-internal.h>
21 #include <babeltrace/compat/memstream-internal.h>
22 #include <babeltrace/ctf-ir/trace.h>
29 #define PRINT_ERR_STREAM mdec->err_stream
30 #define PRINT_PREFIX "metadata-decoder"
33 #define TSDL_MAGIC 0x75d11d57
35 struct ctf_metadata_decoder
{
36 struct ctf_visitor_generate_ir
*visitor
;
43 struct packet_header
{
47 uint32_t content_size
;
49 uint8_t compression_scheme
;
50 uint8_t encryption_scheme
;
51 uint8_t checksum_scheme
;
54 } __attribute__((__packed__
));
57 bool ctf_metadata_decoder_is_packetized(FILE *fp
, int *byte_order
)
63 len
= fread(&magic
, sizeof(magic
), 1, fp
);
69 if (magic
== TSDL_MAGIC
) {
71 *byte_order
= BYTE_ORDER
;
72 } else if (magic
== GUINT32_SWAP_LE_BE(TSDL_MAGIC
)) {
74 *byte_order
= BYTE_ORDER
== BIG_ENDIAN
?
75 LITTLE_ENDIAN
: BIG_ENDIAN
;
86 bool is_version_valid(unsigned int major
, unsigned int minor
)
88 return major
== 1 && minor
== 8;
92 int decode_packet(struct ctf_metadata_decoder
*mdec
, FILE *in_fp
, FILE *out_fp
,
95 struct packet_header header
;
96 size_t readlen
, writelen
, toread
;
97 uint8_t buf
[512 + 1]; /* + 1 for debug-mode \0 */
100 readlen
= fread(&header
, sizeof(header
), 1, in_fp
);
101 if (feof(in_fp
) != 0) {
108 if (byte_order
!= BYTE_ORDER
) {
109 header
.magic
= GUINT32_SWAP_LE_BE(header
.magic
);
110 header
.checksum
= GUINT32_SWAP_LE_BE(header
.checksum
);
111 header
.content_size
= GUINT32_SWAP_LE_BE(header
.content_size
);
112 header
.packet_size
= GUINT32_SWAP_LE_BE(header
.packet_size
);
115 if (header
.compression_scheme
) {
116 PERR("Metadata packet compression not supported yet\n");
120 if (header
.encryption_scheme
) {
121 PERR("Metadata packet encryption not supported yet\n");
125 if (header
.checksum
|| header
.checksum_scheme
) {
126 PERR("Metadata packet checksum verification not supported yet\n");
130 if (!is_version_valid(header
.major
, header
.minor
)) {
131 PERR("Invalid metadata version: %u.%u\n", header
.major
,
136 /* Set expected trace UUID if not set; otherwise validate it */
138 if (!mdec
->is_uuid_set
) {
139 memcpy(mdec
->uuid
, header
.uuid
, sizeof(header
.uuid
));
140 mdec
->is_uuid_set
= true;
141 } else if (bt_uuid_compare(header
.uuid
, mdec
->uuid
)) {
142 PERR("Metadata UUID mismatch between packets of the same stream\n");
147 if ((header
.content_size
/ CHAR_BIT
) < sizeof(header
)) {
148 PERR("Bad metadata packet content size: %u\n",
149 header
.content_size
);
153 toread
= header
.content_size
/ CHAR_BIT
- sizeof(header
);
156 readlen
= fread(buf
, sizeof(uint8_t),
157 MIN(sizeof(buf
) - 1, toread
), in_fp
);
159 PERR("Cannot read metadata packet buffer (at position %ld)\n",
164 writelen
= fwrite(buf
, sizeof(uint8_t), readlen
, out_fp
);
165 if (writelen
< readlen
|| ferror(out_fp
)) {
166 PERR("Cannot write decoded metadata text to buffer\n");
174 /* Read leftover padding */
175 toread
= (header
.packet_size
- header
.content_size
) /
177 fseek_ret
= fseek(in_fp
, toread
, SEEK_CUR
);
179 PWARN("Missing padding at the end of the metadata file\n");
196 int ctf_metadata_decoder_packetized_file_stream_to_buf_with_mdec(
197 struct ctf_metadata_decoder
*mdec
, FILE *fp
,
198 char **buf
, int byte_order
)
204 size_t packet_index
= 0;
206 out_fp
= bt_open_memstream(buf
, &size
);
207 if (out_fp
== NULL
) {
208 PERR("Cannot open memory stream: %s\n", strerror(errno
));
217 tret
= decode_packet(mdec
, fp
, out_fp
, byte_order
);
219 PERR("Cannot decode packet #%zu\n", packet_index
);
226 /* Make sure the whole string ends with a null character */
227 tret
= fputc('\0', out_fp
);
229 PERR("Cannot append '\\0' to the decoded metadata buffer\n");
233 /* Close stream, which also flushes the buffer */
234 ret
= bt_close_memstream(buf
, &size
, out_fp
);
236 PERR("Cannot close memory stream: %s\n", strerror(errno
));
246 bt_close_memstream(buf
, &size
, out_fp
);
258 int ctf_metadata_decoder_packetized_file_stream_to_buf(
259 FILE *fp
, char **buf
, int byte_order
)
261 return ctf_metadata_decoder_packetized_file_stream_to_buf_with_mdec(
262 NULL
, fp
, buf
, byte_order
);
266 struct ctf_metadata_decoder
*ctf_metadata_decoder_create(FILE *err
,
267 uint64_t clock_class_offset_ns
)
269 struct ctf_metadata_decoder
*mdec
=
270 g_new0(struct ctf_metadata_decoder
, 1);
276 mdec
->err_stream
= err
;
277 mdec
->visitor
= ctf_visitor_generate_ir_create(err
,
278 clock_class_offset_ns
);
279 if (!mdec
->visitor
) {
280 ctf_metadata_decoder_destroy(mdec
);
290 void ctf_metadata_decoder_destroy(struct ctf_metadata_decoder
*mdec
)
296 ctf_visitor_generate_ir_destroy(mdec
->visitor
);
301 enum ctf_metadata_decoder_status
ctf_metadata_decoder_decode(
302 struct ctf_metadata_decoder
*mdec
, FILE *fp
)
304 enum ctf_metadata_decoder_status status
=
305 CTF_METADATA_DECODER_STATUS_OK
;
307 struct ctf_scanner
*scanner
= NULL
;
309 bool close_fp
= false;
313 if (ctf_metadata_decoder_is_packetized(fp
, &mdec
->bo
)) {
314 PDBG("Metadata stream is packetized\n");
315 ret
= ctf_metadata_decoder_packetized_file_stream_to_buf_with_mdec(
316 mdec
, fp
, &buf
, mdec
->bo
);
319 status
= CTF_METADATA_DECODER_STATUS_ERROR
;
323 /* Convert the real file pointer to a memory file pointer */
324 fp
= bt_fmemopen(buf
, strlen(buf
), "rb");
327 PERR("Cannot memory-open metadata buffer: %s\n",
329 status
= CTF_METADATA_DECODER_STATUS_ERROR
;
333 unsigned int major
, minor
;
335 const long init_pos
= ftell(fp
);
337 PDBG("Metadata stream is plain text\n");
339 /* Check text-only metadata header and version */
340 nr_items
= fscanf(fp
, "/* CTF %10u.%10u", &major
, &minor
);
342 PWARN("Ill-shapen or missing \"/* CTF major.minor\" header in plain text metadata file stream\n");
345 PDBG("Metadata version: %u.%u\n", major
, minor
);
347 if (!is_version_valid(major
, minor
)) {
348 PERR("Invalid metadata version: %u.%u\n", major
, minor
);
349 status
= CTF_METADATA_DECODER_STATUS_INVAL_VERSION
;
353 if (fseek(fp
, init_pos
, SEEK_SET
)) {
354 PERR("Cannot seek metadata file stream to initial position: %s\n",
356 status
= CTF_METADATA_DECODER_STATUS_ERROR
;
361 /* Allocate a scanner and append the metadata text content */
362 scanner
= ctf_scanner_alloc();
364 PERR("Cannot allocate a metadata lexical scanner\n");
365 status
= CTF_METADATA_DECODER_STATUS_ERROR
;
370 ret
= ctf_scanner_append_ast(scanner
, fp
);
372 PERR("Cannot create the metadata AST\n");
373 status
= CTF_METADATA_DECODER_STATUS_INCOMPLETE
;
377 ret
= ctf_visitor_semantic_check(stderr
, 0, &scanner
->ast
->root
);
379 PERR("Metadata semantic validation failed\n");
380 status
= CTF_METADATA_DECODER_STATUS_ERROR
;
384 ret
= ctf_visitor_generate_ir_visit_node(mdec
->visitor
,
385 &scanner
->ast
->root
);
391 PDBG("While visiting AST: incomplete data\n");
392 status
= CTF_METADATA_DECODER_STATUS_INCOMPLETE
;
395 PERR("Cannot visit AST node to create CTF IR objects\n");
396 status
= CTF_METADATA_DECODER_STATUS_IR_VISITOR_ERROR
;
402 ctf_scanner_free(scanner
);
405 if (fp
&& close_fp
) {
407 PERR("Cannot close metadata file stream\n");
419 struct bt_ctf_trace
*ctf_metadata_decoder_get_trace(
420 struct ctf_metadata_decoder
*mdec
)
422 return ctf_visitor_generate_ir_get_trace(mdec
->visitor
);