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

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

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

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

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

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

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

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

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

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

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

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

128 129
#define GOODCOPY_LEN 128

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

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

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

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

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

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

#define TUN_NUM_FLOW_ENTRIES 1024

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

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

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

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

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

J
Jason Wang 已提交
251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268
bool tun_is_xdp_buff(void *ptr)
{
	return (unsigned long)ptr & TUN_XDP_FLAG;
}
EXPORT_SYMBOL(tun_is_xdp_buff);

void *tun_xdp_to_ptr(void *ptr)
{
	return (void *)((unsigned long)ptr | TUN_XDP_FLAG);
}
EXPORT_SYMBOL(tun_xdp_to_ptr);

void *tun_ptr_to_xdp(void *ptr)
{
	return (void *)((unsigned long)ptr & ~TUN_XDP_FLAG);
}
EXPORT_SYMBOL(tun_ptr_to_xdp);

269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311
static int tun_napi_receive(struct napi_struct *napi, int budget)
{
	struct tun_file *tfile = container_of(napi, struct tun_file, napi);
	struct sk_buff_head *queue = &tfile->sk.sk_write_queue;
	struct sk_buff_head process_queue;
	struct sk_buff *skb;
	int received = 0;

	__skb_queue_head_init(&process_queue);

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

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

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

	return received;
}

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

	received = tun_napi_receive(napi, budget);

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

	return received;
}

static void tun_napi_init(struct tun_struct *tun, struct tun_file *tfile,
			  bool napi_en)
{
312
	tfile->napi_enabled = napi_en;
313 314 315 316
	if (napi_en) {
		netif_napi_add(tun->dev, &tfile->napi, tun_napi_poll,
			       NAPI_POLL_WEIGHT);
		napi_enable(&tfile->napi);
317
		mutex_init(&tfile->napi_mutex);
318 319 320 321 322
	}
}

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

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

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

338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385
#ifdef CONFIG_TUN_VNET_CROSS_LE
static inline bool tun_legacy_is_little_endian(struct tun_struct *tun)
{
	return tun->flags & TUN_VNET_BE ? false :
		virtio_legacy_is_little_endian();
}

static long tun_get_vnet_be(struct tun_struct *tun, int __user *argp)
{
	int be = !!(tun->flags & TUN_VNET_BE);

	if (put_user(be, argp))
		return -EFAULT;

	return 0;
}

static long tun_set_vnet_be(struct tun_struct *tun, int __user *argp)
{
	int be;

	if (get_user(be, argp))
		return -EFAULT;

	if (be)
		tun->flags |= TUN_VNET_BE;
	else
		tun->flags &= ~TUN_VNET_BE;

	return 0;
}
#else
static inline bool tun_legacy_is_little_endian(struct tun_struct *tun)
{
	return virtio_legacy_is_little_endian();
}

static long tun_get_vnet_be(struct tun_struct *tun, int __user *argp)
{
	return -EINVAL;
}

static long tun_set_vnet_be(struct tun_struct *tun, int __user *argp)
{
	return -EINVAL;
}
#endif /* CONFIG_TUN_VNET_CROSS_LE */

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

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

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

J
Jason Wang 已提交
402 403 404 405 406 407 408 409 410
static inline u32 tun_hashfn(u32 rxhash)
{
	return rxhash & 0x3ff;
}

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

411
	hlist_for_each_entry_rcu(e, head, hash_link) {
J
Jason Wang 已提交
412 413 414 415 416 417 418 419 420 421
		if (e->rxhash == rxhash)
			return e;
	}
	return NULL;
}

static struct tun_flow_entry *tun_flow_create(struct tun_struct *tun,
					      struct hlist_head *head,
					      u32 rxhash, u16 queue_index)
{
422 423
	struct tun_flow_entry *e = kmalloc(sizeof(*e), GFP_ATOMIC);

J
Jason Wang 已提交
424 425 426 427 428
	if (e) {
		tun_debug(KERN_INFO, tun, "create flow: hash %u index %u\n",
			  rxhash, queue_index);
		e->updated = jiffies;
		e->rxhash = rxhash;
429
		e->rps_rxhash = 0;
J
Jason Wang 已提交
430 431 432
		e->queue_index = queue_index;
		e->tun = tun;
		hlist_add_head_rcu(&e->hash_link, head);
433
		++tun->flow_count;
J
Jason Wang 已提交
434 435 436 437 438 439 440 441 442
	}
	return e;
}

static void tun_flow_delete(struct tun_struct *tun, struct tun_flow_entry *e)
{
	tun_debug(KERN_INFO, tun, "delete flow: hash %u index %u\n",
		  e->rxhash, e->queue_index);
	hlist_del_rcu(&e->hash_link);
443
	kfree_rcu(e, rcu);
444
	--tun->flow_count;
J
Jason Wang 已提交
445 446 447 448 449 450 451 452 453
}

static void tun_flow_flush(struct tun_struct *tun)
{
	int i;

	spin_lock_bh(&tun->lock);
	for (i = 0; i < TUN_NUM_FLOW_ENTRIES; i++) {
		struct tun_flow_entry *e;
454
		struct hlist_node *n;
J
Jason Wang 已提交
455

456
		hlist_for_each_entry_safe(e, n, &tun->flows[i], hash_link)
J
Jason Wang 已提交
457 458 459 460 461 462 463 464 465 466 467 468
			tun_flow_delete(tun, e);
	}
	spin_unlock_bh(&tun->lock);
}

