1f4886857b444f7927a2fca9ec6f42ccec08a1fc
[firefly-linux-kernel-4.4.55.git] / net / ipv6 / ping.c
1 /*
2  * INET         An implementation of the TCP/IP protocol suite for the LINUX
3  *              operating system.  INET is implemented using the  BSD Socket
4  *              interface as the means of communication with the user level.
5  *
6  *              "Ping" sockets
7  *
8  *              This program is free software; you can redistribute it and/or
9  *              modify it under the terms of the GNU General Public License
10  *              as published by the Free Software Foundation; either version
11  *              2 of the License, or (at your option) any later version.
12  *
13  * Based on ipv4/ping.c code.
14  *
15  * Authors:     Lorenzo Colitti (IPv6 support)
16  *              Vasiliy Kulikov / Openwall (IPv4 implementation, for Linux 2.6),
17  *              Pavel Kankovsky (IPv4 implementation, for Linux 2.4.32)
18  *
19  */
20
21 #include <net/addrconf.h>
22 #include <net/ipv6.h>
23 #include <net/ip6_route.h>
24 #include <net/protocol.h>
25 #include <net/udp.h>
26 #include <net/transp_v6.h>
27 #include <net/ping.h>
28
29 struct proto pingv6_prot = {
30         .name =         "PINGv6",
31         .owner =        THIS_MODULE,
32         .init =         ping_init_sock,
33         .close =        ping_close,
34         .connect =      ip6_datagram_connect,
35         .disconnect =   udp_disconnect,
36         .setsockopt =   ipv6_setsockopt,
37         .getsockopt =   ipv6_getsockopt,
38         .sendmsg =      ping_v6_sendmsg,
39         .recvmsg =      ping_recvmsg,
40         .bind =         ping_bind,
41         .backlog_rcv =  ping_queue_rcv_skb,
42         .hash =         ping_hash,
43         .unhash =       ping_unhash,
44         .get_port =     ping_get_port,
45         .obj_size =     sizeof(struct raw6_sock),
46 };
47 EXPORT_SYMBOL_GPL(pingv6_prot);
48
49 static struct inet_protosw pingv6_protosw = {
50         .type =      SOCK_DGRAM,
51         .protocol =  IPPROTO_ICMPV6,
52         .prot =      &pingv6_prot,
53         .ops =       &inet6_dgram_ops,
54         .no_check =  UDP_CSUM_DEFAULT,
55         .flags =     INET_PROTOSW_REUSE,
56 };
57
58
59 /* Compatibility glue so we can support IPv6 when it's compiled as a module */
60 int dummy_ipv6_recv_error(struct sock *sk, struct msghdr *msg, int len)
61 {
62         return -EAFNOSUPPORT;
63 }
64 int dummy_ip6_datagram_recv_ctl(struct sock *sk, struct msghdr *msg,
65                                  struct sk_buff *skb)
66 {
67         return -EAFNOSUPPORT;
68 }
69 int dummy_icmpv6_err_convert(u8 type, u8 code, int *err)
70 {
71         return -EAFNOSUPPORT;
72 }
73 void dummy_ipv6_icmp_error(struct sock *sk, struct sk_buff *skb, int err,
74                             __be16 port, u32 info, u8 *payload) {}
75 int dummy_ipv6_chk_addr(struct net *net, const struct in6_addr *addr,
76                         struct net_device *dev, int strict)
77 {
78         return 0;
79 }
80
81 int __init pingv6_init(void)
82 {
83         pingv6_ops.ipv6_recv_error = ipv6_recv_error;
84         pingv6_ops.ip6_datagram_recv_ctl = ip6_datagram_recv_ctl;
85         pingv6_ops.icmpv6_err_convert = icmpv6_err_convert;
86         pingv6_ops.ipv6_icmp_error = ipv6_icmp_error;
87         pingv6_ops.ipv6_chk_addr = ipv6_chk_addr;
88         return inet6_register_protosw(&pingv6_protosw);
89 }
90
91 /* This never gets called because it's not possible to unload the ipv6 module,
92  * but just in case.
93  */
94 void pingv6_exit(void)
95 {
96         pingv6_ops.ipv6_recv_error = dummy_ipv6_recv_error;
97         pingv6_ops.ip6_datagram_recv_ctl = dummy_ip6_datagram_recv_ctl;
98         pingv6_ops.icmpv6_err_convert = dummy_icmpv6_err_convert;
99         pingv6_ops.ipv6_icmp_error = dummy_ipv6_icmp_error;
100         pingv6_ops.ipv6_chk_addr = dummy_ipv6_chk_addr;
101         inet6_unregister_protosw(&pingv6_protosw);
102 }
103
104 int ping_v6_sendmsg(struct kiocb *iocb, struct sock *sk, struct msghdr *msg,
105                     size_t len)
106 {
107         struct inet_sock *inet = inet_sk(sk);
108         struct ipv6_pinfo *np = inet6_sk(sk);
109         struct icmp6hdr user_icmph;
110         int addr_type;
111         struct in6_addr *daddr;
112         int iif = 0;
113         struct flowi6 fl6;
114         int err;
115         int hlimit;
116         struct dst_entry *dst;
117         struct rt6_info *rt;
118         struct pingfakehdr pfh;
119
120         pr_debug("ping_v6_sendmsg(sk=%p,sk->num=%u)\n", inet, inet->inet_num);
121
122         err = ping_common_sendmsg(AF_INET6, msg, len, &user_icmph,
123                                   sizeof(user_icmph));
124         if (err)
125                 return err;
126
127         if (msg->msg_name) {
128                 struct sockaddr_in6 *u = (struct sockaddr_in6 *) msg->msg_name;
129                 if (msg->msg_namelen < sizeof(struct sockaddr_in6) ||
130                     u->sin6_family != AF_INET6) {
131                         return -EINVAL;
132                 }
133                 if (sk->sk_bound_dev_if &&
134                     sk->sk_bound_dev_if != u->sin6_scope_id) {
135                         return -EINVAL;
136                 }
137                 daddr = &(u->sin6_addr);
138                 iif = u->sin6_scope_id;
139         } else {
140                 if (sk->sk_state != TCP_ESTABLISHED)
141                         return -EDESTADDRREQ;
142                 daddr = &np->daddr;
143         }
144
145         if (!iif)
146                 iif = sk->sk_bound_dev_if;
147
148         addr_type = ipv6_addr_type(daddr);
149         if (__ipv6_addr_needs_scope_id(addr_type) && !iif)
150                 return -EINVAL;
151         if (addr_type & IPV6_ADDR_MAPPED)
152                 return -EINVAL;
153
154         /* TODO: use ip6_datagram_send_ctl to get options from cmsg */
155
156         memset(&fl6, 0, sizeof(fl6));
157
158         fl6.flowi6_proto = IPPROTO_ICMPV6;
159         fl6.saddr = np->saddr;
160         fl6.daddr = *daddr;
161         fl6.fl6_icmp_type = user_icmph.icmp6_type;
162         fl6.fl6_icmp_code = user_icmph.icmp6_code;
163         security_sk_classify_flow(sk, flowi6_to_flowi(&fl6));
164
165         if (!fl6.flowi6_oif && ipv6_addr_is_multicast(&fl6.daddr))
166                 fl6.flowi6_oif = np->mcast_oif;
167         else if (!fl6.flowi6_oif)
168                 fl6.flowi6_oif = np->ucast_oif;
169
170         dst = ip6_sk_dst_lookup_flow(sk, &fl6,  daddr, 1);
171         if (IS_ERR(dst))
172                 return PTR_ERR(dst);
173         rt = (struct rt6_info *) dst;
174
175         np = inet6_sk(sk);
176         if (!np)
177                 return -EBADF;
178
179         if (!fl6.flowi6_oif && ipv6_addr_is_multicast(&fl6.daddr))
180                 fl6.flowi6_oif = np->mcast_oif;
181         else if (!fl6.flowi6_oif)
182                 fl6.flowi6_oif = np->ucast_oif;
183
184         pfh.icmph.type = user_icmph.icmp6_type;
185         pfh.icmph.code = user_icmph.icmp6_code;
186         pfh.icmph.checksum = 0;
187         pfh.icmph.un.echo.id = inet->inet_sport;
188         pfh.icmph.un.echo.sequence = user_icmph.icmp6_sequence;
189         pfh.iov = msg->msg_iov;
190         pfh.wcheck = 0;
191         pfh.family = AF_INET6;
192
193         if (ipv6_addr_is_multicast(&fl6.daddr))
194                 hlimit = np->mcast_hops;
195         else
196                 hlimit = np->hop_limit;
197         if (hlimit < 0)
198                 hlimit = ip6_dst_hoplimit(dst);
199
200         err = ip6_append_data(sk, ping_getfrag, &pfh, len,
201                               0, hlimit,
202                               np->tclass, NULL, &fl6, rt,
203                               MSG_DONTWAIT, np->dontfrag);
204
205         if (err) {
206                 ICMP6_INC_STATS_BH(sock_net(sk), rt->rt6i_idev,
207                                    ICMP6_MIB_OUTERRORS);
208                 ip6_flush_pending_frames(sk);
209         } else {
210                 err = icmpv6_push_pending_frames(sk, &fl6,
211                                                  (struct icmp6hdr *) &pfh.icmph,
212                                                  len);
213         }
214
215         if (err)
216                 return err;
217
218         return len;
219 }