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