tun.c 81.9 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
/*
 *  TUN - Universal TUN/TAP device driver.
 *  Copyright (C) 1999-2002 Maxim Krasnyansky <maxk@qualcomm.com>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 *  GNU General Public License for more details.
 *
 *  $Id: tun.c,v 1.15 2002/03/01 02:44:24 maxk Exp $
 */

/*
 *  Changes:
 *
21 22 23
 *  Mike Kershaw <dragorn@kismetwireless.net> 2005/08/14
 *    Add TUNSETLINK ioctl to set the link encapsulation
 *
L
Linus Torvalds 已提交
24
 *  Mark Smith <markzzzsmith@yahoo.com.au>
J
Joe Perches 已提交
25
 *    Use eth_random_addr() for tap MAC address.
L
Linus Torvalds 已提交
26 27 28 29 30 31 32 33 34 35 36
 *
 *  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.
 */

37 38
#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt

L
Linus Torvalds 已提交
39 40 41 42 43 44 45 46
#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>
47
#include <linux/sched/signal.h>
L
Linus Torvalds 已提交
48 49 50 51 52 53 54 55 56 57 58
#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>
59
#include <linux/compat.h>
L
Linus Torvalds 已提交
60 61 62 63
#include <linux/if.h>
#include <linux/if_arp.h>
#include <linux/if_ether.h>
#include <linux/if_tun.h>
J
Jason Wang 已提交
64
#include <linux/if_vlan.h>
L
Linus Torvalds 已提交
65
#include <linux/crc32.h>
66
#include <linux/nsproxy.h>
67
#include <linux/virtio_net.h>
M
Michael S. Tsirkin 已提交
68
#include <linux/rcupdate.h>
69
#include <net/net_namespace.h>
70
#include <net/netns/generic.h>
71
#include <net/rtnetlink.h>
72
#include <net/sock.h>
73
#include <net/xdp.h>
74
#include <linux/seq_file.h>
H
Herbert Xu 已提交
75
#include <linux/uio.h>
J
Jason Wang 已提交
76
#include <linux/skb_array.h>
J
Jason Wang 已提交
77 78
#include <linux/bpf.h>
#include <linux/bpf_trace.h>
79
#include <linux/mutex.h>
L
Linus Torvalds 已提交
80

81
#include <linux/uaccess.h>
82
#include <linux/proc_fs.h>
L
Linus Torvalds 已提交
83

84 85 86
/* Uncomment to enable debugging */
/* #define TUN_DEBUG 1 */

L
Linus Torvalds 已提交
87 88
#ifdef TUN_DEBUG
static int debug;
89

90 91 92 93 94 95 96 97 98 99
#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)
100
#else
101 102 103 104 105 106 107 108 109 110
#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)
111 112
#endif

J
Jason Wang 已提交
113
#define TUN_HEADROOM 256
114
#define TUN_RX_PAD (NET_IP_ALIGN + NET_SKB_PAD)
115

116 117 118 119 120 121
/* 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 已提交
122 123
/* High bits in flags field are unused. */
#define TUN_VNET_LE     0x80000000
124
#define TUN_VNET_BE     0x40000000
125 126

#define TUN_FEATURES (IFF_NO_PI | IFF_ONE_QUEUE | IFF_VNET_HDR | \
127 128
		      IFF_MULTI_QUEUE | IFF_NAPI | IFF_NAPI_FRAGS)

129 130
#define GOODCOPY_LEN 128

131 132 133 134 135 136 137
#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];
};

138 139 140
/* 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
141
#define MAX_TAP_FLOWS  4096
J
Jason Wang 已提交
142

J
Jason Wang 已提交
143 144
#define TUN_FLOW_EXPIRE (3 * HZ)

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

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

	u32 rxhash;
194
	u32 rps_rxhash;
J
Jason Wang 已提交
195 196 197 198 199 200
	int queue_index;
	unsigned long updated;
};

#define TUN_NUM_FLOW_ENTRIES 1024

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

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

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

222
	int			align;
223
	int			vnet_hdr_sz;
J
Jason Wang 已提交
224 225 226 227 228
	int			sndbuf;
	struct tap_filter	txflt;
	struct sock_fprog	fprog;
	/* protected by rtnl lock */
	bool			filter_attached;
229 230
#ifdef TUN_DEBUG
	int debug;
L
Linus Torvalds 已提交
231
#endif
J
Jason Wang 已提交
232 233 234 235
	spinlock_t lock;
	struct hlist_head flows[TUN_NUM_FLOW_ENTRIES];
	struct timer_list flow_gc_timer;
	unsigned long ageing_time;
J
Jason Wang 已提交
236 237
	unsigned int numdisabled;
	struct list_head disabled;
238
	void *security;
239
	u32 flow_count;
J
Jason Wang 已提交
240
	u32 rx_batched;
241
	struct tun_pcpu_stats __percpu *pcpu_stats;
J
Jason Wang 已提交
242
	struct bpf_prog __rcu *xdp_prog;
243
	struct tun_prog __rcu *steering_prog;
244
	struct tun_prog __rcu *filter_prog;
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 311 312
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,
			  bool napi_en)
{
313
	tfile->napi_enabled = napi_en;
314 315 316 317
	if (napi_en) {
		netif_napi_add(tun->dev, &tfile->napi, tun_napi_poll,
			       NAPI_POLL_WEIGHT);
		napi_enable(&tfile->napi);
318
		mutex_init(&tfile->napi_mutex);
319 320 321 322 323
	}
}

static void tun_napi_disable(struct tun_struct *tun, struct tun_file *tfile)
{
324
	if (tfile->napi_enabled)
325 326 327 328 329
		napi_disable(&tfile->napi);
}

static void tun_napi_del(struct tun_struct *tun, struct tun_file *tfile)
{
330
	if (tfile->napi_enabled)
331 332 333
		netif_napi_del(&tfile->napi);
}

334 335 336 337 338
static bool tun_napi_frags_enabled(const struct tun_struct *tun)
{
	return READ_ONCE(tun->flags) & IFF_NAPI_FRAGS;
}

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 405 406 407 408 409 410 411
static inline u32 tun_hashfn(u32 rxhash)
{
	return rxhash & 0x3ff;
}

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 523 524 525 526 527 528 529 530 531 532 533

	if (!rxhash)
		return;
	else
		head = &tun->flows[tun_hashfn(rxhash)];

	rcu_read_lock();

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

550 551 552 553 554 555
/**
 * 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)
{
556
	if (unlikely(e->rps_rxhash != hash))
557 558 559
		e->rps_rxhash = hash;
}

J
Jason Wang 已提交
560
/* We try to identify a flow through its rxhash first. The reason that
S
stephen hemminger 已提交
561
 * we do not check rxq no. is because some cards(e.g 82599), chooses
J
Jason Wang 已提交
562 563 564 565 566
 * 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
 * different rxq no. here. If we could not get rxhash, then we would
 * hope the rxq no. may help here.
 */
567
static u16 tun_automq_select_queue(struct tun_struct *tun, struct sk_buff *skb)
J
Jason Wang 已提交
568
{
J
Jason Wang 已提交
569
	struct tun_flow_entry *e;
J
Jason Wang 已提交
570 571 572
	u32 txq = 0;
	u32 numqueues = 0;

573
	numqueues = READ_ONCE(tun->numqueues);
J
Jason Wang 已提交
574

J
Jason Wang 已提交
575
	txq = __skb_get_hash_symmetric(skb);
J
Jason Wang 已提交
576
	if (txq) {
J
Jason Wang 已提交
577
		e = tun_flow_find(&tun->flows[tun_hashfn(txq)], txq);
578 579
		if (e) {
			tun_flow_save_rps_rxhash(e, txq);
580
			txq = e->queue_index;
581
		} else
J
Jason Wang 已提交
582 583
			/* use multiply and shift instead of expensive divide */
			txq = ((u64)txq * numqueues) >> 32;
J
Jason Wang 已提交
584 585 586 587 588 589 590 591 592
	} else if (likely(skb_rx_queue_recorded(skb))) {
		txq = skb_get_rx_queue(skb);
		while (unlikely(txq >= numqueues))
			txq -= numqueues;
	}

	return txq;
}

593 594
static u16 tun_ebpf_select_queue(struct tun_struct *tun, struct sk_buff *skb)
{
595
	struct tun_prog *prog;
596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620
	u16 ret = 0;

	prog = rcu_dereference(tun->steering_prog);
	if (prog)
		ret = bpf_prog_run_clear_cb(prog->prog, skb);

	return ret % tun->numqueues;
}

static u16 tun_select_queue(struct net_device *dev, struct sk_buff *skb,
			    void *accel_priv, select_queue_fallback_t fallback)
{
	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;
}

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

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

J
Jason Wang 已提交
631 632 633 634 635 636
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 已提交
637 638 639 640 641 642 643
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 已提交
644
static struct tun_struct *tun_enable_queue(struct tun_file *tfile)
J
Jason Wang 已提交
645 646 647 648 649 650 651 652 653
{
	struct tun_struct *tun = tfile->detached;

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

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

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

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

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

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

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

684 685
	tun = rtnl_dereference(tfile->tun);

686 687 688 689 690
	if (tun && clean) {
		tun_napi_disable(tun, tfile);
		tun_napi_del(tun, tfile);
	}

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

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

		--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);
755
		tun_napi_disable(tun, 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]);
771
		tun_napi_del(tun, 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 790
static int tun_attach(struct tun_struct *tun, struct file *file,
		      bool skip_filter, bool napi)
791
{
E
Eric W. Biederman 已提交
792
	struct tun_file *tfile = file->private_data;
J
Jason Wang 已提交
793
	struct net_device *dev = tun->dev;
794
	int err;
795

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

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

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

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

	err = 0;
J
Jason Wang 已提交
814

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

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

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

	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;
846 847 848 849 850 851
		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 已提交
852 853 854
		err = 0;
	}

855
	rcu_assign_pointer(tfile->tun, tun);
J
Jason Wang 已提交
856 857
	rcu_assign_pointer(tun->tfiles[tun->numqueues], tfile);
	tun->numqueues++;
858

859
	if (tfile->detached) {
J
Jason Wang 已提交
860
		tun_enable_queue(tfile);
861
	} else {
J
Jason Wang 已提交
862
		sock_hold(&tfile->sk);
863 864
		tun_napi_init(tun, tfile, napi);
	}
J
Jason Wang 已提交
865

J
Jason Wang 已提交
866
	tun_set_real_num_queues(tun);
867

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

out:
	return err;
E
Eric W. Biederman 已提交
874 875
}

876
static struct tun_struct *tun_get(struct tun_file *tfile)
E
Eric W. Biederman 已提交
877
{
878
	struct tun_struct *tun;
879

880 881 882 883 884
	rcu_read_lock();
	tun = rcu_dereference(tfile->tun);
	if (tun)
		dev_hold(tun->dev);
	rcu_read_unlock();
885 886

	return tun;
E
Eric W. Biederman 已提交
887 888 889 890
}

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

894
/* TAP filtering */
895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922
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;
923 924 925
	addr = memdup_user(arg + sizeof(uf), alen);
	if (IS_ERR(addr))
		return PTR_ERR(addr);
