/*
 * begin.c - implementation of the elf_begin(3) and elf_memory(3) functions.
 * Copyright (C) 1995 - 2004 Michael Riepe
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 * 
 * This library 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
 * Library General Public License for more details.
 * 
 * You should have received a copy of the GNU Library General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
 */

#include <private.h>

#ifndef lint
static const char rcsid[] = "@(#) $Id: begin.c,v 1.21 2008年05月23日 08:15:34 michael Exp $";
#endif /* lint */

#if HAVE_AR_H
#include <ar.h>
#else /* HAVE_AR_H */

#define ARMAG	"!<arch>\n"
#define SARMAG	8

struct ar_hdr {
 char ar_name[16];
 char ar_date[12];
 char ar_uid[6];
 char ar_gid[6];
 char ar_mode[8];
 char ar_size[10];
 char ar_fmag[2];
};

#define ARFMAG	"`\n"

#endif /* HAVE_AR_H */

static const Elf _elf_init = INIT_ELF;
static const char fmag[] = ARFMAG;

static unsigned long
getnum(const char *str, size_t len, int base, size_t *err) {
 unsigned long result = 0;

 while (len && *str == ' ') {
	str++; len--;
 }
 while (len && *str >= '0' && (*str - '0') < base) {
	result = base * result + *str++ - '0'; len--;
 }
 while (len && *str == ' ') {
	str++; len--;
 }
 if (len) {
	*err = len;
 }
 return result;
}

static void
_elf_init_ar(Elf *elf) {
 struct ar_hdr *hdr;
 size_t offset;
 size_t size;
 size_t err = 0;

 elf->e_kind = ELF_K_AR;
 elf->e_idlen = SARMAG;
 elf->e_off = SARMAG;

 /* process special members */
 offset = SARMAG;
 while (!elf->e_strtab && offset + sizeof(*hdr) <= elf->e_size) {
	hdr = (struct ar_hdr*)(elf->e_data + offset);
	if (memcmp(hdr->ar_fmag, fmag, sizeof(fmag) - 1)) {
	 break;
	}
	if (hdr->ar_name[0] != '/') {
	 break;
	}
	size = getnum(hdr->ar_size, sizeof(hdr->ar_size), 10, &err);
	if (err || !size) {
	 break;
	}
	offset += sizeof(*hdr);
	if (offset + size > elf->e_size) {
	 break;
	}
	if (hdr->ar_name[1] == '/' && hdr->ar_name[2] == ' ') {
	 elf->e_strtab = elf->e_data + offset;
	 elf->e_strlen = size;
	 break;
	}
	if (hdr->ar_name[1] != ' ') {
	 break;
	}
	/*
	 * Windows (.lib) archives provide two symbol tables
	 * The first one is the one we want.
	 */
	if (!elf->e_symtab) {
	 elf->e_symtab = elf->e_data + offset;
	 elf->e_symlen = size;
	}
	offset += size + (size & 1);
 }
}

