source: trunk/src/kernel32/winimagebase.h@ 3886

/* $Id: winimagebase.h,v 1.13 2000-06-28 18:08:35 sandervl Exp $ */

/*
 * Win32 PE Image base class
 *
 * Copyright 1998-1999 Sander van Leeuwen (sandervl@xs4all.nl)
 *
 *
 * Project Odin Software License can be found in LICENSE.TXT
 *
 */
#ifndef __WINIMAGEBASE_H__
#define __WINIMAGEBASE_H__

#include <peexe.h>
#include "queue.h"

#define MAGIC_WINIMAGE          0x11223344

#ifndef CCHMAXPATH
#define CCHMAXPATH 260
#endif

#define LANG_GETFIRST           0xF0000000
#define ID_GETFIRST             0xF0000000

#ifndef ENUMRESNAMEPROC
    typedef BOOL (* CALLBACK ENUMRESTYPEPROCA)(HMODULE,LPSTR,LONG);
    typedef BOOL (* CALLBACK ENUMRESTYPEPROCW)(HMODULE,LPWSTR,LONG);
    typedef BOOL (* CALLBACK ENUMRESNAMEPROCA)(HMODULE,LPCSTR,LPSTR,LONG);
    typedef BOOL (* CALLBACK ENUMRESNAMEPROCW)(HMODULE,LPCWSTR,LPWSTR,LONG);
    typedef BOOL (* CALLBACK ENUMRESLANGPROCA)(HMODULE,LPCSTR,LPCSTR,WORD,LONG);
    typedef BOOL (* CALLBACK ENUMRESLANGPROCW)(HMODULE,LPCWSTR,LPCWSTR,WORD,LONG);
#endif

class Win32Resource;
class Win32DllBase;

class Win32ImageBase
{
protected:
        DWORD magic;
public:
        void checkObject()
        {
                if(magic != MAGIC_WINIMAGE) {
                        eprintf(("Corrupt this pointer %X %X!!", this, magic));
                        DebugInt3();
                }
        };

public:
         // Constructors and destructors
         Win32ImageBase(HINSTANCE hInstance);
virtual ~Win32ImageBase();

        ULONG     getError()          { return errorState; };
        HINSTANCE getInstanceHandle() { return hinstance; };

//Returns required OS version for this image
virtual ULONG getVersion();

virtual void  setFullPath(char *name);
        char *getFullPath()           { return fullpath; };

        char *getModuleName()         { return szModule; };

        //findResource returns the pointer of the resource's IMAGE_RESOURCE_DATA_ENTRY structure
        HRSRC findResourceA(LPCSTR lpszName, LPSTR lpszType, ULONG lang = LANG_GETFIRST);
        HRSRC findResourceW(LPWSTR lpszName, LPWSTR lpszType, ULONG lang = LANG_GETFIRST);

        ULONG getResourceSizeA(LPSTR lpszName, LPSTR lpszType, ULONG lang = LANG_GETFIRST);
        ULONG getResourceSizeW(LPWSTR lpszName, LPWSTR lpszType, ULONG lang = LANG_GETFIRST);
        BOOL  enumResourceNamesA(HMODULE hmod, LPCTSTR  lpszType, ENUMRESNAMEPROCA lpEnumFunc, LONG lParam);
        BOOL  enumResourceNamesW(HMODULE hmod, LPCWSTR  lpszType, ENUMRESNAMEPROCW lpEnumFunc, LONG lParam);
        BOOL  enumResourceTypesA(HMODULE hmod, ENUMRESTYPEPROCA lpEnumFunc, 
                                 LONG lParam);
        BOOL  enumResourceTypesW(HMODULE hmod, ENUMRESTYPEPROCW lpEnumFunc, 
                                 LONG lParam);

        ULONG getVersionSize();
        BOOL  getVersionStruct(char *verstruct, ULONG bufLength);

        //Returns pointer to data of resource handle
        char *getResourceAddr(HRSRC hResource);
        ULONG getResourceSize(HRSRC hResource);