static void tun_flow_delete_by_queue(struct tun_struct *tun, u16 queue_index)
{
	int i;

	spin_lock_bh(&tun->lock);
	for (i = 0; i < TUN_NUM_FLOW_ENTRIES; i++) {
		struct tun_flow_entry *e;
469
		struct hlist_node *n;
J
Jason Wang 已提交
470

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

479
static void tun_flow_cleanup(struct timer_list *t)
J
Jason Wang 已提交
480
{
481
	struct tun_struct *tun = from_timer(tun, t, flow_gc_timer);
J
Jason Wang 已提交
482 483 484 485 486 487 488
	unsigned long delay = tun->ageing_time;
	unsigned long next_timer = jiffies + delay;
	unsigned long count = 0;
	int i;

	tun_debug(KERN_INFO, tun, "tun_flow_cleanup\n");

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

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

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

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

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

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

	rcu_read_lock();

528 529 530
	/* We may get a very small possibility of OOO during switching, not
	 * worth to optimize.*/
	if (tun->numqueues == 1 || tfile->detached)
J
Jason Wang 已提交
531 532 533 534 535 536 537
		goto unlock;

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

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

unlock:
	rcu_read_unlock();
}

555 556 557 558 559 560
/**
 * Save the hash received in the stack receive path and update the
 * flow_hash table accordingly.
 */
static inline void tun_flow_save_rps_rxhash(struct tun_flow_entry *e, u32 hash)
{
561
	if (unlikely(e->rps_rxhash != hash))
562 563 564
		e->rps_rxhash = hash;
}

J
Jason Wang 已提交
565
/* We try to identify a flow through its rxhash first. The reason that
S
stephen hemminger 已提交
566
 * we do not check rxq no. is because some cards(e.g 82599), chooses
J
Jason Wang 已提交
567 568 569 570 571
 * the rxq based on the txq where the last packet of the flow comes. As
 * the userspace application move between processors, we may get a
 * different rxq no. here. If we could not get rxhash, then we would
 * hope the rxq no. may help here.
 */
572
static u16 tun_automq_select_queue(struct tun_struct *tun, struct sk_buff *skb)
J
Jason Wang 已提交
573
{
J
Jason Wang 已提交
574
	struct tun_flow_entry *e;
J
Jason Wang 已提交
575 576 577
	u32 txq = 0;
	u32 numqueues = 0;

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

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

	return txq;
}

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

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

	return ret % tun->numqueues;
}

static u16 tun_select_queue(struct net_device *dev, struct sk_buff *skb,
			    void *accel_priv, select_queue_fallback_t fallback)
{
	struct tun_struct *tun = netdev_priv(dev);
	u16 ret;

	rcu_read_lock();
	if (rcu_dereference(tun->steering_prog))
		ret = tun_ebpf_select_queue(tun, skb);
	else
		ret = tun_automq_select_queue(tun, skb);
	rcu_read_unlock();

	return ret;
}

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

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

J
Jason Wang 已提交
636 637 638 639 640 641
static void tun_set_real_num_queues(struct tun_struct *tun)
{
	netif_set_real_num_tx_queues(tun->dev, tun->numqueues);
	netif_set_real_num_rx_queues(tun->dev, tun->numqueues);
}

J
Jason Wang 已提交
642 643 644 645 646 647 648
static void tun_disable_queue(struct tun_struct *tun, struct tun_file *tfile)
{
	tfile->detached = tun;
	list_add_tail(&tfile->next, &tun->disabled);
	++tun->numdisabled;
}

J
Jason Wang 已提交
649
static struct tun_struct *tun_enable_queue(struct tun_file *tfile)
J
Jason Wang 已提交
650 651 652 653 654 655 656 657 658
{
	struct tun_struct *tun = tfile->detached;

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

J
Jason Wang 已提交
659 660 661 662 663 664 665 666 667 668 669 670 671
static void tun_ptr_free(void *ptr)
{
	if (!ptr)
		return;
	if (tun_is_xdp_buff(ptr)) {
		struct xdp_buff *xdp = tun_ptr_to_xdp(ptr);

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

	err = 0;
J
Jason Wang 已提交
818

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

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

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

	if (tfile->detached) {
		/* Re-attach detached tfile, updating XDP queue_index */
		WARN_ON(!xdp_rxq_info_is_reg(&tfile->xdp_rxq));

		if (tfile->xdp_rxq.queue_index    != tfile->queue_index)
			tfile->xdp_rxq.queue_index = tfile->queue_index;
	} else {
		/* Setup XDP RX-queue info, for new tfile getting attached */
		err = xdp_rxq_info_reg(&tfile->xdp_rxq,
				       tun->dev, tfile->queue_index);
		if (err < 0)
			goto out;
		err = 0;
	}

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

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

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

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

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

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

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

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

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

891
/* TAP filtering */
892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919
static void addr_hash_set(u32 *mask, const u8 *addr)
{
	int n = ether_crc(ETH_ALEN, addr) >> 26;
	mask[n >> 5] |= (1 << (n & 31));
}

static unsigned int addr_hash_test(const u32 *mask, const u8 *addr)
{
	int n = ether_crc(ETH_ALEN, addr) >> 26;
	return mask[n >> 5] & (1 << (n & 31));
}

static int update_filter(struct tap_filter *filter, void __user *arg)
{
	struct { u8 u[ETH_ALEN]; } *addr;
	struct tun_filter uf;
	int err, alen, n, nexact;

	if (copy_from_user(&uf, arg, sizeof(uf)))
		return -EFAULT;

	if (!uf.count) {
		/* Disabled */
		filter->count = 0;
		return 0;
	}

	alen = ETH_ALEN * uf.count;
920 921 922
	addr = memdup_user(arg + sizeof(uf), alen);
	if (IS_ERR(addr))
		return PTR_ERR(addr);
923 924 925 926 927 928 929 930 931 932 933 934 935

	/* The filter is updated without holding any locks. Which is
	 * perfectly safe. We disable it first and in the worst
	 * case we'll accept a few undesired packets. */
	filter->count = 0;
	wmb();

	/* Use first set of addresses as an exact filter */
	for (n = 0; n < uf.count && n < FLT_EXACT_COUNT; n++)
		memcpy(filter->addr[n], addr[n].u, ETH_ALEN);

	nexact = n;

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

	/* For ALLMULTI just set the mask to all ones.
	 * This overrides the mask populated above. */
	if ((uf.flags & TUN_FLT_ALLMULTI))
		memset(filter->mask, ~0, sizeof(filter->mask));

	/* Now enable the filter */
	wmb();
	filter->count = nexact;

	/* Return the number of exact filters */
	err = nexact;
958
free_addr:
959 960 961 962 963 964 965 966 967 968 969 970 971 972
	kfree(addr);
	return err;
}

/* Returns: 0 - drop, !=0 - accept */
static int run_filter(struct tap_filter *filter, const struct sk_buff *skb)
{
	/* Cannot use eth_hdr(skb) here because skb_mac_hdr() is incorrect
	 * at this point. */
	struct ethhdr *eh = (struct ethhdr *) skb->data;
	int i;

	/* Exact match */
	for (i = 0; i < filter->count; i++)
973
		if (ether_addr_equal(eh->h_dest, filter->addr[i]))
974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994
			return 1;

	/* Inexact match (multicast only) */
	if (is_multicast_ether_addr(eh->h_dest))
		return addr_hash_test(filter->mask, eh->h_dest);

	return 0;
}

/*
 * Checks whether the packet is accepted or not.
 * Returns: 0 - drop, !=0 - accept
 */
static int check_filter(struct tap_filter *filter, const struct sk_buff *skb)
{
	if (!filter->count)
		return 1;

	return run_filter(filter, skb);
}

L
Linus Torvalds 已提交
995 996
/* Network device part of the driver */

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

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

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

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

	for (i = 0; i < tun->numqueues; i++) {
		struct tun_file *tfile;

		tfile = rtnl_dereference(tun->tfiles[i]);
		tfile->socket.sk->sk_write_space(tfile->socket.sk);
	}

L
Linus Torvalds 已提交
1020 1021 1022 1023 1024 1025
	return 0;
}

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

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

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

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

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

	return len;
}

1064 1065 1066 1067 1068 1069
/* Net device start xmit */
static netdev_tx_t tun_net_xmit(struct sk_buff *skb, struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
	int txq = skb->queue_mapping;
	struct tun_file *tfile;
1070
	int len = skb->len;
1071 1072 1073 1074 1075

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

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

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

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

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

1086 1087 1088 1089 1090 1091
	/* Drop if the filter does not like it.
	 * This is a noop if the filter is disabled.
	 * Filter can be enabled only for the TAP devices. */
	if (!check_filter(&tun->txflt, skb))
		goto drop;

J
Jason Wang 已提交
1092 1093
	if (tfile->socket.sk->sk_filter &&
	    sk_filter(tfile->socket.sk, skb))
M
Michael S. Tsirkin 已提交
1094 1095
		goto drop;

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

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

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

1108
	skb_tx_timestamp(skb);
1109

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

1115 1116
	nf_reset(skb);

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

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

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

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

1136
static void tun_net_mclist(struct net_device *dev)
L
Linus Torvalds 已提交
1137
{
1138 1139 1140 1141 1142
	/*
	 * This callback is supposed to deal with mc filter in
	 * _rx_ path and has nothing to do with the _tx_ path.
	 * In rx path we always accept everything userspace gives us.
	 */
L
Linus Torvalds 已提交
1143 1144
}

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

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

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

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

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

static void tun_set_headroom(struct net_device *dev, int new_hr)
{
	struct tun_struct *tun = netdev_priv(dev);

	if (new_hr < NET_SKB_PAD)
		new_hr = NET_SKB_PAD;

	tun->align = new_hr;
}

1200
static void
1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235
tun_net_get_stats64(struct net_device *dev, struct rtnl_link_stats64 *stats)
{
	u32 rx_dropped = 0, tx_dropped = 0, rx_frame_errors = 0;
	struct tun_struct *tun = netdev_priv(dev);
	struct tun_pcpu_stats *p;
	int i;

	for_each_possible_cpu(i) {
		u64 rxpackets, rxbytes, txpackets, txbytes;
		unsigned int start;

		p = per_cpu_ptr(tun->pcpu_stats, i);
		do {
			start = u64_stats_fetch_begin(&p->syncp);
			rxpackets	= p->rx_packets;
			rxbytes		= p->rx_bytes;
			txpackets	= p->tx_packets;
			txbytes		= p->tx_bytes;
		} while (u64_stats_fetch_retry(&p->syncp, start));

		stats->rx_packets	+= rxpackets;
		stats->rx_bytes		+= rxbytes;
		stats->tx_packets	+= txpackets;
		stats->tx_bytes		+= txbytes;

		/* u32 counters */
		rx_dropped	+= p->rx_dropped;
		rx_frame_errors	+= p->rx_frame_errors;
		tx_dropped	+= p->tx_dropped;
	}
	stats->rx_dropped  = rx_dropped;
	stats->rx_frame_errors = rx_frame_errors;
	stats->tx_dropped = tx_dropped;
}

J
Jason Wang 已提交
1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261
static int tun_xdp_set(struct net_device *dev, struct bpf_prog *prog,
		       struct netlink_ext_ack *extack)
{
	struct tun_struct *tun = netdev_priv(dev);
	struct bpf_prog *old_prog;

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

	return 0;
}

static u32 tun_xdp_query(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
	const struct bpf_prog *xdp_prog;

	xdp_prog = rtnl_dereference(tun->xdp_prog);
	if (xdp_prog)
		return xdp_prog->aux->id;

	return 0;
}

1262
static int tun_xdp(struct net_device *dev, struct netdev_bpf *xdp)
J
Jason Wang 已提交
1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275
{
	switch (xdp->command) {
	case XDP_SETUP_PROG:
		return tun_xdp_set(dev, xdp->prog, xdp->extack);
	case XDP_QUERY_PROG:
		xdp->prog_id = tun_xdp_query(dev);
		xdp->prog_attached = !!xdp->prog_id;
		return 0;
	default:
		return -EINVAL;
	}
}

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

J
Jason Wang 已提交
1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350
static int tun_xdp_xmit(struct net_device *dev, struct xdp_buff *xdp)
{
	struct tun_struct *tun = netdev_priv(dev);
	struct xdp_buff *buff = xdp->data_hard_start;
	int headroom = xdp->data - xdp->data_hard_start;
	struct tun_file *tfile;
	u32 numqueues;
	int ret = 0;

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

	*buff = *xdp;

	rcu_read_lock();

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

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

out:
	rcu_read_unlock();
	return ret;
}

static void tun_xdp_flush(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
	struct tun_file *tfile;
	u32 numqueues;

	rcu_read_lock();

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

	tfile = rcu_dereference(tun->tfiles[smp_processor_id() %
					    numqueues]);
	/* Notify and wake up reader process */
	if (tfile->flags & TUN_FASYNC)
		kill_fasync(&tfile->fasync, SIGIO, POLL_IN);
	tfile->socket.sk->sk_data_ready(tfile->socket.sk);

out:
	rcu_read_unlock();
}

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

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

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

	tun->ageing_time = TUN_FLOW_EXPIRE;
1380 1381 1382
	timer_setup(&tun->flow_gc_timer, tun_flow_cleanup, 0);
	mod_timer(&tun->flow_gc_timer,
		  round_jiffies_up(jiffies + tun->ageing_time));
J
Jason Wang 已提交
1383 1384 1385 1386 1387 1388 1389 1390
}

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

1391 1392 1393
#define MIN_MTU 68
#define MAX_MTU 65535

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

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

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

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

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

1420
		eth_hw_addr_random(dev);
1421

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

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

/* Character device part */

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

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

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

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

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

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

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

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

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

1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491
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++) {
1492
		struct page_frag *pfrag = &current->task_frag;
1493 1494 1495 1496 1497 1498 1499
		size_t fragsz = it->iov[i].iov_len;

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

1500
		if (!skb_page_frag_refill(fragsz, pfrag, GFP_KERNEL)) {
1501 1502 1503 1504
			err = -ENOMEM;
			goto free;
		}

1505 1506 1507 1508
		skb_fill_page_desc(skb, i - 1, pfrag->page,
				   pfrag->offset, fragsz);
		page_ref_inc(pfrag->page);
		pfrag->offset += fragsz;
1509 1510 1511 1512 1513 1514 1515 1516 1517
	}

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

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

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

1532
	skb = sock_alloc_send_pskb(sk, prepad + linear, len - linear, noblock,
1533
				   &err, 0);
1534
	if (!skb)
1535
		return ERR_PTR(err);
1536 1537 1538

	skb_reserve(skb, prepad);
	skb_put(skb, linear);
1539 1540
	skb->data_len = len - linear;
	skb->len += len - linear;
1541 1542 1543 1544

	return skb;
}