static Elf_Arhdr*
_elf_arhdr(Elf *arf) {
 struct ar_hdr *hdr;
 Elf_Arhdr *arhdr;
 size_t namelen;
 size_t tmp;
 char *name;
 size_t err = 0;

 if (arf->e_off == arf->e_size) {
	/* no error! */
	return NULL;
 }
 if (arf->e_off < 0 || arf->e_off > arf->e_size) {
	seterr(ERROR_OUTSIDE);
	return NULL;
 }
 if (arf->e_off + sizeof(*hdr) > arf->e_size) {
	seterr(ERROR_TRUNC_ARHDR);
	return NULL;
 }
 elf_assert(arf->e_data != NULL);
 hdr = (struct ar_hdr*)(arf->e_data + arf->e_off);
 if (memcmp(hdr->ar_fmag, fmag, sizeof(fmag) - 1)) {
	seterr(ERROR_ARFMAG);
	return NULL;
 }

 name = hdr->ar_name;
 for (namelen = sizeof(hdr->ar_name); namelen > 0; namelen--) {
	if (name[namelen - 1] != ' ') {
	 break;
	}
 }
 if (name[0] == '/') {
	if (name[1] >= '0' && name[1] <= '9') {
	 if (!arf->e_strtab) {
		seterr(ERROR_ARSTRTAB);
		return NULL;
	 }
	 tmp = getnum(&name[1], namelen - 1, 10, &err);
	 if (err) {
		seterr(ERROR_ARSPECIAL);
		return NULL;
	 }
	 if (tmp < 0 || tmp >= arf->e_strlen) {
		seterr(ERROR_ARSTRTAB);
		return NULL;
	 }
	 for (namelen = tmp; namelen < arf->e_strlen; namelen++) {
		if (arf->e_strtab[namelen] == '/') {
		 break;
		}
	 }
	 if (namelen == arf->e_strlen) {
		seterr(ERROR_ARSTRTAB);
		return NULL;
	 }
	 name = arf->e_strtab + tmp;
	 namelen -= tmp;
	}
	else if (namelen != 1 && !(namelen == 2 && name[1] == '/')) {
	 seterr(ERROR_ARSPECIAL);
	 return NULL;
	}
 }
 else if (namelen > 0 && name[namelen - 1] == '/') {
	namelen--;
 }
 /* XXX some broken software omits the trailing slash
 else {
	namelen = 0;
 }
 */

 if (!(arhdr = (Elf_Arhdr*)malloc(sizeof(*arhdr) +
		 sizeof(hdr->ar_name) + namelen + 2))) {
	seterr(ERROR_MEM_ARHDR);
	return NULL;
 }

 arhdr->ar_name = NULL;
 arhdr->ar_rawname = (char*)(arhdr + 1);
 arhdr->ar_date = getnum(hdr->ar_date, sizeof(hdr->ar_date), 10, &err);
 arhdr->ar_uid = getnum(hdr->ar_uid, sizeof(hdr->ar_uid), 10, &err);
 arhdr->ar_gid = getnum(hdr->ar_gid, sizeof(hdr->ar_gid), 10, &err);
 arhdr->ar_mode = getnum(hdr->ar_mode, sizeof(hdr->ar_mode), 8, &err);
 arhdr->ar_size = getnum(hdr->ar_size, sizeof(hdr->ar_size), 10, &err);
 if (err) {
	free(arhdr);
	seterr(ERROR_ARHDR);
	return NULL;
 }
 if (arf->e_off + sizeof(struct ar_hdr) + arhdr->ar_size > arf->e_size) {
	free(arhdr);
	seterr(ERROR_TRUNC_MEMBER);
	return NULL;
 }

 memcpy(arhdr->ar_rawname, hdr->ar_name, sizeof(hdr->ar_name));
 arhdr->ar_rawname[sizeof(hdr->ar_name)] = '0円';

 if (namelen) {
	arhdr->ar_name = arhdr->ar_rawname + sizeof(hdr->ar_name) + 1;
	memcpy(arhdr->ar_name, name, namelen);
	arhdr->ar_name[namelen] = '0円';
 }

 return arhdr;
}

static void
_elf_check_type(Elf *elf, size_t size) {
 elf->e_idlen = size;
 if (size >= EI_NIDENT && !memcmp(elf->e_data, ELFMAG, SELFMAG)) {
	elf->e_kind = ELF_K_ELF;
	elf->e_idlen = EI_NIDENT;
	elf->e_class = elf->e_data[EI_CLASS];
	elf->e_encoding = elf->e_data[EI_DATA];
	elf->e_version = elf->e_data[EI_VERSION];
 }
 else if (size >= SARMAG && !memcmp(elf->e_data, ARMAG, SARMAG)) {
	_elf_init_ar(elf);
 }
}

