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

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

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

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

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

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

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

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

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

93 94 95 96 97 98 99 100 101 102
#define tun_debug(level, tun, fmt, args...)			\
do {								\
	if (tun->debug)						\
		netdev_printk(level, tun->dev, fmt, ##args);	\
} while (0)
#define DBG1(level, fmt, args...)				\
do {								\
	if (debug == 2)						\
		printk(level fmt, ##args);			\
} while (0)
103
#else
104 105 106 107 108 109 110 111 112 113
#define tun_debug(level, tun, fmt, args...)			\
do {								\
	if (0)							\
		netdev_printk(level, tun->dev, fmt, ##args);	\
} while (0)
#define DBG1(level, fmt, args...)				\
do {								\
	if (0)							\
		printk(level fmt, ##args);			\
} while (0)
114 115
#endif

116
#define TUN_RX_PAD (NET_IP_ALIGN + NET_SKB_PAD)
117

118 119 120 121 122 123
/* TUN device flags */

/* IFF_ATTACH_QUEUE is never stored in device flags,
 * overload it to mean fasync when stored there.
 */
#define TUN_FASYNC	IFF_ATTACH_QUEUE
M
Michael S. Tsirkin 已提交
124 125
/* High bits in flags field are unused. */
#define TUN_VNET_LE     0x80000000
126
#define TUN_VNET_BE     0x40000000
127 128

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

131 132
#define GOODCOPY_LEN 128

133 134 135 136 137 138 139
#define FLT_EXACT_COUNT 8
struct tap_filter {
	unsigned int    count;    /* Number of addrs. Zero means disabled */
	u32             mask[2];  /* Mask of the hashed addrs */
	unsigned char	addr[FLT_EXACT_COUNT][ETH_ALEN];
};

140 141 142
/* MAX_TAP_QUEUES 256 is chosen to allow rx/tx queues to be equal
 * to max number of VCPUs in guest. */
#define MAX_TAP_QUEUES 256
143
#define MAX_TAP_FLOWS  4096
J
Jason Wang 已提交
144

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

147 148 149 150 151 152 153 154 155 156 157
struct tun_pcpu_stats {
	u64 rx_packets;
	u64 rx_bytes;
	u64 tx_packets;
	u64 tx_bytes;
	struct u64_stats_sync syncp;
	u32 rx_dropped;
	u32 tx_dropped;
	u32 rx_frame_errors;
};

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

J
Jason Wang 已提交
190 191 192 193 194 195
struct tun_flow_entry {
	struct hlist_node hash_link;
	struct rcu_head rcu;
	struct tun_struct *tun;

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

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

204
struct tun_prog {
205 206 207 208
	struct rcu_head rcu;
	struct bpf_prog *prog;
};

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

	struct net_device	*dev;
221
	netdev_features_t	set_features;
222
#define TUN_USER_FEATURES (NETIF_F_HW_CSUM|NETIF_F_TSO_ECN|NETIF_F_TSO| \
223
			  NETIF_F_TSO6)
224

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

251 252 253
struct veth {
	__be16 h_vlan_proto;
	__be16 h_vlan_TCI;
254
};
L
Linus Torvalds 已提交
255

256
bool tun_is_xdp_frame(void *ptr)
J
Jason Wang 已提交
257 258 259
{
	return (unsigned long)ptr & TUN_XDP_FLAG;
}
260
EXPORT_SYMBOL(tun_is_xdp_frame);
J
Jason Wang 已提交
261 262 263 264 265 266 267 268 269 270 271 272 273

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

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 312 313 314 315 316
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)
{
317
	tfile->napi_enabled = napi_en;
318 319 320 321
	if (napi_en) {
		netif_napi_add(tun->dev, &tfile->napi, tun_napi_poll,
			       NAPI_POLL_WEIGHT);
		napi_enable(&tfile->napi);
322
		mutex_init(&tfile->napi_mutex);
323 324 325 326 327
	}
}

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

static void tun_napi_del(struct tun_struct *tun, struct tun_file *tfile)
{
334
	if (tfile->napi_enabled)
335 336 337
		netif_napi_del(&tfile->napi);
}

338 339 340 341 342
static bool tun_napi_frags_enabled(const struct tun_struct *tun)
{
	return READ_ONCE(tun->flags) & IFF_NAPI_FRAGS;
}

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 386 387 388 389 390
#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 */

391 392
static inline bool tun_is_little_endian(struct tun_struct *tun)
{
393
	return tun->flags & TUN_VNET_LE ||
394
		tun_legacy_is_little_endian(tun);
395 396
}

397 398
static inline u16 tun16_to_cpu(struct tun_struct *tun, __virtio16 val)
{
399
	return __virtio16_to_cpu(tun_is_little_endian(tun), val);
400 401 402 403
}

static inline __virtio16 cpu_to_tun16(struct tun_struct *tun, u16 val)
{
404
	return __cpu_to_virtio16(tun_is_little_endian(tun), val);
405 406
}

J
Jason Wang 已提交
407 408
static inline u32 tun_hashfn(u32 rxhash)
{
L
Li RongQing 已提交
409
	return rxhash & TUN_MASK_FLOW_ENTRIES;
J
Jason Wang 已提交
410 411 412 413 414 415
}

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

416
	hlist_for_each_entry_rcu(e, head, hash_link) {
J
Jason Wang 已提交
417 418 419 420 421 422 423 424 425 426
		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)
{
427 428
	struct tun_flow_entry *e = kmalloc(sizeof(*e), GFP_ATOMIC);

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

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

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

476
		hlist_for_each_entry_safe(e, n, &tun->flows[i], hash_link) {
J
Jason Wang 已提交
477 478 479 480 481 482 483
			if (e->queue_index == queue_index)
				tun_flow_delete(tun, e);
		}
	}
	spin_unlock_bh(&tun->lock);
}

484
static void tun_flow_cleanup(struct timer_list *t)
J
Jason Wang 已提交
485
{
486
	struct tun_struct *tun = from_timer(tun, t, flow_gc_timer);
J
Jason Wang 已提交
487 488 489 490 491 492 493
	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");

494
	spin_lock(&tun->lock);
J
Jason Wang 已提交
495 496
	for (i = 0; i < TUN_NUM_FLOW_ENTRIES; i++) {
		struct tun_flow_entry *e;
497
		struct hlist_node *n;
J
Jason Wang 已提交
498

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

J
Jason Wang 已提交
502
			this_timer = e->updated + delay;
503
			if (time_before_eq(this_timer, jiffies)) {
J
Jason Wang 已提交
504
				tun_flow_delete(tun, e);
505 506 507 508
				continue;
			}
			count++;
			if (time_before(this_timer, next_timer))
J
Jason Wang 已提交
509 510 511 512 513 514
				next_timer = this_timer;
		}
	}

	if (count)
		mod_timer(&tun->flow_gc_timer, round_jiffies_up(next_timer));
515
	spin_unlock(&tun->lock);
J
Jason Wang 已提交
516 517
}

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

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

	rcu_read_lock();

	e = tun_flow_find(head, rxhash);
	if (likely(e)) {
		/* TODO: keep queueing to old queue until it's empty? */
		e->queue_index = queue_index;
		e->updated = jiffies;
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
			tun_flow_create(tun, head, rxhash, queue_index);

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

	rcu_read_unlock();
}

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

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

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

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

	return txq;
}

597 598
static u16 tun_ebpf_select_queue(struct tun_struct *tun, struct sk_buff *skb)
{
599
	struct tun_prog *prog;
600 601 602 603 604 605 606 607 608 609
	u16 ret = 0;

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

	return ret % tun->numqueues;
}

static u16 tun_select_queue(struct net_device *dev, struct sk_buff *skb,
610 611
			    struct net_device *sb_dev,
			    select_queue_fallback_t fallback)
612 613 614 615 616 617 618 619 620 621 622 623 624 625
{
	struct tun_struct *tun = netdev_priv(dev);
	u16 ret;

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

	return ret;
}

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

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

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

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

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

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

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

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

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

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

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

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

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

691 692 693 694 695
	if (tun && clean) {
		tun_napi_disable(tun, tfile);
		tun_napi_del(tun, tfile);
	}

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

	err = 0;
J
Jason Wang 已提交
819

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

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

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

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

		if (tfile->xdp_rxq.queue_index    != tfile->queue_index)
			tfile->xdp_rxq.queue_index = tfile->queue_index;
	} else {
		/* Setup XDP RX-queue info, for new tfile getting attached */
		err = xdp_rxq_info_reg(&tfile->xdp_rxq,
				       tun->dev, tfile->queue_index);
		if (err < 0)
			goto out;
851 852 853 854 855 856
		err = xdp_rxq_info_reg_mem_model(&tfile->xdp_rxq,
						 MEM_TYPE_PAGE_SHARED, NULL);
		if (err < 0) {
			xdp_rxq_info_unreg(&tfile->xdp_rxq);
			goto out;
		}
J
Jesper Dangaard Brouer 已提交
857 858 859
		err = 0;
	}

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

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

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

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

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

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

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

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

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

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

902
/* TAP filtering */
903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930
static void addr_hash_set(u32 *mask, const u8 *addr)
{
	int n = ether_crc(ETH_ALEN, addr) >> 26;
	mask[n >> 5] |= (1 << (n & 31));
}

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

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

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

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

	alen = ETH_ALEN * uf.count;
931 932 933
	addr = memdup_user(arg + sizeof(uf), alen);
	if (IS_ERR(addr))
		return PTR_ERR(addr);
934 935 936 937 938 939 940 941 942 943 944 945 946

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

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

	nexact = n;

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

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

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

	/* Return the number of exact filters */
	err = nexact;
969
free_addr:
970 971 972 973 974 975 976 977 978 979 980 981 982 983
	kfree(addr);
	return err;
}

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

	/* Exact match */
	for (i = 0; i < filter->count; i++)
984
		if (ether_addr_equal(eh->h_dest, filter->addr[i]))
985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005
			return 1;

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

	return 0;
}

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

	return run_filter(filter, skb);
}

L
Linus Torvalds 已提交
1006 1007
/* Network device part of the driver */

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

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

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

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

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

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

L
Linus Torvalds 已提交
1031 1032 1033 1034 1035 1036
	return 0;
}

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

