tun.c 77.7 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 <linux/seq_file.h>
H
Herbert Xu 已提交
74
#include <linux/uio.h>
J
Jason Wang 已提交
75
#include <linux/skb_array.h>
J
Jason Wang 已提交
76 77
#include <linux/bpf.h>
#include <linux/bpf_trace.h>
78
#include <linux/mutex.h>
L
Linus Torvalds 已提交
79

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

82 83 84
/* Uncomment to enable debugging */
/* #define TUN_DEBUG 1 */

L
Linus Torvalds 已提交
85 86
#ifdef TUN_DEBUG
static int debug;
87

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

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

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

#define TUN_FEATURES (IFF_NO_PI | IFF_ONE_QUEUE | IFF_VNET_HDR | \
125 126
		      IFF_MULTI_QUEUE | IFF_NAPI | IFF_NAPI_FRAGS)

127 128
#define GOODCOPY_LEN 128

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

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

J
Jason Wang 已提交
141 142
#define TUN_FLOW_EXPIRE (3 * HZ)

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

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

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

#define TUN_NUM_FLOW_ENTRIES 1024

199 200 201 202 203
struct tun_steering_prog {
	struct rcu_head rcu;
	struct bpf_prog *prog;
};

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

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

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

J
Jason Wang 已提交
244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261
bool tun_is_xdp_buff(void *ptr)
{
	return (unsigned long)ptr & TUN_XDP_FLAG;
}
EXPORT_SYMBOL(tun_is_xdp_buff);

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

262 263 264 265 266 267 268 269 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
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)
{
305
	tfile->napi_enabled = napi_en;
306 307 308 309
	if (napi_en) {
		netif_napi_add(tun->dev, &tfile->napi, tun_napi_poll,
			       NAPI_POLL_WEIGHT);
		napi_enable(&tfile->napi);
310
		mutex_init(&tfile->napi_mutex);
311 312 313 314 315
	}
}

static void tun_napi_disable(struct tun_struct *tun, struct tun_file *tfile)
{
316
	if (tfile->napi_enabled)
317 318 319 320 321
		napi_disable(&tfile->napi);
}

static void tun_napi_del(struct tun_struct *tun, struct tun_file *tfile)
{
322
	if (tfile->napi_enabled)
323 324 325
		netif_napi_del(&tfile->napi);
}

326 327 328 329 330
static bool tun_napi_frags_enabled(const struct tun_struct *tun)
{
	return READ_ONCE(tun->flags) & IFF_NAPI_FRAGS;
}

331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378
#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 */

379 380
static inline bool tun_is_little_endian(struct tun_struct *tun)
{
381
	return tun->flags & TUN_VNET_LE ||
382
		tun_legacy_is_little_endian(tun);
383 384
}

385 386
static inline u16 tun16_to_cpu(struct tun_struct *tun, __virtio16 val)
{
387
	return __virtio16_to_cpu(tun_is_little_endian(tun), val);
388 389 390 391
}

static inline __virtio16 cpu_to_tun16(struct tun_struct *tun, u16 val)
{
392
	return __cpu_to_virtio16(tun_is_little_endian(tun), val);
393 394
}

J
Jason Wang 已提交
395 396 397 398 399 400 401 402 403
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;

404
	hlist_for_each_entry_rcu(e, head, hash_link) {
J
Jason Wang 已提交
405 406 407 408 409 410 411 412 413 414
		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)
{
415 416
	struct tun_flow_entry *e = kmalloc(sizeof(*e), GFP_ATOMIC);

J
Jason Wang 已提交
417 418 419 420 421
	if (e) {
		tun_debug(KERN_INFO, tun, "create flow: hash %u index %u\n",
			  rxhash, queue_index);
		e->updated = jiffies;
		e->rxhash = rxhash;
422
		e->rps_rxhash = 0;
J
Jason Wang 已提交
423 424 425
		e->queue_index = queue_index;
		e->tun = tun;
		hlist_add_head_rcu(&e->hash_link, head);
426
		++tun->flow_count;
J
Jason Wang 已提交
427 428 429 430 431 432 433 434 435
	}
	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);
436
	kfree_rcu(e, rcu);
437
	--tun->flow_count;
J
Jason Wang 已提交
438 439 440 441 442 443 444 445 446
}

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;
447
		struct hlist_node *n;
J
Jason Wang 已提交
448

449
		hlist_for_each_entry_safe(e, n, &tun->flows[i], hash_link)
J
Jason Wang 已提交
450 451 452 453 454 455 456 457 458 459 460 461
			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;
462
		struct hlist_node *n;
J
Jason Wang 已提交
463

464
		hlist_for_each_entry_safe(e, n, &tun->flows[i], hash_link) {
J
Jason Wang 已提交
465 466 467 468 469 470 471
			if (e->queue_index == queue_index)
				tun_flow_delete(tun, e);
		}
	}
	spin_unlock_bh(&tun->lock);
}

472
static void tun_flow_cleanup(struct timer_list *t)
J
Jason Wang 已提交
473
{
474
	struct tun_struct *tun = from_timer(tun, t, flow_gc_timer);
J
Jason Wang 已提交
475 476 477 478 479 480 481
	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");

482
	spin_lock(&tun->lock);
J
Jason Wang 已提交
483 484
	for (i = 0; i < TUN_NUM_FLOW_ENTRIES; i++) {
		struct tun_flow_entry *e;
485
		struct hlist_node *n;
J
Jason Wang 已提交
486

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

J
Jason Wang 已提交
490
			this_timer = e->updated + delay;
491
			if (time_before_eq(this_timer, jiffies)) {
J
Jason Wang 已提交
492
				tun_flow_delete(tun, e);
493 494 495 496
				continue;
			}
			count++;
			if (time_before(this_timer, next_timer))
J
Jason Wang 已提交
497 498 499 500 501 502
				next_timer = this_timer;
		}
	}

	if (count)
		mod_timer(&tun->flow_gc_timer, round_jiffies_up(next_timer));
503
	spin_unlock(&tun->lock);
J
Jason Wang 已提交
504 505
}

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

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

	rcu_read_lock();

521 522 523
	/* We may get a very small possibility of OOO during switching, not
	 * worth to optimize.*/
	if (tun->numqueues == 1 || tfile->detached)
J
Jason Wang 已提交
524 525 526 527 528 529 530
		goto unlock;

	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;
531
		sock_rps_record_flow_hash(e->rps_rxhash);
J
Jason Wang 已提交
532 533
	} else {
		spin_lock_bh(&tun->lock);
534 535
		if (!tun_flow_find(head, rxhash) &&
		    tun->flow_count < MAX_TAP_FLOWS)
J
Jason Wang 已提交
536 537 538 539 540 541 542 543 544 545 546 547
			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);
	}

unlock:
	rcu_read_unlock();
}

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

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

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

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

	return txq;
}

591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618
static u16 tun_ebpf_select_queue(struct tun_struct *tun, struct sk_buff *skb)
{
	struct tun_steering_prog *prog;
	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;
}

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

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

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

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

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

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

J
Jason Wang 已提交
652 653 654 655 656 657 658 659 660 661 662 663 664
static void tun_ptr_free(void *ptr)
{
	if (!ptr)
		return;
	if (tun_is_xdp_buff(ptr)) {
		struct xdp_buff *xdp = tun_ptr_to_xdp(ptr);

		put_page(virt_to_head_page(xdp->data));
	} else {
		__skb_array_destroy_skb(ptr);
	}
}

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

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

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

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

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

683 684 685 686 687
	if (tun && clean) {
		tun_napi_disable(tun, tfile);
		tun_napi_del(tun, tfile);
	}

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

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

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

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

	if (clean) {
715 716 717
		if (tun && tun->numqueues == 0 && tun->numdisabled == 0) {
			netif_carrier_off(tun->dev);

M
Michael S. Tsirkin 已提交
718
			if (!(tun->flags & IFF_PERSIST) &&
719
			    tun->dev->reg_state == NETREG_REGISTERED)
J
Jason Wang 已提交
720
				unregister_netdevice(tun->dev);
721
		}
J
Jesper Dangaard Brouer 已提交
722
		if (tun) {
J
Jason Wang 已提交
723
			ptr_ring_cleanup(&tfile->tx_ring, tun_ptr_free);
J
Jesper Dangaard Brouer 已提交
724 725
			xdp_rxq_info_unreg(&tfile->xdp_rxq);
		}
726
		sock_put(&tfile->sk);
J
Jason Wang 已提交
727 728 729 730 731 732 733 734 735 736 737 738 739
	}
}

static void tun_detach(struct tun_file *tfile, bool clean)
{
	rtnl_lock();
	__tun_detach(tfile, clean);
	rtnl_unlock();
}

static void tun_detach_all(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
J
Jason Wang 已提交
740
	struct tun_file *tfile, *tmp;
J
Jason Wang 已提交
741 742 743
	int i, n = tun->numqueues;

	for (i = 0; i < n; i++) {
744
		tfile = rtnl_dereference(tun->tfiles[i]);
J
Jason Wang 已提交
745
		BUG_ON(!tfile);
746
		tun_napi_disable(tun, tfile);
747
		tfile->socket.sk->sk_shutdown = RCV_SHUTDOWN;
748
		tfile->socket.sk->sk_data_ready(tfile->socket.sk);
749
		RCU_INIT_POINTER(tfile->tun, NULL);
J
Jason Wang 已提交
750 751
		--tun->numqueues;
	}
752
	list_for_each_entry(tfile, &tun->disabled, next) {
753
		tfile->socket.sk->sk_shutdown = RCV_SHUTDOWN;
754
		tfile->socket.sk->sk_data_ready(tfile->socket.sk);
755
		RCU_INIT_POINTER(tfile->tun, NULL);
756
	}
J
Jason Wang 已提交
757 758 759 760
	BUG_ON(tun->numqueues != 0);

	synchronize_net();
	for (i = 0; i < n; i++) {
761
		tfile = rtnl_dereference(tun->tfiles[i]);
762
		tun_napi_del(tun, tfile);
J
Jason Wang 已提交
763
		/* Drop read queue */
764
		tun_queue_purge(tfile);
J
Jesper Dangaard Brouer 已提交
765
		xdp_rxq_info_unreg(&tfile->xdp_rxq);
J
Jason Wang 已提交
766 767
		sock_put(&tfile->sk);
	}
J
Jason Wang 已提交
768 769
	list_for_each_entry_safe(tfile, tmp, &tun->disabled, next) {
		tun_enable_queue(tfile);
770
		tun_queue_purge(tfile);
J
Jesper Dangaard Brouer 已提交
771
		xdp_rxq_info_unreg(&tfile->xdp_rxq);
J
Jason Wang 已提交
772 773 774
		sock_put(&tfile->sk);
	}
	BUG_ON(tun->numdisabled != 0);
J
Jason Wang 已提交
775

M
Michael S. Tsirkin 已提交
776
	if (tun->flags & IFF_PERSIST)
J
Jason Wang 已提交
777
		module_put(THIS_MODULE);
J
Jason Wang 已提交
778 779
}

