5 Aug
2006
5 Aug
'06
4:29 p.m.
Author: ion
Date: Sat Aug 5 20:29:40 2006
New Revision: 23485
URL: http://svn.reactos.org/svn/reactos?rev=23485&view=rev
Log:
- Add cmhive.c now that the code mostly makes sense. (requires changes in cmlib to
compile).
Added:
branches/alex-cm-branch/reactos/ntoskrnl/cm/cmhive.c
Modified:
branches/alex-cm-branch/reactos/ntoskrnl/cm/cm.h
Modified: branches/alex-cm-branch/reactos/ntoskrnl/cm/cm.h
URL:
http://svn.reactos.org/svn/reactos/branches/alex-cm-branch/reactos/ntoskrnl…
==============================================================================
--- branches/alex-cm-branch/reactos/ntoskrnl/cm/cm.h (original)
+++ branches/alex-cm-branch/reactos/ntoskrnl/cm/cm.h Sat Aug 5 20:29:40 2006
@@ -6,7 +6,7 @@
* PROGRAMMERS: Alex Ionescu (alex.ionescu(a)reactos.org)
*/
-//#include "cmlib.h"
+#include "cmlib.h"
//
// Define this if you want debugging support
@@ -66,6 +66,13 @@
#define HFILE_TYPE_LOG 2
#define HFILE_TYPE_EXTERNAL 3
#define HFILE_TYPE_MAX 4
+
+//
+// Hive sizes
+//
+#define HBLOCK_SIZE 0x1000
+#define HSECTOR_SIZE 0x200
+#define HSECTOR_COUNT 8
//
// CM_KEY_CONTROL_BLOCK Flags
Added: branches/alex-cm-branch/reactos/ntoskrnl/cm/cmhive.c
URL:
http://svn.reactos.org/svn/reactos/branches/alex-cm-branch/reactos/ntoskrnl…
==============================================================================
--- branches/alex-cm-branch/reactos/ntoskrnl/cm/cmhive.c (added)
+++ branches/alex-cm-branch/reactos/ntoskrnl/cm/cmhive.c Sat Aug 5 20:29:40 2006
@@ -1,0 +1,325 @@
+/*
+ * PROJECT: ReactOS Kernel
+ * LICENSE: GPL - See COPYING in the top level directory
+ * FILE: ntoskrnl/cm/cmhive.c
+ * PURPOSE: Routines for managing Registry Hives in general
+ * PROGRAMMERS: Alex Ionescu (alex.ionescu(a)reactos.org)
+ */
+
+/* INCLUDES ******************************************************************/
+
+#include <ntoskrnl.h>
+#define NDEBUG
+#include <debug.h>
+
+/* GLOBALS *******************************************************************/
+
+EX_PUSH_LOCK CmpHiveListHeadLock;
+LIST_ENTRY CmpHiveListHead;
+
+/* FUNCTIONS *****************************************************************/
+
+PVOID
+NTAPI
+CmpAllocate(IN ULONG Size,
+ IN BOOLEAN Paged)
+{
+ /* FIXME: TODO */
+ DbgBreakPoint();
+ return NULL;
+}
+
+VOID
+NTAPI
+CmpFree(IN PVOID Block)
+{
+ /* FIXME: TODO */
+ DbgBreakPoint();
+}
+
+BOOLEAN
+NTAPI
+CmpFileRead(IN PHHIVE Hive,
+ IN ULONG FileType,
+ IN ULONG FileOffset,
+ IN PVOID Buffer,
+ IN ULONG BufferLength)
+{
+ /* FIXME: TODO */
+ DbgBreakPoint();
+ return FALSE;
+}
+
+BOOLEAN
+NTAPI
+CmpFileWrite(IN PHHIVE Hive,
+ IN ULONG FileType,
+ IN ULONG FileOffset,
+ IN PVOID Buffer,
+ IN ULONG BufferLength)
+{
+ /* FIXME: TODO */
+ DbgBreakPoint();
+ return FALSE;
+}
+
+BOOLEAN
+NTAPI
+CmpFileSetSize(IN PHHIVE Hive,
+ IN ULONG FileType,
+ IN ULONG FileSize)
+{
+ /* FIXME: TODO */
+ DbgBreakPoint();
+ return FALSE;
+}
+
+BOOLEAN
+NTAPI
+CmpFileFlush(IN PHHIVE Hive,
+ IN ULONG FileType)
+{
+ /* FIXME: TODO */
+ DbgBreakPoint();
+ return FALSE;
+}
+
+VOID
+NTAPI
+CmpInitializeSecurityCache(IN PCMHIVE Hive)
+{
+ ULONG i;
+ PAGED_CODE();
+
+ /* Initialize defaults */
+ Hive->SecurityCache = NULL;
+ Hive->SecurityCount = 0;
+ Hive->SecurityHitHint = -1;
+ Hive->SecurityCacheSize = 0;
+
+ /* Loop the lists */
+ for (i = 0; i < CMP_SECURITY_HASH_LISTS; i++)
+ {
+ /* Initialize this list */
+ InitializeListHead(&Hive->SecurityHash[i]);
+ }
+}
+
+VOID
+NTAPI
+CmpInitializeHiveViewList(IN PCMHIVE Hive)
+{
+ PAGED_CODE();
+
+ /* Initialize the lists */
+ InitializeListHead(&Hive->PinViewListHead);
+ InitializeListHead(&Hive->LRUViewListHead);
+
+ /* Setup defaults */
+ Hive->MappedViews = 0;
+ Hive->PinnedViews = 0;
+ Hive->UseCount = 0;
+}
+
+NTSTATUS
+NTAPI
+CmpInitializeHive(OUT PCMHIVE *CmHive,
+ IN ULONG Operation,
+ IN ULONG Flags,
+ IN ULONG FileType,
+ IN PVOID Context,
+ IN HANDLE Primary,
+ IN HANDLE Alternate,
+ IN HANDLE Log,
+ IN HANDLE External,
+ IN PUNICODE_STRING FileName)
+{
+ PCMHIVE Hive;
+ IO_STATUS_BLOCK IoStatusBlock;
+ FILE_FS_SIZE_INFORMATION FileSizeInformation;
+ NTSTATUS Status;
+ ULONG Cluster;
+
+ /*
+ * The following are invalid:
+ * An alternate hive that's also a log file.
+ * An external hive that is also internal.
+ * An alternate hive or a log hive that's not a primary hive too.
+ * A volatile hive that's linked to permanent storage.
+ * An in-memory initailization without context data.
+ * A log hive or alternative hive that's not linked to a correct file type.
+ */
+ if ((Alternate && Log) ||
+ (External && (Primary || Alternate || Log)) ||
+ (Alternate && !Primary) ||
+ (Log && !Primary) ||
+ ((Flags & HIVE_VOLATILE) &&
+ (Alternate || Primary || External || Log)) ||
+ ((Operation == HINIT_MEMORY) && (!Context)) ||
+ (Log && (FileType != HFILE_TYPE_LOG)) ||
+ (Alternate && (FileType != HFILE_TYPE_ALTERNATE)))
+ {
+ /* Fail the request */
+ return STATUS_INVALID_PARAMETER;
+ }
+
+ /* Check if this is a primary hive */
+ if (Primary)
+ {
+ /* Get the cluster size */
+ Status = ZwQueryVolumeInformationFile(Primary,
+ &IoStatusBlock,
+ &FileSizeInformation,
+ sizeof(FILE_FS_SIZE_INFORMATION),
+ FileFsSizeInformation);
+ if (!NT_SUCCESS(Status)) return Status;
+
+ /* Make sure it's not larger then the block size */
+ if (FileSizeInformation.BytesPerSector > HBLOCK_SIZE)
+ {
+ /* Fail */
+ return STATUS_REGISTRY_IO_FAILED;
+ }
+
+ /* Otherwise, calculate the cluster */
+ Cluster = FileSizeInformation.BytesPerSector / HSECTOR_SIZE;
+ Cluster = max(1, Cluster);
+ }
+ else
+ {
+ /* Otherwise use cluster 1 */
+ Cluster = 1;
+ }
+
+ /* Allocate a hive */
+ Hive = ExAllocatePoolWithTag(PagedPool,
+ sizeof(CMHIVE),
+ TAG_CM);
+ if (!Hive) return STATUS_INSUFFICIENT_RESOURCES;
+
+ /* Setup null fields */
+ Hive->UnloadEvent = NULL;
+ Hive->RootKcb = NULL;
+ Hive->Frozen = FALSE;
+ Hive->UnloadWorkItem = NULL;
+ Hive->GrowOnlyMode = FALSE;
+ Hive->GrowOffset = 0;
+ Hive->CellRemapArray = NULL;
+ Hive->UseCountLog.Next = 0;
+ Hive->LockHiveLog.Next = 0;
+ Hive->FileObject = NULL;
+ Hive->NotifyList.Flink = NULL;
+ Hive->NotifyList.Blink = NULL;
+
+ /* Set loading flag */
+ Hive->HiveIsLoading = TRUE;
+
+ /* Set the current thread as creator */
+ Hive->CreatorOwner = KeGetCurrentThread();
+
+ /* Initialize lists */
+ InitializeListHead(&Hive->KcbConvertListHead);
+ InitializeListHead(&Hive->KnodeConvertListHead);
+ InitializeListHead(&Hive->TrustClassEntry);
+
+ /* Allocate the view log */
+ Hive->ViewLock = ExAllocatePoolWithTag(NonPagedPool,
+ sizeof(KGUARDED_MUTEX),
+ TAG_CM);
+ if (!Hive->ViewLock) return STATUS_INSUFFICIENT_RESOURCES;
+
+ /* Allocate the flush lock */
+ Hive->FlusherLock = ExAllocatePoolWithTag(NonPagedPool,
+ sizeof(ERESOURCE),
+ TAG_CM);
+ if (!Hive->FlusherLock) return STATUS_INSUFFICIENT_RESOURCES;
+
+ /* Setup the handles */
+ Hive->FileHandles[HFILE_TYPE_PRIMARY] = Primary;
+ Hive->FileHandles[HFILE_TYPE_ALTERNATE] = Alternate;
+ Hive->FileHandles[HFILE_TYPE_LOG] = Log;
+ Hive->FileHandles[HFILE_TYPE_EXTERNAL] = External;
+
+ /* Initailize the guarded mutex */
+ KeInitializeGuardedMutex(Hive->ViewLock);
+ Hive->ViewLockOwner = NULL;
+
+ /* Initialize the flush lock */
+ ExInitializeResourceLite(Hive->FlusherLock);
+
+ /* Setup hive locks */
+ ExInitializePushLock(&Hive->HiveLock);
+ Hive->HiveLockOwner = NULL;
+ ExInitializePushLock(&Hive->WriterLock);
+ Hive->WriterLockOwner = NULL;
+ ExInitializePushLock(&Hive->SecurityLock);
+ Hive->HiveSecurityLockOwner = NULL;
+
+ /* Clear file names */
+ RtlInitEmptyUnicodeString(&Hive->FileUserName, NULL, 0);
+ RtlInitEmptyUnicodeString(&Hive->FileFullPath, NULL, 0);
+
+ /* Initialize the view list */
+ CmpInitializeHiveViewList(Hive);
+
+ /* Initailize the security cache */
+ CmpInitializeSecurityCache(Hive);
+
+ /* Setup flags */
+ Hive->Flags = 0;
+ Hive->FlushCount = 0;
+
+ /* Initialize the hive */
+ Status = HvInitialize(&Hive->Hive,
+ Operation,
+ Flags,
+ FileType,
+ Context,
+ CmpAllocate,
+ CmpFree,
+ CmpFileRead,
+ CmpFileWrite,
+ CmpFileSetSize,
+ CmpFileFlush,
+ Cluster,
+ FileName);
+ if (NT_SUCCESS(Status))
+ {
+ /* Free all alocations */
+ ExFreePool(Hive->ViewLock);
+ ExFreePool(Hive->FlusherLock);
+ ExFreePool(Hive);
+ return Status;
+ }
+
+ /* Check if we should verify the registry */
+ if ((Operation == HINIT_FILE) ||
+ (Operation == HINIT_MEMORY) ||
+ (Operation == HINIT_MEMORY_INPLACE) ||
+ (Operation == HINIT_MAPFILE))
+ {
+ /* Verify integrity */
+ if (CmCheckRegistry(Hive, TRUE))
+ {
+ /* Free all alocations */
+ ExFreePool(Hive->ViewLock);
+ ExFreePool(Hive->FlusherLock);
+ ExFreePool(Hive);
+ return STATUS_REGISTRY_CORRUPT;
+ }
+ }
+
+ /* Lock the hive list */
+ ExAcquirePushLockExclusive(&CmpHiveListHeadLock);
+
+ /* Insert this hive */
+ InsertHeadList(&CmpHiveListHead, &Hive->HiveList);
+
+ /* Release the lock */
+ ExReleasePushLock(&CmpHiveListHeadLock);
+
+ /* Return the hive and success */
+ *CmHive = Hive;
+ return STATUS_SUCCESS;
+}
+