Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-12-28 18:15:11 +00:00
Code Issues Projects Releases Packages Wiki Activity

Added some error checks.

Browse source 
svn path=/trunk/; revision=1973
This commit is contained in:
Eric Kohl 2001-06-16 09:49:45 +00:00
parent 9d04225d93
commit 9d72cf785d
1 changed files with 26 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
reactos/ntoskrnl/cm/regfile.c
View file

@ -87,18 +87,18 @@ NTSTATUS CmiCreateNewRegFile( HANDLE FileHandle )
RootKeyBlock = (PKEY_BLOCK) (tBuf+REG_BLOCK_SIZE+REG_HEAP_BLOCK_DATA_OFFSET); RootKeyBlock = (PKEY_BLOCK) (tBuf+REG_BLOCK_SIZE+REG_HEAP_BLOCK_DATA_OFFSET);
FreeSubBlock = (PFREE_SUB_BLOCK) (tBuf+REG_BLOCK_SIZE+REG_HEAP_BLOCK_DATA_OFFSET+sizeof(KEY_BLOCK)); FreeSubBlock = (PFREE_SUB_BLOCK) (tBuf+REG_BLOCK_SIZE+REG_HEAP_BLOCK_DATA_OFFSET+sizeof(KEY_BLOCK));
CmiCreateDefaultHeaderBlock(HeaderBlock); CmiCreateDefaultHeaderBlock(HeaderBlock);
CmiCreateDefaultHeapBlock(HeapBlock); CmiCreateDefaultHeapBlock(HeapBlock);
CmiCreateDefaultRootKeyBlock(RootKeyBlock); CmiCreateDefaultRootKeyBlock(RootKeyBlock);
HeapBlock->BlockOffset = 0; //First block. HeapBlock->BlockOffset = 0; //First block.
HeaderBlock->RootKeyBlock = REG_HEAP_BLOCK_DATA_OFFSET; //Offset to root key block. HeaderBlock->RootKeyBlock = REG_HEAP_BLOCK_DATA_OFFSET; //Offset to root key block.
//The rest of the block is free //The rest of the block is free
FreeSubBlock->SubBlockSize = REG_BLOCK_SIZE-(REG_HEAP_BLOCK_DATA_OFFSET+sizeof(KEY_BLOCK)); FreeSubBlock->SubBlockSize = REG_BLOCK_SIZE-(REG_HEAP_BLOCK_DATA_OFFSET+sizeof(KEY_BLOCK));
Status = ZwWriteFile( FileHandle, NULL, NULL, NULL, Status = ZwWriteFile( FileHandle, NULL, NULL, NULL,
&IoStatusBlock, tBuf, 2*REG_BLOCK_SIZE, 0, NULL ); &IoStatusBlock, tBuf, 2*REG_BLOCK_SIZE, 0, NULL );
ExFreePool( tBuf ); ExFreePool( tBuf );
return Status; return Status;
} }
@ -115,8 +115,12 @@ CmiCreateRegistry(PWSTR Filename)
DWORD FreeOffset; DWORD FreeOffset;
int i, j; int i, j;
BLOCK_OFFSET BlockOffset; BLOCK_OFFSET BlockOffset;
DPRINT1("CmiCreateRegistry() Filename '%S'\n", Filename);
RegistryFile = ExAllocatePool(NonPagedPool, sizeof(REGISTRY_FILE)); RegistryFile = ExAllocatePool(NonPagedPool, sizeof(REGISTRY_FILE));
CHECKPOINT1;
if (RegistryFile == NULL)
return NULL;
CHECKPOINT1;
if (Filename != NULL) if (Filename != NULL)
{ {
UNICODE_STRING TmpFileName; UNICODE_STRING TmpFileName;
@ -136,7 +140,7 @@ CmiCreateRegistry(PWSTR Filename)
NULL, NULL,
NULL); NULL);
Status = NtCreateFile( &FileHandle, Status = NtCreateFile( &FileHandle,
FILE_ALL_ACCESS, FILE_ALL_ACCESS,
&ObjectAttributes, &ObjectAttributes,
&IoSB, &IoSB,
@ -176,8 +180,21 @@ CmiCreateRegistry(PWSTR Filename)
RegistryFile->HeaderBlock, RegistryFile->HeaderBlock,
sizeof(HEADER_BLOCK), sizeof(HEADER_BLOCK),
&fileOffset, 0); &fileOffset, 0);
ZwQueryInformationFile(FileHandle,&IoSB,&fsi if (!NT_SUCCESS(Status))
{
ExFreePool(RegistryFile->Filename);
RegistryFile->Filename = NULL;
return NULL;
}
Status = ZwQueryInformationFile(FileHandle,&IoSB,&fsi
,sizeof(fsi),FileStandardInformation); ,sizeof(fsi),FileStandardInformation);
if (!NT_SUCCESS(Status))
{
ExFreePool(RegistryFile->Filename);
RegistryFile->Filename = NULL;
return NULL;
}
RegistryFile->FileSize = fsi.EndOfFile.u.LowPart; RegistryFile->FileSize = fsi.EndOfFile.u.LowPart;
RegistryFile->BlockListSize = RegistryFile->FileSize / 4096 -1; RegistryFile->BlockListSize = RegistryFile->FileSize / 4096 -1;
// RegistryFile->NumberOfBlocks = RegistryFile->BlockListSize; // RegistryFile->NumberOfBlocks = RegistryFile->BlockListSize;
@ -196,7 +213,7 @@ CmiCreateRegistry(PWSTR Filename)
ZwClose(FileHandle); ZwClose(FileHandle);
return NULL; return NULL;
} }
Status = ZwReadFile(FileHandle, Status = ZwReadFile(FileHandle,
0, 0, 0, 0, 0, 0, 0, 0,
RegistryFile->BlockList [0], RegistryFile->BlockList [0],
RegistryFile->FileSize-4096, RegistryFile->FileSize-4096,
@ -269,7 +286,7 @@ CmiCreateRegistry(PWSTR Filename)
return RegistryFile; return RegistryFile;
} }
ULONG ULONG
CmiGetMaxNameLength(PREGISTRY_FILE RegistryFile, CmiGetMaxNameLength(PREGISTRY_FILE RegistryFile,
PKEY_BLOCK KeyBlock) PKEY_BLOCK KeyBlock)
{ {