780 781
static int tun_attach(struct tun_struct *tun, struct file *file,
		      bool skip_filter, bool napi)
782
{
E
Eric W. Biederman 已提交
783
	struct tun_file *tfile = file->private_data;
J
Jason Wang 已提交
784
	struct net_device *dev = tun->dev;
785
	int err;
786

787 788 789 790
	err = security_tun_dev_attach(tfile->socket.sk, tun->security);
	if (err < 0)
		goto out;

791
	err = -EINVAL;
792
	if (rtnl_dereference(tfile->tun) && !tfile->detached)
793 794 795
		goto out;

	err = -EBUSY;
M
Michael S. Tsirkin 已提交
796
	if (!(tun->flags & IFF_MULTI_QUEUE) && tun->numqueues == 1)
J
Jason Wang 已提交
797 798 799
		goto out;

	err = -E2BIG;
J
Jason Wang 已提交
800 801
	if (!tfile->detached &&
	    tun->numqueues + tun->numdisabled == MAX_TAP_QUEUES)
802 803 804
		goto out;

	err = 0;
J
Jason Wang 已提交
805

S
stephen hemminger 已提交
806
	/* Re-attach the filter to persist device */
807
	if (!skip_filter && (tun->filter_attached == true)) {
808 809 810
		lock_sock(tfile->socket.sk);
		err = sk_attach_filter(&tun->fprog, tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
811 812 813
		if (!err)
			goto out;
	}
J
Jason Wang 已提交
814 815

	if (!tfile->detached &&
816
	    ptr_ring_init(&tfile->tx_ring, dev->tx_queue_len, GFP_KERNEL)) {
J
Jason Wang 已提交
817 818 819 820
		err = -ENOMEM;
		goto out;
	}

J
Jason Wang 已提交
821
	tfile->queue_index = tun->numqueues;
822
	tfile->socket.sk->sk_shutdown &= ~RCV_SHUTDOWN;
J
Jesper Dangaard Brouer 已提交
823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838

	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;
		err = 0;
	}

839
	rcu_assign_pointer(tfile->tun, tun);
J
Jason Wang 已提交
840 841
	rcu_assign_pointer(tun->tfiles[tun->numqueues], tfile);
	tun->numqueues++;
842

843
	if (tfile->detached) {
J
Jason Wang 已提交
844
		tun_enable_queue(tfile);
845
	} else {
J
Jason Wang 已提交
846
		sock_hold(&tfile->sk);
847 848
		tun_napi_init(tun, tfile, napi);
	}
J
Jason Wang 已提交
849

J
Jason Wang 已提交
850
	tun_set_real_num_queues(tun);
851

J
Jason Wang 已提交
852 853 854 855 856 857
	/* device is allowed to go away first, so no need to hold extra
	 * refcnt.
	 */

out:
	return err;
E
Eric W. Biederman 已提交
858 859
}

860
static struct tun_struct *tun_get(struct tun_file *tfile)
E
Eric W. Biederman 已提交
861
{
862
	struct tun_struct *tun;
863

864 865 866 867 868
	rcu_read_lock();
	tun = rcu_dereference(tfile->tun);
	if (tun)
		dev_hold(tun->dev);
	rcu_read_unlock();
869 870

	return tun;
E
Eric W. Biederman 已提交
871 872 873 874
}

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

878
/* TAP filtering */
879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906
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;
907 908 909
	addr = memdup_user(arg + sizeof(uf), alen);
	if (IS_ERR(addr))
		return PTR_ERR(addr);
910 911 912 913 914 915 916 917 918 919 920 921 922

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

923 924
	/* Remaining multicast addresses are hashed,
	 * unicast will leave the filter disabled. */
925
	memset(filter->mask, 0, sizeof(filter->mask));
926 927 928
	for (; n < uf.count; n++) {
		if (!is_multicast_ether_addr(addr[n].u)) {
			err = 0; /* no filter */
929
			goto free_addr;
930
		}
931
		addr_hash_set(filter->mask, addr[n].u);
932
	}
933 934 935 936 937 938 939 940 941 942 943 944

	/* 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;
945
free_addr:
946 947 948 949 950 951 952 953 954 955 956 957 958 959
	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++)
960
		if (ether_addr_equal(eh->h_dest, filter->addr[i]))
961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981
			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 已提交
982 983
/* Network device part of the driver */

984
static const struct ethtool_ops tun_ethtool_ops;
L
Linus Torvalds 已提交
985

986 987 988
/* Net device detach from fd. */
static void tun_net_uninit(struct net_device *dev)
{
J
Jason Wang 已提交
989
	tun_detach_all(dev);
990 991
}

L
Linus Torvalds 已提交
992 993 994
/* Net device open. */
static int tun_net_open(struct net_device *dev)
{
995 996 997
	struct tun_struct *tun = netdev_priv(dev);
	int i;

J
Jason Wang 已提交
998
	netif_tx_start_all_queues(dev);
999 1000 1001 1002 1003 1004 1005 1006

	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 已提交
1007 1008 1009 1010 1011 1012
	return 0;
}

/* Net device close. */
static int tun_net_close(struct net_device *dev)
{
J
Jason Wang 已提交
1013
	netif_tx_stop_all_queues(dev);
L
Linus Torvalds 已提交
1014 1015 1016 1017
	return 0;
}

/* Net device start xmit */
1018
static void tun_automq_xmit(struct tun_struct *tun, struct sk_buff *skb)
L
Linus Torvalds 已提交
1019
{
1020
#ifdef CONFIG_RPS
1021
	if (tun->numqueues == 1 && static_key_false(&rps_needed)) {
1022 1023 1024 1025 1026
		/* 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 已提交
1027
		rxhash = __skb_get_hash_symmetric(skb);
1028 1029 1030 1031 1032 1033 1034 1035
		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);
		}
	}
1036
#endif
1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049
}

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

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

	/* Drop packet if interface is not attached */
1050
	if (txq >= tun->numqueues)
1051 1052 1053 1054
		goto drop;

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

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

J
Jason Wang 已提交
1058 1059
	BUG_ON(!tfile);

1060 1061 1062 1063 1064 1065
	/* 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 已提交
1066 1067
	if (tfile->socket.sk->sk_filter &&
	    sk_filter(tfile->socket.sk, skb))
M
Michael S. Tsirkin 已提交
1068 1069
		goto drop;

W
Willem de Bruijn 已提交
1070
	if (unlikely(skb_orphan_frags_rx(skb, GFP_ATOMIC)))
1071 1072
		goto drop;

1073
	skb_tx_timestamp(skb);
1074

M
Michael S. Tsirkin 已提交
1075
	/* Orphan the skb - required as we might hang on to it
1076 1077
	 * for indefinite time.
	 */
M
Michael S. Tsirkin 已提交
1078 1079
	skb_orphan(skb);

1080 1081
	nf_reset(skb);

1082
	if (ptr_ring_produce(&tfile->tx_ring, skb))
J
Jason Wang 已提交
1083
		goto drop;
L
Linus Torvalds 已提交
1084 1085

	/* Notify and wake up reader process */
J
Jason Wang 已提交
1086 1087
	if (tfile->flags & TUN_FASYNC)
		kill_fasync(&tfile->fasync, SIGIO, POLL_IN);
1088
	tfile->socket.sk->sk_data_ready(tfile->socket.sk);
1089 1090

	rcu_read_unlock();
1091
	return NETDEV_TX_OK;
L
Linus Torvalds 已提交
1092 1093

drop:
1094
	this_cpu_inc(tun->pcpu_stats->tx_dropped);
1095
	skb_tx_error(skb);
L
Linus Torvalds 已提交
1096
	kfree_skb(skb);
1097
	rcu_read_unlock();
1098
	return NET_XMIT_DROP;
L
Linus Torvalds 已提交
1099 1100
}

1101
static void tun_net_mclist(struct net_device *dev)
L
Linus Torvalds 已提交
1102
{
1103 1104 1105 1106 1107
	/*
	 * 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 已提交
1108 1109
}

1110 1111
static netdev_features_t tun_net_fix_features(struct net_device *dev,
	netdev_features_t features)
1112 1113 1114 1115 1116
{
	struct tun_struct *tun = netdev_priv(dev);

	return (features & tun->set_features) | (features & ~TUN_USER_FEATURES);
}
1117 1118 1119 1120 1121 1122 1123
#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
1124 1125 1126
	 * 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.
1127 1128 1129
	 * 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
1130
	 * If NAPI is enabled, however, we need to schedule polling for all
1131 1132
	 * queues unless we are using napi_gro_frags(), which we call in
	 * process context and not in NAPI context.
1133
	 */
1134 1135 1136 1137 1138 1139
	struct tun_struct *tun = netdev_priv(dev);

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

1140 1141 1142
		if (tun_napi_frags_enabled(tun))
			return;

1143 1144 1145
		rcu_read_lock();
		for (i = 0; i < tun->numqueues; i++) {
			tfile = rcu_dereference(tun->tfiles[i]);
1146 1147
			if (tfile->napi_enabled)
				napi_schedule(&tfile->napi);
1148 1149 1150
		}
		rcu_read_unlock();
	}
1151 1152 1153
	return;
}
#endif
1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164

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

1165
static void
1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200
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 已提交
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
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;
}