Elf*
elf_begin(int fd, Elf_Cmd cmd, Elf *ref) {
 Elf_Arhdr *arhdr = NULL;
 size_t size = 0;
 off_t off;
 Elf *elf;

 elf_assert(_elf_init.e_magic == ELF_MAGIC);
 if (_elf_version == EV_NONE) {
	seterr(ERROR_VERSION_UNSET);
	return NULL;
 }
 else if (cmd == ELF_C_NULL) {
	return NULL;
 }
 else if (cmd == ELF_C_WRITE) {
	ref = NULL;
 }
 else if (cmd != ELF_C_READ && cmd != ELF_C_RDWR) {
	seterr(ERROR_INVALID_CMD);
	return NULL;
 }
 else if (ref) {
	elf_assert(ref->e_magic == ELF_MAGIC);
	if (!ref->e_readable || (cmd == ELF_C_RDWR && !ref->e_writable)) {
	 seterr(ERROR_CMDMISMATCH);
	 return NULL;
	}
	if (ref->e_kind != ELF_K_AR) {
	 ref->e_count++;
	 return ref;
	}
	if (cmd == ELF_C_RDWR) {
	 seterr(ERROR_MEMBERWRITE);
	 return NULL;
	}
	if (ref->e_memory) {
	 fd = ref->e_fd;
	}
	else if (fd != ref->e_fd) {
	 seterr(ERROR_FDMISMATCH);
	 return NULL;
	}
	if (!(arhdr = _elf_arhdr(ref))) {
	 return NULL;
	}
	size = arhdr->ar_size;
 }
 else if ((off = lseek(fd, (off_t)0, SEEK_END)) == (off_t)-1
	 || (off_t)(size = off) != off) {
	seterr(ERROR_IO_GETSIZE);
	return NULL;
 }

 if (!(elf = (Elf*)malloc(sizeof(Elf)))) {
	seterr(ERROR_MEM_ELF);
	return NULL;
 }
 *elf = _elf_init;
 elf->e_fd = fd;
 elf->e_parent = ref;
 elf->e_size = elf->e_dsize = size;

 if (cmd != ELF_C_READ) {
	elf->e_writable = 1;
 }
 if (cmd != ELF_C_WRITE) {
	elf->e_readable = 1;
 }
 else {
	return elf;
 }

 if (ref) {
	size_t offset = ref->e_off + sizeof(struct ar_hdr);
	Elf *xelf;

	elf_assert(arhdr);
	elf->e_arhdr = arhdr;
	elf->e_base = ref->e_base + offset;
	/*
	 * Share the archive's memory image. To avoid
	 * multiple independent elf descriptors if the
	 * same member is requested twice, scan the list
	 * of open members for duplicates.
	 *
	 * I don't know how SVR4 handles this case. Don't rely on it.
	 */
	for (xelf = ref->e_members; xelf; xelf = xelf->e_link) {
	 elf_assert(xelf->e_parent == ref);
	 if (xelf->e_base == elf->e_base) {
		free(arhdr);
		free(elf);
		xelf->e_count++;
		return xelf;
	 }
	}
	if (size == 0) {
	 elf->e_data = NULL;
	}
#if 1
	else {
	 /*
	 * Archive members may be misaligned. Freezing them will
	 * cause libelf to allocate buffers for translated data,
	 * which should be properly aligned in all cases.
	 */
	 elf_assert(!ref->e_cooked);
	 elf->e_data = elf->e_rawdata = ref->e_data + offset;
	}
#else
	else if (ref->e_data == ref->e_rawdata) {
	 elf_assert(!ref->e_cooked);
	 /*
	 * archive is frozen - freeze member, too
	 */
	 elf->e_data = elf->e_rawdata = ref->e_data + offset;
	}
	else {
	 elf_assert(!ref->e_memory);
	 elf->e_data = ref->e_data + offset;
	 /*
	 * The member's memory image may have been modified if
	 * the member has been processed before. Since we need the
	 * original image, we have to re-read the archive file.
	 * Will fail if the archive's file descriptor is disabled.
	 */
	 if (!ref->e_cooked) {
		ref->e_cooked = 1;
	 }
	 else if (!_elf_read(ref, elf->e_data, offset, size)) {
		free(arhdr);
		free(elf);
		return NULL;
	 }
	}
#endif
	elf->e_next = offset + size + (size & 1);
	elf->e_disabled = ref->e_disabled;
	elf->e_memory = ref->e_memory;
	/* parent/child linking */
	elf->e_link = ref->e_members;
	ref->e_members = elf;
	ref->e_count++;
	/* Slowaris compatibility - do not rely on this! */
	ref->e_off = elf->e_next;
 }
 else if (size) {
#if HAVE_MMAP
	/*
	 * Using mmap on writable files will interfere with elf_update
	 */
	if (!elf->e_writable && (elf->e_data = _elf_mmap(elf))) {
	 elf->e_unmap_data = 1;
	}
	else
#endif /* HAVE_MMAP */
	if (!(elf->e_data = _elf_read(elf, NULL, 0, size))) {
	 free(elf);
	 return NULL;
	}
 }

 _elf_check_type(elf, size);
 return elf;
}

Elf*
elf_memory(char *image, size_t size) {
 Elf *elf;

 elf_assert(_elf_init.e_magic == ELF_MAGIC);
 if (_elf_version == EV_NONE) {
	seterr(ERROR_VERSION_UNSET);
	return NULL;
 }
 else if (size == 0 || image == NULL) {
	/* TODO: set error code? */
	return NULL;
 }

 if (!(elf = (Elf*)malloc(sizeof(Elf)))) {
	seterr(ERROR_MEM_ELF);
	return NULL;
 }
 *elf = _elf_init;
 elf->e_size = elf->e_dsize = size;
 elf->e_data = elf->e_rawdata = image;
 elf->e_readable = 1;
 elf->e_disabled = 1;
 elf->e_memory = 1;

 _elf_check_type(elf, size);
 return elf;
}

#if __LIBELF64

int
gelf_getclass(Elf *elf) {
 if (elf && elf->e_kind == ELF_K_ELF && valid_class(elf->e_class)) {
	return elf->e_class;
 }
 return ELFCLASSNONE;
}

#endif /* __LIBELF64 */