Function open_archive_chunk

ogsr_engine/xrCore/arc_db.cpp:7–68

#include "trivial_encryptor.h"

static IReader* open_archive_chunk(void* ptr, u32 ID, const char* archiveName, size_t archiveSize)
{
    const bool shouldDecrypt = !strstr(archiveName, ".xdb");

    u32 dwType = INVALID_SET_FILE_POINTER;
    DWORD read_byte;
    u32 dwPtr = SetFilePointer(ptr, 0, nullptr, FILE_BEGIN);
    R_ASSERT(dwPtr != INVALID_SET_FILE_POINTER, archiveName, Debug.error2string(GetLastError()));
    while (true)
    {
        bool res = ReadFile(ptr, &dwType, 4, &read_byte, nullptr);
        R_ASSERT(res && read_byte == 4, archiveName, Debug.error2string(GetLastError()));

        u32 tempSize = 0;
        res = ReadFile(ptr, &tempSize, 4, &read_byte, nullptr);
        size_t dwSize = tempSize;
        R_ASSERT(res && read_byte == 4, archiveName, Debug.error2string(GetLastError()));

        if ((dwType & ~CFS_CompressMark) == ID)
        {
            u8* src_data = xr_alloc<u8>(dwSize);
            res = ReadFile(ptr, src_data, dwSize, &read_byte, nullptr);
            R_ASSERT(res && read_byte == dwSize, archiveName, Debug.error2string(GetLastError()));
            if (dwType & CFS_CompressMark)
            {
                BYTE* dest{};
                size_t dest_sz{};

                if (shouldDecrypt)
                    g_trivial_encryptor.decode(src_data, dwSize, src_data); // Try WW key first

                bool result = _decompressLZ(&dest, &dest_sz, src_data, dwSize, archiveSize);

                if (!result && shouldDecrypt) // Let's try to decode with Rus key
                {
                    MsgDbg("~~[%s]: decoding of [%s] with WW key failed, trying RU key...", __FUNCTION__, archiveName);
                    g_trivial_encryptor.encode(src_data, dwSize, src_data); // rollback
                    g_trivial_encryptor.decode(src_data, dwSize, src_data, trivial_encryptor::key_flag::russian);
                    result = _decompressLZ(&dest, &dest_sz, src_data, dwSize, archiveSize);

                    if (!result)
                    {
                        g_trivial_encryptor.encode(src_data, dwSize, src_data, trivial_encryptor::key_flag::russian); // rollback

                        result = _decompressLZ(&dest, &dest_sz, src_data, dwSize, archiveSize);
                        if (result)
                            MsgDbg("--[%s]: found CS/COP archive [%s]", __FUNCTION__, archiveName);
                    }
                }
                
                CHECK_OR_EXIT(result, make_string("[%s] Can't decompress archive [%s]", __FUNCTION__, archiveName));

                xr_free(src_data);
                return xr_new<CTempReader>(dest, dest_sz, 0);
            }
            return xr_new<CTempReader>(src_data, dwSize, 0);
        }