1227
static int tun_xdp(struct net_device *dev, struct netdev_bpf *xdp)
J
Jason Wang 已提交
1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240
{
	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 已提交
1241
static const struct net_device_ops tun_netdev_ops = {
1242
	.ndo_uninit		= tun_net_uninit,
S
Stephen Hemminger 已提交
1243 1244
	.ndo_open		= tun_net_open,
	.ndo_stop		= tun_net_close,
1245
	.ndo_start_xmit		= tun_net_xmit,
1246
	.ndo_fix_features	= tun_net_fix_features,
J
Jason Wang 已提交
1247
	.ndo_select_queue	= tun_select_queue,
1248 1249 1250
#ifdef CONFIG_NET_POLL_CONTROLLER
	.ndo_poll_controller	= tun_poll_controller,
#endif
1251
	.ndo_set_rx_headroom	= tun_set_headroom,
1252
	.ndo_get_stats64	= tun_net_get_stats64,
S
Stephen Hemminger 已提交
1253 1254
};

J
Jason Wang 已提交
1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315
static int tun_xdp_xmit(struct net_device *dev, struct xdp_buff *xdp)
{
	struct tun_struct *tun = netdev_priv(dev);
	struct xdp_buff *buff = xdp->data_hard_start;
	int headroom = xdp->data - xdp->data_hard_start;
	struct tun_file *tfile;
	u32 numqueues;
	int ret = 0;

	/* Assure headroom is available and buff is properly aligned */
	if (unlikely(headroom < sizeof(*xdp) || tun_is_xdp_buff(xdp)))
		return -ENOSPC;

	*buff = *xdp;

	rcu_read_lock();

	numqueues = READ_ONCE(tun->numqueues);
	if (!numqueues) {
		ret = -ENOSPC;
		goto out;
	}

	tfile = rcu_dereference(tun->tfiles[smp_processor_id() %
					    numqueues]);
	/* Encode the XDP flag into lowest bit for consumer to differ
	 * XDP buffer from sk_buff.
	 */
	if (ptr_ring_produce(&tfile->tx_ring, tun_xdp_to_ptr(buff))) {
		this_cpu_inc(tun->pcpu_stats->tx_dropped);
		ret = -ENOSPC;
	}

out:
	rcu_read_unlock();
	return ret;
}

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 已提交
1316
static const struct net_device_ops tap_netdev_ops = {
1317
	.ndo_uninit		= tun_net_uninit,
S
Stephen Hemminger 已提交
1318 1319
	.ndo_open		= tun_net_open,
	.ndo_stop		= tun_net_close,
1320
	.ndo_start_xmit		= tun_net_xmit,
1321
	.ndo_fix_features	= tun_net_fix_features,
1322
	.ndo_set_rx_mode	= tun_net_mclist,
S
Stephen Hemminger 已提交
1323 1324
	.ndo_set_mac_address	= eth_mac_addr,
	.ndo_validate_addr	= eth_validate_addr,
J
Jason Wang 已提交
1325
	.ndo_select_queue	= tun_select_queue,
1326 1327 1328
#ifdef CONFIG_NET_POLL_CONTROLLER
	.ndo_poll_controller	= tun_poll_controller,
#endif
1329
	.ndo_features_check	= passthru_features_check,
1330
	.ndo_set_rx_headroom	= tun_set_headroom,
1331
	.ndo_get_stats64	= tun_net_get_stats64,
1332
	.ndo_bpf		= tun_xdp,
J
Jason Wang 已提交
1333 1334
	.ndo_xdp_xmit		= tun_xdp_xmit,
	.ndo_xdp_flush		= tun_xdp_flush,
S
Stephen Hemminger 已提交
1335 1336
};

1337
static void tun_flow_init(struct tun_struct *tun)
J
Jason Wang 已提交
1338 1339 1340 1341 1342 1343 1344
{
	int i;

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

	tun->ageing_time = TUN_FLOW_EXPIRE;
1345 1346 1347
	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 已提交
1348 1349 1350 1351 1352 1353 1354 1355
}

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

1356 1357 1358
#define MIN_MTU 68
#define MAX_MTU 65535

L
Linus Torvalds 已提交
1359 1360 1361 1362
/* Initialize net device. */
static void tun_net_init(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
1363

L
Linus Torvalds 已提交
1364
	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
1365
	case IFF_TUN:
S
Stephen Hemminger 已提交
1366 1367
		dev->netdev_ops = &tun_netdev_ops;

L
Linus Torvalds 已提交
1368 1369 1370 1371 1372 1373
		/* Point-to-Point TUN Device */
		dev->hard_header_len = 0;
		dev->addr_len = 0;
		dev->mtu = 1500;

		/* Zero header length */
1374
		dev->type = ARPHRD_NONE;
L
Linus Torvalds 已提交
1375 1376 1377
		dev->flags = IFF_POINTOPOINT | IFF_NOARP | IFF_MULTICAST;
		break;

M
Michael S. Tsirkin 已提交
1378
	case IFF_TAP:
K
Kusanagi Kouichi 已提交
1379
		dev->netdev_ops = &tap_netdev_ops;
L
Linus Torvalds 已提交
1380 1381
		/* Ethernet TAP Device */
		ether_setup(dev);
1382
		dev->priv_flags &= ~IFF_TX_SKB_SHARING;
1383
		dev->priv_flags |= IFF_LIVE_ADDR_CHANGE;
1384

1385
		eth_hw_addr_random(dev);
1386

L
Linus Torvalds 已提交
1387 1388
		break;
	}
1389 1390 1391

	dev->min_mtu = MIN_MTU;
	dev->max_mtu = MAX_MTU - dev->hard_header_len;
L
Linus Torvalds 已提交
1392 1393 1394 1395 1396
}

/* Character device part */

/* Poll */
J
Jason Wang 已提交
1397
static unsigned int tun_chr_poll(struct file *file, poll_table *wait)
1398
{
1399
	struct tun_file *tfile = file->private_data;
1400
	struct tun_struct *tun = tun_get(tfile);
1401
	struct sock *sk;
1402
	unsigned int mask = 0;
L
Linus Torvalds 已提交
1403 1404

	if (!tun)
1405
		return POLLERR;
L
Linus Torvalds 已提交
1406

J
Jason Wang 已提交
1407
	sk = tfile->socket.sk;
1408

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

1411
	poll_wait(file, sk_sleep(sk), wait);
1412

1413
	if (!ptr_ring_empty(&tfile->tx_ring))
L
Linus Torvalds 已提交
1414 1415
		mask |= POLLIN | POLLRDNORM;

1416 1417 1418 1419
	if (tun->dev->flags & IFF_UP &&
	    (sock_writeable(sk) ||
	     (!test_and_set_bit(SOCKWQ_ASYNC_NOSPACE, &sk->sk_socket->flags) &&
	      sock_writeable(sk))))
1420 1421
		mask |= POLLOUT | POLLWRNORM;

1422 1423 1424
	if (tun->dev->reg_state != NETREG_REGISTERED)
		mask = POLLERR;

E
Eric W. Biederman 已提交
1425
	tun_put(tun);
L
Linus Torvalds 已提交
1426 1427 1428
	return mask;
}

1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486
static struct sk_buff *tun_napi_alloc_frags(struct tun_file *tfile,
					    size_t len,
					    const struct iov_iter *it)
{
	struct sk_buff *skb;
	size_t linear;
	int err;
	int i;

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

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

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

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

	for (i = 1; i < it->nr_segs; i++) {
		size_t fragsz = it->iov[i].iov_len;
		unsigned long offset;
		struct page *page;
		void *data;

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

		local_bh_disable();
		data = napi_alloc_frag(fragsz);
		local_bh_enable();
		if (!data) {
			err = -ENOMEM;
			goto free;
		}

		page = virt_to_head_page(data);
		offset = data - page_address(page);
		skb_fill_page_desc(skb, i - 1, page, offset, fragsz);
	}

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

1487 1488
/* 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 已提交
1489
static struct sk_buff *tun_alloc_skb(struct tun_file *tfile,
1490 1491
				     size_t prepad, size_t len,
				     size_t linear, int noblock)
1492
{
J
Jason Wang 已提交
1493
	struct sock *sk = tfile->socket.sk;
1494
	struct sk_buff *skb;
1495
	int err;
1496 1497

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

1501
	skb = sock_alloc_send_pskb(sk, prepad + linear, len - linear, noblock,
1502
				   &err, 0);
1503
	if (!skb)
1504
		return ERR_PTR(err);
1505 1506 1507

	skb_reserve(skb, prepad);
	skb_put(skb, linear);
1508 1509
	skb->data_len = len - linear;
	skb->len += len - linear;
1510 1511 1512 1513

	return skb;
}

J
Jason Wang 已提交
1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549
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();
	}
}

1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571
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 已提交
1572 1573
static struct sk_buff *tun_build_skb(struct tun_struct *tun,
				     struct tun_file *tfile,
1574
				     struct iov_iter *from,
J
Jason Wang 已提交
1575
				     struct virtio_net_hdr *hdr,
J
Jason Wang 已提交
1576
				     int len, int *skb_xdp)
1577
{
1578
	struct page_frag *alloc_frag = &current->task_frag;
1579
	struct sk_buff *skb;
J
Jason Wang 已提交
1580
	struct bpf_prog *xdp_prog;
1581
	int buflen = SKB_DATA_ALIGN(sizeof(struct skb_shared_info));
J
Jason Wang 已提交
1582
	unsigned int delta = 0;
1583 1584
	char *buf;
	size_t copied;
J
Jason Wang 已提交
1585
	bool xdp_xmit = false;
1586 1587 1588 1589 1590 1591 1592 1593
	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();
1594

1595
	alloc_frag->offset = ALIGN((u64)alloc_frag->offset, SMP_CACHE_BYTES);
1596 1597 1598 1599 1600
	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,
1601
				     alloc_frag->offset + pad,
1602 1603 1604 1605
				     len, from);
	if (copied != len)
		return ERR_PTR(-EFAULT);

1606 1607 1608 1609 1610
	/* 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 已提交
1611
		*skb_xdp = 1;
J
Jason Wang 已提交
1612
	else
J
Jason Wang 已提交
1613
		*skb_xdp = 0;
J
Jason Wang 已提交
1614 1615 1616

	rcu_read_lock();
	xdp_prog = rcu_dereference(tun->xdp_prog);
J
Jason Wang 已提交
1617
	if (xdp_prog && !*skb_xdp) {
J
Jason Wang 已提交
1618 1619 1620 1621 1622
		struct xdp_buff xdp;
		void *orig_data;
		u32 act;

		xdp.data_hard_start = buf;
1623
		xdp.data = buf + pad;
1624
		xdp_set_data_meta_invalid(&xdp);
J
Jason Wang 已提交
1625
		xdp.data_end = xdp.data + len;
J
Jesper Dangaard Brouer 已提交
1626
		xdp.rxq = &tfile->xdp_rxq;
J
Jason Wang 已提交
1627 1628 1629 1630 1631 1632 1633 1634 1635 1636
		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);
			if (err)
				goto err_redirect;
1637
			rcu_read_unlock();
J
Jason Wang 已提交
1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655
			return NULL;
		case XDP_TX:
			xdp_xmit = true;
			/* fall through */
		case XDP_PASS:
			delta = orig_data - xdp.data;
			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;
		}
	}

1656
	skb = build_skb(buf, buflen);
J
Jason Wang 已提交
1657 1658
	if (!skb) {
		rcu_read_unlock();
1659
		return ERR_PTR(-ENOMEM);
J
Jason Wang 已提交
1660
	}
1661

1662
	skb_reserve(skb, pad - delta);
J
Jason Wang 已提交
1663
	skb_put(skb, len + delta);
1664 1665 1666
	get_page(alloc_frag->page);
	alloc_frag->offset += buflen;

J
Jason Wang 已提交
1667 1668 1669
	if (xdp_xmit) {
		skb->dev = tun->dev;
		generic_xdp_tx(skb, xdp_prog);
1670
		rcu_read_unlock();
J
Jason Wang 已提交
1671 1672 1673 1674 1675
		return NULL;
	}

	rcu_read_unlock();

1676
	return skb;
J
Jason Wang 已提交
1677 1678 1679 1680 1681 1682 1683

err_redirect:
	put_page(alloc_frag->page);
err_xdp:
	rcu_read_unlock();
	this_cpu_inc(tun->pcpu_stats->rx_dropped);
	return NULL;
1684 1685
}

L
Linus Torvalds 已提交
1686
/* Get packet from user space buffer */
J
Jason Wang 已提交
1687
static ssize_t tun_get_user(struct tun_struct *tun, struct tun_file *tfile,
1688
			    void *msg_control, struct iov_iter *from,
J
Jason Wang 已提交
1689
			    int noblock, bool more)
