blob: 3ca4f890371a1708c1854adbcdffa97cf02f534c [file] [log] [blame]
Thomas Gleixner2874c5f2019-05-27 08:55:01 +02001// SPDX-License-Identifier: GPL-2.0-or-later
Linus Torvalds1da177e2005-04-16 15:20:36 -07002/*
3 * INET An implementation of the TCP/IP protocol suite for the LINUX
4 * operating system. INET is implemented using the BSD Socket
5 * interface as the means of communication with the user level.
6 *
7 * PACKET - implements raw packet sockets.
8 *
Jesper Juhl02c30a82005-05-05 16:16:16 -07009 * Authors: Ross Biro
Linus Torvalds1da177e2005-04-16 15:20:36 -070010 * Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG>
11 * Alan Cox, <gw4pts@gw4pts.ampr.org>
12 *
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +090013 * Fixes:
Linus Torvalds1da177e2005-04-16 15:20:36 -070014 * Alan Cox : verify_area() now used correctly
15 * Alan Cox : new skbuff lists, look ma no backlogs!
16 * Alan Cox : tidied skbuff lists.
17 * Alan Cox : Now uses generic datagram routines I
18 * added. Also fixed the peek/read crash
19 * from all old Linux datagram code.
20 * Alan Cox : Uses the improved datagram code.
21 * Alan Cox : Added NULL's for socket options.
22 * Alan Cox : Re-commented the code.
23 * Alan Cox : Use new kernel side addressing
24 * Rob Janssen : Correct MTU usage.
25 * Dave Platt : Counter leaks caused by incorrect
26 * interrupt locking and some slightly
27 * dubious gcc output. Can you read
28 * compiler: it said _VOLATILE_
29 * Richard Kooijman : Timestamp fixes.
30 * Alan Cox : New buffers. Use sk->mac.raw.
31 * Alan Cox : sendmsg/recvmsg support.
32 * Alan Cox : Protocol setting support
33 * Alexey Kuznetsov : Untied from IPv4 stack.
34 * Cyrus Durgin : Fixed kerneld for kmod.
35 * Michal Ostrowski : Module initialization cleanup.
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +090036 * Ulises Alonso : Frame number limit removal and
Linus Torvalds1da177e2005-04-16 15:20:36 -070037 * packet_set_ring memory leak.
Eric W. Biederman0fb375f2005-09-21 00:11:37 -070038 * Eric Biederman : Allow for > 8 byte hardware addresses.
39 * The convention is that longer addresses
40 * will simply extend the hardware address
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +090041 * byte arrays at the end of sockaddr_ll
Eric W. Biederman0fb375f2005-09-21 00:11:37 -070042 * and packet_mreq.
Johann Baudy69e3c752009-05-18 22:11:22 -070043 * Johann Baudy : Added TX RING.
chetan lokef6fb8f12011-08-19 10:18:16 +000044 * Chetan Loke : Implemented TPACKET_V3 block abstraction
45 * layer.
46 * Copyright (C) 2011, <lokec@ccs.neu.edu>
Linus Torvalds1da177e2005-04-16 15:20:36 -070047 */
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +090048
Baruch Siachdc41c4a2021-09-09 20:49:47 +030049#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
50
Jakub Kicinskicc698372020-11-20 14:50:52 -080051#include <linux/ethtool.h>
Jakub Kicinskib6459412021-12-28 16:49:13 -080052#include <linux/filter.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070053#include <linux/types.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070054#include <linux/mm.h>
Randy Dunlap4fc268d2006-01-11 12:17:47 -080055#include <linux/capability.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070056#include <linux/fcntl.h>
57#include <linux/socket.h>
58#include <linux/in.h>
59#include <linux/inet.h>
60#include <linux/netdevice.h>
61#include <linux/if_packet.h>
62#include <linux/wireless.h>
Herbert Xuffbc6112007-02-04 23:33:10 -080063#include <linux/kernel.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070064#include <linux/kmod.h>
Tejun Heo5a0e3ad2010-03-24 17:04:11 +090065#include <linux/slab.h>
Neil Horman0e3125c2010-11-16 10:26:47 -080066#include <linux/vmalloc.h>
Eric W. Biederman457c4cb2007-09-12 12:01:34 +020067#include <net/net_namespace.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070068#include <net/ip.h>
69#include <net/protocol.h>
70#include <linux/skbuff.h>
71#include <net/sock.h>
72#include <linux/errno.h>
73#include <linux/timer.h>
Linus Torvalds7c0f6ba2016-12-24 11:46:01 -080074#include <linux/uaccess.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070075#include <asm/ioctls.h>
76#include <asm/page.h>
Al Viroa1f8e7f72006-10-19 16:08:53 -040077#include <asm/cacheflush.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070078#include <asm/io.h>
79#include <linux/proc_fs.h>
80#include <linux/seq_file.h>
81#include <linux/poll.h>
82#include <linux/module.h>
83#include <linux/init.h>
Herbert Xu905db442009-01-30 14:12:06 -080084#include <linux/mutex.h>
Eric Dumazet05423b22009-10-26 18:40:35 -070085#include <linux/if_vlan.h>
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -080086#include <linux/virtio_net.h>
Richard Cochraned85b562010-04-07 22:41:28 +000087#include <linux/errqueue.h>
Scott McMillan614f60f2010-06-02 05:53:56 -070088#include <linux/net_tstamp.h>
Daniel Borkmannb0138402014-01-15 16:25:36 +010089#include <linux/percpu.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070090#ifdef CONFIG_INET
91#include <net/inet_common.h>
92#endif
Willem de Bruijn47dceb82015-08-14 22:31:34 -040093#include <linux/bpf.h>
Willem de Bruijn719c44d2016-06-07 12:06:34 -040094#include <net/compat.h>
Pablo Neira Ayuso0d7308c2021-10-08 22:06:04 +020095#include <linux/netfilter_netdev.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070096
Pavel Emelyanov2787b042012-08-13 05:49:39 +000097#include "internal.h"
98
Linus Torvalds1da177e2005-04-16 15:20:36 -070099/*
Linus Torvalds1da177e2005-04-16 15:20:36 -0700100 Assumptions:
Eyal Birgerd5496992020-11-21 08:28:17 +0200101 - If the device has no dev->header_ops->create, there is no LL header
102 visible above the device. In this case, its hard_header_len should be 0.
Xie Heb4c58812020-09-14 00:41:54 -0700103 The device may prepend its own header internally. In this case, its
104 needed_headroom should be set to the space needed for it to add its
105 internal header.
106 For example, a WiFi driver pretending to be an Ethernet driver should
107 set its hard_header_len to be the Ethernet header length, and set its
108 needed_headroom to be (the real WiFi header length - the fake Ethernet
109 header length).
Linus Torvalds1da177e2005-04-16 15:20:36 -0700110 - packet socket receives packets with pulled ll header,
111 so that SOCK_RAW should push it back.
112
113On receive:
114-----------
115
Eyal Birgerd5496992020-11-21 08:28:17 +0200116Incoming, dev_has_header(dev) == true
Arnaldo Carvalho de Melob0e380b2007-04-10 21:21:55 -0700117 mac_header -> ll header
118 data -> data
Linus Torvalds1da177e2005-04-16 15:20:36 -0700119
Eyal Birgerd5496992020-11-21 08:28:17 +0200120Outgoing, dev_has_header(dev) == true
Arnaldo Carvalho de Melob0e380b2007-04-10 21:21:55 -0700121 mac_header -> ll header
122 data -> ll header
Linus Torvalds1da177e2005-04-16 15:20:36 -0700123
Eyal Birgerd5496992020-11-21 08:28:17 +0200124Incoming, dev_has_header(dev) == false
Xie Heb79a80b2020-09-16 05:23:08 -0700125 mac_header -> data
126 However drivers often make it point to the ll header.
127 This is incorrect because the ll header should be invisible to us.
Arnaldo Carvalho de Melob0e380b2007-04-10 21:21:55 -0700128 data -> data
Linus Torvalds1da177e2005-04-16 15:20:36 -0700129
Eyal Birgerd5496992020-11-21 08:28:17 +0200130Outgoing, dev_has_header(dev) == false
Xie Heb79a80b2020-09-16 05:23:08 -0700131 mac_header -> data. ll header is invisible to us.
Arnaldo Carvalho de Melob0e380b2007-04-10 21:21:55 -0700132 data -> data
Linus Torvalds1da177e2005-04-16 15:20:36 -0700133
134Resume
Eyal Birgerd5496992020-11-21 08:28:17 +0200135 If dev_has_header(dev) == false we are unable to restore the ll header,
Xie Heb79a80b2020-09-16 05:23:08 -0700136 because it is invisible to us.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700137
138
139On transmit:
140------------
141
Xie He21c859742021-02-05 14:41:24 -0800142dev_has_header(dev) == true
Arnaldo Carvalho de Melob0e380b2007-04-10 21:21:55 -0700143 mac_header -> ll header
144 data -> ll header
Linus Torvalds1da177e2005-04-16 15:20:36 -0700145
Xie He21c859742021-02-05 14:41:24 -0800146dev_has_header(dev) == false (ll header is invisible to us)
Arnaldo Carvalho de Melob0e380b2007-04-10 21:21:55 -0700147 mac_header -> data
148 data -> data
Linus Torvalds1da177e2005-04-16 15:20:36 -0700149
Xie He09599722020-09-18 06:56:16 -0700150 We should set network_header on output to the correct position,
Linus Torvalds1da177e2005-04-16 15:20:36 -0700151 packet classifier depends on it.
152 */
153
Linus Torvalds1da177e2005-04-16 15:20:36 -0700154/* Private packet socket structures. */
155
Eric W. Biederman0fb375f2005-09-21 00:11:37 -0700156/* identical to struct packet_mreq except it has
157 * a longer address field.
158 */
Eric Dumazet40d4e3d2009-07-21 21:57:59 +0000159struct packet_mreq_max {
Eric W. Biederman0fb375f2005-09-21 00:11:37 -0700160 int mr_ifindex;
161 unsigned short mr_type;
162 unsigned short mr_alen;
163 unsigned char mr_address[MAX_ADDR_LEN];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700164};
David S. Millera2efcfa2007-05-29 13:12:50 -0700165
Daniel Borkmann184f4892013-04-16 01:57:46 +0000166union tpacket_uhdr {
167 struct tpacket_hdr *h1;
168 struct tpacket2_hdr *h2;
169 struct tpacket3_hdr *h3;
170 void *raw;
171};
172
chetan lokef6fb8f12011-08-19 10:18:16 +0000173static int packet_set_ring(struct sock *sk, union tpacket_req_u *req_u,
Johann Baudy69e3c752009-05-18 22:11:22 -0700174 int closing, int tx_ring);
175
chetan lokef6fb8f12011-08-19 10:18:16 +0000176#define V3_ALIGNMENT (8)
177
chetan lokebc59ba32011-08-25 10:43:30 +0000178#define BLK_HDR_LEN (ALIGN(sizeof(struct tpacket_block_desc), V3_ALIGNMENT))
chetan lokef6fb8f12011-08-19 10:18:16 +0000179
180#define BLK_PLUS_PRIV(sz_of_priv) \
181 (BLK_HDR_LEN + ALIGN((sz_of_priv), V3_ALIGNMENT))
182
chetan lokef6fb8f12011-08-19 10:18:16 +0000183#define BLOCK_STATUS(x) ((x)->hdr.bh1.block_status)
184#define BLOCK_NUM_PKTS(x) ((x)->hdr.bh1.num_pkts)
185#define BLOCK_O2FP(x) ((x)->hdr.bh1.offset_to_first_pkt)
186#define BLOCK_LEN(x) ((x)->hdr.bh1.blk_len)
187#define BLOCK_SNUM(x) ((x)->hdr.bh1.seq_num)
188#define BLOCK_O2PRIV(x) ((x)->offset_to_priv)
chetan lokef6fb8f12011-08-19 10:18:16 +0000189
Johann Baudy69e3c752009-05-18 22:11:22 -0700190struct packet_sock;
Willem de Bruijn77f65eb2013-03-19 10:18:11 +0000191static int tpacket_rcv(struct sk_buff *skb, struct net_device *dev,
192 struct packet_type *pt, struct net_device *orig_dev);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700193
chetan lokef6fb8f12011-08-19 10:18:16 +0000194static void *packet_previous_frame(struct packet_sock *po,
195 struct packet_ring_buffer *rb,
196 int status);
197static void packet_increment_head(struct packet_ring_buffer *buff);
Rosen, Rami878cd3b2017-05-24 18:34:11 +0300198static int prb_curr_blk_in_use(struct tpacket_block_desc *);
chetan lokebc59ba32011-08-25 10:43:30 +0000199static void *prb_dispatch_next_block(struct tpacket_kbdq_core *,
chetan lokef6fb8f12011-08-19 10:18:16 +0000200 struct packet_sock *);
chetan lokebc59ba32011-08-25 10:43:30 +0000201static void prb_retire_current_block(struct tpacket_kbdq_core *,
chetan lokef6fb8f12011-08-19 10:18:16 +0000202 struct packet_sock *, unsigned int status);
chetan lokebc59ba32011-08-25 10:43:30 +0000203static int prb_queue_frozen(struct tpacket_kbdq_core *);
204static void prb_open_block(struct tpacket_kbdq_core *,
205 struct tpacket_block_desc *);
Kees Cook17bfd8c2017-10-24 01:46:26 -0700206static void prb_retire_rx_blk_timer_expired(struct timer_list *);
chetan lokebc59ba32011-08-25 10:43:30 +0000207static void _prb_refresh_rx_retire_blk_timer(struct tpacket_kbdq_core *);
chetan lokebc59ba32011-08-25 10:43:30 +0000208static void prb_fill_rxhash(struct tpacket_kbdq_core *, struct tpacket3_hdr *);
209static void prb_clear_rxhash(struct tpacket_kbdq_core *,
210 struct tpacket3_hdr *);
211static void prb_fill_vlan_info(struct tpacket_kbdq_core *,
212 struct tpacket3_hdr *);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700213static void packet_flush_mclist(struct sock *sk);
Magnus Karlsson865b03f2018-05-02 13:01:33 +0200214static u16 packet_pick_tx_queue(struct sk_buff *skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700215
Herbert Xuffbc6112007-02-04 23:33:10 -0800216struct packet_skb_cb {
Herbert Xuffbc6112007-02-04 23:33:10 -0800217 union {
218 struct sockaddr_pkt pkt;
Eyal Birger2472d762015-03-01 14:58:28 +0200219 union {
220 /* Trick: alias skb original length with
221 * ll.sll_family and ll.protocol in order
222 * to save room.
223 */
224 unsigned int origlen;
225 struct sockaddr_ll ll;
226 };
Herbert Xuffbc6112007-02-04 23:33:10 -0800227 } sa;
228};
229
David Woodhoused3869ef2015-09-23 19:45:08 +0100230#define vio_le() virtio_legacy_is_little_endian()
231
Herbert Xuffbc6112007-02-04 23:33:10 -0800232#define PACKET_SKB_CB(__skb) ((struct packet_skb_cb *)((__skb)->cb))
Herbert Xu8dc41942007-02-04 23:31:32 -0800233
chetan lokebc59ba32011-08-25 10:43:30 +0000234#define GET_PBDQC_FROM_RB(x) ((struct tpacket_kbdq_core *)(&(x)->prb_bdqc))
chetan lokef6fb8f12011-08-19 10:18:16 +0000235#define GET_PBLOCK_DESC(x, bid) \
chetan lokebc59ba32011-08-25 10:43:30 +0000236 ((struct tpacket_block_desc *)((x)->pkbdq[(bid)].buffer))
chetan lokef6fb8f12011-08-19 10:18:16 +0000237#define GET_CURR_PBLOCK_DESC_FROM_CORE(x) \
chetan lokebc59ba32011-08-25 10:43:30 +0000238 ((struct tpacket_block_desc *)((x)->pkbdq[(x)->kactive_blk_num].buffer))
chetan lokef6fb8f12011-08-19 10:18:16 +0000239#define GET_NEXT_PRB_BLK_NUM(x) \
240 (((x)->kactive_blk_num < ((x)->knum_blocks-1)) ? \
241 ((x)->kactive_blk_num+1) : 0)
242
David S. Millerdc99f602011-07-05 01:45:05 -0700243static void __fanout_unlink(struct sock *sk, struct packet_sock *po);
244static void __fanout_link(struct sock *sk, struct packet_sock *po);
245
Pablo Neira Ayuso0d7308c2021-10-08 22:06:04 +0200246#ifdef CONFIG_NETFILTER_EGRESS
247static noinline struct sk_buff *nf_hook_direct_egress(struct sk_buff *skb)
248{
249 struct sk_buff *next, *head = NULL, *tail;
250 int rc;
251
252 rcu_read_lock();
253 for (; skb != NULL; skb = next) {
254 next = skb->next;
255 skb_mark_not_on_list(skb);
256
257 if (!nf_hook_egress(skb, &rc, skb->dev))
258 continue;
259
260 if (!head)
261 head = skb;
262 else
263 tail->next = skb;
264
265 tail = skb;
266 }
267 rcu_read_unlock();
268
269 return head;
270}
271#endif
272
Daniel Borkmannd346a3f2013-12-06 11:36:17 +0100273static int packet_direct_xmit(struct sk_buff *skb)
274{
Pablo Neira Ayuso0d7308c2021-10-08 22:06:04 +0200275#ifdef CONFIG_NETFILTER_EGRESS
276 if (nf_hook_egress_active()) {
277 skb = nf_hook_direct_egress(skb);
278 if (!skb)
279 return NET_XMIT_DROP;
280 }
281#endif
Magnus Karlsson865b03f2018-05-02 13:01:33 +0200282 return dev_direct_xmit(skb, packet_pick_tx_queue(skb));
Daniel Borkmannd346a3f2013-12-06 11:36:17 +0100283}
284
Daniel Borkmann66e56cd2013-12-06 11:36:15 +0100285static struct net_device *packet_cached_dev_get(struct packet_sock *po)
286{
287 struct net_device *dev;
288
289 rcu_read_lock();
290 dev = rcu_dereference(po->cached_dev);
Yajun Deng1160dfa2021-08-05 19:55:27 +0800291 dev_hold(dev);
Daniel Borkmann66e56cd2013-12-06 11:36:15 +0100292 rcu_read_unlock();
293
294 return dev;
295}
296
297static void packet_cached_dev_assign(struct packet_sock *po,
298 struct net_device *dev)
299{
300 rcu_assign_pointer(po->cached_dev, dev);
301}
302
303static void packet_cached_dev_reset(struct packet_sock *po)
304{
305 RCU_INIT_POINTER(po->cached_dev, NULL);
306}
307
Daniel Borkmannd346a3f2013-12-06 11:36:17 +0100308static bool packet_use_direct_xmit(const struct packet_sock *po)
309{
310 return po->xmit == packet_direct_xmit;
311}
312
Magnus Karlsson865b03f2018-05-02 13:01:33 +0200313static u16 packet_pick_tx_queue(struct sk_buff *skb)
Daniel Borkmann0fd5d572014-02-16 15:55:22 +0100314{
Magnus Karlsson865b03f2018-05-02 13:01:33 +0200315 struct net_device *dev = skb->dev;
Daniel Borkmann0fd5d572014-02-16 15:55:22 +0100316 const struct net_device_ops *ops = dev->netdev_ops;
Paolo Abenib71b5832019-03-20 11:02:05 +0100317 int cpu = raw_smp_processor_id();
Daniel Borkmann0fd5d572014-02-16 15:55:22 +0100318 u16 queue_index;
319
Paolo Abenib71b5832019-03-20 11:02:05 +0100320#ifdef CONFIG_XPS
321 skb->sender_cpu = cpu + 1;
322#endif
323 skb_record_rx_queue(skb, cpu % dev->real_num_tx_queues);
Daniel Borkmann0fd5d572014-02-16 15:55:22 +0100324 if (ops->ndo_select_queue) {
Paolo Abenia350ecc2019-03-20 11:02:06 +0100325 queue_index = ops->ndo_select_queue(dev, skb, NULL);
Daniel Borkmann0fd5d572014-02-16 15:55:22 +0100326 queue_index = netdev_cap_txqueue(dev, queue_index);
327 } else {
Paolo Abenib71b5832019-03-20 11:02:05 +0100328 queue_index = netdev_pick_tx(dev, skb, NULL);
Daniel Borkmann0fd5d572014-02-16 15:55:22 +0100329 }
330
Magnus Karlsson865b03f2018-05-02 13:01:33 +0200331 return queue_index;
Daniel Borkmann0fd5d572014-02-16 15:55:22 +0100332}
333
Willem de Bruijna6361f02018-04-23 17:37:03 -0400334/* __register_prot_hook must be invoked through register_prot_hook
David S. Millerce06b032011-07-04 01:44:29 -0700335 * or from a context in which asynchronous accesses to the packet
336 * socket is not possible (packet_create()).
337 */
Willem de Bruijna6361f02018-04-23 17:37:03 -0400338static void __register_prot_hook(struct sock *sk)
David S. Millerce06b032011-07-04 01:44:29 -0700339{
340 struct packet_sock *po = pkt_sk(sk);
Daniel Borkmanne40526c2013-11-21 16:50:58 +0100341
David S. Millerce06b032011-07-04 01:44:29 -0700342 if (!po->running) {
Daniel Borkmann66e56cd2013-12-06 11:36:15 +0100343 if (po->fanout)
David S. Millerdc99f602011-07-05 01:45:05 -0700344 __fanout_link(sk, po);
Daniel Borkmann66e56cd2013-12-06 11:36:15 +0100345 else
David S. Millerdc99f602011-07-05 01:45:05 -0700346 dev_add_pack(&po->prot_hook);
Daniel Borkmanne40526c2013-11-21 16:50:58 +0100347
David S. Millerce06b032011-07-04 01:44:29 -0700348 sock_hold(sk);
349 po->running = 1;
350 }
351}
352
Willem de Bruijna6361f02018-04-23 17:37:03 -0400353static void register_prot_hook(struct sock *sk)
354{
355 lockdep_assert_held_once(&pkt_sk(sk)->bind_lock);
356 __register_prot_hook(sk);
357}
358
359/* If the sync parameter is true, we will temporarily drop
David S. Millerce06b032011-07-04 01:44:29 -0700360 * the po->bind_lock and do a synchronize_net to make sure no
361 * asynchronous packet processing paths still refer to the elements
362 * of po->prot_hook. If the sync parameter is false, it is the
363 * callers responsibility to take care of this.
364 */
365static void __unregister_prot_hook(struct sock *sk, bool sync)
366{
367 struct packet_sock *po = pkt_sk(sk);
368
Willem de Bruijna6361f02018-04-23 17:37:03 -0400369 lockdep_assert_held_once(&po->bind_lock);
370
David S. Millerce06b032011-07-04 01:44:29 -0700371 po->running = 0;
Daniel Borkmann66e56cd2013-12-06 11:36:15 +0100372
373 if (po->fanout)
David S. Millerdc99f602011-07-05 01:45:05 -0700374 __fanout_unlink(sk, po);
Daniel Borkmann66e56cd2013-12-06 11:36:15 +0100375 else
David S. Millerdc99f602011-07-05 01:45:05 -0700376 __dev_remove_pack(&po->prot_hook);
Daniel Borkmanne40526c2013-11-21 16:50:58 +0100377
David S. Millerce06b032011-07-04 01:44:29 -0700378 __sock_put(sk);
379
380 if (sync) {
381 spin_unlock(&po->bind_lock);
382 synchronize_net();
383 spin_lock(&po->bind_lock);
384 }
385}
386
387static void unregister_prot_hook(struct sock *sk, bool sync)
388{
389 struct packet_sock *po = pkt_sk(sk);
390
391 if (po->running)
392 __unregister_prot_hook(sk, sync);
393}
394
Michael S. Tsirkin6e580402014-11-24 13:32:16 +0200395static inline struct page * __pure pgv_to_page(void *addr)
Changli Gao0af55bb2010-12-01 02:52:20 +0000396{
397 if (is_vmalloc_addr(addr))
398 return vmalloc_to_page(addr);
399 return virt_to_page(addr);
400}
401
Patrick McHardybbd6ef82008-07-14 22:50:15 -0700402static void __packet_set_status(struct packet_sock *po, void *frame, int status)
403{
Daniel Borkmann184f4892013-04-16 01:57:46 +0000404 union tpacket_uhdr h;
Patrick McHardybbd6ef82008-07-14 22:50:15 -0700405
406 h.raw = frame;
407 switch (po->tp_version) {
408 case TPACKET_V1:
409 h.h1->tp_status = status;
Changli Gao0af55bb2010-12-01 02:52:20 +0000410 flush_dcache_page(pgv_to_page(&h.h1->tp_status));
Patrick McHardybbd6ef82008-07-14 22:50:15 -0700411 break;
412 case TPACKET_V2:
413 h.h2->tp_status = status;
Changli Gao0af55bb2010-12-01 02:52:20 +0000414 flush_dcache_page(pgv_to_page(&h.h2->tp_status));
Patrick McHardybbd6ef82008-07-14 22:50:15 -0700415 break;
chetan lokef6fb8f12011-08-19 10:18:16 +0000416 case TPACKET_V3:
Sowmini Varadhan7f953ab2017-01-03 06:31:47 -0800417 h.h3->tp_status = status;
418 flush_dcache_page(pgv_to_page(&h.h3->tp_status));
419 break;
Johann Baudy69e3c752009-05-18 22:11:22 -0700420 default:
chetan lokef6fb8f12011-08-19 10:18:16 +0000421 WARN(1, "TPACKET version not supported.\n");
Johann Baudy69e3c752009-05-18 22:11:22 -0700422 BUG();
423 }
424
425 smp_wmb();
426}
427
Eric Dumazet96f657e2019-06-12 09:52:26 -0700428static int __packet_get_status(const struct packet_sock *po, void *frame)
Johann Baudy69e3c752009-05-18 22:11:22 -0700429{
Daniel Borkmann184f4892013-04-16 01:57:46 +0000430 union tpacket_uhdr h;
Johann Baudy69e3c752009-05-18 22:11:22 -0700431
432 smp_rmb();
433
434 h.raw = frame;
435 switch (po->tp_version) {
436 case TPACKET_V1:
Changli Gao0af55bb2010-12-01 02:52:20 +0000437 flush_dcache_page(pgv_to_page(&h.h1->tp_status));
Johann Baudy69e3c752009-05-18 22:11:22 -0700438 return h.h1->tp_status;
439 case TPACKET_V2:
Changli Gao0af55bb2010-12-01 02:52:20 +0000440 flush_dcache_page(pgv_to_page(&h.h2->tp_status));
Johann Baudy69e3c752009-05-18 22:11:22 -0700441 return h.h2->tp_status;
chetan lokef6fb8f12011-08-19 10:18:16 +0000442 case TPACKET_V3:
Sowmini Varadhan7f953ab2017-01-03 06:31:47 -0800443 flush_dcache_page(pgv_to_page(&h.h3->tp_status));
444 return h.h3->tp_status;
Johann Baudy69e3c752009-05-18 22:11:22 -0700445 default:
chetan lokef6fb8f12011-08-19 10:18:16 +0000446 WARN(1, "TPACKET version not supported.\n");
Johann Baudy69e3c752009-05-18 22:11:22 -0700447 BUG();
448 return 0;
Patrick McHardybbd6ef82008-07-14 22:50:15 -0700449 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700450}
Johann Baudy69e3c752009-05-18 22:11:22 -0700451
Arnd Bergmannd413fcb2017-11-27 10:09:24 +0100452static __u32 tpacket_get_timestamp(struct sk_buff *skb, struct timespec64 *ts,
Daniel Borkmannb9c32fb2013-04-23 00:39:31 +0000453 unsigned int flags)
Daniel Borkmann7a513842013-04-23 00:39:29 +0000454{
455 struct skb_shared_hwtstamps *shhwtstamps = skb_hwtstamps(skb);
456
Willem de Bruijn68a360e2014-07-25 18:01:31 -0400457 if (shhwtstamps &&
458 (flags & SOF_TIMESTAMPING_RAW_HARDWARE) &&
Arnd Bergmannd413fcb2017-11-27 10:09:24 +0100459 ktime_to_timespec64_cond(shhwtstamps->hwtstamp, ts))
Willem de Bruijn68a360e2014-07-25 18:01:31 -0400460 return TP_STATUS_TS_RAW_HARDWARE;
Daniel Borkmann7a513842013-04-23 00:39:29 +0000461
Richard Sanger171c3b12021-05-12 13:31:22 +1200462 if ((flags & SOF_TIMESTAMPING_SOFTWARE) &&
463 ktime_to_timespec64_cond(skb->tstamp, ts))
Daniel Borkmannb9c32fb2013-04-23 00:39:31 +0000464 return TP_STATUS_TS_SOFTWARE;
Daniel Borkmann7a513842013-04-23 00:39:29 +0000465
Daniel Borkmannb9c32fb2013-04-23 00:39:31 +0000466 return 0;
Daniel Borkmann7a513842013-04-23 00:39:29 +0000467}
468
Daniel Borkmannb9c32fb2013-04-23 00:39:31 +0000469static __u32 __packet_set_timestamp(struct packet_sock *po, void *frame,
470 struct sk_buff *skb)
Willem de Bruijn2e313962013-04-23 00:39:28 +0000471{
472 union tpacket_uhdr h;
Arnd Bergmannd413fcb2017-11-27 10:09:24 +0100473 struct timespec64 ts;
Daniel Borkmannb9c32fb2013-04-23 00:39:31 +0000474 __u32 ts_status;
Willem de Bruijn2e313962013-04-23 00:39:28 +0000475
Daniel Borkmannb9c32fb2013-04-23 00:39:31 +0000476 if (!(ts_status = tpacket_get_timestamp(skb, &ts, po->tp_tstamp)))
477 return 0;
Willem de Bruijn2e313962013-04-23 00:39:28 +0000478
479 h.raw = frame;
Arnd Bergmannd413fcb2017-11-27 10:09:24 +0100480 /*
481 * versions 1 through 3 overflow the timestamps in y2106, since they
482 * all store the seconds in a 32-bit unsigned integer.
483 * If we create a version 4, that should have a 64-bit timestamp,
484 * either 64-bit seconds + 32-bit nanoseconds, or just 64-bit
485 * nanoseconds.
486 */
Willem de Bruijn2e313962013-04-23 00:39:28 +0000487 switch (po->tp_version) {
488 case TPACKET_V1:
489 h.h1->tp_sec = ts.tv_sec;
490 h.h1->tp_usec = ts.tv_nsec / NSEC_PER_USEC;
491 break;
492 case TPACKET_V2:
493 h.h2->tp_sec = ts.tv_sec;
494 h.h2->tp_nsec = ts.tv_nsec;
495 break;
496 case TPACKET_V3:
Daniel Borkmann57ea8842017-01-05 02:34:28 +0100497 h.h3->tp_sec = ts.tv_sec;
498 h.h3->tp_nsec = ts.tv_nsec;
499 break;
Willem de Bruijn2e313962013-04-23 00:39:28 +0000500 default:
501 WARN(1, "TPACKET version not supported.\n");
502 BUG();
503 }
504
505 /* one flush is safe, as both fields always lie on the same cacheline */
506 flush_dcache_page(pgv_to_page(&h.h1->tp_sec));
507 smp_wmb();
Daniel Borkmannb9c32fb2013-04-23 00:39:31 +0000508
509 return ts_status;
Willem de Bruijn2e313962013-04-23 00:39:28 +0000510}
511
Eric Dumazetd4b5bd92019-06-12 09:52:27 -0700512static void *packet_lookup_frame(const struct packet_sock *po,
513 const struct packet_ring_buffer *rb,
514 unsigned int position,
515 int status)
Johann Baudy69e3c752009-05-18 22:11:22 -0700516{
517 unsigned int pg_vec_pos, frame_offset;
Daniel Borkmann184f4892013-04-16 01:57:46 +0000518 union tpacket_uhdr h;
Johann Baudy69e3c752009-05-18 22:11:22 -0700519
520 pg_vec_pos = position / rb->frames_per_block;
521 frame_offset = position % rb->frames_per_block;
522
Neil Horman0e3125c2010-11-16 10:26:47 -0800523 h.raw = rb->pg_vec[pg_vec_pos].buffer +
524 (frame_offset * rb->frame_size);
Johann Baudy69e3c752009-05-18 22:11:22 -0700525
526 if (status != __packet_get_status(po, h.raw))
527 return NULL;
528
529 return h.raw;
530}
531
Olof Johanssoneea49cc92011-11-02 11:00:49 +0000532static void *packet_current_frame(struct packet_sock *po,
Johann Baudy69e3c752009-05-18 22:11:22 -0700533 struct packet_ring_buffer *rb,
534 int status)
535{
536 return packet_lookup_frame(po, rb, rb->head, status);
537}
538
chetan lokebc59ba32011-08-25 10:43:30 +0000539static void prb_del_retire_blk_timer(struct tpacket_kbdq_core *pkc)
chetan lokef6fb8f12011-08-19 10:18:16 +0000540{
541 del_timer_sync(&pkc->retire_blk_timer);
542}
543
544static void prb_shutdown_retire_blk_timer(struct packet_sock *po,
chetan lokef6fb8f12011-08-19 10:18:16 +0000545 struct sk_buff_head *rb_queue)
546{
chetan lokebc59ba32011-08-25 10:43:30 +0000547 struct tpacket_kbdq_core *pkc;
chetan lokef6fb8f12011-08-19 10:18:16 +0000548
Tobias Klauser73d0fcf2015-07-28 14:21:26 +0200549 pkc = GET_PBDQC_FROM_RB(&po->rx_ring);
chetan lokef6fb8f12011-08-19 10:18:16 +0000550
Veaceslav Falicoec6f8092013-11-29 09:53:23 +0100551 spin_lock_bh(&rb_queue->lock);
chetan lokef6fb8f12011-08-19 10:18:16 +0000552 pkc->delete_blk_timer = 1;
Veaceslav Falicoec6f8092013-11-29 09:53:23 +0100553 spin_unlock_bh(&rb_queue->lock);
chetan lokef6fb8f12011-08-19 10:18:16 +0000554
555 prb_del_retire_blk_timer(pkc);
556}
557
Maninder Singhe8e85cc2015-06-22 12:39:16 +0530558static void prb_setup_retire_blk_timer(struct packet_sock *po)
chetan lokef6fb8f12011-08-19 10:18:16 +0000559{
chetan lokebc59ba32011-08-25 10:43:30 +0000560 struct tpacket_kbdq_core *pkc;
chetan lokef6fb8f12011-08-19 10:18:16 +0000561
Maninder Singhe8e85cc2015-06-22 12:39:16 +0530562 pkc = GET_PBDQC_FROM_RB(&po->rx_ring);
Kees Cook17bfd8c2017-10-24 01:46:26 -0700563 timer_setup(&pkc->retire_blk_timer, prb_retire_rx_blk_timer_expired,
564 0);
565 pkc->retire_blk_timer.expires = jiffies;
chetan lokef6fb8f12011-08-19 10:18:16 +0000566}
567
568static int prb_calc_retire_blk_tmo(struct packet_sock *po,
569 int blk_size_in_bytes)
570{
571 struct net_device *dev;
Mao Wenan0914d2b2019-12-23 18:42:57 +0800572 unsigned int mbits, div;
David Decotigny7cad1ba2016-02-24 10:58:10 -0800573 struct ethtool_link_ksettings ecmd;
Jiri Pirko4bc71cb2011-09-03 03:34:30 +0000574 int err;
chetan lokef6fb8f12011-08-19 10:18:16 +0000575
Jiri Pirko4bc71cb2011-09-03 03:34:30 +0000576 rtnl_lock();
577 dev = __dev_get_by_index(sock_net(&po->sk), po->ifindex);
578 if (unlikely(!dev)) {
579 rtnl_unlock();
chetan lokef6fb8f12011-08-19 10:18:16 +0000580 return DEFAULT_PRB_RETIRE_TOV;
Jiri Pirko4bc71cb2011-09-03 03:34:30 +0000581 }
David Decotigny7cad1ba2016-02-24 10:58:10 -0800582 err = __ethtool_get_link_ksettings(dev, &ecmd);
Jiri Pirko4bc71cb2011-09-03 03:34:30 +0000583 rtnl_unlock();
Mao Wenan0914d2b2019-12-23 18:42:57 +0800584 if (err)
Mao Wenanb43d1f92019-12-09 21:31:25 +0800585 return DEFAULT_PRB_RETIRE_TOV;
chetan lokef6fb8f12011-08-19 10:18:16 +0000586
Mao Wenan0914d2b2019-12-23 18:42:57 +0800587 /* If the link speed is so slow you don't really
588 * need to worry about perf anyways
589 */
590 if (ecmd.base.speed < SPEED_1000 ||
591 ecmd.base.speed == SPEED_UNKNOWN)
592 return DEFAULT_PRB_RETIRE_TOV;
593
594 div = ecmd.base.speed / 1000;
chetan lokef6fb8f12011-08-19 10:18:16 +0000595 mbits = (blk_size_in_bytes * 8) / (1024 * 1024);
596
597 if (div)
598 mbits /= div;
599
chetan lokef6fb8f12011-08-19 10:18:16 +0000600 if (div)
Mao Wenan0914d2b2019-12-23 18:42:57 +0800601 return mbits + 1;
602 return mbits;
chetan lokef6fb8f12011-08-19 10:18:16 +0000603}
604
chetan lokebc59ba32011-08-25 10:43:30 +0000605static void prb_init_ft_ops(struct tpacket_kbdq_core *p1,
chetan lokef6fb8f12011-08-19 10:18:16 +0000606 union tpacket_req_u *req_u)
607{
608 p1->feature_req_word = req_u->req3.tp_feature_req_word;
609}
610
611static void init_prb_bdqc(struct packet_sock *po,
612 struct packet_ring_buffer *rb,
613 struct pgv *pg_vec,
Maninder Singhe8e85cc2015-06-22 12:39:16 +0530614 union tpacket_req_u *req_u)
chetan lokef6fb8f12011-08-19 10:18:16 +0000615{
Duan Jiong22781a52013-12-06 13:29:36 +0800616 struct tpacket_kbdq_core *p1 = GET_PBDQC_FROM_RB(rb);
chetan lokebc59ba32011-08-25 10:43:30 +0000617 struct tpacket_block_desc *pbd;
chetan lokef6fb8f12011-08-19 10:18:16 +0000618
619 memset(p1, 0x0, sizeof(*p1));
620
621 p1->knxt_seq_num = 1;
622 p1->pkbdq = pg_vec;
chetan lokebc59ba32011-08-25 10:43:30 +0000623 pbd = (struct tpacket_block_desc *)pg_vec[0].buffer;
Joe Perchese3192692012-06-03 17:41:40 +0000624 p1->pkblk_start = pg_vec[0].buffer;
chetan lokef6fb8f12011-08-19 10:18:16 +0000625 p1->kblk_size = req_u->req3.tp_block_size;
626 p1->knum_blocks = req_u->req3.tp_block_nr;
627 p1->hdrlen = po->tp_hdrlen;
628 p1->version = po->tp_version;
629 p1->last_kactive_blk_num = 0;
Daniel Borkmannee80fbf2013-04-19 06:12:29 +0000630 po->stats.stats3.tp_freeze_q_cnt = 0;
chetan lokef6fb8f12011-08-19 10:18:16 +0000631 if (req_u->req3.tp_retire_blk_tov)
632 p1->retire_blk_tov = req_u->req3.tp_retire_blk_tov;
633 else
634 p1->retire_blk_tov = prb_calc_retire_blk_tmo(po,
635 req_u->req3.tp_block_size);
636 p1->tov_in_jiffies = msecs_to_jiffies(p1->retire_blk_tov);
637 p1->blk_sizeof_priv = req_u->req3.tp_sizeof_priv;
John Ogness632ca502020-07-07 17:28:04 +0206638 rwlock_init(&p1->blk_fill_in_prog_lock);
chetan lokef6fb8f12011-08-19 10:18:16 +0000639
Eric Dumazetdc808112014-08-15 09:16:04 -0700640 p1->max_frame_len = p1->kblk_size - BLK_PLUS_PRIV(p1->blk_sizeof_priv);
chetan lokef6fb8f12011-08-19 10:18:16 +0000641 prb_init_ft_ops(p1, req_u);
Maninder Singhe8e85cc2015-06-22 12:39:16 +0530642 prb_setup_retire_blk_timer(po);
chetan lokef6fb8f12011-08-19 10:18:16 +0000643 prb_open_block(p1, pbd);
644}
645
646/* Do NOT update the last_blk_num first.
647 * Assumes sk_buff_head lock is held.
648 */
chetan lokebc59ba32011-08-25 10:43:30 +0000649static void _prb_refresh_rx_retire_blk_timer(struct tpacket_kbdq_core *pkc)
chetan lokef6fb8f12011-08-19 10:18:16 +0000650{
651 mod_timer(&pkc->retire_blk_timer,
652 jiffies + pkc->tov_in_jiffies);
653 pkc->last_kactive_blk_num = pkc->kactive_blk_num;
654}
655
656/*
657 * Timer logic:
658 * 1) We refresh the timer only when we open a block.
659 * By doing this we don't waste cycles refreshing the timer
660 * on packet-by-packet basis.
661 *
662 * With a 1MB block-size, on a 1Gbps line, it will take
663 * i) ~8 ms to fill a block + ii) memcpy etc.
664 * In this cut we are not accounting for the memcpy time.
665 *
666 * So, if the user sets the 'tmo' to 10ms then the timer
667 * will never fire while the block is still getting filled
668 * (which is what we want). However, the user could choose
669 * to close a block early and that's fine.
670 *
671 * But when the timer does fire, we check whether or not to refresh it.
672 * Since the tmo granularity is in msecs, it is not too expensive
673 * to refresh the timer, lets say every '8' msecs.
674 * Either the user can set the 'tmo' or we can derive it based on
675 * a) line-speed and b) block-size.
676 * prb_calc_retire_blk_tmo() calculates the tmo.
677 *
678 */
Kees Cook17bfd8c2017-10-24 01:46:26 -0700679static void prb_retire_rx_blk_timer_expired(struct timer_list *t)
chetan lokef6fb8f12011-08-19 10:18:16 +0000680{
Kees Cook17bfd8c2017-10-24 01:46:26 -0700681 struct packet_sock *po =
682 from_timer(po, t, rx_ring.prb_bdqc.retire_blk_timer);
Duan Jiong22781a52013-12-06 13:29:36 +0800683 struct tpacket_kbdq_core *pkc = GET_PBDQC_FROM_RB(&po->rx_ring);
chetan lokef6fb8f12011-08-19 10:18:16 +0000684 unsigned int frozen;
chetan lokebc59ba32011-08-25 10:43:30 +0000685 struct tpacket_block_desc *pbd;
chetan lokef6fb8f12011-08-19 10:18:16 +0000686
687 spin_lock(&po->sk.sk_receive_queue.lock);
688
689 frozen = prb_queue_frozen(pkc);
690 pbd = GET_CURR_PBLOCK_DESC_FROM_CORE(pkc);
691
692 if (unlikely(pkc->delete_blk_timer))
693 goto out;
694
695 /* We only need to plug the race when the block is partially filled.
696 * tpacket_rcv:
697 * lock(); increment BLOCK_NUM_PKTS; unlock()
698 * copy_bits() is in progress ...
699 * timer fires on other cpu:
700 * we can't retire the current block because copy_bits
701 * is in progress.
702 *
703 */
704 if (BLOCK_NUM_PKTS(pbd)) {
John Ogness632ca502020-07-07 17:28:04 +0206705 /* Waiting for skb_copy_bits to finish... */
706 write_lock(&pkc->blk_fill_in_prog_lock);
707 write_unlock(&pkc->blk_fill_in_prog_lock);
chetan lokef6fb8f12011-08-19 10:18:16 +0000708 }
709
710 if (pkc->last_kactive_blk_num == pkc->kactive_blk_num) {
711 if (!frozen) {
Alexander Drozdov41a50d62015-02-24 08:18:28 +0300712 if (!BLOCK_NUM_PKTS(pbd)) {
713 /* An empty block. Just refresh the timer. */
714 goto refresh_timer;
715 }
chetan lokef6fb8f12011-08-19 10:18:16 +0000716 prb_retire_current_block(pkc, po, TP_STATUS_BLK_TMO);
717 if (!prb_dispatch_next_block(pkc, po))
718 goto refresh_timer;
719 else
720 goto out;
721 } else {
722 /* Case 1. Queue was frozen because user-space was
723 * lagging behind.
724 */
Rosen, Rami878cd3b2017-05-24 18:34:11 +0300725 if (prb_curr_blk_in_use(pbd)) {
chetan lokef6fb8f12011-08-19 10:18:16 +0000726 /*
727 * Ok, user-space is still behind.
728 * So just refresh the timer.
729 */
730 goto refresh_timer;
731 } else {
732 /* Case 2. queue was frozen,user-space caught up,
733 * now the link went idle && the timer fired.
734 * We don't have a block to close.So we open this
735 * block and restart the timer.
736 * opening a block thaws the queue,restarts timer
737 * Thawing/timer-refresh is a side effect.
738 */
739 prb_open_block(pkc, pbd);
740 goto out;
741 }
742 }
743 }
744
745refresh_timer:
746 _prb_refresh_rx_retire_blk_timer(pkc);
747
748out:
749 spin_unlock(&po->sk.sk_receive_queue.lock);
750}
751
Olof Johanssoneea49cc92011-11-02 11:00:49 +0000752static void prb_flush_block(struct tpacket_kbdq_core *pkc1,
chetan lokebc59ba32011-08-25 10:43:30 +0000753 struct tpacket_block_desc *pbd1, __u32 status)
chetan lokef6fb8f12011-08-19 10:18:16 +0000754{
755 /* Flush everything minus the block header */
756
757#if ARCH_IMPLEMENTS_FLUSH_DCACHE_PAGE == 1
758 u8 *start, *end;
759
760 start = (u8 *)pbd1;
761
762 /* Skip the block header(we know header WILL fit in 4K) */
763 start += PAGE_SIZE;
764
765 end = (u8 *)PAGE_ALIGN((unsigned long)pkc1->pkblk_end);
766 for (; start < end; start += PAGE_SIZE)
767 flush_dcache_page(pgv_to_page(start));
768
769 smp_wmb();
770#endif
771
772 /* Now update the block status. */
773
774 BLOCK_STATUS(pbd1) = status;
775
776 /* Flush the block header */
777
778#if ARCH_IMPLEMENTS_FLUSH_DCACHE_PAGE == 1
779 start = (u8 *)pbd1;
780 flush_dcache_page(pgv_to_page(start));
781
782 smp_wmb();
783#endif
784}
785
786/*
787 * Side effect:
788 *
789 * 1) flush the block
790 * 2) Increment active_blk_num
791 *
792 * Note:We DONT refresh the timer on purpose.
793 * Because almost always the next block will be opened.
794 */
chetan lokebc59ba32011-08-25 10:43:30 +0000795static void prb_close_block(struct tpacket_kbdq_core *pkc1,
796 struct tpacket_block_desc *pbd1,
chetan lokef6fb8f12011-08-19 10:18:16 +0000797 struct packet_sock *po, unsigned int stat)
798{
799 __u32 status = TP_STATUS_USER | stat;
800
801 struct tpacket3_hdr *last_pkt;
chetan lokebc59ba32011-08-25 10:43:30 +0000802 struct tpacket_hdr_v1 *h1 = &pbd1->hdr.bh1;
Dan Collinsda413ee2014-12-19 16:49:25 +1300803 struct sock *sk = &po->sk;
chetan lokef6fb8f12011-08-19 10:18:16 +0000804
Eric Dumazet8e8e2952019-06-12 09:52:30 -0700805 if (atomic_read(&po->tp_drops))
chetan lokef6fb8f12011-08-19 10:18:16 +0000806 status |= TP_STATUS_LOSING;
807
808 last_pkt = (struct tpacket3_hdr *)pkc1->prev;
809 last_pkt->tp_next_offset = 0;
810
811 /* Get the ts of the last pkt */
812 if (BLOCK_NUM_PKTS(pbd1)) {
813 h1->ts_last_pkt.ts_sec = last_pkt->tp_sec;
814 h1->ts_last_pkt.ts_nsec = last_pkt->tp_nsec;
815 } else {
Alexander Drozdov41a50d62015-02-24 08:18:28 +0300816 /* Ok, we tmo'd - so get the current time.
817 *
818 * It shouldn't really happen as we don't close empty
819 * blocks. See prb_retire_rx_blk_timer_expired().
820 */
Arnd Bergmannd413fcb2017-11-27 10:09:24 +0100821 struct timespec64 ts;
822 ktime_get_real_ts64(&ts);
chetan lokef6fb8f12011-08-19 10:18:16 +0000823 h1->ts_last_pkt.ts_sec = ts.tv_sec;
824 h1->ts_last_pkt.ts_nsec = ts.tv_nsec;
825 }
826
827 smp_wmb();
828
829 /* Flush the block */
830 prb_flush_block(pkc1, pbd1, status);
831
Dan Collinsda413ee2014-12-19 16:49:25 +1300832 sk->sk_data_ready(sk);
833
chetan lokef6fb8f12011-08-19 10:18:16 +0000834 pkc1->kactive_blk_num = GET_NEXT_PRB_BLK_NUM(pkc1);
835}
836
Olof Johanssoneea49cc92011-11-02 11:00:49 +0000837static void prb_thaw_queue(struct tpacket_kbdq_core *pkc)
chetan lokef6fb8f12011-08-19 10:18:16 +0000838{
839 pkc->reset_pending_on_curr_blk = 0;
840}
841
842/*
843 * Side effect of opening a block:
844 *
845 * 1) prb_queue is thawed.
846 * 2) retire_blk_timer is refreshed.
847 *
848 */
chetan lokebc59ba32011-08-25 10:43:30 +0000849static void prb_open_block(struct tpacket_kbdq_core *pkc1,
850 struct tpacket_block_desc *pbd1)
chetan lokef6fb8f12011-08-19 10:18:16 +0000851{
Arnd Bergmannd413fcb2017-11-27 10:09:24 +0100852 struct timespec64 ts;
chetan lokebc59ba32011-08-25 10:43:30 +0000853 struct tpacket_hdr_v1 *h1 = &pbd1->hdr.bh1;
chetan lokef6fb8f12011-08-19 10:18:16 +0000854
855 smp_rmb();
856
Daniel Borkmann8da30562013-05-03 02:57:00 +0000857 /* We could have just memset this but we will lose the
858 * flexibility of making the priv area sticky
859 */
chetan lokef6fb8f12011-08-19 10:18:16 +0000860
Daniel Borkmann8da30562013-05-03 02:57:00 +0000861 BLOCK_SNUM(pbd1) = pkc1->knxt_seq_num++;
862 BLOCK_NUM_PKTS(pbd1) = 0;
863 BLOCK_LEN(pbd1) = BLK_PLUS_PRIV(pkc1->blk_sizeof_priv);
chetan lokef6fb8f12011-08-19 10:18:16 +0000864
Arnd Bergmannd413fcb2017-11-27 10:09:24 +0100865 ktime_get_real_ts64(&ts);
chetan lokef6fb8f12011-08-19 10:18:16 +0000866
Daniel Borkmann8da30562013-05-03 02:57:00 +0000867 h1->ts_first_pkt.ts_sec = ts.tv_sec;
868 h1->ts_first_pkt.ts_nsec = ts.tv_nsec;
chetan lokef6fb8f12011-08-19 10:18:16 +0000869
Daniel Borkmann8da30562013-05-03 02:57:00 +0000870 pkc1->pkblk_start = (char *)pbd1;
871 pkc1->nxt_offset = pkc1->pkblk_start + BLK_PLUS_PRIV(pkc1->blk_sizeof_priv);
872
873 BLOCK_O2FP(pbd1) = (__u32)BLK_PLUS_PRIV(pkc1->blk_sizeof_priv);
874 BLOCK_O2PRIV(pbd1) = BLK_HDR_LEN;
875
876 pbd1->version = pkc1->version;
877 pkc1->prev = pkc1->nxt_offset;
878 pkc1->pkblk_end = pkc1->pkblk_start + pkc1->kblk_size;
879
880 prb_thaw_queue(pkc1);
881 _prb_refresh_rx_retire_blk_timer(pkc1);
882
883 smp_wmb();
chetan lokef6fb8f12011-08-19 10:18:16 +0000884}
885
886/*
887 * Queue freeze logic:
888 * 1) Assume tp_block_nr = 8 blocks.
889 * 2) At time 't0', user opens Rx ring.
890 * 3) Some time past 't0', kernel starts filling blocks starting from 0 .. 7
891 * 4) user-space is either sleeping or processing block '0'.
892 * 5) tpacket_rcv is currently filling block '7', since there is no space left,
893 * it will close block-7,loop around and try to fill block '0'.
894 * call-flow:
895 * __packet_lookup_frame_in_block
896 * prb_retire_current_block()
897 * prb_dispatch_next_block()
898 * |->(BLOCK_STATUS == USER) evaluates to true
899 * 5.1) Since block-0 is currently in-use, we just freeze the queue.
900 * 6) Now there are two cases:
901 * 6.1) Link goes idle right after the queue is frozen.
902 * But remember, the last open_block() refreshed the timer.
903 * When this timer expires,it will refresh itself so that we can
904 * re-open block-0 in near future.
905 * 6.2) Link is busy and keeps on receiving packets. This is a simple
906 * case and __packet_lookup_frame_in_block will check if block-0
907 * is free and can now be re-used.
908 */
Olof Johanssoneea49cc92011-11-02 11:00:49 +0000909static void prb_freeze_queue(struct tpacket_kbdq_core *pkc,
chetan lokef6fb8f12011-08-19 10:18:16 +0000910 struct packet_sock *po)
911{
912 pkc->reset_pending_on_curr_blk = 1;
Daniel Borkmannee80fbf2013-04-19 06:12:29 +0000913 po->stats.stats3.tp_freeze_q_cnt++;
chetan lokef6fb8f12011-08-19 10:18:16 +0000914}
915
916#define TOTAL_PKT_LEN_INCL_ALIGN(length) (ALIGN((length), V3_ALIGNMENT))
917
918/*
919 * If the next block is free then we will dispatch it
920 * and return a good offset.
921 * Else, we will freeze the queue.
922 * So, caller must check the return value.
923 */
chetan lokebc59ba32011-08-25 10:43:30 +0000924static void *prb_dispatch_next_block(struct tpacket_kbdq_core *pkc,
chetan lokef6fb8f12011-08-19 10:18:16 +0000925 struct packet_sock *po)
926{
chetan lokebc59ba32011-08-25 10:43:30 +0000927 struct tpacket_block_desc *pbd;
chetan lokef6fb8f12011-08-19 10:18:16 +0000928
929 smp_rmb();
930
931 /* 1. Get current block num */
932 pbd = GET_CURR_PBLOCK_DESC_FROM_CORE(pkc);
933
934 /* 2. If this block is currently in_use then freeze the queue */
935 if (TP_STATUS_USER & BLOCK_STATUS(pbd)) {
936 prb_freeze_queue(pkc, po);
937 return NULL;
938 }
939
940 /*
941 * 3.
942 * open this block and return the offset where the first packet
943 * needs to get stored.
944 */
945 prb_open_block(pkc, pbd);
946 return (void *)pkc->nxt_offset;
947}
948
chetan lokebc59ba32011-08-25 10:43:30 +0000949static void prb_retire_current_block(struct tpacket_kbdq_core *pkc,
chetan lokef6fb8f12011-08-19 10:18:16 +0000950 struct packet_sock *po, unsigned int status)
951{
chetan lokebc59ba32011-08-25 10:43:30 +0000952 struct tpacket_block_desc *pbd = GET_CURR_PBLOCK_DESC_FROM_CORE(pkc);
chetan lokef6fb8f12011-08-19 10:18:16 +0000953
954 /* retire/close the current block */
955 if (likely(TP_STATUS_KERNEL == BLOCK_STATUS(pbd))) {
956 /*
957 * Plug the case where copy_bits() is in progress on
958 * cpu-0 and tpacket_rcv() got invoked on cpu-1, didn't
959 * have space to copy the pkt in the current block and
960 * called prb_retire_current_block()
961 *
962 * We don't need to worry about the TMO case because
963 * the timer-handler already handled this case.
964 */
965 if (!(status & TP_STATUS_BLK_TMO)) {
John Ogness632ca502020-07-07 17:28:04 +0206966 /* Waiting for skb_copy_bits to finish... */
967 write_lock(&pkc->blk_fill_in_prog_lock);
968 write_unlock(&pkc->blk_fill_in_prog_lock);
chetan lokef6fb8f12011-08-19 10:18:16 +0000969 }
970 prb_close_block(pkc, pbd, po, status);
971 return;
972 }
chetan lokef6fb8f12011-08-19 10:18:16 +0000973}
974
Rosen, Rami878cd3b2017-05-24 18:34:11 +0300975static int prb_curr_blk_in_use(struct tpacket_block_desc *pbd)
chetan lokef6fb8f12011-08-19 10:18:16 +0000976{
977 return TP_STATUS_USER & BLOCK_STATUS(pbd);
978}
979
Olof Johanssoneea49cc92011-11-02 11:00:49 +0000980static int prb_queue_frozen(struct tpacket_kbdq_core *pkc)
chetan lokef6fb8f12011-08-19 10:18:16 +0000981{
982 return pkc->reset_pending_on_curr_blk;
983}
984
Olof Johanssoneea49cc92011-11-02 11:00:49 +0000985static void prb_clear_blk_fill_status(struct packet_ring_buffer *rb)
John Ogness88fd1cb2020-08-13 21:45:25 +0206986 __releases(&pkc->blk_fill_in_prog_lock)
chetan lokef6fb8f12011-08-19 10:18:16 +0000987{
chetan lokebc59ba32011-08-25 10:43:30 +0000988 struct tpacket_kbdq_core *pkc = GET_PBDQC_FROM_RB(rb);
John Ogness632ca502020-07-07 17:28:04 +0206989
990 read_unlock(&pkc->blk_fill_in_prog_lock);
chetan lokef6fb8f12011-08-19 10:18:16 +0000991}
992
Olof Johanssoneea49cc92011-11-02 11:00:49 +0000993static void prb_fill_rxhash(struct tpacket_kbdq_core *pkc,
chetan lokef6fb8f12011-08-19 10:18:16 +0000994 struct tpacket3_hdr *ppd)
995{
Tom Herbert3958afa1b2013-12-15 22:12:06 -0800996 ppd->hv1.tp_rxhash = skb_get_hash(pkc->skb);
chetan lokef6fb8f12011-08-19 10:18:16 +0000997}
998
Olof Johanssoneea49cc92011-11-02 11:00:49 +0000999static void prb_clear_rxhash(struct tpacket_kbdq_core *pkc,
chetan lokef6fb8f12011-08-19 10:18:16 +00001000 struct tpacket3_hdr *ppd)
1001{
1002 ppd->hv1.tp_rxhash = 0;
1003}
1004
Olof Johanssoneea49cc92011-11-02 11:00:49 +00001005static void prb_fill_vlan_info(struct tpacket_kbdq_core *pkc,
chetan lokef6fb8f12011-08-19 10:18:16 +00001006 struct tpacket3_hdr *ppd)
1007{
Jiri Pirkodf8a39d2015-01-13 17:13:44 +01001008 if (skb_vlan_tag_present(pkc->skb)) {
1009 ppd->hv1.tp_vlan_tci = skb_vlan_tag_get(pkc->skb);
Atzm Watanabea0cdfcf2013-12-17 22:53:40 +09001010 ppd->hv1.tp_vlan_tpid = ntohs(pkc->skb->vlan_proto);
1011 ppd->tp_status = TP_STATUS_VLAN_VALID | TP_STATUS_VLAN_TPID_VALID;
chetan lokef6fb8f12011-08-19 10:18:16 +00001012 } else {
danborkmann@iogearbox.net9e670302012-08-20 03:34:03 +00001013 ppd->hv1.tp_vlan_tci = 0;
Atzm Watanabea0cdfcf2013-12-17 22:53:40 +09001014 ppd->hv1.tp_vlan_tpid = 0;
danborkmann@iogearbox.net9e670302012-08-20 03:34:03 +00001015 ppd->tp_status = TP_STATUS_AVAILABLE;
chetan lokef6fb8f12011-08-19 10:18:16 +00001016 }
1017}
1018
chetan lokebc59ba32011-08-25 10:43:30 +00001019static void prb_run_all_ft_ops(struct tpacket_kbdq_core *pkc,
chetan lokef6fb8f12011-08-19 10:18:16 +00001020 struct tpacket3_hdr *ppd)
1021{
Atzm Watanabea0cdfcf2013-12-17 22:53:40 +09001022 ppd->hv1.tp_padding = 0;
chetan lokef6fb8f12011-08-19 10:18:16 +00001023 prb_fill_vlan_info(pkc, ppd);
1024
1025 if (pkc->feature_req_word & TP_FT_REQ_FILL_RXHASH)
1026 prb_fill_rxhash(pkc, ppd);
1027 else
1028 prb_clear_rxhash(pkc, ppd);
1029}
1030
Olof Johanssoneea49cc92011-11-02 11:00:49 +00001031static void prb_fill_curr_block(char *curr,
chetan lokebc59ba32011-08-25 10:43:30 +00001032 struct tpacket_kbdq_core *pkc,
1033 struct tpacket_block_desc *pbd,
chetan lokef6fb8f12011-08-19 10:18:16 +00001034 unsigned int len)
John Ogness88fd1cb2020-08-13 21:45:25 +02061035 __acquires(&pkc->blk_fill_in_prog_lock)
chetan lokef6fb8f12011-08-19 10:18:16 +00001036{
1037 struct tpacket3_hdr *ppd;
1038
1039 ppd = (struct tpacket3_hdr *)curr;
1040 ppd->tp_next_offset = TOTAL_PKT_LEN_INCL_ALIGN(len);
1041 pkc->prev = curr;
1042 pkc->nxt_offset += TOTAL_PKT_LEN_INCL_ALIGN(len);
1043 BLOCK_LEN(pbd) += TOTAL_PKT_LEN_INCL_ALIGN(len);
1044 BLOCK_NUM_PKTS(pbd) += 1;
John Ogness632ca502020-07-07 17:28:04 +02061045 read_lock(&pkc->blk_fill_in_prog_lock);
chetan lokef6fb8f12011-08-19 10:18:16 +00001046 prb_run_all_ft_ops(pkc, ppd);
1047}
1048
1049/* Assumes caller has the sk->rx_queue.lock */
1050static void *__packet_lookup_frame_in_block(struct packet_sock *po,
1051 struct sk_buff *skb,
chetan lokef6fb8f12011-08-19 10:18:16 +00001052 unsigned int len
1053 )
1054{
chetan lokebc59ba32011-08-25 10:43:30 +00001055 struct tpacket_kbdq_core *pkc;
1056 struct tpacket_block_desc *pbd;
chetan lokef6fb8f12011-08-19 10:18:16 +00001057 char *curr, *end;
1058
Joe Perchese3192692012-06-03 17:41:40 +00001059 pkc = GET_PBDQC_FROM_RB(&po->rx_ring);
chetan lokef6fb8f12011-08-19 10:18:16 +00001060 pbd = GET_CURR_PBLOCK_DESC_FROM_CORE(pkc);
1061
1062 /* Queue is frozen when user space is lagging behind */
1063 if (prb_queue_frozen(pkc)) {
1064 /*
1065 * Check if that last block which caused the queue to freeze,
1066 * is still in_use by user-space.
1067 */
Rosen, Rami878cd3b2017-05-24 18:34:11 +03001068 if (prb_curr_blk_in_use(pbd)) {
chetan lokef6fb8f12011-08-19 10:18:16 +00001069 /* Can't record this packet */
1070 return NULL;
1071 } else {
1072 /*
1073 * Ok, the block was released by user-space.
1074 * Now let's open that block.
1075 * opening a block also thaws the queue.
1076 * Thawing is a side effect.
1077 */
1078 prb_open_block(pkc, pbd);
1079 }
1080 }
1081
1082 smp_mb();
1083 curr = pkc->nxt_offset;
1084 pkc->skb = skb;
Joe Perchese3192692012-06-03 17:41:40 +00001085 end = (char *)pbd + pkc->kblk_size;
chetan lokef6fb8f12011-08-19 10:18:16 +00001086
1087 /* first try the current block */
1088 if (curr+TOTAL_PKT_LEN_INCL_ALIGN(len) < end) {
1089 prb_fill_curr_block(curr, pkc, pbd, len);
1090 return (void *)curr;
1091 }
1092
1093 /* Ok, close the current block */
1094 prb_retire_current_block(pkc, po, 0);
1095
1096 /* Now, try to dispatch the next block */
1097 curr = (char *)prb_dispatch_next_block(pkc, po);
1098 if (curr) {
1099 pbd = GET_CURR_PBLOCK_DESC_FROM_CORE(pkc);
1100 prb_fill_curr_block(curr, pkc, pbd, len);
1101 return (void *)curr;
1102 }
1103
1104 /*
1105 * No free blocks are available.user_space hasn't caught up yet.
1106 * Queue was just frozen and now this packet will get dropped.
1107 */
1108 return NULL;
1109}
1110
Olof Johanssoneea49cc92011-11-02 11:00:49 +00001111static void *packet_current_rx_frame(struct packet_sock *po,
chetan lokef6fb8f12011-08-19 10:18:16 +00001112 struct sk_buff *skb,
1113 int status, unsigned int len)
1114{
1115 char *curr = NULL;
1116 switch (po->tp_version) {
1117 case TPACKET_V1:
1118 case TPACKET_V2:
1119 curr = packet_lookup_frame(po, &po->rx_ring,
1120 po->rx_ring.head, status);
1121 return curr;
1122 case TPACKET_V3:
Mao Wenan46088052019-06-11 09:32:13 +08001123 return __packet_lookup_frame_in_block(po, skb, len);
chetan lokef6fb8f12011-08-19 10:18:16 +00001124 default:
1125 WARN(1, "TPACKET version not supported\n");
1126 BUG();
Ying Xue99aa3472012-08-06 16:27:10 +00001127 return NULL;
chetan lokef6fb8f12011-08-19 10:18:16 +00001128 }
1129}
1130
Eric Dumazetdcf70ce2019-06-12 09:52:28 -07001131static void *prb_lookup_block(const struct packet_sock *po,
1132 const struct packet_ring_buffer *rb,
1133 unsigned int idx,
1134 int status)
chetan lokef6fb8f12011-08-19 10:18:16 +00001135{
chetan lokebc59ba32011-08-25 10:43:30 +00001136 struct tpacket_kbdq_core *pkc = GET_PBDQC_FROM_RB(rb);
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001137 struct tpacket_block_desc *pbd = GET_PBLOCK_DESC(pkc, idx);
chetan lokef6fb8f12011-08-19 10:18:16 +00001138
1139 if (status != BLOCK_STATUS(pbd))
1140 return NULL;
1141 return pbd;
1142}
1143
Olof Johanssoneea49cc92011-11-02 11:00:49 +00001144static int prb_previous_blk_num(struct packet_ring_buffer *rb)
chetan lokef6fb8f12011-08-19 10:18:16 +00001145{
1146 unsigned int prev;
1147 if (rb->prb_bdqc.kactive_blk_num)
1148 prev = rb->prb_bdqc.kactive_blk_num-1;
1149 else
1150 prev = rb->prb_bdqc.knum_blocks-1;
1151 return prev;
1152}
1153
1154/* Assumes caller has held the rx_queue.lock */
Olof Johanssoneea49cc92011-11-02 11:00:49 +00001155static void *__prb_previous_block(struct packet_sock *po,
chetan lokef6fb8f12011-08-19 10:18:16 +00001156 struct packet_ring_buffer *rb,
1157 int status)
1158{
1159 unsigned int previous = prb_previous_blk_num(rb);
1160 return prb_lookup_block(po, rb, previous, status);
1161}
1162
Olof Johanssoneea49cc92011-11-02 11:00:49 +00001163static void *packet_previous_rx_frame(struct packet_sock *po,
chetan lokef6fb8f12011-08-19 10:18:16 +00001164 struct packet_ring_buffer *rb,
1165 int status)
1166{
1167 if (po->tp_version <= TPACKET_V2)
1168 return packet_previous_frame(po, rb, status);
1169
1170 return __prb_previous_block(po, rb, status);
1171}
1172
Olof Johanssoneea49cc92011-11-02 11:00:49 +00001173static void packet_increment_rx_head(struct packet_sock *po,
chetan lokef6fb8f12011-08-19 10:18:16 +00001174 struct packet_ring_buffer *rb)
1175{
1176 switch (po->tp_version) {
1177 case TPACKET_V1:
1178 case TPACKET_V2:
1179 return packet_increment_head(rb);
1180 case TPACKET_V3:
1181 default:
1182 WARN(1, "TPACKET version not supported.\n");
1183 BUG();
1184 return;
1185 }
1186}
1187
Olof Johanssoneea49cc92011-11-02 11:00:49 +00001188static void *packet_previous_frame(struct packet_sock *po,
Johann Baudy69e3c752009-05-18 22:11:22 -07001189 struct packet_ring_buffer *rb,
1190 int status)
1191{
1192 unsigned int previous = rb->head ? rb->head - 1 : rb->frame_max;
1193 return packet_lookup_frame(po, rb, previous, status);
1194}
1195
Olof Johanssoneea49cc92011-11-02 11:00:49 +00001196static void packet_increment_head(struct packet_ring_buffer *buff)
Johann Baudy69e3c752009-05-18 22:11:22 -07001197{
1198 buff->head = buff->head != buff->frame_max ? buff->head+1 : 0;
1199}
1200
Daniel Borkmannb0138402014-01-15 16:25:36 +01001201static void packet_inc_pending(struct packet_ring_buffer *rb)
1202{
1203 this_cpu_inc(*rb->pending_refcnt);
1204}
1205
1206static void packet_dec_pending(struct packet_ring_buffer *rb)
1207{
1208 this_cpu_dec(*rb->pending_refcnt);
1209}
1210
1211static unsigned int packet_read_pending(const struct packet_ring_buffer *rb)
1212{
1213 unsigned int refcnt = 0;
1214 int cpu;
1215
1216 /* We don't use pending refcount in rx_ring. */
1217 if (rb->pending_refcnt == NULL)
1218 return 0;
1219
1220 for_each_possible_cpu(cpu)
1221 refcnt += *per_cpu_ptr(rb->pending_refcnt, cpu);
1222
1223 return refcnt;
1224}
1225
1226static int packet_alloc_pending(struct packet_sock *po)
1227{
1228 po->rx_ring.pending_refcnt = NULL;
1229
1230 po->tx_ring.pending_refcnt = alloc_percpu(unsigned int);
1231 if (unlikely(po->tx_ring.pending_refcnt == NULL))
1232 return -ENOBUFS;
1233
1234 return 0;
1235}
1236
1237static void packet_free_pending(struct packet_sock *po)
1238{
1239 free_percpu(po->tx_ring.pending_refcnt);
1240}
1241
Willem de Bruijn99547292015-05-12 11:56:47 -04001242#define ROOM_POW_OFF 2
1243#define ROOM_NONE 0x0
1244#define ROOM_LOW 0x1
1245#define ROOM_NORMAL 0x2
1246
Eric Dumazetd4b5bd92019-06-12 09:52:27 -07001247static bool __tpacket_has_room(const struct packet_sock *po, int pow_off)
Willem de Bruijn99547292015-05-12 11:56:47 -04001248{
1249 int idx, len;
1250
Eric Dumazetd4b5bd92019-06-12 09:52:27 -07001251 len = READ_ONCE(po->rx_ring.frame_max) + 1;
1252 idx = READ_ONCE(po->rx_ring.head);
Willem de Bruijn99547292015-05-12 11:56:47 -04001253 if (pow_off)
1254 idx += len >> pow_off;
1255 if (idx >= len)
1256 idx -= len;
1257 return packet_lookup_frame(po, &po->rx_ring, idx, TP_STATUS_KERNEL);
1258}
1259
Eric Dumazetdcf70ce2019-06-12 09:52:28 -07001260static bool __tpacket_v3_has_room(const struct packet_sock *po, int pow_off)
Willem de Bruijn99547292015-05-12 11:56:47 -04001261{
1262 int idx, len;
1263
Eric Dumazetdcf70ce2019-06-12 09:52:28 -07001264 len = READ_ONCE(po->rx_ring.prb_bdqc.knum_blocks);
1265 idx = READ_ONCE(po->rx_ring.prb_bdqc.kactive_blk_num);
Willem de Bruijn99547292015-05-12 11:56:47 -04001266 if (pow_off)
1267 idx += len >> pow_off;
1268 if (idx >= len)
1269 idx -= len;
1270 return prb_lookup_block(po, &po->rx_ring, idx, TP_STATUS_KERNEL);
1271}
1272
Eric Dumazet0338a142019-06-12 09:52:29 -07001273static int __packet_rcv_has_room(const struct packet_sock *po,
1274 const struct sk_buff *skb)
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001275{
Eric Dumazet0338a142019-06-12 09:52:29 -07001276 const struct sock *sk = &po->sk;
Willem de Bruijn99547292015-05-12 11:56:47 -04001277 int ret = ROOM_NONE;
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001278
Willem de Bruijn99547292015-05-12 11:56:47 -04001279 if (po->prot_hook.func != tpacket_rcv) {
Eric Dumazet0338a142019-06-12 09:52:29 -07001280 int rcvbuf = READ_ONCE(sk->sk_rcvbuf);
1281 int avail = rcvbuf - atomic_read(&sk->sk_rmem_alloc)
1282 - (skb ? skb->truesize : 0);
1283
1284 if (avail > (rcvbuf >> ROOM_POW_OFF))
Willem de Bruijn99547292015-05-12 11:56:47 -04001285 return ROOM_NORMAL;
1286 else if (avail > 0)
1287 return ROOM_LOW;
1288 else
1289 return ROOM_NONE;
1290 }
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001291
Willem de Bruijn99547292015-05-12 11:56:47 -04001292 if (po->tp_version == TPACKET_V3) {
1293 if (__tpacket_v3_has_room(po, ROOM_POW_OFF))
1294 ret = ROOM_NORMAL;
1295 else if (__tpacket_v3_has_room(po, 0))
1296 ret = ROOM_LOW;
1297 } else {
1298 if (__tpacket_has_room(po, ROOM_POW_OFF))
1299 ret = ROOM_NORMAL;
1300 else if (__tpacket_has_room(po, 0))
1301 ret = ROOM_LOW;
1302 }
Willem de Bruijn2ccdbaa2015-05-12 11:56:48 -04001303
1304 return ret;
1305}
1306
1307static int packet_rcv_has_room(struct packet_sock *po, struct sk_buff *skb)
1308{
Eric Dumazet3a2bb842019-06-12 09:52:32 -07001309 int pressure, ret;
Willem de Bruijn2ccdbaa2015-05-12 11:56:48 -04001310
Willem de Bruijn54d7c012015-05-14 15:25:02 -04001311 ret = __packet_rcv_has_room(po, skb);
Eric Dumazet3a2bb842019-06-12 09:52:32 -07001312 pressure = ret != ROOM_NORMAL;
1313
1314 if (READ_ONCE(po->pressure) != pressure)
1315 WRITE_ONCE(po->pressure, pressure);
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001316
Willem de Bruijn99547292015-05-12 11:56:47 -04001317 return ret;
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001318}
1319
Eric Dumazet9bb6cd62019-06-12 09:52:33 -07001320static void packet_rcv_try_clear_pressure(struct packet_sock *po)
1321{
1322 if (READ_ONCE(po->pressure) &&
1323 __packet_rcv_has_room(po, NULL) == ROOM_NORMAL)
1324 WRITE_ONCE(po->pressure, 0);
1325}
1326
Linus Torvalds1da177e2005-04-16 15:20:36 -07001327static void packet_sock_destruct(struct sock *sk)
1328{
Richard Cochraned85b562010-04-07 22:41:28 +00001329 skb_queue_purge(&sk->sk_error_queue);
1330
Ilpo Järvinen547b7922008-07-25 21:43:18 -07001331 WARN_ON(atomic_read(&sk->sk_rmem_alloc));
Reshetova, Elena14afee42017-06-30 13:08:00 +03001332 WARN_ON(refcount_read(&sk->sk_wmem_alloc));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001333
1334 if (!sock_flag(sk, SOCK_DEAD)) {
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00001335 pr_err("Attempt to release alive packet socket: %p\n", sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001336 return;
1337 }
1338
Pavel Emelyanov17ab56a2007-11-10 21:38:48 -08001339 sk_refcnt_debug_dec(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001340}
1341
Willem de Bruijn3b3a5b02015-05-12 11:56:49 -04001342static bool fanout_flow_is_huge(struct packet_sock *po, struct sk_buff *skb)
1343{
Eric Dumazetb756ad92019-11-08 05:07:46 -08001344 u32 *history = po->rollover->history;
1345 u32 victim, rxhash;
Willem de Bruijn3b3a5b02015-05-12 11:56:49 -04001346 int i, count = 0;
1347
1348 rxhash = skb_get_hash(skb);
1349 for (i = 0; i < ROLLOVER_HLEN; i++)
Eric Dumazetb756ad92019-11-08 05:07:46 -08001350 if (READ_ONCE(history[i]) == rxhash)
Willem de Bruijn3b3a5b02015-05-12 11:56:49 -04001351 count++;
1352
Eric Dumazetb756ad92019-11-08 05:07:46 -08001353 victim = prandom_u32() % ROLLOVER_HLEN;
1354
1355 /* Avoid dirtying the cache line if possible */
1356 if (READ_ONCE(history[victim]) != rxhash)
1357 WRITE_ONCE(history[victim], rxhash);
1358
Willem de Bruijn3b3a5b02015-05-12 11:56:49 -04001359 return count > (ROLLOVER_HLEN >> 1);
1360}
1361
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001362static unsigned int fanout_demux_hash(struct packet_fanout *f,
1363 struct sk_buff *skb,
1364 unsigned int num)
David S. Millerdc99f602011-07-05 01:45:05 -07001365{
David S. Millereb70db82016-07-01 16:07:50 -04001366 return reciprocal_scale(__skb_get_hash_symmetric(skb), num);
David S. Millerdc99f602011-07-05 01:45:05 -07001367}
1368
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001369static unsigned int fanout_demux_lb(struct packet_fanout *f,
1370 struct sk_buff *skb,
1371 unsigned int num)
David S. Millerdc99f602011-07-05 01:45:05 -07001372{
Willem de Bruijn468479e2015-06-17 15:59:34 -04001373 unsigned int val = atomic_inc_return(&f->rr_cur);
David S. Millerdc99f602011-07-05 01:45:05 -07001374
Willem de Bruijn468479e2015-06-17 15:59:34 -04001375 return val % num;
David S. Millerdc99f602011-07-05 01:45:05 -07001376}
1377
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001378static unsigned int fanout_demux_cpu(struct packet_fanout *f,
1379 struct sk_buff *skb,
1380 unsigned int num)
David S. Miller95ec3eb2011-07-06 01:56:38 -07001381{
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001382 return smp_processor_id() % num;
1383}
David S. Miller95ec3eb2011-07-06 01:56:38 -07001384
Daniel Borkmann5df0ddf2013-08-28 22:13:09 +02001385static unsigned int fanout_demux_rnd(struct packet_fanout *f,
1386 struct sk_buff *skb,
1387 unsigned int num)
1388{
Daniel Borkmannf337db62014-01-22 02:29:39 +01001389 return prandom_u32_max(num);
Daniel Borkmann5df0ddf2013-08-28 22:13:09 +02001390}
1391
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001392static unsigned int fanout_demux_rollover(struct packet_fanout *f,
1393 struct sk_buff *skb,
Willem de Bruijnad377ca2015-05-12 11:56:45 -04001394 unsigned int idx, bool try_self,
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001395 unsigned int num)
1396{
Willem de Bruijn4633c9e2015-05-17 19:44:02 -04001397 struct packet_sock *po, *po_next, *po_skip = NULL;
Willem de Bruijna9b63912015-05-12 11:56:50 -04001398 unsigned int i, j, room = ROOM_NONE;
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001399
Eric Dumazet94f633e2021-04-14 12:36:44 -07001400 po = pkt_sk(rcu_dereference(f->arr[idx]));
Willem de Bruijn3b3a5b02015-05-12 11:56:49 -04001401
1402 if (try_self) {
1403 room = packet_rcv_has_room(po, skb);
1404 if (room == ROOM_NORMAL ||
1405 (room == ROOM_LOW && !fanout_flow_is_huge(po, skb)))
1406 return idx;
Willem de Bruijn4633c9e2015-05-17 19:44:02 -04001407 po_skip = po;
Willem de Bruijn3b3a5b02015-05-12 11:56:49 -04001408 }
Willem de Bruijnad377ca2015-05-12 11:56:45 -04001409
Willem de Bruijn0648ab72015-05-12 11:56:46 -04001410 i = j = min_t(int, po->rollover->sock, num - 1);
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001411 do {
Eric Dumazet94f633e2021-04-14 12:36:44 -07001412 po_next = pkt_sk(rcu_dereference(f->arr[i]));
Eric Dumazet3a2bb842019-06-12 09:52:32 -07001413 if (po_next != po_skip && !READ_ONCE(po_next->pressure) &&
Willem de Bruijn2ccdbaa2015-05-12 11:56:48 -04001414 packet_rcv_has_room(po_next, skb) == ROOM_NORMAL) {
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001415 if (i != j)
Willem de Bruijn0648ab72015-05-12 11:56:46 -04001416 po->rollover->sock = i;
Willem de Bruijna9b63912015-05-12 11:56:50 -04001417 atomic_long_inc(&po->rollover->num);
1418 if (room == ROOM_LOW)
1419 atomic_long_inc(&po->rollover->num_huge);
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001420 return i;
1421 }
Willem de Bruijnad377ca2015-05-12 11:56:45 -04001422
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001423 if (++i == num)
1424 i = 0;
1425 } while (i != j);
1426
Willem de Bruijna9b63912015-05-12 11:56:50 -04001427 atomic_long_inc(&po->rollover->num_failed);
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001428 return idx;
1429}
1430
Neil Horman2d360972014-01-22 16:01:44 -05001431static unsigned int fanout_demux_qm(struct packet_fanout *f,
1432 struct sk_buff *skb,
1433 unsigned int num)
1434{
1435 return skb_get_queue_mapping(skb) % num;
1436}
1437
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001438static unsigned int fanout_demux_bpf(struct packet_fanout *f,
1439 struct sk_buff *skb,
1440 unsigned int num)
1441{
1442 struct bpf_prog *prog;
1443 unsigned int ret = 0;
1444
1445 rcu_read_lock();
1446 prog = rcu_dereference(f->bpf_prog);
1447 if (prog)
Alexei Starovoitovff936a02015-10-07 10:55:41 -07001448 ret = bpf_prog_run_clear_cb(prog, skb) % num;
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001449 rcu_read_unlock();
1450
1451 return ret;
1452}
1453
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001454static bool fanout_has_flag(struct packet_fanout *f, u16 flag)
1455{
1456 return f->flags & (flag >> 8);
David S. Miller95ec3eb2011-07-06 01:56:38 -07001457}
1458
David S. Miller95ec3eb2011-07-06 01:56:38 -07001459static int packet_rcv_fanout(struct sk_buff *skb, struct net_device *dev,
1460 struct packet_type *pt, struct net_device *orig_dev)
David S. Millerdc99f602011-07-05 01:45:05 -07001461{
1462 struct packet_fanout *f = pt->af_packet_priv;
Eric Dumazetf98f4512015-06-16 07:59:11 -07001463 unsigned int num = READ_ONCE(f->num_members);
Eric W. Biederman19bcf9f2015-10-09 13:44:54 -05001464 struct net *net = read_pnet(&f->net);
David S. Millerdc99f602011-07-05 01:45:05 -07001465 struct packet_sock *po;
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001466 unsigned int idx;
David S. Millerdc99f602011-07-05 01:45:05 -07001467
Eric W. Biederman19bcf9f2015-10-09 13:44:54 -05001468 if (!net_eq(dev_net(dev), net) || !num) {
David S. Millerdc99f602011-07-05 01:45:05 -07001469 kfree_skb(skb);
1470 return 0;
1471 }
1472
Alexander Drozdov3f34b242015-02-20 08:24:27 +03001473 if (fanout_has_flag(f, PACKET_FANOUT_FLAG_DEFRAG)) {
Eric W. Biederman19bcf9f2015-10-09 13:44:54 -05001474 skb = ip_check_defrag(net, skb, IP_DEFRAG_AF_PACKET);
Alexander Drozdov3f34b242015-02-20 08:24:27 +03001475 if (!skb)
1476 return 0;
1477 }
David S. Miller95ec3eb2011-07-06 01:56:38 -07001478 switch (f->type) {
1479 case PACKET_FANOUT_HASH:
1480 default:
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001481 idx = fanout_demux_hash(f, skb, num);
David S. Miller95ec3eb2011-07-06 01:56:38 -07001482 break;
1483 case PACKET_FANOUT_LB:
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001484 idx = fanout_demux_lb(f, skb, num);
David S. Miller95ec3eb2011-07-06 01:56:38 -07001485 break;
1486 case PACKET_FANOUT_CPU:
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001487 idx = fanout_demux_cpu(f, skb, num);
1488 break;
Daniel Borkmann5df0ddf2013-08-28 22:13:09 +02001489 case PACKET_FANOUT_RND:
1490 idx = fanout_demux_rnd(f, skb, num);
1491 break;
Neil Horman2d360972014-01-22 16:01:44 -05001492 case PACKET_FANOUT_QM:
1493 idx = fanout_demux_qm(f, skb, num);
1494 break;
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001495 case PACKET_FANOUT_ROLLOVER:
Willem de Bruijnad377ca2015-05-12 11:56:45 -04001496 idx = fanout_demux_rollover(f, skb, 0, false, num);
David S. Miller95ec3eb2011-07-06 01:56:38 -07001497 break;
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001498 case PACKET_FANOUT_CBPF:
Willem de Bruijnf2e52092015-08-14 22:31:35 -04001499 case PACKET_FANOUT_EBPF:
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001500 idx = fanout_demux_bpf(f, skb, num);
1501 break;
David S. Miller7736d332011-07-05 01:43:20 -07001502 }
1503
Willem de Bruijnad377ca2015-05-12 11:56:45 -04001504 if (fanout_has_flag(f, PACKET_FANOUT_FLAG_ROLLOVER))
1505 idx = fanout_demux_rollover(f, skb, idx, true, num);
David S. Millerdc99f602011-07-05 01:45:05 -07001506
Eric Dumazet94f633e2021-04-14 12:36:44 -07001507 po = pkt_sk(rcu_dereference(f->arr[idx]));
David S. Millerdc99f602011-07-05 01:45:05 -07001508 return po->prot_hook.func(skb, dev, &po->prot_hook, orig_dev);
1509}
1510
Pavel Emelyanovfff33212012-08-16 05:36:48 +00001511DEFINE_MUTEX(fanout_mutex);
1512EXPORT_SYMBOL_GPL(fanout_mutex);
David S. Millerdc99f602011-07-05 01:45:05 -07001513static LIST_HEAD(fanout_list);
Mike Maloney4a69a862017-04-21 10:56:11 -04001514static u16 fanout_next_id;
David S. Millerdc99f602011-07-05 01:45:05 -07001515
1516static void __fanout_link(struct sock *sk, struct packet_sock *po)
1517{
1518 struct packet_fanout *f = po->fanout;
1519
1520 spin_lock(&f->lock);
Eric Dumazet94f633e2021-04-14 12:36:44 -07001521 rcu_assign_pointer(f->arr[f->num_members], sk);
David S. Millerdc99f602011-07-05 01:45:05 -07001522 smp_wmb();
1523 f->num_members++;
Anoob Soman2bd624b2017-02-15 20:25:39 +00001524 if (f->num_members == 1)
1525 dev_add_pack(&f->prot_hook);
David S. Millerdc99f602011-07-05 01:45:05 -07001526 spin_unlock(&f->lock);
1527}
1528
1529static void __fanout_unlink(struct sock *sk, struct packet_sock *po)
1530{
1531 struct packet_fanout *f = po->fanout;
1532 int i;
1533
1534 spin_lock(&f->lock);
1535 for (i = 0; i < f->num_members; i++) {
Eric Dumazet94f633e2021-04-14 12:36:44 -07001536 if (rcu_dereference_protected(f->arr[i],
1537 lockdep_is_held(&f->lock)) == sk)
David S. Millerdc99f602011-07-05 01:45:05 -07001538 break;
1539 }
1540 BUG_ON(i >= f->num_members);
Eric Dumazet94f633e2021-04-14 12:36:44 -07001541 rcu_assign_pointer(f->arr[i],
1542 rcu_dereference_protected(f->arr[f->num_members - 1],
1543 lockdep_is_held(&f->lock)));
David S. Millerdc99f602011-07-05 01:45:05 -07001544 f->num_members--;
Anoob Soman2bd624b2017-02-15 20:25:39 +00001545 if (f->num_members == 0)
1546 __dev_remove_pack(&f->prot_hook);
David S. Millerdc99f602011-07-05 01:45:05 -07001547 spin_unlock(&f->lock);
1548}
1549
Weilong Chend4dd8ae2013-12-23 11:31:38 +08001550static bool match_fanout_group(struct packet_type *ptype, struct sock *sk)
Eric Leblondc0de08d2012-08-16 22:02:58 +00001551{
Eric Dumazet161642e2015-10-09 11:29:32 -07001552 if (sk->sk_family != PF_PACKET)
1553 return false;
Eric Leblondc0de08d2012-08-16 22:02:58 +00001554
Eric Dumazet161642e2015-10-09 11:29:32 -07001555 return ptype->af_packet_priv == pkt_sk(sk)->fanout;
Eric Leblondc0de08d2012-08-16 22:02:58 +00001556}
1557
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001558static void fanout_init_data(struct packet_fanout *f)
1559{
1560 switch (f->type) {
1561 case PACKET_FANOUT_LB:
1562 atomic_set(&f->rr_cur, 0);
1563 break;
1564 case PACKET_FANOUT_CBPF:
Willem de Bruijnf2e52092015-08-14 22:31:35 -04001565 case PACKET_FANOUT_EBPF:
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001566 RCU_INIT_POINTER(f->bpf_prog, NULL);
1567 break;
1568 }
1569}
1570
1571static void __fanout_set_data_bpf(struct packet_fanout *f, struct bpf_prog *new)
1572{
1573 struct bpf_prog *old;
1574
1575 spin_lock(&f->lock);
1576 old = rcu_dereference_protected(f->bpf_prog, lockdep_is_held(&f->lock));
1577 rcu_assign_pointer(f->bpf_prog, new);
1578 spin_unlock(&f->lock);
1579
1580 if (old) {
1581 synchronize_net();
1582 bpf_prog_destroy(old);
1583 }
1584}
1585
Christoph Hellwigb1ea9ff2020-07-23 08:08:47 +02001586static int fanout_set_data_cbpf(struct packet_sock *po, sockptr_t data,
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001587 unsigned int len)
1588{
1589 struct bpf_prog *new;
1590 struct sock_fprog fprog;
1591 int ret;
1592
1593 if (sock_flag(&po->sk, SOCK_FILTER_LOCKED))
1594 return -EPERM;
Christoph Hellwig4d295e52020-07-17 08:23:13 +02001595
1596 ret = copy_bpf_fprog_from_user(&fprog, data, len);
1597 if (ret)
1598 return ret;
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001599
Daniel Borkmannbab18992015-10-02 15:17:33 +02001600 ret = bpf_prog_create_from_user(&new, &fprog, NULL, false);
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001601 if (ret)
1602 return ret;
1603
1604 __fanout_set_data_bpf(po->fanout, new);
1605 return 0;
1606}
1607
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02001608static int fanout_set_data_ebpf(struct packet_sock *po, sockptr_t data,
Willem de Bruijnf2e52092015-08-14 22:31:35 -04001609 unsigned int len)
1610{
1611 struct bpf_prog *new;
1612 u32 fd;
1613
1614 if (sock_flag(&po->sk, SOCK_FILTER_LOCKED))
1615 return -EPERM;
1616 if (len != sizeof(fd))
1617 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02001618 if (copy_from_sockptr(&fd, data, len))
Willem de Bruijnf2e52092015-08-14 22:31:35 -04001619 return -EFAULT;
1620
Daniel Borkmann113214b2016-06-30 17:24:44 +02001621 new = bpf_prog_get_type(fd, BPF_PROG_TYPE_SOCKET_FILTER);
Willem de Bruijnf2e52092015-08-14 22:31:35 -04001622 if (IS_ERR(new))
1623 return PTR_ERR(new);
Willem de Bruijnf2e52092015-08-14 22:31:35 -04001624
1625 __fanout_set_data_bpf(po->fanout, new);
1626 return 0;
1627}
1628
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02001629static int fanout_set_data(struct packet_sock *po, sockptr_t data,
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001630 unsigned int len)
1631{
1632 switch (po->fanout->type) {
1633 case PACKET_FANOUT_CBPF:
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02001634 return fanout_set_data_cbpf(po, data, len);
Willem de Bruijnf2e52092015-08-14 22:31:35 -04001635 case PACKET_FANOUT_EBPF:
1636 return fanout_set_data_ebpf(po, data, len);
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001637 default:
1638 return -EINVAL;
zhong jiang07d53ae2018-08-04 19:41:41 +08001639 }
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001640}
1641
1642static void fanout_release_data(struct packet_fanout *f)
1643{
1644 switch (f->type) {
1645 case PACKET_FANOUT_CBPF:
Willem de Bruijnf2e52092015-08-14 22:31:35 -04001646 case PACKET_FANOUT_EBPF:
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001647 __fanout_set_data_bpf(f, NULL);
zhong jiang07d53ae2018-08-04 19:41:41 +08001648 }
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001649}
1650
Mike Maloney4a69a862017-04-21 10:56:11 -04001651static bool __fanout_id_is_free(struct sock *sk, u16 candidate_id)
1652{
1653 struct packet_fanout *f;
1654
1655 list_for_each_entry(f, &fanout_list, list) {
1656 if (f->id == candidate_id &&
1657 read_pnet(&f->net) == sock_net(sk)) {
1658 return false;
1659 }
1660 }
1661 return true;
1662}
1663
1664static bool fanout_find_new_id(struct sock *sk, u16 *new_id)
1665{
1666 u16 id = fanout_next_id;
1667
1668 do {
1669 if (__fanout_id_is_free(sk, id)) {
1670 *new_id = id;
1671 fanout_next_id = id + 1;
1672 return true;
1673 }
1674
1675 id++;
1676 } while (id != fanout_next_id);
1677
1678 return false;
1679}
1680
Tanner Love9c661b02020-11-06 13:07:40 -05001681static int fanout_add(struct sock *sk, struct fanout_args *args)
David S. Millerdc99f602011-07-05 01:45:05 -07001682{
Eric Dumazetd199fab2017-02-14 09:03:51 -08001683 struct packet_rollover *rollover = NULL;
David S. Millerdc99f602011-07-05 01:45:05 -07001684 struct packet_sock *po = pkt_sk(sk);
Tanner Love9c661b02020-11-06 13:07:40 -05001685 u16 type_flags = args->type_flags;
David S. Millerdc99f602011-07-05 01:45:05 -07001686 struct packet_fanout *f, *match;
David S. Miller7736d332011-07-05 01:43:20 -07001687 u8 type = type_flags & 0xff;
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001688 u8 flags = type_flags >> 8;
Tanner Love9c661b02020-11-06 13:07:40 -05001689 u16 id = args->id;
David S. Millerdc99f602011-07-05 01:45:05 -07001690 int err;
1691
1692 switch (type) {
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001693 case PACKET_FANOUT_ROLLOVER:
1694 if (type_flags & PACKET_FANOUT_FLAG_ROLLOVER)
1695 return -EINVAL;
Gustavo A. R. Silva5af5a022020-11-20 12:38:20 -06001696 break;
David S. Millerdc99f602011-07-05 01:45:05 -07001697 case PACKET_FANOUT_HASH:
1698 case PACKET_FANOUT_LB:
David S. Miller95ec3eb2011-07-06 01:56:38 -07001699 case PACKET_FANOUT_CPU:
Daniel Borkmann5df0ddf2013-08-28 22:13:09 +02001700 case PACKET_FANOUT_RND:
Neil Horman2d360972014-01-22 16:01:44 -05001701 case PACKET_FANOUT_QM:
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001702 case PACKET_FANOUT_CBPF:
Willem de Bruijnf2e52092015-08-14 22:31:35 -04001703 case PACKET_FANOUT_EBPF:
David S. Millerdc99f602011-07-05 01:45:05 -07001704 break;
1705 default:
1706 return -EINVAL;
1707 }
1708
Eric Dumazetd199fab2017-02-14 09:03:51 -08001709 mutex_lock(&fanout_mutex);
David S. Millerdc99f602011-07-05 01:45:05 -07001710
Eric Dumazetd199fab2017-02-14 09:03:51 -08001711 err = -EALREADY;
David S. Millerdc99f602011-07-05 01:45:05 -07001712 if (po->fanout)
Eric Dumazetd199fab2017-02-14 09:03:51 -08001713 goto out;
David S. Millerdc99f602011-07-05 01:45:05 -07001714
Willem de Bruijn4633c9e2015-05-17 19:44:02 -04001715 if (type == PACKET_FANOUT_ROLLOVER ||
1716 (type_flags & PACKET_FANOUT_FLAG_ROLLOVER)) {
Eric Dumazetd199fab2017-02-14 09:03:51 -08001717 err = -ENOMEM;
1718 rollover = kzalloc(sizeof(*rollover), GFP_KERNEL);
1719 if (!rollover)
1720 goto out;
1721 atomic_long_set(&rollover->num, 0);
1722 atomic_long_set(&rollover->num_huge, 0);
1723 atomic_long_set(&rollover->num_failed, 0);
Willem de Bruijn0648ab72015-05-12 11:56:46 -04001724 }
1725
Mike Maloney4a69a862017-04-21 10:56:11 -04001726 if (type_flags & PACKET_FANOUT_FLAG_UNIQUEID) {
1727 if (id != 0) {
1728 err = -EINVAL;
1729 goto out;
1730 }
1731 if (!fanout_find_new_id(sk, &id)) {
1732 err = -ENOMEM;
1733 goto out;
1734 }
1735 /* ephemeral flag for the first socket in the group: drop it */
1736 flags &= ~(PACKET_FANOUT_FLAG_UNIQUEID >> 8);
1737 }
1738
David S. Millerdc99f602011-07-05 01:45:05 -07001739 match = NULL;
1740 list_for_each_entry(f, &fanout_list, list) {
1741 if (f->id == id &&
1742 read_pnet(&f->net) == sock_net(sk)) {
1743 match = f;
1744 break;
1745 }
1746 }
Eric Dumazetafe62c62011-07-07 06:41:29 -07001747 err = -EINVAL;
Tanner Love9c661b02020-11-06 13:07:40 -05001748 if (match) {
1749 if (match->flags != flags)
1750 goto out;
1751 if (args->max_num_members &&
1752 args->max_num_members != match->max_num_members)
1753 goto out;
1754 } else {
1755 if (args->max_num_members > PACKET_FANOUT_MAX)
1756 goto out;
1757 if (!args->max_num_members)
1758 /* legacy PACKET_FANOUT_MAX */
1759 args->max_num_members = 256;
Eric Dumazetafe62c62011-07-07 06:41:29 -07001760 err = -ENOMEM;
Tanner Love9c661b02020-11-06 13:07:40 -05001761 match = kvzalloc(struct_size(match, arr, args->max_num_members),
1762 GFP_KERNEL);
Eric Dumazetafe62c62011-07-07 06:41:29 -07001763 if (!match)
1764 goto out;
1765 write_pnet(&match->net, sock_net(sk));
1766 match->id = id;
1767 match->type = type;
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00001768 match->flags = flags;
Eric Dumazetafe62c62011-07-07 06:41:29 -07001769 INIT_LIST_HEAD(&match->list);
1770 spin_lock_init(&match->lock);
Reshetova, Elenafb5c2c12017-06-30 13:08:10 +03001771 refcount_set(&match->sk_ref, 0);
Willem de Bruijn47dceb82015-08-14 22:31:34 -04001772 fanout_init_data(match);
Eric Dumazetafe62c62011-07-07 06:41:29 -07001773 match->prot_hook.type = po->prot_hook.type;
1774 match->prot_hook.dev = po->prot_hook.dev;
1775 match->prot_hook.func = packet_rcv_fanout;
1776 match->prot_hook.af_packet_priv = match;
Eric Leblondc0de08d2012-08-16 22:02:58 +00001777 match->prot_hook.id_match = match_fanout_group;
Tanner Love9c661b02020-11-06 13:07:40 -05001778 match->max_num_members = args->max_num_members;
Eric Dumazetafe62c62011-07-07 06:41:29 -07001779 list_add(&match->list, &fanout_list);
1780 }
1781 err = -EINVAL;
Willem de Bruijn008ba2a2017-09-14 17:14:41 -04001782
1783 spin_lock(&po->bind_lock);
1784 if (po->running &&
1785 match->type == type &&
Eric Dumazetafe62c62011-07-07 06:41:29 -07001786 match->prot_hook.type == po->prot_hook.type &&
1787 match->prot_hook.dev == po->prot_hook.dev) {
1788 err = -ENOSPC;
Tanner Love9c661b02020-11-06 13:07:40 -05001789 if (refcount_read(&match->sk_ref) < match->max_num_members) {
Eric Dumazetafe62c62011-07-07 06:41:29 -07001790 __dev_remove_pack(&po->prot_hook);
1791 po->fanout = match;
Mike Maloney57f015f2017-11-28 10:44:29 -05001792 po->rollover = rollover;
1793 rollover = NULL;
Reshetova, Elenafb5c2c12017-06-30 13:08:10 +03001794 refcount_set(&match->sk_ref, refcount_read(&match->sk_ref) + 1);
Eric Dumazetafe62c62011-07-07 06:41:29 -07001795 __fanout_link(sk, po);
1796 err = 0;
David S. Millerdc99f602011-07-05 01:45:05 -07001797 }
1798 }
Willem de Bruijn008ba2a2017-09-14 17:14:41 -04001799 spin_unlock(&po->bind_lock);
1800
1801 if (err && !refcount_read(&match->sk_ref)) {
1802 list_del(&match->list);
Tanner Love9c661b02020-11-06 13:07:40 -05001803 kvfree(match);
Willem de Bruijn008ba2a2017-09-14 17:14:41 -04001804 }
1805
Eric Dumazetafe62c62011-07-07 06:41:29 -07001806out:
Mike Maloney57f015f2017-11-28 10:44:29 -05001807 kfree(rollover);
Eric Dumazetd199fab2017-02-14 09:03:51 -08001808 mutex_unlock(&fanout_mutex);
David S. Millerdc99f602011-07-05 01:45:05 -07001809 return err;
1810}
1811
Anoob Soman2bd624b2017-02-15 20:25:39 +00001812/* If pkt_sk(sk)->fanout->sk_ref is zero, this function removes
1813 * pkt_sk(sk)->fanout from fanout_list and returns pkt_sk(sk)->fanout.
1814 * It is the responsibility of the caller to call fanout_release_data() and
1815 * free the returned packet_fanout (after synchronize_net())
1816 */
1817static struct packet_fanout *fanout_release(struct sock *sk)
David S. Millerdc99f602011-07-05 01:45:05 -07001818{
1819 struct packet_sock *po = pkt_sk(sk);
1820 struct packet_fanout *f;
1821
Pavel Emelyanovfff33212012-08-16 05:36:48 +00001822 mutex_lock(&fanout_mutex);
Eric Dumazetd199fab2017-02-14 09:03:51 -08001823 f = po->fanout;
1824 if (f) {
1825 po->fanout = NULL;
David S. Millerdc99f602011-07-05 01:45:05 -07001826
Reshetova, Elenafb5c2c12017-06-30 13:08:10 +03001827 if (refcount_dec_and_test(&f->sk_ref))
Eric Dumazetd199fab2017-02-14 09:03:51 -08001828 list_del(&f->list);
Anoob Soman2bd624b2017-02-15 20:25:39 +00001829 else
1830 f = NULL;
David S. Millerdc99f602011-07-05 01:45:05 -07001831 }
1832 mutex_unlock(&fanout_mutex);
Anoob Soman2bd624b2017-02-15 20:25:39 +00001833
1834 return f;
David S. Millerdc99f602011-07-05 01:45:05 -07001835}
Linus Torvalds1da177e2005-04-16 15:20:36 -07001836
Daniel Borkmann3c70c132015-11-11 23:25:42 +01001837static bool packet_extra_vlan_len_allowed(const struct net_device *dev,
1838 struct sk_buff *skb)
1839{
1840 /* Earlier code assumed this would be a VLAN pkt, double-check
1841 * this now that we have the actual packet in hand. We can only
1842 * do this check on Ethernet devices.
1843 */
1844 if (unlikely(dev->type != ARPHRD_ETHER))
1845 return false;
1846
1847 skb_reset_mac_header(skb);
1848 return likely(eth_hdr(skb)->h_proto == htons(ETH_P_8021Q));
1849}
1850
Eric Dumazet90ddc4f2005-12-22 12:49:22 -08001851static const struct proto_ops packet_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001852
Eric Dumazet90ddc4f2005-12-22 12:49:22 -08001853static const struct proto_ops packet_ops_spkt;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001854
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00001855static int packet_rcv_spkt(struct sk_buff *skb, struct net_device *dev,
1856 struct packet_type *pt, struct net_device *orig_dev)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001857{
1858 struct sock *sk;
1859 struct sockaddr_pkt *spkt;
1860
1861 /*
1862 * When we registered the protocol we saved the socket in the data
1863 * field for just this event.
1864 */
1865
1866 sk = pt->af_packet_priv;
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09001867
Linus Torvalds1da177e2005-04-16 15:20:36 -07001868 /*
1869 * Yank back the headers [hope the device set this
1870 * right or kerboom...]
1871 *
1872 * Incoming packets have ll header pulled,
1873 * push it back.
1874 *
Arnaldo Carvalho de Melo98e399f2007-03-19 15:33:04 -07001875 * For outgoing ones skb->data == skb_mac_header(skb)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001876 * so that this procedure is noop.
1877 */
1878
1879 if (skb->pkt_type == PACKET_LOOPBACK)
1880 goto out;
1881
Octavian Purdila09ad9bc2009-11-25 15:14:13 -08001882 if (!net_eq(dev_net(dev), sock_net(sk)))
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08001883 goto out;
1884
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00001885 skb = skb_share_check(skb, GFP_ATOMIC);
1886 if (skb == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001887 goto oom;
1888
1889 /* drop any routing info */
Eric Dumazetadf30902009-06-02 05:19:30 +00001890 skb_dst_drop(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001891
Phil Oester84531c22005-07-12 11:57:52 -07001892 /* drop conntrack reference */
Florian Westphal895b5c92019-09-29 20:54:03 +02001893 nf_reset_ct(skb);
Phil Oester84531c22005-07-12 11:57:52 -07001894
Herbert Xuffbc6112007-02-04 23:33:10 -08001895 spkt = &PACKET_SKB_CB(skb)->sa.pkt;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001896
Arnaldo Carvalho de Melo98e399f2007-03-19 15:33:04 -07001897 skb_push(skb, skb->data - skb_mac_header(skb));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001898
1899 /*
1900 * The SOCK_PACKET socket receives _all_ frames.
1901 */
1902
1903 spkt->spkt_family = dev->type;
1904 strlcpy(spkt->spkt_device, dev->name, sizeof(spkt->spkt_device));
1905 spkt->spkt_protocol = skb->protocol;
1906
1907 /*
1908 * Charge the memory to the socket. This is done specifically
1909 * to prevent sockets using all the memory up.
1910 */
1911
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00001912 if (sock_queue_rcv_skb(sk, skb) == 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001913 return 0;
1914
1915out:
1916 kfree_skb(skb);
1917oom:
1918 return 0;
1919}
1920
Maxim Mikityanskiy75c65772019-02-21 12:40:01 +00001921static void packet_parse_headers(struct sk_buff *skb, struct socket *sock)
1922{
Yoshiki Komachi18bed892019-03-18 14:39:52 +09001923 if ((!skb->protocol || skb->protocol == htons(ETH_P_ALL)) &&
1924 sock->type == SOCK_RAW) {
Maxim Mikityanskiy75c65772019-02-21 12:40:01 +00001925 skb_reset_mac_header(skb);
1926 skb->protocol = dev_parse_header_protocol(skb);
1927 }
1928
1929 skb_probe_transport_header(skb);
1930}
Linus Torvalds1da177e2005-04-16 15:20:36 -07001931
1932/*
1933 * Output a raw packet to a device layer. This bypasses all the other
1934 * protocol layers and you must therefore supply it with a complete frame
1935 */
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09001936
Ying Xue1b784142015-03-02 15:37:48 +08001937static int packet_sendmsg_spkt(struct socket *sock, struct msghdr *msg,
1938 size_t len)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001939{
1940 struct sock *sk = sock->sk;
Steffen Hurrle342dfc32014-01-17 22:53:15 +01001941 DECLARE_SOCKADDR(struct sockaddr_pkt *, saddr, msg->msg_name);
Eric Dumazet1a35ca82009-12-15 05:47:03 +00001942 struct sk_buff *skb = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001943 struct net_device *dev;
Soheil Hassas Yeganehc14ac942016-04-02 23:08:12 -04001944 struct sockcm_cookie sockc;
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00001945 __be16 proto = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001946 int err;
Ben Greear3bdc0eb2012-02-11 15:39:30 +00001947 int extra_len = 0;
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09001948
Linus Torvalds1da177e2005-04-16 15:20:36 -07001949 /*
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09001950 * Get and verify the address.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001951 */
1952
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00001953 if (saddr) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001954 if (msg->msg_namelen < sizeof(struct sockaddr))
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00001955 return -EINVAL;
1956 if (msg->msg_namelen == sizeof(struct sockaddr_pkt))
1957 proto = saddr->spkt_protocol;
1958 } else
1959 return -ENOTCONN; /* SOCK_PACKET must be sent giving an address */
Linus Torvalds1da177e2005-04-16 15:20:36 -07001960
1961 /*
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09001962 * Find the device first to size check it
Linus Torvalds1da177e2005-04-16 15:20:36 -07001963 */
1964
danborkmann@iogearbox.netde74e922012-06-10 08:59:28 +00001965 saddr->spkt_device[sizeof(saddr->spkt_device) - 1] = 0;
Eric Dumazet1a35ca82009-12-15 05:47:03 +00001966retry:
Eric Dumazet654d1f82009-11-02 10:43:32 +01001967 rcu_read_lock();
1968 dev = dev_get_by_name_rcu(sock_net(sk), saddr->spkt_device);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001969 err = -ENODEV;
1970 if (dev == NULL)
1971 goto out_unlock;
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09001972
David S. Millerd5e76b02007-01-25 19:30:36 -08001973 err = -ENETDOWN;
1974 if (!(dev->flags & IFF_UP))
1975 goto out_unlock;
1976
Linus Torvalds1da177e2005-04-16 15:20:36 -07001977 /*
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00001978 * You may not queue a frame bigger than the mtu. This is the lowest level
1979 * raw protocol and you must do your own fragmentation at this level.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001980 */
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09001981
Ben Greear3bdc0eb2012-02-11 15:39:30 +00001982 if (unlikely(sock_flag(sk, SOCK_NOFCS))) {
1983 if (!netif_supports_nofcs(dev)) {
1984 err = -EPROTONOSUPPORT;
1985 goto out_unlock;
1986 }
1987 extra_len = 4; /* We're doing our own CRC */
1988 }
1989
Linus Torvalds1da177e2005-04-16 15:20:36 -07001990 err = -EMSGSIZE;
Ben Greear3bdc0eb2012-02-11 15:39:30 +00001991 if (len > dev->mtu + dev->hard_header_len + VLAN_HLEN + extra_len)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001992 goto out_unlock;
1993
Eric Dumazet1a35ca82009-12-15 05:47:03 +00001994 if (!skb) {
1995 size_t reserved = LL_RESERVED_SPACE(dev);
Herbert Xu4ce40912011-11-18 02:20:05 +00001996 int tlen = dev->needed_tailroom;
Eric Dumazet1a35ca82009-12-15 05:47:03 +00001997 unsigned int hhlen = dev->header_ops ? dev->hard_header_len : 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001998
Eric Dumazet1a35ca82009-12-15 05:47:03 +00001999 rcu_read_unlock();
Herbert Xu4ce40912011-11-18 02:20:05 +00002000 skb = sock_wmalloc(sk, len + reserved + tlen, 0, GFP_KERNEL);
Eric Dumazet1a35ca82009-12-15 05:47:03 +00002001 if (skb == NULL)
2002 return -ENOBUFS;
2003 /* FIXME: Save some space for broken drivers that write a hard
2004 * header at transmission time by themselves. PPP is the notable
2005 * one here. This should really be fixed at the driver level.
2006 */
2007 skb_reserve(skb, reserved);
2008 skb_reset_network_header(skb);
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09002009
Eric Dumazet1a35ca82009-12-15 05:47:03 +00002010 /* Try to align data part correctly */
2011 if (hhlen) {
2012 skb->data -= hhlen;
2013 skb->tail -= hhlen;
2014 if (len < hhlen)
2015 skb_reset_network_header(skb);
2016 }
Al Viro6ce8e9c2014-04-06 21:25:44 -04002017 err = memcpy_from_msg(skb_put(skb, len), msg, len);
Eric Dumazet1a35ca82009-12-15 05:47:03 +00002018 if (err)
2019 goto out_free;
2020 goto retry;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002021 }
2022
Willem de Bruijn9ed988c2016-03-09 21:58:34 -05002023 if (!dev_validate_header(dev, skb->data, len)) {
2024 err = -EINVAL;
2025 goto out_unlock;
2026 }
Daniel Borkmann3c70c132015-11-11 23:25:42 +01002027 if (len > (dev->mtu + dev->hard_header_len + extra_len) &&
2028 !packet_extra_vlan_len_allowed(dev, skb)) {
2029 err = -EMSGSIZE;
2030 goto out_unlock;
Ben Greear57f89bf2011-02-11 09:35:18 +00002031 }
Eric Dumazet1a35ca82009-12-15 05:47:03 +00002032
Willem de Bruijn657a0662018-07-06 10:12:56 -04002033 sockcm_init(&sockc, sk);
Soheil Hassas Yeganehc14ac942016-04-02 23:08:12 -04002034 if (msg->msg_controllen) {
2035 err = sock_cmsg_send(sk, msg, &sockc);
Soheil Hassas Yeganehf8e77182016-07-20 18:01:18 -04002036 if (unlikely(err))
Soheil Hassas Yeganehc14ac942016-04-02 23:08:12 -04002037 goto out_unlock;
Soheil Hassas Yeganehc14ac942016-04-02 23:08:12 -04002038 }
2039
Linus Torvalds1da177e2005-04-16 15:20:36 -07002040 skb->protocol = proto;
2041 skb->dev = dev;
2042 skb->priority = sk->sk_priority;
Eric Dumazet2d37a182009-10-01 19:14:46 +00002043 skb->mark = sk->sk_mark;
Richard Cochran3d0ba8c2018-07-03 15:42:51 -07002044 skb->tstamp = sockc.transmit_time;
Daniel Borkmannbf84a0102013-04-14 08:08:13 +00002045
Willem de Bruijn8f932f72018-12-17 12:24:00 -05002046 skb_setup_tx_timestamp(skb, sockc.tsflags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002047
Ben Greear3bdc0eb2012-02-11 15:39:30 +00002048 if (unlikely(extra_len == 4))
2049 skb->no_fcs = 1;
2050
Maxim Mikityanskiy75c65772019-02-21 12:40:01 +00002051 packet_parse_headers(skb, sock);
Jason Wangc1aad272013-03-25 20:19:57 +00002052
Linus Torvalds1da177e2005-04-16 15:20:36 -07002053 dev_queue_xmit(skb);
Eric Dumazet654d1f82009-11-02 10:43:32 +01002054 rcu_read_unlock();
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00002055 return len;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002056
Linus Torvalds1da177e2005-04-16 15:20:36 -07002057out_unlock:
Eric Dumazet654d1f82009-11-02 10:43:32 +01002058 rcu_read_unlock();
Eric Dumazet1a35ca82009-12-15 05:47:03 +00002059out_free:
2060 kfree_skb(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002061 return err;
2062}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002063
Alexei Starovoitovff936a02015-10-07 10:55:41 -07002064static unsigned int run_filter(struct sk_buff *skb,
2065 const struct sock *sk,
2066 unsigned int res)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002067{
2068 struct sk_filter *filter;
2069
Eric Dumazet80f8f102011-01-18 07:46:52 +00002070 rcu_read_lock();
2071 filter = rcu_dereference(sk->sk_filter);
David S. Millerdbcb5852007-01-24 15:21:02 -08002072 if (filter != NULL)
Alexei Starovoitovff936a02015-10-07 10:55:41 -07002073 res = bpf_prog_run_clear_cb(filter->prog, skb);
Eric Dumazet80f8f102011-01-18 07:46:52 +00002074 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002075
David S. Millerdbcb5852007-01-24 15:21:02 -08002076 return res;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002077}
2078
Willem de Bruijn16cc1402016-02-03 18:02:14 -05002079static int packet_rcv_vnet(struct msghdr *msg, const struct sk_buff *skb,
2080 size_t *len)
2081{
2082 struct virtio_net_hdr vnet_hdr;
2083
2084 if (*len < sizeof(vnet_hdr))
2085 return -EINVAL;
2086 *len -= sizeof(vnet_hdr);
2087
Willem de Bruijnfd3a8862018-06-06 11:23:01 -04002088 if (virtio_net_hdr_from_skb(skb, &vnet_hdr, vio_le(), true, 0))
Willem de Bruijn16cc1402016-02-03 18:02:14 -05002089 return -EINVAL;
2090
2091 return memcpy_to_msg(msg, (void *)&vnet_hdr, sizeof(vnet_hdr));
2092}
2093
Linus Torvalds1da177e2005-04-16 15:20:36 -07002094/*
Eric Dumazet62ab0812010-12-06 20:50:09 +00002095 * This function makes lazy skb cloning in hope that most of packets
2096 * are discarded by BPF.
2097 *
2098 * Note tricky part: we DO mangle shared skb! skb->data, skb->len
2099 * and skb->cb are mangled. It works because (and until) packets
2100 * falling here are owned by current CPU. Output packets are cloned
2101 * by dev_queue_xmit_nit(), input packets are processed by net_bh
Wang Hai0e4161d2021-03-24 14:19:31 +08002102 * sequentially, so that if we return skb to original state on exit,
Eric Dumazet62ab0812010-12-06 20:50:09 +00002103 * we will not harm anyone.
Linus Torvalds1da177e2005-04-16 15:20:36 -07002104 */
2105
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00002106static int packet_rcv(struct sk_buff *skb, struct net_device *dev,
2107 struct packet_type *pt, struct net_device *orig_dev)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002108{
2109 struct sock *sk;
2110 struct sockaddr_ll *sll;
2111 struct packet_sock *po;
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00002112 u8 *skb_head = skb->data;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002113 int skb_len = skb->len;
David S. Millerdbcb5852007-01-24 15:21:02 -08002114 unsigned int snaplen, res;
Weongyo Jeongda378452016-04-14 14:10:04 -07002115 bool is_drop_n_account = false;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002116
2117 if (skb->pkt_type == PACKET_LOOPBACK)
2118 goto drop;
2119
2120 sk = pt->af_packet_priv;
2121 po = pkt_sk(sk);
2122
Octavian Purdila09ad9bc2009-11-25 15:14:13 -08002123 if (!net_eq(dev_net(dev), sock_net(sk)))
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08002124 goto drop;
2125
Linus Torvalds1da177e2005-04-16 15:20:36 -07002126 skb->dev = dev;
2127
Eyal Birgerd5496992020-11-21 08:28:17 +02002128 if (dev_has_header(dev)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002129 /* The device has an explicit notion of ll header,
Eric Dumazet62ab0812010-12-06 20:50:09 +00002130 * exported to higher levels.
2131 *
2132 * Otherwise, the device hides details of its frame
2133 * structure, so that corresponding packet head is
2134 * never delivered to user.
Linus Torvalds1da177e2005-04-16 15:20:36 -07002135 */
2136 if (sk->sk_type != SOCK_DGRAM)
Arnaldo Carvalho de Melo98e399f2007-03-19 15:33:04 -07002137 skb_push(skb, skb->data - skb_mac_header(skb));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002138 else if (skb->pkt_type == PACKET_OUTGOING) {
2139 /* Special case: outgoing packets have ll header at head */
Arnaldo Carvalho de Melobbe735e2007-03-10 22:16:10 -03002140 skb_pull(skb, skb_network_offset(skb));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002141 }
2142 }
2143
2144 snaplen = skb->len;
2145
David S. Millerdbcb5852007-01-24 15:21:02 -08002146 res = run_filter(skb, sk, snaplen);
2147 if (!res)
Dmitry Mishinfda9ef52006-08-31 15:28:39 -07002148 goto drop_n_restore;
David S. Millerdbcb5852007-01-24 15:21:02 -08002149 if (snaplen > res)
2150 snaplen = res;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002151
Eric Dumazet0fd7bac2011-12-21 07:11:44 +00002152 if (atomic_read(&sk->sk_rmem_alloc) >= sk->sk_rcvbuf)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002153 goto drop_n_acct;
2154
2155 if (skb_shared(skb)) {
2156 struct sk_buff *nskb = skb_clone(skb, GFP_ATOMIC);
2157 if (nskb == NULL)
2158 goto drop_n_acct;
2159
2160 if (skb_head != skb->data) {
2161 skb->data = skb_head;
2162 skb->len = skb_len;
2163 }
Eric Dumazetabc4e4f2012-04-19 02:24:42 +00002164 consume_skb(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002165 skb = nskb;
2166 }
2167
Eyal Birgerb4772ef2015-03-01 14:58:29 +02002168 sock_skb_cb_check_size(sizeof(*PACKET_SKB_CB(skb)) + MAX_ADDR_LEN - 8);
Herbert Xuffbc6112007-02-04 23:33:10 -08002169
2170 sll = &PACKET_SKB_CB(skb)->sa.ll;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002171 sll->sll_hatype = dev->type;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002172 sll->sll_pkttype = skb->pkt_type;
Peter P Waskiewicz Jr8032b462007-11-10 22:03:25 -08002173 if (unlikely(po->origdev))
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07002174 sll->sll_ifindex = orig_dev->ifindex;
2175 else
2176 sll->sll_ifindex = dev->ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002177
Stephen Hemmingerb95cce32007-09-26 22:13:38 -07002178 sll->sll_halen = dev_parse_header(skb, sll->sll_addr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002179
Eyal Birger2472d762015-03-01 14:58:28 +02002180 /* sll->sll_family and sll->sll_protocol are set in packet_recvmsg().
2181 * Use their space for storing the original skb length.
2182 */
2183 PACKET_SKB_CB(skb)->sa.origlen = skb->len;
Herbert Xu8dc41942007-02-04 23:31:32 -08002184
Linus Torvalds1da177e2005-04-16 15:20:36 -07002185 if (pskb_trim(skb, snaplen))
2186 goto drop_n_acct;
2187
2188 skb_set_owner_r(skb, sk);
2189 skb->dev = NULL;
Eric Dumazetadf30902009-06-02 05:19:30 +00002190 skb_dst_drop(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002191
Phil Oester84531c22005-07-12 11:57:52 -07002192 /* drop conntrack reference */
Florian Westphal895b5c92019-09-29 20:54:03 +02002193 nf_reset_ct(skb);
Phil Oester84531c22005-07-12 11:57:52 -07002194
Linus Torvalds1da177e2005-04-16 15:20:36 -07002195 spin_lock(&sk->sk_receive_queue.lock);
Daniel Borkmannee80fbf2013-04-19 06:12:29 +00002196 po->stats.stats1.tp_packets++;
Eyal Birger3bc3b962015-03-01 14:58:30 +02002197 sock_skb_set_dropcount(sk, skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002198 __skb_queue_tail(&sk->sk_receive_queue, skb);
2199 spin_unlock(&sk->sk_receive_queue.lock);
David S. Miller676d2362014-04-11 16:15:36 -04002200 sk->sk_data_ready(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002201 return 0;
2202
2203drop_n_acct:
Weongyo Jeongda378452016-04-14 14:10:04 -07002204 is_drop_n_account = true;
Eric Dumazet8e8e2952019-06-12 09:52:30 -07002205 atomic_inc(&po->tp_drops);
Willem de Bruijn7091fbd2011-09-30 10:38:28 +00002206 atomic_inc(&sk->sk_drops);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002207
2208drop_n_restore:
2209 if (skb_head != skb->data && skb_shared(skb)) {
2210 skb->data = skb_head;
2211 skb->len = skb_len;
2212 }
2213drop:
Weongyo Jeongda378452016-04-14 14:10:04 -07002214 if (!is_drop_n_account)
2215 consume_skb(skb);
2216 else
2217 kfree_skb(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002218 return 0;
2219}
2220
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00002221static int tpacket_rcv(struct sk_buff *skb, struct net_device *dev,
2222 struct packet_type *pt, struct net_device *orig_dev)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002223{
2224 struct sock *sk;
2225 struct packet_sock *po;
2226 struct sockaddr_ll *sll;
Daniel Borkmann184f4892013-04-16 01:57:46 +00002227 union tpacket_uhdr h;
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00002228 u8 *skb_head = skb->data;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002229 int skb_len = skb->len;
David S. Millerdbcb5852007-01-24 15:21:02 -08002230 unsigned int snaplen, res;
chetan lokef6fb8f12011-08-19 10:18:16 +00002231 unsigned long status = TP_STATUS_USER;
Or Cohenacf69c92020-09-03 21:05:28 -07002232 unsigned short macoff, hdrlen;
2233 unsigned int netoff;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002234 struct sk_buff *copy_skb = NULL;
Arnd Bergmannd413fcb2017-11-27 10:09:24 +01002235 struct timespec64 ts;
Daniel Borkmannb9c32fb2013-04-23 00:39:31 +00002236 __u32 ts_status;
Weongyo Jeongda378452016-04-14 14:10:04 -07002237 bool is_drop_n_account = false;
Willem de Bruijn61fad682020-03-13 12:18:09 -04002238 unsigned int slot_id = 0;
Benjamin Poirieredbd58b2017-08-28 14:29:41 -04002239 bool do_vnet = false;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002240
Atzm Watanabe51846352013-12-17 22:53:32 +09002241 /* struct tpacket{2,3}_hdr is aligned to a multiple of TPACKET_ALIGNMENT.
2242 * We may add members to them until current aligned size without forcing
2243 * userspace to call getsockopt(..., PACKET_HDRLEN, ...).
2244 */
2245 BUILD_BUG_ON(TPACKET_ALIGN(sizeof(*h.h2)) != 32);
2246 BUILD_BUG_ON(TPACKET_ALIGN(sizeof(*h.h3)) != 48);
2247
Linus Torvalds1da177e2005-04-16 15:20:36 -07002248 if (skb->pkt_type == PACKET_LOOPBACK)
2249 goto drop;
2250
2251 sk = pt->af_packet_priv;
2252 po = pkt_sk(sk);
2253
Octavian Purdila09ad9bc2009-11-25 15:14:13 -08002254 if (!net_eq(dev_net(dev), sock_net(sk)))
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08002255 goto drop;
2256
Eyal Birgerd5496992020-11-21 08:28:17 +02002257 if (dev_has_header(dev)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002258 if (sk->sk_type != SOCK_DGRAM)
Arnaldo Carvalho de Melo98e399f2007-03-19 15:33:04 -07002259 skb_push(skb, skb->data - skb_mac_header(skb));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002260 else if (skb->pkt_type == PACKET_OUTGOING) {
2261 /* Special case: outgoing packets have ll header at head */
Arnaldo Carvalho de Melobbe735e2007-03-10 22:16:10 -03002262 skb_pull(skb, skb_network_offset(skb));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002263 }
2264 }
2265
2266 snaplen = skb->len;
2267
David S. Millerdbcb5852007-01-24 15:21:02 -08002268 res = run_filter(skb, sk, snaplen);
2269 if (!res)
Dmitry Mishinfda9ef52006-08-31 15:28:39 -07002270 goto drop_n_restore;
Alexander Drozdov68c2e5d2015-03-23 09:11:12 +03002271
Eric Dumazet2c51c622019-06-12 09:52:31 -07002272 /* If we are flooded, just give up */
2273 if (__packet_rcv_has_room(po, skb) == ROOM_NONE) {
2274 atomic_inc(&po->tp_drops);
2275 goto drop_n_restore;
2276 }
2277
Alexander Drozdov68c2e5d2015-03-23 09:11:12 +03002278 if (skb->ip_summed == CHECKSUM_PARTIAL)
2279 status |= TP_STATUS_CSUMNOTREADY;
Alexander Drozdov682f0482015-03-23 09:11:13 +03002280 else if (skb->pkt_type != PACKET_OUTGOING &&
2281 (skb->ip_summed == CHECKSUM_COMPLETE ||
2282 skb_csum_unnecessary(skb)))
2283 status |= TP_STATUS_CSUM_VALID;
Alexander Drozdov68c2e5d2015-03-23 09:11:12 +03002284
David S. Millerdbcb5852007-01-24 15:21:02 -08002285 if (snaplen > res)
2286 snaplen = res;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002287
2288 if (sk->sk_type == SOCK_DGRAM) {
Patrick McHardy8913336a2008-07-18 18:05:19 -07002289 macoff = netoff = TPACKET_ALIGN(po->tp_hdrlen) + 16 +
2290 po->tp_reserve;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002291 } else {
Eric Dumazet95c96172012-04-15 05:58:06 +00002292 unsigned int maclen = skb_network_offset(skb);
Patrick McHardybbd6ef82008-07-14 22:50:15 -07002293 netoff = TPACKET_ALIGN(po->tp_hdrlen +
Patrick McHardy8913336a2008-07-18 18:05:19 -07002294 (maclen < 16 ? 16 : maclen)) +
Willem de Bruijn58d19b12016-02-03 18:02:15 -05002295 po->tp_reserve;
Benjamin Poirieredbd58b2017-08-28 14:29:41 -04002296 if (po->has_vnet_hdr) {
Willem de Bruijn58d19b12016-02-03 18:02:15 -05002297 netoff += sizeof(struct virtio_net_hdr);
Benjamin Poirieredbd58b2017-08-28 14:29:41 -04002298 do_vnet = true;
2299 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002300 macoff = netoff - maclen;
2301 }
Or Cohenacf69c92020-09-03 21:05:28 -07002302 if (netoff > USHRT_MAX) {
2303 atomic_inc(&po->tp_drops);
2304 goto drop_n_restore;
2305 }
chetan lokef6fb8f12011-08-19 10:18:16 +00002306 if (po->tp_version <= TPACKET_V2) {
2307 if (macoff + snaplen > po->rx_ring.frame_size) {
2308 if (po->copy_thresh &&
Eric Dumazet0fd7bac2011-12-21 07:11:44 +00002309 atomic_read(&sk->sk_rmem_alloc) < sk->sk_rcvbuf) {
chetan lokef6fb8f12011-08-19 10:18:16 +00002310 if (skb_shared(skb)) {
2311 copy_skb = skb_clone(skb, GFP_ATOMIC);
2312 } else {
2313 copy_skb = skb_get(skb);
2314 skb_head = skb->data;
2315 }
2316 if (copy_skb)
2317 skb_set_owner_r(copy_skb, sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002318 }
chetan lokef6fb8f12011-08-19 10:18:16 +00002319 snaplen = po->rx_ring.frame_size - macoff;
Benjamin Poirieredbd58b2017-08-28 14:29:41 -04002320 if ((int)snaplen < 0) {
chetan lokef6fb8f12011-08-19 10:18:16 +00002321 snaplen = 0;
Benjamin Poirieredbd58b2017-08-28 14:29:41 -04002322 do_vnet = false;
2323 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002324 }
Eric Dumazetdc808112014-08-15 09:16:04 -07002325 } else if (unlikely(macoff + snaplen >
2326 GET_PBDQC_FROM_RB(&po->rx_ring)->max_frame_len)) {
2327 u32 nval;
2328
2329 nval = GET_PBDQC_FROM_RB(&po->rx_ring)->max_frame_len - macoff;
2330 pr_err_once("tpacket_rcv: packet too big, clamped from %u to %u. macoff=%u\n",
2331 snaplen, nval, macoff);
2332 snaplen = nval;
2333 if (unlikely((int)snaplen < 0)) {
2334 snaplen = 0;
2335 macoff = GET_PBDQC_FROM_RB(&po->rx_ring)->max_frame_len;
Benjamin Poirieredbd58b2017-08-28 14:29:41 -04002336 do_vnet = false;
Eric Dumazetdc808112014-08-15 09:16:04 -07002337 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002338 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002339 spin_lock(&sk->sk_receive_queue.lock);
chetan lokef6fb8f12011-08-19 10:18:16 +00002340 h.raw = packet_current_rx_frame(po, skb,
2341 TP_STATUS_KERNEL, (macoff+snaplen));
Patrick McHardybbd6ef82008-07-14 22:50:15 -07002342 if (!h.raw)
Willem de Bruijn58d19b12016-02-03 18:02:15 -05002343 goto drop_n_account;
Willem de Bruijn46e4c422020-03-09 11:34:35 -04002344
Willem de Bruijn61fad682020-03-13 12:18:09 -04002345 if (po->tp_version <= TPACKET_V2) {
2346 slot_id = po->rx_ring.head;
2347 if (test_bit(slot_id, po->rx_ring.rx_owner_map))
2348 goto drop_n_account;
2349 __set_bit(slot_id, po->rx_ring.rx_owner_map);
2350 }
2351
Willem de Bruijn46e4c422020-03-09 11:34:35 -04002352 if (do_vnet &&
2353 virtio_net_hdr_from_skb(skb, h.raw + macoff -
2354 sizeof(struct virtio_net_hdr),
John Ogness88fd1cb2020-08-13 21:45:25 +02062355 vio_le(), true, 0)) {
2356 if (po->tp_version == TPACKET_V3)
2357 prb_clear_blk_fill_status(&po->rx_ring);
Willem de Bruijn46e4c422020-03-09 11:34:35 -04002358 goto drop_n_account;
John Ogness88fd1cb2020-08-13 21:45:25 +02062359 }
Willem de Bruijn46e4c422020-03-09 11:34:35 -04002360
chetan lokef6fb8f12011-08-19 10:18:16 +00002361 if (po->tp_version <= TPACKET_V2) {
2362 packet_increment_rx_head(po, &po->rx_ring);
2363 /*
2364 * LOSING will be reported till you read the stats,
2365 * because it's COR - Clear On Read.
2366 * Anyways, moving it for V1/V2 only as V3 doesn't need this
2367 * at packet level.
2368 */
Eric Dumazet8e8e2952019-06-12 09:52:30 -07002369 if (atomic_read(&po->tp_drops))
chetan lokef6fb8f12011-08-19 10:18:16 +00002370 status |= TP_STATUS_LOSING;
2371 }
Eric Dumazet945d0152018-06-21 14:16:02 -07002372
Daniel Borkmannee80fbf2013-04-19 06:12:29 +00002373 po->stats.stats1.tp_packets++;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002374 if (copy_skb) {
2375 status |= TP_STATUS_COPY;
2376 __skb_queue_tail(&sk->sk_receive_queue, copy_skb);
2377 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002378 spin_unlock(&sk->sk_receive_queue.lock);
2379
Patrick McHardybbd6ef82008-07-14 22:50:15 -07002380 skb_copy_bits(skb, 0, h.raw + macoff, snaplen);
Daniel Borkmannb9c32fb2013-04-23 00:39:31 +00002381
Richard Sanger171c3b12021-05-12 13:31:22 +12002382 /* Always timestamp; prefer an existing software timestamp taken
2383 * closer to the time of capture.
2384 */
2385 ts_status = tpacket_get_timestamp(skb, &ts,
2386 po->tp_tstamp | SOF_TIMESTAMPING_SOFTWARE);
2387 if (!ts_status)
Arnd Bergmannd413fcb2017-11-27 10:09:24 +01002388 ktime_get_real_ts64(&ts);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002389
Daniel Borkmannb9c32fb2013-04-23 00:39:31 +00002390 status |= ts_status;
2391
Patrick McHardybbd6ef82008-07-14 22:50:15 -07002392 switch (po->tp_version) {
2393 case TPACKET_V1:
2394 h.h1->tp_len = skb->len;
2395 h.h1->tp_snaplen = snaplen;
2396 h.h1->tp_mac = macoff;
2397 h.h1->tp_net = netoff;
Daniel Borkmann4b457bd2013-04-16 01:29:11 +00002398 h.h1->tp_sec = ts.tv_sec;
2399 h.h1->tp_usec = ts.tv_nsec / NSEC_PER_USEC;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07002400 hdrlen = sizeof(*h.h1);
2401 break;
2402 case TPACKET_V2:
2403 h.h2->tp_len = skb->len;
2404 h.h2->tp_snaplen = snaplen;
2405 h.h2->tp_mac = macoff;
2406 h.h2->tp_net = netoff;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07002407 h.h2->tp_sec = ts.tv_sec;
2408 h.h2->tp_nsec = ts.tv_nsec;
Jiri Pirkodf8a39d2015-01-13 17:13:44 +01002409 if (skb_vlan_tag_present(skb)) {
2410 h.h2->tp_vlan_tci = skb_vlan_tag_get(skb);
Atzm Watanabea0cdfcf2013-12-17 22:53:40 +09002411 h.h2->tp_vlan_tpid = ntohs(skb->vlan_proto);
2412 status |= TP_STATUS_VLAN_VALID | TP_STATUS_VLAN_TPID_VALID;
Ben Greeara3bcc232011-06-01 06:49:10 +00002413 } else {
2414 h.h2->tp_vlan_tci = 0;
Atzm Watanabea0cdfcf2013-12-17 22:53:40 +09002415 h.h2->tp_vlan_tpid = 0;
Ben Greeara3bcc232011-06-01 06:49:10 +00002416 }
Atzm Watanabee4d26f42013-12-17 22:53:36 +09002417 memset(h.h2->tp_padding, 0, sizeof(h.h2->tp_padding));
Patrick McHardybbd6ef82008-07-14 22:50:15 -07002418 hdrlen = sizeof(*h.h2);
2419 break;
chetan lokef6fb8f12011-08-19 10:18:16 +00002420 case TPACKET_V3:
2421 /* tp_nxt_offset,vlan are already populated above.
2422 * So DONT clear those fields here
2423 */
2424 h.h3->tp_status |= status;
2425 h.h3->tp_len = skb->len;
2426 h.h3->tp_snaplen = snaplen;
2427 h.h3->tp_mac = macoff;
2428 h.h3->tp_net = netoff;
chetan lokef6fb8f12011-08-19 10:18:16 +00002429 h.h3->tp_sec = ts.tv_sec;
2430 h.h3->tp_nsec = ts.tv_nsec;
Atzm Watanabee4d26f42013-12-17 22:53:36 +09002431 memset(h.h3->tp_padding, 0, sizeof(h.h3->tp_padding));
chetan lokef6fb8f12011-08-19 10:18:16 +00002432 hdrlen = sizeof(*h.h3);
2433 break;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07002434 default:
2435 BUG();
2436 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002437
Patrick McHardybbd6ef82008-07-14 22:50:15 -07002438 sll = h.raw + TPACKET_ALIGN(hdrlen);
Stephen Hemmingerb95cce32007-09-26 22:13:38 -07002439 sll->sll_halen = dev_parse_header(skb, sll->sll_addr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002440 sll->sll_family = AF_PACKET;
2441 sll->sll_hatype = dev->type;
2442 sll->sll_protocol = skb->protocol;
2443 sll->sll_pkttype = skb->pkt_type;
Peter P Waskiewicz Jr8032b462007-11-10 22:03:25 -08002444 if (unlikely(po->origdev))
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07002445 sll->sll_ifindex = orig_dev->ifindex;
2446 else
2447 sll->sll_ifindex = dev->ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002448
Ralf Baechlee16aa202006-12-07 00:11:33 -08002449 smp_mb();
Daniel Borkmannf0d4eb22014-01-19 11:46:53 +01002450
Changli Gaof6dafa92010-12-07 04:26:16 +00002451#if ARCH_IMPLEMENTS_FLUSH_DCACHE_PAGE == 1
Daniel Borkmannf0d4eb22014-01-19 11:46:53 +01002452 if (po->tp_version <= TPACKET_V2) {
Changli Gao0af55bb2010-12-01 02:52:20 +00002453 u8 *start, *end;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002454
Daniel Borkmannf0d4eb22014-01-19 11:46:53 +01002455 end = (u8 *) PAGE_ALIGN((unsigned long) h.raw +
2456 macoff + snaplen);
2457
2458 for (start = h.raw; start < end; start += PAGE_SIZE)
2459 flush_dcache_page(pgv_to_page(start));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002460 }
Daniel Borkmannf0d4eb22014-01-19 11:46:53 +01002461 smp_wmb();
Changli Gaof6dafa92010-12-07 04:26:16 +00002462#endif
Daniel Borkmannf0d4eb22014-01-19 11:46:53 +01002463
Dan Collinsda413ee2014-12-19 16:49:25 +13002464 if (po->tp_version <= TPACKET_V2) {
Willem de Bruijn61fad682020-03-13 12:18:09 -04002465 spin_lock(&sk->sk_receive_queue.lock);
chetan lokef6fb8f12011-08-19 10:18:16 +00002466 __packet_set_status(po, h.raw, status);
Willem de Bruijn61fad682020-03-13 12:18:09 -04002467 __clear_bit(slot_id, po->rx_ring.rx_owner_map);
2468 spin_unlock(&sk->sk_receive_queue.lock);
Dan Collinsda413ee2014-12-19 16:49:25 +13002469 sk->sk_data_ready(sk);
John Ogness88fd1cb2020-08-13 21:45:25 +02062470 } else if (po->tp_version == TPACKET_V3) {
chetan lokef6fb8f12011-08-19 10:18:16 +00002471 prb_clear_blk_fill_status(&po->rx_ring);
Dan Collinsda413ee2014-12-19 16:49:25 +13002472 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002473
2474drop_n_restore:
2475 if (skb_head != skb->data && skb_shared(skb)) {
2476 skb->data = skb_head;
2477 skb->len = skb_len;
2478 }
2479drop:
Weongyo Jeongda378452016-04-14 14:10:04 -07002480 if (!is_drop_n_account)
2481 consume_skb(skb);
2482 else
2483 kfree_skb(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002484 return 0;
2485
Willem de Bruijn58d19b12016-02-03 18:02:15 -05002486drop_n_account:
Linus Torvalds1da177e2005-04-16 15:20:36 -07002487 spin_unlock(&sk->sk_receive_queue.lock);
Eric Dumazet8e8e2952019-06-12 09:52:30 -07002488 atomic_inc(&po->tp_drops);
2489 is_drop_n_account = true;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002490
David S. Miller676d2362014-04-11 16:15:36 -04002491 sk->sk_data_ready(sk);
Wei Yongjunacb5d752009-02-25 00:36:42 +00002492 kfree_skb(copy_skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002493 goto drop_n_restore;
2494}
2495
Johann Baudy69e3c752009-05-18 22:11:22 -07002496static void tpacket_destruct_skb(struct sk_buff *skb)
2497{
2498 struct packet_sock *po = pkt_sk(skb->sk);
Johann Baudy69e3c752009-05-18 22:11:22 -07002499
Johann Baudy69e3c752009-05-18 22:11:22 -07002500 if (likely(po->tx_ring.pg_vec)) {
Daniel Borkmannf0d4eb22014-01-19 11:46:53 +01002501 void *ph;
Daniel Borkmannb9c32fb2013-04-23 00:39:31 +00002502 __u32 ts;
2503
Willem de Bruijn5cd8d462018-11-20 13:00:18 -05002504 ph = skb_zcopy_get_nouarg(skb);
Daniel Borkmannb0138402014-01-15 16:25:36 +01002505 packet_dec_pending(&po->tx_ring);
Daniel Borkmannb9c32fb2013-04-23 00:39:31 +00002506
2507 ts = __packet_set_timestamp(po, ph, skb);
2508 __packet_set_status(po, ph, TP_STATUS_AVAILABLE | ts);
Neil Horman89ed5b52019-06-25 17:57:49 -04002509
2510 if (!packet_read_pending(&po->tx_ring))
2511 complete(&po->skb_completion);
Johann Baudy69e3c752009-05-18 22:11:22 -07002512 }
2513
2514 sock_wfree(skb);
2515}
2516
Willem de Bruijn16cc1402016-02-03 18:02:14 -05002517static int __packet_snd_vnet_parse(struct virtio_net_hdr *vnet_hdr, size_t len)
2518{
Willem de Bruijn16cc1402016-02-03 18:02:14 -05002519 if ((vnet_hdr->flags & VIRTIO_NET_HDR_F_NEEDS_CSUM) &&
2520 (__virtio16_to_cpu(vio_le(), vnet_hdr->csum_start) +
2521 __virtio16_to_cpu(vio_le(), vnet_hdr->csum_offset) + 2 >
2522 __virtio16_to_cpu(vio_le(), vnet_hdr->hdr_len)))
2523 vnet_hdr->hdr_len = __cpu_to_virtio16(vio_le(),
2524 __virtio16_to_cpu(vio_le(), vnet_hdr->csum_start) +
2525 __virtio16_to_cpu(vio_le(), vnet_hdr->csum_offset) + 2);
2526
2527 if (__virtio16_to_cpu(vio_le(), vnet_hdr->hdr_len) > len)
2528 return -EINVAL;
2529
Willem de Bruijn16cc1402016-02-03 18:02:14 -05002530 return 0;
2531}
2532
2533static int packet_snd_vnet_parse(struct msghdr *msg, size_t *len,
2534 struct virtio_net_hdr *vnet_hdr)
2535{
Willem de Bruijn16cc1402016-02-03 18:02:14 -05002536 if (*len < sizeof(*vnet_hdr))
2537 return -EINVAL;
2538 *len -= sizeof(*vnet_hdr);
2539
Al Virocbbd26b2016-11-01 22:09:04 -04002540 if (!copy_from_iter_full(vnet_hdr, sizeof(*vnet_hdr), &msg->msg_iter))
Willem de Bruijn16cc1402016-02-03 18:02:14 -05002541 return -EFAULT;
2542
2543 return __packet_snd_vnet_parse(vnet_hdr, *len);
2544}
2545
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00002546static int tpacket_fill_skb(struct packet_sock *po, struct sk_buff *skb,
Willem de Bruijn8d39b4a2016-02-03 18:02:16 -05002547 void *frame, struct net_device *dev, void *data, int tp_len,
Soheil Hassas Yeganehc14ac942016-04-02 23:08:12 -04002548 __be16 proto, unsigned char *addr, int hlen, int copylen,
2549 const struct sockcm_cookie *sockc)
Johann Baudy69e3c752009-05-18 22:11:22 -07002550{
Daniel Borkmann184f4892013-04-16 01:57:46 +00002551 union tpacket_uhdr ph;
Willem de Bruijn8d39b4a2016-02-03 18:02:16 -05002552 int to_write, offset, len, nr_frags, len_max;
Johann Baudy69e3c752009-05-18 22:11:22 -07002553 struct socket *sock = po->sk.sk_socket;
2554 struct page *page;
Johann Baudy69e3c752009-05-18 22:11:22 -07002555 int err;
2556
2557 ph.raw = frame;
2558
2559 skb->protocol = proto;
2560 skb->dev = dev;
2561 skb->priority = po->sk.sk_priority;
Eric Dumazet2d37a182009-10-01 19:14:46 +00002562 skb->mark = po->sk.sk_mark;
Richard Cochran3d0ba8c2018-07-03 15:42:51 -07002563 skb->tstamp = sockc->transmit_time;
Willem de Bruijn8f932f72018-12-17 12:24:00 -05002564 skb_setup_tx_timestamp(skb, sockc->tsflags);
Willem de Bruijn5cd8d462018-11-20 13:00:18 -05002565 skb_zcopy_set_nouarg(skb, ph.raw);
Johann Baudy69e3c752009-05-18 22:11:22 -07002566
Herbert Xuae641942011-11-18 02:20:04 +00002567 skb_reserve(skb, hlen);
Johann Baudy69e3c752009-05-18 22:11:22 -07002568 skb_reset_network_header(skb);
Jason Wangc1aad272013-03-25 20:19:57 +00002569
Johann Baudy69e3c752009-05-18 22:11:22 -07002570 to_write = tp_len;
2571
2572 if (sock->type == SOCK_DGRAM) {
2573 err = dev_hard_header(skb, dev, ntohs(proto), addr,
2574 NULL, tp_len);
2575 if (unlikely(err < 0))
2576 return -EINVAL;
Willem de Bruijn1d036d22016-02-03 18:02:17 -05002577 } else if (copylen) {
Willem de Bruijn9ed988c2016-03-09 21:58:34 -05002578 int hdrlen = min_t(int, copylen, tp_len);
2579
Johann Baudy69e3c752009-05-18 22:11:22 -07002580 skb_push(skb, dev->hard_header_len);
Willem de Bruijn1d036d22016-02-03 18:02:17 -05002581 skb_put(skb, copylen - dev->hard_header_len);
Willem de Bruijn9ed988c2016-03-09 21:58:34 -05002582 err = skb_store_bits(skb, 0, data, hdrlen);
Johann Baudy69e3c752009-05-18 22:11:22 -07002583 if (unlikely(err))
2584 return err;
Willem de Bruijn9ed988c2016-03-09 21:58:34 -05002585 if (!dev_validate_header(dev, skb->data, hdrlen))
2586 return -EINVAL;
Johann Baudy69e3c752009-05-18 22:11:22 -07002587
Willem de Bruijn9ed988c2016-03-09 21:58:34 -05002588 data += hdrlen;
2589 to_write -= hdrlen;
Johann Baudy69e3c752009-05-18 22:11:22 -07002590 }
2591
Johann Baudy69e3c752009-05-18 22:11:22 -07002592 offset = offset_in_page(data);
2593 len_max = PAGE_SIZE - offset;
2594 len = ((to_write > len_max) ? len_max : to_write);
2595
2596 skb->data_len = to_write;
2597 skb->len += to_write;
2598 skb->truesize += to_write;
Reshetova, Elena14afee42017-06-30 13:08:00 +03002599 refcount_add(to_write, &po->sk.sk_wmem_alloc);
Johann Baudy69e3c752009-05-18 22:11:22 -07002600
2601 while (likely(to_write)) {
2602 nr_frags = skb_shinfo(skb)->nr_frags;
2603
2604 if (unlikely(nr_frags >= MAX_SKB_FRAGS)) {
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00002605 pr_err("Packet exceed the number of skb frags(%lu)\n",
2606 MAX_SKB_FRAGS);
Johann Baudy69e3c752009-05-18 22:11:22 -07002607 return -EFAULT;
2608 }
2609
Changli Gao0af55bb2010-12-01 02:52:20 +00002610 page = pgv_to_page(data);
2611 data += len;
Johann Baudy69e3c752009-05-18 22:11:22 -07002612 flush_dcache_page(page);
2613 get_page(page);
Changli Gao0af55bb2010-12-01 02:52:20 +00002614 skb_fill_page_desc(skb, nr_frags, page, offset, len);
Johann Baudy69e3c752009-05-18 22:11:22 -07002615 to_write -= len;
2616 offset = 0;
2617 len_max = PAGE_SIZE;
2618 len = ((to_write > len_max) ? len_max : to_write);
2619 }
2620
Maxim Mikityanskiy75c65772019-02-21 12:40:01 +00002621 packet_parse_headers(skb, sock);
Daniel Borkmannefdfa2f2015-11-11 23:25:40 +01002622
Johann Baudy69e3c752009-05-18 22:11:22 -07002623 return tp_len;
2624}
2625
Willem de Bruijn8d39b4a2016-02-03 18:02:16 -05002626static int tpacket_parse_header(struct packet_sock *po, void *frame,
2627 int size_max, void **data)
2628{
2629 union tpacket_uhdr ph;
2630 int tp_len, off;
2631
2632 ph.raw = frame;
2633
2634 switch (po->tp_version) {
Sowmini Varadhan7f953ab2017-01-03 06:31:47 -08002635 case TPACKET_V3:
2636 if (ph.h3->tp_next_offset != 0) {
2637 pr_warn_once("variable sized slot not supported");
2638 return -EINVAL;
2639 }
2640 tp_len = ph.h3->tp_len;
2641 break;
Willem de Bruijn8d39b4a2016-02-03 18:02:16 -05002642 case TPACKET_V2:
2643 tp_len = ph.h2->tp_len;
2644 break;
2645 default:
2646 tp_len = ph.h1->tp_len;
2647 break;
2648 }
2649 if (unlikely(tp_len > size_max)) {
2650 pr_err("packet size is too long (%d > %d)\n", tp_len, size_max);
2651 return -EMSGSIZE;
2652 }
2653
2654 if (unlikely(po->tp_tx_has_off)) {
2655 int off_min, off_max;
2656
2657 off_min = po->tp_hdrlen - sizeof(struct sockaddr_ll);
2658 off_max = po->tx_ring.frame_size - tp_len;
2659 if (po->sk.sk_type == SOCK_DGRAM) {
2660 switch (po->tp_version) {
Sowmini Varadhan7f953ab2017-01-03 06:31:47 -08002661 case TPACKET_V3:
2662 off = ph.h3->tp_net;
2663 break;
Willem de Bruijn8d39b4a2016-02-03 18:02:16 -05002664 case TPACKET_V2:
2665 off = ph.h2->tp_net;
2666 break;
2667 default:
2668 off = ph.h1->tp_net;
2669 break;
2670 }
2671 } else {
2672 switch (po->tp_version) {
Sowmini Varadhan7f953ab2017-01-03 06:31:47 -08002673 case TPACKET_V3:
2674 off = ph.h3->tp_mac;
2675 break;
Willem de Bruijn8d39b4a2016-02-03 18:02:16 -05002676 case TPACKET_V2:
2677 off = ph.h2->tp_mac;
2678 break;
2679 default:
2680 off = ph.h1->tp_mac;
2681 break;
2682 }
2683 }
2684 if (unlikely((off < off_min) || (off_max < off)))
2685 return -EINVAL;
2686 } else {
2687 off = po->tp_hdrlen - sizeof(struct sockaddr_ll);
2688 }
2689
2690 *data = frame + off;
2691 return tp_len;
2692}
2693
Johann Baudy69e3c752009-05-18 22:11:22 -07002694static int tpacket_snd(struct packet_sock *po, struct msghdr *msg)
2695{
Neil Horman89ed5b52019-06-25 17:57:49 -04002696 struct sk_buff *skb = NULL;
Johann Baudy69e3c752009-05-18 22:11:22 -07002697 struct net_device *dev;
Willem de Bruijn1d036d22016-02-03 18:02:17 -05002698 struct virtio_net_hdr *vnet_hdr = NULL;
Soheil Hassas Yeganehc14ac942016-04-02 23:08:12 -04002699 struct sockcm_cookie sockc;
Johann Baudy69e3c752009-05-18 22:11:22 -07002700 __be16 proto;
David S. Miller09effa62013-08-07 17:11:00 -07002701 int err, reserve = 0;
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00002702 void *ph;
Steffen Hurrle342dfc32014-01-17 22:53:15 +01002703 DECLARE_SOCKADDR(struct sockaddr_ll *, saddr, msg->msg_name);
Daniel Borkmann87a2fd22014-01-15 16:25:35 +01002704 bool need_wait = !(msg->msg_flags & MSG_DONTWAIT);
Willem de Bruijn486efdc2019-04-29 11:53:18 -04002705 unsigned char *addr = NULL;
Johann Baudy69e3c752009-05-18 22:11:22 -07002706 int tp_len, size_max;
Willem de Bruijn8d39b4a2016-02-03 18:02:16 -05002707 void *data;
Johann Baudy69e3c752009-05-18 22:11:22 -07002708 int len_sum = 0;
danborkmann@iogearbox.net9e670302012-08-20 03:34:03 +00002709 int status = TP_STATUS_AVAILABLE;
Willem de Bruijn1d036d22016-02-03 18:02:17 -05002710 int hlen, tlen, copylen = 0;
Neil Horman89ed5b52019-06-25 17:57:49 -04002711 long timeo = 0;
Johann Baudy69e3c752009-05-18 22:11:22 -07002712
Johann Baudy69e3c752009-05-18 22:11:22 -07002713 mutex_lock(&po->pg_vec_lock);
2714
Eric Dumazet32d31822019-08-14 02:11:57 -07002715 /* packet_sendmsg() check on tx_ring.pg_vec was lockless,
2716 * we need to confirm it under protection of pg_vec_lock.
2717 */
2718 if (unlikely(!po->tx_ring.pg_vec)) {
2719 err = -EBUSY;
2720 goto out;
2721 }
Daniel Borkmann66e56cd2013-12-06 11:36:15 +01002722 if (likely(saddr == NULL)) {
Daniel Borkmanne40526c2013-11-21 16:50:58 +01002723 dev = packet_cached_dev_get(po);
Eric Dumazetc7d2ef52021-06-16 06:42:01 -07002724 proto = READ_ONCE(po->num);
Johann Baudy69e3c752009-05-18 22:11:22 -07002725 } else {
2726 err = -EINVAL;
2727 if (msg->msg_namelen < sizeof(struct sockaddr_ll))
2728 goto out;
2729 if (msg->msg_namelen < (saddr->sll_halen
2730 + offsetof(struct sockaddr_ll,
2731 sll_addr)))
2732 goto out;
Johann Baudy69e3c752009-05-18 22:11:22 -07002733 proto = saddr->sll_protocol;
Ben Greear827d9782011-06-01 07:18:53 +00002734 dev = dev_get_by_index(sock_net(&po->sk), saddr->sll_ifindex);
Willem de Bruijn486efdc2019-04-29 11:53:18 -04002735 if (po->sk.sk_socket->type == SOCK_DGRAM) {
2736 if (dev && msg->msg_namelen < dev->addr_len +
2737 offsetof(struct sockaddr_ll, sll_addr))
2738 goto out_put;
2739 addr = saddr->sll_addr;
2740 }
Johann Baudy69e3c752009-05-18 22:11:22 -07002741 }
2742
Johann Baudy69e3c752009-05-18 22:11:22 -07002743 err = -ENXIO;
2744 if (unlikely(dev == NULL))
2745 goto out;
Johann Baudy69e3c752009-05-18 22:11:22 -07002746 err = -ENETDOWN;
2747 if (unlikely(!(dev->flags & IFF_UP)))
2748 goto out_put;
2749
Willem de Bruijn657a0662018-07-06 10:12:56 -04002750 sockcm_init(&sockc, &po->sk);
Douglas Caetano dos Santosd19b1832017-05-12 15:19:15 -03002751 if (msg->msg_controllen) {
2752 err = sock_cmsg_send(&po->sk, msg, &sockc);
2753 if (unlikely(err))
2754 goto out_put;
2755 }
2756
Daniel Borkmann5cfb4c82015-11-11 23:25:44 +01002757 if (po->sk.sk_socket->type == SOCK_RAW)
2758 reserve = dev->hard_header_len;
Johann Baudy69e3c752009-05-18 22:11:22 -07002759 size_max = po->tx_ring.frame_size
Gabor Gombasb5dd8842009-10-29 03:19:11 -07002760 - (po->tp_hdrlen - sizeof(struct sockaddr_ll));
Johann Baudy69e3c752009-05-18 22:11:22 -07002761
Willem de Bruijn1d036d22016-02-03 18:02:17 -05002762 if ((size_max > dev->mtu + reserve + VLAN_HLEN) && !po->has_vnet_hdr)
Daniel Borkmann5cfb4c82015-11-11 23:25:44 +01002763 size_max = dev->mtu + reserve + VLAN_HLEN;
David S. Miller09effa62013-08-07 17:11:00 -07002764
Neil Horman89ed5b52019-06-25 17:57:49 -04002765 reinit_completion(&po->skb_completion);
2766
Johann Baudy69e3c752009-05-18 22:11:22 -07002767 do {
2768 ph = packet_current_frame(po, &po->tx_ring,
Daniel Borkmann87a2fd22014-01-15 16:25:35 +01002769 TP_STATUS_SEND_REQUEST);
Johann Baudy69e3c752009-05-18 22:11:22 -07002770 if (unlikely(ph == NULL)) {
Neil Horman89ed5b52019-06-25 17:57:49 -04002771 if (need_wait && skb) {
2772 timeo = sock_sndtimeo(&po->sk, msg->msg_flags & MSG_DONTWAIT);
2773 timeo = wait_for_completion_interruptible_timeout(&po->skb_completion, timeo);
2774 if (timeo <= 0) {
2775 err = !timeo ? -ETIMEDOUT : -ERESTARTSYS;
2776 goto out_put;
2777 }
2778 }
2779 /* check for additional frames */
Johann Baudy69e3c752009-05-18 22:11:22 -07002780 continue;
2781 }
2782
Willem de Bruijn8d39b4a2016-02-03 18:02:16 -05002783 skb = NULL;
2784 tp_len = tpacket_parse_header(po, ph, size_max, &data);
2785 if (tp_len < 0)
2786 goto tpacket_error;
2787
Johann Baudy69e3c752009-05-18 22:11:22 -07002788 status = TP_STATUS_SEND_REQUEST;
Herbert Xuae641942011-11-18 02:20:04 +00002789 hlen = LL_RESERVED_SPACE(dev);
2790 tlen = dev->needed_tailroom;
Willem de Bruijn1d036d22016-02-03 18:02:17 -05002791 if (po->has_vnet_hdr) {
2792 vnet_hdr = data;
2793 data += sizeof(*vnet_hdr);
2794 tp_len -= sizeof(*vnet_hdr);
2795 if (tp_len < 0 ||
2796 __packet_snd_vnet_parse(vnet_hdr, tp_len)) {
2797 tp_len = -EINVAL;
2798 goto tpacket_error;
2799 }
2800 copylen = __virtio16_to_cpu(vio_le(),
2801 vnet_hdr->hdr_len);
2802 }
Willem de Bruijn9ed988c2016-03-09 21:58:34 -05002803 copylen = max_t(int, copylen, dev->hard_header_len);
Johann Baudy69e3c752009-05-18 22:11:22 -07002804 skb = sock_alloc_send_skb(&po->sk,
Willem de Bruijn1d036d22016-02-03 18:02:17 -05002805 hlen + tlen + sizeof(struct sockaddr_ll) +
2806 (copylen - dev->hard_header_len),
Kretschmer, Mathiasfbf33a22015-05-08 15:44:37 +02002807 !need_wait, &err);
Johann Baudy69e3c752009-05-18 22:11:22 -07002808
Kretschmer, Mathiasfbf33a22015-05-08 15:44:37 +02002809 if (unlikely(skb == NULL)) {
2810 /* we assume the socket was initially writeable ... */
2811 if (likely(len_sum > 0))
2812 err = len_sum;
Johann Baudy69e3c752009-05-18 22:11:22 -07002813 goto out_status;
Kretschmer, Mathiasfbf33a22015-05-08 15:44:37 +02002814 }
Willem de Bruijn8d39b4a2016-02-03 18:02:16 -05002815 tp_len = tpacket_fill_skb(po, skb, ph, dev, data, tp_len, proto,
Soheil Hassas Yeganehc14ac942016-04-02 23:08:12 -04002816 addr, hlen, copylen, &sockc);
Alexander Drozdovdbd46ab2015-07-28 13:57:01 +03002817 if (likely(tp_len >= 0) &&
Daniel Borkmann5cfb4c82015-11-11 23:25:44 +01002818 tp_len > dev->mtu + reserve &&
Willem de Bruijn1d036d22016-02-03 18:02:17 -05002819 !po->has_vnet_hdr &&
Daniel Borkmann3c70c132015-11-11 23:25:42 +01002820 !packet_extra_vlan_len_allowed(dev, skb))
2821 tp_len = -EMSGSIZE;
Johann Baudy69e3c752009-05-18 22:11:22 -07002822
2823 if (unlikely(tp_len < 0)) {
Willem de Bruijn8d39b4a2016-02-03 18:02:16 -05002824tpacket_error:
Johann Baudy69e3c752009-05-18 22:11:22 -07002825 if (po->tp_loss) {
2826 __packet_set_status(po, ph,
2827 TP_STATUS_AVAILABLE);
2828 packet_increment_head(&po->tx_ring);
2829 kfree_skb(skb);
2830 continue;
2831 } else {
2832 status = TP_STATUS_WRONG_FORMAT;
2833 err = tp_len;
2834 goto out_status;
2835 }
2836 }
2837
Jianfeng Tan9d2f67e2018-09-29 15:41:27 +00002838 if (po->has_vnet_hdr) {
2839 if (virtio_net_hdr_to_skb(skb, vnet_hdr, vio_le())) {
2840 tp_len = -EINVAL;
2841 goto tpacket_error;
2842 }
2843 virtio_net_hdr_set_proto(skb, vnet_hdr);
Willem de Bruijn1d036d22016-02-03 18:02:17 -05002844 }
2845
Johann Baudy69e3c752009-05-18 22:11:22 -07002846 skb->destructor = tpacket_destruct_skb;
2847 __packet_set_status(po, ph, TP_STATUS_SENDING);
Daniel Borkmannb0138402014-01-15 16:25:36 +01002848 packet_inc_pending(&po->tx_ring);
Johann Baudy69e3c752009-05-18 22:11:22 -07002849
2850 status = TP_STATUS_SEND_REQUEST;
Daniel Borkmannd346a3f2013-12-06 11:36:17 +01002851 err = po->xmit(skb);
Jarek Poplawskieb70df12010-01-10 22:04:19 +00002852 if (unlikely(err > 0)) {
2853 err = net_xmit_errno(err);
2854 if (err && __packet_get_status(po, ph) ==
2855 TP_STATUS_AVAILABLE) {
2856 /* skb was destructed already */
2857 skb = NULL;
2858 goto out_status;
2859 }
2860 /*
2861 * skb was dropped but not destructed yet;
2862 * let's treat it like congestion or err < 0
2863 */
2864 err = 0;
2865 }
Johann Baudy69e3c752009-05-18 22:11:22 -07002866 packet_increment_head(&po->tx_ring);
2867 len_sum += tp_len;
Daniel Borkmannb0138402014-01-15 16:25:36 +01002868 } while (likely((ph != NULL) ||
2869 /* Note: packet_read_pending() might be slow if we have
2870 * to call it as it's per_cpu variable, but in fast-path
2871 * we already short-circuit the loop with the first
2872 * condition, and luckily don't have to go that path
2873 * anyway.
2874 */
2875 (need_wait && packet_read_pending(&po->tx_ring))));
Johann Baudy69e3c752009-05-18 22:11:22 -07002876
2877 err = len_sum;
2878 goto out_put;
2879
Johann Baudy69e3c752009-05-18 22:11:22 -07002880out_status:
2881 __packet_set_status(po, ph, status);
2882 kfree_skb(skb);
2883out_put:
Daniel Borkmanne40526c2013-11-21 16:50:58 +01002884 dev_put(dev);
Johann Baudy69e3c752009-05-18 22:11:22 -07002885out:
2886 mutex_unlock(&po->pg_vec_lock);
2887 return err;
2888}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002889
Olof Johanssoneea49cc92011-11-02 11:00:49 +00002890static struct sk_buff *packet_alloc_skb(struct sock *sk, size_t prepad,
2891 size_t reserve, size_t len,
2892 size_t linear, int noblock,
2893 int *err)
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08002894{
2895 struct sk_buff *skb;
2896
2897 /* Under a page? Don't bother with paged skb. */
2898 if (prepad + len < PAGE_SIZE || !linear)
2899 linear = len;
2900
2901 skb = sock_alloc_send_pskb(sk, prepad + linear, len - linear, noblock,
Eric Dumazet28d64272013-08-08 14:38:47 -07002902 err, 0);
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08002903 if (!skb)
2904 return NULL;
2905
2906 skb_reserve(skb, reserve);
2907 skb_put(skb, linear);
2908 skb->data_len = len - linear;
2909 skb->len += len - linear;
2910
2911 return skb;
2912}
2913
Daniel Borkmannd346a3f2013-12-06 11:36:17 +01002914static int packet_snd(struct socket *sock, struct msghdr *msg, size_t len)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002915{
2916 struct sock *sk = sock->sk;
Steffen Hurrle342dfc32014-01-17 22:53:15 +01002917 DECLARE_SOCKADDR(struct sockaddr_ll *, saddr, msg->msg_name);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002918 struct sk_buff *skb;
2919 struct net_device *dev;
Al Viro0e11c912006-11-08 00:26:29 -08002920 __be16 proto;
Willem de Bruijn486efdc2019-04-29 11:53:18 -04002921 unsigned char *addr = NULL;
Ben Greear827d9782011-06-01 07:18:53 +00002922 int err, reserve = 0;
Edward Jeec7d39e32015-10-08 14:56:49 -07002923 struct sockcm_cookie sockc;
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08002924 struct virtio_net_hdr vnet_hdr = { 0 };
2925 int offset = 0;
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08002926 struct packet_sock *po = pkt_sk(sk);
Willem de Bruijnda7c9562017-09-26 12:20:17 -04002927 bool has_vnet_hdr = false;
Willem de Bruijn57031eb2017-02-07 15:57:21 -05002928 int hlen, tlen, linear;
Ben Greear3bdc0eb2012-02-11 15:39:30 +00002929 int extra_len = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002930
2931 /*
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09002932 * Get and verify the address.
Linus Torvalds1da177e2005-04-16 15:20:36 -07002933 */
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09002934
Daniel Borkmann66e56cd2013-12-06 11:36:15 +01002935 if (likely(saddr == NULL)) {
Daniel Borkmanne40526c2013-11-21 16:50:58 +01002936 dev = packet_cached_dev_get(po);
Eric Dumazetc7d2ef52021-06-16 06:42:01 -07002937 proto = READ_ONCE(po->num);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002938 } else {
2939 err = -EINVAL;
2940 if (msg->msg_namelen < sizeof(struct sockaddr_ll))
2941 goto out;
Eric W. Biederman0fb375f2005-09-21 00:11:37 -07002942 if (msg->msg_namelen < (saddr->sll_halen + offsetof(struct sockaddr_ll, sll_addr)))
2943 goto out;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002944 proto = saddr->sll_protocol;
Ben Greear827d9782011-06-01 07:18:53 +00002945 dev = dev_get_by_index(sock_net(sk), saddr->sll_ifindex);
Willem de Bruijn486efdc2019-04-29 11:53:18 -04002946 if (sock->type == SOCK_DGRAM) {
2947 if (dev && msg->msg_namelen < dev->addr_len +
2948 offsetof(struct sockaddr_ll, sll_addr))
2949 goto out_unlock;
2950 addr = saddr->sll_addr;
2951 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002952 }
2953
Linus Torvalds1da177e2005-04-16 15:20:36 -07002954 err = -ENXIO;
Daniel Borkmanne40526c2013-11-21 16:50:58 +01002955 if (unlikely(dev == NULL))
Linus Torvalds1da177e2005-04-16 15:20:36 -07002956 goto out_unlock;
Daniel Borkmanne40526c2013-11-21 16:50:58 +01002957 err = -ENETDOWN;
2958 if (unlikely(!(dev->flags & IFF_UP)))
2959 goto out_unlock;
2960
Willem de Bruijn657a0662018-07-06 10:12:56 -04002961 sockcm_init(&sockc, sk);
Edward Jeec7d39e32015-10-08 14:56:49 -07002962 sockc.mark = sk->sk_mark;
2963 if (msg->msg_controllen) {
2964 err = sock_cmsg_send(sk, msg, &sockc);
2965 if (unlikely(err))
2966 goto out_unlock;
2967 }
2968
Linus Torvalds1da177e2005-04-16 15:20:36 -07002969 if (sock->type == SOCK_RAW)
2970 reserve = dev->hard_header_len;
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08002971 if (po->has_vnet_hdr) {
Willem de Bruijn16cc1402016-02-03 18:02:14 -05002972 err = packet_snd_vnet_parse(msg, &len, &vnet_hdr);
2973 if (err)
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08002974 goto out_unlock;
Willem de Bruijnda7c9562017-09-26 12:20:17 -04002975 has_vnet_hdr = true;
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08002976 }
2977
Ben Greear3bdc0eb2012-02-11 15:39:30 +00002978 if (unlikely(sock_flag(sk, SOCK_NOFCS))) {
2979 if (!netif_supports_nofcs(dev)) {
2980 err = -EPROTONOSUPPORT;
2981 goto out_unlock;
2982 }
2983 extra_len = 4; /* We're doing our own CRC */
2984 }
2985
Linus Torvalds1da177e2005-04-16 15:20:36 -07002986 err = -EMSGSIZE;
Willem de Bruijn16cc1402016-02-03 18:02:14 -05002987 if (!vnet_hdr.gso_type &&
2988 (len > dev->mtu + reserve + VLAN_HLEN + extra_len))
Linus Torvalds1da177e2005-04-16 15:20:36 -07002989 goto out_unlock;
2990
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08002991 err = -ENOBUFS;
Herbert Xuae641942011-11-18 02:20:04 +00002992 hlen = LL_RESERVED_SPACE(dev);
2993 tlen = dev->needed_tailroom;
Willem de Bruijn57031eb2017-02-07 15:57:21 -05002994 linear = __virtio16_to_cpu(vio_le(), vnet_hdr.hdr_len);
2995 linear = max(linear, min_t(int, len, dev->hard_header_len));
2996 skb = packet_alloc_skb(sk, hlen + tlen, hlen, len, linear,
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08002997 msg->msg_flags & MSG_DONTWAIT, &err);
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00002998 if (skb == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002999 goto out_unlock;
3000
Willem de Bruijnb84bbaf2018-05-11 13:24:25 -04003001 skb_reset_network_header(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003002
Stephen Hemminger0c4e8582007-10-09 01:36:32 -07003003 err = -EINVAL;
Willem de Bruijn9c707762014-11-19 13:10:16 -05003004 if (sock->type == SOCK_DGRAM) {
3005 offset = dev_hard_header(skb, dev, ntohs(proto), addr, NULL, len);
Christoph Jaeger46d2cfb2015-01-11 13:01:16 -05003006 if (unlikely(offset < 0))
Willem de Bruijn9c707762014-11-19 13:10:16 -05003007 goto out_free;
Willem de Bruijnb84bbaf2018-05-11 13:24:25 -04003008 } else if (reserve) {
Willem de Bruijn9aad13b2018-05-24 18:10:30 -04003009 skb_reserve(skb, -reserve);
Nicolas Dichtel88a81212019-01-17 11:27:22 +01003010 if (len < reserve + sizeof(struct ipv6hdr) &&
3011 dev->min_header_len != dev->hard_header_len)
Willem de Bruijn993675a2018-07-11 12:00:45 -04003012 skb_reset_network_header(skb);
Willem de Bruijn9c707762014-11-19 13:10:16 -05003013 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07003014
3015 /* Returns -EFAULT on error */
Al Viroc0371da2014-11-24 10:42:55 -05003016 err = skb_copy_datagram_from_iter(skb, offset, &msg->msg_iter, len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003017 if (err)
3018 goto out_free;
Daniel Borkmannbf84a0102013-04-14 08:08:13 +00003019
Willem de Bruijn9ed988c2016-03-09 21:58:34 -05003020 if (sock->type == SOCK_RAW &&
3021 !dev_validate_header(dev, skb->data, len)) {
3022 err = -EINVAL;
3023 goto out_free;
3024 }
3025
Willem de Bruijn8f932f72018-12-17 12:24:00 -05003026 skb_setup_tx_timestamp(skb, sockc.tsflags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003027
Willem de Bruijn16cc1402016-02-03 18:02:14 -05003028 if (!vnet_hdr.gso_type && (len > dev->mtu + reserve + extra_len) &&
Daniel Borkmann3c70c132015-11-11 23:25:42 +01003029 !packet_extra_vlan_len_allowed(dev, skb)) {
3030 err = -EMSGSIZE;
3031 goto out_free;
Ben Greear57f89bf2011-02-11 09:35:18 +00003032 }
3033
David S. Miller09effa62013-08-07 17:11:00 -07003034 skb->protocol = proto;
3035 skb->dev = dev;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003036 skb->priority = sk->sk_priority;
Edward Jeec7d39e32015-10-08 14:56:49 -07003037 skb->mark = sockc.mark;
Richard Cochran3d0ba8c2018-07-03 15:42:51 -07003038 skb->tstamp = sockc.transmit_time;
Daniel Borkmann0fd5d572014-02-16 15:55:22 +01003039
Willem de Bruijnda7c9562017-09-26 12:20:17 -04003040 if (has_vnet_hdr) {
Jarno Rajahalmedb60eb5f2016-11-18 15:40:41 -08003041 err = virtio_net_hdr_to_skb(skb, &vnet_hdr, vio_le());
Willem de Bruijn16cc1402016-02-03 18:02:14 -05003042 if (err)
3043 goto out_free;
3044 len += sizeof(vnet_hdr);
Jianfeng Tan9d2f67e2018-09-29 15:41:27 +00003045 virtio_net_hdr_set_proto(skb, &vnet_hdr);
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003046 }
3047
Maxim Mikityanskiy75c65772019-02-21 12:40:01 +00003048 packet_parse_headers(skb, sock);
Daniel Borkmann8fd6c802015-11-11 23:25:41 +01003049
Ben Greear3bdc0eb2012-02-11 15:39:30 +00003050 if (unlikely(extra_len == 4))
3051 skb->no_fcs = 1;
3052
Daniel Borkmannd346a3f2013-12-06 11:36:17 +01003053 err = po->xmit(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003054 if (err > 0 && (err = net_xmit_errno(err)) != 0)
3055 goto out_unlock;
3056
Daniel Borkmanne40526c2013-11-21 16:50:58 +01003057 dev_put(dev);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003058
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003059 return len;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003060
3061out_free:
3062 kfree_skb(skb);
3063out_unlock:
Yajun Deng1160dfa2021-08-05 19:55:27 +08003064 dev_put(dev);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003065out:
3066 return err;
3067}
3068
Ying Xue1b784142015-03-02 15:37:48 +08003069static int packet_sendmsg(struct socket *sock, struct msghdr *msg, size_t len)
Johann Baudy69e3c752009-05-18 22:11:22 -07003070{
Johann Baudy69e3c752009-05-18 22:11:22 -07003071 struct sock *sk = sock->sk;
3072 struct packet_sock *po = pkt_sk(sk);
Daniel Borkmannd346a3f2013-12-06 11:36:17 +01003073
Eric Dumazetd1b5bee2021-06-10 09:00:12 -07003074 /* Reading tx_ring.pg_vec without holding pg_vec_lock is racy.
3075 * tpacket_snd() will redo the check safely.
3076 */
3077 if (data_race(po->tx_ring.pg_vec))
Johann Baudy69e3c752009-05-18 22:11:22 -07003078 return tpacket_snd(po, msg);
Eric Dumazetd1b5bee2021-06-10 09:00:12 -07003079
3080 return packet_snd(sock, msg, len);
Johann Baudy69e3c752009-05-18 22:11:22 -07003081}
3082
Linus Torvalds1da177e2005-04-16 15:20:36 -07003083/*
3084 * Close a PACKET socket. This is fairly simple. We immediately go
3085 * to 'closed' state and remove our protocol entry in the device list.
3086 */
3087
3088static int packet_release(struct socket *sock)
3089{
3090 struct sock *sk = sock->sk;
3091 struct packet_sock *po;
Anoob Soman2bd624b2017-02-15 20:25:39 +00003092 struct packet_fanout *f;
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08003093 struct net *net;
chetan lokef6fb8f12011-08-19 10:18:16 +00003094 union tpacket_req_u req_u;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003095
3096 if (!sk)
3097 return 0;
3098
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09003099 net = sock_net(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003100 po = pkt_sk(sk);
3101
Pavel Emelyanov0fa7fa92012-08-21 01:06:47 +00003102 mutex_lock(&net->packet.sklist_lock);
stephen hemminger808f5112010-02-22 07:57:18 +00003103 sk_del_node_init_rcu(sk);
Pavel Emelyanov0fa7fa92012-08-21 01:06:47 +00003104 mutex_unlock(&net->packet.sklist_lock);
3105
Eric Dumazet920de802008-11-24 00:09:29 -08003106 sock_prot_inuse_add(net, sk->sk_prot, -1);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003107
stephen hemminger808f5112010-02-22 07:57:18 +00003108 spin_lock(&po->bind_lock);
David S. Millerce06b032011-07-04 01:44:29 -07003109 unregister_prot_hook(sk, false);
Daniel Borkmann66e56cd2013-12-06 11:36:15 +01003110 packet_cached_dev_reset(po);
3111
Ben Greear160ff182011-06-01 07:18:52 +00003112 if (po->prot_hook.dev) {
3113 dev_put(po->prot_hook.dev);
3114 po->prot_hook.dev = NULL;
3115 }
stephen hemminger808f5112010-02-22 07:57:18 +00003116 spin_unlock(&po->bind_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003117
Linus Torvalds1da177e2005-04-16 15:20:36 -07003118 packet_flush_mclist(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003119
Eric Dumazet5171b372018-04-15 17:52:04 -07003120 lock_sock(sk);
Phil Sutter9665d5d2013-02-01 07:21:41 +00003121 if (po->rx_ring.pg_vec) {
3122 memset(&req_u, 0, sizeof(req_u));
chetan lokef6fb8f12011-08-19 10:18:16 +00003123 packet_set_ring(sk, &req_u, 1, 0);
Phil Sutter9665d5d2013-02-01 07:21:41 +00003124 }
Johann Baudy69e3c752009-05-18 22:11:22 -07003125
Phil Sutter9665d5d2013-02-01 07:21:41 +00003126 if (po->tx_ring.pg_vec) {
3127 memset(&req_u, 0, sizeof(req_u));
chetan lokef6fb8f12011-08-19 10:18:16 +00003128 packet_set_ring(sk, &req_u, 1, 1);
Phil Sutter9665d5d2013-02-01 07:21:41 +00003129 }
Eric Dumazet5171b372018-04-15 17:52:04 -07003130 release_sock(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003131
Anoob Soman2bd624b2017-02-15 20:25:39 +00003132 f = fanout_release(sk);
David S. Millerdc99f602011-07-05 01:45:05 -07003133
stephen hemminger808f5112010-02-22 07:57:18 +00003134 synchronize_net();
Anoob Soman2bd624b2017-02-15 20:25:39 +00003135
Willem de Bruijnafa09252019-05-31 12:37:23 -04003136 kfree(po->rollover);
Anoob Soman2bd624b2017-02-15 20:25:39 +00003137 if (f) {
3138 fanout_release_data(f);
Tanner Love9c661b02020-11-06 13:07:40 -05003139 kvfree(f);
Anoob Soman2bd624b2017-02-15 20:25:39 +00003140 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07003141 /*
3142 * Now the socket is dead. No more input will appear.
3143 */
Linus Torvalds1da177e2005-04-16 15:20:36 -07003144 sock_orphan(sk);
3145 sock->sk = NULL;
3146
3147 /* Purge queues */
3148
3149 skb_queue_purge(&sk->sk_receive_queue);
Daniel Borkmannb0138402014-01-15 16:25:36 +01003150 packet_free_pending(po);
Pavel Emelyanov17ab56a2007-11-10 21:38:48 -08003151 sk_refcnt_debug_release(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003152
3153 sock_put(sk);
3154 return 0;
3155}
3156
3157/*
3158 * Attach a packet hook.
3159 */
3160
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003161static int packet_do_bind(struct sock *sk, const char *name, int ifindex,
3162 __be16 proto)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003163{
3164 struct packet_sock *po = pkt_sk(sk);
Lars Westerhoff158cd4a2015-07-28 01:32:21 +03003165 struct net_device *dev_curr;
Daniel Borkmann902fefb2014-01-15 16:25:34 +01003166 __be16 proto_curr;
3167 bool need_rehook;
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003168 struct net_device *dev = NULL;
3169 int ret = 0;
3170 bool unlisted = false;
David S. Millerdc99f602011-07-05 01:45:05 -07003171
Linus Torvalds1da177e2005-04-16 15:20:36 -07003172 lock_sock(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003173 spin_lock(&po->bind_lock);
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003174 rcu_read_lock();
3175
Willem de Bruijn49716132017-09-26 12:19:37 -04003176 if (po->fanout) {
3177 ret = -EINVAL;
3178 goto out_unlock;
3179 }
3180
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003181 if (name) {
3182 dev = dev_get_by_name_rcu(sock_net(sk), name);
3183 if (!dev) {
3184 ret = -ENODEV;
3185 goto out_unlock;
3186 }
3187 } else if (ifindex) {
3188 dev = dev_get_by_index_rcu(sock_net(sk), ifindex);
3189 if (!dev) {
3190 ret = -ENODEV;
3191 goto out_unlock;
3192 }
3193 }
3194
Yajun Deng1160dfa2021-08-05 19:55:27 +08003195 dev_hold(dev);
Daniel Borkmann66e56cd2013-12-06 11:36:15 +01003196
Daniel Borkmann902fefb2014-01-15 16:25:34 +01003197 proto_curr = po->prot_hook.type;
3198 dev_curr = po->prot_hook.dev;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003199
Daniel Borkmann902fefb2014-01-15 16:25:34 +01003200 need_rehook = proto_curr != proto || dev_curr != dev;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003201
Daniel Borkmann902fefb2014-01-15 16:25:34 +01003202 if (need_rehook) {
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003203 if (po->running) {
3204 rcu_read_unlock();
Eric Dumazet15fe0762017-11-28 08:03:30 -08003205 /* prevents packet_notifier() from calling
3206 * register_prot_hook()
3207 */
Eric Dumazetc7d2ef52021-06-16 06:42:01 -07003208 WRITE_ONCE(po->num, 0);
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003209 __unregister_prot_hook(sk, true);
3210 rcu_read_lock();
3211 dev_curr = po->prot_hook.dev;
3212 if (dev)
3213 unlisted = !dev_get_by_index_rcu(sock_net(sk),
3214 dev->ifindex);
3215 }
Daniel Borkmann66e56cd2013-12-06 11:36:15 +01003216
Eric Dumazet15fe0762017-11-28 08:03:30 -08003217 BUG_ON(po->running);
Eric Dumazetc7d2ef52021-06-16 06:42:01 -07003218 WRITE_ONCE(po->num, proto);
Daniel Borkmann902fefb2014-01-15 16:25:34 +01003219 po->prot_hook.type = proto;
Daniel Borkmann902fefb2014-01-15 16:25:34 +01003220
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003221 if (unlikely(unlisted)) {
3222 dev_put(dev);
3223 po->prot_hook.dev = NULL;
Eric Dumazete032f7c2021-06-16 06:42:02 -07003224 WRITE_ONCE(po->ifindex, -1);
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003225 packet_cached_dev_reset(po);
3226 } else {
3227 po->prot_hook.dev = dev;
Eric Dumazete032f7c2021-06-16 06:42:02 -07003228 WRITE_ONCE(po->ifindex, dev ? dev->ifindex : 0);
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003229 packet_cached_dev_assign(po, dev);
3230 }
Daniel Borkmann902fefb2014-01-15 16:25:34 +01003231 }
Yajun Deng1160dfa2021-08-05 19:55:27 +08003232 dev_put(dev_curr);
Daniel Borkmann902fefb2014-01-15 16:25:34 +01003233
3234 if (proto == 0 || !need_rehook)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003235 goto out_unlock;
3236
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003237 if (!unlisted && (!dev || (dev->flags & IFF_UP))) {
David S. Millerce06b032011-07-04 01:44:29 -07003238 register_prot_hook(sk);
Urs Thuermannbe85d4a2007-11-12 21:05:20 -08003239 } else {
3240 sk->sk_err = ENETDOWN;
3241 if (!sock_flag(sk, SOCK_DEAD))
Alexander Aringe3ae2362021-06-27 18:48:21 -04003242 sk_error_report(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003243 }
3244
3245out_unlock:
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003246 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07003247 spin_unlock(&po->bind_lock);
3248 release_sock(sk);
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003249 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003250}
3251
3252/*
3253 * Bind a packet socket to a device
3254 */
3255
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003256static int packet_bind_spkt(struct socket *sock, struct sockaddr *uaddr,
3257 int addr_len)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003258{
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003259 struct sock *sk = sock->sk;
Alexander Potapenko540e2892017-03-01 12:57:20 +01003260 char name[sizeof(uaddr->sa_data) + 1];
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09003261
Linus Torvalds1da177e2005-04-16 15:20:36 -07003262 /*
3263 * Check legality
3264 */
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09003265
Kris Katterjohn8ae55f02006-01-23 16:28:02 -08003266 if (addr_len != sizeof(struct sockaddr))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003267 return -EINVAL;
Alexander Potapenko540e2892017-03-01 12:57:20 +01003268 /* uaddr->sa_data comes from the userspace, it's not guaranteed to be
3269 * zero-terminated.
3270 */
3271 memcpy(name, uaddr->sa_data, sizeof(uaddr->sa_data));
3272 name[sizeof(uaddr->sa_data)] = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003273
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003274 return packet_do_bind(sk, name, 0, pkt_sk(sk)->num);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003275}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003276
3277static int packet_bind(struct socket *sock, struct sockaddr *uaddr, int addr_len)
3278{
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003279 struct sockaddr_ll *sll = (struct sockaddr_ll *)uaddr;
3280 struct sock *sk = sock->sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003281
3282 /*
3283 * Check legality
3284 */
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09003285
Linus Torvalds1da177e2005-04-16 15:20:36 -07003286 if (addr_len < sizeof(struct sockaddr_ll))
3287 return -EINVAL;
3288 if (sll->sll_family != AF_PACKET)
3289 return -EINVAL;
3290
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003291 return packet_do_bind(sk, NULL, sll->sll_ifindex,
3292 sll->sll_protocol ? : pkt_sk(sk)->num);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003293}
3294
3295static struct proto packet_proto = {
3296 .name = "PACKET",
3297 .owner = THIS_MODULE,
3298 .obj_size = sizeof(struct packet_sock),
3299};
3300
3301/*
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09003302 * Create a packet of type SOCK_PACKET.
Linus Torvalds1da177e2005-04-16 15:20:36 -07003303 */
3304
Eric Paris3f378b62009-11-05 22:18:14 -08003305static int packet_create(struct net *net, struct socket *sock, int protocol,
3306 int kern)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003307{
3308 struct sock *sk;
3309 struct packet_sock *po;
Al Viro0e11c912006-11-08 00:26:29 -08003310 __be16 proto = (__force __be16)protocol; /* weird, but documented */
Linus Torvalds1da177e2005-04-16 15:20:36 -07003311 int err;
3312
Eric W. Biedermandf008c92012-11-16 03:03:07 +00003313 if (!ns_capable(net->user_ns, CAP_NET_RAW))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003314 return -EPERM;
David S. Millerbe020972007-05-29 13:16:31 -07003315 if (sock->type != SOCK_DGRAM && sock->type != SOCK_RAW &&
3316 sock->type != SOCK_PACKET)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003317 return -ESOCKTNOSUPPORT;
3318
3319 sock->state = SS_UNCONNECTED;
3320
3321 err = -ENOBUFS;
Eric W. Biederman11aa9c22015-05-08 21:09:13 -05003322 sk = sk_alloc(net, PF_PACKET, GFP_KERNEL, &packet_proto, kern);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003323 if (sk == NULL)
3324 goto out;
3325
3326 sock->ops = &packet_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003327 if (sock->type == SOCK_PACKET)
3328 sock->ops = &packet_ops_spkt;
David S. Millerbe020972007-05-29 13:16:31 -07003329
Linus Torvalds1da177e2005-04-16 15:20:36 -07003330 sock_init_data(sock, sk);
3331
3332 po = pkt_sk(sk);
Neil Horman89ed5b52019-06-25 17:57:49 -04003333 init_completion(&po->skb_completion);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003334 sk->sk_family = PF_PACKET;
Al Viro0e11c912006-11-08 00:26:29 -08003335 po->num = proto;
Daniel Borkmannd346a3f2013-12-06 11:36:17 +01003336 po->xmit = dev_queue_xmit;
Daniel Borkmann66e56cd2013-12-06 11:36:15 +01003337
Daniel Borkmannb0138402014-01-15 16:25:36 +01003338 err = packet_alloc_pending(po);
3339 if (err)
3340 goto out2;
3341
Daniel Borkmann66e56cd2013-12-06 11:36:15 +01003342 packet_cached_dev_reset(po);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003343
3344 sk->sk_destruct = packet_sock_destruct;
Pavel Emelyanov17ab56a2007-11-10 21:38:48 -08003345 sk_refcnt_debug_inc(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003346
3347 /*
3348 * Attach a protocol block
3349 */
3350
3351 spin_lock_init(&po->bind_lock);
Herbert Xu905db442009-01-30 14:12:06 -08003352 mutex_init(&po->pg_vec_lock);
Willem de Bruijn0648ab72015-05-12 11:56:46 -04003353 po->rollover = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003354 po->prot_hook.func = packet_rcv;
David S. Millerbe020972007-05-29 13:16:31 -07003355
Linus Torvalds1da177e2005-04-16 15:20:36 -07003356 if (sock->type == SOCK_PACKET)
3357 po->prot_hook.func = packet_rcv_spkt;
David S. Millerbe020972007-05-29 13:16:31 -07003358
Linus Torvalds1da177e2005-04-16 15:20:36 -07003359 po->prot_hook.af_packet_priv = sk;
3360
Al Viro0e11c912006-11-08 00:26:29 -08003361 if (proto) {
3362 po->prot_hook.type = proto;
Willem de Bruijna6361f02018-04-23 17:37:03 -04003363 __register_prot_hook(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003364 }
3365
Pavel Emelyanov0fa7fa92012-08-21 01:06:47 +00003366 mutex_lock(&net->packet.sklist_lock);
Maxime Chevalliera4dc6a42019-03-16 14:41:30 +01003367 sk_add_node_tail_rcu(sk, &net->packet.sklist);
Pavel Emelyanov0fa7fa92012-08-21 01:06:47 +00003368 mutex_unlock(&net->packet.sklist_lock);
3369
Eric Dumazet36804532008-11-19 14:25:35 -08003370 sock_prot_inuse_add(net, &packet_proto, 1);
stephen hemminger808f5112010-02-22 07:57:18 +00003371
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003372 return 0;
Daniel Borkmannb0138402014-01-15 16:25:36 +01003373out2:
3374 sk_free(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003375out:
3376 return err;
3377}
3378
3379/*
3380 * Pull a packet from our receive queue and hand it to the user.
3381 * If necessary we block.
3382 */
3383
Ying Xue1b784142015-03-02 15:37:48 +08003384static int packet_recvmsg(struct socket *sock, struct msghdr *msg, size_t len,
3385 int flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003386{
3387 struct sock *sk = sock->sk;
3388 struct sk_buff *skb;
3389 int copied, err;
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003390 int vnet_hdr_len = 0;
Eyal Birger2472d762015-03-01 14:58:28 +02003391 unsigned int origlen = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003392
3393 err = -EINVAL;
Richard Cochraned85b562010-04-07 22:41:28 +00003394 if (flags & ~(MSG_PEEK|MSG_DONTWAIT|MSG_TRUNC|MSG_CMSG_COMPAT|MSG_ERRQUEUE))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003395 goto out;
3396
3397#if 0
3398 /* What error should we return now? EUNATTACH? */
3399 if (pkt_sk(sk)->ifindex < 0)
3400 return -ENODEV;
3401#endif
3402
Richard Cochraned85b562010-04-07 22:41:28 +00003403 if (flags & MSG_ERRQUEUE) {
Richard Cochrancb820f82013-07-19 19:40:09 +02003404 err = sock_recv_errqueue(sk, msg, len,
3405 SOL_PACKET, PACKET_TX_TIMESTAMP);
Richard Cochraned85b562010-04-07 22:41:28 +00003406 goto out;
3407 }
3408
Linus Torvalds1da177e2005-04-16 15:20:36 -07003409 /*
Linus Torvalds1da177e2005-04-16 15:20:36 -07003410 * Call the generic datagram receiver. This handles all sorts
3411 * of horrible races and re-entrancy so we can forget about it
3412 * in the protocol layers.
3413 *
3414 * Now it will return ENETDOWN, if device have just gone down,
3415 * but then it will block.
3416 */
3417
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003418 skb = skb_recv_datagram(sk, flags, flags & MSG_DONTWAIT, &err);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003419
3420 /*
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09003421 * An error occurred so return it. Because skb_recv_datagram()
Linus Torvalds1da177e2005-04-16 15:20:36 -07003422 * handles the blocking we don't see and worry about blocking
3423 * retries.
3424 */
3425
Kris Katterjohn8ae55f02006-01-23 16:28:02 -08003426 if (skb == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003427 goto out;
3428
Eric Dumazet9bb6cd62019-06-12 09:52:33 -07003429 packet_rcv_try_clear_pressure(pkt_sk(sk));
Willem de Bruijn2ccdbaa2015-05-12 11:56:48 -04003430
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003431 if (pkt_sk(sk)->has_vnet_hdr) {
Willem de Bruijn16cc1402016-02-03 18:02:14 -05003432 err = packet_rcv_vnet(msg, skb, &len);
3433 if (err)
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003434 goto out_free;
Willem de Bruijn16cc1402016-02-03 18:02:14 -05003435 vnet_hdr_len = sizeof(struct virtio_net_hdr);
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003436 }
3437
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003438 /* You lose any data beyond the buffer you gave. If it worries
3439 * a user program they can ask the device for its MTU
3440 * anyway.
Eric W. Biederman0fb375f2005-09-21 00:11:37 -07003441 */
Linus Torvalds1da177e2005-04-16 15:20:36 -07003442 copied = skb->len;
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003443 if (copied > len) {
3444 copied = len;
3445 msg->msg_flags |= MSG_TRUNC;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003446 }
3447
David S. Miller51f3d022014-11-05 16:46:40 -05003448 err = skb_copy_datagram_msg(skb, 0, msg, copied);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003449 if (err)
3450 goto out_free;
3451
Eyal Birger2472d762015-03-01 14:58:28 +02003452 if (sock->type != SOCK_PACKET) {
3453 struct sockaddr_ll *sll = &PACKET_SKB_CB(skb)->sa.ll;
3454
3455 /* Original length was stored in sockaddr_ll fields */
3456 origlen = PACKET_SKB_CB(skb)->sa.origlen;
3457 sll->sll_family = AF_PACKET;
3458 sll->sll_protocol = skb->protocol;
3459 }
3460
Neil Horman3b885782009-10-12 13:26:31 -07003461 sock_recv_ts_and_drops(msg, sk, skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003462
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003463 if (msg->msg_name) {
Willem de Bruijnb2cf86e2019-04-29 11:46:55 -04003464 int copy_len;
3465
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003466 /* If the address length field is there to be filled
3467 * in, we fill it in now.
3468 */
3469 if (sock->type == SOCK_PACKET) {
Steffen Hurrle342dfc32014-01-17 22:53:15 +01003470 __sockaddr_check_size(sizeof(struct sockaddr_pkt));
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003471 msg->msg_namelen = sizeof(struct sockaddr_pkt);
Willem de Bruijnb2cf86e2019-04-29 11:46:55 -04003472 copy_len = msg->msg_namelen;
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003473 } else {
3474 struct sockaddr_ll *sll = &PACKET_SKB_CB(skb)->sa.ll;
Eyal Birger2472d762015-03-01 14:58:28 +02003475
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003476 msg->msg_namelen = sll->sll_halen +
3477 offsetof(struct sockaddr_ll, sll_addr);
Willem de Bruijnb2cf86e2019-04-29 11:46:55 -04003478 copy_len = msg->msg_namelen;
3479 if (msg->msg_namelen < sizeof(struct sockaddr_ll)) {
3480 memset(msg->msg_name +
3481 offsetof(struct sockaddr_ll, sll_addr),
3482 0, sizeof(sll->sll_addr));
3483 msg->msg_namelen = sizeof(struct sockaddr_ll);
3484 }
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003485 }
Willem de Bruijnb2cf86e2019-04-29 11:46:55 -04003486 memcpy(msg->msg_name, &PACKET_SKB_CB(skb)->sa, copy_len);
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003487 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07003488
Herbert Xu8dc41942007-02-04 23:31:32 -08003489 if (pkt_sk(sk)->auxdata) {
Herbert Xuffbc6112007-02-04 23:33:10 -08003490 struct tpacket_auxdata aux;
3491
3492 aux.tp_status = TP_STATUS_USER;
3493 if (skb->ip_summed == CHECKSUM_PARTIAL)
3494 aux.tp_status |= TP_STATUS_CSUMNOTREADY;
Alexander Drozdov682f0482015-03-23 09:11:13 +03003495 else if (skb->pkt_type != PACKET_OUTGOING &&
3496 (skb->ip_summed == CHECKSUM_COMPLETE ||
3497 skb_csum_unnecessary(skb)))
3498 aux.tp_status |= TP_STATUS_CSUM_VALID;
3499
Eyal Birger2472d762015-03-01 14:58:28 +02003500 aux.tp_len = origlen;
Herbert Xuffbc6112007-02-04 23:33:10 -08003501 aux.tp_snaplen = skb->len;
3502 aux.tp_mac = 0;
Arnaldo Carvalho de Melobbe735e2007-03-10 22:16:10 -03003503 aux.tp_net = skb_network_offset(skb);
Jiri Pirkodf8a39d2015-01-13 17:13:44 +01003504 if (skb_vlan_tag_present(skb)) {
3505 aux.tp_vlan_tci = skb_vlan_tag_get(skb);
Atzm Watanabea0cdfcf2013-12-17 22:53:40 +09003506 aux.tp_vlan_tpid = ntohs(skb->vlan_proto);
3507 aux.tp_status |= TP_STATUS_VLAN_VALID | TP_STATUS_VLAN_TPID_VALID;
Ben Greeara3bcc232011-06-01 06:49:10 +00003508 } else {
3509 aux.tp_vlan_tci = 0;
Atzm Watanabea0cdfcf2013-12-17 22:53:40 +09003510 aux.tp_vlan_tpid = 0;
Ben Greeara3bcc232011-06-01 06:49:10 +00003511 }
Herbert Xuffbc6112007-02-04 23:33:10 -08003512 put_cmsg(msg, SOL_PACKET, PACKET_AUXDATA, sizeof(aux), &aux);
Herbert Xu8dc41942007-02-04 23:31:32 -08003513 }
3514
Linus Torvalds1da177e2005-04-16 15:20:36 -07003515 /*
3516 * Free or return the buffer as appropriate. Again this
3517 * hides all the races and re-entrancy issues from us.
3518 */
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003519 err = vnet_hdr_len + ((flags&MSG_TRUNC) ? skb->len : copied);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003520
3521out_free:
3522 skb_free_datagram(sk, skb);
3523out:
3524 return err;
3525}
3526
Linus Torvalds1da177e2005-04-16 15:20:36 -07003527static int packet_getname_spkt(struct socket *sock, struct sockaddr *uaddr,
Denys Vlasenko9b2c45d2018-02-12 20:00:20 +01003528 int peer)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003529{
3530 struct net_device *dev;
3531 struct sock *sk = sock->sk;
3532
3533 if (peer)
3534 return -EOPNOTSUPP;
3535
3536 uaddr->sa_family = AF_PACKET;
Daniel Borkmann2dc85bf2013-06-12 16:02:27 +02003537 memset(uaddr->sa_data, 0, sizeof(uaddr->sa_data));
Eric Dumazet654d1f82009-11-02 10:43:32 +01003538 rcu_read_lock();
Eric Dumazete032f7c2021-06-16 06:42:02 -07003539 dev = dev_get_by_index_rcu(sock_net(sk), READ_ONCE(pkt_sk(sk)->ifindex));
Eric Dumazet654d1f82009-11-02 10:43:32 +01003540 if (dev)
Daniel Borkmann2dc85bf2013-06-12 16:02:27 +02003541 strlcpy(uaddr->sa_data, dev->name, sizeof(uaddr->sa_data));
Eric Dumazet654d1f82009-11-02 10:43:32 +01003542 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07003543
Denys Vlasenko9b2c45d2018-02-12 20:00:20 +01003544 return sizeof(*uaddr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003545}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003546
3547static int packet_getname(struct socket *sock, struct sockaddr *uaddr,
Denys Vlasenko9b2c45d2018-02-12 20:00:20 +01003548 int peer)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003549{
3550 struct net_device *dev;
3551 struct sock *sk = sock->sk;
3552 struct packet_sock *po = pkt_sk(sk);
Cyrill Gorcunov13cfa972009-11-08 05:51:19 +00003553 DECLARE_SOCKADDR(struct sockaddr_ll *, sll, uaddr);
Eric Dumazete032f7c2021-06-16 06:42:02 -07003554 int ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003555
3556 if (peer)
3557 return -EOPNOTSUPP;
3558
Eric Dumazete032f7c2021-06-16 06:42:02 -07003559 ifindex = READ_ONCE(po->ifindex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003560 sll->sll_family = AF_PACKET;
Eric Dumazete032f7c2021-06-16 06:42:02 -07003561 sll->sll_ifindex = ifindex;
Eric Dumazetc7d2ef52021-06-16 06:42:01 -07003562 sll->sll_protocol = READ_ONCE(po->num);
Vasiliy Kulikov67286642010-11-10 12:09:10 -08003563 sll->sll_pkttype = 0;
Eric Dumazet654d1f82009-11-02 10:43:32 +01003564 rcu_read_lock();
Eric Dumazete032f7c2021-06-16 06:42:02 -07003565 dev = dev_get_by_index_rcu(sock_net(sk), ifindex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003566 if (dev) {
3567 sll->sll_hatype = dev->type;
3568 sll->sll_halen = dev->addr_len;
3569 memcpy(sll->sll_addr, dev->dev_addr, dev->addr_len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003570 } else {
3571 sll->sll_hatype = 0; /* Bad: we have no ARPHRD_UNSPEC */
3572 sll->sll_halen = 0;
3573 }
Eric Dumazet654d1f82009-11-02 10:43:32 +01003574 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07003575
Denys Vlasenko9b2c45d2018-02-12 20:00:20 +01003576 return offsetof(struct sockaddr_ll, sll_addr) + sll->sll_halen;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003577}
3578
Wang Chen2aeb0b82008-07-14 20:49:46 -07003579static int packet_dev_mc(struct net_device *dev, struct packet_mclist *i,
3580 int what)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003581{
3582 switch (i->type) {
3583 case PACKET_MR_MULTICAST:
Jiri Pirko11625632010-03-02 20:40:01 +00003584 if (i->alen != dev->addr_len)
3585 return -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003586 if (what > 0)
Jiri Pirko22bedad32010-04-01 21:22:57 +00003587 return dev_mc_add(dev, i->addr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003588 else
Jiri Pirko22bedad32010-04-01 21:22:57 +00003589 return dev_mc_del(dev, i->addr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003590 break;
3591 case PACKET_MR_PROMISC:
Wang Chen2aeb0b82008-07-14 20:49:46 -07003592 return dev_set_promiscuity(dev, what);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003593 case PACKET_MR_ALLMULTI:
Wang Chen2aeb0b82008-07-14 20:49:46 -07003594 return dev_set_allmulti(dev, what);
Eric W. Biedermand95ed922009-05-19 18:27:17 +00003595 case PACKET_MR_UNICAST:
Jiri Pirko11625632010-03-02 20:40:01 +00003596 if (i->alen != dev->addr_len)
3597 return -EINVAL;
Eric W. Biedermand95ed922009-05-19 18:27:17 +00003598 if (what > 0)
Jiri Pirkoa748ee22010-04-01 21:22:09 +00003599 return dev_uc_add(dev, i->addr);
Eric W. Biedermand95ed922009-05-19 18:27:17 +00003600 else
Jiri Pirkoa748ee22010-04-01 21:22:09 +00003601 return dev_uc_del(dev, i->addr);
Eric W. Biedermand95ed922009-05-19 18:27:17 +00003602 break;
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003603 default:
3604 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003605 }
Wang Chen2aeb0b82008-07-14 20:49:46 -07003606 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003607}
3608
Francesco Ruggeri82f17092015-03-09 11:51:04 -07003609static void packet_dev_mclist_delete(struct net_device *dev,
3610 struct packet_mclist **mlp)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003611{
Francesco Ruggeri82f17092015-03-09 11:51:04 -07003612 struct packet_mclist *ml;
3613
3614 while ((ml = *mlp) != NULL) {
3615 if (ml->ifindex == dev->ifindex) {
3616 packet_dev_mc(dev, ml, -1);
3617 *mlp = ml->next;
3618 kfree(ml);
3619 } else
3620 mlp = &ml->next;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003621 }
3622}
3623
Eric W. Biederman0fb375f2005-09-21 00:11:37 -07003624static int packet_mc_add(struct sock *sk, struct packet_mreq_max *mreq)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003625{
3626 struct packet_sock *po = pkt_sk(sk);
3627 struct packet_mclist *ml, *i;
3628 struct net_device *dev;
3629 int err;
3630
3631 rtnl_lock();
3632
3633 err = -ENODEV;
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09003634 dev = __dev_get_by_index(sock_net(sk), mreq->mr_ifindex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003635 if (!dev)
3636 goto done;
3637
3638 err = -EINVAL;
Jiri Pirko11625632010-03-02 20:40:01 +00003639 if (mreq->mr_alen > dev->addr_len)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003640 goto done;
3641
3642 err = -ENOBUFS;
Kris Katterjohn8b3a7002006-01-11 15:56:43 -08003643 i = kmalloc(sizeof(*i), GFP_KERNEL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003644 if (i == NULL)
3645 goto done;
3646
3647 err = 0;
3648 for (ml = po->mclist; ml; ml = ml->next) {
3649 if (ml->ifindex == mreq->mr_ifindex &&
3650 ml->type == mreq->mr_type &&
3651 ml->alen == mreq->mr_alen &&
3652 memcmp(ml->addr, mreq->mr_address, ml->alen) == 0) {
3653 ml->count++;
3654 /* Free the new element ... */
3655 kfree(i);
3656 goto done;
3657 }
3658 }
3659
3660 i->type = mreq->mr_type;
3661 i->ifindex = mreq->mr_ifindex;
3662 i->alen = mreq->mr_alen;
3663 memcpy(i->addr, mreq->mr_address, i->alen);
Mathias Krause309cf372016-04-10 12:52:28 +02003664 memset(i->addr + i->alen, 0, sizeof(i->addr) - i->alen);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003665 i->count = 1;
3666 i->next = po->mclist;
3667 po->mclist = i;
Wang Chen2aeb0b82008-07-14 20:49:46 -07003668 err = packet_dev_mc(dev, i, 1);
3669 if (err) {
3670 po->mclist = i->next;
3671 kfree(i);
3672 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07003673
3674done:
3675 rtnl_unlock();
3676 return err;
3677}
3678
Eric W. Biederman0fb375f2005-09-21 00:11:37 -07003679static int packet_mc_drop(struct sock *sk, struct packet_mreq_max *mreq)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003680{
3681 struct packet_mclist *ml, **mlp;
3682
3683 rtnl_lock();
3684
3685 for (mlp = &pkt_sk(sk)->mclist; (ml = *mlp) != NULL; mlp = &ml->next) {
3686 if (ml->ifindex == mreq->mr_ifindex &&
3687 ml->type == mreq->mr_type &&
3688 ml->alen == mreq->mr_alen &&
3689 memcmp(ml->addr, mreq->mr_address, ml->alen) == 0) {
3690 if (--ml->count == 0) {
3691 struct net_device *dev;
3692 *mlp = ml->next;
Eric Dumazetad959e72009-10-16 06:38:46 +00003693 dev = __dev_get_by_index(sock_net(sk), ml->ifindex);
3694 if (dev)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003695 packet_dev_mc(dev, ml, -1);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003696 kfree(ml);
3697 }
Francesco Ruggeri82f17092015-03-09 11:51:04 -07003698 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003699 }
3700 }
3701 rtnl_unlock();
Francesco Ruggeri82f17092015-03-09 11:51:04 -07003702 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003703}
3704
3705static void packet_flush_mclist(struct sock *sk)
3706{
3707 struct packet_sock *po = pkt_sk(sk);
3708 struct packet_mclist *ml;
3709
3710 if (!po->mclist)
3711 return;
3712
3713 rtnl_lock();
3714 while ((ml = po->mclist) != NULL) {
3715 struct net_device *dev;
3716
3717 po->mclist = ml->next;
Eric Dumazetad959e72009-10-16 06:38:46 +00003718 dev = __dev_get_by_index(sock_net(sk), ml->ifindex);
3719 if (dev != NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003720 packet_dev_mc(dev, ml, -1);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003721 kfree(ml);
3722 }
3723 rtnl_unlock();
3724}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003725
3726static int
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003727packet_setsockopt(struct socket *sock, int level, int optname, sockptr_t optval,
3728 unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003729{
3730 struct sock *sk = sock->sk;
Herbert Xu8dc41942007-02-04 23:31:32 -08003731 struct packet_sock *po = pkt_sk(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003732 int ret;
3733
3734 if (level != SOL_PACKET)
3735 return -ENOPROTOOPT;
3736
Johann Baudy69e3c752009-05-18 22:11:22 -07003737 switch (optname) {
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09003738 case PACKET_ADD_MEMBERSHIP:
Linus Torvalds1da177e2005-04-16 15:20:36 -07003739 case PACKET_DROP_MEMBERSHIP:
3740 {
Eric W. Biederman0fb375f2005-09-21 00:11:37 -07003741 struct packet_mreq_max mreq;
3742 int len = optlen;
3743 memset(&mreq, 0, sizeof(mreq));
3744 if (len < sizeof(struct packet_mreq))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003745 return -EINVAL;
Eric W. Biederman0fb375f2005-09-21 00:11:37 -07003746 if (len > sizeof(mreq))
3747 len = sizeof(mreq);
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003748 if (copy_from_sockptr(&mreq, optval, len))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003749 return -EFAULT;
Eric W. Biederman0fb375f2005-09-21 00:11:37 -07003750 if (len < (mreq.mr_alen + offsetof(struct packet_mreq, mr_address)))
3751 return -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003752 if (optname == PACKET_ADD_MEMBERSHIP)
3753 ret = packet_mc_add(sk, &mreq);
3754 else
3755 ret = packet_mc_drop(sk, &mreq);
3756 return ret;
3757 }
David S. Millera2efcfa2007-05-29 13:12:50 -07003758
Linus Torvalds1da177e2005-04-16 15:20:36 -07003759 case PACKET_RX_RING:
Johann Baudy69e3c752009-05-18 22:11:22 -07003760 case PACKET_TX_RING:
Linus Torvalds1da177e2005-04-16 15:20:36 -07003761 {
chetan lokef6fb8f12011-08-19 10:18:16 +00003762 union tpacket_req_u req_u;
3763 int len;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003764
Eric Dumazet5171b372018-04-15 17:52:04 -07003765 lock_sock(sk);
chetan lokef6fb8f12011-08-19 10:18:16 +00003766 switch (po->tp_version) {
3767 case TPACKET_V1:
3768 case TPACKET_V2:
3769 len = sizeof(req_u.req);
3770 break;
3771 case TPACKET_V3:
3772 default:
3773 len = sizeof(req_u.req3);
3774 break;
3775 }
Eric Dumazet5171b372018-04-15 17:52:04 -07003776 if (optlen < len) {
3777 ret = -EINVAL;
3778 } else {
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003779 if (copy_from_sockptr(&req_u.req, optval, len))
Eric Dumazet5171b372018-04-15 17:52:04 -07003780 ret = -EFAULT;
3781 else
3782 ret = packet_set_ring(sk, &req_u, 0,
3783 optname == PACKET_TX_RING);
3784 }
3785 release_sock(sk);
3786 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003787 }
3788 case PACKET_COPY_THRESH:
3789 {
3790 int val;
3791
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003792 if (optlen != sizeof(val))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003793 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003794 if (copy_from_sockptr(&val, optval, sizeof(val)))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003795 return -EFAULT;
3796
3797 pkt_sk(sk)->copy_thresh = val;
3798 return 0;
3799 }
Patrick McHardybbd6ef82008-07-14 22:50:15 -07003800 case PACKET_VERSION:
3801 {
3802 int val;
3803
3804 if (optlen != sizeof(val))
3805 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003806 if (copy_from_sockptr(&val, optval, sizeof(val)))
Patrick McHardybbd6ef82008-07-14 22:50:15 -07003807 return -EFAULT;
3808 switch (val) {
3809 case TPACKET_V1:
3810 case TPACKET_V2:
chetan lokef6fb8f12011-08-19 10:18:16 +00003811 case TPACKET_V3:
Philip Pettersson84ac7262016-11-30 14:55:36 -08003812 break;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07003813 default:
3814 return -EINVAL;
3815 }
Philip Pettersson84ac7262016-11-30 14:55:36 -08003816 lock_sock(sk);
3817 if (po->rx_ring.pg_vec || po->tx_ring.pg_vec) {
3818 ret = -EBUSY;
3819 } else {
3820 po->tp_version = val;
3821 ret = 0;
3822 }
3823 release_sock(sk);
3824 return ret;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07003825 }
Patrick McHardy8913336a2008-07-18 18:05:19 -07003826 case PACKET_RESERVE:
3827 {
3828 unsigned int val;
3829
3830 if (optlen != sizeof(val))
3831 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003832 if (copy_from_sockptr(&val, optval, sizeof(val)))
Patrick McHardy8913336a2008-07-18 18:05:19 -07003833 return -EFAULT;
Andrey Konovalovbcc53642017-03-29 16:11:22 +02003834 if (val > INT_MAX)
3835 return -EINVAL;
Willem de Bruijnc27927e2017-08-10 12:41:58 -04003836 lock_sock(sk);
3837 if (po->rx_ring.pg_vec || po->tx_ring.pg_vec) {
3838 ret = -EBUSY;
3839 } else {
3840 po->tp_reserve = val;
3841 ret = 0;
3842 }
3843 release_sock(sk);
3844 return ret;
Patrick McHardy8913336a2008-07-18 18:05:19 -07003845 }
Johann Baudy69e3c752009-05-18 22:11:22 -07003846 case PACKET_LOSS:
3847 {
3848 unsigned int val;
3849
3850 if (optlen != sizeof(val))
3851 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003852 if (copy_from_sockptr(&val, optval, sizeof(val)))
Johann Baudy69e3c752009-05-18 22:11:22 -07003853 return -EFAULT;
Willem de Bruijna6361f02018-04-23 17:37:03 -04003854
3855 lock_sock(sk);
3856 if (po->rx_ring.pg_vec || po->tx_ring.pg_vec) {
3857 ret = -EBUSY;
3858 } else {
3859 po->tp_loss = !!val;
3860 ret = 0;
3861 }
3862 release_sock(sk);
3863 return ret;
Johann Baudy69e3c752009-05-18 22:11:22 -07003864 }
Herbert Xu8dc41942007-02-04 23:31:32 -08003865 case PACKET_AUXDATA:
3866 {
3867 int val;
3868
3869 if (optlen < sizeof(val))
3870 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003871 if (copy_from_sockptr(&val, optval, sizeof(val)))
Herbert Xu8dc41942007-02-04 23:31:32 -08003872 return -EFAULT;
3873
Willem de Bruijna6361f02018-04-23 17:37:03 -04003874 lock_sock(sk);
Herbert Xu8dc41942007-02-04 23:31:32 -08003875 po->auxdata = !!val;
Willem de Bruijna6361f02018-04-23 17:37:03 -04003876 release_sock(sk);
Herbert Xu8dc41942007-02-04 23:31:32 -08003877 return 0;
3878 }
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07003879 case PACKET_ORIGDEV:
3880 {
3881 int val;
3882
3883 if (optlen < sizeof(val))
3884 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003885 if (copy_from_sockptr(&val, optval, sizeof(val)))
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07003886 return -EFAULT;
3887
Willem de Bruijna6361f02018-04-23 17:37:03 -04003888 lock_sock(sk);
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07003889 po->origdev = !!val;
Willem de Bruijna6361f02018-04-23 17:37:03 -04003890 release_sock(sk);
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07003891 return 0;
3892 }
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003893 case PACKET_VNET_HDR:
3894 {
3895 int val;
3896
3897 if (sock->type != SOCK_RAW)
3898 return -EINVAL;
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003899 if (optlen < sizeof(val))
3900 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003901 if (copy_from_sockptr(&val, optval, sizeof(val)))
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003902 return -EFAULT;
3903
Willem de Bruijna6361f02018-04-23 17:37:03 -04003904 lock_sock(sk);
3905 if (po->rx_ring.pg_vec || po->tx_ring.pg_vec) {
3906 ret = -EBUSY;
3907 } else {
3908 po->has_vnet_hdr = !!val;
3909 ret = 0;
3910 }
3911 release_sock(sk);
3912 return ret;
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003913 }
Scott McMillan614f60f2010-06-02 05:53:56 -07003914 case PACKET_TIMESTAMP:
3915 {
3916 int val;
3917
3918 if (optlen != sizeof(val))
3919 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003920 if (copy_from_sockptr(&val, optval, sizeof(val)))
Scott McMillan614f60f2010-06-02 05:53:56 -07003921 return -EFAULT;
3922
3923 po->tp_tstamp = val;
3924 return 0;
3925 }
David S. Millerdc99f602011-07-05 01:45:05 -07003926 case PACKET_FANOUT:
3927 {
Tanner Love9c661b02020-11-06 13:07:40 -05003928 struct fanout_args args = { 0 };
David S. Millerdc99f602011-07-05 01:45:05 -07003929
Tanner Love9c661b02020-11-06 13:07:40 -05003930 if (optlen != sizeof(int) && optlen != sizeof(args))
David S. Millerdc99f602011-07-05 01:45:05 -07003931 return -EINVAL;
Tanner Love9c661b02020-11-06 13:07:40 -05003932 if (copy_from_sockptr(&args, optval, optlen))
David S. Millerdc99f602011-07-05 01:45:05 -07003933 return -EFAULT;
3934
Tanner Love9c661b02020-11-06 13:07:40 -05003935 return fanout_add(sk, &args);
David S. Millerdc99f602011-07-05 01:45:05 -07003936 }
Willem de Bruijn47dceb82015-08-14 22:31:34 -04003937 case PACKET_FANOUT_DATA:
3938 {
3939 if (!po->fanout)
3940 return -EINVAL;
3941
3942 return fanout_set_data(po, optval, optlen);
3943 }
Vincent Whitchurchfa788d92018-09-03 16:23:36 +02003944 case PACKET_IGNORE_OUTGOING:
3945 {
3946 int val;
3947
3948 if (optlen != sizeof(val))
3949 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003950 if (copy_from_sockptr(&val, optval, sizeof(val)))
Vincent Whitchurchfa788d92018-09-03 16:23:36 +02003951 return -EFAULT;
3952 if (val < 0 || val > 1)
3953 return -EINVAL;
3954
3955 po->prot_hook.ignore_outgoing = !!val;
3956 return 0;
3957 }
Paul Chavent5920cd3a2012-11-06 23:10:47 +00003958 case PACKET_TX_HAS_OFF:
3959 {
3960 unsigned int val;
3961
3962 if (optlen != sizeof(val))
3963 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003964 if (copy_from_sockptr(&val, optval, sizeof(val)))
Paul Chavent5920cd3a2012-11-06 23:10:47 +00003965 return -EFAULT;
Willem de Bruijna6361f02018-04-23 17:37:03 -04003966
3967 lock_sock(sk);
Jiapeng Chong25c55b32021-05-17 18:15:25 +08003968 if (!po->rx_ring.pg_vec && !po->tx_ring.pg_vec)
Willem de Bruijna6361f02018-04-23 17:37:03 -04003969 po->tp_tx_has_off = !!val;
Jiapeng Chong25c55b32021-05-17 18:15:25 +08003970
Willem de Bruijna6361f02018-04-23 17:37:03 -04003971 release_sock(sk);
Paul Chavent5920cd3a2012-11-06 23:10:47 +00003972 return 0;
3973 }
Daniel Borkmannd346a3f2013-12-06 11:36:17 +01003974 case PACKET_QDISC_BYPASS:
3975 {
3976 int val;
3977
3978 if (optlen != sizeof(val))
3979 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003980 if (copy_from_sockptr(&val, optval, sizeof(val)))
Daniel Borkmannd346a3f2013-12-06 11:36:17 +01003981 return -EFAULT;
3982
3983 po->xmit = val ? packet_direct_xmit : dev_queue_xmit;
3984 return 0;
3985 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07003986 default:
3987 return -ENOPROTOOPT;
3988 }
3989}
3990
3991static int packet_getsockopt(struct socket *sock, int level, int optname,
3992 char __user *optval, int __user *optlen)
3993{
3994 int len;
Eric Dumazetc06fff62012-04-19 21:56:11 +00003995 int val, lv = sizeof(val);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003996 struct sock *sk = sock->sk;
3997 struct packet_sock *po = pkt_sk(sk);
Eric Dumazetc06fff62012-04-19 21:56:11 +00003998 void *data = &val;
Daniel Borkmannee80fbf2013-04-19 06:12:29 +00003999 union tpacket_stats_u st;
Willem de Bruijna9b63912015-05-12 11:56:50 -04004000 struct tpacket_rollover_stats rstats;
Eric Dumazet8e8e2952019-06-12 09:52:30 -07004001 int drops;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004002
4003 if (level != SOL_PACKET)
4004 return -ENOPROTOOPT;
4005
Kris Katterjohn8ae55f02006-01-23 16:28:02 -08004006 if (get_user(len, optlen))
4007 return -EFAULT;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004008
4009 if (len < 0)
4010 return -EINVAL;
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09004011
Johann Baudy69e3c752009-05-18 22:11:22 -07004012 switch (optname) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004013 case PACKET_STATISTICS:
Linus Torvalds1da177e2005-04-16 15:20:36 -07004014 spin_lock_bh(&sk->sk_receive_queue.lock);
Daniel Borkmannee80fbf2013-04-19 06:12:29 +00004015 memcpy(&st, &po->stats, sizeof(st));
4016 memset(&po->stats, 0, sizeof(po->stats));
4017 spin_unlock_bh(&sk->sk_receive_queue.lock);
Eric Dumazet8e8e2952019-06-12 09:52:30 -07004018 drops = atomic_xchg(&po->tp_drops, 0);
Daniel Borkmannee80fbf2013-04-19 06:12:29 +00004019
chetan lokef6fb8f12011-08-19 10:18:16 +00004020 if (po->tp_version == TPACKET_V3) {
Eric Dumazetc06fff62012-04-19 21:56:11 +00004021 lv = sizeof(struct tpacket_stats_v3);
Eric Dumazet8e8e2952019-06-12 09:52:30 -07004022 st.stats3.tp_drops = drops;
4023 st.stats3.tp_packets += drops;
Daniel Borkmannee80fbf2013-04-19 06:12:29 +00004024 data = &st.stats3;
chetan lokef6fb8f12011-08-19 10:18:16 +00004025 } else {
Eric Dumazetc06fff62012-04-19 21:56:11 +00004026 lv = sizeof(struct tpacket_stats);
Eric Dumazet8e8e2952019-06-12 09:52:30 -07004027 st.stats1.tp_drops = drops;
4028 st.stats1.tp_packets += drops;
Daniel Borkmannee80fbf2013-04-19 06:12:29 +00004029 data = &st.stats1;
chetan lokef6fb8f12011-08-19 10:18:16 +00004030 }
Daniel Borkmannee80fbf2013-04-19 06:12:29 +00004031
Linus Torvalds1da177e2005-04-16 15:20:36 -07004032 break;
Herbert Xu8dc41942007-02-04 23:31:32 -08004033 case PACKET_AUXDATA:
Herbert Xu8dc41942007-02-04 23:31:32 -08004034 val = po->auxdata;
Herbert Xu8dc41942007-02-04 23:31:32 -08004035 break;
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07004036 case PACKET_ORIGDEV:
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07004037 val = po->origdev;
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07004038 break;
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08004039 case PACKET_VNET_HDR:
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08004040 val = po->has_vnet_hdr;
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08004041 break;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004042 case PACKET_VERSION:
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004043 val = po->tp_version;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004044 break;
4045 case PACKET_HDRLEN:
4046 if (len > sizeof(int))
4047 len = sizeof(int);
Alexander Potapenkofd2c83b2017-04-25 18:51:46 +02004048 if (len < sizeof(int))
4049 return -EINVAL;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004050 if (copy_from_user(&val, optval, len))
4051 return -EFAULT;
4052 switch (val) {
4053 case TPACKET_V1:
4054 val = sizeof(struct tpacket_hdr);
4055 break;
4056 case TPACKET_V2:
4057 val = sizeof(struct tpacket2_hdr);
4058 break;
chetan lokef6fb8f12011-08-19 10:18:16 +00004059 case TPACKET_V3:
4060 val = sizeof(struct tpacket3_hdr);
4061 break;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004062 default:
4063 return -EINVAL;
4064 }
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004065 break;
Patrick McHardy8913336a2008-07-18 18:05:19 -07004066 case PACKET_RESERVE:
Patrick McHardy8913336a2008-07-18 18:05:19 -07004067 val = po->tp_reserve;
Patrick McHardy8913336a2008-07-18 18:05:19 -07004068 break;
Johann Baudy69e3c752009-05-18 22:11:22 -07004069 case PACKET_LOSS:
Johann Baudy69e3c752009-05-18 22:11:22 -07004070 val = po->tp_loss;
Johann Baudy69e3c752009-05-18 22:11:22 -07004071 break;
Scott McMillan614f60f2010-06-02 05:53:56 -07004072 case PACKET_TIMESTAMP:
Scott McMillan614f60f2010-06-02 05:53:56 -07004073 val = po->tp_tstamp;
Scott McMillan614f60f2010-06-02 05:53:56 -07004074 break;
David S. Millerdc99f602011-07-05 01:45:05 -07004075 case PACKET_FANOUT:
David S. Millerdc99f602011-07-05 01:45:05 -07004076 val = (po->fanout ?
4077 ((u32)po->fanout->id |
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00004078 ((u32)po->fanout->type << 16) |
4079 ((u32)po->fanout->flags << 24)) :
David S. Millerdc99f602011-07-05 01:45:05 -07004080 0);
David S. Millerdc99f602011-07-05 01:45:05 -07004081 break;
Vincent Whitchurchfa788d92018-09-03 16:23:36 +02004082 case PACKET_IGNORE_OUTGOING:
4083 val = po->prot_hook.ignore_outgoing;
4084 break;
Willem de Bruijna9b63912015-05-12 11:56:50 -04004085 case PACKET_ROLLOVER_STATS:
Mike Maloney57f015f2017-11-28 10:44:29 -05004086 if (!po->rollover)
Willem de Bruijna9b63912015-05-12 11:56:50 -04004087 return -EINVAL;
Mike Maloney57f015f2017-11-28 10:44:29 -05004088 rstats.tp_all = atomic_long_read(&po->rollover->num);
4089 rstats.tp_huge = atomic_long_read(&po->rollover->num_huge);
4090 rstats.tp_failed = atomic_long_read(&po->rollover->num_failed);
4091 data = &rstats;
4092 lv = sizeof(rstats);
Willem de Bruijna9b63912015-05-12 11:56:50 -04004093 break;
Paul Chavent5920cd3a2012-11-06 23:10:47 +00004094 case PACKET_TX_HAS_OFF:
4095 val = po->tp_tx_has_off;
4096 break;
Daniel Borkmannd346a3f2013-12-06 11:36:17 +01004097 case PACKET_QDISC_BYPASS:
4098 val = packet_use_direct_xmit(po);
4099 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004100 default:
4101 return -ENOPROTOOPT;
4102 }
4103
Eric Dumazetc06fff62012-04-19 21:56:11 +00004104 if (len > lv)
4105 len = lv;
Kris Katterjohn8ae55f02006-01-23 16:28:02 -08004106 if (put_user(len, optlen))
4107 return -EFAULT;
Herbert Xu8dc41942007-02-04 23:31:32 -08004108 if (copy_to_user(optval, data, len))
4109 return -EFAULT;
Kris Katterjohn8ae55f02006-01-23 16:28:02 -08004110 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004111}
4112
Jiri Pirko351638e2013-05-28 01:30:21 +00004113static int packet_notifier(struct notifier_block *this,
4114 unsigned long msg, void *ptr)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004115{
4116 struct sock *sk;
Jiri Pirko351638e2013-05-28 01:30:21 +00004117 struct net_device *dev = netdev_notifier_info_to_dev(ptr);
YOSHIFUJI Hideakic346dca2008-03-25 21:47:49 +09004118 struct net *net = dev_net(dev);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004119
stephen hemminger808f5112010-02-22 07:57:18 +00004120 rcu_read_lock();
Sasha Levinb67bfe02013-02-27 17:06:00 -08004121 sk_for_each_rcu(sk, &net->packet.sklist) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004122 struct packet_sock *po = pkt_sk(sk);
4123
4124 switch (msg) {
4125 case NETDEV_UNREGISTER:
Linus Torvalds1da177e2005-04-16 15:20:36 -07004126 if (po->mclist)
Francesco Ruggeri82f17092015-03-09 11:51:04 -07004127 packet_dev_mclist_delete(dev, &po->mclist);
Gustavo A. R. Silvadf561f662020-08-23 17:36:59 -05004128 fallthrough;
David S. Millera2efcfa2007-05-29 13:12:50 -07004129
Linus Torvalds1da177e2005-04-16 15:20:36 -07004130 case NETDEV_DOWN:
4131 if (dev->ifindex == po->ifindex) {
4132 spin_lock(&po->bind_lock);
4133 if (po->running) {
David S. Millerce06b032011-07-04 01:44:29 -07004134 __unregister_prot_hook(sk, false);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004135 sk->sk_err = ENETDOWN;
4136 if (!sock_flag(sk, SOCK_DEAD))
Alexander Aringe3ae2362021-06-27 18:48:21 -04004137 sk_error_report(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004138 }
4139 if (msg == NETDEV_UNREGISTER) {
Daniel Borkmann66e56cd2013-12-06 11:36:15 +01004140 packet_cached_dev_reset(po);
Eric Dumazete032f7c2021-06-16 06:42:02 -07004141 WRITE_ONCE(po->ifindex, -1);
Yajun Deng1160dfa2021-08-05 19:55:27 +08004142 dev_put(po->prot_hook.dev);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004143 po->prot_hook.dev = NULL;
4144 }
4145 spin_unlock(&po->bind_lock);
4146 }
4147 break;
4148 case NETDEV_UP:
stephen hemminger808f5112010-02-22 07:57:18 +00004149 if (dev->ifindex == po->ifindex) {
4150 spin_lock(&po->bind_lock);
David S. Millerce06b032011-07-04 01:44:29 -07004151 if (po->num)
4152 register_prot_hook(sk);
stephen hemminger808f5112010-02-22 07:57:18 +00004153 spin_unlock(&po->bind_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004154 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004155 break;
4156 }
4157 }
stephen hemminger808f5112010-02-22 07:57:18 +00004158 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004159 return NOTIFY_DONE;
4160}
4161
4162
4163static int packet_ioctl(struct socket *sock, unsigned int cmd,
4164 unsigned long arg)
4165{
4166 struct sock *sk = sock->sk;
4167
Johann Baudy69e3c752009-05-18 22:11:22 -07004168 switch (cmd) {
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004169 case SIOCOUTQ:
4170 {
4171 int amount = sk_wmem_alloc_get(sk);
Eric Dumazet31e6d362009-06-17 19:05:41 -07004172
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004173 return put_user(amount, (int __user *)arg);
4174 }
4175 case SIOCINQ:
4176 {
4177 struct sk_buff *skb;
4178 int amount = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004179
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004180 spin_lock_bh(&sk->sk_receive_queue.lock);
4181 skb = skb_peek(&sk->sk_receive_queue);
4182 if (skb)
4183 amount = skb->len;
4184 spin_unlock_bh(&sk->sk_receive_queue.lock);
4185 return put_user(amount, (int __user *)arg);
4186 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004187#ifdef CONFIG_INET
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004188 case SIOCADDRT:
4189 case SIOCDELRT:
4190 case SIOCDARP:
4191 case SIOCGARP:
4192 case SIOCSARP:
4193 case SIOCGIFADDR:
4194 case SIOCSIFADDR:
4195 case SIOCGIFBRDADDR:
4196 case SIOCSIFBRDADDR:
4197 case SIOCGIFNETMASK:
4198 case SIOCSIFNETMASK:
4199 case SIOCGIFDSTADDR:
4200 case SIOCSIFDSTADDR:
4201 case SIOCSIFFLAGS:
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004202 return inet_dgram_ops.ioctl(sock, cmd, arg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004203#endif
4204
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004205 default:
4206 return -ENOIOCTLCMD;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004207 }
4208 return 0;
4209}
4210
Linus Torvaldsa11e1d42018-06-28 09:43:44 -07004211static __poll_t packet_poll(struct file *file, struct socket *sock,
4212 poll_table *wait)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004213{
4214 struct sock *sk = sock->sk;
4215 struct packet_sock *po = pkt_sk(sk);
Linus Torvaldsa11e1d42018-06-28 09:43:44 -07004216 __poll_t mask = datagram_poll(file, sock, wait);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004217
4218 spin_lock_bh(&sk->sk_receive_queue.lock);
Johann Baudy69e3c752009-05-18 22:11:22 -07004219 if (po->rx_ring.pg_vec) {
chetan lokef6fb8f12011-08-19 10:18:16 +00004220 if (!packet_previous_rx_frame(po, &po->rx_ring,
4221 TP_STATUS_KERNEL))
Linus Torvaldsa9a08842018-02-11 14:34:03 -08004222 mask |= EPOLLIN | EPOLLRDNORM;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004223 }
Eric Dumazet9bb6cd62019-06-12 09:52:33 -07004224 packet_rcv_try_clear_pressure(po);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004225 spin_unlock_bh(&sk->sk_receive_queue.lock);
Johann Baudy69e3c752009-05-18 22:11:22 -07004226 spin_lock_bh(&sk->sk_write_queue.lock);
4227 if (po->tx_ring.pg_vec) {
4228 if (packet_current_frame(po, &po->tx_ring, TP_STATUS_AVAILABLE))
Linus Torvaldsa9a08842018-02-11 14:34:03 -08004229 mask |= EPOLLOUT | EPOLLWRNORM;
Johann Baudy69e3c752009-05-18 22:11:22 -07004230 }
4231 spin_unlock_bh(&sk->sk_write_queue.lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004232 return mask;
4233}
4234
4235
4236/* Dirty? Well, I still did not learn better way to account
4237 * for user mmaps.
4238 */
4239
4240static void packet_mm_open(struct vm_area_struct *vma)
4241{
4242 struct file *file = vma->vm_file;
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004243 struct socket *sock = file->private_data;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004244 struct sock *sk = sock->sk;
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09004245
Linus Torvalds1da177e2005-04-16 15:20:36 -07004246 if (sk)
4247 atomic_inc(&pkt_sk(sk)->mapped);
4248}
4249
4250static void packet_mm_close(struct vm_area_struct *vma)
4251{
4252 struct file *file = vma->vm_file;
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004253 struct socket *sock = file->private_data;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004254 struct sock *sk = sock->sk;
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09004255
Linus Torvalds1da177e2005-04-16 15:20:36 -07004256 if (sk)
4257 atomic_dec(&pkt_sk(sk)->mapped);
4258}
4259
Alexey Dobriyanf0f37e2f2009-09-27 22:29:37 +04004260static const struct vm_operations_struct packet_mmap_ops = {
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004261 .open = packet_mm_open,
4262 .close = packet_mm_close,
Linus Torvalds1da177e2005-04-16 15:20:36 -07004263};
4264
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004265static void free_pg_vec(struct pgv *pg_vec, unsigned int order,
4266 unsigned int len)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004267{
4268 int i;
4269
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004270 for (i = 0; i < len; i++) {
Neil Horman0e3125c2010-11-16 10:26:47 -08004271 if (likely(pg_vec[i].buffer)) {
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004272 if (is_vmalloc_addr(pg_vec[i].buffer))
4273 vfree(pg_vec[i].buffer);
4274 else
4275 free_pages((unsigned long)pg_vec[i].buffer,
4276 order);
Neil Horman0e3125c2010-11-16 10:26:47 -08004277 pg_vec[i].buffer = NULL;
4278 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004279 }
4280 kfree(pg_vec);
4281}
4282
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004283static char *alloc_one_pg_vec_page(unsigned long order)
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004284{
Daniel Borkmannf0d4eb22014-01-19 11:46:53 +01004285 char *buffer;
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004286 gfp_t gfp_flags = GFP_KERNEL | __GFP_COMP |
4287 __GFP_ZERO | __GFP_NOWARN | __GFP_NORETRY;
Eric Dumazet719bfea2009-04-15 03:39:52 -07004288
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004289 buffer = (char *) __get_free_pages(gfp_flags, order);
Neil Horman0e3125c2010-11-16 10:26:47 -08004290 if (buffer)
4291 return buffer;
4292
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004293 /* __get_free_pages failed, fall back to vmalloc */
4294 buffer = vzalloc(array_size((1 << order), PAGE_SIZE));
4295 if (buffer)
4296 return buffer;
Neil Horman0e3125c2010-11-16 10:26:47 -08004297
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004298 /* vmalloc failed, lets dig into swap here */
4299 gfp_flags &= ~__GFP_NORETRY;
4300 buffer = (char *) __get_free_pages(gfp_flags, order);
4301 if (buffer)
4302 return buffer;
4303
4304 /* complete and utter failure */
4305 return NULL;
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004306}
4307
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004308static struct pgv *alloc_pg_vec(struct tpacket_req *req, int order)
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004309{
4310 unsigned int block_nr = req->tp_block_nr;
Neil Horman0e3125c2010-11-16 10:26:47 -08004311 struct pgv *pg_vec;
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004312 int i;
4313
Christoph Paasch398f0132019-03-18 23:14:52 -07004314 pg_vec = kcalloc(block_nr, sizeof(struct pgv), GFP_KERNEL | __GFP_NOWARN);
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004315 if (unlikely(!pg_vec))
4316 goto out;
4317
4318 for (i = 0; i < block_nr; i++) {
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004319 pg_vec[i].buffer = alloc_one_pg_vec_page(order);
Neil Horman0e3125c2010-11-16 10:26:47 -08004320 if (unlikely(!pg_vec[i].buffer))
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004321 goto out_free_pgvec;
4322 }
4323
4324out:
4325 return pg_vec;
4326
4327out_free_pgvec:
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004328 free_pg_vec(pg_vec, order, block_nr);
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004329 pg_vec = NULL;
4330 goto out;
4331}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004332
chetan lokef6fb8f12011-08-19 10:18:16 +00004333static int packet_set_ring(struct sock *sk, union tpacket_req_u *req_u,
Johann Baudy69e3c752009-05-18 22:11:22 -07004334 int closing, int tx_ring)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004335{
Neil Horman0e3125c2010-11-16 10:26:47 -08004336 struct pgv *pg_vec = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004337 struct packet_sock *po = pkt_sk(sk);
Willem de Bruijn61fad682020-03-13 12:18:09 -04004338 unsigned long *rx_owner_map = NULL;
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004339 int was_running, order = 0;
Johann Baudy69e3c752009-05-18 22:11:22 -07004340 struct packet_ring_buffer *rb;
4341 struct sk_buff_head *rb_queue;
Al Viro0e11c912006-11-08 00:26:29 -08004342 __be16 num;
Colin Ian King2a6d6c32020-07-01 16:04:33 +01004343 int err;
chetan lokef6fb8f12011-08-19 10:18:16 +00004344 /* Added to avoid minimal code churn */
4345 struct tpacket_req *req = &req_u->req;
4346
Johann Baudy69e3c752009-05-18 22:11:22 -07004347 rb = tx_ring ? &po->tx_ring : &po->rx_ring;
4348 rb_queue = tx_ring ? &sk->sk_write_queue : &sk->sk_receive_queue;
4349
4350 err = -EBUSY;
4351 if (!closing) {
4352 if (atomic_read(&po->mapped))
4353 goto out;
Daniel Borkmannb0138402014-01-15 16:25:36 +01004354 if (packet_read_pending(rb))
Johann Baudy69e3c752009-05-18 22:11:22 -07004355 goto out;
4356 }
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09004357
Linus Torvalds1da177e2005-04-16 15:20:36 -07004358 if (req->tp_block_nr) {
Willem de Bruijn4576cd42018-08-06 10:38:34 -04004359 unsigned int min_frame_size;
4360
Linus Torvalds1da177e2005-04-16 15:20:36 -07004361 /* Sanity tests and some calculations */
Johann Baudy69e3c752009-05-18 22:11:22 -07004362 err = -EBUSY;
4363 if (unlikely(rb->pg_vec))
4364 goto out;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004365
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004366 switch (po->tp_version) {
4367 case TPACKET_V1:
4368 po->tp_hdrlen = TPACKET_HDRLEN;
4369 break;
4370 case TPACKET_V2:
4371 po->tp_hdrlen = TPACKET2_HDRLEN;
4372 break;
chetan lokef6fb8f12011-08-19 10:18:16 +00004373 case TPACKET_V3:
4374 po->tp_hdrlen = TPACKET3_HDRLEN;
4375 break;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004376 }
4377
Johann Baudy69e3c752009-05-18 22:11:22 -07004378 err = -EINVAL;
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004379 if (unlikely((int)req->tp_block_size <= 0))
Johann Baudy69e3c752009-05-18 22:11:22 -07004380 goto out;
Tobias Klauser90836b62015-11-17 10:40:21 +01004381 if (unlikely(!PAGE_ALIGNED(req->tp_block_size)))
Johann Baudy69e3c752009-05-18 22:11:22 -07004382 goto out;
Willem de Bruijn4576cd42018-08-06 10:38:34 -04004383 min_frame_size = po->tp_hdrlen + po->tp_reserve;
Eric Dumazetdc808112014-08-15 09:16:04 -07004384 if (po->tp_version >= TPACKET_V3 &&
Willem de Bruijn4576cd42018-08-06 10:38:34 -04004385 req->tp_block_size <
4386 BLK_PLUS_PRIV((u64)req_u->req3.tp_sizeof_priv) + min_frame_size)
Eric Dumazetdc808112014-08-15 09:16:04 -07004387 goto out;
Willem de Bruijn4576cd42018-08-06 10:38:34 -04004388 if (unlikely(req->tp_frame_size < min_frame_size))
Johann Baudy69e3c752009-05-18 22:11:22 -07004389 goto out;
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004390 if (unlikely(req->tp_frame_size & (TPACKET_ALIGNMENT - 1)))
Johann Baudy69e3c752009-05-18 22:11:22 -07004391 goto out;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004392
Tobias Klauser4194b492015-11-17 10:38:36 +01004393 rb->frames_per_block = req->tp_block_size / req->tp_frame_size;
4394 if (unlikely(rb->frames_per_block == 0))
Johann Baudy69e3c752009-05-18 22:11:22 -07004395 goto out;
Kal Conleyfc628142019-02-10 09:57:11 +01004396 if (unlikely(rb->frames_per_block > UINT_MAX / req->tp_block_nr))
Andrey Konovalov8f8d28e2017-03-29 16:11:21 +02004397 goto out;
Johann Baudy69e3c752009-05-18 22:11:22 -07004398 if (unlikely((rb->frames_per_block * req->tp_block_nr) !=
4399 req->tp_frame_nr))
4400 goto out;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004401
4402 err = -ENOMEM;
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004403 order = get_order(req->tp_block_size);
4404 pg_vec = alloc_pg_vec(req, order);
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004405 if (unlikely(!pg_vec))
Linus Torvalds1da177e2005-04-16 15:20:36 -07004406 goto out;
chetan lokef6fb8f12011-08-19 10:18:16 +00004407 switch (po->tp_version) {
4408 case TPACKET_V3:
Sowmini Varadhan7f953ab2017-01-03 06:31:47 -08004409 /* Block transmit is not supported yet */
4410 if (!tx_ring) {
Maninder Singhe8e85cc2015-06-22 12:39:16 +05304411 init_prb_bdqc(po, rb, pg_vec, req_u);
Sowmini Varadhan7f953ab2017-01-03 06:31:47 -08004412 } else {
4413 struct tpacket_req3 *req3 = &req_u->req3;
4414
4415 if (req3->tp_retire_blk_tov ||
4416 req3->tp_sizeof_priv ||
4417 req3->tp_feature_req_word) {
4418 err = -EINVAL;
Eric Dumazet55655e32019-06-24 02:38:20 -07004419 goto out_free_pg_vec;
Sowmini Varadhan7f953ab2017-01-03 06:31:47 -08004420 }
4421 }
Dan Carpenterd7cf0c32014-02-18 15:20:51 +03004422 break;
chetan lokef6fb8f12011-08-19 10:18:16 +00004423 default:
Willem de Bruijn61fad682020-03-13 12:18:09 -04004424 if (!tx_ring) {
4425 rx_owner_map = bitmap_alloc(req->tp_frame_nr,
4426 GFP_KERNEL | __GFP_NOWARN | __GFP_ZERO);
4427 if (!rx_owner_map)
4428 goto out_free_pg_vec;
4429 }
chetan lokef6fb8f12011-08-19 10:18:16 +00004430 break;
4431 }
Johann Baudy69e3c752009-05-18 22:11:22 -07004432 }
4433 /* Done */
4434 else {
4435 err = -EINVAL;
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004436 if (unlikely(req->tp_frame_nr))
Johann Baudy69e3c752009-05-18 22:11:22 -07004437 goto out;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004438 }
4439
Linus Torvalds1da177e2005-04-16 15:20:36 -07004440
4441 /* Detach socket from network */
4442 spin_lock(&po->bind_lock);
4443 was_running = po->running;
4444 num = po->num;
4445 if (was_running) {
Eric Dumazetc7d2ef52021-06-16 06:42:01 -07004446 WRITE_ONCE(po->num, 0);
David S. Millerce06b032011-07-04 01:44:29 -07004447 __unregister_prot_hook(sk, false);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004448 }
4449 spin_unlock(&po->bind_lock);
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09004450
Linus Torvalds1da177e2005-04-16 15:20:36 -07004451 synchronize_net();
4452
4453 err = -EBUSY;
Herbert Xu905db442009-01-30 14:12:06 -08004454 mutex_lock(&po->pg_vec_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004455 if (closing || atomic_read(&po->mapped) == 0) {
4456 err = 0;
Johann Baudy69e3c752009-05-18 22:11:22 -07004457 spin_lock_bh(&rb_queue->lock);
Changli Gaoc053fd92010-12-10 16:02:20 -08004458 swap(rb->pg_vec, pg_vec);
Willem de Bruijn61fad682020-03-13 12:18:09 -04004459 if (po->tp_version <= TPACKET_V2)
4460 swap(rb->rx_owner_map, rx_owner_map);
Johann Baudy69e3c752009-05-18 22:11:22 -07004461 rb->frame_max = (req->tp_frame_nr - 1);
4462 rb->head = 0;
4463 rb->frame_size = req->tp_frame_size;
4464 spin_unlock_bh(&rb_queue->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004465
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004466 swap(rb->pg_vec_order, order);
Changli Gaoc053fd92010-12-10 16:02:20 -08004467 swap(rb->pg_vec_len, req->tp_block_nr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004468
Johann Baudy69e3c752009-05-18 22:11:22 -07004469 rb->pg_vec_pages = req->tp_block_size/PAGE_SIZE;
4470 po->prot_hook.func = (po->rx_ring.pg_vec) ?
4471 tpacket_rcv : packet_rcv;
4472 skb_queue_purge(rb_queue);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004473 if (atomic_read(&po->mapped))
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004474 pr_err("packet_mmap: vma is busy: %d\n",
4475 atomic_read(&po->mapped));
Linus Torvalds1da177e2005-04-16 15:20:36 -07004476 }
Herbert Xu905db442009-01-30 14:12:06 -08004477 mutex_unlock(&po->pg_vec_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004478
4479 spin_lock(&po->bind_lock);
David S. Millerce06b032011-07-04 01:44:29 -07004480 if (was_running) {
Eric Dumazetc7d2ef52021-06-16 06:42:01 -07004481 WRITE_ONCE(po->num, num);
David S. Millerce06b032011-07-04 01:44:29 -07004482 register_prot_hook(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004483 }
4484 spin_unlock(&po->bind_lock);
WANG Congc800aaf2017-07-24 10:07:32 -07004485 if (pg_vec && (po->tp_version > TPACKET_V2)) {
chetan lokef6fb8f12011-08-19 10:18:16 +00004486 /* Because we don't support block-based V3 on tx-ring */
4487 if (!tx_ring)
Tobias Klauser73d0fcf2015-07-28 14:21:26 +02004488 prb_shutdown_retire_blk_timer(po, rb_queue);
chetan lokef6fb8f12011-08-19 10:18:16 +00004489 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004490
Eric Dumazet55655e32019-06-24 02:38:20 -07004491out_free_pg_vec:
Willem de Bruijn61fad682020-03-13 12:18:09 -04004492 bitmap_free(rx_owner_map);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004493 if (pg_vec)
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004494 free_pg_vec(pg_vec, order, req->tp_block_nr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004495out:
4496 return err;
4497}
4498
Johann Baudy69e3c752009-05-18 22:11:22 -07004499static int packet_mmap(struct file *file, struct socket *sock,
4500 struct vm_area_struct *vma)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004501{
4502 struct sock *sk = sock->sk;
4503 struct packet_sock *po = pkt_sk(sk);
Johann Baudy69e3c752009-05-18 22:11:22 -07004504 unsigned long size, expected_size;
4505 struct packet_ring_buffer *rb;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004506 unsigned long start;
4507 int err = -EINVAL;
4508 int i;
4509
4510 if (vma->vm_pgoff)
4511 return -EINVAL;
4512
Herbert Xu905db442009-01-30 14:12:06 -08004513 mutex_lock(&po->pg_vec_lock);
Johann Baudy69e3c752009-05-18 22:11:22 -07004514
4515 expected_size = 0;
4516 for (rb = &po->rx_ring; rb <= &po->tx_ring; rb++) {
4517 if (rb->pg_vec) {
4518 expected_size += rb->pg_vec_len
4519 * rb->pg_vec_pages
4520 * PAGE_SIZE;
4521 }
4522 }
4523
4524 if (expected_size == 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004525 goto out;
Johann Baudy69e3c752009-05-18 22:11:22 -07004526
4527 size = vma->vm_end - vma->vm_start;
4528 if (size != expected_size)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004529 goto out;
4530
Linus Torvalds1da177e2005-04-16 15:20:36 -07004531 start = vma->vm_start;
Johann Baudy69e3c752009-05-18 22:11:22 -07004532 for (rb = &po->rx_ring; rb <= &po->tx_ring; rb++) {
4533 if (rb->pg_vec == NULL)
4534 continue;
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004535
Johann Baudy69e3c752009-05-18 22:11:22 -07004536 for (i = 0; i < rb->pg_vec_len; i++) {
Neil Horman0e3125c2010-11-16 10:26:47 -08004537 struct page *page;
4538 void *kaddr = rb->pg_vec[i].buffer;
Johann Baudy69e3c752009-05-18 22:11:22 -07004539 int pg_num;
4540
Changli Gaoc56b4d92010-12-01 02:52:57 +00004541 for (pg_num = 0; pg_num < rb->pg_vec_pages; pg_num++) {
4542 page = pgv_to_page(kaddr);
Johann Baudy69e3c752009-05-18 22:11:22 -07004543 err = vm_insert_page(vma, start, page);
4544 if (unlikely(err))
4545 goto out;
4546 start += PAGE_SIZE;
Neil Horman0e3125c2010-11-16 10:26:47 -08004547 kaddr += PAGE_SIZE;
Johann Baudy69e3c752009-05-18 22:11:22 -07004548 }
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004549 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004550 }
Johann Baudy69e3c752009-05-18 22:11:22 -07004551
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004552 atomic_inc(&po->mapped);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004553 vma->vm_ops = &packet_mmap_ops;
4554 err = 0;
4555
4556out:
Herbert Xu905db442009-01-30 14:12:06 -08004557 mutex_unlock(&po->pg_vec_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004558 return err;
4559}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004560
Eric Dumazet90ddc4f2005-12-22 12:49:22 -08004561static const struct proto_ops packet_ops_spkt = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004562 .family = PF_PACKET,
4563 .owner = THIS_MODULE,
4564 .release = packet_release,
4565 .bind = packet_bind_spkt,
4566 .connect = sock_no_connect,
4567 .socketpair = sock_no_socketpair,
4568 .accept = sock_no_accept,
4569 .getname = packet_getname_spkt,
Linus Torvaldsa11e1d42018-06-28 09:43:44 -07004570 .poll = datagram_poll,
Linus Torvalds1da177e2005-04-16 15:20:36 -07004571 .ioctl = packet_ioctl,
Arnd Bergmannc7cbdbf2019-04-17 22:51:48 +02004572 .gettstamp = sock_gettstamp,
Linus Torvalds1da177e2005-04-16 15:20:36 -07004573 .listen = sock_no_listen,
4574 .shutdown = sock_no_shutdown,
Linus Torvalds1da177e2005-04-16 15:20:36 -07004575 .sendmsg = packet_sendmsg_spkt,
4576 .recvmsg = packet_recvmsg,
4577 .mmap = sock_no_mmap,
4578 .sendpage = sock_no_sendpage,
4579};
Linus Torvalds1da177e2005-04-16 15:20:36 -07004580
Eric Dumazet90ddc4f2005-12-22 12:49:22 -08004581static const struct proto_ops packet_ops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004582 .family = PF_PACKET,
4583 .owner = THIS_MODULE,
4584 .release = packet_release,
4585 .bind = packet_bind,
4586 .connect = sock_no_connect,
4587 .socketpair = sock_no_socketpair,
4588 .accept = sock_no_accept,
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09004589 .getname = packet_getname,
Linus Torvaldsa11e1d42018-06-28 09:43:44 -07004590 .poll = packet_poll,
Linus Torvalds1da177e2005-04-16 15:20:36 -07004591 .ioctl = packet_ioctl,
Arnd Bergmannc7cbdbf2019-04-17 22:51:48 +02004592 .gettstamp = sock_gettstamp,
Linus Torvalds1da177e2005-04-16 15:20:36 -07004593 .listen = sock_no_listen,
4594 .shutdown = sock_no_shutdown,
4595 .setsockopt = packet_setsockopt,
4596 .getsockopt = packet_getsockopt,
4597 .sendmsg = packet_sendmsg,
4598 .recvmsg = packet_recvmsg,
4599 .mmap = packet_mmap,
4600 .sendpage = sock_no_sendpage,
4601};
4602
Stephen Hemmingerec1b4cf2009-10-05 05:58:39 +00004603static const struct net_proto_family packet_family_ops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004604 .family = PF_PACKET,
4605 .create = packet_create,
4606 .owner = THIS_MODULE,
4607};
4608
4609static struct notifier_block packet_netdev_notifier = {
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004610 .notifier_call = packet_notifier,
Linus Torvalds1da177e2005-04-16 15:20:36 -07004611};
4612
4613#ifdef CONFIG_PROC_FS
Linus Torvalds1da177e2005-04-16 15:20:36 -07004614
4615static void *packet_seq_start(struct seq_file *seq, loff_t *pos)
stephen hemminger808f5112010-02-22 07:57:18 +00004616 __acquires(RCU)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004617{
Denis V. Luneve372c412007-11-19 22:31:54 -08004618 struct net *net = seq_file_net(seq);
stephen hemminger808f5112010-02-22 07:57:18 +00004619
4620 rcu_read_lock();
4621 return seq_hlist_start_head_rcu(&net->packet.sklist, *pos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004622}
4623
4624static void *packet_seq_next(struct seq_file *seq, void *v, loff_t *pos)
4625{
Herbert Xu1bf40952007-12-16 14:04:02 -08004626 struct net *net = seq_file_net(seq);
stephen hemminger808f5112010-02-22 07:57:18 +00004627 return seq_hlist_next_rcu(v, &net->packet.sklist, pos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004628}
4629
4630static void packet_seq_stop(struct seq_file *seq, void *v)
stephen hemminger808f5112010-02-22 07:57:18 +00004631 __releases(RCU)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004632{
stephen hemminger808f5112010-02-22 07:57:18 +00004633 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004634}
4635
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09004636static int packet_seq_show(struct seq_file *seq, void *v)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004637{
4638 if (v == SEQ_START_TOKEN)
Baruch Siachabdcd062020-12-16 09:28:04 +02004639 seq_printf(seq,
4640 "%*sRefCnt Type Proto Iface R Rmem User Inode\n",
4641 IS_ENABLED(CONFIG_64BIT) ? -17 : -9, "sk");
Linus Torvalds1da177e2005-04-16 15:20:36 -07004642 else {
Li Zefanb7ceabd2010-02-08 23:19:29 +00004643 struct sock *s = sk_entry(v);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004644 const struct packet_sock *po = pkt_sk(s);
4645
4646 seq_printf(seq,
Dan Rosenberg71338aa2011-05-23 12:17:35 +00004647 "%pK %-6d %-4d %04x %-5d %1d %-6u %-6u %-6lu\n",
Linus Torvalds1da177e2005-04-16 15:20:36 -07004648 s,
Reshetova, Elena41c6d652017-06-30 13:08:01 +03004649 refcount_read(&s->sk_refcnt),
Linus Torvalds1da177e2005-04-16 15:20:36 -07004650 s->sk_type,
Eric Dumazetc7d2ef52021-06-16 06:42:01 -07004651 ntohs(READ_ONCE(po->num)),
Eric Dumazete032f7c2021-06-16 06:42:02 -07004652 READ_ONCE(po->ifindex),
Linus Torvalds1da177e2005-04-16 15:20:36 -07004653 po->running,
4654 atomic_read(&s->sk_rmem_alloc),
Eric W. Biedermana7cb5a42012-05-24 01:10:10 -06004655 from_kuid_munged(seq_user_ns(seq), sock_i_uid(s)),
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004656 sock_i_ino(s));
Linus Torvalds1da177e2005-04-16 15:20:36 -07004657 }
4658
4659 return 0;
4660}
4661
Philippe De Muyter56b3d972007-07-10 23:07:31 -07004662static const struct seq_operations packet_seq_ops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004663 .start = packet_seq_start,
4664 .next = packet_seq_next,
4665 .stop = packet_seq_stop,
4666 .show = packet_seq_show,
4667};
Linus Torvalds1da177e2005-04-16 15:20:36 -07004668#endif
4669
Alexey Dobriyan2c8c1e72010-01-17 03:35:32 +00004670static int __net_init packet_net_init(struct net *net)
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08004671{
Pavel Emelyanov0fa7fa92012-08-21 01:06:47 +00004672 mutex_init(&net->packet.sklist_lock);
Denis V. Lunev2aaef4e2007-12-11 04:19:54 -08004673 INIT_HLIST_HEAD(&net->packet.sklist);
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08004674
Yonatan Linika268e0f2020-12-14 22:25:50 +02004675#ifdef CONFIG_PROC_FS
Christoph Hellwigc3506372018-04-10 19:42:55 +02004676 if (!proc_create_net("packet", 0, net->proc_net, &packet_seq_ops,
4677 sizeof(struct seq_net_private)))
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08004678 return -ENOMEM;
Yonatan Linika268e0f2020-12-14 22:25:50 +02004679#endif /* CONFIG_PROC_FS */
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08004680
4681 return 0;
4682}
4683
Alexey Dobriyan2c8c1e72010-01-17 03:35:32 +00004684static void __net_exit packet_net_exit(struct net *net)
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08004685{
Gao fengece31ff2013-02-18 01:34:56 +00004686 remove_proc_entry("packet", net->proc_net);
Vasily Averin669f8f12017-11-12 22:27:49 +03004687 WARN_ON_ONCE(!hlist_empty(&net->packet.sklist));
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08004688}
4689
4690static struct pernet_operations packet_net_ops = {
4691 .init = packet_net_init,
4692 .exit = packet_net_exit,
4693};
4694
4695
Linus Torvalds1da177e2005-04-16 15:20:36 -07004696static void __exit packet_exit(void)
4697{
Linus Torvalds1da177e2005-04-16 15:20:36 -07004698 unregister_netdevice_notifier(&packet_netdev_notifier);
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08004699 unregister_pernet_subsys(&packet_net_ops);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004700 sock_unregister(PF_PACKET);
4701 proto_unregister(&packet_proto);
4702}
4703
4704static int __init packet_init(void)
4705{
YueHaibing36096f22019-05-09 22:52:20 +08004706 int rc;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004707
YueHaibing36096f22019-05-09 22:52:20 +08004708 rc = proto_register(&packet_proto, 0);
4709 if (rc)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004710 goto out;
YueHaibing36096f22019-05-09 22:52:20 +08004711 rc = sock_register(&packet_family_ops);
4712 if (rc)
4713 goto out_proto;
4714 rc = register_pernet_subsys(&packet_net_ops);
4715 if (rc)
4716 goto out_sock;
4717 rc = register_netdevice_notifier(&packet_netdev_notifier);
4718 if (rc)
4719 goto out_pernet;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004720
YueHaibing36096f22019-05-09 22:52:20 +08004721 return 0;
4722
4723out_pernet:
4724 unregister_pernet_subsys(&packet_net_ops);
4725out_sock:
4726 sock_unregister(PF_PACKET);
4727out_proto:
4728 proto_unregister(&packet_proto);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004729out:
4730 return rc;
4731}
4732
4733module_init(packet_init);
4734module_exit(packet_exit);
4735MODULE_LICENSE("GPL");
4736MODULE_ALIAS_NETPROTO(PF_PACKET);