926 927 928 929 930 931 932 933 934 935 936 937 938

	/* 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;

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

	/* 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;
961
free_addr:
962 963 964 965 966 967 968 969 970 971 972 973 974 975
	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++)
976
		if (ether_addr_equal(eh->h_dest, filter->addr[i]))
977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997
			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 已提交
998 999
/* Network device part of the driver */

1000
static const struct ethtool_ops tun_ethtool_ops;
L
Linus Torvalds 已提交
1001

1002 1003 1004
/* Net device detach from fd. */
static void tun_net_uninit(struct net_device *dev)
{
J
Jason Wang 已提交
1005
	tun_detach_all(dev);
1006 1007
}

L
Linus Torvalds 已提交
1008 1009 1010
/* Net device open. */
static int tun_net_open(struct net_device *dev)
{
1011 1012 1013
	struct tun_struct *tun = netdev_priv(dev);
	int i;

J
Jason Wang 已提交
1014
	netif_tx_start_all_queues(dev);
1015 1016 1017 1018 1019 1020 1021 1022

	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);
	}

L
Linus Torvalds 已提交
1023 1024 1025 1026 1027 1028
	return 0;
}

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

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

J
Jason Wang 已提交
1043
		rxhash = __skb_get_hash_symmetric(skb);
1044 1045 1046 1047 1048 1049 1050 1051
		if (rxhash) {
			struct tun_flow_entry *e;
			e = tun_flow_find(&tun->flows[tun_hashfn(rxhash)],
					rxhash);
			if (e)
				tun_flow_save_rps_rxhash(e, rxhash);
		}
	}
1052
#endif
1053 1054
}

1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066
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;
}

1067 1068 1069 1070 1071 1072
/* 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;
1073
	int len = skb->len;
1074 1075 1076 1077 1078

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

	/* Drop packet if interface is not attached */
1079
	if (txq >= tun->numqueues)
1080 1081 1082 1083
		goto drop;

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

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

J
Jason Wang 已提交
1087 1088
	BUG_ON(!tfile);

1089 1090 1091 1092 1093 1094
	/* 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 已提交
1095 1096
	if (tfile->socket.sk->sk_filter &&
	    sk_filter(tfile->socket.sk, skb))
M
Michael S. Tsirkin 已提交
1097 1098
		goto drop;

1099
	len = run_ebpf_filter(tun, skb, len);
B
Bjørn Mork 已提交
1100
	if (len == 0 || pskb_trim(skb, len))
1101 1102
		goto drop;

W
Willem de Bruijn 已提交
1103
	if (unlikely(skb_orphan_frags_rx(skb, GFP_ATOMIC)))
1104 1105
		goto drop;

1106
	skb_tx_timestamp(skb);
1107

M
Michael S. Tsirkin 已提交
1108
	/* Orphan the skb - required as we might hang on to it
1109 1110
	 * for indefinite time.
	 */
M
Michael S. Tsirkin 已提交
1111 1112
	skb_orphan(skb);

1113 1114
	nf_reset(skb);

1115
	if (ptr_ring_produce(&tfile->tx_ring, skb))
J
Jason Wang 已提交
1116
		goto drop;
L
Linus Torvalds 已提交
1117 1118

	/* Notify and wake up reader process */
J
Jason Wang 已提交
1119 1120
	if (tfile->flags & TUN_FASYNC)
		kill_fasync(&tfile->fasync, SIGIO, POLL_IN);
1121
	tfile->socket.sk->sk_data_ready(tfile->socket.sk);
1122 1123

	rcu_read_unlock();
1124
	return NETDEV_TX_OK;
L
Linus Torvalds 已提交
1125 1126

drop:
1127
	this_cpu_inc(tun->pcpu_stats->tx_dropped);
1128
	skb_tx_error(skb);
L
Linus Torvalds 已提交
1129
	kfree_skb(skb);
1130
	rcu_read_unlock();
1131
	return NET_XMIT_DROP;
L
Linus Torvalds 已提交
1132 1133
}

1134
static void tun_net_mclist(struct net_device *dev)
L
Linus Torvalds 已提交
1135
{
1136 1137 1138 1139 1140
	/*
	 * 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 已提交
1141 1142
}

1143 1144
static netdev_features_t tun_net_fix_features(struct net_device *dev,
	netdev_features_t features)
1145 1146 1147 1148 1149
{
	struct tun_struct *tun = netdev_priv(dev);

	return (features & tun->set_features) | (features & ~TUN_USER_FEATURES);
}
1150 1151 1152 1153 1154 1155 1156
#ifdef CONFIG_NET_POLL_CONTROLLER
static void tun_poll_controller(struct net_device *dev)
{
	/*
	 * Tun only receives frames when:
	 * 1) the char device endpoint gets data from user space
	 * 2) the tun socket gets a sendmsg call from user space
1157 1158 1159
	 * If NAPI is not enabled, since both of those are synchronous
	 * operations, we are guaranteed never to have pending data when we poll
	 * for it so there is nothing to do here but return.
1160 1161 1162
	 * We need this though so netpoll recognizes us as an interface that
	 * supports polling, which enables bridge devices in virt setups to
	 * still use netconsole
1163
	 * If NAPI is enabled, however, we need to schedule polling for all
1164 1165
	 * queues unless we are using napi_gro_frags(), which we call in
	 * process context and not in NAPI context.
1166
	 */
1167 1168 1169 1170 1171 1172
	struct tun_struct *tun = netdev_priv(dev);

	if (tun->flags & IFF_NAPI) {
		struct tun_file *tfile;
		int i;

1173 1174 1175
		if (tun_napi_frags_enabled(tun))
			return;

1176 1177 1178
		rcu_read_lock();
		for (i = 0; i < tun->numqueues; i++) {
			tfile = rcu_dereference(tun->tfiles[i]);
1179 1180
			if (tfile->napi_enabled)
				napi_schedule(&tfile->napi);
1181 1182 1183
		}
		rcu_read_unlock();
	}
1184 1185 1186
	return;
}
#endif
1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197

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;
}

1198
static void
1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233
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 已提交
1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259
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);
	struct bpf_prog *old_prog;

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

	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;
}

1260
static int tun_xdp(struct net_device *dev, struct netdev_bpf *xdp)
J
Jason Wang 已提交
1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273
{
	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);
		xdp->prog_attached = !!xdp->prog_id;
		return 0;
	default:
		return -EINVAL;
	}
}

S
Stephen Hemminger 已提交
1274
static const struct net_device_ops tun_netdev_ops = {
1275
	.ndo_uninit		= tun_net_uninit,
S
Stephen Hemminger 已提交
1276 1277
	.ndo_open		= tun_net_open,
	.ndo_stop		= tun_net_close,
1278
	.ndo_start_xmit		= tun_net_xmit,
1279
	.ndo_fix_features	= tun_net_fix_features,
J
Jason Wang 已提交
1280
	.ndo_select_queue	= tun_select_queue,
1281 1282 1283
#ifdef CONFIG_NET_POLL_CONTROLLER
	.ndo_poll_controller	= tun_poll_controller,
#endif
1284
	.ndo_set_rx_headroom	= tun_set_headroom,
1285
	.ndo_get_stats64	= tun_net_get_stats64,
S
Stephen Hemminger 已提交
1286 1287
};

1288 1289
static int tun_xdp_xmit(struct net_device *dev, int n,
			struct xdp_frame **frames, u32 flags)
J
Jason Wang 已提交
1290 1291 1292 1293
{
	struct tun_struct *tun = netdev_priv(dev);
	struct tun_file *tfile;
	u32 numqueues;
1294 1295 1296
	int drops = 0;
	int cnt = n;
	int i;
J
Jason Wang 已提交
1297

1298 1299 1300
	if (unlikely(flags & ~XDP_XMIT_FLAGS_NONE))
		return -EINVAL;

J
Jason Wang 已提交
1301 1302 1303 1304
	rcu_read_lock();

	numqueues = READ_ONCE(tun->numqueues);
	if (!numqueues) {
1305 1306
		rcu_read_unlock();
		return -ENXIO; /* Caller will free/return all frames */
J
Jason Wang 已提交
1307 1308 1309 1310
	}

	tfile = rcu_dereference(tun->tfiles[smp_processor_id() %
					    numqueues]);
1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324

	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 已提交
1325
	}
1326
	spin_unlock(&tfile->tx_ring.producer_lock);
J
Jason Wang 已提交
1327 1328

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

1332 1333 1334 1335 1336 1337 1338
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;

1339
	return tun_xdp_xmit(dev, 1, &frame, 0);
1340 1341
}

