commit 10a90fff54090ca880d4ffbee3ad0dfbf31d11e5
author Ashlesh Gawande <agawande@memphis.edu> Thu Nov 01 16:11:51 2018 -0500
committer Ashlesh Gawande <agawande@memphis.edu> Thu Dec 20 21:40:55 2018 -0600
tree 835938ff3a543f03fa8a8bea580f20765596cfed
parent 78b94adbc2afde4be71901cb4959c507b45fce70

iblt: add zlib compression and decompression

refs: #4671

Change-Id: I44609dfdf820cccdb2cc6ff437c88d3880a41d9d

PSync/detail/iblt.cpp

diff --git a/PSync/detail/iblt.cpp b/PSync/detail/iblt.cpp
index d2a2161..5de61ca 100644
--- a/PSync/detail/iblt.cpp
+++ b/PSync/detail/iblt.cpp
@@ -46,10 +46,15 @@
 #include "PSync/detail/iblt.hpp"
 #include "PSync/detail/util.hpp"
 
-#include <sstream>
+#include <boost/iostreams/device/array.hpp>
+#include <boost/iostreams/filtering_stream.hpp>
+#include <boost/iostreams/filter/zlib.hpp>
+#include <boost/iostreams/copy.hpp>
 
 namespace psync {
 
+namespace bio = boost::iostreams;
+
 const size_t N_HASH(3);
 const size_t N_HASHCHECK(11);
 
@@ -227,7 +232,7 @@
   size_t unitSize = (32 * 3) / 8; // hard coding
   size_t tableSize = unitSize * n;
 
-  std::vector <uint8_t> table(tableSize);
+  std::vector<char> table(tableSize);
 
   for (size_t i = 0; i < n; i++) {
     // table[i*12],   table[i*12+1], table[i*12+2], table[i*12+3] --> hashTable[i].count
@@ -252,13 +257,31 @@
     table[(i * unitSize) + 11] = 0xFF & (m_hashTable[i].keyCheck >> 24);
   }
 
-  name.append(table.begin(), table.end());
+  bio::filtering_streambuf<bio::input> in;
+  in.push(bio::zlib_compressor());
+  in.push(bio::array_source(table.data(), table.size()));
+
+  std::stringstream sstream;
+  bio::copy(in, sstream);
+
+  std::string compressedIBF = sstream.str();
+  name.append(compressedIBF.begin(), compressedIBF.end());
 }
 
 std::vector<uint32_t>
 IBLT::extractValueFromName(const ndn::name::Component& ibltName) const
 {
-  std::vector<uint8_t> ibltValues(ibltName.value_begin(), ibltName.value_end());
+  std::string compressed(ibltName.value_begin(), ibltName.value_end());
+
+  bio::filtering_streambuf<bio::input> in;
+  in.push(bio::zlib_decompressor());
+  in.push(bio::array_source(compressed.data(), compressed.size()));
+
+  std::stringstream sstream;
+  bio::copy(in, sstream);
+  std::string ibltStr = sstream.str();
+
+  std::vector<uint8_t> ibltValues(ibltStr.begin(), ibltStr.end());
   size_t n = ibltValues.size() / 4;
 
   std::vector<uint32_t> values(n, 0);