lib: add bt_{graph,query_executor}_add_interrupter()

[babeltrace.git] / src / plugins / ctf / lttng-live / viewer-connection.c

diff --git a/src/plugins/ctf/lttng-live/viewer-connection.c b/src/plugins/ctf/lttng-live/viewer-connection.c
index 57271b2a5485bdd163377ed7334c24a22c2e3677..42ea78c0ad7fd9cca3d4287fff86c91e22960cbd 100644 (file)
--- a/src/plugins/ctf/lttng-live/viewer-connection.c
+++ b/src/plugins/ctf/lttng-live/viewer-connection.c
@@ -67,7 +67,7 @@ ssize_t lttng_live_recv(struct live_viewer_connection *viewer_connection,
                }
                if (ret == BT_SOCKET_ERROR && bt_socket_interrupted()) {
                        if (!viewer_connection->in_query &&
-                                       lttng_live_graph_is_canceled(lttng_live_msg_iter->lttng_live_comp)) {
+                                       lttng_live_graph_is_canceled(lttng_live_msg_iter)) {
                                break;
                        } else {
                                continue;
@@ -93,7 +93,7 @@ ssize_t lttng_live_send(struct live_viewer_connection *viewer_connection,
                ret = bt_socket_send_nosigpipe(sock, buf, len);
                if (ret == BT_SOCKET_ERROR && bt_socket_interrupted()) {
                        if (!viewer_connection->in_query &&
-                                       lttng_live_graph_is_canceled(lttng_live_msg_iter->lttng_live_comp)) {
+                                       lttng_live_graph_is_canceled(lttng_live_msg_iter)) {
                                break;
                        } else {
                                continue;
@@ -149,10 +149,10 @@ int parse_url(struct live_viewer_connection *viewer_connection)
                        "proto : %s",
                        viewer_connection->relay_hostname->str,
                        viewer_connection->port,
-                       viewer_connection->target_hostname == NULL ?
-                               "<none>" : viewer_connection->target_hostname->str,
-                       viewer_connection->session_name == NULL ?
-                               "<none>" : viewer_connection->session_name->str,
+                       !viewer_connection->target_hostname ?
+                       "<none>" : viewer_connection->target_hostname->str,
+                       !viewer_connection->session_name ?
+                       "<none>" : viewer_connection->session_name->str,
                        lttng_live_url_parts.proto->str);
        ret = 0;
 
@@ -347,9 +347,8 @@ int list_update_session(bt_value *results,
                hostname_str = bt_value_string_get(hostname);
                session_name_str = bt_value_string_get(session_name);
 
-               if (!strcmp(session->hostname, hostname_str)
-                               && !strcmp(session->session_name,
-                                       session_name_str)) {
+               if (strcmp(session->hostname, hostname_str) == 0
+                               && strcmp(session->session_name, session_name_str) == 0) {
                        int64_t val;
                        uint32_t streams = be32toh(session->streams);
                        uint32_t clients = be32toh(session->clients);
@@ -362,10 +361,10 @@ int list_update_session(bt_value *results,
                                ret = -1;
                                goto end;
                        }
-                       val = bt_value_signed_integer_get(btval);
+                       val = bt_value_integer_signed_get(btval);
                        /* sum */
                        val += streams;
-                       bt_value_signed_integer_set(btval, val);
+                       bt_value_integer_signed_set(btval, val);
 
                        btval = bt_value_map_borrow_entry_value(map, "client-count");
                        if (!btval) {
@@ -373,10 +372,10 @@ int list_update_session(bt_value *results,
                                ret = -1;
                                goto end;
                        }
-                       val = bt_value_signed_integer_get(btval);
+                       val = bt_value_integer_signed_get(btval);
                        /* max */
                        val = bt_max_t(int64_t, clients, val);
-                       bt_value_signed_integer_set(btval, val);
+                       bt_value_integer_signed_set(btval, val);
                }
 
                if (found) {
@@ -395,7 +394,8 @@ int list_append_session(bt_value *results,
                struct live_viewer_connection *viewer_connection)
 {
        int ret = 0;
-       bt_value_status ret_status;
+       bt_value_map_insert_entry_status insert_status;
+       bt_value_array_append_element_status append_status;
        bt_value *map = NULL;
        GString *url = NULL;
        bool found = false;
@@ -431,8 +431,8 @@ int list_append_session(bt_value *results,
        g_string_append_c(url, '/');
        g_string_append(url, session->session_name);
 
-       ret_status = bt_value_map_insert_string_entry(map, "url", url->str);
-       if (ret_status != BT_VALUE_STATUS_OK) {
+       insert_status = bt_value_map_insert_string_entry(map, "url", url->str);
+       if (insert_status != BT_VALUE_MAP_INSERT_ENTRY_STATUS_OK) {
                BT_COMP_LOGE_STR("Error inserting \"url\" entry.");
                ret = -1;
                goto end;
@@ -442,9 +442,9 @@ int list_append_session(bt_value *results,
         * key = "target-hostname",
         * value = <string>,
         */
-       ret_status = bt_value_map_insert_string_entry(map, "target-hostname",
+       insert_status = bt_value_map_insert_string_entry(map, "target-hostname",
                session->hostname);
-       if (ret_status != BT_VALUE_STATUS_OK) {
+       if (insert_status != BT_VALUE_MAP_INSERT_ENTRY_STATUS_OK) {
                BT_COMP_LOGE_STR("Error inserting \"target-hostname\" entry.");
                ret = -1;
                goto end;
@@ -454,9 +454,9 @@ int list_append_session(bt_value *results,
         * key = "session-name",
         * value = <string>,
         */
-       ret_status = bt_value_map_insert_string_entry(map, "session-name",
+       insert_status = bt_value_map_insert_string_entry(map, "session-name",
                session->session_name);
-       if (ret_status != BT_VALUE_STATUS_OK) {
+       if (insert_status != BT_VALUE_MAP_INSERT_ENTRY_STATUS_OK) {
                BT_COMP_LOGE_STR("Error inserting \"session-name\" entry.");
                ret = -1;
                goto end;
@@ -469,9 +469,9 @@ int list_append_session(bt_value *results,
        {
                uint32_t live_timer = be32toh(session->live_timer);
 
-               ret_status = bt_value_map_insert_signed_integer_entry(
+               insert_status = bt_value_map_insert_signed_integer_entry(
                        map, "timer-us", live_timer);
-               if (ret_status != BT_VALUE_STATUS_OK) {
+               if (insert_status != BT_VALUE_MAP_INSERT_ENTRY_STATUS_OK) {
                        BT_COMP_LOGE_STR("Error inserting \"timer-us\" entry.");
                        ret = -1;
                        goto end;
@@ -485,9 +485,9 @@ int list_append_session(bt_value *results,
        {
                uint32_t streams = be32toh(session->streams);
 
-               ret_status = bt_value_map_insert_signed_integer_entry(map,
+               insert_status = bt_value_map_insert_signed_integer_entry(map,
                        "stream-count", streams);
-               if (ret_status != BT_VALUE_STATUS_OK) {
+               if (insert_status != BT_VALUE_MAP_INSERT_ENTRY_STATUS_OK) {
                        BT_COMP_LOGE_STR("Error inserting \"stream-count\" entry.");
                        ret = -1;
                        goto end;
@@ -501,17 +501,17 @@ int list_append_session(bt_value *results,
        {
                uint32_t clients = be32toh(session->clients);
 
-               ret_status = bt_value_map_insert_signed_integer_entry(map,
+               insert_status = bt_value_map_insert_signed_integer_entry(map,
                        "client-count", clients);
-               if (ret_status != BT_VALUE_STATUS_OK) {
+               if (insert_status != BT_VALUE_MAP_INSERT_ENTRY_STATUS_OK) {
                        BT_COMP_LOGE_STR("Error inserting \"client-count\" entry.");
                        ret = -1;
                        goto end;
                }
        }
 
-       ret_status = bt_value_array_append_element(results, map);
-       if (ret_status != BT_VALUE_STATUS_OK) {
+       append_status = bt_value_array_append_element(results, map);
+       if (append_status != BT_VALUE_ARRAY_APPEND_ELEMENT_STATUS_OK) {
                BT_COMP_LOGE_STR("Error appending map to results.");
                ret = -1;
        }
@@ -561,11 +561,12 @@ end:
  */
 
 BT_HIDDEN
-bt_query_status live_viewer_connection_list_sessions(
+bt_component_class_query_method_status live_viewer_connection_list_sessions(
                struct live_viewer_connection *viewer_connection,
                const bt_value **user_result)
 {
-       bt_query_status status = BT_QUERY_STATUS_OK;
+       bt_component_class_query_method_status status =
+               BT_COMPONENT_CLASS_QUERY_METHOD_STATUS_OK;
        bt_value *result = NULL;
        struct lttng_viewer_cmd cmd;
        struct lttng_viewer_list_sessions list;
@@ -579,7 +580,7 @@ bt_query_status live_viewer_connection_list_sessions(
        result = bt_value_array_create();
        if (!result) {
                BT_COMP_LOGE("Error creating array");
-               status = BT_QUERY_STATUS_NOMEM;
+               status = BT_COMPONENT_CLASS_QUERY_METHOD_STATUS_MEMORY_ERROR;
                goto error;
        }
 
@@ -590,7 +591,7 @@ bt_query_status live_viewer_connection_list_sessions(
        ret_len = lttng_live_send(viewer_connection, &cmd, sizeof(cmd));
        if (ret_len == BT_SOCKET_ERROR) {
                BT_COMP_LOGE("Error sending cmd: %s", bt_socket_errormsg());
-               status = BT_QUERY_STATUS_ERROR;
+               status = BT_COMPONENT_CLASS_QUERY_METHOD_STATUS_ERROR;
                goto error;
        }
        BT_ASSERT(ret_len == sizeof(cmd));
@@ -598,12 +599,12 @@ bt_query_status live_viewer_connection_list_sessions(
        ret_len = lttng_live_recv(viewer_connection, &list, sizeof(list));
        if (ret_len == 0) {
                BT_COMP_LOGI("Remote side has closed connection");
-               status = BT_QUERY_STATUS_ERROR;
+               status = BT_COMPONENT_CLASS_QUERY_METHOD_STATUS_ERROR;
                goto error;
        }
        if (ret_len == BT_SOCKET_ERROR) {
                BT_COMP_LOGE("Error receiving session list: %s", bt_socket_errormsg());
-               status = BT_QUERY_STATUS_ERROR;
+               status = BT_COMPONENT_CLASS_QUERY_METHOD_STATUS_ERROR;
                goto error;
        }
        BT_ASSERT(ret_len == sizeof(list));
@@ -616,12 +617,12 @@ bt_query_status live_viewer_connection_list_sessions(
                        sizeof(lsession));
                if (ret_len == 0) {
                        BT_COMP_LOGI("Remote side has closed connection");
-                       status = BT_QUERY_STATUS_ERROR;
+                       status = BT_COMPONENT_CLASS_QUERY_METHOD_STATUS_ERROR;
                        goto error;
                }
                if (ret_len == BT_SOCKET_ERROR) {
                        BT_COMP_LOGE("Error receiving session: %s", bt_socket_errormsg());
-                       status = BT_QUERY_STATUS_ERROR;
+                       status = BT_COMPONENT_CLASS_QUERY_METHOD_STATUS_ERROR;
                        goto error;
                }
                BT_ASSERT(ret_len == sizeof(lsession));
@@ -629,7 +630,7 @@ bt_query_status live_viewer_connection_list_sessions(
                lsession.session_name[LTTNG_VIEWER_NAME_MAX - 1] = '\0';
                if (list_append_session(result, viewer_connection->url,
                                &lsession, viewer_connection)) {
-                       status = BT_QUERY_STATUS_ERROR;
+                       status = BT_COMPONENT_CLASS_QUERY_METHOD_STATUS_ERROR;
                        goto error;
                }
        }
@@ -1124,8 +1125,6 @@ enum lttng_live_iterator_status lttng_live_get_next_index(
        struct lttng_live_trace *trace = stream->trace;
        const size_t cmd_buf_len = sizeof(cmd) + sizeof(rq);
        char cmd_buf[cmd_buf_len];
-       struct lttng_live_component *lttng_live =
-               lttng_live_msg_iter->lttng_live_comp;
 
        cmd.cmd = htobe32(LTTNG_VIEWER_GET_NEXT_INDEX);
        cmd.data_size = htobe64((uint64_t) sizeof(rq));
@@ -1238,7 +1237,7 @@ end:
        return retstatus;
 
 error:
-       if (lttng_live_graph_is_canceled(lttng_live)) {
+       if (lttng_live_graph_is_canceled(lttng_live_msg_iter)) {
                retstatus = LTTNG_LIVE_ITERATOR_STATUS_AGAIN;
        } else {
                retstatus = LTTNG_LIVE_ITERATOR_STATUS_ERROR;
@@ -1263,8 +1262,6 @@ enum bt_msg_iter_medium_status lttng_live_get_stream_bytes(
        struct lttng_live_trace *trace = stream->trace;
        const size_t cmd_buf_len = sizeof(cmd) + sizeof(rq);
        char cmd_buf[cmd_buf_len];
-       struct lttng_live_component *lttng_live =
-               lttng_live_msg_iter->lttng_live_comp;
 
        BT_COMP_LOGD("lttng_live_get_stream_bytes: offset=%" PRIu64 ", req_len=%" PRIu64,
                        offset, req_len);
@@ -1363,7 +1360,7 @@ end:
        return retstatus;
 
 error:
-       if (lttng_live_graph_is_canceled(lttng_live)) {
+       if (lttng_live_graph_is_canceled(lttng_live_msg_iter)) {
                retstatus = BT_MSG_ITER_MEDIUM_STATUS_AGAIN;
        } else {
                retstatus = BT_MSG_ITER_MEDIUM_STATUS_ERROR;
@@ -1388,8 +1385,6 @@ enum lttng_live_iterator_status lttng_live_get_new_streams(
                session->lttng_live_msg_iter;
        struct live_viewer_connection *viewer_connection =
                lttng_live_msg_iter->viewer_connection;
-       struct lttng_live_component *lttng_live =
-               lttng_live_msg_iter->lttng_live_comp;
        uint32_t streams_count;
        const size_t cmd_buf_len = sizeof(cmd) + sizeof(rq);
        char cmd_buf[cmd_buf_len];
@@ -1460,7 +1455,7 @@ end:
        return status;
 
 error:
-       if (lttng_live_graph_is_canceled(lttng_live)) {
+       if (lttng_live_graph_is_canceled(lttng_live_msg_iter)) {
                status = LTTNG_LIVE_ITERATOR_STATUS_AGAIN;
        } else {
                status = LTTNG_LIVE_ITERATOR_STATUS_ERROR;
@@ -1477,7 +1472,7 @@ struct live_viewer_connection *live_viewer_connection_create(
 
        viewer_connection = g_new0(struct live_viewer_connection, 1);
 
-       if (bt_socket_init() != 0) {
+       if (bt_socket_init(lttng_live_msg_iter->log_level) != 0) {
                goto error;
        }