/* Net device start xmit */
1042
static void tun_automq_xmit(struct tun_struct *tun, struct sk_buff *skb)
L
Linus Torvalds 已提交
1043
{
1044
#ifdef CONFIG_RPS
1045
	if (tun->numqueues == 1 && static_key_false(&rps_needed)) {
1046 1047 1048 1049 1050
		/* 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 已提交
1051
		rxhash = __skb_get_hash_symmetric(skb);
1052 1053 1054 1055 1056 1057 1058 1059
		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);
		}
	}
1060
#endif
1061 1062
}

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

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

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

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

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

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

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

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

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

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

1114
	skb_tx_timestamp(skb);
1115

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

1121 1122
	nf_reset(skb);

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

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

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

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

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

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

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

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

1181 1182 1183
		if (tun_napi_frags_enabled(tun))
			return;

1184 1185 1186
		rcu_read_lock();
		for (i = 0; i < tun->numqueues; i++) {
			tfile = rcu_dereference(tun->tfiles[i]);
1187 1188
			if (tfile->napi_enabled)
				napi_schedule(&tfile->napi);
1189 1190 1191
		}
		rcu_read_unlock();
	}
1192 1193 1194
	return;
}
#endif
1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205

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

1206
static void
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 1236 1237 1238 1239 1240 1241
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 已提交
1242 1243 1244 1245
static int tun_xdp_set(struct net_device *dev, struct bpf_prog *prog,
		       struct netlink_ext_ack *extack)
{
	struct tun_struct *tun = netdev_priv(dev);
J
Jason Wang 已提交
1246
	struct tun_file *tfile;
J
Jason Wang 已提交
1247
	struct bpf_prog *old_prog;
J
Jason Wang 已提交
1248
	int i;
J
Jason Wang 已提交
1249 1250 1251 1252 1253 1254

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

J
Jason Wang 已提交
1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268
	for (i = 0; i < tun->numqueues; i++) {
		tfile = rtnl_dereference(tun->tfiles[i]);
		if (prog)
			sock_set_flag(&tfile->sk, SOCK_XDP);
		else
			sock_reset_flag(&tfile->sk, SOCK_XDP);
	}
	list_for_each_entry(tfile, &tun->disabled, next) {
		if (prog)
			sock_set_flag(&tfile->sk, SOCK_XDP);
		else
			sock_reset_flag(&tfile->sk, SOCK_XDP);
	}

J
Jason Wang 已提交
1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283
	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;
}

1284
static int tun_xdp(struct net_device *dev, struct netdev_bpf *xdp)
J
Jason Wang 已提交
1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296
{
	switch (xdp->command) {
	case XDP_SETUP_PROG:
		return tun_xdp_set(dev, xdp->prog, xdp->extack);
	case XDP_QUERY_PROG:
		xdp->prog_id = tun_xdp_query(dev);
		return 0;
	default:
		return -EINVAL;
	}
}

S
Stephen Hemminger 已提交
1297
static const struct net_device_ops tun_netdev_ops = {
1298
	.ndo_uninit		= tun_net_uninit,
S
Stephen Hemminger 已提交
1299 1300
	.ndo_open		= tun_net_open,
	.ndo_stop		= tun_net_close,
1301
	.ndo_start_xmit		= tun_net_xmit,
1302
	.ndo_fix_features	= tun_net_fix_features,
J
Jason Wang 已提交
1303
	.ndo_select_queue	= tun_select_queue,
1304 1305 1306
#ifdef CONFIG_NET_POLL_CONTROLLER
	.ndo_poll_controller	= tun_poll_controller,
#endif
1307
	.ndo_set_rx_headroom	= tun_set_headroom,
1308
	.ndo_get_stats64	= tun_net_get_stats64,
S
Stephen Hemminger 已提交
1309 1310
};

1311 1312 1313 1314 1315 1316 1317 1318
static void __tun_xdp_flush_tfile(struct tun_file *tfile)
{
	/* Notify and wake up reader process */
	if (tfile->flags & TUN_FASYNC)
		kill_fasync(&tfile->fasync, SIGIO, POLL_IN);
	tfile->socket.sk->sk_data_ready(tfile->socket.sk);
}

1319 1320
static int tun_xdp_xmit(struct net_device *dev, int n,
			struct xdp_frame **frames, u32 flags)
J
Jason Wang 已提交
1321 1322 1323 1324
{
	struct tun_struct *tun = netdev_priv(dev);
	struct tun_file *tfile;
	u32 numqueues;
1325 1326 1327
	int drops = 0;
	int cnt = n;
	int i;
J
Jason Wang 已提交
1328

1329
	if (unlikely(flags & ~XDP_XMIT_FLAGS_MASK))
1330 1331
		return -EINVAL;

J
Jason Wang 已提交
1332 1333 1334 1335
	rcu_read_lock();

	numqueues = READ_ONCE(tun->numqueues);
	if (!numqueues) {
1336 1337
		rcu_read_unlock();
		return -ENXIO; /* Caller will free/return all frames */
J
Jason Wang 已提交
1338 1339 1340 1341
	}

	tfile = rcu_dereference(tun->tfiles[smp_processor_id() %
					    numqueues]);
1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355

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

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

1359 1360 1361
	if (flags & XDP_XMIT_FLUSH)
		__tun_xdp_flush_tfile(tfile);

J
Jason Wang 已提交
1362
	rcu_read_unlock();
1363
	return cnt - drops;
J
Jason Wang 已提交
1364 1365
}

1366 1367 1368 1369 1370 1371 1372
static int tun_xdp_tx(struct net_device *dev, struct xdp_buff *xdp)
{
	struct xdp_frame *frame = convert_to_xdp_frame(xdp);

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

1373
	return tun_xdp_xmit(dev, 1, &frame, XDP_XMIT_FLUSH);
J
Jason Wang 已提交
1374 1375
}

S
Stephen Hemminger 已提交
1376
static const struct net_device_ops tap_netdev_ops = {
1377
	.ndo_uninit		= tun_net_uninit,
S
Stephen Hemminger 已提交
1378 1379
	.ndo_open		= tun_net_open,
	.ndo_stop		= tun_net_close,
1380
	.ndo_start_xmit		= tun_net_xmit,
1381
	.ndo_fix_features	= tun_net_fix_features,
1382
	.ndo_set_rx_mode	= tun_net_mclist,
S
Stephen Hemminger 已提交
1383 1384
	.ndo_set_mac_address	= eth_mac_addr,
	.ndo_validate_addr	= eth_validate_addr,
J
Jason Wang 已提交
1385
	.ndo_select_queue	= tun_select_queue,
1386 1387 1388
#ifdef CONFIG_NET_POLL_CONTROLLER
	.ndo_poll_controller	= tun_poll_controller,
#endif
1389
	.ndo_features_check	= passthru_features_check,
1390
	.ndo_set_rx_headroom	= tun_set_headroom,
1391
	.ndo_get_stats64	= tun_net_get_stats64,
1392
	.ndo_bpf		= tun_xdp,
J
Jason Wang 已提交
1393
	.ndo_xdp_xmit		= tun_xdp_xmit,
S
Stephen Hemminger 已提交
1394 1395
};

1396
static void tun_flow_init(struct tun_struct *tun)
J
Jason Wang 已提交
1397 1398 1399 1400 1401 1402 1403
{
	int i;

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

	tun->ageing_time = TUN_FLOW_EXPIRE;
1404 1405 1406
	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 已提交
1407 1408 1409 1410 1411 1412 1413 1414
}

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

1415 1416 1417
#define MIN_MTU 68
#define MAX_MTU 65535

L
Linus Torvalds 已提交
1418 1419 1420 1421
/* Initialize net device. */
static void tun_net_init(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
1422

L
Linus Torvalds 已提交
1423
	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
1424
	case IFF_TUN:
S
Stephen Hemminger 已提交
1425 1426
		dev->netdev_ops = &tun_netdev_ops;

L
Linus Torvalds 已提交
1427 1428 1429 1430 1431 1432
		/* Point-to-Point TUN Device */
		dev->hard_header_len = 0;
		dev->addr_len = 0;
		dev->mtu = 1500;

		/* Zero header length */
1433
		dev->type = ARPHRD_NONE;
L
Linus Torvalds 已提交
1434 1435 1436
		dev->flags = IFF_POINTOPOINT | IFF_NOARP | IFF_MULTICAST;
		break;

M
Michael S. Tsirkin 已提交
1437
	case IFF_TAP:
K
Kusanagi Kouichi 已提交
1438
		dev->netdev_ops = &tap_netdev_ops;
L
Linus Torvalds 已提交
1439 1440
		/* Ethernet TAP Device */
		ether_setup(dev);
1441
		dev->priv_flags &= ~IFF_TX_SKB_SHARING;
1442
		dev->priv_flags |= IFF_LIVE_ADDR_CHANGE;
1443

1444
		eth_hw_addr_random(dev);
1445

L
Linus Torvalds 已提交
1446 1447
		break;
	}
1448 1449 1450

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

1453 1454 1455 1456 1457 1458 1459
static bool tun_sock_writeable(struct tun_struct *tun, struct tun_file *tfile)
{
	struct sock *sk = tfile->socket.sk;

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

L
Linus Torvalds 已提交
1460 1461 1462
/* Character device part */

/* Poll */
1463
static __poll_t tun_chr_poll(struct file *file, poll_table *wait)
1464
{
1465
	struct tun_file *tfile = file->private_data;
1466
	struct tun_struct *tun = tun_get(tfile);
1467
	struct sock *sk;
1468
	__poll_t mask = 0;
L
Linus Torvalds 已提交
1469 1470

	if (!tun)
1471
		return EPOLLERR;
L
Linus Torvalds 已提交
1472

J
Jason Wang 已提交
1473
	sk = tfile->socket.sk;
1474

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

1477
	poll_wait(file, sk_sleep(sk), wait);
1478

1479
	if (!ptr_ring_empty(&tfile->tx_ring))
1480
		mask |= EPOLLIN | EPOLLRDNORM;
L
Linus Torvalds 已提交
1481

1482 1483 1484 1485 1486 1487 1488 1489
	/* Make sure SOCKWQ_ASYNC_NOSPACE is set if not writable to
	 * guarantee EPOLLOUT to be raised by either here or
	 * tun_sock_write_space(). Then process could get notification
	 * after it writes to a down device and meets -EIO.
	 */
	if (tun_sock_writeable(tun, tfile) ||
	    (!test_and_set_bit(SOCKWQ_ASYNC_NOSPACE, &sk->sk_socket->flags) &&
	     tun_sock_writeable(tun, tfile)))
1490
		mask |= EPOLLOUT | EPOLLWRNORM;
1491

1492
	if (tun->dev->reg_state != NETREG_REGISTERED)
1493
		mask = EPOLLERR;
1494

E
Eric W. Biederman 已提交
1495
	tun_put(tun);
L
Linus Torvalds 已提交
1496 1497 1498
	return mask;
}

1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526
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++) {
1527
		struct page_frag *pfrag = &current->task_frag;
1528 1529 1530 1531 1532 1533 1534
		size_t fragsz = it->iov[i].iov_len;

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

1535
		if (!skb_page_frag_refill(fragsz, pfrag, GFP_KERNEL)) {
1536 1537 1538 1539
			err = -ENOMEM;
			goto free;
		}

1540 1541 1542 1543
		skb_fill_page_desc(skb, i - 1, pfrag->page,
				   pfrag->offset, fragsz);
		page_ref_inc(pfrag->page);
		pfrag->offset += fragsz;
1544 1545 1546 1547 1548 1549 1550 1551 1552
	}

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

1553 1554
/* 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 已提交
1555
static struct sk_buff *tun_alloc_skb(struct tun_file *tfile,
1556 1557
				     size_t prepad, size_t len,
				     size_t linear, int noblock)
1558
{
J
Jason Wang 已提交
1559
	struct sock *sk = tfile->socket.sk;
1560
	struct sk_buff *skb;
1561
	int err;
1562 1563

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

1567
	skb = sock_alloc_send_pskb(sk, prepad + linear, len - linear, noblock,
1568
				   &err, 0);
1569
	if (!skb)
1570
		return ERR_PTR(err);
1571 1572 1573

	skb_reserve(skb, prepad);
	skb_put(skb, linear);
1574 1575
	skb->data_len = len - linear;
	skb->len += len - linear;
1576 1577 1578 1579

	return skb;
}

J
Jason Wang 已提交
1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615
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();
	}
}

1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637
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;
}

1638
static struct sk_buff *__tun_build_skb(struct page_frag *alloc_frag, char *buf,
J
Jason Wang 已提交
1639
				       int buflen, int len, int pad)
1640 1641 1642 1643 1644 1645
{
	struct sk_buff *skb = build_skb(buf, buflen);

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

J
Jason Wang 已提交
1646
	skb_reserve(skb, pad);
1647 1648 1649 1650 1651 1652 1653 1654
	skb_put(skb, len);

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

	return skb;
}

J
Jason Wang 已提交
1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687
static int tun_xdp_act(struct tun_struct *tun, struct bpf_prog *xdp_prog,
		       struct xdp_buff *xdp, u32 act)
{
	int err;

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

	return act;
}

J
Jason Wang 已提交
1688 1689
static struct sk_buff *tun_build_skb(struct tun_struct *tun,
				     struct tun_file *tfile,
1690
				     struct iov_iter *from,
J
Jason Wang 已提交
1691
				     struct virtio_net_hdr *hdr,
J
Jason Wang 已提交
1692
				     int len, int *skb_xdp)
1693
{
1694
	struct page_frag *alloc_frag = &current->task_frag;
J
Jason Wang 已提交
1695
	struct bpf_prog *xdp_prog;
1696
	int buflen = SKB_DATA_ALIGN(sizeof(struct skb_shared_info));
1697 1698
	char *buf;
	size_t copied;
J
Jason Wang 已提交
1699 1700
	int pad = TUN_RX_PAD;
	int err = 0;
1701 1702 1703 1704

	rcu_read_lock();
	xdp_prog = rcu_dereference(tun->xdp_prog);
	if (xdp_prog)
1705
		pad += XDP_PACKET_HEADROOM;
1706 1707
	buflen += SKB_DATA_ALIGN(len + pad);
	rcu_read_unlock();
1708

1709
	alloc_frag->offset = ALIGN((u64)alloc_frag->offset, SMP_CACHE_BYTES);
1710 1711 1712 1713 1714
	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,
1715
				     alloc_frag->offset + pad,
1716 1717 1718 1719
				     len, from);
	if (copied != len)
		return ERR_PTR(-EFAULT);

1720 1721 1722 1723
	/* 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.
	 */
1724
	if (hdr->gso_type || !xdp_prog) {
J
Jason Wang 已提交
1725
		*skb_xdp = 1;
J
Jason Wang 已提交
1726
		return __tun_build_skb(alloc_frag, buf, buflen, len, pad);
1727 1728 1729
	}

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

1731
	local_bh_disable();
J
Jason Wang 已提交
1732 1733
	rcu_read_lock();
	xdp_prog = rcu_dereference(tun->xdp_prog);
J
Jason Wang 已提交
1734
	if (xdp_prog) {
J
Jason Wang 已提交
1735 1736 1737 1738
		struct xdp_buff xdp;
		u32 act;

		xdp.data_hard_start = buf;
1739
		xdp.data = buf + pad;
1740
		xdp_set_data_meta_invalid(&xdp);
J
Jason Wang 已提交
1741
		xdp.data_end = xdp.data + len;
J
Jesper Dangaard Brouer 已提交
1742
		xdp.rxq = &tfile->xdp_rxq;
J
Jason Wang 已提交
1743

J
Jason Wang 已提交
1744 1745
		act = bpf_prog_run_xdp(xdp_prog, &xdp);
		if (act == XDP_REDIRECT || act == XDP_TX) {
J
Jason Wang 已提交
1746 1747
			get_page(alloc_frag->page);
			alloc_frag->offset += buflen;
J
Jason Wang 已提交
1748
		}
J
Jason Wang 已提交
1749 1750 1751 1752 1753 1754 1755 1756
		err = tun_xdp_act(tun, xdp_prog, &xdp, act);
		if (err < 0)
			goto err_xdp;
		if (err != XDP_PASS)
			goto out;

		pad = xdp.data - xdp.data_hard_start;
		len = xdp.data_end - xdp.data;
J
Jason Wang 已提交
1757
	}
1758 1759
	rcu_read_unlock();
	local_bh_enable();
J
Jason Wang 已提交
1760

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

J
Jason Wang 已提交
1763
err_xdp:
J
Jason Wang 已提交
1764
	put_page(alloc_frag->page);
1765
out:
J
Jason Wang 已提交
1766
	rcu_read_unlock();
1767
	local_bh_enable();
J
Jason Wang 已提交
1768
	return NULL;
1769 1770
}

