source: trunk/include/winimagebase.h@ 2606

/* $Id: winimagebase.h,v 1.8 1999-11-30 19:39:14 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>

#define MAGIC_WINIMAGE          0x11223344

#ifndef CCHMAXPATH
#define CCHMAXPATH 260
#endif

#define LANG_GETFIRST           0x80000000
#define ID_GETFIRST             LANG_GETFIRST
#define IDLANG_GETFIRST         LANG_GETFIRST

#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 Win32ImageBase
{
#ifdef DEBUG
protected:
        DWORD magic;
public:
        void checkObject()
        {
                if(magic != MAGIC_WINIMAGE) {
                        eprintf(("Corrupt this pointer %X %X!!", this, magic));
                        DebugInt3();
                }
        };
#endif

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; };

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

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

static  BOOL  isPEImage(char *szFileName);

        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 ULONG     getApi(char *name)  = 0;
virtual ULONG     getApi(int ordinal) = 0;

virtual BOOL  isDll() = 0;

static Win32ImageBase * findModule(HMODULE hModule);

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

        Win32Resource        *winres;

        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

        ULONG getPEResourceSize(ULONG id, ULONG type, ULONG lang = LANG_GETFIRST);

        PIMAGE_RESOURCE_DATA_ENTRY getPEResourceEntry(ULONG id, ULONG type, ULONG lang = LANG_GETFIRST);
        PIMAGE_RESOURCE_DATA_ENTRY ProcessResSubDir(PIMAGE_RESOURCE_DIRECTORY prdType,
                                                    ULONG *nodeData, int level);
        PIMAGE_RESOURCE_DIRECTORY  getResSubDirW(PIMAGE_RESOURCE_DIRECTORY pResDir, LPCWSTR lpszName);
        PIMAGE_RESOURCE_DIRECTORY  getResSubDirA(PIMAGE_RESOURCE_DIRECTORY pResDir, LPCTSTR lpszName);

        PIMAGE_RESOURCE_DIRECTORY pResDir;

        //substracted from RVA data offsets
        ULONG                     ulRVAResourceSection;

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__