ldrkStuff.cpp revision cf3aad827eee194a3e6e68796710074b44164371
/* $Id$ */
/** @file
* innotek Portable Runtime - Binary Image Loader, kLdr Interface.
*/
/*
* Copyright (C) 2006-2007 innotek GmbH
*
* This file is part of VirtualBox Open Source Edition (OSE), as
* available from http://www.virtualbox.org. This file is free software;
* General Public License as published by the Free Software Foundation,
* in version 2 as it comes in the "COPYING" file of the VirtualBox OSE
* distribution. VirtualBox OSE is distributed in the hope that it will
* be useful, but WITHOUT ANY WARRANTY of any kind.
*/
/*******************************************************************************
* Header Files *
*******************************************************************************/
#define LOG_GROUP RTLOGGROUP_LDR
#define KLDR_ALREADY_INCLUDE_STD_TYPES
#define KLDR_NO_KLDR_H_INCLUDES
/*******************************************************************************
* Structures and Typedefs *
*******************************************************************************/
/**
* kLdr file provider.
*/
typedef struct RTKLDRRDR
{
/** The core. */
/** The IPRT bit reader. */
} RTKLDRRDR, *PRTKLDRRDR;
/**
* IPRT module.
*/
typedef struct RTLDRMODKLDR
{
/** The Core module structure. */
/** The kLdr module. */
/**
* Arguments for a RTLDRMODKLDR callback wrapper.
*/
typedef struct RTLDRMODKLDRARGS
{
union
{
} u;
void *pvUser;
const void *pvBits;
/*******************************************************************************
* Internal Functions *
*******************************************************************************/
static int rtkldrConvertError(int krc);
static int rtkldrConvertErrorFromIPRT(int rc);
static int rtkldrRdrMap( PKRDR pRdr, void **ppvBase, KU32 cSegments, PCKLDRSEG paSegments, KBOOL fFixed);
static int rtkldrRdrProtect( PKRDR pRdr, void *pvBase, KU32 cSegments, PCKLDRSEG paSegments, KBOOL fUnprotectOrProtect);
static DECLCALLBACK(int) rtkldrEnumSymbols(PRTLDRMODINTERNAL pMod, unsigned fFlags, const void *pvBits,
static DECLCALLBACK(int) rtkldrRelocate(PRTLDRMODINTERNAL pMod, void *pvBits, RTUINTPTR NewBaseAddress,
static int rtkldrGetImportWrapper(PKLDRMOD pMod, uint32_t iImport, uint32_t iSymbol, const char *pchSymbol, KSIZE cchSymbol,
static DECLCALLBACK(int) rtkldrGetSymbolEx(PRTLDRMODINTERNAL pMod, const void *pvBits, RTUINTPTR BaseAddress,
/**
* Converts a kLdr error code to an IPRT one.
*/
static int rtkldrConvertError(int krc)
{
if (!krc)
return VINF_SUCCESS;
switch (krc)
{
case KERR_INVALID_PARAMETER: return VERR_INVALID_PARAMETER;
case KERR_INVALID_HANDLE: return VERR_INVALID_HANDLE;
case KERR_NO_MEMORY: return VERR_NO_MEMORY;
case KLDR_ERR_UNKNOWN_FORMAT:
case KLDR_ERR_MZ_NOT_SUPPORTED: return VERR_MZ_EXE_NOT_SUPPORTED;
case KLDR_ERR_NE_NOT_SUPPORTED: return VERR_NE_EXE_NOT_SUPPORTED;
case KLDR_ERR_LX_NOT_SUPPORTED: return VERR_LX_EXE_NOT_SUPPORTED;
case KLDR_ERR_LE_NOT_SUPPORTED: return VERR_LE_EXE_NOT_SUPPORTED;
case KLDR_ERR_PE_NOT_SUPPORTED: return VERR_PE_EXE_NOT_SUPPORTED;
case KLDR_ERR_ELF_NOT_SUPPORTED: return VERR_ELF_EXE_NOT_SUPPORTED;
case KLDR_ERR_MACHO_NOT_SUPPORTED: return VERR_INVALID_EXE_SIGNATURE;
case KLDR_ERR_AOUT_NOT_SUPPORTED: return VERR_AOUT_EXE_NOT_SUPPORTED;
case KLDR_ERR_MODULE_NOT_FOUND: return VERR_MODULE_NOT_FOUND;
case KLDR_ERR_PREREQUISITE_MODULE_NOT_FOUND: return VERR_MODULE_NOT_FOUND;
case KLDR_ERR_MAIN_STACK_ALLOC_FAILED: return VERR_NO_MEMORY;
case KERR_BUFFER_OVERFLOW: return VERR_BUFFER_OVERFLOW;
case KLDR_ERR_SYMBOL_NOT_FOUND: return VERR_SYMBOL_NOT_FOUND;
case KLDR_ERR_FORWARDER_SYMBOL: return VERR_BAD_EXE_FORMAT;
case KLDR_ERR_BAD_FIXUP: return VERR_BAD_EXE_FORMAT;
case KLDR_ERR_IMPORT_ORDINAL_OUT_OF_BOUNDS: return VERR_BAD_EXE_FORMAT;
case KLDR_ERR_NO_DEBUG_INFO: return VERR_FILE_NOT_FOUND;
case KLDR_ERR_ALREADY_MAPPED: return VERR_WRONG_ORDER;
case KLDR_ERR_NOT_MAPPED: return VERR_WRONG_ORDER;
case KLDR_ERR_ADDRESS_OVERFLOW: return VERR_NUMBER_TOO_BIG;
case KLDR_ERR_NOT_DLL:
case KLDR_ERR_NOT_EXE:
return VERR_GENERAL_FAILURE;
case KLDR_ERR_PE_BAD_FIXUP:
case KLDR_ERR_PE_BAD_IMPORT:
return VERR_GENERAL_FAILURE;
case KLDR_ERR_LX_BAD_HEADER:
case KLDR_ERR_LX_BAD_PAGE_MAP:
case KLDR_ERR_LX_BAD_ITERDATA:
case KLDR_ERR_LX_BAD_BUNDLE:
case KLDR_ERR_LX_NO_SONAME:
case KLDR_ERR_LX_BAD_SONAME:
return VERR_GENERAL_FAILURE;
default:
if (RT_FAILURE(krc))
return krc;
return VERR_NO_TRANSLATION;
}
}
/**
* Converts a IPRT error code to an kLdr one.
*/
static int rtkldrConvertErrorFromIPRT(int rc)
{
if (RT_SUCCESS(rc))
return 0;
switch (rc)
{
case VERR_NO_MEMORY: return KERR_NO_MEMORY;
case VERR_INVALID_PARAMETER: return KERR_INVALID_PARAMETER;
case VERR_INVALID_HANDLE: return KERR_INVALID_HANDLE;
case VERR_BUFFER_OVERFLOW: return KERR_BUFFER_OVERFLOW;
default:
return rc;
}
}
/**
* The file reader operations.
* We provide our own based on IPRT instead of using the kLdr ones.
*/
{
/* .pszName = */ "IPRT",
/* .pNext = */ NULL,
/* .pfnCreate = */ rtkldrRdrCreate,
/* .pfnDestroy = */ rtkldrRdrDestroy,
/* .pfnRead = */ rtkldrRdrRead,
/* .pfnAllMap = */ rtkldrRdrAllMap,
/* .pfnAllUnmap = */ rtkldrRdrAllUnmap,
/* .pfnSize = */ rtkldrRdrSize,
/* .pfnTell = */ rtkldrRdrTell,
/* .pfnName = */ rtkldrRdrName,
/* .pfnNativeFH = */ rtkldrRdrNativeFH,
/* .pfnPageSize = */ rtkldrRdrPageSize,
/* .pfnMap = */ rtkldrRdrMap,
/* .pfnRefresh = */ rtkldrRdrRefresh,
/* .pfnProtect = */ rtkldrRdrProtect,
/* .pfnUnmap = */ rtkldrRdrUnmap,
/* .pfnDone = */ rtkldrRdrDone,
/* .u32Dummy = */ 42
};
/** @copydoc KLDRRDROPS::pfnCreate
* @remark This is a dummy which isn't used. */
{
return -1;
}
/** @copydoc KLDRRDROPS::pfnDestroy */
{
return rtkldrConvertErrorFromIPRT(rc);
}
/** @copydoc KLDRRDROPS::pfnRead */
{
return rtkldrConvertErrorFromIPRT(rc);
}
/** @copydoc KLDRRDROPS::pfnAllMap */
{
return rtkldrConvertErrorFromIPRT(rc);
}
/** @copydoc KLDRRDROPS::pfnAllUnmap */
{
return rtkldrConvertErrorFromIPRT(rc);
}
/** @copydoc KLDRRDROPS::pfnSize */
{
}
/** @copydoc KLDRRDROPS::pfnTell */
{
}
/** @copydoc KLDRRDROPS::pfnName */
{
}
/** @copydoc KLDRRDROPS::pfnNativeFH */
{
AssertFailed();
return -1;
}
/** @copydoc KLDRRDROPS::pfnPageSize */
{
return PAGE_SIZE;
}
/** @copydoc KLDRRDROPS::pfnMap */
static int rtkldrRdrMap( PKRDR pRdr, void **ppvBase, KU32 cSegments, PCKLDRSEG paSegments, KBOOL fFixed)
{
//PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
AssertFailed();
return -1;
}
/** @copydoc KLDRRDROPS::pfnRefresh */
{
//PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
AssertFailed();
return -1;
}
/** @copydoc KLDRRDROPS::pfnProtect */
static int rtkldrRdrProtect( PKRDR pRdr, void *pvBase, KU32 cSegments, PCKLDRSEG paSegments, KBOOL fUnprotectOrProtect)
{
//PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
AssertFailed();
return -1;
}
/** @copydoc KLDRRDROPS::pfnUnmap */
{
//PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
AssertFailed();
return -1;
}
/** @copydoc KLDRRDROPS::pfnDone */
{
//PRTLDRREADER pReader = ((PRTKLDRRDR)pRdr)->pReader;
}
/**
* Operations for a kLdr module.
*/
static const RTLDROPS g_rtkldrOps =
{
"kLdr",
NULL,
/* ext */
42
};
/** @copydoc RTLDROPS::pfnClose */
{
return rtkldrConvertError(rc);
}
/** @copydoc RTLDROPS::pfnDone */
{
return rtkldrConvertError(rc);
}
/** @copydoc RTLDROPS::pfnEnumSymbols */
static DECLCALLBACK(int) rtkldrEnumSymbols(PRTLDRMODINTERNAL pMod, unsigned fFlags, const void *pvBits, RTUINTPTR BaseAddress,
{
return rtkldrConvertError(rc);
}
/** @copydoc FNKLDRMODENUMSYMS */
{
/* If not zero terminated we'll have to use a temporary buffer. */
{
}
/* skip the underscore prefix. */
if (*pszSymbol == '_')
pszSymbol++;
#endif
if (RT_FAILURE(rc))
return rc; /* don't bother converting. */
return 0;
}
/** @copydoc RTLDROPS::pfnGetImageSize */
{
return kLdrModSize(pModkLdr);
}
/** @copydoc RTLDROPS::pfnGetBits */
{
return rtkldrConvertError(rc);
}
/** @copydoc RTLDROPS::pfnRelocate */
static DECLCALLBACK(int) rtkldrRelocate(PRTLDRMODINTERNAL pMod, void *pvBits, RTUINTPTR NewBaseAddress,
{
int rc = kLdrModRelocateBits(pModkLdr, pvBits, NewBaseAddress, OldBaseAddress, rtkldrGetImportWrapper, &Args);
return rtkldrConvertError(rc);
}
/** @copydoc FNKLDRMODGETIMPORT */
static int rtkldrGetImportWrapper(PKLDRMOD pMod, uint32_t iImport, uint32_t iSymbol, const char *pchSymbol, KSIZE cchSymbol,
{
/* If not zero terminated we'll have to use a temporary buffer. */
{
}
/* skip the underscore prefix. */
if (*pszSymbol == '_')
pszSymbol++;
#endif
/* get the import module name - TODO: cache this */
if (iImport != NIL_KLDRMOD_IMPORT)
{
if (rc)
return rc;
}
/* do the query */
int rc = pArgs->u.pfnGetImport(&pArgs->pMod->Core, pszModule, pszSymbol, pszSymbol ? ~0 : iSymbol, &Value, pArgs->pvUser);
if (RT_SUCCESS(rc))
{
return 0;
}
return rtkldrConvertErrorFromIPRT(rc);
}
/** @copydoc RTLDROPS::pfnGetSymbolEx */
static DECLCALLBACK(int) rtkldrGetSymbolEx(PRTLDRMODINTERNAL pMod, const void *pvBits, RTUINTPTR BaseAddress,
{
/*
* Add underscore prefix.
*/
if (pszSymbol)
{
*psz = '_';
}
#endif
if (!rc)
{
return VINF_SUCCESS;
}
return rtkldrConvertError(rc);
}
/**
* Open a image using kLdr.
*
* @returns iprt status code.
* @param pReader The loader reader instance which will provide the raw image bits.
* @param phLdrMod Where to store the handle.
*/
{
/* Create a rtkldrRdr instance. */
if (!pRdr)
return VERR_NO_MEMORY;
/* Try open it. */
if (!krc)
{
/* Create a module wrapper for it. */
if (pNewMod)
{
return VINF_SUCCESS;
}
}
return rtkldrConvertError(krc);
}