220 lines
7.3 KiB
C++
220 lines
7.3 KiB
C++
#ifndef RKCOMM_HEADER
|
|
#define RKCOMM_HEADER
|
|
#include "DefineHeader.h"
|
|
typedef enum
|
|
{
|
|
USB_BULK_READ = 0,
|
|
USB_BULK_WRITE,
|
|
USB_CONTROL,
|
|
} USB_ACCESS_TYPE;
|
|
typedef enum
|
|
{
|
|
TU_NONE_SUBCODE = 0,
|
|
TU_ERASESYSTEM_SUBCODE = 0xFE,
|
|
TU_LOWERFORMAT_SUBCODE = 0xFD,
|
|
TU_ERASEUSERDATA_SUBCODE = 0xFB,
|
|
TU_GETUSERSECTOR_SUBCODE = 0xF9
|
|
} TESTUNIT_SUBCODE;
|
|
typedef enum
|
|
{
|
|
RST_NONE_SUBCODE = 0,
|
|
RST_RESETMSC_SUBCODE,
|
|
RST_POWEROFF_SUBCODE,
|
|
RST_RESETMASKROM_SUBCODE,
|
|
RST_DISCONNECTRESET_SUBCODE
|
|
} RESET_SUBCODE;
|
|
typedef enum
|
|
{
|
|
RWMETHOD_IMAGE = 0,
|
|
RWMETHOD_LBA,
|
|
} RW_SUBCODE;
|
|
/* ÃüÁî */
|
|
typedef enum
|
|
{
|
|
TEST_UNIT_READY = 0,
|
|
READ_FLASH_ID = 0x01,
|
|
TEST_BAD_BLOCK = 0x03,
|
|
READ_SECTOR = 0x04,
|
|
WRITE_SECTOR = 0x05,
|
|
ERASE_NORMAL = 0x06,
|
|
ERASE_FORCE = 0x0B,
|
|
READ_LBA = 0x14,
|
|
WRITE_LBA = 0x15,
|
|
ERASE_SYSTEMDISK = 0x16,
|
|
READ_SDRAM = 0x17,
|
|
WRITE_SDRAM = 0x18,
|
|
EXECUTE_SDRAM = 0x19,
|
|
READ_FLASH_INFO = 0x1A,
|
|
READ_CHIP_INFO = 0x1B,
|
|
SET_RESET_FLAG = 0x1E,
|
|
WRITE_EFUSE = 0x1F,
|
|
READ_EFUSE = 0x20,
|
|
READ_SPI_FLASH = 0x21,
|
|
WRITE_SPI_FLASH = 0x22,
|
|
SWITCH_MASKROM = 0xFE,
|
|
DEVICE_RESET = 0xFF
|
|
} USB_OPERATION_CODE;
|
|
|
|
#pragma pack(1)
|
|
|
|
typedef struct
|
|
{
|
|
BYTE ucOperCode;
|
|
BYTE ucReserved;
|
|
DWORD dwAddress;
|
|
BYTE ucReserved2;
|
|
USHORT usLength;
|
|
BYTE ucReserved3[7];
|
|
} CBWCB, *PCBWCB;
|
|
|
|
typedef struct
|
|
{
|
|
DWORD dwCBWSignature;
|
|
DWORD dwCBWTag;
|
|
DWORD dwCBWTransferLength;
|
|
BYTE ucCBWFlags;
|
|
BYTE ucCBWLUN;
|
|
BYTE ucCBWCBLength;
|
|
CBWCB cbwcb;
|
|
} CBW, *PCBW;
|
|
|
|
typedef struct
|
|
{
|
|
DWORD dwCSWSignature;
|
|
DWORD dwCSWTag;
|
|
DWORD dwCBWDataResidue;
|
|
BYTE ucCSWStatus;
|
|
} CSW, *PCSW;
|
|
typedef struct
|
|
{
|
|
UINT uiSize;
|
|
UINT uiCrc;
|
|
UINT uiBlock[5];
|
|
} STRUCT_END_WRITE_SECTOR, *PSTRUCT_END_WRITE_SECTOR;
|
|
|
|
#pragma pack()
|
|
#define NAND_DRIVER_DEV "/dev/rknand_sys_storage"
|
|
#define NAND_DRIVER_DEV_VENDOR "/dev/rknand_sys_storage"
|
|
#define NAND_DRIVER_DEV_LBA "/dev/rkflash0"
|
|
#define EMMC_DRIVER_DEV "/dev/rknand_sys_storage"
|
|
#define EMMC_DRIVER_DEV_VENDOR "/dev/vendor_storage"
|
|
//#define EMMC_DRIVER_DEV_LBA "/dev/block/mmcblk0"
|
|
#define EMMC_DRIVER_DEV_LBA "/dev/mmcblk0"
|
|
#define EMMC_POINT_NAME "emmc_point_name"
|
|
|
|
#define READ_SECTOR_IO _IOW('r', READ_SECTOR, unsigned int)
|
|
#define WRITE_SECTOR_IO _IOW('r', WRITE_SECTOR, unsigned int)
|
|
#define READ_LBA_IO _IOW('r', READ_LBA, unsigned int)
|
|
#define WRITE_LBA_IO _IOW('r', WRITE_LBA, unsigned int)
|
|
#define START_WRITE_SECTOR_IO _IOW('r', 0x51, unsigned int)
|
|
#define END_WRITE_SECTOR_IO _IOW('r', 0x52, unsigned int)
|
|
#define GET_FLASH_INFO_IO _IOW('r', READ_FLASH_INFO, unsigned int)
|
|
#define GET_BAD_BLOCK_IO _IOW('r', TEST_BAD_BLOCK, unsigned int)
|
|
#define GET_LOCK_FLAG_IO _IOW('r', 0x53, unsigned int)
|
|
#define GET_PUBLIC_KEY_IO _IOW('r', 0x54, unsigned int)
|
|
|
|
#define DISABLE_NAND_LBA_WRITE_IO _IO('V',0)
|
|
#define ENABLE_NAND_LBA_WRITE_IO _IO('V',1)
|
|
#define DISABLE_NAND_LBA_READ_IO _IO('V',2)
|
|
#define ENABLE_NAND_LBA_READ_IO _IO('V',3)
|
|
|
|
|
|
#define CMD_TIMEOUT 0
|
|
|
|
#define CBW_SIGN 0x43425355 /* "USBC" */
|
|
#define CSW_SIGN 0x53425355 /* "USBS" */
|
|
|
|
|
|
#define DIRECTION_OUT 0x00
|
|
#define DIRECTION_IN 0x80
|
|
#define MAX_TEST_BLOCKS 512
|
|
#define MAX_ERASE_BLOCKS 128
|
|
#define MAX_CLEAR_LEN 16*1024
|
|
|
|
#ifndef ERR_SUCCESS
|
|
#define ERR_SUCCESS 0
|
|
#endif
|
|
#define ERR_DEVICE_READY 0
|
|
#define ERR_DEVICE_OPEN_FAILED -1
|
|
#define ERR_CSW_OPEN_FAILED -2
|
|
#define ERR_DEVICE_WRITE_FAILED -3
|
|
#define ERR_DEVICE_READ_FAILED -4
|
|
#define ERR_CMD_NOTMATCH -5
|
|
#define ERR_DEVICE_UNREADY -6
|
|
#define ERR_FOUND_BAD_BLOCK -7
|
|
#define ERR_FAILED -8
|
|
#define ERR_CROSS_BORDER -9
|
|
#define ERR_DEVICE_NOT_SUPPORT -10
|
|
#define ERR_REQUEST_NOT_SUPPORT -11
|
|
#define ERR_REQUEST_FAIL -12
|
|
#define ERR_BUFFER_NOT_ENOUGH -13
|
|
|
|
#define UFI_CHECK_SIGN(cbw, csw) ((CSW_SIGN == (csw).dwCSWSignature) && ((csw).dwCSWTag == (cbw).dwCBWTag))
|
|
|
|
class CRKLog;
|
|
class CRKComm
|
|
{
|
|
public:
|
|
virtual int RKU_EraseBlock(BYTE ucFlashCS, DWORD dwPos, DWORD dwCount, BYTE ucEraseType) = 0;
|
|
virtual int RKU_ReadChipInfo(BYTE *lpBuffer) = 0;
|
|
virtual int RKU_ReadFlashID(BYTE *lpBuffer) = 0;
|
|
virtual int RKU_ReadFlashInfo(BYTE *lpBuffer, UINT *puiRead = NULL) = 0;
|
|
virtual int RKU_ReadLBA(DWORD dwPos, DWORD dwCount, BYTE *lpBuffer, BYTE bySubCode = RWMETHOD_IMAGE) = 0;
|
|
virtual int RKU_ReadSector(DWORD dwPos, DWORD dwCount, BYTE *lpBuffer) = 0;
|
|
virtual int RKU_ResetDevice(BYTE bySubCode = RST_NONE_SUBCODE) = 0;
|
|
virtual int RKU_TestBadBlock(BYTE ucFlashCS, DWORD dwPos, DWORD dwCount, BYTE *lpBuffer) = 0;
|
|
virtual int RKU_TestDeviceReady(DWORD *dwTotal = NULL, DWORD *dwCurrent = NULL, BYTE bySubCode = TU_NONE_SUBCODE) = 0;
|
|
virtual int RKU_WriteLBA(DWORD dwPos, DWORD dwCount, BYTE *lpBuffer, BYTE bySubCode = RWMETHOD_IMAGE) = 0;
|
|
virtual int RKU_LoaderWriteLBA(DWORD dwPos, DWORD dwCount, BYTE *lpBuffer, BYTE bySubCode = RWMETHOD_IMAGE) = 0;
|
|
virtual int RKU_WriteSector(DWORD dwPos, DWORD dwCount, BYTE *lpBuffer) = 0;
|
|
virtual int RKU_EndWriteSector(BYTE *lpBuffer) = 0;
|
|
virtual int RKU_GetLockFlag(BYTE *lpBuffer) = 0;
|
|
virtual int RKU_GetPublicKey(BYTE *lpBuffer) = 0;
|
|
virtual void RKU_ReopenLBAHandle() = 0;
|
|
virtual int RKU_ShowNandLBADevice() = 0;
|
|
virtual bool RKU_IsEmmcFlash() = 0;
|
|
CRKComm(CRKLog *pLog);
|
|
virtual ~CRKComm();
|
|
protected:
|
|
CRKLog *m_log;
|
|
bool m_bEmmc;
|
|
int m_hDev;
|
|
int m_hLbaDev;
|
|
private:
|
|
|
|
};
|
|
class CRKUsbComm: public CRKComm
|
|
{
|
|
public:
|
|
virtual int RKU_EraseBlock(BYTE ucFlashCS, DWORD dwPos, DWORD dwCount, BYTE ucEraseType);
|
|
virtual int RKU_ReadChipInfo(BYTE *lpBuffer);
|
|
virtual int RKU_ReadFlashID(BYTE *lpBuffer);
|
|
virtual int RKU_ReadFlashInfo(BYTE *lpBuffer, UINT *puiRead = NULL);
|
|
virtual int RKU_ReadLBA(DWORD dwPos, DWORD dwCount, BYTE *lpBuffer, BYTE bySubCode = RWMETHOD_IMAGE);
|
|
virtual int RKU_ReadSector(DWORD dwPos, DWORD dwCount, BYTE *lpBuffer);
|
|
virtual int RKU_ResetDevice(BYTE bySubCode = RST_NONE_SUBCODE);
|
|
virtual int RKU_TestBadBlock(BYTE ucFlashCS, DWORD dwPos, DWORD dwCount, BYTE *lpBuffer);
|
|
virtual int RKU_TestDeviceReady(DWORD *dwTotal = NULL, DWORD *dwCurrent = NULL, BYTE bySubCode = TU_NONE_SUBCODE);
|
|
virtual int RKU_WriteLBA(DWORD dwPos, DWORD dwCount, BYTE *lpBuffer, BYTE bySubCode = RWMETHOD_IMAGE);
|
|
virtual int RKU_LoaderWriteLBA(DWORD dwPos, DWORD dwCount, BYTE *lpBuffer, BYTE bySubCode = RWMETHOD_IMAGE);
|
|
virtual int RKU_WriteSector(DWORD dwPos, DWORD dwCount, BYTE *lpBuffer);
|
|
virtual int RKU_EndWriteSector(BYTE *lpBuffer);
|
|
virtual int RKU_GetLockFlag(BYTE *lpBuffer);
|
|
virtual int RKU_GetPublicKey(BYTE *lpBuffer);
|
|
virtual void RKU_ReopenLBAHandle();
|
|
virtual int RKU_ShowNandLBADevice();
|
|
virtual bool RKU_IsEmmcFlash();
|
|
CRKUsbComm(CRKLog *pLog);
|
|
~CRKUsbComm();
|
|
|
|
protected:
|
|
STRUCT_RKDEVICE_DESC m_deviceDesc;
|
|
CRKLog *m_log;
|
|
private:
|
|
bool CtrlNandLbaWrite(bool bEnable = true);
|
|
bool CtrlNandLbaRead(bool bEnable = true);
|
|
|
|
};
|
|
|
|
#endif
|