| --- /dev/null |
| +++ b/extensions/rtsp/Kbuild |
| @@ -0,0 +1,4 @@ |
| +# -*- Makefile -*- |
| + |
| +obj-m += nf_nat_rtsp.o |
| +obj-m += nf_conntrack_rtsp.o |
| --- /dev/null |
| +++ b/extensions/rtsp/netfilter_helpers.h |
| @@ -0,0 +1,144 @@ |
| +/* |
| + * Helpers for netfiler modules. This file provides implementations for basic |
| + * functions such as strncasecmp(), etc. |
| + * |
| + * gcc will warn for defined but unused functions, so we only include the |
| + * functions requested. The following macros are used: |
| + * NF_NEED_STRNCASECMP nf_strncasecmp() |
| + * NF_NEED_STRTOU16 nf_strtou16() |
| + * NF_NEED_STRTOU32 nf_strtou32() |
| + */ |
| +#ifndef _NETFILTER_HELPERS_H |
| +#define _NETFILTER_HELPERS_H |
| + |
| +/* Only include these functions for kernel code. */ |
| +#ifdef __KERNEL__ |
| + |
| +#include <net/netfilter/nf_conntrack_expect.h> |
| + |
| +#include <linux/ctype.h> |
| +#define iseol(c) ( (c) == '\r' || (c) == '\n' ) |
| + |
| +/* |
| + * The standard strncasecmp() |
| + */ |
| +#ifdef NF_NEED_STRNCASECMP |
| +static int |
| +nf_strncasecmp(const char* s1, const char* s2, u_int32_t len) |
| +{ |
| + if (s1 == NULL || s2 == NULL) |
| + { |
| + if (s1 == NULL && s2 == NULL) |
| + { |
| + return 0; |
| + } |
| + return (s1 == NULL) ? -1 : 1; |
| + } |
| + while (len > 0 && tolower(*s1) == tolower(*s2)) |
| + { |
| + len--; |
| + s1++; |
| + s2++; |
| + } |
| + return ( (len == 0) ? 0 : (tolower(*s1) - tolower(*s2)) ); |
| +} |
| +#endif /* NF_NEED_STRNCASECMP */ |
| + |
| +/* |
| + * Parse a string containing a 16-bit unsigned integer. |
| + * Returns the number of chars used, or zero if no number is found. |
| + */ |
| +#ifdef NF_NEED_STRTOU16 |
| +static int |
| +nf_strtou16(const char* pbuf, u_int16_t* pval) |
| +{ |
| + int n = 0; |
| + |
| + *pval = 0; |
| + while (isdigit(pbuf[n])) |
| + { |
| + *pval = (*pval * 10) + (pbuf[n] - '0'); |
| + n++; |
| + } |
| + |
| + return n; |
| +} |
| +#endif /* NF_NEED_STRTOU16 */ |
| + |
| +/* |
| + * Parse a string containing a 32-bit unsigned integer. |
| + * Returns the number of chars used, or zero if no number is found. |
| + */ |
| +#ifdef NF_NEED_STRTOU32 |
| +static int |
| +nf_strtou32(const char* pbuf, u_int32_t* pval) |
| +{ |
| + int n = 0; |
| + |
| + *pval = 0; |
| + while (pbuf[n] >= '0' && pbuf[n] <= '9') |
| + { |
| + *pval = (*pval * 10) + (pbuf[n] - '0'); |
| + n++; |
| + } |
| + |
| + return n; |
| +} |
| +#endif /* NF_NEED_STRTOU32 */ |
| + |
| +/* |
| + * Given a buffer and length, advance to the next line and mark the current |
| + * line. |
| + */ |
| +#ifdef NF_NEED_NEXTLINE |
| +static int |
| +nf_nextline(char* p, uint len, uint* poff, uint* plineoff, uint* plinelen) |
| +{ |
| + uint off = *poff; |
| + uint physlen = 0; |
| + |
| + if (off >= len) |
| + { |
| + return 0; |
| + } |
| + |
| + while (p[off] != '\n') |
| + { |
| + if (len-off <= 1) |
| + { |
| + return 0; |
| + } |
| + |
| + physlen++; |
| + off++; |
| + } |
| + |
| + /* if we saw a crlf, physlen needs adjusted */ |
| + if (physlen > 0 && p[off] == '\n' && p[off-1] == '\r') |
| + { |
| + physlen--; |
| + } |
| + |
| + /* advance past the newline */ |
| + off++; |
| + |
| + *plineoff = *poff; |
| + *plinelen = physlen; |
| + *poff = off; |
| + |
| + return 1; |
| +} |
| +#endif /* NF_NEED_NEXTLINE */ |
| + |
| +static inline int rtsp_nf_ct_expect_related(struct nf_conntrack_expect *expect) |
| +{ |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(5,3,0) |
| + return nf_ct_expect_related(expect, 0); |
| +#else |
| + return nf_ct_expect_related(expect); |
| +#endif |
| +} |
| + |
| +#endif /* __KERNEL__ */ |
| + |
| +#endif /* _NETFILTER_HELPERS_H */ |
| --- /dev/null |
| +++ b/extensions/rtsp/netfilter_mime.h |
| @@ -0,0 +1,89 @@ |
| +/* |
| + * MIME functions for netfilter modules. This file provides implementations |
| + * for basic MIME parsing. MIME headers are used in many protocols, such as |
| + * HTTP, RTSP, SIP, etc. |
| + * |
| + * gcc will warn for defined but unused functions, so we only include the |
| + * functions requested. The following macros are used: |
| + * NF_NEED_MIME_NEXTLINE nf_mime_nextline() |
| + */ |
| +#ifndef _NETFILTER_MIME_H |
| +#define _NETFILTER_MIME_H |
| + |
| +/* Only include these functions for kernel code. */ |
| +#ifdef __KERNEL__ |
| + |
| +#include <linux/ctype.h> |
| + |
| +/* |
| + * Given a buffer and length, advance to the next line and mark the current |
| + * line. If the current line is empty, *plinelen will be set to zero. If |
| + * not, it will be set to the actual line length (including CRLF). |
| + * |
| + * 'line' in this context means logical line (includes LWS continuations). |
| + * Returns 1 on success, 0 on failure. |
| + */ |
| +#ifdef NF_NEED_MIME_NEXTLINE |
| +static int |
| +nf_mime_nextline(char* p, uint len, uint* poff, uint* plineoff, uint* plinelen) |
| +{ |
| + uint off = *poff; |
| + uint physlen = 0; |
| + int is_first_line = 1; |
| + |
| + if (off >= len) |
| + { |
| + return 0; |
| + } |
| + |
| + do |
| + { |
| + while (p[off] != '\n') |
| + { |
| + if (len-off <= 1) |
| + { |
| + return 0; |
| + } |
| + |
| + physlen++; |
| + off++; |
| + } |
| + |
| + /* if we saw a crlf, physlen needs adjusted */ |
| + if (physlen > 0 && p[off] == '\n' && p[off-1] == '\r') |
| + { |
| + physlen--; |
| + } |
| + |
| + /* advance past the newline */ |
| + off++; |
| + |
| + /* check for an empty line */ |
| + if (physlen == 0) |
| + { |
| + break; |
| + } |
| + |
| + /* check for colon on the first physical line */ |
| + if (is_first_line) |
| + { |
| + is_first_line = 0; |
| + if (memchr(p+(*poff), ':', physlen) == NULL) |
| + { |
| + return 0; |
| + } |
| + } |
| + } |
| + while (p[off] == ' ' || p[off] == '\t'); |
| + |
| + *plineoff = *poff; |
| + *plinelen = (physlen == 0) ? 0 : (off - *poff); |
| + *poff = off; |
| + |
| + return 1; |
| +} |
| +#endif /* NF_NEED_MIME_NEXTLINE */ |
| + |
| +#endif /* __KERNEL__ */ |
| + |
| +#endif /* _NETFILTER_MIME_H */ |
| --- /dev/null |
| +++ b/extensions/rtsp/nf_conntrack_rtsp.c |
| @@ -0,0 +1,756 @@ |
| +/* |
| + * RTSP extension for IP connection tracking |
| + * (C) 2003 by Tom Marshall <tmarshall at real.com> |
| + * |
| + * 2005-02-13: Harald Welte <laforge at netfilter.org> |
| + * - port to 2.6 |
| + * - update to recent post-2.6.11 api changes |
| + * 2006-09-14: Steven Van Acker <deepstar at singularity.be> |
| + * - removed calls to NAT code from conntrack helper: NAT no longer needed to use rtsp-conntrack |
| + * 2007-04-18: Michael Guntsche <mike at it-loops.com> |
| + * - Port to new NF API |
| + * 2013-03-04: Il'inykh Sergey <sergeyi at inango-sw.com>. Inango Systems Ltd |
| + * - fixed rtcp nat mapping and other port mapping fixes |
| + * - simple TEARDOWN request handling |
| + * - codestyle fixes and other less significant bug fixes |
| + * 2018-04-17: Alin Nastac <alin.nastac at gmail.com> |
| + * Hans Dedecker <dedeckeh at gmail.com> |
| + * - use IP address read from SETUP URI in expected connections |
| + * 2018-04-18: Hans Dedecker <dedeckeh at gmail.com> |
| + * - update RTP expected connection source IP based on SOURCE |
| + * in the SETUP reply message |
| + * 2018-08-03: Alin Nastac <alin.nastac at gmail.com> |
| + * Hans Dedecker <dedeckeh at gmail.com> |
| + * - parse non-standard destination=address:port format |
| + * |
| + * based on ip_conntrack_irc.c |
| + * |
| + * This program is free software; you can redistribute it and/or |
| + * modify it under the terms of the GNU General Public License |
| + * as published by the Free Software Foundation; either version |
| + * 2 of the License, or (at your option) any later version. |
| + * |
| + * Module load syntax: |
| + * insmod nf_conntrack_rtsp.o ports=port1,port2,...port<MAX_PORTS> |
| + * max_outstanding=n setup_timeout=secs |
| + * |
| + * If no ports are specified, the default will be port 554. |
| + * |
| + * With max_outstanding you can define the maximum number of not yet |
| + * answered SETUP requests per RTSP session (default 8). |
| + * With setup_timeout you can specify how long the system waits for |
| + * an expected data channel (default 300 seconds). |
| + * |
| + */ |
| + |
| +#include <linux/module.h> |
| +#include <linux/netfilter.h> |
| +#include <linux/ip.h> |
| +#include <linux/inet.h> |
| +#include <net/tcp.h> |
| + |
| +#include <net/netfilter/nf_conntrack.h> |
| +#include <net/netfilter/nf_conntrack_helper.h> |
| +#include <net/netfilter/nf_conntrack_zones.h> |
| +#include "nf_conntrack_rtsp.h" |
| + |
| +#define NF_NEED_STRNCASECMP |
| +#define NF_NEED_STRTOU16 |
| +#define NF_NEED_STRTOU32 |
| +#define NF_NEED_NEXTLINE |
| +#include "netfilter_helpers.h" |
| +#define NF_NEED_MIME_NEXTLINE |
| +#include "netfilter_mime.h" |
| + |
| +#include <linux/ctype.h> |
| + |
| +#define MAX_PORTS 8 |
| +static int ports[MAX_PORTS]; |
| +static int num_ports = 0; |
| +static int max_outstanding = 8; |
| +static unsigned int setup_timeout = 300; |
| + |
| +MODULE_AUTHOR("Tom Marshall <tmarshall at real.com>"); |
| +MODULE_DESCRIPTION("RTSP connection tracking module"); |
| +MODULE_LICENSE("GPL"); |
| +module_param_array(ports, int, &num_ports, 0400); |
| +MODULE_PARM_DESC(ports, "port numbers of RTSP servers"); |
| +module_param(max_outstanding, int, 0400); |
| +MODULE_PARM_DESC(max_outstanding, "max number of outstanding SETUP requests per RTSP session"); |
| +module_param(setup_timeout, uint, 0400); |
| +MODULE_PARM_DESC(setup_timeout, "timeout on for unestablished data channels"); |
| + |
| +static char *rtsp_buffer; |
| +static DEFINE_SPINLOCK(rtsp_buffer_lock); |
| + |
| +static struct nf_conntrack_expect_policy rtsp_exp_policy; |
| + |
| +unsigned int (*nf_nat_rtsp_hook)(struct sk_buff *skb, |
| + enum ip_conntrack_info ctinfo, |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + unsigned int protoff, |
| +#endif |
| + unsigned int matchoff, unsigned int matchlen, |
| + struct ip_ct_rtsp_expect* prtspexp, |
| + struct nf_conntrack_expect *rtp_exp, |
| + struct nf_conntrack_expect *rtcp_exp); |
| + |
| +EXPORT_SYMBOL_GPL(nf_nat_rtsp_hook); |
| + |
| +/* |
| + * Max mappings we will allow for one RTSP connection (for RTP, the number |
| + * of allocated ports is twice this value). Note that SMIL burns a lot of |
| + * ports so keep this reasonably high. If this is too low, you will see a |
| + * lot of "no free client map entries" messages. |
| + */ |
| +#define MAX_PORT_MAPS 16 |
| + |
| +/*** default port list was here in the masq code: 554, 3030, 4040 ***/ |
| + |
| +#define SKIP_WSPACE(ptr,len,off) while(off < len && isspace(*(ptr+off))) { off++; } |
| + |
| +/* |
| + * Parse an RTSP packet. |
| + * |
| + * Returns zero if parsing failed. |
| + * |
| + * Parameters: |
| + * IN ptcp tcp data pointer |
| + * IN tcplen tcp data len |
| + * IN/OUT ptcpoff points to current tcp offset |
| + * OUT phdrsoff set to offset of rtsp headers |
| + * OUT phdrslen set to length of rtsp headers |
| + * OUT pcseqoff set to offset of CSeq header |
| + * OUT pcseqlen set to length of CSeq header |
| + */ |
| +static int |
| +rtsp_parse_message(char* ptcp, uint tcplen, uint* ptcpoff, |
| + uint* phdrsoff, uint* phdrslen, |
| + uint* pcseqoff, uint* pcseqlen, |
| + uint* transoff, uint* translen) |
| +{ |
| + uint entitylen = 0; |
| + uint lineoff; |
| + uint linelen; |
| + |
| + if (!nf_nextline(ptcp, tcplen, ptcpoff, &lineoff, &linelen)) |
| + return 0; |
| + |
| + *phdrsoff = *ptcpoff; |
| + while (nf_mime_nextline(ptcp, tcplen, ptcpoff, &lineoff, &linelen)) { |
| + if (linelen == 0) { |
| + if (entitylen > 0) |
| + *ptcpoff += min(entitylen, tcplen - *ptcpoff); |
| + break; |
| + } |
| + if (lineoff+linelen > tcplen) { |
| + pr_info("!! overrun !!\n"); |
| + break; |
| + } |
| + |
| + if (nf_strncasecmp(ptcp+lineoff, "CSeq:", 5) == 0) { |
| + *pcseqoff = lineoff; |
| + *pcseqlen = linelen; |
| + } |
| + |
| + if (nf_strncasecmp(ptcp+lineoff, "Transport:", 10) == 0) { |
| + *transoff = lineoff; |
| + *translen = linelen; |
| + } |
| + |
| + if (nf_strncasecmp(ptcp+lineoff, "Content-Length:", 15) == 0) { |
| + uint off = lineoff+15; |
| + SKIP_WSPACE(ptcp+lineoff, linelen, off); |
| + nf_strtou32(ptcp+off, &entitylen); |
| + } |
| + } |
| + *phdrslen = (*ptcpoff) - (*phdrsoff); |
| + |
| + return 1; |
| +} |
| + |
| +/* |
| + * Find lo/hi client ports and/or source (if any) in transport header |
| + * In: |
| + * ptcp, tcplen = packet |
| + * tranoff, tranlen = buffer to search |
| + * |
| + * Out: |
| + * pport_lo, pport_hi = lo/hi ports (host endian) |
| + * srvaddr |
| + * |
| + * Returns nonzero if any client ports found |
| + * |
| + * Note: it is valid (and expected) for the client to request multiple |
| + * transports, so we need to parse the entire line. |
| + */ |
| +static int |
| +rtsp_parse_transport(char* ptran, uint tranlen, |
| + struct ip_ct_rtsp_expect* prtspexp) |
| +{ |
| + int rc = 0; |
| + uint off = 0; |
| + |
| + if (tranlen < 10 || !iseol(ptran[tranlen-1]) || |
| + nf_strncasecmp(ptran, "Transport:", 10) != 0) { |
| + pr_info("sanity check failed\n"); |
| + return 0; |
| + } |
| + |
| + pr_debug("tran='%.*s'\n", (int)tranlen, ptran); |
| + off += 10; |
| + SKIP_WSPACE(ptran, tranlen, off); |
| + |
| + /* Transport: tran;field;field=val,tran;field;field=val,... */ |
| + while (off < tranlen) { |
| + const char* pparamend; |
| + uint nextparamoff; |
| + |
| + pparamend = memchr(ptran+off, ',', tranlen-off); |
| + pparamend = (pparamend == NULL) ? ptran+tranlen : pparamend+1; |
| + nextparamoff = pparamend-ptran; |
| + |
| + while (off < nextparamoff) { |
| + const char* pfieldend; |
| + uint nextfieldoff; |
| + |
| + pfieldend = memchr(ptran+off, ';', nextparamoff-off); |
| + nextfieldoff = (pfieldend == NULL) ? nextparamoff : pfieldend-ptran+1; |
| + |
| + if (strncmp(ptran+off, "client_port=", 12) == 0) { |
| + u_int16_t port; |
| + uint numlen; |
| + |
| + off += 12; |
| + numlen = nf_strtou16(ptran+off, &port); |
| + off += numlen; |
| + if (prtspexp->loport != 0 && prtspexp->loport != port) |
| + pr_debug("multiple ports found, port %hu ignored\n", port); |
| + else { |
| + pr_debug("lo port found : %hu\n", port); |
| + prtspexp->loport = prtspexp->hiport = port; |
| + if (ptran[off] == '-') { |
| + off++; |
| + numlen = nf_strtou16(ptran+off, &port); |
| + off += numlen; |
| + prtspexp->pbtype = pb_range; |
| + prtspexp->hiport = port; |
| + |
| + // If we have a range, assume rtp: |
| + // loport must be even, hiport must be loport+1 |
| + if ((prtspexp->loport & 0x0001) != 0 || |
| + prtspexp->hiport != prtspexp->loport+1) { |
| + pr_debug("incorrect range: %hu-%hu, correcting\n", |
| + prtspexp->loport, prtspexp->hiport); |
| + prtspexp->loport &= 0xfffe; |
| + prtspexp->hiport = prtspexp->loport+1; |
| + } |
| + } else if (ptran[off] == '/') { |
| + off++; |
| + numlen = nf_strtou16(ptran+off, &port); |
| + off += numlen; |
| + prtspexp->pbtype = pb_discon; |
| + prtspexp->hiport = port; |
| + } |
| + rc = 1; |
| + } |
| + } else if (strncmp(ptran+off, "source=", 7) == 0) { |
| + uint srcaddrlen; |
| + |
| + off += 7; |
| + srcaddrlen = nextfieldoff - off - 1; |
| + |
| + if (in4_pton(ptran + off, srcaddrlen, |
| + (u8 *)&prtspexp->srvaddr.in, |
| + -1, NULL)) |
| + pr_debug("source found : %pI4\n", |
| + &prtspexp->srvaddr.ip); |
| + } else if (nextfieldoff - off > 12 && strncmp(ptran+off, "destination=", 12) == 0) { |
| + const char *psep; |
| + u_int16_t port; |
| + |
| + off += 12; |
| + |
| + if (in4_pton(ptran+off, nextfieldoff - off - 1, (u8 *)&prtspexp->cltaddr.in, -1, NULL)) { |
| + pr_debug("destination found : %pI4\n", &prtspexp->cltaddr.ip); |
| + |
| + /* |
| + * Some RTSP clients(mostly STBs) use non-standard destination parameters: |
| + * destination=address:port |
| + */ |
| + psep = memchr(ptran+off, ':', nextfieldoff-off); |
| + if (psep != NULL && nf_strtou16(psep + 1, &port)) { |
| + if (prtspexp->loport != 0 && prtspexp->loport != port) |
| + pr_debug("multiple ports found, port %hu ignored\n", port); |
| + else { |
| + pr_debug("lo port found : %hu\n", port); |
| + prtspexp->loport = prtspexp->hiport = port; |
| + } |
| + } |
| + } |
| + } |
| + |
| + /* |
| + * Note we don't look for the destination parameter here. |
| + * If we are using NAT, the NAT module will handle it. If not, |
| + * and the client is sending packets elsewhere, the expectation |
| + * will quietly time out. |
| + */ |
| + |
| + off = nextfieldoff; |
| + } |
| + |
| + off = nextparamoff; |
| + } |
| + |
| + return rc; |
| +} |
| + |
| + |
| +/*** conntrack functions ***/ |
| + |
| +/* outbound packet: client->server */ |
| + |
| +static inline int |
| +help_out(struct sk_buff *skb, unsigned char *rb_ptr, unsigned int datalen, |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + struct nf_conn *ct, enum ip_conntrack_info ctinfo, |
| + unsigned int protoff) |
| +#else |
| + struct nf_conn *ct, enum ip_conntrack_info ctinfo) |
| +#endif |
| +{ |
| + struct ip_ct_rtsp_expect expinfo; |
| + |
| + int dir = CTINFO2DIR(ctinfo); /* = IP_CT_DIR_ORIGINAL */ |
| + //struct tcphdr* tcph = (void*)iph + iph->ihl * 4; |
| + //uint tcplen = pktlen - iph->ihl * 4; |
| + char* pdata = rb_ptr; |
| + //uint datalen = tcplen - tcph->doff * 4; |
| + uint dataoff = 0; |
| + int ret = NF_ACCEPT; |
| + |
| + struct nf_conntrack_expect *rtp_exp; |
| + struct nf_conntrack_expect *rtcp_exp = NULL; |
| + |
| + __be16 be_loport; |
| + __be16 be_hiport; |
| + |
| + typeof(nf_nat_rtsp_hook) nf_nat_rtsp; |
| + |
| + memset(&expinfo, 0, sizeof(expinfo)); |
| + |
| + while (dataoff < datalen) { |
| + uint cmdoff = dataoff; |
| + uint hdrsoff = 0; |
| + uint hdrslen = 0; |
| + uint cseqoff = 0; |
| + uint cseqlen = 0; |
| + uint transoff = 0; |
| + uint translen = 0; |
| + uint off; |
| + union nf_inet_addr srvaddr; |
| + |
| + if (!rtsp_parse_message(pdata, datalen, &dataoff, |
| + &hdrsoff, &hdrslen, |
| + &cseqoff, &cseqlen, |
| + &transoff, &translen)) |
| + break; /* not a valid message */ |
| + |
| + if (strncmp(pdata+cmdoff, "TEARDOWN ", 9) == 0) { |
| + pr_debug("teardown handled\n"); |
| + nf_ct_remove_expectations(ct); /* FIXME must be session id aware */ |
| + break; |
| + } |
| + |
| + if (strncmp(pdata+cmdoff, "SETUP ", 6) != 0) |
| + continue; /* not a SETUP message */ |
| + |
| + srvaddr = ct->tuplehash[!dir].tuple.src.u3; |
| + |
| + /* try to get RTP media source from SETUP URI */ |
| + off = cmdoff + 6; |
| + while (off < datalen) { |
| + if (strncmp(pdata+off, "://", 3) == 0) { |
| + off += 3; |
| + cmdoff = off; |
| + |
| + while (off < datalen) { |
| + if (pdata[off] == ':' || |
| + pdata[off] == '/' || |
| + pdata[off] == ' ') { |
| + in4_pton(pdata + cmdoff, |
| + off - cmdoff, |
| + (u8 *)&srvaddr.in, |
| + -1, NULL); |
| + break; |
| + } |
| + off++; |
| + } |
| + break; |
| + } |
| + off++; |
| + } |
| + |
| + pr_debug("found a setup message\n"); |
| + |
| + off = 0; |
| + if(translen) |
| + rtsp_parse_transport(pdata+transoff, translen, &expinfo); |
| + |
| + if (expinfo.loport == 0) { |
| + pr_debug("no udp transports found\n"); |
| + continue; /* no udp transports found */ |
| + } |
| + |
| + pr_debug("udp transport found, ports=(%d,%hu,%hu)\n", |
| + (int)expinfo.pbtype, expinfo.loport, expinfo.hiport); |
| + |
| + |
| + be_loport = htons(expinfo.loport); |
| + |
| + rtp_exp = nf_ct_expect_alloc(ct); |
| + if (rtp_exp == NULL) { |
| + ret = NF_DROP; |
| + goto out; |
| + } |
| + |
| + nf_ct_expect_init(rtp_exp, NF_CT_EXPECT_CLASS_DEFAULT, |
| + nf_ct_l3num(ct), NULL, |
| + &ct->tuplehash[!dir].tuple.dst.u3, |
| + IPPROTO_UDP, NULL, &be_loport); |
| + |
| + rtp_exp->flags = 0; |
| + |
| + if (expinfo.pbtype == pb_range || expinfo.pbtype == pb_discon) { |
| + pr_debug("setup expectation for rtcp\n"); |
| + |
| + be_hiport = htons(expinfo.hiport); |
| + rtcp_exp = nf_ct_expect_alloc(ct); |
| + if (rtcp_exp == NULL) { |
| + ret = NF_DROP; |
| + goto out1; |
| + } |
| + |
| + nf_ct_expect_init(rtcp_exp, NF_CT_EXPECT_CLASS_DEFAULT, |
| + nf_ct_l3num(ct), NULL, |
| + &ct->tuplehash[!dir].tuple.dst.u3, |
| + IPPROTO_UDP, NULL, &be_hiport); |
| + |
| + rtcp_exp->flags = 0; |
| + |
| + pr_debug("expect_related %pI4:%u-%u-%pI4:%u-%u\n", |
| + &rtp_exp->tuple.src.u3.ip, |
| + ntohs(rtp_exp->tuple.src.u.udp.port), |
| + ntohs(rtcp_exp->tuple.src.u.udp.port), |
| + &rtp_exp->tuple.dst.u3.ip, |
| + ntohs(rtp_exp->tuple.dst.u.udp.port), |
| + ntohs(rtcp_exp->tuple.dst.u.udp.port)); |
| + } else { |
| + pr_debug("expect_related %pI4:%u-%pI4:%u\n", |
| + &rtp_exp->tuple.src.u3.ip, |
| + ntohs(rtp_exp->tuple.src.u.udp.port), |
| + &rtp_exp->tuple.dst.u3.ip, |
| + ntohs(rtp_exp->tuple.dst.u.udp.port)); |
| + } |
| + |
| + nf_nat_rtsp = rcu_dereference(nf_nat_rtsp_hook); |
| + if (nf_nat_rtsp && ct->status & IPS_NAT_MASK) |
| + /* pass the request off to the nat helper */ |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + ret = nf_nat_rtsp(skb, ctinfo, protoff, hdrsoff, hdrslen, |
| + &expinfo, rtp_exp, rtcp_exp); |
| +#else |
| + ret = nf_nat_rtsp(skb, ctinfo, hdrsoff, hdrslen, |
| + &expinfo, rtp_exp, rtcp_exp); |
| +#endif |
| + else { |
| + if (rtsp_nf_ct_expect_related(rtp_exp) == 0) { |
| + if (rtcp_exp && rtsp_nf_ct_expect_related(rtcp_exp) != 0) { |
| + nf_ct_unexpect_related(rtp_exp); |
| + pr_info("nf_conntrack_expect_related failed for rtcp\n"); |
| + ret = NF_DROP; |
| + } |
| + } else { |
| + pr_info("nf_conntrack_expect_related failed for rtp\n"); |
| + ret = NF_DROP; |
| + } |
| + } |
| + if (rtcp_exp) { |
| + nf_ct_expect_put(rtcp_exp); |
| + } |
| +out1: |
| + nf_ct_expect_put(rtp_exp); |
| + goto out; |
| + } |
| +out: |
| + |
| + return ret; |
| +} |
| + |
| + |
| +static inline int |
| +help_in(struct sk_buff *skb, unsigned char *rb_ptr, unsigned int datalen, |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + struct nf_conn *ct, enum ip_conntrack_info ctinfo, |
| + unsigned int protoff) |
| +#else |
| + struct nf_conn *ct, enum ip_conntrack_info ctinfo) |
| +#endif |
| + { |
| + struct ip_ct_rtsp_expect expinfo; |
| + union nf_inet_addr srvaddr; |
| + int dir = CTINFO2DIR(ctinfo); /* = IP_CT_DIR_ORIGINAL */ |
| + int len; |
| + char* pdata = rb_ptr; |
| + uint dataoff = 0; |
| + int ret = NF_ACCEPT; |
| + u_int8_t family; |
| + struct nf_conntrack_expect *exp_ct = NULL; |
| + struct nf_conntrack_tuple t; |
| + struct net *net = nf_ct_net(ct); |
| + |
| + memset(&expinfo, 0, sizeof(expinfo)); |
| + |
| + while (dataoff < datalen) { |
| + uint cmdoff = dataoff; |
| + uint hdrsoff = 0; |
| + uint hdrslen = 0; |
| + uint cseqoff = 0; |
| + uint cseqlen = 0; |
| + uint transoff = 0; |
| + uint translen = 0; |
| + |
| + if (!rtsp_parse_message(pdata, datalen, &dataoff, |
| + &hdrsoff, &hdrslen, |
| + &cseqoff, &cseqlen, |
| + &transoff, &translen)) |
| + break; /* not a valid message */ |
| + |
| + if (strncmp(pdata+cmdoff, "RTSP/", 5) == 0 && translen) { |
| + union nf_inet_addr zeroaddr; |
| + |
| + memset(&zeroaddr, 0, sizeof(zeroaddr)); |
| + |
| + if (!rtsp_parse_transport(pdata+transoff, translen, &expinfo)) |
| + continue; |
| + |
| + srvaddr = expinfo.srvaddr; |
| + |
| + if (nf_inet_addr_cmp(&srvaddr, &ct->tuplehash[dir].tuple.src.u3) || |
| + nf_inet_addr_cmp(&srvaddr, &zeroaddr)) |
| + continue; |
| + } else |
| + continue; /* not valid RTSP reply */ |
| + |
| + if (expinfo.loport == 0) { |
| + pr_debug("no udp transports found\n"); |
| + continue; /* no udp transports found */ |
| + } |
| + |
| + family = nf_ct_l3num(ct); |
| + if (family == AF_INET) |
| + len = 4; |
| + else |
| + len = 16; |
| + |
| + /* replace rtp expect src addr */ |
| + t.src.l3num = family; |
| + t.dst.protonum = IPPROTO_UDP; |
| + |
| + memcpy(&t.src.u3, &ct->tuplehash[dir].tuple.src.u3, len); |
| + if (sizeof(t.src.u3) > len) |
| + /* address needs to be cleared for nf_ct_tuple_equal */ |
| + memset((void *)&t.src.u3 + len, 0, sizeof(t.src.u3) - len); |
| + |
| + t.src.u.all = 0; |
| + |
| + memcpy(&t.dst.u3, &ct->tuplehash[dir].tuple.dst.u3, len); |
| + if (sizeof(t.dst.u3) > len) |
| + /* address needs to be cleared for nf_ct_tuple_equal */ |
| + memset((void *)&t.dst.u3 + len, 0, sizeof(t.dst.u3) - len); |
| + |
| + t.dst.u.all = htons(expinfo.loport); |
| + |
| + /* get the rtp expect and replace the srcaddr with RTP server addr */ |
| + exp_ct = nf_ct_expect_find_get(net, nf_ct_zone(ct), &t); |
| + if (exp_ct) { |
| + memcpy(&exp_ct->tuple.src.u3, &srvaddr, len); |
| + if (sizeof(exp_ct->tuple.src.u3) > len) |
| + /* address needs to be cleared for nf_ct_tuple_equal */ |
| + memset((void *)&exp_ct->tuple.src.u3 + len, 0, |
| + sizeof(exp_ct->tuple.src.u3) - len); |
| + } else |
| + goto out; |
| + |
| + /* replace rtcp expect src addr */ |
| + if (expinfo.pbtype == pb_range || expinfo.pbtype == pb_discon) { |
| + t.dst.u.all = htons(expinfo.hiport); |
| + |
| + /* get the rtcp expect and replace the srcaddr with RTP server addr */ |
| + exp_ct = nf_ct_expect_find_get(net, nf_ct_zone(ct), &t); |
| + if (exp_ct) { |
| + memcpy(&exp_ct->tuple.src.u3, &srvaddr, len); |
| + if (sizeof(exp_ct->tuple.src.u3) > len) |
| + /* address needs to be cleared for nf_ct_tuple_equal */ |
| + memset((void *)&exp_ct->tuple.src.u3 + len, 0, |
| + sizeof(exp_ct->tuple.src.u3) - len); |
| + } else |
| + goto out; |
| + } |
| + |
| + goto out; |
| + } |
| +out: |
| + return ret; |
| + } |
| + |
| +static int help(struct sk_buff *skb, unsigned int protoff, |
| + struct nf_conn *ct, enum ip_conntrack_info ctinfo) |
| +{ |
| + struct tcphdr _tcph, *th; |
| + unsigned int dataoff, datalen; |
| + char *rb_ptr; |
| + int ret = NF_DROP; |
| + |
| + /* Until there's been traffic both ways, don't look in packets. */ |
| + if (ctinfo != IP_CT_ESTABLISHED && |
| + ctinfo != IP_CT_ESTABLISHED + IP_CT_IS_REPLY) { |
| + pr_debug("conntrackinfo = %u\n", ctinfo); |
| + return NF_ACCEPT; |
| + } |
| + |
| + /* Not whole TCP header? */ |
| + th = skb_header_pointer(skb, protoff, sizeof(_tcph), &_tcph); |
| + |
| + if (!th) |
| + return NF_ACCEPT; |
| + |
| + /* No data ? */ |
| + dataoff = protoff + th->doff*4; |
| + datalen = skb->len - dataoff; |
| + if (dataoff >= skb->len) |
| + return NF_ACCEPT; |
| + |
| + spin_lock_bh(&rtsp_buffer_lock); |
| + rb_ptr = skb_header_pointer(skb, dataoff, |
| + skb->len - dataoff, rtsp_buffer); |
| + BUG_ON(rb_ptr == NULL); |
| + |
| +#if 0 |
| + /* Checksum invalid? Ignore. */ |
| + /* FIXME: Source route IP option packets --RR */ |
| + if (tcp_v4_check(tcph, tcplen, iph->saddr, iph->daddr, |
| + csum_partial((char*)tcph, tcplen, 0))) |
| + { |
| + DEBUGP("bad csum: %p %u %u.%u.%u.%u %u.%u.%u.%u\n", |
| + tcph, tcplen, NIPQUAD(iph->saddr), NIPQUAD(iph->daddr)); |
| + return NF_ACCEPT; |
| + } |
| +#endif |
| + |
| + switch (CTINFO2DIR(ctinfo)) { |
| + case IP_CT_DIR_ORIGINAL: |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + ret = help_out(skb, rb_ptr, datalen, ct, ctinfo, protoff); |
| +#else |
| + ret = help_out(skb, rb_ptr, datalen, ct, ctinfo); |
| +#endif |
| + break; |
| + case IP_CT_DIR_REPLY: |
| + pr_debug("IP_CT_DIR_REPLY\n"); |
| + /* inbound packet: server->client */ |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + ret = help_in(skb, rb_ptr, datalen, ct, ctinfo, protoff); |
| +#else |
| + ret = help_in(skb, rb_ptr, datalen, ct, ctinfo); |
| +#endif |
| + break; |
| + default: |
| + break; |
| + } |
| + |
| + spin_unlock_bh(&rtsp_buffer_lock); |
| + |
| + return ret; |
| +} |
| + |
| +static struct nf_conntrack_helper rtsp_helpers[MAX_PORTS]; |
| +static char rtsp_names[MAX_PORTS][10]; |
| + |
| +/* This function is intentionally _NOT_ defined as __exit */ |
| +static void |
| +fini(void) |
| +{ |
| + int i; |
| + for (i = 0; i < num_ports; i++) { |
| + pr_debug("unregistering port %d\n", ports[i]); |
| + nf_conntrack_helper_unregister(&rtsp_helpers[i]); |
| + } |
| + kfree(rtsp_buffer); |
| +} |
| + |
| +static int __init |
| +init(void) |
| +{ |
| + int i, ret; |
| + struct nf_conntrack_helper *hlpr; |
| + char *tmpname; |
| + |
| + printk("nf_conntrack_rtsp v" IP_NF_RTSP_VERSION " loading\n"); |
| + |
| + if (max_outstanding < 1) { |
| + printk("nf_conntrack_rtsp: max_outstanding must be a positive integer\n"); |
| + return -EBUSY; |
| + } |
| + |
| + rtsp_exp_policy.max_expected = max_outstanding; |
| + rtsp_exp_policy.timeout = setup_timeout; |
| + |
| + rtsp_buffer = kmalloc(65536, GFP_KERNEL); |
| + if (!rtsp_buffer) |
| + return -ENOMEM; |
| + |
| + /* If no port given, default to standard rtsp port */ |
| + if (ports[0] == 0) { |
| + ports[0] = RTSP_PORT; |
| + num_ports = 1; |
| + } |
| + |
| + for (i = 0; (i < MAX_PORTS) && ports[i]; i++) { |
| + hlpr = &rtsp_helpers[i]; |
| + memset(hlpr, 0, sizeof(struct nf_conntrack_helper)); |
| + hlpr->tuple.src.l3num = AF_INET; |
| + hlpr->tuple.src.u.tcp.port = htons(ports[i]); |
| + hlpr->tuple.dst.protonum = IPPROTO_TCP; |
| + hlpr->expect_policy = &rtsp_exp_policy; |
| + hlpr->me = THIS_MODULE; |
| + hlpr->help = help; |
| + |
| + tmpname = &rtsp_names[i][0]; |
| + if (ports[i] == RTSP_PORT) { |
| + sprintf(tmpname, "rtsp"); |
| + } else { |
| + sprintf(tmpname, "rtsp-%d", i); |
| + } |
| + |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,6,0) |
| + strlcpy(hlpr->name, tmpname, sizeof(hlpr->name)); |
| +#else |
| + hlpr->name = tmpname; |
| +#endif |
| + pr_debug("port #%d: %d\n", i, ports[i]); |
| + |
| + ret = nf_conntrack_helper_register(hlpr); |
| + |
| + if (ret) { |
| + printk("nf_conntrack_rtsp: ERROR registering port %d\n", ports[i]); |
| + fini(); |
| + return -EBUSY; |
| + } |
| + } |
| + return 0; |
| +} |
| + |
| +module_init(init); |
| +module_exit(fini); |
| --- /dev/null |
| +++ b/extensions/rtsp/nf_conntrack_rtsp.h |
| @@ -0,0 +1,74 @@ |
| +/* |
| + * RTSP extension for IP connection tracking. |
| + * (C) 2003 by Tom Marshall <tmarshall at real.com> |
| + * based on ip_conntrack_irc.h |
| + * |
| + * This program is free software; you can redistribute it and/or |
| + * modify it under the terms of the GNU General Public License |
| + * as published by the Free Software Foundation; either version |
| + * 2 of the License, or (at your option) any later version. |
| + * |
| + * 2013-03-04: Il'inykh Sergey <sergeyi at inango-sw.com>. Inango Systems Ltd |
| + * - conditional compilation for kernel 3.7 |
| + * - port mapping improvements |
| +*/ |
| +#ifndef _IP_CONNTRACK_RTSP_H |
| +#define _IP_CONNTRACK_RTSP_H |
| + |
| +#include <linux/version.h> |
| + |
| +//#define IP_NF_RTSP_DEBUG 1 |
| +#define IP_NF_RTSP_VERSION "0.7" |
| + |
| +#ifdef __KERNEL__ |
| +/* port block types */ |
| +typedef enum { |
| + pb_single, /* client_port=x */ |
| + pb_range, /* client_port=x-y */ |
| + pb_discon /* client_port=x/y (rtspbis) */ |
| +} portblock_t; |
| + |
| +/* We record seq number and length of rtsp headers here, all in host order. */ |
| + |
| +/* |
| + * This structure is per expected connection. It is a member of struct |
| + * ip_conntrack_expect. The TCP SEQ for the conntrack expect is stored |
| + * there and we are expected to only store the length of the data which |
| + * needs replaced. If a packet contains multiple RTSP messages, we create |
| + * one expected connection per message. |
| + * |
| + * We use these variables to mark the entire header block. This may seem |
| + * like overkill, but the nature of RTSP requires it. A header may appear |
| + * multiple times in a message. We must treat two Transport headers the |
| + * same as one Transport header with two entries. |
| + */ |
| +struct ip_ct_rtsp_expect |
| +{ |
| + u_int32_t len; /* length of header block */ |
| + portblock_t pbtype; /* Type of port block that was requested */ |
| + u_int16_t loport; /* Port that was requested, low or first */ |
| + u_int16_t hiport; /* Port that was requested, high or second */ |
| + union nf_inet_addr srvaddr; /* src address in SETUP reply */ |
| + union nf_inet_addr cltaddr; /* destination address */ |
| +#if 0 |
| + uint method; /* RTSP method */ |
| + uint cseq; /* CSeq from request */ |
| +#endif |
| +}; |
| + |
| +extern unsigned int (*nf_nat_rtsp_hook)(struct sk_buff *skb, |
| + enum ip_conntrack_info ctinfo, |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + unsigned int protoff, |
| +#endif |
| + unsigned int matchoff, |
| + unsigned int matchlen, |
| + struct ip_ct_rtsp_expect *prtspexp, |
| + struct nf_conntrack_expect *rtp_exp, |
| + struct nf_conntrack_expect *rtcp_exp); |
| + |
| +#define RTSP_PORT 554 |
| + |
| +#endif /* __KERNEL__ */ |
| + |
| +#endif /* _IP_CONNTRACK_RTSP_H */ |
| --- /dev/null |
| +++ b/extensions/rtsp/nf_nat_rtsp.c |
| @@ -0,0 +1,640 @@ |
| +/* |
| + * RTSP extension for TCP NAT alteration |
| + * (C) 2003 by Tom Marshall <tmarshall at real.com> |
| + * |
| + * 2013-03-04: Il'inykh Sergey <sergeyi at inango-sw.com>. Inango Systems Ltd |
| + * - fixed rtcp nat mapping and other port mapping fixes |
| + * - fixed system hard lock because of bug in the parser |
| + * - codestyle fixes and less significant fixes |
| + * |
| + * based on ip_nat_irc.c |
| + * |
| + * This program is free software; you can redistribute it and/or |
| + * modify it under the terms of the GNU General Public License |
| + * as published by the Free Software Foundation; either version |
| + * 2 of the License, or (at your option) any later version. |
| + * |
| + * Module load syntax: |
| + * insmod nf_nat_rtsp.o ports=port1,port2,...port<MAX_PORTS> |
| + * stunaddr=<address> |
| + * destaction=[auto|strip|none] |
| + * |
| + * If no ports are specified, the default will be port 554 only. |
| + * |
| + * stunaddr specifies the address used to detect that a client is using STUN. |
| + * If this address is seen in the destination parameter, it is assumed that |
| + * the client has already punched a UDP hole in the firewall, so we don't |
| + * mangle the client_port. If none is specified, it is autodetected. It |
| + * only needs to be set if you have multiple levels of NAT. It should be |
| + * set to the external address that the STUN clients detect. Note that in |
| + * this case, it will not be possible for clients to use UDP with servers |
| + * between the NATs. |
| + * |
| + * If no destaction is specified, auto is used. |
| + * destaction=auto: strip destination parameter if it is not stunaddr. |
| + * destaction=strip: always strip destination parameter (not recommended). |
| + * destaction=none: do not touch destination parameter (not recommended). |
| + */ |
| + |
| +#include <linux/module.h> |
| +#include <linux/version.h> |
| +#include <net/tcp.h> |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| +# include <net/netfilter/nf_nat.h> |
| +#else |
| +# include <net/netfilter/nf_nat_rule.h> |
| +#endif |
| +#include <net/netfilter/nf_nat_helper.h> |
| +#include "nf_conntrack_rtsp.h" |
| + |
| +#include <linux/inet.h> |
| +#include <linux/ctype.h> |
| +#define NF_NEED_STRNCASECMP |
| +#define NF_NEED_STRTOU16 |
| +#include "netfilter_helpers.h" |
| +#define NF_NEED_MIME_NEXTLINE |
| +#include "netfilter_mime.h" |
| + |
| +#define MAX_PORTS 8 |
| +#define DSTACT_AUTO 0 |
| +#define DSTACT_STRIP 1 |
| +#define DSTACT_NONE 2 |
| + |
| +static char* stunaddr = NULL; |
| +static char* destaction = NULL; |
| + |
| +static u_int32_t extip = 0; |
| +static int dstact = 0; |
| + |
| +static void nf_nat_rtsp_expected(struct nf_conn* ct, struct nf_conntrack_expect *exp); |
| + |
| +MODULE_AUTHOR("Tom Marshall <tmarshall at real.com>"); |
| +MODULE_DESCRIPTION("RTSP network address translation module"); |
| +MODULE_LICENSE("GPL"); |
| +module_param(stunaddr, charp, 0644); |
| +MODULE_PARM_DESC(stunaddr, "Address for detecting STUN"); |
| +module_param(destaction, charp, 0644); |
| +MODULE_PARM_DESC(destaction, "Action for destination parameter (auto/strip/none)"); |
| + |
| +#define SKIP_WSPACE(ptr,len,off) while(off < len && isspace(*(ptr+off))) { off++; } |
| + |
| +/*** helper functions ***/ |
| + |
| +static void |
| +get_skb_tcpdata(struct sk_buff* skb, char** pptcpdata, uint* ptcpdatalen) |
| +{ |
| + struct iphdr* iph = ip_hdr(skb); |
| + struct tcphdr* tcph = (void *)iph + ip_hdrlen(skb); |
| + |
| + *pptcpdata = (char*)tcph + tcph->doff*4; |
| + *ptcpdatalen = ((char*)skb_transport_header(skb) + skb->len) - *pptcpdata; |
| +} |
| + |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| +/* copy of sip_sprintf_addr */ |
| +static int rtsp_sprintf_addr(const struct nf_conn *ct, char *buffer, |
| + const union nf_inet_addr *addr, bool delim) |
| +{ |
| + if (nf_ct_l3num(ct) == NFPROTO_IPV4) { |
| + return sprintf(buffer, "%pI4", &addr->ip); |
| + } else { |
| + if (delim) |
| + return sprintf(buffer, "[%pI6c]", &addr->ip6); |
| + else |
| + return sprintf(buffer, "%pI6c", &addr->ip6); |
| + } |
| +} |
| +#endif |
| + |
| +/*** nat functions ***/ |
| + |
| +/* |
| + * Mangle the "Transport:" header: |
| + * - Replace all occurences of "client_port=<spec>" |
| + * - Handle destination parameter |
| + * |
| + * In: |
| + * ct, ctinfo = conntrack context |
| + * skb = packet |
| + * tranoff = Transport header offset from TCP data |
| + * tranlen = Transport header length (incl. CRLF) |
| + * rport_lo = replacement low port (host endian) |
| + * rport_hi = replacement high port (host endian) |
| + * |
| + * Returns packet size difference. |
| + * |
| + * Assumes that a complete transport header is present, ending with CR or LF |
| + */ |
| +static int |
| +rtsp_mangle_tran(enum ip_conntrack_info ctinfo, |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + unsigned int protoff, |
| +#endif |
| + struct nf_conntrack_expect* rtp_exp, |
| + struct nf_conntrack_expect* rtcp_exp, |
| + struct ip_ct_rtsp_expect* prtspexp, |
| + struct sk_buff* skb, uint tranoff, uint tranlen) |
| +{ |
| + char* ptcp; |
| + uint tcplen; |
| + char* ptran; |
| + char rbuf1[16]; /* Replacement buffer (one port) */ |
| + uint rbuf1len; /* Replacement len (one port) */ |
| + char rbufa[16]; /* Replacement buffer (all ports) */ |
| + uint rbufalen; /* Replacement len (all ports) */ |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + union nf_inet_addr newip; |
| +#else |
| + u_int32_t newip; |
| +#endif |
| + u_int16_t loport, hiport; |
| + uint off = 0; |
| + uint diff; /* Number of bytes we removed */ |
| + |
| + struct nf_conn *ct = rtp_exp->master; |
| + /* struct nf_conn *ct = nf_ct_get(skb, &ctinfo); */ |
| + struct nf_conntrack_tuple *rtp_t; |
| + |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + char szextaddr[INET6_ADDRSTRLEN + 16]; |
| +#else |
| + char szextaddr[INET_ADDRSTRLEN + 16]; |
| +#endif |
| + uint extaddrlen; |
| + int is_stun; |
| + |
| + get_skb_tcpdata(skb, &ptcp, &tcplen); |
| + ptran = ptcp+tranoff; |
| + |
| + if (tranoff+tranlen > tcplen || tcplen-tranoff < tranlen || |
| + tranlen < 10 || !iseol(ptran[tranlen-1]) || |
| + nf_strncasecmp(ptran, "Transport:", 10) != 0) { |
| + pr_info("sanity check failed\n"); |
| + return 0; |
| + } |
| + off += 10; |
| + SKIP_WSPACE(ptcp+tranoff, tranlen, off); |
| + |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + newip = ct->tuplehash[IP_CT_DIR_REPLY].tuple.dst.u3; |
| + rtp_t = &rtp_exp->tuple; |
| + rtp_t->dst.u3 = newip; |
| + if (rtcp_exp) { |
| + rtcp_exp->tuple.dst.u3 = newip; |
| + } |
| + extaddrlen = rtsp_sprintf_addr(ct, szextaddr, &newip, true); // FIXME handle extip |
| + pr_debug("stunaddr=%s (auto)\n", szextaddr); |
| +#else |
| + newip = ct->tuplehash[IP_CT_DIR_REPLY].tuple.dst.u3.ip; |
| + rtp_t = &rtp_exp->tuple; |
| + rtp_t->dst.u3.ip = newip; |
| + if (rtcp_exp) { |
| + rtcp_exp->tuple.dst.u3.ip = newip; |
| + } |
| + extaddrlen = extip ? sprintf(szextaddr, "%pI4", &extip) |
| + : sprintf(szextaddr, "%pI4", &newip); |
| + pr_debug("stunaddr=%s (%s)\n", szextaddr, (extip?"forced":"auto")); |
| +#endif |
| + hiport = 0; |
| + rbuf1len = rbufalen = 0; |
| + switch (prtspexp->pbtype) { |
| + case pb_single: |
| + for (loport = prtspexp->loport; loport != 0; loport++) { /* XXX: improper wrap? */ |
| + rtp_t->dst.u.udp.port = htons(loport); |
| + if (rtsp_nf_ct_expect_related(rtp_exp) == 0) { |
| + pr_debug("using port %hu\n", loport); |
| + break; |
| + } |
| + } |
| + if (loport != 0) { |
| + rbuf1len = sprintf(rbuf1, "%hu", loport); |
| + rbufalen = sprintf(rbufa, "%hu", loport); |
| + } |
| + break; |
| + case pb_range: |
| + for (loport = prtspexp->loport; loport != 0; loport += 2) { /* XXX: improper wrap? */ |
| + rtp_t->dst.u.udp.port = htons(loport); |
| + if (rtsp_nf_ct_expect_related(rtp_exp) != 0) { |
| + continue; |
| + } |
| + hiport = loport + 1; |
| + rtcp_exp->tuple.dst.u.udp.port = htons(hiport); |
| + if (rtsp_nf_ct_expect_related(rtcp_exp) != 0) { |
| + nf_ct_unexpect_related(rtp_exp); |
| + continue; |
| + } |
| + |
| + /* FIXME: invalid print in case of ipv6 */ |
| + pr_debug("nat expect_related %pI4:%u-%u-%pI4:%u-%u\n", |
| + &rtp_exp->tuple.src.u3.ip, |
| + ntohs(rtp_exp->tuple.src.u.udp.port), |
| + ntohs(rtcp_exp->tuple.src.u.udp.port), |
| + &rtp_exp->tuple.dst.u3.ip, |
| + ntohs(rtp_exp->tuple.dst.u.udp.port), |
| + ntohs(rtcp_exp->tuple.dst.u.udp.port)); |
| + break; |
| + } |
| + if (loport != 0) { |
| + rbuf1len = sprintf(rbuf1, "%hu", loport); |
| + rbufalen = sprintf(rbufa, "%hu-%hu", loport, hiport); |
| + } |
| + break; |
| + case pb_discon: |
| + for (loport = prtspexp->loport; loport != 0; loport++) { /* XXX: improper wrap? */ |
| + rtp_t->dst.u.udp.port = htons(loport); |
| + if (rtsp_nf_ct_expect_related(rtp_exp) == 0) { |
| + pr_debug("using port %hu (1 of 2)\n", loport); |
| + break; |
| + } |
| + } |
| + for (hiport = prtspexp->hiport; hiport != 0; hiport++) { /* XXX: improper wrap? */ |
| + rtcp_exp->tuple.dst.u.udp.port = htons(hiport); |
| + if (rtsp_nf_ct_expect_related(rtcp_exp) == 0) { |
| + pr_debug("using port %hu (2 of 2)\n", hiport); |
| + break; |
| + } |
| + } |
| + if (loport != 0 && hiport != 0) { |
| + rbuf1len = sprintf(rbuf1, "%hu", loport); |
| + rbufalen = sprintf(rbufa, hiport == loport+1 ? |
| + "%hu-%hu":"%hu/%hu", loport, hiport); |
| + } else { |
| + if (loport != 0) |
| + nf_ct_unexpect_related(rtp_exp); |
| + if (hiport != 0) |
| + nf_ct_unexpect_related(rtcp_exp); |
| + } |
| + break; |
| + } |
| + |
| + if (rbuf1len == 0) |
| + return 0; /* cannot get replacement port(s) */ |
| + |
| + /* Transport: tran;field;field=val,tran;field;field=val,... |
| + `off` is set to the start of Transport value from start of line |
| + */ |
| + while (off < tranlen) { |
| + uint saveoff; |
| + const char* pparamend; |
| + uint nextparamoff; |
| + |
| + pparamend = memchr(ptran+off, ',', tranlen-off); |
| + pparamend = (pparamend == NULL) ? ptran+tranlen : pparamend+1; |
| + nextparamoff = pparamend-ptran; |
| + |
| + /* |
| + * We pass over each param twice. On the first pass, we look for a |
| + * destination= field. It is handled by the security policy. If it |
| + * is present, allowed, and equal to our external address, we assume |
| + * that STUN is being used and we leave the client_port= field alone. |
| + */ |
| + is_stun = 0; |
| + saveoff = off; |
| + while (off < nextparamoff) { |
| + const char* pfieldend; |
| + uint nextfieldoff; |
| + |
| + pfieldend = memchr(ptran+off, ';', nextparamoff-off); |
| + nextfieldoff = (pfieldend == NULL) ? nextparamoff : pfieldend-ptran+1; |
| + SKIP_WSPACE(ptran, nextfieldoff, off); |
| + |
| + if (dstact != DSTACT_NONE && nextfieldoff - off > 12 && strncmp(ptran+off, "destination=", 12) == 0) { |
| + if (strncmp(ptran+off+12, szextaddr, extaddrlen) == 0) |
| + is_stun = 1; |
| + |
| + if (dstact == DSTACT_STRIP || (dstact == DSTACT_AUTO && !is_stun)) { |
| + uint dstoff = (ptran-ptcp)+off; |
| + uint dstlen = nextfieldoff-off; |
| + char* pdstrep = NULL; |
| + uint dstreplen = 0; |
| + diff = dstlen; |
| + if (dstact == DSTACT_AUTO && !is_stun) { |
| + const char* psep = memchr(ptran+off, ':', dstlen); |
| + u_int16_t port; |
| + |
| + dstoff += 12; |
| + dstlen -= 13; |
| + pdstrep = szextaddr; |
| + |
| + if (psep != NULL && nf_strtou16(psep + 1, &port)) { |
| + pr_debug("RTSP: replace dst addr&port\n"); |
| + |
| + if (port != prtspexp->loport) { |
| + pr_debug("multiple ports found, port %hu ignored\n", port); |
| + dstreplen = extaddrlen; |
| + } else { |
| + sprintf(szextaddr+extaddrlen, ":%s", rbuf1); |
| + dstreplen = extaddrlen+1+rbuf1len; |
| + } |
| + } else { |
| + pr_debug("RTSP: replace dst addr\n"); |
| + dstreplen = extaddrlen; |
| + } |
| + diff = nextfieldoff-off-13-dstreplen; |
| + } |
| + |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + if (!nf_nat_mangle_tcp_packet(skb, ct, ctinfo, protoff, |
| + dstoff, dstlen, pdstrep, dstreplen)) { |
| +#else |
| + if (!nf_nat_mangle_tcp_packet(skb, ct, ctinfo, |
| + dstoff, dstlen, pdstrep, dstreplen)) { |
| +#endif |
| + /* mangle failed, all we can do is bail */ |
| + nf_ct_unexpect_related(rtp_exp); |
| + if (rtcp_exp) |
| + nf_ct_unexpect_related(rtcp_exp); |
| + return 0; |
| + } |
| + get_skb_tcpdata(skb, &ptcp, &tcplen); |
| + ptran = ptcp+tranoff; |
| + tranlen -= diff; |
| + nextparamoff -= diff; |
| + nextfieldoff -= diff; |
| + } |
| + } |
| + |
| + off = nextfieldoff; |
| + } |
| + |
| + if (is_stun) |
| + continue; |
| + |
| + off = saveoff; |
| + while (off < nextparamoff) { |
| + const char* pfieldend; |
| + uint nextfieldoff; |
| + |
| + pfieldend = memchr(ptran+off, ';', nextparamoff-off); |
| + nextfieldoff = (pfieldend == NULL) ? nextparamoff : pfieldend-ptran+1; |
| + |
| + if (nextfieldoff - off > 12 && strncmp(ptran+off, "client_port=", 12) == 0) { |
| + u_int16_t port; |
| + uint numlen; |
| + uint origoff; |
| + uint origlen; |
| + char* rbuf = rbuf1; |
| + uint rbuflen = rbuf1len; |
| + |
| + off += 12; |
| + origoff = (ptran-ptcp)+off; |
| + origlen = 0; |
| + numlen = nf_strtou16(ptran+off, &port); |
| + off += numlen; |
| + origlen += numlen; |
| + if (port != prtspexp->loport) { |
| + pr_debug("multiple ports found, port %hu ignored\n", port); |
| + } else { |
| + if (ptran[off] == '-' || ptran[off] == '/') { |
| + off++; |
| + origlen++; |
| + numlen = nf_strtou16(ptran+off, &port); |
| + off += numlen; |
| + origlen += numlen; |
| + rbuf = rbufa; |
| + rbuflen = rbufalen; |
| + } |
| + |
| + /* |
| + * note we cannot just memcpy() if the sizes are the same. |
| + * the mangle function does skb resizing, checks for a |
| + * cloned skb, and updates the checksums. |
| + * |
| + * parameter 4 below is offset from start of tcp data. |
| + */ |
| + diff = origlen-rbuflen; |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + if (!nf_nat_mangle_tcp_packet(skb, ct, ctinfo, protoff, |
| + origoff, origlen, rbuf, rbuflen)) { |
| +#else |
| + if (!nf_nat_mangle_tcp_packet(skb, ct, ctinfo, |
| + origoff, origlen, rbuf, rbuflen)) { |
| +#endif |
| + /* mangle failed, all we can do is bail */ |
| + nf_ct_unexpect_related(rtp_exp); |
| + if (rtcp_exp) |
| + nf_ct_unexpect_related(rtcp_exp); |
| + return 0; |
| + } |
| + get_skb_tcpdata(skb, &ptcp, &tcplen); |
| + ptran = ptcp+tranoff; |
| + tranlen -= diff; |
| + nextparamoff -= diff; |
| + nextfieldoff -= diff; |
| + } |
| + } |
| + |
| + off = nextfieldoff; |
| + } |
| + |
| + off = nextparamoff; |
| + } |
| + |
| + return 1; |
| +} |
| + |
| +static uint |
| +help_out(struct sk_buff *skb, enum ip_conntrack_info ctinfo, |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + unsigned int protoff, |
| +#endif |
| + unsigned int matchoff, unsigned int matchlen, |
| + struct ip_ct_rtsp_expect* prtspexp, |
| + struct nf_conntrack_expect* rtp_exp, |
| + struct nf_conntrack_expect* rtcp_exp) |
| +{ |
| + char* ptcp; |
| + uint tcplen; |
| + uint hdrsoff; |
| + uint hdrslen; |
| + uint lineoff; |
| + uint linelen; |
| + uint off; |
| + int dir = CTINFO2DIR(ctinfo); |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + union nf_inet_addr saddr = rtp_exp->master->tuplehash[dir].tuple.src.u3; |
| +#else |
| + __be32 saddr = rtp_exp->master->tuplehash[dir].tuple.src.u3.ip; |
| +#endif |
| + |
| + //struct iphdr* iph = (struct iphdr*)(*pskb)->nh.iph; |
| + //struct tcphdr* tcph = (struct tcphdr*)((void*)iph + iph->ihl*4); |
| + |
| + get_skb_tcpdata(skb, &ptcp, &tcplen); |
| + hdrsoff = matchoff;//exp->seq - ntohl(tcph->seq); |
| + hdrslen = matchlen; |
| + off = hdrsoff; |
| + pr_debug("NAT rtsp help_out\n"); |
| + |
| + while (nf_mime_nextline(ptcp, hdrsoff+hdrslen, &off, &lineoff, &linelen)) { |
| + if (linelen == 0) |
| + break; |
| + |
| + if (off > hdrsoff+hdrslen) { |
| + pr_info("!! overrun !!"); |
| + break; |
| + } |
| + pr_debug("hdr: len=%u, %.*s", linelen, (int)linelen, ptcp+lineoff); |
| + |
| + if (nf_strncasecmp(ptcp+lineoff, "Transport:", 10) == 0) { |
| + uint oldtcplen = tcplen; |
| + pr_debug("hdr: Transport\n"); |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + if (!rtsp_mangle_tran(ctinfo, protoff, rtp_exp, rtcp_exp, |
| + prtspexp, skb, lineoff, linelen)) { |
| +#else |
| + if (!rtsp_mangle_tran(ctinfo, rtp_exp, rtcp_exp, prtspexp, |
| + skb, lineoff, linelen)) { |
| +#endif |
| + pr_debug("hdr: Transport mangle failed"); |
| + break; |
| + } |
| + rtp_exp->expectfn = nf_nat_rtsp_expected; |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + rtp_exp->saved_addr = saddr; |
| +#else |
| + rtp_exp->saved_ip = saddr; |
| +#endif |
| + rtp_exp->saved_proto.udp.port = htons(prtspexp->loport); |
| + rtp_exp->dir = !dir; |
| + if (rtcp_exp) { |
| + rtcp_exp->expectfn = nf_nat_rtsp_expected; |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + rtcp_exp->saved_addr = saddr; |
| +#else |
| + rtcp_exp->saved_ip = saddr; |
| +#endif |
| + rtcp_exp->saved_proto.udp.port = htons(prtspexp->hiport); |
| + rtcp_exp->dir = !dir; |
| + } |
| + get_skb_tcpdata(skb, &ptcp, &tcplen); |
| + hdrslen -= (oldtcplen-tcplen); |
| + off -= (oldtcplen-tcplen); |
| + lineoff -= (oldtcplen-tcplen); |
| + linelen -= (oldtcplen-tcplen); |
| + pr_debug("rep: len=%u, %.*s", linelen, (int)linelen, ptcp+lineoff); |
| + } |
| + } |
| + |
| + return NF_ACCEPT; |
| +} |
| + |
| +static unsigned int |
| +nf_nat_rtsp(struct sk_buff *skb, enum ip_conntrack_info ctinfo, |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + unsigned int protoff, |
| +#endif |
| + unsigned int matchoff, unsigned int matchlen, |
| + struct ip_ct_rtsp_expect* prtspexp, |
| + struct nf_conntrack_expect* rtp_exp, |
| + struct nf_conntrack_expect* rtcp_exp) |
| +{ |
| + int dir = CTINFO2DIR(ctinfo); |
| + int rc = NF_ACCEPT; |
| + |
| + switch (dir) { |
| + case IP_CT_DIR_ORIGINAL: |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + rc = help_out(skb, ctinfo, protoff, matchoff, matchlen, prtspexp, |
| + rtp_exp, rtcp_exp); |
| +#else |
| + rc = help_out(skb, ctinfo, matchoff, matchlen, prtspexp, |
| + rtp_exp, rtcp_exp); |
| +#endif |
| + break; |
| + case IP_CT_DIR_REPLY: |
| + pr_debug("unmangle ! %u\n", ctinfo); |
| + /* XXX: unmangle */ |
| + rc = NF_ACCEPT; |
| + break; |
| + } |
| + //UNLOCK_BH(&ip_rtsp_lock); |
| + |
| + return rc; |
| +} |
| + |
| +static void nf_nat_rtsp_expected(struct nf_conn* ct, struct nf_conntrack_expect *exp) |
| +{ |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(4,18,0) |
| + struct nf_nat_range2 range; |
| +#elif LINUX_VERSION_CODE < KERNEL_VERSION(3,3,0) || LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + struct nf_nat_range range; |
| +#else |
| + struct nf_nat_ipv4_range range; |
| +#endif |
| + |
| + /* This must be a fresh one. */ |
| + BUG_ON(ct->status & IPS_NAT_DONE_MASK); |
| + |
| + /* For DST manip, map port here to where it's expected. */ |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + range.min_proto = range.max_proto = exp->saved_proto; |
| + range.min_addr = range.max_addr = exp->saved_addr; |
| +#else |
| + range.min = range.max = exp->saved_proto; |
| + range.min_ip = range.max_ip = exp->saved_ip; |
| +#endif |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,3,0) |
| + range.flags = (NF_NAT_RANGE_MAP_IPS | NF_NAT_RANGE_PROTO_SPECIFIED); |
| + nf_nat_setup_info(ct, &range, NF_NAT_MANIP_DST); |
| +#else |
| + range.flags = (IP_NAT_RANGE_MAP_IPS | IP_NAT_RANGE_PROTO_SPECIFIED); |
| + nf_nat_setup_info(ct, &range, IP_NAT_MANIP_DST); |
| +#endif |
| + |
| + /* Change src to where master sends to, but only if the connection |
| + * actually came from the same source. */ |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,7,0) |
| + if (nf_inet_addr_cmp(&ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.src.u3, |
| + &ct->master->tuplehash[exp->dir].tuple.src.u3)) { |
| + range.min_addr = range.max_addr |
| + = ct->master->tuplehash[!exp->dir].tuple.dst.u3; |
| +#else |
| + if (ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.src.u3.ip == |
| + ct->master->tuplehash[exp->dir].tuple.src.u3.ip) { |
| + range.min_ip = range.max_ip |
| + = ct->master->tuplehash[!exp->dir].tuple.dst.u3.ip; |
| +#endif |
| +#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,3,0) |
| + range.flags = NF_NAT_RANGE_MAP_IPS; |
| + nf_nat_setup_info(ct, &range, NF_NAT_MANIP_SRC); |
| +#else |
| + range.flags = IP_NAT_RANGE_MAP_IPS; |
| + nf_nat_setup_info(ct, &range, IP_NAT_MANIP_SRC); |
| +#endif |
| + } |
| +} |
| + |
| + |
| +static void __exit fini(void) |
| +{ |
| + rcu_assign_pointer(nf_nat_rtsp_hook, NULL); |
| + synchronize_net(); |
| +} |
| + |
| +static int __init init(void) |
| +{ |
| + printk("nf_nat_rtsp v" IP_NF_RTSP_VERSION " loading\n"); |
| + |
| + BUG_ON(nf_nat_rtsp_hook); |
| + rcu_assign_pointer(nf_nat_rtsp_hook, nf_nat_rtsp); |
| + |
| + if (stunaddr != NULL) |
| + extip = in_aton(stunaddr); |
| + |
| + if (destaction != NULL) { |
| + if (strcmp(destaction, "auto") == 0) |
| + dstact = DSTACT_AUTO; |
| + |
| + if (strcmp(destaction, "strip") == 0) |
| + dstact = DSTACT_STRIP; |
| + |
| + if (strcmp(destaction, "none") == 0) |
| + dstact = DSTACT_NONE; |
| + } |
| + |
| + return 0; |
| +} |
| + |
| +module_init(init); |
| +module_exit(fini); |
| --- a/extensions/Kbuild |
| +++ b/extensions/Kbuild |
| @@ -27,6 +27,7 @@ obj-${build_lscan} += xt_lscan.o |
| obj-${build_pknock} += pknock/ |
| obj-${build_psd} += xt_psd.o |
| obj-${build_quota2} += xt_quota2.o |
| +obj-${build_rtsp} += rtsp/ |
| |
| -include ${M}/*.Kbuild |
| -include ${M}/Kbuild.* |
| --- a/mconfig |
| +++ b/mconfig |
| @@ -23,3 +23,4 @@ build_lscan=m |
| build_pknock=m |
| build_psd=m |
| build_quota2=m |
| +build_rtsp=m |