tun.c 85.6 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
	int queue_index;
204
	unsigned long updated ____cacheline_aligned_in_smp;
J
Jason Wang 已提交
205 206 207
};

#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
	head = &tun->flows[tun_hashfn(rxhash)];
J
Jason Wang 已提交
533 534 535 536 537 538

	rcu_read_lock();

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

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

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

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

J
Jason Wang 已提交
583
	txq = __skb_get_hash_symmetric(skb);
584 585 586 587 588 589 590
	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 已提交
591 592 593 594 595
	}

	return txq;
}

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

	err = 0;
J
Jason Wang 已提交
818

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

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

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

	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;
850 851 852 853 854 855
		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 已提交
856 857 858
		err = 0;
	}

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

J
Jason Wang 已提交
866 867 868
	if (rtnl_dereference(tun->xdp_prog))
		sock_set_flag(&tfile->sk, SOCK_XDP);

J
Jason Wang 已提交
869
	tun_set_real_num_queues(tun);
870

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

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

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

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

	return tun;
E
Eric W. Biederman 已提交
897 898 899 900
}

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

904
/* TAP filtering */
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 931 932
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;
933 934 935
	addr = memdup_user(arg + sizeof(uf), alen);
	if (IS_ERR(addr))
		return PTR_ERR(addr);
936 937 938 939 940 941 942 943 944 945 946 947 948

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

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

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

1010
static const struct ethtool_ops tun_ethtool_ops;
L
Linus Torvalds 已提交
1011

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

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

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

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

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

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

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

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

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

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

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

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

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

J
Jason Wang 已提交
1094 1095
	BUG_ON(!tfile);

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

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

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

1113
	skb_tx_timestamp(skb);
1114

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

1120 1121
	nf_reset(skb);

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

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

	rcu_read_unlock();
1131
	return NETDEV_TX_OK;
L
Linus Torvalds 已提交
1132 1133

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

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

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

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

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

1168
static void
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 1202 1203
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 已提交
1204 1205 1206 1207
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 已提交
1208
	struct tun_file *tfile;
J
Jason Wang 已提交
1209
	struct bpf_prog *old_prog;
J
Jason Wang 已提交
1210
	int i;
J
Jason Wang 已提交
1211 1212 1213 1214 1215 1216

	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 已提交
1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230
	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 已提交
1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245
	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;
}

1246
static int tun_xdp(struct net_device *dev, struct netdev_bpf *xdp)
J
Jason Wang 已提交
1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258
{
	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 已提交
1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273
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 已提交
1274
static const struct net_device_ops tun_netdev_ops = {
1275
	.ndo_uninit		= tun_net_uninit,
S
Stephen Hemminger 已提交
1276 1277
	.ndo_open		= tun_net_open,
	.ndo_stop		= tun_net_close,
1278
	.ndo_start_xmit		= tun_net_xmit,
1279
	.ndo_fix_features	= tun_net_fix_features,
J
Jason Wang 已提交
1280
	.ndo_select_queue	= tun_select_queue,
1281
	.ndo_set_rx_headroom	= tun_set_headroom,
1282
	.ndo_get_stats64	= tun_net_get_stats64,
N
Nicolas Dichtel 已提交
1283
	.ndo_change_carrier	= tun_net_change_carrier,
S
Stephen Hemminger 已提交
1284 1285
};

1286 1287 1288 1289 1290 1291 1292 1293
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);
}

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

1304
	if (unlikely(flags & ~XDP_XMIT_FLAGS_MASK))
1305 1306
		return -EINVAL;

J
Jason Wang 已提交
1307 1308 1309 1310
	rcu_read_lock();

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

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

	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 已提交
1331
	}
1332
	spin_unlock(&tfile->tx_ring.producer_lock);
J
Jason Wang 已提交
1333

1334 1335 1336
	if (flags & XDP_XMIT_FLUSH)
		__tun_xdp_flush_tfile(tfile);

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

1341 1342 1343 1344 1345 1346 1347
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;

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

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

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

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

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

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

1388 1389 1390
#define MIN_MTU 68
#define MAX_MTU 65535

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

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

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

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

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

1417
		eth_hw_addr_random(dev);
1418

L
Linus Torvalds 已提交
1419 1420
		break;
	}
