1 /* (C) 1999-2001 Paul `Rusty' Russell
2 * (C) 2002-2004 Netfilter Core Team <coreteam@netfilter.org>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License version 2 as
6 * published by the Free Software Foundation.
9 #include <linux/types.h>
10 #include <linux/init.h>
11 #include <linux/netfilter.h>
13 #include <linux/udp.h>
16 #include <linux/netfilter_ipv4/ip_nat.h>
17 #include <linux/netfilter_ipv4/ip_nat_core.h>
18 #include <linux/netfilter_ipv4/ip_nat_rule.h>
19 #include <linux/netfilter_ipv4/ip_nat_protocol.h>
22 udp_in_range(const struct ip_conntrack_tuple
*tuple
,
23 enum ip_nat_manip_type maniptype
,
24 const union ip_conntrack_manip_proto
*min
,
25 const union ip_conntrack_manip_proto
*max
)
29 if (maniptype
== IP_NAT_MANIP_SRC
)
30 port
= tuple
->src
.u
.udp
.port
;
32 port
= tuple
->dst
.u
.udp
.port
;
34 return ntohs(port
) >= ntohs(min
->udp
.port
)
35 && ntohs(port
) <= ntohs(max
->udp
.port
);
39 udp_unique_tuple(struct ip_conntrack_tuple
*tuple
,
40 const struct ip_nat_range
*range
,
41 enum ip_nat_manip_type maniptype
,
42 const struct ip_conntrack
*conntrack
)
44 static u_int16_t port
;
46 unsigned int range_size
, min
, i
;
48 if (maniptype
== IP_NAT_MANIP_SRC
)
49 portptr
= &tuple
->src
.u
.udp
.port
;
51 portptr
= &tuple
->dst
.u
.udp
.port
;
53 /* If no range specified... */
54 if (!(range
->flags
& IP_NAT_RANGE_PROTO_SPECIFIED
)) {
55 /* If it's dst rewrite, can't change port */
56 if (maniptype
== IP_NAT_MANIP_DST
)
59 if (ntohs(*portptr
) < 1024) {
60 /* Loose convention: >> 512 is credential passing */
61 if (ntohs(*portptr
)<512) {
63 range_size
= 511 - min
+ 1;
66 range_size
= 1023 - min
+ 1;
70 range_size
= 65535 - 1024 + 1;
73 min
= ntohs(range
->min
.udp
.port
);
74 range_size
= ntohs(range
->max
.udp
.port
) - min
+ 1;
77 for (i
= 0; i
< range_size
; i
++, port
++) {
78 *portptr
= htons(min
+ port
% range_size
);
79 if (!ip_nat_used_tuple(tuple
, conntrack
))
86 udp_manip_pkt(struct sk_buff
**pskb
,
87 unsigned int iphdroff
,
88 const struct ip_conntrack_tuple
*tuple
,
89 enum ip_nat_manip_type maniptype
)
91 struct iphdr
*iph
= (struct iphdr
*)((*pskb
)->data
+ iphdroff
);
93 unsigned int hdroff
= iphdroff
+ iph
->ihl
*4;
95 __be16
*portptr
, newport
;
97 if (!skb_make_writable(pskb
, hdroff
+ sizeof(*hdr
)))
100 iph
= (struct iphdr
*)((*pskb
)->data
+ iphdroff
);
101 hdr
= (struct udphdr
*)((*pskb
)->data
+ hdroff
);
103 if (maniptype
== IP_NAT_MANIP_SRC
) {
104 /* Get rid of src ip and src pt */
106 newip
= tuple
->src
.ip
;
107 newport
= tuple
->src
.u
.udp
.port
;
108 portptr
= &hdr
->source
;
110 /* Get rid of dst ip and dst pt */
112 newip
= tuple
->dst
.ip
;
113 newport
= tuple
->dst
.u
.udp
.port
;
114 portptr
= &hdr
->dest
;
117 if (hdr
->check
|| (*pskb
)->ip_summed
== CHECKSUM_PARTIAL
) {
118 nf_proto_csum_replace4(&hdr
->check
, *pskb
, oldip
, newip
, 1);
119 nf_proto_csum_replace2(&hdr
->check
, *pskb
, *portptr
, newport
, 0);
121 hdr
->check
= CSUM_MANGLED_0
;
127 struct ip_nat_protocol ip_nat_protocol_udp
= {
129 .protonum
= IPPROTO_UDP
,
131 .manip_pkt
= udp_manip_pkt
,
132 .in_range
= udp_in_range
,
133 .unique_tuple
= udp_unique_tuple
,
134 #if defined(CONFIG_IP_NF_CONNTRACK_NETLINK) || \
135 defined(CONFIG_IP_NF_CONNTRACK_NETLINK_MODULE)
136 .range_to_nfattr
= ip_nat_port_range_to_nfattr
,
137 .nfattr_to_range
= ip_nat_port_nfattr_to_range
,