#include "RKBoot.h"
extern UINT CRC_32(PBYTE pData, UINT ulSize, UINT uiPreviousValue = 0);

bool CRKBoot::GetRc4DisableFlag()
{
    return m_bRc4Disable;
}
bool CRKBoot::GetSignFlag()
{
    return m_bSignFlag;
}
DWORD CRKBoot::GetVersion()
{
    return m_version;
}
DWORD CRKBoot::GetMergeVersion()
{
    return m_mergeVersion;
}
STRUCT_RKTIME CRKBoot::GetReleaseTime()
{
    return m_releaseTime;
}
ENUM_RKDEVICE_TYPE CRKBoot::GetSupportDevice()
{
    return m_supportDevice;
}
UCHAR CRKBoot::GetEntry471Count()
{
    return m_471Count;
}
UCHAR CRKBoot::GetEntry472Count()
{
    return m_472Count;
}
UCHAR CRKBoot::GetEntryLoaderCount()
{
    return m_loaderCount;
}
bool CRKBoot::CrcCheck()
{
    UINT *pOldCrc, ulNewCrc;
    pOldCrc = (UINT *)(m_BootData + (m_BootSize - 4));
    ulNewCrc = CRC_32(m_BootData, m_BootSize - 4);
    return (*pOldCrc == ulNewCrc) ? true : false;
}
bool CRKBoot::SaveEntryFile(ENUM_RKBOOTENTRY type, UCHAR ucIndex, tstring fileName)
{
    DWORD dwOffset;
    UCHAR ucCount, ucSize;
    switch (type)
    {
        case ENTRY471:
            dwOffset = m_471Offset;
            ucCount = m_471Count;
            ucSize = m_471Size;
            break;
        case ENTRY472:
            dwOffset = m_472Offset;
            ucCount = m_472Count;
            ucSize = m_472Size;
            break;
        case ENTRYLOADER:
            dwOffset = m_loaderOffset;
            ucCount = m_loaderCount;
            ucSize = m_loaderSize;
            break;
        default:
            return false;
    }
    if (ucIndex >= ucCount)
    {
        return false;
    }
    PSTRUCT_RKBOOT_ENTRY pEntry;
    pEntry = (PSTRUCT_RKBOOT_ENTRY)(m_BootData + dwOffset + (ucSize * ucIndex));
    FILE *file = NULL;
    file = fopen(fileName.c_str(), _T("wb+"));
    if (!file)
    {
        return false;
    }
    fwrite(m_BootData + pEntry->dwDataOffset, 1, pEntry->dwDataSize, file);
    fclose(file);
    return true;
}
bool CRKBoot::GetEntryProperty(ENUM_RKBOOTENTRY type, UCHAR ucIndex, DWORD &dwSize, DWORD &dwDelay, tchar *pName)
{
    DWORD dwOffset;
    UCHAR ucCount, ucSize;
    switch (type)
    {
        case ENTRY471:
            dwOffset = m_471Offset;
            ucCount = m_471Count;
            ucSize = m_471Size;
            break;
        case ENTRY472:
            dwOffset = m_472Offset;
            ucCount = m_472Count;
            ucSize = m_472Size;
            break;
        case ENTRYLOADER:
            dwOffset = m_loaderOffset;
            ucCount = m_loaderCount;
            ucSize = m_loaderSize;//Loader³¤¶ÈÉú³ÉʱÒѾ­512¶ÔÆë
            break;
        default:
            return false;
    }
    if (ucIndex >= ucCount)
    {
        return false;
    }
    PSTRUCT_RKBOOT_ENTRY pEntry;
    pEntry = (PSTRUCT_RKBOOT_ENTRY)(m_BootData + dwOffset + (ucSize * ucIndex));
    dwDelay = pEntry->dwDataDelay;
    dwSize = pEntry->dwDataSize;
    if (pName)
    {
        WCHAR_To_char(pEntry->szName, pName, 20);
    }
    return true;
}
bool CRKBoot::GetEntryData(ENUM_RKBOOTENTRY type, UCHAR ucIndex, PBYTE lpData)
{
    DWORD dwOffset;
    UCHAR ucCount, ucSize;
    switch (type)
    {
        case ENTRY471:
            dwOffset = m_471Offset;
            ucCount = m_471Count;
            ucSize = m_471Size;
            break;
        case ENTRY472:
            dwOffset = m_472Offset;
            ucCount = m_472Count;
            ucSize = m_472Size;
            break;
        case ENTRYLOADER:
            dwOffset = m_loaderOffset;
            ucCount = m_loaderCount;
            ucSize = m_loaderSize;
            break;
        default:
            return false;
    }
    if (ucIndex >= ucCount)
    {
        return false;
    }
    PSTRUCT_RKBOOT_ENTRY pEntry;
    pEntry = (PSTRUCT_RKBOOT_ENTRY)(m_BootData + dwOffset + (ucSize * ucIndex));
    memcpy(lpData, m_BootData + pEntry->dwDataOffset, pEntry->dwDataSize);
    return true;
}
CHAR CRKBoot::GetIndexByName(ENUM_RKBOOTENTRY type, tchar *pName)
{
    DWORD dwOffset;
    UCHAR ucCount, ucSize;
    switch (type)
    {
        case ENTRY471:
            dwOffset = m_471Offset;
            ucCount = m_471Count;
            ucSize = m_471Size;
            break;
        case ENTRY472:
            dwOffset = m_472Offset;
            ucCount = m_472Count;
            ucSize = m_472Size;
            break;
        case ENTRYLOADER:
            dwOffset = m_loaderOffset;
            ucCount = m_loaderCount;
            ucSize = m_loaderSize;
            break;
        default:
            return -1;
    }

    for (UCHAR i = 0; i < ucCount; i++)
    {
        PSTRUCT_RKBOOT_ENTRY pEntry;
        pEntry = (PSTRUCT_RKBOOT_ENTRY)(m_BootData + dwOffset + (ucSize * i));

        char szName[20];
        WCHAR_To_char(pEntry->szName, szName, 20);
        if (_tcsicmp(pName, szName) == 0)
        {
            return i;
        }
    }
    return -1;
}

