add support for defining global host info
authorFelix Fietkau <nbd@nbd.name>
Tue, 6 Jul 2021 17:44:01 +0000 (19:44 +0200)
committerFelix Fietkau <nbd@nbd.name>
Tue, 6 Jul 2021 17:44:02 +0000 (19:44 +0200)
Change the '*' in the node info update cmd to refer to host info instead

Signed-off-by: Felix Fietkau <nbd@nbd.name>
local_node.c
main.c
monitor.c
node.h
parse.c
remote.c
remote.h
ubus.c
usteer.h

index 97cb716e8c4aafc51c89a87b05a705d32da78082..985b37b068a711d8ffc787d08fdfe982d4459daf 100644 (file)
@@ -506,23 +506,6 @@ node_list_cb(struct ubus_context *ctx, struct ubus_object_data *obj, void *priv)
        usteer_register_node(ctx, obj->path, obj->id);
 }
 
-void usteer_local_node_update_node_info(struct usteer_local_node *ln)
-{
-       struct blob_attr *val;
-       const char *name;
-
-       blob_buf_init(&b, 0);
-       kvlist_for_each(&ln->node_info, name, val)
-               blobmsg_add_field(&b, blobmsg_type(val), name,
-                                 blobmsg_data(val), blobmsg_len(val));
-
-       val = b.head;
-       if (!blobmsg_len(val))
-               val = NULL;
-
-       usteer_node_set_blob(&ln->node.node_info, val);
-}
-
 void config_set_node_up_script(struct blob_attr *data)
 {
        const char *val;
diff --git a/main.c b/main.c
index a26818e0d3cc596cfe08f8e22fcc52b576c2ac6e..9e35bf93b3a58fc090c3e481f6be0165339b92a3 100644 (file)
--- a/main.c
+++ b/main.c
@@ -26,6 +26,7 @@
 
 struct ubus_context *ubus_ctx;
 struct usteer_config config = {};
+struct blob_attr *host_info_blob;
 uint64_t current_time;
 
 LIST_HEAD(node_handlers);
index bf02ed7af29ca2635568bd1fa0b288b4b60c436c..625783762bb74421ae32837ad81c9121896524cb 100644 (file)
--- a/monitor.c
+++ b/monitor.c
@@ -117,6 +117,11 @@ decode_packet(struct blob_attr *data)
        }
 
        fprintf(stderr, "id=%08x, seq=%d\n", msg.id, msg.seq);
+       if (msg.host_info) {
+               char *data = blobmsg_format_json(msg.host_info, true);
+               fprintf(stderr, "\tHost info: %s\n", data);
+               free(data);
+       }
 
        blob_for_each_attr(cur, msg.nodes, rem)
                decode_node(cur);
diff --git a/node.h b/node.h
index 9d5b7aa630c1282a08c6cd20af8812e610dcfd1d..d177ced87c57c790bd95d3c808d4422adb844226 100644 (file)
--- a/node.h
+++ b/node.h
@@ -70,6 +70,7 @@ struct usteer_remote_host {
        struct avl_node avl;
 
        struct list_head nodes;
+       struct blob_attr *host_info;
        char *addr;
 };
 
diff --git a/parse.c b/parse.c
index d8bd3599ea971786899a3998df3f333e1ec47935..0ff2dfb8d3f6a65eec0e77c3c0bbd7bc170376da 100644 (file)
--- a/parse.c
+++ b/parse.c
@@ -26,6 +26,7 @@ bool parse_apmsg(struct apmsg *msg, struct blob_attr *data)
                [APMSG_ID] = { .type = BLOB_ATTR_INT32 },
                [APMSG_SEQ] = { .type = BLOB_ATTR_INT32 },
                [APMSG_NODES] = { .type = BLOB_ATTR_NESTED },
+               [APMSG_HOST_INFO] = { .type = BLOB_ATTR_NESTED },
        };
        struct blob_attr *tb[__APMSG_MAX];
 
@@ -36,6 +37,7 @@ bool parse_apmsg(struct apmsg *msg, struct blob_attr *data)
        msg->id = blob_get_int32(tb[APMSG_ID]);
        msg->seq = blob_get_int32(tb[APMSG_SEQ]);
        msg->nodes = tb[APMSG_NODES];
+       msg->host_info = tb[APMSG_HOST_INFO];
 
        return true;
 }
index e5b27972d652f6cb9b88bce96317f036dbdfafd5..de183a4b8eb0def9ca278a8f413926de14dde187 100644 (file)
--- a/remote.c
+++ b/remote.c
@@ -305,6 +305,7 @@ interface_recv_msg(struct interface *iface, struct in_addr *addr, void *buf, int
        inet_ntop(AF_INET, addr, addr_str, sizeof(addr_str));
 
        host = interface_get_host(addr_str, msg.id);
+       usteer_node_set_blob(&host->host_info, msg.host_info);
 
        blob_for_each_attr(cur, msg.nodes, rem)
                interface_add_node(host, cur);
@@ -492,6 +493,10 @@ usteer_update_init(void)
        blob_buf_init(&buf, 0);
        blob_put_int32(&buf, APMSG_ID, local_id);
        blob_put_int32(&buf, APMSG_SEQ, ++msg_seq);
+       if (host_info_blob)
+               blob_put(&buf, APMSG_HOST_INFO,
+                        blob_data(host_info_blob),
+                        blob_len(host_info_blob));
 
        return blob_nest_start(&buf, APMSG_NODES);
 }
