* Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
-#define _GNU_SOURCE
#include <assert.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
-#include <time.h>
+#include <common/compat/time.h>
#include <sys/types.h>
#include <inttypes.h>
#include <stdlib.h>
#include <lttng/lttng.h>
#include <urcu/list.h>
-#include <bin/lttng-sessiond/session.h>
#include <common/common.h>
-#include <bin/lttng-relayd/lttng-viewer.h>
+#include <bin/lttng-relayd/lttng-viewer-abi.h>
#include <common/index/ctf-index.h>
+#include <common/compat/endian.h>
+
#define SESSION1 "test1"
#define RELAYD_URL "net://localhost"
#define LIVE_TIMER 2000000
/* Number of TAP tests in this file */
-#define NUM_TESTS 7
+#define NUM_TESTS 11
#define mmap_size 524288
-int ust_consumerd32_fd;
-int ust_consumerd64_fd;
-
static int control_sock;
struct live_session *session;
static int first_packet_offset;
static int first_packet_len;
-static int first_packet_stream_id;
+static int first_packet_stream_id = -1;
struct viewer_stream {
uint64_t id;
uint64_t stream_count;
};
+static
+ssize_t lttng_live_recv(int fd, void *buf, size_t len)
+{
+ ssize_t ret;
+ size_t copied = 0, to_copy = len;
+
+ do {
+ ret = recv(fd, buf + copied, to_copy, 0);
+ if (ret > 0) {
+ assert(ret <= to_copy);
+ copied += ret;
+ to_copy -= ret;
+ }
+ } while ((ret > 0 && to_copy > 0)
+ || (ret < 0 && errno == EINTR));
+ if (ret > 0)
+ ret = copied;
+ /* ret = 0 means orderly shutdown, ret < 0 is error. */
+ return ret;
+}
+
+static
+ssize_t lttng_live_send(int fd, const void *buf, size_t len)
+{
+ ssize_t ret;
+
+ do {
+ ret = send(fd, buf, len, MSG_NOSIGNAL);
+ } while (ret < 0 && errno == EINTR);
+ return ret;
+}
+
static
int connect_viewer(char *hostname)
{
}
if ((control_sock = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
- perror("Socket");
+ PERROR("Socket");
ret = -1;
goto end;
}
if (connect(control_sock, (struct sockaddr *) &server_addr,
sizeof(struct sockaddr)) == -1) {
- perror("Connect");
+ PERROR("Connect");
ret = -1;
goto end;
}
{
struct lttng_viewer_cmd cmd;
struct lttng_viewer_connect connect;
- int ret;
+ ssize_t ret_len;
- cmd.cmd = htobe32(VIEWER_CONNECT);
- cmd.data_size = sizeof(connect);
- cmd.cmd_version = 0;
+ cmd.cmd = htobe32(LTTNG_VIEWER_CONNECT);
+ cmd.data_size = htobe64(sizeof(connect));
+ cmd.cmd_version = htobe32(0);
memset(&connect, 0, sizeof(connect));
connect.major = htobe32(VERSION_MAJOR);
connect.minor = htobe32(VERSION_MINOR);
- connect.type = htobe32(VIEWER_CLIENT_COMMAND);
+ connect.type = htobe32(LTTNG_VIEWER_CLIENT_COMMAND);
- do {
- ret = send(control_sock, &cmd, sizeof(cmd), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error sending cmd\n");
+ ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
+ if (ret_len < 0) {
+ diag("Error sending cmd");
goto error;
}
- do {
- ret = send(control_sock, &connect, sizeof(connect), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error sending version\n");
+ ret_len = lttng_live_send(control_sock, &connect, sizeof(connect));
+ if (ret_len < 0) {
+ diag("Error sending version");
goto error;
}
- do {
- ret = recv(control_sock, &connect, sizeof(connect), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error receiving version\n");
+ ret_len = lttng_live_recv(control_sock, &connect, sizeof(connect));
+ if (ret_len == 0) {
+ diag("[error] Remote side has closed connection");
goto error;
}
- ret = 0;
+ if (ret_len < 0) {
+ diag("Error receiving version");
+ goto error;
+ }
+ return 0;
error:
- return ret;
+ return -1;
}
/*
* Returns the number of sessions, should be 1 during the unit test.
*/
-int list_sessions(int *session_id)
+int list_sessions(uint64_t *session_id)
{
struct lttng_viewer_cmd cmd;
struct lttng_viewer_list_sessions list;
struct lttng_viewer_session lsession;
- int i, ret;
+ int i;
+ ssize_t ret_len;
int first_session = 0;
- cmd.cmd = htobe32(VIEWER_LIST_SESSIONS);
- cmd.data_size = 0;
- cmd.cmd_version = 0;
+ cmd.cmd = htobe32(LTTNG_VIEWER_LIST_SESSIONS);
+ cmd.data_size = htobe64(0);
+ cmd.cmd_version = htobe32(0);
- do {
- ret = send(control_sock, &cmd, sizeof(cmd), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error sending cmd\n");
+ ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
+ if (ret_len < 0) {
+ diag("Error sending cmd");
goto error;
}
- do {
- ret = recv(control_sock, &list, sizeof(list), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error receiving session list\n");
+ ret_len = lttng_live_recv(control_sock, &list, sizeof(list));
+ if (ret_len == 0) {
+ diag("[error] Remote side has closed connection");
+ goto error;
+ }
+ if (ret_len < 0) {
+ diag("Error receiving session list");
goto error;
}
for (i = 0; i < be32toh(list.sessions_count); i++) {
- do {
- ret = recv(control_sock, &lsession, sizeof(lsession), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error receiving session\n");
+ ret_len = lttng_live_recv(control_sock, &lsession, sizeof(lsession));
+ if (ret_len < 0) {
+ diag("Error receiving session");
goto error;
}
if (lsession.streams > 0 && first_session <= 0) {
}
}
- ret = be32toh(list.sessions_count);
+ return be32toh(list.sessions_count);
error:
- return ret;
+ return -1;
+}
+
+int create_viewer_session(void)
+{
+ struct lttng_viewer_cmd cmd;
+ struct lttng_viewer_create_session_response resp;
+ ssize_t ret_len;
+
+ cmd.cmd = htobe32(LTTNG_VIEWER_CREATE_SESSION);
+ cmd.data_size = htobe64(0);
+ cmd.cmd_version = htobe32(0);
+
+ ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
+ if (ret_len < 0) {
+ diag("[error] Error sending cmd");
+ goto error;
+ }
+ assert(ret_len == sizeof(cmd));
+
+ ret_len = lttng_live_recv(control_sock, &resp, sizeof(resp));
+ if (ret_len == 0) {
+ diag("[error] Remote side has closed connection");
+ goto error;
+ }
+ if (ret_len < 0) {
+ diag("[error] Error receiving create session reply");
+ goto error;
+ }
+ assert(ret_len == sizeof(resp));
+
+ if (be32toh(resp.status) != LTTNG_VIEWER_CREATE_SESSION_OK) {
+ diag("[error] Error creating viewer session");
+ goto error;
+ }
+ return 0;
+
+error:
+ return -1;
}
-int attach_session(int id)
+int attach_session(uint64_t id)
{
struct lttng_viewer_cmd cmd;
struct lttng_viewer_attach_session_request rq;
struct lttng_viewer_attach_session_response rp;
struct lttng_viewer_stream stream;
- int ret, i;
+ int i;
+ ssize_t ret_len;
session = zmalloc(sizeof(struct live_session));
if (!session) {
- ret = -1;
goto error;
}
- cmd.cmd = htobe32(VIEWER_ATTACH_SESSION);
- cmd.data_size = sizeof(rq);
- cmd.cmd_version = 0;
+ cmd.cmd = htobe32(LTTNG_VIEWER_ATTACH_SESSION);
+ cmd.data_size = htobe64(sizeof(rq));
+ cmd.cmd_version = htobe32(0);
memset(&rq, 0, sizeof(rq));
rq.session_id = htobe64(id);
- rq.seek = htobe32(VIEWER_SEEK_BEGINNING);
+ rq.seek = htobe32(LTTNG_VIEWER_SEEK_BEGINNING);
- do {
- ret = send(control_sock, &cmd, sizeof(cmd), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error sending cmd\n");
+ ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
+ if (ret_len < 0) {
+ diag("Error sending cmd LTTNG_VIEWER_ATTACH_SESSION");
goto error;
}
- do {
- ret = send(control_sock, &rq, sizeof(rq), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error sending attach request\n");
+ ret_len = lttng_live_send(control_sock, &rq, sizeof(rq));
+ if (ret_len < 0) {
+ diag("Error sending attach request");
goto error;
}
- do {
- ret = recv(control_sock, &rp, sizeof(rp), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error receiving attach response\n");
+ ret_len = lttng_live_recv(control_sock, &rp, sizeof(rp));
+ if (ret_len == 0) {
+ diag("[error] Remote side has closed connection");
goto error;
}
- if (be32toh(rp.status) != VIEWER_ATTACH_OK) {
- ret = -1;
- goto end;
+ if (ret_len < 0) {
+ diag("Error receiving attach response");
+ goto error;
+ }
+ if (be32toh(rp.status) != LTTNG_VIEWER_ATTACH_OK) {
+ goto error;
}
session->stream_count = be32toh(rp.streams_count);
+ if (session->stream_count == 0) {
+ diag("Got session stream count == 0");
+ goto error;
+ }
session->streams = zmalloc(session->stream_count *
sizeof(struct viewer_stream));
if (!session->streams) {
- ret = -1;
goto error;
}
for (i = 0; i < be32toh(rp.streams_count); i++) {
- do {
- ret = recv(control_sock, &stream, sizeof(stream), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error receiving stream\n");
+ ret_len = lttng_live_recv(control_sock, &stream, sizeof(stream));
+ if (ret_len == 0) {
+ diag("[error] Remote side has closed connection");
+ goto error;
+ }
+ if (ret_len < 0) {
+ diag("Error receiving stream");
goto error;
}
session->streams[i].id = be64toh(stream.id);
session->streams[i].ctf_trace_id = be64toh(stream.ctf_trace_id);
session->streams[i].first_read = 1;
- session->streams[i].mmap_base = mmap(NULL, mmap_size, PROT_READ | PROT_WRITE,
+ session->streams[i].mmap_base = mmap(NULL, mmap_size,
+ PROT_READ | PROT_WRITE,
MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
if (session->streams[i].mmap_base == MAP_FAILED) {
- fprintf(stderr, "mmap error\n");
- ret = -1;
+ diag("mmap error");
goto error;
}
session->streams[i].metadata_flag = 1;
}
}
- ret = session->stream_count;
+ return session->stream_count;
-end:
error:
- return ret;
+ return -1;
}
int get_metadata(void)
struct lttng_viewer_cmd cmd;
struct lttng_viewer_get_metadata rq;
struct lttng_viewer_metadata_packet rp;
+ ssize_t ret_len;
int ret;
uint64_t i;
char *data = NULL;
uint64_t len = 0;
int metadata_stream_id = -1;
- cmd.cmd = htobe32(VIEWER_GET_METADATA);
- cmd.data_size = sizeof(rq);
- cmd.cmd_version = 0;
+ cmd.cmd = htobe32(LTTNG_VIEWER_GET_METADATA);
+ cmd.data_size = htobe64(sizeof(rq));
+ cmd.cmd_version = htobe32(0);
for (i = 0; i < session->stream_count; i++) {
if (session->streams[i].metadata_flag) {
}
if (metadata_stream_id < 0) {
- fprintf(stderr, "No metadata stream found\n");
- ret = -1;
+ diag("No metadata stream found");
goto error;
}
rq.stream_id = htobe64(session->streams[metadata_stream_id].id);
- do {
- ret = send(control_sock, &cmd, sizeof(cmd), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error sending cmd\n");
+retry:
+ ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
+ if (ret_len < 0) {
+ diag("Error sending cmd");
goto error;
}
- do {
- ret = send(control_sock, &rq, sizeof(rq), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error sending get_metadata request\n");
+ ret_len = lttng_live_send(control_sock, &rq, sizeof(rq));
+ if (ret_len < 0) {
+ diag("Error sending get_metadata request");
goto error;
}
- do {
- ret = recv(control_sock, &rp, sizeof(rp), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error receiving metadata response\n");
+ ret_len = lttng_live_recv(control_sock, &rp, sizeof(rp));
+ if (ret_len == 0) {
+ diag("[error] Remote side has closed connection");
+ goto error;
+ }
+ if (ret_len < 0) {
+ diag("Error receiving metadata response");
goto error;
}
switch (be32toh(rp.status)) {
- case VIEWER_METADATA_OK:
- break;
- case VIEWER_NO_NEW_METADATA:
- fprintf(stderr, "NO NEW\n");
- ret = -1;
- goto end;
- case VIEWER_METADATA_ERR:
- fprintf(stderr, "ERR\n");
- ret = -1;
- goto end;
- default:
- fprintf(stderr, "UNKNOWN\n");
- ret = -1;
- goto end;
+ case LTTNG_VIEWER_METADATA_OK:
+ break;
+ case LTTNG_VIEWER_NO_NEW_METADATA:
+ diag("Got LTTNG_VIEWER_NO_NEW_METADATA:");
+ usleep(50);
+ goto retry;
+ case LTTNG_VIEWER_METADATA_ERR:
+ diag("Got LTTNG_VIEWER_METADATA_ERR:");
+ goto error;
+ default:
+ diag("Got unknown status during LTTNG_VIEWER_GET_METADATA");
+ goto error;
}
len = be64toh(rp.len);
if (len <= 0) {
- goto end;
+ goto error;
}
data = zmalloc(len);
if (!data) {
- perror("relay data zmalloc");
+ PERROR("relay data zmalloc");
goto error;
}
- do {
- ret = recv(control_sock, data, len, MSG_WAITALL);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error receiving trace packet\n");
- free(data);
- goto error;
+ ret_len = lttng_live_recv(control_sock, data, len);
+ if (ret_len == 0) {
+ diag("[error] Remote side has closed connection");
+ goto error_free_data;
+ }
+ if (ret_len < 0) {
+ diag("Error receiving trace packet");
+ goto error_free_data;
}
free(data);
+ ret = len;
- ret = (int) len;
-end:
-error:
return ret;
+
+error_free_data:
+ free(data);
+error:
+ return -1;
}
int get_next_index(void)
struct lttng_viewer_cmd cmd;
struct lttng_viewer_get_next_index rq;
struct lttng_viewer_index rp;
- int ret;
+ ssize_t ret_len;
int id;
- cmd.cmd = htobe32(VIEWER_GET_NEXT_INDEX);
- cmd.data_size = sizeof(rq);
- cmd.cmd_version = 0;
+ cmd.cmd = htobe32(LTTNG_VIEWER_GET_NEXT_INDEX);
+ cmd.data_size = htobe64(sizeof(rq));
+ cmd.cmd_version = htobe32(0);
for (id = 0; id < session->stream_count; id++) {
if (session->streams[id].metadata_flag) {
continue;
}
+ memset(&rq, 0, sizeof(rq));
rq.stream_id = htobe64(session->streams[id].id);
retry:
- do {
- ret = send(control_sock, &cmd, sizeof(cmd), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error sending cmd\n");
+ ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
+ if (ret_len < 0) {
+ diag("Error sending cmd");
+ goto error;
+ }
+ ret_len = lttng_live_send(control_sock, &rq, sizeof(rq));
+ if (ret_len < 0) {
+ diag("Error sending get_next_index request");
goto error;
}
- do {
- ret = send(control_sock, &rq, sizeof(rq), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error sending get_next_index request\n");
+ ret_len = lttng_live_recv(control_sock, &rp, sizeof(rp));
+ if (ret_len == 0) {
+ diag("[error] Remote side has closed connection");
goto error;
}
- do {
- ret = recv(control_sock, &rp, sizeof(rp), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error receiving index response\n");
+ if (ret_len < 0) {
+ diag("Error receiving index response");
goto error;
}
rp.flags = be32toh(rp.flags);
switch (be32toh(rp.status)) {
- case VIEWER_INDEX_INACTIVE:
- fprintf(stderr, "(INACTIVE)\n");
- break;
- case VIEWER_INDEX_OK:
- break;
- case VIEWER_INDEX_RETRY:
- sleep(1);
- goto retry;
- case VIEWER_INDEX_HUP:
- fprintf(stderr, "(HUP)\n");
- session->streams[id].id = -1ULL;
- session->streams[id].fd = -1;
- break;
- case VIEWER_INDEX_ERR:
- fprintf(stderr, "(ERR)\n");
- ret = -1;
- goto error;
- default:
- fprintf(stderr, "SHOULD NOT HAPPEN\n");
- ret = -1;
- goto error;
+ case LTTNG_VIEWER_INDEX_INACTIVE:
+ /* Skip this stream. */
+ diag("Got LTTNG_VIEWER_INDEX_INACTIVE");
+ continue;
+ case LTTNG_VIEWER_INDEX_OK:
+ break;
+ case LTTNG_VIEWER_INDEX_RETRY:
+ sleep(1);
+ goto retry;
+ case LTTNG_VIEWER_INDEX_HUP:
+ diag("Got LTTNG_VIEWER_INDEX_HUP");
+ session->streams[id].id = -1ULL;
+ session->streams[id].fd = -1;
+ goto error;
+ case LTTNG_VIEWER_INDEX_ERR:
+ diag("Got LTTNG_VIEWER_INDEX_ERR");
+ goto error;
+ default:
+ diag("Unknown reply status during LTTNG_VIEWER_GET_NEXT_INDEX (%d)", be32toh(rp.status));
+ goto error;
}
- if (!first_packet_stream_id) {
+ if (first_packet_stream_id < 0) {
+ /*
+ * Initialize the first packet stream id. That is,
+ * the first active stream encoutered.
+ */
first_packet_offset = be64toh(rp.offset);
first_packet_len = be64toh(rp.packet_size) / CHAR_BIT;
first_packet_stream_id = id;
+ diag("Got first packet index with offset %d and len %d",
+ first_packet_offset, first_packet_len);
}
}
- ret = 0;
+ return 0;
error:
- return ret;
+ return -1;
}
static
struct lttng_viewer_cmd cmd;
struct lttng_viewer_get_packet rq;
struct lttng_viewer_trace_packet rp;
- int ret;
+ ssize_t ret_len;
- cmd.cmd = htobe32(VIEWER_GET_PACKET);
- cmd.data_size = sizeof(rq);
- cmd.cmd_version = 0;
+ cmd.cmd = htobe32(LTTNG_VIEWER_GET_PACKET);
+ cmd.data_size = htobe64(sizeof(rq));
+ cmd.cmd_version = htobe32(0);
+ memset(&rq, 0, sizeof(rq));
rq.stream_id = htobe64(session->streams[id].id);
/* Already in big endian. */
rq.offset = offset;
rq.len = htobe32(len);
- do {
- ret = send(control_sock, &cmd, sizeof(cmd), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error sending cmd\n");
+ ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
+ if (ret_len < 0) {
+ diag("Error sending cmd");
goto error;
}
- do {
- ret = send(control_sock, &rq, sizeof(rq), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error sending get_data_packet request\n");
+ ret_len = lttng_live_send(control_sock, &rq, sizeof(rq));
+ if (ret_len < 0) {
+ diag("Error sending get_data_packet request");
goto error;
}
- do {
- ret = recv(control_sock, &rp, sizeof(rp), 0);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error receiving data response\n");
+ ret_len = lttng_live_recv(control_sock, &rp, sizeof(rp));
+ if (ret_len == 0) {
+ diag("[error] Remote side has closed connection");
+ goto error;
+ }
+ if (ret_len < 0) {
+ diag("Error receiving data response");
goto error;
}
rp.flags = be32toh(rp.flags);
switch (be32toh(rp.status)) {
- case VIEWER_GET_PACKET_OK:
+ case LTTNG_VIEWER_GET_PACKET_OK:
+ len = be32toh(rp.len);
+ if (len == 0) {
+ diag("Got LTTNG_VIEWER_GET_PACKET_OK, but len == 0");
+ goto error;
+ }
break;
- case VIEWER_GET_PACKET_RETRY:
- fprintf(stderr, "RETRY\n");
- ret = -1;
- goto end;
- case VIEWER_GET_PACKET_ERR:
+ case LTTNG_VIEWER_GET_PACKET_RETRY:
+ diag("Got LTTNG_VIEWER_GET_PACKET_RETRY:");
+ goto error;
+ case LTTNG_VIEWER_GET_PACKET_ERR:
if (rp.flags & LTTNG_VIEWER_FLAG_NEW_METADATA) {
- fprintf(stderr, "NEW_METADATA\n");
- ret = 0;
+ diag("Got LTTNG_VIEWER_GET_PACKET_ERR with NEW_METADATA flag");
goto end;
}
- fprintf(stderr, "ERR\n");
- ret = -1;
- goto end;
+ diag("Got LTTNG_VIEWER_GET_PACKET_ERR:");
+ goto error;
default:
- fprintf(stderr, "UNKNOWN\n");
- ret = -1;
- goto end;
+ diag("Got unknown status code during LTTNG_VIEWER_GET_PACKET");
+ goto error;
}
- len = be32toh(rp.len);
- if (len <= 0) {
- goto end;
+ if (len > mmap_size) {
+ diag("mmap_size not big enough");
+ goto error;
}
- if (len > mmap_size) {
- fprintf(stderr, "mmap_size not big enough\n");
- ret = -1;
+ ret_len = lttng_live_recv(control_sock, session->streams[id].mmap_base, len);
+ if (ret_len == 0) {
+ diag("[error] Remote side has closed connection");
+ goto error;
+ }
+ if (ret_len < 0) {
+ diag("Error receiving trace packet");
goto error;
}
+end:
+ return 0;
+error:
+ return -1;
+}
- do {
- ret = recv(control_sock, session->streams[id].mmap_base, len, MSG_WAITALL);
- } while (ret < 0 && errno == EINTR);
- if (ret < 0) {
- fprintf(stderr, "Error receiving trace packet\n");
+int detach_viewer_session(uint64_t id)
+{
+ struct lttng_viewer_cmd cmd;
+ struct lttng_viewer_detach_session_response resp;
+ struct lttng_viewer_detach_session_request rq;
+ int ret;
+ ssize_t ret_len;
+
+ cmd.cmd = htobe32(LTTNG_VIEWER_DETACH_SESSION);
+ cmd.data_size = htobe64(sizeof(rq));
+ cmd.cmd_version = htobe32(0);
+
+ memset(&rq, 0, sizeof(rq));
+ rq.session_id = htobe64(id);
+
+ ret_len = lttng_live_send(control_sock, &cmd, sizeof(cmd));
+ if (ret_len < 0) {
+ fprintf(stderr, "[error] Error sending cmd\n");
+ ret = ret_len;
goto error;
}
- ret = len;
-end:
+ ret_len = lttng_live_send(control_sock, &rq, sizeof(rq));
+ if (ret_len < 0) {
+ fprintf(stderr, "Error sending attach request\n");
+ ret = ret_len;
+ goto error;
+ }
+
+ ret_len = lttng_live_recv(control_sock, &resp, sizeof(resp));
+ if (ret_len < 0) {
+ fprintf(stderr, "[error] Error receiving detach session reply\n");
+ ret = ret_len;
+ goto error;
+ }
+
+ if (be32toh(resp.status) != LTTNG_VIEWER_DETACH_SESSION_OK) {
+ fprintf(stderr, "[error] Error detaching viewer session\n");
+ ret = -1;
+ goto error;
+ }
+ ret = 0;
+
error:
return ret;
}
int main(int argc, char **argv)
{
int ret;
- int session_id;
+ uint64_t session_id;
plan_tests(NUM_TESTS);
ret = list_sessions(&session_id);
ok(ret > 0, "List sessions : %d session(s)", ret);
+ if (ret < 0) {
+ goto end;
+ }
+
+ ret = create_viewer_session();
+ ok(ret == 0, "Create viewer session");
ret = attach_session(session_id);
- ok(ret > 0, "Attach to session, %d streams received", ret);
+ ok(ret > 0, "Attach to session, %d stream(s) received", ret);
ret = get_metadata();
ok(ret > 0, "Get metadata, received %d bytes", ret);
ret = get_data_packet(first_packet_stream_id, first_packet_offset,
first_packet_len);
- ok(ret == first_packet_len,
+ ok(ret == 0,
"Get one data packet for stream %d, offset %d, len %d",
first_packet_stream_id, first_packet_offset,
first_packet_len);
+ ret = detach_viewer_session(session_id);
+ ok(ret == 0, "Detach viewer session");
+
+ ret = list_sessions(&session_id);
+ ok(ret > 0, "List sessions : %d session(s)", ret);
+
+ ret = attach_session(session_id);
+ ok(ret > 0, "Attach to session, %d streams received", ret);
+end:
return exit_status();
}