busybox/util-linux/readprofile.c

/* vi: set sw=4 ts=4: */
/*
 *  readprofile.c - used to read /proc/profile
 *
 *  Copyright (C) 1994,1996 Alessandro Rubini (rubini@ipvvis.unipv.it)
 *
 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
 */

/*
 * 1999-02-22 Arkadiusz Mickiewicz <misiek@pld.ORG.PL>
 * - added Native Language Support
 * 1999-09-01 Stephane Eranian <eranian@cello.hpl.hp.com>
 * - 64bit clean patch
 * 3Feb2001 Andrew Morton <andrewm@uow.edu.au>
 * - -M option to write profile multiplier.
 * 2001-11-07 Werner Almesberger <wa@almesberger.net>
 * - byte order auto-detection and -n option
 * 2001-11-09 Werner Almesberger <wa@almesberger.net>
 * - skip step size (index 0)
 * 2002-03-09 John Levon <moz@compsoc.man.ac.uk>
 * - make maplineno do something
 * 2002-11-28 Mads Martin Joergensen +
 * - also try /boot/System.map-`uname -r`
 * 2003-04-09 Werner Almesberger <wa@almesberger.net>
 * - fixed off-by eight error and improved heuristics in byte order detection
 * 2003-08-12 Nikita Danilov <Nikita@Namesys.COM>
 * - added -s option; example of use:
 * "readprofile -s -m /boot/System.map-test | grep __d_lookup | sort -n -k3"
 *
 * Taken from util-linux and adapted for busybox by
 * Paul Mundt <lethal@linux-sh.org>.
 */

#include "libbb.h"
#include <sys/utsname.h>

#define S_LEN 128

/* These are the defaults */
static const char defaultmap[] ALIGN1 = "/boot/System.map";
static const char defaultpro[] ALIGN1 = "/proc/profile";

