[mirror_lxc.git] / src / lxc / lxc_create.c

/*
 *
 * Copyright © 2013 Serge Hallyn <serge.hallyn@ubuntu.com>.
 * Copyright © 2013 Canonical Ltd.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2, as
 * published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include <lxc/lxccontainer.h>

#include <stdio.h>
#include <libgen.h>
#include <unistd.h>
#include <ctype.h>
#include <fcntl.h>
#include <sys/types.h>
#include <stdint.h>

#include "lxc.h"
#include "log.h"
#include "bdev.h"
#include "arguments.h"
#include "utils.h"

lxc_log_define(lxc_create_ui, lxc);

static uint64_t get_fssize(char *s)
{
	uint64_t ret;
	char *end;

	ret = strtoull(s, &end, 0);
	if (end == s)
	{
		fprintf(stderr, "Invalid blockdev size '%s', using default size\n", s);
		return 0;
	}
	while (isblank(*end))
		end++;
	if (*end == '\0')
		ret *= 1024ULL * 1024ULL; // MB by default
	else if (*end == 'b' || *end == 'B')
		ret *= 1ULL;
	else if (*end == 'k' || *end == 'K')
		ret *= 1024ULL;
	else if (*end == 'm' || *end == 'M')
		ret *= 1024ULL * 1024ULL;
	else if (*end == 'g' || *end == 'G')
		ret *= 1024ULL * 1024ULL * 1024ULL;
	else if (*end == 't' || *end == 'T')
		ret *= 1024ULL * 1024ULL * 1024ULL * 1024ULL;
	else
	{
		fprintf(stderr, "Invalid blockdev unit size '%c' in '%s', using default size\n", *end, s);
		return 0;
	}
	return ret;
}

static int my_parser(struct lxc_arguments* args, int c, char* arg)
{
	switch (c) {
	case 'B': args->bdevtype = arg; break;
	case 'f': args->configfile = arg; break;
	case 't': args->template = arg; break;
	case '0': args->lvname = arg; break;
	case '1': args->vgname = arg; break;
	case '2': args->thinpool = arg; break;
	case '3': args->fstype = arg; break;
	case '4': args->fssize = get_fssize(arg); break;
	case '5': args->zfsroot = arg; break;
	case '6': args->dir = arg; break;
	}
	return 0;
}

static const struct option my_longopts[] = {
	{"bdev", required_argument, 0, 'B'},
	{"config", required_argument, 0, 'f'},
	{"template", required_argument, 0, 't'},
	{"lvname", required_argument, 0, '0'},
	{"vgname", required_argument, 0, '1'},
	{"thinpool", required_argument, 0, '2'},
	{"fstype", required_argument, 0, '3'},
	{"fssize", required_argument, 0, '4'},
	{"zfsroot", required_argument, 0, '5'},
	{"dir", required_argument, 0, '6'},
	LXC_COMMON_OPTIONS
};

static void create_helpfn(const struct lxc_arguments *args) {
	char *argv[3], *path;
	pid_t pid;

	if (!args->template)
		return;

	pid = fork();
	if (pid) {
		wait_for_pid(pid);
		return;
	}

	path = get_template_path(args->template);

	argv[0] = path;
	argv[1] = "-h";
	argv[2] = NULL;
	execv(path, argv);
	ERROR("Error executing %s -h", path);
	exit(1);
}

static struct lxc_arguments my_args = {
	.progname = "lxc-create",
	.helpfn   = create_helpfn,
	.help     = "\
--name=NAME -t template [-w] [-r] [-P lxcpath]\n\
\n\
lxc-create creates a container\n\
\n\
Options :\n\
  -n, --name=NAME    NAME of the container\n\
  -f, --config=file  Initial configuration file\n\
  -t, --template=t   Template to use to setup container\n\
  -B, --bdev=BDEV    Backing store type to use\n\
  -P, --lxcpath=PATH Place container under PATH\n\
  --lvname=LVNAME    Use LVM lv name LVNAME\n\
                     (Default: container name)\n\
  --vgname=VG        Use LVM vg called VG\n\
                     (Default: lxc)\n\
  --thinpool=TP      Use LVM thin pool called TP\n\
                     (Default: lxc)\n\
  --fstype=TYPE      Create fstype TYPE\n\
                     (Default: ext3)\n\
  --fssize=SIZE[U]   Create filesystem of size SIZE * unit U (bBkKmMgGtT)\n\
                     (Default: 1G, default unit: M)\n\
  --dir=DIR          Place rootfs directory under DIR\n\
  --zfsroot=PATH     Create zfs under given zfsroot\n\
                     (Default: tank/lxc)\n",
	.options  = my_longopts,
	.parser   = my_parser,
	.checker  = NULL,
};

static bool validate_bdev_args(struct lxc_arguments *a)
{
	if (strcmp(a->bdevtype, "best") != 0) {
		if (a->fstype || a->fssize) {
			if (strcmp(a->bdevtype, "lvm") != 0 &&
			    strcmp(a->bdevtype, "loop") != 0) {
				fprintf(stderr, "filesystem type and size are only valid with block devices\n");
				return false;
			}
		}
		if (strcmp(a->bdevtype, "lvm") != 0) {
			if (a->lvname || a->vgname || a->thinpool) {
				fprintf(stderr, "--lvname, --vgname and --thinpool are only valid with -B lvm\n");
				return false;
			}
		}
		if (strcmp(a->bdevtype, "zfs") != 0) {
			if (a->zfsroot) {
				fprintf(stderr, "zfsroot is only valid with -B zfs\n");
				return false;
			}
		}
	}
	return true;
}

int main(int argc, char *argv[])
{
	struct lxc_container *c;
	struct bdev_specs spec;
	int flags = 0;

	if (lxc_arguments_parse(&my_args, argc, argv))
		exit(1);

	if (!my_args.log_file)
		my_args.log_file = "none";

	if (lxc_log_init(my_args.name, my_args.log_file, my_args.log_priority,
			 my_args.progname, my_args.quiet, my_args.lxcpath[0]))
		exit(1);
	lxc_log_options_no_override();

	if (!my_args.template) {
		fprintf(stderr, "A template must be specified.\n");
		fprintf(stderr, "Use \"none\" if you really want a container without a rootfs.\n");
		exit(1);
	}

	if (strcmp(my_args.template, "none") == 0)
		my_args.template = NULL;

	memset(&spec, 0, sizeof(spec));
	if (!my_args.bdevtype)
		my_args.bdevtype = "_unset";
	if (!validate_bdev_args(&my_args))
		exit(1);

	if (strcmp(my_args.bdevtype, "none") == 0)
		my_args.bdevtype = "dir";

	if (geteuid()) {
		if (mkdir_p(my_args.lxcpath[0], 0755)) {
			exit(1);
		}
		if (access(my_args.lxcpath[0], O_RDWR) < 0) {
			fprintf(stderr, "You lack access to %s\n", my_args.lxcpath[0]);
			exit(1);
		}
		if (strcmp(my_args.bdevtype, "dir") && strcmp(my_args.bdevtype, "_unset") &&
				strcmp(my_args.bdevtype, "btrfs")) {
			fprintf(stderr, "Unprivileged users cannot create %s containers", my_args.bdevtype);
			exit(1);
		}
	}


	c = lxc_container_new(my_args.name, my_args.lxcpath[0]);
	if (!c) {
		fprintf(stderr, "System error loading container\n");
		exit(1);
	}
	if (c->is_defined(c)) {
		fprintf(stderr, "Container already exists\n");
		exit(1);
	}
	if (my_args.configfile)
		c->load_config(c, my_args.configfile);
	else
		c->load_config(c, lxc_global_config_value("lxc.default_config"));

	if (my_args.fstype)
		spec.fstype = my_args.fstype;
	if (my_args.fssize)
		spec.fssize = my_args.fssize;

	if (strcmp(my_args.bdevtype, "zfs") == 0 || strcmp(my_args.bdevtype, "best") == 0) {
		if (my_args.zfsroot)
			spec.zfs.zfsroot = my_args.zfsroot;
	}
	if (strcmp(my_args.bdevtype, "lvm") == 0 || strcmp(my_args.bdevtype, "best") == 0) {
		if (my_args.lvname)
			spec.lvm.lv = my_args.lvname;
		if (my_args.vgname)
			spec.lvm.vg = my_args.vgname;
		if (my_args.thinpool)
			spec.lvm.thinpool = my_args.thinpool;
	}
	if (my_args.dir) {
		spec.dir = my_args.dir;
	}

	if (strcmp(my_args.bdevtype, "_unset") == 0)
		my_args.bdevtype = NULL;
	if (my_args.quiet)
		flags = LXC_CREATE_QUIET;
	if (!c->create(c, my_args.template, my_args.bdevtype, &spec, flags, &argv[optind])) {
		ERROR("Error creating container %s", c->name);
		lxc_container_put(c);
		exit(1);
	}
	INFO("container %s created", c->name);
	exit(0);
}
Commit	Line	Data
1897e3bc SH	1	/*
	2	*
	3	* Copyright © 2013 Serge Hallyn <serge.hallyn@ubuntu.com>.
	4	* Copyright © 2013 Canonical Ltd.
	5	*
	6	* This program is free software; you can redistribute it and/or modify
	7	* it under the terms of the GNU General Public License version 2, as
	8	* published by the Free Software Foundation.
	9	*
	10	* This program is distributed in the hope that it will be useful,
	11	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	12	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	13	* GNU General Public License for more details.
	14	*
	15	* You should have received a copy of the GNU General Public License along
	16	* with this program; if not, write to the Free Software Foundation, Inc.,
	17	* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
	18	*/
	19