L
Linus Torvalds 已提交
1771
/* Get packet from user space buffer */
J
Jason Wang 已提交
1772
static ssize_t tun_get_user(struct tun_struct *tun, struct tun_file *tfile,
1773
			    void *msg_control, struct iov_iter *from,
J
Jason Wang 已提交
1774
			    int noblock, bool more)
L
Linus Torvalds 已提交
1775
{
1776
	struct tun_pi pi = { 0, cpu_to_be16(ETH_P_IP) };
L
Linus Torvalds 已提交
1777
	struct sk_buff *skb;
1778
	size_t total_len = iov_iter_count(from);
1779
	size_t len = total_len, align = tun->align, linear;
1780
	struct virtio_net_hdr gso = { 0 };
1781
	struct tun_pcpu_stats *stats;
1782
	int good_linear;
1783 1784 1785
	int copylen;
	bool zerocopy = false;
	int err;
1786
	u32 rxhash = 0;
J
Jason Wang 已提交
1787
	int skb_xdp = 1;
1788
	bool frags = tun_napi_frags_enabled(tun);
L
Linus Torvalds 已提交
1789

E
Eric Dumazet 已提交
1790 1791 1792
	if (!(tun->dev->flags & IFF_UP))
		return -EIO;

M
Michael S. Tsirkin 已提交
1793
	if (!(tun->flags & IFF_NO_PI)) {
1794
		if (len < sizeof(pi))
L
Linus Torvalds 已提交
1795
			return -EINVAL;
1796
		len -= sizeof(pi);
L
Linus Torvalds 已提交
1797

1798
		if (!copy_from_iter_full(&pi, sizeof(pi), from))
L
Linus Torvalds 已提交
1799 1800 1801
			return -EFAULT;
	}

M
Michael S. Tsirkin 已提交
1802
	if (tun->flags & IFF_VNET_HDR) {
W
Willem de Bruijn 已提交
1803 1804 1805
		int vnet_hdr_sz = READ_ONCE(tun->vnet_hdr_sz);

		if (len < vnet_hdr_sz)
1806
			return -EINVAL;
W
Willem de Bruijn 已提交
1807
		len -= vnet_hdr_sz;
1808

1809
		if (!copy_from_iter_full(&gso, sizeof(gso), from))
1810 1811
			return -EFAULT;

1812
		if ((gso.flags & VIRTIO_NET_HDR_F_NEEDS_CSUM) &&
1813 1814
		    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);
1815

1816
		if (tun16_to_cpu(tun, gso.hdr_len) > len)
1817
			return -EINVAL;
W
Willem de Bruijn 已提交
1818
		iov_iter_advance(from, vnet_hdr_sz - sizeof(gso));
1819 1820
	}

M
Michael S. Tsirkin 已提交
1821
	if ((tun->flags & TUN_TYPE_MASK) == IFF_TAP) {
1822
		align += NET_IP_ALIGN;
H
Herbert Xu 已提交
1823
		if (unlikely(len < ETH_HLEN ||
1824
			     (gso.hdr_len && tun16_to_cpu(tun, gso.hdr_len) < ETH_HLEN)))
1825 1826
			return -EINVAL;
	}
1827

1828 1829
	good_linear = SKB_MAX_HEAD(align);

1830
	if (msg_control) {
1831 1832
		struct iov_iter i = *from;

1833 1834
		/* There are 256 bytes to be copied in skb, so there is
		 * enough room for skb expand head in case it is used.
1835 1836
		 * The rest of the buffer is mapped from userspace.
		 */
1837
		copylen = gso.hdr_len ? tun16_to_cpu(tun, gso.hdr_len) : GOODCOPY_LEN;
1838 1839
		if (copylen > good_linear)
			copylen = good_linear;
1840
		linear = copylen;
1841 1842
		iov_iter_advance(&i, copylen);
		if (iov_iter_npages(&i, INT_MAX) <= MAX_SKB_FRAGS)
1843 1844 1845
			zerocopy = true;
	}

1846
	if (!frags && tun_can_build_skb(tun, tfile, len, noblock, zerocopy)) {
J
Jason Wang 已提交
1847 1848 1849 1850 1851
		/* 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);
1852
		if (IS_ERR(skb)) {
1853
			this_cpu_inc(tun->pcpu_stats->rx_dropped);
1854 1855
			return PTR_ERR(skb);
		}
J
Jason Wang 已提交
1856 1857
		if (!skb)
			return total_len;
1858 1859 1860 1861 1862 1863 1864 1865
	} 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 已提交
1866

1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879
		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);
		}

1880 1881 1882
		if (IS_ERR(skb)) {
			if (PTR_ERR(skb) != -EAGAIN)
				this_cpu_inc(tun->pcpu_stats->rx_dropped);
1883 1884
			if (frags)
				mutex_unlock(&tfile->napi_mutex);
1885 1886
			return PTR_ERR(skb);
		}
1887

1888 1889 1890 1891 1892 1893 1894 1895
		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);
1896 1897 1898 1899 1900
			if (frags) {
				tfile->napi.skb = NULL;
				mutex_unlock(&tfile->napi_mutex);
			}

1901 1902
			return -EFAULT;
		}
D
Dave Jones 已提交
1903
	}
L
Linus Torvalds 已提交
1904

1905
	if (virtio_net_hdr_to_skb(skb, &gso, tun_is_little_endian(tun))) {
1906 1907
		this_cpu_inc(tun->pcpu_stats->rx_frame_errors);
		kfree_skb(skb);
1908 1909 1910 1911 1912
		if (frags) {
			tfile->napi.skb = NULL;
			mutex_unlock(&tfile->napi_mutex);
		}

1913 1914 1915
		return -EINVAL;
	}

L
Linus Torvalds 已提交
1916
	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
1917 1918
	case IFF_TUN:
		if (tun->flags & IFF_NO_PI) {
1919 1920 1921 1922
			u8 ip_version = skb->len ? (skb->data[0] >> 4) : 0;

			switch (ip_version) {
			case 4:
1923 1924
				pi.proto = htons(ETH_P_IP);
				break;
1925
			case 6:
1926 1927 1928
				pi.proto = htons(ETH_P_IPV6);
				break;
			default:
1929
				this_cpu_inc(tun->pcpu_stats->rx_dropped);
1930 1931 1932 1933 1934
				kfree_skb(skb);
				return -EINVAL;
			}
		}

1935
		skb_reset_mac_header(skb);
L
Linus Torvalds 已提交
1936
		skb->protocol = pi.proto;
1937
		skb->dev = tun->dev;
L
Linus Torvalds 已提交
1938
		break;
M
Michael S. Tsirkin 已提交
1939
	case IFF_TAP:
1940 1941
		if (!frags)
			skb->protocol = eth_type_trans(skb, tun->dev);
L
Linus Torvalds 已提交
1942
		break;
1943
	}
L
Linus Torvalds 已提交
1944

1945 1946 1947 1948
	/* 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;
1949
		skb_shinfo(skb)->tx_flags |= SKBTX_SHARED_FRAG;
1950 1951 1952
	} else if (msg_control) {
		struct ubuf_info *uarg = msg_control;
		uarg->callback(uarg, false);
1953 1954
	}

1955
	skb_reset_network_header(skb);
1956
	skb_probe_transport_header(skb, 0);
1957

J
Jason Wang 已提交
1958
	if (skb_xdp) {
J
Jason Wang 已提交
1959 1960 1961
		struct bpf_prog *xdp_prog;
		int ret;

1962
		local_bh_disable();
J
Jason Wang 已提交
1963 1964 1965 1966 1967 1968
		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();
1969
				local_bh_enable();
J
Jason Wang 已提交
1970 1971 1972 1973
				return total_len;
			}
		}
		rcu_read_unlock();
1974
		local_bh_enable();
J
Jason Wang 已提交
1975 1976
	}

1977 1978 1979 1980 1981 1982
	/* Compute the costly rx hash only if needed for flow updates.
	 * We may get a very small possibility of OOO during switching, not
	 * worth to optimize.
	 */
	if (!rcu_access_pointer(tun->steering_prog) && tun->numqueues > 1 &&
	    !tfile->detached)
1983
		rxhash = __skb_get_hash_symmetric(skb);
1984

1985 1986 1987 1988
	if (frags) {
		/* Exercise flow dissector code path. */
		u32 headlen = eth_get_headlen(skb->data, skb_headlen(skb));

1989
		if (unlikely(headlen > skb_headlen(skb))) {
1990 1991 1992 1993 1994 1995 1996 1997 1998 1999 2000
			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);
2001
	} else if (tfile->napi_enabled) {
2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018
		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);
	}
2019

2020 2021 2022 2023 2024 2025
	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 已提交
2026

2027 2028 2029
	if (rxhash)
		tun_flow_update(tun, rxhash, tfile);

2030
	return total_len;
2031
}
L
Linus Torvalds 已提交
2032