1421 1422 1423

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

1426 1427 1428 1429 1430 1431 1432
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 已提交
1433 1434 1435
/* Character device part */

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

	if (!tun)
1444
		return EPOLLERR;
L
Linus Torvalds 已提交
1445

J
Jason Wang 已提交
1446
	sk = tfile->socket.sk;
1447

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

1450
	poll_wait(file, sk_sleep(sk), wait);
1451

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

1455 1456 1457 1458 1459 1460 1461 1462
	/* 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)))
1463
		mask |= EPOLLOUT | EPOLLWRNORM;
1464

1465
	if (tun->dev->reg_state != NETREG_REGISTERED)
1466
		mask = EPOLLERR;
1467

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

1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500
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;
1501 1502
		struct page *page;
		void *frag;
1503 1504 1505 1506 1507

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

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

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

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

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

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

	return skb;
}

J
Jason Wang 已提交
1552 1553 1554 1555 1556 1557 1558 1559 1560 1561
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 已提交
1562
		skb_record_rx_queue(skb, tfile->queue_index);
J
Jason Wang 已提交
1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581
		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 已提交
1582 1583
		while ((nskb = __skb_dequeue(&process_queue))) {
			skb_record_rx_queue(nskb, tfile->queue_index);
J
Jason Wang 已提交
1584
			netif_receive_skb(nskb);
M
Matthew Cover 已提交
1585 1586
		}
		skb_record_rx_queue(skb, tfile->queue_index);
J
Jason Wang 已提交
1587 1588 1589 1590 1591
		netif_receive_skb(skb);
		local_bh_enable();
	}
}

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

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

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

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

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

	return skb;
}

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

1805 1806
	good_linear = SKB_MAX_HEAD(align);

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

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

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

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

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

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

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

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

1890 1891 1892
		return -EINVAL;
	}

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

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

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

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

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

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

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

1954 1955 1956 1957 1958 1959
	/* 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)
1960
		rxhash = __skb_get_hash_symmetric(skb);
1961

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

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

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

2004 2005 2006
	if (rxhash)
		tun_flow_update(tun, rxhash, tfile);

2007
	return total_len;
2008
}
L
Linus Torvalds 已提交
2009

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

	if (!tun)
		return -EBADFD;

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

	tun_put(tun);
	return result;
L
Linus Torvalds 已提交
2025 2026
}

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

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

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

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

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

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

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

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

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

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

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

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

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

		iov_iter_advance(iter, vnet_hdr_sz - sizeof(gso));
2122 2123
	}

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

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

		vlan_offset = offsetof(struct vlan_ethhdr, h_vlan_proto);

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

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

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

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

	return total;
}

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

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

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

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

		schedule();
	}

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

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

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

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

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

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

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

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

2233 2234 2235
	return ret;
}

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

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

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

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

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

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

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

	if (old)
2281
		call_rcu(&old->rcu, tun_prog_free);
2282 2283 2284 2285

	return 0;
}

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

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

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

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

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

2313 2314 2315
/* Trivial set of netlink ops to allow deleting tun or tap
 * device with netlink.
 */
2316 2317
static int tun_validate(struct nlattr *tb[], struct nlattr *data[],
			struct netlink_ext_ack *extack)
2318
{
2319 2320 2321
	NL_SET_ERR_MSG(extack,
		       "tun/tap creation via rtnetlink is not supported.");
	return -EOPNOTSUPP;
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 2376 2377
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;
}

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

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

	if (!sock_writeable(sk))
		return;

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

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

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

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

2413 2414
static int tun_xdp_one(struct tun_struct *tun,
		       struct tun_file *tfile,
2415 2416
		       struct xdp_buff *xdp, int *flush,
		       struct tun_page *tpage)
2417
{
2418
	unsigned int datasize = xdp->data_end - xdp->data;
2419 2420 2421 2422 2423 2424 2425 2426 2427
	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;
2428
	struct page *page;
2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454

	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:
2455 2456 2457 2458 2459 2460 2461 2462
			page = virt_to_head_page(xdp->data);
			if (tpage->page == page) {
				++tpage->count;
			} else {
				tun_put_page(tpage);
				tpage->page = page;
				tpage->count = 1;
			}
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 2491 2492 2493
			return 0;
		}
	}

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

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

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

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

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

