libubus: restore uloop_cancelled on timeout
[project/ubus.git] / ubusd_event.c
1 /*
2  * Copyright (C) 2011 Felix Fietkau <nbd@openwrt.org>
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU Lesser General Public License version 2.1
6  * as published by the Free Software Foundation
7  *
8  * This program is distributed in the hope that it will be useful,
9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11  * GNU General Public License for more details.
12  */
13
14 #include <arpa/inet.h>
15 #include "ubusd.h"
16
17 static struct avl_tree patterns;
18 static struct ubus_object *event_obj;
19 static int event_seq = 0;
20 static int obj_event_seq = 1;
21
22 struct event_source {
23         struct list_head list;
24         struct ubus_object *obj;
25         struct avl_node avl;
26         bool partial;
27 };
28
29 static void ubusd_delete_event_source(struct event_source *evs)
30 {
31         list_del(&evs->list);
32         avl_delete(&patterns, &evs->avl);
33         free(evs);
34 }
35
36 void ubusd_event_cleanup_object(struct ubus_object *obj)
37 {
38         struct event_source *ev;
39
40         while (!list_empty(&obj->events)) {
41                 ev = list_first_entry(&obj->events, struct event_source, list);
42                 ubusd_delete_event_source(ev);
43         }
44 }
45
46 enum {
47         EVREG_PATTERN,
48         EVREG_OBJECT,
49         EVREG_LAST,
50 };
51
52 static struct blobmsg_policy evr_policy[] = {
53         [EVREG_PATTERN] = { .name = "pattern", .type = BLOBMSG_TYPE_STRING },
54         [EVREG_OBJECT] = { .name = "object", .type = BLOBMSG_TYPE_INT32 },
55 };
56
57 static int ubusd_alloc_event_pattern(struct ubus_client *cl, struct blob_attr *msg)
58 {
59         struct event_source *ev;
60         struct ubus_object *obj;
61         struct blob_attr *attr[EVREG_LAST];
62         char *pattern, *name;
63         uint32_t id;
64         bool partial = false;
65         int len;
66
67         blobmsg_parse(evr_policy, EVREG_LAST, attr, blob_data(msg), blob_len(msg));
68         if (!attr[EVREG_OBJECT] || !attr[EVREG_PATTERN])
69                 return UBUS_STATUS_INVALID_ARGUMENT;
70
71         id = blobmsg_get_u32(attr[EVREG_OBJECT]);
72         if (id < UBUS_SYSTEM_OBJECT_MAX)
73                 return UBUS_STATUS_PERMISSION_DENIED;
74
75         obj = ubusd_find_object(id);
76         if (!obj)
77                 return UBUS_STATUS_NOT_FOUND;
78
79         if (obj->client != cl)
80                 return UBUS_STATUS_PERMISSION_DENIED;
81
82         pattern = blobmsg_data(attr[EVREG_PATTERN]);
83
84         len = strlen(pattern);
85         if (pattern[len - 1] == '*') {
86                 partial = true;
87                 pattern[len - 1] = 0;
88                 len--;
89         }
90
91         ev = calloc(1, sizeof(*ev) + len + 1);
92         if (!ev)
93                 return UBUS_STATUS_NO_DATA;
94
95         list_add(&ev->list, &obj->events);
96         ev->obj = obj;
97         ev->partial = partial;
98         name = (char *) (ev + 1);
99         strcpy(name, pattern);
100         ev->avl.key = name;
101         avl_insert(&patterns, &ev->avl);
102
103         return 0;
104 }
105
106 typedef struct ubus_msg_buf *(*event_fill_cb)(void *priv, const char *id);
107
108 static void ubusd_send_event_msg(struct ubus_msg_buf **ub, struct ubus_client *cl,
109                                  struct ubus_object *obj, const char *id,
110                                  event_fill_cb fill_cb, void *cb_priv)
111 {
112         uint32_t *objid_ptr;
113
114         /* do not loop back events */
115         if (obj->client == cl)
116             return;
117
118         /* do not send duplicate events */
119         if (obj->event_seen == obj_event_seq)
120                 return;
121
122         obj->event_seen = obj_event_seq;
123
124         if (!*ub) {
125                 *ub = fill_cb(cb_priv, id);
126                 (*ub)->hdr.type = UBUS_MSG_INVOKE;
127                 (*ub)->hdr.peer = 0;
128         }
129
130         objid_ptr = blob_data(blob_data((*ub)->data));
131         *objid_ptr = htonl(obj->id.id);
132
133         (*ub)->hdr.seq = ++event_seq;
134         ubus_msg_send(obj->client, *ub, false);
135 }
136
137 static bool strmatch_len(const char *s1, const char *s2, int *len)
138 {
139         for (*len = 0; s1[*len] == s2[*len]; (*len)++)
140                 if (!s1[*len])
141                         return true;
142
143         return false;
144 }
145
146 static int ubusd_send_event(struct ubus_client *cl, const char *id,
147                             event_fill_cb fill_cb, void *cb_priv)
148 {
149         struct ubus_msg_buf *ub = NULL;
150         struct event_source *ev;
151         int match_len = 0;
152
153         obj_event_seq++;
154
155         /*
156          * Since this tree is sorted alphabetically, we can only expect to find
157          * matching entries as long as the number of matching characters
158          * between the pattern string and our string is monotonically increasing.
159          */
160         avl_for_each_element(&patterns, ev, avl) {
161                 const char *key = ev->avl.key;
162                 int cur_match_len;
163                 bool full_match;
164
165                 full_match = strmatch_len(id, key, &cur_match_len);
166                 if (cur_match_len < match_len)
167                         break;
168
169                 match_len = cur_match_len;
170
171                 if (!full_match) {
172                         if (!ev->partial)
173                                 continue;
174
175                         if (match_len != strlen(key))
176                                 continue;
177                 }
178
179                 ubusd_send_event_msg(&ub, cl, ev->obj, id, fill_cb, cb_priv);
180         }
181
182         if (ub)
183                 ubus_msg_free(ub);
184
185         return 0;
186 }
187
188 enum {
189         EVMSG_ID,
190         EVMSG_DATA,
191         EVMSG_LAST,
192 };
193
194 static struct blobmsg_policy ev_policy[] = {
195         [EVMSG_ID] = { .name = "id", .type = BLOBMSG_TYPE_STRING },
196         [EVMSG_DATA] = { .name = "data", .type = BLOBMSG_TYPE_TABLE },
197 };
198
199 static struct ubus_msg_buf *
200 ubusd_create_event_from_msg(void *priv, const char *id)
201 {
202         struct blob_attr *msg = priv;
203
204         blob_buf_init(&b, 0);
205         blob_put_int32(&b, UBUS_ATTR_OBJID, 0);
206         blob_put_string(&b, UBUS_ATTR_METHOD, id);
207         blob_put(&b, UBUS_ATTR_DATA, blobmsg_data(msg), blobmsg_data_len(msg));
208
209         return ubus_msg_new(b.head, blob_raw_len(b.head), true);
210 }
211
212 static int ubusd_forward_event(struct ubus_client *cl, struct blob_attr *msg)
213 {
214         struct blob_attr *data;
215         struct blob_attr *attr[EVMSG_LAST];
216         const char *id;
217
218         blobmsg_parse(ev_policy, EVMSG_LAST, attr, blob_data(msg), blob_len(msg));
219         if (!attr[EVMSG_ID] || !attr[EVMSG_DATA])
220                 return UBUS_STATUS_INVALID_ARGUMENT;
221
222         id = blobmsg_data(attr[EVMSG_ID]);
223         data = attr[EVMSG_DATA];
224
225         if (!strncmp(id, "ubus.", 5))
226                 return UBUS_STATUS_PERMISSION_DENIED;
227
228         return ubusd_send_event(cl, id, ubusd_create_event_from_msg, data);
229 }
230
231 static int ubusd_event_recv(struct ubus_client *cl, const char *method, struct blob_attr *msg)
232 {
233         if (!strcmp(method, "register"))
234                 return ubusd_alloc_event_pattern(cl, msg);
235
236         if (!strcmp(method, "send"))
237                 return ubusd_forward_event(cl, msg);
238
239         return UBUS_STATUS_INVALID_COMMAND;
240 }
241
242 static struct ubus_msg_buf *
243 ubusd_create_object_event_msg(void *priv, const char *id)
244 {
245         struct ubus_object *obj = priv;
246         void *s;
247
248         blob_buf_init(&b, 0);
249         blob_put_int32(&b, UBUS_ATTR_OBJID, 0);
250         blob_put_string(&b, UBUS_ATTR_METHOD, id);
251         s = blob_nest_start(&b, UBUS_ATTR_DATA);
252         blobmsg_add_u32(&b, "id", obj->id.id);
253         blobmsg_add_string(&b, "path", obj->path.key);
254         blob_nest_end(&b, s);
255
256         return ubus_msg_new(b.head, blob_raw_len(b.head), true);
257 }
258
259 void ubusd_send_obj_event(struct ubus_object *obj, bool add)
260 {
261         const char *id = add ? "ubus.object.add" : "ubus.object.remove";
262
263         ubusd_send_event(NULL, id, ubusd_create_object_event_msg, obj);
264 }
265
266 void ubusd_event_init(void)
267 {
268         ubus_init_string_tree(&patterns, true);
269         event_obj = ubusd_create_object_internal(NULL, UBUS_SYSTEM_OBJECT_EVENT);
270         event_obj->recv_msg = ubusd_event_recv;
271 }
272