Cleanup: ust start/stop trace

Move setting/clearing the session "active" state from cmd.c to
cmd_start_trace()/cmd_stop_trace() for better encapsulation of
behavior.

Reduces the amount of code to maintain in the catch-all cmd.c.

Signed-off-by: Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
Signed-off-by: Jérémie Galarneau <jeremie.galarneau@efficios.com>

diff --git a/src/bin/lttng-sessiond/cmd.c b/src/bin/lttng-sessiond/cmd.c
index 507274f35568bc421887ebfb88674d476297af1d..d3c8376be1a01cf691f4a02be1ffe93c09c4a6b0 100644 (file)
--- a/src/bin/lttng-sessiond/cmd.c
+++ b/src/bin/lttng-sessiond/cmd.c
@@ -2684,12 +2684,6 @@ int cmd_start_trace(struct ltt_session *session)
 
        /* Flag session that trace should start automatically */
        if (usess) {
-               /*
-                * Even though the start trace might fail, flag this session active so
-                * other application coming in are started by default.
-                */
-               usess->active = 1;
-
                ret = ust_app_start_trace_all(usess);
                if (ret < 0) {
                        ret = LTTNG_ERR_UST_START_FAIL;
@@ -2781,12 +2775,6 @@ int cmd_stop_trace(struct ltt_session *session)
        }
 
        if (usess && usess->active) {
-               /*
-                * Even though the stop trace might fail, flag this session inactive so
-                * other application coming in are not started by default.
-                */
-               usess->active = 0;
-
                ret = ust_app_stop_trace_all(usess);
                if (ret < 0) {
                        ret = LTTNG_ERR_UST_STOP_FAIL;

diff --git a/src/bin/lttng-sessiond/ust-app.c b/src/bin/lttng-sessiond/ust-app.c
index 8c5f3f6afc0e3dfdb7ad225f2963bce7bd526624..f88df7406ea567049321b6c508fdc5b249a4ff35 100644 (file)
--- a/src/bin/lttng-sessiond/ust-app.c
+++ b/src/bin/lttng-sessiond/ust-app.c
@@ -4996,6 +4996,12 @@ int ust_app_start_trace_all(struct ltt_ust_session *usess)
 
        DBG("Starting all UST traces");
 
+       /*
+        * Even though the start trace might fail, flag this session active so
+        * other application coming in are started by default.
+        */
+       usess->active = 1;
+
        rcu_read_lock();
 
        /*
@@ -5031,6 +5037,12 @@ int ust_app_stop_trace_all(struct ltt_ust_session *usess)
 
        DBG("Stopping all UST traces");
 
+       /*
+        * Even though the stop trace might fail, flag this session inactive so
+        * other application coming in are not started by default.
+        */
+       usess->active = 0;
+
        rcu_read_lock();
 
        cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {