/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <libelf.h>
#include <gelf.h>
#include <errno.h>

/*
 * symdef is a very simplified version of nm.  It is used by upgrade and
 * create_ramdisk in situations where we can't guarantee that nm will be around.
 *
 * Two arguments are expected: a binary and a symbol name.  If the symbol is
 * found in the name table of the binary, 0 is returned.  If it is not found,
 * 1 is returned.  If an error occurs, a message is printed to stderr and -1
 * is returned.
 */


static void
usage(void)
{
	(void) fprintf(stderr, "USAGE: symdef file_name symbol\n");
}

int
main(int argc, char *argv[])
{
	int	fd = 0;
	int	rv = 1;
	uint_t	cnt, symcnt;
	Elf	*elfp = NULL;
	Elf_Scn	*scn = NULL;
	size_t	shstrndx;
	GElf_Ehdr	ehdr;
	GElf_Shdr	shdr;
	GElf_Sym	sym;
	Elf32_Word	shndx;
	Elf_Data	*symdata, *shndxdata;

	if (argc != 3) {
		usage();
		return (-1);
	}

	fd = open(argv[1], O_RDONLY);
	if (fd == -1) {
		(void) fprintf(stderr, "%s\n", strerror(errno));
		rv = -1;
		goto done;
	}
	if (elf_version(EV_CURRENT) == EV_NONE) {
		(void) fprintf(stderr, "Elf library version out of date\n");
		rv = -1;
		goto done;
	}
	elfp = elf_begin(fd, ELF_C_READ, NULL);
	if ((elfp == NULL) || (elf_kind(elfp) != ELF_K_ELF) ||
	    ((gelf_getehdr(elfp, &ehdr)) == NULL) ||
	    (elf_getshstrndx(elfp, &shstrndx) == 0))
		goto done;

	while ((scn = elf_nextscn(elfp, scn)) != NULL) {
		if ((gelf_getshdr(scn, &shdr) == NULL) ||
		    ((shdr.sh_type != SHT_SYMTAB) &&
		    (shdr.sh_type != SHT_DYNSYM)) ||
		    ((symdata = elf_getdata(scn, NULL)) == NULL))
			continue;
		symcnt = shdr.sh_size / shdr.sh_entsize;
		shndxdata = NULL;
		for (cnt = 0; cnt < symcnt; cnt++) {
			if ((gelf_getsymshndx(symdata, shndxdata, cnt,
			    &sym, &shndx) != NULL) &&
			    (strcmp(argv[2], elf_strptr(elfp, shdr.sh_link,
			    sym.st_name)) == 0)) {
				rv = 0;
				goto done;
			}
		}
	}
done:
	if (elfp)
		(void) elf_end(elfp);
	if (fd != -1)
		(void) close(fd);
	return (rv);
}