Add unpacktool

unpacktool/CompactProLZHDecompressor.cpp

@@ -0,0 +1,129 @@
+#include "CompactProLZHDecompressor.h"
+
+#include "CSInputBuffer.h"
+#include "PrefixCode.h"
+
+CompactProLZHDecompressor::CompactProLZHDecompressor(int blocksize)
+	: LZSSDecompressor(8192)
+	, literalcode(nullptr)
+	, lengthcode(nullptr)
+	, offsetcode(nullptr)
+	, blocksize(blocksize)
+	, blockcount(0)
+	, blockstart(0)
+{
+}
+
+CompactProLZHDecompressor::~CompactProLZHDecompressor()
+{
+	delete literalcode;
+	delete lengthcode;
+	delete offsetcode;
+}
+
+bool CompactProLZHDecompressor::resetLZSSHandle()
+{
+	blockcount = blocksize;
+	blockstart = 0;
+
+	return true;
+}
+
+bool CompactProLZHDecompressor::nextLiteralOrOffset(int *offset, int *length, int &result)
+{
+	if (blockcount >= blocksize)
+	{
+		if (blockstart)
+		{
+			// Don't let your bad implementations leak into your file formats, people!
+			if (!CSInputSkipToByteBoundary(input))
+				return false;
+
+			if ((CSInputBufferOffset(input) - blockstart) & 1)
+			{
+				if (!CSInputSkipBytes(input, 3))
+					return false;
+			}
+			else
+			{
+				if (!CSInputSkipBytes(input, 2))
+					return false;
+			}
+		}
+
+		delete literalcode;
+		delete lengthcode;
+		delete offsetcode;
+		literalcode = lengthcode = offsetcode = nullptr;
+		literalcode = allocAndParseCodeOfSize(256);
+		if (!literalcode)
+			return false;
+		lengthcode = allocAndParseCodeOfSize(64);
+		if (!lengthcode)
+			return false;
+		offsetcode = allocAndParseCodeOfSize(128);
+		if (!offsetcode)
+			return false;
+		blockcount = 0;
+		blockstart = CSInputBufferOffset(input);
+	}
+
+	unsigned int nextBit;
+	if (!CSInputNextBit(input, nextBit))
+		return false;
+
+	if (nextBit)
+	{
+		blockcount += 2;
+		return CSInputNextSymbolUsingCode(input, literalcode, result);
+	}
+	else
+	{
+		blockcount += 3;
+
+		if (!CSInputNextSymbolUsingCode(input, lengthcode, *length))
+			return false;
+
+		int offsetSym;
+		if (!CSInputNextSymbolUsingCode(input, offsetcode, offsetSym))
+			return false;
+
+		unsigned int offsetBits;
+		if (!CSInputNextBitString(input, 6, offsetBits))
+			return false;
+
+		*offset = offsetSym << 6;
+		*offset |= offsetBits;
+
+		result = XADLZSSMatch;
+		return true;
+	}
+
+	result = XADLZSSEnd;
+	return true;
+}
+
+XADPrefixCode *CompactProLZHDecompressor::allocAndParseCodeOfSize(int size)
+{
+	int numbytes;
+	if (!CSInputNextByte(input, numbytes))
+		return nullptr;
+
+	if (numbytes * 2 > size)
+		return nullptr;
+
+	std::vector<int> codelengths;
+	codelengths.resize(size);
+
+	for (int i = 0; i < numbytes; i++)
+	{
+		int val;
+		if (!CSInputNextByte(input, val))
+			return nullptr;
+		codelengths[2 * i] = val >> 4;
+		codelengths[2 * i + 1] = val & 0x0f;
+	}
+	for (int i = numbytes * 2; i < size; i++) codelengths[i] = 0;
+
+	return XADPrefixCode::prefixCodeWithLengths(codelengths.data(), size, 15, true);
+}