Fix: test all close return values in sessiond
[lttng-tools.git] / src / bin / lttng-sessiond / ust-app.c
index 62816cecb89f4423970db66177e7821d025983a1..6e1dd60fd6b060a1437589a6594b30c0e69deefe 100644 (file)
@@ -198,7 +198,10 @@ void delete_ust_app(struct ust_app *app)
         * closing this socket, otherwise an application could re-use the socket ID
         * and race with the teardown, using the same hash table entry.
         */
-       close(sock);
+       ret = close(sock);
+       if (ret) {
+               PERROR("close");
+       }
 
        DBG2("UST app pid %d deleted", app->key.pid);
        free(app);
@@ -393,7 +396,7 @@ int create_ust_channel_context(struct ust_app_channel *ua_chan,
 
        ua_ctx->handle = ua_ctx->obj->handle;
 
-       DBG2("UST app context added to channel %s successfully", ua_chan->name);
+       DBG2("UST app context created successfully for channel %s", ua_chan->name);
 
 error:
        return ret;
@@ -416,7 +419,7 @@ int create_ust_event_context(struct ust_app_event *ua_event,
 
        ua_ctx->handle = ua_ctx->obj->handle;
 
-       DBG2("UST app context added to event %s successfully", ua_event->name);
+       DBG2("UST app context created successfully for event %s", ua_event->name);
 
 error:
        return ret;
@@ -865,6 +868,8 @@ static struct ust_app_session *create_ust_app_session(
                ret = ustctl_create_session(app->key.sock);
                if (ret < 0) {
                        ERR("Creating session for app pid %d", app->key.pid);
+                       /* This means that the tracer is gone... */
+                       ua_sess = (void*) -1UL;
                        goto error;
                }
 
@@ -1266,13 +1271,27 @@ error:
 int ust_app_register(struct ust_register_msg *msg, int sock)
 {
        struct ust_app *lta;
+       int ret;
 
        if ((msg->bits_per_long == 64 && ust_consumerd64_fd == -EINVAL)
                        || (msg->bits_per_long == 32 && ust_consumerd32_fd == -EINVAL)) {
                ERR("Registration failed: application \"%s\" (pid: %d) has "
                        "%d-bit long, but no consumerd for this long size is available.\n",
                        msg->name, msg->pid, msg->bits_per_long);
-               close(sock);
+               ret = close(sock);
+               if (ret) {
+                       PERROR("close");
+               }
+               return -EINVAL;
+       }
+       if (msg->major != LTTNG_UST_COMM_MAJOR) {
+               ERR("Registration failed: application \"%s\" (pid: %d) has "
+                       "communication protocol version %u.%u, but sessiond supports 2.x.\n",
+                       msg->name, msg->pid, msg->major, msg->minor);
+               ret = close(sock);
+               if (ret) {
+                       PERROR("close");
+               }
                return -EINVAL;
        }
        lta = zmalloc(sizeof(struct ust_app));
@@ -1420,8 +1439,7 @@ int ust_app_list_events(struct lttng_event **events)
                                }
                        }
                        memcpy(tmp[count].name, uiter.name, LTTNG_UST_SYM_NAME_LEN);
-                       memcpy(tmp[count].loglevel, uiter.loglevel, LTTNG_UST_SYM_NAME_LEN);
-                       tmp[count].loglevel_value = uiter.loglevel_value;
+                       tmp[count].loglevel = uiter.loglevel;
                        tmp[count].type = LTTNG_UST_TRACEPOINT;
                        tmp[count].pid = app->key.pid;
                        tmp[count].enabled = -1;
@@ -1751,8 +1769,11 @@ int ust_app_create_channel_glb(struct ltt_ust_session *usess,
                 */
                ua_sess = create_ust_app_session(usess, app);
                if (ua_sess == NULL) {
-                       /* Major problem here and it's maybe the tracer or malloc() */
+                       /* The malloc() failed. */
                        goto error;
+               } else if (ua_sess == (void *) -1UL) {
+                       /* The application's socket is not valid. Contiuing */
+                       continue;
                }
 
                /* Create channel onto application */
@@ -1924,6 +1945,9 @@ int ust_app_start_trace(struct ltt_ust_session *usess, struct ust_app *app)
                goto skip_setup;
        }
 
+       /* Indicate that the session has been started once */
+       ua_sess->started = 1;
+
        ret = create_ust_app_metadata(ua_sess, usess->pathname, app);
        if (ret < 0) {
                goto error_rcu_unlock;
@@ -1980,7 +2004,6 @@ int ust_app_start_trace(struct ltt_ust_session *usess, struct ust_app *app)
        if (ret < 0) {
                goto error_rcu_unlock;
        }
-       ua_sess->started = 1;
 
 skip_setup:
        /* This start the UST tracing */
@@ -2026,10 +2049,12 @@ int ust_app_stop_trace(struct ltt_ust_session *usess, struct ust_app *app)
                goto error_rcu_unlock;
        }
 
-       /* Not started, continuing. */
-       if (ua_sess->started == 0) {
-               goto end;
-       }
+       /*
+        * If started = 0, it means that stop trace has been called for a session
+        * that was never started. This is a code flow error and should never
+        * happen.
+        */
+       assert(ua_sess->started == 1);
 
        /* This inhibits UST tracing */
        ret = ustctl_stop_session(app->key.sock, ua_sess->handle);
@@ -2060,8 +2085,6 @@ int ust_app_stop_trace(struct ltt_ust_session *usess, struct ust_app *app)
                                ret);
        }
 
