X-Git-Url: https://git.teslayout.com/public/public/public/?p=anna.git;a=blobdiff_plain;f=example%2Fdiameter%2Flauncher%2FMyLocalServer.cpp;h=31873c690f772ff1f05bccc2fa7ab972fe345be1;hp=42b8b1e233c0e761fc46be7674ae43dbdef0fcb3;hb=HEAD;hpb=7681cb4079366eb6908dd5d0dc0124c3fcef4b6b

diff --git a/example/diameter/launcher/MyLocalServer.cpp b/example/diameter/launcher/MyLocalServer.cpp
index 42b8b1e..31873c6 100644
--- a/example/diameter/launcher/MyLocalServer.cpp
+++ b/example/diameter/launcher/MyLocalServer.cpp
@@ -24,13 +24,16 @@
 #include <anna/testing/TestManager.hpp>
 
 
-void MyLocalServer::eventRequest(anna::diameter::comm::ServerSession *serverSession, const anna::DataBlock &message)
+void MyLocalServer::eventRequest(anna::diameter::comm::ServerSession *serverSession, const anna::DataBlock &message, const anna::diameter::comm::OriginHost *myNode)
 noexcept(false) {
   LOGMETHOD(anna::TraceMethod tm("launcher::MyLocalServer", "eventRequest", ANNA_FILE_LOCATION));
+
+  // Protection:
+  if (!myNode) return;
+
   // Performance stats:
   Launcher& my_app = static_cast <Launcher&>(anna::app::functions::getApp());
-  anna::diameter::comm::OriginHost *my_node = my_app.getOriginHost(getEngine()->getOriginHostName());
-  anna::diameter::codec::Engine *codecEngine = my_node->getCodecEngine();
+  anna::diameter::codec::Engine *codecEngine = myNode->getCodecEngine();
 
   // CommandId:
   anna::diameter::CommandId cid = anna::diameter::codec::functions::getCommandId(message);
@@ -46,27 +49,28 @@ noexcept(false) {
   );
 
   // Write reception
-  if(my_node->logEnabled()) my_node->writeLogFile(message, "recvfc", serverSession->asString());
+  if(myNode->logEnabled()) myNode->writeLogFile(message, "recvfc", serverSession->asString());
 
   // If no answer is programmed and entity is configured, the failed request would be forwarded even being wrong (delegates at the end point)
   int code = cid.first;
-  anna::diameter::codec::Message *programmed_answer = a_reactingAnswers.getMessage(code);
+  anna::diameter::codec::Message *programmed_answer = (const_cast<anna::diameter::comm::OriginHost*>(myNode))->getReactingAnswers()->getMessage(code);
+
   bool programmed = (programmed_answer != NULL);
 
-  anna::diameter::comm::Entity *entity = my_node->getEntity();
+  anna::diameter::comm::Entity *entity = myNode->getEntity();
   if(!programmed && entity) {  // forward condition (no programmed answer + entity available)
-    anna::diameter::comm::Message *msg = my_node->createCommMessage();
+    anna::diameter::comm::Message *msg = (const_cast<anna::diameter::comm::OriginHost*>(myNode))->createCommMessage();
     msg->forwardEndToEnd(); // end-to-end will be kept
     msg->setBody(message);
     msg->setRequestServerSessionKey(serverSession->getKey());
     bool success = entity->send(msg);
 
     // Detailed log:
-    if(my_node->logEnabled()) {
+    if(myNode->logEnabled()) {
       anna::diameter::comm::Server *usedServer = entity->getLastUsedResource();
       anna::diameter::comm::ClientSession *usedClientSession = usedServer ? usedServer->getLastUsedResource() : NULL;
       std::string detail = usedClientSession ? usedClientSession->asString() : "[null client session]"; // esto no deberia ocurrir
-      my_node->writeLogFile(message, (success ? "fwd2e" : "fwd2eError"), detail); // forwarded
+      myNode->writeLogFile(message, (success ? "fwd2e" : "fwd2eError"), detail); // forwarded
     }
 
 
@@ -90,7 +94,7 @@ noexcept(false) {
     // Decode
     try { codecMsg.decode(message, answer_message); } catch(anna::RuntimeException &ex) { ex.trace(); }
 
-    answer_message->setStandardToAnswer(codecMsg, my_node->getCommEngine()->getOriginHostName(), my_node->getCommEngine()->getOriginRealmName());
+    answer_message->setStandardToAnswer(codecMsg, myNode->getCommEngine()->getOriginHostName(), myNode->getCommEngine()->getOriginRealmName());
     analysisOK = (answer_message->getResultCode() == anna::diameter::helpers::base::AVPVALUES__Result_Code::DIAMETER_SUCCESS);
   }
 
@@ -116,34 +120,33 @@ noexcept(false) {
 
   anna::diameter::comm::Message *msg;
   try {
-    msg = my_node->createCommMessage();
+    msg = (const_cast<anna::diameter::comm::OriginHost*>(myNode))->createCommMessage();
     msg->setBody(answer_message->code());
     /* response = NULL =*/serverSession->send(msg);
 
-    if(my_node->logEnabled()) my_node->writeLogFile(*answer_message, "sent2c", serverSession->asString());
+    if(myNode->logEnabled()) myNode->writeLogFile(*answer_message, "sent2c", serverSession->asString());
   } catch(anna::RuntimeException &ex) {
     ex.trace();
-    if(my_node->logEnabled()) my_node->writeLogFile(*answer_message, "send2cError", serverSession->asString());
+    if(myNode->logEnabled()) myNode->writeLogFile(*answer_message, "send2cError", serverSession->asString());
   }
 
   // release msg
-  my_node->releaseCommMessage(msg);
+  (const_cast<anna::diameter::comm::OriginHost*>(myNode))->releaseCommMessage(msg);
 
   // Restore validation mode
   codecEngine->setValidationMode(backupVM);
 
   // Pop front the reacting answer:
-  if(analysisOK && programmed) a_reactingAnswers.nextMessage(code);
+  if(analysisOK && programmed) (const_cast<anna::diameter::comm::OriginHost*>(myNode))->getReactingAnswers()->nextMessage(code);
 
   // Testing:
   anna::testing::TestManager::instantiate().receiveDiameterMessage(message, serverSession);
 }
 
-void MyLocalServer::eventResponse(const anna::diameter::comm::Response &response)
+void MyLocalServer::eventResponse(const anna::diameter::comm::Response &response, const anna::diameter::comm::OriginHost *myNode)
 noexcept(false) {
   LOGMETHOD(anna::TraceMethod tm("launcher::MyLocalServer", "eventResponse", ANNA_FILE_LOCATION));
   Launcher& my_app = static_cast <Launcher&>(anna::app::functions::getApp());
-  anna::diameter::comm::OriginHost * my_node = my_app.getOriginHost(getEngine()->getOriginHostName());
   anna::diameter::comm::ClassCode::_v code = response.getClassCode();
   anna::diameter::comm::Response::ResultCode::_v result = response.getResultCode();
   anna::diameter::comm::Message* request = const_cast<anna::diameter::comm::Message*>(response.getRequest());
@@ -168,6 +171,9 @@ noexcept(false) {
     anna::Logger::debug(msg, ANNA_FILE_LOCATION);
   );
 
+  // Protection:
+  if (!myNode) return;
+
   if(isUnavailable) {
     //if (isApplicationMessage)
     LOGWARNING(anna::Logger::warning("Diameter client unavailable for Diameter Request", ANNA_FILE_LOCATION));
@@ -178,7 +184,7 @@ noexcept(false) {
     LOGWARNING(anna::Logger::warning("Context Expired for Diameter Request which was sent to the client", ANNA_FILE_LOCATION));
 
     if(request_cid != anna::diameter::helpers::base::COMMANDID__Capabilities_Exchange_Request) {  // don't trace CEA
-      if(my_node->logEnabled()) my_node->writeLogFile(*request, "req2c-expired", serverSession->asString());
+      if(myNode->logEnabled()) myNode->writeLogFile(*request, "req2c-expired", serverSession->asString());
     }
   }
 
@@ -190,20 +196,20 @@ noexcept(false) {
     );
 
     // Write reception
-    if(my_node->logEnabled()) my_node->writeLogFile(*message, "recvfc", serverSession->asString());
+    if(myNode->logEnabled()) myNode->writeLogFile(*message, "recvfc", serverSession->asString());
 
     // This is not very usual, but answers could arrive from clients:
-    anna::diameter::comm::Entity *entity = my_node->getEntity();
+    anna::diameter::comm::Entity *entity = myNode->getEntity();
 
     if(entity) {
-      anna::diameter::comm::ClientSession *usedClientSession = my_node->getCommEngine()->findClientSession(request->getRequestClientSessionKey());
+      anna::diameter::comm::ClientSession *usedClientSession = myNode->getCommEngine()->findClientSession(request->getRequestClientSessionKey());
       std::string detail;
 
-      if(my_node->logEnabled()) detail = usedClientSession ? usedClientSession->asString() : "[null client session]";  // this should not happen
+      if(myNode->logEnabled()) detail = usedClientSession ? usedClientSession->asString() : "[null client session]";  // this should not happen
       anna::diameter::comm::Message *msg;
 
       try {
-        msg = my_node->createCommMessage();
+        msg = (const_cast<anna::diameter::comm::OriginHost*>(myNode))->createCommMessage();
         msg->forwardEndToEnd(); // end-to-end will be kept
         msg->setBody(*message);
 
@@ -214,15 +220,15 @@ noexcept(false) {
         //msg->setRequestClientSessionKey(request->getRequestClientSessionKey());
         //bool success = entity->send(msg);
 
-        if(my_node->logEnabled()) my_node->writeLogFile(*message, "fwd2e", detail);  // forwarded
+        if(myNode->logEnabled()) myNode->writeLogFile(*message, "fwd2e", detail);  // forwarded
       } catch(anna::RuntimeException &ex) {
         ex.trace();
-        if(my_node->logEnabled()) my_node->writeLogFile(*message, "fwd2eError", detail);  // forwarded
+        if(myNode->logEnabled()) myNode->writeLogFile(*message, "fwd2eError", detail);  // forwarded
       }
 
       // release msgs
-      my_node->releaseCommMessage(msg);
-      my_node->releaseCommMessage(request);
+      (const_cast<anna::diameter::comm::OriginHost*>(myNode))->releaseCommMessage(msg);
+      (const_cast<anna::diameter::comm::OriginHost*>(myNode))->releaseCommMessage(request);
     }
   }
 
@@ -230,12 +236,11 @@ noexcept(false) {
   if(isOK) anna::testing::TestManager::instantiate().receiveDiameterMessage(*message, serverSession);
 }
 
-void MyLocalServer::eventUnknownResponse(anna::diameter::comm::ServerSession *serverSession, const anna::DataBlock &message)
+void MyLocalServer::eventUnknownResponse(anna::diameter::comm::ServerSession *serverSession, const anna::DataBlock &message, const anna::diameter::comm::OriginHost *myNode)
 noexcept(false) {
   LOGMETHOD(anna::TraceMethod tm("launcher::MyLocalServer", "eventUnknownResponse", ANNA_FILE_LOCATION));
   // Performance stats:
   Launcher& my_app = static_cast <Launcher&>(anna::app::functions::getApp());
-  anna::diameter::comm::OriginHost *my_node = my_app.getOriginHost(getEngine()->getOriginHostName());
   // CommandId:
   anna::diameter::CommandId cid = anna::diameter::codec::functions::getCommandId(message);
   LOGDEBUG
@@ -249,15 +254,17 @@ noexcept(false) {
     anna::Logger::debug(msg, ANNA_FILE_LOCATION);
   );
 
-  if(my_node->logEnabled()) my_node->writeLogFile(message, "recvfc-ans-unknown", serverSession->asString());
+  // Protection:
+  if (!myNode) return;
+
+  if(myNode->logEnabled()) myNode->writeLogFile(message, "recvfc-ans-unknown", serverSession->asString());
 }
 
-void MyLocalServer::eventDPA(anna::diameter::comm::ServerSession *serverSession, const anna::DataBlock &message)
+void MyLocalServer::eventDPA(anna::diameter::comm::ServerSession *serverSession, const anna::DataBlock &message, const anna::diameter::comm::OriginHost *myNode)
 noexcept(false) {
   LOGMETHOD(anna::TraceMethod tm("launcher::MyLocalServer", "eventDPA", ANNA_FILE_LOCATION));
   // Performance stats:
   Launcher& my_app = static_cast <Launcher&>(anna::app::functions::getApp());
-  anna::diameter::comm::OriginHost *my_node = my_app.getOriginHost(getEngine()->getOriginHostName());
   // CommandId:
   anna::diameter::CommandId cid = anna::diameter::codec::functions::getCommandId(message);
   LOGDEBUG
@@ -271,8 +278,11 @@ noexcept(false) {
     anna::Logger::debug(msg, ANNA_FILE_LOCATION);
   );
 
+  // Protection:
+  if (!myNode) return;
+
   // Write reception
-  if(my_node->logEnabled()) my_node->writeLogFile(message, "recvfc", serverSession->asString());
+  if(myNode->logEnabled()) myNode->writeLogFile(message, "recvfc", serverSession->asString());
 
   // Testing:
   anna::testing::TestManager::instantiate().receiveDiameterMessage(message, serverSession);