948955a2	20	#include <lxc/lxccontainer.h>
1897e3bc SH	21
	22	#include <stdio.h>
	23	#include <libgen.h>
	24	#include <unistd.h>
	25	#include <ctype.h>
95ee490b	26	#include <fcntl.h>
1897e3bc	27	#include <sys/types.h>
d659597e	28	#include <stdint.h>
1897e3bc	29
f2363e38 ÇO	30	#include "lxc.h"
	31	#include "log.h"
	32	#include "bdev.h"
1897e3bc SH	33	#include "arguments.h"
	34	#include "utils.h"
	35
6ea518f6	36	lxc_log_define(lxc_create_ui, lxc);
1897e3bc	37
d659597e	38	static uint64_t get_fssize(char *s)
1897e3bc	39	{
d659597e	40	uint64_t ret;
1897e3bc SH	41	char *end;
1897e3bc SH	42
d659597e	43	ret = strtoull(s, &end, 0);
1897e3bc	44	if (end == s)
d659597e SA	45	{
d659597e SA	46	fprintf(stderr, "Invalid blockdev size '%s', using default size\n", s);
1897e3bc	47	return 0;
d659597e	48	}
1897e3bc SH	49	while (isblank(*end))
1897e3bc SH	50	end++;
d659597e SA	51	if (*end == '\0')
	52	ret = 1024ULL 1024ULL; // MB by default
	53	else if (end == 'b' \|\| end == 'B')
	54	ret *= 1ULL;
1897e3bc	55	else if (end == 'k' \|\| end == 'K')
d659597e SA	56	ret *= 1024ULL;
	57	else if (end == 'm' \|\| end == 'M')
	58	ret = 1024ULL 1024ULL;
	59	else if (end == 'g' \|\| end == 'G')
	60	ret = 1024ULL 1024ULL * 1024ULL;
	61	else if (end == 't' \|\| end == 'T')
	62	ret = 1024ULL 1024ULL * 1024ULL * 1024ULL;
	63	else
5e8757ed	64	{
d659597e SA	65	fprintf(stderr, "Invalid blockdev unit size '%c' in '%s', using default size\n", *end, s);
	66	return 0;
	67	}
1897e3bc SH	68	return ret;
	69	}
	70
	71	static int my_parser(struct lxc_arguments* args, int c, char* arg)
	72	{
	73	switch (c) {
	74	case 'B': args->bdevtype = arg; break;
	75	case 'f': args->configfile = arg; break;
	76	case 't': args->template = arg; break;
	77	case '0': args->lvname = arg; break;
	78	case '1': args->vgname = arg; break;
f99c386b SS	79	case '2': args->thinpool = arg; break;
	80	case '3': args->fstype = arg; break;
	81	case '4': args->fssize = get_fssize(arg); break;
	82	case '5': args->zfsroot = arg; break;
	83	case '6': args->dir = arg; break;
1897e3bc SH	84	}
	85	return 0;
	86	}
	87
	88	static const struct option my_longopts[] = {
	89	{"bdev", required_argument, 0, 'B'},
	90	{"config", required_argument, 0, 'f'},
	91	{"template", required_argument, 0, 't'},
	92	{"lvname", required_argument, 0, '0'},
	93	{"vgname", required_argument, 0, '1'},
f99c386b SS	94	{"thinpool", required_argument, 0, '2'},
	95	{"fstype", required_argument, 0, '3'},
	96	{"fssize", required_argument, 0, '4'},
	97	{"zfsroot", required_argument, 0, '5'},
	98	{"dir", required_argument, 0, '6'},
1897e3bc SH	99	LXC_COMMON_OPTIONS
	100	};
	101