J
Jason Wang 已提交
1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364
static void tun_xdp_flush(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
	struct tun_file *tfile;
	u32 numqueues;

	rcu_read_lock();

	numqueues = READ_ONCE(tun->numqueues);
	if (!numqueues)
		goto out;

	tfile = rcu_dereference(tun->tfiles[smp_processor_id() %
					    numqueues]);
	/* 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);

out:
	rcu_read_unlock();
}

S
Stephen Hemminger 已提交
1365
static const struct net_device_ops tap_netdev_ops = {
1366
	.ndo_uninit		= tun_net_uninit,
S
Stephen Hemminger 已提交
1367 1368
	.ndo_open		= tun_net_open,
	.ndo_stop		= tun_net_close,
1369
	.ndo_start_xmit		= tun_net_xmit,
1370
	.ndo_fix_features	= tun_net_fix_features,
1371
	.ndo_set_rx_mode	= tun_net_mclist,
S
Stephen Hemminger 已提交
1372 1373
	.ndo_set_mac_address	= eth_mac_addr,
	.ndo_validate_addr	= eth_validate_addr,
J
Jason Wang 已提交
1374
	.ndo_select_queue	= tun_select_queue,
1375 1376 1377
#ifdef CONFIG_NET_POLL_CONTROLLER
	.ndo_poll_controller	= tun_poll_controller,
#endif
1378
	.ndo_features_check	= passthru_features_check,
1379
	.ndo_set_rx_headroom	= tun_set_headroom,
1380
	.ndo_get_stats64	= tun_net_get_stats64,
1381
	.ndo_bpf		= tun_xdp,
J
Jason Wang 已提交
1382 1383
	.ndo_xdp_xmit		= tun_xdp_xmit,
	.ndo_xdp_flush		= tun_xdp_flush,
S
Stephen Hemminger 已提交
1384 1385
};

1386
static void tun_flow_init(struct tun_struct *tun)
J
Jason Wang 已提交
1387 1388 1389 1390 1391 1392 1393
{
	int i;

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

	tun->ageing_time = TUN_FLOW_EXPIRE;
1394 1395 1396
	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 已提交
1397 1398 1399 1400 1401 1402 1403 1404
}

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

1405 1406 1407
#define MIN_MTU 68
#define MAX_MTU 65535

L
Linus Torvalds 已提交
1408 1409 1410 1411
/* Initialize net device. */
static void tun_net_init(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
1412

L
Linus Torvalds 已提交
1413
	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
1414
	case IFF_TUN:
S
Stephen Hemminger 已提交
1415 1416
		dev->netdev_ops = &tun_netdev_ops;

L
Linus Torvalds 已提交
1417 1418 1419 1420 1421 1422
		/* Point-to-Point TUN Device */
		dev->hard_header_len = 0;
		dev->addr_len = 0;
		dev->mtu = 1500;

		/* Zero header length */
1423
		dev->type = ARPHRD_NONE;
L
Linus Torvalds 已提交
1424 1425 1426
		dev->flags = IFF_POINTOPOINT | IFF_NOARP | IFF_MULTICAST;
		break;

M
Michael S. Tsirkin 已提交
1427
	case IFF_TAP:
K
Kusanagi Kouichi 已提交
1428
		dev->netdev_ops = &tap_netdev_ops;
L
Linus Torvalds 已提交
1429 1430
		/* Ethernet TAP Device */
		ether_setup(dev);
1431
		dev->priv_flags &= ~IFF_TX_SKB_SHARING;
1432
		dev->priv_flags |= IFF_LIVE_ADDR_CHANGE;
1433

1434
		eth_hw_addr_random(dev);
1435

L
Linus Torvalds 已提交
1436 1437
		break;
	}
1438 1439 1440

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

1443 1444 1445 1446 1447 1448 1449
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 已提交
1450 1451 1452
/* Character device part */

/* Poll */
1453
static __poll_t tun_chr_poll(struct file *file, poll_table *wait)
1454
{
1455
	struct tun_file *tfile = file->private_data;
1456
	struct tun_struct *tun = tun_get(tfile);
1457
	struct sock *sk;
1458
	__poll_t mask = 0;
L
Linus Torvalds 已提交
1459 1460

	if (!tun)
1461
		return EPOLLERR;
L
Linus Torvalds 已提交
1462

J
Jason Wang 已提交
1463
	sk = tfile->socket.sk;
1464

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

1467
	poll_wait(file, sk_sleep(sk), wait);
1468

1469
	if (!ptr_ring_empty(&tfile->tx_ring))
1470
		mask |= EPOLLIN | EPOLLRDNORM;
L
Linus Torvalds 已提交
1471

1472 1473 1474 1475 1476 1477 1478 1479
	/* 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)))
1480
		mask |= EPOLLOUT | EPOLLWRNORM;
1481

1482
	if (tun->dev->reg_state != NETREG_REGISTERED)
1483
		mask = EPOLLERR;
1484

E
Eric W. Biederman 已提交
1485
	tun_put(tun);
L
Linus Torvalds 已提交
1486 1487 1488
	return mask;
}

1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516
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++) {
1517
		struct page_frag *pfrag = &current->task_frag;
1518 1519 1520 1521 1522 1523 1524
		size_t fragsz = it->iov[i].iov_len;

		if (fragsz == 0 || fragsz > PAGE_SIZE) {
			err = -EINVAL;
			goto free;
		}

1525
		if (!skb_page_frag_refill(fragsz, pfrag, GFP_KERNEL)) {
1526 1527 1528 1529
			err = -ENOMEM;
			goto free;
		}

1530 1531 1532 1533
		skb_fill_page_desc(skb, i - 1, pfrag->page,
				   pfrag->offset, fragsz);
		page_ref_inc(pfrag->page);
		pfrag->offset += fragsz;
1534 1535 1536 1537 1538 1539 1540 1541 1542
	}

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

1543 1544
/* 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 已提交
1545
static struct sk_buff *tun_alloc_skb(struct tun_file *tfile,
1546 1547
				     size_t prepad, size_t len,
				     size_t linear, int noblock)
1548
{
J
Jason Wang 已提交
1549
	struct sock *sk = tfile->socket.sk;
1550
	struct sk_buff *skb;
1551
	int err;
1552 1553

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

1557
	skb = sock_alloc_send_pskb(sk, prepad + linear, len - linear, noblock,
1558
				   &err, 0);
1559
	if (!skb)
1560
		return ERR_PTR(err);
1561 1562 1563

	skb_reserve(skb, prepad);
	skb_put(skb, linear);
1564 1565
	skb->data_len = len - linear;
	skb->len += len - linear;
1566 1567 1568 1569

	return skb;
}

J
Jason Wang 已提交
1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605
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();
		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();
		while ((nskb = __skb_dequeue(&process_queue)))
			netif_receive_skb(nskb);
		netif_receive_skb(skb);
		local_bh_enable();
	}
}

1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627
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;
}

J
Jason Wang 已提交
1628 1629
static struct sk_buff *tun_build_skb(struct tun_struct *tun,
				     struct tun_file *tfile,
1630
				     struct iov_iter *from,
J
Jason Wang 已提交
1631
				     struct virtio_net_hdr *hdr,
J
Jason Wang 已提交
1632
				     int len, int *skb_xdp)
1633
{
1634
	struct page_frag *alloc_frag = &current->task_frag;
1635
	struct sk_buff *skb;
J
Jason Wang 已提交
1636
	struct bpf_prog *xdp_prog;
1637
	int buflen = SKB_DATA_ALIGN(sizeof(struct skb_shared_info));
J
Jason Wang 已提交
1638
	unsigned int delta = 0;
1639 1640
	char *buf;
	size_t copied;
1641 1642 1643 1644 1645 1646 1647 1648
	int err, pad = TUN_RX_PAD;

	rcu_read_lock();
	xdp_prog = rcu_dereference(tun->xdp_prog);
	if (xdp_prog)
		pad += TUN_HEADROOM;
	buflen += SKB_DATA_ALIGN(len + pad);
	rcu_read_unlock();
1649

1650
	alloc_frag->offset = ALIGN((u64)alloc_frag->offset, SMP_CACHE_BYTES);
1651 1652 1653 1654 1655
	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,
1656
				     alloc_frag->offset + pad,
1657 1658 1659 1660
				     len, from);
	if (copied != len)
		return ERR_PTR(-EFAULT);

1661 1662 1663 1664 1665
	/* 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.
	 */
	if (hdr->gso_type || !xdp_prog)
J
Jason Wang 已提交
1666
		*skb_xdp = 1;
J
Jason Wang 已提交
1667
	else
J
Jason Wang 已提交
1668
		*skb_xdp = 0;
J
Jason Wang 已提交
1669

1670
	preempt_disable();
J
Jason Wang 已提交
1671 1672
	rcu_read_lock();
	xdp_prog = rcu_dereference(tun->xdp_prog);
J
Jason Wang 已提交
1673
	if (xdp_prog && !*skb_xdp) {
J
Jason Wang 已提交
1674 1675 1676 1677 1678
		struct xdp_buff xdp;
		void *orig_data;
		u32 act;

		xdp.data_hard_start = buf;
1679
		xdp.data = buf + pad;
1680
		xdp_set_data_meta_invalid(&xdp);
J
Jason Wang 已提交
1681
		xdp.data_end = xdp.data + len;
J
Jesper Dangaard Brouer 已提交
1682
		xdp.rxq = &tfile->xdp_rxq;
J
Jason Wang 已提交
1683 1684 1685 1686 1687 1688 1689 1690
		orig_data = xdp.data;
		act = bpf_prog_run_xdp(xdp_prog, &xdp);

		switch (act) {
		case XDP_REDIRECT:
			get_page(alloc_frag->page);
			alloc_frag->offset += buflen;
			err = xdp_do_redirect(tun->dev, &xdp, xdp_prog);
1691
			xdp_do_flush_map();
J
Jason Wang 已提交
1692 1693
			if (err)
				goto err_redirect;
1694
			rcu_read_unlock();
1695
			preempt_enable();
J
Jason Wang 已提交
1696 1697
			return NULL;
		case XDP_TX:
J
Jason Wang 已提交
1698 1699
			get_page(alloc_frag->page);
			alloc_frag->offset += buflen;
1700
			if (tun_xdp_tx(tun->dev, &xdp))
J
Jason Wang 已提交
1701 1702 1703 1704 1705
				goto err_redirect;
			tun_xdp_flush(tun->dev);
			rcu_read_unlock();
			preempt_enable();
			return NULL;
J
Jason Wang 已提交
1706 1707
		case XDP_PASS:
			delta = orig_data - xdp.data;
1708
			len = xdp.data_end - xdp.data;
J
Jason Wang 已提交
1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720
			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:
			goto err_xdp;
		}
	}

1721
	skb = build_skb(buf, buflen);
J
Jason Wang 已提交
1722 1723
	if (!skb) {
		rcu_read_unlock();
1724
		preempt_enable();
1725
		return ERR_PTR(-ENOMEM);
J
Jason Wang 已提交
1726
	}
1727

1728
	skb_reserve(skb, pad - delta);
1729
	skb_put(skb, len);
1730 1731 1732
	get_page(alloc_frag->page);
	alloc_frag->offset += buflen;

J
Jason Wang 已提交
1733
	rcu_read_unlock();
1734
	preempt_enable();
J
Jason Wang 已提交
1735

1736
	return skb;
J
Jason Wang 已提交
1737 1738 1739 1740 1741

err_redirect:
	put_page(alloc_frag->page);
err_xdp:
	rcu_read_unlock();
1742
	preempt_enable();
J
Jason Wang 已提交
1743 1744
	this_cpu_inc(tun->pcpu_stats->rx_dropped);
	return NULL;
1745 1746
}

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

E
Eric Dumazet 已提交
1766 1767 1768
	if (!(tun->dev->flags & IFF_UP))
		return -EIO;

M
Michael S. Tsirkin 已提交
1769
	if (!(tun->flags & IFF_NO_PI)) {
1770
		if (len < sizeof(pi))
L
Linus Torvalds 已提交
1771
			return -EINVAL;
1772
		len -= sizeof(pi);
L
Linus Torvalds 已提交
1773

1774
		if (!copy_from_iter_full(&pi, sizeof(pi), from))
L
Linus Torvalds 已提交
1775 1776 1777
			return -EFAULT;
	}

M
Michael S. Tsirkin 已提交
1778
	if (tun->flags & IFF_VNET_HDR) {
W
Willem de Bruijn 已提交
1779 1780 1781
		int vnet_hdr_sz = READ_ONCE(tun->vnet_hdr_sz);

		if (len < vnet_hdr_sz)
1782
			return -EINVAL;
W
Willem de Bruijn 已提交
1783
		len -= vnet_hdr_sz;
1784

1785
		if (!copy_from_iter_full(&gso, sizeof(gso), from))
1786 1787
			return -EFAULT;

1788
		if ((gso.flags & VIRTIO_NET_HDR_F_NEEDS_CSUM) &&
1789 1790
		    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);
1791

1792
		if (tun16_to_cpu(tun, gso.hdr_len) > len)
1793
			return -EINVAL;
W
Willem de Bruijn 已提交
1794
		iov_iter_advance(from, vnet_hdr_sz - sizeof(gso));
1795 1796
	}

M
Michael S. Tsirkin 已提交
1797
	if ((tun->flags & TUN_TYPE_MASK) == IFF_TAP) {
1798
		align += NET_IP_ALIGN;
H
Herbert Xu 已提交
1799
		if (unlikely(len < ETH_HLEN ||
1800
			     (gso.hdr_len && tun16_to_cpu(tun, gso.hdr_len) < ETH_HLEN)))
1801 1802
			return -EINVAL;
	}
1803

1804 1805
	good_linear = SKB_MAX_HEAD(align);

1806
	if (msg_control) {
1807 1808
		struct iov_iter i = *from;

1809 1810
		/* There are 256 bytes to be copied in skb, so there is
		 * enough room for skb expand head in case it is used.
1811 1812
		 * The rest of the buffer is mapped from userspace.
		 */
1813
		copylen = gso.hdr_len ? tun16_to_cpu(tun, gso.hdr_len) : GOODCOPY_LEN;
1814 1815
		if (copylen > good_linear)
			copylen = good_linear;
1816
		linear = copylen;
1817 1818
		iov_iter_advance(&i, copylen);
		if (iov_iter_npages(&i, INT_MAX) <= MAX_SKB_FRAGS)
1819 1820 1821
			zerocopy = true;
	}

1822
	if (!frags && tun_can_build_skb(tun, tfile, len, noblock, zerocopy)) {
J
Jason Wang 已提交
1823 1824 1825 1826 1827
		/* 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);
1828
		if (IS_ERR(skb)) {
1829
			this_cpu_inc(tun->pcpu_stats->rx_dropped);
1830 1831
			return PTR_ERR(skb);
		}
J
Jason Wang 已提交
1832 1833
		if (!skb)
			return total_len;
1834 1835 1836 1837 1838 1839 1840 1841
	} 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 已提交
1842

1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855
		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);
		}

1856 1857 1858
		if (IS_ERR(skb)) {
			if (PTR_ERR(skb) != -EAGAIN)
				this_cpu_inc(tun->pcpu_stats->rx_dropped);
1859 1860
			if (frags)
				mutex_unlock(&tfile->napi_mutex);
1861 1862
			return PTR_ERR(skb);
		}
1863

1864 1865 1866 1867 1868 1869 1870 1871
		if (zerocopy)
			err = zerocopy_sg_from_iter(skb, from);
		else
			err = skb_copy_datagram_from_iter(skb, 0, from, len);

		if (err) {
			this_cpu_inc(tun->pcpu_stats->rx_dropped);
			kfree_skb(skb);
1872 1873 1874 1875 1876
			if (frags) {
				tfile->napi.skb = NULL;
				mutex_unlock(&tfile->napi_mutex);
			}

1877 1878
			return -EFAULT;
		}
D
Dave Jones 已提交
1879
	}
L
Linus Torvalds 已提交
1880

1881
	if (virtio_net_hdr_to_skb(skb, &gso, tun_is_little_endian(tun))) {
1882 1883
		this_cpu_inc(tun->pcpu_stats->rx_frame_errors);
		kfree_skb(skb);
1884 1885 1886 1887 1888
		if (frags) {
			tfile->napi.skb = NULL;
			mutex_unlock(&tfile->napi_mutex);
		}

1889 1890 1891
		return -EINVAL;
	}

L
Linus Torvalds 已提交
1892
	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
1893 1894
	case IFF_TUN:
		if (tun->flags & IFF_NO_PI) {
1895 1896 1897 1898
			u8 ip_version = skb->len ? (skb->data[0] >> 4) : 0;

			switch (ip_version) {
			case 4:
1899 1900
				pi.proto = htons(ETH_P_IP);
				break;
1901
			case 6:
1902 1903 1904
				pi.proto = htons(ETH_P_IPV6);
				break;
			default:
1905
				this_cpu_inc(tun->pcpu_stats->rx_dropped);
1906 1907 1908 1909 1910
				kfree_skb(skb);
				return -EINVAL;
			}
		}

1911
		skb_reset_mac_header(skb);
L
Linus Torvalds 已提交
1912
		skb->protocol = pi.proto;
1913
		skb->dev = tun->dev;
L
Linus Torvalds 已提交
1914
		break;
M
Michael S. Tsirkin 已提交
1915
	case IFF_TAP:
1916 1917
		if (!frags)
			skb->protocol = eth_type_trans(skb, tun->dev);
L
Linus Torvalds 已提交
1918
		break;
1919
	}
L
Linus Torvalds 已提交
1920

1921 1922 1923 1924
	/* 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;
1925
		skb_shinfo(skb)->tx_flags |= SKBTX_SHARED_FRAG;
1926 1927 1928
	} else if (msg_control) {
		struct ubuf_info *uarg = msg_control;
		uarg->callback(uarg, false);
1929 1930
	}

1931
	skb_reset_network_header(skb);
1932
	skb_probe_transport_header(skb, 0);
1933

J
Jason Wang 已提交
1934
	if (skb_xdp) {
J
Jason Wang 已提交
1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949
		struct bpf_prog *xdp_prog;
		int ret;

		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();
				return total_len;
			}
		}
		rcu_read_unlock();
	}

1950 1951 1952 1953 1954 1955
	/* 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)
1956
		rxhash = __skb_get_hash_symmetric(skb);
1957

1958 1959 1960 1961
	if (frags) {
		/* Exercise flow dissector code path. */
		u32 headlen = eth_get_headlen(skb->data, skb_headlen(skb));

1962
		if (unlikely(headlen > skb_headlen(skb))) {
1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973
			this_cpu_inc(tun->pcpu_stats->rx_dropped);
			napi_free_frags(&tfile->napi);
			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);
1974
	} else if (tfile->napi_enabled) {
1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990 1991
		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);
	}
