/*
=======================================
KrKr XP3 文件读取
=======================================
*/
CR_API sFMT_PRT*
load_krkr_xp3 (
__CR_IO__ iDATIN* datin,
__CR_IN__ const sLOADER* param
)
{
int16u idx;
int16u len;
int32u attr;
int64u tots;
int64u tsz1;
int64u tsz2;
int64u pksz;
int64u unsz;
leng_t back;
sARRAY list;
byte_t* info;
byte_t* pntr;
wide_t* name;
sXP3_HDR head;
iPAK_XP3* port;
sFMT_PRT* rett;
sPAK_XP3_FILE temp;
/* 必须使用自己私有的读取接口 */
datin = create_file_inX(param);
if (datin == NULL)
return (NULL);
array_initT(&list, sPAK_XP3_FILE);
list.free = xp3_free;
/* 读取文件头信息 */
if (!(CR_VCALL(datin)->geType(datin, &head, sXP3_HDR)))
goto _failure1;
if (mem_cmp(head.tag, "XP3\r\n \n\x1A\x8B\x67\x01", 11) != 0)
goto _failure1;
head.idx_pos = QWORD_LE(head.idx_pos);
/* 读取文件索引表 */
if (!CR_VCALL(datin)->seek64(datin, head.idx_pos, SEEK_SET))
goto _failure1;
if (!CR_VCALL(datin)->getb_no(datin, head.tag))
goto _failure1;
if (head.tag[0])
{
/* ZLib 压缩的索引表 */
if (!CR_VCALL(datin)->getq_le(datin, &pksz))
goto _failure1;
if (!CR_VCALL(datin)->getq_le(datin, &unsz))
goto _failure1;
pntr = (byte_t*)mem_malloc64(pksz);
if (pntr == NULL)
goto _failure1;
back = CR_VCALL(datin)->read(datin, pntr, (leng_t)pksz);
if (back != (leng_t)pksz) {
mem_free(pntr);
goto _failure1;
}
info = (byte_t*)mem_malloc64(unsz);
if (info == NULL) {
mem_free(pntr);
goto _failure1;
}
back = uncompr_zlib(info, (leng_t)unsz, pntr, (leng_t)pksz);
mem_free(pntr);
if (back != (leng_t)unsz)
goto _failure2;
}
else
{
/* 直接读起索引表 */
if (!CR_VCALL(datin)->getq_le(datin, &unsz))
goto _failure1;
info = (byte_t*)mem_malloc64(unsz);
if (info == NULL)
goto _failure1;
back = CR_VCALL(datin)->read(datin, info, (leng_t)unsz);
if (back != (leng_t)unsz)
goto _failure2;
}
/* 设置加密类型 (如果有的话) */
if (param->aprm != NULL &&
*(byte_t*)param->aprm != 0x00)
attr = PAK_FILE_ENC;
else
attr = 0;
/* 加载文件信息表 */
for (pntr = info, pksz = 0; pksz < unsz;)
{
/* "File" 数据块 */
if (mem_cmp(pntr, "File", 4) != 0)
break;
mem_cpy(&tots, pntr + 4, sizeof(int64u));
tots = QWORD_LE(tots);
pntr += 12;
pksz += 12;
if (tots <= 12 + 22 + 12 + 28 || tots > unsz - pksz)
goto _failure2;
/* "info" 文件总信息 */
if (mem_cmp(pntr, "info", 4) != 0)
goto _failure2;
mem_cpy(&tsz1, pntr + 4, sizeof(int64u));
tsz1 = QWORD_LE(tsz1);
pntr += 12;
pksz += 12;
if (tsz1 <= 22 || tsz1 > tots - 12 - 12 - 28)
goto _failure2;
/* 填充一些文件信息结构成员 */
struct_zero(&temp, sPAK_XP3_FILE);
mem_cpy(&temp.protect, pntr + 0, sizeof(int32u));
mem_cpy(&temp.base.size, pntr + 4, sizeof(int64u));
mem_cpy(&temp.base.pack, pntr + 12, sizeof(int64u));
temp.base.skip = sizeof(sPAK_XP3_FILE);
temp.base.attr = attr;
temp.protect = DWORD_LE(temp.protect);
temp.base.size = QWORD_LE(temp.base.size);
temp.base.pack = QWORD_LE(temp.base.pack);
/* 读取文件名 UTF-16 无\0结尾 */
mem_cpy(&len, pntr + 20, sizeof(int16u));
len = WORD_LE(len);
if ((int64u)len > tsz1 - 22)
goto _failure2;
name = str_allocW(len + 1);
if (name == NULL)
goto _failure2;
for (idx = 0; idx < len; idx++) {
mem_cpy(&name[idx], pntr + 22 + ((leng_t)idx) * 2, 2);
name[idx] = WORD_LE(name[idx]);
}
name[idx] = 0x0000;
temp.base.name = utf16_to_utf8(name);
mem_free(name);
if (temp.base.name == NULL)
goto _failure2;
pntr += (leng_t)tsz1;
pksz += (leng_t)tsz1;
/* "segm" 文件分段信息 */
if (mem_cmp(pntr, "segm", 4) != 0)
goto _failure3;
mem_cpy(&tsz2, pntr + 4, sizeof(int64u));
tsz2 = QWORD_LE(tsz2);
pntr += 12;
pksz += 12;
if (tsz2 < 28 || tsz2 % 28 != 0 ||
tsz2 > tots - 12 - tsz1 - 12)
goto _failure3;
/* 读取所有分段信息 */
temp.segm_lst = mem_talloc64(tsz2 / 28, sPAK_XP3_SEGM);
if (temp.segm_lst == NULL)
goto _failure3;
temp.segm_cnt = (leng_t)(tsz2 / 28);
mem_cpy(temp.segm_lst, pntr, (leng_t)tsz2);
for (back = 0; back < temp.segm_cnt; back++) {
temp.segm_lst[back].zlib = DWORD_LE(temp.segm_lst[back].zlib);
temp.segm_lst[back].offset = QWORD_LE(temp.segm_lst[back].offset);
temp.segm_lst[back].unsize = QWORD_LE(temp.segm_lst[back].unsize);
temp.segm_lst[back].pksize = QWORD_LE(temp.segm_lst[back].pksize);
if (temp.segm_lst[back].unsize > temp.base.size)
goto _failure4;
if (temp.segm_lst[back].pksize > temp.base.pack)
goto _failure4;
}
pntr += (leng_t)tsz2;
pksz += (leng_t)tsz2;
/* 其他非重要数据段 */
if (tots > 12 + tsz1 + 12 + tsz2)
{
tots -= 12 + tsz1 + 12 + tsz2;
/* "adlr" 附加数据 */
if (mem_cmp(pntr, "adlr", 4) == 0) {
if (tots < 16)
goto _failure4;
mem_cpy(&temp.adlr_key, pntr + 4, sizeof(int32u));
temp.adlr_key = DWORD_LE(temp.adlr_key);
}
pntr += (leng_t)tots;
pksz += (leng_t)tots;
}
/* 用第一段设置剩下的成员 */
temp.base.offs = temp.segm_lst[0].offset;
if (temp.segm_lst[0].zlib) {
if (attr == 0)
temp.base.memo = "ZLib";
else
temp.base.memo = "ZLib + Encrypt";
temp.base.attr |= PAK_FILE_CMP;
}
else {
if (attr == 0)
temp.base.memo = "Store";
else
temp.base.memo = "Encrypt";
}
/* 文件信息压入列表 */
if (array_push_growT(&list, sPAK_XP3_FILE, &temp) == NULL)
goto _failure4;
}
mem_free(info);
/* 固定一下列表大小 */
if (!array_no_growT(&list, sPAK_XP3_FILE))
goto _failure1;
/* 生成读包接口对象 */
port = struct_new(iPAK_XP3);
if (port == NULL)
goto _failure1;
port->m_file = datin;
port->m_cnt = array_get_sizeT(&list, sPAK_XP3_FILE);
port->pack.__filelst__ = array_get_dataT(&list, sPAK_FILE);
port->pack.__vptr__ = &s_pack_vtbl;
if (!pack_init_list((iPACKAGE*)port, TRUE)) {
mem_free(port);
goto _failure1;
}
/* 设置解密回调 (传入名称字符串) */
if (attr != 0) {
decode_setup(port, (ansi_t*)param->aprm);
}
else {
port->decode_file = NULL;
port->decode_block = NULL;
}
/* 返回读取的文件数据 */
rett = struct_new(sFMT_PRT);
if (rett == NULL) {
iPAK_XP3_release((iPACKAGE*)port);
return (NULL);
}
rett->type = CR_FMTZ_PRT;
rett->port = (iPORT*)port;
rett->more = "iPACKAGE";
rett->infor = "KiriKiri2 XP3 Archive (XP3)";
return (rett);
_failure4:
mem_free(temp.segm_lst);
_failure3:
mem_free(temp.base.name);
_failure2:
mem_free(info);
_failure1:
array_freeT(&list, sPAK_XP3_FILE);
CR_VCALL(datin)->release(datin);
return (NULL);
}
/*
---------------------------------------
文件处理回调
---------------------------------------
*/
static bool_t do_each_file (void_t *param, sSEARCHa *finfo)
{
wide_t *tmp;
sCP_PARAM *ctx;
leng_t idx, len;
ansi_t *src, *dst;
/* 统计文件总数 */
ctx = (sCP_PARAM*)param;
ctx->tot += 1;
/* 跳过空文件和只读文件 */
if ((finfo->size == 0) ||
(finfo->attr & CR_FILE_ATTR_RDO))
return (TRUE);
/* 过滤掉大文件 */
if (finfo->size > CR_M2B(4))
return (TRUE);
if (ctx->del_u8_bom && finfo->size <= 3)
return (TRUE);
/* 加载整个文件 */
src = file_load_as_strA(finfo->name);
if (src == NULL)
return (TRUE);
/* 跳过二进制文件 */
len = str_lenA(src);
if (len != finfo->size)
goto _func_out;
/* 删除 UTF-8 文件的 BOM 标志 */
if (ctx->del_u8_bom) {
if (chr_cmpA(src, BOM_UTF8, 3) != 0)
goto _func_out;
if (!is_utf8_file(src))
goto _func_out;
if (!file_saveA(finfo->name, src + 3, len - 3))
goto _func_out;
ctx->cvt += 1;
printf("%s\n", finfo->name);
goto _func_out;
}
/* 检查是否是 UTF-8, 跳过不用转换的文件 */
if (ctx->skip_utf8 && is_utf8_file(src))
goto _func_out;
/* 转换到 UTF-16 编码 */
len = MultiByteToWideChar(ctx->from, MB_ERR_INVALID_CHARS,
src, -1, NULL, 0);
if (len == 0)
goto _func_out;
if (ctx->to == CR_UTF16LE || ctx->to == CR_UTF16BE) {
tmp = str_allocW(len + 1);
if (tmp == NULL)
goto _func_out;
len = MultiByteToWideChar(ctx->from, 0, src, -1, tmp + 1, len);
mem_cpy(tmp, BOM_UTF16LE, 2);
if (ctx->to == CR_UTF16BE) {
for (idx = 0; idx < len; idx++)
tmp[idx] = xchg_int16u(tmp[idx]);
}
len = len * 2 + 1;
dst = (ansi_t*)tmp;
mem_free(src);
}
else {
tmp = str_allocW(len);
if (tmp == NULL)
goto _func_out;
MultiByteToWideChar(ctx->from, 0, src, -1, tmp, len);
mem_free(src);
/* 转换到目标编码 */
len = WideCharToMultiByte(ctx->to, 0, tmp, -1, NULL, 0, NULL, NULL);
if (len == 0) {
mem_free(tmp);
return (TRUE);
}
dst = str_allocA(len);
if (dst == NULL) {
mem_free(tmp);
return (TRUE);
}
len = WideCharToMultiByte(ctx->to, 0, tmp, -1, dst, len, NULL, NULL);
mem_free(tmp);
}
/* 写入整个文件 (去掉后面的0) */
if (!file_saveA(finfo->name, dst, len - 1)) {
mem_free(dst);
return (TRUE);
}
ctx->cvt += 1;
printf("%s\n", finfo->name);
mem_free(dst);
return (TRUE);
_func_out:
mem_free(src);
return (TRUE);
}
