File: decompress.cc

package info (click to toggle)
goldendict 1.5.0~rc2%2Bgit20181207%2Bds-1
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 16,496 kB
  • sloc: cpp: 58,990; ansic: 11,311; xml: 488; makefile: 77; sh: 42
file content (134 lines) | stat: -rw-r--r-- 3,062 bytes parent folder | download | duplicates (2)
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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
#include "decompress.hh"
#include "zlib.h"
#include "bzlib.h"

#ifdef MAKE_ZIM_SUPPORT
#include "lzma.h"
#endif

#define CHUNK_SIZE 2048

QByteArray zlibDecompress( const char * bufptr, unsigned length )
{
z_stream zs;
char buf[CHUNK_SIZE];
QByteArray str;
int res;
    memset( &zs, 0, sizeof(zs) );
    zs.next_in = (Bytef *)bufptr;
    zs.avail_in = length;
    while( 1 )
    {
        res = inflateInit( &zs );
        if( res != Z_OK )
            break;
        while( res != Z_STREAM_END )
        {
            zs.next_out = (Bytef *)buf;
            zs.avail_out = CHUNK_SIZE;
            res = inflate( &zs, Z_SYNC_FLUSH );
            str.append( buf, CHUNK_SIZE - zs.avail_out );
            if( res != Z_OK && res != Z_STREAM_END )
                 break;
        }
        break;
    }
    inflateEnd( &zs );
    if( res != Z_STREAM_END )
        str.clear();
    return str;
}

string decompressZlib( const char * bufptr, unsigned length )
{
  QByteArray b = zlibDecompress( bufptr, length );
  return string( b.constData(), b.size() );
}

string decompressBzip2( const char * bufptr, unsigned length )
{
bz_stream zs;
char buf[CHUNK_SIZE];
string str;
int res;
    memset( &zs, 0, sizeof(zs) );
    zs.next_in = (char *)bufptr;
    zs.avail_in = length;
    zs.total_in_lo32 = length;
    while( 1 )
    {
        res = BZ2_bzDecompressInit( &zs, 0, 0 );
        if( res != BZ_OK )
            break;
        while( res != BZ_STREAM_END )
        {
            zs.next_out = buf;
            zs.avail_out = CHUNK_SIZE;
            zs.total_out_lo32 = length;
            res = BZ2_bzDecompress( &zs );
            str.append( buf, CHUNK_SIZE - zs.avail_out );
            if( res != BZ_OK && res != BZ_STREAM_END )
                break;
        }
        break;
    }
    BZ2_bzDecompressEnd( &zs );
    if( res != BZ_STREAM_END )
        str.clear();
    return str;
}

#ifdef MAKE_ZIM_SUPPORT

string decompressLzma2( const char * bufptr, unsigned length,
                        bool raw_decoder )
{
string str;
lzma_ret res;
char buf[CHUNK_SIZE];

  lzma_stream strm = LZMA_STREAM_INIT;
  strm.next_in = reinterpret_cast< const uint8_t * >( bufptr );
  strm.avail_in = length;

  lzma_options_lzma opt;
  lzma_filter filters[ 2 ];

  if( raw_decoder )
  {
    lzma_lzma_preset(&opt, LZMA_PRESET_DEFAULT);

    filters[ 0 ].id = LZMA_FILTER_LZMA2;
    filters[ 0 ].options = &opt;
    filters[ 1 ].id = LZMA_VLI_UNKNOWN;
  }

  while( 1 )
  {
    if( raw_decoder )
      res = lzma_raw_decoder( &strm, filters );
    else
      res = lzma_stream_decoder( &strm, UINT64_MAX, 0 );

    if( res != LZMA_OK )
      break;

    while ( res != LZMA_STREAM_END )
    {
      strm.next_out = reinterpret_cast< uint8_t * >( buf );
      strm.avail_out = CHUNK_SIZE;
      res = lzma_code( &strm, LZMA_RUN );
      str.append( buf, CHUNK_SIZE - strm.avail_out );
      if( res != LZMA_OK && res != LZMA_STREAM_END )
        break;
    }
    lzma_end( &strm );
    if( res != LZMA_STREAM_END )
      str.clear();

    break;
  }
  return str;
}

#endif