pana_client_fsm.cxx

/* BEGIN_COPYRIGHT                                                        */
/*                                                                        */
/* Open Diameter: Open-source software for the Diameter and               */
/*                Diameter related protocols                              */
/*                                                                        */
/* Copyright (C) 2002-2004 Open Diameter Project                          */
/*                                                                        */
/* This library is free software; you can redistribute it and/or modify   */
/* it under the terms of the GNU Lesser General Public License as         */
/* published by the Free Software Foundation; either version 2.1 of the   */
/* License, or (at your option) any later version.                        */
/*                                                                        */
/* This library is distributed in the hope that it will be useful,        */
/* but WITHOUT ANY WARRANTY; without even the implied warranty of         */
/* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU      */
/* Lesser General Public License for more details.                        */
/*                                                                        */
/* You should have received a copy of the GNU Lesser General Public       */
/* License along with this library; if not, write to the Free Software    */
/* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307    */
/* USA.                                                                   */
/*                                                                        */
/* In addition, when you copy and redistribute some or the entire part of */
/* the source code of this software with or without modification, you     */
/* MUST include this copyright notice in each copy.                       */
/*                                                                        */
/* If you make any changes that are appeared to be useful, please send    */
/* sources that include the changed part to                               */
/* diameter-developers@lists.sourceforge.net so that we can reflect your  */
/* changes to one unified version of this software.                       */
/*                                                                        */
/* END_COPYRIGHT                                                          */


#include "pana_client_fsm.h"

PANA_ClientStateTable PANA_PacSession::m_StateTable;
typedef PANA_SessionRxInterface<PANA_Client> PANA_ClientSessionRxInterface;

class PANA_CsmRxStartMsg : public PANA_ClientSessionRxInterface
{
   public:
      PANA_CsmRxStartMsg(PANA_Client &c, PANA_PacSession &s) : 
         PANA_ClientSessionRxInterface(c), m_Session(s) { }
      virtual void operator()(PANA_Message &msg) { 
         // first level validation
         m_arg.ValidateMessage(msg);
         // second level validation
         if (! msg.flags().request) {
            throw (PANA_Exception(PANA_Exception::INVALID_MESSAGE, 
                                 "client received invalid message in discovery state"));
         }
         // validate sequence number
         if (msg.tseq() == 0 || msg.rseq() != 0) {
            throw (PANA_Exception(PANA_Exception::INVALID_MESSAGE, 
                                 "Rseq or Tseq field of start answer message is zero"));
         }
         m_arg.m_RxMessageQueue.push_back(&msg);
         m_Session.Notify(PANA_RX_START_REQUEST); 
      }
      virtual PANA_ClientSessionRxInterface *clone() { return (new PANA_CsmRxStartMsg(*this)); }
   private:
      PANA_PacSession &m_Session;
};

class PANA_CsmRxAuthMsg : public PANA_ClientSessionRxInterface
{
   public:
      PANA_CsmRxAuthMsg(PANA_Client &c, PANA_PacSession &s) : 
         PANA_ClientSessionRxInterface(c), m_Session(s) { }
      virtual void operator()(PANA_Message &msg) { 
         // first level validation
         m_arg.ValidateMessage(msg);
         // second level validation
         if (! msg.flags().request) {
            throw (PANA_Exception(PANA_Exception::INVALID_MESSAGE, 
                                 "client received invalid message"));
         }
         m_arg.m_RxMessageQueue.push_back(&msg); 
         m_Session.Notify(PANA_RX_AUTH_REQUEST); 
      }
      virtual PANA_ClientSessionRxInterface *clone() { return (new PANA_CsmRxAuthMsg(*this)); }
   private:
      PANA_PacSession &m_Session;
};