f002c8a7 SH	102	static void create_helpfn(const struct lxc_arguments *args) {
f002c8a7 SH	103	char argv[3], path;
f002c8a7 SH	104	pid_t pid;
	105
	106	if (!args->template)
	107	return;
19668d8b SG	108
	109	pid = fork();
	110	if (pid) {
f002c8a7	111	wait_for_pid(pid);
19668d8b SG	112	return;
	113	}
	114
6010a416	115	path = get_template_path(args->template);
f002c8a7 SH	116
	117	argv[0] = path;
	118	argv[1] = "-h";
	119	argv[2] = NULL;
	120	execv(path, argv);
	121	ERROR("Error executing %s -h", path);
	122	exit(1);
	123	}
	124
1897e3bc SH	125	static struct lxc_arguments my_args = {
1897e3bc SH	126	.progname = "lxc-create",
f002c8a7	127	.helpfn = create_helpfn,
1897e3bc	128	.help = "\
d3de16bb	129	--name=NAME -t template [-w] [-r] [-P lxcpath]\n\
1897e3bc	130	\n\
3155e7f9	131	lxc-create creates a container\n\
1897e3bc SH	132	\n\
1897e3bc SH	133	Options :\n\
5e8757ed	134	-n, --name=NAME NAME of the container\n\
e3f46bfb SH	135	-f, --config=file Initial configuration file\n\
	136	-t, --template=t Template to use to setup container\n\
	137	-B, --bdev=BDEV Backing store type to use\n\
	138	-P, --lxcpath=PATH Place container under PATH\n\
	139	--lvname=LVNAME Use LVM lv name LVNAME\n\
	140	(Default: container name)\n\
	141	--vgname=VG Use LVM vg called VG\n\
d659597e	142	(Default: lxc)\n\
f99c386b	143	--thinpool=TP Use LVM thin pool called TP\n\
d659597e	144	(Default: lxc)\n\
e3f46bfb	145	--fstype=TYPE Create fstype TYPE\n\
d659597e SA	146	(Default: ext3)\n\
	147	--fssize=SIZE[U] Create filesystem of size SIZE * unit U (bBkKmMgGtT)\n\
	148	(Default: 1G, default unit: M)\n\
e3f46bfb SH	149	--dir=DIR Place rootfs directory under DIR\n\
e3f46bfb SH	150	--zfsroot=PATH Create zfs under given zfsroot\n\
d659597e	151	(Default: tank/lxc)\n",
1897e3bc SH	152	.options = my_longopts,
	153	.parser = my_parser,
	154	.checker = NULL,
	155	};
	156
