tun.c 78.8 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;
J
Jason Wang 已提交
184
	int xdp_pending_pkts;
E
Eric W. Biederman 已提交
185 186
};

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

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

#define TUN_NUM_FLOW_ENTRIES 1024

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

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

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

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

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

J
Jason Wang 已提交
251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268
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);

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 305 306 307 308 309 310 311
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)
{
312
	tfile->napi_enabled = napi_en;
313 314 315 316
	if (napi_en) {
		netif_napi_add(tun->dev, &tfile->napi, tun_napi_poll,
			       NAPI_POLL_WEIGHT);
		napi_enable(&tfile->napi);
317
		mutex_init(&tfile->napi_mutex);
318 319 320 321 322
	}
}

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

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

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

338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385
#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 */

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

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

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

J
Jason Wang 已提交
402 403 404 405 406 407 408 409 410
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;

411
	hlist_for_each_entry_rcu(e, head, hash_link) {
J
Jason Wang 已提交
412 413 414 415 416 417 418 419 420 421
		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)
{
422 423
	struct tun_flow_entry *e = kmalloc(sizeof(*e), GFP_ATOMIC);

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

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

456
		hlist_for_each_entry_safe(e, n, &tun->flows[i], hash_link)
J
Jason Wang 已提交
457 458 459 460 461 462 463 464 465 466 467 468
			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;
469
		struct hlist_node *n;
J
Jason Wang 已提交
470

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

479
static void tun_flow_cleanup(struct timer_list *t)
J
Jason Wang 已提交
480
{
481
	struct tun_struct *tun = from_timer(tun, t, flow_gc_timer);
J
Jason Wang 已提交
482 483 484 485 486 487 488
	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");

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

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

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

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

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

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

	rcu_read_lock();

528 529 530
	/* We may get a very small possibility of OOO during switching, not
	 * worth to optimize.*/
	if (tun->numqueues == 1 || tfile->detached)
J
Jason Wang 已提交
531 532 533 534 535 536 537
		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;
538
		sock_rps_record_flow_hash(e->rps_rxhash);
J
Jason Wang 已提交
539 540
	} else {
		spin_lock_bh(&tun->lock);
541 542
		if (!tun_flow_find(head, rxhash) &&
		    tun->flow_count < MAX_TAP_FLOWS)
J
Jason Wang 已提交
543 544 545 546 547 548 549 550 551 552 553 554
			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();
}

555 556 557 558 559 560
/**
 * 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)
{
561
	if (unlikely(e->rps_rxhash != hash))
562 563 564
		e->rps_rxhash = hash;
}

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

578
	numqueues = READ_ONCE(tun->numqueues);
J
Jason Wang 已提交
579

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

	return txq;
}

598 599
static u16 tun_ebpf_select_queue(struct tun_struct *tun, struct sk_buff *skb)
{
600
	struct tun_prog *prog;
601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625
	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;
}

626 627 628
static inline bool tun_not_capable(struct tun_struct *tun)
{
	const struct cred *cred = current_cred();
629
	struct net *net = dev_net(tun->dev);
630 631 632

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

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

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

J
Jason Wang 已提交
659 660 661 662 663 664 665 666 667 668 669 670 671
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);
	}
}

672 673
static void tun_queue_purge(struct tun_file *tfile)
{
J
Jason Wang 已提交
674
	void *ptr;
J
Jason Wang 已提交
675

J
Jason Wang 已提交
676 677
	while ((ptr = ptr_ring_consume(&tfile->tx_ring)) != NULL)
		tun_ptr_free(ptr);
J
Jason Wang 已提交
678

J
Jason Wang 已提交
679
	skb_queue_purge(&tfile->sk.sk_write_queue);
680 681 682
	skb_queue_purge(&tfile->sk.sk_error_queue);
}

683
static void tun_cleanup_tx_ring(struct tun_file *tfile)
684
{
685 686 687 688
	if (tfile->tx_ring.queue) {
		ptr_ring_cleanup(&tfile->tx_ring, tun_ptr_free);
		xdp_rxq_info_unreg(&tfile->xdp_rxq);
		memset(&tfile->tx_ring, 0, sizeof(tfile->tx_ring));
689 690 691
	}
}

J
Jason Wang 已提交
692 693 694 695 696
static void __tun_detach(struct tun_file *tfile, bool clean)
{
	struct tun_file *ntfile;
	struct tun_struct *tun;

697 698
	tun = rtnl_dereference(tfile->tun);

699 700 701 702 703
	if (tun && clean) {
		tun_napi_disable(tun, tfile);
		tun_napi_del(tun, tfile);
	}

704
	if (tun && !tfile->detached) {
J
Jason Wang 已提交
705 706 707 708 709
		u16 index = tfile->queue_index;
		BUG_ON(index >= tun->numqueues);

		rcu_assign_pointer(tun->tfiles[index],
				   tun->tfiles[tun->numqueues - 1]);
710
		ntfile = rtnl_dereference(tun->tfiles[index]);
J
Jason Wang 已提交
711 712 713
		ntfile->queue_index = index;

		--tun->numqueues;
714
		if (clean) {
715
			RCU_INIT_POINTER(tfile->tun, NULL);
J
Jason Wang 已提交
716
			sock_put(&tfile->sk);
717
		} else
J
Jason Wang 已提交
718
			tun_disable_queue(tun, tfile);
J
Jason Wang 已提交
719 720

		synchronize_net();
J
Jason Wang 已提交
721
		tun_flow_delete_by_queue(tun, tun->numqueues + 1);
J
Jason Wang 已提交
722
		/* Drop read queue */
723
		tun_queue_purge(tfile);
J
Jason Wang 已提交
724
		tun_set_real_num_queues(tun);
J
Jason Wang 已提交
725
	} else if (tfile->detached && clean) {
J
Jason Wang 已提交
726
		tun = tun_enable_queue(tfile);
J
Jason Wang 已提交
727 728
		sock_put(&tfile->sk);
	}
J
Jason Wang 已提交
729 730

	if (clean) {
731 732 733
		if (tun && tun->numqueues == 0 && tun->numdisabled == 0) {
			netif_carrier_off(tun->dev);

M
Michael S. Tsirkin 已提交
734
			if (!(tun->flags & IFF_PERSIST) &&
735
			    tun->dev->reg_state == NETREG_REGISTERED)
J
Jason Wang 已提交
736
				unregister_netdevice(tun->dev);
737
		}
738
		tun_cleanup_tx_ring(tfile);
739
		sock_put(&tfile->sk);
J
Jason Wang 已提交
740 741 742 743 744 745 746 747 748 749 750 751 752
	}
}

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 已提交
753
	struct tun_file *tfile, *tmp;
J
Jason Wang 已提交
754 755 756
	int i, n = tun->numqueues;

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

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

M
Michael S. Tsirkin 已提交
789
	if (tun->flags & IFF_PERSIST)
J
Jason Wang 已提交
790
		module_put(THIS_MODULE);
J
Jason Wang 已提交
791 792
}

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

800 801 802 803
	err = security_tun_dev_attach(tfile->socket.sk, tun->security);
	if (err < 0)
		goto out;

804
	err = -EINVAL;
805
	if (rtnl_dereference(tfile->tun) && !tfile->detached)
806 807 808
		goto out;

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

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

	err = 0;
J
Jason Wang 已提交
818

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

	if (!tfile->detached &&
829
	    ptr_ring_init(&tfile->tx_ring, dev->tx_queue_len, GFP_KERNEL)) {
J
Jason Wang 已提交
830 831 832 833
		err = -ENOMEM;
		goto out;
	}

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

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

852
	rcu_assign_pointer(tfile->tun, tun);
J
Jason Wang 已提交
853 854
	rcu_assign_pointer(tun->tfiles[tun->numqueues], tfile);
	tun->numqueues++;
855

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

J
Jason Wang 已提交
863
	tun_set_real_num_queues(tun);
864

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

out:
	return err;
E
Eric W. Biederman 已提交
871 872
}

873
static struct tun_struct *tun_get(struct tun_file *tfile)
E
Eric W. Biederman 已提交
874
{
875
	struct tun_struct *tun;
876

877 878 879 880 881
	rcu_read_lock();
	tun = rcu_dereference(tfile->tun);
	if (tun)
		dev_hold(tun->dev);
	rcu_read_unlock();
882 883

	return tun;
E
Eric W. Biederman 已提交
884 885 886 887
}

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

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

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

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

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

997
static const struct ethtool_ops tun_ethtool_ops;
L
Linus Torvalds 已提交
998

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

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

J
Jason Wang 已提交
1011
	netif_tx_start_all_queues(dev);
1012 1013 1014 1015 1016 1017 1018 1019

	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 已提交
1020 1021 1022 1023 1024 1025
	return 0;
}

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

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

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

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

	return len;
}

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

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

	/* Drop packet if interface is not attached */
1076
	if (txq >= tun->numqueues)
1077 1078 1079 1080
		goto drop;

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

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

J
Jason Wang 已提交
1084 1085
	BUG_ON(!tfile);

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

1096 1097 1098 1099 1100 1101 1102 1103 1104
	len = run_ebpf_filter(tun, skb, len);

	/* Trim extra bytes since we may insert vlan proto & TCI
	 * in tun_put_user().
	 */
	len -= skb_vlan_tag_present(skb) ? sizeof(struct veth) : 0;
	if (len <= 0 || pskb_trim(skb, len))
		goto drop;

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

1108
	skb_tx_timestamp(skb);
1109

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

1115 1116
	nf_reset(skb);

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

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

	rcu_read_unlock();