J
Jason Wang 已提交
1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580
static void tun_rx_batched(struct tun_struct *tun, struct tun_file *tfile,
			   struct sk_buff *skb, int more)
{
	struct sk_buff_head *queue = &tfile->sk.sk_write_queue;
	struct sk_buff_head process_queue;
	u32 rx_batched = tun->rx_batched;
	bool rcv = false;

	if (!rx_batched || (!more && skb_queue_empty(queue))) {
		local_bh_disable();
		netif_receive_skb(skb);
		local_bh_enable();
		return;
	}

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

	if (rcv) {
		struct sk_buff *nskb;

		local_bh_disable();
		while ((nskb = __skb_dequeue(&process_queue)))
			netif_receive_skb(nskb);
		netif_receive_skb(skb);
		local_bh_enable();
	}
}

1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602
static bool tun_can_build_skb(struct tun_struct *tun, struct tun_file *tfile,
			      int len, int noblock, bool zerocopy)
{
	if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
		return false;

	if (tfile->socket.sk->sk_sndbuf != INT_MAX)
		return false;

	if (!noblock)
		return false;

	if (zerocopy)
		return false;

	if (SKB_DATA_ALIGN(len + TUN_RX_PAD) +
	    SKB_DATA_ALIGN(sizeof(struct skb_shared_info)) > PAGE_SIZE)
		return false;

	return true;
}

J
Jason Wang 已提交
1603 1604
static struct sk_buff *tun_build_skb(struct tun_struct *tun,
				     struct tun_file *tfile,
1605
				     struct iov_iter *from,
J
Jason Wang 已提交
1606
				     struct virtio_net_hdr *hdr,
J
Jason Wang 已提交
1607
				     int len, int *skb_xdp)
1608
{
1609
	struct page_frag *alloc_frag = &current->task_frag;
1610
	struct sk_buff *skb;
J
Jason Wang 已提交
1611
	struct bpf_prog *xdp_prog;
1612
	int buflen = SKB_DATA_ALIGN(sizeof(struct skb_shared_info));
J
Jason Wang 已提交
1613
	unsigned int delta = 0;
1614 1615
	char *buf;
	size_t copied;
1616 1617 1618 1619 1620 1621 1622 1623
	int err, pad = TUN_RX_PAD;

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

1625
	alloc_frag->offset = ALIGN((u64)alloc_frag->offset, SMP_CACHE_BYTES);
1626 1627 1628 1629 1630
	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,
1631
				     alloc_frag->offset + pad,
1632 1633 1634 1635
				     len, from);
	if (copied != len)
		return ERR_PTR(-EFAULT);

1636 1637 1638 1639 1640
	/* There's a small window that XDP may be set after the check
	 * of xdp_prog above, this should be rare and for simplicity
	 * we do XDP on skb in case the headroom is not enough.
	 */
	if (hdr->gso_type || !xdp_prog)
J
Jason Wang 已提交
1641
		*skb_xdp = 1;
J
Jason Wang 已提交
1642
	else
J
Jason Wang 已提交
1643
		*skb_xdp = 0;
J
Jason Wang 已提交
1644

1645
	preempt_disable();
J
Jason Wang 已提交
1646 1647
	rcu_read_lock();
	xdp_prog = rcu_dereference(tun->xdp_prog);
J
Jason Wang 已提交
1648
	if (xdp_prog && !*skb_xdp) {
J
Jason Wang 已提交
1649 1650 1651 1652 1653
		struct xdp_buff xdp;
		void *orig_data;
		u32 act;

		xdp.data_hard_start = buf;
1654
		xdp.data = buf + pad;
1655
		xdp_set_data_meta_invalid(&xdp);
J
Jason Wang 已提交
1656
		xdp.data_end = xdp.data + len;
J
Jesper Dangaard Brouer 已提交
1657
		xdp.rxq = &tfile->xdp_rxq;
J
Jason Wang 已提交
1658 1659 1660 1661 1662 1663 1664 1665
		orig_data = xdp.data;
		act = bpf_prog_run_xdp(xdp_prog, &xdp);

		switch (act) {
		case XDP_REDIRECT:
			get_page(alloc_frag->page);
			alloc_frag->offset += buflen;
			err = xdp_do_redirect(tun->dev, &xdp, xdp_prog);
1666
			xdp_do_flush_map();
J
Jason Wang 已提交
1667 1668
			if (err)
				goto err_redirect;
1669
			rcu_read_unlock();
1670
			preempt_enable();
J
Jason Wang 已提交
1671 1672
			return NULL;
		case XDP_TX:
J
Jason Wang 已提交
1673 1674 1675 1676 1677 1678 1679 1680
			get_page(alloc_frag->page);
			alloc_frag->offset += buflen;
			if (tun_xdp_xmit(tun->dev, &xdp))
				goto err_redirect;
			tun_xdp_flush(tun->dev);
			rcu_read_unlock();
			preempt_enable();
			return NULL;
J
Jason Wang 已提交
1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694
		case XDP_PASS:
			delta = orig_data - xdp.data;
			break;
		default:
			bpf_warn_invalid_xdp_action(act);
			/* fall through */
		case XDP_ABORTED:
			trace_xdp_exception(tun->dev, xdp_prog, act);
			/* fall through */
		case XDP_DROP:
			goto err_xdp;
		}
	}

1695
	skb = build_skb(buf, buflen);
J
Jason Wang 已提交
1696 1697
	if (!skb) {
		rcu_read_unlock();
1698
		preempt_enable();
1699
		return ERR_PTR(-ENOMEM);
J
Jason Wang 已提交
1700
	}
1701

1702
	skb_reserve(skb, pad - delta);
J
Jason Wang 已提交
1703
	skb_put(skb, len + delta);
1704 1705 1706
	get_page(alloc_frag->page);
	alloc_frag->offset += buflen;

J
Jason Wang 已提交
1707
	rcu_read_unlock();
1708
	preempt_enable();
J
Jason Wang 已提交
1709

1710
	return skb;
J
Jason Wang 已提交
1711 1712 1713 1714 1715

err_redirect:
	put_page(alloc_frag->page);
err_xdp:
	rcu_read_unlock();
1716
	preempt_enable();
J
Jason Wang 已提交
1717 1718
	this_cpu_inc(tun->pcpu_stats->rx_dropped);
	return NULL;
1719 1720
}

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

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

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

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

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

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

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

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

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

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

1778 1779
	good_linear = SKB_MAX_HEAD(align);

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

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

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

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

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

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

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

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

1863 1864 1865
		return -EINVAL;
	}

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

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

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

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

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

J
Jason Wang 已提交
1908
	if (skb_xdp) {
J
Jason Wang 已提交
1909 1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922 1923
		struct bpf_prog *xdp_prog;
		int ret;

		rcu_read_lock();
		xdp_prog = rcu_dereference(tun->xdp_prog);
		if (xdp_prog) {
			ret = do_xdp_generic(xdp_prog, skb);
			if (ret != XDP_PASS) {
				rcu_read_unlock();
				return total_len;
			}
		}
		rcu_read_unlock();
	}

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

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

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

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

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

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

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

	if (!tun)
		return -EBADFD;

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

	tun_put(tun);
	return result;
L
Linus Torvalds 已提交
1992 1993
}

J
Jason Wang 已提交
1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027
static ssize_t tun_put_user_xdp(struct tun_struct *tun,
				struct tun_file *tfile,
				struct xdp_buff *xdp,
				struct iov_iter *iter)
{
	int vnet_hdr_sz = 0;
	size_t size = xdp->data_end - xdp->data;
	struct tun_pcpu_stats *stats;
	size_t ret;

	if (tun->flags & IFF_VNET_HDR) {
		struct virtio_net_hdr gso = { 0 };

		vnet_hdr_sz = READ_ONCE(tun->vnet_hdr_sz);
		if (unlikely(iov_iter_count(iter) < vnet_hdr_sz))
			return -EINVAL;
		if (unlikely(copy_to_iter(&gso, sizeof(gso), iter) !=
			     sizeof(gso)))
			return -EFAULT;
		iov_iter_advance(iter, vnet_hdr_sz - sizeof(gso));
	}

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

	stats = get_cpu_ptr(tun->pcpu_stats);
	u64_stats_update_begin(&stats->syncp);
	stats->tx_packets++;
	stats->tx_bytes += ret;
	u64_stats_update_end(&stats->syncp);
	put_cpu_ptr(tun->pcpu_stats);

	return ret;
}

L
Linus Torvalds 已提交
2028
/* Put packet to the user space buffer */
2029
static ssize_t tun_put_user(struct tun_struct *tun,
J
Jason Wang 已提交
2030
			    struct tun_file *tfile,
2031
			    struct sk_buff *skb,
H
Herbert Xu 已提交
2032
			    struct iov_iter *iter)
