X-Git-Url: https://git.teslayout.com/public/public/public/?a=blobdiff_plain;f=source%2Fdiameter.comm%2FEngine.cpp;h=56aa1b64f0123904d05a2bc810e54e63bf81d805;hb=d723d5bf571eb48c641b092058eaa38bb6c4fcc8;hp=53de6ac0a51e1570a24a235e0f131db6bf13ae23;hpb=129500a50678c43ff28fb0054d6197899b8c0b2c;p=anna.git diff --git a/source/diameter.comm/Engine.cpp b/source/diameter.comm/Engine.cpp index 53de6ac..56aa1b6 100644 --- a/source/diameter.comm/Engine.cpp +++ b/source/diameter.comm/Engine.cpp @@ -5,16 +5,17 @@ // See project site at http://redmine.teslayout.com/projects/anna-suite // // See accompanying file LICENSE or copy at http://www.teslayout.com/projects/public/anna.LICENSE // +// Standard +#include // rand() -#include +#include #include #include #include #include #include #include - #include #include #include @@ -29,15 +30,24 @@ #include #include #include +#include // STD #include using namespace std; -using namespace anna::diameter::comm; +using namespace anna::diameter; -Engine::Engine(const char *className) : +namespace anna { + namespace diameter { + namespace stack { + class Dictionary; + } + } +} + +comm::Engine::Engine(const char *className, const stack::Dictionary *baseProtocolDictionary) : anna::app::Component(className), a_autoBind(true), a_availableForEntities(false), @@ -48,19 +58,34 @@ Engine::Engine(const char *className) : // a_dwa(true), a_watchdogPeriod(ClientSession::DefaultWatchdogPeriod), a_maxConnectionDelay(anna::comm::ClientSocket::DefaultMaxConnectionDelay /* 200 ms*/), - a_numberOfClientSessionsPerServer(1) { + a_numberOfClientSessionsPerServer(1), + a_baseProtocolCodecEngine((std::string("baseProtocolCodecEngine_for_") + std::string(className)).c_str(), baseProtocolDictionary) +{ anna::diameter::sccs::activate(); - a_realm = anna::functions::getDomainname(); - a_host = anna::functions::getHostname(); + a_originRealm = anna::functions::getDomainname(); + a_originHost = anna::functions::getHostname(); + a_ceaPathfile = ""; + + // Internal base protocol codec engine: + a_baseProtocolCodecEngine.setValidationMode(anna::diameter::codec::Engine::ValidationMode::Always); // default was: after decoding } -Server* Engine::allocateServer() throw() { return a_serversRecycler.create(); } -void Engine::releaseServer(Server *server) throw() { a_serversRecycler.release(server); } -ClientSession* Engine::allocateClientSession() throw() { return a_clientSessionsRecycler.create(); } -void Engine::releaseClientSession(ClientSession *clientSession) throw() { a_clientSessionsRecycler.release(clientSession); } + +void comm::Engine::assertBaseProtocolHealth() throw(anna::RuntimeException) { + if (!getBaseProtocolCodecEngine()->getDictionary()) + throw anna::RuntimeException("Invalid diameter::comm::Engine object: base protocol dictionary provided on constructor was NULL", ANNA_FILE_LOCATION); + // it would be interesting to check and identify certain base protocol elements in the dictionary ... + // but these things will be checked in runtime and will fail if they should. +} -void Engine::setCERandDWR(const anna::DataBlock & cer, const anna::DataBlock & dwr) throw(anna::RuntimeException) { +comm::Server* comm::Engine::allocateServer() throw() { return a_serversRecycler.create(); } +void comm::Engine::releaseServer(Server *server) throw() { a_serversRecycler.release(server); } +comm::ClientSession* comm::Engine::allocateClientSession() throw() { return a_clientSessionsRecycler.create(); } +void comm::Engine::releaseClientSession(ClientSession *clientSession) throw() { a_clientSessionsRecycler.release(clientSession); } + + +void comm::Engine::setClientCERandDWR(const anna::DataBlock & cer, const anna::DataBlock & dwr) throw(anna::RuntimeException) { if(codec::functions::getCommandId(cer) != helpers::base::COMMANDID__Capabilities_Exchange_Request) { throw anna::RuntimeException("The message provided as 'CER' is not a Capabilities-Exchange-Request", ANNA_FILE_LOCATION); } @@ -73,20 +98,87 @@ void Engine::setCERandDWR(const anna::DataBlock & cer, const anna::DataBlock & d a_dwr = dwr; } -//void Engine::setCEAandDWA(const anna::DataBlock & cea, const anna::DataBlock & dwa) throw(anna::RuntimeException) { -// if (codec::functions::getCommandId(cea) != helpers::base::COMMANDID__Capabilities_Exchange_Answer) { -// throw anna::RuntimeException("The message provided as 'CEA' is not a Capabilities-Exchange-Answer", ANNA_FILE_LOCATION); -// } -// -// if (codec::functions::getCommandId(dwa) != helpers::base::COMMANDID__Device_Watchdog_Answer) { -// throw anna::RuntimeException("The message provided as 'DWA' is not a Device-Watchdog-Answer", ANNA_FILE_LOCATION); -// } -// -// a_cea = cea; -// a_dwa = dwa; -//} +void comm::Engine::setClientCERandDWR(const std::string & cer, const std::string & dwr) throw(anna::RuntimeException) { + + // Check for base protocol codec engine health: + assertBaseProtocolHealth(); + + // Build CER + // ::= < Diameter Header: 257, REQ > + // { Origin-Host } 264 diameterIdentity + // { Origin-Realm } 296 idem + // 1* { Host-IP-Address } 257, address + // { Vendor-Id } 266 Unsigned32 + // { Product-Name } 269 UTF8String + // [Origin-State-Id] 278 Unsigned32 + // * [ Supported-Vendor-Id ] 265 Unsigned32 + // * [ Auth-Application-Id ] 258 Unsigned32 + // * [Acct-Application-Id] 259 Unsigned32 + anna::diameter::codec::Message diameterCER(getBaseProtocolCodecEngine()); + int applicationId = 0 /*anna::diameter::helpers::APPID__3GPP_Rx*/; // Unsigned32 + std::string OH = getOriginHostName(); + std::string OR = getOriginRealmName(); + std::string hostIP = anna::functions::getHostnameIP(); // Address + int vendorId = anna::diameter::helpers::VENDORID__tgpp; // Unsigned32 + std::string productName = "ANNA Diameter Client"; // UTF8String + bool encodeDefault = false; + + if (cer != "") { + try { + diameterCER.loadXML(cer); + } catch(anna::RuntimeException &ex) { + //ex.trace(); + encodeDefault = true; + LOGWARNING(anna::Logger::warning("CER file not found or unable to parse. Encoding harcoded default version ...", ANNA_FILE_LOCATION)); + } + } + else { + encodeDefault = true; + } + + if(encodeDefault) { + diameterCER.setId(anna::diameter::helpers::base::COMMANDID__Capabilities_Exchange_Request); + diameterCER.setApplicationId(applicationId); + diameterCER.addAvp(anna::diameter::helpers::base::AVPID__Origin_Host)->getDiameterIdentity()->setValue(OH); + diameterCER.addAvp(anna::diameter::helpers::base::AVPID__Origin_Realm)->getDiameterIdentity()->setValue(OR); + diameterCER.addAvp(anna::diameter::helpers::base::AVPID__Host_IP_Address)->getAddress()->fromPrintableString(hostIP.c_str()); // supported by Address class, anyway is better to provide "1|" + diameterCER.addAvp(anna::diameter::helpers::base::AVPID__Vendor_Id)->getUnsigned32()->setValue(vendorId); + diameterCER.addAvp(anna::diameter::helpers::base::AVPID__Product_Name)->getUTF8String()->setValue(productName); + diameterCER.addAvp(anna::diameter::helpers::base::AVPID__Auth_Application_Id)->getUnsigned32()->setValue(applicationId); + } -void Engine::setWatchdogPeriod(const anna::Millisecond & wp) throw(anna::RuntimeException) { + // Build DWR + // ::= < Diameter Header: 280, REQ > + // { Origin-Host } + // { Origin-Realm } + anna::diameter::codec::Message diameterDWR(getBaseProtocolCodecEngine()); + encodeDefault = false; + + if (dwr != "") { + try { + diameterDWR.loadXML(dwr); + } catch(anna::RuntimeException &ex) { + //ex.trace(); + encodeDefault = true; + LOGWARNING(anna::Logger::warning("DWR file not found or unable to parse. Encoding harcoded default version ...", ANNA_FILE_LOCATION)); + } + } + else { + encodeDefault = true; + } + + if(encodeDefault) { + diameterDWR.setId(anna::diameter::helpers::base::COMMANDID__Device_Watchdog_Request); + diameterDWR.setApplicationId(applicationId); + diameterDWR.addAvp(anna::diameter::helpers::base::AVPID__Origin_Host)->getDiameterIdentity()->setValue(OH); + diameterDWR.addAvp(anna::diameter::helpers::base::AVPID__Origin_Realm)->getDiameterIdentity()->setValue(OR); + } + + // Assignment for internal encoded versions: + setClientCERandDWR(diameterCER.code(), diameterDWR.code()); +} + +void comm::Engine::setWatchdogPeriod(const anna::Millisecond & wp) throw(anna::RuntimeException) { if(wp < ClientSession::DefaultWatchdogPeriod) { throw anna::RuntimeException(anna::functions::asString("Please set watchdog period over %s", ClientSession::DefaultWatchdogPeriod.asString().c_str()), ANNA_FILE_LOCATION); } @@ -94,7 +186,7 @@ void Engine::setWatchdogPeriod(const anna::Millisecond & wp) throw(anna::Runtime a_watchdogPeriod = wp; } -void Engine::checkEntityCollision(const socket_v &v) throw(anna::RuntimeException) { +void comm::Engine::checkEntityCollision(const socket_v &v) throw(anna::RuntimeException) { socket_v::const_iterator it; socket_v::const_iterator it_min(v.begin()); socket_v::const_iterator it_max(v.end()); @@ -115,7 +207,7 @@ void Engine::checkEntityCollision(const socket_v &v) throw(anna::RuntimeExceptio throw anna::RuntimeException("diameter::comm::Engine::checkEntityCollision: Provided addresses list (sockets) must have all items different", ANNA_FILE_LOCATION); } -Entity* Engine::createEntity(const socket_v & socketList, const std::string &description) +comm::Entity* comm::Engine::createEntity(const socket_v & socketList, const std::string &description) throw(anna::RuntimeException) { Entity* result(NULL); anna::Guard guard(this, "anna::diameter::comm::Engine::createEntity"); @@ -164,7 +256,7 @@ throw(anna::RuntimeException) { } -LocalServer *Engine::createLocalServer(const std::string & addr, int port, int maxConnections, const anna::Millisecond & allowedInactivityTime, int category, const std::string & description) +comm::LocalServer *comm::Engine::createLocalServer(const std::string & addr, int port, int maxConnections, const anna::Millisecond & allowedInactivityTime, int category, const std::string & description) throw(anna::RuntimeException) { LocalServer* result(NULL); anna::Guard guard(this, "anna::diameter::comm::Engine::createLocalServer"); @@ -182,7 +274,7 @@ throw(anna::RuntimeException) { result->setCategory(category); result->setDescription(description); result->setAllowedInactivityTime(allowedInactivityTime); - result->initializeStatisticConcepts(); + result->initializeStatisticResources(); // Los saco con metodos virtuales readXXX del motor: // if ((a_cea.isEmpty()) || (a_dwa.isEmpty())) // throw anna::RuntimeException("Must define valid CEA and DWA messages by mean setCEAandDWA()", ANNA_FILE_LOCATION); @@ -199,7 +291,7 @@ throw(anna::RuntimeException) { } -Entity* Engine::createEntity(const std::string & addr1, int port1, const std::string & addr2, int port2, const std::string &description) +comm::Entity* comm::Engine::createEntity(const std::string & addr1, int port1, const std::string & addr2, int port2, const std::string &description) throw(anna::RuntimeException) { socket_v dualList; dualList.push_back(socket_t(addr1, port1)); @@ -208,7 +300,7 @@ throw(anna::RuntimeException) { } -Server* Engine::createServer(Entity *entity, const socket_t & socket) +comm::Server* comm::Engine::createServer(Entity *entity, const socket_t & socket) throw(anna::RuntimeException) { Server* result(NULL); anna::Guard guard(this, "anna::diameter::comm::Engine::createServer"); @@ -223,7 +315,7 @@ throw(anna::RuntimeException) { result->a_socket = socket; result->setMaxClientSessions(a_numberOfClientSessionsPerServer /* engine */); result->a_engine = this; - result->initializeStatisticConcepts(); + result->initializeStatisticResources(); for(int k = 0; k < a_numberOfClientSessionsPerServer; k++) result->addClientSession(k); @@ -240,7 +332,7 @@ throw(anna::RuntimeException) { // Lohacemos privado -ClientSession* Engine::createClientSession(Server *server, int socketId) +comm::ClientSession* comm::Engine::createClientSession(Server *server, int socketId) throw(anna::RuntimeException) { ClientSession* result(NULL); anna::Guard guard(this, "anna::diameter::comm::Engine::createClientSession"); @@ -286,7 +378,7 @@ throw(anna::RuntimeException) { } -bool Engine::broadcastEntities(const Message* message) throw(anna::RuntimeException) { +bool comm::Engine::broadcastEntities(const Message* message) throw(anna::RuntimeException) { LOGMETHOD(anna::TraceMethod tttm("diameter::comm::Engine", "broadcastEntities", ANNA_FILE_LOCATION)); bool allok = true; bool ok; @@ -305,7 +397,7 @@ bool Engine::broadcastEntities(const Message* message) throw(anna::RuntimeExcept return allok; } -bool Engine::broadcastLocalServers(const Message* message) throw(anna::RuntimeException) { +bool comm::Engine::broadcastLocalServers(const Message* message) throw(anna::RuntimeException) { LOGMETHOD(anna::TraceMethod tttm("diameter::comm::Engine", "broadcastLocalServers", ANNA_FILE_LOCATION)); bool allok = true; bool ok; @@ -324,7 +416,7 @@ bool Engine::broadcastLocalServers(const Message* message) throw(anna::RuntimeEx return allok; } -bool Engine::bind() throw(anna::RuntimeException) { +bool comm::Engine::bind() throw(anna::RuntimeException) { LOGMETHOD(anna::TraceMethod tttm("diameter::comm::Engine", "bind", ANNA_FILE_LOCATION)); bool result = true; // all OK return @@ -340,12 +432,12 @@ bool Engine::bind() throw(anna::RuntimeException) { return result; } -ClientSession* Engine::findClientSession(const std::string & addr, int port, int socketId, anna::Exception::Mode::_v emode) +comm::ClientSession* comm::Engine::findClientSession(const std::string & addr, int port, int socketId, anna::Exception::Mode::_v emode) throw(anna::RuntimeException) { return findClientSession(ClientSession::getKey(addr, port, socketId), emode); } -ClientSession* Engine::findClientSession(const std::string & key, anna::Exception::Mode::_v emode) +comm::ClientSession* comm::Engine::findClientSession(const std::string & key, anna::Exception::Mode::_v emode) throw(anna::RuntimeException) { anna::Guard guard(this, "anna::diameter::comm::Engine::findClientSession"); clientSession_iterator ii = clientSession_find(key); @@ -369,7 +461,7 @@ throw(anna::RuntimeException) { } -Server* Engine::findServer(const std::string & addr, int port, anna::Exception::Mode::_v emode) +comm::Server* comm::Engine::findServer(const std::string & addr, int port, anna::Exception::Mode::_v emode) throw(anna::RuntimeException) { anna::Guard guard(this, "anna::diameter::comm::Engine::findServer"); server_iterator ii = server_find(server_key(addr, port)); @@ -393,7 +485,7 @@ throw(anna::RuntimeException) { return NULL; } -Entity* Engine::findEntity(const socket_v & socketList, anna::Exception::Mode::_v emode) +comm::Entity* comm::Engine::findEntity(const socket_v & socketList, anna::Exception::Mode::_v emode) throw(anna::RuntimeException) { anna::Guard guard(this, "anna::diameter::comm::Engine::findEntity"); entity_key key(getEntityKey(socketList)); @@ -417,7 +509,7 @@ throw(anna::RuntimeException) { return NULL; } -Entity* Engine::findEntity(const std::string & addr1, int port1, const std::string & addr2, int port2, anna::Exception::Mode::_v emode) +comm::Entity* comm::Engine::findEntity(const std::string & addr1, int port1, const std::string & addr2, int port2, anna::Exception::Mode::_v emode) throw(anna::RuntimeException) { socket_v dualList; dualList.push_back(socket_t(addr1, port1)); @@ -440,7 +532,7 @@ throw(anna::RuntimeException) { //} -LocalServer* Engine::findLocalServer(const std::string & addr, int port, anna::Exception::Mode::_v emode) +comm::LocalServer* comm::Engine::findLocalServer(const std::string & addr, int port, anna::Exception::Mode::_v emode) throw(anna::RuntimeException) { anna::Guard guard(this, "anna::diameter::comm::Engine::findLocalServer"); socket_t key(addr, port); @@ -466,7 +558,7 @@ throw(anna::RuntimeException) { } -ServerSession* Engine::findServerSession(int socketId, anna::Exception::Mode::_v emode) throw(anna::RuntimeException) { +comm::ServerSession* comm::Engine::findServerSession(int socketId, anna::Exception::Mode::_v emode) throw(anna::RuntimeException) { anna::Guard guard(this, "anna::diameter::comm::Engine::findServerSession"); ServerSession *result; @@ -493,7 +585,7 @@ ServerSession* Engine::findServerSession(int socketId, anna::Exception::Mode::_v } -void Engine::closeClientSession(ClientSession* clientSession, bool destroy) +void comm::Engine::closeClientSession(comm::ClientSession* clientSession, bool destroy) throw(anna::RuntimeException) { if(clientSession == NULL) return; @@ -531,7 +623,7 @@ throw(anna::RuntimeException) { -void Engine::closeServer(Server* server, bool destroy) +void comm::Engine::closeServer(comm::Server* server, bool destroy) throw(anna::RuntimeException) { if(server == NULL) return; @@ -563,7 +655,7 @@ throw(anna::RuntimeException) { } -void Engine::closeEntity(Entity* entity, bool destroy) +void comm::Engine::closeEntity(comm::Entity* entity, bool destroy) throw(anna::RuntimeException) { if(entity == NULL) return; @@ -598,7 +690,7 @@ throw(anna::RuntimeException) { -void Engine::closeLocalServer(LocalServer* localServer, bool destroy) +void comm::Engine::closeLocalServer(comm::LocalServer* localServer, bool destroy) throw(anna::RuntimeException) { if(localServer == NULL) return; @@ -631,7 +723,7 @@ throw(anna::RuntimeException) { -void Engine::closeEntities(bool destroy) throw(anna::RuntimeException) { +void comm::Engine::closeEntities(bool destroy) throw(anna::RuntimeException) { LOGMETHOD(anna::TraceMethod tttm("diameter::comm::Engine", "closeEntities", ANNA_FILE_LOCATION)); anna::Guard guard(this, "anna::diameter::comm::Engine::closeEntities"); @@ -639,7 +731,7 @@ void Engine::closeEntities(bool destroy) throw(anna::RuntimeException) { closeEntity(entity(it), destroy); } -void Engine::closeLocalServers(bool destroy) throw(anna::RuntimeException) { +void comm::Engine::closeLocalServers(bool destroy) throw(anna::RuntimeException) { LOGMETHOD(anna::TraceMethod tttm("diameter::comm::Engine", "closeLocalServers", ANNA_FILE_LOCATION)); anna::Guard guard(this, "anna::diameter::comm::Engine::closeLocalServers"); @@ -647,7 +739,7 @@ void Engine::closeLocalServers(bool destroy) throw(anna::RuntimeException) { closeLocalServer(localServer(it), destroy); } -void Engine::eraseDeprecatedIdleEntities() throw() { +void comm::Engine::eraseDeprecatedIdleEntities() throw() { LOGMETHOD(anna::TraceMethod tttm("diameter::comm::Engine", "eraseDeprecatedIdleEntities", ANNA_FILE_LOCATION)); Entity *et; @@ -658,7 +750,7 @@ void Engine::eraseDeprecatedIdleEntities() throw() { } } -int Engine::getOTARequestsForEntities() const throw() { +int comm::Engine::getOTARequestsForEntities() const throw() { int result = 0; for(const_entity_iterator it = entity_begin(), maxii = entity_end(); it != maxii; it ++) @@ -667,7 +759,7 @@ int Engine::getOTARequestsForEntities() const throw() { return result; } -int Engine::getOTARequestsForLocalServers() const throw() { +int comm::Engine::getOTARequestsForLocalServers() const throw() { int result = 0; for(const_localServer_iterator it = localServer_begin(), maxii = localServer_end(); it != maxii; it ++) @@ -677,31 +769,31 @@ int Engine::getOTARequestsForLocalServers() const throw() { } -void Engine::setRealm(const std::string & name) throw() { - a_realm = ((name != "") ? name : anna::functions::getDomainname()); +void comm::Engine::setOriginRealmName(const std::string & originRealmName) throw() { + a_originRealm = ((originRealmName != "") ? originRealmName : anna::functions::getDomainname()); } -void Engine::setHost(const std::string & name) throw() { - a_host = ((name != "") ? name : anna::functions::getHostname()); +void comm::Engine::setOriginHostName(const std::string & originHostName) throw() { + a_originHost = ((originHostName != "") ? originHostName : anna::functions::getHostname()); } -void Engine::raiseAutoRecovery(bool autoRecovery) throw(anna::RuntimeException) { +void comm::Engine::raiseAutoRecovery(bool autoRecovery) throw(anna::RuntimeException) { LOGMETHOD(anna::TraceMethod tttm("diameter::comm::Engine", "raiseAutoRecovery", ANNA_FILE_LOCATION)); for(entity_iterator it = entity_begin(), maxii = entity_end(); it != maxii; it ++) entity(it)->raiseAutoRecovery(autoRecovery); } -void Engine::do_stop() +void comm::Engine::do_stop() throw() { LOGMETHOD(anna::TraceMethod tttm("diameter::comm::Engine", "do_stop", ANNA_FILE_LOCATION)); close(true /* destroy */); } -std::string Engine::asString(void) const throw() { +std::string comm::Engine::asString(void) const throw() { std::string trace; trace = "\n================================"; trace += "\nDiameter comm Engine information"; @@ -742,7 +834,7 @@ std::string Engine::asString(void) const throw() { } -anna::xml::Node* Engine::asXML(anna::xml::Node* parent) const +anna::xml::Node* comm::Engine::asXML(anna::xml::Node* parent) const throw() { parent = anna::app::Component::asXML(parent); anna::xml::Node* result = parent->createChild("diameter.comm.Engine"); @@ -765,33 +857,57 @@ throw() { for(const_localServer_iterator it = localServer_begin(); it != localServer_end(); it++) localServer(it)->asXML(localServers); + // DRA Basics + // Aspect: + // + // + // + // + // + // + for (dr_dh_server_sessions_it_t drit = a_dr_dh_server_sessions.begin(); drit != a_dr_dh_server_sessions.end(); drit++) { + anna::xml::Node* remoteRealm = result->createChild("Engine.RemoteRealm"); + remoteRealm->createAttribute("Name", drit->first); + dh_server_sessions_map_t *dhServerSessions = (dh_server_sessions_map_t *)&(drit->second); + for (dh_server_sessions_it_t dhit = dhServerSessions->begin(); dhit != dhServerSessions->end(); dhit++) { + anna::xml::Node* remoteRealmHost = remoteRealm->createChild("Engine.RemoteRealmHost"); + remoteRealmHost->createAttribute("Name", dhit->first); + server_sessions_vector_t *serverSessions = (server_sessions_vector_t *)&(dhit->second); + for (server_sessions_it_t ssit = serverSessions->begin(); ssit != serverSessions->end(); ssit++) { + std::string socket = anna::functions::socketLiteralAsString((*ssit)->getAddress(), (*ssit)->getPort()); + std::string ss_desc = socket + anna::functions::asString("|ServerSessionId:%d", (*ssit)->getSocketId()); + remoteRealmHost->createAttribute("ServerSession", ss_desc); + } + } + } + return result; } -Engine::clientSession_iterator Engine::clientSession_find(const clientSession_key &key) throw() { +comm::Engine::clientSession_iterator comm::Engine::clientSession_find(const clientSession_key &key) throw() { return a_clientSessions.find(key); } -Engine::server_iterator Engine::server_find(const server_key &key) throw() { +comm::Engine::server_iterator comm::Engine::server_find(const server_key &key) throw() { return a_servers.find(key); } -Engine::entity_iterator Engine::entity_find(const entity_key &key) throw() { +comm::Engine::entity_iterator comm::Engine::entity_find(const entity_key &key) throw() { return a_entities.find(key); } -Engine::localServer_iterator Engine::localServer_find(const socket_t &key) throw() { +comm::Engine::localServer_iterator comm::Engine::localServer_find(const socket_t &key) throw() { return a_localServers.find(key); } -Engine::entity_key Engine::getEntityKey(const std::string & addr1, int port1, const std::string & addr2, int port2) const throw() { +comm::Engine::entity_key comm::Engine::getEntityKey(const std::string & addr1, int port1, const std::string & addr2, int port2) const throw() { socket_v dualList; dualList.push_back(socket_t(addr1, port1)); dualList.push_back(socket_t(addr2, port2)); return (getEntityKey(dualList)); } -Engine::entity_key Engine::getEntityKey(const socket_v &v) const throw() { +comm::Engine::entity_key comm::Engine::getEntityKey(const socket_v &v) const throw() { std::string result; socket_v::const_iterator it; socket_v::const_iterator it_min(v.begin()); @@ -808,11 +924,13 @@ Engine::entity_key Engine::getEntityKey(const socket_v &v) const throw() { } -void Engine::availabilityLostForEntities() throw() { +void comm::Engine::availabilityLostForEntities() throw() { a_availableForEntities = false; LOGDEBUG( - std::string msg = "diameter::comm::Engine { Realm: "; - msg += getRealm(); + std::string msg = "diameter::comm::Engine { Origin-Realm: "; + msg += getOriginRealmName(); + msg += " | Origin-Host: "; + msg += getOriginHostName(); msg += " } has lost its availability for entities"; anna::Logger::debug(msg, ANNA_FILE_LOCATION); ); @@ -825,11 +943,13 @@ void Engine::availabilityLostForEntities() throw() { } -void Engine::availabilityRecoveredForEntities() throw() { +void comm::Engine::availabilityRecoveredForEntities() throw() { a_availableForEntities = true; LOGDEBUG( - std::string msg = "diameter::comm::Engine { Realm: "; - msg += getRealm(); + std::string msg = "diameter::comm::Engine { Origin-Realm: "; + msg += getOriginRealmName(); + msg += " | Origin-Host: "; + msg += getOriginHostName(); msg += " } has recovered its availability for entities"; anna::Logger::debug(msg, ANNA_FILE_LOCATION); ); @@ -842,11 +962,13 @@ void Engine::availabilityRecoveredForEntities() throw() { } -void Engine::availabilityLostForLocalServers() throw() { +void comm::Engine::availabilityLostForLocalServers() throw() { a_availableForLocalServers = false; LOGDEBUG( - std::string msg = "diameter::comm::Engine { Realm: "; - msg += getRealm(); + std::string msg = "diameter::comm::Engine { Origin-Realm: "; + msg += getOriginRealmName(); + msg += " | Origin-Host: "; + msg += getOriginHostName(); msg += " } has lost its availability for local servers"; anna::Logger::debug(msg, ANNA_FILE_LOCATION); ); @@ -859,11 +981,13 @@ void Engine::availabilityLostForLocalServers() throw() { } -void Engine::availabilityRecoveredForLocalServers() throw() { +void comm::Engine::availabilityRecoveredForLocalServers() throw() { a_availableForLocalServers = true; LOGDEBUG( - std::string msg = "diameter::comm::Engine { Realm: "; - msg += getRealm(); + std::string msg = "diameter::comm::Engine { Origin-Realm: "; + msg += getOriginRealmName(); + msg += " | Origin-Host: "; + msg += getOriginHostName(); msg += " } has recovered its availability for local servers"; anna::Logger::debug(msg, ANNA_FILE_LOCATION); ); @@ -876,7 +1000,7 @@ void Engine::availabilityRecoveredForLocalServers() throw() { } -bool Engine::refreshAvailabilityForEntities() throw() { +bool comm::Engine::refreshAvailabilityForEntities() throw() { // Here available if(a_availableForEntities) { // check not-bound state for all client-sessions: bool isolate = true; @@ -902,7 +1026,7 @@ bool Engine::refreshAvailabilityForEntities() throw() { return false; } -bool Engine::refreshAvailabilityForLocalServers() throw() { +bool comm::Engine::refreshAvailabilityForLocalServers() throw() { // Here available if(a_availableForLocalServers) { // check not-bound state for all client-sessions: bool isolate = true; @@ -929,7 +1053,11 @@ bool Engine::refreshAvailabilityForLocalServers() throw() { } -void Engine::readDPA(anna::DataBlock &dpa, const anna::DataBlock & dpr) throw() { +void comm::Engine::readDPA(anna::DataBlock &dpa, const anna::DataBlock & dpr) throw() { + + // Check for base protocol codec engine health: + assertBaseProtocolHealth(); + // Default DPA implementation: // // 'Disconnect-Peer-Answer' (282,answer) @@ -939,10 +1067,10 @@ void Engine::readDPA(anna::DataBlock &dpa, const anna::DataBlock & dpr) throw() // [Error-Message].................................(281,0) // *[Failed-AVP]....................................(279,0) try { - anna::diameter::codec::Message diameterDPA; - anna::diameter::codec::Avp avpRC; - anna::diameter::codec::Avp avpOH; - anna::diameter::codec::Avp avpOR; + anna::diameter::codec::Message diameterDPA(getBaseProtocolCodecEngine()); + anna::diameter::codec::Avp avpRC(getBaseProtocolCodecEngine()); + anna::diameter::codec::Avp avpOH(getBaseProtocolCodecEngine()); + anna::diameter::codec::Avp avpOR(getBaseProtocolCodecEngine()); // Message header diameterDPA.setId(anna::diameter::helpers::base::COMMANDID__Disconnect_Peer_Answer); diameterDPA.setVersion(1); @@ -956,23 +1084,49 @@ void Engine::readDPA(anna::DataBlock &dpa, const anna::DataBlock & dpr) throw() // Origin-Host avpOH.setId(anna::diameter::helpers::base::AVPID__Origin_Host); avpOH.setMandatoryBit(); - avpOH.getDiameterIdentity()->fromPrintableString(a_host.c_str()); + avpOH.getDiameterIdentity()->fromPrintableString(a_originHost.c_str()); // Origin-Realm avpOR.setId(anna::diameter::helpers::base::AVPID__Origin_Realm); avpOR.setMandatoryBit(); - avpOR.getDiameterIdentity()->fromPrintableString(a_realm.c_str()); + avpOR.getDiameterIdentity()->fromPrintableString(a_originRealm.c_str()); diameterDPA.addAvp(&avpRC); diameterDPA.addAvp(&avpOH); diameterDPA.addAvp(&avpOR); // Encode dpa = diameterDPA.code(); } catch(anna::RuntimeException &ex) { - ex.trace(); + std::string msg = ex.getText(); + msg += " | Use diameter::comm::Engine::setBaseProtocolCodecEngine() to allow internal base protocol messages encoding (unable to answer with DPA)"; + anna::Logger::error(msg, ANNA_FILE_LOCATION); + //throw anna::RuntimeException(msg, ANNA_FILE_LOCATION); } } -void Engine::readCEA(anna::DataBlock &cea, const anna::DataBlock & cer) throw() { +void comm::Engine::readCEA(anna::DataBlock &cea, const anna::DataBlock &cer) throw() { + + // Check for base protocol codec engine health: + assertBaseProtocolHealth(); + + if (a_ceaPathfile != "") { + anna::diameter::codec::Message diameterCEA(getBaseProtocolCodecEngine()); + + try { + diameterCEA.loadXML(a_ceaPathfile); + diameterCEA.setHopByHop(anna::diameter::codec::functions::getHopByHop(cer)); + diameterCEA.setEndToEnd(anna::diameter::codec::functions::getEndToEnd(cer)); + cea = diameterCEA.code(); + + } catch(anna::RuntimeException &ex) { + ex.trace(); + LOGWARNING(anna::Logger::warning("CEA file not found or unable to parse. Encoding harcoded default version ...", ANNA_FILE_LOCATION)); + //return anna::diameter::comm::Engine::readCEA(cea, cer); + // will fail with empty cea + } + + return; + } + // Default CEA implementation: // // 'Capabilities-Exchange-Answer' (257,answer) @@ -993,10 +1147,10 @@ void Engine::readCEA(anna::DataBlock &cea, const anna::DataBlock & cer) throw() // [Firmware-Revision].............................(267,0) // *[AVP]...........................................(0,0) try { - anna::diameter::codec::Message diameterCEA; - anna::diameter::codec::Avp avpRC; - anna::diameter::codec::Avp avpOH; - anna::diameter::codec::Avp avpOR; + anna::diameter::codec::Message diameterCEA(getBaseProtocolCodecEngine()); + anna::diameter::codec::Avp avpRC(getBaseProtocolCodecEngine()); + anna::diameter::codec::Avp avpOH(getBaseProtocolCodecEngine()); + anna::diameter::codec::Avp avpOR(getBaseProtocolCodecEngine()); // Message header diameterCEA.setId(anna::diameter::helpers::base::COMMANDID__Capabilities_Exchange_Answer); diameterCEA.setVersion(1); @@ -1010,11 +1164,11 @@ void Engine::readCEA(anna::DataBlock &cea, const anna::DataBlock & cer) throw() // Origin-Host avpOH.setId(anna::diameter::helpers::base::AVPID__Origin_Host); avpOH.setMandatoryBit(); - avpOH.getDiameterIdentity()->fromPrintableString(a_host.c_str()); + avpOH.getDiameterIdentity()->fromPrintableString(a_originHost.c_str()); // Origin-Realm avpOR.setId(anna::diameter::helpers::base::AVPID__Origin_Realm); avpOR.setMandatoryBit(); - avpOR.getDiameterIdentity()->fromPrintableString(a_realm.c_str()); + avpOR.getDiameterIdentity()->fromPrintableString(a_originRealm.c_str()); diameterCEA.addAvp(&avpRC); diameterCEA.addAvp(&avpOH); diameterCEA.addAvp(&avpOR); @@ -1025,17 +1179,83 @@ void Engine::readCEA(anna::DataBlock &cea, const anna::DataBlock & cer) throw() int vendorId = anna::diameter::helpers::VENDORID__tgpp; // Unsigned32 diameterCEA.addAvp(anna::diameter::helpers::base::AVPID__Vendor_Id)->getUnsigned32()->setValue(vendorId); // Product-Name - std::string productName = "OCS Diameter Server"; // UTF8String + std::string productName = "Diameter Server"; // UTF8String diameterCEA.addAvp(anna::diameter::helpers::base::AVPID__Product_Name)->getUTF8String()->setValue(productName); // Encode cea = diameterCEA.code(); } catch(anna::RuntimeException &ex) { + std::string msg = ex.getText(); + msg += " | Use diameter::comm::Engine::setBaseProtocolCodecEngine() to allow internal base protocol messages encoding (unable to answer with CEA)"; + anna::Logger::error(msg, ANNA_FILE_LOCATION); + //throw anna::RuntimeException(msg, ANNA_FILE_LOCATION); + } +} + +void comm::Engine::manageDrDhServerSession(ServerSession *ss, bool register_or_desregister) throw() { + + // Decode CER (TODO: use raw buffer helpers) + std::string destinationRealm, destinationHost; + codec::Message codecMsg(getBaseProtocolCodecEngine()); + try { + codecMsg.decode(ss->a_cer); + destinationRealm = codecMsg.getAvp(anna::diameter::helpers::base::AVPID__Origin_Realm)->getDiameterIdentity()->getValue(); + destinationHost = codecMsg.getAvp(anna::diameter::helpers::base::AVPID__Origin_Host)->getDiameterIdentity()->getValue(); + } + catch(anna::RuntimeException &ex) { ex.trace(); + return; + } + + dr_dh_server_sessions_nc_it_t drit = a_dr_dh_server_sessions.find(destinationRealm); + if (drit != a_dr_dh_server_sessions.end()) { // found + dh_server_sessions_map_t *dhServerSessions = (dh_server_sessions_map_t *)&(drit->second); + dh_server_sessions_nc_it_t dhit = dhServerSessions->find(destinationHost); + if (dhit != dhServerSessions->end()) { // found + server_sessions_vector_t *serverSessions = (server_sessions_vector_t *)&(dhit->second); + if (register_or_desregister) { // REGISTER + serverSessions->push_back(ss); + } + else { // DESREGISTER + // Sequential search the specific server session: + for (server_sessions_nc_it_t ssit = serverSessions->begin(); ssit != serverSessions->end(); ssit++) { + if ((*ssit)->getAddress() != ss->getAddress()) continue; + if ((*ssit)->getPort() != ss->getPort()) continue; + if ((*ssit)->getSocketId() != ss->getSocketId()) continue; + serverSessions->erase(ssit); // if it is the last server session removed in DR-DH path, the XML will show this tree empty + // (it could be a hint for past registerings): + // + // + // + // + // + // + + break; + } + } + } + else { + if (!register_or_desregister) return; // strange (host not found) + server_sessions_vector_t ssVector; + ssVector.push_back(ss); + (*dhServerSessions)[destinationHost] = ssVector; + } + } + else { + if (!register_or_desregister) return; // strange (realm not found) + server_sessions_vector_t ssVector; + ssVector.push_back(ss); + dh_server_sessions_map_t dhServerSessions; + dhServerSessions[destinationHost] = ssVector; + a_dr_dh_server_sessions[destinationRealm] = dhServerSessions; } } +void comm::Engine::readDWA(anna::DataBlock &dwa, const anna::DataBlock & dwr) throw() { + + // Check for base protocol codec engine health: + assertBaseProtocolHealth(); -void Engine::readDWA(anna::DataBlock &dwa, const anna::DataBlock & dwr) throw() { // Default DWA implementation: // // 'Device-Watchdog-Answer' (280,answer) @@ -1046,10 +1266,10 @@ void Engine::readDWA(anna::DataBlock &dwa, const anna::DataBlock & dwr) throw() // *[Failed-AVP]....................................(279,0) // [Origin-State-Id]...............................(278,0) try { - anna::diameter::codec::Message diameterDWA; - anna::diameter::codec::Avp avpRC; - anna::diameter::codec::Avp avpOH; - anna::diameter::codec::Avp avpOR; + anna::diameter::codec::Message diameterDWA(getBaseProtocolCodecEngine()); + anna::diameter::codec::Avp avpRC(getBaseProtocolCodecEngine()); + anna::diameter::codec::Avp avpOH(getBaseProtocolCodecEngine()); + anna::diameter::codec::Avp avpOR(getBaseProtocolCodecEngine()); // Message header diameterDWA.setId(anna::diameter::helpers::base::COMMANDID__Device_Watchdog_Answer); diameterDWA.setVersion(1); @@ -1063,22 +1283,25 @@ void Engine::readDWA(anna::DataBlock &dwa, const anna::DataBlock & dwr) throw() // Origin-Host avpOH.setId(anna::diameter::helpers::base::AVPID__Origin_Host); avpOH.setMandatoryBit(); - avpOH.getDiameterIdentity()->fromPrintableString(a_host.c_str()); + avpOH.getDiameterIdentity()->fromPrintableString(a_originHost.c_str()); // Origin-Realm avpOR.setId(anna::diameter::helpers::base::AVPID__Origin_Realm); avpOR.setMandatoryBit(); - avpOR.getDiameterIdentity()->fromPrintableString(a_realm.c_str()); + avpOR.getDiameterIdentity()->fromPrintableString(a_originRealm.c_str()); diameterDWA.addAvp(&avpRC); diameterDWA.addAvp(&avpOH); diameterDWA.addAvp(&avpOR); // Encode dwa = diameterDWA.code(); } catch(anna::RuntimeException &ex) { - ex.trace(); + std::string msg = ex.getText(); + msg += " | Use diameter::comm::Engine::setBaseProtocolCodecEngine() to allow internal base protocol messages encoding (unable to answer with DWA)"; + anna::Logger::error(msg, ANNA_FILE_LOCATION); + //throw anna::RuntimeException(msg, ANNA_FILE_LOCATION); } } -void Engine::resetStatistics() throw() { +void comm::Engine::resetStatistics() throw() { for(server_iterator it = server_begin(), maxii = server_end(); it != maxii; it ++) server(it)->resetStatistics(); @@ -1086,4 +1309,58 @@ void Engine::resetStatistics() throw() { localServer(it)->resetStatistics(); } +void comm::Engine::do_initialize() throw(RuntimeException) { + LOGMETHOD(anna::TraceMethod tttm("diameter::comm::Engine", "do_initialize", ANNA_FILE_LOCATION)); + LOGDEBUG(anna::Logger::debug("Nothing special done on component initialization", ANNA_FILE_LOCATION)); +} + +void comm::Engine::lazyInitialize() throw(RuntimeException) { + LOGMETHOD(anna::TraceMethod tttm("diameter::comm::Engine", "lazyInitialize", ANNA_FILE_LOCATION)); + anna::app::Component::initialize(); // this will invoke do_initialize +} +// Not tested yet +const comm::Response* comm::Engine::sendRealmHost(const Message* message, const std::string &destinationRealm, const std::string &destinationHost) throw(anna::RuntimeException) { + + if (destinationRealm == "") + throw anna::RuntimeException("Unable to resolve the destination: empty provided Destination-Realm name", ANNA_FILE_LOCATION); + + // Get the server sessions which fulfill the restrictions: + dr_dh_server_sessions_it_t drit = a_dr_dh_server_sessions.find(destinationRealm); + if (drit == a_dr_dh_server_sessions.end()) + throw anna::RuntimeException(anna::functions::asString("Unable to resolve the destination: Destination-Realm name is not registered (no remote clients have been connected to '%s')", destinationRealm.c_str()), ANNA_FILE_LOCATION); + + dh_server_sessions_map_t *dhServerSessions = (dh_server_sessions_map_t *)&(drit->second); + // randomize between all server sessions for all hosts: + dh_server_sessions_nc_it_t dhit; + int hostsN = dhServerSessions->size(); + if (hostsN == 0) // avoids next division by cero (rand() % 0) + throw anna::RuntimeException(anna::functions::asString("Unable to resolve the destination: neither Destination-Host currently connected to Destination-Realm '%s'", destinationRealm.c_str()), ANNA_FILE_LOCATION); + + if (destinationHost == "") { + // in this case, randomize the host: + dhit = dhServerSessions->begin(); + int randomHostIndx = rand() % hostsN; // number between 0 and the number of hosts - 1 + std::advance (dhit, randomHostIndx); + } + else { + dhit = dhServerSessions->find(destinationHost); + if (dhit == dhServerSessions->end()) + throw anna::RuntimeException(anna::functions::asString("Unable to resolve the destination: Destination-Host '%s' is not registered for Destination-Realm '%s'", destinationHost.c_str(), destinationRealm.c_str()), ANNA_FILE_LOCATION); + } + + // Now, randomize the available server sessions: + server_sessions_vector_t *serverSessions = (server_sessions_vector_t *)&(dhit->second); + int serverSessionN = serverSessions->size(); + if (serverSessionN == 0) { // avoids next division by cero (rand() % 0) + std::string aux = ""; + if (destinationHost != "") { aux = "to Destination-Host '"; aux += destinationHost; aux += "'"; } + std::string msg = anna::functions::asString("Unable to resolve the destination: neither server session currently connected%s within Destination-Realm '%s'", aux.c_str(), destinationRealm.c_str()); + throw anna::RuntimeException(msg, ANNA_FILE_LOCATION); + } + + server_sessions_nc_it_t ssit = serverSessions->begin(); + int randomServerSessionIndx = rand() % serverSessionN; // number between 0 and the number of server sessions - 1 + std::advance (ssit, randomServerSessionIndx); + return (*ssit)->send(message); +}