Codebase list sipp / master actions.hpp
master

Tree @master (Download .tar.gz)

actions.hpp @masterraw · history · blame

/*
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program 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 General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 *  Authors : Benjamin GAUTHIER - 24 Mar 2004
 *            Joseph BANINO
 *            Olivier JACQUES
 *            Richard GAYRAUD
 *            From Hewlett Packard Company.
 */

#ifndef _CACTIONS
#define _CACTIONS

#include "variables.hpp"
#include "message.hpp"
class CSample;

#ifdef PCAPPLAY
#include "prepare_pcap.h"
#endif

#define MAX_ACTION_MESSAGE 3

class CAction
{
  public:
    enum T_ActionType
    {
      E_AT_NO_ACTION = 0,
      E_AT_ASSIGN_FROM_REGEXP,
      E_AT_CHECK,
      E_AT_ASSIGN_FROM_VALUE,
      E_AT_ASSIGN_FROM_SAMPLE,
      E_AT_ASSIGN_FROM_STRING,
      E_AT_ASSIGN_FROM_INDEX,
      E_AT_ASSIGN_FROM_GETTIMEOFDAY,
      E_AT_JUMP,
      E_AT_LOOKUP,
      E_AT_INSERT,
      E_AT_REPLACE,
      E_AT_PAUSE_RESTORE,
      E_AT_LOG_TO_FILE,
      E_AT_LOG_WARNING,
      E_AT_LOG_ERROR,
      E_AT_EXECUTE_CMD,
      E_AT_EXEC_INTCMD,
      E_AT_VAR_ADD,
      E_AT_VAR_SUBTRACT,
      E_AT_VAR_MULTIPLY,
      E_AT_VAR_DIVIDE,
      E_AT_VAR_TEST,
      E_AT_VAR_TO_DOUBLE,
      E_AT_VAR_STRCMP,
      E_AT_VAR_TRIM,
      E_AT_VERIFY_AUTH,
      E_AT_SET_DEST,
      E_AT_CLOSE_CON,
#ifdef PCAPPLAY
      E_AT_PLAY_PCAP_AUDIO,
      E_AT_PLAY_PCAP_VIDEO,
#endif
      E_AT_NB_ACTION
    };

    enum T_LookingPlace
    {
      E_LP_MSG = 0,
      E_LP_HDR,
      E_LP_BODY,
      E_LP_VAR,
      E_LP_NB_LOOKING_PLACE
    };

    enum T_Comparator
    {
      E_C_EQ,
      E_C_NE,
      E_C_GT,
      E_C_LT,
      E_C_GEQ,
      E_C_LEQ,
      E_C_NB_COMPARATOR
    };

    enum T_IntCmdType
    {
      E_INTCMD_INVALID = 0,
      E_INTCMD_STOPCALL,
      E_INTCMD_STOP_ALL,
      E_INTCMD_STOP_NOW
    };

    typedef struct _T_Action
    {
    } T_Action;

    void afficheInfo();
    const char *comparatorToString(T_Comparator comp);
    bool compare(VariableTable *variableTable);

    T_ActionType   getActionType();
    T_VarType      getVarType();
    T_LookingPlace getLookingPlace();
    T_Comparator   getComparator();
    bool           getCheckIt();
    bool           getCheckItInverse();
    bool           getCaseIndep();
    bool           getHeadersOnly();
    int            getVarId();
    int            getVarInId();
    int            getVarIn2Id();
    int            getOccurence();
    char*          getLookingChar();
    char*          getRegularExpression();
    SendingMessage *getMessage(int n = 0);  /* log specific function  */
    T_IntCmdType   getIntCmd();   /* exec specific function */
#ifdef PCAPPLAY
    pcap_pkts     *getPcapPkts(); /* send_packets specific function */
#endif

    void setActionType   (T_ActionType   P_value);
    void setLookingPlace (T_LookingPlace P_value);
    void setComparator   (T_Comparator   P_value);
    void setCheckIt      (bool           P_value);
    void setCheckItInverse (bool           P_value);
    void setVarId        (int            P_value);
    void setVarInId      (int            P_value);
    void setVarIn2Id      (int            P_value);
    void setLookingChar  (char*          P_value);
    void setAction       (CAction        P_action);
    void setCaseIndep    (bool           P_action);
    void setOccurence   (int            P_value);
    void setHeadersOnly  (bool           P_value);
    void setScenario     (scenario *     P_scenario);
    void setRegExp       (char*		 P_value);  /* ereg specific function. */
    int  executeRegExp   (char* P_string, VariableTable *P_callVarTable);
    void setMessage      (char*          P_value, int n = 0);  /* log specific function  */
    void setIntCmd       (T_IntCmdType   P_type );  /* exec specific function */
    void setDistribution (CSample *      P_value);  /* sample specific function  */
    void setDoubleValue  (double         P_value);  /* assign value specific function  */
    void setStringValue  (char *         P_value);  /* strcmp value specific function  */
#ifdef PCAPPLAY
    void setPcapArgs     (char *         P_value);  /* send_packets specific function */
    void setPcapArgs     (pcap_pkts   *  P_value);  /* send_packets specific function */
#endif

    void setSubVarId     (int P_value);
    int  getSubVarId     (int P_index);
    void setNbSubVarId   (int P_value);
    int  getNbSubVarId   ();
    int* getSubVarId() ;
    CSample *getDistribution ();  /* sample specific function  */
    double getDoubleValue ();  /* assign value specific function  */
    char * getStringValue ();  /* strcmp specific function  */

    CAction(scenario *scenario);
    ~CAction();

  private:
      T_ActionType   M_action;
      T_LookingPlace M_lookingPlace;
      T_Comparator   M_comp;
      bool           M_checkIt;
      bool           M_checkItInverse;
      bool           M_caseIndep;
      bool           M_headersOnly;
      int            M_varId;
      int            M_varInId;
      int            M_varIn2Id;
      int            M_occurence;
      int            M_nbSubVarId;
      int            M_maxNbSubVarId;
      int *          M_subVarId;

      char*          M_lookingChar;
      /* log specific member  */
      SendingMessage *M_message[MAX_ACTION_MESSAGE];
      char *	     M_message_str[MAX_ACTION_MESSAGE];
      /* exec specific member */
      T_IntCmdType   M_IntCmd;
      /* sample specific member. */
      CSample	     *M_distribution;
      /* assign value specific member. */
      double         M_doubleValue;
      /* strcmp specific member. */
      char *         M_stringValue;
      /* what scenario we belong to. */
      scenario *     M_scenario;
      /* Our regular expression. */
      bool	     M_regExpSet;
      regex_t        M_internalRegExp;
      char *         M_regularExpression;
#ifdef PCAPPLAY
      /* pcap specific member */
      pcap_pkts  *   M_pcapArgs;
#endif
      void setSubString(char** P_target, char* P_source, int P_start, int P_stop);
};

class CActions
{
  public:
    void afficheInfo();
    void setAction(CAction *P_action);
    void reset();
    int  getActionSize();
    CAction* getAction(int i);
    CActions();
    ~CActions();
  
  private:
    CAction ** M_actionList;
    int        M_nbAction;
    int        M_currentSettedAction;
};

#endif