2033
static ssize_t tun_chr_write_iter(struct kiocb *iocb, struct iov_iter *from)
L
Linus Torvalds 已提交
2034
{
2035
	struct file *file = iocb->ki_filp;
J
Jason Wang 已提交
2036
	struct tun_file *tfile = file->private_data;
2037
	struct tun_struct *tun = tun_get(tfile);
E
Eric W. Biederman 已提交
2038
	ssize_t result;
L
Linus Torvalds 已提交
2039 2040 2041 2042

	if (!tun)
		return -EBADFD;

J
Jason Wang 已提交
2043 2044
	result = tun_get_user(tun, tfile, NULL, from,
			      file->f_flags & O_NONBLOCK, false);
E
Eric W. Biederman 已提交
2045 2046 2047

	tun_put(tun);
	return result;
L
Linus Torvalds 已提交
2048 2049
}

J
Jason Wang 已提交
2050 2051
static ssize_t tun_put_user_xdp(struct tun_struct *tun,
				struct tun_file *tfile,
2052
				struct xdp_frame *xdp_frame,
J
Jason Wang 已提交
2053 2054 2055
				struct iov_iter *iter)
{
	int vnet_hdr_sz = 0;
2056
	size_t size = xdp_frame->len;
J
Jason Wang 已提交
2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068 2069 2070 2071
	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));
	}

2072
	ret = copy_to_iter(xdp_frame->data, size, iter) + vnet_hdr_sz;
J
Jason Wang 已提交
2073 2074 2075 2076 2077 2078 2079 2080 2081 2082 2083

	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 已提交
2084
/* Put packet to the user space buffer */
2085
static ssize_t tun_put_user(struct tun_struct *tun,
J
Jason Wang 已提交
2086
			    struct tun_file *tfile,
2087
			    struct sk_buff *skb,
H
Herbert Xu 已提交
2088
			    struct iov_iter *iter)
L
Linus Torvalds 已提交
2089 2090
{
	struct tun_pi pi = { 0, skb->protocol };
2091
	struct tun_pcpu_stats *stats;
H
Herbert Xu 已提交
2092
	ssize_t total;
2093
	int vlan_offset = 0;
2094
	int vlan_hlen = 0;
H
Herbert Xu 已提交
2095
	int vnet_hdr_sz = 0;
2096

2097
	if (skb_vlan_tag_present(skb))
2098
		vlan_hlen = VLAN_HLEN;
L
Linus Torvalds 已提交
2099

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

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

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

H
Herbert Xu 已提交
2109 2110
		total += sizeof(pi);
		if (iov_iter_count(iter) < total) {
L
Linus Torvalds 已提交
2111 2112 2113
			/* Packet will be striped */
			pi.flags |= TUN_PKT_STRIP;
		}
2114

H
Herbert Xu 已提交
2115
		if (copy_to_iter(&pi, sizeof(pi), iter) != sizeof(pi))
L
Linus Torvalds 已提交
2116
			return -EFAULT;
2117
	}
L
Linus Torvalds 已提交
2118

H
Herbert Xu 已提交
2119
	if (vnet_hdr_sz) {
2120
		struct virtio_net_hdr gso;
2121

H
Herbert Xu 已提交
2122
		if (iov_iter_count(iter) < vnet_hdr_sz)
2123 2124
			return -EINVAL;

2125
		if (virtio_net_hdr_from_skb(skb, &gso,
2126 2127
					    tun_is_little_endian(tun), true,
					    vlan_hlen)) {
2128
			struct skb_shared_info *sinfo = skb_shinfo(skb);
2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139
			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;
		}
2140

H
Herbert Xu 已提交
2141
		if (copy_to_iter(&gso, sizeof(gso), iter) != sizeof(gso))
2142
			return -EFAULT;
2143 2144

		iov_iter_advance(iter, vnet_hdr_sz - sizeof(gso));
2145 2146
	}

2147
	if (vlan_hlen) {
H
Herbert Xu 已提交
2148
		int ret;
2149
		struct veth veth;
J
Jason Wang 已提交
2150 2151

		veth.h_vlan_proto = skb->vlan_proto;
2152
		veth.h_vlan_TCI = htons(skb_vlan_tag_get(skb));
J
Jason Wang 已提交
2153 2154 2155

		vlan_offset = offsetof(struct vlan_ethhdr, h_vlan_proto);

H
Herbert Xu 已提交
2156 2157
		ret = skb_copy_datagram_iter(skb, 0, iter, vlan_offset);
		if (ret || !iov_iter_count(iter))
J
Jason Wang 已提交
2158 2159
			goto done;

H
Herbert Xu 已提交
2160 2161
		ret = copy_to_iter(&veth, sizeof(veth), iter);
		if (ret != sizeof(veth) || !iov_iter_count(iter))
J
Jason Wang 已提交
2162 2163
			goto done;
	}
L
Linus Torvalds 已提交
2164

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

J
Jason Wang 已提交
2167
done:
2168 2169 2170 2171 2172 2173 2174
	/* 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 已提交
2175 2176 2177 2178

	return total;
}

J
Jason Wang 已提交
2179
static void *tun_ring_recv(struct tun_file *tfile, int noblock, int *err)
J
Jason Wang 已提交
2180 2181
{
	DECLARE_WAITQUEUE(wait, current);
J
Jason Wang 已提交
2182
	void *ptr = NULL;
J
Jason Wang 已提交
2183
	int error = 0;
J
Jason Wang 已提交
2184

J
Jason Wang 已提交
2185 2186
	ptr = ptr_ring_consume(&tfile->tx_ring);
	if (ptr)
J
Jason Wang 已提交
2187 2188
		goto out;
	if (noblock) {
J
Jason Wang 已提交
2189
		error = -EAGAIN;
J
Jason Wang 已提交
2190 2191 2192 2193 2194 2195 2196
		goto out;
	}

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

	while (1) {
J
Jason Wang 已提交
2197 2198
		ptr = ptr_ring_consume(&tfile->tx_ring);
		if (ptr)
J
Jason Wang 已提交
2199 2200
			break;
		if (signal_pending(current)) {
J
Jason Wang 已提交
2201
			error = -ERESTARTSYS;
J
Jason Wang 已提交
2202 2203 2204
			break;
		}
		if (tfile->socket.sk->sk_shutdown & RCV_SHUTDOWN) {
J
Jason Wang 已提交
2205
			error = -EFAULT;
J
Jason Wang 已提交
2206 2207 2208 2209 2210 2211 2212 2213 2214 2215
			break;
		}

		schedule();
	}

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

out:
J
Jason Wang 已提交
2216
	*err = error;
J
Jason Wang 已提交
2217
	return ptr;
J
Jason Wang 已提交
2218 2219
}

J
Jason Wang 已提交
2220
static ssize_t tun_do_read(struct tun_struct *tun, struct tun_file *tfile,
2221
			   struct iov_iter *to,
J
Jason Wang 已提交
2222
			   int noblock, void *ptr)
L
Linus Torvalds 已提交
2223
{
2224
	ssize_t ret;
J
Jason Wang 已提交
2225
	int err;
L
Linus Torvalds 已提交
2226

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

W
Wei Xu 已提交
2229
	if (!iov_iter_count(to)) {
J
Jason Wang 已提交
2230
		tun_ptr_free(ptr);
2231
		return 0;
W
Wei Xu 已提交
2232
	}
L
Linus Torvalds 已提交
2233

J
Jason Wang 已提交
2234
	if (!ptr) {
2235
		/* Read frames from ring */
J
Jason Wang 已提交
2236 2237
		ptr = tun_ring_recv(tfile, noblock, &err);
		if (!ptr)