1126
	return NETDEV_TX_OK;
L
Linus Torvalds 已提交
1127 1128

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

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

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

	return (features & tun->set_features) | (features & ~TUN_USER_FEATURES);
}
1152 1153 1154 1155 1156 1157 1158
#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
1159 1160 1161
	 * 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.
1162 1163 1164
	 * 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
1165
	 * If NAPI is enabled, however, we need to schedule polling for all
1166 1167
	 * queues unless we are using napi_gro_frags(), which we call in
	 * process context and not in NAPI context.
1168
	 */
1169 1170 1171 1172 1173 1174
	struct tun_struct *tun = netdev_priv(dev);

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

1175 1176 1177
		if (tun_napi_frags_enabled(tun))
			return;

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

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

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

1262
static int tun_xdp(struct net_device *dev, struct netdev_bpf *xdp)
J
Jason Wang 已提交
1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275
{
	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 已提交
1276
static const struct net_device_ops tun_netdev_ops = {
1277
	.ndo_uninit		= tun_net_uninit,
S
Stephen Hemminger 已提交
1278 1279
	.ndo_open		= tun_net_open,
	.ndo_stop		= tun_net_close,
1280
	.ndo_start_xmit		= tun_net_xmit,
1281
	.ndo_fix_features	= tun_net_fix_features,
J
Jason Wang 已提交
1282
	.ndo_select_queue	= tun_select_queue,
1283 1284 1285
#ifdef CONFIG_NET_POLL_CONTROLLER
	.ndo_poll_controller	= tun_poll_controller,
#endif
1286
	.ndo_set_rx_headroom	= tun_set_headroom,
1287
	.ndo_get_stats64	= tun_net_get_stats64,
S
Stephen Hemminger 已提交
1288 1289
};

J
Jason Wang 已提交
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 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350
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 已提交
1351
static const struct net_device_ops tap_netdev_ops = {
1352
	.ndo_uninit		= tun_net_uninit,
S
Stephen Hemminger 已提交
1353 1354
	.ndo_open		= tun_net_open,
	.ndo_stop		= tun_net_close,
1355
	.ndo_start_xmit		= tun_net_xmit,
1356
	.ndo_fix_features	= tun_net_fix_features,
1357
	.ndo_set_rx_mode	= tun_net_mclist,
S
Stephen Hemminger 已提交
1358 1359
	.ndo_set_mac_address	= eth_mac_addr,
	.ndo_validate_addr	= eth_validate_addr,
J
Jason Wang 已提交
1360
	.ndo_select_queue	= tun_select_queue,
1361 1362 1363
#ifdef CONFIG_NET_POLL_CONTROLLER
	.ndo_poll_controller	= tun_poll_controller,
#endif
1364
	.ndo_features_check	= passthru_features_check,
1365
	.ndo_set_rx_headroom	= tun_set_headroom,
1366
	.ndo_get_stats64	= tun_net_get_stats64,
1367
	.ndo_bpf		= tun_xdp,
J
Jason Wang 已提交
1368 1369
	.ndo_xdp_xmit		= tun_xdp_xmit,
	.ndo_xdp_flush		= tun_xdp_flush,
S
Stephen Hemminger 已提交
1370 1371
};

1372
static void tun_flow_init(struct tun_struct *tun)
J
Jason Wang 已提交
1373 1374 1375 1376 1377 1378 1379
{
	int i;

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

	tun->ageing_time = TUN_FLOW_EXPIRE;
1380 1381 1382
	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 已提交
1383 1384 1385 1386 1387 1388 1389 1390
}

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

1391 1392 1393
#define MIN_MTU 68
#define MAX_MTU 65535

L
Linus Torvalds 已提交
1394 1395 1396 1397
/* Initialize net device. */
static void tun_net_init(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
1398

L
Linus Torvalds 已提交
1399
	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
1400
	case IFF_TUN:
S
Stephen Hemminger 已提交
1401 1402
		dev->netdev_ops = &tun_netdev_ops;

L
Linus Torvalds 已提交
1403 1404 1405 1406 1407 1408
		/* Point-to-Point TUN Device */
		dev->hard_header_len = 0;
		dev->addr_len = 0;
		dev->mtu = 1500;

		/* Zero header length */
1409
		dev->type = ARPHRD_NONE;
L
Linus Torvalds 已提交
1410 1411 1412
		dev->flags = IFF_POINTOPOINT | IFF_NOARP | IFF_MULTICAST;
		break;

M
Michael S. Tsirkin 已提交
1413
	case IFF_TAP:
K
Kusanagi Kouichi 已提交
1414
		dev->netdev_ops = &tap_netdev_ops;
L
Linus Torvalds 已提交
1415 1416
		/* Ethernet TAP Device */
		ether_setup(dev);
1417
		dev->priv_flags &= ~IFF_TX_SKB_SHARING;
1418
		dev->priv_flags |= IFF_LIVE_ADDR_CHANGE;
1419

1420
		eth_hw_addr_random(dev);
1421

L
Linus Torvalds 已提交
1422 1423
		break;
	}
1424 1425 1426

	dev->min_mtu = MIN_MTU;
	dev->max_mtu = MAX_MTU - dev->hard_header_len;
L
Linus Torvalds 已提交
1427 1428 1429 1430 1431
}

/* Character device part */

/* Poll */
1432
static __poll_t tun_chr_poll(struct file *file, poll_table *wait)
1433
{
1434
	struct tun_file *tfile = file->private_data;
1435
	struct tun_struct *tun = tun_get(tfile);
1436
	struct sock *sk;
1437
	__poll_t mask = 0;
L
Linus Torvalds 已提交
1438 1439

	if (!tun)
1440
		return POLLERR;
L
Linus Torvalds 已提交
1441

J
Jason Wang 已提交
1442
	sk = tfile->socket.sk;
1443

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

1446
	poll_wait(file, sk_sleep(sk), wait);
1447

1448
	if (!ptr_ring_empty(&tfile->tx_ring))
L
Linus Torvalds 已提交
1449 1450
		mask |= POLLIN | POLLRDNORM;

1451 1452 1453 1454
	if (tun->dev->flags & IFF_UP &&
	    (sock_writeable(sk) ||
	     (!test_and_set_bit(SOCKWQ_ASYNC_NOSPACE, &sk->sk_socket->flags) &&
	      sock_writeable(sk))))
1455 1456
		mask |= POLLOUT | POLLWRNORM;

1457 1458 1459
	if (tun->dev->reg_state != NETREG_REGISTERED)
		mask = POLLERR;

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

1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521
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);
}

1522 1523
/* 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 已提交
1524
static struct sk_buff *tun_alloc_skb(struct tun_file *tfile,
1525 1526
				     size_t prepad, size_t len,
				     size_t linear, int noblock)
1527
{
J
Jason Wang 已提交
1528
	struct sock *sk = tfile->socket.sk;
1529
	struct sk_buff *skb;
1530
	int err;
1531 1532

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

1536
	skb = sock_alloc_send_pskb(sk, prepad + linear, len - linear, noblock,
1537
				   &err, 0);
1538
	if (!skb)
1539
		return ERR_PTR(err);
1540 1541 1542

	skb_reserve(skb, prepad);
	skb_put(skb, linear);
1543 1544
	skb->data_len = len - linear;
	skb->len += len - linear;
1545 1546 1547 1548

	return skb;
}

J
Jason Wang 已提交
1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584
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();
	}
}

1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606
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 已提交
1607 1608
static struct sk_buff *tun_build_skb(struct tun_struct *tun,
				     struct tun_file *tfile,
1609
				     struct iov_iter *from,
J
Jason Wang 已提交
1610
				     struct virtio_net_hdr *hdr,
J
Jason Wang 已提交
1611
				     int len, int *skb_xdp)
1612
{
1613
	struct page_frag *alloc_frag = &current->task_frag;
1614
	struct sk_buff *skb;
J
Jason Wang 已提交
1615
	struct bpf_prog *xdp_prog;
1616
	int buflen = SKB_DATA_ALIGN(sizeof(struct skb_shared_info));
J
Jason Wang 已提交
1617
	unsigned int delta = 0;
1618 1619
	char *buf;
	size_t copied;
J
Jason Wang 已提交
1620
	bool xdp_xmit = false;
1621 1622 1623 1624 1625 1626 1627 1628
	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();
1629

1630
	alloc_frag->offset = ALIGN((u64)alloc_frag->offset, SMP_CACHE_BYTES);
1631 1632 1633 1634 1635
	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,
1636
				     alloc_frag->offset + pad,
1637 1638 1639 1640
				     len, from);
	if (copied != len)
		return ERR_PTR(-EFAULT);

1641 1642 1643 1644 1645
	/* 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 已提交
1646
		*skb_xdp = 1;
J
Jason Wang 已提交
1647
	else
J
Jason Wang 已提交
1648
		*skb_xdp = 0;
J
Jason Wang 已提交
1649 1650 1651

	rcu_read_lock();
	xdp_prog = rcu_dereference(tun->xdp_prog);
J
Jason Wang 已提交
1652
	if (xdp_prog && !*skb_xdp) {
J
Jason Wang 已提交
1653 1654 1655 1656 1657
		struct xdp_buff xdp;
		void *orig_data;
		u32 act;

		xdp.data_hard_start = buf;
1658
		xdp.data = buf + pad;
1659
		xdp_set_data_meta_invalid(&xdp);
J
Jason Wang 已提交
1660
		xdp.data_end = xdp.data + len;
J
Jesper Dangaard Brouer 已提交
1661
		xdp.rxq = &tfile->xdp_rxq;
J
Jason Wang 已提交
1662 1663 1664 1665 1666 1667 1668
		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;
J
Jason Wang 已提交
1669
			++tfile->xdp_pending_pkts;
J
Jason Wang 已提交
1670 1671 1672
			err = xdp_do_redirect(tun->dev, &xdp, xdp_prog);
			if (err)
				goto err_redirect;
1673
			rcu_read_unlock();
J
Jason Wang 已提交
1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691
			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;
		}
	}

1692
	skb = build_skb(buf, buflen);
J
Jason Wang 已提交
1693 1694
	if (!skb) {
		rcu_read_unlock();
1695
		return ERR_PTR(-ENOMEM);
J
Jason Wang 已提交
1696
	}
1697

1698
	skb_reserve(skb, pad - delta);
J
Jason Wang 已提交
1699
	skb_put(skb, len + delta);
1700 1701 1702
	get_page(alloc_frag->page);
	alloc_frag->offset += buflen;

J
Jason Wang 已提交
1703 1704 1705
	if (xdp_xmit) {
		skb->dev = tun->dev;
		generic_xdp_tx(skb, xdp_prog);
1706
		rcu_read_unlock();
J
Jason Wang 已提交
1707 1708 1709 1710 1711
		return NULL;
	}

	rcu_read_unlock();

1712
	return skb;
J
Jason Wang 已提交
1713 1714 1715 1716 1717 1718 1719

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

L
Linus Torvalds 已提交
1722
/* Get packet from user space buffer */
J
Jason Wang 已提交
1723
static ssize_t tun_get_user(struct tun_struct *tun, struct tun_file *tfile,
1724
			    void *msg_control, struct iov_iter *from,
J
Jason Wang 已提交
1725
			    int noblock, bool more)
