ibrcommon/ibrcommon/net/TLSStream.h

Go to the documentation of this file.

/*
 * TLSStream.h
 *
 *  Created on: Mar 24, 2011
 *      Author: roettger
 */

#ifndef TLSSTREAM_H_
#define TLSSTREAM_H_

#include <streambuf>
#include <iostream>
#include <memory>
#include <openssl/ssl.h>
#include "ibrcommon/thread/Mutex.h"
#include "ibrcommon/data/File.h"
#include "ibrcommon/net/iostreamBIO.h"

namespace ibrcommon
{
        class TLSStream : public std::streambuf, public std::iostream
        {
        public:
                typedef std::char_traits<char> traits;

                TLSStream(std::iostream *stream);
                virtual ~TLSStream();

                void setServer(bool val);

            static void init(X509 *certificate, EVP_PKEY *privateKey, ibrcommon::File trustedCAPath, bool enableEncryption = false);

            static void flushInitialization();

            static bool isInitialized();

            void close();

                static const size_t BUFF_SIZE = 5120;

                X509 *activate();

        protected:
                virtual int sync();
                virtual int overflow(int = std::char_traits<char>::eof());
                virtual int underflow();

        private:
                std::string switchTLSError(int errnumber);

                static bool _initialized;
                /* this second initialized variable is needed, because init() can fail and SSL_library_init() is not reentrant. */
                static bool _SSL_initialized;
                static ibrcommon::Mutex _initialization_lock;

                bool _activated;
                ibrcommon::Mutex _activation_lock;

                // Input buffer
                char *in_buf_;
                // Output buffer
                char *out_buf_;

                std::iostream *_stream;
                /* indicates if this node is the server in the underlying tcp connection */
                bool _server;

                static SSL_CTX *_ssl_ctx;
                SSL *_ssl;
                X509 *_peer_cert;
                std::auto_ptr<iostreamBIO> _iostreamBIO;
        };
}

#endif /* TLSSTREAM_H_ */