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

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

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

L
Linus Torvalds 已提交
39 40 41 42 43 44 45 46
#define DRV_NAME	"tun"
#define DRV_VERSION	"1.6"
#define DRV_DESCRIPTION	"Universal TUN/TAP device driver"
#define DRV_COPYRIGHT	"(C) 1999-2004 Max Krasnyansky <maxk@qualcomm.com>"

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

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

84 85 86
static void tun_default_link_ksettings(struct net_device *dev,
				       struct ethtool_link_ksettings *cmd);

87 88 89
/* Uncomment to enable debugging */
/* #define TUN_DEBUG 1 */

L
Linus Torvalds 已提交
90 91
#ifdef TUN_DEBUG
static int debug;
92

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

116
#define TUN_RX_PAD (NET_IP_ALIGN + NET_SKB_PAD)
117

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

#define TUN_FEATURES (IFF_NO_PI | IFF_ONE_QUEUE | IFF_VNET_HDR | \
129 130
		      IFF_MULTI_QUEUE | IFF_NAPI | IFF_NAPI_FRAGS)

131 132
#define GOODCOPY_LEN 128

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

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

J
Jason Wang 已提交
145 146
#define TUN_FLOW_EXPIRE (3 * HZ)

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

191 192 193 194 195
struct tun_page {
	struct page *page;
	int count;
};

J
Jason Wang 已提交
196 197 198 199 200 201
struct tun_flow_entry {
	struct hlist_node hash_link;
	struct rcu_head rcu;
	struct tun_struct *tun;

	u32 rxhash;
202
	u32 rps_rxhash;
J
Jason Wang 已提交
203 204 205 206 207
	int queue_index;
	unsigned long updated;
};

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

210
struct tun_prog {
211 212 213 214
	struct rcu_head rcu;
	struct bpf_prog *prog;
};

J
Jason Wang 已提交
215
/* Since the socket were moved to tun_file, to preserve the behavior of persist
R
Rami Rosen 已提交
216
 * device, socket filter, sndbuf and vnet header size were restore when the
J
Jason Wang 已提交
217 218
 * file were attached to a persist device.
 */
219
struct tun_struct {
J
Jason Wang 已提交
220 221
	struct tun_file __rcu	*tfiles[MAX_TAP_QUEUES];
	unsigned int            numqueues;
222
	unsigned int 		flags;
223 224
	kuid_t			owner;
	kgid_t			group;
225 226

	struct net_device	*dev;
227
	netdev_features_t	set_features;
228
#define TUN_USER_FEATURES (NETIF_F_HW_CSUM|NETIF_F_TSO_ECN|NETIF_F_TSO| \
229
			  NETIF_F_TSO6)
230

231
	int			align;
232
	int			vnet_hdr_sz;
J
Jason Wang 已提交
233 234 235 236 237
	int			sndbuf;
	struct tap_filter	txflt;
	struct sock_fprog	fprog;
	/* protected by rtnl lock */
	bool			filter_attached;
238 239
#ifdef TUN_DEBUG
	int debug;
L
Linus Torvalds 已提交
240
#endif
J
Jason Wang 已提交
241 242 243 244
	spinlock_t lock;
	struct hlist_head flows[TUN_NUM_FLOW_ENTRIES];
	struct timer_list flow_gc_timer;
	unsigned long ageing_time;
J
Jason Wang 已提交
245 246
	unsigned int numdisabled;
	struct list_head disabled;
247
	void *security;
248
	u32 flow_count;
J
Jason Wang 已提交
249
	u32 rx_batched;
250
	struct tun_pcpu_stats __percpu *pcpu_stats;
J
Jason Wang 已提交
251
	struct bpf_prog __rcu *xdp_prog;
252
	struct tun_prog __rcu *steering_prog;
253
	struct tun_prog __rcu *filter_prog;
254
	struct ethtool_link_ksettings link_ksettings;
255
};
L
Linus Torvalds 已提交
256

257 258 259
struct veth {
	__be16 h_vlan_proto;
	__be16 h_vlan_TCI;
260
};
L
Linus Torvalds 已提交
261

262
bool tun_is_xdp_frame(void *ptr)
J
Jason Wang 已提交
263 264 265
{
	return (unsigned long)ptr & TUN_XDP_FLAG;
}
266
EXPORT_SYMBOL(tun_is_xdp_frame);
J
Jason Wang 已提交
267 268 269 270 271 272 273 274 275 276 277 278 279

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

280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320
static int tun_napi_receive(struct napi_struct *napi, int budget)
{
	struct tun_file *tfile = container_of(napi, struct tun_file, napi);
	struct sk_buff_head *queue = &tfile->sk.sk_write_queue;
	struct sk_buff_head process_queue;
	struct sk_buff *skb;
	int received = 0;

	__skb_queue_head_init(&process_queue);

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

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

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

	return received;
}

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

	received = tun_napi_receive(napi, budget);

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

	return received;
}

static void tun_napi_init(struct tun_struct *tun, struct tun_file *tfile,
E
Eric Dumazet 已提交
321
			  bool napi_en, bool napi_frags)
322
{
323
	tfile->napi_enabled = napi_en;
E
Eric Dumazet 已提交
324
	tfile->napi_frags_enabled = napi_en && napi_frags;
325 326 327 328 329 330 331
	if (napi_en) {
		netif_napi_add(tun->dev, &tfile->napi, tun_napi_poll,
			       NAPI_POLL_WEIGHT);
		napi_enable(&tfile->napi);
	}
}

E
Eric Dumazet 已提交
332
static void tun_napi_disable(struct tun_file *tfile)
333
{
334
	if (tfile->napi_enabled)
335 336 337
		napi_disable(&tfile->napi);
}

E
Eric Dumazet 已提交
338
static void tun_napi_del(struct tun_file *tfile)
339
{
340
	if (tfile->napi_enabled)
341 342 343
		netif_napi_del(&tfile->napi);
}

E
Eric Dumazet 已提交
344
static bool tun_napi_frags_enabled(const struct tun_file *tfile)
345
{
E
Eric Dumazet 已提交
346
	return tfile->napi_frags_enabled;
347 348
}

349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396
#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 */

397 398
static inline bool tun_is_little_endian(struct tun_struct *tun)
{
399
	return tun->flags & TUN_VNET_LE ||
400
		tun_legacy_is_little_endian(tun);
401 402
}

403 404
static inline u16 tun16_to_cpu(struct tun_struct *tun, __virtio16 val)
{
405
	return __virtio16_to_cpu(tun_is_little_endian(tun), val);
406 407 408 409
}

static inline __virtio16 cpu_to_tun16(struct tun_struct *tun, u16 val)
{
410
	return __cpu_to_virtio16(tun_is_little_endian(tun), val);
411 412
}

J
Jason Wang 已提交
413 414
static inline u32 tun_hashfn(u32 rxhash)
{
L
Li RongQing 已提交
415
	return rxhash & TUN_MASK_FLOW_ENTRIES;
J
Jason Wang 已提交
416 417 418 419 420 421
}

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

422
	hlist_for_each_entry_rcu(e, head, hash_link) {
J
Jason Wang 已提交
423 424 425 426 427 428 429 430 431 432
		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)
{
433 434
	struct tun_flow_entry *e = kmalloc(sizeof(*e), GFP_ATOMIC);

J
Jason Wang 已提交
435 436 437 438 439
	if (e) {
		tun_debug(KERN_INFO, tun, "create flow: hash %u index %u\n",
			  rxhash, queue_index);
		e->updated = jiffies;
		e->rxhash = rxhash;
440
		e->rps_rxhash = 0;
J
Jason Wang 已提交
441 442 443
		e->queue_index = queue_index;
		e->tun = tun;
		hlist_add_head_rcu(&e->hash_link, head);
444
		++tun->flow_count;
J
Jason Wang 已提交
445 446 447 448 449 450 451 452 453
	}
	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);
454
	kfree_rcu(e, rcu);
455
	--tun->flow_count;
J
Jason Wang 已提交
456 457 458 459 460 461 462 463 464
}

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

467
		hlist_for_each_entry_safe(e, n, &tun->flows[i], hash_link)
J
Jason Wang 已提交
468 469 470 471 472 473 474 475 476 477 478 479
			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;
480
		struct hlist_node *n;
J
Jason Wang 已提交
481

482
		hlist_for_each_entry_safe(e, n, &tun->flows[i], hash_link) {
J
Jason Wang 已提交
483 484 485 486 487 488 489
			if (e->queue_index == queue_index)
				tun_flow_delete(tun, e);
		}
	}
	spin_unlock_bh(&tun->lock);
}

490
static void tun_flow_cleanup(struct timer_list *t)
J
Jason Wang 已提交
491
{
492
	struct tun_struct *tun = from_timer(tun, t, flow_gc_timer);
J
Jason Wang 已提交
493 494 495 496 497 498 499
	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");

500
	spin_lock(&tun->lock);
J
Jason Wang 已提交
501 502
	for (i = 0; i < TUN_NUM_FLOW_ENTRIES; i++) {
		struct tun_flow_entry *e;
503
		struct hlist_node *n;
J
Jason Wang 已提交
504

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

J
Jason Wang 已提交
508
			this_timer = e->updated + delay;
509
			if (time_before_eq(this_timer, jiffies)) {
J
Jason Wang 已提交
510
				tun_flow_delete(tun, e);
511 512 513 514
				continue;
			}
			count++;
			if (time_before(this_timer, next_timer))
J
Jason Wang 已提交
515 516 517 518 519 520
				next_timer = this_timer;
		}
	}

	if (count)
		mod_timer(&tun->flow_gc_timer, round_jiffies_up(next_timer));
521
	spin_unlock(&tun->lock);
J
Jason Wang 已提交
522 523
}

524
static void tun_flow_update(struct tun_struct *tun, u32 rxhash,
525
			    struct tun_file *tfile)
