X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=lttng%2Flttng.c;h=16487ff74a99cb2149bda0964af415ee2721868d;hb=1657e9bb835a28e66396deac871270fe373d5ff4;hp=52c3e3873d27ce967f6cb3294126b64d269144b8;hpb=1c9f794140bd5bf0854e4790cc7ec4a5d3bb1134;p=lttng-tools.git

diff --git a/lttng/lttng.c b/lttng/lttng.c
index 52c3e3873..16487ff74 100644
--- a/lttng/lttng.c
+++ b/lttng/lttng.c
@@ -42,10 +42,12 @@ static char *progname;
 static int process_client_opt(void);
 static int process_opt_list_apps(void);
 static int process_opt_list_sessions(void);
+static int process_opt_list_traces(void);
 static int process_opt_create_session(void);
 static void sighandler(int sig);
 static int set_signal_handler(void);
 static int get_cmdline_by_pid(pid_t pid, char **cmdline);
+static int validate_options(void);
 
 /*
  *  start_client
@@ -79,6 +81,13 @@ static int process_client_opt(void)
 		}
 	}
 
+	if (opt_list_traces) {
+		ret = process_opt_list_traces();
+		if (ret < 0) {
+			goto end;
+		}
+	}
+
 	if (opt_create_session != NULL) {
 		ret = process_opt_create_session();
 		if (ret < 0) {
@@ -95,6 +104,7 @@ static int process_client_opt(void)
 	}
 
 	if (opt_session_uuid != NULL) {
+		DBG("Set session uuid to %s", opt_session_uuid);
 		lttng_set_current_session_uuid(opt_session_uuid);
 	}
 
@@ -104,7 +114,16 @@ static int process_client_opt(void)
 		if (ret < 0) {
 			goto end;
 		}
-		MSG("Trace created successfully!\nUse --start PID to start tracing");
+		MSG("Trace created successfully!\nUse --start PID to start tracing.");
+	}
+
+	if (opt_start_trace) {
+		DBG("Start trace for pid %d", opt_start_trace);
+		ret = lttng_ust_start_trace(opt_start_trace);
+		if (ret < 0) {
+			goto end;
+		}
+		MSG("Trace started successfully!");
 	}
 
 	return 0;
@@ -114,6 +133,45 @@ end:
 	return ret;
 }
 
+/*
+ *  process_opt_list_traces
+ *
+ *  Get list of all traces for a specific session uuid.
+ */
+static int process_opt_list_traces(void)
+{
+	int ret, i;
+	uuid_t uuid;
+	struct lttng_trace *traces;
+
+	uuid_parse(opt_session_uuid, uuid);
+	ret = lttng_list_traces(&uuid, &traces);
+	if (ret < 0) {
+		goto error;
+	}
+
+	MSG("Userspace traces:");
+	for (i = 0; i < ret; i++) {
+		if (traces[i].type == USERSPACE) {
+			MSG("\t%d) %s (pid: %d)", i, traces[i].name, traces[i].pid);
+		} else {
+			break;
+		}
+	}
+
+	MSG("Kernel traces:");
+	for (;i < ret; i++) {
+		if (traces[i].type == KERNEL) {
+			MSG("\t%d) %s", i, traces[i].name);
+		}
+	}
+
+	free(traces);
+
+error:
+	return ret;
+}
+
 /*
  *  process_opt_create_session
  *
@@ -242,6 +300,29 @@ not_running:
 	return 0;
 }
 
+/*
+ *  validate_options
+ *
+ *  Make sure that all options passed to the command line
+ *  are compatible with each others.
+ *
+ *  On error, return -1
+ *  On success, return 0
+ */
+static int validate_options(void)
+{
+	if ((opt_session_uuid == NULL) &&
+			(opt_create_trace || opt_start_trace || opt_list_traces)) {
+		ERR("You need to specify a session UUID.\nPlease use --session UUID to do so.");
+		goto error;
+	}
+
+	return 0;
+
+error:
+	return -1;
+}
+
 /*
  *  spawn_sessiond
  *
@@ -376,15 +457,17 @@ end:
  */
 static void sighandler(int sig)
 {
-	DBG("%d received", sig);
 	switch (sig) {
 		case SIGTERM:
+			DBG("SIGTERM catched");
 			clean_exit(EXIT_FAILURE);
 			break;
 		case SIGCHLD:
 			/* Notify is done */
+			DBG("SIGCHLD catched");
 			break;
 		default:
+			DBG("Unknown signal %d catched", sig);
 			break;
 	}
 
@@ -421,6 +504,11 @@ int main(int argc, char *argv[])
 		clean_exit(EXIT_FAILURE);
 	}
 
+	ret = validate_options();
+	if (ret < 0) {
+		return EXIT_FAILURE;
+	}
+
 	ret = set_signal_handler();
 	if (ret < 0) {
 		clean_exit(ret);