bootdata.c File Reference

#include <rtl.h>
#include <debug.h>

Include dependency graph for bootdata.c:

Go to the source code of this file.

Classes
struct	_RTL_BSD_ITEM_TABLE_ENTRY

Macros
#define	NDEBUG

Typedefs
typedef struct _RTL_BSD_ITEM_TABLE_ENTRY	RTL_BSD_ITEM_TABLE_ENTRY

Functions
NTSTATUS NTAPI	RtlCreateBootStatusDataFile (VOID)
NTSTATUS NTAPI	RtlGetSetBootStatusData (_In_ HANDLE FileHandle, _In_ BOOLEAN Read, _In_ RTL_BSD_ITEM_TYPE DataClass, _In_ PVOID Buffer, _In_ ULONG BufferSize, _Out_opt_ PULONG ReturnLength)
NTSTATUS NTAPI	RtlLockBootStatusData (_Out_ PHANDLE FileHandle)
NTSTATUS NTAPI	RtlUnlockBootStatusData (_In_ HANDLE FileHandle)

Variables
PRTL_BSD_DATA	DummyBsd
RTL_BSD_ITEM_TABLE_ENTRY	BsdItemTable [RtlBsdItemMax]

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 13 of file bootdata.c.

Typedef Documentation

RTL_BSD_ITEM_TABLE_ENTRY

typedef struct _RTL_BSD_ITEM_TABLE_ENTRY RTL_BSD_ITEM_TABLE_ENTRY

Function Documentation

RtlCreateBootStatusDataFile()

NTSTATUS NTAPI RtlCreateBootStatusDataFile

(

VOID

)

Definition at line 98 of file bootdata.c.

{
    IO_STATUS_BLOCK IoStatusBlock;
    LARGE_INTEGER AllocationSize;
    LARGE_INTEGER ByteOffset;
    UNICODE_STRING FileName =
        RTL_CONSTANT_STRING(L"\\SystemRoot\\bootstat.dat");
    OBJECT_ATTRIBUTES ObjectAttributes =
        RTL_CONSTANT_OBJECT_ATTRIBUTES(&FileName, OBJ_CASE_INSENSITIVE);
    HANDLE FileHandle;
    NTSTATUS Status;
    RTL_BSD_DATA InitialBsd;
 
    /* Create the boot status data file */
    AllocationSize.QuadPart = 0x800;
    DBG_UNREFERENCED_LOCAL_VARIABLE(AllocationSize);
    Status = ZwCreateFile(&FileHandle,
                          FILE_GENERIC_READ | FILE_GENERIC_WRITE,
                          &ObjectAttributes,
                          &IoStatusBlock,
                          NULL, //&AllocationSize,
                          FILE_ATTRIBUTE_SYSTEM,
                          0,
                          FILE_CREATE,
                          FILE_SYNCHRONOUS_IO_NONALERT,
                          NULL,
                          0);
    if (NT_SUCCESS(Status))
    {
        /* Setup a sane looking initial BSD */
        RtlZeroMemory(&InitialBsd, sizeof(InitialBsd));
        InitialBsd.Version = sizeof(InitialBsd);
        InitialBsd.ProductType = NtProductWinNt;
        InitialBsd.AabEnabled = 1;
        InitialBsd.AabTimeout = 30;
        InitialBsd.LastBootSucceeded = TRUE;
 
        /* Write it to disk */
        ByteOffset.QuadPart = 0;
        Status = ZwWriteFile(FileHandle,
                             NULL,
                             NULL,
                             NULL,
                             &IoStatusBlock,
                             &InitialBsd,
                             sizeof(InitialBsd),
                             &ByteOffset,
                             NULL);
    }
 
    /* Close the file */
    ZwClose(FileHandle);
 
    return Status;
}

Referenced by ItemCompletionThread().

RtlGetSetBootStatusData()

