pe.c File Reference

#include <ntifs.h>
#include <ndk/ntndk.h>
#include <reactos/rossym.h>
#include "rossympriv.h"
#include <ntimage.h>
#include <debug.h>
#include "dwarf.h"
#include "pe.h"

Include dependency graph for pe.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Functions
PeSect *	pesection (Pe *pe, const char *name)
u16int	peget2 (const unsigned char *ptr)
u32int	peget4 (const unsigned char *ptr)
u64int	peget8 (const unsigned char *ptr)
int	readn (void *filectx, char *buffer, ulong size)
int	seek (void *filectx, ulong position, int origin)
static int	readblock (void *fd, DwarfBlock *b, ulong off, ulong len)
int	loaddisksection (Pe *pe, char *name, DwarfBlock *b)
int	loadmemsection (Pe *pe, char *name, DwarfBlock *b)
void *	RosSymAllocMemZero (ulong size, ulong count)
int	GetStrnlen (const char *string, int maxlen)
void	pefree (Pe *pe)
void	xfree (void *v)
ulong	pefindrva (struct _IMAGE_SECTION_HEADER *SectionHeaders, int NumberOfSections, ulong TargetPhysical)

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 8 of file pe.c.

Function Documentation

GetStrnlen()

int GetStrnlen	(	const char *	string,
		int	maxlen
	)

Definition at line 103 of file pe.c.

                                               {
    int i;
    for (i = 0; i < maxlen && string[i]; i++);
    return i;
}

Referenced by RosSymCreateFromFile(), and RosSymCreateFromMem().

loaddisksection()

int loaddisksection	(	Pe *	pe,
		char *	name,
		DwarfBlock *	b
	)

Definition at line 72 of file pe.c.

{
    PeSect *s;
    if((s = pesection(pe, name)) == nil)
        return -1;
    return readblock(pe->fd, b, s->PointerToRawData, s->SizeOfRawData);
}

Referenced by RosSymCreateFromFile().

loadmemsection()

int loadmemsection	(	Pe *	pe,
		char *	name,
		DwarfBlock *	b
	)

Definition at line 81 of file pe.c.

{
    PeSect *s;
 
    if((s = pesection(pe, name)) == nil)
        return -1;
    DPRINT("Loading section %s (ImageBase %x RVA %x)\n", name, pe->fd, s->VirtualAddress);
    b->data = RosSymAllocMem(s->SizeOfRawData);
    b->len = s->SizeOfRawData;
    PCHAR DataSource = ((char *)pe->fd) + s->VirtualAddress;
    DPRINT("Copying to %x from %x (%x)\n", DataSource, b->data, b->len);
    RtlCopyMemory(b->data, DataSource, s->SizeOfRawData);
 
    return s->SizeOfRawData;
}

Referenced by RosSymCreateFromMem().

pefindrva()

ulong pefindrva	(	struct _IMAGE_SECTION_HEADER *	SectionHeaders,
		int	NumberOfSections,
		ulong	TargetPhysical
	)

Definition at line 126 of file pe.c.

                                                                                                          {
    int i;
    DPRINT("Finding RVA for Physical %x\n", TargetPhysical);
    for (i = 0; i < NumberOfSections; i++) {
        DPRINT("Section %d name %s Raw %x Virt %x\n",
               i,
               ANSI_NAME_STRING(&SectionHeaders[i])->Buffer,
               SectionHeaders[i].PointerToRawData,
               SectionHeaders[i].VirtualAddress);
        if (TargetPhysical >= SectionHeaders[i].PointerToRawData &&
            TargetPhysical < SectionHeaders[i].PointerToRawData + SectionHeaders[i].SizeOfRawData) {
            DPRINT("RVA %x\n", TargetPhysical - SectionHeaders[i].PointerToRawData + SectionHeaders[i].VirtualAddress);
            return TargetPhysical - SectionHeaders[i].PointerToRawData + SectionHeaders[i].VirtualAddress;
        }
    }
    return nil;
}

Referenced by RosSymCreateFromMem().

pefree()

void pefree

(

Pe *

pe

)

Definition at line 109 of file pe.c.

                    {
    int i;
    for (i = 0; i < pe->nsections; i++) {
        RtlFreeAnsiString(ANSI_NAME_STRING(&pe->sect[i]));
    }
    for (i = 0; i < pe->nsymbols; i++) {
        free(pe->symtab[i].name);
    }
    free(pe->symtab);
    free(pe->sect);
    free(pe);
}

Referenced by dwarfclose().

peget2()

u16int peget2

(

const unsigned char *

ptr

)

Definition at line 35 of file pe.c.

                                        {
    return *((u16int*)ptr);
}

Referenced by RosSymCreateFromFile(), and RosSymCreateFromMem().

peget4()

u32int peget4

(

const unsigned char *

ptr

)

Definition at line 39 of file pe.c.

                                        {
    return *((u32int*)ptr);
}

Referenced by RosSymCreateFromFile(), and RosSymCreateFromMem().

peget8()

u64int peget8

(

const unsigned char *

ptr

)

Definition at line 43 of file pe.c.

                                        {
    return *((u64int*)ptr);
}

Referenced by RosSymCreateFromFile(), and RosSymCreateFromMem().

pesection()

PeSect * pesection	(	Pe *	pe,
		const char *	name
	)

Definition at line 15 of file pe.c.

{
    int i;
    ANSI_STRING WantName;
    RtlInitAnsiString(&WantName, name);
    DPRINT("Searching for section %s\n", name);
    for (i = 0; i < pe->nsections; i++) {
        PANSI_STRING AnsiString = ANSI_NAME_STRING(&pe->sect[i]);
        if (WantName.Length == AnsiString->Length &&
            !memcmp(AnsiString->Buffer, name, WantName.Length)) {
            DPRINT("Found %s (%d) @ %x (%x)\n", name, i,
                   ((PCHAR)pe->imagebase)+pe->sect[i].VirtualAddress,
                   pe->sect[i].SizeOfRawData);
            return &pe->sect[i];
        }
    }
    DPRINT("%s not found\n", name);
    return nil;
}

Referenced by loaddisksection(), and loadmemsection().

readblock()

static int readblock ( void *  fd, DwarfBlock *  b, ulong  off, ulong  len  )

static

Definition at line 57 of file pe.c.

{
    b->data = malloc(len);
    if(b->data == nil)
        return -1;
    if(!seek(fd, off, 0) || !readn(fd, (char *)b->data, len)){
        free(b->data);
        b->data = nil;
        return -1;
    }
    b->len = len;
    return 0;
}

Referenced by loaddisksection().

readn()

int readn	(	void *	filectx,
		char *	buffer,
		ulong	size
	)

Definition at line 47 of file pe.c.

                                                   {
    return RosSymReadFile(filectx, buffer, size);
}

Referenced by readblock().

RosSymAllocMemZero()

void * RosSymAllocMemZero	(	ulong	size,
		ulong	count
	)

Definition at line 97 of file pe.c.

                                                  {
    void *res = RosSymAllocMem(size * count);
    if (res) memset(res, 0, size * count);
    return res;
}

seek()

int seek	(	void *	filectx,
		ulong	position,
		int	origin
	)

Definition at line 51 of file pe.c.

                                                    {
    assert(origin == 0);
    return RosSymSeekFile(filectx, position);
}

Referenced by readblock().

xfree()

void xfree

(

void *

v

)

Definition at line 122 of file pe.c.

                    {
    if (v) RosSymFreeMem(v);
}