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