J
Jason Wang 已提交
526 527 528 529
{
	struct hlist_head *head;
	struct tun_flow_entry *e;
	unsigned long delay = tun->ageing_time;
530
	u16 queue_index = tfile->queue_index;
J
Jason Wang 已提交
531 532 533 534 535 536 537 538 539 540 541 542 543

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

	rcu_read_lock();

	e = tun_flow_find(head, rxhash);
	if (likely(e)) {
		/* TODO: keep queueing to old queue until it's empty? */
		e->queue_index = queue_index;
		e->updated = jiffies;
544
		sock_rps_record_flow_hash(e->rps_rxhash);
J
Jason Wang 已提交
545 546
	} else {
		spin_lock_bh(&tun->lock);
547 548
		if (!tun_flow_find(head, rxhash) &&
		    tun->flow_count < MAX_TAP_FLOWS)
J
Jason Wang 已提交
549 550 551 552 553 554 555 556 557 558 559
			tun_flow_create(tun, head, rxhash, queue_index);

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

	rcu_read_unlock();
}

560 561 562 563 564 565
/**
 * 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)
{
566
	if (unlikely(e->rps_rxhash != hash))
567 568 569
		e->rps_rxhash = hash;
}

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

582
	numqueues = READ_ONCE(tun->numqueues);
J
Jason Wang 已提交
583

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

	return txq;
}

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

659
void tun_ptr_free(void *ptr)
J
Jason Wang 已提交
660 661 662
{
	if (!ptr)
		return;
663 664
	if (tun_is_xdp_frame(ptr)) {
		struct xdp_frame *xdpf = tun_ptr_to_xdp(ptr);
J
Jason Wang 已提交
665

666
		xdp_return_frame(xdpf);
J
Jason Wang 已提交
667 668 669 670
	} else {
		__skb_array_destroy_skb(ptr);
	}
}
671
EXPORT_SYMBOL_GPL(tun_ptr_free);
J
Jason Wang 已提交
672

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

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

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

J
Jason Wang 已提交
684 685 686 687 688
static void __tun_detach(struct tun_file *tfile, bool clean)
{
	struct tun_file *ntfile;
	struct tun_struct *tun;

689 690
	tun = rtnl_dereference(tfile->tun);

691
	if (tun && clean) {
E
Eric Dumazet 已提交
692 693
		tun_napi_disable(tfile);
		tun_napi_del(tfile);
694 695
	}

696
	if (tun && !tfile->detached) {
J
Jason Wang 已提交
697 698 699 700 701
		u16 index = tfile->queue_index;
		BUG_ON(index >= tun->numqueues);

		rcu_assign_pointer(tun->tfiles[index],
				   tun->tfiles[tun->numqueues - 1]);
702
		ntfile = rtnl_dereference(tun->tfiles[index]);
J
Jason Wang 已提交
703 704 705
		ntfile->queue_index = index;

		--tun->numqueues;
706
		if (clean) {
707
			RCU_INIT_POINTER(tfile->tun, NULL);
J
Jason Wang 已提交
708
			sock_put(&tfile->sk);
709
		} else
J
Jason Wang 已提交
710
			tun_disable_queue(tun, tfile);
J
Jason Wang 已提交
711 712

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

	if (clean) {
723 724 725
		if (tun && tun->numqueues == 0 && tun->numdisabled == 0) {
			netif_carrier_off(tun->dev);

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

static void tun_detach(struct tun_file *tfile, bool clean)
{
739 740 741
	struct tun_struct *tun;
	struct net_device *dev;

J
Jason Wang 已提交
742
	rtnl_lock();
743 744
	tun = rtnl_dereference(tfile->tun);
	dev = tun ? tun->dev : NULL;
J
Jason Wang 已提交
745
	__tun_detach(tfile, clean);
746 747
	if (dev)
		netdev_state_change(dev);
J
Jason Wang 已提交
748 749 750 751 752 753
	rtnl_unlock();
}

static void tun_detach_all(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
J
Jason Wang 已提交
754
	struct tun_file *tfile, *tmp;
J
Jason Wang 已提交
755 756 757
	int i, n = tun->numqueues;

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

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

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

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

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

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

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

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

	err = 0;
J
Jason Wang 已提交
819

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

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

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

	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;
851 852 853 854 855 856
		err = xdp_rxq_info_reg_mem_model(&tfile->xdp_rxq,
						 MEM_TYPE_PAGE_SHARED, NULL);
		if (err < 0) {
			xdp_rxq_info_unreg(&tfile->xdp_rxq);
			goto out;
		}
J
Jesper Dangaard Brouer 已提交
857 858 859
		err = 0;
	}

860
	rcu_assign_pointer(tfile->tun, tun);
J
Jason Wang 已提交
861 862
	rcu_assign_pointer(tun->tfiles[tun->numqueues], tfile);
	tun->numqueues++;
863

864
	if (tfile->detached) {
J
Jason Wang 已提交
865
		tun_enable_queue(tfile);
866
	} else {
J
Jason Wang 已提交
867
		sock_hold(&tfile->sk);
E
Eric Dumazet 已提交
868
		tun_napi_init(tun, tfile, napi, napi_frags);
869
	}
J
Jason Wang 已提交
870

J
Jason Wang 已提交
871 872 873
	if (rtnl_dereference(tun->xdp_prog))
		sock_set_flag(&tfile->sk, SOCK_XDP);

J
Jason Wang 已提交
874
	tun_set_real_num_queues(tun);
875

J
Jason Wang 已提交
876 877 878 879 880 881
	/* device is allowed to go away first, so no need to hold extra
	 * refcnt.
	 */

out:
	return err;
E
Eric W. Biederman 已提交
882 883
}

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

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

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

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

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

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

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

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

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

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

L
Linus Torvalds 已提交
1016 1017 1018
/* Net device open. */
static int tun_net_open(struct net_device *dev)
{
1019 1020 1021
	struct tun_struct *tun = netdev_priv(dev);
	int i;

J
Jason Wang 已提交
1022
	netif_tx_start_all_queues(dev);
1023 1024 1025 1026 1027 1028 1029 1030

	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 已提交
1031 1032 1033 1034 1035 1036
	return 0;
}

/* Net device close. */
static int tun_net_close(struct net_device *dev)
{
J
Jason Wang 已提交
1037
	netif_tx_stop_all_queues(dev);
L
Linus Torvalds 已提交
1038 1039 1040 1041
	return 0;
}

/* Net device start xmit */
1042
static void tun_automq_xmit(struct tun_struct *tun, struct sk_buff *skb)
L
Linus Torvalds 已提交
1043
{
1044
#ifdef CONFIG_RPS
1045
	if (tun->numqueues == 1 && static_key_false(&rps_needed)) {
1046 1047 1048
		/* Select queue was not called for the skbuff, so we extract the
		 * RPS hash and save it into the flow_table here.
		 */
1049
		struct tun_flow_entry *e;
1050 1051
		__u32 rxhash;

J
Jason Wang 已提交
1052
		rxhash = __skb_get_hash_symmetric(skb);
1053 1054 1055
		e = tun_flow_find(&tun->flows[tun_hashfn(rxhash)], rxhash);
		if (e)
			tun_flow_save_rps_rxhash(e, rxhash);
1056
	}
1057
#endif
1058 1059
}

1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071
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;
}

1072 1073 1074 1075 1076 1077
/* 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;
1078
	int len = skb->len;
1079 1080 1081 1082 1083

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

	/* Drop packet if interface is not attached */
1084
	if (txq >= tun->numqueues)
1085 1086 1087 1088
		goto drop;

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

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

J
Jason Wang 已提交
1092 1093
	BUG_ON(!tfile);

1094 1095 1096 1097 1098 1099
	/* 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 已提交
1100 1101
	if (tfile->socket.sk->sk_filter &&
	    sk_filter(tfile->socket.sk, skb))
M
Michael S. Tsirkin 已提交
1102 1103
		goto drop;

1104
	len = run_ebpf_filter(tun, skb, len);
B
Bjørn Mork 已提交
1105
	if (len == 0 || pskb_trim(skb, len))
1106 1107
		goto drop;

W
Willem de Bruijn 已提交
1108
	if (unlikely(skb_orphan_frags_rx(skb, GFP_ATOMIC)))
1109 1110
		goto drop;

1111
	skb_tx_timestamp(skb);
1112

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

1118 1119
	nf_reset(skb);

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

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

	rcu_read_unlock();
1129
	return NETDEV_TX_OK;
L
Linus Torvalds 已提交
1130 1131

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

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

1148 1149
static netdev_features_t tun_net_fix_features(struct net_device *dev,
	netdev_features_t features)
1150 1151 1152 1153 1154
{
	struct tun_struct *tun = netdev_priv(dev);

	return (features & tun->set_features) | (features & ~TUN_USER_FEATURES);
}
1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165

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

1166
static void
1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201
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 已提交
1202 1203 1204 1205
static int tun_xdp_set(struct net_device *dev, struct bpf_prog *prog,
		       struct netlink_ext_ack *extack)
{
	struct tun_struct *tun = netdev_priv(dev);
J
Jason Wang 已提交
1206
	struct tun_file *tfile;
J
Jason Wang 已提交
1207
	struct bpf_prog *old_prog;
J
Jason Wang 已提交
1208
	int i;
J
Jason Wang 已提交
1209 1210 1211 1212 1213 1214

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

J
Jason Wang 已提交
1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228
	for (i = 0; i < tun->numqueues; i++) {
		tfile = rtnl_dereference(tun->tfiles[i]);
		if (prog)
			sock_set_flag(&tfile->sk, SOCK_XDP);
		else
			sock_reset_flag(&tfile->sk, SOCK_XDP);
	}
	list_for_each_entry(tfile, &tun->disabled, next) {
		if (prog)
			sock_set_flag(&tfile->sk, SOCK_XDP);
		else
			sock_reset_flag(&tfile->sk, SOCK_XDP);
	}

J
Jason Wang 已提交
1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243
	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;
}

1244
static int tun_xdp(struct net_device *dev, struct netdev_bpf *xdp)
J
Jason Wang 已提交
1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256
{
	switch (xdp->command) {
	case XDP_SETUP_PROG:
		return tun_xdp_set(dev, xdp->prog, xdp->extack);
	case XDP_QUERY_PROG:
		xdp->prog_id = tun_xdp_query(dev);
		return 0;
	default:
		return -EINVAL;
	}
}

N
Nicolas Dichtel 已提交
1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271
static int tun_net_change_carrier(struct net_device *dev, bool new_carrier)
{
	if (new_carrier) {
		struct tun_struct *tun = netdev_priv(dev);

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

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

S
Stephen Hemminger 已提交
1272
static const struct net_device_ops tun_netdev_ops = {
1273
	.ndo_uninit		= tun_net_uninit,
S
Stephen Hemminger 已提交
1274 1275
	.ndo_open		= tun_net_open,
	.ndo_stop		= tun_net_close,
1276
	.ndo_start_xmit		= tun_net_xmit,
1277
	.ndo_fix_features	= tun_net_fix_features,
J
Jason Wang 已提交
1278
	.ndo_select_queue	= tun_select_queue,
1279
	.ndo_set_rx_headroom	= tun_set_headroom,
1280
	.ndo_get_stats64	= tun_net_get_stats64,
N
Nicolas Dichtel 已提交
1281
	.ndo_change_carrier	= tun_net_change_carrier,
S
Stephen Hemminger 已提交
1282 1283
};

1284 1285 1286 1287 1288 1289 1290 1291
static void __tun_xdp_flush_tfile(struct tun_file *tfile)
{
	/* Notify and wake up reader process */
	if (tfile->flags & TUN_FASYNC)
		kill_fasync(&tfile->fasync, SIGIO, POLL_IN);
	tfile->socket.sk->sk_data_ready(tfile->socket.sk);
}

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

