[sr-dev] git:master:b04744aa: corex: added set_send_socket() and set_recv_socket()

[Daniel-Constantin Mierla]

Module: kamailio
Branch: master
Commit: b04744aaf95b8ef791a69c93554919af47f67652
URL: https://github.com/kamailio/kamailio/commit/b04744aaf95b8ef791a69c93554919af47f67652

Author: Daniel-Constantin Mierla <miconda at gmail.com>
Committer: Daniel-Constantin Mierla <miconda at gmail.com>
Date: 2018-04-09T12:11:31+02:00

corex: added set_send_socket() and set_recv_socket()

- options to set send and recv sockets with parameters containing
variables

---

Modified: src/modules/corex/corex_mod.c

---

Diff:  https://github.com/kamailio/kamailio/commit/b04744aaf95b8ef791a69c93554919af47f67652.diff
Patch: https://github.com/kamailio/kamailio/commit/b04744aaf95b8ef791a69c93554919af47f67652.patch

---

diff --git a/src/modules/corex/corex_mod.c b/src/modules/corex/corex_mod.c
index d80004dd05..d2632dbdab 100644
--- a/src/modules/corex/corex_mod.c
+++ b/src/modules/corex/corex_mod.c
@@ -49,6 +49,8 @@ static int w_file_write(sip_msg_t *msg, char *fn, char *vn);
 static int w_isxflagset(struct sip_msg *msg, char *flag, str *s2);
 static int w_resetxflag(struct sip_msg *msg, char *flag, str *s2);
 static int w_setxflag(struct sip_msg *msg, char *flag, char *s2);
+static int w_set_send_socket(sip_msg_t *msg, char *psock, char *p2);
+static int w_set_recv_socket(sip_msg_t *msg, char *psock, char *p2);
 
 static int fixup_file_op(void** param, int param_no);
 
@@ -100,6 +102,10 @@ static cmd_export_t cmds[]={
 		0, ANY_ROUTE },
 	{"isxflagset", (cmd_function)w_isxflagset,      1,fixup_igp_null,
 		0, ANY_ROUTE },
+	{"set_send_socket", (cmd_function)w_set_send_socket, 1, fixup_spve_null,
+		0, ANY_ROUTE },
+	{"set_recv_socket", (cmd_function)w_set_recv_socket, 1, fixup_spve_null,
+		0, ANY_ROUTE },
 
 	{0, 0, 0, 0, 0, 0}
 };
@@ -540,6 +546,96 @@ static int w_setxflag(sip_msg_t *msg, char *flag, char *s2)
 	return ki_setxflag(msg, fval);
 }
 
+/**
+ *
+ */
+static int ki_set_socket_helper(sip_msg_t *msg, str *ssock, int smode)
+{
+	socket_info_t *si;
+	receive_info_t ri;
+	int port, proto;
+	str host;
+
+	if(msg==NULL) {
+		LM_ERR("bad parameters\n");
+		return -1;
+	}
+
+	if(ssock==NULL || ssock->len<=0) {
+		reset_force_socket(msg);
+		return 1;
+	}
+
+	if (parse_phostport(ssock->s, &host.s, &host.len, &port, &proto) < 0) {
+		LM_ERR("invalid socket specification [%.*s] (%d)\n",
+				ssock->len, ssock->s, smode);
+		goto error;
+	}
+	si = grep_sock_info(&host, (unsigned short)port, (unsigned short)proto);
+	if (si!=NULL) {
+		if(smode==0) {
+			/* send socket */
+			set_force_socket(msg, si);
+		} else {
+			/* recv socket */
+			msg->rcv.bind_address = si;
+			msg->rcv.dst_port = si->port_no;
+			msg->rcv.dst_ip = si->address;
+			msg->rcv.proto = si->proto;
+			msg->rcv.proto_reserved1 = 0;
+			msg->rcv.proto_reserved2 = 0;
+		}
+	} else {
+		LM_WARN("no socket found to match [%.*s] (%d)\n",
+				ssock->len, ssock->s, smode);
+	}
+	return 1;
+error:
+	return -1;
+}
+
+/**
+ *
+ */
+static int ki_set_send_socket(sip_msg_t *msg, str *ssock)
+{
+	return ki_set_socket_helper(msg, ssock, 0);
+}
+
+/**
+ *
+ */
+static int w_set_send_socket(sip_msg_t *msg, char *psock, char *p2)
+{
+	str ssock;
+	if (fixup_get_svalue(msg, (gparam_t*)psock, &ssock) != 0 || ssock.len<=0) {
+		LM_ERR("cannot get socket address value\n");
+		return -1;
+	}
+	return ki_set_send_socket(msg, &ssock);
+}
+
+/**
+ *
+ */
+static int ki_set_recv_socket(sip_msg_t *msg, str *ssock)
+{
+	return ki_set_socket_helper(msg, ssock, 1);
+}
+
+/**
+ *
+ */
+static int w_set_recv_socket(sip_msg_t *msg, char *psock, char *p2)
+{
+	str ssock;
+	if (fixup_get_svalue(msg, (gparam_t*)psock, &ssock) != 0 || ssock.len<=0) {
+		LM_ERR("cannot get socket address value\n");
+		return -1;
+	}
+	return ki_set_recv_socket(msg, &ssock);
+}
+
 /**
  *
  */
@@ -575,6 +671,16 @@ static sr_kemi_t sr_kemi_corex_exports[] = {
 		{ SR_KEMIP_INT, SR_KEMIP_NONE, SR_KEMIP_NONE,
 			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
 	},
+	{ str_init("corex"), str_init("set_send_socket"),
+		SR_KEMIP_INT, ki_set_send_socket,
+		{ SR_KEMIP_STR, SR_KEMIP_NONE, SR_KEMIP_NONE,
+			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
+	},
+	{ str_init("corex"), str_init("set_recv_socket"),
+		SR_KEMIP_INT, ki_set_recv_socket,
+		{ SR_KEMIP_STR, SR_KEMIP_NONE, SR_KEMIP_NONE,
+			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
+	},
 
 	{ {0, 0}, {0, 0}, 0, NULL, { 0, 0, 0, 0, 0, 0 } }
 };