L
Linus Torvalds 已提交
1690
{
1691
	struct tun_pi pi = { 0, cpu_to_be16(ETH_P_IP) };
L
Linus Torvalds 已提交
1692
	struct sk_buff *skb;
1693
	size_t total_len = iov_iter_count(from);
1694
	size_t len = total_len, align = tun->align, linear;
1695
	struct virtio_net_hdr gso = { 0 };
1696
	struct tun_pcpu_stats *stats;
1697
	int good_linear;
1698 1699 1700
	int copylen;
	bool zerocopy = false;
	int err;
1701
	u32 rxhash = 0;
J
Jason Wang 已提交
1702
	int skb_xdp = 1;
1703
	bool frags = tun_napi_frags_enabled(tun);
L
Linus Torvalds 已提交
1704

E
Eric Dumazet 已提交
1705 1706 1707
	if (!(tun->dev->flags & IFF_UP))
		return -EIO;

M
Michael S. Tsirkin 已提交
1708
	if (!(tun->flags & IFF_NO_PI)) {
1709
		if (len < sizeof(pi))
L
Linus Torvalds 已提交
1710
			return -EINVAL;
1711
		len -= sizeof(pi);
L
Linus Torvalds 已提交
1712

1713
		if (!copy_from_iter_full(&pi, sizeof(pi), from))
L
Linus Torvalds 已提交
1714 1715 1716
			return -EFAULT;
	}

M
Michael S. Tsirkin 已提交
1717
	if (tun->flags & IFF_VNET_HDR) {
W
Willem de Bruijn 已提交
1718 1719 1720
		int vnet_hdr_sz = READ_ONCE(tun->vnet_hdr_sz);

		if (len < vnet_hdr_sz)
1721
			return -EINVAL;
W
Willem de Bruijn 已提交
1722
		len -= vnet_hdr_sz;
1723

1724
		if (!copy_from_iter_full(&gso, sizeof(gso), from))
1725 1726
			return -EFAULT;

1727
		if ((gso.flags & VIRTIO_NET_HDR_F_NEEDS_CSUM) &&
1728 1729
		    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);
1730

1731
		if (tun16_to_cpu(tun, gso.hdr_len) > len)
1732
			return -EINVAL;
W
Willem de Bruijn 已提交
1733
		iov_iter_advance(from, vnet_hdr_sz - sizeof(gso));
1734 1735
	}

M
Michael S. Tsirkin 已提交
1736
	if ((tun->flags & TUN_TYPE_MASK) == IFF_TAP) {
1737
		align += NET_IP_ALIGN;
H
Herbert Xu 已提交
1738
		if (unlikely(len < ETH_HLEN ||
1739
			     (gso.hdr_len && tun16_to_cpu(tun, gso.hdr_len) < ETH_HLEN)))
1740 1741
			return -EINVAL;
	}
1742

1743 1744
	good_linear = SKB_MAX_HEAD(align);

1745
	if (msg_control) {
1746 1747
		struct iov_iter i = *from;

1748 1749
		/* There are 256 bytes to be copied in skb, so there is
		 * enough room for skb expand head in case it is used.
1750 1751
		 * The rest of the buffer is mapped from userspace.
		 */
1752
		copylen = gso.hdr_len ? tun16_to_cpu(tun, gso.hdr_len) : GOODCOPY_LEN;
1753 1754
		if (copylen > good_linear)
			copylen = good_linear;
1755
		linear = copylen;
1756 1757
		iov_iter_advance(&i, copylen);
		if (iov_iter_npages(&i, INT_MAX) <= MAX_SKB_FRAGS)
1758 1759 1760
			zerocopy = true;
	}

1761
	if (!frags && tun_can_build_skb(tun, tfile, len, noblock, zerocopy)) {
J
Jason Wang 已提交
1762 1763 1764 1765 1766
		/* 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);
1767
		if (IS_ERR(skb)) {
1768
			this_cpu_inc(tun->pcpu_stats->rx_dropped);
1769 1770
			return PTR_ERR(skb);
		}
J
Jason Wang 已提交
1771 1772
		if (!skb)
			return total_len;
1773 1774 1775 1776 1777 1778 1779 1780
	} 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 已提交
1781

1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794
		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);
		}

1795 1796 1797
		if (IS_ERR(skb)) {
			if (PTR_ERR(skb) != -EAGAIN)
				this_cpu_inc(tun->pcpu_stats->rx_dropped);
1798 1799
			if (frags)
				mutex_unlock(&tfile->napi_mutex);
1800 1801
			return PTR_ERR(skb);
		}
1802

1803 1804 1805 1806 1807 1808 1809 1810
		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);
1811 1812 1813 1814 1815
			if (frags) {
				tfile->napi.skb = NULL;
				mutex_unlock(&tfile->napi_mutex);
			}

1816 1817
			return -EFAULT;
		}
D
Dave Jones 已提交
1818
	}
L
Linus Torvalds 已提交
1819

1820
	if (virtio_net_hdr_to_skb(skb, &gso, tun_is_little_endian(tun))) {
1821 1822
		this_cpu_inc(tun->pcpu_stats->rx_frame_errors);
		kfree_skb(skb);
1823 1824 1825 1826 1827
		if (frags) {
			tfile->napi.skb = NULL;
			mutex_unlock(&tfile->napi_mutex);
		}

1828 1829 1830
		return -EINVAL;
	}

L
Linus Torvalds 已提交
1831
	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
1832 1833
	case IFF_TUN:
		if (tun->flags & IFF_NO_PI) {
1834 1835 1836 1837
			u8 ip_version = skb->len ? (skb->data[0] >> 4) : 0;

			switch (ip_version) {
			case 4:
1838 1839
				pi.proto = htons(ETH_P_IP);
				break;
1840
			case 6:
1841 1842 1843
				pi.proto = htons(ETH_P_IPV6);
				break;
			default:
1844
				this_cpu_inc(tun->pcpu_stats->rx_dropped);
1845 1846 1847 1848 1849
				kfree_skb(skb);
				return -EINVAL;
			}
		}

1850
		skb_reset_mac_header(skb);
L
Linus Torvalds 已提交
1851
		skb->protocol = pi.proto;
1852
		skb->dev = tun->dev;
L
Linus Torvalds 已提交
1853
		break;
M
Michael S. Tsirkin 已提交
1854
	case IFF_TAP:
1855 1856
		if (!frags)
			skb->protocol = eth_type_trans(skb, tun->dev);
L
Linus Torvalds 已提交
1857
		break;
1858
	}
L
Linus Torvalds 已提交
1859

1860 1861 1862 1863
	/* 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;
1864
		skb_shinfo(skb)->tx_flags |= SKBTX_SHARED_FRAG;
1865 1866 1867
	} else if (msg_control) {
		struct ubuf_info *uarg = msg_control;
		uarg->callback(uarg, false);
1868 1869
	}

1870
	skb_reset_network_header(skb);
1871
	skb_probe_transport_header(skb, 0);
1872

J
Jason Wang 已提交
1873
	if (skb_xdp) {
J
Jason Wang 已提交
1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888
		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();
	}

1889 1890 1891 1892
	rcu_read_lock();
	if (!rcu_dereference(tun->steering_prog))
		rxhash = __skb_get_hash_symmetric(skb);
	rcu_read_unlock();
1893

1894 1895 1896 1897
	if (frags) {
		/* Exercise flow dissector code path. */
		u32 headlen = eth_get_headlen(skb->data, skb_headlen(skb));

1898
		if (unlikely(headlen > skb_headlen(skb))) {
1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909
			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);
1910
	} else if (tfile->napi_enabled) {
1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925 1926 1927
		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);
	}
1928

1929 1930 1931 1932 1933 1934
	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 已提交
1935

1936 1937 1938
	if (rxhash)
		tun_flow_update(tun, rxhash, tfile);

1939
	return total_len;
1940
}
L
Linus Torvalds 已提交
1941

1942
static ssize_t tun_chr_write_iter(struct kiocb *iocb, struct iov_iter *from)
L
Linus Torvalds 已提交
1943
{
1944
	struct file *file = iocb->ki_filp;
J
Jason Wang 已提交
1945
	struct tun_file *tfile = file->private_data;
1946
	struct tun_struct *tun = tun_get(tfile);
E
Eric W. Biederman 已提交
1947
	ssize_t result;
L
Linus Torvalds 已提交
1948 1949 1950 1951

	if (!tun)
		return -EBADFD;

J
Jason Wang 已提交
1952 1953
	result = tun_get_user(tun, tfile, NULL, from,
			      file->f_flags & O_NONBLOCK, false);
E
Eric W. Biederman 已提交
1954 1955 1956

	tun_put(tun);
	return result;
L
Linus Torvalds 已提交
1957 1958
}

J
Jason Wang 已提交
1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992
static ssize_t tun_put_user_xdp(struct tun_struct *tun,
				struct tun_file *tfile,
				struct xdp_buff *xdp,
				struct iov_iter *iter)
{
	int vnet_hdr_sz = 0;
	size_t size = xdp->data_end - xdp->data;
	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));
	}

	ret = copy_to_iter(xdp->data, size, iter) + vnet_hdr_sz;

	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 已提交
1993
/* Put packet to the user space buffer */
1994
static ssize_t tun_put_user(struct tun_struct *tun,
J
Jason Wang 已提交
1995
			    struct tun_file *tfile,
1996
			    struct sk_buff *skb,
H
Herbert Xu 已提交
1997
			    struct iov_iter *iter)
L
Linus Torvalds 已提交
1998 1999
{
	struct tun_pi pi = { 0, skb->protocol };
2000
	struct tun_pcpu_stats *stats;
H
Herbert Xu 已提交
2001
	ssize_t total;
2002
	int vlan_offset = 0;
2003
	int vlan_hlen = 0;
H
Herbert Xu 已提交
2004
	int vnet_hdr_sz = 0;
2005

2006
	if (skb_vlan_tag_present(skb))
2007
		vlan_hlen = VLAN_HLEN;
L
Linus Torvalds 已提交
2008

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

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

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

H
Herbert Xu 已提交
2018 2019
		total += sizeof(pi);
		if (iov_iter_count(iter) < total) {
L
Linus Torvalds 已提交
2020 2021 2022
			/* Packet will be striped */
			pi.flags |= TUN_PKT_STRIP;
		}
2023

H
Herbert Xu 已提交
2024
		if (copy_to_iter(&pi, sizeof(pi), iter) != sizeof(pi))
L
Linus Torvalds 已提交
2025
			return -EFAULT;
2026
	}
L
Linus Torvalds 已提交
2027

H
Herbert Xu 已提交
2028
	if (vnet_hdr_sz) {
2029
		struct virtio_net_hdr gso;
2030

H
Herbert Xu 已提交
2031
		if (iov_iter_count(iter) < vnet_hdr_sz)
2032 2033
			return -EINVAL;

2034
		if (virtio_net_hdr_from_skb(skb, &gso,
2035
					    tun_is_little_endian(tun), true)) {
2036
			struct skb_shared_info *sinfo = skb_shinfo(skb);
2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047
			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;
		}
2048

H
Herbert Xu 已提交
2049
		if (copy_to_iter(&gso, sizeof(gso), iter) != sizeof(gso))
2050
			return -EFAULT;
2051 2052

		iov_iter_advance(iter, vnet_hdr_sz - sizeof(gso));
2053 2054
	}