L
Linus Torvalds 已提交
2033 2034
{
	struct tun_pi pi = { 0, skb->protocol };
2035
	struct tun_pcpu_stats *stats;
H
Herbert Xu 已提交
2036
	ssize_t total;
2037
	int vlan_offset = 0;
2038
	int vlan_hlen = 0;
H
Herbert Xu 已提交
2039
	int vnet_hdr_sz = 0;
2040

2041
	if (skb_vlan_tag_present(skb))
2042
		vlan_hlen = VLAN_HLEN;
L
Linus Torvalds 已提交
2043

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

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

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

H
Herbert Xu 已提交
2053 2054
		total += sizeof(pi);
		if (iov_iter_count(iter) < total) {
L
Linus Torvalds 已提交
2055 2056 2057
			/* Packet will be striped */
			pi.flags |= TUN_PKT_STRIP;
		}
2058

H
Herbert Xu 已提交
2059
		if (copy_to_iter(&pi, sizeof(pi), iter) != sizeof(pi))
L
Linus Torvalds 已提交
2060
			return -EFAULT;
2061
	}
L
Linus Torvalds 已提交
2062

H
Herbert Xu 已提交
2063
	if (vnet_hdr_sz) {
2064
		struct virtio_net_hdr gso;
2065

H
Herbert Xu 已提交
2066
		if (iov_iter_count(iter) < vnet_hdr_sz)
2067 2068
			return -EINVAL;

2069
		if (virtio_net_hdr_from_skb(skb, &gso,
2070
					    tun_is_little_endian(tun), true)) {
2071
			struct skb_shared_info *sinfo = skb_shinfo(skb);
2072 2073 2074 2075 2076 2077 2078 2079 2080 2081 2082
			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;
		}
2083

H
Herbert Xu 已提交
2084
		if (copy_to_iter(&gso, sizeof(gso), iter) != sizeof(gso))
2085
			return -EFAULT;
2086 2087

		iov_iter_advance(iter, vnet_hdr_sz - sizeof(gso));
2088 2089
	}

2090
	if (vlan_hlen) {
H
Herbert Xu 已提交
2091
		int ret;
2092
		struct veth veth;
J
Jason Wang 已提交
2093 2094

		veth.h_vlan_proto = skb->vlan_proto;
2095
		veth.h_vlan_TCI = htons(skb_vlan_tag_get(skb));
J
Jason Wang 已提交
2096 2097 2098

		vlan_offset = offsetof(struct vlan_ethhdr, h_vlan_proto);

H
Herbert Xu 已提交
2099 2100
		ret = skb_copy_datagram_iter(skb, 0, iter, vlan_offset);
		if (ret || !iov_iter_count(iter))
J
Jason Wang 已提交
2101 2102
			goto done;

H
Herbert Xu 已提交
2103 2104
		ret = copy_to_iter(&veth, sizeof(veth), iter);
		if (ret != sizeof(veth) || !iov_iter_count(iter))
J
Jason Wang 已提交
2105 2106
			goto done;
	}
L
Linus Torvalds 已提交
2107

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

J
Jason Wang 已提交
2110
done:
2111 2112 2113 2114 2115 2116 2117
	/* 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 已提交
2118 2119 2120 2121

	return total;
}

J
Jason Wang 已提交
2122
static void *tun_ring_recv(struct tun_file *tfile, int noblock, int *err)
J
Jason Wang 已提交
2123 2124
{
	DECLARE_WAITQUEUE(wait, current);
J
Jason Wang 已提交
2125
	void *ptr = NULL;
J
Jason Wang 已提交
2126
	int error = 0;
J
Jason Wang 已提交
2127

J
Jason Wang 已提交
2128 2129
	ptr = ptr_ring_consume(&tfile->tx_ring);
	if (ptr)
J
Jason Wang 已提交
2130 2131
		goto out;
	if (noblock) {
J
Jason Wang 已提交
2132
		error = -EAGAIN;
J
Jason Wang 已提交
2133 2134 2135 2136 2137 2138 2139
		goto out;
	}

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

	while (1) {
J
Jason Wang 已提交
2140 2141
		ptr = ptr_ring_consume(&tfile->tx_ring);
		if (ptr)
J
Jason Wang 已提交
2142 2143
			break;
		if (signal_pending(current)) {
J
Jason Wang 已提交
2144
			error = -ERESTARTSYS;
J
Jason Wang 已提交
2145 2146 2147
			break;
		}
		if (tfile->socket.sk->sk_shutdown & RCV_SHUTDOWN) {
J
Jason Wang 已提交
2148
			error = -EFAULT;
J
Jason Wang 已提交
2149 2150 2151 2152 2153 2154 2155 2156 2157 2158
			break;
		}

		schedule();
	}

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

out:
J
Jason Wang 已提交
2159
	*err = error;
J
Jason Wang 已提交
2160
	return ptr;
J
Jason Wang 已提交
2161 2162
}

J
Jason Wang 已提交
2163
static ssize_t tun_do_read(struct tun_struct *tun, struct tun_file *tfile,
2164
			   struct iov_iter *to,
J
Jason Wang 已提交
2165
			   int noblock, void *ptr)
L
Linus Torvalds 已提交
2166
{
2167
	ssize_t ret;
J
Jason Wang 已提交
2168
	int err;
L
Linus Torvalds 已提交
2169

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

W
Wei Xu 已提交
2172
	if (!iov_iter_count(to)) {
J
Jason Wang 已提交
2173
		tun_ptr_free(ptr);
2174
		return 0;
W
Wei Xu 已提交
2175
	}
L
Linus Torvalds 已提交
2176

J
Jason Wang 已提交
2177
	if (!ptr) {
2178
		/* Read frames from ring */
J
Jason Wang 已提交
2179 2180
		ptr = tun_ring_recv(tfile, noblock, &err);
		if (!ptr)
2181 2182
			return err;
	}
H
Herbert Xu 已提交
2183

J
Jason Wang 已提交
2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197
	if (tun_is_xdp_buff(ptr)) {
		struct xdp_buff *xdp = tun_ptr_to_xdp(ptr);

		ret = tun_put_user_xdp(tun, tfile, xdp, to);
		put_page(virt_to_head_page(xdp->data));
	} else {
		struct sk_buff *skb = ptr;

		ret = tun_put_user(tun, tfile, skb, to);
		if (unlikely(ret < 0))
			kfree_skb(skb);
		else
			consume_skb(skb);
	}
L
Linus Torvalds 已提交
2198

2199 2200 2201
	return ret;
}

2202
static ssize_t tun_chr_read_iter(struct kiocb *iocb, struct iov_iter *to)
2203 2204 2205
{
	struct file *file = iocb->ki_filp;
	struct tun_file *tfile = file->private_data;
2206
	struct tun_struct *tun = tun_get(tfile);
2207
	ssize_t len = iov_iter_count(to), ret;
2208 2209 2210

	if (!tun)
		return -EBADFD;
2211
	ret = tun_do_read(tun, tfile, to, file->f_flags & O_NONBLOCK, NULL);
2212
	ret = min_t(ssize_t, ret, len);
Z
Zhi Yong Wu 已提交
2213 2214
	if (ret > 0)
		iocb->ki_pos = ret;
E
Eric W. Biederman 已提交
2215
	tun_put(tun);
L
Linus Torvalds 已提交
2216 2217 2218
	return ret;
}

2219
static void tun_prog_free(struct rcu_head *rcu)
2220
{
2221
	struct tun_prog *prog = container_of(rcu, struct tun_prog, rcu);
2222 2223 2224 2225 2226

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

J
Jason Wang 已提交
2227 2228
static int __tun_set_ebpf(struct tun_struct *tun,
			  struct tun_prog __rcu **prog_p,
2229
			  struct bpf_prog *prog)
2230
{
2231
	struct tun_prog *old, *new = NULL;
2232 2233 2234 2235 2236 2237 2238 2239

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

2240
	spin_lock_bh(&tun->lock);
2241
	old = rcu_dereference_protected(*prog_p,
2242
					lockdep_is_held(&tun->lock));
2243
	rcu_assign_pointer(*prog_p, new);
2244
	spin_unlock_bh(&tun->lock);
2245 2246

	if (old)
2247
		call_rcu(&old->rcu, tun_prog_free);
2248 2249 2250 2251

	return 0;
}

J
Jason Wang 已提交
2252 2253 2254 2255
static void tun_free_netdev(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

J
Jason Wang 已提交
2256
	BUG_ON(!(list_empty(&tun->disabled)));
2257
	free_percpu(tun->pcpu_stats);
J
Jason Wang 已提交
2258
	tun_flow_uninit(tun);
2259
	security_tun_dev_free_security(tun->security);
2260
	__tun_set_ebpf(tun, &tun->steering_prog, NULL);
2261
	__tun_set_ebpf(tun, &tun->filter_prog, NULL);
J
Jason Wang 已提交
2262 2263
}

L
Linus Torvalds 已提交
2264 2265 2266 2267
static void tun_setup(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

2268 2269
	tun->owner = INVALID_UID;
	tun->group = INVALID_GID;
L
Linus Torvalds 已提交
2270 2271

	dev->ethtool_ops = &tun_ethtool_ops;
2272 2273
	dev->needs_free_netdev = true;
	dev->priv_destructor = tun_free_netdev;
J
Jason Wang 已提交
2274 2275
	/* We prefer our own queue length */
	dev->tx_queue_len = TUN_READQ_SIZE;
L
Linus Torvalds 已提交
2276 2277
}

2278 2279 2280
/* Trivial set of netlink ops to allow deleting tun or tap
 * device with netlink.
 */
2281 2282
static int tun_validate(struct nlattr *tb[], struct nlattr *data[],
			struct netlink_ext_ack *extack)
2283 2284 2285 2286
{
	return -EINVAL;
}

2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340
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;
}