2494 2495
	if (!rcu_dereference(tun->steering_prog) && tun->numqueues > 1 &&
	    !tfile->detached)
2496 2497
		rxhash = __skb_get_hash_symmetric(skb);

M
Matthew Cover 已提交
2498
	skb_record_rx_queue(skb, tfile->queue_index);
2499 2500
	netif_receive_skb(skb);

2501 2502 2503 2504
	/* No need for get_cpu_ptr() here since this function is
	 * always called with bh disabled
	 */
	stats = this_cpu_ptr(tun->pcpu_stats);
2505 2506
	u64_stats_update_begin(&stats->syncp);
	stats->rx_packets++;
2507
	stats->rx_bytes += datasize;
2508 2509 2510 2511 2512 2513 2514 2515 2516
	u64_stats_update_end(&stats->syncp);

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

out:
	return err;
}

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

	if (!tun)
		return -EBADFD;
2527

2528
	if (ctl && (ctl->type == TUN_MSG_PTR)) {
2529
		struct tun_page tpage;
2530 2531 2532
		int n = ctl->num;
		int flush = 0;

2533 2534
		memset(&tpage, 0, sizeof(tpage));

2535 2536 2537 2538 2539
		local_bh_disable();
		rcu_read_lock();

		for (i = 0; i < n; i++) {
			xdp = &((struct xdp_buff *)ctl->ptr)[i];
2540
			tun_xdp_one(tun, tfile, xdp, &flush, &tpage);
2541 2542 2543 2544 2545 2546 2547 2548
		}

		if (flush)
			xdp_do_flush_map();

		rcu_read_unlock();
		local_bh_enable();

2549 2550
		tun_put_page(&tpage);

2551 2552 2553
		ret = total_len;
		goto out;
	}
2554 2555

	ret = tun_get_user(tun, tfile, ctl ? ctl->ptr : NULL, &m->msg_iter,
J
Jason Wang 已提交
2556 2557
			   m->msg_flags & MSG_DONTWAIT,
			   m->msg_flags & MSG_MORE);
2558
out:
J
Jason Wang 已提交
2559 2560
	tun_put(tun);
	return ret;
2561 2562
}

2563
static int tun_recvmsg(struct socket *sock, struct msghdr *m, size_t total_len,
2564 2565
		       int flags)
{
J
Jason Wang 已提交
2566
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2567
	struct tun_struct *tun = tun_get(tfile);
J
Jason Wang 已提交
2568
	void *ptr = m->msg_control;
2569
	int ret;
J
Jason Wang 已提交
2570

W
Wei Xu 已提交
2571 2572
	if (!tun) {
		ret = -EBADFD;
J
Jason Wang 已提交
2573
		goto out_free;
W
Wei Xu 已提交
2574
	}
J
Jason Wang 已提交
2575

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

out_put_tun:
	tun_put(tun);
J
Jason Wang 已提交
2596 2597
out_free:
	tun_ptr_free(ptr);
W
Wei Xu 已提交
2598
	return ret;
2599 2600
}

J
Jason Wang 已提交
2601 2602 2603
static int tun_ptr_peek_len(void *ptr)
{
	if (likely(ptr)) {
2604 2605
		if (tun_is_xdp_frame(ptr)) {
			struct xdp_frame *xdpf = tun_ptr_to_xdp(ptr);
J
Jason Wang 已提交
2606

2607
			return xdpf->len;
J
Jason Wang 已提交
2608 2609 2610 2611 2612 2613 2614
		}
		return __skb_array_len_with_tag(ptr);
	} else {
		return 0;
	}
}

J
Jason Wang 已提交
2615 2616 2617 2618 2619 2620
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;

2621
	tun = tun_get(tfile);
J
Jason Wang 已提交
2622 2623 2624
	if (!tun)
		return 0;

J
Jason Wang 已提交
2625
	ret = PTR_RING_PEEK_CALL(&tfile->tx_ring, tun_ptr_peek_len);
J
Jason Wang 已提交
2626 2627 2628 2629 2630
	tun_put(tun);

	return ret;
}

