blob: 5bd409ab4cc2001f2ac2d045e77f96c8bbba956a [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) {
Eric Dumazetf1d92682021-12-14 07:09:33 -08003113 dev_put_track(po->prot_hook.dev, &po->prot_hook.dev_tracker);
Ben Greear160ff182011-06-01 07:18:52 +00003114 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);
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003165 struct net_device *dev = NULL;
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003166 bool unlisted = false;
Eric Dumazetbf440772022-01-07 10:39:53 -08003167 bool need_rehook;
3168 int ret = 0;
David S. Millerdc99f602011-07-05 01:45:05 -07003169
Linus Torvalds1da177e2005-04-16 15:20:36 -07003170 lock_sock(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003171 spin_lock(&po->bind_lock);
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003172 rcu_read_lock();
3173
Willem de Bruijn49716132017-09-26 12:19:37 -04003174 if (po->fanout) {
3175 ret = -EINVAL;
3176 goto out_unlock;
3177 }
3178
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003179 if (name) {
3180 dev = dev_get_by_name_rcu(sock_net(sk), name);
3181 if (!dev) {
3182 ret = -ENODEV;
3183 goto out_unlock;
3184 }
3185 } else if (ifindex) {
3186 dev = dev_get_by_index_rcu(sock_net(sk), ifindex);
3187 if (!dev) {
3188 ret = -ENODEV;
3189 goto out_unlock;
3190 }
3191 }
3192
Eric Dumazetbf440772022-01-07 10:39:53 -08003193 need_rehook = po->prot_hook.type != proto || po->prot_hook.dev != dev;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003194
Daniel Borkmann902fefb2014-01-15 16:25:34 +01003195 if (need_rehook) {
Eric Dumazetbf440772022-01-07 10:39:53 -08003196 dev_hold(dev);
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003197 if (po->running) {
3198 rcu_read_unlock();
Eric Dumazet15fe0762017-11-28 08:03:30 -08003199 /* prevents packet_notifier() from calling
3200 * register_prot_hook()
3201 */
Eric Dumazetc7d2ef52021-06-16 06:42:01 -07003202 WRITE_ONCE(po->num, 0);
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003203 __unregister_prot_hook(sk, true);
3204 rcu_read_lock();
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003205 if (dev)
3206 unlisted = !dev_get_by_index_rcu(sock_net(sk),
3207 dev->ifindex);
3208 }
Daniel Borkmann66e56cd2013-12-06 11:36:15 +01003209
Eric Dumazet15fe0762017-11-28 08:03:30 -08003210 BUG_ON(po->running);
Eric Dumazetc7d2ef52021-06-16 06:42:01 -07003211 WRITE_ONCE(po->num, proto);
Daniel Borkmann902fefb2014-01-15 16:25:34 +01003212 po->prot_hook.type = proto;
Daniel Borkmann902fefb2014-01-15 16:25:34 +01003213
Eric Dumazetbf440772022-01-07 10:39:53 -08003214 dev_put_track(po->prot_hook.dev, &po->prot_hook.dev_tracker);
Eric Dumazetf1d92682021-12-14 07:09:33 -08003215
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003216 if (unlikely(unlisted)) {
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003217 po->prot_hook.dev = NULL;
Eric Dumazete032f7c2021-06-16 06:42:02 -07003218 WRITE_ONCE(po->ifindex, -1);
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003219 packet_cached_dev_reset(po);
3220 } else {
Eric Dumazetbf440772022-01-07 10:39:53 -08003221 dev_hold_track(dev, &po->prot_hook.dev_tracker,
3222 GFP_ATOMIC);
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003223 po->prot_hook.dev = dev;
Eric Dumazete032f7c2021-06-16 06:42:02 -07003224 WRITE_ONCE(po->ifindex, dev ? dev->ifindex : 0);
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003225 packet_cached_dev_assign(po, dev);
3226 }
Eric Dumazetbf440772022-01-07 10:39:53 -08003227 dev_put(dev);
Daniel Borkmann902fefb2014-01-15 16:25:34 +01003228 }
3229
3230 if (proto == 0 || !need_rehook)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003231 goto out_unlock;
3232
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003233 if (!unlisted && (!dev || (dev->flags & IFF_UP))) {
David S. Millerce06b032011-07-04 01:44:29 -07003234 register_prot_hook(sk);
Urs Thuermannbe85d4a2007-11-12 21:05:20 -08003235 } else {
3236 sk->sk_err = ENETDOWN;
3237 if (!sock_flag(sk, SOCK_DEAD))
Alexander Aringe3ae2362021-06-27 18:48:21 -04003238 sk_error_report(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003239 }
3240
3241out_unlock:
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003242 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07003243 spin_unlock(&po->bind_lock);
3244 release_sock(sk);
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003245 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003246}
3247
3248/*
3249 * Bind a packet socket to a device
3250 */
3251
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003252static int packet_bind_spkt(struct socket *sock, struct sockaddr *uaddr,
3253 int addr_len)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003254{
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003255 struct sock *sk = sock->sk;
Alexander Potapenko540e2892017-03-01 12:57:20 +01003256 char name[sizeof(uaddr->sa_data) + 1];
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09003257
Linus Torvalds1da177e2005-04-16 15:20:36 -07003258 /*
3259 * Check legality
3260 */
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09003261
Kris Katterjohn8ae55f02006-01-23 16:28:02 -08003262 if (addr_len != sizeof(struct sockaddr))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003263 return -EINVAL;
Alexander Potapenko540e2892017-03-01 12:57:20 +01003264 /* uaddr->sa_data comes from the userspace, it's not guaranteed to be
3265 * zero-terminated.
3266 */
3267 memcpy(name, uaddr->sa_data, sizeof(uaddr->sa_data));
3268 name[sizeof(uaddr->sa_data)] = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003269
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003270 return packet_do_bind(sk, name, 0, pkt_sk(sk)->num);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003271}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003272
3273static int packet_bind(struct socket *sock, struct sockaddr *uaddr, int addr_len)
3274{
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003275 struct sockaddr_ll *sll = (struct sockaddr_ll *)uaddr;
3276 struct sock *sk = sock->sk;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003277
3278 /*
3279 * Check legality
3280 */
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09003281
Linus Torvalds1da177e2005-04-16 15:20:36 -07003282 if (addr_len < sizeof(struct sockaddr_ll))
3283 return -EINVAL;
3284 if (sll->sll_family != AF_PACKET)
3285 return -EINVAL;
3286
Francesco Ruggeri30f7ea12015-11-05 08:16:14 -08003287 return packet_do_bind(sk, NULL, sll->sll_ifindex,
3288 sll->sll_protocol ? : pkt_sk(sk)->num);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003289}
3290
3291static struct proto packet_proto = {
3292 .name = "PACKET",
3293 .owner = THIS_MODULE,
3294 .obj_size = sizeof(struct packet_sock),
3295};
3296
3297/*
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09003298 * Create a packet of type SOCK_PACKET.
Linus Torvalds1da177e2005-04-16 15:20:36 -07003299 */
3300
Eric Paris3f378b62009-11-05 22:18:14 -08003301static int packet_create(struct net *net, struct socket *sock, int protocol,
3302 int kern)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003303{
3304 struct sock *sk;
3305 struct packet_sock *po;
Al Viro0e11c912006-11-08 00:26:29 -08003306 __be16 proto = (__force __be16)protocol; /* weird, but documented */
Linus Torvalds1da177e2005-04-16 15:20:36 -07003307 int err;
3308
Eric W. Biedermandf008c92012-11-16 03:03:07 +00003309 if (!ns_capable(net->user_ns, CAP_NET_RAW))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003310 return -EPERM;
David S. Millerbe020972007-05-29 13:16:31 -07003311 if (sock->type != SOCK_DGRAM && sock->type != SOCK_RAW &&
3312 sock->type != SOCK_PACKET)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003313 return -ESOCKTNOSUPPORT;
3314
3315 sock->state = SS_UNCONNECTED;
3316
3317 err = -ENOBUFS;
Eric W. Biederman11aa9c22015-05-08 21:09:13 -05003318 sk = sk_alloc(net, PF_PACKET, GFP_KERNEL, &packet_proto, kern);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003319 if (sk == NULL)
3320 goto out;
3321
3322 sock->ops = &packet_ops;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003323 if (sock->type == SOCK_PACKET)
3324 sock->ops = &packet_ops_spkt;
David S. Millerbe020972007-05-29 13:16:31 -07003325
Linus Torvalds1da177e2005-04-16 15:20:36 -07003326 sock_init_data(sock, sk);
3327
3328 po = pkt_sk(sk);
Neil Horman89ed5b52019-06-25 17:57:49 -04003329 init_completion(&po->skb_completion);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003330 sk->sk_family = PF_PACKET;
Al Viro0e11c912006-11-08 00:26:29 -08003331 po->num = proto;
Daniel Borkmannd346a3f2013-12-06 11:36:17 +01003332 po->xmit = dev_queue_xmit;
Daniel Borkmann66e56cd2013-12-06 11:36:15 +01003333
Daniel Borkmannb0138402014-01-15 16:25:36 +01003334 err = packet_alloc_pending(po);
3335 if (err)
3336 goto out2;
3337
Daniel Borkmann66e56cd2013-12-06 11:36:15 +01003338 packet_cached_dev_reset(po);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003339
3340 sk->sk_destruct = packet_sock_destruct;
Pavel Emelyanov17ab56a2007-11-10 21:38:48 -08003341 sk_refcnt_debug_inc(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003342
3343 /*
3344 * Attach a protocol block
3345 */
3346
3347 spin_lock_init(&po->bind_lock);
Herbert Xu905db442009-01-30 14:12:06 -08003348 mutex_init(&po->pg_vec_lock);
Willem de Bruijn0648ab72015-05-12 11:56:46 -04003349 po->rollover = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003350 po->prot_hook.func = packet_rcv;
David S. Millerbe020972007-05-29 13:16:31 -07003351
Linus Torvalds1da177e2005-04-16 15:20:36 -07003352 if (sock->type == SOCK_PACKET)
3353 po->prot_hook.func = packet_rcv_spkt;
David S. Millerbe020972007-05-29 13:16:31 -07003354
Linus Torvalds1da177e2005-04-16 15:20:36 -07003355 po->prot_hook.af_packet_priv = sk;
3356
Al Viro0e11c912006-11-08 00:26:29 -08003357 if (proto) {
3358 po->prot_hook.type = proto;
Willem de Bruijna6361f02018-04-23 17:37:03 -04003359 __register_prot_hook(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003360 }
3361
Pavel Emelyanov0fa7fa92012-08-21 01:06:47 +00003362 mutex_lock(&net->packet.sklist_lock);
Maxime Chevalliera4dc6a42019-03-16 14:41:30 +01003363 sk_add_node_tail_rcu(sk, &net->packet.sklist);
Pavel Emelyanov0fa7fa92012-08-21 01:06:47 +00003364 mutex_unlock(&net->packet.sklist_lock);
3365
Eric Dumazet36804532008-11-19 14:25:35 -08003366 sock_prot_inuse_add(net, &packet_proto, 1);
stephen hemminger808f5112010-02-22 07:57:18 +00003367
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003368 return 0;
Daniel Borkmannb0138402014-01-15 16:25:36 +01003369out2:
3370 sk_free(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003371out:
3372 return err;
3373}
3374
3375/*
3376 * Pull a packet from our receive queue and hand it to the user.
3377 * If necessary we block.
3378 */
3379
Ying Xue1b784142015-03-02 15:37:48 +08003380static int packet_recvmsg(struct socket *sock, struct msghdr *msg, size_t len,
3381 int flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003382{
3383 struct sock *sk = sock->sk;
3384 struct sk_buff *skb;
3385 int copied, err;
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003386 int vnet_hdr_len = 0;
Eyal Birger2472d762015-03-01 14:58:28 +02003387 unsigned int origlen = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003388
3389 err = -EINVAL;
Richard Cochraned85b562010-04-07 22:41:28 +00003390 if (flags & ~(MSG_PEEK|MSG_DONTWAIT|MSG_TRUNC|MSG_CMSG_COMPAT|MSG_ERRQUEUE))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003391 goto out;
3392
3393#if 0
3394 /* What error should we return now? EUNATTACH? */
3395 if (pkt_sk(sk)->ifindex < 0)
3396 return -ENODEV;
3397#endif
3398
Richard Cochraned85b562010-04-07 22:41:28 +00003399 if (flags & MSG_ERRQUEUE) {
Richard Cochrancb820f82013-07-19 19:40:09 +02003400 err = sock_recv_errqueue(sk, msg, len,
3401 SOL_PACKET, PACKET_TX_TIMESTAMP);
Richard Cochraned85b562010-04-07 22:41:28 +00003402 goto out;
3403 }
3404
Linus Torvalds1da177e2005-04-16 15:20:36 -07003405 /*
Linus Torvalds1da177e2005-04-16 15:20:36 -07003406 * Call the generic datagram receiver. This handles all sorts
3407 * of horrible races and re-entrancy so we can forget about it
3408 * in the protocol layers.
3409 *
3410 * Now it will return ENETDOWN, if device have just gone down,
3411 * but then it will block.
3412 */
3413
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003414 skb = skb_recv_datagram(sk, flags, flags & MSG_DONTWAIT, &err);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003415
3416 /*
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09003417 * An error occurred so return it. Because skb_recv_datagram()
Linus Torvalds1da177e2005-04-16 15:20:36 -07003418 * handles the blocking we don't see and worry about blocking
3419 * retries.
3420 */
3421
Kris Katterjohn8ae55f02006-01-23 16:28:02 -08003422 if (skb == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003423 goto out;
3424
Eric Dumazet9bb6cd62019-06-12 09:52:33 -07003425 packet_rcv_try_clear_pressure(pkt_sk(sk));
Willem de Bruijn2ccdbaa2015-05-12 11:56:48 -04003426
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003427 if (pkt_sk(sk)->has_vnet_hdr) {
Willem de Bruijn16cc1402016-02-03 18:02:14 -05003428 err = packet_rcv_vnet(msg, skb, &len);
3429 if (err)
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003430 goto out_free;
Willem de Bruijn16cc1402016-02-03 18:02:14 -05003431 vnet_hdr_len = sizeof(struct virtio_net_hdr);
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003432 }
3433
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003434 /* You lose any data beyond the buffer you gave. If it worries
3435 * a user program they can ask the device for its MTU
3436 * anyway.
Eric W. Biederman0fb375f2005-09-21 00:11:37 -07003437 */
Linus Torvalds1da177e2005-04-16 15:20:36 -07003438 copied = skb->len;
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003439 if (copied > len) {
3440 copied = len;
3441 msg->msg_flags |= MSG_TRUNC;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003442 }
3443
David S. Miller51f3d022014-11-05 16:46:40 -05003444 err = skb_copy_datagram_msg(skb, 0, msg, copied);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003445 if (err)
3446 goto out_free;
3447
Eyal Birger2472d762015-03-01 14:58:28 +02003448 if (sock->type != SOCK_PACKET) {
3449 struct sockaddr_ll *sll = &PACKET_SKB_CB(skb)->sa.ll;
3450
3451 /* Original length was stored in sockaddr_ll fields */
3452 origlen = PACKET_SKB_CB(skb)->sa.origlen;
3453 sll->sll_family = AF_PACKET;
3454 sll->sll_protocol = skb->protocol;
3455 }
3456
Neil Horman3b885782009-10-12 13:26:31 -07003457 sock_recv_ts_and_drops(msg, sk, skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003458
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003459 if (msg->msg_name) {
Willem de Bruijnb2cf86e2019-04-29 11:46:55 -04003460 int copy_len;
3461
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003462 /* If the address length field is there to be filled
3463 * in, we fill it in now.
3464 */
3465 if (sock->type == SOCK_PACKET) {
Steffen Hurrle342dfc32014-01-17 22:53:15 +01003466 __sockaddr_check_size(sizeof(struct sockaddr_pkt));
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003467 msg->msg_namelen = sizeof(struct sockaddr_pkt);
Willem de Bruijnb2cf86e2019-04-29 11:46:55 -04003468 copy_len = msg->msg_namelen;
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003469 } else {
3470 struct sockaddr_ll *sll = &PACKET_SKB_CB(skb)->sa.ll;
Eyal Birger2472d762015-03-01 14:58:28 +02003471
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003472 msg->msg_namelen = sll->sll_halen +
3473 offsetof(struct sockaddr_ll, sll_addr);
Willem de Bruijnb2cf86e2019-04-29 11:46:55 -04003474 copy_len = msg->msg_namelen;
3475 if (msg->msg_namelen < sizeof(struct sockaddr_ll)) {
3476 memset(msg->msg_name +
3477 offsetof(struct sockaddr_ll, sll_addr),
3478 0, sizeof(sll->sll_addr));
3479 msg->msg_namelen = sizeof(struct sockaddr_ll);
3480 }
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003481 }
Willem de Bruijnb2cf86e2019-04-29 11:46:55 -04003482 memcpy(msg->msg_name, &PACKET_SKB_CB(skb)->sa, copy_len);
Hannes Frederic Sowaf3d33422013-11-21 03:14:22 +01003483 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07003484
Herbert Xu8dc41942007-02-04 23:31:32 -08003485 if (pkt_sk(sk)->auxdata) {
Herbert Xuffbc6112007-02-04 23:33:10 -08003486 struct tpacket_auxdata aux;
3487
3488 aux.tp_status = TP_STATUS_USER;
3489 if (skb->ip_summed == CHECKSUM_PARTIAL)
3490 aux.tp_status |= TP_STATUS_CSUMNOTREADY;
Alexander Drozdov682f0482015-03-23 09:11:13 +03003491 else if (skb->pkt_type != PACKET_OUTGOING &&
3492 (skb->ip_summed == CHECKSUM_COMPLETE ||
3493 skb_csum_unnecessary(skb)))
3494 aux.tp_status |= TP_STATUS_CSUM_VALID;
3495
Eyal Birger2472d762015-03-01 14:58:28 +02003496 aux.tp_len = origlen;
Herbert Xuffbc6112007-02-04 23:33:10 -08003497 aux.tp_snaplen = skb->len;
3498 aux.tp_mac = 0;
Arnaldo Carvalho de Melobbe735e2007-03-10 22:16:10 -03003499 aux.tp_net = skb_network_offset(skb);
Jiri Pirkodf8a39d2015-01-13 17:13:44 +01003500 if (skb_vlan_tag_present(skb)) {
3501 aux.tp_vlan_tci = skb_vlan_tag_get(skb);
Atzm Watanabea0cdfcf2013-12-17 22:53:40 +09003502 aux.tp_vlan_tpid = ntohs(skb->vlan_proto);
3503 aux.tp_status |= TP_STATUS_VLAN_VALID | TP_STATUS_VLAN_TPID_VALID;
Ben Greeara3bcc232011-06-01 06:49:10 +00003504 } else {
3505 aux.tp_vlan_tci = 0;
Atzm Watanabea0cdfcf2013-12-17 22:53:40 +09003506 aux.tp_vlan_tpid = 0;
Ben Greeara3bcc232011-06-01 06:49:10 +00003507 }
Herbert Xuffbc6112007-02-04 23:33:10 -08003508 put_cmsg(msg, SOL_PACKET, PACKET_AUXDATA, sizeof(aux), &aux);
Herbert Xu8dc41942007-02-04 23:31:32 -08003509 }
3510
Linus Torvalds1da177e2005-04-16 15:20:36 -07003511 /*
3512 * Free or return the buffer as appropriate. Again this
3513 * hides all the races and re-entrancy issues from us.
3514 */
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003515 err = vnet_hdr_len + ((flags&MSG_TRUNC) ? skb->len : copied);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003516
3517out_free:
3518 skb_free_datagram(sk, skb);
3519out:
3520 return err;
3521}
3522
Linus Torvalds1da177e2005-04-16 15:20:36 -07003523static int packet_getname_spkt(struct socket *sock, struct sockaddr *uaddr,
Denys Vlasenko9b2c45d2018-02-12 20:00:20 +01003524 int peer)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003525{
3526 struct net_device *dev;
3527 struct sock *sk = sock->sk;
3528
3529 if (peer)
3530 return -EOPNOTSUPP;
3531
3532 uaddr->sa_family = AF_PACKET;
Daniel Borkmann2dc85bf2013-06-12 16:02:27 +02003533 memset(uaddr->sa_data, 0, sizeof(uaddr->sa_data));
Eric Dumazet654d1f82009-11-02 10:43:32 +01003534 rcu_read_lock();
Eric Dumazete032f7c2021-06-16 06:42:02 -07003535 dev = dev_get_by_index_rcu(sock_net(sk), READ_ONCE(pkt_sk(sk)->ifindex));
Eric Dumazet654d1f82009-11-02 10:43:32 +01003536 if (dev)
Daniel Borkmann2dc85bf2013-06-12 16:02:27 +02003537 strlcpy(uaddr->sa_data, dev->name, sizeof(uaddr->sa_data));
Eric Dumazet654d1f82009-11-02 10:43:32 +01003538 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07003539
Denys Vlasenko9b2c45d2018-02-12 20:00:20 +01003540 return sizeof(*uaddr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003541}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003542
3543static int packet_getname(struct socket *sock, struct sockaddr *uaddr,
Denys Vlasenko9b2c45d2018-02-12 20:00:20 +01003544 int peer)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003545{
3546 struct net_device *dev;
3547 struct sock *sk = sock->sk;
3548 struct packet_sock *po = pkt_sk(sk);
Cyrill Gorcunov13cfa972009-11-08 05:51:19 +00003549 DECLARE_SOCKADDR(struct sockaddr_ll *, sll, uaddr);
Eric Dumazete032f7c2021-06-16 06:42:02 -07003550 int ifindex;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003551
3552 if (peer)
3553 return -EOPNOTSUPP;
3554
Eric Dumazete032f7c2021-06-16 06:42:02 -07003555 ifindex = READ_ONCE(po->ifindex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003556 sll->sll_family = AF_PACKET;
Eric Dumazete032f7c2021-06-16 06:42:02 -07003557 sll->sll_ifindex = ifindex;
Eric Dumazetc7d2ef52021-06-16 06:42:01 -07003558 sll->sll_protocol = READ_ONCE(po->num);
Vasiliy Kulikov67286642010-11-10 12:09:10 -08003559 sll->sll_pkttype = 0;
Eric Dumazet654d1f82009-11-02 10:43:32 +01003560 rcu_read_lock();
Eric Dumazete032f7c2021-06-16 06:42:02 -07003561 dev = dev_get_by_index_rcu(sock_net(sk), ifindex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003562 if (dev) {
3563 sll->sll_hatype = dev->type;
3564 sll->sll_halen = dev->addr_len;
3565 memcpy(sll->sll_addr, dev->dev_addr, dev->addr_len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003566 } else {
3567 sll->sll_hatype = 0; /* Bad: we have no ARPHRD_UNSPEC */
3568 sll->sll_halen = 0;
3569 }
Eric Dumazet654d1f82009-11-02 10:43:32 +01003570 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07003571
Denys Vlasenko9b2c45d2018-02-12 20:00:20 +01003572 return offsetof(struct sockaddr_ll, sll_addr) + sll->sll_halen;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003573}
3574
Wang Chen2aeb0b82008-07-14 20:49:46 -07003575static int packet_dev_mc(struct net_device *dev, struct packet_mclist *i,
3576 int what)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003577{
3578 switch (i->type) {
3579 case PACKET_MR_MULTICAST:
Jiri Pirko11625632010-03-02 20:40:01 +00003580 if (i->alen != dev->addr_len)
3581 return -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003582 if (what > 0)
Jiri Pirko22bedad32010-04-01 21:22:57 +00003583 return dev_mc_add(dev, i->addr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003584 else
Jiri Pirko22bedad32010-04-01 21:22:57 +00003585 return dev_mc_del(dev, i->addr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003586 break;
3587 case PACKET_MR_PROMISC:
Wang Chen2aeb0b82008-07-14 20:49:46 -07003588 return dev_set_promiscuity(dev, what);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003589 case PACKET_MR_ALLMULTI:
Wang Chen2aeb0b82008-07-14 20:49:46 -07003590 return dev_set_allmulti(dev, what);
Eric W. Biedermand95ed922009-05-19 18:27:17 +00003591 case PACKET_MR_UNICAST:
Jiri Pirko11625632010-03-02 20:40:01 +00003592 if (i->alen != dev->addr_len)
3593 return -EINVAL;
Eric W. Biedermand95ed922009-05-19 18:27:17 +00003594 if (what > 0)
Jiri Pirkoa748ee22010-04-01 21:22:09 +00003595 return dev_uc_add(dev, i->addr);
Eric W. Biedermand95ed922009-05-19 18:27:17 +00003596 else
Jiri Pirkoa748ee22010-04-01 21:22:09 +00003597 return dev_uc_del(dev, i->addr);
Eric W. Biedermand95ed922009-05-19 18:27:17 +00003598 break;
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003599 default:
3600 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003601 }
Wang Chen2aeb0b82008-07-14 20:49:46 -07003602 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003603}
3604
Francesco Ruggeri82f17092015-03-09 11:51:04 -07003605static void packet_dev_mclist_delete(struct net_device *dev,
3606 struct packet_mclist **mlp)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003607{
Francesco Ruggeri82f17092015-03-09 11:51:04 -07003608 struct packet_mclist *ml;
3609
3610 while ((ml = *mlp) != NULL) {
3611 if (ml->ifindex == dev->ifindex) {
3612 packet_dev_mc(dev, ml, -1);
3613 *mlp = ml->next;
3614 kfree(ml);
3615 } else
3616 mlp = &ml->next;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003617 }
3618}
3619
Eric W. Biederman0fb375f2005-09-21 00:11:37 -07003620static int packet_mc_add(struct sock *sk, struct packet_mreq_max *mreq)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003621{
3622 struct packet_sock *po = pkt_sk(sk);
3623 struct packet_mclist *ml, *i;
3624 struct net_device *dev;
3625 int err;
3626
3627 rtnl_lock();
3628
3629 err = -ENODEV;
YOSHIFUJI Hideaki3b1e0a62008-03-26 02:26:21 +09003630 dev = __dev_get_by_index(sock_net(sk), mreq->mr_ifindex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003631 if (!dev)
3632 goto done;
3633
3634 err = -EINVAL;
Jiri Pirko11625632010-03-02 20:40:01 +00003635 if (mreq->mr_alen > dev->addr_len)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003636 goto done;
3637
3638 err = -ENOBUFS;
Kris Katterjohn8b3a7002006-01-11 15:56:43 -08003639 i = kmalloc(sizeof(*i), GFP_KERNEL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003640 if (i == NULL)
3641 goto done;
3642
3643 err = 0;
3644 for (ml = po->mclist; ml; ml = ml->next) {
3645 if (ml->ifindex == mreq->mr_ifindex &&
3646 ml->type == mreq->mr_type &&
3647 ml->alen == mreq->mr_alen &&
3648 memcmp(ml->addr, mreq->mr_address, ml->alen) == 0) {
3649 ml->count++;
3650 /* Free the new element ... */
3651 kfree(i);
3652 goto done;
3653 }
3654 }
3655
3656 i->type = mreq->mr_type;
3657 i->ifindex = mreq->mr_ifindex;
3658 i->alen = mreq->mr_alen;
3659 memcpy(i->addr, mreq->mr_address, i->alen);
Mathias Krause309cf372016-04-10 12:52:28 +02003660 memset(i->addr + i->alen, 0, sizeof(i->addr) - i->alen);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003661 i->count = 1;
3662 i->next = po->mclist;
3663 po->mclist = i;
Wang Chen2aeb0b82008-07-14 20:49:46 -07003664 err = packet_dev_mc(dev, i, 1);
3665 if (err) {
3666 po->mclist = i->next;
3667 kfree(i);
3668 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07003669
3670done:
3671 rtnl_unlock();
3672 return err;
3673}
3674
Eric W. Biederman0fb375f2005-09-21 00:11:37 -07003675static int packet_mc_drop(struct sock *sk, struct packet_mreq_max *mreq)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003676{
3677 struct packet_mclist *ml, **mlp;
3678
3679 rtnl_lock();
3680
3681 for (mlp = &pkt_sk(sk)->mclist; (ml = *mlp) != NULL; mlp = &ml->next) {
3682 if (ml->ifindex == mreq->mr_ifindex &&
3683 ml->type == mreq->mr_type &&
3684 ml->alen == mreq->mr_alen &&
3685 memcmp(ml->addr, mreq->mr_address, ml->alen) == 0) {
3686 if (--ml->count == 0) {
3687 struct net_device *dev;
3688 *mlp = ml->next;
Eric Dumazetad959e72009-10-16 06:38:46 +00003689 dev = __dev_get_by_index(sock_net(sk), ml->ifindex);
3690 if (dev)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003691 packet_dev_mc(dev, ml, -1);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003692 kfree(ml);
3693 }
Francesco Ruggeri82f17092015-03-09 11:51:04 -07003694 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003695 }
3696 }
3697 rtnl_unlock();
Francesco Ruggeri82f17092015-03-09 11:51:04 -07003698 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003699}
3700
3701static void packet_flush_mclist(struct sock *sk)
3702{
3703 struct packet_sock *po = pkt_sk(sk);
3704 struct packet_mclist *ml;
3705
3706 if (!po->mclist)
3707 return;
3708
3709 rtnl_lock();
3710 while ((ml = po->mclist) != NULL) {
3711 struct net_device *dev;
3712
3713 po->mclist = ml->next;
Eric Dumazetad959e72009-10-16 06:38:46 +00003714 dev = __dev_get_by_index(sock_net(sk), ml->ifindex);
3715 if (dev != NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003716 packet_dev_mc(dev, ml, -1);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003717 kfree(ml);
3718 }
3719 rtnl_unlock();
3720}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003721
3722static int
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003723packet_setsockopt(struct socket *sock, int level, int optname, sockptr_t optval,
3724 unsigned int optlen)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003725{
3726 struct sock *sk = sock->sk;
Herbert Xu8dc41942007-02-04 23:31:32 -08003727 struct packet_sock *po = pkt_sk(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003728 int ret;
3729
3730 if (level != SOL_PACKET)
3731 return -ENOPROTOOPT;
3732
Johann Baudy69e3c752009-05-18 22:11:22 -07003733 switch (optname) {
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09003734 case PACKET_ADD_MEMBERSHIP:
Linus Torvalds1da177e2005-04-16 15:20:36 -07003735 case PACKET_DROP_MEMBERSHIP:
3736 {
Eric W. Biederman0fb375f2005-09-21 00:11:37 -07003737 struct packet_mreq_max mreq;
3738 int len = optlen;
3739 memset(&mreq, 0, sizeof(mreq));
3740 if (len < sizeof(struct packet_mreq))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003741 return -EINVAL;
Eric W. Biederman0fb375f2005-09-21 00:11:37 -07003742 if (len > sizeof(mreq))
3743 len = sizeof(mreq);
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003744 if (copy_from_sockptr(&mreq, optval, len))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003745 return -EFAULT;
Eric W. Biederman0fb375f2005-09-21 00:11:37 -07003746 if (len < (mreq.mr_alen + offsetof(struct packet_mreq, mr_address)))
3747 return -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003748 if (optname == PACKET_ADD_MEMBERSHIP)
3749 ret = packet_mc_add(sk, &mreq);
3750 else
3751 ret = packet_mc_drop(sk, &mreq);
3752 return ret;
3753 }
David S. Millera2efcfa2007-05-29 13:12:50 -07003754
Linus Torvalds1da177e2005-04-16 15:20:36 -07003755 case PACKET_RX_RING:
Johann Baudy69e3c752009-05-18 22:11:22 -07003756 case PACKET_TX_RING:
Linus Torvalds1da177e2005-04-16 15:20:36 -07003757 {
chetan lokef6fb8f12011-08-19 10:18:16 +00003758 union tpacket_req_u req_u;
3759 int len;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003760
Eric Dumazet5171b372018-04-15 17:52:04 -07003761 lock_sock(sk);
chetan lokef6fb8f12011-08-19 10:18:16 +00003762 switch (po->tp_version) {
3763 case TPACKET_V1:
3764 case TPACKET_V2:
3765 len = sizeof(req_u.req);
3766 break;
3767 case TPACKET_V3:
3768 default:
3769 len = sizeof(req_u.req3);
3770 break;
3771 }
Eric Dumazet5171b372018-04-15 17:52:04 -07003772 if (optlen < len) {
3773 ret = -EINVAL;
3774 } else {
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003775 if (copy_from_sockptr(&req_u.req, optval, len))
Eric Dumazet5171b372018-04-15 17:52:04 -07003776 ret = -EFAULT;
3777 else
3778 ret = packet_set_ring(sk, &req_u, 0,
3779 optname == PACKET_TX_RING);
3780 }
3781 release_sock(sk);
3782 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003783 }
3784 case PACKET_COPY_THRESH:
3785 {
3786 int val;
3787
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00003788 if (optlen != sizeof(val))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003789 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003790 if (copy_from_sockptr(&val, optval, sizeof(val)))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003791 return -EFAULT;
3792
3793 pkt_sk(sk)->copy_thresh = val;
3794 return 0;
3795 }
Patrick McHardybbd6ef82008-07-14 22:50:15 -07003796 case PACKET_VERSION:
3797 {
3798 int val;
3799
3800 if (optlen != sizeof(val))
3801 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003802 if (copy_from_sockptr(&val, optval, sizeof(val)))
Patrick McHardybbd6ef82008-07-14 22:50:15 -07003803 return -EFAULT;
3804 switch (val) {
3805 case TPACKET_V1:
3806 case TPACKET_V2:
chetan lokef6fb8f12011-08-19 10:18:16 +00003807 case TPACKET_V3:
Philip Pettersson84ac7262016-11-30 14:55:36 -08003808 break;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07003809 default:
3810 return -EINVAL;
3811 }
Philip Pettersson84ac7262016-11-30 14:55:36 -08003812 lock_sock(sk);
3813 if (po->rx_ring.pg_vec || po->tx_ring.pg_vec) {
3814 ret = -EBUSY;
3815 } else {
3816 po->tp_version = val;
3817 ret = 0;
3818 }
3819 release_sock(sk);
3820 return ret;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07003821 }
Patrick McHardy8913336a2008-07-18 18:05:19 -07003822 case PACKET_RESERVE:
3823 {
3824 unsigned int val;
3825
3826 if (optlen != sizeof(val))
3827 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003828 if (copy_from_sockptr(&val, optval, sizeof(val)))
Patrick McHardy8913336a2008-07-18 18:05:19 -07003829 return -EFAULT;
Andrey Konovalovbcc53642017-03-29 16:11:22 +02003830 if (val > INT_MAX)
3831 return -EINVAL;
Willem de Bruijnc27927e2017-08-10 12:41:58 -04003832 lock_sock(sk);
3833 if (po->rx_ring.pg_vec || po->tx_ring.pg_vec) {
3834 ret = -EBUSY;
3835 } else {
3836 po->tp_reserve = val;
3837 ret = 0;
3838 }
3839 release_sock(sk);
3840 return ret;
Patrick McHardy8913336a2008-07-18 18:05:19 -07003841 }
Johann Baudy69e3c752009-05-18 22:11:22 -07003842 case PACKET_LOSS:
3843 {
3844 unsigned int val;
3845
3846 if (optlen != sizeof(val))
3847 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003848 if (copy_from_sockptr(&val, optval, sizeof(val)))
Johann Baudy69e3c752009-05-18 22:11:22 -07003849 return -EFAULT;
Willem de Bruijna6361f02018-04-23 17:37:03 -04003850
3851 lock_sock(sk);
3852 if (po->rx_ring.pg_vec || po->tx_ring.pg_vec) {
3853 ret = -EBUSY;
3854 } else {
3855 po->tp_loss = !!val;
3856 ret = 0;
3857 }
3858 release_sock(sk);
3859 return ret;
Johann Baudy69e3c752009-05-18 22:11:22 -07003860 }
Herbert Xu8dc41942007-02-04 23:31:32 -08003861 case PACKET_AUXDATA:
3862 {
3863 int val;
3864
3865 if (optlen < sizeof(val))
3866 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003867 if (copy_from_sockptr(&val, optval, sizeof(val)))
Herbert Xu8dc41942007-02-04 23:31:32 -08003868 return -EFAULT;
3869
Willem de Bruijna6361f02018-04-23 17:37:03 -04003870 lock_sock(sk);
Herbert Xu8dc41942007-02-04 23:31:32 -08003871 po->auxdata = !!val;
Willem de Bruijna6361f02018-04-23 17:37:03 -04003872 release_sock(sk);
Herbert Xu8dc41942007-02-04 23:31:32 -08003873 return 0;
3874 }
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07003875 case PACKET_ORIGDEV:
3876 {
3877 int val;
3878
3879 if (optlen < sizeof(val))
3880 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003881 if (copy_from_sockptr(&val, optval, sizeof(val)))
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07003882 return -EFAULT;
3883
Willem de Bruijna6361f02018-04-23 17:37:03 -04003884 lock_sock(sk);
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07003885 po->origdev = !!val;
Willem de Bruijna6361f02018-04-23 17:37:03 -04003886 release_sock(sk);
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07003887 return 0;
3888 }
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003889 case PACKET_VNET_HDR:
3890 {
3891 int val;
3892
3893 if (sock->type != SOCK_RAW)
3894 return -EINVAL;
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003895 if (optlen < sizeof(val))
3896 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003897 if (copy_from_sockptr(&val, optval, sizeof(val)))
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003898 return -EFAULT;
3899
Willem de Bruijna6361f02018-04-23 17:37:03 -04003900 lock_sock(sk);
3901 if (po->rx_ring.pg_vec || po->tx_ring.pg_vec) {
3902 ret = -EBUSY;
3903 } else {
3904 po->has_vnet_hdr = !!val;
3905 ret = 0;
3906 }
3907 release_sock(sk);
3908 return ret;
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08003909 }
Scott McMillan614f60f2010-06-02 05:53:56 -07003910 case PACKET_TIMESTAMP:
3911 {
3912 int val;
3913
3914 if (optlen != sizeof(val))
3915 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003916 if (copy_from_sockptr(&val, optval, sizeof(val)))
Scott McMillan614f60f2010-06-02 05:53:56 -07003917 return -EFAULT;
3918
3919 po->tp_tstamp = val;
3920 return 0;
3921 }
David S. Millerdc99f602011-07-05 01:45:05 -07003922 case PACKET_FANOUT:
3923 {
Tanner Love9c661b02020-11-06 13:07:40 -05003924 struct fanout_args args = { 0 };
David S. Millerdc99f602011-07-05 01:45:05 -07003925
Tanner Love9c661b02020-11-06 13:07:40 -05003926 if (optlen != sizeof(int) && optlen != sizeof(args))
David S. Millerdc99f602011-07-05 01:45:05 -07003927 return -EINVAL;
Tanner Love9c661b02020-11-06 13:07:40 -05003928 if (copy_from_sockptr(&args, optval, optlen))
David S. Millerdc99f602011-07-05 01:45:05 -07003929 return -EFAULT;
3930
Tanner Love9c661b02020-11-06 13:07:40 -05003931 return fanout_add(sk, &args);
David S. Millerdc99f602011-07-05 01:45:05 -07003932 }
Willem de Bruijn47dceb82015-08-14 22:31:34 -04003933 case PACKET_FANOUT_DATA:
3934 {
3935 if (!po->fanout)
3936 return -EINVAL;
3937
3938 return fanout_set_data(po, optval, optlen);
3939 }
Vincent Whitchurchfa788d92018-09-03 16:23:36 +02003940 case PACKET_IGNORE_OUTGOING:
3941 {
3942 int val;
3943
3944 if (optlen != sizeof(val))
3945 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003946 if (copy_from_sockptr(&val, optval, sizeof(val)))
Vincent Whitchurchfa788d92018-09-03 16:23:36 +02003947 return -EFAULT;
3948 if (val < 0 || val > 1)
3949 return -EINVAL;
3950
3951 po->prot_hook.ignore_outgoing = !!val;
3952 return 0;
3953 }
Paul Chavent5920cd3a2012-11-06 23:10:47 +00003954 case PACKET_TX_HAS_OFF:
3955 {
3956 unsigned int val;
3957
3958 if (optlen != sizeof(val))
3959 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003960 if (copy_from_sockptr(&val, optval, sizeof(val)))
Paul Chavent5920cd3a2012-11-06 23:10:47 +00003961 return -EFAULT;
Willem de Bruijna6361f02018-04-23 17:37:03 -04003962
3963 lock_sock(sk);
Jiapeng Chong25c55b32021-05-17 18:15:25 +08003964 if (!po->rx_ring.pg_vec && !po->tx_ring.pg_vec)
Willem de Bruijna6361f02018-04-23 17:37:03 -04003965 po->tp_tx_has_off = !!val;
Jiapeng Chong25c55b32021-05-17 18:15:25 +08003966
Willem de Bruijna6361f02018-04-23 17:37:03 -04003967 release_sock(sk);
Paul Chavent5920cd3a2012-11-06 23:10:47 +00003968 return 0;
3969 }
Daniel Borkmannd346a3f2013-12-06 11:36:17 +01003970 case PACKET_QDISC_BYPASS:
3971 {
3972 int val;
3973
3974 if (optlen != sizeof(val))
3975 return -EINVAL;
Christoph Hellwiga7b75c52020-07-23 08:09:07 +02003976 if (copy_from_sockptr(&val, optval, sizeof(val)))
Daniel Borkmannd346a3f2013-12-06 11:36:17 +01003977 return -EFAULT;
3978
3979 po->xmit = val ? packet_direct_xmit : dev_queue_xmit;
3980 return 0;
3981 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07003982 default:
3983 return -ENOPROTOOPT;
3984 }
3985}
3986
3987static int packet_getsockopt(struct socket *sock, int level, int optname,
3988 char __user *optval, int __user *optlen)
3989{
3990 int len;
Eric Dumazetc06fff62012-04-19 21:56:11 +00003991 int val, lv = sizeof(val);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003992 struct sock *sk = sock->sk;
3993 struct packet_sock *po = pkt_sk(sk);
Eric Dumazetc06fff62012-04-19 21:56:11 +00003994 void *data = &val;
Daniel Borkmannee80fbf2013-04-19 06:12:29 +00003995 union tpacket_stats_u st;
Willem de Bruijna9b63912015-05-12 11:56:50 -04003996 struct tpacket_rollover_stats rstats;
Eric Dumazet8e8e2952019-06-12 09:52:30 -07003997 int drops;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003998
3999 if (level != SOL_PACKET)
4000 return -ENOPROTOOPT;
4001
Kris Katterjohn8ae55f02006-01-23 16:28:02 -08004002 if (get_user(len, optlen))
4003 return -EFAULT;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004004
4005 if (len < 0)
4006 return -EINVAL;
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09004007
Johann Baudy69e3c752009-05-18 22:11:22 -07004008 switch (optname) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004009 case PACKET_STATISTICS:
Linus Torvalds1da177e2005-04-16 15:20:36 -07004010 spin_lock_bh(&sk->sk_receive_queue.lock);
Daniel Borkmannee80fbf2013-04-19 06:12:29 +00004011 memcpy(&st, &po->stats, sizeof(st));
4012 memset(&po->stats, 0, sizeof(po->stats));
4013 spin_unlock_bh(&sk->sk_receive_queue.lock);
Eric Dumazet8e8e2952019-06-12 09:52:30 -07004014 drops = atomic_xchg(&po->tp_drops, 0);
Daniel Borkmannee80fbf2013-04-19 06:12:29 +00004015
chetan lokef6fb8f12011-08-19 10:18:16 +00004016 if (po->tp_version == TPACKET_V3) {
Eric Dumazetc06fff62012-04-19 21:56:11 +00004017 lv = sizeof(struct tpacket_stats_v3);
Eric Dumazet8e8e2952019-06-12 09:52:30 -07004018 st.stats3.tp_drops = drops;
4019 st.stats3.tp_packets += drops;
Daniel Borkmannee80fbf2013-04-19 06:12:29 +00004020 data = &st.stats3;
chetan lokef6fb8f12011-08-19 10:18:16 +00004021 } else {
Eric Dumazetc06fff62012-04-19 21:56:11 +00004022 lv = sizeof(struct tpacket_stats);
Eric Dumazet8e8e2952019-06-12 09:52:30 -07004023 st.stats1.tp_drops = drops;
4024 st.stats1.tp_packets += drops;
Daniel Borkmannee80fbf2013-04-19 06:12:29 +00004025 data = &st.stats1;
chetan lokef6fb8f12011-08-19 10:18:16 +00004026 }
Daniel Borkmannee80fbf2013-04-19 06:12:29 +00004027
Linus Torvalds1da177e2005-04-16 15:20:36 -07004028 break;
Herbert Xu8dc41942007-02-04 23:31:32 -08004029 case PACKET_AUXDATA:
Herbert Xu8dc41942007-02-04 23:31:32 -08004030 val = po->auxdata;
Herbert Xu8dc41942007-02-04 23:31:32 -08004031 break;
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07004032 case PACKET_ORIGDEV:
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07004033 val = po->origdev;
Peter P. Waskiewicz Jr80feaac2007-04-20 16:05:39 -07004034 break;
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08004035 case PACKET_VNET_HDR:
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08004036 val = po->has_vnet_hdr;
Sridhar Samudralabfd5f4a2010-02-04 20:24:10 -08004037 break;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004038 case PACKET_VERSION:
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004039 val = po->tp_version;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004040 break;
4041 case PACKET_HDRLEN:
4042 if (len > sizeof(int))
4043 len = sizeof(int);
Alexander Potapenkofd2c83b2017-04-25 18:51:46 +02004044 if (len < sizeof(int))
4045 return -EINVAL;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004046 if (copy_from_user(&val, optval, len))
4047 return -EFAULT;
4048 switch (val) {
4049 case TPACKET_V1:
4050 val = sizeof(struct tpacket_hdr);
4051 break;
4052 case TPACKET_V2:
4053 val = sizeof(struct tpacket2_hdr);
4054 break;
chetan lokef6fb8f12011-08-19 10:18:16 +00004055 case TPACKET_V3:
4056 val = sizeof(struct tpacket3_hdr);
4057 break;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004058 default:
4059 return -EINVAL;
4060 }
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004061 break;
Patrick McHardy8913336a2008-07-18 18:05:19 -07004062 case PACKET_RESERVE:
Patrick McHardy8913336a2008-07-18 18:05:19 -07004063 val = po->tp_reserve;
Patrick McHardy8913336a2008-07-18 18:05:19 -07004064 break;
Johann Baudy69e3c752009-05-18 22:11:22 -07004065 case PACKET_LOSS:
Johann Baudy69e3c752009-05-18 22:11:22 -07004066 val = po->tp_loss;
Johann Baudy69e3c752009-05-18 22:11:22 -07004067 break;
Scott McMillan614f60f2010-06-02 05:53:56 -07004068 case PACKET_TIMESTAMP:
Scott McMillan614f60f2010-06-02 05:53:56 -07004069 val = po->tp_tstamp;
Scott McMillan614f60f2010-06-02 05:53:56 -07004070 break;
David S. Millerdc99f602011-07-05 01:45:05 -07004071 case PACKET_FANOUT:
David S. Millerdc99f602011-07-05 01:45:05 -07004072 val = (po->fanout ?
4073 ((u32)po->fanout->id |
Willem de Bruijn77f65eb2013-03-19 10:18:11 +00004074 ((u32)po->fanout->type << 16) |
4075 ((u32)po->fanout->flags << 24)) :
David S. Millerdc99f602011-07-05 01:45:05 -07004076 0);
David S. Millerdc99f602011-07-05 01:45:05 -07004077 break;
Vincent Whitchurchfa788d92018-09-03 16:23:36 +02004078 case PACKET_IGNORE_OUTGOING:
4079 val = po->prot_hook.ignore_outgoing;
4080 break;
Willem de Bruijna9b63912015-05-12 11:56:50 -04004081 case PACKET_ROLLOVER_STATS:
Mike Maloney57f015f2017-11-28 10:44:29 -05004082 if (!po->rollover)
Willem de Bruijna9b63912015-05-12 11:56:50 -04004083 return -EINVAL;
Mike Maloney57f015f2017-11-28 10:44:29 -05004084 rstats.tp_all = atomic_long_read(&po->rollover->num);
4085 rstats.tp_huge = atomic_long_read(&po->rollover->num_huge);
4086 rstats.tp_failed = atomic_long_read(&po->rollover->num_failed);
4087 data = &rstats;
4088 lv = sizeof(rstats);
Willem de Bruijna9b63912015-05-12 11:56:50 -04004089 break;
Paul Chavent5920cd3a2012-11-06 23:10:47 +00004090 case PACKET_TX_HAS_OFF:
4091 val = po->tp_tx_has_off;
4092 break;
Daniel Borkmannd346a3f2013-12-06 11:36:17 +01004093 case PACKET_QDISC_BYPASS:
4094 val = packet_use_direct_xmit(po);
4095 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004096 default:
4097 return -ENOPROTOOPT;
4098 }
4099
Eric Dumazetc06fff62012-04-19 21:56:11 +00004100 if (len > lv)
4101 len = lv;
Kris Katterjohn8ae55f02006-01-23 16:28:02 -08004102 if (put_user(len, optlen))
4103 return -EFAULT;
Herbert Xu8dc41942007-02-04 23:31:32 -08004104 if (copy_to_user(optval, data, len))
4105 return -EFAULT;
Kris Katterjohn8ae55f02006-01-23 16:28:02 -08004106 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004107}
4108
Jiri Pirko351638e2013-05-28 01:30:21 +00004109static int packet_notifier(struct notifier_block *this,
4110 unsigned long msg, void *ptr)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004111{
4112 struct sock *sk;
Jiri Pirko351638e2013-05-28 01:30:21 +00004113 struct net_device *dev = netdev_notifier_info_to_dev(ptr);
YOSHIFUJI Hideakic346dca2008-03-25 21:47:49 +09004114 struct net *net = dev_net(dev);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004115
stephen hemminger808f5112010-02-22 07:57:18 +00004116 rcu_read_lock();
Sasha Levinb67bfe02013-02-27 17:06:00 -08004117 sk_for_each_rcu(sk, &net->packet.sklist) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004118 struct packet_sock *po = pkt_sk(sk);
4119
4120 switch (msg) {
4121 case NETDEV_UNREGISTER:
Linus Torvalds1da177e2005-04-16 15:20:36 -07004122 if (po->mclist)
Francesco Ruggeri82f17092015-03-09 11:51:04 -07004123 packet_dev_mclist_delete(dev, &po->mclist);
Gustavo A. R. Silvadf561f662020-08-23 17:36:59 -05004124 fallthrough;
David S. Millera2efcfa2007-05-29 13:12:50 -07004125
Linus Torvalds1da177e2005-04-16 15:20:36 -07004126 case NETDEV_DOWN:
4127 if (dev->ifindex == po->ifindex) {
4128 spin_lock(&po->bind_lock);
4129 if (po->running) {
David S. Millerce06b032011-07-04 01:44:29 -07004130 __unregister_prot_hook(sk, false);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004131 sk->sk_err = ENETDOWN;
4132 if (!sock_flag(sk, SOCK_DEAD))
Alexander Aringe3ae2362021-06-27 18:48:21 -04004133 sk_error_report(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004134 }
4135 if (msg == NETDEV_UNREGISTER) {
Daniel Borkmann66e56cd2013-12-06 11:36:15 +01004136 packet_cached_dev_reset(po);
Eric Dumazete032f7c2021-06-16 06:42:02 -07004137 WRITE_ONCE(po->ifindex, -1);
Eric Dumazetf1d92682021-12-14 07:09:33 -08004138 dev_put_track(po->prot_hook.dev,
4139 &po->prot_hook.dev_tracker);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004140 po->prot_hook.dev = NULL;
4141 }
4142 spin_unlock(&po->bind_lock);
4143 }
4144 break;
4145 case NETDEV_UP:
stephen hemminger808f5112010-02-22 07:57:18 +00004146 if (dev->ifindex == po->ifindex) {
4147 spin_lock(&po->bind_lock);
David S. Millerce06b032011-07-04 01:44:29 -07004148 if (po->num)
4149 register_prot_hook(sk);
stephen hemminger808f5112010-02-22 07:57:18 +00004150 spin_unlock(&po->bind_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004151 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004152 break;
4153 }
4154 }
stephen hemminger808f5112010-02-22 07:57:18 +00004155 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004156 return NOTIFY_DONE;
4157}
4158
4159
4160static int packet_ioctl(struct socket *sock, unsigned int cmd,
4161 unsigned long arg)
4162{
4163 struct sock *sk = sock->sk;
4164
Johann Baudy69e3c752009-05-18 22:11:22 -07004165 switch (cmd) {
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004166 case SIOCOUTQ:
4167 {
4168 int amount = sk_wmem_alloc_get(sk);
Eric Dumazet31e6d362009-06-17 19:05:41 -07004169
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004170 return put_user(amount, (int __user *)arg);
4171 }
4172 case SIOCINQ:
4173 {
4174 struct sk_buff *skb;
4175 int amount = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004176
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004177 spin_lock_bh(&sk->sk_receive_queue.lock);
4178 skb = skb_peek(&sk->sk_receive_queue);
4179 if (skb)
4180 amount = skb->len;
4181 spin_unlock_bh(&sk->sk_receive_queue.lock);
4182 return put_user(amount, (int __user *)arg);
4183 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004184#ifdef CONFIG_INET
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004185 case SIOCADDRT:
4186 case SIOCDELRT:
4187 case SIOCDARP:
4188 case SIOCGARP:
4189 case SIOCSARP:
4190 case SIOCGIFADDR:
4191 case SIOCSIFADDR:
4192 case SIOCGIFBRDADDR:
4193 case SIOCSIFBRDADDR:
4194 case SIOCGIFNETMASK:
4195 case SIOCSIFNETMASK:
4196 case SIOCGIFDSTADDR:
4197 case SIOCSIFDSTADDR:
4198 case SIOCSIFFLAGS:
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004199 return inet_dgram_ops.ioctl(sock, cmd, arg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004200#endif
4201
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004202 default:
4203 return -ENOIOCTLCMD;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004204 }
4205 return 0;
4206}
4207
Linus Torvaldsa11e1d42018-06-28 09:43:44 -07004208static __poll_t packet_poll(struct file *file, struct socket *sock,
4209 poll_table *wait)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004210{
4211 struct sock *sk = sock->sk;
4212 struct packet_sock *po = pkt_sk(sk);
Linus Torvaldsa11e1d42018-06-28 09:43:44 -07004213 __poll_t mask = datagram_poll(file, sock, wait);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004214
4215 spin_lock_bh(&sk->sk_receive_queue.lock);
Johann Baudy69e3c752009-05-18 22:11:22 -07004216 if (po->rx_ring.pg_vec) {
chetan lokef6fb8f12011-08-19 10:18:16 +00004217 if (!packet_previous_rx_frame(po, &po->rx_ring,
4218 TP_STATUS_KERNEL))
Linus Torvaldsa9a08842018-02-11 14:34:03 -08004219 mask |= EPOLLIN | EPOLLRDNORM;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004220 }
Eric Dumazet9bb6cd62019-06-12 09:52:33 -07004221 packet_rcv_try_clear_pressure(po);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004222 spin_unlock_bh(&sk->sk_receive_queue.lock);
Johann Baudy69e3c752009-05-18 22:11:22 -07004223 spin_lock_bh(&sk->sk_write_queue.lock);
4224 if (po->tx_ring.pg_vec) {
4225 if (packet_current_frame(po, &po->tx_ring, TP_STATUS_AVAILABLE))
Linus Torvaldsa9a08842018-02-11 14:34:03 -08004226 mask |= EPOLLOUT | EPOLLWRNORM;
Johann Baudy69e3c752009-05-18 22:11:22 -07004227 }
4228 spin_unlock_bh(&sk->sk_write_queue.lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004229 return mask;
4230}
4231
4232
4233/* Dirty? Well, I still did not learn better way to account
4234 * for user mmaps.
4235 */
4236
4237static void packet_mm_open(struct vm_area_struct *vma)
4238{
4239 struct file *file = vma->vm_file;
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004240 struct socket *sock = file->private_data;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004241 struct sock *sk = sock->sk;
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09004242
Linus Torvalds1da177e2005-04-16 15:20:36 -07004243 if (sk)
4244 atomic_inc(&pkt_sk(sk)->mapped);
4245}
4246
4247static void packet_mm_close(struct vm_area_struct *vma)
4248{
4249 struct file *file = vma->vm_file;
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004250 struct socket *sock = file->private_data;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004251 struct sock *sk = sock->sk;
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09004252
Linus Torvalds1da177e2005-04-16 15:20:36 -07004253 if (sk)
4254 atomic_dec(&pkt_sk(sk)->mapped);
4255}
4256
Alexey Dobriyanf0f37e2f2009-09-27 22:29:37 +04004257static const struct vm_operations_struct packet_mmap_ops = {
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004258 .open = packet_mm_open,
4259 .close = packet_mm_close,
Linus Torvalds1da177e2005-04-16 15:20:36 -07004260};
4261
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004262static void free_pg_vec(struct pgv *pg_vec, unsigned int order,
4263 unsigned int len)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004264{
4265 int i;
4266
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004267 for (i = 0; i < len; i++) {
Neil Horman0e3125c2010-11-16 10:26:47 -08004268 if (likely(pg_vec[i].buffer)) {
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004269 if (is_vmalloc_addr(pg_vec[i].buffer))
4270 vfree(pg_vec[i].buffer);
4271 else
4272 free_pages((unsigned long)pg_vec[i].buffer,
4273 order);
Neil Horman0e3125c2010-11-16 10:26:47 -08004274 pg_vec[i].buffer = NULL;
4275 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004276 }
4277 kfree(pg_vec);
4278}
4279
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004280static char *alloc_one_pg_vec_page(unsigned long order)
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004281{
Daniel Borkmannf0d4eb22014-01-19 11:46:53 +01004282 char *buffer;
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004283 gfp_t gfp_flags = GFP_KERNEL | __GFP_COMP |
4284 __GFP_ZERO | __GFP_NOWARN | __GFP_NORETRY;
Eric Dumazet719bfea2009-04-15 03:39:52 -07004285
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004286 buffer = (char *) __get_free_pages(gfp_flags, order);
Neil Horman0e3125c2010-11-16 10:26:47 -08004287 if (buffer)
4288 return buffer;
4289
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004290 /* __get_free_pages failed, fall back to vmalloc */
4291 buffer = vzalloc(array_size((1 << order), PAGE_SIZE));
4292 if (buffer)
4293 return buffer;
Neil Horman0e3125c2010-11-16 10:26:47 -08004294
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004295 /* vmalloc failed, lets dig into swap here */
4296 gfp_flags &= ~__GFP_NORETRY;
4297 buffer = (char *) __get_free_pages(gfp_flags, order);
4298 if (buffer)
4299 return buffer;
4300
4301 /* complete and utter failure */
4302 return NULL;
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004303}
4304
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004305static struct pgv *alloc_pg_vec(struct tpacket_req *req, int order)
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004306{
4307 unsigned int block_nr = req->tp_block_nr;
Neil Horman0e3125c2010-11-16 10:26:47 -08004308 struct pgv *pg_vec;
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004309 int i;
4310
Christoph Paasch398f0132019-03-18 23:14:52 -07004311 pg_vec = kcalloc(block_nr, sizeof(struct pgv), GFP_KERNEL | __GFP_NOWARN);
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004312 if (unlikely(!pg_vec))
4313 goto out;
4314
4315 for (i = 0; i < block_nr; i++) {
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004316 pg_vec[i].buffer = alloc_one_pg_vec_page(order);
Neil Horman0e3125c2010-11-16 10:26:47 -08004317 if (unlikely(!pg_vec[i].buffer))
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004318 goto out_free_pgvec;
4319 }
4320
4321out:
4322 return pg_vec;
4323
4324out_free_pgvec:
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004325 free_pg_vec(pg_vec, order, block_nr);
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004326 pg_vec = NULL;
4327 goto out;
4328}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004329
chetan lokef6fb8f12011-08-19 10:18:16 +00004330static int packet_set_ring(struct sock *sk, union tpacket_req_u *req_u,
Johann Baudy69e3c752009-05-18 22:11:22 -07004331 int closing, int tx_ring)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004332{
Neil Horman0e3125c2010-11-16 10:26:47 -08004333 struct pgv *pg_vec = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004334 struct packet_sock *po = pkt_sk(sk);
Willem de Bruijn61fad682020-03-13 12:18:09 -04004335 unsigned long *rx_owner_map = NULL;
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004336 int was_running, order = 0;
Johann Baudy69e3c752009-05-18 22:11:22 -07004337 struct packet_ring_buffer *rb;
4338 struct sk_buff_head *rb_queue;
Al Viro0e11c912006-11-08 00:26:29 -08004339 __be16 num;
Colin Ian King2a6d6c32020-07-01 16:04:33 +01004340 int err;
chetan lokef6fb8f12011-08-19 10:18:16 +00004341 /* Added to avoid minimal code churn */
4342 struct tpacket_req *req = &req_u->req;
4343
Johann Baudy69e3c752009-05-18 22:11:22 -07004344 rb = tx_ring ? &po->tx_ring : &po->rx_ring;
4345 rb_queue = tx_ring ? &sk->sk_write_queue : &sk->sk_receive_queue;
4346
4347 err = -EBUSY;
4348 if (!closing) {
4349 if (atomic_read(&po->mapped))
4350 goto out;
Daniel Borkmannb0138402014-01-15 16:25:36 +01004351 if (packet_read_pending(rb))
Johann Baudy69e3c752009-05-18 22:11:22 -07004352 goto out;
4353 }
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09004354
Linus Torvalds1da177e2005-04-16 15:20:36 -07004355 if (req->tp_block_nr) {
Willem de Bruijn4576cd42018-08-06 10:38:34 -04004356 unsigned int min_frame_size;
4357
Linus Torvalds1da177e2005-04-16 15:20:36 -07004358 /* Sanity tests and some calculations */
Johann Baudy69e3c752009-05-18 22:11:22 -07004359 err = -EBUSY;
4360 if (unlikely(rb->pg_vec))
4361 goto out;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004362
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004363 switch (po->tp_version) {
4364 case TPACKET_V1:
4365 po->tp_hdrlen = TPACKET_HDRLEN;
4366 break;
4367 case TPACKET_V2:
4368 po->tp_hdrlen = TPACKET2_HDRLEN;
4369 break;
chetan lokef6fb8f12011-08-19 10:18:16 +00004370 case TPACKET_V3:
4371 po->tp_hdrlen = TPACKET3_HDRLEN;
4372 break;
Patrick McHardybbd6ef82008-07-14 22:50:15 -07004373 }
4374
Johann Baudy69e3c752009-05-18 22:11:22 -07004375 err = -EINVAL;
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004376 if (unlikely((int)req->tp_block_size <= 0))
Johann Baudy69e3c752009-05-18 22:11:22 -07004377 goto out;
Tobias Klauser90836b62015-11-17 10:40:21 +01004378 if (unlikely(!PAGE_ALIGNED(req->tp_block_size)))
Johann Baudy69e3c752009-05-18 22:11:22 -07004379 goto out;
Willem de Bruijn4576cd42018-08-06 10:38:34 -04004380 min_frame_size = po->tp_hdrlen + po->tp_reserve;
Eric Dumazetdc808112014-08-15 09:16:04 -07004381 if (po->tp_version >= TPACKET_V3 &&
Willem de Bruijn4576cd42018-08-06 10:38:34 -04004382 req->tp_block_size <
4383 BLK_PLUS_PRIV((u64)req_u->req3.tp_sizeof_priv) + min_frame_size)
Eric Dumazetdc808112014-08-15 09:16:04 -07004384 goto out;
Willem de Bruijn4576cd42018-08-06 10:38:34 -04004385 if (unlikely(req->tp_frame_size < min_frame_size))
Johann Baudy69e3c752009-05-18 22:11:22 -07004386 goto out;
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004387 if (unlikely(req->tp_frame_size & (TPACKET_ALIGNMENT - 1)))
Johann Baudy69e3c752009-05-18 22:11:22 -07004388 goto out;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004389
Tobias Klauser4194b492015-11-17 10:38:36 +01004390 rb->frames_per_block = req->tp_block_size / req->tp_frame_size;
4391 if (unlikely(rb->frames_per_block == 0))
Johann Baudy69e3c752009-05-18 22:11:22 -07004392 goto out;
Kal Conleyfc628142019-02-10 09:57:11 +01004393 if (unlikely(rb->frames_per_block > UINT_MAX / req->tp_block_nr))
Andrey Konovalov8f8d28e2017-03-29 16:11:21 +02004394 goto out;
Johann Baudy69e3c752009-05-18 22:11:22 -07004395 if (unlikely((rb->frames_per_block * req->tp_block_nr) !=
4396 req->tp_frame_nr))
4397 goto out;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004398
4399 err = -ENOMEM;
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004400 order = get_order(req->tp_block_size);
4401 pg_vec = alloc_pg_vec(req, order);
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004402 if (unlikely(!pg_vec))
Linus Torvalds1da177e2005-04-16 15:20:36 -07004403 goto out;
chetan lokef6fb8f12011-08-19 10:18:16 +00004404 switch (po->tp_version) {
4405 case TPACKET_V3:
Sowmini Varadhan7f953ab2017-01-03 06:31:47 -08004406 /* Block transmit is not supported yet */
4407 if (!tx_ring) {
Maninder Singhe8e85cc2015-06-22 12:39:16 +05304408 init_prb_bdqc(po, rb, pg_vec, req_u);
Sowmini Varadhan7f953ab2017-01-03 06:31:47 -08004409 } else {
4410 struct tpacket_req3 *req3 = &req_u->req3;
4411
4412 if (req3->tp_retire_blk_tov ||
4413 req3->tp_sizeof_priv ||
4414 req3->tp_feature_req_word) {
4415 err = -EINVAL;
Eric Dumazet55655e32019-06-24 02:38:20 -07004416 goto out_free_pg_vec;
Sowmini Varadhan7f953ab2017-01-03 06:31:47 -08004417 }
4418 }
Dan Carpenterd7cf0c32014-02-18 15:20:51 +03004419 break;
chetan lokef6fb8f12011-08-19 10:18:16 +00004420 default:
Willem de Bruijn61fad682020-03-13 12:18:09 -04004421 if (!tx_ring) {
4422 rx_owner_map = bitmap_alloc(req->tp_frame_nr,
4423 GFP_KERNEL | __GFP_NOWARN | __GFP_ZERO);
4424 if (!rx_owner_map)
4425 goto out_free_pg_vec;
4426 }
chetan lokef6fb8f12011-08-19 10:18:16 +00004427 break;
4428 }
Johann Baudy69e3c752009-05-18 22:11:22 -07004429 }
4430 /* Done */
4431 else {
4432 err = -EINVAL;
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004433 if (unlikely(req->tp_frame_nr))
Johann Baudy69e3c752009-05-18 22:11:22 -07004434 goto out;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004435 }
4436
Linus Torvalds1da177e2005-04-16 15:20:36 -07004437
4438 /* Detach socket from network */
4439 spin_lock(&po->bind_lock);
4440 was_running = po->running;
4441 num = po->num;
4442 if (was_running) {
Eric Dumazetc7d2ef52021-06-16 06:42:01 -07004443 WRITE_ONCE(po->num, 0);
David S. Millerce06b032011-07-04 01:44:29 -07004444 __unregister_prot_hook(sk, false);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004445 }
4446 spin_unlock(&po->bind_lock);
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09004447
Linus Torvalds1da177e2005-04-16 15:20:36 -07004448 synchronize_net();
4449
4450 err = -EBUSY;
Herbert Xu905db442009-01-30 14:12:06 -08004451 mutex_lock(&po->pg_vec_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004452 if (closing || atomic_read(&po->mapped) == 0) {
4453 err = 0;
Johann Baudy69e3c752009-05-18 22:11:22 -07004454 spin_lock_bh(&rb_queue->lock);
Changli Gaoc053fd92010-12-10 16:02:20 -08004455 swap(rb->pg_vec, pg_vec);
Willem de Bruijn61fad682020-03-13 12:18:09 -04004456 if (po->tp_version <= TPACKET_V2)
4457 swap(rb->rx_owner_map, rx_owner_map);
Johann Baudy69e3c752009-05-18 22:11:22 -07004458 rb->frame_max = (req->tp_frame_nr - 1);
4459 rb->head = 0;
4460 rb->frame_size = req->tp_frame_size;
4461 spin_unlock_bh(&rb_queue->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004462
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004463 swap(rb->pg_vec_order, order);
Changli Gaoc053fd92010-12-10 16:02:20 -08004464 swap(rb->pg_vec_len, req->tp_block_nr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004465
Johann Baudy69e3c752009-05-18 22:11:22 -07004466 rb->pg_vec_pages = req->tp_block_size/PAGE_SIZE;
4467 po->prot_hook.func = (po->rx_ring.pg_vec) ?
4468 tpacket_rcv : packet_rcv;
4469 skb_queue_purge(rb_queue);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004470 if (atomic_read(&po->mapped))
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004471 pr_err("packet_mmap: vma is busy: %d\n",
4472 atomic_read(&po->mapped));
Linus Torvalds1da177e2005-04-16 15:20:36 -07004473 }
Herbert Xu905db442009-01-30 14:12:06 -08004474 mutex_unlock(&po->pg_vec_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004475
4476 spin_lock(&po->bind_lock);
David S. Millerce06b032011-07-04 01:44:29 -07004477 if (was_running) {
Eric Dumazetc7d2ef52021-06-16 06:42:01 -07004478 WRITE_ONCE(po->num, num);
David S. Millerce06b032011-07-04 01:44:29 -07004479 register_prot_hook(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004480 }
4481 spin_unlock(&po->bind_lock);
WANG Congc800aaf2017-07-24 10:07:32 -07004482 if (pg_vec && (po->tp_version > TPACKET_V2)) {
chetan lokef6fb8f12011-08-19 10:18:16 +00004483 /* Because we don't support block-based V3 on tx-ring */
4484 if (!tx_ring)
Tobias Klauser73d0fcf2015-07-28 14:21:26 +02004485 prb_shutdown_retire_blk_timer(po, rb_queue);
chetan lokef6fb8f12011-08-19 10:18:16 +00004486 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004487
Eric Dumazet55655e32019-06-24 02:38:20 -07004488out_free_pg_vec:
Willem de Bruijnec6af092021-12-15 09:39:37 -05004489 if (pg_vec) {
4490 bitmap_free(rx_owner_map);
Eric Dumazet3a7ad062018-08-29 11:50:12 -07004491 free_pg_vec(pg_vec, order, req->tp_block_nr);
Willem de Bruijnec6af092021-12-15 09:39:37 -05004492 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004493out:
4494 return err;
4495}
4496
Johann Baudy69e3c752009-05-18 22:11:22 -07004497static int packet_mmap(struct file *file, struct socket *sock,
4498 struct vm_area_struct *vma)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004499{
4500 struct sock *sk = sock->sk;
4501 struct packet_sock *po = pkt_sk(sk);
Johann Baudy69e3c752009-05-18 22:11:22 -07004502 unsigned long size, expected_size;
4503 struct packet_ring_buffer *rb;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004504 unsigned long start;
4505 int err = -EINVAL;
4506 int i;
4507
4508 if (vma->vm_pgoff)
4509 return -EINVAL;
4510
Herbert Xu905db442009-01-30 14:12:06 -08004511 mutex_lock(&po->pg_vec_lock);
Johann Baudy69e3c752009-05-18 22:11:22 -07004512
4513 expected_size = 0;
4514 for (rb = &po->rx_ring; rb <= &po->tx_ring; rb++) {
4515 if (rb->pg_vec) {
4516 expected_size += rb->pg_vec_len
4517 * rb->pg_vec_pages
4518 * PAGE_SIZE;
4519 }
4520 }
4521
4522 if (expected_size == 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004523 goto out;
Johann Baudy69e3c752009-05-18 22:11:22 -07004524
4525 size = vma->vm_end - vma->vm_start;
4526 if (size != expected_size)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004527 goto out;
4528
Linus Torvalds1da177e2005-04-16 15:20:36 -07004529 start = vma->vm_start;
Johann Baudy69e3c752009-05-18 22:11:22 -07004530 for (rb = &po->rx_ring; rb <= &po->tx_ring; rb++) {
4531 if (rb->pg_vec == NULL)
4532 continue;
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004533
Johann Baudy69e3c752009-05-18 22:11:22 -07004534 for (i = 0; i < rb->pg_vec_len; i++) {
Neil Horman0e3125c2010-11-16 10:26:47 -08004535 struct page *page;
4536 void *kaddr = rb->pg_vec[i].buffer;
Johann Baudy69e3c752009-05-18 22:11:22 -07004537 int pg_num;
4538
Changli Gaoc56b4d92010-12-01 02:52:57 +00004539 for (pg_num = 0; pg_num < rb->pg_vec_pages; pg_num++) {
4540 page = pgv_to_page(kaddr);
Johann Baudy69e3c752009-05-18 22:11:22 -07004541 err = vm_insert_page(vma, start, page);
4542 if (unlikely(err))
4543 goto out;
4544 start += PAGE_SIZE;
Neil Horman0e3125c2010-11-16 10:26:47 -08004545 kaddr += PAGE_SIZE;
Johann Baudy69e3c752009-05-18 22:11:22 -07004546 }
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004547 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004548 }
Johann Baudy69e3c752009-05-18 22:11:22 -07004549
David S. Miller4ebf0ae2005-12-06 16:38:35 -08004550 atomic_inc(&po->mapped);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004551 vma->vm_ops = &packet_mmap_ops;
4552 err = 0;
4553
4554out:
Herbert Xu905db442009-01-30 14:12:06 -08004555 mutex_unlock(&po->pg_vec_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004556 return err;
4557}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004558
Eric Dumazet90ddc4f2005-12-22 12:49:22 -08004559static const struct proto_ops packet_ops_spkt = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004560 .family = PF_PACKET,
4561 .owner = THIS_MODULE,
4562 .release = packet_release,
4563 .bind = packet_bind_spkt,
4564 .connect = sock_no_connect,
4565 .socketpair = sock_no_socketpair,
4566 .accept = sock_no_accept,
4567 .getname = packet_getname_spkt,
Linus Torvaldsa11e1d42018-06-28 09:43:44 -07004568 .poll = datagram_poll,
Linus Torvalds1da177e2005-04-16 15:20:36 -07004569 .ioctl = packet_ioctl,
Arnd Bergmannc7cbdbf2019-04-17 22:51:48 +02004570 .gettstamp = sock_gettstamp,
Linus Torvalds1da177e2005-04-16 15:20:36 -07004571 .listen = sock_no_listen,
4572 .shutdown = sock_no_shutdown,
Linus Torvalds1da177e2005-04-16 15:20:36 -07004573 .sendmsg = packet_sendmsg_spkt,
4574 .recvmsg = packet_recvmsg,
4575 .mmap = sock_no_mmap,
4576 .sendpage = sock_no_sendpage,
4577};
Linus Torvalds1da177e2005-04-16 15:20:36 -07004578
Eric Dumazet90ddc4f2005-12-22 12:49:22 -08004579static const struct proto_ops packet_ops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004580 .family = PF_PACKET,
4581 .owner = THIS_MODULE,
4582 .release = packet_release,
4583 .bind = packet_bind,
4584 .connect = sock_no_connect,
4585 .socketpair = sock_no_socketpair,
4586 .accept = sock_no_accept,
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09004587 .getname = packet_getname,
Linus Torvaldsa11e1d42018-06-28 09:43:44 -07004588 .poll = packet_poll,
Linus Torvalds1da177e2005-04-16 15:20:36 -07004589 .ioctl = packet_ioctl,
Arnd Bergmannc7cbdbf2019-04-17 22:51:48 +02004590 .gettstamp = sock_gettstamp,
Linus Torvalds1da177e2005-04-16 15:20:36 -07004591 .listen = sock_no_listen,
4592 .shutdown = sock_no_shutdown,
4593 .setsockopt = packet_setsockopt,
4594 .getsockopt = packet_getsockopt,
4595 .sendmsg = packet_sendmsg,
4596 .recvmsg = packet_recvmsg,
4597 .mmap = packet_mmap,
4598 .sendpage = sock_no_sendpage,
4599};
4600
Stephen Hemmingerec1b4cf2009-10-05 05:58:39 +00004601static const struct net_proto_family packet_family_ops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004602 .family = PF_PACKET,
4603 .create = packet_create,
4604 .owner = THIS_MODULE,
4605};
4606
4607static struct notifier_block packet_netdev_notifier = {
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004608 .notifier_call = packet_notifier,
Linus Torvalds1da177e2005-04-16 15:20:36 -07004609};
4610
4611#ifdef CONFIG_PROC_FS
Linus Torvalds1da177e2005-04-16 15:20:36 -07004612
4613static void *packet_seq_start(struct seq_file *seq, loff_t *pos)
stephen hemminger808f5112010-02-22 07:57:18 +00004614 __acquires(RCU)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004615{
Denis V. Luneve372c412007-11-19 22:31:54 -08004616 struct net *net = seq_file_net(seq);
stephen hemminger808f5112010-02-22 07:57:18 +00004617
4618 rcu_read_lock();
4619 return seq_hlist_start_head_rcu(&net->packet.sklist, *pos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004620}
4621
4622static void *packet_seq_next(struct seq_file *seq, void *v, loff_t *pos)
4623{
Herbert Xu1bf40952007-12-16 14:04:02 -08004624 struct net *net = seq_file_net(seq);
stephen hemminger808f5112010-02-22 07:57:18 +00004625 return seq_hlist_next_rcu(v, &net->packet.sklist, pos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004626}
4627
4628static void packet_seq_stop(struct seq_file *seq, void *v)
stephen hemminger808f5112010-02-22 07:57:18 +00004629 __releases(RCU)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004630{
stephen hemminger808f5112010-02-22 07:57:18 +00004631 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004632}
4633
YOSHIFUJI Hideaki1ce4f282007-02-09 23:25:10 +09004634static int packet_seq_show(struct seq_file *seq, void *v)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004635{
4636 if (v == SEQ_START_TOKEN)
Baruch Siachabdcd062020-12-16 09:28:04 +02004637 seq_printf(seq,
4638 "%*sRefCnt Type Proto Iface R Rmem User Inode\n",
4639 IS_ENABLED(CONFIG_64BIT) ? -17 : -9, "sk");
Linus Torvalds1da177e2005-04-16 15:20:36 -07004640 else {
Li Zefanb7ceabd2010-02-08 23:19:29 +00004641 struct sock *s = sk_entry(v);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004642 const struct packet_sock *po = pkt_sk(s);
4643
4644 seq_printf(seq,
Dan Rosenberg71338aa2011-05-23 12:17:35 +00004645 "%pK %-6d %-4d %04x %-5d %1d %-6u %-6u %-6lu\n",
Linus Torvalds1da177e2005-04-16 15:20:36 -07004646 s,
Reshetova, Elena41c6d652017-06-30 13:08:01 +03004647 refcount_read(&s->sk_refcnt),
Linus Torvalds1da177e2005-04-16 15:20:36 -07004648 s->sk_type,
Eric Dumazetc7d2ef52021-06-16 06:42:01 -07004649 ntohs(READ_ONCE(po->num)),
Eric Dumazete032f7c2021-06-16 06:42:02 -07004650 READ_ONCE(po->ifindex),
Linus Torvalds1da177e2005-04-16 15:20:36 -07004651 po->running,
4652 atomic_read(&s->sk_rmem_alloc),
Eric W. Biedermana7cb5a42012-05-24 01:10:10 -06004653 from_kuid_munged(seq_user_ns(seq), sock_i_uid(s)),
Eric Dumazet40d4e3d2009-07-21 21:57:59 +00004654 sock_i_ino(s));
Linus Torvalds1da177e2005-04-16 15:20:36 -07004655 }
4656
4657 return 0;
4658}
4659
Philippe De Muyter56b3d972007-07-10 23:07:31 -07004660static const struct seq_operations packet_seq_ops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004661 .start = packet_seq_start,
4662 .next = packet_seq_next,
4663 .stop = packet_seq_stop,
4664 .show = packet_seq_show,
4665};
Linus Torvalds1da177e2005-04-16 15:20:36 -07004666#endif
4667
Alexey Dobriyan2c8c1e72010-01-17 03:35:32 +00004668static int __net_init packet_net_init(struct net *net)
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08004669{
Pavel Emelyanov0fa7fa92012-08-21 01:06:47 +00004670 mutex_init(&net->packet.sklist_lock);
Denis V. Lunev2aaef4e2007-12-11 04:19:54 -08004671 INIT_HLIST_HEAD(&net->packet.sklist);
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08004672
Yonatan Linika268e0f2020-12-14 22:25:50 +02004673#ifdef CONFIG_PROC_FS
Christoph Hellwigc3506372018-04-10 19:42:55 +02004674 if (!proc_create_net("packet", 0, net->proc_net, &packet_seq_ops,
4675 sizeof(struct seq_net_private)))
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08004676 return -ENOMEM;
Yonatan Linika268e0f2020-12-14 22:25:50 +02004677#endif /* CONFIG_PROC_FS */
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08004678
4679 return 0;
4680}
4681
Alexey Dobriyan2c8c1e72010-01-17 03:35:32 +00004682static void __net_exit packet_net_exit(struct net *net)
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08004683{
Gao fengece31ff2013-02-18 01:34:56 +00004684 remove_proc_entry("packet", net->proc_net);
Vasily Averin669f8f12017-11-12 22:27:49 +03004685 WARN_ON_ONCE(!hlist_empty(&net->packet.sklist));
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08004686}
4687
4688static struct pernet_operations packet_net_ops = {
4689 .init = packet_net_init,
4690 .exit = packet_net_exit,
4691};
4692
4693
Linus Torvalds1da177e2005-04-16 15:20:36 -07004694static void __exit packet_exit(void)
4695{
Linus Torvalds1da177e2005-04-16 15:20:36 -07004696 unregister_netdevice_notifier(&packet_netdev_notifier);
Denis V. Lunevd12d01d2007-11-19 22:28:35 -08004697 unregister_pernet_subsys(&packet_net_ops);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004698 sock_unregister(PF_PACKET);
4699 proto_unregister(&packet_proto);
4700}
4701
4702static int __init packet_init(void)
4703{
YueHaibing36096f22019-05-09 22:52:20 +08004704 int rc;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004705
YueHaibing36096f22019-05-09 22:52:20 +08004706 rc = proto_register(&packet_proto, 0);
4707 if (rc)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004708 goto out;
YueHaibing36096f22019-05-09 22:52:20 +08004709 rc = sock_register(&packet_family_ops);
4710 if (rc)
4711 goto out_proto;
4712 rc = register_pernet_subsys(&packet_net_ops);
4713 if (rc)
4714 goto out_sock;
4715 rc = register_netdevice_notifier(&packet_netdev_notifier);
4716 if (rc)
4717 goto out_pernet;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004718
YueHaibing36096f22019-05-09 22:52:20 +08004719 return 0;
4720
4721out_pernet:
4722 unregister_pernet_subsys(&packet_net_ops);
4723out_sock:
4724 sock_unregister(PF_PACKET);
4725out_proto:
4726 proto_unregister(&packet_proto);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004727out:
4728 return rc;
4729}
4730
4731module_init(packet_init);
4732module_exit(packet_exit);
4733MODULE_LICENSE("GPL");
4734MODULE_ALIAS_NETPROTO(PF_PACKET);