L
Linus Torvalds 已提交
1726
{
1727
	struct tun_pi pi = { 0, cpu_to_be16(ETH_P_IP) };
L
Linus Torvalds 已提交
1728
	struct sk_buff *skb;
1729
	size_t total_len = iov_iter_count(from);
1730
	size_t len = total_len, align = tun->align, linear;
1731
	struct virtio_net_hdr gso = { 0 };
1732
	struct tun_pcpu_stats *stats;
1733
	int good_linear;
1734 1735 1736
	int copylen;
	bool zerocopy = false;
	int err;
1737
	u32 rxhash = 0;
J
Jason Wang 已提交
1738
	int skb_xdp = 1;
1739
	bool frags = tun_napi_frags_enabled(tun);
L
Linus Torvalds 已提交
1740

E
Eric Dumazet 已提交
1741 1742 1743
	if (!(tun->dev->flags & IFF_UP))
		return -EIO;

M
Michael S. Tsirkin 已提交
1744
	if (!(tun->flags & IFF_NO_PI)) {
1745
		if (len < sizeof(pi))
L
Linus Torvalds 已提交
1746
			return -EINVAL;
1747
		len -= sizeof(pi);
L
Linus Torvalds 已提交
1748

1749
		if (!copy_from_iter_full(&pi, sizeof(pi), from))
L
Linus Torvalds 已提交
1750 1751 1752
			return -EFAULT;
	}

M
Michael S. Tsirkin 已提交
1753
	if (tun->flags & IFF_VNET_HDR) {
W
Willem de Bruijn 已提交
1754 1755 1756
		int vnet_hdr_sz = READ_ONCE(tun->vnet_hdr_sz);

		if (len < vnet_hdr_sz)
1757
			return -EINVAL;
W
Willem de Bruijn 已提交
1758
		len -= vnet_hdr_sz;
1759

1760
		if (!copy_from_iter_full(&gso, sizeof(gso), from))
1761 1762
			return -EFAULT;

1763
		if ((gso.flags & VIRTIO_NET_HDR_F_NEEDS_CSUM) &&
1764 1765
		    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);
1766

1767
		if (tun16_to_cpu(tun, gso.hdr_len) > len)
1768
			return -EINVAL;
W
Willem de Bruijn 已提交
1769
		iov_iter_advance(from, vnet_hdr_sz - sizeof(gso));
1770 1771
	}

M
Michael S. Tsirkin 已提交
1772
	if ((tun->flags & TUN_TYPE_MASK) == IFF_TAP) {
1773
		align += NET_IP_ALIGN;
H
Herbert Xu 已提交
1774
		if (unlikely(len < ETH_HLEN ||
1775
			     (gso.hdr_len && tun16_to_cpu(tun, gso.hdr_len) < ETH_HLEN)))
1776 1777
			return -EINVAL;
	}
1778

1779 1780
	good_linear = SKB_MAX_HEAD(align);

1781
	if (msg_control) {
1782 1783
		struct iov_iter i = *from;

1784 1785
		/* There are 256 bytes to be copied in skb, so there is
		 * enough room for skb expand head in case it is used.
1786 1787
		 * The rest of the buffer is mapped from userspace.
		 */
1788
		copylen = gso.hdr_len ? tun16_to_cpu(tun, gso.hdr_len) : GOODCOPY_LEN;
1789 1790
		if (copylen > good_linear)
			copylen = good_linear;
1791
		linear = copylen;
1792 1793
		iov_iter_advance(&i, copylen);
		if (iov_iter_npages(&i, INT_MAX) <= MAX_SKB_FRAGS)
1794 1795 1796
			zerocopy = true;
	}

1797
	if (!frags && tun_can_build_skb(tun, tfile, len, noblock, zerocopy)) {
J
Jason Wang 已提交
1798 1799 1800 1801 1802
		/* 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);
1803
		if (IS_ERR(skb)) {
1804
			this_cpu_inc(tun->pcpu_stats->rx_dropped);
1805 1806
			return PTR_ERR(skb);
		}
J
Jason Wang 已提交
1807 1808
		if (!skb)
			return total_len;
1809 1810 1811 1812 1813 1814 1815 1816
	} 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 已提交
1817

1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830
		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);
		}

1831 1832 1833
		if (IS_ERR(skb)) {
			if (PTR_ERR(skb) != -EAGAIN)
				this_cpu_inc(tun->pcpu_stats->rx_dropped);
1834 1835
			if (frags)
				mutex_unlock(&tfile->napi_mutex);
1836 1837
			return PTR_ERR(skb);
		}
1838

1839 1840 1841 1842 1843 1844 1845 1846
		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);
1847 1848 1849 1850 1851
			if (frags) {
				tfile->napi.skb = NULL;
				mutex_unlock(&tfile->napi_mutex);
			}

1852 1853
			return -EFAULT;
		}
D
Dave Jones 已提交
1854
	}
L
Linus Torvalds 已提交
1855

1856
	if (virtio_net_hdr_to_skb(skb, &gso, tun_is_little_endian(tun))) {
1857 1858
		this_cpu_inc(tun->pcpu_stats->rx_frame_errors);
		kfree_skb(skb);
1859 1860 1861 1862 1863
		if (frags) {
			tfile->napi.skb = NULL;
			mutex_unlock(&tfile->napi_mutex);
		}

1864 1865 1866
		return -EINVAL;
	}

L
Linus Torvalds 已提交
1867
	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
1868 1869
	case IFF_TUN:
		if (tun->flags & IFF_NO_PI) {
1870 1871 1872 1873
			u8 ip_version = skb->len ? (skb->data[0] >> 4) : 0;

			switch (ip_version) {
			case 4:
1874 1875
				pi.proto = htons(ETH_P_IP);
				break;
1876
			case 6:
1877 1878 1879
				pi.proto = htons(ETH_P_IPV6);
				break;
			default:
1880
				this_cpu_inc(tun->pcpu_stats->rx_dropped);
1881 1882 1883 1884 1885
				kfree_skb(skb);
				return -EINVAL;
			}
		}

1886
		skb_reset_mac_header(skb);
L
Linus Torvalds 已提交
1887
		skb->protocol = pi.proto;
1888
		skb->dev = tun->dev;
L
Linus Torvalds 已提交
1889
		break;
M
Michael S. Tsirkin 已提交
1890
	case IFF_TAP:
1891 1892
		if (!frags)
			skb->protocol = eth_type_trans(skb, tun->dev);
L
Linus Torvalds 已提交
1893
		break;
1894
	}
L
Linus Torvalds 已提交
1895

1896 1897 1898 1899
	/* 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;
1900
		skb_shinfo(skb)->tx_flags |= SKBTX_SHARED_FRAG;
1901 1902 1903
	} else if (msg_control) {
		struct ubuf_info *uarg = msg_control;
		uarg->callback(uarg, false);
1904 1905
	}

1906
	skb_reset_network_header(skb);
1907
	skb_probe_transport_header(skb, 0);
1908

J
Jason Wang 已提交
1909
	if (skb_xdp) {
J
Jason Wang 已提交
1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924
		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();
	}

1925 1926 1927 1928
	rcu_read_lock();
	if (!rcu_dereference(tun->steering_prog))
		rxhash = __skb_get_hash_symmetric(skb);
	rcu_read_unlock();
1929

1930 1931 1932 1933
	if (frags) {
		/* Exercise flow dissector code path. */
		u32 headlen = eth_get_headlen(skb->data, skb_headlen(skb));

1934
		if (unlikely(headlen > skb_headlen(skb))) {
1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945
			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);
1946
	} else if (tfile->napi_enabled) {
1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963
		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);
	}
1964

1965 1966 1967 1968 1969 1970
	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 已提交
1971

1972 1973 1974
	if (rxhash)
		tun_flow_update(tun, rxhash, tfile);

1975
	return total_len;
1976
}
L
Linus Torvalds 已提交
1977