2631 2632
/* Ops structure to mimic raw sockets with tun */
static const struct proto_ops tun_socket_ops = {
J
Jason Wang 已提交
2633
	.peek_len = tun_peek_len,
2634 2635 2636 2637
	.sendmsg = tun_sendmsg,
	.recvmsg = tun_recvmsg,
};

2638 2639 2640
static struct proto tun_proto = {
	.name		= "tun",
	.owner		= THIS_MODULE,
J
Jason Wang 已提交
2641
	.obj_size	= sizeof(struct tun_file),
2642
};
2643

2644 2645
static int tun_flags(struct tun_struct *tun)
{
2646
	return tun->flags & (TUN_FEATURES | IFF_PERSIST | IFF_TUN | IFF_TAP);
2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657 2658 2659
}

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));
2660 2661 2662 2663
	return uid_valid(tun->owner)?
		sprintf(buf, "%u\n",
			from_kuid_munged(current_user_ns(), tun->owner)):
		sprintf(buf, "-1\n");
2664 2665 2666 2667 2668 2669
}

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));
2670 2671 2672 2673
	return gid_valid(tun->group) ?
		sprintf(buf, "%u\n",
			from_kgid_munged(current_user_ns(), tun->group)):
		sprintf(buf, "-1\n");
2674 2675 2676 2677 2678 2679
}

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

2680 2681 2682 2683 2684 2685 2686 2687 2688 2689 2690
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
};

2691
static int tun_set_iff(struct net *net, struct file *file, struct ifreq *ifr)
L
Linus Torvalds 已提交
2692 2693
{
	struct tun_struct *tun;
J
Jason Wang 已提交
2694
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
2695 2696 2697
	struct net_device *dev;
	int err;

2698 2699 2700
	if (tfile->detached)
		return -EINVAL;

2701 2702 2703 2704 2705 2706 2707 2708 2709
	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;
	}