NTSTATUS NTAPI RtlGetSetBootStatusData	(	_In_ HANDLE	FileHandle,
		_In_ BOOLEAN	Read,
		_In_ RTL_BSD_ITEM_TYPE	DataClass,
		_In_ PVOID	Buffer,
		_In_ ULONG	BufferSize,
		_Out_opt_ PULONG	ReturnLength
	)

Definition at line 161 of file bootdata.c.

{
    IO_STATUS_BLOCK IoStatusBlock;
    LARGE_INTEGER ByteOffset;
    NTSTATUS Status;
 
    DPRINT("RtlGetSetBootStatusData (%p %u %d %p %lu %p)\n",
           FileHandle, Read, DataClass, Buffer, BufferSize, ReturnLength);
 
    if (DataClass >= RtlBsdItemMax)
    {
        return STATUS_INVALID_PARAMETER;
    }
 
    if (BufferSize > BsdItemTable[DataClass].Size)
    {
        return STATUS_BUFFER_TOO_SMALL;
    }
 
    ByteOffset.HighPart = 0;
    ByteOffset.LowPart = BsdItemTable[DataClass].Offset;
 
    if (Read)
    {
        Status = ZwReadFile(FileHandle,
                            NULL,
                            NULL,
                            NULL,
                            &IoStatusBlock,
                            Buffer,
                            BufferSize,
                            &ByteOffset,
                            NULL);
    }
    else
    {
        Status = ZwWriteFile(FileHandle,
                             NULL,
                             NULL,
                             NULL,
                             &IoStatusBlock,
                             Buffer,
                             BufferSize,
                             &ByteOffset,
                             NULL);
    }
 
    if (NT_SUCCESS(Status))
    {
        if (ReturnLength)
        {
            *ReturnLength = BsdItemTable[DataClass].Size;
        }
    }
 
    return Status;
}

Referenced by SmpRestoreBootStatusData(), and SmpSaveAndClearBootStatusData().

RtlLockBootStatusData()

NTSTATUS NTAPI RtlLockBootStatusData

(

_Out_ PHANDLE

FileHandle

)

Definition at line 231 of file bootdata.c.

{
    UNICODE_STRING FileName =
        RTL_CONSTANT_STRING(L"\\SystemRoot\\bootstat.dat");
    OBJECT_ATTRIBUTES ObjectAttributes =
        RTL_CONSTANT_OBJECT_ATTRIBUTES(&FileName, OBJ_CASE_INSENSITIVE);
    HANDLE LocalFileHandle;
    NTSTATUS Status;
    IO_STATUS_BLOCK IoStatusBlock;
 
    /* Intialize the file handle */
    *FileHandle = NULL;
 
    /* Open the boot status data file */
    Status = ZwOpenFile(&LocalFileHandle,
                        FILE_ALL_ACCESS,
                        &ObjectAttributes,
                        &IoStatusBlock,
                        0,
                        FILE_SYNCHRONOUS_IO_NONALERT);
    if (NT_SUCCESS(Status))
    {
        /* Return the file handle */
        *FileHandle = LocalFileHandle;
    }
 
    return Status;
}

Referenced by SmpRestoreBootStatusData(), and SmpSaveAndClearBootStatusData().

RtlUnlockBootStatusData()

NTSTATUS NTAPI RtlUnlockBootStatusData

(

_In_ HANDLE

FileHandle

)

Definition at line 267 of file bootdata.c.

{
    IO_STATUS_BLOCK IoStatusBlock;
 
    /* Flush the file and close it */
    ZwFlushBuffersFile(FileHandle, &IoStatusBlock);
    return ZwClose(FileHandle);
}

Referenced by SmpRestoreBootStatusData(), and SmpSaveAndClearBootStatusData().

Variable Documentation

BsdItemTable

RTL_BSD_ITEM_TABLE_ENTRY BsdItemTable[RtlBsdItemMax]

Definition at line 25 of file bootdata.c.

Referenced by RtlGetSetBootStatusData().

DummyBsd

PRTL_BSD_DATA DummyBsd

Definition at line 24 of file bootdata.c.