src/socketoutputstream.cpp

/***************************************************************************
           socketoutputstream.cpp  -  class SocketOutputStream
                             -------------------
    begin                : ven mai 9 2003
    copyright            : (C) 2003 by Michael CATANZARITI
    email                : mcatan@free.fr
 ***************************************************************************/

/***************************************************************************
 * Copyright (C) The Apache Software Foundation. All rights reserved.      *
 *                                                                         *
 * This software is published under the terms of the Apache Software       *
 * License version 1.1, a copy of which has been included with this        *
 * distribution in the LICENSE.txt file.                                   *
 ***************************************************************************/

#include <log4cxx/helpers/socketoutputstream.h>
#include <log4cxx/helpers/socket.h>
#include <log4cxx/helpers/loglog.h>

using namespace log4cxx;
using namespace log4cxx::helpers;

#define INCREMENT 512

SocketOutputStream::SocketOutputStream(SocketPtr socket)
: socket(socket), beg(0), cur(0), end(0)
{
}

SocketOutputStream::~SocketOutputStream()
{
	delete [] beg;
}

void SocketOutputStream::write(const void * buffer, int len)
{
//	LOGLOG_DEBUG (_T("SocketOutputStream writing ") << len << _T(" bytes."));
	if (cur + len > end)
	{
		if (beg == 0)
		{
			size_t size = ((len > INCREMENT) ? len : INCREMENT);
//			LOGLOG_DEBUG (_T("SocketOutputStream growing ") << size << _T(" bytes."));
			beg = new unsigned char[size];
			end = beg + size;
			cur = beg;
		}
		else
		{
			size_t size = end - beg + ((len > INCREMENT) ? len : INCREMENT);
			unsigned char * old = beg;
//			LOGLOG_DEBUG (_T("SocketOutputStream growing ") <<
//				((len > INCREMENT) ? len : INCREMENT) << _T(" bytes."));
			beg = new unsigned char[size];
			memcpy(beg, old, cur - old);
			cur = beg + (cur - old);
			end = beg + size;
			delete [] old;
		}		
	}

	memcpy(cur, buffer, len);
	cur+= len;
}

void SocketOutputStream::write(unsigned int value)
{
	write(&value, sizeof(value));
}

void SocketOutputStream::write(int value)
{
	write(&value, sizeof(value));
}

void SocketOutputStream::write(unsigned long value)
{
	write(&value, sizeof(value));
}

void SocketOutputStream::write(long value)
{
	write(&value, sizeof(value));
}

void SocketOutputStream::write(const tstring& value)
{
	tstring::size_type size;

	size = value.size();
	write(&size, sizeof(tstring::size_type));
	if (size > 0)
	{
		if (size > 1024)
		{
			size = 1024;
		}
		
		write(value.c_str(), size * sizeof(TCHAR));
	}
}

void SocketOutputStream::close()
{
	// seek to begin
	cur = beg;

	// dereference socket
	socket = 0;
}

void SocketOutputStream::flush()
{
	// write to socket
	socket->write(beg, cur - beg);

	// seek to begin
	cur = beg;
}