1992

1993 1994 1995 1996 1997 1998
	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 已提交
1999

2000 2001 2002
	if (rxhash)
		tun_flow_update(tun, rxhash, tfile);

2003
	return total_len;
2004
}
L
Linus Torvalds 已提交
2005

2006
static ssize_t tun_chr_write_iter(struct kiocb *iocb, struct iov_iter *from)
L
Linus Torvalds 已提交
2007
{
2008
	struct file *file = iocb->ki_filp;
J
Jason Wang 已提交
2009
	struct tun_file *tfile = file->private_data;
2010
	struct tun_struct *tun = tun_get(tfile);
E
Eric W. Biederman 已提交
2011
	ssize_t result;
L
Linus Torvalds 已提交
2012 2013 2014 2015

	if (!tun)
		return -EBADFD;

J
Jason Wang 已提交
2016 2017
	result = tun_get_user(tun, tfile, NULL, from,
			      file->f_flags & O_NONBLOCK, false);
E
Eric W. Biederman 已提交
2018 2019 2020

	tun_put(tun);
	return result;
L
Linus Torvalds 已提交
2021 2022
}

J
Jason Wang 已提交
2023 2024
static ssize_t tun_put_user_xdp(struct tun_struct *tun,
				struct tun_file *tfile,
2025
				struct xdp_frame *xdp_frame,
J
Jason Wang 已提交
2026 2027 2028
				struct iov_iter *iter)
{
	int vnet_hdr_sz = 0;
2029
	size_t size = xdp_frame->len;
J
Jason Wang 已提交
2030 2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044
	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));
	}

2045
	ret = copy_to_iter(xdp_frame->data, size, iter) + vnet_hdr_sz;
J
Jason Wang 已提交
2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056

	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 已提交
2057
/* Put packet to the user space buffer */
2058
static ssize_t tun_put_user(struct tun_struct *tun,
J
Jason Wang 已提交
2059
			    struct tun_file *tfile,
2060
			    struct sk_buff *skb,
H
Herbert Xu 已提交
2061
			    struct iov_iter *iter)
L
Linus Torvalds 已提交
2062 2063
{
	struct tun_pi pi = { 0, skb->protocol };
2064
	struct tun_pcpu_stats *stats;
H
Herbert Xu 已提交
2065
	ssize_t total;
2066
	int vlan_offset = 0;
2067
	int vlan_hlen = 0;
H
Herbert Xu 已提交
2068
	int vnet_hdr_sz = 0;
2069

2070
	if (skb_vlan_tag_present(skb))
2071
		vlan_hlen = VLAN_HLEN;
L
Linus Torvalds 已提交
2072

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

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

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

H
Herbert Xu 已提交
2082 2083
		total += sizeof(pi);
		if (iov_iter_count(iter) < total) {
L
Linus Torvalds 已提交
2084 2085 2086
			/* Packet will be striped */
			pi.flags |= TUN_PKT_STRIP;
		}
2087

H
Herbert Xu 已提交
2088
		if (copy_to_iter(&pi, sizeof(pi), iter) != sizeof(pi))
L
Linus Torvalds 已提交
2089
			return -EFAULT;
2090
	}
L
Linus Torvalds 已提交
2091

H
Herbert Xu 已提交
2092
	if (vnet_hdr_sz) {
2093
		struct virtio_net_hdr gso;
2094

H
Herbert Xu 已提交
2095
		if (iov_iter_count(iter) < vnet_hdr_sz)
2096 2097
			return -EINVAL;

2098
		if (virtio_net_hdr_from_skb(skb, &gso,
2099
					    tun_is_little_endian(tun), true)) {
2100
			struct skb_shared_info *sinfo = skb_shinfo(skb);
2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111
			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;
		}
2112

H
Herbert Xu 已提交
2113
		if (copy_to_iter(&gso, sizeof(gso), iter) != sizeof(gso))
2114
			return -EFAULT;
2115 2116

		iov_iter_advance(iter, vnet_hdr_sz - sizeof(gso));
2117 2118
	}

2119
	if (vlan_hlen) {
H
Herbert Xu 已提交
2120
		int ret;
2121
		struct veth veth;
J
Jason Wang 已提交
2122 2123

		veth.h_vlan_proto = skb->vlan_proto;
2124
		veth.h_vlan_TCI = htons(skb_vlan_tag_get(skb));
J
Jason Wang 已提交
2125 2126 2127

		vlan_offset = offsetof(struct vlan_ethhdr, h_vlan_proto);

H
Herbert Xu 已提交
2128 2129
		ret = skb_copy_datagram_iter(skb, 0, iter, vlan_offset);
		if (ret || !iov_iter_count(iter))
J
Jason Wang 已提交
2130 2131
			goto done;

H
Herbert Xu 已提交
2132 2133
		ret = copy_to_iter(&veth, sizeof(veth), iter);
		if (ret != sizeof(veth) || !iov_iter_count(iter))
J
Jason Wang 已提交
2134 2135
			goto done;
	}
L
Linus Torvalds 已提交
2136

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