2710 2711
	dev = __dev_get_by_name(net, ifr->ifr_name);
	if (dev) {
2712 2713
		if (ifr->ifr_flags & IFF_TUN_EXCL)
			return -EBUSY;
2714 2715 2716 2717 2718 2719 2720
		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;

2721
		if (!!(ifr->ifr_flags & IFF_MULTI_QUEUE) !=
M
Michael S. Tsirkin 已提交
2722
		    !!(tun->flags & IFF_MULTI_QUEUE))
2723 2724
			return -EINVAL;

2725
		if (tun_not_capable(tun))
P
Paul Moore 已提交
2726
			return -EPERM;
2727
		err = security_tun_dev_open(tun->security);
P
Paul Moore 已提交
2728 2729 2730
		if (err < 0)
			return err;

2731
		err = tun_attach(tun, file, ifr->ifr_flags & IFF_NOFILTER,
E
Eric Dumazet 已提交
2732 2733
				 ifr->ifr_flags & IFF_NAPI,
				 ifr->ifr_flags & IFF_NAPI_FRAGS);
2734 2735
		if (err < 0)
			return err;
J
Jason Wang 已提交
2736

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

		tun->flags = (tun->flags & ~TUN_FEATURES) |
			      (ifr->ifr_flags & TUN_FEATURES);
2748 2749 2750

		netdev_state_change(dev);
	} else {
L
Linus Torvalds 已提交
2751 2752
		char *name;
		unsigned long flags = 0;
2753 2754
		int queues = ifr->ifr_flags & IFF_MULTI_QUEUE ?
			     MAX_TAP_QUEUES : 1;
L
Linus Torvalds 已提交
2755

2756
		if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
2757
			return -EPERM;
P
Paul Moore 已提交
2758 2759 2760
		err = security_tun_dev_create();
		if (err < 0)
			return err;
2761

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

L
Linus Torvalds 已提交
2774 2775 2776
		if (*ifr->ifr_name)
			name = ifr->ifr_name;

J
Jason Wang 已提交
2777
		dev = alloc_netdev_mqs(sizeof(struct tun_struct), name,
2778 2779
				       NET_NAME_UNKNOWN, tun_setup, queues,
				       queues);
2780

L
Linus Torvalds 已提交
2781 2782
		if (!dev)
			return -ENOMEM;
2783
		err = dev_get_valid_name(net, dev, name);
2784
		if (err < 0)
2785
			goto err_free_dev;
L
Linus Torvalds 已提交
2786

2787
		dev_net_set(dev, net);
2788
		dev->rtnl_link_ops = &tun_link_ops;
2789
		dev->ifindex = tfile->ifindex;
2790
		dev->sysfs_groups[0] = &tun_attr_group;
S
Stephen Hemminger 已提交
2791

L
Linus Torvalds 已提交
2792 2793 2794
		tun = netdev_priv(dev);
		tun->dev = dev;
		tun->flags = flags;
2795
		tun->txflt.count = 0;
2796
		tun->vnet_hdr_sz = sizeof(struct virtio_net_hdr);
2797

2798
		tun->align = NET_SKB_PAD;
J
Jason Wang 已提交
2799 2800
		tun->filter_attached = false;
		tun->sndbuf = tfile->socket.sk->sk_sndbuf;
J
Jason Wang 已提交
2801
		tun->rx_batched = 0;
2802
		RCU_INIT_POINTER(tun->steering_prog, NULL);
2803

2804 2805 2806 2807 2808 2809
		tun->pcpu_stats = netdev_alloc_pcpu_stats(struct tun_pcpu_stats);
		if (!tun->pcpu_stats) {
			err = -ENOMEM;
			goto err_free_dev;
		}

J
Jason Wang 已提交
2810 2811
		spin_lock_init(&tun->lock);

2812 2813
		err = security_tun_dev_alloc_security(&tun->security);
		if (err < 0)
2814
			goto err_free_stat;
P
Paul Moore 已提交
2815

L
Linus Torvalds 已提交
2816
		tun_net_init(dev);
2817
		tun_flow_init(tun);
J
Jason Wang 已提交
2818

2819
		dev->hw_features = NETIF_F_SG | NETIF_F_FRAGLIST |
J
Jason Wang 已提交
2820 2821
				   TUN_USER_FEATURES | NETIF_F_HW_VLAN_CTAG_TX |
				   NETIF_F_HW_VLAN_STAG_TX;
2822
		dev->features = dev->hw_features | NETIF_F_LLTX;
2823 2824 2825
		dev->vlan_features = dev->features &
				     ~(NETIF_F_HW_VLAN_CTAG_TX |
				       NETIF_F_HW_VLAN_STAG_TX);
2826

2827 2828 2829
		tun->flags = (tun->flags & ~TUN_FEATURES) |
			      (ifr->ifr_flags & TUN_FEATURES);

J
Jason Wang 已提交
2830
		INIT_LIST_HEAD(&tun->disabled);
E
Eric Dumazet 已提交
2831 2832
		err = tun_attach(tun, file, false, ifr->ifr_flags & IFF_NAPI,
				 ifr->ifr_flags & IFF_NAPI_FRAGS);
2833
		if (err < 0)
2834
			goto err_free_flow;
2835

L
Linus Torvalds 已提交
2836 2837
		err = register_netdevice(tun->dev);
		if (err < 0)
2838
			goto err_detach;
L
Linus Torvalds 已提交
2839 2840
	}

2841 2842
	netif_carrier_on(tun->dev);

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

2845 2846 2847 2848
	/* Make sure persistent devices do not get stuck in
	 * xoff state.
	 */
	if (netif_running(tun->dev))
J
Jason Wang 已提交
2849
		netif_tx_wake_all_queues(tun->dev);
2850

L
Linus Torvalds 已提交
2851 2852 2853
	strcpy(ifr->ifr_name, tun->dev->name);
	return 0;

2854 2855
err_detach:
	tun_detach_all(dev);
2856 2857 2858
	/* register_netdevice() already called tun_free_netdev() */
	goto err_free_dev;

2859 2860 2861
err_free_flow:
	tun_flow_uninit(tun);
	security_tun_dev_free_security(tun->security);
2862 2863
err_free_stat:
	free_percpu(tun->pcpu_stats);
2864
err_free_dev:
L
Linus Torvalds 已提交
2865 2866 2867 2868
	free_netdev(dev);
	return err;
}

R
Rami Rosen 已提交
2869
static void tun_get_iff(struct net *net, struct tun_struct *tun,
2870
		       struct ifreq *ifr)
