1/*-
2 * Copyright (c) 2015 Netflix, Inc.
3 * Written by: Scott Long <scottl@freebsd.org>
4 *
5 * Copyright (c) 2008 Yahoo!, Inc.
6 * All rights reserved.
7 * Written by: John Baldwin <jhb@FreeBSD.org>
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
11 * are met:
12 * 1. Redistributions of source code must retain the above copyright
13 *    notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 *    notice, this list of conditions and the following disclaimer in the
16 *    documentation and/or other materials provided with the distribution.
17 * 3. Neither the name of the author nor the names of any co-contributors
18 *    may be used to endorse or promote products derived from this software
19 *    without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 */
33
34#include <sys/param.h>
35#include <sys/errno.h>
36#include <err.h>
37#include <inttypes.h>
38#include <paths.h>
39#include <stdio.h>
40#include <stdlib.h>
41#include <string.h>
42#include <unistd.h>
43#include "mpsutil.h"
44
45SET_DECLARE(MPS_DATASET(top), struct mpsutil_command);
46SET_DECLARE(MPS_DATASET(usage), struct mpsutil_usage);
47
48int mps_unit;
49int is_mps;
50
51static void
52usage(void)
53{
54	struct mpsutil_usage **cmd;
55	const char *args, *desc;
56
57	fprintf(stderr, "usage: %s [-u unit] <command> ...\n\n", getprogname());
58	fprintf(stderr, "Commands include:\n");
59	SET_FOREACH(cmd, MPS_DATASET(usage)) {
60		if (*cmd == NULL) {
61			fprintf(stderr, "\n");
62		} else {
63			(*cmd)->handler(&args, &desc);
64			if (strncmp((*cmd)->set, "top", 3) == 0)
65				fprintf(stderr, "%-16s %-28s%s\n",
66				    (*cmd)->name, args, desc);
67			else
68				fprintf(stderr, "%-5s %-10s %-28s%s\n",
69				    (*cmd)->set, (*cmd)->name, args, desc);
70		}
71	}
72	exit(1);
73}
74
75static int
76version(int ac, char **av)
77{
78
79	printf("%s: version %s", MPSUTIL_VERSION, getprogname());
80#ifdef DEBUG
81	printf(" (DEBUG)");
82#endif
83	printf("\n");
84	return (0);
85}
86
87MPS_COMMAND(top, version, version, "", "Version number")
88
89int
90main(int ac, char **av)
91{
92	struct mpsutil_command **cmd;
93	uintmax_t unit;
94	char *end;
95	int ch;
96
97	is_mps = !strcmp(getprogname(), "mpsutil");
98
99	while ((ch = getopt(ac, av, "u:h?")) != -1) {
100		switch (ch) {
101		case 'u':
102			if (strncmp(optarg, _PATH_DEV, strlen(_PATH_DEV)) == 0) {
103				optarg += strlen(_PATH_DEV);
104				if (strncmp(optarg, is_mps ? "mps" : "mpr", 3) != 0)
105					errx(1, "Invalid device: %s", optarg);
106			}
107			if (strncmp(optarg, is_mps ? "mps" : "mpr", 3) == 0)
108				optarg += 3;
109			unit = strtoumax(optarg, &end, 10);
110			if (*end != '\0' || unit == UINTMAX_MAX || unit > INT_MAX)
111				errx(1, "Invalid unit: %s", optarg);
112			mps_unit = unit;
113			break;
114		case 'h':
115		case '?':
116			usage();
117			return (1);
118		}
119	}
120
121	av += optind;
122	ac -= optind;
123
124	/* getopt() eats av[0], so we can't use mpt_table_handler() directly. */
125	if (ac == 0) {
126		usage();
127		return (1);
128	}
129
130	SET_FOREACH(cmd, MPS_DATASET(top)) {
131		if (strcmp((*cmd)->name, av[0]) == 0) {
132			if ((*cmd)->handler(ac, av))
133				return (1);
134			else
135				return (0);
136		}
137	}
138	warnx("Unknown command %s.", av[0]);
139	return (1);
140}
141
142int
143mps_table_handler(struct mpsutil_command **start, struct mpsutil_command **end,
144    int ac, char **av)
145{
146	struct mpsutil_command **cmd;
147
148	if (ac < 2) {
149		warnx("The %s command requires a sub-command.", av[0]);
150		return (EINVAL);
151	}
152	for (cmd = start; cmd < end; cmd++) {
153		if (strcmp((*cmd)->name, av[1]) == 0)
154			return ((*cmd)->handler(ac - 1, av + 1));
155	}
156
157	warnx("%s is not a valid sub-command of %s.", av[1], av[0]);
158	return (ENOENT);
159}
160
161void
162hexdump(const void *ptr, int length, const char *hdr, int flags)
163{
164	int i, j, k;
165	int cols;
166	const unsigned char *cp;
167	char delim;
168
169	if ((flags & HD_DELIM_MASK) != 0)
170		delim = (flags & HD_DELIM_MASK) >> 8;
171	else
172		delim = ' ';
173
174	if ((flags & HD_COLUMN_MASK) != 0)
175		cols = flags & HD_COLUMN_MASK;
176	else
177		cols = 16;
178
179	cp = ptr;
180	for (i = 0; i < length; i+= cols) {
181		if (hdr != NULL)
182			printf("%s", hdr);
183
184		if ((flags & HD_OMIT_COUNT) == 0)
185			printf("%04x  ", i);
186
187		if ((flags & HD_OMIT_HEX) == 0) {
188			for (j = 0; j < cols; j++) {
189				if (flags & HD_REVERSED)
190					k = i + (cols - 1 - j);
191				else
192					k = i + j;
193				if (k < length)
194					printf("%c%02x", delim, cp[k]);
195				else
196					printf("   ");
197			}
198		}
199
200		if ((flags & HD_OMIT_CHARS) == 0) {
201			printf("  |");
202			for (j = 0; j < cols; j++) {
203				if (flags & HD_REVERSED)
204					k = i + (cols - 1 - j);
205				else
206					k = i + j;
207				if (k >= length)
208					printf(" ");
209				else if (cp[k] >= ' ' && cp[k] <= '~')
210					printf("%c", cp[k]);
211				else
212					printf(".");
213			}
214			printf("|");
215		}
216		printf("\n");
217	}
218}
219
220#define PCHAR(c) { if (retval < tmpsz) { *outbuf++ = (c); retval++; } }
221
222int
223mps_parse_flags(uintmax_t num, const char *q, char *outbuf, int tmpsz)
224{
225	int n, tmp, retval = 0;
226
227	if (num == 0)
228		return (retval);
229
230	/* %b conversion flag format. */
231	tmp = retval;
232	while (*q) {
233		n = *q++;
234		if (num & (1 << (n - 1))) {
235			PCHAR(retval != tmp ?  ',' : '<');
236			for (; (n = *q) > ' '; ++q)
237				PCHAR(n);
238		} else
239			for (; *q > ' '; ++q)
240				continue;
241	}
242	if (retval != tmp)
243		PCHAR('>');
244
245	return (retval);
246}
247
248