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.
9 * Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG>
10 * Alan Cox, <gw4pts@gw4pts.ampr.org>
13 * Alan Cox : verify_area() now used correctly
14 * Alan Cox : new skbuff lists, look ma no backlogs!
15 * Alan Cox : tidied skbuff lists.
16 * Alan Cox : Now uses generic datagram routines I
17 * added. Also fixed the peek/read crash
18 * from all old Linux datagram code.
19 * Alan Cox : Uses the improved datagram code.
20 * Alan Cox : Added NULL's for socket options.
21 * Alan Cox : Re-commented the code.
22 * Alan Cox : Use new kernel side addressing
23 * Rob Janssen : Correct MTU usage.
24 * Dave Platt : Counter leaks caused by incorrect
25 * interrupt locking and some slightly
26 * dubious gcc output. Can you read
27 * compiler: it said _VOLATILE_
28 * Richard Kooijman : Timestamp fixes.
29 * Alan Cox : New buffers. Use sk->mac.raw.
30 * Alan Cox : sendmsg/recvmsg support.
31 * Alan Cox : Protocol setting support
32 * Alexey Kuznetsov : Untied from IPv4 stack.
33 * Cyrus Durgin : Fixed kerneld for kmod.
34 * Michal Ostrowski : Module initialization cleanup.
35 * Ulises Alonso : Frame number limit removal and
36 * packet_set_ring memory leak.
37 * Eric Biederman : Allow for > 8 byte hardware addresses.
38 * The convention is that longer addresses
39 * will simply extend the hardware address
40 * byte arrays at the end of sockaddr_ll
42 * Johann Baudy : Added TX RING.
43 * Chetan Loke : Implemented TPACKET_V3 block abstraction
45 * Copyright (C) 2011, <lokec@ccs.neu.edu>
48 * This program is free software; you can redistribute it and/or
49 * modify it under the terms of the GNU General Public License
50 * as published by the Free Software Foundation; either version
51 * 2 of the License, or (at your option) any later version.
55 #include <linux/types.h>
57 #include <linux/capability.h>
58 #include <linux/fcntl.h>
59 #include <linux/socket.h>
61 #include <linux/inet.h>
62 #include <linux/netdevice.h>
63 #include <linux/if_packet.h>
64 #include <linux/wireless.h>
65 #include <linux/kernel.h>
66 #include <linux/kmod.h>
67 #include <linux/slab.h>
68 #include <linux/vmalloc.h>
69 #include <net/net_namespace.h>
71 #include <net/protocol.h>
72 #include <linux/skbuff.h>
74 #include <linux/errno.h>
75 #include <linux/timer.h>
76 #include <linux/uaccess.h>
77 #include <asm/ioctls.h>
79 #include <asm/cacheflush.h>
81 #include <linux/proc_fs.h>
82 #include <linux/seq_file.h>
83 #include <linux/poll.h>
84 #include <linux/module.h>
85 #include <linux/init.h>
86 #include <linux/mutex.h>
87 #include <linux/if_vlan.h>
88 #include <linux/virtio_net.h>
89 #include <linux/errqueue.h>
90 #include <linux/net_tstamp.h>
91 #include <linux/percpu.h>
93 #include <net/inet_common.h>
95 #include <linux/bpf.h>
96 #include <net/compat.h>
102 - if device has no dev->hard_header routine, it adds and removes ll header
103 inside itself. In this case ll header is invisible outside of device,
104 but higher levels still should reserve dev->hard_header_len.
105 Some devices are enough clever to reallocate skb, when header
106 will not fit to reserved space (tunnel), another ones are silly
108 - packet socket receives packets with pulled ll header,
109 so that SOCK_RAW should push it back.
114 Incoming, dev->hard_header!=NULL
115 mac_header -> ll header
118 Outgoing, dev->hard_header!=NULL
119 mac_header -> ll header
122 Incoming, dev->hard_header==NULL
123 mac_header -> UNKNOWN position. It is very likely, that it points to ll
124 header. PPP makes it, that is wrong, because introduce
125 assymetry between rx and tx paths.
128 Outgoing, dev->hard_header==NULL
129 mac_header -> data. ll header is still not built!
133 If dev->hard_header==NULL we are unlikely to restore sensible ll header.
139 dev->hard_header != NULL
140 mac_header -> ll header
143 dev->hard_header == NULL (ll header is added by device, we cannot control it)
147 We should set nh.raw on output to correct posistion,
148 packet classifier depends on it.
151 /* Private packet socket structures. */
153 /* identical to struct packet_mreq except it has
154 * a longer address field.
156 struct packet_mreq_max
{
158 unsigned short mr_type
;
159 unsigned short mr_alen
;
160 unsigned char mr_address
[MAX_ADDR_LEN
];
164 struct tpacket_hdr
*h1
;
165 struct tpacket2_hdr
*h2
;
166 struct tpacket3_hdr
*h3
;
170 static int packet_set_ring(struct sock
*sk
, union tpacket_req_u
*req_u
,
171 int closing
, int tx_ring
);
173 #define V3_ALIGNMENT (8)
175 #define BLK_HDR_LEN (ALIGN(sizeof(struct tpacket_block_desc), V3_ALIGNMENT))
177 #define BLK_PLUS_PRIV(sz_of_priv) \
178 (BLK_HDR_LEN + ALIGN((sz_of_priv), V3_ALIGNMENT))
180 #define PGV_FROM_VMALLOC 1
182 #define BLOCK_STATUS(x) ((x)->hdr.bh1.block_status)
183 #define BLOCK_NUM_PKTS(x) ((x)->hdr.bh1.num_pkts)
184 #define BLOCK_O2FP(x) ((x)->hdr.bh1.offset_to_first_pkt)
185 #define BLOCK_LEN(x) ((x)->hdr.bh1.blk_len)
186 #define BLOCK_SNUM(x) ((x)->hdr.bh1.seq_num)
187 #define BLOCK_O2PRIV(x) ((x)->offset_to_priv)
188 #define BLOCK_PRIV(x) ((void *)((char *)(x) + BLOCK_O2PRIV(x)))
191 static int tpacket_snd(struct packet_sock
*po
, struct msghdr
*msg
);
192 static int tpacket_rcv(struct sk_buff
*skb
, struct net_device
*dev
,
193 struct packet_type
*pt
, struct net_device
*orig_dev
);
195 static void *packet_previous_frame(struct packet_sock
*po
,
196 struct packet_ring_buffer
*rb
,
198 static void packet_increment_head(struct packet_ring_buffer
*buff
);
199 static int prb_curr_blk_in_use(struct tpacket_block_desc
*);
200 static void *prb_dispatch_next_block(struct tpacket_kbdq_core
*,
201 struct packet_sock
*);
202 static void prb_retire_current_block(struct tpacket_kbdq_core
*,
203 struct packet_sock
*, unsigned int status
);
204 static int prb_queue_frozen(struct tpacket_kbdq_core
*);
205 static void prb_open_block(struct tpacket_kbdq_core
*,
206 struct tpacket_block_desc
*);
207 static void prb_retire_rx_blk_timer_expired(unsigned long);
208 static void _prb_refresh_rx_retire_blk_timer(struct tpacket_kbdq_core
*);
209 static void prb_init_blk_timer(struct packet_sock
*,
210 struct tpacket_kbdq_core
*,
211 void (*func
) (unsigned long));
212 static void prb_fill_rxhash(struct tpacket_kbdq_core
*, struct tpacket3_hdr
*);
213 static void prb_clear_rxhash(struct tpacket_kbdq_core
*,
214 struct tpacket3_hdr
*);
215 static void prb_fill_vlan_info(struct tpacket_kbdq_core
*,
216 struct tpacket3_hdr
*);
217 static void packet_flush_mclist(struct sock
*sk
);
219 struct packet_skb_cb
{
221 struct sockaddr_pkt pkt
;
223 /* Trick: alias skb original length with
224 * ll.sll_family and ll.protocol in order
227 unsigned int origlen
;
228 struct sockaddr_ll ll
;
233 #define vio_le() virtio_legacy_is_little_endian()
235 #define PACKET_SKB_CB(__skb) ((struct packet_skb_cb *)((__skb)->cb))
237 #define GET_PBDQC_FROM_RB(x) ((struct tpacket_kbdq_core *)(&(x)->prb_bdqc))
238 #define GET_PBLOCK_DESC(x, bid) \
239 ((struct tpacket_block_desc *)((x)->pkbdq[(bid)].buffer))
240 #define GET_CURR_PBLOCK_DESC_FROM_CORE(x) \
241 ((struct tpacket_block_desc *)((x)->pkbdq[(x)->kactive_blk_num].buffer))
242 #define GET_NEXT_PRB_BLK_NUM(x) \
243 (((x)->kactive_blk_num < ((x)->knum_blocks-1)) ? \
244 ((x)->kactive_blk_num+1) : 0)
246 static void __fanout_unlink(struct sock
*sk
, struct packet_sock
*po
);
247 static void __fanout_link(struct sock
*sk
, struct packet_sock
*po
);
249 static int packet_direct_xmit(struct sk_buff
*skb
)
251 struct net_device
*dev
= skb
->dev
;
252 struct sk_buff
*orig_skb
= skb
;
253 struct netdev_queue
*txq
;
254 int ret
= NETDEV_TX_BUSY
;
256 if (unlikely(!netif_running(dev
) ||
257 !netif_carrier_ok(dev
)))
260 skb
= validate_xmit_skb_list(skb
, dev
);
264 txq
= skb_get_tx_queue(dev
, skb
);
268 HARD_TX_LOCK(dev
, txq
, smp_processor_id());
269 if (!netif_xmit_frozen_or_drv_stopped(txq
))
270 ret
= netdev_start_xmit(skb
, dev
, txq
, false);
271 HARD_TX_UNLOCK(dev
, txq
);
275 if (!dev_xmit_complete(ret
))
280 atomic_long_inc(&dev
->tx_dropped
);
282 return NET_XMIT_DROP
;
285 static struct net_device
*packet_cached_dev_get(struct packet_sock
*po
)
287 struct net_device
*dev
;
290 dev
= rcu_dereference(po
->cached_dev
);
298 static void packet_cached_dev_assign(struct packet_sock
*po
,
299 struct net_device
*dev
)
301 rcu_assign_pointer(po
->cached_dev
, dev
);
304 static void packet_cached_dev_reset(struct packet_sock
*po
)
306 RCU_INIT_POINTER(po
->cached_dev
, NULL
);
309 static bool packet_use_direct_xmit(const struct packet_sock
*po
)
311 return po
->xmit
== packet_direct_xmit
;
314 static u16
__packet_pick_tx_queue(struct net_device
*dev
, struct sk_buff
*skb
)
316 return (u16
) raw_smp_processor_id() % dev
->real_num_tx_queues
;
319 static void packet_pick_tx_queue(struct net_device
*dev
, struct sk_buff
*skb
)
321 const struct net_device_ops
*ops
= dev
->netdev_ops
;
324 if (ops
->ndo_select_queue
) {
325 queue_index
= ops
->ndo_select_queue(dev
, skb
, NULL
,
326 __packet_pick_tx_queue
);
327 queue_index
= netdev_cap_txqueue(dev
, queue_index
);
329 queue_index
= __packet_pick_tx_queue(dev
, skb
);
332 skb_set_queue_mapping(skb
, queue_index
);
335 /* register_prot_hook must be invoked with the po->bind_lock held,
336 * or from a context in which asynchronous accesses to the packet
337 * socket is not possible (packet_create()).
339 static void register_prot_hook(struct sock
*sk
)
341 struct packet_sock
*po
= pkt_sk(sk
);
345 __fanout_link(sk
, po
);
347 dev_add_pack(&po
->prot_hook
);
354 /* {,__}unregister_prot_hook() must be invoked with the po->bind_lock
355 * held. If the sync parameter is true, we will temporarily drop
356 * the po->bind_lock and do a synchronize_net to make sure no
357 * asynchronous packet processing paths still refer to the elements
358 * of po->prot_hook. If the sync parameter is false, it is the
359 * callers responsibility to take care of this.
361 static void __unregister_prot_hook(struct sock
*sk
, bool sync
)
363 struct packet_sock
*po
= pkt_sk(sk
);
368 __fanout_unlink(sk
, po
);
370 __dev_remove_pack(&po
->prot_hook
);
375 spin_unlock(&po
->bind_lock
);
377 spin_lock(&po
->bind_lock
);
381 static void unregister_prot_hook(struct sock
*sk
, bool sync
)
383 struct packet_sock
*po
= pkt_sk(sk
);
386 __unregister_prot_hook(sk
, sync
);
389 static inline struct page
* __pure
pgv_to_page(void *addr
)
391 if (is_vmalloc_addr(addr
))
392 return vmalloc_to_page(addr
);
393 return virt_to_page(addr
);
396 static void __packet_set_status(struct packet_sock
*po
, void *frame
, int status
)
398 union tpacket_uhdr h
;
401 switch (po
->tp_version
) {
403 h
.h1
->tp_status
= status
;
404 flush_dcache_page(pgv_to_page(&h
.h1
->tp_status
));
407 h
.h2
->tp_status
= status
;
408 flush_dcache_page(pgv_to_page(&h
.h2
->tp_status
));
411 h
.h3
->tp_status
= status
;
412 flush_dcache_page(pgv_to_page(&h
.h3
->tp_status
));
415 WARN(1, "TPACKET version not supported.\n");
422 static int __packet_get_status(struct packet_sock
*po
, void *frame
)
424 union tpacket_uhdr h
;
429 switch (po
->tp_version
) {
431 flush_dcache_page(pgv_to_page(&h
.h1
->tp_status
));
432 return h
.h1
->tp_status
;
434 flush_dcache_page(pgv_to_page(&h
.h2
->tp_status
));
435 return h
.h2
->tp_status
;
437 flush_dcache_page(pgv_to_page(&h
.h3
->tp_status
));
438 return h
.h3
->tp_status
;
440 WARN(1, "TPACKET version not supported.\n");
446 static __u32
tpacket_get_timestamp(struct sk_buff
*skb
, struct timespec
*ts
,
449 struct skb_shared_hwtstamps
*shhwtstamps
= skb_hwtstamps(skb
);
452 (flags
& SOF_TIMESTAMPING_RAW_HARDWARE
) &&
453 ktime_to_timespec_cond(shhwtstamps
->hwtstamp
, ts
))
454 return TP_STATUS_TS_RAW_HARDWARE
;
456 if (ktime_to_timespec_cond(skb
->tstamp
, ts
))
457 return TP_STATUS_TS_SOFTWARE
;
462 static __u32
__packet_set_timestamp(struct packet_sock
*po
, void *frame
,
465 union tpacket_uhdr h
;
469 if (!(ts_status
= tpacket_get_timestamp(skb
, &ts
, po
->tp_tstamp
)))
473 switch (po
->tp_version
) {
475 h
.h1
->tp_sec
= ts
.tv_sec
;
476 h
.h1
->tp_usec
= ts
.tv_nsec
/ NSEC_PER_USEC
;
479 h
.h2
->tp_sec
= ts
.tv_sec
;
480 h
.h2
->tp_nsec
= ts
.tv_nsec
;
483 h
.h3
->tp_sec
= ts
.tv_sec
;
484 h
.h3
->tp_nsec
= ts
.tv_nsec
;
487 WARN(1, "TPACKET version not supported.\n");
491 /* one flush is safe, as both fields always lie on the same cacheline */
492 flush_dcache_page(pgv_to_page(&h
.h1
->tp_sec
));
498 static void *packet_lookup_frame(struct packet_sock
*po
,
499 struct packet_ring_buffer
*rb
,
500 unsigned int position
,
503 unsigned int pg_vec_pos
, frame_offset
;
504 union tpacket_uhdr h
;
506 pg_vec_pos
= position
/ rb
->frames_per_block
;
507 frame_offset
= position
% rb
->frames_per_block
;
509 h
.raw
= rb
->pg_vec
[pg_vec_pos
].buffer
+
510 (frame_offset
* rb
->frame_size
);
512 if (status
!= __packet_get_status(po
, h
.raw
))
518 static void *packet_current_frame(struct packet_sock
*po
,
519 struct packet_ring_buffer
*rb
,
522 return packet_lookup_frame(po
, rb
, rb
->head
, status
);
525 static void prb_del_retire_blk_timer(struct tpacket_kbdq_core
*pkc
)
527 del_timer_sync(&pkc
->retire_blk_timer
);
530 static void prb_shutdown_retire_blk_timer(struct packet_sock
*po
,
531 struct sk_buff_head
*rb_queue
)
533 struct tpacket_kbdq_core
*pkc
;
535 pkc
= GET_PBDQC_FROM_RB(&po
->rx_ring
);
537 spin_lock_bh(&rb_queue
->lock
);
538 pkc
->delete_blk_timer
= 1;
539 spin_unlock_bh(&rb_queue
->lock
);
541 prb_del_retire_blk_timer(pkc
);
544 static void prb_init_blk_timer(struct packet_sock
*po
,
545 struct tpacket_kbdq_core
*pkc
,
546 void (*func
) (unsigned long))
548 init_timer(&pkc
->retire_blk_timer
);
549 pkc
->retire_blk_timer
.data
= (long)po
;
550 pkc
->retire_blk_timer
.function
= func
;
551 pkc
->retire_blk_timer
.expires
= jiffies
;
554 static void prb_setup_retire_blk_timer(struct packet_sock
*po
)
556 struct tpacket_kbdq_core
*pkc
;
558 pkc
= GET_PBDQC_FROM_RB(&po
->rx_ring
);
559 prb_init_blk_timer(po
, pkc
, prb_retire_rx_blk_timer_expired
);
562 static int prb_calc_retire_blk_tmo(struct packet_sock
*po
,
563 int blk_size_in_bytes
)
565 struct net_device
*dev
;
566 unsigned int mbits
= 0, msec
= 0, div
= 0, tmo
= 0;
567 struct ethtool_link_ksettings ecmd
;
571 dev
= __dev_get_by_index(sock_net(&po
->sk
), po
->ifindex
);
572 if (unlikely(!dev
)) {
574 return DEFAULT_PRB_RETIRE_TOV
;
576 err
= __ethtool_get_link_ksettings(dev
, &ecmd
);
580 * If the link speed is so slow you don't really
581 * need to worry about perf anyways
583 if (ecmd
.base
.speed
< SPEED_1000
||
584 ecmd
.base
.speed
== SPEED_UNKNOWN
) {
585 return DEFAULT_PRB_RETIRE_TOV
;
588 div
= ecmd
.base
.speed
/ 1000;
592 mbits
= (blk_size_in_bytes
* 8) / (1024 * 1024);
604 static void prb_init_ft_ops(struct tpacket_kbdq_core
*p1
,
605 union tpacket_req_u
*req_u
)
607 p1
->feature_req_word
= req_u
->req3
.tp_feature_req_word
;
610 static void init_prb_bdqc(struct packet_sock
*po
,
611 struct packet_ring_buffer
*rb
,
613 union tpacket_req_u
*req_u
)
615 struct tpacket_kbdq_core
*p1
= GET_PBDQC_FROM_RB(rb
);
616 struct tpacket_block_desc
*pbd
;
618 memset(p1
, 0x0, sizeof(*p1
));
620 p1
->knxt_seq_num
= 1;
622 pbd
= (struct tpacket_block_desc
*)pg_vec
[0].buffer
;
623 p1
->pkblk_start
= pg_vec
[0].buffer
;
624 p1
->kblk_size
= req_u
->req3
.tp_block_size
;
625 p1
->knum_blocks
= req_u
->req3
.tp_block_nr
;
626 p1
->hdrlen
= po
->tp_hdrlen
;
627 p1
->version
= po
->tp_version
;
628 p1
->last_kactive_blk_num
= 0;
629 po
->stats
.stats3
.tp_freeze_q_cnt
= 0;
630 if (req_u
->req3
.tp_retire_blk_tov
)
631 p1
->retire_blk_tov
= req_u
->req3
.tp_retire_blk_tov
;
633 p1
->retire_blk_tov
= prb_calc_retire_blk_tmo(po
,
634 req_u
->req3
.tp_block_size
);
635 p1
->tov_in_jiffies
= msecs_to_jiffies(p1
->retire_blk_tov
);
636 p1
->blk_sizeof_priv
= req_u
->req3
.tp_sizeof_priv
;
638 p1
->max_frame_len
= p1
->kblk_size
- BLK_PLUS_PRIV(p1
->blk_sizeof_priv
);
639 prb_init_ft_ops(p1
, req_u
);
640 prb_setup_retire_blk_timer(po
);
641 prb_open_block(p1
, pbd
);
644 /* Do NOT update the last_blk_num first.
645 * Assumes sk_buff_head lock is held.
647 static void _prb_refresh_rx_retire_blk_timer(struct tpacket_kbdq_core
*pkc
)
649 mod_timer(&pkc
->retire_blk_timer
,
650 jiffies
+ pkc
->tov_in_jiffies
);
651 pkc
->last_kactive_blk_num
= pkc
->kactive_blk_num
;
656 * 1) We refresh the timer only when we open a block.
657 * By doing this we don't waste cycles refreshing the timer
658 * on packet-by-packet basis.
660 * With a 1MB block-size, on a 1Gbps line, it will take
661 * i) ~8 ms to fill a block + ii) memcpy etc.
662 * In this cut we are not accounting for the memcpy time.
664 * So, if the user sets the 'tmo' to 10ms then the timer
665 * will never fire while the block is still getting filled
666 * (which is what we want). However, the user could choose
667 * to close a block early and that's fine.
669 * But when the timer does fire, we check whether or not to refresh it.
670 * Since the tmo granularity is in msecs, it is not too expensive
671 * to refresh the timer, lets say every '8' msecs.
672 * Either the user can set the 'tmo' or we can derive it based on
673 * a) line-speed and b) block-size.
674 * prb_calc_retire_blk_tmo() calculates the tmo.
677 static void prb_retire_rx_blk_timer_expired(unsigned long data
)
679 struct packet_sock
*po
= (struct packet_sock
*)data
;
680 struct tpacket_kbdq_core
*pkc
= GET_PBDQC_FROM_RB(&po
->rx_ring
);
682 struct tpacket_block_desc
*pbd
;
684 spin_lock(&po
->sk
.sk_receive_queue
.lock
);
686 frozen
= prb_queue_frozen(pkc
);
687 pbd
= GET_CURR_PBLOCK_DESC_FROM_CORE(pkc
);
689 if (unlikely(pkc
->delete_blk_timer
))
692 /* We only need to plug the race when the block is partially filled.
694 * lock(); increment BLOCK_NUM_PKTS; unlock()
695 * copy_bits() is in progress ...
696 * timer fires on other cpu:
697 * we can't retire the current block because copy_bits
701 if (BLOCK_NUM_PKTS(pbd
)) {
702 while (atomic_read(&pkc
->blk_fill_in_prog
)) {
703 /* Waiting for skb_copy_bits to finish... */
708 if (pkc
->last_kactive_blk_num
== pkc
->kactive_blk_num
) {
710 if (!BLOCK_NUM_PKTS(pbd
)) {
711 /* An empty block. Just refresh the timer. */
714 prb_retire_current_block(pkc
, po
, TP_STATUS_BLK_TMO
);
715 if (!prb_dispatch_next_block(pkc
, po
))
720 /* Case 1. Queue was frozen because user-space was
723 if (prb_curr_blk_in_use(pbd
)) {
725 * Ok, user-space is still behind.
726 * So just refresh the timer.
730 /* Case 2. queue was frozen,user-space caught up,
731 * now the link went idle && the timer fired.
732 * We don't have a block to close.So we open this
733 * block and restart the timer.
734 * opening a block thaws the queue,restarts timer
735 * Thawing/timer-refresh is a side effect.
737 prb_open_block(pkc
, pbd
);
744 _prb_refresh_rx_retire_blk_timer(pkc
);
747 spin_unlock(&po
->sk
.sk_receive_queue
.lock
);
750 static void prb_flush_block(struct tpacket_kbdq_core
*pkc1
,
751 struct tpacket_block_desc
*pbd1
, __u32 status
)
753 /* Flush everything minus the block header */
755 #if ARCH_IMPLEMENTS_FLUSH_DCACHE_PAGE == 1
760 /* Skip the block header(we know header WILL fit in 4K) */
763 end
= (u8
*)PAGE_ALIGN((unsigned long)pkc1
->pkblk_end
);
764 for (; start
< end
; start
+= PAGE_SIZE
)
765 flush_dcache_page(pgv_to_page(start
));
770 /* Now update the block status. */
772 BLOCK_STATUS(pbd1
) = status
;
774 /* Flush the block header */
776 #if ARCH_IMPLEMENTS_FLUSH_DCACHE_PAGE == 1
778 flush_dcache_page(pgv_to_page(start
));
788 * 2) Increment active_blk_num
790 * Note:We DONT refresh the timer on purpose.
791 * Because almost always the next block will be opened.
793 static void prb_close_block(struct tpacket_kbdq_core
*pkc1
,
794 struct tpacket_block_desc
*pbd1
,
795 struct packet_sock
*po
, unsigned int stat
)
797 __u32 status
= TP_STATUS_USER
| stat
;
799 struct tpacket3_hdr
*last_pkt
;
800 struct tpacket_hdr_v1
*h1
= &pbd1
->hdr
.bh1
;
801 struct sock
*sk
= &po
->sk
;
803 if (po
->stats
.stats3
.tp_drops
)
804 status
|= TP_STATUS_LOSING
;
806 last_pkt
= (struct tpacket3_hdr
*)pkc1
->prev
;
807 last_pkt
->tp_next_offset
= 0;
809 /* Get the ts of the last pkt */
810 if (BLOCK_NUM_PKTS(pbd1
)) {
811 h1
->ts_last_pkt
.ts_sec
= last_pkt
->tp_sec
;
812 h1
->ts_last_pkt
.ts_nsec
= last_pkt
->tp_nsec
;
814 /* Ok, we tmo'd - so get the current time.
816 * It shouldn't really happen as we don't close empty
817 * blocks. See prb_retire_rx_blk_timer_expired().
821 h1
->ts_last_pkt
.ts_sec
= ts
.tv_sec
;
822 h1
->ts_last_pkt
.ts_nsec
= ts
.tv_nsec
;
827 /* Flush the block */
828 prb_flush_block(pkc1
, pbd1
, status
);
830 sk
->sk_data_ready(sk
);
832 pkc1
->kactive_blk_num
= GET_NEXT_PRB_BLK_NUM(pkc1
);
835 static void prb_thaw_queue(struct tpacket_kbdq_core
*pkc
)
837 pkc
->reset_pending_on_curr_blk
= 0;
841 * Side effect of opening a block:
843 * 1) prb_queue is thawed.
844 * 2) retire_blk_timer is refreshed.
847 static void prb_open_block(struct tpacket_kbdq_core
*pkc1
,
848 struct tpacket_block_desc
*pbd1
)
851 struct tpacket_hdr_v1
*h1
= &pbd1
->hdr
.bh1
;
855 /* We could have just memset this but we will lose the
856 * flexibility of making the priv area sticky
859 BLOCK_SNUM(pbd1
) = pkc1
->knxt_seq_num
++;
860 BLOCK_NUM_PKTS(pbd1
) = 0;
861 BLOCK_LEN(pbd1
) = BLK_PLUS_PRIV(pkc1
->blk_sizeof_priv
);
865 h1
->ts_first_pkt
.ts_sec
= ts
.tv_sec
;
866 h1
->ts_first_pkt
.ts_nsec
= ts
.tv_nsec
;
868 pkc1
->pkblk_start
= (char *)pbd1
;
869 pkc1
->nxt_offset
= pkc1
->pkblk_start
+ BLK_PLUS_PRIV(pkc1
->blk_sizeof_priv
);
871 BLOCK_O2FP(pbd1
) = (__u32
)BLK_PLUS_PRIV(pkc1
->blk_sizeof_priv
);
872 BLOCK_O2PRIV(pbd1
) = BLK_HDR_LEN
;
874 pbd1
->version
= pkc1
->version
;
875 pkc1
->prev
= pkc1
->nxt_offset
;
876 pkc1
->pkblk_end
= pkc1
->pkblk_start
+ pkc1
->kblk_size
;
878 prb_thaw_queue(pkc1
);
879 _prb_refresh_rx_retire_blk_timer(pkc1
);
885 * Queue freeze logic:
886 * 1) Assume tp_block_nr = 8 blocks.
887 * 2) At time 't0', user opens Rx ring.
888 * 3) Some time past 't0', kernel starts filling blocks starting from 0 .. 7
889 * 4) user-space is either sleeping or processing block '0'.
890 * 5) tpacket_rcv is currently filling block '7', since there is no space left,
891 * it will close block-7,loop around and try to fill block '0'.
893 * __packet_lookup_frame_in_block
894 * prb_retire_current_block()
895 * prb_dispatch_next_block()
896 * |->(BLOCK_STATUS == USER) evaluates to true
897 * 5.1) Since block-0 is currently in-use, we just freeze the queue.
898 * 6) Now there are two cases:
899 * 6.1) Link goes idle right after the queue is frozen.
900 * But remember, the last open_block() refreshed the timer.
901 * When this timer expires,it will refresh itself so that we can
902 * re-open block-0 in near future.
903 * 6.2) Link is busy and keeps on receiving packets. This is a simple
904 * case and __packet_lookup_frame_in_block will check if block-0
905 * is free and can now be re-used.
907 static void prb_freeze_queue(struct tpacket_kbdq_core
*pkc
,
908 struct packet_sock
*po
)
910 pkc
->reset_pending_on_curr_blk
= 1;
911 po
->stats
.stats3
.tp_freeze_q_cnt
++;
914 #define TOTAL_PKT_LEN_INCL_ALIGN(length) (ALIGN((length), V3_ALIGNMENT))
917 * If the next block is free then we will dispatch it
918 * and return a good offset.
919 * Else, we will freeze the queue.
920 * So, caller must check the return value.
922 static void *prb_dispatch_next_block(struct tpacket_kbdq_core
*pkc
,
923 struct packet_sock
*po
)
925 struct tpacket_block_desc
*pbd
;
929 /* 1. Get current block num */
930 pbd
= GET_CURR_PBLOCK_DESC_FROM_CORE(pkc
);
932 /* 2. If this block is currently in_use then freeze the queue */
933 if (TP_STATUS_USER
& BLOCK_STATUS(pbd
)) {
934 prb_freeze_queue(pkc
, po
);
940 * open this block and return the offset where the first packet
941 * needs to get stored.
943 prb_open_block(pkc
, pbd
);
944 return (void *)pkc
->nxt_offset
;
947 static void prb_retire_current_block(struct tpacket_kbdq_core
*pkc
,
948 struct packet_sock
*po
, unsigned int status
)
950 struct tpacket_block_desc
*pbd
= GET_CURR_PBLOCK_DESC_FROM_CORE(pkc
);
952 /* retire/close the current block */
953 if (likely(TP_STATUS_KERNEL
== BLOCK_STATUS(pbd
))) {
955 * Plug the case where copy_bits() is in progress on
956 * cpu-0 and tpacket_rcv() got invoked on cpu-1, didn't
957 * have space to copy the pkt in the current block and
958 * called prb_retire_current_block()
960 * We don't need to worry about the TMO case because
961 * the timer-handler already handled this case.
963 if (!(status
& TP_STATUS_BLK_TMO
)) {
964 while (atomic_read(&pkc
->blk_fill_in_prog
)) {
965 /* Waiting for skb_copy_bits to finish... */
969 prb_close_block(pkc
, pbd
, po
, status
);
974 static int prb_curr_blk_in_use(struct tpacket_block_desc
*pbd
)
976 return TP_STATUS_USER
& BLOCK_STATUS(pbd
);
979 static int prb_queue_frozen(struct tpacket_kbdq_core
*pkc
)
981 return pkc
->reset_pending_on_curr_blk
;
984 static void prb_clear_blk_fill_status(struct packet_ring_buffer
*rb
)
986 struct tpacket_kbdq_core
*pkc
= GET_PBDQC_FROM_RB(rb
);
987 atomic_dec(&pkc
->blk_fill_in_prog
);
990 static void prb_fill_rxhash(struct tpacket_kbdq_core
*pkc
,
991 struct tpacket3_hdr
*ppd
)
993 ppd
->hv1
.tp_rxhash
= skb_get_hash(pkc
->skb
);
996 static void prb_clear_rxhash(struct tpacket_kbdq_core
*pkc
,
997 struct tpacket3_hdr
*ppd
)
999 ppd
->hv1
.tp_rxhash
= 0;
1002 static void prb_fill_vlan_info(struct tpacket_kbdq_core
*pkc
,
1003 struct tpacket3_hdr
*ppd
)
1005 if (skb_vlan_tag_present(pkc
->skb
)) {
1006 ppd
->hv1
.tp_vlan_tci
= skb_vlan_tag_get(pkc
->skb
);
1007 ppd
->hv1
.tp_vlan_tpid
= ntohs(pkc
->skb
->vlan_proto
);
1008 ppd
->tp_status
= TP_STATUS_VLAN_VALID
| TP_STATUS_VLAN_TPID_VALID
;
1010 ppd
->hv1
.tp_vlan_tci
= 0;
1011 ppd
->hv1
.tp_vlan_tpid
= 0;
1012 ppd
->tp_status
= TP_STATUS_AVAILABLE
;
1016 static void prb_run_all_ft_ops(struct tpacket_kbdq_core
*pkc
,
1017 struct tpacket3_hdr
*ppd
)
1019 ppd
->hv1
.tp_padding
= 0;
1020 prb_fill_vlan_info(pkc
, ppd
);
1022 if (pkc
->feature_req_word
& TP_FT_REQ_FILL_RXHASH
)
1023 prb_fill_rxhash(pkc
, ppd
);
1025 prb_clear_rxhash(pkc
, ppd
);
1028 static void prb_fill_curr_block(char *curr
,
1029 struct tpacket_kbdq_core
*pkc
,
1030 struct tpacket_block_desc
*pbd
,
1033 struct tpacket3_hdr
*ppd
;
1035 ppd
= (struct tpacket3_hdr
*)curr
;
1036 ppd
->tp_next_offset
= TOTAL_PKT_LEN_INCL_ALIGN(len
);
1038 pkc
->nxt_offset
+= TOTAL_PKT_LEN_INCL_ALIGN(len
);
1039 BLOCK_LEN(pbd
) += TOTAL_PKT_LEN_INCL_ALIGN(len
);
1040 BLOCK_NUM_PKTS(pbd
) += 1;
1041 atomic_inc(&pkc
->blk_fill_in_prog
);
1042 prb_run_all_ft_ops(pkc
, ppd
);
1045 /* Assumes caller has the sk->rx_queue.lock */
1046 static void *__packet_lookup_frame_in_block(struct packet_sock
*po
,
1047 struct sk_buff
*skb
,
1052 struct tpacket_kbdq_core
*pkc
;
1053 struct tpacket_block_desc
*pbd
;
1056 pkc
= GET_PBDQC_FROM_RB(&po
->rx_ring
);
1057 pbd
= GET_CURR_PBLOCK_DESC_FROM_CORE(pkc
);
1059 /* Queue is frozen when user space is lagging behind */
1060 if (prb_queue_frozen(pkc
)) {
1062 * Check if that last block which caused the queue to freeze,
1063 * is still in_use by user-space.
1065 if (prb_curr_blk_in_use(pbd
)) {
1066 /* Can't record this packet */
1070 * Ok, the block was released by user-space.
1071 * Now let's open that block.
1072 * opening a block also thaws the queue.
1073 * Thawing is a side effect.
1075 prb_open_block(pkc
, pbd
);
1080 curr
= pkc
->nxt_offset
;
1082 end
= (char *)pbd
+ pkc
->kblk_size
;
1084 /* first try the current block */
1085 if (curr
+TOTAL_PKT_LEN_INCL_ALIGN(len
) < end
) {
1086 prb_fill_curr_block(curr
, pkc
, pbd
, len
);
1087 return (void *)curr
;
1090 /* Ok, close the current block */
1091 prb_retire_current_block(pkc
, po
, 0);
1093 /* Now, try to dispatch the next block */
1094 curr
= (char *)prb_dispatch_next_block(pkc
, po
);
1096 pbd
= GET_CURR_PBLOCK_DESC_FROM_CORE(pkc
);
1097 prb_fill_curr_block(curr
, pkc
, pbd
, len
);
1098 return (void *)curr
;
1102 * No free blocks are available.user_space hasn't caught up yet.
1103 * Queue was just frozen and now this packet will get dropped.
1108 static void *packet_current_rx_frame(struct packet_sock
*po
,
1109 struct sk_buff
*skb
,
1110 int status
, unsigned int len
)
1113 switch (po
->tp_version
) {
1116 curr
= packet_lookup_frame(po
, &po
->rx_ring
,
1117 po
->rx_ring
.head
, status
);
1120 return __packet_lookup_frame_in_block(po
, skb
, status
, len
);
1122 WARN(1, "TPACKET version not supported\n");
1128 static void *prb_lookup_block(struct packet_sock
*po
,
1129 struct packet_ring_buffer
*rb
,
1133 struct tpacket_kbdq_core
*pkc
= GET_PBDQC_FROM_RB(rb
);
1134 struct tpacket_block_desc
*pbd
= GET_PBLOCK_DESC(pkc
, idx
);
1136 if (status
!= BLOCK_STATUS(pbd
))
1141 static int prb_previous_blk_num(struct packet_ring_buffer
*rb
)
1144 if (rb
->prb_bdqc
.kactive_blk_num
)
1145 prev
= rb
->prb_bdqc
.kactive_blk_num
-1;
1147 prev
= rb
->prb_bdqc
.knum_blocks
-1;
1151 /* Assumes caller has held the rx_queue.lock */
1152 static void *__prb_previous_block(struct packet_sock
*po
,
1153 struct packet_ring_buffer
*rb
,
1156 unsigned int previous
= prb_previous_blk_num(rb
);
1157 return prb_lookup_block(po
, rb
, previous
, status
);
1160 static void *packet_previous_rx_frame(struct packet_sock
*po
,
1161 struct packet_ring_buffer
*rb
,
1164 if (po
->tp_version
<= TPACKET_V2
)
1165 return packet_previous_frame(po
, rb
, status
);
1167 return __prb_previous_block(po
, rb
, status
);
1170 static void packet_increment_rx_head(struct packet_sock
*po
,
1171 struct packet_ring_buffer
*rb
)
1173 switch (po
->tp_version
) {
1176 return packet_increment_head(rb
);
1179 WARN(1, "TPACKET version not supported.\n");
1185 static void *packet_previous_frame(struct packet_sock
*po
,
1186 struct packet_ring_buffer
*rb
,
1189 unsigned int previous
= rb
->head
? rb
->head
- 1 : rb
->frame_max
;
1190 return packet_lookup_frame(po
, rb
, previous
, status
);
1193 static void packet_increment_head(struct packet_ring_buffer
*buff
)
1195 buff
->head
= buff
->head
!= buff
->frame_max
? buff
->head
+1 : 0;
1198 static void packet_inc_pending(struct packet_ring_buffer
*rb
)
1200 this_cpu_inc(*rb
->pending_refcnt
);
1203 static void packet_dec_pending(struct packet_ring_buffer
*rb
)
1205 this_cpu_dec(*rb
->pending_refcnt
);
1208 static unsigned int packet_read_pending(const struct packet_ring_buffer
*rb
)
1210 unsigned int refcnt
= 0;
1213 /* We don't use pending refcount in rx_ring. */
1214 if (rb
->pending_refcnt
== NULL
)
1217 for_each_possible_cpu(cpu
)
1218 refcnt
+= *per_cpu_ptr(rb
->pending_refcnt
, cpu
);
1223 static int packet_alloc_pending(struct packet_sock
*po
)
1225 po
->rx_ring
.pending_refcnt
= NULL
;
1227 po
->tx_ring
.pending_refcnt
= alloc_percpu(unsigned int);
1228 if (unlikely(po
->tx_ring
.pending_refcnt
== NULL
))
1234 static void packet_free_pending(struct packet_sock
*po
)
1236 free_percpu(po
->tx_ring
.pending_refcnt
);
1239 #define ROOM_POW_OFF 2
1240 #define ROOM_NONE 0x0
1241 #define ROOM_LOW 0x1
1242 #define ROOM_NORMAL 0x2
1244 static bool __tpacket_has_room(struct packet_sock
*po
, int pow_off
)
1248 len
= po
->rx_ring
.frame_max
+ 1;
1249 idx
= po
->rx_ring
.head
;
1251 idx
+= len
>> pow_off
;
1254 return packet_lookup_frame(po
, &po
->rx_ring
, idx
, TP_STATUS_KERNEL
);
1257 static bool __tpacket_v3_has_room(struct packet_sock
*po
, int pow_off
)
1261 len
= po
->rx_ring
.prb_bdqc
.knum_blocks
;
1262 idx
= po
->rx_ring
.prb_bdqc
.kactive_blk_num
;
1264 idx
+= len
>> pow_off
;
1267 return prb_lookup_block(po
, &po
->rx_ring
, idx
, TP_STATUS_KERNEL
);
1270 static int __packet_rcv_has_room(struct packet_sock
*po
, struct sk_buff
*skb
)
1272 struct sock
*sk
= &po
->sk
;
1273 int ret
= ROOM_NONE
;
1275 if (po
->prot_hook
.func
!= tpacket_rcv
) {
1276 int avail
= sk
->sk_rcvbuf
- atomic_read(&sk
->sk_rmem_alloc
)
1277 - (skb
? skb
->truesize
: 0);
1278 if (avail
> (sk
->sk_rcvbuf
>> ROOM_POW_OFF
))
1286 if (po
->tp_version
== TPACKET_V3
) {
1287 if (__tpacket_v3_has_room(po
, ROOM_POW_OFF
))
1289 else if (__tpacket_v3_has_room(po
, 0))
1292 if (__tpacket_has_room(po
, ROOM_POW_OFF
))
1294 else if (__tpacket_has_room(po
, 0))
1301 static int packet_rcv_has_room(struct packet_sock
*po
, struct sk_buff
*skb
)
1306 spin_lock_bh(&po
->sk
.sk_receive_queue
.lock
);
1307 ret
= __packet_rcv_has_room(po
, skb
);
1308 has_room
= ret
== ROOM_NORMAL
;
1309 if (po
->pressure
== has_room
)
1310 po
->pressure
= !has_room
;
1311 spin_unlock_bh(&po
->sk
.sk_receive_queue
.lock
);
1316 static void packet_sock_destruct(struct sock
*sk
)
1318 skb_queue_purge(&sk
->sk_error_queue
);
1320 WARN_ON(atomic_read(&sk
->sk_rmem_alloc
));
1321 WARN_ON(atomic_read(&sk
->sk_wmem_alloc
));
1323 if (!sock_flag(sk
, SOCK_DEAD
)) {
1324 pr_err("Attempt to release alive packet socket: %p\n", sk
);
1328 sk_refcnt_debug_dec(sk
);
1331 static bool fanout_flow_is_huge(struct packet_sock
*po
, struct sk_buff
*skb
)
1336 rxhash
= skb_get_hash(skb
);
1337 for (i
= 0; i
< ROLLOVER_HLEN
; i
++)
1338 if (po
->rollover
->history
[i
] == rxhash
)
1341 po
->rollover
->history
[prandom_u32() % ROLLOVER_HLEN
] = rxhash
;
1342 return count
> (ROLLOVER_HLEN
>> 1);
1345 static unsigned int fanout_demux_hash(struct packet_fanout
*f
,
1346 struct sk_buff
*skb
,
1349 return reciprocal_scale(__skb_get_hash_symmetric(skb
), num
);
1352 static unsigned int fanout_demux_lb(struct packet_fanout
*f
,
1353 struct sk_buff
*skb
,
1356 unsigned int val
= atomic_inc_return(&f
->rr_cur
);
1361 static unsigned int fanout_demux_cpu(struct packet_fanout
*f
,
1362 struct sk_buff
*skb
,
1365 return smp_processor_id() % num
;
1368 static unsigned int fanout_demux_rnd(struct packet_fanout
*f
,
1369 struct sk_buff
*skb
,
1372 return prandom_u32_max(num
);
1375 static unsigned int fanout_demux_rollover(struct packet_fanout
*f
,
1376 struct sk_buff
*skb
,
1377 unsigned int idx
, bool try_self
,
1380 struct packet_sock
*po
, *po_next
, *po_skip
= NULL
;
1381 unsigned int i
, j
, room
= ROOM_NONE
;
1383 po
= pkt_sk(f
->arr
[idx
]);
1386 room
= packet_rcv_has_room(po
, skb
);
1387 if (room
== ROOM_NORMAL
||
1388 (room
== ROOM_LOW
&& !fanout_flow_is_huge(po
, skb
)))
1393 i
= j
= min_t(int, po
->rollover
->sock
, num
- 1);
1395 po_next
= pkt_sk(f
->arr
[i
]);
1396 if (po_next
!= po_skip
&& !po_next
->pressure
&&
1397 packet_rcv_has_room(po_next
, skb
) == ROOM_NORMAL
) {
1399 po
->rollover
->sock
= i
;
1400 atomic_long_inc(&po
->rollover
->num
);
1401 if (room
== ROOM_LOW
)
1402 atomic_long_inc(&po
->rollover
->num_huge
);
1410 atomic_long_inc(&po
->rollover
->num_failed
);
1414 static unsigned int fanout_demux_qm(struct packet_fanout
*f
,
1415 struct sk_buff
*skb
,
1418 return skb_get_queue_mapping(skb
) % num
;
1421 static unsigned int fanout_demux_bpf(struct packet_fanout
*f
,
1422 struct sk_buff
*skb
,
1425 struct bpf_prog
*prog
;
1426 unsigned int ret
= 0;
1429 prog
= rcu_dereference(f
->bpf_prog
);
1431 ret
= bpf_prog_run_clear_cb(prog
, skb
) % num
;
1437 static bool fanout_has_flag(struct packet_fanout
*f
, u16 flag
)
1439 return f
->flags
& (flag
>> 8);
1442 static int packet_rcv_fanout(struct sk_buff
*skb
, struct net_device
*dev
,
1443 struct packet_type
*pt
, struct net_device
*orig_dev
)
1445 struct packet_fanout
*f
= pt
->af_packet_priv
;
1446 unsigned int num
= READ_ONCE(f
->num_members
);
1447 struct net
*net
= read_pnet(&f
->net
);
1448 struct packet_sock
*po
;
1451 if (!net_eq(dev_net(dev
), net
) || !num
) {
1456 if (fanout_has_flag(f
, PACKET_FANOUT_FLAG_DEFRAG
)) {
1457 skb
= ip_check_defrag(net
, skb
, IP_DEFRAG_AF_PACKET
);
1462 case PACKET_FANOUT_HASH
:
1464 idx
= fanout_demux_hash(f
, skb
, num
);
1466 case PACKET_FANOUT_LB
:
1467 idx
= fanout_demux_lb(f
, skb
, num
);
1469 case PACKET_FANOUT_CPU
:
1470 idx
= fanout_demux_cpu(f
, skb
, num
);
1472 case PACKET_FANOUT_RND
:
1473 idx
= fanout_demux_rnd(f
, skb
, num
);
1475 case PACKET_FANOUT_QM
:
1476 idx
= fanout_demux_qm(f
, skb
, num
);
1478 case PACKET_FANOUT_ROLLOVER
:
1479 idx
= fanout_demux_rollover(f
, skb
, 0, false, num
);
1481 case PACKET_FANOUT_CBPF
:
1482 case PACKET_FANOUT_EBPF
:
1483 idx
= fanout_demux_bpf(f
, skb
, num
);
1487 if (fanout_has_flag(f
, PACKET_FANOUT_FLAG_ROLLOVER
))
1488 idx
= fanout_demux_rollover(f
, skb
, idx
, true, num
);
1490 po
= pkt_sk(f
->arr
[idx
]);
1491 return po
->prot_hook
.func(skb
, dev
, &po
->prot_hook
, orig_dev
);
1494 DEFINE_MUTEX(fanout_mutex
);
1495 EXPORT_SYMBOL_GPL(fanout_mutex
);
1496 static LIST_HEAD(fanout_list
);
1497 static u16 fanout_next_id
;
1499 static void __fanout_link(struct sock
*sk
, struct packet_sock
*po
)
1501 struct packet_fanout
*f
= po
->fanout
;
1503 spin_lock(&f
->lock
);
1504 f
->arr
[f
->num_members
] = sk
;
1507 if (f
->num_members
== 1)
1508 dev_add_pack(&f
->prot_hook
);
1509 spin_unlock(&f
->lock
);
1512 static void __fanout_unlink(struct sock
*sk
, struct packet_sock
*po
)
1514 struct packet_fanout
*f
= po
->fanout
;
1517 spin_lock(&f
->lock
);
1518 for (i
= 0; i
< f
->num_members
; i
++) {
1519 if (f
->arr
[i
] == sk
)
1522 BUG_ON(i
>= f
->num_members
);
1523 f
->arr
[i
] = f
->arr
[f
->num_members
- 1];
1525 if (f
->num_members
== 0)
1526 __dev_remove_pack(&f
->prot_hook
);
1527 spin_unlock(&f
->lock
);
1530 static bool match_fanout_group(struct packet_type
*ptype
, struct sock
*sk
)
1532 if (sk
->sk_family
!= PF_PACKET
)
1535 return ptype
->af_packet_priv
== pkt_sk(sk
)->fanout
;
1538 static void fanout_init_data(struct packet_fanout
*f
)
1541 case PACKET_FANOUT_LB
:
1542 atomic_set(&f
->rr_cur
, 0);
1544 case PACKET_FANOUT_CBPF
:
1545 case PACKET_FANOUT_EBPF
:
1546 RCU_INIT_POINTER(f
->bpf_prog
, NULL
);
1551 static void __fanout_set_data_bpf(struct packet_fanout
*f
, struct bpf_prog
*new)
1553 struct bpf_prog
*old
;
1555 spin_lock(&f
->lock
);
1556 old
= rcu_dereference_protected(f
->bpf_prog
, lockdep_is_held(&f
->lock
));
1557 rcu_assign_pointer(f
->bpf_prog
, new);
1558 spin_unlock(&f
->lock
);
1562 bpf_prog_destroy(old
);
1566 static int fanout_set_data_cbpf(struct packet_sock
*po
, char __user
*data
,
1569 struct bpf_prog
*new;
1570 struct sock_fprog fprog
;
1573 if (sock_flag(&po
->sk
, SOCK_FILTER_LOCKED
))
1575 if (len
!= sizeof(fprog
))
1577 if (copy_from_user(&fprog
, data
, len
))
1580 ret
= bpf_prog_create_from_user(&new, &fprog
, NULL
, false);
1584 __fanout_set_data_bpf(po
->fanout
, new);
1588 static int fanout_set_data_ebpf(struct packet_sock
*po
, char __user
*data
,
1591 struct bpf_prog
*new;
1594 if (sock_flag(&po
->sk
, SOCK_FILTER_LOCKED
))
1596 if (len
!= sizeof(fd
))
1598 if (copy_from_user(&fd
, data
, len
))
1601 new = bpf_prog_get_type(fd
, BPF_PROG_TYPE_SOCKET_FILTER
);
1603 return PTR_ERR(new);
1605 __fanout_set_data_bpf(po
->fanout
, new);
1609 static int fanout_set_data(struct packet_sock
*po
, char __user
*data
,
1612 switch (po
->fanout
->type
) {
1613 case PACKET_FANOUT_CBPF
:
1614 return fanout_set_data_cbpf(po
, data
, len
);
1615 case PACKET_FANOUT_EBPF
:
1616 return fanout_set_data_ebpf(po
, data
, len
);
1622 static void fanout_release_data(struct packet_fanout
*f
)
1625 case PACKET_FANOUT_CBPF
:
1626 case PACKET_FANOUT_EBPF
:
1627 __fanout_set_data_bpf(f
, NULL
);
1631 static bool __fanout_id_is_free(struct sock
*sk
, u16 candidate_id
)
1633 struct packet_fanout
*f
;
1635 list_for_each_entry(f
, &fanout_list
, list
) {
1636 if (f
->id
== candidate_id
&&
1637 read_pnet(&f
->net
) == sock_net(sk
)) {
1644 static bool fanout_find_new_id(struct sock
*sk
, u16
*new_id
)
1646 u16 id
= fanout_next_id
;
1649 if (__fanout_id_is_free(sk
, id
)) {
1651 fanout_next_id
= id
+ 1;
1656 } while (id
!= fanout_next_id
);
1661 static int fanout_add(struct sock
*sk
, u16 id
, u16 type_flags
)
1663 struct packet_rollover
*rollover
= NULL
;
1664 struct packet_sock
*po
= pkt_sk(sk
);
1665 struct packet_fanout
*f
, *match
;
1666 u8 type
= type_flags
& 0xff;
1667 u8 flags
= type_flags
>> 8;
1671 case PACKET_FANOUT_ROLLOVER
:
1672 if (type_flags
& PACKET_FANOUT_FLAG_ROLLOVER
)
1674 case PACKET_FANOUT_HASH
:
1675 case PACKET_FANOUT_LB
:
1676 case PACKET_FANOUT_CPU
:
1677 case PACKET_FANOUT_RND
:
1678 case PACKET_FANOUT_QM
:
1679 case PACKET_FANOUT_CBPF
:
1680 case PACKET_FANOUT_EBPF
:
1686 mutex_lock(&fanout_mutex
);
1696 if (type
== PACKET_FANOUT_ROLLOVER
||
1697 (type_flags
& PACKET_FANOUT_FLAG_ROLLOVER
)) {
1699 rollover
= kzalloc(sizeof(*rollover
), GFP_KERNEL
);
1702 atomic_long_set(&rollover
->num
, 0);
1703 atomic_long_set(&rollover
->num_huge
, 0);
1704 atomic_long_set(&rollover
->num_failed
, 0);
1705 po
->rollover
= rollover
;
1708 if (type_flags
& PACKET_FANOUT_FLAG_UNIQUEID
) {
1713 if (!fanout_find_new_id(sk
, &id
)) {
1717 /* ephemeral flag for the first socket in the group: drop it */
1718 flags
&= ~(PACKET_FANOUT_FLAG_UNIQUEID
>> 8);
1722 list_for_each_entry(f
, &fanout_list
, list
) {
1724 read_pnet(&f
->net
) == sock_net(sk
)) {
1730 if (match
&& match
->flags
!= flags
)
1734 match
= kzalloc(sizeof(*match
), GFP_KERNEL
);
1737 write_pnet(&match
->net
, sock_net(sk
));
1740 match
->flags
= flags
;
1741 INIT_LIST_HEAD(&match
->list
);
1742 spin_lock_init(&match
->lock
);
1743 atomic_set(&match
->sk_ref
, 0);
1744 fanout_init_data(match
);
1745 match
->prot_hook
.type
= po
->prot_hook
.type
;
1746 match
->prot_hook
.dev
= po
->prot_hook
.dev
;
1747 match
->prot_hook
.func
= packet_rcv_fanout
;
1748 match
->prot_hook
.af_packet_priv
= match
;
1749 match
->prot_hook
.id_match
= match_fanout_group
;
1750 list_add(&match
->list
, &fanout_list
);
1753 if (match
->type
== type
&&
1754 match
->prot_hook
.type
== po
->prot_hook
.type
&&
1755 match
->prot_hook
.dev
== po
->prot_hook
.dev
) {
1757 if (atomic_read(&match
->sk_ref
) < PACKET_FANOUT_MAX
) {
1758 __dev_remove_pack(&po
->prot_hook
);
1760 atomic_inc(&match
->sk_ref
);
1761 __fanout_link(sk
, po
);
1766 if (err
&& rollover
) {
1768 po
->rollover
= NULL
;
1770 mutex_unlock(&fanout_mutex
);
1774 /* If pkt_sk(sk)->fanout->sk_ref is zero, this function removes
1775 * pkt_sk(sk)->fanout from fanout_list and returns pkt_sk(sk)->fanout.
1776 * It is the responsibility of the caller to call fanout_release_data() and
1777 * free the returned packet_fanout (after synchronize_net())
1779 static struct packet_fanout
*fanout_release(struct sock
*sk
)
1781 struct packet_sock
*po
= pkt_sk(sk
);
1782 struct packet_fanout
*f
;
1784 mutex_lock(&fanout_mutex
);
1789 if (atomic_dec_and_test(&f
->sk_ref
))
1795 kfree_rcu(po
->rollover
, rcu
);
1797 mutex_unlock(&fanout_mutex
);
1802 static bool packet_extra_vlan_len_allowed(const struct net_device
*dev
,
1803 struct sk_buff
*skb
)
1805 /* Earlier code assumed this would be a VLAN pkt, double-check
1806 * this now that we have the actual packet in hand. We can only
1807 * do this check on Ethernet devices.
1809 if (unlikely(dev
->type
!= ARPHRD_ETHER
))
1812 skb_reset_mac_header(skb
);
1813 return likely(eth_hdr(skb
)->h_proto
== htons(ETH_P_8021Q
));
1816 static const struct proto_ops packet_ops
;
1818 static const struct proto_ops packet_ops_spkt
;
1820 static int packet_rcv_spkt(struct sk_buff
*skb
, struct net_device
*dev
,
1821 struct packet_type
*pt
, struct net_device
*orig_dev
)
1824 struct sockaddr_pkt
*spkt
;
1827 * When we registered the protocol we saved the socket in the data
1828 * field for just this event.
1831 sk
= pt
->af_packet_priv
;
1834 * Yank back the headers [hope the device set this
1835 * right or kerboom...]
1837 * Incoming packets have ll header pulled,
1840 * For outgoing ones skb->data == skb_mac_header(skb)
1841 * so that this procedure is noop.
1844 if (skb
->pkt_type
== PACKET_LOOPBACK
)
1847 if (!net_eq(dev_net(dev
), sock_net(sk
)))
1850 skb
= skb_share_check(skb
, GFP_ATOMIC
);
1854 /* drop any routing info */
1857 /* drop conntrack reference */
1860 spkt
= &PACKET_SKB_CB(skb
)->sa
.pkt
;
1862 skb_push(skb
, skb
->data
- skb_mac_header(skb
));
1865 * The SOCK_PACKET socket receives _all_ frames.
1868 spkt
->spkt_family
= dev
->type
;
1869 strlcpy(spkt
->spkt_device
, dev
->name
, sizeof(spkt
->spkt_device
));
1870 spkt
->spkt_protocol
= skb
->protocol
;
1873 * Charge the memory to the socket. This is done specifically
1874 * to prevent sockets using all the memory up.
1877 if (sock_queue_rcv_skb(sk
, skb
) == 0)
1888 * Output a raw packet to a device layer. This bypasses all the other
1889 * protocol layers and you must therefore supply it with a complete frame
1892 static int packet_sendmsg_spkt(struct socket
*sock
, struct msghdr
*msg
,
1895 struct sock
*sk
= sock
->sk
;
1896 DECLARE_SOCKADDR(struct sockaddr_pkt
*, saddr
, msg
->msg_name
);
1897 struct sk_buff
*skb
= NULL
;
1898 struct net_device
*dev
;
1899 struct sockcm_cookie sockc
;
1905 * Get and verify the address.
1909 if (msg
->msg_namelen
< sizeof(struct sockaddr
))
1911 if (msg
->msg_namelen
== sizeof(struct sockaddr_pkt
))
1912 proto
= saddr
->spkt_protocol
;
1914 return -ENOTCONN
; /* SOCK_PACKET must be sent giving an address */
1917 * Find the device first to size check it
1920 saddr
->spkt_device
[sizeof(saddr
->spkt_device
) - 1] = 0;
1923 dev
= dev_get_by_name_rcu(sock_net(sk
), saddr
->spkt_device
);
1929 if (!(dev
->flags
& IFF_UP
))
1933 * You may not queue a frame bigger than the mtu. This is the lowest level
1934 * raw protocol and you must do your own fragmentation at this level.
1937 if (unlikely(sock_flag(sk
, SOCK_NOFCS
))) {
1938 if (!netif_supports_nofcs(dev
)) {
1939 err
= -EPROTONOSUPPORT
;
1942 extra_len
= 4; /* We're doing our own CRC */
1946 if (len
> dev
->mtu
+ dev
->hard_header_len
+ VLAN_HLEN
+ extra_len
)
1950 size_t reserved
= LL_RESERVED_SPACE(dev
);
1951 int tlen
= dev
->needed_tailroom
;
1952 unsigned int hhlen
= dev
->header_ops
? dev
->hard_header_len
: 0;
1955 skb
= sock_wmalloc(sk
, len
+ reserved
+ tlen
, 0, GFP_KERNEL
);
1958 /* FIXME: Save some space for broken drivers that write a hard
1959 * header at transmission time by themselves. PPP is the notable
1960 * one here. This should really be fixed at the driver level.
1962 skb_reserve(skb
, reserved
);
1963 skb_reset_network_header(skb
);
1965 /* Try to align data part correctly */
1970 skb_reset_network_header(skb
);
1972 err
= memcpy_from_msg(skb_put(skb
, len
), msg
, len
);
1978 if (!dev_validate_header(dev
, skb
->data
, len
)) {
1982 if (len
> (dev
->mtu
+ dev
->hard_header_len
+ extra_len
) &&
1983 !packet_extra_vlan_len_allowed(dev
, skb
)) {
1988 sockc
.tsflags
= sk
->sk_tsflags
;
1989 if (msg
->msg_controllen
) {
1990 err
= sock_cmsg_send(sk
, msg
, &sockc
);
1995 skb
->protocol
= proto
;
1997 skb
->priority
= sk
->sk_priority
;
1998 skb
->mark
= sk
->sk_mark
;
2000 sock_tx_timestamp(sk
, sockc
.tsflags
, &skb_shinfo(skb
)->tx_flags
);
2002 if (unlikely(extra_len
== 4))
2005 skb_probe_transport_header(skb
, 0);
2007 dev_queue_xmit(skb
);
2018 static unsigned int run_filter(struct sk_buff
*skb
,
2019 const struct sock
*sk
,
2022 struct sk_filter
*filter
;
2025 filter
= rcu_dereference(sk
->sk_filter
);
2027 res
= bpf_prog_run_clear_cb(filter
->prog
, skb
);
2033 static int packet_rcv_vnet(struct msghdr
*msg
, const struct sk_buff
*skb
,
2036 struct virtio_net_hdr vnet_hdr
;
2038 if (*len
< sizeof(vnet_hdr
))
2040 *len
-= sizeof(vnet_hdr
);
2042 if (virtio_net_hdr_from_skb(skb
, &vnet_hdr
, vio_le(), true))
2045 return memcpy_to_msg(msg
, (void *)&vnet_hdr
, sizeof(vnet_hdr
));
2049 * This function makes lazy skb cloning in hope that most of packets
2050 * are discarded by BPF.
2052 * Note tricky part: we DO mangle shared skb! skb->data, skb->len
2053 * and skb->cb are mangled. It works because (and until) packets
2054 * falling here are owned by current CPU. Output packets are cloned
2055 * by dev_queue_xmit_nit(), input packets are processed by net_bh
2056 * sequencially, so that if we return skb to original state on exit,
2057 * we will not harm anyone.
2060 static int packet_rcv(struct sk_buff
*skb
, struct net_device
*dev
,
2061 struct packet_type
*pt
, struct net_device
*orig_dev
)
2064 struct sockaddr_ll
*sll
;
2065 struct packet_sock
*po
;
2066 u8
*skb_head
= skb
->data
;
2067 int skb_len
= skb
->len
;
2068 unsigned int snaplen
, res
;
2069 bool is_drop_n_account
= false;
2071 if (skb
->pkt_type
== PACKET_LOOPBACK
)
2074 sk
= pt
->af_packet_priv
;
2077 if (!net_eq(dev_net(dev
), sock_net(sk
)))
2082 if (dev
->header_ops
) {
2083 /* The device has an explicit notion of ll header,
2084 * exported to higher levels.
2086 * Otherwise, the device hides details of its frame
2087 * structure, so that corresponding packet head is
2088 * never delivered to user.
2090 if (sk
->sk_type
!= SOCK_DGRAM
)
2091 skb_push(skb
, skb
->data
- skb_mac_header(skb
));
2092 else if (skb
->pkt_type
== PACKET_OUTGOING
) {
2093 /* Special case: outgoing packets have ll header at head */
2094 skb_pull(skb
, skb_network_offset(skb
));
2100 res
= run_filter(skb
, sk
, snaplen
);
2102 goto drop_n_restore
;
2106 if (atomic_read(&sk
->sk_rmem_alloc
) >= sk
->sk_rcvbuf
)
2109 if (skb_shared(skb
)) {
2110 struct sk_buff
*nskb
= skb_clone(skb
, GFP_ATOMIC
);
2114 if (skb_head
!= skb
->data
) {
2115 skb
->data
= skb_head
;
2122 sock_skb_cb_check_size(sizeof(*PACKET_SKB_CB(skb
)) + MAX_ADDR_LEN
- 8);
2124 sll
= &PACKET_SKB_CB(skb
)->sa
.ll
;
2125 sll
->sll_hatype
= dev
->type
;
2126 sll
->sll_pkttype
= skb
->pkt_type
;
2127 if (unlikely(po
->origdev
))
2128 sll
->sll_ifindex
= orig_dev
->ifindex
;
2130 sll
->sll_ifindex
= dev
->ifindex
;
2132 sll
->sll_halen
= dev_parse_header(skb
, sll
->sll_addr
);
2134 /* sll->sll_family and sll->sll_protocol are set in packet_recvmsg().
2135 * Use their space for storing the original skb length.
2137 PACKET_SKB_CB(skb
)->sa
.origlen
= skb
->len
;
2139 if (pskb_trim(skb
, snaplen
))
2142 skb_set_owner_r(skb
, sk
);
2146 /* drop conntrack reference */
2149 spin_lock(&sk
->sk_receive_queue
.lock
);
2150 po
->stats
.stats1
.tp_packets
++;
2151 sock_skb_set_dropcount(sk
, skb
);
2152 __skb_queue_tail(&sk
->sk_receive_queue
, skb
);
2153 spin_unlock(&sk
->sk_receive_queue
.lock
);
2154 sk
->sk_data_ready(sk
);
2158 is_drop_n_account
= true;
2159 spin_lock(&sk
->sk_receive_queue
.lock
);
2160 po
->stats
.stats1
.tp_drops
++;
2161 atomic_inc(&sk
->sk_drops
);
2162 spin_unlock(&sk
->sk_receive_queue
.lock
);
2165 if (skb_head
!= skb
->data
&& skb_shared(skb
)) {
2166 skb
->data
= skb_head
;
2170 if (!is_drop_n_account
)
2177 static int tpacket_rcv(struct sk_buff
*skb
, struct net_device
*dev
,
2178 struct packet_type
*pt
, struct net_device
*orig_dev
)
2181 struct packet_sock
*po
;
2182 struct sockaddr_ll
*sll
;
2183 union tpacket_uhdr h
;
2184 u8
*skb_head
= skb
->data
;
2185 int skb_len
= skb
->len
;
2186 unsigned int snaplen
, res
;
2187 unsigned long status
= TP_STATUS_USER
;
2188 unsigned short macoff
, netoff
, hdrlen
;
2189 struct sk_buff
*copy_skb
= NULL
;
2192 bool is_drop_n_account
= false;
2194 /* struct tpacket{2,3}_hdr is aligned to a multiple of TPACKET_ALIGNMENT.
2195 * We may add members to them until current aligned size without forcing
2196 * userspace to call getsockopt(..., PACKET_HDRLEN, ...).
2198 BUILD_BUG_ON(TPACKET_ALIGN(sizeof(*h
.h2
)) != 32);
2199 BUILD_BUG_ON(TPACKET_ALIGN(sizeof(*h
.h3
)) != 48);
2201 if (skb
->pkt_type
== PACKET_LOOPBACK
)
2204 sk
= pt
->af_packet_priv
;
2207 if (!net_eq(dev_net(dev
), sock_net(sk
)))
2210 if (dev
->header_ops
) {
2211 if (sk
->sk_type
!= SOCK_DGRAM
)
2212 skb_push(skb
, skb
->data
- skb_mac_header(skb
));
2213 else if (skb
->pkt_type
== PACKET_OUTGOING
) {
2214 /* Special case: outgoing packets have ll header at head */
2215 skb_pull(skb
, skb_network_offset(skb
));
2221 res
= run_filter(skb
, sk
, snaplen
);
2223 goto drop_n_restore
;
2225 if (skb
->ip_summed
== CHECKSUM_PARTIAL
)
2226 status
|= TP_STATUS_CSUMNOTREADY
;
2227 else if (skb
->pkt_type
!= PACKET_OUTGOING
&&
2228 (skb
->ip_summed
== CHECKSUM_COMPLETE
||
2229 skb_csum_unnecessary(skb
)))
2230 status
|= TP_STATUS_CSUM_VALID
;
2235 if (sk
->sk_type
== SOCK_DGRAM
) {
2236 macoff
= netoff
= TPACKET_ALIGN(po
->tp_hdrlen
) + 16 +
2239 unsigned int maclen
= skb_network_offset(skb
);
2240 netoff
= TPACKET_ALIGN(po
->tp_hdrlen
+
2241 (maclen
< 16 ? 16 : maclen
)) +
2243 if (po
->has_vnet_hdr
)
2244 netoff
+= sizeof(struct virtio_net_hdr
);
2245 macoff
= netoff
- maclen
;
2247 if (po
->tp_version
<= TPACKET_V2
) {
2248 if (macoff
+ snaplen
> po
->rx_ring
.frame_size
) {
2249 if (po
->copy_thresh
&&
2250 atomic_read(&sk
->sk_rmem_alloc
) < sk
->sk_rcvbuf
) {
2251 if (skb_shared(skb
)) {
2252 copy_skb
= skb_clone(skb
, GFP_ATOMIC
);
2254 copy_skb
= skb_get(skb
);
2255 skb_head
= skb
->data
;
2258 skb_set_owner_r(copy_skb
, sk
);
2260 snaplen
= po
->rx_ring
.frame_size
- macoff
;
2261 if ((int)snaplen
< 0)
2264 } else if (unlikely(macoff
+ snaplen
>
2265 GET_PBDQC_FROM_RB(&po
->rx_ring
)->max_frame_len
)) {
2268 nval
= GET_PBDQC_FROM_RB(&po
->rx_ring
)->max_frame_len
- macoff
;
2269 pr_err_once("tpacket_rcv: packet too big, clamped from %u to %u. macoff=%u\n",
2270 snaplen
, nval
, macoff
);
2272 if (unlikely((int)snaplen
< 0)) {
2274 macoff
= GET_PBDQC_FROM_RB(&po
->rx_ring
)->max_frame_len
;
2277 spin_lock(&sk
->sk_receive_queue
.lock
);
2278 h
.raw
= packet_current_rx_frame(po
, skb
,
2279 TP_STATUS_KERNEL
, (macoff
+snaplen
));
2281 goto drop_n_account
;
2282 if (po
->tp_version
<= TPACKET_V2
) {
2283 packet_increment_rx_head(po
, &po
->rx_ring
);
2285 * LOSING will be reported till you read the stats,
2286 * because it's COR - Clear On Read.
2287 * Anyways, moving it for V1/V2 only as V3 doesn't need this
2290 if (po
->stats
.stats1
.tp_drops
)
2291 status
|= TP_STATUS_LOSING
;
2293 po
->stats
.stats1
.tp_packets
++;
2295 status
|= TP_STATUS_COPY
;
2296 __skb_queue_tail(&sk
->sk_receive_queue
, copy_skb
);
2298 spin_unlock(&sk
->sk_receive_queue
.lock
);
2300 if (po
->has_vnet_hdr
) {
2301 if (virtio_net_hdr_from_skb(skb
, h
.raw
+ macoff
-
2302 sizeof(struct virtio_net_hdr
),
2304 spin_lock(&sk
->sk_receive_queue
.lock
);
2305 goto drop_n_account
;
2309 skb_copy_bits(skb
, 0, h
.raw
+ macoff
, snaplen
);
2311 if (!(ts_status
= tpacket_get_timestamp(skb
, &ts
, po
->tp_tstamp
)))
2312 getnstimeofday(&ts
);
2314 status
|= ts_status
;
2316 switch (po
->tp_version
) {
2318 h
.h1
->tp_len
= skb
->len
;
2319 h
.h1
->tp_snaplen
= snaplen
;
2320 h
.h1
->tp_mac
= macoff
;
2321 h
.h1
->tp_net
= netoff
;
2322 h
.h1
->tp_sec
= ts
.tv_sec
;
2323 h
.h1
->tp_usec
= ts
.tv_nsec
/ NSEC_PER_USEC
;
2324 hdrlen
= sizeof(*h
.h1
);
2327 h
.h2
->tp_len
= skb
->len
;
2328 h
.h2
->tp_snaplen
= snaplen
;
2329 h
.h2
->tp_mac
= macoff
;
2330 h
.h2
->tp_net
= netoff
;
2331 h
.h2
->tp_sec
= ts
.tv_sec
;
2332 h
.h2
->tp_nsec
= ts
.tv_nsec
;
2333 if (skb_vlan_tag_present(skb
)) {
2334 h
.h2
->tp_vlan_tci
= skb_vlan_tag_get(skb
);
2335 h
.h2
->tp_vlan_tpid
= ntohs(skb
->vlan_proto
);
2336 status
|= TP_STATUS_VLAN_VALID
| TP_STATUS_VLAN_TPID_VALID
;
2338 h
.h2
->tp_vlan_tci
= 0;
2339 h
.h2
->tp_vlan_tpid
= 0;
2341 memset(h
.h2
->tp_padding
, 0, sizeof(h
.h2
->tp_padding
));
2342 hdrlen
= sizeof(*h
.h2
);
2345 /* tp_nxt_offset,vlan are already populated above.
2346 * So DONT clear those fields here
2348 h
.h3
->tp_status
|= status
;
2349 h
.h3
->tp_len
= skb
->len
;
2350 h
.h3
->tp_snaplen
= snaplen
;
2351 h
.h3
->tp_mac
= macoff
;
2352 h
.h3
->tp_net
= netoff
;
2353 h
.h3
->tp_sec
= ts
.tv_sec
;
2354 h
.h3
->tp_nsec
= ts
.tv_nsec
;
2355 memset(h
.h3
->tp_padding
, 0, sizeof(h
.h3
->tp_padding
));
2356 hdrlen
= sizeof(*h
.h3
);
2362 sll
= h
.raw
+ TPACKET_ALIGN(hdrlen
);
2363 sll
->sll_halen
= dev_parse_header(skb
, sll
->sll_addr
);
2364 sll
->sll_family
= AF_PACKET
;
2365 sll
->sll_hatype
= dev
->type
;
2366 sll
->sll_protocol
= skb
->protocol
;
2367 sll
->sll_pkttype
= skb
->pkt_type
;
2368 if (unlikely(po
->origdev
))
2369 sll
->sll_ifindex
= orig_dev
->ifindex
;
2371 sll
->sll_ifindex
= dev
->ifindex
;
2375 #if ARCH_IMPLEMENTS_FLUSH_DCACHE_PAGE == 1
2376 if (po
->tp_version
<= TPACKET_V2
) {
2379 end
= (u8
*) PAGE_ALIGN((unsigned long) h
.raw
+
2382 for (start
= h
.raw
; start
< end
; start
+= PAGE_SIZE
)
2383 flush_dcache_page(pgv_to_page(start
));
2388 if (po
->tp_version
<= TPACKET_V2
) {
2389 __packet_set_status(po
, h
.raw
, status
);
2390 sk
->sk_data_ready(sk
);
2392 prb_clear_blk_fill_status(&po
->rx_ring
);
2396 if (skb_head
!= skb
->data
&& skb_shared(skb
)) {
2397 skb
->data
= skb_head
;
2401 if (!is_drop_n_account
)
2408 is_drop_n_account
= true;
2409 po
->stats
.stats1
.tp_drops
++;
2410 spin_unlock(&sk
->sk_receive_queue
.lock
);
2412 sk
->sk_data_ready(sk
);
2413 kfree_skb(copy_skb
);
2414 goto drop_n_restore
;
2417 static void tpacket_destruct_skb(struct sk_buff
*skb
)
2419 struct packet_sock
*po
= pkt_sk(skb
->sk
);
2421 if (likely(po
->tx_ring
.pg_vec
)) {
2425 ph
= skb_shinfo(skb
)->destructor_arg
;
2426 packet_dec_pending(&po
->tx_ring
);
2428 ts
= __packet_set_timestamp(po
, ph
, skb
);
2429 __packet_set_status(po
, ph
, TP_STATUS_AVAILABLE
| ts
);
2435 static void tpacket_set_protocol(const struct net_device
*dev
,
2436 struct sk_buff
*skb
)
2438 if (dev
->type
== ARPHRD_ETHER
) {
2439 skb_reset_mac_header(skb
);
2440 skb
->protocol
= eth_hdr(skb
)->h_proto
;
2444 static int __packet_snd_vnet_parse(struct virtio_net_hdr
*vnet_hdr
, size_t len
)
2446 if ((vnet_hdr
->flags
& VIRTIO_NET_HDR_F_NEEDS_CSUM
) &&
2447 (__virtio16_to_cpu(vio_le(), vnet_hdr
->csum_start
) +
2448 __virtio16_to_cpu(vio_le(), vnet_hdr
->csum_offset
) + 2 >
2449 __virtio16_to_cpu(vio_le(), vnet_hdr
->hdr_len
)))
2450 vnet_hdr
->hdr_len
= __cpu_to_virtio16(vio_le(),
2451 __virtio16_to_cpu(vio_le(), vnet_hdr
->csum_start
) +
2452 __virtio16_to_cpu(vio_le(), vnet_hdr
->csum_offset
) + 2);
2454 if (__virtio16_to_cpu(vio_le(), vnet_hdr
->hdr_len
) > len
)
2460 static int packet_snd_vnet_parse(struct msghdr
*msg
, size_t *len
,
2461 struct virtio_net_hdr
*vnet_hdr
)
2463 if (*len
< sizeof(*vnet_hdr
))
2465 *len
-= sizeof(*vnet_hdr
);
2467 if (!copy_from_iter_full(vnet_hdr
, sizeof(*vnet_hdr
), &msg
->msg_iter
))
2470 return __packet_snd_vnet_parse(vnet_hdr
, *len
);
2473 static int tpacket_fill_skb(struct packet_sock
*po
, struct sk_buff
*skb
,
2474 void *frame
, struct net_device
*dev
, void *data
, int tp_len
,
2475 __be16 proto
, unsigned char *addr
, int hlen
, int copylen
,
2476 const struct sockcm_cookie
*sockc
)
2478 union tpacket_uhdr ph
;
2479 int to_write
, offset
, len
, nr_frags
, len_max
;
2480 struct socket
*sock
= po
->sk
.sk_socket
;
2486 skb
->protocol
= proto
;
2488 skb
->priority
= po
->sk
.sk_priority
;
2489 skb
->mark
= po
->sk
.sk_mark
;
2490 sock_tx_timestamp(&po
->sk
, sockc
->tsflags
, &skb_shinfo(skb
)->tx_flags
);
2491 skb_shinfo(skb
)->destructor_arg
= ph
.raw
;
2493 skb_reserve(skb
, hlen
);
2494 skb_reset_network_header(skb
);
2498 if (sock
->type
== SOCK_DGRAM
) {
2499 err
= dev_hard_header(skb
, dev
, ntohs(proto
), addr
,
2501 if (unlikely(err
< 0))
2503 } else if (copylen
) {
2504 int hdrlen
= min_t(int, copylen
, tp_len
);
2506 skb_push(skb
, dev
->hard_header_len
);
2507 skb_put(skb
, copylen
- dev
->hard_header_len
);
2508 err
= skb_store_bits(skb
, 0, data
, hdrlen
);
2511 if (!dev_validate_header(dev
, skb
->data
, hdrlen
))
2514 tpacket_set_protocol(dev
, skb
);
2520 offset
= offset_in_page(data
);
2521 len_max
= PAGE_SIZE
- offset
;
2522 len
= ((to_write
> len_max
) ? len_max
: to_write
);
2524 skb
->data_len
= to_write
;
2525 skb
->len
+= to_write
;
2526 skb
->truesize
+= to_write
;
2527 atomic_add(to_write
, &po
->sk
.sk_wmem_alloc
);
2529 while (likely(to_write
)) {
2530 nr_frags
= skb_shinfo(skb
)->nr_frags
;
2532 if (unlikely(nr_frags
>= MAX_SKB_FRAGS
)) {
2533 pr_err("Packet exceed the number of skb frags(%lu)\n",
2538 page
= pgv_to_page(data
);
2540 flush_dcache_page(page
);
2542 skb_fill_page_desc(skb
, nr_frags
, page
, offset
, len
);
2545 len_max
= PAGE_SIZE
;
2546 len
= ((to_write
> len_max
) ? len_max
: to_write
);
2549 skb_probe_transport_header(skb
, 0);
2554 static int tpacket_parse_header(struct packet_sock
*po
, void *frame
,
2555 int size_max
, void **data
)
2557 union tpacket_uhdr ph
;
2562 switch (po
->tp_version
) {
2564 if (ph
.h3
->tp_next_offset
!= 0) {
2565 pr_warn_once("variable sized slot not supported");
2568 tp_len
= ph
.h3
->tp_len
;
2571 tp_len
= ph
.h2
->tp_len
;
2574 tp_len
= ph
.h1
->tp_len
;
2577 if (unlikely(tp_len
> size_max
)) {
2578 pr_err("packet size is too long (%d > %d)\n", tp_len
, size_max
);
2582 if (unlikely(po
->tp_tx_has_off
)) {
2583 int off_min
, off_max
;
2585 off_min
= po
->tp_hdrlen
- sizeof(struct sockaddr_ll
);
2586 off_max
= po
->tx_ring
.frame_size
- tp_len
;
2587 if (po
->sk
.sk_type
== SOCK_DGRAM
) {
2588 switch (po
->tp_version
) {
2590 off
= ph
.h3
->tp_net
;
2593 off
= ph
.h2
->tp_net
;
2596 off
= ph
.h1
->tp_net
;
2600 switch (po
->tp_version
) {
2602 off
= ph
.h3
->tp_mac
;
2605 off
= ph
.h2
->tp_mac
;
2608 off
= ph
.h1
->tp_mac
;
2612 if (unlikely((off
< off_min
) || (off_max
< off
)))
2615 off
= po
->tp_hdrlen
- sizeof(struct sockaddr_ll
);
2618 *data
= frame
+ off
;
2622 static int tpacket_snd(struct packet_sock
*po
, struct msghdr
*msg
)
2624 struct sk_buff
*skb
;
2625 struct net_device
*dev
;
2626 struct virtio_net_hdr
*vnet_hdr
= NULL
;
2627 struct sockcm_cookie sockc
;
2629 int err
, reserve
= 0;
2631 DECLARE_SOCKADDR(struct sockaddr_ll
*, saddr
, msg
->msg_name
);
2632 bool need_wait
= !(msg
->msg_flags
& MSG_DONTWAIT
);
2633 int tp_len
, size_max
;
2634 unsigned char *addr
;
2637 int status
= TP_STATUS_AVAILABLE
;
2638 int hlen
, tlen
, copylen
= 0;
2640 mutex_lock(&po
->pg_vec_lock
);
2642 if (likely(saddr
== NULL
)) {
2643 dev
= packet_cached_dev_get(po
);
2648 if (msg
->msg_namelen
< sizeof(struct sockaddr_ll
))
2650 if (msg
->msg_namelen
< (saddr
->sll_halen
2651 + offsetof(struct sockaddr_ll
,
2654 proto
= saddr
->sll_protocol
;
2655 addr
= saddr
->sll_addr
;
2656 dev
= dev_get_by_index(sock_net(&po
->sk
), saddr
->sll_ifindex
);
2660 if (unlikely(dev
== NULL
))
2663 if (unlikely(!(dev
->flags
& IFF_UP
)))
2666 sockc
.tsflags
= po
->sk
.sk_tsflags
;
2667 if (msg
->msg_controllen
) {
2668 err
= sock_cmsg_send(&po
->sk
, msg
, &sockc
);
2673 if (po
->sk
.sk_socket
->type
== SOCK_RAW
)
2674 reserve
= dev
->hard_header_len
;
2675 size_max
= po
->tx_ring
.frame_size
2676 - (po
->tp_hdrlen
- sizeof(struct sockaddr_ll
));
2678 if ((size_max
> dev
->mtu
+ reserve
+ VLAN_HLEN
) && !po
->has_vnet_hdr
)
2679 size_max
= dev
->mtu
+ reserve
+ VLAN_HLEN
;
2682 ph
= packet_current_frame(po
, &po
->tx_ring
,
2683 TP_STATUS_SEND_REQUEST
);
2684 if (unlikely(ph
== NULL
)) {
2685 if (need_wait
&& need_resched())
2691 tp_len
= tpacket_parse_header(po
, ph
, size_max
, &data
);
2695 status
= TP_STATUS_SEND_REQUEST
;
2696 hlen
= LL_RESERVED_SPACE(dev
);
2697 tlen
= dev
->needed_tailroom
;
2698 if (po
->has_vnet_hdr
) {
2700 data
+= sizeof(*vnet_hdr
);
2701 tp_len
-= sizeof(*vnet_hdr
);
2703 __packet_snd_vnet_parse(vnet_hdr
, tp_len
)) {
2707 copylen
= __virtio16_to_cpu(vio_le(),
2710 copylen
= max_t(int, copylen
, dev
->hard_header_len
);
2711 skb
= sock_alloc_send_skb(&po
->sk
,
2712 hlen
+ tlen
+ sizeof(struct sockaddr_ll
) +
2713 (copylen
- dev
->hard_header_len
),
2716 if (unlikely(skb
== NULL
)) {
2717 /* we assume the socket was initially writeable ... */
2718 if (likely(len_sum
> 0))
2722 tp_len
= tpacket_fill_skb(po
, skb
, ph
, dev
, data
, tp_len
, proto
,
2723 addr
, hlen
, copylen
, &sockc
);
2724 if (likely(tp_len
>= 0) &&
2725 tp_len
> dev
->mtu
+ reserve
&&
2726 !po
->has_vnet_hdr
&&
2727 !packet_extra_vlan_len_allowed(dev
, skb
))
2730 if (unlikely(tp_len
< 0)) {
2733 __packet_set_status(po
, ph
,
2734 TP_STATUS_AVAILABLE
);
2735 packet_increment_head(&po
->tx_ring
);
2739 status
= TP_STATUS_WRONG_FORMAT
;
2745 if (po
->has_vnet_hdr
&& virtio_net_hdr_to_skb(skb
, vnet_hdr
,
2751 packet_pick_tx_queue(dev
, skb
);
2753 skb
->destructor
= tpacket_destruct_skb
;
2754 __packet_set_status(po
, ph
, TP_STATUS_SENDING
);
2755 packet_inc_pending(&po
->tx_ring
);
2757 status
= TP_STATUS_SEND_REQUEST
;
2758 err
= po
->xmit(skb
);
2759 if (unlikely(err
> 0)) {
2760 err
= net_xmit_errno(err
);
2761 if (err
&& __packet_get_status(po
, ph
) ==
2762 TP_STATUS_AVAILABLE
) {
2763 /* skb was destructed already */
2768 * skb was dropped but not destructed yet;
2769 * let's treat it like congestion or err < 0
2773 packet_increment_head(&po
->tx_ring
);
2775 } while (likely((ph
!= NULL
) ||
2776 /* Note: packet_read_pending() might be slow if we have
2777 * to call it as it's per_cpu variable, but in fast-path
2778 * we already short-circuit the loop with the first
2779 * condition, and luckily don't have to go that path
2782 (need_wait
&& packet_read_pending(&po
->tx_ring
))));
2788 __packet_set_status(po
, ph
, status
);
2793 mutex_unlock(&po
->pg_vec_lock
);
2797 static struct sk_buff
*packet_alloc_skb(struct sock
*sk
, size_t prepad
,
2798 size_t reserve
, size_t len
,
2799 size_t linear
, int noblock
,
2802 struct sk_buff
*skb
;
2804 /* Under a page? Don't bother with paged skb. */
2805 if (prepad
+ len
< PAGE_SIZE
|| !linear
)
2808 skb
= sock_alloc_send_pskb(sk
, prepad
+ linear
, len
- linear
, noblock
,
2813 skb_reserve(skb
, reserve
);
2814 skb_put(skb
, linear
);
2815 skb
->data_len
= len
- linear
;
2816 skb
->len
+= len
- linear
;
2821 static int packet_snd(struct socket
*sock
, struct msghdr
*msg
, size_t len
)
2823 struct sock
*sk
= sock
->sk
;
2824 DECLARE_SOCKADDR(struct sockaddr_ll
*, saddr
, msg
->msg_name
);
2825 struct sk_buff
*skb
;
2826 struct net_device
*dev
;
2828 unsigned char *addr
;
2829 int err
, reserve
= 0;
2830 struct sockcm_cookie sockc
;
2831 struct virtio_net_hdr vnet_hdr
= { 0 };
2833 struct packet_sock
*po
= pkt_sk(sk
);
2834 int hlen
, tlen
, linear
;
2838 * Get and verify the address.
2841 if (likely(saddr
== NULL
)) {
2842 dev
= packet_cached_dev_get(po
);
2847 if (msg
->msg_namelen
< sizeof(struct sockaddr_ll
))
2849 if (msg
->msg_namelen
< (saddr
->sll_halen
+ offsetof(struct sockaddr_ll
, sll_addr
)))
2851 proto
= saddr
->sll_protocol
;
2852 addr
= saddr
->sll_addr
;
2853 dev
= dev_get_by_index(sock_net(sk
), saddr
->sll_ifindex
);
2857 if (unlikely(dev
== NULL
))
2860 if (unlikely(!(dev
->flags
& IFF_UP
)))
2863 sockc
.tsflags
= sk
->sk_tsflags
;
2864 sockc
.mark
= sk
->sk_mark
;
2865 if (msg
->msg_controllen
) {
2866 err
= sock_cmsg_send(sk
, msg
, &sockc
);
2871 if (sock
->type
== SOCK_RAW
)
2872 reserve
= dev
->hard_header_len
;
2873 if (po
->has_vnet_hdr
) {
2874 err
= packet_snd_vnet_parse(msg
, &len
, &vnet_hdr
);
2879 if (unlikely(sock_flag(sk
, SOCK_NOFCS
))) {
2880 if (!netif_supports_nofcs(dev
)) {
2881 err
= -EPROTONOSUPPORT
;
2884 extra_len
= 4; /* We're doing our own CRC */
2888 if (!vnet_hdr
.gso_type
&&
2889 (len
> dev
->mtu
+ reserve
+ VLAN_HLEN
+ extra_len
))
2893 hlen
= LL_RESERVED_SPACE(dev
);
2894 tlen
= dev
->needed_tailroom
;
2895 linear
= __virtio16_to_cpu(vio_le(), vnet_hdr
.hdr_len
);
2896 linear
= max(linear
, min_t(int, len
, dev
->hard_header_len
));
2897 skb
= packet_alloc_skb(sk
, hlen
+ tlen
, hlen
, len
, linear
,
2898 msg
->msg_flags
& MSG_DONTWAIT
, &err
);
2902 skb_set_network_header(skb
, reserve
);
2905 if (sock
->type
== SOCK_DGRAM
) {
2906 offset
= dev_hard_header(skb
, dev
, ntohs(proto
), addr
, NULL
, len
);
2907 if (unlikely(offset
< 0))
2911 /* Returns -EFAULT on error */
2912 err
= skb_copy_datagram_from_iter(skb
, offset
, &msg
->msg_iter
, len
);
2916 if (sock
->type
== SOCK_RAW
&&
2917 !dev_validate_header(dev
, skb
->data
, len
)) {
2922 sock_tx_timestamp(sk
, sockc
.tsflags
, &skb_shinfo(skb
)->tx_flags
);
2924 if (!vnet_hdr
.gso_type
&& (len
> dev
->mtu
+ reserve
+ extra_len
) &&
2925 !packet_extra_vlan_len_allowed(dev
, skb
)) {
2930 skb
->protocol
= proto
;
2932 skb
->priority
= sk
->sk_priority
;
2933 skb
->mark
= sockc
.mark
;
2935 packet_pick_tx_queue(dev
, skb
);
2937 if (po
->has_vnet_hdr
) {
2938 err
= virtio_net_hdr_to_skb(skb
, &vnet_hdr
, vio_le());
2941 len
+= sizeof(vnet_hdr
);
2944 skb_probe_transport_header(skb
, reserve
);
2946 if (unlikely(extra_len
== 4))
2949 err
= po
->xmit(skb
);
2950 if (err
> 0 && (err
= net_xmit_errno(err
)) != 0)
2966 static int packet_sendmsg(struct socket
*sock
, struct msghdr
*msg
, size_t len
)
2968 struct sock
*sk
= sock
->sk
;
2969 struct packet_sock
*po
= pkt_sk(sk
);
2971 if (po
->tx_ring
.pg_vec
)
2972 return tpacket_snd(po
, msg
);
2974 return packet_snd(sock
, msg
, len
);
2978 * Close a PACKET socket. This is fairly simple. We immediately go
2979 * to 'closed' state and remove our protocol entry in the device list.
2982 static int packet_release(struct socket
*sock
)
2984 struct sock
*sk
= sock
->sk
;
2985 struct packet_sock
*po
;
2986 struct packet_fanout
*f
;
2988 union tpacket_req_u req_u
;
2996 mutex_lock(&net
->packet
.sklist_lock
);
2997 sk_del_node_init_rcu(sk
);
2998 mutex_unlock(&net
->packet
.sklist_lock
);
3001 sock_prot_inuse_add(net
, sk
->sk_prot
, -1);
3004 spin_lock(&po
->bind_lock
);
3005 unregister_prot_hook(sk
, false);
3006 packet_cached_dev_reset(po
);
3008 if (po
->prot_hook
.dev
) {
3009 dev_put(po
->prot_hook
.dev
);
3010 po
->prot_hook
.dev
= NULL
;
3012 spin_unlock(&po
->bind_lock
);
3014 packet_flush_mclist(sk
);
3016 if (po
->rx_ring
.pg_vec
) {
3017 memset(&req_u
, 0, sizeof(req_u
));
3018 packet_set_ring(sk
, &req_u
, 1, 0);
3021 if (po
->tx_ring
.pg_vec
) {
3022 memset(&req_u
, 0, sizeof(req_u
));
3023 packet_set_ring(sk
, &req_u
, 1, 1);
3026 f
= fanout_release(sk
);
3031 fanout_release_data(f
);
3035 * Now the socket is dead. No more input will appear.
3042 skb_queue_purge(&sk
->sk_receive_queue
);
3043 packet_free_pending(po
);
3044 sk_refcnt_debug_release(sk
);
3051 * Attach a packet hook.
3054 static int packet_do_bind(struct sock
*sk
, const char *name
, int ifindex
,
3057 struct packet_sock
*po
= pkt_sk(sk
);
3058 struct net_device
*dev_curr
;
3061 struct net_device
*dev
= NULL
;
3063 bool unlisted
= false;
3069 spin_lock(&po
->bind_lock
);
3073 dev
= dev_get_by_name_rcu(sock_net(sk
), name
);
3078 } else if (ifindex
) {
3079 dev
= dev_get_by_index_rcu(sock_net(sk
), ifindex
);
3089 proto_curr
= po
->prot_hook
.type
;
3090 dev_curr
= po
->prot_hook
.dev
;
3092 need_rehook
= proto_curr
!= proto
|| dev_curr
!= dev
;
3097 __unregister_prot_hook(sk
, true);
3099 dev_curr
= po
->prot_hook
.dev
;
3101 unlisted
= !dev_get_by_index_rcu(sock_net(sk
),
3106 po
->prot_hook
.type
= proto
;
3108 if (unlikely(unlisted
)) {
3110 po
->prot_hook
.dev
= NULL
;
3112 packet_cached_dev_reset(po
);
3114 po
->prot_hook
.dev
= dev
;
3115 po
->ifindex
= dev
? dev
->ifindex
: 0;
3116 packet_cached_dev_assign(po
, dev
);
3122 if (proto
== 0 || !need_rehook
)
3125 if (!unlisted
&& (!dev
|| (dev
->flags
& IFF_UP
))) {
3126 register_prot_hook(sk
);
3128 sk
->sk_err
= ENETDOWN
;
3129 if (!sock_flag(sk
, SOCK_DEAD
))
3130 sk
->sk_error_report(sk
);
3135 spin_unlock(&po
->bind_lock
);
3141 * Bind a packet socket to a device
3144 static int packet_bind_spkt(struct socket
*sock
, struct sockaddr
*uaddr
,
3147 struct sock
*sk
= sock
->sk
;
3148 char name
[sizeof(uaddr
->sa_data
) + 1];
3154 if (addr_len
!= sizeof(struct sockaddr
))
3156 /* uaddr->sa_data comes from the userspace, it's not guaranteed to be
3159 memcpy(name
, uaddr
->sa_data
, sizeof(uaddr
->sa_data
));
3160 name
[sizeof(uaddr
->sa_data
)] = 0;
3162 return packet_do_bind(sk
, name
, 0, pkt_sk(sk
)->num
);
3165 static int packet_bind(struct socket
*sock
, struct sockaddr
*uaddr
, int addr_len
)
3167 struct sockaddr_ll
*sll
= (struct sockaddr_ll
*)uaddr
;
3168 struct sock
*sk
= sock
->sk
;
3174 if (addr_len
< sizeof(struct sockaddr_ll
))
3176 if (sll
->sll_family
!= AF_PACKET
)
3179 return packet_do_bind(sk
, NULL
, sll
->sll_ifindex
,
3180 sll
->sll_protocol
? : pkt_sk(sk
)->num
);
3183 static struct proto packet_proto
= {
3185 .owner
= THIS_MODULE
,
3186 .obj_size
= sizeof(struct packet_sock
),
3190 * Create a packet of type SOCK_PACKET.
3193 static int packet_create(struct net
*net
, struct socket
*sock
, int protocol
,
3197 struct packet_sock
*po
;
3198 __be16 proto
= (__force __be16
)protocol
; /* weird, but documented */
3201 if (!ns_capable(net
->user_ns
, CAP_NET_RAW
))
3203 if (sock
->type
!= SOCK_DGRAM
&& sock
->type
!= SOCK_RAW
&&
3204 sock
->type
!= SOCK_PACKET
)
3205 return -ESOCKTNOSUPPORT
;
3207 sock
->state
= SS_UNCONNECTED
;
3210 sk
= sk_alloc(net
, PF_PACKET
, GFP_KERNEL
, &packet_proto
, kern
);
3214 sock
->ops
= &packet_ops
;
3215 if (sock
->type
== SOCK_PACKET
)
3216 sock
->ops
= &packet_ops_spkt
;
3218 sock_init_data(sock
, sk
);
3221 sk
->sk_family
= PF_PACKET
;
3223 po
->xmit
= dev_queue_xmit
;
3225 err
= packet_alloc_pending(po
);
3229 packet_cached_dev_reset(po
);
3231 sk
->sk_destruct
= packet_sock_destruct
;
3232 sk_refcnt_debug_inc(sk
);
3235 * Attach a protocol block
3238 spin_lock_init(&po
->bind_lock
);
3239 mutex_init(&po
->pg_vec_lock
);
3240 po
->rollover
= NULL
;
3241 po
->prot_hook
.func
= packet_rcv
;
3243 if (sock
->type
== SOCK_PACKET
)
3244 po
->prot_hook
.func
= packet_rcv_spkt
;
3246 po
->prot_hook
.af_packet_priv
= sk
;
3249 po
->prot_hook
.type
= proto
;
3250 register_prot_hook(sk
);
3253 mutex_lock(&net
->packet
.sklist_lock
);
3254 sk_add_node_rcu(sk
, &net
->packet
.sklist
);
3255 mutex_unlock(&net
->packet
.sklist_lock
);
3258 sock_prot_inuse_add(net
, &packet_proto
, 1);
3269 * Pull a packet from our receive queue and hand it to the user.
3270 * If necessary we block.
3273 static int packet_recvmsg(struct socket
*sock
, struct msghdr
*msg
, size_t len
,
3276 struct sock
*sk
= sock
->sk
;
3277 struct sk_buff
*skb
;
3279 int vnet_hdr_len
= 0;
3280 unsigned int origlen
= 0;
3283 if (flags
& ~(MSG_PEEK
|MSG_DONTWAIT
|MSG_TRUNC
|MSG_CMSG_COMPAT
|MSG_ERRQUEUE
))
3287 /* What error should we return now? EUNATTACH? */
3288 if (pkt_sk(sk
)->ifindex
< 0)
3292 if (flags
& MSG_ERRQUEUE
) {
3293 err
= sock_recv_errqueue(sk
, msg
, len
,
3294 SOL_PACKET
, PACKET_TX_TIMESTAMP
);
3299 * Call the generic datagram receiver. This handles all sorts
3300 * of horrible races and re-entrancy so we can forget about it
3301 * in the protocol layers.
3303 * Now it will return ENETDOWN, if device have just gone down,
3304 * but then it will block.
3307 skb
= skb_recv_datagram(sk
, flags
, flags
& MSG_DONTWAIT
, &err
);
3310 * An error occurred so return it. Because skb_recv_datagram()
3311 * handles the blocking we don't see and worry about blocking
3318 if (pkt_sk(sk
)->pressure
)
3319 packet_rcv_has_room(pkt_sk(sk
), NULL
);
3321 if (pkt_sk(sk
)->has_vnet_hdr
) {
3322 err
= packet_rcv_vnet(msg
, skb
, &len
);
3325 vnet_hdr_len
= sizeof(struct virtio_net_hdr
);
3328 /* You lose any data beyond the buffer you gave. If it worries
3329 * a user program they can ask the device for its MTU
3335 msg
->msg_flags
|= MSG_TRUNC
;
3338 err
= skb_copy_datagram_msg(skb
, 0, msg
, copied
);
3342 if (sock
->type
!= SOCK_PACKET
) {
3343 struct sockaddr_ll
*sll
= &PACKET_SKB_CB(skb
)->sa
.ll
;
3345 /* Original length was stored in sockaddr_ll fields */
3346 origlen
= PACKET_SKB_CB(skb
)->sa
.origlen
;
3347 sll
->sll_family
= AF_PACKET
;
3348 sll
->sll_protocol
= skb
->protocol
;
3351 sock_recv_ts_and_drops(msg
, sk
, skb
);
3353 if (msg
->msg_name
) {
3354 /* If the address length field is there to be filled
3355 * in, we fill it in now.
3357 if (sock
->type
== SOCK_PACKET
) {
3358 __sockaddr_check_size(sizeof(struct sockaddr_pkt
));
3359 msg
->msg_namelen
= sizeof(struct sockaddr_pkt
);
3361 struct sockaddr_ll
*sll
= &PACKET_SKB_CB(skb
)->sa
.ll
;
3363 msg
->msg_namelen
= sll
->sll_halen
+
3364 offsetof(struct sockaddr_ll
, sll_addr
);
3366 memcpy(msg
->msg_name
, &PACKET_SKB_CB(skb
)->sa
,
3370 if (pkt_sk(sk
)->auxdata
) {
3371 struct tpacket_auxdata aux
;
3373 aux
.tp_status
= TP_STATUS_USER
;
3374 if (skb
->ip_summed
== CHECKSUM_PARTIAL
)
3375 aux
.tp_status
|= TP_STATUS_CSUMNOTREADY
;
3376 else if (skb
->pkt_type
!= PACKET_OUTGOING
&&
3377 (skb
->ip_summed
== CHECKSUM_COMPLETE
||
3378 skb_csum_unnecessary(skb
)))
3379 aux
.tp_status
|= TP_STATUS_CSUM_VALID
;
3381 aux
.tp_len
= origlen
;
3382 aux
.tp_snaplen
= skb
->len
;
3384 aux
.tp_net
= skb_network_offset(skb
);
3385 if (skb_vlan_tag_present(skb
)) {
3386 aux
.tp_vlan_tci
= skb_vlan_tag_get(skb
);
3387 aux
.tp_vlan_tpid
= ntohs(skb
->vlan_proto
);
3388 aux
.tp_status
|= TP_STATUS_VLAN_VALID
| TP_STATUS_VLAN_TPID_VALID
;
3390 aux
.tp_vlan_tci
= 0;
3391 aux
.tp_vlan_tpid
= 0;
3393 put_cmsg(msg
, SOL_PACKET
, PACKET_AUXDATA
, sizeof(aux
), &aux
);
3397 * Free or return the buffer as appropriate. Again this
3398 * hides all the races and re-entrancy issues from us.
3400 err
= vnet_hdr_len
+ ((flags
&MSG_TRUNC
) ? skb
->len
: copied
);
3403 skb_free_datagram(sk
, skb
);
3408 static int packet_getname_spkt(struct socket
*sock
, struct sockaddr
*uaddr
,
3409 int *uaddr_len
, int peer
)
3411 struct net_device
*dev
;
3412 struct sock
*sk
= sock
->sk
;
3417 uaddr
->sa_family
= AF_PACKET
;
3418 memset(uaddr
->sa_data
, 0, sizeof(uaddr
->sa_data
));
3420 dev
= dev_get_by_index_rcu(sock_net(sk
), pkt_sk(sk
)->ifindex
);
3422 strlcpy(uaddr
->sa_data
, dev
->name
, sizeof(uaddr
->sa_data
));
3424 *uaddr_len
= sizeof(*uaddr
);
3429 static int packet_getname(struct socket
*sock
, struct sockaddr
*uaddr
,
3430 int *uaddr_len
, int peer
)
3432 struct net_device
*dev
;
3433 struct sock
*sk
= sock
->sk
;
3434 struct packet_sock
*po
= pkt_sk(sk
);
3435 DECLARE_SOCKADDR(struct sockaddr_ll
*, sll
, uaddr
);
3440 sll
->sll_family
= AF_PACKET
;
3441 sll
->sll_ifindex
= po
->ifindex
;
3442 sll
->sll_protocol
= po
->num
;
3443 sll
->sll_pkttype
= 0;
3445 dev
= dev_get_by_index_rcu(sock_net(sk
), po
->ifindex
);
3447 sll
->sll_hatype
= dev
->type
;
3448 sll
->sll_halen
= dev
->addr_len
;
3449 memcpy(sll
->sll_addr
, dev
->dev_addr
, dev
->addr_len
);
3451 sll
->sll_hatype
= 0; /* Bad: we have no ARPHRD_UNSPEC */
3455 *uaddr_len
= offsetof(struct sockaddr_ll
, sll_addr
) + sll
->sll_halen
;
3460 static int packet_dev_mc(struct net_device
*dev
, struct packet_mclist
*i
,
3464 case PACKET_MR_MULTICAST
:
3465 if (i
->alen
!= dev
->addr_len
)
3468 return dev_mc_add(dev
, i
->addr
);
3470 return dev_mc_del(dev
, i
->addr
);
3472 case PACKET_MR_PROMISC
:
3473 return dev_set_promiscuity(dev
, what
);
3474 case PACKET_MR_ALLMULTI
:
3475 return dev_set_allmulti(dev
, what
);
3476 case PACKET_MR_UNICAST
:
3477 if (i
->alen
!= dev
->addr_len
)
3480 return dev_uc_add(dev
, i
->addr
);
3482 return dev_uc_del(dev
, i
->addr
);
3490 static void packet_dev_mclist_delete(struct net_device
*dev
,
3491 struct packet_mclist
**mlp
)
3493 struct packet_mclist
*ml
;
3495 while ((ml
= *mlp
) != NULL
) {
3496 if (ml
->ifindex
== dev
->ifindex
) {
3497 packet_dev_mc(dev
, ml
, -1);
3505 static int packet_mc_add(struct sock
*sk
, struct packet_mreq_max
*mreq
)
3507 struct packet_sock
*po
= pkt_sk(sk
);
3508 struct packet_mclist
*ml
, *i
;
3509 struct net_device
*dev
;
3515 dev
= __dev_get_by_index(sock_net(sk
), mreq
->mr_ifindex
);
3520 if (mreq
->mr_alen
> dev
->addr_len
)
3524 i
= kmalloc(sizeof(*i
), GFP_KERNEL
);
3529 for (ml
= po
->mclist
; ml
; ml
= ml
->next
) {
3530 if (ml
->ifindex
== mreq
->mr_ifindex
&&
3531 ml
->type
== mreq
->mr_type
&&
3532 ml
->alen
== mreq
->mr_alen
&&
3533 memcmp(ml
->addr
, mreq
->mr_address
, ml
->alen
) == 0) {
3535 /* Free the new element ... */
3541 i
->type
= mreq
->mr_type
;
3542 i
->ifindex
= mreq
->mr_ifindex
;
3543 i
->alen
= mreq
->mr_alen
;
3544 memcpy(i
->addr
, mreq
->mr_address
, i
->alen
);
3545 memset(i
->addr
+ i
->alen
, 0, sizeof(i
->addr
) - i
->alen
);
3547 i
->next
= po
->mclist
;
3549 err
= packet_dev_mc(dev
, i
, 1);
3551 po
->mclist
= i
->next
;
3560 static int packet_mc_drop(struct sock
*sk
, struct packet_mreq_max
*mreq
)
3562 struct packet_mclist
*ml
, **mlp
;
3566 for (mlp
= &pkt_sk(sk
)->mclist
; (ml
= *mlp
) != NULL
; mlp
= &ml
->next
) {
3567 if (ml
->ifindex
== mreq
->mr_ifindex
&&
3568 ml
->type
== mreq
->mr_type
&&
3569 ml
->alen
== mreq
->mr_alen
&&
3570 memcmp(ml
->addr
, mreq
->mr_address
, ml
->alen
) == 0) {
3571 if (--ml
->count
== 0) {
3572 struct net_device
*dev
;
3574 dev
= __dev_get_by_index(sock_net(sk
), ml
->ifindex
);
3576 packet_dev_mc(dev
, ml
, -1);
3586 static void packet_flush_mclist(struct sock
*sk
)
3588 struct packet_sock
*po
= pkt_sk(sk
);
3589 struct packet_mclist
*ml
;
3595 while ((ml
= po
->mclist
) != NULL
) {
3596 struct net_device
*dev
;
3598 po
->mclist
= ml
->next
;
3599 dev
= __dev_get_by_index(sock_net(sk
), ml
->ifindex
);
3601 packet_dev_mc(dev
, ml
, -1);
3608 packet_setsockopt(struct socket
*sock
, int level
, int optname
, char __user
*optval
, unsigned int optlen
)
3610 struct sock
*sk
= sock
->sk
;
3611 struct packet_sock
*po
= pkt_sk(sk
);
3614 if (level
!= SOL_PACKET
)
3615 return -ENOPROTOOPT
;
3618 case PACKET_ADD_MEMBERSHIP
:
3619 case PACKET_DROP_MEMBERSHIP
:
3621 struct packet_mreq_max mreq
;
3623 memset(&mreq
, 0, sizeof(mreq
));
3624 if (len
< sizeof(struct packet_mreq
))
3626 if (len
> sizeof(mreq
))
3628 if (copy_from_user(&mreq
, optval
, len
))
3630 if (len
< (mreq
.mr_alen
+ offsetof(struct packet_mreq
, mr_address
)))
3632 if (optname
== PACKET_ADD_MEMBERSHIP
)
3633 ret
= packet_mc_add(sk
, &mreq
);
3635 ret
= packet_mc_drop(sk
, &mreq
);
3639 case PACKET_RX_RING
:
3640 case PACKET_TX_RING
:
3642 union tpacket_req_u req_u
;
3645 switch (po
->tp_version
) {
3648 len
= sizeof(req_u
.req
);
3652 len
= sizeof(req_u
.req3
);
3657 if (copy_from_user(&req_u
.req
, optval
, len
))
3659 return packet_set_ring(sk
, &req_u
, 0,
3660 optname
== PACKET_TX_RING
);
3662 case PACKET_COPY_THRESH
:
3666 if (optlen
!= sizeof(val
))
3668 if (copy_from_user(&val
, optval
, sizeof(val
)))
3671 pkt_sk(sk
)->copy_thresh
= val
;
3674 case PACKET_VERSION
:
3678 if (optlen
!= sizeof(val
))
3680 if (copy_from_user(&val
, optval
, sizeof(val
)))
3691 if (po
->rx_ring
.pg_vec
|| po
->tx_ring
.pg_vec
) {
3694 po
->tp_version
= val
;
3700 case PACKET_RESERVE
:
3704 if (optlen
!= sizeof(val
))
3706 if (po
->rx_ring
.pg_vec
|| po
->tx_ring
.pg_vec
)
3708 if (copy_from_user(&val
, optval
, sizeof(val
)))
3712 po
->tp_reserve
= val
;
3719 if (optlen
!= sizeof(val
))
3721 if (po
->rx_ring
.pg_vec
|| po
->tx_ring
.pg_vec
)
3723 if (copy_from_user(&val
, optval
, sizeof(val
)))
3725 po
->tp_loss
= !!val
;
3728 case PACKET_AUXDATA
:
3732 if (optlen
< sizeof(val
))
3734 if (copy_from_user(&val
, optval
, sizeof(val
)))
3737 po
->auxdata
= !!val
;
3740 case PACKET_ORIGDEV
:
3744 if (optlen
< sizeof(val
))
3746 if (copy_from_user(&val
, optval
, sizeof(val
)))
3749 po
->origdev
= !!val
;
3752 case PACKET_VNET_HDR
:
3756 if (sock
->type
!= SOCK_RAW
)
3758 if (po
->rx_ring
.pg_vec
|| po
->tx_ring
.pg_vec
)
3760 if (optlen
< sizeof(val
))
3762 if (copy_from_user(&val
, optval
, sizeof(val
)))
3765 po
->has_vnet_hdr
= !!val
;
3768 case PACKET_TIMESTAMP
:
3772 if (optlen
!= sizeof(val
))
3774 if (copy_from_user(&val
, optval
, sizeof(val
)))
3777 po
->tp_tstamp
= val
;
3784 if (optlen
!= sizeof(val
))
3786 if (copy_from_user(&val
, optval
, sizeof(val
)))
3789 return fanout_add(sk
, val
& 0xffff, val
>> 16);
3791 case PACKET_FANOUT_DATA
:
3796 return fanout_set_data(po
, optval
, optlen
);
3798 case PACKET_TX_HAS_OFF
:
3802 if (optlen
!= sizeof(val
))
3804 if (po
->rx_ring
.pg_vec
|| po
->tx_ring
.pg_vec
)
3806 if (copy_from_user(&val
, optval
, sizeof(val
)))
3808 po
->tp_tx_has_off
= !!val
;
3811 case PACKET_QDISC_BYPASS
:
3815 if (optlen
!= sizeof(val
))
3817 if (copy_from_user(&val
, optval
, sizeof(val
)))
3820 po
->xmit
= val
? packet_direct_xmit
: dev_queue_xmit
;
3824 return -ENOPROTOOPT
;
3828 static int packet_getsockopt(struct socket
*sock
, int level
, int optname
,
3829 char __user
*optval
, int __user
*optlen
)
3832 int val
, lv
= sizeof(val
);
3833 struct sock
*sk
= sock
->sk
;
3834 struct packet_sock
*po
= pkt_sk(sk
);
3836 union tpacket_stats_u st
;
3837 struct tpacket_rollover_stats rstats
;
3839 if (level
!= SOL_PACKET
)
3840 return -ENOPROTOOPT
;
3842 if (get_user(len
, optlen
))
3849 case PACKET_STATISTICS
:
3850 spin_lock_bh(&sk
->sk_receive_queue
.lock
);
3851 memcpy(&st
, &po
->stats
, sizeof(st
));
3852 memset(&po
->stats
, 0, sizeof(po
->stats
));
3853 spin_unlock_bh(&sk
->sk_receive_queue
.lock
);
3855 if (po
->tp_version
== TPACKET_V3
) {
3856 lv
= sizeof(struct tpacket_stats_v3
);
3857 st
.stats3
.tp_packets
+= st
.stats3
.tp_drops
;
3860 lv
= sizeof(struct tpacket_stats
);
3861 st
.stats1
.tp_packets
+= st
.stats1
.tp_drops
;
3866 case PACKET_AUXDATA
:
3869 case PACKET_ORIGDEV
:
3872 case PACKET_VNET_HDR
:
3873 val
= po
->has_vnet_hdr
;
3875 case PACKET_VERSION
:
3876 val
= po
->tp_version
;
3879 if (len
> sizeof(int))
3881 if (len
< sizeof(int))
3883 if (copy_from_user(&val
, optval
, len
))
3887 val
= sizeof(struct tpacket_hdr
);
3890 val
= sizeof(struct tpacket2_hdr
);
3893 val
= sizeof(struct tpacket3_hdr
);
3899 case PACKET_RESERVE
:
3900 val
= po
->tp_reserve
;
3905 case PACKET_TIMESTAMP
:
3906 val
= po
->tp_tstamp
;
3910 ((u32
)po
->fanout
->id
|
3911 ((u32
)po
->fanout
->type
<< 16) |
3912 ((u32
)po
->fanout
->flags
<< 24)) :
3915 case PACKET_ROLLOVER_STATS
:
3918 rstats
.tp_all
= atomic_long_read(&po
->rollover
->num
);
3919 rstats
.tp_huge
= atomic_long_read(&po
->rollover
->num_huge
);
3920 rstats
.tp_failed
= atomic_long_read(&po
->rollover
->num_failed
);
3922 lv
= sizeof(rstats
);
3924 case PACKET_TX_HAS_OFF
:
3925 val
= po
->tp_tx_has_off
;
3927 case PACKET_QDISC_BYPASS
:
3928 val
= packet_use_direct_xmit(po
);
3931 return -ENOPROTOOPT
;
3936 if (put_user(len
, optlen
))
3938 if (copy_to_user(optval
, data
, len
))
3944 #ifdef CONFIG_COMPAT
3945 static int compat_packet_setsockopt(struct socket
*sock
, int level
, int optname
,
3946 char __user
*optval
, unsigned int optlen
)
3948 struct packet_sock
*po
= pkt_sk(sock
->sk
);
3950 if (level
!= SOL_PACKET
)
3951 return -ENOPROTOOPT
;
3953 if (optname
== PACKET_FANOUT_DATA
&&
3954 po
->fanout
&& po
->fanout
->type
== PACKET_FANOUT_CBPF
) {
3955 optval
= (char __user
*)get_compat_bpf_fprog(optval
);
3958 optlen
= sizeof(struct sock_fprog
);
3961 return packet_setsockopt(sock
, level
, optname
, optval
, optlen
);
3965 static int packet_notifier(struct notifier_block
*this,
3966 unsigned long msg
, void *ptr
)
3969 struct net_device
*dev
= netdev_notifier_info_to_dev(ptr
);
3970 struct net
*net
= dev_net(dev
);
3973 sk_for_each_rcu(sk
, &net
->packet
.sklist
) {
3974 struct packet_sock
*po
= pkt_sk(sk
);
3977 case NETDEV_UNREGISTER
:
3979 packet_dev_mclist_delete(dev
, &po
->mclist
);
3983 if (dev
->ifindex
== po
->ifindex
) {
3984 spin_lock(&po
->bind_lock
);
3986 __unregister_prot_hook(sk
, false);
3987 sk
->sk_err
= ENETDOWN
;
3988 if (!sock_flag(sk
, SOCK_DEAD
))
3989 sk
->sk_error_report(sk
);
3991 if (msg
== NETDEV_UNREGISTER
) {
3992 packet_cached_dev_reset(po
);
3994 if (po
->prot_hook
.dev
)
3995 dev_put(po
->prot_hook
.dev
);
3996 po
->prot_hook
.dev
= NULL
;
3998 spin_unlock(&po
->bind_lock
);
4002 if (dev
->ifindex
== po
->ifindex
) {
4003 spin_lock(&po
->bind_lock
);
4005 register_prot_hook(sk
);
4006 spin_unlock(&po
->bind_lock
);
4016 static int packet_ioctl(struct socket
*sock
, unsigned int cmd
,
4019 struct sock
*sk
= sock
->sk
;
4024 int amount
= sk_wmem_alloc_get(sk
);
4026 return put_user(amount
, (int __user
*)arg
);
4030 struct sk_buff
*skb
;
4033 spin_lock_bh(&sk
->sk_receive_queue
.lock
);
4034 skb
= skb_peek(&sk
->sk_receive_queue
);
4037 spin_unlock_bh(&sk
->sk_receive_queue
.lock
);
4038 return put_user(amount
, (int __user
*)arg
);
4041 return sock_get_timestamp(sk
, (struct timeval __user
*)arg
);
4043 return sock_get_timestampns(sk
, (struct timespec __user
*)arg
);
4053 case SIOCGIFBRDADDR
:
4054 case SIOCSIFBRDADDR
:
4055 case SIOCGIFNETMASK
:
4056 case SIOCSIFNETMASK
:
4057 case SIOCGIFDSTADDR
:
4058 case SIOCSIFDSTADDR
:
4060 return inet_dgram_ops
.ioctl(sock
, cmd
, arg
);
4064 return -ENOIOCTLCMD
;
4069 static unsigned int packet_poll(struct file
*file
, struct socket
*sock
,
4072 struct sock
*sk
= sock
->sk
;
4073 struct packet_sock
*po
= pkt_sk(sk
);
4074 unsigned int mask
= datagram_poll(file
, sock
, wait
);
4076 spin_lock_bh(&sk
->sk_receive_queue
.lock
);
4077 if (po
->rx_ring
.pg_vec
) {
4078 if (!packet_previous_rx_frame(po
, &po
->rx_ring
,
4080 mask
|= POLLIN
| POLLRDNORM
;
4082 if (po
->pressure
&& __packet_rcv_has_room(po
, NULL
) == ROOM_NORMAL
)
4084 spin_unlock_bh(&sk
->sk_receive_queue
.lock
);
4085 spin_lock_bh(&sk
->sk_write_queue
.lock
);
4086 if (po
->tx_ring
.pg_vec
) {
4087 if (packet_current_frame(po
, &po
->tx_ring
, TP_STATUS_AVAILABLE
))
4088 mask
|= POLLOUT
| POLLWRNORM
;
4090 spin_unlock_bh(&sk
->sk_write_queue
.lock
);
4095 /* Dirty? Well, I still did not learn better way to account
4099 static void packet_mm_open(struct vm_area_struct
*vma
)
4101 struct file
*file
= vma
->vm_file
;
4102 struct socket
*sock
= file
->private_data
;
4103 struct sock
*sk
= sock
->sk
;
4106 atomic_inc(&pkt_sk(sk
)->mapped
);
4109 static void packet_mm_close(struct vm_area_struct
*vma
)
4111 struct file
*file
= vma
->vm_file
;
4112 struct socket
*sock
= file
->private_data
;
4113 struct sock
*sk
= sock
->sk
;
4116 atomic_dec(&pkt_sk(sk
)->mapped
);
4119 static const struct vm_operations_struct packet_mmap_ops
= {
4120 .open
= packet_mm_open
,
4121 .close
= packet_mm_close
,
4124 static void free_pg_vec(struct pgv
*pg_vec
, unsigned int order
,
4129 for (i
= 0; i
< len
; i
++) {
4130 if (likely(pg_vec
[i
].buffer
)) {
4131 if (is_vmalloc_addr(pg_vec
[i
].buffer
))
4132 vfree(pg_vec
[i
].buffer
);
4134 free_pages((unsigned long)pg_vec
[i
].buffer
,
4136 pg_vec
[i
].buffer
= NULL
;
4142 static char *alloc_one_pg_vec_page(unsigned long order
)
4145 gfp_t gfp_flags
= GFP_KERNEL
| __GFP_COMP
|
4146 __GFP_ZERO
| __GFP_NOWARN
| __GFP_NORETRY
;
4148 buffer
= (char *) __get_free_pages(gfp_flags
, order
);
4152 /* __get_free_pages failed, fall back to vmalloc */
4153 buffer
= vzalloc((1 << order
) * PAGE_SIZE
);
4157 /* vmalloc failed, lets dig into swap here */
4158 gfp_flags
&= ~__GFP_NORETRY
;
4159 buffer
= (char *) __get_free_pages(gfp_flags
, order
);
4163 /* complete and utter failure */
4167 static struct pgv
*alloc_pg_vec(struct tpacket_req
*req
, int order
)
4169 unsigned int block_nr
= req
->tp_block_nr
;
4173 pg_vec
= kcalloc(block_nr
, sizeof(struct pgv
), GFP_KERNEL
);
4174 if (unlikely(!pg_vec
))
4177 for (i
= 0; i
< block_nr
; i
++) {
4178 pg_vec
[i
].buffer
= alloc_one_pg_vec_page(order
);
4179 if (unlikely(!pg_vec
[i
].buffer
))
4180 goto out_free_pgvec
;
4187 free_pg_vec(pg_vec
, order
, block_nr
);
4192 static int packet_set_ring(struct sock
*sk
, union tpacket_req_u
*req_u
,
4193 int closing
, int tx_ring
)
4195 struct pgv
*pg_vec
= NULL
;
4196 struct packet_sock
*po
= pkt_sk(sk
);
4197 int was_running
, order
= 0;
4198 struct packet_ring_buffer
*rb
;
4199 struct sk_buff_head
*rb_queue
;
4202 /* Added to avoid minimal code churn */
4203 struct tpacket_req
*req
= &req_u
->req
;
4207 rb
= tx_ring
? &po
->tx_ring
: &po
->rx_ring
;
4208 rb_queue
= tx_ring
? &sk
->sk_write_queue
: &sk
->sk_receive_queue
;
4212 if (atomic_read(&po
->mapped
))
4214 if (packet_read_pending(rb
))
4218 if (req
->tp_block_nr
) {
4219 /* Sanity tests and some calculations */
4221 if (unlikely(rb
->pg_vec
))
4224 switch (po
->tp_version
) {
4226 po
->tp_hdrlen
= TPACKET_HDRLEN
;
4229 po
->tp_hdrlen
= TPACKET2_HDRLEN
;
4232 po
->tp_hdrlen
= TPACKET3_HDRLEN
;
4237 if (unlikely((int)req
->tp_block_size
<= 0))
4239 if (unlikely(!PAGE_ALIGNED(req
->tp_block_size
)))
4241 if (po
->tp_version
>= TPACKET_V3
&&
4242 req
->tp_block_size
<=
4243 BLK_PLUS_PRIV((u64
)req_u
->req3
.tp_sizeof_priv
))
4245 if (unlikely(req
->tp_frame_size
< po
->tp_hdrlen
+
4248 if (unlikely(req
->tp_frame_size
& (TPACKET_ALIGNMENT
- 1)))
4251 rb
->frames_per_block
= req
->tp_block_size
/ req
->tp_frame_size
;
4252 if (unlikely(rb
->frames_per_block
== 0))
4254 if (unlikely(req
->tp_block_size
> UINT_MAX
/ req
->tp_block_nr
))
4256 if (unlikely((rb
->frames_per_block
* req
->tp_block_nr
) !=
4261 order
= get_order(req
->tp_block_size
);
4262 pg_vec
= alloc_pg_vec(req
, order
);
4263 if (unlikely(!pg_vec
))
4265 switch (po
->tp_version
) {
4267 /* Block transmit is not supported yet */
4269 init_prb_bdqc(po
, rb
, pg_vec
, req_u
);
4271 struct tpacket_req3
*req3
= &req_u
->req3
;
4273 if (req3
->tp_retire_blk_tov
||
4274 req3
->tp_sizeof_priv
||
4275 req3
->tp_feature_req_word
) {
4288 if (unlikely(req
->tp_frame_nr
))
4293 /* Detach socket from network */
4294 spin_lock(&po
->bind_lock
);
4295 was_running
= po
->running
;
4299 __unregister_prot_hook(sk
, false);
4301 spin_unlock(&po
->bind_lock
);
4306 mutex_lock(&po
->pg_vec_lock
);
4307 if (closing
|| atomic_read(&po
->mapped
) == 0) {
4309 spin_lock_bh(&rb_queue
->lock
);
4310 swap(rb
->pg_vec
, pg_vec
);
4311 rb
->frame_max
= (req
->tp_frame_nr
- 1);
4313 rb
->frame_size
= req
->tp_frame_size
;
4314 spin_unlock_bh(&rb_queue
->lock
);
4316 swap(rb
->pg_vec_order
, order
);
4317 swap(rb
->pg_vec_len
, req
->tp_block_nr
);
4319 rb
->pg_vec_pages
= req
->tp_block_size
/PAGE_SIZE
;
4320 po
->prot_hook
.func
= (po
->rx_ring
.pg_vec
) ?
4321 tpacket_rcv
: packet_rcv
;
4322 skb_queue_purge(rb_queue
);
4323 if (atomic_read(&po
->mapped
))
4324 pr_err("packet_mmap: vma is busy: %d\n",
4325 atomic_read(&po
->mapped
));
4327 mutex_unlock(&po
->pg_vec_lock
);
4329 spin_lock(&po
->bind_lock
);
4332 register_prot_hook(sk
);
4334 spin_unlock(&po
->bind_lock
);
4335 if (closing
&& (po
->tp_version
> TPACKET_V2
)) {
4336 /* Because we don't support block-based V3 on tx-ring */
4338 prb_shutdown_retire_blk_timer(po
, rb_queue
);
4342 free_pg_vec(pg_vec
, order
, req
->tp_block_nr
);
4348 static int packet_mmap(struct file
*file
, struct socket
*sock
,
4349 struct vm_area_struct
*vma
)
4351 struct sock
*sk
= sock
->sk
;
4352 struct packet_sock
*po
= pkt_sk(sk
);
4353 unsigned long size
, expected_size
;
4354 struct packet_ring_buffer
*rb
;
4355 unsigned long start
;
4362 mutex_lock(&po
->pg_vec_lock
);
4365 for (rb
= &po
->rx_ring
; rb
<= &po
->tx_ring
; rb
++) {
4367 expected_size
+= rb
->pg_vec_len
4373 if (expected_size
== 0)
4376 size
= vma
->vm_end
- vma
->vm_start
;
4377 if (size
!= expected_size
)
4380 start
= vma
->vm_start
;
4381 for (rb
= &po
->rx_ring
; rb
<= &po
->tx_ring
; rb
++) {
4382 if (rb
->pg_vec
== NULL
)
4385 for (i
= 0; i
< rb
->pg_vec_len
; i
++) {
4387 void *kaddr
= rb
->pg_vec
[i
].buffer
;
4390 for (pg_num
= 0; pg_num
< rb
->pg_vec_pages
; pg_num
++) {
4391 page
= pgv_to_page(kaddr
);
4392 err
= vm_insert_page(vma
, start
, page
);
4401 atomic_inc(&po
->mapped
);
4402 vma
->vm_ops
= &packet_mmap_ops
;
4406 mutex_unlock(&po
->pg_vec_lock
);
4410 static const struct proto_ops packet_ops_spkt
= {
4411 .family
= PF_PACKET
,
4412 .owner
= THIS_MODULE
,
4413 .release
= packet_release
,
4414 .bind
= packet_bind_spkt
,
4415 .connect
= sock_no_connect
,
4416 .socketpair
= sock_no_socketpair
,
4417 .accept
= sock_no_accept
,
4418 .getname
= packet_getname_spkt
,
4419 .poll
= datagram_poll
,
4420 .ioctl
= packet_ioctl
,
4421 .listen
= sock_no_listen
,
4422 .shutdown
= sock_no_shutdown
,
4423 .setsockopt
= sock_no_setsockopt
,
4424 .getsockopt
= sock_no_getsockopt
,
4425 .sendmsg
= packet_sendmsg_spkt
,
4426 .recvmsg
= packet_recvmsg
,
4427 .mmap
= sock_no_mmap
,
4428 .sendpage
= sock_no_sendpage
,
4431 static const struct proto_ops packet_ops
= {
4432 .family
= PF_PACKET
,
4433 .owner
= THIS_MODULE
,
4434 .release
= packet_release
,
4435 .bind
= packet_bind
,
4436 .connect
= sock_no_connect
,
4437 .socketpair
= sock_no_socketpair
,
4438 .accept
= sock_no_accept
,
4439 .getname
= packet_getname
,
4440 .poll
= packet_poll
,
4441 .ioctl
= packet_ioctl
,
4442 .listen
= sock_no_listen
,
4443 .shutdown
= sock_no_shutdown
,
4444 .setsockopt
= packet_setsockopt
,
4445 .getsockopt
= packet_getsockopt
,
4446 #ifdef CONFIG_COMPAT
4447 .compat_setsockopt
= compat_packet_setsockopt
,
4449 .sendmsg
= packet_sendmsg
,
4450 .recvmsg
= packet_recvmsg
,
4451 .mmap
= packet_mmap
,
4452 .sendpage
= sock_no_sendpage
,
4455 static const struct net_proto_family packet_family_ops
= {
4456 .family
= PF_PACKET
,
4457 .create
= packet_create
,
4458 .owner
= THIS_MODULE
,
4461 static struct notifier_block packet_netdev_notifier
= {
4462 .notifier_call
= packet_notifier
,
4465 #ifdef CONFIG_PROC_FS
4467 static void *packet_seq_start(struct seq_file
*seq
, loff_t
*pos
)
4470 struct net
*net
= seq_file_net(seq
);
4473 return seq_hlist_start_head_rcu(&net
->packet
.sklist
, *pos
);
4476 static void *packet_seq_next(struct seq_file
*seq
, void *v
, loff_t
*pos
)
4478 struct net
*net
= seq_file_net(seq
);
4479 return seq_hlist_next_rcu(v
, &net
->packet
.sklist
, pos
);
4482 static void packet_seq_stop(struct seq_file
*seq
, void *v
)
4488 static int packet_seq_show(struct seq_file
*seq
, void *v
)
4490 if (v
== SEQ_START_TOKEN
)
4491 seq_puts(seq
, "sk RefCnt Type Proto Iface R Rmem User Inode\n");
4493 struct sock
*s
= sk_entry(v
);
4494 const struct packet_sock
*po
= pkt_sk(s
);
4497 "%pK %-6d %-4d %04x %-5d %1d %-6u %-6u %-6lu\n",
4499 atomic_read(&s
->sk_refcnt
),
4504 atomic_read(&s
->sk_rmem_alloc
),
4505 from_kuid_munged(seq_user_ns(seq
), sock_i_uid(s
)),
4512 static const struct seq_operations packet_seq_ops
= {
4513 .start
= packet_seq_start
,
4514 .next
= packet_seq_next
,
4515 .stop
= packet_seq_stop
,
4516 .show
= packet_seq_show
,
4519 static int packet_seq_open(struct inode
*inode
, struct file
*file
)
4521 return seq_open_net(inode
, file
, &packet_seq_ops
,
4522 sizeof(struct seq_net_private
));
4525 static const struct file_operations packet_seq_fops
= {
4526 .owner
= THIS_MODULE
,
4527 .open
= packet_seq_open
,
4529 .llseek
= seq_lseek
,
4530 .release
= seq_release_net
,
4535 static int __net_init
packet_net_init(struct net
*net
)
4537 mutex_init(&net
->packet
.sklist_lock
);
4538 INIT_HLIST_HEAD(&net
->packet
.sklist
);
4540 if (!proc_create("packet", 0, net
->proc_net
, &packet_seq_fops
))
4546 static void __net_exit
packet_net_exit(struct net
*net
)
4548 remove_proc_entry("packet", net
->proc_net
);
4551 static struct pernet_operations packet_net_ops
= {
4552 .init
= packet_net_init
,
4553 .exit
= packet_net_exit
,
4557 static void __exit
packet_exit(void)
4559 unregister_netdevice_notifier(&packet_netdev_notifier
);
4560 unregister_pernet_subsys(&packet_net_ops
);
4561 sock_unregister(PF_PACKET
);
4562 proto_unregister(&packet_proto
);
4565 static int __init
packet_init(void)
4567 int rc
= proto_register(&packet_proto
, 0);
4572 sock_register(&packet_family_ops
);
4573 register_pernet_subsys(&packet_net_ops
);
4574 register_netdevice_notifier(&packet_netdev_notifier
);
4579 module_init(packet_init
);
4580 module_exit(packet_exit
);
4581 MODULE_LICENSE("GPL");
4582 MODULE_ALIAS_NETPROTO(PF_PACKET
);