khanat-opennel-code/code/nel/src/net/module_socket.cpp

// NeL - MMORPG Framework <http://dev.ryzom.com/projects/nel/>
// Copyright (C) 2010  Winch Gate Property Limited
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.


#include "stdnet.h"
#include "nel/net/module_socket.h"
#include "nel/net/module.h"
#include "nel/net/module_manager.h"
#include "nel/net/module_gateway.h"
#include "nel/net/module_common.h"

using namespace std;
using namespace NLMISC;

namespace NLNET
{

	CModuleSocket::CModuleSocket()
		: _SocketRegistered(false)
	{
	}

	CModuleSocket::~CModuleSocket()
	{
		unregisterSocket();
	}


	void CModuleSocket::registerSocket()
	{
		if (!_SocketRegistered)
		{
			_SocketRegistered = true;
			IModuleManager::getInstance().registerModuleSocket(this);
		}
	}

	void CModuleSocket::unregisterSocket()
	{
		if (_SocketRegistered)
		{
			IModuleManager::getInstance().unregisterModuleSocket(this);
			_SocketRegistered = false;
		}
	}

	void CModuleSocket::_onModulePlugged(const TModulePtr &pluggedModule)
	{
		TPluggedModules::TBToAMap::const_iterator it(_PluggedModules.getBToAMap().find(pluggedModule));
		if (it != _PluggedModules.getBToAMap().end())
		{
			throw IModule::EModuleAlreadyPluggedHere();
		}

		_PluggedModules.add(pluggedModule->getModuleId(), pluggedModule);

		// callback socket implementation
		onModulePlugged(pluggedModule);
	}

	void CModuleSocket::_onModuleUnplugged(const TModulePtr &pluggedModule)
	{
		TPluggedModules::TBToAMap::const_iterator it(_PluggedModules.getBToAMap().find(pluggedModule));
		if (it == _PluggedModules.getBToAMap().end())
		{
			throw EModuleNotPluggedHere();
		}

		// callback socket implementation
		onModuleUnplugged(pluggedModule);

		_PluggedModules.removeWithB(pluggedModule);
	}

	void CModuleSocket::sendModuleMessage(IModule *senderModule, TModuleId destModuleProxyId, const NLNET::CMessage &message )
			throw (EModuleNotPluggedHere)
	{
		TPluggedModules::TBToAMap::const_iterator it(_PluggedModules.getBToAMap().find(senderModule));
		if (it == _PluggedModules.getBToAMap().end())
		{
			throw EModuleNotPluggedHere();
		}

		// forward to socket implementation
		_sendModuleMessage(senderModule, destModuleProxyId, message);

	}

	void CModuleSocket::broadcastModuleMessage(IModule *senderModule, const NLNET::CMessage &message)
			throw (EModuleNotPluggedHere)
	{
		TPluggedModules::TBToAMap::const_iterator it(_PluggedModules.getBToAMap().find(senderModule));
		if (it == _PluggedModules.getBToAMap().end())
		{
			throw EModuleNotPluggedHere();
		}

		// forward to socket implementation
		_broadcastModuleMessage(senderModule, message);
	}

} // namespace NLNET