[sr-dev] git:master: core: helper function to get source address as uri or proxy format

Module: sip-router Branch: master Commit: 3f3c865ed04e1b0b8eb3762622662a76ce92b2e2 URL: http://git.sip-router.org/cgi-bin/gitweb.cgi/sip-router/?a=commit;h=3f3c865… Author: Daniel-Constantin Mierla <miconda(a)gmail.com> Committer: Daniel-Constantin Mierla <miconda(a)gmail.com> Date: Wed Oct 2 14:23:53 2013 +0200 core: helper function to get source address as uri or proxy format --- dset.c | 114 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ dset.h | 5 +++ 2 files changed, 119 insertions(+), 0 deletions(-) diff --git a/dset.c b/dset.c index 52521ce..6ca3c02 100644 --- a/dset.c +++ b/dset.c @@ -605,3 +605,117 @@ int rewrite_uri(struct sip_msg* _m, str* _s) return 1; } +/** + * return src ip, port and proto as a SIP uri or proxy address + * - value stored in a static buffer + * - mode=0 return uri, mode=1 return proxy address + */ +int msg_get_src_addr(sip_msg_t *msg, str *uri, int mode) +{ + static char buf[80]; + char* p; + str ip, port; + int len; + str proto; + + if (msg==NULL || uri==NULL) { + LM_ERR("invalid parameter value\n"); + return -1; + } + + ip.s = ip_addr2a(&msg->rcv.src_ip); + ip.len = strlen(ip.s); + + port.s = int2str(msg->rcv.src_port, &port.len); + + switch(msg->rcv.proto) { + case PROTO_NONE: + case PROTO_UDP: + if(mode==0) { + proto.s = 0; /* Do not add transport parameter, UDP is default */ + proto.len = 0; + } else { + proto.s = "udp"; + proto.len = 3; + } + break; + + case PROTO_TCP: + proto.s = "tcp"; + proto.len = 3; + break; + + case PROTO_TLS: + proto.s = "tls"; + proto.len = 3; + break; + + case PROTO_SCTP: + proto.s = "sctp"; + proto.len = 4; + break; + + case PROTO_WS: + case PROTO_WSS: + proto.s = "ws"; + proto.len = 2; + break; + + default: + LM_ERR("unknown transport protocol\n"); + return -1; + } + + len = ip.len + 2*(msg->rcv.src_ip.af==AF_INET6)+ 1 + port.len; + if (mode==0) { + len += 4; + if(proto.s) { + len += TRANSPORT_PARAM_LEN; + len += proto.len; + } + } else { + len += proto.len + 1; + } + + if (len > 79) { + LM_ERR("buffer too small\n"); + return -1; + } + + p = buf; + if(mode==0) { + memcpy(p, "sip:", 4); + p += 4; + } else { + memcpy(p, proto.s, proto.len); + p += proto.len; + *p++ = ':'; + } + + if (msg->rcv.src_ip.af==AF_INET6) + *p++ = '['; + memcpy(p, ip.s, ip.len); + p += ip.len; + if (msg->rcv.src_ip.af==AF_INET6) + *p++ = ']'; + + *p++ = ':'; + + memcpy(p, port.s, port.len); + p += port.len; + + if (mode==0 && proto.s) { + memcpy(p, TRANSPORT_PARAM, TRANSPORT_PARAM_LEN); + p += TRANSPORT_PARAM_LEN; + + memcpy(p, proto.s, proto.len); + p += proto.len; + } + + uri->s = buf; + uri->len = len; + uri->s[uri->len] = '\0'; + + return 0; +} + diff --git a/dset.h b/dset.h index dc15eba..c811cdb 100644 --- a/dset.h +++ b/dset.h @@ -170,6 +170,11 @@ void set_ruri_q(qvalue_t q); /*! \brief + * Get src ip, port and proto as SIP uri or proxy address + */ +int msg_get_src_addr(sip_msg_t *msg, str *uri, int mode); + +/*! \brief * Get the q value of the Request-URI */ qvalue_t get_ruri_q(void);