1978
static ssize_t tun_chr_write_iter(struct kiocb *iocb, struct iov_iter *from)
L
Linus Torvalds 已提交
1979
{
1980
	struct file *file = iocb->ki_filp;
J
Jason Wang 已提交
1981
	struct tun_file *tfile = file->private_data;
1982
	struct tun_struct *tun = tun_get(tfile);
E
Eric W. Biederman 已提交
1983
	ssize_t result;
L
Linus Torvalds 已提交
1984 1985 1986 1987

	if (!tun)
		return -EBADFD;

J
Jason Wang 已提交
1988 1989
	result = tun_get_user(tun, tfile, NULL, from,
			      file->f_flags & O_NONBLOCK, false);
E
Eric W. Biederman 已提交
1990

J
Jason Wang 已提交
1991 1992 1993 1994 1995
	if (tfile->xdp_pending_pkts) {
		tfile->xdp_pending_pkts = 0;
		xdp_do_flush_map();
	}

E
Eric W. Biederman 已提交
1996 1997
	tun_put(tun);
	return result;
L
Linus Torvalds 已提交
1998 1999
}

J
Jason Wang 已提交
2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033
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 已提交
2034
/* Put packet to the user space buffer */
2035
static ssize_t tun_put_user(struct tun_struct *tun,
J
Jason Wang 已提交
2036
			    struct tun_file *tfile,
2037
			    struct sk_buff *skb,
H
Herbert Xu 已提交
2038
			    struct iov_iter *iter)
L
Linus Torvalds 已提交
2039 2040
{
	struct tun_pi pi = { 0, skb->protocol };
2041
	struct tun_pcpu_stats *stats;
H
Herbert Xu 已提交
2042
	ssize_t total;
2043
	int vlan_offset = 0;
2044
	int vlan_hlen = 0;
H
Herbert Xu 已提交
2045
	int vnet_hdr_sz = 0;
2046

2047
	if (skb_vlan_tag_present(skb))
2048
		vlan_hlen = VLAN_HLEN;
L
Linus Torvalds 已提交
2049

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

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

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

H
Herbert Xu 已提交
2059 2060
		total += sizeof(pi);
		if (iov_iter_count(iter) < total) {
L
Linus Torvalds 已提交
2061 2062 2063
			/* Packet will be striped */
			pi.flags |= TUN_PKT_STRIP;
		}
2064

H
Herbert Xu 已提交
2065
		if (copy_to_iter(&pi, sizeof(pi), iter) != sizeof(pi))
L
Linus Torvalds 已提交
2066
			return -EFAULT;
2067
	}
L
Linus Torvalds 已提交
2068

H
Herbert Xu 已提交
2069
	if (vnet_hdr_sz) {
2070
		struct virtio_net_hdr gso;
2071

H
Herbert Xu 已提交
2072
		if (iov_iter_count(iter) < vnet_hdr_sz)
2073 2074
			return -EINVAL;

2075
		if (virtio_net_hdr_from_skb(skb, &gso,
2076
					    tun_is_little_endian(tun), true)) {
2077
			struct skb_shared_info *sinfo = skb_shinfo(skb);
2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088
			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;
		}
2089

H
Herbert Xu 已提交
2090
		if (copy_to_iter(&gso, sizeof(gso), iter) != sizeof(gso))
2091
			return -EFAULT;
2092 2093

		iov_iter_advance(iter, vnet_hdr_sz - sizeof(gso));
2094 2095
	}

2096
	if (vlan_hlen) {
H
Herbert Xu 已提交
2097
		int ret;
2098
		struct veth veth;
J
Jason Wang 已提交
2099 2100

		veth.h_vlan_proto = skb->vlan_proto;
2101
		veth.h_vlan_TCI = htons(skb_vlan_tag_get(skb));
J
Jason Wang 已提交
2102 2103 2104

		vlan_offset = offsetof(struct vlan_ethhdr, h_vlan_proto);

H
Herbert Xu 已提交
2105 2106
		ret = skb_copy_datagram_iter(skb, 0, iter, vlan_offset);
		if (ret || !iov_iter_count(iter))
J
Jason Wang 已提交
2107 2108
			goto done;

H
Herbert Xu 已提交
2109 2110
		ret = copy_to_iter(&veth, sizeof(veth), iter);
		if (ret != sizeof(veth) || !iov_iter_count(iter))
J
Jason Wang 已提交
2111 2112
			goto done;
	}
L
Linus Torvalds 已提交
2113

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

J
Jason Wang 已提交
2116
done:
2117 2118 2119 2120 2121 2122 2123
	/* 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 已提交
2124 2125 2126 2127

	return total;
}

J
Jason Wang 已提交
2128
static void *tun_ring_recv(struct tun_file *tfile, int noblock, int *err)
J
Jason Wang 已提交
2129 2130
{
	DECLARE_WAITQUEUE(wait, current);
J
Jason Wang 已提交
2131
	void *ptr = NULL;
J
Jason Wang 已提交
2132
	int error = 0;
J
Jason Wang 已提交
2133

J
Jason Wang 已提交
2134 2135
	ptr = ptr_ring_consume(&tfile->tx_ring);
	if (ptr)
J
Jason Wang 已提交
2136 2137
		goto out;
	if (noblock) {
J
Jason Wang 已提交
2138
		error = -EAGAIN;
J
Jason Wang 已提交
2139 2140 2141 2142 2143 2144 2145
		goto out;
	}

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

	while (1) {
J
Jason Wang 已提交
2146 2147
		ptr = ptr_ring_consume(&tfile->tx_ring);
		if (ptr)
J
Jason Wang 已提交
2148 2149
			break;
		if (signal_pending(current)) {
J
Jason Wang 已提交
2150
			error = -ERESTARTSYS;
J
Jason Wang 已提交
2151 2152 2153
			break;
		}
		if (tfile->socket.sk->sk_shutdown & RCV_SHUTDOWN) {
J
Jason Wang 已提交
2154
			error = -EFAULT;
J
Jason Wang 已提交
2155 2156 2157 2158 2159 2160 2161 2162 2163 2164
			break;
		}

		schedule();
	}

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

out:
J
Jason Wang 已提交
2165
	*err = error;
J
Jason Wang 已提交
2166
	return ptr;
J
Jason Wang 已提交
2167 2168
}

J
Jason Wang 已提交
2169
static ssize_t tun_do_read(struct tun_struct *tun, struct tun_file *tfile,
2170
			   struct iov_iter *to,
J
Jason Wang 已提交
2171
			   int noblock, void *ptr)
L
Linus Torvalds 已提交
2172
{
2173
	ssize_t ret;
J
Jason Wang 已提交
2174
	int err;
L
Linus Torvalds 已提交
2175

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

W
Wei Xu 已提交
2178
	if (!iov_iter_count(to)) {
J
Jason Wang 已提交
2179
		tun_ptr_free(ptr);
2180
		return 0;
W
Wei Xu 已提交
2181
	}
L
Linus Torvalds 已提交
2182

J
Jason Wang 已提交
2183
	if (!ptr) {
2184
		/* Read frames from ring */
J
Jason Wang 已提交
2185 2186
		ptr = tun_ring_recv(tfile, noblock, &err);
		if (!ptr)
2187 2188
			return err;
	}
H
Herbert Xu 已提交
2189

J
Jason Wang 已提交
2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203
	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 已提交
2204

2205 2206 2207
	return ret;
}

2208
static ssize_t tun_chr_read_iter(struct kiocb *iocb, struct iov_iter *to)
2209 2210 2211
{
	struct file *file = iocb->ki_filp;
	struct tun_file *tfile = file->private_data;
2212
	struct tun_struct *tun = tun_get(tfile);
2213
	ssize_t len = iov_iter_count(to), ret;
2214 2215 2216

	if (!tun)
		return -EBADFD;
2217
	ret = tun_do_read(tun, tfile, to, file->f_flags & O_NONBLOCK, NULL);
2218
	ret = min_t(ssize_t, ret, len);
Z
Zhi Yong Wu 已提交
2219 2220
	if (ret > 0)
		iocb->ki_pos = ret;
E
Eric W. Biederman 已提交
2221
	tun_put(tun);
L
Linus Torvalds 已提交
2222 2223 2224
	return ret;
}

