mboot.c

Go to the documentation of this file.

/*
 *  FreeLoader
 *  Copyright (C) 1998-2003  Brian Palmer  <brianp@sginet.com>
 *  Copyright (C) 2005       Alex Ionescu  <alex@relsoft.net>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License along
 *  with this program; if not, write to the Free Software Foundation, Inc.,
 *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include <freeldr.h>
#include <elf/elf.h>
#include <elf/reactos.h>
#include <of.h>
#include "ppcmmu/mmu.h"
#include "compat.h"

#include <debug.h>

/* We'll check this to see if we're in OFW land */
extern of_proxy ofproxy;

PVOID KernelMemory = NULL;

/* Bits to shift to convert a Virtual Address into an Offset in the Page Table */
#define PFN_SHIFT 12

/* Bits to shift to convert a Virtual Address into an Offset in the Page Directory */
#define PDE_SHIFT 22
#define PDE_SHIFT_PAE 18

#define STARTUP_BASE                0xC0000000
#define HYPERSPACE_BASE             0xC0400000
#define HYPERSPACE_PAE_BASE         0xC0800000
#define APIC_BASE                   0xFEC00000
#define KPCR_BASE                   0xFF000000

#define LowMemPageTableIndex        0
#define StartupPageTableIndex       (STARTUP_BASE >> 22)
#define HyperspacePageTableIndex    (HYPERSPACE_BASE >> 22)
#define KpcrPageTableIndex          (KPCR_BASE >> 22)
#define ApicPageTableIndex          (APIC_BASE >> 22)

#define BAT_GRANULARITY             (64 * 1024)
#define KernelMemorySize            (8 * 1024 * 1024)
#define XROUNDUP(x,n)               ((((ULONG)x) + ((n) - 1)) & (~((n) - 1)))

#define TAG_MBOOT 'oobM'

char reactos_module_strings[64][256];    // Array to hold module names

/* Load Address of Next Module */
ULONG_PTR NextModuleBase = 0;

/* Currently Opened Module */
PLOADER_MODULE CurrentModule = NULL;

/* Unrelocated Kernel Base in Virtual Memory */
ULONG_PTR KernelBase;

/* Wether PAE is to be used or not */
BOOLEAN PaeModeEnabled;

/* Kernel Entrypoint in Physical Memory */
ULONG_PTR KernelEntryPoint;

/* Dummy to bring in memmove */
PVOID memmove_dummy = memmove;

PLOADER_MODULE
NTAPI
LdrGetModuleObject(PCHAR ModuleName);

NTSTATUS
NTAPI
LdrPEFixupImports(IN PVOID DllBase,
                  IN PCHAR DllName);

VOID PpcInitializeMmu(int max);

/* FUNCTIONS *****************************************************************/

/*++
 * FrLdrStartup
 * INTERNAL
 *
 *     Prepares the system for loading the Kernel.
 *
 * Params:
 *     Magic - Multiboot Magic
 *
 * Returns:
 *     None.
 *
 * Remarks:
 *     None.
 *
 *--*/

typedef void (*KernelEntryFn)( void * );

int MmuPageMiss(int trapCode, ppc_trap_frame_t *trap)
{
    int i;
    printf("TRAP %x\n", trapCode);
    for( i = 0; i < 40; i++ )
    printf("r[%d] %x\n", i, trap->gpr[i]);
    printf("HALT!\n");
    while(1);
}

typedef struct _ppc_map_set_t {
    int mapsize;
    int usecount;
    ppc_map_info_t *info;
} ppc_map_set_t;

extern int mmu_handle;
paddr_t MmuTranslate(paddr_t possibly_virtual)
{
    if (ofproxy)
    {
        /* Openfirmware takes liberties with boot-time memory.
         * if you're in a unitary kernel, it's not as difficult, but since
         * we rely on loading things into virtual space from here, we need
         * to detect the mappings so far.
         */
        int args[2];
        args[0] = possibly_virtual;
        args[1] = 1; /* Marker to tell we want a physical addr */
        return (paddr_t)ofw_callmethod_ret("translate", mmu_handle, 2, args, 3);
    }
    else
    {
        /* Other booters don't remap ram */
        return possibly_virtual;
    }
}