J
Jason Wang 已提交
2139
done:
2140 2141 2142 2143 2144 2145 2146
	/* 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 已提交
2147 2148 2149 2150

	return total;
}

J
Jason Wang 已提交
2151
static void *tun_ring_recv(struct tun_file *tfile, int noblock, int *err)
J
Jason Wang 已提交
2152 2153
{
	DECLARE_WAITQUEUE(wait, current);
J
Jason Wang 已提交
2154
	void *ptr = NULL;
J
Jason Wang 已提交
2155
	int error = 0;
J
Jason Wang 已提交
2156

J
Jason Wang 已提交
2157 2158
	ptr = ptr_ring_consume(&tfile->tx_ring);
	if (ptr)
J
Jason Wang 已提交
2159 2160
		goto out;
	if (noblock) {
J
Jason Wang 已提交
2161
		error = -EAGAIN;
J
Jason Wang 已提交
2162 2163 2164 2165 2166 2167 2168
		goto out;
	}

	add_wait_queue(&tfile->wq.wait, &wait);
	current->state = TASK_INTERRUPTIBLE;

	while (1) {
J
Jason Wang 已提交
2169 2170
		ptr = ptr_ring_consume(&tfile->tx_ring);
		if (ptr)
J
Jason Wang 已提交
2171 2172
			break;
		if (signal_pending(current)) {
J
Jason Wang 已提交
2173
			error = -ERESTARTSYS;
J
Jason Wang 已提交
2174 2175 2176
			break;
		}
		if (tfile->socket.sk->sk_shutdown & RCV_SHUTDOWN) {
J
Jason Wang 已提交
2177
			error = -EFAULT;
J
Jason Wang 已提交
2178 2179 2180 2181 2182 2183 2184 2185 2186 2187
			break;
		}

		schedule();
	}

	current->state = TASK_RUNNING;
	remove_wait_queue(&tfile->wq.wait, &wait);

out:
J
Jason Wang 已提交
2188
	*err = error;
J
Jason Wang 已提交
2189
	return ptr;
J
Jason Wang 已提交
2190 2191
}

J
Jason Wang 已提交
2192
static ssize_t tun_do_read(struct tun_struct *tun, struct tun_file *tfile,
2193
			   struct iov_iter *to,
J
Jason Wang 已提交
2194
			   int noblock, void *ptr)
L
Linus Torvalds 已提交
2195
{
2196
	ssize_t ret;
J
Jason Wang 已提交
2197
	int err;
L
Linus Torvalds 已提交
2198

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

W
Wei Xu 已提交
2201
	if (!iov_iter_count(to)) {
J
Jason Wang 已提交
2202
		tun_ptr_free(ptr);
2203
		return 0;
W
Wei Xu 已提交
2204
	}
L
Linus Torvalds 已提交
2205

J
Jason Wang 已提交
2206
	if (!ptr) {
2207
		/* Read frames from ring */
J
Jason Wang 已提交
2208 2209
		ptr = tun_ring_recv(tfile, noblock, &err);
		if (!ptr)
2210 2211
			return err;
	}
H
Herbert Xu 已提交
2212

2213 2214
	if (tun_is_xdp_frame(ptr)) {
		struct xdp_frame *xdpf = tun_ptr_to_xdp(ptr);
J
Jason Wang 已提交
2215

2216
		ret = tun_put_user_xdp(tun, tfile, xdpf, to);
2217
		xdp_return_frame(xdpf);
J
Jason Wang 已提交
2218 2219 2220 2221 2222 2223 2224 2225 2226
	} 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 已提交
2227

2228 2229 2230
	return ret;
}

2231
static ssize_t tun_chr_read_iter(struct kiocb *iocb, struct iov_iter *to)
2232 2233 2234
{
	struct file *file = iocb->ki_filp;
	struct tun_file *tfile = file->private_data;
2235
	struct tun_struct *tun = tun_get(tfile);
2236
	ssize_t len = iov_iter_count(to), ret;
2237 2238 2239

	if (!tun)
		return -EBADFD;
2240
	ret = tun_do_read(tun, tfile, to, file->f_flags & O_NONBLOCK, NULL);
2241
	ret = min_t(ssize_t, ret, len);
Z
Zhi Yong Wu 已提交
2242 2243
	if (ret > 0)
		iocb->ki_pos = ret;
E
Eric W. Biederman 已提交
2244
	tun_put(tun);
L
Linus Torvalds 已提交
2245 2246 2247
	return ret;
}

2248
static void tun_prog_free(struct rcu_head *rcu)
2249
{
2250
	struct tun_prog *prog = container_of(rcu, struct tun_prog, rcu);
2251 2252 2253 2254 2255

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

J
Jason Wang 已提交
2256 2257
static int __tun_set_ebpf(struct tun_struct *tun,
			  struct tun_prog __rcu **prog_p,
2258
			  struct bpf_prog *prog)
2259
{
2260
	struct tun_prog *old, *new = NULL;
2261 2262 2263 2264 2265 2266 2267 2268

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

2269
	spin_lock_bh(&tun->lock);
2270
	old = rcu_dereference_protected(*prog_p,
2271
					lockdep_is_held(&tun->lock));
2272
	rcu_assign_pointer(*prog_p, new);
2273
	spin_unlock_bh(&tun->lock);
2274 2275

	if (old)
2276
		call_rcu(&old->rcu, tun_prog_free);
2277 2278 2279 2280

	return 0;
}

J
Jason Wang 已提交
2281 2282 2283 2284
static void tun_free_netdev(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

J
Jason Wang 已提交
2285
	BUG_ON(!(list_empty(&tun->disabled)));
2286
	free_percpu(tun->pcpu_stats);
J
Jason Wang 已提交
2287
	tun_flow_uninit(tun);
2288
	security_tun_dev_free_security(tun->security);
2289
	__tun_set_ebpf(tun, &tun->steering_prog, NULL);
2290
	__tun_set_ebpf(tun, &tun->filter_prog, NULL);
J
Jason Wang 已提交
2291 2292
}

L
Linus Torvalds 已提交
2293 2294 2295 2296
static void tun_setup(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

2297 2298
	tun->owner = INVALID_UID;
	tun->group = INVALID_GID;
L
Linus Torvalds 已提交
2299 2300

	dev->ethtool_ops = &tun_ethtool_ops;
2301 2302
	dev->needs_free_netdev = true;
	dev->priv_destructor = tun_free_netdev;
J
Jason Wang 已提交
2303 2304
	/* We prefer our own queue length */
	dev->tx_queue_len = TUN_READQ_SIZE;
L
Linus Torvalds 已提交
2305 2306
}

2307 2308 2309
/* Trivial set of netlink ops to allow deleting tun or tap
 * device with netlink.
 */
2310 2311
static int tun_validate(struct nlattr *tb[], struct nlattr *data[],
			struct netlink_ext_ack *extack)
2312 2313 2314 2315
{
	return -EINVAL;
}

2316 2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 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
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;
}

2370 2371 2372 2373 2374
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,
2375 2376
	.get_size       = tun_get_size,
	.fill_info      = tun_fill_info,
2377 2378
};

2379 2380
static void tun_sock_write_space(struct sock *sk)
{
J
Jason Wang 已提交
2381
	struct tun_file *tfile;
2382
	wait_queue_head_t *wqueue;
2383 2384 2385 2386

	if (!sock_writeable(sk))
		return;

2387
	if (!test_and_clear_bit(SOCKWQ_ASYNC_NOSPACE, &sk->sk_socket->flags))
2388 2389
		return;

2390 2391
	wqueue = sk_sleep(sk);
	if (wqueue && waitqueue_active(wqueue))
2392 2393
		wake_up_interruptible_sync_poll(wqueue, EPOLLOUT |
						EPOLLWRNORM | EPOLLWRBAND);
H
Herbert Xu 已提交
2394

J
Jason Wang 已提交
2395 2396
	tfile = container_of(sk, struct tun_file, sk);
	kill_fasync(&tfile->fasync, SIGIO, POLL_OUT);
2397 2398
}

2399
static int tun_sendmsg(struct socket *sock, struct msghdr *m, size_t total_len)
2400
{
J
Jason Wang 已提交
2401 2402
	int ret;
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2403
	struct tun_struct *tun = tun_get(tfile);
J
Jason Wang 已提交
2404 2405 2406

	if (!tun)
		return -EBADFD;
2407

A
Al Viro 已提交
2408
	ret = tun_get_user(tun, tfile, m->msg_control, &m->msg_iter,
J
Jason Wang 已提交
2409 2410
			   m->msg_flags & MSG_DONTWAIT,
			   m->msg_flags & MSG_MORE);
J
Jason Wang 已提交
2411 2412
	tun_put(tun);
	return ret;
2413 2414
}

2415
static int tun_recvmsg(struct socket *sock, struct msghdr *m, size_t total_len,
2416 2417
		       int flags)
{
J
Jason Wang 已提交
2418
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2419
	struct tun_struct *tun = tun_get(tfile);
J
Jason Wang 已提交
2420
	void *ptr = m->msg_control;
2421
	int ret;
J
Jason Wang 已提交
2422

W
Wei Xu 已提交
2423 2424
	if (!tun) {
		ret = -EBADFD;
J
Jason Wang 已提交
2425
		goto out_free;
W
Wei Xu 已提交
2426
	}
J
Jason Wang 已提交
2427

2428
	if (flags & ~(MSG_DONTWAIT|MSG_TRUNC|MSG_ERRQUEUE)) {
2429
		ret = -EINVAL;
W
Wei Xu 已提交
2430
		goto out_put_tun;
2431
	}
2432 2433 2434 2435 2436
	if (flags & MSG_ERRQUEUE) {
		ret = sock_recv_errqueue(sock->sk, m, total_len,
					 SOL_PACKET, TUN_TX_TIMESTAMP);
		goto out;
	}
J
Jason Wang 已提交
2437
	ret = tun_do_read(tun, tfile, &m->msg_iter, flags & MSG_DONTWAIT, ptr);
2438
	if (ret > (ssize_t)total_len) {
2439 2440 2441
		m->msg_flags |= MSG_TRUNC;
		ret = flags & MSG_TRUNC ? ret : total_len;
	}
2442
out:
J
Jason Wang 已提交
2443
	tun_put(tun);
2444
	return ret;
W
Wei Xu 已提交
2445 2446 2447

out_put_tun:
	tun_put(tun);
J
Jason Wang 已提交
2448 2449
out_free:
	tun_ptr_free(ptr);
W
Wei Xu 已提交
2450
	return ret;
2451 2452
}

J
Jason Wang 已提交
2453 2454 2455
static int tun_ptr_peek_len(void *ptr)
{
	if (likely(ptr)) {
2456 2457
		if (tun_is_xdp_frame(ptr)) {
			struct xdp_frame *xdpf = tun_ptr_to_xdp(ptr);
J
Jason Wang 已提交
2458

2459
			return xdpf->len;
J
Jason Wang 已提交
2460 2461 2462 2463 2464 2465 2466
		}
		return __skb_array_len_with_tag(ptr);
	} else {
		return 0;
	}
}

J
Jason Wang 已提交
2467 2468 2469 2470 2471 2472
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;

2473
	tun = tun_get(tfile);
J
Jason Wang 已提交
2474 2475 2476
	if (!tun)
		return 0;

J
Jason Wang 已提交
2477
	ret = PTR_RING_PEEK_CALL(&tfile->tx_ring, tun_ptr_peek_len);
J
Jason Wang 已提交
2478 2479 2480 2481 2482
	tun_put(tun);

	return ret;
}

2483 2484
/* Ops structure to mimic raw sockets with tun */
static const struct proto_ops tun_socket_ops = {
J
Jason Wang 已提交
2485
	.peek_len = tun_peek_len,
2486 2487 2488 2489
	.sendmsg = tun_sendmsg,
	.recvmsg = tun_recvmsg,
};

2490 2491 2492
static struct proto tun_proto = {
	.name		= "tun",
	.owner		= THIS_MODULE,
J
Jason Wang 已提交
2493
	.obj_size	= sizeof(struct tun_file),
2494
};
2495

