1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0-only
2*4882a593Smuzhiyun /*
3*4882a593Smuzhiyun * (C) 2011 Thomas Renninger <trenn@suse.de>, Novell Inc.
4*4882a593Smuzhiyun */
5*4882a593Smuzhiyun
6*4882a593Smuzhiyun
7*4882a593Smuzhiyun #include <unistd.h>
8*4882a593Smuzhiyun #include <stdio.h>
9*4882a593Smuzhiyun #include <stdlib.h>
10*4882a593Smuzhiyun #include <errno.h>
11*4882a593Smuzhiyun #include <string.h>
12*4882a593Smuzhiyun #include <getopt.h>
13*4882a593Smuzhiyun #include <sys/utsname.h>
14*4882a593Smuzhiyun
15*4882a593Smuzhiyun #include "helpers/helpers.h"
16*4882a593Smuzhiyun #include "helpers/sysfs.h"
17*4882a593Smuzhiyun #include "helpers/bitmask.h"
18*4882a593Smuzhiyun
19*4882a593Smuzhiyun static struct option set_opts[] = {
20*4882a593Smuzhiyun {"perf-bias", required_argument, NULL, 'b'},
21*4882a593Smuzhiyun { },
22*4882a593Smuzhiyun };
23*4882a593Smuzhiyun
print_wrong_arg_exit(void)24*4882a593Smuzhiyun static void print_wrong_arg_exit(void)
25*4882a593Smuzhiyun {
26*4882a593Smuzhiyun printf(_("invalid or unknown argument\n"));
27*4882a593Smuzhiyun exit(EXIT_FAILURE);
28*4882a593Smuzhiyun }
29*4882a593Smuzhiyun
cmd_set(int argc,char ** argv)30*4882a593Smuzhiyun int cmd_set(int argc, char **argv)
31*4882a593Smuzhiyun {
32*4882a593Smuzhiyun extern char *optarg;
33*4882a593Smuzhiyun extern int optind, opterr, optopt;
34*4882a593Smuzhiyun unsigned int cpu;
35*4882a593Smuzhiyun struct utsname uts;
36*4882a593Smuzhiyun
37*4882a593Smuzhiyun union {
38*4882a593Smuzhiyun struct {
39*4882a593Smuzhiyun int perf_bias:1;
40*4882a593Smuzhiyun };
41*4882a593Smuzhiyun int params;
42*4882a593Smuzhiyun } params;
43*4882a593Smuzhiyun int perf_bias = 0;
44*4882a593Smuzhiyun int ret = 0;
45*4882a593Smuzhiyun
46*4882a593Smuzhiyun ret = uname(&uts);
47*4882a593Smuzhiyun if (!ret && (!strcmp(uts.machine, "ppc64le") ||
48*4882a593Smuzhiyun !strcmp(uts.machine, "ppc64"))) {
49*4882a593Smuzhiyun fprintf(stderr, _("Subcommand not supported on POWER.\n"));
50*4882a593Smuzhiyun return ret;
51*4882a593Smuzhiyun }
52*4882a593Smuzhiyun
53*4882a593Smuzhiyun setlocale(LC_ALL, "");
54*4882a593Smuzhiyun textdomain(PACKAGE);
55*4882a593Smuzhiyun
56*4882a593Smuzhiyun params.params = 0;
57*4882a593Smuzhiyun /* parameter parsing */
58*4882a593Smuzhiyun while ((ret = getopt_long(argc, argv, "b:",
59*4882a593Smuzhiyun set_opts, NULL)) != -1) {
60*4882a593Smuzhiyun switch (ret) {
61*4882a593Smuzhiyun case 'b':
62*4882a593Smuzhiyun if (params.perf_bias)
63*4882a593Smuzhiyun print_wrong_arg_exit();
64*4882a593Smuzhiyun perf_bias = atoi(optarg);
65*4882a593Smuzhiyun if (perf_bias < 0 || perf_bias > 15) {
66*4882a593Smuzhiyun printf(_("--perf-bias param out "
67*4882a593Smuzhiyun "of range [0-%d]\n"), 15);
68*4882a593Smuzhiyun print_wrong_arg_exit();
69*4882a593Smuzhiyun }
70*4882a593Smuzhiyun params.perf_bias = 1;
71*4882a593Smuzhiyun break;
72*4882a593Smuzhiyun default:
73*4882a593Smuzhiyun print_wrong_arg_exit();
74*4882a593Smuzhiyun }
75*4882a593Smuzhiyun }
76*4882a593Smuzhiyun
77*4882a593Smuzhiyun if (!params.params)
78*4882a593Smuzhiyun print_wrong_arg_exit();
79*4882a593Smuzhiyun
80*4882a593Smuzhiyun /* Default is: set all CPUs */
81*4882a593Smuzhiyun if (bitmask_isallclear(cpus_chosen))
82*4882a593Smuzhiyun bitmask_setall(cpus_chosen);
83*4882a593Smuzhiyun
84*4882a593Smuzhiyun /* loop over CPUs */
85*4882a593Smuzhiyun for (cpu = bitmask_first(cpus_chosen);
86*4882a593Smuzhiyun cpu <= bitmask_last(cpus_chosen); cpu++) {
87*4882a593Smuzhiyun
88*4882a593Smuzhiyun if (!bitmask_isbitset(cpus_chosen, cpu))
89*4882a593Smuzhiyun continue;
90*4882a593Smuzhiyun
91*4882a593Smuzhiyun if (sysfs_is_cpu_online(cpu) != 1){
92*4882a593Smuzhiyun fprintf(stderr, _("Cannot set values on CPU %d:"), cpu);
93*4882a593Smuzhiyun fprintf(stderr, _(" *is offline\n"));
94*4882a593Smuzhiyun continue;
95*4882a593Smuzhiyun }
96*4882a593Smuzhiyun
97*4882a593Smuzhiyun if (params.perf_bias) {
98*4882a593Smuzhiyun ret = msr_intel_set_perf_bias(cpu, perf_bias);
99*4882a593Smuzhiyun if (ret) {
100*4882a593Smuzhiyun fprintf(stderr, _("Error setting perf-bias "
101*4882a593Smuzhiyun "value on CPU %d\n"), cpu);
102*4882a593Smuzhiyun break;
103*4882a593Smuzhiyun }
104*4882a593Smuzhiyun }
105*4882a593Smuzhiyun }
106*4882a593Smuzhiyun return ret;
107*4882a593Smuzhiyun }
108