encode.h

// :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

#define BUFFERSIZE 16777216

#include <iostream>

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;
        }
    };

} // namespace base64

#endif // BASE64_ENCODE_H