1302
	if (unlikely(flags & ~XDP_XMIT_FLAGS_MASK))
1303 1304
		return -EINVAL;

J
Jason Wang 已提交
1305 1306 1307 1308
	rcu_read_lock();

	numqueues = READ_ONCE(tun->numqueues);
	if (!numqueues) {
1309 1310
		rcu_read_unlock();
		return -ENXIO; /* Caller will free/return all frames */
J
Jason Wang 已提交
1311 1312 1313 1314
	}

	tfile = rcu_dereference(tun->tfiles[smp_processor_id() %
					    numqueues]);
1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328

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

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

1332 1333 1334
	if (flags & XDP_XMIT_FLUSH)
		__tun_xdp_flush_tfile(tfile);

J
Jason Wang 已提交
1335
	rcu_read_unlock();
1336
	return cnt - drops;
J
Jason Wang 已提交
1337 1338
}

1339 1340 1341 1342 1343 1344 1345
static int tun_xdp_tx(struct net_device *dev, struct xdp_buff *xdp)
{
	struct xdp_frame *frame = convert_to_xdp_frame(xdp);

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

1346
	return tun_xdp_xmit(dev, 1, &frame, XDP_XMIT_FLUSH);
J
Jason Wang 已提交
1347 1348
}

S
Stephen Hemminger 已提交
1349
static const struct net_device_ops tap_netdev_ops = {
1350
	.ndo_uninit		= tun_net_uninit,
S
Stephen Hemminger 已提交
1351 1352
	.ndo_open		= tun_net_open,
	.ndo_stop		= tun_net_close,
1353
	.ndo_start_xmit		= tun_net_xmit,
1354
	.ndo_fix_features	= tun_net_fix_features,
1355
	.ndo_set_rx_mode	= tun_net_mclist,
S
Stephen Hemminger 已提交
1356 1357
	.ndo_set_mac_address	= eth_mac_addr,
	.ndo_validate_addr	= eth_validate_addr,
J
Jason Wang 已提交
1358
	.ndo_select_queue	= tun_select_queue,
1359
	.ndo_features_check	= passthru_features_check,
1360
	.ndo_set_rx_headroom	= tun_set_headroom,
1361
	.ndo_get_stats64	= tun_net_get_stats64,
1362
	.ndo_bpf		= tun_xdp,
J
Jason Wang 已提交
1363
	.ndo_xdp_xmit		= tun_xdp_xmit,
N
Nicolas Dichtel 已提交
1364
	.ndo_change_carrier	= tun_net_change_carrier,
S
Stephen Hemminger 已提交
1365 1366
};

1367
static void tun_flow_init(struct tun_struct *tun)
J
Jason Wang 已提交
1368 1369 1370 1371 1372 1373 1374
{
	int i;

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

	tun->ageing_time = TUN_FLOW_EXPIRE;
1375 1376 1377
	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 已提交
1378 1379 1380 1381 1382 1383 1384 1385
}

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

1386 1387 1388
#define MIN_MTU 68
#define MAX_MTU 65535

L
Linus Torvalds 已提交
1389 1390 1391 1392
/* Initialize net device. */
static void tun_net_init(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
1393

L
Linus Torvalds 已提交
1394
	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
1395
	case IFF_TUN:
S
Stephen Hemminger 已提交
1396 1397
		dev->netdev_ops = &tun_netdev_ops;

L
Linus Torvalds 已提交
1398 1399 1400 1401 1402 1403
		/* Point-to-Point TUN Device */
		dev->hard_header_len = 0;
		dev->addr_len = 0;
		dev->mtu = 1500;

		/* Zero header length */
1404
		dev->type = ARPHRD_NONE;
L
Linus Torvalds 已提交
1405 1406 1407
		dev->flags = IFF_POINTOPOINT | IFF_NOARP | IFF_MULTICAST;
		break;

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

1415
		eth_hw_addr_random(dev);
1416

L
Linus Torvalds 已提交
1417 1418
		break;
	}
1419 1420 1421

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

1424 1425 1426 1427 1428 1429 1430
static bool tun_sock_writeable(struct tun_struct *tun, struct tun_file *tfile)
{
	struct sock *sk = tfile->socket.sk;

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

L
Linus Torvalds 已提交
1431 1432 1433
/* Character device part */

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

	if (!tun)
1442
		return EPOLLERR;
L
Linus Torvalds 已提交
1443

J
Jason Wang 已提交
1444
	sk = tfile->socket.sk;
1445

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

1448
	poll_wait(file, sk_sleep(sk), wait);
1449

1450
	if (!ptr_ring_empty(&tfile->tx_ring))
1451
		mask |= EPOLLIN | EPOLLRDNORM;
L
Linus Torvalds 已提交
1452

1453 1454 1455 1456 1457 1458 1459 1460
	/* Make sure SOCKWQ_ASYNC_NOSPACE is set if not writable to
	 * guarantee EPOLLOUT to be raised by either here or
	 * tun_sock_write_space(). Then process could get notification
	 * after it writes to a down device and meets -EIO.
	 */
	if (tun_sock_writeable(tun, tfile) ||
	    (!test_and_set_bit(SOCKWQ_ASYNC_NOSPACE, &sk->sk_socket->flags) &&
	     tun_sock_writeable(tun, tfile)))
1461
		mask |= EPOLLOUT | EPOLLWRNORM;
1462

1463
	if (tun->dev->reg_state != NETREG_REGISTERED)
1464
		mask = EPOLLERR;
1465

E
Eric W. Biederman 已提交
1466
	tun_put(tun);
L
Linus Torvalds 已提交
1467 1468 1469
	return mask;
}

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
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;
1499 1500
		struct page *page;
		void *frag;
1501 1502 1503 1504 1505

		if (fragsz == 0 || fragsz > PAGE_SIZE) {
			err = -EINVAL;
			goto free;
		}
1506 1507
		frag = netdev_alloc_frag(fragsz);
		if (!frag) {
1508 1509 1510
			err = -ENOMEM;
			goto free;
		}
1511 1512 1513
		page = virt_to_head_page(frag);
		skb_fill_page_desc(skb, i - 1, page,
				   frag - page_address(page), fragsz);
1514 1515 1516 1517 1518 1519 1520 1521 1522
	}

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

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

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

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

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

	return skb;
}

J
Jason Wang 已提交
1550 1551 1552 1553 1554 1555 1556 1557 1558 1559
static void tun_rx_batched(struct tun_struct *tun, struct tun_file *tfile,
			   struct sk_buff *skb, int more)
{
	struct sk_buff_head *queue = &tfile->sk.sk_write_queue;
	struct sk_buff_head process_queue;
	u32 rx_batched = tun->rx_batched;
	bool rcv = false;

	if (!rx_batched || (!more && skb_queue_empty(queue))) {
		local_bh_disable();
M
Matthew Cover 已提交
1560
		skb_record_rx_queue(skb, tfile->queue_index);
J
Jason Wang 已提交
1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579
		netif_receive_skb(skb);
		local_bh_enable();
		return;
	}

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

	if (rcv) {
		struct sk_buff *nskb;

		local_bh_disable();
M
Matthew Cover 已提交
1580 1581
		while ((nskb = __skb_dequeue(&process_queue))) {
			skb_record_rx_queue(nskb, tfile->queue_index);
J
Jason Wang 已提交
1582
			netif_receive_skb(nskb);
M
Matthew Cover 已提交
1583 1584
		}
		skb_record_rx_queue(skb, tfile->queue_index);
J
Jason Wang 已提交
1585 1586 1587 1588 1589
		netif_receive_skb(skb);
		local_bh_enable();
	}
}

1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611
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;
}

1612
static struct sk_buff *__tun_build_skb(struct page_frag *alloc_frag, char *buf,
J
Jason Wang 已提交
1613
				       int buflen, int len, int pad)
1614 1615 1616 1617 1618 1619
{
	struct sk_buff *skb = build_skb(buf, buflen);

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

J
Jason Wang 已提交
1620
	skb_reserve(skb, pad);
1621 1622 1623 1624 1625 1626 1627 1628
	skb_put(skb, len);

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

	return skb;
}

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

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

	return act;
}

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

	rcu_read_lock();
	xdp_prog = rcu_dereference(tun->xdp_prog);
	if (xdp_prog)