2871
{
2872
	tun_debug(KERN_INFO, tun, "tun_get_iff\n");
2873 2874 2875

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

2876
	ifr->ifr_flags = tun_flags(tun);
2877 2878 2879

}

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

	if (arg & TUN_F_CSUM) {
2887
		features |= NETIF_F_HW_CSUM;
2888 2889 2890 2891 2892 2893 2894 2895 2896 2897 2898 2899 2900
		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);
		}
2901 2902

		arg &= ~TUN_F_UFO;
2903 2904 2905 2906 2907 2908 2909
	}

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

2910
	tun->set_features = features;
2911 2912
	tun->dev->wanted_features &= ~TUN_USER_FEATURES;
	tun->dev->wanted_features |= features;
2913
	netdev_update_features(tun->dev);
2914 2915 2916 2917

	return 0;
}

J
Jason Wang 已提交
2918 2919 2920 2921 2922 2923
static void tun_detach_filter(struct tun_struct *tun, int n)
{
	int i;
	struct tun_file *tfile;

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

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

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

2992 2993 2994
	if (ret >= 0)
		netdev_state_change(tun->dev);

2995
unlock:
2996 2997 2998 2999
	rtnl_unlock();
	return ret;
}

3000 3001
static int tun_set_ebpf(struct tun_struct *tun, struct tun_prog **prog_p,
			void __user *data)
3002 3003 3004 3005 3006 3007 3008 3009 3010 3011 3012 3013 3014 3015 3016
{
	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);
	}

3017
	return __tun_set_ebpf(tun, prog_p, prog);
3018 3019
}

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

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

J
Jason Wang 已提交
3059
	ret = 0;
3060 3061
	rtnl_lock();

3062
	tun = tun_get(tfile);