2238 2239
			return err;
	}
H
Herbert Xu 已提交
2240

2241 2242
	if (tun_is_xdp_frame(ptr)) {
		struct xdp_frame *xdpf = tun_ptr_to_xdp(ptr);
J
Jason Wang 已提交
2243

2244
		ret = tun_put_user_xdp(tun, tfile, xdpf, to);
2245
		xdp_return_frame(xdpf);
J
Jason Wang 已提交
2246 2247 2248 2249 2250 2251 2252 2253 2254
	} 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 已提交
2255

2256 2257 2258
	return ret;
}

2259
static ssize_t tun_chr_read_iter(struct kiocb *iocb, struct iov_iter *to)
2260 2261 2262
{
	struct file *file = iocb->ki_filp;
	struct tun_file *tfile = file->private_data;
2263
	struct tun_struct *tun = tun_get(tfile);
2264
	ssize_t len = iov_iter_count(to), ret;
2265 2266 2267

	if (!tun)
		return -EBADFD;
2268
	ret = tun_do_read(tun, tfile, to, file->f_flags & O_NONBLOCK, NULL);
2269
	ret = min_t(ssize_t, ret, len);
Z
Zhi Yong Wu 已提交
2270 2271
	if (ret > 0)
		iocb->ki_pos = ret;
E
Eric W. Biederman 已提交
2272
	tun_put(tun);
L
Linus Torvalds 已提交
2273 2274 2275
	return ret;
}

2276
static void tun_prog_free(struct rcu_head *rcu)
2277
{
2278
	struct tun_prog *prog = container_of(rcu, struct tun_prog, rcu);
2279 2280 2281 2282 2283

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

J
Jason Wang 已提交
2284 2285
static int __tun_set_ebpf(struct tun_struct *tun,
			  struct tun_prog __rcu **prog_p,
2286
			  struct bpf_prog *prog)
2287
{
2288
	struct tun_prog *old, *new = NULL;
2289 2290 2291 2292 2293 2294 2295 2296

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

2297
	spin_lock_bh(&tun->lock);
2298
	old = rcu_dereference_protected(*prog_p,
2299
					lockdep_is_held(&tun->lock));
2300
	rcu_assign_pointer(*prog_p, new);
2301
	spin_unlock_bh(&tun->lock);
2302 2303

	if (old)
2304
		call_rcu(&old->rcu, tun_prog_free);
2305 2306 2307 2308

	return 0;
}

J
Jason Wang 已提交
2309 2310 2311 2312
static void tun_free_netdev(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

J
Jason Wang 已提交
2313
	BUG_ON(!(list_empty(&tun->disabled)));
2314
	free_percpu(tun->pcpu_stats);
J
Jason Wang 已提交
2315
	tun_flow_uninit(tun);
2316
	security_tun_dev_free_security(tun->security);
2317
	__tun_set_ebpf(tun, &tun->steering_prog, NULL);
2318
	__tun_set_ebpf(tun, &tun->filter_prog, NULL);
J
Jason Wang 已提交
2319 2320
}

L
Linus Torvalds 已提交
2321 2322 2323 2324
static void tun_setup(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

2325 2326
	tun->owner = INVALID_UID;
	tun->group = INVALID_GID;
2327
	tun_default_link_ksettings(dev, &tun->link_ksettings);
L
Linus Torvalds 已提交
2328 2329

	dev->ethtool_ops = &tun_ethtool_ops;
2330 2331
	dev->needs_free_netdev = true;
	dev->priv_destructor = tun_free_netdev;
J
Jason Wang 已提交
2332 2333
	/* We prefer our own queue length */
	dev->tx_queue_len = TUN_READQ_SIZE;
L
Linus Torvalds 已提交
2334 2335
}

2336 2337 2338
/* Trivial set of netlink ops to allow deleting tun or tap
 * device with netlink.
 */
2339 2340
static int tun_validate(struct nlattr *tb[], struct nlattr *data[],
			struct netlink_ext_ack *extack)
2341 2342 2343 2344
{
	return -EINVAL;
}

2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398
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;
}

2399 2400 2401 2402 2403
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,
2404 2405
	.get_size       = tun_get_size,
	.fill_info      = tun_fill_info,
2406 2407
};

2408 2409
static void tun_sock_write_space(struct sock *sk)
{
J
Jason Wang 已提交
2410
	struct tun_file *tfile;
2411
	wait_queue_head_t *wqueue;
2412 2413 2414 2415

	if (!sock_writeable(sk))
		return;

2416
	if (!test_and_clear_bit(SOCKWQ_ASYNC_NOSPACE, &sk->sk_socket->flags))
2417 2418
		return;

2419 2420
	wqueue = sk_sleep(sk);
	if (wqueue && waitqueue_active(wqueue))
2421 2422
		wake_up_interruptible_sync_poll(wqueue, EPOLLOUT |
						EPOLLWRNORM | EPOLLWRBAND);
H
Herbert Xu 已提交
2423

J
Jason Wang 已提交
2424 2425
	tfile = container_of(sk, struct tun_file, sk);
	kill_fasync(&tfile->fasync, SIGIO, POLL_OUT);
2426 2427
}

2428
static int tun_sendmsg(struct socket *sock, struct msghdr *m, size_t total_len)
2429
{
J
Jason Wang 已提交
2430 2431
	int ret;
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2432
	struct tun_struct *tun = tun_get(tfile);
J
Jason Wang 已提交
2433 2434 2435

	if (!tun)
		return -EBADFD;
2436

A
Al Viro 已提交
2437
	ret = tun_get_user(tun, tfile, m->msg_control, &m->msg_iter,
J
Jason Wang 已提交
2438 2439
			   m->msg_flags & MSG_DONTWAIT,
			   m->msg_flags & MSG_MORE);
J
Jason Wang 已提交
2440 2441
	tun_put(tun);
	return ret;
2442 2443
}

2444
static int tun_recvmsg(struct socket *sock, struct msghdr *m, size_t total_len,
2445 2446
		       int flags)
{
J
Jason Wang 已提交
2447
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2448
	struct tun_struct *tun = tun_get(tfile);
J
Jason Wang 已提交
2449
	void *ptr = m->msg_control;
2450
	int ret;
J
Jason Wang 已提交
2451

W
Wei Xu 已提交
2452 2453
	if (!tun) {
		ret = -EBADFD;
J
Jason Wang 已提交
2454
		goto out_free;
W
Wei Xu 已提交
2455
	}
J
Jason Wang 已提交
2456

2457
	if (flags & ~(MSG_DONTWAIT|MSG_TRUNC|MSG_ERRQUEUE)) {
2458
		ret = -EINVAL;
W
Wei Xu 已提交
2459
		goto out_put_tun;
2460
	}
2461 2462 2463 2464 2465
	if (flags & MSG_ERRQUEUE) {
		ret = sock_recv_errqueue(sock->sk, m, total_len,
					 SOL_PACKET, TUN_TX_TIMESTAMP);
		goto out;
	}
J
Jason Wang 已提交
2466
	ret = tun_do_read(tun, tfile, &m->msg_iter, flags & MSG_DONTWAIT, ptr);
2467
	if (ret > (ssize_t)total_len) {
2468 2469 2470
		m->msg_flags |= MSG_TRUNC;
		ret = flags & MSG_TRUNC ? ret : total_len;
	}
2471
out:
J
Jason Wang 已提交
2472
	tun_put(tun);
2473
	return ret;
W
Wei Xu 已提交
2474 2475 2476

out_put_tun:
	tun_put(tun);
J
Jason Wang 已提交
2477 2478
out_free:
	tun_ptr_free(ptr);
W
Wei Xu 已提交
2479
	return ret;
2480 2481
}

J
Jason Wang 已提交
2482 2483 2484
static int tun_ptr_peek_len(void *ptr)
{
	if (likely(ptr)) {
2485 2486
		if (tun_is_xdp_frame(ptr)) {
			struct xdp_frame *xdpf = tun_ptr_to_xdp(ptr);
J
Jason Wang 已提交
2487

2488
			return xdpf->len;
J
Jason Wang 已提交
2489 2490 2491 2492 2493 2494 2495
		}
		return __skb_array_len_with_tag(ptr);
	} else {
		return 0;
	}
}

J
Jason Wang 已提交
2496 2497 2498 2499 2500 2501
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;

2502
	tun = tun_get(tfile);
J
Jason Wang 已提交
2503 2504 2505
	if (!tun)
		return 0;

J
Jason Wang 已提交
2506
	ret = PTR_RING_PEEK_CALL(&tfile->tx_ring, tun_ptr_peek_len);
J
Jason Wang 已提交
2507 2508 2509 2510 2511
	tun_put(tun);

	return ret;
}

2512 2513
/* Ops structure to mimic raw sockets with tun */
static const struct proto_ops tun_socket_ops = {
J
Jason Wang 已提交
2514
	.peek_len = tun_peek_len,
2515 2516 2517 2518
	.sendmsg = tun_sendmsg,
	.recvmsg = tun_recvmsg,
};

2519 2520 2521
static struct proto tun_proto = {
	.name		= "tun",
	.owner		= THIS_MODULE,
J
Jason Wang 已提交
2522
	.obj_size	= sizeof(struct tun_file),
2523
};
2524

2525 2526
static int tun_flags(struct tun_struct *tun)
{
2527
	return tun->flags & (TUN_FEATURES | IFF_PERSIST | IFF_TUN | IFF_TAP);
2528 2529 2530 2531 2532 2533 2534 2535 2536 2537 2538 2539 2540
}

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));
2541 2542 2543 2544
	return uid_valid(tun->owner)?
		sprintf(buf, "%u\n",
			from_kuid_munged(current_user_ns(), tun->owner)):
		sprintf(buf, "-1\n");
2545 2546 2547 2548 2549 2550
}

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));
2551 2552 2553 2554
	return gid_valid(tun->group) ?
		sprintf(buf, "%u\n",
			from_kgid_munged(current_user_ns(), tun->group)):
		sprintf(buf, "-1\n");
2555 2556 2557 2558 2559 2560
}

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

2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571
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
};

2572
static int tun_set_iff(struct net *net, struct file *file, struct ifreq *ifr)
L
Linus Torvalds 已提交
2573 2574
{
	struct tun_struct *tun;
J
Jason Wang 已提交
2575
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
2576 2577 2578
	struct net_device *dev;
	int err;

2579 2580 2581
	if (tfile->detached)
		return -EINVAL;

2582 2583 2584 2585 2586 2587 2588 2589 2590
	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;
	}

