DPDK  20.05.0
examples/vm_power_manager/guest_cli/main.c
/* SPDX-License-Identifier: BSD-3-Clause
* Copyright(c) 2010-2014 Intel Corporation
*/
#include <stdio.h>
#include <stdlib.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)
{
printf("Received signal %d, exiting...\n", signo);
unsigned lcore_id;
RTE_LCORE_FOREACH(lcore_id) {
rte_power_exit(lcore_id);
}
}
#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();
ret = set_policy_defaults(policy);
if (ret != 0) {
printf("Failed to set policy defaults\n");
return -1;
}
argvopt = argv;
while ((opt = getopt_long(argc, argvopt, "n:b:q:p:",
lgopts, &option_index)) != EOF) {
switch (opt) {
/* portmask */
case 'n':
strlcpy(policy->vm_name, optarg, VM_MAX_NAME_SZ);
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);
if (set_policy_mac(i, idx++) != 0) {
printf("Cannot set policy MAC");
return -1;
}
}
}
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)
{
int ret;
unsigned lcore_id;
ret = rte_eal_init(argc, argv);
if (ret < 0)
rte_panic("Cannot init EAL\n");
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);
}
run_cli(NULL);
return 0;
}