Module: kamailio Branch: master Commit: 857800a368c6bb70b5ec057c89ffca7c30556c43 URL: https://github.com/kamailio/kamailio/commit/857800a368c6bb70b5ec057c89ffca7… Author: Daniel-Constantin Mierla <miconda(a)gmail.com> Committer: Daniel-Constantin Mierla <miconda(a)gmail.com> Date: 2017-05-28T11:42:04+02:00 pdt: exported functions to kemi interface --- Modified: src/modules/pdt/pdt.c --- Diff: https://github.com/kamailio/kamailio/commit/857800a368c6bb70b5ec057c89ffca7… Patch: https://github.com/kamailio/kamailio/commit/857800a368c6bb70b5ec057c89ffca7… --- diff --git a/src/modules/pdt/pdt.c b/src/modules/pdt/pdt.c index b8d72fe45c..6505d8a3d5 100644 --- a/src/modules/pdt/pdt.c +++ b/src/modules/pdt/pdt.c @@ -44,6 +44,7 @@ #include "../../core/parser/parse_from.h" #include "../../core/rpc.h" #include "../../core/rpc_lookup.h" +#include "../../core/kemi.h" #include "pdtree.h" @@ -304,36 +305,22 @@ static int w_prefix2domain_1(struct sip_msg* msg, char* mode, char* str2) return pd_translate(msg, &sdall, md, 0); } -static int w_prefix2domain_2(struct sip_msg* msg, char* mode, char* sdm) +static int ki_prefix2domain(sip_msg_t* msg, int m, int s) { - int m, s, f; + int f; str sdomain={"*",1}; sip_uri_t *furi; - if(fixup_get_ivalue(msg, (gparam_p)mode, &m)!=0) - { - LM_ERR("no mode value\n"); - return -1; - } - if(m!=1 && m!=2) m = 0; - if(fixup_get_ivalue(msg, (gparam_p)sdm, &s)!=0) - { - LM_ERR("no multi-domain mode value\n"); - return -1; - } - if(s!=1 && s!=2) s = 0; f = 0; - if(s==1 || s==2) - { + if(s==1 || s==2) { /* take the domain from FROM uri as sdomain */ - if((furi = parse_from_uri(msg))==NULL) - { + if((furi = parse_from_uri(msg))==NULL) { LM_ERR("cannot parse FROM header URI\n"); return -1; } @@ -344,6 +331,24 @@ static int w_prefix2domain_2(struct sip_msg* msg, char* mode, char* sdm) return pd_translate(msg, &sdomain, m, f); } +static int w_prefix2domain_2(struct sip_msg* msg, char* mode, char* sdm) +{ + int m, s; + + if(fixup_get_ivalue(msg, (gparam_p)mode, &m)!=0) + { + LM_ERR("no mode value\n"); + return -1; + } + + if(fixup_get_ivalue(msg, (gparam_p)sdm, &s)!=0) + { + LM_ERR("no multi-domain mode value\n"); + return -1; + } + return ki_prefix2domain(msg, m, s); +} + /** * @brief change the r-uri domain based on source domain and prefix * @@ -477,6 +482,17 @@ static int w_pd_translate(sip_msg_t* msg, char* sdomain, char* mode) } /** + * + */ +static int ki_pd_translate(sip_msg_t* msg, str* sd, int md) +{ + if(md!=1 && md!=2) + return pd_translate(msg, sd, 0, 0);; + + return pd_translate(msg, sd, md, 0); +} + +/** * change the uri according to update mode */ static int update_new_uri(struct sip_msg *msg, int plen, str *d, int mode) @@ -901,3 +917,32 @@ static int pdt_init_rpc(void) } return 0; } + +/** + * + */ +/* clang-format off */ +static sr_kemi_t sr_kemi_pdt_exports[] = { + { str_init("pdt"), str_init("pd_translate"), + SR_KEMIP_INT, ki_pd_translate, + { SR_KEMIP_STR, SR_KEMIP_INT, SR_KEMIP_NONE, + SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE } + }, + { str_init("pdt"), str_init("pprefix2domain"), + SR_KEMIP_INT, ki_prefix2domain, + { SR_KEMIP_INT, SR_KEMIP_INT, SR_KEMIP_NONE, + SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE } + }, + + { {0, 0}, {0, 0}, 0, NULL, { 0, 0, 0, 0, 0, 0 } } +}; +/* clang-format on */ + +/** + * + */ +int mod_register(char *path, int *dlflags, void *p1, void *p2) +{ + sr_kemi_modules_add(sr_kemi_pdt_exports); + return 0; +} \ No newline at end of file