/*

LzmaStateTest.c

Test application for LZMA Decoder (State version)

This file written and distributed to public domain by Igor Pavlov.

This file is part of LZMA SDK 4.26 (2005-08-02)

*/

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include "LzmaStateDecode.h"

const char *kCantReadMessage = "Can not read input file";

const char *kCantWriteMessage = "Can not write output file";

const char *kCantAllocateMessage = "Can not allocate memory";

#define kInBufferSize (1 << 15)

#define kOutBufferSize (1 << 15)

unsigned char g_InBuffer[kInBufferSize];

unsigned char g_OutBuffer[kOutBufferSize];

size_t MyReadFile(FILE *file, void *data, size_t size)

{ return fread(data, 1, size, file); }

int MyReadFileAndCheck(FILE *file, void *data, size_t size)

{ return (MyReadFile(file, data, size) == size); }

int PrintError(char *buffer, const char *message)

{

sprintf(buffer + strlen(buffer), "\nError: ");

sprintf(buffer + strlen(buffer), message);

return 1;

}

int main3(FILE *inFile, FILE *outFile, char *rs)

{

/* We use two 32-bit integers to construct 64-bit integer for file size.

You can remove outSizeHigh, if you don't need >= 4GB supporting,

or you can use UInt64 outSize, if your compiler supports 64-bit integers*/

UInt32 outSize = 0;

UInt32 outSizeHigh = 0;

int waitEOS = 1;

/* waitEOS = 1, if there is no uncompressed size in headers,

so decoder will wait EOS (End of Stream Marker) in compressed stream */

int i;

int res = 0;

CLzmaDecoderState state; /* it's about 140 bytes structure, if int is 32-bit */

unsigned char properties[LZMA_PROPERTIES_SIZE];

SizeT inAvail = 0;

unsigned char *inBuffer = 0;

if (sizeof(UInt32) < 4)

return PrintError(rs, "LZMA decoder needs correct UInt32");

/* Read LZMA properties for compressed stream */

if (!MyReadFileAndCheck(inFile, properties, sizeof(properties)))

return PrintError(rs, kCantReadMessage);

/* Read uncompressed size */

for (i = 0; i < 8; i++)

{

unsigned char b;

if (!MyReadFileAndCheck(inFile, &b, 1))

return PrintError(rs, kCantReadMessage);

if (b != 0xFF)

waitEOS = 0;

if (i < 4)

outSize += (UInt32)(b) << (i * 8);

else

outSizeHigh += (UInt32)(b) << ((i - 4) * 8);

}

/* Decode LZMA properties and allocate memory */

if (LzmaDecodeProperties(&state.Properties, properties, LZMA_PROPERTIES_SIZE) != LZMA_RESULT_OK)

return PrintError(rs, "Incorrect stream properties");

state.Probs = (CProb *)malloc(LzmaGetNumProbs(&state.Properties) * sizeof(CProb));

if (state.Probs == 0)

return PrintError(rs, kCantAllocateMessage);

if (state.Properties.DictionarySize == 0)

state.Dictionary = 0;

else

{

state.Dictionary = (unsigned char *)malloc(state.Properties.DictionarySize);

if (state.Dictionary == 0)

{

free(state.Probs);

return PrintError(rs, kCantAllocateMessage);

}

}

/* Decompress */

LzmaDecoderInit(&state);

do

{

SizeT inProcessed, outProcessed;

int finishDecoding;

UInt32 outAvail = kOutBufferSize;

if (!waitEOS && outSizeHigh == 0 && outAvail > outSize)

outAvail = outSize;

if (inAvail == 0)

{

inAvail = (SizeT)MyReadFile(inFile, g_InBuffer, kInBufferSize);

inBuffer = g_InBuffer;

}

finishDecoding = (inAvail == 0);

res = LzmaDecode(&state,

inBuffer, inAvail, &inProcessed,

g_OutBuffer, outAvail, &outProcessed,

finishDecoding);

if (res != 0)

{

sprintf(rs + strlen(rs), "\nDecoding error = %d\n", res);

res = 1;

break;

}

inAvail -= inProcessed;

inBuffer += inProcessed;

if (outFile != 0)

if (fwrite(g_OutBuffer, 1, outProcessed, outFile) != outProcessed)

{

PrintError(rs, kCantWriteMessage);

res = 1;

break;

}

if (outSize < outProcessed)

outSizeHigh--;

outSize -= (UInt32)outProcessed;

outSize &= 0xFFFFFFFF;

if (outProcessed == 0 && finishDecoding)

{

if (!waitEOS && (outSize != 0 || outSizeHigh != 0))

res = 1;

break;

}

}

while ((outSize != 0 && outSizeHigh == 0) || outSizeHigh != 0 || waitEOS);

free(state.Dictionary);

free(state.Probs);

return res;

}

int main2(int numArgs, const char *args[], char *rs)

{

FILE *inFile = 0;

FILE *outFile = 0;

int res;

sprintf(rs + strlen(rs), "\nLZMA Decoder 4.26 Copyright (c) 1999-2005 Igor Pavlov 2005-08-02\n");

if (numArgs < 2 || numArgs > 3)

{

sprintf(rs + strlen(rs), "\nUsage: lzmadec file.lzma [outFile]\n");

return 1;

}

inFile = fopen(args[1], "rb");

if (inFile == 0)

return PrintError(rs, "Can not open input file");

if (numArgs > 2)

{

outFile = fopen(args[2], "wb+");

if (outFile == 0)

return PrintError(rs, "Can not open output file");

}

res = main3(inFile, outFile, rs);

if (outFile != 0)

fclose(outFile);

fclose(inFile);

return res;

}

int main(int numArgs, const char *args[])

{

char rs[800] = { 0 };

int res = main2(numArgs, args, rs);

printf(rs);

return res;

}