Commit | Line | Data |
---|---|---|
bc37ae52 JG |
1 | /* |
2 | * trace.c | |
3 | * | |
4 | * Babeltrace CTF IR - Trace | |
5 | * | |
6 | * Copyright 2014 Jérémie Galarneau <jeremie.galarneau@efficios.com> | |
7 | * | |
8 | * Author: Jérémie Galarneau <jeremie.galarneau@efficios.com> | |
9 | * | |
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: | |
16 | * | |
17 | * The above copyright notice and this permission notice shall be included in | |
18 | * all copies or substantial portions of the Software. | |
19 | * | |
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 | |
26 | * SOFTWARE. | |
27 | */ | |
28 | ||
29 | #include <babeltrace/ctf-ir/trace-internal.h> | |
30 | #include <babeltrace/ctf-ir/clock-internal.h> | |
31 | #include <babeltrace/ctf-ir/stream-internal.h> | |
32 | #include <babeltrace/ctf-ir/stream-class-internal.h> | |
33 | #include <babeltrace/ctf-writer/functor-internal.h> | |
34 | #include <babeltrace/ctf-ir/event-types-internal.h> | |
44e0a4f5 | 35 | #include <babeltrace/ctf-ir/attributes-internal.h> |
654c1444 | 36 | #include <babeltrace/ctf-ir/utils.h> |
bc37ae52 | 37 | #include <babeltrace/compiler.h> |
7f800dc7 | 38 | #include <babeltrace/objects.h> |
bc37ae52 JG |
39 | |
40 | #define DEFAULT_IDENTIFIER_SIZE 128 | |
41 | #define DEFAULT_METADATA_STRING_SIZE 4096 | |
42 | ||
bc37ae52 JG |
43 | static |
44 | void bt_ctf_trace_destroy(struct bt_ctf_ref *ref); | |
45 | static | |
46 | int init_trace_packet_header(struct bt_ctf_trace *trace); | |
f67f3a6e JG |
47 | static |
48 | void bt_ctf_trace_freeze(struct bt_ctf_trace *trace); | |
bc37ae52 | 49 | |
bc37ae52 JG |
50 | static |
51 | const unsigned int field_type_aliases_alignments[] = { | |
52 | [FIELD_TYPE_ALIAS_UINT5_T] = 1, | |
53 | [FIELD_TYPE_ALIAS_UINT8_T ... FIELD_TYPE_ALIAS_UINT16_T] = 8, | |
54 | [FIELD_TYPE_ALIAS_UINT27_T] = 1, | |
55 | [FIELD_TYPE_ALIAS_UINT32_T ... FIELD_TYPE_ALIAS_UINT64_T] = 8, | |
56 | }; | |
57 | ||
58 | static | |
59 | const unsigned int field_type_aliases_sizes[] = { | |
60 | [FIELD_TYPE_ALIAS_UINT5_T] = 5, | |
61 | [FIELD_TYPE_ALIAS_UINT8_T] = 8, | |
62 | [FIELD_TYPE_ALIAS_UINT16_T] = 16, | |
63 | [FIELD_TYPE_ALIAS_UINT27_T] = 27, | |
64 | [FIELD_TYPE_ALIAS_UINT32_T] = 32, | |
65 | [FIELD_TYPE_ALIAS_UINT64_T] = 64, | |
66 | }; | |
67 | ||
d3814b54 JG |
68 | static |
69 | void put_stream_class(struct bt_ctf_stream_class *stream_class) | |
70 | { | |
71 | (void) bt_ctf_stream_class_set_trace(stream_class, NULL); | |
72 | bt_ctf_stream_class_put(stream_class); | |
73 | } | |
74 | ||
bc37ae52 JG |
75 | struct bt_ctf_trace *bt_ctf_trace_create(void) |
76 | { | |
77 | struct bt_ctf_trace *trace = NULL; | |
78 | ||
79 | trace = g_new0(struct bt_ctf_trace, 1); | |
80 | if (!trace) { | |
81 | goto error; | |
82 | } | |
83 | ||
84 | bt_ctf_trace_set_byte_order(trace, BT_CTF_BYTE_ORDER_NATIVE); | |
85 | bt_ctf_ref_init(&trace->ref_count); | |
bc37ae52 | 86 | trace->clocks = g_ptr_array_new_with_free_func( |
d3814b54 | 87 | (GDestroyNotify) bt_ctf_clock_put); |
bc37ae52 | 88 | trace->streams = g_ptr_array_new_with_free_func( |
d3814b54 | 89 | (GDestroyNotify) bt_ctf_stream_put); |
bc37ae52 | 90 | trace->stream_classes = g_ptr_array_new_with_free_func( |
d3814b54 | 91 | (GDestroyNotify) put_stream_class); |
7f800dc7 | 92 | if (!trace->clocks || !trace->stream_classes || !trace->streams) { |
bc37ae52 JG |
93 | goto error_destroy; |
94 | } | |
95 | ||
96 | /* Generate a trace UUID */ | |
97 | uuid_generate(trace->uuid); | |
98 | if (init_trace_packet_header(trace)) { | |
99 | goto error_destroy; | |
100 | } | |
101 | ||
7f800dc7 PP |
102 | /* Create the environment array object */ |
103 | trace->environment = bt_ctf_attributes_create(); | |
104 | if (!trace->environment) { | |
105 | goto error_destroy; | |
106 | } | |
107 | ||
bc37ae52 JG |
108 | return trace; |
109 | ||
110 | error_destroy: | |
111 | bt_ctf_trace_destroy(&trace->ref_count); | |
112 | trace = NULL; | |
113 | error: | |
114 | return trace; | |
115 | } | |
116 | ||
117 | void bt_ctf_trace_destroy(struct bt_ctf_ref *ref) | |
118 | { | |
119 | struct bt_ctf_trace *trace; | |
120 | ||
121 | if (!ref) { | |
122 | return; | |
123 | } | |
124 | ||
125 | trace = container_of(ref, struct bt_ctf_trace, ref_count); | |
126 | if (trace->environment) { | |
7f800dc7 | 127 | bt_ctf_attributes_destroy(trace->environment); |
bc37ae52 JG |
128 | } |
129 | ||
130 | if (trace->clocks) { | |
131 | g_ptr_array_free(trace->clocks, TRUE); | |
132 | } | |
133 | ||
134 | if (trace->streams) { | |
135 | g_ptr_array_free(trace->streams, TRUE); | |
136 | } | |
137 | ||
138 | if (trace->stream_classes) { | |
139 | g_ptr_array_free(trace->stream_classes, TRUE); | |
140 | } | |
141 | ||
d246b111 | 142 | bt_ctf_field_type_put(trace->packet_header_type); |
bc37ae52 JG |
143 | g_free(trace); |
144 | } | |
145 | ||
146 | struct bt_ctf_stream *bt_ctf_trace_create_stream(struct bt_ctf_trace *trace, | |
147 | struct bt_ctf_stream_class *stream_class) | |
148 | { | |
149 | int ret; | |
150 | int stream_class_found = 0; | |
151 | size_t i; | |
152 | struct bt_ctf_stream *stream = NULL; | |
153 | ||
154 | if (!trace || !stream_class) { | |
155 | goto error; | |
156 | } | |
157 | ||
d246b111 | 158 | stream = bt_ctf_stream_create(stream_class, trace); |
bc37ae52 JG |
159 | if (!stream) { |
160 | goto error; | |
161 | } | |
162 | ||
163 | for (i = 0; i < trace->stream_classes->len; i++) { | |
164 | if (trace->stream_classes->pdata[i] == stream_class) { | |
165 | stream_class_found = 1; | |
166 | } | |
167 | } | |
168 | ||
169 | if (!stream_class_found) { | |
ef0c4a15 JG |
170 | ret = bt_ctf_trace_add_stream_class(trace, stream_class); |
171 | if (ret) { | |
172 | goto error; | |
bc37ae52 | 173 | } |
bc37ae52 JG |
174 | } |
175 | ||
176 | bt_ctf_stream_get(stream); | |
177 | g_ptr_array_add(trace->streams, stream); | |
c35a1669 | 178 | |
bc37ae52 | 179 | return stream; |
bc37ae52 JG |
180 | error: |
181 | bt_ctf_stream_put(stream); | |
182 | return NULL; | |
183 | } | |
184 | ||
7f800dc7 PP |
185 | int bt_ctf_trace_set_environment_field(struct bt_ctf_trace *trace, |
186 | const char *name, struct bt_object *value) | |
bc37ae52 | 187 | { |
bc37ae52 JG |
188 | int ret = 0; |
189 | ||
a0d12916 | 190 | if (!trace || !name || !value || |
7f800dc7 PP |
191 | bt_ctf_validate_identifier(name) || |
192 | !(bt_object_is_integer(value) || bt_object_is_string(value))) { | |
bc37ae52 | 193 | ret = -1; |
7f800dc7 | 194 | goto end; |
bc37ae52 JG |
195 | } |
196 | ||
197 | if (strchr(name, ' ')) { | |
198 | ret = -1; | |
7f800dc7 | 199 | goto end; |
bc37ae52 JG |
200 | } |
201 | ||
a0d12916 JG |
202 | if (trace->frozen) { |
203 | /* | |
204 | * New environment fields may be added to a frozen trace, | |
205 | * but existing fields may not be changed. | |
206 | * | |
207 | * The object passed is frozen like all other attributes. | |
208 | */ | |
209 | struct bt_object *attribute = | |
210 | bt_ctf_attributes_get_field_value_by_name( | |
211 | trace->environment, name); | |
212 | ||
213 | if (attribute) { | |
214 | BT_OBJECT_PUT(attribute); | |
215 | ret = -1; | |
216 | goto end; | |
217 | } | |
218 | ||
219 | bt_object_freeze(value); | |
220 | } | |
221 | ||
7f800dc7 PP |
222 | ret = bt_ctf_attributes_set_field_value(trace->environment, name, |
223 | value); | |
bc37ae52 | 224 | |
7f800dc7 PP |
225 | end: |
226 | return ret; | |
227 | } | |
bc37ae52 | 228 | |
7f800dc7 PP |
229 | int bt_ctf_trace_set_environment_field_string(struct bt_ctf_trace *trace, |
230 | const char *name, const char *value) | |
231 | { | |
232 | int ret = 0; | |
233 | struct bt_object *env_value_string_obj = NULL; | |
234 | ||
235 | if (!trace || !name || !value) { | |
bc37ae52 | 236 | ret = -1; |
7f800dc7 | 237 | goto end; |
bc37ae52 JG |
238 | } |
239 | ||
a0d12916 JG |
240 | if (trace->frozen) { |
241 | /* | |
242 | * New environment fields may be added to a frozen trace, | |
243 | * but existing fields may not be changed. | |
244 | */ | |
245 | struct bt_object *attribute = | |
246 | bt_ctf_attributes_get_field_value_by_name( | |
247 | trace->environment, name); | |
248 | ||
249 | if (attribute) { | |
250 | BT_OBJECT_PUT(attribute); | |
251 | ret = -1; | |
252 | goto end; | |
253 | } | |
254 | } | |
255 | ||
7f800dc7 | 256 | env_value_string_obj = bt_object_string_create_init(value); |
bc37ae52 | 257 | |
7f800dc7 PP |
258 | if (!env_value_string_obj) { |
259 | ret = -1; | |
260 | goto end; | |
bc37ae52 JG |
261 | } |
262 | ||
a0d12916 JG |
263 | if (trace->frozen) { |
264 | bt_object_freeze(env_value_string_obj); | |
265 | } | |
7f800dc7 PP |
266 | ret = bt_ctf_trace_set_environment_field(trace, name, |
267 | env_value_string_obj); | |
268 | ||
269 | end: | |
270 | BT_OBJECT_PUT(env_value_string_obj); | |
3487c9f3 | 271 | |
3487c9f3 JG |
272 | return ret; |
273 | } | |
274 | ||
7f800dc7 PP |
275 | int bt_ctf_trace_set_environment_field_integer(struct bt_ctf_trace *trace, |
276 | const char *name, int64_t value) | |
3487c9f3 | 277 | { |
3487c9f3 | 278 | int ret = 0; |
7f800dc7 | 279 | struct bt_object *env_value_integer_obj = NULL; |
3487c9f3 JG |
280 | |
281 | if (!trace || !name) { | |
282 | ret = -1; | |
7f800dc7 | 283 | goto end; |
3487c9f3 JG |
284 | } |
285 | ||
a0d12916 JG |
286 | if (trace->frozen) { |
287 | /* | |
288 | * New environment fields may be added to a frozen trace, | |
289 | * but existing fields may not be changed. | |
290 | */ | |
291 | struct bt_object *attribute = | |
292 | bt_ctf_attributes_get_field_value_by_name( | |
293 | trace->environment, name); | |
294 | ||
295 | if (attribute) { | |
296 | BT_OBJECT_PUT(attribute); | |
297 | ret = -1; | |
298 | goto end; | |
299 | } | |
300 | } | |
301 | ||
7f800dc7 PP |
302 | env_value_integer_obj = bt_object_integer_create_init(value); |
303 | if (!env_value_integer_obj) { | |
3487c9f3 | 304 | ret = -1; |
7f800dc7 | 305 | goto end; |
3487c9f3 JG |
306 | } |
307 | ||
7f800dc7 PP |
308 | ret = bt_ctf_trace_set_environment_field(trace, name, |
309 | env_value_integer_obj); | |
a0d12916 JG |
310 | if (trace->frozen) { |
311 | bt_object_freeze(env_value_integer_obj); | |
312 | } | |
7f800dc7 PP |
313 | end: |
314 | BT_OBJECT_PUT(env_value_integer_obj); | |
bc37ae52 | 315 | |
bc37ae52 JG |
316 | return ret; |
317 | } | |
318 | ||
e6fa2160 JG |
319 | int bt_ctf_trace_get_environment_field_count(struct bt_ctf_trace *trace) |
320 | { | |
321 | int ret = 0; | |
322 | ||
323 | if (!trace) { | |
324 | ret = -1; | |
325 | goto end; | |
326 | } | |
327 | ||
7f800dc7 | 328 | ret = bt_ctf_attributes_get_count(trace->environment); |
e6fa2160 | 329 | |
e6fa2160 | 330 | end: |
7f800dc7 | 331 | return ret; |
e6fa2160 JG |
332 | } |
333 | ||
334 | const char * | |
335 | bt_ctf_trace_get_environment_field_name(struct bt_ctf_trace *trace, | |
336 | int index) | |
337 | { | |
e6fa2160 JG |
338 | const char *ret = NULL; |
339 | ||
7f800dc7 | 340 | if (!trace) { |
e6fa2160 JG |
341 | goto end; |
342 | } | |
343 | ||
7f800dc7 PP |
344 | ret = bt_ctf_attributes_get_field_name(trace->environment, index); |
345 | ||
e6fa2160 JG |
346 | end: |
347 | return ret; | |
348 | } | |
349 | ||
7f800dc7 PP |
350 | struct bt_object *bt_ctf_trace_get_environment_field_value( |
351 | struct bt_ctf_trace *trace, int index) | |
e6fa2160 | 352 | { |
7f800dc7 | 353 | struct bt_object *ret = NULL; |
e6fa2160 | 354 | |
7f800dc7 | 355 | if (!trace) { |
e6fa2160 JG |
356 | goto end; |
357 | } | |
358 | ||
7f800dc7 PP |
359 | ret = bt_ctf_attributes_get_field_value(trace->environment, index); |
360 | ||
e6fa2160 JG |
361 | end: |
362 | return ret; | |
363 | } | |
364 | ||
7f800dc7 PP |
365 | struct bt_object *bt_ctf_trace_get_environment_field_value_by_name( |
366 | struct bt_ctf_trace *trace, const char *name) | |
e6fa2160 | 367 | { |
7f800dc7 | 368 | struct bt_object *ret = NULL; |
e6fa2160 | 369 | |
7f800dc7 | 370 | if (!trace || !name) { |
e6fa2160 JG |
371 | goto end; |
372 | } | |
373 | ||
7f800dc7 PP |
374 | ret = bt_ctf_attributes_get_field_value_by_name(trace->environment, |
375 | name); | |
376 | ||
e6fa2160 JG |
377 | end: |
378 | return ret; | |
379 | } | |
380 | ||
bc37ae52 JG |
381 | int bt_ctf_trace_add_clock(struct bt_ctf_trace *trace, |
382 | struct bt_ctf_clock *clock) | |
383 | { | |
384 | int ret = 0; | |
385 | struct search_query query = { .value = clock, .found = 0 }; | |
386 | ||
387 | if (!trace || !clock) { | |
388 | ret = -1; | |
389 | goto end; | |
390 | } | |
391 | ||
392 | /* Check for duplicate clocks */ | |
393 | g_ptr_array_foreach(trace->clocks, value_exists, &query); | |
394 | if (query.found) { | |
395 | ret = -1; | |
396 | goto end; | |
397 | } | |
398 | ||
399 | bt_ctf_clock_get(clock); | |
400 | g_ptr_array_add(trace->clocks, clock); | |
401 | end: | |
402 | return ret; | |
403 | } | |
404 | ||
884cd6c3 JG |
405 | int bt_ctf_trace_get_clock_count(struct bt_ctf_trace *trace) |
406 | { | |
407 | int ret = -1; | |
408 | ||
409 | if (!trace) { | |
410 | goto end; | |
411 | } | |
412 | ||
413 | ret = trace->clocks->len; | |
414 | end: | |
415 | return ret; | |
416 | } | |
417 | ||
418 | struct bt_ctf_clock *bt_ctf_trace_get_clock(struct bt_ctf_trace *trace, | |
419 | int index) | |
420 | { | |
421 | struct bt_ctf_clock *clock = NULL; | |
422 | ||
423 | if (!trace || index < 0 || index >= trace->clocks->len) { | |
424 | goto end; | |
425 | } | |
426 | ||
427 | clock = g_ptr_array_index(trace->clocks, index); | |
428 | bt_ctf_clock_get(clock); | |
429 | end: | |
430 | return clock; | |
431 | } | |
432 | ||
ef0c4a15 JG |
433 | int bt_ctf_trace_add_stream_class(struct bt_ctf_trace *trace, |
434 | struct bt_ctf_stream_class *stream_class) | |
435 | { | |
436 | int ret, i; | |
437 | int64_t stream_id; | |
438 | ||
439 | if (!trace || !stream_class) { | |
440 | ret = -1; | |
441 | goto end; | |
442 | } | |
443 | ||
444 | for (i = 0; i < trace->stream_classes->len; i++) { | |
445 | if (trace->stream_classes->pdata[i] == stream_class) { | |
446 | ret = -1; | |
447 | goto end; | |
448 | } | |
449 | } | |
450 | ||
451 | stream_id = bt_ctf_stream_class_get_id(stream_class); | |
452 | if (stream_id < 0) { | |
453 | stream_id = trace->next_stream_id++; | |
454 | ||
455 | /* Try to assign a new stream id */ | |
456 | for (i = 0; i < trace->stream_classes->len; i++) { | |
457 | if (stream_id == bt_ctf_stream_class_get_id( | |
458 | trace->stream_classes->pdata[i])) { | |
459 | /* Duplicate stream id found */ | |
460 | ret = -1; | |
461 | goto end; | |
462 | } | |
463 | } | |
464 | ||
465 | if (_bt_ctf_stream_class_set_id(stream_class, | |
de866173 | 466 | stream_id)) { |
ef0c4a15 JG |
467 | /* TODO Should retry with a different stream id */ |
468 | ret = -1; | |
469 | goto end; | |
470 | } | |
471 | } | |
472 | ||
d3814b54 JG |
473 | /* Set weak reference to trace in stream class */ |
474 | ret = bt_ctf_stream_class_set_trace(stream_class, trace); | |
475 | if (ret) { | |
476 | /* Stream class already part of another trace */ | |
477 | goto end; | |
478 | } | |
479 | ||
ef0c4a15 JG |
480 | bt_ctf_stream_class_get(stream_class); |
481 | g_ptr_array_add(trace->stream_classes, stream_class); | |
482 | ||
483 | /* | |
484 | * Freeze the trace and its packet header. | |
485 | * | |
486 | * All field type byte orders set as "native" byte ordering can now be | |
487 | * safely set to trace's own endianness, including the stream class'. | |
488 | */ | |
489 | bt_ctf_field_type_set_native_byte_order(trace->packet_header_type, | |
490 | trace->byte_order); | |
491 | ret = bt_ctf_stream_class_set_byte_order(stream_class, | |
492 | trace->byte_order == LITTLE_ENDIAN ? | |
493 | BT_CTF_BYTE_ORDER_LITTLE_ENDIAN : BT_CTF_BYTE_ORDER_BIG_ENDIAN); | |
494 | if (ret) { | |
495 | goto end; | |
496 | } | |
ef0c4a15 | 497 | |
f67f3a6e JG |
498 | bt_ctf_stream_class_freeze(stream_class); |
499 | bt_ctf_trace_freeze(trace); | |
ef0c4a15 | 500 | end: |
d3814b54 JG |
501 | if (ret) { |
502 | (void) bt_ctf_stream_class_set_trace(stream_class, NULL); | |
503 | } | |
ef0c4a15 JG |
504 | return ret; |
505 | } | |
506 | ||
507 | int bt_ctf_trace_get_stream_class_count(struct bt_ctf_trace *trace) | |
508 | { | |
509 | int ret; | |
510 | ||
511 | if (!trace) { | |
512 | ret = -1; | |
513 | goto end; | |
514 | } | |
515 | ||
516 | ret = trace->stream_classes->len; | |
517 | end: | |
518 | return ret; | |
519 | } | |
520 | ||
521 | struct bt_ctf_stream_class *bt_ctf_trace_get_stream_class( | |
522 | struct bt_ctf_trace *trace, int index) | |
523 | { | |
524 | struct bt_ctf_stream_class *stream_class = NULL; | |
525 | ||
526 | if (!trace || index < 0 || index >= trace->stream_classes->len) { | |
527 | goto end; | |
528 | } | |
529 | ||
530 | stream_class = g_ptr_array_index(trace->stream_classes, index); | |
531 | bt_ctf_stream_class_get(stream_class); | |
532 | end: | |
533 | return stream_class; | |
534 | } | |
535 | ||
7e4347a3 PP |
536 | struct bt_ctf_clock *bt_ctf_trace_get_clock_by_name( |
537 | struct bt_ctf_trace *trace, const char *name) | |
538 | { | |
539 | size_t i; | |
540 | struct bt_ctf_clock *clock = NULL; | |
541 | ||
542 | if (!trace || !name) { | |
543 | goto end; | |
544 | } | |
545 | ||
546 | for (i = 0; i < trace->clocks->len; ++i) { | |
547 | struct bt_ctf_clock *cur_clk = | |
548 | g_ptr_array_index(trace->clocks, i); | |
549 | const char *cur_clk_name = bt_ctf_clock_get_name(cur_clk); | |
550 | ||
551 | if (!cur_clk_name) { | |
552 | goto end; | |
553 | } | |
554 | ||
555 | if (!strcmp(cur_clk_name, name)) { | |
556 | clock = cur_clk; | |
557 | bt_ctf_clock_get(clock); | |
558 | goto end; | |
559 | } | |
560 | } | |
561 | ||
562 | end: | |
563 | return clock; | |
564 | } | |
565 | ||
bc37ae52 JG |
566 | BT_HIDDEN |
567 | const char *get_byte_order_string(int byte_order) | |
568 | { | |
569 | const char *string; | |
570 | ||
571 | switch (byte_order) { | |
572 | case LITTLE_ENDIAN: | |
573 | string = "le"; | |
574 | break; | |
575 | case BIG_ENDIAN: | |
576 | string = "be"; | |
577 | break; | |
578 | default: | |
579 | string = "unknown"; | |
580 | break; | |
581 | } | |
582 | ||
583 | return string; | |
584 | } | |
585 | ||
586 | static | |
587 | int append_trace_metadata(struct bt_ctf_trace *trace, | |
588 | struct metadata_context *context) | |
589 | { | |
590 | unsigned char *uuid = trace->uuid; | |
591 | int ret; | |
592 | ||
593 | g_string_append(context->string, "trace {\n"); | |
594 | ||
595 | g_string_append(context->string, "\tmajor = 1;\n"); | |
596 | g_string_append(context->string, "\tminor = 8;\n"); | |
597 | ||
598 | g_string_append_printf(context->string, | |
599 | "\tuuid = \"%02x%02x%02x%02x-%02x%02x-%02x%02x-%02x%02x-%02x%02x%02x%02x%02x%02x\";\n", | |
600 | uuid[0], uuid[1], uuid[2], uuid[3], | |
601 | uuid[4], uuid[5], uuid[6], uuid[7], | |
602 | uuid[8], uuid[9], uuid[10], uuid[11], | |
603 | uuid[12], uuid[13], uuid[14], uuid[15]); | |
604 | g_string_append_printf(context->string, "\tbyte_order = %s;\n", | |
605 | get_byte_order_string(trace->byte_order)); | |
606 | ||
607 | g_string_append(context->string, "\tpacket.header := "); | |
608 | context->current_indentation_level++; | |
609 | g_string_assign(context->field_name, ""); | |
d246b111 | 610 | ret = bt_ctf_field_type_serialize(trace->packet_header_type, |
bc37ae52 JG |
611 | context); |
612 | if (ret) { | |
613 | goto end; | |
614 | } | |
615 | context->current_indentation_level--; | |
616 | ||
617 | g_string_append(context->string, ";\n};\n\n"); | |
618 | end: | |
619 | return ret; | |
620 | } | |
621 | ||
bc37ae52 JG |
622 | static |
623 | void append_env_metadata(struct bt_ctf_trace *trace, | |
624 | struct metadata_context *context) | |
625 | { | |
7f800dc7 PP |
626 | int i; |
627 | int env_size; | |
628 | ||
629 | env_size = bt_ctf_attributes_get_count(trace->environment); | |
630 | ||
631 | if (env_size <= 0) { | |
bc37ae52 JG |
632 | return; |
633 | } | |
634 | ||
635 | g_string_append(context->string, "env {\n"); | |
7f800dc7 PP |
636 | |
637 | for (i = 0; i < env_size; ++i) { | |
638 | struct bt_object *env_field_value_obj = NULL; | |
639 | const char *entry_name; | |
7f800dc7 PP |
640 | |
641 | entry_name = bt_ctf_attributes_get_field_name( | |
642 | trace->environment, i); | |
643 | env_field_value_obj = bt_ctf_attributes_get_field_value( | |
644 | trace->environment, i); | |
645 | ||
646 | if (!entry_name || !env_field_value_obj) { | |
647 | goto loop_next; | |
648 | } | |
649 | ||
650 | switch (bt_object_get_type(env_field_value_obj)) { | |
651 | case BT_OBJECT_TYPE_INTEGER: | |
b8248cc0 PP |
652 | { |
653 | int ret; | |
654 | int64_t int_value; | |
655 | ||
7f800dc7 PP |
656 | ret = bt_object_integer_get(env_field_value_obj, |
657 | &int_value); | |
658 | ||
659 | if (ret) { | |
660 | goto loop_next; | |
661 | } | |
662 | ||
663 | g_string_append_printf(context->string, | |
664 | "\t%s = %" PRId64 ";\n", entry_name, | |
665 | int_value); | |
666 | break; | |
b8248cc0 | 667 | } |
7f800dc7 PP |
668 | case BT_OBJECT_TYPE_STRING: |
669 | { | |
670 | int ret; | |
671 | const char *str_value; | |
672 | char *escaped_str = NULL; | |
673 | ||
674 | ret = bt_object_string_get(env_field_value_obj, | |
675 | &str_value); | |
676 | ||
677 | if (ret) { | |
678 | goto loop_next; | |
679 | } | |
680 | ||
681 | escaped_str = g_strescape(str_value, NULL); | |
682 | ||
683 | if (!escaped_str) { | |
684 | goto loop_next; | |
685 | } | |
686 | ||
687 | g_string_append_printf(context->string, | |
688 | "\t%s = \"%s\";\n", entry_name, escaped_str); | |
689 | free(escaped_str); | |
690 | break; | |
691 | } | |
692 | ||
693 | default: | |
694 | goto loop_next; | |
695 | } | |
696 | ||
697 | loop_next: | |
698 | BT_OBJECT_PUT(env_field_value_obj); | |
699 | } | |
700 | ||
bc37ae52 JG |
701 | g_string_append(context->string, "};\n\n"); |
702 | } | |
703 | ||
704 | char *bt_ctf_trace_get_metadata_string(struct bt_ctf_trace *trace) | |
705 | { | |
706 | char *metadata = NULL; | |
707 | struct metadata_context *context = NULL; | |
708 | int err = 0; | |
709 | size_t i; | |
710 | ||
711 | if (!trace) { | |
712 | goto end; | |
713 | } | |
714 | ||
715 | context = g_new0(struct metadata_context, 1); | |
716 | if (!context) { | |
717 | goto end; | |
718 | } | |
719 | ||
720 | context->field_name = g_string_sized_new(DEFAULT_IDENTIFIER_SIZE); | |
721 | context->string = g_string_sized_new(DEFAULT_METADATA_STRING_SIZE); | |
722 | g_string_append(context->string, "/* CTF 1.8 */\n\n"); | |
723 | if (append_trace_metadata(trace, context)) { | |
724 | goto error; | |
725 | } | |
726 | append_env_metadata(trace, context); | |
727 | g_ptr_array_foreach(trace->clocks, | |
728 | (GFunc)bt_ctf_clock_serialize, context); | |
729 | ||
730 | for (i = 0; i < trace->stream_classes->len; i++) { | |
731 | err = bt_ctf_stream_class_serialize( | |
732 | trace->stream_classes->pdata[i], context); | |
733 | if (err) { | |
734 | goto error; | |
735 | } | |
736 | } | |
737 | ||
738 | metadata = context->string->str; | |
739 | error: | |
740 | g_string_free(context->string, err ? TRUE : FALSE); | |
741 | g_string_free(context->field_name, TRUE); | |
742 | g_free(context); | |
743 | end: | |
744 | return metadata; | |
745 | } | |
746 | ||
4ed90fb3 JG |
747 | enum bt_ctf_byte_order bt_ctf_trace_get_byte_order(struct bt_ctf_trace *trace) |
748 | { | |
749 | enum bt_ctf_byte_order ret = BT_CTF_BYTE_ORDER_UNKNOWN; | |
750 | ||
751 | if (!trace) { | |
752 | goto end; | |
753 | } | |
754 | ||
755 | switch (trace->byte_order) { | |
756 | case BIG_ENDIAN: | |
757 | ret = BT_CTF_BYTE_ORDER_BIG_ENDIAN; | |
758 | break; | |
759 | case LITTLE_ENDIAN: | |
760 | ret = BT_CTF_BYTE_ORDER_LITTLE_ENDIAN; | |
761 | break; | |
762 | default: | |
763 | break; | |
764 | } | |
765 | end: | |
766 | return ret; | |
767 | } | |
768 | ||
bc37ae52 JG |
769 | int bt_ctf_trace_set_byte_order(struct bt_ctf_trace *trace, |
770 | enum bt_ctf_byte_order byte_order) | |
771 | { | |
772 | int ret = 0; | |
773 | int internal_byte_order; | |
774 | ||
775 | if (!trace || trace->frozen) { | |
776 | ret = -1; | |
777 | goto end; | |
778 | } | |
779 | ||
780 | switch (byte_order) { | |
781 | case BT_CTF_BYTE_ORDER_NATIVE: | |
c35a1669 JG |
782 | /* |
783 | * This doesn't make sense since the CTF specification defines | |
784 | * the "native" byte order as "the byte order described in the | |
785 | * trace description". However, this behavior had been | |
786 | * implemented as part of v1.2 and is kept to maintain | |
787 | * compatibility. | |
788 | * | |
789 | * This may be changed on a major version bump only. | |
790 | */ | |
791 | internal_byte_order = (G_BYTE_ORDER == G_LITTLE_ENDIAN) ? | |
bc37ae52 JG |
792 | LITTLE_ENDIAN : BIG_ENDIAN; |
793 | break; | |
794 | case BT_CTF_BYTE_ORDER_LITTLE_ENDIAN: | |
795 | internal_byte_order = LITTLE_ENDIAN; | |
796 | break; | |
797 | case BT_CTF_BYTE_ORDER_BIG_ENDIAN: | |
798 | case BT_CTF_BYTE_ORDER_NETWORK: | |
799 | internal_byte_order = BIG_ENDIAN; | |
800 | break; | |
801 | default: | |
802 | ret = -1; | |
803 | goto end; | |
804 | } | |
805 | ||
806 | trace->byte_order = internal_byte_order; | |
bc37ae52 JG |
807 | end: |
808 | return ret; | |
809 | } | |
810 | ||
d246b111 JG |
811 | struct bt_ctf_field_type *bt_ctf_trace_get_packet_header_type( |
812 | struct bt_ctf_trace *trace) | |
813 | { | |
814 | struct bt_ctf_field_type *field_type = NULL; | |
815 | ||
816 | if (!trace) { | |
817 | goto end; | |
818 | } | |
819 | ||
820 | bt_ctf_field_type_get(trace->packet_header_type); | |
821 | field_type = trace->packet_header_type; | |
822 | end: | |
823 | return field_type; | |
824 | } | |
825 | ||
826 | int bt_ctf_trace_set_packet_header_type(struct bt_ctf_trace *trace, | |
827 | struct bt_ctf_field_type *packet_header_type) | |
828 | { | |
829 | int ret = 0; | |
830 | ||
831 | if (!trace || !packet_header_type || trace->frozen) { | |
832 | ret = -1; | |
833 | goto end; | |
834 | } | |
835 | ||
836 | /* packet_header_type must be a structure */ | |
837 | if (bt_ctf_field_type_get_type_id(packet_header_type) != | |
838 | CTF_TYPE_STRUCT) { | |
839 | ret = -1; | |
840 | goto end; | |
841 | } | |
842 | ||
843 | bt_ctf_field_type_get(packet_header_type); | |
844 | bt_ctf_field_type_put(trace->packet_header_type); | |
845 | trace->packet_header_type = packet_header_type; | |
846 | end: | |
847 | return ret; | |
848 | } | |
849 | ||
bc37ae52 JG |
850 | void bt_ctf_trace_get(struct bt_ctf_trace *trace) |
851 | { | |
852 | if (!trace) { | |
853 | return; | |
854 | } | |
855 | ||
856 | bt_ctf_ref_get(&trace->ref_count); | |
857 | } | |
858 | ||
859 | void bt_ctf_trace_put(struct bt_ctf_trace *trace) | |
860 | { | |
861 | if (!trace) { | |
862 | return; | |
863 | } | |
864 | ||
865 | bt_ctf_ref_put(&trace->ref_count, bt_ctf_trace_destroy); | |
866 | } | |
867 | ||
bc37ae52 JG |
868 | BT_HIDDEN |
869 | struct bt_ctf_field_type *get_field_type(enum field_type_alias alias) | |
870 | { | |
871 | unsigned int alignment, size; | |
872 | struct bt_ctf_field_type *field_type; | |
873 | ||
874 | if (alias >= NR_FIELD_TYPE_ALIAS) { | |
875 | return NULL; | |
876 | } | |
877 | ||
878 | alignment = field_type_aliases_alignments[alias]; | |
879 | size = field_type_aliases_sizes[alias]; | |
880 | field_type = bt_ctf_field_type_integer_create(size); | |
881 | bt_ctf_field_type_set_alignment(field_type, alignment); | |
882 | return field_type; | |
883 | } | |
884 | ||
f67f3a6e JG |
885 | static |
886 | void bt_ctf_trace_freeze(struct bt_ctf_trace *trace) | |
887 | { | |
888 | bt_ctf_attributes_freeze(trace->environment); | |
889 | trace->frozen = 1; | |
890 | } | |
891 | ||
bc37ae52 JG |
892 | static |
893 | int init_trace_packet_header(struct bt_ctf_trace *trace) | |
894 | { | |
bc37ae52 | 895 | int ret = 0; |
d246b111 | 896 | struct bt_ctf_field *magic = NULL, *uuid_array = NULL; |
bc37ae52 JG |
897 | struct bt_ctf_field_type *_uint32_t = |
898 | get_field_type(FIELD_TYPE_ALIAS_UINT32_T); | |
899 | struct bt_ctf_field_type *_uint8_t = | |
900 | get_field_type(FIELD_TYPE_ALIAS_UINT8_T); | |
901 | struct bt_ctf_field_type *trace_packet_header_type = | |
902 | bt_ctf_field_type_structure_create(); | |
903 | struct bt_ctf_field_type *uuid_array_type = | |
904 | bt_ctf_field_type_array_create(_uint8_t, 16); | |
905 | ||
906 | if (!trace_packet_header_type || !uuid_array_type) { | |
907 | ret = -1; | |
908 | goto end; | |
909 | } | |
910 | ||
bc37ae52 JG |
911 | ret = bt_ctf_field_type_structure_add_field(trace_packet_header_type, |
912 | _uint32_t, "magic"); | |
913 | if (ret) { | |
914 | goto end; | |
915 | } | |
916 | ||
917 | ret = bt_ctf_field_type_structure_add_field(trace_packet_header_type, | |
918 | uuid_array_type, "uuid"); | |
919 | if (ret) { | |
920 | goto end; | |
921 | } | |
922 | ||
923 | ret = bt_ctf_field_type_structure_add_field(trace_packet_header_type, | |
924 | _uint32_t, "stream_id"); | |
925 | if (ret) { | |
926 | goto end; | |
927 | } | |
928 | ||
662e778c JG |
929 | ret = bt_ctf_trace_set_packet_header_type(trace, |
930 | trace_packet_header_type); | |
931 | if (ret) { | |
932 | goto end; | |
933 | } | |
bc37ae52 JG |
934 | end: |
935 | bt_ctf_field_type_put(uuid_array_type); | |
936 | bt_ctf_field_type_put(_uint32_t); | |
937 | bt_ctf_field_type_put(_uint8_t); | |
938 | bt_ctf_field_put(magic); | |
939 | bt_ctf_field_put(uuid_array); | |
662e778c | 940 | bt_ctf_field_type_put(trace_packet_header_type); |
bc37ae52 JG |
941 | |
942 | return ret; | |
943 | } |