Commit | Line | Data |
---|---|---|
19ce87a4 JD |
1 | /* |
2 | * copy.c | |
3 | * | |
4 | * Babeltrace Copy Trace Structure | |
5 | * | |
6 | * Copyright 2017 Julien Desfossez <jdesfossez@efficios.com> | |
7 | * | |
8 | * Author: Julien Desfossez <jdesfossez@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> | |
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> | |
38 | ||
39 | #include <ctfcopytrace.h> | |
40 | #include "iterator.h" | |
41 | ||
42 | static | |
43 | struct bt_ctf_packet *lookup_packet(struct trimmer_iterator *trim_it, | |
44 | struct bt_ctf_packet *packet) | |
45 | { | |
46 | return (struct bt_ctf_packet *) g_hash_table_lookup( | |
47 | trim_it->packet_map, | |
48 | (gpointer) packet); | |
49 | } | |
50 | ||
51 | static | |
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) | |
55 | { | |
9ae49d3d | 56 | struct bt_ctf_packet *writer_packet = NULL; |
19ce87a4 JD |
57 | |
58 | writer_packet = bt_ctf_packet_create(stream); | |
59 | if (!writer_packet) { | |
60 | fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__, | |
61 | __FILE__, __LINE__); | |
62 | goto end; | |
63 | } | |
64 | g_hash_table_insert(trim_it->packet_map, (gpointer) packet, writer_packet); | |
65 | ||
66 | end: | |
67 | return writer_packet; | |
68 | } | |
69 | ||
70 | BT_HIDDEN | |
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) | |
74 | { | |
75 | enum bt_component_status ret; | |
9ae49d3d JD |
76 | struct bt_ctf_field *packet_context = NULL, *writer_packet_context = NULL; |
77 | struct bt_ctf_field_type *struct_type = NULL, *field_type = NULL; | |
78 | struct bt_ctf_field *field = NULL, *writer_field = NULL; | |
19ce87a4 JD |
79 | int nr_fields, i, int_ret; |
80 | ||
81 | packet_context = bt_ctf_packet_get_context(writer_packet); | |
82 | if (!packet_context) { | |
19ce87a4 JD |
83 | fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__, |
84 | __LINE__); | |
9ae49d3d | 85 | goto error; |
19ce87a4 JD |
86 | } |
87 | ||
88 | struct_type = bt_ctf_field_get_type(packet_context); | |
89 | if (!struct_type) { | |
19ce87a4 JD |
90 | fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__, |
91 | __LINE__); | |
9ae49d3d | 92 | goto error; |
19ce87a4 JD |
93 | } |
94 | ||
95 | writer_packet_context = bt_ctf_packet_get_context(writer_packet); | |
96 | if (!writer_packet_context) { | |
19ce87a4 JD |
97 | fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__, |
98 | __LINE__); | |
9ae49d3d | 99 | goto error; |
19ce87a4 JD |
100 | } |
101 | ||
102 | nr_fields = bt_ctf_field_type_structure_get_field_count(struct_type); | |
103 | for (i = 0; i < nr_fields; i++) { | |
19ce87a4 JD |
104 | const char *field_name; |
105 | ||
106 | field = bt_ctf_field_structure_get_field_by_index( | |
107 | packet_context, i); | |
108 | if (!field) { | |
19ce87a4 JD |
109 | fprintf(err, "[error] %s in %s:%d\n", __func__, |
110 | __FILE__, __LINE__); | |
9ae49d3d | 111 | goto error; |
19ce87a4 JD |
112 | } |
113 | if (bt_ctf_field_type_structure_get_field(struct_type, | |
114 | &field_name, &field_type, i) < 0) { | |
19ce87a4 JD |
115 | fprintf(err, "[error] %s in %s:%d\n", __func__, |
116 | __FILE__, __LINE__); | |
9ae49d3d | 117 | goto error; |
19ce87a4 JD |
118 | } |
119 | if (strcmp(field_name, name)) { | |
9ae49d3d JD |
120 | BT_PUT(field_type); |
121 | BT_PUT(field); | |
19ce87a4 JD |
122 | continue; |
123 | } | |
1487a16a | 124 | if (bt_ctf_field_type_get_type_id(field_type) != BT_CTF_FIELD_TYPE_ID_INTEGER) { |
19ce87a4 | 125 | fprintf(err, "[error] Unexpected packet context field type\n"); |
9ae49d3d | 126 | goto error; |
19ce87a4 JD |
127 | } |
128 | writer_field = bt_ctf_field_structure_get_field(writer_packet_context, | |
129 | field_name); | |
130 | if (!writer_field) { | |
19ce87a4 JD |
131 | fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__, |
132 | __LINE__); | |
9ae49d3d | 133 | goto error; |
19ce87a4 JD |
134 | } |
135 | ||
136 | int_ret = bt_ctf_field_unsigned_integer_set_value(writer_field, value); | |
19ce87a4 | 137 | if (int_ret < 0) { |
19ce87a4 JD |
138 | fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__, |
139 | __LINE__); | |
9ae49d3d | 140 | goto error; |
19ce87a4 | 141 | } |
9ae49d3d JD |
142 | BT_PUT(writer_field); |
143 | BT_PUT(field_type); | |
144 | BT_PUT(field); | |
19ce87a4 JD |
145 | } |
146 | ||
147 | ret = BT_COMPONENT_STATUS_OK; | |
9ae49d3d | 148 | goto end; |
19ce87a4 | 149 | |
9ae49d3d JD |
150 | error: |
151 | bt_put(writer_field); | |
152 | bt_put(field_type); | |
153 | bt_put(field); | |
154 | ret = BT_COMPONENT_STATUS_ERROR; | |
155 | end: | |
19ce87a4 | 156 | bt_put(struct_type); |
19ce87a4 | 157 | bt_put(packet_context); |
19ce87a4 JD |
158 | return ret; |
159 | } | |
160 | ||
161 | BT_HIDDEN | |
162 | struct bt_ctf_packet *trimmer_new_packet( | |
163 | struct trimmer_iterator *trim_it, | |
164 | struct bt_ctf_packet *packet) | |
165 | { | |
9ae49d3d | 166 | struct bt_ctf_stream *stream = NULL; |
19ce87a4 JD |
167 | struct bt_ctf_packet *writer_packet = NULL; |
168 | int int_ret; | |
169 | ||
170 | stream = bt_ctf_packet_get_stream(packet); | |
171 | if (!stream) { | |
172 | fprintf(trim_it->err, "[error] %s in %s:%d\n", | |
173 | __func__, __FILE__, __LINE__); | |
9ae49d3d | 174 | goto error; |
19ce87a4 JD |
175 | } |
176 | ||
177 | /* | |
178 | * If a packet was already opened, close it and remove it from | |
179 | * the HT. | |
180 | */ | |
181 | writer_packet = lookup_packet(trim_it, packet); | |
182 | if (writer_packet) { | |
183 | g_hash_table_remove(trim_it->packet_map, packet); | |
9ae49d3d | 184 | BT_PUT(writer_packet); |
19ce87a4 JD |
185 | } |
186 | ||
187 | writer_packet = insert_new_packet(trim_it, packet, stream); | |
188 | if (!writer_packet) { | |
189 | fprintf(trim_it->err, "[error] %s in %s:%d\n", | |
190 | __func__, __FILE__, __LINE__); | |
9ae49d3d | 191 | goto error; |
19ce87a4 JD |
192 | } |
193 | bt_get(writer_packet); | |
194 | ||
9877e1aa JD |
195 | int_ret = ctf_packet_copy_context(trim_it->err, packet, |
196 | stream, writer_packet); | |
197 | if (int_ret < 0) { | |
19ce87a4 JD |
198 | fprintf(trim_it->err, "[error] %s in %s:%d\n", |
199 | __func__, __FILE__, __LINE__); | |
9ae49d3d | 200 | goto error; |
19ce87a4 JD |
201 | } |
202 | ||
9ae49d3d JD |
203 | goto end; |
204 | ||
205 | error: | |
206 | BT_PUT(writer_packet); | |
207 | end: | |
19ce87a4 | 208 | bt_put(stream); |
19ce87a4 JD |
209 | return writer_packet; |
210 | } | |
211 | ||
212 | BT_HIDDEN | |
213 | struct bt_ctf_packet *trimmer_close_packet( | |
214 | struct trimmer_iterator *trim_it, | |
215 | struct bt_ctf_packet *packet) | |
216 | { | |
9ae49d3d | 217 | struct bt_ctf_packet *writer_packet = NULL; |
19ce87a4 JD |
218 | |
219 | writer_packet = lookup_packet(trim_it, packet); | |
220 | if (!writer_packet) { | |
221 | fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__, | |
222 | __FILE__, __LINE__); | |
223 | goto end; | |
224 | } | |
225 | ||
226 | g_hash_table_remove(trim_it->packet_map, packet); | |
227 | ||
228 | end: | |
229 | return writer_packet; | |
230 | } | |
231 | ||
232 | BT_HIDDEN | |
233 | struct bt_ctf_event *trimmer_output_event( | |
234 | struct trimmer_iterator *trim_it, | |
235 | struct bt_ctf_event *event) | |
236 | { | |
9ae49d3d | 237 | struct bt_ctf_event_class *event_class = NULL; |
19ce87a4 | 238 | struct bt_ctf_event *writer_event = NULL; |
9ae49d3d | 239 | struct bt_ctf_packet *packet = NULL, *writer_packet = NULL; |
19ce87a4 JD |
240 | const char *event_name; |
241 | int int_ret; | |
242 | ||
243 | event_class = bt_ctf_event_get_class(event); | |
244 | if (!event_class) { | |
245 | fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__, | |
246 | __FILE__, __LINE__); | |
9ae49d3d | 247 | goto error; |
19ce87a4 JD |
248 | } |
249 | ||
250 | event_name = bt_ctf_event_class_get_name(event_class); | |
251 | if (!event_name) { | |
252 | fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__, | |
253 | __FILE__, __LINE__); | |
9ae49d3d | 254 | goto error; |
19ce87a4 JD |
255 | } |
256 | ||
257 | writer_event = ctf_copy_event(trim_it->err, event, event_class, false); | |
258 | if (!writer_event) { | |
259 | fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__, | |
260 | __FILE__, __LINE__); | |
261 | fprintf(trim_it->err, "[error] Failed to copy event %s\n", | |
262 | bt_ctf_event_class_get_name(event_class)); | |
9ae49d3d | 263 | goto error; |
19ce87a4 JD |
264 | } |
265 | ||
266 | packet = bt_ctf_event_get_packet(event); | |
267 | if (!packet) { | |
19ce87a4 JD |
268 | fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__, |
269 | __FILE__, __LINE__); | |
9ae49d3d | 270 | goto error; |
19ce87a4 JD |
271 | } |
272 | ||
273 | writer_packet = lookup_packet(trim_it, packet); | |
274 | if (!writer_packet) { | |
19ce87a4 JD |
275 | fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__, |
276 | __FILE__, __LINE__); | |
9ae49d3d | 277 | goto error; |
19ce87a4 JD |
278 | } |
279 | bt_get(writer_packet); | |
280 | ||
281 | int_ret = bt_ctf_event_set_packet(writer_event, writer_packet); | |
282 | if (int_ret < 0) { | |
19ce87a4 JD |
283 | fprintf(trim_it->err, "[error] %s in %s:%d\n", __func__, |
284 | __FILE__, __LINE__); | |
285 | fprintf(trim_it->err, "[error] Failed to append event %s\n", | |
286 | bt_ctf_event_class_get_name(event_class)); | |
9ae49d3d | 287 | goto error; |
19ce87a4 JD |
288 | } |
289 | ||
9ae49d3d JD |
290 | /* We keep the reference on the writer_event to create a notification. */ |
291 | goto end; | |
19ce87a4 | 292 | |
9ae49d3d JD |
293 | error: |
294 | BT_PUT(writer_event); | |
295 | end: | |
19ce87a4 | 296 | bt_put(writer_packet); |
19ce87a4 | 297 | bt_put(packet); |
19ce87a4 | 298 | bt_put(event_class); |
19ce87a4 JD |
299 | return writer_event; |
300 | } |