New upstream version 18.08

[deb_dpdk.git] / examples / vm_power_manager / guest_cli / main.c

diff --git a/examples/vm_power_manager/guest_cli/main.c b/examples/vm_power_manager/guest_cli/main.c
index b17936d..36365b1 100644 (file)
--- a/examples/vm_power_manager/guest_cli/main.c
+++ b/examples/vm_power_manager/guest_cli/main.c
@@ -2,23 +2,20 @@
  * Copyright(c) 2010-2014 Intel Corporation
  */
 
-/*
 #include <stdio.h>
-#include <string.h>
-#include <stdint.h>
-#include <sys/epoll.h>
-#include <fcntl.h>
-#include <unistd.h>
 #include <stdlib.h>
-#include <errno.h>
-*/
 #include <signal.h>
+#include <getopt.h>
+#include <string.h>
 
 #include <rte_lcore.h>
 #include <rte_power.h>
 #include <rte_debug.h>
+#include <rte_eal.h>
+#include <rte_log.h>
 
 #include "vm_power_cli_guest.h"
+#include "parse.h"
 
 static void
 sig_handler(int signo)
@@ -32,6 +29,136 @@ sig_handler(int signo)
 
 }
 
+#define MAX_HOURS 24
+
+/* Parse the argument given in the command line of the application */
+static int
+parse_args(int argc, char **argv)
+{
+       int opt, ret;
+       char **argvopt;
+       int option_index;
+       char *prgname = argv[0];
+       const struct option lgopts[] = {
+               { "vm-name", required_argument, 0, 'n'},
+               { "busy-hours", required_argument, 0, 'b'},
+               { "quiet-hours", required_argument, 0, 'q'},
+               { "port-list", required_argument, 0, 'p'},
+               { "vcpu-list", required_argument, 0, 'l'},
+               { "policy", required_argument, 0, 'o'},
+               {NULL, 0, 0, 0}
+       };
+       struct channel_packet *policy;
+       unsigned short int hours[MAX_HOURS];
+       unsigned short int cores[MAX_VCPU_PER_VM];
+       unsigned short int ports[MAX_VCPU_PER_VM];
+       int i, cnt, idx;
+
+       policy = get_policy();
+       set_policy_defaults(policy);
+
+       argvopt = argv;
+
+       while ((opt = getopt_long(argc, argvopt, "n:b:q:p:",
+                                 lgopts, &option_index)) != EOF) {
+
+               switch (opt) {
+               /* portmask */
+               case 'n':
+                       strcpy(policy->vm_name, optarg);
+                       printf("Setting VM Name to [%s]\n", policy->vm_name);
+                       break;
+               case 'b':
+               case 'q':
+                       //printf("***Processing set using [%s]\n", optarg);
+                       cnt = parse_set(optarg, hours, MAX_HOURS);
+                       if (cnt < 0) {
+                               printf("Invalid value passed to quiet/busy hours - [%s]\n",
+                                               optarg);
+                               break;
+                       }
+                       idx = 0;
+                       for (i = 0; i < MAX_HOURS; i++) {
+                               if (hours[i]) {
+                                       if (opt == 'b') {
+                                               printf("***Busy Hour %d\n", i);
+                                               policy->timer_policy.busy_hours
+                                                       [idx++] = i;
+                                       } else {
+                                               printf("***Quiet Hour %d\n", i);
+                                               policy->timer_policy.quiet_hours
+                                                       [idx++] = i;
+                                       }
+                               }
+                       }
+                       break;
+               case 'l':
+                       cnt = parse_set(optarg, cores, MAX_VCPU_PER_VM);
+                       if (cnt < 0) {
+                               printf("Invalid value passed to vcpu-list - [%s]\n",
+                                               optarg);
+                               break;
+                       }
+                       idx = 0;
+                       for (i = 0; i < MAX_VCPU_PER_VM; i++) {
+                               if (cores[i]) {
+                                       printf("***Using core %d\n", i);
+                                       policy->vcpu_to_control[idx++] = i;
+                               }
+                       }
+                       policy->num_vcpu = idx;
+                       printf("Total cores: %d\n", idx);
+                       break;
+               case 'p':
+                       cnt = parse_set(optarg, ports, MAX_VCPU_PER_VM);
+                       if (cnt < 0) {
+                               printf("Invalid value passed to port-list - [%s]\n",
+                                               optarg);
+                               break;
+                       }
+                       idx = 0;
+                       for (i = 0; i < MAX_VCPU_PER_VM; i++) {
+                               if (ports[i]) {
+                                       printf("***Using port %d\n", i);
+                                       set_policy_mac(i, idx++);
+                               }
+                       }
+                       policy->nb_mac_to_monitor = idx;
+                       printf("Total Ports: %d\n", idx);
+                       break;
+               case 'o':
+                       if (!strcmp(optarg, "TRAFFIC"))
+                               policy->policy_to_use = TRAFFIC;
+                       else if (!strcmp(optarg, "TIME"))
+                               policy->policy_to_use = TIME;
+                       else if (!strcmp(optarg, "WORKLOAD"))
+                               policy->policy_to_use = WORKLOAD;
+                       else if (!strcmp(optarg, "BRANCH_RATIO"))
+                               policy->policy_to_use = BRANCH_RATIO;
+                       else {
+                               printf("Invalid policy specified: %s\n",
+                                               optarg);
+                               return -1;
+                       }
+                       break;
+               /* long options */
+
+               case 0:
+                       break;
+
+               default:
+                       return -1;
+               }
+       }
+
+       if (optind >= 0)
+               argv[optind-1] = prgname;
+
+       ret = optind-1;
+       optind = 0; /* reset getopt lib */
+       return ret;
+}
+
 int
 main(int argc, char **argv)
 {
@@ -45,6 +172,14 @@ main(int argc, char **argv)
        signal(SIGINT, sig_handler);
        signal(SIGTERM, sig_handler);
 
+       argc -= ret;
+       argv += ret;
+
+       /* parse application arguments (after the EAL ones) */
+       ret = parse_args(argc, argv);
+       if (ret < 0)
+               rte_exit(EXIT_FAILURE, "Invalid arguments\n");
+
        rte_power_set_env(PM_ENV_KVM_VM);
        RTE_LCORE_FOREACH(lcore_id) {
                rte_power_init(lcore_id);