DGNum/rpcd
6
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

luci2: implement sshkeys_get and sshkeys_set calls

Browse source
This commit is contained in:
Jo-Philipp Wich 2013-07-01 17:28:30 +02:00
parent f1aa60efc5
commit e8daa19a06
1 changed files with 79 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

80
luci2.c
View file

@ -21,6 +21,7 @@
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <dirent.h>
@ -54,6 +55,15 @@ static const struct blobmsg_policy rpc_init_policy[__RPC_I_MAX] = {
[RPC_I_ACTION] = { .name = "action", .type = BLOBMSG_TYPE_STRING },
};
enum {
RPC_K_KEYS,
__RPC_K_MAX
};
static const struct blobmsg_policy rpc_sshkey_policy[__RPC_K_MAX] = {
[RPC_K_KEYS] = { .name = "keys", .type = BLOBMSG_TYPE_ARRAY },
};
static int
rpc_errno_status(void)
@ -440,6 +450,71 @@ rpc_luci2_init_action(struct ubus_context *ctx, struct ubus_object *obj,
}
}
static int
rpc_luci2_sshkeys_get(struct ubus_context *ctx, struct ubus_object *obj,
struct ubus_request_data *req, const char *method,
struct blob_attr *msg)
{
FILE *f;
void *c;
char *p, line[4096];
if (!(f = fopen("/etc/dropbear/authorized_keys", "r")))
return rpc_errno_status();
blob_buf_init(&buf, 0);
c = blobmsg_open_array(&buf, "keys");
while (fgets(line, sizeof(line) - 1, f))
{
for (p = line + strlen(line) - 1; (p > line) && isspace(*p); p--)
*p = 0;
for (p = line; isspace(*p); p++)
*p = 0;
if (*p)
blobmsg_add_string(&buf, NULL, p);
}
blobmsg_close_array(&buf, c);
fclose(f);
ubus_send_reply(ctx, req, buf.head);
return 0;
}
static int
rpc_luci2_sshkeys_set(struct ubus_context *ctx, struct ubus_object *obj,
struct ubus_request_data *req, const char *method,
struct blob_attr *msg)
{
FILE *f;
int rem;
struct blob_attr *cur, *tb[__RPC_K_MAX];
blobmsg_parse(rpc_sshkey_policy, __RPC_K_MAX, tb,
blob_data(msg), blob_len(msg));
if (!tb[RPC_K_KEYS])
return UBUS_STATUS_INVALID_ARGUMENT;
if (!(f = fopen("/etc/dropbear/authorized_keys", "w")))
return rpc_errno_status();
blobmsg_for_each_attr(cur, tb[RPC_K_KEYS], rem)
{
if (blobmsg_type(cur) != BLOBMSG_TYPE_STRING)
continue;
fwrite(blobmsg_data(cur), blobmsg_data_len(cur) - 1, 1, f);
fwrite("\n", 1, 1, f);
}
fclose(f);
return 0;
}
static FILE *
dnsmasq_leasefile(void)
@ -971,7 +1046,10 @@ int rpc_luci2_api_init(struct ubus_context *ctx)
rpc_signal_policy),
UBUS_METHOD_NOARG("init_list", rpc_luci2_init_list),
UBUS_METHOD("init_action", rpc_luci2_init_action,
rpc_init_policy)
rpc_init_policy),
UBUS_METHOD_NOARG("sshkeys_get", rpc_luci2_sshkeys_get),
UBUS_METHOD("sshkeys_set", rpc_luci2_sshkeys_set,
rpc_sshkey_policy)
};
static struct ubus_object_type luci2_system_type =