2341 2342 2343 2344 2345
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,
2346 2347
	.get_size       = tun_get_size,
	.fill_info      = tun_fill_info,
2348 2349
};

2350 2351
static void tun_sock_write_space(struct sock *sk)
{
J
Jason Wang 已提交
2352
	struct tun_file *tfile;
2353
	wait_queue_head_t *wqueue;
2354 2355 2356 2357

	if (!sock_writeable(sk))
		return;

2358
	if (!test_and_clear_bit(SOCKWQ_ASYNC_NOSPACE, &sk->sk_socket->flags))
2359 2360
		return;

2361 2362
	wqueue = sk_sleep(sk);
	if (wqueue && waitqueue_active(wqueue))
2363 2364
		wake_up_interruptible_sync_poll(wqueue, EPOLLOUT |
						EPOLLWRNORM | EPOLLWRBAND);
H
Herbert Xu 已提交
2365

J
Jason Wang 已提交
2366 2367
	tfile = container_of(sk, struct tun_file, sk);
	kill_fasync(&tfile->fasync, SIGIO, POLL_OUT);
2368 2369
}

2370
static int tun_sendmsg(struct socket *sock, struct msghdr *m, size_t total_len)
2371
{
J
Jason Wang 已提交
2372 2373
	int ret;
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2374
	struct tun_struct *tun = tun_get(tfile);
J
Jason Wang 已提交
2375 2376 2377

	if (!tun)
		return -EBADFD;
2378

A
Al Viro 已提交
2379
	ret = tun_get_user(tun, tfile, m->msg_control, &m->msg_iter,
J
Jason Wang 已提交
2380 2381
			   m->msg_flags & MSG_DONTWAIT,
			   m->msg_flags & MSG_MORE);
J
Jason Wang 已提交
2382 2383
	tun_put(tun);
	return ret;
2384 2385
}

2386
static int tun_recvmsg(struct socket *sock, struct msghdr *m, size_t total_len,
2387 2388
		       int flags)
{
J
Jason Wang 已提交
2389
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2390
	struct tun_struct *tun = tun_get(tfile);
J
Jason Wang 已提交
2391
	void *ptr = m->msg_control;
2392
	int ret;
J
Jason Wang 已提交
2393

W
Wei Xu 已提交
2394 2395
	if (!tun) {
		ret = -EBADFD;
J
Jason Wang 已提交
2396
		goto out_free;
W
Wei Xu 已提交
2397
	}
J
Jason Wang 已提交
2398

2399
	if (flags & ~(MSG_DONTWAIT|MSG_TRUNC|MSG_ERRQUEUE)) {
2400
		ret = -EINVAL;
W
Wei Xu 已提交
2401
		goto out_put_tun;
2402
	}
2403 2404 2405 2406 2407
	if (flags & MSG_ERRQUEUE) {
		ret = sock_recv_errqueue(sock->sk, m, total_len,
					 SOL_PACKET, TUN_TX_TIMESTAMP);
		goto out;
	}
J
Jason Wang 已提交
2408
	ret = tun_do_read(tun, tfile, &m->msg_iter, flags & MSG_DONTWAIT, ptr);
2409
	if (ret > (ssize_t)total_len) {
2410 2411 2412
		m->msg_flags |= MSG_TRUNC;
		ret = flags & MSG_TRUNC ? ret : total_len;
	}
2413
out:
J
Jason Wang 已提交
2414
	tun_put(tun);
2415
	return ret;
W
Wei Xu 已提交
2416 2417 2418

out_put_tun:
	tun_put(tun);
J
Jason Wang 已提交
2419 2420
out_free:
	tun_ptr_free(ptr);
W
Wei Xu 已提交
2421
	return ret;
2422 2423
}

J
Jason Wang 已提交
2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437
static int tun_ptr_peek_len(void *ptr)
{
	if (likely(ptr)) {
		if (tun_is_xdp_buff(ptr)) {
			struct xdp_buff *xdp = tun_ptr_to_xdp(ptr);

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

J
Jason Wang 已提交
2438 2439 2440 2441 2442 2443
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;

2444
	tun = tun_get(tfile);
J
Jason Wang 已提交
2445 2446 2447
	if (!tun)
		return 0;

J
Jason Wang 已提交
2448
	ret = PTR_RING_PEEK_CALL(&tfile->tx_ring, tun_ptr_peek_len);
J
Jason Wang 已提交
2449 2450 2451 2452 2453
	tun_put(tun);

	return ret;
}

2454 2455
/* Ops structure to mimic raw sockets with tun */
static const struct proto_ops tun_socket_ops = {
J
Jason Wang 已提交
2456
	.peek_len = tun_peek_len,
2457 2458 2459 2460
	.sendmsg = tun_sendmsg,
	.recvmsg = tun_recvmsg,
};

2461 2462 2463
static struct proto tun_proto = {
	.name		= "tun",
	.owner		= THIS_MODULE,
J
Jason Wang 已提交
2464
	.obj_size	= sizeof(struct tun_file),
2465
};
2466

2467 2468
static int tun_flags(struct tun_struct *tun)
{
2469
	return tun->flags & (TUN_FEATURES | IFF_PERSIST | IFF_TUN | IFF_TAP);
2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482
}

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));
2483 2484 2485 2486
	return uid_valid(tun->owner)?
		sprintf(buf, "%u\n",
			from_kuid_munged(current_user_ns(), tun->owner)):
		sprintf(buf, "-1\n");
2487 2488 2489 2490 2491 2492
}

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));
2493 2494 2495 2496
	return gid_valid(tun->group) ?
		sprintf(buf, "%u\n",
			from_kgid_munged(current_user_ns(), tun->group)):
		sprintf(buf, "-1\n");
2497 2498 2499 2500 2501 2502
}

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

2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513
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
};

2514
static int tun_set_iff(struct net *net, struct file *file, struct ifreq *ifr)
L
Linus Torvalds 已提交
2515 2516
{
	struct tun_struct *tun;
J
Jason Wang 已提交
2517
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
2518 2519 2520
	struct net_device *dev;
	int err;

2521 2522 2523
	if (tfile->detached)
		return -EINVAL;

2524 2525 2526 2527 2528 2529 2530 2531 2532
	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;
	}

2533 2534
	dev = __dev_get_by_name(net, ifr->ifr_name);
	if (dev) {
2535 2536
		if (ifr->ifr_flags & IFF_TUN_EXCL)
			return -EBUSY;
2537 2538 2539 2540 2541 2542 2543
		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;

2544
		if (!!(ifr->ifr_flags & IFF_MULTI_QUEUE) !=
M
Michael S. Tsirkin 已提交
2545
		    !!(tun->flags & IFF_MULTI_QUEUE))
2546 2547
			return -EINVAL;

2548
		if (tun_not_capable(tun))
P
Paul Moore 已提交
2549
			return -EPERM;
2550
		err = security_tun_dev_open(tun->security);
P
Paul Moore 已提交
2551 2552 2553
		if (err < 0)
			return err;

2554 2555
		err = tun_attach(tun, file, ifr->ifr_flags & IFF_NOFILTER,
				 ifr->ifr_flags & IFF_NAPI);
2556 2557
		if (err < 0)
			return err;
J
Jason Wang 已提交
2558

M
Michael S. Tsirkin 已提交
2559
		if (tun->flags & IFF_MULTI_QUEUE &&
2560 2561 2562 2563 2564 2565
		    (tun->numqueues + tun->numdisabled > 1)) {
			/* One or more queue has already been attached, no need
			 * to initialize the device again.
			 */
			return 0;
		}
2566
	}
