ACE/protocols/ace/INet/HTTP_IOStream.cpp

   1 #include "ace/INet/HTTP_IOStream.h"
   2 #include "ace/INet/IOS_util.h"
   3
   4 #if !defined (__ACE_INLINE__)
   5 #include "ace/INet/HTTP_IOStream.inl"
   6 #endif
   7
   8 #include "ace/Truncate.h"
   9
  10
  11 ACE_BEGIN_VERSIONED_NAMESPACE_DECL
  12
  13 namespace ACE
  14 {
  15   namespace HTTP
  16   {
  17     StreamBuffer::StreamBuffer (std::iostream& stream, StreamBuffer::policy_type* policy)
  18       : ACE::IOS::BufferedStreamBuffer (BUFFER_SIZE,
  19                                         std::ios::in | std::ios::out),
  20         stream_ (stream),
  21         policy_ (policy)
  22       {
  23         if (this->policy_)
  24           this->policy_->set_stream_buffer (this);
  25       }
  26
  27     StreamBuffer::~StreamBuffer ()
  28       {
  29         if (this->policy_)
  30           delete this->policy_;
  31       }
  32
  33     int StreamBuffer::read_from_stream (char* buffer, std::streamsize length)
  34       {
  35         if (this->policy_)
  36           return this->policy_->read_from_stream (buffer, length);
  37         else
  38           return this->read_from_stream_i (buffer, length);
  39       }
  40
  41     int StreamBuffer::read_from_stream_i (char* buffer, std::streamsize length)
  42       {
  43         this->stream_.read (buffer, length);
  44         return ACE_Utils::truncate_cast<int> (this->stream_.gcount ());
  45       }
  46
  47     int StreamBuffer::write_to_stream (const char* buffer, std::streamsize length)
  48       {
  49         if (this->policy_)
  50           return this->policy_->write_to_stream (buffer, length);
  51         else
  52           return this->write_to_stream_i (buffer, length);
  53       }
  54
  55     int StreamBuffer::write_to_stream_i (const char* buffer, std::streamsize length)
  56       {
  57         this->stream_.write (buffer, length);
  58         return this->stream_.good () ? ACE_Utils::truncate_cast<int> (length) : -1;
  59       }
  60
  61     int StreamBuffer::sync ()
  62       {
  63         if (ACE::IOS::BufferedStreamBuffer::sync () == -1)
  64           return -1;
  65         return this->stream_.sync ();
  66       }
  67
  68     IOS::IOS (std::iostream& stream, StreamBuffer::policy_type* policy)
  69       : streambuf_ (stream, policy)
  70       {
  71         ace_ios_init (&this->streambuf_);
  72       }
  73
  74     IOS::~IOS ()
  75       {
  76         try
  77           {
  78             this->streambuf_.sync();
  79           }
  80         catch (...)
  81           {
  82           }
  83       }
  84
  85     OStream::OStream(std::iostream& stream, StreamBuffer::policy_type* policy)
  86       : IOS (stream, policy), std::ostream (&streambuf_)
  87       {
  88       }
  89
  90     OStream::~OStream()
  91       {
  92       }
  93
  94     IStream::IStream(std::iostream& stream, StreamBuffer::policy_type* policy)
  95       : IOS (stream, policy), std::istream (&streambuf_)
  96       {
  97       }
  98
  99     IStream::~IStream()
 100       {
 101       }
 102
 103   }
 104 }
 105
 106 ACE_END_VERSIONED_NAMESPACE_DECL