cmwraper.c File Reference

#include "edithive.h"
#include "nturtl.h"
#include "stdlib.h"
#include "stdio.h"

Go to the source code of this file.

Functions
ULONG	DbgPrint (IN PCH Format,...)
NTSTATUS	MyCmpInitHiveFromFile (IN PUNICODE_STRING FileName, OUT PCMHIVE *CmHive, OUT PBOOLEAN Allocate)
VOID	CmpLazyFlush (VOID)
VOID	CmpFreeSecurityDescriptor (IN PHHIVE Hive, IN HCELL_INDEX Cell)
VOID	CmpReportNotify (UNICODE_STRING Name, PHHIVE Hive, HCELL_INDEX Cell, ULONG Filter)
VOID	CmpLockRegistry (VOID)
BOOLEAN	CmpTryLockRegistryExclusive (IN BOOLEAN CanWait)
VOID	CmpUnlockRegistry ()
BOOLEAN	CmpTestRegistryLock ()
BOOLEAN	CmpTestRegistryLockExclusive ()
LONG	KeReleaseMutex (IN PKMUTEX Mutex, IN BOOLEAN Wait)
NTSTATUS	KeWaitForSingleObject (IN PVOID Object, IN KWAIT_REASON WaitReason, IN KPROCESSOR_MODE WaitMode, IN BOOLEAN Alertable, IN PLARGE_INTEGER Timeout OPTIONAL)
BOOLEAN	CmpValidateHiveSecurityDescriptors (IN PHHIVE Hive)
VOID	KeBugCheck (IN ULONG BugCheckCode)
VOID	KeBugCheckEx (IN ULONG BugCheckCode, IN ULONG Arg1, IN ULONG Arg2, IN ULONG Arg3, IN ULONG Arg4)
VOID	KeQuerySystemTime (OUT PLARGE_INTEGER SystemTime)
PVOID	ExAllocatePool (IN POOL_TYPE PoolType, IN ULONG NumberOfBytes)
VOID	ExFreePool (IN PVOID P)
NTSTATUS	CmpWorkerCommand (IN OUT PREGISTRY_COMMAND Command)
NTSTATUS	CmpLinkHiveToMaster (PUNICODE_STRING LinkName, HANDLE RootDirectory, PCMHIVE CmHive, BOOLEAN Allocate, PSECURITY_DESCRIPTOR SecurityDescriptor)
BOOLEAN	CmpFileSetSize (PHHIVE Hive, ULONG FileType, ULONG FileSize)
Variables
ULONG	UsedStorage
CCHAR	KiFindFirstSetRight [256]
ULONG	MmSizeOfPagedPoolInBytes = 0xffffffff
GENERIC_MAPPING	CmpKeyMapping
BOOLEAN	CmpNoWrite = FALSE
ULONG	CmLogLevel = 0
ULONG	CmLogSelect = 0
PCMHIVE	CmpMasterHive = NULL
LIST_ENTRY	CmpHiveListHead

---

Function Documentation

BOOLEAN CmpFileSetSize (  PHHIVE  Hive, ULONG  FileType, ULONG  FileSize )

Definition at line 509 of file cmwraper.c. References CML_MAJOR, CMLOG, CmpDoFileSetSize(), CMS_IO_ERROR, FALSE, Hive, NT_SUCCESS, NTSTATUS(), and TRUE. Referenced by CmpInitializeHive(), and CmpInitializeHiveList(). 00516 : 00517 00518 This routine sets the size of a file. It must not return until 00519 the size is guaranteed, therefore, it does a flush. 00520 00521 It is environment specific. 00522 00523 This routine will force execution to the correct thread context. 00524 00525 Arguments: 00526 00527 Hive - Hive we are doing I/O for 00528 00529 FileType - which supporting file to use 00530 00531 FileSize - 32 bit value to set the file's size to 00532 00533 Return Value: 00534 00535 FALSE if failure 00536 TRUE if success 00537 00538 --*/ 00539 { 00540 NTSTATUS status; 00541 00542 status = CmpDoFileSetSize(Hive, FileType, FileSize); 00543 if (!NT_SUCCESS(status)) { 00544 CMLOG(CML_MAJOR, CMS_IO_ERROR) { 00545 KdPrint(("CmpFileSetSize:\n\t")); 00546 KdPrint(("Failure: status = %08lx ", status)); 00547 } 00548 return FALSE; 00549 } 00550 00551 return TRUE; 00552 } }

VOID CmpFreeSecurityDescriptor (  IN PHHIVE  Hive, IN HCELL_INDEX  Cell )

Definition at line 105 of file cmwraper.c. Referenced by CmpFreeKeyByCell(), and CmpFreeKeyValues(). { return; }

