1// SPDX-License-Identifier: GPL-2.0-only
2/* SIP extension for NAT alteration.
3 *
4 * (C) 2005 by Christian Hentschel <chentschel@arnet.com.ar>
5 * based on RR's ip_nat_ftp.c and other modules.
6 * (C) 2007 United Security Providers
7 * (C) 2007, 2008, 2011, 2012 Patrick McHardy <kaber@trash.net>
8 */
9
10#include <linux/module.h>
11#include <linux/skbuff.h>
12#include <linux/inet.h>
13#include <linux/udp.h>
14#include <linux/tcp.h>
15
16#include <net/netfilter/nf_nat.h>
17#include <net/netfilter/nf_nat_helper.h>
18#include <net/netfilter/nf_conntrack_core.h>
19#include <net/netfilter/nf_conntrack_helper.h>
20#include <net/netfilter/nf_conntrack_expect.h>
21#include <net/netfilter/nf_conntrack_seqadj.h>
22#include <linux/netfilter/nf_conntrack_sip.h>
23
24#define NAT_HELPER_NAME "sip"
25
26MODULE_LICENSE("GPL");
27MODULE_AUTHOR("Christian Hentschel <chentschel@arnet.com.ar>");
28MODULE_DESCRIPTION("SIP NAT helper");
29MODULE_ALIAS_NF_NAT_HELPER(NAT_HELPER_NAME);
30
31static struct nf_conntrack_nat_helper nat_helper_sip =
32	NF_CT_NAT_HELPER_INIT(NAT_HELPER_NAME);
33
34static unsigned int mangle_packet(struct sk_buff *skb, unsigned int protoff,
35				  unsigned int dataoff,
36				  const char **dptr, unsigned int *datalen,
37				  unsigned int matchoff, unsigned int matchlen,
38				  const char *buffer, unsigned int buflen)
39{
40	enum ip_conntrack_info ctinfo;
41	struct nf_conn *ct = nf_ct_get(skb, &ctinfo);
42	struct tcphdr *th;
43	unsigned int baseoff;
44
45	if (nf_ct_protonum(ct) == IPPROTO_TCP) {
46		th = (struct tcphdr *)(skb->data + protoff);
47		baseoff = protoff + th->doff * 4;
48		matchoff += dataoff - baseoff;
49
50		if (!__nf_nat_mangle_tcp_packet(skb, ct, ctinfo,
51						protoff, matchoff, matchlen,
52						buffer, buflen, false))
53			return 0;
54	} else {
55		baseoff = protoff + sizeof(struct udphdr);
56		matchoff += dataoff - baseoff;
57
58		if (!nf_nat_mangle_udp_packet(skb, ct, ctinfo,
59					      protoff, matchoff, matchlen,
60					      buffer, buflen))
61			return 0;
62	}
63
64	/* Reload data pointer and adjust datalen value */
65	*dptr = skb->data + dataoff;
66	*datalen += buflen - matchlen;
67	return 1;
68}
69
70static int sip_sprintf_addr(const struct nf_conn *ct, char *buffer,
71			    const union nf_inet_addr *addr, bool delim)
72{
73	if (nf_ct_l3num(ct) == NFPROTO_IPV4)
74		return sprintf(buffer, "%pI4", &addr->ip);
75	else {
76		if (delim)
77			return sprintf(buffer, "[%pI6c]", &addr->ip6);
78		else
79			return sprintf(buffer, "%pI6c", &addr->ip6);
80	}
81}
82
83static int sip_sprintf_addr_port(const struct nf_conn *ct, char *buffer,
84				 const union nf_inet_addr *addr, u16 port)
85{
86	if (nf_ct_l3num(ct) == NFPROTO_IPV4)
87		return sprintf(buffer, "%pI4:%u", &addr->ip, port);
88	else
89		return sprintf(buffer, "[%pI6c]:%u", &addr->ip6, port);
90}
91
92static int map_addr(struct sk_buff *skb, unsigned int protoff,
93		    unsigned int dataoff,
94		    const char **dptr, unsigned int *datalen,
95		    unsigned int matchoff, unsigned int matchlen,
96		    union nf_inet_addr *addr, __be16 port)
97{
98	enum ip_conntrack_info ctinfo;
99	struct nf_conn *ct = nf_ct_get(skb, &ctinfo);
100	enum ip_conntrack_dir dir = CTINFO2DIR(ctinfo);
101	struct nf_ct_sip_master *ct_sip_info = nfct_help_data(ct);
102	char buffer[INET6_ADDRSTRLEN + sizeof("[]:nnnnn")];
103	unsigned int buflen;
104	union nf_inet_addr newaddr;
105	__be16 newport;
106
107	if (nf_inet_addr_cmp(&ct->tuplehash[dir].tuple.src.u3, addr) &&
108	    ct->tuplehash[dir].tuple.src.u.udp.port == port) {
109		newaddr = ct->tuplehash[!dir].tuple.dst.u3;
110		newport = ct->tuplehash[!dir].tuple.dst.u.udp.port;
111	} else if (nf_inet_addr_cmp(&ct->tuplehash[dir].tuple.dst.u3, addr) &&
112		   ct->tuplehash[dir].tuple.dst.u.udp.port == port) {
113		newaddr = ct->tuplehash[!dir].tuple.src.u3;
114		newport = ct_sip_info->forced_dport ? :
115			  ct->tuplehash[!dir].tuple.src.u.udp.port;
116	} else
117		return 1;
118
119	if (nf_inet_addr_cmp(&newaddr, addr) && newport == port)
120		return 1;
121
122	buflen = sip_sprintf_addr_port(ct, buffer, &newaddr, ntohs(newport));
123	return mangle_packet(skb, protoff, dataoff, dptr, datalen,
124			     matchoff, matchlen, buffer, buflen);
125}
126
127static int map_sip_addr(struct sk_buff *skb, unsigned int protoff,
128			unsigned int dataoff,
129			const char **dptr, unsigned int *datalen,
130			enum sip_header_types type)
131{
132	enum ip_conntrack_info ctinfo;
133	struct nf_conn *ct = nf_ct_get(skb, &ctinfo);
134	unsigned int matchlen, matchoff;
135	union nf_inet_addr addr;
136	__be16 port;
137
138	if (ct_sip_parse_header_uri(ct, *dptr, NULL, *datalen, type, NULL,
139				    &matchoff, &matchlen, &addr, &port) <= 0)
140		return 1;
141	return map_addr(skb, protoff, dataoff, dptr, datalen,
142			matchoff, matchlen, &addr, port);
143}
144
145static unsigned int nf_nat_sip(struct sk_buff *skb, unsigned int protoff,
146			       unsigned int dataoff,
147			       const char **dptr, unsigned int *datalen)
148{
149	enum ip_conntrack_info ctinfo;
150	struct nf_conn *ct = nf_ct_get(skb, &ctinfo);
151	enum ip_conntrack_dir dir = CTINFO2DIR(ctinfo);
152	struct nf_ct_sip_master *ct_sip_info = nfct_help_data(ct);
153	unsigned int coff, matchoff, matchlen;
154	enum sip_header_types hdr;
155	union nf_inet_addr addr;
156	__be16 port;
157	int request, in_header;
158
159	/* Basic rules: requests and responses. */
160	if (strncasecmp(*dptr, "SIP/2.0", strlen("SIP/2.0")) != 0) {
161		if (ct_sip_parse_request(ct, *dptr, *datalen,
162					 &matchoff, &matchlen,
163					 &addr, &port) > 0 &&
164		    !map_addr(skb, protoff, dataoff, dptr, datalen,
165			      matchoff, matchlen, &addr, port)) {
166			nf_ct_helper_log(skb, ct, "cannot mangle SIP message");
167			return NF_DROP;
168		}
169		request = 1;
170	} else
171		request = 0;
172
173	if (nf_ct_protonum(ct) == IPPROTO_TCP)
174		hdr = SIP_HDR_VIA_TCP;
175	else
176		hdr = SIP_HDR_VIA_UDP;
177
178	/* Translate topmost Via header and parameters */
179	if (ct_sip_parse_header_uri(ct, *dptr, NULL, *datalen,
180				    hdr, NULL, &matchoff, &matchlen,
181				    &addr, &port) > 0) {
182		unsigned int olen, matchend, poff, plen, buflen, n;
183		char buffer[INET6_ADDRSTRLEN + sizeof("[]:nnnnn")];
184
185		/* We're only interested in headers related to this
186		 * connection */
187		if (request) {
188			if (!nf_inet_addr_cmp(&addr,
189					&ct->tuplehash[dir].tuple.src.u3) ||
190			    port != ct->tuplehash[dir].tuple.src.u.udp.port)
191				goto next;
192		} else {
193			if (!nf_inet_addr_cmp(&addr,
194					&ct->tuplehash[dir].tuple.dst.u3) ||
195			    port != ct->tuplehash[dir].tuple.dst.u.udp.port)
196				goto next;
197		}
198
199		olen = *datalen;
200		if (!map_addr(skb, protoff, dataoff, dptr, datalen,
201			      matchoff, matchlen, &addr, port)) {
202			nf_ct_helper_log(skb, ct, "cannot mangle Via header");
203			return NF_DROP;
204		}
205
206		matchend = matchoff + matchlen + *datalen - olen;
207
208		/* The maddr= parameter (RFC 2361) specifies where to send
209		 * the reply. */
210		if (ct_sip_parse_address_param(ct, *dptr, matchend, *datalen,
211					       "maddr=", &poff, &plen,
212					       &addr, true) > 0 &&
213		    nf_inet_addr_cmp(&addr, &ct->tuplehash[dir].tuple.src.u3) &&
214		    !nf_inet_addr_cmp(&addr, &ct->tuplehash[!dir].tuple.dst.u3)) {
215			buflen = sip_sprintf_addr(ct, buffer,
216					&ct->tuplehash[!dir].tuple.dst.u3,
217					true);
218			if (!mangle_packet(skb, protoff, dataoff, dptr, datalen,
219					   poff, plen, buffer, buflen)) {
220				nf_ct_helper_log(skb, ct, "cannot mangle maddr");
221				return NF_DROP;
222			}
223		}
224
225		/* The received= parameter (RFC 2361) contains the address
226		 * from which the server received the request. */
227		if (ct_sip_parse_address_param(ct, *dptr, matchend, *datalen,
228					       "received=", &poff, &plen,
229					       &addr, false) > 0 &&
230		    nf_inet_addr_cmp(&addr, &ct->tuplehash[dir].tuple.dst.u3) &&
231		    !nf_inet_addr_cmp(&addr, &ct->tuplehash[!dir].tuple.src.u3)) {
232			buflen = sip_sprintf_addr(ct, buffer,
233					&ct->tuplehash[!dir].tuple.src.u3,
234					false);
235			if (!mangle_packet(skb, protoff, dataoff, dptr, datalen,
236					   poff, plen, buffer, buflen)) {
237				nf_ct_helper_log(skb, ct, "cannot mangle received");
238				return NF_DROP;
239			}
240		}
241
242		/* The rport= parameter (RFC 3581) contains the port number
243		 * from which the server received the request. */
244		if (ct_sip_parse_numerical_param(ct, *dptr, matchend, *datalen,
245						 "rport=", &poff, &plen,
246						 &n) > 0 &&
247		    htons(n) == ct->tuplehash[dir].tuple.dst.u.udp.port &&
248		    htons(n) != ct->tuplehash[!dir].tuple.src.u.udp.port) {
249			__be16 p = ct->tuplehash[!dir].tuple.src.u.udp.port;
250			buflen = sprintf(buffer, "%u", ntohs(p));
251			if (!mangle_packet(skb, protoff, dataoff, dptr, datalen,
252					   poff, plen, buffer, buflen)) {
253				nf_ct_helper_log(skb, ct, "cannot mangle rport");
254				return NF_DROP;
255			}
256		}
257	}
258
259next:
260	/* Translate Contact headers */
261	coff = 0;
262	in_header = 0;
263	while (ct_sip_parse_header_uri(ct, *dptr, &coff, *datalen,
264				       SIP_HDR_CONTACT, &in_header,
265				       &matchoff, &matchlen,
266				       &addr, &port) > 0) {
267		if (!map_addr(skb, protoff, dataoff, dptr, datalen,
268			      matchoff, matchlen,
269			      &addr, port)) {
270			nf_ct_helper_log(skb, ct, "cannot mangle contact");
271			return NF_DROP;
272		}
273	}
274
275	if (!map_sip_addr(skb, protoff, dataoff, dptr, datalen, SIP_HDR_FROM) ||
276	    !map_sip_addr(skb, protoff, dataoff, dptr, datalen, SIP_HDR_TO)) {
277		nf_ct_helper_log(skb, ct, "cannot mangle SIP from/to");
278		return NF_DROP;
279	}
280
281	/* Mangle destination port for Cisco phones, then fix up checksums */
282	if (dir == IP_CT_DIR_REPLY && ct_sip_info->forced_dport) {
283		struct udphdr *uh;
284
285		if (skb_ensure_writable(skb, skb->len)) {
286			nf_ct_helper_log(skb, ct, "cannot mangle packet");
287			return NF_DROP;
288		}
289
290		uh = (void *)skb->data + protoff;
291		uh->dest = ct_sip_info->forced_dport;
292
293		if (!nf_nat_mangle_udp_packet(skb, ct, ctinfo, protoff,
294					      0, 0, NULL, 0)) {
295			nf_ct_helper_log(skb, ct, "cannot mangle packet");
296			return NF_DROP;
297		}
298	}
299
300	return NF_ACCEPT;
301}
302
303static void nf_nat_sip_seq_adjust(struct sk_buff *skb, unsigned int protoff,
304				  s16 off)
305{
306	enum ip_conntrack_info ctinfo;
307	struct nf_conn *ct = nf_ct_get(skb, &ctinfo);
308	const struct tcphdr *th;
309
310	if (nf_ct_protonum(ct) != IPPROTO_TCP || off == 0)
311		return;
312
313	th = (struct tcphdr *)(skb->data + protoff);
314	nf_ct_seqadj_set(ct, ctinfo, th->seq, off);
315}
316
317/* Handles expected signalling connections and media streams */
318static void nf_nat_sip_expected(struct nf_conn *ct,
319				struct nf_conntrack_expect *exp)
320{
321	struct nf_conn_help *help = nfct_help(ct->master);
322	struct nf_conntrack_expect *pair_exp;
323	int range_set_for_snat = 0;
324	struct nf_nat_range2 range;
325
326	/* This must be a fresh one. */
327	BUG_ON(ct->status & IPS_NAT_DONE_MASK);
328
329	/* For DST manip, map port here to where it's expected. */
330	range.flags = (NF_NAT_RANGE_MAP_IPS | NF_NAT_RANGE_PROTO_SPECIFIED);
331	range.min_proto = range.max_proto = exp->saved_proto;
332	range.min_addr = range.max_addr = exp->saved_addr;
333	nf_nat_setup_info(ct, &range, NF_NAT_MANIP_DST);
334
335	/* Do media streams SRC manip according with the parameters
336	 * found in the paired expectation.
337	 */
338	if (exp->class != SIP_EXPECT_SIGNALLING) {
339		spin_lock_bh(&nf_conntrack_expect_lock);
340		hlist_for_each_entry(pair_exp, &help->expectations, lnode) {
341			if (pair_exp->tuple.src.l3num == nf_ct_l3num(ct) &&
342			    pair_exp->tuple.dst.protonum == ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.dst.protonum &&
343			    nf_inet_addr_cmp(&ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.src.u3, &pair_exp->saved_addr) &&
344			    ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.src.u.all == pair_exp->saved_proto.all) {
345				range.flags = (NF_NAT_RANGE_MAP_IPS | NF_NAT_RANGE_PROTO_SPECIFIED);
346				range.min_proto.all = range.max_proto.all = pair_exp->tuple.dst.u.all;
347				range.min_addr = range.max_addr = pair_exp->tuple.dst.u3;
348				range_set_for_snat = 1;
349				break;
350			}
351		}
352		spin_unlock_bh(&nf_conntrack_expect_lock);
353	}
354
355	/* When no paired expectation has been found, change src to
356	 * where master sends to, but only if the connection actually came
357	 * from the same source.
358	 */
359	if (!range_set_for_snat &&
360	    nf_inet_addr_cmp(&ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.src.u3,
361			     &ct->master->tuplehash[exp->dir].tuple.src.u3)) {
362		range.flags = NF_NAT_RANGE_MAP_IPS;
363		range.min_addr = range.max_addr
364			= ct->master->tuplehash[!exp->dir].tuple.dst.u3;
365		range_set_for_snat = 1;
366	}
367
368	/* Perform SRC manip. */
369	if (range_set_for_snat)
370		nf_nat_setup_info(ct, &range, NF_NAT_MANIP_SRC);
371}
372
373static unsigned int nf_nat_sip_expect(struct sk_buff *skb, unsigned int protoff,
374				      unsigned int dataoff,
375				      const char **dptr, unsigned int *datalen,
376				      struct nf_conntrack_expect *exp,
377				      unsigned int matchoff,
378				      unsigned int matchlen)
379{
380	enum ip_conntrack_info ctinfo;
381	struct nf_conn *ct = nf_ct_get(skb, &ctinfo);
382	enum ip_conntrack_dir dir = CTINFO2DIR(ctinfo);
383	struct nf_ct_sip_master *ct_sip_info = nfct_help_data(ct);
384	union nf_inet_addr newaddr;
385	u_int16_t port;
386	__be16 srcport;
387	char buffer[INET6_ADDRSTRLEN + sizeof("[]:nnnnn")];
388	unsigned int buflen;
389
390	/* Connection will come from reply */
391	if (nf_inet_addr_cmp(&ct->tuplehash[dir].tuple.src.u3,
392			     &ct->tuplehash[!dir].tuple.dst.u3))
393		newaddr = exp->tuple.dst.u3;
394	else
395		newaddr = ct->tuplehash[!dir].tuple.dst.u3;
396
397	/* If the signalling port matches the connection's source port in the
398	 * original direction, try to use the destination port in the opposite
399	 * direction. */
400	srcport = ct_sip_info->forced_dport ? :
401		  ct->tuplehash[dir].tuple.src.u.udp.port;
402	if (exp->tuple.dst.u.udp.port == srcport)
403		port = ntohs(ct->tuplehash[!dir].tuple.dst.u.udp.port);
404	else
405		port = ntohs(exp->tuple.dst.u.udp.port);
406
407	exp->saved_addr = exp->tuple.dst.u3;
408	exp->tuple.dst.u3 = newaddr;
409	exp->saved_proto.udp.port = exp->tuple.dst.u.udp.port;
410	exp->dir = !dir;
411	exp->expectfn = nf_nat_sip_expected;
412
413	port = nf_nat_exp_find_port(exp, port);
414	if (port == 0) {
415		nf_ct_helper_log(skb, ct, "all ports in use for SIP");
416		return NF_DROP;
417	}
418
419	if (!nf_inet_addr_cmp(&exp->tuple.dst.u3, &exp->saved_addr) ||
420	    exp->tuple.dst.u.udp.port != exp->saved_proto.udp.port) {
421		buflen = sip_sprintf_addr_port(ct, buffer, &newaddr, port);
422		if (!mangle_packet(skb, protoff, dataoff, dptr, datalen,
423				   matchoff, matchlen, buffer, buflen)) {
424			nf_ct_helper_log(skb, ct, "cannot mangle packet");
425			goto err;
426		}
427	}
428	return NF_ACCEPT;
429
430err:
431	nf_ct_unexpect_related(exp);
432	return NF_DROP;
433}
434
435static int mangle_content_len(struct sk_buff *skb, unsigned int protoff,
436			      unsigned int dataoff,
437			      const char **dptr, unsigned int *datalen)
438{
439	enum ip_conntrack_info ctinfo;
440	struct nf_conn *ct = nf_ct_get(skb, &ctinfo);
441	unsigned int matchoff, matchlen;
442	char buffer[sizeof("65536")];
443	int buflen, c_len;
444
445	/* Get actual SDP length */
446	if (ct_sip_get_sdp_header(ct, *dptr, 0, *datalen,
447				  SDP_HDR_VERSION, SDP_HDR_UNSPEC,
448				  &matchoff, &matchlen) <= 0)
449		return 0;
450	c_len = *datalen - matchoff + strlen("v=");
451
452	/* Now, update SDP length */
453	if (ct_sip_get_header(ct, *dptr, 0, *datalen, SIP_HDR_CONTENT_LENGTH,
454			      &matchoff, &matchlen) <= 0)
455		return 0;
456
457	buflen = sprintf(buffer, "%u", c_len);
458	return mangle_packet(skb, protoff, dataoff, dptr, datalen,
459			     matchoff, matchlen, buffer, buflen);
460}
461
462static int mangle_sdp_packet(struct sk_buff *skb, unsigned int protoff,
463			     unsigned int dataoff,
464			     const char **dptr, unsigned int *datalen,
465			     unsigned int sdpoff,
466			     enum sdp_header_types type,
467			     enum sdp_header_types term,
468			     char *buffer, int buflen)
469{
470	enum ip_conntrack_info ctinfo;
471	struct nf_conn *ct = nf_ct_get(skb, &ctinfo);
472	unsigned int matchlen, matchoff;
473
474	if (ct_sip_get_sdp_header(ct, *dptr, sdpoff, *datalen, type, term,
475				  &matchoff, &matchlen) <= 0)
476		return -ENOENT;
477	return mangle_packet(skb, protoff, dataoff, dptr, datalen,
478			     matchoff, matchlen, buffer, buflen) ? 0 : -EINVAL;
479}
480
481static unsigned int nf_nat_sdp_addr(struct sk_buff *skb, unsigned int protoff,
482				    unsigned int dataoff,
483				    const char **dptr, unsigned int *datalen,
484				    unsigned int sdpoff,
485				    enum sdp_header_types type,
486				    enum sdp_header_types term,
487				    const union nf_inet_addr *addr)
488{
489	enum ip_conntrack_info ctinfo;
490	struct nf_conn *ct = nf_ct_get(skb, &ctinfo);
491	char buffer[INET6_ADDRSTRLEN];
492	unsigned int buflen;
493
494	buflen = sip_sprintf_addr(ct, buffer, addr, false);
495	if (mangle_sdp_packet(skb, protoff, dataoff, dptr, datalen,
496			      sdpoff, type, term, buffer, buflen))
497		return 0;
498
499	return mangle_content_len(skb, protoff, dataoff, dptr, datalen);
500}
501
502static unsigned int nf_nat_sdp_port(struct sk_buff *skb, unsigned int protoff,
503				    unsigned int dataoff,
504				    const char **dptr, unsigned int *datalen,
505				    unsigned int matchoff,
506				    unsigned int matchlen,
507				    u_int16_t port)
508{
509	char buffer[sizeof("nnnnn")];
510	unsigned int buflen;
511
512	buflen = sprintf(buffer, "%u", port);
513	if (!mangle_packet(skb, protoff, dataoff, dptr, datalen,
514			   matchoff, matchlen, buffer, buflen))
515		return 0;
516
517	return mangle_content_len(skb, protoff, dataoff, dptr, datalen);
518}
519
520static unsigned int nf_nat_sdp_session(struct sk_buff *skb, unsigned int protoff,
521				       unsigned int dataoff,
522				       const char **dptr, unsigned int *datalen,
523				       unsigned int sdpoff,
524				       const union nf_inet_addr *addr)
525{
526	enum ip_conntrack_info ctinfo;
527	struct nf_conn *ct = nf_ct_get(skb, &ctinfo);
528	char buffer[INET6_ADDRSTRLEN];
529	unsigned int buflen;
530
531	/* Mangle session description owner and contact addresses */
532	buflen = sip_sprintf_addr(ct, buffer, addr, false);
533	if (mangle_sdp_packet(skb, protoff, dataoff, dptr, datalen, sdpoff,
534			      SDP_HDR_OWNER, SDP_HDR_MEDIA, buffer, buflen))
535		return 0;
536
537	switch (mangle_sdp_packet(skb, protoff, dataoff, dptr, datalen, sdpoff,
538				  SDP_HDR_CONNECTION, SDP_HDR_MEDIA,
539				  buffer, buflen)) {
540	case 0:
541	/*
542	 * RFC 2327:
543	 *
544	 * Session description
545	 *
546	 * c=* (connection information - not required if included in all media)
547	 */
548	case -ENOENT:
549		break;
550	default:
551		return 0;
552	}
553
554	return mangle_content_len(skb, protoff, dataoff, dptr, datalen);
555}
556
557/* So, this packet has hit the connection tracking matching code.
558   Mangle it, and change the expectation to match the new version. */
559static unsigned int nf_nat_sdp_media(struct sk_buff *skb, unsigned int protoff,
560				     unsigned int dataoff,
561				     const char **dptr, unsigned int *datalen,
562				     struct nf_conntrack_expect *rtp_exp,
563				     struct nf_conntrack_expect *rtcp_exp,
564				     unsigned int mediaoff,
565				     unsigned int medialen,
566				     union nf_inet_addr *rtp_addr)
567{
568	enum ip_conntrack_info ctinfo;
569	struct nf_conn *ct = nf_ct_get(skb, &ctinfo);
570	enum ip_conntrack_dir dir = CTINFO2DIR(ctinfo);
571	u_int16_t port;
572
573	/* Connection will come from reply */
574	if (nf_inet_addr_cmp(&ct->tuplehash[dir].tuple.src.u3,
575			     &ct->tuplehash[!dir].tuple.dst.u3))
576		*rtp_addr = rtp_exp->tuple.dst.u3;
577	else
578		*rtp_addr = ct->tuplehash[!dir].tuple.dst.u3;
579
580	rtp_exp->saved_addr = rtp_exp->tuple.dst.u3;
581	rtp_exp->tuple.dst.u3 = *rtp_addr;
582	rtp_exp->saved_proto.udp.port = rtp_exp->tuple.dst.u.udp.port;
583	rtp_exp->dir = !dir;
584	rtp_exp->expectfn = nf_nat_sip_expected;
585
586	rtcp_exp->saved_addr = rtcp_exp->tuple.dst.u3;
587	rtcp_exp->tuple.dst.u3 = *rtp_addr;
588	rtcp_exp->saved_proto.udp.port = rtcp_exp->tuple.dst.u.udp.port;
589	rtcp_exp->dir = !dir;
590	rtcp_exp->expectfn = nf_nat_sip_expected;
591
592	/* Try to get same pair of ports: if not, try to change them. */
593	for (port = ntohs(rtp_exp->tuple.dst.u.udp.port);
594	     port != 0; port += 2) {
595		int ret;
596
597		rtp_exp->tuple.dst.u.udp.port = htons(port);
598		ret = nf_ct_expect_related(rtp_exp,
599					   NF_CT_EXP_F_SKIP_MASTER);
600		if (ret == -EBUSY)
601			continue;
602		else if (ret < 0) {
603			port = 0;
604			break;
605		}
606		rtcp_exp->tuple.dst.u.udp.port = htons(port + 1);
607		ret = nf_ct_expect_related(rtcp_exp,
608					   NF_CT_EXP_F_SKIP_MASTER);
609		if (ret == 0)
610			break;
611		else if (ret == -EBUSY) {
612			nf_ct_unexpect_related(rtp_exp);
613			continue;
614		} else if (ret < 0) {
615			nf_ct_unexpect_related(rtp_exp);
616			port = 0;
617			break;
618		}
619	}
620
621	if (port == 0) {
622		nf_ct_helper_log(skb, ct, "all ports in use for SDP media");
623		goto err1;
624	}
625
626	/* Update media port. */
627	if (rtp_exp->tuple.dst.u.udp.port != rtp_exp->saved_proto.udp.port &&
628	    !nf_nat_sdp_port(skb, protoff, dataoff, dptr, datalen,
629			     mediaoff, medialen, port)) {
630		nf_ct_helper_log(skb, ct, "cannot mangle SDP message");
631		goto err2;
632	}
633
634	return NF_ACCEPT;
635
636err2:
637	nf_ct_unexpect_related(rtp_exp);
638	nf_ct_unexpect_related(rtcp_exp);
639err1:
640	return NF_DROP;
641}
642
643static struct nf_ct_helper_expectfn sip_nat = {
644	.name		= "sip",
645	.expectfn	= nf_nat_sip_expected,
646};
647
648static void __exit nf_nat_sip_fini(void)
649{
650	nf_nat_helper_unregister(&nat_helper_sip);
651	RCU_INIT_POINTER(nf_nat_sip_hooks, NULL);
652	nf_ct_helper_expectfn_unregister(&sip_nat);
653	synchronize_rcu();
654}
655
656static const struct nf_nat_sip_hooks sip_hooks = {
657	.msg		= nf_nat_sip,
658	.seq_adjust	= nf_nat_sip_seq_adjust,
659	.expect		= nf_nat_sip_expect,
660	.sdp_addr	= nf_nat_sdp_addr,
661	.sdp_port	= nf_nat_sdp_port,
662	.sdp_session	= nf_nat_sdp_session,
663	.sdp_media	= nf_nat_sdp_media,
664};
665
666static int __init nf_nat_sip_init(void)
667{
668	BUG_ON(nf_nat_sip_hooks != NULL);
669	nf_nat_helper_register(&nat_helper_sip);
670	RCU_INIT_POINTER(nf_nat_sip_hooks, &sip_hooks);
671	nf_ct_helper_expectfn_register(&sip_nat);
672	return 0;
673}
674
675module_init(nf_nat_sip_init);
676module_exit(nf_nat_sip_fini);
677