2591 2592
	dev = __dev_get_by_name(net, ifr->ifr_name);
	if (dev) {
2593 2594
		if (ifr->ifr_flags & IFF_TUN_EXCL)
			return -EBUSY;
2595 2596 2597 2598 2599 2600 2601
		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;

2602
		if (!!(ifr->ifr_flags & IFF_MULTI_QUEUE) !=
M
Michael S. Tsirkin 已提交
2603
		    !!(tun->flags & IFF_MULTI_QUEUE))
2604 2605
			return -EINVAL;

2606
		if (tun_not_capable(tun))
P
Paul Moore 已提交
2607
			return -EPERM;
2608
		err = security_tun_dev_open(tun->security);
P
Paul Moore 已提交
2609 2610 2611
		if (err < 0)
			return err;

2612 2613
		err = tun_attach(tun, file, ifr->ifr_flags & IFF_NOFILTER,
				 ifr->ifr_flags & IFF_NAPI);
2614 2615
		if (err < 0)
			return err;
J
Jason Wang 已提交
2616

M
Michael S. Tsirkin 已提交
2617
		if (tun->flags & IFF_MULTI_QUEUE &&
2618 2619 2620 2621
		    (tun->numqueues + tun->numdisabled > 1)) {
			/* One or more queue has already been attached, no need
			 * to initialize the device again.
			 */
2622
			netdev_state_change(dev);
2623 2624
			return 0;
		}
2625 2626 2627

		tun->flags = (tun->flags & ~TUN_FEATURES) |
			      (ifr->ifr_flags & TUN_FEATURES);
2628 2629 2630

		netdev_state_change(dev);
	} else {
L
Linus Torvalds 已提交
2631 2632
		char *name;
		unsigned long flags = 0;
2633 2634
		int queues = ifr->ifr_flags & IFF_MULTI_QUEUE ?
			     MAX_TAP_QUEUES : 1;
L
Linus Torvalds 已提交
2635

2636
		if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
2637
			return -EPERM;
P
Paul Moore 已提交
2638 2639 2640
		err = security_tun_dev_create();
		if (err < 0)
			return err;
2641

L
Linus Torvalds 已提交
2642 2643 2644
		/* Set dev type */
		if (ifr->ifr_flags & IFF_TUN) {
			/* TUN device */
M
Michael S. Tsirkin 已提交
2645
			flags |= IFF_TUN;
L
Linus Torvalds 已提交
2646 2647 2648
			name = "tun%d";
		} else if (ifr->ifr_flags & IFF_TAP) {
			/* TAP device */
M
Michael S. Tsirkin 已提交
2649
			flags |= IFF_TAP;
L
Linus Torvalds 已提交
2650
			name = "tap%d";
2651
		} else
2652
			return -EINVAL;
2653

L
Linus Torvalds 已提交
2654 2655 2656
		if (*ifr->ifr_name)
			name = ifr->ifr_name;

J
Jason Wang 已提交
2657
		dev = alloc_netdev_mqs(sizeof(struct tun_struct), name,
2658 2659
				       NET_NAME_UNKNOWN, tun_setup, queues,
				       queues);
2660

L
Linus Torvalds 已提交
2661 2662
		if (!dev)
			return -ENOMEM;
2663
		err = dev_get_valid_name(net, dev, name);
2664
		if (err < 0)
2665
			goto err_free_dev;
L
Linus Torvalds 已提交
2666

2667
		dev_net_set(dev, net);
2668
		dev->rtnl_link_ops = &tun_link_ops;
2669
		dev->ifindex = tfile->ifindex;
2670
		dev->sysfs_groups[0] = &tun_attr_group;
S
Stephen Hemminger 已提交
2671

L
Linus Torvalds 已提交
2672 2673 2674
		tun = netdev_priv(dev);
		tun->dev = dev;
		tun->flags = flags;
2675
		tun->txflt.count = 0;
2676
		tun->vnet_hdr_sz = sizeof(struct virtio_net_hdr);
2677

2678
		tun->align = NET_SKB_PAD;
J
Jason Wang 已提交
2679 2680
		tun->filter_attached = false;
		tun->sndbuf = tfile->socket.sk->sk_sndbuf;
J
Jason Wang 已提交
2681
		tun->rx_batched = 0;
2682
		RCU_INIT_POINTER(tun->steering_prog, NULL);
2683

2684 2685 2686 2687 2688 2689
		tun->pcpu_stats = netdev_alloc_pcpu_stats(struct tun_pcpu_stats);
		if (!tun->pcpu_stats) {
			err = -ENOMEM;
			goto err_free_dev;
		}

J
Jason Wang 已提交
2690 2691
		spin_lock_init(&tun->lock);

2692 2693
		err = security_tun_dev_alloc_security(&tun->security);
		if (err < 0)
2694
			goto err_free_stat;
P
Paul Moore 已提交
2695

L
Linus Torvalds 已提交
2696
		tun_net_init(dev);
2697
		tun_flow_init(tun);
J
Jason Wang 已提交
2698

2699
		dev->hw_features = NETIF_F_SG | NETIF_F_FRAGLIST |
J
Jason Wang 已提交
2700 2701
				   TUN_USER_FEATURES | NETIF_F_HW_VLAN_CTAG_TX |
				   NETIF_F_HW_VLAN_STAG_TX;
2702
		dev->features = dev->hw_features | NETIF_F_LLTX;
2703 2704 2705
		dev->vlan_features = dev->features &
				     ~(NETIF_F_HW_VLAN_CTAG_TX |
				       NETIF_F_HW_VLAN_STAG_TX);
2706

2707 2708 2709
		tun->flags = (tun->flags & ~TUN_FEATURES) |
			      (ifr->ifr_flags & TUN_FEATURES);

J
Jason Wang 已提交
2710
		INIT_LIST_HEAD(&tun->disabled);
2711
		err = tun_attach(tun, file, false, ifr->ifr_flags & IFF_NAPI);
2712
		if (err < 0)
2713
			goto err_free_flow;
2714

L
Linus Torvalds 已提交
2715 2716
		err = register_netdevice(tun->dev);
		if (err < 0)
2717
			goto err_detach;
L
Linus Torvalds 已提交
2718 2719
	}

2720 2721
	netif_carrier_on(tun->dev);

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

2724 2725 2726 2727
	/* Make sure persistent devices do not get stuck in
	 * xoff state.
	 */
	if (netif_running(tun->dev))
J
Jason Wang 已提交
2728
		netif_tx_wake_all_queues(tun->dev);
2729

L
Linus Torvalds 已提交
2730 2731 2732
	strcpy(ifr->ifr_name, tun->dev->name);
	return 0;

2733 2734
err_detach:
	tun_detach_all(dev);
2735 2736 2737
	/* register_netdevice() already called tun_free_netdev() */
	goto err_free_dev;

2738 2739 2740
err_free_flow:
	tun_flow_uninit(tun);
	security_tun_dev_free_security(tun->security);
2741 2742
err_free_stat:
	free_percpu(tun->pcpu_stats);
2743
err_free_dev:
L
Linus Torvalds 已提交
2744 2745 2746 2747
	free_netdev(dev);
	return err;
}

R
Rami Rosen 已提交
2748
static void tun_get_iff(struct net *net, struct tun_struct *tun,
2749
		       struct ifreq *ifr)
2750
{
2751
	tun_debug(KERN_INFO, tun, "tun_get_iff\n");
2752 2753 2754

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

2755
	ifr->ifr_flags = tun_flags(tun);
2756 2757 2758

}

2759 2760
/* This is like a cut-down ethtool ops, except done via tun fd so no
 * privs required. */
2761
static int set_offload(struct tun_struct *tun, unsigned long arg)
2762
{
2763
	netdev_features_t features = 0;
2764 2765

	if (arg & TUN_F_CSUM) {
2766
		features |= NETIF_F_HW_CSUM;
2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777 2778 2779
		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);
		}
2780 2781

		arg &= ~TUN_F_UFO;
2782 2783 2784 2785 2786 2787 2788
	}

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

2789
	tun->set_features = features;
2790 2791
	tun->dev->wanted_features &= ~TUN_USER_FEATURES;
	tun->dev->wanted_features |= features;
2792
	netdev_update_features(tun->dev);
2793 2794 2795 2796

	return 0;
}

J
Jason Wang 已提交
2797 2798 2799 2800 2801 2802
static void tun_detach_filter(struct tun_struct *tun, int n)
{
	int i;
	struct tun_file *tfile;

	for (i = 0; i < n; i++) {
2803
		tfile = rtnl_dereference(tun->tfiles[i]);
2804 2805 2806
		lock_sock(tfile->socket.sk);
		sk_detach_filter(tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2807 2808 2809 2810 2811 2812 2813 2814 2815 2816 2817
	}

	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++) {
2818
		tfile = rtnl_dereference(tun->tfiles[i]);
2819 2820 2821
		lock_sock(tfile->socket.sk);
		ret = sk_attach_filter(&tun->fprog, tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2822 2823 2824 2825 2826 2827 2828 2829 2830 2831 2832 2833 2834 2835 2836 2837
		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++) {
2838
		tfile = rtnl_dereference(tun->tfiles[i]);
J
Jason Wang 已提交
2839 2840 2841 2842
		tfile->socket.sk->sk_sndbuf = tun->sndbuf;
	}
}

2843 2844 2845 2846 2847 2848 2849 2850 2851
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 已提交
2852
		tun = tfile->detached;
2853
		if (!tun) {
2854
			ret = -EINVAL;
2855 2856 2857 2858 2859
			goto unlock;
		}
		ret = security_tun_dev_attach_queue(tun->security);
		if (ret < 0)
			goto unlock;
2860
		ret = tun_attach(tun, file, false, tun->flags & IFF_NAPI);
J
Jason Wang 已提交
2861
	} else if (ifr->ifr_flags & IFF_DETACH_QUEUE) {
2862
		tun = rtnl_dereference(tfile->tun);
M
Michael S. Tsirkin 已提交
2863
		if (!tun || !(tun->flags & IFF_MULTI_QUEUE) || tfile->detached)
J
Jason Wang 已提交
2864 2865 2866 2867
			ret = -EINVAL;
		else
			__tun_detach(tfile, false);
	} else
2868 2869
		ret = -EINVAL;

2870 2871 2872
	if (ret >= 0)
		netdev_state_change(tun->dev);

2873
unlock:
2874 2875 2876 2877
	rtnl_unlock();
	return ret;
}

2878 2879
static int tun_set_ebpf(struct tun_struct *tun, struct tun_prog **prog_p,
			void __user *data)
2880 2881 2882 2883 2884 2885 2886 2887 2888 2889 2890 2891 2892 2893 2894
{
	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);
	}

2895
	return __tun_set_ebpf(tun, prog_p, prog);
2896 2897
}

2898 2899
static long __tun_chr_ioctl(struct file *file, unsigned int cmd,
			    unsigned long arg, int ifreq_len)
