[deliverable/linux.git] / tools / perf / util / symbol-minimal.c

#include "symbol.h"
#include "util.h"

#include <stdio.h>
#include <fcntl.h>
#include <string.h>
#include <byteswap.h>
#include <sys/stat.h>


static bool check_need_swap(int file_endian)
{
	const int data = 1;
	u8 *check = (u8 *)&data;
	int host_endian;

	if (check[0] == 1)
		host_endian = ELFDATA2LSB;
	else
		host_endian = ELFDATA2MSB;

	return host_endian != file_endian;
}

#define NOTE_ALIGN(sz) (((sz) + 3) & ~3)

#define NT_GNU_BUILD_ID	3

static int read_build_id(void *note_data, size_t note_len, void *bf,
			 size_t size, bool need_swap)
{
	struct {
		u32 n_namesz;
		u32 n_descsz;
		u32 n_type;
	} *nhdr;
	void *ptr;

	ptr = note_data;
	while (ptr < (note_data + note_len)) {
		const char *name;
		size_t namesz, descsz;

		nhdr = ptr;
		if (need_swap) {
			nhdr->n_namesz = bswap_32(nhdr->n_namesz);
			nhdr->n_descsz = bswap_32(nhdr->n_descsz);
			nhdr->n_type = bswap_32(nhdr->n_type);
		}

		namesz = NOTE_ALIGN(nhdr->n_namesz);
		descsz = NOTE_ALIGN(nhdr->n_descsz);

		ptr += sizeof(*nhdr);
		name = ptr;
		ptr += namesz;
		if (nhdr->n_type == NT_GNU_BUILD_ID &&
		    nhdr->n_namesz == sizeof("GNU")) {
			if (memcmp(name, "GNU", sizeof("GNU")) == 0) {
				size_t sz = min(size, descsz);
				memcpy(bf, ptr, sz);
				memset(bf + sz, 0, size - sz);
				return 0;
			}
		}
		ptr += descsz;
	}

	return -1;
}

int filename__read_debuglink(const char *filename __maybe_unused,
			     char *debuglink __maybe_unused,
			     size_t size __maybe_unused)
{
	return -1;
}

/*
 * Just try PT_NOTE header otherwise fails
 */
