/*
 * Copyright (C) 2008, 2009 The Android Open Source Project
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *  * Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *  * 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "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
 * COPYRIGHT OWNER OR CONTRIBUTORS 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.
 */

#if defined __ANDROID__ && !defined XASH_64BIT
#include <string.h>
#include <android/log.h>
#include <dlfcn.h>
#include "linker.h"

static Elf_Sym* soinfo_elf_lookup(soinfo* si, unsigned hash, const char* name) {
    Elf_Sym* symtab = si->symtab;
    const char* strtab = si->strtab;

	if( si->nbucket == 0 )
		return NULL;

    for (unsigned n = si->bucket[hash % si->nbucket]; n != 0; n = si->chain[n]) {
        Elf_Sym* s = symtab + n;
        if (strcmp(strtab + s->st_name, name)) continue;

            /* only concern ourselves with global and weak symbol definitions */
        switch (ELF_ST_BIND(s->st_info)) {
        case STB_GLOBAL:
        case STB_WEAK:
            if (s->st_shndx == SHN_UNDEF) {
                continue;
            }
            return s;
        }
    }

    return NULL;
}

static Elf_Sym* soinfo_elf_lookup_reverse(soinfo* si, size_t addr) {
	Elf_Sym* symtab = si->symtab;

	if( si->nbucket == 0 )
		return NULL;

	for(int j = 0; j < si->nbucket; j++)
	for (unsigned n = si->bucket[j]; n != 0; n = si->chain[n]) {
		Elf_Sym* s = symtab + n;
		if (s->st_value != addr) continue;

			/* only concern ourselves with global and weak symbol definitions */
		switch (ELF_ST_BIND(s->st_info)) {
		case STB_GLOBAL:
		case STB_LOCAL:
		case STB_WEAK:
			if (s->st_shndx == SHN_UNDEF) {
				continue;
			}
			return s;
		}
	}

	return NULL;
}


static unsigned elfhash(const char* _name) {
    const unsigned char* name = (const unsigned char*) _name;
    unsigned h = 0, g;

    while(*name) {
        h = (h << 4) + *name++;
        g = h & 0xf0000000;
        h ^= g;
        h ^= g >> 24;
    }
    return h;
}

/* This is used by dlsym(3).  It performs symbol lookup only within the
   specified soinfo object and not in any of its dependencies.

   TODO: Only looking in the specified soinfo seems wrong. dlsym(3) says
   that it should do a breadth first search through the dependency
   tree. This agrees with the ELF spec (aka System V Application
   Binary Interface) where in Chapter 5 it discuss resolving "Shared
   Object Dependencies" in breadth first search order.
 */
static Elf_Sym* dlsym_handle_lookup(soinfo* si, const char* name) {
    return soinfo_elf_lookup(si, elfhash(name), name);
}

extern "C" void *ANDROID_LoadLibrary( const char *dllname );
static int dladdr_fallback( const void *addr, Dl_info *info )
{
	static soinfo *server_info;
	Elf_Sym *sym;

	if( !server_info )
		server_info = (soinfo*)ANDROID_LoadLibrary( "server" );
	if( !server_info )
		return 0;
	//__android_log_print( ANDROID_LOG_ERROR, "dladdr_fb", "%p %p\n", addr, server_info );

	sym =  soinfo_elf_lookup_reverse( server_info, ((char*)addr) - ((char*)server_info->base ));
	//__android_log_print( ANDROID_LOG_ERROR, "dladdr_fb", "sym %p %p\n", addr, sym );
	if( sym )
	{
		info->dli_sname = server_info->strtab + sym->st_name;
		info->dli_fname = "server";
		info->dli_fbase = (void*)server_info->base;
		info->dli_saddr = (void*)addr;
		//__android_log_print( ANDROID_LOG_ERROR, "dladdr_fb", "name %p %s\n", addr, info->dli_sname );
		return 1;
	}

	return 0;
}



extern "C" int __attribute__((visibility("hidden"))) dladdr( const void *addr, Dl_info *info )
{
	static int (*pDladdr)( const void *addr, Dl_info *info );
	// __android_log_print( ANDROID_LOG_ERROR, "dladdr", "dladdr %p %p %p\n", addr, pDladdr, &dladdr );

	if(!pDladdr)
	{
		void *lib = dlopen( "libdl.so", RTLD_NOW );
		if( lib )
			*(void**)&pDladdr = dlsym( lib, "dladdr" );
		if( (void*)pDladdr == (void*)&dladdr )
			pDladdr = 0;
		if( !pDladdr )
			pDladdr = dladdr_fallback;
	}
	// __android_log_print( ANDROID_LOG_ERROR, "dladdr", "dladdr %p\n", addr );
	return pDladdr( addr, info );
}


extern "C" void* dlsym_weak(void* handle, const char* symbol) {

  soinfo* found = NULL;
  Elf_Sym* sym = NULL;
  found = reinterpret_cast<soinfo*>(handle);
  sym = dlsym_handle_lookup(found, symbol);

  if (sym != NULL) {
    return reinterpret_cast<void*>(sym->st_value + found->base/*load_bias*/);
  }
  __android_log_print(ANDROID_LOG_ERROR, "dlsym-weak", "Failed when looking up %s\n", symbol);
  return NULL;
}
#endif