1678
		pad += XDP_PACKET_HEADROOM;
1679 1680
	buflen += SKB_DATA_ALIGN(len + pad);
	rcu_read_unlock();
1681

1682
	alloc_frag->offset = ALIGN((u64)alloc_frag->offset, SMP_CACHE_BYTES);
1683 1684 1685 1686 1687
	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,
1688
				     alloc_frag->offset + pad,
1689 1690 1691 1692
				     len, from);
	if (copied != len)
		return ERR_PTR(-EFAULT);

1693 1694 1695 1696
	/* 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.
	 */
1697
	if (hdr->gso_type || !xdp_prog) {
J
Jason Wang 已提交
1698
		*skb_xdp = 1;
J
Jason Wang 已提交
1699
		return __tun_build_skb(alloc_frag, buf, buflen, len, pad);
1700 1701 1702
	}

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

1704
	local_bh_disable();
J
Jason Wang 已提交
1705 1706
	rcu_read_lock();
	xdp_prog = rcu_dereference(tun->xdp_prog);
J
Jason Wang 已提交
1707
	if (xdp_prog) {
J
Jason Wang 已提交
1708 1709 1710 1711
		struct xdp_buff xdp;
		u32 act;

		xdp.data_hard_start = buf;
1712
		xdp.data = buf + pad;
1713
		xdp_set_data_meta_invalid(&xdp);
J
Jason Wang 已提交
1714
		xdp.data_end = xdp.data + len;
J
Jesper Dangaard Brouer 已提交
1715
		xdp.rxq = &tfile->xdp_rxq;
J
Jason Wang 已提交
1716

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

		pad = xdp.data - xdp.data_hard_start;
		len = xdp.data_end - xdp.data;
J
Jason Wang 已提交
1732
	}
1733 1734
	rcu_read_unlock();
	local_bh_enable();
J
Jason Wang 已提交
1735

J
Jason Wang 已提交
1736
	return __tun_build_skb(alloc_frag, buf, buflen, len, pad);
J
Jason Wang 已提交
1737

J
Jason Wang 已提交
1738
err_xdp:
J
Jason Wang 已提交
1739
	put_page(alloc_frag->page);
1740
out:
J
Jason Wang 已提交
1741
	rcu_read_unlock();
1742
	local_bh_enable();
J
Jason Wang 已提交
1743
	return NULL;
1744 1745
}

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

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

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

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

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

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

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

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

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

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

1803 1804
	good_linear = SKB_MAX_HEAD(align);

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

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

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

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

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

1863 1864 1865 1866 1867 1868 1869 1870
		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);
1871 1872 1873 1874 1875
			if (frags) {
				tfile->napi.skb = NULL;
				mutex_unlock(&tfile->napi_mutex);
			}

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

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

1888 1889 1890
		return -EINVAL;
	}

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

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

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

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

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

J
Jason Wang 已提交
1933
	if (skb_xdp) {
J
Jason Wang 已提交
1934 1935 1936
		struct bpf_prog *xdp_prog;
		int ret;

1937
		local_bh_disable();
J
Jason Wang 已提交
1938 1939 1940 1941 1942 1943
		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();
1944
				local_bh_enable();
J
Jason Wang 已提交
1945 1946 1947 1948
				return total_len;
			}
		}
		rcu_read_unlock();
1949
		local_bh_enable();
J
Jason Wang 已提交
1950 1951
	}

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

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

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

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

2002 2003 2004
	if (rxhash)
		tun_flow_update(tun, rxhash, tfile);

2005
	return total_len;
2006
}
L
Linus Torvalds 已提交
2007

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

	if (!tun)
		return -EBADFD;

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

	tun_put(tun);
	return result;
L
Linus Torvalds 已提交
2023 2024
}

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

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

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

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

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

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

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

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

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

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

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

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

H
Herbert Xu 已提交
2116
		if (copy_to_iter(&gso, sizeof(gso), iter) != sizeof(gso))
2117
			return -EFAULT;
2118 2119

		iov_iter_advance(iter, vnet_hdr_sz - sizeof(gso));
2120 2121
	}

2122
	if (vlan_hlen) {
H
Herbert Xu 已提交
2123
		int ret;
2124
		struct veth veth;
J
Jason Wang 已提交
2125 2126

		veth.h_vlan_proto = skb->vlan_proto;
2127
		veth.h_vlan_TCI = htons(skb_vlan_tag_get(skb));
J
Jason Wang 已提交
2128 2129 2130

		vlan_offset = offsetof(struct vlan_ethhdr, h_vlan_proto);

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

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

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

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

	return total;
}

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

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

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

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

		schedule();
	}

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

out:
J
Jason Wang 已提交
2191
	*err = error;
J
Jason Wang 已提交
2192
	return ptr;
J
Jason Wang 已提交
2193 2194
}

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

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

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

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

2216 2217
	if (tun_is_xdp_frame(ptr)) {
		struct xdp_frame *xdpf = tun_ptr_to_xdp(ptr);
J
Jason Wang 已提交
2218

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

2231 2232 2233
	return ret;
}

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

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

2251
static void tun_prog_free(struct rcu_head *rcu)
2252
{
2253
	struct tun_prog *prog = container_of(rcu, struct tun_prog, rcu);
2254 2255 2256 2257 2258

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

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

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

2272
	spin_lock_bh(&tun->lock);
2273
	old = rcu_dereference_protected(*prog_p,
2274
					lockdep_is_held(&tun->lock));
2275
	rcu_assign_pointer(*prog_p, new);
2276
	spin_unlock_bh(&tun->lock);
2277 2278

	if (old)
2279
		call_rcu(&old->rcu, tun_prog_free);
2280 2281 2282 2283

	return 0;
}

