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
 20struct 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
 28static 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 */
 35struct cpupower_cpu_info cpupower_cpu_info;
 36int run_as_root;
 37/* Affected cpus chosen by -c/--cpu param */
 38struct bitmask *cpus_chosen;
 39
 40#ifdef DEBUG
 41int be_verbose;
 42#endif
 43
 44static void print_help(void);
 45
 46static 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
 57static 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
 73static 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
 98static 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
109static void print_version(void)
110{
111	printf(PACKAGE " " VERSION "\n");
112	printf(_("Report errors and bugs to %s, please.\n"), PACKAGE_BUGREPORT);
113}
114
115static 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
167int 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}