package/source/zipapi/Deflater.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/Deflater.hxx>
  21 #include <zlib.h>
  22 #include <com/sun/star/packages/zip/ZipConstants.hpp>
  23 #include <osl/diagnose.h>
  24 #include <string.h>
  25
  26 using namespace com::sun::star::packages::zip::ZipConstants;
  27 using namespace com::sun::star;
  28 using namespace ZipUtils;
  29
  30 /** Provides general purpose compression using the ZLIB compression
  31  * library.
  32  */
  33
  34 Deflater::~Deflater()
  35 {
  36     end();
  37 }
  38 void Deflater::init (sal_Int32 nLevelArg, bool bNowrap)
  39 {
  40     pStream.reset(new z_stream);
  41     /* Memset it to 0...sets zalloc/zfree/opaque to NULL */
  42     memset (pStream.get(), 0, sizeof(*pStream));
  43
  44     switch (deflateInit2(pStream.get(), nLevelArg, Z_DEFLATED, bNowrap? -MAX_WBITS : MAX_WBITS,
  45                 DEF_MEM_LEVEL, DEFAULT_STRATEGY))
  46     {
  47         case Z_OK:
  48             break;
  49         case Z_MEM_ERROR:
  50             pStream.reset();
  51             break;
  52         case Z_STREAM_ERROR:
  53             pStream.reset();
  54             break;
  55         default:
  56              break;
  57     }
  58 }
  59
  60 Deflater::Deflater(sal_Int32 nSetLevel, bool bNowrap)
  61 : bFinish(false)
  62 , bFinished(false)
  63 , nOffset(0)
  64 , nLength(0)
  65 {
  66     init(nSetLevel, bNowrap);
  67 }
  68
  69 sal_Int32 Deflater::doDeflateBytes (uno::Sequence < sal_Int8 > &rBuffer, sal_Int32 nNewOffset, sal_Int32 nNewLength)
  70 {
  71     sal_Int32 nResult;
  72     pStream->next_in   = reinterpret_cast<unsigned char*>(sInBuffer.getArray()) + nOffset;
  73     pStream->next_out  = reinterpret_cast<unsigned char*>(rBuffer.getArray())+nNewOffset;
  74     pStream->avail_in  = nLength;
  75     pStream->avail_out = nNewLength;
  76
  77 #if !defined Z_PREFIX
  78     nResult = deflate(pStream.get(), bFinish ? Z_FINISH : Z_NO_FLUSH);
  79 #else
  80     nResult = z_deflate(pStream.get(), bFinish ? Z_FINISH : Z_NO_FLUSH);
  81 #endif
  82     switch (nResult)
  83     {
  84         case Z_STREAM_END:
  85             bFinished = true;
  86             SAL_FALLTHROUGH;
  87         case Z_OK:
  88             nOffset += nLength - pStream->avail_in;
  89             nLength = pStream->avail_in;
  90             return nNewLength - pStream->avail_out;
  91         default:
  92             return 0;
  93     }
  94 }
  95
  96 void Deflater::setInputSegment( const uno::Sequence< sal_Int8 >& rBuffer )
  97 {
  98     sInBuffer = rBuffer;
  99     nOffset = 0;
 100     nLength = rBuffer.getLength();
 101 }
 102
 103 bool Deflater::needsInput(  )
 104 {
 105     return nLength <=0;
 106 }
 107 void Deflater::finish(  )
 108 {
 109     bFinish = true;
 110 }
 111 sal_Int32 Deflater::doDeflateSegment( uno::Sequence< sal_Int8 >& rBuffer, sal_Int32 nNewOffset, sal_Int32 nNewLength )
 112 {
 113     OSL_ASSERT( !(nNewOffset < 0 || nNewLength < 0 || nNewOffset + nNewLength > rBuffer.getLength()));
 114     return doDeflateBytes(rBuffer, nNewOffset, nNewLength);
 115 }
 116 sal_Int64 Deflater::getTotalIn(  )
 117 {
 118     return pStream->total_in; // FIXME64: zlib doesn't look 64bit clean here
 119 }
 120 sal_Int64 Deflater::getTotalOut(  )
 121 {
 122     return pStream->total_out; // FIXME64: zlib doesn't look 64bit clean here
 123 }
 124 void Deflater::reset(  )
 125 {
 126 #if !defined Z_PREFIX
 127     deflateReset(pStream.get());
 128 #else
 129     z_deflateReset(pStream.get());
 130 #endif
 131     bFinish = false;
 132     bFinished = false;
 133     nOffset = nLength = 0;
 134 }
 135 void Deflater::end(  )
 136 {
 137     if (pStream)
 138     {
 139 #if !defined Z_PREFIX
 140         deflateEnd(pStream.get());
 141 #else
 142         z_deflateEnd(pStream.get());
 143 #endif
 144         pStream.reset();
 145     }
 146 }
 147
 148 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */