2025-05-10 21:49:39 +08:00

727 lines
18 KiB
C++

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include "RKComm.h"
#include "RKLog.h"
#include "RKAndroidDevice.h"
CRKComm::CRKComm(CRKLog *pLog)
{
m_log = pLog;
m_bEmmc = false;
m_hDev = m_hLbaDev = -1;
}
CRKComm::~CRKComm()
{
}
CRKUsbComm::CRKUsbComm(CRKLog *pLog): CRKComm(pLog)
{
//char bootmode[100];
//property_get("ro.bootmode", bootmode, "unknown");
//if(!strcmp(bootmode, "emmc"))
// m_bEmmc = true;
//else
// m_bEmmc = false;
char *emmc_point = getenv(EMMC_POINT_NAME);
m_hLbaDev = open(emmc_point, O_RDWR | O_SYNC, 0);
if (m_hLbaDev < 0)
{
if (pLog)
{
pLog->Record(_T("INFO:is nand devices..."));
}
m_bEmmc = false;
}
else
{
if (pLog)
{
pLog->Record(_T("INFO:is emmc devices..."));
}
m_bEmmc = true;
close(m_hLbaDev);
}
m_log = pLog;
if (m_bEmmc)
{
if (pLog)
{
pLog->Record(_T("INFO:CRKUsbComm-->is emmc."));
}
m_hDev = open(EMMC_DRIVER_DEV_VENDOR, O_RDWR, 0);
if (m_hDev < 0)
{
if (pLog)
{
pLog->Record(_T("ERROR:CRKUsbComm-->open %s failed,err=%s"), EMMC_DRIVER_DEV_VENDOR, strerror(errno));
pLog->Record(_T("ERROR:CRKUsbComm-->try to read %s."), EMMC_DRIVER_DEV);
}
m_hDev = open(EMMC_DRIVER_DEV, O_RDWR, 0);
if (m_hDev < 0)
{
if (pLog)
{
pLog->Record(_T("ERROR:CRKUsbComm-->open %s failed,err=%s"), EMMC_DRIVER_DEV, strerror(errno));
pLog->Record(_T("ERROR:CRKUsbComm-->please to check drmboot.ko."));
}
}
else
{
if (pLog)
{
pLog->Record(_T("INFO:CRKUsbComm-->%s=%d"), EMMC_DRIVER_DEV, m_hDev);
}
}
}
else
{
if (pLog)
{
pLog->Record(_T("INFO:CRKUsbComm-->%s=%d"), EMMC_DRIVER_DEV_VENDOR, m_hDev);
}
}
//get EMMC_DRIVER_DEV_LBA from
m_hLbaDev = open(emmc_point, O_RDWR | O_SYNC, 0);
if (m_hLbaDev < 0)
{
if (pLog)
{
pLog->Record(_T("ERROR:CRKUsbComm-->open %s failed,err=%d"), emmc_point, errno);
}
}
else
{
if (pLog)
{
pLog->Record(_T("INFO:CRKUsbComm-->%s=%d"), emmc_point, m_hLbaDev);
}
}
}
else
{
if (pLog)
{
pLog->Record(_T("INFO:CRKUsbComm-->is nand."));
}
m_hDev = open(NAND_DRIVER_DEV_VENDOR, O_RDWR, 0);
if (m_hDev < 0)
{
if (pLog)
{
pLog->Record(_T("ERROR:CRKUsbComm-->open %s failed,err=%d"), NAND_DRIVER_DEV_VENDOR, strerror(errno));
pLog->Record(_T("ERROR:CRKUsbComm-->try to read from %s."), NAND_DRIVER_DEV_VENDOR);
}
m_hDev = open(NAND_DRIVER_DEV, O_RDWR, 0);
if (pLog)
{
pLog->Record(_T("ERROR:CRKUsbComm-->open %s failed,err=%d"), NAND_DRIVER_DEV, strerror(errno));
}
else
{
if (pLog)
{
pLog->Record(_T("INFO:CRKUsbComm-->%s=%d"), NAND_DRIVER_DEV, m_hDev);
}
}
}
else
{
if (pLog)
{
pLog->Record(_T("INFO:CRKUsbComm-->%s=%d"), NAND_DRIVER_DEV_VENDOR, m_hDev);
}
}
}
}
void CRKUsbComm::RKU_ReopenLBAHandle()
{
if (m_bEmmc)
{
return;
}
if (m_hLbaDev > 0)
{
close(m_hLbaDev);
m_hLbaDev = -1;
}
// if (m_bEmmc)
// {
// m_hLbaDev= open(EMMC_DRIVER_DEV_LBA,O_RDWR|O_SYNC,0);
// if (m_hLbaDev<0)
// {
// if (m_log)
// m_log->Record(_T("ERROR:RKU_ReopenLBAHandle-->open %s failed,err=%d"),EMMC_DRIVER_DEV_LBA,errno);
// }
// else
// {
// if (m_log)
// m_log->Record(_T("INFO:RKU_ReopenLBAHandle-->%s=%d"),EMMC_DRIVER_DEV_LBA,m_hLbaDev);
// }
// }
// else
// {
m_hLbaDev = open(NAND_DRIVER_DEV_LBA, O_RDWR | O_SYNC, 0);
if (m_hLbaDev < 0)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_ReopenLBAHandle-->open %s failed,err=%d"), NAND_DRIVER_DEV_LBA, errno);
}
}
else
{
if (m_log)
{
m_log->Record(_T("INFO:RKU_ReopenLBAHandle-->%s=%d"), NAND_DRIVER_DEV_LBA, m_hLbaDev);
}
}
// }
}
int CRKUsbComm::RKU_ShowNandLBADevice()
{
if (m_bEmmc)
{
return ERR_SUCCESS;
}
BYTE blockState[64];
memset(blockState, 0, 64);
int iRet;
iRet = RKU_TestBadBlock(0, 0, MAX_TEST_BLOCKS, blockState);
if (iRet != ERR_SUCCESS)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_ShowNandLBADevice-->RKU_TestBadBlock failed,ret=%d"), iRet);
}
}
return iRet;
}
bool CRKUsbComm::RKU_IsEmmcFlash()
{
return m_bEmmc ? true : false;
}
CRKUsbComm::~CRKUsbComm()
{
if (m_hDev > 0)
{
close(m_hDev);
}
if (m_hLbaDev > 0)
{
// if (!m_bEmmc)
// {
// CtrlNandLbaRead(false);
// CtrlNandLbaWrite(false);
// }
close(m_hLbaDev);
}
}
int CRKUsbComm::RKU_EraseBlock(BYTE ucFlashCS, DWORD dwPos, DWORD dwCount, BYTE ucEraseType)
{
return ERR_SUCCESS;
}
int CRKUsbComm::RKU_ReadChipInfo(BYTE *lpBuffer)
{
return ERR_SUCCESS;
}
int CRKUsbComm::RKU_ReadFlashID(BYTE *lpBuffer)
{
return ERR_SUCCESS;
}
void rknand_print_hex_data(char *s, unsigned int *buf, unsigned int len)
{
unsigned int i, j, count;
printf("%s\n", s);
for (i = 0; i < len; i += 4)
{
printf("%08x %08x %08x %08x\n", buf[i], buf[i + 1], buf[i + 2], buf[i + 3]);
}
}
int CRKUsbComm::RKU_ReadFlashInfo(BYTE *lpBuffer, UINT *puiRead)
{
long long ret;
#if 0 //close by chad.ma
if (m_hDev < 0)
{
return ERR_DEVICE_OPEN_FAILED;
}
ret = ioctl(m_hDev, GET_FLASH_INFO_IO, lpBuffer);
if (ret)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_ReadFlashInfo ioctl failed,err=%d"), errno);
}
return ERR_FAILED;
}
*puiRead = 11;
#else
/////////////////////////////////////////////////////////////////
// get flashsize directly
m_log->Record(_T("INFO: m_bEmmc = %d, m_hLbaDev = %d"), m_bEmmc, m_hLbaDev);
if (m_hLbaDev < 0)
{
m_hLbaDev = open(NAND_DRIVER_DEV_LBA, O_RDWR | O_SYNC, 0);
if (m_hLbaDev < 0)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_ReadFlashInfo-->open %s failed,err=%d"), NAND_DRIVER_DEV_LBA, errno);
}
return ERR_FAILED;
}
else
{
if (m_log)
{
m_log->Record(_T("INFO:RKU_ReadFlashInfo-->open %s ok,handle=%d"), NAND_DRIVER_DEV_LBA, m_hLbaDev);
}
ret = lseek64(m_hLbaDev, 0, SEEK_END);
if (ret < 0)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_ReadFlashInfo-->get %s file length fail"), NAND_DRIVER_DEV_LBA);
}
return ERR_FAILED;
}
else
{
char str[20] = {0};
lseek64(m_hLbaDev, 0, SEEK_SET); //reset the cfo to begin
snprintf(str, sizeof(str), "%d", ret / 1024);
*(UINT *)lpBuffer = (ret / 1024);
}
}
}
else
{
ret = lseek64(m_hLbaDev, 0, SEEK_END);
m_log->Record(_T("INFO: lseek64 result = %lld"), ret);
if (ret < 0)
{
if (m_log)
{
if (m_bEmmc)
m_log->Record(_T("ERROR:RKU_ReadFlashInfo-->get %s file length fail"),
getenv(EMMC_POINT_NAME));
else
m_log->Record(_T("ERROR:RKU_ReadFlashInfo-->get %s file length fail"),
NAND_DRIVER_DEV_LBA);
}
return ERR_FAILED;
}
else
{
char str[20] = {0};
lseek64(m_hLbaDev, 0, SEEK_SET); //reset the cfo to begin
snprintf(str, sizeof(str), "%d", ret / 1024);
*(UINT *)lpBuffer = (ret / 1024);
}
}
#endif
return ERR_SUCCESS;
}
int CRKUsbComm::RKU_ReadLBA(DWORD dwPos, DWORD dwCount, BYTE *lpBuffer, BYTE bySubCode)
{
long long ret;
long long dwPosBuf;
if (m_hLbaDev < 0)
{
if (!m_bEmmc)
{
m_hLbaDev = open(NAND_DRIVER_DEV_LBA, O_RDWR | O_SYNC, 0);
if (m_hLbaDev < 0)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_ReadLBA-->open %s failed,err=%d"), NAND_DRIVER_DEV_LBA, errno);
}
return ERR_DEVICE_OPEN_FAILED;
}
else
{
if (m_log)
{
m_log->Record(_T("INFO:RKU_ReadLBA-->open %s ok,handle=%d"), NAND_DRIVER_DEV_LBA, m_hLbaDev);
}
}
}
else
{
return ERR_DEVICE_OPEN_FAILED;
}
}
if (m_bEmmc && !CRKAndroidDevice::bGptFlag)
{
dwPos += 8192;
}
dwPosBuf = dwPos;
ret = lseek64(m_hLbaDev, (off64_t)dwPosBuf * 512, SEEK_SET);
if (ret < 0)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_ReadLBA seek failed,err=%d,ret=%lld."), errno, ret);
m_log->Record(_T("the dwPosBuf = dwPosBuf*512,dwPosBuf:%lld!"), dwPosBuf * 512);
}
return ERR_FAILED;
}
ret = read(m_hLbaDev, lpBuffer, dwCount * 512);
if (ret != dwCount * 512)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_ReadLBA read failed,err=%d"), errno);
}
return ERR_FAILED;
}
return ERR_SUCCESS;
}
int CRKUsbComm::RKU_ReadSector(DWORD dwPos, DWORD dwCount, BYTE *lpBuffer)
{
int ret;
if (m_hDev < 0)
{
return ERR_DEVICE_OPEN_FAILED;
}
DWORD *pOffsetSec = (DWORD *)(lpBuffer);
DWORD *pCountSec = (DWORD *)(lpBuffer + 4);
*pOffsetSec = dwPos;
*pCountSec = dwCount;
ret = ioctl(m_hDev, READ_SECTOR_IO, lpBuffer);
if (ret)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_ReadSector failed,err=%d"), errno);
}
return ERR_FAILED;
}
return ERR_SUCCESS;
}
int CRKUsbComm::RKU_ResetDevice(BYTE bySubCode)
{
return ERR_SUCCESS;
}
int CRKUsbComm::RKU_TestBadBlock(BYTE ucFlashCS, DWORD dwPos, DWORD dwCount, BYTE *lpBuffer)
{
int ret;
if (m_hDev < 0)
{
return ERR_DEVICE_OPEN_FAILED;
}
ret = ioctl(m_hDev, GET_BAD_BLOCK_IO, lpBuffer);
if (ret)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_TestBadBlock failed,err=%d"), errno);
}
return ERR_FAILED;
}
if (m_log)
{
string strOutput;
m_log->PrintBuffer(strOutput, lpBuffer, 64);
m_log->Record(_T("INFO:BadBlockState:\r\n%s"), strOutput.c_str());
}
return ERR_SUCCESS;
}
int CRKUsbComm::RKU_TestDeviceReady(DWORD *dwTotal, DWORD *dwCurrent, BYTE bySubCode)
{
return ERR_DEVICE_READY;
}
int CRKUsbComm::RKU_WriteLBA(DWORD dwPos, DWORD dwCount, BYTE *lpBuffer, BYTE bySubCode)
{
long long ret;
long long dwPosBuf;
if (m_hLbaDev < 0)
{
if (!m_bEmmc)
{
m_hLbaDev = open(NAND_DRIVER_DEV_LBA, O_RDWR | O_SYNC, 0);
if (m_hLbaDev < 0)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_WriteLBA-->open %s failed,err=%d"), NAND_DRIVER_DEV_LBA, errno);
}
return ERR_DEVICE_OPEN_FAILED;
}
else
{
if (m_log)
{
m_log->Record(_T("INFO:RKU_WriteLBA-->open %s ok,handle=%d"), NAND_DRIVER_DEV_LBA, m_hLbaDev);
}
}
}
else
{
return ERR_DEVICE_OPEN_FAILED;
}
}
if (m_bEmmc && !CRKAndroidDevice::bGptFlag)
{
dwPos += 8192;
}
dwPosBuf = dwPos;
ret = lseek64(m_hLbaDev, (off64_t)dwPosBuf * 512, SEEK_SET);
if (ret < 0)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_WriteLBA seek failed,err=%d,ret:%lld"), errno, ret);
m_log->Record(_T("the dwPosBuf = dwPosBuf*512,dwPosBuf:%lld!"), dwPosBuf * 512);
}
return ERR_FAILED;
}
ret = write(m_hLbaDev, lpBuffer, dwCount * 512);
if (ret != dwCount * 512)
{
sleep(1);
if (m_log)
{
m_log->Record(_T("ERROR:RKU_WriteLBA write failed,err=%d"), errno);
}
return ERR_FAILED;
}
return ERR_SUCCESS;
}
int CRKUsbComm::RKU_LoaderWriteLBA(DWORD dwPos, DWORD dwCount, BYTE *lpBuffer, BYTE bySubCode)
{
long long ret;
long long dwPosBuf;
if (m_hLbaDev < 0)
{
if (!m_bEmmc)
{
m_hLbaDev = open(NAND_DRIVER_DEV_LBA, O_RDWR | O_SYNC, 0);
if (m_hLbaDev < 0)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_WriteLBA-->open %s failed,err=%d"), NAND_DRIVER_DEV_LBA, errno);
}
return ERR_DEVICE_OPEN_FAILED;
}
else
{
if (m_log)
{
m_log->Record(_T("INFO:RKU_WriteLBA-->open %s ok,handle=%d"), NAND_DRIVER_DEV_LBA, m_hLbaDev);
}
}
}
else
{
return ERR_DEVICE_OPEN_FAILED;
}
}
dwPosBuf = dwPos;
//if (m_log)
// m_log->Record(_T("INFO: dwPosBuf = %d ,will seek to pos = 0x%08x"), dwPosBuf, dwPosBuf*512);
ret = lseek64(m_hLbaDev, (off64_t)dwPosBuf * 512, SEEK_SET);
if (ret < 0)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_WriteLBA seek failed,err=%d,ret:%lld"), errno, ret);
m_log->Record(_T("the dwPosBuf = dwPosBuf*512,dwPosBuf:%lld!"), dwPosBuf * 512);
}
return ERR_FAILED;
}
ret = write(m_hLbaDev, lpBuffer, dwCount * 512);
if (ret != dwCount * 512)
{
sleep(1);
if (m_log)
{
m_log->Record(_T("ERROR:RKU_WriteLBA write failed,err=%d"), errno);
}
return ERR_FAILED;
}
return ERR_SUCCESS;
}
int CRKUsbComm::RKU_WriteSector(DWORD dwPos, DWORD dwCount, BYTE *lpBuffer)
{
int ret;
if (m_hDev < 0)
{
return ERR_DEVICE_OPEN_FAILED;
}
DWORD *pOffset = (DWORD *)(lpBuffer);
DWORD *pCount = (DWORD *)(lpBuffer + 4);
*pOffset = dwPos;
*pCount = dwCount;
ret = ioctl(m_hDev, WRITE_SECTOR_IO, lpBuffer);
if (ret)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_WriteSector failed,err=%d"), errno);
}
return ERR_FAILED;
}
return ERR_SUCCESS;
}
int CRKUsbComm::RKU_EndWriteSector(BYTE *lpBuffer)
{
int ret;
if (m_hDev < 0)
{
return ERR_DEVICE_OPEN_FAILED;
}
ret = ioctl(m_hDev, END_WRITE_SECTOR_IO, lpBuffer);
if (ret)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_EndWriteSector failed,err=%d"), errno);
}
return ERR_FAILED;
}
return ERR_SUCCESS;
}
int CRKUsbComm::RKU_GetLockFlag(BYTE *lpBuffer)
{
int ret;
if (m_hDev < 0)
{
return ERR_DEVICE_OPEN_FAILED;
}
ret = ioctl(m_hDev, GET_LOCK_FLAG_IO, lpBuffer);
if (ret)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_GetLockFlag failed,err=%d"), errno);
}
return ERR_FAILED;
}
DWORD *pFlag = (DWORD *)lpBuffer;
if (m_log)
{
m_log->Record(_T("INFO:LockFlag:0x%08x"), *pFlag);
}
return ERR_SUCCESS;
}
int CRKUsbComm::RKU_GetPublicKey(BYTE *lpBuffer)
{
int ret;
if (m_hDev < 0)
{
return ERR_DEVICE_OPEN_FAILED;
}
ret = ioctl(m_hDev, GET_PUBLIC_KEY_IO, lpBuffer);
if (ret)
{
if (m_log)
{
m_log->Record(_T("ERROR:RKU_GetPublicKey failed,err=%d"), errno);
}
return ERR_FAILED;
}
return ERR_SUCCESS;
}
bool CRKUsbComm::CtrlNandLbaWrite(bool bEnable)
{
int ret;
if (m_bEmmc)
{
return false;
}
if (m_hLbaDev < 0)
{
return false;
}
if (bEnable)
{
ret = ioctl(m_hLbaDev, ENABLE_NAND_LBA_WRITE_IO);
}
else
{
ret = ioctl(m_hLbaDev, DISABLE_NAND_LBA_WRITE_IO);
}
if (ret)
{
if (m_log)
{
m_log->Record(_T("ERROR:CtrlNandLbaWrite failed,enable=%d,err=%d"), bEnable, errno);
}
return false;
}
return true;
}
bool CRKUsbComm::CtrlNandLbaRead(bool bEnable)
{
int ret;
if (m_bEmmc)
{
return false;
}
if (m_hLbaDev < 0)
{
return false;
}
if (bEnable)
{
ret = ioctl(m_hLbaDev, ENABLE_NAND_LBA_READ_IO);
}
else
{
ret = ioctl(m_hLbaDev, DISABLE_NAND_LBA_READ_IO);
}
if (ret)
{
if (m_log)
{
m_log->Record(_T("ERROR:CtrlNandLbaRead failed,enable=%d,err=%d"), bEnable, errno);
}
return false;
}
return true;
}