int filename__read_build_id(const char *filename, void *bf, size_t size)
{
	FILE *fp;
	int ret = -1;
	bool need_swap = false;
	u8 e_ident[EI_NIDENT];
	size_t buf_size;
	void *buf;
	int i;

	fp = fopen(filename, "r");
	if (fp == NULL)
		return -1;

	if (fread(e_ident, sizeof(e_ident), 1, fp) != 1)
		goto out;

	if (memcmp(e_ident, ELFMAG, SELFMAG) ||
	    e_ident[EI_VERSION] != EV_CURRENT)
		goto out;

	need_swap = check_need_swap(e_ident[EI_DATA]);

	/* for simplicity */
	fseek(fp, 0, SEEK_SET);

	if (e_ident[EI_CLASS] == ELFCLASS32) {
		Elf32_Ehdr ehdr;
		Elf32_Phdr *phdr;

		if (fread(&ehdr, sizeof(ehdr), 1, fp) != 1)
			goto out;

		if (need_swap) {
			ehdr.e_phoff = bswap_32(ehdr.e_phoff);
			ehdr.e_phentsize = bswap_16(ehdr.e_phentsize);
			ehdr.e_phnum = bswap_16(ehdr.e_phnum);
		}

		buf_size = ehdr.e_phentsize * ehdr.e_phnum;
		buf = malloc(buf_size);
		if (buf == NULL)
			goto out;

		fseek(fp, ehdr.e_phoff, SEEK_SET);
		if (fread(buf, buf_size, 1, fp) != 1)
			goto out_free;

		for (i = 0, phdr = buf; i < ehdr.e_phnum; i++, phdr++) {
			void *tmp;

			if (need_swap) {
				phdr->p_type = bswap_32(phdr->p_type);
				phdr->p_offset = bswap_32(phdr->p_offset);
				phdr->p_filesz = bswap_32(phdr->p_filesz);
			}

			if (phdr->p_type != PT_NOTE)
				continue;

			buf_size = phdr->p_filesz;
			tmp = realloc(buf, buf_size);
			if (tmp == NULL)
				goto out_free;

			buf = tmp;
			fseek(fp, phdr->p_offset, SEEK_SET);
			if (fread(buf, buf_size, 1, fp) != 1)
				goto out_free;

			ret = read_build_id(buf, buf_size, bf, size, need_swap);
			if (ret == 0)
				ret = size;
			break;
		}
	} else {
		Elf64_Ehdr ehdr;
		Elf64_Phdr *phdr;

		if (fread(&ehdr, sizeof(ehdr), 1, fp) != 1)
			goto out;

		if (need_swap) {
			ehdr.e_phoff = bswap_64(ehdr.e_phoff);
			ehdr.e_phentsize = bswap_16(ehdr.e_phentsize);
			ehdr.e_phnum = bswap_16(ehdr.e_phnum);
		}

		buf_size = ehdr.e_phentsize * ehdr.e_phnum;
		buf = malloc(buf_size);
		if (buf == NULL)
			goto out;

		fseek(fp, ehdr.e_phoff, SEEK_SET);
		if (fread(buf, buf_size, 1, fp) != 1)
			goto out_free;

		for (i = 0, phdr = buf; i < ehdr.e_phnum; i++, phdr++) {
			void *tmp;

			if (need_swap) {
				phdr->p_type = bswap_32(phdr->p_type);
				phdr->p_offset = bswap_64(phdr->p_offset);
				phdr->p_filesz = bswap_64(phdr->p_filesz);
			}

			if (phdr->p_type != PT_NOTE)
				continue;

			buf_size = phdr->p_filesz;
			tmp = realloc(buf, buf_size);
			if (tmp == NULL)
				goto out_free;

			buf = tmp;
			fseek(fp, phdr->p_offset, SEEK_SET);
			if (fread(buf, buf_size, 1, fp) != 1)
				goto out_free;

			ret = read_build_id(buf, buf_size, bf, size, need_swap);
			if (ret == 0)
				ret = size;
			break;
		}
	}
out_free:
	free(buf);
out:
	fclose(fp);
	return ret;
}

int sysfs__read_build_id(const char *filename, void *build_id, size_t size)
{
	int fd;
	int ret = -1;
	struct stat stbuf;
	size_t buf_size;
	void *buf;

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

	if (fstat(fd, &stbuf) < 0)
		goto out;

	buf_size = stbuf.st_size;
	buf = malloc(buf_size);
	if (buf == NULL)
		goto out;

	if (read(fd, buf, buf_size) != (ssize_t) buf_size)
		goto out_free;

	ret = read_build_id(buf, buf_size, build_id, size, false);
out_free:
	free(buf);
out:
	close(fd);
	return ret;
}

int symsrc__init(struct symsrc *ss, struct dso *dso __maybe_unused,
		 const char *name,
	         enum dso_binary_type type)
{
	int fd = open(name, O_RDONLY);
	if (fd < 0)
		return -1;

	ss->name = strdup(name);
	if (!ss->name)
		goto out_close;

	ss->fd = fd;
	ss->type = type;

	return 0;
out_close:
	close(fd);
	return -1;
}

bool symsrc__possibly_runtime(struct symsrc *ss __maybe_unused)
{
	/* Assume all sym sources could be a runtime image. */
	return true;
}

bool symsrc__has_symtab(struct symsrc *ss __maybe_unused)
{
	return false;
}

void symsrc__destroy(struct symsrc *ss)
{
	zfree(&ss->name);
	close(ss->fd);
}

