init.c File Reference

#include "precomp.h"
#include <winreg.h>
#include <ndk/cmfuncs.h>
#include <ndk/obfuncs.h>
#include <debug.h>

Include dependency graph for init.c:

Go to the source code of this file.

Macros
#define	NTOS_MODE_USER
#define	NDEBUG

Functions
PIFS_PROVIDER	LoadProvider (IN PWCHAR FileSystem)
BOOLEAN	UnloadProvider (PIFS_PROVIDER Provider)
static BOOLEAN	AddProvider (IN PCUNICODE_STRING FileSystem, IN PWCHAR DllFile)
static BOOLEAN	InitializeFmIfsOnce (VOID)
static VOID	UninitializeFmIfsOnce (VOID)
BOOLEAN NTAPI	InitializeFmIfs (IN PVOID hinstDll, IN DWORD dwReason, IN PVOID reserved)

Variables
static ULONG	FmIfsAttached = 0
LIST_ENTRY	ProviderListHead

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 19 of file init.c.

NTOS_MODE_USER

#define NTOS_MODE_USER

Definition at line 15 of file init.c.

Function Documentation

AddProvider()

static BOOLEAN AddProvider ( IN PCUNICODE_STRING  FileSystem, IN PWCHAR  DllFile  )

static

Definition at line 89 of file init.c.

{
    PIFS_PROVIDER Provider = NULL;
    ULONG ProcCount = 0;
    HMODULE hMod = NULL;
    BOOLEAN ret = FALSE;
 
    DPRINT("AddProvider(%wZ %S)\n", FileSystem, DllFile);
 
    hMod = LoadLibraryW(DllFile);
    if (!hMod)
        goto cleanup;
 
    if (GetProcAddress(hMod, "Chkdsk"))
        ProcCount++;
    if (GetProcAddress(hMod, "ChkdskEx"))
        ProcCount++;
    if (GetProcAddress(hMod, "Extend"))
        ProcCount++;
    if (GetProcAddress(hMod, "Format"))
        ProcCount++;
    if (GetProcAddress(hMod, "FormatEx"))
        ProcCount++;
 
    DPRINT("ProcCount %lu\n", ProcCount);
    if (ProcCount == 0)
        goto cleanup;
 
    Provider = (PIFS_PROVIDER)RtlAllocateHeap(
        RtlGetProcessHeap(),
        HEAP_ZERO_MEMORY,
        sizeof(IFS_PROVIDER));
    if (!Provider)
        goto cleanup;
 
    Provider->Name = (PWSTR)RtlAllocateHeap(RtlGetProcessHeap(),
                                            HEAP_ZERO_MEMORY,
                                            FileSystem->Length + sizeof(UNICODE_NULL));
    if (!Provider->Name)
        goto cleanup;
 
    RtlCopyMemory(Provider->Name, FileSystem->Buffer, FileSystem->Length);
 
    Provider->DllFile = (PWSTR)RtlAllocateHeap(RtlGetProcessHeap(),
                                               HEAP_ZERO_MEMORY,
                                               (wcslen(DllFile) + 1) * sizeof(WCHAR));
    if (!Provider->DllFile)
        goto cleanup;
 
    wcscpy(Provider->DllFile, DllFile);
 
    InsertTailList(&ProviderListHead, &Provider->ListEntry);
    ret = TRUE;
 
    DPRINT("AddProvider success\n");
 
cleanup:
    if (hMod)
        FreeLibrary(hMod);
 
    if (!ret)
    {
        if (Provider)
        {
            if (Provider->Name)
                RtlFreeHeap(RtlGetProcessHeap(), 0, Provider->Name);
            if (Provider->DllFile)
                RtlFreeHeap(RtlGetProcessHeap(), 0, Provider->DllFile);
            RtlFreeHeap(RtlGetProcessHeap(), 0, Provider);
        }
    }
    return ret;
}

Referenced by InitializeFmIfsOnce().

InitializeFmIfs()

BOOLEAN NTAPI InitializeFmIfs	(	IN PVOID	hinstDll,
		IN DWORD	dwReason,
		IN PVOID	reserved
	)

Definition at line 256 of file init.c.

{
    switch (dwReason)
    {
        case DLL_PROCESS_ATTACH:
            if (FmIfsAttached == 0)
            {
                if (InitializeFmIfsOnce() == FALSE)
                    return FALSE;
 
                FmIfsAttached++;
            }
            break;
 
        case DLL_THREAD_ATTACH:
            break;
 
        case DLL_THREAD_DETACH:
            break;
 
        case DLL_PROCESS_DETACH:
            FmIfsAttached--;
            if (FmIfsAttached == 0)
                UninitializeFmIfsOnce();
            break;
    }
 
    return TRUE;
}

InitializeFmIfsOnce()

static BOOLEAN InitializeFmIfsOnce ( VOID  )

static

Definition at line 167 of file init.c.

