Merge git://bedivere.hansenpartnership.com/git/scsi-rc-fixes-2.6
[linux-2.6.git] / tools / power / cpupower / utils / cpupower.c
1 /*
2  *  (C) 2010,2011       Thomas Renninger <trenn@suse.de>, Novell Inc.
3  *
4  *  Licensed under the terms of the GNU GPL License version 2.
5  *
6  *  Ideas taken over from the perf userspace tool (included in the Linus
7  *  kernel git repo): subcommand builtins and param parsing.
8  */
9
10 #include <stdio.h>
11 #include <stdlib.h>
12 #include <string.h>
13 #include <unistd.h>
14 #include <errno.h>
15
16 #include "builtin.h"
17 #include "helpers/helpers.h"
18 #include "helpers/bitmask.h"
19
20 struct cmd_struct {
21         const char *cmd;
22         int (*main)(int, const char **);
23         int needs_root;
24 };
25
26 #define ARRAY_SIZE(x) (sizeof(x)/sizeof(x[0]))
27
28 static int cmd_help(int argc, const char **argv);
29
30 /* Global cpu_info object available for all binaries
31  * Info only retrieved from CPU 0
32  *
33  * Values will be zero/unknown on non X86 archs
34  */
35 struct cpupower_cpu_info cpupower_cpu_info;
36 int run_as_root;
37 /* Affected cpus chosen by -c/--cpu param */
38 struct bitmask *cpus_chosen;
39
40 #ifdef DEBUG
41 int be_verbose;
42 #endif
43
44 static void print_help(void);
45
46 static struct cmd_struct commands[] = {
47         { "frequency-info",     cmd_freq_info,  0       },
48         { "frequency-set",      cmd_freq_set,   1       },
49         { "idle-info",          cmd_idle_info,  0       },
50         { "set",                cmd_set,        1       },
51         { "info",               cmd_info,       0       },
52         { "monitor",            cmd_monitor,    0       },
53         { "help",               cmd_help,       0       },
54         /*      { "bench",      cmd_bench,      1       }, */
55 };
56
57 static void print_help(void)
58 {
59         unsigned int i;
60
61 #ifdef DEBUG
62         printf(_("Usage:\tcpupower [-d|--debug] [-c|--cpu cpulist ] <command> [<args>]\n"));
63 #else
64         printf(_("Usage:\tcpupower [-c|--cpu cpulist ] <command> [<args>]\n"));
65 #endif
66         printf(_("Supported commands are:\n"));
67         for (i = 0; i < ARRAY_SIZE(commands); i++)
68                 printf("\t%s\n", commands[i].cmd);
69         printf(_("\nNot all commands can make use of the -c cpulist option.\n"));
70         printf(_("\nUse 'cpupower help <command>' for getting help for above commands.\n"));
71 }
72
73 static int print_man_page(const char *subpage)
74 {
75         int len;
76         char *page;
77
78         len = 10; /* enough for "cpupower-" */
79         if (subpage != NULL)
80                 len += strlen(subpage);
81
82         page = malloc(len);
83         if (!page)
84                 return -ENOMEM;
85
86         sprintf(page, "cpupower");
87         if ((subpage != NULL) && strcmp(subpage, "help")) {
88                 strcat(page, "-");
89                 strcat(page, subpage);
90         }
91
92         execlp("man", "man", page, NULL);
93
94         /* should not be reached */
95         return -EINVAL;
96 }
97
98 static int cmd_help(int argc, const char **argv)
99 {
100         if (argc > 1) {
101                 print_man_page(argv[1]); /* exits within execlp() */
102                 return EXIT_FAILURE;
103         }
104
105         print_help();
106         return EXIT_SUCCESS;
107 }
108
109 static void print_version(void)
110 {
111         printf(PACKAGE " " VERSION "\n");
112         printf(_("Report errors and bugs to %s, please.\n"), PACKAGE_BUGREPORT);
113 }
114
115 static void handle_options(int *argc, const char ***argv)
116 {
117         int ret, x, new_argc = 0;
118
119         if (*argc < 1)
120                 return;
121
122         for (x = 0;  x < *argc && ((*argv)[x])[0] == '-'; x++) {
123                 const char *param = (*argv)[x];
124                 if (!strcmp(param, "-h") || !strcmp(param, "--help")) {
125                         print_help();
126                         exit(EXIT_SUCCESS);
127                 } else if (!strcmp(param, "-c") || !strcmp(param, "--cpu")) {
128                         if (*argc < 2) {
129                                 print_help();
130                                 exit(EXIT_FAILURE);
131                         }
132                         if (!strcmp((*argv)[x+1], "all"))
133                                 bitmask_setall(cpus_chosen);
134                         else {
135                                 ret = bitmask_parselist(
136                                                 (*argv)[x+1], cpus_chosen);
137                                 if (ret < 0) {
138                                         fprintf(stderr, _("Error parsing cpu "
139                                                           "list\n"));
140                                         exit(EXIT_FAILURE);
141                                 }
142                         }
143                         x += 1;
144                         /* Cut out param: cpupower -c 1 info -> cpupower info */
145                         new_argc += 2;
146                         continue;
147                 } else if (!strcmp(param, "-v") ||
148                         !strcmp(param, "--version")) {
149                         print_version();
150                         exit(EXIT_SUCCESS);
151 #ifdef DEBUG
152                 } else if (!strcmp(param, "-d") || !strcmp(param, "--debug")) {
153                         be_verbose = 1;
154                         new_argc++;
155                         continue;
156 #endif
157                 } else {
158                         fprintf(stderr, "Unknown option: %s\n", param);
159                         print_help();
160                         exit(EXIT_FAILURE);
161                 }
162         }
163         *argc -= new_argc;
164         *argv += new_argc;
165 }
166
167 int main(int argc, const char *argv[])
168 {
169         const char *cmd;
170         unsigned int i, ret;
171
172         cpus_chosen = bitmask_alloc(sysconf(_SC_NPROCESSORS_CONF));
173
174         argc--;
175         argv += 1;
176
177         handle_options(&argc, &argv);
178
179         cmd = argv[0];
180
181         if (argc < 1) {
182                 print_help();
183                 return EXIT_FAILURE;
184         }
185
186         setlocale(LC_ALL, "");
187         textdomain(PACKAGE);
188
189         /* Turn "perf cmd --help" into "perf help cmd" */
190         if (argc > 1 && !strcmp(argv[1], "--help")) {
191                 argv[1] = argv[0];
192                 argv[0] = cmd = "help";
193         }
194
195         get_cpu_info(0, &cpupower_cpu_info);
196         run_as_root = !getuid();
197
198         for (i = 0; i < ARRAY_SIZE(commands); i++) {
199                 struct cmd_struct *p = commands + i;
200                 if (strcmp(p->cmd, cmd))
201                         continue;
202                 if (!run_as_root && p->needs_root) {
203                         fprintf(stderr, _("Subcommand %s needs root "
204                                           "privileges\n"), cmd);
205                         return EXIT_FAILURE;
206                 }
207                 ret = p->main(argc, argv);
208                 if (cpus_chosen)
209                         bitmask_free(cpus_chosen);
210                 return ret;
211         }
212         print_help();
213         return EXIT_FAILURE;
214 }