VOID
NTAPI
FrLdrAddPageMapping(ppc_map_set_t *set, int proc, paddr_t phys, vaddr_t virt)
{
    int j;
    paddr_t page = ROUND_DOWN(phys, (1<<PFN_SHIFT));

    if (virt == 0)
        virt = ROUND_DOWN(page, (1<<PFN_SHIFT));
    else
        virt = ROUND_DOWN(virt, (1<<PFN_SHIFT));

    page = MmuTranslate(page);

    //printf("Mapping virt [%x] to phys [%x (from) %x]\n", virt, page, phys);

    for( j = 0; j < set->usecount; j++ )
    {
        if(set->info[j].addr == page) return;
    }

    if (!set->mapsize)
    {
        set->mapsize = 0x80;
        set->info = MmAllocateMemory(0x80 * sizeof(*set->info));
    }
    else if (set->mapsize <= set->usecount)
    {
        ppc_map_info_t *newinfo = MmAllocateMemory(set->mapsize * 2 * sizeof(*set->info));
        memcpy(newinfo, set->info, set->mapsize * sizeof(*set->info));
        MmFreeMemory(set->info);
        set->info = newinfo;
        set->mapsize *= 2;
    }

    set->info[set->usecount].flags = MMU_ALL_RW;
    set->info[set->usecount].proc = proc;
    set->info[set->usecount].addr = virt;
    set->info[set->usecount].phys = page;
    set->usecount++;
}

extern int _start[], _end[];

VOID
NTAPI
FrLdrStartup(ULONG Magic)
{
    ULONG_PTR i, tmp, OldModCount = 0;
    PCHAR ModHeader;
    CHAR ModulesTreated[64] = { 0 };
    ULONG NumberOfEntries = 0, UsedEntries = 0;
    PPAGE_LOOKUP_TABLE_ITEM FreeLdrMap = MmGetMemoryMap(&NumberOfEntries);
    ppc_map_set_t memmap = { };

    printf("FrLdrStartup\n");

    /* Disable EE */
    __asm__("mfmsr %0" : "=r" (tmp));
    tmp &= 0x7fff;
    __asm__("mtmsr %0" : : "r" (tmp));

    while(OldModCount != LoaderBlock.ModsCount)
    {
        printf("Added %d modules last pass\n",
               LoaderBlock.ModsCount - OldModCount);

        OldModCount = LoaderBlock.ModsCount;

        for(i = 0; i < LoaderBlock.ModsCount; i++)
        {
            if (!ModulesTreated[i])
            {
                ModulesTreated[i] = 1;
                ModHeader = ((PCHAR)reactos_modules[i].ModStart);
                if(ModHeader[0] == 'M' && ModHeader[1] == 'Z')
                    LdrPEFixupImports
                        ((PVOID)reactos_modules[i].ModStart,
                         (PCHAR)reactos_modules[i].String);
            }
        }
    }

    printf("Starting mmu\n");

    PpcInitializeMmu(0);

    printf("Allocating vsid 0 (kernel)\n");
    MmuAllocVsid(0, 0xff00);

    /* We'll use vsid 1 for freeldr (expendable) */
    printf("Allocating vsid 1 (freeldr)\n");
    MmuAllocVsid(1, 0xff);

    printf("Mapping Freeldr Code (%x-%x)\n", _start, _end);

    /* Map memory zones */
    /* Freeldr itself */
    for( i = (int)_start;
         i < (int)_end;
         i += (1<<PFN_SHIFT) ) {
        FrLdrAddPageMapping(&memmap, 1, i, 0);
    }

    printf("KernelBase %x\n", KernelBase);

    /* Heap pages -- this gets the entire freeldr heap */
    for( i = 0; i < NumberOfEntries; i++ ) {
        tmp = i<<PFN_SHIFT;
        if (FreeLdrMap[i].PageAllocated == LoaderSystemCode) {
            UsedEntries++;
            if (tmp >= (ULONG)KernelMemory &&
                tmp <  (ULONG)KernelMemory + KernelMemorySize) {
                FrLdrAddPageMapping(&memmap, 0, tmp, KernelBase + tmp - (ULONG)KernelMemory);
            } else {
                FrLdrAddPageMapping(&memmap, 1, tmp, 0);
            }
        }
    }

    MmuMapPage(memmap.info, memmap.usecount);

    printf("Finished Mapping the Freeldr Heap (used %d pages)\n", UsedEntries);

    printf("Setting initial segments\n");
    MmuSetVsid(0, 8, 1);
    MmuSetVsid(8, 16, 0);

    printf("Segments set!\n");

    MmuTurnOn((KernelEntryFn)KernelEntryPoint, &LoaderBlock);

    /* Nothing more */
    while(1);
}

