// NeL - MMORPG Framework // 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 . #include "stdnet.h" #include "nel/misc/entity_id.h" // for createMessage() #include "nel/misc/variable.h" #include "nel/misc/thread.h" #include "nel/misc/mutex.h" #include "nel/net/unified_network.h" #include "nel/net/module_common.h" #include "nel/net/naming_client.h" #ifdef NL_OS_UNIX #include #endif using namespace std; using namespace NLMISC; namespace NLNET { static size_t ThreadCreator = 0; static const uint64 AppIdDeadConnection = 0xDEAD; uint32 TotalCallbackCalled = 0; uint32 TimeInCallback =0; #ifdef NL_OS_UNIX /// Yield method (Unix only) CVariable UseYieldMethod("nel", "UseYieldMethod", "0=select 1=usleep 2=nanosleep 3=sched_yield 4=none", 0, 0, true ); #endif /// Reduce sending lag CVariable FlushSendsBeforeSleep("nel", "FlushSendsBeforeSleep", "If true, send buffers will be flushed before sleep, not in next update", true, 0, true ); /// Network congestion monitoring CVariable L5TotalBytesInLowLevelSendQueues("nel", "L5TotalBytesInLowLevelSendQueues", "Number of bytes pending in send queues (postponed by non-blocking send()) for network congestion monitoring. N/A if FlushSendsBeforeSleep disabled)", 0, 0, true ); /// Receiving size limit CVariablePtr DefaultMaxExpectedBlockSize("nel", "DefaultMaxExpectedBlockSize", "If receiving more than this value in bytes, the connection will be dropped", &CBufNetBase::DefaultMaxExpectedBlockSize, true ); /// Sending size limit CVariablePtr DefaultMaxSentBlockSize("nel", "DefaultMaxSentBlockSize", "If sending more than this value in bytes, the program may be stopped", &CBufNetBase::DefaultMaxSentBlockSize, true ); #define AUTOCHECK_DISPLAY nlwarning //#define AUTOCHECK_DISPLAY CUnifiedNetwork::getInstance()->displayInternalTables (), nlerror const TServiceId TServiceId::InvalidId = TServiceId(uint16(~0)); // // Callbacks from NAMING SERVICE // // when a service registers void uNetRegistrationBroadcast(const string &name, TServiceId sid, const vector &addr) { nldebug ("HNETL5: + naming %s-%u '%s'", name.c_str(), sid.get(), vectorCInetAddressToString(addr).c_str ()); CUnifiedNetwork *uni= CUnifiedNetwork::getInstance(); if (uni->_SId == sid) { // it's me! don't add me!!! return; } // add the unified connection if(sid.get() >= uni->_IdCnx.size ()) uni->_IdCnx.resize (sid.get()+1); if (uni->_IdCnx[sid.get()].State == CUnifiedNetwork::CUnifiedConnection::NotUsed) { uni->_IdCnx[sid.get()] = CUnifiedNetwork::CUnifiedConnection(name, sid, false); uni->_UsedConnection.push_back (sid); } if (!uni->_IdCnx[sid.get()].ExtAddress.empty ()) AUTOCHECK_DISPLAY ("HNETL5: %s-%u already inserted in the table with '%s'", name.c_str(), sid.get(), vectorCInetAddressToString (uni->_IdCnx[sid.get()].ExtAddress).c_str ()); // set the list of external addresses nlassert (!addr.empty()); uni->_IdCnx[sid.get()].ExtAddress = addr; // associate nid with ext address uni->_IdCnx[sid.get()].setupNetworkAssociation (uni->_NetworkAssociations, uni->_DefaultNetwork); } // when a service unregisters void uNetUnregistrationBroadcast(const string &name, TServiceId sid, const vector &addr) { nldebug ("HNETL5: - naming %s-%u '%s'", name.c_str(), sid.get(), vectorCInetAddressToString (addr).c_str ()); // get the service connection CUnifiedNetwork *uni = CUnifiedNetwork::getInstance(); CUnifiedNetwork::CUnifiedConnection *uc = uni->getUnifiedConnection (sid); if (uc == 0) return; // should never happen, the getUnifiedConnection() will generate a AUTOCHECK_DISPLAY // call the user callback uni->callServiceDownCallback(uc->ServiceName, uc->ServiceId); if(!uc->Connections.empty ()) { // set all connection to dead, now, all messages received on this socket will be ignored and closed for (uint i = 0; i < uc->Connections.size (); ++i) { if (uc->Connections[i].valid()) uc->Connections[i].setAppId (AppIdDeadConnection); } // // It's the first connection that added the _NamedCnx so if there s no connection, no need to // remove entry in _NamedCnx // uni->removeNamedCnx (uc->ServiceName, uc->ServiceId); } // remove the _UsedConnection bool found = false; for (vector::iterator it = uni->_UsedConnection.begin (); it != uni->_UsedConnection.end(); it++) { if (*it == uc->ServiceId) { found = true; uni->_UsedConnection.erase (it); break; } } if (!found) AUTOCHECK_DISPLAY ("HNETL5: can't find the sid %hu in the _UsedConnection", uc->ServiceId.get()); // reset the unified connection uc->reset (); } // // Callbacks from connection/disconnection services // void uncbConnection(TSockId from, void * /* arg */) { nlinfo ("HNETL5: + connec '%s'", from->asString().c_str()); from->setAppId (AppIdDeadConnection); } void uncbDisconnection(TSockId from, void * /* arg */) { if(from->appId () == AppIdDeadConnection) { nlinfo ("HNETL5: - connec '%s'", from->asString().c_str()); } else { CUnifiedNetwork *uni = CUnifiedNetwork::getInstance(); TServiceId sid(uint16(from->appId())); CUnifiedNetwork::CUnifiedConnection *uc = uni->getUnifiedConnection (sid); if (uc == 0) { nlinfo ("HNETL5: - connec '%s' sid %hu", from->asString().c_str(), sid.get()); } else { nlinfo ("HNETL5: - connec '%s' %s-%hu", from->asString().c_str(), uc->ServiceName.c_str (), sid.get()); if (uc->IsExternal) { if (!uc->AutoRetry) { // If it s a external service with no auto retry, remove the connection // call the user callback uni->callServiceDownCallback(uc->ServiceName, uc->ServiceId); uni->removeNamedCnx (uc->ServiceName, uc->ServiceId); uni->_ConnectionToReset.push_back(uc->ServiceId); } else { // call the user callback uni->callServiceDownCallback(uc->ServiceName, uc->ServiceId); } } else { // reset the connection uint i; for (i = 0; i < uc->Connections.size (); i++) { if (uc->Connections[i].valid() && uc->Connections[i].CbNetBase->getSockId(uc->Connections[i].HostId) == from) { if (uc->Connections[i].IsServerConnection) { // we have to remove the stuffs now because HostId will not be accessible later uc->Connections[i].reset(); } else { // if it s a client, we can't delete now because the callback client is currently in use // only disconnect if(uc->Connections[i].CbNetBase->connected ()) { uc->Connections[i].CbNetBase->disconnect (uc->Connections[i].HostId); } } break; } } if (i == uc->Connections.size ()) { AUTOCHECK_DISPLAY ("HNETL5: received a disconnection from a service but the connection is not in my list!"); } } } from->setAppId (AppIdDeadConnection); } } // // Callback from identification services // void uncbServiceIdentification(CMessage &msgin, TSockId from, CCallbackNetBase &netbase) { string inSName; TServiceId inSid; if (from->appId () != AppIdDeadConnection) AUTOCHECK_DISPLAY ("HNETL5: received a connect ident from an unknown connection 0x%"NL_I64"X", from->appId ()); // recover the service name and id msgin.serial(inSName); msgin.serial(inSid); uint8 pos; msgin.serial (pos); bool isExternal; msgin.serial (isExternal); nlinfo ("HNETL5: + connect ident '%s' %s-%hu pos %hu ext %d", from->asString().c_str(), inSName.c_str(), inSid.get(), (uint16)pos, (uint8)isExternal); if(isExternal) { nlassert (pos == 0); } if (inSid.get() == 0) { if (isExternal) { inSid = CUnifiedNetwork::getInstance ()->_ExtSId; CUnifiedNetwork::getInstance ()->_ExtSId.set(CUnifiedNetwork::getInstance ()->_ExtSId.get()+1); // the following was an nlwarning but this is in fact desired behavior and not an error so we have changed to nlinfo nlinfo ("HNETL5: Received a connection from a service with a SId 0, we give him the SId %d", inSid.get()); } else { nlwarning ("HNETL5: Received a connection from a service with a SId 0 and wasn't external, disconnecting it"); netbase.disconnect(); return; } } from->setAppId(inSid.get()); // add a new connection to the list CUnifiedNetwork *uni= CUnifiedNetwork::getInstance(); if(inSid.get() >= uni->_IdCnx.size ()) { uni->_IdCnx.resize (inSid.get()+1); } switch(uni->_IdCnx[inSid.get()].State) { case CUnifiedNetwork::CUnifiedConnection::NotUsed: // add the new unified connection uni->_IdCnx[inSid.get()] = CUnifiedNetwork::CUnifiedConnection(inSName, inSid, isExternal); uni->_UsedConnection.push_back (inSid); break; default: break; } if (uni->_IdCnx[inSid.get()].IsExternal != isExternal) { AUTOCHECK_DISPLAY ("HNETL5: Receive a connection that is not totally external %d %d", uni->_IdCnx[inSid.get()].IsExternal, isExternal); return; } bool FirstConnection = (uni->_IdCnx[inSid.get()].Connections.size () == 0); // add the connection to the already inserted unified connection if (pos >= uni->_IdCnx[inSid.get()].Connections.size ()) { uni->_IdCnx[inSid.get()].Connections.resize(pos+1); } uni->_IdCnx[inSid.get()].Connections[pos] = CUnifiedNetwork::CUnifiedConnection::TConnection(&netbase, from); // If the connection is external, we'll never receive the ExtAddress by the naming service, so add it manually if (isExternal) { uni->_IdCnx[inSid.get()].ExtAddress.push_back (netbase.hostAddress (from)); uni->_IdCnx[inSid.get()].setupNetworkAssociation (uni->_NetworkAssociations, uni->_DefaultNetwork); } // send the callback to the user with the first connection if (FirstConnection) { // insert the name in the map to be able to send message with the name uni->addNamedCnx (inSName, inSid); uni->callServiceUpCallback (inSName, inSid); } } // the callbacks wrapper void uncbMsgProcessing(CMessage &msgin, TSockId from, CCallbackNetBase &/* netbase */) { if (from->appId() == AppIdDeadConnection) { AUTOCHECK_DISPLAY ("HNETL5: Receive a message from a dead connection"); return; } CUnifiedNetwork *uni = CUnifiedNetwork::getInstance(); TServiceId sid(uint16(from->appId())); CUnifiedNetwork::TMsgMappedCallback::iterator itcb; itcb = uni->_Callbacks.find(msgin.getName()); if (itcb == uni->_Callbacks.end()) { // the callback doesn't exist nlwarning ("HNETL5: Can't find callback '%s' called by service %hu", msgin.getName().c_str(), sid.get()); } else { CUnifiedNetwork::CUnifiedConnection *uc = uni->getUnifiedConnection (sid); if (uc == 0) { nlwarning ("HNETL5: Received a message from a service %hu that is not ready (bad appid? 0x%"NL_I64"X)", sid.get(), from->appId ()); return; } if((*itcb).second == 0) { nlwarning ("HNETL5: Received message %s from a service %hu but the associated callback is NULL", msgin.getName ().c_str(), sid.get()); return; } { static map timers; map::iterator it; { H_AUTO(L5UCHTimerOverhead); string callbackName = "USRCB_" + msgin.getName(); it = timers.find(callbackName); if(it == timers.end()) { it = timers.insert(make_pair(callbackName, CHTimer(NULL))).first; (*it).second.setName((*it).first.c_str()); } } { H_AUTO(L5UserCallback); TTime before = CTime::getLocalTime(); (*it).second.before(); // const std::string &cbName = itcb->first; (*itcb).second (msgin, uc->ServiceName, sid); (*it).second.after(); TTime after = CTime::getLocalTime(); // sum the time used to do callback TimeInCallback += uint32((after-before)); } } uc->TotalCallbackCalled++; TotalCallbackCalled++; } } TCallbackItem unServerCbArray[] = { { "UN_SIDENT", uncbServiceIdentification } }; // // Alive check thread // class CAliveCheck : public NLMISC::IRunnable { public: CAliveCheck() : ExitRequired(false) { } virtual void run(); virtual ~CAliveCheck() { } volatile bool ExitRequired; static CAliveCheck* Thread; struct CCheckAddress { CCheckAddress() : ConnectionId(0xDEAD), NeedCheck(false), AddressValid(false) { } CInetAddress Address; std::string ServiceName; TServiceId ServiceId; uint ConnectionId; uint ConnectionIndex; volatile bool NeedCheck; volatile bool AddressValid; }; CCheckAddress CheckList[128]; void checkService(CInetAddress address, uint connectionId, uint connectionIndex, const std::string &service, TServiceId id); }; CAliveCheck* CAliveCheck::Thread = NULL; IThread* AliveThread = NULL; void CAliveCheck::run() { // setup thread Thread = this; TTime t = CTime::getLocalTime(); while (!ExitRequired) { if (CTime::getLocalTime() - t < 10000) { nlSleep(100); continue; } uint i; for (i=0; i laddr = CInetAddress::localAddresses(); _RecordingState = rec; _Name = shortName; _SId = sid; if (addr != 0) _NamingServiceAddr = *addr; // if the address isn't 0, uses the naming service if (_NamingServiceAddr.isValid ()) { // connect the callback to know when a new service comes in or goes down CNamingClient::setRegistrationBroadcastCallback(uNetRegistrationBroadcast); CNamingClient::setUnregistrationBroadcastCallback(uNetUnregistrationBroadcast); // connect to the naming service (may generate a ESocketConnectionFailed exception) CNamingClient::connect(_NamingServiceAddr, _RecordingState, laddr); if (port == 0) port = CNamingClient::queryServicePort (); } #ifdef NL_OS_UNIX /// Init the main pipe to select() on data available if ( ::pipe( _MainDataAvailablePipe ) != 0 ) nlwarning( "Unable to create main D.A. pipe" ); //nldebug( "Pipe: created" ); #endif // setup the server callback only if server port != 0, otherwise there's no server callback _ServerPort = port; if(_ServerPort != 0) { nlassert (_CbServer == 0); _CbServer = new CCallbackServer( CCallbackNetBase::Off, "", true, false ); // don't init one pipe per connection #ifdef NL_OS_UNIX _CbServer->setExternalPipeForDataAvailable( _MainDataAvailablePipe ); // the main pipe is shared for all connections //nldebug( "Pipe: set (server %p)", _CbServer ); #endif bool retry = false; do { retry = false; try { _CbServer->init(port); } catch (ESocket &) { nlwarning("Failed to init the listen socket on port %u, is the service already running ?", port); // wait a little before retrying nlSleep(5000); retry = true; } } while(retry); _CbServer->addCallbackArray(unServerCbArray, 1); // the service ident callback _CbServer->setDefaultCallback(uncbMsgProcessing); // the default callback wrapper _CbServer->setConnectionCallback(uncbConnection, NULL); _CbServer->setDisconnectionCallback(uncbDisconnection, NULL); } else { nlinfo ("HNETL5: ServerPort is 0 so I don't create a CCallbackServer"); } if (CNamingClient::connected()) { // register the service for (uint i = 0; i < laddr.size(); i++) laddr[i].setPort(_ServerPort); if (_SId.get() == 0) { if ( ! CNamingClient::registerService(_Name, laddr, _SId) ) { nlinfo ("HNETL5: Registration denied"); return false; } } else { CNamingClient::registerServiceWithSId(_Name, laddr, _SId); } sid = _SId; nlinfo ("HNETL5: Server '%s' added, registered and listen to port %hu", _Name.c_str (), _ServerPort); } AliveThread = IThread::create(new CAliveCheck(), 1024*4); AliveThread->start(); _Initialised = true; return true; } void CUnifiedNetwork::connect() { nlassertex(_Initialised == true, ("Try to CUnifiedNetwork::connect() whereas it is not initialised yet")); if (ThreadCreator != NLMISC::getThreadId()) nlwarning ("HNETL5: Multithread access but this class is not thread safe thread creator = %u thread used = %u", ThreadCreator, NLMISC::getThreadId()); if (CNamingClient::connected()) { // get the services list const list &services = CNamingClient::getRegisteredServices(); // connects to the registered services list::const_iterator its; // don't connect to itself for (its = services.begin(); its != services.end(); ++its) { if (_SId != (*its).SId) { // add service with name, address, ident, not external, service id, and not autoretry (obsolete) // we put the last true because the name callback should already inserted it by uNetRegistrationBroadcast() addService((*its).Name, (*its).Addr, true, false, (*its).SId, false, true); } else { // don't process services received after mine because they'll connect to me break; } } } } void CUnifiedNetwork::release(bool mustFlushSendQueues, const std::vector& namesOfOnlyServiceToFlushSending) { if (!_Initialised) return; // the commands can't be invoked unregisterCommandsHandler(); // terminate the auto reconnection thread if (CAliveCheck::Thread) { CAliveCheck::Thread->ExitRequired = true; AliveThread->wait(); delete CAliveCheck::Thread; delete AliveThread; } if (ThreadCreator != NLMISC::getThreadId()) nlwarning ("HNETL5: Multithread access but this class is not thread safe thread creator = %u thread used = %u", ThreadCreator, NLMISC::getThreadId()); // Ensure all outgoing data are sent before disconnecting, if requested if ( mustFlushSendQueues ) { nlinfo( "HNETL5: Flushing sending queues..." ); float totalBytes = 0; uint bytesRemaining, i=0; while ( (bytesRemaining = tryFlushAllQueues( namesOfOnlyServiceToFlushSending )) != 0 ) { if ( i == 0 ) totalBytes = (float)bytesRemaining; if ( i % 20 == 0 ) { nldebug( "%.1f%% of %.3f KB flushed so far", // display without HNETL5 to bypass filters! ((float)(bytesRemaining-totalBytes))/totalBytes, totalBytes / 1024.0f ); } ++i; nlSleep( 100 ); } nldebug( "HNETL5: Flush done in %u steps", i+1 ); } // disconnect all clients if(_CbServer) { _CbServer->disconnect(InvalidSockId); delete _CbServer; _CbServer = 0; } // disconnect all connections to servers for (uint i = 0; i<_IdCnx.size(); ++i) { if (_IdCnx[i].State != CUnifiedNetwork::CUnifiedConnection::NotUsed) { for(uint j = 0 ; j < _IdCnx[i].Connections.size (); j++) { if (_IdCnx[i].Connections[j].valid() && !_IdCnx[i].Connections[j].IsServerConnection) { if (_IdCnx[i].Connections[j].CbNetBase->connected ()) _IdCnx[i].Connections[j].CbNetBase->disconnect(); delete _IdCnx[i].Connections[j].CbNetBase; } } _IdCnx[i].Connections.clear (); } } // clear all other data _IdCnx.clear(); _UsedConnection.clear (); _NamedCnx.clear(); _UpCallbacks.clear(); _DownCallbacks.clear(); _Callbacks.clear(); // disconnect the connection with the naming service if (CNamingClient::connected ()) CNamingClient::disconnect (); #ifdef NL_OS_UNIX ::close( _MainDataAvailablePipe[PipeRead] ); ::close( _MainDataAvailablePipe[PipeWrite] ); #endif } void CUnifiedNetwork::addService(const string &name, const CInetAddress &addr, bool sendId, bool external, TServiceId sid, bool autoRetry, bool shouldBeAlreayInserted) { vector addrs; addrs.push_back (addr); addService (name, addrs, sendId, external, sid, autoRetry, shouldBeAlreayInserted); } void CUnifiedNetwork::addService(const string &name, const vector &addr, bool sendId, bool external, TServiceId sid, bool autoRetry, bool shouldBeAlreayInserted) { nlassertex(_Initialised == true, ("Try to CUnifiedNetwork::addService() whereas it is not initialised yet")); if (ThreadCreator != NLMISC::getThreadId()) nlwarning ("HNETL5: Multithread access but this class is not thread safe thread creator = %u thread used = %u", ThreadCreator, NLMISC::getThreadId()); if (external) { sid = _ExtSId; _ExtSId.set(_ExtSId.get()+1); } nlinfo("HNETL5: addService %s-%hu '%s'", name.c_str(), sid.get(), vectorCInetAddressToString(addr).c_str()); if (external && addr.size () != 1) { AUTOCHECK_DISPLAY ("HNETL5: Can't add external service with more than one connection"); } // add the entry in the unified connection table if (sid.get() >= _IdCnx.size()) _IdCnx.resize(sid.get()+1); CUnifiedConnection *uc = &_IdCnx[sid.get()]; // at this point it s possible that the service already added in the _IdCnx by the uNetRegistrationBroadcast() if (shouldBeAlreayInserted && _IdCnx[sid.get()].State == CUnifiedNetwork::CUnifiedConnection::NotUsed) AUTOCHECK_DISPLAY ("HNETL5: the unified connection should already set by the naming reg broadcast and is not (%hu)", sid.get()); if (!shouldBeAlreayInserted && _IdCnx[sid.get()].State == CUnifiedNetwork::CUnifiedConnection::Ready) AUTOCHECK_DISPLAY ("HNETL5: the unified connection should not already set but is (%hu)", sid.get()); if (_IdCnx[sid.get()].State == CUnifiedNetwork::CUnifiedConnection::NotUsed) { *uc = CUnifiedConnection(name, sid, external); _UsedConnection.push_back (sid); } else { // If the entry already set, check that all is correct if (name != uc->ServiceName) AUTOCHECK_DISPLAY ("HNETL5: name are different in addService %s %s", name.c_str (), uc->ServiceName.c_str ()); if (sid != uc->ServiceId) AUTOCHECK_DISPLAY ("HNETL5: sid are different in addService %hu %hu", sid.get(), uc->ServiceId.get()); if (addr != uc->ExtAddress) AUTOCHECK_DISPLAY ("HNETL5: external addr are different in addService '%s' '%s'", vectorCInetAddressToString(addr).c_str(), vectorCInetAddressToString(uc->ExtAddress).c_str ()); } uc->AutoRetry = autoRetry; uc->SendId = sendId; uc->ExtAddress = addr; nlassert (!addr.empty()); // associate nid with ext address uc->setupNetworkAssociation (_NetworkAssociations, _DefaultNetwork); // connect to all connection bool connectSuccess; if (uc->Connections.size () < addr.size ()) { uc->Connections.resize (addr.size ()); } vector laddr = CInetAddress::localAddresses(); for (uint i = 0; i < addr.size(); i++) { // first we have to look if we have a network that can established the connection uint j = 0; // it s 127.0.0.1, it s ok if (!addr[i].is127001 ()) { for (j = 0; j < laddr.size (); j++) { if (laddr[j].internalNetAddress () == addr[i].internalNetAddress ()) { // it's ok, we can try break; } } // If we don't found a valid network, we'll try with the first one. // It's happen, for example, when you try to connect to a service that is not in the network but use IP translation if (j == laddr.size ()) { nlwarning ("HNETL5: I can't access '%s' because I haven't a net card on this network, we'll use the first network", addr[i].asString ().c_str ()); j = 0; } } // Create a new connection with the service, setup callback and connect CCallbackClient *cbc = new CCallbackClient( CCallbackNetBase::Off, "", true, false ); // don't init one pipe per connection #ifdef NL_OS_UNIX cbc->setExternalPipeForDataAvailable( _MainDataAvailablePipe ); // the main pipe is shared for all connections //nldebug( "Pipe: set (client %p)", cbc ); #endif cbc->setDisconnectionCallback(uncbDisconnection, NULL); cbc->setDefaultCallback(uncbMsgProcessing); cbc->getSockId()->setAppId(sid.get()); try { cbc->connect(addr[i]); connectSuccess = true; } catch (ESocketConnectionFailed &e) { nlwarning ("HNETL5: can't connect to %s (sid %u) now (%s) '%s'", name.c_str(), sid.get(), e.what (), addr[i].asString ().c_str()); connectSuccess = false; } if (!connectSuccess && !autoRetry) { nlwarning ("HNETL5: Can't add service because no retry and can't connect"); delete cbc; } else { uc->Connections[i] = CUnifiedNetwork::CUnifiedConnection::TConnection(cbc); } if (connectSuccess && sendId) { // send identification to the service CMessage msg("UN_SIDENT"); msg.serial(_Name); TServiceId ssid = _SId; if (uc->IsExternal) { // in the case that the service is external, we can't send our sid because the external service can // have other connection with the same sid (for example, LS can have 2 WS with same sid => sid = 0 and leave // the other side to find a good number ssid.set(0); } msg.serial(ssid); // serializes a 16 bits service id uint8 pos = uint8(j); msg.serial(pos); // send the position in the connection table msg.serial (uc->IsExternal); cbc->send (msg); } } if (addr.size () != uc->Connections.size()) { nlwarning ("HNETL5: Can't connect to all connections to the service %d/%d", addr.size (), uc->Connections.size()); } bool cntok = false; for (uint j = 0; j < uc->Connections.size(); j++) { if (uc->Connections[j].CbNetBase != NULL) { if (uc->Connections[j].CbNetBase->connected ()) { cntok = true; break; } } } if (cntok) { // add the name only if at least one connection is ok addNamedCnx (name, sid); callServiceUpCallback (name, sid); // global callback ("*") will be called even for external service } nldebug ("HNETL5: addService was successful"); } // // // #define TIME_BLOCK(tick, instr) \ { \ TTicks _time_block_before = CTime::getPerformanceTime(); \ instr ; \ TTicks _time_block_after = CTime::getPerformanceTime(); \ tick += (_time_block_after - _before); \ } /* * */ void CUnifiedNetwork::update(TTime timeout) { H_AUTO(CUnifiedNetworkUpdate); H_BEFORE(UNMisc1); nlassertex(_Initialised == true, ("Try to CUnifiedNetwork::update() whereas it is not initialised yet")); if (ThreadCreator != NLMISC::getThreadId()) nlwarning ("HNETL5: Multithread access but this class is not thread safe thread creator = %u thread used = %u", ThreadCreator, NLMISC::getThreadId()); bool enableRetry; // true every 5 seconds to reconnect if necessary // Compute the real timeout based on the next update timeout TTime t0 = CTime::getLocalTime (); if (timeout > 0) { if (_NextUpdateTime == 0) { _NextUpdateTime = t0 + timeout; } else { TTime err = t0 - _NextUpdateTime; _NextUpdateTime += timeout; // if we are too late, resync to the next value while (err > timeout) { err -= timeout; _NextUpdateTime += timeout; } timeout -= err; if (timeout < 0) timeout = 0; } } TTime remainingTime = timeout; // check if we need to retry to connect to the client enableRetry = (t0-_LastRetry > 5000); if (enableRetry) _LastRetry = t0; H_AFTER(UNMisc1); H_BEFORE(UNNamingCheck); // Try to reconnect to the naming service if connection lost if (_NamingServiceAddr.isValid ()) { if (CNamingClient::connected ()) { CNamingClient::update (); } else if (enableRetry) { H_AUTO(L5NSReconnect); try { vector laddr = CInetAddress::localAddresses(); CNamingClient::connect (_NamingServiceAddr, _RecordingState, laddr); // re-register the service for (uint i = 0; i < laddr.size(); i++) laddr[i].setPort(_ServerPort); CNamingClient::resendRegisteration (_Name, laddr, _SId); } catch (ESocketConnectionFailed &) { nlwarning ("HNETL5: Could not connect to the Naming Service (%s). Retrying in a few seconds...", _NamingServiceAddr.asString().c_str()); } } } H_AFTER(UNNamingCheck); H_BEFORE(UNUpdateCnx); for(;;) { H_AUTO(L5OneLoopUpdate); if (CAliveCheck::Thread) { uint i; for (i=0; iCheckList)/sizeof(CAliveCheck::Thread->CheckList[0]); ++i) { CAliveCheck::CCheckAddress &address = CAliveCheck::Thread->CheckList[i]; if (address.NeedCheck && address.AddressValid) { CUnifiedConnection &uc = _IdCnx[address.ConnectionId]; if (uc.ServiceName == address.ServiceName && uc.ServiceId == address.ServiceId && uc.ValidRequested) { uc.ValidRequested = false; uc.ValidExternal = true; } address.NeedCheck = false; address.AddressValid = false; } } } // update all server connections if (_CbServer) { _CbServer->update2((sint32)remainingTime, 0); } // update all client connections for (uint k = 0; k<_UsedConnection.size(); ++k) { H_AUTO(UNBrowseConnections); nlassert (_IdCnx[_UsedConnection[k].get()].State == CUnifiedNetwork::CUnifiedConnection::Ready); for (uint j = 0; j < _IdCnx[_UsedConnection[k].get()].Connections.size (); j++) { // WARNING : don't take a reference in the outside loop because // _IdCnx can be resized by execution of a callback CUnifiedConnection &uc = _IdCnx[_UsedConnection[k].get()]; nlassert(_IdCnx[_UsedConnection[k].get()].Connections.size() > j); CUnifiedConnection::TConnection &conn = _IdCnx[_UsedConnection[k].get()].Connections[j]; H_AUTO(UNBrowseSubConnections); if (!conn.valid()) continue; if (conn.IsServerConnection) continue; nlassert(uc.Connections.size() > j); if (conn.CbNetBase->connected ()) { nlassert(uc.Connections.size() > j); conn.CbNetBase->update2((sint32)remainingTime, 0); } else if (enableRetry && uc.AutoRetry) { if (uc.ValidExternal) { uc.ValidExternal = false; uc.ValidRequested = false; autoReconnect( uc, j ); } else if (!uc.ValidRequested && CAliveCheck::Thread) { uc.ValidRequested = true; CAliveCheck::Thread->checkService(uc.ExtAddress[j], _UsedConnection[k].get(), j, uc.ServiceName, uc.ServiceId); } } } } // reset closed client connection for (uint i=0; i<_ConnectionToReset.size(); ++i) { // remove the _UsedConnection bool found = false; for (vector::iterator it = _UsedConnection.begin (); it != _UsedConnection.end(); it++) { if (*it == _IdCnx[_ConnectionToReset[i].get()].ServiceId) { found = true; _UsedConnection.erase (it); break; } } if (!found) AUTOCHECK_DISPLAY ("HNETL5: can't find the sid %hu in the _UsedConnection", _IdCnx[_ConnectionToReset[i].get()].ServiceId.get()); _IdCnx[_ConnectionToReset[i].get()].reset(); } _ConnectionToReset.clear(); enableRetry = false; if ( FlushSendsBeforeSleep.get() ) { // Flush all connections L5TotalBytesInLowLevelSendQueues = tryFlushAllQueues(); } // t0 --------- previousTime -------------------------- t0 + timeout // prevRemainingTime // // t0 -------------- currentTime ---------------------- t0 + timeout // remainingTime TTime prevRemainingTime = remainingTime; TTime currentTime = CTime::getLocalTime(); remainingTime = t0 + timeout - currentTime; // If it's the end (or if the Unix system time was changed forwards), don't sleep (currentTime > t0 + timeout) if ( remainingTime <= 0 ) break; // If the Unix system time was changed backwards, don't wait more than requested and don't provide an erroneous time to the sleep function that would fail (currentTime < previousTime) if ( remainingTime > prevRemainingTime ) { // Restart at previousTime nldebug( "Backward time sync detected (at least -%"NL_I64"d ms)", remainingTime - prevRemainingTime ); remainingTime = prevRemainingTime; t0 = currentTime - (timeout - remainingTime); } #ifdef NL_OS_UNIX // Sleep until the time expires or we receive a message H_BEFORE(L5UpdateSleep); switch ( UseYieldMethod.get() ) { case 0: sleepUntilDataAvailable( remainingTime ); break; // accurate sleep with select() case 1: ::usleep(1000); break; // 20 ms case 2: nlSleep(1); break; // 10 ms (by nanosleep, but 20 ms measured on kernel 2.4.20) case 3: ::sched_yield(); break; // makes all slow (at least on kernel 2.4.20) ! default: break; // don't sleep at all, makes all slow! } H_AFTER(L5UpdateSleep); #else // Enable windows multithreading before rescanning all connections H_TIME(L5UpdateSleep, nlSleep(1);); // 0 (yield) would be too harmful to other applications #endif } H_AFTER(UNUpdateCnx); H_TIME(UNAutoCheck, autoCheck();); } /* * Auto-reconnect */ void CUnifiedNetwork::autoReconnect( CUnifiedConnection &uc, uint connectionIndex ) { H_AUTO(L5AutoReconnect); try { CCallbackClient *cbc = (CCallbackClient *)uc.Connections[connectionIndex].CbNetBase; cbc->connect(uc.ExtAddress[connectionIndex]); uc.Connections[connectionIndex].CbNetBase->getSockId()->setAppId(uc.ServiceId.get()); nlinfo ("HNETL5: reconnection to %s-%hu success", uc.ServiceName.c_str(), uc.ServiceId.get()); // add the name only if at least one connection is ok if (!haveNamedCnx (uc.ServiceName, uc.ServiceId)) addNamedCnx (uc.ServiceName, uc.ServiceId); // resend the identification is necessary if (uc.SendId) { // send identification to the service CMessage msg("UN_SIDENT"); msg.serial(_Name); TServiceId ssid = _SId; if (uc.IsExternal) { // in the case that the service is external, we can't send our sid because the external service can // have other connection with the same sid (for example, LS can have 2 WS with same sid => sid = 0 and leave // the other side to find a good number ssid.set(0); } msg.serial(ssid); // serializes a 16 bits service id uint8 pos = uint8(connectionIndex); msg.serial(pos); // send the position in the connection table msg.serial (uc.IsExternal); uc.Connections[connectionIndex].CbNetBase->send (msg, uc.Connections[connectionIndex].HostId); } // call the user callback callServiceUpCallback (uc.ServiceName, uc.ServiceId); } catch (ESocketConnectionFailed &e) { #if FINAL_VERSION nlinfo ("HNETL5: can't connect to %s-%hu now (%s)", uc.ServiceName.c_str(), uc.ServiceId.get(), e.what ()); #else nlwarning ("HNETL5: can't connect to %s-%hu now (%s)", uc.ServiceName.c_str(), uc.ServiceId.get(), e.what ()); #endif } } #ifdef NL_OS_UNIX /* * */ void CUnifiedNetwork::sleepUntilDataAvailable( TTime msecMax ) { // Prevent looping infinitely if an erroneous time was provided if ( msecMax > 999 ) // limit not told in Linux man but here: http://docs.hp.com/en/B9106-90009/select.2.html msecMax = 999; // Prepare for select() fd_set readers; FD_ZERO( &readers ); FD_SET( _MainDataAvailablePipe[PipeRead], &readers ); SOCKET descmax = _MainDataAvailablePipe[PipeRead] + 1; // Select timeval tv; tv.tv_sec = 0; tv.tv_usec = msecMax * 1000; //nldebug( "Select %u ms", (uint)msecMax ); //TTime before = CTime::getLocalTime(); int res = ::select( descmax+1, &readers, NULL, NULL, &tv ); if ( res == -1 ) nlwarning( "HNETL5: Select failed in sleepUntilDataAvailable"); //nldebug( "Slept %u ms", (uint)(CTime::getLocalTime()-before) ); } #endif bool CUnifiedNetwork::isConnectionConnected(TServiceId sid) const { // a Connected connection is a connection that is Ready but that is not yet connected (serverUp will be called latter via L5). return sid.get() < _IdCnx.size() && _IdCnx[sid.get()].State == CUnifiedConnection::Ready && !_IdCnx[sid.get()].Connections.empty(); } // // // uint8 CUnifiedNetwork::findConnectionId (TServiceId sid, uint8 nid) { if (_IdCnx[sid.get()].Connections.size () == 0) { nlwarning ("HNETL5: Can't send message to %s because no connection are available", _IdCnx[sid.get()].ServiceName.c_str ()); return 0xFF; } // by default, connection id will be the default one uint8 connectionId = _IdCnx[sid.get()].DefaultNetwork; if (nid == 0xFF) { // it s often happen because they didn't set a good network configuration, so it s in debug to disable it easily //nldebug ("HNETL5: nid %hu, will use the default connection %hu", (uint16)nid, (uint16)connectionId); } else if (nid >= _IdCnx[sid.get()].NetworkConnectionAssociations.size()) { nlwarning ("HNETL5: No net association for nid %hu, use the default connection %hu", (uint16)nid, (uint16)connectionId); } else { if (_IdCnx[sid.get()].NetworkConnectionAssociations[nid] >= _IdCnx[sid.get()].Connections.size ()) { nlwarning ("HNETL5: Can't send message to %s because nid %d point on a bad connection (%d and only have %d cnx), use default connection", _IdCnx[sid.get()].ServiceName.c_str (), nid, connectionId, _IdCnx[sid.get()].Connections.size ()); } else { connectionId = _IdCnx[sid.get()].NetworkConnectionAssociations[nid]; } } if (connectionId >= _IdCnx[sid.get()].Connections.size() || !_IdCnx[sid.get()].Connections[connectionId].valid() || !_IdCnx[sid.get()].Connections[connectionId].CbNetBase->connected()) { // there's a problem with the selected connectionID, so try to find a valid one nlwarning ("HNETL5: Can't find selected connection id %hu to send message to %s because connection is not valid or connected, find a valid connection id", (uint16)connectionId, _IdCnx[sid.get()].ServiceName.c_str ()); for (connectionId = 0; connectionId < _IdCnx[sid.get()].Connections.size(); connectionId++) { if (_IdCnx[sid.get()].Connections[connectionId].valid() && _IdCnx[sid.get()].Connections[connectionId].CbNetBase->connected()) { // we found one at last, use this one //nldebug ("HNETL5: Ok, we found a valid connectionid, use %hu", (uint16)connectionId); break; } } if (connectionId == _IdCnx[sid.get()].Connections.size()) { nlwarning ("HNETL5: Can't send message to %s because default connection is not exist, valid or connected", _IdCnx[sid.get()].ServiceName.c_str ()); return 0xFF; } } return connectionId; } // // // uint CUnifiedNetwork::send(const string &serviceName, const CMessage &msgout, bool warnIfNotFound, uint8 nid) { nlassertex(_Initialised == true, ("Try to CUnifiedNetwork::send(const string&, const CMessage&) whereas it is not initialised yet")); if (ThreadCreator != NLMISC::getThreadId()) nlwarning ("HNETL5: Multithread access but this class is not thread safe thread creator = %u thread used = %u", ThreadCreator, NLMISC::getThreadId()); TNameMappedConnection::const_iterator it; pair range; range = _NamedCnx.equal_range(serviceName); uint found = 0; if (range.first != _NamedCnx.end()) { for (it=range.first; it!=range.second; ++it) { TServiceId sid = it->second; if (sid.get() >= _IdCnx.size () || _IdCnx[sid.get()].State != CUnifiedNetwork::CUnifiedConnection::Ready) { // It often happen when the service is down (connection broke and the naming not already say that it s down) // In this case, just warn nlwarning ("HNETL5: Can't send %s to the service '%s' because it was in the _NamedCnx but not in _IdCnx (means that the service is down)", msgout.getName().c_str(), serviceName.c_str ()); return false; } ++found; uint8 connectionId = findConnectionId (sid, nid); if (connectionId == 0xff) // failed { nlwarning ("HNETL5: Can't send %s message to %hu because no connection available", msgout.getName().c_str(), sid.get()); continue; } _IdCnx[sid.get()].Connections[connectionId].CbNetBase->send (msgout, _IdCnx[sid.get()].Connections[connectionId].HostId); } } if (!found && warnIfNotFound) nlwarning ("HNETL5: can't find service %s to send message %s", serviceName.c_str(), msgout.getName().c_str()); return found; } bool CUnifiedNetwork::send(TServiceId sid, const CMessage &msgout, uint8 nid) { nlassertex(_Initialised == true, ("Try to CUnifiedNetwork::send(TServiceId, const CMessage&) whereas it is not initialised yet")); if (ThreadCreator != NLMISC::getThreadId()) nlwarning ("HNETL5: Multithread access but this class is not thread safe thread creator = %u thread used = %u", ThreadCreator, NLMISC::getThreadId()); if (sid.get() >= _IdCnx.size () || _IdCnx[sid.get()].State != CUnifiedNetwork::CUnifiedConnection::Ready) { // Happens when trying to send a message to an unknown service id nlwarning ("HNETL5: Can't send %s to the service '%hu' because not in _IdCnx", msgout.getName().c_str(), sid.get()); return false; } uint8 connectionId = findConnectionId (sid, nid); if (connectionId == 0xff) // failed { nlwarning ("HNETL5: Can't send %s to the service '%hu' because no connection available", msgout.getName().c_str(), sid.get()); return false; } _IdCnx[sid.get()].Connections[connectionId].CbNetBase->send (msgout, _IdCnx[sid.get()].Connections[connectionId].HostId); return true; } void CUnifiedNetwork::sendAll(const CMessage &msgout, uint8 nid) { nlassertex(_Initialised == true, ("Try to CUnifiedNetwork::send(const CMessage&) whereas it is not initialised yet")); if (ThreadCreator != NLMISC::getThreadId()) nlwarning ("HNETL5: Multithread access but this class is not thread safe thread creator = %u thread used = %u", ThreadCreator, NLMISC::getThreadId()); for (TServiceId::size_type i=0; i<_IdCnx.size(); ++i) { if (_IdCnx[i].State == CUnifiedNetwork::CUnifiedConnection::Ready) { uint8 connectionId = findConnectionId (TServiceId(i), nid); if (connectionId == 0xff) // failed { nlwarning ("HNETL5: Can't send message to %u because no connection available", i); continue; } _IdCnx[i].Connections[connectionId].CbNetBase->send (msgout, _IdCnx[i].Connections[connectionId].HostId); } } } /* Flush all the sending queues, and report the number of bytes still pending. * To ensure all data are sent before stopping a service, you may want to repeat * calling this method evenly until it returns 0. * \param namesOfOnlyServiceToFlushSending If not empty, only the send queues to the * services specified (by short name) will be flushed. */ uint CUnifiedNetwork::tryFlushAllQueues(const std::vector& namesOfOnlyServiceToFlushSending) { H_AUTO(L5FlushAll); uint bytesRemaining = 0; for (uint k = 0; k<_UsedConnection.size(); ++k) { H_AUTO(UNFABrowseConnections); CUnifiedConnection &uc = _IdCnx[_UsedConnection[k].get()]; // Skip the connection if it is not found in the 'only' list (except if the list is empty) if ( (! namesOfOnlyServiceToFlushSending.empty()) && (std::find( namesOfOnlyServiceToFlushSending.begin(), namesOfOnlyServiceToFlushSending.end(), uc.ServiceName ) == namesOfOnlyServiceToFlushSending.end()) ) continue; nlassert (uc.State == CUnifiedNetwork::CUnifiedConnection::Ready); for (uint j = 0; j < uc.Connections.size (); j++) { H_AUTO(UNFABrowseSubConnections); if (!uc.Connections[j].valid()) continue; if (uc.Connections[j].CbNetBase->connected ()) { uint bytesRemainingLocal; uc.Connections[j].CbNetBase->flush(uc.Connections[j].HostId, &bytesRemainingLocal); bytesRemaining += bytesRemainingLocal; } } } return bytesRemaining; } // // // void CUnifiedNetwork::addCallbackArray (const TUnifiedCallbackItem *callbackarray, sint arraysize) { uint i; for (i=0; i<(uint)arraysize; ++i) _Callbacks.insert(make_pair(string(callbackarray[i].Key),callbackarray[i].Callback)); } void CUnifiedNetwork::setServiceUpCallback (const string &serviceName, TUnifiedNetCallback cb, void *arg, bool back) { nlassert (cb != NULL); if (serviceName == "*") { if (back) _UpUniCallback.push_back (make_pair(cb, arg)); else _UpUniCallback.insert (_UpUniCallback.begin(), make_pair(cb, arg)); } else { if (back) _UpCallbacks[serviceName].push_back (make_pair(cb, arg)); else _UpCallbacks[serviceName].insert (_UpCallbacks[serviceName].begin(), make_pair(cb, arg)); } } void CUnifiedNetwork::removeServiceUpCallback (const string &serviceName, TUnifiedNetCallback cb, void *arg) { if (serviceName == "*") { uint i; for (i=0; i<_UpUniCallback.size(); ++i) { if (_UpUniCallback[i].first == cb && _UpUniCallback[i].second == arg) { // we found it _UpUniCallback.erase(_UpUniCallback.begin()+i); break; } } if (i == _UpUniCallback.size()) { nlwarning("HNETL5 : can't remove service up callback, not found"); } } else { if (_UpCallbacks.find(serviceName) != _UpCallbacks.end()) { std::list &list = _UpCallbacks[serviceName]; std::list::iterator first(list.begin()), last(list.end()); for (; first != last; ++first) { if (first->first == cb && first->second == arg) { list.erase(first); break; } } if (first == last) { nlwarning("HNETL5 : can't remove service up callback, not found"); } if (list.empty()) { // no more callback for this service _UpCallbacks.erase(serviceName); } } else { nlwarning("HNETL5 : can't remove service up callback, not found"); } } } void CUnifiedNetwork::setServiceDownCallback (const string &serviceName, TUnifiedNetCallback cb, void *arg, bool back) { nlassert (cb != NULL); if (serviceName == "*") { if (back) _DownUniCallback.push_back (make_pair(cb, arg)); else _DownUniCallback.insert (_DownUniCallback.begin(), make_pair(cb, arg)); } else { if (back) _DownCallbacks[serviceName].push_back (make_pair(cb, arg)); else _DownCallbacks[serviceName].insert (_DownCallbacks[serviceName].begin(), make_pair(cb, arg)); } } void CUnifiedNetwork::removeServiceDownCallback (const string &serviceName, TUnifiedNetCallback cb, void *arg) { if (serviceName == "*") { uint i; for (i=0; i<_DownUniCallback.size(); ++i) { if (_DownUniCallback[i].first == cb && _DownUniCallback[i].second == arg) { // we found it _DownUniCallback.erase(_DownUniCallback.begin()+i); break; } } if (i == _DownUniCallback.size()) { nlwarning("HNETL5 : can't remove service down callback, not found"); } } else { if (_DownCallbacks.find(serviceName) != _DownCallbacks.end()) { std::list &list = _DownCallbacks[serviceName]; std::list::iterator first(list.begin()), last(list.end()); for (; first != last; ++first) { if (first->first == cb && first->second == arg) { list.erase(first); break; } } if (first == last) { nlwarning("HNETL5 : can't remove service down callback, not found"); } if (list.empty()) { // no more callback for this service _DownCallbacks.erase(serviceName); } } else { nlwarning("HNETL5 : can't remove service down callback, not found"); } } } // // // uint64 CUnifiedNetwork::getBytesSent () { uint64 sent = 0; uint j; for (vector::iterator it = _UsedConnection.begin (); it != _UsedConnection.end(); it++) { if (_IdCnx[it->get()].State == CUnifiedNetwork::CUnifiedConnection::Ready) for (j=0; j<_IdCnx[it->get()].Connections.size (); ++j) if(_IdCnx[it->get()].Connections[j].valid () && !_IdCnx[it->get()].Connections[j].IsServerConnection) sent += _IdCnx[it->get()].Connections[j].CbNetBase->getBytesSent(); } if(_CbServer) sent += _CbServer->getBytesSent(); return sent; } uint64 CUnifiedNetwork::getBytesReceived () { uint64 received = 0; uint j; for (vector::iterator it = _UsedConnection.begin (); it != _UsedConnection.end(); it++) { if (_IdCnx[it->get()].State == CUnifiedNetwork::CUnifiedConnection::Ready) for (j=0; j<_IdCnx[it->get()].Connections.size (); ++j) if(_IdCnx[it->get()].Connections[j].valid () && !_IdCnx[it->get()].Connections[j].IsServerConnection) received += _IdCnx[it->get()].Connections[j].CbNetBase->getBytesReceived(); } if (_CbServer) received += _CbServer->getBytesReceived(); return received; } uint64 CUnifiedNetwork::getSendQueueSize () { uint64 sent = 0; uint j; for (vector::iterator it = _UsedConnection.begin (); it != _UsedConnection.end(); it++) { if (_IdCnx[it->get()].State == CUnifiedNetwork::CUnifiedConnection::Ready) for (j=0; j<_IdCnx[it->get()].Connections.size (); ++j) if(_IdCnx[it->get()].Connections[j].valid () && !_IdCnx[it->get()].Connections[j].IsServerConnection) sent += _IdCnx[it->get()].Connections[j].CbNetBase->getSendQueueSize(); } if (_CbServer) sent += _CbServer->getSendQueueSize(); return sent; } uint64 CUnifiedNetwork::getReceiveQueueSize () { uint64 received = 0; uint j; for (vector::iterator it = _UsedConnection.begin (); it != _UsedConnection.end(); it++) { if (_IdCnx[it->get()].State == CUnifiedNetwork::CUnifiedConnection::Ready) for (j=0; j<_IdCnx[it->get()].Connections.size (); ++j) if(_IdCnx[it->get()].Connections[j].valid () && !_IdCnx[it->get()].Connections[j].IsServerConnection) received += _IdCnx[it->get()].Connections[j].CbNetBase->getReceiveQueueSize(); } if (_CbServer) received += _CbServer->getReceiveQueueSize(); return received; } CCallbackNetBase *CUnifiedNetwork::getNetBase(const std::string &name, TSockId &host, uint8 nid) { nlassertex(_Initialised == true, ("Try to CUnifiedNetwork::getNetBase() whereas it is not initialised yet")); if (ThreadCreator != NLMISC::getThreadId()) nlwarning ("HNETL5: Multithread access but this class is not thread safe thread creator = %u thread used = %u", ThreadCreator, NLMISC::getThreadId()); sint count = (sint)_NamedCnx.count(name); if (count <= 0) { nlwarning ("HNETL5: couldn't access the service %s", name.c_str()); host = InvalidSockId; return NULL; } else if (count > 1) { nlwarning ("HNETL5: %d services %s to getNetBase, returns the first valid", count, name.c_str()); } TNameMappedConnection::const_iterator itnmc = _NamedCnx.find(name); uint8 connectionId = findConnectionId ((*itnmc).second, nid); if (connectionId == 0xff) // failed { nlwarning ("HNETL5: Can't getNetBase %s because no connection available", name.c_str()); host = InvalidSockId; return NULL; } host = _IdCnx[itnmc->second.get()].Connections[connectionId].HostId; return _IdCnx[itnmc->second.get()].Connections[connectionId].CbNetBase; } CCallbackNetBase *CUnifiedNetwork::getNetBase(TServiceId sid, TSockId &host, uint8 nid) { nlassertex(_Initialised == true, ("Try to CUnifiedNetwork::getNetBase() whereas it is not initialised yet")); if (ThreadCreator != NLMISC::getThreadId()) nlwarning ("HNETL5: Multithread access but this class is not thread safe thread creator = %u thread used = %u", ThreadCreator, NLMISC::getThreadId()); if (sid.get() >= _IdCnx.size () || _IdCnx[sid.get()].State != CUnifiedNetwork::CUnifiedConnection::Ready) { nlwarning ("HNETL5: Can't get net base to the service '%hu' because not in _IdCnx", sid.get()); host = InvalidSockId; return NULL; } uint8 connectionId = findConnectionId (sid, nid); if (connectionId == 0xff) // failed { nlwarning ("HNETL5: Can't getNetBase %hu because no connection available", sid.get()); host = InvalidSockId; return NULL; } host = _IdCnx[sid.get()].Connections[connectionId].HostId; return _IdCnx[sid.get()].Connections[connectionId].CbNetBase; } TUnifiedMsgCallback CUnifiedNetwork::findCallback (const std::string &callbackName) { TMsgMappedCallback::iterator itcb = _Callbacks.find(callbackName); if (itcb == _Callbacks.end()) return NULL; else return (*itcb).second; } bool CUnifiedNetwork::isServiceLocal (const std::string &serviceName) { // it s me, of course we are local if (serviceName == _Name) return true; pair range; range = _NamedCnx.equal_range(serviceName); if (range.first != _NamedCnx.end()) { TServiceId sid = (*(range.first)).second; return isServiceLocal (sid); } return false; } bool CUnifiedNetwork::isServiceLocal (TServiceId sid) { // it s me, of course we are local if (sid == _SId) return true; if (sid.get() >= _IdCnx.size () || _IdCnx[sid.get()].State != CUnifiedNetwork::CUnifiedConnection::Ready) { return false; } vector laddr = CInetAddress::localAddresses(); for (uint i = 0; i < laddr.size(); i++) { for (uint j = 0; j < _IdCnx[sid.get()].ExtAddress.size(); j++) { if (_IdCnx[sid.get()].ExtAddress[j].is127001 ()) return true; if (_IdCnx[sid.get()].ExtAddress[j].internalIPAddress () == laddr[i].internalIPAddress ()) return true; } } return false; } /* * Return the name of the specified service, or "" if not found */ std::string CUnifiedNetwork::getServiceName(TServiceId sid) { string s; CUnifiedConnection *c = getUnifiedConnection(sid, false); if (c) s = c->ServiceName; return s; } /* * Return a string identifying the service, using the format "NAME-sid" (or "sid" only if not found) */ std::string CUnifiedNetwork::getServiceUnifiedName(TServiceId sid) { string s; CUnifiedConnection *c = getUnifiedConnection(sid, false); if (c) s = c->ServiceName + "-"; s += toString(sid.get()); return s; } // // // //CUnifiedNetwork *CUnifiedNetwork::_Instance = NULL; NLMISC_SAFE_SINGLETON_IMPL(CUnifiedNetwork); bool CUnifiedNetwork::isUsed () { return (_Instance != NULL); } // // // CUnifiedNetwork::CUnifiedConnection *CUnifiedNetwork::getUnifiedConnection (TServiceId sid, bool warn) { if (sid.get() < _IdCnx.size () && _IdCnx[sid.get()].State == CUnifiedConnection::Ready) { if (sid != _IdCnx[sid.get()].ServiceId) { AUTOCHECK_DISPLAY ("HNETL5: Sid index %hu is not the same that in the entry %hu", sid.get(), _IdCnx[sid.get()].ServiceId.get()); return NULL; } return &_IdCnx[sid.get()]; } else { if ( warn ) nlwarning ("HNETL5: Try to get a bad unified connection (sid %hu is not in the table)", sid.get()); return NULL; } } void CUnifiedNetwork::autoCheck() { H_AUTO(L5UpdateAutoCheck); uint i, j; for (i = 0; i < _IdCnx.size (); i++) { if (_IdCnx[i].State == CUnifiedNetwork::CUnifiedConnection::Ready) { _IdCnx[i].AutoCheck = 1; } else { _IdCnx[i].AutoCheck = 0; } } TNameMappedConnection::iterator itn; for (itn = _NamedCnx.begin(); itn != _NamedCnx.end(); ++itn) { if ((*itn).first != _IdCnx[itn->second.get()].ServiceName) AUTOCHECK_DISPLAY ("HLNET5: problem with name synchro between _NameCnx '%s' and _IdCnx '%s' '%hd'", (*itn).first.c_str(), _IdCnx[itn->second.get()].ServiceName.c_str (), itn->second.get()); if (_IdCnx[itn->second.get()].AutoCheck == 0) AUTOCHECK_DISPLAY ("HLNET5: problem with name synchro between _NameCnx '%s' and _IdCnx '%s' '%hd'", (*itn).first.c_str(), _IdCnx[itn->second.get()].ServiceName.c_str (), itn->second.get()); if (_IdCnx[itn->second.get()].AutoCheck > 1) AUTOCHECK_DISPLAY ("HLNET5: problem with name synchro between _NameCnx '%s' and _IdCnx '%s' '%hd' more than one entry is named with the same name", (*itn).first.c_str(), _IdCnx[itn->second.get()].ServiceName.c_str (), itn->second.get()); _IdCnx[itn->second.get()].AutoCheck++; } for (i = 0; i < _UsedConnection.size (); i++) { if (_IdCnx[_UsedConnection[i].get()].State != CUnifiedNetwork::CUnifiedConnection::Ready) AUTOCHECK_DISPLAY ("HLNET5: problem with the _UsedConnection syncro sid %d is not used in _IdCnx", _UsedConnection[i].get()); } for (i = 0; i < _IdCnx.size (); i++) { if (_IdCnx[i].State == CUnifiedNetwork::CUnifiedConnection::Ready) { for (j = 0; j < _UsedConnection.size (); j++) { if (_UsedConnection[j].get() == i) break; } if (j == _UsedConnection.size ()) AUTOCHECK_DISPLAY ("HLNET5: problem with the _UsedConnection syncro sid %d is not in _UsedConnection", i); } } for (i = 0; i < _IdCnx.size (); i++) { if (_IdCnx[i].State == CUnifiedNetwork::CUnifiedConnection::NotUsed) { if (_IdCnx[i].ServiceName != "DEAD") AUTOCHECK_DISPLAY ("HLNET5: sid %d name should be DEAD and is '%s'", i, _IdCnx[i].ServiceName.c_str ()); if (_IdCnx[i].ServiceId.get() != 0xDEAD) AUTOCHECK_DISPLAY ("HLNET5: sid %d sid should be 0xDEAD and is 0x%X", i, _IdCnx[i].ServiceId.get()); if (!_IdCnx[i].Connections.empty ()) AUTOCHECK_DISPLAY ("HLNET5: sid %d connection size should be 0 and is %d", i, _IdCnx[i].Connections.size ()); if (!_IdCnx[i].ExtAddress.empty ()) AUTOCHECK_DISPLAY ("HLNET5: sid %d ext addr size should be 0 and is %d", i, _IdCnx[i].ExtAddress.size ()); if (_IdCnx[i].AutoCheck != 0) AUTOCHECK_DISPLAY ("HLNET5: sid %d prob with syncro with _NamedCnx", i); } else if (_IdCnx[i].State == CUnifiedNetwork::CUnifiedConnection::Ready) { if (_IdCnx[i].ServiceId.get() != i) AUTOCHECK_DISPLAY ("HNETL5: Bad syncro sid index sid entry for %d %d", i, _IdCnx[i].ServiceId.get()); if (_IdCnx[i].ServiceName == "DEAD") AUTOCHECK_DISPLAY ("HLNET5: sid %d name should not be DEAD and is '%s'", i, _IdCnx[i].ServiceName.c_str ()); if (_IdCnx[i].ServiceId.get() == 0xDEAD) AUTOCHECK_DISPLAY ("HLNET5: sid %d sid should not be 0xDEAD and is 0x%X", i, _IdCnx[i].ServiceId.get()); if (!_IdCnx[i].ExtAddress.empty () && _IdCnx[i].Connections.size () > _IdCnx[i].ExtAddress.size()) AUTOCHECK_DISPLAY ("HLNET5: sid %d ext addr size should not be 0 and is %d", i, _IdCnx[i].ExtAddress.size ()); if (_IdCnx[i].AutoRetry == true && _IdCnx[i].Connections.size () > 1) AUTOCHECK_DISPLAY ("HLNET5: sid %d auto retry with more than one connection %d", i, _IdCnx[i].Connections.size ()); if (_IdCnx[i].AutoRetry == true && _IdCnx[i].IsExternal == false) AUTOCHECK_DISPLAY ("HLNET5: sid %d auto retry with internal connection", i); if (_IdCnx[i].AutoRetry == true && _IdCnx[i].Connections[0].valid() == false) AUTOCHECK_DISPLAY ("HLNET5: sid %d auto retry with invalid connection", i); for (j = 0; j < _IdCnx[i].Connections.size (); j++) { if (_IdCnx[i].Connections[j].valid() && !_IdCnx[i].Connections[j].IsServerConnection && _IdCnx[i].Connections[j].CbNetBase->connected () && _IdCnx[i].Connections[j].getAppId() != i) AUTOCHECK_DISPLAY ("HLNET5: sid %d bad appid %"NL_I64"X", i, _IdCnx[i].Connections[j].getAppId()); } for (j = 0; j < _IdCnx[i].NetworkConnectionAssociations.size (); j++) { if (_IdCnx[i].NetworkConnectionAssociations[j] != 0) { if (_NetworkAssociations[j] != _IdCnx[i].ExtAddress[_IdCnx[i].NetworkConnectionAssociations[j]].internalNetAddress ()) AUTOCHECK_DISPLAY ("HLNET5: sid %d nid %d have address 0x%08x and is not the good connection net 0x%08x", i, j, _NetworkAssociations[j], _IdCnx[i].ExtAddress[_IdCnx[i].NetworkConnectionAssociations[j]].internalNetAddress ()); } } } } } void CUnifiedNetwork::displayInternalTables (NLMISC::CLog *log) { uint i, j; log->displayNL ("%d Named Connections:", _NamedCnx.size ()); for (TNameMappedConnection::iterator it = _NamedCnx.begin(); it != _NamedCnx.end (); it++) { log->displayNL ("> '%s' -> %hu", (*it).first.c_str(), it->second.get()); } uint nbused = 0; for (i = 0; i < _IdCnx.size (); i++) { if(_IdCnx[i].State != CUnifiedNetwork::CUnifiedConnection::NotUsed) nbused++; } log->displayNL ("%u/%u Unified Connections:", nbused, _IdCnx.size ()); for (i = 0; i < _IdCnx.size (); i++) { if(_IdCnx[i].State != CUnifiedNetwork::CUnifiedConnection::NotUsed) { _IdCnx[i].display (false, log); for (j = 0; j < _IdCnx[i].NetworkConnectionAssociations.size (); j++) { log->displayNL (" * nid %d -> cnxn %hu", j, (uint16)_IdCnx[i].NetworkConnectionAssociations[j]); } } } log->displayNL ("%u Used Unified Connections:", _UsedConnection.size()); for (i = 0; i < _UsedConnection.size (); i++) { log->displayNL ("> %hu", _UsedConnection[i].get()); } log->displayNL ("%u Network Associations:", _NetworkAssociations.size()); for (i = 0; i < _NetworkAssociations.size (); i++) { log->displayNL ("> 0x%08x -> '%s'", _NetworkAssociations[i], internalIPAddressToString (_NetworkAssociations[i]).c_str ()); } } bool CUnifiedNetwork::haveNamedCnx (const std::string &name, TServiceId sid) { CUnifiedNetwork::TNameMappedConnection::iterator it; pair range; range = _NamedCnx.equal_range(name); if (range.first != range.second) { for (it=range.first; it!=range.second && (*it).second!=sid; ++it) ; return (it != range.second); } return false; } void CUnifiedNetwork::addNamedCnx (const std::string &name, TServiceId sid) { // check if not already inserted CUnifiedNetwork::TNameMappedConnection::iterator it; pair range; range = _NamedCnx.equal_range(name); if (range.first != range.second) { for (it=range.first; it!=range.second && (*it).second!=sid; ++it) ; if (it != range.second) { AUTOCHECK_DISPLAY ("HNETL5: Try to add 2 times the same connection %s-%hu", name.c_str(), sid.get()); return; } } // insert the name in the map to be able to send message with the name _NamedCnx.insert(make_pair(name, sid)); } void CUnifiedNetwork::removeNamedCnx (const std::string &name, TServiceId sid) { // get all map nodes of that service name CUnifiedNetwork::TNameMappedConnection::iterator it; pair range; range = _NamedCnx.equal_range(name); // assume not empty if (range.first == range.second) { AUTOCHECK_DISPLAY ("HNETL5: The unified connection %s-%hu wasn't on the _NamedCnx", name.c_str(), sid.get()); return; } // select good service id for (it=range.first; it!=range.second && (*it).second!=sid; ++it) ; // assume id exists if (it == range.second) { AUTOCHECK_DISPLAY ("HNETL5: The unified connection %s-%hu wasn't on the _NamedCnx", name.c_str(), sid.get()); return; } // remove service for map _NamedCnx.erase(it); } void CUnifiedNetwork::addNetworkAssociation (const string &networkName, uint8 nid) { if (nid >= _NetworkAssociations.size ()) _NetworkAssociations.resize (nid+1, 0xFF); _NetworkAssociations[nid] = stringToInternalIPAddress (networkName); nlinfo ("HNETL5: Associate network '%s' 0x%08x '%s' to nid %hu", networkName.c_str(), _NetworkAssociations[nid], internalIPAddressToString (_NetworkAssociations[nid]).c_str(), (uint16)nid); } void CUnifiedNetwork::callServiceUpCallback (const std::string &serviceName, TServiceId sid, bool callGlobalCallback) { // now we warn the user CUnifiedNetwork::TNameMappedCallback::iterator it = _UpCallbacks.find(serviceName); if (it != _UpCallbacks.end()) { // call it for (list::iterator it2 = (*it).second.begin(); it2 != (*it).second.end(); it2++) { TUnifiedNetCallback cb = (*it2).first; if (cb) cb(serviceName, sid, (*it2).second); else nlwarning ("HNETL5: User set an empty callback for '%s' service up", serviceName.c_str()); } } if(callGlobalCallback) { for (uint c = 0; c < _UpUniCallback.size (); c++) { if (_UpUniCallback[c].first != NULL) _UpUniCallback[c].first (serviceName, sid, _UpUniCallback[c].second); else nlwarning ("HNETL5: User set an empty callback for '*' service up"); } } } void CUnifiedNetwork::callServiceDownCallback (const std::string &serviceName, TServiceId sid, bool callGlobalCallback) { // now we warn the user CUnifiedNetwork::TNameMappedCallback::iterator it = _DownCallbacks.find(serviceName); if (it != _DownCallbacks.end()) { // call it for (list::iterator it2 = (*it).second.begin(); it2 != (*it).second.end(); it2++) { TUnifiedNetCallback cb = (*it2).first; if (cb) cb(serviceName, sid, (*it2).second); else nlwarning ("HNETL5: User set an empty callback for '%s' service down", serviceName.c_str()); } } if(callGlobalCallback) { for (uint c = 0; c < _DownUniCallback.size (); c++) { if (_DownUniCallback[c].first != NULL) _DownUniCallback[c].first (serviceName, sid, _DownUniCallback[c].second); else nlwarning ("HNETL5: User set an empty callback for '*' service down"); } } } void CUnifiedNetwork::CUnifiedConnection::display (bool full, CLog *log) { log->displayNL ("> %s-%hu %s %s %s (%d ExtAddr %d Cnx) TotalCb %d", ServiceName.c_str (), ServiceId.get(), IsExternal?"External":"NotExternal", AutoRetry?"AutoRetry":"NoAutoRetry", SendId?"SendId":"NoSendId", ExtAddress.size (), Connections.size (), TotalCallbackCalled); uint maxc = (uint)std::max (ExtAddress.size (), Connections.size ()); for (uint j = 0; j < maxc; j++) { string base; if(j < ExtAddress.size ()) { base += ExtAddress[j].asString (); } else { base += "NotValid"; } string ext; if(j < Connections.size () && Connections[j].valid()) { if(Connections[j].IsServerConnection) { ext += "Server "; } else { ext += "Client "; } ext += Connections[j].CbNetBase->getSockId (Connections[j].HostId)->asString (); ext += " AppId:" + toString(Connections[j].getAppId()); if (Connections[j].CbNetBase->connected ()) ext += " Connected"; else ext += " NotConnected"; } else { ext += "NotValid"; } log->displayNL (" - %s %s", base.c_str (), ext.c_str ()); if(full) { log->displayNL (" * ReceiveQueueStat"); Connections[j].CbNetBase->displayReceiveQueueStat(log); log->displayNL (" * SendQueueStat"); Connections[j].CbNetBase->displaySendQueueStat(log, Connections[j].HostId); log->displayNL (" * ThreadStat"); Connections[j].CbNetBase->displayThreadStat(log); } } } // // Commands // bool createMessage (CMessage &msgout, const vector &args, CLog &log) { for (uint i = 2; i < args.size (); i+=2) { string type = args[i+0]; string value = args[i+1]; if (type == "s8") { sint8 v; fromString(value, v); msgout.serial (v); } else if (type == "s16") { sint16 v; fromString(value, v); msgout.serial (v); } else if (type == "s32") { sint32 v; fromString(value, v); msgout.serial (v); } else if (type == "s64") { sint64 v; fromString(value, v); msgout.serial (v); } else if (type == "u8") { uint8 v; fromString(value, v); msgout.serial (v); } else if (type == "u16") { uint16 v; fromString(value, v); msgout.serial (v); } else if (type == "u32") { uint32 v; fromString(value, v); msgout.serial (v); } else if (type == "u64") { uint64 v; fromString(value, v); msgout.serial (v); } else if (type == "f") { float v; fromString(value, v); msgout.serial (v); } else if (type == "d") { double v; fromString(value, v); msgout.serial (v); } else if (type == "b") { bool v; fromString(value, v); msgout.serial (v); } else if (type == "s") { msgout.serial (value); } else if (type == "e") { CEntityId e; e.fromString(value.c_str()); msgout.serial(e); } else { log.displayNL ("type '%s' is not a valid type", type.c_str()); return false; } } return true; } // // Commands and Variables // NLMISC_CATEGORISED_VARIABLE(nel, uint32, TotalCallbackCalled, "Total callback called number on layer 5"); NLMISC_CATEGORISED_DYNVARIABLE(nel, uint64, SendQueueSize, "current size in bytes of all send queues") { nlunreferenced(human); if (get) { if (!CUnifiedNetwork::isUsed ()) *pointer = 0; else *pointer = CUnifiedNetwork::getInstance()->getSendQueueSize(); } } NLMISC_CATEGORISED_DYNVARIABLE(nel, uint64, ReceiveQueueSize, "current size in bytes of all receive queues") { nlunreferenced(human); if (get) { if (!CUnifiedNetwork::isUsed ()) *pointer = 0; else *pointer = CUnifiedNetwork::getInstance()->getReceiveQueueSize(); } } NLMISC_CATEGORISED_DYNVARIABLE(nel, uint64, ReceivedBytes, "total of bytes received by this service") { nlunreferenced(human); if (get) { if (!CUnifiedNetwork::isUsed ()) *pointer = 0; else *pointer = CUnifiedNetwork::getInstance()->getBytesReceived (); } } NLMISC_CATEGORISED_DYNVARIABLE(nel, uint64, SentBytes, "total of bytes sent by this service") { nlunreferenced(human); if (get) { if (!CUnifiedNetwork::isUsed ()) *pointer = 0; else *pointer = CUnifiedNetwork::getInstance()->getBytesSent (); } } /* * Simulate a message that comes from the network. * * for the bool (b type), you must set the value to 1 or 0 * for the string (s type), we don't manage space inside a string * for stl containers, you have first to put a u32 type that is the size of the container and after all elements * (ex: if you want to put a vector that have 3 elements: u32 3 u16 10 u16 11 u16 12) * * ex: msgin 128 REGISTER u32 10 u32 541 u32 45 * You'll receive a fake message REGISTER that seems to come from the service number 128 with 3 uint32. * */ NLMISC_CATEGORISED_COMMAND(nel, msgin, "Simulate an input message from another service (ex: msgin 128 REGISTER u32 10 b 1 f 1.5)", "| [ ]*") { nlunreferenced(rawCommandString); nlunreferenced(quiet); nlunreferenced(human); if(args.size() < 2) return false; if (!CUnifiedNetwork::isUsed ()) { log.displayNL("Can't do that because the service doesn't use CUnifiedNetwork"); return false; } uint16 sId; fromString(args[0], sId); TServiceId serviceId(sId); string serviceName = args[0].c_str(); string messageName = args[1].c_str(); if (serviceId.get() > 255) { log.displayNL ("Service Id %d must be between [1;255]", serviceId.get()); return false; } if ((args.size()-2) % 2 != 0) { log.displayNL ("The number of parameter must be a multiple of 2"); return false; } CMessage msg (messageName); if (!createMessage (msg, args, log)) return false; msg.invert (); TUnifiedMsgCallback cb = CUnifiedNetwork::getInstance()->findCallback (messageName); if (cb == NULL) { log.displayNL ("Callback for message '%s' is not found", messageName.c_str()); } else { cb (msg, serviceName, serviceId); } return true; } /* * Create a message and send it to the specified service * * for the bool (b type), you must set the value to 1 or 0 * for the string (s type), we don't manage space inside a string * for stl containers, you have first to put a u32 type that is the size of the container and after all elements * (ex: if you want to put a vector that have 3 elements: u32 3 u16 10 u16 11 u16 12) * * ex: msgout 128 REGISTER u32 10 u32 541 u32 45 * You'll send a real message REGISTER to the service number 128 with 3 uint32. * */ NLMISC_CATEGORISED_COMMAND(nel, msgout, "Send a message to a specified service (ex: msgout 128 REGISTER u32 10 b 1 f 1.5)", "| [ ]*") { nlunreferenced(rawCommandString); nlunreferenced(quiet); nlunreferenced(human); if(args.size() < 2) return false; if (!CUnifiedNetwork::isUsed ()) { log.displayNL("Can't do that because the service doesn't use CUnifiedNetwork"); return false; } uint16 nId; fromString(args[0], nId); TServiceId serviceId(nId); string serviceName = args[0].c_str(); string messageName = args[1].c_str(); if (serviceId.get() > 255) { log.displayNL ("Service Id %d must be between [1;255]", serviceId.get()); return false; } if ((args.size()-2) % 2 != 0) { log.displayNL ("The number of parameter must be a multiple of 2"); return false; } CMessage msg (messageName); if (!createMessage (msg, args, log)) return false; TSockId host = InvalidSockId; CCallbackNetBase *cnb = NULL; if (serviceId.get() != 0) cnb = CUnifiedNetwork::getInstance()->getNetBase (serviceId, host); else cnb = CUnifiedNetwork::getInstance()->getNetBase (serviceName, host); if (cnb == NULL) { log.displayNL ("'%s' is a bad or ", args[0].c_str()); return false; } cnb->send (msg, host); return true; } NLMISC_CATEGORISED_COMMAND(nel, l5QueuesStats, "Displays queues stats of network layer5", "") { nlunreferenced(rawCommandString); nlunreferenced(quiet); nlunreferenced(human); if(args.size() != 0) return false; if (!CUnifiedNetwork::isUsed ()) { log.displayNL("Can't display internal table because layer5 is not used"); return false; } log.displayNL ("%u Unified Connections:", CUnifiedNetwork::getInstance()->_IdCnx.size ()); for (uint i = 0; i < CUnifiedNetwork::getInstance()->_IdCnx.size (); i++) { if(CUnifiedNetwork::getInstance()->_IdCnx[i].State != CUnifiedNetwork::CUnifiedConnection::NotUsed) { CUnifiedNetwork::getInstance()->_IdCnx[i].display (true, &log); } } return true; } NLMISC_CATEGORISED_COMMAND(nel, l5InternalTables, "Displays internal table of network layer5", "") { nlunreferenced(rawCommandString); nlunreferenced(quiet); nlunreferenced(human); if(args.size() != 0) return false; if (!CUnifiedNetwork::isUsed ()) { log.displayNL("Can't display internal table because layer5 is not used"); return false; } CUnifiedNetwork::getInstance ()->displayInternalTables(&log); return true; } NLMISC_CATEGORISED_COMMAND(nel, l5Callback, "Displays all callback registered in layer5", "") { nlunreferenced(rawCommandString); nlunreferenced(quiet); nlunreferenced(human); if(args.size() != 0) return false; if (!CUnifiedNetwork::isUsed ()) { log.displayNL("Can't display internal table because layer5 is not used"); return false; } log.displayNL ("There're %d registered callbacks:", CUnifiedNetwork::getInstance()->_Callbacks.size()); uint i = 0; for (CUnifiedNetwork::TMsgMappedCallback::iterator it = CUnifiedNetwork::getInstance()->_Callbacks.begin(); it != CUnifiedNetwork::getInstance()->_Callbacks.end(); it++) { log.displayNL (" %d '%s' %s", i++, (*it).first.c_str(), ((*it).second == NULL?"have a NULL address":"")); } return true; } NLMISC_CATEGORISED_COMMAND(nel, isServiceLocal, "Says if a service is local or not compare with this service", "|") { nlunreferenced(human); nlunreferenced(quiet); nlunreferenced(rawCommandString); if(args.size() != 1) return false; if (!CUnifiedNetwork::isUsed ()) { log.displayNL("Can't do that because the service doesn't use CUnifiedNetwork"); return false; } uint16 nId; fromString(args[0], nId); TServiceId sid(nId); if (sid.get() > 0) { log.displayNL ("Service %s-%hu and sid %s are %son the same computer", CUnifiedNetwork::getInstance ()->_Name.c_str(), CUnifiedNetwork::getInstance ()->_SId.get(), args[0].c_str(), CUnifiedNetwork::getInstance ()->isServiceLocal (sid)?"":"not "); } else { log.displayNL ("Service %s-%hu and %s are %son the same computer", CUnifiedNetwork::getInstance ()->_Name.c_str(), CUnifiedNetwork::getInstance ()->_SId.get(), args[0].c_str(), CUnifiedNetwork::getInstance ()->isServiceLocal (args[0])?"":"not "); } return true; } NLMISC_CLASS_COMMAND_IMPL(CUnifiedNetwork, addService) { nlunreferenced(human); nlunreferenced(quiet); nlunreferenced(args); TParsedCommandLine pcl; pcl.parseParamList(rawCommandString); if (pcl.SubParams.size() != 2) return false; // syntax is as follow : // ( address= [sid=] [sendId] [external] [autoRetry] ) TParsedCommandLine * serviceInfo = pcl.SubParams[1]; const TParsedCommandLine *address = serviceInfo->getParam("address"); if (address == NULL) { log.displayNL("Can't find param 'address'"); return false; } CInetAddress ia(address->ParamValue);; if (!ia.isValid()) { log.displayNL("Can't parse internet address in '%s'", address->ParamValue.c_str()); return false; } TServiceId serviceId(0); const TParsedCommandLine *sid = serviceInfo->getParam("sid"); if (sid != NULL) { uint16 nId; fromString(sid->ParamValue, nId); serviceId.set(nId); } bool sendId = serviceInfo->getParam("sendId") != NULL; bool external = serviceInfo->getParam("external") != NULL; bool autoRetry = serviceInfo->getParam("autoRetry") != NULL; log.displayNL("Adding service '%s' as sid %u with [sendId = %s], [external = %s], [autoRetry = %s]", serviceInfo->ParamName.c_str(), sid, sendId ? "YES" : "NO", external ? "YES" : "NO", autoRetry ? "YES" : "NO" ); addService(serviceInfo->ParamName, ia, sendId, external, serviceId, autoRetry, false); return true; } } // NLNET