3063 3064 3065 3066 3067
	if (cmd == TUNSETIFF) {
		ret = -EEXIST;
		if (tun)
			goto unlock;

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

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

3072 3073
		if (ret)
			goto unlock;
L
Linus Torvalds 已提交
3074

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

3093
	ret = -EBADFD;
L
Linus Torvalds 已提交
3094
	if (!tun)
3095
		goto unlock;
L
Linus Torvalds 已提交
3096

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

E
Eric W. Biederman 已提交
3099
	ret = 0;
L
Linus Torvalds 已提交
3100
	switch (cmd) {
3101
	case TUNGETIFF:
R
Rami Rosen 已提交
3102
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
3103

3104 3105
		if (tfile->detached)
			ifr.ifr_flags |= IFF_DETACH_QUEUE;
3106 3107
		if (!tfile->socket.sk->sk_filter)
			ifr.ifr_flags |= IFF_NOFILTER;
3108

3109
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
3110
			ret = -EFAULT;
3111 3112
		break;

L
Linus Torvalds 已提交
3113 3114 3115
	case TUNSETNOCSUM:
		/* Disable/Enable checksum */

3116 3117
		/* [unimplemented] */
		tun_debug(KERN_INFO, tun, "ignored: set checksum %s\n",
3118
			  arg ? "disabled" : "enabled");
L
Linus Torvalds 已提交
3119 3120 3121
		break;

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

3136 3137
		tun_debug(KERN_INFO, tun, "persist %s\n",
			  arg ? "enabled" : "disabled");
L
Linus Torvalds 已提交
3138 3139 3140 3141
		break;

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

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

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

L
Linus Torvalds 已提交
3180 3181 3182 3183 3184
#ifdef TUN_DEBUG
	case TUNSETDEBUG:
		tun->debug = arg;
		break;
#endif
3185
	case TUNSETOFFLOAD:
3186
		ret = set_offload(tun, arg);
E
Eric W. Biederman 已提交
3187
		break;
3188

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

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

	case SIOCSIFHWADDR:
3206
		/* Set hw address */
3207 3208
		tun_debug(KERN_DEBUG, tun, "set hw address: %pM\n",
			  ifr.ifr_hwaddr.sa_data);
3209

3210
		ret = dev_set_mac_address(tun->dev, &ifr.ifr_hwaddr, NULL);
E
Eric W. Biederman 已提交
3211
		break;
3212 3213

	case TUNGETSNDBUF:
J
Jason Wang 已提交
3214
		sndbuf = tfile->socket.sk->sk_sndbuf;
3215 3216 3217 3218 3219 3220 3221 3222 3223
		if (copy_to_user(argp, &sndbuf, sizeof(sndbuf)))
			ret = -EFAULT;
		break;

	case TUNSETSNDBUF:
		if (copy_from_user(&sndbuf, argp, sizeof(sndbuf))) {
			ret = -EFAULT;
			break;
		}
3224 3225 3226 3227
		if (sndbuf <= 0) {
			ret = -EINVAL;
			break;
		}
3228

J
Jason Wang 已提交
3229 3230
		tun->sndbuf = sndbuf;
		tun_set_sndbuf(tun);
3231 3232
		break;

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

3269 3270 3271 3272 3273 3274 3275 3276
	case TUNGETVNETBE:
		ret = tun_get_vnet_be(tun, argp);
		break;

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

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

J
Jason Wang 已提交
3286
		ret = tun_attach_filter(tun);
M
Michael S. Tsirkin 已提交
3287 3288 3289 3290 3291
		break;

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

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

3308
	case TUNSETSTEERINGEBPF:
3309
		ret = tun_set_ebpf(tun, &tun->steering_prog, argp);
3310 3311
		break;

3312 3313 3314 3315
	case TUNSETFILTEREBPF:
		ret = tun_set_ebpf(tun, &tun->filter_prog, argp);
		break;

N
Nicolas Dichtel 已提交
3316 3317 3318 3319 3320 3321 3322 3323
	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 已提交
3324
	default:
E
Eric W. Biederman 已提交
3325 3326
		ret = -EINVAL;
		break;
3327
	}
L
Linus Torvalds 已提交
3328

3329 3330 3331
	if (do_notify)
		netdev_state_change(tun->dev);

3332 3333 3334 3335
unlock:
	rtnl_unlock();
	if (tun)
		tun_put(tun);
E
Eric W. Biederman 已提交
3336
	return ret;
L
Linus Torvalds 已提交
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 3369 3370 3371 3372 3373
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 已提交
3374 3375
static int tun_chr_fasync(int fd, struct file *file, int on)
{
J
Jason Wang 已提交
3376
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
3377 3378
	int ret;

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

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

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

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

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

3408
	mutex_init(&tfile->napi_mutex);
3409
	RCU_INIT_POINTER(tfile->tun, NULL);
J
Jason Wang 已提交
3410
	tfile->flags = 0;
3411
	tfile->ifindex = 0;
J
Jason Wang 已提交
3412 3413

	init_waitqueue_head(&tfile->wq.wait);
3414
	RCU_INIT_POINTER(tfile->socket.wq, &tfile->wq);
J
Jason Wang 已提交
3415 3416 3417 3418 3419 3420 3421 3422 3423

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

3427 3428
	sock_set_flag(&tfile->sk, SOCK_ZEROCOPY);

L
Linus Torvalds 已提交
3429 3430 3431 3432 3433
	return 0;
}

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

J
Jason Wang 已提交
3436
	tun_detach(tfile, true);
L
Linus Torvalds 已提交
3437 3438 3439 3440

	return 0;
}

3441
#ifdef CONFIG_PROC_FS
3442
static void tun_chr_show_fdinfo(struct seq_file *m, struct file *file)
3443
{
3444
	struct tun_file *tfile = file->private_data;
3445 3446 3447 3448 3449 3450
	struct tun_struct *tun;
	struct ifreq ifr;

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

	rtnl_lock();
3451
	tun = tun_get(tfile);
3452 3453 3454 3455 3456 3457 3458
	if (tun)
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
	rtnl_unlock();

	if (tun)
		tun_put(tun);

3459
	seq_printf(m, "iff:\t%s\n", ifr.ifr_name);
3460 3461 3462
}
#endif

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

static struct miscdevice tun_miscdev = {
	.minor = TUN_MINOR,
	.name = "tun",
3484
	.nodename = "net/tun",
L
Linus Torvalds 已提交
3485 3486 3487 3488 3489
	.fops = &tun_fops,
};

/* ethtool interface */

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

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 已提交
3517 3518 3519 3520 3521 3522 3523
	return 0;
}

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

3524 3525
	strlcpy(info->driver, DRV_NAME, sizeof(info->driver));
	strlcpy(info->version, DRV_VERSION, sizeof(info->version));
