Compress all instructions (closes #19 github issue)

src/network/Makefile.am

@@ -3,4 +3,4 @@ AM_CXXFLAGS = -pedantic -Wno-long-long -Werror -Wall -Wextra -Weffc++ -fno-defau
 
 noinst_LIBRARIES = libmoshnetwork.a
 
-libmoshnetwork_a_SOURCES = network.cc network.h networktransport.cc networktransport.h transportfragment.cc transportfragment.h transportsender.cc transportsender.h transportstate.h
+libmoshnetwork_a_SOURCES = network.cc network.h networktransport.cc networktransport.h transportfragment.cc transportfragment.h transportsender.cc transportsender.h transportstate.h compressor.cc compressor.h

src/network/compressor.cc

@@ -0,0 +1,32 @@
+#include <zlib.h>
+#include <assert.h>
+
+#include "compressor.h"
+
+using namespace Network;
+using namespace std;
+
+string Compressor::compress_str( const string input )
+{
+  size_t len = BUFFER_SIZE;
+  assert( Z_OK == compress( buffer, &len,
+			    reinterpret_cast<const unsigned char *>( input.data() ),
+			    input.size() ) );
+  return string( reinterpret_cast<char *>( buffer ), len );
+}
+
+string Compressor::uncompress_str( const string input )
+{
+  size_t len = BUFFER_SIZE;
+  assert( Z_OK == uncompress( buffer, &len,
+			      reinterpret_cast<const unsigned char *>( input.data() ),
+			      input.size() ) );
+  return string( reinterpret_cast<char *>( buffer ), len );
+}
+
+/* construct on first use */
+Compressor & Network::get_compressor( void )
+{
+  static Compressor the_compressor;
+  return the_compressor;
+}

src/network/compressor.h

@@ -0,0 +1,28 @@
+#ifndef COMPRESSOR_H
+#define COMPRESSOR_H
+
+#include <string>
+
+namespace Network {
+  class Compressor {
+  private:
+    static const int BUFFER_SIZE = 2048 * 2048; /* effective limit on terminal size */
+
+    unsigned char *buffer;
+
+  public:
+    Compressor() : buffer( NULL ) { buffer = new unsigned char[ BUFFER_SIZE ]; }
+    ~Compressor() { if ( buffer ) { delete[] buffer; } }
+
+    std::string compress_str( const std::string input );
+    std::string uncompress_str( const std::string input );
+
+    /* unused */
+    Compressor( const Compressor & );
+    Compressor & operator=( const Compressor & );
+  };
+
+  Compressor & get_compressor( void );
+}
+
+#endif

src/network/transportfragment.cc

@@ -21,6 +21,7 @@
 
 #include "transportfragment.h"
 #include "transportinstruction.pb.h"
+#include "compressor.h"
 
 using namespace Network;
 using namespace TransportBuffers;
@@ -121,7 +122,7 @@ Instruction FragmentAssembly::get_assembly( void )
   }
 
   Instruction ret;
-  assert( ret.ParseFromString( encoded ) );
+  assert( ret.ParseFromString( get_compressor().uncompress_str( encoded ) ) );
 
   fragments.clear();
   fragments_arrived = 0;
@@ -155,7 +156,7 @@ vector<Fragment> Fragmenter::make_fragments( const Instruction &inst, int MTU )
   last_instruction = inst;
   last_MTU = MTU;
 
-  string payload = inst.SerializeAsString();
+  string payload = get_compressor().compress_str( inst.SerializeAsString() );
   uint16_t fragment_num = 0;
   vector<Fragment> ret;