VOID CmpLazyFlush (  VOID   )

Definition at line 97 of file cmwraper.c. Referenced by CmpInitializeHiveList(), CmpLazyFlushWorker(), and HvMarkDirty(). { }

NTSTATUS CmpLinkHiveToMaster (  PUNICODE_STRING  LinkName, HANDLE  RootDirectory, PCMHIVE  CmHive, BOOLEAN  Allocate, PSECURITY_DESCRIPTOR  SecurityDescriptor )

Definition at line 496 of file cmwraper.c. Referenced by CmInitSystem1(), CmLoadKey(), CmpInitializeHiveList(), CmpInitializeSystemHive(), and CmpLoadHiveVolatile(). { return( STATUS_SUCCESS ); }

VOID CmpLockRegistry (  VOID   )

Definition at line 124 of file cmwraper.c. Referenced by CmEnumerateKey(), CmEnumerateValueKey(), CmpCloseKeyObject(), CmpDeleteKeyObject(), CmpLazyFlushWorker(), CmpQueryKeyName(), CmpSecurityMethod(), CmQueryKey(), CmQueryMultipleValueKey(), CmQueryValueKey(), CmSetValueKey(), NtFlushKey(), NtNotifyChangeMultipleKeys(), and NtOpenKey(). { return; }

VOID CmpReportNotify (  UNICODE_STRING  Name, PHHIVE  Hive, HCELL_INDEX  Cell, ULONG  Filter )

Definition at line 114 of file cmwraper.c. Referenced by CmDeleteKey(), CmDeleteValueKey(), CmpDoCreateChild(), CmpLinkHiveToMaster(), CmpSecurityMethod(), CmRestoreKey(), CmSetValueKey(), and NtUnloadKey(). { }

BOOLEAN CmpTestRegistryLock (   )

Definition at line 144 of file cmwraper.c. References TRUE. { return TRUE; }

BOOLEAN CmpTestRegistryLockExclusive (   )

Definition at line 150 of file cmwraper.c. References TRUE. { return TRUE; }

BOOLEAN CmpTryLockRegistryExclusive (  IN BOOLEAN  CanWait  )

Definition at line 130 of file cmwraper.c. References TRUE. { return TRUE; }

VOID CmpUnlockRegistry (   )

Definition at line 138 of file cmwraper.c. Referenced by CmDeleteKey(), CmDeleteValueKey(), CmEnumerateKey(), CmEnumerateValueKey(), CmGetSystemDriverList(), CmInitSystem1(), CmLoadKey(), CmNotifyRunDown(), CmpCloneControlSet(), CmpCloneHwProfile(), CmpCloseKeyObject(), CmpDeleteKeyObject(), CmpFileSetSize(), CmpLazyFlushWorker(), CmpLoadHiveVolatile(), CmpQueryKeyName(), CmpRefreshHive(), CmpSaveBootControlSet(), CmpSaveKeyByFileCopy(), CmpSecurityMethod(), CmQueryKey(), CmQueryMultipleValueKey(), CmQueryValueKey(), CmReplaceKey(), CmRestoreKey(), CmSaveKey(), CmSaveMergedKeys(), CmSetLastWriteTimeKey(), CmSetValueKey(), CmShutdownSystem(), CmUnloadKey(), NtCreateKey(), NtFlushKey(), NtInitializeRegistry(), NtNotifyChangeMultipleKeys(), NtOpenKey(), NtQueryOpenSubKeys(), and NtUnloadKey(). { }

BOOLEAN CmpValidateHiveSecurityDescriptors (  IN PHHIVE  Hive  )

