tahoma2d/toonz/sources/common/tipc/tipcsrv.cpp

#include "assert.h"

//Qt includes
#include <QDataStream>
#include <QDebug>

#include "tipc.h"
#include "tipcmsg.h"

#include "tipcsrvP.h"
#include "tipcsrv.h"

//*******************************************************************************
//    Diagnostics stuff
//*******************************************************************************

//#define TIPC_DEBUG

#ifdef TIPC_DEBUG
#define tipc_debug(expr) expr
#else
#define tipc_debug(expr)
#endif

#ifdef TIPC_DEBUG
#include <QTime>
#endif

//*******************************************************************************
//    tipc::SocketListener implementation
//*******************************************************************************

void tipc::SocketController::onReadyRead()
{
	//Deliver the message to the server for interpretation.
	m_server->dispatchSocket(m_socket);
}

//-----------------------------------------------------------------------

void tipc::SocketController::onDisconnected()
{
	m_socket->QObject::disconnect(SIGNAL(readyRead()));

	//Auto-delete this
	delete this;
}

//*******************************************************************************
//    Server implementation
//*******************************************************************************

tipc::Server::Server()
	: m_lock(false)
{
	connect(this, SIGNAL(newConnection()), this, SLOT(onNewConnection()));

	//Add default parsers
	addParser(new DefaultMessageParser<SHMEM_REQUEST>);
	addParser(new DefaultMessageParser<SHMEM_RELEASE>);
	addParser(new DefaultMessageParser<TMPFILE_REQUEST>);
	addParser(new DefaultMessageParser<TMPFILE_RELEASE>);
	addParser(new DefaultMessageParser<QUIT_ON_ERROR>);
}

//-----------------------------------------------------------------------

tipc::Server::~Server()
{
	//Release parsers
	QHash<QString, MessageParser *>::iterator it;
	for (it = m_parsers.begin(); it != m_parsers.end(); ++it)
		delete it.value();
}

//-----------------------------------------------------------------------

void tipc::Server::addParser(MessageParser *parser)
{
	m_parsers.insert(parser->header(), parser);
}

//-----------------------------------------------------------------------

void tipc::Server::removeParser(QString header)
{
	MessageParser *parser = m_parsers.take(header);
	if (parser)
		delete parser;
}

//-----------------------------------------------------------------------

void tipc::Server::onNewConnection()
{
	tipc_debug(qDebug("new connection"));

	//Accept the connection
	QLocalSocket *socket = nextPendingConnection();

	//Allocate a controller for the socket
	SocketController *controller = new SocketController;
	controller->m_server = this;
	controller->m_socket = socket;

	//Connect the controller to the socket's signals
	connect(socket, SIGNAL(readyRead()), controller, SLOT(onReadyRead()));
	connect(socket, SIGNAL(disconnected()), controller, SLOT(onDisconnected()));
	connect(socket, SIGNAL(disconnected()), socket, SLOT(deleteLater()));
	connect(socket, SIGNAL(error(QLocalSocket::LocalSocketError)),
			this, SLOT(onError(QLocalSocket::LocalSocketError)));
}

//-----------------------------------------------------------------------

void tipc::Server::onError(QLocalSocket::LocalSocketError error)
{
	tipc_debug(qDebug() << "Server error #" << error << ": " << errorString());
}

//-----------------------------------------------------------------------

void tipc::Server::dispatchSocket(QLocalSocket *socket)
{
	//The lock is established when a message is currently being processed.
	//Returning if the lock is set avoids having recursive message processing;
	//which is possible if a parser expects further message packets.
	if (m_lock)
		return;

	tipc::Stream stream(socket);
	QString header;

	while (socket->bytesAvailable() > 0) {
		if (!stream.messageReady())
			return;

		Message msg;

		stream >> msg;
		msg >> header;
		assert(!header.isEmpty());

		tipc_debug(qDebug() << header << endl);

		QHash<QString, MessageParser *>::iterator it = m_parsers.find(header);
		if (it == m_parsers.end()) {
			tipc_debug(qDebug() << "Error: Unrecognized command" << endl);
			continue;
		}

		m_lock = true;

		MessageParser *parser = it.value();
		parser->m_socket = socket;
		parser->m_stream = &stream;
		parser->operator()(msg);

		m_lock = false;

		//The Message has been read and processed. Send the reply.
		if (msg.ba().size() > 0)
			stream << msg;
	}
}