#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
+#include <inttypes.h>
#include <common/common.h>
#include <common/defaults.h>
ret = 0;
} else {
ret = -reply.ret_code;
- DBG("Consumer ret code %d", reply.ret_code);
+ DBG("Consumer ret code %d", ret);
}
end:
return ret;
}
+/*
+ * Once the ASK_CHANNEL command is sent to the consumer, the channel
+ * information are sent back. This call receives that data and populates key
+ * and stream_count.
+ *
+ * On success return 0 and both key and stream_count are set. On error, a
+ * negative value is sent back and both parameters are untouched.
+ */
+int consumer_recv_status_channel(struct consumer_socket *sock,
+ uint64_t *key, unsigned int *stream_count)
+{
+ int ret;
+ struct lttcomm_consumer_status_channel reply;
+
+ assert(sock);
+ assert(stream_count);
+ assert(key);
+
+ ret = lttcomm_recv_unix_sock(sock->fd, &reply, sizeof(reply));
+ if (ret <= 0) {
+ if (ret == 0) {
+ /* Orderly shutdown. Don't return 0 which means success. */
+ ret = -1;
+ }
+ /* The above call will print a PERROR on error. */
+ DBG("Fail to receive status reply on sock %d", sock->fd);
+ goto end;
+ }
+
+ /* An error is possible so don't touch the key and stream_count. */
+ if (reply.ret_code != LTTNG_OK) {
+ ret = -1;
+ goto end;
+ }
+
+ *key = reply.key;
+ *stream_count = reply.stream_count;
+
+end:
+ return ret;
+}
+
/*
* Send destroy relayd command to consumer.
*
assert(consumer);
assert(sock);
- DBG2("Sending destroy relayd command to consumer...");
+ DBG2("Sending destroy relayd command to consumer sock %d", sock->fd);
/* Bail out if consumer is disabled */
if (!consumer->enabled) {
/* By default, consumer output is enabled */
output->enabled = 1;
output->type = type;
- output->net_seq_index = -1;
+ output->net_seq_index = (uint64_t) -1ULL;
output->socks = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
return ret;
}
+/*
+ * Consumer send communication message structure to consumer.
+ */
+int consumer_send_msg(struct consumer_socket *sock,
+ struct lttcomm_consumer_msg *msg)
+{
+ int ret;
+
+ assert(msg);
+ assert(sock);
+ assert(sock->fd >= 0);
+
+ ret = lttcomm_send_unix_sock(sock->fd, msg,
+ sizeof(struct lttcomm_consumer_msg));
+ if (ret < 0) {
+ /* The above call will print a PERROR on error. */
+ DBG("Error when sending consumer channel on sock %d", sock->fd);
+ goto error;
+ }
+
+ ret = consumer_recv_status_reply(sock);
+
+error:
+ return ret;
+}
+
/*
* Consumer send channel communication message structure to consumer.
*/
return ret;
}
+/*
+ * Populate the given consumer msg structure with the ask_channel command
+ * information.
+ */
+void consumer_init_ask_channel_comm_msg(struct lttcomm_consumer_msg *msg,
+ uint64_t subbuf_size,
+ uint64_t num_subbuf,
+ int overwrite,
+ unsigned int switch_timer_interval,
+ unsigned int read_timer_interval,
+ int output,
+ int type,
+ uint64_t session_id,
+ const char *pathname,
+ const char *name,
+ uid_t uid,
+ gid_t gid,
+ uint64_t relayd_id,
+ uint64_t key,
+ unsigned char *uuid)
+{
+ assert(msg);
+
+ /* Zeroed structure */
+ memset(msg, 0, sizeof(struct lttcomm_consumer_msg));
+
+ msg->cmd_type = LTTNG_CONSUMER_ASK_CHANNEL_CREATION;
+ msg->u.ask_channel.subbuf_size = subbuf_size;
+ msg->u.ask_channel.num_subbuf = num_subbuf ;
+ msg->u.ask_channel.overwrite = overwrite;
+ msg->u.ask_channel.switch_timer_interval = switch_timer_interval;
+ msg->u.ask_channel.read_timer_interval = read_timer_interval;
+ msg->u.ask_channel.output = output;
+ msg->u.ask_channel.type = type;
+ msg->u.ask_channel.session_id = session_id;
+ msg->u.ask_channel.uid = uid;
+ msg->u.ask_channel.gid = gid;
+ msg->u.ask_channel.relayd_id = relayd_id;
+ msg->u.ask_channel.key = key;
+
+ memcpy(msg->u.ask_channel.uuid, uuid, sizeof(msg->u.ask_channel.uuid));
+
+ strncpy(msg->u.ask_channel.pathname, pathname,
+ sizeof(msg->u.ask_channel.pathname));
+ msg->u.ask_channel.pathname[sizeof(msg->u.ask_channel.pathname)-1] = '\0';
+
+ strncpy(msg->u.ask_channel.name, name, sizeof(msg->u.ask_channel.name));
+ msg->u.ask_channel.name[sizeof(msg->u.ask_channel.name) - 1] = '\0';
+}
+
/*
* Init channel communication message structure.
*/
void consumer_init_channel_comm_msg(struct lttcomm_consumer_msg *msg,
enum lttng_consumer_command cmd,
- int channel_key,
- uint64_t max_sb_size,
- uint64_t mmap_len,
+ uint64_t channel_key,
+ uint64_t session_id,
+ const char *pathname,
+ uid_t uid,
+ gid_t gid,
+ uint64_t relayd_id,
const char *name,
- unsigned int nb_init_streams)
+ unsigned int nb_init_streams,
+ enum lttng_event_output output,
+ int type)
{
assert(msg);
- /* TODO: Args validation */
-
/* Zeroed structure */
memset(msg, 0, sizeof(struct lttcomm_consumer_msg));
/* Send channel */
msg->cmd_type = cmd;
msg->u.channel.channel_key = channel_key;
- msg->u.channel.max_sb_size = max_sb_size;
- msg->u.channel.mmap_len = mmap_len;
+ msg->u.channel.session_id = session_id;
+ msg->u.channel.uid = uid;
+ msg->u.channel.gid = gid;
+ msg->u.channel.relayd_id = relayd_id;
msg->u.channel.nb_init_streams = nb_init_streams;
+ msg->u.channel.output = output;
+ msg->u.channel.type = type;
+
+ strncpy(msg->u.channel.pathname, pathname,
+ sizeof(msg->u.channel.pathname));
+ msg->u.channel.pathname[sizeof(msg->u.channel.pathname) - 1] = '\0';
+
+ strncpy(msg->u.channel.name, name, sizeof(msg->u.channel.name));
+ msg->u.channel.name[sizeof(msg->u.channel.name) - 1] = '\0';
}
/*
*/
void consumer_init_stream_comm_msg(struct lttcomm_consumer_msg *msg,
enum lttng_consumer_command cmd,
- int channel_key,
- int stream_key,
- uint32_t state,
- enum lttng_event_output output,
- uint64_t mmap_len,
- uid_t uid,
- gid_t gid,
- int net_index,
- unsigned int metadata_flag,
- const char *name,
- const char *pathname,
- unsigned int session_id)
+ uint64_t channel_key,
+ uint64_t stream_key,
+ int cpu)
{
assert(msg);
memset(msg, 0, sizeof(struct lttcomm_consumer_msg));
- /* TODO: Args validation */
-
msg->cmd_type = cmd;
msg->u.stream.channel_key = channel_key;
msg->u.stream.stream_key = stream_key;
- msg->u.stream.state = state;
- msg->u.stream.output = output;
- msg->u.stream.mmap_len = mmap_len;
- msg->u.stream.uid = uid;
- msg->u.stream.gid = gid;
- msg->u.stream.net_index = net_index;
- msg->u.stream.metadata_flag = metadata_flag;
- msg->u.stream.session_id = (uint64_t) session_id;
- strncpy(msg->u.stream.name, name, sizeof(msg->u.stream.name));
- msg->u.stream.name[sizeof(msg->u.stream.name) - 1] = '\0';
- strncpy(msg->u.stream.path_name, pathname,
- sizeof(msg->u.stream.path_name));
- msg->u.stream.path_name[sizeof(msg->u.stream.path_name) - 1] = '\0';
+ msg->u.stream.cpu = cpu;
}
/*
assert(msg);
assert(dst);
assert(sock);
-
- switch (dst->type) {
- case CONSUMER_DST_NET:
- /* Consumer should send the stream on the network. */
- msg->u.stream.net_index = dst->net_seq_index;
- break;
- case CONSUMER_DST_LOCAL:
- /* Add stream file name to stream path */
- strncat(msg->u.stream.path_name, "/",
- sizeof(msg->u.stream.path_name) -
- strlen(msg->u.stream.path_name) - 1);
- strncat(msg->u.stream.path_name, msg->u.stream.name,
- sizeof(msg->u.stream.path_name) -
- strlen(msg->u.stream.path_name) - 1);
- msg->u.stream.path_name[sizeof(msg->u.stream.path_name) - 1] = '\0';
- /* Indicate that the stream is NOT network */
- msg->u.stream.net_index = -1;
- break;
- default:
- ERR("Consumer unknown output type (%d)", dst->type);
- ret = -1;
- goto error;
- }
+ assert(fds);
/* Send on socket */
ret = lttcomm_send_unix_sock(sock->fd, msg,
*/
int consumer_send_relayd_socket(struct consumer_socket *consumer_sock,
struct lttcomm_sock *sock, struct consumer_output *consumer,
- enum lttng_stream_type type, unsigned int session_id)
+ enum lttng_stream_type type, uint64_t session_id)
{
int ret;
struct lttcomm_consumer_msg msg;
* This function has a different behavior with the consumer i.e. that it waits
* for a reply from the consumer if yes or no the data is pending.
*/
-int consumer_is_data_pending(unsigned int id,
+int consumer_is_data_pending(uint64_t session_id,
struct consumer_output *consumer)
{
int ret;
msg.cmd_type = LTTNG_CONSUMER_DATA_PENDING;
- msg.u.data_pending.session_id = (uint64_t) id;
+ msg.u.data_pending.session_id = session_id;
- DBG3("Consumer data pending for id %u", id);
+ DBG3("Consumer data pending for id %" PRIu64, session_id);
/* Send command for each consumer */
rcu_read_lock();
}
rcu_read_unlock();
- DBG("Consumer data is %s pending for session id %u",
- ret_code == 1 ? "" : "NOT", id);
+ DBG("Consumer data is %s pending for session id %" PRIu64,
+ ret_code == 1 ? "" : "NOT", session_id);
return ret_code;
error_unlock:
projects / lttng-tools.git / blobdiff