L
Linus Torvalds 已提交
2900
{
E
Eric W. Biederman 已提交
2901
	struct tun_file *tfile = file->private_data;
2902
	struct net *net = sock_net(&tfile->sk);
E
Eric W. Biederman 已提交
2903
	struct tun_struct *tun;
L
Linus Torvalds 已提交
2904 2905
	void __user* argp = (void __user*)arg;
	struct ifreq ifr;
2906 2907
	kuid_t owner;
	kgid_t group;
2908
	int sndbuf;
2909
	int vnet_hdr_sz;
2910
	unsigned int ifindex;
M
Michael S. Tsirkin 已提交
2911
	int le;
2912
	int ret;
2913
	bool do_notify = false;
L
Linus Torvalds 已提交
2914

2915 2916
	if (cmd == TUNSETIFF || cmd == TUNSETQUEUE ||
	    (_IOC_TYPE(cmd) == SOCK_IOC_TYPE && cmd != SIOCGSKNS)) {
2917
		if (copy_from_user(&ifr, argp, ifreq_len))
L
Linus Torvalds 已提交
2918
			return -EFAULT;
D
David S. Miller 已提交
2919
	} else {
2920
		memset(&ifr, 0, sizeof(ifr));
D
David S. Miller 已提交
2921
	}
E
Eric W. Biederman 已提交
2922 2923 2924
	if (cmd == TUNGETFEATURES) {
		/* Currently this just means: "what IFF flags are valid?".
		 * This is needed because we never checked for invalid flags on
2925 2926 2927
		 * TUNSETIFF.
		 */
		return put_user(IFF_TUN | IFF_TAP | TUN_FEATURES,
E
Eric W. Biederman 已提交
2928
				(unsigned int __user*)argp);
2929
	} else if (cmd == TUNSETQUEUE) {
2930
		return tun_set_queue(file, &ifr);
2931 2932 2933 2934 2935
	} else if (cmd == SIOCGSKNS) {
		if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
			return -EPERM;
		return open_related_ns(&net->ns, get_net_ns);
	}
E
Eric W. Biederman 已提交
2936

J
Jason Wang 已提交
2937
	ret = 0;
2938 2939
	rtnl_lock();

2940
	tun = tun_get(tfile);
2941 2942 2943 2944 2945
	if (cmd == TUNSETIFF) {
		ret = -EEXIST;
		if (tun)
			goto unlock;

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

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

2950 2951
		if (ret)
			goto unlock;
L
Linus Torvalds 已提交
2952

2953
		if (copy_to_user(argp, &ifr, ifreq_len))
2954 2955
			ret = -EFAULT;
		goto unlock;
L
Linus Torvalds 已提交
2956
	}
2957 2958 2959 2960 2961 2962 2963 2964 2965 2966 2967 2968 2969
	if (cmd == TUNSETIFINDEX) {
		ret = -EPERM;
		if (tun)
			goto unlock;

		ret = -EFAULT;
		if (copy_from_user(&ifindex, argp, sizeof(ifindex)))
			goto unlock;

		ret = 0;
		tfile->ifindex = ifindex;
		goto unlock;
	}
L
Linus Torvalds 已提交
2970

2971
	ret = -EBADFD;
L
Linus Torvalds 已提交
2972
	if (!tun)
2973
		goto unlock;
L
Linus Torvalds 已提交
2974

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

E
Eric W. Biederman 已提交
2977
	ret = 0;
L
Linus Torvalds 已提交
2978
	switch (cmd) {
2979
	case TUNGETIFF:
R
Rami Rosen 已提交
2980
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
2981

2982 2983
		if (tfile->detached)
			ifr.ifr_flags |= IFF_DETACH_QUEUE;
2984 2985
		if (!tfile->socket.sk->sk_filter)
			ifr.ifr_flags |= IFF_NOFILTER;
2986

2987
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
2988
			ret = -EFAULT;
2989 2990
		break;

L
Linus Torvalds 已提交
2991 2992 2993
	case TUNSETNOCSUM:
		/* Disable/Enable checksum */

2994 2995
		/* [unimplemented] */
		tun_debug(KERN_INFO, tun, "ignored: set checksum %s\n",
2996
			  arg ? "disabled" : "enabled");
L
Linus Torvalds 已提交
2997 2998 2999
		break;

	case TUNSETPERSIST:
J
Jason Wang 已提交
3000 3001 3002
		/* Disable/Enable persist mode. Keep an extra reference to the
		 * module to prevent the module being unprobed.
		 */
M
Michael S. Tsirkin 已提交
3003 3004
		if (arg && !(tun->flags & IFF_PERSIST)) {
			tun->flags |= IFF_PERSIST;
J
Jason Wang 已提交
3005
			__module_get(THIS_MODULE);
3006
			do_notify = true;
J
Jason Wang 已提交
3007
		}
M
Michael S. Tsirkin 已提交
3008 3009
		if (!arg && (tun->flags & IFF_PERSIST)) {
			tun->flags &= ~IFF_PERSIST;
J
Jason Wang 已提交
3010
			module_put(THIS_MODULE);
3011
			do_notify = true;
J
Jason Wang 已提交
3012
		}
L
Linus Torvalds 已提交
3013

3014 3015
		tun_debug(KERN_INFO, tun, "persist %s\n",
			  arg ? "enabled" : "disabled");
L
Linus Torvalds 已提交
3016 3017 3018 3019
		break;

	case TUNSETOWNER:
		/* Set owner of the device */
3020 3021 3022 3023 3024 3025
		owner = make_kuid(current_user_ns(), arg);
		if (!uid_valid(owner)) {
			ret = -EINVAL;
			break;
		}
		tun->owner = owner;
3026
		do_notify = true;
3027
		tun_debug(KERN_INFO, tun, "owner set to %u\n",
3028
			  from_kuid(&init_user_ns, tun->owner));
L
Linus Torvalds 已提交
3029 3030
		break;

3031 3032
	case TUNSETGROUP:
		/* Set group of the device */
3033 3034 3035 3036 3037 3038
		group = make_kgid(current_user_ns(), arg);
		if (!gid_valid(group)) {
			ret = -EINVAL;
			break;
		}
		tun->group = group;
3039
		do_notify = true;
3040
		tun_debug(KERN_INFO, tun, "group set to %u\n",
3041
			  from_kgid(&init_user_ns, tun->group));
3042 3043
		break;

3044 3045 3046
	case TUNSETLINK:
		/* Only allow setting the type when the interface is down */
		if (tun->dev->flags & IFF_UP) {
3047 3048
			tun_debug(KERN_INFO, tun,
				  "Linktype set failed because interface is up\n");
3049
			ret = -EBUSY;
3050 3051
		} else {
			tun->dev->type = (int) arg;
3052 3053
			tun_debug(KERN_INFO, tun, "linktype set to %d\n",
				  tun->dev->type);
3054
			ret = 0;
3055
		}
E
Eric W. Biederman 已提交
3056
		break;
3057

L
Linus Torvalds 已提交
3058 3059 3060 3061 3062
#ifdef TUN_DEBUG
	case TUNSETDEBUG:
		tun->debug = arg;
		break;
#endif
3063
	case TUNSETOFFLOAD:
3064
		ret = set_offload(tun, arg);
E
Eric W. Biederman 已提交
3065
		break;
3066

3067 3068
	case TUNSETTXFILTER:
		/* Can be set only for TAPs */
E
Eric W. Biederman 已提交
3069
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3070
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
E
Eric W. Biederman 已提交
3071
			break;
H
Harvey Harrison 已提交
3072
		ret = update_filter(&tun->txflt, (void __user *)arg);
E
Eric W. Biederman 已提交
3073
		break;
L
Linus Torvalds 已提交
3074 3075

	case SIOCGIFHWADDR:
3076
		/* Get hw address */
3077 3078
		memcpy(ifr.ifr_hwaddr.sa_data, tun->dev->dev_addr, ETH_ALEN);
		ifr.ifr_hwaddr.sa_family = tun->dev->type;
3079
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
3080 3081
			ret = -EFAULT;
		break;
L
Linus Torvalds 已提交
3082 3083

	case SIOCSIFHWADDR:
3084
		/* Set hw address */
3085 3086
		tun_debug(KERN_DEBUG, tun, "set hw address: %pM\n",
			  ifr.ifr_hwaddr.sa_data);
3087 3088

		ret = dev_set_mac_address(tun->dev, &ifr.ifr_hwaddr);
E
Eric W. Biederman 已提交
3089
		break;
3090 3091

	case TUNGETSNDBUF:
J
Jason Wang 已提交
3092
		sndbuf = tfile->socket.sk->sk_sndbuf;
3093 3094 3095 3096 3097 3098 3099 3100 3101
		if (copy_to_user(argp, &sndbuf, sizeof(sndbuf)))
			ret = -EFAULT;
		break;

	case TUNSETSNDBUF:
		if (copy_from_user(&sndbuf, argp, sizeof(sndbuf))) {
			ret = -EFAULT;
			break;
		}
3102 3103 3104 3105
		if (sndbuf <= 0) {
			ret = -EINVAL;
			break;
		}
3106

J
Jason Wang 已提交
3107 3108
		tun->sndbuf = sndbuf;
		tun_set_sndbuf(tun);
3109 3110
		break;

3111 3112 3113 3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125 3126 3127 3128 3129
	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 已提交
3130 3131 3132 3133 3134 3135 3136 3137 3138 3139 3140 3141 3142 3143 3144 3145 3146
	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;

3147 3148 3149 3150 3151 3152 3153 3154
	case TUNGETVNETBE:
		ret = tun_get_vnet_be(tun, argp);
		break;

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

M
Michael S. Tsirkin 已提交
3155 3156 3157
	case TUNATTACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3158
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
3159 3160
			break;
		ret = -EFAULT;
J
Jason Wang 已提交
3161
		if (copy_from_user(&tun->fprog, argp, sizeof(tun->fprog)))
M
Michael S. Tsirkin 已提交
3162 3163
			break;

J
Jason Wang 已提交
3164
		ret = tun_attach_filter(tun);
M
Michael S. Tsirkin 已提交
3165 3166 3167 3168 3169
		break;

	case TUNDETACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3170
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
3171
			break;
J
Jason Wang 已提交
3172 3173
		ret = 0;
		tun_detach_filter(tun, tun->numqueues);
M
Michael S. Tsirkin 已提交
3174 3175
		break;

P
Pavel Emelyanov 已提交
3176 3177
	case TUNGETFILTER:
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3178
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
P
Pavel Emelyanov 已提交
3179 3180 3181 3182 3183 3184 3185
			break;
		ret = -EFAULT;
		if (copy_to_user(argp, &tun->fprog, sizeof(tun->fprog)))
			break;
		ret = 0;
		break;

3186
	case TUNSETSTEERINGEBPF:
3187
		ret = tun_set_ebpf(tun, &tun->steering_prog, argp);
3188 3189
		break;

3190 3191 3192 3193
	case TUNSETFILTEREBPF:
		ret = tun_set_ebpf(tun, &tun->filter_prog, argp);
		break;

L
Linus Torvalds 已提交
3194
	default:
E
Eric W. Biederman 已提交
3195 3196
		ret = -EINVAL;
		break;
3197
	}
L
Linus Torvalds 已提交
3198

3199 3200 3201
	if (do_notify)
		netdev_state_change(tun->dev);

3202 3203 3204 3205
unlock:
	rtnl_unlock();
	if (tun)
		tun_put(tun);
E
Eric W. Biederman 已提交
3206
	return ret;
L
Linus Torvalds 已提交
3207 3208
}

