--- /dev/null
+/** @file\r
+ Platform Flash Access library.\r
+\r
+ Copyright (c) 2016, Intel Corporation. All rights reserved.<BR>\r
+ This program and the accompanying materials\r
+ are licensed and made available under the terms and conditions of the BSD License\r
+ which accompanies this distribution. The full text of the license may be found at\r
+ http://opensource.org/licenses/bsd-license.php\r
+\r
+ THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
+ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
+\r
+**/\r
+\r
+#include <PiDxe.h>\r
+\r
+#include <Library/BaseLib.h>\r
+#include <Library/BaseMemoryLib.h>\r
+#include <Library/DebugLib.h>\r
+#include <Library/PcdLib.h>\r
+#include <Library/PlatformFlashAccessLib.h>\r
+#include <Library/UefiBootServicesTableLib.h>\r
+#include <Protocol/Spi.h>\r
+\r
+//\r
+// SPI default opcode slots\r
+//\r
+#define SPI_OPCODE_JEDEC_ID_INDEX 0\r
+#define SPI_OPCODE_READ_ID_INDEX 1\r
+#define SPI_OPCODE_WRITE_S_INDEX 2\r
+#define SPI_OPCODE_WRITE_INDEX 3\r
+#define SPI_OPCODE_READ_INDEX 4\r
+#define SPI_OPCODE_ERASE_INDEX 5\r
+#define SPI_OPCODE_READ_S_INDEX 6\r
+#define SPI_OPCODE_CHIP_ERASE_INDEX 7\r
+\r
+#define SPI_ERASE_SECTOR_SIZE SIZE_4KB //This is the chipset requirement\r
+\r
+STATIC EFI_PHYSICAL_ADDRESS mInternalFdAddress;\r
+EFI_SPI_PROTOCOL *mSpiProtocol;\r
+\r
+/**\r
+ Writes specified number of bytes from the input buffer to the address\r
+\r
+ @param[in] WriteAddress The flash address to be written.\r
+ @param[in, out] NumBytes The number of bytes.\r
+ @param[in] Buffer The data buffer to be written.\r
+\r
+ @return The status of flash write.\r
+**/\r
+EFI_STATUS\r
+FlashFdWrite (\r
+ IN UINTN WriteAddress,\r
+ IN OUT UINTN *NumBytes,\r
+ IN UINT8 *Buffer\r
+ )\r
+{\r
+ EFI_STATUS Status;\r
+\r
+ Status = EFI_SUCCESS;\r
+\r
+ Status = mSpiProtocol->Execute (\r
+ mSpiProtocol,\r
+ SPI_OPCODE_WRITE_INDEX, // OpcodeIndex\r
+ 0, // PrefixOpcodeIndex\r
+ TRUE, // DataCycle\r
+ TRUE, // Atomic\r
+ TRUE, // ShiftOut\r
+ WriteAddress, // Address\r
+ (UINT32) (*NumBytes), // Data Number\r
+ Buffer,\r
+ EnumSpiRegionBios\r
+ );\r
+ DEBUG((DEBUG_INFO, "FlashFdWrite - 0x%x - %r\n", (UINTN)WriteAddress, Status));\r
+\r
+ AsmWbinvd ();\r
+\r
+ return Status;\r
+}\r
+\r
+/**\r
+ Erase a certain block from address LbaWriteAddress\r
+\r
+ @param[in] WriteAddress The flash address to be erased.\r
+\r
+ @return The status of flash erase.\r
+**/\r
+EFI_STATUS\r
+FlashFdErase (\r
+ IN UINTN WriteAddress\r
+ )\r
+{\r
+ EFI_STATUS Status;\r
+\r
+ Status = mSpiProtocol->Execute (\r
+ mSpiProtocol,\r
+ SPI_OPCODE_ERASE_INDEX, // OpcodeIndex\r
+ 0, // PrefixOpcodeIndex\r
+ FALSE, // DataCycle\r
+ TRUE, // Atomic\r
+ FALSE, // ShiftOut\r
+ WriteAddress, // Address\r
+ 0, // Data Number\r
+ NULL,\r
+ EnumSpiRegionBios // SPI_REGION_TYPE\r
+ );\r
+ DEBUG((DEBUG_INFO, "FlashFdErase - 0x%x - %r\n", (UINTN)WriteAddress, Status));\r
+\r
+ AsmWbinvd ();\r
+\r
+ return Status;\r
+}\r
+\r
+/**\r
+ Perform flash write opreation.\r
+\r
+ @param[in] FirmwareType The type of firmware.\r
+ @param[in] FlashAddress The address of flash device to be accessed.\r
+ @param[in] FlashAddressType The type of flash device address.\r
+ @param[in] Buffer The pointer to the data buffer.\r
+ @param[in] Length The length of data buffer in bytes.\r
+\r
+ @retval EFI_SUCCESS The operation returns successfully.\r
+ @retval EFI_WRITE_PROTECTED The flash device is read only.\r
+ @retval EFI_UNSUPPORTED The flash device access is unsupported.\r
+ @retval EFI_INVALID_PARAMETER The input parameter is not valid.\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+PerformFlashWrite (\r
+ IN PLATFORM_FIRMWARE_TYPE FirmwareType,\r
+ IN EFI_PHYSICAL_ADDRESS FlashAddress,\r
+ IN FLASH_ADDRESS_TYPE FlashAddressType,\r
+ IN VOID *Buffer,\r
+ IN UINTN Length\r
+ )\r
+{\r
+ EFI_STATUS Status;\r
+ UINTN SectorNum;\r
+ UINTN Index;\r
+ UINTN NumBytes;\r
+\r
+ DEBUG((DEBUG_INFO, "PerformFlashWrite - 0x%x(%x) - 0x%x\n", (UINTN)FlashAddress, (UINTN)FlashAddressType, Length));\r
+ if (FlashAddressType == FlashAddressTypeAbsoluteAddress) {\r
+ FlashAddress = FlashAddress - mInternalFdAddress;\r
+ }\r
+\r
+ //\r
+ // Erase & Write\r
+ //\r
+ SectorNum = Length / SPI_ERASE_SECTOR_SIZE;\r
+ for (Index = 0; Index < SectorNum; Index++){\r
+ if (CompareMem(\r
+ (UINT8 *)(UINTN)(FlashAddress + mInternalFdAddress) + Index * SPI_ERASE_SECTOR_SIZE,\r
+ (UINT8 *)Buffer + Index * SPI_ERASE_SECTOR_SIZE,\r
+ SPI_ERASE_SECTOR_SIZE) == 0) {\r
+ DEBUG((DEBUG_INFO, "Sector - 0x%x - skip\n", Index));\r
+ continue;\r
+ }\r
+ DEBUG((DEBUG_INFO, "Sector - 0x%x - update...\n", Index));\r
+\r
+ Status = FlashFdErase (\r
+ (UINTN)FlashAddress + Index * SPI_ERASE_SECTOR_SIZE\r
+ );\r
+ if (Status != EFI_SUCCESS){\r
+ break;\r
+ }\r
+ NumBytes = SPI_ERASE_SECTOR_SIZE;\r
+ Status = FlashFdWrite (\r
+ (UINTN)FlashAddress + Index * SPI_ERASE_SECTOR_SIZE,\r
+ &NumBytes,\r
+ (UINT8 *)Buffer + Index * SPI_ERASE_SECTOR_SIZE\r
+ );\r
+ if (Status != EFI_SUCCESS){\r
+ break;\r
+ }\r
+ }\r
+\r
+ return EFI_SUCCESS;\r
+}\r
+\r
+/**\r
+ Platform Flash Access Lib Constructor.\r
+\r
+ @param[in] ImageHandle The firmware allocated handle for the EFI image.\r
+ @param[in] SystemTable A pointer to the EFI System Table.\r
+\r
+ @retval EFI_SUCCESS Constructor returns successfully.\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+PerformFlashAccessLibConstructor (\r
+ IN EFI_HANDLE ImageHandle,\r
+ IN EFI_SYSTEM_TABLE *SystemTable\r
+ )\r
+{\r
+ EFI_STATUS Status;\r
+\r
+ mInternalFdAddress = (EFI_PHYSICAL_ADDRESS)(UINTN)PcdGet32(PcdFlashAreaBaseAddress);\r
+ DEBUG((DEBUG_INFO, "PcdFlashAreaBaseAddress - 0x%x\n", mInternalFdAddress));\r
+\r
+ Status = gBS->LocateProtocol(&gEfiSpiProtocolGuid, NULL, (VOID **)&mSpiProtocol);\r
+ ASSERT_EFI_ERROR(Status);\r
+\r
+ return EFI_SUCCESS;\r
+}\r