Fix errors in elf2kip

This commit is contained in:
Michael Scire 2018-05-01 00:30:58 -06:00
parent 92268de97b
commit 648a5037e8

View File

@ -460,7 +460,7 @@ int main(int argc, char* argv[]) {
KipHeader kip_hdr = {0}; KipHeader kip_hdr = {0};
memcpy(kip_hdr.Magic, "KIP1", 4); memcpy(kip_hdr.Magic, "KIP1", 4);
kip_hdr.Flags = 0x1F; kip_hdr.Flags = 0x3F;
if (sizeof(KipHeader) != 0x100) { if (sizeof(KipHeader) != 0x100) {
fprintf(stderr, "Bad compile environment!\n"); fprintf(stderr, "Bad compile environment!\n");
@ -538,13 +538,12 @@ int main(int argc, char* argv[]) {
} else { } else {
kip_hdr.Segments[i].DecompSz = 0; kip_hdr.Segments[i].DecompSz = 0;
} }
kip_hdr.Segments[i].DstOff = phdr->p_vaddr;
kip_hdr.Segments[i].CompSz = 0; kip_hdr.Segments[i].CompSz = 0;
break; break;
} }
FileOffsets[i] = phdr->p_vaddr; FileOffsets[i] = file_off;
kip_hdr.Segments[i].DecompSz = (phdr->p_filesz + 0xFFF) & ~0xFFF; kip_hdr.Segments[i].DecompSz = phdr->p_filesz;
buf[i] = malloc(kip_hdr.Segments[i].DecompSz); buf[i] = malloc(kip_hdr.Segments[i].DecompSz);
if (buf[i] == NULL) { if (buf[i] == NULL) {
@ -573,7 +572,7 @@ int main(int argc, char* argv[]) {
for (i=0; i<3; i++) for (i=0; i<3; i++)
{ {
fseek(out, sizeof(kip_hdr) + kip_hdr.Segments[i].DstOff, SEEK_SET); fseek(out, sizeof(kip_hdr) + FileOffsets[i], SEEK_SET);
fwrite(cmp[i], kip_hdr.Segments[i].CompSz, 1, out); fwrite(cmp[i], kip_hdr.Segments[i].CompSz, 1, out);
} }