[sr-dev] git:adubovikov/homer: core: prepare raw_sock for monitoring port usage

Module: sip-router Branch: adubovikov/homer Commit: 9a0390327614065f0992c032bfa474f2e4df2301 URL: http://git.sip-router.org/cgi-bin/gitweb.cgi/sip-router/?a=commit;h=9a03903… Author: Alexandr Dubovikov &lt;alexandr.dubovikov(a)gmail.com&gt; Committer: Alexandr Dubovikov &lt;alexandr.dubovikov(a)gmail.com&gt; Date: Thu Sep 22 16:11:45 2011 +0200 core: prepare raw_sock for monitoring port usage --- raw_sock.c | 42 +++++++++++++++++++++++++++++++++++++----- raw_sock.h | 1 + 2 files changed, 38 insertions(+), 5 deletions(-) diff --git a/raw_sock.c b/raw_sock.c index b514a87..601d321 100644 --- a/raw_sock.c +++ b/raw_sock.c @@ -85,6 +85,8 @@ #endif /* __OS_* */ int raw_ipip = 0; /* set if raw socket is in capture mode for IPIP */ +int raw_moni = 0; /* set if raw socket is in capture mode for monitor port */ +#define ETHHDR 14 /* sizeof of ethhdr structure */ /** create and return a raw socket. @@ -244,13 +246,20 @@ int recvpkt4(int sock, char* buf, int len, union sockaddr_union* from, rcv_msg.msg_iovlen=1; ret=-2; /* no PKT_INFO or AF mismatch */ retry: - n=recvmsg(sock, &rcv_msg, MSG_WAITALL); + n=recvmsg(sock, &rcv_msg, raw_moni ? 0 : MSG_WAITALL); if (unlikely(n==-1)){ if (errno==EINTR) goto retry; ret=n; goto end; } + + /* RAW MONI */ + if(raw_moni) { + ret = n; + goto end; + } + /* find the pkt info */ for (cmsg=CMSG_FIRSTHDR(&rcv_msg); cmsg; cmsg=CMSG_NXTHDR(&rcv_msg, cmsg)){ #ifdef IP_PKTINFO @@ -309,7 +318,7 @@ int raw_udp4_recv(int rsock, char** buf, int len, union sockaddr_union* from, int n; unsigned short dst_port; unsigned short src_port; - struct ip_addr dst_ip; + struct ip_addr dst_ip, src_ip; char* end; char* udph_start; char* udp_payload; @@ -321,13 +330,15 @@ int raw_udp4_recv(int rsock, char** buf, int len, union sockaddr_union* from, if (unlikely(n<0)) goto error; end=*buf+n; - if (unlikely(n<((sizeof(struct ip) * raw_ipip ? 2 : 1)+sizeof(struct udphdr)))) { + if (unlikely(n<((sizeof(struct ip) * raw_ipip ? 2 : 1)+sizeof(struct udphdr) + (raw_moni ? ETHHDR : 0)))) { n=-3; goto error; } if(raw_ipip) - *buf = *buf + sizeof(struct ip); + *buf = *buf + sizeof(struct ip); + else if(raw_moni) + *buf = *buf + ETHHDR; /* FIXME: if initial buffer is aligned, one could skip the memcpy and directly cast ip and udphdr pointer to the memory */ @@ -345,7 +356,7 @@ int raw_udp4_recv(int rsock, char** buf, int len, union sockaddr_union* from, n=-3; goto error; }else{ - ERR("udp length too small: %d/%d\n", + if(!raw_moni) ERR("udp length too small: %d/%d\n", (int)udp_len, (int)(end-udph_start)); n=-3; goto error; @@ -363,8 +374,29 @@ int raw_udp4_recv(int rsock, char** buf, int len, union sockaddr_union* from, ip_addr2su(to, &dst_ip, dst_port); /* fill src_port */ src_port=ntohs(udph.uh_sport); + /* Need to fill from structure */ + if(raw_moni) { + src_ip.af=AF_INET; + src_ip.len=4; + src_ip.u.addr32[0]=iph.ip_src.s_addr; + ip_addr2su(from, &src_ip, src_port); + } su_setport(from, src_port); if (likely(rf)) { + + /* check moni mode */ + if(raw_moni) { + if(dst_port && src_port && + ( ((dst_port >= rf->port1) && (dst_port <= rf->port2)) || + ((src_port >= rf->port1) && (src_port <= rf->port2)) + )) { + /* match */ + goto error; + } + n=-4; + goto error; + } + su2ip_addr(&dst_ip, to); if ( (dst_port && rf->port1 && ((dst_port<rf->port1) || (dst_port>rf->port2)) ) || diff --git a/raw_sock.h b/raw_sock.h index 868399f..cd72954 100644 --- a/raw_sock.h +++ b/raw_sock.h @@ -40,6 +40,7 @@ struct raw_filter{ }; extern int raw_ipip; +extern int raw_moni; int raw_socket(int proto, struct ip_addr* ip, str* iface, int iphdr_incl); int raw_udp4_socket(struct ip_addr* ip, str* iface, int iphdr_incl);