[mirror_ubuntu-artful-kernel.git] / tools / perf / util / util.c

#include "../perf.h"
#include "util.h"
#include "debug.h"
#include <api/fs/fs.h>
#include <sys/mman.h>
#include <sys/utsname.h>
#ifdef HAVE_BACKTRACE_SUPPORT
#include <execinfo.h>
#endif
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <limits.h>
#include <byteswap.h>
#include <linux/kernel.h>
#include <unistd.h>
#include "callchain.h"

struct callchain_param	callchain_param = {
	.mode	= CHAIN_GRAPH_ABS,
	.min_percent = 0.5,
	.order  = ORDER_CALLEE,
	.key	= CCKEY_FUNCTION
};

/*
 * XXX We need to find a better place for these things...
 */
unsigned int page_size;
int cacheline_size;

bool test_attr__enabled;

bool perf_host  = true;
bool perf_guest = false;

void event_attr_init(struct perf_event_attr *attr)
{
	if (!perf_host)
		attr->exclude_host  = 1;
	if (!perf_guest)
		attr->exclude_guest = 1;
	/* to capture ABI version */
	attr->size = sizeof(*attr);
}

int mkdir_p(char *path, mode_t mode)
{
	struct stat st;
	int err;
	char *d = path;

	if (*d != '/')
		return -1;

	if (stat(path, &st) == 0)
		return 0;

	while (*++d == '/');

	while ((d = strchr(d, '/'))) {
		*d = '\0';
		err = stat(path, &st) && mkdir(path, mode);
		*d++ = '/';
		if (err)
			return -1;
		while (*d == '/')
			++d;
	}
	return (stat(path, &st) && mkdir(path, mode)) ? -1 : 0;
}

int rm_rf(char *path)
{
	DIR *dir;
	int ret = 0;
	struct dirent *d;
	char namebuf[PATH_MAX];

	dir = opendir(path);
	if (dir == NULL)
		return 0;

	while ((d = readdir(dir)) != NULL && !ret) {
		struct stat statbuf;

		if (!strcmp(d->d_name, ".") || !strcmp(d->d_name, ".."))
			continue;

		scnprintf(namebuf, sizeof(namebuf), "%s/%s",
			  path, d->d_name);

		ret = stat(namebuf, &statbuf);
		if (ret < 0) {
			pr_debug("stat failed: %s\n", namebuf);
			break;
		}

		if (S_ISREG(statbuf.st_mode))
			ret = unlink(namebuf);
		else if (S_ISDIR(statbuf.st_mode))
			ret = rm_rf(namebuf);
		else {
			pr_debug("unknown file: %s\n", namebuf);
			ret = -1;
		}
	}
	closedir(dir);

	if (ret < 0)
		return ret;

	return rmdir(path);
}

static int slow_copyfile(const char *from, const char *to)
{
	int err = -1;
	char *line = NULL;
	size_t n;
	FILE *from_fp = fopen(from, "r"), *to_fp;

	if (from_fp == NULL)
		goto out;

	to_fp = fopen(to, "w");
	if (to_fp == NULL)
		goto out_fclose_from;

	while (getline(&line, &n, from_fp) > 0)
		if (fputs(line, to_fp) == EOF)
			goto out_fclose_to;
	err = 0;
out_fclose_to:
	fclose(to_fp);
	free(line);
out_fclose_from:
	fclose(from_fp);
out:
	return err;
}

int copyfile_offset(int ifd, loff_t off_in, int ofd, loff_t off_out, u64 size)
{
	void *ptr;
	loff_t pgoff;

	pgoff = off_in & ~(page_size - 1);
	off_in -= pgoff;

	ptr = mmap(NULL, off_in + size, PROT_READ, MAP_PRIVATE, ifd, pgoff);
	if (ptr == MAP_FAILED)
		return -1;

	while (size) {
		ssize_t ret = pwrite(ofd, ptr + off_in, size, off_out);
		if (ret < 0 && errno == EINTR)
			continue;
		if (ret <= 0)
			break;

		size -= ret;
		off_in += ret;
		off_out -= ret;
	}
	munmap(ptr, off_in + size);

	return size ? -1 : 0;
}

int copyfile_mode(const char *from, const char *to, mode_t mode)
{
	int fromfd, tofd;
	struct stat st;
	int err = -1;
	char *tmp = NULL, *ptr = NULL;

	if (stat(from, &st))
		goto out;

	/* extra 'x' at the end is to reserve space for '.' */
	if (asprintf(&tmp, "%s.XXXXXXx", to) < 0) {
		tmp = NULL;
		goto out;
	}
	ptr = strrchr(tmp, '/');
	if (!ptr)
		goto out;
	ptr = memmove(ptr + 1, ptr, strlen(ptr) - 1);
	*ptr = '.';

	tofd = mkstemp(tmp);
	if (tofd < 0)
		goto out;

	if (fchmod(tofd, mode))
		goto out_close_to;

	if (st.st_size == 0) { /* /proc? do it slowly... */
		err = slow_copyfile(from, tmp);
		goto out_close_to;
	}

	fromfd = open(from, O_RDONLY);
	if (fromfd < 0)
		goto out_close_to;

	err = copyfile_offset(fromfd, 0, tofd, 0, st.st_size);

	close(fromfd);
out_close_to:
	close(tofd);
	if (!err)
		err = link(tmp, to);
	unlink(tmp);
out:
	free(tmp);
	return err;
}