2055
	if (vlan_hlen) {
H
Herbert Xu 已提交
2056
		int ret;
J
Jason Wang 已提交
2057 2058 2059 2060 2061 2062
		struct {
			__be16 h_vlan_proto;
			__be16 h_vlan_TCI;
		} veth;

		veth.h_vlan_proto = skb->vlan_proto;
2063
		veth.h_vlan_TCI = htons(skb_vlan_tag_get(skb));
J
Jason Wang 已提交
2064 2065 2066

		vlan_offset = offsetof(struct vlan_ethhdr, h_vlan_proto);

H
Herbert Xu 已提交
2067 2068
		ret = skb_copy_datagram_iter(skb, 0, iter, vlan_offset);
		if (ret || !iov_iter_count(iter))
J
Jason Wang 已提交
2069 2070
			goto done;

H
Herbert Xu 已提交
2071 2072
		ret = copy_to_iter(&veth, sizeof(veth), iter);
		if (ret != sizeof(veth) || !iov_iter_count(iter))
J
Jason Wang 已提交
2073 2074
			goto done;
	}
L
Linus Torvalds 已提交
2075

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

J
Jason Wang 已提交
2078
done:
2079 2080 2081 2082 2083 2084 2085
	/* 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 已提交
2086 2087 2088 2089

	return total;
}

J
Jason Wang 已提交
2090
static void *tun_ring_recv(struct tun_file *tfile, int noblock, int *err)
J
Jason Wang 已提交
2091 2092
{
	DECLARE_WAITQUEUE(wait, current);
J
Jason Wang 已提交
2093
	void *ptr = NULL;
J
Jason Wang 已提交
2094
	int error = 0;
J
Jason Wang 已提交
2095

J
Jason Wang 已提交
2096 2097
	ptr = ptr_ring_consume(&tfile->tx_ring);
	if (ptr)
J
Jason Wang 已提交
2098 2099
		goto out;
	if (noblock) {
J
Jason Wang 已提交
2100
		error = -EAGAIN;
J
Jason Wang 已提交
2101 2102 2103 2104 2105 2106 2107
		goto out;
	}

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

	while (1) {
J
Jason Wang 已提交
2108 2109
		ptr = ptr_ring_consume(&tfile->tx_ring);
		if (ptr)
J
Jason Wang 已提交
2110 2111
			break;
		if (signal_pending(current)) {
J
Jason Wang 已提交
2112
			error = -ERESTARTSYS;
J
Jason Wang 已提交
2113 2114 2115
			break;
		}
		if (tfile->socket.sk->sk_shutdown & RCV_SHUTDOWN) {
J
Jason Wang 已提交
2116
			error = -EFAULT;
J
Jason Wang 已提交
2117 2118 2119 2120 2121 2122 2123 2124 2125 2126
			break;
		}

		schedule();
	}

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

out:
J
Jason Wang 已提交
2127
	*err = error;
J
Jason Wang 已提交
2128
	return ptr;
J
Jason Wang 已提交
2129 2130
}

J
Jason Wang 已提交
2131
static ssize_t tun_do_read(struct tun_struct *tun, struct tun_file *tfile,
2132
			   struct iov_iter *to,
J
Jason Wang 已提交
2133
			   int noblock, void *ptr)
L
Linus Torvalds 已提交
2134
{
2135
	ssize_t ret;
J
Jason Wang 已提交
2136
	int err;
L
Linus Torvalds 已提交
2137

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

W
Wei Xu 已提交
2140
	if (!iov_iter_count(to)) {
J
Jason Wang 已提交
2141
		tun_ptr_free(ptr);
2142
		return 0;
W
Wei Xu 已提交
2143
	}
L
Linus Torvalds 已提交
2144

J
Jason Wang 已提交
2145
	if (!ptr) {
2146
		/* Read frames from ring */
J
Jason Wang 已提交
2147 2148
		ptr = tun_ring_recv(tfile, noblock, &err);
		if (!ptr)
2149 2150
			return err;
	}
H
Herbert Xu 已提交
2151

J
Jason Wang 已提交
2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165
	if (tun_is_xdp_buff(ptr)) {
		struct xdp_buff *xdp = tun_ptr_to_xdp(ptr);

		ret = tun_put_user_xdp(tun, tfile, xdp, to);
		put_page(virt_to_head_page(xdp->data));
	} 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 已提交
2166

2167 2168 2169
	return ret;
}

2170
static ssize_t tun_chr_read_iter(struct kiocb *iocb, struct iov_iter *to)
2171 2172 2173
{
	struct file *file = iocb->ki_filp;
	struct tun_file *tfile = file->private_data;
2174
	struct tun_struct *tun = tun_get(tfile);
2175
	ssize_t len = iov_iter_count(to), ret;
2176 2177 2178

	if (!tun)
		return -EBADFD;
2179
	ret = tun_do_read(tun, tfile, to, file->f_flags & O_NONBLOCK, NULL);
2180
	ret = min_t(ssize_t, ret, len);
Z
Zhi Yong Wu 已提交
2181 2182
	if (ret > 0)
		iocb->ki_pos = ret;
E
Eric W. Biederman 已提交
2183
	tun_put(tun);
L
Linus Torvalds 已提交
2184 2185 2186
	return ret;
}

2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207
static void tun_steering_prog_free(struct rcu_head *rcu)
{
	struct tun_steering_prog *prog = container_of(rcu,
					 struct tun_steering_prog, rcu);

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

static int __tun_set_steering_ebpf(struct tun_struct *tun,
				   struct bpf_prog *prog)
{
	struct tun_steering_prog *old, *new = NULL;

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

2208 2209 2210
	spin_lock_bh(&tun->lock);
	old = rcu_dereference_protected(tun->steering_prog,
					lockdep_is_held(&tun->lock));
2211
	rcu_assign_pointer(tun->steering_prog, new);
2212
	spin_unlock_bh(&tun->lock);
2213 2214 2215 2216 2217 2218 2219

	if (old)
		call_rcu(&old->rcu, tun_steering_prog_free);

	return 0;
}

J
Jason Wang 已提交
2220 2221 2222 2223
static void tun_free_netdev(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

J
Jason Wang 已提交
2224
	BUG_ON(!(list_empty(&tun->disabled)));
2225
	free_percpu(tun->pcpu_stats);
J
Jason Wang 已提交
2226
	tun_flow_uninit(tun);
2227
	security_tun_dev_free_security(tun->security);
2228
	__tun_set_steering_ebpf(tun, NULL);
J
Jason Wang 已提交
2229 2230
}

L
Linus Torvalds 已提交
2231 2232 2233 2234
static void tun_setup(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

2235 2236
	tun->owner = INVALID_UID;
	tun->group = INVALID_GID;
L
Linus Torvalds 已提交
2237 2238

	dev->ethtool_ops = &tun_ethtool_ops;
2239 2240
	dev->needs_free_netdev = true;
	dev->priv_destructor = tun_free_netdev;
J
Jason Wang 已提交
2241 2242
	/* We prefer our own queue length */
	dev->tx_queue_len = TUN_READQ_SIZE;
L
Linus Torvalds 已提交
2243 2244
}

2245 2246 2247
/* Trivial set of netlink ops to allow deleting tun or tap
 * device with netlink.
 */
2248 2249
static int tun_validate(struct nlattr *tb[], struct nlattr *data[],
			struct netlink_ext_ack *extack)
2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260
{
	return -EINVAL;
}

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

2261 2262
static void tun_sock_write_space(struct sock *sk)
{
J
Jason Wang 已提交
2263
	struct tun_file *tfile;
2264
	wait_queue_head_t *wqueue;
2265 2266 2267 2268

	if (!sock_writeable(sk))
		return;

2269
	if (!test_and_clear_bit(SOCKWQ_ASYNC_NOSPACE, &sk->sk_socket->flags))
2270 2271
		return;

2272 2273 2274
	wqueue = sk_sleep(sk);
	if (wqueue && waitqueue_active(wqueue))
		wake_up_interruptible_sync_poll(wqueue, POLLOUT |
2275
						POLLWRNORM | POLLWRBAND);
H
Herbert Xu 已提交
2276

J
Jason Wang 已提交
2277 2278
	tfile = container_of(sk, struct tun_file, sk);
	kill_fasync(&tfile->fasync, SIGIO, POLL_OUT);
2279 2280
}

2281
static int tun_sendmsg(struct socket *sock, struct msghdr *m, size_t total_len)
2282
{
J
Jason Wang 已提交
2283 2284
	int ret;
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2285
	struct tun_struct *tun = tun_get(tfile);
J
Jason Wang 已提交
2286 2287 2288

	if (!tun)
		return -EBADFD;
2289

A
Al Viro 已提交
2290
	ret = tun_get_user(tun, tfile, m->msg_control, &m->msg_iter,
J
Jason Wang 已提交
2291 2292
			   m->msg_flags & MSG_DONTWAIT,
			   m->msg_flags & MSG_MORE);
J
Jason Wang 已提交
2293 2294
	tun_put(tun);
	return ret;
2295 2296
}

2297
static int tun_recvmsg(struct socket *sock, struct msghdr *m, size_t total_len,
2298 2299
		       int flags)
{
J
Jason Wang 已提交
2300
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2301
	struct tun_struct *tun = tun_get(tfile);
J
Jason Wang 已提交
2302
	void *ptr = m->msg_control;
2303
	int ret;
J
Jason Wang 已提交
2304

W
Wei Xu 已提交
2305 2306
	if (!tun) {
		ret = -EBADFD;
J
Jason Wang 已提交
2307
		goto out_free;
W
Wei Xu 已提交
2308
	}
J
Jason Wang 已提交
2309

2310
	if (flags & ~(MSG_DONTWAIT|MSG_TRUNC|MSG_ERRQUEUE)) {
2311
		ret = -EINVAL;
W
Wei Xu 已提交
2312
		goto out_put_tun;
2313
	}
2314 2315 2316 2317 2318
	if (flags & MSG_ERRQUEUE) {
		ret = sock_recv_errqueue(sock->sk, m, total_len,
					 SOL_PACKET, TUN_TX_TIMESTAMP);
		goto out;
	}
J
Jason Wang 已提交
2319
	ret = tun_do_read(tun, tfile, &m->msg_iter, flags & MSG_DONTWAIT, ptr);
2320
	if (ret > (ssize_t)total_len) {
2321 2322 2323
		m->msg_flags |= MSG_TRUNC;
		ret = flags & MSG_TRUNC ? ret : total_len;
	}
2324
out:
J
Jason Wang 已提交
2325
	tun_put(tun);
2326
	return ret;
W
Wei Xu 已提交
2327 2328 2329

out_put_tun:
	tun_put(tun);
J
Jason Wang 已提交
2330 2331
out_free:
	tun_ptr_free(ptr);
W
Wei Xu 已提交
2332
	return ret;
2333 2334
}

J
Jason Wang 已提交
2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348
static int tun_ptr_peek_len(void *ptr)
{
	if (likely(ptr)) {
		if (tun_is_xdp_buff(ptr)) {
			struct xdp_buff *xdp = tun_ptr_to_xdp(ptr);

			return xdp->data_end - xdp->data;
		}
		return __skb_array_len_with_tag(ptr);
	} else {
		return 0;
	}
}

J
Jason Wang 已提交
2349 2350 2351 2352 2353 2354
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;

2355
	tun = tun_get(tfile);
J
Jason Wang 已提交
2356 2357 2358
	if (!tun)
		return 0;

J
Jason Wang 已提交
2359
	ret = PTR_RING_PEEK_CALL(&tfile->tx_ring, tun_ptr_peek_len);
J
Jason Wang 已提交
2360 2361 2362 2363 2364
	tun_put(tun);

	return ret;
}

