example/diameter/launcher/MyDiameterEntity.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 example_diameter_launcher_MyDiameterEntity_hpp
  10 #define example_diameter_launcher_MyDiameterEntity_hpp
  11
  12 // Project
  13 #include <anna/diameter.comm/Entity.hpp>
  14
  15 // Process
  16 #include "ProgrammedAnswers.hpp"
  17
  18 namespace anna {
  19   namespace diameter {
  20     namespace codec {
  21       class Engine;
  22     }
  23   }
  24 }
  25
  26 class MyDiameterEntity : public anna::diameter::comm::Entity {
  27
  28   anna::diameter::codec::Engine * a_codecEngine; // for automatic answers (failed-avp), write logs, etc.
  29
  30   void eventRequestRetransmission(const anna::diameter::comm::ClientSession *, anna::diameter::comm::Message*) throw();
  31   void eventResponse(const anna::diameter::comm::Response&) throw(anna::RuntimeException);
  32   void eventRequest(anna::diameter::comm::ClientSession *, const anna::DataBlock&) throw(anna::RuntimeException);
  33   void eventUnknownResponse(anna::diameter::comm::ClientSession *, const anna::DataBlock&) throw(anna::RuntimeException);
  34   void eventDPA(anna::diameter::comm::ClientSession *, const anna::DataBlock&) throw(anna::RuntimeException);
  35
  36   // Reimplementation
  37   int readSocketId(const anna::diameter::comm::Message* message, int maxClientSessions) const throw();
  38
  39 public:
  40
  41   ProgrammedAnswers a_reactingAnswers;
  42   void setCodecEngine(anna::diameter::codec::Engine *codecEngine) throw() { a_codecEngine = codecEngine; a_reactingAnswers.setCodecEngine(codecEngine); }
  43   ProgrammedAnswers *getReactingAnswers() throw() { return (ProgrammedAnswers *)&a_reactingAnswers; }
  44 };
  45
  46 #endif