Commit | Line | Data |
---|---|---|
273b65be JG |
1 | /* |
2 | * stream.c | |
3 | * | |
d2dc44b6 | 4 | * Babeltrace CTF IR - Stream |
273b65be | 5 | * |
de9dd397 | 6 | * Copyright 2013, 2014 Jérémie Galarneau <jeremie.galarneau@efficios.com> |
273b65be JG |
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 | ||
ac0c6bdd PP |
29 | #include <babeltrace/ctf-ir/clock-class.h> |
30 | #include <babeltrace/ctf-writer/clock.h> | |
31 | #include <babeltrace/ctf-writer/clock-internal.h> | |
273b65be | 32 | #include <babeltrace/ctf-writer/event.h> |
adc315b8 | 33 | #include <babeltrace/ctf-ir/event-internal.h> |
2e33ac5a PP |
34 | #include <babeltrace/ctf-ir/field-types-internal.h> |
35 | #include <babeltrace/ctf-ir/fields-internal.h> | |
3f043b05 JG |
36 | #include <babeltrace/ctf-ir/stream.h> |
37 | #include <babeltrace/ctf-ir/stream-internal.h> | |
adc315b8 | 38 | #include <babeltrace/ctf-ir/stream-class-internal.h> |
319fd969 PP |
39 | #include <babeltrace/ctf-ir/trace-internal.h> |
40 | #include <babeltrace/ctf-writer/writer-internal.h> | |
83509119 | 41 | #include <babeltrace/ref.h> |
273b65be | 42 | #include <babeltrace/ctf-writer/functor-internal.h> |
3d9990ac PP |
43 | #include <babeltrace/compiler-internal.h> |
44 | #include <babeltrace/align-internal.h> | |
dc3fffef | 45 | #include <inttypes.h> |
273b65be JG |
46 | |
47 | static | |
83509119 | 48 | void bt_ctf_stream_destroy(struct bt_object *obj); |
273b65be | 49 | static |
af9296f3 JG |
50 | int try_set_structure_field_integer(struct bt_ctf_field *, char *, uint64_t); |
51 | static | |
273b65be JG |
52 | int set_structure_field_integer(struct bt_ctf_field *, char *, uint64_t); |
53 | ||
ac0c6bdd PP |
54 | static |
55 | int set_integer_field_value(struct bt_ctf_field* field, uint64_t value) | |
56 | { | |
57 | int ret = 0; | |
58 | struct bt_ctf_field_type *field_type = NULL; | |
59 | ||
60 | if (!field) { | |
61 | ret = -1; | |
62 | goto end; | |
63 | } | |
64 | ||
65 | field_type = bt_ctf_field_get_type(field); | |
66 | assert(field_type); | |
67 | ||
68 | if (bt_ctf_field_type_get_type_id(field_type) != | |
1487a16a | 69 | BT_CTF_FIELD_TYPE_ID_INTEGER) { |
ac0c6bdd PP |
70 | /* Not an integer and the value is unset, error. */ |
71 | ret = -1; | |
72 | goto end; | |
73 | } | |
74 | ||
75 | if (bt_ctf_field_type_integer_get_signed(field_type)) { | |
76 | ret = bt_ctf_field_signed_integer_set_value(field, (int64_t) value); | |
77 | if (ret) { | |
78 | /* Value is out of range, error. */ | |
79 | goto end; | |
80 | } | |
81 | } else { | |
82 | ret = bt_ctf_field_unsigned_integer_set_value(field, value); | |
83 | if (ret) { | |
84 | /* Value is out of range, error. */ | |
85 | goto end; | |
86 | } | |
87 | } | |
88 | end: | |
89 | bt_put(field_type); | |
90 | return ret; | |
91 | } | |
92 | ||
d246b111 JG |
93 | static |
94 | int set_packet_header_magic(struct bt_ctf_stream *stream) | |
95 | { | |
96 | int ret = 0; | |
97 | struct bt_ctf_field_type *magic_field_type = NULL; | |
98 | struct bt_ctf_field *magic_field = bt_ctf_field_structure_get_field( | |
99 | stream->packet_header, "magic"); | |
100 | ||
101 | if (!magic_field) { | |
102 | /* No magic field found. Not an error, skip. */ | |
103 | goto end; | |
104 | } | |
105 | ||
fc47320a | 106 | if (bt_ctf_field_is_set(magic_field)) { |
d246b111 JG |
107 | /* Value already set. Not an error, skip. */ |
108 | goto end; | |
109 | } | |
110 | ||
111 | magic_field_type = bt_ctf_field_get_type(magic_field); | |
112 | assert(magic_field_type); | |
113 | ||
114 | if (bt_ctf_field_type_get_type_id(magic_field_type) != | |
1487a16a | 115 | BT_CTF_FIELD_TYPE_ID_INTEGER) { |
d246b111 JG |
116 | /* Magic field is not an integer. Not an error, skip. */ |
117 | goto end; | |
118 | } | |
119 | ||
120 | if (bt_ctf_field_type_integer_get_size(magic_field_type) != 32) { | |
121 | /* | |
122 | * Magic field is not of the expected size. | |
123 | * Not an error, skip. | |
124 | */ | |
125 | goto end; | |
126 | } | |
127 | ||
128 | ret = bt_ctf_field_type_integer_get_signed(magic_field_type); | |
129 | assert(ret >= 0); | |
130 | if (ret) { | |
131 | ret = bt_ctf_field_signed_integer_set_value(magic_field, | |
132 | (int64_t) 0xC1FC1FC1); | |
133 | } else { | |
134 | ret = bt_ctf_field_unsigned_integer_set_value(magic_field, | |
135 | (uint64_t) 0xC1FC1FC1); | |
136 | } | |
137 | end: | |
83509119 JG |
138 | bt_put(magic_field); |
139 | bt_put(magic_field_type); | |
d246b111 JG |
140 | return ret; |
141 | } | |
142 | ||
143 | static | |
144 | int set_packet_header_uuid(struct bt_ctf_stream *stream) | |
145 | { | |
146 | int i, ret = 0; | |
e6a8e8e4 | 147 | struct bt_ctf_trace *trace = NULL; |
d246b111 JG |
148 | struct bt_ctf_field_type *uuid_field_type = NULL; |
149 | struct bt_ctf_field_type *element_field_type = NULL; | |
150 | struct bt_ctf_field *uuid_field = bt_ctf_field_structure_get_field( | |
151 | stream->packet_header, "uuid"); | |
152 | ||
153 | if (!uuid_field) { | |
154 | /* No uuid field found. Not an error, skip. */ | |
155 | goto end; | |
156 | } | |
157 | ||
fc47320a | 158 | if (bt_ctf_field_is_set(uuid_field)) { |
d246b111 JG |
159 | /* Value already set. Not an error, skip. */ |
160 | goto end; | |
161 | } | |
162 | ||
163 | uuid_field_type = bt_ctf_field_get_type(uuid_field); | |
164 | assert(uuid_field_type); | |
165 | if (bt_ctf_field_type_get_type_id(uuid_field_type) != | |
1487a16a | 166 | BT_CTF_FIELD_TYPE_ID_ARRAY) { |
d246b111 JG |
167 | /* UUID field is not an array. Not an error, skip. */ |
168 | goto end; | |
169 | } | |
170 | ||
171 | if (bt_ctf_field_type_array_get_length(uuid_field_type) != 16) { | |
172 | /* | |
173 | * UUID field is not of the expected size. | |
174 | * Not an error, skip. | |
175 | */ | |
176 | goto end; | |
177 | } | |
178 | ||
179 | element_field_type = bt_ctf_field_type_array_get_element_type( | |
180 | uuid_field_type); | |
181 | assert(element_field_type); | |
182 | if (bt_ctf_field_type_get_type_id(element_field_type) != | |
1487a16a | 183 | BT_CTF_FIELD_TYPE_ID_INTEGER) { |
d246b111 JG |
184 | /* UUID array elements are not integers. Not an error, skip */ |
185 | goto end; | |
186 | } | |
187 | ||
e6a8e8e4 | 188 | trace = (struct bt_ctf_trace *) bt_object_get_parent(stream); |
d246b111 JG |
189 | for (i = 0; i < 16; i++) { |
190 | struct bt_ctf_field *uuid_element = | |
191 | bt_ctf_field_array_get_field(uuid_field, i); | |
192 | ||
193 | ret = bt_ctf_field_type_integer_get_signed(element_field_type); | |
194 | assert(ret >= 0); | |
195 | ||
196 | if (ret) { | |
197 | ret = bt_ctf_field_signed_integer_set_value( | |
e6a8e8e4 | 198 | uuid_element, (int64_t) trace->uuid[i]); |
d246b111 JG |
199 | } else { |
200 | ret = bt_ctf_field_unsigned_integer_set_value( | |
201 | uuid_element, | |
e6a8e8e4 | 202 | (uint64_t) trace->uuid[i]); |
d246b111 | 203 | } |
83509119 | 204 | bt_put(uuid_element); |
d246b111 JG |
205 | if (ret) { |
206 | goto end; | |
207 | } | |
208 | } | |
209 | ||
210 | end: | |
83509119 JG |
211 | bt_put(uuid_field); |
212 | bt_put(uuid_field_type); | |
213 | bt_put(element_field_type); | |
e6a8e8e4 | 214 | BT_PUT(trace); |
d246b111 JG |
215 | return ret; |
216 | } | |
217 | static | |
218 | int set_packet_header_stream_id(struct bt_ctf_stream *stream) | |
219 | { | |
220 | int ret = 0; | |
221 | uint32_t stream_id; | |
222 | struct bt_ctf_field_type *stream_id_field_type = NULL; | |
223 | struct bt_ctf_field *stream_id_field = bt_ctf_field_structure_get_field( | |
224 | stream->packet_header, "stream_id"); | |
225 | ||
226 | if (!stream_id_field) { | |
227 | /* No stream_id field found. Not an error, skip. */ | |
228 | goto end; | |
229 | } | |
230 | ||
fc47320a | 231 | if (bt_ctf_field_is_set(stream_id_field)) { |
d246b111 JG |
232 | /* Value already set. Not an error, skip. */ |
233 | goto end; | |
234 | } | |
235 | ||
236 | stream_id_field_type = bt_ctf_field_get_type(stream_id_field); | |
237 | assert(stream_id_field_type); | |
238 | if (bt_ctf_field_type_get_type_id(stream_id_field_type) != | |
1487a16a | 239 | BT_CTF_FIELD_TYPE_ID_INTEGER) { |
d246b111 JG |
240 | /* stream_id field is not an integer. Not an error, skip. */ |
241 | goto end; | |
242 | } | |
243 | ||
244 | stream_id = stream->stream_class->id; | |
245 | ret = bt_ctf_field_type_integer_get_signed(stream_id_field_type); | |
246 | assert(ret >= 0); | |
247 | if (ret) { | |
248 | ret = bt_ctf_field_signed_integer_set_value(stream_id_field, | |
249 | (int64_t) stream_id); | |
250 | } else { | |
251 | ret = bt_ctf_field_unsigned_integer_set_value(stream_id_field, | |
252 | (uint64_t) stream_id); | |
253 | } | |
254 | end: | |
83509119 JG |
255 | bt_put(stream_id_field); |
256 | bt_put(stream_id_field_type); | |
d246b111 JG |
257 | return ret; |
258 | } | |
259 | ||
260 | static | |
261 | int set_packet_header(struct bt_ctf_stream *stream) | |
262 | { | |
263 | int ret; | |
264 | ||
265 | ret = set_packet_header_magic(stream); | |
266 | if (ret) { | |
267 | goto end; | |
268 | } | |
269 | ||
270 | ret = set_packet_header_uuid(stream); | |
271 | if (ret) { | |
272 | goto end; | |
273 | } | |
274 | ||
275 | ret = set_packet_header_stream_id(stream); | |
276 | if (ret) { | |
277 | goto end; | |
278 | } | |
279 | end: | |
280 | return ret; | |
281 | } | |
282 | ||
123fbdec | 283 | static |
e6a8e8e4 | 284 | void release_event(struct bt_ctf_event *event) |
123fbdec | 285 | { |
e6a8e8e4 JG |
286 | if (bt_object_get_ref_count(event)) { |
287 | /* | |
288 | * The event is being orphaned, but it must guarantee the | |
289 | * existence of its event class for the duration of its | |
290 | * lifetime. | |
291 | */ | |
292 | bt_get(event->event_class); | |
293 | BT_PUT(event->base.parent); | |
294 | } else { | |
295 | bt_object_release(event); | |
296 | } | |
123fbdec JG |
297 | } |
298 | ||
319fd969 PP |
299 | static |
300 | int create_stream_file(struct bt_ctf_writer *writer, | |
301 | struct bt_ctf_stream *stream) | |
302 | { | |
303 | int fd; | |
304 | GString *filename = g_string_new(stream->stream_class->name->str); | |
305 | ||
306 | if (stream->stream_class->name->len == 0) { | |
307 | int64_t ret; | |
308 | ||
309 | ret = bt_ctf_stream_class_get_id(stream->stream_class); | |
310 | if (ret < 0) { | |
311 | fd = -1; | |
312 | goto error; | |
313 | } | |
314 | ||
315 | g_string_printf(filename, "stream_%" PRId64, ret); | |
316 | } | |
317 | ||
318 | g_string_append_printf(filename, "_%" PRIu32, stream->id); | |
319 | fd = openat(writer->trace_dir_fd, filename->str, | |
320 | O_RDWR | O_CREAT | O_TRUNC, | |
321 | S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP); | |
322 | error: | |
323 | g_string_free(filename, TRUE); | |
324 | return fd; | |
325 | } | |
326 | ||
327 | static | |
328 | int set_stream_fd(struct bt_ctf_stream *stream, int fd) | |
329 | { | |
330 | int ret = 0; | |
331 | ||
332 | if (stream->pos.fd != -1) { | |
333 | ret = -1; | |
334 | goto end; | |
335 | } | |
336 | ||
dc3fffef | 337 | (void) bt_ctf_stream_pos_init(&stream->pos, fd, O_RDWR); |
319fd969 PP |
338 | stream->pos.fd = fd; |
339 | end: | |
340 | return ret; | |
341 | } | |
342 | ||
273b65be | 343 | struct bt_ctf_stream *bt_ctf_stream_create( |
b71d7298 PP |
344 | struct bt_ctf_stream_class *stream_class, |
345 | const char *name) | |
273b65be | 346 | { |
12c8a1a3 | 347 | int ret; |
273b65be | 348 | struct bt_ctf_stream *stream = NULL; |
319fd969 PP |
349 | struct bt_ctf_trace *trace = NULL; |
350 | struct bt_ctf_writer *writer = NULL; | |
273b65be | 351 | |
319fd969 | 352 | if (!stream_class) { |
b71d7298 | 353 | goto error; |
319fd969 PP |
354 | } |
355 | ||
356 | trace = bt_ctf_stream_class_get_trace(stream_class); | |
357 | if (!trace) { | |
b71d7298 | 358 | goto error; |
273b65be JG |
359 | } |
360 | ||
361 | stream = g_new0(struct bt_ctf_stream, 1); | |
362 | if (!stream) { | |
b71d7298 | 363 | goto error; |
273b65be JG |
364 | } |
365 | ||
83509119 | 366 | bt_object_init(stream, bt_ctf_stream_destroy); |
e6a8e8e4 JG |
367 | /* |
368 | * Acquire reference to parent since stream will become publicly | |
369 | * reachable; it needs its parent to remain valid. | |
370 | */ | |
371 | bt_object_set_parent(stream, trace); | |
273b65be JG |
372 | stream->id = stream_class->next_stream_id++; |
373 | stream->stream_class = stream_class; | |
319fd969 | 374 | stream->pos.fd = -1; |
d246b111 | 375 | |
b71d7298 PP |
376 | if (name) { |
377 | stream->name = g_string_new(name); | |
378 | if (!stream->name) { | |
379 | goto error; | |
380 | } | |
381 | } | |
382 | ||
319fd969 PP |
383 | if (trace->is_created_by_writer) { |
384 | int fd; | |
385 | writer = (struct bt_ctf_writer *) | |
386 | bt_object_get_parent(trace); | |
387 | ||
388 | assert(writer); | |
98edd02c JG |
389 | if (stream_class->packet_context_type) { |
390 | stream->packet_context = bt_ctf_field_create( | |
391 | stream_class->packet_context_type); | |
392 | if (!stream->packet_context) { | |
393 | goto error; | |
394 | } | |
319fd969 | 395 | |
98edd02c | 396 | /* Initialize events_discarded */ |
af9296f3 JG |
397 | ret = try_set_structure_field_integer( |
398 | stream->packet_context, | |
98edd02c | 399 | "events_discarded", 0); |
af9296f3 | 400 | if (ret != 1) { |
98edd02c JG |
401 | goto error; |
402 | } | |
319fd969 PP |
403 | } |
404 | ||
405 | stream->events = g_ptr_array_new_with_free_func( | |
406 | (GDestroyNotify) release_event); | |
407 | if (!stream->events) { | |
408 | goto error; | |
409 | } | |
410 | ||
411 | /* A trace is not allowed to have a NULL packet header */ | |
412 | assert(trace->packet_header_type); | |
413 | stream->packet_header = | |
414 | bt_ctf_field_create(trace->packet_header_type); | |
415 | if (!stream->packet_header) { | |
416 | goto error; | |
417 | } | |
418 | ||
419 | /* | |
420 | * Attempt to populate the default trace packet header fields | |
421 | * (magic, uuid and stream_id). This will _not_ fail shall the | |
422 | * fields not be found or be of an incompatible type; they will | |
423 | * simply not be populated automatically. The user will have to | |
424 | * make sure to set the trace packet header fields himself | |
425 | * before flushing. | |
426 | */ | |
427 | ret = set_packet_header(stream); | |
428 | if (ret) { | |
429 | goto error; | |
430 | } | |
431 | ||
432 | /* Create file associated with this stream */ | |
433 | fd = create_stream_file(writer, stream); | |
434 | if (fd < 0) { | |
435 | goto error; | |
436 | } | |
437 | ||
438 | ret = set_stream_fd(stream, fd); | |
439 | if (ret) { | |
440 | goto error; | |
441 | } | |
442 | ||
443 | /* Freeze the writer */ | |
444 | bt_ctf_writer_freeze(writer); | |
445 | } else { | |
446 | /* Non-writer stream indicated by a negative FD */ | |
447 | ret = set_stream_fd(stream, -1); | |
448 | if (ret) { | |
449 | goto error; | |
450 | } | |
d246b111 | 451 | } |
319fd969 PP |
452 | |
453 | /* Add this stream to the trace's streams */ | |
454 | g_ptr_array_add(trace->streams, stream); | |
455 | ||
319fd969 PP |
456 | BT_PUT(trace); |
457 | BT_PUT(writer); | |
273b65be | 458 | return stream; |
83509119 JG |
459 | error: |
460 | BT_PUT(stream); | |
319fd969 PP |
461 | BT_PUT(trace); |
462 | BT_PUT(writer); | |
83509119 | 463 | return stream; |
273b65be JG |
464 | } |
465 | ||
3baf0856 JG |
466 | struct bt_ctf_stream_class *bt_ctf_stream_get_class( |
467 | struct bt_ctf_stream *stream) | |
468 | { | |
469 | struct bt_ctf_stream_class *stream_class = NULL; | |
470 | ||
471 | if (!stream) { | |
472 | goto end; | |
473 | } | |
474 | ||
475 | stream_class = stream->stream_class; | |
83509119 | 476 | bt_get(stream_class); |
3baf0856 JG |
477 | end: |
478 | return stream_class; | |
479 | } | |
480 | ||
a78a2e25 JG |
481 | int bt_ctf_stream_get_discarded_events_count( |
482 | struct bt_ctf_stream *stream, uint64_t *count) | |
483 | { | |
484 | int64_t ret = 0; | |
12c8a1a3 JG |
485 | int field_signed; |
486 | struct bt_ctf_field *events_discarded_field = NULL; | |
487 | struct bt_ctf_field_type *events_discarded_field_type = NULL; | |
a78a2e25 | 488 | |
1579cde5 PP |
489 | if (!stream || !count || !stream->packet_context || |
490 | stream->pos.fd < 0) { | |
a78a2e25 JG |
491 | ret = -1; |
492 | goto end; | |
493 | } | |
494 | ||
12c8a1a3 JG |
495 | events_discarded_field = bt_ctf_field_structure_get_field( |
496 | stream->packet_context, "events_discarded"); | |
497 | if (!events_discarded_field) { | |
498 | ret = -1; | |
499 | goto end; | |
500 | } | |
501 | ||
502 | events_discarded_field_type = bt_ctf_field_get_type( | |
503 | events_discarded_field); | |
504 | if (!events_discarded_field_type) { | |
505 | ret = -1; | |
506 | goto end; | |
507 | } | |
508 | ||
509 | field_signed = bt_ctf_field_type_integer_get_signed( | |
510 | events_discarded_field_type); | |
511 | if (field_signed < 0) { | |
512 | ret = field_signed; | |
513 | goto end; | |
514 | } | |
515 | ||
516 | if (field_signed) { | |
517 | int64_t signed_count; | |
518 | ||
519 | ret = bt_ctf_field_signed_integer_get_value( | |
520 | events_discarded_field, &signed_count); | |
521 | if (ret) { | |
522 | goto end; | |
523 | } | |
524 | if (signed_count < 0) { | |
525 | /* Invalid value */ | |
526 | ret = -1; | |
527 | goto end; | |
528 | } | |
529 | *count = (uint64_t) signed_count; | |
530 | } else { | |
531 | ret = bt_ctf_field_unsigned_integer_get_value( | |
532 | events_discarded_field, count); | |
533 | if (ret) { | |
534 | goto end; | |
535 | } | |
536 | } | |
a78a2e25 | 537 | end: |
83509119 JG |
538 | bt_put(events_discarded_field); |
539 | bt_put(events_discarded_field_type); | |
a78a2e25 JG |
540 | return ret; |
541 | } | |
542 | ||
273b65be JG |
543 | void bt_ctf_stream_append_discarded_events(struct bt_ctf_stream *stream, |
544 | uint64_t event_count) | |
545 | { | |
12c8a1a3 JG |
546 | int ret; |
547 | int field_signed; | |
548 | uint64_t previous_count; | |
549 | uint64_t new_count; | |
550 | struct bt_ctf_field *events_discarded_field = NULL; | |
551 | struct bt_ctf_field_type *events_discarded_field_type = NULL; | |
552 | ||
1579cde5 | 553 | if (!stream || !stream->packet_context || stream->pos.fd < 0) { |
12c8a1a3 JG |
554 | goto end; |
555 | } | |
556 | ||
557 | ret = bt_ctf_stream_get_discarded_events_count(stream, | |
558 | &previous_count); | |
559 | if (ret) { | |
560 | goto end; | |
561 | } | |
562 | ||
563 | events_discarded_field = bt_ctf_field_structure_get_field( | |
564 | stream->packet_context, "events_discarded"); | |
565 | if (!events_discarded_field) { | |
566 | goto end; | |
567 | } | |
568 | ||
569 | events_discarded_field_type = bt_ctf_field_get_type( | |
570 | events_discarded_field); | |
571 | if (!events_discarded_field_type) { | |
572 | goto end; | |
573 | } | |
574 | ||
575 | field_signed = bt_ctf_field_type_integer_get_signed( | |
576 | events_discarded_field_type); | |
577 | if (field_signed < 0) { | |
578 | goto end; | |
273b65be JG |
579 | } |
580 | ||
12c8a1a3 JG |
581 | new_count = previous_count + event_count; |
582 | if (field_signed) { | |
583 | ret = bt_ctf_field_signed_integer_set_value( | |
584 | events_discarded_field, (int64_t) new_count); | |
585 | if (ret) { | |
586 | goto end; | |
587 | } | |
588 | } else { | |
589 | ret = bt_ctf_field_unsigned_integer_set_value( | |
590 | events_discarded_field, new_count); | |
591 | if (ret) { | |
592 | goto end; | |
593 | } | |
594 | } | |
595 | ||
596 | end: | |
83509119 JG |
597 | bt_put(events_discarded_field); |
598 | bt_put(events_discarded_field_type); | |
273b65be JG |
599 | } |
600 | ||
ac0c6bdd PP |
601 | static int auto_populate_event_header(struct bt_ctf_stream *stream, |
602 | struct bt_ctf_event *event) | |
603 | { | |
604 | int ret = 0; | |
605 | struct bt_ctf_field *id_field = NULL, *timestamp_field = NULL; | |
606 | struct bt_ctf_clock_class *mapped_clock_class = NULL; | |
607 | ||
608 | if (!event || event->frozen) { | |
609 | ret = -1; | |
610 | goto end; | |
611 | } | |
612 | ||
613 | /* | |
614 | * The condition to automatically set the ID are: | |
615 | * | |
616 | * 1. The event header field "id" exists and is an integer | |
617 | * field. | |
618 | * 2. The event header field "id" is NOT set. | |
619 | */ | |
620 | id_field = bt_ctf_field_structure_get_field(event->event_header, "id"); | |
621 | if (id_field && !bt_ctf_field_is_set(id_field)) { | |
622 | ret = set_integer_field_value(id_field, | |
623 | (uint64_t) bt_ctf_event_class_get_id( | |
624 | event->event_class)); | |
625 | if (ret) { | |
626 | goto end; | |
627 | } | |
628 | } | |
629 | ||
630 | /* | |
631 | * The conditions to automatically set the timestamp are: | |
632 | * | |
633 | * 1. The event header field "timestamp" exists and is an | |
634 | * integer field. | |
635 | * 2. This stream's class has a registered clock (set with | |
636 | * bt_ctf_stream_class_set_clock()). | |
637 | * 3. The event header field "timestamp" has its type mapped to | |
638 | * a clock class which is also the clock class of this | |
639 | * stream's class's registered clock. | |
640 | * 4. The event header field "timestamp" is NOT set. | |
641 | */ | |
642 | timestamp_field = bt_ctf_field_structure_get_field(event->event_header, | |
643 | "timestamp"); | |
644 | if (timestamp_field && !bt_ctf_field_is_set(timestamp_field) && | |
645 | stream->stream_class->clock) { | |
646 | struct bt_ctf_clock_class *stream_class_clock_class = | |
647 | stream->stream_class->clock->clock_class; | |
648 | struct bt_ctf_field_type *timestamp_field_type = | |
649 | bt_ctf_field_get_type(timestamp_field); | |
650 | ||
651 | assert(timestamp_field_type); | |
652 | mapped_clock_class = | |
653 | bt_ctf_field_type_integer_get_mapped_clock_class( | |
654 | timestamp_field_type); | |
655 | BT_PUT(timestamp_field_type); | |
656 | if (mapped_clock_class == stream_class_clock_class) { | |
657 | uint64_t timestamp; | |
658 | ||
659 | ret = bt_ctf_clock_get_value( | |
660 | stream->stream_class->clock, | |
661 | ×tamp); | |
662 | if (ret) { | |
663 | goto end; | |
664 | } | |
665 | ||
666 | ret = set_integer_field_value(timestamp_field, | |
667 | timestamp); | |
668 | if (ret) { | |
669 | goto end; | |
670 | } | |
671 | } | |
672 | } | |
673 | ||
674 | end: | |
675 | bt_put(id_field); | |
676 | bt_put(timestamp_field); | |
677 | bt_put(mapped_clock_class); | |
678 | return ret; | |
679 | } | |
680 | ||
273b65be JG |
681 | int bt_ctf_stream_append_event(struct bt_ctf_stream *stream, |
682 | struct bt_ctf_event *event) | |
683 | { | |
684 | int ret = 0; | |
273b65be | 685 | |
1579cde5 | 686 | if (!stream || !event || stream->pos.fd < 0) { |
273b65be JG |
687 | ret = -1; |
688 | goto end; | |
689 | } | |
690 | ||
fa29ba83 PP |
691 | /* |
692 | * The event is not supposed to have a parent stream at this | |
693 | * point. The only other way an event can have a parent stream | |
694 | * is if it was assigned when setting a packet to the event, | |
695 | * in which case the packet's stream is not a writer stream, | |
696 | * and thus the user is trying to append an event which belongs | |
697 | * to another stream. | |
698 | */ | |
699 | if (event->base.parent) { | |
700 | ret = -1; | |
701 | goto end; | |
702 | } | |
703 | ||
e6a8e8e4 | 704 | bt_object_set_parent(event, stream); |
ac0c6bdd | 705 | ret = auto_populate_event_header(stream, event); |
662e778c | 706 | if (ret) { |
37f30168 | 707 | goto error; |
662e778c JG |
708 | } |
709 | ||
5fd2e9fd | 710 | /* Make sure the various scopes of the event are set */ |
273b65be JG |
711 | ret = bt_ctf_event_validate(event); |
712 | if (ret) { | |
37f30168 | 713 | goto error; |
273b65be JG |
714 | } |
715 | ||
0d688c15 PP |
716 | /* Save the new event and freeze it */ |
717 | bt_ctf_event_freeze(event); | |
273b65be | 718 | g_ptr_array_add(stream->events, event); |
5fd2e9fd | 719 | |
e6a8e8e4 JG |
720 | /* |
721 | * Event had to hold a reference to its event class as long as it wasn't | |
722 | * part of the same trace hierarchy. From now on, the event and its | |
723 | * class share the same lifetime guarantees and the reference is no | |
724 | * longer needed. | |
725 | */ | |
726 | bt_put(event->event_class); | |
37f30168 | 727 | |
273b65be | 728 | end: |
37f30168 PP |
729 | return ret; |
730 | ||
731 | error: | |
732 | /* | |
733 | * Orphan the event; we were not successful in associating it to | |
734 | * a stream. | |
735 | */ | |
736 | bt_object_set_parent(event, NULL); | |
737 | ||
273b65be JG |
738 | return ret; |
739 | } | |
740 | ||
12c8a1a3 JG |
741 | struct bt_ctf_field *bt_ctf_stream_get_packet_context( |
742 | struct bt_ctf_stream *stream) | |
743 | { | |
744 | struct bt_ctf_field *packet_context = NULL; | |
745 | ||
1579cde5 | 746 | if (!stream || stream->pos.fd < 0) { |
12c8a1a3 JG |
747 | goto end; |
748 | } | |
749 | ||
750 | packet_context = stream->packet_context; | |
12c8a1a3 | 751 | if (packet_context) { |
83509119 | 752 | bt_get(packet_context); |
12c8a1a3 | 753 | } |
34629a55 | 754 | end: |
12c8a1a3 JG |
755 | return packet_context; |
756 | } | |
757 | ||
758 | int bt_ctf_stream_set_packet_context(struct bt_ctf_stream *stream, | |
759 | struct bt_ctf_field *field) | |
760 | { | |
761 | int ret = 0; | |
762 | struct bt_ctf_field_type *field_type; | |
763 | ||
835b2d10 | 764 | if (!stream || stream->pos.fd < 0) { |
12c8a1a3 JG |
765 | ret = -1; |
766 | goto end; | |
767 | } | |
768 | ||
769 | field_type = bt_ctf_field_get_type(field); | |
09840de5 PP |
770 | if (bt_ctf_field_type_compare(field_type, |
771 | stream->stream_class->packet_context_type)) { | |
12c8a1a3 JG |
772 | ret = -1; |
773 | goto end; | |
774 | } | |
775 | ||
83509119 | 776 | bt_put(field_type); |
83509119 | 777 | bt_put(stream->packet_context); |
835b2d10 | 778 | stream->packet_context = bt_get(field); |
12c8a1a3 JG |
779 | end: |
780 | return ret; | |
781 | } | |
782 | ||
263a7df5 JG |
783 | struct bt_ctf_field *bt_ctf_stream_get_packet_header( |
784 | struct bt_ctf_stream *stream) | |
785 | { | |
786 | struct bt_ctf_field *packet_header = NULL; | |
787 | ||
1579cde5 | 788 | if (!stream || stream->pos.fd < 0) { |
263a7df5 JG |
789 | goto end; |
790 | } | |
791 | ||
792 | packet_header = stream->packet_header; | |
793 | if (packet_header) { | |
83509119 | 794 | bt_get(packet_header); |
263a7df5 JG |
795 | } |
796 | end: | |
797 | return packet_header; | |
798 | } | |
799 | ||
800 | int bt_ctf_stream_set_packet_header(struct bt_ctf_stream *stream, | |
801 | struct bt_ctf_field *field) | |
802 | { | |
803 | int ret = 0; | |
e6a8e8e4 | 804 | struct bt_ctf_trace *trace = NULL; |
263a7df5 JG |
805 | struct bt_ctf_field_type *field_type = NULL; |
806 | ||
835b2d10 | 807 | if (!stream || stream->pos.fd < 0) { |
263a7df5 JG |
808 | ret = -1; |
809 | goto end; | |
810 | } | |
811 | ||
e6a8e8e4 | 812 | trace = (struct bt_ctf_trace *) bt_object_get_parent(stream); |
263a7df5 | 813 | field_type = bt_ctf_field_get_type(field); |
09840de5 | 814 | if (bt_ctf_field_type_compare(field_type, trace->packet_header_type)) { |
263a7df5 JG |
815 | ret = -1; |
816 | goto end; | |
817 | } | |
818 | ||
83509119 | 819 | bt_put(stream->packet_header); |
835b2d10 | 820 | stream->packet_header = bt_get(field); |
263a7df5 | 821 | end: |
e6a8e8e4 | 822 | BT_PUT(trace); |
83509119 | 823 | bt_put(field_type); |
263a7df5 JG |
824 | return ret; |
825 | } | |
826 | ||
9a220c32 JG |
827 | static |
828 | int get_event_header_timestamp(struct bt_ctf_field *event_header, uint64_t *timestamp) | |
829 | { | |
830 | int ret = 0; | |
831 | struct bt_ctf_field *timestamp_field = NULL; | |
832 | struct bt_ctf_field_type *timestamp_field_type = NULL; | |
833 | ||
834 | timestamp_field = bt_ctf_field_structure_get_field(event_header, | |
835 | "timestamp"); | |
836 | if (!timestamp_field) { | |
837 | ret = -1; | |
838 | goto end; | |
839 | } | |
840 | ||
841 | timestamp_field_type = bt_ctf_field_get_type(timestamp_field); | |
842 | assert(timestamp_field_type); | |
843 | if (bt_ctf_field_type_get_type_id(timestamp_field_type) != | |
1487a16a | 844 | BT_CTF_FIELD_TYPE_ID_INTEGER) { |
9a220c32 JG |
845 | ret = -1; |
846 | goto end; | |
847 | } | |
848 | ||
849 | if (bt_ctf_field_type_integer_get_signed(timestamp_field_type)) { | |
850 | int64_t val; | |
851 | ||
852 | ret = bt_ctf_field_signed_integer_get_value(timestamp_field, | |
853 | &val); | |
854 | if (ret) { | |
855 | goto end; | |
856 | } | |
857 | *timestamp = (uint64_t) val; | |
858 | } else { | |
859 | ret = bt_ctf_field_unsigned_integer_get_value(timestamp_field, | |
860 | timestamp); | |
861 | if (ret) { | |
862 | goto end; | |
863 | } | |
864 | } | |
865 | end: | |
83509119 JG |
866 | bt_put(timestamp_field); |
867 | bt_put(timestamp_field_type); | |
9a220c32 JG |
868 | return ret; |
869 | } | |
870 | ||
c9af50d1 JG |
871 | static |
872 | void reset_structure_field(struct bt_ctf_field *structure, const char *name) | |
873 | { | |
874 | struct bt_ctf_field *member; | |
875 | ||
876 | member = bt_ctf_field_structure_get_field(structure, name); | |
877 | assert(member); | |
878 | (void) bt_ctf_field_reset(member); | |
e027ff8c | 879 | bt_put(member); |
c9af50d1 JG |
880 | } |
881 | ||
273b65be JG |
882 | int bt_ctf_stream_flush(struct bt_ctf_stream *stream) |
883 | { | |
884 | int ret = 0; | |
885 | size_t i; | |
c9af50d1 | 886 | uint64_t timestamp_begin, timestamp_end; |
273b65be | 887 | struct bt_ctf_field *integer = NULL; |
dc3fffef PP |
888 | struct bt_ctf_stream_pos packet_context_pos; |
889 | struct bt_ctf_trace *trace; | |
890 | enum bt_ctf_byte_order native_byte_order; | |
98edd02c | 891 | bool empty_packet; |
0686ef94 | 892 | uint64_t packet_size_bits; |
c9af50d1 JG |
893 | struct { |
894 | bool timestamp_begin; | |
895 | bool timestamp_end; | |
896 | bool content_size; | |
897 | bool packet_size; | |
898 | } auto_set_fields = { 0 }; | |
273b65be | 899 | |
bc37ae52 JG |
900 | if (!stream || stream->pos.fd < 0) { |
901 | /* | |
902 | * Stream does not have an associated fd. It is, | |
903 | * therefore, not a stream being used to write events. | |
904 | */ | |
273b65be JG |
905 | ret = -1; |
906 | goto end; | |
907 | } | |
908 | ||
664f50c8 JG |
909 | if (!stream->packet_context && stream->flushed_packet_count > 0) { |
910 | /* | |
911 | * A stream without a packet context, and thus without | |
912 | * content and packet size members, can't have more than | |
913 | * one packet. | |
914 | */ | |
915 | ret = -1; | |
916 | goto end; | |
917 | } | |
918 | ||
dc3fffef PP |
919 | trace = bt_ctf_stream_class_borrow_trace(stream->stream_class); |
920 | assert(trace); | |
921 | native_byte_order = bt_ctf_trace_get_byte_order(trace); | |
98edd02c | 922 | empty_packet = (stream->events->len == 0); |
d246b111 | 923 | |
28362f2b | 924 | /* mmap the next packet */ |
dc3fffef PP |
925 | bt_ctf_stream_pos_packet_seek(&stream->pos, 0, SEEK_CUR); |
926 | ret = bt_ctf_field_serialize(stream->packet_header, &stream->pos, | |
927 | native_byte_order); | |
d246b111 JG |
928 | if (ret) { |
929 | goto end; | |
273b65be JG |
930 | } |
931 | ||
98edd02c JG |
932 | if (stream->packet_context) { |
933 | /* Set the default context attributes if present and unset. */ | |
934 | if (!empty_packet && !get_event_header_timestamp( | |
935 | ((struct bt_ctf_event *) g_ptr_array_index( | |
936 | stream->events, 0))->event_header, ×tamp_begin)) { | |
af9296f3 JG |
937 | ret = try_set_structure_field_integer( |
938 | stream->packet_context, | |
98edd02c | 939 | "timestamp_begin", timestamp_begin); |
af9296f3 | 940 | if (ret < 0) { |
98edd02c JG |
941 | goto end; |
942 | } | |
c9af50d1 | 943 | auto_set_fields.timestamp_begin = ret == 1; |
98edd02c JG |
944 | } |
945 | ||
946 | if (!empty_packet && !get_event_header_timestamp( | |
947 | ((struct bt_ctf_event *) g_ptr_array_index( | |
948 | stream->events, stream->events->len - 1))->event_header, | |
949 | ×tamp_end)) { | |
950 | ||
af9296f3 JG |
951 | ret = try_set_structure_field_integer( |
952 | stream->packet_context, | |
98edd02c | 953 | "timestamp_end", timestamp_end); |
af9296f3 | 954 | if (ret < 0) { |
98edd02c JG |
955 | goto end; |
956 | } | |
c9af50d1 | 957 | auto_set_fields.timestamp_end = ret == 1; |
98edd02c | 958 | } |
af9296f3 | 959 | ret = try_set_structure_field_integer(stream->packet_context, |
98edd02c | 960 | "content_size", UINT64_MAX); |
af9296f3 | 961 | if (ret < 0) { |
9a220c32 JG |
962 | goto end; |
963 | } | |
c9af50d1 | 964 | auto_set_fields.content_size = ret == 1; |
273b65be | 965 | |
af9296f3 | 966 | ret = try_set_structure_field_integer(stream->packet_context, |
98edd02c | 967 | "packet_size", UINT64_MAX); |
af9296f3 | 968 | if (ret < 0) { |
98edd02c JG |
969 | goto end; |
970 | } | |
c9af50d1 | 971 | auto_set_fields.packet_size = ret == 1; |
12c8a1a3 | 972 | |
10cb7b41 JG |
973 | /* Write packet context */ |
974 | memcpy(&packet_context_pos, &stream->pos, | |
dc3fffef | 975 | sizeof(packet_context_pos)); |
10cb7b41 | 976 | ret = bt_ctf_field_serialize(stream->packet_context, |
dc3fffef | 977 | &stream->pos, native_byte_order); |
98edd02c JG |
978 | if (ret) { |
979 | goto end; | |
980 | } | |
12c8a1a3 JG |
981 | } |
982 | ||
273b65be JG |
983 | for (i = 0; i < stream->events->len; i++) { |
984 | struct bt_ctf_event *event = g_ptr_array_index( | |
985 | stream->events, i); | |
273b65be | 986 | |
273b65be | 987 | /* Write event header */ |
662e778c | 988 | ret = bt_ctf_field_serialize(event->event_header, |
dc3fffef | 989 | &stream->pos, native_byte_order); |
273b65be JG |
990 | if (ret) { |
991 | goto end; | |
992 | } | |
993 | ||
8bfa3f9c | 994 | /* Write stream event context */ |
5fd2e9fd | 995 | if (event->stream_event_context) { |
8bfa3f9c | 996 | ret = bt_ctf_field_serialize( |
dc3fffef PP |
997 | event->stream_event_context, &stream->pos, |
998 | native_byte_order); | |
8bfa3f9c JG |
999 | if (ret) { |
1000 | goto end; | |
1001 | } | |
1002 | } | |
1003 | ||
273b65be | 1004 | /* Write event content */ |
dc3fffef PP |
1005 | ret = bt_ctf_event_serialize(event, &stream->pos, |
1006 | native_byte_order); | |
273b65be JG |
1007 | if (ret) { |
1008 | goto end; | |
1009 | } | |
1010 | } | |
1011 | ||
0686ef94 JG |
1012 | /* Rounded-up in case content_size is not byte-aligned. */ |
1013 | packet_size_bits = (stream->pos.offset + (CHAR_BIT - 1)) & | |
1014 | ~(CHAR_BIT - 1); | |
1015 | stream->pos.packet_size = packet_size_bits; | |
1016 | ||
98edd02c JG |
1017 | if (stream->packet_context) { |
1018 | /* | |
1019 | * Update the packet total size and content size and overwrite | |
1020 | * the packet context. | |
1021 | * Copy base_mma as the packet may have been remapped (e.g. when | |
1022 | * a packet is resized). | |
1023 | */ | |
1024 | packet_context_pos.base_mma = stream->pos.base_mma; | |
c9af50d1 JG |
1025 | if (auto_set_fields.content_size) { |
1026 | ret = set_structure_field_integer( | |
1027 | stream->packet_context, | |
1028 | "content_size", stream->pos.offset); | |
1029 | if (ret < 0) { | |
1030 | goto end; | |
1031 | } | |
98edd02c | 1032 | } |
273b65be | 1033 | |
c9af50d1 JG |
1034 | if (auto_set_fields.packet_size) { |
1035 | ret = set_structure_field_integer(stream->packet_context, | |
1036 | "packet_size", packet_size_bits); | |
1037 | if (ret < 0) { | |
1038 | goto end; | |
1039 | } | |
98edd02c | 1040 | } |
273b65be | 1041 | |
98edd02c | 1042 | ret = bt_ctf_field_serialize(stream->packet_context, |
dc3fffef | 1043 | &packet_context_pos, native_byte_order); |
98edd02c JG |
1044 | if (ret) { |
1045 | goto end; | |
1046 | } | |
273b65be JG |
1047 | } |
1048 | ||
1049 | g_ptr_array_set_size(stream->events, 0); | |
1050 | stream->flushed_packet_count++; | |
0686ef94 | 1051 | stream->size += packet_size_bits / CHAR_BIT; |
273b65be | 1052 | end: |
c9af50d1 JG |
1053 | /* Reset automatically-set fields. */ |
1054 | if (auto_set_fields.timestamp_begin) { | |
1055 | reset_structure_field(stream->packet_context, | |
1056 | "timestamp_begin"); | |
1057 | } | |
1058 | if (auto_set_fields.timestamp_end) { | |
1059 | reset_structure_field(stream->packet_context, | |
1060 | "timestamp_end"); | |
1061 | } | |
1062 | if (auto_set_fields.packet_size) { | |
1063 | reset_structure_field(stream->packet_context, | |
1064 | "packet_size"); | |
1065 | } | |
1066 | if (auto_set_fields.content_size) { | |
1067 | reset_structure_field(stream->packet_context, | |
1068 | "content_size"); | |
1069 | } | |
83509119 | 1070 | bt_put(integer); |
cd7d8fb7 JG |
1071 | |
1072 | if (ret < 0) { | |
1073 | /* | |
1074 | * We failed to write the packet. Its size is therefore set to 0 | |
1075 | * to ensure the next mapping is done in the same place rather | |
1076 | * than advancing by "stream->pos.packet_size", which would | |
1077 | * leave a corrupted packet in the trace. | |
1078 | */ | |
1079 | stream->pos.packet_size = 0; | |
1080 | } | |
273b65be JG |
1081 | return ret; |
1082 | } | |
1083 | ||
1084 | void bt_ctf_stream_get(struct bt_ctf_stream *stream) | |
1085 | { | |
83509119 | 1086 | bt_get(stream); |
273b65be JG |
1087 | } |
1088 | ||
1089 | void bt_ctf_stream_put(struct bt_ctf_stream *stream) | |
1090 | { | |
83509119 | 1091 | bt_put(stream); |
273b65be JG |
1092 | } |
1093 | ||
1094 | static | |
83509119 | 1095 | void bt_ctf_stream_destroy(struct bt_object *obj) |
273b65be JG |
1096 | { |
1097 | struct bt_ctf_stream *stream; | |
273b65be | 1098 | |
83509119 | 1099 | stream = container_of(obj, struct bt_ctf_stream, base); |
dc3fffef | 1100 | (void) bt_ctf_stream_pos_fini(&stream->pos); |
0686ef94 JG |
1101 | if (stream->pos.fd >= 0) { |
1102 | int ret; | |
1103 | ||
1104 | /* | |
1105 | * Truncate the file's size to the minimum required to fit the | |
1106 | * last packet as we might have grown it too much on the last | |
1107 | * mmap. | |
1108 | */ | |
1109 | do { | |
1110 | ret = ftruncate(stream->pos.fd, stream->size); | |
1111 | } while (ret == -1 && errno == EINTR); | |
1112 | if (ret) { | |
1113 | perror("ftruncate"); | |
1114 | } | |
1115 | ||
1116 | if (close(stream->pos.fd)) { | |
1117 | perror("close"); | |
1118 | } | |
9f56e450 | 1119 | } |
12c8a1a3 | 1120 | |
12c8a1a3 JG |
1121 | if (stream->events) { |
1122 | g_ptr_array_free(stream->events, TRUE); | |
1123 | } | |
b71d7298 PP |
1124 | |
1125 | if (stream->name) { | |
1126 | g_string_free(stream->name, TRUE); | |
1127 | } | |
41ac640a | 1128 | |
83509119 JG |
1129 | bt_put(stream->packet_header); |
1130 | bt_put(stream->packet_context); | |
273b65be JG |
1131 | g_free(stream); |
1132 | } | |
1133 | ||
273b65be | 1134 | static |
af9296f3 JG |
1135 | int _set_structure_field_integer(struct bt_ctf_field *structure, char *name, |
1136 | uint64_t value, bool force) | |
273b65be JG |
1137 | { |
1138 | int ret = 0; | |
d7b1ea66 | 1139 | struct bt_ctf_field_type *field_type = NULL; |
273b65be JG |
1140 | struct bt_ctf_field *integer = |
1141 | bt_ctf_field_structure_get_field(structure, name); | |
12c8a1a3 JG |
1142 | |
1143 | if (!structure || !name) { | |
273b65be JG |
1144 | ret = -1; |
1145 | goto end; | |
1146 | } | |
1147 | ||
12c8a1a3 JG |
1148 | if (!integer) { |
1149 | /* Field not found, not an error. */ | |
1150 | goto end; | |
1151 | } | |
1152 | ||
1153 | /* Make sure the payload has not already been set. */ | |
af9296f3 | 1154 | if (!force && bt_ctf_field_is_set(integer)) { |
12c8a1a3 JG |
1155 | /* Payload already set, not an error */ |
1156 | goto end; | |
1157 | } | |
1158 | ||
d7b1ea66 | 1159 | field_type = bt_ctf_field_get_type(integer); |
d7b1ea66 | 1160 | assert(field_type); |
1487a16a | 1161 | if (bt_ctf_field_type_get_type_id(field_type) != BT_CTF_FIELD_TYPE_ID_INTEGER) { |
d7b1ea66 JG |
1162 | /* |
1163 | * The user most likely meant for us to populate this field | |
1164 | * automatically. However, we can only do this if the field | |
1165 | * is an integer. Return an error. | |
1166 | */ | |
1167 | ret = -1; | |
1168 | goto end; | |
1169 | } | |
1170 | ||
1171 | if (bt_ctf_field_type_integer_get_signed(field_type)) { | |
1172 | ret = bt_ctf_field_signed_integer_set_value(integer, | |
1173 | (int64_t) value); | |
1174 | } else { | |
1175 | ret = bt_ctf_field_unsigned_integer_set_value(integer, value); | |
1176 | } | |
af9296f3 | 1177 | ret = !ret ? 1 : ret; |
273b65be | 1178 | end: |
83509119 JG |
1179 | bt_put(integer); |
1180 | bt_put(field_type); | |
273b65be JG |
1181 | return ret; |
1182 | } | |
b71d7298 | 1183 | |
af9296f3 JG |
1184 | static |
1185 | int set_structure_field_integer(struct bt_ctf_field *structure, char *name, | |
1186 | uint64_t value) | |
1187 | { | |
1188 | return _set_structure_field_integer(structure, name, value, true); | |
1189 | } | |
1190 | ||
1191 | /* | |
1192 | * Returns the following codes: | |
1193 | * 1 if the field was found and set, | |
1194 | * 0 if nothing was done (field not found, or was already set), | |
1195 | * <0 if an error was encoutered | |
1196 | */ | |
1197 | static | |
1198 | int try_set_structure_field_integer(struct bt_ctf_field *structure, char *name, | |
1199 | uint64_t value) | |
1200 | { | |
1201 | return _set_structure_field_integer(structure, name, value, false); | |
1202 | } | |
1203 | ||
b71d7298 PP |
1204 | const char *bt_ctf_stream_get_name(struct bt_ctf_stream *stream) |
1205 | { | |
1206 | const char *name = NULL; | |
1207 | ||
1208 | if (!stream) { | |
1209 | goto end; | |
1210 | } | |
1211 | ||
1212 | name = stream->name ? stream->name->str : NULL; | |
1213 | ||
1214 | end: | |
1215 | return name; | |
1216 | } | |
98a4cbef PP |
1217 | |
1218 | int bt_ctf_stream_is_writer(struct bt_ctf_stream *stream) | |
1219 | { | |
1220 | int ret = -1; | |
1221 | ||
1222 | if (!stream) { | |
1223 | goto end; | |
1224 | } | |
1225 | ||
1226 | ret = (stream->pos.fd >= 0); | |
1227 | ||
1228 | end: | |
1229 | return ret; | |
1230 | } |