/*++
 * FrLdrSetupPae
 * INTERNAL
 *
 *     Configures PAE on a MP System, and sets the PDBR if it's supported, or if
 *     the system is UP.
 *
 * Params:
 *     Magic - Multiboot Magic
 *
 * Returns:
 *     None.
 *
 * Remarks:
 *     None.
 *
 *--*/
VOID
FASTCALL
FrLdrSetupPae(ULONG Magic)
{
}

/*++
 * FrLdrGetKernelBase
 * INTERNAL
 *
 *     Gets the Kernel Base to use.
 *
 * Params:
 *
 * Returns:
 *     None.
 *
 * Remarks:
 *     Sets both the FreeLdr internal variable as well as the one which
 *     will be used by the Kernel.
 *
 *--*/
VOID
FASTCALL
FrLdrGetKernelBase(VOID)
{
    PCHAR p;

    /* Default kernel base at 2GB */
    KernelBase = 0x80800000;

    /* Set KernelBase */
    LoaderBlock.KernelBase = 0x80000000;

    /* Read Command Line */
    p = (PCHAR)LoaderBlock.CommandLine;
    while ((p = strchr(p, '/')) != NULL) {

        /* Find "/3GB" */
        if (!_strnicmp(p + 1, "3GB", 3)) {

            /* Make sure there's nothing following it */
            if (p[4] == ' ' || p[4] == 0) {

                /* Use 3GB */
                KernelBase = 0xE0000000;
                LoaderBlock.KernelBase = 0xC0000000;
            }
        }

        p++;
    }
}

/*++
 * FrLdrGetPaeMode
 * INTERNAL
 *
 *     Determines whether PAE mode should be enabled or not.
 *
 * Params:
 *     None.
 *
 * Returns:
 *     None.
 *
 * Remarks:
 *     None.
 *
 *--*/
VOID
FASTCALL
FrLdrGetPaeMode(VOID)
{
}

/*++
 * FrLdrSetupPageDirectory
 * INTERNAL
 *
 *     Sets up the ReactOS Startup Page Directory.
 *
 * Params:
 *     None.
 *
 * Returns:
 *     None.
 *
 * Remarks:
 *     We are setting PDEs, but using the equivalent (for our purpose) PTE structure.
 *     As such, please note that PageFrameNumber == PageEntryNumber.
 *
 *--*/
VOID
FASTCALL
FrLdrSetupPageDirectory(VOID)
{
}

/*++
 * FrLdrMapModule
 * INTERNAL
 *
 *     Loads the indicated elf image as PE.  The target will appear to be
 *     a PE image whose ImageBase has ever been KernelAddr.
 *
 * Params:
 *     Image -- File to load
 *     ImageName -- Name of image for the modules list
 *     MemLoadAddr -- Freeldr address of module
 *     KernelAddr -- Kernel address of module
 *--*/
