25777a60feee0d1a479d3e44b09c3948cab91f0c
[project/rpcd.git] / session.c
1 /*
2  * rpcd - UBUS RPC server
3  *
4  *   Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
5  *   Copyright (C) 2013-2014 Jo-Philipp Wich <jow@openwrt.org>
6  *
7  * Permission to use, copy, modify, and/or distribute this software for any
8  * purpose with or without fee is hereby granted, provided that the above
9  * copyright notice and this permission notice appear in all copies.
10  *
11  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
12  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
14  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
16  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
17  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18  */
19
20 #define _GNU_SOURCE     /* crypt() */
21
22 #include <libubox/avl-cmp.h>
23 #include <libubox/blobmsg.h>
24 #include <libubox/utils.h>
25 #include <libubus.h>
26 #include <fnmatch.h>
27 #include <glob.h>
28 #include <uci.h>
29
30 #ifdef HAVE_SHADOW
31 #include <shadow.h>
32 #endif
33
34 #include <rpcd/session.h>
35
36 static struct avl_tree sessions;
37 static struct blob_buf buf;
38
39 static LIST_HEAD(create_callbacks);
40 static LIST_HEAD(destroy_callbacks);
41
42 static const struct blobmsg_policy new_policy = {
43         .name = "timeout", .type = BLOBMSG_TYPE_INT32
44 };
45
46 static const struct blobmsg_policy sid_policy = {
47         .name = "ubus_rpc_session", .type = BLOBMSG_TYPE_STRING
48 };
49
50 enum {
51         RPC_SS_SID,
52         RPC_SS_VALUES,
53         __RPC_SS_MAX,
54 };
55 static const struct blobmsg_policy set_policy[__RPC_SS_MAX] = {
56         [RPC_SS_SID] = { .name = "ubus_rpc_session", .type = BLOBMSG_TYPE_STRING },
57         [RPC_SS_VALUES] = { .name = "values", .type = BLOBMSG_TYPE_TABLE },
58 };
59
60 enum {
61         RPC_SG_SID,
62         RPC_SG_KEYS,
63         __RPC_SG_MAX,
64 };
65 static const struct blobmsg_policy get_policy[__RPC_SG_MAX] = {
66         [RPC_SG_SID] = { .name = "ubus_rpc_session", .type = BLOBMSG_TYPE_STRING },
67         [RPC_SG_KEYS] = { .name = "keys", .type = BLOBMSG_TYPE_ARRAY },
68 };
69
70 enum {
71         RPC_SA_SID,
72         RPC_SA_SCOPE,
73         RPC_SA_OBJECTS,
74         __RPC_SA_MAX,
75 };
76 static const struct blobmsg_policy acl_policy[__RPC_SA_MAX] = {
77         [RPC_SA_SID] = { .name = "ubus_rpc_session", .type = BLOBMSG_TYPE_STRING },
78         [RPC_SA_SCOPE] = { .name = "scope", .type = BLOBMSG_TYPE_STRING },
79         [RPC_SA_OBJECTS] = { .name = "objects", .type = BLOBMSG_TYPE_ARRAY },
80 };
81
82 enum {
83         RPC_SP_SID,
84         RPC_SP_SCOPE,
85         RPC_SP_OBJECT,
86         RPC_SP_FUNCTION,
87         __RPC_SP_MAX,
88 };
89 static const struct blobmsg_policy perm_policy[__RPC_SP_MAX] = {
90         [RPC_SP_SID] = { .name = "ubus_rpc_session", .type = BLOBMSG_TYPE_STRING },
91         [RPC_SP_SCOPE] = { .name = "scope", .type = BLOBMSG_TYPE_STRING },
92         [RPC_SP_OBJECT] = { .name = "object", .type = BLOBMSG_TYPE_STRING },
93         [RPC_SP_FUNCTION] = { .name = "function", .type = BLOBMSG_TYPE_STRING },
94 };
95
96 enum {
97         RPC_DUMP_SID,
98         RPC_DUMP_TIMEOUT,
99         RPC_DUMP_EXPIRES,
100         RPC_DUMP_DATA,
101         __RPC_DUMP_MAX,
102 };
103 static const struct blobmsg_policy dump_policy[__RPC_DUMP_MAX] = {
104         [RPC_DUMP_SID] = { .name = "ubus_rpc_session", .type = BLOBMSG_TYPE_STRING },
105         [RPC_DUMP_TIMEOUT] = { .name = "timeout", .type = BLOBMSG_TYPE_INT32 },
106         [RPC_DUMP_EXPIRES] = { .name = "expires", .type = BLOBMSG_TYPE_INT32 },
107         [RPC_DUMP_DATA] = { .name = "data", .type = BLOBMSG_TYPE_TABLE },
108 };
109
110 enum {
111         RPC_L_USERNAME,
112         RPC_L_PASSWORD,
113         RPC_L_TIMEOUT,
114         __RPC_L_MAX,
115 };
116 static const struct blobmsg_policy login_policy[__RPC_L_MAX] = {
117         [RPC_L_USERNAME] = { .name = "username", .type = BLOBMSG_TYPE_STRING },
118         [RPC_L_PASSWORD] = { .name = "password", .type = BLOBMSG_TYPE_STRING },
119         [RPC_L_TIMEOUT]  = { .name = "timeout", .type = BLOBMSG_TYPE_INT32 },
120 };
121
122 /*
123  * Keys in the AVL tree contain all pattern characters up to the first wildcard.
124  * To look up entries, start with the last entry that has a key less than or
125  * equal to the method name, then work backwards as long as the AVL key still
126  * matches its counterpart in the object name
127  */
128 #define uh_foreach_matching_acl_prefix(_acl, _avl, _obj, _func)         \
129         for (_acl = avl_find_le_element(_avl, _obj, _acl, avl);                 \
130              _acl;                                                                                                              \
131              _acl = avl_is_first(_avl, &(_acl)->avl) ? NULL :                   \
132                     avl_prev_element((_acl), avl))
133
134 #define uh_foreach_matching_acl(_acl, _avl, _obj, _func)                        \
135         uh_foreach_matching_acl_prefix(_acl, _avl, _obj, _func)                 \
136                 if (!strncmp((_acl)->object, _obj, (_acl)->sort_len) &&         \
137                     !fnmatch((_acl)->object, (_obj), FNM_NOESCAPE) &&           \
138                     !fnmatch((_acl)->function, (_func), FNM_NOESCAPE))
139
140 static void
141 rpc_random(char *dest)
142 {
143         unsigned char buf[16] = { 0 };
144         FILE *f;
145         int i;
146
147         f = fopen("/dev/urandom", "r");
148         if (!f)
149                 return;
150
151         fread(buf, 1, sizeof(buf), f);
152         fclose(f);
153
154         for (i = 0; i < sizeof(buf); i++)
155                 sprintf(dest + (i<<1), "%02x", buf[i]);
156 }
157
158 static void
159 rpc_session_dump_data(struct rpc_session *ses, struct blob_buf *b)
160 {
161         struct rpc_session_data *d;
162
163         avl_for_each_element(&ses->data, d, avl) {
164                 blobmsg_add_field(b, blobmsg_type(d->attr), blobmsg_name(d->attr),
165                                   blobmsg_data(d->attr), blobmsg_data_len(d->attr));
166         }
167 }
168
169 static void
170 rpc_session_dump_acls(struct rpc_session *ses, struct blob_buf *b)
171 {
172         struct rpc_session_acl *acl;
173         struct rpc_session_acl_scope *acl_scope;
174         const char *lastobj = NULL;
175         const char *lastscope = NULL;
176         void *c = NULL, *d = NULL;
177
178         avl_for_each_element(&ses->acls, acl_scope, avl) {
179                 if (!lastscope || strcmp(acl_scope->avl.key, lastscope))
180                 {
181                         if (c) blobmsg_close_table(b, c);
182                         c = blobmsg_open_table(b, acl_scope->avl.key);
183                         lastobj = NULL;
184                 }
185
186                 d = NULL;
187
188                 avl_for_each_element(&acl_scope->acls, acl, avl) {
189                         if (!lastobj || strcmp(acl->object, lastobj))
190                         {
191                                 if (d) blobmsg_close_array(b, d);
192                                 d = blobmsg_open_array(b, acl->object);
193                         }
194
195                         blobmsg_add_string(b, NULL, acl->function);
196                         lastobj = acl->object;
197                 }
198
199                 if (d) blobmsg_close_array(b, d);
200         }
201
202         if (c) blobmsg_close_table(b, c);
203 }
204
205 static void
206 rpc_session_to_blob(struct rpc_session *ses, bool acls)
207 {
208         void *c;
209
210         blob_buf_init(&buf, 0);
211
212         blobmsg_add_string(&buf, "ubus_rpc_session", ses->id);
213         blobmsg_add_u32(&buf, "timeout", ses->timeout);
214         blobmsg_add_u32(&buf, "expires", uloop_timeout_remaining(&ses->t) / 1000);
215
216         if (acls) {
217                 c = blobmsg_open_table(&buf, "acls");
218                 rpc_session_dump_acls(ses, &buf);
219                 blobmsg_close_table(&buf, c);
220         }
221
222         c = blobmsg_open_table(&buf, "data");
223         rpc_session_dump_data(ses, &buf);
224         blobmsg_close_table(&buf, c);
225 }
226
227 static void
228 rpc_session_dump(struct rpc_session *ses, struct ubus_context *ctx,
229                  struct ubus_request_data *req)
230 {
231         rpc_session_to_blob(ses, true);
232
233         ubus_send_reply(ctx, req, buf.head);
234 }
235
236 static void
237 rpc_touch_session(struct rpc_session *ses)
238 {
239         if (ses->timeout > 0)
240                 uloop_timeout_set(&ses->t, ses->timeout * 1000);
241 }
242
243 static void
244 rpc_session_destroy(struct rpc_session *ses)
245 {
246         struct rpc_session_acl *acl, *nacl;
247         struct rpc_session_acl_scope *acl_scope, *nacl_scope;
248         struct rpc_session_data *data, *ndata;
249         struct rpc_session_cb *cb;
250
251         list_for_each_entry(cb, &destroy_callbacks, list)
252                 cb->cb(ses, cb->priv);
253
254         uloop_timeout_cancel(&ses->t);
255
256         avl_for_each_element_safe(&ses->acls, acl_scope, avl, nacl_scope) {
257                 avl_remove_all_elements(&acl_scope->acls, acl, avl, nacl)
258                         free(acl);
259
260                 avl_delete(&ses->acls, &acl_scope->avl);
261                 free(acl_scope);
262         }
263
264         avl_remove_all_elements(&ses->data, data, avl, ndata)
265                 free(data);
266
267         avl_delete(&sessions, &ses->avl);
268         free(ses);
269 }
270
271 static void rpc_session_timeout(struct uloop_timeout *t)
272 {
273         struct rpc_session *ses;
274
275         ses = container_of(t, struct rpc_session, t);
276         rpc_session_destroy(ses);
277 }
278
279 static struct rpc_session *
280 rpc_session_new(void)
281 {
282         struct rpc_session *ses;
283
284         ses = calloc(1, sizeof(*ses));
285
286         if (!ses)
287                 return NULL;
288
289         ses->avl.key = ses->id;
290
291         avl_init(&ses->acls, avl_strcmp, true, NULL);
292         avl_init(&ses->data, avl_strcmp, false, NULL);
293
294         ses->t.cb = rpc_session_timeout;
295
296         return ses;
297 }
298
299 static struct rpc_session *
300 rpc_session_create(int timeout)
301 {
302         struct rpc_session *ses;
303         struct rpc_session_cb *cb;
304
305         ses = rpc_session_new();
306
307         if (!ses)
308                 return NULL;
309
310         rpc_random(ses->id);
311
312         ses->timeout = timeout;
313
314         avl_insert(&sessions, &ses->avl);
315
316         rpc_touch_session(ses);
317
318         list_for_each_entry(cb, &create_callbacks, list)
319                 cb->cb(ses, cb->priv);
320
321         return ses;
322 }
323
324 static struct rpc_session *
325 rpc_session_get(const char *id)
326 {
327         struct rpc_session *ses;
328
329         ses = avl_find_element(&sessions, id, ses, avl);
330         if (!ses)
331                 return NULL;
332
333         rpc_touch_session(ses);
334         return ses;
335 }
336
337 static int
338 rpc_handle_create(struct ubus_context *ctx, struct ubus_object *obj,
339                   struct ubus_request_data *req, const char *method,
340                   struct blob_attr *msg)
341 {
342         struct rpc_session *ses;
343         struct blob_attr *tb;
344         int timeout = RPC_DEFAULT_SESSION_TIMEOUT;
345
346         blobmsg_parse(&new_policy, 1, &tb, blob_data(msg), blob_len(msg));
347         if (tb)
348                 timeout = blobmsg_get_u32(tb);
349
350         ses = rpc_session_create(timeout);
351         if (ses)
352                 rpc_session_dump(ses, ctx, req);
353
354         return 0;
355 }
356
357 static int
358 rpc_handle_list(struct ubus_context *ctx, struct ubus_object *obj,
359                 struct ubus_request_data *req, const char *method,
360                 struct blob_attr *msg)
361 {
362         struct rpc_session *ses;
363         struct blob_attr *tb;
364
365         blobmsg_parse(&sid_policy, 1, &tb, blob_data(msg), blob_len(msg));
366
367         if (!tb) {
368                 avl_for_each_element(&sessions, ses, avl)
369                         rpc_session_dump(ses, ctx, req);
370                 return 0;
371         }
372
373         ses = rpc_session_get(blobmsg_data(tb));
374         if (!ses)
375                 return UBUS_STATUS_NOT_FOUND;
376
377         rpc_session_dump(ses, ctx, req);
378
379         return 0;
380 }
381
382 static int
383 uh_id_len(const char *str)
384 {
385         return strcspn(str, "*?[");
386 }
387
388 static int
389 rpc_session_grant(struct rpc_session *ses,
390                   const char *scope, const char *object, const char *function)
391 {
392         struct rpc_session_acl *acl;
393         struct rpc_session_acl_scope *acl_scope;
394         char *new_scope, *new_obj, *new_func, *new_id;
395         int id_len;
396
397         if (!object || !function)
398                 return UBUS_STATUS_INVALID_ARGUMENT;
399
400         acl_scope = avl_find_element(&ses->acls, scope, acl_scope, avl);
401
402         if (acl_scope) {
403                 uh_foreach_matching_acl_prefix(acl, &acl_scope->acls, object, function) {
404                         if (!strcmp(acl->object, object) &&
405                                 !strcmp(acl->function, function))
406                                 return 0;
407                 }
408         }
409
410         if (!acl_scope) {
411                 acl_scope = calloc_a(sizeof(*acl_scope),
412                                      &new_scope, strlen(scope) + 1);
413
414                 if (!acl_scope)
415                         return UBUS_STATUS_UNKNOWN_ERROR;
416
417                 acl_scope->avl.key = strcpy(new_scope, scope);
418                 avl_init(&acl_scope->acls, avl_strcmp, true, NULL);
419                 avl_insert(&ses->acls, &acl_scope->avl);
420         }
421
422         id_len = uh_id_len(object);
423         acl = calloc_a(sizeof(*acl),
424                 &new_obj, strlen(object) + 1,
425                 &new_func, strlen(function) + 1,
426                 &new_id, id_len + 1);
427
428         if (!acl)
429                 return UBUS_STATUS_UNKNOWN_ERROR;
430
431         acl->object = strcpy(new_obj, object);
432         acl->function = strcpy(new_func, function);
433         acl->avl.key = strncpy(new_id, object, id_len);
434         avl_insert(&acl_scope->acls, &acl->avl);
435
436         return 0;
437 }
438
439 static int
440 rpc_session_revoke(struct rpc_session *ses,
441                    const char *scope, const char *object, const char *function)
442 {
443         struct rpc_session_acl *acl, *next;
444         struct rpc_session_acl_scope *acl_scope;
445         int id_len;
446         char *id;
447
448         acl_scope = avl_find_element(&ses->acls, scope, acl_scope, avl);
449
450         if (!acl_scope)
451                 return 0;
452
453         if (!object && !function) {
454                 avl_remove_all_elements(&acl_scope->acls, acl, avl, next)
455                         free(acl);
456                 avl_delete(&ses->acls, &acl_scope->avl);
457                 free(acl_scope);
458                 return 0;
459         }
460
461         id_len = uh_id_len(object);
462         id = alloca(id_len + 1);
463         strncpy(id, object, id_len);
464         id[id_len] = 0;
465
466         acl = avl_find_element(&acl_scope->acls, id, acl, avl);
467         while (acl) {
468                 if (!avl_is_last(&acl_scope->acls, &acl->avl))
469                         next = avl_next_element(acl, avl);
470                 else
471                         next = NULL;
472
473                 if (strcmp(id, acl->avl.key) != 0)
474                         break;
475
476                 if (!strcmp(acl->object, object) &&
477                     !strcmp(acl->function, function)) {
478                         avl_delete(&acl_scope->acls, &acl->avl);
479                         free(acl);
480                 }
481                 acl = next;
482         }
483
484         if (avl_is_empty(&acl_scope->acls)) {
485                 avl_delete(&ses->acls, &acl_scope->avl);
486                 free(acl_scope);
487         }
488
489         return 0;
490 }
491
492
493 static int
494 rpc_handle_acl(struct ubus_context *ctx, struct ubus_object *obj,
495                struct ubus_request_data *req, const char *method,
496                struct blob_attr *msg)
497 {
498         struct rpc_session *ses;
499         struct blob_attr *tb[__RPC_SA_MAX];
500         struct blob_attr *attr, *sattr;
501         const char *object, *function;
502         const char *scope = "ubus";
503         int rem1, rem2;
504
505         int (*cb)(struct rpc_session *ses,
506                   const char *scope, const char *object, const char *function);
507
508         blobmsg_parse(acl_policy, __RPC_SA_MAX, tb, blob_data(msg), blob_len(msg));
509
510         if (!tb[RPC_SA_SID])
511                 return UBUS_STATUS_INVALID_ARGUMENT;
512
513         ses = rpc_session_get(blobmsg_data(tb[RPC_SA_SID]));
514         if (!ses)
515                 return UBUS_STATUS_NOT_FOUND;
516
517         if (tb[RPC_SA_SCOPE])
518                 scope = blobmsg_data(tb[RPC_SA_SCOPE]);
519
520         if (!strcmp(method, "grant"))
521                 cb = rpc_session_grant;
522         else
523                 cb = rpc_session_revoke;
524
525         if (!tb[RPC_SA_OBJECTS])
526                 return cb(ses, scope, NULL, NULL);
527
528         blobmsg_for_each_attr(attr, tb[RPC_SA_OBJECTS], rem1) {
529                 if (blobmsg_type(attr) != BLOBMSG_TYPE_ARRAY)
530                         continue;
531
532                 object = NULL;
533                 function = NULL;
534
535                 blobmsg_for_each_attr(sattr, attr, rem2) {
536                         if (blobmsg_type(sattr) != BLOBMSG_TYPE_STRING)
537                                 continue;
538
539                         if (!object)
540                                 object = blobmsg_data(sattr);
541                         else if (!function)
542                                 function = blobmsg_data(sattr);
543                         else
544                                 break;
545                 }
546
547                 if (object && function)
548                         cb(ses, scope, object, function);
549         }
550
551         return 0;
552 }
553
554 static bool
555 rpc_session_acl_allowed(struct rpc_session *ses, const char *scope,
556                         const char *obj, const char *fun)
557 {
558         struct rpc_session_acl *acl;
559         struct rpc_session_acl_scope *acl_scope;
560
561         acl_scope = avl_find_element(&ses->acls, scope, acl_scope, avl);
562
563         if (acl_scope) {
564                 uh_foreach_matching_acl(acl, &acl_scope->acls, obj, fun)
565                         return true;
566         }
567
568         return false;
569 }
570
571 static int
572 rpc_handle_access(struct ubus_context *ctx, struct ubus_object *obj,
573                   struct ubus_request_data *req, const char *method,
574                   struct blob_attr *msg)
575 {
576         struct rpc_session *ses;
577         struct blob_attr *tb[__RPC_SP_MAX];
578         const char *scope = "ubus";
579         bool allow;
580
581         blobmsg_parse(perm_policy, __RPC_SP_MAX, tb, blob_data(msg), blob_len(msg));
582
583         if (!tb[RPC_SP_SID] || !tb[RPC_SP_OBJECT] || !tb[RPC_SP_FUNCTION])
584                 return UBUS_STATUS_INVALID_ARGUMENT;
585
586         ses = rpc_session_get(blobmsg_data(tb[RPC_SP_SID]));
587         if (!ses)
588                 return UBUS_STATUS_NOT_FOUND;
589
590         if (tb[RPC_SP_SCOPE])
591                 scope = blobmsg_data(tb[RPC_SP_SCOPE]);
592
593         allow = rpc_session_acl_allowed(ses, scope,
594                                                                         blobmsg_data(tb[RPC_SP_OBJECT]),
595                                                                         blobmsg_data(tb[RPC_SP_FUNCTION]));
596
597         blob_buf_init(&buf, 0);
598         blobmsg_add_u8(&buf, "access", allow);
599         ubus_send_reply(ctx, req, buf.head);
600
601         return 0;
602 }
603
604 static void
605 rpc_session_set(struct rpc_session *ses, const char *key, struct blob_attr *val)
606 {
607         struct rpc_session_data *data;
608
609         data = avl_find_element(&ses->data, key, data, avl);
610         if (data) {
611                 avl_delete(&ses->data, &data->avl);
612                 free(data);
613         }
614
615         data = calloc(1, sizeof(*data) + blob_pad_len(val));
616         if (!data)
617                 return;
618
619         memcpy(data->attr, val, blob_pad_len(val));
620         data->avl.key = blobmsg_name(data->attr);
621         avl_insert(&ses->data, &data->avl);
622 }
623
624 static int
625 rpc_handle_set(struct ubus_context *ctx, struct ubus_object *obj,
626                struct ubus_request_data *req, const char *method,
627                struct blob_attr *msg)
628 {
629         struct rpc_session *ses;
630         struct blob_attr *tb[__RPC_SS_MAX];
631         struct blob_attr *attr;
632         int rem;
633
634         blobmsg_parse(set_policy, __RPC_SS_MAX, tb, blob_data(msg), blob_len(msg));
635
636         if (!tb[RPC_SS_SID] || !tb[RPC_SS_VALUES])
637                 return UBUS_STATUS_INVALID_ARGUMENT;
638
639         ses = rpc_session_get(blobmsg_data(tb[RPC_SS_SID]));
640         if (!ses)
641                 return UBUS_STATUS_NOT_FOUND;
642
643         blobmsg_for_each_attr(attr, tb[RPC_SS_VALUES], rem) {
644                 if (!blobmsg_name(attr)[0])
645                         continue;
646
647                 rpc_session_set(ses, blobmsg_name(attr), attr);
648         }
649
650         return 0;
651 }
652
653 static int
654 rpc_handle_get(struct ubus_context *ctx, struct ubus_object *obj,
655                struct ubus_request_data *req, const char *method,
656                struct blob_attr *msg)
657 {
658         struct rpc_session *ses;
659         struct rpc_session_data *data;
660         struct blob_attr *tb[__RPC_SG_MAX];
661         struct blob_attr *attr;
662         void *c;
663         int rem;
664
665         blobmsg_parse(get_policy, __RPC_SG_MAX, tb, blob_data(msg), blob_len(msg));
666
667         if (!tb[RPC_SG_SID])
668                 return UBUS_STATUS_INVALID_ARGUMENT;
669
670         ses = rpc_session_get(blobmsg_data(tb[RPC_SG_SID]));
671         if (!ses)
672                 return UBUS_STATUS_NOT_FOUND;
673
674         blob_buf_init(&buf, 0);
675         c = blobmsg_open_table(&buf, "values");
676
677         if (tb[RPC_SG_KEYS])
678                 blobmsg_for_each_attr(attr, tb[RPC_SG_KEYS], rem) {
679                         if (blobmsg_type(attr) != BLOBMSG_TYPE_STRING)
680                                 continue;
681
682                         data = avl_find_element(&ses->data, blobmsg_data(attr), data, avl);
683                         if (!data)
684                                 continue;
685
686                         blobmsg_add_field(&buf, blobmsg_type(data->attr),
687                                           blobmsg_name(data->attr),
688                                           blobmsg_data(data->attr),
689                                           blobmsg_data_len(data->attr));
690                 }
691         else
692                 rpc_session_dump_data(ses, &buf);
693
694         blobmsg_close_table(&buf, c);
695         ubus_send_reply(ctx, req, buf.head);
696
697         return 0;
698 }
699
700 static int
701 rpc_handle_unset(struct ubus_context *ctx, struct ubus_object *obj,
702                  struct ubus_request_data *req, const char *method,
703                  struct blob_attr *msg)
704 {
705         struct rpc_session *ses;
706         struct rpc_session_data *data, *ndata;
707         struct blob_attr *tb[__RPC_SA_MAX];
708         struct blob_attr *attr;
709         int rem;
710
711         blobmsg_parse(get_policy, __RPC_SG_MAX, tb, blob_data(msg), blob_len(msg));
712
713         if (!tb[RPC_SG_SID])
714                 return UBUS_STATUS_INVALID_ARGUMENT;
715
716         ses = rpc_session_get(blobmsg_data(tb[RPC_SG_SID]));
717         if (!ses)
718                 return UBUS_STATUS_NOT_FOUND;
719
720         if (!tb[RPC_SG_KEYS]) {
721                 avl_remove_all_elements(&ses->data, data, avl, ndata)
722                         free(data);
723                 return 0;
724         }
725
726         blobmsg_for_each_attr(attr, tb[RPC_SG_KEYS], rem) {
727                 if (blobmsg_type(attr) != BLOBMSG_TYPE_STRING)
728                         continue;
729
730                 data = avl_find_element(&ses->data, blobmsg_data(attr), data, avl);
731                 if (!data)
732                         continue;
733
734                 avl_delete(&ses->data, &data->avl);
735                 free(data);
736         }
737
738         return 0;
739 }
740
741 static int
742 rpc_handle_destroy(struct ubus_context *ctx, struct ubus_object *obj,
743                    struct ubus_request_data *req, const char *method,
744                    struct blob_attr *msg)
745 {
746         struct rpc_session *ses;
747         struct blob_attr *tb;
748
749         blobmsg_parse(&sid_policy, 1, &tb, blob_data(msg), blob_len(msg));
750
751         if (!tb)
752                 return UBUS_STATUS_INVALID_ARGUMENT;
753
754         if (!strcmp(blobmsg_get_string(tb), RPC_DEFAULT_SESSION_ID))
755                 return UBUS_STATUS_PERMISSION_DENIED;
756
757         ses = rpc_session_get(blobmsg_data(tb));
758         if (!ses)
759                 return UBUS_STATUS_NOT_FOUND;
760
761         rpc_session_destroy(ses);
762
763         return 0;
764 }
765
766
767 static bool
768 rpc_login_test_password(const char *hash, const char *password)
769 {
770         char *crypt_hash;
771
772         /* password is not set */
773         if (!hash || !*hash || !strcmp(hash, "!") || !strcmp(hash, "x"))
774         {
775                 return true;
776         }
777
778         /* password hash refers to shadow/passwd */
779         else if (!strncmp(hash, "$p$", 3))
780         {
781 #ifdef HAVE_SHADOW
782                 struct spwd *sp = getspnam(hash + 3);
783
784                 if (!sp)
785                         return false;
786
787                 return rpc_login_test_password(sp->sp_pwdp, password);
788 #else
789                 struct passwd *pw = getpwnam(hash + 3);
790
791                 if (!pw)
792                         return false;
793
794                 return rpc_login_test_password(pw->pw_passwd, password);
795 #endif
796         }
797
798         crypt_hash = crypt(password, hash);
799
800         return !strcmp(crypt_hash, hash);
801 }
802
803 static struct uci_section *
804 rpc_login_test_login(struct uci_context *uci,
805                      const char *username, const char *password)
806 {
807         struct uci_package *p = NULL;
808         struct uci_section *s;
809         struct uci_element *e;
810         struct uci_ptr ptr = { .package = "rpcd" };
811
812         uci_load(uci, ptr.package, &p);
813
814         if (!p)
815                 return false;
816
817         uci_foreach_element(&p->sections, e)
818         {
819                 s = uci_to_section(e);
820
821                 if (strcmp(s->type, "login"))
822                         continue;
823
824                 ptr.section = s->e.name;
825                 ptr.s = NULL;
826
827                 /* test for matching username */
828                 ptr.option = "username";
829                 ptr.o = NULL;
830
831                 if (uci_lookup_ptr(uci, &ptr, NULL, true))
832                         continue;
833
834                 if (ptr.o->type != UCI_TYPE_STRING)
835                         continue;
836
837                 if (strcmp(ptr.o->v.string, username))
838                         continue;
839
840                 /* If password is NULL, we're restoring ACLs for an existing session,
841                  * in this case do not check the password again. */
842                 if (!password)
843                         return ptr.s;
844
845                 /* test for matching password */
846                 ptr.option = "password";
847                 ptr.o = NULL;
848
849                 if (uci_lookup_ptr(uci, &ptr, NULL, true))
850                         continue;
851
852                 if (ptr.o->type != UCI_TYPE_STRING)
853                         continue;
854
855                 if (rpc_login_test_password(ptr.o->v.string, password))
856                         return ptr.s;
857         }
858
859         return NULL;
860 }
861
862 static bool
863 rpc_login_test_permission(struct uci_section *s,
864                           const char *perm, const char *group)
865 {
866         const char *p;
867         struct uci_option *o;
868         struct uci_element *e, *l;
869
870         /* If the login section is not provided, we're setting up acls for the
871          * default session, in this case uncondionally allow access to the
872          * "unauthenticated" access group */
873         if (!s) {
874                 return !strcmp(group, "unauthenticated");
875         }
876
877         uci_foreach_element(&s->options, e)
878         {
879                 o = uci_to_option(e);
880
881                 if (o->type != UCI_TYPE_LIST)
882                         continue;
883
884                 if (strcmp(o->e.name, perm))
885                         continue;
886
887                 /* Match negative expressions first. If a negative expression matches
888                  * the current group name then deny access. */
889                 uci_foreach_element(&o->v.list, l) {
890                         p = l->name;
891
892                         if (!p || *p != '!')
893                                 continue;
894
895                         while (isspace(*++p));
896
897                         if (!*p)
898                                 continue;
899
900                         if (!fnmatch(p, group, 0))
901                                 return false;
902                 }
903
904                 uci_foreach_element(&o->v.list, l) {
905                         if (!l->name || !*l->name || *l->name == '!')
906                                 continue;
907
908                         if (!fnmatch(l->name, group, 0))
909                                 return true;
910                 }
911         }
912
913         /* make sure that write permission implies read permission */
914         if (!strcmp(perm, "read"))
915                 return rpc_login_test_permission(s, "write", group);
916
917         return false;
918 }
919
920 static void
921 rpc_login_setup_acl_scope(struct rpc_session *ses,
922                           struct blob_attr *acl_perm,
923                           struct blob_attr *acl_scope)
924 {
925         struct blob_attr *acl_obj, *acl_func;
926         int rem, rem2;
927
928         /*
929          * Parse ACL scopes in table notation.
930          *
931          *      "<scope>": {
932          *              "<object>": [
933          *                      "<function>",
934          *                      "<function>",
935          *                      ...
936          *              ]
937          *      }
938          */
939         if (blobmsg_type(acl_scope) == BLOBMSG_TYPE_TABLE) {
940                 blobmsg_for_each_attr(acl_obj, acl_scope, rem) {
941                         if (blobmsg_type(acl_obj) != BLOBMSG_TYPE_ARRAY)
942                                 continue;
943
944                         blobmsg_for_each_attr(acl_func, acl_obj, rem2) {
945                                 if (blobmsg_type(acl_func) != BLOBMSG_TYPE_STRING)
946                                         continue;
947
948                                 rpc_session_grant(ses, blobmsg_name(acl_scope),
949                                                        blobmsg_name(acl_obj),
950                                                        blobmsg_data(acl_func));
951                         }
952                 }
953         }
954
955         /*
956          * Parse ACL scopes in array notation. The permission ("read" or "write")
957          * will be used as function name for each object.
958          *
959          *      "<scope>": [
960          *              "<object>",
961          *              "<object>",
962          *              ...
963          *      ]
964          */
965         else if (blobmsg_type(acl_scope) == BLOBMSG_TYPE_ARRAY) {
966                 blobmsg_for_each_attr(acl_obj, acl_scope, rem) {
967                         if (blobmsg_type(acl_obj) != BLOBMSG_TYPE_STRING)
968                                 continue;
969
970                         rpc_session_grant(ses, blobmsg_name(acl_scope),
971                                                blobmsg_data(acl_obj),
972                                                blobmsg_name(acl_perm));
973                 }
974         }
975 }
976
977 static void
978 rpc_login_setup_acl_file(struct rpc_session *ses, struct uci_section *login,
979                          const char *path)
980 {
981         struct blob_buf acl = { 0 };
982         struct blob_attr *acl_group, *acl_perm, *acl_scope;
983         int rem, rem2, rem3;
984
985         blob_buf_init(&acl, 0);
986
987         if (!blobmsg_add_json_from_file(&acl, path)) {
988                 fprintf(stderr, "Failed to parse %s\n", path);
989                 goto out;
990         }
991
992         /* Iterate access groups in toplevel object */
993         blob_for_each_attr(acl_group, acl.head, rem) {
994                 /* Iterate permission objects in each access group object */
995                 blobmsg_for_each_attr(acl_perm, acl_group, rem2) {
996                         if (blobmsg_type(acl_perm) != BLOBMSG_TYPE_TABLE)
997                                 continue;
998
999                         /* Only "read" and "write" permissions are defined */
1000                         if (strcmp(blobmsg_name(acl_perm), "read") &&
1001                                 strcmp(blobmsg_name(acl_perm), "write"))
1002                                 continue;
1003
1004                         /*
1005                          * Check if the current user context specifies the current
1006                          * "read" or "write" permission in the given access group.
1007                          */
1008                         if (!rpc_login_test_permission(login, blobmsg_name(acl_perm),
1009                                                               blobmsg_name(acl_group)))
1010                                 continue;
1011
1012                         /* Iterate scope objects within the permission object */
1013                         blobmsg_for_each_attr(acl_scope, acl_perm, rem3) {
1014                                 /* Setup the scopes of the access group */
1015                                 rpc_login_setup_acl_scope(ses, acl_perm, acl_scope);
1016
1017                                 /*
1018                                  * Add the access group itself as object to the "access-group"
1019                                  * meta scope and the the permission level ("read" or "write")
1020                                  * as function, so
1021                                  *      "<group>": {
1022                                  *              "<permission>": {
1023                                  *                      "<scope>": ...
1024                                  *              }
1025                                  *      }
1026                                  * becomes
1027                                  *      "access-group": {
1028                                  *              "<group>": [
1029                                  *                      "<permission>"
1030                                  *              ]
1031                                  *      }
1032                                  *
1033                                  * This allows session clients to easily query the allowed
1034                                  * access groups without having to test access of each single
1035                                  * <scope>/<object>/<function> tuple defined in a group.
1036                                  */
1037                                 rpc_session_grant(ses, "access-group",
1038                                                        blobmsg_name(acl_group),
1039                                                        blobmsg_name(acl_perm));
1040                         }
1041                 }
1042         }
1043
1044 out:
1045         blob_buf_free(&acl);
1046 }
1047
1048 static void
1049 rpc_login_setup_acls(struct rpc_session *ses, struct uci_section *login)
1050 {
1051         int i;
1052         glob_t gl;
1053
1054         if (glob(RPC_SESSION_ACL_DIR "/*.json", 0, NULL, &gl))
1055                 return;
1056
1057         for (i = 0; i < gl.gl_pathc; i++)
1058                 rpc_login_setup_acl_file(ses, login, gl.gl_pathv[i]);
1059
1060         globfree(&gl);
1061 }
1062
1063 static int
1064 rpc_handle_login(struct ubus_context *ctx, struct ubus_object *obj,
1065                  struct ubus_request_data *req, const char *method,
1066                  struct blob_attr *msg)
1067 {
1068         struct uci_context *uci = NULL;
1069         struct uci_section *login;
1070         struct rpc_session *ses;
1071         struct blob_attr *tb[__RPC_L_MAX];
1072         int timeout = RPC_DEFAULT_SESSION_TIMEOUT;
1073         int rv = 0;
1074
1075         blobmsg_parse(login_policy, __RPC_L_MAX, tb, blob_data(msg), blob_len(msg));
1076
1077         if (!tb[RPC_L_USERNAME] || !tb[RPC_L_PASSWORD]) {
1078                 rv = UBUS_STATUS_INVALID_ARGUMENT;
1079                 goto out;
1080         }
1081
1082         uci = uci_alloc_context();
1083
1084         if (!uci) {
1085                 rv = UBUS_STATUS_UNKNOWN_ERROR;
1086                 goto out;
1087         }
1088
1089         login = rpc_login_test_login(uci, blobmsg_get_string(tb[RPC_L_USERNAME]),
1090                                           blobmsg_get_string(tb[RPC_L_PASSWORD]));
1091
1092         if (!login) {
1093                 rv = UBUS_STATUS_PERMISSION_DENIED;
1094                 goto out;
1095         }
1096
1097         if (tb[RPC_L_TIMEOUT])
1098                 timeout = blobmsg_get_u32(tb[RPC_L_TIMEOUT]);
1099
1100         ses = rpc_session_create(timeout);
1101
1102         if (!ses) {
1103                 rv = UBUS_STATUS_UNKNOWN_ERROR;
1104                 goto out;
1105         }
1106
1107         rpc_login_setup_acls(ses, login);
1108
1109         rpc_session_set(ses, "user", tb[RPC_L_USERNAME]);
1110         rpc_session_dump(ses, ctx, req);
1111
1112 out:
1113         if (uci)
1114                 uci_free_context(uci);
1115
1116         return rv;
1117 }
1118
1119
1120 static bool
1121 rpc_validate_sid(const char *id)
1122 {
1123         if (!id)
1124                 return false;
1125
1126         if (strlen(id) != RPC_SID_LEN)
1127                 return false;
1128
1129         while (*id)
1130                 if (!isxdigit(*id++))
1131                         return false;
1132
1133         return true;
1134 }
1135
1136 static int
1137 rpc_blob_to_file(const char *path, struct blob_attr *attr)
1138 {
1139         int fd, len;
1140
1141         fd = open(path, O_WRONLY | O_CREAT | O_EXCL, 0600);
1142
1143         if (fd < 0)
1144                 return fd;
1145
1146         len = write(fd, attr, blob_pad_len(attr));
1147
1148         close(fd);
1149
1150         if (len != blob_pad_len(attr))
1151         {
1152                 unlink(path);
1153                 return -1;
1154         }
1155
1156         return len;
1157 }
1158
1159 static struct blob_attr *
1160 rpc_blob_from_file(const char *path)
1161 {
1162         int fd = -1, len;
1163         struct stat s;
1164         struct blob_attr head, *attr = NULL;
1165
1166         if (stat(path, &s) || !S_ISREG(s.st_mode))
1167                 return NULL;
1168
1169         fd = open(path, O_RDONLY);
1170
1171         if (fd < 0)
1172                 goto fail;
1173
1174         len = read(fd, &head, sizeof(head));
1175
1176         if (len != sizeof(head) || blob_pad_len(&head) != s.st_size)
1177                 goto fail;
1178
1179         attr = calloc(1, s.st_size);
1180
1181         if (!attr)
1182                 goto fail;
1183
1184         memcpy(attr, &head, sizeof(head));
1185
1186         len += read(fd, (char *)attr + sizeof(head), s.st_size - sizeof(head));
1187
1188         if (len != blob_pad_len(&head))
1189                 goto fail;
1190
1191         close(fd);
1192
1193         return attr;
1194
1195 fail:
1196         if (fd >= 0)
1197                 close(fd);
1198
1199         if (attr)
1200                 free(attr);
1201
1202         return NULL;
1203 }
1204
1205 static bool
1206 rpc_session_from_blob(struct uci_context *uci, struct blob_attr *attr)
1207 {
1208         int i, rem;
1209         const char *user = NULL;
1210         struct rpc_session *ses;
1211         struct uci_section *login;
1212         struct blob_attr *tb[__RPC_DUMP_MAX], *data;
1213
1214         blobmsg_parse(dump_policy, __RPC_DUMP_MAX, tb,
1215                       blob_data(attr), blob_len(attr));
1216
1217         for (i = 0; i < __RPC_DUMP_MAX; i++)
1218                 if (!tb[i])
1219                         return false;
1220
1221         ses = rpc_session_new();
1222
1223         if (!ses)
1224                 return false;
1225
1226         memcpy(ses->id, blobmsg_data(tb[RPC_DUMP_SID]), RPC_SID_LEN);
1227
1228         ses->timeout = blobmsg_get_u32(tb[RPC_DUMP_TIMEOUT]);
1229
1230         blobmsg_for_each_attr(data, tb[RPC_DUMP_DATA], rem) {
1231                 rpc_session_set(ses, blobmsg_name(data), data);
1232
1233                 if (!strcmp(blobmsg_name(data), "username"))
1234                         user = blobmsg_get_string(data);
1235         }
1236
1237         if (uci && user) {
1238                 login = rpc_login_test_login(uci, user, NULL);
1239                 if (login)
1240                         rpc_login_setup_acls(ses, login);
1241         }
1242
1243         avl_insert(&sessions, &ses->avl);
1244
1245         uloop_timeout_set(&ses->t, blobmsg_get_u32(tb[RPC_DUMP_EXPIRES]) * 1000);
1246
1247         return true;
1248 }
1249
1250 int rpc_session_api_init(struct ubus_context *ctx)
1251 {
1252         struct rpc_session *ses;
1253
1254         static const struct ubus_method session_methods[] = {
1255                 UBUS_METHOD("create",  rpc_handle_create,  &new_policy),
1256                 UBUS_METHOD("list",    rpc_handle_list,    &sid_policy),
1257                 UBUS_METHOD("grant",   rpc_handle_acl,     acl_policy),
1258                 UBUS_METHOD("revoke",  rpc_handle_acl,     acl_policy),
1259                 UBUS_METHOD("access",  rpc_handle_access,  perm_policy),
1260                 UBUS_METHOD("set",     rpc_handle_set,     set_policy),
1261                 UBUS_METHOD("get",     rpc_handle_get,     get_policy),
1262                 UBUS_METHOD("unset",   rpc_handle_unset,   get_policy),
1263                 UBUS_METHOD("destroy", rpc_handle_destroy, &sid_policy),
1264                 UBUS_METHOD("login",   rpc_handle_login,   login_policy),
1265         };
1266
1267         static struct ubus_object_type session_type =
1268                 UBUS_OBJECT_TYPE("luci-rpc-session", session_methods);
1269
1270         static struct ubus_object obj = {
1271                 .name = "session",
1272                 .type = &session_type,
1273                 .methods = session_methods,
1274                 .n_methods = ARRAY_SIZE(session_methods),
1275         };
1276
1277         avl_init(&sessions, avl_strcmp, false, NULL);
1278
1279         /* setup the default session */
1280         ses = rpc_session_new();
1281
1282         if (ses) {
1283                 strcpy(ses->id, RPC_DEFAULT_SESSION_ID);
1284                 rpc_login_setup_acls(ses, NULL);
1285                 avl_insert(&sessions, &ses->avl);
1286         }
1287
1288         return ubus_add_object(ctx, &obj);
1289 }
1290
1291 bool rpc_session_access(const char *sid, const char *scope,
1292                         const char *object, const char *function)
1293 {
1294         struct rpc_session *ses = rpc_session_get(sid);
1295
1296         if (!ses)
1297                 return false;
1298
1299         return rpc_session_acl_allowed(ses, scope, object, function);
1300 }
1301
1302 void rpc_session_create_cb(struct rpc_session_cb *cb)
1303 {
1304         if (cb && cb->cb)
1305                 list_add(&cb->list, &create_callbacks);
1306 }
1307
1308 void rpc_session_destroy_cb(struct rpc_session_cb *cb)
1309 {
1310         if (cb && cb->cb)
1311                 list_add(&cb->list, &destroy_callbacks);
1312 }
1313
1314 void rpc_session_freeze(void)
1315 {
1316         struct stat s;
1317         struct rpc_session *ses;
1318         char path[PATH_MAX];
1319
1320         if (stat(RPC_SESSION_DIRECTORY, &s))
1321                 mkdir(RPC_SESSION_DIRECTORY, 0700);
1322
1323         avl_for_each_element(&sessions, ses, avl) {
1324                 /* skip default session */
1325                 if (!strcmp(ses->id, RPC_DEFAULT_SESSION_ID))
1326                         continue;
1327
1328                 snprintf(path, sizeof(path) - 1, RPC_SESSION_DIRECTORY "/%s", ses->id);
1329                 rpc_session_to_blob(ses, false);
1330                 rpc_blob_to_file(path, buf.head);
1331         }
1332 }
1333
1334 void rpc_session_thaw(void)
1335 {
1336         DIR *d;
1337         char path[PATH_MAX];
1338         struct dirent *e;
1339         struct blob_attr *attr;
1340         struct uci_context *uci;
1341
1342         d = opendir(RPC_SESSION_DIRECTORY);
1343
1344         if (!d)
1345                 return;
1346
1347         uci = uci_alloc_context();
1348
1349         if (!uci)
1350                 return;
1351
1352         while ((e = readdir(d)) != NULL) {
1353                 if (!rpc_validate_sid(e->d_name))
1354                         continue;
1355
1356                 snprintf(path, sizeof(path) - 1,
1357                          RPC_SESSION_DIRECTORY "/%s", e->d_name);
1358
1359                 attr = rpc_blob_from_file(path);
1360
1361                 if (attr) {
1362                         rpc_session_from_blob(uci, attr);
1363                         free(attr);
1364                 }
1365
1366                 unlink(path);
1367         }
1368
1369         closedir(d);
1370
1371         uci_free_context(uci);
1372 }