modify.c File Reference

#include <stdarg.h>
#include "windef.h"
#include "winbase.h"
#include "winternl.h"
#include "winerror.h"
#include "wine/debug.h"
#include "imagehlp.h"

Include dependency graph for modify.c:

Go to the source code of this file.

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (imagehlp)
static WORD	CalcCheckSum (DWORD StartValue, LPVOID BaseAddress, DWORD WordCount)
BOOL WINAPI	BindImage (PCSTR ImageName, PCSTR DllPath, PCSTR SymbolPath)
BOOL WINAPI	BindImageEx (DWORD Flags, PCSTR ImageName, PCSTR DllPath, PCSTR SymbolPath, PIMAGEHLP_STATUS_ROUTINE StatusRoutine)
PIMAGE_NT_HEADERS WINAPI	CheckSumMappedFile (LPVOID BaseAddress, DWORD FileLength, LPDWORD HeaderSum, LPDWORD CheckSum)
DWORD WINAPI	MapFileAndCheckSumA (PCSTR Filename, PDWORD HeaderSum, PDWORD CheckSum)
DWORD WINAPI	MapFileAndCheckSumW (PCWSTR Filename, PDWORD HeaderSum, PDWORD CheckSum)
BOOL WINAPI	ReBaseImage (PCSTR CurrentImageName, PCSTR SymbolPath, BOOL fReBase, BOOL fRebaseSysfileOk, BOOL fGoingDown, ULONG CheckImageSize, ULONG *OldImageSize, ULONG_PTR *OldImageBase, ULONG *NewImageSize, ULONG_PTR *NewImageBase, ULONG TimeStamp)
BOOL WINAPI	RemovePrivateCvSymbolic (PCHAR DebugData, PCHAR *NewDebugData, ULONG *NewDebugSize)
VOID WINAPI	RemoveRelocations (PCHAR ImageName)
BOOL WINAPI	SplitSymbols (PSTR ImageName, PCSTR SymbolsPath, PSTR SymbolFilePath, ULONG Flags)
BOOL WINAPI	UpdateDebugInfoFile (PCSTR ImageFileName, PCSTR SymbolPath, PSTR DebugFilePath, PIMAGE_NT_HEADERS32 NtHeaders)
BOOL WINAPI	UpdateDebugInfoFileEx (PCSTR ImageFileName, PCSTR SymbolPath, PSTR DebugFilePath, PIMAGE_NT_HEADERS32 NtHeaders, DWORD OldChecksum)

Function Documentation

BindImage()

BOOL WINAPI BindImage	(	PCSTR	ImageName,
		PCSTR	DllPath,
		PCSTR	SymbolPath
	)

Definition at line 40 of file modify.c.

{
  return BindImageEx(0, ImageName, DllPath, SymbolPath, NULL);
}

Referenced by msi_bind_image().

BindImageEx()

BOOL WINAPI BindImageEx	(	DWORD	Flags,
		PCSTR	ImageName,
		PCSTR	DllPath,
		PCSTR	SymbolPath,
		PIMAGEHLP_STATUS_ROUTINE	StatusRoutine
	)

Definition at line 67 of file modify.c.