#define ELF_SECTION(n) ((Elf32_Shdr*)(sptr + (n * shsize)))
#define COFF_FIRST_SECTION(h) ((PIMAGE_SECTION_HEADER) ((DWORD)h+FIELD_OFFSET(IMAGE_NT_HEADERS,OptionalHeader)+(SWAPW(((PIMAGE_NT_HEADERS)(h))->FileHeader.SizeOfOptionalHeader))))

BOOLEAN
NTAPI
FrLdrMapModule(FILE *KernelImage, PCHAR ImageName, PCHAR MemLoadAddr, ULONG KernelAddr)
{
    PIMAGE_DOS_HEADER ImageHeader = 0;
    PIMAGE_NT_HEADERS NtHeader = 0;
    PIMAGE_SECTION_HEADER Section;
    ULONG SectionCount;
    ULONG ImageSize;
    INT i, j;
    PLOADER_MODULE ModuleData;
    //int phsize, phnum;
    int shsize, shnum, relsize, SectionAddr = 0;
    PCHAR sptr;
    Elf32_Ehdr ehdr;
    Elf32_Shdr *shdr;
    LARGE_INTEGER Position;
    PSTR TempName;

    TempName = strrchr(ImageName, '\\');
    if(TempName) TempName++; else TempName = (PSTR)ImageName;
    ModuleData = LdrGetModuleObject(TempName);

    if(ModuleData)
    {
    return TRUE;
    }

    if(!KernelAddr)
    KernelAddr = (ULONG)NextModuleBase - (ULONG)KernelMemory + KernelBase;
    if(!MemLoadAddr)
    MemLoadAddr = (PCHAR)NextModuleBase;

    ModuleData = &reactos_modules[LoaderBlock.ModsCount];
    //printf("Loading file (elf at %x)\n", KernelAddr);

    /* Load the first 1024 bytes of the kernel image so we can read the PE header */
    if (ArcRead(KernelImage, &ehdr, sizeof(ehdr), NULL) != ESUCCESS) {

        /* Fail if we couldn't read */
    printf("Couldn't read the elf header\n");
        return FALSE;
    }

    /* Start by getting elf headers */
    //phsize = ehdr.e_phentsize;
    //phnum = ehdr.e_phnum;
    shsize = ehdr.e_shentsize;
    shnum = ehdr.e_shnum;
    sptr = (PCHAR)FrLdrTempAlloc(shnum * shsize, TAG_MBOOT);

    /* Read section headers */
    Position.QuadPart = ehdr.e_shoff;
    ArcSeek(KernelImage, &Position, SeekAbsolute);
    ArcRead(KernelImage, sptr, shsize * shnum, NULL);

    /* Now we'll get the PE Header */
    for( i = 0; i < shnum; i++ )
    {
    shdr = ELF_SECTION(i);
    shdr->sh_addr = 0;

    /* Find the PE Header */
    if (shdr->sh_type == TYPE_PEHEADER)
    {
        Position.QuadPart = shdr->sh_offset;
        ArcSeek(KernelImage, &Position, SeekAbsolute);
        ArcRead(KernelImage, MemLoadAddr, shdr->sh_size, NULL);
        ImageHeader = (PIMAGE_DOS_HEADER)MemLoadAddr;
        NtHeader = (PIMAGE_NT_HEADERS)((PCHAR)MemLoadAddr + SWAPD(ImageHeader->e_lfanew));
#if 0
        printf("NtHeader at %x\n", SWAPD(ImageHeader->e_lfanew));
        printf("SectionAlignment %x\n",
           SWAPD(NtHeader->OptionalHeader.SectionAlignment));
        SectionAddr = ROUND_UP
        (shdr->sh_size, SWAPD(NtHeader->OptionalHeader.SectionAlignment));
        printf("Header ends at %x\n", SectionAddr);
#endif
        break;
    }
    }

    if(i == shnum)
    {
    printf("No peheader section encountered :-(\n");
    return 0;
    }
#if 0
    else
    {
        printf("DOS SIG: %s\n", (PCHAR)MemLoadAddr);
    }
#endif

    /* Save the Image Base */
    NtHeader->OptionalHeader.ImageBase = SWAPD(KernelAddr);

    /* Load the file image */
    Section = COFF_FIRST_SECTION(NtHeader);
    SectionCount = SWAPW(NtHeader->FileHeader.NumberOfSections);

    /* Walk each section */
    for (i=0; i < SectionCount; i++, Section++)
    {
    shdr = ELF_SECTION((SWAPD(Section->PointerToRawData)+1));

    shdr->sh_addr = SectionAddr = SWAPD(Section->VirtualAddress);
    shdr->sh_addr += KernelAddr;

    Section->PointerToRawData = SWAPD((Section->VirtualAddress - KernelAddr));

    if (shdr->sh_type != SHT_NOBITS)
    {
        /* Content area */
        printf("Loading section %d at %x (real: %x:%d)\n", i, KernelAddr + SectionAddr, MemLoadAddr+SectionAddr, shdr->sh_size);
        Position.QuadPart = shdr->sh_offset;
        ArcSeek(KernelImage, &Position, SeekAbsolute);
        ArcRead(KernelImage, MemLoadAddr + SectionAddr, shdr->sh_size, NULL);
    }
    else
    {
        /* Zero it out */
        printf("BSS section %d at %x\n", i, KernelAddr + SectionAddr);
        memset(MemLoadAddr + SectionAddr, 0,
           ROUND_UP(shdr->sh_size,
                SWAPD(NtHeader->OptionalHeader.SectionAlignment)));
        }
    }

    ImageSize = SWAPD(NtHeader->OptionalHeader.SizeOfImage);
    printf("Total image size is %x\n", ImageSize);

    /* Handle relocation sections */
    for (i = 0; i < shnum; i++) {
    Elf32_Rela reloc = { };
    ULONG *Target32;
    USHORT *Target16;
    int numreloc, relstart, targetSection;
    Elf32_Sym symbol;
    PCHAR RelocSection, SymbolSection;

    shdr = ELF_SECTION(i);
    /* Only relocs here */
    if((shdr->sh_type != SHT_REL) &&
       (shdr->sh_type != SHT_RELA)) continue;

    relstart = shdr->sh_offset;
    relsize = shdr->sh_type == SHT_RELA ? 12 : 8;
    numreloc = shdr->sh_size / relsize;
    targetSection = shdr->sh_info;

    if (!ELF_SECTION(targetSection)->sh_addr) continue;

    RelocSection = FrLdrTempAlloc(shdr->sh_size, TAG_MBOOT);
    Position.QuadPart = relstart;
    ArcSeek(KernelImage, &Position, SeekAbsolute);
    ArcRead(KernelImage, RelocSection, shdr->sh_size, NULL);

    /* Get the symbol section */
    shdr = ELF_SECTION(shdr->sh_link);

    SymbolSection = FrLdrTempAlloc(shdr->sh_size, TAG_MBOOT);
    Position.QuadPart = shdr->sh_offset;
    ArcSeek(KernelImage, &Position, SeekAbsolute);
    ArcRead(KernelImage, SymbolSection, shdr->sh_size, NULL);

    for(j = 0; j < numreloc; j++)
    {
        ULONG S,A,P;

        /* Get the reloc */
        memcpy(&reloc, RelocSection + (j * relsize), sizeof(reloc));

        /* Get the symbol */
        memcpy(&symbol, SymbolSection + (ELF32_R_SYM(reloc.r_info) * sizeof(symbol)), sizeof(symbol));

        /* Compute addends */
        S = symbol.st_value + ELF_SECTION(symbol.st_shndx)->sh_addr;
        A = reloc.r_addend;
        P = reloc.r_offset + ELF_SECTION(targetSection)->sh_addr;

#if 0
        printf("Symbol[%d] %d -> %d(%x:%x) -> %x(+%x)@%x\n",
           ELF32_R_TYPE(reloc.r_info),
           ELF32_R_SYM(reloc.r_info),
           symbol.st_shndx,
           ELF_SECTION(symbol.st_shndx)->sh_addr,
           symbol.st_value,
           S,
           A,
           P);
#endif

        Target32 = (ULONG*)(((PCHAR)MemLoadAddr) + (P - KernelAddr));
        Target16 = (USHORT *)Target32;

        switch (ELF32_R_TYPE(reloc.r_info))
        {
        case R_PPC_NONE:
        break;
        case R_PPC_ADDR32:
        *Target32 = S + A;
        break;
        case R_PPC_REL32:
        *Target32 = S + A - P;
        break;
        case R_PPC_UADDR32: /* Special: Treat as RVA */
        *Target32 = S + A - KernelAddr;
        break;
        case R_PPC_ADDR24:
        *Target32 = (ADDR24_MASK & (S+A)) | (*Target32 & ~ADDR24_MASK);
        break;
        case R_PPC_REL24:
        *Target32 = (ADDR24_MASK & (S+A-P)) | (*Target32 & ~ADDR24_MASK);
        break;
        case R_PPC_ADDR16_LO:
        *Target16 = S + A;
        break;
        case R_PPC_ADDR16_HA:
        *Target16 = (S + A + 0x8000) >> 16;
        break;
        default:
        break;
        }

#if 0
        printf("reloc[%d:%x]: (type %x sym %d val %d) off %x add %x (old %x new %x)\n",
           j,
           ((ULONG)Target32) - ((ULONG)MemLoadAddr),
           ELF32_R_TYPE(reloc.r_info),
           ELF32_R_SYM(reloc.r_info),
           symbol.st_value,
           reloc.r_offset, reloc.r_addend,
           x, *Target32);
#endif
    }

    FrLdrTempFree(SymbolSection, TAG_MBOOT);
    FrLdrTempFree(RelocSection, TAG_MBOOT);
    }

    FrLdrTempFree(sptr, TAG_MBOOT);

    ModuleData->ModStart = (ULONG)MemLoadAddr;
    /* Increase the next Load Base */
    NextModuleBase = ROUND_UP((ULONG)MemLoadAddr + ImageSize, PAGE_SIZE);
    ModuleData->ModEnd = NextModuleBase;
    ModuleData->String = (ULONG)MmAllocateMemory(strlen(ImageName)+1);
    strcpy((PCHAR)ModuleData->String, ImageName);
    printf("Module %s (%x-%x) next at %x\n",
       ModuleData->String,
       ModuleData->ModStart,
       ModuleData->ModEnd,
       NextModuleBase);
    LoaderBlock.ModsCount++;

    /* Return Success */
    return TRUE;
}