2496 2497
static int tun_flags(struct tun_struct *tun)
{
2498
	return tun->flags & (TUN_FEATURES | IFF_PERSIST | IFF_TUN | IFF_TAP);
2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511
}

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));
2512 2513 2514 2515
	return uid_valid(tun->owner)?
		sprintf(buf, "%u\n",
			from_kuid_munged(current_user_ns(), tun->owner)):
		sprintf(buf, "-1\n");
2516 2517 2518 2519 2520 2521
}

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));
2522 2523 2524 2525
	return gid_valid(tun->group) ?
		sprintf(buf, "%u\n",
			from_kgid_munged(current_user_ns(), tun->group)):
		sprintf(buf, "-1\n");
2526 2527 2528 2529 2530 2531
}

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);

2532 2533 2534 2535 2536 2537 2538 2539 2540 2541 2542
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
};

2543
static int tun_set_iff(struct net *net, struct file *file, struct ifreq *ifr)
L
Linus Torvalds 已提交
2544 2545
{
	struct tun_struct *tun;
J
Jason Wang 已提交
2546
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
2547 2548 2549
	struct net_device *dev;
	int err;

2550 2551 2552
	if (tfile->detached)
		return -EINVAL;

2553 2554 2555 2556 2557 2558 2559 2560 2561
	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;
	}

2562 2563
	dev = __dev_get_by_name(net, ifr->ifr_name);
	if (dev) {
2564 2565
		if (ifr->ifr_flags & IFF_TUN_EXCL)
			return -EBUSY;
2566 2567 2568 2569 2570 2571 2572
		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;

2573
		if (!!(ifr->ifr_flags & IFF_MULTI_QUEUE) !=
M
Michael S. Tsirkin 已提交
2574
		    !!(tun->flags & IFF_MULTI_QUEUE))
2575 2576
			return -EINVAL;

2577
		if (tun_not_capable(tun))
P
Paul Moore 已提交
2578
			return -EPERM;
2579
		err = security_tun_dev_open(tun->security);
P
Paul Moore 已提交
2580 2581 2582
		if (err < 0)
			return err;

2583 2584
		err = tun_attach(tun, file, ifr->ifr_flags & IFF_NOFILTER,
				 ifr->ifr_flags & IFF_NAPI);
2585 2586
		if (err < 0)
			return err;
J
Jason Wang 已提交
2587

M
Michael S. Tsirkin 已提交
2588
		if (tun->flags & IFF_MULTI_QUEUE &&
2589 2590 2591 2592
		    (tun->numqueues + tun->numdisabled > 1)) {
			/* One or more queue has already been attached, no need
			 * to initialize the device again.
			 */
2593
			netdev_state_change(dev);
2594 2595
			return 0;
		}
2596 2597 2598

		tun->flags = (tun->flags & ~TUN_FEATURES) |
			      (ifr->ifr_flags & TUN_FEATURES);
2599 2600 2601

		netdev_state_change(dev);
	} else {
L
Linus Torvalds 已提交
2602 2603
		char *name;
		unsigned long flags = 0;
2604 2605
		int queues = ifr->ifr_flags & IFF_MULTI_QUEUE ?
			     MAX_TAP_QUEUES : 1;
L
Linus Torvalds 已提交
2606

2607
		if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
2608
			return -EPERM;
P
Paul Moore 已提交
2609 2610 2611
		err = security_tun_dev_create();
		if (err < 0)
			return err;
2612

L
Linus Torvalds 已提交
2613 2614 2615
		/* Set dev type */
		if (ifr->ifr_flags & IFF_TUN) {
			/* TUN device */
M
Michael S. Tsirkin 已提交
2616
			flags |= IFF_TUN;
L
Linus Torvalds 已提交
2617 2618 2619
			name = "tun%d";
		} else if (ifr->ifr_flags & IFF_TAP) {
			/* TAP device */
M
Michael S. Tsirkin 已提交
2620
			flags |= IFF_TAP;
L
Linus Torvalds 已提交
2621
			name = "tap%d";
2622
		} else
2623
			return -EINVAL;
2624

L
Linus Torvalds 已提交
2625 2626 2627
		if (*ifr->ifr_name)
			name = ifr->ifr_name;

J
Jason Wang 已提交
2628
		dev = alloc_netdev_mqs(sizeof(struct tun_struct), name,
2629 2630
				       NET_NAME_UNKNOWN, tun_setup, queues,
				       queues);
2631

L
Linus Torvalds 已提交
2632 2633
		if (!dev)
			return -ENOMEM;
2634
		err = dev_get_valid_name(net, dev, name);
2635
		if (err < 0)
2636
			goto err_free_dev;
L
Linus Torvalds 已提交
2637

2638
		dev_net_set(dev, net);
2639
		dev->rtnl_link_ops = &tun_link_ops;
2640
		dev->ifindex = tfile->ifindex;
2641
		dev->sysfs_groups[0] = &tun_attr_group;
S
Stephen Hemminger 已提交
2642

L
Linus Torvalds 已提交
2643 2644 2645
		tun = netdev_priv(dev);
		tun->dev = dev;
		tun->flags = flags;
2646
		tun->txflt.count = 0;
2647
		tun->vnet_hdr_sz = sizeof(struct virtio_net_hdr);
2648

2649
		tun->align = NET_SKB_PAD;
J
Jason Wang 已提交
2650 2651
		tun->filter_attached = false;
		tun->sndbuf = tfile->socket.sk->sk_sndbuf;
J
Jason Wang 已提交
2652
		tun->rx_batched = 0;
2653
		RCU_INIT_POINTER(tun->steering_prog, NULL);
2654

2655 2656 2657 2658 2659 2660
		tun->pcpu_stats = netdev_alloc_pcpu_stats(struct tun_pcpu_stats);
		if (!tun->pcpu_stats) {
			err = -ENOMEM;
			goto err_free_dev;
		}

J
Jason Wang 已提交
2661 2662
		spin_lock_init(&tun->lock);

2663 2664
		err = security_tun_dev_alloc_security(&tun->security);
		if (err < 0)
2665
			goto err_free_stat;
P
Paul Moore 已提交
2666

L
Linus Torvalds 已提交
2667
		tun_net_init(dev);
2668
		tun_flow_init(tun);
J
Jason Wang 已提交
2669

2670
		dev->hw_features = NETIF_F_SG | NETIF_F_FRAGLIST |
J
Jason Wang 已提交
2671 2672
				   TUN_USER_FEATURES | NETIF_F_HW_VLAN_CTAG_TX |
				   NETIF_F_HW_VLAN_STAG_TX;
2673
		dev->features = dev->hw_features | NETIF_F_LLTX;
2674 2675 2676
		dev->vlan_features = dev->features &
				     ~(NETIF_F_HW_VLAN_CTAG_TX |
				       NETIF_F_HW_VLAN_STAG_TX);
2677

2678 2679 2680
		tun->flags = (tun->flags & ~TUN_FEATURES) |
			      (ifr->ifr_flags & TUN_FEATURES);

J
Jason Wang 已提交
2681
		INIT_LIST_HEAD(&tun->disabled);
2682
		err = tun_attach(tun, file, false, ifr->ifr_flags & IFF_NAPI);
2683
		if (err < 0)
2684
			goto err_free_flow;
2685

L
Linus Torvalds 已提交
2686 2687
		err = register_netdevice(tun->dev);
		if (err < 0)
2688
			goto err_detach;
L
Linus Torvalds 已提交
2689 2690
	}

2691 2692
	netif_carrier_on(tun->dev);

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

2695 2696 2697 2698
	/* Make sure persistent devices do not get stuck in
	 * xoff state.
	 */
	if (netif_running(tun->dev))
J
Jason Wang 已提交
2699
		netif_tx_wake_all_queues(tun->dev);
2700

L
Linus Torvalds 已提交
2701 2702 2703
	strcpy(ifr->ifr_name, tun->dev->name);
	return 0;

2704 2705
err_detach:
	tun_detach_all(dev);
2706 2707 2708
	/* register_netdevice() already called tun_free_netdev() */
	goto err_free_dev;

2709 2710 2711
err_free_flow:
	tun_flow_uninit(tun);
	security_tun_dev_free_security(tun->security);
2712 2713
err_free_stat:
	free_percpu(tun->pcpu_stats);
2714
err_free_dev:
L
Linus Torvalds 已提交
2715 2716 2717 2718
	free_netdev(dev);
	return err;
}

R
Rami Rosen 已提交
2719
static void tun_get_iff(struct net *net, struct tun_struct *tun,
2720
		       struct ifreq *ifr)
2721
{
2722
	tun_debug(KERN_INFO, tun, "tun_get_iff\n");
2723 2724 2725

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

2726
	ifr->ifr_flags = tun_flags(tun);
2727 2728 2729

}

2730 2731
/* This is like a cut-down ethtool ops, except done via tun fd so no
 * privs required. */
2732
static int set_offload(struct tun_struct *tun, unsigned long arg)
2733
{
2734
	netdev_features_t features = 0;
2735 2736

	if (arg & TUN_F_CSUM) {
2737
		features |= NETIF_F_HW_CSUM;
2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750
		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);
		}
2751 2752

		arg &= ~TUN_F_UFO;
2753 2754 2755 2756 2757 2758 2759
	}

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

2760
	tun->set_features = features;
2761 2762
	tun->dev->wanted_features &= ~TUN_USER_FEATURES;
	tun->dev->wanted_features |= features;
2763
	netdev_update_features(tun->dev);
2764 2765 2766 2767

	return 0;
}

J
Jason Wang 已提交
2768 2769 2770 2771 2772 2773
static void tun_detach_filter(struct tun_struct *tun, int n)
{
	int i;
	struct tun_file *tfile;

	for (i = 0; i < n; i++) {
2774
		tfile = rtnl_dereference(tun->tfiles[i]);
2775 2776 2777
		lock_sock(tfile->socket.sk);
		sk_detach_filter(tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788
	}

	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++) {
2789
		tfile = rtnl_dereference(tun->tfiles[i]);
2790 2791 2792
		lock_sock(tfile->socket.sk);
		ret = sk_attach_filter(&tun->fprog, tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804 2805 2806 2807 2808
		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++) {
2809
		tfile = rtnl_dereference(tun->tfiles[i]);
J
Jason Wang 已提交
2810 2811 2812 2813
		tfile->socket.sk->sk_sndbuf = tun->sndbuf;
	}
}

2814 2815 2816 2817 2818 2819 2820 2821 2822
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 已提交
2823
		tun = tfile->detached;
2824
		if (!tun) {
2825
			ret = -EINVAL;
2826 2827 2828 2829 2830
			goto unlock;
		}
		ret = security_tun_dev_attach_queue(tun->security);
		if (ret < 0)
			goto unlock;
2831
		ret = tun_attach(tun, file, false, tun->flags & IFF_NAPI);
J
Jason Wang 已提交
2832
	} else if (ifr->ifr_flags & IFF_DETACH_QUEUE) {
2833
		tun = rtnl_dereference(tfile->tun);
M
Michael S. Tsirkin 已提交
2834
		if (!tun || !(tun->flags & IFF_MULTI_QUEUE) || tfile->detached)
J
Jason Wang 已提交
2835 2836 2837 2838
			ret = -EINVAL;
		else
			__tun_detach(tfile, false);
	} else
2839 2840
		ret = -EINVAL;