J
Jason Wang 已提交
2284 2285 2286 2287
static void tun_free_netdev(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

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

L
Linus Torvalds 已提交
2296 2297 2298 2299
static void tun_setup(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

2300 2301
	tun->owner = INVALID_UID;
	tun->group = INVALID_GID;
2302
	tun_default_link_ksettings(dev, &tun->link_ksettings);
L
Linus Torvalds 已提交
2303 2304

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

2311 2312 2313
/* Trivial set of netlink ops to allow deleting tun or tap
 * device with netlink.
 */
2314 2315
static int tun_validate(struct nlattr *tb[], struct nlattr *data[],
			struct netlink_ext_ack *extack)
2316
{
2317 2318
	if (!data)
		return 0;
2319 2320 2321
	return -EINVAL;
}

2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372 2373 2374 2375
static size_t tun_get_size(const struct net_device *dev)
{
	BUILD_BUG_ON(sizeof(u32) != sizeof(uid_t));
	BUILD_BUG_ON(sizeof(u32) != sizeof(gid_t));

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

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

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

	return 0;

nla_put_failure:
	return -EMSGSIZE;
}

2376 2377 2378 2379 2380
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,
2381 2382
	.get_size       = tun_get_size,
	.fill_info      = tun_fill_info,
2383 2384
};

2385 2386
static void tun_sock_write_space(struct sock *sk)
{
J
Jason Wang 已提交
2387
	struct tun_file *tfile;
2388
	wait_queue_head_t *wqueue;
2389 2390 2391 2392

	if (!sock_writeable(sk))
		return;

2393
	if (!test_and_clear_bit(SOCKWQ_ASYNC_NOSPACE, &sk->sk_socket->flags))
2394 2395
		return;

2396 2397
	wqueue = sk_sleep(sk);
	if (wqueue && waitqueue_active(wqueue))
2398 2399
		wake_up_interruptible_sync_poll(wqueue, EPOLLOUT |
						EPOLLWRNORM | EPOLLWRBAND);
H
Herbert Xu 已提交
2400

J
Jason Wang 已提交
2401 2402
	tfile = container_of(sk, struct tun_file, sk);
	kill_fasync(&tfile->fasync, SIGIO, POLL_OUT);
2403 2404
}

2405 2406 2407 2408 2409 2410
static void tun_put_page(struct tun_page *tpage)
{
	if (tpage->page)
		__page_frag_cache_drain(tpage->page, tpage->count);
}

2411 2412
static int tun_xdp_one(struct tun_struct *tun,
		       struct tun_file *tfile,
2413 2414
		       struct xdp_buff *xdp, int *flush,
		       struct tun_page *tpage)
2415 2416 2417 2418 2419 2420 2421 2422 2423 2424
{
	struct tun_xdp_hdr *hdr = xdp->data_hard_start;
	struct virtio_net_hdr *gso = &hdr->gso;
	struct tun_pcpu_stats *stats;
	struct bpf_prog *xdp_prog;
	struct sk_buff *skb = NULL;
	u32 rxhash = 0, act;
	int buflen = hdr->buflen;
	int err = 0;
	bool skb_xdp = false;
2425
	struct page *page;
2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451

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

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

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

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

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

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

	skb->protocol = eth_type_trans(skb, tun->dev);
	skb_reset_network_header(skb);
	skb_probe_transport_header(skb, 0);

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

2491 2492
	if (!rcu_dereference(tun->steering_prog) && tun->numqueues > 1 &&
	    !tfile->detached)
2493 2494
		rxhash = __skb_get_hash_symmetric(skb);

M
Matthew Cover 已提交
2495
	skb_record_rx_queue(skb, tfile->queue_index);
2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511
	netif_receive_skb(skb);

	stats = get_cpu_ptr(tun->pcpu_stats);
	u64_stats_update_begin(&stats->syncp);
	stats->rx_packets++;
	stats->rx_bytes += skb->len;
	u64_stats_update_end(&stats->syncp);
	put_cpu_ptr(stats);

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

out:
	return err;
}

2512
static int tun_sendmsg(struct socket *sock, struct msghdr *m, size_t total_len)
2513
{
2514
	int ret, i;
J
Jason Wang 已提交
2515
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2516
	struct tun_struct *tun = tun_get(tfile);
2517
	struct tun_msg_ctl *ctl = m->msg_control;
2518
	struct xdp_buff *xdp;
J
Jason Wang 已提交
2519 2520 2521

	if (!tun)
		return -EBADFD;
2522

2523
	if (ctl && (ctl->type == TUN_MSG_PTR)) {
2524
		struct tun_page tpage;
2525 2526 2527
		int n = ctl->num;
		int flush = 0;

2528 2529
		memset(&tpage, 0, sizeof(tpage));

2530 2531 2532 2533 2534
		local_bh_disable();
		rcu_read_lock();

		for (i = 0; i < n; i++) {
			xdp = &((struct xdp_buff *)ctl->ptr)[i];
2535
			tun_xdp_one(tun, tfile, xdp, &flush, &tpage);
2536 2537 2538 2539 2540 2541 2542 2543
		}

		if (flush)
			xdp_do_flush_map();

		rcu_read_unlock();
		local_bh_enable();

2544 2545
		tun_put_page(&tpage);

2546 2547 2548
		ret = total_len;
		goto out;
	}
2549 2550

	ret = tun_get_user(tun, tfile, ctl ? ctl->ptr : NULL, &m->msg_iter,
J
Jason Wang 已提交
2551 2552
			   m->msg_flags & MSG_DONTWAIT,
			   m->msg_flags & MSG_MORE);
2553
out:
J
Jason Wang 已提交
2554 2555
	tun_put(tun);
	return ret;
2556 2557
}

2558
static int tun_recvmsg(struct socket *sock, struct msghdr *m, size_t total_len,
2559 2560
		       int flags)
{
J
Jason Wang 已提交
2561
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2562
	struct tun_struct *tun = tun_get(tfile);
J
Jason Wang 已提交
2563
	void *ptr = m->msg_control;
2564
	int ret;
J
Jason Wang 已提交
2565

W
Wei Xu 已提交
2566 2567
	if (!tun) {
		ret = -EBADFD;
J
Jason Wang 已提交
2568
		goto out_free;
W
Wei Xu 已提交
2569
	}
J
Jason Wang 已提交
2570

2571
	if (flags & ~(MSG_DONTWAIT|MSG_TRUNC|MSG_ERRQUEUE)) {
2572
		ret = -EINVAL;
W
Wei Xu 已提交
2573
		goto out_put_tun;
2574
	}
2575 2576 2577 2578 2579
	if (flags & MSG_ERRQUEUE) {
		ret = sock_recv_errqueue(sock->sk, m, total_len,
					 SOL_PACKET, TUN_TX_TIMESTAMP);
		goto out;
	}
J
Jason Wang 已提交
2580
	ret = tun_do_read(tun, tfile, &m->msg_iter, flags & MSG_DONTWAIT, ptr);
2581
	if (ret > (ssize_t)total_len) {
2582 2583 2584
		m->msg_flags |= MSG_TRUNC;
		ret = flags & MSG_TRUNC ? ret : total_len;
	}
2585
out:
J
Jason Wang 已提交
2586
	tun_put(tun);
2587
	return ret;
W
Wei Xu 已提交
2588 2589 2590

out_put_tun:
	tun_put(tun);
J
Jason Wang 已提交
2591 2592
out_free:
	tun_ptr_free(ptr);
W
Wei Xu 已提交
2593
	return ret;
2594 2595
}

J
Jason Wang 已提交
2596 2597 2598
static int tun_ptr_peek_len(void *ptr)
{
	if (likely(ptr)) {
2599 2600
		if (tun_is_xdp_frame(ptr)) {
			struct xdp_frame *xdpf = tun_ptr_to_xdp(ptr);
J
Jason Wang 已提交
2601

2602
			return xdpf->len;
J
Jason Wang 已提交
2603 2604 2605 2606 2607 2608 2609
		}
		return __skb_array_len_with_tag(ptr);
	} else {
		return 0;
	}
}

J
Jason Wang 已提交
2610 2611 2612 2613 2614 2615
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;

2616
	tun = tun_get(tfile);
J
Jason Wang 已提交
2617 2618 2619
	if (!tun)
		return 0;

J
Jason Wang 已提交
2620
	ret = PTR_RING_PEEK_CALL(&tfile->tx_ring, tun_ptr_peek_len);
J
Jason Wang 已提交
2621 2622 2623 2624 2625
	tun_put(tun);

	return ret;
}

2626 2627
/* Ops structure to mimic raw sockets with tun */
static const struct proto_ops tun_socket_ops = {
J
Jason Wang 已提交
2628
	.peek_len = tun_peek_len,
2629 2630 2631 2632
	.sendmsg = tun_sendmsg,
	.recvmsg = tun_recvmsg,
};

2633 2634 2635
static struct proto tun_proto = {
	.name		= "tun",
	.owner		= THIS_MODULE,
J
Jason Wang 已提交
2636
	.obj_size	= sizeof(struct tun_file),
2637
};
2638

2639 2640
static int tun_flags(struct tun_struct *tun)
{
2641
	return tun->flags & (TUN_FEATURES | IFF_PERSIST | IFF_TUN | IFF_TAP);
2642 2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654
}

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));
2655 2656 2657 2658
	return uid_valid(tun->owner)?
		sprintf(buf, "%u\n",
			from_kuid_munged(current_user_ns(), tun->owner)):
		sprintf(buf, "-1\n");
2659 2660 2661 2662 2663 2664
}

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));
2665 2666 2667 2668
	return gid_valid(tun->group) ?
		sprintf(buf, "%u\n",
			from_kgid_munged(current_user_ns(), tun->group)):
		sprintf(buf, "-1\n");
2669 2670 2671 2672 2673 2674
}

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

2675 2676 2677 2678 2679 2680 2681 2682 2683 2684 2685
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
};

2686
static int tun_set_iff(struct net *net, struct file *file, struct ifreq *ifr)
L
Linus Torvalds 已提交
2687 2688
{
	struct tun_struct *tun;
J
Jason Wang 已提交
2689
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
2690 2691 2692
	struct net_device *dev;
	int err;

2693 2694 2695
	if (tfile->detached)
		return -EINVAL;

2696 2697 2698 2699 2700 2701 2702 2703 2704
	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;
	}

2705 2706
	dev = __dev_get_by_name(net, ifr->ifr_name);
	if (dev) {
2707 2708
		if (ifr->ifr_flags & IFF_TUN_EXCL)
			return -EBUSY;
2709 2710 2711 2712 2713 2714 2715
		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;

2716
		if (!!(ifr->ifr_flags & IFF_MULTI_QUEUE) !=
M
Michael S. Tsirkin 已提交
2717
		    !!(tun->flags & IFF_MULTI_QUEUE))
2718 2719
			return -EINVAL;

2720
		if (tun_not_capable(tun))
P
Paul Moore 已提交
2721
			return -EPERM;
2722
		err = security_tun_dev_open(tun->security);
P
Paul Moore 已提交
2723 2724 2725
		if (err < 0)
			return err;

2726
		err = tun_attach(tun, file, ifr->ifr_flags & IFF_NOFILTER,
E
Eric Dumazet 已提交
2727 2728
				 ifr->ifr_flags & IFF_NAPI,
				 ifr->ifr_flags & IFF_NAPI_FRAGS);
2729 2730
		if (err < 0)
			return err;
J
Jason Wang 已提交
2731

M
Michael S. Tsirkin 已提交
2732
		if (tun->flags & IFF_MULTI_QUEUE &&
2733 2734 2735 2736
		    (tun->numqueues + tun->numdisabled > 1)) {
			/* One or more queue has already been attached, no need
			 * to initialize the device again.
			 */
2737
			netdev_state_change(dev);
2738 2739
			return 0;
		}
2740 2741 2742

		tun->flags = (tun->flags & ~TUN_FEATURES) |
			      (ifr->ifr_flags & TUN_FEATURES);
2743 2744 2745

		netdev_state_change(dev);
	} else {
L
Linus Torvalds 已提交
2746 2747
		char *name;
		unsigned long flags = 0;
2748 2749
		int queues = ifr->ifr_flags & IFF_MULTI_QUEUE ?
			     MAX_TAP_QUEUES : 1;
L
Linus Torvalds 已提交
2750

2751
		if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
2752
			return -EPERM;
P
Paul Moore 已提交
2753 2754 2755
		err = security_tun_dev_create();
		if (err < 0)
			return err;
2756

L
Linus Torvalds 已提交
2757 2758 2759
		/* Set dev type */
		if (ifr->ifr_flags & IFF_TUN) {
			/* TUN device */
M
Michael S. Tsirkin 已提交
2760
			flags |= IFF_TUN;
L
Linus Torvalds 已提交
2761 2762 2763
			name = "tun%d";
		} else if (ifr->ifr_flags & IFF_TAP) {
			/* TAP device */
M
Michael S. Tsirkin 已提交
2764
			flags |= IFF_TAP;
L
Linus Torvalds 已提交
2765
			name = "tap%d";
2766
		} else
2767
			return -EINVAL;
2768

L
Linus Torvalds 已提交
2769 2770 2771
		if (*ifr->ifr_name)
			name = ifr->ifr_name;

J
Jason Wang 已提交
2772
		dev = alloc_netdev_mqs(sizeof(struct tun_struct), name,
2773 2774
				       NET_NAME_UNKNOWN, tun_setup, queues,
				       queues);
2775

L
Linus Torvalds 已提交
2776 2777
		if (!dev)
			return -ENOMEM;
2778
		err = dev_get_valid_name(net, dev, name);
2779
		if (err < 0)
2780
			goto err_free_dev;
L
Linus Torvalds 已提交
2781

2782
		dev_net_set(dev, net);
2783
		dev->rtnl_link_ops = &tun_link_ops;
2784
		dev->ifindex = tfile->ifindex;
2785
		dev->sysfs_groups[0] = &tun_attr_group;
S
Stephen Hemminger 已提交
2786

L
Linus Torvalds 已提交
2787 2788 2789
		tun = netdev_priv(dev);
		tun->dev = dev;
		tun->flags = flags;
2790
		tun->txflt.count = 0;
2791
		tun->vnet_hdr_sz = sizeof(struct virtio_net_hdr);
2792

2793
		tun->align = NET_SKB_PAD;
J
Jason Wang 已提交
2794 2795
		tun->filter_attached = false;
		tun->sndbuf = tfile->socket.sk->sk_sndbuf;
J
Jason Wang 已提交
2796
		tun->rx_batched = 0;
2797
		RCU_INIT_POINTER(tun->steering_prog, NULL);
2798

2799 2800 2801 2802 2803 2804
		tun->pcpu_stats = netdev_alloc_pcpu_stats(struct tun_pcpu_stats);
		if (!tun->pcpu_stats) {
			err = -ENOMEM;
			goto err_free_dev;
		}

J
Jason Wang 已提交
2805 2806
		spin_lock_init(&tun->lock);

2807 2808
		err = security_tun_dev_alloc_security(&tun->security);
		if (err < 0)
2809
			goto err_free_stat;
P
Paul Moore 已提交
2810

L
Linus Torvalds 已提交
2811
		tun_net_init(dev);
2812
		tun_flow_init(tun);
J
Jason Wang 已提交
2813

2814
		dev->hw_features = NETIF_F_SG | NETIF_F_FRAGLIST |
J
Jason Wang 已提交
2815 2816
				   TUN_USER_FEATURES | NETIF_F_HW_VLAN_CTAG_TX |
				   NETIF_F_HW_VLAN_STAG_TX;
2817
		dev->features = dev->hw_features | NETIF_F_LLTX;
2818 2819 2820
		dev->vlan_features = dev->features &
				     ~(NETIF_F_HW_VLAN_CTAG_TX |
				       NETIF_F_HW_VLAN_STAG_TX);
2821

2822 2823 2824
		tun->flags = (tun->flags & ~TUN_FEATURES) |
			      (ifr->ifr_flags & TUN_FEATURES);

J
Jason Wang 已提交
2825
		INIT_LIST_HEAD(&tun->disabled);
E
Eric Dumazet 已提交
2826 2827
		err = tun_attach(tun, file, false, ifr->ifr_flags & IFF_NAPI,
				 ifr->ifr_flags & IFF_NAPI_FRAGS);
2828
		if (err < 0)
2829
			goto err_free_flow;
2830

L
Linus Torvalds 已提交
2831 2832
		err = register_netdevice(tun->dev);
		if (err < 0)
2833
			goto err_detach;
L
Linus Torvalds 已提交
2834 2835
	}

