X-Git-Url: http://git.linux-vserver.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=util-vserver%2Fsrc%2Fvsched.c;h=3ccc9aa575051305307e03bad2cfaff706004929;hb=a917f24ef7b003dcef54a3db2644cf9cb4bc2db2;hp=1b69318e3c7c9ec4e8996d6c81543ec6e2137ed4;hpb=a45068f573f3051db224977a8c0439233319054f;p=util-vserver.git diff --git a/util-vserver/src/vsched.c b/util-vserver/src/vsched.c index 1b69318..3ccc9aa 100644 --- a/util-vserver/src/vsched.c +++ b/util-vserver/src/vsched.c @@ -50,6 +50,7 @@ CMDLINE_OPTIONS[] = { { "help", no_argument, 0, CMD_HELP }, { "version", no_argument, 0, CMD_VERSION }, { "ctx", required_argument, 0, CMD_XID }, + { "xid", required_argument, 0, CMD_XID }, { "fill-rate", required_argument, 0, CMD_FRATE }, { "interval", required_argument, 0, CMD_INTERVAL }, { "tokens", required_argument, 0, CMD_TOKENS }, @@ -67,7 +68,7 @@ showHelp(int fd, char const *cmd, int res) WRITE_MSG(fd, "Usage:\n "); WRITE_STR(fd, cmd); WRITE_MSG(fd, - " [--ctx ] [--fill-rate ] [--interval ] [--tokens ] [--tokens_min ] [--tokens_max ] [--cpu_mask ] [--] [ *]\n" + " [--xid ] [--fill-rate ] [--interval ] [--tokens ] [--tokens_min ] [--tokens_max ] [--cpu_mask ] [--] [ *]\n" "\n" "Please report bugs to " PACKAGE_BUGREPORT "\n"); @@ -89,6 +90,7 @@ int main(int argc, char *argv[]) { xid_t xid = VC_NOCTX; struct vc_set_sched sched = { 0,0,0,0,0,0 }; + bool do_it = false; while (1) { int c = getopt_long(argc, argv, "+", CMDLINE_OPTIONS, 0); @@ -97,13 +99,13 @@ int main(int argc, char *argv[]) switch (c) { case CMD_HELP : showHelp(1, argv[0], 0); case CMD_VERSION : showVersion(); - case CMD_XID : xid = atoi(optarg); break; - case CMD_FRATE : sched.fill_rate = atoi(optarg); break; - case CMD_INTERVAL : sched.interval = atoi(optarg); break; - case CMD_TOKENS : sched.tokens = atoi(optarg); break; - case CMD_TOK_MIN : sched.tokens_min = atoi(optarg); break; - case CMD_TOK_MAX : sched.tokens_max = atoi(optarg); break; - case CMD_CPU_MASK : sched.cpu_mask = atoi(optarg); break; + case CMD_XID : xid = Evc_xidopt2xid(optarg,true); break; + case CMD_FRATE : sched.fill_rate = atoi(optarg); do_it = true; break; + case CMD_INTERVAL : sched.interval = atoi(optarg); do_it = true; break; + case CMD_TOKENS : sched.tokens = atoi(optarg); do_it = true; break; + case CMD_TOK_MIN : sched.tokens_min = atoi(optarg); do_it = true; break; + case CMD_TOK_MAX : sched.tokens_max = atoi(optarg); do_it = true; break; + case CMD_CPU_MASK : sched.cpu_mask = atoi(optarg); do_it = true; break; default : WRITE_MSG(2, "Try '"); WRITE_STR(2, argv[0]); @@ -114,14 +116,19 @@ int main(int argc, char *argv[]) } if (xid==VC_NOCTX && optind==argc) { - WRITE_MSG(2, "Neither '--xid' nor a program was specified; try '--help' for more information\n"); - exit(255); + WRITE_MSG(2, "Without a program, '--xid' must be used; try '--help' for more information\n"); + exit(wrapper_exit_code); + } + + if (!do_it && optind==argc) { + WRITE_MSG(2, "Neither an option nor a program was specified; try '--help' for more information\n"); + exit(wrapper_exit_code); } if (xid==VC_NOCTX) xid = Evc_get_task_xid(0); - if (vc_set_sched(xid, &sched)==-1) { + if (do_it && vc_set_sched(xid, &sched)==-1) { perror("vc_set_sched()"); exit(255); }