/*++
 * FrLdrMapKernel
 * INTERNAL
 *
 *     Maps the Kernel into memory, does PE Section Mapping, initializes the
 *     uninitialized data sections, and relocates the image.
 *
 * Params:
 *     KernelImage - FILE Structure representing the ntoskrnl image file.
 *
 * Returns:
 *     TRUE if the Kernel was mapped.
 *
 * Remarks:
 *     None.
 *
 *--*/
BOOLEAN
NTAPI
FrLdrMapKernel(FILE *KernelImage)
{
    /* Get Kernel Base */
    FrLdrGetKernelBase();

    /* Allocate kernel memory */
    KernelMemory = MmAllocateMemory(KernelMemorySize);

    return FrLdrMapModule(KernelImage, "ntoskrnl.exe", KernelMemory, KernelBase);
}

ULONG_PTR
NTAPI
FrLdrLoadModule(FILE *ModuleImage,
                PCSTR ModuleName,
                PULONG ModuleSize)
{
    ARC_STATUS Status;
    FILEINFORMATION FileInfo;
    ULONG LocalModuleSize;
    ULONG_PTR ThisModuleBase = NextModuleBase;
    PLOADER_MODULE ModuleData;
    PSTR NameBuffer;
    PSTR TempName;

    /* Get current module data structure and module name string array */
    ModuleData = &reactos_modules[LoaderBlock.ModsCount];

    /* Get only the Module Name */
    do {

        TempName = strchr(ModuleName, '\\');

        if(TempName) {
            ModuleName = TempName + 1;
        }

    } while(TempName);
    NameBuffer = reactos_module_strings[LoaderBlock.ModsCount];

    /* Get Module Size */
    Status = ArcGetFileInformation(ModuleImage, &FileInfo);
    if (Status != ESUCCESS || FileInfo.EndingAddress.HighPart != 0)
        LocalModuleSize = 0;
    else
        LocalModuleSize = FileInfo.EndingAddress.LowPart;

    /* Fill out Module Data Structure */
    ModuleData->ModStart = NextModuleBase;
    ModuleData->ModEnd = NextModuleBase + LocalModuleSize;

    /* Save name */
    strcpy(NameBuffer, ModuleName);
    ModuleData->String = (ULONG_PTR)NameBuffer;

    /* Load the file image */
    ArcRead(ModuleImage, (PVOID)NextModuleBase, LocalModuleSize, NULL);

    /* Move to next memory block and increase Module Count */
    NextModuleBase = ROUND_UP(ModuleData->ModEnd, PAGE_SIZE);
    LoaderBlock.ModsCount++;

    /* Return Module Size if required */
    if (ModuleSize != NULL) {
        *ModuleSize = LocalModuleSize;
    }

    printf("Module %s (%x-%x) next at %x\n",
       ModuleData->String,
       ModuleData->ModStart,
       ModuleData->ModEnd,
       NextModuleBase);

    return ThisModuleBase;
}

