X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=liblttngctl%2Fliblttngctl.c;h=7c5af7fe784e3169ed841e66413e6048e149f953;hp=658f646f0fe7fc4c49e21d3917b7b08592959def;hb=87378cf56f08f23cd6715a923ba10d98343902d7;hpb=5716705821202372fd16168f66f347ba293ef6b4

diff --git a/liblttngctl/liblttngctl.c b/liblttngctl/liblttngctl.c
index 658f646f0..7c5af7fe7 100644
--- a/liblttngctl/liblttngctl.c
+++ b/liblttngctl/liblttngctl.c
@@ -34,33 +34,29 @@ static int sessiond_socket;
 static char sessiond_sock_path[PATH_MAX];
 
 /* Communication structure to ltt-sessiond */
-static struct lttcomm_lttng_msg llm;
 static struct lttcomm_session_msg lsm;
+static struct lttcomm_lttng_msg llm;
 
 /* Prototypes */
 static int check_tracing_group(const char *grp_name);
-static int ask_sessiond(void);
-static int recvfrom_sessiond(void);
+static int ask_sessiond(enum lttcomm_command_type lct, void **buf);
+static int recv_data_sessiond(void *buf, size_t len);
+static int send_data_sessiond(void);
 static int set_session_daemon_path(void);
-static void reset_data_struct(void);
-
-int lttng_connect_sessiond(void);
-int lttng_create_session(const char *name, char *session_id);
-int lttng_check_session_daemon(void);
 
 /* Variables */
 static char *tracing_group;
 static int connected;
 
 /*
- *  ask_sessiond
+ *  send_data_sessiond
  *
  *  Send lttcomm_session_msg to the session daemon.
  *
  *  On success, return 0
  *  On error, return error code
  */
-static int ask_sessiond(void)
+static int send_data_sessiond(void)
 {
 	int ret;
 
@@ -76,14 +72,14 @@ end:
 }
 
 /*
- *  recvfrom_sessiond
+ *  recv_data_sessiond
  *
  *  Receive data from the sessiond socket.
  *
  *  On success, return 0
  *  On error, return recv() error code
  */
-static int recvfrom_sessiond(void)
+static int recv_data_sessiond(void *buf, size_t len)
 {
 	int ret;
 
@@ -92,18 +88,68 @@ static int recvfrom_sessiond(void)
 		goto end;
 	}
 
-	ret = lttcomm_recv_unix_sock(sessiond_socket, &llm, sizeof(llm));
+	ret = lttcomm_recv_unix_sock(sessiond_socket, buf, len);
+	if (ret < 0) {
+		goto end;
+	}
+
+end:
+	return ret;
+}
+
+/*
+ *  ask_sessiond
+ *
+ *  Ask the session daemon a specific command
+ *  and put the data into buf.
+ *
+ *  Return size of data (only payload, not header).
+ */
+static int ask_sessiond(enum lttcomm_command_type lct, void **buf)
+{
+	int ret;
+	size_t size;
+	void *data = NULL;
+
+	lsm.cmd_type = lct;
+
+	/* Send command to session daemon */
+	ret = send_data_sessiond();
+	if (ret < 0) {
+		goto end;
+	}
+
+	/* Get header from data transmission */
+	ret = recv_data_sessiond(&llm, sizeof(llm));
 	if (ret < 0) {
 		goto end;
 	}
 
-	/* Check return code */
+	/* Check error code if OK */
 	if (llm.ret_code != LTTCOMM_OK) {
 		ret = -llm.ret_code;
 		goto end;
 	}
 
+	size = llm.size_payload;
+	if (size == 0) {
+		goto end;
+	}
+
+	data = (void*) malloc(size);
+
+	/* Get payload data */
+	ret = recv_data_sessiond(data, size);
+	if (ret < 0) {
+		goto end;
+	}
+
+	*buf = data;
+	ret = size;
+
 end:
+	/* Reset lsm data struct */
+	memset(&lsm, 0, sizeof(lsm));
 	return ret;
 }
 
@@ -122,82 +168,67 @@ const char *lttng_get_readable_code(int code)
 }
 
 /*
- *  lttng_create_session
+ *  lttng_ust_list_apps
  *
- *  Create a tracing session using "name" to the session daemon.
- *  If no name is given, the auto session creation is set and
- *  the daemon will take care of finding a name.
+ *  Ask the session daemon for all UST traceable
+ *  applications.
  *
- *  On success, return 0 and session_id point to the uuid str.
- *  On error, negative value is returned.
+ *  Return the number of pids.
+ *  On error, return negative value.
  */
