ibrcommon/ibrcommon/ssl/CipherStream.cpp

Go to the documentation of this file.

/*
 * CipherStream.cpp
 *
 *  Created on: 13.07.2010
 *      Author: morgenro
 */

#include "ibrcommon/ssl/CipherStream.h"

namespace ibrcommon
{
        CipherStream::CipherStream(std::ostream &stream, const CipherMode mode, const size_t buffer)
         : std::ostream(this), _stream(stream), _mode(mode), data_buf_(new char[buffer]), data_size_(buffer)
        {
                setp(data_buf_, data_buf_ + data_size_ - 1);
        }

        CipherStream::~CipherStream()
        {
                delete[] data_buf_;
        }

        int CipherStream::sync()
        {
                int ret = std::char_traits<char>::eq_int_type(this->overflow(
                                std::char_traits<char>::eof()), std::char_traits<char>::eof()) ? -1
                                : 0;

                return ret;
        }

        int CipherStream::overflow(int c)
        {
                char *ibegin = data_buf_;
                char *iend = pptr();

                // mark the buffer as free
                setp(data_buf_, data_buf_ + data_size_ - 1);

                if (!std::char_traits<char>::eq_int_type(c, std::char_traits<char>::eof()))
                {
                        *iend++ = std::char_traits<char>::to_char_type(c);
                }

                // if there is nothing to send, just return
                if ((iend - ibegin) == 0)
                {
                        return std::char_traits<char>::not_eof(c);
                }

                // do cipher stuff
                switch (_mode)
                {
                case CIPHER_ENCRYPT:
                        encrypt(data_buf_, (iend - ibegin));
                        break;

                case CIPHER_DECRYPT:
                        decrypt(data_buf_, (iend - ibegin));
                        break;
                }

                // write result to stream
                _stream.write(data_buf_, (iend - ibegin));

                return std::char_traits<char>::not_eof(c);
        }
}