PIFS_PROVIDER
GetProvider(
- IN PWCHAR FileSystem)
+ 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;
+ 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
+
+static
+BOOLEAN
AddProvider(
- IN PCUNICODE_STRING FileSystem,
- IN PWCHAR DllFile)
+ 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->ChkdskEx = (CHKDSKEX)GetProcAddress(hMod, "ChkdskEx");
- //Provider->Extend = (EXTEND)GetProcAddress(hMod, "Extend");
- Provider->FormatEx = (FORMATEX)GetProcAddress(hMod, "FormatEx");
-
- RtlCopyMemory(Provider->Name, FileSystem->Buffer, FileSystem->Length);
-
- InsertTailList(&ProviderListHead, &Provider->ListEntry);
- ret = TRUE;
+ 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->ChkdskEx = (CHKDSKEX)GetProcAddress(hMod, "ChkdskEx");
+ //Provider->Extend = (EXTEND)GetProcAddress(hMod, "Extend");
+ Provider->FormatEx = (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;
+ if (!ret)
+ {
+ if (hMod)
+ FreeLibrary(hMod);
+ if (Provider)
+ RtlFreeHeap(RtlGetProcessHeap(), 0, Provider);
+ }
+ return ret;
}
-static BOOLEAN
-InitializeFmIfsOnce(void)
+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;
+ 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
+BOOLEAN
+NTAPI
InitializeFmIfs(
- IN PVOID hinstDll,
- IN DWORD dwReason,
- IN PVOID reserved)
+ IN PVOID hinstDll,
+ IN DWORD dwReason,
+ IN PVOID reserved)
{
- switch (dwReason)
- {
- case DLL_PROCESS_ATTACH:
- if (FALSE == FmIfsInitialized)
- {
- if (FALSE == InitializeFmIfsOnce())
- {
- return FALSE;
- }
-
- FmIfsInitialized = TRUE;
- }
- break;
-
- case DLL_THREAD_ATTACH:
- break;
-
- case DLL_THREAD_DETACH:
- break;
-
- case DLL_PROCESS_DETACH:
- break;
- }
-
- return TRUE;
+ 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 */
projects / reactos.git / blobdiff