int copyfile(const char *from, const char *to)
{
	return copyfile_mode(from, to, 0755);
}

unsigned long convert_unit(unsigned long value, char *unit)
{
	*unit = ' ';

	if (value > 1000) {
		value /= 1000;
		*unit = 'K';
	}

	if (value > 1000) {
		value /= 1000;
		*unit = 'M';
	}

	if (value > 1000) {
		value /= 1000;
		*unit = 'G';
	}

	return value;
}

static ssize_t ion(bool is_read, int fd, void *buf, size_t n)
{
	void *buf_start = buf;
	size_t left = n;

	while (left) {
		ssize_t ret = is_read ? read(fd, buf, left) :
					write(fd, buf, left);

		if (ret < 0 && errno == EINTR)
			continue;
		if (ret <= 0)
			return ret;

		left -= ret;
		buf  += ret;
	}

	BUG_ON((size_t)(buf - buf_start) != n);
	return n;
}

/*
 * Read exactly 'n' bytes or return an error.
 */
ssize_t readn(int fd, void *buf, size_t n)
{
	return ion(true, fd, buf, n);
}

/*
 * Write exactly 'n' bytes or return an error.
 */
ssize_t writen(int fd, void *buf, size_t n)
{
	return ion(false, fd, buf, n);
}

size_t hex_width(u64 v)
{
	size_t n = 1;

	while ((v >>= 4))
		++n;

	return n;
}

static int hex(char ch)
{
	if ((ch >= '0') && (ch <= '9'))
		return ch - '0';
	if ((ch >= 'a') && (ch <= 'f'))
		return ch - 'a' + 10;
	if ((ch >= 'A') && (ch <= 'F'))
		return ch - 'A' + 10;
	return -1;
}

/*
 * While we find nice hex chars, build a long_val.
 * Return number of chars processed.
 */
int hex2u64(const char *ptr, u64 *long_val)
{
	const char *p = ptr;
	*long_val = 0;

	while (*p) {
		const int hex_val = hex(*p);

		if (hex_val < 0)
			break;

		*long_val = (*long_val << 4) | hex_val;
		p++;
	}

	return p - ptr;
}

/* Obtain a backtrace and print it to stdout. */
#ifdef HAVE_BACKTRACE_SUPPORT
void dump_stack(void)
{
	void *array[16];
	size_t size = backtrace(array, ARRAY_SIZE(array));
	char **strings = backtrace_symbols(array, size);
	size_t i;

	printf("Obtained %zd stack frames.\n", size);

	for (i = 0; i < size; i++)
		printf("%s\n", strings[i]);

	free(strings);
}
#else
void dump_stack(void) {}
#endif

void sighandler_dump_stack(int sig)
{
	psignal(sig, "perf");
	dump_stack();
	exit(sig);
}

void get_term_dimensions(struct winsize *ws)
{
	char *s = getenv("LINES");

	if (s != NULL) {
		ws->ws_row = atoi(s);
		s = getenv("COLUMNS");
		if (s != NULL) {
			ws->ws_col = atoi(s);
			if (ws->ws_row && ws->ws_col)
				return;
		}
	}
#ifdef TIOCGWINSZ
	if (ioctl(1, TIOCGWINSZ, ws) == 0 &&
	    ws->ws_row && ws->ws_col)
		return;
#endif
	ws->ws_row = 25;
	ws->ws_col = 80;
}

void set_term_quiet_input(struct termios *old)
{
	struct termios tc;

	tcgetattr(0, old);
	tc = *old;
	tc.c_lflag &= ~(ICANON | ECHO);
	tc.c_cc[VMIN] = 0;
	tc.c_cc[VTIME] = 0;
	tcsetattr(0, TCSANOW, &tc);
}

int parse_nsec_time(const char *str, u64 *ptime)
{
	u64 time_sec, time_nsec;
	char *end;

	time_sec = strtoul(str, &end, 10);
	if (*end != '.' && *end != '\0')
		return -1;

	if (*end == '.') {
		int i;
		char nsec_buf[10];

		if (strlen(++end) > 9)
			return -1;

		strncpy(nsec_buf, end, 9);
		nsec_buf[9] = '\0';

		/* make it nsec precision */
		for (i = strlen(nsec_buf); i < 9; i++)
			nsec_buf[i] = '0';

		time_nsec = strtoul(nsec_buf, &end, 10);
		if (*end != '\0')
			return -1;
	} else
		time_nsec = 0;

	*ptime = time_sec * NSEC_PER_SEC + time_nsec;
	return 0;
}