-       ua_sess->started = 0;
-
 end:
        rcu_read_unlock();
        return 0;
@@ -2203,11 +2226,12 @@ int ust_app_destroy_trace_all(struct ltt_ust_session *usess)
 void ust_app_global_update(struct ltt_ust_session *usess, int sock)
 {
        int ret = 0;
-       struct lttng_ht_iter iter, uiter;
+       struct lttng_ht_iter iter, uiter, iter_ctx;
        struct ust_app *app;
        struct ust_app_session *ua_sess;
        struct ust_app_channel *ua_chan;
        struct ust_app_event *ua_event;
+       struct ust_app_ctx *ua_ctx;
 
        if (usess == NULL) {
                ERR("No UST session on global update. Returning");
@@ -2247,6 +2271,16 @@ void ust_app_global_update(struct ltt_ust_session *usess, int sock)
                        continue;
                }
 
+               cds_lfht_for_each_entry(ua_chan->ctx->ht, &iter_ctx.iter, ua_ctx,
+                               node.node) {
+                       ret = create_ust_channel_context(ua_chan, ua_ctx, app);
+                       if (ret < 0) {
+                               /* FIXME: Should we quit here or continue... */
+                               continue;
+                       }
+               }
+
+
                /* For each events */
                cds_lfht_for_each_entry(ua_chan->events->ht, &uiter.iter, ua_event,
                                node.node) {
@@ -2255,6 +2289,16 @@ void ust_app_global_update(struct ltt_ust_session *usess, int sock)
                                /* FIXME: Should we quit here or continue... */
                                continue;
                        }
+
+                       /* Add context on events. */
+                       cds_lfht_for_each_entry(ua_event->ctx->ht, &iter_ctx.iter,
+                                       ua_ctx, node.node) {
+                               ret = create_ust_event_context(ua_event, ua_ctx, app);
+                               if (ret < 0) {
+                                       /* FIXME: Should we quit here or continue... */
+                                       continue;
+                               }
+                       }
                }
        }
 
@@ -2538,3 +2582,46 @@ error:
        rcu_read_unlock();
        return -1;
 }
+
+/*
+ * Calibrate registered applications.
+ */
+int ust_app_calibrate_glb(struct lttng_ust_calibrate *calibrate)
+{
+       int ret = 0;
+       struct lttng_ht_iter iter;
+       struct ust_app *app;
+
+       rcu_read_lock();
+
+       cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, node.node) {
+               if (!app->compatible) {
+                       /*
+                        * TODO: In time, we should notice the caller of this error by
+                        * telling him that this is a version error.
+                        */
+                       continue;
+               }
+
+               ret = ustctl_calibrate(app->key.sock, calibrate);
+               if (ret < 0) {
+                       switch (ret) {
+                       case -ENOSYS:
+                               /* Means that it's not implemented on the tracer side. */
+                               ret = 0;
+                               break;
+                       default:
+                               /* TODO: Report error to user */
+                               DBG2("Calibrate app PID %d returned with error %d",
+                                               app->key.pid, ret);
+                               break;
+                       }
+               }
+       }
+
+       DBG("UST app global domain calibration finished");
+
+       rcu_read_unlock();
+
+       return ret;
+}
This page took 0.025703 seconds and 4 git commands to generate.