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

Author: Daniel-Constantin Mierla <mico...@gmail.com>
Committer: Daniel-Constantin Mierla <mico...@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 } }
 };


_______________________________________________
Kamailio (SER) - Development Mailing List
sr-dev@lists.kamailio.org
https://lists.kamailio.org/cgi-bin/mailman/listinfo/sr-dev

Reply via email to