2365 2366
/* Ops structure to mimic raw sockets with tun */
static const struct proto_ops tun_socket_ops = {
J
Jason Wang 已提交
2367
	.peek_len = tun_peek_len,
2368 2369 2370 2371
	.sendmsg = tun_sendmsg,
	.recvmsg = tun_recvmsg,
};

2372 2373 2374
static struct proto tun_proto = {
	.name		= "tun",
	.owner		= THIS_MODULE,
J
Jason Wang 已提交
2375
	.obj_size	= sizeof(struct tun_file),
2376
};
2377

2378 2379
static int tun_flags(struct tun_struct *tun)
{
2380
	return tun->flags & (TUN_FEATURES | IFF_PERSIST | IFF_TUN | IFF_TAP);
2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393
}

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));
2394 2395 2396 2397
	return uid_valid(tun->owner)?
		sprintf(buf, "%u\n",
			from_kuid_munged(current_user_ns(), tun->owner)):
		sprintf(buf, "-1\n");
2398 2399 2400 2401 2402 2403
}

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));
2404 2405 2406 2407
	return gid_valid(tun->group) ?
		sprintf(buf, "%u\n",
			from_kgid_munged(current_user_ns(), tun->group)):
		sprintf(buf, "-1\n");
2408 2409 2410 2411 2412 2413
}

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

2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424
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
};

2425
static int tun_set_iff(struct net *net, struct file *file, struct ifreq *ifr)
L
Linus Torvalds 已提交
2426 2427
{
	struct tun_struct *tun;
J
Jason Wang 已提交
2428
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
2429 2430 2431
	struct net_device *dev;
	int err;

2432 2433 2434
	if (tfile->detached)
		return -EINVAL;

2435 2436 2437 2438 2439 2440 2441 2442 2443
	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;
	}

2444 2445
	dev = __dev_get_by_name(net, ifr->ifr_name);
	if (dev) {
2446 2447
		if (ifr->ifr_flags & IFF_TUN_EXCL)
			return -EBUSY;
2448 2449 2450 2451 2452 2453 2454
		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;

2455
		if (!!(ifr->ifr_flags & IFF_MULTI_QUEUE) !=
M
Michael S. Tsirkin 已提交
2456
		    !!(tun->flags & IFF_MULTI_QUEUE))
2457 2458
			return -EINVAL;

2459
		if (tun_not_capable(tun))
P
Paul Moore 已提交
2460
			return -EPERM;
2461
		err = security_tun_dev_open(tun->security);
P
Paul Moore 已提交
2462 2463 2464
		if (err < 0)
			return err;

2465 2466
		err = tun_attach(tun, file, ifr->ifr_flags & IFF_NOFILTER,
				 ifr->ifr_flags & IFF_NAPI);
2467 2468
		if (err < 0)
			return err;
J
Jason Wang 已提交
2469

M
Michael S. Tsirkin 已提交
2470
		if (tun->flags & IFF_MULTI_QUEUE &&
2471 2472 2473 2474 2475 2476
		    (tun->numqueues + tun->numdisabled > 1)) {
			/* One or more queue has already been attached, no need
			 * to initialize the device again.
			 */
			return 0;
		}
2477
	}
L
Linus Torvalds 已提交
2478 2479 2480
	else {
		char *name;
		unsigned long flags = 0;
2481 2482
		int queues = ifr->ifr_flags & IFF_MULTI_QUEUE ?
			     MAX_TAP_QUEUES : 1;
L
Linus Torvalds 已提交
2483

2484
		if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
2485
			return -EPERM;
P
Paul Moore 已提交
2486 2487 2488
		err = security_tun_dev_create();
		if (err < 0)
			return err;
2489

L
Linus Torvalds 已提交
2490 2491 2492
		/* Set dev type */
		if (ifr->ifr_flags & IFF_TUN) {
			/* TUN device */
M
Michael S. Tsirkin 已提交
2493
			flags |= IFF_TUN;
L
Linus Torvalds 已提交
2494 2495 2496
			name = "tun%d";
		} else if (ifr->ifr_flags & IFF_TAP) {
			/* TAP device */
M
Michael S. Tsirkin 已提交
2497
			flags |= IFF_TAP;
L
Linus Torvalds 已提交
2498
			name = "tap%d";
2499
		} else
2500
			return -EINVAL;
2501

L
Linus Torvalds 已提交
2502 2503 2504
		if (*ifr->ifr_name)
			name = ifr->ifr_name;

J
Jason Wang 已提交
2505
		dev = alloc_netdev_mqs(sizeof(struct tun_struct), name,
2506 2507
				       NET_NAME_UNKNOWN, tun_setup, queues,
				       queues);
2508

L
Linus Torvalds 已提交
2509 2510
		if (!dev)
			return -ENOMEM;
2511
		err = dev_get_valid_name(net, dev, name);
2512
		if (err < 0)
2513
			goto err_free_dev;
L
Linus Torvalds 已提交
2514

2515
		dev_net_set(dev, net);
2516
		dev->rtnl_link_ops = &tun_link_ops;
2517
		dev->ifindex = tfile->ifindex;
2518
		dev->sysfs_groups[0] = &tun_attr_group;
S
Stephen Hemminger 已提交
2519

L
Linus Torvalds 已提交
2520 2521 2522
		tun = netdev_priv(dev);
		tun->dev = dev;
		tun->flags = flags;
2523
		tun->txflt.count = 0;
2524
		tun->vnet_hdr_sz = sizeof(struct virtio_net_hdr);
2525

2526
		tun->align = NET_SKB_PAD;
J
Jason Wang 已提交
2527 2528
		tun->filter_attached = false;
		tun->sndbuf = tfile->socket.sk->sk_sndbuf;
J
Jason Wang 已提交
2529
		tun->rx_batched = 0;
2530
		RCU_INIT_POINTER(tun->steering_prog, NULL);
2531

2532 2533 2534 2535 2536 2537
		tun->pcpu_stats = netdev_alloc_pcpu_stats(struct tun_pcpu_stats);
		if (!tun->pcpu_stats) {
			err = -ENOMEM;
			goto err_free_dev;
		}

J
Jason Wang 已提交
2538 2539
		spin_lock_init(&tun->lock);

2540 2541
		err = security_tun_dev_alloc_security(&tun->security);
		if (err < 0)
2542
			goto err_free_stat;
P
Paul Moore 已提交
2543

L
Linus Torvalds 已提交
2544
		tun_net_init(dev);
2545
		tun_flow_init(tun);
J
Jason Wang 已提交
2546

2547
		dev->hw_features = NETIF_F_SG | NETIF_F_FRAGLIST |
J
Jason Wang 已提交
2548 2549
				   TUN_USER_FEATURES | NETIF_F_HW_VLAN_CTAG_TX |
				   NETIF_F_HW_VLAN_STAG_TX;
2550
		dev->features = dev->hw_features | NETIF_F_LLTX;
2551 2552 2553
		dev->vlan_features = dev->features &
				     ~(NETIF_F_HW_VLAN_CTAG_TX |
				       NETIF_F_HW_VLAN_STAG_TX);
2554

J
Jason Wang 已提交
2555
		INIT_LIST_HEAD(&tun->disabled);
2556
		err = tun_attach(tun, file, false, ifr->ifr_flags & IFF_NAPI);
2557
		if (err < 0)
2558
			goto err_free_flow;
2559

L
Linus Torvalds 已提交
2560 2561
		err = register_netdevice(tun->dev);
		if (err < 0)
2562
			goto err_detach;
L
Linus Torvalds 已提交
2563 2564
	}

2565 2566
	netif_carrier_on(tun->dev);

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

2569 2570
	tun->flags = (tun->flags & ~TUN_FEATURES) |
		(ifr->ifr_flags & TUN_FEATURES);
J
Jason Wang 已提交
2571

2572 2573 2574 2575
	/* Make sure persistent devices do not get stuck in
	 * xoff state.
	 */
	if (netif_running(tun->dev))
J
Jason Wang 已提交
2576
		netif_tx_wake_all_queues(tun->dev);
2577

L
Linus Torvalds 已提交
2578 2579 2580
	strcpy(ifr->ifr_name, tun->dev->name);
	return 0;

2581 2582
err_detach:
	tun_detach_all(dev);
2583 2584 2585
	/* register_netdevice() already called tun_free_netdev() */
	goto err_free_dev;

2586 2587 2588
err_free_flow:
	tun_flow_uninit(tun);
	security_tun_dev_free_security(tun->security);
2589 2590
err_free_stat:
	free_percpu(tun->pcpu_stats);
2591
err_free_dev:
L
Linus Torvalds 已提交
2592 2593 2594 2595
	free_netdev(dev);
	return err;
}

R
Rami Rosen 已提交
2596
static void tun_get_iff(struct net *net, struct tun_struct *tun,
2597
		       struct ifreq *ifr)
2598
{
2599
	tun_debug(KERN_INFO, tun, "tun_get_iff\n");
2600 2601 2602

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

2603
	ifr->ifr_flags = tun_flags(tun);
2604 2605 2606

}

2607 2608
/* This is like a cut-down ethtool ops, except done via tun fd so no
 * privs required. */
2609
static int set_offload(struct tun_struct *tun, unsigned long arg)
2610
{
2611
	netdev_features_t features = 0;
2612 2613

	if (arg & TUN_F_CSUM) {
2614
		features |= NETIF_F_HW_CSUM;
2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627
		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);
		}
2628 2629

		arg &= ~TUN_F_UFO;
2630 2631 2632 2633 2634 2635 2636
	}

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

2637
	tun->set_features = features;
2638 2639
	tun->dev->wanted_features &= ~TUN_USER_FEATURES;
	tun->dev->wanted_features |= features;
2640
	netdev_update_features(tun->dev);
2641 2642 2643 2644

	return 0;
}