int dso__synthesize_plt_symbols(struct dso *dso __maybe_unused,
				struct symsrc *ss __maybe_unused,
				struct map *map __maybe_unused,
				symbol_filter_t filter __maybe_unused)
{
	return 0;
}

static int fd__is_64_bit(int fd)
{
	u8 e_ident[EI_NIDENT];

	if (lseek(fd, 0, SEEK_SET))
		return -1;

	if (readn(fd, e_ident, sizeof(e_ident)) != sizeof(e_ident))
		return -1;

	if (memcmp(e_ident, ELFMAG, SELFMAG) ||
	    e_ident[EI_VERSION] != EV_CURRENT)
		return -1;

	return e_ident[EI_CLASS] == ELFCLASS64;
}

enum dso_type dso__type_fd(int fd)
{
	Elf64_Ehdr ehdr;
	int ret;

	ret = fd__is_64_bit(fd);
	if (ret < 0)
		return DSO__TYPE_UNKNOWN;

	if (ret)
		return DSO__TYPE_64BIT;

	if (readn(fd, &ehdr, sizeof(ehdr)) != sizeof(ehdr))
		return DSO__TYPE_UNKNOWN;

	if (ehdr.e_machine == EM_X86_64)
		return DSO__TYPE_X32BIT;

	return DSO__TYPE_32BIT;
}

int dso__load_sym(struct dso *dso, struct map *map __maybe_unused,
		  struct symsrc *ss,
		  struct symsrc *runtime_ss __maybe_unused,
		  symbol_filter_t filter __maybe_unused,
		  int kmodule __maybe_unused)
{
	unsigned char *build_id[BUILD_ID_SIZE];
	int ret;

	ret = fd__is_64_bit(ss->fd);
	if (ret >= 0)
		dso->is_64_bit = ret;

	if (filename__read_build_id(ss->name, build_id, BUILD_ID_SIZE) > 0) {
		dso__set_build_id(dso, build_id);
		return 1;
	}
	return 0;
}

int file__read_maps(int fd __maybe_unused, bool exe __maybe_unused,
		    mapfn_t mapfn __maybe_unused, void *data __maybe_unused,
		    bool *is_64_bit __maybe_unused)
{
	return -1;
}

int kcore_extract__create(struct kcore_extract *kce __maybe_unused)
{
	return -1;
}

void kcore_extract__delete(struct kcore_extract *kce __maybe_unused)
{
}

int kcore_copy(const char *from_dir __maybe_unused,
	       const char *to_dir __maybe_unused)
{
	return -1;
}

void symbol__elf_init(void)
{
}
Commit	Line	Data
393be2e3	1	#include "symbol.h"
74cf249d	2	#include "util.h"
393be2e3	3
b691f643 NK	4	#include <stdio.h>
	5	#include <fcntl.h>
	6	#include <string.h>
	7	#include <byteswap.h>
	8	#include <sys/stat.h>
393be2e3	9
b691f643 NK	10
b691f643 NK	11	static bool check_need_swap(int file_endian)
393be2e3	12	{
b691f643 NK	13	const int data = 1;
	14	u8 check = (u8 )&data;
	15	int host_endian;
	16
	17	if (check[0] == 1)
	18	host_endian = ELFDATA2LSB;
	19	else
	20	host_endian = ELFDATA2MSB;
	21
	22	return host_endian != file_endian;
393be2e3 NK	23	}
393be2e3 NK	24
b691f643 NK	25	#define NOTE_ALIGN(sz) (((sz) + 3) & ~3)
	26
	27	#define NT_GNU_BUILD_ID 3
	28
	29	static int read_build_id(void note_data, size_t note_len, void bf,
	30	size_t size, bool need_swap)