        //returns ERROR_SUCCESS or error code
static  ULONG isPEImage(char *szFileName);
static  BOOL  findDll(const char *pszFileName, char *pszFullName,
                      int cchFullName, const char *pszAltPath = NULL);

        void  setEntryPoint(ULONG startAddress) { entryPoint = startAddress; };

        void  setTLSAddress(LPVOID dwTlsAddress)        { tlsAddress = dwTlsAddress; };
        void  setTLSIndexAddr(LPDWORD dwTlsIndexAddr)   { tlsIndexAddr = dwTlsIndexAddr; };
        void  setTLSInitSize(ULONG dwTlsSize)           { tlsInitSize = dwTlsSize; };
        void  setTLSTotalSize(ULONG dwTlsSize)          { tlsTotalSize = dwTlsSize; };
        void  setTLSCallBackAddr(PIMAGE_TLS_CALLBACK *dwTlsCallBackAddr)
        {
           tlsCallBackAddr = dwTlsCallBackAddr;
        };

        void  tlsAttachThread();        //setup TLS structures for new thread
        void  tlsDetachThread();        //destroy TLS structures

virtual BOOL  insideModule(ULONG address);
virtual BOOL  insideModuleCode(ULONG address);

virtual ULONG getApi(char *name)  = 0;
virtual ULONG getApi(int ordinal) = 0;

virtual ULONG getImageSize();

virtual BOOL  isDll() = 0;

static Win32ImageBase * findModule(HMODULE hModule);

        //Add image to dependency list of this image
        void  addDependency(Win32DllBase *dll);
        BOOL  dependsOn(Win32DllBase *dll);

protected:
        void tlsAlloc();                //Allocate TLS index for this module
        void tlsDelete();               //Destroy TLS index for this module

        ULONG                 errorState, entryPoint;

        char                 *fullpath;
        char                  szModule[CCHMAXPATH];
        char                  szFileName[CCHMAXPATH];

        HINSTANCE             hinstance;

        LPVOID                tlsAddress;               //address of TLS data
        LPDWORD               tlsIndexAddr;             //address of DWORD that receives the TLS index
        ULONG                 tlsInitSize;              //size of initialized TLS memory block
        ULONG                 tlsTotalSize;             //size of TLS memory block
        PIMAGE_TLS_CALLBACK  *tlsCallBackAddr;  //ptr to TLS callback array
        ULONG                 tlsIndex;         //module TLS index

        PIMAGE_RESOURCE_DIRECTORY  getResSubDirW(PIMAGE_RESOURCE_DIRECTORY pResDir, LPCWSTR lpszName);
        PIMAGE_RESOURCE_DIRECTORY  getResSubDirA(PIMAGE_RESOURCE_DIRECTORY pResDir, LPCTSTR lpszName);

        PIMAGE_RESOURCE_DATA_ENTRY getResDataLang(PIMAGE_RESOURCE_DIRECTORY pResDir, ULONG language, BOOL fGetDefault = FALSE);

        PIMAGE_RESOURCE_DIRECTORY pResRootDir;

        //substracted from RVA data offsets
        ULONG                     ulRVAResourceSection;

        //linked list of dlls loaded on behalf of this executable image (dll or exe)
        Queue                      loadedDlls;
private:

        friend class Win32Resource;
        friend ULONG SYSTEM GetVersionSize(char *modname);
};

//SvL: This structure is placed at the end of the first page of the image (header
//     page), so we can determine the Win32Image pointer from a HINSTANCE variable
//     (which is actually the address of the win32 module)
typedef struct
{
  ULONG           magic1;
  Win32ImageBase *image;
  ULONG           magic2;
} WINIMAGE_LOOKUP;

#define WINIMAGE_LOOKUPADDR(a)  (WINIMAGE_LOOKUP *)((ULONG)a + PAGE_SIZE - sizeof(WINIMAGE_LOOKUP))

#endif //__WINIMAGEBASE_H__