74a3920a	157	static bool validate_bdev_args(struct lxc_arguments *a)
1897e3bc	158	{
72e99249 SS	159	if (strcmp(a->bdevtype, "best") != 0) {
	160	if (a->fstype \|\| a->fssize) {
	161	if (strcmp(a->bdevtype, "lvm") != 0 &&
	162	strcmp(a->bdevtype, "loop") != 0) {
	163	fprintf(stderr, "filesystem type and size are only valid with block devices\n");
	164	return false;
	165	}
1897e3bc	166	}
72e99249 SS	167	if (strcmp(a->bdevtype, "lvm") != 0) {
	168	if (a->lvname \|\| a->vgname \|\| a->thinpool) {
	169	fprintf(stderr, "--lvname, --vgname and --thinpool are only valid with -B lvm\n");
	170	return false;
	171	}
1897e3bc	172	}
72e99249 SS	173	if (strcmp(a->bdevtype, "zfs") != 0) {
	174	if (a->zfsroot) {
	175	fprintf(stderr, "zfsroot is only valid with -B zfs\n");
	176	return false;
	177	}
1897e3bc SH	178	}
	179	}
	180	return true;
	181	}
	182
1897e3bc SH	183	int main(int argc, char *argv[])
	184	{
	185	struct lxc_container *c;
	186	struct bdev_specs spec;
dc23c1c8	187	int flags = 0;
1897e3bc	188
1897e3bc SH	189	if (lxc_arguments_parse(&my_args, argc, argv))
	190	exit(1);
	191
f5abd74d SG	192	if (!my_args.log_file)
	193	my_args.log_file = "none";
	194
1897e3bc SH	195	if (lxc_log_init(my_args.name, my_args.log_file, my_args.log_priority,
	196	my_args.progname, my_args.quiet, my_args.lxcpath[0]))
	197	exit(1);
6edbfc86	198	lxc_log_options_no_override();
1897e3bc	199
d3de16bb SG	200	if (!my_args.template) {
	201	fprintf(stderr, "A template must be specified.\n");
	202	fprintf(stderr, "Use \"none\" if you really want a container without a rootfs.\n");
	203	exit(1);
	204	}
	205
	206	if (strcmp(my_args.template, "none") == 0)
	207	my_args.template = NULL;
	208
1897e3bc SH	209	memset(&spec, 0, sizeof(spec));
	210	if (!my_args.bdevtype)
	211	my_args.bdevtype = "_unset";
	212	if (!validate_bdev_args(&my_args))
	213	exit(1);
	214
50040b5e SH	215	if (strcmp(my_args.bdevtype, "none") == 0)
	216	my_args.bdevtype = "dir";
	217
460bcbd8	218	if (geteuid()) {
040f1c40 SG	219	if (mkdir_p(my_args.lxcpath[0], 0755)) {
	220	exit(1);
	221	}
460bcbd8 SH	222	if (access(my_args.lxcpath[0], O_RDWR) < 0) {
	223	fprintf(stderr, "You lack access to %s\n", my_args.lxcpath[0]);
	224	exit(1);
	225	}
2659c7cb SH	226	if (strcmp(my_args.bdevtype, "dir") && strcmp(my_args.bdevtype, "_unset") &&
	227	strcmp(my_args.bdevtype, "btrfs")) {
	228	fprintf(stderr, "Unprivileged users cannot create %s containers", my_args.bdevtype);
460bcbd8 SH	229	exit(1);
	230	}
	231	}
	232
	233
1897e3bc SH	234	c = lxc_container_new(my_args.name, my_args.lxcpath[0]);
	235	if (!c) {
	236	fprintf(stderr, "System error loading container\n");
	237	exit(1);
	238	}
	239	if (c->is_defined(c)) {
	240	fprintf(stderr, "Container already exists\n");
	241	exit(1);
	242	}
	243	if (my_args.configfile)
	244	c->load_config(c, my_args.configfile);
	245	else
dad87e3b	246	c->load_config(c, lxc_global_config_value("lxc.default_config"));
1897e3bc	247
72e99249 SS	248	if (my_args.fstype)
	249	spec.fstype = my_args.fstype;
	250	if (my_args.fssize)
	251	spec.fssize = my_args.fssize;
	252
	253	if (strcmp(my_args.bdevtype, "zfs") == 0 \|\| strcmp(my_args.bdevtype, "best") == 0) {
1897e3bc	254	if (my_args.zfsroot)
72e99249 SS	255	spec.zfs.zfsroot = my_args.zfsroot;
	256	}
	257	if (strcmp(my_args.bdevtype, "lvm") == 0 \|\| strcmp(my_args.bdevtype, "best") == 0) {
1897e3bc	258	if (my_args.lvname)
72e99249	259	spec.lvm.lv = my_args.lvname;
1897e3bc	260	if (my_args.vgname)
72e99249	261	spec.lvm.vg = my_args.vgname;
f99c386b	262	if (my_args.thinpool)
72e99249 SS	263	spec.lvm.thinpool = my_args.thinpool;
	264	}
	265	if (my_args.dir) {
7bb87886	266	spec.dir = my_args.dir;
1897e3bc SH	267	}
	268
	269	if (strcmp(my_args.bdevtype, "_unset") == 0)
	270	my_args.bdevtype = NULL;
dc23c1c8 SH	271	if (my_args.quiet)
	272	flags = LXC_CREATE_QUIET;
	273	if (!c->create(c, my_args.template, my_args.bdevtype, &spec, flags, &argv[optind])) {
1897e3bc SH	274	ERROR("Error creating container %s", c->name);
	275	lxc_container_put(c);
	276	exit(1);
	277	}
	278	INFO("container %s created", c->name);
	279	exit(0);
	280	}