X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fbin%2Flttng-sessiond%2Fcmd.c;h=696e650debb70e6c0f9724b506a1b1413f0bd3a7;hp=2dcc52b7a2aff880bdfbb1c9fc61cb8ece35c27d;hb=6bad3253d952934f18be50e032b3d495e7a54613;hpb=971da06a61170301dc28c2203ef0679c3024e21d

diff --git a/src/bin/lttng-sessiond/cmd.c b/src/bin/lttng-sessiond/cmd.c
index 2dcc52b7a..696e650de 100644
--- a/src/bin/lttng-sessiond/cmd.c
+++ b/src/bin/lttng-sessiond/cmd.c
@@ -220,7 +220,9 @@ static int list_lttng_agent_events(struct agent *agt,
 
 	DBG3("Listing agent events");
 
+	rcu_read_lock();
 	nb_event = lttng_ht_get_count(agt->events);
+	rcu_read_unlock();
 	if (nb_event == 0) {
 		ret = nb_event;
 		goto error;
@@ -1040,7 +1042,8 @@ int cmd_disable_event(struct ltt_session *session, int domain,
 
 	event_name = event->name;
 
-	if (event->loglevel_type || event->loglevel || event->enabled
+	/* Error out on unhandled search criteria */
+	if (event->loglevel_type || event->loglevel != -1 || event->enabled
 			|| event->pid || event->filter || event->exclusion) {
 		return LTTNG_ERR_UNK;
 	}
@@ -2415,12 +2418,14 @@ ssize_t cmd_list_domains(struct ltt_session *session,
 		DBG3("Listing domains found UST global domain");
 		nb_dom++;
 
+		rcu_read_lock();
 		cds_lfht_for_each_entry(session->ust_session->agents->ht, &iter.iter,
 				agt, node.node) {
 			if (agt->being_used) {
 				nb_dom++;
 			}
 		}
+		rcu_read_unlock();
 	}
 
 	if (!nb_dom) {
@@ -2443,6 +2448,7 @@ ssize_t cmd_list_domains(struct ltt_session *session,
 		(*domains)[index].buf_type = session->ust_session->buffer_type;
 		index++;
 
+		rcu_read_lock();
 		cds_lfht_for_each_entry(session->ust_session->agents->ht, &iter.iter,
 				agt, node.node) {
 			if (agt->being_used) {
@@ -2451,6 +2457,7 @@ ssize_t cmd_list_domains(struct ltt_session *session,
 				index++;
 			}
 		}
+		rcu_read_unlock();
 	}
 end:
 	return nb_dom;
@@ -2482,8 +2489,10 @@ ssize_t cmd_list_channels(int domain, struct ltt_session *session,
 		break;
 	case LTTNG_DOMAIN_UST:
 		if (session->ust_session != NULL) {
+			rcu_read_lock();
 			nb_chan = lttng_ht_get_count(
-					session->ust_session->domain_global.channels);
+				session->ust_session->domain_global.channels);
+			rcu_read_unlock();
 		}
 		DBG3("Number of UST global channels %zd", nb_chan);
 		if (nb_chan < 0) {
@@ -2544,10 +2553,12 @@ ssize_t cmd_list_events(int domain, struct ltt_session *session,
 			struct lttng_ht_iter iter;
 			struct agent *agt;
 
+			rcu_read_lock();
 			cds_lfht_for_each_entry(session->ust_session->agents->ht,
 					&iter.iter, agt, node.node) {
 				nb_event = list_lttng_agent_events(agt, events);
 			}
+			rcu_read_unlock();
 		}
 		break;
 	default:
@@ -2796,7 +2807,7 @@ ssize_t cmd_snapshot_list_outputs(struct ltt_session *session,
 		struct lttng_snapshot_output **outputs)
 {
 	int ret, idx = 0;
-	struct lttng_snapshot_output *list;
+	struct lttng_snapshot_output *list = NULL;
 	struct lttng_ht_iter iter;
 	struct snapshot_output *output;
 
@@ -2810,7 +2821,7 @@ ssize_t cmd_snapshot_list_outputs(struct ltt_session *session,
 	 * set in no output mode.
 	 */
 	if (session->output_traces) {
-		ret = LTTNG_ERR_EPERM;
+		ret = -LTTNG_ERR_EPERM;
 		goto error;
 	}
 
@@ -2821,11 +2832,12 @@ ssize_t cmd_snapshot_list_outputs(struct ltt_session *session,
 
 	list = zmalloc(session->snapshot.nb_output * sizeof(*list));
 	if (!list) {
-		ret = LTTNG_ERR_NOMEM;
+		ret = -LTTNG_ERR_NOMEM;
 		goto error;
 	}
 
 	/* Copy list from session to the new list object. */
+	rcu_read_lock();
 	cds_lfht_for_each_entry(session->snapshot.output_ht->ht, &iter.iter,
 			output, node.node) {
 		assert(output->consumer);
@@ -2840,28 +2852,28 @@ ssize_t cmd_snapshot_list_outputs(struct ltt_session *session,
 			ret = uri_to_str_url(&output->consumer->dst.net.control,
 					list[idx].ctrl_url, sizeof(list[idx].ctrl_url));
 			if (ret < 0) {
-				ret = LTTNG_ERR_NOMEM;
-				goto free_error;
+				ret = -LTTNG_ERR_NOMEM;
+				goto error;
 			}
 
 			/* Data URI. */
 			ret = uri_to_str_url(&output->consumer->dst.net.data,
 					list[idx].data_url, sizeof(list[idx].data_url));
 			if (ret < 0) {
-				ret = LTTNG_ERR_NOMEM;
-				goto free_error;
+				ret = -LTTNG_ERR_NOMEM;
+				goto error;
 			}
 		}
 		idx++;
 	}
 
 	*outputs = list;
-	return session->snapshot.nb_output;
-
-free_error:
-	free(list);
+	list = NULL;
+	ret = session->snapshot.nb_output;
 error:
-	return -ret;
+	free(list);
+	rcu_read_unlock();
+	return ret;
 }
 
 /*
@@ -3054,12 +3066,14 @@ static uint64_t get_session_max_subbuf_size(struct ltt_session *session)
 		struct ltt_ust_channel *uchan;
 		struct ltt_ust_session *usess = session->ust_session;
 
+		rcu_read_lock();
 		cds_lfht_for_each_entry(usess->domain_global.channels->ht, &iter.iter,
 				uchan, node.node) {
 			if (uchan->attr.subbuf_size > max_size) {
 				max_size = uchan->attr.subbuf_size;
 			}
 		}
+		rcu_read_unlock();
 	}
 
 	return max_size;