{
    OBJECT_ATTRIBUTES ObjectAttributes;
    UNICODE_STRING RegistryPath
        = RTL_CONSTANT_STRING(L"\\REGISTRY\\Machine\\SOFTWARE\\ReactOS\\ReactOS\\CurrentVersion\\IFS");
    HANDLE hKey = NULL;
    PKEY_VALUE_FULL_INFORMATION Buffer;
    ULONG BufferSize = sizeof(KEY_VALUE_FULL_INFORMATION) + MAX_PATH;
    ULONG RequiredSize;
    ULONG i = 0;
    UNICODE_STRING Name;
    UNICODE_STRING Data;
    NTSTATUS Status;
 
    InitializeListHead(&ProviderListHead);
 
    /* Read IFS providers from HKLM\SOFTWARE\ReactOS\ReactOS\CurrentVersion\IFS */
    InitializeObjectAttributes(&ObjectAttributes, &RegistryPath, 0, NULL, NULL);
    Status = NtOpenKey(&hKey, KEY_QUERY_VALUE, &ObjectAttributes);
    if (Status == STATUS_OBJECT_NAME_NOT_FOUND)
        return TRUE;
    else if (!NT_SUCCESS(Status))
        return FALSE;
 
    Buffer = (PKEY_VALUE_FULL_INFORMATION)RtlAllocateHeap(
        RtlGetProcessHeap(),
        0,
        BufferSize);
    if (!Buffer)
    {
        NtClose(hKey);
        return FALSE;
    }
 
    while (TRUE)
    {
        Status = NtEnumerateValueKey(
            hKey,
            i++,
            KeyValueFullInformation,
            Buffer,
            BufferSize,
            &RequiredSize);
        if (Status == STATUS_BUFFER_OVERFLOW)
            continue;
        else if (!NT_SUCCESS(Status))
            break;
        else if (Buffer->Type != REG_SZ)
            continue;
 
        Name.Length = Name.MaximumLength = Buffer->NameLength;
        Name.Buffer = Buffer->Name;
        Data.Length = Data.MaximumLength = Buffer->DataLength;
        Data.Buffer = (PWCHAR)((ULONG_PTR)Buffer + Buffer->DataOffset);
        if (Data.Length > sizeof(WCHAR) && Data.Buffer[Data.Length / sizeof(WCHAR) - 1] == UNICODE_NULL)
            Data.Length -= sizeof(WCHAR);
 
        AddProvider(&Name, Data.Buffer);
    }
 
    NtClose(hKey);
    RtlFreeHeap(RtlGetProcessHeap(), 0, Buffer);
    return TRUE;
}

Referenced by InitializeFmIfs().

LoadProvider()

PIFS_PROVIDER LoadProvider

(

IN PWCHAR

FileSystem

)

Definition at line 26 of file init.c.

{
    PLIST_ENTRY ListEntry;
    PIFS_PROVIDER Provider;
 
    DPRINT("LoadProvider(%S)\n", FileSystem);
 
    ListEntry = ProviderListHead.Flink;
    while (ListEntry != &ProviderListHead)
    {
        Provider = CONTAINING_RECORD(ListEntry, IFS_PROVIDER, ListEntry);
        if (_wcsicmp(Provider->Name, FileSystem) == 0)
        {
            DPRINT("Found it!\n");
            if (!Provider->hModule)
            {
                Provider->hModule = LoadLibraryW(Provider->DllFile);
                if (!Provider->hModule)
                {
                    DPRINT("Dll Loading failed!\n");
                    return NULL;
                }
 
                /* Get function pointers */
                Provider->Chkdsk = (PULIB_CHKDSK)GetProcAddress(Provider->hModule, "Chkdsk");
                //Provider->ChkdskEx = (PULIB_CHKDSKEX)GetProcAddress(Provider->hModule, "ChkdskEx");
                //Provider->Extend = (PULIB_EXTEND)GetProcAddress(Provider->hModule, "Extend");
                Provider->Format = (PULIB_FORMAT)GetProcAddress(Provider->hModule, "Format");
                //Provider->FormatEx = (PULIB_FORMATEX)GetProcAddress(Provider->hModule, "FormatEx");
            }
            return Provider;
        }
        ListEntry = ListEntry->Flink;
    }
 
    /* Provider not found */
    return NULL;
}

Referenced by Chkdsk(), and FormatEx().

UninitializeFmIfsOnce()

static VOID UninitializeFmIfsOnce ( VOID  )

static

Definition at line 234 of file init.c.

{
    PLIST_ENTRY ListEntry;
    PIFS_PROVIDER Provider;
 
    while (!IsListEmpty(&ProviderListHead))
    {
        ListEntry = RemoveTailList(&ProviderListHead);
        Provider = CONTAINING_RECORD(ListEntry, IFS_PROVIDER, ListEntry);
        if (Provider->Name)
            RtlFreeHeap(RtlGetProcessHeap(), 0, Provider->Name);
        if (Provider->DllFile)
            RtlFreeHeap(RtlGetProcessHeap(), 0, Provider->DllFile);
        if (Provider->hModule)
            FreeLibrary(Provider->hModule);
        RtlFreeHeap(RtlGetProcessHeap(), 0, Provider);
    }
}

Referenced by InitializeFmIfs().

UnloadProvider()

BOOLEAN UnloadProvider

(

PIFS_PROVIDER

Provider

)

Definition at line 67 of file init.c.

{
    DPRINT("UnloadProvider(%S)\n", Provider->Name);
 
    if (Provider->hModule)
    {
        FreeLibrary(Provider->hModule);
        Provider->hModule = NULL;
 
        Provider->Chkdsk =  NULL;
        Provider->ChkdskEx = NULL;
        Provider->Extend =  NULL;
        Provider->Format =  NULL;
        Provider->FormatEx = NULL;
    }
 
    return TRUE;
}

Referenced by Chkdsk(), and FormatEx().

Variable Documentation

FmIfsAttached

ULONG FmIfsAttached = 0

static

Definition at line 22 of file init.c.

Referenced by InitializeFmIfs().

ProviderListHead

LIST_ENTRY ProviderListHead

Definition at line 23 of file init.c.

Referenced by AddProvider(), InitializeFmIfsOnce(), LoadProvider(), QueryAvailableFileSystemFormat(), and UninitializeFmIfsOnce().