L
Linus Torvalds 已提交
2567 2568 2569
	else {
		char *name;
		unsigned long flags = 0;
2570 2571
		int queues = ifr->ifr_flags & IFF_MULTI_QUEUE ?
			     MAX_TAP_QUEUES : 1;
L
Linus Torvalds 已提交
2572

2573
		if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
2574
			return -EPERM;
P
Paul Moore 已提交
2575 2576 2577
		err = security_tun_dev_create();
		if (err < 0)
			return err;
2578

L
Linus Torvalds 已提交
2579 2580 2581
		/* Set dev type */
		if (ifr->ifr_flags & IFF_TUN) {
			/* TUN device */
M
Michael S. Tsirkin 已提交
2582
			flags |= IFF_TUN;
L
Linus Torvalds 已提交
2583 2584 2585
			name = "tun%d";
		} else if (ifr->ifr_flags & IFF_TAP) {
			/* TAP device */
M
Michael S. Tsirkin 已提交
2586
			flags |= IFF_TAP;
L
Linus Torvalds 已提交
2587
			name = "tap%d";
2588
		} else
2589
			return -EINVAL;
2590

L
Linus Torvalds 已提交
2591 2592 2593
		if (*ifr->ifr_name)
			name = ifr->ifr_name;

J
Jason Wang 已提交
2594
		dev = alloc_netdev_mqs(sizeof(struct tun_struct), name,
2595 2596
				       NET_NAME_UNKNOWN, tun_setup, queues,
				       queues);
2597

L
Linus Torvalds 已提交
2598 2599
		if (!dev)
			return -ENOMEM;
2600
		err = dev_get_valid_name(net, dev, name);
2601
		if (err < 0)
2602
			goto err_free_dev;
L
Linus Torvalds 已提交
2603

2604
		dev_net_set(dev, net);
2605
		dev->rtnl_link_ops = &tun_link_ops;
2606
		dev->ifindex = tfile->ifindex;
2607
		dev->sysfs_groups[0] = &tun_attr_group;
S
Stephen Hemminger 已提交
2608

L
Linus Torvalds 已提交
2609 2610 2611
		tun = netdev_priv(dev);
		tun->dev = dev;
		tun->flags = flags;
2612
		tun->txflt.count = 0;
2613
		tun->vnet_hdr_sz = sizeof(struct virtio_net_hdr);
2614

2615
		tun->align = NET_SKB_PAD;
J
Jason Wang 已提交
2616 2617
		tun->filter_attached = false;
		tun->sndbuf = tfile->socket.sk->sk_sndbuf;
J
Jason Wang 已提交
2618
		tun->rx_batched = 0;
2619
		RCU_INIT_POINTER(tun->steering_prog, NULL);
2620

2621 2622 2623 2624 2625 2626
		tun->pcpu_stats = netdev_alloc_pcpu_stats(struct tun_pcpu_stats);
		if (!tun->pcpu_stats) {
			err = -ENOMEM;
			goto err_free_dev;
		}

J
Jason Wang 已提交
2627 2628
		spin_lock_init(&tun->lock);

2629 2630
		err = security_tun_dev_alloc_security(&tun->security);
		if (err < 0)
2631
			goto err_free_stat;
P
Paul Moore 已提交
2632

L
Linus Torvalds 已提交
2633
		tun_net_init(dev);
2634
		tun_flow_init(tun);
J
Jason Wang 已提交
2635

2636
		dev->hw_features = NETIF_F_SG | NETIF_F_FRAGLIST |
J
Jason Wang 已提交
2637 2638
				   TUN_USER_FEATURES | NETIF_F_HW_VLAN_CTAG_TX |
				   NETIF_F_HW_VLAN_STAG_TX;
2639
		dev->features = dev->hw_features | NETIF_F_LLTX;
2640 2641 2642
		dev->vlan_features = dev->features &
				     ~(NETIF_F_HW_VLAN_CTAG_TX |
				       NETIF_F_HW_VLAN_STAG_TX);
2643

J
Jason Wang 已提交
2644
		INIT_LIST_HEAD(&tun->disabled);
2645
		err = tun_attach(tun, file, false, ifr->ifr_flags & IFF_NAPI);
2646
		if (err < 0)
2647
			goto err_free_flow;
2648

L
Linus Torvalds 已提交
2649 2650
		err = register_netdevice(tun->dev);
		if (err < 0)
2651
			goto err_detach;
L
Linus Torvalds 已提交
2652 2653
	}

2654 2655
	netif_carrier_on(tun->dev);

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

2658 2659
	tun->flags = (tun->flags & ~TUN_FEATURES) |
		(ifr->ifr_flags & TUN_FEATURES);
J
Jason Wang 已提交
2660

2661 2662 2663 2664
	/* Make sure persistent devices do not get stuck in
	 * xoff state.
	 */
	if (netif_running(tun->dev))
J
Jason Wang 已提交
2665
		netif_tx_wake_all_queues(tun->dev);
2666

L
Linus Torvalds 已提交
2667 2668 2669
	strcpy(ifr->ifr_name, tun->dev->name);
	return 0;

2670 2671
err_detach:
	tun_detach_all(dev);
2672 2673 2674
	/* register_netdevice() already called tun_free_netdev() */
	goto err_free_dev;

2675 2676 2677
err_free_flow:
	tun_flow_uninit(tun);
	security_tun_dev_free_security(tun->security);
2678 2679
err_free_stat:
	free_percpu(tun->pcpu_stats);
2680
err_free_dev:
L
Linus Torvalds 已提交
2681 2682 2683 2684
	free_netdev(dev);
	return err;
}

R
Rami Rosen 已提交
2685
static void tun_get_iff(struct net *net, struct tun_struct *tun,
2686
		       struct ifreq *ifr)
2687
{
2688
	tun_debug(KERN_INFO, tun, "tun_get_iff\n");
2689 2690 2691

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

2692
	ifr->ifr_flags = tun_flags(tun);
2693 2694 2695

}

2696 2697
/* This is like a cut-down ethtool ops, except done via tun fd so no
 * privs required. */
2698
static int set_offload(struct tun_struct *tun, unsigned long arg)
2699
{
2700
	netdev_features_t features = 0;
2701 2702

	if (arg & TUN_F_CSUM) {
2703
		features |= NETIF_F_HW_CSUM;
2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714 2715 2716
		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);
		}
2717 2718

		arg &= ~TUN_F_UFO;
2719 2720 2721 2722 2723 2724 2725
	}

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

2726
	tun->set_features = features;
2727 2728
	tun->dev->wanted_features &= ~TUN_USER_FEATURES;
	tun->dev->wanted_features |= features;
2729
	netdev_update_features(tun->dev);
2730 2731 2732 2733

	return 0;
}

J
Jason Wang 已提交
2734 2735 2736 2737 2738 2739
static void tun_detach_filter(struct tun_struct *tun, int n)
{
	int i;
	struct tun_file *tfile;

	for (i = 0; i < n; i++) {
2740
		tfile = rtnl_dereference(tun->tfiles[i]);
2741 2742 2743
		lock_sock(tfile->socket.sk);
		sk_detach_filter(tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754
	}

	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++) {
2755
		tfile = rtnl_dereference(tun->tfiles[i]);
2756 2757 2758
		lock_sock(tfile->socket.sk);
		ret = sk_attach_filter(&tun->fprog, tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772 2773 2774
		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++) {
2775
		tfile = rtnl_dereference(tun->tfiles[i]);
J
Jason Wang 已提交
2776 2777 2778 2779
		tfile->socket.sk->sk_sndbuf = tun->sndbuf;
	}
}

2780 2781 2782 2783 2784 2785 2786 2787 2788
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 已提交
2789
		tun = tfile->detached;
2790
		if (!tun) {
2791
			ret = -EINVAL;
2792 2793 2794 2795 2796
			goto unlock;
		}
		ret = security_tun_dev_attach_queue(tun->security);
		if (ret < 0)
			goto unlock;
2797
		ret = tun_attach(tun, file, false, tun->flags & IFF_NAPI);
J
Jason Wang 已提交
2798
	} else if (ifr->ifr_flags & IFF_DETACH_QUEUE) {
2799
		tun = rtnl_dereference(tfile->tun);
M
Michael S. Tsirkin 已提交
2800
		if (!tun || !(tun->flags & IFF_MULTI_QUEUE) || tfile->detached)
J
Jason Wang 已提交
2801 2802 2803 2804
			ret = -EINVAL;
		else
			__tun_detach(tfile, false);
	} else
2805 2806
		ret = -EINVAL;

2807
unlock:
2808 2809 2810 2811
	rtnl_unlock();
	return ret;
}

2812 2813
static int tun_set_ebpf(struct tun_struct *tun, struct tun_prog **prog_p,
			void __user *data)
2814 2815 2816 2817 2818 2819 2820 2821 2822 2823 2824 2825 2826 2827 2828
{
	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);
	}

2829
	return __tun_set_ebpf(tun, prog_p, prog);
2830 2831
}

2832 2833
static long __tun_chr_ioctl(struct file *file, unsigned int cmd,
			    unsigned long arg, int ifreq_len)
L
Linus Torvalds 已提交
2834
{
E
Eric W. Biederman 已提交
2835
	struct tun_file *tfile = file->private_data;
E
Eric W. Biederman 已提交
2836
	struct tun_struct *tun;
L
Linus Torvalds 已提交
2837 2838
	void __user* argp = (void __user*)arg;
	struct ifreq ifr;
2839
	struct net *net;
2840 2841
	kuid_t owner;
	kgid_t group;
2842
	int sndbuf;
2843
	int vnet_hdr_sz;
2844
	unsigned int ifindex;
M
Michael S. Tsirkin 已提交
2845
	int le;
2846
	int ret;
L
Linus Torvalds 已提交
2847

2848 2849
	if (cmd == TUNSETIFF || cmd == TUNSETQUEUE ||
	    (_IOC_TYPE(cmd) == SOCK_IOC_TYPE && cmd != SIOCGSKNS)) {
2850
		if (copy_from_user(&ifr, argp, ifreq_len))
L
Linus Torvalds 已提交
2851
			return -EFAULT;
D
David S. Miller 已提交
2852
	} else {
2853
		memset(&ifr, 0, sizeof(ifr));
D
David S. Miller 已提交
2854
	}
E
Eric W. Biederman 已提交
2855 2856 2857
	if (cmd == TUNGETFEATURES) {
		/* Currently this just means: "what IFF flags are valid?".
		 * This is needed because we never checked for invalid flags on
2858 2859 2860
		 * TUNSETIFF.
		 */
		return put_user(IFF_TUN | IFF_TAP | TUN_FEATURES,
E
Eric W. Biederman 已提交
2861
				(unsigned int __user*)argp);
2862 2863
	} else if (cmd == TUNSETQUEUE)
		return tun_set_queue(file, &ifr);
E
Eric W. Biederman 已提交
2864

J
Jason Wang 已提交
2865
	ret = 0;
2866 2867
	rtnl_lock();

2868
	tun = tun_get(tfile);
2869
	net = sock_net(&tfile->sk);
2870 2871 2872 2873 2874
	if (cmd == TUNSETIFF) {
		ret = -EEXIST;
		if (tun)
			goto unlock;

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

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

2879 2880
		if (ret)
			goto unlock;
L
Linus Torvalds 已提交
2881

2882
		if (copy_to_user(argp, &ifr, ifreq_len))
2883 2884
			ret = -EFAULT;
		goto unlock;
L
Linus Torvalds 已提交
2885
	}
2886 2887 2888 2889 2890 2891 2892 2893 2894 2895 2896 2897 2898
	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;
	}
