Skip to content

Commit

Permalink
ubus: filter interfaces
Browse files Browse the repository at this point in the history
  • Loading branch information
blocktrron committed Apr 19, 2024
1 parent 3bd14cc commit f7dadd8
Show file tree
Hide file tree
Showing 4 changed files with 85 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/daemon.c
Original file line number Diff line number Diff line change
Expand Up @@ -141,6 +141,8 @@ static void nw_daemon_collect_information(struct uloop_timeout *timeout) {

static int start_daemon() {
struct nw instance = {};

INIT_LIST_HEAD(&instance.enabled_interfaces);

uloop_init();

Expand Down
6 changes: 6 additions & 0 deletions src/daemon.h
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,15 @@

#include "node-whisperer.h"

struct nw_enabled_interface {
struct list_head list;
char *name;
};

struct nw {
struct ubus_context *ubus_ctx;
struct uloop_timeout update_timeout;
struct list_head enabled_interfaces;

struct {
uint8_t *buf;
Expand Down
20 changes: 20 additions & 0 deletions src/interface.c
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
#include "node-whisperer.h"
#include "daemon.h"

static LIST_HEAD(nw_interfaces);
static struct blob_buf b;
Expand All @@ -23,8 +24,22 @@ static void nw_interface_handle_remove(struct ubus_context *ctx,
nw_interface_remove(ctx, iface);
}

static int nw_interface_enabled(struct nw *instance, const char *name)
{
struct nw_enabled_interface *iface;

list_for_each_entry(iface, &instance->enabled_interfaces, list) {
if (!strcmp(iface->name, name)) {
return 1;
}
}

return 0;
}

int nw_interface_update(struct ubus_context *ctx, char *vendor_elements)
{
struct nw *instance = container_of(ctx, struct nw, ubus_ctx);
struct nw_interface *iface;
int ret;

Expand All @@ -40,6 +55,11 @@ int nw_interface_update(struct ubus_context *ctx, char *vendor_elements)
/* Delete element */
blob_buf_init(&b, 0);
blobmsg_add_string(&b, "vendor_elements", "");

if (!nw_interface_enabled(instance, iface->ubus.name)) {
continue;
}

ret = ubus_invoke(ctx, iface->ubus.id, "set_vendor_elements", b.head, NULL, NULL, 1000);
if (ret) {
log_error("Failed to reset vendor elements for id=%d name=%s code=%d", iface->ubus.id, iface->ubus.name, ret);
Expand Down
57 changes: 57 additions & 0 deletions src/ubus.c
Original file line number Diff line number Diff line change
@@ -1,10 +1,66 @@
#include <string.h>

#include "node-whisperer.h"
#include "daemon.h"

extern struct nw_information_source information_sources[];

static struct blob_buf b;

static struct blobmsg_policy set_interfaces_policy[] = {
{ .name = "interfaces", .type = BLOBMSG_TYPE_ARRAY, },
};

static int nw_ubus_set_interfaces(struct ubus_context *ctx, struct ubus_object *obj,
struct ubus_request_data *req, const char *method,
struct blob_attr *msg)
{
struct nw *nw = container_of(ctx, struct nw, ubus_ctx);
struct nw_enabled_interface *iface;
const char *ifname;
struct blob_attr *cur;
struct blob_attr *tb[1];
size_t rem;

/* Parse message */
blobmsg_parse(set_interfaces_policy, ARRAY_SIZE(set_interfaces_policy), tb, blob_data(msg), blob_len(msg));
if (!tb[0])
return UBUS_STATUS_INVALID_ARGUMENT;

/* Clear list */
while (!list_empty(&nw->enabled_interfaces)) {
iface = list_first_entry(&nw->enabled_interfaces, struct nw_enabled_interface, list);
list_del(&iface->list);
free(iface->name);
free(iface);
}

/* Add new interfaces */
blobmsg_for_each_attr(cur, tb[0], rem) {
if (blobmsg_type(cur) != BLOBMSG_TYPE_STRING)
return UBUS_STATUS_INVALID_ARGUMENT;

ifname = blobmsg_get_string(cur);
if (!ifname)
return UBUS_STATUS_INVALID_ARGUMENT;

iface = calloc(1, sizeof(*iface));
if (!iface)
return UBUS_STATUS_UNKNOWN_ERROR;

iface->name = strdup(ifname);
if (!iface->name) {
free(iface);
return UBUS_STATUS_UNKNOWN_ERROR;
}

list_add_tail(&iface->list, &nw->enabled_interfaces);
}

return 0;
}


static struct blobmsg_policy source_toggle_arg[] = {
{ .name = "information_sources", .type = BLOBMSG_TYPE_ARRAY, },
};
Expand Down Expand Up @@ -109,6 +165,7 @@ static int nw_ubus_statistics(struct ubus_context *ctx, struct ubus_object *obj,
}

static const struct ubus_method nw_methods[] = {
UBUS_METHOD("set_interfaces", nw_ubus_set_interfaces, set_interfaces_policy),
UBUS_METHOD("set_sources", nw_ubus_enable_source, source_toggle_arg),
UBUS_METHOD_NOARG("get_sources", nw_ubus_get_sources),
UBUS_METHOD_NOARG("statistics", nw_ubus_statistics),
Expand Down

0 comments on commit f7dadd8

Please sign in to comment.