J
Jason Wang 已提交
2645 2646 2647 2648 2649 2650
static void tun_detach_filter(struct tun_struct *tun, int n)
{
	int i;
	struct tun_file *tfile;

	for (i = 0; i < n; i++) {
2651
		tfile = rtnl_dereference(tun->tfiles[i]);
2652 2653 2654
		lock_sock(tfile->socket.sk);
		sk_detach_filter(tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665
	}

	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++) {
2666
		tfile = rtnl_dereference(tun->tfiles[i]);
2667 2668 2669
		lock_sock(tfile->socket.sk);
		ret = sk_attach_filter(&tun->fprog, tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2670 2671 2672 2673 2674 2675 2676 2677 2678 2679 2680 2681 2682 2683 2684 2685
		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++) {
2686
		tfile = rtnl_dereference(tun->tfiles[i]);
J
Jason Wang 已提交
2687 2688 2689 2690
		tfile->socket.sk->sk_sndbuf = tun->sndbuf;
	}
}

2691 2692 2693 2694 2695 2696 2697 2698 2699
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 已提交
2700
		tun = tfile->detached;
2701
		if (!tun) {
2702
			ret = -EINVAL;
2703 2704 2705 2706 2707
			goto unlock;
		}
		ret = security_tun_dev_attach_queue(tun->security);
		if (ret < 0)
			goto unlock;
2708
		ret = tun_attach(tun, file, false, tun->flags & IFF_NAPI);
J
Jason Wang 已提交
2709
	} else if (ifr->ifr_flags & IFF_DETACH_QUEUE) {
2710
		tun = rtnl_dereference(tfile->tun);
M
Michael S. Tsirkin 已提交
2711
		if (!tun || !(tun->flags & IFF_MULTI_QUEUE) || tfile->detached)
J
Jason Wang 已提交
2712 2713 2714 2715
			ret = -EINVAL;
		else
			__tun_detach(tfile, false);
	} else
2716 2717
		ret = -EINVAL;

2718
unlock:
2719 2720 2721 2722
	rtnl_unlock();
	return ret;
}

2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736 2737 2738 2739 2740 2741
static int tun_set_steering_ebpf(struct tun_struct *tun, void __user *data)
{
	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);
	}

	return __tun_set_steering_ebpf(tun, prog);
}

2742 2743
static long __tun_chr_ioctl(struct file *file, unsigned int cmd,
			    unsigned long arg, int ifreq_len)
L
Linus Torvalds 已提交
2744
{
E
Eric W. Biederman 已提交
2745
	struct tun_file *tfile = file->private_data;
E
Eric W. Biederman 已提交
2746
	struct tun_struct *tun;
L
Linus Torvalds 已提交
2747 2748
	void __user* argp = (void __user*)arg;
	struct ifreq ifr;
2749 2750
	kuid_t owner;
	kgid_t group;
2751
	int sndbuf;
2752
	int vnet_hdr_sz;
2753
	unsigned int ifindex;
M
Michael S. Tsirkin 已提交
2754
	int le;
2755
	int ret;
L
Linus Torvalds 已提交
2756

2757
	if (cmd == TUNSETIFF || cmd == TUNSETQUEUE || _IOC_TYPE(cmd) == SOCK_IOC_TYPE) {
2758
		if (copy_from_user(&ifr, argp, ifreq_len))
L
Linus Torvalds 已提交
2759
			return -EFAULT;
D
David S. Miller 已提交
2760
	} else {
2761
		memset(&ifr, 0, sizeof(ifr));
D
David S. Miller 已提交
2762
	}
E
Eric W. Biederman 已提交
2763 2764 2765
	if (cmd == TUNGETFEATURES) {
		/* Currently this just means: "what IFF flags are valid?".
		 * This is needed because we never checked for invalid flags on
2766 2767 2768
		 * TUNSETIFF.
		 */
		return put_user(IFF_TUN | IFF_TAP | TUN_FEATURES,
E
Eric W. Biederman 已提交
2769
				(unsigned int __user*)argp);
2770 2771
	} else if (cmd == TUNSETQUEUE)
		return tun_set_queue(file, &ifr);
E
Eric W. Biederman 已提交
2772

J
Jason Wang 已提交
2773
	ret = 0;
2774 2775
	rtnl_lock();

2776
	tun = tun_get(tfile);
2777 2778 2779 2780 2781
	if (cmd == TUNSETIFF) {
		ret = -EEXIST;
		if (tun)
			goto unlock;

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

2784
		ret = tun_set_iff(sock_net(&tfile->sk), file, &ifr);
L
Linus Torvalds 已提交
2785

2786 2787
		if (ret)
			goto unlock;
L
Linus Torvalds 已提交
2788

2789
		if (copy_to_user(argp, &ifr, ifreq_len))
2790 2791
			ret = -EFAULT;
		goto unlock;
L
Linus Torvalds 已提交
2792
	}
2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804 2805
	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 已提交
2806

2807
	ret = -EBADFD;
L
Linus Torvalds 已提交
2808
	if (!tun)
2809
		goto unlock;
L
Linus Torvalds 已提交
2810

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

E
Eric W. Biederman 已提交
2813
	ret = 0;
L
Linus Torvalds 已提交
2814
	switch (cmd) {
2815
	case TUNGETIFF:
R
Rami Rosen 已提交
2816
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
2817

2818 2819
		if (tfile->detached)
			ifr.ifr_flags |= IFF_DETACH_QUEUE;
2820 2821
		if (!tfile->socket.sk->sk_filter)
			ifr.ifr_flags |= IFF_NOFILTER;
2822

2823
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
2824
			ret = -EFAULT;
2825 2826
		break;

L
Linus Torvalds 已提交
2827 2828 2829
	case TUNSETNOCSUM:
		/* Disable/Enable checksum */

2830 2831
		/* [unimplemented] */
		tun_debug(KERN_INFO, tun, "ignored: set checksum %s\n",
2832
			  arg ? "disabled" : "enabled");
L
Linus Torvalds 已提交
2833 2834 2835
		break;

	case TUNSETPERSIST:
J
Jason Wang 已提交
2836 2837 2838
		/* Disable/Enable persist mode. Keep an extra reference to the
		 * module to prevent the module being unprobed.
		 */
M
Michael S. Tsirkin 已提交
2839 2840
		if (arg && !(tun->flags & IFF_PERSIST)) {
			tun->flags |= IFF_PERSIST;
J
Jason Wang 已提交
2841
			__module_get(THIS_MODULE);
J
Jason Wang 已提交
2842
		}
M
Michael S. Tsirkin 已提交
2843 2844
		if (!arg && (tun->flags & IFF_PERSIST)) {
			tun->flags &= ~IFF_PERSIST;
J
Jason Wang 已提交
2845 2846
			module_put(THIS_MODULE);
		}
L
Linus Torvalds 已提交
2847

2848 2849
		tun_debug(KERN_INFO, tun, "persist %s\n",
			  arg ? "enabled" : "disabled");
L
Linus Torvalds 已提交
2850 2851 2852 2853
		break;

	case TUNSETOWNER:
		/* Set owner of the device */
2854 2855 2856 2857 2858 2859
		owner = make_kuid(current_user_ns(), arg);
		if (!uid_valid(owner)) {
			ret = -EINVAL;
			break;
		}
		tun->owner = owner;
2860
		tun_debug(KERN_INFO, tun, "owner set to %u\n",
2861
			  from_kuid(&init_user_ns, tun->owner));
L
Linus Torvalds 已提交
2862 2863
		break;

2864 2865
	case TUNSETGROUP:
		/* Set group of the device */
2866 2867 2868 2869 2870 2871
		group = make_kgid(current_user_ns(), arg);
		if (!gid_valid(group)) {
			ret = -EINVAL;
			break;
		}
		tun->group = group;
2872
		tun_debug(KERN_INFO, tun, "group set to %u\n",
2873
			  from_kgid(&init_user_ns, tun->group));
2874 2875
		break;

2876 2877 2878
	case TUNSETLINK:
		/* Only allow setting the type when the interface is down */
		if (tun->dev->flags & IFF_UP) {
2879 2880
			tun_debug(KERN_INFO, tun,
				  "Linktype set failed because interface is up\n");
2881
			ret = -EBUSY;
2882 2883
		} else {
			tun->dev->type = (int) arg;
2884 2885
			tun_debug(KERN_INFO, tun, "linktype set to %d\n",
				  tun->dev->type);
2886
			ret = 0;
2887
		}
E
Eric W. Biederman 已提交
2888
		break;
2889

L
Linus Torvalds 已提交
2890 2891 2892 2893 2894
#ifdef TUN_DEBUG
	case TUNSETDEBUG:
		tun->debug = arg;
		break;
#endif
2895
	case TUNSETOFFLOAD:
2896
		ret = set_offload(tun, arg);
E
Eric W. Biederman 已提交
2897
		break;
2898

2899 2900
	case TUNSETTXFILTER:
		/* Can be set only for TAPs */
E
Eric W. Biederman 已提交
2901
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
2902
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
E
Eric W. Biederman 已提交
2903
			break;
H
Harvey Harrison 已提交
2904
		ret = update_filter(&tun->txflt, (void __user *)arg);
E
Eric W. Biederman 已提交
2905
		break;
L
Linus Torvalds 已提交
2906 2907

	case SIOCGIFHWADDR:
2908
		/* Get hw address */
2909 2910
		memcpy(ifr.ifr_hwaddr.sa_data, tun->dev->dev_addr, ETH_ALEN);
		ifr.ifr_hwaddr.sa_family = tun->dev->type;
2911
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
2912 2913
			ret = -EFAULT;
		break;
L
Linus Torvalds 已提交
2914 2915

	case SIOCSIFHWADDR:
2916
		/* Set hw address */
2917 2918
		tun_debug(KERN_DEBUG, tun, "set hw address: %pM\n",
			  ifr.ifr_hwaddr.sa_data);
2919 2920

		ret = dev_set_mac_address(tun->dev, &ifr.ifr_hwaddr);
E
Eric W. Biederman 已提交
2921
		break;
2922 2923

	case TUNGETSNDBUF:
J
Jason Wang 已提交
2924
		sndbuf = tfile->socket.sk->sk_sndbuf;
2925 2926 2927 2928 2929 2930 2931 2932 2933
		if (copy_to_user(argp, &sndbuf, sizeof(sndbuf)))
			ret = -EFAULT;
		break;

	case TUNSETSNDBUF:
		if (copy_from_user(&sndbuf, argp, sizeof(sndbuf))) {
			ret = -EFAULT;
			break;
		}
2934 2935 2936 2937
		if (sndbuf <= 0) {
			ret = -EINVAL;
			break;
		}
2938

J
Jason Wang 已提交
2939 2940
		tun->sndbuf = sndbuf;
		tun_set_sndbuf(tun);
2941 2942
		break;

2943 2944 2945 2946 2947 2948 2949 2950 2951 2952 2953 2954 2955 2956 2957 2958 2959 2960 2961
	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 已提交
2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973 2974 2975 2976 2977 2978
	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;

2979 2980 2981 2982 2983 2984 2985 2986
	case TUNGETVNETBE:
		ret = tun_get_vnet_be(tun, argp);
		break;

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

M
Michael S. Tsirkin 已提交
2987 2988 2989
	case TUNATTACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
2990
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
2991 2992
			break;
		ret = -EFAULT;
J
Jason Wang 已提交
2993
		if (copy_from_user(&tun->fprog, argp, sizeof(tun->fprog)))
M
Michael S. Tsirkin 已提交
2994 2995
			break;

J
Jason Wang 已提交
2996
		ret = tun_attach_filter(tun);
M
Michael S. Tsirkin 已提交
2997 2998 2999 3000 3001
		break;

	case TUNDETACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3002
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
3003
			break;
J
Jason Wang 已提交
3004 3005
		ret = 0;
		tun_detach_filter(tun, tun->numqueues);
M
Michael S. Tsirkin 已提交
3006 3007
		break;

P
Pavel Emelyanov 已提交
3008 3009
	case TUNGETFILTER:
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3010
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
P
Pavel Emelyanov 已提交
3011 3012 3013 3014 3015 3016 3017
			break;
		ret = -EFAULT;
		if (copy_to_user(argp, &tun->fprog, sizeof(tun->fprog)))
			break;
		ret = 0;
		break;

3018 3019 3020 3021
	case TUNSETSTEERINGEBPF:
		ret = tun_set_steering_ebpf(tun, argp);
		break;

L
Linus Torvalds 已提交
3022
	default:
E
Eric W. Biederman 已提交
3023 3024
		ret = -EINVAL;
		break;
3025
	}
L
Linus Torvalds 已提交
3026

3027 3028 3029 3030
unlock:
	rtnl_unlock();
	if (tun)
		tun_put(tun);
E
Eric W. Biederman 已提交
3031
	return ret;
L
Linus Torvalds 已提交
3032 3033
}