393be2e3	31	{
b691f643 NK	32	struct {
	33	u32 n_namesz;
	34	u32 n_descsz;
	35	u32 n_type;
	36	} *nhdr;
	37	void *ptr;
	38
	39	ptr = note_data;
	40	while (ptr < (note_data + note_len)) {
	41	const char *name;
	42	size_t namesz, descsz;
	43
	44	nhdr = ptr;
	45	if (need_swap) {
	46	nhdr->n_namesz = bswap_32(nhdr->n_namesz);
	47	nhdr->n_descsz = bswap_32(nhdr->n_descsz);
	48	nhdr->n_type = bswap_32(nhdr->n_type);
	49	}
	50
	51	namesz = NOTE_ALIGN(nhdr->n_namesz);
	52	descsz = NOTE_ALIGN(nhdr->n_descsz);
	53
	54	ptr += sizeof(*nhdr);
	55	name = ptr;
	56	ptr += namesz;
	57	if (nhdr->n_type == NT_GNU_BUILD_ID &&
	58	nhdr->n_namesz == sizeof("GNU")) {
	59	if (memcmp(name, "GNU", sizeof("GNU")) == 0) {
	60	size_t sz = min(size, descsz);
	61	memcpy(bf, ptr, sz);
	62	memset(bf + sz, 0, size - sz);
	63	return 0;
	64	}
	65	}
	66	ptr += descsz;
	67	}
	68
393be2e3 NK	69	return -1;
	70	}
	71
1d037ca1 IT	72	int filename__read_debuglink(const char *filename __maybe_unused,
	73	char *debuglink __maybe_unused,
	74	size_t size __maybe_unused)
393be2e3 NK	75	{
	76	return -1;
	77	}
	78