L
Linus Torvalds 已提交
3526 3527

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

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

J
Jason Wang 已提交
3590 3591 3592 3593
static int tun_queue_resize(struct tun_struct *tun)
{
	struct net_device *dev = tun->dev;
	struct tun_file *tfile;
3594
	struct ptr_ring **rings;
J
Jason Wang 已提交
3595 3596 3597
	int n = tun->numqueues + tun->numdisabled;
	int ret, i;

3598 3599
	rings = kmalloc_array(n, sizeof(*rings), GFP_KERNEL);
	if (!rings)
J
Jason Wang 已提交
3600 3601 3602 3603
		return -ENOMEM;

	for (i = 0; i < tun->numqueues; i++) {
		tfile = rtnl_dereference(tun->tfiles[i]);
3604
		rings[i] = &tfile->tx_ring;
J
Jason Wang 已提交
3605 3606
	}
	list_for_each_entry(tfile, &tun->disabled, next)
3607
		rings[i++] = &tfile->tx_ring;
J
Jason Wang 已提交
3608

3609 3610
	ret = ptr_ring_resize_multiple(rings, n,
				       dev->tx_queue_len, GFP_KERNEL,
J
Jason Wang 已提交
3611
				       tun_ptr_free);
J
Jason Wang 已提交
3612

3613
	kfree(rings);
J
Jason Wang 已提交
3614 3615 3616 3617 3618 3619 3620 3621 3622
	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);

3623 3624 3625
	if (dev->rtnl_link_ops != &tun_link_ops)
		return NOTIFY_DONE;

J
Jason Wang 已提交
3626 3627 3628 3629 3630 3631 3632 3633 3634 3635 3636 3637 3638 3639 3640
	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,
};
3641

L
Linus Torvalds 已提交
3642 3643 3644 3645
static int __init tun_init(void)
{
	int ret = 0;

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

3648
	ret = rtnl_link_register(&tun_link_ops);
3649
	if (ret) {
3650
		pr_err("Can't register link_ops\n");
3651
		goto err_linkops;
3652 3653
	}

L
Linus Torvalds 已提交
3654
	ret = misc_register(&tun_miscdev);
3655
	if (ret) {
3656
		pr_err("Can't register misc device %d\n", TUN_MINOR);
3657 3658
		goto err_misc;
	}
J
Jason Wang 已提交
3659

3660 3661 3662 3663 3664 3665
	ret = register_netdevice_notifier(&tun_notifier_block);
	if (ret) {
		pr_err("Can't register netdevice notifier\n");
		goto err_notifier;
	}

3666
	return  0;
3667 3668 3669

err_notifier:
	misc_deregister(&tun_miscdev);
3670
err_misc:
3671 3672
	rtnl_link_unregister(&tun_link_ops);
err_linkops:
L
Linus Torvalds 已提交
3673 3674 3675 3676 3677
	return ret;
}

static void tun_cleanup(void)
{
3678
	misc_deregister(&tun_miscdev);
3679
	rtnl_link_unregister(&tun_link_ops);
J
Jason Wang 已提交
3680
	unregister_netdevice_notifier(&tun_notifier_block);
L
Linus Torvalds 已提交
3681 3682
}

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

3699
struct ptr_ring *tun_get_tx_ring(struct file *file)
J
Jason Wang 已提交
3700 3701 3702 3703 3704 3705 3706 3707
{
	struct tun_file *tfile;

	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
	tfile = file->private_data;
	if (!tfile)
		return ERR_PTR(-EBADFD);
3708
	return &tfile->tx_ring;
J
Jason Wang 已提交
3709
}
3710
EXPORT_SYMBOL_GPL(tun_get_tx_ring);
J
Jason Wang 已提交
3711

L
Linus Torvalds 已提交
3712 3713 3714 3715 3716 3717
module_init(tun_init);
module_exit(tun_cleanup);
MODULE_DESCRIPTION(DRV_DESCRIPTION);
MODULE_AUTHOR(DRV_COPYRIGHT);
MODULE_LICENSE("GPL");
MODULE_ALIAS_MISCDEV(TUN_MINOR);
3718
MODULE_ALIAS("devname:net/tun");