#ifndef RKDEVICE_HEADER
#define RKDEVICE_HEADER
#include "RKImage.h"
#include "RKComm.h"
#include "RKLog.h"
#include "DefineHeader.h"



#define RK28_SEC3_RESERVED_LEN 419
#define RKANDROID_SEC3_RESERVED_LEN 419

#define RK28_SEC2_RESERVED_LEN 473
#define RKANDROID_SEC2_RESERVED_LEN 473


#define RKDEVICE_CUSTOMDATA_LEN 512
#define RKDEVICE_SN_LEN 30
#define RKDEVICE_UID_LEN 30
#define RKDEVICE_MAC_LEN 6
#define RKDEVICE_BT_LEN 6
#define RKDEVICE_IMEI_LEN 15

#define SECTOR_SIZE 512
#define SPARE_SIZE 16
#define IDB_BLOCKS 5
#define CHIPINFO_LEN 16
#define IDBLOCK_TOP 50
#define WBBUFFER_BOTTOM 5
#define WBBUFFER_TOP 10
#define CALC_UNIT(a, b)     ((a>0)?((a-1)/b+1):(a))
#define BYTE2SECTOR(x)      (CALC_UNIT(x, SECTOR_SIZE))
#define PAGEALIGN(x)        (CALC_UNIT(x, 4))
//#define MAX_TEST_BLOCKS       512
//#define MAX_ERASE_BLOCKS  128
#define MAX_WRITE_SECTOR    16
#define IOCTL_TRANSFER_SIZE     4096

#define CRC_TAG "CRC"
#pragma pack(1)
typedef struct _STRUCT_FLASH_INFO
{
    tchar   szManufacturerName[16];
    UINT    uiFlashSize;  //MB
    USHORT  usBlockSize;//KB
    UINT    uiPageSize;   //KB
    UINT    uiSectorPerBlock;
    BYTE    blockState[IDBLOCK_TOP];
    UINT    uiBlockNum;
    BYTE    bECCBits;
    BYTE    bAccessTime;  //Á½´Î·ÃÎÊFlashµÄ¼ä¸ôʱ¼ä
    BYTE    bFlashCS;  // FlashƬѡ(FlashƬѡ´æÔÚÖÃ1£¬·ñÔòÖÃ0)
    USHORT  usValidSecPerBlock;//ÿ¿é¿ÉÒÔʹÓõÄÉÈÇøÊý=¿é´óС/Ò³´óС*4
    USHORT  usPhyBlokcPerIDB;//ÿ¸öIDBlockÕ¼ÓõÄÎïÀí¿éÊýÁ¿
    UINT    uiSecNumPerIDB;//ÿ¸öIDBlockÕ¼ÓõÄÉÈÇøÊý
} STRUCT_FLASH_INFO, *PSTRUCT_FLASH_INFO;
typedef struct _STRUCT_FLASHINFO_CMD
{
    UINT    uiFlashSize;    // Flash´óС£¨ÒÔSectorΪµ¥Î»£©
    USHORT  usBlockSize;    // ÎïÀíµÄBlock´óС£¨ÒÔSectorΪµ¥Î»£©
    BYTE    bPageSize;      // ÎïÀíµÄPage´óС£¨ÒÔSectorΪµ¥Î»£©
    BYTE    bECCBits;       // 8/14
    BYTE    bAccessTime;    // Á½´Î·ÃÎÊFlashµÄ¼ä¸ôʱ¼ä
    BYTE    bManufCode;     // ³§ÉÌʶ±ðÂë
    BYTE    bFlashCS;       // FlashƬѡ(ÈôFlashƬѡ´æÔÚ£¬Ôò½«ÏàÓ¦µÄBitÖÃ1£¬·ñÔòÖÃ0)
    BYTE    reserved[501];
} STRUCT_FLASHINFO_CMD, *PSTRUCT_FLASHINFO_CMD;
typedef struct
{
    BYTE bFlashCS;
    UINT uiBlockNum;
    USHORT usBlockStateSize;
    PBYTE pBlockStateData;
} STRUCT_BLOCK_STATE, *PSTRUCT_BLOCK_STATE;
#pragma pack()
typedef struct
{
    USHORT usVid;
    USHORT usPid;
} STRUCT_DEVICE_PROP;
typedef vector<STRUCT_DEVICE_PROP> DEVICE_PROP_SET;

class CRKDevice
{
public:

    ENUM_OS_TYPE GetOsType();
    void SetOsType(ENUM_OS_TYPE value);
    property<CRKDevice, ENUM_OS_TYPE, READ_WRITE> OsType;

    void SetUid(PBYTE value);
    property<CRKDevice, PBYTE, WRITE_ONLY> Uid;

    void SetPrepareEraseFlag(bool value);
    property<CRKDevice, bool, WRITE_ONLY> PrepareEraseFlag;

    void SetWorkFlow(UINT value);
    property<CRKDevice, UINT, WRITE_ONLY> WorkFlow;

    void SetMiscModifyFlag(ENUM_MISC_MODIFY_FLAG value);
    property<CRKDevice, ENUM_MISC_MODIFY_FLAG, WRITE_ONLY> MiscModifyFlag;

