modify.c File Reference

#include <stdarg.h>
#include "windef.h"
#include "winbase.h"
#include "winternl.h"
#include "winerror.h"
#include "winver.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, const char *module, const char *dll_path, const char *symbol_path, PIMAGEHLP_STATUS_ROUTINE cb)
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 39 of file modify.c.

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

Referenced by bind_image().

BindImageEx()

BOOL WINAPI BindImageEx	(	DWORD	flags,
		const char *	module,
		const char *	dll_path,
		const char *	symbol_path,
		PIMAGEHLP_STATUS_ROUTINE	cb
	)

Definition at line 48 of file modify.c.

{
    const IMAGE_IMPORT_DESCRIPTOR *import;
    LOADED_IMAGE image;
    ULONG size;
 
    TRACE("flags %#lx, module %s, dll_path %s, symbol_path %s, cb %p.\n",
            flags, debugstr_a(module), debugstr_a(dll_path), debugstr_a(symbol_path), cb);
 
    if (!(flags & BIND_NO_UPDATE))
        FIXME("Image modification is not implemented.\n");
    if (flags & ~BIND_NO_UPDATE)
        FIXME("Ignoring flags %#lx.\n", flags);
 
    if (!MapAndLoad(module, dll_path, &image, TRUE, TRUE))
        return FALSE;
 
    if (!(import = ImageDirectoryEntryToData(image.MappedAddress, FALSE, IMAGE_DIRECTORY_ENTRY_IMPORT, &size)))
    {
        UnMapAndLoad(&image);
        return TRUE; /* no imports */
    }
 
    if (image.FileHeader->OptionalHeader.Magic != IMAGE_NT_OPTIONAL_HDR_MAGIC)
    {
        FIXME("Unhandled architecture %#x.\n", image.FileHeader->OptionalHeader.Magic);
        UnMapAndLoad(&image);
        return TRUE;
    }
 
    for (; import->Name && import->FirstThunk; ++import)
    {
        char full_path[MAX_PATH];
        IMAGE_THUNK_DATA *thunk;
        const char *dll_name;
        DWORD thunk_rva;
 
        if (!(dll_name = ImageRvaToVa(image.FileHeader, image.MappedAddress, import->Name, 0)))
        {
            ERR("Failed to get VA for import name RVA %#lx.\n", import->Name);
            continue;
        }
 
        if (cb) cb(BindImportModule, module, dll_name, 0, 0);
 
        if (!SearchPathA(dll_path, dll_name, 0, sizeof(full_path), full_path, 0))
        {
            ERR("Import %s was not found.\n", debugstr_a(dll_path));
            continue;
        }
 
        thunk_rva = import->OriginalFirstThunk ? import->OriginalFirstThunk : import->FirstThunk;
        if (!(thunk = ImageRvaToVa(image.FileHeader, image.MappedAddress, thunk_rva, 0)))
        {
            ERR("Failed to get VA for import thunk RVA %#lx.\n", thunk_rva);
            continue;
        }
 
        for (; thunk->u1.Ordinal; ++thunk)
        {
            if (IMAGE_SNAP_BY_ORDINAL(thunk->u1.Ordinal))
            {
                /* FIXME: We apparently need to subtract the actual module's
                 * ordinal base. */
                FIXME("Ordinal imports are not implemented.\n");
            }
            else
            {
                IMAGE_IMPORT_BY_NAME *name;
 
                if (!(name = ImageRvaToVa(image.FileHeader, image.MappedAddress, thunk->u1.AddressOfData, 0)))
                {
                    ERR("Failed to get VA for name RVA %#Ix.\n", (size_t)thunk->u1.AddressOfData);
                    continue;
                }
 
                if (cb) cb(BindImportProcedure, module, full_path, 0, (ULONG_PTR)name->Name);
            }
        }
    }
 
    UnMapAndLoad(&image);
    return TRUE;
}

Referenced by BindImage(), and test_bind_image_ex().

CalcCheckSum()

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

static

Definition at line 138 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 164 of file modify.c.

{
  PIMAGE_NT_HEADERS header;
  DWORD CalcSum;
  DWORD HdrSum;
 
  TRACE("(%p, %ld, %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(), MapFileAndCheckSumW(), and test_pe_checksum().

MapFileAndCheckSumA()

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

Definition at line 214 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 280 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 346 of file modify.c.

{
  FIXME(
    "(%s, %s, %d, %d, %d, %ld, %p, %p, %p, %p, %ld): 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 365 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 378 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 387 of file modify.c.

{
  FIXME("(%s, %s, %s, %ld): 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 402 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 417 of file modify.c.

{
  FIXME("(%s, %s, %s, %p, %ld): 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

)