4 static struct avl_tree patterns
;
5 static LIST_HEAD(catch_all
);
6 static struct ubus_object
*event_obj
;
7 static int event_seq
= 0;
8 static int obj_event_seq
= 0;
16 struct list_head list
;
17 struct ubus_object
*obj
;
25 struct list_head list
;
30 static void ubusd_delete_event_source(struct event_source
*evs
)
35 avl_delete(&patterns
, &evs
->pattern
.avl
);
38 list_del(&evs
->catchall
.list
);
44 void ubusd_event_cleanup_object(struct ubus_object
*obj
)
46 struct event_source
*ev
;
48 while (!list_empty(&obj
->events
)) {
49 ev
= list_first_entry(&obj
->events
, struct event_source
, list
);
50 ubusd_delete_event_source(ev
);
60 static struct blobmsg_policy evr_policy
[] = {
61 [EVREG_PATTERN
] = { .name
= "pattern", .type
= BLOBMSG_TYPE_STRING
},
62 [EVREG_OBJECT
] = { .name
= "object", .type
= BLOBMSG_TYPE_INT32
},
66 static struct event_source
*ubusd_alloc_event_source(struct ubus_object
*obj
, enum evs_type type
, int datalen
)
68 struct event_source
*evs
;
70 evs
= calloc(1, sizeof(*evs
) + datalen
);
71 list_add(&evs
->list
, &obj
->events
);
77 static int ubusd_alloc_catchall(struct ubus_object
*obj
)
79 struct event_source
*evs
;
81 evs
= ubusd_alloc_event_source(obj
, EVS_CATCHALL
, 0);
82 list_add(&evs
->catchall
.list
, &catch_all
);
87 static int ubusd_alloc_event_pattern(struct ubus_client
*cl
, struct blob_attr
*msg
)
89 struct event_source
*ev
;
90 struct ubus_object
*obj
;
91 struct blob_attr
*attr
[EVREG_LAST
];
97 blobmsg_parse(evr_policy
, EVREG_LAST
, attr
, blob_data(msg
), blob_len(msg
));
98 if (!attr
[EVREG_OBJECT
])
99 return UBUS_STATUS_INVALID_ARGUMENT
;
101 id
= blobmsg_get_u32(attr
[EVREG_OBJECT
]);
102 if (id
< UBUS_SYSTEM_OBJECT_MAX
)
103 return UBUS_STATUS_PERMISSION_DENIED
;
105 obj
= ubusd_find_object(id
);
107 return UBUS_STATUS_NOT_FOUND
;
109 if (obj
->client
!= cl
)
110 return UBUS_STATUS_PERMISSION_DENIED
;
112 if (!attr
[EVREG_PATTERN
])
113 return ubusd_alloc_catchall(obj
);
115 pattern
= blobmsg_data(attr
[EVREG_PATTERN
]);
117 len
= strlen(pattern
);
118 if (pattern
[len
- 1] == '*') {
120 pattern
[len
- 1] = 0;
124 ev
= ubusd_alloc_event_source(obj
, EVS_PATTERN
, len
+ 1);
125 ev
->pattern
.partial
= partial
;
126 ev
->pattern
.avl
.key
= (void *) (ev
+ 1);
127 strcpy(ev
->pattern
.avl
.key
, pattern
);
128 avl_insert(&patterns
, &ev
->pattern
.avl
);
133 typedef struct ubus_msg_buf
*(*event_fill_cb
)(void *priv
, const char *id
);
135 static void ubusd_send_event_msg(struct ubus_msg_buf
**ub
, struct ubus_client
*cl
,
136 struct ubus_object
*obj
, const char *id
,
137 event_fill_cb fill_cb
, void *cb_priv
)
141 /* do not loop back events */
142 if (obj
->client
== cl
)
145 /* do not send duplicate events */
146 if (obj
->event_seen
== obj_event_seq
)
149 obj
->event_seen
= obj_event_seq
;
152 *ub
= fill_cb(cb_priv
, id
);
153 (*ub
)->hdr
.type
= UBUS_MSG_INVOKE
;
157 objid_ptr
= blob_data(blob_data((*ub
)->data
));
158 *objid_ptr
= htonl(obj
->id
.id
);
160 (*ub
)->hdr
.seq
= ++event_seq
;
161 ubus_msg_send(obj
->client
, *ub
, false);
164 bool strmatch_len(const char *s1
, const char *s2
, int *len
)
166 for (*len
= 0; s1
[*len
] == s2
[*len
]; (*len
)++)
173 static int ubusd_send_event(struct ubus_client
*cl
, const char *id
,
174 event_fill_cb fill_cb
, void *cb_priv
)
176 struct ubus_msg_buf
*ub
= NULL
;
177 struct event_source
*ev
;
180 list_for_each_entry(ev
, &catch_all
, catchall
.list
)
181 ubusd_send_event_msg(&ub
, cl
, ev
->obj
, id
, fill_cb
, cb_priv
);
186 * Since this tree is sorted alphabetically, we can only expect to find
187 * matching entries as long as the number of matching characters
188 * between the pattern string and our string is monotonically increasing.
190 avl_for_each_element(&patterns
, ev
, pattern
.avl
) {
191 const char *key
= ev
->pattern
.avl
.key
;
195 full_match
= strmatch_len(id
, key
, &cur_match_len
);
196 if (cur_match_len
< match_len
)
199 match_len
= cur_match_len
;
202 if (!ev
->pattern
.partial
)
205 if (match_len
!= strlen(key
))
209 ubusd_send_event_msg(&ub
, cl
, ev
->obj
, id
, fill_cb
, cb_priv
);
224 static struct blobmsg_policy ev_policy
[] = {
225 [EVMSG_ID
] = { .name
= "id", .type
= BLOBMSG_TYPE_STRING
},
226 [EVMSG_DATA
] = { .name
= "data", .type
= BLOBMSG_TYPE_TABLE
},
229 static struct ubus_msg_buf
*
230 ubusd_create_event_from_msg(void *priv
, const char *id
)
232 struct blob_attr
*msg
= priv
;
234 blob_buf_init(&b
, 0);
235 blob_put_int32(&b
, UBUS_ATTR_OBJID
, 0);
236 blob_put_string(&b
, UBUS_ATTR_METHOD
, id
);
237 blob_put(&b
, UBUS_ATTR_DATA
, blobmsg_data(msg
), blobmsg_data_len(msg
));
239 return ubus_msg_new(b
.head
, blob_raw_len(b
.head
), true);
242 static int ubusd_forward_event(struct ubus_client
*cl
, struct blob_attr
*msg
)
244 struct blob_attr
*data
;
245 struct blob_attr
*attr
[EVMSG_LAST
];
248 blobmsg_parse(ev_policy
, EVMSG_LAST
, attr
, blob_data(msg
), blob_len(msg
));
249 if (!attr
[EVMSG_ID
] || !attr
[EVMSG_DATA
])
250 return UBUS_STATUS_INVALID_ARGUMENT
;
252 id
= blobmsg_data(attr
[EVMSG_ID
]);
253 data
= attr
[EVMSG_DATA
];
255 if (!strncmp(id
, "ubus.", 5))
256 return UBUS_STATUS_PERMISSION_DENIED
;
258 return ubusd_send_event(cl
, id
, ubusd_create_event_from_msg
, data
);
261 static int ubusd_event_recv(struct ubus_client
*cl
, const char *method
, struct blob_attr
*msg
)
263 if (!strcmp(method
, "register"))
264 return ubusd_alloc_event_pattern(cl
, msg
);
266 if (!strcmp(method
, "send"))
267 return ubusd_forward_event(cl
, msg
);
269 return UBUS_STATUS_INVALID_COMMAND
;
272 static struct ubus_msg_buf
*
273 ubusd_create_object_event_msg(void *priv
, const char *id
)
275 struct ubus_object
*obj
= priv
;
278 blob_buf_init(&b
, 0);
279 blob_put_int32(&b
, UBUS_ATTR_OBJID
, 0);
280 blob_put_string(&b
, UBUS_ATTR_METHOD
, id
);
281 s
= blob_nest_start(&b
, UBUS_ATTR_DATA
);
282 blobmsg_add_u32(&b
, "id", obj
->id
.id
);
283 blobmsg_add_string(&b
, "path", obj
->path
.key
);
284 blob_nest_end(&b
, s
);
286 return ubus_msg_new(b
.head
, blob_raw_len(b
.head
), true);
289 void ubusd_send_obj_event(struct ubus_object
*obj
, bool add
)
291 const char *id
= add
? "ubus.object.add" : "ubus.object.remove";
293 ubusd_send_event(NULL
, id
, ubusd_create_object_event_msg
, obj
);
296 void ubusd_event_init(void)
298 ubus_init_string_tree(&patterns
, true);
299 event_obj
= ubusd_create_object_internal(NULL
, UBUS_SYSTEM_OBJECT_EVENT
);
300 event_obj
->recv_msg
= ubusd_event_recv
;