yate/engine/Compressor.cpp

106 lines
3.2 KiB
C++

/**
* Compressor.cpp
* This file is part of the YATE Project http://YATE.null.ro
*
* Yet Another Telephony Engine - a fully featured software PBX and IVR
* Copyright (C) 2004-2010 Null Team
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.
*/
#include "yatengine.h"
using namespace TelEngine;
// Compress the input buffer, flush all data,
// append compressed data to the received data block
int Compressor::compress(const void* buf, unsigned int len, DataBlock& dest)
{
XDebug(DebugAll,"Compressor(%s)::compress(%p,%u) dest len %u [%p]",
toString().c_str(),buf,len,dest.length(),this);
if (!(buf && len)) {
buf = 0;
len = 0;
}
const unsigned char* wrBuf = (const unsigned char*)buf;
int ret = 0;
while (true) {
int wr = 0;
if (len) {
wr = writeComp(wrBuf + ret,len,false);
XDebug(DebugAll,"Compressor(%s)::compress() wrote %d [%p]",
toString().c_str(),wr,this);
if (wr > 0) {
ret += wr;
len -= wr;
}
}
int rd = readComp(dest,true);
XDebug(DebugAll,"Compressor(%s)::compress() read %d [%p]",
toString().c_str(),rd,this);
// Don't stop if write succeeded and was partial
if (rd >= 0 && wr >= 0 && len)
continue;
// Set return value to error or 0 if nothing wrote so far
if (!ret)
ret = wr;
break;
}
XDebug(DebugAll,"Compressor(%s)::compress(%p,%u) returning %d dest len %u [%p]",
toString().c_str(),buf,len,ret,dest.length(),this);
return ret;
}
// Decompress the input buffer, flush all data,
// append decompressed data to the received data block
int Compressor::decompress(const void* buf, unsigned int len, DataBlock& dest)
{
XDebug(DebugAll,"Compressor(%s)::decompress(%p,%u) dest len %u [%p]",
toString().c_str(),buf,len,dest.length(),this);
if (!(buf && len)) {
buf = 0;
len = 0;
}
const unsigned char* wrBuf = (const unsigned char*)buf;
int ret = 0;
while (true) {
int wr = 0;
if (len) {
wr = writeDecomp(wrBuf + ret,len,false);
XDebug(DebugAll,"Compressor(%s)::decompress() wrote %d [%p]",
toString().c_str(),wr,this);
if (wr > 0) {
ret += wr;
len -= wr;
}
}
int rd = readDecomp(dest,true);
XDebug(DebugAll,"Compressor(%s)::decompress() read %d [%p]",
toString().c_str(),rd,this);
// Don't stop if write succeeded and was partial
if (rd >= 0 && wr >= 0 && len)
continue;
// Set return value to error or 0 if nothing wrote so far
if (!ret)
ret = wr;
break;
}
XDebug(DebugAll,"Compressor(%s)::decompress(%p,%u) returning %d dest len %u [%p]",
toString().c_str(),buf,len,ret,dest.length(),this);
return ret;
}
/* vi: set ts=8 sw=4 sts=4 noet: */