#include "precomp.h"
#include "bootcode.h"
#include <debug.h>

Include dependency graph for bootcode.c:

Macros
#define	NDEBUG

Functions
NTSTATUS	ReadBootCodeByHandle (IN OUT PBOOTCODE BootCodeInfo, IN HANDLE FileHandle, IN ULONG Length OPTIONAL)

NTSTATUS	ReadBootCodeFromFile (IN OUT PBOOTCODE BootCodeInfo, IN PUNICODE_STRING FilePath, IN ULONG Length OPTIONAL)

VOID	FreeBootCode (IN OUT PBOOTCODE BootCodeInfo)

Macro Definition Documentation

◆ NDEBUG

#define NDEBUG

Definition at line 14 of file bootcode.c.

Function Documentation

◆ FreeBootCode()

VOID FreeBootCode ( IN OUT PBOOTCODE BootCodeInfo )

Definition at line 104 of file bootcode.c.

{
    ASSERT(BootCodeInfo);
 
    /* Update the bootcode information */
    if (BootCodeInfo->BootCode)
        RtlFreeHeap(ProcessHeap, 0, BootCodeInfo->BootCode);
    BootCodeInfo->BootCode = NULL;
     BootCodeInfo->Length = 0; 
}

Referenced by InstallBtrfsBootCode(), InstallFat32BootCode(), InstallFatBootCode(), InstallMbrBootCode(), InstallNtfsBootCode(), IsThereAValidBootSector(), and SaveBootSector().

◆ ReadBootCodeByHandle()

NTSTATUS ReadBootCodeByHandle	(	IN OUT PBOOTCODE	BootCodeInfo,
		IN HANDLE	FileHandle,
		IN ULONG Length	OPTIONAL
	)

Definition at line 21 of file bootcode.c.

{
    NTSTATUS Status;
    PVOID BootCode;
    IO_STATUS_BLOCK IoStatusBlock;
    LARGE_INTEGER FileOffset;
 
    ASSERT(BootCodeInfo);
 
    /* Normalize the bootcode length */
    if (Length == 0 || Length == (ULONG)-1)
        Length = SECTORSIZE;
 
    /* Allocate a buffer for the bootcode */
    BootCode = RtlAllocateHeap(ProcessHeap, HEAP_ZERO_MEMORY, Length);
    if (BootCode == NULL)
        return STATUS_INSUFFICIENT_RESOURCES;
 
    /* Read the bootcode from the file into the buffer */
    FileOffset.QuadPart = 0ULL;
    Status = NtReadFile(FileHandle,
                        NULL,
                        NULL,
                        NULL,
                        &IoStatusBlock,
                        BootCode,
                        Length,
                        &FileOffset,
                        NULL);
    if (!NT_SUCCESS(Status))
    {
        RtlFreeHeap(ProcessHeap, 0, BootCode);
        return Status;
    }
 
    /* Update the bootcode information */
    if (BootCodeInfo->BootCode)
        RtlFreeHeap(ProcessHeap, 0, BootCodeInfo->BootCode);
    BootCodeInfo->BootCode = BootCode;
     BootCodeInfo->Length = Length; 
 
    return STATUS_SUCCESS;
}

Referenced by InstallFat32BootCode(), InstallFatBootCode(), InstallMbrBootCode(), InstallNtfsBootCode(), and ReadBootCodeFromFile().

◆ ReadBootCodeFromFile()

NTSTATUS ReadBootCodeFromFile	(	IN OUT PBOOTCODE	BootCodeInfo,
		IN PUNICODE_STRING	FilePath,
		IN ULONG Length	OPTIONAL
	)

Definition at line 69 of file bootcode.c.

{
    NTSTATUS Status;
    OBJECT_ATTRIBUTES ObjectAttributes;
    IO_STATUS_BLOCK IoStatusBlock;
    HANDLE FileHandle;
 
    ASSERT(BootCodeInfo);
 
    /* Open the file */
    InitializeObjectAttributes(&ObjectAttributes,
                               FilePath,
                               OBJ_CASE_INSENSITIVE,
                               NULL,
                               NULL);
    Status = NtOpenFile(&FileHandle,
                        GENERIC_READ | SYNCHRONIZE,
                        &ObjectAttributes,
                        &IoStatusBlock,
                        FILE_SHARE_READ | FILE_SHARE_WRITE, // Is FILE_SHARE_WRITE necessary?
                        FILE_SYNCHRONOUS_IO_NONALERT);
    if (!NT_SUCCESS(Status))
        return Status;
 
    Status = ReadBootCodeByHandle(BootCodeInfo, FileHandle, Length);
 
    /* Close the file and return */
    NtClose(FileHandle);
    return Status;
}