int readprofile_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int readprofile_main(int argc UNUSED_PARAM, char **argv)
{
	FILE *map;
	const char *mapFile, *proFile;
	unsigned long indx = 1;
	size_t len;
	uint64_t add0 = 0;
	unsigned int step;
	unsigned int *buf, total, fn_len;
	unsigned long long fn_add, next_add;     /* current and next address */
	char fn_name[S_LEN], next_name[S_LEN];   /* current and next name */
	char mapline[S_LEN];
	char mode[8];
	int maplineno = 1;
	int header_printed;
	int multiplier = 0;
	unsigned opt;
	enum {
		OPT_M = (1 << 0),
		OPT_m = (1 << 1),
		OPT_p = (1 << 2),
		OPT_n = (1 << 3),
		OPT_a = (1 << 4),
		OPT_b = (1 << 5),
		OPT_s = (1 << 6),
		OPT_i = (1 << 7),
		OPT_r = (1 << 8),
		OPT_v = (1 << 9),
	};
#define optMult    (opt & OPT_M)
#define optNative  (opt & OPT_n)
#define optAll     (opt & OPT_a)
#define optBins    (opt & OPT_b)
#define optSub     (opt & OPT_s)
#define optInfo    (opt & OPT_i)
#define optReset   (opt & OPT_r)
#define optVerbose (opt & OPT_v)

#define next (current^1)

	proFile = defaultpro;
	mapFile = defaultmap;

	opt_complementary = "M+"; /* -M N */
	opt = getopt32(argv, "M:m:p:nabsirv", &multiplier, &mapFile, &proFile);

	if (opt & (OPT_M|OPT_r)) { /* mult or reset, or both */
		int fd, to_write;

		/*
		 * When writing the multiplier, if the length of the write is
		 * not sizeof(int), the multiplier is not changed
		 */
		to_write = sizeof(int);
		if (!optMult)
			to_write = 1;	/* sth different from sizeof(int) */

		fd = xopen(defaultpro, O_WRONLY);
		xwrite(fd, &multiplier, to_write);
		close(fd);
		return EXIT_SUCCESS;
	}

	/*
	 * Use an fd for the profiling buffer, to skip stdio overhead
	 */
	len = MAXINT(ssize_t);
	buf = xmalloc_xopen_read_close(proFile, &len);
	if (!optNative) {
		int entries = len / sizeof(*buf);
		int big = 0, small = 0, i;
		unsigned *p;

		for (p = buf+1; p < buf+entries; p++) {
			if (*p & ~0U << (sizeof(*buf)*4))
				big++;
			if (*p & ((1 << (sizeof(*buf)*4))-1))
				small++;
		}
		if (big > small) {
			bb_error_msg("assuming reversed byte order, "
				"use -n to force native byte order");
			for (p = buf; p < buf+entries; p++)
				for (i = 0; i < sizeof(*buf)/2; i++) {
					unsigned char *b = (unsigned char *) p;
					unsigned char tmp;

					tmp = b[i];
					b[i] = b[sizeof(*buf)-i-1];
					b[sizeof(*buf)-i-1] = tmp;
				}
		}
	}

	step = buf[0];
	if (optInfo) {
		printf("Sampling_step: %i\n", step);
		return EXIT_SUCCESS;
	}

	total = 0;

	map = xfopen_for_read(mapFile);

	while (fgets(mapline, S_LEN, map)) {
		if (sscanf(mapline, "%llx %s %s", &fn_add, mode, fn_name) != 3)
			bb_error_msg_and_die("%s(%i): wrong map line",
					     mapFile, maplineno);

		if (!strcmp(fn_name, "_stext")) /* only elf works like this */ {
			add0 = fn_add;
			break;
		}
		maplineno++;
	}

	if (!add0)
		bb_error_msg_and_die("can't find \"_stext\" in %s", mapFile);

	/*
	 * Main loop.
	 */
	while (fgets(mapline, S_LEN, map)) {
		unsigned int this = 0;

		if (sscanf(mapline, "%llx %s %s", &next_add, mode, next_name) != 3)
			bb_error_msg_and_die("%s(%i): wrong map line",
					mapFile, maplineno);

		header_printed = 0;

		/* ignore any LEADING (before a '[tT]' symbol is found)
		   Absolute symbols */
		if ((*mode == 'A' || *mode == '?') && total == 0) continue;
		if (*mode != 'T' && *mode != 't' &&
		    *mode != 'W' && *mode != 'w')
			break;	/* only text is profiled */

		if (indx >= len / sizeof(*buf))
			bb_error_msg_and_die("profile address out of range. "
					     "Wrong map file?");

		while (indx < (next_add-add0)/step) {
			if (optBins && (buf[indx] || optAll)) {
				if (!header_printed) {
					printf("%s:\n", fn_name);
					header_printed = 1;
				}
				printf("\t%"PRIx64"\t%u\n", (indx - 1)*step + add0, buf[indx]);
			}
			this += buf[indx++];
		}
		total += this;

		if (optBins) {
			if (optVerbose || this > 0)
				printf("  total\t\t\t\t%u\n", this);
		} else if ((this || optAll)
		        && (fn_len = next_add-fn_add) != 0
		) {
			if (optVerbose)
				printf("%016llx %-40s %6i %8.4f\n", fn_add,
				       fn_name, this, this/(double)fn_len);
			else
				printf("%6i %-40s %8.4f\n",
				       this, fn_name, this/(double)fn_len);
			if (optSub) {
				unsigned long long scan;

				for (scan = (fn_add-add0)/step + 1;
				     scan < (next_add-add0)/step; scan++) {
					unsigned long long addr;

					addr = (scan - 1)*step + add0;
					printf("\t%#llx\t%s+%#llx\t%u\n",
					       addr, fn_name, addr - fn_add,
					       buf[scan]);
				}
			}
		}

		fn_add = next_add;
		strcpy(fn_name, next_name);

		maplineno++;
	}

	/* clock ticks, out of kernel text - probably modules */
	printf("%6i %s\n", buf[len/sizeof(*buf)-1], "*unknown*");

	/* trailer */
	if (optVerbose)
		printf("%016x %-40s %6i %8.4f\n",
		       0, "total", total, total/(double)(fn_add-add0));
	else
		printf("%6i %-40s %8.4f\n",
		       total, "total", total/(double)(fn_add-add0));

	fclose(map);
	free(buf);

	return EXIT_SUCCESS;
}
1	niro	532	/* vi: set sw=4 ts=4: */
2			/*
3			* readprofile.c - used to read /proc/profile
4			*
5			* Copyright (C) 1994,1996 Alessandro Rubini (rubini@ipvvis.unipv.it)
6			*
7			* Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
8			*/
9
10			/*
11	niro	816	* 1999-02-22 Arkadiusz Mickiewicz <misiek@pld.ORG.PL>
12	niro	532	* - added Native Language Support
13			* 1999-09-01 Stephane Eranian <eranian@cello.hpl.hp.com>
14			* - 64bit clean patch
15			* 3Feb2001 Andrew Morton <andrewm@uow.edu.au>
16			* - -M option to write profile multiplier.
17			* 2001-11-07 Werner Almesberger <wa@almesberger.net>
18			* - byte order auto-detection and -n option
19			* 2001-11-09 Werner Almesberger <wa@almesberger.net>
20			* - skip step size (index 0)
21			* 2002-03-09 John Levon <moz@compsoc.man.ac.uk>
22			* - make maplineno do something
23			* 2002-11-28 Mads Martin Joergensen +
24			* - also try /boot/System.map-`uname -r`
25			* 2003-04-09 Werner Almesberger <wa@almesberger.net>
26			* - fixed off-by eight error and improved heuristics in byte order detection
27			* 2003-08-12 Nikita Danilov <Nikita@Namesys.COM>
28			* - added -s option; example of use:
29			* "readprofile -s -m /boot/System.map-test \| grep __d_lookup \| sort -n -k3"
30			*
31			* Taken from util-linux and adapted for busybox by
32			* Paul Mundt <lethal@linux-sh.org>.
33			*/
34
35	niro	816	#include "libbb.h"
36	niro	532	#include <sys/utsname.h>
37
38			#define S_LEN 128
39
40			/* These are the defaults */
41	niro	816	static const char defaultmap[] ALIGN1 = "/boot/System.map";
42			static const char defaultpro[] ALIGN1 = "/proc/profile";
43	niro	532
44	niro	816	int readprofile_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
45			int readprofile_main(int argc UNUSED_PARAM, char **argv)
46	niro	532	{
47			FILE *map;
48	niro	816	const char mapFile, proFile;
49	niro	532	unsigned long indx = 1;
50			size_t len;
51			uint64_t add0 = 0;
52			unsigned int step;
53			unsigned int *buf, total, fn_len;
54			unsigned long long fn_add, next_add; /* current and next address */
55			char fn_name[S_LEN], next_name[S_LEN]; /* current and next name */
56			char mapline[S_LEN];
57			char mode[8];
58			int maplineno = 1;
59			int header_printed;
60	niro	816	int multiplier = 0;
61			unsigned opt;
62			enum {
63			OPT_M = (1 << 0),
64			OPT_m = (1 << 1),
65			OPT_p = (1 << 2),
66			OPT_n = (1 << 3),
67			OPT_a = (1 << 4),
68			OPT_b = (1 << 5),
69			OPT_s = (1 << 6),
70			OPT_i = (1 << 7),
71			OPT_r = (1 << 8),
72			OPT_v = (1 << 9),
73			};
74			#define optMult (opt & OPT_M)
75			#define optNative (opt & OPT_n)
76			#define optAll (opt & OPT_a)
77			#define optBins (opt & OPT_b)
78			#define optSub (opt & OPT_s)
79			#define optInfo (opt & OPT_i)
80			#define optReset (opt & OPT_r)
81			#define optVerbose (opt & OPT_v)
82	niro	532
83			#define next (current^1)
84
85			proFile = defaultpro;
86			mapFile = defaultmap;
87
88	niro	816	opt_complementary = "M+"; /* -M N */
89			opt = getopt32(argv, "M:m:p:nabsirv", &multiplier, &mapFile, &proFile);
90	niro	532
91	niro	816	if (opt & (OPT_M\|OPT_r)) { /* mult or reset, or both */
92			int fd, to_write;
93	niro	532
94			/*
95			* When writing the multiplier, if the length of the write is
96			* not sizeof(int), the multiplier is not changed
97			*/
98	niro	816	to_write = sizeof(int);
99			if (!optMult)
100	niro	532	to_write = 1; /* sth different from sizeof(int) */
101
102			fd = xopen(defaultpro, O_WRONLY);
103	niro	816	xwrite(fd, &multiplier, to_write);
104	niro	532	close(fd);
105			return EXIT_SUCCESS;
106			}
107
108			/*
109			* Use an fd for the profiling buffer, to skip stdio overhead
110			*/
111	niro	816	len = MAXINT(ssize_t);
112			buf = xmalloc_xopen_read_close(proFile, &len);
113	niro	532	if (!optNative) {
114	niro	816	int entries = len / sizeof(*buf);
115	niro	532	int big = 0, small = 0, i;
116			unsigned *p;
117
118			for (p = buf+1; p < buf+entries; p++) {
119			if (p & ~0U << (sizeof(buf)*4))
120			big++;
121			if (p & ((1 << (sizeof(buf)*4))-1))
122			small++;
123			}
124			if (big > small) {
125			bb_error_msg("assuming reversed byte order, "
126			"use -n to force native byte order");
127			for (p = buf; p < buf+entries; p++)
128			for (i = 0; i < sizeof(*buf)/2; i++) {
129			unsigned char b = (unsigned char ) p;
130			unsigned char tmp;
131
132			tmp = b[i];
133			b[i] = b[sizeof(*buf)-i-1];
134			b[sizeof(*buf)-i-1] = tmp;
135			}
136			}
137			}
138
139			step = buf[0];
140			if (optInfo) {
141			printf("Sampling_step: %i\n", step);
142			return EXIT_SUCCESS;
143			}
144
145			total = 0;
146
147	niro	816	map = xfopen_for_read(mapFile);
148	niro	532
149			while (fgets(mapline, S_LEN, map)) {
150			if (sscanf(mapline, "%llx %s %s", &fn_add, mode, fn_name) != 3)
151			bb_error_msg_and_die("%s(%i): wrong map line",
152			mapFile, maplineno);
153
154			if (!strcmp(fn_name, "_stext")) /* only elf works like this */ {
155			add0 = fn_add;
156			break;
157			}
158			maplineno++;
159			}
160
161			if (!add0)
162			bb_error_msg_and_die("can't find \"_stext\" in %s", mapFile);
163
164			/*
165			* Main loop.
166			*/
167			while (fgets(mapline, S_LEN, map)) {
168			unsigned int this = 0;
169
170			if (sscanf(mapline, "%llx %s %s", &next_add, mode, next_name) != 3)
171			bb_error_msg_and_die("%s(%i): wrong map line",
172			mapFile, maplineno);
173
174			header_printed = 0;
175
176			/* ignore any LEADING (before a '[tT]' symbol is found)
177			Absolute symbols */
178			if ((mode == 'A' \|\| mode == '?') && total == 0) continue;
179			if (mode != 'T' && mode != 't' &&
180			mode != 'W' && mode != 'w')
181			break; /* only text is profiled */
182
183			if (indx >= len / sizeof(*buf))
184			bb_error_msg_and_die("profile address out of range. "
185			"Wrong map file?");
186
187			while (indx < (next_add-add0)/step) {
188			if (optBins && (buf[indx] \|\| optAll)) {
189			if (!header_printed) {
190			printf("%s:\n", fn_name);
191			header_printed = 1;
192			}
193			printf("\t%"PRIx64"\t%u\n", (indx - 1)*step + add0, buf[indx]);
194			}
195			this += buf[indx++];
196			}
197			total += this;
198
199			if (optBins) {
200			if (optVerbose \|\| this > 0)
201			printf(" total\t\t\t\t%u\n", this);
202	niro	816	} else if ((this \|\| optAll)
203			&& (fn_len = next_add-fn_add) != 0
204			) {
205	niro	532	if (optVerbose)
206			printf("%016llx %-40s %6i %8.4f\n", fn_add,
207			fn_name, this, this/(double)fn_len);
208			else
209			printf("%6i %-40s %8.4f\n",
210			this, fn_name, this/(double)fn_len);
211			if (optSub) {
212			unsigned long long scan;
213
214			for (scan = (fn_add-add0)/step + 1;
215			scan < (next_add-add0)/step; scan++) {
216			unsigned long long addr;
217
218			addr = (scan - 1)*step + add0;
219			printf("\t%#llx\t%s+%#llx\t%u\n",
220			addr, fn_name, addr - fn_add,
221			buf[scan]);
222			}
223			}
224			}
225
226			fn_add = next_add;
227			strcpy(fn_name, next_name);
228
229			maplineno++;
230			}
231
232			/* clock ticks, out of kernel text - probably modules */
233			printf("%6i %s\n", buf[len/sizeof(buf)-1], "unknown*");
234
235			/* trailer */
236			if (optVerbose)
237			printf("%016x %-40s %6i %8.4f\n",
238			0, "total", total, total/(double)(fn_add-add0));
239			else
240			printf("%6i %-40s %8.4f\n",
241			total, "total", total/(double)(fn_add-add0));
242
243			fclose(map);
244			free(buf);
245
246			return EXIT_SUCCESS;
247			}