File indexing completed on 2024-06-02 04:07:14

 #include "xmpp/zlib/zlibdecompressor.h"
 
 #include <QtDebug>
 #include <QIODevice>
 #include <zlib.h>
 
 #include "xmpp/zlib/common.h"
 
 ZLibDecompressor::ZLibDecompressor(QIODevice* device) : device_(device)
 {
     zlib_stream_ = (z_stream*) malloc(sizeof(z_stream));
     initZStream(zlib_stream_);
     int result = inflateInit(zlib_stream_);
     Q_ASSERT(result == Z_OK);
     Q_UNUSED(result);
     connect(device, &QIODevice::aboutToClose, this, &ZLibDecompressor::flush);
     flushed_ = false;
 }
 
 ZLibDecompressor::~ZLibDecompressor()
 {
     flush();
     free(zlib_stream_);
 }
 
 void ZLibDecompressor::flush()
 {
     if (flushed_)
         return;
     
     // Flush
     write(QByteArray(),true);
     int result = inflateEnd(zlib_stream_);
     if (result != Z_OK) 
         qWarning("compressor.c: inflateEnd failed (%d)", result);
     
     flushed_ = true;
 }
 
 int ZLibDecompressor::write(const QByteArray& input)
 {
     return write(input,false);
 }
 
 int ZLibDecompressor::write(const QByteArray& input, bool flush)
 {
     int result;
     zlib_stream_->avail_in = input.size();
     zlib_stream_->next_in = (Bytef*) input.data();
     QByteArray output;
 
     // Write the data
     int output_position = 0;
     do {
         output.resize(output_position + CHUNK_SIZE);
         zlib_stream_->avail_out = CHUNK_SIZE;
         zlib_stream_->next_out = (Bytef*) (output.data() + output_position);
         result = inflate(zlib_stream_,(flush ? Z_FINISH : Z_NO_FLUSH));
         if (result == Z_STREAM_ERROR) {
             qWarning("compressor.cpp: Error ('%s')", zlib_stream_->msg);
             return result;
         }
         output_position += CHUNK_SIZE;
     }
     while (zlib_stream_->avail_out == 0);
     //Q_ASSERT(zlib_stream_->avail_in == 0);
     if (zlib_stream_->avail_in != 0) {
         qWarning() << "ZLibDecompressor: Unexpected state: avail_in=" << zlib_stream_->avail_in << ",avail_out=" << zlib_stream_->avail_out << ",result=" << result;
         return Z_STREAM_ERROR; // FIXME: Should probably return 'result'
     }
     output_position -= zlib_stream_->avail_out;
 
     // Flush the data
     if (!flush) {
         do {
             output.resize(output_position + CHUNK_SIZE);
             zlib_stream_->avail_out = CHUNK_SIZE;
             zlib_stream_->next_out = (Bytef*) (output.data() + output_position);
             result = inflate(zlib_stream_,Z_SYNC_FLUSH);
             if (result == Z_STREAM_ERROR) {
                 qWarning("compressor.cpp: Error ('%s')", zlib_stream_->msg);
                 return result;
             }
             output_position += CHUNK_SIZE;
         }
         while (zlib_stream_->avail_out == 0);
         output_position -= zlib_stream_->avail_out;
     }
     output.resize(output_position);
 
     // Write the compressed data
     device_->write(output);
     return 0;
 }