Definition at line 175 of file cmwraper.c. References _HHIVE::BaseBlock, _CM_KEY_SECURITY::Blink, CML_FLOW, CML_MAJOR, CML_MINOR, CMLOG, CmpDumpSecurityDescriptor, CMS_SEC, _CM_KEY_SECURITY::Descriptor, _CM_KEY_SECURITY::DescriptorLength, FALSE, _CM_KEY_SECURITY::Flink, HCELL_INDEX, Hive, HvGetCell, HvIsCellAllocated(), PCM_KEY_SECURITY, _CM_KEY_SECURITY::ReferenceCount, _HBASE_BLOCK::RootCell, _CM_KEY_NODE::Security, SetUsed, SeValidSecurityDescriptor(), and TRUE. Referenced by CmCheckRegistry(). { PCM_KEY_NODE RootNode; PCM_KEY_SECURITY SecurityCell; HCELL_INDEX ListAnchor; HCELL_INDEX NextCell; HCELL_INDEX LastCell; BOOLEAN ValidHive = TRUE; CMLOG(CML_FLOW, CMS_SEC) { KdPrint(("CmpValidateHiveSecurityDescriptor: Hive = %lx\n",(ULONG)Hive)); } RootNode = (PCM_KEY_NODE) HvGetCell(Hive, Hive->BaseBlock->RootCell); ListAnchor = NextCell = RootNode->u1.s1.Security; do { SecurityCell = (PCM_KEY_SECURITY) HvGetCell(Hive, NextCell); if (NextCell != ListAnchor) { // // Check to make sure that our Blink points to where we just // came from. // if (SecurityCell->Blink != LastCell) { CMLOG(CML_MAJOR, CMS_SEC) { KdPrint((" Invalid Blink (%ld) on security cell %ld\n",SecurityCell->Blink, NextCell)); KdPrint((" should point to %ld\n", LastCell)); } ValidHive = FALSE; } } CMLOG(CML_MINOR, CMS_SEC) { KdPrint(("CmpValidSD: SD shared by %d nodes\n",SecurityCell->ReferenceCount)); } // SetUsed(Hive, NextCell); LastCell = NextCell; NextCell = SecurityCell->Flink; } while ( NextCell != ListAnchor ); return(TRUE); }

NTSTATUS CmpWorkerCommand (  IN OUT PREGISTRY_COMMAND  Command  )

Definition at line 323 of file cmwraper.c. References CmFlushKey(), CmpDoFileSetSize(), _CMHIVE::FileHandles, FileName, HFILE_TYPE_MAX, MyCmpInitHiveFromFile(), NtClose(), NULL, REG_CMD_FILE_SET_SIZE, REG_CMD_FLUSH_KEY, REG_CMD_HIVE_CLOSE, REG_CMD_HIVE_OPEN, and REG_CMD_SHUTDOWN. : This routine just encapsulates all the necessary synchronization for sending a command to the worker thread. Arguments: Command - Supplies a pointer to an initialized REGISTRY_COMMAND structure which will be copied into the global communication structure. Return Value: NTSTATUS = Command.Status --*/ { PCMHIVE CmHive; PUNICODE_STRING FileName; ULONG i; switch (Command->Command) { case REG_CMD_FLUSH_KEY: return CmFlushKey(Command->Hive, Command->Cell); break; case REG_CMD_FILE_SET_SIZE: return CmpDoFileSetSize( Command->Hive, Command->FileType, Command->FileSize ); break; case REG_CMD_HIVE_OPEN: // // Open the file. // FileName = Command->FileAttributes->ObjectName; return MyCmpInitHiveFromFile(FileName, &Command->CmHive, &Command->Allocate); break; case REG_CMD_HIVE_CLOSE: // // Close the files associated with this hive. // CmHive = Command->CmHive; for (i=0; i<HFILE_TYPE_MAX; i++) { if (CmHive->FileHandles[i] != NULL) { NtClose(CmHive->FileHandles[i]); } } return STATUS_SUCCESS; break; case REG_CMD_SHUTDOWN: // // shut down the registry // break; default: return STATUS_INVALID_PARAMETER; } }

ULONG DbgPrint (  IN PCH  Format,   ... )

Definition at line 50 of file cmwraper.c. References Buffer. { va_list arglist; UCHAR Buffer[512]; STRING Output; // // Format the output into a buffer and then print it. // va_start(arglist, Format); Output.Length = _vsnprintf(Buffer, sizeof(Buffer), Format, arglist); Output.Buffer = Buffer; printf("%s", Buffer); return 0; }

PVOID ExAllocatePool (  IN POOL_TYPE  PoolType, IN ULONG  NumberOfBytes )

Definition at line 278 of file cmwraper.c. References HBLOCK_SIZE, NT_SUCCESS, NtAllocateVirtualMemory(), NTSTATUS(), NULL, ROUND_UP, and Size. { PVOID Address = NULL; ULONG Size; NTSTATUS status; Size = ROUND_UP(NumberOfBytes, HBLOCK_SIZE); status = NtAllocateVirtualMemory( NtCurrentProcess(), &Address, 0, &Size, MEM_COMMIT, PAGE_READWRITE ); if (!NT_SUCCESS(status)) { return NULL; } return Address; }

VOID ExFreePool (  IN PVOID  P  )

Definition at line 304 of file cmwraper.c. Referenced by ExInitializeNPagedLookasideList(), and ExInitializePagedLookasideList(). { ULONG size; size = HBLOCK_SIZE; // if it was really more than 1 page, well, too bad NtFreeVirtualMemory( NtCurrentProcess(), &P, &size, MEM_DECOMMIT ); return; }

VOID KeBugCheck (  IN ULONG  BugCheckCode  )

Definition at line 218 of file cmwraper.c. Referenced by KiInitializeKernel(). { printf("BugCheck: code = %08lx\n", BugCheckCode); exit(1); }

VOID KeBugCheckEx (  IN ULONG  BugCheckCode, IN ULONG  Arg1, IN ULONG  Arg2, IN ULONG  Arg3, IN ULONG  Arg4 )

Definition at line 227 of file cmwraper.c. References exit. { printf("BugCheck: code = %08lx\n", BugCheckCode); printf("Args =%08lx %08lx %08lx %08lx\n", Arg1, Arg2, Arg3, Arg4); exit(1); }

VOID KeQuerySystemTime (  OUT PLARGE_INTEGER  SystemTime  )

Definition at line 242 of file cmwraper.c. { NtQuerySystemTime(SystemTime); }

LONG KeReleaseMutex (  IN PKMUTEX  Mutex, IN BOOLEAN  Wait )

Definition at line 155 of file cmwraper.c. References PKMUTEX. { return(0); }

NTSTATUS KeWaitForSingleObject (  IN PVOID  Object, IN KWAIT_REASON  WaitReason, IN KPROCESSOR_MODE  WaitMode, IN BOOLEAN  Alertable, IN PLARGE_INTEGER Timeout  OPTIONAL )

Definition at line 163 of file cmwraper.c. { return(STATUS_SUCCESS); }

NTSTATUS MyCmpInitHiveFromFile (  IN PUNICODE_STRING  FileName, OUT PCMHIVE *  CmHive, OUT PBOOLEAN  Allocate )

Definition at line 405 of file cmwraper.c. References CmpInitializeHive(), CmpOpenHiveFiles(), FALSE, FileName, HFILE_TYPE_LOG, HFILE_TYPE_PRIMARY, HINIT_CREATE, HINIT_FILE, L, NT_SUCCESS, NtClose(), NTSTATUS(), NULL, Status, and TRUE. Referenced by CmpWorkerCommand(). : This routine opens a file and log, allocates a CMHIVE, and initializes it. Arguments: FileName - Supplies name of file to be loaded. CmHive - Returns pointer to initialized hive (if successful) Allocate - Returns whether the hive was allocated or existing. Return Value: NTSTATUS --*/ { PCMHIVE NewHive; ULONG Disposition; ULONG SecondaryDisposition; HANDLE PrimaryHandle; HANDLE LogHandle; NTSTATUS Status; ULONG FileType; ULONG Operation; BOOLEAN Success; *CmHive = NULL; Status = CmpOpenHiveFiles(FileName, L".log", &PrimaryHandle, &LogHandle, &Disposition, &SecondaryDisposition, TRUE, NULL ); if (!NT_SUCCESS(Status)) { return(Status); } if (LogHandle == NULL) { FileType = HFILE_TYPE_PRIMARY; } else { FileType = HFILE_TYPE_LOG; } if (Disposition == FILE_CREATED) { Operation = HINIT_CREATE; *Allocate = TRUE; } else { Operation = HINIT_FILE; *Allocate = FALSE; } Success = CmpInitializeHive(&NewHive, Operation, FALSE, FileType, NULL, PrimaryHandle, NULL, LogHandle, NULL, NULL); if (!Success) { NtClose(PrimaryHandle); if (LogHandle != NULL) { NtClose(LogHandle); } return(STATUS_REGISTRY_CORRUPT); } else { *CmHive = NewHive; return(STATUS_SUCCESS); } }

---

Variable Documentation

ULONG CmLogLevel = 0

Definition at line 83 of file cmwraper.c.

ULONG CmLogSelect = 0

Definition at line 84 of file cmwraper.c.

LIST_ENTRY CmpHiveListHead

Definition at line 86 of file cmwraper.c.

GENERIC_MAPPING CmpKeyMapping

Initial value: { KEY_READ, KEY_WRITE, KEY_EXECUTE, KEY_ALL_ACCESS } Definition at line 76 of file cmwraper.c. Referenced by CmpCreateObjectTypes(), and EhpAttachSecurity().

PCMHIVE CmpMasterHive = NULL

Definition at line 85 of file cmwraper.c.

BOOLEAN CmpNoWrite = FALSE

Definition at line 82 of file cmwraper.c.

CCHAR KiFindFirstSetRight[256]

Initial value: { 0, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0, 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0, 5, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0, 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0, 6, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0, 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0, 5, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0, 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0, 7, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0, 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0, 5, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0, 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0, 6, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0, 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0, 5, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0, 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0} Definition at line 28 of file cmwraper.c.

ULONG MmSizeOfPagedPoolInBytes = 0xffffffff

Definition at line 47 of file cmwraper.c.

ULONG UsedStorage

Definition at line 26 of file cmwraper.c. Referenced by CmCheckRegistry().

---