unsigned long parse_tag_value(const char *str, struct parse_tag *tags)
{
	struct parse_tag *i = tags;

	while (i->tag) {
		char *s;

		s = strchr(str, i->tag);
		if (s) {
			unsigned long int value;
			char *endptr;

			value = strtoul(str, &endptr, 10);
			if (s != endptr)
				break;

			if (value > ULONG_MAX / i->mult)
				break;
			value *= i->mult;
			return value;
		}
		i++;
	}

	return (unsigned long) -1;
}

int get_stack_size(const char *str, unsigned long *_size)
{
	char *endptr;
	unsigned long size;
	unsigned long max_size = round_down(USHRT_MAX, sizeof(u64));

	size = strtoul(str, &endptr, 0);

	do {
		if (*endptr)
			break;

		size = round_up(size, sizeof(u64));
		if (!size || size > max_size)
			break;

		*_size = size;
		return 0;

	} while (0);

	pr_err("callchain: Incorrect stack dump size (max %ld): %s\n",
	       max_size, str);
	return -1;
}

int parse_callchain_record(const char *arg, struct callchain_param *param)
{
	char *tok, *name, *saveptr = NULL;
	char *buf;
	int ret = -1;

	/* We need buffer that we know we can write to. */
	buf = malloc(strlen(arg) + 1);
	if (!buf)
		return -ENOMEM;

	strcpy(buf, arg);

	tok = strtok_r((char *)buf, ",", &saveptr);
	name = tok ? : (char *)buf;

	do {
		/* Framepointer style */
		if (!strncmp(name, "fp", sizeof("fp"))) {
			if (!strtok_r(NULL, ",", &saveptr)) {
				param->record_mode = CALLCHAIN_FP;
				ret = 0;
			} else
				pr_err("callchain: No more arguments "
				       "needed for --call-graph fp\n");
			break;

#ifdef HAVE_DWARF_UNWIND_SUPPORT
		/* Dwarf style */
		} else if (!strncmp(name, "dwarf", sizeof("dwarf"))) {
			const unsigned long default_stack_dump_size = 8192;

			ret = 0;
			param->record_mode = CALLCHAIN_DWARF;
			param->dump_size = default_stack_dump_size;

			tok = strtok_r(NULL, ",", &saveptr);
			if (tok) {
				unsigned long size = 0;

				ret = get_stack_size(tok, &size);
				param->dump_size = size;
			}
#endif /* HAVE_DWARF_UNWIND_SUPPORT */
		} else if (!strncmp(name, "lbr", sizeof("lbr"))) {
			if (!strtok_r(NULL, ",", &saveptr)) {
				param->record_mode = CALLCHAIN_LBR;
				ret = 0;
			} else
				pr_err("callchain: No more arguments "
					"needed for --call-graph lbr\n");
			break;
		} else {
			pr_err("callchain: Unknown --call-graph option "
			       "value: %s\n", arg);
			break;
		}

	} while (0);

	free(buf);
	return ret;
}

int filename__read_str(const char *filename, char **buf, size_t *sizep)
{
	size_t size = 0, alloc_size = 0;
	void *bf = NULL, *nbf;
	int fd, n, err = 0;
	char sbuf[STRERR_BUFSIZE];

	fd = open(filename, O_RDONLY);
	if (fd < 0)
		return -errno;

	do {
		if (size == alloc_size) {
			alloc_size += BUFSIZ;
			nbf = realloc(bf, alloc_size);
			if (!nbf) {
				err = -ENOMEM;
				break;
			}

			bf = nbf;
		}

		n = read(fd, bf + size, alloc_size - size);
		if (n < 0) {
			if (size) {
				pr_warning("read failed %d: %s\n", errno,
					 strerror_r(errno, sbuf, sizeof(sbuf)));
				err = 0;
			} else
				err = -errno;

			break;
		}

		size += n;
	} while (n > 0);

	if (!err) {
		*sizep = size;
		*buf   = bf;
	} else
		free(bf);

	close(fd);
	return err;
}

const char *get_filename_for_perf_kvm(void)
{
	const char *filename;

	if (perf_host && !perf_guest)
		filename = strdup("perf.data.host");
	else if (!perf_host && perf_guest)
		filename = strdup("perf.data.guest");
	else
		filename = strdup("perf.data.kvm");

	return filename;
}

int perf_event_paranoid(void)
{
	int value;

	if (sysctl__read_int("kernel/perf_event_paranoid", &value))
		return INT_MAX;

	return value;
}

void mem_bswap_32(void *src, int byte_size)
{
	u32 *m = src;
	while (byte_size > 0) {
		*m = bswap_32(*m);
		byte_size -= sizeof(u32);
		++m;
	}
}