{
    LOADED_IMAGE loaded_image;
    const IMAGE_IMPORT_DESCRIPTOR *import_desc;
    ULONG size;
 
    FIXME("(%d, %s, %s, %s, %p): semi-stub\n",
        Flags, debugstr_a(ImageName), debugstr_a(DllPath),
        debugstr_a(SymbolPath), StatusRoutine
    );
 
    if (!(MapAndLoad(ImageName, DllPath, &loaded_image, TRUE, TRUE))) return FALSE;
 
    if (!(import_desc = RtlImageDirectoryEntryToData((HMODULE)loaded_image.MappedAddress, FALSE,
                                                     IMAGE_DIRECTORY_ENTRY_IMPORT, &size)))
    {
        UnMapAndLoad(&loaded_image);
        return TRUE; /* No imported modules means nothing to bind, so we're done. */
    }
 
    /* FIXME: Does native imagehlp support both 32-bit and 64-bit PE executables? */
#ifdef _WIN64
    if (loaded_image.FileHeader->OptionalHeader.Magic != IMAGE_NT_OPTIONAL_HDR64_MAGIC)
#else
    if (loaded_image.FileHeader->OptionalHeader.Magic != IMAGE_NT_OPTIONAL_HDR32_MAGIC)
#endif
    {
        FIXME("Wrong architecture in PE header, unable to enumerate imports\n");
        UnMapAndLoad(&loaded_image);
        return TRUE;
    }
 
    for (; import_desc->Name && import_desc->FirstThunk; ++import_desc)
    {
        IMAGE_THUNK_DATA *thunk;
        char dll_fullname[MAX_PATH];
        const char *dll_name;
 
        if (!(dll_name = ImageRvaToVa(loaded_image.FileHeader, loaded_image.MappedAddress,
                                      import_desc->Name, 0)))
        {
            UnMapAndLoad(&loaded_image);
            SetLastError(ERROR_INVALID_ACCESS); /* FIXME */
            return FALSE;
        }
 
        if (StatusRoutine)
            StatusRoutine(BindImportModule, ImageName, dll_name, 0, 0);
 
        if (!SearchPathA(DllPath, dll_name, 0, sizeof(dll_fullname), dll_fullname, 0))
        {
            UnMapAndLoad(&loaded_image);
            SetLastError(ERROR_FILE_NOT_FOUND);
            return FALSE;
        }
 
        if (!(thunk = ImageRvaToVa(loaded_image.FileHeader, loaded_image.MappedAddress,
                                   import_desc->OriginalFirstThunk ? import_desc->OriginalFirstThunk :
                                   import_desc->FirstThunk, 0)))
        {
            ERR("Can't grab thunk data of %s, going to next imported DLL\n", dll_name);
            continue;
        }
 
        for (; thunk->u1.Ordinal; ++thunk)
        {
            /* Ignoring ordinal imports for now */
            if(!IMAGE_SNAP_BY_ORDINAL(thunk->u1.Ordinal))
            {
                IMAGE_IMPORT_BY_NAME *iibn;
 
                if (!(iibn = ImageRvaToVa(loaded_image.FileHeader, loaded_image.MappedAddress,
                                          thunk->u1.AddressOfData, 0)))
                {
                    ERR("Can't grab import by name info, skipping to next ordinal\n");
                    continue;
                }
 
                if (StatusRoutine)
                    StatusRoutine(BindImportProcedure, ImageName, dll_fullname, 0, (ULONG_PTR)iibn->Name);
            }
        }
    }
 
    UnMapAndLoad(&loaded_image);
    return TRUE;
}

Referenced by BindImage().

CalcCheckSum()

static WORD CalcCheckSum ( DWORD  StartValue, LPVOID  BaseAddress, DWORD  WordCount  )

static

Definition at line 161 of file modify.c.

{
   LPWORD Ptr;
   DWORD Sum;
   DWORD i;
 
   Sum = StartValue;
   Ptr = (LPWORD)BaseAddress;
   for (i = 0; i < WordCount; i++)
     {
    Sum += *Ptr;
    if (HIWORD(Sum) != 0)
      {
         Sum = LOWORD(Sum) + HIWORD(Sum);
      }
    Ptr++;
     }
 
   return (WORD)(LOWORD(Sum) + HIWORD(Sum));
}

Referenced by CheckSumMappedFile().

CheckSumMappedFile()

PIMAGE_NT_HEADERS WINAPI CheckSumMappedFile	(	LPVOID	BaseAddress,
		DWORD	FileLength,
		LPDWORD	HeaderSum,
		LPDWORD	CheckSum
	)

Definition at line 187 of file modify.c.

