fs-sink: fix check trace completed when static is set
[babeltrace.git] / plugins / ctf / fs-sink / write.c
CommitLineData
bc506aa5
JD
1/*
2 * writer.c
3 *
4 * Babeltrace CTF Writer Output Plugin Event Handling
5 *
6 * Copyright 2016 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/event.h>
30#include <babeltrace/ctf-ir/packet.h>
31#include <babeltrace/ctf-ir/event-class.h>
32#include <babeltrace/ctf-ir/stream.h>
33#include <babeltrace/ctf-ir/stream-class.h>
ac0c6bdd 34#include <babeltrace/ctf-ir/clock-class.h>
bc506aa5
JD
35#include <babeltrace/ctf-ir/fields.h>
36#include <babeltrace/ctf-writer/stream-class.h>
37#include <babeltrace/ctf-writer/stream.h>
f3168545 38#include <assert.h>
bc506aa5 39
91b73004
JD
40#include <ctfcopytrace.h>
41
bc506aa5
JD
42#include "writer.h"
43
a619fcb7
JD
44static
45void unref_stream_class(struct bt_ctf_stream_class *writer_stream_class)
46{
47 bt_put(writer_stream_class);
48}
49
50static
51void unref_stream(struct bt_ctf_stream_class *writer_stream)
52{
53 bt_put(writer_stream);
54}
55
56gboolean empty_ht(gpointer key, gpointer value, gpointer user_data)
57{
58 return TRUE;
59}
60
cdcf612f
JD
61void destroy_stream_state_key(gpointer key)
62{
63 g_free((enum fs_writer_stream_state *) key);
64}
65
6f25f7c4
JD
66void check_completed_trace(gpointer key, gpointer value, gpointer user_data)
67{
68 enum fs_writer_stream_state *state = value;
69 int *trace_completed = user_data;
70
71 if (*state != FS_WRITER_COMPLETED_STREAM) {
72 *trace_completed = 0;
73 }
74}
75
f3168545
JD
76static
77void trace_is_static_listener(struct bt_ctf_trace *trace, void *data)
78{
6f25f7c4
JD
79 struct fs_writer *fs_writer = data;
80 int trace_completed = 1;
81
82 fs_writer->trace_static = 1;
83
84 g_hash_table_foreach(fs_writer->stream_states,
85 check_completed_trace, &trace_completed);
86 if (trace_completed) {
87 writer_close(fs_writer->writer_component, fs_writer);
88 g_hash_table_remove(fs_writer->writer_component->trace_map,
89 fs_writer->trace);
90 }
f3168545
JD
91}
92
bc506aa5 93static
91b73004
JD
94struct bt_ctf_stream_class *insert_new_stream_class(
95 struct writer_component *writer_component,
a619fcb7 96 struct fs_writer *fs_writer,
91b73004 97 struct bt_ctf_stream_class *stream_class)
bc506aa5 98{
b2f1f465 99 struct bt_ctf_stream_class *writer_stream_class = NULL;
9ae49d3d 100 struct bt_ctf_trace *trace = NULL, *writer_trace = NULL;
a619fcb7 101 struct bt_ctf_writer *ctf_writer = fs_writer->writer;
3241bc18 102 enum bt_component_status ret;
bc506aa5 103
3241bc18
JD
104 trace = bt_ctf_stream_class_get_trace(stream_class);
105 if (!trace) {
91b73004
JD
106 fprintf(writer_component->err,
107 "[error] %s in %s:%d\n", __func__, __FILE__,
bc506aa5 108 __LINE__);
9ae49d3d 109 goto error;
bc506aa5
JD
110 }
111
91b73004
JD
112 writer_trace = bt_ctf_writer_get_trace(ctf_writer);
113 if (!writer_trace) {
91b73004
JD
114 fprintf(writer_component->err,
115 "[error] %s in %s:%d\n", __func__, __FILE__,
d00b90bf 116 __LINE__);
9ae49d3d 117 goto error;
bc506aa5
JD
118 }
119
91b73004
JD
120 ret = ctf_copy_clock_classes(writer_component->err, writer_trace,
121 writer_stream_class, trace);
bc506aa5 122 if (ret != BT_COMPONENT_STATUS_OK) {
91b73004
JD
123 fprintf(writer_component->err,
124 "[error] %s in %s:%d\n", __func__, __FILE__,
bc506aa5 125 __LINE__);
9ae49d3d 126 goto error;
b2f1f465
JD
127 }
128
129 writer_stream_class = ctf_copy_stream_class(writer_component->err,
130 stream_class, writer_trace, true);
131 if (!writer_stream_class) {
132 fprintf(writer_component->err, "[error] Failed to copy stream class\n");
133 fprintf(writer_component->err, "[error] %s in %s:%d\n",
134 __func__, __FILE__, __LINE__);
9ae49d3d 135 goto error;
bc506aa5 136 }
bc506aa5 137
a619fcb7 138 g_hash_table_insert(fs_writer->stream_class_map,
bc506aa5
JD
139 (gpointer) stream_class, writer_stream_class);
140
9ae49d3d
JD
141 goto end;
142
143error:
144 BT_PUT(writer_stream_class);
145end:
b2f1f465 146 bt_put(writer_trace);
b2f1f465 147 bt_put(trace);
bc506aa5
JD
148 return writer_stream_class;
149}
150
2881e5b5
JD
151static
152enum fs_writer_stream_state *insert_new_stream_state(
153 struct writer_component *writer_component,
154 struct fs_writer *fs_writer, struct bt_ctf_stream *stream)
155{
156 enum fs_writer_stream_state *v = NULL;
157
158 v = g_new0(enum fs_writer_stream_state, 1);
159 if (!v) {
160 fprintf(writer_component->err,
161 "[error] %s in %s:%d\n", __func__,
162 __FILE__, __LINE__);
163 }
164 *v = FS_WRITER_UNKNOWN_STREAM;
165
166 g_hash_table_insert(fs_writer->stream_states, stream, v);
167
168 return v;
169}
170
9b4f54fb
JD
171static
172int make_trace_path(struct writer_component *writer_component,
173 struct bt_ctf_trace *trace, char *trace_path)
174{
175 int ret;
176 const char *trace_name;
177
178 trace_name = bt_ctf_trace_get_name(trace);
179 if (!trace_name) {
180 trace_name = writer_component->trace_name_base->str;
181 }
182 /* XXX: we might have to skip the first level, TBD. */
183
184 /* Sanitize the trace name. */
185 if (strlen(trace_name) == 2 && !strcmp(trace_name, "..")) {
186 fprintf(writer_component->err, "[error] Trace name cannot "
187 "be \"..\"\n");
188 goto error;
189 }
190
191 if (strstr(trace_name, "../")) {
192 fprintf(writer_component->err, "[error] Trace name cannot "
193 "contain \"../\", received \"%s\"\n",
194 trace_name);
195 goto error;
196
197 }
198
199 snprintf(trace_path, PATH_MAX, "%s/%s",
200 writer_component->base_path->str,
201 trace_name);
202 if (g_file_test(trace_path, G_FILE_TEST_EXISTS)) {
203 int i = 0;
204 do {
205 snprintf(trace_path, PATH_MAX, "%s/%s-%d",
206 writer_component->base_path->str,
207 trace_name, ++i);
208 } while (g_file_test(trace_path, G_FILE_TEST_EXISTS) && i < INT_MAX);
209 if (i == INT_MAX) {
210 fprintf(writer_component->err, "[error] Unable to find "
211 "a unique trace path\n");
212 goto error;
213 }
214 }
215
216 ret = 0;
217
218 goto end;
219
220error:
221 ret = -1;
222end:
223 return ret;
224}
225
bc506aa5 226static
f3168545 227struct fs_writer *insert_new_writer(
bc506aa5
JD
228 struct writer_component *writer_component,
229 struct bt_ctf_trace *trace)
230{
9ae49d3d
JD
231 struct bt_ctf_writer *ctf_writer = NULL;
232 struct bt_ctf_trace *writer_trace = NULL;
9b4f54fb 233 char trace_path[PATH_MAX];
bc506aa5 234 enum bt_component_status ret;
cdcf612f
JD
235 struct bt_ctf_stream *stream = NULL;
236 struct fs_writer *fs_writer = NULL;
237 int nr_stream, i;
bc506aa5 238
9b4f54fb
JD
239 ret = make_trace_path(writer_component, trace, trace_path);
240 if (ret) {
241 fprintf(writer_component->err, "[error] %s in %s:%d\n",
242 __func__, __FILE__, __LINE__);
243 goto error;
244 }
245
246 printf("ctf.fs sink creating trace in %s\n", trace_path);
9057f037 247
9b4f54fb 248 ctf_writer = bt_ctf_writer_create(trace_path);
bc506aa5
JD
249 if (!ctf_writer) {
250 fprintf(writer_component->err, "[error] %s in %s:%d\n",
251 __func__, __FILE__, __LINE__);
9ae49d3d 252 goto error;
bc506aa5
JD
253 }
254
91b73004
JD
255 writer_trace = bt_ctf_writer_get_trace(ctf_writer);
256 if (!writer_trace) {
91b73004
JD
257 fprintf(writer_component->err,
258 "[error] %s in %s:%d\n", __func__, __FILE__,
259 __LINE__);
9ae49d3d 260 goto error;
91b73004
JD
261 }
262
263 ret = ctf_copy_trace(writer_component->err, trace, writer_trace);
bc506aa5
JD
264 if (ret != BT_COMPONENT_STATUS_OK) {
265 fprintf(writer_component->err, "[error] Failed to copy trace\n");
266 fprintf(writer_component->err, "[error] %s in %s:%d\n",
267 __func__, __FILE__, __LINE__);
268 BT_PUT(ctf_writer);
f3168545 269 goto error;
bc506aa5 270 }
f3168545
JD
271
272 fs_writer = g_new0(struct fs_writer, 1);
273 if (!fs_writer) {
274 fprintf(writer_component->err,
275 "[error] %s in %s:%d\n", __func__, __FILE__,
276 __LINE__);
277 goto error;
278 }
279 fs_writer->writer = ctf_writer;
cdcf612f
JD
280 fs_writer->trace = trace;
281 fs_writer->writer_trace = writer_trace;
6f25f7c4 282 fs_writer->writer_component = writer_component;
cdcf612f 283 BT_PUT(writer_trace);
a619fcb7
JD
284 fs_writer->stream_class_map = g_hash_table_new_full(g_direct_hash,
285 g_direct_equal, NULL, (GDestroyNotify) unref_stream_class);
286 fs_writer->stream_map = g_hash_table_new_full(g_direct_hash,
287 g_direct_equal, NULL, (GDestroyNotify) unref_stream);
cdcf612f
JD
288 fs_writer->stream_states = g_hash_table_new_full(g_direct_hash,
289 g_direct_equal, NULL, destroy_stream_state_key);
290
291 /* Set all the existing streams in the unknown state. */
292 nr_stream = bt_ctf_trace_get_stream_count(trace);
293 for (i = 0; i < nr_stream; i++) {
cdcf612f
JD
294 stream = bt_ctf_trace_get_stream_by_index(trace, i);
295 if (!stream) {
296 fprintf(writer_component->err,
297 "[error] %s in %s:%d\n", __func__,
298 __FILE__, __LINE__);
299 goto error;
300 }
2881e5b5 301 insert_new_stream_state(writer_component, fs_writer, stream);
cdcf612f
JD
302 BT_PUT(stream);
303 }
304
305 /* Check if the trace is already static or register a listener. */
f3168545
JD
306 if (bt_ctf_trace_is_static(trace)) {
307 fs_writer->trace_static = 1;
308 fs_writer->static_listener_id = -1;
309 } else {
310 ret = bt_ctf_trace_add_is_static_listener(trace,
6f25f7c4 311 trace_is_static_listener, fs_writer);
f3168545
JD
312 if (ret < 0) {
313 fprintf(writer_component->err,
314 "[error] %s in %s:%d\n", __func__, __FILE__,
315 __LINE__);
f3168545
JD
316 goto error;
317 }
318 fs_writer->static_listener_id = ret;
319 }
bc506aa5
JD
320
321 g_hash_table_insert(writer_component->trace_map, (gpointer) trace,
f3168545 322 fs_writer);
bc506aa5 323
9ae49d3d
JD
324 goto end;
325
326error:
cdcf612f
JD
327 g_free(fs_writer);
328 fs_writer = NULL;
9ae49d3d 329 bt_put(writer_trace);
cdcf612f 330 bt_put(stream);
9ae49d3d 331 BT_PUT(ctf_writer);
bc506aa5 332end:
f3168545 333 return fs_writer;
bc506aa5
JD
334}
335
336static
f3168545 337struct fs_writer *get_fs_writer(struct writer_component *writer_component,
bc506aa5
JD
338 struct bt_ctf_stream_class *stream_class)
339{
9ae49d3d 340 struct bt_ctf_trace *trace = NULL;
f3168545 341 struct fs_writer *fs_writer;
bc506aa5
JD
342
343 trace = bt_ctf_stream_class_get_trace(stream_class);
344 if (!trace) {
bc506aa5
JD
345 fprintf(writer_component->err, "[error] %s in %s:%d\n",
346 __func__, __FILE__, __LINE__);
9ae49d3d 347 goto error;
bc506aa5
JD
348 }
349
f3168545 350 fs_writer = g_hash_table_lookup(writer_component->trace_map,
bc506aa5 351 (gpointer) trace);
f3168545
JD
352 if (!fs_writer) {
353 fs_writer = insert_new_writer(writer_component, trace);
bc506aa5 354 }
9ae49d3d
JD
355 BT_PUT(trace);
356 goto end;
bc506aa5 357
9ae49d3d 358error:
f3168545 359 fs_writer = NULL;
bc506aa5 360end:
f3168545 361 return fs_writer;
bc506aa5
JD
362}
363
a619fcb7
JD
364static
365struct fs_writer *get_fs_writer_from_stream(
366 struct writer_component *writer_component,
367 struct bt_ctf_stream *stream)
368{
369 struct bt_ctf_stream_class *stream_class = NULL;
370 struct fs_writer *fs_writer;
371
372 stream_class = bt_ctf_stream_get_class(stream);
373 if (!stream_class) {
374 fprintf(writer_component->err, "[error] %s in %s:%d\n",
375 __func__, __FILE__, __LINE__);
376 goto error;
377 }
378
379 fs_writer = get_fs_writer(writer_component, stream_class);
380 goto end;
381
382error:
383 fs_writer = NULL;
384
385end:
386 bt_put(stream_class);
387 return fs_writer;
388}
389
390static
391struct bt_ctf_stream_class *lookup_stream_class(
392 struct writer_component *writer_component,
393 struct bt_ctf_stream_class *stream_class)
394{
395 struct fs_writer *fs_writer = get_fs_writer(
396 writer_component, stream_class);
397 assert(fs_writer);
398 return (struct bt_ctf_stream_class *) g_hash_table_lookup(
399 fs_writer->stream_class_map, (gpointer) stream_class);
400}
401
402static
403struct bt_ctf_stream *lookup_stream(struct writer_component *writer_component,
404 struct bt_ctf_stream *stream)
405{
406 struct fs_writer *fs_writer = get_fs_writer_from_stream(
407 writer_component, stream);
408 assert(fs_writer);
409 return (struct bt_ctf_stream *) g_hash_table_lookup(
410 fs_writer->stream_map, (gpointer) stream);
411}
412
413static
414struct bt_ctf_stream *insert_new_stream(
415 struct writer_component *writer_component,
416 struct fs_writer *fs_writer,
417 struct bt_ctf_stream_class *stream_class,
418 struct bt_ctf_stream *stream)
419{
420 struct bt_ctf_stream *writer_stream = NULL;
421 struct bt_ctf_stream_class *writer_stream_class = NULL;
422 struct bt_ctf_writer *ctf_writer = bt_get(fs_writer->writer);
423
424 writer_stream_class = lookup_stream_class(writer_component,
425 stream_class);
426 if (!writer_stream_class) {
427 writer_stream_class = insert_new_stream_class(
428 writer_component, fs_writer, stream_class);
429 if (!writer_stream_class) {
430 fprintf(writer_component->err, "[error] %s in %s:%d\n",
431 __func__, __FILE__, __LINE__);
432 goto error;
433 }
434 }
435 bt_get(writer_stream_class);
436
437 writer_stream = bt_ctf_writer_create_stream(ctf_writer,
438 writer_stream_class);
439 if (!writer_stream) {
440 fprintf(writer_component->err, "[error] %s in %s:%d\n",
441 __func__, __FILE__, __LINE__);
442 goto error;
443 }
444
445 g_hash_table_insert(fs_writer->stream_map, (gpointer) stream,
446 writer_stream);
447
448 goto end;
449
450error:
451 BT_PUT(writer_stream);
452end:
453 bt_put(ctf_writer);
454 bt_put(writer_stream_class);
455 return writer_stream;
456}
457
458static
459struct bt_ctf_event_class *get_event_class(struct writer_component *writer_component,
460 struct bt_ctf_stream_class *writer_stream_class,
461 struct bt_ctf_event_class *event_class)
462{
463 return bt_ctf_stream_class_get_event_class_by_id(writer_stream_class,
464 bt_ctf_event_class_get_id(event_class));
465}
466
bc506aa5
JD
467static
468struct bt_ctf_stream *get_writer_stream(
469 struct writer_component *writer_component,
470 struct bt_ctf_packet *packet, struct bt_ctf_stream *stream)
471{
9ae49d3d 472 struct bt_ctf_stream *writer_stream = NULL;
bc506aa5 473
f384901f
JD
474 writer_stream = lookup_stream(writer_component, stream);
475 if (!writer_stream) {
bc506aa5
JD
476 fprintf(writer_component->err, "[error] %s in %s:%d\n",
477 __func__, __FILE__, __LINE__);
9ae49d3d 478 goto error;
bc506aa5 479 }
9ae49d3d
JD
480 bt_get(writer_stream);
481
482 goto end;
bc506aa5 483
9ae49d3d
JD
484error:
485 BT_PUT(writer_stream);
486end:
bc506aa5
JD
487 return writer_stream;
488}
489
f3168545 490BT_HIDDEN
a619fcb7
JD
491void writer_close(struct writer_component *writer_component,
492 struct fs_writer *fs_writer)
f3168545 493{
f3168545 494 if (fs_writer->static_listener_id > 0) {
a619fcb7 495 bt_ctf_trace_remove_is_static_listener(fs_writer->trace,
f3168545
JD
496 fs_writer->static_listener_id);
497 }
a619fcb7
JD
498
499 /* Empty the stream class HT. */
500 g_hash_table_foreach_remove(fs_writer->stream_class_map,
501 empty_ht, NULL);
502 g_hash_table_destroy(fs_writer->stream_class_map);
503
504 /* Empty the stream HT. */
505 g_hash_table_foreach_remove(fs_writer->stream_map,
506 empty_ht, NULL);
507 g_hash_table_destroy(fs_writer->stream_map);
cdcf612f
JD
508
509 /* Empty the stream state HT. */
510 g_hash_table_foreach_remove(fs_writer->stream_states,
511 empty_ht, NULL);
512 g_hash_table_destroy(fs_writer->stream_states);
f3168545
JD
513}
514
f384901f
JD
515BT_HIDDEN
516enum bt_component_status writer_stream_begin(
517 struct writer_component *writer_component,
518 struct bt_ctf_stream *stream)
519{
520 struct bt_ctf_stream_class *stream_class = NULL;
521 struct fs_writer *fs_writer;
f384901f
JD
522 struct bt_ctf_stream *writer_stream = NULL;
523 enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
cdcf612f 524 enum fs_writer_stream_state *state;
f384901f
JD
525
526 stream_class = bt_ctf_stream_get_class(stream);
527 if (!stream_class) {
528 fprintf(writer_component->err, "[error] %s in %s:%d\n",
529 __func__, __FILE__, __LINE__);
530 goto error;
531 }
532
533 fs_writer = get_fs_writer(writer_component, stream_class);
534 if (!fs_writer) {
535 fprintf(writer_component->err, "[error] %s in %s:%d\n",
536 __func__, __FILE__, __LINE__);
537 goto error;
538 }
cdcf612f
JD
539
540 /* Set the stream as active */
541 state = g_hash_table_lookup(fs_writer->stream_states, stream);
2881e5b5
JD
542 if (!state) {
543 if (fs_writer->trace_static) {
544 fprintf(writer_component->err, "[error] Adding a new "
545 "stream on a static trace\n");
546 goto error;
547 }
548 state = insert_new_stream_state(writer_component, fs_writer,
549 stream);
550 }
cdcf612f
JD
551 if (*state != FS_WRITER_UNKNOWN_STREAM) {
552 fprintf(writer_component->err, "[error] Unexpected stream "
553 "state %d\n", *state);
554 goto error;
555 }
556 *state = FS_WRITER_ACTIVE_STREAM;
557
a619fcb7 558 writer_stream = insert_new_stream(writer_component, fs_writer,
f384901f
JD
559 stream_class, stream);
560 if (!writer_stream) {
561 fprintf(writer_component->err, "[error] %s in %s:%d\n",
562 __func__, __FILE__, __LINE__);
563 goto error;
564 }
565 fs_writer->active_streams++;
566
567 goto end;
568
569error:
570 ret = BT_COMPONENT_STATUS_ERROR;
571end:
f384901f
JD
572 bt_put(stream_class);
573 return ret;
574}
575
f3168545
JD
576BT_HIDDEN
577enum bt_component_status writer_stream_end(
578 struct writer_component *writer_component,
579 struct bt_ctf_stream *stream)
580{
581 struct bt_ctf_stream_class *stream_class = NULL;
582 struct fs_writer *fs_writer;
583 struct bt_ctf_trace *trace = NULL;
584 enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
cdcf612f 585 enum fs_writer_stream_state *state;
f3168545 586
f3168545
JD
587 stream_class = bt_ctf_stream_get_class(stream);
588 if (!stream_class) {
589 fprintf(writer_component->err, "[error] %s in %s:%d\n",
590 __func__, __FILE__, __LINE__);
591 goto error;
592 }
593
594 fs_writer = get_fs_writer(writer_component, stream_class);
595 if (!fs_writer) {
596 fprintf(writer_component->err, "[error] %s in %s:%d\n",
597 __func__, __FILE__, __LINE__);
598 goto error;
599 }
cdcf612f
JD
600
601 state = g_hash_table_lookup(fs_writer->stream_states, stream);
602 if (*state != FS_WRITER_ACTIVE_STREAM) {
603 fprintf(writer_component->err, "[error] Unexpected stream "
604 "state %d\n", *state);
605 goto error;
606 }
607 *state = FS_WRITER_COMPLETED_STREAM;
608
a619fcb7 609 g_hash_table_remove(fs_writer->stream_map, stream);
f3168545 610
cdcf612f
JD
611 if (fs_writer->trace_static) {
612 int trace_completed = 1;
613
614 g_hash_table_foreach(fs_writer->stream_states,
615 check_completed_trace, &trace_completed);
616 if (trace_completed) {
617 writer_close(writer_component, fs_writer);
618 g_hash_table_remove(writer_component->trace_map,
619 fs_writer->trace);
620 }
f3168545
JD
621 }
622
623 goto end;
624
625error:
626 ret = BT_COMPONENT_STATUS_ERROR;
627end:
628 BT_PUT(trace);
629 BT_PUT(stream_class);
630 return ret;
631}
632
bc506aa5
JD
633BT_HIDDEN
634enum bt_component_status writer_new_packet(
635 struct writer_component *writer_component,
636 struct bt_ctf_packet *packet)
637{
9ae49d3d
JD
638 struct bt_ctf_stream *stream = NULL, *writer_stream = NULL;
639 struct bt_ctf_field *writer_packet_context = NULL;
bc506aa5 640 enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
b2f1f465 641 int int_ret;
bc506aa5
JD
642
643 stream = bt_ctf_packet_get_stream(packet);
644 if (!stream) {
bc506aa5
JD
645 fprintf(writer_component->err, "[error] %s in %s:%d\n",
646 __func__, __FILE__, __LINE__);
9ae49d3d 647 goto error;
bc506aa5
JD
648 }
649
bc506aa5
JD
650 writer_stream = get_writer_stream(writer_component, packet, stream);
651 if (!writer_stream) {
bc506aa5
JD
652 fprintf(writer_component->err, "[error] %s in %s:%d\n",
653 __func__, __FILE__, __LINE__);
9ae49d3d 654 goto error;
bc506aa5 655 }
9ae49d3d 656 BT_PUT(stream);
bc506aa5 657
b2f1f465 658 writer_packet_context = ctf_copy_packet_context(writer_component->err,
79697a58 659 packet, writer_stream);
b2f1f465 660 if (!writer_packet_context) {
91b73004
JD
661 fprintf(writer_component->err, "[error] %s in %s:%d\n",
662 __func__, __FILE__, __LINE__);
9ae49d3d 663 goto error;
bc506aa5
JD
664 }
665
b2f1f465
JD
666 int_ret = bt_ctf_stream_set_packet_context(writer_stream,
667 writer_packet_context);
668 if (int_ret < 0) {
b2f1f465
JD
669 fprintf(writer_component->err, "[error] %s in %s:%d\n", __func__,
670 __FILE__, __LINE__);
9ae49d3d 671 goto error;
b2f1f465 672 }
9ae49d3d
JD
673 BT_PUT(writer_stream);
674 BT_PUT(writer_packet_context);
b2f1f465 675
9ae49d3d 676 goto end;
bc506aa5 677
9ae49d3d
JD
678error:
679 ret = BT_COMPONENT_STATUS_ERROR;
680end:
681 bt_put(writer_stream);
b2f1f465 682 bt_put(writer_packet_context);
91b73004 683 bt_put(stream);
bc506aa5
JD
684 return ret;
685}
686
687BT_HIDDEN
688enum bt_component_status writer_close_packet(
689 struct writer_component *writer_component,
690 struct bt_ctf_packet *packet)
691{
9ae49d3d 692 struct bt_ctf_stream *stream = NULL, *writer_stream = NULL;
bc506aa5
JD
693 enum bt_component_status ret;
694
695 stream = bt_ctf_packet_get_stream(packet);
696 if (!stream) {
bc506aa5
JD
697 fprintf(writer_component->err, "[error] %s in %s:%d\n",
698 __func__, __FILE__, __LINE__);
9ae49d3d 699 goto error;
bc506aa5
JD
700 }
701
702 writer_stream = lookup_stream(writer_component, stream);
703 if (!writer_stream) {
bc506aa5
JD
704 fprintf(writer_component->err, "[error] %s in %s:%d\n",
705 __func__, __FILE__, __LINE__);
9ae49d3d 706 goto error;
bc506aa5 707 }
9ae49d3d 708 BT_PUT(stream);
bc506aa5 709
9ae49d3d 710 bt_get(writer_stream);
bc506aa5 711
bc506aa5
JD
712 ret = bt_ctf_stream_flush(writer_stream);
713 if (ret < 0) {
714 fprintf(writer_component->err,
715 "[error] Failed to flush packet\n");
9ae49d3d 716 goto error;
bc506aa5 717 }
9ae49d3d 718 BT_PUT(writer_stream);
bc506aa5
JD
719
720 ret = BT_COMPONENT_STATUS_OK;
9ae49d3d 721 goto end;
bc506aa5 722
9ae49d3d
JD
723error:
724 ret = BT_COMPONENT_STATUS_ERROR;
725end:
bc506aa5 726 bt_put(writer_stream);
bc506aa5 727 bt_put(stream);
bc506aa5
JD
728 return ret;
729}
730
bc506aa5
JD
731BT_HIDDEN
732enum bt_component_status writer_output_event(
733 struct writer_component *writer_component,
734 struct bt_ctf_event *event)
735{
736 enum bt_component_status ret;
9ae49d3d
JD
737 struct bt_ctf_event_class *event_class = NULL, *writer_event_class = NULL;
738 struct bt_ctf_stream *stream = NULL, *writer_stream = NULL;
739 struct bt_ctf_stream_class *stream_class = NULL, *writer_stream_class = NULL;
740 struct bt_ctf_event *writer_event = NULL;
bc506aa5
JD
741 const char *event_name;
742 int int_ret;
743
744 event_class = bt_ctf_event_get_class(event);
745 if (!event_class) {
bc506aa5
JD
746 fprintf(writer_component->err, "[error] %s in %s:%d\n", __func__,
747 __FILE__, __LINE__);
9ae49d3d 748 goto error;
bc506aa5
JD
749 }
750
751 event_name = bt_ctf_event_class_get_name(event_class);
752 if (!event_name) {
bc506aa5
JD
753 fprintf(writer_component->err, "[error] %s in %s:%d\n", __func__,
754 __FILE__, __LINE__);
9ae49d3d 755 goto error;
bc506aa5 756 }
bc506aa5
JD
757
758 stream = bt_ctf_event_get_stream(event);
759 if (!stream) {
bc506aa5
JD
760 fprintf(writer_component->err, "[error] %s in %s:%d\n", __func__,
761 __FILE__, __LINE__);
9ae49d3d 762 goto error;
bc506aa5
JD
763 }
764
765 writer_stream = lookup_stream(writer_component, stream);
766 if (!writer_stream || !bt_get(writer_stream)) {
bc506aa5
JD
767 fprintf(writer_component->err, "[error] %s in %s:%d\n", __func__,
768 __FILE__, __LINE__);
9ae49d3d 769 goto error;
bc506aa5
JD
770 }
771
772 stream_class = bt_ctf_event_class_get_stream_class(event_class);
773 if (!stream_class) {
bc506aa5
JD
774 fprintf(writer_component->err, "[error] %s in %s:%d\n", __func__,
775 __FILE__, __LINE__);
9ae49d3d 776 goto error;
bc506aa5
JD
777 }
778
a619fcb7 779 writer_stream_class = lookup_stream_class(writer_component, stream_class);
bc506aa5 780 if (!writer_stream_class || !bt_get(writer_stream_class)) {
bc506aa5
JD
781 fprintf(writer_component->err, "[error] %s in %s:%d\n", __func__,
782 __FILE__, __LINE__);
9ae49d3d 783 goto error;
bc506aa5
JD
784 }
785
786 writer_event_class = get_event_class(writer_component,
787 writer_stream_class, event_class);
788 if (!writer_event_class) {
af34e875
JD
789 writer_event_class = ctf_copy_event_class(writer_component->err,
790 event_class);
791 if (!writer_event_class) {
af34e875
JD
792 fprintf(writer_component->err, "[error] %s in %s:%d\n",
793 __func__, __FILE__, __LINE__);
9ae49d3d 794 goto error;
af34e875
JD
795 }
796 int_ret = bt_ctf_stream_class_add_event_class(
797 writer_stream_class, writer_event_class);
798 if (int_ret) {
af34e875
JD
799 fprintf(writer_component->err, "[error] %s in %s:%d\n",
800 __func__, __FILE__, __LINE__);
9ae49d3d 801 goto error;
af34e875 802 }
bc506aa5
JD
803 }
804
b2f1f465
JD
805 writer_event = ctf_copy_event(writer_component->err, event,
806 writer_event_class, true);
bc506aa5 807 if (!writer_event) {
bc506aa5
JD
808 fprintf(writer_component->err, "[error] %s in %s:%d\n", __func__,
809 __FILE__, __LINE__);
810 fprintf(writer_component->err, "[error] Failed to copy event %s\n",
811 bt_ctf_event_class_get_name(writer_event_class));
9ae49d3d 812 goto error;
bc506aa5
JD
813 }
814
815 int_ret = bt_ctf_stream_append_event(writer_stream, writer_event);
816 if (int_ret < 0) {
bc506aa5
JD
817 fprintf(writer_component->err, "[error] %s in %s:%d\n", __func__,
818 __FILE__, __LINE__);
819 fprintf(writer_component->err, "[error] Failed to append event %s\n",
820 bt_ctf_event_class_get_name(writer_event_class));
9ae49d3d 821 goto error;
bc506aa5
JD
822 }
823
824 ret = BT_COMPONENT_STATUS_OK;
9ae49d3d 825 goto end;
bc506aa5 826
9ae49d3d
JD
827error:
828 ret = BT_COMPONENT_STATUS_ERROR;
829end:
bc506aa5 830 bt_put(writer_event);
bc506aa5 831 bt_put(writer_event_class);
bc506aa5 832 bt_put(writer_stream_class);
bc506aa5 833 bt_put(stream_class);
bc506aa5 834 bt_put(writer_stream);
bc506aa5 835 bt_put(stream);
bc506aa5 836 bt_put(event_class);
bc506aa5
JD
837 return ret;
838}
This page took 0.067185 seconds and 4 git commands to generate.