void mem_bswap_64(void *src, int byte_size)
{
	u64 *m = src;

	while (byte_size > 0) {
		*m = bswap_64(*m);
		byte_size -= sizeof(u64);
		++m;
	}
}

bool find_process(const char *name)
{
	size_t len = strlen(name);
	DIR *dir;
	struct dirent *d;
	int ret = -1;

	dir = opendir(procfs__mountpoint());
	if (!dir)
		return false;

	/* Walk through the directory. */
	while (ret && (d = readdir(dir)) != NULL) {
		char path[PATH_MAX];
		char *data;
		size_t size;

		if ((d->d_type != DT_DIR) ||
		     !strcmp(".", d->d_name) ||
		     !strcmp("..", d->d_name))
			continue;

		scnprintf(path, sizeof(path), "%s/%s/comm",
			  procfs__mountpoint(), d->d_name);

		if (filename__read_str(path, &data, &size))
			continue;

		ret = strncmp(name, data, len);
		free(data);
	}

	closedir(dir);
	return ret ? false : true;
}

int
fetch_kernel_version(unsigned int *puint, char *str,
		     size_t str_size)
{
	struct utsname utsname;
	int version, patchlevel, sublevel, err;

	if (uname(&utsname))
		return -1;

	if (str && str_size) {
		strncpy(str, utsname.release, str_size);
		str[str_size - 1] = '\0';
	}

	err = sscanf(utsname.release, "%d.%d.%d",
		     &version, &patchlevel, &sublevel);

	if (err != 3) {
		pr_debug("Unablt to get kernel version from uname '%s'\n",
			 utsname.release);
		return -1;
	}

	if (puint)
		*puint = (version << 16) + (patchlevel << 8) + sublevel;
	return 0;
}
Commit	Line	Data
1aed2671	1	#include "../perf.h"
4cf40131	2	#include "util.h"
84f5d36f	3	#include "debug.h"
cd0cfad7	4	#include <api/fs/fs.h>
69e3f52d	5	#include <sys/mman.h>
07bc5c69	6	#include <sys/utsname.h>
89fe808a	7	#ifdef HAVE_BACKTRACE_SUPPORT
dc4552bf	8	#include <execinfo.h>
c9f08bee	9	#endif
dc4552bf ACM	10	#include <stdio.h>
dc4552bf ACM	11	#include <stdlib.h>
cef82c9f JO	12	#include <string.h>
cef82c9f JO	13	#include <errno.h>
1a47245d	14	#include <limits.h>
71db07b1	15	#include <byteswap.h>
838d1452	16	#include <linux/kernel.h>
9398c484	17	#include <unistd.h>
23aadb1f JO	18	#include "callchain.h"
	19
	20	struct callchain_param callchain_param = {
def02db0	21	.mode = CHAIN_GRAPH_ABS,
23aadb1f	22	.min_percent = 0.5,
792aeafa	23	.order = ORDER_CALLEE,
23aadb1f JO	24	.key = CCKEY_FUNCTION
23aadb1f JO	25	};
4cf40131	26
1aed2671 JR	27	/*
	28	* XXX We need to find a better place for these things...
	29	*/
0c1fe6b2	30	unsigned int page_size;
2b1b7100	31	int cacheline_size;
0c1fe6b2	32
0c6332e9 ACM	33	bool test_attr__enabled;
0c6332e9 ACM	34
1aed2671	35	bool perf_host = true;
c4a7dca9	36	bool perf_guest = false;
1aed2671 JR	37
	38	void event_attr_init(struct perf_event_attr *attr)
	39	{
	40	if (!perf_host)
	41	attr->exclude_host = 1;
	42	if (!perf_guest)
	43	attr->exclude_guest = 1;
7e1ccd38 SE	44	/* to capture ABI version */
7e1ccd38 SE	45	attr->size = sizeof(*attr);
1aed2671 JR	46	}
1aed2671 JR	47
4cf40131 ACM	48	int mkdir_p(char *path, mode_t mode)
	49	{
	50	struct stat st;
	51	int err;
	52	char *d = path;
	53
	54	if (*d != '/')
	55	return -1;
	56
	57	if (stat(path, &st) == 0)
	58	return 0;
	59
	60	while (*++d == '/');
	61
	62	while ((d = strchr(d, '/'))) {
	63	*d = '\0';
	64	err = stat(path, &st) && mkdir(path, mode);
	65	*d++ = '/';
	66	if (err)
	67	return -1;
	68	while (*d == '/')
	69	++d;
	70	}
	71	return (stat(path, &st) && mkdir(path, mode)) ? -1 : 0;
	72	}
	73
