mirror of
https://github.com/reactos/reactos.git
synced 2024-10-31 03:48:17 +00:00
8d3e80e437
[AUTOCHK] Add also support for scanning FATX volumes. The Format(), FormatEx(), Chkdsk(), ChkdskEx() functions exposed by the U*.DLL user-mode FS library dlls are different (and have different prototypes) than the similarly-named functions exported by FMIFS.DLL . In particular, what we used to call "xxxChkdskEx()" and "xxxFormatEx()" in our U*.DLL libraries actually correspond more, from their arguments, to the "Chkdsk()" and "Format()" functions in Windows' U*.DLL . Their *Ex() counterparts instead take most of the parameters through a structure passed by pointer. On FMIFS.DLL side, while FMIFS!Chkdsk() calls U*.DLL!Chkdsk() and FMIFS!ChkdskEx() calls U*.DLL!ChkdskEx() (and we do not implement these *Ex() functions at the moment), both FMIFS!Format() and FMIFS!FormatEx() call U*.DLL!Format() instead, while FMIFS!FormatEx2() calls U*.DLL!FormatEx() (that we do not implement yet either) !! To improve that, refactor the calls to these U*.DLL functions so as to respect the more compatible prototypes: They contain the correct number of parameters in a compatible order. However, some of the parameters do not have the same types yet: the strings are kept here in PUNICODE_STRINGS, while on Windows they are passed via an undocumented DSTRING struct, and the FMIFS callback is instead a MESSAGE struct/class on Windows. Finally, the MEDIA_TYPE parameter in U*.DLL!Format() is equivalent, yet not fully 100% in 1-to-1 correspondence, with the FMIFS_MEDIA_FLAG used in the corresponding FMIFS.DLL functions. One thing to notice is that the U*.DLL!Format() (and the Ex) functions support a BOOLEAN (a flag resp.) for telling that a backwards-compatible FS version should be used instead of the (default) latest FS version. This is used e.g. by the FAT FS, where by default FAT32 is selected (depending also on other constraints like, the disk and the partition sizes), unless that bit is set in which case, FAT16 (or 12) is used.
192 lines
4.7 KiB
C
192 lines
4.7 KiB
C
/*
|
|
* COPYRIGHT: See COPYING in the top level directory
|
|
* PROJECT: File Management IFS Utility functions
|
|
* FILE: reactos/dll/win32/fmifs/init.c
|
|
* PURPOSE: Initialisation
|
|
*
|
|
* PROGRAMMERS: Emanuele Aliberti
|
|
* Hervé Poussineau (hpoussin@reactos.org)
|
|
*/
|
|
|
|
#include "precomp.h"
|
|
|
|
#include <winreg.h>
|
|
|
|
#define NTOS_MODE_USER
|
|
#include <ndk/cmfuncs.h>
|
|
#include <ndk/obfuncs.h>
|
|
|
|
static BOOLEAN FmIfsInitialized = FALSE;
|
|
LIST_ENTRY ProviderListHead;
|
|
|
|
PIFS_PROVIDER
|
|
GetProvider(
|
|
IN PWCHAR FileSystem)
|
|
{
|
|
PLIST_ENTRY ListEntry;
|
|
PIFS_PROVIDER Provider;
|
|
|
|
ListEntry = ProviderListHead.Flink;
|
|
while (ListEntry != &ProviderListHead)
|
|
{
|
|
Provider = CONTAINING_RECORD(ListEntry, IFS_PROVIDER, ListEntry);
|
|
if (_wcsicmp(Provider->Name, FileSystem) == 0)
|
|
return Provider;
|
|
ListEntry = ListEntry->Flink;
|
|
}
|
|
|
|
/* Provider not found */
|
|
return NULL;
|
|
}
|
|
|
|
|
|
static
|
|
BOOLEAN
|
|
AddProvider(
|
|
IN PCUNICODE_STRING FileSystem,
|
|
IN PWCHAR DllFile)
|
|
{
|
|
PIFS_PROVIDER Provider = NULL;
|
|
ULONG RequiredSize;
|
|
HMODULE hMod = NULL;
|
|
BOOLEAN ret = FALSE;
|
|
|
|
hMod = LoadLibraryW(DllFile);
|
|
if (!hMod)
|
|
goto cleanup;
|
|
|
|
RequiredSize = FIELD_OFFSET(IFS_PROVIDER, Name)
|
|
+ FileSystem->Length + sizeof(UNICODE_NULL);
|
|
Provider = (PIFS_PROVIDER)RtlAllocateHeap(
|
|
RtlGetProcessHeap(),
|
|
0,
|
|
RequiredSize);
|
|
if (!Provider)
|
|
goto cleanup;
|
|
RtlZeroMemory(Provider, RequiredSize);
|
|
|
|
/* Get function pointers */
|
|
Provider->Chkdsk = (PULIB_CHKDSK)GetProcAddress(hMod, "Chkdsk");
|
|
//Provider->ChkdskEx = (PULIB_CHKDSKEX)GetProcAddress(hMod, "ChkdskEx");
|
|
//Provider->Extend = (PULIB_EXTEND)GetProcAddress(hMod, "Extend");
|
|
Provider->Format = (PULIB_FORMAT)GetProcAddress(hMod, "Format");
|
|
//Provider->FormatEx = (PULIB_FORMATEX)GetProcAddress(hMod, "FormatEx");
|
|
|
|
RtlCopyMemory(Provider->Name, FileSystem->Buffer, FileSystem->Length);
|
|
|
|
InsertTailList(&ProviderListHead, &Provider->ListEntry);
|
|
ret = TRUE;
|
|
|
|
cleanup:
|
|
if (!ret)
|
|
{
|
|
if (hMod)
|
|
FreeLibrary(hMod);
|
|
if (Provider)
|
|
RtlFreeHeap(RtlGetProcessHeap(), 0, Provider);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
static
|
|
BOOLEAN
|
|
InitializeFmIfsOnce(VOID)
|
|
{
|
|
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;
|
|
}
|
|
|
|
/* FMIFS.8 */
|
|
BOOLEAN
|
|
NTAPI
|
|
InitializeFmIfs(
|
|
IN PVOID hinstDll,
|
|
IN DWORD dwReason,
|
|
IN PVOID reserved)
|
|
{
|
|
switch (dwReason)
|
|
{
|
|
case DLL_PROCESS_ATTACH:
|
|
if (FmIfsInitialized == FALSE)
|
|
{
|
|
if (InitializeFmIfsOnce() == FALSE)
|
|
return FALSE;
|
|
|
|
FmIfsInitialized = TRUE;
|
|
}
|
|
break;
|
|
|
|
case DLL_THREAD_ATTACH:
|
|
break;
|
|
|
|
case DLL_THREAD_DETACH:
|
|
break;
|
|
|
|
case DLL_PROCESS_DETACH:
|
|
break;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
/* EOF */
|