runtime/zlib/contrib/iostream/zfstream.cpp

   1
   2 #include "zfstream.h"
   3
   4 gzfilebuf::gzfilebuf() :
   5   file(NULL),
   6   mode(0),
   7   own_file_descriptor(0)
   8 { }
   9
  10 gzfilebuf::~gzfilebuf() {
  11
  12   sync();
  13   if ( own_file_descriptor )
  14     close();
  15
  16 }
  17
  18 gzfilebuf *gzfilebuf::open( const char *name,
  19                             int io_mode ) {
  20
  21   if ( is_open() )
  22     return NULL;
  23
  24   char char_mode[10];
  25   char *p = char_mode;
  26
  27   if ( io_mode & ios::in ) {
  28     mode = ios::in;
  29     *p++ = 'r';
  30   } else if ( io_mode & ios::app ) {
  31     mode = ios::app;
  32     *p++ = 'a';
  33   } else {
  34     mode = ios::out;
  35     *p++ = 'w';
  36   }
  37
  38   if ( io_mode & ios::binary ) {
  39     mode |= ios::binary;
  40     *p++ = 'b';
  41   }
  42
  43   // Hard code the compression level
  44   if ( io_mode & (ios::out|ios::app )) {
  45     *p++ = '9';
  46   }
  47
  48   // Put the end-of-string indicator
  49   *p = '\0';
  50
  51   if ( (file = gzopen(name, char_mode)) == NULL )
  52     return NULL;
  53
  54   own_file_descriptor = 1;
  55
  56   return this;
  57
  58 }
  59
  60 gzfilebuf *gzfilebuf::attach( int file_descriptor,
  61                               int io_mode ) {
  62
  63   if ( is_open() )
  64     return NULL;
  65
  66   char char_mode[10];
  67   char *p = char_mode;
  68
  69   if ( io_mode & ios::in ) {
  70     mode = ios::in;
  71     *p++ = 'r';
  72   } else if ( io_mode & ios::app ) {
  73     mode = ios::app;
  74     *p++ = 'a';
  75   } else {
  76     mode = ios::out;
  77     *p++ = 'w';
  78   }
  79
  80   if ( io_mode & ios::binary ) {
  81     mode |= ios::binary;
  82     *p++ = 'b';
  83   }
  84
  85   // Hard code the compression level
  86   if ( io_mode & (ios::out|ios::app )) {
  87     *p++ = '9';
  88   }
  89
  90   // Put the end-of-string indicator
  91   *p = '\0';
  92
  93   if ( (file = gzdopen(file_descriptor, char_mode)) == NULL )
  94     return NULL;
  95
  96   own_file_descriptor = 0;
  97
  98   return this;
  99
 100 }
 101
 102 gzfilebuf *gzfilebuf::close() {
 103
 104   if ( is_open() ) {
 105
 106     sync();
 107     gzclose( file );
 108     file = NULL;
 109
 110   }
 111
 112   return this;
 113
 114 }
 115
 116 int gzfilebuf::setcompressionlevel( int comp_level ) {
 117
 118   return gzsetparams(file, comp_level, -2);
 119
 120 }
 121
 122 int gzfilebuf::setcompressionstrategy( int comp_strategy ) {
 123
 124   return gzsetparams(file, -2, comp_strategy);
 125
 126 }
 127
 128
 129 streampos gzfilebuf::seekoff( streamoff off, ios::seek_dir dir, int which ) {
 130
 131   return streampos(EOF);
 132
 133 }
 134
 135 int gzfilebuf::underflow() {
 136
 137   // If the file hasn't been opened for reading, error.
 138   if ( !is_open() || !(mode & ios::in) )
 139     return EOF;
 140
 141   // if a buffer doesn't exists, allocate one.
 142   if ( !base() ) {
 143
 144     if ( (allocate()) == EOF )
 145       return EOF;
 146     setp(0,0);
 147
 148   } else {
 149
 150     if ( in_avail() )
 151       return (unsigned char) *gptr();
 152
 153     if ( out_waiting() ) {
 154       if ( flushbuf() == EOF )
 155         return EOF;
 156     }
 157
 158   }
 159
 160   // Attempt to fill the buffer.
 161
 162   int result = fillbuf();
 163   if ( result == EOF ) {
 164     // disable get area
 165     setg(0,0,0);
 166     return EOF;
 167   }
 168
 169   return (unsigned char) *gptr();
 170
 171 }
 172
 173 int gzfilebuf::overflow( int c ) {
 174
 175   if ( !is_open() || !(mode & ios::out) )
 176     return EOF;
 177
 178   if ( !base() ) {
 179     if ( allocate() == EOF )
 180       return EOF;
 181     setg(0,0,0);
 182   } else {
 183     if (in_avail()) {
 184         return EOF;
 185     }
 186     if (out_waiting()) {
 187       if (flushbuf() == EOF)
 188         return EOF;
 189     }
 190   }
 191
 192   int bl = blen();
 193   setp( base(), base() + bl);
 194
 195   if ( c != EOF ) {
 196
 197     *pptr() = c;
 198     pbump(1);
 199
 200   }
 201
 202   return 0;
 203
 204 }
 205
 206 int gzfilebuf::sync() {
 207
 208   if ( !is_open() )
 209     return EOF;
 210
 211   if ( out_waiting() )
 212     return flushbuf();
 213
 214   return 0;
 215
 216 }
 217
 218 int gzfilebuf::flushbuf() {
 219
 220   int n;
 221   char *q;
 222
 223   q = pbase();
 224   n = pptr() - q;
 225
 226   if ( gzwrite( file, q, n) < n )
 227     return EOF;
 228
 229   setp(0,0);
 230
 231   return 0;
 232
 233 }
 234
 235 int gzfilebuf::fillbuf() {
 236
 237   int required;
 238   char *p;
 239
 240   p = base();
 241
 242   required = blen();
 243
 244   int t = gzread( file, p, required );
 245
 246   if ( t <= 0) return EOF;
 247
 248   setg( base(), base(), base()+t);
 249
 250   return t;
 251
 252 }
 253
 254 gzfilestream_common::gzfilestream_common() :
 255   ios( gzfilestream_common::rdbuf() )
 256 { }
 257
 258 gzfilestream_common::~gzfilestream_common()
 259 { }
 260
 261 void gzfilestream_common::attach( int fd, int io_mode ) {
 262
 263   if ( !buffer.attach( fd, io_mode) )
 264     clear( ios::failbit | ios::badbit );
 265   else
 266     clear();
 267
 268 }
 269
 270 void gzfilestream_common::open( const char *name, int io_mode ) {
 271
 272   if ( !buffer.open( name, io_mode ) )
 273     clear( ios::failbit | ios::badbit );
 274   else
 275     clear();
 276
 277 }
 278
 279 void gzfilestream_common::close() {
 280
 281   if ( !buffer.close() )
 282     clear( ios::failbit | ios::badbit );
 283
 284 }
 285
 286 gzfilebuf *gzfilestream_common::rdbuf()
 287 {
 288   return &buffer;
 289 }
 290
 291 gzifstream::gzifstream() :
 292   ios( gzfilestream_common::rdbuf() )
 293 {
 294   clear( ios::badbit );
 295 }
 296
 297 gzifstream::gzifstream( const char *name, int io_mode ) :
 298   ios( gzfilestream_common::rdbuf() )
 299 {
 300   gzfilestream_common::open( name, io_mode );
 301 }
 302
 303 gzifstream::gzifstream( int fd, int io_mode ) :
 304   ios( gzfilestream_common::rdbuf() )
 305 {
 306   gzfilestream_common::attach( fd, io_mode );
 307 }
 308
 309 gzifstream::~gzifstream() { }
 310
 311 gzofstream::gzofstream() :
 312   ios( gzfilestream_common::rdbuf() )
 313 {
 314   clear( ios::badbit );
 315 }
 316
 317 gzofstream::gzofstream( const char *name, int io_mode ) :
 318   ios( gzfilestream_common::rdbuf() )
 319 {
 320   gzfilestream_common::open( name, io_mode );
 321 }
 322
 323 gzofstream::gzofstream( int fd, int io_mode ) :
 324   ios( gzfilestream_common::rdbuf() )
 325 {
 326   gzfilestream_common::attach( fd, io_mode );
 327 }
 328
 329 gzofstream::~gzofstream() { }