4 * Babeltrace Copy Trace Structure
6 * Copyright 2017 Julien Desfossez <jdesfossez@efficios.com>
8 * Author: Julien Desfossez <jdesfossez@efficios.com>
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:
17 * The above copyright notice and this permission notice shall be included in
18 * all copies or substantial portions of the Software.
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
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>
34 #include <babeltrace/ctf-ir/clock-class.h>
35 #include <babeltrace/ctf-ir/fields.h>
36 #include <babeltrace/ctf-writer/stream-class.h>
37 #include <babeltrace/ctf-writer/stream.h>
39 #include <ctfcopytrace.h>
43 struct bt_ctf_packet
*lookup_packet(struct trimmer_iterator
*trim_it
,
44 struct bt_ctf_packet
*packet
)
46 return (struct bt_ctf_packet
*) g_hash_table_lookup(
52 struct bt_ctf_packet
*insert_new_packet(struct trimmer_iterator
*trim_it
,
53 struct bt_ctf_packet
*packet
,
54 struct bt_ctf_stream
*stream
)
56 struct bt_ctf_packet
*writer_packet
;
58 writer_packet
= bt_ctf_packet_create(stream
);
60 fprintf(trim_it
->err
, "[error] %s in %s:%d\n", __func__
,
64 g_hash_table_insert(trim_it
->packet_map
, (gpointer
) packet
, writer_packet
);
71 enum bt_component_status
update_packet_context_field(FILE *err
,
72 struct bt_ctf_packet
*writer_packet
,
73 const char *name
, int64_t value
)
75 enum bt_component_status ret
;
76 struct bt_ctf_field
*packet_context
, *writer_packet_context
;
77 struct bt_ctf_field_type
*struct_type
, *writer_packet_context_type
;
78 struct bt_ctf_stream_class
*stream_class
;
79 struct bt_ctf_stream
*stream
;
80 int nr_fields
, i
, int_ret
;
82 packet_context
= bt_ctf_packet_get_context(writer_packet
);
83 if (!packet_context
) {
84 ret
= BT_COMPONENT_STATUS_ERROR
;
85 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
90 stream
= bt_ctf_packet_get_stream(writer_packet
);
92 ret
= BT_COMPONENT_STATUS_ERROR
;
93 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
95 goto end_put_packet_context
;
98 stream_class
= bt_ctf_stream_get_class(stream
);
100 ret
= BT_COMPONENT_STATUS_ERROR
;
101 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
106 writer_packet_context_type
= bt_ctf_stream_class_get_packet_context_type(
108 if (!writer_packet_context_type
) {
109 ret
= BT_COMPONENT_STATUS_ERROR
;
110 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
112 goto end_put_stream_class
;
115 struct_type
= bt_ctf_field_get_type(packet_context
);
117 ret
= BT_COMPONENT_STATUS_ERROR
;
118 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
120 goto end_put_writer_packet_context_type
;
123 writer_packet_context
= bt_ctf_packet_get_context(writer_packet
);
124 if (!writer_packet_context
) {
125 ret
= BT_COMPONENT_STATUS_ERROR
;
126 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
128 goto end_put_struct_type
;
131 nr_fields
= bt_ctf_field_type_structure_get_field_count(struct_type
);
132 for (i
= 0; i
< nr_fields
; i
++) {
133 struct bt_ctf_field
*field
, *writer_field
;
134 struct bt_ctf_field_type
*field_type
;
135 const char *field_name
;
137 field
= bt_ctf_field_structure_get_field_by_index(
140 ret
= BT_COMPONENT_STATUS_ERROR
;
141 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
143 goto end_put_writer_packet_context
;
145 if (bt_ctf_field_type_structure_get_field(struct_type
,
146 &field_name
, &field_type
, i
) < 0) {
147 ret
= BT_COMPONENT_STATUS_ERROR
;
149 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
151 goto end_put_writer_packet_context
;
153 if (strcmp(field_name
, name
)) {
158 if (bt_ctf_field_type_get_type_id(field_type
) != BT_CTF_TYPE_ID_INTEGER
) {
159 fprintf(err
, "[error] Unexpected packet context field type\n");
161 ret
= BT_COMPONENT_STATUS_ERROR
;
162 goto end_put_writer_packet_context
;
164 writer_field
= bt_ctf_field_structure_get_field(writer_packet_context
,
167 ret
= BT_COMPONENT_STATUS_ERROR
;
168 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
173 int_ret
= bt_ctf_field_unsigned_integer_set_value(writer_field
, value
);
174 bt_put(writer_field
);
176 ret
= BT_COMPONENT_STATUS_ERROR
;
177 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
179 goto end_put_writer_packet_context
;
186 ret
= BT_COMPONENT_STATUS_OK
;
188 end_put_writer_packet_context
:
189 bt_put(writer_packet_context
);
192 end_put_writer_packet_context_type
:
193 bt_put(writer_packet_context_type
);
194 end_put_stream_class
:
195 bt_put(stream_class
);
196 end_put_packet_context
:
197 bt_put(packet_context
);
205 struct bt_ctf_packet
*trimmer_new_packet(
206 struct trimmer_iterator
*trim_it
,
207 struct bt_ctf_packet
*packet
)
209 struct bt_ctf_stream
*stream
;
210 struct bt_ctf_field
*writer_packet_context
;
211 struct bt_ctf_packet
*writer_packet
= NULL
;
214 stream
= bt_ctf_packet_get_stream(packet
);
216 fprintf(trim_it
->err
, "[error] %s in %s:%d\n",
217 __func__
, __FILE__
, __LINE__
);
222 * If a packet was already opened, close it and remove it from
225 writer_packet
= lookup_packet(trim_it
, packet
);
227 g_hash_table_remove(trim_it
->packet_map
, packet
);
228 bt_put(writer_packet
);
231 writer_packet
= insert_new_packet(trim_it
, packet
, stream
);
232 if (!writer_packet
) {
233 fprintf(trim_it
->err
, "[error] %s in %s:%d\n",
234 __func__
, __FILE__
, __LINE__
);
237 bt_get(writer_packet
);
239 writer_packet_context
= ctf_copy_packet_context(trim_it
->err
, packet
,
241 if (!writer_packet_context
) {
242 BT_PUT(writer_packet
);
243 fprintf(trim_it
->err
, "[error] %s in %s:%d\n",
244 __func__
, __FILE__
, __LINE__
);
248 int_ret
= bt_ctf_packet_set_context(writer_packet
, writer_packet_context
);
250 BT_PUT(writer_packet
);
251 fprintf(trim_it
->err
, "[error] %s in %s:%d\n",
252 __func__
, __FILE__
, __LINE__
);
253 goto end_put_writer_packet_context
;
256 end_put_writer_packet_context
:
257 bt_put(writer_packet_context
);
261 return writer_packet
;
265 struct bt_ctf_packet
*trimmer_close_packet(
266 struct trimmer_iterator
*trim_it
,
267 struct bt_ctf_packet
*packet
)
269 struct bt_ctf_packet
*writer_packet
;
271 writer_packet
= lookup_packet(trim_it
, packet
);
272 if (!writer_packet
) {
273 fprintf(trim_it
->err
, "[error] %s in %s:%d\n", __func__
,
278 g_hash_table_remove(trim_it
->packet_map
, packet
);
281 return writer_packet
;
285 struct bt_ctf_event
*trimmer_output_event(
286 struct trimmer_iterator
*trim_it
,
287 struct bt_ctf_event
*event
)
289 struct bt_ctf_event_class
*event_class
;
290 struct bt_ctf_stream
*stream
;
291 struct bt_ctf_stream_class
*stream_class
;
292 struct bt_ctf_event
*writer_event
= NULL
;
293 struct bt_ctf_packet
*packet
, *writer_packet
;
294 const char *event_name
;
297 event_class
= bt_ctf_event_get_class(event
);
299 fprintf(trim_it
->err
, "[error] %s in %s:%d\n", __func__
,
304 event_name
= bt_ctf_event_class_get_name(event_class
);
306 fprintf(trim_it
->err
, "[error] %s in %s:%d\n", __func__
,
308 goto end_put_event_class
;
311 stream
= bt_ctf_event_get_stream(event
);
313 fprintf(trim_it
->err
, "[error] %s in %s:%d\n", __func__
,
315 goto end_put_event_class
;
318 stream_class
= bt_ctf_event_class_get_stream_class(event_class
);
320 fprintf(trim_it
->err
, "[error] %s in %s:%d\n", __func__
,
325 writer_event
= ctf_copy_event(trim_it
->err
, event
, event_class
, false);
327 fprintf(trim_it
->err
, "[error] %s in %s:%d\n", __func__
,
329 fprintf(trim_it
->err
, "[error] Failed to copy event %s\n",
330 bt_ctf_event_class_get_name(event_class
));
331 goto end_put_stream_class
;
334 packet
= bt_ctf_event_get_packet(event
);
336 BT_PUT(writer_event
);
337 fprintf(trim_it
->err
, "[error] %s in %s:%d\n", __func__
,
339 goto end_put_stream_class
;
342 writer_packet
= lookup_packet(trim_it
, packet
);
343 if (!writer_packet
) {
344 BT_PUT(writer_event
);
345 fprintf(trim_it
->err
, "[error] %s in %s:%d\n", __func__
,
349 bt_get(writer_packet
);
351 int_ret
= bt_ctf_event_set_packet(writer_event
, writer_packet
);
353 BT_PUT(writer_event
);
354 fprintf(trim_it
->err
, "[error] %s in %s:%d\n", __func__
,
356 fprintf(trim_it
->err
, "[error] Failed to append event %s\n",
357 bt_ctf_event_class_get_name(event_class
));
358 goto end_put_writer_packet
;
361 /* We keep the reference on the writer event */
363 end_put_writer_packet
:
364 bt_put(writer_packet
);
367 end_put_stream_class
:
368 bt_put(stream_class
);