#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;
|
}
|
