You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
1.6 KiB
86 lines
1.6 KiB
![]()
7 years ago
|
#include <unistd.h>
|
||
|
#include <fcntl.h>
|
||
|
#include <iostream>
|
||
|
#include <fstream>
|
||
|
#include <functional>
|
||
|
#include <cassert>
|
||
|
#include "Base.h"
|
||
|
|
||
|
const size_t BUFFSZ = 1024;
|
||
|
|
||
|
static int printHelp(const char * exe, int exitcode)
|
||
|
{
|
||
|
std::cout << "usage: " << exe << " [-d] [filename]" << std::endl;
|
||
|
return exitcode;
|
||
|
}
|
||
|
|
||
|
template <typename InCh, typename OutCh, size_t isz, size_t osz>
|
||
|
static int operate(std::function<std::size_t(const InCh *, size_t, OutCh *, size_t)> f, int infile, int outfile)
|
||
|
{
|
||
|
InCh inbuf[isz];
|
||
|
OutCh outbuf[osz];
|
||
|
ssize_t sz;
|
||
|
size_t outsz;
|
||
|
while((sz = read(infile, inbuf, sizeof(inbuf))) > 0)
|
||
|
{
|
||
|
outsz = f(inbuf, sz, outbuf, sizeof(outbuf));
|
||
|
if(outsz && outsz <= sizeof(outbuf))
|
||
|
{
|
||
|
write(outfile, outbuf, outsz);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
return -1;
|
||
|
}
|
||
|
}
|
||
|
return errno;
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
int main(int argc, char * argv[])
|
||
|
{
|
||
|
int opt;
|
||
|
bool decode = false;
|
||
|
int infile = 0;
|
||
|
while((opt = getopt(argc, argv, "dh")) != -1)
|
||
|
{
|
||
|
switch(opt)
|
||
|
{
|
||
|
case 'h':
|
||
|
return printHelp(argv[0], 0);
|
||
|
case 'd':
|
||
|
decode = true;
|
||
|
break;
|
||
|
default:
|
||
|
continue;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (argc - optind > 1)
|
||
|
{
|
||
|
return printHelp(argv[0], -1);
|
||
|
}
|
||
|
|
||
|
if (optind < argc)
|
||
|
{
|
||
|
infile = open(argv[optind], O_RDONLY);
|
||
|
if(infile == -1) {
|
||
|
perror(argv[optind]);
|
||
|
return -1;
|
||
|
}
|
||
|
}
|
||
|
int retcode = 0;
|
||
|
if(decode)
|
||
|
{
|
||
|
retcode = operate<char, uint8_t, BUFFSZ*4, BUFFSZ*3>(i2p::data::Base64ToByteStream, infile, 1);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
retcode = operate<uint8_t, char, BUFFSZ*3, BUFFSZ*4>(&i2p::data::ByteStreamToBase64, infile, 1);
|
||
|
}
|
||
|
close(infile);
|
||
|
return retcode;
|
||
|
}
|