source/core/util/ZBlock.cpp

   1 // ANNA - Anna is Not Nothingness Anymore
   2 //
   3 // (c) Copyright 2005-2014 Eduardo Ramos Testillano & Francisco Ruiz Rayo
   4 //
   5 // https://bitbucket.org/testillano/anna
   6 //
   7 // Redistribution and use in source and binary forms, with or without
   8 // modification, are permitted provided that the following conditions
   9 // are met:
  10 //
  11 //     * Redistributions of source code must retain the above copyright
  12 // notice, this list of conditions and the following disclaimer.
  13 //     * Redistributions in binary form must reproduce the above
  14 // copyright notice, this list of conditions and the following disclaimer
  15 // in the documentation and/or other materials provided with the
  16 // distribution.
  17 //     * Neither the name of Google Inc. nor the names of its
  18 // contributors may be used to endorse or promote products derived from
  19 // this software without specific prior written permission.
  20 //
  21 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  24 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  25 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  26 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  27 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  28 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  29 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  30 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  31 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32 //
  33 // Authors: eduardo.ramos.testillano@gmail.com
  34 //          cisco.tierra@gmail.com
  35
  36
  37 #include <arpa/inet.h>
  38
  39 #include <zlib.h>
  40
  41 #include <anna/core/util/ZBlock.hpp>
  42
  43 using namespace std;
  44 using namespace anna;
  45
  46 /*
  47  * El bloque de datos retornoado contiene:
  48  * <int que indica el tamaño original del buffer> + buffer comprimido.
  49  */
  50 const DataBlock& ZBlock::compress(const DataBlock& data, const Mode::_v mode)
  51 throw(RuntimeException) {
  52   if(&data == this)
  53     throw RuntimeException("ZBlock::compress | Source and target object cannot be the same", ANNA_FILE_LOCATION);
  54
  55   int originalSize;
  56
  57   /*
  58    * Si la fuente de datos está vacía no hay que hacer nada
  59    */
  60   if((originalSize = data.getSize()) == 0) {
  61     DataBlock::clear();
  62     return *this;
  63   }
  64
  65   /*
  66    * Según la documentación el buffer destino debe ser, como mínimo un 0.1% mayor que el original + 12 bytes
  67    * Lo incrementamos en un 1% para simplificar
  68    */
  69   uLong maxSize = sizeof(int) + 12 + originalSize + (originalSize * 101) / 100;
  70   DataBlock::allocate(maxSize);
  71   char* buffer = (char*) DataBlock::getData();
  72   int aux = htonl(originalSize);
  73   register char* w((char*) &aux);
  74   /* Establece el tamaño original del buffer */
  75   buffer [0] = *w;
  76   buffer [1] = *(w + 1);
  77   buffer [2] = *(w + 2);
  78   buffer [3] = *(w + 3);
  79   int rr;
  80
  81   if(mode == Mode::Default)
  82     rr = ::compress((Bytef*) buffer + sizeof(int), &maxSize, (Bytef*) data.getData(), originalSize);
  83   else
  84     rr = ::compress2((Bytef*) buffer + sizeof(int), &maxSize, (Bytef*) data.getData(), originalSize, mode);
  85
  86   if(rr != Z_OK) {
  87     string msg("ZBlock::compress | ");
  88     msg += zError(rr);
  89     throw RuntimeException(msg, ANNA_FILE_LOCATION);
  90   }
  91
  92   DataBlock::setSize((int) maxSize + sizeof(int));
  93   return *this;
  94 }
  95
  96 const DataBlock& ZBlock::uncompress(const DataBlock& zdata)
  97 throw(RuntimeException) {
  98   if(&zdata == this)
  99     throw RuntimeException("ZBlock::uncompress | Source and target object cannot be the same", ANNA_FILE_LOCATION);
 100
 101   /*
 102    * Si la fuente de datos está vacía no hay que hacer nada
 103    */
 104   if(zdata.getSize() < sizeof(int)) {
 105     DataBlock::clear();
 106     return *this;
 107   }
 108
 109   char* buffer = (char*) zdata.getData();
 110   int aux;
 111   uLong size;
 112   register char* w((char*) &aux);
 113   *w  = *buffer;
 114   *(w + 1) = *(buffer + 1);
 115   *(w + 2) = *(buffer + 2);
 116   *(w + 3) = *(buffer + 3);
 117   size = ntohl(aux);
 118   DataBlock::allocate((int) size);
 119   int rr = ::uncompress((Bytef*) DataBlock::getData(), &size, (const Bytef*) zdata.getData() + sizeof(int), zdata.getSize() - sizeof(int));
 120
 121   if(rr != Z_OK) {
 122     string msg("ZBlock::uncompress | ");
 123     msg += zError(rr);
 124     throw RuntimeException(msg, ANNA_FILE_LOCATION);
 125   }
 126
 127   DataBlock::setSize((int) size);
 128   return *this;
 129 }