Merge git://git.kernel.org/pub/scm/linux/kernel/git/mingo/linux-2.6-sched
[linux-2.6/linux-acpi-2.6/ibm-acpi-2.6.git] / net / packet / af_packet.c
blobb8b827c7062d3a7a50f6032f76cdd61bfdf634dd
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.
6 * PACKET - implements raw packet sockets.
8 * Version: $Id: af_packet.c,v 1.61 2002/02/08 03:57:19 davem Exp $
10 * Authors: Ross Biro
11 * Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG>
12 * Alan Cox, <gw4pts@gw4pts.ampr.org>
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.
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.
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>
83 #ifdef CONFIG_INET
84 #include <net/inet_common.h>
85 #endif
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.
98 On receive:
99 -----------
101 Incoming, dev->hard_header!=NULL
102 mac_header -> ll header
103 data -> data
105 Outgoing, dev->hard_header!=NULL
106 mac_header -> ll header
107 data -> ll header
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
115 Outgoing, dev->hard_header==NULL
116 mac_header -> data. ll header is still not built!
117 data -> data
119 Resume
120 If dev->hard_header==NULL we are unlikely to restore sensible ll header.
123 On transmit:
124 ------------
126 dev->hard_header != NULL
127 mac_header -> ll header
128 data -> ll header
130 dev->hard_header == NULL (ll header is added by device, we cannot control it)
131 mac_header -> data
132 data -> data
134 We should set nh.raw on output to correct posistion,
135 packet classifier depends on it.
138 /* Private packet socket structures. */
140 struct packet_mclist
142 struct packet_mclist *next;
143 int ifindex;
144 int count;
145 unsigned short type;
146 unsigned short alen;
147 unsigned char addr[MAX_ADDR_LEN];
149 /* identical to struct packet_mreq except it has
150 * a longer address field.
152 struct packet_mreq_max
154 int mr_ifindex;
155 unsigned short mr_type;
156 unsigned short mr_alen;
157 unsigned char mr_address[MAX_ADDR_LEN];
160 #ifdef CONFIG_PACKET_MMAP
161 static int packet_set_ring(struct sock *sk, struct tpacket_req *req, int closing);
162 #endif
164 static void packet_flush_mclist(struct sock *sk);
166 struct packet_sock {
167 /* struct sock has to be the first member of packet_sock */
168 struct sock sk;
169 struct tpacket_stats stats;
170 #ifdef CONFIG_PACKET_MMAP
171 char * *pg_vec;
172 unsigned int head;
173 unsigned int frames_per_block;
174 unsigned int frame_size;
175 unsigned int frame_max;
176 int copy_thresh;
177 #endif
178 struct packet_type prot_hook;
179 spinlock_t bind_lock;
180 unsigned int running:1, /* prot_hook is attached*/
181 auxdata:1,
182 origdev:1;
183 int ifindex; /* bound device */
184 __be16 num;
185 struct packet_mclist *mclist;
186 #ifdef CONFIG_PACKET_MMAP
187 atomic_t mapped;
188 unsigned int pg_vec_order;
189 unsigned int pg_vec_pages;
190 unsigned int pg_vec_len;
191 #endif
194 struct packet_skb_cb {
195 unsigned int origlen;
196 union {
197 struct sockaddr_pkt pkt;
198 struct sockaddr_ll ll;
199 } sa;
202 #define PACKET_SKB_CB(__skb) ((struct packet_skb_cb *)((__skb)->cb))
204 #ifdef CONFIG_PACKET_MMAP
206 static inline struct tpacket_hdr *packet_lookup_frame(struct packet_sock *po, unsigned int position)
208 unsigned int pg_vec_pos, frame_offset;
210 pg_vec_pos = position / po->frames_per_block;
211 frame_offset = position % po->frames_per_block;
213 return (struct tpacket_hdr *)(po->pg_vec[pg_vec_pos] + (frame_offset * po->frame_size));
215 #endif
217 static inline struct packet_sock *pkt_sk(struct sock *sk)
219 return (struct packet_sock *)sk;
222 static void packet_sock_destruct(struct sock *sk)
224 BUG_TRAP(!atomic_read(&sk->sk_rmem_alloc));
225 BUG_TRAP(!atomic_read(&sk->sk_wmem_alloc));
227 if (!sock_flag(sk, SOCK_DEAD)) {
228 printk("Attempt to release alive packet socket: %p\n", sk);
229 return;
232 sk_refcnt_debug_dec(sk);
236 static const struct proto_ops packet_ops;
238 static const struct proto_ops packet_ops_spkt;
240 static int packet_rcv_spkt(struct sk_buff *skb, struct net_device *dev, struct packet_type *pt, struct net_device *orig_dev)
242 struct sock *sk;
243 struct sockaddr_pkt *spkt;
246 * When we registered the protocol we saved the socket in the data
247 * field for just this event.
250 sk = pt->af_packet_priv;
253 * Yank back the headers [hope the device set this
254 * right or kerboom...]
256 * Incoming packets have ll header pulled,
257 * push it back.
259 * For outgoing ones skb->data == skb_mac_header(skb)
260 * so that this procedure is noop.
263 if (skb->pkt_type == PACKET_LOOPBACK)
264 goto out;
266 if (dev->nd_net != sk->sk_net)
267 goto out;
269 if ((skb = skb_share_check(skb, GFP_ATOMIC)) == NULL)
270 goto oom;
272 /* drop any routing info */
273 dst_release(skb->dst);
274 skb->dst = NULL;
276 /* drop conntrack reference */
277 nf_reset(skb);
279 spkt = &PACKET_SKB_CB(skb)->sa.pkt;
281 skb_push(skb, skb->data - skb_mac_header(skb));
284 * The SOCK_PACKET socket receives _all_ frames.
287 spkt->spkt_family = dev->type;
288 strlcpy(spkt->spkt_device, dev->name, sizeof(spkt->spkt_device));
289 spkt->spkt_protocol = skb->protocol;
292 * Charge the memory to the socket. This is done specifically
293 * to prevent sockets using all the memory up.
296 if (sock_queue_rcv_skb(sk,skb) == 0)
297 return 0;
299 out:
300 kfree_skb(skb);
301 oom:
302 return 0;
307 * Output a raw packet to a device layer. This bypasses all the other
308 * protocol layers and you must therefore supply it with a complete frame
311 static int packet_sendmsg_spkt(struct kiocb *iocb, struct socket *sock,
312 struct msghdr *msg, size_t len)
314 struct sock *sk = sock->sk;
315 struct sockaddr_pkt *saddr=(struct sockaddr_pkt *)msg->msg_name;
316 struct sk_buff *skb;
317 struct net_device *dev;
318 __be16 proto=0;
319 int err;
322 * Get and verify the address.
325 if (saddr)
327 if (msg->msg_namelen < sizeof(struct sockaddr))
328 return(-EINVAL);
329 if (msg->msg_namelen==sizeof(struct sockaddr_pkt))
330 proto=saddr->spkt_protocol;
332 else
333 return(-ENOTCONN); /* SOCK_PACKET must be sent giving an address */
336 * Find the device first to size check it
339 saddr->spkt_device[13] = 0;
340 dev = dev_get_by_name(sk->sk_net, saddr->spkt_device);
341 err = -ENODEV;
342 if (dev == NULL)
343 goto out_unlock;
345 err = -ENETDOWN;
346 if (!(dev->flags & IFF_UP))
347 goto out_unlock;
350 * You may not queue a frame bigger than the mtu. This is the lowest level
351 * raw protocol and you must do your own fragmentation at this level.
354 err = -EMSGSIZE;
355 if (len > dev->mtu + dev->hard_header_len)
356 goto out_unlock;
358 err = -ENOBUFS;
359 skb = sock_wmalloc(sk, len + LL_RESERVED_SPACE(dev), 0, GFP_KERNEL);
362 * If the write buffer is full, then tough. At this level the user gets to
363 * deal with the problem - do your own algorithmic backoffs. That's far
364 * more flexible.
367 if (skb == NULL)
368 goto out_unlock;
371 * Fill it in
374 /* FIXME: Save some space for broken drivers that write a
375 * hard header at transmission time by themselves. PPP is the
376 * notable one here. This should really be fixed at the driver level.
378 skb_reserve(skb, LL_RESERVED_SPACE(dev));
379 skb_reset_network_header(skb);
381 /* Try to align data part correctly */
382 if (dev->header_ops) {
383 skb->data -= dev->hard_header_len;
384 skb->tail -= dev->hard_header_len;
385 if (len < dev->hard_header_len)
386 skb_reset_network_header(skb);
389 /* Returns -EFAULT on error */
390 err = memcpy_fromiovec(skb_put(skb,len), msg->msg_iov, len);
391 skb->protocol = proto;
392 skb->dev = dev;
393 skb->priority = sk->sk_priority;
394 if (err)
395 goto out_free;
398 * Now send it
401 dev_queue_xmit(skb);
402 dev_put(dev);
403 return(len);
405 out_free:
406 kfree_skb(skb);
407 out_unlock:
408 if (dev)
409 dev_put(dev);
410 return err;
413 static inline unsigned int run_filter(struct sk_buff *skb, struct sock *sk,
414 unsigned int res)
416 struct sk_filter *filter;
418 rcu_read_lock_bh();
419 filter = rcu_dereference(sk->sk_filter);
420 if (filter != NULL)
421 res = sk_run_filter(skb, filter->insns, filter->len);
422 rcu_read_unlock_bh();
424 return res;
428 This function makes lazy skb cloning in hope that most of packets
429 are discarded by BPF.
431 Note tricky part: we DO mangle shared skb! skb->data, skb->len
432 and skb->cb are mangled. It works because (and until) packets
433 falling here are owned by current CPU. Output packets are cloned
434 by dev_queue_xmit_nit(), input packets are processed by net_bh
435 sequencially, so that if we return skb to original state on exit,
436 we will not harm anyone.
439 static int packet_rcv(struct sk_buff *skb, struct net_device *dev, struct packet_type *pt, struct net_device *orig_dev)
441 struct sock *sk;
442 struct sockaddr_ll *sll;
443 struct packet_sock *po;
444 u8 * skb_head = skb->data;
445 int skb_len = skb->len;
446 unsigned int snaplen, res;
448 if (skb->pkt_type == PACKET_LOOPBACK)
449 goto drop;
451 sk = pt->af_packet_priv;
452 po = pkt_sk(sk);
454 if (dev->nd_net != sk->sk_net)
455 goto drop;
457 skb->dev = dev;
459 if (dev->header_ops) {
460 /* The device has an explicit notion of ll header,
461 exported to higher levels.
463 Otherwise, the device hides datails of it frame
464 structure, so that corresponding packet head
465 never delivered to user.
467 if (sk->sk_type != SOCK_DGRAM)
468 skb_push(skb, skb->data - skb_mac_header(skb));
469 else if (skb->pkt_type == PACKET_OUTGOING) {
470 /* Special case: outgoing packets have ll header at head */
471 skb_pull(skb, skb_network_offset(skb));
475 snaplen = skb->len;
477 res = run_filter(skb, sk, snaplen);
478 if (!res)
479 goto drop_n_restore;
480 if (snaplen > res)
481 snaplen = res;
483 if (atomic_read(&sk->sk_rmem_alloc) + skb->truesize >=
484 (unsigned)sk->sk_rcvbuf)
485 goto drop_n_acct;
487 if (skb_shared(skb)) {
488 struct sk_buff *nskb = skb_clone(skb, GFP_ATOMIC);
489 if (nskb == NULL)
490 goto drop_n_acct;
492 if (skb_head != skb->data) {
493 skb->data = skb_head;
494 skb->len = skb_len;
496 kfree_skb(skb);
497 skb = nskb;
500 BUILD_BUG_ON(sizeof(*PACKET_SKB_CB(skb)) + MAX_ADDR_LEN - 8 >
501 sizeof(skb->cb));
503 sll = &PACKET_SKB_CB(skb)->sa.ll;
504 sll->sll_family = AF_PACKET;
505 sll->sll_hatype = dev->type;
506 sll->sll_protocol = skb->protocol;
507 sll->sll_pkttype = skb->pkt_type;
508 if (unlikely(po->origdev))
509 sll->sll_ifindex = orig_dev->ifindex;
510 else
511 sll->sll_ifindex = dev->ifindex;
513 sll->sll_halen = dev_parse_header(skb, sll->sll_addr);
515 PACKET_SKB_CB(skb)->origlen = skb->len;
517 if (pskb_trim(skb, snaplen))
518 goto drop_n_acct;
520 skb_set_owner_r(skb, sk);
521 skb->dev = NULL;
522 dst_release(skb->dst);
523 skb->dst = NULL;
525 /* drop conntrack reference */
526 nf_reset(skb);
528 spin_lock(&sk->sk_receive_queue.lock);
529 po->stats.tp_packets++;
530 __skb_queue_tail(&sk->sk_receive_queue, skb);
531 spin_unlock(&sk->sk_receive_queue.lock);
532 sk->sk_data_ready(sk, skb->len);
533 return 0;
535 drop_n_acct:
536 spin_lock(&sk->sk_receive_queue.lock);
537 po->stats.tp_drops++;
538 spin_unlock(&sk->sk_receive_queue.lock);
540 drop_n_restore:
541 if (skb_head != skb->data && skb_shared(skb)) {
542 skb->data = skb_head;
543 skb->len = skb_len;
545 drop:
546 kfree_skb(skb);
547 return 0;
550 #ifdef CONFIG_PACKET_MMAP
551 static int tpacket_rcv(struct sk_buff *skb, struct net_device *dev, struct packet_type *pt, struct net_device *orig_dev)
553 struct sock *sk;
554 struct packet_sock *po;
555 struct sockaddr_ll *sll;
556 struct tpacket_hdr *h;
557 u8 * skb_head = skb->data;
558 int skb_len = skb->len;
559 unsigned int snaplen, res;
560 unsigned long status = TP_STATUS_LOSING|TP_STATUS_USER;
561 unsigned short macoff, netoff;
562 struct sk_buff *copy_skb = NULL;
563 struct timeval tv;
565 if (skb->pkt_type == PACKET_LOOPBACK)
566 goto drop;
568 sk = pt->af_packet_priv;
569 po = pkt_sk(sk);
571 if (dev->nd_net != sk->sk_net)
572 goto drop;
574 if (dev->header_ops) {
575 if (sk->sk_type != SOCK_DGRAM)
576 skb_push(skb, skb->data - skb_mac_header(skb));
577 else if (skb->pkt_type == PACKET_OUTGOING) {
578 /* Special case: outgoing packets have ll header at head */
579 skb_pull(skb, skb_network_offset(skb));
583 if (skb->ip_summed == CHECKSUM_PARTIAL)
584 status |= TP_STATUS_CSUMNOTREADY;
586 snaplen = skb->len;
588 res = run_filter(skb, sk, snaplen);
589 if (!res)
590 goto drop_n_restore;
591 if (snaplen > res)
592 snaplen = res;
594 if (sk->sk_type == SOCK_DGRAM) {
595 macoff = netoff = TPACKET_ALIGN(TPACKET_HDRLEN) + 16;
596 } else {
597 unsigned maclen = skb_network_offset(skb);
598 netoff = TPACKET_ALIGN(TPACKET_HDRLEN + (maclen < 16 ? 16 : maclen));
599 macoff = netoff - maclen;
602 if (macoff + snaplen > po->frame_size) {
603 if (po->copy_thresh &&
604 atomic_read(&sk->sk_rmem_alloc) + skb->truesize <
605 (unsigned)sk->sk_rcvbuf) {
606 if (skb_shared(skb)) {
607 copy_skb = skb_clone(skb, GFP_ATOMIC);
608 } else {
609 copy_skb = skb_get(skb);
610 skb_head = skb->data;
612 if (copy_skb)
613 skb_set_owner_r(copy_skb, sk);
615 snaplen = po->frame_size - macoff;
616 if ((int)snaplen < 0)
617 snaplen = 0;
620 spin_lock(&sk->sk_receive_queue.lock);
621 h = packet_lookup_frame(po, po->head);
623 if (h->tp_status)
624 goto ring_is_full;
625 po->head = po->head != po->frame_max ? po->head+1 : 0;
626 po->stats.tp_packets++;
627 if (copy_skb) {
628 status |= TP_STATUS_COPY;
629 __skb_queue_tail(&sk->sk_receive_queue, copy_skb);
631 if (!po->stats.tp_drops)
632 status &= ~TP_STATUS_LOSING;
633 spin_unlock(&sk->sk_receive_queue.lock);
635 skb_copy_bits(skb, 0, (u8*)h + macoff, snaplen);
637 h->tp_len = skb->len;
638 h->tp_snaplen = snaplen;
639 h->tp_mac = macoff;
640 h->tp_net = netoff;
641 if (skb->tstamp.tv64)
642 tv = ktime_to_timeval(skb->tstamp);
643 else
644 do_gettimeofday(&tv);
645 h->tp_sec = tv.tv_sec;
646 h->tp_usec = tv.tv_usec;
648 sll = (struct sockaddr_ll*)((u8*)h + TPACKET_ALIGN(sizeof(*h)));
649 sll->sll_halen = dev_parse_header(skb, sll->sll_addr);
650 sll->sll_family = AF_PACKET;
651 sll->sll_hatype = dev->type;
652 sll->sll_protocol = skb->protocol;
653 sll->sll_pkttype = skb->pkt_type;
654 if (unlikely(po->origdev))
655 sll->sll_ifindex = orig_dev->ifindex;
656 else
657 sll->sll_ifindex = dev->ifindex;
659 h->tp_status = status;
660 smp_mb();
663 struct page *p_start, *p_end;
664 u8 *h_end = (u8 *)h + macoff + snaplen - 1;
666 p_start = virt_to_page(h);
667 p_end = virt_to_page(h_end);
668 while (p_start <= p_end) {
669 flush_dcache_page(p_start);
670 p_start++;
674 sk->sk_data_ready(sk, 0);
676 drop_n_restore:
677 if (skb_head != skb->data && skb_shared(skb)) {
678 skb->data = skb_head;
679 skb->len = skb_len;
681 drop:
682 kfree_skb(skb);
683 return 0;
685 ring_is_full:
686 po->stats.tp_drops++;
687 spin_unlock(&sk->sk_receive_queue.lock);
689 sk->sk_data_ready(sk, 0);
690 if (copy_skb)
691 kfree_skb(copy_skb);
692 goto drop_n_restore;
695 #endif
698 static int packet_sendmsg(struct kiocb *iocb, struct socket *sock,
699 struct msghdr *msg, size_t len)
701 struct sock *sk = sock->sk;
702 struct sockaddr_ll *saddr=(struct sockaddr_ll *)msg->msg_name;
703 struct sk_buff *skb;
704 struct net_device *dev;
705 __be16 proto;
706 unsigned char *addr;
707 int ifindex, err, reserve = 0;
710 * Get and verify the address.
713 if (saddr == NULL) {
714 struct packet_sock *po = pkt_sk(sk);
716 ifindex = po->ifindex;
717 proto = po->num;
718 addr = NULL;
719 } else {
720 err = -EINVAL;
721 if (msg->msg_namelen < sizeof(struct sockaddr_ll))
722 goto out;
723 if (msg->msg_namelen < (saddr->sll_halen + offsetof(struct sockaddr_ll, sll_addr)))
724 goto out;
725 ifindex = saddr->sll_ifindex;
726 proto = saddr->sll_protocol;
727 addr = saddr->sll_addr;
731 dev = dev_get_by_index(sk->sk_net, ifindex);
732 err = -ENXIO;
733 if (dev == NULL)
734 goto out_unlock;
735 if (sock->type == SOCK_RAW)
736 reserve = dev->hard_header_len;
738 err = -ENETDOWN;
739 if (!(dev->flags & IFF_UP))
740 goto out_unlock;
742 err = -EMSGSIZE;
743 if (len > dev->mtu+reserve)
744 goto out_unlock;
746 skb = sock_alloc_send_skb(sk, len + LL_RESERVED_SPACE(dev),
747 msg->msg_flags & MSG_DONTWAIT, &err);
748 if (skb==NULL)
749 goto out_unlock;
751 skb_reserve(skb, LL_RESERVED_SPACE(dev));
752 skb_reset_network_header(skb);
754 err = -EINVAL;
755 if (sock->type == SOCK_DGRAM &&
756 dev_hard_header(skb, dev, ntohs(proto), addr, NULL, len) < 0)
757 goto out_free;
759 /* Returns -EFAULT on error */
760 err = memcpy_fromiovec(skb_put(skb,len), msg->msg_iov, len);
761 if (err)
762 goto out_free;
764 skb->protocol = proto;
765 skb->dev = dev;
766 skb->priority = sk->sk_priority;
769 * Now send it
772 err = dev_queue_xmit(skb);
773 if (err > 0 && (err = net_xmit_errno(err)) != 0)
774 goto out_unlock;
776 dev_put(dev);
778 return(len);
780 out_free:
781 kfree_skb(skb);
782 out_unlock:
783 if (dev)
784 dev_put(dev);
785 out:
786 return err;
790 * Close a PACKET socket. This is fairly simple. We immediately go
791 * to 'closed' state and remove our protocol entry in the device list.
794 static int packet_release(struct socket *sock)
796 struct sock *sk = sock->sk;
797 struct packet_sock *po;
798 struct net *net;
800 if (!sk)
801 return 0;
803 net = sk->sk_net;
804 po = pkt_sk(sk);
806 write_lock_bh(&net->packet.sklist_lock);
807 sk_del_node_init(sk);
808 write_unlock_bh(&net->packet.sklist_lock);
811 * Unhook packet receive handler.
814 if (po->running) {
816 * Remove the protocol hook
818 dev_remove_pack(&po->prot_hook);
819 po->running = 0;
820 po->num = 0;
821 __sock_put(sk);
824 packet_flush_mclist(sk);
826 #ifdef CONFIG_PACKET_MMAP
827 if (po->pg_vec) {
828 struct tpacket_req req;
829 memset(&req, 0, sizeof(req));
830 packet_set_ring(sk, &req, 1);
832 #endif
835 * Now the socket is dead. No more input will appear.
838 sock_orphan(sk);
839 sock->sk = NULL;
841 /* Purge queues */
843 skb_queue_purge(&sk->sk_receive_queue);
844 sk_refcnt_debug_release(sk);
846 sock_put(sk);
847 return 0;
851 * Attach a packet hook.
854 static int packet_do_bind(struct sock *sk, struct net_device *dev, __be16 protocol)
856 struct packet_sock *po = pkt_sk(sk);
858 * Detach an existing hook if present.
861 lock_sock(sk);
863 spin_lock(&po->bind_lock);
864 if (po->running) {
865 __sock_put(sk);
866 po->running = 0;
867 po->num = 0;
868 spin_unlock(&po->bind_lock);
869 dev_remove_pack(&po->prot_hook);
870 spin_lock(&po->bind_lock);
873 po->num = protocol;
874 po->prot_hook.type = protocol;
875 po->prot_hook.dev = dev;
877 po->ifindex = dev ? dev->ifindex : 0;
879 if (protocol == 0)
880 goto out_unlock;
882 if (!dev || (dev->flags & IFF_UP)) {
883 dev_add_pack(&po->prot_hook);
884 sock_hold(sk);
885 po->running = 1;
886 } else {
887 sk->sk_err = ENETDOWN;
888 if (!sock_flag(sk, SOCK_DEAD))
889 sk->sk_error_report(sk);
892 out_unlock:
893 spin_unlock(&po->bind_lock);
894 release_sock(sk);
895 return 0;
899 * Bind a packet socket to a device
902 static int packet_bind_spkt(struct socket *sock, struct sockaddr *uaddr, int addr_len)
904 struct sock *sk=sock->sk;
905 char name[15];
906 struct net_device *dev;
907 int err = -ENODEV;
910 * Check legality
913 if (addr_len != sizeof(struct sockaddr))
914 return -EINVAL;
915 strlcpy(name,uaddr->sa_data,sizeof(name));
917 dev = dev_get_by_name(sk->sk_net, name);
918 if (dev) {
919 err = packet_do_bind(sk, dev, pkt_sk(sk)->num);
920 dev_put(dev);
922 return err;
925 static int packet_bind(struct socket *sock, struct sockaddr *uaddr, int addr_len)
927 struct sockaddr_ll *sll = (struct sockaddr_ll*)uaddr;
928 struct sock *sk=sock->sk;
929 struct net_device *dev = NULL;
930 int err;
934 * Check legality
937 if (addr_len < sizeof(struct sockaddr_ll))
938 return -EINVAL;
939 if (sll->sll_family != AF_PACKET)
940 return -EINVAL;
942 if (sll->sll_ifindex) {
943 err = -ENODEV;
944 dev = dev_get_by_index(sk->sk_net, sll->sll_ifindex);
945 if (dev == NULL)
946 goto out;
948 err = packet_do_bind(sk, dev, sll->sll_protocol ? : pkt_sk(sk)->num);
949 if (dev)
950 dev_put(dev);
952 out:
953 return err;
956 static struct proto packet_proto = {
957 .name = "PACKET",
958 .owner = THIS_MODULE,
959 .obj_size = sizeof(struct packet_sock),
963 * Create a packet of type SOCK_PACKET.
966 static int packet_create(struct net *net, struct socket *sock, int protocol)
968 struct sock *sk;
969 struct packet_sock *po;
970 __be16 proto = (__force __be16)protocol; /* weird, but documented */
971 int err;
973 if (!capable(CAP_NET_RAW))
974 return -EPERM;
975 if (sock->type != SOCK_DGRAM && sock->type != SOCK_RAW &&
976 sock->type != SOCK_PACKET)
977 return -ESOCKTNOSUPPORT;
979 sock->state = SS_UNCONNECTED;
981 err = -ENOBUFS;
982 sk = sk_alloc(net, PF_PACKET, GFP_KERNEL, &packet_proto);
983 if (sk == NULL)
984 goto out;
986 sock->ops = &packet_ops;
987 if (sock->type == SOCK_PACKET)
988 sock->ops = &packet_ops_spkt;
990 sock_init_data(sock, sk);
992 po = pkt_sk(sk);
993 sk->sk_family = PF_PACKET;
994 po->num = proto;
996 sk->sk_destruct = packet_sock_destruct;
997 sk_refcnt_debug_inc(sk);
1000 * Attach a protocol block
1003 spin_lock_init(&po->bind_lock);
1004 po->prot_hook.func = packet_rcv;
1006 if (sock->type == SOCK_PACKET)
1007 po->prot_hook.func = packet_rcv_spkt;
1009 po->prot_hook.af_packet_priv = sk;
1011 if (proto) {
1012 po->prot_hook.type = proto;
1013 dev_add_pack(&po->prot_hook);
1014 sock_hold(sk);
1015 po->running = 1;
1018 write_lock_bh(&net->packet.sklist_lock);
1019 sk_add_node(sk, &net->packet.sklist);
1020 write_unlock_bh(&net->packet.sklist_lock);
1021 return(0);
1022 out:
1023 return err;
1027 * Pull a packet from our receive queue and hand it to the user.
1028 * If necessary we block.
1031 static int packet_recvmsg(struct kiocb *iocb, struct socket *sock,
1032 struct msghdr *msg, size_t len, int flags)
1034 struct sock *sk = sock->sk;
1035 struct sk_buff *skb;
1036 int copied, err;
1037 struct sockaddr_ll *sll;
1039 err = -EINVAL;
1040 if (flags & ~(MSG_PEEK|MSG_DONTWAIT|MSG_TRUNC|MSG_CMSG_COMPAT))
1041 goto out;
1043 #if 0
1044 /* What error should we return now? EUNATTACH? */
1045 if (pkt_sk(sk)->ifindex < 0)
1046 return -ENODEV;
1047 #endif
1050 * Call the generic datagram receiver. This handles all sorts
1051 * of horrible races and re-entrancy so we can forget about it
1052 * in the protocol layers.
1054 * Now it will return ENETDOWN, if device have just gone down,
1055 * but then it will block.
1058 skb=skb_recv_datagram(sk,flags,flags&MSG_DONTWAIT,&err);
1061 * An error occurred so return it. Because skb_recv_datagram()
1062 * handles the blocking we don't see and worry about blocking
1063 * retries.
1066 if (skb == NULL)
1067 goto out;
1070 * If the address length field is there to be filled in, we fill
1071 * it in now.
1074 sll = &PACKET_SKB_CB(skb)->sa.ll;
1075 if (sock->type == SOCK_PACKET)
1076 msg->msg_namelen = sizeof(struct sockaddr_pkt);
1077 else
1078 msg->msg_namelen = sll->sll_halen + offsetof(struct sockaddr_ll, sll_addr);
1081 * You lose any data beyond the buffer you gave. If it worries a
1082 * user program they can ask the device for its MTU anyway.
1085 copied = skb->len;
1086 if (copied > len)
1088 copied=len;
1089 msg->msg_flags|=MSG_TRUNC;
1092 err = skb_copy_datagram_iovec(skb, 0, msg->msg_iov, copied);
1093 if (err)
1094 goto out_free;
1096 sock_recv_timestamp(msg, sk, skb);
1098 if (msg->msg_name)
1099 memcpy(msg->msg_name, &PACKET_SKB_CB(skb)->sa,
1100 msg->msg_namelen);
1102 if (pkt_sk(sk)->auxdata) {
1103 struct tpacket_auxdata aux;
1105 aux.tp_status = TP_STATUS_USER;
1106 if (skb->ip_summed == CHECKSUM_PARTIAL)
1107 aux.tp_status |= TP_STATUS_CSUMNOTREADY;
1108 aux.tp_len = PACKET_SKB_CB(skb)->origlen;
1109 aux.tp_snaplen = skb->len;
1110 aux.tp_mac = 0;
1111 aux.tp_net = skb_network_offset(skb);
1113 put_cmsg(msg, SOL_PACKET, PACKET_AUXDATA, sizeof(aux), &aux);
1117 * Free or return the buffer as appropriate. Again this
1118 * hides all the races and re-entrancy issues from us.
1120 err = (flags&MSG_TRUNC) ? skb->len : copied;
1122 out_free:
1123 skb_free_datagram(sk, skb);
1124 out:
1125 return err;
1128 static int packet_getname_spkt(struct socket *sock, struct sockaddr *uaddr,
1129 int *uaddr_len, int peer)
1131 struct net_device *dev;
1132 struct sock *sk = sock->sk;
1134 if (peer)
1135 return -EOPNOTSUPP;
1137 uaddr->sa_family = AF_PACKET;
1138 dev = dev_get_by_index(sk->sk_net, pkt_sk(sk)->ifindex);
1139 if (dev) {
1140 strlcpy(uaddr->sa_data, dev->name, 15);
1141 dev_put(dev);
1142 } else
1143 memset(uaddr->sa_data, 0, 14);
1144 *uaddr_len = sizeof(*uaddr);
1146 return 0;
1149 static int packet_getname(struct socket *sock, struct sockaddr *uaddr,
1150 int *uaddr_len, int peer)
1152 struct net_device *dev;
1153 struct sock *sk = sock->sk;
1154 struct packet_sock *po = pkt_sk(sk);
1155 struct sockaddr_ll *sll = (struct sockaddr_ll*)uaddr;
1157 if (peer)
1158 return -EOPNOTSUPP;
1160 sll->sll_family = AF_PACKET;
1161 sll->sll_ifindex = po->ifindex;
1162 sll->sll_protocol = po->num;
1163 dev = dev_get_by_index(sk->sk_net, po->ifindex);
1164 if (dev) {
1165 sll->sll_hatype = dev->type;
1166 sll->sll_halen = dev->addr_len;
1167 memcpy(sll->sll_addr, dev->dev_addr, dev->addr_len);
1168 dev_put(dev);
1169 } else {
1170 sll->sll_hatype = 0; /* Bad: we have no ARPHRD_UNSPEC */
1171 sll->sll_halen = 0;
1173 *uaddr_len = offsetof(struct sockaddr_ll, sll_addr) + sll->sll_halen;
1175 return 0;
1178 static void packet_dev_mc(struct net_device *dev, struct packet_mclist *i, int what)
1180 switch (i->type) {
1181 case PACKET_MR_MULTICAST:
1182 if (what > 0)
1183 dev_mc_add(dev, i->addr, i->alen, 0);
1184 else
1185 dev_mc_delete(dev, i->addr, i->alen, 0);
1186 break;
1187 case PACKET_MR_PROMISC:
1188 dev_set_promiscuity(dev, what);
1189 break;
1190 case PACKET_MR_ALLMULTI:
1191 dev_set_allmulti(dev, what);
1192 break;
1193 default:;
1197 static void packet_dev_mclist(struct net_device *dev, struct packet_mclist *i, int what)
1199 for ( ; i; i=i->next) {
1200 if (i->ifindex == dev->ifindex)
1201 packet_dev_mc(dev, i, what);
1205 static int packet_mc_add(struct sock *sk, struct packet_mreq_max *mreq)
1207 struct packet_sock *po = pkt_sk(sk);
1208 struct packet_mclist *ml, *i;
1209 struct net_device *dev;
1210 int err;
1212 rtnl_lock();
1214 err = -ENODEV;
1215 dev = __dev_get_by_index(sk->sk_net, mreq->mr_ifindex);
1216 if (!dev)
1217 goto done;
1219 err = -EINVAL;
1220 if (mreq->mr_alen > dev->addr_len)
1221 goto done;
1223 err = -ENOBUFS;
1224 i = kmalloc(sizeof(*i), GFP_KERNEL);
1225 if (i == NULL)
1226 goto done;
1228 err = 0;
1229 for (ml = po->mclist; ml; ml = ml->next) {
1230 if (ml->ifindex == mreq->mr_ifindex &&
1231 ml->type == mreq->mr_type &&
1232 ml->alen == mreq->mr_alen &&
1233 memcmp(ml->addr, mreq->mr_address, ml->alen) == 0) {
1234 ml->count++;
1235 /* Free the new element ... */
1236 kfree(i);
1237 goto done;
1241 i->type = mreq->mr_type;
1242 i->ifindex = mreq->mr_ifindex;
1243 i->alen = mreq->mr_alen;
1244 memcpy(i->addr, mreq->mr_address, i->alen);
1245 i->count = 1;
1246 i->next = po->mclist;
1247 po->mclist = i;
1248 packet_dev_mc(dev, i, +1);
1250 done:
1251 rtnl_unlock();
1252 return err;
1255 static int packet_mc_drop(struct sock *sk, struct packet_mreq_max *mreq)
1257 struct packet_mclist *ml, **mlp;
1259 rtnl_lock();
1261 for (mlp = &pkt_sk(sk)->mclist; (ml = *mlp) != NULL; mlp = &ml->next) {
1262 if (ml->ifindex == mreq->mr_ifindex &&
1263 ml->type == mreq->mr_type &&
1264 ml->alen == mreq->mr_alen &&
1265 memcmp(ml->addr, mreq->mr_address, ml->alen) == 0) {
1266 if (--ml->count == 0) {
1267 struct net_device *dev;
1268 *mlp = ml->next;
1269 dev = dev_get_by_index(sk->sk_net, ml->ifindex);
1270 if (dev) {
1271 packet_dev_mc(dev, ml, -1);
1272 dev_put(dev);
1274 kfree(ml);
1276 rtnl_unlock();
1277 return 0;
1280 rtnl_unlock();
1281 return -EADDRNOTAVAIL;
1284 static void packet_flush_mclist(struct sock *sk)
1286 struct packet_sock *po = pkt_sk(sk);
1287 struct packet_mclist *ml;
1289 if (!po->mclist)
1290 return;
1292 rtnl_lock();
1293 while ((ml = po->mclist) != NULL) {
1294 struct net_device *dev;
1296 po->mclist = ml->next;
1297 if ((dev = dev_get_by_index(sk->sk_net, ml->ifindex)) != NULL) {
1298 packet_dev_mc(dev, ml, -1);
1299 dev_put(dev);
1301 kfree(ml);
1303 rtnl_unlock();
1306 static int
1307 packet_setsockopt(struct socket *sock, int level, int optname, char __user *optval, int optlen)
1309 struct sock *sk = sock->sk;
1310 struct packet_sock *po = pkt_sk(sk);
1311 int ret;
1313 if (level != SOL_PACKET)
1314 return -ENOPROTOOPT;
1316 switch(optname) {
1317 case PACKET_ADD_MEMBERSHIP:
1318 case PACKET_DROP_MEMBERSHIP:
1320 struct packet_mreq_max mreq;
1321 int len = optlen;
1322 memset(&mreq, 0, sizeof(mreq));
1323 if (len < sizeof(struct packet_mreq))
1324 return -EINVAL;
1325 if (len > sizeof(mreq))
1326 len = sizeof(mreq);
1327 if (copy_from_user(&mreq,optval,len))
1328 return -EFAULT;
1329 if (len < (mreq.mr_alen + offsetof(struct packet_mreq, mr_address)))
1330 return -EINVAL;
1331 if (optname == PACKET_ADD_MEMBERSHIP)
1332 ret = packet_mc_add(sk, &mreq);
1333 else
1334 ret = packet_mc_drop(sk, &mreq);
1335 return ret;
1338 #ifdef CONFIG_PACKET_MMAP
1339 case PACKET_RX_RING:
1341 struct tpacket_req req;
1343 if (optlen<sizeof(req))
1344 return -EINVAL;
1345 if (copy_from_user(&req,optval,sizeof(req)))
1346 return -EFAULT;
1347 return packet_set_ring(sk, &req, 0);
1349 case PACKET_COPY_THRESH:
1351 int val;
1353 if (optlen!=sizeof(val))
1354 return -EINVAL;
1355 if (copy_from_user(&val,optval,sizeof(val)))
1356 return -EFAULT;
1358 pkt_sk(sk)->copy_thresh = val;
1359 return 0;
1361 #endif
1362 case PACKET_AUXDATA:
1364 int val;
1366 if (optlen < sizeof(val))
1367 return -EINVAL;
1368 if (copy_from_user(&val, optval, sizeof(val)))
1369 return -EFAULT;
1371 po->auxdata = !!val;
1372 return 0;
1374 case PACKET_ORIGDEV:
1376 int val;
1378 if (optlen < sizeof(val))
1379 return -EINVAL;
1380 if (copy_from_user(&val, optval, sizeof(val)))
1381 return -EFAULT;
1383 po->origdev = !!val;
1384 return 0;
1386 default:
1387 return -ENOPROTOOPT;
1391 static int packet_getsockopt(struct socket *sock, int level, int optname,
1392 char __user *optval, int __user *optlen)
1394 int len;
1395 int val;
1396 struct sock *sk = sock->sk;
1397 struct packet_sock *po = pkt_sk(sk);
1398 void *data;
1399 struct tpacket_stats st;
1401 if (level != SOL_PACKET)
1402 return -ENOPROTOOPT;
1404 if (get_user(len, optlen))
1405 return -EFAULT;
1407 if (len < 0)
1408 return -EINVAL;
1410 switch(optname) {
1411 case PACKET_STATISTICS:
1412 if (len > sizeof(struct tpacket_stats))
1413 len = sizeof(struct tpacket_stats);
1414 spin_lock_bh(&sk->sk_receive_queue.lock);
1415 st = po->stats;
1416 memset(&po->stats, 0, sizeof(st));
1417 spin_unlock_bh(&sk->sk_receive_queue.lock);
1418 st.tp_packets += st.tp_drops;
1420 data = &st;
1421 break;
1422 case PACKET_AUXDATA:
1423 if (len > sizeof(int))
1424 len = sizeof(int);
1425 val = po->auxdata;
1427 data = &val;
1428 break;
1429 case PACKET_ORIGDEV:
1430 if (len > sizeof(int))
1431 len = sizeof(int);
1432 val = po->origdev;
1434 data = &val;
1435 break;
1436 default:
1437 return -ENOPROTOOPT;
1440 if (put_user(len, optlen))
1441 return -EFAULT;
1442 if (copy_to_user(optval, data, len))
1443 return -EFAULT;
1444 return 0;
1448 static int packet_notifier(struct notifier_block *this, unsigned long msg, void *data)
1450 struct sock *sk;
1451 struct hlist_node *node;
1452 struct net_device *dev = data;
1453 struct net *net = dev->nd_net;
1455 read_lock(&net->packet.sklist_lock);
1456 sk_for_each(sk, node, &net->packet.sklist) {
1457 struct packet_sock *po = pkt_sk(sk);
1459 switch (msg) {
1460 case NETDEV_UNREGISTER:
1461 if (po->mclist)
1462 packet_dev_mclist(dev, po->mclist, -1);
1463 /* fallthrough */
1465 case NETDEV_DOWN:
1466 if (dev->ifindex == po->ifindex) {
1467 spin_lock(&po->bind_lock);
1468 if (po->running) {
1469 __dev_remove_pack(&po->prot_hook);
1470 __sock_put(sk);
1471 po->running = 0;
1472 sk->sk_err = ENETDOWN;
1473 if (!sock_flag(sk, SOCK_DEAD))
1474 sk->sk_error_report(sk);
1476 if (msg == NETDEV_UNREGISTER) {
1477 po->ifindex = -1;
1478 po->prot_hook.dev = NULL;
1480 spin_unlock(&po->bind_lock);
1482 break;
1483 case NETDEV_UP:
1484 spin_lock(&po->bind_lock);
1485 if (dev->ifindex == po->ifindex && po->num &&
1486 !po->running) {
1487 dev_add_pack(&po->prot_hook);
1488 sock_hold(sk);
1489 po->running = 1;
1491 spin_unlock(&po->bind_lock);
1492 break;
1495 read_unlock(&net->packet.sklist_lock);
1496 return NOTIFY_DONE;
1500 static int packet_ioctl(struct socket *sock, unsigned int cmd,
1501 unsigned long arg)
1503 struct sock *sk = sock->sk;
1505 switch(cmd) {
1506 case SIOCOUTQ:
1508 int amount = atomic_read(&sk->sk_wmem_alloc);
1509 return put_user(amount, (int __user *)arg);
1511 case SIOCINQ:
1513 struct sk_buff *skb;
1514 int amount = 0;
1516 spin_lock_bh(&sk->sk_receive_queue.lock);
1517 skb = skb_peek(&sk->sk_receive_queue);
1518 if (skb)
1519 amount = skb->len;
1520 spin_unlock_bh(&sk->sk_receive_queue.lock);
1521 return put_user(amount, (int __user *)arg);
1523 case SIOCGSTAMP:
1524 return sock_get_timestamp(sk, (struct timeval __user *)arg);
1525 case SIOCGSTAMPNS:
1526 return sock_get_timestampns(sk, (struct timespec __user *)arg);
1528 #ifdef CONFIG_INET
1529 case SIOCADDRT:
1530 case SIOCDELRT:
1531 case SIOCDARP:
1532 case SIOCGARP:
1533 case SIOCSARP:
1534 case SIOCGIFADDR:
1535 case SIOCSIFADDR:
1536 case SIOCGIFBRDADDR:
1537 case SIOCSIFBRDADDR:
1538 case SIOCGIFNETMASK:
1539 case SIOCSIFNETMASK:
1540 case SIOCGIFDSTADDR:
1541 case SIOCSIFDSTADDR:
1542 case SIOCSIFFLAGS:
1543 if (sk->sk_net != &init_net)
1544 return -ENOIOCTLCMD;
1545 return inet_dgram_ops.ioctl(sock, cmd, arg);
1546 #endif
1548 default:
1549 return -ENOIOCTLCMD;
1551 return 0;
1554 #ifndef CONFIG_PACKET_MMAP
1555 #define packet_mmap sock_no_mmap
1556 #define packet_poll datagram_poll
1557 #else
1559 static unsigned int packet_poll(struct file * file, struct socket *sock,
1560 poll_table *wait)
1562 struct sock *sk = sock->sk;
1563 struct packet_sock *po = pkt_sk(sk);
1564 unsigned int mask = datagram_poll(file, sock, wait);
1566 spin_lock_bh(&sk->sk_receive_queue.lock);
1567 if (po->pg_vec) {
1568 unsigned last = po->head ? po->head-1 : po->frame_max;
1569 struct tpacket_hdr *h;
1571 h = packet_lookup_frame(po, last);
1573 if (h->tp_status)
1574 mask |= POLLIN | POLLRDNORM;
1576 spin_unlock_bh(&sk->sk_receive_queue.lock);
1577 return mask;
1581 /* Dirty? Well, I still did not learn better way to account
1582 * for user mmaps.
1585 static void packet_mm_open(struct vm_area_struct *vma)
1587 struct file *file = vma->vm_file;
1588 struct socket * sock = file->private_data;
1589 struct sock *sk = sock->sk;
1591 if (sk)
1592 atomic_inc(&pkt_sk(sk)->mapped);
1595 static void packet_mm_close(struct vm_area_struct *vma)
1597 struct file *file = vma->vm_file;
1598 struct socket * sock = file->private_data;
1599 struct sock *sk = sock->sk;
1601 if (sk)
1602 atomic_dec(&pkt_sk(sk)->mapped);
1605 static struct vm_operations_struct packet_mmap_ops = {
1606 .open = packet_mm_open,
1607 .close =packet_mm_close,
1610 static void free_pg_vec(char **pg_vec, unsigned int order, unsigned int len)
1612 int i;
1614 for (i = 0; i < len; i++) {
1615 if (likely(pg_vec[i]))
1616 free_pages((unsigned long) pg_vec[i], order);
1618 kfree(pg_vec);
1621 static inline char *alloc_one_pg_vec_page(unsigned long order)
1623 return (char *) __get_free_pages(GFP_KERNEL | __GFP_COMP | __GFP_ZERO,
1624 order);
1627 static char **alloc_pg_vec(struct tpacket_req *req, int order)
1629 unsigned int block_nr = req->tp_block_nr;
1630 char **pg_vec;
1631 int i;
1633 pg_vec = kzalloc(block_nr * sizeof(char *), GFP_KERNEL);
1634 if (unlikely(!pg_vec))
1635 goto out;
1637 for (i = 0; i < block_nr; i++) {
1638 pg_vec[i] = alloc_one_pg_vec_page(order);
1639 if (unlikely(!pg_vec[i]))
1640 goto out_free_pgvec;
1643 out:
1644 return pg_vec;
1646 out_free_pgvec:
1647 free_pg_vec(pg_vec, order, block_nr);
1648 pg_vec = NULL;
1649 goto out;
1652 static int packet_set_ring(struct sock *sk, struct tpacket_req *req, int closing)
1654 char **pg_vec = NULL;
1655 struct packet_sock *po = pkt_sk(sk);
1656 int was_running, order = 0;
1657 __be16 num;
1658 int err = 0;
1660 if (req->tp_block_nr) {
1661 int i, l;
1663 /* Sanity tests and some calculations */
1665 if (unlikely(po->pg_vec))
1666 return -EBUSY;
1668 if (unlikely((int)req->tp_block_size <= 0))
1669 return -EINVAL;
1670 if (unlikely(req->tp_block_size & (PAGE_SIZE - 1)))
1671 return -EINVAL;
1672 if (unlikely(req->tp_frame_size < TPACKET_HDRLEN))
1673 return -EINVAL;
1674 if (unlikely(req->tp_frame_size & (TPACKET_ALIGNMENT - 1)))
1675 return -EINVAL;
1677 po->frames_per_block = req->tp_block_size/req->tp_frame_size;
1678 if (unlikely(po->frames_per_block <= 0))
1679 return -EINVAL;
1680 if (unlikely((po->frames_per_block * req->tp_block_nr) !=
1681 req->tp_frame_nr))
1682 return -EINVAL;
1684 err = -ENOMEM;
1685 order = get_order(req->tp_block_size);
1686 pg_vec = alloc_pg_vec(req, order);
1687 if (unlikely(!pg_vec))
1688 goto out;
1690 l = 0;
1691 for (i = 0; i < req->tp_block_nr; i++) {
1692 char *ptr = pg_vec[i];
1693 struct tpacket_hdr *header;
1694 int k;
1696 for (k = 0; k < po->frames_per_block; k++) {
1697 header = (struct tpacket_hdr *) ptr;
1698 header->tp_status = TP_STATUS_KERNEL;
1699 ptr += req->tp_frame_size;
1702 /* Done */
1703 } else {
1704 if (unlikely(req->tp_frame_nr))
1705 return -EINVAL;
1708 lock_sock(sk);
1710 /* Detach socket from network */
1711 spin_lock(&po->bind_lock);
1712 was_running = po->running;
1713 num = po->num;
1714 if (was_running) {
1715 __dev_remove_pack(&po->prot_hook);
1716 po->num = 0;
1717 po->running = 0;
1718 __sock_put(sk);
1720 spin_unlock(&po->bind_lock);
1722 synchronize_net();
1724 err = -EBUSY;
1725 if (closing || atomic_read(&po->mapped) == 0) {
1726 err = 0;
1727 #define XC(a, b) ({ __typeof__ ((a)) __t; __t = (a); (a) = (b); __t; })
1729 spin_lock_bh(&sk->sk_receive_queue.lock);
1730 pg_vec = XC(po->pg_vec, pg_vec);
1731 po->frame_max = (req->tp_frame_nr - 1);
1732 po->head = 0;
1733 po->frame_size = req->tp_frame_size;
1734 spin_unlock_bh(&sk->sk_receive_queue.lock);
1736 order = XC(po->pg_vec_order, order);
1737 req->tp_block_nr = XC(po->pg_vec_len, req->tp_block_nr);
1739 po->pg_vec_pages = req->tp_block_size/PAGE_SIZE;
1740 po->prot_hook.func = po->pg_vec ? tpacket_rcv : packet_rcv;
1741 skb_queue_purge(&sk->sk_receive_queue);
1742 #undef XC
1743 if (atomic_read(&po->mapped))
1744 printk(KERN_DEBUG "packet_mmap: vma is busy: %d\n", atomic_read(&po->mapped));
1747 spin_lock(&po->bind_lock);
1748 if (was_running && !po->running) {
1749 sock_hold(sk);
1750 po->running = 1;
1751 po->num = num;
1752 dev_add_pack(&po->prot_hook);
1754 spin_unlock(&po->bind_lock);
1756 release_sock(sk);
1758 if (pg_vec)
1759 free_pg_vec(pg_vec, order, req->tp_block_nr);
1760 out:
1761 return err;
1764 static int packet_mmap(struct file *file, struct socket *sock, struct vm_area_struct *vma)
1766 struct sock *sk = sock->sk;
1767 struct packet_sock *po = pkt_sk(sk);
1768 unsigned long size;
1769 unsigned long start;
1770 int err = -EINVAL;
1771 int i;
1773 if (vma->vm_pgoff)
1774 return -EINVAL;
1776 size = vma->vm_end - vma->vm_start;
1778 lock_sock(sk);
1779 if (po->pg_vec == NULL)
1780 goto out;
1781 if (size != po->pg_vec_len*po->pg_vec_pages*PAGE_SIZE)
1782 goto out;
1784 start = vma->vm_start;
1785 for (i = 0; i < po->pg_vec_len; i++) {
1786 struct page *page = virt_to_page(po->pg_vec[i]);
1787 int pg_num;
1789 for (pg_num = 0; pg_num < po->pg_vec_pages; pg_num++, page++) {
1790 err = vm_insert_page(vma, start, page);
1791 if (unlikely(err))
1792 goto out;
1793 start += PAGE_SIZE;
1796 atomic_inc(&po->mapped);
1797 vma->vm_ops = &packet_mmap_ops;
1798 err = 0;
1800 out:
1801 release_sock(sk);
1802 return err;
1804 #endif
1807 static const struct proto_ops packet_ops_spkt = {
1808 .family = PF_PACKET,
1809 .owner = THIS_MODULE,
1810 .release = packet_release,
1811 .bind = packet_bind_spkt,
1812 .connect = sock_no_connect,
1813 .socketpair = sock_no_socketpair,
1814 .accept = sock_no_accept,
1815 .getname = packet_getname_spkt,
1816 .poll = datagram_poll,
1817 .ioctl = packet_ioctl,
1818 .listen = sock_no_listen,
1819 .shutdown = sock_no_shutdown,
1820 .setsockopt = sock_no_setsockopt,
1821 .getsockopt = sock_no_getsockopt,
1822 .sendmsg = packet_sendmsg_spkt,
1823 .recvmsg = packet_recvmsg,
1824 .mmap = sock_no_mmap,
1825 .sendpage = sock_no_sendpage,
1828 static const struct proto_ops packet_ops = {
1829 .family = PF_PACKET,
1830 .owner = THIS_MODULE,
1831 .release = packet_release,
1832 .bind = packet_bind,
1833 .connect = sock_no_connect,
1834 .socketpair = sock_no_socketpair,
1835 .accept = sock_no_accept,
1836 .getname = packet_getname,
1837 .poll = packet_poll,
1838 .ioctl = packet_ioctl,
1839 .listen = sock_no_listen,
1840 .shutdown = sock_no_shutdown,
1841 .setsockopt = packet_setsockopt,
1842 .getsockopt = packet_getsockopt,
1843 .sendmsg = packet_sendmsg,
1844 .recvmsg = packet_recvmsg,
1845 .mmap = packet_mmap,
1846 .sendpage = sock_no_sendpage,
1849 static struct net_proto_family packet_family_ops = {
1850 .family = PF_PACKET,
1851 .create = packet_create,
1852 .owner = THIS_MODULE,
1855 static struct notifier_block packet_netdev_notifier = {
1856 .notifier_call =packet_notifier,
1859 #ifdef CONFIG_PROC_FS
1860 static inline struct sock *packet_seq_idx(struct net *net, loff_t off)
1862 struct sock *s;
1863 struct hlist_node *node;
1865 sk_for_each(s, node, &net->packet.sklist) {
1866 if (!off--)
1867 return s;
1869 return NULL;
1872 static void *packet_seq_start(struct seq_file *seq, loff_t *pos)
1873 __acquires(seq_file_net(seq)->packet.sklist_lock)
1875 struct net *net = seq_file_net(seq);
1876 read_lock(&net->packet.sklist_lock);
1877 return *pos ? packet_seq_idx(net, *pos - 1) : SEQ_START_TOKEN;
1880 static void *packet_seq_next(struct seq_file *seq, void *v, loff_t *pos)
1882 struct net *net = seq_file_net(seq);
1883 ++*pos;
1884 return (v == SEQ_START_TOKEN)
1885 ? sk_head(&net->packet.sklist)
1886 : sk_next((struct sock*)v) ;
1889 static void packet_seq_stop(struct seq_file *seq, void *v)
1890 __releases(seq_file_net(seq)->packet.sklist_lock)
1892 struct net *net = seq_file_net(seq);
1893 read_unlock(&net->packet.sklist_lock);
1896 static int packet_seq_show(struct seq_file *seq, void *v)
1898 if (v == SEQ_START_TOKEN)
1899 seq_puts(seq, "sk RefCnt Type Proto Iface R Rmem User Inode\n");
1900 else {
1901 struct sock *s = v;
1902 const struct packet_sock *po = pkt_sk(s);
1904 seq_printf(seq,
1905 "%p %-6d %-4d %04x %-5d %1d %-6u %-6u %-6lu\n",
1907 atomic_read(&s->sk_refcnt),
1908 s->sk_type,
1909 ntohs(po->num),
1910 po->ifindex,
1911 po->running,
1912 atomic_read(&s->sk_rmem_alloc),
1913 sock_i_uid(s),
1914 sock_i_ino(s) );
1917 return 0;
1920 static const struct seq_operations packet_seq_ops = {
1921 .start = packet_seq_start,
1922 .next = packet_seq_next,
1923 .stop = packet_seq_stop,
1924 .show = packet_seq_show,
1927 static int packet_seq_open(struct inode *inode, struct file *file)
1929 return seq_open_net(inode, file, &packet_seq_ops,
1930 sizeof(struct seq_net_private));
1933 static const struct file_operations packet_seq_fops = {
1934 .owner = THIS_MODULE,
1935 .open = packet_seq_open,
1936 .read = seq_read,
1937 .llseek = seq_lseek,
1938 .release = seq_release_net,
1941 #endif
1943 static int packet_net_init(struct net *net)
1945 rwlock_init(&net->packet.sklist_lock);
1946 INIT_HLIST_HEAD(&net->packet.sklist);
1948 if (!proc_net_fops_create(net, "packet", 0, &packet_seq_fops))
1949 return -ENOMEM;
1951 return 0;
1954 static void packet_net_exit(struct net *net)
1956 proc_net_remove(net, "packet");
1959 static struct pernet_operations packet_net_ops = {
1960 .init = packet_net_init,
1961 .exit = packet_net_exit,
1965 static void __exit packet_exit(void)
1967 unregister_netdevice_notifier(&packet_netdev_notifier);
1968 unregister_pernet_subsys(&packet_net_ops);
1969 sock_unregister(PF_PACKET);
1970 proto_unregister(&packet_proto);
1973 static int __init packet_init(void)
1975 int rc = proto_register(&packet_proto, 0);
1977 if (rc != 0)
1978 goto out;
1980 sock_register(&packet_family_ops);
1981 register_pernet_subsys(&packet_net_ops);
1982 register_netdevice_notifier(&packet_netdev_notifier);
1983 out:
1984 return rc;
1987 module_init(packet_init);
1988 module_exit(packet_exit);
1989 MODULE_LICENSE("GPL");
1990 MODULE_ALIAS_NETPROTO(PF_PACKET);