2 * Copyright 2016 - Philippe Proulx <pproulx@efficios.com>
3 * Copyright 2010-2011 - EfficiOS Inc. and Linux Foundation
5 * Some functions are based on older functions written by Mathieu Desnoyers.
7 * Permission is hereby granted, free of charge, to any person obtaining a copy
8 * of this software and associated documentation files (the "Software"), to deal
9 * in the Software without restriction, including without limitation the rights
10 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11 * copies of the Software, and to permit persons to whom the Software is
12 * furnished to do so, subject to the following conditions:
14 * The above copyright notice and this permission notice shall be included in
15 * all copies or substantial portions of the Software.
17 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
20 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
26 #define BT_LOG_TAG "PLUGIN-CTF-LTTNG-LIVE-SRC-METADATA"
34 #include <babeltrace/compat/uuid-internal.h>
35 #include <babeltrace/compat/memstream-internal.h>
36 #include <babeltrace/graph/graph.h>
39 #include "../common/metadata/decoder.h"
41 #define TSDL_MAGIC 0x75d11d57
43 struct packet_header
{
47 uint32_t content_size
;
49 uint8_t compression_scheme
;
50 uint8_t encryption_scheme
;
51 uint8_t checksum_scheme
;
54 } __attribute__((__packed__
));
57 enum bt_ctf_lttng_live_iterator_status
lttng_live_update_clock_map(
58 struct lttng_live_trace
*trace
)
60 enum bt_ctf_lttng_live_iterator_status status
=
61 BT_CTF_LTTNG_LIVE_ITERATOR_STATUS_OK
;
65 BT_PUT(trace
->cc_prio_map
);
66 trace
->cc_prio_map
= bt_clock_class_priority_map_create();
67 if (!trace
->cc_prio_map
) {
71 count
= bt_ctf_trace_get_clock_class_count(trace
->trace
);
74 for (i
= 0; i
< count
; i
++) {
75 struct bt_ctf_clock_class
*clock_class
=
76 bt_ctf_trace_get_clock_class_by_index(trace
->trace
, i
);
79 ret
= bt_clock_class_priority_map_add_clock_class(
80 trace
->cc_prio_map
, clock_class
, 0);
90 status
= BT_CTF_LTTNG_LIVE_ITERATOR_STATUS_ERROR
;
96 enum bt_ctf_lttng_live_iterator_status
lttng_live_metadata_update(
97 struct lttng_live_trace
*trace
)
99 struct lttng_live_session
*session
= trace
->session
;
100 struct lttng_live_metadata
*metadata
= trace
->metadata
;
102 size_t size
, len_read
= 0;
103 char *metadata_buf
= NULL
;
105 enum ctf_metadata_decoder_status decoder_status
;
106 enum bt_ctf_lttng_live_iterator_status status
=
107 BT_CTF_LTTNG_LIVE_ITERATOR_STATUS_OK
;
109 /* No metadata stream yet. */
111 if (session
->new_streams_needed
) {
112 status
= BT_CTF_LTTNG_LIVE_ITERATOR_STATUS_AGAIN
;
114 session
->new_streams_needed
= true;
115 status
= BT_CTF_LTTNG_LIVE_ITERATOR_STATUS_CONTINUE
;
120 if (!trace
->new_metadata_needed
) {
124 /* Open for writing */
125 fp
= bt_open_memstream(&metadata_buf
, &size
);
127 BT_LOGE("Metadata open_memstream: %s", strerror(errno
));
131 /* Grab all available metadata. */
134 * get_one_metadata_packet returns the number of bytes
135 * received, 0 when we have received everything, a
136 * negative value on error.
138 ret
= lttng_live_get_one_metadata_packet(trace
, fp
);
145 * Consider metadata closed as soon as we get an error reading
146 * it (e.g. cannot be found).
149 if (!metadata
->closed
) {
150 metadata
->closed
= true;
152 * Release our reference on the trace as soon as
153 * we know the metadata stream is not available
154 * anymore. This won't necessarily teardown the
155 * metadata objects immediately, but only when
156 * the data streams are done.
158 lttng_live_unref_trace(metadata
->trace
);
160 if (errno
== EINTR
) {
161 if (bt_graph_is_canceled(session
->lttng_live
->graph
)) {
162 status
= BT_CTF_LTTNG_LIVE_ITERATOR_STATUS_AGAIN
;
168 if (bt_close_memstream(&metadata_buf
, &size
, fp
)) {
169 BT_LOGE("bt_close_memstream: %s", strerror(errno
));
176 status
= BT_CTF_LTTNG_LIVE_ITERATOR_STATUS_AGAIN
;
179 trace
->new_metadata_needed
= false;
183 if (babeltrace_debug
) {
187 fp
= bt_fmemopen(metadata_buf
, len_read
, "rb");
189 BT_LOGE("Cannot memory-open metadata buffer: %s",
194 decoder_status
= ctf_metadata_decoder_decode(metadata
->decoder
, fp
);
195 switch (decoder_status
) {
196 case CTF_METADATA_DECODER_STATUS_OK
:
197 BT_PUT(trace
->trace
);
198 trace
->trace
= ctf_metadata_decoder_get_trace(metadata
->decoder
);
199 trace
->new_metadata_needed
= false;
200 status
= lttng_live_update_clock_map(trace
);
201 if (status
!= BT_CTF_LTTNG_LIVE_ITERATOR_STATUS_OK
) {
205 case CTF_METADATA_DECODER_STATUS_INCOMPLETE
:
206 status
= BT_CTF_LTTNG_LIVE_ITERATOR_STATUS_AGAIN
;
208 case CTF_METADATA_DECODER_STATUS_ERROR
:
209 case CTF_METADATA_DECODER_STATUS_INVAL_VERSION
:
210 case CTF_METADATA_DECODER_STATUS_IR_VISITOR_ERROR
:
216 status
= BT_CTF_LTTNG_LIVE_ITERATOR_STATUS_ERROR
;
221 closeret
= fclose(fp
);
223 BT_LOGE("Error on fclose");
230 int lttng_live_metadata_create_stream(struct lttng_live_session
*session
,
231 uint64_t ctf_trace_id
,
234 struct lttng_live_metadata
*metadata
= NULL
;
235 struct lttng_live_trace
*trace
;
237 metadata
= g_new0(struct lttng_live_metadata
, 1);
241 metadata
->stream_id
= stream_id
;
242 //TODO: add clock offset option
243 //TODO: add (preferably unique) trace's name
244 metadata
->decoder
= ctf_metadata_decoder_create(stderr
, 0,
246 if (!metadata
->decoder
) {
249 trace
= lttng_live_ref_trace(session
, ctf_trace_id
);
253 metadata
->trace
= trace
;
254 trace
->metadata
= metadata
;
258 ctf_metadata_decoder_destroy(metadata
->decoder
);
264 void lttng_live_metadata_fini(struct lttng_live_trace
*trace
)
266 struct lttng_live_metadata
*metadata
= trace
->metadata
;
271 if (metadata
->text
) {
272 free(metadata
->text
);
274 ctf_metadata_decoder_destroy(metadata
->decoder
);
275 trace
->metadata
= NULL
;
276 lttng_live_unref_trace(trace
);
277 if (!metadata
->closed
) {
278 lttng_live_unref_trace(metadata
->trace
);