/*

	Copyright(C) 2016, Red Hat, Inc., Stanislav Kozina

	This program is free software: you can redistribute it and/or modify

	it under the terms of the GNU General Public License as published by

	the Free Software Foundation, either version 3 of the License, or

	(at your option) any later version.

	This program is distributed in the hope that it will be useful,

	but WITHOUT ANY WARRANTY; without even the implied warranty of

	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License

	along with this program.  If not, see <http://www.gnu.org/licenses/>.

*/

#ifndef KSYMTAB_H_

#define	KSYMTAB_H_

#include <stdint.h>

struct elf_data {

	Elf *elf;

	GElf_Ehdr *ehdr;

	size_t shstrndx;

	const char *strtab;

	size_t strtab_size;

	int fd;

};

struct ksymtab;

struct ksym {

	uint64_t value;

	bool mark;

	char *link;

	struct ksymtab *ksymtab;

	char key[];

};

static inline bool ksymtab_ksym_is_marked(struct ksym *ksym)

{

	return ksym->mark;

}

static inline const char *ksymtab_ksym_get_name(struct ksym *ksym)

{

	return ksym->key;

}

static inline uint64_t ksymtab_ksym_get_value(struct ksym *ksym)

{

	return ksym->value;

}

static inline char *ksymtab_ksym_get_link(struct ksym *ksym)

{

	return ksym->link;

}

static inline void ksymtab_ksym_set_link(struct ksym *ksym, const char *link)

{

	if (ksym->link)

		free(ksym->link);

	ksym->link = safe_strdup_or_null(link);

}

extern void ksymtab_free(struct ksymtab *);

extern struct elf_data *elf_open(const char *);

extern int elf_get_exported(struct elf_data *, struct ksymtab **,

			    struct ksymtab **);

extern void elf_close(struct elf_data *);

extern int elf_get_endianness(struct elf_data *, unsigned int *);

extern struct ksym *ksymtab_find(struct ksymtab *, const char *);

extern size_t ksymtab_len(struct ksymtab *);

extern struct ksymtab *ksymtab_new(size_t);

extern struct ksym *ksymtab_add_sym(struct ksymtab *,

				    const char *, size_t, uint64_t);

extern struct ksym *ksymtab_copy_sym(struct ksymtab *, struct ksym *);

extern void ksymtab_for_each(struct ksymtab *,

			     void (*f)(struct ksym *, void *),

			     void *);

extern size_t ksymtab_mark_count(struct ksymtab *);

extern void ksymtab_ksym_mark(struct ksym *);

#endif /* KSYMTAB_H_ */