2836 2837
	netif_carrier_on(tun->dev);

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

2840 2841 2842 2843
	/* Make sure persistent devices do not get stuck in
	 * xoff state.
	 */
	if (netif_running(tun->dev))
J
Jason Wang 已提交
2844
		netif_tx_wake_all_queues(tun->dev);
2845

L
Linus Torvalds 已提交
2846 2847 2848
	strcpy(ifr->ifr_name, tun->dev->name);
	return 0;

2849 2850
err_detach:
	tun_detach_all(dev);
2851 2852 2853
	/* register_netdevice() already called tun_free_netdev() */
	goto err_free_dev;

2854 2855 2856
err_free_flow:
	tun_flow_uninit(tun);
	security_tun_dev_free_security(tun->security);
2857 2858
err_free_stat:
	free_percpu(tun->pcpu_stats);
2859
err_free_dev:
L
Linus Torvalds 已提交
2860 2861 2862 2863
	free_netdev(dev);
	return err;
}

R
Rami Rosen 已提交
2864
static void tun_get_iff(struct net *net, struct tun_struct *tun,
2865
		       struct ifreq *ifr)
2866
{
2867
	tun_debug(KERN_INFO, tun, "tun_get_iff\n");
2868 2869 2870

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

2871
	ifr->ifr_flags = tun_flags(tun);
2872 2873 2874

}

2875 2876
/* This is like a cut-down ethtool ops, except done via tun fd so no
 * privs required. */
2877
static int set_offload(struct tun_struct *tun, unsigned long arg)
2878
{
2879
	netdev_features_t features = 0;
2880 2881

	if (arg & TUN_F_CSUM) {
2882
		features |= NETIF_F_HW_CSUM;
2883 2884 2885 2886 2887 2888 2889 2890 2891 2892 2893 2894 2895
		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);
		}
2896 2897

		arg &= ~TUN_F_UFO;
2898 2899 2900 2901 2902 2903 2904
	}

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

2905
	tun->set_features = features;
2906 2907
	tun->dev->wanted_features &= ~TUN_USER_FEATURES;
	tun->dev->wanted_features |= features;
2908
	netdev_update_features(tun->dev);
2909 2910 2911 2912

	return 0;
}

J
Jason Wang 已提交
2913 2914 2915 2916 2917 2918
static void tun_detach_filter(struct tun_struct *tun, int n)
{
	int i;
	struct tun_file *tfile;

	for (i = 0; i < n; i++) {
2919
		tfile = rtnl_dereference(tun->tfiles[i]);
2920 2921 2922
		lock_sock(tfile->socket.sk);
		sk_detach_filter(tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2923 2924 2925 2926 2927 2928 2929 2930 2931 2932 2933
	}

	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++) {
2934
		tfile = rtnl_dereference(tun->tfiles[i]);
2935 2936 2937
		lock_sock(tfile->socket.sk);
		ret = sk_attach_filter(&tun->fprog, tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2938 2939 2940 2941 2942 2943 2944 2945 2946 2947 2948 2949 2950 2951 2952 2953
		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++) {
2954
		tfile = rtnl_dereference(tun->tfiles[i]);
J
Jason Wang 已提交
2955 2956 2957 2958
		tfile->socket.sk->sk_sndbuf = tun->sndbuf;
	}
}

2959 2960 2961 2962 2963 2964 2965 2966 2967
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 已提交
2968
		tun = tfile->detached;
2969
		if (!tun) {
2970
			ret = -EINVAL;
2971 2972 2973 2974 2975
			goto unlock;
		}
		ret = security_tun_dev_attach_queue(tun->security);
		if (ret < 0)
			goto unlock;
E
Eric Dumazet 已提交
2976 2977
		ret = tun_attach(tun, file, false, tun->flags & IFF_NAPI,
				 tun->flags & IFF_NAPI_FRAGS);
J
Jason Wang 已提交
2978
	} else if (ifr->ifr_flags & IFF_DETACH_QUEUE) {
2979
		tun = rtnl_dereference(tfile->tun);
M
Michael S. Tsirkin 已提交
2980
		if (!tun || !(tun->flags & IFF_MULTI_QUEUE) || tfile->detached)
J
Jason Wang 已提交
2981 2982 2983 2984
			ret = -EINVAL;
		else
			__tun_detach(tfile, false);
	} else
2985 2986
		ret = -EINVAL;

2987 2988 2989
	if (ret >= 0)
		netdev_state_change(tun->dev);

2990
unlock:
2991 2992 2993 2994
	rtnl_unlock();
	return ret;
}

2995 2996
static int tun_set_ebpf(struct tun_struct *tun, struct tun_prog **prog_p,
			void __user *data)
2997 2998 2999 3000 3001 3002 3003 3004 3005 3006 3007 3008 3009 3010 3011
{
	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);
	}

3012
	return __tun_set_ebpf(tun, prog_p, prog);
3013 3014
}

3015 3016
static long __tun_chr_ioctl(struct file *file, unsigned int cmd,
			    unsigned long arg, int ifreq_len)
