mirror of
https://github.com/tianocore/edk2
synced 2026-08-27 00:23:19 -04:00
77585e5004
Add the positive-path host-based tests for the shared GptLib parser (extracted as part of the parser security hardening), ensuring the tightened checks in PartitionValidGptTable(), PartitionCheckGptEntry() and PartitionRestoreGptTable() do not falsely reject well-formed GPTs. The tests run against an in-memory mock disk and cover accepted primary/backup headers, boundary but legal header/entry sizes, correct entry-status flagging on valid entries, and primary/backup restore round-trips. Signed-off-by: Richard Lyu <richard.lyu@suse.com>
110 lines
3.1 KiB
C
110 lines
3.1 KiB
C
/** @file
|
|
Shared mock disk and GPT construction helpers used by the GptLib
|
|
host-based unit tests.
|
|
|
|
Copyright (c) 2026, SUSE LLC. All rights reserved.<BR>
|
|
SPDX-License-Identifier: BSD-2-Clause-Patent
|
|
**/
|
|
|
|
#pragma once
|
|
|
|
#include <Uefi.h>
|
|
#include <Protocol/BlockIo.h>
|
|
#include <Protocol/DiskIo.h>
|
|
#include <Library/GptLib.h>
|
|
|
|
#define SECTOR_SIZE 512U
|
|
#define DISK_SECTORS 4096U
|
|
#define DISK_IMAGE_SIZE (DISK_SECTORS * SECTOR_SIZE)
|
|
#define LAST_LBA (DISK_SECTORS - 1U)
|
|
#define NUM_PARTITION_ENTRIES 128U
|
|
#define PARTITION_ENTRY_SIZE 128U
|
|
#define PART_ARRAY_SIZE (NUM_PARTITION_ENTRIES * PARTITION_ENTRY_SIZE)
|
|
#define FIRST_USABLE_LBA 34U
|
|
#define LAST_USABLE_LBA (DISK_SECTORS - 34U)
|
|
|
|
//
|
|
// The only GPT header revision defined by the UEFI specification, and the
|
|
// minimum on-disk header size (through PartitionEntryArrayCRC32).
|
|
//
|
|
#define TEST_GPT_REVISION_V1 0x00010000U
|
|
#define TEST_GPT_HEADER_MIN_SIZE 92U
|
|
|
|
extern UINT8 mDiskImage[DISK_IMAGE_SIZE];
|
|
|
|
extern EFI_BLOCK_IO_MEDIA mBlockIoMedia;
|
|
extern EFI_BLOCK_IO_PROTOCOL mBlockIo;
|
|
extern EFI_DISK_IO_PROTOCOL mDiskIo;
|
|
extern BOOLEAN mWriteProtected;
|
|
|
|
extern CONST EFI_GUID mEspTypeGuid;
|
|
extern CONST EFI_GUID mPartitionGuid1;
|
|
extern CONST EFI_GUID mPartitionGuid2;
|
|
|
|
UINT8 *
|
|
GetDiskLba (
|
|
IN EFI_LBA Lba
|
|
);
|
|
|
|
EFI_PARTITION_TABLE_HEADER *
|
|
GetPrimaryHeader (
|
|
VOID
|
|
);
|
|
|
|
VOID
|
|
UpdateGptHeaderCrc (
|
|
IN OUT EFI_PARTITION_TABLE_HEADER *Header
|
|
);
|
|
|
|
VOID
|
|
FillPartitionEntry (
|
|
IN OUT EFI_PARTITION_ENTRY *Entry,
|
|
IN CONST EFI_GUID *UniquePartitionGuid,
|
|
IN EFI_LBA StartingLba,
|
|
IN EFI_LBA EndingLba
|
|
);
|
|
|
|
/**
|
|
Write a self-consistent GPT table (header plus partition entry array) to the
|
|
mock disk at the requested locations, computing correct header and
|
|
entry-array CRC32 values.
|
|
|
|
@param[in] HeaderLba LBA where the GPT header is written. Also stored
|
|
in the header's MyLBA field.
|
|
@param[in] AlternateLba Value stored in the header's AlternateLBA field.
|
|
@param[in] EntryArrayLba LBA where the partition entry array is written.
|
|
@param[in] NumEntries NumberOfPartitionEntries value.
|
|
@param[in] EntrySize SizeOfPartitionEntry value.
|
|
@param[in] NumPartitions Number of non-empty partition entries to populate.
|
|
**/
|
|
VOID
|
|
WriteGptTableAt (
|
|
IN EFI_LBA HeaderLba,
|
|
IN EFI_LBA AlternateLba,
|
|
IN EFI_LBA EntryArrayLba,
|
|
IN UINT32 NumEntries,
|
|
IN UINT32 EntrySize,
|
|
IN UINT32 NumPartitions
|
|
);
|
|
|
|
/**
|
|
Reset the mock disk to a single valid primary GPT: header at LBA 1, a
|
|
128 x 128-byte entry array at LBA 2 holding two partitions.
|
|
**/
|
|
VOID
|
|
SetupDiskWithValidPrimary (
|
|
VOID
|
|
);
|
|
|
|
BOOLEAN
|
|
ValidateAt (
|
|
IN EFI_LBA Lba,
|
|
OUT EFI_PARTITION_TABLE_HEADER *Header
|
|
);
|
|
|
|
VOID
|
|
InitCheckEntryHeader (
|
|
OUT EFI_PARTITION_TABLE_HEADER *Header,
|
|
IN UINT32 NumEntries,
|
|
IN UINT32 EntrySize
|
|
);
|