ACE/protocols/ace/INet/SSLSock_IOStream.h

   1 /**
   2  * @file SSLSock_IOStream.h
   3  *
   4  * @author Martin Corino <mcorino@remedy.nl>
   5  */
   6
   7 #ifndef ACE_IOS_SSLSOCK_IOSTREAM_H
   8 #define ACE_IOS_SSLSOCK_IOSTREAM_H
   9
  10 #include /**/ "ace/pre.h"
  11
  12 #include /**/ "ace/config-all.h"
  13
  14 #if !defined (ACE_LACKS_PRAGMA_ONCE)
  15 # pragma once
  16 #endif /* ACE_LACKS_PRAGMA_ONCE */
  17
  18 #include "ace/SSL/SSL_SOCK_Stream.h"
  19 #include "ace/INet/BidirStreamBuffer.h"
  20 #include "ace/INet/StreamHandler.h"
  21 #include "ace/INet/StreamInterceptor.h"
  22 #include <istream>
  23 #include <ostream>
  24
  25 ACE_BEGIN_VERSIONED_NAMESPACE_DECL
  26
  27 namespace ACE
  28   {
  29     namespace IOS
  30       {
  31         /**
  32         * @class ACE_IOS_SSLSock_StreamBuffer
  33         *
  34         * @brief
  35         *
  36         */
  37         template <ACE_SYNCH_DECL>
  38         class SSLSock_StreamBufferBase
  39           : public BidirStreamBuffer<StreamHandler<ACE_SSL_SOCK_Stream, ACE_SYNCH_USE> >
  40           {
  41             public:
  42               typedef StreamHandler<ACE_SSL_SOCK_Stream, ACE_SYNCH_USE> stream_type;
  43
  44               SSLSock_StreamBufferBase (stream_type* stream);
  45               virtual ~SSLSock_StreamBufferBase ();
  46
  47             private:
  48               enum
  49               {
  50                 BUFFER_SIZE = 1024
  51               };
  52           };
  53
  54         /**
  55         * @class ACE_IOS_SSLSock_IOS
  56         *
  57         * @brief
  58         *
  59         */
  60         template <ACE_SYNCH_DECL>
  61         class SSLSock_IOSBase
  62           : public virtual std::ios
  63           {
  64             public:
  65               typedef SSLSock_StreamBufferBase<ACE_SYNCH_USE> buffer_type;
  66               typedef typename buffer_type::stream_type stream_type;
  67
  68               SSLSock_IOSBase (stream_type* stream);
  69               ~SSLSock_IOSBase ();
  70
  71               buffer_type* rdbuf ();
  72
  73               void close ();
  74
  75               const stream_type& stream () const;
  76
  77             protected:
  78               buffer_type streambuf_;
  79           };
  80
  81         /**
  82         * @class ACE_IOS_SSLSock_OStream
  83         *
  84         * @brief
  85         *
  86         */
  87         template <ACE_SYNCH_DECL>
  88         class SSLSock_OStreamBase
  89           : public SSLSock_IOSBase<ACE_SYNCH_USE>, public std::ostream
  90           {
  91             public:
  92               typedef SSLSock_IOSBase<ACE_SYNCH_USE> ios_base;
  93               typedef typename ios_base::stream_type stream_type;
  94               typedef SSLSock_StreamBufferBase<ACE_SYNCH_USE> buffer_type;
  95
  96               explicit SSLSock_OStreamBase(stream_type* stream);
  97
  98               buffer_type* rdbuf ()
  99               { return ios_base::rdbuf (); }
 100
 101               ~SSLSock_OStreamBase();
 102
 103               void set_interceptor (typename buffer_type::interceptor_type& interceptor);
 104           };
 105
 106         /**
 107         * @class ACE_IOS_SSLSock_IStream
 108         *
 109         * @brief
 110         *
 111         */
 112         template <ACE_SYNCH_DECL>
 113         class SSLSock_IStreamBase
 114           : public SSLSock_IOSBase<ACE_SYNCH_USE>, public std::istream
 115           {
 116             public:
 117               typedef SSLSock_IOSBase<ACE_SYNCH_USE> ios_base;
 118               typedef typename ios_base::stream_type stream_type;
 119               typedef SSLSock_StreamBufferBase<ACE_SYNCH_USE> buffer_type;
 120
 121               explicit SSLSock_IStreamBase(stream_type* stream);
 122
 123               buffer_type* rdbuf ()
 124               { return ios_base::rdbuf (); }
 125
 126               ~SSLSock_IStreamBase();
 127
 128               void set_interceptor (typename buffer_type::interceptor_type& interceptor);
 129           };
 130
 131         /**
 132         * @class ACE_IOS_SSLSock_IOStream
 133         *
 134         * @brief
 135         *
 136         */
 137         template <ACE_SYNCH_DECL>
 138         class SSLSock_IOStreamBase
 139           : public SSLSock_IOSBase<ACE_SYNCH_USE>, public std::iostream
 140           {
 141             public:
 142               typedef SSLSock_IOSBase<ACE_SYNCH_USE> ios_base;
 143               typedef typename ios_base::stream_type stream_type;
 144               typedef SSLSock_StreamBufferBase<ACE_SYNCH_USE> buffer_type;
 145
 146               explicit SSLSock_IOStreamBase(stream_type* stream);
 147
 148               buffer_type* rdbuf ()
 149               { return ios_base::rdbuf (); }
 150
 151               ~SSLSock_IOStreamBase();
 152
 153               void set_interceptor (typename buffer_type::interceptor_type& interceptor);
 154           };
 155
 156         typedef SSLSock_StreamBufferBase<ACE_NULL_SYNCH> SSLSock_StreamBuffer;
 157         typedef SSLSock_IOSBase<ACE_NULL_SYNCH> SSLSock_IOS;
 158         typedef SSLSock_IStreamBase<ACE_NULL_SYNCH> SSLSock_IStream;
 159         typedef SSLSock_OStreamBase<ACE_NULL_SYNCH> SSLSock_OStream;
 160         typedef SSLSock_IOStreamBase<ACE_NULL_SYNCH> SSLSock_IOStream;
 161       }
 162   }
 163
 164 ACE_END_VERSIONED_NAMESPACE_DECL
 165
 166 #include "ace/INet/SSLSock_IOStream.cpp"
 167
 168 #include /**/ "ace/post.h"
 169 #endif /* ACE_IOS_SSLSOCK_IOSTREAM_H */