L
Linus Torvalds 已提交
3017
{
E
Eric W. Biederman 已提交
3018
	struct tun_file *tfile = file->private_data;
3019
	struct net *net = sock_net(&tfile->sk);
E
Eric W. Biederman 已提交
3020
	struct tun_struct *tun;
L
Linus Torvalds 已提交
3021
	void __user* argp = (void __user*)arg;
N
Nicolas Dichtel 已提交
3022
	unsigned int ifindex, carrier;
L
Linus Torvalds 已提交
3023
	struct ifreq ifr;
3024 3025
	kuid_t owner;
	kgid_t group;
3026
	int sndbuf;
3027
	int vnet_hdr_sz;
M
Michael S. Tsirkin 已提交
3028
	int le;
3029
	int ret;
3030
	bool do_notify = false;
L
Linus Torvalds 已提交
3031

3032 3033
	if (cmd == TUNSETIFF || cmd == TUNSETQUEUE ||
	    (_IOC_TYPE(cmd) == SOCK_IOC_TYPE && cmd != SIOCGSKNS)) {
3034
		if (copy_from_user(&ifr, argp, ifreq_len))
L
Linus Torvalds 已提交
3035
			return -EFAULT;
D
David S. Miller 已提交
3036
	} else {
3037
		memset(&ifr, 0, sizeof(ifr));
D
David S. Miller 已提交
3038
	}
E
Eric W. Biederman 已提交
3039 3040 3041
	if (cmd == TUNGETFEATURES) {
		/* Currently this just means: "what IFF flags are valid?".
		 * This is needed because we never checked for invalid flags on
3042 3043 3044
		 * TUNSETIFF.
		 */
		return put_user(IFF_TUN | IFF_TAP | TUN_FEATURES,
E
Eric W. Biederman 已提交
3045
				(unsigned int __user*)argp);
3046
	} else if (cmd == TUNSETQUEUE) {
3047
		return tun_set_queue(file, &ifr);
3048 3049 3050 3051 3052
	} else if (cmd == SIOCGSKNS) {
		if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
			return -EPERM;
		return open_related_ns(&net->ns, get_net_ns);
	}
E
Eric W. Biederman 已提交
3053

J
Jason Wang 已提交
3054
	ret = 0;
3055 3056
	rtnl_lock();

3057
	tun = tun_get(tfile);
3058 3059 3060 3061 3062
	if (cmd == TUNSETIFF) {
		ret = -EEXIST;
		if (tun)
			goto unlock;

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

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

3067 3068
		if (ret)
			goto unlock;
L
Linus Torvalds 已提交
3069

3070
		if (copy_to_user(argp, &ifr, ifreq_len))
3071 3072
			ret = -EFAULT;
		goto unlock;
L
Linus Torvalds 已提交
3073
	}
3074 3075 3076 3077 3078 3079 3080 3081 3082 3083 3084 3085 3086
	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 已提交
3087

3088
	ret = -EBADFD;
L
Linus Torvalds 已提交
3089
	if (!tun)
3090
		goto unlock;
L
Linus Torvalds 已提交
3091

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

E
Eric W. Biederman 已提交
3094
	ret = 0;
L
Linus Torvalds 已提交
3095
	switch (cmd) {
3096
	case TUNGETIFF:
R
Rami Rosen 已提交
3097
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
3098

3099 3100
		if (tfile->detached)
			ifr.ifr_flags |= IFF_DETACH_QUEUE;
3101 3102
		if (!tfile->socket.sk->sk_filter)
			ifr.ifr_flags |= IFF_NOFILTER;
3103

3104
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
3105
			ret = -EFAULT;
3106 3107
		break;

L
Linus Torvalds 已提交
3108 3109 3110
	case TUNSETNOCSUM:
		/* Disable/Enable checksum */

3111 3112
		/* [unimplemented] */
		tun_debug(KERN_INFO, tun, "ignored: set checksum %s\n",
3113
			  arg ? "disabled" : "enabled");
L
Linus Torvalds 已提交
3114 3115 3116
		break;

	case TUNSETPERSIST:
J
Jason Wang 已提交
3117 3118 3119
		/* Disable/Enable persist mode. Keep an extra reference to the
		 * module to prevent the module being unprobed.
		 */
M
Michael S. Tsirkin 已提交
3120 3121
		if (arg && !(tun->flags & IFF_PERSIST)) {
			tun->flags |= IFF_PERSIST;
J
Jason Wang 已提交
3122
			__module_get(THIS_MODULE);
3123
			do_notify = true;
J
Jason Wang 已提交
3124
		}
M
Michael S. Tsirkin 已提交
3125 3126
		if (!arg && (tun->flags & IFF_PERSIST)) {
			tun->flags &= ~IFF_PERSIST;
J
Jason Wang 已提交
3127
			module_put(THIS_MODULE);
3128
			do_notify = true;
J
Jason Wang 已提交
3129
		}
L
Linus Torvalds 已提交
3130

3131 3132
		tun_debug(KERN_INFO, tun, "persist %s\n",
			  arg ? "enabled" : "disabled");
L
Linus Torvalds 已提交
3133 3134 3135 3136
		break;

	case TUNSETOWNER:
		/* Set owner of the device */
3137 3138 3139 3140 3141 3142
		owner = make_kuid(current_user_ns(), arg);
		if (!uid_valid(owner)) {
			ret = -EINVAL;
			break;
		}
		tun->owner = owner;
3143
		do_notify = true;
3144
		tun_debug(KERN_INFO, tun, "owner set to %u\n",
3145
			  from_kuid(&init_user_ns, tun->owner));
L
Linus Torvalds 已提交
3146 3147
		break;

3148 3149
	case TUNSETGROUP:
		/* Set group of the device */
3150 3151 3152 3153 3154 3155
		group = make_kgid(current_user_ns(), arg);
		if (!gid_valid(group)) {
			ret = -EINVAL;
			break;
		}
		tun->group = group;
3156
		do_notify = true;
3157
		tun_debug(KERN_INFO, tun, "group set to %u\n",
3158
			  from_kgid(&init_user_ns, tun->group));
3159 3160
		break;

3161 3162 3163
	case TUNSETLINK:
		/* Only allow setting the type when the interface is down */
		if (tun->dev->flags & IFF_UP) {
3164 3165
			tun_debug(KERN_INFO, tun,
				  "Linktype set failed because interface is up\n");
3166
			ret = -EBUSY;
3167 3168
		} else {
			tun->dev->type = (int) arg;
3169 3170
			tun_debug(KERN_INFO, tun, "linktype set to %d\n",
				  tun->dev->type);
3171
			ret = 0;
3172
		}
E
Eric W. Biederman 已提交
3173
		break;
3174

L
Linus Torvalds 已提交
3175 3176 3177 3178 3179
#ifdef TUN_DEBUG
	case TUNSETDEBUG:
		tun->debug = arg;
		break;
#endif
3180
	case TUNSETOFFLOAD:
3181
		ret = set_offload(tun, arg);
E
Eric W. Biederman 已提交
3182
		break;
3183

3184 3185
	case TUNSETTXFILTER:
		/* Can be set only for TAPs */
E
Eric W. Biederman 已提交
3186
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3187
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
E
Eric W. Biederman 已提交
3188
			break;
H
Harvey Harrison 已提交
3189
		ret = update_filter(&tun->txflt, (void __user *)arg);
E
Eric W. Biederman 已提交
3190
		break;
L
Linus Torvalds 已提交
3191 3192

	case SIOCGIFHWADDR:
3193
		/* Get hw address */
3194 3195
		memcpy(ifr.ifr_hwaddr.sa_data, tun->dev->dev_addr, ETH_ALEN);
		ifr.ifr_hwaddr.sa_family = tun->dev->type;
3196
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
3197 3198
			ret = -EFAULT;
		break;
L
Linus Torvalds 已提交
3199 3200

	case SIOCSIFHWADDR:
3201
		/* Set hw address */
3202 3203
		tun_debug(KERN_DEBUG, tun, "set hw address: %pM\n",
			  ifr.ifr_hwaddr.sa_data);
3204 3205

		ret = dev_set_mac_address(tun->dev, &ifr.ifr_hwaddr);
E
Eric W. Biederman 已提交
3206
		break;
3207 3208

	case TUNGETSNDBUF:
J
Jason Wang 已提交
3209
		sndbuf = tfile->socket.sk->sk_sndbuf;
3210 3211 3212 3213 3214 3215 3216 3217 3218
		if (copy_to_user(argp, &sndbuf, sizeof(sndbuf)))
			ret = -EFAULT;
		break;

	case TUNSETSNDBUF:
		if (copy_from_user(&sndbuf, argp, sizeof(sndbuf))) {
			ret = -EFAULT;
			break;
		}
3219 3220 3221 3222
		if (sndbuf <= 0) {
			ret = -EINVAL;
			break;
		}
3223

J
Jason Wang 已提交
3224 3225
		tun->sndbuf = sndbuf;
		tun_set_sndbuf(tun);
3226 3227
		break;

3228 3229 3230 3231 3232 3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246
	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 已提交
3247 3248 3249 3250 3251 3252 3253 3254 3255 3256 3257 3258 3259 3260 3261 3262 3263
	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;

3264 3265 3266 3267 3268 3269 3270 3271
	case TUNGETVNETBE:
		ret = tun_get_vnet_be(tun, argp);
		break;

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

M
Michael S. Tsirkin 已提交
3272 3273 3274
	case TUNATTACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3275
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
3276 3277
			break;
		ret = -EFAULT;
J
Jason Wang 已提交
3278
		if (copy_from_user(&tun->fprog, argp, sizeof(tun->fprog)))
M
Michael S. Tsirkin 已提交
3279 3280
			break;

J
Jason Wang 已提交
3281
		ret = tun_attach_filter(tun);
M
Michael S. Tsirkin 已提交
3282 3283 3284 3285 3286
		break;

	case TUNDETACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3287
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
3288
			break;
J
Jason Wang 已提交
3289 3290
		ret = 0;
		tun_detach_filter(tun, tun->numqueues);
M
Michael S. Tsirkin 已提交
3291 3292
		break;

P
Pavel Emelyanov 已提交
3293 3294
	case TUNGETFILTER:
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3295
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
P
Pavel Emelyanov 已提交
3296 3297 3298 3299 3300 3301 3302
			break;
		ret = -EFAULT;
		if (copy_to_user(argp, &tun->fprog, sizeof(tun->fprog)))
			break;
		ret = 0;
		break;

3303
	case TUNSETSTEERINGEBPF:
3304
		ret = tun_set_ebpf(tun, &tun->steering_prog, argp);
3305 3306
		break;

3307 3308 3309 3310
	case TUNSETFILTEREBPF:
		ret = tun_set_ebpf(tun, &tun->filter_prog, argp);
		break;

N
Nicolas Dichtel 已提交
3311 3312 3313 3314 3315 3316 3317 3318
	case TUNSETCARRIER:
		ret = -EFAULT;
		if (copy_from_user(&carrier, argp, sizeof(carrier)))
			goto unlock;

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

L
Linus Torvalds 已提交
3319
	default:
E
Eric W. Biederman 已提交
3320 3321
		ret = -EINVAL;
		break;
3322
	}
L
Linus Torvalds 已提交
3323

3324 3325 3326
	if (do_notify)
		netdev_state_change(tun->dev);

3327 3328 3329 3330
unlock:
	rtnl_unlock();
	if (tun)
		tun_put(tun);
E
Eric W. Biederman 已提交
3331
	return ret;
L
Linus Torvalds 已提交
3332 3333
}

3334 3335 3336 3337 3338 3339 3340 3341 3342 3343 3344 3345 3346 3347 3348 3349 3350 3351 3352 3353 3354 3355 3356 3357 3358 3359 3360 3361 3362 3363 3364 3365 3366 3367 3368
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 已提交
3369 3370
static int tun_chr_fasync(int fd, struct file *file, int on)
{
J
Jason Wang 已提交
3371
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
3372 3373
	int ret;

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

L
Linus Torvalds 已提交
3377
	if (on) {
3378
		__f_setown(file, task_pid(current), PIDTYPE_TGID, 0);
J
Jason Wang 已提交
3379
		tfile->flags |= TUN_FASYNC;
3380
	} else
J
Jason Wang 已提交
3381
		tfile->flags &= ~TUN_FASYNC;
J
Jonathan Corbet 已提交
3382 3383 3384
	ret = 0;
out:
	return ret;
L
Linus Torvalds 已提交
3385 3386 3387 3388
}

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

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