2899 2900 2901 2902 2903 2904 2905 2906
	if (cmd == SIOCGSKNS) {
		ret = -EPERM;
		if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
			goto unlock;

		ret = open_related_ns(&net->ns, get_net_ns);
		goto unlock;
	}
L
Linus Torvalds 已提交
2907

2908
	ret = -EBADFD;
L
Linus Torvalds 已提交
2909
	if (!tun)
2910
		goto unlock;
L
Linus Torvalds 已提交
2911

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

E
Eric W. Biederman 已提交
2914
	ret = 0;
L
Linus Torvalds 已提交
2915
	switch (cmd) {
2916
	case TUNGETIFF:
R
Rami Rosen 已提交
2917
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
2918

2919 2920
		if (tfile->detached)
			ifr.ifr_flags |= IFF_DETACH_QUEUE;
2921 2922
		if (!tfile->socket.sk->sk_filter)
			ifr.ifr_flags |= IFF_NOFILTER;
2923

2924
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
2925
			ret = -EFAULT;
2926 2927
		break;

L
Linus Torvalds 已提交
2928 2929 2930
	case TUNSETNOCSUM:
		/* Disable/Enable checksum */

2931 2932
		/* [unimplemented] */
		tun_debug(KERN_INFO, tun, "ignored: set checksum %s\n",
2933
			  arg ? "disabled" : "enabled");
L
Linus Torvalds 已提交
2934 2935 2936
		break;

	case TUNSETPERSIST:
J
Jason Wang 已提交
2937 2938 2939
		/* Disable/Enable persist mode. Keep an extra reference to the
		 * module to prevent the module being unprobed.
		 */
M
Michael S. Tsirkin 已提交
2940 2941
		if (arg && !(tun->flags & IFF_PERSIST)) {
			tun->flags |= IFF_PERSIST;
J
Jason Wang 已提交
2942
			__module_get(THIS_MODULE);
J
Jason Wang 已提交
2943
		}
M
Michael S. Tsirkin 已提交
2944 2945
		if (!arg && (tun->flags & IFF_PERSIST)) {
			tun->flags &= ~IFF_PERSIST;
J
Jason Wang 已提交
2946 2947
			module_put(THIS_MODULE);
		}
L
Linus Torvalds 已提交
2948

2949 2950
		tun_debug(KERN_INFO, tun, "persist %s\n",
			  arg ? "enabled" : "disabled");
L
Linus Torvalds 已提交
2951 2952 2953 2954
		break;

	case TUNSETOWNER:
		/* Set owner of the device */
2955 2956 2957 2958 2959 2960
		owner = make_kuid(current_user_ns(), arg);
		if (!uid_valid(owner)) {
			ret = -EINVAL;
			break;
		}
		tun->owner = owner;
2961
		tun_debug(KERN_INFO, tun, "owner set to %u\n",
2962
			  from_kuid(&init_user_ns, tun->owner));
L
Linus Torvalds 已提交
2963 2964
		break;

2965 2966
	case TUNSETGROUP:
		/* Set group of the device */
2967 2968 2969 2970 2971 2972
		group = make_kgid(current_user_ns(), arg);
		if (!gid_valid(group)) {
			ret = -EINVAL;
			break;
		}
		tun->group = group;
2973
		tun_debug(KERN_INFO, tun, "group set to %u\n",
2974
			  from_kgid(&init_user_ns, tun->group));
2975 2976
		break;

2977 2978 2979
	case TUNSETLINK:
		/* Only allow setting the type when the interface is down */
		if (tun->dev->flags & IFF_UP) {
2980 2981
			tun_debug(KERN_INFO, tun,
				  "Linktype set failed because interface is up\n");
2982
			ret = -EBUSY;
2983 2984
		} else {
			tun->dev->type = (int) arg;
2985 2986
			tun_debug(KERN_INFO, tun, "linktype set to %d\n",
				  tun->dev->type);
2987
			ret = 0;
2988
		}
E
Eric W. Biederman 已提交
2989
		break;
2990

L
Linus Torvalds 已提交
2991 2992 2993 2994 2995
#ifdef TUN_DEBUG
	case TUNSETDEBUG:
		tun->debug = arg;
		break;
#endif
2996
	case TUNSETOFFLOAD:
2997
		ret = set_offload(tun, arg);
E
Eric W. Biederman 已提交
2998
		break;
2999

3000 3001
	case TUNSETTXFILTER:
		/* Can be set only for TAPs */
E
Eric W. Biederman 已提交
3002
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3003
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
E
Eric W. Biederman 已提交
3004
			break;
H
Harvey Harrison 已提交
3005
		ret = update_filter(&tun->txflt, (void __user *)arg);
E
Eric W. Biederman 已提交
3006
		break;
L
Linus Torvalds 已提交
3007 3008

	case SIOCGIFHWADDR:
3009
		/* Get hw address */
3010 3011
		memcpy(ifr.ifr_hwaddr.sa_data, tun->dev->dev_addr, ETH_ALEN);
		ifr.ifr_hwaddr.sa_family = tun->dev->type;
3012
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
3013 3014
			ret = -EFAULT;
		break;
L
Linus Torvalds 已提交
3015 3016

	case SIOCSIFHWADDR:
3017
		/* Set hw address */
3018 3019
		tun_debug(KERN_DEBUG, tun, "set hw address: %pM\n",
			  ifr.ifr_hwaddr.sa_data);
3020 3021

		ret = dev_set_mac_address(tun->dev, &ifr.ifr_hwaddr);
E
Eric W. Biederman 已提交
3022
		break;
3023 3024

	case TUNGETSNDBUF:
J
Jason Wang 已提交
3025
		sndbuf = tfile->socket.sk->sk_sndbuf;
3026 3027 3028 3029 3030 3031 3032 3033 3034
		if (copy_to_user(argp, &sndbuf, sizeof(sndbuf)))
			ret = -EFAULT;
		break;

	case TUNSETSNDBUF:
		if (copy_from_user(&sndbuf, argp, sizeof(sndbuf))) {
			ret = -EFAULT;
			break;
		}
3035 3036 3037 3038
		if (sndbuf <= 0) {
			ret = -EINVAL;
			break;
		}
3039

J
Jason Wang 已提交
3040 3041
		tun->sndbuf = sndbuf;
		tun_set_sndbuf(tun);
3042 3043
		break;

3044 3045 3046 3047 3048 3049 3050 3051 3052 3053 3054 3055 3056 3057 3058 3059 3060 3061 3062
	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 已提交
3063 3064 3065 3066 3067 3068 3069 3070 3071 3072 3073 3074 3075 3076 3077 3078 3079
	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;

3080 3081 3082 3083 3084 3085 3086 3087
	case TUNGETVNETBE:
		ret = tun_get_vnet_be(tun, argp);
		break;

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

M
Michael S. Tsirkin 已提交
3088 3089 3090
	case TUNATTACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3091
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
3092 3093
			break;
		ret = -EFAULT;
J
Jason Wang 已提交
3094
		if (copy_from_user(&tun->fprog, argp, sizeof(tun->fprog)))
M
Michael S. Tsirkin 已提交
3095 3096
			break;

J
Jason Wang 已提交
3097
		ret = tun_attach_filter(tun);
M
Michael S. Tsirkin 已提交
3098 3099 3100 3101 3102
		break;

	case TUNDETACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3103
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
3104
			break;
J
Jason Wang 已提交
3105 3106
		ret = 0;
		tun_detach_filter(tun, tun->numqueues);
M
Michael S. Tsirkin 已提交
3107 3108
		break;

P
Pavel Emelyanov 已提交
3109 3110
	case TUNGETFILTER:
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3111
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
P
Pavel Emelyanov 已提交
3112 3113 3114 3115 3116 3117 3118
			break;
		ret = -EFAULT;
		if (copy_to_user(argp, &tun->fprog, sizeof(tun->fprog)))
			break;
		ret = 0;
		break;

3119
	case TUNSETSTEERINGEBPF:
3120
		ret = tun_set_ebpf(tun, &tun->steering_prog, argp);
3121 3122
		break;

3123 3124 3125 3126
	case TUNSETFILTEREBPF:
		ret = tun_set_ebpf(tun, &tun->filter_prog, argp);
		break;

L
Linus Torvalds 已提交
3127
	default:
E
Eric W. Biederman 已提交
3128 3129
		ret = -EINVAL;
		break;
3130
	}
L
Linus Torvalds 已提交
3131

3132 3133 3134 3135
unlock:
	rtnl_unlock();
	if (tun)
		tun_put(tun);
E
Eric W. Biederman 已提交
3136
	return ret;
L
Linus Torvalds 已提交
3137 3138
}

3139 3140 3141 3142 3143 3144 3145 3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156 3157 3158 3159 3160 3161 3162 3163 3164 3165 3166 3167 3168 3169 3170 3171 3172 3173
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 已提交
3174 3175
static int tun_chr_fasync(int fd, struct file *file, int on)
{
J
Jason Wang 已提交
3176
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
3177 3178
	int ret;

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

L
Linus Torvalds 已提交
3182
	if (on) {
3183
		__f_setown(file, task_pid(current), PIDTYPE_PID, 0);
J
Jason Wang 已提交
3184
		tfile->flags |= TUN_FASYNC;
3185
	} else
J
Jason Wang 已提交
3186
		tfile->flags &= ~TUN_FASYNC;
J
Jonathan Corbet 已提交
3187 3188 3189
	ret = 0;
out:
	return ret;
L
Linus Torvalds 已提交
3190 3191 3192 3193
}

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

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

