ArmPlatformPkg/Drivers/NorFlashDxe: Directly implement DiskIO protocol
[mirror_edk2.git] / ArmPlatformPkg / Drivers / NorFlashDxe / NorFlashFvbDxe.c
index 9ad0e14..bf42085 100644 (file)
@@ -531,201 +531,11 @@ FvbWrite (
   IN        UINT8                                 *Buffer\r
   )\r
 {\r
-  EFI_STATUS  TempStatus;\r
-  UINT32      Tmp;\r
-  UINT32      TmpBuf;\r
-  UINT32      WordToWrite;\r
-  UINT32      Mask;\r
-  UINTN       DoErase;\r
-  UINTN       BytesToWrite;\r
-  UINTN       CurOffset;\r
-  UINTN       WordAddr;\r
-  UINTN       BlockSize;\r
   NOR_FLASH_INSTANCE *Instance;\r
-  UINTN       BlockAddress;\r
-  UINTN       PrevBlockAddress;\r
 \r
-  PrevBlockAddress = 0;\r
+  Instance = INSTANCE_FROM_FVB_THIS (This);\r
 \r
-  Instance = INSTANCE_FROM_FVB_THIS(This);\r
-\r
-  if (!Instance->Initialized && Instance->Initialize) {\r
-    Instance->Initialize(Instance);\r
-  }\r
-\r
-  DEBUG ((DEBUG_BLKIO, "FvbWrite(Parameters: Lba=%ld, Offset=0x%x, *NumBytes=0x%x, Buffer @ 0x%08x)\n", Instance->StartLba + Lba, Offset, *NumBytes, Buffer));\r
-\r
-  // Detect WriteDisabled state\r
-  if (Instance->Media.ReadOnly == TRUE) {\r
-    DEBUG ((EFI_D_ERROR, "FvbWrite: ERROR - Can not write: Device is in WriteDisabled state.\n"));\r
-    // It is in WriteDisabled state, return an error right away\r
-    return EFI_ACCESS_DENIED;\r
-  }\r
-\r
-  // Cache the block size to avoid de-referencing pointers all the time\r
-  BlockSize = Instance->Media.BlockSize;\r
-\r
-  // The write must not span block boundaries.\r
-  // We need to check each variable individually because adding two large values together overflows.\r
-  if ( ( Offset               >= BlockSize ) ||\r
-       ( *NumBytes            >  BlockSize ) ||\r
-       ( (Offset + *NumBytes) >  BlockSize )    ) {\r
-    DEBUG ((EFI_D_ERROR, "FvbWrite: ERROR - EFI_BAD_BUFFER_SIZE: (Offset=0x%x + NumBytes=0x%x) > BlockSize=0x%x\n", Offset, *NumBytes, BlockSize ));\r
-    return EFI_BAD_BUFFER_SIZE;\r
-  }\r
-\r
-  // We must have some bytes to write\r
-  if (*NumBytes == 0) {\r
-    DEBUG ((EFI_D_ERROR, "FvbWrite: ERROR - EFI_BAD_BUFFER_SIZE: (Offset=0x%x + NumBytes=0x%x) > BlockSize=0x%x\n", Offset, *NumBytes, BlockSize ));\r
-    return EFI_BAD_BUFFER_SIZE;\r
-  }\r
-\r
-  // Pick 128bytes as a good start for word operations as opposed to erasing the\r
-  // block and writing the data regardless if an erase is really needed.\r
-  // It looks like most individual NV variable writes are smaller than 128bytes.\r
-  if (*NumBytes <= 128) {\r
-    // Check to see if we need to erase before programming the data into NOR.\r
-    // If the destination bits are only changing from 1s to 0s we can just write.\r
-    // After a block is erased all bits in the block is set to 1.\r
-    // If any byte requires us to erase we just give up and rewrite all of it.\r
-    DoErase = 0;\r
-    BytesToWrite = *NumBytes;\r
-    CurOffset = Offset;\r
-\r
-    while (BytesToWrite > 0) {\r
-      // Read full word from NOR, splice as required. A word is the smallest\r
-      // unit we can write.\r
-      TempStatus = NorFlashRead (Instance, Instance->StartLba + Lba,\r
-                                 CurOffset & ~(0x3), sizeof(Tmp), &Tmp);\r
-      if (EFI_ERROR (TempStatus)) {\r
-        return EFI_DEVICE_ERROR;\r
-      }\r
-\r
-      // Physical address of word in NOR to write.\r
-      WordAddr = (CurOffset & ~(0x3)) + GET_NOR_BLOCK_ADDRESS (Instance->RegionBaseAddress,\r
-                                                               Lba, BlockSize);\r
-      // The word of data that is to be written.\r
-      TmpBuf = *((UINT32*)(Buffer + (*NumBytes - BytesToWrite)));\r
-\r
-      // First do word aligned chunks.\r
-      if ((CurOffset & 0x3) == 0) {\r
-        if (BytesToWrite >= 4) {\r
-          // Is the destination still in 'erased' state?\r
-          if (~Tmp != 0) {\r
-            // Check to see if we are only changing bits to zero.\r
-            if ((Tmp ^ TmpBuf) & TmpBuf) {\r
-              DoErase = 1;\r
-              break;\r
-            }\r
-          }\r
-          // Write this word to NOR\r
-          WordToWrite = TmpBuf;\r
-          CurOffset += sizeof(TmpBuf);\r
-          BytesToWrite -= sizeof(TmpBuf);\r
-        } else {\r
-          // BytesToWrite < 4. Do small writes and left-overs\r
-          Mask = ~((~0) << (BytesToWrite * 8));\r
-          // Mask out the bytes we want.\r
-          TmpBuf &= Mask;\r
-          // Is the destination still in 'erased' state?\r
-          if ((Tmp & Mask) != Mask) {\r
-            // Check to see if we are only changing bits to zero.\r
-            if ((Tmp ^ TmpBuf) & TmpBuf) {\r
-              DoErase = 1;\r
-              break;\r
-            }\r
-          }\r
-          // Merge old and new data. Write merged word to NOR\r
-          WordToWrite = (Tmp & ~Mask) | TmpBuf;\r
-          CurOffset += BytesToWrite;\r
-          BytesToWrite = 0;\r
-        }\r
-      } else {\r
-        // Do multiple words, but starting unaligned.\r
-        if (BytesToWrite > (4 - (CurOffset & 0x3))) {\r
-          Mask = ~((~0) << ((CurOffset & 0x3) * 8));\r
-          // Mask out the bytes we want.\r
-          TmpBuf = (TmpBuf << ((CurOffset & 0x3) * 8)) & Mask;\r
-          // Is the destination still in 'erased' state?\r
-          if ((Tmp & Mask) != Mask) {\r
-            // Check to see if we are only changing bits to zero.\r
-            if ((Tmp ^ TmpBuf) & TmpBuf) {\r
-              DoErase = 1;\r
-              break;\r
-            }\r
-          }\r
-          // Merge old and new data. Write merged word to NOR\r
-          WordToWrite = (Tmp & ~Mask) | TmpBuf;\r
-          BytesToWrite -= (4 - (CurOffset & 0x3));\r
-          CurOffset += (4 - (CurOffset & 0x3));\r
-        } else {\r
-          // Unaligned and fits in one word.\r
-          Mask = (~((~0) << (BytesToWrite * 8))) << ((CurOffset & 0x3) * 8);\r
-          // Mask out the bytes we want.\r
-          TmpBuf = (TmpBuf << ((CurOffset & 0x3) * 8)) & Mask;\r
-          // Is the destination still in 'erased' state?\r
-          if ((Tmp & Mask) != Mask) {\r
-            // Check to see if we are only changing bits to zero.\r
-            if ((Tmp ^ TmpBuf) & TmpBuf) {\r
-              DoErase = 1;\r
-              break;\r
-            }\r
-          }\r
-          // Merge old and new data. Write merged word to NOR\r
-          WordToWrite = (Tmp & ~Mask) | TmpBuf;\r
-          CurOffset += BytesToWrite;\r
-          BytesToWrite = 0;\r
-        }\r
-      }\r
-\r
-      //\r
-      // Write the word to NOR.\r
-      //\r
-\r
-      BlockAddress = GET_NOR_BLOCK_ADDRESS (Instance->RegionBaseAddress, Lba, BlockSize);\r
-      if (BlockAddress != PrevBlockAddress) {\r
-        TempStatus = NorFlashUnlockSingleBlockIfNecessary (Instance, BlockAddress);\r
-        if (EFI_ERROR (TempStatus)) {\r
-          return EFI_DEVICE_ERROR;\r
-        }\r
-        PrevBlockAddress = BlockAddress;\r
-      }\r
-      TempStatus = NorFlashWriteSingleWord (Instance, WordAddr, WordToWrite);\r
-      if (EFI_ERROR (TempStatus)) {\r
-        return EFI_DEVICE_ERROR;\r
-      }\r
-    }\r
-    // Exit if we got here and could write all the data. Otherwise do the\r
-    // Erase-Write cycle.\r
-    if (!DoErase) {\r
-      return EFI_SUCCESS;\r
-    }\r
-  }\r
-\r
-  // Check we did get some memory. Buffer is BlockSize.\r
-  if (Instance->FvbBuffer == NULL) {\r
-    DEBUG ((EFI_D_ERROR, "FvbWrite: ERROR - Buffer not ready\n"));\r
-    return EFI_DEVICE_ERROR;\r
-  }\r
-\r
-  // Read NOR Flash data into shadow buffer\r
-  TempStatus = NorFlashReadBlocks (Instance, Instance->StartLba + Lba, BlockSize, Instance->FvbBuffer);\r
-  if (EFI_ERROR (TempStatus)) {\r
-    // Return one of the pre-approved error statuses\r
-    return EFI_DEVICE_ERROR;\r
-  }\r
-\r
-  // Put the data at the appropriate location inside the buffer area\r
-  CopyMem ((VOID*)((UINTN)Instance->FvbBuffer + Offset), Buffer, *NumBytes);\r
-\r
-  // Write the modified buffer back to the NorFlash\r
-  TempStatus = NorFlashWriteBlocks (Instance, Instance->StartLba + Lba, BlockSize, Instance->FvbBuffer);\r
-  if (EFI_ERROR (TempStatus)) {\r
-    // Return one of the pre-approved error statuses\r
-    return EFI_DEVICE_ERROR;\r
-  }\r
-\r
-  return EFI_SUCCESS;\r
+  return NorFlashWriteSingleBlock (Instance, Lba, Offset, NumBytes, Buffer);\r
 }\r
 \r
 /**\r