include/anna/diameter.comm/Message.hpp

   1 // ANNA - Anna is Not Nothingness Anymore                                                         //
   2 //                                                                                                //
   3 // (c) Copyright 2005-2015 Eduardo Ramos Testillano & Francisco Ruiz Rayo                         //
   4 //                                                                                                //
   5 // See project site at http://redmine.teslayout.com/projects/anna-suite                           //
   6 // See accompanying file LICENSE or copy at http://www.teslayout.com/projects/public/anna.LICENSE //
   7
   8
   9 #ifndef anna_diameter_comm_Message_hpp
  10 #define anna_diameter_comm_Message_hpp
  11
  12 // Local
  13 #include <anna/diameter/defines.hpp>
  14 #include <anna/diameter.comm/ClassCode.hpp>
  15
  16 #include <anna/core/RuntimeException.hpp>
  17 #include <anna/comm/Message.hpp>
  18 #include <anna/core/util/Millisecond.hpp>
  19 #include <anna/core/functions.hpp>
  20
  21
  22
  23 namespace anna {
  24 namespace xml {
  25 class Node;
  26 }
  27 }
  28
  29
  30
  31 namespace anna {
  32
  33 namespace diameter {
  34
  35 namespace comm {
  36
  37
  38 class ClientSession;
  39 class ServerSession;
  40
  41 /**
  42    Messages launched to diameter servers
  43    Could be proxied (end-to-end kept) or not (by default).
  44 */
  45 class Message : public anna::comm::Message {
  46 public:
  47
  48   /**
  49    * Define las acciones a realizar en caso de que el temporizador de la petici�n expire.
  50    */
  51   struct OnExpiry { enum _v { Abandon, Ignore }; };
  52
  53   /**
  54      Constructor.
  55      \param onExpiry Indica la acci�n a realizar si el temporizador de esta transaci�n expira.
  56   */
  57   Message(const OnExpiry::_v onExpiry = OnExpiry::Ignore) : anna::comm::Message(StatusCodeBuffer::Reserve),
  58     a_classCode(ClassCode::ApplicationMessage),
  59     a_onExpiry(onExpiry) { initialize(); }
  60
  61   /**
  62      Devuelve el tipo de la clase de esta peticion indicada en el contructor.
  63      \return El tipo de la clase de esta peticion indicada en el contructor.
  64   */
  65   const ClassCode::_v & getClassCode() const throw() { return a_classCode; }
  66
  67   /**
  68    * Devuelve la acci�n a realizar en caso de que el temporizador asociado a esta petici�n expire.
  69    * \return la acci�n a realizar en caso de que el temporizador asociado a esta petici�n expire.
  70    */
  71   OnExpiry::_v getOnExpiry() const throw() { return a_onExpiry; }
  72
  73   /**
  74    * Establece la acci�n a realizar en caso de que el temporizador asociado a esta petici�n expire.
  75    * \param onExpiry Indica la acci�n a realizar en caso de que el temporizador asociado a esta petici�n expire.
  76    *
  77    * \warning Establecer el valor OnExpiry::Ignore podr�a causar p�rdida de memoria y uso innecesario de recursos.
  78    */
  79   void setOnExpiry(const OnExpiry::_v onExpiry) throw() { a_onExpiry = onExpiry; }
  80
  81   // Internal use (CER message)
  82   void setClassCode(const ClassCode::_v & classCode) throw() { a_classCode = classCode; }
  83
  84
  85   /**
  86      Class string representation
  87      \return String with relevant information for this instance.
  88   */
  89   virtual std::string asString() const throw();
  90
  91   /**
  92      Class xml representation
  93      \param parent Parent XML node on which hold this instance information.
  94      \return XML document with relevant information for this instance.
  95   */
  96   virtual anna::xml::Node* asXML(anna::xml::Node* parent) const throw();
  97
  98
  99   // Helpers
 100   HopByHop getHopByHop() const throw();
 101   EndToEnd getEndToEnd() const throw();
 102   HopByHop getRequestHopByHop() const throw() { return a_requestHopByHop; }
 103   EndToEnd getRequestEndToEnd() const throw() { return a_requestEndToEnd; }
 104   void setRequestHopByHop(HopByHop hbh) throw() { a_requestHopByHop = hbh; }
 105   void setRequestEndToEnd(EndToEnd ete) throw() { a_requestEndToEnd = ete; }
 106   CommandId getCommandId(bool &isRequest) const throw();
 107   CommandId getCommandId() const throw() { bool dummy; return getCommandId(dummy); }
 108
 109   bool fixRequestSequence(HopByHop hbh, EndToEnd ete) throw();
 110
 111   /** Diameter agents CANNOT modify the end-to-end. True value stands for intermediate agents, false for request originators */
 112   bool isProxied() const throw() { return a_proxied; }
 113
 114   /** Diameter agents CANNOT modify the end-to-end. True value stands for intermediate agents, false for request originators */
 115   void setProxied(bool proxied = true) throw() { a_proxied = proxied; }
 116
 117
 118   // Statistics
 119   void updateRequestTimestampMs(void) throw() { a_request_timestamp_ms = anna::functions::millisecond(); }
 120   const anna::Millisecond & getRequestTimestampMs() const throw() { return (a_request_timestamp_ms); }
 121
 122   int getRetries() const throw() { return a_retries; }
 123   void setRetries(int value) throw() { a_retries = value; }
 124
 125
 126   int getRequestServerSessionKey() const throw() { return a_requestServerSessionKey; }
 127
 128   /** Application specific socket id to keep origin track for request which came from a specific client, at asyncronous contexts (process with both diameter interfaces: client & entities) */
 129   void setRequestServerSessionKey(int value) throw() { a_requestServerSessionKey = value; }
 130
 131   const std::string & getRequestClientSessionKey() const throw() { return a_requestClientSessionKey; }
 132
 133   /** Application specific socket id to keep origin track for request which came from a specific server (entity), at asyncronous contexts (process with both diameter interfaces: client & entities) */
 134   void setRequestClientSessionKey(const std::string & value) throw() { a_requestClientSessionKey = value; }
 135
 136   /** Initializes class information */
 137   void initialize() throw() {
 138     a_retries = 0;
 139     a_requestServerSessionKey = -1; // means unknown/unset
 140     a_requestClientSessionKey = ""; // means unknown/unset
 141     a_requestHopByHop = 0;
 142     a_requestEndToEnd = 0;
 143     a_proxied = false;
 144   }
 145
 146
 147 protected:
 148   /**
 149      Constructor.
 150      \param classCode Tipo de clase de esta peticion.
 151      \param onExpiry Indica la acci�n a realizar si el temporizador de esta transaci�n expira.
 152   */
 153   Message(const ClassCode::_v & classCode, const OnExpiry::_v onExpiry = OnExpiry::Ignore) : anna::comm::Message(StatusCodeBuffer::Reserve),
 154     a_classCode(classCode),
 155     a_onExpiry(onExpiry) { initialize(); }
 156
 157
 158 private:
 159   ClassCode::_v a_classCode;
 160   OnExpiry::_v a_onExpiry;
 161   anna::Millisecond a_request_timestamp_ms;    // Lapsed timestamp milliseconds at request event. Used for statistic purposes.
 162   int a_retries;
 163   int a_requestServerSessionKey;            // useful to resolve origin of messages when diameter sessions are involved in the other side (if another interface, inherit from this Message class and put/carry context information)
 164   std::string a_requestClientSessionKey;    // idem for request which was received from servers
 165   HopByHop a_requestHopByHop; // application backup for hop-by-hop in order to restore on answer receive
 166   EndToEnd a_requestEndToEnd; // application backup for end-to-end in order to restore on answer receive
 167   bool a_proxied; // end-to-end will be kept
 168
 169   void send(ClientSession&) const throw(anna::RuntimeException);
 170   void send(ServerSession&) const throw(anna::RuntimeException);
 171   void restoreSequencesAfterFix() throw();
 172
 173   static const char* asText(const OnExpiry::_v) throw();
 174
 175   friend class Session;
 176   friend class ClientSession;
 177   friend class ServerSession;
 178 };
 179
 180 }
 181 }
 182 }
 183
 184 #endif
 185