2225
static void tun_prog_free(struct rcu_head *rcu)
2226
{
2227
	struct tun_prog *prog = container_of(rcu, struct tun_prog, rcu);
2228 2229 2230 2231 2232

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

J
Jason Wang 已提交
2233 2234
static int __tun_set_ebpf(struct tun_struct *tun,
			  struct tun_prog __rcu **prog_p,
2235
			  struct bpf_prog *prog)
2236
{
2237
	struct tun_prog *old, *new = NULL;
2238 2239 2240 2241 2242 2243 2244 2245

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

2246
	spin_lock_bh(&tun->lock);
2247
	old = rcu_dereference_protected(*prog_p,
2248
					lockdep_is_held(&tun->lock));
2249
	rcu_assign_pointer(*prog_p, new);
2250
	spin_unlock_bh(&tun->lock);
2251 2252

	if (old)
2253
		call_rcu(&old->rcu, tun_prog_free);
2254 2255 2256 2257

	return 0;
}

J
Jason Wang 已提交
2258 2259 2260 2261
static void tun_free_netdev(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

J
Jason Wang 已提交
2262
	BUG_ON(!(list_empty(&tun->disabled)));
2263
	free_percpu(tun->pcpu_stats);
J
Jason Wang 已提交
2264
	tun_flow_uninit(tun);
2265
	security_tun_dev_free_security(tun->security);
2266
	__tun_set_ebpf(tun, &tun->steering_prog, NULL);
2267
	__tun_set_ebpf(tun, &tun->filter_prog, NULL);
J
Jason Wang 已提交
2268 2269
}

L
Linus Torvalds 已提交
2270 2271 2272 2273
static void tun_setup(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

2274 2275
	tun->owner = INVALID_UID;
	tun->group = INVALID_GID;
L
Linus Torvalds 已提交
2276 2277

	dev->ethtool_ops = &tun_ethtool_ops;
2278 2279
	dev->needs_free_netdev = true;
	dev->priv_destructor = tun_free_netdev;
J
Jason Wang 已提交
2280 2281
	/* We prefer our own queue length */
	dev->tx_queue_len = TUN_READQ_SIZE;
L
Linus Torvalds 已提交
2282 2283
}

2284 2285 2286
/* Trivial set of netlink ops to allow deleting tun or tap
 * device with netlink.
 */
2287 2288
static int tun_validate(struct nlattr *tb[], struct nlattr *data[],
			struct netlink_ext_ack *extack)
2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299
{
	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,
};

2300 2301
static void tun_sock_write_space(struct sock *sk)
{
J
Jason Wang 已提交
2302
	struct tun_file *tfile;
2303
	wait_queue_head_t *wqueue;
2304 2305 2306 2307

	if (!sock_writeable(sk))
		return;

2308
	if (!test_and_clear_bit(SOCKWQ_ASYNC_NOSPACE, &sk->sk_socket->flags))
2309 2310
		return;

2311 2312 2313
	wqueue = sk_sleep(sk);
	if (wqueue && waitqueue_active(wqueue))
		wake_up_interruptible_sync_poll(wqueue, POLLOUT |
2314
						POLLWRNORM | POLLWRBAND);
H
Herbert Xu 已提交
2315

J
Jason Wang 已提交
2316 2317
	tfile = container_of(sk, struct tun_file, sk);
	kill_fasync(&tfile->fasync, SIGIO, POLL_OUT);
2318 2319
}

2320
static int tun_sendmsg(struct socket *sock, struct msghdr *m, size_t total_len)
2321
{
J
Jason Wang 已提交
2322 2323
	int ret;
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2324
	struct tun_struct *tun = tun_get(tfile);
J
Jason Wang 已提交
2325 2326 2327

	if (!tun)
		return -EBADFD;
2328

A
Al Viro 已提交
2329
	ret = tun_get_user(tun, tfile, m->msg_control, &m->msg_iter,
J
Jason Wang 已提交
2330 2331
			   m->msg_flags & MSG_DONTWAIT,
			   m->msg_flags & MSG_MORE);
J
Jason Wang 已提交
2332 2333 2334 2335 2336 2337 2338

	if (tfile->xdp_pending_pkts >= NAPI_POLL_WEIGHT ||
	    !(m->msg_flags & MSG_MORE)) {
		tfile->xdp_pending_pkts = 0;
		xdp_do_flush_map();
	}

J
Jason Wang 已提交
2339 2340
	tun_put(tun);
	return ret;
2341 2342
}

2343
static int tun_recvmsg(struct socket *sock, struct msghdr *m, size_t total_len,
2344 2345
		       int flags)
{
J
Jason Wang 已提交
2346
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2347
	struct tun_struct *tun = tun_get(tfile);
J
Jason Wang 已提交
2348
	void *ptr = m->msg_control;
2349
	int ret;
J
Jason Wang 已提交
2350

W
Wei Xu 已提交
2351 2352
	if (!tun) {
		ret = -EBADFD;
J
Jason Wang 已提交
2353
		goto out_free;
W
Wei Xu 已提交
2354
	}
J
Jason Wang 已提交
2355

2356
	if (flags & ~(MSG_DONTWAIT|MSG_TRUNC|MSG_ERRQUEUE)) {
2357
		ret = -EINVAL;
W
Wei Xu 已提交
2358
		goto out_put_tun;
2359
	}
2360 2361 2362 2363 2364
	if (flags & MSG_ERRQUEUE) {
		ret = sock_recv_errqueue(sock->sk, m, total_len,
					 SOL_PACKET, TUN_TX_TIMESTAMP);
		goto out;
	}
J
Jason Wang 已提交
2365
	ret = tun_do_read(tun, tfile, &m->msg_iter, flags & MSG_DONTWAIT, ptr);
2366
	if (ret > (ssize_t)total_len) {
2367 2368 2369
		m->msg_flags |= MSG_TRUNC;
		ret = flags & MSG_TRUNC ? ret : total_len;
	}
2370
out:
J
Jason Wang 已提交
2371
	tun_put(tun);
2372
	return ret;
W
Wei Xu 已提交
2373 2374 2375

out_put_tun:
	tun_put(tun);
J
Jason Wang 已提交
2376 2377
out_free:
	tun_ptr_free(ptr);
W
Wei Xu 已提交
2378
	return ret;
2379 2380
}

J
Jason Wang 已提交
2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394
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 已提交
2395 2396 2397 2398 2399 2400
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;

2401
	tun = tun_get(tfile);
J
Jason Wang 已提交
2402 2403 2404
	if (!tun)
		return 0;

J
Jason Wang 已提交
2405
	ret = PTR_RING_PEEK_CALL(&tfile->tx_ring, tun_ptr_peek_len);
J
Jason Wang 已提交
2406 2407 2408 2409 2410
	tun_put(tun);

	return ret;
}

2411 2412
/* Ops structure to mimic raw sockets with tun */
static const struct proto_ops tun_socket_ops = {
J
Jason Wang 已提交
2413
	.peek_len = tun_peek_len,
2414 2415 2416 2417
	.sendmsg = tun_sendmsg,
	.recvmsg = tun_recvmsg,
};

2418 2419 2420
static struct proto tun_proto = {
	.name		= "tun",
	.owner		= THIS_MODULE,
J
Jason Wang 已提交
2421
	.obj_size	= sizeof(struct tun_file),
2422
};
2423

2424 2425
static int tun_flags(struct tun_struct *tun)
{
2426
	return tun->flags & (TUN_FEATURES | IFF_PERSIST | IFF_TUN | IFF_TAP);
2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439
}

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));
2440 2441 2442 2443
	return uid_valid(tun->owner)?
		sprintf(buf, "%u\n",
			from_kuid_munged(current_user_ns(), tun->owner)):
		sprintf(buf, "-1\n");
2444 2445 2446 2447 2448 2449
}

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));
2450 2451 2452 2453
	return gid_valid(tun->group) ?
		sprintf(buf, "%u\n",
			from_kgid_munged(current_user_ns(), tun->group)):
		sprintf(buf, "-1\n");
2454 2455 2456 2457 2458 2459
}

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

2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470
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
};

2471
static int tun_set_iff(struct net *net, struct file *file, struct ifreq *ifr)
L
Linus Torvalds 已提交
2472 2473
{
	struct tun_struct *tun;
J
Jason Wang 已提交
2474
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
2475 2476 2477
	struct net_device *dev;
	int err;

2478 2479 2480
	if (tfile->detached)
		return -EINVAL;

2481 2482 2483 2484 2485 2486 2487 2488 2489
	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;
	}

2490 2491
	dev = __dev_get_by_name(net, ifr->ifr_name);
	if (dev) {
2492 2493
		if (ifr->ifr_flags & IFF_TUN_EXCL)
			return -EBUSY;
2494 2495 2496 2497 2498 2499 2500
		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;

2501
		if (!!(ifr->ifr_flags & IFF_MULTI_QUEUE) !=
M
Michael S. Tsirkin 已提交
2502
		    !!(tun->flags & IFF_MULTI_QUEUE))
2503 2504
			return -EINVAL;

2505
		if (tun_not_capable(tun))
P
Paul Moore 已提交
2506
			return -EPERM;
2507
		err = security_tun_dev_open(tun->security);
P
Paul Moore 已提交
2508 2509 2510
		if (err < 0)
			return err;

2511 2512
		err = tun_attach(tun, file, ifr->ifr_flags & IFF_NOFILTER,
				 ifr->ifr_flags & IFF_NAPI);
2513 2514
		if (err < 0)
			return err;
J
Jason Wang 已提交
2515

M
Michael S. Tsirkin 已提交
2516
		if (tun->flags & IFF_MULTI_QUEUE &&
2517 2518 2519 2520 2521 2522
		    (tun->numqueues + tun->numdisabled > 1)) {
			/* One or more queue has already been attached, no need
			 * to initialize the device again.
			 */
			return 0;
		}
2523
	}
