disk.c File Reference

#include <freeldr.h>
#include <debug.h>
#include "part_mbr.h"
#include "part_gpt.h"

Include dependency graph for disk.c:

Go to the source code of this file.

Macros
#define	GUID_FORMAT_STR   "%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x"
#define	GUID_ELEMENTS(Guid)
#define	FIRST_PARTITION   1

Functions
	DBG_DEFAULT_CHANNEL (DISK)
BOOLEAN	DiskReadGptHeader (_In_ UCHAR DriveNumber, _Out_ PGPT_TABLE_HEADER GptHeader)
LONG	DiskReportError (_In_ BOOLEAN bShowError)
VOID	DiskError (_In_ PCSTR ErrorString, _In_ ULONG ErrorCode)
ARC_STATUS	DiskInitialize (_In_ UCHAR DriveNumber, _In_ PCSTR DeviceName, _In_ CONFIGURATION_TYPE DeviceType, _In_ const DEVVTBL *FuncTable, _Out_opt_ PULONG pChecksum, _Out_opt_ PULONG pSignature, _Out_opt_ PBOOLEAN pValidPartitionTable)

Variables
static LONG	lReportError = 0

Macro Definition Documentation

FIRST_PARTITION

#define FIRST_PARTITION   1

Definition at line 31 of file disk.c.

GUID_ELEMENTS

#define GUID_ELEMENTS

(

Guid

)

Value:

    (Guid)->Data1, (Guid)->Data2, (Guid)->Data3, \
    (Guid)->Data4[0], (Guid)->Data4[1], (Guid)->Data4[2], (Guid)->Data4[3], \
    (Guid)->Data4[4], (Guid)->Data4[5], (Guid)->Data4[6], (Guid)->Data4[7]

Definition at line 26 of file disk.c.

GUID_FORMAT_STR

#define GUID_FORMAT_STR   "%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x"

Definition at line 25 of file disk.c.

Function Documentation

DBG_DEFAULT_CHANNEL()

DBG_DEFAULT_CHANNEL

(

DISK

)

DiskError()

VOID DiskError	(	_In_ PCSTR	ErrorString,
		_In_ ULONG	ErrorCode
	)

Definition at line 48 of file disk.c.

{
    PCSTR ErrorDescription;
    CHAR ErrorCodeString[200];
 
    if (lReportError < 0)
        return;
 
    ErrorDescription = DiskGetErrorCodeString(ErrorCode);
    if (ErrorDescription)
    {
        RtlStringCbPrintfA(ErrorCodeString, sizeof(ErrorCodeString),
                           "%s\n\nError Code: 0x%lx\nError: %s",
                           ErrorString, ErrorCode, ErrorDescription);
    }
    else
    {
        RtlStringCbCopyA(ErrorCodeString, sizeof(ErrorCodeString), ErrorString);
    }
 
    ERR("%s\n", ErrorCodeString);
    UiMessageBox(ErrorCodeString);
}

Referenced by Pc98DiskReadLogicalSectorsCHS(), Pc98DiskReadLogicalSectorsLBA(), PcDiskReadLogicalSectorsCHS(), PcDiskReadLogicalSectorsLBA(), and XboxDiskReadLogicalSectors().

DiskInitialize()

ARC_STATUS DiskInitialize	(	_In_ UCHAR	DriveNumber,
		_In_ PCSTR	DeviceName,
		_In_ CONFIGURATION_TYPE	DeviceType,
		_In_ const DEVVTBL *	FuncTable,
		_Out_opt_ PULONG	pChecksum,
		_Out_opt_ PULONG	pSignature,
		_Out_opt_ PBOOLEAN	pValidPartitionTable
	)

Definition at line 78 of file disk.c.

