tun.c 86.1 KB
Newer Older
1
// SPDX-License-Identifier: GPL-2.0-or-later
L
Linus Torvalds 已提交
2 3 4 5 6 7 8 9 10 11
/*
 *  TUN - Universal TUN/TAP device driver.
 *  Copyright (C) 1999-2002 Maxim Krasnyansky <maxk@qualcomm.com>
 *
 *  $Id: tun.c,v 1.15 2002/03/01 02:44:24 maxk Exp $
 */

/*
 *  Changes:
 *
12 13 14
 *  Mike Kershaw <dragorn@kismetwireless.net> 2005/08/14
 *    Add TUNSETLINK ioctl to set the link encapsulation
 *
L
Linus Torvalds 已提交
15
 *  Mark Smith <markzzzsmith@yahoo.com.au>
J
Joe Perches 已提交
16
 *    Use eth_random_addr() for tap MAC address.
L
Linus Torvalds 已提交
17 18 19 20 21 22 23 24 25 26 27
 *
 *  Harald Roelle <harald.roelle@ifi.lmu.de>  2004/04/20
 *    Fixes in packet dropping, queue length setting and queue wakeup.
 *    Increased default tx queue length.
 *    Added ethtool API.
 *    Minor cleanups
 *
 *  Daniel Podlejski <underley@underley.eu.org>
 *    Modifications for 2.3.99-pre5 kernel.
 */

28 29
#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt

L
Linus Torvalds 已提交
30 31 32 33 34 35 36 37
#define DRV_NAME	"tun"
#define DRV_VERSION	"1.6"
#define DRV_DESCRIPTION	"Universal TUN/TAP device driver"
#define DRV_COPYRIGHT	"(C) 1999-2004 Max Krasnyansky <maxk@qualcomm.com>"

#include <linux/module.h>
#include <linux/errno.h>
#include <linux/kernel.h>
38
#include <linux/sched/signal.h>
L
Linus Torvalds 已提交
39 40 41 42 43 44 45 46 47 48 49
#include <linux/major.h>
#include <linux/slab.h>
#include <linux/poll.h>
#include <linux/fcntl.h>
#include <linux/init.h>
#include <linux/skbuff.h>
#include <linux/netdevice.h>
#include <linux/etherdevice.h>
#include <linux/miscdevice.h>
#include <linux/ethtool.h>
#include <linux/rtnetlink.h>
50
#include <linux/compat.h>
L
Linus Torvalds 已提交
51 52 53 54
#include <linux/if.h>
#include <linux/if_arp.h>
#include <linux/if_ether.h>
#include <linux/if_tun.h>
J
Jason Wang 已提交
55
#include <linux/if_vlan.h>
L
Linus Torvalds 已提交
56
#include <linux/crc32.h>
57
#include <linux/nsproxy.h>
58
#include <linux/virtio_net.h>
M
Michael S. Tsirkin 已提交
59
#include <linux/rcupdate.h>
60
#include <net/net_namespace.h>
61
#include <net/netns/generic.h>
62
#include <net/rtnetlink.h>
63
#include <net/sock.h>
64
#include <net/xdp.h>
65
#include <linux/seq_file.h>
H
Herbert Xu 已提交
66
#include <linux/uio.h>
J
Jason Wang 已提交
67
#include <linux/skb_array.h>
J
Jason Wang 已提交
68 69
#include <linux/bpf.h>
#include <linux/bpf_trace.h>
70
#include <linux/mutex.h>
L
Linus Torvalds 已提交
71

72
#include <linux/uaccess.h>
73
#include <linux/proc_fs.h>
L
Linus Torvalds 已提交
74

75 76 77
static void tun_default_link_ksettings(struct net_device *dev,
				       struct ethtool_link_ksettings *cmd);

78 79 80
/* Uncomment to enable debugging */
/* #define TUN_DEBUG 1 */

L
Linus Torvalds 已提交
81 82
#ifdef TUN_DEBUG
static int debug;
83