L
Linus Torvalds 已提交
2524 2525 2526
	else {
		char *name;
		unsigned long flags = 0;
2527 2528
		int queues = ifr->ifr_flags & IFF_MULTI_QUEUE ?
			     MAX_TAP_QUEUES : 1;
L
Linus Torvalds 已提交
2529

2530
		if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
2531
			return -EPERM;
P
Paul Moore 已提交
2532 2533 2534
		err = security_tun_dev_create();
		if (err < 0)
			return err;
2535

L
Linus Torvalds 已提交
2536 2537 2538
		/* Set dev type */
		if (ifr->ifr_flags & IFF_TUN) {
			/* TUN device */
M
Michael S. Tsirkin 已提交
2539
			flags |= IFF_TUN;
L
Linus Torvalds 已提交
2540 2541 2542
			name = "tun%d";
		} else if (ifr->ifr_flags & IFF_TAP) {
			/* TAP device */
M
Michael S. Tsirkin 已提交
2543
			flags |= IFF_TAP;
L
Linus Torvalds 已提交
2544
			name = "tap%d";
2545
		} else
2546
			return -EINVAL;
2547

L
Linus Torvalds 已提交
2548 2549 2550
		if (*ifr->ifr_name)
			name = ifr->ifr_name;

J
Jason Wang 已提交
2551
		dev = alloc_netdev_mqs(sizeof(struct tun_struct), name,
2552 2553
				       NET_NAME_UNKNOWN, tun_setup, queues,
				       queues);
2554

L
Linus Torvalds 已提交
2555 2556
		if (!dev)
			return -ENOMEM;
2557
		err = dev_get_valid_name(net, dev, name);
2558
		if (err < 0)
2559
			goto err_free_dev;
L
Linus Torvalds 已提交
2560

2561
		dev_net_set(dev, net);
2562
		dev->rtnl_link_ops = &tun_link_ops;
2563
		dev->ifindex = tfile->ifindex;
2564
		dev->sysfs_groups[0] = &tun_attr_group;
S
Stephen Hemminger 已提交
2565

L
Linus Torvalds 已提交
2566 2567 2568
		tun = netdev_priv(dev);
		tun->dev = dev;
		tun->flags = flags;
2569
		tun->txflt.count = 0;
2570
		tun->vnet_hdr_sz = sizeof(struct virtio_net_hdr);
2571

2572
		tun->align = NET_SKB_PAD;
J
Jason Wang 已提交
2573 2574
		tun->filter_attached = false;
		tun->sndbuf = tfile->socket.sk->sk_sndbuf;
J
Jason Wang 已提交
2575
		tun->rx_batched = 0;
2576
		RCU_INIT_POINTER(tun->steering_prog, NULL);
2577

2578 2579 2580 2581 2582 2583
		tun->pcpu_stats = netdev_alloc_pcpu_stats(struct tun_pcpu_stats);
		if (!tun->pcpu_stats) {
			err = -ENOMEM;
			goto err_free_dev;
		}

J
Jason Wang 已提交
2584 2585
		spin_lock_init(&tun->lock);

2586 2587
		err = security_tun_dev_alloc_security(&tun->security);
		if (err < 0)
2588
			goto err_free_stat;
P
Paul Moore 已提交
2589

L
Linus Torvalds 已提交
2590
		tun_net_init(dev);
2591
		tun_flow_init(tun);
J
Jason Wang 已提交
2592

2593
		dev->hw_features = NETIF_F_SG | NETIF_F_FRAGLIST |
J
Jason Wang 已提交
2594 2595
				   TUN_USER_FEATURES | NETIF_F_HW_VLAN_CTAG_TX |
				   NETIF_F_HW_VLAN_STAG_TX;
2596
		dev->features = dev->hw_features | NETIF_F_LLTX;
2597 2598 2599
		dev->vlan_features = dev->features &
				     ~(NETIF_F_HW_VLAN_CTAG_TX |
				       NETIF_F_HW_VLAN_STAG_TX);
2600

J
Jason Wang 已提交
2601
		INIT_LIST_HEAD(&tun->disabled);
2602
		err = tun_attach(tun, file, false, ifr->ifr_flags & IFF_NAPI);
2603
		if (err < 0)
2604
			goto err_free_flow;
2605

L
Linus Torvalds 已提交
2606 2607
		err = register_netdevice(tun->dev);
		if (err < 0)
2608
			goto err_detach;
L
Linus Torvalds 已提交
2609 2610
	}

2611 2612
	netif_carrier_on(tun->dev);

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

2615 2616
	tun->flags = (tun->flags & ~TUN_FEATURES) |
		(ifr->ifr_flags & TUN_FEATURES);
J
Jason Wang 已提交
2617

2618 2619 2620 2621
	/* Make sure persistent devices do not get stuck in
	 * xoff state.
	 */
	if (netif_running(tun->dev))
J
Jason Wang 已提交
2622
		netif_tx_wake_all_queues(tun->dev);
2623

L
Linus Torvalds 已提交
2624 2625 2626
	strcpy(ifr->ifr_name, tun->dev->name);
	return 0;

2627 2628
err_detach:
	tun_detach_all(dev);
2629 2630 2631
	/* register_netdevice() already called tun_free_netdev() */
	goto err_free_dev;

2632 2633 2634
err_free_flow:
	tun_flow_uninit(tun);
	security_tun_dev_free_security(tun->security);
2635 2636
err_free_stat:
	free_percpu(tun->pcpu_stats);
2637
err_free_dev:
L
Linus Torvalds 已提交
2638 2639 2640 2641
	free_netdev(dev);
	return err;
}

R
Rami Rosen 已提交
2642
static void tun_get_iff(struct net *net, struct tun_struct *tun,
2643
		       struct ifreq *ifr)
2644
{
2645
	tun_debug(KERN_INFO, tun, "tun_get_iff\n");
2646 2647 2648

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

2649
	ifr->ifr_flags = tun_flags(tun);
2650 2651 2652

}

2653 2654
/* This is like a cut-down ethtool ops, except done via tun fd so no
 * privs required. */
2655
static int set_offload(struct tun_struct *tun, unsigned long arg)
2656
{
2657
	netdev_features_t features = 0;
2658 2659

	if (arg & TUN_F_CSUM) {
2660
		features |= NETIF_F_HW_CSUM;
2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671 2672 2673
		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);
		}
2674 2675

		arg &= ~TUN_F_UFO;
2676 2677 2678 2679 2680 2681 2682
	}

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

2683
	tun->set_features = features;
2684 2685
	tun->dev->wanted_features &= ~TUN_USER_FEATURES;
	tun->dev->wanted_features |= features;
2686
	netdev_update_features(tun->dev);
2687 2688 2689 2690

	return 0;
}

J
Jason Wang 已提交
2691 2692 2693 2694 2695 2696
static void tun_detach_filter(struct tun_struct *tun, int n)
{
	int i;
	struct tun_file *tfile;

	for (i = 0; i < n; i++) {
2697
		tfile = rtnl_dereference(tun->tfiles[i]);
2698 2699 2700
		lock_sock(tfile->socket.sk);
		sk_detach_filter(tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711
	}

	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++) {
2712
		tfile = rtnl_dereference(tun->tfiles[i]);
2713 2714 2715
		lock_sock(tfile->socket.sk);
		ret = sk_attach_filter(&tun->fprog, tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2716 2717 2718 2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731
		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++) {
2732
		tfile = rtnl_dereference(tun->tfiles[i]);
J
Jason Wang 已提交
2733 2734 2735 2736
		tfile->socket.sk->sk_sndbuf = tun->sndbuf;
	}
}

2737 2738 2739 2740 2741 2742 2743 2744 2745
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 已提交
2746
		tun = tfile->detached;
2747
		if (!tun) {
2748
			ret = -EINVAL;
2749 2750 2751 2752 2753
			goto unlock;
		}
		ret = security_tun_dev_attach_queue(tun->security);
		if (ret < 0)
			goto unlock;
2754
		ret = tun_attach(tun, file, false, tun->flags & IFF_NAPI);
J
Jason Wang 已提交
2755
	} else if (ifr->ifr_flags & IFF_DETACH_QUEUE) {
2756
		tun = rtnl_dereference(tfile->tun);
M
Michael S. Tsirkin 已提交
2757
		if (!tun || !(tun->flags & IFF_MULTI_QUEUE) || tfile->detached)
J
Jason Wang 已提交
2758 2759 2760 2761
			ret = -EINVAL;
		else
			__tun_detach(tfile, false);
	} else
2762 2763
		ret = -EINVAL;

2764
unlock:
2765 2766 2767 2768
	rtnl_unlock();
	return ret;
}

2769 2770
static int tun_set_ebpf(struct tun_struct *tun, struct tun_prog **prog_p,
			void __user *data)
2771 2772 2773 2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785
{
	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);
	}

2786
	return __tun_set_ebpf(tun, prog_p, prog);
2787 2788
}

2789 2790
static long __tun_chr_ioctl(struct file *file, unsigned int cmd,
			    unsigned long arg, int ifreq_len)