b691f643 NK	79	/*
	80	* Just try PT_NOTE header otherwise fails
	81	*/
	82	int filename__read_build_id(const char filename, void bf, size_t size)
	83	{
	84	FILE *fp;
	85	int ret = -1;
	86	bool need_swap = false;
	87	u8 e_ident[EI_NIDENT];
	88	size_t buf_size;
	89	void *buf;
	90	int i;
	91
	92	fp = fopen(filename, "r");
	93	if (fp == NULL)
	94	return -1;
	95
	96	if (fread(e_ident, sizeof(e_ident), 1, fp) != 1)
	97	goto out;
	98
	99	if (memcmp(e_ident, ELFMAG, SELFMAG) \|\|
	100	e_ident[EI_VERSION] != EV_CURRENT)
	101	goto out;
	102
	103	need_swap = check_need_swap(e_ident[EI_DATA]);
	104
	105	/* for simplicity */
	106	fseek(fp, 0, SEEK_SET);
	107
	108	if (e_ident[EI_CLASS] == ELFCLASS32) {
	109	Elf32_Ehdr ehdr;
	110	Elf32_Phdr *phdr;
	111
	112	if (fread(&ehdr, sizeof(ehdr), 1, fp) != 1)
	113	goto out;
	114
	115	if (need_swap) {
	116	ehdr.e_phoff = bswap_32(ehdr.e_phoff);
	117	ehdr.e_phentsize = bswap_16(ehdr.e_phentsize);
	118	ehdr.e_phnum = bswap_16(ehdr.e_phnum);
	119	}
	120
	121	buf_size = ehdr.e_phentsize * ehdr.e_phnum;
	122	buf = malloc(buf_size);
	123	if (buf == NULL)
	124	goto out;
	125
	126	fseek(fp, ehdr.e_phoff, SEEK_SET);
	127	if (fread(buf, buf_size, 1, fp) != 1)
	128	goto out_free;
	129
	130	for (i = 0, phdr = buf; i < ehdr.e_phnum; i++, phdr++) {
	131	void *tmp;
	132
	133	if (need_swap) {
	134	phdr->p_type = bswap_32(phdr->p_type);
	135	phdr->p_offset = bswap_32(phdr->p_offset);
	136	phdr->p_filesz = bswap_32(phdr->p_filesz);
	137	}
	138
	139	if (phdr->p_type != PT_NOTE)
	140	continue;
	141
	142	buf_size = phdr->p_filesz;
143	tmp = realloc(buf, buf_size);
144	if (tmp == NULL)
145	goto out_free;
146
147	buf = tmp;
148	fseek(fp, phdr->p_offset, SEEK_SET);
149	if (fread(buf, buf_size, 1, fp) != 1)
150	goto out_free;
151
152	ret = read_build_id(buf, buf_size, bf, size, need_swap);
153	if (ret == 0)
154	ret = size;
155	break;
156	}
157	} else {
158	Elf64_Ehdr ehdr;
159	Elf64_Phdr *phdr;
160
161	if (fread(&ehdr, sizeof(ehdr), 1, fp) != 1)
162	goto out;
163
164	if (need_swap) {
165	ehdr.e_phoff = bswap_64(ehdr.e_phoff);
166	ehdr.e_phentsize = bswap_16(ehdr.e_phentsize);
167	ehdr.e_phnum = bswap_16(ehdr.e_phnum);
168	}
169
170	buf_size = ehdr.e_phentsize * ehdr.e_phnum;
171	buf = malloc(buf_size);
172	if (buf == NULL)
173	goto out;
174
175	fseek(fp, ehdr.e_phoff, SEEK_SET);
176	if (fread(buf, buf_size, 1, fp) != 1)
177	goto out_free;
178
179	for (i = 0, phdr = buf; i < ehdr.e_phnum; i++, phdr++) {
180	void *tmp;
181
182	if (need_swap) {
183	phdr->p_type = bswap_32(phdr->p_type);
184	phdr->p_offset = bswap_64(phdr->p_offset);
185	phdr->p_filesz = bswap_64(phdr->p_filesz);
186	}
187
188	if (phdr->p_type != PT_NOTE)
189	continue;
190
191	buf_size = phdr->p_filesz;
192	tmp = realloc(buf, buf_size);
193	if (tmp == NULL)
194	goto out_free;
195
196	buf = tmp;
197	fseek(fp, phdr->p_offset, SEEK_SET);
198	if (fread(buf, buf_size, 1, fp) != 1)
199	goto out_free;
200
201	ret = read_build_id(buf, buf_size, bf, size, need_swap);
202	if (ret == 0)
203	ret = size;
204	break;
205	}
206	}
207	out_free:
208	free(buf);
209	out:
210	fclose(fp);
211	return ret;
212	}
213
214	int sysfs__read_build_id(const char filename, void build_id, size_t size)
215	{
216	int fd;
217	int ret = -1;
218	struct stat stbuf;
219	size_t buf_size;
220	void *buf;
221
222	fd = open(filename, O_RDONLY);
223	if (fd < 0)
224	return -1;
225
226	if (fstat(fd, &stbuf) < 0)
227	goto out;
228
229	buf_size = stbuf.st_size;
230	buf = malloc(buf_size);
231	if (buf == NULL)
232	goto out;
233
234	if (read(fd, buf, buf_size) != (ssize_t) buf_size)
235	goto out_free;
236
237	ret = read_build_id(buf, buf_size, build_id, size, false);
238	out_free:
239	free(buf);
240	out:
241	close(fd);
242	return ret;
243	}
244
1d037ca1 IT	245	int symsrc__init(struct symsrc ss, struct dso dso __maybe_unused,
1d037ca1 IT	246	const char *name,
b68e2f91 CS	247	enum dso_binary_type type)
	248	{
	249	int fd = open(name, O_RDONLY);
	250	if (fd < 0)
	251	return -1;
	252
	253	ss->name = strdup(name);
	254	if (!ss->name)
	255	goto out_close;
	256
779e24e2	257	ss->fd = fd;
b68e2f91 CS	258	ss->type = type;
	259
	260	return 0;
	261	out_close:
	262	close(fd);
	263	return -1;
	264	}
	265
1d037ca1	266	bool symsrc__possibly_runtime(struct symsrc *ss __maybe_unused)
3aafe5ae CS	267	{
	268	/* Assume all sym sources could be a runtime image. */
	269	return true;
	270	}
	271
