package/source/zipapi/Inflater.cxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /*
   3  * This file is part of the LibreOffice project.
   4  *
   5  * This Source Code Form is subject to the terms of the Mozilla Public
   6  * License, v. 2.0. If a copy of the MPL was not distributed with this
   7  * file, You can obtain one at http://mozilla.org/MPL/2.0/.
   8  *
   9  * This file incorporates work covered by the following license notice:
  10  *
  11  *   Licensed to the Apache Software Foundation (ASF) under one or more
  12  *   contributor license agreements. See the NOTICE file distributed
  13  *   with this work for additional information regarding copyright
  14  *   ownership. The ASF licenses this file to you under the Apache
  15  *   License, Version 2.0 (the "License"); you may not use this file
  16  *   except in compliance with the License. You may obtain a copy of
  17  *   the License at http://www.apache.org/licenses/LICENSE-2.0 .
  18  */
  19
  20 #include <package/Inflater.hxx>
  21 #include <zlib.h>
  22 #include <string.h>
  23
  24 using namespace com::sun::star::uno;
  25 using namespace ZipUtils;
  26
  27 /** Provides general purpose decompression using the ZLIB library */
  28
  29 Inflater::Inflater(bool bNoWrap)
  30 : bFinished(false),
  31   bNeedDict(false),
  32   nOffset(0),
  33   nLength(0),
  34   nLastInflateError(0)
  35 {
  36     pStream.reset(new z_stream);
  37     /* memset to 0 to set zalloc/opaque etc */
  38     memset (pStream.get(), 0, sizeof(*pStream));
  39     sal_Int32 nRes;
  40     nRes = inflateInit2(pStream.get(), bNoWrap ? -MAX_WBITS : MAX_WBITS);
  41     switch (nRes)
  42     {
  43         case Z_OK:
  44             break;
  45         case Z_MEM_ERROR:
  46             pStream.reset();
  47             break;
  48         case Z_STREAM_ERROR:
  49             pStream.reset();
  50             break;
  51         default:
  52             break;
  53     }
  54 }
  55
  56 Inflater::~Inflater()
  57 {
  58     end();
  59 }
  60
  61 void Inflater::setInput( const Sequence< sal_Int8 >& rBuffer )
  62 {
  63     sInBuffer = rBuffer;
  64     nOffset = 0;
  65     nLength = rBuffer.getLength();
  66 }
  67
  68
  69 sal_Int32 Inflater::doInflateSegment( Sequence< sal_Int8 >& rBuffer, sal_Int32 nNewOffset, sal_Int32 nNewLength )
  70 {
  71     if (nNewOffset < 0 || nNewLength < 0 || nNewOffset + nNewLength > rBuffer.getLength())
  72     {
  73         // do error handling
  74     }
  75     return doInflateBytes(rBuffer, nNewOffset, nNewLength);
  76 }
  77
  78 void Inflater::end(  )
  79 {
  80     if (pStream)
  81     {
  82 #if !defined Z_PREFIX
  83         inflateEnd(pStream.get());
  84 #else
  85         z_inflateEnd(pStream.get());
  86 #endif
  87         pStream.reset();
  88     }
  89 }
  90
  91 sal_Int32 Inflater::doInflateBytes (Sequence < sal_Int8 >  &rBuffer, sal_Int32 nNewOffset, sal_Int32 nNewLength)
  92 {
  93     if ( !pStream )
  94     {
  95         nLastInflateError = Z_STREAM_ERROR;
  96         return 0;
  97     }
  98
  99     nLastInflateError = 0;
 100
 101     pStream->next_in   = reinterpret_cast<const unsigned char*>( sInBuffer.getConstArray() + nOffset );
 102     pStream->avail_in  = nLength;
 103     pStream->next_out  = reinterpret_cast < unsigned char* > ( rBuffer.getArray() + nNewOffset );
 104     pStream->avail_out = nNewLength;
 105
 106 #if !defined Z_PREFIX
 107     sal_Int32 nResult = ::inflate(pStream.get(), Z_PARTIAL_FLUSH);
 108 #else
 109     sal_Int32 nResult = ::z_inflate(pStream.get(), Z_PARTIAL_FLUSH);
 110 #endif
 111
 112     switch (nResult)
 113     {
 114         case Z_STREAM_END:
 115             bFinished = true;
 116             [[fallthrough]];
 117         case Z_OK:
 118             nOffset += nLength - pStream->avail_in;
 119             nLength = pStream->avail_in;
 120             return nNewLength - pStream->avail_out;
 121
 122         case Z_NEED_DICT:
 123             bNeedDict = true;
 124             nOffset += nLength - pStream->avail_in;
 125             nLength = pStream->avail_in;
 126             return 0;
 127
 128         default:
 129             // it is no error, if there is no input or no output
 130             if ( nLength && nNewLength )
 131                 nLastInflateError = nResult;
 132     }
 133
 134     return 0;
 135 }
 136
 137 InflaterBytes::InflaterBytes(bool bNoWrap)
 138 : bFinished(false),
 139   nOffset(0),
 140   nLength(0),
 141   sInBuffer(nullptr)
 142 {
 143     pStream.reset(new z_stream);
 144     /* memset to 0 to set zalloc/opaque etc */
 145     memset (pStream.get(), 0, sizeof(*pStream));
 146     sal_Int32 nRes;
 147     nRes = inflateInit2(pStream.get(), bNoWrap ? -MAX_WBITS : MAX_WBITS);
 148     switch (nRes)
 149     {
 150         case Z_OK:
 151             break;
 152         case Z_MEM_ERROR:
 153             pStream.reset();
 154             break;
 155         case Z_STREAM_ERROR:
 156             pStream.reset();
 157             break;
 158         default:
 159             break;
 160     }
 161 }
 162
 163 InflaterBytes::~InflaterBytes()
 164 {
 165     end();
 166 }
 167
 168 void InflaterBytes::setInput( const sal_Int8* rBuffer, sal_Int32 nBufLen )
 169 {
 170     sInBuffer = rBuffer;
 171     nOffset = 0;
 172     nLength = nBufLen;
 173 }
 174
 175
 176 sal_Int32 InflaterBytes::doInflateSegment( sal_Int8* pOutBuffer, sal_Int32 nBufLen, sal_Int32 nNewOffset, sal_Int32 nNewLength )
 177 {
 178     if (nNewOffset < 0 || nNewLength < 0 || nNewOffset + nNewLength > nBufLen)
 179     {
 180         // do error handling
 181     }
 182     return doInflateBytes(pOutBuffer, nNewOffset, nNewLength);
 183 }
 184
 185 void InflaterBytes::end(  )
 186 {
 187     if (pStream)
 188     {
 189 #if !defined Z_PREFIX
 190         inflateEnd(pStream.get());
 191 #else
 192         z_inflateEnd(pStream.get());
 193 #endif
 194         pStream.reset();
 195     }
 196 }
 197
 198 sal_Int32 InflaterBytes::doInflateBytes (sal_Int8* pOutBuffer, sal_Int32 nNewOffset, sal_Int32 nNewLength)
 199 {
 200     if ( !pStream )
 201     {
 202         return 0;
 203     }
 204
 205     pStream->next_in   = reinterpret_cast<const unsigned char*>( sInBuffer + nOffset );
 206     pStream->avail_in  = nLength;
 207     pStream->next_out  = reinterpret_cast < unsigned char* > ( pOutBuffer + nNewOffset );
 208     pStream->avail_out = nNewLength;
 209
 210 #if !defined Z_PREFIX
 211     sal_Int32 nResult = ::inflate(pStream.get(), Z_PARTIAL_FLUSH);
 212 #else
 213     sal_Int32 nResult = ::z_inflate(pStream.get(), Z_PARTIAL_FLUSH);
 214 #endif
 215
 216     switch (nResult)
 217     {
 218         case Z_STREAM_END:
 219             bFinished = true;
 220             [[fallthrough]];
 221         case Z_OK:
 222             nOffset += nLength - pStream->avail_in;
 223             nLength = pStream->avail_in;
 224             return nNewLength - pStream->avail_out;
 225
 226         case Z_NEED_DICT:
 227             nOffset += nLength - pStream->avail_in;
 228             nLength = pStream->avail_in;
 229             return 0;
 230
 231         default:
 232             // it is no error, if there is no input or no output
 233             break;
 234     }
 235
 236     return 0;
 237 }
 238
 239 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */