int main(int argc, char **argv)
{
struct stat st;
char path[64];
int ret, opt, veid, verbose = 0;
vps_param *param;
while ((opt = getopt(argc, argv, "vh")) > 0) {
switch(opt) {
case 'v':
verbose = 1;
break;
case 'h':
usage(0);
break;
default :
exit(1);
}
}
if (optind == argc)
usage(1);
if (parse_int(argv[optind], &veid)) {
fprintf(stderr, "Invalid CT ID: %s\n", argv[optind]);
exit(1);
}
get_vps_conf_path(veid, path, sizeof(path));
if (stat(path, &st)) {
fprintf(stderr, "Container configuration file: %s not found\n",
path);
exit(1);
}
param = init_vps_param();
if (vps_parse_config(veid, path, param, NULL))
exit(1);
ret = calculate(veid, ¶m->res.ub, verbose);
free_vps_param(param);
exit(ret);
}
int main(int argc, char *argv[], char *envp[])
{
act_t action = -1;
int verbose = 0;
int verbose_custom = 0;
int quiet = 0;
int veid, ret, skiplock = 0;
char buf[256];
vps_param *gparam = NULL, *vps_p = NULL, *cmd_p = NULL;
const char *action_nm;
struct sigaction act;
char *name = NULL, *opt;
_proc_title = argv[0];
_proc_title_len = envp[0] - argv[0];
gparam = init_vps_param();
vps_p = init_vps_param();
cmd_p = init_vps_param();
sigemptyset(&act.sa_mask);
act.sa_handler = SIG_IGN;
act.sa_flags = 0;
sigaction(SIGPIPE, &act, NULL);
while (argc > 1) {
opt = argv[1];
if (!strcmp(opt, "--verbose")) {
verbose++;
verbose_custom = 1;
} else if (!strcmp(opt, "--quiet"))
quiet = 1;
else if (!strcmp(opt, "--version")) {
version(stdout);
exit(0);
} else if (!strcmp(opt, "--skiplock"))
skiplock = YES;
else
break;
argc--; argv++;
}
if (argc <= 1)
usage(VZ_INVALID_PARAMETER_SYNTAX);
action_nm = argv[1];
init_log(NULL, 0, 1, verbose, quiet, NULL);
if (!strcmp(argv[1], "set")) {
init_modules(&g_action, "set");
action = ACTION_SET;
} else if (!strcmp(argv[1], "create")) {
init_modules(&g_action, "create");
action = ACTION_CREATE;
} else if (!strcmp(argv[1], "start")) {
init_modules(&g_action, "set");
action = ACTION_START;
} else if (!strcmp(argv[1], "stop")) {
init_modules(&g_action, "set");
action = ACTION_STOP;
} else if (!strcmp(argv[1], "restart")) {
action = ACTION_RESTART;
} else if (!strcmp(argv[1], "destroy") || !strcmp(argv[1], "delete")) {
action = ACTION_DESTROY;
} else if (!strcmp(argv[1], "mount")) {
action = ACTION_MOUNT;
} else if (!strcmp(argv[1], "umount")) {
action = ACTION_UMOUNT;
} else if (!strcmp(argv[1], "exec3")) {
action = ACTION_EXEC3;
} else if (!strcmp(argv[1], "exec2")) {
action = ACTION_EXEC2;
} else if (!strcmp(argv[1], "exec")) {
action = ACTION_EXEC;
} else if (!strcmp(argv[1], "runscript")) {
action = ACTION_RUNSCRIPT;
} else if (!strcmp(argv[1], "enter")) {
action = ACTION_ENTER;
} else if (!strcmp(argv[1], "console")) {
action = ACTION_CONSOLE;
#ifdef HAVE_PLOOP
} else if (!strcmp(argv[1], "convert")) {
action = ACTION_CONVERT;
} else if (!strcmp(argv[1], "compact")) {
action = ACTION_COMPACT;
#endif
} else if (!strcmp(argv[1], "status")) {
action = ACTION_STATUS;
quiet = 1;
} else if (!strcmp(argv[1], "suspend") || !strcmp(argv[1], "chkpnt")) {
action = ACTION_SUSPEND;
} else if (!strcmp(argv[1], "resume") || !strcmp(argv[1], "restore")) {
action = ACTION_RESUME;
} else if (!strcmp(argv[1], "quotaon")) {
action = ACTION_QUOTAON;
} else if (!strcmp(argv[1], "quotaoff")) {
action = ACTION_QUOTAOFF;
} else if (!strcmp(argv[1], "quotainit")) {
action = ACTION_QUOTAINIT;
#ifdef HAVE_PLOOP
} else if (!strcmp(argv[1], "snapshot")) {
action = ACTION_SNAPSHOT_CREATE;
} else if (!strcmp(argv[1], "snapshot-switch")) {
action = ACTION_SNAPSHOT_SWITCH;
} else if (!strcmp(argv[1], "snapshot-delete")) {
action = ACTION_SNAPSHOT_DELETE;
} else if (!strcmp(argv[1], "snapshot-list")) {
action = ACTION_SNAPSHOT_LIST;
} else if (!strcmp(argv[1], "snapshot-mount")) {
action = ACTION_SNAPSHOT_MOUNT;
} else if (!strcmp(argv[1], "snapshot-umount")) {
action = ACTION_SNAPSHOT_UMOUNT;
#endif
} else if (!strcmp(argv[1], "--help")) {
usage(0);
} else {
init_modules(&g_action, action_nm);
action = ACTION_CUSTOM;
if (!g_action.mod_count) {
fprintf(stderr, "Bad command: %s\n", argv[1]);
ret = VZ_INVALID_PARAMETER_SYNTAX;
goto error;
}
}
if (argc < 3) {
fprintf(stderr, "CT ID missing\n");
ret = VZ_INVALID_PARAMETER_VALUE;
goto error;
}
if (parse_int(argv[2], &veid)) {
name = strdup(argv[2]);
veid = get_veid_by_name(name);
}
if (veid < 0 || veid > VEID_MAX) {
fprintf(stderr, "Bad CT ID %s\n", argv[2]);
ret = VZ_INVALID_PARAMETER_VALUE;
goto error;
}
argc -= 2; argv += 2;
/* getopt_long() prints argv[0] when reporting errors */
argv[0] = _proc_title;
/* Read global config file */
if (vps_parse_config(veid, GLOBAL_CFG, gparam, &g_action)) {
ret = VZ_NOCONFIG;
goto error;
}
init_log(gparam->log.log_file, veid, gparam->log.enable != NO,
gparam->log.level, quiet, "vzctl");
/* Set verbose level from global config if not overwriten
by --verbose
*/
if (!verbose_custom && gparam->log.verbose != NULL) {
verbose = *gparam->log.verbose;
verbose_custom = 1;
}
if (verbose < -1)
verbose = -1;
if (verbose_custom)
set_log_verbose(verbose);
if ((ret = parse_action_opt(veid, action, argc, argv, cmd_p,
action_nm)))
{
goto error;
}
if (veid == 0 && action != ACTION_SET) {
fprintf(stderr, "Only set actions are allowed for CT0\n");
ret = VZ_INVALID_PARAMETER_VALUE;
goto error;
} else if (veid < 0) {
fprintf(stderr, "Bad CT ID %d\n", veid);
ret = VZ_INVALID_PARAMETER_VALUE;
goto error;
}
get_vps_conf_path(veid, buf, sizeof(buf));
if (stat_file(buf) == 1) {
if (vps_parse_config(veid, buf, vps_p, &g_action)) {
ret = VZ_NOCONFIG;
goto error;
}
if (name != NULL &&
vps_p->res.name.name != NULL &&
strcmp(name, vps_p->res.name.name))
{
logger(-1, 0, "Unable to find container by name %s",
name);
ret = VZ_INVALID_PARAMETER_VALUE;
goto error;
}
} else if (action != ACTION_CREATE &&
action != ACTION_STATUS &&
action != ACTION_SET)
{
logger(-1, 0, "Container config file does not exist");
ret = VZ_NOVECONFIG;
goto error;
}
merge_vps_param(gparam, vps_p);
merge_global_param(cmd_p, gparam);
ret = run_action(veid, action, gparam, vps_p, cmd_p, argc, argv,
skiplock);
error:
free_modules(&g_action);
free_vps_param(gparam);
free_vps_param(vps_p);
free_vps_param(cmd_p);
free_log();
free(name);
return ret;
}
int main(int argc, char *argv[], char *envp[])
{
int action = 0;
int verbose = 0;
int verbose_tmp;
int verbose_custom = 0;
int quiet = 0;
int veid, ret, skiplock = 0;
char buf[256];
vps_param *gparam = NULL, *vps_p = NULL, *cmd_p = NULL;
const char *action_nm;
struct sigaction act;
char *name = NULL, *opt;
_proc_title = argv[0];
_proc_title_len = envp[0] - argv[0];
gparam = init_vps_param();
vps_p = init_vps_param();
cmd_p = init_vps_param();
sigemptyset(&act.sa_mask);
act.sa_handler = SIG_IGN;
act.sa_flags = 0;
sigaction(SIGPIPE, &act, NULL);
while (argc > 1) {
opt = argv[1];
if (!strcmp(opt, "--verbose")) {
if (argc > 2 &&
!parse_int(argv[2], &verbose_tmp))
{
verbose += verbose_tmp;
argc--; argv++;
} else {
verbose++;
}
verbose_custom = 1;
} else if (!strncmp(opt, "--verbose=", 10)) {
if (parse_int(opt + 10, &verbose_tmp)) {
fprintf(stderr, "Invalid value for"
" --verbose\n");
exit(VZ_INVALID_PARAMETER_VALUE);
}
verbose += verbose_tmp;
verbose_custom = 1;
} else if (!strcmp(opt, "--quiet"))
quiet = 1;
else if (!strcmp(opt, "--version")) {
version(stdout);
exit(0);
} else if (!strcmp(opt, "--skiplock"))
skiplock = YES;
else
break;
argc--; argv++;
}
if (argc <= 1)
usage(VZ_INVALID_PARAMETER_SYNTAX);
action_nm = argv[1];
init_log(NULL, 0, 1, verbose, 0, NULL);
if (!strcmp(argv[1], "set")) {
init_modules(&g_action, "set");
action = ACTION_SET;
// status = ST_SET;
} else if (!strcmp(argv[1], "create")) {
init_modules(&g_action, "create");
action = ACTION_CREATE;
// status = ST_CREATE;
} else if (!strcmp(argv[1], "start")) {
init_modules(&g_action, "set");
action = ACTION_START;
// status = ST_START;
} else if (!strcmp(argv[1], "stop")) {
init_modules(&g_action, "set");
action = ACTION_STOP;
// status = ST_STOP;
} else if (!strcmp(argv[1], "restart")) {
action = ACTION_RESTART;
// status = ST_RESTART;
} else if (!strcmp(argv[1], "destroy")) {
action = ACTION_DESTROY;
// status = ST_DESTROY;
} else if (!strcmp(argv[1], "mount")) {
action = ACTION_MOUNT;
// status = ST_MOUNT;
} else if (!strcmp(argv[1], "umount")) {
action = ACTION_UMOUNT;
// status = ST_UMOUNT;
} else if (!strcmp(argv[1], "exec3")) {
action = ACTION_EXEC3;
} else if (!strcmp(argv[1], "exec2")) {
action = ACTION_EXEC2;
} else if (!strcmp(argv[1], "exec")) {
action = ACTION_EXEC;
} else if (!strcmp(argv[1], "runscript")) {
action = ACTION_RUNSCRIPT;
} else if (!strcmp(argv[1], "enter")) {
action = ACTION_ENTER;
} else if (!strcmp(argv[1], "status")) {
action = ACTION_STATUS;
quiet = 1;
} else if (!strcmp(argv[1], "chkpnt")) {
action = ACTION_CHKPNT;
} else if (!strcmp(argv[1], "restore")) {
action = ACTION_RESTORE;
} else if (!strcmp(argv[1], "quotaon")) {
action = ACTION_QUOTAON;
} else if (!strcmp(argv[1], "quotaoff")) {
action = ACTION_QUOTAOFF;
} else if (!strcmp(argv[1], "quotainit")) {
action = ACTION_QUOTAINIT;
} else if (!strcmp(argv[1], "--help")) {
usage(0);
} else {
init_modules(&g_action, action_nm);
action = ACTION_CUSTOM;
if (!g_action.mod_count) {
fprintf(stderr, "Bad command: %s\n", argv[1]);
ret = VZ_INVALID_PARAMETER_SYNTAX;
goto error;
}
}
if (argc < 3) {
fprintf(stderr, "CT ID missing\n");
ret = VZ_INVALID_PARAMETER_VALUE;
goto error;
}
if (parse_int(argv[2], &veid)) {
name = strdup(argv[2]);
veid = get_veid_by_name(name);
if (veid < 0) {
fprintf(stderr, "Bad CT ID %s\n", argv[2]);
ret = VZ_INVALID_PARAMETER_VALUE;
goto error;
}
}
argc -= 2; argv += 2;
/* Read global config file */
if (vps_parse_config(veid, GLOBAL_CFG, gparam, &g_action)) {
fprintf(stderr, "Global configuration file %s not found\n",
GLOBAL_CFG);
ret = VZ_NOCONFIG;
goto error;
}
init_log(gparam->log.log_file, veid, gparam->log.enable != NO,
gparam->log.level, quiet, "vzctl");
/* Set verbose level from global config if not overwriten
by --verbose
*/
if (!verbose_custom && gparam->log.verbose != NULL) {
verbose = *gparam->log.verbose;
verbose_custom = 1;
}
if (verbose < -1)
verbose = -1;
if (verbose_custom)
set_log_verbose(verbose);
if ((ret = parse_action_opt(veid, action, argc, argv, cmd_p,
action_nm)))
{
goto error;
}
if (veid == 0 && action != ACTION_SET) {
fprintf(stderr, "Only set actions are allowed for CT0\n");
ret = VZ_INVALID_PARAMETER_VALUE;
goto error;
} else if (veid < 0) {
fprintf(stderr, "Bad CT ID %d\n", veid);
ret = VZ_INVALID_PARAMETER_VALUE;
goto error;
}
get_vps_conf_path(veid, buf, sizeof(buf));
if (stat_file(buf)) {
if (vps_parse_config(veid, buf, vps_p, &g_action)) {
logger(-1, 0, "Error in config file %s",
buf);
ret = VZ_NOCONFIG;
goto error;
}
if (name != NULL &&
vps_p->res.name.name != NULL &&
strcmp(name, vps_p->res.name.name))
{
logger(-1, 0, "Unable to find container by name %s",
name);
ret = VZ_INVALID_PARAMETER_VALUE;
goto error;
}
} else if (action != ACTION_CREATE &&
action != ACTION_STATUS &&
action != ACTION_SET)
{
logger(-1, 0, "Container config file does not exist");
ret = VZ_NOVECONFIG;
goto error;
}
merge_vps_param(gparam, vps_p);
merge_global_param(cmd_p, gparam);
ret = run_action(veid, action, gparam, vps_p, cmd_p, argc-1, argv+1,
skiplock);
error:
free_modules(&g_action);
free_vps_param(gparam);
free_vps_param(vps_p);
free_vps_param(cmd_p);
free_log();
if (name != NULL) free(name);
return ret;
}
