nel/src/net/module_socket.cpp

   1 // NeL - MMORPG Framework <http://dev.ryzom.com/projects/nel/>
   2 // Copyright (C) 2010  Winch Gate Property Limited
   3 //
   4 // This program is free software: you can redistribute it and/or modify
   5 // it under the terms of the GNU Affero General Public License as
   6 // published by the Free Software Foundation, either version 3 of the
   7 // License, or (at your option) any later version.
   8 //
   9 // This program is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU Affero General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU Affero General Public License
  15 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17
  18 #include "stdnet.h"
  19 #include "nel/net/module_socket.h"
  20 #include "nel/net/module.h"
  21 #include "nel/net/module_manager.h"
  22 #include "nel/net/module_gateway.h"
  23 #include "nel/net/module_common.h"
  24
  25 using namespace std;
  26 using namespace NLMISC;
  27
  28 namespace NLNET
  29 {
  30
  31         CModuleSocket::CModuleSocket()
  32                 : _SocketRegistered(false)
  33         {
  34         }
  35
  36         CModuleSocket::~CModuleSocket()
  37         {
  38                 unregisterSocket();
  39         }
  40
  41
  42         void CModuleSocket::registerSocket()
  43         {
  44                 if (!_SocketRegistered)
  45                 {
  46                         _SocketRegistered = true;
  47                         IModuleManager::getInstance().registerModuleSocket(this);
  48                 }
  49         }
  50
  51         void CModuleSocket::unregisterSocket()
  52         {
  53                 if (_SocketRegistered)
  54                 {
  55                         IModuleManager::getInstance().unregisterModuleSocket(this);
  56                         _SocketRegistered = false;
  57                 }
  58         }
  59
  60         void CModuleSocket::_onModulePlugged(const TModulePtr &pluggedModule)
  61         {
  62                 TPluggedModules::TBToAMap::const_iterator it(_PluggedModules.getBToAMap().find(pluggedModule));
  63                 if (it != _PluggedModules.getBToAMap().end())
  64                 {
  65                         throw IModule::EModuleAlreadyPluggedHere();
  66                 }
  67
  68                 _PluggedModules.add(pluggedModule->getModuleId(), pluggedModule);
  69
  70                 // callback socket implementation
  71                 onModulePlugged(pluggedModule);
  72         }
  73
  74         void CModuleSocket::_onModuleUnplugged(const TModulePtr &pluggedModule)
  75         {
  76                 TPluggedModules::TBToAMap::const_iterator it(_PluggedModules.getBToAMap().find(pluggedModule));
  77                 if (it == _PluggedModules.getBToAMap().end())
  78                 {
  79                         throw EModuleNotPluggedHere();
  80                 }
  81
  82                 // callback socket implementation
  83                 onModuleUnplugged(pluggedModule);
  84
  85                 _PluggedModules.removeWithB(pluggedModule);
  86         }
  87
  88         void CModuleSocket::sendModuleMessage(IModule *senderModule, TModuleId destModuleProxyId, const NLNET::CMessage &message)
  89         {
  90                 TPluggedModules::TBToAMap::const_iterator it(_PluggedModules.getBToAMap().find(senderModule));
  91                 if (it == _PluggedModules.getBToAMap().end())
  92                 {
  93                         throw EModuleNotPluggedHere();
  94                 }
  95
  96                 // forward to socket implementation
  97                 _sendModuleMessage(senderModule, destModuleProxyId, message);
  98
  99         }
 100
 101         void CModuleSocket::broadcastModuleMessage(IModule *senderModule, const NLNET::CMessage &message)
 102         {
 103                 TPluggedModules::TBToAMap::const_iterator it(_PluggedModules.getBToAMap().find(senderModule));
 104                 if (it == _PluggedModules.getBToAMap().end())
 105                 {
 106                         throw EModuleNotPluggedHere();
 107                 }
 108
 109                 // forward to socket implementation
 110                 _broadcastModuleMessage(senderModule, message);
 111         }
 112
 113 } // namespace NLNET