Module: kamailio Branch: master Commit: 21ad31f75e11adf2789c387b356cc5af5d29e3b4 URL: https://github.com/kamailio/kamailio/commit/21ad31f75e11adf2789c387b356cc5a… Author: Daniel-Constantin Mierla <miconda(a)gmail.com> Committer: Daniel-Constantin Mierla <miconda(a)gmail.com> Date: 2018-07-17T07:50:50+02:00 textops: added replace_hdrs(re, sval) - replace matching regexp with sval inside sip headers part --- Modified: src/modules/textops/textops.c --- Diff: https://github.com/kamailio/kamailio/commit/21ad31f75e11adf2789c387b356cc5a… Patch: https://github.com/kamailio/kamailio/commit/21ad31f75e11adf2789c387b356cc5a… --- diff --git a/src/modules/textops/textops.c b/src/modules/textops/textops.c index a986f78ed3..8da8998803 100644 --- a/src/modules/textops/textops.c +++ b/src/modules/textops/textops.c @@ -91,6 +91,7 @@ static int replace_f(struct sip_msg*, char*, char*); static int replace_str_f(struct sip_msg*, char*, char*, char*); static int replace_body_f(struct sip_msg*, char*, char*); static int replace_body_str_f(struct sip_msg*, char*, char*, char*); +static int replace_hdrs_f(struct sip_msg*, char*, char*); static int replace_hdrs_str_f(struct sip_msg*, char*, char*, char*); static int replace_all_f(struct sip_msg*, char*, char*); static int replace_body_all_f(struct sip_msg*, char*, char*); @@ -190,6 +191,9 @@ static cmd_export_t cmds[]={ {"replace_body_str", (cmd_function)replace_body_str_f,3, fixup_spve_all, fixup_free_spve_all, ANY_ROUTE}, + {"replace_hdrs", (cmd_function)replace_hdrs_f, 2, + fixup_regexp_none, fixup_free_regexp_none, + ANY_ROUTE}, {"replace_hdrs_str", (cmd_function)replace_hdrs_str_f,3, fixup_spve_all, fixup_free_spve_all, ANY_ROUTE}, @@ -1014,6 +1018,87 @@ static int replace_body_str_f(sip_msg_t* msg, char* pmkey, char* prval, char* pr return ki_replace_body_str(msg, &mkey, &rval, &rmode); } +static int replace_hdrs_helper(sip_msg_t* msg, regex_t *re, str *val) +{ + struct lump* l; + regmatch_t pmatch; + char* s; + int off; + str lbuf; + char bk; + + if ( parse_headers(msg, HDR_EOH_F, 0)==-1 ) { + LM_ERR("failed to parse to end of headers\n"); + return -1; + } + + lbuf.s = get_header(msg); + lbuf.len = (int)(msg->unparsed - lbuf.s); + + if (lbuf.len==0) { + LM_DBG("message headers part has zero length\n"); + return -1; + } + + bk = lbuf.s[lbuf.len]; + lbuf.s[lbuf.len] = '\0'; + if (regexec(re, lbuf.s, 1, &pmatch, 0)!=0) { + lbuf.s[lbuf.len] = bk; + return -1; + } + lbuf.s[lbuf.len] = bk; + + off=lbuf.s-msg->buf; + + if (pmatch.rm_so!=-1){ + if ((l=del_lump(msg, pmatch.rm_so+off, + pmatch.rm_eo-pmatch.rm_so, 0))==0) + return -1; + s=pkg_malloc(val->len+1); + if (s==0){ + LM_ERR("memory allocation failure\n"); + return -1; + } + memcpy(s, val->s, val->len); + if (insert_new_lump_after(l, s, val->len, 0)==0){ + LM_ERR("could not insert new lump\n"); + pkg_free(s); + return -1; + } + + return 1; + } + return -1; +} + +static int replace_hdrs_f(struct sip_msg* msg, char* key, char* str2) +{ + str val; + + val.s = str2; + val.len = strlen(val.s); + + return replace_hdrs_helper(msg, (regex_t*)key, &val); +} + +static int ki_replace_hdrs(sip_msg_t* msg, str* sre, str* sval) +{ + regex_t mre; + int ret; + + memset(&mre, 0, sizeof(regex_t)); + if (regcomp(&mre, sre->s, REG_EXTENDED|REG_ICASE|REG_NEWLINE)!=0) { + LM_ERR("failed to compile regex: %.*s\n", sre->len, sre->s); + return -1; + } + + ret = replace_hdrs_helper(msg, &mre, sval); + + regfree(&mre); + + return ret; +} + static int ki_replace_hdrs_str(sip_msg_t* msg, str* mkey, str* rval, str *rmode) { str lbuf; @@ -4468,6 +4553,11 @@ static sr_kemi_t sr_kemi_textops_exports[] = { { SR_KEMIP_STR, SR_KEMIP_STR, SR_KEMIP_STR, SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE } }, + { str_init("textops"), str_init("replace_hdrs"), + SR_KEMIP_INT, ki_replace_hdrs, + { SR_KEMIP_STR, SR_KEMIP_STR, SR_KEMIP_NONE, + SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE } + }, { str_init("textops"), str_init("replace_hdrs_str"), SR_KEMIP_INT, ki_replace_hdrs_str, { SR_KEMIP_STR, SR_KEMIP_STR, SR_KEMIP_STR,