123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374 |
- /*
- * rpcd - UBUS RPC server
- *
- * Copyright (C) 2013-2014 Jo-Philipp Wich <jow@openwrt.org>
- *
- * Permission to use, copy, modify, and/or distribute this software for any
- * purpose with or without fee is hereby granted, provided that the above
- * copyright notice and this permission notice appear in all copies.
- *
- * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
- * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
- * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
- * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
- * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
- * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
- * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
- */
- #include <libubus.h>
- #include <rpcd/exec.h>
- #include <rpcd/plugin.h>
- #include <rpcd/session.h>
- #include <sys/reboot.h>
- static const struct rpc_daemon_ops *ops;
- enum {
- RPC_P_USER,
- RPC_P_PASSWORD,
- __RPC_P_MAX
- };
- static const struct blobmsg_policy rpc_password_policy[__RPC_P_MAX] = {
- [RPC_P_USER] = { .name = "user", .type = BLOBMSG_TYPE_STRING },
- [RPC_P_PASSWORD] = { .name = "password", .type = BLOBMSG_TYPE_STRING },
- };
- enum {
- RPC_UPGRADE_KEEP,
- __RPC_UPGRADE_MAX
- };
- static const struct blobmsg_policy rpc_upgrade_policy[__RPC_UPGRADE_MAX] = {
- [RPC_UPGRADE_KEEP] = { .name = "keep", .type = BLOBMSG_TYPE_BOOL },
- };
- enum {
- RPC_PACKAGELIST_ALL,
- __RPC_PACKAGELIST_MAX
- };
- static const struct blobmsg_policy rpc_packagelist_policy[__RPC_PACKAGELIST_MAX] = {
- [RPC_PACKAGELIST_ALL] = { .name = "all", .type = BLOBMSG_TYPE_BOOL },
- };
- static int
- rpc_errno_status(void)
- {
- switch (errno)
- {
- case EACCES:
- return UBUS_STATUS_PERMISSION_DENIED;
- case ENOTDIR:
- return UBUS_STATUS_INVALID_ARGUMENT;
- case ENOENT:
- return UBUS_STATUS_NOT_FOUND;
- case EINVAL:
- return UBUS_STATUS_INVALID_ARGUMENT;
- default:
- return UBUS_STATUS_UNKNOWN_ERROR;
- }
- }
- static int
- rpc_cgi_password_set(struct ubus_context *ctx, struct ubus_object *obj,
- struct ubus_request_data *req, const char *method,
- struct blob_attr *msg)
- {
- pid_t pid;
- int fd, fds[2];
- struct stat s;
- struct blob_attr *tb[__RPC_P_MAX];
- ssize_t n;
- int ret;
- const char *const passwd = "/bin/passwd";
- const struct timespec ts = {0, 100 * 1000 * 1000};
- blobmsg_parse(rpc_password_policy, __RPC_P_MAX, tb,
- blob_data(msg), blob_len(msg));
- if (!tb[RPC_P_USER] || !tb[RPC_P_PASSWORD])
- return UBUS_STATUS_INVALID_ARGUMENT;
- if (stat(passwd, &s))
- return UBUS_STATUS_NOT_FOUND;
- if (!(s.st_mode & S_IXUSR))
- return UBUS_STATUS_PERMISSION_DENIED;
- if (pipe(fds))
- return rpc_errno_status();
- switch ((pid = fork()))
- {
- case -1:
- close(fds[0]);
- close(fds[1]);
- return rpc_errno_status();
- case 0:
- uloop_done();
- dup2(fds[0], 0);
- close(fds[0]);
- close(fds[1]);
- if ((fd = open("/dev/null", O_RDWR)) > -1)
- {
- dup2(fd, 1);
- dup2(fd, 2);
- close(fd);
- }
- ret = chdir("/");
- if (ret < 0)
- return rpc_errno_status();
- if (execl(passwd, passwd,
- blobmsg_data(tb[RPC_P_USER]), NULL))
- return rpc_errno_status();
- default:
- close(fds[0]);
- n = write(fds[1], blobmsg_data(tb[RPC_P_PASSWORD]),
- blobmsg_data_len(tb[RPC_P_PASSWORD]) - 1);
- if (n < 0)
- return rpc_errno_status();
- n = write(fds[1], "\n", 1);
- if (n < 0)
- return rpc_errno_status();
- nanosleep(&ts, NULL);
- n = write(fds[1], blobmsg_data(tb[RPC_P_PASSWORD]),
- blobmsg_data_len(tb[RPC_P_PASSWORD]) - 1);
- if (n < 0)
- return rpc_errno_status();
- n = write(fds[1], "\n", 1);
- if (n < 0)
- return rpc_errno_status();
- close(fds[1]);
- waitpid(pid, NULL, 0);
- return 0;
- }
- }
- static int
- rpc_sys_packagelist(struct ubus_context *ctx, struct ubus_object *obj,
- struct ubus_request_data *req, const char *method,
- struct blob_attr *msg)
- {
- struct blob_attr *tb[__RPC_PACKAGELIST_MAX];
- int all = false;
- struct blob_buf buf = { 0 };
- char var[256], pkg[128], ver[128];
- char *tmp, *p1, *p2, *p3;
- void *tbl;
- blobmsg_parse(rpc_packagelist_policy, __RPC_PACKAGELIST_MAX, tb,
- blob_data(msg), blob_len(msg));
- if (tb[RPC_PACKAGELIST_ALL] && blobmsg_get_bool(tb[RPC_PACKAGELIST_ALL]))
- all = true;
- FILE *f = fopen("/usr/lib/opkg/status", "r");
- if (!f)
- return UBUS_STATUS_NOT_FOUND;
- blob_buf_init(&buf, 0);
- tbl = blobmsg_open_table(&buf, "packages");
- pkg[0] = ver[0] = '\0';
- while(fgets(var, sizeof(var), f)) {
- p1 = strchr(var, ' ');
- p2 = p3 = NULL;
- if (!p1)
- goto procstr;
- *p1++ = '\0';
- p2 = strchr(p1, ' ');
- if (!p2) {
- tmp = strchr(p1, '\n');
- if (tmp)
- *tmp = '\0';
- goto procstr;
- }
- *p2++ = '\0';
- p3 = strchr(p2, ' ');
- if (!p3) {
- tmp = strchr(p2, '\n');
- if (tmp)
- *tmp = '\0';
- goto procstr;
- }
- *p3++ = '\0';
- tmp = strchr(p3, '\n');
- if (tmp)
- *tmp = '\0';
- procstr:
- if (!p1)
- continue;
- if (!strcmp(var, "Package:")) {
- strncpy(pkg, p1, sizeof(pkg));
- continue;
- }
- /* If there is ABIVersion, remove that suffix */
- if (!strcmp(var, "ABIVersion:")) {
- if (strlen(pkg) <= strlen(p1))
- continue;
- tmp = &pkg[strlen(pkg) - strlen(p1)];
- if (strncmp(p1, tmp, strlen(p1)))
- continue;
- *tmp = '\0';
- continue;
- }
- if (!strcmp(var, "Version:")) {
- strncpy(ver, p1, sizeof(ver));
- continue;
- }
- if (p2 && p3 &&
- !strcmp(var, "Status:") &&
- !strcmp(p1, "install") &&
- (all || strstr(p2, "user")) &&
- !strcmp(p3, "installed") && pkg[0] && ver[0]) {
- blobmsg_add_string(&buf, pkg, ver);
- pkg[0] = ver[0] = '\0';
- }
- }
- blobmsg_close_table(&buf, tbl);
- ubus_send_reply(ctx, req, buf.head);
- blob_buf_free(&buf);
- fclose(f);
- return 0;
- }
- static int
- rpc_sys_upgrade_test(struct ubus_context *ctx, struct ubus_object *obj,
- struct ubus_request_data *req, const char *method,
- struct blob_attr *msg)
- {
- const char *cmd[4] = { "sysupgrade", "--test", "/tmp/firmware.bin", NULL };
- return ops->exec(cmd, NULL, NULL, NULL, NULL, NULL, ctx, req);
- }
- static int
- rpc_sys_upgrade_start(struct ubus_context *ctx, struct ubus_object *obj,
- struct ubus_request_data *req, const char *method,
- struct blob_attr *msg)
- {
- struct blob_attr *tb[__RPC_UPGRADE_MAX];
- char * const cmd[4] = { "/sbin/sysupgrade", "-n", "/tmp/firmware.bin", NULL };
- char * const cmd_keep[3] = { "/sbin/sysupgrade", "/tmp/firmware.bin", NULL };
- char * const * c = cmd;
- blobmsg_parse(rpc_upgrade_policy, __RPC_UPGRADE_MAX, tb,
- blob_data(msg), blob_len(msg));
- if (tb[RPC_UPGRADE_KEEP] && blobmsg_get_bool(tb[RPC_UPGRADE_KEEP]))
- c = cmd_keep;
- if (!fork()) {
- /* wait for the RPC call to complete */
- sleep(2);
- return execv(c[0], c);
- }
- return 0;
- }
- static int
- rpc_sys_upgrade_clean(struct ubus_context *ctx, struct ubus_object *obj,
- struct ubus_request_data *req, const char *method,
- struct blob_attr *msg)
- {
- if (unlink("/tmp/firmware.bin"))
- return rpc_errno_status();
- return 0;
- }
- static int
- rpc_sys_factory(struct ubus_context *ctx, struct ubus_object *obj,
- struct ubus_request_data *req, const char *method,
- struct blob_attr *msg)
- {
- char * const cmd[4] = { "/sbin/jffs2reset", "-y", "-r", NULL };
- if (!fork()) {
- /* wait for the RPC call to complete */
- sleep(2);
- return execv(cmd[0], cmd);
- }
- return 0;
- }
- static int
- rpc_sys_reboot(struct ubus_context *ctx, struct ubus_object *obj,
- struct ubus_request_data *req, const char *method,
- struct blob_attr *msg)
- {
- if (!fork()) {
- sync();
- sleep(2);
- reboot(RB_AUTOBOOT);
- while (1)
- ;
- }
- return 0;
- }
- static int
- rpc_sys_api_init(const struct rpc_daemon_ops *o, struct ubus_context *ctx)
- {
- static const struct ubus_method sys_methods[] = {
- UBUS_METHOD("packagelist", rpc_sys_packagelist, rpc_packagelist_policy),
- UBUS_METHOD("password_set", rpc_cgi_password_set, rpc_password_policy),
- UBUS_METHOD_NOARG("upgrade_test", rpc_sys_upgrade_test),
- UBUS_METHOD("upgrade_start", rpc_sys_upgrade_start,
- rpc_upgrade_policy),
- UBUS_METHOD_NOARG("upgrade_clean", rpc_sys_upgrade_clean),
- UBUS_METHOD_NOARG("factory", rpc_sys_factory),
- UBUS_METHOD_NOARG("reboot", rpc_sys_reboot),
- };
- static struct ubus_object_type sys_type =
- UBUS_OBJECT_TYPE("luci-rpc-sys", sys_methods);
- static struct ubus_object obj = {
- .name = "rpc-sys",
- .type = &sys_type,
- .methods = sys_methods,
- .n_methods = ARRAY_SIZE(sys_methods),
- };
- ops = o;
- return ubus_add_object(ctx, &obj);
- }
- struct rpc_plugin rpc_plugin = {
- .init = rpc_sys_api_init
- };
|