2841 2842 2843
	if (ret >= 0)
		netdev_state_change(tun->dev);

2844
unlock:
2845 2846 2847 2848
	rtnl_unlock();
	return ret;
}

2849 2850
static int tun_set_ebpf(struct tun_struct *tun, struct tun_prog **prog_p,
			void __user *data)
2851 2852 2853 2854 2855 2856 2857 2858 2859 2860 2861 2862 2863 2864 2865
{
	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);
	}

2866
	return __tun_set_ebpf(tun, prog_p, prog);
2867 2868
}

2869 2870
static long __tun_chr_ioctl(struct file *file, unsigned int cmd,
			    unsigned long arg, int ifreq_len)
L
Linus Torvalds 已提交
2871
{
E
Eric W. Biederman 已提交
2872
	struct tun_file *tfile = file->private_data;
2873
	struct net *net = sock_net(&tfile->sk);
E
Eric W. Biederman 已提交
2874
	struct tun_struct *tun;
L
Linus Torvalds 已提交
2875 2876
	void __user* argp = (void __user*)arg;
	struct ifreq ifr;
2877 2878
	kuid_t owner;
	kgid_t group;
2879
	int sndbuf;
2880
	int vnet_hdr_sz;
2881
	unsigned int ifindex;
M
Michael S. Tsirkin 已提交
2882
	int le;
2883
	int ret;
2884
	bool do_notify = false;
L
Linus Torvalds 已提交
2885

2886 2887
	if (cmd == TUNSETIFF || cmd == TUNSETQUEUE ||
	    (_IOC_TYPE(cmd) == SOCK_IOC_TYPE && cmd != SIOCGSKNS)) {
2888
		if (copy_from_user(&ifr, argp, ifreq_len))
L
Linus Torvalds 已提交
2889
			return -EFAULT;
D
David S. Miller 已提交
2890
	} else {
2891
		memset(&ifr, 0, sizeof(ifr));
D
David S. Miller 已提交
2892
	}
E
Eric W. Biederman 已提交
2893 2894 2895
	if (cmd == TUNGETFEATURES) {
		/* Currently this just means: "what IFF flags are valid?".
		 * This is needed because we never checked for invalid flags on
2896 2897 2898
		 * TUNSETIFF.
		 */
		return put_user(IFF_TUN | IFF_TAP | TUN_FEATURES,
E
Eric W. Biederman 已提交
2899
				(unsigned int __user*)argp);
2900
	} else if (cmd == TUNSETQUEUE) {
2901
		return tun_set_queue(file, &ifr);
2902 2903 2904 2905 2906
	} 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 已提交
2907

J
Jason Wang 已提交
2908
	ret = 0;
2909 2910
	rtnl_lock();

2911
	tun = tun_get(tfile);
2912 2913 2914 2915 2916
	if (cmd == TUNSETIFF) {
		ret = -EEXIST;
		if (tun)
			goto unlock;

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

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

2921 2922
		if (ret)
			goto unlock;
L
Linus Torvalds 已提交
2923

2924
		if (copy_to_user(argp, &ifr, ifreq_len))
2925 2926
			ret = -EFAULT;
		goto unlock;
L
Linus Torvalds 已提交
2927
	}
2928 2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939 2940
	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 已提交
2941

2942
	ret = -EBADFD;
L
Linus Torvalds 已提交
2943
	if (!tun)
2944
		goto unlock;
L
Linus Torvalds 已提交
2945

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

E
Eric W. Biederman 已提交
2948
	ret = 0;
L
Linus Torvalds 已提交
2949
	switch (cmd) {
2950
	case TUNGETIFF:
R
Rami Rosen 已提交
2951
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
2952

2953 2954
		if (tfile->detached)
			ifr.ifr_flags |= IFF_DETACH_QUEUE;
2955 2956
		if (!tfile->socket.sk->sk_filter)
			ifr.ifr_flags |= IFF_NOFILTER;
2957

2958
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
2959
			ret = -EFAULT;
2960 2961
		break;

L
Linus Torvalds 已提交
2962 2963 2964
	case TUNSETNOCSUM:
		/* Disable/Enable checksum */

2965 2966
		/* [unimplemented] */
		tun_debug(KERN_INFO, tun, "ignored: set checksum %s\n",
2967
			  arg ? "disabled" : "enabled");
L
Linus Torvalds 已提交
2968 2969 2970
		break;

	case TUNSETPERSIST:
J
Jason Wang 已提交
2971 2972 2973
		/* Disable/Enable persist mode. Keep an extra reference to the
		 * module to prevent the module being unprobed.
		 */
M
Michael S. Tsirkin 已提交
2974 2975
		if (arg && !(tun->flags & IFF_PERSIST)) {
			tun->flags |= IFF_PERSIST;
J
Jason Wang 已提交
2976
			__module_get(THIS_MODULE);
2977
			do_notify = true;
J
Jason Wang 已提交
2978
		}
M
Michael S. Tsirkin 已提交
2979 2980
		if (!arg && (tun->flags & IFF_PERSIST)) {
			tun->flags &= ~IFF_PERSIST;
J
Jason Wang 已提交
2981
			module_put(THIS_MODULE);
2982
			do_notify = true;
J
Jason Wang 已提交
2983
		}
L
Linus Torvalds 已提交
2984

2985 2986
		tun_debug(KERN_INFO, tun, "persist %s\n",
			  arg ? "enabled" : "disabled");
L
Linus Torvalds 已提交
2987 2988 2989 2990
		break;

	case TUNSETOWNER:
		/* Set owner of the device */
2991 2992 2993 2994 2995 2996
		owner = make_kuid(current_user_ns(), arg);
		if (!uid_valid(owner)) {
			ret = -EINVAL;
			break;
		}
		tun->owner = owner;
2997
		do_notify = true;
2998
		tun_debug(KERN_INFO, tun, "owner set to %u\n",
2999
			  from_kuid(&init_user_ns, tun->owner));
L
Linus Torvalds 已提交
3000 3001
		break;

3002 3003
	case TUNSETGROUP:
		/* Set group of the device */
3004 3005 3006 3007 3008 3009
		group = make_kgid(current_user_ns(), arg);
		if (!gid_valid(group)) {
			ret = -EINVAL;
			break;
		}
		tun->group = group;
3010
		do_notify = true;
3011
		tun_debug(KERN_INFO, tun, "group set to %u\n",
3012
			  from_kgid(&init_user_ns, tun->group));
3013 3014
		break;

3015 3016 3017
	case TUNSETLINK:
		/* Only allow setting the type when the interface is down */
		if (tun->dev->flags & IFF_UP) {
3018 3019
			tun_debug(KERN_INFO, tun,
				  "Linktype set failed because interface is up\n");
3020
			ret = -EBUSY;
3021 3022
		} else {
			tun->dev->type = (int) arg;
3023 3024
			tun_debug(KERN_INFO, tun, "linktype set to %d\n",
				  tun->dev->type);
3025
			ret = 0;
3026
		}
E
Eric W. Biederman 已提交
3027
		break;
3028

L
Linus Torvalds 已提交
3029 3030 3031 3032 3033
#ifdef TUN_DEBUG
	case TUNSETDEBUG:
		tun->debug = arg;
		break;
#endif
3034
	case TUNSETOFFLOAD:
3035
		ret = set_offload(tun, arg);
E
Eric W. Biederman 已提交
3036
		break;
3037

3038 3039
	case TUNSETTXFILTER:
		/* Can be set only for TAPs */
E
Eric W. Biederman 已提交
3040
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3041
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
E
Eric W. Biederman 已提交
3042
			break;
H
Harvey Harrison 已提交
3043
		ret = update_filter(&tun->txflt, (void __user *)arg);
E
Eric W. Biederman 已提交
3044
		break;
L
Linus Torvalds 已提交
3045 3046

	case SIOCGIFHWADDR:
3047
		/* Get hw address */
3048 3049
		memcpy(ifr.ifr_hwaddr.sa_data, tun->dev->dev_addr, ETH_ALEN);
		ifr.ifr_hwaddr.sa_family = tun->dev->type;
3050
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
3051 3052
			ret = -EFAULT;
		break;
L
Linus Torvalds 已提交
3053 3054

	case SIOCSIFHWADDR:
3055
		/* Set hw address */
3056 3057
		tun_debug(KERN_DEBUG, tun, "set hw address: %pM\n",
			  ifr.ifr_hwaddr.sa_data);
3058 3059

		ret = dev_set_mac_address(tun->dev, &ifr.ifr_hwaddr);
E
Eric W. Biederman 已提交
3060
		break;
3061 3062

	case TUNGETSNDBUF:
J
Jason Wang 已提交
3063
		sndbuf = tfile->socket.sk->sk_sndbuf;
3064 3065 3066 3067 3068 3069 3070 3071 3072
		if (copy_to_user(argp, &sndbuf, sizeof(sndbuf)))
			ret = -EFAULT;
		break;

	case TUNSETSNDBUF:
		if (copy_from_user(&sndbuf, argp, sizeof(sndbuf))) {
			ret = -EFAULT;
			break;
		}
3073 3074 3075 3076
		if (sndbuf <= 0) {
			ret = -EINVAL;
			break;
		}
3077

J
Jason Wang 已提交
3078 3079
		tun->sndbuf = sndbuf;
		tun_set_sndbuf(tun);
3080 3081
		break;

3082 3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094 3095 3096 3097 3098 3099 3100
	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 已提交
3101 3102 3103 3104 3105 3106 3107 3108 3109 3110 3111 3112 3113 3114 3115 3116 3117
	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;

3118 3119 3120 3121 3122 3123 3124 3125
	case TUNGETVNETBE:
		ret = tun_get_vnet_be(tun, argp);
		break;

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

M
Michael S. Tsirkin 已提交
3126 3127 3128
	case TUNATTACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3129
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
3130 3131
			break;
		ret = -EFAULT;
J
Jason Wang 已提交
3132
		if (copy_from_user(&tun->fprog, argp, sizeof(tun->fprog)))
M
Michael S. Tsirkin 已提交
3133 3134
			break;

J
Jason Wang 已提交
3135
		ret = tun_attach_filter(tun);
M
Michael S. Tsirkin 已提交
3136 3137 3138 3139 3140
		break;

	case TUNDETACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3141
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
3142
			break;
J
Jason Wang 已提交
3143 3144
		ret = 0;
		tun_detach_filter(tun, tun->numqueues);
M
Michael S. Tsirkin 已提交
3145 3146
		break;

P
Pavel Emelyanov 已提交
3147 3148
	case TUNGETFILTER:
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3149
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
P
Pavel Emelyanov 已提交
3150 3151 3152 3153 3154 3155 3156
			break;
		ret = -EFAULT;
		if (copy_to_user(argp, &tun->fprog, sizeof(tun->fprog)))
			break;
		ret = 0;
		break;

3157
	case TUNSETSTEERINGEBPF:
3158
		ret = tun_set_ebpf(tun, &tun->steering_prog, argp);
3159 3160
		break;

3161 3162 3163 3164
	case TUNSETFILTEREBPF:
		ret = tun_set_ebpf(tun, &tun->filter_prog, argp);
		break;

L
Linus Torvalds 已提交
3165
	default:
E
Eric W. Biederman 已提交
3166 3167
		ret = -EINVAL;
		break;
3168
	}
L
Linus Torvalds 已提交
3169

3170 3171 3172
	if (do_notify)
		netdev_state_change(tun->dev);

3173 3174 3175 3176
unlock:
	rtnl_unlock();
	if (tun)
		tun_put(tun);
E
Eric W. Biederman 已提交
3177
	return ret;
L
Linus Torvalds 已提交
3178 3179
}

