mirror of
https://github.com/CTCaer/hekate.git
synced 2024-11-23 02:16:41 +00:00
157 lines
3.1 KiB
C
157 lines
3.1 KiB
C
#include <string.h>
|
|
#include <stdlib.h>
|
|
|
|
#include <bdk.h>
|
|
|
|
#include "pkg2_ini_kippatch.h"
|
|
#include <libs/fatfs/ff.h>
|
|
|
|
#define KPS(x) ((u32)(x) << 29)
|
|
|
|
static u8 *_htoa(u8 *result, const char *ptr, u8 byte_len)
|
|
{
|
|
char ch = *ptr;
|
|
u32 ascii_len = byte_len * 2;
|
|
if (!result)
|
|
result = malloc(byte_len);
|
|
u8 *dst = result;
|
|
|
|
while (ch == ' ' || ch == '\t')
|
|
ch = *(++ptr);
|
|
|
|
bool shift = true;
|
|
while (ascii_len)
|
|
{
|
|
u8 tmp = 0;
|
|
if (ch >= '0' && ch <= '9')
|
|
tmp = (ch - '0');
|
|
else if (ch >= 'A' && ch <= 'F')
|
|
tmp = (ch - 'A' + 10);
|
|
else if (ch >= 'a' && ch <= 'f')
|
|
tmp = (ch - 'a' + 10);
|
|
|
|
if (shift)
|
|
*dst = (tmp << 4) & 0xF0;
|
|
else
|
|
{
|
|
*dst |= (tmp & 0x0F);
|
|
dst++;
|
|
}
|
|
|
|
ascii_len--;
|
|
ch = *(++ptr);
|
|
shift = !shift;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
static char *_strdup(char *str)
|
|
{
|
|
if (!str)
|
|
return NULL;
|
|
if (str[0] == ' ' && (strlen(str) > 1))
|
|
str++;
|
|
char *res = (char *)malloc(strlen(str) + 1);
|
|
strcpy(res, str);
|
|
if (res[strlen(res) - 1] == ' ' && (strlen(res) > 1))
|
|
res[strlen(res) - 1] = 0;
|
|
|
|
return res;
|
|
}
|
|
|
|
static u32 _find_patch_section_name(char *lbuf, u32 lblen, char schar)
|
|
{
|
|
u32 i;
|
|
for (i = 0; i < lblen && lbuf[i] != schar && lbuf[i] != '\n' && lbuf[i] != '\r'; i++)
|
|
;
|
|
lbuf[i] = 0;
|
|
|
|
return i;
|
|
}
|
|
|
|
static ini_kip_sec_t *_ini_create_kip_section(link_t *dst, ini_kip_sec_t *ksec, char *name)
|
|
{
|
|
if (ksec)
|
|
list_append(dst, &ksec->link);
|
|
|
|
ksec = (ini_kip_sec_t *)calloc(sizeof(ini_kip_sec_t), 1);
|
|
u32 i = _find_patch_section_name(name, strlen(name), ':') + 1;
|
|
ksec->name = _strdup(name);
|
|
|
|
// Get hash section.
|
|
_htoa(ksec->hash, &name[i], 8);
|
|
|
|
return ksec;
|
|
}
|
|
|
|
int ini_patch_parse(link_t *dst, char *ini_path)
|
|
{
|
|
FIL fp;
|
|
u32 lblen;
|
|
char lbuf[512];
|
|
ini_kip_sec_t *ksec = NULL;
|
|
|
|
// Open ini.
|
|
if (f_open(&fp, ini_path, FA_READ) != FR_OK)
|
|
return 0;
|
|
|
|
do
|
|
{
|
|
// Fetch one line.
|
|
lbuf[0] = 0;
|
|
f_gets(lbuf, 512, &fp);
|
|
lblen = strlen(lbuf);
|
|
|
|
// Remove trailing newline. Depends on 'FF_USE_STRFUNC 2' that removes \r.
|
|
if (lblen && lbuf[lblen - 1] == '\n')
|
|
lbuf[lblen - 1] = 0;
|
|
|
|
if (lblen > 2 && lbuf[0] == '[') // Create new section.
|
|
{
|
|
_find_patch_section_name(lbuf, lblen, ']');
|
|
|
|
ksec = _ini_create_kip_section(dst, ksec, &lbuf[1]);
|
|
list_init(&ksec->pts);
|
|
}
|
|
else if (ksec && lbuf[0] == '.') //Extract key/value.
|
|
{
|
|
u32 tmp = 0;
|
|
u32 i = _find_patch_section_name(lbuf, lblen, '=');
|
|
|
|
ini_patchset_t *pt = (ini_patchset_t *)calloc(sizeof(ini_patchset_t), 1);
|
|
|
|
pt->name = _strdup(&lbuf[1]);
|
|
|
|
u8 kip_sidx = lbuf[i + 1] - '0';
|
|
|
|
if (kip_sidx < 6)
|
|
{
|
|
pt->offset = KPS(kip_sidx);
|
|
tmp = _find_patch_section_name(&lbuf[i + 3], lblen, ':');
|
|
pt->offset |= strtol(&lbuf[i + 3], NULL, 16);
|
|
|
|
i += tmp + 4;
|
|
|
|
tmp = _find_patch_section_name(&lbuf[i], lblen, ':');
|
|
pt->length = strtol(&lbuf[i], NULL, 16);
|
|
|
|
i += tmp + 1;
|
|
|
|
tmp = _find_patch_section_name(&lbuf[i], lblen, ',');
|
|
pt->srcData = _htoa(NULL, &lbuf[i], pt->length);
|
|
i += tmp + 1;
|
|
pt->dstData = _htoa(NULL, &lbuf[i], pt->length);
|
|
}
|
|
|
|
list_append(&ksec->pts, &pt->link);
|
|
}
|
|
} while (!f_eof(&fp));
|
|
|
|
f_close(&fp);
|
|
|
|
if (ksec)
|
|
list_append(dst, &ksec->link);
|
|
|
|
return 1;
|
|
}
|