android13/bootable/recovery/rkupdate/RKDevice.cpp

675 lines
14 KiB
C++
Executable File
Raw Blame History

/*
* Copyright 2019 Rockchip Electronics Co., Ltd
* Dayao Ji <jdy@rock-chips.com>
*
* SPDX-License-Identifier: GPL-2.0+
*/
#include "rkupdate/RKDevice.h"
const tchar* szManufName[] =
{
_T("SAMSUNG"),
_T("TOSHIBA"),
_T("HYNIX"),
_T("INFINEON"),
_T("MICRON"),
_T("RENESAS"),
_T("ST"),
_T("INTEL")
};
void CRKDevice::SetPrepareEraseFlag(bool value)
{
m_bEraseInPrepare = value;
}
void CRKDevice::SetWorkFlow(UINT value)
{
m_uiWorkFlow = value;
}
void CRKDevice::SetUid(PBYTE value)
{
if (value)
{
if (!m_uid)
{
m_uid = new BYTE[RKDEVICE_UID_LEN];
memset(m_uid,0,RKDEVICE_UID_LEN);
}
memcpy(m_uid,value,RKDEVICE_UID_LEN);
}
else
{
if (m_uid)
{
delete []m_uid;
}
m_uid = value;
}
}
void CRKDevice::SetMiscModifyFlag(ENUM_MISC_MODIFY_FLAG value)
{
m_emMiscModifyFlag = value;
}
void CRKDevice::SetOsType(ENUM_OS_TYPE value)
{
m_os = value;
}
ENUM_OS_TYPE CRKDevice::GetOsType()
{
return m_os;
}
CRKLog* CRKDevice::GetLogObjectPointer()
{
return m_pLog;
}
CRKComm* CRKDevice::GetCommObjectPointer()
{
return m_pComm;
}
CRKDevice::CRKDevice(STRUCT_RKDEVICE_DESC &device)
{
OsType.setContainer(this);
OsType.getter(&CRKDevice::GetOsType);
OsType.setter(&CRKDevice::SetOsType);
Uid.setContainer(this);
Uid.setter(&CRKDevice::SetUid);
PrepareEraseFlag.setContainer(this);
PrepareEraseFlag.setter(&CRKDevice::SetPrepareEraseFlag);
WorkFlow.setContainer(this);
WorkFlow.setter(&CRKDevice::SetWorkFlow);
MiscModifyFlag.setContainer(this);
MiscModifyFlag.setter(&CRKDevice::SetMiscModifyFlag);
LogObjectPointer.setContainer(this);
LogObjectPointer.getter(&CRKDevice::GetLogObjectPointer);
CommObjectPointer.setContainer(this);
CommObjectPointer.getter(&CRKDevice::GetCommObjectPointer);
m_usb = device.emUsbType;
m_device = device.emDeviceType;
m_bcdUsb = device.usbcdUsb;
memset(m_idBlockOffset,0,sizeof(DWORD)*5);
memset(m_flashInfo.blockState,0,IDBLOCK_TOP);
m_flashInfo.usPhyBlokcPerIDB = 1;
m_flashInfo.uiSecNumPerIDB = 0;
m_chipData = NULL;
m_pImage = NULL;
m_pLog = NULL;
m_pComm = NULL;
m_customData = NULL;
m_customDataSize = 0;
m_customDataOffset = 0;
m_sn = NULL;
m_snSize = 0;
m_mac = NULL;
m_imei = NULL;
m_blueTooth = NULL;
m_uid = NULL;
m_sysDiskSize = 0;
m_cfgDiskSize = 0;
m_bGetNewDiskSizeFlag = true;
m_pBlockState = NULL;
m_bWriteBack = false;
m_pFlashInfoData = NULL;
m_usFlashInfoDataLen = 0;
m_usFlashInfoDataOffset = 0;
memset(m_backupBuffer,0,SECTOR_SIZE);
memset(m_backupBuffer+SECTOR_SIZE,0xFF,SPARE_SIZE);
m_bUidWriteOK = false;
m_remallocDisk = false;
m_emMiscModifyFlag = MISC_MODIFY_NONE;
m_bQuickCheckMode = false;
m_bExistSector3Crc = false;
m_bEmmc = false;
m_usSector3Crc = 0;
m_usWriteBackCrc = 0;
m_usWriteBackCustomDataOffset = 0;
m_usWriteBackCustomDataSize = 0;
}
CRKDevice::~CRKDevice()
{
if (m_pComm)
{
delete m_pComm;
m_pComm = NULL;
}
if (m_chipData)
{
delete []m_chipData;
m_chipData = NULL;
}
if (m_customData)
{
delete []m_customData;
m_customData = NULL;
}
if (m_sn)
{
delete []m_sn;
m_sn = NULL;
}
if (m_mac)
{
delete []m_mac;
m_mac = NULL;
}
if (m_imei)
{
delete []m_imei;
m_imei = NULL;
}
if (m_blueTooth)
{
delete []m_blueTooth;
m_blueTooth = NULL;
}
if (m_uid)
{
delete []m_uid;
m_uid = NULL;
}
if (m_pFlashInfoData)
{
delete []m_pFlashInfoData;
m_pFlashInfoData = NULL;
}
}
bool CRKDevice::SetObject(CRKImage *pImage,CRKComm *pComm,CRKLog *pLog)
{
/*pImage<67><65><EFBFBD><EFBFBD>Ϊ<EFBFBD><CEAA>,<2C><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ɲ<EFBFBD><C9B2>ù̼<C3B9><CCBC><EFBFBD><EFBFBD><EFBFBD><EFBFBD>IJ<EFBFBD><C4B2><EFBFBD>*/
if (!pComm)
{
return false;
}
m_pImage = pImage;
m_pComm = pComm;
m_pLog = pLog;
if (m_pImage)
{
m_os = m_pImage->OsType;
}
else
m_os = RK_OS;
return true;
}
int CRKDevice::EraseEmmcBlock(UCHAR ucFlashCS,DWORD dwPos,DWORD dwCount)
{
int sectorOffset,nWrittenBlcok,iRet;
BYTE emptyData[4*(SECTOR_SIZE+SPARE_SIZE)];
memset(emptyData,0xff,4*(SECTOR_SIZE+SPARE_SIZE));
nWrittenBlcok = 0;
while (dwCount>0)
{
sectorOffset = (ucFlashCS*m_flashInfo.uiBlockNum+dwPos+nWrittenBlcok)*m_flashInfo.uiSectorPerBlock;
iRet = m_pComm->RKU_WriteSector(sectorOffset,4,emptyData);
if ((iRet!=ERR_SUCCESS)&&(iRet!=ERR_FOUND_BAD_BLOCK))
{
if (m_pLog)
{
m_pLog->Record(_T("ERROR:EraseEmmcBlock-->RKU_WriteSector failed,RetCode(%d)"),iRet);
}
return iRet;
}
dwCount--;
nWrittenBlcok++;
}
return ERR_SUCCESS;
}
bool CRKDevice::GetFlashInfo()
{
STRUCT_FLASHINFO_CMD info;
//BYTE flashID[5];
int iRet;
UINT uiRead;
iRet = m_pComm->RKU_ReadFlashInfo((PBYTE)&info,&uiRead);
if( ERR_SUCCESS == iRet )
{
if (m_pLog)
{
tstring strFlashInfo;
m_pLog->PrintBuffer(strFlashInfo,(PBYTE)&info,11);
m_pLog->Record(_T("INFO:FlashInfo:%s"),strFlashInfo.c_str());
}
if ((info.usBlockSize==0)||(info.bPageSize==0))
{
if (m_pLog)
{
m_pLog->Record(_T("ERROR:GetFlashInfo-->RKU_ReadFlashInfo failed,pagesize or blocksize is zero"));
}
return false;
}
if ((info.bManufCode>=0)&&(info.bManufCode<=7))
{
_tcscpy(m_flashInfo.szManufacturerName,szManufName[info.bManufCode]);
}
else
{
_tcscpy(m_flashInfo.szManufacturerName,_T("UNKNOWN"));
}
m_flashInfo.uiFlashSize = m_pComm->RKU_GetFlashSize()/2/1024;//MB
//m_flashInfo.uiFlashSize = info.uiFlashSize/2/1024;//MB
m_pLog->Record(_T("INFO:m_flashInfo.uiFlashSize=%u MB, RKU_GetFlashSize=%lld sectors"),m_flashInfo.uiFlashSize, m_pComm->RKU_GetFlashSize());
m_flashInfo.uiPageSize = info.bPageSize/2;//KB
m_flashInfo.usBlockSize = info.usBlockSize/2;//KB
m_flashInfo.bECCBits = info.bECCBits;
m_flashInfo.bAccessTime = info.bAccessTime;
m_flashInfo.uiBlockNum = m_flashInfo.uiFlashSize*1024/m_flashInfo.usBlockSize;
m_flashInfo.uiSectorPerBlock = info.usBlockSize;
m_flashInfo.bFlashCS = info.bFlashCS;
m_flashInfo.usValidSecPerBlock = (info.usBlockSize/info.bPageSize)*4;
if (m_pFlashInfoData)
{
delete []m_pFlashInfoData;
m_pFlashInfoData = NULL;
}
m_usFlashInfoDataLen = BYTE2SECTOR(uiRead);
m_pFlashInfoData = new BYTE[SECTOR_SIZE*m_usFlashInfoDataLen];
memset(m_pFlashInfoData,0,SECTOR_SIZE*m_usFlashInfoDataLen);
memcpy(m_pFlashInfoData,(PBYTE)&info,uiRead);
}
else
{
if (m_pLog)
{
m_pLog->Record(_T("ERROR:GetFlashInfo-->RKU_ReadFlashInfo failed,RetCode(%d)"),iRet);
}
return false;
}
return true;
}
bool CRKDevice::BuildBlockStateMap(BYTE bFlashCS)
{
BYTE blockState[64];
int iRet,i,j ;
memset(blockState, 0, 64);
iRet = m_pComm->RKU_TestBadBlock( bFlashCS, 0, MAX_TEST_BLOCKS, blockState);
if(ERR_SUCCESS == iRet)//<2F>޻<EFBFBD><DEBB><EFBFBD>
{
// return true;
// }
// else if(ERR_FOUND_BAD_BLOCK == iRet)//<2F>л<EFBFBD><D0BB><EFBFBD>
// {
for(i=0; i<64; i++)
{
for(j=0; j<8; j++)
{
if( blockState[i] & (1<<j) )
m_flashInfo.blockState[i*8+j]=1;
if (i*8+j>(IDBLOCK_TOP-2))
{
break;
}
}
if (j<8)
{
break;
}
}
return true;
}
else//<2F><><EFBFBD><EFBFBD>ʧ<EFBFBD><CAA7>
{
if (m_pLog)
{
m_pLog->Record(_T("ERROR:BuildBlockStateMap-->RKU_TestBadBlock failed,RetCode(%d)"),iRet);
}
return false;
}
}
int CRKDevice::ReadMutilSector(DWORD dwPos,DWORD dwCount,PBYTE lpBuffer)
{
DWORD dwReadSector=0,dwMaxReadWriteOnce;
int iUsedSecCount,iUsedBlockCount,iValidSecCount;
int iRet=0,iCurPos;
iUsedBlockCount = dwPos / m_flashInfo.uiSectorPerBlock;
iUsedSecCount = dwPos - (iUsedBlockCount*m_flashInfo.uiSectorPerBlock);
iValidSecCount = m_flashInfo.usValidSecPerBlock-iUsedSecCount;
dwMaxReadWriteOnce = MAX_WRITE_SECTOR;
while(dwCount>0)
{
dwReadSector = (dwCount>=dwMaxReadWriteOnce) ? dwMaxReadWriteOnce : dwCount;
if (dwReadSector>iValidSecCount)
{
dwReadSector = iValidSecCount;
}
iCurPos = iUsedBlockCount*m_flashInfo.uiSectorPerBlock+(m_flashInfo.usValidSecPerBlock-iValidSecCount);
iRet = m_pComm->RKU_ReadSector(iCurPos, dwReadSector,lpBuffer);
if( iRet!=ERR_SUCCESS )
{
if (m_pLog)
{
m_pLog->Record(_T("ERROR:ReadMutilSector-->RKU_ReadSector failed,RetCode(%d)"),iRet);
}
break;
}
dwCount -= dwReadSector;
iUsedSecCount += dwReadSector;
iValidSecCount -= dwReadSector;
if (iValidSecCount<=0)
{
iUsedBlockCount++;
iValidSecCount = m_flashInfo.usValidSecPerBlock;
}
lpBuffer += dwReadSector * 512;
}
return iRet;
}
bool CRKDevice::EraseMutilBlock(BYTE bFlashCS,DWORD dwPos,DWORD dwCount,bool bForce)
{
DWORD dwTimes = 0;
UCHAR eraseType;
eraseType = bForce?ERASE_FORCE:ERASE_NORMAL;
int iRet;
while(dwCount >= MAX_ERASE_BLOCKS)
{
iRet = m_pComm->RKU_EraseBlock( bFlashCS, dwPos+dwTimes*MAX_ERASE_BLOCKS, MAX_ERASE_BLOCKS, eraseType);
if(ERR_FOUND_BAD_BLOCK == iRet)
{
dwCount -= MAX_ERASE_BLOCKS;
++dwTimes;
}else if(ERR_SUCCESS == iRet)
{
dwCount -= MAX_ERASE_BLOCKS;
++dwTimes;
}
else
{//<2F><><EFBFBD><EFBFBD>ʧ<EFBFBD><CAA7>
if (m_pLog)
{
m_pLog->Record(_T("ERROR:EraseMutilBlock-->RKU_EraseBlock failed,RetCode(%d)"),iRet);
}
return false;
}
}
if(dwCount>0)
{
iRet = m_pComm->RKU_EraseBlock( bFlashCS, dwPos+dwTimes*MAX_ERASE_BLOCKS, dwCount, eraseType);
if(ERR_SUCCESS == iRet)
{
dwCount = 0;
}
else if(ERR_FOUND_BAD_BLOCK == iRet)
{
dwCount = 0;
}
else
{//<2F><><EFBFBD><EFBFBD>ʧ<EFBFBD><CAA7>
if (m_pLog)
{
m_pLog->Record(_T("ERROR:EraseMutilBlock-->RKU_EraseBlock failed,RetCode(%d)"),iRet);
}
return false;
}
}
return true;
}
bool CRKDevice::CheckChip()
{
int iRet;
BYTE bChipInfo[CHIPINFO_LEN];
ENUM_RKDEVICE_TYPE curDeviceType=RKNONE_DEVICE;
memset(bChipInfo,0,CHIPINFO_LEN);
iRet = m_pComm->RKU_ReadChipInfo(bChipInfo);
if (iRet==ERR_SUCCESS)
{
if (!m_chipData)
{
m_chipData = new BYTE[CHIPINFO_LEN];
}
memset(m_chipData,0,CHIPINFO_LEN);
memcpy(m_chipData,bChipInfo,CHIPINFO_LEN);
DWORD *pValue;
pValue = (DWORD *)(&bChipInfo[0]);
if ((ENUM_RKDEVICE_TYPE)(*pValue)==m_device)
{
return true;
}
if (*pValue==0x524B3237)
{
curDeviceType = RK27_DEVICE;
}
else if (*pValue==0x32373341)
{
curDeviceType = RKCAYMAN_DEVICE;
}
else if (*pValue==0x524B3238)
{
curDeviceType = RK28_DEVICE;
}
else if (*pValue==0x32383158)
{
curDeviceType = RK281X_DEVICE;
}
else if (*pValue==0x32383242)
{
curDeviceType = RKPANDA_DEVICE;
}
else if (*pValue==0x32393058)
{
curDeviceType = RK29_DEVICE;
}
else if (*pValue==0x32393258)
{
curDeviceType = RK292X_DEVICE;
}
else if (*pValue==0x33303041)
{
curDeviceType = RK30_DEVICE;
}
else if (*pValue==0x33313041)
{
curDeviceType = RK30B_DEVICE;
}
else if (*pValue==0x33313042)
{
curDeviceType = RK31_DEVICE;
}
else if (*pValue==0x33323041)
{
curDeviceType = RK32_DEVICE;
}
else if (*pValue==0x32363243)
{
curDeviceType = RKSMART_DEVICE;
}
else if (*pValue==0x6E616E6F)
{
curDeviceType = RKNANO_DEVICE;
}
else if (*pValue==0x4E4F5243)
{
curDeviceType = RKCROWN_DEVICE;
}
if (curDeviceType==m_device)
{
return true;
}
else
{
if (m_pLog)
{
m_pLog->Record(_T("ERROR:CheckChip-->Chip is not match,firmware(0x%x),device(0x%x)"),m_device,*pValue);
}
return false;
}
}
else
{
if (m_pLog)
{
m_pLog->Record(_T("ERROR:CheckChip-->RKU_ReadChipInfo failed,RetCode(%d)"),iRet);
}
return false;
}
}
CHAR CRKDevice::FindValidBlocks(char bBegin, char bLen)
{
char bCount = 0;
char bIndex = bBegin;
while(bBegin < IDBLOCK_TOP)
{
if(0 == m_flashInfo.blockState[(int)bBegin++])
++bCount;
else
{
bCount = 0;
bIndex = bBegin;
}
if(bCount >= bLen)
break;
}
if(bBegin >= IDBLOCK_TOP)
bIndex = -1;
return bIndex;
}
USHORT UshortToBCD(USHORT num)
{
USHORT bcd = 0;
bcd = (num % 10) | ( ((num/10 )% 10)<< 4 )|( ((num/100) %10) << 8)|( ((num/1000) %10) << 12);
return bcd;
}
BYTE ByteToBCD(BYTE num)
{
BYTE bcd = 0;
bcd = (num % 10) | ( ((num/10 )% 10)<< 4 );
return bcd;
}
BYTE CRKDevice::RandomByte(BYTE bLowLimit,BYTE bHighLimit)
{
BYTE k;
double d;
d = (double)rand() / ((double)RAND_MAX+1);
k = (BYTE)( d*(bHighLimit-bLowLimit+1) );
return (bLowLimit+k);
}
bool CRKDevice::CheckCrc16(PBYTE pCheckData,USHORT usDataLength,USHORT usOldCrc)
{
USHORT usNewCrc;
usNewCrc = CRC_16(pCheckData,usDataLength);
return (usNewCrc==usOldCrc)?true:false;
}
bool CRKDevice::CheckUid(BYTE uidSize,BYTE *pUid)
{
if (uidSize!=RKDEVICE_UID_LEN)
{
return false;
}
USHORT oldCrc,newCrc;
oldCrc = *(USHORT *)(pUid+RKDEVICE_UID_LEN-2);
newCrc = CRC_CCITT(pUid,RKDEVICE_UID_LEN-2);
if (oldCrc!=newCrc)
{
return false;
}
return true;
}
bool CRKDevice::GetIDBData(UINT uiIDBCount,PBYTE lpBuf,UINT uiSecCount)
{
PBYTE pIDB;
pIDB = new BYTE[uiSecCount*512];
memset( pIDB, 0, uiSecCount*512 );
int i,j,iResult;
int nSrc=-1,nDst=-1;
bool bRet;
for (i=0;i<uiIDBCount;i++)
{
if (nSrc==-1)
{
iResult = ReadMutilSector( m_flashInfo.uiSectorPerBlock*m_idBlockOffset[i], uiSecCount, lpBuf );
if (iResult!=ERR_SUCCESS)
{
if (m_pLog)
{
m_pLog->Record(_T("ERROR:GetIDBData-->RKU_ReadSector failed,RetCode(%d)"),iResult);
}
continue;
}
nSrc = i;
continue;
}
if (nDst==-1)
{
iResult = ReadMutilSector( m_flashInfo.uiSectorPerBlock*m_idBlockOffset[i], uiSecCount, pIDB );
if (iResult!=ERR_SUCCESS)
{
if (m_pLog)
{
m_pLog->Record(_T("ERROR:GetIDBData-->RKU_ReadSector failed,RetCode(%d)"),iResult);
}
continue;
}
nDst = i;
}
bRet = true;
for(j=0;j<uiSecCount;j++)
{
bRet = memcmp(lpBuf+512*j,pIDB+512*j,512)==0;
if (!bRet)
break;
}
if (bRet)
{//<2F><>ͬ
delete []pIDB;
pIDB = NULL;
return true;
}
else
{
if (m_pLog)
{
m_pLog->Record(_T("ERROR:GetIDBData-->memcmp failed,src(%d),Dst(%d)"),nSrc,nDst);
}
memcpy(lpBuf,pIDB,512*uiSecCount);
nSrc = nDst;
nDst = -1;
continue;
}
}
delete []pIDB;
pIDB = NULL;
if (nSrc!=-1)
{
return true;
}
return false;
}
bool CRKDevice::GetWriteBackData(UINT uiIDBCount,PBYTE lpBuf)
{
bool bRet;
bRet = GetIDBData(uiIDBCount,lpBuf,4);
return bRet;
}