3209 3210 3211 3212 3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224 3225 3226 3227 3228 3229 3230 3231 3232 3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243
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 已提交
3244 3245
static int tun_chr_fasync(int fd, struct file *file, int on)
{
J
Jason Wang 已提交
3246
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
3247 3248
	int ret;

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

L
Linus Torvalds 已提交
3252
	if (on) {
3253
		__f_setown(file, task_pid(current), PIDTYPE_TGID, 0);
J
Jason Wang 已提交
3254
		tfile->flags |= TUN_FASYNC;
3255
	} else
J
Jason Wang 已提交
3256
		tfile->flags &= ~TUN_FASYNC;
J
Jonathan Corbet 已提交
3257 3258 3259
	ret = 0;
out:
	return ret;
L
Linus Torvalds 已提交
3260 3261 3262 3263
}

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

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

3269
	tfile = (struct tun_file *)sk_alloc(net, AF_UNSPEC, GFP_KERNEL,
3270
					    &tun_proto, 0);
E
Eric W. Biederman 已提交
3271 3272
	if (!tfile)
		return -ENOMEM;
3273 3274 3275 3276 3277
	if (ptr_ring_init(&tfile->tx_ring, 0, GFP_KERNEL)) {
		sk_free(&tfile->sk);
		return -ENOMEM;
	}

3278
	RCU_INIT_POINTER(tfile->tun, NULL);
J
Jason Wang 已提交
3279
	tfile->flags = 0;
3280
	tfile->ifindex = 0;
J
Jason Wang 已提交
3281 3282

	init_waitqueue_head(&tfile->wq.wait);
3283
	RCU_INIT_POINTER(tfile->socket.wq, &tfile->wq);
J
Jason Wang 已提交
3284 3285 3286 3287 3288 3289 3290 3291 3292

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

3296 3297
	sock_set_flag(&tfile->sk, SOCK_ZEROCOPY);

L
Linus Torvalds 已提交
3298 3299 3300 3301 3302
	return 0;
}

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

J
Jason Wang 已提交
3305
	tun_detach(tfile, true);
L
Linus Torvalds 已提交
3306 3307 3308 3309

	return 0;
}

3310
#ifdef CONFIG_PROC_FS
3311
static void tun_chr_show_fdinfo(struct seq_file *m, struct file *file)
3312
{
3313
	struct tun_file *tfile = file->private_data;
3314 3315 3316 3317 3318 3319
	struct tun_struct *tun;
	struct ifreq ifr;

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

	rtnl_lock();
3320
	tun = tun_get(tfile);
3321 3322 3323 3324 3325 3326 3327
	if (tun)
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
	rtnl_unlock();

	if (tun)
		tun_put(tun);

3328
	seq_printf(m, "iff:\t%s\n", ifr.ifr_name);
3329 3330 3331
}
#endif

3332
static const struct file_operations tun_fops = {
3333
	.owner	= THIS_MODULE,
L
Linus Torvalds 已提交
3334
	.llseek = no_llseek,
3335
	.read_iter  = tun_chr_read_iter,
3336
	.write_iter = tun_chr_write_iter,
L
Linus Torvalds 已提交
3337
	.poll	= tun_chr_poll,
3338 3339 3340 3341
	.unlocked_ioctl	= tun_chr_ioctl,
#ifdef CONFIG_COMPAT
	.compat_ioctl = tun_chr_compat_ioctl,
#endif
L
Linus Torvalds 已提交
3342 3343
	.open	= tun_chr_open,
	.release = tun_chr_close,
3344 3345 3346 3347
	.fasync = tun_chr_fasync,
#ifdef CONFIG_PROC_FS
	.show_fdinfo = tun_chr_show_fdinfo,
#endif
L
Linus Torvalds 已提交
3348 3349 3350 3351 3352
};

static struct miscdevice tun_miscdev = {
	.minor = TUN_MINOR,
	.name = "tun",
3353
	.nodename = "net/tun",
L
Linus Torvalds 已提交
3354 3355 3356 3357 3358
	.fops = &tun_fops,
};

/* ethtool interface */

3359 3360
static void tun_default_link_ksettings(struct net_device *dev,
				       struct ethtool_link_ksettings *cmd)
3361 3362 3363 3364 3365 3366 3367 3368
{
	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;
3369 3370 3371 3372 3373 3374 3375 3376 3377 3378 3379 3380 3381 3382 3383 3384 3385
}

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

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

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

	memcpy(&tun->link_ksettings, cmd, sizeof(*cmd));
L
Linus Torvalds 已提交
3386 3387 3388 3389 3390 3391 3392
	return 0;
}

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

3393 3394
	strlcpy(info->driver, DRV_NAME, sizeof(info->driver));
	strlcpy(info->version, DRV_VERSION, sizeof(info->version));
L
Linus Torvalds 已提交
3395 3396

	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
3397
	case IFF_TUN:
3398
		strlcpy(info->bus_info, "tun", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3399
		break;
M
Michael S. Tsirkin 已提交
3400
	case IFF_TAP:
3401
		strlcpy(info->bus_info, "tap", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3402 3403 3404 3405 3406 3407 3408 3409 3410 3411 3412 3413 3414 3415 3416 3417 3418 3419 3420 3421 3422 3423
		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 已提交
3424 3425 3426 3427 3428 3429 3430 3431 3432 3433 3434 3435 3436 3437 3438 3439 3440 3441 3442 3443 3444 3445 3446
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;
}

3447
static const struct ethtool_ops tun_ethtool_ops = {
L
Linus Torvalds 已提交
3448 3449 3450
	.get_drvinfo	= tun_get_drvinfo,
	.get_msglevel	= tun_get_msglevel,
	.set_msglevel	= tun_set_msglevel,
3451
	.get_link	= ethtool_op_get_link,
3452
	.get_ts_info	= ethtool_op_get_ts_info,
J
Jason Wang 已提交
3453 3454
	.get_coalesce   = tun_get_coalesce,
	.set_coalesce   = tun_set_coalesce,
3455
	.get_link_ksettings = tun_get_link_ksettings,
3456
	.set_link_ksettings = tun_set_link_ksettings,
L
Linus Torvalds 已提交
3457 3458
};

J
Jason Wang 已提交
3459 3460 3461 3462
static int tun_queue_resize(struct tun_struct *tun)
{
	struct net_device *dev = tun->dev;
	struct tun_file *tfile;
3463
	struct ptr_ring **rings;
J
Jason Wang 已提交
3464 3465 3466
	int n = tun->numqueues + tun->numdisabled;
	int ret, i;

3467 3468
	rings = kmalloc_array(n, sizeof(*rings), GFP_KERNEL);
	if (!rings)
J
Jason Wang 已提交
3469 3470 3471 3472
		return -ENOMEM;

	for (i = 0; i < tun->numqueues; i++) {
		tfile = rtnl_dereference(tun->tfiles[i]);
3473
		rings[i] = &tfile->tx_ring;
J
Jason Wang 已提交
3474 3475
	}
	list_for_each_entry(tfile, &tun->disabled, next)
3476
		rings[i++] = &tfile->tx_ring;
J
Jason Wang 已提交
3477

3478 3479
	ret = ptr_ring_resize_multiple(rings, n,
				       dev->tx_queue_len, GFP_KERNEL,
J
Jason Wang 已提交
3480
				       tun_ptr_free);
J
Jason Wang 已提交
3481

3482
	kfree(rings);
J
Jason Wang 已提交
3483 3484 3485 3486 3487 3488 3489 3490 3491
	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);

3492 3493 3494
	if (dev->rtnl_link_ops != &tun_link_ops)
		return NOTIFY_DONE;

J
Jason Wang 已提交
3495 3496 3497 3498 3499 3500 3501 3502 3503 3504 3505 3506 3507 3508 3509
	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,
};
3510

L
Linus Torvalds 已提交
3511 3512 3513 3514
static int __init tun_init(void)
{
	int ret = 0;

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

3517
	ret = rtnl_link_register(&tun_link_ops);
3518
	if (ret) {
3519
		pr_err("Can't register link_ops\n");
3520
		goto err_linkops;
3521 3522
	}

L
Linus Torvalds 已提交
3523
	ret = misc_register(&tun_miscdev);
3524
	if (ret) {
3525
		pr_err("Can't register misc device %d\n", TUN_MINOR);
3526 3527
		goto err_misc;
	}
J
Jason Wang 已提交
3528

3529 3530 3531 3532 3533 3534
	ret = register_netdevice_notifier(&tun_notifier_block);
	if (ret) {
		pr_err("Can't register netdevice notifier\n");
		goto err_notifier;
	}

3535
	return  0;
3536 3537 3538

err_notifier:
	misc_deregister(&tun_miscdev);
3539
err_misc:
3540 3541
	rtnl_link_unregister(&tun_link_ops);
err_linkops:
L
Linus Torvalds 已提交
3542 3543 3544 3545 3546
	return ret;
}

static void tun_cleanup(void)
{
3547
	misc_deregister(&tun_miscdev);
3548
	rtnl_link_unregister(&tun_link_ops);
J
Jason Wang 已提交
3549
	unregister_netdevice_notifier(&tun_notifier_block);
L
Linus Torvalds 已提交
3550 3551
}

3552 3553 3554 3555 3556 3557
/* 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)
{
3558
	struct tun_file *tfile;
3559 3560
	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
3561 3562
	tfile = file->private_data;
	if (!tfile)
3563
		return ERR_PTR(-EBADFD);
J
Jason Wang 已提交
3564
	return &tfile->socket;
3565 3566 3567
}
EXPORT_SYMBOL_GPL(tun_get_socket);

3568
struct ptr_ring *tun_get_tx_ring(struct file *file)
J
Jason Wang 已提交
3569 3570 3571 3572 3573 3574 3575 3576
{
	struct tun_file *tfile;

	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
	tfile = file->private_data;
	if (!tfile)
		return ERR_PTR(-EBADFD);
3577
	return &tfile->tx_ring;
J
Jason Wang 已提交
3578
}
3579
EXPORT_SYMBOL_GPL(tun_get_tx_ring);
J
Jason Wang 已提交
3580

L
Linus Torvalds 已提交
3581 3582 3583 3584 3585 3586
module_init(tun_init);
module_exit(tun_cleanup);
MODULE_DESCRIPTION(DRV_DESCRIPTION);
MODULE_AUTHOR(DRV_COPYRIGHT);
MODULE_LICENSE("GPL");
MODULE_ALIAS_MISCDEV(TUN_MINOR);
3587
MODULE_ALIAS("devname:net/tun");