4f80ad5021
git-svn-id: http://yate.null.ro/svn/yate/trunk@5201 acf43c95-373e-0410-b603-e72c3f656dc1
902 lines
24 KiB
C++
902 lines
24 KiB
C++
/**
|
|
* regexroute.cpp
|
|
* This file is part of the YATE Project http://YATE.null.ro
|
|
*
|
|
* Regular expressions based routing
|
|
*
|
|
* Yet Another Telephony Engine - a fully featured software PBX and IVR
|
|
* Copyright (C) 2004-2006 Null Team
|
|
*
|
|
* 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#include <yatephone.h>
|
|
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
using namespace TelEngine;
|
|
namespace { // anonymous
|
|
|
|
#define DEFAULT_RULE "^\\(false\\|no\\|off\\|disable\\|f\\|0*\\)$^"
|
|
#define BLOCK_STACK 10
|
|
|
|
static Configuration s_cfg;
|
|
static const char* s_trackName = 0;
|
|
static bool s_extended;
|
|
static bool s_insensitive;
|
|
static bool s_prerouteall;
|
|
static int s_maxDepth = 5;
|
|
static String s_defRule;
|
|
static Mutex s_mutex(true,"RegexRoute");
|
|
static ObjList s_extra;
|
|
static NamedList s_vars("");
|
|
static int s_dispatching = 0;
|
|
|
|
class RouteHandler : public MessageHandler
|
|
{
|
|
public:
|
|
RouteHandler(int prio)
|
|
: MessageHandler("call.route",prio,s_trackName)
|
|
{ }
|
|
virtual bool received(Message &msg);
|
|
};
|
|
|
|
static String& vars(String& s, String* vName = 0)
|
|
{
|
|
if (s.startSkip("$",false)) {
|
|
s.trimBlanks();
|
|
if (vName)
|
|
*vName = s;
|
|
s = s_vars.getValue(s);
|
|
}
|
|
return s;
|
|
}
|
|
|
|
enum {
|
|
OPER_ADD,
|
|
OPER_SUB,
|
|
OPER_MUL,
|
|
OPER_DIV,
|
|
OPER_MOD,
|
|
OPER_EQ,
|
|
OPER_NE,
|
|
OPER_GT,
|
|
OPER_LT,
|
|
OPER_GE,
|
|
OPER_LE,
|
|
};
|
|
|
|
static void mathOper(String& str, String& par, int sep, int oper)
|
|
{
|
|
str = par.substr(0,sep);
|
|
par = par.substr(sep+1);
|
|
int len = str.length();
|
|
sep = par.find(',');
|
|
if (sep >= 0) {
|
|
String tmp = par.substr(sep+1);
|
|
len = vars(tmp).toInteger();
|
|
par = par.substr(0,sep);
|
|
}
|
|
int p1 = vars(str).toInteger(0,10);
|
|
int p2 = vars(par).toInteger(0,10);
|
|
switch (oper) {
|
|
case OPER_ADD:
|
|
str = p1+p2;
|
|
break;
|
|
case OPER_SUB:
|
|
str = p1-p2;
|
|
break;
|
|
case OPER_MUL:
|
|
str = p1*p2;
|
|
break;
|
|
case OPER_DIV:
|
|
str = p2 ? p1/p2 : 0;
|
|
break;
|
|
case OPER_MOD:
|
|
str = p2 ? p1%p2 : 0;
|
|
break;
|
|
case OPER_EQ:
|
|
str = (p1 == p2);
|
|
len = 0;
|
|
return;
|
|
case OPER_NE:
|
|
str = (p1 != p2);
|
|
len = 0;
|
|
return;
|
|
case OPER_GT:
|
|
str = (p1 > p2);
|
|
len = 0;
|
|
return;
|
|
case OPER_LT:
|
|
str = (p1 < p2);
|
|
len = 0;
|
|
return;
|
|
case OPER_GE:
|
|
str = (p1 >= p2);
|
|
len = 0;
|
|
return;
|
|
case OPER_LE:
|
|
str = (p1 <= p2);
|
|
len = 0;
|
|
return;
|
|
}
|
|
len -= (int)str.length();
|
|
if (len > 0) {
|
|
// left pad the result to the desired length
|
|
String tmp('0',len);
|
|
if (str[0] == '-')
|
|
str = "-" + tmp + str.substr(1);
|
|
else
|
|
str = tmp + str;
|
|
}
|
|
}
|
|
|
|
static void evalFunc(String& str, Message& msg)
|
|
{
|
|
if (str.null())
|
|
str = ";";
|
|
else if (str == "$")
|
|
;
|
|
else if (str.startSkip("++",false)) {
|
|
String tmp;
|
|
str = vars(str,&tmp).toInteger(0,10) + 1;
|
|
if (tmp)
|
|
s_vars.setParam(tmp,str);
|
|
}
|
|
else if (str.startSkip("--",false)) {
|
|
String tmp;
|
|
str = vars(str,&tmp).toInteger(0,10) - 1;
|
|
if (tmp)
|
|
s_vars.setParam(tmp,str);
|
|
}
|
|
else {
|
|
bool bare = true;
|
|
int sep = str.find(',');
|
|
String par;
|
|
if (sep > 0) {
|
|
bare = false;
|
|
par = str.substr(sep+1);
|
|
str = str.substr(0,sep);
|
|
sep = par.find(',');
|
|
}
|
|
if (str == "length")
|
|
str = vars(par).length();
|
|
else if (str == "upper")
|
|
str = vars(par).toUpper();
|
|
else if (str == "lower")
|
|
str = vars(par).toLower();
|
|
else if (str == "chr")
|
|
str = static_cast<char>(0xff & vars(par).toInteger());
|
|
else if ((sep >= 0) && ((str == "streq") || (str == "strne"))) {
|
|
bool ret = (str == "strne");
|
|
str = par.substr(sep+1);
|
|
par = par.substr(0,sep);
|
|
vars(str);
|
|
vars(par);
|
|
ret ^= (str == par);
|
|
str = ret;
|
|
}
|
|
else if ((sep >= 0) && (str == "strpos")) {
|
|
str = par.substr(sep+1);
|
|
par = par.substr(0,sep);
|
|
vars(str);
|
|
vars(par);
|
|
str = str.find(par);
|
|
}
|
|
else if ((sep >= 0) && ((str == "add") || (str == "+")))
|
|
mathOper(str,par,sep,OPER_ADD);
|
|
else if ((sep >= 0) && ((str == "sub") || (str == "-")))
|
|
mathOper(str,par,sep,OPER_SUB);
|
|
else if ((sep >= 0) && ((str == "mul") || (str == "*")))
|
|
mathOper(str,par,sep,OPER_MUL);
|
|
else if ((sep >= 0) && ((str == "div") || (str == "/")))
|
|
mathOper(str,par,sep,OPER_DIV);
|
|
else if ((sep >= 0) && ((str == "mod") || (str == "%")))
|
|
mathOper(str,par,sep,OPER_MOD);
|
|
else if ((sep >= 0) && (str == "eq"))
|
|
mathOper(str,par,sep,OPER_EQ);
|
|
else if ((sep >= 0) && (str == "ne"))
|
|
mathOper(str,par,sep,OPER_NE);
|
|
else if ((sep >= 0) && ((str == "gt") || (str == ">")))
|
|
mathOper(str,par,sep,OPER_GT);
|
|
else if ((sep >= 0) && ((str == "lt") || (str == "<")))
|
|
mathOper(str,par,sep,OPER_LT);
|
|
else if ((sep >= 0) && (str == "ge"))
|
|
mathOper(str,par,sep,OPER_GE);
|
|
else if ((sep >= 0) && (str == "le"))
|
|
mathOper(str,par,sep,OPER_LE);
|
|
else if (str == "random") {
|
|
str.clear();
|
|
vars(par);
|
|
for (unsigned int i = 0; i < par.length(); i++) {
|
|
if (par.at(i) == '?')
|
|
str << (int)(Random::random() % 10);
|
|
else
|
|
str << par.at(i);
|
|
}
|
|
}
|
|
else if (str == "hex") {
|
|
int len = 0;
|
|
if (sep >= 0) {
|
|
len = par.substr(sep+1).toInteger();
|
|
par = par.substr(0,sep);
|
|
}
|
|
int val = par.toInteger();
|
|
unsigned char buf[4];
|
|
buf[0] = (unsigned char)val;
|
|
buf[1] = (unsigned char)(val >> 8);
|
|
buf[2] = (unsigned char)(val >> 16);
|
|
buf[3] = (unsigned char)(val >> 24);
|
|
if (len > 4)
|
|
len = 4;
|
|
else if (len <= 0) {
|
|
if (buf[3])
|
|
len = 4;
|
|
else if (buf[2])
|
|
len = 3;
|
|
else if (buf[1])
|
|
len = 2;
|
|
else
|
|
len = 1;
|
|
}
|
|
str.hexify(&buf,len,' ');
|
|
}
|
|
else if ((sep > 0) && ((str == "index") || (str == "rotate"))) {
|
|
bool rotate = (str == "rotate");
|
|
String vname;
|
|
str = par.substr(0,sep);
|
|
par = par.substr(sep+1).trimBlanks();
|
|
int idx = vars(str,&vname).toInteger(0,10);
|
|
ObjList* lst = par.split(',');
|
|
str.clear();
|
|
par.clear();
|
|
unsigned int n = lst->count();
|
|
if (n) {
|
|
int i = idx % n;
|
|
for (ObjList* l = lst->skipNull(); l; l = l->skipNext()) {
|
|
String* s = static_cast<String*>(l->get());
|
|
vars(*s);
|
|
if (rotate) {
|
|
if (i > 0)
|
|
par.append(*s," ");
|
|
else
|
|
str.append(*s," ");
|
|
}
|
|
else if (0 == i) {
|
|
str = *s;
|
|
break;
|
|
}
|
|
i--;
|
|
}
|
|
str.append(par," ");
|
|
// auto increment the index variable if any
|
|
if (vname) {
|
|
par = (idx + 1) % n;
|
|
s_vars.setParam(vname,par);
|
|
}
|
|
}
|
|
lst->destruct();
|
|
}
|
|
else if ((sep >= 0) && (str == "config")) {
|
|
str = par.substr(0,sep).trimBlanks();
|
|
par = par.substr(sep+1).trimBlanks();
|
|
str = Engine::config().getValue(str,par);
|
|
}
|
|
else if (str == "engine")
|
|
str = Engine::runParams().getValue(vars(par));
|
|
else if (str == "message") {
|
|
if (sep >= 0) {
|
|
str = par.substr(sep+1).trimBlanks();
|
|
par = par.substr(0,sep).trimBlanks();
|
|
}
|
|
else
|
|
str.clear();
|
|
if (par.null() || par == "name")
|
|
str = msg;
|
|
else if (par == "time")
|
|
str = msg.msgTime().sec();
|
|
else if (par == "broadcast")
|
|
str = msg.broadcast();
|
|
else if (par == "count")
|
|
str = msg.count();
|
|
else if (par == "parameters") {
|
|
par = str;
|
|
if (par.null())
|
|
par = ",";
|
|
str.clear();
|
|
unsigned int n = msg.length();
|
|
for (unsigned int i = 0; i < n; i++) {
|
|
NamedString* s = msg.getParam(i);
|
|
if (s && s->name())
|
|
str.append(s->name(),par);
|
|
}
|
|
}
|
|
else
|
|
str.clear();
|
|
}
|
|
else if (str == "runid") {
|
|
str.clear();
|
|
str << Engine::runId();
|
|
}
|
|
else if (str == "nodename")
|
|
str = Engine::nodeName();
|
|
else if (str == "threadname")
|
|
str = Thread::currentName();
|
|
else if ((sep >= 0) && (str == "transcode")) {
|
|
str = par.substr(0,sep);
|
|
par = par.substr(sep+1).trimBlanks();
|
|
ObjList* fmts = DataTranslator::allFormats(par,
|
|
(str.find('e') < 0),
|
|
(str.find('r') < 0),
|
|
(str.find('c') < 0));
|
|
str.clear();
|
|
str.append(fmts,",");
|
|
TelEngine::destruct(fmts);
|
|
}
|
|
else if (str == "dispatching")
|
|
str = s_dispatching;
|
|
else if (bare && str.trimBlanks())
|
|
str = s_vars.getValue(str);
|
|
else {
|
|
Debug("RegexRoute",DebugWarn,"Invalid function '%s'",str.c_str());
|
|
str.clear();
|
|
}
|
|
}
|
|
}
|
|
|
|
// handle $(function) replacements
|
|
static void replaceFuncs(String &str, Message& msg)
|
|
{
|
|
int p1;
|
|
while ((p1 = str.find("$(")) >= 0) {
|
|
int p2 = str.find(')',p1+2);
|
|
if (p2 > 0) {
|
|
String v = str.substr(p1+2,p2-p1-2);
|
|
v.trimBlanks();
|
|
DDebug("RegexRoute",DebugAll,"Replacing function '%s'",
|
|
v.c_str());
|
|
evalFunc(v,msg);
|
|
str = str.substr(0,p1) + v + str.substr(p2+1);
|
|
}
|
|
else {
|
|
Debug("RegexRoute",DebugWarn,"Unmatched function end: '%s'",str.c_str()+p1);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
// handle ;paramname[=value] assignments
|
|
static void setMessage(const String& match, Message& msg, String& line, Message* target = 0)
|
|
{
|
|
if (!target)
|
|
target = &msg;
|
|
ObjList *strs = line.split(';');
|
|
bool first = true;
|
|
for (ObjList *p = strs; p; p=p->next()) {
|
|
String *s = static_cast<String*>(p->get());
|
|
if (s) {
|
|
*s = match.replaceMatches(*s);
|
|
msg.replaceParams(*s);
|
|
replaceFuncs(*s,msg);
|
|
}
|
|
if (first) {
|
|
first = false;
|
|
line = s ? *s : String::empty();
|
|
continue;
|
|
}
|
|
if (s && !s->trimBlanks().null()) {
|
|
int q = s->find('=');
|
|
if (q > 0) {
|
|
String n = s->substr(0,q);
|
|
String v = s->substr(q+1);
|
|
n.trimBlanks();
|
|
v.trimBlanks();
|
|
DDebug("RegexRoute",DebugAll,"Setting '%s' to '%s'",n.c_str(),v.c_str());
|
|
if (n.startSkip("$",false))
|
|
s_vars.setParam(n,v);
|
|
else
|
|
target->setParam(n,v);
|
|
}
|
|
else {
|
|
DDebug("RegexRoute",DebugAll,"Clearing parameter '%s'",s->c_str());
|
|
if (s->startSkip("$",false))
|
|
s_vars.clearParam(*s);
|
|
else
|
|
target->clearParam(*s);
|
|
}
|
|
}
|
|
}
|
|
strs->destruct();
|
|
}
|
|
|
|
// helper function to set the default regexp
|
|
static void setDefault(Regexp& reg)
|
|
{
|
|
if (s_defRule.null())
|
|
return;
|
|
if (reg.null())
|
|
reg = s_defRule;
|
|
else if (reg == "^") {
|
|
// deal with double '^' at end
|
|
if (s_defRule.endsWith("^"))
|
|
reg.assign(s_defRule,s_defRule.length()-1);
|
|
else
|
|
reg = s_defRule + reg;
|
|
}
|
|
}
|
|
|
|
// helper function to process one match attempt
|
|
static bool oneMatch(Message& msg, Regexp& reg, String& match, const String& context, unsigned int rule)
|
|
{
|
|
if (reg.startsWith("${")) {
|
|
// handle special matching by param ${paramname}regexp
|
|
int p = reg.find('}');
|
|
if (p < 3) {
|
|
Debug("RegexRoute",DebugWarn,"Invalid parameter match '%s' in rule #%u in context '%s'",
|
|
reg.c_str(),rule,context.c_str());
|
|
return false;
|
|
}
|
|
match = reg.substr(2,p-2);
|
|
reg = reg.substr(p+1);
|
|
match.trimBlanks();
|
|
reg.trimBlanks();
|
|
String def;
|
|
p = match.find('$');
|
|
if (p >= 0) {
|
|
// param is in ${<name>$<default>} format
|
|
def = match.substr(p+1);
|
|
match = match.substr(0,p);
|
|
match.trimBlanks();
|
|
}
|
|
setDefault(reg);
|
|
if (match.null() || reg.null()) {
|
|
Debug("RegexRoute",DebugWarn,"Missing parameter or rule in rule #%u in context '%s'",
|
|
rule,context.c_str());
|
|
return false;
|
|
}
|
|
DDebug("RegexRoute",DebugAll,"Using message parameter '%s' default '%s'",
|
|
match.c_str(),def.c_str());
|
|
match = msg.getValue(match,def);
|
|
}
|
|
else if (reg.startsWith("$(")) {
|
|
// handle special matching by param $(function)regexp
|
|
int p = reg.find(')');
|
|
if (p < 3) {
|
|
Debug("RegexRoute",DebugWarn,"Invalid function match '%s' in rule #%u in context '%s'",
|
|
reg.c_str(),rule,context.c_str());
|
|
return false;
|
|
}
|
|
match = reg.substr(0,p+1);
|
|
reg = reg.substr(p+1);
|
|
reg.trimBlanks();
|
|
setDefault(reg);
|
|
if (reg.null()) {
|
|
Debug("RegexRoute",DebugWarn,"Missing rule in rule #%u in context '%s'",
|
|
rule,context.c_str());
|
|
return false;
|
|
}
|
|
DDebug("RegexRoute",DebugAll,"Using function '%s'",match.c_str());
|
|
msg.replaceParams(match);
|
|
replaceFuncs(match,msg);
|
|
}
|
|
match.trimBlanks();
|
|
|
|
bool doMatch = true;
|
|
if (reg.endsWith("^")) {
|
|
// reverse match on final ^ (makes no sense in a regexp)
|
|
doMatch = false;
|
|
reg = reg.substr(0,reg.length()-1);
|
|
}
|
|
return (match.matches(reg) == doMatch);
|
|
}
|
|
|
|
enum BlockState {
|
|
BlockRun = 0,
|
|
BlockSkip = 1,
|
|
BlockDone = 2
|
|
};
|
|
|
|
// process one context, can call itself recursively
|
|
static bool oneContext(Message &msg, String &str, const String &context, String &ret, int depth = 0)
|
|
{
|
|
if (context.null())
|
|
return false;
|
|
if (depth > s_maxDepth) {
|
|
Debug("RegexRoute",DebugWarn,"Possible loop detected, current context '%s'",context.c_str());
|
|
return false;
|
|
}
|
|
NamedList *l = s_cfg.getSection(context);
|
|
if (l) {
|
|
unsigned int blockDepth = 0;
|
|
BlockState blockStack[BLOCK_STACK];
|
|
unsigned int len = l->length();
|
|
for (unsigned int i = 0; i < len; i++) {
|
|
const NamedString* n = l->getParam(i);
|
|
if (!n)
|
|
continue;
|
|
BlockState blockThis = (blockDepth > 0) ? blockStack[blockDepth-1] : BlockRun;
|
|
BlockState blockLast = BlockSkip;
|
|
Regexp reg(n->name(),s_extended,s_insensitive);
|
|
if (reg.startSkip("}")) {
|
|
if (!blockDepth) {
|
|
Debug("RegexRoute",DebugWarn,"Got '}' outside block in line #%u in context '%s'",
|
|
i+1,context.c_str());
|
|
continue;
|
|
}
|
|
if (reg.trimBlanks().null())
|
|
reg = ".*";
|
|
blockDepth--;
|
|
blockLast = blockThis;
|
|
blockThis = (blockDepth > 0) ? blockStack[blockDepth-1] : BlockRun;
|
|
}
|
|
static Regexp s_blockStart("\\(=[[:space:]]*\\)\\?{$");
|
|
if (s_blockStart.matches(*n)) {
|
|
// start of a new block
|
|
if (blockDepth >= BLOCK_STACK) {
|
|
Debug("RegexRoute",DebugWarn,"Block stack overflow in line #%u in context '%s'",
|
|
i+1,context.c_str());
|
|
return false;
|
|
}
|
|
// assume block is done
|
|
BlockState blockEnter = BlockDone;
|
|
if (BlockRun == blockThis) {
|
|
// if we returned from a false inner block to a true outer block
|
|
if (BlockSkip == blockLast)
|
|
blockEnter = BlockSkip;
|
|
else
|
|
blockThis = BlockDone;
|
|
}
|
|
blockStack[blockDepth++] = blockEnter;
|
|
}
|
|
XDebug("RegexRoute",DebugAll,"%s:%d(%u:%s) %s=%s",context.c_str(),i+1,
|
|
blockDepth,String::boolText(BlockRun == blockThis),
|
|
n->name().c_str(),n->c_str());
|
|
if (BlockRun != blockThis)
|
|
continue;
|
|
|
|
String val(*n);
|
|
String match;
|
|
bool ok;
|
|
do {
|
|
match = str;
|
|
ok = oneMatch(msg,reg,match,context,i+1);
|
|
if (ok) {
|
|
if (val.startSkip("or")) {
|
|
do {
|
|
int p = val.find('=');
|
|
if (p < 0) {
|
|
Debug("RegexRoute",DebugWarn,"Malformed 'or' rule #%u in context '%s'",
|
|
i+1,context.c_str());
|
|
ok = false;
|
|
break;
|
|
}
|
|
val = val.substr(p+1);
|
|
val.trimBlanks();
|
|
} while (ok && (val.startSkip("or") || val.startSkip("if") || val.startSkip("and")));
|
|
break;
|
|
}
|
|
if (!(val.startSkip("if") || val.startSkip("and")))
|
|
break;
|
|
}
|
|
else if (val.startSkip("or"))
|
|
ok = true;
|
|
if (ok) {
|
|
int p = val.find('=');
|
|
if (p >= 1) {
|
|
reg = val.substr(0,p);
|
|
val = val.substr(p+1);
|
|
reg.trimBlanks();
|
|
val.trimBlanks();
|
|
if (!reg.null()) {
|
|
NDebug("RegexRoute",DebugAll,"Secondary match rule '%s' by rule #%u in context '%s'",
|
|
reg.c_str(),i+1,context.c_str());
|
|
continue;
|
|
}
|
|
}
|
|
Debug("RegexRoute",DebugWarn,"Missing 'if' in rule #%u in context '%s'",
|
|
i+1,context.c_str());
|
|
ok = false;
|
|
}
|
|
} while (ok);
|
|
if (!ok)
|
|
continue;
|
|
|
|
if (val.startSkip("echo") || val.startSkip("output")) {
|
|
// special case: display the line but don't set params
|
|
val = match.replaceMatches(val);
|
|
msg.replaceParams(val);
|
|
replaceFuncs(val,msg);
|
|
Output("%s",val.safe());
|
|
continue;
|
|
}
|
|
else if (val == "{") {
|
|
// mark block as being processed now
|
|
if (blockDepth)
|
|
blockStack[blockDepth-1] = BlockRun;
|
|
else
|
|
Debug("RegexRoute",DebugWarn,"Got '{' outside block in line #%u in context '%s'",
|
|
i+1,context.c_str());
|
|
continue;
|
|
}
|
|
bool disp = val.startSkip("dispatch");
|
|
if (disp || val.startSkip("enqueue")) {
|
|
// special case: enqueue or dispatch a new message
|
|
if (val && (val[0] != ';')) {
|
|
Message* m = new Message("");
|
|
// parameters are set in the new message
|
|
setMessage(match,msg,val,m);
|
|
val.trimBlanks();
|
|
if (val) {
|
|
*m = val;
|
|
m->userData(msg.userData());
|
|
NDebug("RegexRoute",DebugAll,"%s new message '%s' by rule #%u '%s' in context '%s'",
|
|
(disp ? "Dispatching" : "Enqueueing"),
|
|
val.c_str(),i+1,n->name().c_str(),context.c_str());
|
|
if (disp) {
|
|
s_dispatching++;
|
|
Engine::dispatch(m);
|
|
s_dispatching--;
|
|
}
|
|
else {
|
|
Engine::enqueue(m);
|
|
m = 0;
|
|
}
|
|
}
|
|
TelEngine::destruct(m);
|
|
}
|
|
continue;
|
|
}
|
|
setMessage(match,msg,val);
|
|
val.trimBlanks();
|
|
if (val.null()) {
|
|
// special case: do nothing on empty target
|
|
continue;
|
|
}
|
|
else if (val.startSkip("return")) {
|
|
bool ok = val.toBoolean();
|
|
NDebug("RegexRoute",DebugAll,"Returning %s from context '%s'",
|
|
String::boolText(ok),context.c_str());
|
|
return ok;
|
|
}
|
|
else if (val.startSkip("goto") || val.startSkip("jump")) {
|
|
NDebug("RegexRoute",DebugAll,"Jumping to context '%s' by rule #%u '%s'",
|
|
val.c_str(),i+1,n->name().c_str());
|
|
return oneContext(msg,str,val,ret,depth+1);
|
|
}
|
|
else if (val.startSkip("include") || val.startSkip("call")) {
|
|
NDebug("RegexRoute",DebugAll,"Including context '%s' by rule #%u '%s'",
|
|
val.c_str(),i+1,n->name().c_str());
|
|
if (oneContext(msg,str,val,ret,depth+1)) {
|
|
DDebug("RegexRoute",DebugAll,"Returning true from context '%s'", context.c_str());
|
|
return true;
|
|
}
|
|
}
|
|
else if (val.startSkip("match") || val.startSkip("newmatch")) {
|
|
if (!val.null()) {
|
|
NDebug("RegexRoute",DebugAll,"Setting match string '%s' by rule #%u '%s' in context '%s'",
|
|
val.c_str(),i+1,n->name().c_str(),context.c_str());
|
|
str = val;
|
|
}
|
|
}
|
|
else if (val.startSkip("rename")) {
|
|
if (!val.null()) {
|
|
NDebug("RegexRoute",DebugAll,"Renaming message '%s' to '%s' by rule #%u '%s' in context '%s'",
|
|
msg.c_str(),val.c_str(),i+1,n->name().c_str(),context.c_str());
|
|
msg = val;
|
|
}
|
|
}
|
|
else {
|
|
DDebug("RegexRoute",DebugAll,"Returning '%s' for '%s' in context '%s' by rule #%u '%s'",
|
|
val.c_str(),str.c_str(),context.c_str(),i+1,n->name().c_str());
|
|
ret = val;
|
|
return true;
|
|
}
|
|
}
|
|
if (blockDepth)
|
|
Debug("RegexRoute",DebugWarn,"There are %u blocks still open at end of context '%s'",
|
|
blockDepth,context.c_str());
|
|
}
|
|
DDebug("RegexRoute",DebugAll,"Returning false at end of context '%s'", context.c_str());
|
|
return false;
|
|
}
|
|
|
|
|
|
bool RouteHandler::received(Message &msg)
|
|
{
|
|
u_int64_t tmr = Time::now();
|
|
String called(msg.getValue(YSTRING("called")));
|
|
if (called.null())
|
|
return false;
|
|
const char *context = msg.getValue(YSTRING("context"),"default");
|
|
Lock lock(s_mutex);
|
|
if (oneContext(msg,called,context,msg.retValue())) {
|
|
Debug(DebugInfo,"Routing call to '%s' in context '%s' via '%s' in " FMT64U " usec",
|
|
called.c_str(),context,msg.retValue().c_str(),Time::now()-tmr);
|
|
return true;
|
|
}
|
|
Debug(DebugInfo,"Could not route call to '%s' in context '%s', wasted " FMT64U " usec",
|
|
called.c_str(),context,Time::now()-tmr);
|
|
return false;
|
|
};
|
|
|
|
|
|
class PrerouteHandler : public MessageHandler
|
|
{
|
|
public:
|
|
PrerouteHandler(int prio)
|
|
: MessageHandler("call.preroute",prio,s_trackName)
|
|
{ }
|
|
virtual bool received(Message &msg);
|
|
};
|
|
|
|
bool PrerouteHandler::received(Message &msg)
|
|
{
|
|
u_int64_t tmr = Time::now();
|
|
// return immediately if there is already a context
|
|
if (!s_prerouteall && msg.getValue(YSTRING("context")))
|
|
return false;
|
|
|
|
String caller(msg.getValue(YSTRING("caller")));
|
|
if (!s_prerouteall && caller.null())
|
|
return false;
|
|
|
|
String ret;
|
|
Lock lock(s_mutex);
|
|
if (oneContext(msg,caller,"contexts",ret)) {
|
|
Debug(DebugInfo,"Classifying caller '%s' in context '%s' in " FMT64 " usec",
|
|
caller.c_str(),ret.c_str(),Time::now()-tmr);
|
|
if (ret == YSTRING("-") || ret == YSTRING("error"))
|
|
msg.retValue() = ret;
|
|
else
|
|
msg.setParam("context",ret);
|
|
return true;
|
|
}
|
|
Debug(DebugInfo,"Could not classify call from '%s', wasted " FMT64 " usec",
|
|
caller.c_str(),Time::now()-tmr);
|
|
return false;
|
|
};
|
|
|
|
|
|
class GenericHandler : public MessageHandler
|
|
{
|
|
public:
|
|
GenericHandler(const char* name, int prio, const char* context, const char* match)
|
|
: MessageHandler(name,prio,s_trackName),
|
|
m_context(context), m_match(match)
|
|
{
|
|
Debug(DebugAll,"Generic handler for '%s' prio %d to [%s] match '%s%s%s' [%p]",
|
|
c_str(),prio,context,
|
|
(match ? "${" : ""),(match ? match : c_str()),(match ? "}" : ""),
|
|
this);
|
|
s_extra.append(this);
|
|
}
|
|
~GenericHandler()
|
|
{ s_extra.remove(this,false); }
|
|
virtual bool received(Message &msg);
|
|
private:
|
|
String m_context;
|
|
String m_match;
|
|
};
|
|
|
|
bool GenericHandler::received(Message &msg)
|
|
{
|
|
DDebug(DebugAll,"Handling message '%s' [%p]",c_str(),this);
|
|
String what(m_match);
|
|
if (what)
|
|
what = msg.getValue(what);
|
|
else
|
|
what = *this;
|
|
Lock lock(s_mutex);
|
|
return oneContext(msg,what,m_context,msg.retValue());
|
|
}
|
|
|
|
|
|
class RegexRoutePlugin : public Plugin
|
|
{
|
|
public:
|
|
RegexRoutePlugin();
|
|
virtual void initialize();
|
|
private:
|
|
void initVars(NamedList* sect);
|
|
MessageHandler *m_preroute, *m_route;
|
|
bool m_first;
|
|
};
|
|
|
|
RegexRoutePlugin::RegexRoutePlugin()
|
|
: Plugin("regexroute"),
|
|
m_preroute(0), m_route(0), m_first(true)
|
|
{
|
|
Output("Loaded module RegexRoute");
|
|
}
|
|
|
|
void RegexRoutePlugin::initVars(NamedList* sect)
|
|
{
|
|
if (!sect)
|
|
return;
|
|
unsigned int len = sect->length();
|
|
for (unsigned int i=0; i<len; i++) {
|
|
NamedString* n = sect->getParam(i);
|
|
if (n)
|
|
s_vars.setParam(n->name(),*n);
|
|
}
|
|
}
|
|
|
|
void RegexRoutePlugin::initialize()
|
|
{
|
|
Output("Initializing module RegexRoute");
|
|
TelEngine::destruct(m_preroute);
|
|
TelEngine::destruct(m_route);
|
|
s_extra.clear();
|
|
Lock lock(s_mutex);
|
|
s_cfg = Engine::configFile("regexroute");
|
|
s_cfg.load();
|
|
if (m_first) {
|
|
m_first = false;
|
|
initVars(s_cfg.getSection("$once"));
|
|
}
|
|
initVars(s_cfg.getSection("$init"));
|
|
s_trackName = s_cfg.getBoolValue("priorities","trackparam",true) ?
|
|
name().c_str() : (const char*)0;
|
|
s_extended = s_cfg.getBoolValue("priorities","extended",false);
|
|
s_insensitive = s_cfg.getBoolValue("priorities","insensitive",false);
|
|
s_prerouteall = s_cfg.getBoolValue("priorities","prerouteall",false);
|
|
unsigned priority = s_cfg.getIntValue("priorities","preroute",100);
|
|
if (priority) {
|
|
m_preroute = new PrerouteHandler(priority);
|
|
Engine::install(m_preroute);
|
|
}
|
|
priority = s_cfg.getIntValue("priorities","route",100);
|
|
if (priority) {
|
|
m_route = new RouteHandler(priority);
|
|
Engine::install(m_route);
|
|
}
|
|
int depth = s_cfg.getIntValue("priorities","maxdepth",5);
|
|
if (depth < 5)
|
|
depth = 5;
|
|
else if (depth > 100)
|
|
depth = 100;
|
|
s_maxDepth = depth;
|
|
s_defRule = s_cfg.getValue("priorities","defaultrule",DEFAULT_RULE);
|
|
NamedList* l = s_cfg.getSection("extra");
|
|
if (l) {
|
|
unsigned int len = l->length();
|
|
for (unsigned int i=0; i<len; i++) {
|
|
NamedString* n = l->getParam(i);
|
|
if (n) {
|
|
// message=priority[,[parameter][,context]]
|
|
ObjList* o = n->split(',');
|
|
const String* s = static_cast<const String*>(o->at(0));
|
|
int prio = s ? s->toInteger(100) : 100;
|
|
const char* match = TelEngine::c_str(static_cast<const String*>(o->at(1)));
|
|
const char* context = TelEngine::c_str(static_cast<const String*>(o->at(2)));
|
|
if (TelEngine::null(context))
|
|
context = n->name().c_str();
|
|
if (s_cfg.getSection(context))
|
|
Engine::install(new GenericHandler(n->name(),prio,context,match));
|
|
else
|
|
Debug(DebugWarn,"Missing context [%s] for handling %s",context,n->name().c_str());
|
|
TelEngine::destruct(o);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
INIT_PLUGIN(RegexRoutePlugin);
|
|
|
|
}; // anonymous namespace
|
|
|
|
/* vi: set ts=8 sw=4 sts=4 noet: */
|