{
  PIMAGE_NT_HEADERS header;
  DWORD CalcSum;
  DWORD HdrSum;
 
  TRACE("(%p, %d, %p, %p)\n", BaseAddress, FileLength, HeaderSum, CheckSum);
 
  CalcSum = CalcCheckSum(0, BaseAddress, (FileLength + 1) / sizeof(WORD));
  header = RtlImageNtHeader(BaseAddress);
 
  if (!header)
    return NULL;
 
  *HeaderSum = HdrSum = header->OptionalHeader.CheckSum;
 
  /* Subtract image checksum from calculated checksum. */
  /* fix low word of checksum */
  if (LOWORD(CalcSum) >= LOWORD(HdrSum))
  {
    CalcSum -= LOWORD(HdrSum);
  }
  else
  {
    CalcSum = ((LOWORD(CalcSum) - LOWORD(HdrSum)) & 0xFFFF) - 1;
  }
 
   /* fix high word of checksum */
  if (LOWORD(CalcSum) >= HIWORD(HdrSum))
  {
    CalcSum -= HIWORD(HdrSum);
  }
  else
  {
    CalcSum = ((LOWORD(CalcSum) - HIWORD(HdrSum)) & 0xFFFF) - 1;
  }
 
  /* add file length */
  CalcSum += FileLength;
 
  *CheckSum = CalcSum;
 
  return header;
}

Referenced by IMAGEHLP_RecalculateChecksum(), MapFileAndCheckSumA(), and MapFileAndCheckSumW().

MapFileAndCheckSumA()

DWORD WINAPI MapFileAndCheckSumA	(	PCSTR	Filename,
		PDWORD	HeaderSum,
		PDWORD	CheckSum
	)

Definition at line 237 of file modify.c.

{
  HANDLE hFile;
  HANDLE hMapping;
  LPVOID BaseAddress;
  DWORD FileLength;
 
  TRACE("(%s, %p, %p): stub\n",
    debugstr_a(Filename), HeaderSum, CheckSum
  );
 
  hFile = CreateFileA(Filename,
              GENERIC_READ,
              FILE_SHARE_READ | FILE_SHARE_WRITE,
              NULL,
              OPEN_EXISTING,
              FILE_ATTRIBUTE_NORMAL,
              0);
  if (hFile == INVALID_HANDLE_VALUE)
  {
    return CHECKSUM_OPEN_FAILURE;
  }
 
  hMapping = CreateFileMappingW(hFile,
                   NULL,
                   PAGE_READONLY,
                   0,
                   0,
                   NULL);
  if (hMapping == 0)
  {
    CloseHandle(hFile);
    return CHECKSUM_MAP_FAILURE;
  }
 
  BaseAddress = MapViewOfFile(hMapping,
                  FILE_MAP_READ,
                  0,
                  0,
                  0);
  if (BaseAddress == 0)
  {
    CloseHandle(hMapping);
    CloseHandle(hFile);
    return CHECKSUM_MAPVIEW_FAILURE;
  }
 
  FileLength = GetFileSize(hFile,
               NULL);
 
  CheckSumMappedFile(BaseAddress,
             FileLength,
             HeaderSum,
             CheckSum);
 
  UnmapViewOfFile(BaseAddress);
  CloseHandle(hMapping);
  CloseHandle(hFile);
 
  return 0;
}

MapFileAndCheckSumW()

DWORD WINAPI MapFileAndCheckSumW	(	PCWSTR	Filename,
		PDWORD	HeaderSum,
		PDWORD	CheckSum
	)

Definition at line 303 of file modify.c.

{
  HANDLE hFile;
  HANDLE hMapping;
  LPVOID BaseAddress;
  DWORD FileLength;
 
  TRACE("(%s, %p, %p): stub\n",
    debugstr_w(Filename), HeaderSum, CheckSum
  );
 
  hFile = CreateFileW(Filename,
              GENERIC_READ,
              FILE_SHARE_READ | FILE_SHARE_WRITE,
              NULL,
              OPEN_EXISTING,
              FILE_ATTRIBUTE_NORMAL,
              0);
  if (hFile == INVALID_HANDLE_VALUE)
  {
  return CHECKSUM_OPEN_FAILURE;
  }
 
  hMapping = CreateFileMappingW(hFile,
                   NULL,
                   PAGE_READONLY,
                   0,
                   0,
                   NULL);
  if (hMapping == 0)
  {
    CloseHandle(hFile);
    return CHECKSUM_MAP_FAILURE;
  }
 
  BaseAddress = MapViewOfFile(hMapping,
                  FILE_MAP_READ,
                  0,
                  0,
                  0);
  if (BaseAddress == 0)
  {
    CloseHandle(hMapping);
    CloseHandle(hFile);
    return CHECKSUM_MAPVIEW_FAILURE;
  }
 
  FileLength = GetFileSize(hFile,
               NULL);
 
  CheckSumMappedFile(BaseAddress,
             FileLength,
             HeaderSum,
             CheckSum);
 
  UnmapViewOfFile(BaseAddress);
  CloseHandle(hMapping);
  CloseHandle(hFile);
 
  return 0;
}

