|
|
@ -0,0 +1,427 @@ |
|
|
|
/* |
|
|
|
* rpcd-lxc-plugin |
|
|
|
* |
|
|
|
* Copyright (C) 2014 Cisco Systems, Inc. |
|
|
|
* Author: Luka Perkov <luka@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 <lxc/lxccontainer.h> |
|
|
|
|
|
|
|
#include <rpcd/plugin.h> |
|
|
|
|
|
|
|
static struct blob_buf buf; |
|
|
|
|
|
|
|
struct rpc_lxc { |
|
|
|
/* ubus options */ |
|
|
|
char *name; |
|
|
|
char *config; |
|
|
|
/* lxc container */ |
|
|
|
struct lxc_container *container; |
|
|
|
}; |
|
|
|
|
|
|
|
enum { |
|
|
|
RPC_LXC_NAME, |
|
|
|
RPC_LXC_CONFIG, |
|
|
|
__RPC_LXC_MAX, |
|
|
|
}; |
|
|
|
|
|
|
|
enum { |
|
|
|
RPC_LXC_SHUTDOWN_NAME, |
|
|
|
RPC_LXC_SHUTDOWN_CONFIG, |
|
|
|
RPC_LXC_SHUTDOWN_TIMEOUT, |
|
|
|
__RPC_LXC_SHUTDOWN_MAX, |
|
|
|
}; |
|
|
|
|
|
|
|
enum { |
|
|
|
RPC_LXC_RENAME_NAME, |
|
|
|
RPC_LXC_RENAME_CONFIG, |
|
|
|
RPC_LXC_RENAME_NEWNAME, |
|
|
|
__RPC_LXC_RENAME_MAX, |
|
|
|
}; |
|
|
|
|
|
|
|
static const struct blobmsg_policy rpc_lxc_min_policy[__RPC_LXC_MAX] = { |
|
|
|
[RPC_LXC_NAME] = { .name = "name", .type = BLOBMSG_TYPE_STRING }, |
|
|
|
[RPC_LXC_CONFIG] = { .name = "config", .type = BLOBMSG_TYPE_STRING }, |
|
|
|
}; |
|
|
|
|
|
|
|
static const struct blobmsg_policy rpc_lxc_shutdown_policy[__RPC_LXC_SHUTDOWN_MAX] = { |
|
|
|
[RPC_LXC_SHUTDOWN_NAME] = { .name = "name", .type = BLOBMSG_TYPE_STRING }, |
|
|
|
[RPC_LXC_SHUTDOWN_CONFIG] = { .name = "config", .type = BLOBMSG_TYPE_STRING }, |
|
|
|
[RPC_LXC_SHUTDOWN_TIMEOUT] = { .name = "timeout", .type = BLOBMSG_TYPE_INT32 }, |
|
|
|
}; |
|
|
|
|
|
|
|
static const struct blobmsg_policy rpc_lxc_rename_policy[__RPC_LXC_RENAME_MAX] = { |
|
|
|
[RPC_LXC_RENAME_NAME] = { .name = "name", .type = BLOBMSG_TYPE_STRING }, |
|
|
|
[RPC_LXC_RENAME_CONFIG] = { .name = "config", .type = BLOBMSG_TYPE_STRING }, |
|
|
|
[RPC_LXC_RENAME_NEWNAME] = { .name = "newname", .type = BLOBMSG_TYPE_STRING }, |
|
|
|
}; |
|
|
|
|
|
|
|
static struct rpc_lxc * |
|
|
|
rpc_lxc_init(struct blob_attr *tb[__RPC_LXC_MAX]) |
|
|
|
{ |
|
|
|
struct rpc_lxc *l = NULL; |
|
|
|
|
|
|
|
l = calloc(1, sizeof(struct rpc_lxc)); |
|
|
|
if (!l) return NULL; |
|
|
|
|
|
|
|
if (tb[RPC_LXC_NAME]) { |
|
|
|
l->name = blobmsg_data(tb[RPC_LXC_NAME]); |
|
|
|
} else { |
|
|
|
goto error; |
|
|
|
} |
|
|
|
|
|
|
|
if (tb[RPC_LXC_CONFIG]) { |
|
|
|
l->config = blobmsg_data(tb[RPC_LXC_CONFIG]); |
|
|
|
} else { |
|
|
|
l->config = NULL; |
|
|
|
} |
|
|
|
|
|
|
|
l->container = lxc_container_new(l->name, l->config); |
|
|
|
if (!l->container) { |
|
|
|
goto error; |
|
|
|
} |
|
|
|
|
|
|
|
return l; |
|
|
|
error: |
|
|
|
free(l); |
|
|
|
return NULL; |
|
|
|
} |
|
|
|
|
|
|
|
static void |
|
|
|
rpc_lxc_done(struct rpc_lxc *l) |
|
|
|
{ |
|
|
|
if (l) { |
|
|
|
lxc_container_put(l->container); |
|
|
|
free(l); |
|
|
|
} |
|
|
|
|
|
|
|
return; |
|
|
|
} |
|
|
|
|
|
|
|
static int |
|
|
|
rpc_lxc_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_LXC_MAX]; |
|
|
|
struct rpc_lxc *l = NULL; |
|
|
|
int rc; |
|
|
|
|
|
|
|
blobmsg_parse(rpc_lxc_min_policy, __RPC_LXC_MAX, tb, blob_data(msg), blob_len(msg)); |
|
|
|
|
|
|
|
l = rpc_lxc_init(tb); |
|
|
|
if (!l) return UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
|
|
|
|
if (l->container->is_running(l->container)) { |
|
|
|
rc = UBUS_STATUS_UNKNOWN_ERROR; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
if (!l->container->start(l->container, 0, NULL)) { |
|
|
|
rc = UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
rc = UBUS_STATUS_OK; |
|
|
|
out: |
|
|
|
rpc_lxc_done(l); |
|
|
|
return rc; |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
static int |
|
|
|
rpc_lxc_reboot(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_LXC_MAX]; |
|
|
|
struct rpc_lxc *l = NULL; |
|
|
|
int rc; |
|
|
|
|
|
|
|
blobmsg_parse(rpc_lxc_min_policy, __RPC_LXC_MAX, tb, blob_data(msg), blob_len(msg)); |
|
|
|
|
|
|
|
l = rpc_lxc_init(tb); |
|
|
|
if (!l) return UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
|
|
|
|
if (!l->container->is_running(l->container)) { |
|
|
|
rc = UBUS_STATUS_UNKNOWN_ERROR; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
if (!l->container->reboot(l->container)) { |
|
|
|
rc = UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
rc = UBUS_STATUS_OK; |
|
|
|
out: |
|
|
|
rpc_lxc_done(l); |
|
|
|
return rc; |
|
|
|
} |
|
|
|
|
|
|
|
static int |
|
|
|
rpc_lxc_shutdown(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_LXC_SHUTDOWN_MAX]; |
|
|
|
struct rpc_lxc *l = NULL; |
|
|
|
int rc; |
|
|
|
|
|
|
|
blobmsg_parse(rpc_lxc_shutdown_policy, __RPC_LXC_SHUTDOWN_MAX, tb, blob_data(msg), blob_len(msg)); |
|
|
|
|
|
|
|
l = rpc_lxc_init(tb); |
|
|
|
if (!l) return UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
|
|
|
|
if (!l->container->is_running(l->container)) { |
|
|
|
rc = UBUS_STATUS_UNKNOWN_ERROR; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
/* define default timeout */ |
|
|
|
int timeout = 30; |
|
|
|
if (tb[RPC_LXC_SHUTDOWN_TIMEOUT]) { |
|
|
|
timeout = blobmsg_get_u32(tb[RPC_LXC_SHUTDOWN_TIMEOUT]); |
|
|
|
} |
|
|
|
|
|
|
|
if (!l->container->shutdown(l->container, timeout)) { |
|
|
|
rc = UBUS_STATUS_UNKNOWN_ERROR; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
rc = UBUS_STATUS_OK; |
|
|
|
out: |
|
|
|
rpc_lxc_done(l); |
|
|
|
return rc; |
|
|
|
} |
|
|
|
|
|
|
|
static int |
|
|
|
rpc_lxc_stop(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_LXC_MAX]; |
|
|
|
struct rpc_lxc *l = NULL; |
|
|
|
int rc; |
|
|
|
|
|
|
|
blobmsg_parse(rpc_lxc_min_policy, __RPC_LXC_MAX, tb, blob_data(msg), blob_len(msg)); |
|
|
|
|
|
|
|
l = rpc_lxc_init(tb); |
|
|
|
if (!l) return UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
|
|
|
|
if (!l->container->is_running(l->container)) { |
|
|
|
rc = UBUS_STATUS_UNKNOWN_ERROR; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
if (!l->container->stop(l->container)) { |
|
|
|
rc = UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
rc = UBUS_STATUS_OK; |
|
|
|
out: |
|
|
|
rpc_lxc_done(l); |
|
|
|
return rc; |
|
|
|
} |
|
|
|
|
|
|
|
static int |
|
|
|
rpc_lxc_freeze(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_LXC_MAX]; |
|
|
|
struct rpc_lxc *l = NULL; |
|
|
|
int rc; |
|
|
|
|
|
|
|
blobmsg_parse(rpc_lxc_min_policy, __RPC_LXC_MAX, tb, blob_data(msg), blob_len(msg)); |
|
|
|
|
|
|
|
l = rpc_lxc_init(tb); |
|
|
|
if (!l) return UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
|
|
|
|
if (!l->container->is_running(l->container)) { |
|
|
|
rc = UBUS_STATUS_UNKNOWN_ERROR; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
if (!l->container->freeze(l->container)) { |
|
|
|
rc = UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
rc = UBUS_STATUS_OK; |
|
|
|
out: |
|
|
|
rpc_lxc_done(l); |
|
|
|
return rc; |
|
|
|
} |
|
|
|
|
|
|
|
static int |
|
|
|
rpc_lxc_unfreeze(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_LXC_MAX]; |
|
|
|
struct rpc_lxc *l = NULL; |
|
|
|
int rc; |
|
|
|
|
|
|
|
blobmsg_parse(rpc_lxc_min_policy, __RPC_LXC_MAX, tb, blob_data(msg), blob_len(msg)); |
|
|
|
|
|
|
|
l = rpc_lxc_init(tb); |
|
|
|
if (!l) return UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
|
|
|
|
if (!l->container->is_running(l->container)) { |
|
|
|
rc = UBUS_STATUS_UNKNOWN_ERROR; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
if (!l->container->unfreeze(l->container)) { |
|
|
|
rc = UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
rc = UBUS_STATUS_OK; |
|
|
|
out: |
|
|
|
rpc_lxc_done(l); |
|
|
|
return rc; |
|
|
|
} |
|
|
|
|
|
|
|
static int |
|
|
|
rpc_lxc_rename(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_LXC_RENAME_MAX]; |
|
|
|
struct rpc_lxc *l = NULL; |
|
|
|
int rc; |
|
|
|
|
|
|
|
blobmsg_parse(rpc_lxc_rename_policy, __RPC_LXC_RENAME_MAX, tb, blob_data(msg), blob_len(msg)); |
|
|
|
|
|
|
|
l = rpc_lxc_init(tb); |
|
|
|
if (!l) return UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
|
|
|
|
if (!tb[RPC_LXC_RENAME_NEWNAME]) { |
|
|
|
rc = UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
if (l->container->is_running(l->container)) { |
|
|
|
rc = UBUS_STATUS_UNKNOWN_ERROR; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
char *newname = blobmsg_data(tb[RPC_LXC_RENAME_NEWNAME]); |
|
|
|
if (!newname || !l->container->rename(l->container, newname)) { |
|
|
|
rc = UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
rc = UBUS_STATUS_OK; |
|
|
|
out: |
|
|
|
rpc_lxc_done(l); |
|
|
|
return rc; |
|
|
|
} |
|
|
|
|
|
|
|
static int |
|
|
|
rpc_lxc_destroy(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_LXC_MAX]; |
|
|
|
struct rpc_lxc *l = NULL; |
|
|
|
int rc; |
|
|
|
|
|
|
|
blobmsg_parse(rpc_lxc_min_policy, __RPC_LXC_MAX, tb, blob_data(msg), blob_len(msg)); |
|
|
|
|
|
|
|
l = rpc_lxc_init(tb); |
|
|
|
if (!l) return UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
|
|
|
|
if (l->container->is_running(l->container)) { |
|
|
|
rc = UBUS_STATUS_UNKNOWN_ERROR; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
if (!l->container->destroy(l->container)) { |
|
|
|
rc = UBUS_STATUS_INVALID_ARGUMENT; |
|
|
|
goto out; |
|
|
|
} |
|
|
|
|
|
|
|
rc = UBUS_STATUS_OK; |
|
|
|
out: |
|
|
|
rpc_lxc_done(l); |
|
|
|
return rc; |
|
|
|
} |
|
|
|
|
|
|
|
static int |
|
|
|
rpc_lxc_list(struct ubus_context *ctx, struct ubus_object *obj, |
|
|
|
struct ubus_request_data *req, const char *method, |
|
|
|
struct blob_attr *msg) |
|
|
|
{ |
|
|
|
|
|
|
|
blob_buf_init(&buf, 0); |
|
|
|
|
|
|
|
int rc; |
|
|
|
char **names; |
|
|
|
struct lxc_container **cret; |
|
|
|
|
|
|
|
rc = list_all_containers(NULL, &names, &cret); |
|
|
|
if (rc == -1) |
|
|
|
return UBUS_STATUS_UNKNOWN_ERROR; |
|
|
|
|
|
|
|
for (int i = 0; i < rc; i++) { |
|
|
|
struct lxc_container *c = cret[i]; |
|
|
|
blobmsg_add_string(&buf, names[i], c->state(c)); |
|
|
|
|
|
|
|
free(names[i]); |
|
|
|
lxc_container_put(c); |
|
|
|
} |
|
|
|
|
|
|
|
ubus_send_reply(ctx, req, buf.head); |
|
|
|
|
|
|
|
return UBUS_STATUS_OK; |
|
|
|
} |
|
|
|
|
|
|
|
static int |
|
|
|
rpc_lxc_api_init(const struct rpc_daemon_ops *o, struct ubus_context *ctx) |
|
|
|
{ |
|
|
|
static const struct ubus_method lxc_methods[] = { |
|
|
|
UBUS_METHOD("start", rpc_lxc_start, rpc_lxc_min_policy), |
|
|
|
UBUS_METHOD("reboot", rpc_lxc_reboot, rpc_lxc_min_policy), |
|
|
|
UBUS_METHOD("shutdown", rpc_lxc_shutdown, rpc_lxc_shutdown_policy), |
|
|
|
UBUS_METHOD("stop", rpc_lxc_stop, rpc_lxc_min_policy), |
|
|
|
UBUS_METHOD("freeze", rpc_lxc_freeze, rpc_lxc_min_policy), |
|
|
|
UBUS_METHOD("unfreeze", rpc_lxc_unfreeze, rpc_lxc_min_policy), |
|
|
|
UBUS_METHOD("rename", rpc_lxc_rename, rpc_lxc_rename_policy), |
|
|
|
UBUS_METHOD("destroy", rpc_lxc_destroy, rpc_lxc_min_policy), |
|
|
|
UBUS_METHOD_NOARG("list", rpc_lxc_list), |
|
|
|
}; |
|
|
|
|
|
|
|
static struct ubus_object_type lxc_type = |
|
|
|
UBUS_OBJECT_TYPE("luci-rpc-lxc", lxc_methods); |
|
|
|
|
|
|
|
static struct ubus_object obj = { |
|
|
|
.name = "lxc", |
|
|
|
.type = &lxc_type, |
|
|
|
.methods = lxc_methods, |
|
|
|
.n_methods = ARRAY_SIZE(lxc_methods), |
|
|
|
}; |
|
|
|
|
|
|
|
return ubus_add_object(ctx, &obj); |
|
|
|
} |
|
|
|
|
|
|
|
struct rpc_plugin rpc_plugin = { |
|
|
|
.init = rpc_lxc_api_init |
|
|
|
}; |