L
Linus Torvalds 已提交
2791
{
E
Eric W. Biederman 已提交
2792
	struct tun_file *tfile = file->private_data;
E
Eric W. Biederman 已提交
2793
	struct tun_struct *tun;
L
Linus Torvalds 已提交
2794 2795
	void __user* argp = (void __user*)arg;
	struct ifreq ifr;
2796 2797
	kuid_t owner;
	kgid_t group;
2798
	int sndbuf;
2799
	int vnet_hdr_sz;
2800
	unsigned int ifindex;
M
Michael S. Tsirkin 已提交
2801
	int le;
2802
	int ret;
L
Linus Torvalds 已提交
2803

2804
	if (cmd == TUNSETIFF || cmd == TUNSETQUEUE || _IOC_TYPE(cmd) == SOCK_IOC_TYPE) {
2805
		if (copy_from_user(&ifr, argp, ifreq_len))
L
Linus Torvalds 已提交
2806
			return -EFAULT;
D
David S. Miller 已提交
2807
	} else {
2808
		memset(&ifr, 0, sizeof(ifr));
D
David S. Miller 已提交
2809
	}
E
Eric W. Biederman 已提交
2810 2811 2812
	if (cmd == TUNGETFEATURES) {
		/* Currently this just means: "what IFF flags are valid?".
		 * This is needed because we never checked for invalid flags on
2813 2814 2815
		 * TUNSETIFF.
		 */
		return put_user(IFF_TUN | IFF_TAP | TUN_FEATURES,
E
Eric W. Biederman 已提交
2816
				(unsigned int __user*)argp);
2817 2818
	} else if (cmd == TUNSETQUEUE)
		return tun_set_queue(file, &ifr);
E
Eric W. Biederman 已提交
2819

J
Jason Wang 已提交
2820
	ret = 0;
2821 2822
	rtnl_lock();

2823
	tun = tun_get(tfile);
2824 2825 2826 2827 2828
	if (cmd == TUNSETIFF) {
		ret = -EEXIST;
		if (tun)
			goto unlock;

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

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

2833 2834
		if (ret)
			goto unlock;
L
Linus Torvalds 已提交
2835

2836
		if (copy_to_user(argp, &ifr, ifreq_len))
2837 2838
			ret = -EFAULT;
		goto unlock;
L
Linus Torvalds 已提交
2839
	}
2840 2841 2842 2843 2844 2845 2846 2847 2848 2849 2850 2851 2852
	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 已提交
2853

2854
	ret = -EBADFD;
L
Linus Torvalds 已提交
2855
	if (!tun)
2856
		goto unlock;
L
Linus Torvalds 已提交
2857

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

E
Eric W. Biederman 已提交
2860
	ret = 0;
L
Linus Torvalds 已提交
2861
	switch (cmd) {
2862
	case TUNGETIFF:
R
Rami Rosen 已提交
2863
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
2864

2865 2866
		if (tfile->detached)
			ifr.ifr_flags |= IFF_DETACH_QUEUE;
2867 2868
		if (!tfile->socket.sk->sk_filter)
			ifr.ifr_flags |= IFF_NOFILTER;
2869

2870
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
2871
			ret = -EFAULT;
2872 2873
		break;

L
Linus Torvalds 已提交
2874 2875 2876
	case TUNSETNOCSUM:
		/* Disable/Enable checksum */

2877 2878
		/* [unimplemented] */
		tun_debug(KERN_INFO, tun, "ignored: set checksum %s\n",
2879
			  arg ? "disabled" : "enabled");
L
Linus Torvalds 已提交
2880 2881 2882
		break;

	case TUNSETPERSIST:
J
Jason Wang 已提交
2883 2884 2885
		/* Disable/Enable persist mode. Keep an extra reference to the
		 * module to prevent the module being unprobed.
		 */
M
Michael S. Tsirkin 已提交
2886 2887
		if (arg && !(tun->flags & IFF_PERSIST)) {
			tun->flags |= IFF_PERSIST;
J
Jason Wang 已提交
2888
			__module_get(THIS_MODULE);
J
Jason Wang 已提交
2889
		}
M
Michael S. Tsirkin 已提交
2890 2891
		if (!arg && (tun->flags & IFF_PERSIST)) {
			tun->flags &= ~IFF_PERSIST;
J
Jason Wang 已提交
2892 2893
			module_put(THIS_MODULE);
		}
L
Linus Torvalds 已提交
2894

2895 2896
		tun_debug(KERN_INFO, tun, "persist %s\n",
			  arg ? "enabled" : "disabled");
L
Linus Torvalds 已提交
2897 2898 2899 2900
		break;

	case TUNSETOWNER:
		/* Set owner of the device */
2901 2902 2903 2904 2905 2906
		owner = make_kuid(current_user_ns(), arg);
		if (!uid_valid(owner)) {
			ret = -EINVAL;
			break;
		}
		tun->owner = owner;
2907
		tun_debug(KERN_INFO, tun, "owner set to %u\n",
2908
			  from_kuid(&init_user_ns, tun->owner));
L
Linus Torvalds 已提交
2909 2910
		break;

2911 2912
	case TUNSETGROUP:
		/* Set group of the device */
2913 2914 2915 2916 2917 2918
		group = make_kgid(current_user_ns(), arg);
		if (!gid_valid(group)) {
			ret = -EINVAL;
			break;
		}
		tun->group = group;
2919
		tun_debug(KERN_INFO, tun, "group set to %u\n",
2920
			  from_kgid(&init_user_ns, tun->group));
2921 2922
		break;

2923 2924 2925
	case TUNSETLINK:
		/* Only allow setting the type when the interface is down */
		if (tun->dev->flags & IFF_UP) {
2926 2927
			tun_debug(KERN_INFO, tun,
				  "Linktype set failed because interface is up\n");
2928
			ret = -EBUSY;
2929 2930
		} else {
			tun->dev->type = (int) arg;
2931 2932
			tun_debug(KERN_INFO, tun, "linktype set to %d\n",
				  tun->dev->type);
2933
			ret = 0;
2934
		}
E
Eric W. Biederman 已提交
2935
		break;
2936

L
Linus Torvalds 已提交
2937 2938 2939 2940 2941
#ifdef TUN_DEBUG
	case TUNSETDEBUG:
		tun->debug = arg;
		break;
#endif
2942
	case TUNSETOFFLOAD:
2943
		ret = set_offload(tun, arg);
E
Eric W. Biederman 已提交
2944
		break;
2945

2946 2947
	case TUNSETTXFILTER:
		/* Can be set only for TAPs */
E
Eric W. Biederman 已提交
2948
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
2949
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
E
Eric W. Biederman 已提交
2950
			break;
H
Harvey Harrison 已提交
2951
		ret = update_filter(&tun->txflt, (void __user *)arg);
E
Eric W. Biederman 已提交
2952
		break;
L
Linus Torvalds 已提交
2953 2954

	case SIOCGIFHWADDR:
2955
		/* Get hw address */
2956 2957
		memcpy(ifr.ifr_hwaddr.sa_data, tun->dev->dev_addr, ETH_ALEN);
		ifr.ifr_hwaddr.sa_family = tun->dev->type;
2958
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
2959 2960
			ret = -EFAULT;
		break;
L
Linus Torvalds 已提交
2961 2962

	case SIOCSIFHWADDR:
2963
		/* Set hw address */
2964 2965
		tun_debug(KERN_DEBUG, tun, "set hw address: %pM\n",
			  ifr.ifr_hwaddr.sa_data);
2966 2967

		ret = dev_set_mac_address(tun->dev, &ifr.ifr_hwaddr);
E
Eric W. Biederman 已提交
2968
		break;
2969 2970

	case TUNGETSNDBUF:
J
Jason Wang 已提交
2971
		sndbuf = tfile->socket.sk->sk_sndbuf;
2972 2973 2974 2975 2976 2977 2978 2979 2980
		if (copy_to_user(argp, &sndbuf, sizeof(sndbuf)))
			ret = -EFAULT;
		break;

	case TUNSETSNDBUF:
		if (copy_from_user(&sndbuf, argp, sizeof(sndbuf))) {
			ret = -EFAULT;
			break;
		}
2981 2982 2983 2984
		if (sndbuf <= 0) {
			ret = -EINVAL;
			break;
		}
2985

J
Jason Wang 已提交
2986 2987
		tun->sndbuf = sndbuf;
		tun_set_sndbuf(tun);
2988 2989
		break;

2990 2991 2992 2993 2994 2995 2996 2997 2998 2999 3000 3001 3002 3003 3004 3005 3006 3007 3008
	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 已提交
3009 3010 3011 3012 3013 3014 3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025
	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;

3026 3027 3028 3029 3030 3031 3032 3033
	case TUNGETVNETBE:
		ret = tun_get_vnet_be(tun, argp);
		break;

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

M
Michael S. Tsirkin 已提交
3034 3035 3036
	case TUNATTACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3037
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
3038 3039
			break;
		ret = -EFAULT;
J
Jason Wang 已提交
3040
		if (copy_from_user(&tun->fprog, argp, sizeof(tun->fprog)))
M
Michael S. Tsirkin 已提交
3041 3042
			break;

J
Jason Wang 已提交
3043
		ret = tun_attach_filter(tun);
M
Michael S. Tsirkin 已提交
3044 3045 3046 3047 3048
		break;

	case TUNDETACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3049
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
3050
			break;
J
Jason Wang 已提交
3051 3052
		ret = 0;
		tun_detach_filter(tun, tun->numqueues);
M
Michael S. Tsirkin 已提交
3053 3054
		break;

P
Pavel Emelyanov 已提交
3055 3056
	case TUNGETFILTER:
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3057
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
P
Pavel Emelyanov 已提交
3058 3059 3060 3061 3062 3063 3064
			break;
		ret = -EFAULT;
		if (copy_to_user(argp, &tun->fprog, sizeof(tun->fprog)))
			break;
		ret = 0;
		break;

3065
	case TUNSETSTEERINGEBPF:
3066
		ret = tun_set_ebpf(tun, &tun->steering_prog, argp);
3067 3068
		break;

3069 3070 3071 3072
	case TUNSETFILTEREBPF:
		ret = tun_set_ebpf(tun, &tun->filter_prog, argp);
		break;

L
Linus Torvalds 已提交
3073
	default:
E
Eric W. Biederman 已提交
3074 3075
		ret = -EINVAL;
		break;
3076
	}
L
Linus Torvalds 已提交
3077

3078 3079 3080 3081
unlock:
	rtnl_unlock();
	if (tun)
		tun_put(tun);
E
Eric W. Biederman 已提交
3082
	return ret;
L
Linus Torvalds 已提交
3083 3084
}