-int lttng_create_session(const char *name, char *session_id)
+int lttng_ust_list_apps(pid_t **pids)
 {
 	int ret;
 
-	lsm.cmd_type = LTTNG_CREATE_SESSION;
-	if (name == NULL) {
-		lsm.u.create_session.auto_session = 1;
-	} else {
-		strncpy(lsm.session_name, name, strlen(name));
-		lsm.u.create_session.auto_session = 0;
+	ret = ask_sessiond(UST_LIST_APPS, (void**) pids);
+	if (ret < 0) {
+		return ret;
 	}
 
-	/* Ask the session daemon */
-	ret = ask_sessiond();
+	return ret / sizeof(pid_t);
+}
+
+/*
+ *  lttng_create_session
+ *
+ *  Create a brand new session using name. Allocate
+ *  the session_id param pointing to the UUID.
+ */
+int lttng_create_session(char *name, uuid_t *session_id)
+{
+	int ret;
+
+	strncpy(lsm.session_name, name, sizeof(lsm.session_name));
+	lsm.session_name[sizeof(lsm.session_name) - 1] = '\0';
+
+	ret = ask_sessiond(LTTNG_CREATE_SESSION, NULL);
 	if (ret < 0) {
 		goto end;
 	}
 
-	/* Unparse session ID */
-	uuid_unparse(llm.session_id, session_id);
+	uuid_copy(*session_id, llm.session_id);
 
 end:
-	reset_data_struct();
-
 	return ret;
 }
 
 /*
- *  lttng_ust_list_apps
- *
- *  Ask the session daemon for all UST traceable
- *  applications.
+ *  lttng_destroy_session
  *
- *  Return the size of pids.
+ *  Destroy session using name.
  */
-size_t lttng_ust_list_apps(pid_t **pids)
+int lttng_destroy_session(uuid_t *uuid)
 {
-	int ret, first = 0;
-	size_t size = 0;
-	pid_t *p = NULL;
+	int ret;
 
-	lsm.cmd_type = UST_LIST_APPS;
+	uuid_copy(lsm.session_id, *uuid);
 
-	ret = ask_sessiond();
+	ret = ask_sessiond(LTTNG_DESTROY_SESSION, NULL);
 	if (ret < 0) {
-		goto error;
+		goto end;
 	}
 
-	do {
-		ret = recvfrom_sessiond();
-		if (ret < 0) {
-			goto error;
-		}
-
-		if (first == 0) {
-			first = 1;
-			size = llm.num_pckt;
-			p = malloc(sizeof(pid_t) * size);
-		}
-		p[size - llm.num_pckt] = llm.u.list_apps.pid;
-	} while ((llm.num_pckt-1) != 0);
-
-	*pids = p;
-
-	return size;
-
-error:
+end:
 	return ret;
 }
 
@@ -206,44 +237,19 @@ error:
  *
  *  Ask the session daemon for all available sessions.
  *
- *  Return number of sessions
+ *  Return number of session.
+ *  On error, return negative value.
  */
-size_t lttng_list_sessions(struct lttng_session **sessions)
+int lttng_list_sessions(struct lttng_session **sessions)
 {
-	int ret, first = 0;
-	size_t size = 0;
-	struct lttng_session *ls = NULL;
-
-	lsm.cmd_type = LTTNG_LIST_SESSIONS;
+	int ret;
 
-	ret = ask_sessiond();
+	ret = ask_sessiond(LTTNG_LIST_SESSIONS, (void**) sessions);
 	if (ret < 0) {
-		goto error;
+		return ret;
 	}
 
-	do {
-		ret = recvfrom_sessiond();
-		if (ret < 0) {
-			goto error;
-		}
-
-		if (first == 0) {
-			first = 1;
-			size = llm.num_pckt;
-			ls = malloc(sizeof(struct lttng_session) * size);
-		}
-		strncpy(ls[size - llm.num_pckt].name, llm.u.list_sessions.name,
-				sizeof(ls[size - llm.num_pckt].name));
-		strncpy(ls[size - llm.num_pckt].uuid, llm.u.list_sessions.uuid,
-				sizeof(ls[size - llm.num_pckt].uuid));
-	} while ((llm.num_pckt - 1) != 0);
-
-	*sessions = ls;
-
-	return size;
-
-error:
-	return ret;
+	return ret / sizeof(struct lttng_session);
 }
 
 /*
@@ -274,6 +280,34 @@ int lttng_connect_sessiond(void)
 	return 0;
 }
 
+/*
+ *  lttng_disconnect_sessiond
+ *
+ *  Clean disconnect the session daemon.
+ */
+int lttng_disconnect_sessiond(void)
+{
+	int ret = 0;
+
+	if (connected) {
+		ret = lttcomm_close_unix_sock(sessiond_socket);
+		sessiond_socket = 0;
+		connected = 0;
+	}
+
+	return ret;
+}
+
+/*
+ *  lttng_set_current_session_uuid
+ *
+ *  Set the session uuid for current lsm.
+ */
+void lttng_set_current_session_uuid(char *uuid)
+{
+	uuid_parse(uuid, lsm.session_id);
+}
+
 /*
  *  lttng_set_tracing_group
  *
@@ -313,17 +347,6 @@ int lttng_check_session_daemon(void)
 	return 0;
 }
 
-/*
- *  reset_data_struct
- *
- *  Reset session daemon structures.
- */
-static void reset_data_struct(void)
-{
-	memset(&lsm, 0, sizeof(lsm));
-	memset(&llm, 0, sizeof(llm));
-}
-
 /*
  *  set_session_daemon_path
  *