/*-
 * Modified 2002 Team LinBSD
 * Copyright (c) 2000, 2001 David O'Brien
 * Copyright (c) 1996 Søren Schmidt
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer
 *    in this position and unchanged.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. The name of the author may not be used to endorse or promote products
 *    derived from this software withough specific prior written permission
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/cdefs.h>
#include <sys/types.h>
#include <sys/errno.h>
#ifdef __linux__
#include <elf.h>
#else
#include <sys/elf_common.h>
#endif
#include <err.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>


#ifndef ELFOSABI_NONE
#define ELFOSABI_NONE   0       /* UNIX System V ABI */
#define ELFOSABI_HPUX   1       /* HP-UX operating system */
#define ELFOSABI_NETBSD 2       /* NetBSD */
#define ELFOSABI_LINUX  3       /* GNU/Linux */
#define ELFOSABI_HURD   4       /* GNU/Hurd */
#define ELFOSABI_SOLARIS 6      /* Solaris */
#define ELFOSABI_AIX    7       /* AIX */
#define ELFOSABI_IRIX   8       /* IRIX */
#define ELFOSABI_FREEBSD 9      /* FreeBSD */
#define ELFOSABI_TRU64  10      /* TRU64 UNIX */
#define ELFOSABI_MODESTO 11     /* Novell Modesto */
#define ELFOSABI_OPENBSD 12     /* OpenBSD */
#define ELFOSABI_STANDALONE 255 /* Standalone (embedded) application */
#define ELFOSABI_ARM   97       /* ARM */
#endif

static int elftype(const char *);
static const char *iselftype(int);
static void printelftypes(void);
static void usage(void);

struct ELFtypes {
        const char *str;
        int value;
};

/* XXX - any more types? */
static struct ELFtypes elftypes[] = {
	{"SVR4", ELFOSABI_NONE},
	{"HP-UX", ELFOSABI_HPUX},
	{"NetBSD", ELFOSABI_NETBSD},
	{"Linux", ELFOSABI_LINUX},
#ifdef ELFOSABI_HURD
	{"Hurd", ELFOSABI_HURD},
#endif
	{"Solaris", ELFOSABI_SOLARIS},
	{"AIX", ELFOSABI_AIX},
	{"IRIX", ELFOSABI_IRIX},
	{"FreeBSD", ELFOSABI_FREEBSD},
	{"TRU64", ELFOSABI_TRU64},
	{"Novell", ELFOSABI_MODESTO},
	{"OpenBSD", ELFOSABI_OPENBSD},
	{"Standalone", ELFOSABI_STANDALONE},
	{"ARM", ELFOSABI_ARM}
};

int
main(int argc, char **argv)
{

        const char *strtype = "FreeBSD";
        int type = ELFOSABI_FREEBSD;
        int retval = 0;
        int ch, change = 0, verbose = 0, force = 0, listed = 0;

        while ((ch = getopt(argc, argv, "f:lt:v")) != -1)
                switch (ch) {
                case 'f':
                        if (change)
                                errx(1, "f option incompatible with t option");
                        force = 1;
                        type = atoi(optarg);
                        if (errno == ERANGE || type < 0 || type > 255) {
                                warnx("invalid argument to option f: %s",
                                    optarg);
                                usage();
                        }
                        break;
                case 'l':
                        printelftypes();
                        listed = 1;
                        break;
                case 'v':
                        verbose = 1;
                        break;
                case 't':
                        if (force)
                                errx(1, "t option incompatible with f option");
                        change = 1;
                        strtype = optarg;
                        break;
                default:
                        usage();
        }
        argc -= optind;
        argv += optind;
        if (!argc) {
                if (listed)
                        exit(0);
                else {
                        warnx("no file(s) specified");
                        usage();
                }
        }

        if (!force && (type = elftype(strtype)) == -1) {
                warnx("invalid ELF type '%s'", strtype);
                printelftypes();
                usage();
        }

        while (argc) {
                int fd;
                char buffer[EI_NIDENT];

                if ((fd = open(argv[0], change || force ? O_RDWR : O_RDONLY, 0)) < 0) {
                        warn("error opening file %s", argv[0]);
                        retval = 1;
                        goto fail;
                }
                if (read(fd, buffer, EI_NIDENT) < EI_NIDENT) {
                        warnx("file '%s' too short", argv[0]);
                        retval = 1;
                        goto fail;
                }
                if (buffer[0] != ELFMAG0 || buffer[1] != ELFMAG1 ||
                    buffer[2] != ELFMAG2 || buffer[3] != ELFMAG3) {
                        warnx("file '%s' is not ELF format", argv[0]);
                        retval = 1;
                        goto fail;
                }
                if (!change && !force) {
                        fprintf(stdout,
                                "File '%s' is of brand '%s' (%u).\n",
                                argv[0], iselftype(buffer[EI_OSABI]),
                                buffer[EI_OSABI]);
                        if (!iselftype(type)) {
                                warnx("ELF ABI Brand '%u' is unknown",
                                      type);
                                printelftypes();
                        }
                }
                else {
                        buffer[EI_OSABI] = type;
                        lseek(fd, 0, SEEK_SET);
                        if (write(fd, buffer, EI_NIDENT) != EI_NIDENT) {
                                warn("error writing %s %d", argv[0], fd);
                                retval = 1;
                                goto fail;
                        }
                }
fail:
                close(fd);
                argc--;
                argv++;
        }

        return retval;
}

static void
usage(void)
{
	fprintf(stderr, "usage: brandelf [-f ELF ABI number] [-v] [-l] [-t string] file ...\n");
	exit(1);
}

static const char *
iselftype(int etype)
{
        size_t elfwalk;

        for (elfwalk = 0;
             elfwalk < sizeof(elftypes)/sizeof(elftypes[0]);
             elfwalk++)
                if (etype == elftypes[elfwalk].value)
                        return elftypes[elfwalk].str;
        return 0;
}

static int
elftype(const char *elfstrtype)
{
        size_t elfwalk;

        for (elfwalk = 0;
             elfwalk < sizeof(elftypes)/sizeof(elftypes[0]);
             elfwalk++)
                if (strcmp(elfstrtype, elftypes[elfwalk].str) == 0)
                        return elftypes[elfwalk].value;
        return -1;
}

static void
printelftypes(void)
{
        size_t elfwalk;

        fprintf(stderr, "Known ELF types are:\n");
        for (elfwalk = 0;
             elfwalk < sizeof(elftypes)/sizeof(elftypes[0]);
             elfwalk++)
                fprintf(stderr, "\t%s(%u) \n", elftypes[elfwalk].str, 
                        elftypes[elfwalk].value);
        fprintf(stderr, "\n");
}