mirror of
https://github.com/moparisthebest/mailiverse
synced 2024-11-06 09:25:01 -05:00
71 lines
1.4 KiB
C++
71 lines
1.4 KiB
C++
/*
|
|
* (C) 2009 Jack Lloyd
|
|
*
|
|
* Distributed under the terms of the Botan license
|
|
*/
|
|
|
|
#include <botan/botan.h>
|
|
#include <botan/serpent.h>
|
|
#include <botan/package.h>
|
|
|
|
#include <iostream>
|
|
#include <fstream>
|
|
#include <vector>
|
|
|
|
using namespace Botan;
|
|
|
|
namespace {
|
|
|
|
std::vector<byte> slurp_file(const std::string& filename)
|
|
{
|
|
std::ifstream in(filename.c_str(), std::ios::binary);
|
|
|
|
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());
|
|
|
|
aont_package(rng, new Serpent,
|
|
&input[0], input.size(),
|
|
&output[0]);
|
|
|
|
std::vector<byte> unpackage_output(output.size() - cipher->block_size());
|
|
|
|
aont_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";
|
|
}
|