0b1de0be NK	74	int rm_rf(char *path)
	75	{
	76	DIR *dir;
	77	int ret = 0;
	78	struct dirent *d;
	79	char namebuf[PATH_MAX];
	80
	81	dir = opendir(path);
	82	if (dir == NULL)
	83	return 0;
	84
	85	while ((d = readdir(dir)) != NULL && !ret) {
	86	struct stat statbuf;
	87
	88	if (!strcmp(d->d_name, ".") \|\| !strcmp(d->d_name, ".."))
	89	continue;
	90
	91	scnprintf(namebuf, sizeof(namebuf), "%s/%s",
	92	path, d->d_name);
	93
	94	ret = stat(namebuf, &statbuf);
	95	if (ret < 0) {
	96	pr_debug("stat failed: %s\n", namebuf);
	97	break;
	98	}
	99
	100	if (S_ISREG(statbuf.st_mode))
	101	ret = unlink(namebuf);
	102	else if (S_ISDIR(statbuf.st_mode))
	103	ret = rm_rf(namebuf);
	104	else {
	105	pr_debug("unknown file: %s\n", namebuf);
	106	ret = -1;
	107	}
	108	}
	109	closedir(dir);
	110
	111	if (ret < 0)
	112	return ret;
	113
	114	return rmdir(path);
	115	}
	116
d7c72606	117	static int slow_copyfile(const char from, const char to)
9e201442	118	{
9a17d726	119	int err = -1;
9e201442 ACM	120	char *line = NULL;
	121	size_t n;
	122	FILE from_fp = fopen(from, "r"), to_fp;
	123
	124	if (from_fp == NULL)
	125	goto out;
	126
	127	to_fp = fopen(to, "w");
	128	if (to_fp == NULL)
	129	goto out_fclose_from;
	130
	131	while (getline(&line, &n, from_fp) > 0)
	132	if (fputs(line, to_fp) == EOF)
	133	goto out_fclose_to;
	134	err = 0;
	135	out_fclose_to:
	136	fclose(to_fp);
	137	free(line);
	138	out_fclose_from:
	139	fclose(from_fp);
	140	out:
	141	return err;
	142	}
	143
9c9f5a2f NK	144	int copyfile_offset(int ifd, loff_t off_in, int ofd, loff_t off_out, u64 size)
	145	{
	146	void *ptr;
	147	loff_t pgoff;
	148
	149	pgoff = off_in & ~(page_size - 1);
	150	off_in -= pgoff;
	151
	152	ptr = mmap(NULL, off_in + size, PROT_READ, MAP_PRIVATE, ifd, pgoff);
	153	if (ptr == MAP_FAILED)
	154	return -1;
	155
	156	while (size) {
	157	ssize_t ret = pwrite(ofd, ptr + off_in, size, off_out);
	158	if (ret < 0 && errno == EINTR)
	159	continue;
	160	if (ret <= 0)
	161	break;
	162
	163	size -= ret;
	164	off_in += ret;
	165	off_out -= ret;
	166	}
	167	munmap(ptr, off_in + size);
	168
	169	return size ? -1 : 0;
	170	}
	171
9a17d726	172	int copyfile_mode(const char from, const char to, mode_t mode)
4cf40131 ACM	173	{
	174	int fromfd, tofd;
	175	struct stat st;
4cf40131	176	int err = -1;
d7c72606	177	char tmp = NULL, ptr = NULL;
4cf40131 ACM	178
	179	if (stat(from, &st))
	180	goto out;
	181
d7c72606 MV	182	/* extra 'x' at the end is to reserve space for '.' */
	183	if (asprintf(&tmp, "%s.XXXXXXx", to) < 0) {
	184	tmp = NULL;
4cf40131	185	goto out;
d7c72606 MV	186	}
	187	ptr = strrchr(tmp, '/');
	188	if (!ptr)
	189	goto out;
	190	ptr = memmove(ptr + 1, ptr, strlen(ptr) - 1);
	191	*ptr = '.';
4cf40131	192
d7c72606	193	tofd = mkstemp(tmp);
4cf40131	194	if (tofd < 0)
d7c72606 MV	195	goto out;
	196
	197	if (fchmod(tofd, mode))
	198	goto out_close_to;
	199
	200	if (st.st_size == 0) { /* /proc? do it slowly... */
	201	err = slow_copyfile(from, tmp);
	202	goto out_close_to;
	203	}
	204
	205	fromfd = open(from, O_RDONLY);
	206	if (fromfd < 0)
	207	goto out_close_to;
4cf40131	208
9c9f5a2f	209	err = copyfile_offset(fromfd, 0, tofd, 0, st.st_size);
4cf40131	210
4cf40131	211	close(fromfd);
d7c72606 MV	212	out_close_to:
	213	close(tofd);
	214	if (!err)
	215	err = link(tmp, to);
	216	unlink(tmp);
4cf40131	217	out:
d7c72606	218	free(tmp);
4cf40131 ACM	219	return err;
4cf40131 ACM	220	}
c82ee828	221
9a17d726 AH	222	int copyfile(const char from, const char to)
	223	{
	224	return copyfile_mode(from, to, 0755);
	225	}
	226
