Add example of the package transform

1 files changed, 61 insertions, 0 deletions

diff --git a/doc/examples/package.cpp b/doc/examples/package.cpp
new file mode 100644
index 000000000..981abaa31
--- /dev/null
+++ b/doc/examples/package.cpp
@@ -0,0 +1,61 @@
+
+#include <botan/botan.h>
+#include <botan/serpent.h>
+#include <botan/package.h>
+
+#include <iostream>
+#include <fstream>
+#include <vector>
+
+using namespace Botan;
+
+std::vector<byte> slurp_file(const std::string& filename)
+   {
+   std::ifstream in(filename.c_str());
+
+   std::vector<byte> out;
+   byte buf[4096] = { 0 };
+
+   while(in.good())
+      {
+      in.read((char*)buf, sizeof(buf));
+      ssize_t got = in.gcount();
+
+      out.insert(out.end(), buf, buf+got);
+      }
+
+   return out;
+   }
+
+int main(int argc, char* argv[])
+   {
+   if(argc != 2)
+      {
+      std::cout << "Usage: " << argv[0] << " filename\n";
+      return 1;
+      }
+
+   LibraryInitializer init;
+
+   AutoSeeded_RNG rng;
+
+   BlockCipher* cipher = new Serpent;
+
+   std::vector<byte> input = slurp_file(argv[1]);
+   std::vector<byte> output(input.size() + cipher->BLOCK_SIZE);
+
+   AllOrNothingTransform::package(rng, new Serpent,
+                                  &input[0], input.size(),
+                                  &output[0]);
+
+   std::vector<byte> unpackage_output(output.size() - cipher->BLOCK_SIZE);
+
+   AllOrNothingTransform::unpackage(new Serpent,
+                                    &output[0], output.size(),
+                                    &unpackage_output[0]);
+
+   if(unpackage_output == input)
+      std::cout << Package/unpackage worked\n";
+   else
+      std::cout << "Something went wrong :(\n";
+   }