git:master:857800a3: pdt: exported functions to kemi interface - sr-dev

28 May 2017

Module: kamailio
Branch: master
Commit: 857800a368c6bb70b5ec057c89ffca7c30556c43
URL: https://github.com/kamailio/kamailio/commit/857800a368c6bb70b5ec057c89ffca7c...
Author: Daniel-Constantin Mierla miconda@gmail.com
Committer: Daniel-Constantin Mierla miconda@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/857800a368c6bb70b5ec057c89ffca7c...
Patch: https://github.com/kamailio/kamailio/commit/857800a368c6bb70b5ec057c89ffca7c...
---

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

    