c82ee828 ACM	227	unsigned long convert_unit(unsigned long value, char *unit)
	228	{
	229	*unit = ' ';
	230
	231	if (value > 1000) {
	232	value /= 1000;
	233	*unit = 'K';
	234	}
	235
	236	if (value > 1000) {
	237	value /= 1000;
	238	*unit = 'M';
	239	}
	240
	241	if (value > 1000) {
	242	value /= 1000;
	243	*unit = 'G';
	244	}
	245
	246	return value;
	247	}
1e7972cc	248
bc3a502b	249	static ssize_t ion(bool is_read, int fd, void *buf, size_t n)
1e7972cc ACM	250	{
1e7972cc ACM	251	void *buf_start = buf;
838d1452	252	size_t left = n;
1e7972cc	253
838d1452	254	while (left) {
bc3a502b JO	255	ssize_t ret = is_read ? read(fd, buf, left) :
bc3a502b JO	256	write(fd, buf, left);
1e7972cc	257
e148c760 NK	258	if (ret < 0 && errno == EINTR)
e148c760 NK	259	continue;
1e7972cc ACM	260	if (ret <= 0)
	261	return ret;
	262
838d1452 JO	263	left -= ret;
838d1452 JO	264	buf += ret;
1e7972cc ACM	265	}
1e7972cc ACM	266
838d1452 JO	267	BUG_ON((size_t)(buf - buf_start) != n);
838d1452 JO	268	return n;
1e7972cc	269	}
61e04b33	270
bc3a502b JO	271	/*
	272	* Read exactly 'n' bytes or return an error.
	273	*/
	274	ssize_t readn(int fd, void *buf, size_t n)
	275	{
	276	return ion(true, fd, buf, n);
	277	}
	278
	279	/*
	280	* Write exactly 'n' bytes or return an error.
	281	*/
	282	ssize_t writen(int fd, void *buf, size_t n)
	283	{
	284	return ion(false, fd, buf, n);
	285	}
	286
61e04b33 ACM	287	size_t hex_width(u64 v)
	288	{
	289	size_t n = 1;
	290
	291	while ((v >>= 4))
	292	++n;
	293
	294	return n;
	295	}
dc4552bf	296
b2aff5f6 JO	297	static int hex(char ch)
	298	{
	299	if ((ch >= '0') && (ch <= '9'))
	300	return ch - '0';
	301	if ((ch >= 'a') && (ch <= 'f'))
	302	return ch - 'a' + 10;
	303	if ((ch >= 'A') && (ch <= 'F'))
	304	return ch - 'A' + 10;
	305	return -1;
	306	}
	307
	308	/*
	309	* While we find nice hex chars, build a long_val.
	310	* Return number of chars processed.
	311	*/
	312	int hex2u64(const char ptr, u64 long_val)
	313	{
	314	const char *p = ptr;
	315	*long_val = 0;
	316
	317	while (*p) {
	318	const int hex_val = hex(*p);
	319
	320	if (hex_val < 0)
	321	break;
	322
	323	long_val = (long_val << 4) \| hex_val;
	324	p++;
	325	}
	326
	327	return p - ptr;
	328	}
	329
dc4552bf	330	/* Obtain a backtrace and print it to stdout. */
89fe808a	331	#ifdef HAVE_BACKTRACE_SUPPORT
dc4552bf ACM	332	void dump_stack(void)
	333	{
	334	void *array[16];
	335	size_t size = backtrace(array, ARRAY_SIZE(array));
	336	char **strings = backtrace_symbols(array, size);
	337	size_t i;
	338
	339	printf("Obtained %zd stack frames.\n", size);
	340
	341	for (i = 0; i < size; i++)
	342	printf("%s\n", strings[i]);
	343
	344	free(strings);
	345	}
c9f08bee IT	346	#else
	347	void dump_stack(void) {}
	348	#endif
2c803e52	349
07c1a0da ACM	350	void sighandler_dump_stack(int sig)
	351	{
	352	psignal(sig, "perf");
	353	dump_stack();
	354	exit(sig);
	355	}
	356
2c803e52 DA	357	void get_term_dimensions(struct winsize *ws)
	358	{
	359	char *s = getenv("LINES");
	360
	361	if (s != NULL) {
	362	ws->ws_row = atoi(s);
	363	s = getenv("COLUMNS");
	364	if (s != NULL) {
	365	ws->ws_col = atoi(s);
	366	if (ws->ws_row && ws->ws_col)
	367	return;
	368	}
	369	}
	370	#ifdef TIOCGWINSZ
	371	if (ioctl(1, TIOCGWINSZ, ws) == 0 &&
	372	ws->ws_row && ws->ws_col)
	373	return;
	374	#endif
	375	ws->ws_row = 25;
	376	ws->ws_col = 80;
	377	}
