[TCP]: Merge exit paths in tcp_v4_conn_request.
[linux-2.6] / net / ipv4 / raw.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  *              RAW - implementation of IP "raw" sockets.
7  *
8  * Version:     $Id: raw.c,v 1.64 2002/02/01 22:01:04 davem Exp $
9  *
10  * Authors:     Ross Biro
11  *              Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG>
12  *
13  * Fixes:
14  *              Alan Cox        :       verify_area() fixed up
15  *              Alan Cox        :       ICMP error handling
16  *              Alan Cox        :       EMSGSIZE if you send too big a packet
17  *              Alan Cox        :       Now uses generic datagrams and shared
18  *                                      skbuff library. No more peek crashes,
19  *                                      no more backlogs
20  *              Alan Cox        :       Checks sk->broadcast.
21  *              Alan Cox        :       Uses skb_free_datagram/skb_copy_datagram
22  *              Alan Cox        :       Raw passes ip options too
23  *              Alan Cox        :       Setsocketopt added
24  *              Alan Cox        :       Fixed error return for broadcasts
25  *              Alan Cox        :       Removed wake_up calls
26  *              Alan Cox        :       Use ttl/tos
27  *              Alan Cox        :       Cleaned up old debugging
28  *              Alan Cox        :       Use new kernel side addresses
29  *      Arnt Gulbrandsen        :       Fixed MSG_DONTROUTE in raw sockets.
30  *              Alan Cox        :       BSD style RAW socket demultiplexing.
31  *              Alan Cox        :       Beginnings of mrouted support.
32  *              Alan Cox        :       Added IP_HDRINCL option.
33  *              Alan Cox        :       Skip broadcast check if BSDism set.
34  *              David S. Miller :       New socket lookup architecture.
35  *
36  *              This program is free software; you can redistribute it and/or
37  *              modify it under the terms of the GNU General Public License
38  *              as published by the Free Software Foundation; either version
39  *              2 of the License, or (at your option) any later version.
40  */
41
42 #include <linux/types.h>
43 #include <asm/atomic.h>
44 #include <asm/byteorder.h>
45 #include <asm/current.h>
46 #include <asm/uaccess.h>
47 #include <asm/ioctls.h>
48 #include <linux/stddef.h>
49 #include <linux/slab.h>
50 #include <linux/errno.h>
51 #include <linux/aio.h>
52 #include <linux/kernel.h>
53 #include <linux/spinlock.h>
54 #include <linux/sockios.h>
55 #include <linux/socket.h>
56 #include <linux/in.h>
57 #include <linux/mroute.h>
58 #include <linux/netdevice.h>
59 #include <linux/in_route.h>
60 #include <linux/route.h>
61 #include <linux/skbuff.h>
62 #include <net/net_namespace.h>
63 #include <net/dst.h>
64 #include <net/sock.h>
65 #include <linux/gfp.h>
66 #include <linux/ip.h>
67 #include <linux/net.h>
68 #include <net/ip.h>
69 #include <net/icmp.h>
70 #include <net/udp.h>
71 #include <net/raw.h>
72 #include <net/snmp.h>
73 #include <net/tcp_states.h>
74 #include <net/inet_common.h>
75 #include <net/checksum.h>
76 #include <net/xfrm.h>
77 #include <linux/rtnetlink.h>
78 #include <linux/proc_fs.h>
79 #include <linux/seq_file.h>
80 #include <linux/netfilter.h>
81 #include <linux/netfilter_ipv4.h>
82
83 static struct raw_hashinfo raw_v4_hashinfo = {
84         .lock = __RW_LOCK_UNLOCKED(),
85 };
86
87 void raw_hash_sk(struct sock *sk, struct raw_hashinfo *h)
88 {
89         struct hlist_head *head;
90
91         head = &h->ht[inet_sk(sk)->num & (RAW_HTABLE_SIZE - 1)];
92
93         write_lock_bh(&h->lock);
94         sk_add_node(sk, head);
95         sock_prot_inuse_add(sk->sk_prot, 1);
96         write_unlock_bh(&h->lock);
97 }
98 EXPORT_SYMBOL_GPL(raw_hash_sk);
99
100 void raw_unhash_sk(struct sock *sk, struct raw_hashinfo *h)
101 {
102         write_lock_bh(&h->lock);
103         if (sk_del_node_init(sk))
104                 sock_prot_inuse_add(sk->sk_prot, -1);
105         write_unlock_bh(&h->lock);
106 }
107 EXPORT_SYMBOL_GPL(raw_unhash_sk);
108
109 static void raw_v4_hash(struct sock *sk)
110 {
111         raw_hash_sk(sk, &raw_v4_hashinfo);
112 }
113
114 static void raw_v4_unhash(struct sock *sk)
115 {
116         raw_unhash_sk(sk, &raw_v4_hashinfo);
117 }
118
119 static struct sock *__raw_v4_lookup(struct net *net, struct sock *sk,
120                 unsigned short num, __be32 raddr, __be32 laddr, int dif)
121 {
122         struct hlist_node *node;
123
124         sk_for_each_from(sk, node) {
125                 struct inet_sock *inet = inet_sk(sk);
126
127                 if (sk->sk_net == net && inet->num == num               &&
128                     !(inet->daddr && inet->daddr != raddr)              &&
129                     !(inet->rcv_saddr && inet->rcv_saddr != laddr)      &&
130                     !(sk->sk_bound_dev_if && sk->sk_bound_dev_if != dif))
131                         goto found; /* gotcha */
132         }
133         sk = NULL;
134 found:
135         return sk;
136 }
137
138 /*
139  *      0 - deliver
140  *      1 - block
141  */
142 static __inline__ int icmp_filter(struct sock *sk, struct sk_buff *skb)
143 {
144         int type;
145
146         if (!pskb_may_pull(skb, sizeof(struct icmphdr)))
147                 return 1;
148
149         type = icmp_hdr(skb)->type;
150         if (type < 32) {
151                 __u32 data = raw_sk(sk)->filter.data;
152
153                 return ((1 << type) & data) != 0;
154         }
155
156         /* Do not block unknown ICMP types */
157         return 0;
158 }
159
160 /* IP input processing comes here for RAW socket delivery.
161  * Caller owns SKB, so we must make clones.
162  *
163  * RFC 1122: SHOULD pass TOS value up to the transport layer.
164  * -> It does. And not only TOS, but all IP header.
165  */
166 static int raw_v4_input(struct sk_buff *skb, struct iphdr *iph, int hash)
167 {
168         struct sock *sk;
169         struct hlist_head *head;
170         int delivered = 0;
171         struct net *net;
172
173         read_lock(&raw_v4_hashinfo.lock);
174         head = &raw_v4_hashinfo.ht[hash];
175         if (hlist_empty(head))
176                 goto out;
177
178         net = skb->dev->nd_net;
179         sk = __raw_v4_lookup(net, __sk_head(head), iph->protocol,
180                              iph->saddr, iph->daddr,
181                              skb->dev->ifindex);
182
183         while (sk) {
184                 delivered = 1;
185                 if (iph->protocol != IPPROTO_ICMP || !icmp_filter(sk, skb)) {
186                         struct sk_buff *clone = skb_clone(skb, GFP_ATOMIC);
187
188                         /* Not releasing hash table! */
189                         if (clone)
190                                 raw_rcv(sk, clone);
191                 }
192                 sk = __raw_v4_lookup(net, sk_next(sk), iph->protocol,
193                                      iph->saddr, iph->daddr,
194                                      skb->dev->ifindex);
195         }
196 out:
197         read_unlock(&raw_v4_hashinfo.lock);
198         return delivered;
199 }
200
201 int raw_local_deliver(struct sk_buff *skb, int protocol)
202 {
203         int hash;
204         struct sock *raw_sk;
205
206         hash = protocol & (RAW_HTABLE_SIZE - 1);
207         raw_sk = sk_head(&raw_v4_hashinfo.ht[hash]);
208
209         /* If there maybe a raw socket we must check - if not we
210          * don't care less
211          */
212         if (raw_sk && !raw_v4_input(skb, ip_hdr(skb), hash))
213                 raw_sk = NULL;
214
215         return raw_sk != NULL;
216
217 }
218
219 static void raw_err(struct sock *sk, struct sk_buff *skb, u32 info)
220 {
221         struct inet_sock *inet = inet_sk(sk);
222         const int type = icmp_hdr(skb)->type;
223         const int code = icmp_hdr(skb)->code;
224         int err = 0;
225         int harderr = 0;
226
227         /* Report error on raw socket, if:
228            1. User requested ip_recverr.
229            2. Socket is connected (otherwise the error indication
230               is useless without ip_recverr and error is hard.
231          */
232         if (!inet->recverr && sk->sk_state != TCP_ESTABLISHED)
233                 return;
234
235         switch (type) {
236         default:
237         case ICMP_TIME_EXCEEDED:
238                 err = EHOSTUNREACH;
239                 break;
240         case ICMP_SOURCE_QUENCH:
241                 return;
242         case ICMP_PARAMETERPROB:
243                 err = EPROTO;
244                 harderr = 1;
245                 break;
246         case ICMP_DEST_UNREACH:
247                 err = EHOSTUNREACH;
248                 if (code > NR_ICMP_UNREACH)
249                         break;
250                 err = icmp_err_convert[code].errno;
251                 harderr = icmp_err_convert[code].fatal;
252                 if (code == ICMP_FRAG_NEEDED) {
253                         harderr = inet->pmtudisc != IP_PMTUDISC_DONT;
254                         err = EMSGSIZE;
255                 }
256         }
257
258         if (inet->recverr) {
259                 struct iphdr *iph = (struct iphdr*)skb->data;
260                 u8 *payload = skb->data + (iph->ihl << 2);
261
262                 if (inet->hdrincl)
263                         payload = skb->data;
264                 ip_icmp_error(sk, skb, err, 0, info, payload);
265         }
266
267         if (inet->recverr || harderr) {
268                 sk->sk_err = err;
269                 sk->sk_error_report(sk);
270         }
271 }
272
273 void raw_icmp_error(struct sk_buff *skb, int protocol, u32 info)
274 {
275         int hash;
276         struct sock *raw_sk;
277         struct iphdr *iph;
278         struct net *net;
279
280         hash = protocol & (RAW_HTABLE_SIZE - 1);
281
282         read_lock(&raw_v4_hashinfo.lock);
283         raw_sk = sk_head(&raw_v4_hashinfo.ht[hash]);
284         if (raw_sk != NULL) {
285                 iph = (struct iphdr *)skb->data;
286                 net = skb->dev->nd_net;
287
288                 while ((raw_sk = __raw_v4_lookup(net, raw_sk, protocol,
289                                                 iph->daddr, iph->saddr,
290                                                 skb->dev->ifindex)) != NULL) {
291                         raw_err(raw_sk, skb, info);
292                         raw_sk = sk_next(raw_sk);
293                         iph = (struct iphdr *)skb->data;
294                 }
295         }
296         read_unlock(&raw_v4_hashinfo.lock);
297 }
298
299 static int raw_rcv_skb(struct sock * sk, struct sk_buff * skb)
300 {
301         /* Charge it to the socket. */
302
303         if (sock_queue_rcv_skb(sk, skb) < 0) {
304                 atomic_inc(&sk->sk_drops);
305                 kfree_skb(skb);
306                 return NET_RX_DROP;
307         }
308
309         return NET_RX_SUCCESS;
310 }
311
312 int raw_rcv(struct sock *sk, struct sk_buff *skb)
313 {
314         if (!xfrm4_policy_check(sk, XFRM_POLICY_IN, skb)) {
315                 atomic_inc(&sk->sk_drops);
316                 kfree_skb(skb);
317                 return NET_RX_DROP;
318         }
319         nf_reset(skb);
320
321         skb_push(skb, skb->data - skb_network_header(skb));
322
323         raw_rcv_skb(sk, skb);
324         return 0;
325 }
326
327 static int raw_send_hdrinc(struct sock *sk, void *from, size_t length,
328                         struct rtable *rt,
329                         unsigned int flags)
330 {
331         struct inet_sock *inet = inet_sk(sk);
332         int hh_len;
333         struct iphdr *iph;
334         struct sk_buff *skb;
335         unsigned int iphlen;
336         int err;
337
338         if (length > rt->u.dst.dev->mtu) {
339                 ip_local_error(sk, EMSGSIZE, rt->rt_dst, inet->dport,
340                                rt->u.dst.dev->mtu);
341                 return -EMSGSIZE;
342         }
343         if (flags&MSG_PROBE)
344                 goto out;
345
346         hh_len = LL_RESERVED_SPACE(rt->u.dst.dev);
347
348         skb = sock_alloc_send_skb(sk, length+hh_len+15,
349                                   flags&MSG_DONTWAIT, &err);
350         if (skb == NULL)
351                 goto error;
352         skb_reserve(skb, hh_len);
353
354         skb->priority = sk->sk_priority;
355         skb->mark = sk->sk_mark;
356         skb->dst = dst_clone(&rt->u.dst);
357
358         skb_reset_network_header(skb);
359         iph = ip_hdr(skb);
360         skb_put(skb, length);
361
362         skb->ip_summed = CHECKSUM_NONE;
363
364         skb->transport_header = skb->network_header;
365         err = memcpy_fromiovecend((void *)iph, from, 0, length);
366         if (err)
367                 goto error_fault;
368
369         /* We don't modify invalid header */
370         iphlen = iph->ihl * 4;
371         if (iphlen >= sizeof(*iph) && iphlen <= length) {
372                 if (!iph->saddr)
373                         iph->saddr = rt->rt_src;
374                 iph->check   = 0;
375                 iph->tot_len = htons(length);
376                 if (!iph->id)
377                         ip_select_ident(iph, &rt->u.dst, NULL);
378
379                 iph->check = ip_fast_csum((unsigned char *)iph, iph->ihl);
380         }
381         if (iph->protocol == IPPROTO_ICMP)
382                 icmp_out_count(((struct icmphdr *)
383                         skb_transport_header(skb))->type);
384
385         err = NF_HOOK(PF_INET, NF_INET_LOCAL_OUT, skb, NULL, rt->u.dst.dev,
386                       dst_output);
387         if (err > 0)
388                 err = inet->recverr ? net_xmit_errno(err) : 0;
389         if (err)
390                 goto error;
391 out:
392         return 0;
393
394 error_fault:
395         err = -EFAULT;
396         kfree_skb(skb);
397 error:
398         IP_INC_STATS(IPSTATS_MIB_OUTDISCARDS);
399         return err;
400 }
401
402 static int raw_probe_proto_opt(struct flowi *fl, struct msghdr *msg)
403 {
404         struct iovec *iov;
405         u8 __user *type = NULL;
406         u8 __user *code = NULL;
407         int probed = 0;
408         unsigned int i;
409
410         if (!msg->msg_iov)
411                 return 0;
412
413         for (i = 0; i < msg->msg_iovlen; i++) {
414                 iov = &msg->msg_iov[i];
415                 if (!iov)
416                         continue;
417
418                 switch (fl->proto) {
419                 case IPPROTO_ICMP:
420                         /* check if one-byte field is readable or not. */
421                         if (iov->iov_base && iov->iov_len < 1)
422                                 break;
423
424                         if (!type) {
425                                 type = iov->iov_base;
426                                 /* check if code field is readable or not. */
427                                 if (iov->iov_len > 1)
428                                         code = type + 1;
429                         } else if (!code)
430                                 code = iov->iov_base;
431
432                         if (type && code) {
433                                 if (get_user(fl->fl_icmp_type, type) ||
434                                     get_user(fl->fl_icmp_code, code))
435                                         return -EFAULT;
436                                 probed = 1;
437                         }
438                         break;
439                 default:
440                         probed = 1;
441                         break;
442                 }
443                 if (probed)
444                         break;
445         }
446         return 0;
447 }
448
449 static int raw_sendmsg(struct kiocb *iocb, struct sock *sk, struct msghdr *msg,
450                        size_t len)
451 {
452         struct inet_sock *inet = inet_sk(sk);
453         struct ipcm_cookie ipc;
454         struct rtable *rt = NULL;
455         int free = 0;
456         __be32 daddr;
457         __be32 saddr;
458         u8  tos;
459         int err;
460
461         err = -EMSGSIZE;
462         if (len > 0xFFFF)
463                 goto out;
464
465         /*
466          *      Check the flags.
467          */
468
469         err = -EOPNOTSUPP;
470         if (msg->msg_flags & MSG_OOB)   /* Mirror BSD error message */
471                 goto out;               /* compatibility */
472
473         /*
474          *      Get and verify the address.
475          */
476
477         if (msg->msg_namelen) {
478                 struct sockaddr_in *usin = (struct sockaddr_in*)msg->msg_name;
479                 err = -EINVAL;
480                 if (msg->msg_namelen < sizeof(*usin))
481                         goto out;
482                 if (usin->sin_family != AF_INET) {
483                         static int complained;
484                         if (!complained++)
485                                 printk(KERN_INFO "%s forgot to set AF_INET in "
486                                                  "raw sendmsg. Fix it!\n",
487                                                  current->comm);
488                         err = -EAFNOSUPPORT;
489                         if (usin->sin_family)
490                                 goto out;
491                 }
492                 daddr = usin->sin_addr.s_addr;
493                 /* ANK: I did not forget to get protocol from port field.
494                  * I just do not know, who uses this weirdness.
495                  * IP_HDRINCL is much more convenient.
496                  */
497         } else {
498                 err = -EDESTADDRREQ;
499                 if (sk->sk_state != TCP_ESTABLISHED)
500                         goto out;
501                 daddr = inet->daddr;
502         }
503
504         ipc.addr = inet->saddr;
505         ipc.opt = NULL;
506         ipc.oif = sk->sk_bound_dev_if;
507
508         if (msg->msg_controllen) {
509                 err = ip_cmsg_send(msg, &ipc);
510                 if (err)
511                         goto out;
512                 if (ipc.opt)
513                         free = 1;
514         }
515
516         saddr = ipc.addr;
517         ipc.addr = daddr;
518
519         if (!ipc.opt)
520                 ipc.opt = inet->opt;
521
522         if (ipc.opt) {
523                 err = -EINVAL;
524                 /* Linux does not mangle headers on raw sockets,
525                  * so that IP options + IP_HDRINCL is non-sense.
526                  */
527                 if (inet->hdrincl)
528                         goto done;
529                 if (ipc.opt->srr) {
530                         if (!daddr)
531                                 goto done;
532                         daddr = ipc.opt->faddr;
533                 }
534         }
535         tos = RT_CONN_FLAGS(sk);
536         if (msg->msg_flags & MSG_DONTROUTE)
537                 tos |= RTO_ONLINK;
538
539         if (ipv4_is_multicast(daddr)) {
540                 if (!ipc.oif)
541                         ipc.oif = inet->mc_index;
542                 if (!saddr)
543                         saddr = inet->mc_addr;
544         }
545
546         {
547                 struct flowi fl = { .oif = ipc.oif,
548                                     .mark = sk->sk_mark,
549                                     .nl_u = { .ip4_u =
550                                               { .daddr = daddr,
551                                                 .saddr = saddr,
552                                                 .tos = tos } },
553                                     .proto = inet->hdrincl ? IPPROTO_RAW :
554                                                              sk->sk_protocol,
555                                   };
556                 if (!inet->hdrincl) {
557                         err = raw_probe_proto_opt(&fl, msg);
558                         if (err)
559                                 goto done;
560                 }
561
562                 security_sk_classify_flow(sk, &fl);
563                 err = ip_route_output_flow(&init_net, &rt, &fl, sk, 1);
564         }
565         if (err)
566                 goto done;
567
568         err = -EACCES;
569         if (rt->rt_flags & RTCF_BROADCAST && !sock_flag(sk, SOCK_BROADCAST))
570                 goto done;
571
572         if (msg->msg_flags & MSG_CONFIRM)
573                 goto do_confirm;
574 back_from_confirm:
575
576         if (inet->hdrincl)
577                 err = raw_send_hdrinc(sk, msg->msg_iov, len,
578                                         rt, msg->msg_flags);
579
580          else {
581                 if (!ipc.addr)
582                         ipc.addr = rt->rt_dst;
583                 lock_sock(sk);
584                 err = ip_append_data(sk, ip_generic_getfrag, msg->msg_iov, len, 0,
585                                         &ipc, rt, msg->msg_flags);
586                 if (err)
587                         ip_flush_pending_frames(sk);
588                 else if (!(msg->msg_flags & MSG_MORE))
589                         err = ip_push_pending_frames(sk);
590                 release_sock(sk);
591         }
592 done:
593         if (free)
594                 kfree(ipc.opt);
595         ip_rt_put(rt);
596
597 out:
598         if (err < 0)
599                 return err;
600         return len;
601
602 do_confirm:
603         dst_confirm(&rt->u.dst);
604         if (!(msg->msg_flags & MSG_PROBE) || len)
605                 goto back_from_confirm;
606         err = 0;
607         goto done;
608 }
609
610 static void raw_close(struct sock *sk, long timeout)
611 {
612         /*
613          * Raw sockets may have direct kernel refereneces. Kill them.
614          */
615         ip_ra_control(sk, 0, NULL);
616
617         sk_common_release(sk);
618 }
619
620 /* This gets rid of all the nasties in af_inet. -DaveM */
621 static int raw_bind(struct sock *sk, struct sockaddr *uaddr, int addr_len)
622 {
623         struct inet_sock *inet = inet_sk(sk);
624         struct sockaddr_in *addr = (struct sockaddr_in *) uaddr;
625         int ret = -EINVAL;
626         int chk_addr_ret;
627
628         if (sk->sk_state != TCP_CLOSE || addr_len < sizeof(struct sockaddr_in))
629                 goto out;
630         chk_addr_ret = inet_addr_type(sk->sk_net, addr->sin_addr.s_addr);
631         ret = -EADDRNOTAVAIL;
632         if (addr->sin_addr.s_addr && chk_addr_ret != RTN_LOCAL &&
633             chk_addr_ret != RTN_MULTICAST && chk_addr_ret != RTN_BROADCAST)
634                 goto out;
635         inet->rcv_saddr = inet->saddr = addr->sin_addr.s_addr;
636         if (chk_addr_ret == RTN_MULTICAST || chk_addr_ret == RTN_BROADCAST)
637                 inet->saddr = 0;  /* Use device */
638         sk_dst_reset(sk);
639         ret = 0;
640 out:    return ret;
641 }
642
643 /*
644  *      This should be easy, if there is something there
645  *      we return it, otherwise we block.
646  */
647
648 static int raw_recvmsg(struct kiocb *iocb, struct sock *sk, struct msghdr *msg,
649                        size_t len, int noblock, int flags, int *addr_len)
650 {
651         struct inet_sock *inet = inet_sk(sk);
652         size_t copied = 0;
653         int err = -EOPNOTSUPP;
654         struct sockaddr_in *sin = (struct sockaddr_in *)msg->msg_name;
655         struct sk_buff *skb;
656
657         if (flags & MSG_OOB)
658                 goto out;
659
660         if (addr_len)
661                 *addr_len = sizeof(*sin);
662
663         if (flags & MSG_ERRQUEUE) {
664                 err = ip_recv_error(sk, msg, len);
665                 goto out;
666         }
667
668         skb = skb_recv_datagram(sk, flags, noblock, &err);
669         if (!skb)
670                 goto out;
671
672         copied = skb->len;
673         if (len < copied) {
674                 msg->msg_flags |= MSG_TRUNC;
675                 copied = len;
676         }
677
678         err = skb_copy_datagram_iovec(skb, 0, msg->msg_iov, copied);
679         if (err)
680                 goto done;
681
682         sock_recv_timestamp(msg, sk, skb);
683
684         /* Copy the address. */
685         if (sin) {
686                 sin->sin_family = AF_INET;
687                 sin->sin_addr.s_addr = ip_hdr(skb)->saddr;
688                 sin->sin_port = 0;
689                 memset(&sin->sin_zero, 0, sizeof(sin->sin_zero));
690         }
691         if (inet->cmsg_flags)
692                 ip_cmsg_recv(msg, skb);
693         if (flags & MSG_TRUNC)
694                 copied = skb->len;
695 done:
696         skb_free_datagram(sk, skb);
697 out:
698         if (err)
699                 return err;
700         return copied;
701 }
702
703 static int raw_init(struct sock *sk)
704 {
705         struct raw_sock *rp = raw_sk(sk);
706
707         if (inet_sk(sk)->num == IPPROTO_ICMP)
708                 memset(&rp->filter, 0, sizeof(rp->filter));
709         return 0;
710 }
711
712 static int raw_seticmpfilter(struct sock *sk, char __user *optval, int optlen)
713 {
714         if (optlen > sizeof(struct icmp_filter))
715                 optlen = sizeof(struct icmp_filter);
716         if (copy_from_user(&raw_sk(sk)->filter, optval, optlen))
717                 return -EFAULT;
718         return 0;
719 }
720
721 static int raw_geticmpfilter(struct sock *sk, char __user *optval, int __user *optlen)
722 {
723         int len, ret = -EFAULT;
724
725         if (get_user(len, optlen))
726                 goto out;
727         ret = -EINVAL;
728         if (len < 0)
729                 goto out;
730         if (len > sizeof(struct icmp_filter))
731                 len = sizeof(struct icmp_filter);
732         ret = -EFAULT;
733         if (put_user(len, optlen) ||
734             copy_to_user(optval, &raw_sk(sk)->filter, len))
735                 goto out;
736         ret = 0;
737 out:    return ret;
738 }
739
740 static int do_raw_setsockopt(struct sock *sk, int level, int optname,
741                           char __user *optval, int optlen)
742 {
743         if (optname == ICMP_FILTER) {
744                 if (inet_sk(sk)->num != IPPROTO_ICMP)
745                         return -EOPNOTSUPP;
746                 else
747                         return raw_seticmpfilter(sk, optval, optlen);
748         }
749         return -ENOPROTOOPT;
750 }
751
752 static int raw_setsockopt(struct sock *sk, int level, int optname,
753                           char __user *optval, int optlen)
754 {
755         if (level != SOL_RAW)
756                 return ip_setsockopt(sk, level, optname, optval, optlen);
757         return do_raw_setsockopt(sk, level, optname, optval, optlen);
758 }
759
760 #ifdef CONFIG_COMPAT
761 static int compat_raw_setsockopt(struct sock *sk, int level, int optname,
762                                  char __user *optval, int optlen)
763 {
764         if (level != SOL_RAW)
765                 return compat_ip_setsockopt(sk, level, optname, optval, optlen);
766         return do_raw_setsockopt(sk, level, optname, optval, optlen);
767 }
768 #endif
769
770 static int do_raw_getsockopt(struct sock *sk, int level, int optname,
771                           char __user *optval, int __user *optlen)
772 {
773         if (optname == ICMP_FILTER) {
774                 if (inet_sk(sk)->num != IPPROTO_ICMP)
775                         return -EOPNOTSUPP;
776                 else
777                         return raw_geticmpfilter(sk, optval, optlen);
778         }
779         return -ENOPROTOOPT;
780 }
781
782 static int raw_getsockopt(struct sock *sk, int level, int optname,
783                           char __user *optval, int __user *optlen)
784 {
785         if (level != SOL_RAW)
786                 return ip_getsockopt(sk, level, optname, optval, optlen);
787         return do_raw_getsockopt(sk, level, optname, optval, optlen);
788 }
789
790 #ifdef CONFIG_COMPAT
791 static int compat_raw_getsockopt(struct sock *sk, int level, int optname,
792                                  char __user *optval, int __user *optlen)
793 {
794         if (level != SOL_RAW)
795                 return compat_ip_getsockopt(sk, level, optname, optval, optlen);
796         return do_raw_getsockopt(sk, level, optname, optval, optlen);
797 }
798 #endif
799
800 static int raw_ioctl(struct sock *sk, int cmd, unsigned long arg)
801 {
802         switch (cmd) {
803                 case SIOCOUTQ: {
804                         int amount = atomic_read(&sk->sk_wmem_alloc);
805                         return put_user(amount, (int __user *)arg);
806                 }
807                 case SIOCINQ: {
808                         struct sk_buff *skb;
809                         int amount = 0;
810
811                         spin_lock_bh(&sk->sk_receive_queue.lock);
812                         skb = skb_peek(&sk->sk_receive_queue);
813                         if (skb != NULL)
814                                 amount = skb->len;
815                         spin_unlock_bh(&sk->sk_receive_queue.lock);
816                         return put_user(amount, (int __user *)arg);
817                 }
818
819                 default:
820 #ifdef CONFIG_IP_MROUTE
821                         return ipmr_ioctl(sk, cmd, (void __user *)arg);
822 #else
823                         return -ENOIOCTLCMD;
824 #endif
825         }
826 }
827
828 DEFINE_PROTO_INUSE(raw)
829
830 struct proto raw_prot = {
831         .name              = "RAW",
832         .owner             = THIS_MODULE,
833         .close             = raw_close,
834         .connect           = ip4_datagram_connect,
835         .disconnect        = udp_disconnect,
836         .ioctl             = raw_ioctl,
837         .init              = raw_init,
838         .setsockopt        = raw_setsockopt,
839         .getsockopt        = raw_getsockopt,
840         .sendmsg           = raw_sendmsg,
841         .recvmsg           = raw_recvmsg,
842         .bind              = raw_bind,
843         .backlog_rcv       = raw_rcv_skb,
844         .hash              = raw_v4_hash,
845         .unhash            = raw_v4_unhash,
846         .obj_size          = sizeof(struct raw_sock),
847 #ifdef CONFIG_COMPAT
848         .compat_setsockopt = compat_raw_setsockopt,
849         .compat_getsockopt = compat_raw_getsockopt,
850 #endif
851         REF_PROTO_INUSE(raw)
852 };
853
854 #ifdef CONFIG_PROC_FS
855 static struct sock *raw_get_first(struct seq_file *seq)
856 {
857         struct sock *sk;
858         struct raw_iter_state* state = raw_seq_private(seq);
859
860         for (state->bucket = 0; state->bucket < RAW_HTABLE_SIZE;
861                         ++state->bucket) {
862                 struct hlist_node *node;
863
864                 sk_for_each(sk, node, &state->h->ht[state->bucket])
865                         if (sk->sk_net == state->p.net)
866                                 goto found;
867         }
868         sk = NULL;
869 found:
870         return sk;
871 }
872
873 static struct sock *raw_get_next(struct seq_file *seq, struct sock *sk)
874 {
875         struct raw_iter_state* state = raw_seq_private(seq);
876
877         do {
878                 sk = sk_next(sk);
879 try_again:
880                 ;
881         } while (sk && sk->sk_net != state->p.net);
882
883         if (!sk && ++state->bucket < RAW_HTABLE_SIZE) {
884                 sk = sk_head(&state->h->ht[state->bucket]);
885                 goto try_again;
886         }
887         return sk;
888 }
889
890 static struct sock *raw_get_idx(struct seq_file *seq, loff_t pos)
891 {
892         struct sock *sk = raw_get_first(seq);
893
894         if (sk)
895                 while (pos && (sk = raw_get_next(seq, sk)) != NULL)
896                         --pos;
897         return pos ? NULL : sk;
898 }
899
900 void *raw_seq_start(struct seq_file *seq, loff_t *pos)
901 {
902         struct raw_iter_state *state = raw_seq_private(seq);
903
904         read_lock(&state->h->lock);
905         return *pos ? raw_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
906 }
907 EXPORT_SYMBOL_GPL(raw_seq_start);
908
909 void *raw_seq_next(struct seq_file *seq, void *v, loff_t *pos)
910 {
911         struct sock *sk;
912
913         if (v == SEQ_START_TOKEN)
914                 sk = raw_get_first(seq);
915         else
916                 sk = raw_get_next(seq, v);
917         ++*pos;
918         return sk;
919 }
920 EXPORT_SYMBOL_GPL(raw_seq_next);
921
922 void raw_seq_stop(struct seq_file *seq, void *v)
923 {
924         struct raw_iter_state *state = raw_seq_private(seq);
925
926         read_unlock(&state->h->lock);
927 }
928 EXPORT_SYMBOL_GPL(raw_seq_stop);
929
930 static void raw_sock_seq_show(struct seq_file *seq, struct sock *sp, int i)
931 {
932         struct inet_sock *inet = inet_sk(sp);
933         __be32 dest = inet->daddr,
934                src = inet->rcv_saddr;
935         __u16 destp = 0,
936               srcp  = inet->num;
937
938         seq_printf(seq, "%4d: %08X:%04X %08X:%04X"
939                 " %02X %08X:%08X %02X:%08lX %08X %5d %8d %lu %d %p %d",
940                 i, src, srcp, dest, destp, sp->sk_state,
941                 atomic_read(&sp->sk_wmem_alloc),
942                 atomic_read(&sp->sk_rmem_alloc),
943                 0, 0L, 0, sock_i_uid(sp), 0, sock_i_ino(sp),
944                 atomic_read(&sp->sk_refcnt), sp, atomic_read(&sp->sk_drops));
945 }
946
947 static int raw_seq_show(struct seq_file *seq, void *v)
948 {
949         if (v == SEQ_START_TOKEN)
950                 seq_printf(seq, "  sl  local_address rem_address   st tx_queue "
951                                 "rx_queue tr tm->when retrnsmt   uid  timeout "
952                                 "inode  drops\n");
953         else
954                 raw_sock_seq_show(seq, v, raw_seq_private(seq)->bucket);
955         return 0;
956 }
957
958 static const struct seq_operations raw_seq_ops = {
959         .start = raw_seq_start,
960         .next  = raw_seq_next,
961         .stop  = raw_seq_stop,
962         .show  = raw_seq_show,
963 };
964
965 int raw_seq_open(struct inode *ino, struct file *file,
966                  struct raw_hashinfo *h, const struct seq_operations *ops)
967 {
968         int err;
969         struct raw_iter_state *i;
970
971         err = seq_open_net(ino, file, ops, sizeof(struct raw_iter_state));
972         if (err < 0)
973                 return err;
974
975         i = raw_seq_private((struct seq_file *)file->private_data);
976         i->h = h;
977         return 0;
978 }
979 EXPORT_SYMBOL_GPL(raw_seq_open);
980
981 static int raw_v4_seq_open(struct inode *inode, struct file *file)
982 {
983         return raw_seq_open(inode, file, &raw_v4_hashinfo, &raw_seq_ops);
984 }
985
986 static const struct file_operations raw_seq_fops = {
987         .owner   = THIS_MODULE,
988         .open    = raw_v4_seq_open,
989         .read    = seq_read,
990         .llseek  = seq_lseek,
991         .release = seq_release_net,
992 };
993
994 static __net_init int raw_init_net(struct net *net)
995 {
996         if (!proc_net_fops_create(net, "raw", S_IRUGO, &raw_seq_fops))
997                 return -ENOMEM;
998
999         return 0;
1000 }
1001
1002 static __net_exit void raw_exit_net(struct net *net)
1003 {
1004         proc_net_remove(net, "raw");
1005 }
1006
1007 static __net_initdata struct pernet_operations raw_net_ops = {
1008         .init = raw_init_net,
1009         .exit = raw_exit_net,
1010 };
1011
1012 int __init raw_proc_init(void)
1013 {
1014         return register_pernet_subsys(&raw_net_ops);
1015 }
1016
1017 void __init raw_proc_exit(void)
1018 {
1019         unregister_pernet_subsys(&raw_net_ops);
1020 }
1021 #endif /* CONFIG_PROC_FS */