bool CRKBoot::IsNewIDBFlag()
{
    return m_NewIDBFlag;
}

CRKBoot::~CRKBoot()
{
    if (m_BootData != NULL)
    {
        delete []m_BootData;
    }
}

CRKBoot::CRKBoot(PBYTE lpBootData, DWORD dwBootSize, bool &bCheck)
{
    Rc4DisableFlag.setContainer(this);
    Rc4DisableFlag.getter(&CRKBoot::GetRc4DisableFlag);
    SignFlag.setContainer(this);
    SignFlag.getter(&CRKBoot::GetSignFlag);
    Version.setContainer(this);
    Version.getter(&CRKBoot::GetVersion);
    MergeVersion.setContainer(this);
    MergeVersion.getter(&CRKBoot::GetMergeVersion);
    ReleaseTime.setContainer(this);
    ReleaseTime.getter(&CRKBoot::GetReleaseTime);
    SupportDevice.setContainer(this);
    SupportDevice.getter(&CRKBoot::GetSupportDevice);
    Entry471Count.setContainer(this);
    Entry471Count.getter(&CRKBoot::GetEntry471Count);
    Entry472Count.setContainer(this);
    Entry472Count.getter(&CRKBoot::GetEntry472Count);
    EntryLoaderCount.setContainer(this);
    EntryLoaderCount.getter(&CRKBoot::GetEntryLoaderCount);
    bCheck = true;
    if (lpBootData != NULL)
    {
        m_BootData = lpBootData;
        m_BootSize = dwBootSize;
        bCheck = CrcCheck();
        if (!bCheck)
        {
            printf("CRKBoot : CrcCheck happen error\n");
            return;
        }
        PSTRUCT_RKBOOT_HEAD pBootHead;
        pBootHead = (PSTRUCT_RKBOOT_HEAD)(m_BootData);
        if (pBootHead->uiTag != 0x544F4F42  && (pBootHead->uiTag != 0x2052444C)) //add suppoty rk356x new loader
        {
            bCheck = false;
            printf("CRKBoot : pBootHead->uiTag!=0x544F4F42 or 0x2052444C\n");
            return;
        }

        if (0x2052444C == pBootHead->uiTag)
        {
            m_NewIDBFlag = true;
            printf("RKBoot:this is new IDB flag\n");
        }
        else
        {
            m_NewIDBFlag = false;
        }

        if (pBootHead->ucRc4Flag)
        {
            m_bRc4Disable = true;
        }
        else
        {
            m_bRc4Disable = false;
        }
        if (pBootHead->ucSignFlag == 'S')
        {
            m_bSignFlag = true;
        }
        else
        {
            m_bSignFlag = false;
        }
        m_version = pBootHead->dwVersion;
        m_mergeVersion = pBootHead->dwMergeVersion;
        m_BootHeadSize = pBootHead->usSize;
        m_releaseTime.usYear = pBootHead->stReleaseTime.usYear;
        m_releaseTime.ucMonth = pBootHead->stReleaseTime.ucMonth;
        m_releaseTime.ucDay = pBootHead->stReleaseTime.ucDay;
        m_releaseTime.ucHour = pBootHead->stReleaseTime.ucHour;
        m_releaseTime.ucMinute = pBootHead->stReleaseTime.ucMinute;
        m_releaseTime.ucSecond = pBootHead->stReleaseTime.ucSecond;
        m_supportDevice = pBootHead->emSupportChip;

        m_471Offset = pBootHead->dw471EntryOffset;
        m_471Count = pBootHead->uc471EntryCount;
        m_471Size = pBootHead->uc471EntrySize;

        m_472Offset = pBootHead->dw472EntryOffset;
        m_472Count = pBootHead->uc472EntryCount;
        m_472Size = pBootHead->uc472EntrySize;

        m_loaderOffset = pBootHead->dwLoaderEntryOffset;
        m_loaderCount = pBootHead->ucLoaderEntryCount;
        m_loaderSize = pBootHead->ucLoaderEntrySize;

        memcpy(m_crc, m_BootData + (m_BootSize - 4), 4);
    }
    else
    {
        bCheck = false;
        printf("CRKBoot : lpBootData = null\n");
        m_BootData = NULL;
    }
}

void CRKBoot::WCHAR_To_char(WCHAR *src, char *dst, int len)
{
    int i;
    memset(dst, 0, len * sizeof(char));
    for (i = 0; i < len; i++)
    {
        memcpy(dst, src, 1);
        src++;
        dst++;
    }
}