3085 3086 3087 3088 3089 3090 3091 3092 3093 3094 3095 3096 3097 3098 3099 3100 3101 3102 3103 3104 3105 3106 3107 3108 3109 3110 3111 3112 3113 3114 3115 3116 3117 3118 3119
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 已提交
3120 3121
static int tun_chr_fasync(int fd, struct file *file, int on)
{
J
Jason Wang 已提交
3122
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
3123 3124
	int ret;

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

L
Linus Torvalds 已提交
3128
	if (on) {
3129
		__f_setown(file, task_pid(current), PIDTYPE_PID, 0);
J
Jason Wang 已提交
3130
		tfile->flags |= TUN_FASYNC;
3131
	} else
J
Jason Wang 已提交
3132
		tfile->flags &= ~TUN_FASYNC;
J
Jonathan Corbet 已提交
3133 3134 3135
	ret = 0;
out:
	return ret;
L
Linus Torvalds 已提交
3136 3137 3138 3139
}

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

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

3145
	tfile = (struct tun_file *)sk_alloc(net, AF_UNSPEC, GFP_KERNEL,
3146
					    &tun_proto, 0);
E
Eric W. Biederman 已提交
3147 3148
	if (!tfile)
		return -ENOMEM;
3149
	RCU_INIT_POINTER(tfile->tun, NULL);
J
Jason Wang 已提交
3150
	tfile->flags = 0;
3151
	tfile->ifindex = 0;
J
Jason Wang 已提交
3152 3153

	init_waitqueue_head(&tfile->wq.wait);
3154
	RCU_INIT_POINTER(tfile->socket.wq, &tfile->wq);
J
Jason Wang 已提交
3155 3156 3157 3158 3159 3160 3161 3162 3163

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

3167 3168
	sock_set_flag(&tfile->sk, SOCK_ZEROCOPY);

3169
	memset(&tfile->tx_ring, 0, sizeof(tfile->tx_ring));
J
Jason Wang 已提交
3170
	tfile->xdp_pending_pkts = 0;
3171

L
Linus Torvalds 已提交
3172 3173 3174 3175 3176
	return 0;
}

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

J
Jason Wang 已提交
3179
	tun_detach(tfile, true);
L
Linus Torvalds 已提交
3180 3181 3182 3183

	return 0;
}

3184
#ifdef CONFIG_PROC_FS
3185
static void tun_chr_show_fdinfo(struct seq_file *m, struct file *file)
3186
{
3187
	struct tun_file *tfile = file->private_data;
3188 3189 3190 3191 3192 3193
	struct tun_struct *tun;
	struct ifreq ifr;

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

	rtnl_lock();
3194
	tun = tun_get(tfile);
3195 3196 3197 3198 3199 3200 3201
	if (tun)
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
	rtnl_unlock();

	if (tun)
		tun_put(tun);

3202
	seq_printf(m, "iff:\t%s\n", ifr.ifr_name);
3203 3204 3205
}
#endif

3206
static const struct file_operations tun_fops = {
3207
	.owner	= THIS_MODULE,
L
Linus Torvalds 已提交
3208
	.llseek = no_llseek,
3209
	.read_iter  = tun_chr_read_iter,
3210
	.write_iter = tun_chr_write_iter,
L
Linus Torvalds 已提交
3211
	.poll	= tun_chr_poll,
3212 3213 3214 3215
	.unlocked_ioctl	= tun_chr_ioctl,
#ifdef CONFIG_COMPAT
	.compat_ioctl = tun_chr_compat_ioctl,
#endif
L
Linus Torvalds 已提交
3216 3217
	.open	= tun_chr_open,
	.release = tun_chr_close,
3218 3219 3220 3221
	.fasync = tun_chr_fasync,
#ifdef CONFIG_PROC_FS
	.show_fdinfo = tun_chr_show_fdinfo,
#endif
L
Linus Torvalds 已提交
3222 3223 3224 3225 3226
};

static struct miscdevice tun_miscdev = {
	.minor = TUN_MINOR,
	.name = "tun",
3227
	.nodename = "net/tun",
L
Linus Torvalds 已提交
3228 3229 3230 3231 3232
	.fops = &tun_fops,
};

/* ethtool interface */

3233 3234 3235 3236 3237 3238 3239 3240 3241 3242
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 已提交
3243 3244 3245 3246 3247 3248 3249
	return 0;
}

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

3250 3251
	strlcpy(info->driver, DRV_NAME, sizeof(info->driver));
	strlcpy(info->version, DRV_VERSION, sizeof(info->version));
L
Linus Torvalds 已提交
3252 3253

	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
3254
	case IFF_TUN:
3255
		strlcpy(info->bus_info, "tun", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3256
		break;
M
Michael S. Tsirkin 已提交
3257
	case IFF_TAP:
3258
		strlcpy(info->bus_info, "tap", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3259 3260 3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280
		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 已提交
3281 3282 3283 3284 3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297 3298 3299 3300 3301 3302 3303
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;
}

3304
static const struct ethtool_ops tun_ethtool_ops = {
L
Linus Torvalds 已提交
3305 3306 3307
	.get_drvinfo	= tun_get_drvinfo,
	.get_msglevel	= tun_get_msglevel,
	.set_msglevel	= tun_set_msglevel,
3308
	.get_link	= ethtool_op_get_link,
3309
	.get_ts_info	= ethtool_op_get_ts_info,
J
Jason Wang 已提交
3310 3311
	.get_coalesce   = tun_get_coalesce,
	.set_coalesce   = tun_set_coalesce,
3312
	.get_link_ksettings = tun_get_link_ksettings,
L
Linus Torvalds 已提交
3313 3314
};

J
Jason Wang 已提交
3315 3316 3317 3318
static int tun_queue_resize(struct tun_struct *tun)
{
	struct net_device *dev = tun->dev;
	struct tun_file *tfile;
3319
	struct ptr_ring **rings;
J
Jason Wang 已提交
3320 3321 3322
	int n = tun->numqueues + tun->numdisabled;
	int ret, i;

3323 3324
	rings = kmalloc_array(n, sizeof(*rings), GFP_KERNEL);
	if (!rings)
J
Jason Wang 已提交
3325 3326 3327 3328
		return -ENOMEM;

	for (i = 0; i < tun->numqueues; i++) {
		tfile = rtnl_dereference(tun->tfiles[i]);
3329
		rings[i] = &tfile->tx_ring;
J
Jason Wang 已提交
3330 3331
	}
	list_for_each_entry(tfile, &tun->disabled, next)
3332
		rings[i++] = &tfile->tx_ring;
J
Jason Wang 已提交
3333

3334 3335
	ret = ptr_ring_resize_multiple(rings, n,
				       dev->tx_queue_len, GFP_KERNEL,
J
Jason Wang 已提交
3336
				       tun_ptr_free);
J
Jason Wang 已提交
3337

3338
	kfree(rings);
J
Jason Wang 已提交
3339 3340 3341 3342 3343 3344 3345 3346 3347
	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);

3348 3349 3350
	if (dev->rtnl_link_ops != &tun_link_ops)
		return NOTIFY_DONE;

J
Jason Wang 已提交
3351 3352 3353 3354 3355 3356 3357 3358 3359 3360 3361 3362 3363 3364 3365
	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,
};
3366

L
Linus Torvalds 已提交
3367 3368 3369 3370
static int __init tun_init(void)
{
	int ret = 0;

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

3373
	ret = rtnl_link_register(&tun_link_ops);
3374
	if (ret) {
3375
		pr_err("Can't register link_ops\n");
3376
		goto err_linkops;
3377 3378
	}

L
Linus Torvalds 已提交
3379
	ret = misc_register(&tun_miscdev);
3380
	if (ret) {
3381
		pr_err("Can't register misc device %d\n", TUN_MINOR);
3382 3383
		goto err_misc;
	}
J
Jason Wang 已提交
3384

3385 3386 3387 3388 3389 3390
	ret = register_netdevice_notifier(&tun_notifier_block);
	if (ret) {
		pr_err("Can't register netdevice notifier\n");
		goto err_notifier;
	}

3391
	return  0;
3392 3393 3394

err_notifier:
	misc_deregister(&tun_miscdev);
3395
err_misc:
3396 3397
	rtnl_link_unregister(&tun_link_ops);
err_linkops:
L
Linus Torvalds 已提交
3398 3399 3400 3401 3402
	return ret;
}

static void tun_cleanup(void)
{
3403
	misc_deregister(&tun_miscdev);
3404
	rtnl_link_unregister(&tun_link_ops);
J
Jason Wang 已提交
3405
	unregister_netdevice_notifier(&tun_notifier_block);
L
Linus Torvalds 已提交
3406 3407
}

3408 3409 3410 3411 3412 3413
/* 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)
{
3414
	struct tun_file *tfile;
3415 3416
	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
3417 3418
	tfile = file->private_data;
	if (!tfile)
3419
		return ERR_PTR(-EBADFD);
J
Jason Wang 已提交
3420
	return &tfile->socket;
3421 3422 3423
}
EXPORT_SYMBOL_GPL(tun_get_socket);

3424
struct ptr_ring *tun_get_tx_ring(struct file *file)
J
Jason Wang 已提交
3425 3426 3427 3428 3429 3430 3431 3432
{
	struct tun_file *tfile;

	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
	tfile = file->private_data;
	if (!tfile)
		return ERR_PTR(-EBADFD);
3433
	return &tfile->tx_ring;
J
Jason Wang 已提交
3434
}
3435
EXPORT_SYMBOL_GPL(tun_get_tx_ring);
J
Jason Wang 已提交
3436

L
Linus Torvalds 已提交
3437 3438 3439 3440 3441 3442
module_init(tun_init);
module_exit(tun_cleanup);
MODULE_DESCRIPTION(DRV_DESCRIPTION);
MODULE_AUTHOR(DRV_COPYRIGHT);
MODULE_LICENSE("GPL");
MODULE_ALIAS_MISCDEV(TUN_MINOR);
3443
MODULE_ALIAS("devname:net/tun");