bridge: simplify ip_mc_check_igmp() and ipv6_mc_check_mld() calls
[linux-2.6-microblaze.git] / net / ipv6 / mcast_snoop.c
1 /* Copyright (C) 2010: YOSHIFUJI Hideaki <yoshfuji@linux-ipv6.org>
2  * Copyright (C) 2015: Linus Lüssing <linus.luessing@c0d3.blue>
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of version 2 of the GNU General Public
6  * License as published by the Free Software Foundation.
7  *
8  * This program is distributed in the hope that it will be useful, but
9  * WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11  * General Public License for more details.
12  *
13  * You should have received a copy of the GNU General Public License
14  * along with this program; if not, see <http://www.gnu.org/licenses/>.
15  *
16  *
17  * Based on the MLD support added to br_multicast.c by YOSHIFUJI Hideaki.
18  */
19
20 #include <linux/skbuff.h>
21 #include <net/ipv6.h>
22 #include <net/mld.h>
23 #include <net/addrconf.h>
24 #include <net/ip6_checksum.h>
25
26 static int ipv6_mc_check_ip6hdr(struct sk_buff *skb)
27 {
28         const struct ipv6hdr *ip6h;
29         unsigned int len;
30         unsigned int offset = skb_network_offset(skb) + sizeof(*ip6h);
31
32         if (!pskb_may_pull(skb, offset))
33                 return -EINVAL;
34
35         ip6h = ipv6_hdr(skb);
36
37         if (ip6h->version != 6)
38                 return -EINVAL;
39
40         len = offset + ntohs(ip6h->payload_len);
41         if (skb->len < len || len <= offset)
42                 return -EINVAL;
43
44         return 0;
45 }
46
47 static int ipv6_mc_check_exthdrs(struct sk_buff *skb)
48 {
49         const struct ipv6hdr *ip6h;
50         int offset;
51         u8 nexthdr;
52         __be16 frag_off;
53
54         ip6h = ipv6_hdr(skb);
55
56         if (ip6h->nexthdr != IPPROTO_HOPOPTS)
57                 return -ENOMSG;
58
59         nexthdr = ip6h->nexthdr;
60         offset = skb_network_offset(skb) + sizeof(*ip6h);
61         offset = ipv6_skip_exthdr(skb, offset, &nexthdr, &frag_off);
62
63         if (offset < 0)
64                 return -EINVAL;
65
66         if (nexthdr != IPPROTO_ICMPV6)
67                 return -ENOMSG;
68
69         skb_set_transport_header(skb, offset);
70
71         return 0;
72 }
73
74 static int ipv6_mc_check_mld_reportv2(struct sk_buff *skb)
75 {
76         unsigned int len = skb_transport_offset(skb);
77
78         len += sizeof(struct mld2_report);
79
80         return pskb_may_pull(skb, len) ? 0 : -EINVAL;
81 }
82
83 static int ipv6_mc_check_mld_query(struct sk_buff *skb)
84 {
85         struct mld_msg *mld;
86         unsigned int len = skb_transport_offset(skb);
87
88         /* RFC2710+RFC3810 (MLDv1+MLDv2) require link-local source addresses */
89         if (!(ipv6_addr_type(&ipv6_hdr(skb)->saddr) & IPV6_ADDR_LINKLOCAL))
90                 return -EINVAL;
91
92         len += sizeof(struct mld_msg);
93         if (skb->len < len)
94                 return -EINVAL;
95
96         /* MLDv1? */
97         if (skb->len != len) {
98                 /* or MLDv2? */
99                 len += sizeof(struct mld2_query) - sizeof(struct mld_msg);
100                 if (skb->len < len || !pskb_may_pull(skb, len))
101                         return -EINVAL;
102         }
103
104         mld = (struct mld_msg *)skb_transport_header(skb);
105
106         /* RFC2710+RFC3810 (MLDv1+MLDv2) require the multicast link layer
107          * all-nodes destination address (ff02::1) for general queries
108          */
109         if (ipv6_addr_any(&mld->mld_mca) &&
110             !ipv6_addr_is_ll_all_nodes(&ipv6_hdr(skb)->daddr))
111                 return -EINVAL;
112
113         return 0;
114 }
115
116 static int ipv6_mc_check_mld_msg(struct sk_buff *skb)
117 {
118         struct mld_msg *mld = (struct mld_msg *)skb_transport_header(skb);
119
120         switch (mld->mld_type) {
121         case ICMPV6_MGM_REDUCTION:
122         case ICMPV6_MGM_REPORT:
123                 /* fall through */
124                 return 0;
125         case ICMPV6_MLD2_REPORT:
126                 return ipv6_mc_check_mld_reportv2(skb);
127         case ICMPV6_MGM_QUERY:
128                 return ipv6_mc_check_mld_query(skb);
129         default:
130                 return -ENOMSG;
131         }
132 }
133
134 static inline __sum16 ipv6_mc_validate_checksum(struct sk_buff *skb)
135 {
136         return skb_checksum_validate(skb, IPPROTO_ICMPV6, ip6_compute_pseudo);
137 }
138
139 static int __ipv6_mc_check_mld(struct sk_buff *skb)
140
141 {
142         struct sk_buff *skb_chk = NULL;
143         unsigned int transport_len;
144         unsigned int len = skb_transport_offset(skb) + sizeof(struct mld_msg);
145         int ret = -EINVAL;
146
147         transport_len = ntohs(ipv6_hdr(skb)->payload_len);
148         transport_len -= skb_transport_offset(skb) - sizeof(struct ipv6hdr);
149
150         skb_chk = skb_checksum_trimmed(skb, transport_len,
151                                        ipv6_mc_validate_checksum);
152         if (!skb_chk)
153                 goto err;
154
155         if (!pskb_may_pull(skb_chk, len))
156                 goto err;
157
158         ret = ipv6_mc_check_mld_msg(skb_chk);
159         if (ret)
160                 goto err;
161
162         ret = 0;
163
164 err:
165         if (skb_chk && skb_chk != skb)
166                 kfree_skb(skb_chk);
167
168         return ret;
169 }
170
171 /**
172  * ipv6_mc_check_mld - checks whether this is a sane MLD packet
173  * @skb: the skb to validate
174  *
175  * Checks whether an IPv6 packet is a valid MLD packet. If so sets
176  * skb transport header accordingly and returns zero.
177  *
178  * -EINVAL: A broken packet was detected, i.e. it violates some internet
179  *  standard
180  * -ENOMSG: IP header validation succeeded but it is not an MLD packet.
181  * -ENOMEM: A memory allocation failure happened.
182  *
183  * Caller needs to set the skb network header and free any returned skb if it
184  * differs from the provided skb.
185  */
186 int ipv6_mc_check_mld(struct sk_buff *skb)
187 {
188         int ret;
189
190         ret = ipv6_mc_check_ip6hdr(skb);
191         if (ret < 0)
192                 return ret;
193
194         ret = ipv6_mc_check_exthdrs(skb);
195         if (ret < 0)
196                 return ret;
197
198         return __ipv6_mc_check_mld(skb);
199 }
200 EXPORT_SYMBOL(ipv6_mc_check_mld);