X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=src%2Fbin%2Flttng-sessiond%2Fcmd.c;h=91ea88cb7c593a5c5f8cb010016a4f5f9816b8d4;hb=ccf10263bcd2ca4667b9e1fc4dab64a3c8d8c4d8;hp=06b015f937e3e12f2d2a360a3ac6223d3588b64b;hpb=d07ceecd2f35ccf383c7529a18dcc9b9c4b8cb17;p=lttng-tools.git

diff --git a/src/bin/lttng-sessiond/cmd.c b/src/bin/lttng-sessiond/cmd.c
index 06b015f93..91ea88cb7 100644
--- a/src/bin/lttng-sessiond/cmd.c
+++ b/src/bin/lttng-sessiond/cmd.c
@@ -929,6 +929,90 @@ error:
 	return ret;
 }
 
+/*
+ * Command LTTNG_TRACK_PID processed by the client thread.
+ */
+int cmd_track_pid(struct ltt_session *session, int domain, int pid)
+{
+	int ret;
+
+	rcu_read_lock();
+
+	switch (domain) {
+	case LTTNG_DOMAIN_KERNEL:
+	{
+		struct ltt_kernel_session *ksess;
+
+		ksess = session->kernel_session;
+
+		ret = kernel_track_pid(ksess, pid);
+		if (ret != LTTNG_OK) {
+			goto error;
+		}
+
+		kernel_wait_quiescent(kernel_tracer_fd);
+		break;
+	}
+#if 0
+	case LTTNG_DOMAIN_UST:
+	case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
+	case LTTNG_DOMAIN_UST_EXEC_NAME:
+	case LTTNG_DOMAIN_UST_PID:
+#endif
+	default:
+		ret = LTTNG_ERR_UNKNOWN_DOMAIN;
+		goto error;
+	}
+
+	ret = LTTNG_OK;
+
+error:
+	rcu_read_unlock();
+	return ret;
+}
+
+/*
+ * Command LTTNG_UNTRACK_PID processed by the client thread.
+ */
+int cmd_untrack_pid(struct ltt_session *session, int domain, int pid)
+{
+	int ret;
+
+	rcu_read_lock();
+
+	switch (domain) {
+	case LTTNG_DOMAIN_KERNEL:
+	{
+		struct ltt_kernel_session *ksess;
+
+		ksess = session->kernel_session;
+
+		ret = kernel_untrack_pid(ksess, pid);
+		if (ret != LTTNG_OK) {
+			goto error;
+		}
+
+		kernel_wait_quiescent(kernel_tracer_fd);
+		break;
+	}
+#if 0
+	case LTTNG_DOMAIN_UST:
+	case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
+	case LTTNG_DOMAIN_UST_EXEC_NAME:
+	case LTTNG_DOMAIN_UST_PID:
+#endif
+	default:
+		ret = LTTNG_ERR_UNKNOWN_DOMAIN;
+		goto error;
+	}
+
+	ret = LTTNG_OK;
+
+error:
+	rcu_read_unlock();
+	return ret;
+}
+
 /*
  * Command LTTNG_ENABLE_CHANNEL processed by the client thread.
  *
@@ -3152,7 +3236,7 @@ int cmd_snapshot_record(struct ltt_session *session,
 	int ret = LTTNG_OK;
 	unsigned int use_tmp_output = 0;
 	struct snapshot_output tmp_output;
-	unsigned int nb_streams, snapshot_success = 0;
+	unsigned int snapshot_success = 0;
 
 	assert(session);
 	assert(output);
@@ -3333,6 +3417,29 @@ error:
 	return ret;
 }
 
+/*
+ * Command LTTNG_SET_SESSION_SHM_PATH processed by the client thread.
+ */
+int cmd_set_session_shm_path(struct ltt_session *session,
+		const char *shm_path)
+{
+	/* Safety net */
+	assert(session);
+
+	/*
+	 * Can only set shm path before session is started.
+	 */
+	if (session->has_been_started) {
+		return LTTNG_ERR_SESSION_STARTED;
+	}
+
+	strncpy(session->shm_path, shm_path,
+		sizeof(session->shm_path));
+	session->shm_path[sizeof(session->shm_path) - 1] = '\0';
+
+	return 0;
+}
+
 /*
  * Init command subsystem.
  */