123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864 |
- using System;
- using CommonMPQ.SharpZipLib.Checksums;
- using CommonMPQ.SharpZipLib.Zip.Compression.Streams;
- namespace CommonMPQ.SharpZipLib.Zip.Compression
- {
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- public class Inflater
- {
- #region Constants/Readonly
-
-
-
- static readonly int[] CPLENS = {
- 3, 4, 5, 6, 7, 8, 9, 10, 11, 13, 15, 17, 19, 23, 27, 31,
- 35, 43, 51, 59, 67, 83, 99, 115, 131, 163, 195, 227, 258
- };
-
-
-
-
- static readonly int[] CPLEXT = {
- 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2,
- 3, 3, 3, 3, 4, 4, 4, 4, 5, 5, 5, 5, 0
- };
-
-
-
-
- static readonly int[] CPDIST = {
- 1, 2, 3, 4, 5, 7, 9, 13, 17, 25, 33, 49, 65, 97, 129, 193,
- 257, 385, 513, 769, 1025, 1537, 2049, 3073, 4097, 6145,
- 8193, 12289, 16385, 24577
- };
-
-
-
-
- static readonly int[] CPDEXT = {
- 0, 0, 0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6,
- 7, 7, 8, 8, 9, 9, 10, 10, 11, 11,
- 12, 12, 13, 13
- };
-
-
-
-
- const int DECODE_HEADER = 0;
- const int DECODE_DICT = 1;
- const int DECODE_BLOCKS = 2;
- const int DECODE_STORED_LEN1 = 3;
- const int DECODE_STORED_LEN2 = 4;
- const int DECODE_STORED = 5;
- const int DECODE_DYN_HEADER = 6;
- const int DECODE_HUFFMAN = 7;
- const int DECODE_HUFFMAN_LENBITS = 8;
- const int DECODE_HUFFMAN_DIST = 9;
- const int DECODE_HUFFMAN_DISTBITS = 10;
- const int DECODE_CHKSUM = 11;
- const int FINISHED = 12;
- #endregion
- #region Instance Fields
-
-
-
- int mode;
-
-
-
-
-
-
-
- int readAdler;
-
-
-
-
-
-
- int neededBits;
- int repLength;
- int repDist;
- int uncomprLen;
-
-
-
-
-
-
- bool isLastBlock;
-
-
-
-
- long totalOut;
-
-
-
-
-
-
- long totalIn;
-
-
-
-
-
-
- bool noHeader;
-
- StreamManipulator input;
- OutputWindow outputWindow;
- InflaterDynHeader dynHeader;
- InflaterHuffmanTree litlenTree, distTree;
- Adler32 adler;
- #endregion
-
- #region Constructors
-
-
-
-
- public Inflater() : this(false)
- {
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
- public Inflater(bool noHeader)
- {
- this.noHeader = noHeader;
- this.adler = new Adler32();
- input = new StreamManipulator();
- outputWindow = new OutputWindow();
- mode = noHeader ? DECODE_BLOCKS : DECODE_HEADER;
- }
- #endregion
-
-
-
-
- public void Reset()
- {
- mode = noHeader ? DECODE_BLOCKS : DECODE_HEADER;
- totalIn = 0;
- totalOut = 0;
- input.Reset();
- outputWindow.Reset();
- dynHeader = null;
- litlenTree = null;
- distTree = null;
- isLastBlock = false;
- adler.Reset();
- }
-
-
-
-
-
-
-
-
-
-
- private bool DecodeHeader()
- {
- int header = input.PeekBits(16);
- if (header < 0) {
- return false;
- }
- input.DropBits(16);
-
-
- header = ((header << 8) | (header >> 8)) & 0xffff;
- if (header % 31 != 0) {
- throw new SharpZipBaseException("Header checksum illegal");
- }
-
- if ((header & 0x0f00) != (Deflater.DEFLATED << 8)) {
- throw new SharpZipBaseException("Compression Method unknown");
- }
-
-
-
- if ((header & 0x0020) == 0) {
- mode = DECODE_BLOCKS;
- } else {
- mode = DECODE_DICT;
- neededBits = 32;
- }
- return true;
- }
-
-
-
-
-
-
-
- private bool DecodeDict()
- {
- while (neededBits > 0) {
- int dictByte = input.PeekBits(8);
- if (dictByte < 0) {
- return false;
- }
- input.DropBits(8);
- readAdler = (readAdler << 8) | dictByte;
- neededBits -= 8;
- }
- return false;
- }
-
-
-
-
-
-
-
-
-
-
-
- private bool DecodeHuffman()
- {
- int free = outputWindow.GetFreeSpace();
- while (free >= 258)
- {
- int symbol;
- switch (mode)
- {
- case DECODE_HUFFMAN:
-
- while (((symbol = litlenTree.GetSymbol(input)) & ~0xff) == 0)
- {
- outputWindow.Write(symbol);
- if (--free < 258)
- {
- return true;
- }
- }
-
- if (symbol < 257)
- {
- if (symbol < 0)
- {
- return false;
- }
- else
- {
-
- distTree = null;
- litlenTree = null;
- mode = DECODE_BLOCKS;
- return true;
- }
- }
-
- try
- {
- repLength = CPLENS[symbol - 257];
- neededBits = CPLEXT[symbol - 257];
- }
- catch (Exception)
- {
- throw new SharpZipBaseException("Illegal rep length code");
- }
- goto case DECODE_HUFFMAN_LENBITS;
-
- case DECODE_HUFFMAN_LENBITS:
- if (neededBits > 0)
- {
- mode = DECODE_HUFFMAN_LENBITS;
- int i = input.PeekBits(neededBits);
- if (i < 0)
- {
- return false;
- }
- input.DropBits(neededBits);
- repLength += i;
- }
- mode = DECODE_HUFFMAN_DIST;
- goto case DECODE_HUFFMAN_DIST;
-
- case DECODE_HUFFMAN_DIST:
- symbol = distTree.GetSymbol(input);
- if (symbol < 0)
- {
- return false;
- }
-
- try
- {
- repDist = CPDIST[symbol];
- neededBits = CPDEXT[symbol];
- }
- catch (Exception)
- {
- throw new SharpZipBaseException("Illegal rep dist code");
- }
-
- goto case DECODE_HUFFMAN_DISTBITS;
-
- case DECODE_HUFFMAN_DISTBITS:
- if (neededBits > 0)
- {
- mode = DECODE_HUFFMAN_DISTBITS;
- int i = input.PeekBits(neededBits);
- if (i < 0)
- {
- return false;
- }
- input.DropBits(neededBits);
- repDist += i;
- }
-
- outputWindow.Repeat(repLength, repDist);
- free -= repLength;
- mode = DECODE_HUFFMAN;
- break;
-
- default:
- throw new SharpZipBaseException("Inflater unknown mode");
- }
- }
- return true;
- }
-
-
-
-
-
-
-
-
-
-
- private bool DecodeChksum()
- {
- while (neededBits > 0) {
- int chkByte = input.PeekBits(8);
- if (chkByte < 0) {
- return false;
- }
- input.DropBits(8);
- readAdler = (readAdler << 8) | chkByte;
- neededBits -= 8;
- }
- if ((int) adler.Value != readAdler) {
- throw new SharpZipBaseException("Adler chksum doesn't match: " + (int)adler.Value + " vs. " + readAdler);
- }
- mode = FINISHED;
- return false;
- }
-
-
-
-
-
-
-
-
-
-
- private bool Decode()
- {
- switch (mode) {
- case DECODE_HEADER:
- return DecodeHeader();
- case DECODE_DICT:
- return DecodeDict();
- case DECODE_CHKSUM:
- return DecodeChksum();
-
- case DECODE_BLOCKS:
- if (isLastBlock) {
- if (noHeader) {
- mode = FINISHED;
- return false;
- } else {
- input.SkipToByteBoundary();
- neededBits = 32;
- mode = DECODE_CHKSUM;
- return true;
- }
- }
-
- int type = input.PeekBits(3);
- if (type < 0) {
- return false;
- }
- input.DropBits(3);
-
- if ((type & 1) != 0) {
- isLastBlock = true;
- }
- switch (type >> 1){
- case DeflaterConstants.STORED_BLOCK:
- input.SkipToByteBoundary();
- mode = DECODE_STORED_LEN1;
- break;
- case DeflaterConstants.STATIC_TREES:
- litlenTree = InflaterHuffmanTree.defLitLenTree;
- distTree = InflaterHuffmanTree.defDistTree;
- mode = DECODE_HUFFMAN;
- break;
- case DeflaterConstants.DYN_TREES:
- dynHeader = new InflaterDynHeader();
- mode = DECODE_DYN_HEADER;
- break;
- default:
- throw new SharpZipBaseException("Unknown block type " + type);
- }
- return true;
-
- case DECODE_STORED_LEN1:
- {
- if ((uncomprLen = input.PeekBits(16)) < 0) {
- return false;
- }
- input.DropBits(16);
- mode = DECODE_STORED_LEN2;
- }
- goto case DECODE_STORED_LEN2;
-
- case DECODE_STORED_LEN2:
- {
- int nlen = input.PeekBits(16);
- if (nlen < 0) {
- return false;
- }
- input.DropBits(16);
- if (nlen != (uncomprLen ^ 0xffff)) {
- throw new SharpZipBaseException("broken uncompressed block");
- }
- mode = DECODE_STORED;
- }
- goto case DECODE_STORED;
-
- case DECODE_STORED:
- {
- int more = outputWindow.CopyStored(input, uncomprLen);
- uncomprLen -= more;
- if (uncomprLen == 0) {
- mode = DECODE_BLOCKS;
- return true;
- }
- return !input.IsNeedingInput;
- }
-
- case DECODE_DYN_HEADER:
- if (!dynHeader.Decode(input)) {
- return false;
- }
-
- litlenTree = dynHeader.BuildLitLenTree();
- distTree = dynHeader.BuildDistTree();
- mode = DECODE_HUFFMAN;
- goto case DECODE_HUFFMAN;
-
- case DECODE_HUFFMAN:
- case DECODE_HUFFMAN_LENBITS:
- case DECODE_HUFFMAN_DIST:
- case DECODE_HUFFMAN_DISTBITS:
- return DecodeHuffman();
-
- case FINISHED:
- return false;
-
- default:
- throw new SharpZipBaseException("Inflater.Decode unknown mode");
- }
- }
-
-
-
-
-
-
-
-
-
-
- public void SetDictionary(byte[] buffer)
- {
- SetDictionary(buffer, 0, buffer.Length);
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- public void SetDictionary(byte[] buffer, int index, int count)
- {
- if ( buffer == null ) {
- throw new ArgumentNullException("buffer");
- }
- if ( index < 0 ) {
- throw new ArgumentOutOfRangeException("index");
- }
- if ( count < 0 ) {
- throw new ArgumentOutOfRangeException("count");
- }
- if (!IsNeedingDictionary) {
- throw new InvalidOperationException("Dictionary is not needed");
- }
-
- adler.Update(buffer, index, count);
- if ((int)adler.Value != readAdler) {
- throw new SharpZipBaseException("Wrong adler checksum");
- }
- adler.Reset();
- outputWindow.CopyDict(buffer, index, count);
- mode = DECODE_BLOCKS;
- }
-
-
-
-
-
-
-
-
- public void SetInput(byte[] buffer)
- {
- SetInput(buffer, 0, buffer.Length);
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- public void SetInput(byte[] buffer, int index, int count)
- {
- input.SetInput(buffer, index, count);
- totalIn += (long)count;
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- public int Inflate(byte[] buffer)
- {
- if ( buffer == null )
- {
- throw new ArgumentNullException("buffer");
- }
- return Inflate(buffer, 0, buffer.Length);
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- public int Inflate(byte[] buffer, int offset, int count)
- {
- if ( buffer == null )
- {
- throw new ArgumentNullException("buffer");
- }
- if ( count < 0 ) {
- #if NETCF_1_0
- throw new ArgumentOutOfRangeException("count");
- #else
- throw new ArgumentOutOfRangeException("count", "count cannot be negative");
- #endif
- }
- if ( offset < 0 ) {
- #if NETCF_1_0
- throw new ArgumentOutOfRangeException("offset");
- #else
- throw new ArgumentOutOfRangeException("offset", "offset cannot be negative");
- #endif
- }
- if ( offset + count > buffer.Length ) {
- throw new ArgumentException("count exceeds buffer bounds");
- }
-
- if (count == 0)
- {
- if (!IsFinished) {
- Decode();
- }
- return 0;
- }
- int bytesCopied = 0;
- do {
- if (mode != DECODE_CHKSUM) {
-
- int more = outputWindow.CopyOutput(buffer, offset, count);
- if ( more > 0 ) {
- adler.Update(buffer, offset, more);
- offset += more;
- bytesCopied += more;
- totalOut += (long)more;
- count -= more;
- if (count == 0) {
- return bytesCopied;
- }
- }
- }
- } while (Decode() || ((outputWindow.GetAvailable() > 0) && (mode != DECODE_CHKSUM)));
- return bytesCopied;
- }
-
-
-
-
-
-
- public bool IsNeedingInput {
- get {
- return input.IsNeedingInput;
- }
- }
-
-
-
-
- public bool IsNeedingDictionary {
- get {
- return mode == DECODE_DICT && neededBits == 0;
- }
- }
-
-
-
-
-
- public bool IsFinished {
- get {
- return mode == FINISHED && outputWindow.GetAvailable() == 0;
- }
- }
-
-
-
-
-
-
-
-
-
-
- public int Adler {
- get {
- return IsNeedingDictionary ? readAdler : (int) adler.Value;
- }
- }
-
-
-
-
-
-
-
- public long TotalOut {
- get {
- return totalOut;
- }
- }
-
-
-
-
-
-
-
- public long TotalIn {
- get {
- return totalIn - (long)RemainingInput;
- }
- }
-
-
-
-
-
-
-
-
-
- public int RemainingInput {
-
- get {
- return input.AvailableBytes;
- }
- }
- }
- }
|