class PANA_CsmRxBindMsg : public PANA_ClientSessionRxInterface
{
   public:
      PANA_CsmRxBindMsg(PANA_Client &c, PANA_PacSession &s) : 
         PANA_ClientSessionRxInterface(c), m_Session(s) { }
      virtual void operator()(PANA_Message &msg) {
         // first level validation
         m_arg.ValidateMessage(msg);
         // second level validation
         if (! msg.flags().request) {
            throw (PANA_Exception(PANA_Exception::INVALID_MESSAGE, 
                                 "client received invalid message"));
         }
         // compute the event
         AAAAvpContainer* c_rcode = msg.avpList().search("Result-Code");
         if (c_rcode == NULL) {
            throw (PANA_Exception(PANA_Exception::INVALID_MESSAGE, 
                                 "PANA client received bind msg with no result code"));
         }
         diameter_unsigned32_t &rcode = (*c_rcode)[0]->dataRef(Type2Type<diameter_unsigned32_t>());
    
         AAA_Event ev = PANA_RX_BIND_FAILURE;
         switch (m_arg.BindResult(rcode)) {
            case PANA_BIND_SUCCESS_INITIAL:
            case PANA_BIND_SUCCESS_FINAL: ev = PANA_RX_BIND_FINAL; break;
            case PANA_BIND_INTERIM:       ev = PANA_RX_BIND_FIRST; break;
            case PANA_BIND_FAILED:        ev = PANA_RX_BIND_FAILURE; break;
            default: break;
               throw (PANA_Exception(PANA_Exception::INVALID_MESSAGE, 
                                    "PANA client received invalid bind event"));
         }
         m_arg.m_RxMessageQueue.push_back(&msg);         
         m_Session.Notify(ev);
      };
      virtual PANA_ClientSessionRxInterface *clone() { return (new PANA_CsmRxBindMsg(*this)); }
   private:
      PANA_PacSession &m_Session;
};

class PANA_CsmRxReAuthMsg : public PANA_ClientSessionRxInterface
{
   public:
      PANA_CsmRxReAuthMsg(PANA_Client &c, PANA_PacSession &s) : 
         PANA_ClientSessionRxInterface(c), m_Session(s) { }
      virtual void operator()(PANA_Message &msg) {
         // first level validation
         m_arg.ValidateMessage(msg);
         // second level validation
         if (msg.flags().request && m_Session.IsSessionResumption()) {
            m_Session.Notify(PANA_RX_REAUTH_REQUEST);
         }
         else {
            m_arg.m_RxMessageQueue.push_back(&msg); 
            m_Session.Notify(PANA_RX_REAUTH_ANSWER); 
         }
      };
      virtual PANA_ClientSessionRxInterface *clone() { return (new PANA_CsmRxReAuthMsg(*this)); }
   private:
      PANA_PacSession &m_Session;
};

class PANA_CsmRxTermMsg : public PANA_ClientSessionRxInterface
{
   public:
      PANA_CsmRxTermMsg(PANA_Client &c, PANA_PacSession &s) : 
         PANA_ClientSessionRxInterface(c), m_Session(s) { }
      virtual void operator()(PANA_Message &msg) {
         // first level validation
         m_arg.ValidateMessage(msg);
         // second level validation
         m_arg.m_RxMessageQueue.push_back(&msg); 
         (msg.flags().request) ? m_Session.Notify(PANA_RX_TERM_REQUEST) :
                                 m_Session.Notify(PANA_RX_TERM_ANSWER); 
      };
      virtual PANA_ClientSessionRxInterface *clone() { return (new PANA_CsmRxTermMsg(*this)); }
   private:
      PANA_PacSession &m_Session;
};

void PANA_PacSession::InitializeMsgMaps()
{
   PANA_CsmRxStartMsg startMsg(m_Client, *this);
   m_MsgHandlers.Register(PANA_MTYPE_START, startMsg);

   PANA_CsmRxAuthMsg authMsg(m_Client, *this);
   m_MsgHandlers.Register(PANA_MTYPE_AUTH, authMsg);

   PANA_CsmRxBindMsg bindMsg(m_Client, *this);
   m_MsgHandlers.Register(PANA_MTYPE_BIND, bindMsg);

   PANA_CsmRxReAuthMsg reAuthMsg(m_Client, *this);
   m_MsgHandlers.Register(PANA_MTYPE_REAUTH, reAuthMsg);

   PANA_CsmRxTermMsg termMsg(m_Client, *this);
   m_MsgHandlers.Register(PANA_MTYPE_TERM, termMsg);
}

void PANA_PacSession::FlushMsgMaps()
{
   m_MsgHandlers.Remove(PANA_MTYPE_START);
   m_MsgHandlers.Remove(PANA_MTYPE_AUTH);
   m_MsgHandlers.Remove(PANA_MTYPE_BIND);
   m_MsgHandlers.Remove(PANA_MTYPE_REAUTH);
   m_MsgHandlers.Remove(PANA_MTYPE_TERM);
}

---