3199
	tfile = (struct tun_file *)sk_alloc(net, AF_UNSPEC, GFP_KERNEL,
3200
					    &tun_proto, 0);
E
Eric W. Biederman 已提交
3201 3202
	if (!tfile)
		return -ENOMEM;
3203
	RCU_INIT_POINTER(tfile->tun, NULL);
J
Jason Wang 已提交
3204
	tfile->flags = 0;
3205
	tfile->ifindex = 0;
J
Jason Wang 已提交
3206 3207

	init_waitqueue_head(&tfile->wq.wait);
3208
	RCU_INIT_POINTER(tfile->socket.wq, &tfile->wq);
J
Jason Wang 已提交
3209 3210 3211 3212 3213 3214 3215 3216 3217

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

3221 3222
	sock_set_flag(&tfile->sk, SOCK_ZEROCOPY);

3223
	memset(&tfile->tx_ring, 0, sizeof(tfile->tx_ring));
3224

L
Linus Torvalds 已提交
3225 3226 3227 3228 3229
	return 0;
}

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

J
Jason Wang 已提交
3232
	tun_detach(tfile, true);
L
Linus Torvalds 已提交
3233 3234 3235 3236

	return 0;
}

3237
#ifdef CONFIG_PROC_FS
3238
static void tun_chr_show_fdinfo(struct seq_file *m, struct file *file)
3239
{
3240
	struct tun_file *tfile = file->private_data;
3241 3242 3243 3244 3245 3246
	struct tun_struct *tun;
	struct ifreq ifr;

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

	rtnl_lock();
3247
	tun = tun_get(tfile);
3248 3249 3250 3251 3252 3253 3254
	if (tun)
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
	rtnl_unlock();

	if (tun)
		tun_put(tun);

3255
	seq_printf(m, "iff:\t%s\n", ifr.ifr_name);
3256 3257 3258
}
#endif

3259
static const struct file_operations tun_fops = {
3260
	.owner	= THIS_MODULE,
L
Linus Torvalds 已提交
3261
	.llseek = no_llseek,
3262
	.read_iter  = tun_chr_read_iter,
3263
	.write_iter = tun_chr_write_iter,
L
Linus Torvalds 已提交
3264
	.poll	= tun_chr_poll,
3265 3266 3267 3268
	.unlocked_ioctl	= tun_chr_ioctl,
#ifdef CONFIG_COMPAT
	.compat_ioctl = tun_chr_compat_ioctl,
#endif
L
Linus Torvalds 已提交
3269 3270
	.open	= tun_chr_open,
	.release = tun_chr_close,
3271 3272 3273 3274
	.fasync = tun_chr_fasync,
#ifdef CONFIG_PROC_FS
	.show_fdinfo = tun_chr_show_fdinfo,
#endif
L
Linus Torvalds 已提交
3275 3276 3277 3278 3279
};

static struct miscdevice tun_miscdev = {
	.minor = TUN_MINOR,
	.name = "tun",
3280
	.nodename = "net/tun",
L
Linus Torvalds 已提交
3281 3282 3283 3284 3285
	.fops = &tun_fops,
};

/* ethtool interface */

3286 3287 3288 3289 3290 3291 3292 3293 3294 3295
static int tun_get_link_ksettings(struct net_device *dev,
				  struct ethtool_link_ksettings *cmd)
{
	ethtool_link_ksettings_zero_link_mode(cmd, supported);
	ethtool_link_ksettings_zero_link_mode(cmd, advertising);
	cmd->base.speed		= SPEED_10;
	cmd->base.duplex	= DUPLEX_FULL;
	cmd->base.port		= PORT_TP;
	cmd->base.phy_address	= 0;
	cmd->base.autoneg	= AUTONEG_DISABLE;
L
Linus Torvalds 已提交
3296 3297 3298 3299 3300 3301 3302
	return 0;
}

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

3303 3304
	strlcpy(info->driver, DRV_NAME, sizeof(info->driver));
	strlcpy(info->version, DRV_VERSION, sizeof(info->version));
L
Linus Torvalds 已提交
3305 3306

	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
3307
	case IFF_TUN:
3308
		strlcpy(info->bus_info, "tun", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3309
		break;
M
Michael S. Tsirkin 已提交
3310
	case IFF_TAP:
3311
		strlcpy(info->bus_info, "tap", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3312 3313 3314 3315 3316 3317 3318 3319 3320 3321 3322 3323 3324 3325 3326 3327 3328 3329 3330 3331 3332 3333
		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 已提交
3334 3335 3336 3337 3338 3339 3340 3341 3342 3343 3344 3345 3346 3347 3348 3349 3350 3351 3352 3353 3354 3355 3356
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;
}

3357
static const struct ethtool_ops tun_ethtool_ops = {
L
Linus Torvalds 已提交
3358 3359 3360
	.get_drvinfo	= tun_get_drvinfo,
	.get_msglevel	= tun_get_msglevel,
	.set_msglevel	= tun_set_msglevel,
3361
	.get_link	= ethtool_op_get_link,
3362
	.get_ts_info	= ethtool_op_get_ts_info,
J
Jason Wang 已提交
3363 3364
	.get_coalesce   = tun_get_coalesce,
	.set_coalesce   = tun_set_coalesce,
3365
	.get_link_ksettings = tun_get_link_ksettings,
L
Linus Torvalds 已提交
3366 3367
};

J
Jason Wang 已提交
3368 3369 3370 3371
static int tun_queue_resize(struct tun_struct *tun)
{
	struct net_device *dev = tun->dev;
	struct tun_file *tfile;
3372
	struct ptr_ring **rings;
J
Jason Wang 已提交
3373 3374 3375
	int n = tun->numqueues + tun->numdisabled;
	int ret, i;

3376 3377
	rings = kmalloc_array(n, sizeof(*rings), GFP_KERNEL);
	if (!rings)
J
Jason Wang 已提交
3378 3379 3380 3381
		return -ENOMEM;

	for (i = 0; i < tun->numqueues; i++) {
		tfile = rtnl_dereference(tun->tfiles[i]);
3382
		rings[i] = &tfile->tx_ring;
J
Jason Wang 已提交
3383 3384
	}
	list_for_each_entry(tfile, &tun->disabled, next)
3385
		rings[i++] = &tfile->tx_ring;
J
Jason Wang 已提交
3386

3387 3388
	ret = ptr_ring_resize_multiple(rings, n,
				       dev->tx_queue_len, GFP_KERNEL,
J
Jason Wang 已提交
3389
				       tun_ptr_free);
J
Jason Wang 已提交
3390

3391
	kfree(rings);
J
Jason Wang 已提交
3392 3393 3394 3395 3396 3397 3398 3399 3400
	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);

3401 3402 3403
	if (dev->rtnl_link_ops != &tun_link_ops)
		return NOTIFY_DONE;

J
Jason Wang 已提交
3404 3405 3406 3407 3408 3409 3410 3411 3412 3413 3414 3415 3416 3417 3418
	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,
};
3419

L
Linus Torvalds 已提交
3420 3421 3422 3423
static int __init tun_init(void)
{
	int ret = 0;

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

3426
	ret = rtnl_link_register(&tun_link_ops);
3427
	if (ret) {
3428
		pr_err("Can't register link_ops\n");
3429
		goto err_linkops;
3430 3431
	}

L
Linus Torvalds 已提交
3432
	ret = misc_register(&tun_miscdev);
3433
	if (ret) {
3434
		pr_err("Can't register misc device %d\n", TUN_MINOR);
3435 3436
		goto err_misc;
	}
J
Jason Wang 已提交
3437

3438 3439 3440 3441 3442 3443
	ret = register_netdevice_notifier(&tun_notifier_block);
	if (ret) {
		pr_err("Can't register netdevice notifier\n");
		goto err_notifier;
	}

3444
	return  0;
3445 3446 3447

err_notifier:
	misc_deregister(&tun_miscdev);
3448
err_misc:
3449 3450
	rtnl_link_unregister(&tun_link_ops);
err_linkops:
L
Linus Torvalds 已提交
3451 3452 3453 3454 3455
	return ret;
}

static void tun_cleanup(void)
{
3456
	misc_deregister(&tun_miscdev);
3457
	rtnl_link_unregister(&tun_link_ops);
J
Jason Wang 已提交
3458
	unregister_netdevice_notifier(&tun_notifier_block);
L
Linus Torvalds 已提交
3459 3460
}

3461 3462 3463 3464 3465 3466
/* 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)
{
3467
	struct tun_file *tfile;
3468 3469
	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
3470 3471
	tfile = file->private_data;
	if (!tfile)
3472
		return ERR_PTR(-EBADFD);
J
Jason Wang 已提交
3473
	return &tfile->socket;
3474 3475 3476
}
EXPORT_SYMBOL_GPL(tun_get_socket);

3477
struct ptr_ring *tun_get_tx_ring(struct file *file)
J
Jason Wang 已提交
3478 3479 3480 3481 3482 3483 3484 3485
{
	struct tun_file *tfile;

	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
	tfile = file->private_data;
	if (!tfile)
		return ERR_PTR(-EBADFD);
3486
	return &tfile->tx_ring;
J
Jason Wang 已提交
3487
}
3488
EXPORT_SYMBOL_GPL(tun_get_tx_ring);
J
Jason Wang 已提交
3489

L
Linus Torvalds 已提交
3490 3491 3492 3493 3494 3495
module_init(tun_init);
module_exit(tun_cleanup);
MODULE_DESCRIPTION(DRV_DESCRIPTION);
MODULE_AUTHOR(DRV_COPYRIGHT);
MODULE_LICENSE("GPL");
MODULE_ALIAS_MISCDEV(TUN_MINOR);
3496
MODULE_ALIAS("devname:net/tun");