index b749f64485591a8e6621ee9fb87d64145159dabf..1c0123396738deb2101efcb6ede20ba5309a0218 100644 (file)
--- a/remote.h
+++ b/remote.h
@@ -26,6 +26,7 @@ enum {
        APMSG_ID,
        APMSG_SEQ,
        APMSG_NODES,
+       APMSG_HOST_INFO,
        __APMSG_MAX
 };
 
@@ -33,6 +34,7 @@ struct apmsg {
        uint32_t id;
        uint32_t seq;
        struct blob_attr *nodes;
+       struct blob_attr *host_info;
 };
 
 enum {
diff --git a/ubus.c b/ubus.c
index 4ae5ef9379ef101d51995f4a0705d447854480f9..bc328f3c4cf916407f9f4090a29d1fd724aaaa43 100644 (file)
--- a/ubus.c
+++ b/ubus.c
@@ -29,6 +29,7 @@
 #include "event.h"
 
 static struct blob_buf b;
+static KVLIST(host_info, kvlist_blob_len);
 
 static int
 usteer_ubus_get_clients(struct ubus_context *ctx, struct ubus_object *obj,
@@ -319,6 +320,10 @@ usteer_ubus_remote_hosts(struct ubus_context *ctx, struct ubus_object *obj,
        avl_for_each_element(&remote_hosts, host, avl) {
                c = blobmsg_open_table(&b, host->addr);
                blobmsg_add_u32(&b, "id", (uint32_t)(uintptr_t)host->avl.key);
+               if (host->host_info)
+                       blobmsg_add_field(&b, BLOBMSG_TYPE_TABLE, "host_info",
+                                         blobmsg_data(host->host_info),
+                                         blobmsg_len(host->host_info));
                blobmsg_close_table(&b, c);
        }
 
@@ -361,20 +366,36 @@ static const struct blobmsg_policy del_node_data_policy[] = {
 };
 
 static void
-__usteer_ubus_update_node_data(struct usteer_local_node *ln, struct blob_attr *data,
-                              bool delete)
+usteer_update_kvlist_data(struct kvlist *kv, struct blob_attr *data,
+                         bool delete)
 {
        struct blob_attr *cur;
        int rem;
 
        blobmsg_for_each_attr(cur, data, rem) {
                if (delete)
-                       kvlist_delete(&ln->node_info, blobmsg_get_string(cur));
+                       kvlist_delete(kv, blobmsg_get_string(cur));
                else
-                       kvlist_set(&ln->node_info, blobmsg_name(cur), cur);
+                       kvlist_set(kv, blobmsg_name(cur), cur);
        }
+}
+
+static void
+usteer_update_kvlist_blob(struct blob_attr **dest, struct kvlist *kv)
+{
+       struct blob_attr *val;
+       const char *name;
+
+       blob_buf_init(&b, 0);
+       kvlist_for_each(kv, name, val)
+               blobmsg_add_field(&b, blobmsg_type(val), name,
+                                 blobmsg_data(val), blobmsg_len(val));
+
+       val = b.head;
+       if (!blobmsg_len(val))
+               val = NULL;
 
-       usteer_local_node_update_node_info(ln);
+       usteer_node_set_blob(dest, val);
 }
 
 static int
@@ -406,13 +427,14 @@ usteer_ubus_update_node_data(struct ubus_context *ctx, struct ubus_object *obj,
                if (!ln)
                        return UBUS_STATUS_NOT_FOUND;
 
-               __usteer_ubus_update_node_data(ln, val, delete);
+               usteer_update_kvlist_data(&ln->node_info, val, delete);
+               usteer_update_kvlist_blob(&ln->node.node_info, &ln->node_info);
 
                return 0;
        }
 
-       avl_for_each_element(&local_nodes, ln, node.avl)
-               __usteer_ubus_update_node_data(ln, val, delete);
+       usteer_update_kvlist_data(&host_info, val, delete);
+       usteer_update_kvlist_blob(&host_info_blob, &host_info);
 
        return 0;
 }
index 81ed1f4e6bec916c05ac52e6431ff2bd945ff2ad..f8ffc8d05fbbcba57dc7d9c74bb04efce270eaa2 100644 (file)
--- a/usteer.h
+++ b/usteer.h
@@ -222,6 +222,7 @@ extern struct avl_tree stations;
 extern struct ubus_object usteer_obj;
 extern uint64_t current_time;
 extern const char * const event_types[__EVENT_TYPE_MAX];
+extern struct blob_attr *host_info_blob;
 
 void usteer_update_time(void);
 void usteer_init_defaults(void);
@@ -230,7 +231,6 @@ bool usteer_handle_sta_event(struct usteer_node *node, const uint8_t *addr,
 
 void usteer_local_nodes_init(struct ubus_context *ctx);
 void usteer_local_node_kick(struct usteer_local_node *ln);
-void usteer_local_node_update_node_info(struct usteer_local_node *ln);
 
 void usteer_ubus_init(struct ubus_context *ctx);
 void usteer_ubus_kick_client(struct sta_info *si);