?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
pasemi_mac: don't enable rx before there are buffers on the ring
[linux-2.6/kvm.git] / net / packet / af_packet.c
blobe11000a8e950950cd5a52ece4597c72761dc8d38
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 * PACKET - implements raw packet sockets.
7 *
8 * Version: $Id: af_packet.c,v 1.61 2002/02/08 03:57:19 davem Exp $
9 *
10 * Authors: Ross Biro
11 * Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG>
12 * Alan Cox, <gw4pts@gw4pts.ampr.org>
13 *
14 * Fixes:
15 * Alan Cox : verify_area() now used correctly
16 * Alan Cox : new skbuff lists, look ma no backlogs!
17 * Alan Cox : tidied skbuff lists.
18 * Alan Cox : Now uses generic datagram routines I
19 * added. Also fixed the peek/read crash
20 * from all old Linux datagram code.
21 * Alan Cox : Uses the improved datagram code.
22 * Alan Cox : Added NULL's for socket options.
23 * Alan Cox : Re-commented the code.
24 * Alan Cox : Use new kernel side addressing
25 * Rob Janssen : Correct MTU usage.
26 * Dave Platt : Counter leaks caused by incorrect
27 * interrupt locking and some slightly
28 * dubious gcc output. Can you read
29 * compiler: it said _VOLATILE_
30 * Richard Kooijman : Timestamp fixes.
31 * Alan Cox : New buffers. Use sk->mac.raw.
32 * Alan Cox : sendmsg/recvmsg support.
33 * Alan Cox : Protocol setting support
34 * Alexey Kuznetsov : Untied from IPv4 stack.
35 * Cyrus Durgin : Fixed kerneld for kmod.
36 * Michal Ostrowski : Module initialization cleanup.
37 * Ulises Alonso : Frame number limit removal and
38 * packet_set_ring memory leak.
39 * Eric Biederman : Allow for > 8 byte hardware addresses.
40 * The convention is that longer addresses
41 * will simply extend the hardware address
42 * byte arrays at the end of sockaddr_ll
43 * and packet_mreq.
44 *
45 * This program is free software; you can redistribute it and/or
46 * modify it under the terms of the GNU General Public License
47 * as published by the Free Software Foundation; either version
48 * 2 of the License, or (at your option) any later version.
49 *
50 */
51
52 #include <linux/types.h>
53 #include <linux/mm.h>
54 #include <linux/capability.h>
55 #include <linux/fcntl.h>
56 #include <linux/socket.h>
57 #include <linux/in.h>
58 #include <linux/inet.h>
59 #include <linux/netdevice.h>
60 #include <linux/if_packet.h>
61 #include <linux/wireless.h>
62 #include <linux/kernel.h>
63 #include <linux/kmod.h>
64 #include <net/net_namespace.h>
65 #include <net/ip.h>
66 #include <net/protocol.h>
67 #include <linux/skbuff.h>
68 #include <net/sock.h>
69 #include <linux/errno.h>
70 #include <linux/timer.h>
71 #include <asm/system.h>
72 #include <asm/uaccess.h>
73 #include <asm/ioctls.h>
74 #include <asm/page.h>
75 #include <asm/cacheflush.h>
76 #include <asm/io.h>
77 #include <linux/proc_fs.h>
78 #include <linux/seq_file.h>
79 #include <linux/poll.h>
80 #include <linux/module.h>
81 #include <linux/init.h>
82
83 #ifdef CONFIG_INET
84 #include <net/inet_common.h>
85 #endif
86
87 /*
88 Assumptions:
89 - if device has no dev->hard_header routine, it adds and removes ll header
90 inside itself. In this case ll header is invisible outside of device,
91 but higher levels still should reserve dev->hard_header_len.
92 Some devices are enough clever to reallocate skb, when header
93 will not fit to reserved space (tunnel), another ones are silly
94 (PPP).
95 - packet socket receives packets with pulled ll header,
96 so that SOCK_RAW should push it back.
97
98 On receive:
99 -----------
100
101 Incoming, dev->hard_header!=NULL
102 mac_header -> ll header
103 data -> data
104
105 Outgoing, dev->hard_header!=NULL
106 mac_header -> ll header
107 data -> ll header
108
109 Incoming, dev->hard_header==NULL
110 mac_header -> UNKNOWN position. It is very likely, that it points to ll
111 header. PPP makes it, that is wrong, because introduce
112 assymetry between rx and tx paths.
113 data -> data
114
115 Outgoing, dev->hard_header==NULL
116 mac_header -> data. ll header is still not built!
117 data -> data
118
119 Resume
120 If dev->hard_header==NULL we are unlikely to restore sensible ll header.
121
122
123 On transmit:
124 ------------
125
126 dev->hard_header != NULL
127 mac_header -> ll header
128 data -> ll header
129
130 dev->hard_header == NULL (ll header is added by device, we cannot control it)
131 mac_header -> data
132 data -> data
133
134 We should set nh.raw on output to correct posistion,
135 packet classifier depends on it.
136 */
137
138 /* List of all packet sockets. */
139 static HLIST_HEAD(packet_sklist);
140 static DEFINE_RWLOCK(packet_sklist_lock);
141
142 static atomic_t packet_socks_nr;
143
144
145 /* Private packet socket structures. */
146
147 struct packet_mclist
148 {
149 struct packet_mclist *next;
150 int ifindex;
151 int count;
152 unsigned short type;
153 unsigned short alen;
154 unsigned char addr[MAX_ADDR_LEN];
155 };
156 /* identical to struct packet_mreq except it has
157 * a longer address field.
158 */
159 struct packet_mreq_max
160 {
161 int mr_ifindex;
162 unsigned short mr_type;
163 unsigned short mr_alen;
164 unsigned char mr_address[MAX_ADDR_LEN];
165 };
166
167 #ifdef CONFIG_PACKET_MMAP
168 static int packet_set_ring(struct sock *sk, struct tpacket_req *req, int closing);
169 #endif
170
171 static void packet_flush_mclist(struct sock *sk);
172
173 struct packet_sock {
174 /* struct sock has to be the first member of packet_sock */
175 struct sock sk;
176 struct tpacket_stats stats;
177 #ifdef CONFIG_PACKET_MMAP
178 char * *pg_vec;
179 unsigned int head;
180 unsigned int frames_per_block;
181 unsigned int frame_size;
182 unsigned int frame_max;
183 int copy_thresh;
184 #endif
185 struct packet_type prot_hook;
186 spinlock_t bind_lock;
187 unsigned int running:1, /* prot_hook is attached*/
188 auxdata:1,
189 origdev:1;
190 int ifindex; /* bound device */
191 __be16 num;
192 struct packet_mclist *mclist;
193 #ifdef CONFIG_PACKET_MMAP
194 atomic_t mapped;
195 unsigned int pg_vec_order;
196 unsigned int pg_vec_pages;
197 unsigned int pg_vec_len;
198 #endif
199 };
200
201 struct packet_skb_cb {
202 unsigned int origlen;
203 union {
204 struct sockaddr_pkt pkt;
205 struct sockaddr_ll ll;
206 } sa;
207 };
208
209 #define PACKET_SKB_CB(__skb) ((struct packet_skb_cb *)((__skb)->cb))
210
211 #ifdef CONFIG_PACKET_MMAP
212
213 static inline struct tpacket_hdr *packet_lookup_frame(struct packet_sock *po, unsigned int position)
214 {
215 unsigned int pg_vec_pos, frame_offset;
216
217 pg_vec_pos = position / po->frames_per_block;
218 frame_offset = position % po->frames_per_block;
219
220 return (struct tpacket_hdr *)(po->pg_vec[pg_vec_pos] + (frame_offset * po->frame_size));
221 }
222 #endif
223
224 static inline struct packet_sock *pkt_sk(struct sock *sk)
225 {
226 return (struct packet_sock *)sk;
227 }
228
229 static void packet_sock_destruct(struct sock *sk)
230 {
231 BUG_TRAP(!atomic_read(&sk->sk_rmem_alloc));
232 BUG_TRAP(!atomic_read(&sk->sk_wmem_alloc));
233
234 if (!sock_flag(sk, SOCK_DEAD)) {
235 printk("Attempt to release alive packet socket: %p\n", sk);
236 return;
237 }
238
239 atomic_dec(&packet_socks_nr);
240 #ifdef PACKET_REFCNT_DEBUG
241 printk(KERN_DEBUG "PACKET socket %p is free, %d are alive\n", sk, atomic_read(&packet_socks_nr));
242 #endif
243 }
244
245
246 static const struct proto_ops packet_ops;
247
248 static const struct proto_ops packet_ops_spkt;
249
250 static int packet_rcv_spkt(struct sk_buff *skb, struct net_device *dev, struct packet_type *pt, struct net_device *orig_dev)
251 {
252 struct sock *sk;
253 struct sockaddr_pkt *spkt;
254
255 if (dev->nd_net != &init_net)
256 goto out;
257
258 /*
259 * When we registered the protocol we saved the socket in the data
260 * field for just this event.
261 */
262
263 sk = pt->af_packet_priv;
264
265 /*
266 * Yank back the headers [hope the device set this
267 * right or kerboom...]
268 *
269 * Incoming packets have ll header pulled,
270 * push it back.
271 *
272 * For outgoing ones skb->data == skb_mac_header(skb)
273 * so that this procedure is noop.
274 */
275
276 if (skb->pkt_type == PACKET_LOOPBACK)
277 goto out;
278
279 if ((skb = skb_share_check(skb, GFP_ATOMIC)) == NULL)
280 goto oom;
281
282 /* drop any routing info */
283 dst_release(skb->dst);
284 skb->dst = NULL;
285
286 /* drop conntrack reference */
287 nf_reset(skb);
288
289 spkt = &PACKET_SKB_CB(skb)->sa.pkt;
290
291 skb_push(skb, skb->data - skb_mac_header(skb));
292
293 /*
294 * The SOCK_PACKET socket receives _all_ frames.
295 */
296
297 spkt->spkt_family = dev->type;
298 strlcpy(spkt->spkt_device, dev->name, sizeof(spkt->spkt_device));
299 spkt->spkt_protocol = skb->protocol;
300
301 /*
302 * Charge the memory to the socket. This is done specifically
303 * to prevent sockets using all the memory up.
304 */
305
306 if (sock_queue_rcv_skb(sk,skb) == 0)
307 return 0;
308
309 out:
310 kfree_skb(skb);
311 oom:
312 return 0;
313 }
314
315
316 /*
317 * Output a raw packet to a device layer. This bypasses all the other
318 * protocol layers and you must therefore supply it with a complete frame
319 */
320
321 static int packet_sendmsg_spkt(struct kiocb *iocb, struct socket *sock,
322 struct msghdr *msg, size_t len)
323 {
324 struct sock *sk = sock->sk;
325 struct sockaddr_pkt *saddr=(struct sockaddr_pkt *)msg->msg_name;
326 struct sk_buff *skb;
327 struct net_device *dev;
328 __be16 proto=0;
329 int err;
330
331 /*
332 * Get and verify the address.
333 */
334
335 if (saddr)
336 {
337 if (msg->msg_namelen < sizeof(struct sockaddr))
338 return(-EINVAL);
339 if (msg->msg_namelen==sizeof(struct sockaddr_pkt))
340 proto=saddr->spkt_protocol;
341 }
342 else
343 return(-ENOTCONN); /* SOCK_PACKET must be sent giving an address */
344
345 /*
346 * Find the device first to size check it
347 */
348
349 saddr->spkt_device[13] = 0;
350 dev = dev_get_by_name(&init_net, saddr->spkt_device);
351 err = -ENODEV;
352 if (dev == NULL)
353 goto out_unlock;
354
355 err = -ENETDOWN;
356 if (!(dev->flags & IFF_UP))
357 goto out_unlock;
358
359 /*
360 * You may not queue a frame bigger than the mtu. This is the lowest level
361 * raw protocol and you must do your own fragmentation at this level.
362 */
363
364 err = -EMSGSIZE;
365 if (len > dev->mtu + dev->hard_header_len)
366 goto out_unlock;
367
368 err = -ENOBUFS;
369 skb = sock_wmalloc(sk, len + LL_RESERVED_SPACE(dev), 0, GFP_KERNEL);
370
371 /*
372 * If the write buffer is full, then tough. At this level the user gets to
373 * deal with the problem - do your own algorithmic backoffs. That's far
374 * more flexible.
375 */
376
377 if (skb == NULL)
378 goto out_unlock;
379
380 /*
381 * Fill it in
382 */
383
384 /* FIXME: Save some space for broken drivers that write a
385 * hard header at transmission time by themselves. PPP is the
386 * notable one here. This should really be fixed at the driver level.
387 */
388 skb_reserve(skb, LL_RESERVED_SPACE(dev));
389 skb_reset_network_header(skb);
390
391 /* Try to align data part correctly */
392 if (dev->header_ops) {
393 skb->data -= dev->hard_header_len;
394 skb->tail -= dev->hard_header_len;
395 if (len < dev->hard_header_len)
396 skb_reset_network_header(skb);
397 }
398
399 /* Returns -EFAULT on error */
400 err = memcpy_fromiovec(skb_put(skb,len), msg->msg_iov, len);
401 skb->protocol = proto;
402 skb->dev = dev;
403 skb->priority = sk->sk_priority;
404 if (err)
405 goto out_free;
406
407 /*
408 * Now send it
409 */
410
411 dev_queue_xmit(skb);
412 dev_put(dev);
413 return(len);
414
415 out_free:
416 kfree_skb(skb);
417 out_unlock:
418 if (dev)
419 dev_put(dev);
420 return err;
421 }
422
423 static inline unsigned int run_filter(struct sk_buff *skb, struct sock *sk,
424 unsigned int res)
425 {
426 struct sk_filter *filter;
427
428 rcu_read_lock_bh();
429 filter = rcu_dereference(sk->sk_filter);
430 if (filter != NULL)
431 res = sk_run_filter(skb, filter->insns, filter->len);
432 rcu_read_unlock_bh();
433
434 return res;
435 }
436
437 /*
438 This function makes lazy skb cloning in hope that most of packets
439 are discarded by BPF.
440
441 Note tricky part: we DO mangle shared skb! skb->data, skb->len
442 and skb->cb are mangled. It works because (and until) packets
443 falling here are owned by current CPU. Output packets are cloned
444 by dev_queue_xmit_nit(), input packets are processed by net_bh
445 sequencially, so that if we return skb to original state on exit,
446 we will not harm anyone.
447 */
448
449 static int packet_rcv(struct sk_buff *skb, struct net_device *dev, struct packet_type *pt, struct net_device *orig_dev)
450 {
451 struct sock *sk;
452 struct sockaddr_ll *sll;
453 struct packet_sock *po;
454 u8 * skb_head = skb->data;
455 int skb_len = skb->len;
456 unsigned int snaplen, res;
457
458 if (dev->nd_net != &init_net)
459 goto drop;
460
461 if (skb->pkt_type == PACKET_LOOPBACK)
462 goto drop;
463
464 sk = pt->af_packet_priv;
465 po = pkt_sk(sk);
466
467 skb->dev = dev;
468
469 if (dev->header_ops) {
470 /* The device has an explicit notion of ll header,
471 exported to higher levels.
472
473 Otherwise, the device hides datails of it frame
474 structure, so that corresponding packet head
475 never delivered to user.
476 */
477 if (sk->sk_type != SOCK_DGRAM)
478 skb_push(skb, skb->data - skb_mac_header(skb));
479 else if (skb->pkt_type == PACKET_OUTGOING) {
480 /* Special case: outgoing packets have ll header at head */
481 skb_pull(skb, skb_network_offset(skb));
482 }
483 }
484
485 snaplen = skb->len;
486
487 res = run_filter(skb, sk, snaplen);
488 if (!res)
489 goto drop_n_restore;
490 if (snaplen > res)
491 snaplen = res;
492
493 if (atomic_read(&sk->sk_rmem_alloc) + skb->truesize >=
494 (unsigned)sk->sk_rcvbuf)
495 goto drop_n_acct;
496
497 if (skb_shared(skb)) {
498 struct sk_buff *nskb = skb_clone(skb, GFP_ATOMIC);
499 if (nskb == NULL)
500 goto drop_n_acct;
501
502 if (skb_head != skb->data) {
503 skb->data = skb_head;
504 skb->len = skb_len;
505 }
506 kfree_skb(skb);
507 skb = nskb;
508 }
509
510 BUILD_BUG_ON(sizeof(*PACKET_SKB_CB(skb)) + MAX_ADDR_LEN - 8 >
511 sizeof(skb->cb));
512
513 sll = &PACKET_SKB_CB(skb)->sa.ll;
514 sll->sll_family = AF_PACKET;
515 sll->sll_hatype = dev->type;
516 sll->sll_protocol = skb->protocol;
517 sll->sll_pkttype = skb->pkt_type;
518 if (unlikely(po->origdev) && skb->pkt_type == PACKET_HOST)
519 sll->sll_ifindex = orig_dev->ifindex;
520 else
521 sll->sll_ifindex = dev->ifindex;
522
523 sll->sll_halen = dev_parse_header(skb, sll->sll_addr);
524
525 PACKET_SKB_CB(skb)->origlen = skb->len;
526
527 if (pskb_trim(skb, snaplen))
528 goto drop_n_acct;
529
530 skb_set_owner_r(skb, sk);
531 skb->dev = NULL;
532 dst_release(skb->dst);
533 skb->dst = NULL;
534
535 /* drop conntrack reference */
536 nf_reset(skb);
537
538 spin_lock(&sk->sk_receive_queue.lock);
539 po->stats.tp_packets++;
540 __skb_queue_tail(&sk->sk_receive_queue, skb);
541 spin_unlock(&sk->sk_receive_queue.lock);
542 sk->sk_data_ready(sk, skb->len);
543 return 0;
544
545 drop_n_acct:
546 spin_lock(&sk->sk_receive_queue.lock);
547 po->stats.tp_drops++;
548 spin_unlock(&sk->sk_receive_queue.lock);
549
550 drop_n_restore:
551 if (skb_head != skb->data && skb_shared(skb)) {
552 skb->data = skb_head;
553 skb->len = skb_len;
554 }
555 drop:
556 kfree_skb(skb);
557 return 0;
558 }
559
560 #ifdef CONFIG_PACKET_MMAP
561 static int tpacket_rcv(struct sk_buff *skb, struct net_device *dev, struct packet_type *pt, struct net_device *orig_dev)
562 {
563 struct sock *sk;
564 struct packet_sock *po;
565 struct sockaddr_ll *sll;
566 struct tpacket_hdr *h;
567 u8 * skb_head = skb->data;
568 int skb_len = skb->len;
569 unsigned int snaplen, res;
570 unsigned long status = TP_STATUS_LOSING|TP_STATUS_USER;
571 unsigned short macoff, netoff;
572 struct sk_buff *copy_skb = NULL;
573 struct timeval tv;
574
575 if (dev->nd_net != &init_net)
576 goto drop;
577
578 if (skb->pkt_type == PACKET_LOOPBACK)
579 goto drop;
580
581 sk = pt->af_packet_priv;
582 po = pkt_sk(sk);
583
584 if (dev->header_ops) {
585 if (sk->sk_type != SOCK_DGRAM)
586 skb_push(skb, skb->data - skb_mac_header(skb));
587 else if (skb->pkt_type == PACKET_OUTGOING) {
588 /* Special case: outgoing packets have ll header at head */
589 skb_pull(skb, skb_network_offset(skb));
590 }
591 }
592
593 if (skb->ip_summed == CHECKSUM_PARTIAL)
594 status |= TP_STATUS_CSUMNOTREADY;
595
596 snaplen = skb->len;
597
598 res = run_filter(skb, sk, snaplen);
599 if (!res)
600 goto drop_n_restore;
601 if (snaplen > res)
602 snaplen = res;
603
604 if (sk->sk_type == SOCK_DGRAM) {
605 macoff = netoff = TPACKET_ALIGN(TPACKET_HDRLEN) + 16;
606 } else {
607 unsigned maclen = skb_network_offset(skb);
608 netoff = TPACKET_ALIGN(TPACKET_HDRLEN + (maclen < 16 ? 16 : maclen));
609 macoff = netoff - maclen;
610 }
611
612 if (macoff + snaplen > po->frame_size) {
613 if (po->copy_thresh &&
614 atomic_read(&sk->sk_rmem_alloc) + skb->truesize <
615 (unsigned)sk->sk_rcvbuf) {
616 if (skb_shared(skb)) {
617 copy_skb = skb_clone(skb, GFP_ATOMIC);
618 } else {
619 copy_skb = skb_get(skb);
620 skb_head = skb->data;
621 }
622 if (copy_skb)
623 skb_set_owner_r(copy_skb, sk);
624 }
625 snaplen = po->frame_size - macoff;
626 if ((int)snaplen < 0)
627 snaplen = 0;
628 }
629
630 spin_lock(&sk->sk_receive_queue.lock);
631 h = packet_lookup_frame(po, po->head);
632
633 if (h->tp_status)
634 goto ring_is_full;
635 po->head = po->head != po->frame_max ? po->head+1 : 0;
636 po->stats.tp_packets++;
637 if (copy_skb) {
638 status |= TP_STATUS_COPY;
639 __skb_queue_tail(&sk->sk_receive_queue, copy_skb);
640 }
641 if (!po->stats.tp_drops)
642 status &= ~TP_STATUS_LOSING;
643 spin_unlock(&sk->sk_receive_queue.lock);
644
645 skb_copy_bits(skb, 0, (u8*)h + macoff, snaplen);
646
647 h->tp_len = skb->len;
648 h->tp_snaplen = snaplen;
649 h->tp_mac = macoff;
650 h->tp_net = netoff;
651 if (skb->tstamp.tv64)
652 tv = ktime_to_timeval(skb->tstamp);
653 else
654 do_gettimeofday(&tv);
655 h->tp_sec = tv.tv_sec;
656 h->tp_usec = tv.tv_usec;
657
658 sll = (struct sockaddr_ll*)((u8*)h + TPACKET_ALIGN(sizeof(*h)));
659 sll->sll_halen = dev_parse_header(skb, sll->sll_addr);
660 sll->sll_family = AF_PACKET;
661 sll->sll_hatype = dev->type;
662 sll->sll_protocol = skb->protocol;
663 sll->sll_pkttype = skb->pkt_type;
664 if (unlikely(po->origdev) && skb->pkt_type == PACKET_HOST)
665 sll->sll_ifindex = orig_dev->ifindex;
666 else
667 sll->sll_ifindex = dev->ifindex;
668
669 h->tp_status = status;
670 smp_mb();
671
672 {
673 struct page *p_start, *p_end;
674 u8 *h_end = (u8 *)h + macoff + snaplen - 1;
675
676 p_start = virt_to_page(h);
677 p_end = virt_to_page(h_end);
678 while (p_start <= p_end) {
679 flush_dcache_page(p_start);
680 p_start++;
681 }
682 }
683
684 sk->sk_data_ready(sk, 0);
685
686 drop_n_restore:
687 if (skb_head != skb->data && skb_shared(skb)) {
688 skb->data = skb_head;
689 skb->len = skb_len;
690 }
691 drop:
692 kfree_skb(skb);
693 return 0;
694
695 ring_is_full:
696 po->stats.tp_drops++;
697 spin_unlock(&sk->sk_receive_queue.lock);
698
699 sk->sk_data_ready(sk, 0);
700 if (copy_skb)
701 kfree_skb(copy_skb);
702 goto drop_n_restore;
703 }
704
705 #endif
706
707
708 static int packet_sendmsg(struct kiocb *iocb, struct socket *sock,
709 struct msghdr *msg, size_t len)
710 {
711 struct sock *sk = sock->sk;
712 struct sockaddr_ll *saddr=(struct sockaddr_ll *)msg->msg_name;
713 struct sk_buff *skb;
714 struct net_device *dev;
715 __be16 proto;
716 unsigned char *addr;
717 int ifindex, err, reserve = 0;
718
719 /*
720 * Get and verify the address.
721 */
722
723 if (saddr == NULL) {
724 struct packet_sock *po = pkt_sk(sk);
725
726 ifindex = po->ifindex;
727 proto = po->num;
728 addr = NULL;
729 } else {
730 err = -EINVAL;
731 if (msg->msg_namelen < sizeof(struct sockaddr_ll))
732 goto out;
733 if (msg->msg_namelen < (saddr->sll_halen + offsetof(struct sockaddr_ll, sll_addr)))
734 goto out;
735 ifindex = saddr->sll_ifindex;
736 proto = saddr->sll_protocol;
737 addr = saddr->sll_addr;
738 }
739
740
741 dev = dev_get_by_index(&init_net, ifindex);
742 err = -ENXIO;
743 if (dev == NULL)
744 goto out_unlock;
745 if (sock->type == SOCK_RAW)
746 reserve = dev->hard_header_len;
747
748 err = -ENETDOWN;
749 if (!(dev->flags & IFF_UP))
750 goto out_unlock;
751
752 err = -EMSGSIZE;
753 if (len > dev->mtu+reserve)
754 goto out_unlock;
755
756 skb = sock_alloc_send_skb(sk, len + LL_RESERVED_SPACE(dev),
757 msg->msg_flags & MSG_DONTWAIT, &err);
758 if (skb==NULL)
759 goto out_unlock;
760
761 skb_reserve(skb, LL_RESERVED_SPACE(dev));
762 skb_reset_network_header(skb);
763
764 err = -EINVAL;
765 if (sock->type == SOCK_DGRAM &&
766 dev_hard_header(skb, dev, ntohs(proto), addr, NULL, len) < 0)
767 goto out_free;
768
769 /* Returns -EFAULT on error */
770 err = memcpy_fromiovec(skb_put(skb,len), msg->msg_iov, len);
771 if (err)
772 goto out_free;
773
774 skb->protocol = proto;
775 skb->dev = dev;
776 skb->priority = sk->sk_priority;
777
778 /*
779 * Now send it
780 */
781
782 err = dev_queue_xmit(skb);
783 if (err > 0 && (err = net_xmit_errno(err)) != 0)
784 goto out_unlock;
785
786 dev_put(dev);
787
788 return(len);
789
790 out_free:
791 kfree_skb(skb);
792 out_unlock:
793 if (dev)
794 dev_put(dev);
795 out:
796 return err;
797 }
798
799 /*
800 * Close a PACKET socket. This is fairly simple. We immediately go
801 * to 'closed' state and remove our protocol entry in the device list.
802 */
803
804 static int packet_release(struct socket *sock)
805 {
806 struct sock *sk = sock->sk;
807 struct packet_sock *po;
808
809 if (!sk)
810 return 0;
811
812 po = pkt_sk(sk);
813
814 write_lock_bh(&packet_sklist_lock);
815 sk_del_node_init(sk);
816 write_unlock_bh(&packet_sklist_lock);
817
818 /*
819 * Unhook packet receive handler.
820 */
821
822 if (po->running) {
823 /*
824 * Remove the protocol hook
825 */
826 dev_remove_pack(&po->prot_hook);
827 po->running = 0;
828 po->num = 0;
829 __sock_put(sk);
830 }
831
832 packet_flush_mclist(sk);
833
834 #ifdef CONFIG_PACKET_MMAP
835 if (po->pg_vec) {
836 struct tpacket_req req;
837 memset(&req, 0, sizeof(req));
838 packet_set_ring(sk, &req, 1);
839 }
840 #endif
841
842 /*
843 * Now the socket is dead. No more input will appear.
844 */
845
846 sock_orphan(sk);
847 sock->sk = NULL;
848
849 /* Purge queues */
850
851 skb_queue_purge(&sk->sk_receive_queue);
852
853 sock_put(sk);
854 return 0;
855 }
856
857 /*
858 * Attach a packet hook.
859 */
860
861 static int packet_do_bind(struct sock *sk, struct net_device *dev, __be16 protocol)
862 {
863 struct packet_sock *po = pkt_sk(sk);
864 /*
865 * Detach an existing hook if present.
866 */
867
868 lock_sock(sk);
869
870 spin_lock(&po->bind_lock);
871 if (po->running) {
872 __sock_put(sk);
873 po->running = 0;
874 po->num = 0;
875 spin_unlock(&po->bind_lock);
876 dev_remove_pack(&po->prot_hook);
877 spin_lock(&po->bind_lock);
878 }
879
880 po->num = protocol;
881 po->prot_hook.type = protocol;
882 po->prot_hook.dev = dev;
883
884 po->ifindex = dev ? dev->ifindex : 0;
885
886 if (protocol == 0)
887 goto out_unlock;
888
889 if (dev) {
890 if (dev->flags&IFF_UP) {
891 dev_add_pack(&po->prot_hook);
892 sock_hold(sk);
893 po->running = 1;
894 } else {
895 sk->sk_err = ENETDOWN;
896 if (!sock_flag(sk, SOCK_DEAD))
897 sk->sk_error_report(sk);
898 }
899 } else {
900 dev_add_pack(&po->prot_hook);
901 sock_hold(sk);
902 po->running = 1;
903 }
904
905 out_unlock:
906 spin_unlock(&po->bind_lock);
907 release_sock(sk);
908 return 0;
909 }
910
911 /*
912 * Bind a packet socket to a device
913 */
914
915 static int packet_bind_spkt(struct socket *sock, struct sockaddr *uaddr, int addr_len)
916 {
917 struct sock *sk=sock->sk;
918 char name[15];
919 struct net_device *dev;
920 int err = -ENODEV;
921
922 /*
923 * Check legality
924 */
925
926 if (addr_len != sizeof(struct sockaddr))
927 return -EINVAL;
928 strlcpy(name,uaddr->sa_data,sizeof(name));
929
930 dev = dev_get_by_name(&init_net, name);
931 if (dev) {
932 err = packet_do_bind(sk, dev, pkt_sk(sk)->num);
933 dev_put(dev);
934 }
935 return err;
936 }
937
938 static int packet_bind(struct socket *sock, struct sockaddr *uaddr, int addr_len)
939 {
940 struct sockaddr_ll *sll = (struct sockaddr_ll*)uaddr;
941 struct sock *sk=sock->sk;
942 struct net_device *dev = NULL;
943 int err;
944
945
946 /*
947 * Check legality
948 */
949
950 if (addr_len < sizeof(struct sockaddr_ll))
951 return -EINVAL;
952 if (sll->sll_family != AF_PACKET)
953 return -EINVAL;
954
955 if (sll->sll_ifindex) {
956 err = -ENODEV;
957 dev = dev_get_by_index(&init_net, sll->sll_ifindex);
958 if (dev == NULL)
959 goto out;
960 }
961 err = packet_do_bind(sk, dev, sll->sll_protocol ? : pkt_sk(sk)->num);
962 if (dev)
963 dev_put(dev);
964
965 out:
966 return err;
967 }
968
969 static struct proto packet_proto = {
970 .name = "PACKET",
971 .owner = THIS_MODULE,
972 .obj_size = sizeof(struct packet_sock),
973 };
974
975 /*
976 * Create a packet of type SOCK_PACKET.
977 */
978
979 static int packet_create(struct net *net, struct socket *sock, int protocol)
980 {
981 struct sock *sk;
982 struct packet_sock *po;
983 __be16 proto = (__force __be16)protocol; /* weird, but documented */
984 int err;
985
986 if (net != &init_net)
987 return -EAFNOSUPPORT;
988
989 if (!capable(CAP_NET_RAW))
990 return -EPERM;
991 if (sock->type != SOCK_DGRAM && sock->type != SOCK_RAW &&
992 sock->type != SOCK_PACKET)
993 return -ESOCKTNOSUPPORT;
994
995 sock->state = SS_UNCONNECTED;
996
997 err = -ENOBUFS;
998 sk = sk_alloc(net, PF_PACKET, GFP_KERNEL, &packet_proto, 1);
999 if (sk == NULL)
1000 goto out;
1001
1002 sock->ops = &packet_ops;
1003 if (sock->type == SOCK_PACKET)
1004 sock->ops = &packet_ops_spkt;
1005
1006 sock_init_data(sock, sk);
1007
1008 po = pkt_sk(sk);
1009 sk->sk_family = PF_PACKET;
1010 po->num = proto;
1011
1012 sk->sk_destruct = packet_sock_destruct;
1013 atomic_inc(&packet_socks_nr);
1014
1015 /*
1016 * Attach a protocol block
1017 */
1018
1019 spin_lock_init(&po->bind_lock);
1020 po->prot_hook.func = packet_rcv;
1021
1022 if (sock->type == SOCK_PACKET)
1023 po->prot_hook.func = packet_rcv_spkt;
1024
1025 po->prot_hook.af_packet_priv = sk;
1026
1027 if (proto) {
1028 po->prot_hook.type = proto;
1029 dev_add_pack(&po->prot_hook);
1030 sock_hold(sk);
1031 po->running = 1;
1032 }
1033
1034 write_lock_bh(&packet_sklist_lock);
1035 sk_add_node(sk, &packet_sklist);
1036 write_unlock_bh(&packet_sklist_lock);
1037 return(0);
1038 out:
1039 return err;
1040 }
1041
1042 /*
1043 * Pull a packet from our receive queue and hand it to the user.
1044 * If necessary we block.
1045 */
1046
1047 static int packet_recvmsg(struct kiocb *iocb, struct socket *sock,
1048 struct msghdr *msg, size_t len, int flags)
1049 {
1050 struct sock *sk = sock->sk;
1051 struct sk_buff *skb;
1052 int copied, err;
1053 struct sockaddr_ll *sll;
1054
1055 err = -EINVAL;
1056 if (flags & ~(MSG_PEEK|MSG_DONTWAIT|MSG_TRUNC|MSG_CMSG_COMPAT))
1057 goto out;
1058
1059 #if 0
1060 /* What error should we return now? EUNATTACH? */
1061 if (pkt_sk(sk)->ifindex < 0)
1062 return -ENODEV;
1063 #endif
1064
1065 /*
1066 * Call the generic datagram receiver. This handles all sorts
1067 * of horrible races and re-entrancy so we can forget about it
1068 * in the protocol layers.
1069 *
1070 * Now it will return ENETDOWN, if device have just gone down,
1071 * but then it will block.
1072 */
1073
1074 skb=skb_recv_datagram(sk,flags,flags&MSG_DONTWAIT,&err);
1075
1076 /*
1077 * An error occurred so return it. Because skb_recv_datagram()
1078 * handles the blocking we don't see and worry about blocking
1079 * retries.
1080 */
1081
1082 if (skb == NULL)
1083 goto out;
1084
1085 /*
1086 * If the address length field is there to be filled in, we fill
1087 * it in now.
1088 */
1089
1090 sll = &PACKET_SKB_CB(skb)->sa.ll;
1091 if (sock->type == SOCK_PACKET)
1092 msg->msg_namelen = sizeof(struct sockaddr_pkt);
1093 else
1094 msg->msg_namelen = sll->sll_halen + offsetof(struct sockaddr_ll, sll_addr);
1095
1096 /*
1097 * You lose any data beyond the buffer you gave. If it worries a
1098 * user program they can ask the device for its MTU anyway.
1099 */
1100
1101 copied = skb->len;
1102 if (copied > len)
1103 {
1104 copied=len;
1105 msg->msg_flags|=MSG_TRUNC;
1106 }
1107
1108 err = skb_copy_datagram_iovec(skb, 0, msg->msg_iov, copied);
1109 if (err)
1110 goto out_free;
1111
1112 sock_recv_timestamp(msg, sk, skb);
1113
1114 if (msg->msg_name)
1115 memcpy(msg->msg_name, &PACKET_SKB_CB(skb)->sa,
1116 msg->msg_namelen);
1117
1118 if (pkt_sk(sk)->auxdata) {
1119 struct tpacket_auxdata aux;
1120
1121 aux.tp_status = TP_STATUS_USER;
1122 if (skb->ip_summed == CHECKSUM_PARTIAL)
1123 aux.tp_status |= TP_STATUS_CSUMNOTREADY;
1124 aux.tp_len = PACKET_SKB_CB(skb)->origlen;
1125 aux.tp_snaplen = skb->len;
1126 aux.tp_mac = 0;
1127 aux.tp_net = skb_network_offset(skb);
1128
1129 put_cmsg(msg, SOL_PACKET, PACKET_AUXDATA, sizeof(aux), &aux);
1130 }
1131
1132 /*
1133 * Free or return the buffer as appropriate. Again this
1134 * hides all the races and re-entrancy issues from us.
1135 */
1136 err = (flags&MSG_TRUNC) ? skb->len : copied;
1137
1138 out_free:
1139 skb_free_datagram(sk, skb);
1140 out:
1141 return err;
1142 }
1143
1144 static int packet_getname_spkt(struct socket *sock, struct sockaddr *uaddr,
1145 int *uaddr_len, int peer)
1146 {
1147 struct net_device *dev;
1148 struct sock *sk = sock->sk;
1149
1150 if (peer)
1151 return -EOPNOTSUPP;
1152
1153 uaddr->sa_family = AF_PACKET;
1154 dev = dev_get_by_index(&init_net, pkt_sk(sk)->ifindex);
1155 if (dev) {
1156 strlcpy(uaddr->sa_data, dev->name, 15);
1157 dev_put(dev);
1158 } else
1159 memset(uaddr->sa_data, 0, 14);
1160 *uaddr_len = sizeof(*uaddr);
1161
1162 return 0;
1163 }
1164
1165 static int packet_getname(struct socket *sock, struct sockaddr *uaddr,
1166 int *uaddr_len, int peer)
1167 {
1168 struct net_device *dev;
1169 struct sock *sk = sock->sk;
1170 struct packet_sock *po = pkt_sk(sk);
1171 struct sockaddr_ll *sll = (struct sockaddr_ll*)uaddr;
1172
1173 if (peer)
1174 return -EOPNOTSUPP;
1175
1176 sll->sll_family = AF_PACKET;
1177 sll->sll_ifindex = po->ifindex;
1178 sll->sll_protocol = po->num;
1179 dev = dev_get_by_index(&init_net, po->ifindex);
1180 if (dev) {
1181 sll->sll_hatype = dev->type;
1182 sll->sll_halen = dev->addr_len;
1183 memcpy(sll->sll_addr, dev->dev_addr, dev->addr_len);
1184 dev_put(dev);
1185 } else {
1186 sll->sll_hatype = 0; /* Bad: we have no ARPHRD_UNSPEC */
1187 sll->sll_halen = 0;
1188 }
1189 *uaddr_len = offsetof(struct sockaddr_ll, sll_addr) + sll->sll_halen;
1190
1191 return 0;
1192 }
1193
1194 static void packet_dev_mc(struct net_device *dev, struct packet_mclist *i, int what)
1195 {
1196 switch (i->type) {
1197 case PACKET_MR_MULTICAST:
1198 if (what > 0)
1199 dev_mc_add(dev, i->addr, i->alen, 0);
1200 else
1201 dev_mc_delete(dev, i->addr, i->alen, 0);
1202 break;
1203 case PACKET_MR_PROMISC:
1204 dev_set_promiscuity(dev, what);
1205 break;
1206 case PACKET_MR_ALLMULTI:
1207 dev_set_allmulti(dev, what);
1208 break;
1209 default:;
1210 }
1211 }
1212
1213 static void packet_dev_mclist(struct net_device *dev, struct packet_mclist *i, int what)
1214 {
1215 for ( ; i; i=i->next) {
1216 if (i->ifindex == dev->ifindex)
1217 packet_dev_mc(dev, i, what);
1218 }
1219 }
1220
1221 static int packet_mc_add(struct sock *sk, struct packet_mreq_max *mreq)
1222 {
1223 struct packet_sock *po = pkt_sk(sk);
1224 struct packet_mclist *ml, *i;
1225 struct net_device *dev;
1226 int err;
1227
1228 rtnl_lock();
1229
1230 err = -ENODEV;
1231 dev = __dev_get_by_index(&init_net, mreq->mr_ifindex);
1232 if (!dev)
1233 goto done;
1234
1235 err = -EINVAL;
1236 if (mreq->mr_alen > dev->addr_len)
1237 goto done;
1238
1239 err = -ENOBUFS;
1240 i = kmalloc(sizeof(*i), GFP_KERNEL);
1241 if (i == NULL)
1242 goto done;
1243
1244 err = 0;
1245 for (ml = po->mclist; ml; ml = ml->next) {
1246 if (ml->ifindex == mreq->mr_ifindex &&
1247 ml->type == mreq->mr_type &&
1248 ml->alen == mreq->mr_alen &&
1249 memcmp(ml->addr, mreq->mr_address, ml->alen) == 0) {
1250 ml->count++;
1251 /* Free the new element ... */
1252 kfree(i);
1253 goto done;
1254 }
1255 }
1256
1257 i->type = mreq->mr_type;
1258 i->ifindex = mreq->mr_ifindex;
1259 i->alen = mreq->mr_alen;
1260 memcpy(i->addr, mreq->mr_address, i->alen);
1261 i->count = 1;
1262 i->next = po->mclist;
1263 po->mclist = i;
1264 packet_dev_mc(dev, i, +1);
1265
1266 done:
1267 rtnl_unlock();
1268 return err;
1269 }
1270
1271 static int packet_mc_drop(struct sock *sk, struct packet_mreq_max *mreq)
1272 {
1273 struct packet_mclist *ml, **mlp;
1274
1275 rtnl_lock();
1276
1277 for (mlp = &pkt_sk(sk)->mclist; (ml = *mlp) != NULL; mlp = &ml->next) {
1278 if (ml->ifindex == mreq->mr_ifindex &&
1279 ml->type == mreq->mr_type &&
1280 ml->alen == mreq->mr_alen &&
1281 memcmp(ml->addr, mreq->mr_address, ml->alen) == 0) {
1282 if (--ml->count == 0) {
1283 struct net_device *dev;
1284 *mlp = ml->next;
1285 dev = dev_get_by_index(&init_net, ml->ifindex);
1286 if (dev) {
1287 packet_dev_mc(dev, ml, -1);
1288 dev_put(dev);
1289 }
1290 kfree(ml);
1291 }
1292 rtnl_unlock();
1293 return 0;
1294 }
1295 }
1296 rtnl_unlock();
1297 return -EADDRNOTAVAIL;
1298 }
1299
1300 static void packet_flush_mclist(struct sock *sk)
1301 {
1302 struct packet_sock *po = pkt_sk(sk);
1303 struct packet_mclist *ml;
1304
1305 if (!po->mclist)
1306 return;
1307
1308 rtnl_lock();
1309 while ((ml = po->mclist) != NULL) {
1310 struct net_device *dev;
1311
1312 po->mclist = ml->next;
1313 if ((dev = dev_get_by_index(&init_net, ml->ifindex)) != NULL) {
1314 packet_dev_mc(dev, ml, -1);
1315 dev_put(dev);
1316 }
1317 kfree(ml);
1318 }
1319 rtnl_unlock();
1320 }
1321
1322 static int
1323 packet_setsockopt(struct socket *sock, int level, int optname, char __user *optval, int optlen)
1324 {
1325 struct sock *sk = sock->sk;
1326 struct packet_sock *po = pkt_sk(sk);
1327 int ret;
1328
1329 if (level != SOL_PACKET)
1330 return -ENOPROTOOPT;
1331
1332 switch(optname) {
1333 case PACKET_ADD_MEMBERSHIP:
1334 case PACKET_DROP_MEMBERSHIP:
1335 {
1336 struct packet_mreq_max mreq;
1337 int len = optlen;
1338 memset(&mreq, 0, sizeof(mreq));
1339 if (len < sizeof(struct packet_mreq))
1340 return -EINVAL;
1341 if (len > sizeof(mreq))
1342 len = sizeof(mreq);
1343 if (copy_from_user(&mreq,optval,len))
1344 return -EFAULT;
1345 if (len < (mreq.mr_alen + offsetof(struct packet_mreq, mr_address)))
1346 return -EINVAL;
1347 if (optname == PACKET_ADD_MEMBERSHIP)
1348 ret = packet_mc_add(sk, &mreq);
1349 else
1350 ret = packet_mc_drop(sk, &mreq);
1351 return ret;
1352 }
1353
1354 #ifdef CONFIG_PACKET_MMAP
1355 case PACKET_RX_RING:
1356 {
1357 struct tpacket_req req;
1358
1359 if (optlen<sizeof(req))
1360 return -EINVAL;
1361 if (copy_from_user(&req,optval,sizeof(req)))
1362 return -EFAULT;
1363 return packet_set_ring(sk, &req, 0);
1364 }
1365 case PACKET_COPY_THRESH:
1366 {
1367 int val;
1368
1369 if (optlen!=sizeof(val))
1370 return -EINVAL;
1371 if (copy_from_user(&val,optval,sizeof(val)))
1372 return -EFAULT;
1373
1374 pkt_sk(sk)->copy_thresh = val;
1375 return 0;
1376 }
1377 #endif
1378 case PACKET_AUXDATA:
1379 {
1380 int val;
1381
1382 if (optlen < sizeof(val))
1383 return -EINVAL;
1384 if (copy_from_user(&val, optval, sizeof(val)))
1385 return -EFAULT;
1386
1387 po->auxdata = !!val;
1388 return 0;
1389 }
1390 case PACKET_ORIGDEV:
1391 {
1392 int val;
1393
1394 if (optlen < sizeof(val))
1395 return -EINVAL;
1396 if (copy_from_user(&val, optval, sizeof(val)))
1397 return -EFAULT;
1398
1399 po->origdev = !!val;
1400 return 0;
1401 }
1402 default:
1403 return -ENOPROTOOPT;
1404 }
1405 }
1406
1407 static int packet_getsockopt(struct socket *sock, int level, int optname,
1408 char __user *optval, int __user *optlen)
1409 {
1410 int len;
1411 int val;
1412 struct sock *sk = sock->sk;
1413 struct packet_sock *po = pkt_sk(sk);
1414 void *data;
1415 struct tpacket_stats st;
1416
1417 if (level != SOL_PACKET)
1418 return -ENOPROTOOPT;
1419
1420 if (get_user(len, optlen))
1421 return -EFAULT;
1422
1423 if (len < 0)
1424 return -EINVAL;
1425
1426 switch(optname) {
1427 case PACKET_STATISTICS:
1428 if (len > sizeof(struct tpacket_stats))
1429 len = sizeof(struct tpacket_stats);
1430 spin_lock_bh(&sk->sk_receive_queue.lock);
1431 st = po->stats;
1432 memset(&po->stats, 0, sizeof(st));
1433 spin_unlock_bh(&sk->sk_receive_queue.lock);
1434 st.tp_packets += st.tp_drops;
1435
1436 data = &st;
1437 break;
1438 case PACKET_AUXDATA:
1439 if (len > sizeof(int))
1440 len = sizeof(int);
1441 val = po->auxdata;
1442
1443 data = &val;
1444 break;
1445 case PACKET_ORIGDEV:
1446 if (len > sizeof(int))
1447 len = sizeof(int);
1448 val = po->origdev;
1449
1450 data = &val;
1451 break;
1452 default:
1453 return -ENOPROTOOPT;
1454 }
1455
1456 if (put_user(len, optlen))
1457 return -EFAULT;
1458 if (copy_to_user(optval, data, len))
1459 return -EFAULT;
1460 return 0;
1461 }
1462
1463
1464 static int packet_notifier(struct notifier_block *this, unsigned long msg, void *data)
1465 {
1466 struct sock *sk;
1467 struct hlist_node *node;
1468 struct net_device *dev = data;
1469
1470 if (dev->nd_net != &init_net)
1471 return NOTIFY_DONE;
1472
1473 read_lock(&packet_sklist_lock);
1474 sk_for_each(sk, node, &packet_sklist) {
1475 struct packet_sock *po = pkt_sk(sk);
1476
1477 switch (msg) {
1478 case NETDEV_UNREGISTER:
1479 if (po->mclist)
1480 packet_dev_mclist(dev, po->mclist, -1);
1481 /* fallthrough */
1482
1483 case NETDEV_DOWN:
1484 if (dev->ifindex == po->ifindex) {
1485 spin_lock(&po->bind_lock);
1486 if (po->running) {
1487 __dev_remove_pack(&po->prot_hook);
1488 __sock_put(sk);
1489 po->running = 0;
1490 sk->sk_err = ENETDOWN;
1491 if (!sock_flag(sk, SOCK_DEAD))
1492 sk->sk_error_report(sk);
1493 }
1494 if (msg == NETDEV_UNREGISTER) {
1495 po->ifindex = -1;
1496 po->prot_hook.dev = NULL;
1497 }
1498 spin_unlock(&po->bind_lock);
1499 }
1500 break;
1501 case NETDEV_UP:
1502 spin_lock(&po->bind_lock);
1503 if (dev->ifindex == po->ifindex && po->num &&
1504 !po->running) {
1505 dev_add_pack(&po->prot_hook);
1506 sock_hold(sk);
1507 po->running = 1;
1508 }
1509 spin_unlock(&po->bind_lock);
1510 break;
1511 }
1512 }
1513 read_unlock(&packet_sklist_lock);
1514 return NOTIFY_DONE;
1515 }
1516
1517
1518 static int packet_ioctl(struct socket *sock, unsigned int cmd,
1519 unsigned long arg)
1520 {
1521 struct sock *sk = sock->sk;
1522
1523 switch(cmd) {
1524 case SIOCOUTQ:
1525 {
1526 int amount = atomic_read(&sk->sk_wmem_alloc);
1527 return put_user(amount, (int __user *)arg);
1528 }
1529 case SIOCINQ:
1530 {
1531 struct sk_buff *skb;
1532 int amount = 0;
1533
1534 spin_lock_bh(&sk->sk_receive_queue.lock);
1535 skb = skb_peek(&sk->sk_receive_queue);
1536 if (skb)
1537 amount = skb->len;
1538 spin_unlock_bh(&sk->sk_receive_queue.lock);
1539 return put_user(amount, (int __user *)arg);
1540 }
1541 case SIOCGSTAMP:
1542 return sock_get_timestamp(sk, (struct timeval __user *)arg);
1543 case SIOCGSTAMPNS:
1544 return sock_get_timestampns(sk, (struct timespec __user *)arg);
1545
1546 #ifdef CONFIG_INET
1547 case SIOCADDRT:
1548 case SIOCDELRT:
1549 case SIOCDARP:
1550 case SIOCGARP:
1551 case SIOCSARP:
1552 case SIOCGIFADDR:
1553 case SIOCSIFADDR:
1554 case SIOCGIFBRDADDR:
1555 case SIOCSIFBRDADDR:
1556 case SIOCGIFNETMASK:
1557 case SIOCSIFNETMASK:
1558 case SIOCGIFDSTADDR:
1559 case SIOCSIFDSTADDR:
1560 case SIOCSIFFLAGS:
1561 return inet_dgram_ops.ioctl(sock, cmd, arg);
1562 #endif
1563
1564 default:
1565 return -ENOIOCTLCMD;
1566 }
1567 return 0;
1568 }
1569
1570 #ifndef CONFIG_PACKET_MMAP
1571 #define packet_mmap sock_no_mmap
1572 #define packet_poll datagram_poll
1573 #else
1574
1575 static unsigned int packet_poll(struct file * file, struct socket *sock,
1576 poll_table *wait)
1577 {
1578 struct sock *sk = sock->sk;
1579 struct packet_sock *po = pkt_sk(sk);
1580 unsigned int mask = datagram_poll(file, sock, wait);
1581
1582 spin_lock_bh(&sk->sk_receive_queue.lock);
1583 if (po->pg_vec) {
1584 unsigned last = po->head ? po->head-1 : po->frame_max;
1585 struct tpacket_hdr *h;
1586
1587 h = packet_lookup_frame(po, last);
1588
1589 if (h->tp_status)
1590 mask |= POLLIN | POLLRDNORM;
1591 }
1592 spin_unlock_bh(&sk->sk_receive_queue.lock);
1593 return mask;
1594 }
1595
1596
1597 /* Dirty? Well, I still did not learn better way to account
1598 * for user mmaps.
1599 */
1600
1601 static void packet_mm_open(struct vm_area_struct *vma)
1602 {
1603 struct file *file = vma->vm_file;
1604 struct socket * sock = file->private_data;
1605 struct sock *sk = sock->sk;
1606
1607 if (sk)
1608 atomic_inc(&pkt_sk(sk)->mapped);
1609 }
1610
1611 static void packet_mm_close(struct vm_area_struct *vma)
1612 {
1613 struct file *file = vma->vm_file;
1614 struct socket * sock = file->private_data;
1615 struct sock *sk = sock->sk;
1616
1617 if (sk)
1618 atomic_dec(&pkt_sk(sk)->mapped);
1619 }
1620
1621 static struct vm_operations_struct packet_mmap_ops = {
1622 .open = packet_mm_open,
1623 .close =packet_mm_close,
1624 };
1625
1626 static inline struct page *pg_vec_endpage(char *one_pg_vec, unsigned int order)
1627 {
1628 return virt_to_page(one_pg_vec + (PAGE_SIZE << order) - 1);
1629 }
1630
1631 static void free_pg_vec(char **pg_vec, unsigned int order, unsigned int len)
1632 {
1633 int i;
1634
1635 for (i = 0; i < len; i++) {
1636 if (likely(pg_vec[i]))
1637 free_pages((unsigned long) pg_vec[i], order);
1638 }
1639 kfree(pg_vec);
1640 }
1641
1642 static inline char *alloc_one_pg_vec_page(unsigned long order)
1643 {
1644 return (char *) __get_free_pages(GFP_KERNEL | __GFP_COMP | __GFP_ZERO,
1645 order);
1646 }
1647
1648 static char **alloc_pg_vec(struct tpacket_req *req, int order)
1649 {
1650 unsigned int block_nr = req->tp_block_nr;
1651 char **pg_vec;
1652 int i;
1653
1654 pg_vec = kzalloc(block_nr * sizeof(char *), GFP_KERNEL);
1655 if (unlikely(!pg_vec))
1656 goto out;
1657
1658 for (i = 0; i < block_nr; i++) {
1659 pg_vec[i] = alloc_one_pg_vec_page(order);
1660 if (unlikely(!pg_vec[i]))
1661 goto out_free_pgvec;
1662 }
1663
1664 out:
1665 return pg_vec;
1666
1667 out_free_pgvec:
1668 free_pg_vec(pg_vec, order, block_nr);
1669 pg_vec = NULL;
1670 goto out;
1671 }
1672
1673 static int packet_set_ring(struct sock *sk, struct tpacket_req *req, int closing)
1674 {
1675 char **pg_vec = NULL;
1676 struct packet_sock *po = pkt_sk(sk);
1677 int was_running, order = 0;
1678 __be16 num;
1679 int err = 0;
1680
1681 if (req->tp_block_nr) {
1682 int i, l;
1683
1684 /* Sanity tests and some calculations */
1685
1686 if (unlikely(po->pg_vec))
1687 return -EBUSY;
1688
1689 if (unlikely((int)req->tp_block_size <= 0))
1690 return -EINVAL;
1691 if (unlikely(req->tp_block_size & (PAGE_SIZE - 1)))
1692 return -EINVAL;
1693 if (unlikely(req->tp_frame_size < TPACKET_HDRLEN))
1694 return -EINVAL;
1695 if (unlikely(req->tp_frame_size & (TPACKET_ALIGNMENT - 1)))
1696 return -EINVAL;
1697
1698 po->frames_per_block = req->tp_block_size/req->tp_frame_size;
1699 if (unlikely(po->frames_per_block <= 0))
1700 return -EINVAL;
1701 if (unlikely((po->frames_per_block * req->tp_block_nr) !=
1702 req->tp_frame_nr))
1703 return -EINVAL;
1704
1705 err = -ENOMEM;
1706 order = get_order(req->tp_block_size);
1707 pg_vec = alloc_pg_vec(req, order);
1708 if (unlikely(!pg_vec))
1709 goto out;
1710
1711 l = 0;
1712 for (i = 0; i < req->tp_block_nr; i++) {
1713 char *ptr = pg_vec[i];
1714 struct tpacket_hdr *header;
1715 int k;
1716
1717 for (k = 0; k < po->frames_per_block; k++) {
1718 header = (struct tpacket_hdr *) ptr;
1719 header->tp_status = TP_STATUS_KERNEL;
1720 ptr += req->tp_frame_size;
1721 }
1722 }
1723 /* Done */
1724 } else {
1725 if (unlikely(req->tp_frame_nr))
1726 return -EINVAL;
1727 }
1728
1729 lock_sock(sk);
1730
1731 /* Detach socket from network */
1732 spin_lock(&po->bind_lock);
1733 was_running = po->running;
1734 num = po->num;
1735 if (was_running) {
1736 __dev_remove_pack(&po->prot_hook);
1737 po->num = 0;
1738 po->running = 0;
1739 __sock_put(sk);
1740 }
1741 spin_unlock(&po->bind_lock);
1742
1743 synchronize_net();
1744
1745 err = -EBUSY;
1746 if (closing || atomic_read(&po->mapped) == 0) {
1747 err = 0;
1748 #define XC(a, b) ({ __typeof__ ((a)) __t; __t = (a); (a) = (b); __t; })
1749
1750 spin_lock_bh(&sk->sk_receive_queue.lock);
1751 pg_vec = XC(po->pg_vec, pg_vec);
1752 po->frame_max = (req->tp_frame_nr - 1);
1753 po->head = 0;
1754 po->frame_size = req->tp_frame_size;
1755 spin_unlock_bh(&sk->sk_receive_queue.lock);
1756
1757 order = XC(po->pg_vec_order, order);
1758 req->tp_block_nr = XC(po->pg_vec_len, req->tp_block_nr);
1759
1760 po->pg_vec_pages = req->tp_block_size/PAGE_SIZE;
1761 po->prot_hook.func = po->pg_vec ? tpacket_rcv : packet_rcv;
1762 skb_queue_purge(&sk->sk_receive_queue);
1763 #undef XC
1764 if (atomic_read(&po->mapped))
1765 printk(KERN_DEBUG "packet_mmap: vma is busy: %d\n", atomic_read(&po->mapped));
1766 }
1767
1768 spin_lock(&po->bind_lock);
1769 if (was_running && !po->running) {
1770 sock_hold(sk);
1771 po->running = 1;
1772 po->num = num;
1773 dev_add_pack(&po->prot_hook);
1774 }
1775 spin_unlock(&po->bind_lock);
1776
1777 release_sock(sk);
1778
1779 if (pg_vec)
1780 free_pg_vec(pg_vec, order, req->tp_block_nr);
1781 out:
1782 return err;
1783 }
1784
1785 static int packet_mmap(struct file *file, struct socket *sock, struct vm_area_struct *vma)
1786 {
1787 struct sock *sk = sock->sk;
1788 struct packet_sock *po = pkt_sk(sk);
1789 unsigned long size;
1790 unsigned long start;
1791 int err = -EINVAL;
1792 int i;
1793
1794 if (vma->vm_pgoff)
1795 return -EINVAL;
1796
1797 size = vma->vm_end - vma->vm_start;
1798
1799 lock_sock(sk);
1800 if (po->pg_vec == NULL)
1801 goto out;
1802 if (size != po->pg_vec_len*po->pg_vec_pages*PAGE_SIZE)
1803 goto out;
1804
1805 start = vma->vm_start;
1806 for (i = 0; i < po->pg_vec_len; i++) {
1807 struct page *page = virt_to_page(po->pg_vec[i]);
1808 int pg_num;
1809
1810 for (pg_num = 0; pg_num < po->pg_vec_pages; pg_num++, page++) {
1811 err = vm_insert_page(vma, start, page);
1812 if (unlikely(err))
1813 goto out;
1814 start += PAGE_SIZE;
1815 }
1816 }
1817 atomic_inc(&po->mapped);
1818 vma->vm_ops = &packet_mmap_ops;
1819 err = 0;
1820
1821 out:
1822 release_sock(sk);
1823 return err;
1824 }
1825 #endif
1826
1827
1828 static const struct proto_ops packet_ops_spkt = {
1829 .family = PF_PACKET,
1830 .owner = THIS_MODULE,
1831 .release = packet_release,
1832 .bind = packet_bind_spkt,
1833 .connect = sock_no_connect,
1834 .socketpair = sock_no_socketpair,
1835 .accept = sock_no_accept,
1836 .getname = packet_getname_spkt,
1837 .poll = datagram_poll,
1838 .ioctl = packet_ioctl,
1839 .listen = sock_no_listen,
1840 .shutdown = sock_no_shutdown,
1841 .setsockopt = sock_no_setsockopt,
1842 .getsockopt = sock_no_getsockopt,
1843 .sendmsg = packet_sendmsg_spkt,
1844 .recvmsg = packet_recvmsg,
1845 .mmap = sock_no_mmap,
1846 .sendpage = sock_no_sendpage,
1847 };
1848
1849 static const struct proto_ops packet_ops = {
1850 .family = PF_PACKET,
1851 .owner = THIS_MODULE,
1852 .release = packet_release,
1853 .bind = packet_bind,
1854 .connect = sock_no_connect,
1855 .socketpair = sock_no_socketpair,
1856 .accept = sock_no_accept,
1857 .getname = packet_getname,
1858 .poll = packet_poll,
1859 .ioctl = packet_ioctl,
1860 .listen = sock_no_listen,
1861 .shutdown = sock_no_shutdown,
1862 .setsockopt = packet_setsockopt,
1863 .getsockopt = packet_getsockopt,
1864 .sendmsg = packet_sendmsg,
1865 .recvmsg = packet_recvmsg,
1866 .mmap = packet_mmap,
1867 .sendpage = sock_no_sendpage,
1868 };
1869
1870 static struct net_proto_family packet_family_ops = {
1871 .family = PF_PACKET,
1872 .create = packet_create,
1873 .owner = THIS_MODULE,
1874 };
1875
1876 static struct notifier_block packet_netdev_notifier = {
1877 .notifier_call =packet_notifier,
1878 };
1879
1880 #ifdef CONFIG_PROC_FS
1881 static inline struct sock *packet_seq_idx(loff_t off)
1882 {
1883 struct sock *s;
1884 struct hlist_node *node;
1885
1886 sk_for_each(s, node, &packet_sklist) {
1887 if (!off--)
1888 return s;
1889 }
1890 return NULL;
1891 }
1892
1893 static void *packet_seq_start(struct seq_file *seq, loff_t *pos)
1894 {
1895 read_lock(&packet_sklist_lock);
1896 return *pos ? packet_seq_idx(*pos - 1) : SEQ_START_TOKEN;
1897 }
1898
1899 static void *packet_seq_next(struct seq_file *seq, void *v, loff_t *pos)
1900 {
1901 ++*pos;
1902 return (v == SEQ_START_TOKEN)
1903 ? sk_head(&packet_sklist)
1904 : sk_next((struct sock*)v) ;
1905 }
1906
1907 static void packet_seq_stop(struct seq_file *seq, void *v)
1908 {
1909 read_unlock(&packet_sklist_lock);
1910 }
1911
1912 static int packet_seq_show(struct seq_file *seq, void *v)
1913 {
1914 if (v == SEQ_START_TOKEN)
1915 seq_puts(seq, "sk RefCnt Type Proto Iface R Rmem User Inode\n");
1916 else {
1917 struct sock *s = v;
1918 const struct packet_sock *po = pkt_sk(s);
1919
1920 seq_printf(seq,
1921 "%p %-6d %-4d %04x %-5d %1d %-6u %-6u %-6lu\n",
1922 s,
1923 atomic_read(&s->sk_refcnt),
1924 s->sk_type,
1925 ntohs(po->num),
1926 po->ifindex,
1927 po->running,
1928 atomic_read(&s->sk_rmem_alloc),
1929 sock_i_uid(s),
1930 sock_i_ino(s) );
1931 }
1932
1933 return 0;
1934 }
1935
1936 static const struct seq_operations packet_seq_ops = {
1937 .start = packet_seq_start,
1938 .next = packet_seq_next,
1939 .stop = packet_seq_stop,
1940 .show = packet_seq_show,
1941 };
1942
1943 static int packet_seq_open(struct inode *inode, struct file *file)
1944 {
1945 return seq_open(file, &packet_seq_ops);
1946 }
1947
1948 static const struct file_operations packet_seq_fops = {
1949 .owner = THIS_MODULE,
1950 .open = packet_seq_open,
1951 .read = seq_read,
1952 .llseek = seq_lseek,
1953 .release = seq_release,
1954 };
1955
1956 #endif
1957
1958 static void __exit packet_exit(void)
1959 {
1960 proc_net_remove(&init_net, "packet");
1961 unregister_netdevice_notifier(&packet_netdev_notifier);
1962 sock_unregister(PF_PACKET);
1963 proto_unregister(&packet_proto);
1964 }
1965
1966 static int __init packet_init(void)
1967 {
1968 int rc = proto_register(&packet_proto, 0);
1969
1970 if (rc != 0)
1971 goto out;
1972
1973 sock_register(&packet_family_ops);
1974 register_netdevice_notifier(&packet_netdev_notifier);
1975 proc_net_fops_create(&init_net, "packet", 0, &packet_seq_fops);
1976 out:
1977 return rc;
1978 }
1979
1980 module_init(packet_init);
1981 module_exit(packet_exit);
1982 MODULE_LICENSE("GPL");
1983 MODULE_ALIAS_NETPROTO(PF_PACKET);
kernel-based virtual machine