1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
//========= Copyright Valve Corporation, All rights reserved. ============//
//
// LZMA Codec interface for engine.
//
// LZMA SDK 9.38 beta
// 2015-01-03 : Igor Pavlov : Public domain
// http://www.7-zip.org/
//
//========================================================================//
#ifndef _LZMADECODER_H
#define _LZMADECODER_H
#pragma once
// Thanks for the useful define namespacing, LZMA
#include "../../utils/lzma/C/7zVersion.h"
#define LZMA_SDK_VERSION_MAJOR MY_VER_MAJOR
#define LZMA_SDK_VERSION_MINOR MY_VER_MINOR
#if !defined( _X360 )
#define LZMA_ID (('A'<<24)|('M'<<16)|('Z'<<8)|('L'))
#else
#define LZMA_ID (('L'<<24)|('Z'<<16)|('M'<<8)|('A'))
#endif
// bind the buffer for correct identification
#pragma pack(1)
struct lzma_header_t
{
unsigned int id;
unsigned int actualSize; // always little endian
unsigned int lzmaSize; // always little endian
unsigned char properties[5];
};
#pragma pack()
class CLZMAStream;
class CLZMA
{
public:
static unsigned int Uncompress( unsigned char *pInput, unsigned char *pOutput );
static bool IsCompressed( unsigned char *pInput );
static unsigned int GetActualSize( unsigned char *pInput );
};
// For files besides the implementation, we forward declare a dummy struct. We can't unconditionally forward declare
// this because LzmaEnc.h typedefs this directly to an unnamed struct :-/
#ifndef CLzmaDec_t
struct _CLzmaDec_t;
#define CLzmaDec_t struct _CLzmaDec_t
#endif
class CLZMAStream
{
public:
CLZMAStream();
~CLZMAStream();
// Initialize a stream to read data from a LZMA style zip file, passing the original size from the zip headers.
// Streams with a source-engine style header (lzma_header_t) do not need an init call.
void InitZIPHeader( unsigned int nCompressedSize, unsigned int nOriginalSize );
// Attempt to read up to nMaxInputBytes from the compressed stream, writing up to nMaxOutputBytes to pOutput.
// Makes progress until blocked on input or output.
// Returns false if read stops due to an error or if called at EOF (GetExpectedBytesRemaining == 0)
bool Read( unsigned char *pInput, unsigned int nMaxInputBytes,
unsigned char *pOutput, unsigned int nMaxOutputBytes,
/* out */ unsigned int &nCompressedBytesRead, /* out */ unsigned int &nOutputBytesWritten );
// Get the expected uncompressed bytes yet to be read from this stream. Returns false if not yet known, such as
// before being fed the header.
bool GetExpectedBytesRemaining( /* out */ unsigned int &nBytesRemaining );
private:
enum eHeaderParse
{
eHeaderParse_OK,
eHeaderParse_Fail,
eHeaderParse_NeedMoreBytes
};
eHeaderParse TryParseHeader( unsigned char *pInput, unsigned int nBytesAvailable, /* out */ unsigned int &nBytesConsumed );
void FreeDecoderState();
bool CreateDecoderState( const unsigned char *pProperties );
// Init from a zip-embedded LZMA stream. Requires the original size be passed from zip headers.
CLzmaDec_t *m_pDecoderState;
unsigned int m_nActualSize;
unsigned int m_nActualBytesRead;
unsigned int m_nCompressedSize;
unsigned int m_nCompressedBytesRead;
// If we have read past the header
bool m_bParsedHeader : 1;
// If InitZIPHeader() was called. We're expecting a zip-style header and have size information.
bool m_bZIPStyleHeader : 1;
};
#endif
|