* SOFTWARE.
*/
+#define BT_COMP_LOG_SELF_COMP self_comp
+#define BT_LOG_OUTPUT_LEVEL log_level
#define BT_LOG_TAG "PLUGIN/SRC.CTF.LTTNG-LIVE/DS"
-#include "logging.h"
+#include "logging/comp-logging.h"
+#include <inttypes.h>
#include <stdio.h>
-#include <stdint.h>
#include <stdlib.h>
-#include <stdbool.h>
+
#include <glib.h>
-#include <inttypes.h>
-#include "compat/mman.h"
+
#include <babeltrace2/babeltrace.h>
+
#include "../common/msg-iter/msg-iter.h"
#include "common/assert.h"
-
+#include "compat/mman.h"
#include "data-stream.h"
#define STREAM_NAME_PREFIX "stream-"
static
-enum bt_msg_iter_medium_status medop_request_bytes(
+enum ctf_msg_iter_medium_status medop_request_bytes(
size_t request_sz, uint8_t **buffer_addr,
size_t *buffer_sz, void *data)
{
- enum bt_msg_iter_medium_status status =
- BT_MSG_ITER_MEDIUM_STATUS_OK;
+ enum ctf_msg_iter_medium_status status =
+ CTF_MSG_ITER_MEDIUM_STATUS_OK;
struct lttng_live_stream_iterator *stream = data;
struct lttng_live_trace *trace = stream->trace;
struct lttng_live_session *session = trace->session;
uint64_t len_left;
uint64_t read_len;
+ BT_ASSERT(request_sz);
+
+ if (stream->has_stream_hung_up) {
+ status = CTF_MSG_ITER_MEDIUM_STATUS_EOF;
+ goto end;
+ }
+
len_left = stream->base_offset + stream->len - stream->offset;
if (!len_left) {
stream->state = LTTNG_LIVE_STREAM_ACTIVE_NO_DATA;
- status = BT_MSG_ITER_MEDIUM_STATUS_AGAIN;
- return status;
+ status = CTF_MSG_ITER_MEDIUM_STATUS_AGAIN;
+ goto end;
}
+
read_len = MIN(request_sz, stream->buflen);
read_len = MIN(read_len, len_left);
status = lttng_live_get_stream_bytes(live_msg_iter,
*buffer_addr = stream->buf;
*buffer_sz = recv_len;
stream->offset += recv_len;
+end:
return status;
}
int64_t stream_id, void *data)
{
struct lttng_live_stream_iterator *lttng_live_stream = data;
+ bt_logging_level log_level = lttng_live_stream->log_level;
+ bt_self_component *self_comp = lttng_live_stream->self_comp;
if (!lttng_live_stream->stream) {
uint64_t stream_class_id = bt_stream_class_get_id(stream_class);
- BT_LOGI("Creating stream %s (ID: %" PRIu64 ") out of stream "
+ BT_COMP_LOGI("Creating stream %s (ID: %" PRIu64 ") out of stream "
"class %" PRId64, lttng_live_stream->name->str,
stream_id, stream_class_id);
}
if (!lttng_live_stream->stream) {
- BT_LOGE("Cannot create stream %s (stream class ID "
+ BT_COMP_LOGE_APPEND_CAUSE(self_comp,
+ "Cannot create stream %s (stream class ID "
"%" PRId64 ", stream ID %" PRIu64 ")",
lttng_live_stream->name->str,
stream_class_id, stream_id);
+ goto end;
}
+
bt_stream_set_name(lttng_live_stream->stream,
lttng_live_stream->name->str);
}
+end:
return lttng_live_stream->stream;
}
-static struct bt_msg_iter_medium_ops medops = {
+static struct ctf_msg_iter_medium_ops medops = {
.request_bytes = medop_request_bytes,
.seek = NULL,
.borrow_stream = medop_borrow_stream,
BT_HIDDEN
enum lttng_live_iterator_status lttng_live_lazy_msg_init(
- struct lttng_live_session *session)
+ struct lttng_live_session *session,
+ bt_self_message_iterator *self_msg_iter)
{
struct lttng_live_component *lttng_live =
session->lttng_live_msg_iter->lttng_live_comp;
uint64_t trace_idx, stream_iter_idx;
+ bt_logging_level log_level = session->log_level;
+ bt_self_component *self_comp = session->self_comp;
if (!session->lazy_stream_msg_init) {
return LTTNG_LIVE_ITERATOR_STATUS_OK;
}
+ BT_COMP_LOGD("Lazily initializing self message iterator for live session: "
+ "session-id=%"PRIu64", self-msg-iter-addr=%p", session->id,
+ self_msg_iter);
+
for (trace_idx = 0; trace_idx < session->traces->len; trace_idx++) {
struct lttng_live_trace *trace =
g_ptr_array_index(session->traces, trace_idx);
struct ctf_trace_class *ctf_tc;
struct lttng_live_stream_iterator *stream_iter =
g_ptr_array_index(trace->stream_iterators,
- stream_iter_idx);
+ stream_iter_idx);
if (stream_iter->msg_iter) {
continue;
}
ctf_tc = ctf_metadata_decoder_borrow_ctf_trace_class(
- trace->metadata->decoder);
- stream_iter->msg_iter = bt_msg_iter_create(ctf_tc,
- lttng_live->max_query_size, medops,
- stream_iter, BT_LOG_OUTPUT_LEVEL);
+ trace->metadata->decoder);
+ BT_COMP_LOGD("Creating CTF message iterator: "
+ "session-id=%"PRIu64", ctf-tc-addr=%p, "
+ "stream-iter-name=%s, self-msg-iter-addr=%p",
+ session->id, ctf_tc, stream_iter->name->str, self_msg_iter);
+ stream_iter->msg_iter = ctf_msg_iter_create(ctf_tc,
+ lttng_live->max_query_size, medops, stream_iter,
+ log_level, self_comp, self_msg_iter);
if (!stream_iter->msg_iter) {
+ BT_COMP_LOGE_APPEND_CAUSE(self_comp,
+ "Failed to create CTF message iterator");
goto error;
}
+
+ ctf_msg_iter_set_emit_stream_end_message(
+ stream_iter->msg_iter, true);
+ ctf_msg_iter_set_emit_stream_beginning_message(
+ stream_iter->msg_iter, true);
}
}
struct lttng_live_stream_iterator *lttng_live_stream_iterator_create(
struct lttng_live_session *session,
uint64_t ctf_trace_id,
- uint64_t stream_id)
+ uint64_t stream_id,
+ bt_self_message_iterator *self_msg_iter)
{
struct lttng_live_stream_iterator *stream_iter;
struct lttng_live_component *lttng_live;
struct lttng_live_trace *trace;
+ bt_logging_level log_level;
+ bt_self_component *self_comp;
BT_ASSERT(session);
BT_ASSERT(session->lttng_live_msg_iter);
BT_ASSERT(session->lttng_live_msg_iter->lttng_live_comp);
+ log_level = session->log_level;
+ self_comp = session->self_comp;
lttng_live = session->lttng_live_msg_iter->lttng_live_comp;
stream_iter = g_new0(struct lttng_live_stream_iterator, 1);
if (!stream_iter) {
+ BT_COMP_LOGE_APPEND_CAUSE(self_comp,
+ "Failed to allocate struct lttng_live_stream_iterator");
goto error;
}
- trace = lttng_live_borrow_trace(session, ctf_trace_id);
+ stream_iter->log_level = log_level;
+ stream_iter->self_comp = self_comp;
+ trace = lttng_live_session_borrow_or_create_trace_by_id(session, ctf_trace_id);
if (!trace) {
+ BT_COMP_LOGE_APPEND_CAUSE(self_comp,
+ "Failed to borrow CTF trace.");
goto error;
}
if (trace->trace) {
struct ctf_trace_class *ctf_tc =
ctf_metadata_decoder_borrow_ctf_trace_class(
- trace->metadata->decoder);
+ trace->metadata->decoder);
BT_ASSERT(!stream_iter->msg_iter);
- stream_iter->msg_iter = bt_msg_iter_create(ctf_tc,
- lttng_live->max_query_size, medops,
- stream_iter, BT_LOG_OUTPUT_LEVEL);
+ stream_iter->msg_iter = ctf_msg_iter_create(ctf_tc,
+ lttng_live->max_query_size, medops, stream_iter,
+ log_level, self_comp, self_msg_iter);
if (!stream_iter->msg_iter) {
+ BT_COMP_LOGE_APPEND_CAUSE(self_comp,
+ "Failed to create CTF message iterator");
goto error;
}
+
+ ctf_msg_iter_set_emit_stream_end_message(
+ stream_iter->msg_iter, true);
+ ctf_msg_iter_set_emit_stream_beginning_message(
+ stream_iter->msg_iter, true);
}
stream_iter->buf = g_new0(uint8_t, lttng_live->max_query_size);
if (!stream_iter->buf) {
+ BT_COMP_LOGE_APPEND_CAUSE(self_comp,
+ "Failed to allocate live stream iterator buffer");
goto error;
}
stream_iter->buflen = lttng_live->max_query_size;
stream_iter->name = g_string_new(NULL);
if (!stream_iter->name) {
+ BT_COMP_LOGE_APPEND_CAUSE(self_comp,
+ "Failed to allocate live stream iterator name buffer");
goto error;
}
g_string_printf(stream_iter->name, STREAM_NAME_PREFIX "%" PRIu64,
- stream_iter->viewer_stream_id);
+ stream_iter->viewer_stream_id);
g_ptr_array_add(trace->stream_iterators, stream_iter);
/* Track the number of active stream iterator. */
}
if (stream_iter->msg_iter) {
- bt_msg_iter_destroy(stream_iter->msg_iter);
- }
- if (stream_iter->buf) {
- g_free(stream_iter->buf);
+ ctf_msg_iter_destroy(stream_iter->msg_iter);
}
+ g_free(stream_iter->buf);
if (stream_iter->name) {
g_string_free(stream_iter->name, TRUE);
}
/* Track the number of active stream iterator. */
stream_iter->trace->session->lttng_live_msg_iter->active_stream_iter--;
- /*
- * Ensure we poke the trace metadata in the future, which is
- * required to release the metadata reference on the trace.
- */
- stream_iter->trace->new_metadata_needed = true;
g_free(stream_iter);
}