3394
	tfile = (struct tun_file *)sk_alloc(net, AF_UNSPEC, GFP_KERNEL,
3395
					    &tun_proto, 0);
E
Eric W. Biederman 已提交
3396 3397
	if (!tfile)
		return -ENOMEM;
3398 3399 3400 3401 3402
	if (ptr_ring_init(&tfile->tx_ring, 0, GFP_KERNEL)) {
		sk_free(&tfile->sk);
		return -ENOMEM;
	}

3403
	mutex_init(&tfile->napi_mutex);
3404
	RCU_INIT_POINTER(tfile->tun, NULL);
J
Jason Wang 已提交
3405
	tfile->flags = 0;
3406
	tfile->ifindex = 0;
J
Jason Wang 已提交
3407 3408

	init_waitqueue_head(&tfile->wq.wait);
3409
	RCU_INIT_POINTER(tfile->socket.wq, &tfile->wq);
J
Jason Wang 已提交
3410 3411 3412 3413 3414 3415 3416 3417 3418

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

3422 3423
	sock_set_flag(&tfile->sk, SOCK_ZEROCOPY);

L
Linus Torvalds 已提交
3424 3425 3426 3427 3428
	return 0;
}

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

J
Jason Wang 已提交
3431
	tun_detach(tfile, true);
L
Linus Torvalds 已提交
3432 3433 3434 3435

	return 0;
}

3436
#ifdef CONFIG_PROC_FS
3437
static void tun_chr_show_fdinfo(struct seq_file *m, struct file *file)
3438
{
3439
	struct tun_file *tfile = file->private_data;
3440 3441 3442 3443 3444 3445
	struct tun_struct *tun;
	struct ifreq ifr;

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

	rtnl_lock();
3446
	tun = tun_get(tfile);
3447 3448 3449 3450 3451 3452 3453
	if (tun)
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
	rtnl_unlock();

	if (tun)
		tun_put(tun);

3454
	seq_printf(m, "iff:\t%s\n", ifr.ifr_name);
3455 3456 3457
}
#endif

3458
static const struct file_operations tun_fops = {
3459
	.owner	= THIS_MODULE,
L
Linus Torvalds 已提交
3460
	.llseek = no_llseek,
3461
	.read_iter  = tun_chr_read_iter,
3462
	.write_iter = tun_chr_write_iter,
L
Linus Torvalds 已提交
3463
	.poll	= tun_chr_poll,
3464 3465 3466 3467
	.unlocked_ioctl	= tun_chr_ioctl,
#ifdef CONFIG_COMPAT
	.compat_ioctl = tun_chr_compat_ioctl,
#endif
L
Linus Torvalds 已提交
3468 3469
	.open	= tun_chr_open,
	.release = tun_chr_close,
3470 3471 3472 3473
	.fasync = tun_chr_fasync,
#ifdef CONFIG_PROC_FS
	.show_fdinfo = tun_chr_show_fdinfo,
#endif
L
Linus Torvalds 已提交
3474 3475 3476 3477 3478
};

static struct miscdevice tun_miscdev = {
	.minor = TUN_MINOR,
	.name = "tun",
3479
	.nodename = "net/tun",
L
Linus Torvalds 已提交
3480 3481 3482 3483 3484
	.fops = &tun_fops,
};

/* ethtool interface */

3485 3486
static void tun_default_link_ksettings(struct net_device *dev,
				       struct ethtool_link_ksettings *cmd)
3487 3488 3489 3490 3491 3492 3493 3494
{
	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;
3495 3496 3497 3498 3499 3500 3501 3502 3503 3504 3505 3506 3507 3508 3509 3510 3511
}

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

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

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

	memcpy(&tun->link_ksettings, cmd, sizeof(*cmd));
L
Linus Torvalds 已提交
3512 3513 3514 3515 3516 3517 3518
	return 0;
}

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

3519 3520
	strlcpy(info->driver, DRV_NAME, sizeof(info->driver));
	strlcpy(info->version, DRV_VERSION, sizeof(info->version));
L
Linus Torvalds 已提交
3521 3522

	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
3523
	case IFF_TUN:
3524
		strlcpy(info->bus_info, "tun", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3525
		break;
M
Michael S. Tsirkin 已提交
3526
	case IFF_TAP:
3527
		strlcpy(info->bus_info, "tap", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3528 3529 3530 3531 3532 3533 3534 3535 3536 3537 3538 3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549
		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 已提交
3550 3551 3552 3553 3554 3555 3556 3557 3558 3559 3560 3561 3562 3563 3564 3565 3566 3567 3568 3569 3570 3571 3572
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;
}

3573
static const struct ethtool_ops tun_ethtool_ops = {
L
Linus Torvalds 已提交
3574 3575 3576
	.get_drvinfo	= tun_get_drvinfo,
	.get_msglevel	= tun_get_msglevel,
	.set_msglevel	= tun_set_msglevel,
3577
	.get_link	= ethtool_op_get_link,
3578
	.get_ts_info	= ethtool_op_get_ts_info,
J
Jason Wang 已提交
3579 3580
	.get_coalesce   = tun_get_coalesce,
	.set_coalesce   = tun_set_coalesce,
3581
	.get_link_ksettings = tun_get_link_ksettings,
3582
	.set_link_ksettings = tun_set_link_ksettings,
L
Linus Torvalds 已提交
3583 3584
};

J
Jason Wang 已提交
3585 3586 3587 3588
static int tun_queue_resize(struct tun_struct *tun)
{
	struct net_device *dev = tun->dev;
	struct tun_file *tfile;
3589
	struct ptr_ring **rings;
J
Jason Wang 已提交
3590 3591 3592
	int n = tun->numqueues + tun->numdisabled;
	int ret, i;

3593 3594
	rings = kmalloc_array(n, sizeof(*rings), GFP_KERNEL);
	if (!rings)
J
Jason Wang 已提交
3595 3596 3597 3598
		return -ENOMEM;

	for (i = 0; i < tun->numqueues; i++) {
		tfile = rtnl_dereference(tun->tfiles[i]);
3599
		rings[i] = &tfile->tx_ring;
J
Jason Wang 已提交
3600 3601
	}
	list_for_each_entry(tfile, &tun->disabled, next)
3602
		rings[i++] = &tfile->tx_ring;
J
Jason Wang 已提交
3603

3604 3605
	ret = ptr_ring_resize_multiple(rings, n,
				       dev->tx_queue_len, GFP_KERNEL,
J
Jason Wang 已提交
3606
				       tun_ptr_free);
J
Jason Wang 已提交
3607

3608
	kfree(rings);
J
Jason Wang 已提交
3609 3610 3611 3612 3613 3614 3615 3616 3617
	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);

3618 3619 3620
	if (dev->rtnl_link_ops != &tun_link_ops)
		return NOTIFY_DONE;

J
Jason Wang 已提交
3621 3622 3623 3624 3625 3626 3627 3628 3629 3630 3631 3632 3633 3634 3635
	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,
};
3636

L
Linus Torvalds 已提交
3637 3638 3639 3640
static int __init tun_init(void)
{
	int ret = 0;

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

3643
	ret = rtnl_link_register(&tun_link_ops);
3644
	if (ret) {
3645
		pr_err("Can't register link_ops\n");
3646
		goto err_linkops;
3647 3648
	}

L
Linus Torvalds 已提交
3649
	ret = misc_register(&tun_miscdev);
3650
	if (ret) {
3651
		pr_err("Can't register misc device %d\n", TUN_MINOR);
3652 3653
		goto err_misc;
	}
J
Jason Wang 已提交
3654

3655 3656 3657 3658 3659 3660
	ret = register_netdevice_notifier(&tun_notifier_block);
	if (ret) {
		pr_err("Can't register netdevice notifier\n");
		goto err_notifier;
	}

3661
	return  0;
3662 3663 3664

err_notifier:
	misc_deregister(&tun_miscdev);
3665
err_misc:
3666 3667
	rtnl_link_unregister(&tun_link_ops);
err_linkops:
L
Linus Torvalds 已提交
3668 3669 3670 3671 3672
	return ret;
}

static void tun_cleanup(void)
{
3673
	misc_deregister(&tun_miscdev);
3674
	rtnl_link_unregister(&tun_link_ops);
J
Jason Wang 已提交
3675
	unregister_netdevice_notifier(&tun_notifier_block);
L
Linus Torvalds 已提交
3676 3677
}

3678 3679 3680 3681 3682 3683
/* 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)
{
3684
	struct tun_file *tfile;
3685 3686
	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
3687 3688
	tfile = file->private_data;
	if (!tfile)
3689
		return ERR_PTR(-EBADFD);
J
Jason Wang 已提交
3690
	return &tfile->socket;
3691 3692 3693
}
EXPORT_SYMBOL_GPL(tun_get_socket);

3694
struct ptr_ring *tun_get_tx_ring(struct file *file)
J
Jason Wang 已提交
3695 3696 3697 3698 3699 3700 3701 3702
{
	struct tun_file *tfile;

	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
	tfile = file->private_data;
	if (!tfile)
		return ERR_PTR(-EBADFD);
3703
	return &tfile->tx_ring;
J
Jason Wang 已提交
3704
}
3705
EXPORT_SYMBOL_GPL(tun_get_tx_ring);
J
Jason Wang 已提交
3706

L
Linus Torvalds 已提交
3707 3708 3709 3710 3711 3712
module_init(tun_init);
module_exit(tun_cleanup);
MODULE_DESCRIPTION(DRV_DESCRIPTION);
MODULE_AUTHOR(DRV_COPYRIGHT);
MODULE_LICENSE("GPL");
MODULE_ALIAS_MISCDEV(TUN_MINOR);
3713
MODULE_ALIAS("devname:net/tun");