use $(LIBENSCVECTOR) instead of libensc_vector.a
[util-vserver.git] / util-vserver / src / vsched.c
index c78f717..a898090 100644 (file)
 #define CMD_VERSION            0x1001
 #define CMD_XID                        0x4000
 #define CMD_FRATE              0x4001
-#define CMD_PERIOD             0x4002
-#define CMD_FLEVEL             0x4003
-#define CMD_BSIZE              0x4004
+#define CMD_INTERVAL           0x4002
+#define CMD_TOKENS             0x4003
+#define CMD_TOK_MIN            0x4004
+#define CMD_TOK_MAX            0x4005
+#define CMD_CPU_MASK           0x4006
+#define CMD_PRIO_BIAS          0x4007
 
 int                    wrapper_exit_code = 255;
 
@@ -47,11 +50,20 @@ struct option const
 CMDLINE_OPTIONS[] = {
   { "help",     no_argument,  0, CMD_HELP },
   { "version",  no_argument,  0, CMD_VERSION },
-  { "ctx",         required_argument, 0, CMD_XID },
-  { "fill-rate",   required_argument, 0, CMD_FRATE },
-  { "period",      required_argument, 0, CMD_PERIOD },
-  { "fill-level",  required_argument, 0, CMD_FLEVEL },
-  { "bucket-size", required_argument, 0, CMD_BSIZE },
+  { "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 },
+  { "tokens_min",    required_argument, 0, CMD_TOK_MIN },
+  { "tokens-min",    required_argument, 0, CMD_TOK_MIN },
+  { "tokens_max",    required_argument, 0, CMD_TOK_MAX },
+  { "tokens-max",    required_argument, 0, CMD_TOK_MAX },
+  { "prio_bias",     required_argument, 0, CMD_PRIO_BIAS },
+  { "prio-bias",     required_argument, 0, CMD_PRIO_BIAS },
+  { "priority_bias", required_argument, 0, CMD_PRIO_BIAS },
+  { "priority-bias", required_argument, 0, CMD_PRIO_BIAS },
+  { "cpu_mask",      required_argument, 0, CMD_CPU_MASK },
   {0,0,0,0}
 };
 
@@ -63,7 +75,7 @@ showHelp(int fd, char const *cmd, int res)
   WRITE_MSG(fd, "Usage:\n  ");
   WRITE_STR(fd, cmd);
   WRITE_MSG(fd,
-           " [--ctx <xid>] [--fill-rate <rate>] [--period <period>] [--fill-level <level>] [--bucket-size <size>] [--] [<command> <args>*]\n"
+           " [--xid <xid>] [--fill-rate <rate>] [--interval <interval>] [--tokens <tokens>] [--tokens-min <tokens>] [--tokens-max <tokens>] [--prio-bias <bias>] [--] [<command> <args>*]\n"
            "\n"
            "Please report bugs to " PACKAGE_BUGREPORT "\n");
 
@@ -81,10 +93,16 @@ showVersion()
   exit(0);
 }
 
+#define SETVAL(ATTR,MASK) \
+  sched.ATTR      = atoi(optarg); \
+  sched.set_mask |= MASK;
+
 int main(int argc, char *argv[])
 {
   xid_t                        xid   = VC_NOCTX;
-  struct vc_set_sched  sched = { 0,0,0,0 };
+  struct vc_set_sched  sched = {
+    .set_mask = 0
+  };
   
   while (1) {
     int                c = getopt_long(argc, argv, "+", CMDLINE_OPTIONS, 0);
@@ -93,11 +111,16 @@ 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_PERIOD  :  sched.period      = atoi(optarg); break;
-      case CMD_FLEVEL  :  sched.fill_level  = atoi(optarg); break;
-      case CMD_BSIZE   :  sched.bucket_size = atoi(optarg); break;
+      case CMD_XID     :  xid = Evc_xidopt2xid(optarg,true);         break;
+      case CMD_FRATE   :  SETVAL(fill_rate,     VC_VXSM_FILL_RATE);  break;
+      case CMD_INTERVAL        :  SETVAL(interval,      VC_VXSM_INTERVAL);   break;
+      case CMD_TOKENS  :  SETVAL(tokens,        VC_VXSM_TOKENS);     break;
+      case CMD_TOK_MIN :  SETVAL(tokens_min,    VC_VXSM_TOKENS_MIN); break;
+      case CMD_TOK_MAX :  SETVAL(tokens_max,    VC_VXSM_TOKENS_MAX); break;
+      case CMD_PRIO_BIAS:  SETVAL(priority_bias, VC_VXSM_PRIO_BIAS);  break;
+      case CMD_CPU_MASK        :
+       WRITE_MSG(2, "vsched: WARNING: the '--cpu_mask' parameter is deprecated and will not have any effect\n");
+       break;
       default          :
        WRITE_MSG(2, "Try '");
        WRITE_STR(2, argv[0]);
@@ -108,17 +131,25 @@ 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 (sched.set_mask==0 && 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 (sched.set_mask!=0 && vc_set_sched(xid, &sched)==-1) {
     perror("vc_set_sched()");
     exit(255);
   }
 
-  EexecvpD(argv[optind],argv+optind);
+  if (optind<argc)
+    EexecvpD(argv[optind],argv+optind);
+
+  return EXIT_SUCCESS;
 }