1355915a	378
9398c484 JO	379	void set_term_quiet_input(struct termios *old)
	380	{
	381	struct termios tc;
	382
	383	tcgetattr(0, old);
	384	tc = *old;
	385	tc.c_lflag &= ~(ICANON \| ECHO);
	386	tc.c_cc[VMIN] = 0;
	387	tc.c_cc[VTIME] = 0;
	388	tcsetattr(0, TCSANOW, &tc);
	389	}
	390
3b47abe1 NK	391	int parse_nsec_time(const char str, u64 ptime)
	392	{
	393	u64 time_sec, time_nsec;
	394	char *end;
	395
	396	time_sec = strtoul(str, &end, 10);
	397	if (end != '.' && end != '\0')
	398	return -1;
	399
	400	if (*end == '.') {
	401	int i;
	402	char nsec_buf[10];
	403
	404	if (strlen(++end) > 9)
	405	return -1;
	406
	407	strncpy(nsec_buf, end, 9);
	408	nsec_buf[9] = '\0';
	409
	410	/* make it nsec precision */
	411	for (i = strlen(nsec_buf); i < 9; i++)
	412	nsec_buf[i] = '0';
	413
	414	time_nsec = strtoul(nsec_buf, &end, 10);
	415	if (*end != '\0')
	416	return -1;
	417	} else
	418	time_nsec = 0;
	419
	420	ptime = time_sec NSEC_PER_SEC + time_nsec;
	421	return 0;
	422	}
27050f53 JO	423
	424	unsigned long parse_tag_value(const char str, struct parse_tag tags)
	425	{
	426	struct parse_tag *i = tags;
	427
	428	while (i->tag) {
	429	char *s;
	430
	431	s = strchr(str, i->tag);
	432	if (s) {
	433	unsigned long int value;
	434	char *endptr;
	435
	436	value = strtoul(str, &endptr, 10);
	437	if (s != endptr)
	438	break;
	439
56921bec AH	440	if (value > ULONG_MAX / i->mult)
56921bec AH	441	break;
27050f53 JO	442	value *= i->mult;
	443	return value;
	444	}
	445	i++;
	446	}
	447
	448	return (unsigned long) -1;
	449	}
97a07f10	450
076a30c4 KL	451	int get_stack_size(const char str, unsigned long _size)
	452	{
	453	char *endptr;
	454	unsigned long size;
	455	unsigned long max_size = round_down(USHRT_MAX, sizeof(u64));
	456
	457	size = strtoul(str, &endptr, 0);
	458
	459	do {
	460	if (*endptr)
	461	break;
	462
	463	size = round_up(size, sizeof(u64));
	464	if (!size \|\| size > max_size)
	465	break;
	466
	467	*_size = size;
	468	return 0;
	469
	470	} while (0);
	471
	472	pr_err("callchain: Incorrect stack dump size (max %ld): %s\n",
	473	max_size, str);
	474	return -1;
	475	}
	476
	477	int parse_callchain_record(const char arg, struct callchain_param param)
	478	{
	479	char tok, name, *saveptr = NULL;
	480	char *buf;
	481	int ret = -1;
	482
	483	/* We need buffer that we know we can write to. */
	484	buf = malloc(strlen(arg) + 1);
	485	if (!buf)
	486	return -ENOMEM;
	487
	488	strcpy(buf, arg);
	489
	490	tok = strtok_r((char *)buf, ",", &saveptr);
	491	name = tok ? : (char *)buf;
	492
	493	do {
	494	/* Framepointer style */
	495	if (!strncmp(name, "fp", sizeof("fp"))) {
	496	if (!strtok_r(NULL, ",", &saveptr)) {
	497	param->record_mode = CALLCHAIN_FP;
	498	ret = 0;
	499	} else
	500	pr_err("callchain: No more arguments "
	501	"needed for --call-graph fp\n");
	502	break;
	503
	504	#ifdef HAVE_DWARF_UNWIND_SUPPORT
	505	/* Dwarf style */
	506	} else if (!strncmp(name, "dwarf", sizeof("dwarf"))) {
	507	const unsigned long default_stack_dump_size = 8192;
	508
	509	ret = 0;
	510	param->record_mode = CALLCHAIN_DWARF;
	511	param->dump_size = default_stack_dump_size;
	512
	513	tok = strtok_r(NULL, ",", &saveptr);
	514	if (tok) {
515	unsigned long size = 0;
516
517	ret = get_stack_size(tok, &size);
518	param->dump_size = size;
519	}
520	#endif /* HAVE_DWARF_UNWIND_SUPPORT */
521	} else if (!strncmp(name, "lbr", sizeof("lbr"))) {
522	if (!strtok_r(NULL, ",", &saveptr)) {
523	param->record_mode = CALLCHAIN_LBR;
524	ret = 0;
525	} else
526	pr_err("callchain: No more arguments "
527	"needed for --call-graph lbr\n");
528	break;
529	} else {
530	pr_err("callchain: Unknown --call-graph option "
531	"value: %s\n", arg);
532	break;
533	}
534
535	} while (0);
536
537	free(buf);
538	return ret;
539	}
540
cef82c9f JO	541	int filename__read_str(const char filename, char buf, size_t sizep)
	542	{
	543	size_t size = 0, alloc_size = 0;
	544	void bf = NULL, nbf;
	545	int fd, n, err = 0;
6e81c74c	546	char sbuf[STRERR_BUFSIZE];
cef82c9f JO	547
	548	fd = open(filename, O_RDONLY);
	549	if (fd < 0)
	550	return -errno;
	551
	552	do {
	553	if (size == alloc_size) {
	554	alloc_size += BUFSIZ;
	555	nbf = realloc(bf, alloc_size);
	556	if (!nbf) {
	557	err = -ENOMEM;
	558	break;
	559	}
	560
	561	bf = nbf;
	562	}
	563
	564	n = read(fd, bf + size, alloc_size - size);
	565	if (n < 0) {
	566	if (size) {
6e81c74c MH	567	pr_warning("read failed %d: %s\n", errno,
6e81c74c MH	568	strerror_r(errno, sbuf, sizeof(sbuf)));
cef82c9f JO	569	err = 0;
	570	} else
	571	err = -errno;
	572
	573	break;
	574	}
	575
	576	size += n;
	577	} while (n > 0);
	578
	579	if (!err) {
	580	*sizep = size;
	581	*buf = bf;
	582	} else
	583	free(bf);
	584
	585	close(fd);
	586	return err;
	587	}
