00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00036 #include "OW_config.h"
00037 #ifdef OW_HAVE_ZLIB_H
00038 #include "OW_HTTPDeflateIStream.hpp"
00039 #include "OW_HTTPException.hpp"
00040
00041 namespace OW_NAMESPACE
00042 {
00043
00044 using std::istream;
00045 HTTPDeflateIStreamBuffer::HTTPDeflateIStreamBuffer(istream& istr)
00046 : BaseStreamBuffer(HTTP_BUF_SIZE, "in")
00047 , m_istr(istr)
00048 {
00049 m_zstr.opaque = Z_NULL;
00050 m_zstr.zfree = Z_NULL;
00051 m_zstr.zalloc = Z_NULL;
00052 m_zstr.avail_in = 0;
00053 int rval = inflateInit(&m_zstr);
00054 if (rval != Z_OK)
00055 {
00056 String msg = "Error: inflateInit returned " + String(rval);
00057 if (m_zstr.msg)
00058 {
00059 msg += String(": ") + String(m_zstr.msg);
00060 }
00061 OW_THROW(HTTPException, msg.c_str());
00062 }
00063 }
00065 HTTPDeflateIStreamBuffer::~HTTPDeflateIStreamBuffer()
00066 {
00067 inflateEnd(&m_zstr);
00068 }
00070 int
00071 HTTPDeflateIStreamBuffer::buffer_from_device(char* c, int n)
00072 {
00073 if (n < 1)
00074 {
00075 return 0;
00076 }
00077 m_zstr.avail_out = n;
00078 m_zstr.next_out = reinterpret_cast<Bytef*>(c);
00079 int bytesRead = 0;
00080 while (m_zstr.avail_out > 0)
00081 {
00082 if (m_zstr.avail_in == 0)
00083 {
00084 if (m_istr)
00085 {
00086 m_istr.read(reinterpret_cast<char*>(m_inBuf), m_inBufSize);
00087 m_zstr.avail_in = m_istr.gcount();
00088 m_zstr.next_in = m_inBuf;
00089 }
00090 }
00091 int rv = inflate(&m_zstr, Z_SYNC_FLUSH);
00092 bytesRead = n - m_zstr.avail_out;
00093 if (rv != Z_OK)
00094 {
00095 break;
00096 }
00097 }
00098 if (bytesRead > 0)
00099 {
00100 return bytesRead;
00101 }
00102 else
00103 {
00104 return -1;
00105 }
00106
00107
00108
00109
00110
00111
00112
00113
00114
00115
00116
00117
00118
00119
00120
00121
00122
00123
00124
00125 }
00127 HTTPDeflateIStream::HTTPDeflateIStream(
00128 const CIMProtocolIStreamIFCRef& istr)
00129 : HTTPDeflateIStreamBase(*istr)
00130 , CIMProtocolIStreamIFC(&m_strbuf)
00131 , m_istr(istr)
00132 {
00133 }
00134
00136
00137 void HTTPDeflateIStream::checkForError() const
00138 {
00139 m_istr->checkForError();
00140 }
00141
00143 CIMProtocolIStreamIFCRef
00144 HTTPDeflateIStream::getInputStreamOrig()
00145 {
00146 return m_istr;
00147 }
00148
00149 }
00150
00151 #endif // #ifdef OW_HAVE_ZLIB_H
00152