1/*
2LzmaStateTest.c
3Test application for LZMA Decoder (State version)
4
5This file written and distributed to public domain by Igor Pavlov.
6This file is part of LZMA SDK 4.26 (2005-08-02)
7*/
8
9#include <stdio.h>
10#include <stdlib.h>
11#include <string.h>
12
13#include "LzmaStateDecode.h"
14
15const char *kCantReadMessage = "Can not read input file";
16const char *kCantWriteMessage = "Can not write output file";
17const char *kCantAllocateMessage = "Can not allocate memory";
18
19#define kInBufferSize (1 << 15)
20#define kOutBufferSize (1 << 15)
21
22unsigned char g_InBuffer[kInBufferSize];
23unsigned char g_OutBuffer[kOutBufferSize];
24
25size_t MyReadFile(FILE *file, void *data, size_t size)
26  { return fread(data, 1, size, file); }
27
28int MyReadFileAndCheck(FILE *file, void *data, size_t size)
29  { return (MyReadFile(file, data, size) == size); }
30
31int PrintError(char *buffer, const char *message)
32{
33  sprintf(buffer + strlen(buffer), "\nError: ");
34  sprintf(buffer + strlen(buffer), message);
35  return 1;
36}
37
38int main3(FILE *inFile, FILE *outFile, char *rs)
39{
40  /* We use two 32-bit integers to construct 64-bit integer for file size.
41     You can remove outSizeHigh, if you don't need >= 4GB supporting,
42     or you can use UInt64 outSize, if your compiler supports 64-bit integers*/
43  UInt32 outSize = 0;
44  UInt32 outSizeHigh = 0;
45
46  int waitEOS = 1;
47  /* waitEOS = 1, if there is no uncompressed size in headers,
48   so decoder will wait EOS (End of Stream Marker) in compressed stream */
49
50  int i;
51  int res = 0;
52  CLzmaDecoderState state;  /* it's about 140 bytes structure, if int is 32-bit */
53  unsigned char properties[LZMA_PROPERTIES_SIZE];
54  SizeT inAvail = 0;
55  unsigned char *inBuffer = 0;
56
57  if (sizeof(UInt32) < 4)
58    return PrintError(rs, "LZMA decoder needs correct UInt32");
59
60  /* Read LZMA properties for compressed stream */
61
62  if (!MyReadFileAndCheck(inFile, properties, sizeof(properties)))
63    return PrintError(rs, kCantReadMessage);
64
65  /* Read uncompressed size */
66
67  for (i = 0; i < 8; i++)
68  {
69    unsigned char b;
70    if (!MyReadFileAndCheck(inFile, &b, 1))
71      return PrintError(rs, kCantReadMessage);
72    if (b != 0xFF)
73      waitEOS = 0;
74    if (i < 4)
75      outSize += (UInt32)(b) << (i * 8);
76    else
77      outSizeHigh += (UInt32)(b) << ((i - 4) * 8);
78  }
79
80  /* Decode LZMA properties and allocate memory */
81
82  if (LzmaDecodeProperties(&state.Properties, properties, LZMA_PROPERTIES_SIZE) != LZMA_RESULT_OK)
83    return PrintError(rs, "Incorrect stream properties");
84  state.Probs = (CProb *)malloc(LzmaGetNumProbs(&state.Properties) * sizeof(CProb));
85  if (state.Probs == 0)
86    return PrintError(rs, kCantAllocateMessage);
87
88  if (state.Properties.DictionarySize == 0)
89    state.Dictionary = 0;
90  else
91  {
92    state.Dictionary = (unsigned char *)malloc(state.Properties.DictionarySize);
93    if (state.Dictionary == 0)
94    {
95      free(state.Probs);
96      return PrintError(rs, kCantAllocateMessage);
97    }
98  }
99
100  /* Decompress */
101
102  LzmaDecoderInit(&state);
103
104  do
105  {
106    SizeT inProcessed, outProcessed;
107    int finishDecoding;
108    UInt32 outAvail = kOutBufferSize;
109    if (!waitEOS && outSizeHigh == 0 && outAvail > outSize)
110      outAvail = outSize;
111    if (inAvail == 0)
112    {
113      inAvail = (SizeT)MyReadFile(inFile, g_InBuffer, kInBufferSize);
114      inBuffer = g_InBuffer;
115    }
116    finishDecoding = (inAvail == 0);
117    res = LzmaDecode(&state,
118        inBuffer, inAvail, &inProcessed,
119        g_OutBuffer, outAvail, &outProcessed,
120        finishDecoding);
121    if (res != 0)
122    {
123      sprintf(rs + strlen(rs), "\nDecoding error = %d\n", res);
124      res = 1;
125      break;
126    }
127    inAvail -= inProcessed;
128    inBuffer += inProcessed;
129
130    if (outFile != 0)
131      if (fwrite(g_OutBuffer, 1, outProcessed, outFile) != outProcessed)
132      {
133        PrintError(rs, kCantWriteMessage);
134        res = 1;
135        break;
136      }
137
138    if (outSize < outProcessed)
139      outSizeHigh--;
140    outSize -= (UInt32)outProcessed;
141    outSize &= 0xFFFFFFFF;
142
143    if (outProcessed == 0 && finishDecoding)
144    {
145      if (!waitEOS && (outSize != 0 || outSizeHigh != 0))
146        res = 1;
147      break;
148    }
149  }
150  while ((outSize != 0 && outSizeHigh == 0) || outSizeHigh != 0  || waitEOS);
151
152  free(state.Dictionary);
153  free(state.Probs);
154  return res;
155}
156
157int main2(int numArgs, const char *args[], char *rs)
158{
159  FILE *inFile = 0;
160  FILE *outFile = 0;
161  int res;
162
163  sprintf(rs + strlen(rs), "\nLZMA Decoder 4.26 Copyright (c) 1999-2005 Igor Pavlov  2005-08-02\n");
164  if (numArgs < 2 || numArgs > 3)
165  {
166    sprintf(rs + strlen(rs), "\nUsage:  lzmadec file.lzma [outFile]\n");
167    return 1;
168  }
169
170  inFile = fopen(args[1], "rb");
171  if (inFile == 0)
172    return PrintError(rs, "Can not open input file");
173
174  if (numArgs > 2)
175  {
176    outFile = fopen(args[2], "wb+");
177    if (outFile == 0)
178      return PrintError(rs, "Can not open output file");
179  }
180
181  res = main3(inFile, outFile, rs);
182
183  if (outFile != 0)
184    fclose(outFile);
185  fclose(inFile);
186  return res;
187}
188
189int main(int numArgs, const char *args[])
190{
191  char rs[800] = { 0 };
192  int res = main2(numArgs, args, rs);
193  printf(rs);
194  return res;
195}
196