foundationdb/fdbclient/libb64/encode.h

73 lines
1.6 KiB
C++

// :mode=c++:
/*
encode.h - c++ wrapper for a base64 encoding algorithm
This is part of the libb64 project, and has been placed in the public domain.
For details, see http://sourceforge.net/projects/libb64
*/
#ifndef BASE64_ENCODE_H
#define BASE64_ENCODE_H
#include <iostream>
#include <sstream>
#define BUFFERSIZE 8192
namespace base64 {
extern "C" {
#include "cencode.h"
}
struct encoder {
base64_encodestate _state;
int _buffersize;
encoder(int buffersize_in = BUFFERSIZE) : _buffersize(buffersize_in) {}
int encode(char value_in) { return base64_encode_value(value_in); }
int encode(const char* code_in, const int length_in, char* plaintext_out) {
return base64_encode_block(code_in, length_in, plaintext_out, &_state);
}
int encode_end(char* plaintext_out) { return base64_encode_blockend(plaintext_out, &_state); }
void encode(std::istream& istream_in, std::ostream& ostream_in) {
base64_init_encodestate(&_state);
//
const int N = _buffersize;
char* plaintext = new char[N];
char* code = new char[2 * N];
int plainlength;
int codelength;
do {
istream_in.read(plaintext, N);
plainlength = istream_in.gcount();
//
codelength = encode(plaintext, plainlength, code);
ostream_in.write(code, codelength);
} while (istream_in.good() && plainlength > 0);
codelength = encode_end(code);
ostream_in.write(code, codelength);
//
base64_init_encodestate(&_state);
delete[] code;
delete[] plaintext;
}
static std::string from_string(std::string s) {
std::stringstream in(s);
std::stringstream out;
encoder enc;
enc.encode(in, out);
return out.str();
}
};
} // namespace base64
#endif // BASE64_ENCODE_H