4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync/** @file
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync The library instance provides security service of TPM measure boot.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncCopyright (c) 2009 - 2012, Intel Corporation. All rights reserved.<BR>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncThis program and the accompanying materials
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncare licensed and made available under the terms and conditions of the BSD License
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncwhich accompanies this distribution. The full text of the license may be found at
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsynchttp://opensource.org/licenses/bsd-license.php
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncTHE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncWITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync**/
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <PiDxe.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <Protocol/TcgService.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <Protocol/FirmwareVolume2.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <Protocol/BlockIo.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <Protocol/DiskIo.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <Protocol/DevicePathToText.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <Library/BaseLib.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <Library/DebugLib.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <Library/BaseMemoryLib.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <Library/MemoryAllocationLib.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <Library/DevicePathLib.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <Library/UefiBootServicesTableLib.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <Library/BaseCryptLib.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <Library/PeCoffLib.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <Library/SecurityManagementLib.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync//
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync// Flag to check GPT partition. It only need be measured once.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync//
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncBOOLEAN mMeasureGptTableFlag = FALSE;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncEFI_GUID mZeroGuid = {0, 0, 0, {0, 0, 0, 0, 0, 0, 0, 0}};
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncUINTN mMeasureGptCount = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncVOID *mFileBuffer;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncUINTN mImageSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync/**
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Reads contents of a PE/COFF image in memory buffer.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param FileHandle Pointer to the file handle to read the PE/COFF image.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param FileOffset Offset into the PE/COFF image to begin the read operation.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param ReadSize On input, the size in bytes of the requested read operation.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync On output, the number of bytes actually read.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param Buffer Output buffer that contains the data read from the PE/COFF image.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_SUCCESS The specified portion of the PE/COFF image was read and the size
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync**/
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncEFI_STATUS
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncEFIAPI
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncDxeTpmMeasureBootLibImageRead (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN VOID *FileHandle,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN UINTN FileOffset,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN OUT UINTN *ReadSize,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync OUT VOID *Buffer
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync )
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync{
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN EndPosition;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (FileHandle == NULL || ReadSize == NULL || Buffer == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_INVALID_PARAMETER;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (MAX_ADDRESS - FileOffset < *ReadSize) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_INVALID_PARAMETER;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EndPosition = FileOffset + *ReadSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (EndPosition > mImageSize) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync *ReadSize = (UINT32)(mImageSize - FileOffset);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (FileOffset >= mImageSize) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync *ReadSize = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CopyMem (Buffer, (UINT8 *)((UINTN) FileHandle + FileOffset), *ReadSize);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_SUCCESS;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync}
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync/**
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Measure GPT table data into TPM log.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param TcgProtocol Pointer to the located TCG protocol instance.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param GptHandle Handle that GPT partition was installed.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_SUCCESS Successfully measure GPT table.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_UNSUPPORTED Not support GPT table on the given handle.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_DEVICE_ERROR Can't get GPT table because device error.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_OUT_OF_RESOURCES No enough resource to measure GPT table.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval other error value
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync**/
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncEFI_STATUS
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncEFIAPI
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncTcgMeasureGptTable (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN EFI_TCG_PROTOCOL *TcgProtocol,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN EFI_HANDLE GptHandle
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync )
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync{
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_STATUS Status;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_BLOCK_IO_PROTOCOL *BlockIo;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_DISK_IO_PROTOCOL *DiskIo;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_PARTITION_TABLE_HEADER *PrimaryHeader;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_PARTITION_ENTRY *PartitionEntry;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINT8 *EntryPtr;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN NumberOfPartition;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINT32 Index;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TCG_PCR_EVENT *TcgEvent;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_GPT_DATA *GptData;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINT32 EventSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINT32 EventNumber;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_PHYSICAL_ADDRESS EventLogLastEntry;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (mMeasureGptCount > 0) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_SUCCESS;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = gBS->HandleProtocol (GptHandle, &gEfiBlockIoProtocolGuid, (VOID**)&BlockIo);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (EFI_ERROR (Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_UNSUPPORTED;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = gBS->HandleProtocol (GptHandle, &gEfiDiskIoProtocolGuid, (VOID**)&DiskIo);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (EFI_ERROR (Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_UNSUPPORTED;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Read the EFI Partition Table Header
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync PrimaryHeader = (EFI_PARTITION_TABLE_HEADER *) AllocatePool (BlockIo->Media->BlockSize);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (PrimaryHeader == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_OUT_OF_RESOURCES;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = DiskIo->ReadDisk (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DiskIo,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync BlockIo->Media->MediaId,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync 1 * BlockIo->Media->BlockSize,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync BlockIo->Media->BlockSize,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (UINT8 *)PrimaryHeader
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync );
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (EFI_ERROR (Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DEBUG ((EFI_D_ERROR, "Failed to Read Partition Table Header!\n"));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool (PrimaryHeader);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_DEVICE_ERROR;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Read the partition entry.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EntryPtr = (UINT8 *)AllocatePool (PrimaryHeader->NumberOfPartitionEntries * PrimaryHeader->SizeOfPartitionEntry);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (EntryPtr == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool (PrimaryHeader);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_OUT_OF_RESOURCES;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = DiskIo->ReadDisk (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DiskIo,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync BlockIo->Media->MediaId,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync MultU64x32(PrimaryHeader->PartitionEntryLBA, BlockIo->Media->BlockSize),
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync PrimaryHeader->NumberOfPartitionEntries * PrimaryHeader->SizeOfPartitionEntry,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EntryPtr
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync );
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (EFI_ERROR (Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool (PrimaryHeader);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool (EntryPtr);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_DEVICE_ERROR;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Count the valid partition
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync PartitionEntry = (EFI_PARTITION_ENTRY *)EntryPtr;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync NumberOfPartition = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync for (Index = 0; Index < PrimaryHeader->NumberOfPartitionEntries; Index++) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!CompareGuid (&PartitionEntry->PartitionTypeGUID, &mZeroGuid)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync NumberOfPartition++;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync PartitionEntry++;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Parepare Data for Measurement
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EventSize = (UINT32)(sizeof (EFI_GPT_DATA) - sizeof (GptData->Partitions)
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync + NumberOfPartition * PrimaryHeader->SizeOfPartitionEntry);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgEvent = (TCG_PCR_EVENT *) AllocateZeroPool (EventSize + sizeof (TCG_PCR_EVENT));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (TcgEvent == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool (PrimaryHeader);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool (EntryPtr);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_OUT_OF_RESOURCES;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgEvent->PCRIndex = 5;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgEvent->EventType = EV_EFI_GPT_EVENT;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgEvent->EventSize = EventSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync GptData = (EFI_GPT_DATA *) TcgEvent->Event;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Copy the EFI_PARTITION_TABLE_HEADER and NumberOfPartition
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CopyMem ((UINT8 *)GptData, (UINT8*)PrimaryHeader, sizeof (EFI_PARTITION_TABLE_HEADER));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync GptData->NumberOfPartitions = NumberOfPartition;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Copy the valid partition entry
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync PartitionEntry = (EFI_PARTITION_ENTRY*)EntryPtr;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync NumberOfPartition = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync for (Index = 0; Index < PrimaryHeader->NumberOfPartitionEntries; Index++) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!CompareGuid (&PartitionEntry->PartitionTypeGUID, &mZeroGuid)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CopyMem (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (UINT8 *)&GptData->Partitions + NumberOfPartition * sizeof (EFI_PARTITION_ENTRY),
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (UINT8 *)PartitionEntry,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync sizeof (EFI_PARTITION_ENTRY)
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync );
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync NumberOfPartition++;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync PartitionEntry++;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Measure the GPT data
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EventNumber = 1;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = TcgProtocol->HashLogExtendEvent (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgProtocol,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (EFI_PHYSICAL_ADDRESS) (UINTN) (VOID *) GptData,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (UINT64) TcgEvent->EventSize,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TPM_ALG_SHA,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgEvent,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &EventNumber,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &EventLogLastEntry
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync );
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!EFI_ERROR (Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync mMeasureGptCount++;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool (PrimaryHeader);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool (EntryPtr);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool (TcgEvent);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return Status;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync}
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync/**
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Measure PE image into TPM log based on the authenticode image hashing in
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync PE/COFF Specification 8.0 Appendix A.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] TcgProtocol Pointer to the located TCG protocol instance.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] ImageAddress Start address of image buffer.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] ImageSize Image size
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] LinkTimeBase Address that the image is loaded into memory.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] ImageType Image subsystem type.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] FilePath File path is corresponding to the input image.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_SUCCESS Successfully measure image.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_OUT_OF_RESOURCES No enough resource to measure image.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_UNSUPPORTED ImageType is unsupported or PE image is mal-format.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval other error value
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync**/
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncEFI_STATUS
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncEFIAPI
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncTcgMeasurePeImage (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN EFI_TCG_PROTOCOL *TcgProtocol,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN EFI_PHYSICAL_ADDRESS ImageAddress,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN UINTN ImageSize,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN UINTN LinkTimeBase,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN UINT16 ImageType,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN EFI_DEVICE_PATH_PROTOCOL *FilePath
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync )
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync{
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_STATUS Status;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TCG_PCR_EVENT *TcgEvent;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_IMAGE_LOAD_EVENT *ImageLoad;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINT32 FilePathSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync VOID *Sha1Ctx;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN CtxSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_IMAGE_DOS_HEADER *DosHdr;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINT32 PeCoffHeaderOffset;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_IMAGE_SECTION_HEADER *Section;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINT8 *HashBase;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN HashSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN SumOfBytesHashed;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_IMAGE_SECTION_HEADER *SectionHeader;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN Index;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN Pos;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINT16 Magic;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINT32 EventSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINT32 EventNumber;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_PHYSICAL_ADDRESS EventLogLastEntry;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION Hdr;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINT32 NumberOfRvaAndSizes;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync BOOLEAN HashStatus;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINT32 CertSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = EFI_UNSUPPORTED;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ImageLoad = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync SectionHeader = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Sha1Ctx = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FilePathSize = (UINT32) GetDevicePathSize (FilePath);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Determine destination PCR by BootPolicy
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EventSize = sizeof (*ImageLoad) - sizeof (ImageLoad->DevicePath) + FilePathSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgEvent = AllocateZeroPool (EventSize + sizeof (TCG_PCR_EVENT));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (TcgEvent == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_OUT_OF_RESOURCES;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgEvent->EventSize = EventSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ImageLoad = (EFI_IMAGE_LOAD_EVENT *) TcgEvent->Event;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync switch (ImageType) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync case EFI_IMAGE_SUBSYSTEM_EFI_APPLICATION:
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgEvent->EventType = EV_EFI_BOOT_SERVICES_APPLICATION;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgEvent->PCRIndex = 4;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync break;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync case EFI_IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVER:
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgEvent->EventType = EV_EFI_BOOT_SERVICES_DRIVER;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgEvent->PCRIndex = 2;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync break;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync case EFI_IMAGE_SUBSYSTEM_EFI_RUNTIME_DRIVER:
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgEvent->EventType = EV_EFI_RUNTIME_SERVICES_DRIVER;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgEvent->PCRIndex = 2;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync break;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync default:
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DEBUG ((
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_D_ERROR,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync "TcgMeasurePeImage: Unknown subsystem type %d",
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ImageType
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto Finish;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ImageLoad->ImageLocationInMemory = ImageAddress;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ImageLoad->ImageLengthInMemory = ImageSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ImageLoad->ImageLinkTimeAddress = LinkTimeBase;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ImageLoad->LengthOfDevicePath = FilePathSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CopyMem (ImageLoad->DevicePath, FilePath, FilePathSize);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Check PE/COFF image
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DosHdr = (EFI_IMAGE_DOS_HEADER *) (UINTN) ImageAddress;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync PeCoffHeaderOffset = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (DosHdr->e_magic == EFI_IMAGE_DOS_SIGNATURE) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync PeCoffHeaderOffset = DosHdr->e_lfanew;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)((UINT8 *) (UINTN) ImageAddress + PeCoffHeaderOffset);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Hdr.Pe32->Signature != EFI_IMAGE_NT_SIGNATURE) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto Finish;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // PE/COFF Image Measurement
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // NOTE: The following codes/steps are based upon the authenticode image hashing in
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // PE/COFF Specification 8.0 Appendix A.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 1. Load the image header into memory.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 2. Initialize a SHA hash context.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CtxSize = Sha1GetContextSize ();
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Sha1Ctx = AllocatePool (CtxSize);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Sha1Ctx == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = EFI_OUT_OF_RESOURCES;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto Finish;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashStatus = Sha1Init (Sha1Ctx);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!HashStatus) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto Finish;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Measuring PE/COFF Image Header;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // But CheckSum field and SECURITY data directory (certificate) are excluded
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Magic = Hdr.Pe32->OptionalHeader.Magic;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 3. Calculate the distance from the base of the image header to the image checksum address.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 4. Hash the image header from its base to beginning of the image checksum.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashBase = (UINT8 *) (UINTN) ImageAddress;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Magic == EFI_IMAGE_NT_OPTIONAL_HDR32_MAGIC) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Use PE32 offset
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync NumberOfRvaAndSizes = Hdr.Pe32->OptionalHeader.NumberOfRvaAndSizes;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashSize = (UINTN) ((UINT8 *)(&Hdr.Pe32->OptionalHeader.CheckSum) - HashBase);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Use PE32+ offset
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync NumberOfRvaAndSizes = Hdr.Pe32Plus->OptionalHeader.NumberOfRvaAndSizes;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashSize = (UINTN) ((UINT8 *)(&Hdr.Pe32Plus->OptionalHeader.CheckSum) - HashBase);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashStatus = Sha1Update (Sha1Ctx, HashBase, HashSize);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!HashStatus) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto Finish;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 5. Skip over the image checksum (it occupies a single ULONG).
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (NumberOfRvaAndSizes <= EFI_IMAGE_DIRECTORY_ENTRY_SECURITY) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 6. Since there is no Cert Directory in optional header, hash everything
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // from the end of the checksum to the end of image header.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Magic == EFI_IMAGE_NT_OPTIONAL_HDR32_MAGIC) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Use PE32 offset.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashBase = (UINT8 *) &Hdr.Pe32->OptionalHeader.CheckSum + sizeof (UINT32);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashSize = Hdr.Pe32->OptionalHeader.SizeOfHeaders - (UINTN) (HashBase - ImageAddress);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Use PE32+ offset.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashBase = (UINT8 *) &Hdr.Pe32Plus->OptionalHeader.CheckSum + sizeof (UINT32);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashSize = Hdr.Pe32Plus->OptionalHeader.SizeOfHeaders - (UINTN) (HashBase - ImageAddress);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (HashSize != 0) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashStatus = Sha1Update (Sha1Ctx, HashBase, HashSize);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!HashStatus) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto Finish;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 7. Hash everything from the end of the checksum to the start of the Cert Directory.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Magic == EFI_IMAGE_NT_OPTIONAL_HDR32_MAGIC) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Use PE32 offset
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashBase = (UINT8 *) &Hdr.Pe32->OptionalHeader.CheckSum + sizeof (UINT32);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashSize = (UINTN) ((UINT8 *)(&Hdr.Pe32->OptionalHeader.DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_SECURITY]) - HashBase);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Use PE32+ offset
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashBase = (UINT8 *) &Hdr.Pe32Plus->OptionalHeader.CheckSum + sizeof (UINT32);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashSize = (UINTN) ((UINT8 *)(&Hdr.Pe32Plus->OptionalHeader.DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_SECURITY]) - HashBase);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (HashSize != 0) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashStatus = Sha1Update (Sha1Ctx, HashBase, HashSize);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!HashStatus) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto Finish;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 8. Skip over the Cert Directory. (It is sizeof(IMAGE_DATA_DIRECTORY) bytes.)
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 9. Hash everything from the end of the Cert Directory to the end of image header.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Magic == EFI_IMAGE_NT_OPTIONAL_HDR32_MAGIC) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Use PE32 offset
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashBase = (UINT8 *) &Hdr.Pe32->OptionalHeader.DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_SECURITY + 1];
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashSize = Hdr.Pe32->OptionalHeader.SizeOfHeaders - (UINTN) (HashBase - ImageAddress);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Use PE32+ offset
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashBase = (UINT8 *) &Hdr.Pe32Plus->OptionalHeader.DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_SECURITY + 1];
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashSize = Hdr.Pe32Plus->OptionalHeader.SizeOfHeaders - (UINTN) (HashBase - ImageAddress);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (HashSize != 0) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashStatus = Sha1Update (Sha1Ctx, HashBase, HashSize);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!HashStatus) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto Finish;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 10. Set the SUM_OF_BYTES_HASHED to the size of the header
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Magic == EFI_IMAGE_NT_OPTIONAL_HDR32_MAGIC) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Use PE32 offset
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync SumOfBytesHashed = Hdr.Pe32->OptionalHeader.SizeOfHeaders;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Use PE32+ offset
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync SumOfBytesHashed = Hdr.Pe32Plus->OptionalHeader.SizeOfHeaders;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 11. Build a temporary table of pointers to all the IMAGE_SECTION_HEADER
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // structures in the image. The 'NumberOfSections' field of the image
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // header indicates how big the table should be. Do not include any
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // IMAGE_SECTION_HEADERs in the table whose 'SizeOfRawData' field is zero.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync SectionHeader = (EFI_IMAGE_SECTION_HEADER *) AllocateZeroPool (sizeof (EFI_IMAGE_SECTION_HEADER) * Hdr.Pe32->FileHeader.NumberOfSections);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (SectionHeader == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = EFI_OUT_OF_RESOURCES;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto Finish;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 12. Using the 'PointerToRawData' in the referenced section headers as
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // a key, arrange the elements in the table in ascending order. In other
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // words, sort the section headers according to the disk-file offset of
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // the section.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Section = (EFI_IMAGE_SECTION_HEADER *) (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (UINT8 *) (UINTN) ImageAddress +
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync PeCoffHeaderOffset +
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync sizeof(UINT32) +
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync sizeof(EFI_IMAGE_FILE_HEADER) +
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Hdr.Pe32->FileHeader.SizeOfOptionalHeader
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync );
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync for (Index = 0; Index < Hdr.Pe32->FileHeader.NumberOfSections; Index++) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Pos = Index;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync while ((Pos > 0) && (Section->PointerToRawData < SectionHeader[Pos - 1].PointerToRawData)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CopyMem (&SectionHeader[Pos], &SectionHeader[Pos - 1], sizeof(EFI_IMAGE_SECTION_HEADER));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Pos--;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CopyMem (&SectionHeader[Pos], Section, sizeof(EFI_IMAGE_SECTION_HEADER));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Section += 1;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 13. Walk through the sorted table, bring the corresponding section
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // into memory, and hash the entire section (using the 'SizeOfRawData'
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // field in the section header to determine the amount of data to hash).
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 14. Add the section's 'SizeOfRawData' to SUM_OF_BYTES_HASHED .
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 15. Repeat steps 13 and 14 for all the sections in the sorted table.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync for (Index = 0; Index < Hdr.Pe32->FileHeader.NumberOfSections; Index++) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Section = (EFI_IMAGE_SECTION_HEADER *) &SectionHeader[Index];
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Section->SizeOfRawData == 0) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync continue;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashBase = (UINT8 *) (UINTN) ImageAddress + Section->PointerToRawData;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashSize = (UINTN) Section->SizeOfRawData;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashStatus = Sha1Update (Sha1Ctx, HashBase, HashSize);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!HashStatus) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto Finish;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync SumOfBytesHashed += HashSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 16. If the file size is greater than SUM_OF_BYTES_HASHED, there is extra
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // data in the file that needs to be added to the hash. This data begins
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // at file offset SUM_OF_BYTES_HASHED and its length is:
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // FileSize - (CertDirectory->Size)
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ImageSize > SumOfBytesHashed) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashBase = (UINT8 *) (UINTN) ImageAddress + SumOfBytesHashed;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (NumberOfRvaAndSizes <= EFI_IMAGE_DIRECTORY_ENTRY_SECURITY) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CertSize = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Magic == EFI_IMAGE_NT_OPTIONAL_HDR32_MAGIC) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Use PE32 offset.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CertSize = Hdr.Pe32->OptionalHeader.DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_SECURITY].Size;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Use PE32+ offset.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CertSize = Hdr.Pe32Plus->OptionalHeader.DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_SECURITY].Size;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ImageSize > CertSize + SumOfBytesHashed) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashSize = (UINTN) (ImageSize - CertSize - SumOfBytesHashed);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashStatus = Sha1Update (Sha1Ctx, HashBase, HashSize);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!HashStatus) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto Finish;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else if (ImageSize < CertSize + SumOfBytesHashed) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto Finish;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 17. Finalize the SHA hash.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HashStatus = Sha1Final (Sha1Ctx, (UINT8 *) &TcgEvent->Digest);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!HashStatus) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto Finish;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Log the PE data
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EventNumber = 1;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = TcgProtocol->HashLogExtendEvent (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgProtocol,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (EFI_PHYSICAL_ADDRESS) (UINTN) (VOID *) NULL,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync 0,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TPM_ALG_SHA,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgEvent,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &EventNumber,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &EventLogLastEntry
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync );
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncFinish:
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool (TcgEvent);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (SectionHeader != NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool (SectionHeader);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Sha1Ctx != NULL ) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool (Sha1Ctx);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return Status;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync}
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync/**
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync The security handler is used to abstract platform-specific policy
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync from the DXE core response to an attempt to use a file that returns a
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync given status for the authentication check from the section extraction protocol.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync The possible responses in a given SAP implementation may include locking
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync flash upon failure to authenticate, attestation logging for all signed drivers,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync and other exception operations. The File parameter allows for possible logging
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync within the SAP of the driver.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync If File is NULL, then EFI_INVALID_PARAMETER is returned.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync If the file specified by File with an authentication status specified by
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync AuthenticationStatus is safe for the DXE Core to use, then EFI_SUCCESS is returned.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync If the file specified by File with an authentication status specified by
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync AuthenticationStatus is not safe for the DXE Core to use under any circumstances,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync then EFI_ACCESS_DENIED is returned.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync If the file specified by File with an authentication status specified by
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync AuthenticationStatus is not safe for the DXE Core to use right now, but it
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync might be possible to use it at a future time, then EFI_SECURITY_VIOLATION is
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync returned.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in, out] AuthenticationStatus This is the authentication status returned
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync from the securitymeasurement services for the
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync input file.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] File This is a pointer to the device path of the file that is
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync being dispatched. This will optionally be used for logging.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] FileBuffer File buffer matches the input file device path.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] FileSize Size of File buffer matches the input file device path.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_SUCCESS The file specified by File did authenticate, and the
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync platform policy dictates that the DXE Core may use File.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_INVALID_PARAMETER File is NULL.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_SECURITY_VIOLATION The file specified by File did not authenticate, and
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync the platform policy dictates that File should be placed
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync in the untrusted state. A file may be promoted from
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync the untrusted to the trusted state at a future time
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync with a call to the Trust() DXE Service.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_ACCESS_DENIED The file specified by File did not authenticate, and
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync the platform policy dictates that File should not be
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync used for any purpose.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync**/
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncEFI_STATUS
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncEFIAPI
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncDxeTpmMeasureBootHandler (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN OUT UINT32 AuthenticationStatus,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN CONST EFI_DEVICE_PATH_PROTOCOL *File,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN VOID *FileBuffer OPTIONAL,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN UINTN FileSize OPTIONAL
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync )
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync{
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_TCG_PROTOCOL *TcgProtocol;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_STATUS Status;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TCG_EFI_BOOT_SERVICE_CAPABILITY ProtocolCapability;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINT32 TCGFeatureFlags;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_PHYSICAL_ADDRESS EventLogLocation;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_PHYSICAL_ADDRESS EventLogLastEntry;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_DEVICE_PATH_PROTOCOL *DevicePathNode;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_DEVICE_PATH_PROTOCOL *OrigDevicePathNode;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_HANDLE Handle;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync BOOLEAN ApplicationRequired;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync PE_COFF_LOADER_IMAGE_CONTEXT ImageContext;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (File == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_INVALID_PARAMETER;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = gBS->LocateProtocol (&gEfiTcgProtocolGuid, NULL, (VOID **) &TcgProtocol);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (EFI_ERROR (Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // TCG protocol is not installed. So, TPM is not present.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Don't do any measurement, and directly return EFI_SUCCESS.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_SUCCESS;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ProtocolCapability.Size = (UINT8) sizeof (ProtocolCapability);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = TcgProtocol->StatusCheck (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgProtocol,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &ProtocolCapability,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &TCGFeatureFlags,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &EventLogLocation,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &EventLogLastEntry
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync );
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (EFI_ERROR (Status) || ProtocolCapability.TPMDeactivatedFlag) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // TPM device doesn't work or activate.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_SUCCESS;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Copy File Device Path
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync OrigDevicePathNode = DuplicateDevicePath (File);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ASSERT (OrigDevicePathNode != NULL);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 1. Check whether this device path support BlockIo protocol.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Is so, this device path may be a GPT device path.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DevicePathNode = OrigDevicePathNode;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = gBS->LocateDevicePath (&gEfiBlockIoProtocolGuid, &DevicePathNode, &Handle);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!EFI_ERROR (Status) && !mMeasureGptTableFlag) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Find the gpt partion on the given devicepath
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DevicePathNode = OrigDevicePathNode;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync while (!IsDevicePathEnd (DevicePathNode)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Find the Gpt partition
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (DevicePathType (DevicePathNode) == MEDIA_DEVICE_PATH &&
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DevicePathSubType (DevicePathNode) == MEDIA_HARDDRIVE_DP) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Check whether it is a gpt partition or not
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (((HARDDRIVE_DEVICE_PATH *) DevicePathNode)->MBRType == MBR_TYPE_EFI_PARTITION_TABLE_HEADER &&
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ((HARDDRIVE_DEVICE_PATH *) DevicePathNode)->SignatureType == SIGNATURE_TYPE_GUID) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Change the partition device path to its parent device path (disk) and get the handle.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DevicePathNode->Type = END_DEVICE_PATH_TYPE;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DevicePathNode->SubType = END_ENTIRE_DEVICE_PATH_SUBTYPE;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DevicePathNode = OrigDevicePathNode;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = gBS->LocateDevicePath (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &gEfiDiskIoProtocolGuid,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &DevicePathNode,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &Handle
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync );
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!EFI_ERROR (Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Measure GPT disk.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = TcgMeasureGptTable (TcgProtocol, Handle);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!EFI_ERROR (Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // GPT disk check done.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync mMeasureGptTableFlag = TRUE;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool (OrigDevicePathNode);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync OrigDevicePathNode = DuplicateDevicePath (File);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ASSERT (OrigDevicePathNode != NULL);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync break;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DevicePathNode = NextDevicePathNode (DevicePathNode);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // 2. Measure PE image.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ApplicationRequired = FALSE;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Check whether this device path support FV2 protocol.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DevicePathNode = OrigDevicePathNode;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = gBS->LocateDevicePath (&gEfiFirmwareVolume2ProtocolGuid, &DevicePathNode, &Handle);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!EFI_ERROR (Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Don't check FV image, and directly return EFI_SUCCESS.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // It can be extended to the specific FV authentication according to the different requirement.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (IsDevicePathEnd (DevicePathNode)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return EFI_SUCCESS;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // The image from Firmware image will not be mearsured.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Current policy doesn't measure PeImage from Firmware if it is driver
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // If the got PeImage is application, it will be still be measured.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ApplicationRequired = TRUE;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // File is not found.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (FileBuffer == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = EFI_SECURITY_VIOLATION;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto Finish;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync mImageSize = FileSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync mFileBuffer = FileBuffer;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Measure PE Image
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DevicePathNode = OrigDevicePathNode;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ZeroMem (&ImageContext, sizeof (ImageContext));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ImageContext.Handle = (VOID *) FileBuffer;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ImageContext.ImageRead = (PE_COFF_LOADER_READ_FILE) DxeTpmMeasureBootLibImageRead;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Get information about the image being loaded
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = PeCoffLoaderGetImageInfo (&ImageContext);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (EFI_ERROR (Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // The information can't be got from the invalid PeImage
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto Finish;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Measure only application if Application flag is set
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Measure drivers and applications if Application flag is not set
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if ((!ApplicationRequired) ||
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (ApplicationRequired && ImageContext.ImageType == EFI_IMAGE_SUBSYSTEM_EFI_APPLICATION)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Print the image path to be measured.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DEBUG_CODE_BEGIN ();
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CHAR16 *ToText;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_DEVICE_PATH_TO_TEXT_PROTOCOL *DevPathToText;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = gBS->LocateProtocol (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &gEfiDevicePathToTextProtocolGuid,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync NULL,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (VOID **) &DevPathToText
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync );
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!EFI_ERROR (Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ToText = DevPathToText->ConvertDevicePathToText (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DevicePathNode,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FALSE,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TRUE
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync );
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ToText != NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DEBUG ((DEBUG_INFO, "The measured image path is %s.\n", ToText));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DEBUG_CODE_END ();
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Measure PE image into TPM log.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = TcgMeasurePeImage (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TcgProtocol,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (EFI_PHYSICAL_ADDRESS) (UINTN) FileBuffer,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FileSize,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (UINTN) ImageContext.ImageAddress,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ImageContext.ImageType,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DevicePathNode
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync );
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Done, free the allocated resource.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncFinish:
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool (OrigDevicePathNode);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return Status;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync}
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync/**
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Register the security handler to provide TPM measure boot service.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param ImageHandle ImageHandle of the loaded driver.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param SystemTable Pointer to the EFI System Table.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_SUCCESS Register successfully.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_OUT_OF_RESOURCES No enough memory to register this handler.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync**/
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncEFI_STATUS
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncEFIAPI
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncDxeTpmMeasureBootLibConstructor (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN EFI_HANDLE ImageHandle,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN EFI_SYSTEM_TABLE *SystemTable
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync )
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync{
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return RegisterSecurityHandler (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DxeTpmMeasureBootHandler,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_AUTH_OPERATION_MEASURE_IMAGE | EFI_AUTH_OPERATION_IMAGE_REQUIRED
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync );
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync}