Fixed Failed-AVP bug (feature). Now wrong avps inside grouped are correctly tracked
[anna.git] / source / diameter / codec / Message.cpp
index 6490b81..3a989cf 100644 (file)
@@ -1,8 +1,8 @@
-// ANNA - Anna is Not 'N' Anymore
+// ANNA - Anna is Not Nothingness Anymore
 //
 // (c) Copyright 2005-2014 Eduardo Ramos Testillano & Francisco Ruiz Rayo
 //
-// https://bitbucket.org/testillano/anna
+// http://redmine.teslayout.com/projects/anna-suite
 //
 // Redistribution and use in source and binary forms, with or without
 // modification, are permitted provided that the following conditions
@@ -14,7 +14,7 @@
 // copyright notice, this list of conditions and the following disclaimer
 // in the documentation and/or other materials provided with the
 // distribution.
-//     * Neither the name of Google Inc. nor the names of its
+//     *  Neither the name of the copyright holder nor the names of its
 // contributors may be used to endorse or promote products derived from
 // this software without specific prior written permission.
 //
@@ -59,6 +59,7 @@
 
 // STL
 #include <string>
+#include <vector>
 
 
 using namespace anna;
@@ -300,7 +301,7 @@ void Message::decode(const anna::DataBlock &db, Message *ptrAnswer) throw(anna::
   if(db.getSize() < HeaderLength) {
     oamModule.activateAlarm(OamModule::Alarm::MessageDecode__NotEnoughBytesToCoverMessageHeaderLength);
     oamModule.count(OamModule::Counter::MessageDecode__NotEnoughBytesToCoverMessageHeaderLength);
-    // DIAMETER_INVALID_MESSAGE_LENGTH; // no podré construir un answer fiable, así que no registro el result-code
+    // DIAMETER_INVALID_MESSAGE_LENGTH; // no podr� construir un answer fiable, as� que no registro el result-code
     throw anna::RuntimeException("Not enough bytes to cover message header length (20 bytes)", ANNA_FILE_LOCATION);
   }
 
@@ -369,17 +370,17 @@ void Message::decode(const anna::DataBlock &db, Message *ptrAnswer) throw(anna::
   Avp* avp;
   anna::DataBlock db_aux;
 
+  // Parent information:
+  parent_t parent;
+  parent.setMessage(a_id);
+
+
   while(avpPos < dataBytes) {
     try {
       avp =  getEngine()->allocateAvp();
       db_aux.assign(startData + avpPos, dataBytes - avpPos /* is valid to pass total length (indeed i don't know the real avp length) because it will be limited and this has deep copy disabled (no memory is reserved) */);
-      avp -> decode(db_aux, answer);
+      avp -> decode(db_aux, parent, answer);
     } catch(anna::RuntimeException &ex) {
-      if(answer) {
-        answer->setResultCode(helpers::base::AVPVALUES__Result_Code::DIAMETER_INVALID_AVP_VALUE); // unspecified error ...
-        answer->setNewFailedAvp(avp->getId());
-      }
-
       getEngine()->releaseAvp(avp);
       LOGWARNING(
         anna::Logger::warning(ex.getText(), ANNA_FILE_LOCATION);
@@ -457,14 +458,55 @@ int Message::getResultCode() const throw() {
 
 
 //------------------------------------------------------------------------------
-//------------------------------------------------------ Message::addFailedAVP()
+//------------------------------------------------------ Message::setFailedAvp()
 //------------------------------------------------------------------------------
-Avp * Message::addFailedAVP() throw() {
-  Avp *result = getAvp(helpers::base::AVPID__Failed_AVP, 1, anna::Exception::Mode::Ignore);
+void Message::setFailedAvp(const parent_t &parent, AvpId wrong) throw(anna::RuntimeException) {
 
-  if(!result) result = addAvp(helpers::base::AVPID__Failed_AVP);
+  if(isRequest()) return;
 
-  return result;
+// RFC 6733:
+//
+//  7.5.  Failed-AVP AVP
+//
+//     The Failed-AVP AVP (AVP Code 279) is of type Grouped and provides
+//     debugging information in cases where a request is rejected or not
+//     fully processed due to erroneous information in a specific AVP.  The
+//     value of the Result-Code AVP will provide information on the reason
+//     for the Failed-AVP AVP.  A Diameter answer message SHOULD contain an
+//     instance of the Failed-AVP AVP that corresponds to the error
+//     indicated by the Result-Code AVP.  For practical purposes, this
+//     Failed-AVP would typically refer to the first AVP processing error
+//     that a Diameter node encounters.
+
+  // Although the Failed-AVP definition has cardinality 1* and Failed-AVP itself is defined in
+  // most of the command codes as *[Failed-AVP], i think this is not a deliberate ambiguity.
+  // Probably the RFC wants to give freedom to the application layer, but it is recommended to
+  // have only one child (wrong avp) inside a unique message Failed-AVP to ease the Result-Code
+  // correspondence. Anyway, this behaviour could be easily  opened commenting condition block (*).
+  Avp *theFailedAvp = getAvp(helpers::base::AVPID__Failed_AVP, 1, anna::Exception::Mode::Ignore);
+  if (theFailedAvp) {
+       LOGDEBUG(anna::Logger::debug("Failed-AVP has already been added. RFC 6733 Section 7.5 recommends to store only the first error found", ANNA_FILE_LOCATION));
+    return;
+  }
+
+  // Section 7.5 RFC 6733: A Diameter message SHOULD contain one Failed-AVP AVP
+  theFailedAvp = addAvp(helpers::base::AVPID__Failed_AVP);
+  Avp *leaf = theFailedAvp;
+
+  LOGDEBUG(
+    std::string msg = "Adding to Failed-AVP, the wrong avp ";
+    msg += anna::diameter::functions::avpIdAsPairString(wrong);
+    msg += " found inside ";
+    msg += parent.asString();
+
+    anna::Logger::debug(msg, ANNA_FILE_LOCATION);
+  );
+
+  std::vector<AvpId>::const_iterator it;
+  for(it = parent.AvpsId.begin(); it != parent.AvpsId.end(); it++)
+       leaf = leaf->addAvp(*it);
+
+  leaf->addAvp(wrong);
 }
 
 
@@ -480,7 +522,8 @@ void Message::setStandardToAnswer(const Message &request, const std::string &ori
   const Avp *reqSessionId = request.getAvp(helpers::base::AVPID__Session_Id, 1, anna::Exception::Mode::Ignore);
 
   if(reqSessionId)
-    addAvp(helpers::base::AVPID__Session_Id)->getUTF8String()->setValue(reqSessionId->getUTF8String()->getValue());
+       if(!getAvp(helpers::base::AVPID__Session_Id, 1, anna::Exception::Mode::Ignore))
+      addAvp(helpers::base::AVPID__Session_Id)->getUTF8String()->setValue(reqSessionId->getUTF8String()->getValue());
 
   // Origin-Host & Realm
   if(!getAvp(helpers::base::AVPID__Origin_Host, 1, anna::Exception::Mode::Ignore))
@@ -543,7 +586,6 @@ bool Message::valid(Message *ptrAnswer) const throw(anna::RuntimeException) {
   OamModule &oamModule = OamModule::instantiate();
   // Dictionary stack command:
   const stack::Command *stackCommand = getStackCommand();
-  std::string me;
   // Only build answer for a request:
   Message *answer = isRequest() ? ptrAnswer : NULL;
 
@@ -551,7 +593,7 @@ bool Message::valid(Message *ptrAnswer) const throw(anna::RuntimeException) {
 
   if(!stackCommand) {
     // OAM
-    me = anna::diameter::functions::commandIdAsPairString(a_id);
+    std::string me = anna::diameter::functions::commandIdAsPairString(a_id);
     oamModule.activateAlarm(OamModule::Alarm::MessageValidation__UnknownOperation__s__UnableToValidate, STRING_WITH_QUOTATION_MARKS__C_STR(me));
     oamModule.count(OamModule::Counter::MessageValidation__UnknownOperationUnableToValidate);
 
@@ -561,7 +603,10 @@ bool Message::valid(Message *ptrAnswer) const throw(anna::RuntimeException) {
     return false;
   }
 
-  me = stackCommand->getName();
+  // Parent information:
+  parent_t me;
+  me.setMessage(a_id, stackCommand->getName().c_str());
+
   //////////////////////////////
   // Flags coherence checking //
   //////////////////////////////
@@ -570,12 +615,12 @@ bool Message::valid(Message *ptrAnswer) const throw(anna::RuntimeException) {
 
   if(!result) {
     // OAM & Depth management
-    oamModule.activateAlarm(OamModule::Alarm::MessageValidation__Operation__s__HaveIncoherentFlags__d__, STRING_WITH_QUOTATION_MARKS__C_STR(me), (int)a_flags);
+    oamModule.activateAlarm(OamModule::Alarm::MessageValidation__Operation__s__HaveIncoherentFlags__d__, STRING_WITH_QUOTATION_MARKS__C_STR(me.asString()), (int)a_flags);
     oamModule.count(OamModule::Counter::MessageValidation__OperationHaveIncoherentFlags);
 
     if(answer) answer->setResultCode(rc);
 
-    getEngine()->validationAnomaly(anna::functions::asString("Operation %s have incoherent flags (%d)", STRING_WITH_QUOTATION_MARKS__C_STR(me), (int)a_flags));
+    getEngine()->validationAnomaly(anna::functions::asString("Operation %s have incoherent flags (%d)", STRING_WITH_QUOTATION_MARKS__C_STR(me.asString()), (int)a_flags));
   }
 
   ////////////////////
@@ -811,11 +856,13 @@ void Message::fromXML(const anna::xml::Node* messageNode) throw(anna::RuntimeExc
   // Application-id
   u_aux = appid->getIntegerValue();
 
+  /*
   if(u_aux < 0) {
     std::string msg = "Error processing command <application-id '"; msg += appid->getValue();
     msg += "': negative values are not allowed";
     throw anna::RuntimeException(msg, ANNA_FILE_LOCATION);
   }
+  */
 
   setApplicationId(u_aux);
 
@@ -823,11 +870,13 @@ void Message::fromXML(const anna::xml::Node* messageNode) throw(anna::RuntimeExc
   if(hbh) {
     u_aux = hbh->getIntegerValue();
 
+    /*
     if(u_aux < 0) {
       std::string msg = "Error processing command <hop-by-hop-id '"; msg += hbh->getValue();
       msg += "': negative values are not allowed";
       throw anna::RuntimeException(msg, ANNA_FILE_LOCATION);
     }
+    */
   } else u_aux = 0;
 
   setHopByHop(u_aux);
@@ -836,11 +885,13 @@ void Message::fromXML(const anna::xml::Node* messageNode) throw(anna::RuntimeExc
   if(ete) {
     u_aux = ete->getIntegerValue();
 
+    /*
     if(u_aux < 0) {
       std::string msg = "Error processing command <end-to-end-id '"; msg += ete->getValue();
       msg += "': negative values are not allowed";
       throw anna::RuntimeException(msg, ANNA_FILE_LOCATION);
     }
+    */
   } else u_aux = 0;
 
   setEndToEnd(u_aux);