84 85 86 87 88 89 90 91 92 93
#define tun_debug(level, tun, fmt, args...)			\
do {								\
	if (tun->debug)						\
		netdev_printk(level, tun->dev, fmt, ##args);	\
} while (0)
#define DBG1(level, fmt, args...)				\
do {								\
	if (debug == 2)						\
		printk(level fmt, ##args);			\
} while (0)
94
#else
95 96 97 98 99 100 101 102 103 104
#define tun_debug(level, tun, fmt, args...)			\
do {								\
	if (0)							\
		netdev_printk(level, tun->dev, fmt, ##args);	\
} while (0)
#define DBG1(level, fmt, args...)				\
do {								\
	if (0)							\
		printk(level fmt, ##args);			\
} while (0)
105 106
#endif

107
#define TUN_RX_PAD (NET_IP_ALIGN + NET_SKB_PAD)
108

109 110 111 112 113 114
/* TUN device flags */

/* IFF_ATTACH_QUEUE is never stored in device flags,
 * overload it to mean fasync when stored there.
 */
#define TUN_FASYNC	IFF_ATTACH_QUEUE
M
Michael S. Tsirkin 已提交
115 116
/* High bits in flags field are unused. */
#define TUN_VNET_LE     0x80000000
117
#define TUN_VNET_BE     0x40000000
118 119

#define TUN_FEATURES (IFF_NO_PI | IFF_ONE_QUEUE | IFF_VNET_HDR | \
120 121
		      IFF_MULTI_QUEUE | IFF_NAPI | IFF_NAPI_FRAGS)

122 123
#define GOODCOPY_LEN 128

124 125 126 127 128 129 130
#define FLT_EXACT_COUNT 8
struct tap_filter {
	unsigned int    count;    /* Number of addrs. Zero means disabled */
	u32             mask[2];  /* Mask of the hashed addrs */
	unsigned char	addr[FLT_EXACT_COUNT][ETH_ALEN];
};

131 132 133
/* MAX_TAP_QUEUES 256 is chosen to allow rx/tx queues to be equal
 * to max number of VCPUs in guest. */
#define MAX_TAP_QUEUES 256
134
#define MAX_TAP_FLOWS  4096
J
Jason Wang 已提交
135

J
Jason Wang 已提交
136 137
#define TUN_FLOW_EXPIRE (3 * HZ)

138 139 140 141 142 143 144 145 146 147 148
struct tun_pcpu_stats {
	u64 rx_packets;
	u64 rx_bytes;
	u64 tx_packets;
	u64 tx_bytes;
	struct u64_stats_sync syncp;
	u32 rx_dropped;
	u32 tx_dropped;
	u32 rx_frame_errors;
};

J
Jason Wang 已提交
149
/* A tun_file connects an open character device to a tuntap netdevice. It
S
stephen hemminger 已提交
150
 * also contains all socket related structures (except sock_fprog and tap_filter)
J
Jason Wang 已提交
151 152
 * to serve as one transmit queue for tuntap device. The sock_fprog and
 * tap_filter were kept in tun_struct since they were used for filtering for the
R
Rami Rosen 已提交
153
 * netdevice not for a specific queue (at least I didn't see the requirement for
J
Jason Wang 已提交
154
 * this).
155 156
 *
 * RCU usage:
R
Rami Rosen 已提交
157
 * The tun_file and tun_struct are loosely coupled, the pointer from one to the
158
 * other can only be read while rcu_read_lock or rtnl_lock is held.
J
Jason Wang 已提交
159
 */
E
Eric W. Biederman 已提交
160
struct tun_file {
J
Jason Wang 已提交
161 162
	struct sock sk;
	struct socket socket;
163
	struct tun_struct __rcu *tun;
J
Jason Wang 已提交
164 165 166
	struct fasync_struct *fasync;
	/* only used for fasnyc */
	unsigned int flags;
167 168 169 170
	union {
		u16 queue_index;
		unsigned int ifindex;
	};
171
	struct napi_struct napi;
172
	bool napi_enabled;
E
Eric Dumazet 已提交
173
	bool napi_frags_enabled;
174
	struct mutex napi_mutex;	/* Protects access to the above napi */
J
Jason Wang 已提交
175 176
	struct list_head next;
	struct tun_struct *detached;
177
	struct ptr_ring tx_ring;
J
Jesper Dangaard Brouer 已提交
178
	struct xdp_rxq_info xdp_rxq;
E
Eric W. Biederman 已提交
179 180
};

181 182 183 184 185
struct tun_page {
	struct page *page;
	int count;
};

J
Jason Wang 已提交
186 187 188 189 190 191
struct tun_flow_entry {
	struct hlist_node hash_link;
	struct rcu_head rcu;
	struct tun_struct *tun;

	u32 rxhash;
192
	u32 rps_rxhash;
J
Jason Wang 已提交
193
	int queue_index;
194
	unsigned long updated ____cacheline_aligned_in_smp;
J
Jason Wang 已提交
195 196 197
};

#define TUN_NUM_FLOW_ENTRIES 1024
L
Li RongQing 已提交
198
#define TUN_MASK_FLOW_ENTRIES (TUN_NUM_FLOW_ENTRIES - 1)
J
Jason Wang 已提交
199

200
struct tun_prog {
201 202 203 204
	struct rcu_head rcu;
	struct bpf_prog *prog;
};

J
Jason Wang 已提交
205
/* Since the socket were moved to tun_file, to preserve the behavior of persist
R
Rami Rosen 已提交
206
 * device, socket filter, sndbuf and vnet header size were restore when the
J
Jason Wang 已提交
207 208
 * file were attached to a persist device.
 */
209
struct tun_struct {
J
Jason Wang 已提交
210 211
	struct tun_file __rcu	*tfiles[MAX_TAP_QUEUES];
	unsigned int            numqueues;
212
	unsigned int 		flags;
213 214
	kuid_t			owner;
	kgid_t			group;
215 216

	struct net_device	*dev;
217
	netdev_features_t	set_features;
218
#define TUN_USER_FEATURES (NETIF_F_HW_CSUM|NETIF_F_TSO_ECN|NETIF_F_TSO| \
219
			  NETIF_F_TSO6)
220

221
	int			align;
222
	int			vnet_hdr_sz;
J
Jason Wang 已提交
223 224 225 226 227
	int			sndbuf;
	struct tap_filter	txflt;
	struct sock_fprog	fprog;
	/* protected by rtnl lock */
	bool			filter_attached;
228 229
#ifdef TUN_DEBUG
	int debug;
L
Linus Torvalds 已提交
230
#endif
J
Jason Wang 已提交
231 232 233 234
	spinlock_t lock;
	struct hlist_head flows[TUN_NUM_FLOW_ENTRIES];
	struct timer_list flow_gc_timer;
	unsigned long ageing_time;
J
Jason Wang 已提交
235 236
	unsigned int numdisabled;
	struct list_head disabled;
237
	void *security;
238
	u32 flow_count;
J
Jason Wang 已提交
239
	u32 rx_batched;
240
	struct tun_pcpu_stats __percpu *pcpu_stats;
J
Jason Wang 已提交
241
	struct bpf_prog __rcu *xdp_prog;
242
	struct tun_prog __rcu *steering_prog;
243
	struct tun_prog __rcu *filter_prog;
244
	struct ethtool_link_ksettings link_ksettings;
245
};
L
Linus Torvalds 已提交
246

247 248 249
struct veth {
	__be16 h_vlan_proto;
	__be16 h_vlan_TCI;
250
};
L
Linus Torvalds 已提交
251

252
bool tun_is_xdp_frame(void *ptr)
J
Jason Wang 已提交
253 254 255
{
	return (unsigned long)ptr & TUN_XDP_FLAG;
}
256
EXPORT_SYMBOL(tun_is_xdp_frame);
J
Jason Wang 已提交
257 258 259 260 261 262 263 264 265 266 267 268 269

void *tun_xdp_to_ptr(void *ptr)
{
	return (void *)((unsigned long)ptr | TUN_XDP_FLAG);
}
EXPORT_SYMBOL(tun_xdp_to_ptr);

void *tun_ptr_to_xdp(void *ptr)
{
	return (void *)((unsigned long)ptr & ~TUN_XDP_FLAG);
}
EXPORT_SYMBOL(tun_ptr_to_xdp);

270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310
static int tun_napi_receive(struct napi_struct *napi, int budget)
{
	struct tun_file *tfile = container_of(napi, struct tun_file, napi);
	struct sk_buff_head *queue = &tfile->sk.sk_write_queue;
	struct sk_buff_head process_queue;
	struct sk_buff *skb;
	int received = 0;

	__skb_queue_head_init(&process_queue);

	spin_lock(&queue->lock);
	skb_queue_splice_tail_init(queue, &process_queue);
	spin_unlock(&queue->lock);

	while (received < budget && (skb = __skb_dequeue(&process_queue))) {
		napi_gro_receive(napi, skb);
		++received;
	}

	if (!skb_queue_empty(&process_queue)) {
		spin_lock(&queue->lock);
		skb_queue_splice(&process_queue, queue);
		spin_unlock(&queue->lock);
	}

	return received;
}

static int tun_napi_poll(struct napi_struct *napi, int budget)
{
	unsigned int received;

	received = tun_napi_receive(napi, budget);

	if (received < budget)
		napi_complete_done(napi, received);

	return received;
}

static void tun_napi_init(struct tun_struct *tun, struct tun_file *tfile,
E
Eric Dumazet 已提交
311
			  bool napi_en, bool napi_frags)
312
{
313
	tfile->napi_enabled = napi_en;
E
Eric Dumazet 已提交
314
	tfile->napi_frags_enabled = napi_en && napi_frags;
315 316 317 318 319 320 321
	if (napi_en) {
		netif_napi_add(tun->dev, &tfile->napi, tun_napi_poll,
			       NAPI_POLL_WEIGHT);
		napi_enable(&tfile->napi);
	}
}

E
Eric Dumazet 已提交
322
static void tun_napi_disable(struct tun_file *tfile)
323
{
324
	if (tfile->napi_enabled)
325 326 327
		napi_disable(&tfile->napi);
}

E
Eric Dumazet 已提交
328
static void tun_napi_del(struct tun_file *tfile)
329
{
330
	if (tfile->napi_enabled)
331 332 333
		netif_napi_del(&tfile->napi);
}

E
Eric Dumazet 已提交
334
static bool tun_napi_frags_enabled(const struct tun_file *tfile)
335
{
E
Eric Dumazet 已提交
336
	return tfile->napi_frags_enabled;
337 338
}

339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386
#ifdef CONFIG_TUN_VNET_CROSS_LE
static inline bool tun_legacy_is_little_endian(struct tun_struct *tun)
{
	return tun->flags & TUN_VNET_BE ? false :
		virtio_legacy_is_little_endian();
}

static long tun_get_vnet_be(struct tun_struct *tun, int __user *argp)
{
	int be = !!(tun->flags & TUN_VNET_BE);

	if (put_user(be, argp))
		return -EFAULT;

	return 0;
}

static long tun_set_vnet_be(struct tun_struct *tun, int __user *argp)
{
	int be;

	if (get_user(be, argp))
		return -EFAULT;

	if (be)
		tun->flags |= TUN_VNET_BE;
	else
		tun->flags &= ~TUN_VNET_BE;

	return 0;
}
#else
static inline bool tun_legacy_is_little_endian(struct tun_struct *tun)
{
	return virtio_legacy_is_little_endian();
}

static long tun_get_vnet_be(struct tun_struct *tun, int __user *argp)
{
	return -EINVAL;
}

static long tun_set_vnet_be(struct tun_struct *tun, int __user *argp)
{
	return -EINVAL;
}
#endif /* CONFIG_TUN_VNET_CROSS_LE */

387 388
static inline bool tun_is_little_endian(struct tun_struct *tun)
{
389
	return tun->flags & TUN_VNET_LE ||
390
		tun_legacy_is_little_endian(tun);
391 392
}

393 394
static inline u16 tun16_to_cpu(struct tun_struct *tun, __virtio16 val)
{
395
	return __virtio16_to_cpu(tun_is_little_endian(tun), val);
396 397 398 399
}

static inline __virtio16 cpu_to_tun16(struct tun_struct *tun, u16 val)
{
400
	return __cpu_to_virtio16(tun_is_little_endian(tun), val);
401 402
}

J
Jason Wang 已提交
403 404
static inline u32 tun_hashfn(u32 rxhash)
{
L
Li RongQing 已提交
405
	return rxhash & TUN_MASK_FLOW_ENTRIES;
J
Jason Wang 已提交
406 407 408 409 410 411
}

static struct tun_flow_entry *tun_flow_find(struct hlist_head *head, u32 rxhash)
{
	struct tun_flow_entry *e;

412
	hlist_for_each_entry_rcu(e, head, hash_link) {
J
Jason Wang 已提交
413 414 415 416 417 418 419 420 421 422
		if (e->rxhash == rxhash)
			return e;
	}
	return NULL;
}

static struct tun_flow_entry *tun_flow_create(struct tun_struct *tun,
					      struct hlist_head *head,
					      u32 rxhash, u16 queue_index)
{
423 424
	struct tun_flow_entry *e = kmalloc(sizeof(*e), GFP_ATOMIC);

J
Jason Wang 已提交
425 426 427 428 429
	if (e) {
		tun_debug(KERN_INFO, tun, "create flow: hash %u index %u\n",
			  rxhash, queue_index);
		e->updated = jiffies;
		e->rxhash = rxhash;
430
		e->rps_rxhash = 0;
J
Jason Wang 已提交
431 432 433
		e->queue_index = queue_index;
		e->tun = tun;
		hlist_add_head_rcu(&e->hash_link, head);
434
		++tun->flow_count;
J
Jason Wang 已提交
435 436 437 438 439 440 441 442 443
	}
	return e;
}

static void tun_flow_delete(struct tun_struct *tun, struct tun_flow_entry *e)
{
	tun_debug(KERN_INFO, tun, "delete flow: hash %u index %u\n",
		  e->rxhash, e->queue_index);
	hlist_del_rcu(&e->hash_link);
444
	kfree_rcu(e, rcu);
445
	--tun->flow_count;
J
Jason Wang 已提交
446 447 448 449 450 451 452 453 454
}

static void tun_flow_flush(struct tun_struct *tun)
{
	int i;

	spin_lock_bh(&tun->lock);
	for (i = 0; i < TUN_NUM_FLOW_ENTRIES; i++) {
		struct tun_flow_entry *e;
455
		struct hlist_node *n;
J
Jason Wang 已提交
456

457
		hlist_for_each_entry_safe(e, n, &tun->flows[i], hash_link)
J
Jason Wang 已提交
458 459 460 461 462 463 464 465 466 467 468 469
			tun_flow_delete(tun, e);
	}
	spin_unlock_bh(&tun->lock);
}

static void tun_flow_delete_by_queue(struct tun_struct *tun, u16 queue_index)
{
	int i;

	spin_lock_bh(&tun->lock);
	for (i = 0; i < TUN_NUM_FLOW_ENTRIES; i++) {
		struct tun_flow_entry *e;
470
		struct hlist_node *n;
J
Jason Wang 已提交
471

472
		hlist_for_each_entry_safe(e, n, &tun->flows[i], hash_link) {
J
Jason Wang 已提交
473 474 475 476 477 478 479
			if (e->queue_index == queue_index)
				tun_flow_delete(tun, e);
		}
	}
	spin_unlock_bh(&tun->lock);
}

480
static void tun_flow_cleanup(struct timer_list *t)
J
Jason Wang 已提交
481
{
482
	struct tun_struct *tun = from_timer(tun, t, flow_gc_timer);
J
Jason Wang 已提交
483 484 485 486 487 488 489
	unsigned long delay = tun->ageing_time;
	unsigned long next_timer = jiffies + delay;
	unsigned long count = 0;
	int i;

	tun_debug(KERN_INFO, tun, "tun_flow_cleanup\n");

490
	spin_lock(&tun->lock);
J
Jason Wang 已提交
491 492
	for (i = 0; i < TUN_NUM_FLOW_ENTRIES; i++) {
		struct tun_flow_entry *e;
493
		struct hlist_node *n;
J
Jason Wang 已提交
494

495
		hlist_for_each_entry_safe(e, n, &tun->flows[i], hash_link) {
J
Jason Wang 已提交
496
			unsigned long this_timer;
497

J
Jason Wang 已提交
498
			this_timer = e->updated + delay;
499
			if (time_before_eq(this_timer, jiffies)) {
J
Jason Wang 已提交
500
				tun_flow_delete(tun, e);
501 502 503 504
				continue;
			}
			count++;
			if (time_before(this_timer, next_timer))
J
Jason Wang 已提交
505 506 507 508 509 510
				next_timer = this_timer;
		}
	}

	if (count)
		mod_timer(&tun->flow_gc_timer, round_jiffies_up(next_timer));
511
	spin_unlock(&tun->lock);
J
Jason Wang 已提交
512 513
}

514
static void tun_flow_update(struct tun_struct *tun, u32 rxhash,
515
			    struct tun_file *tfile)
J
Jason Wang 已提交
516 517 518 519
{
	struct hlist_head *head;
	struct tun_flow_entry *e;
	unsigned long delay = tun->ageing_time;
520
	u16 queue_index = tfile->queue_index;
J
Jason Wang 已提交
521

522
	head = &tun->flows[tun_hashfn(rxhash)];
J
Jason Wang 已提交
523 524 525 526 527 528

	rcu_read_lock();

	e = tun_flow_find(head, rxhash);
	if (likely(e)) {
		/* TODO: keep queueing to old queue until it's empty? */
529 530
		if (READ_ONCE(e->queue_index) != queue_index)
			WRITE_ONCE(e->queue_index, queue_index);
531 532
		if (e->updated != jiffies)
			e->updated = jiffies;
533
		sock_rps_record_flow_hash(e->rps_rxhash);
J
Jason Wang 已提交
534 535
	} else {
		spin_lock_bh(&tun->lock);
536 537
		if (!tun_flow_find(head, rxhash) &&
		    tun->flow_count < MAX_TAP_FLOWS)
J
Jason Wang 已提交
538 539 540 541 542 543 544 545 546 547 548
			tun_flow_create(tun, head, rxhash, queue_index);

		if (!timer_pending(&tun->flow_gc_timer))
			mod_timer(&tun->flow_gc_timer,
				  round_jiffies_up(jiffies + delay));
		spin_unlock_bh(&tun->lock);
	}

	rcu_read_unlock();
}

549 550 551 552 553 554
/**
 * Save the hash received in the stack receive path and update the
 * flow_hash table accordingly.
 */
static inline void tun_flow_save_rps_rxhash(struct tun_flow_entry *e, u32 hash)
{
555
	if (unlikely(e->rps_rxhash != hash))
556 557 558
		e->rps_rxhash = hash;
}

559
/* We try to identify a flow through its rxhash. The reason that
S
stephen hemminger 已提交
560
 * we do not check rxq no. is because some cards(e.g 82599), chooses
J
Jason Wang 已提交
561 562
 * the rxq based on the txq where the last packet of the flow comes. As
 * the userspace application move between processors, we may get a
563
 * different rxq no. here.
J
Jason Wang 已提交
564
 */
565
static u16 tun_automq_select_queue(struct tun_struct *tun, struct sk_buff *skb)
J
Jason Wang 已提交
566
{
J
Jason Wang 已提交
567
	struct tun_flow_entry *e;
J
Jason Wang 已提交
568 569 570
	u32 txq = 0;
	u32 numqueues = 0;

571
	numqueues = READ_ONCE(tun->numqueues);
J
Jason Wang 已提交
572

J
Jason Wang 已提交
573
	txq = __skb_get_hash_symmetric(skb);
574 575 576 577 578 579 580
	e = tun_flow_find(&tun->flows[tun_hashfn(txq)], txq);
	if (e) {
		tun_flow_save_rps_rxhash(e, txq);
		txq = e->queue_index;
	} else {
		/* use multiply and shift instead of expensive divide */
		txq = ((u64)txq * numqueues) >> 32;
J
Jason Wang 已提交
581 582 583 584 585
	}

	return txq;
}

586 587
static u16 tun_ebpf_select_queue(struct tun_struct *tun, struct sk_buff *skb)
{
588
	struct tun_prog *prog;
589
	u32 numqueues;
590 591
	u16 ret = 0;

592 593 594 595
	numqueues = READ_ONCE(tun->numqueues);
	if (!numqueues)
		return 0;

596 597 598 599
	prog = rcu_dereference(tun->steering_prog);
	if (prog)
		ret = bpf_prog_run_clear_cb(prog->prog, skb);

600
	return ret % numqueues;
601 602 603
}

static u16 tun_select_queue(struct net_device *dev, struct sk_buff *skb,
604
			    struct net_device *sb_dev)
605 606 607 608 609 610 611 612 613 614 615 616 617 618
{
	struct tun_struct *tun = netdev_priv(dev);
	u16 ret;

	rcu_read_lock();
	if (rcu_dereference(tun->steering_prog))
		ret = tun_ebpf_select_queue(tun, skb);
	else
		ret = tun_automq_select_queue(tun, skb);
	rcu_read_unlock();

	return ret;
}

619 620 621
static inline bool tun_not_capable(struct tun_struct *tun)
{
	const struct cred *cred = current_cred();
622
	struct net *net = dev_net(tun->dev);
623 624 625

	return ((uid_valid(tun->owner) && !uid_eq(cred->euid, tun->owner)) ||
		  (gid_valid(tun->group) && !in_egroup_p(tun->group))) &&
626
		!ns_capable(net->user_ns, CAP_NET_ADMIN);
627 628
}

J
Jason Wang 已提交
629 630 631 632 633 634
static void tun_set_real_num_queues(struct tun_struct *tun)
{
	netif_set_real_num_tx_queues(tun->dev, tun->numqueues);
	netif_set_real_num_rx_queues(tun->dev, tun->numqueues);
}

J
Jason Wang 已提交
635 636 637 638 639 640 641
static void tun_disable_queue(struct tun_struct *tun, struct tun_file *tfile)
{
	tfile->detached = tun;
	list_add_tail(&tfile->next, &tun->disabled);
	++tun->numdisabled;
}

J
Jason Wang 已提交
642
static struct tun_struct *tun_enable_queue(struct tun_file *tfile)
J
Jason Wang 已提交
643 644 645 646 647 648 649 650 651
{
	struct tun_struct *tun = tfile->detached;

	tfile->detached = NULL;
	list_del_init(&tfile->next);
	--tun->numdisabled;
	return tun;
}

652
void tun_ptr_free(void *ptr)
J
Jason Wang 已提交
653 654 655
{
	if (!ptr)
		return;
656 657
	if (tun_is_xdp_frame(ptr)) {
		struct xdp_frame *xdpf = tun_ptr_to_xdp(ptr);
J
Jason Wang 已提交
658

659
		xdp_return_frame(xdpf);
J
Jason Wang 已提交
660 661 662 663
	} else {
		__skb_array_destroy_skb(ptr);
	}
}
664
EXPORT_SYMBOL_GPL(tun_ptr_free);
J
Jason Wang 已提交
665

666 667
static void tun_queue_purge(struct tun_file *tfile)
{
J
Jason Wang 已提交
668
	void *ptr;
J
Jason Wang 已提交
669

J
Jason Wang 已提交
670 671
	while ((ptr = ptr_ring_consume(&tfile->tx_ring)) != NULL)
		tun_ptr_free(ptr);
J
Jason Wang 已提交
672

J
Jason Wang 已提交
673
	skb_queue_purge(&tfile->sk.sk_write_queue);
674 675 676
	skb_queue_purge(&tfile->sk.sk_error_queue);
}

J
Jason Wang 已提交
677 678 679 680 681
static void __tun_detach(struct tun_file *tfile, bool clean)
{
	struct tun_file *ntfile;
	struct tun_struct *tun;

682 683
	tun = rtnl_dereference(tfile->tun);

684
	if (tun && clean) {
E
Eric Dumazet 已提交
685 686
		tun_napi_disable(tfile);
		tun_napi_del(tfile);
687 688
	}

689
	if (tun && !tfile->detached) {
J
Jason Wang 已提交
690 691 692 693 694
		u16 index = tfile->queue_index;
		BUG_ON(index >= tun->numqueues);

		rcu_assign_pointer(tun->tfiles[index],
				   tun->tfiles[tun->numqueues - 1]);
695
		ntfile = rtnl_dereference(tun->tfiles[index]);
J
Jason Wang 已提交
696
		ntfile->queue_index = index;
697 698
		rcu_assign_pointer(tun->tfiles[tun->numqueues - 1],
				   NULL);
J
Jason Wang 已提交
699 700

		--tun->numqueues;
701
		if (clean) {
702
			RCU_INIT_POINTER(tfile->tun, NULL);
J
Jason Wang 已提交
703
			sock_put(&tfile->sk);
704
		} else
J
Jason Wang 已提交
705
			tun_disable_queue(tun, tfile);
J
Jason Wang 已提交
706 707

		synchronize_net();
J
Jason Wang 已提交
708
		tun_flow_delete_by_queue(tun, tun->numqueues + 1);
J
Jason Wang 已提交
709
		/* Drop read queue */
710
		tun_queue_purge(tfile);
J
Jason Wang 已提交
711
		tun_set_real_num_queues(tun);
J
Jason Wang 已提交
712
	} else if (tfile->detached && clean) {
J
Jason Wang 已提交
713
		tun = tun_enable_queue(tfile);
J
Jason Wang 已提交
714 715
		sock_put(&tfile->sk);
	}
J
Jason Wang 已提交
716 717

	if (clean) {
718 719 720
		if (tun && tun->numqueues == 0 && tun->numdisabled == 0) {
			netif_carrier_off(tun->dev);

M
Michael S. Tsirkin 已提交
721
			if (!(tun->flags & IFF_PERSIST) &&
722
			    tun->dev->reg_state == NETREG_REGISTERED)
J
Jason Wang 已提交
723
				unregister_netdevice(tun->dev);
724
		}
725 726
		if (tun)
			xdp_rxq_info_unreg(&tfile->xdp_rxq);
727
		ptr_ring_cleanup(&tfile->tx_ring, tun_ptr_free);
728
		sock_put(&tfile->sk);
J
Jason Wang 已提交
729 730 731 732 733
	}
}

static void tun_detach(struct tun_file *tfile, bool clean)
{
734 735 736
	struct tun_struct *tun;
	struct net_device *dev;

J
Jason Wang 已提交
737
	rtnl_lock();
738 739
	tun = rtnl_dereference(tfile->tun);
	dev = tun ? tun->dev : NULL;
J
Jason Wang 已提交
740
	__tun_detach(tfile, clean);
741 742
	if (dev)
		netdev_state_change(dev);
J
Jason Wang 已提交
743 744 745 746 747 748
	rtnl_unlock();
}

static void tun_detach_all(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
J
Jason Wang 已提交
749
	struct tun_file *tfile, *tmp;
J
Jason Wang 已提交
750 751 752
	int i, n = tun->numqueues;

	for (i = 0; i < n; i++) {
753
		tfile = rtnl_dereference(tun->tfiles[i]);
J
Jason Wang 已提交
754
		BUG_ON(!tfile);
E
Eric Dumazet 已提交
755
		tun_napi_disable(tfile);
756
		tfile->socket.sk->sk_shutdown = RCV_SHUTDOWN;
757
		tfile->socket.sk->sk_data_ready(tfile->socket.sk);
758
		RCU_INIT_POINTER(tfile->tun, NULL);
J
Jason Wang 已提交
759 760
		--tun->numqueues;
	}
761
	list_for_each_entry(tfile, &tun->disabled, next) {
762
		tfile->socket.sk->sk_shutdown = RCV_SHUTDOWN;
763
		tfile->socket.sk->sk_data_ready(tfile->socket.sk);
764
		RCU_INIT_POINTER(tfile->tun, NULL);
765
	}
J
Jason Wang 已提交
766 767 768 769
	BUG_ON(tun->numqueues != 0);

	synchronize_net();
	for (i = 0; i < n; i++) {
770
		tfile = rtnl_dereference(tun->tfiles[i]);
E
Eric Dumazet 已提交
771
		tun_napi_del(tfile);
J
Jason Wang 已提交
772
		/* Drop read queue */
773
		tun_queue_purge(tfile);
774
		xdp_rxq_info_unreg(&tfile->xdp_rxq);
J
Jason Wang 已提交
775 776
		sock_put(&tfile->sk);
	}
J
Jason Wang 已提交
777 778
	list_for_each_entry_safe(tfile, tmp, &tun->disabled, next) {
		tun_enable_queue(tfile);
779
		tun_queue_purge(tfile);
780
		xdp_rxq_info_unreg(&tfile->xdp_rxq);
J
Jason Wang 已提交
781 782 783
		sock_put(&tfile->sk);
	}
	BUG_ON(tun->numdisabled != 0);
J
Jason Wang 已提交
784

M
Michael S. Tsirkin 已提交
785
	if (tun->flags & IFF_PERSIST)
J
Jason Wang 已提交
786
		module_put(THIS_MODULE);
J
Jason Wang 已提交
787 788
}

789
static int tun_attach(struct tun_struct *tun, struct file *file,
790 791
		      bool skip_filter, bool napi, bool napi_frags,
		      bool publish_tun)
792
{
E
Eric W. Biederman 已提交
793
	struct tun_file *tfile = file->private_data;
J
Jason Wang 已提交
794
	struct net_device *dev = tun->dev;
795
	int err;
796

797 798 799 800
	err = security_tun_dev_attach(tfile->socket.sk, tun->security);
	if (err < 0)
		goto out;

801
	err = -EINVAL;
802
	if (rtnl_dereference(tfile->tun) && !tfile->detached)
803 804 805
		goto out;

	err = -EBUSY;
M
Michael S. Tsirkin 已提交
806
	if (!(tun->flags & IFF_MULTI_QUEUE) && tun->numqueues == 1)
J
Jason Wang 已提交
807 808 809
		goto out;

	err = -E2BIG;
J
Jason Wang 已提交
810 811
	if (!tfile->detached &&
	    tun->numqueues + tun->numdisabled == MAX_TAP_QUEUES)
812 813 814
		goto out;

	err = 0;
J
Jason Wang 已提交
815

S
stephen hemminger 已提交
816
	/* Re-attach the filter to persist device */
817
	if (!skip_filter && (tun->filter_attached == true)) {
818 819 820
		lock_sock(tfile->socket.sk);
		err = sk_attach_filter(&tun->fprog, tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
821 822 823
		if (!err)
			goto out;
	}
J
Jason Wang 已提交
824 825

	if (!tfile->detached &&
826 827
	    ptr_ring_resize(&tfile->tx_ring, dev->tx_queue_len,
			    GFP_KERNEL, tun_ptr_free)) {
J
Jason Wang 已提交
828 829 830 831
		err = -ENOMEM;
		goto out;
	}

J
Jason Wang 已提交
832
	tfile->queue_index = tun->numqueues;
833
	tfile->socket.sk->sk_shutdown &= ~RCV_SHUTDOWN;
J
Jesper Dangaard Brouer 已提交
834 835 836 837 838 839 840 841 842 843 844 845 846

	if (tfile->detached) {
		/* Re-attach detached tfile, updating XDP queue_index */
		WARN_ON(!xdp_rxq_info_is_reg(&tfile->xdp_rxq));

		if (tfile->xdp_rxq.queue_index    != tfile->queue_index)
			tfile->xdp_rxq.queue_index = tfile->queue_index;
	} else {
		/* Setup XDP RX-queue info, for new tfile getting attached */
		err = xdp_rxq_info_reg(&tfile->xdp_rxq,
				       tun->dev, tfile->queue_index);
		if (err < 0)
			goto out;
847 848 849 850 851 852
		err = xdp_rxq_info_reg_mem_model(&tfile->xdp_rxq,
						 MEM_TYPE_PAGE_SHARED, NULL);
		if (err < 0) {
			xdp_rxq_info_unreg(&tfile->xdp_rxq);
			goto out;
		}
J
Jesper Dangaard Brouer 已提交
853 854 855
		err = 0;
	}

856
	if (tfile->detached) {
J
Jason Wang 已提交
857
		tun_enable_queue(tfile);
858
	} else {
J
Jason Wang 已提交
859
		sock_hold(&tfile->sk);
E
Eric Dumazet 已提交
860
		tun_napi_init(tun, tfile, napi, napi_frags);
861
	}
J
Jason Wang 已提交
862

J
Jason Wang 已提交
863 864 865
	if (rtnl_dereference(tun->xdp_prog))
		sock_set_flag(&tfile->sk, SOCK_XDP);

J
Jason Wang 已提交
866 867 868 869
	/* device is allowed to go away first, so no need to hold extra
	 * refcnt.
	 */

870 871 872 873
	/* Publish tfile->tun and tun->tfiles only after we've fully
	 * initialized tfile; otherwise we risk using half-initialized
	 * object.
	 */
874 875
	if (publish_tun)
		rcu_assign_pointer(tfile->tun, tun);
876 877
	rcu_assign_pointer(tun->tfiles[tun->numqueues], tfile);
	tun->numqueues++;
878
	tun_set_real_num_queues(tun);
J
Jason Wang 已提交
879 880
out:
	return err;
E
Eric W. Biederman 已提交
881 882
}

883
static struct tun_struct *tun_get(struct tun_file *tfile)
E
Eric W. Biederman 已提交
884
{
885
	struct tun_struct *tun;
886

887 888 889 890 891
	rcu_read_lock();
	tun = rcu_dereference(tfile->tun);
	if (tun)
		dev_hold(tun->dev);
	rcu_read_unlock();
892 893

	return tun;
E
Eric W. Biederman 已提交
894 895 896 897
}

static void tun_put(struct tun_struct *tun)
{
898
	dev_put(tun->dev);
E
Eric W. Biederman 已提交
899 900
}

901
/* TAP filtering */
902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929
static void addr_hash_set(u32 *mask, const u8 *addr)
{
	int n = ether_crc(ETH_ALEN, addr) >> 26;
	mask[n >> 5] |= (1 << (n & 31));
}

static unsigned int addr_hash_test(const u32 *mask, const u8 *addr)
{
	int n = ether_crc(ETH_ALEN, addr) >> 26;
	return mask[n >> 5] & (1 << (n & 31));
}

static int update_filter(struct tap_filter *filter, void __user *arg)
{
	struct { u8 u[ETH_ALEN]; } *addr;
	struct tun_filter uf;
	int err, alen, n, nexact;

	if (copy_from_user(&uf, arg, sizeof(uf)))
		return -EFAULT;

	if (!uf.count) {
		/* Disabled */
		filter->count = 0;
		return 0;
	}

	alen = ETH_ALEN * uf.count;
930 931 932
	addr = memdup_user(arg + sizeof(uf), alen);
	if (IS_ERR(addr))
		return PTR_ERR(addr);
933 934 935 936 937 938 939 940 941 942 943 944 945

	/* The filter is updated without holding any locks. Which is
	 * perfectly safe. We disable it first and in the worst
	 * case we'll accept a few undesired packets. */
	filter->count = 0;
	wmb();

	/* Use first set of addresses as an exact filter */
	for (n = 0; n < uf.count && n < FLT_EXACT_COUNT; n++)
		memcpy(filter->addr[n], addr[n].u, ETH_ALEN);

	nexact = n;

946 947
	/* Remaining multicast addresses are hashed,
	 * unicast will leave the filter disabled. */
948
	memset(filter->mask, 0, sizeof(filter->mask));
949 950 951
	for (; n < uf.count; n++) {
		if (!is_multicast_ether_addr(addr[n].u)) {
			err = 0; /* no filter */
952
			goto free_addr;
953
		}
954
		addr_hash_set(filter->mask, addr[n].u);
955
	}
956 957 958 959 960 961 962 963 964 965 966 967

	/* For ALLMULTI just set the mask to all ones.
	 * This overrides the mask populated above. */
	if ((uf.flags & TUN_FLT_ALLMULTI))
		memset(filter->mask, ~0, sizeof(filter->mask));

	/* Now enable the filter */
	wmb();
	filter->count = nexact;

	/* Return the number of exact filters */
	err = nexact;
968
free_addr:
969 970 971 972 973 974 975 976 977 978 979 980 981 982
	kfree(addr);
	return err;
}

/* Returns: 0 - drop, !=0 - accept */
static int run_filter(struct tap_filter *filter, const struct sk_buff *skb)
{
	/* Cannot use eth_hdr(skb) here because skb_mac_hdr() is incorrect
	 * at this point. */
	struct ethhdr *eh = (struct ethhdr *) skb->data;
	int i;

	/* Exact match */
	for (i = 0; i < filter->count; i++)
983
		if (ether_addr_equal(eh->h_dest, filter->addr[i]))
984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004
			return 1;

	/* Inexact match (multicast only) */
	if (is_multicast_ether_addr(eh->h_dest))
		return addr_hash_test(filter->mask, eh->h_dest);

	return 0;
}

/*
 * Checks whether the packet is accepted or not.
 * Returns: 0 - drop, !=0 - accept
 */
static int check_filter(struct tap_filter *filter, const struct sk_buff *skb)
{
	if (!filter->count)
		return 1;

	return run_filter(filter, skb);
}

L
Linus Torvalds 已提交
1005 1006
/* Network device part of the driver */

1007
static const struct ethtool_ops tun_ethtool_ops;
L
Linus Torvalds 已提交
1008

1009 1010 1011
/* Net device detach from fd. */
static void tun_net_uninit(struct net_device *dev)
{
J
Jason Wang 已提交
1012
	tun_detach_all(dev);
1013 1014
}

L
Linus Torvalds 已提交
1015 1016 1017
/* Net device open. */
static int tun_net_open(struct net_device *dev)
{
J
Jason Wang 已提交
1018
	netif_tx_start_all_queues(dev);
1019

L
Linus Torvalds 已提交
1020 1021 1022 1023 1024 1025
	return 0;
}

/* Net device close. */
static int tun_net_close(struct net_device *dev)
{
J
Jason Wang 已提交
1026
	netif_tx_stop_all_queues(dev);
L
Linus Torvalds 已提交
1027 1028 1029 1030
	return 0;
}

/* Net device start xmit */
1031
static void tun_automq_xmit(struct tun_struct *tun, struct sk_buff *skb)
L
Linus Torvalds 已提交
1032
{
1033
#ifdef CONFIG_RPS
1034
	if (tun->numqueues == 1 && static_branch_unlikely(&rps_needed)) {
1035 1036 1037
		/* Select queue was not called for the skbuff, so we extract the
		 * RPS hash and save it into the flow_table here.
		 */
1038
		struct tun_flow_entry *e;
1039 1040
		__u32 rxhash;

J
Jason Wang 已提交
1041
		rxhash = __skb_get_hash_symmetric(skb);
1042 1043 1044
		e = tun_flow_find(&tun->flows[tun_hashfn(rxhash)], rxhash);
		if (e)
			tun_flow_save_rps_rxhash(e, rxhash);
1045
	}
1046
#endif
1047 1048
}

1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060
static unsigned int run_ebpf_filter(struct tun_struct *tun,
				    struct sk_buff *skb,
				    int len)
{
	struct tun_prog *prog = rcu_dereference(tun->filter_prog);

	if (prog)
		len = bpf_prog_run_clear_cb(prog->prog, skb);

	return len;
}

1061 1062 1063 1064 1065 1066
/* Net device start xmit */
static netdev_tx_t tun_net_xmit(struct sk_buff *skb, struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
	int txq = skb->queue_mapping;
	struct tun_file *tfile;
1067
	int len = skb->len;
1068 1069 1070 1071 1072

	rcu_read_lock();
	tfile = rcu_dereference(tun->tfiles[txq]);

	/* Drop packet if interface is not attached */
1073
	if (!tfile)
1074 1075 1076 1077
		goto drop;

	if (!rcu_dereference(tun->steering_prog))
		tun_automq_xmit(tun, skb);
1078

1079 1080
	tun_debug(KERN_INFO, tun, "tun_net_xmit %d\n", skb->len);

J
Jason Wang 已提交
1081 1082
	BUG_ON(!tfile);

1083 1084 1085 1086 1087 1088
	/* Drop if the filter does not like it.
	 * This is a noop if the filter is disabled.
	 * Filter can be enabled only for the TAP devices. */
	if (!check_filter(&tun->txflt, skb))
		goto drop;

J
Jason Wang 已提交
1089 1090
	if (tfile->socket.sk->sk_filter &&
	    sk_filter(tfile->socket.sk, skb))
M
Michael S. Tsirkin 已提交
1091 1092
		goto drop;

1093
	len = run_ebpf_filter(tun, skb, len);
B
Bjørn Mork 已提交
1094
	if (len == 0 || pskb_trim(skb, len))
1095 1096
		goto drop;

W
Willem de Bruijn 已提交
1097
	if (unlikely(skb_orphan_frags_rx(skb, GFP_ATOMIC)))
1098 1099
		goto drop;

1100
	skb_tx_timestamp(skb);
1101

M
Michael S. Tsirkin 已提交
1102
	/* Orphan the skb - required as we might hang on to it
1103 1104
	 * for indefinite time.
	 */
M
Michael S. Tsirkin 已提交
1105 1106
	skb_orphan(skb);

1107
	nf_reset_ct(skb);
1108

1109
	if (ptr_ring_produce(&tfile->tx_ring, skb))
J
Jason Wang 已提交
1110
		goto drop;
L
Linus Torvalds 已提交
1111 1112

	/* Notify and wake up reader process */
J
Jason Wang 已提交
1113 1114
	if (tfile->flags & TUN_FASYNC)
		kill_fasync(&tfile->fasync, SIGIO, POLL_IN);
1115
	tfile->socket.sk->sk_data_ready(tfile->socket.sk);
1116 1117

	rcu_read_unlock();
1118
	return NETDEV_TX_OK;
L
Linus Torvalds 已提交
1119 1120

drop:
1121
	this_cpu_inc(tun->pcpu_stats->tx_dropped);
1122
	skb_tx_error(skb);
L
Linus Torvalds 已提交
1123
	kfree_skb(skb);
1124
	rcu_read_unlock();
1125
	return NET_XMIT_DROP;
L
Linus Torvalds 已提交
1126 1127
}

1128
static void tun_net_mclist(struct net_device *dev)
L
Linus Torvalds 已提交
1129
{
1130 1131 1132 1133 1134
	/*
	 * This callback is supposed to deal with mc filter in
	 * _rx_ path and has nothing to do with the _tx_ path.
	 * In rx path we always accept everything userspace gives us.
	 */
L
Linus Torvalds 已提交
1135 1136
}

1137 1138
static netdev_features_t tun_net_fix_features(struct net_device *dev,
	netdev_features_t features)
1139 1140 1141 1142 1143
{
	struct tun_struct *tun = netdev_priv(dev);

	return (features & tun->set_features) | (features & ~TUN_USER_FEATURES);
}
1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154

static void tun_set_headroom(struct net_device *dev, int new_hr)
{
	struct tun_struct *tun = netdev_priv(dev);

	if (new_hr < NET_SKB_PAD)
		new_hr = NET_SKB_PAD;

	tun->align = new_hr;
}

1155
static void
1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190
tun_net_get_stats64(struct net_device *dev, struct rtnl_link_stats64 *stats)
{
	u32 rx_dropped = 0, tx_dropped = 0, rx_frame_errors = 0;
	struct tun_struct *tun = netdev_priv(dev);
	struct tun_pcpu_stats *p;
	int i;

	for_each_possible_cpu(i) {
		u64 rxpackets, rxbytes, txpackets, txbytes;
		unsigned int start;

		p = per_cpu_ptr(tun->pcpu_stats, i);
		do {
			start = u64_stats_fetch_begin(&p->syncp);
			rxpackets	= p->rx_packets;
			rxbytes		= p->rx_bytes;
			txpackets	= p->tx_packets;
			txbytes		= p->tx_bytes;
		} while (u64_stats_fetch_retry(&p->syncp, start));

		stats->rx_packets	+= rxpackets;
		stats->rx_bytes		+= rxbytes;
		stats->tx_packets	+= txpackets;
		stats->tx_bytes		+= txbytes;

		/* u32 counters */
		rx_dropped	+= p->rx_dropped;
		rx_frame_errors	+= p->rx_frame_errors;
		tx_dropped	+= p->tx_dropped;
	}
	stats->rx_dropped  = rx_dropped;
	stats->rx_frame_errors = rx_frame_errors;
	stats->tx_dropped = tx_dropped;
}

J
Jason Wang 已提交
1191 1192 1193 1194
static int tun_xdp_set(struct net_device *dev, struct bpf_prog *prog,
		       struct netlink_ext_ack *extack)
{
	struct tun_struct *tun = netdev_priv(dev);
J
Jason Wang 已提交
1195
	struct tun_file *tfile;
J
Jason Wang 已提交
1196
	struct bpf_prog *old_prog;
J
Jason Wang 已提交
1197
	int i;
J
Jason Wang 已提交
1198 1199 1200 1201 1202 1203

	old_prog = rtnl_dereference(tun->xdp_prog);
	rcu_assign_pointer(tun->xdp_prog, prog);
	if (old_prog)
		bpf_prog_put(old_prog);

J
Jason Wang 已提交
1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217
	for (i = 0; i < tun->numqueues; i++) {
		tfile = rtnl_dereference(tun->tfiles[i]);
		if (prog)
			sock_set_flag(&tfile->sk, SOCK_XDP);
		else
			sock_reset_flag(&tfile->sk, SOCK_XDP);
	}
	list_for_each_entry(tfile, &tun->disabled, next) {
		if (prog)
			sock_set_flag(&tfile->sk, SOCK_XDP);
		else
			sock_reset_flag(&tfile->sk, SOCK_XDP);
	}

J
Jason Wang 已提交
1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232
	return 0;
}

static u32 tun_xdp_query(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
	const struct bpf_prog *xdp_prog;

	xdp_prog = rtnl_dereference(tun->xdp_prog);
	if (xdp_prog)
		return xdp_prog->aux->id;

	return 0;
}

1233
static int tun_xdp(struct net_device *dev, struct netdev_bpf *xdp)
J
Jason Wang 已提交
1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245
{
	switch (xdp->command) {
	case XDP_SETUP_PROG:
		return tun_xdp_set(dev, xdp->prog, xdp->extack);
	case XDP_QUERY_PROG:
		xdp->prog_id = tun_xdp_query(dev);
		return 0;
	default:
		return -EINVAL;
	}
}

N
Nicolas Dichtel 已提交
1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260
static int tun_net_change_carrier(struct net_device *dev, bool new_carrier)
{
	if (new_carrier) {
		struct tun_struct *tun = netdev_priv(dev);

		if (!tun->numqueues)
			return -EPERM;

		netif_carrier_on(dev);
	} else {
		netif_carrier_off(dev);
	}
	return 0;
}

S
Stephen Hemminger 已提交
1261
static const struct net_device_ops tun_netdev_ops = {
1262
	.ndo_uninit		= tun_net_uninit,
S
Stephen Hemminger 已提交
1263 1264
	.ndo_open		= tun_net_open,
	.ndo_stop		= tun_net_close,
1265
	.ndo_start_xmit		= tun_net_xmit,
1266
	.ndo_fix_features	= tun_net_fix_features,
J
Jason Wang 已提交
1267
	.ndo_select_queue	= tun_select_queue,
1268
	.ndo_set_rx_headroom	= tun_set_headroom,
1269
	.ndo_get_stats64	= tun_net_get_stats64,
N
Nicolas Dichtel 已提交
1270
	.ndo_change_carrier	= tun_net_change_carrier,
S
Stephen Hemminger 已提交
1271 1272
};

1273 1274 1275 1276 1277 1278 1279 1280
static void __tun_xdp_flush_tfile(struct tun_file *tfile)
{
	/* Notify and wake up reader process */
	if (tfile->flags & TUN_FASYNC)
		kill_fasync(&tfile->fasync, SIGIO, POLL_IN);
	tfile->socket.sk->sk_data_ready(tfile->socket.sk);
}

1281 1282
static int tun_xdp_xmit(struct net_device *dev, int n,
			struct xdp_frame **frames, u32 flags)
J
Jason Wang 已提交
1283 1284 1285 1286
{
	struct tun_struct *tun = netdev_priv(dev);
	struct tun_file *tfile;
	u32 numqueues;
1287 1288 1289
	int drops = 0;
	int cnt = n;
	int i;
J
Jason Wang 已提交
1290

1291
	if (unlikely(flags & ~XDP_XMIT_FLAGS_MASK))
1292 1293
		return -EINVAL;

J
Jason Wang 已提交
1294 1295
	rcu_read_lock();

1296
resample:
J
Jason Wang 已提交
1297 1298
	numqueues = READ_ONCE(tun->numqueues);
	if (!numqueues) {
1299 1300
		rcu_read_unlock();
		return -ENXIO; /* Caller will free/return all frames */
J
Jason Wang 已提交
1301 1302 1303 1304
	}

	tfile = rcu_dereference(tun->tfiles[smp_processor_id() %
					    numqueues]);
1305 1306
	if (unlikely(!tfile))
		goto resample;
1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320

	spin_lock(&tfile->tx_ring.producer_lock);
	for (i = 0; i < n; i++) {
		struct xdp_frame *xdp = frames[i];
		/* Encode the XDP flag into lowest bit for consumer to differ
		 * XDP buffer from sk_buff.
		 */
		void *frame = tun_xdp_to_ptr(xdp);

		if (__ptr_ring_produce(&tfile->tx_ring, frame)) {
			this_cpu_inc(tun->pcpu_stats->tx_dropped);
			xdp_return_frame_rx_napi(xdp);
			drops++;
		}
J
Jason Wang 已提交
1321
	}
1322
	spin_unlock(&tfile->tx_ring.producer_lock);
J
Jason Wang 已提交
1323

1324 1325 1326
	if (flags & XDP_XMIT_FLUSH)
		__tun_xdp_flush_tfile(tfile);

J
Jason Wang 已提交
1327
	rcu_read_unlock();
1328
	return cnt - drops;
J
Jason Wang 已提交
1329 1330
}

1331 1332 1333 1334 1335 1336 1337
static int tun_xdp_tx(struct net_device *dev, struct xdp_buff *xdp)
{
	struct xdp_frame *frame = convert_to_xdp_frame(xdp);

	if (unlikely(!frame))
		return -EOVERFLOW;

1338
	return tun_xdp_xmit(dev, 1, &frame, XDP_XMIT_FLUSH);
J
Jason Wang 已提交
1339 1340
}

S
Stephen Hemminger 已提交
1341
static const struct net_device_ops tap_netdev_ops = {
1342
	.ndo_uninit		= tun_net_uninit,
S
Stephen Hemminger 已提交
1343 1344
	.ndo_open		= tun_net_open,
	.ndo_stop		= tun_net_close,
1345
	.ndo_start_xmit		= tun_net_xmit,
1346
	.ndo_fix_features	= tun_net_fix_features,
1347
	.ndo_set_rx_mode	= tun_net_mclist,
S
Stephen Hemminger 已提交
1348 1349
	.ndo_set_mac_address	= eth_mac_addr,
	.ndo_validate_addr	= eth_validate_addr,
J
Jason Wang 已提交
1350
	.ndo_select_queue	= tun_select_queue,
1351
	.ndo_features_check	= passthru_features_check,
1352
	.ndo_set_rx_headroom	= tun_set_headroom,
1353
	.ndo_get_stats64	= tun_net_get_stats64,
1354
	.ndo_bpf		= tun_xdp,
J
Jason Wang 已提交
1355
	.ndo_xdp_xmit		= tun_xdp_xmit,
N
Nicolas Dichtel 已提交
1356
	.ndo_change_carrier	= tun_net_change_carrier,
S
Stephen Hemminger 已提交
1357 1358
};

1359
static void tun_flow_init(struct tun_struct *tun)
J
Jason Wang 已提交
1360 1361 1362 1363 1364 1365 1366
{
	int i;

	for (i = 0; i < TUN_NUM_FLOW_ENTRIES; i++)
		INIT_HLIST_HEAD(&tun->flows[i]);

	tun->ageing_time = TUN_FLOW_EXPIRE;
1367 1368 1369
	timer_setup(&tun->flow_gc_timer, tun_flow_cleanup, 0);
	mod_timer(&tun->flow_gc_timer,
		  round_jiffies_up(jiffies + tun->ageing_time));
J
Jason Wang 已提交
1370 1371 1372 1373 1374 1375 1376 1377
}

static void tun_flow_uninit(struct tun_struct *tun)
{
	del_timer_sync(&tun->flow_gc_timer);
	tun_flow_flush(tun);
}

1378 1379 1380
#define MIN_MTU 68
#define MAX_MTU 65535

L
Linus Torvalds 已提交
1381 1382 1383 1384
/* Initialize net device. */
static void tun_net_init(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
1385

L
Linus Torvalds 已提交
1386
	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
1387
	case IFF_TUN:
S
Stephen Hemminger 已提交
1388 1389
		dev->netdev_ops = &tun_netdev_ops;

L
Linus Torvalds 已提交
1390 1391 1392 1393 1394 1395
		/* Point-to-Point TUN Device */
		dev->hard_header_len = 0;
		dev->addr_len = 0;
		dev->mtu = 1500;

		/* Zero header length */
1396
		dev->type = ARPHRD_NONE;
L
Linus Torvalds 已提交
1397 1398 1399
		dev->flags = IFF_POINTOPOINT | IFF_NOARP | IFF_MULTICAST;
		break;

M
Michael S. Tsirkin 已提交
1400
	case IFF_TAP:
K
Kusanagi Kouichi 已提交
1401
		dev->netdev_ops = &tap_netdev_ops;
L
Linus Torvalds 已提交
1402 1403
		/* Ethernet TAP Device */
		ether_setup(dev);
1404
		dev->priv_flags &= ~IFF_TX_SKB_SHARING;
1405
		dev->priv_flags |= IFF_LIVE_ADDR_CHANGE;
1406

1407
		eth_hw_addr_random(dev);
1408

L
Linus Torvalds 已提交
1409 1410
		break;
	}
1411 1412 1413

	dev->min_mtu = MIN_MTU;
	dev->max_mtu = MAX_MTU - dev->hard_header_len;
L
Linus Torvalds 已提交
1414 1415
}

1416 1417 1418 1419 1420 1421 1422
static bool tun_sock_writeable(struct tun_struct *tun, struct tun_file *tfile)
{
	struct sock *sk = tfile->socket.sk;

	return (tun->dev->flags & IFF_UP) && sock_writeable(sk);
}

L
Linus Torvalds 已提交
1423 1424 1425
/* Character device part */

/* Poll */
1426
static __poll_t tun_chr_poll(struct file *file, poll_table *wait)
1427
{
1428
	struct tun_file *tfile = file->private_data;
1429
	struct tun_struct *tun = tun_get(tfile);
1430
	struct sock *sk;
1431
	__poll_t mask = 0;
L
Linus Torvalds 已提交
1432 1433

	if (!tun)
1434
		return EPOLLERR;
L
Linus Torvalds 已提交
1435

J
Jason Wang 已提交
1436
	sk = tfile->socket.sk;
1437

1438
	tun_debug(KERN_INFO, tun, "tun_chr_poll\n");
L
Linus Torvalds 已提交
1439

1440
	poll_wait(file, sk_sleep(sk), wait);
1441

1442
	if (!ptr_ring_empty(&tfile->tx_ring))
1443
		mask |= EPOLLIN | EPOLLRDNORM;
L
Linus Torvalds 已提交
1444

1445 1446 1447 1448 1449 1450 1451 1452
	/* Make sure SOCKWQ_ASYNC_NOSPACE is set if not writable to
	 * guarantee EPOLLOUT to be raised by either here or
	 * tun_sock_write_space(). Then process could get notification
	 * after it writes to a down device and meets -EIO.
	 */
	if (tun_sock_writeable(tun, tfile) ||
	    (!test_and_set_bit(SOCKWQ_ASYNC_NOSPACE, &sk->sk_socket->flags) &&
	     tun_sock_writeable(tun, tfile)))
1453
		mask |= EPOLLOUT | EPOLLWRNORM;
1454

1455
	if (tun->dev->reg_state != NETREG_REGISTERED)
1456
		mask = EPOLLERR;
1457

E
Eric W. Biederman 已提交
1458
	tun_put(tun);
L
Linus Torvalds 已提交
1459 1460 1461
	return mask;
}

1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490
static struct sk_buff *tun_napi_alloc_frags(struct tun_file *tfile,
					    size_t len,
					    const struct iov_iter *it)
{
	struct sk_buff *skb;
	size_t linear;
	int err;
	int i;

	if (it->nr_segs > MAX_SKB_FRAGS + 1)
		return ERR_PTR(-ENOMEM);

	local_bh_disable();
	skb = napi_get_frags(&tfile->napi);
	local_bh_enable();
	if (!skb)
		return ERR_PTR(-ENOMEM);

	linear = iov_iter_single_seg_count(it);
	err = __skb_grow(skb, linear);
	if (err)
		goto free;

	skb->len = len;
	skb->data_len = len - linear;
	skb->truesize += skb->data_len;

	for (i = 1; i < it->nr_segs; i++) {
		size_t fragsz = it->iov[i].iov_len;
1491 1492
		struct page *page;
		void *frag;
1493 1494 1495 1496 1497

		if (fragsz == 0 || fragsz > PAGE_SIZE) {
			err = -EINVAL;
			goto free;
		}
1498 1499
		frag = netdev_alloc_frag(fragsz);
		if (!frag) {
1500 1501 1502
			err = -ENOMEM;
			goto free;
		}
1503 1504 1505
		page = virt_to_head_page(frag);
		skb_fill_page_desc(skb, i - 1, page,
				   frag - page_address(page), fragsz);
1506 1507 1508 1509 1510 1511 1512 1513 1514
	}

	return skb;
free:
	/* frees skb and all frags allocated with napi_alloc_frag() */
	napi_free_frags(&tfile->napi);
	return ERR_PTR(err);
}

1515 1516
/* prepad is the amount to reserve at front.  len is length after that.
 * linear is a hint as to how much to copy (usually headers). */
J
Jason Wang 已提交
1517
static struct sk_buff *tun_alloc_skb(struct tun_file *tfile,
1518 1519
				     size_t prepad, size_t len,
				     size_t linear, int noblock)
1520
{
J
Jason Wang 已提交
1521
	struct sock *sk = tfile->socket.sk;
1522
	struct sk_buff *skb;
1523
	int err;
1524 1525

	/* Under a page?  Don't bother with paged skb. */
H
Herbert Xu 已提交
1526
	if (prepad + len < PAGE_SIZE || !linear)
1527
		linear = len;
1528

1529
	skb = sock_alloc_send_pskb(sk, prepad + linear, len - linear, noblock,
1530
				   &err, 0);
1531
	if (!skb)
1532
		return ERR_PTR(err);
1533 1534 1535

	skb_reserve(skb, prepad);
	skb_put(skb, linear);
1536 1537
	skb->data_len = len - linear;
	skb->len += len - linear;
1538 1539 1540 1541

	return skb;
}

J
Jason Wang 已提交
1542 1543 1544 1545 1546 1547 1548 1549 1550 1551
static void tun_rx_batched(struct tun_struct *tun, struct tun_file *tfile,
			   struct sk_buff *skb, int more)
{
	struct sk_buff_head *queue = &tfile->sk.sk_write_queue;
	struct sk_buff_head process_queue;
	u32 rx_batched = tun->rx_batched;
	bool rcv = false;

	if (!rx_batched || (!more && skb_queue_empty(queue))) {
		local_bh_disable();
M
Matthew Cover 已提交
1552
		skb_record_rx_queue(skb, tfile->queue_index);
J
Jason Wang 已提交
1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571
		netif_receive_skb(skb);
		local_bh_enable();
		return;
	}

	spin_lock(&queue->lock);
	if (!more || skb_queue_len(queue) == rx_batched) {
		__skb_queue_head_init(&process_queue);
		skb_queue_splice_tail_init(queue, &process_queue);
		rcv = true;
	} else {
		__skb_queue_tail(queue, skb);
	}
	spin_unlock(&queue->lock);

	if (rcv) {
		struct sk_buff *nskb;

		local_bh_disable();
M
Matthew Cover 已提交
1572 1573
		while ((nskb = __skb_dequeue(&process_queue))) {
			skb_record_rx_queue(nskb, tfile->queue_index);
J
Jason Wang 已提交
1574
			netif_receive_skb(nskb);
M
Matthew Cover 已提交
1575 1576
		}
		skb_record_rx_queue(skb, tfile->queue_index);
J
Jason Wang 已提交
1577 1578 1579 1580 1581
		netif_receive_skb(skb);
		local_bh_enable();
	}
}

1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603
static bool tun_can_build_skb(struct tun_struct *tun, struct tun_file *tfile,
			      int len, int noblock, bool zerocopy)
{
	if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
		return false;

	if (tfile->socket.sk->sk_sndbuf != INT_MAX)
		return false;

	if (!noblock)
		return false;

	if (zerocopy)
		return false;

	if (SKB_DATA_ALIGN(len + TUN_RX_PAD) +
	    SKB_DATA_ALIGN(sizeof(struct skb_shared_info)) > PAGE_SIZE)
		return false;

	return true;
}

1604 1605
static struct sk_buff *__tun_build_skb(struct tun_file *tfile,
				       struct page_frag *alloc_frag, char *buf,
J
Jason Wang 已提交
1606
				       int buflen, int len, int pad)
1607 1608 1609 1610 1611 1612
{
	struct sk_buff *skb = build_skb(buf, buflen);

	if (!skb)
		return ERR_PTR(-ENOMEM);

J
Jason Wang 已提交
1613
	skb_reserve(skb, pad);
1614
	skb_put(skb, len);
1615
	skb_set_owner_w(skb, tfile->socket.sk);
1616 1617 1618 1619 1620 1621 1622

	get_page(alloc_frag->page);
	alloc_frag->offset += buflen;

	return skb;
}

J
Jason Wang 已提交
1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654
static int tun_xdp_act(struct tun_struct *tun, struct bpf_prog *xdp_prog,
		       struct xdp_buff *xdp, u32 act)
{
	int err;

	switch (act) {
	case XDP_REDIRECT:
		err = xdp_do_redirect(tun->dev, xdp, xdp_prog);
		if (err)
			return err;
		break;
	case XDP_TX:
		err = tun_xdp_tx(tun->dev, xdp);
		if (err < 0)
			return err;
		break;
	case XDP_PASS:
		break;
	default:
		bpf_warn_invalid_xdp_action(act);
		/* fall through */
	case XDP_ABORTED:
		trace_xdp_exception(tun->dev, xdp_prog, act);
		/* fall through */
	case XDP_DROP:
		this_cpu_inc(tun->pcpu_stats->rx_dropped);
		break;
	}

	return act;
}

J
Jason Wang 已提交
1655 1656
static struct sk_buff *tun_build_skb(struct tun_struct *tun,
				     struct tun_file *tfile,
1657
				     struct iov_iter *from,
J
Jason Wang 已提交
1658
				     struct virtio_net_hdr *hdr,
J
Jason Wang 已提交
1659
				     int len, int *skb_xdp)
1660
{
1661
	struct page_frag *alloc_frag = &current->task_frag;
J
Jason Wang 已提交
1662
	struct bpf_prog *xdp_prog;
1663
	int buflen = SKB_DATA_ALIGN(sizeof(struct skb_shared_info));
1664 1665
	char *buf;
	size_t copied;
J
Jason Wang 已提交
1666 1667
	int pad = TUN_RX_PAD;
	int err = 0;
1668 1669 1670 1671

	rcu_read_lock();
	xdp_prog = rcu_dereference(tun->xdp_prog);
	if (xdp_prog)
1672
		pad += XDP_PACKET_HEADROOM;
1673 1674
	buflen += SKB_DATA_ALIGN(len + pad);
	rcu_read_unlock();
1675

1676
	alloc_frag->offset = ALIGN((u64)alloc_frag->offset, SMP_CACHE_BYTES);
1677 1678 1679 1680 1681
	if (unlikely(!skb_page_frag_refill(buflen, alloc_frag, GFP_KERNEL)))
		return ERR_PTR(-ENOMEM);

	buf = (char *)page_address(alloc_frag->page) + alloc_frag->offset;
	copied = copy_page_from_iter(alloc_frag->page,
1682
				     alloc_frag->offset + pad,
1683 1684 1685 1686
				     len, from);
	if (copied != len)
		return ERR_PTR(-EFAULT);

1687 1688 1689 1690
	/* There's a small window that XDP may be set after the check
	 * of xdp_prog above, this should be rare and for simplicity
	 * we do XDP on skb in case the headroom is not enough.
	 */
1691
	if (hdr->gso_type || !xdp_prog) {
J
Jason Wang 已提交
1692
		*skb_xdp = 1;
1693 1694
		return __tun_build_skb(tfile, alloc_frag, buf, buflen, len,
				       pad);
1695 1696 1697
	}

	*skb_xdp = 0;
J
Jason Wang 已提交
1698

1699
	local_bh_disable();
J
Jason Wang 已提交
1700 1701
	rcu_read_lock();
	xdp_prog = rcu_dereference(tun->xdp_prog);
J
Jason Wang 已提交
1702
	if (xdp_prog) {
J
Jason Wang 已提交
1703 1704 1705 1706
		struct xdp_buff xdp;
		u32 act;

		xdp.data_hard_start = buf;
1707
		xdp.data = buf + pad;
1708
		xdp_set_data_meta_invalid(&xdp);
J
Jason Wang 已提交
1709
		xdp.data_end = xdp.data + len;
J
Jesper Dangaard Brouer 已提交
1710
		xdp.rxq = &tfile->xdp_rxq;
J
Jason Wang 已提交
1711

J
Jason Wang 已提交
1712 1713
		act = bpf_prog_run_xdp(xdp_prog, &xdp);
		if (act == XDP_REDIRECT || act == XDP_TX) {
J
Jason Wang 已提交
1714 1715
			get_page(alloc_frag->page);
			alloc_frag->offset += buflen;
J
Jason Wang 已提交
1716
		}
J
Jason Wang 已提交
1717 1718 1719
		err = tun_xdp_act(tun, xdp_prog, &xdp, act);
		if (err < 0)
			goto err_xdp;
1720 1721
		if (err == XDP_REDIRECT)
			xdp_do_flush_map();
J
Jason Wang 已提交
1722 1723 1724 1725 1726
		if (err != XDP_PASS)
			goto out;

		pad = xdp.data - xdp.data_hard_start;
		len = xdp.data_end - xdp.data;
J
Jason Wang 已提交
1727
	}
1728 1729
	rcu_read_unlock();
	local_bh_enable();
J
Jason Wang 已提交
1730

1731
	return __tun_build_skb(tfile, alloc_frag, buf, buflen, len, pad);
J
Jason Wang 已提交
1732

J
Jason Wang 已提交
1733
err_xdp:
J
Jason Wang 已提交
1734
	put_page(alloc_frag->page);
1735
out:
J
Jason Wang 已提交
1736
	rcu_read_unlock();
1737
	local_bh_enable();
J
Jason Wang 已提交
1738
	return NULL;
1739 1740
}

L
Linus Torvalds 已提交
1741
/* Get packet from user space buffer */
J
Jason Wang 已提交
1742
static ssize_t tun_get_user(struct tun_struct *tun, struct tun_file *tfile,
1743
			    void *msg_control, struct iov_iter *from,
J
Jason Wang 已提交
1744
			    int noblock, bool more)
L
Linus Torvalds 已提交
1745
{
1746
	struct tun_pi pi = { 0, cpu_to_be16(ETH_P_IP) };
L
Linus Torvalds 已提交
1747
	struct sk_buff *skb;
1748
	size_t total_len = iov_iter_count(from);
1749
	size_t len = total_len, align = tun->align, linear;
1750
	struct virtio_net_hdr gso = { 0 };
1751
	struct tun_pcpu_stats *stats;
1752
	int good_linear;
1753 1754 1755
	int copylen;
	bool zerocopy = false;
	int err;
1756
	u32 rxhash = 0;
J
Jason Wang 已提交
1757
	int skb_xdp = 1;
E
Eric Dumazet 已提交
1758
	bool frags = tun_napi_frags_enabled(tfile);
L
Linus Torvalds 已提交
1759

M
Michael S. Tsirkin 已提交
1760
	if (!(tun->flags & IFF_NO_PI)) {
1761
		if (len < sizeof(pi))
L
Linus Torvalds 已提交
1762
			return -EINVAL;
1763
		len -= sizeof(pi);
L
Linus Torvalds 已提交
1764

1765
		if (!copy_from_iter_full(&pi, sizeof(pi), from))
L
Linus Torvalds 已提交
1766 1767 1768
			return -EFAULT;
	}

M
Michael S. Tsirkin 已提交
1769
	if (tun->flags & IFF_VNET_HDR) {
W
Willem de Bruijn 已提交
1770 1771 1772
		int vnet_hdr_sz = READ_ONCE(tun->vnet_hdr_sz);

		if (len < vnet_hdr_sz)
1773
			return -EINVAL;
W
Willem de Bruijn 已提交
1774
		len -= vnet_hdr_sz;
1775

1776
		if (!copy_from_iter_full(&gso, sizeof(gso), from))
1777 1778
			return -EFAULT;

1779
		if ((gso.flags & VIRTIO_NET_HDR_F_NEEDS_CSUM) &&
1780 1781
		    tun16_to_cpu(tun, gso.csum_start) + tun16_to_cpu(tun, gso.csum_offset) + 2 > tun16_to_cpu(tun, gso.hdr_len))
			gso.hdr_len = cpu_to_tun16(tun, tun16_to_cpu(tun, gso.csum_start) + tun16_to_cpu(tun, gso.csum_offset) + 2);
1782

1783
		if (tun16_to_cpu(tun, gso.hdr_len) > len)
1784
			return -EINVAL;
W
Willem de Bruijn 已提交
1785
		iov_iter_advance(from, vnet_hdr_sz - sizeof(gso));
1786 1787
	}

M
Michael S. Tsirkin 已提交
1788
	if ((tun->flags & TUN_TYPE_MASK) == IFF_TAP) {
1789
		align += NET_IP_ALIGN;
H
Herbert Xu 已提交
1790
		if (unlikely(len < ETH_HLEN ||
1791
			     (gso.hdr_len && tun16_to_cpu(tun, gso.hdr_len) < ETH_HLEN)))
1792 1793
			return -EINVAL;
	}
1794

1795 1796
	good_linear = SKB_MAX_HEAD(align);

1797
	if (msg_control) {
1798 1799
		struct iov_iter i = *from;

1800 1801
		/* There are 256 bytes to be copied in skb, so there is
		 * enough room for skb expand head in case it is used.
1802 1803
		 * The rest of the buffer is mapped from userspace.
		 */
1804
		copylen = gso.hdr_len ? tun16_to_cpu(tun, gso.hdr_len) : GOODCOPY_LEN;
1805 1806
		if (copylen > good_linear)
			copylen = good_linear;
1807
		linear = copylen;
1808 1809
		iov_iter_advance(&i, copylen);
		if (iov_iter_npages(&i, INT_MAX) <= MAX_SKB_FRAGS)
1810 1811 1812
			zerocopy = true;
	}

1813
	if (!frags && tun_can_build_skb(tun, tfile, len, noblock, zerocopy)) {
J
Jason Wang 已提交
1814 1815 1816 1817 1818
		/* For the packet that is not easy to be processed
		 * (e.g gso or jumbo packet), we will do it at after
		 * skb was created with generic XDP routine.
		 */
		skb = tun_build_skb(tun, tfile, from, &gso, len, &skb_xdp);
1819
		if (IS_ERR(skb)) {
1820
			this_cpu_inc(tun->pcpu_stats->rx_dropped);
1821 1822
			return PTR_ERR(skb);
		}
J
Jason Wang 已提交
1823 1824
		if (!skb)
			return total_len;
1825 1826 1827 1828 1829 1830 1831 1832
	} else {
		if (!zerocopy) {
			copylen = len;
			if (tun16_to_cpu(tun, gso.hdr_len) > good_linear)
				linear = good_linear;
			else
				linear = tun16_to_cpu(tun, gso.hdr_len);
		}
L
Linus Torvalds 已提交
1833

1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846
		if (frags) {
			mutex_lock(&tfile->napi_mutex);
			skb = tun_napi_alloc_frags(tfile, copylen, from);
			/* tun_napi_alloc_frags() enforces a layout for the skb.
			 * If zerocopy is enabled, then this layout will be
			 * overwritten by zerocopy_sg_from_iter().
			 */
			zerocopy = false;
		} else {
			skb = tun_alloc_skb(tfile, align, copylen, linear,
					    noblock);
		}

1847 1848 1849
		if (IS_ERR(skb)) {
			if (PTR_ERR(skb) != -EAGAIN)
				this_cpu_inc(tun->pcpu_stats->rx_dropped);
1850 1851
			if (frags)
				mutex_unlock(&tfile->napi_mutex);
1852 1853
			return PTR_ERR(skb);
		}
1854

1855 1856 1857 1858 1859 1860
		if (zerocopy)
			err = zerocopy_sg_from_iter(skb, from);
		else
			err = skb_copy_datagram_from_iter(skb, 0, from, len);

		if (err) {
E
Eric Dumazet 已提交
1861 1862
			err = -EFAULT;
drop:
1863 1864
			this_cpu_inc(tun->pcpu_stats->rx_dropped);
			kfree_skb(skb);
1865 1866 1867 1868 1869
			if (frags) {
				tfile->napi.skb = NULL;
				mutex_unlock(&tfile->napi_mutex);
			}

E
Eric Dumazet 已提交
1870
			return err;
1871
		}
D
Dave Jones 已提交
1872
	}
L
Linus Torvalds 已提交
1873

1874
	if (virtio_net_hdr_to_skb(skb, &gso, tun_is_little_endian(tun))) {
1875 1876
		this_cpu_inc(tun->pcpu_stats->rx_frame_errors);
		kfree_skb(skb);
1877 1878 1879 1880 1881
		if (frags) {
			tfile->napi.skb = NULL;
			mutex_unlock(&tfile->napi_mutex);
		}

1882 1883 1884
		return -EINVAL;
	}

L
Linus Torvalds 已提交
1885
	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
1886 1887
	case IFF_TUN:
		if (tun->flags & IFF_NO_PI) {
1888 1889 1890 1891
			u8 ip_version = skb->len ? (skb->data[0] >> 4) : 0;

			switch (ip_version) {
			case 4:
1892 1893
				pi.proto = htons(ETH_P_IP);
				break;
1894
			case 6:
1895 1896 1897
				pi.proto = htons(ETH_P_IPV6);
				break;
			default:
1898
				this_cpu_inc(tun->pcpu_stats->rx_dropped);
1899 1900 1901 1902 1903
				kfree_skb(skb);
				return -EINVAL;
			}
		}

1904
		skb_reset_mac_header(skb);
L
Linus Torvalds 已提交
1905
		skb->protocol = pi.proto;
1906
		skb->dev = tun->dev;
L
Linus Torvalds 已提交
1907
		break;
M
Michael S. Tsirkin 已提交
1908
	case IFF_TAP:
1909 1910
		if (!frags)
			skb->protocol = eth_type_trans(skb, tun->dev);
L
Linus Torvalds 已提交
1911
		break;
1912
	}
L
Linus Torvalds 已提交
1913

1914 1915 1916 1917
	/* copy skb_ubuf_info for callback when skb has no error */
	if (zerocopy) {
		skb_shinfo(skb)->destructor_arg = msg_control;
		skb_shinfo(skb)->tx_flags |= SKBTX_DEV_ZEROCOPY;
1918
		skb_shinfo(skb)->tx_flags |= SKBTX_SHARED_FRAG;
1919 1920 1921
	} else if (msg_control) {
		struct ubuf_info *uarg = msg_control;
		uarg->callback(uarg, false);
1922 1923
	}

1924
	skb_reset_network_header(skb);
1925
	skb_probe_transport_header(skb);
1926

J
Jason Wang 已提交
1927
	if (skb_xdp) {
J
Jason Wang 已提交
1928 1929 1930
		struct bpf_prog *xdp_prog;
		int ret;

1931
		local_bh_disable();
J
Jason Wang 已提交
1932 1933 1934 1935 1936 1937
		rcu_read_lock();
		xdp_prog = rcu_dereference(tun->xdp_prog);
		if (xdp_prog) {
			ret = do_xdp_generic(xdp_prog, skb);
			if (ret != XDP_PASS) {
				rcu_read_unlock();
1938
				local_bh_enable();
J
Jason Wang 已提交
1939 1940 1941 1942
				return total_len;
			}
		}
		rcu_read_unlock();
1943
		local_bh_enable();
J
Jason Wang 已提交
1944 1945
	}

1946 1947 1948 1949 1950 1951
	/* Compute the costly rx hash only if needed for flow updates.
	 * We may get a very small possibility of OOO during switching, not
	 * worth to optimize.
	 */
	if (!rcu_access_pointer(tun->steering_prog) && tun->numqueues > 1 &&
	    !tfile->detached)
1952
		rxhash = __skb_get_hash_symmetric(skb);
1953

E
Eric Dumazet 已提交
1954 1955 1956
	rcu_read_lock();
	if (unlikely(!(tun->dev->flags & IFF_UP))) {
		err = -EIO;
1957
		rcu_read_unlock();
E
Eric Dumazet 已提交
1958 1959 1960
		goto drop;
	}

1961 1962
	if (frags) {
		/* Exercise flow dissector code path. */
1963 1964
		u32 headlen = eth_get_headlen(tun->dev, skb->data,
					      skb_headlen(skb));
1965

1966
		if (unlikely(headlen > skb_headlen(skb))) {
1967 1968
			this_cpu_inc(tun->pcpu_stats->rx_dropped);
			napi_free_frags(&tfile->napi);
E
Eric Dumazet 已提交
1969
			rcu_read_unlock();
1970 1971 1972 1973 1974 1975 1976 1977 1978
			mutex_unlock(&tfile->napi_mutex);
			WARN_ON(1);
			return -ENOMEM;
		}

		local_bh_disable();
		napi_gro_frags(&tfile->napi);
		local_bh_enable();
		mutex_unlock(&tfile->napi_mutex);
1979
	} else if (tfile->napi_enabled) {
1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995 1996
		struct sk_buff_head *queue = &tfile->sk.sk_write_queue;
		int queue_len;

		spin_lock_bh(&queue->lock);
		__skb_queue_tail(queue, skb);
		queue_len = skb_queue_len(queue);
		spin_unlock(&queue->lock);

		if (!more || queue_len > NAPI_POLL_WEIGHT)
			napi_schedule(&tfile->napi);

		local_bh_enable();
	} else if (!IS_ENABLED(CONFIG_4KSTACKS)) {
		tun_rx_batched(tun, tfile, skb, more);
	} else {
		netif_rx_ni(skb);
	}
E
Eric Dumazet 已提交
1997
	rcu_read_unlock();
1998

1999 2000 2001 2002 2003 2004
	stats = get_cpu_ptr(tun->pcpu_stats);
	u64_stats_update_begin(&stats->syncp);
	stats->rx_packets++;
	stats->rx_bytes += len;
	u64_stats_update_end(&stats->syncp);
	put_cpu_ptr(stats);
L
Linus Torvalds 已提交
2005

2006 2007 2008
	if (rxhash)
		tun_flow_update(tun, rxhash, tfile);

2009
	return total_len;
2010
}
L
Linus Torvalds 已提交
2011

2012
static ssize_t tun_chr_write_iter(struct kiocb *iocb, struct iov_iter *from)
L
Linus Torvalds 已提交
2013
{
2014
	struct file *file = iocb->ki_filp;
J
Jason Wang 已提交
2015
	struct tun_file *tfile = file->private_data;
2016
	struct tun_struct *tun = tun_get(tfile);
E
Eric W. Biederman 已提交
2017
	ssize_t result;
L
Linus Torvalds 已提交
2018 2019 2020 2021

	if (!tun)
		return -EBADFD;

J
Jason Wang 已提交
2022 2023
	result = tun_get_user(tun, tfile, NULL, from,
			      file->f_flags & O_NONBLOCK, false);
E
Eric W. Biederman 已提交
2024 2025 2026

	tun_put(tun);
	return result;
L
Linus Torvalds 已提交
2027 2028
}

J
Jason Wang 已提交
2029 2030
static ssize_t tun_put_user_xdp(struct tun_struct *tun,
				struct tun_file *tfile,
2031
				struct xdp_frame *xdp_frame,
J
Jason Wang 已提交
2032 2033 2034
				struct iov_iter *iter)
{
	int vnet_hdr_sz = 0;
2035
	size_t size = xdp_frame->len;
J
Jason Wang 已提交
2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050
	struct tun_pcpu_stats *stats;
	size_t ret;

	if (tun->flags & IFF_VNET_HDR) {
		struct virtio_net_hdr gso = { 0 };

		vnet_hdr_sz = READ_ONCE(tun->vnet_hdr_sz);
		if (unlikely(iov_iter_count(iter) < vnet_hdr_sz))
			return -EINVAL;
		if (unlikely(copy_to_iter(&gso, sizeof(gso), iter) !=
			     sizeof(gso)))
			return -EFAULT;
		iov_iter_advance(iter, vnet_hdr_sz - sizeof(gso));
	}

2051
	ret = copy_to_iter(xdp_frame->data, size, iter) + vnet_hdr_sz;
J
Jason Wang 已提交
2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062

	stats = get_cpu_ptr(tun->pcpu_stats);
	u64_stats_update_begin(&stats->syncp);
	stats->tx_packets++;
	stats->tx_bytes += ret;
	u64_stats_update_end(&stats->syncp);
	put_cpu_ptr(tun->pcpu_stats);

	return ret;
}

L
Linus Torvalds 已提交
2063
/* Put packet to the user space buffer */
2064
static ssize_t tun_put_user(struct tun_struct *tun,
J
Jason Wang 已提交
2065
			    struct tun_file *tfile,
2066
			    struct sk_buff *skb,
H
Herbert Xu 已提交
2067
			    struct iov_iter *iter)
L
Linus Torvalds 已提交
2068 2069
{
	struct tun_pi pi = { 0, skb->protocol };
2070
	struct tun_pcpu_stats *stats;
H
Herbert Xu 已提交
2071
	ssize_t total;
2072
	int vlan_offset = 0;
2073
	int vlan_hlen = 0;
H
Herbert Xu 已提交
2074
	int vnet_hdr_sz = 0;
2075

2076
	if (skb_vlan_tag_present(skb))
2077
		vlan_hlen = VLAN_HLEN;
L
Linus Torvalds 已提交
2078

M
Michael S. Tsirkin 已提交
2079
	if (tun->flags & IFF_VNET_HDR)
W
Willem de Bruijn 已提交
2080
		vnet_hdr_sz = READ_ONCE(tun->vnet_hdr_sz);
L
Linus Torvalds 已提交
2081

H
Herbert Xu 已提交
2082 2083
	total = skb->len + vlan_hlen + vnet_hdr_sz;

M
Michael S. Tsirkin 已提交
2084
	if (!(tun->flags & IFF_NO_PI)) {
H
Herbert Xu 已提交
2085
		if (iov_iter_count(iter) < sizeof(pi))
L
Linus Torvalds 已提交
2086 2087
			return -EINVAL;

H
Herbert Xu 已提交
2088 2089
		total += sizeof(pi);
		if (iov_iter_count(iter) < total) {
L
Linus Torvalds 已提交
2090 2091 2092
			/* Packet will be striped */
			pi.flags |= TUN_PKT_STRIP;
		}
2093

H
Herbert Xu 已提交
2094
		if (copy_to_iter(&pi, sizeof(pi), iter) != sizeof(pi))
L
Linus Torvalds 已提交
2095
			return -EFAULT;
2096
	}
L
Linus Torvalds 已提交
2097

H
Herbert Xu 已提交
2098
	if (vnet_hdr_sz) {
2099
		struct virtio_net_hdr gso;
2100

H
Herbert Xu 已提交
2101
		if (iov_iter_count(iter) < vnet_hdr_sz)
2102 2103
			return -EINVAL;

2104
		if (virtio_net_hdr_from_skb(skb, &gso,
2105 2106
					    tun_is_little_endian(tun), true,
					    vlan_hlen)) {
2107
			struct skb_shared_info *sinfo = skb_shinfo(skb);
2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118
			pr_err("unexpected GSO type: "
			       "0x%x, gso_size %d, hdr_len %d\n",
			       sinfo->gso_type, tun16_to_cpu(tun, gso.gso_size),
			       tun16_to_cpu(tun, gso.hdr_len));
			print_hex_dump(KERN_ERR, "tun: ",
				       DUMP_PREFIX_NONE,
				       16, 1, skb->head,
				       min((int)tun16_to_cpu(tun, gso.hdr_len), 64), true);
			WARN_ON_ONCE(1);
			return -EINVAL;
		}
2119

H
Herbert Xu 已提交
2120
		if (copy_to_iter(&gso, sizeof(gso), iter) != sizeof(gso))
2121
			return -EFAULT;
2122 2123

		iov_iter_advance(iter, vnet_hdr_sz - sizeof(gso));
2124 2125
	}

2126
	if (vlan_hlen) {
H
Herbert Xu 已提交
2127
		int ret;
2128
		struct veth veth;
J
Jason Wang 已提交
2129 2130

		veth.h_vlan_proto = skb->vlan_proto;
2131
		veth.h_vlan_TCI = htons(skb_vlan_tag_get(skb));
J
Jason Wang 已提交
2132 2133 2134

		vlan_offset = offsetof(struct vlan_ethhdr, h_vlan_proto);

H
Herbert Xu 已提交
2135 2136
		ret = skb_copy_datagram_iter(skb, 0, iter, vlan_offset);
		if (ret || !iov_iter_count(iter))
J
Jason Wang 已提交
2137 2138
			goto done;

H
Herbert Xu 已提交
2139 2140
		ret = copy_to_iter(&veth, sizeof(veth), iter);
		if (ret != sizeof(veth) || !iov_iter_count(iter))
J
Jason Wang 已提交
2141 2142
			goto done;
	}
L
Linus Torvalds 已提交
2143

H
Herbert Xu 已提交
2144
	skb_copy_datagram_iter(skb, vlan_offset, iter, skb->len - vlan_offset);
L
Linus Torvalds 已提交
2145

J
Jason Wang 已提交
2146
done:
2147 2148 2149 2150 2151 2152 2153
	/* caller is in process context, */
	stats = get_cpu_ptr(tun->pcpu_stats);
	u64_stats_update_begin(&stats->syncp);
	stats->tx_packets++;
	stats->tx_bytes += skb->len + vlan_hlen;
	u64_stats_update_end(&stats->syncp);
	put_cpu_ptr(tun->pcpu_stats);
L
Linus Torvalds 已提交
2154 2155 2156 2157

	return total;
}

J
Jason Wang 已提交
2158
static void *tun_ring_recv(struct tun_file *tfile, int noblock, int *err)
J
Jason Wang 已提交
2159 2160
{
	DECLARE_WAITQUEUE(wait, current);
J
Jason Wang 已提交
2161
	void *ptr = NULL;
J
Jason Wang 已提交
2162
	int error = 0;
J
Jason Wang 已提交
2163

J
Jason Wang 已提交
2164 2165
	ptr = ptr_ring_consume(&tfile->tx_ring);
	if (ptr)
J
Jason Wang 已提交
2166 2167
		goto out;
	if (noblock) {
J
Jason Wang 已提交
2168
		error = -EAGAIN;
J
Jason Wang 已提交
2169 2170 2171
		goto out;
	}

2172
	add_wait_queue(&tfile->socket.wq.wait, &wait);
J
Jason Wang 已提交
2173 2174

	while (1) {
T
Timur Celik 已提交
2175
		set_current_state(TASK_INTERRUPTIBLE);
J
Jason Wang 已提交
2176 2177
		ptr = ptr_ring_consume(&tfile->tx_ring);
		if (ptr)
J
Jason Wang 已提交
2178 2179
			break;
		if (signal_pending(current)) {
J
Jason Wang 已提交
2180
			error = -ERESTARTSYS;
J
Jason Wang 已提交
2181 2182 2183
			break;
		}
		if (tfile->socket.sk->sk_shutdown & RCV_SHUTDOWN) {
J
Jason Wang 已提交
2184
			error = -EFAULT;
J
Jason Wang 已提交
2185 2186 2187 2188 2189 2190
			break;
		}

		schedule();
	}

2191
	__set_current_state(TASK_RUNNING);
2192
	remove_wait_queue(&tfile->socket.wq.wait, &wait);
J
Jason Wang 已提交
2193 2194

out:
J
Jason Wang 已提交
2195
	*err = error;
J
Jason Wang 已提交
2196
	return ptr;
J
Jason Wang 已提交
2197 2198
}

J
Jason Wang 已提交
2199
static ssize_t tun_do_read(struct tun_struct *tun, struct tun_file *tfile,
2200
			   struct iov_iter *to,
J
Jason Wang 已提交
2201
			   int noblock, void *ptr)
L
Linus Torvalds 已提交
2202
{
2203
	ssize_t ret;
J
Jason Wang 已提交
2204
	int err;
L
Linus Torvalds 已提交
2205

2206
	tun_debug(KERN_INFO, tun, "tun_do_read\n");
L
Linus Torvalds 已提交
2207

W
Wei Xu 已提交
2208
	if (!iov_iter_count(to)) {
J
Jason Wang 已提交
2209
		tun_ptr_free(ptr);
2210
		return 0;
W
Wei Xu 已提交
2211
	}
L
Linus Torvalds 已提交
2212

J
Jason Wang 已提交
2213
	if (!ptr) {
2214
		/* Read frames from ring */
J
Jason Wang 已提交
2215 2216
		ptr = tun_ring_recv(tfile, noblock, &err);
		if (!ptr)
2217 2218
			return err;
	}
H
Herbert Xu 已提交
2219

2220 2221
	if (tun_is_xdp_frame(ptr)) {
		struct xdp_frame *xdpf = tun_ptr_to_xdp(ptr);
J
Jason Wang 已提交
2222

2223
		ret = tun_put_user_xdp(tun, tfile, xdpf, to);
2224
		xdp_return_frame(xdpf);
J
Jason Wang 已提交
2225 2226 2227 2228 2229 2230 2231 2232 2233
	} else {
		struct sk_buff *skb = ptr;

		ret = tun_put_user(tun, tfile, skb, to);
		if (unlikely(ret < 0))
			kfree_skb(skb);
		else
			consume_skb(skb);
	}
L
Linus Torvalds 已提交
2234

2235 2236 2237
	return ret;
}

2238
static ssize_t tun_chr_read_iter(struct kiocb *iocb, struct iov_iter *to)
2239 2240 2241
{
	struct file *file = iocb->ki_filp;
	struct tun_file *tfile = file->private_data;
2242
	struct tun_struct *tun = tun_get(tfile);
2243
	ssize_t len = iov_iter_count(to), ret;
2244 2245 2246

	if (!tun)
		return -EBADFD;
2247
	ret = tun_do_read(tun, tfile, to, file->f_flags & O_NONBLOCK, NULL);
2248
	ret = min_t(ssize_t, ret, len);
Z
Zhi Yong Wu 已提交
2249 2250
	if (ret > 0)
		iocb->ki_pos = ret;
E
Eric W. Biederman 已提交
2251
	tun_put(tun);
L
Linus Torvalds 已提交
2252 2253 2254
	return ret;
}

2255
static void tun_prog_free(struct rcu_head *rcu)
2256
{
2257
	struct tun_prog *prog = container_of(rcu, struct tun_prog, rcu);
2258 2259 2260 2261 2262

	bpf_prog_destroy(prog->prog);
	kfree(prog);
}

J
Jason Wang 已提交
2263 2264
static int __tun_set_ebpf(struct tun_struct *tun,
			  struct tun_prog __rcu **prog_p,
2265
			  struct bpf_prog *prog)
2266
{
2267
	struct tun_prog *old, *new = NULL;
2268 2269 2270 2271 2272 2273 2274 2275

	if (prog) {
		new = kmalloc(sizeof(*new), GFP_KERNEL);
		if (!new)
			return -ENOMEM;
		new->prog = prog;
	}

2276
	spin_lock_bh(&tun->lock);
2277
	old = rcu_dereference_protected(*prog_p,
2278
					lockdep_is_held(&tun->lock));
2279
	rcu_assign_pointer(*prog_p, new);
2280
	spin_unlock_bh(&tun->lock);
2281 2282

	if (old)
2283
		call_rcu(&old->rcu, tun_prog_free);
2284 2285 2286 2287

	return 0;
}

J
Jason Wang 已提交
2288 2289 2290 2291
static void tun_free_netdev(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

J
Jason Wang 已提交
2292
	BUG_ON(!(list_empty(&tun->disabled)));
E
Eric Dumazet 已提交
2293

2294
	free_percpu(tun->pcpu_stats);
E
Eric Dumazet 已提交
2295 2296 2297 2298 2299
	/* We clear pcpu_stats so that tun_set_iff() can tell if
	 * tun_free_netdev() has been called from register_netdevice().
	 */
	tun->pcpu_stats = NULL;

J
Jason Wang 已提交
2300
	tun_flow_uninit(tun);
2301
	security_tun_dev_free_security(tun->security);
2302
	__tun_set_ebpf(tun, &tun->steering_prog, NULL);
2303
	__tun_set_ebpf(tun, &tun->filter_prog, NULL);
J
Jason Wang 已提交
2304 2305
}

L
Linus Torvalds 已提交
2306 2307 2308 2309
static void tun_setup(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

2310 2311
	tun->owner = INVALID_UID;
	tun->group = INVALID_GID;
2312
	tun_default_link_ksettings(dev, &tun->link_ksettings);
L
Linus Torvalds 已提交
2313 2314

	dev->ethtool_ops = &tun_ethtool_ops;
2315 2316
	dev->needs_free_netdev = true;
	dev->priv_destructor = tun_free_netdev;
J
Jason Wang 已提交
2317 2318
	/* We prefer our own queue length */
	dev->tx_queue_len = TUN_READQ_SIZE;
L
Linus Torvalds 已提交
2319 2320
}

2321 2322 2323
/* Trivial set of netlink ops to allow deleting tun or tap
 * device with netlink.
 */
2324 2325
static int tun_validate(struct nlattr *tb[], struct nlattr *data[],
			struct netlink_ext_ack *extack)
2326
{
2327 2328 2329
	NL_SET_ERR_MSG(extack,
		       "tun/tap creation via rtnetlink is not supported.");
	return -EOPNOTSUPP;
2330 2331
}

2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385
static size_t tun_get_size(const struct net_device *dev)
{
	BUILD_BUG_ON(sizeof(u32) != sizeof(uid_t));
	BUILD_BUG_ON(sizeof(u32) != sizeof(gid_t));

	return nla_total_size(sizeof(uid_t)) + /* OWNER */
	       nla_total_size(sizeof(gid_t)) + /* GROUP */
	       nla_total_size(sizeof(u8)) + /* TYPE */
	       nla_total_size(sizeof(u8)) + /* PI */
	       nla_total_size(sizeof(u8)) + /* VNET_HDR */
	       nla_total_size(sizeof(u8)) + /* PERSIST */
	       nla_total_size(sizeof(u8)) + /* MULTI_QUEUE */
	       nla_total_size(sizeof(u32)) + /* NUM_QUEUES */
	       nla_total_size(sizeof(u32)) + /* NUM_DISABLED_QUEUES */
	       0;
}

static int tun_fill_info(struct sk_buff *skb, const struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

	if (nla_put_u8(skb, IFLA_TUN_TYPE, tun->flags & TUN_TYPE_MASK))
		goto nla_put_failure;
	if (uid_valid(tun->owner) &&
	    nla_put_u32(skb, IFLA_TUN_OWNER,
			from_kuid_munged(current_user_ns(), tun->owner)))
		goto nla_put_failure;
	if (gid_valid(tun->group) &&
	    nla_put_u32(skb, IFLA_TUN_GROUP,
			from_kgid_munged(current_user_ns(), tun->group)))
		goto nla_put_failure;
	if (nla_put_u8(skb, IFLA_TUN_PI, !(tun->flags & IFF_NO_PI)))
		goto nla_put_failure;
	if (nla_put_u8(skb, IFLA_TUN_VNET_HDR, !!(tun->flags & IFF_VNET_HDR)))
		goto nla_put_failure;
	if (nla_put_u8(skb, IFLA_TUN_PERSIST, !!(tun->flags & IFF_PERSIST)))
		goto nla_put_failure;
	if (nla_put_u8(skb, IFLA_TUN_MULTI_QUEUE,
		       !!(tun->flags & IFF_MULTI_QUEUE)))
		goto nla_put_failure;
	if (tun->flags & IFF_MULTI_QUEUE) {
		if (nla_put_u32(skb, IFLA_TUN_NUM_QUEUES, tun->numqueues))
			goto nla_put_failure;
		if (nla_put_u32(skb, IFLA_TUN_NUM_DISABLED_QUEUES,
				tun->numdisabled))
			goto nla_put_failure;
	}

	return 0;

nla_put_failure:
	return -EMSGSIZE;
}

2386 2387 2388 2389 2390
static struct rtnl_link_ops tun_link_ops __read_mostly = {
	.kind		= DRV_NAME,
	.priv_size	= sizeof(struct tun_struct),
	.setup		= tun_setup,
	.validate	= tun_validate,
2391 2392
	.get_size       = tun_get_size,
	.fill_info      = tun_fill_info,
2393 2394
};

2395 2396
static void tun_sock_write_space(struct sock *sk)
{
J
Jason Wang 已提交
2397
	struct tun_file *tfile;
2398
	wait_queue_head_t *wqueue;
2399 2400 2401 2402

	if (!sock_writeable(sk))
		return;

2403
	if (!test_and_clear_bit(SOCKWQ_ASYNC_NOSPACE, &sk->sk_socket->flags))
2404 2405
		return;

2406 2407
	wqueue = sk_sleep(sk);
	if (wqueue && waitqueue_active(wqueue))
2408 2409
		wake_up_interruptible_sync_poll(wqueue, EPOLLOUT |
						EPOLLWRNORM | EPOLLWRBAND);
H
Herbert Xu 已提交
2410

J
Jason Wang 已提交
2411 2412
	tfile = container_of(sk, struct tun_file, sk);
	kill_fasync(&tfile->fasync, SIGIO, POLL_OUT);
2413 2414
}

2415 2416 2417 2418 2419 2420
static void tun_put_page(struct tun_page *tpage)
{
	if (tpage->page)
		__page_frag_cache_drain(tpage->page, tpage->count);
}

2421 2422
static int tun_xdp_one(struct tun_struct *tun,
		       struct tun_file *tfile,
2423 2424
		       struct xdp_buff *xdp, int *flush,
		       struct tun_page *tpage)
2425
{
2426
	unsigned int datasize = xdp->data_end - xdp->data;
2427 2428 2429 2430 2431 2432 2433 2434 2435
	struct tun_xdp_hdr *hdr = xdp->data_hard_start;
	struct virtio_net_hdr *gso = &hdr->gso;
	struct tun_pcpu_stats *stats;
	struct bpf_prog *xdp_prog;
	struct sk_buff *skb = NULL;
	u32 rxhash = 0, act;
	int buflen = hdr->buflen;
	int err = 0;
	bool skb_xdp = false;
2436
	struct page *page;
2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462

	xdp_prog = rcu_dereference(tun->xdp_prog);
	if (xdp_prog) {
		if (gso->gso_type) {
			skb_xdp = true;
			goto build;
		}
		xdp_set_data_meta_invalid(xdp);
		xdp->rxq = &tfile->xdp_rxq;

		act = bpf_prog_run_xdp(xdp_prog, xdp);
		err = tun_xdp_act(tun, xdp_prog, xdp, act);
		if (err < 0) {
			put_page(virt_to_head_page(xdp->data));
			return err;
		}

		switch (err) {
		case XDP_REDIRECT:
			*flush = true;
			/* fall through */
		case XDP_TX:
			return 0;
		case XDP_PASS:
			break;
		default:
2463 2464 2465 2466 2467 2468 2469 2470
			page = virt_to_head_page(xdp->data);
			if (tpage->page == page) {
				++tpage->count;
			} else {
				tun_put_page(tpage);
				tpage->page = page;
				tpage->count = 1;
			}
2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493
			return 0;
		}
	}

build:
	skb = build_skb(xdp->data_hard_start, buflen);
	if (!skb) {
		err = -ENOMEM;
		goto out;
	}

	skb_reserve(skb, xdp->data - xdp->data_hard_start);
	skb_put(skb, xdp->data_end - xdp->data);

	if (virtio_net_hdr_to_skb(skb, gso, tun_is_little_endian(tun))) {
		this_cpu_inc(tun->pcpu_stats->rx_frame_errors);
		kfree_skb(skb);
		err = -EINVAL;
		goto out;
	}

	skb->protocol = eth_type_trans(skb, tun->dev);
	skb_reset_network_header(skb);
2494
	skb_probe_transport_header(skb);
2495 2496 2497 2498 2499 2500 2501

	if (skb_xdp) {
		err = do_xdp_generic(xdp_prog, skb);
		if (err != XDP_PASS)
			goto out;
	}

2502 2503
	if (!rcu_dereference(tun->steering_prog) && tun->numqueues > 1 &&
	    !tfile->detached)
2504 2505
		rxhash = __skb_get_hash_symmetric(skb);

M
Matthew Cover 已提交
2506
	skb_record_rx_queue(skb, tfile->queue_index);
2507 2508
	netif_receive_skb(skb);

2509 2510 2511 2512
	/* No need for get_cpu_ptr() here since this function is
	 * always called with bh disabled
	 */
	stats = this_cpu_ptr(tun->pcpu_stats);
2513 2514
	u64_stats_update_begin(&stats->syncp);
	stats->rx_packets++;
2515
	stats->rx_bytes += datasize;
2516 2517 2518 2519 2520 2521 2522 2523 2524
	u64_stats_update_end(&stats->syncp);

	if (rxhash)
		tun_flow_update(tun, rxhash, tfile);

out:
	return err;
}

2525
static int tun_sendmsg(struct socket *sock, struct msghdr *m, size_t total_len)
2526
{
2527
	int ret, i;
J
Jason Wang 已提交
2528
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2529
	struct tun_struct *tun = tun_get(tfile);
2530
	struct tun_msg_ctl *ctl = m->msg_control;
2531
	struct xdp_buff *xdp;
J
Jason Wang 已提交
2532 2533 2534

	if (!tun)
		return -EBADFD;
2535

2536
	if (ctl && (ctl->type == TUN_MSG_PTR)) {
2537
		struct tun_page tpage;
2538 2539 2540
		int n = ctl->num;
		int flush = 0;

2541 2542
		memset(&tpage, 0, sizeof(tpage));

2543 2544 2545 2546 2547
		local_bh_disable();
		rcu_read_lock();

		for (i = 0; i < n; i++) {
			xdp = &((struct xdp_buff *)ctl->ptr)[i];
2548
			tun_xdp_one(tun, tfile, xdp, &flush, &tpage);
2549 2550 2551 2552 2553 2554 2555 2556
		}

		if (flush)
			xdp_do_flush_map();

		rcu_read_unlock();
		local_bh_enable();

2557 2558
		tun_put_page(&tpage);

2559 2560 2561
		ret = total_len;
		goto out;
	}
2562 2563

	ret = tun_get_user(tun, tfile, ctl ? ctl->ptr : NULL, &m->msg_iter,
J
Jason Wang 已提交
2564 2565
			   m->msg_flags & MSG_DONTWAIT,
			   m->msg_flags & MSG_MORE);
2566
out:
J
Jason Wang 已提交
2567 2568
	tun_put(tun);
	return ret;
2569 2570
}

2571
static int tun_recvmsg(struct socket *sock, struct msghdr *m, size_t total_len,
2572 2573
		       int flags)
{
J
Jason Wang 已提交
2574
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2575
	struct tun_struct *tun = tun_get(tfile);
J
Jason Wang 已提交
2576
	void *ptr = m->msg_control;
2577
	int ret;
J
Jason Wang 已提交
2578

W
Wei Xu 已提交
2579 2580
	if (!tun) {
		ret = -EBADFD;
J
Jason Wang 已提交
2581
		goto out_free;
W
Wei Xu 已提交
2582
	}
J
Jason Wang 已提交
2583

2584
	if (flags & ~(MSG_DONTWAIT|MSG_TRUNC|MSG_ERRQUEUE)) {
2585
		ret = -EINVAL;
W
Wei Xu 已提交
2586
		goto out_put_tun;
2587
	}
2588 2589 2590 2591 2592
	if (flags & MSG_ERRQUEUE) {
		ret = sock_recv_errqueue(sock->sk, m, total_len,
					 SOL_PACKET, TUN_TX_TIMESTAMP);
		goto out;
	}
J
Jason Wang 已提交
2593
	ret = tun_do_read(tun, tfile, &m->msg_iter, flags & MSG_DONTWAIT, ptr);
2594
	if (ret > (ssize_t)total_len) {
2595 2596 2597
		m->msg_flags |= MSG_TRUNC;
		ret = flags & MSG_TRUNC ? ret : total_len;
	}
2598
out:
J
Jason Wang 已提交
2599
	tun_put(tun);
2600
	return ret;
W
Wei Xu 已提交
2601 2602 2603

out_put_tun:
	tun_put(tun);
J
Jason Wang 已提交
2604 2605
out_free:
	tun_ptr_free(ptr);
W
Wei Xu 已提交
2606
	return ret;
2607 2608
}

J
Jason Wang 已提交
2609 2610 2611
static int tun_ptr_peek_len(void *ptr)
{
	if (likely(ptr)) {
2612 2613
		if (tun_is_xdp_frame(ptr)) {
			struct xdp_frame *xdpf = tun_ptr_to_xdp(ptr);
J
Jason Wang 已提交
2614

2615
			return xdpf->len;
J
Jason Wang 已提交
2616 2617 2618 2619 2620 2621 2622
		}
		return __skb_array_len_with_tag(ptr);
	} else {
		return 0;
	}
}

J
Jason Wang 已提交
2623 2624 2625 2626 2627 2628
static int tun_peek_len(struct socket *sock)
{
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
	struct tun_struct *tun;
	int ret = 0;

2629
	tun = tun_get(tfile);
J
Jason Wang 已提交
2630 2631 2632
	if (!tun)
		return 0;

J
Jason Wang 已提交
2633
	ret = PTR_RING_PEEK_CALL(&tfile->tx_ring, tun_ptr_peek_len);
J
Jason Wang 已提交
2634 2635 2636 2637 2638
	tun_put(tun);

	return ret;
}

2639 2640
/* Ops structure to mimic raw sockets with tun */
static const struct proto_ops tun_socket_ops = {
J
Jason Wang 已提交
2641
	.peek_len = tun_peek_len,
2642 2643 2644 2645
	.sendmsg = tun_sendmsg,
	.recvmsg = tun_recvmsg,
};

2646 2647 2648
static struct proto tun_proto = {
	.name		= "tun",
	.owner		= THIS_MODULE,
J
Jason Wang 已提交
2649
	.obj_size	= sizeof(struct tun_file),
2650
};
2651

2652 2653
static int tun_flags(struct tun_struct *tun)
{
2654
	return tun->flags & (TUN_FEATURES | IFF_PERSIST | IFF_TUN | IFF_TAP);
2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667
}

static ssize_t tun_show_flags(struct device *dev, struct device_attribute *attr,
			      char *buf)
{
	struct tun_struct *tun = netdev_priv(to_net_dev(dev));
	return sprintf(buf, "0x%x\n", tun_flags(tun));
}

static ssize_t tun_show_owner(struct device *dev, struct device_attribute *attr,
			      char *buf)
{
	struct tun_struct *tun = netdev_priv(to_net_dev(dev));
2668 2669 2670 2671
	return uid_valid(tun->owner)?
		sprintf(buf, "%u\n",
			from_kuid_munged(current_user_ns(), tun->owner)):
		sprintf(buf, "-1\n");
2672 2673 2674 2675 2676 2677
}

static ssize_t tun_show_group(struct device *dev, struct device_attribute *attr,
			      char *buf)
{
	struct tun_struct *tun = netdev_priv(to_net_dev(dev));
2678 2679 2680 2681
	return gid_valid(tun->group) ?
		sprintf(buf, "%u\n",
			from_kgid_munged(current_user_ns(), tun->group)):
		sprintf(buf, "-1\n");
2682 2683 2684 2685 2686 2687
}

static DEVICE_ATTR(tun_flags, 0444, tun_show_flags, NULL);
static DEVICE_ATTR(owner, 0444, tun_show_owner, NULL);
static DEVICE_ATTR(group, 0444, tun_show_group, NULL);

2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698
static struct attribute *tun_dev_attrs[] = {
	&dev_attr_tun_flags.attr,
	&dev_attr_owner.attr,
	&dev_attr_group.attr,
	NULL
};

static const struct attribute_group tun_attr_group = {
	.attrs = tun_dev_attrs
};

2699
static int tun_set_iff(struct net *net, struct file *file, struct ifreq *ifr)
L
Linus Torvalds 已提交
2700 2701
{
	struct tun_struct *tun;
J
Jason Wang 已提交
2702
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
2703 2704 2705
	struct net_device *dev;
	int err;

2706 2707 2708
	if (tfile->detached)
		return -EINVAL;

2709 2710 2711 2712 2713 2714 2715 2716 2717
	if ((ifr->ifr_flags & IFF_NAPI_FRAGS)) {
		if (!capable(CAP_NET_ADMIN))
			return -EPERM;

		if (!(ifr->ifr_flags & IFF_NAPI) ||
		    (ifr->ifr_flags & TUN_TYPE_MASK) != IFF_TAP)
			return -EINVAL;
	}

2718 2719
	dev = __dev_get_by_name(net, ifr->ifr_name);
	if (dev) {
2720 2721
		if (ifr->ifr_flags & IFF_TUN_EXCL)
			return -EBUSY;
2722 2723 2724 2725 2726 2727 2728
		if ((ifr->ifr_flags & IFF_TUN) && dev->netdev_ops == &tun_netdev_ops)
			tun = netdev_priv(dev);
		else if ((ifr->ifr_flags & IFF_TAP) && dev->netdev_ops == &tap_netdev_ops)
			tun = netdev_priv(dev);
		else
			return -EINVAL;

2729
		if (!!(ifr->ifr_flags & IFF_MULTI_QUEUE) !=
M
Michael S. Tsirkin 已提交
2730
		    !!(tun->flags & IFF_MULTI_QUEUE))
2731 2732
			return -EINVAL;

2733
		if (tun_not_capable(tun))
P
Paul Moore 已提交
2734
			return -EPERM;
2735
		err = security_tun_dev_open(tun->security);
P
Paul Moore 已提交
2736 2737 2738
		if (err < 0)
			return err;

2739
		err = tun_attach(tun, file, ifr->ifr_flags & IFF_NOFILTER,
E
Eric Dumazet 已提交
2740
				 ifr->ifr_flags & IFF_NAPI,
2741
				 ifr->ifr_flags & IFF_NAPI_FRAGS, true);
2742 2743
		if (err < 0)
			return err;
J
Jason Wang 已提交
2744

M
Michael S. Tsirkin 已提交
2745
		if (tun->flags & IFF_MULTI_QUEUE &&
2746 2747 2748 2749
		    (tun->numqueues + tun->numdisabled > 1)) {
			/* One or more queue has already been attached, no need
			 * to initialize the device again.
			 */
2750
			netdev_state_change(dev);
2751 2752
			return 0;
		}
2753 2754 2755

		tun->flags = (tun->flags & ~TUN_FEATURES) |
			      (ifr->ifr_flags & TUN_FEATURES);
2756 2757 2758

		netdev_state_change(dev);
	} else {
L
Linus Torvalds 已提交
2759 2760
		char *name;
		unsigned long flags = 0;
2761 2762
		int queues = ifr->ifr_flags & IFF_MULTI_QUEUE ?
			     MAX_TAP_QUEUES : 1;
L
Linus Torvalds 已提交
2763

2764
		if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
2765
			return -EPERM;
P
Paul Moore 已提交
2766 2767 2768
		err = security_tun_dev_create();
		if (err < 0)
			return err;
2769

L
Linus Torvalds 已提交
2770 2771 2772
		/* Set dev type */
		if (ifr->ifr_flags & IFF_TUN) {
			/* TUN device */
M
Michael S. Tsirkin 已提交
2773
			flags |= IFF_TUN;
L
Linus Torvalds 已提交
2774 2775 2776
			name = "tun%d";
		} else if (ifr->ifr_flags & IFF_TAP) {
			/* TAP device */
M
Michael S. Tsirkin 已提交
2777
			flags |= IFF_TAP;
L
Linus Torvalds 已提交
2778
			name = "tap%d";
2779
		} else
2780
			return -EINVAL;
2781

L
Linus Torvalds 已提交
2782 2783 2784
		if (*ifr->ifr_name)
			name = ifr->ifr_name;

J
Jason Wang 已提交
2785
		dev = alloc_netdev_mqs(sizeof(struct tun_struct), name,
2786 2787
				       NET_NAME_UNKNOWN, tun_setup, queues,
				       queues);
2788

L
Linus Torvalds 已提交
2789 2790 2791
		if (!dev)
			return -ENOMEM;

2792
		dev_net_set(dev, net);
2793
		dev->rtnl_link_ops = &tun_link_ops;
2794
		dev->ifindex = tfile->ifindex;
2795
		dev->sysfs_groups[0] = &tun_attr_group;
S
Stephen Hemminger 已提交
2796

L
Linus Torvalds 已提交
2797 2798 2799
		tun = netdev_priv(dev);
		tun->dev = dev;
		tun->flags = flags;
2800
		tun->txflt.count = 0;
2801
		tun->vnet_hdr_sz = sizeof(struct virtio_net_hdr);
2802

2803
		tun->align = NET_SKB_PAD;
J
Jason Wang 已提交
2804 2805
		tun->filter_attached = false;
		tun->sndbuf = tfile->socket.sk->sk_sndbuf;
J
Jason Wang 已提交
2806
		tun->rx_batched = 0;
2807
		RCU_INIT_POINTER(tun->steering_prog, NULL);
2808

2809 2810 2811 2812 2813 2814
		tun->pcpu_stats = netdev_alloc_pcpu_stats(struct tun_pcpu_stats);
		if (!tun->pcpu_stats) {
			err = -ENOMEM;
			goto err_free_dev;
		}

J
Jason Wang 已提交
2815 2816
		spin_lock_init(&tun->lock);

2817 2818
		err = security_tun_dev_alloc_security(&tun->security);
		if (err < 0)
2819
			goto err_free_stat;
P
Paul Moore 已提交
2820

L
Linus Torvalds 已提交
2821
		tun_net_init(dev);
2822
		tun_flow_init(tun);
J
Jason Wang 已提交
2823

2824
		dev->hw_features = NETIF_F_SG | NETIF_F_FRAGLIST |
J
Jason Wang 已提交
2825 2826
				   TUN_USER_FEATURES | NETIF_F_HW_VLAN_CTAG_TX |
				   NETIF_F_HW_VLAN_STAG_TX;
2827
		dev->features = dev->hw_features | NETIF_F_LLTX;
2828 2829 2830
		dev->vlan_features = dev->features &
				     ~(NETIF_F_HW_VLAN_CTAG_TX |
				       NETIF_F_HW_VLAN_STAG_TX);
2831

2832 2833 2834
		tun->flags = (tun->flags & ~TUN_FEATURES) |
			      (ifr->ifr_flags & TUN_FEATURES);

J
Jason Wang 已提交
2835
		INIT_LIST_HEAD(&tun->disabled);
E
Eric Dumazet 已提交
2836
		err = tun_attach(tun, file, false, ifr->ifr_flags & IFF_NAPI,
2837
				 ifr->ifr_flags & IFF_NAPI_FRAGS, false);
2838
		if (err < 0)
2839
			goto err_free_flow;
2840

L
Linus Torvalds 已提交
2841 2842
		err = register_netdevice(tun->dev);
		if (err < 0)
2843
			goto err_detach;
2844 2845 2846 2847
		/* free_netdev() won't check refcnt, to aovid race
		 * with dev_put() we need publish tun after registration.
		 */
		rcu_assign_pointer(tfile->tun, tun);
L
Linus Torvalds 已提交
2848 2849
	}

2850 2851
	netif_carrier_on(tun->dev);

2852
	tun_debug(KERN_INFO, tun, "tun_set_iff\n");
L
Linus Torvalds 已提交
2853

2854 2855 2856 2857
	/* Make sure persistent devices do not get stuck in
	 * xoff state.
	 */
	if (netif_running(tun->dev))
J
Jason Wang 已提交
2858
		netif_tx_wake_all_queues(tun->dev);
2859

L
Linus Torvalds 已提交
2860 2861 2862
	strcpy(ifr->ifr_name, tun->dev->name);
	return 0;

2863 2864
err_detach:
	tun_detach_all(dev);
E
Eric Dumazet 已提交
2865 2866 2867 2868 2869 2870
	/* We are here because register_netdevice() has failed.
	 * If register_netdevice() already called tun_free_netdev()
	 * while dealing with the error, tun->pcpu_stats has been cleared.
	 */
	if (!tun->pcpu_stats)
		goto err_free_dev;
2871

2872 2873 2874
err_free_flow:
	tun_flow_uninit(tun);
	security_tun_dev_free_security(tun->security);
2875 2876
err_free_stat:
	free_percpu(tun->pcpu_stats);
2877
err_free_dev:
L
Linus Torvalds 已提交
2878 2879 2880 2881
	free_netdev(dev);
	return err;
}

2882
static void tun_get_iff(struct tun_struct *tun, struct ifreq *ifr)
2883
{
2884
	tun_debug(KERN_INFO, tun, "tun_get_iff\n");
2885 2886 2887

	strcpy(ifr->ifr_name, tun->dev->name);

2888
	ifr->ifr_flags = tun_flags(tun);
2889 2890 2891

}

2892 2893
/* This is like a cut-down ethtool ops, except done via tun fd so no
 * privs required. */
2894
static int set_offload(struct tun_struct *tun, unsigned long arg)
2895
{
2896
	netdev_features_t features = 0;
2897 2898

	if (arg & TUN_F_CSUM) {
2899
		features |= NETIF_F_HW_CSUM;
2900 2901 2902 2903 2904 2905 2906 2907 2908 2909 2910 2911 2912
		arg &= ~TUN_F_CSUM;

		if (arg & (TUN_F_TSO4|TUN_F_TSO6)) {
			if (arg & TUN_F_TSO_ECN) {
				features |= NETIF_F_TSO_ECN;
				arg &= ~TUN_F_TSO_ECN;
			}
			if (arg & TUN_F_TSO4)
				features |= NETIF_F_TSO;
			if (arg & TUN_F_TSO6)
				features |= NETIF_F_TSO6;
			arg &= ~(TUN_F_TSO4|TUN_F_TSO6);
		}
2913 2914

		arg &= ~TUN_F_UFO;
2915 2916 2917 2918 2919 2920 2921
	}

	/* This gives the user a way to test for new features in future by
	 * trying to set them. */
	if (arg)
		return -EINVAL;

2922
	tun->set_features = features;
2923 2924
	tun->dev->wanted_features &= ~TUN_USER_FEATURES;
	tun->dev->wanted_features |= features;
2925
	netdev_update_features(tun->dev);
2926 2927 2928 2929

	return 0;
}

J
Jason Wang 已提交
2930 2931 2932 2933 2934 2935
static void tun_detach_filter(struct tun_struct *tun, int n)
{
	int i;
	struct tun_file *tfile;

	for (i = 0; i < n; i++) {
2936
		tfile = rtnl_dereference(tun->tfiles[i]);
2937 2938 2939
		lock_sock(tfile->socket.sk);
		sk_detach_filter(tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2940 2941 2942 2943 2944 2945 2946 2947 2948 2949 2950
	}

	tun->filter_attached = false;
}

static int tun_attach_filter(struct tun_struct *tun)
{
	int i, ret = 0;
	struct tun_file *tfile;

	for (i = 0; i < tun->numqueues; i++) {
2951
		tfile = rtnl_dereference(tun->tfiles[i]);
2952 2953 2954
		lock_sock(tfile->socket.sk);
		ret = sk_attach_filter(&tun->fprog, tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2955 2956 2957 2958 2959 2960 2961 2962 2963 2964 2965 2966 2967 2968 2969 2970
		if (ret) {
			tun_detach_filter(tun, i);
			return ret;
		}
	}

	tun->filter_attached = true;
	return ret;
}

static void tun_set_sndbuf(struct tun_struct *tun)
{
	struct tun_file *tfile;
	int i;

	for (i = 0; i < tun->numqueues; i++) {
2971
		tfile = rtnl_dereference(tun->tfiles[i]);
J
Jason Wang 已提交
2972 2973 2974 2975
		tfile->socket.sk->sk_sndbuf = tun->sndbuf;
	}
}

2976 2977 2978 2979 2980 2981 2982 2983 2984
static int tun_set_queue(struct file *file, struct ifreq *ifr)
{
	struct tun_file *tfile = file->private_data;
	struct tun_struct *tun;
	int ret = 0;

	rtnl_lock();

	if (ifr->ifr_flags & IFF_ATTACH_QUEUE) {
J
Jason Wang 已提交
2985
		tun = tfile->detached;
2986
		if (!tun) {
2987
			ret = -EINVAL;
2988 2989 2990 2991 2992
			goto unlock;
		}
		ret = security_tun_dev_attach_queue(tun->security);
		if (ret < 0)
			goto unlock;
E
Eric Dumazet 已提交
2993
		ret = tun_attach(tun, file, false, tun->flags & IFF_NAPI,
2994
				 tun->flags & IFF_NAPI_FRAGS, true);
J
Jason Wang 已提交
2995
	} else if (ifr->ifr_flags & IFF_DETACH_QUEUE) {
2996
		tun = rtnl_dereference(tfile->tun);
M
Michael S. Tsirkin 已提交
2997
		if (!tun || !(tun->flags & IFF_MULTI_QUEUE) || tfile->detached)
J
Jason Wang 已提交
2998 2999 3000 3001
			ret = -EINVAL;
		else
			__tun_detach(tfile, false);
	} else
3002 3003
		ret = -EINVAL;

3004 3005 3006
	if (ret >= 0)
		netdev_state_change(tun->dev);

3007
unlock:
3008 3009 3010 3011
	rtnl_unlock();
	return ret;
}

3012 3013
static int tun_set_ebpf(struct tun_struct *tun, struct tun_prog **prog_p,
			void __user *data)
3014 3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028
{
	struct bpf_prog *prog;
	int fd;

	if (copy_from_user(&fd, data, sizeof(fd)))
		return -EFAULT;

	if (fd == -1) {
		prog = NULL;
	} else {
		prog = bpf_prog_get_type(fd, BPF_PROG_TYPE_SOCKET_FILTER);
		if (IS_ERR(prog))
			return PTR_ERR(prog);
	}

3029
	return __tun_set_ebpf(tun, prog_p, prog);
3030 3031
}

3032 3033
static long __tun_chr_ioctl(struct file *file, unsigned int cmd,
			    unsigned long arg, int ifreq_len)
L
Linus Torvalds 已提交
3034
{
E
Eric W. Biederman 已提交
3035
	struct tun_file *tfile = file->private_data;
3036
	struct net *net = sock_net(&tfile->sk);
E
Eric W. Biederman 已提交
3037
	struct tun_struct *tun;
L
Linus Torvalds 已提交
3038
	void __user* argp = (void __user*)arg;
N
Nicolas Dichtel 已提交
3039
	unsigned int ifindex, carrier;
L
Linus Torvalds 已提交
3040
	struct ifreq ifr;
3041 3042
	kuid_t owner;
	kgid_t group;
3043
	int sndbuf;
3044
	int vnet_hdr_sz;
M
Michael S. Tsirkin 已提交
3045
	int le;
3046
	int ret;
3047
	bool do_notify = false;
L
Linus Torvalds 已提交
3048

3049 3050
	if (cmd == TUNSETIFF || cmd == TUNSETQUEUE ||
	    (_IOC_TYPE(cmd) == SOCK_IOC_TYPE && cmd != SIOCGSKNS)) {
3051
		if (copy_from_user(&ifr, argp, ifreq_len))
L
Linus Torvalds 已提交
3052
			return -EFAULT;
D
David S. Miller 已提交
3053
	} else {
3054
		memset(&ifr, 0, sizeof(ifr));
D
David S. Miller 已提交
3055
	}
E
Eric W. Biederman 已提交
3056 3057 3058
	if (cmd == TUNGETFEATURES) {
		/* Currently this just means: "what IFF flags are valid?".
		 * This is needed because we never checked for invalid flags on
3059 3060 3061
		 * TUNSETIFF.
		 */
		return put_user(IFF_TUN | IFF_TAP | TUN_FEATURES,
E
Eric W. Biederman 已提交
3062
				(unsigned int __user*)argp);
3063
	} else if (cmd == TUNSETQUEUE) {
3064
		return tun_set_queue(file, &ifr);
3065 3066 3067 3068 3069
	} else if (cmd == SIOCGSKNS) {
		if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
			return -EPERM;
		return open_related_ns(&net->ns, get_net_ns);
	}
E
Eric W. Biederman 已提交
3070

J
Jason Wang 已提交
3071
	ret = 0;
3072 3073
	rtnl_lock();

3074
	tun = tun_get(tfile);
3075 3076 3077 3078 3079
	if (cmd == TUNSETIFF) {
		ret = -EEXIST;
		if (tun)
			goto unlock;

L
Linus Torvalds 已提交
3080 3081
		ifr.ifr_name[IFNAMSIZ-1] = '\0';

3082
		ret = tun_set_iff(net, file, &ifr);
L
Linus Torvalds 已提交
3083

3084 3085
		if (ret)
			goto unlock;
L
Linus Torvalds 已提交
3086

3087
		if (copy_to_user(argp, &ifr, ifreq_len))
3088 3089
			ret = -EFAULT;
		goto unlock;
L
Linus Torvalds 已提交
3090
	}
3091 3092 3093 3094 3095 3096 3097 3098 3099 3100 3101 3102 3103
	if (cmd == TUNSETIFINDEX) {
		ret = -EPERM;
		if (tun)
			goto unlock;

		ret = -EFAULT;
		if (copy_from_user(&ifindex, argp, sizeof(ifindex)))
			goto unlock;

		ret = 0;
		tfile->ifindex = ifindex;
		goto unlock;
	}
L
Linus Torvalds 已提交
3104

3105
	ret = -EBADFD;
L
Linus Torvalds 已提交
3106
	if (!tun)
3107
		goto unlock;
L
Linus Torvalds 已提交
3108

3109
	tun_debug(KERN_INFO, tun, "tun_chr_ioctl cmd %u\n", cmd);
L
Linus Torvalds 已提交
3110

3111
	net = dev_net(tun->dev);
E
Eric W. Biederman 已提交
3112
	ret = 0;
L
Linus Torvalds 已提交
3113
	switch (cmd) {
3114
	case TUNGETIFF:
3115
		tun_get_iff(tun, &ifr);
3116

3117 3118
		if (tfile->detached)
			ifr.ifr_flags |= IFF_DETACH_QUEUE;
3119 3120
		if (!tfile->socket.sk->sk_filter)
			ifr.ifr_flags |= IFF_NOFILTER;
3121

3122
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
3123
			ret = -EFAULT;
3124 3125
		break;

L
Linus Torvalds 已提交
3126 3127 3128
	case TUNSETNOCSUM:
		/* Disable/Enable checksum */

3129 3130
		/* [unimplemented] */
		tun_debug(KERN_INFO, tun, "ignored: set checksum %s\n",
3131
			  arg ? "disabled" : "enabled");
L
Linus Torvalds 已提交
3132 3133 3134
		break;

	case TUNSETPERSIST:
J
Jason Wang 已提交
3135 3136 3137
		/* Disable/Enable persist mode. Keep an extra reference to the
		 * module to prevent the module being unprobed.
		 */
M
Michael S. Tsirkin 已提交
3138 3139
		if (arg && !(tun->flags & IFF_PERSIST)) {
			tun->flags |= IFF_PERSIST;
J
Jason Wang 已提交
3140
			__module_get(THIS_MODULE);
3141
			do_notify = true;
J
Jason Wang 已提交
3142
		}
M
Michael S. Tsirkin 已提交
3143 3144
		if (!arg && (tun->flags & IFF_PERSIST)) {
			tun->flags &= ~IFF_PERSIST;
J
Jason Wang 已提交
3145
			module_put(THIS_MODULE);
3146
			do_notify = true;
J
Jason Wang 已提交
3147
		}
L
Linus Torvalds 已提交
3148

3149 3150
		tun_debug(KERN_INFO, tun, "persist %s\n",
			  arg ? "enabled" : "disabled");
L
Linus Torvalds 已提交
3151 3152 3153 3154
		break;

	case TUNSETOWNER:
		/* Set owner of the device */
3155 3156 3157 3158 3159 3160
		owner = make_kuid(current_user_ns(), arg);
		if (!uid_valid(owner)) {
			ret = -EINVAL;
			break;
		}
		tun->owner = owner;
3161
		do_notify = true;
3162
		tun_debug(KERN_INFO, tun, "owner set to %u\n",
3163
			  from_kuid(&init_user_ns, tun->owner));
L
Linus Torvalds 已提交
3164 3165
		break;

3166 3167
	case TUNSETGROUP:
		/* Set group of the device */
3168 3169 3170 3171 3172 3173
		group = make_kgid(current_user_ns(), arg);
		if (!gid_valid(group)) {
			ret = -EINVAL;
			break;
		}
		tun->group = group;
3174
		do_notify = true;
3175
		tun_debug(KERN_INFO, tun, "group set to %u\n",
3176
			  from_kgid(&init_user_ns, tun->group));
3177 3178
		break;

3179 3180 3181
	case TUNSETLINK:
		/* Only allow setting the type when the interface is down */
		if (tun->dev->flags & IFF_UP) {
3182 3183
			tun_debug(KERN_INFO, tun,
				  "Linktype set failed because interface is up\n");
3184
			ret = -EBUSY;
3185 3186
		} else {
			tun->dev->type = (int) arg;
3187 3188
			tun_debug(KERN_INFO, tun, "linktype set to %d\n",
				  tun->dev->type);
3189
			ret = 0;
3190
		}
E
Eric W. Biederman 已提交
3191
		break;
3192

L
Linus Torvalds 已提交
3193 3194 3195 3196 3197
#ifdef TUN_DEBUG
	case TUNSETDEBUG:
		tun->debug = arg;
		break;
#endif
3198
	case TUNSETOFFLOAD:
3199
		ret = set_offload(tun, arg);
E
Eric W. Biederman 已提交
3200
		break;
3201

3202 3203
	case TUNSETTXFILTER:
		/* Can be set only for TAPs */
E
Eric W. Biederman 已提交
3204
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3205
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
E
Eric W. Biederman 已提交
3206
			break;
H
Harvey Harrison 已提交
3207
		ret = update_filter(&tun->txflt, (void __user *)arg);
E
Eric W. Biederman 已提交
3208
		break;
L
Linus Torvalds 已提交
3209 3210

	case SIOCGIFHWADDR:
3211
		/* Get hw address */
3212 3213
		memcpy(ifr.ifr_hwaddr.sa_data, tun->dev->dev_addr, ETH_ALEN);
		ifr.ifr_hwaddr.sa_family = tun->dev->type;
3214
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
3215 3216
			ret = -EFAULT;
		break;
L
Linus Torvalds 已提交
3217 3218

	case SIOCSIFHWADDR:
3219
		/* Set hw address */
3220 3221
		tun_debug(KERN_DEBUG, tun, "set hw address: %pM\n",
			  ifr.ifr_hwaddr.sa_data);
3222

3223
		ret = dev_set_mac_address(tun->dev, &ifr.ifr_hwaddr, NULL);
E
Eric W. Biederman 已提交
3224
		break;
3225 3226

	case TUNGETSNDBUF:
J
Jason Wang 已提交
3227
		sndbuf = tfile->socket.sk->sk_sndbuf;
3228 3229 3230 3231 3232 3233 3234 3235 3236
		if (copy_to_user(argp, &sndbuf, sizeof(sndbuf)))
			ret = -EFAULT;
		break;

	case TUNSETSNDBUF:
		if (copy_from_user(&sndbuf, argp, sizeof(sndbuf))) {
			ret = -EFAULT;
			break;
		}
3237 3238 3239 3240
		if (sndbuf <= 0) {
			ret = -EINVAL;
			break;
		}
3241

J
Jason Wang 已提交
3242 3243
		tun->sndbuf = sndbuf;
		tun_set_sndbuf(tun);
3244 3245
		break;

3246 3247 3248 3249 3250 3251 3252 3253 3254 3255 3256 3257 3258 3259 3260 3261 3262 3263 3264
	case TUNGETVNETHDRSZ:
		vnet_hdr_sz = tun->vnet_hdr_sz;
		if (copy_to_user(argp, &vnet_hdr_sz, sizeof(vnet_hdr_sz)))
			ret = -EFAULT;
		break;

	case TUNSETVNETHDRSZ:
		if (copy_from_user(&vnet_hdr_sz, argp, sizeof(vnet_hdr_sz))) {
			ret = -EFAULT;
			break;
		}
		if (vnet_hdr_sz < (int)sizeof(struct virtio_net_hdr)) {
			ret = -EINVAL;
			break;
		}

		tun->vnet_hdr_sz = vnet_hdr_sz;
		break;

M
Michael S. Tsirkin 已提交
3265 3266 3267 3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280 3281
	case TUNGETVNETLE:
		le = !!(tun->flags & TUN_VNET_LE);
		if (put_user(le, (int __user *)argp))
			ret = -EFAULT;
		break;

	case TUNSETVNETLE:
		if (get_user(le, (int __user *)argp)) {
			ret = -EFAULT;
			break;
		}
		if (le)
			tun->flags |= TUN_VNET_LE;
		else
			tun->flags &= ~TUN_VNET_LE;
		break;

3282 3283 3284 3285 3286 3287 3288 3289
	case TUNGETVNETBE:
		ret = tun_get_vnet_be(tun, argp);
		break;

	case TUNSETVNETBE:
		ret = tun_set_vnet_be(tun, argp);
		break;

M
Michael S. Tsirkin 已提交
3290 3291 3292
	case TUNATTACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3293
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
3294 3295
			break;
		ret = -EFAULT;
J
Jason Wang 已提交
3296
		if (copy_from_user(&tun->fprog, argp, sizeof(tun->fprog)))
M
Michael S. Tsirkin 已提交
3297 3298
			break;

J
Jason Wang 已提交
3299
		ret = tun_attach_filter(tun);
M
Michael S. Tsirkin 已提交
3300 3301 3302 3303 3304
		break;

	case TUNDETACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3305
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
3306
			break;
J
Jason Wang 已提交
3307 3308
		ret = 0;
		tun_detach_filter(tun, tun->numqueues);
M
Michael S. Tsirkin 已提交
3309 3310
		break;

P
Pavel Emelyanov 已提交
3311 3312
	case TUNGETFILTER:
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3313
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
P
Pavel Emelyanov 已提交
3314 3315 3316 3317 3318 3319 3320
			break;
		ret = -EFAULT;
		if (copy_to_user(argp, &tun->fprog, sizeof(tun->fprog)))
			break;
		ret = 0;
		break;

3321
	case TUNSETSTEERINGEBPF:
3322
		ret = tun_set_ebpf(tun, &tun->steering_prog, argp);
3323 3324
		break;

3325 3326 3327 3328
	case TUNSETFILTEREBPF:
		ret = tun_set_ebpf(tun, &tun->filter_prog, argp);
		break;

N
Nicolas Dichtel 已提交
3329 3330 3331 3332 3333 3334 3335 3336
	case TUNSETCARRIER:
		ret = -EFAULT;
		if (copy_from_user(&carrier, argp, sizeof(carrier)))
			goto unlock;

		ret = tun_net_change_carrier(tun->dev, (bool)carrier);
		break;

3337 3338 3339 3340 3341 3342 3343
	case TUNGETDEVNETNS:
		ret = -EPERM;
		if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
			goto unlock;
		ret = open_related_ns(&net->ns, get_net_ns);
		break;

L
Linus Torvalds 已提交
3344
	default:
E
Eric W. Biederman 已提交
3345 3346
		ret = -EINVAL;
		break;
3347
	}
L
Linus Torvalds 已提交
3348

3349 3350 3351
	if (do_notify)
		netdev_state_change(tun->dev);

3352 3353 3354 3355
unlock:
	rtnl_unlock();
	if (tun)
		tun_put(tun);
E
Eric W. Biederman 已提交
3356
	return ret;
L
Linus Torvalds 已提交
3357 3358
}

3359 3360 3361 3362 3363 3364 3365 3366 3367 3368 3369 3370 3371 3372 3373 3374 3375 3376 3377 3378 3379 3380 3381 3382 3383 3384 3385 3386 3387 3388 3389 3390 3391 3392 3393
static long tun_chr_ioctl(struct file *file,
			  unsigned int cmd, unsigned long arg)
{
	return __tun_chr_ioctl(file, cmd, arg, sizeof (struct ifreq));
}

#ifdef CONFIG_COMPAT
static long tun_chr_compat_ioctl(struct file *file,
			 unsigned int cmd, unsigned long arg)
{
	switch (cmd) {
	case TUNSETIFF:
	case TUNGETIFF:
	case TUNSETTXFILTER:
	case TUNGETSNDBUF:
	case TUNSETSNDBUF:
	case SIOCGIFHWADDR:
	case SIOCSIFHWADDR:
		arg = (unsigned long)compat_ptr(arg);
		break;
	default:
		arg = (compat_ulong_t)arg;
		break;
	}

	/*
	 * compat_ifreq is shorter than ifreq, so we must not access beyond
	 * the end of that structure. All fields that are used in this
	 * driver are compatible though, we don't need to convert the
	 * contents.
	 */
	return __tun_chr_ioctl(file, cmd, arg, sizeof(struct compat_ifreq));
}
#endif /* CONFIG_COMPAT */

L
Linus Torvalds 已提交
3394 3395
static int tun_chr_fasync(int fd, struct file *file, int on)
{
J
Jason Wang 已提交
3396
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
3397 3398
	int ret;

J
Jason Wang 已提交
3399
	if ((ret = fasync_helper(fd, file, on, &tfile->fasync)) < 0)
J
Jonathan Corbet 已提交
3400
		goto out;
3401

L
Linus Torvalds 已提交
3402
	if (on) {
3403
		__f_setown(file, task_pid(current), PIDTYPE_TGID, 0);
J
Jason Wang 已提交
3404
		tfile->flags |= TUN_FASYNC;
3405
	} else
J
Jason Wang 已提交
3406
		tfile->flags &= ~TUN_FASYNC;
J
Jonathan Corbet 已提交
3407 3408 3409
	ret = 0;
out:
	return ret;
L
Linus Torvalds 已提交
3410 3411 3412 3413
}

static int tun_chr_open(struct inode *inode, struct file * file)
{
3414
	struct net *net = current->nsproxy->net_ns;
E
Eric W. Biederman 已提交
3415
	struct tun_file *tfile;
T
Thomas Gleixner 已提交
3416

3417
	DBG1(KERN_INFO, "tunX: tun_chr_open\n");
E
Eric W. Biederman 已提交
3418

3419
	tfile = (struct tun_file *)sk_alloc(net, AF_UNSPEC, GFP_KERNEL,
3420
					    &tun_proto, 0);
E
Eric W. Biederman 已提交
3421 3422
	if (!tfile)
		return -ENOMEM;
3423 3424 3425 3426 3427
	if (ptr_ring_init(&tfile->tx_ring, 0, GFP_KERNEL)) {
		sk_free(&tfile->sk);
		return -ENOMEM;
	}

3428
	mutex_init(&tfile->napi_mutex);
3429
	RCU_INIT_POINTER(tfile->tun, NULL);
J
Jason Wang 已提交
3430
	tfile->flags = 0;
3431
	tfile->ifindex = 0;
J
Jason Wang 已提交
3432

3433
	init_waitqueue_head(&tfile->socket.wq.wait);
J
Jason Wang 已提交
3434 3435 3436 3437 3438 3439 3440 3441 3442

	tfile->socket.file = file;
	tfile->socket.ops = &tun_socket_ops;

	sock_init_data(&tfile->socket, &tfile->sk);

	tfile->sk.sk_write_space = tun_sock_write_space;
	tfile->sk.sk_sndbuf = INT_MAX;

E
Eric W. Biederman 已提交
3443
	file->private_data = tfile;
J
Jason Wang 已提交
3444
	INIT_LIST_HEAD(&tfile->next);
J
Jason Wang 已提交
3445

3446 3447
	sock_set_flag(&tfile->sk, SOCK_ZEROCOPY);

L
Linus Torvalds 已提交
3448 3449 3450 3451 3452
	return 0;
}

static int tun_chr_close(struct inode *inode, struct file *file)
{
E
Eric W. Biederman 已提交
3453
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
3454

J
Jason Wang 已提交
3455
	tun_detach(tfile, true);
L
Linus Torvalds 已提交
3456 3457 3458 3459

	return 0;
}

3460
#ifdef CONFIG_PROC_FS
3461
static void tun_chr_show_fdinfo(struct seq_file *m, struct file *file)
3462
{
3463
	struct tun_file *tfile = file->private_data;
3464 3465 3466 3467 3468 3469
	struct tun_struct *tun;
	struct ifreq ifr;

	memset(&ifr, 0, sizeof(ifr));

	rtnl_lock();
3470
	tun = tun_get(tfile);
3471
	if (tun)
3472
		tun_get_iff(tun, &ifr);
3473 3474 3475 3476 3477
	rtnl_unlock();

	if (tun)
		tun_put(tun);

3478
	seq_printf(m, "iff:\t%s\n", ifr.ifr_name);
3479 3480 3481
}
#endif

3482
static const struct file_operations tun_fops = {
3483
	.owner	= THIS_MODULE,
L
Linus Torvalds 已提交
3484
	.llseek = no_llseek,
3485
	.read_iter  = tun_chr_read_iter,
3486
	.write_iter = tun_chr_write_iter,
L
Linus Torvalds 已提交
3487
	.poll	= tun_chr_poll,
3488 3489 3490 3491
	.unlocked_ioctl	= tun_chr_ioctl,
#ifdef CONFIG_COMPAT
	.compat_ioctl = tun_chr_compat_ioctl,
#endif
L
Linus Torvalds 已提交
3492 3493
	.open	= tun_chr_open,
	.release = tun_chr_close,
3494 3495 3496 3497
	.fasync = tun_chr_fasync,
#ifdef CONFIG_PROC_FS
	.show_fdinfo = tun_chr_show_fdinfo,
#endif
L
Linus Torvalds 已提交
3498 3499 3500 3501 3502
};

static struct miscdevice tun_miscdev = {
	.minor = TUN_MINOR,
	.name = "tun",
3503
	.nodename = "net/tun",
L
Linus Torvalds 已提交
3504 3505 3506 3507 3508
	.fops = &tun_fops,
};

/* ethtool interface */

3509 3510
static void tun_default_link_ksettings(struct net_device *dev,
				       struct ethtool_link_ksettings *cmd)
3511 3512 3513 3514 3515 3516 3517 3518
{
	ethtool_link_ksettings_zero_link_mode(cmd, supported);
	ethtool_link_ksettings_zero_link_mode(cmd, advertising);
	cmd->base.speed		= SPEED_10;
	cmd->base.duplex	= DUPLEX_FULL;
	cmd->base.port		= PORT_TP;
	cmd->base.phy_address	= 0;
	cmd->base.autoneg	= AUTONEG_DISABLE;
3519 3520 3521 3522 3523 3524 3525 3526 3527 3528 3529 3530 3531 3532 3533 3534 3535
}

static int tun_get_link_ksettings(struct net_device *dev,
				  struct ethtool_link_ksettings *cmd)
{
	struct tun_struct *tun = netdev_priv(dev);

	memcpy(cmd, &tun->link_ksettings, sizeof(*cmd));
	return 0;
}

static int tun_set_link_ksettings(struct net_device *dev,
				  const struct ethtool_link_ksettings *cmd)
{
	struct tun_struct *tun = netdev_priv(dev);

	memcpy(&tun->link_ksettings, cmd, sizeof(*cmd));
L
Linus Torvalds 已提交
3536 3537 3538 3539 3540 3541 3542
	return 0;
}

static void tun_get_drvinfo(struct net_device *dev, struct ethtool_drvinfo *info)
{
	struct tun_struct *tun = netdev_priv(dev);

3543 3544
	strlcpy(info->driver, DRV_NAME, sizeof(info->driver));
	strlcpy(info->version, DRV_VERSION, sizeof(info->version));
L
Linus Torvalds 已提交
3545 3546

	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
3547
	case IFF_TUN:
3548
		strlcpy(info->bus_info, "tun", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3549
		break;
M
Michael S. Tsirkin 已提交
3550
	case IFF_TAP:
3551
		strlcpy(info->bus_info, "tap", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3552 3553 3554 3555 3556 3557 3558 3559 3560 3561 3562 3563 3564 3565 3566 3567 3568 3569 3570 3571 3572 3573
		break;
	}
}

static u32 tun_get_msglevel(struct net_device *dev)
{
#ifdef TUN_DEBUG
	struct tun_struct *tun = netdev_priv(dev);
	return tun->debug;
#else
	return -EOPNOTSUPP;
#endif
}

static void tun_set_msglevel(struct net_device *dev, u32 value)
{
#ifdef TUN_DEBUG
	struct tun_struct *tun = netdev_priv(dev);
	tun->debug = value;
#endif
}

J
Jason Wang 已提交
3574 3575 3576 3577 3578 3579 3580 3581 3582 3583 3584 3585 3586 3587 3588 3589 3590 3591 3592 3593 3594 3595 3596
static int tun_get_coalesce(struct net_device *dev,
			    struct ethtool_coalesce *ec)
{
	struct tun_struct *tun = netdev_priv(dev);

	ec->rx_max_coalesced_frames = tun->rx_batched;

	return 0;
}

static int tun_set_coalesce(struct net_device *dev,
			    struct ethtool_coalesce *ec)
{
	struct tun_struct *tun = netdev_priv(dev);

	if (ec->rx_max_coalesced_frames > NAPI_POLL_WEIGHT)
		tun->rx_batched = NAPI_POLL_WEIGHT;
	else
		tun->rx_batched = ec->rx_max_coalesced_frames;

	return 0;
}

3597
static const struct ethtool_ops tun_ethtool_ops = {
L
Linus Torvalds 已提交
3598 3599 3600
	.get_drvinfo	= tun_get_drvinfo,
	.get_msglevel	= tun_get_msglevel,
	.set_msglevel	= tun_set_msglevel,
3601
	.get_link	= ethtool_op_get_link,
3602
	.get_ts_info	= ethtool_op_get_ts_info,
J
Jason Wang 已提交
3603 3604
	.get_coalesce   = tun_get_coalesce,
	.set_coalesce   = tun_set_coalesce,
3605
	.get_link_ksettings = tun_get_link_ksettings,
3606
	.set_link_ksettings = tun_set_link_ksettings,
L
Linus Torvalds 已提交
3607 3608
};

J
Jason Wang 已提交
3609 3610 3611 3612
static int tun_queue_resize(struct tun_struct *tun)
{
	struct net_device *dev = tun->dev;
	struct tun_file *tfile;
3613
	struct ptr_ring **rings;
J
Jason Wang 已提交
3614 3615 3616
	int n = tun->numqueues + tun->numdisabled;
	int ret, i;

3617 3618
	rings = kmalloc_array(n, sizeof(*rings), GFP_KERNEL);
	if (!rings)
J
Jason Wang 已提交
3619 3620 3621 3622
		return -ENOMEM;

	for (i = 0; i < tun->numqueues; i++) {
		tfile = rtnl_dereference(tun->tfiles[i]);
3623
		rings[i] = &tfile->tx_ring;
J
Jason Wang 已提交
3624 3625
	}
	list_for_each_entry(tfile, &tun->disabled, next)
3626
		rings[i++] = &tfile->tx_ring;
J
Jason Wang 已提交
3627

3628 3629
	ret = ptr_ring_resize_multiple(rings, n,
				       dev->tx_queue_len, GFP_KERNEL,
J
Jason Wang 已提交
3630
				       tun_ptr_free);
J
Jason Wang 已提交
3631

3632
	kfree(rings);
J
Jason Wang 已提交
3633 3634 3635 3636 3637 3638 3639 3640
	return ret;
}

static int tun_device_event(struct notifier_block *unused,
			    unsigned long event, void *ptr)
{
	struct net_device *dev = netdev_notifier_info_to_dev(ptr);
	struct tun_struct *tun = netdev_priv(dev);
3641
	int i;
J
Jason Wang 已提交
3642

3643 3644 3645
	if (dev->rtnl_link_ops != &tun_link_ops)
		return NOTIFY_DONE;

J
Jason Wang 已提交
3646 3647 3648 3649 3650
	switch (event) {
	case NETDEV_CHANGE_TX_QUEUE_LEN:
		if (tun_queue_resize(tun))
			return NOTIFY_BAD;
		break;
3651 3652 3653 3654 3655 3656 3657 3658
	case NETDEV_UP:
		for (i = 0; i < tun->numqueues; i++) {
			struct tun_file *tfile;

			tfile = rtnl_dereference(tun->tfiles[i]);
			tfile->socket.sk->sk_write_space(tfile->socket.sk);
		}
		break;
J
Jason Wang 已提交
3659 3660 3661 3662 3663 3664 3665 3666 3667 3668
	default:
		break;
	}

	return NOTIFY_DONE;
}

static struct notifier_block tun_notifier_block __read_mostly = {
	.notifier_call	= tun_device_event,
};
3669

L
Linus Torvalds 已提交
3670 3671 3672 3673
static int __init tun_init(void)
{
	int ret = 0;

3674
	pr_info("%s, %s\n", DRV_DESCRIPTION, DRV_VERSION);
L
Linus Torvalds 已提交
3675

3676
	ret = rtnl_link_register(&tun_link_ops);
3677
	if (ret) {
3678
		pr_err("Can't register link_ops\n");
3679
		goto err_linkops;
3680 3681
	}

L
Linus Torvalds 已提交
3682
	ret = misc_register(&tun_miscdev);
3683
	if (ret) {
3684
		pr_err("Can't register misc device %d\n", TUN_MINOR);
3685 3686
		goto err_misc;
	}
J
Jason Wang 已提交
3687

3688 3689 3690 3691 3692 3693
	ret = register_netdevice_notifier(&tun_notifier_block);
	if (ret) {
		pr_err("Can't register netdevice notifier\n");
		goto err_notifier;
	}

3694
	return  0;
3695 3696 3697

err_notifier:
	misc_deregister(&tun_miscdev);
3698
err_misc:
3699 3700
	rtnl_link_unregister(&tun_link_ops);
err_linkops:
L
Linus Torvalds 已提交
3701 3702 3703 3704 3705
	return ret;
}

static void tun_cleanup(void)
{
3706
	misc_deregister(&tun_miscdev);
3707
	rtnl_link_unregister(&tun_link_ops);
J
Jason Wang 已提交
3708
	unregister_netdevice_notifier(&tun_notifier_block);
L
Linus Torvalds 已提交
3709 3710
}

3711 3712 3713 3714 3715 3716
/* Get an underlying socket object from tun file.  Returns error unless file is
 * attached to a device.  The returned object works like a packet socket, it
 * can be used for sock_sendmsg/sock_recvmsg.  The caller is responsible for
 * holding a reference to the file for as long as the socket is in use. */
struct socket *tun_get_socket(struct file *file)
{
3717
	struct tun_file *tfile;
3718 3719
	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
3720 3721
	tfile = file->private_data;
	if (!tfile)
3722
		return ERR_PTR(-EBADFD);
J
Jason Wang 已提交
3723
	return &tfile->socket;
3724 3725 3726
}
EXPORT_SYMBOL_GPL(tun_get_socket);

3727
struct ptr_ring *tun_get_tx_ring(struct file *file)
J
Jason Wang 已提交
3728 3729 3730 3731 3732 3733 3734 3735
{
	struct tun_file *tfile;

	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
	tfile = file->private_data;
	if (!tfile)
		return ERR_PTR(-EBADFD);
3736
	return &tfile->tx_ring;
J
Jason Wang 已提交
3737
}
3738
EXPORT_SYMBOL_GPL(tun_get_tx_ring);
J
Jason Wang 已提交
3739

L
Linus Torvalds 已提交
3740 3741 3742 3743 3744 3745
module_init(tun_init);
module_exit(tun_cleanup);
MODULE_DESCRIPTION(DRV_DESCRIPTION);
MODULE_AUTHOR(DRV_COPYRIGHT);
MODULE_LICENSE("GPL");
MODULE_ALIAS_MISCDEV(TUN_MINOR);
3746
MODULE_ALIAS("devname:net/tun");