X-Git-Url: https://git.teslayout.com/public/public/public/?a=blobdiff_plain;f=source%2Fdiameter%2Fhelpers%2Fbase%2Ffunctions.cpp;h=51a0370e39de6f2ae6295b04ca055c3c35bf7f86;hb=5a6cba5fde2b2f538a7515f8293cc0a8d9589dfa;hp=20a02570fb98e7fd936b19411abd800539aa53aa;hpb=4c3f0a4d7e4db76996404d80c6f939548fca656f;p=anna.git

diff --git a/source/diameter/helpers/base/functions.cpp b/source/diameter/helpers/base/functions.cpp
index 20a0257..51a0370 100644
--- a/source/diameter/helpers/base/functions.cpp
+++ b/source/diameter/helpers/base/functions.cpp
@@ -26,7 +26,7 @@ using namespace anna::diameter::codec;
 using namespace anna::diameter::helpers::base;
 
 // getters
-U32 anna::diameter::helpers::base::functions::getResultCode(const anna::DataBlock & db) throw(anna::RuntimeException) {
+U32 anna::diameter::helpers::base::functions::getResultCode(const anna::DataBlock & db) noexcept(false) {
   if(db.getSize() < Message::HeaderLength)
     throw anna::RuntimeException("Not enough bytes to cover command header length", ANNA_FILE_LOCATION);
 
@@ -54,7 +54,7 @@ U32 anna::diameter::helpers::base::functions::getResultCode(const anna::DataBloc
 
 
 void anna::diameter::helpers::base::functions::decodeSessionId(const std::string &sessionId,
-    std::string &diameterIdentity, U32 &high, U32 &low, std::string &optional) throw(anna::RuntimeException) {
+    std::string &diameterIdentity, U32 &high, U32 &low, std::string &optional) noexcept(false) {
 // RFC 3588:
 //
 //      8.8.  Session-Id AVP
@@ -125,7 +125,7 @@ void anna::diameter::helpers::base::functions::decodeSessionId(const std::string
 
 
 std::string anna::diameter::helpers::base::functions::getSessionId(const anna::DataBlock & db,
-    std::string &diameterIdentity, U32 &high, U32 &low, std::string &optional) throw(anna::RuntimeException) {
+    std::string &diameterIdentity, U32 &high, U32 &low, std::string &optional) noexcept(false) {
   if(db.getSize() < Message::HeaderLength)
     throw anna::RuntimeException("Not enough bytes to cover command header length", ANNA_FILE_LOCATION);
 
@@ -150,3 +150,25 @@ std::string anna::diameter::helpers::base::functions::getSessionId(const anna::D
   return result;
 }
 
+
+std::string anna::diameter::helpers::base::functions::getOriginHost(const anna::DataBlock & db) noexcept(false) {
+  if(db.getSize() < Message::HeaderLength)
+    throw anna::RuntimeException("Not enough bytes to cover command header length", ANNA_FILE_LOCATION);
+
+  const char *avpsDB = db.getData() + Message::HeaderLength;
+  int avpsLen = db.getSize() - Message::HeaderLength;
+  const char * originHostPtr = anna::diameter::codec::functions::findAVP(avpsDB, avpsLen, AVPID__Origin_Host);
+
+
+  if(originHostPtr == NULL)
+    throw anna::RuntimeException("Origin-Host AVP not found in DataBlock provided", ANNA_FILE_LOCATION);
+
+  // Decoded avp information:
+  AvpId _id;
+  char _flags;
+  int _length;
+  std::string result;
+  anna::diameter::codec::functions::decodeAVP(originHostPtr, _id, _flags, _length, result);
+  return result;
+}
+