Node class, command line redesign. New xml template for process configuration.

[anna.git] / source / diameter.comm / Engine.cpp

diff --git a/source/diameter.comm/Engine.cpp b/source/diameter.comm/Engine.cpp
index 4028991..538001a 100644 (file)
--- a/source/diameter.comm/Engine.cpp
+++ b/source/diameter.comm/Engine.cpp
@@ -61,7 +61,7 @@ ClientSession* Engine::allocateClientSession() throw() { return a_clientSessions
 void Engine::releaseClientSession(ClientSession *clientSession) throw() { a_clientSessionsRecycler.release(clientSession); }
 
 
-void Engine::setCERandDWR(const anna::DataBlock & cer, const anna::DataBlock & dwr) throw(anna::RuntimeException) {
+void 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);
   }
@@ -74,18 +74,86 @@ 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 Engine::setClientCERandDWR(const std::string & cer, const std::string & dwr) throw(anna::RuntimeException) {
+
+  // Check for base protocol codec engine:
+  if (!getBaseProtocolCodecEngine())
+    throw anna::RuntimeException("Use diameter::comm::Engine::setBaseProtocolCodecEngine() to allow base protocol messages encoding, or use setClientCERandDWR(const anna::DataBlock & cer, const anna::DataBlock & dwr) which expect externally encoded messages", ANNA_FILE_LOCATION);
+
+  // Build CER
+  //   <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 = getHost();
+  std::string OR = getRealm();
+  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|<ip address>"
+    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);
+  }
+
+  // Build DWR
+  //   <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 Engine::setWatchdogPeriod(const anna::Millisecond & wp) throw(anna::RuntimeException) {
   if(wp < ClientSession::DefaultWatchdogPeriod) {