1d037ca1	272	bool symsrc__has_symtab(struct symsrc *ss __maybe_unused)
d26cd12b CS	273	{
	274	return false;
	275	}
	276
b68e2f91 CS	277	void symsrc__destroy(struct symsrc *ss)
b68e2f91 CS	278	{
74cf249d	279	zfree(&ss->name);
b68e2f91 CS	280	close(ss->fd);
	281	}
	282
1d037ca1 IT	283	int dso__synthesize_plt_symbols(struct dso *dso __maybe_unused,
	284	struct symsrc *ss __maybe_unused,
	285	struct map *map __maybe_unused,
	286	symbol_filter_t filter __maybe_unused)
393be2e3 NK	287	{
	288	return 0;
	289	}
	290
c6d8f2a4 AH	291	static int fd__is_64_bit(int fd)
	292	{
	293	u8 e_ident[EI_NIDENT];
	294
	295	if (lseek(fd, 0, SEEK_SET))
	296	return -1;
	297
	298	if (readn(fd, e_ident, sizeof(e_ident)) != sizeof(e_ident))
	299	return -1;
	300
	301	if (memcmp(e_ident, ELFMAG, SELFMAG) \|\|
	302	e_ident[EI_VERSION] != EV_CURRENT)
	303	return -1;
	304
	305	return e_ident[EI_CLASS] == ELFCLASS64;
	306	}
	307
2b5b8bb2 AH	308	enum dso_type dso__type_fd(int fd)
	309	{
	310	Elf64_Ehdr ehdr;
	311	int ret;
	312
	313	ret = fd__is_64_bit(fd);
	314	if (ret < 0)
	315	return DSO__TYPE_UNKNOWN;
	316
	317	if (ret)
	318	return DSO__TYPE_64BIT;
	319
	320	if (readn(fd, &ehdr, sizeof(ehdr)) != sizeof(ehdr))
	321	return DSO__TYPE_UNKNOWN;
	322
	323	if (ehdr.e_machine == EM_X86_64)
	324	return DSO__TYPE_X32BIT;
	325
	326	return DSO__TYPE_32BIT;
	327	}
	328
1d037ca1 IT	329	int dso__load_sym(struct dso dso, struct map map __maybe_unused,
	330	struct symsrc *ss,
	331	struct symsrc *runtime_ss __maybe_unused,
	332	symbol_filter_t filter __maybe_unused,
	333	int kmodule __maybe_unused)
393be2e3	334	{
b691f643	335	unsigned char *build_id[BUILD_ID_SIZE];
c6d8f2a4 AH	336	int ret;
	337
	338	ret = fd__is_64_bit(ss->fd);
	339	if (ret >= 0)
	340	dso->is_64_bit = ret;
b691f643	341
b68e2f91	342	if (filename__read_build_id(ss->name, build_id, BUILD_ID_SIZE) > 0) {
b691f643 NK	343	dso__set_build_id(dso, build_id);
	344	return 1;
	345	}
393be2e3 NK	346	return 0;
	347	}
	348
8e0cf965 AH	349	int file__read_maps(int fd __maybe_unused, bool exe __maybe_unused,
	350	mapfn_t mapfn __maybe_unused, void *data __maybe_unused,
	351	bool *is_64_bit __maybe_unused)
	352	{
	353	return -1;
	354	}
	355
afba19d9 AH	356	int kcore_extract__create(struct kcore_extract *kce __maybe_unused)
	357	{
	358	return -1;
	359	}
	360
	361	void kcore_extract__delete(struct kcore_extract *kce __maybe_unused)
	362	{
	363	}
	364
fc1b691d AH	365	int kcore_copy(const char *from_dir __maybe_unused,
	366	const char *to_dir __maybe_unused)
	367	{
	368	return -1;
	369	}
	370
393be2e3 NK	371	void symbol__elf_init(void)
	372	{
	373	}