RetroShare/tests/librssimulator/testing/IsolatedServiceTester.cc

#include <list>
#include "retroshare/rsids.h"
#include "serialiser/rsserial.h"

#include "testing/IsolatedServiceTester.h"

#include "peer/PeerNode.h"

IsolatedServiceTester::IsolatedServiceTester(RsPeerId ownId, std::list<RsPeerId> peers)
{
	mNode = new PeerNode(ownId, peers, false);
	mRsSerialiser = new RsSerialiser();
}


IsolatedServiceTester::~IsolatedServiceTester()
{
	delete mNode;
	delete mRsSerialiser;
}

void    IsolatedServiceTester::addSerialType(RsSerialType *st)
{
	mRsSerialiser->addSerialType(st);
}

bool IsolatedServiceTester::startup()
{
	mNode->notifyOfFriends();
	return true;
}


bool IsolatedServiceTester::tick()
{
	mNode->tick();
	return true;
}

bool IsolatedServiceTester::tickUntilPacket(int max_ticks)
{
	for(int i = 0; i < max_ticks; i++)
	{
		tick();
		if (mNode->haveOutgoingPackets())
		{
			return true;
		}
	}
	return false;
}


RsItem *IsolatedServiceTester::getPacket()
{
	RsRawItem *rawitem = mNode->outgoing();
	if (rawitem)
	{
#ifdef DEBUG
		/* convert back to standard item for convenience */
		std::cerr << "IsolatedServiceTester::getPacket() have RsRawItem";
		std::cerr << std::endl;
#endif

		/* convert to RsServiceItem */
		uint32_t size = rawitem->getRawLength();
		RsItem *item = mRsSerialiser->deserialise(rawitem->getRawData(), &size);
		if ((!item) || (size != rawitem->getRawLength()))
		{
			/* error in conversion */
			std::cerr << "IsolatedServiceTester::getPacket() Error";
			std::cerr << std::endl;
			std::cerr << "IsolatedServiceTester::getPacket() Size: " << size;
			std::cerr << std::endl;
			std::cerr << "IsolatedServiceTester::getPacket() RawLength: " << rawitem->getRawLength();
			std::cerr << std::endl;
			if (item)
			{
				std::cerr << "p3Service::recv() Bad Item:";
				std::cerr << std::endl;
				item->print(std::cerr, 0);
				std::cerr << std::endl;

				delete item;
				item=NULL ;
			}
		}
		else
		{
			item->PeerId(rawitem->PeerId());
		}
		delete rawitem;
		return item;
	}
	return NULL;
}

bool IsolatedServiceTester::sendPacket(RsItem *si)
{
#ifdef DEBUG
	std::cerr << "IsolatedServiceTester::sendPacket()";
	std::cerr << std::endl;
#endif

	/* try to convert */
	uint32_t size = mRsSerialiser->size(si);
	if (!size)
	{
		std::cerr << "IsolatedServiceTester::sendPacket() ERROR size == 0";
		std::cerr << std::endl;

		/* can't convert! */
		delete si;
		return false;
	}

	RsRawItem *raw = new RsRawItem(si->PacketId(), size);
	if (!mRsSerialiser->serialise(si, raw->getRawData(), &size))
	{
		std::cerr << "IsolatedServiceTester::sendPacket() ERROR serialise failed";
		std::cerr << std::endl;

		delete raw;
		raw = NULL;
	}

	if ((raw) && (size != raw->getRawLength()))
	{
		std::cerr << "IsolatedServiceTester::sendPacket() ERROR serialise size mismatch";
		std::cerr << std::endl;

		delete raw;
		raw = NULL;
	}

	if (raw)
	{
#ifdef DEBUG
		std::cerr << "IsolatedServiceTester::sendPacket()";
		std::cerr << std::endl;
		si->print(std::cerr);
		std::cerr << "IsolatedServiceTester::sendPacket() passing to Node: ";
		std::cerr << std::endl;
#endif

		raw->PeerId(si->PeerId());
		mNode->incoming(raw);
	}
	delete si;
	return (raw != NULL);
}