135 lines
4.0 KiB
C++
135 lines
4.0 KiB
C++
#include "lz11.h"
|
|
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
|
|
#include <sstream>
|
|
|
|
#define MIN(a,b) (((a)<(b))?(a):(b))
|
|
|
|
// Ported from: https://github.com/svn2github/3DS-Explorer/blob/master/3DSExplorer/DSDecmp/Formats/Nitro/LZ11.cs
|
|
|
|
u32 lz11_get_occurence_length(u8* newPtr, u32 newLength, u8* oldPtr, u32 oldLength, u32* disp) {
|
|
if(disp != NULL) {
|
|
*disp = 0;
|
|
}
|
|
|
|
if(newLength == 0) {
|
|
return 0;
|
|
}
|
|
|
|
u32 maxLength = 0;
|
|
if(oldLength > 0) {
|
|
for(u32 i = 0; i < oldLength - 1; i++) {
|
|
u8* currentOldStart = oldPtr + i;
|
|
u32 currentLength = 0;
|
|
for(u32 j = 0; j < newLength; j++) {
|
|
if(*(currentOldStart + j) != *(newPtr + j)) {
|
|
break;
|
|
}
|
|
|
|
currentLength++;
|
|
}
|
|
|
|
if(currentLength > maxLength) {
|
|
maxLength = currentLength;
|
|
if(disp != NULL) {
|
|
*disp = oldLength - i;
|
|
}
|
|
|
|
if(maxLength == newLength) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return maxLength;
|
|
}
|
|
|
|
void* lz11_compress(u32* size, void* input, u32 inputSize) {
|
|
if(inputSize > 0xFFFFFF) {
|
|
printf("ERROR: LZ11 input is too large.\n");
|
|
return NULL;
|
|
}
|
|
|
|
std::stringstream ss;
|
|
|
|
u8 header[4] = { 0x11, (u8) (inputSize & 0xFF), (u8) ((inputSize >> 8) & 0xFF), (u8) ((inputSize >> 16) & 0xFF) };
|
|
ss.write((char*) header, 4);
|
|
|
|
u32 compressedLength = 4;
|
|
u8 outbuffer[8 * 4 + 1];
|
|
outbuffer[0] = 0;
|
|
u32 bufferlength = 1;
|
|
u32 bufferedBlocks = 0;
|
|
u32 readBytes = 0;
|
|
while(readBytes < inputSize) {
|
|
if(bufferedBlocks == 8) {
|
|
ss.write((char*) outbuffer, bufferlength);
|
|
compressedLength += bufferlength;
|
|
outbuffer[0] = 0;
|
|
bufferlength = 1;
|
|
bufferedBlocks = 0;
|
|
}
|
|
|
|
u32 disp = 0;
|
|
u32 oldLength = MIN(readBytes, 0x1000);
|
|
u32 length = lz11_get_occurence_length((u8*) input + readBytes, MIN(inputSize - readBytes, 0x10110), (u8*) input + readBytes - oldLength, oldLength, &disp);
|
|
if(length < 3) {
|
|
outbuffer[bufferlength++] = *((u8*) input + (readBytes++));
|
|
} else {
|
|
readBytes += length;
|
|
outbuffer[0] |= (u8)(1 << (7 - bufferedBlocks));
|
|
if(length > 0x110) {
|
|
outbuffer[bufferlength] = 0x10;
|
|
outbuffer[bufferlength] |= (u8)(((length - 0x111) >> 12) & 0x0F);
|
|
bufferlength++;
|
|
outbuffer[bufferlength] = (u8)(((length - 0x111) >> 4) & 0xFF);
|
|
bufferlength++;
|
|
outbuffer[bufferlength] = (u8)(((length - 0x111) << 4) & 0xF0);
|
|
} else if(length > 0x10) {
|
|
outbuffer[bufferlength] = 0x00;
|
|
outbuffer[bufferlength] |= (u8)(((length - 0x111) >> 4) & 0x0F);
|
|
bufferlength++;
|
|
outbuffer[bufferlength] = (u8)(((length - 0x111) << 4) & 0xF0);
|
|
} else {
|
|
outbuffer[bufferlength] = (u8)(((length - 1) << 4) & 0xF0);
|
|
}
|
|
|
|
outbuffer[bufferlength] |= (u8)(((disp - 1) >> 8) & 0x0F);
|
|
bufferlength++;
|
|
outbuffer[bufferlength] = (u8)((disp - 1) & 0xFF);
|
|
bufferlength++;
|
|
}
|
|
|
|
bufferedBlocks++;
|
|
}
|
|
|
|
if(bufferedBlocks > 0) {
|
|
ss.write((char*) outbuffer, bufferlength);
|
|
compressedLength += bufferlength;
|
|
}
|
|
|
|
if(compressedLength % 4 != 0) {
|
|
u32 padLength = 4 - (compressedLength % 4);
|
|
// Small patch to prevent msvc error
|
|
u8* pad = new u8[padLength];
|
|
memset(pad, 0, (size_t) padLength);
|
|
|
|
ss.write((char*) pad, padLength);
|
|
compressedLength += padLength;
|
|
// Need to delete as not dynamic
|
|
delete[] pad;
|
|
}
|
|
|
|
void* buf = malloc((size_t) compressedLength);
|
|
ss.read((char*) buf, compressedLength);
|
|
|
|
if(size != NULL) {
|
|
*size = (u32) compressedLength;
|
|
}
|
|
|
|
return buf;
|
|
}
|