e1a2b174 DY	588
	589	const char *get_filename_for_perf_kvm(void)
	590	{
	591	const char *filename;
	592
	593	if (perf_host && !perf_guest)
	594	filename = strdup("perf.data.host");
	595	else if (!perf_host && perf_guest)
	596	filename = strdup("perf.data.guest");
	597	else
	598	filename = strdup("perf.data.kvm");
	599
	600	return filename;
	601	}
1a47245d AH	602
	603	int perf_event_paranoid(void)
	604	{
1a47245d AH	605	int value;
1a47245d AH	606
ce27309f	607	if (sysctl__read_int("kernel/perf_event_paranoid", &value))
1a47245d AH	608	return INT_MAX;
	609
	610	return value;
	611	}
71db07b1 AH	612
	613	void mem_bswap_32(void *src, int byte_size)
	614	{
	615	u32 *m = src;
	616	while (byte_size > 0) {
	617	m = bswap_32(m);
	618	byte_size -= sizeof(u32);
	619	++m;
	620	}
	621	}
	622
	623	void mem_bswap_64(void *src, int byte_size)
	624	{
	625	u64 *m = src;
	626
	627	while (byte_size > 0) {
	628	m = bswap_64(m);
	629	byte_size -= sizeof(u64);
	630	++m;
	631	}
	632	}
63914aca JO	633
	634	bool find_process(const char *name)
	635	{
	636	size_t len = strlen(name);
	637	DIR *dir;
	638	struct dirent *d;
	639	int ret = -1;
	640
	641	dir = opendir(procfs__mountpoint());
	642	if (!dir)
bf644563	643	return false;
63914aca JO	644
	645	/* Walk through the directory. */
	646	while (ret && (d = readdir(dir)) != NULL) {
	647	char path[PATH_MAX];
	648	char *data;
	649	size_t size;
	650
	651	if ((d->d_type != DT_DIR) \|\|
	652	!strcmp(".", d->d_name) \|\|
	653	!strcmp("..", d->d_name))
	654	continue;
	655
	656	scnprintf(path, sizeof(path), "%s/%s/comm",
	657	procfs__mountpoint(), d->d_name);
	658
	659	if (filename__read_str(path, &data, &size))
	660	continue;
	661
	662	ret = strncmp(name, data, len);
	663	free(data);
	664	}
	665
	666	closedir(dir);
	667	return ret ? false : true;
	668	}
07bc5c69 WN	669
	670	int
	671	fetch_kernel_version(unsigned int puint, char str,
	672	size_t str_size)
	673	{
	674	struct utsname utsname;
	675	int version, patchlevel, sublevel, err;
	676
	677	if (uname(&utsname))
	678	return -1;
	679
	680	if (str && str_size) {
	681	strncpy(str, utsname.release, str_size);
	682	str[str_size - 1] = '\0';
	683	}
	684
	685	err = sscanf(utsname.release, "%d.%d.%d",
	686	&version, &patchlevel, &sublevel);
	687
	688	if (err != 3) {
	689	pr_debug("Unablt to get kernel version from uname '%s'\n",
	690	utsname.release);
	691	return -1;
	692	}
	693
	694	if (puint)
	695	*puint = (version << 16) + (patchlevel << 8) + sublevel;
	696	return 0;
	697	}