3034 3035 3036 3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049 3050 3051 3052 3053 3054 3055 3056 3057 3058 3059 3060 3061 3062 3063 3064 3065 3066 3067 3068
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 已提交
3069 3070
static int tun_chr_fasync(int fd, struct file *file, int on)
{
J
Jason Wang 已提交
3071
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
3072 3073
	int ret;

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

L
Linus Torvalds 已提交
3077
	if (on) {
3078
		__f_setown(file, task_pid(current), PIDTYPE_PID, 0);
J
Jason Wang 已提交
3079
		tfile->flags |= TUN_FASYNC;
3080
	} else
J
Jason Wang 已提交
3081
		tfile->flags &= ~TUN_FASYNC;
J
Jonathan Corbet 已提交
3082 3083 3084
	ret = 0;
out:
	return ret;
L
Linus Torvalds 已提交
3085 3086 3087 3088
}

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

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

3094
	tfile = (struct tun_file *)sk_alloc(net, AF_UNSPEC, GFP_KERNEL,
3095
					    &tun_proto, 0);
E
Eric W. Biederman 已提交
3096 3097
	if (!tfile)
		return -ENOMEM;
3098
	RCU_INIT_POINTER(tfile->tun, NULL);
J
Jason Wang 已提交
3099
	tfile->flags = 0;
3100
	tfile->ifindex = 0;
J
Jason Wang 已提交
3101 3102

	init_waitqueue_head(&tfile->wq.wait);
3103
	RCU_INIT_POINTER(tfile->socket.wq, &tfile->wq);
J
Jason Wang 已提交
3104 3105 3106 3107 3108 3109 3110 3111 3112

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

3116 3117
	sock_set_flag(&tfile->sk, SOCK_ZEROCOPY);

L
Linus Torvalds 已提交
3118 3119 3120 3121 3122
	return 0;
}

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

J
Jason Wang 已提交
3125
	tun_detach(tfile, true);
L
Linus Torvalds 已提交
3126 3127 3128 3129

	return 0;
}

3130
#ifdef CONFIG_PROC_FS
3131
static void tun_chr_show_fdinfo(struct seq_file *m, struct file *file)
3132
{
3133
	struct tun_file *tfile = file->private_data;
3134 3135 3136 3137 3138 3139
	struct tun_struct *tun;
	struct ifreq ifr;

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

	rtnl_lock();
3140
	tun = tun_get(tfile);
3141 3142 3143 3144 3145 3146 3147
	if (tun)
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
	rtnl_unlock();

	if (tun)
		tun_put(tun);

3148
	seq_printf(m, "iff:\t%s\n", ifr.ifr_name);
3149 3150 3151
}
#endif

3152
static const struct file_operations tun_fops = {
3153
	.owner	= THIS_MODULE,
L
Linus Torvalds 已提交
3154
	.llseek = no_llseek,
3155
	.read_iter  = tun_chr_read_iter,
3156
	.write_iter = tun_chr_write_iter,
L
Linus Torvalds 已提交
3157
	.poll	= tun_chr_poll,
3158 3159 3160 3161
	.unlocked_ioctl	= tun_chr_ioctl,
#ifdef CONFIG_COMPAT
	.compat_ioctl = tun_chr_compat_ioctl,
#endif
L
Linus Torvalds 已提交
3162 3163
	.open	= tun_chr_open,
	.release = tun_chr_close,
3164 3165 3166 3167
	.fasync = tun_chr_fasync,
#ifdef CONFIG_PROC_FS
	.show_fdinfo = tun_chr_show_fdinfo,
#endif
L
Linus Torvalds 已提交
3168 3169 3170 3171 3172
};

static struct miscdevice tun_miscdev = {
	.minor = TUN_MINOR,
	.name = "tun",
3173
	.nodename = "net/tun",
L
Linus Torvalds 已提交
3174 3175 3176 3177 3178
	.fops = &tun_fops,
};

/* ethtool interface */

3179 3180 3181 3182 3183 3184 3185 3186 3187 3188
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 已提交
3189 3190 3191 3192 3193 3194 3195
	return 0;
}

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

3196 3197
	strlcpy(info->driver, DRV_NAME, sizeof(info->driver));
	strlcpy(info->version, DRV_VERSION, sizeof(info->version));
L
Linus Torvalds 已提交
3198 3199

	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
3200
	case IFF_TUN:
3201
		strlcpy(info->bus_info, "tun", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3202
		break;
M
Michael S. Tsirkin 已提交
3203
	case IFF_TAP:
3204
		strlcpy(info->bus_info, "tap", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3205 3206 3207 3208 3209 3210 3211 3212 3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224 3225 3226
		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 已提交
3227 3228 3229 3230 3231 3232 3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249
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;
}

3250
static const struct ethtool_ops tun_ethtool_ops = {
L
Linus Torvalds 已提交
3251 3252 3253
	.get_drvinfo	= tun_get_drvinfo,
	.get_msglevel	= tun_get_msglevel,
	.set_msglevel	= tun_set_msglevel,
3254
	.get_link	= ethtool_op_get_link,
3255
	.get_ts_info	= ethtool_op_get_ts_info,
J
Jason Wang 已提交
3256 3257
	.get_coalesce   = tun_get_coalesce,
	.set_coalesce   = tun_set_coalesce,
3258
	.get_link_ksettings = tun_get_link_ksettings,
L
Linus Torvalds 已提交
3259 3260
};

J
Jason Wang 已提交
3261 3262 3263 3264
static int tun_queue_resize(struct tun_struct *tun)
{
	struct net_device *dev = tun->dev;
	struct tun_file *tfile;
3265
	struct ptr_ring **rings;
J
Jason Wang 已提交
3266 3267 3268
	int n = tun->numqueues + tun->numdisabled;
	int ret, i;

3269 3270
	rings = kmalloc_array(n, sizeof(*rings), GFP_KERNEL);
	if (!rings)
J
Jason Wang 已提交
3271 3272 3273 3274
		return -ENOMEM;

	for (i = 0; i < tun->numqueues; i++) {
		tfile = rtnl_dereference(tun->tfiles[i]);
3275
		rings[i] = &tfile->tx_ring;
J
Jason Wang 已提交
3276 3277
	}
	list_for_each_entry(tfile, &tun->disabled, next)
3278
		rings[i++] = &tfile->tx_ring;
J
Jason Wang 已提交
3279

3280 3281
	ret = ptr_ring_resize_multiple(rings, n,
				       dev->tx_queue_len, GFP_KERNEL,
J
Jason Wang 已提交
3282
				       tun_ptr_free);
J
Jason Wang 已提交
3283

3284
	kfree(rings);
J
Jason Wang 已提交
3285 3286 3287 3288 3289 3290 3291 3292 3293
	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);

3294 3295 3296
	if (dev->rtnl_link_ops != &tun_link_ops)
		return NOTIFY_DONE;

J
Jason Wang 已提交
3297 3298 3299 3300 3301 3302 3303 3304 3305 3306 3307 3308 3309 3310 3311
	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,
};
3312

L
Linus Torvalds 已提交
3313 3314 3315 3316
static int __init tun_init(void)
{
	int ret = 0;

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

3319
	ret = rtnl_link_register(&tun_link_ops);
3320
	if (ret) {
3321
		pr_err("Can't register link_ops\n");
3322
		goto err_linkops;
3323 3324
	}

L
Linus Torvalds 已提交
3325
	ret = misc_register(&tun_miscdev);
3326
	if (ret) {
3327
		pr_err("Can't register misc device %d\n", TUN_MINOR);
3328 3329
		goto err_misc;
	}
J
Jason Wang 已提交
3330

3331 3332 3333 3334 3335 3336
	ret = register_netdevice_notifier(&tun_notifier_block);
	if (ret) {
		pr_err("Can't register netdevice notifier\n");
		goto err_notifier;
	}

3337
	return  0;
3338 3339 3340

err_notifier:
	misc_deregister(&tun_miscdev);
3341
err_misc:
3342 3343
	rtnl_link_unregister(&tun_link_ops);
err_linkops:
L
Linus Torvalds 已提交
3344 3345 3346 3347 3348
	return ret;
}

static void tun_cleanup(void)
{
3349
	misc_deregister(&tun_miscdev);
3350
	rtnl_link_unregister(&tun_link_ops);
J
Jason Wang 已提交
3351
	unregister_netdevice_notifier(&tun_notifier_block);
L
Linus Torvalds 已提交
3352 3353
}

3354 3355 3356 3357 3358 3359
/* 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)
{
3360
	struct tun_file *tfile;
3361 3362
	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
3363 3364
	tfile = file->private_data;
	if (!tfile)
3365
		return ERR_PTR(-EBADFD);
J
Jason Wang 已提交
3366
	return &tfile->socket;
3367 3368 3369
}
EXPORT_SYMBOL_GPL(tun_get_socket);

3370
struct ptr_ring *tun_get_tx_ring(struct file *file)
J
Jason Wang 已提交
3371 3372 3373 3374 3375 3376 3377 3378
{
	struct tun_file *tfile;

	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
	tfile = file->private_data;
	if (!tfile)
		return ERR_PTR(-EBADFD);
3379
	return &tfile->tx_ring;
J
Jason Wang 已提交
3380
}
3381
EXPORT_SYMBOL_GPL(tun_get_tx_ring);
J
Jason Wang 已提交
3382

L
Linus Torvalds 已提交
3383 3384 3385 3386 3387 3388
module_init(tun_init);
module_exit(tun_cleanup);
MODULE_DESCRIPTION(DRV_DESCRIPTION);
MODULE_AUTHOR(DRV_COPYRIGHT);
MODULE_LICENSE("GPL");
MODULE_ALIAS_MISCDEV(TUN_MINOR);
3389
MODULE_ALIAS("devname:net/tun");