ReBaseImage()

BOOL WINAPI ReBaseImage	(	PCSTR	CurrentImageName,
		PCSTR	SymbolPath,
		BOOL	fReBase,
		BOOL	fRebaseSysfileOk,
		BOOL	fGoingDown,
		ULONG	CheckImageSize,
		ULONG *	OldImageSize,
		ULONG_PTR *	OldImageBase,
		ULONG *	NewImageSize,
		ULONG_PTR *	NewImageBase,
		ULONG	TimeStamp
	)

Definition at line 369 of file modify.c.

{
  FIXME(
    "(%s, %s, %d, %d, %d, %d, %p, %p, %p, %p, %d): stub\n",
      debugstr_a(CurrentImageName),debugstr_a(SymbolPath), fReBase,
      fRebaseSysfileOk, fGoingDown, CheckImageSize, OldImageSize,
      OldImageBase, NewImageSize, NewImageBase, TimeStamp
  );
  SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
  return FALSE;
}

RemovePrivateCvSymbolic()

BOOL WINAPI RemovePrivateCvSymbolic	(	PCHAR	DebugData,
		PCHAR *	NewDebugData,
		ULONG *	NewDebugSize
	)

Definition at line 388 of file modify.c.

{
  FIXME("(%p, %p, %p): stub\n",
    DebugData, NewDebugData, NewDebugSize
  );
  SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
  return FALSE;
}

RemoveRelocations()

VOID WINAPI RemoveRelocations

(

PCHAR

ImageName

)

Definition at line 401 of file modify.c.

{
  FIXME("(%p): stub\n", ImageName);
  SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
}

SplitSymbols()

BOOL WINAPI SplitSymbols	(	PSTR	ImageName,
		PCSTR	SymbolsPath,
		PSTR	SymbolFilePath,
		ULONG	Flags
	)

Definition at line 410 of file modify.c.

{
  FIXME("(%s, %s, %s, %d): stub\n",
    debugstr_a(ImageName), debugstr_a(SymbolsPath),
    debugstr_a(SymbolFilePath), Flags
  );
  SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
  return FALSE;
}

UpdateDebugInfoFile()

BOOL WINAPI UpdateDebugInfoFile	(	PCSTR	ImageFileName,
		PCSTR	SymbolPath,
		PSTR	DebugFilePath,
		PIMAGE_NT_HEADERS32	NtHeaders
	)

Definition at line 425 of file modify.c.

{
  FIXME("(%s, %s, %s, %p): stub\n",
    debugstr_a(ImageFileName), debugstr_a(SymbolPath),
    debugstr_a(DebugFilePath), NtHeaders
  );
  SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
  return FALSE;
}

UpdateDebugInfoFileEx()

BOOL WINAPI UpdateDebugInfoFileEx	(	PCSTR	ImageFileName,
		PCSTR	SymbolPath,
		PSTR	DebugFilePath,
		PIMAGE_NT_HEADERS32	NtHeaders,
		DWORD	OldChecksum
	)

Definition at line 440 of file modify.c.

{
  FIXME("(%s, %s, %s, %p, %d): stub\n",
    debugstr_a(ImageFileName), debugstr_a(SymbolPath),
    debugstr_a(DebugFilePath), NtHeaders, OldChecksum
  );
  SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
  return FALSE;
}

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

imagehlp

)