+/*
+ * Copyright (C) 2011 Felix Fietkau <nbd@openwrt.org>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License version 2.1
+ * as published by the Free Software Foundation
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ */
+
#include <arpa/inet.h>
#include "ubusd.h"
static struct avl_tree patterns;
-static LIST_HEAD(catch_all);
static struct ubus_object *event_obj;
static int event_seq = 0;
-static int obj_event_seq = 0;
-
-enum evs_type {
- EVS_PATTERN,
- EVS_CATCHALL
-};
+static int obj_event_seq = 1;
struct event_source {
struct list_head list;
struct ubus_object *obj;
- enum evs_type type;
- union {
- struct {
- struct avl_node avl;
- bool partial;
- } pattern;
- struct {
- struct list_head list;
- } catchall;
- };
+ struct avl_node avl;
+ bool partial;
};
static void ubusd_delete_event_source(struct event_source *evs)
{
list_del(&evs->list);
- switch (evs->type) {
- case EVS_PATTERN:
- avl_delete(&patterns, &evs->pattern.avl);
- break;
- case EVS_CATCHALL:
- list_del(&evs->catchall.list);
- break;
- }
+ avl_delete(&patterns, &evs->avl);
free(evs);
}
[EVREG_OBJECT] = { .name = "object", .type = BLOBMSG_TYPE_INT32 },
};
-
-static struct event_source *ubusd_alloc_event_source(struct ubus_object *obj, enum evs_type type, int datalen)
-{
- struct event_source *evs;
-
- evs = calloc(1, sizeof(*evs) + datalen);
- list_add(&evs->list, &obj->events);
- evs->obj = obj;
- evs->type = type;
- return evs;
-}
-
-static int ubusd_alloc_catchall(struct ubus_object *obj)
-{
- struct event_source *evs;
-
- evs = ubusd_alloc_event_source(obj, EVS_CATCHALL, 0);
- list_add(&evs->catchall.list, &catch_all);
-
- return 0;
-}
-
static int ubusd_alloc_event_pattern(struct ubus_client *cl, struct blob_attr *msg)
{
struct event_source *ev;
struct ubus_object *obj;
struct blob_attr *attr[EVREG_LAST];
- char *pattern;
+ char *pattern, *name;
uint32_t id;
bool partial = false;
int len;
blobmsg_parse(evr_policy, EVREG_LAST, attr, blob_data(msg), blob_len(msg));
- if (!attr[EVREG_OBJECT])
+ if (!attr[EVREG_OBJECT] || !attr[EVREG_PATTERN])
return UBUS_STATUS_INVALID_ARGUMENT;
id = blobmsg_get_u32(attr[EVREG_OBJECT]);
if (obj->client != cl)
return UBUS_STATUS_PERMISSION_DENIED;
- if (!attr[EVREG_PATTERN])
- return ubusd_alloc_catchall(obj);
-
pattern = blobmsg_data(attr[EVREG_PATTERN]);
len = strlen(pattern);
len--;
}
- ev = ubusd_alloc_event_source(obj, EVS_PATTERN, len + 1);
- ev->pattern.partial = partial;
- ev->pattern.avl.key = (void *) (ev + 1);
- strcpy(ev->pattern.avl.key, pattern);
- avl_insert(&patterns, &ev->pattern.avl);
+ ev = calloc(1, sizeof(*ev) + len + 1);
+ if (!ev)
+ return UBUS_STATUS_NO_DATA;
+
+ list_add(&ev->list, &obj->events);
+ ev->obj = obj;
+ ev->partial = partial;
+ name = (char *) (ev + 1);
+ strcpy(name, pattern);
+ ev->avl.key = name;
+ avl_insert(&patterns, &ev->avl);
return 0;
}
-enum {
- EVMSG_ID,
- EVMSG_DATA,
- EVMSG_LAST,
-};
-
-static struct blobmsg_policy ev_policy[] = {
- [EVMSG_ID] = { .name = "id", .type = BLOBMSG_TYPE_STRING },
- [EVMSG_DATA] = { .name = "data", .type = BLOBMSG_TYPE_TABLE },
-};
-
static void ubusd_send_event_msg(struct ubus_msg_buf **ub, struct ubus_client *cl,
struct ubus_object *obj, const char *id,
- struct blob_attr *msg)
+ event_fill_cb fill_cb, void *cb_priv)
{
uint32_t *objid_ptr;
obj->event_seen = obj_event_seq;
- if (*ub) {
- objid_ptr = blob_data(blob_data((*ub)->data));
- *objid_ptr = htonl(obj->id.id);
- } else {
- blob_buf_init(&b, 0);
- blob_put_int32(&b, UBUS_ATTR_OBJID, obj->id.id);
- blob_put_string(&b, UBUS_ATTR_METHOD, id);
- blob_put(&b, UBUS_ATTR_DATA, blobmsg_data(msg), blobmsg_data_len(msg));
-
- *ub = ubus_msg_new(b.head, blob_raw_len(b.head), true);
-
+ if (!*ub) {
+ *ub = fill_cb(cb_priv, id);
(*ub)->hdr.type = UBUS_MSG_INVOKE;
(*ub)->hdr.peer = 0;
}
+
+ objid_ptr = blob_data(blob_data((*ub)->data));
+ *objid_ptr = htonl(obj->id.id);
+
(*ub)->hdr.seq = ++event_seq;
ubus_msg_send(obj->client, *ub, false);
}
-bool strmatch_len(const char *s1, const char *s2, int *len)
+static bool strmatch_len(const char *s1, const char *s2, int *len)
{
for (*len = 0; s1[*len] == s2[*len]; (*len)++)
if (!s1[*len])
return false;
}
-static int ubusd_send_event(struct ubus_client *cl, struct blob_attr *msg)
+int ubusd_send_event(struct ubus_client *cl, const char *id,
+ event_fill_cb fill_cb, void *cb_priv)
{
struct ubus_msg_buf *ub = NULL;
struct event_source *ev;
- struct blob_attr *attr[EVMSG_LAST];
- const char *id;
int match_len = 0;
- void *data;
-
- blobmsg_parse(ev_policy, EVMSG_LAST, attr, blob_data(msg), blob_len(msg));
- if (!attr[EVMSG_ID] || !attr[EVMSG_DATA])
- return UBUS_STATUS_INVALID_ARGUMENT;
-
- id = blobmsg_data(attr[EVMSG_ID]);
- data = attr[EVMSG_DATA];
-
- list_for_each_entry(ev, &catch_all, catchall.list)
- ubusd_send_event_msg(&ub, cl, ev->obj, id, data);
obj_event_seq++;
* matching entries as long as the number of matching characters
* between the pattern string and our string is monotonically increasing.
*/
- avl_for_each_element(&patterns, ev, pattern.avl) {
- const char *key = ev->pattern.avl.key;
+ avl_for_each_element(&patterns, ev, avl) {
+ const char *key = ev->avl.key;
int cur_match_len;
bool full_match;
match_len = cur_match_len;
if (!full_match) {
- if (!ev->pattern.partial)
+ if (!ev->partial)
continue;
if (match_len != strlen(key))
continue;
}
- ubusd_send_event_msg(&ub, cl, ev->obj, id, data);
+ ubusd_send_event_msg(&ub, cl, ev->obj, id, fill_cb, cb_priv);
}
if (ub)
return 0;
}
-static int ubusd_event_recv(struct ubus_client *cl, const char *method, struct blob_attr *msg)
+enum {
+ EVMSG_ID,
+ EVMSG_DATA,
+ EVMSG_LAST,
+};
+
+static struct blobmsg_policy ev_policy[] = {
+ [EVMSG_ID] = { .name = "id", .type = BLOBMSG_TYPE_STRING },
+ [EVMSG_DATA] = { .name = "data", .type = BLOBMSG_TYPE_TABLE },
+};
+
+static struct ubus_msg_buf *
+ubusd_create_event_from_msg(void *priv, const char *id)
+{
+ struct blob_attr *msg = priv;
+
+ blob_buf_init(&b, 0);
+ blob_put_int32(&b, UBUS_ATTR_OBJID, 0);
+ blob_put_string(&b, UBUS_ATTR_METHOD, id);
+ blob_put(&b, UBUS_ATTR_DATA, blobmsg_data(msg), blobmsg_data_len(msg));
+
+ return ubus_msg_new(b.head, blob_raw_len(b.head), true);
+}
+
+static int ubusd_forward_event(struct ubus_client *cl, struct blob_attr *msg)
+{
+ struct blob_attr *data;
+ struct blob_attr *attr[EVMSG_LAST];
+ const char *id;
+
+ blobmsg_parse(ev_policy, EVMSG_LAST, attr, blob_data(msg), blob_len(msg));
+ if (!attr[EVMSG_ID] || !attr[EVMSG_DATA])
+ return UBUS_STATUS_INVALID_ARGUMENT;
+
+ id = blobmsg_data(attr[EVMSG_ID]);
+ data = attr[EVMSG_DATA];
+
+ if (!strncmp(id, "ubus.", 5))
+ return UBUS_STATUS_PERMISSION_DENIED;
+
+ return ubusd_send_event(cl, id, ubusd_create_event_from_msg, data);
+}
+
+static int ubusd_event_recv(struct ubus_client *cl, struct ubus_msg_buf *ub, const char *method, struct blob_attr *msg)
{
if (!strcmp(method, "register"))
return ubusd_alloc_event_pattern(cl, msg);
if (!strcmp(method, "send"))
- return ubusd_send_event(cl, msg);
+ return ubusd_forward_event(cl, msg);
return UBUS_STATUS_INVALID_COMMAND;
}
+static struct ubus_msg_buf *
+ubusd_create_object_event_msg(void *priv, const char *id)
+{
+ struct ubus_object *obj = priv;
+ void *s;
+
+ blob_buf_init(&b, 0);
+ blob_put_int32(&b, UBUS_ATTR_OBJID, 0);
+ blob_put_string(&b, UBUS_ATTR_METHOD, id);
+ s = blob_nest_start(&b, UBUS_ATTR_DATA);
+ blobmsg_add_u32(&b, "id", obj->id.id);
+ blobmsg_add_string(&b, "path", obj->path.key);
+ blob_nest_end(&b, s);
+
+ return ubus_msg_new(b.head, blob_raw_len(b.head), true);
+}
+
+void ubusd_send_obj_event(struct ubus_object *obj, bool add)
+{
+ const char *id = add ? "ubus.object.add" : "ubus.object.remove";
+
+ ubusd_send_event(NULL, id, ubusd_create_object_event_msg, obj);
+}
+
void ubusd_event_init(void)
{
ubus_init_string_tree(&patterns, true);
event_obj = ubusd_create_object_internal(NULL, UBUS_SYSTEM_OBJECT_EVENT);
- event_obj->recv_msg = ubusd_event_recv;
+ if (event_obj != NULL)
+ event_obj->recv_msg = ubusd_event_recv;
}