{
    PMASTER_BOOT_RECORD Mbr;
    PULONG Buffer;
    GEOMETRY Geometry;
    ULONGLONG SectorStart;
    ULONG MbrSectorSize;
    ULONG i;
    ULONG Checksum, Signature;
    BOOLEAN ValidPartitionTable;
    BOOLEAN IsCdRom;
    PGUID GptDiskGuid = NULL;
    GPT_TABLE_HEADER GptHeader;
    PARTITION_INFORMATION PartitionEntry;
    CHAR ArcName[MAX_PATH];
    NTSTATUS NtStatus;
 
    TRACE("DiskInitialize(0x%02X, '%s', Type: %lu)\n", DriveNumber, DeviceName, DeviceType);
 
    if (!MachDiskGetDriveGeometry(DriveNumber, &Geometry))
        return FALSE;
 
    IsCdRom = (DeviceType == CdromController);
    if (IsCdRom)
    {
        SectorStart = 16ULL;
        MbrSectorSize = 2048;
    }
    else // (DeviceType == FloppyDiskPeripheral || DiskPeripheral)
    {
        SectorStart = 0ULL;
        MbrSectorSize = 512;
    }
 
    /* Read the MBR */
    if (!MachDiskReadLogicalSectors(DriveNumber, SectorStart, 1, DiskReadBuffer))
    {
        ERR("Reading MBR failed\n");
        return EIO;
    }
 
    Buffer = (PULONG)DiskReadBuffer;
    Mbr = (PMASTER_BOOT_RECORD)DiskReadBuffer;
 
    Signature = Mbr->Signature;
    TRACE("Signature: %x\n", Signature);
 
    /* Calculate the MBR checksum */
    Checksum = 0;
    for (i = 0; i < MbrSectorSize / sizeof(ULONG); i++)
    {
        Checksum += Buffer[i];
    }
    Checksum = ~Checksum + 1;
    TRACE("Checksum: %x\n", Checksum);
 
    ValidPartitionTable = (IsCdRom || (Mbr->MasterBootRecordMagic == 0xAA55));
    TRACE("IsPartitionValid: %s\n", ValidPartitionTable ? "TRUE" : "FALSE");
 
    /* Register the device */
    // NOTE: For now, only register the direct device if it's not a
    // "rigid" disk, i.e. peripherals not of the form 'xxx(y)rdisk(z)'.
    // Rigid disks are registered with the 'partition(0)' suffix below.
    if (DeviceType != DiskPeripheral)
    {
        if (!FsRegisterDevice(DeviceName, FuncTable))
            return ENOMEM;
    }
 
    /* GPT disk signature support */
    if (DeviceType == DiskPeripheral)
    {
        if (DiskReadGptHeader(DriveNumber, &GptHeader))
            GptDiskGuid = &GptHeader.DiskGuid;
    }
    if (GptDiskGuid)
    {
        TRACE("Disk 0x%02X is GPT, DiskGuid: {" GUID_FORMAT_STR "}\n",
              DriveNumber, GUID_ELEMENTS(GptDiskGuid));
    }
 
    /* Fill out the ARC disk block */
    AddReactOSArcDiskInfo(DeviceName, GptDiskGuid, Signature,
                          Checksum, ValidPartitionTable);
 
    if (pChecksum)
        *pChecksum = Checksum;
    if (pSignature)
        *pSignature = Signature;
    if (pValidPartitionTable)
        *pValidPartitionTable = ValidPartitionTable;
 
    /* Don't search for partitions in non-"rigid" disks (floppies, CD-ROMs) */
    if (DeviceType != DiskPeripheral)
        return ESUCCESS;
 
    /* Register the device with the 'partition(0)' suffix */
    NtStatus = RtlStringCbPrintfA(ArcName, sizeof(ArcName), "%spartition(0)", DeviceName);
    if (!NT_SUCCESS(NtStatus))
        return ENAMETOOLONG;
    if (!FsRegisterDevice(ArcName, FuncTable))
        return ENOMEM;
 
    /* Detect disk partition type */
    DiskDetectPartitionType(DriveNumber);
 
    /* Add partitions */
    i = FIRST_PARTITION;
    while (DiskGetPartitionEntry(DriveNumber, Geometry.BytesPerSector, i, &PartitionEntry))
    {
        if (PartitionEntry.PartitionType != PARTITION_ENTRY_UNUSED)
        {
            NtStatus = RtlStringCbPrintfA(ArcName, sizeof(ArcName),
                                          "%spartition(%lu)", DeviceName, i);
            if (!NT_SUCCESS(NtStatus))
                return ENAMETOOLONG;
            if (!FsRegisterDevice(ArcName, FuncTable))
                return ENOMEM;
        }
        i++;
    }
 
    return ESUCCESS;
}

Referenced by GetHarddiskInformation(), PcInitializeBootDevices(), and UefiInitializeBootDevices().

DiskReadGptHeader()

BOOLEAN DiskReadGptHeader	(	_In_ UCHAR	DriveNumber,
		_Out_ PGPT_TABLE_HEADER	GptHeader
	)

Definition at line 22 of file part_gpt.c.

{
    /* Read GPT header (1 sector) from LBA 1 */
    if (!MachDiskReadLogicalSectors(DriveNumber, EFI_HEADER_LOCATION, 1, DiskReadBuffer))
        return FALSE;
    RtlCopyMemory(GptHeader, DiskReadBuffer, sizeof(*GptHeader));
 
    /* Verify GPT signature */
    if (!RtlEqualMemory(GptHeader->Signature, EFI_PARTITION_HEADER_SIGNATURE, 8))
        return FALSE;
 
    /* Verify revision */
    if (GptHeader->Revision != EFI_TABLE_REVISION)
    {
        TRACE("GPT header has unsupported revision: 0x%x\n", GptHeader->Revision);
        return FALSE;
    }
 
    return TRUE;
}

Referenced by DiskGetGptPartitionEntry(), and DiskInitialize().

DiskReportError()

LONG DiskReportError

(

_In_ BOOLEAN

bShowError

)

Definition at line 38 of file disk.c.

{
    /* Set the reference count */
    if (bShowError) ++lReportError;
    else            --lReportError;
    return lReportError;
}

Referenced by EnumerateHarddisks(), GetHarddiskInformation(), PcInitializeBootDevices(), and UefiInitializeBootDevices().

Variable Documentation

lReportError

LONG lReportError = 0

static

Definition at line 35 of file disk.c.

Referenced by DiskError(), and DiskReportError().