You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
sip-tester/actions.cpp

571 lines
17 KiB

/*
* 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.
* Guillaume Teissier from FTR&D
*/
#include "sipp.hpp"
#include <assert.h>
#ifdef PCAPPLAY
#include "prepare_pcap.h"
#endif
static const char* strIntCmd(CAction::T_IntCmdType type)
{
switch (type)
{
case CAction::E_INTCMD_STOPCALL:
return "stop_call";
case CAction::E_INTCMD_STOP_ALL:
return "stop_gracefully";
case CAction::E_INTCMD_STOP_NOW:
return "stop_now";
default:
case CAction::E_INTCMD_INVALID:
return "invalid";
}
return "invalid";
}
const char * CAction::comparatorToString(T_Comparator comp) {
switch(comp) {
case E_C_EQ:
return "==";
case E_C_NE:
return "!=";
case E_C_GT:
return ">";
case E_C_LT:
return "<";
case E_C_GEQ:
return ">=";
case E_C_LEQ:
return "<=";
default:
return "invalid";
}
}
bool CAction::compare(VariableTable *variableTable) {
double lhs = variableTable->getVar(M_varInId)->getDouble();
double rhs = M_varIn2Id ? variableTable->getVar(M_varIn2Id)->getDouble() : M_doubleValue;
switch(M_comp) {
case E_C_EQ:
return lhs == rhs;
case E_C_NE:
return lhs != rhs;
case E_C_GT:
return lhs > rhs;
case E_C_LT:
return lhs < rhs;
case E_C_GEQ:
return lhs >= rhs;
case E_C_LEQ:
return lhs <= rhs;
default:
ERROR("Internal error: Invalid comparison type %d", M_comp);
return false; /* Shut up warning. */
}
}
void CAction::afficheInfo()
{
if (M_action == E_AT_ASSIGN_FROM_REGEXP) {
if(M_lookingPlace == E_LP_MSG) {
printf("Type[%d] - regexp[%s] where[%s] - checkIt[%d] - checkItInverse[%d] - $%s",
M_action,
M_regularExpression,
"Full Msg",
M_checkIt,
M_checkItInverse,
display_scenario->allocVars->getName(M_varId));
} else {
printf("Type[%d] - regexp[%s] where[%s-%s] - checkIt[%d] - checkItInverse[%d] - $%s",
M_action,
M_regularExpression,
"Header",
M_lookingChar,
M_checkIt,
M_checkItInverse, display_scenario->allocVars->getName(M_varId));
}
} else if (M_action == E_AT_EXECUTE_CMD) {
printf("Type[%d] - command[%-32.32s]", M_action, M_message_str[0]);
} else if (M_action == E_AT_EXEC_INTCMD) {
printf("Type[%d] - intcmd[%-32.32s]", M_action, strIntCmd(M_IntCmd));
} else if (M_action == E_AT_LOG_TO_FILE) {
printf("Type[%d] - message[%-32.32s]", M_action, M_message_str[0]);
} else if (M_action == E_AT_LOG_WARNING) {
printf("Type[%d] - warning[%-32.32s]", M_action, M_message_str[0]);
} else if (M_action == E_AT_LOG_ERROR) {
printf("Type[%d] - error[%-32.32s]", M_action, M_message_str[0]);
} else if (M_action == E_AT_ASSIGN_FROM_SAMPLE) {
char tmp[40];
M_distribution->textDescr(tmp, sizeof(tmp));
printf("Type[%d] - sample varId[%s] %s", M_action, display_scenario->allocVars->getName(M_varId), tmp);
} else if (M_action == E_AT_ASSIGN_FROM_VALUE) {
printf("Type[%d] - assign varId[%s] %lf", M_action, display_scenario->allocVars->getName(M_varId), M_doubleValue);
} else if (M_action == E_AT_ASSIGN_FROM_INDEX) {
printf("Type[%d] - assign index[%s]", M_action, display_scenario->allocVars->getName(M_varId));
} else if (M_action == E_AT_ASSIGN_FROM_GETTIMEOFDAY) {
printf("Type[%d] - assign gettimeofday[%s, %s]", M_action, display_scenario->allocVars->getName(M_varId), display_scenario->allocVars->getName(M_subVarId[0]));
} else if (M_action == E_AT_ASSIGN_FROM_STRING) {
printf("Type[%d] - string assign varId[%s] [%-32.32s]", M_action, display_scenario->allocVars->getName(M_varId), M_message_str[0]);
} else if (M_action == E_AT_JUMP) {
printf("Type[%d] - jump varInId[%s] %lf", M_action, display_scenario->allocVars->getName(M_varInId), M_doubleValue);
} else if (M_action == E_AT_PAUSE_RESTORE) {
printf("Type[%d] - restore pause varInId[%s] %lf", M_action, display_scenario->allocVars->getName(M_varInId), M_doubleValue);
} else if (M_action == E_AT_VAR_ADD) {
printf("Type[%d] - add varId[%s] %lf", M_action, display_scenario->allocVars->getName(M_varId), M_doubleValue);
} else if (M_action == E_AT_VAR_MULTIPLY) {
printf("Type[%d] - multiply varId[%s] %lf", M_action, display_scenario->allocVars->getName(M_varId), M_doubleValue);
} else if (M_action == E_AT_VAR_DIVIDE) {
printf("Type[%d] - divide varId[%s] %lf", M_action, display_scenario->allocVars->getName(M_varId), M_doubleValue);
} else if (M_action == E_AT_VAR_TRIM) {
printf("Type[%d] - trim varId[%s]", M_action, display_scenario->allocVars->getName(M_varId));
} else if (M_action == E_AT_VAR_TEST) {
printf("Type[%d] - divide varId[%s] varInId[%s] %s %lf", M_action, display_scenario->allocVars->getName(M_varId), display_scenario->allocVars->getName(M_varInId), comparatorToString(M_comp), M_doubleValue);
} else if (M_action == E_AT_VAR_TO_DOUBLE) {
printf("Type[%d] - toDouble varId[%s]", M_action, display_scenario->allocVars->getName(M_varId));
#ifdef PCAPPLAY
} else if ((M_action == E_AT_PLAY_PCAP_AUDIO) || (M_action == E_AT_PLAY_PCAP_VIDEO)) {
printf("Type[%d] - file[%s]", M_action, M_pcapArgs->file);
#endif
} else {
printf("Type[%d] - unknown action type ... ", M_action);
}
}
CAction::T_ActionType CAction::getActionType() { return(M_action); }
CAction::T_LookingPlace CAction::getLookingPlace() { return(M_lookingPlace); }
CAction::T_IntCmdType CAction::getIntCmd () { return(M_IntCmd); }
CAction::T_Comparator CAction::getComparator () { return(M_comp); }
bool CAction::getCheckIt() { return(M_checkIt); }
bool CAction::getCheckItInverse() { return(M_checkItInverse); }
bool CAction::getCaseIndep() { return(M_caseIndep); }
bool CAction::getHeadersOnly() { return(M_headersOnly); }
int CAction::getOccurence() { return(M_occurence); }
int CAction::getVarId() { return(M_varId); }
int CAction::getVarInId() { return(M_varInId); }
int CAction::getVarIn2Id() { return(M_varIn2Id); }
char* CAction::getLookingChar() { return(M_lookingChar); }
SendingMessage *CAction::getMessage(int n) { return(M_message[n]); }
CSample* CAction::getDistribution() { return(M_distribution); }
double CAction::getDoubleValue() { return(M_doubleValue); }
char* CAction::getStringValue() { return(M_stringValue); }
#ifdef PCAPPLAY
pcap_pkts * CAction::getPcapPkts() { return(M_pcapArgs); }
#endif
void CAction::setActionType (CAction::T_ActionType P_value)
{ M_action = P_value; }
void CAction::setLookingPlace (CAction::T_LookingPlace P_value)
{ M_lookingPlace = P_value; }
void CAction::setCheckIt (bool P_value)
{ M_checkIt = P_value; }
void CAction::setCheckItInverse (bool P_value)
{ M_checkItInverse = P_value; }
void CAction::setVarId (int P_value)
{ M_varId = P_value; }
void CAction::setVarInId (int P_value)
{ M_varInId = P_value; }
void CAction::setVarIn2Id (int P_value)
{ M_varIn2Id = P_value; }
void CAction::setCaseIndep (bool P_value)
{ M_caseIndep = P_value; }
void CAction::setOccurence (int P_value)
{ M_occurence = P_value; }
void CAction::setHeadersOnly (bool P_value)
{ M_headersOnly = P_value; }
void CAction::setIntCmd (T_IntCmdType P_type)
{ M_IntCmd = P_type; }
void CAction::setComparator (T_Comparator P_value)
{ M_comp = P_value; }
/* sample specific function. */
void CAction::setDistribution (CSample *P_value)
{ M_distribution = P_value; }
/* assign from value specific function. */
void CAction::setDoubleValue (double P_value)
{ M_doubleValue = P_value; }
/* strcmp specific function. */
void CAction::setStringValue (char *P_value)
{ M_stringValue = P_value; }
void CAction::setSubVarId (int P_value) {
if ( M_nbSubVarId < M_maxNbSubVarId ) {
M_subVarId[M_nbSubVarId] = P_value;
M_nbSubVarId++;
}
}
int CAction::getSubVarId(int P_index) {
return(M_subVarId[P_index]);
}
int* CAction::getSubVarId() {
return(M_subVarId);
}
void CAction::setNbSubVarId (int P_value) {
M_maxNbSubVarId = P_value;
if(M_subVarId != NULL) {
delete [] M_subVarId;
M_subVarId = NULL;
}
M_subVarId = new int[M_maxNbSubVarId] ;
M_nbSubVarId = 0 ;
}
int CAction::getNbSubVarId () {
return(M_nbSubVarId);
}
void CAction::setLookingChar (char* P_value)
{
if(M_lookingChar != NULL)
{
delete [] M_lookingChar;
M_lookingChar = NULL;
}
if(P_value != NULL)
{
M_lookingChar = new char[strlen(P_value)+1];
strcpy(M_lookingChar, P_value);
}
}
void CAction::setMessage (char* P_value, int n)
{
if(M_message[n] != NULL)
{
delete M_message[n];
M_message[n] = NULL;
}
free(M_message_str[n]);
M_message_str[n] = NULL;
if(P_value != NULL)
{
M_message_str[n] = strdup(P_value);
M_message[n] = new SendingMessage(M_scenario, P_value, true /* skip sanity */);
}
}
void CAction::setRegExp(char *P_value) {
int errorCode;
M_regularExpression = strdup(P_value);
M_regExpSet = true;
errorCode = regcomp(&(M_internalRegExp), M_regularExpression, REGEXP_PARAMS);
if(errorCode != 0)
{
char buffer[MAX_HEADER_LEN];
regerror(errorCode, &M_internalRegExp, buffer, sizeof(buffer));
ERROR("recomp error : regular expression '%s' - error '%s'\n", M_regularExpression, buffer);
}
}
char *CAction::getRegularExpression() {
if (!M_regExpSet) {
ERROR("Trying to get a regular expression for an action that does not have one!");
}
return M_regularExpression;
}
int CAction::executeRegExp(char* P_string, VariableTable *P_callVarTable)
{
regmatch_t pmatch[10];
int error;
int nbOfMatch = 0;
char* result = NULL ;
if (!M_regExpSet) {
ERROR("Trying to perform regular expression match on action that does not have one!");
}
if (getNbSubVarId() > 9) {
ERROR("You can only have nine sub expressions!");
}
memset((void*)pmatch, 0, sizeof(regmatch_t)*10);
error = regexec(&(M_internalRegExp), P_string, 10, pmatch, REGEXP_PARAMS);
if ( error == 0) {
CCallVariable* L_callVar = P_callVarTable->getVar(getVarId());
for(int i = 0; i <= getNbSubVarId(); i++) {
if(pmatch[i].rm_eo == -1) break ;
setSubString(&result, P_string, pmatch[i].rm_so, pmatch[i].rm_eo);
L_callVar->setMatchingValue(result);
if (i == getNbSubVarId())
break ;
L_callVar = P_callVarTable->getVar(getSubVarId(i));
}
}
return(nbOfMatch);
}
void CAction::setSubString(char** P_target, char* P_source, int P_start, int P_stop)
{
int sizeOf;
if(P_source != NULL) {
sizeOf = P_stop - P_start;
(*P_target) = new char[sizeOf + 1];
if (sizeOf > 0) {
memcpy((*P_target), &(P_source[P_start]), sizeOf);
}
(*P_target)[sizeOf] = '\0';
} else {
*P_target = NULL ;
}
}
#ifdef PCAPPLAY
void CAction::setPcapArgs (pcap_pkts * P_value)
{
if(M_pcapArgs != NULL)
{
free(M_pcapArgs);
M_pcapArgs = NULL;
}
if(P_value != NULL)
{
M_pcapArgs = (pcap_pkts *)malloc(sizeof(*M_pcapArgs));
memcpy(M_pcapArgs, P_value, sizeof(*M_pcapArgs));
}
}
void CAction::setPcapArgs (char* P_value)
{
if(M_pcapArgs != NULL)
{
free(M_pcapArgs);
M_pcapArgs = NULL;
}
if(P_value != NULL)
{
M_pcapArgs = (pcap_pkts *) malloc(sizeof(*M_pcapArgs));
if (parse_play_args(P_value, M_pcapArgs) == -1)
{
ERROR("Play pcap error");
}
if (access(M_pcapArgs->file, F_OK)) {
ERROR("Cannot read file %s\n", M_pcapArgs->file);
}
}
}
#endif
void CAction::setScenario(scenario * P_scenario) {
M_scenario = P_scenario;
}
void CAction::setAction(CAction P_action)
{
if (P_action.getActionType() == CAction::E_AT_ASSIGN_FROM_SAMPLE) {
assert(P_action.getDistribution() != NULL);
}
int L_i;
setActionType ( P_action.getActionType() );
setLookingPlace ( P_action.getLookingPlace() );
setVarId ( P_action.getVarId() );
setVarInId ( P_action.getVarInId() );
setDoubleValue ( P_action.getDoubleValue() );
setDistribution ( P_action.getDistribution() );
setScenario ( P_action.M_scenario );
setNbSubVarId ( P_action.getNbSubVarId() );
for (L_i = 0; L_i < P_action.getNbSubVarId() ; L_i++ ) {
setSubVarId (P_action.getSubVarId(L_i));
}
setLookingChar ( P_action.getLookingChar() );
setCheckIt ( P_action.getCheckIt() );
setCheckItInverse ( P_action.getCheckItInverse() );
setCaseIndep ( P_action.getCaseIndep() );
setOccurence ( P_action.getOccurence() );
setHeadersOnly ( P_action.getHeadersOnly() );
for (L_i = 0; L_i < MAX_ACTION_MESSAGE; L_i++) {
setMessage(P_action.M_message_str[L_i], L_i);
}
setRegExp ( P_action.M_regularExpression);
setIntCmd ( P_action.M_IntCmd );
#ifdef PCAPPLAY
setPcapArgs ( P_action.M_pcapArgs );
#endif
}
CAction::CAction(scenario *scenario)
{
M_action = E_AT_NO_ACTION;
M_varId = 0;
M_varInId = 0;
M_varIn2Id = 0;
M_nbSubVarId = 0;
M_maxNbSubVarId = 0;
M_subVarId = NULL;
M_checkIt = false;
M_checkItInverse = false;
M_lookingPlace = E_LP_MSG;
M_lookingChar = NULL;
M_caseIndep = false;
M_occurence = 1;
M_headersOnly = true;
for (int i = 0; i < MAX_ACTION_MESSAGE; i++) {
M_message[i] = NULL;
M_message_str[i] = NULL;
}
M_IntCmd = E_INTCMD_INVALID;
M_doubleValue = 0;
M_stringValue = NULL;
M_distribution = NULL;
#ifdef PCAPPLAY
M_pcapArgs = NULL;
#endif
M_scenario = scenario;
M_regExpSet = false;
M_regularExpression = NULL;
}
CAction::~CAction()
{
if(M_lookingChar != NULL)
{
delete [] M_lookingChar;
M_lookingChar = NULL;
}
for (int i = 0; i < MAX_ACTION_MESSAGE; i++) {
if(M_message[i] != NULL)
{
delete M_message[i];
M_message[i] = NULL;
}
free(M_message_str[i]);
M_message_str[i] = NULL;
}
if(M_subVarId != NULL)
{
delete [] M_subVarId;
M_subVarId = NULL;
}
free(M_stringValue);
#ifdef PCAPPLAY
if (M_pcapArgs != NULL) {
free(M_pcapArgs);
M_pcapArgs = NULL;
}
#endif
if (M_regExpSet) {
regfree(&M_internalRegExp);
free(M_regularExpression);
}
if (M_distribution) {
delete M_distribution;
}
}
/****************************** CActions class ************************/
void CActions::afficheInfo()
{
printf("Action Size = [%d]\n", M_nbAction);
for(int i=0; i<M_nbAction; i++)
{
printf("actionlist[%d] : \n", i);
M_actionList[i]->afficheInfo();
}
}
void CActions::reset()
{
for (int i = 0; i < M_nbAction; i++) {
delete M_actionList[i];
M_actionList[i] = NULL;
}
M_nbAction = 0;
}
int CActions::getActionSize()
{
return(M_nbAction);
}
void CActions::setAction(CAction *P_action)
{
CAction **newActions = new CAction*[M_nbAction + 1];
if (!newActions) {
ERROR("Could not allocate new action list.");
}
for (int i = 0; i < M_nbAction; i++) {
newActions[i] = M_actionList[i];
}
if (M_actionList) {
delete [] M_actionList;
}
M_actionList = newActions;
M_actionList[M_nbAction] = P_action;
M_nbAction++;
}
CAction* CActions::getAction(int i)
{
if(i < M_nbAction)
{
return(M_actionList[i]);
}
else
return(NULL);
}
CActions::CActions()
{
M_nbAction = 0;
M_actionList = NULL;
}
CActions::~CActions()
{
for (int i = 0; i < M_nbAction; i++) {
delete M_actionList[i];
}
delete [] M_actionList;
M_actionList = NULL;
}