VOID SetBadBlock(OAL_BLMENU_ITEM *pMenu)
{
HANDLE hFMD = NULL;
PCI_REG_INFO regInfo;
FlashInfo flashInfo;
BLOCK_ID blockId;
WCHAR szInputLine[16];
UNREFERENCED_PARAMETER(pMenu);
// Open FMD
regInfo.MemBase.Reg[0] = g_ulFlashBase;
hFMD = FMD_Init(NULL, ®Info, NULL);
if (hFMD == NULL)
{
OALLog(L" Oops, can't open FMD driver\r\n");
goto cleanUp;
}
if (!FMD_GetInfo(&flashInfo))
{
OALLog(L" Oops, can't get flash geometry info\r\n");
goto cleanUp;
}
OALLog(L"\r\n Block Number: ");
if (OALBLMenuReadLine(szInputLine, dimof(szInputLine)) == 0)
{
goto cleanUp;
}
// Get sector number
blockId = OALStringToUINT32(szInputLine);
// Check sector number
if (blockId >= flashInfo.dwNumBlocks)
{
OALLog(L" Oops, too big block number\r\n");
goto cleanUp;
}
FMD_SetBlockStatus(blockId, BLOCK_STATUS_BAD);
OALLog(L"\r\n Done\r\n");
cleanUp:
if (hFMD != NULL)
{
FMD_Deinit(hFMD);
}
return;
}
BOOL WriteFlashNK(UINT32 address, UINT32 size)
{
BOOL rc = FALSE;
HANDLE hFMD;
FlashInfo flashInfo;
SectorInfo sectorInfo;
BOOL ok = FALSE;
BLOCK_ID block;
//ROMHDR *pTOC;
UINT32 *pInfo, count, sector;
UINT32 blockSize, sectorSize, sectorsPerBlock;
UINT8 *pData;
OALMSG(OAL_INFO, (L"OEMWriteFlash: Writing NK image to flash\r\n"));
// We need to know sector/block size
if ((hFMD = FMD_Init(NULL, NULL, NULL)) == NULL) {
OALMSG(OAL_ERROR, (L"ERROR: OEMWriteFlash: "
L"FMD_Init call failed\r\n"
));
goto cleanUp;
}
// Get flash info
if (!FMD_GetInfo(&flashInfo)) {
OALMSG(OAL_ERROR, (L"ERROR: EBOOT!OEMWriteFlash: "
L"FMD_GetInfo call failed!\r\n"
));
FMD_Deinit(hFMD);
goto cleanUp;
}
// We don't need access to FMD library
FMD_Deinit(hFMD);
OALMSG(OAL_INFO, (L"OEMWriteFlash: "
L"Flash has %d blocks, %d bytes/block, %d sectors/block\r\n",
flashInfo.dwNumBlocks, flashInfo.dwBytesPerBlock,
flashInfo.wSectorsPerBlock
));
// Make block & sector size ready
blockSize = flashInfo.dwBytesPerBlock;
sectorSize = flashInfo.wDataBytesPerSector;
sectorsPerBlock = flashInfo.wSectorsPerBlock;
// Get data location
pData = OEMMapMemAddr(address, address);
// Verify that we get CE image.
pInfo = (UINT32*)(pData + ROM_SIGNATURE_OFFSET);
if (*pInfo++ != ROM_SIGNATURE) {
OALMSG(OAL_ERROR, (L"ERROR: OEMWriteFlash: "
L"Image Signature not found\r\n"
));
goto cleanUp;
}
pInfo++;
// Skip reserved blocks
block = 0;
while (block < flashInfo.dwNumBlocks) {
if ((FMD_GetBlockStatus(block) & BLOCK_STATUS_BAD) != 0) {
OALMSG(OAL_WARN, (L"WARN: EBOOT!OEMWriteFlash: "
L"Skip bad block %d\r\n", block
));
block++;
continue;
}
if ((FMD_GetBlockStatus(block) & BLOCK_STATUS_RESERVED) == 0) break;
block++;
}
OALMSG(OAL_INFO, (L"OEMWriteFlash: "
L"NK image starts at block %d\r\n", block
));
// Write image
count = 0;
while (count < size && block < flashInfo.dwNumBlocks) {
// If block is bad, we have to offset it
if ((FMD_GetBlockStatus(block) & BLOCK_STATUS_BAD) != 0) {
block++;
OALMSG(OAL_WARN, (L"WARN: EBOOT!OEMWriteFlash: "
L"Skip bad block %d\r\n", block
));
continue;
}
// Erase block
if (!EraseBlock(block)) {
FMD_SetBlockStatus(block, BLOCK_STATUS_BAD);
block++;
OALMSG(OAL_WARN, (L"WARN: EBOOT!OEMWriteFlash: "
L"Block %d erase failed, mark block as bad\r\n", block
));
continue;
}
// Now write sectors
sector = 0;
while (sector < sectorsPerBlock && count < size) {
// Prepare sector info
memset(§orInfo, 0xFF, sizeof(sectorInfo));
sectorInfo.dwReserved1 = 0;
sectorInfo.wReserved2 = 0;
// Write sector
if (!(ok = WriteSector(
block * sectorsPerBlock + sector, pData + count, §orInfo
))) break;
// Move to next sector
count += sectorSize;
sector++;
}
// When sector write failed, mark block as bad and move back
if (!ok) {
OALMSG(OAL_WARN, (L"WARN: EBOOT!OEMWriteFlash: "
L"Block %d sector %d write failed, mark block as bad\r\n",
block, sector
));
// First move back
count -= sector * flashInfo.wDataBytesPerSector;
// Mark block as bad
FMD_SetBlockStatus(block, BLOCK_STATUS_BAD);
}
// We are done with block
block++;
}
// Erase rest of media
while (block < flashInfo.dwNumBlocks) {
// If block is bad, we have to offset it
if ((FMD_GetBlockStatus(block) & BLOCK_STATUS_BAD) != 0) {
block++;
OALMSG(OAL_WARN, (L"WARN: EBOOT!OEMWriteFlash: "
L"Skip bad block %d\r\n", block
));
continue;
}
// When erase failed, mark block as bad and skip it
if (!EraseBlock(block)) {
FMD_SetBlockStatus(block, BLOCK_STATUS_BAD);
block++;
OALMSG(OAL_WARN, (L"WARN: EBOOT!OEMWriteFlash: "
L"Block %d erase failed, mark block as bad\r\n", block
));
continue;
}
// Move to next block
block++;
}
// Close FMD driver
FMD_Deinit(hFMD);
hFMD = NULL;
OALMSG(OAL_INFO, (L"OEMWriteFlash: NK written\r\n"));
// Done
rc = TRUE;
cleanUp:
if (hFMD != NULL) FMD_Deinit(hFMD);
return rc;
}
BOOL
BLReserveBootBlocks(
BOOT_CFG *pBootCfg
)
{
#ifndef BSP_NO_NAND_IN_SDBOOT
BOOL rc = FALSE;
HANDLE hFMD;
PCI_REG_INFO regInfo;
FlashInfo flashInfo;
UINT32 size;
BLOCK_ID firstblock, lastblock;
UINT32 status;
UNREFERENCED_PARAMETER(pBootCfg);
// Automatically mark the bootloader blocks as read-only/reserved
regInfo.MemBase.Reg[0] = g_ulFlashBase;
hFMD = FMD_Init(NULL, ®Info, NULL);
if (hFMD == NULL)
{
OALMSG(OAL_ERROR, (L"ERROR: FMD_Init call failed!\r\n"));
goto cleanUp;
}
// Get flash info
if (!FMD_GetInfo(&flashInfo))
{
OALMSG(OAL_ERROR, (L"ERROR: FMD_GetInfo call failed!\r\n"));
goto cleanUp;
}
// Loop thru the bootloader blocks to ensure they are marked reserved
firstblock = 0;
size = IMAGE_BOOTLOADER_NAND_SIZE;
lastblock = ((size -1) / flashInfo.dwBytesPerBlock) + 1;
OALLog(L"Checking bootloader blocks are marked as reserved (Num = %d)\r\n", lastblock-firstblock);
while (firstblock < lastblock)
{
// If block is bad, we have to offset it
status = FMD_GetBlockStatus(firstblock);
// Skip bad blocks
if ((status & BLOCK_STATUS_BAD) != 0)
{
OALLog(L" Skip bad block %d\r\n", firstblock);
// blocks marked bad would not have been written either, so don't include this
// in the count of blocks that are reserved.
firstblock++;
lastblock++;
continue;
}
// Skip already reserved blocks
if ((status & BLOCK_STATUS_RESERVED) != 0)
{
firstblock++;
continue;
}
// Mark block as read-only & reserved
if (!FMD_SetBlockStatus(firstblock, BLOCK_STATUS_READONLY|BLOCK_STATUS_RESERVED))
{
OALLog(L" Oops, can't mark block %d - as reserved\r\n", firstblock);
}
firstblock++;
OALLog(L".");
}
// Done
rc = TRUE;
OALLog(L"\r\n");
cleanUp:
if (hFMD != NULL) FMD_Deinit(hFMD);
return rc;
#else
UNREFERENCED_PARAMETER(pBootCfg);
// Nothing to do...
return TRUE;
#endif
}
VOID ReserveBlock(OAL_BLMENU_ITEM *pMenu)
{
WCHAR key;
HANDLE hFMD = NULL;
PCI_REG_INFO regInfo;
FlashInfo flashInfo;
BLOCK_ID firstblock, lastblock=0;
WCHAR szInputLine[16];
UINT32 status;
UNREFERENCED_PARAMETER(pMenu);
OALLog(L"\r\n First Block Number: ");
if (OALBLMenuReadLine(szInputLine, dimof(szInputLine)) == 0)
{
goto cleanUp;
}
// Get block number
firstblock = OALStringToUINT32(szInputLine);
OALLog(L"\r\n Last Block Number: ");
if (OALBLMenuReadLine(szInputLine, dimof(szInputLine)) != 0)
{
// Get block number
lastblock = OALStringToUINT32(szInputLine);
}
if (lastblock < firstblock)
{
lastblock=firstblock;
}
// Open FMD
regInfo.MemBase.Reg[0] = g_ulFlashBase;
hFMD = FMD_Init(NULL, ®Info, NULL);
if (hFMD == NULL)
{
OALLog(L" Oops, can't open FMD driver\r\n");
goto cleanUp;
}
if (!FMD_GetInfo(&flashInfo))
{
OALLog(L" Oops, can't get flash geometry info\r\n");
goto cleanUp;
}
if (lastblock >= flashInfo.dwNumBlocks)
{
OALLog(L" Oops, too big block number\r\n");
goto cleanUp;
}
OALLog(L" Do you want mark as reserved block %d-%d [-/y]? ", firstblock, lastblock);
// Get key
key = OALBLMenuReadKey(TRUE);
OALLog(L"%c\r\n", key);
// Depending on result
if (key != L'y' && key != L'Y')
{
goto cleanUp;
}
while (firstblock<=lastblock)
{
// If block is bad, we have to offset it
status = FMD_GetBlockStatus(firstblock);
// Skip bad blocks
if ((status & BLOCK_STATUS_BAD) != 0)
{
OALLog(L" Skip bad block %d\r\n", firstblock);
// NOTE - this will cause a smaller number of blocks to actually be reserved...
firstblock++;
continue;
}
// Skip already reserved blocks
if ((status & BLOCK_STATUS_RESERVED) != 0)
{
OALLog(L" Skip reserved block %d\r\n", firstblock);
firstblock++;
continue;
}
// Mark block as read-only & reserved
if (!FMD_SetBlockStatus(firstblock, BLOCK_STATUS_READONLY|BLOCK_STATUS_RESERVED))
{
OALLog(L" Oops, can't mark block %d - as reserved\r\n", firstblock);
}
firstblock++;
OALLog(L".");
}
OALLog(L" Done\r\n");
cleanUp:
if (hFMD != NULL) FMD_Deinit(hFMD);
return;
}
VOID EraseFlash(OAL_BLMENU_ITEM *pMenu)
{
WCHAR key;
HANDLE hFMD = NULL;
PCI_REG_INFO regInfo;
FlashInfo flashInfo;
BLOCK_ID block;
UINT32 status;
UNREFERENCED_PARAMETER(pMenu);
OALLog(L" Do you want to erase unreserved blocks [-/y]? ");
// Get key
key = OALBLMenuReadKey(TRUE);
OALLog(L"%c\r\n", key);
// Depending on result
if (key != L'y' && key != L'Y') goto cleanUp;
// Open FMD
regInfo.MemBase.Reg[0] = g_ulFlashBase;
hFMD = FMD_Init(NULL, ®Info, NULL);
if (hFMD == NULL)
{
OALLog(L" Oops, can't open FMD driver\r\n");
goto cleanUp;
}
if (!FMD_GetInfo(&flashInfo))
{
OALLog(L" Oops, can't get flash geometry info\r\n");
goto cleanUp;
}
// First offset given
block = 0;
while (block < flashInfo.dwNumBlocks)
{
// If block is bad, we have to offset it
status = FMD_GetBlockStatus(block);
// Skip bad blocks
if ((status & BLOCK_STATUS_BAD) != 0)
{
OALLog(L" Skip bad block %d\r\n", block);
block++;
continue;
}
// Skip reserved blocks
if ((status & BLOCK_STATUS_RESERVED) != 0)
{
OALLog(L" Skip reserved block %d\r\n", block);
block++;
continue;
}
// Erase block
if (!FMD_EraseBlock(block))
{
OALLog(L" Oops, can't erase block %d - mark as bad\r\n", block);
FMD_SetBlockStatus(block, BLOCK_STATUS_BAD);
}
block++;
}
OALLog(L" Done\r\n");
// Block until a keypress
OALBLMenuReadKey(TRUE);
cleanUp:
if (hFMD != NULL) FMD_Deinit(hFMD);
return;
}
