sessions: add scopes for acls, default to "ubus" scope

This commit is contained in:
Jo-Philipp Wich 2013-08-08 18:53:59 +02:00
parent c2c39eeb30
commit c6e81f4b69
2 changed files with 109 additions and 35 deletions

139
session.c
View file

@ -57,22 +57,26 @@ static const struct blobmsg_policy get_policy[__RPC_SG_MAX] = {
enum { enum {
RPC_SA_SID, RPC_SA_SID,
RPC_SA_SCOPE,
RPC_SA_OBJECTS, RPC_SA_OBJECTS,
__RPC_SA_MAX, __RPC_SA_MAX,
}; };
static const struct blobmsg_policy acl_policy[__RPC_SA_MAX] = { static const struct blobmsg_policy acl_policy[__RPC_SA_MAX] = {
[RPC_SA_SID] = { .name = "sid", .type = BLOBMSG_TYPE_STRING }, [RPC_SA_SID] = { .name = "sid", .type = BLOBMSG_TYPE_STRING },
[RPC_SA_SCOPE] = { .name = "scope", .type = BLOBMSG_TYPE_STRING },
[RPC_SA_OBJECTS] = { .name = "objects", .type = BLOBMSG_TYPE_ARRAY }, [RPC_SA_OBJECTS] = { .name = "objects", .type = BLOBMSG_TYPE_ARRAY },
}; };
enum { enum {
RPC_SP_SID, RPC_SP_SID,
RPC_SP_SCOPE,
RPC_SP_OBJECT, RPC_SP_OBJECT,
RPC_SP_FUNCTION, RPC_SP_FUNCTION,
__RPC_SP_MAX, __RPC_SP_MAX,
}; };
static const struct blobmsg_policy perm_policy[__RPC_SP_MAX] = { static const struct blobmsg_policy perm_policy[__RPC_SP_MAX] = {
[RPC_SP_SID] = { .name = "sid", .type = BLOBMSG_TYPE_STRING }, [RPC_SP_SID] = { .name = "sid", .type = BLOBMSG_TYPE_STRING },
[RPC_SP_SCOPE] = { .name = "scope", .type = BLOBMSG_TYPE_STRING },
[RPC_SP_OBJECT] = { .name = "object", .type = BLOBMSG_TYPE_STRING }, [RPC_SP_OBJECT] = { .name = "object", .type = BLOBMSG_TYPE_STRING },
[RPC_SP_FUNCTION] = { .name = "function", .type = BLOBMSG_TYPE_STRING }, [RPC_SP_FUNCTION] = { .name = "function", .type = BLOBMSG_TYPE_STRING },
}; };
@ -83,14 +87,14 @@ static const struct blobmsg_policy perm_policy[__RPC_SP_MAX] = {
* equal to the method name, then work backwards as long as the AVL key still * equal to the method name, then work backwards as long as the AVL key still
* matches its counterpart in the object name * matches its counterpart in the object name
*/ */
#define uh_foreach_matching_acl_prefix(_acl, _ses, _obj, _func) \ #define uh_foreach_matching_acl_prefix(_acl, _avl, _obj, _func) \
for (_acl = avl_find_le_element(&(_ses)->acls, _obj, _acl, avl); \ for (_acl = avl_find_le_element(_avl, _obj, _acl, avl); \
_acl; \ _acl; \
_acl = avl_is_first(&(ses)->acls, &(_acl)->avl) ? NULL : \ _acl = avl_is_first(_avl, &(_acl)->avl) ? NULL : \
avl_prev_element((_acl), avl)) avl_prev_element((_acl), avl))
#define uh_foreach_matching_acl(_acl, _ses, _obj, _func) \ #define uh_foreach_matching_acl(_acl, _avl, _obj, _func) \
uh_foreach_matching_acl_prefix(_acl, _ses, _obj, _func) \ uh_foreach_matching_acl_prefix(_acl, _avl, _obj, _func) \
if (!strncmp((_acl)->object, _obj, (_acl)->sort_len) && \ if (!strncmp((_acl)->object, _obj, (_acl)->sort_len) && \
!fnmatch((_acl)->object, (_obj), FNM_NOESCAPE) && \ !fnmatch((_acl)->object, (_obj), FNM_NOESCAPE) && \
!fnmatch((_acl)->function, (_func), FNM_NOESCAPE)) !fnmatch((_acl)->function, (_func), FNM_NOESCAPE))
@ -128,21 +132,35 @@ static void
rpc_session_dump_acls(struct rpc_session *ses, struct blob_buf *b) rpc_session_dump_acls(struct rpc_session *ses, struct blob_buf *b)
{ {
struct rpc_session_acl *acl; struct rpc_session_acl *acl;
struct rpc_session_acl_scope *acl_scope;
const char *lastobj = NULL; const char *lastobj = NULL;
void *c = NULL; const char *lastscope = NULL;
void *c = NULL, *d = NULL;
avl_for_each_element(&ses->acls, acl, avl) { avl_for_each_element(&ses->acls, acl_scope, avl) {
if (!lastobj || strcmp(acl->object, lastobj)) if (!lastscope || strcmp(acl_scope->avl.key, lastscope))
{ {
if (c) blobmsg_close_array(b, c); if (c) blobmsg_close_table(b, c);
c = blobmsg_open_array(b, acl->object); c = blobmsg_open_table(b, acl_scope->avl.key);
} }
blobmsg_add_string(b, NULL, acl->function); d = NULL;
lastobj = acl->object;
avl_for_each_element(&acl_scope->acls, acl, avl) {
if (!lastobj || strcmp(acl->object, lastobj))
{
if (d) blobmsg_close_array(b, d);
d = blobmsg_open_array(b, acl->object);
}
blobmsg_add_string(b, NULL, acl->function);
lastobj = acl->object;
}
if (d) blobmsg_close_array(b, d);
} }
if (c) blobmsg_close_array(b, c); if (c) blobmsg_close_table(b, c);
} }
static void static void
@ -179,11 +197,18 @@ static void
rpc_session_destroy(struct rpc_session *ses) rpc_session_destroy(struct rpc_session *ses)
{ {
struct rpc_session_acl *acl, *nacl; struct rpc_session_acl *acl, *nacl;
struct rpc_session_acl_scope *acl_scope, *nacl_scope;
struct rpc_session_data *data, *ndata; struct rpc_session_data *data, *ndata;
uloop_timeout_cancel(&ses->t); uloop_timeout_cancel(&ses->t);
avl_remove_all_elements(&ses->acls, acl, avl, nacl)
free(acl); avl_for_each_element_safe(&ses->acls, acl_scope, avl, nacl_scope) {
avl_remove_all_elements(&acl_scope->acls, acl, avl, nacl)
free(acl);
avl_delete(&ses->acls, &acl_scope->avl);
free(acl_scope);
}
avl_remove_all_elements(&ses->data, data, avl, ndata) avl_remove_all_elements(&ses->data, data, avl, ndata)
free(data); free(data);
@ -289,19 +314,36 @@ uh_id_len(const char *str)
static int static int
rpc_session_grant(struct rpc_session *ses, struct ubus_context *ctx, rpc_session_grant(struct rpc_session *ses, struct ubus_context *ctx,
const char *object, const char *function) const char *scope, const char *object, const char *function)
{ {
struct rpc_session_acl *acl; struct rpc_session_acl *acl;
char *new_obj, *new_func, *new_id; struct rpc_session_acl_scope *acl_scope;
char *new_scope, *new_obj, *new_func, *new_id;
int id_len; int id_len;
if (!object || !function) if (!object || !function)
return UBUS_STATUS_INVALID_ARGUMENT; return UBUS_STATUS_INVALID_ARGUMENT;
uh_foreach_matching_acl_prefix(acl, ses, object, function) { acl_scope = avl_find_element(&ses->acls, scope, acl_scope, avl);
if (!strcmp(acl->object, object) &&
!strcmp(acl->function, function)) if (acl_scope) {
return 0; uh_foreach_matching_acl_prefix(acl, &acl_scope->acls, object, function) {
if (!strcmp(acl->object, object) &&
!strcmp(acl->function, function))
return 0;
}
}
if (!acl_scope) {
acl_scope = calloc_a(sizeof(*acl_scope),
&new_scope, strlen(scope) + 1);
if (!acl_scope)
return UBUS_STATUS_UNKNOWN_ERROR;
acl_scope->avl.key = strcpy(new_scope, scope);
avl_init(&acl_scope->acls, avl_strcmp, true, NULL);
avl_insert(&ses->acls, &acl_scope->avl);
} }
id_len = uh_id_len(object); id_len = uh_id_len(object);
@ -316,22 +358,30 @@ rpc_session_grant(struct rpc_session *ses, struct ubus_context *ctx,
acl->object = strcpy(new_obj, object); acl->object = strcpy(new_obj, object);
acl->function = strcpy(new_func, function); acl->function = strcpy(new_func, function);
acl->avl.key = strncpy(new_id, object, id_len); acl->avl.key = strncpy(new_id, object, id_len);
avl_insert(&ses->acls, &acl->avl); avl_insert(&acl_scope->acls, &acl->avl);
return 0; return 0;
} }
static int static int
rpc_session_revoke(struct rpc_session *ses, struct ubus_context *ctx, rpc_session_revoke(struct rpc_session *ses, struct ubus_context *ctx,
const char *object, const char *function) const char *scope, const char *object, const char *function)
{ {
struct rpc_session_acl *acl, *next; struct rpc_session_acl *acl, *next;
struct rpc_session_acl_scope *acl_scope;
int id_len; int id_len;
char *id; char *id;
acl_scope = avl_find_element(&ses->acls, scope, acl_scope, avl);
if (!acl_scope)
return 0;
if (!object && !function) { if (!object && !function) {
avl_remove_all_elements(&ses->acls, acl, avl, next) avl_remove_all_elements(&acl_scope->acls, acl, avl, next)
free(acl); free(acl);
avl_delete(&ses->acls, &acl_scope->avl);
free(acl_scope);
return 0; return 0;
} }
@ -340,9 +390,9 @@ rpc_session_revoke(struct rpc_session *ses, struct ubus_context *ctx,
strncpy(id, object, id_len); strncpy(id, object, id_len);
id[id_len] = 0; id[id_len] = 0;
acl = avl_find_element(&ses->acls, id, acl, avl); acl = avl_find_element(&acl_scope->acls, id, acl, avl);
while (acl) { while (acl) {
if (!avl_is_last(&ses->acls, &acl->avl)) if (!avl_is_last(&acl_scope->acls, &acl->avl))
next = avl_next_element(acl, avl); next = avl_next_element(acl, avl);
else else
next = NULL; next = NULL;
@ -352,12 +402,17 @@ rpc_session_revoke(struct rpc_session *ses, struct ubus_context *ctx,
if (!strcmp(acl->object, object) && if (!strcmp(acl->object, object) &&
!strcmp(acl->function, function)) { !strcmp(acl->function, function)) {
avl_delete(&ses->acls, &acl->avl); avl_delete(&acl_scope->acls, &acl->avl);
free(acl); free(acl);
} }
acl = next; acl = next;
} }
if (avl_is_empty(&acl_scope->acls)) {
avl_delete(&ses->acls, &acl_scope->avl);
free(acl_scope);
}
return 0; return 0;
} }
@ -371,10 +426,11 @@ rpc_handle_acl(struct ubus_context *ctx, struct ubus_object *obj,
struct blob_attr *tb[__RPC_SA_MAX]; struct blob_attr *tb[__RPC_SA_MAX];
struct blob_attr *attr, *sattr; struct blob_attr *attr, *sattr;
const char *object, *function; const char *object, *function;
const char *scope = "ubus";
int rem1, rem2; int rem1, rem2;
int (*cb)(struct rpc_session *ses, struct ubus_context *ctx, int (*cb)(struct rpc_session *ses, struct ubus_context *ctx,
const char *object, const char *function); const char *scope, const char *object, const char *function);
blobmsg_parse(acl_policy, __RPC_SA_MAX, tb, blob_data(msg), blob_len(msg)); blobmsg_parse(acl_policy, __RPC_SA_MAX, tb, blob_data(msg), blob_len(msg));
@ -385,13 +441,16 @@ rpc_handle_acl(struct ubus_context *ctx, struct ubus_object *obj,
if (!ses) if (!ses)
return UBUS_STATUS_NOT_FOUND; return UBUS_STATUS_NOT_FOUND;
if (tb[RPC_SA_SCOPE])
scope = blobmsg_data(tb[RPC_SA_SCOPE]);
if (!strcmp(method, "grant")) if (!strcmp(method, "grant"))
cb = rpc_session_grant; cb = rpc_session_grant;
else else
cb = rpc_session_revoke; cb = rpc_session_revoke;
if (!tb[RPC_SA_OBJECTS]) if (!tb[RPC_SA_OBJECTS])
return cb(ses, ctx, NULL, NULL); return cb(ses, ctx, scope, NULL, NULL);
blobmsg_for_each_attr(attr, tb[RPC_SA_OBJECTS], rem1) { blobmsg_for_each_attr(attr, tb[RPC_SA_OBJECTS], rem1) {
if (blob_id(attr) != BLOBMSG_TYPE_ARRAY) if (blob_id(attr) != BLOBMSG_TYPE_ARRAY)
@ -413,19 +472,25 @@ rpc_handle_acl(struct ubus_context *ctx, struct ubus_object *obj,
} }
if (object && function) if (object && function)
cb(ses, ctx, object, function); cb(ses, ctx, scope, object, function);
} }
return 0; return 0;
} }
static bool static bool
rpc_session_acl_allowed(struct rpc_session *ses, const char *obj, const char *fun) rpc_session_acl_allowed(struct rpc_session *ses, const char *scope,
const char *obj, const char *fun)
{ {
struct rpc_session_acl *acl; struct rpc_session_acl *acl;
struct rpc_session_acl_scope *acl_scope;
uh_foreach_matching_acl(acl, ses, obj, fun) acl_scope = avl_find_element(&ses->acls, scope, acl_scope, avl);
return true;
if (acl_scope) {
uh_foreach_matching_acl(acl, &acl_scope->acls, obj, fun)
return true;
}
return false; return false;
} }
@ -437,6 +502,7 @@ rpc_handle_access(struct ubus_context *ctx, struct ubus_object *obj,
{ {
struct rpc_session *ses; struct rpc_session *ses;
struct blob_attr *tb[__RPC_SP_MAX]; struct blob_attr *tb[__RPC_SP_MAX];
const char *scope = "ubus";
bool allow; bool allow;
blobmsg_parse(perm_policy, __RPC_SP_MAX, tb, blob_data(msg), blob_len(msg)); blobmsg_parse(perm_policy, __RPC_SP_MAX, tb, blob_data(msg), blob_len(msg));
@ -448,7 +514,10 @@ rpc_handle_access(struct ubus_context *ctx, struct ubus_object *obj,
if (!ses) if (!ses)
return UBUS_STATUS_NOT_FOUND; return UBUS_STATUS_NOT_FOUND;
allow = rpc_session_acl_allowed(ses, if (tb[RPC_SP_SCOPE])
scope = blobmsg_data(tb[RPC_SP_SCOPE]);
allow = rpc_session_acl_allowed(ses, scope,
blobmsg_data(tb[RPC_SP_OBJECT]), blobmsg_data(tb[RPC_SP_OBJECT]),
blobmsg_data(tb[RPC_SP_FUNCTION])); blobmsg_data(tb[RPC_SP_FUNCTION]));

View file

@ -42,6 +42,11 @@ struct rpc_session_data {
struct blob_attr attr[]; struct blob_attr attr[];
}; };
struct rpc_session_acl_scope {
struct avl_node avl;
struct avl_tree acls;
};
struct rpc_session_acl { struct rpc_session_acl {
struct avl_node avl; struct avl_node avl;
const char *object; const char *object;