Skip to content

Commit

Permalink
corex: functions to set socket by name
Browse files Browse the repository at this point in the history
  • Loading branch information
@miconda
miconda committed Apr 3, 2020
1 parent 6a7aa74 commit c6c6e03
Showing 1 changed file with 72 additions and 9 deletions.
81 changes: 72 additions & 9 deletions src/modules/corex/corex_mod.c
View file
Expand Up @@ -54,7 +54,9 @@ static int w_isxflagset(struct sip_msg *msg, char *flag, char *s2);
static int w_resetxflag(struct sip_msg *msg, char *flag, char *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_send_socket_name(sip_msg_t *msg, char *psock, char *p2);
static int w_set_recv_socket(sip_msg_t *msg, char *psock, char *p2);
static int w_set_recv_socket_name(sip_msg_t *msg, char *psock, char *p2);
static int w_set_source_address(sip_msg_t *msg, char *paddr, char *p2);
static int w_via_add_srvid(sip_msg_t *msg, char *pflags, char *p2);
static int w_via_add_xavp_params(sip_msg_t *msg, char *pflags, char *p2);
Expand Down Expand Up @@ -114,8 +116,12 @@ static cmd_export_t cmds[]={
0, ANY_ROUTE },
{"set_send_socket", (cmd_function)w_set_send_socket, 1, fixup_spve_null,
0, ANY_ROUTE },
{"set_send_socket_name", (cmd_function)w_set_send_socket_name, 1, fixup_spve_null,
0, ANY_ROUTE },
{"set_recv_socket", (cmd_function)w_set_recv_socket, 1, fixup_spve_null,
0, ANY_ROUTE },
{"set_recv_socket_name", (cmd_function)w_set_recv_socket_name, 1, fixup_spve_null,
0, ANY_ROUTE },
{"set_source_address", (cmd_function)w_set_source_address, 1, fixup_spve_null,
0, ANY_ROUTE },
{"via_add_srvid", (cmd_function)w_via_add_srvid, 1, fixup_igp_null,
Expand Down Expand Up @@ -609,7 +615,7 @@ static int w_setxflag(sip_msg_t *msg, char *flag, char *s2)
/**
*
*/
static int ki_set_socket_helper(sip_msg_t *msg, str *ssock, int smode)
static int ki_set_socket_helper(sip_msg_t *msg, str *ssock, int smode, int sfmt)
{
socket_info_t *si;
int port, proto;
Expand All @@ -625,14 +631,19 @@ static int ki_set_socket_helper(sip_msg_t *msg, str *ssock, int smode)
return 1;
}

if (parse_phostport(ssock->s, &host.s, &host.len, &port, &proto) < 0) {
LM_ERR("invalid socket specification [%.*s] (%d)\n",
LM_DBG("trying to set %s-socket to %s [%.*s] (%d)\n",
(smode==0)?"snd":"rcv", (sfmt==0)?"addr":"name",
ssock->len, ssock->s, smode);
goto error;
if(sfmt == 0) {
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);
} else {
si = ksr_get_socket_by_name(ssock);
}
LM_DBG("trying to set %s-socket to [%.*s] (%d)\n",
(smode==0)?"snd":"rcv", ssock->len, ssock->s, smode);
si = grep_sock_info(&host, (unsigned short)port, (unsigned short)proto);
if (si!=NULL) {
if(smode==0) {
/* send socket */
Expand Down Expand Up @@ -660,7 +671,7 @@ static int ki_set_socket_helper(sip_msg_t *msg, str *ssock, int smode)
*/
static int ki_set_send_socket(sip_msg_t *msg, str *ssock)
{
return ki_set_socket_helper(msg, ssock, 0);
return ki_set_socket_helper(msg, ssock, 0, 0);
}

/**
Expand All @@ -676,12 +687,33 @@ static int w_set_send_socket(sip_msg_t *msg, char *psock, char *p2)
return ki_set_send_socket(msg, &ssock);
}

/**
*
*/
static int ki_set_send_socket_name(sip_msg_t *msg, str *ssock)
{
return ki_set_socket_helper(msg, ssock, 0, 1);
}

/**
*
*/
static int w_set_send_socket_name(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_name(msg, &ssock);
}

/**
*
*/
static int ki_set_recv_socket(sip_msg_t *msg, str *ssock)
{
return ki_set_socket_helper(msg, ssock, 1);
return ki_set_socket_helper(msg, ssock, 1, 0);
}

/**
Expand All @@ -697,6 +729,27 @@ static int w_set_recv_socket(sip_msg_t *msg, char *psock, char *p2)
return ki_set_recv_socket(msg, &ssock);
}

/**
*
*/
static int ki_set_recv_socket_name(sip_msg_t *msg, str *ssock)
{
return ki_set_socket_helper(msg, ssock, 1, 1);
}

/**
*
*/
static int w_set_recv_socket_name(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_name(msg, &ssock);
}

/**
*
*/
Expand Down Expand Up @@ -914,11 +967,21 @@ static sr_kemi_t sr_kemi_corex_exports[] = {
{ SR_KEMIP_STR, SR_KEMIP_NONE, SR_KEMIP_NONE,
SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
},
{ str_init("corex"), str_init("set_send_socket_name"),
SR_KEMIP_INT, ki_set_send_socket_name,
{ 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 }
},
{ str_init("corex"), str_init("set_recv_socket_name"),
SR_KEMIP_INT, ki_set_recv_socket_name,
{ SR_KEMIP_STR, SR_KEMIP_NONE, SR_KEMIP_NONE,
SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
},
{ str_init("corex"), str_init("set_source_address"),
SR_KEMIP_INT, ki_set_source_address,
{ SR_KEMIP_STR, SR_KEMIP_NONE, SR_KEMIP_NONE,
Expand Down

0 comments on commit c6c6e03

Please sign in to comment.