Commit | Line | Data |
---|---|---|
273b65be JG |
1 | /* |
2 | * writer.c | |
3 | * | |
4 | * Babeltrace CTF Writer | |
5 | * | |
de9dd397 | 6 | * Copyright 2013, 2014 Jérémie Galarneau <jeremie.galarneau@efficios.com> |
273b65be JG |
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 | ||
adc315b8 | 29 | #include <babeltrace/ctf-ir/clock-internal.h> |
273b65be | 30 | #include <babeltrace/ctf-writer/writer-internal.h> |
adc315b8 JG |
31 | #include <babeltrace/ctf-ir/event-types-internal.h> |
32 | #include <babeltrace/ctf-ir/event-fields-internal.h> | |
273b65be | 33 | #include <babeltrace/ctf-writer/functor-internal.h> |
adc315b8 | 34 | #include <babeltrace/ctf-ir/stream-class-internal.h> |
3f043b05 | 35 | #include <babeltrace/ctf-ir/stream-internal.h> |
273b65be JG |
36 | #include <babeltrace/compiler.h> |
37 | #include <stdio.h> | |
38 | #include <stdlib.h> | |
39 | #include <sys/stat.h> | |
40 | #include <errno.h> | |
41 | #include <unistd.h> | |
42 | #include <fcntl.h> | |
43 | #include <inttypes.h> | |
44 | ||
273b65be JG |
45 | static |
46 | void bt_ctf_writer_destroy(struct bt_ctf_ref *ref); | |
47 | static | |
273b65be JG |
48 | int create_stream_file(struct bt_ctf_writer *writer, |
49 | struct bt_ctf_stream *stream); | |
50 | static | |
51 | void stream_flush_cb(struct bt_ctf_stream *stream, | |
52 | struct bt_ctf_writer *writer); | |
53 | ||
273b65be JG |
54 | struct bt_ctf_writer *bt_ctf_writer_create(const char *path) |
55 | { | |
56 | struct bt_ctf_writer *writer = NULL; | |
57 | ||
58 | if (!path) { | |
59 | goto error; | |
60 | } | |
61 | ||
62 | writer = g_new0(struct bt_ctf_writer, 1); | |
63 | if (!writer) { | |
64 | goto error; | |
65 | } | |
66 | ||
273b65be JG |
67 | bt_ctf_ref_init(&writer->ref_count); |
68 | writer->path = g_string_new(path); | |
69 | if (!writer->path) { | |
70 | goto error_destroy; | |
71 | } | |
72 | ||
bc37ae52 JG |
73 | writer->trace = bt_ctf_trace_create(); |
74 | if (!writer->trace) { | |
75 | goto error_destroy; | |
76 | } | |
77 | ||
273b65be JG |
78 | /* Create trace directory if necessary and open a metadata file */ |
79 | if (g_mkdir_with_parents(path, S_IRWXU | S_IRWXG)) { | |
80 | perror("g_mkdir_with_parents"); | |
81 | goto error_destroy; | |
82 | } | |
83 | ||
1b8180b9 | 84 | writer->trace_dir_fd = open(path, O_RDONLY, S_IRWXU | S_IRWXG); |
273b65be JG |
85 | if (writer->trace_dir_fd < 0) { |
86 | perror("open"); | |
87 | goto error_destroy; | |
88 | } | |
89 | ||
90 | writer->metadata_fd = openat(writer->trace_dir_fd, "metadata", | |
91 | O_WRONLY | O_CREAT | O_TRUNC, | |
92 | S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP); | |
273b65be JG |
93 | |
94 | return writer; | |
bc37ae52 | 95 | |
273b65be JG |
96 | error_destroy: |
97 | unlinkat(writer->trace_dir_fd, "metadata", 0); | |
98 | bt_ctf_writer_destroy(&writer->ref_count); | |
99 | writer = NULL; | |
100 | error: | |
101 | return writer; | |
102 | } | |
103 | ||
104 | void bt_ctf_writer_destroy(struct bt_ctf_ref *ref) | |
105 | { | |
106 | struct bt_ctf_writer *writer; | |
107 | ||
108 | if (!ref) { | |
109 | return; | |
110 | } | |
111 | ||
112 | writer = container_of(ref, struct bt_ctf_writer, ref_count); | |
113 | bt_ctf_writer_flush_metadata(writer); | |
114 | if (writer->path) { | |
115 | g_string_free(writer->path, TRUE); | |
116 | } | |
117 | ||
118 | if (writer->trace_dir_fd > 0) { | |
9bb7e58b MD |
119 | if (close(writer->trace_dir_fd)) { |
120 | perror("close"); | |
9bb7e58b | 121 | } |
273b65be JG |
122 | } |
123 | ||
124 | if (writer->metadata_fd > 0) { | |
9bb7e58b MD |
125 | if (close(writer->metadata_fd)) { |
126 | perror("close"); | |
9bb7e58b | 127 | } |
273b65be JG |
128 | } |
129 | ||
bc37ae52 | 130 | bt_ctf_trace_put(writer->trace); |
273b65be JG |
131 | g_free(writer); |
132 | } | |
133 | ||
134 | struct bt_ctf_stream *bt_ctf_writer_create_stream(struct bt_ctf_writer *writer, | |
135 | struct bt_ctf_stream_class *stream_class) | |
136 | { | |
273b65be JG |
137 | int stream_fd; |
138 | struct bt_ctf_stream *stream = NULL; | |
139 | ||
140 | if (!writer || !stream_class) { | |
141 | goto error; | |
142 | } | |
143 | ||
bc37ae52 | 144 | stream = bt_ctf_trace_create_stream(writer->trace, stream_class); |
273b65be JG |
145 | if (!stream) { |
146 | goto error; | |
147 | } | |
148 | ||
149 | stream_fd = create_stream_file(writer, stream); | |
150 | if (stream_fd < 0 || bt_ctf_stream_set_fd(stream, stream_fd)) { | |
151 | goto error; | |
152 | } | |
153 | ||
154 | bt_ctf_stream_set_flush_callback(stream, (flush_func)stream_flush_cb, | |
155 | writer); | |
273b65be JG |
156 | writer->frozen = 1; |
157 | return stream; | |
bc37ae52 | 158 | |
273b65be JG |
159 | error: |
160 | bt_ctf_stream_put(stream); | |
161 | return NULL; | |
162 | } | |
163 | ||
164 | int bt_ctf_writer_add_environment_field(struct bt_ctf_writer *writer, | |
165 | const char *name, | |
166 | const char *value) | |
167 | { | |
bc37ae52 | 168 | int ret = -1; |
273b65be | 169 | |
bc37ae52 JG |
170 | if (!writer || !name || !value) { |
171 | goto end; | |
273b65be JG |
172 | } |
173 | ||
bc37ae52 JG |
174 | ret = bt_ctf_trace_add_environment_field(writer->trace, |
175 | name, value); | |
176 | end: | |
273b65be JG |
177 | return ret; |
178 | } | |
179 | ||
180 | int bt_ctf_writer_add_clock(struct bt_ctf_writer *writer, | |
181 | struct bt_ctf_clock *clock) | |
182 | { | |
bc37ae52 | 183 | int ret = -1; |
273b65be JG |
184 | |
185 | if (!writer || !clock) { | |
12af6048 JG |
186 | goto end; |
187 | } | |
273b65be | 188 | |
bc37ae52 | 189 | ret = bt_ctf_trace_add_clock(writer->trace, clock); |
12af6048 JG |
190 | end: |
191 | return ret; | |
273b65be JG |
192 | } |
193 | ||
273b65be JG |
194 | char *bt_ctf_writer_get_metadata_string(struct bt_ctf_writer *writer) |
195 | { | |
bc37ae52 | 196 | char *metadata_string = NULL; |
273b65be JG |
197 | |
198 | if (!writer) { | |
199 | goto end; | |
200 | } | |
201 | ||
bc37ae52 JG |
202 | metadata_string = bt_ctf_trace_get_metadata_string( |
203 | writer->trace); | |
273b65be | 204 | end: |
bc37ae52 | 205 | return metadata_string; |
273b65be JG |
206 | } |
207 | ||
208 | void bt_ctf_writer_flush_metadata(struct bt_ctf_writer *writer) | |
209 | { | |
210 | int ret; | |
211 | char *metadata_string = NULL; | |
212 | ||
213 | if (!writer) { | |
214 | goto end; | |
215 | } | |
216 | ||
bc37ae52 JG |
217 | metadata_string = bt_ctf_trace_get_metadata_string( |
218 | writer->trace); | |
273b65be JG |
219 | if (!metadata_string) { |
220 | goto end; | |
221 | } | |
222 | ||
223 | if (lseek(writer->metadata_fd, 0, SEEK_SET) == (off_t)-1) { | |
224 | perror("lseek"); | |
225 | goto end; | |
226 | } | |
227 | ||
228 | if (ftruncate(writer->metadata_fd, 0)) { | |
229 | perror("ftruncate"); | |
230 | goto end; | |
231 | } | |
232 | ||
233 | ret = write(writer->metadata_fd, metadata_string, | |
234 | strlen(metadata_string)); | |
235 | if (ret < 0) { | |
236 | perror("write"); | |
237 | goto end; | |
238 | } | |
239 | end: | |
240 | g_free(metadata_string); | |
241 | } | |
242 | ||
243 | int bt_ctf_writer_set_byte_order(struct bt_ctf_writer *writer, | |
244 | enum bt_ctf_byte_order byte_order) | |
245 | { | |
246 | int ret = 0; | |
273b65be JG |
247 | |
248 | if (!writer || writer->frozen) { | |
249 | ret = -1; | |
250 | goto end; | |
251 | } | |
252 | ||
bc37ae52 JG |
253 | ret = bt_ctf_trace_set_byte_order(writer->trace, |
254 | byte_order); | |
273b65be JG |
255 | end: |
256 | return ret; | |
257 | } | |
258 | ||
259 | void bt_ctf_writer_get(struct bt_ctf_writer *writer) | |
260 | { | |
261 | if (!writer) { | |
262 | return; | |
263 | } | |
264 | ||
265 | bt_ctf_ref_get(&writer->ref_count); | |
266 | } | |
267 | ||
268 | void bt_ctf_writer_put(struct bt_ctf_writer *writer) | |
269 | { | |
270 | if (!writer) { | |
271 | return; | |
272 | } | |
273 | ||
274 | bt_ctf_ref_put(&writer->ref_count, bt_ctf_writer_destroy); | |
275 | } | |
276 | ||
273b65be JG |
277 | static |
278 | int create_stream_file(struct bt_ctf_writer *writer, | |
279 | struct bt_ctf_stream *stream) | |
280 | { | |
281 | int fd; | |
282 | GString *filename = g_string_new(stream->stream_class->name->str); | |
283 | ||
284 | g_string_append_printf(filename, "_%" PRIu32, stream->id); | |
285 | fd = openat(writer->trace_dir_fd, filename->str, | |
286 | O_RDWR | O_CREAT | O_TRUNC, | |
287 | S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP); | |
288 | g_string_free(filename, TRUE); | |
289 | return fd; | |
290 | } | |
291 | ||
292 | static | |
293 | void stream_flush_cb(struct bt_ctf_stream *stream, struct bt_ctf_writer *writer) | |
294 | { | |
295 | struct bt_ctf_field *stream_id; | |
296 | ||
297 | /* Start a new packet in the stream */ | |
298 | if (stream->flushed_packet_count) { | |
299 | /* ctf_init_pos has already initialized the first packet */ | |
300 | ctf_packet_seek(&stream->pos.parent, 0, SEEK_CUR); | |
301 | } | |
302 | ||
303 | stream_id = bt_ctf_field_structure_get_field( | |
bc37ae52 | 304 | writer->trace->trace_packet_header, "stream_id"); |
c13c7ca0 | 305 | bt_ctf_field_unsigned_integer_set_value(stream_id, stream->stream_class->id); |
273b65be JG |
306 | bt_ctf_field_put(stream_id); |
307 | ||
308 | /* Write the trace_packet_header */ | |
bc37ae52 | 309 | bt_ctf_field_serialize(writer->trace->trace_packet_header, &stream->pos); |
273b65be | 310 | } |