1 files changed, 79 insertions, 0 deletions
diff --git a/src/cmd/compress.cpp b/src/cmd/compress.cpp
new file mode 100644
index 000000000..fc1ece7bd
--- /dev/null
+++ b/src/cmd/compress.cpp
@@ -0,0 +1,79 @@
+
+#include "apps.h"
+
+#include <botan/compression.h>
+#include <fstream>
+
+namespace {
+
+void do_compress(Transformation& comp, std::ifstream& in, std::ostream& out)
+   {
+   secure_vector<byte> buf;
+
+   comp.start();
+
+   while(in.good())
+      {
+      buf.resize(16*1024);
+      in.read(reinterpret_cast<char*>(&buf[0]), buf.size());
+      buf.resize(in.gcount());
+
+      comp.update(buf);
+
+      out.write(reinterpret_cast<const char*>(&buf[0]), buf.size());
+      }
+
+   buf.clear();
+   comp.finish(buf);
+   out.write(reinterpret_cast<const char*>(&buf[0]), buf.size());
+   }
+
+int compress(int argc, char* argv[])
+   {
+   const std::string in_file = argv[1];
+   std::ifstream in(in_file.c_str());
+
+   if(!in.good())
+      {
+      std::cout << "Couldn't read " << in_file << "\n";
+      return 1;
+      }
+
+   const size_t level = 6;
+   const std::string suffix = "gz";
+
+   std::unique_ptr<Transformation> compress(make_compressor(suffix, level));
+
+   const std::string out_file = in_file + "." + suffix;
+   std::ofstream out(out_file.c_str());
+
+   do_compress(*compress, in, out);
+
+   return 0;
+   }
+
+int uncompress(int argc, char* argv[])
+   {
+   const std::string in_file = argv[1];
+   std::ifstream in(in_file.c_str());
+
+   if(!in.good())
+      {
+      std::cout << "Couldn't read " << argv[1] << "\n";
+      return 1;
+      }
+
+   std::ofstream out("out");
+   const std::string suffix = "gz";
+
+   std::unique_ptr<Transformation> decompress(make_decompressor(suffix));
+
+   do_compress(*decompress, in, out);
+
+   return 0;
+   }
+
+REGISTER_APP(compress);
+REGISTER_APP(uncompress);
+
+}