3180 3181 3182 3183 3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196 3197 3198 3199 3200 3201 3202 3203 3204 3205 3206 3207 3208 3209 3210 3211 3212 3213 3214
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 已提交
3215 3216
static int tun_chr_fasync(int fd, struct file *file, int on)
{
J
Jason Wang 已提交
3217
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
3218 3219
	int ret;

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

L
Linus Torvalds 已提交
3223
	if (on) {
3224
		__f_setown(file, task_pid(current), PIDTYPE_PID, 0);
J
Jason Wang 已提交
3225
		tfile->flags |= TUN_FASYNC;
3226
	} else
J
Jason Wang 已提交
3227
		tfile->flags &= ~TUN_FASYNC;
J
Jonathan Corbet 已提交
3228 3229 3230
	ret = 0;
out:
	return ret;
L
Linus Torvalds 已提交
3231 3232 3233 3234
}

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

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

3240
	tfile = (struct tun_file *)sk_alloc(net, AF_UNSPEC, GFP_KERNEL,
3241
					    &tun_proto, 0);
E
Eric W. Biederman 已提交
3242 3243
	if (!tfile)
		return -ENOMEM;
3244 3245 3246 3247 3248
	if (ptr_ring_init(&tfile->tx_ring, 0, GFP_KERNEL)) {
		sk_free(&tfile->sk);
		return -ENOMEM;
	}

3249
	RCU_INIT_POINTER(tfile->tun, NULL);
J
Jason Wang 已提交
3250
	tfile->flags = 0;
3251
	tfile->ifindex = 0;
J
Jason Wang 已提交
3252 3253

	init_waitqueue_head(&tfile->wq.wait);
3254
	RCU_INIT_POINTER(tfile->socket.wq, &tfile->wq);
J
Jason Wang 已提交
3255 3256 3257 3258 3259 3260 3261 3262 3263

	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 已提交
3264
	file->private_data = tfile;
J
Jason Wang 已提交
3265
	INIT_LIST_HEAD(&tfile->next);
J
Jason Wang 已提交
3266

3267 3268
	sock_set_flag(&tfile->sk, SOCK_ZEROCOPY);

L
Linus Torvalds 已提交
3269 3270 3271 3272 3273
	return 0;
}

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

J
Jason Wang 已提交
3276
	tun_detach(tfile, true);
L
Linus Torvalds 已提交
3277 3278 3279 3280

	return 0;
}

3281
#ifdef CONFIG_PROC_FS
3282
static void tun_chr_show_fdinfo(struct seq_file *m, struct file *file)
3283
{
3284
	struct tun_file *tfile = file->private_data;
3285 3286 3287 3288 3289 3290
	struct tun_struct *tun;
	struct ifreq ifr;

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

	rtnl_lock();
3291
	tun = tun_get(tfile);
3292 3293 3294 3295 3296 3297 3298
	if (tun)
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
	rtnl_unlock();

	if (tun)
		tun_put(tun);

3299
	seq_printf(m, "iff:\t%s\n", ifr.ifr_name);
3300 3301 3302
}
#endif

3303
static const struct file_operations tun_fops = {
3304
	.owner	= THIS_MODULE,
L
Linus Torvalds 已提交
3305
	.llseek = no_llseek,
3306
	.read_iter  = tun_chr_read_iter,
3307
	.write_iter = tun_chr_write_iter,
L
Linus Torvalds 已提交
3308
	.poll	= tun_chr_poll,
3309 3310 3311 3312
	.unlocked_ioctl	= tun_chr_ioctl,
#ifdef CONFIG_COMPAT
	.compat_ioctl = tun_chr_compat_ioctl,
#endif
L
Linus Torvalds 已提交
3313 3314
	.open	= tun_chr_open,
	.release = tun_chr_close,
3315 3316 3317 3318
	.fasync = tun_chr_fasync,
#ifdef CONFIG_PROC_FS
	.show_fdinfo = tun_chr_show_fdinfo,
#endif
L
Linus Torvalds 已提交
3319 3320 3321 3322 3323
};

static struct miscdevice tun_miscdev = {
	.minor = TUN_MINOR,
	.name = "tun",
3324
	.nodename = "net/tun",
L
Linus Torvalds 已提交
3325 3326 3327 3328 3329
	.fops = &tun_fops,
};

/* ethtool interface */

3330 3331 3332 3333 3334 3335 3336 3337 3338 3339
static int tun_get_link_ksettings(struct net_device *dev,
				  struct ethtool_link_ksettings *cmd)
{
	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;
L
Linus Torvalds 已提交
3340 3341 3342 3343 3344 3345 3346
	return 0;
}

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

3347 3348
	strlcpy(info->driver, DRV_NAME, sizeof(info->driver));
	strlcpy(info->version, DRV_VERSION, sizeof(info->version));
L
Linus Torvalds 已提交
3349 3350

	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
3351
	case IFF_TUN:
3352
		strlcpy(info->bus_info, "tun", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3353
		break;
M
Michael S. Tsirkin 已提交
3354
	case IFF_TAP:
3355
		strlcpy(info->bus_info, "tap", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3356 3357 3358 3359 3360 3361 3362 3363 3364 3365 3366 3367 3368 3369 3370 3371 3372 3373 3374 3375 3376 3377
		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 已提交
3378 3379 3380 3381 3382 3383 3384 3385 3386 3387 3388 3389 3390 3391 3392 3393 3394 3395 3396 3397 3398 3399 3400
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;
}

3401
static const struct ethtool_ops tun_ethtool_ops = {
L
Linus Torvalds 已提交
3402 3403 3404
	.get_drvinfo	= tun_get_drvinfo,
	.get_msglevel	= tun_get_msglevel,
	.set_msglevel	= tun_set_msglevel,
3405
	.get_link	= ethtool_op_get_link,
3406
	.get_ts_info	= ethtool_op_get_ts_info,
J
Jason Wang 已提交
3407 3408
	.get_coalesce   = tun_get_coalesce,
	.set_coalesce   = tun_set_coalesce,
3409
	.get_link_ksettings = tun_get_link_ksettings,
L
Linus Torvalds 已提交
3410 3411
};

J
Jason Wang 已提交
3412 3413 3414 3415
static int tun_queue_resize(struct tun_struct *tun)
{
	struct net_device *dev = tun->dev;
	struct tun_file *tfile;
3416
	struct ptr_ring **rings;
J
Jason Wang 已提交
3417 3418 3419
	int n = tun->numqueues + tun->numdisabled;
	int ret, i;

3420 3421
	rings = kmalloc_array(n, sizeof(*rings), GFP_KERNEL);
	if (!rings)
J
Jason Wang 已提交
3422 3423 3424 3425
		return -ENOMEM;

	for (i = 0; i < tun->numqueues; i++) {
		tfile = rtnl_dereference(tun->tfiles[i]);
3426
		rings[i] = &tfile->tx_ring;
J
Jason Wang 已提交
3427 3428
	}
	list_for_each_entry(tfile, &tun->disabled, next)
3429
		rings[i++] = &tfile->tx_ring;
J
Jason Wang 已提交
3430

3431 3432
	ret = ptr_ring_resize_multiple(rings, n,
				       dev->tx_queue_len, GFP_KERNEL,
J
Jason Wang 已提交
3433
				       tun_ptr_free);
J
Jason Wang 已提交
3434

3435
	kfree(rings);
J
Jason Wang 已提交
3436 3437 3438 3439 3440 3441 3442 3443 3444
	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);

3445 3446 3447
	if (dev->rtnl_link_ops != &tun_link_ops)
		return NOTIFY_DONE;

J
Jason Wang 已提交
3448 3449 3450 3451 3452 3453 3454 3455 3456 3457 3458 3459 3460 3461 3462
	switch (event) {
	case NETDEV_CHANGE_TX_QUEUE_LEN:
		if (tun_queue_resize(tun))
			return NOTIFY_BAD;
		break;
	default:
		break;
	}

	return NOTIFY_DONE;
}

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

L
Linus Torvalds 已提交
3464 3465 3466 3467
static int __init tun_init(void)
{
	int ret = 0;

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

3470
	ret = rtnl_link_register(&tun_link_ops);
3471
	if (ret) {
3472
		pr_err("Can't register link_ops\n");
3473
		goto err_linkops;
3474 3475
	}

L
Linus Torvalds 已提交
3476
	ret = misc_register(&tun_miscdev);
3477
	if (ret) {
3478
		pr_err("Can't register misc device %d\n", TUN_MINOR);
3479 3480
		goto err_misc;
	}
J
Jason Wang 已提交
3481

3482 3483 3484 3485 3486 3487
	ret = register_netdevice_notifier(&tun_notifier_block);
	if (ret) {
		pr_err("Can't register netdevice notifier\n");
		goto err_notifier;
	}

3488
	return  0;
3489 3490 3491

err_notifier:
	misc_deregister(&tun_miscdev);
3492
err_misc:
3493 3494
	rtnl_link_unregister(&tun_link_ops);
err_linkops:
L
Linus Torvalds 已提交
3495 3496 3497 3498 3499
	return ret;
}

static void tun_cleanup(void)
{
3500
	misc_deregister(&tun_miscdev);
3501
	rtnl_link_unregister(&tun_link_ops);
J
Jason Wang 已提交
3502
	unregister_netdevice_notifier(&tun_notifier_block);
L
Linus Torvalds 已提交
3503 3504
}

3505 3506 3507 3508 3509 3510
/* 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)
{
3511
	struct tun_file *tfile;
3512 3513
	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
3514 3515
	tfile = file->private_data;
	if (!tfile)
3516
		return ERR_PTR(-EBADFD);
J
Jason Wang 已提交
3517
	return &tfile->socket;
3518 3519 3520
}
EXPORT_SYMBOL_GPL(tun_get_socket);

3521
struct ptr_ring *tun_get_tx_ring(struct file *file)
J
Jason Wang 已提交
3522 3523 3524 3525 3526 3527 3528 3529
{
	struct tun_file *tfile;

	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
	tfile = file->private_data;
	if (!tfile)
		return ERR_PTR(-EBADFD);
3530
	return &tfile->tx_ring;
J
Jason Wang 已提交
3531
}
3532
EXPORT_SYMBOL_GPL(tun_get_tx_ring);
J
Jason Wang 已提交
3533

L
Linus Torvalds 已提交
3534 3535 3536 3537 3538 3539
module_init(tun_init);
module_exit(tun_cleanup);
MODULE_DESCRIPTION(DRV_DESCRIPTION);
MODULE_AUTHOR(DRV_COPYRIGHT);
MODULE_LICENSE("GPL");
MODULE_ALIAS_MISCDEV(TUN_MINOR);
3540
MODULE_ALIAS("devname:net/tun");