PVOID
NTAPI
FrLdrMapImage(IN FILE *Image, IN PCHAR ShortName, IN ULONG ImageType)
{
    PVOID Result = NULL;

    printf("Loading image %s (type %d)\n", ShortName, ImageType);

    if (ImageType == 1)
    {
        if(FrLdrMapKernel(Image))
            Result = (PVOID)KernelMemory;
    }
    else
    {
        PVOID ModuleBase = (PVOID)NextModuleBase;

        if(FrLdrMapModule(Image, ShortName, 0, 0))
            Result = ModuleBase;
    }
    return Result;
}

ULONG_PTR
NTAPI
FrLdrCreateModule(PCSTR ModuleName)
{
    PLOADER_MODULE ModuleData;
    PSTR NameBuffer;

    /* Get current module data structure and module name string array */
    ModuleData = &reactos_modules[LoaderBlock.ModsCount];
    NameBuffer = reactos_module_strings[LoaderBlock.ModsCount];

    /* Set up the structure */
    ModuleData->ModStart = NextModuleBase;
    ModuleData->ModEnd = -1;

    /* Copy the name */
    strcpy(NameBuffer, ModuleName);
    ModuleData->String = (ULONG_PTR)NameBuffer;

    /* Set the current Module */
    CurrentModule = ModuleData;

    /* Return Module Base Address */
    return(ModuleData->ModStart);
}

BOOLEAN
NTAPI
FrLdrCloseModule(ULONG_PTR ModuleBase,
                 ULONG ModuleSize)
{
    PLOADER_MODULE ModuleData = CurrentModule;

    /* Make sure a module is opened */
    if (ModuleData) {

        /* Make sure this is the right module and that it hasn't been closed */
        if ((ModuleBase == ModuleData->ModStart) && (ModuleData->ModEnd == MAXULONG_PTR)) {

            /* Close the Module */
            ModuleData->ModEnd = ModuleData->ModStart + ModuleSize;

            /* Set the next Module Base and increase the number of modules */
            NextModuleBase = ROUND_UP(ModuleData->ModEnd, PAGE_SIZE);
            LoaderBlock.ModsCount++;

            /* Close the currently opened module */
            CurrentModule = NULL;

            /* Success */
            return(TRUE);
        }
    }

    /* Failure path */
    return(FALSE);
}