    CRKLog *GetLogObjectPointer();
    property<CRKDevice, CRKLog *, READ_ONLY> LogObjectPointer;

    CRKComm *GetCommObjectPointer();
    property<CRKDevice, CRKComm *, READ_ONLY> CommObjectPointer;

    virtual int DownloadIDBlock() = 0;
    virtual int DownloadImage() = 0;
    virtual int PrepareIDB() = 0;
    virtual bool BufferWriteBack() = 0;


    bool CheckChip();
    bool GetFlashInfo();
    virtual int EraseIDB() = 0;
    virtual int EraseAllBlocks() = 0;
    bool SetObject(CRKImage *pImage, CRKComm *pComm, CRKLog *pLog);
    CRKDevice(STRUCT_RKDEVICE_DESC &device);
    virtual ~CRKDevice();
protected:
    STRUCT_FLASH_INFO m_flashInfo;
    PBYTE   m_pFlashInfoData;
    USHORT  m_usFlashInfoDataOffset;
    USHORT  m_usFlashInfoDataLen;
    DWORD m_idBlockOffset[IDB_BLOCKS];
    PBYTE  m_chipData;
    CRKImage *m_pImage;
    CRKComm  *m_pComm;
    CRKLog   *m_pLog;
    PBYTE  m_customData;
    USHORT m_customDataSize;
    USHORT m_customDataOffset;
    PBYTE  m_sn;
    BYTE   m_snSize;
    PBYTE  m_mac;
    PBYTE  m_blueTooth;
    PBYTE  m_uid;
    PBYTE  m_imei;
    USHORT m_sysDiskSize;
    USHORT m_cfgDiskSize;
    bool   m_bGetNewDiskSizeFlag;
    bool   m_bExistSector3Crc;
    USHORT m_usSector3Crc;
    USHORT m_usWriteBackCrc;
    USHORT m_usWriteBackCustomDataOffset;
    USHORT m_usWriteBackCustomDataSize;
    PSTRUCT_BLOCK_STATE m_pBlockState;
    BYTE   m_backupBuffer[SECTOR_SIZE + SPARE_SIZE];
    bool m_bWriteBack;
    UINT m_uiWorkFlow;
    bool m_bEraseInPrepare;
    bool m_bUidUseFlag;
    bool m_bUidWriteOK;
    bool  m_remallocDisk;
    bool m_bEmmc;
    bool GptFlag;
    bool DirectLBA;
    bool First4Access;
    ENUM_MISC_MODIFY_FLAG m_emMiscModifyFlag;
    bool m_bQuickCheckMode;
    bool BuildBlockStateMap(BYTE bFlashCS);
    int ReadMutilSector(DWORD dwPos, DWORD dwCount, PBYTE lpBuffer);
    bool EraseMutilBlock(BYTE bFlashCS, DWORD dwPos, DWORD dwCount, bool bForce);
    CHAR FindValidBlocks(char bBegin, char bLen);
    BYTE RandomByte(BYTE bLowLimit, BYTE bHighLimit);
    bool CheckCrc16(PBYTE pCheckData, USHORT usDataLength, USHORT usOldCrc);
    bool CheckUid(BYTE uidSize, BYTE *pUid);
    bool GetWriteBackData(UINT uiIDBCount, PBYTE lpBuf);
    bool GetIDBData(UINT uiIDBCount, PBYTE lpBuf, UINT uiSecCount);
    int EraseEmmcBlock(UCHAR ucFlashCS, DWORD dwPos, DWORD dwCount);
    virtual bool FindBackupBuffer() = 0;
    virtual char FindAllIDB() = 0;
    virtual CHAR FindIDBlock(char pos, char &IDBlockPos) = 0;
    virtual bool ReserveIDBlock(char iBlockIndex = 0, char iIdblockPos = 0) = 0;
    virtual bool OffsetIDBlock(char pos) = 0;
    virtual bool MakeSector0(PBYTE pSector) = 0;
    virtual void MakeSector1(PBYTE pSector) = 0;
    virtual bool MakeSector2(PBYTE pSector) = 0;
    virtual bool MakeSector3(PBYTE pSector) = 0;
    virtual int MakeIDBlockData(PBYTE lpIDBlock) = 0;
    virtual bool MakeSpareData(PBYTE lpIDBlock, DWORD dwSectorNum, PBYTE lpSpareBuffer) = 0;
    virtual int WriteIDBlock(PBYTE lpIDBlock, DWORD dwSectorNum, bool bErase) = 0;

private:
    ENUM_RKDEVICE_TYPE m_device;
    ENUM_OS_TYPE m_os;
    ENUM_RKUSB_TYPE m_usb;
    USHORT m_bcdUsb;
};

USHORT UshortToBCD(USHORT num);
BYTE   ByteToBCD(BYTE num);
extern USHORT CRC_16(BYTE *aData, UINT aSize);
extern UINT CRC_32(PBYTE pData, UINT ulSize, UINT uiPreviousValue = 0);
extern void P_RC4(BYTE *buf, USHORT len);
extern void bch_encode(BYTE *encode_in, BYTE *encode_out);
extern USHORT CRC_CCITT(UCHAR *p, UINT CalculateNumber);
extern void generate_gf();
extern void gen_poly();
#endif