tun.c 82.4 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

J
Jason Wang 已提交
116
#define TUN_HEADROOM 256
117
#define TUN_RX_PAD (NET_IP_ALIGN + NET_SKB_PAD)
118

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

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

132 133
#define GOODCOPY_LEN 128

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

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

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

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

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

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

#define TUN_NUM_FLOW_ENTRIES 1024

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 409 410 411 412 413 414 415
static inline u32 tun_hashfn(u32 rxhash)
{
	return rxhash & 0x3ff;
}

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

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 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624
	u16 ret = 0;

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

	return ret % tun->numqueues;
}

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

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

	return ret;
}

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

	err = 0;
J
Jason Wang 已提交
818

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

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

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

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

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

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

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

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

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

out:
	return err;
E
Eric W. Biederman 已提交
878 879
}

880
static struct tun_struct *tun_get(struct tun_file *tfile)
E
Eric W. Biederman 已提交
881
{
882
	struct tun_struct *tun;
883

884 885 886 887 888
	rcu_read_lock();
	tun = rcu_dereference(tfile->tun);
	if (tun)
		dev_hold(tun->dev);
	rcu_read_unlock();
889 890

	return tun;
E
Eric W. Biederman 已提交
891 892 893 894
}

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

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

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

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

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

1004
static const struct ethtool_ops tun_ethtool_ops;
L
Linus Torvalds 已提交
1005

1006 1007 1008
/* Net device detach from fd. */
static void tun_net_uninit(struct net_device *dev)
{
J
Jason Wang 已提交
1009
	tun_detach_all(dev);
1010 1011
}

L
Linus Torvalds 已提交
1012 1013 1014
/* Net device open. */
static int tun_net_open(struct net_device *dev)
{
1015 1016 1017
	struct tun_struct *tun = netdev_priv(dev);
	int i;

J
Jason Wang 已提交
1018
	netif_tx_start_all_queues(dev);
1019 1020 1021 1022 1023 1024 1025 1026

	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 已提交
1027 1028 1029 1030 1031 1032
	return 0;
}

/* Net device close. */
static int tun_net_close(struct net_device *dev)
{
J
Jason Wang 已提交
1033
	netif_tx_stop_all_queues(dev);
L
Linus Torvalds 已提交
1034 1035 1036 1037
	return 0;
}

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

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

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

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

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

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

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

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

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

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

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

1110
	skb_tx_timestamp(skb);
1111

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

1117 1118
	nf_reset(skb);

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

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

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

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

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

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

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

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

1177 1178 1179
		if (tun_napi_frags_enabled(tun))
			return;

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

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

1202
static void
1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237
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 已提交
1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263
static int tun_xdp_set(struct net_device *dev, struct bpf_prog *prog,
		       struct netlink_ext_ack *extack)
{
	struct tun_struct *tun = netdev_priv(dev);
	struct bpf_prog *old_prog;

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

	return 0;
}

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

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

	return 0;
}

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

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

1292 1293 1294 1295 1296 1297 1298 1299
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);
}

1300 1301
static int tun_xdp_xmit(struct net_device *dev, int n,
			struct xdp_frame **frames, u32 flags)
J
Jason Wang 已提交
1302 1303 1304 1305
{
	struct tun_struct *tun = netdev_priv(dev);
	struct tun_file *tfile;
	u32 numqueues;
1306 1307 1308
	int drops = 0;
	int cnt = n;
	int i;
J
Jason Wang 已提交
1309

1310
	if (unlikely(flags & ~XDP_XMIT_FLAGS_MASK))
1311 1312
		return -EINVAL;

J
Jason Wang 已提交
1313 1314 1315 1316
	rcu_read_lock();

	numqueues = READ_ONCE(tun->numqueues);
	if (!numqueues) {
1317 1318
		rcu_read_unlock();
		return -ENXIO; /* Caller will free/return all frames */
J
Jason Wang 已提交
1319 1320 1321 1322
	}

	tfile = rcu_dereference(tun->tfiles[smp_processor_id() %
					    numqueues]);
1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336

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

1340 1341 1342
	if (flags & XDP_XMIT_FLUSH)
		__tun_xdp_flush_tfile(tfile);

J
Jason Wang 已提交
1343
	rcu_read_unlock();
1344
	return cnt - drops;
J
Jason Wang 已提交
1345 1346
}

1347 1348 1349 1350 1351 1352 1353
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;

1354
	return tun_xdp_xmit(dev, 1, &frame, XDP_XMIT_FLUSH);
J
Jason Wang 已提交
1355 1356
}

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

1377
static void tun_flow_init(struct tun_struct *tun)
J
Jason Wang 已提交
1378 1379 1380 1381 1382 1383 1384
{
	int i;

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

	tun->ageing_time = TUN_FLOW_EXPIRE;
1385 1386 1387
	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 已提交
1388 1389 1390 1391 1392 1393 1394 1395
}

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

1396 1397 1398
#define MIN_MTU 68
#define MAX_MTU 65535

L
Linus Torvalds 已提交
1399 1400 1401 1402
/* Initialize net device. */
static void tun_net_init(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);
1403

L
Linus Torvalds 已提交
1404
	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
1405
	case IFF_TUN:
S
Stephen Hemminger 已提交
1406 1407
		dev->netdev_ops = &tun_netdev_ops;

L
Linus Torvalds 已提交
1408 1409 1410 1411 1412 1413
		/* Point-to-Point TUN Device */
		dev->hard_header_len = 0;
		dev->addr_len = 0;
		dev->mtu = 1500;

		/* Zero header length */
1414
		dev->type = ARPHRD_NONE;
L
Linus Torvalds 已提交
1415 1416 1417
		dev->flags = IFF_POINTOPOINT | IFF_NOARP | IFF_MULTICAST;
		break;

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

1425
		eth_hw_addr_random(dev);
1426

L
Linus Torvalds 已提交
1427 1428
		break;
	}
1429 1430 1431

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

1434 1435 1436 1437 1438 1439 1440
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 已提交
1441 1442 1443
/* Character device part */

/* Poll */
1444
static __poll_t tun_chr_poll(struct file *file, poll_table *wait)
1445
{
1446
	struct tun_file *tfile = file->private_data;
1447
	struct tun_struct *tun = tun_get(tfile);
1448
	struct sock *sk;
1449
	__poll_t mask = 0;
L
Linus Torvalds 已提交
1450 1451

	if (!tun)
1452
		return EPOLLERR;
L
Linus Torvalds 已提交
1453

J
Jason Wang 已提交
1454
	sk = tfile->socket.sk;
1455

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

1458
	poll_wait(file, sk_sleep(sk), wait);
1459

1460
	if (!ptr_ring_empty(&tfile->tx_ring))
1461
		mask |= EPOLLIN | EPOLLRDNORM;
L
Linus Torvalds 已提交
1462

1463 1464 1465 1466 1467 1468 1469 1470
	/* 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)))
1471
		mask |= EPOLLOUT | EPOLLWRNORM;
1472

1473
	if (tun->dev->reg_state != NETREG_REGISTERED)
1474
		mask = EPOLLERR;
1475

E
Eric W. Biederman 已提交
1476
	tun_put(tun);
L
Linus Torvalds 已提交
1477 1478 1479
	return mask;
}

1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507
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++) {
1508
		struct page_frag *pfrag = &current->task_frag;
1509 1510 1511 1512 1513 1514 1515
		size_t fragsz = it->iov[i].iov_len;

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

1516
		if (!skb_page_frag_refill(fragsz, pfrag, GFP_KERNEL)) {
1517 1518 1519 1520
			err = -ENOMEM;
			goto free;
		}

1521 1522 1523 1524
		skb_fill_page_desc(skb, i - 1, pfrag->page,
				   pfrag->offset, fragsz);
		page_ref_inc(pfrag->page);
		pfrag->offset += fragsz;
1525 1526 1527 1528 1529 1530 1531 1532 1533
	}

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

1534 1535
/* 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 已提交
1536
static struct sk_buff *tun_alloc_skb(struct tun_file *tfile,
1537 1538
				     size_t prepad, size_t len,
				     size_t linear, int noblock)
1539
{
J
Jason Wang 已提交
1540
	struct sock *sk = tfile->socket.sk;
1541
	struct sk_buff *skb;
1542
	int err;
1543 1544

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

1548
	skb = sock_alloc_send_pskb(sk, prepad + linear, len - linear, noblock,
1549
				   &err, 0);
1550
	if (!skb)
1551
		return ERR_PTR(err);
1552 1553 1554

	skb_reserve(skb, prepad);
	skb_put(skb, linear);
1555 1556
	skb->data_len = len - linear;
	skb->len += len - linear;
1557 1558 1559 1560

	return skb;
}

J
Jason Wang 已提交
1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596
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();
	}
}

1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618
static bool tun_can_build_skb(struct tun_struct *tun, struct tun_file *tfile,
			      int len, int noblock, bool zerocopy)
{
	if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
		return false;

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

	if (!noblock)
		return false;

	if (zerocopy)
		return false;

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

	return true;
}

J
Jason Wang 已提交
1619 1620
static struct sk_buff *tun_build_skb(struct tun_struct *tun,
				     struct tun_file *tfile,
1621
				     struct iov_iter *from,
J
Jason Wang 已提交
1622
				     struct virtio_net_hdr *hdr,
J
Jason Wang 已提交
1623
				     int len, int *skb_xdp)
1624
{
1625
	struct page_frag *alloc_frag = &current->task_frag;
1626
	struct sk_buff *skb;
J
Jason Wang 已提交
1627
	struct bpf_prog *xdp_prog;
1628
	int buflen = SKB_DATA_ALIGN(sizeof(struct skb_shared_info));
J
Jason Wang 已提交
1629
	unsigned int delta = 0;
1630 1631
	char *buf;
	size_t copied;
1632 1633 1634 1635 1636 1637 1638 1639
	int err, pad = TUN_RX_PAD;

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

1641
	alloc_frag->offset = ALIGN((u64)alloc_frag->offset, SMP_CACHE_BYTES);
1642 1643 1644 1645 1646
	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,
1647
				     alloc_frag->offset + pad,
1648 1649 1650 1651
				     len, from);
	if (copied != len)
		return ERR_PTR(-EFAULT);

1652 1653 1654 1655 1656
	/* There's a small window that XDP may be set after the check
	 * of xdp_prog above, this should be rare and for simplicity
	 * we do XDP on skb in case the headroom is not enough.
	 */
	if (hdr->gso_type || !xdp_prog)
J
Jason Wang 已提交
1657
		*skb_xdp = 1;
J
Jason Wang 已提交
1658
	else
J
Jason Wang 已提交
1659
		*skb_xdp = 0;
J
Jason Wang 已提交
1660

1661
	local_bh_disable();
J
Jason Wang 已提交
1662 1663
	rcu_read_lock();
	xdp_prog = rcu_dereference(tun->xdp_prog);
J
Jason Wang 已提交
1664
	if (xdp_prog && !*skb_xdp) {
J
Jason Wang 已提交
1665 1666 1667 1668 1669
		struct xdp_buff xdp;
		void *orig_data;
		u32 act;

		xdp.data_hard_start = buf;
1670
		xdp.data = buf + pad;
1671
		xdp_set_data_meta_invalid(&xdp);
J
Jason Wang 已提交
1672
		xdp.data_end = xdp.data + len;
J
Jesper Dangaard Brouer 已提交
1673
		xdp.rxq = &tfile->xdp_rxq;
J
Jason Wang 已提交
1674 1675 1676 1677 1678 1679 1680 1681
		orig_data = xdp.data;
		act = bpf_prog_run_xdp(xdp_prog, &xdp);

		switch (act) {
		case XDP_REDIRECT:
			get_page(alloc_frag->page);
			alloc_frag->offset += buflen;
			err = xdp_do_redirect(tun->dev, &xdp, xdp_prog);
1682
			xdp_do_flush_map();
J
Jason Wang 已提交
1683 1684
			if (err)
				goto err_redirect;
1685
			rcu_read_unlock();
1686
			local_bh_enable();
J
Jason Wang 已提交
1687 1688
			return NULL;
		case XDP_TX:
J
Jason Wang 已提交
1689 1690
			get_page(alloc_frag->page);
			alloc_frag->offset += buflen;
1691
			if (tun_xdp_tx(tun->dev, &xdp))
J
Jason Wang 已提交
1692 1693
				goto err_redirect;
			rcu_read_unlock();
1694
			local_bh_enable();
J
Jason Wang 已提交
1695
			return NULL;
J
Jason Wang 已提交
1696 1697
		case XDP_PASS:
			delta = orig_data - xdp.data;
1698
			len = xdp.data_end - xdp.data;
J
Jason Wang 已提交
1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710
			break;
		default:
			bpf_warn_invalid_xdp_action(act);
			/* fall through */
		case XDP_ABORTED:
			trace_xdp_exception(tun->dev, xdp_prog, act);
			/* fall through */
		case XDP_DROP:
			goto err_xdp;
		}
	}

1711
	skb = build_skb(buf, buflen);
J
Jason Wang 已提交
1712 1713
	if (!skb) {
		rcu_read_unlock();
1714
		local_bh_enable();
1715
		return ERR_PTR(-ENOMEM);
J
Jason Wang 已提交
1716
	}
1717

1718
	skb_reserve(skb, pad - delta);
1719
	skb_put(skb, len);
1720 1721 1722
	get_page(alloc_frag->page);
	alloc_frag->offset += buflen;

J
Jason Wang 已提交
1723
	rcu_read_unlock();
1724
	local_bh_enable();
J
Jason Wang 已提交
1725

1726
	return skb;
J
Jason Wang 已提交
1727 1728 1729 1730 1731

err_redirect:
	put_page(alloc_frag->page);
err_xdp:
	rcu_read_unlock();
1732
	local_bh_enable();
J
Jason Wang 已提交
1733 1734
	this_cpu_inc(tun->pcpu_stats->rx_dropped);
	return NULL;
1735 1736
}

L
Linus Torvalds 已提交
1737
/* Get packet from user space buffer */
J
Jason Wang 已提交
1738
static ssize_t tun_get_user(struct tun_struct *tun, struct tun_file *tfile,
1739
			    void *msg_control, struct iov_iter *from,
J
Jason Wang 已提交
1740
			    int noblock, bool more)
L
Linus Torvalds 已提交
1741
{
1742
	struct tun_pi pi = { 0, cpu_to_be16(ETH_P_IP) };
L
Linus Torvalds 已提交
1743
	struct sk_buff *skb;
1744
	size_t total_len = iov_iter_count(from);
1745
	size_t len = total_len, align = tun->align, linear;
1746
	struct virtio_net_hdr gso = { 0 };
1747
	struct tun_pcpu_stats *stats;
1748
	int good_linear;
1749 1750 1751
	int copylen;
	bool zerocopy = false;
	int err;
1752
	u32 rxhash = 0;
J
Jason Wang 已提交
1753
	int skb_xdp = 1;
1754
	bool frags = tun_napi_frags_enabled(tun);
L
Linus Torvalds 已提交
1755

E
Eric Dumazet 已提交
1756 1757 1758
	if (!(tun->dev->flags & IFF_UP))
		return -EIO;

M
Michael S. Tsirkin 已提交
1759
	if (!(tun->flags & IFF_NO_PI)) {
1760
		if (len < sizeof(pi))
L
Linus Torvalds 已提交
1761
			return -EINVAL;
1762
		len -= sizeof(pi);
L
Linus Torvalds 已提交
1763

1764
		if (!copy_from_iter_full(&pi, sizeof(pi), from))
L
Linus Torvalds 已提交
1765 1766 1767
			return -EFAULT;
	}

M
Michael S. Tsirkin 已提交
1768
	if (tun->flags & IFF_VNET_HDR) {
W
Willem de Bruijn 已提交
1769 1770 1771
		int vnet_hdr_sz = READ_ONCE(tun->vnet_hdr_sz);

		if (len < vnet_hdr_sz)
1772
			return -EINVAL;
W
Willem de Bruijn 已提交
1773
		len -= vnet_hdr_sz;
1774

1775
		if (!copy_from_iter_full(&gso, sizeof(gso), from))
1776 1777
			return -EFAULT;

1778
		if ((gso.flags & VIRTIO_NET_HDR_F_NEEDS_CSUM) &&
1779 1780
		    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);
1781

1782
		if (tun16_to_cpu(tun, gso.hdr_len) > len)
1783
			return -EINVAL;
W
Willem de Bruijn 已提交
1784
		iov_iter_advance(from, vnet_hdr_sz - sizeof(gso));
1785 1786
	}

M
Michael S. Tsirkin 已提交
1787
	if ((tun->flags & TUN_TYPE_MASK) == IFF_TAP) {
1788
		align += NET_IP_ALIGN;
H
Herbert Xu 已提交
1789
		if (unlikely(len < ETH_HLEN ||
1790
			     (gso.hdr_len && tun16_to_cpu(tun, gso.hdr_len) < ETH_HLEN)))
1791 1792
			return -EINVAL;
	}
1793

1794 1795
	good_linear = SKB_MAX_HEAD(align);

1796
	if (msg_control) {
1797 1798
		struct iov_iter i = *from;

1799 1800
		/* There are 256 bytes to be copied in skb, so there is
		 * enough room for skb expand head in case it is used.
1801 1802
		 * The rest of the buffer is mapped from userspace.
		 */
1803
		copylen = gso.hdr_len ? tun16_to_cpu(tun, gso.hdr_len) : GOODCOPY_LEN;
1804 1805
		if (copylen > good_linear)
			copylen = good_linear;
1806
		linear = copylen;
1807 1808
		iov_iter_advance(&i, copylen);
		if (iov_iter_npages(&i, INT_MAX) <= MAX_SKB_FRAGS)
1809 1810 1811
			zerocopy = true;
	}

1812
	if (!frags && tun_can_build_skb(tun, tfile, len, noblock, zerocopy)) {
J
Jason Wang 已提交
1813 1814 1815 1816 1817
		/* 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);
1818
		if (IS_ERR(skb)) {
1819
			this_cpu_inc(tun->pcpu_stats->rx_dropped);
1820 1821
			return PTR_ERR(skb);
		}
J
Jason Wang 已提交
1822 1823
		if (!skb)
			return total_len;
1824 1825 1826 1827 1828 1829 1830 1831
	} 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 已提交
1832

1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845
		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);
		}

1846 1847 1848
		if (IS_ERR(skb)) {
			if (PTR_ERR(skb) != -EAGAIN)
				this_cpu_inc(tun->pcpu_stats->rx_dropped);
1849 1850
			if (frags)
				mutex_unlock(&tfile->napi_mutex);
1851 1852
			return PTR_ERR(skb);
		}
1853

1854 1855 1856 1857 1858 1859 1860 1861
		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);
1862 1863 1864 1865 1866
			if (frags) {
				tfile->napi.skb = NULL;
				mutex_unlock(&tfile->napi_mutex);
			}

1867 1868
			return -EFAULT;
		}
D
Dave Jones 已提交
1869
	}
L
Linus Torvalds 已提交
1870

1871
	if (virtio_net_hdr_to_skb(skb, &gso, tun_is_little_endian(tun))) {
1872 1873
		this_cpu_inc(tun->pcpu_stats->rx_frame_errors);
		kfree_skb(skb);
1874 1875 1876 1877 1878
		if (frags) {
			tfile->napi.skb = NULL;
			mutex_unlock(&tfile->napi_mutex);
		}

1879 1880 1881
		return -EINVAL;
	}

L
Linus Torvalds 已提交
1882
	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
1883 1884
	case IFF_TUN:
		if (tun->flags & IFF_NO_PI) {
1885 1886 1887 1888
			u8 ip_version = skb->len ? (skb->data[0] >> 4) : 0;

			switch (ip_version) {
			case 4:
1889 1890
				pi.proto = htons(ETH_P_IP);
				break;
1891
			case 6:
1892 1893 1894
				pi.proto = htons(ETH_P_IPV6);
				break;
			default:
1895
				this_cpu_inc(tun->pcpu_stats->rx_dropped);
1896 1897 1898 1899 1900
				kfree_skb(skb);
				return -EINVAL;
			}
		}

1901
		skb_reset_mac_header(skb);
L
Linus Torvalds 已提交
1902
		skb->protocol = pi.proto;
1903
		skb->dev = tun->dev;
L
Linus Torvalds 已提交
1904
		break;
M
Michael S. Tsirkin 已提交
1905
	case IFF_TAP:
1906 1907
		if (!frags)
			skb->protocol = eth_type_trans(skb, tun->dev);
L
Linus Torvalds 已提交
1908
		break;
1909
	}
L
Linus Torvalds 已提交
1910

1911 1912 1913 1914
	/* 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;
1915
		skb_shinfo(skb)->tx_flags |= SKBTX_SHARED_FRAG;
1916 1917 1918
	} else if (msg_control) {
		struct ubuf_info *uarg = msg_control;
		uarg->callback(uarg, false);
1919 1920
	}

1921
	skb_reset_network_header(skb);
1922
	skb_probe_transport_header(skb, 0);
1923

J
Jason Wang 已提交
1924
	if (skb_xdp) {
J
Jason Wang 已提交
1925 1926 1927
		struct bpf_prog *xdp_prog;
		int ret;

1928
		local_bh_disable();
J
Jason Wang 已提交
1929 1930 1931 1932 1933 1934
		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();
1935
				local_bh_enable();
J
Jason Wang 已提交
1936 1937 1938 1939
				return total_len;
			}
		}
		rcu_read_unlock();
1940
		local_bh_enable();
J
Jason Wang 已提交
1941 1942
	}

1943 1944 1945 1946 1947 1948
	/* 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)
1949
		rxhash = __skb_get_hash_symmetric(skb);
1950

1951 1952 1953 1954
	if (frags) {
		/* Exercise flow dissector code path. */
		u32 headlen = eth_get_headlen(skb->data, skb_headlen(skb));

1955
		if (unlikely(headlen > skb_headlen(skb))) {
1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966
			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);
1967
	} else if (tfile->napi_enabled) {
1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984
		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);
	}
1985

1986 1987 1988 1989 1990 1991
	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 已提交
1992

1993 1994 1995
	if (rxhash)
		tun_flow_update(tun, rxhash, tfile);

1996
	return total_len;
1997
}
L
Linus Torvalds 已提交
1998

1999
static ssize_t tun_chr_write_iter(struct kiocb *iocb, struct iov_iter *from)
L
Linus Torvalds 已提交
2000
{
2001
	struct file *file = iocb->ki_filp;
J
Jason Wang 已提交
2002
	struct tun_file *tfile = file->private_data;
2003
	struct tun_struct *tun = tun_get(tfile);
E
Eric W. Biederman 已提交
2004
	ssize_t result;
L
Linus Torvalds 已提交
2005 2006 2007 2008

	if (!tun)
		return -EBADFD;

J
Jason Wang 已提交
2009 2010
	result = tun_get_user(tun, tfile, NULL, from,
			      file->f_flags & O_NONBLOCK, false);
E
Eric W. Biederman 已提交
2011 2012 2013

	tun_put(tun);
	return result;
L
Linus Torvalds 已提交
2014 2015
}

J
Jason Wang 已提交
2016 2017
static ssize_t tun_put_user_xdp(struct tun_struct *tun,
				struct tun_file *tfile,
2018
				struct xdp_frame *xdp_frame,
J
Jason Wang 已提交
2019 2020 2021
				struct iov_iter *iter)
{
	int vnet_hdr_sz = 0;
2022
	size_t size = xdp_frame->len;
J
Jason Wang 已提交
2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037
	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));
	}

2038
	ret = copy_to_iter(xdp_frame->data, size, iter) + vnet_hdr_sz;
J
Jason Wang 已提交
2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049

	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 已提交
2050
/* Put packet to the user space buffer */
2051
static ssize_t tun_put_user(struct tun_struct *tun,
J
Jason Wang 已提交
2052
			    struct tun_file *tfile,
2053
			    struct sk_buff *skb,
H
Herbert Xu 已提交
2054
			    struct iov_iter *iter)
L
Linus Torvalds 已提交
2055 2056
{
	struct tun_pi pi = { 0, skb->protocol };
2057
	struct tun_pcpu_stats *stats;
H
Herbert Xu 已提交
2058
	ssize_t total;
2059
	int vlan_offset = 0;
2060
	int vlan_hlen = 0;
H
Herbert Xu 已提交
2061
	int vnet_hdr_sz = 0;
2062

2063
	if (skb_vlan_tag_present(skb))
2064
		vlan_hlen = VLAN_HLEN;
L
Linus Torvalds 已提交
2065

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

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

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

H
Herbert Xu 已提交
2075 2076
		total += sizeof(pi);
		if (iov_iter_count(iter) < total) {
L
Linus Torvalds 已提交
2077 2078 2079
			/* Packet will be striped */
			pi.flags |= TUN_PKT_STRIP;
		}
2080

H
Herbert Xu 已提交
2081
		if (copy_to_iter(&pi, sizeof(pi), iter) != sizeof(pi))
L
Linus Torvalds 已提交
2082
			return -EFAULT;
2083
	}
L
Linus Torvalds 已提交
2084

H
Herbert Xu 已提交
2085
	if (vnet_hdr_sz) {
2086
		struct virtio_net_hdr gso;
2087

H
Herbert Xu 已提交
2088
		if (iov_iter_count(iter) < vnet_hdr_sz)
2089 2090
			return -EINVAL;

2091
		if (virtio_net_hdr_from_skb(skb, &gso,
2092 2093
					    tun_is_little_endian(tun), true,
					    vlan_hlen)) {
2094
			struct skb_shared_info *sinfo = skb_shinfo(skb);
2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105
			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;
		}
2106

H
Herbert Xu 已提交
2107
		if (copy_to_iter(&gso, sizeof(gso), iter) != sizeof(gso))
2108
			return -EFAULT;
2109 2110

		iov_iter_advance(iter, vnet_hdr_sz - sizeof(gso));
2111 2112
	}

2113
	if (vlan_hlen) {
H
Herbert Xu 已提交
2114
		int ret;
2115
		struct veth veth;
J
Jason Wang 已提交
2116 2117

		veth.h_vlan_proto = skb->vlan_proto;
2118
		veth.h_vlan_TCI = htons(skb_vlan_tag_get(skb));
J
Jason Wang 已提交
2119 2120 2121

		vlan_offset = offsetof(struct vlan_ethhdr, h_vlan_proto);

H
Herbert Xu 已提交
2122 2123
		ret = skb_copy_datagram_iter(skb, 0, iter, vlan_offset);
		if (ret || !iov_iter_count(iter))
J
Jason Wang 已提交
2124 2125
			goto done;

H
Herbert Xu 已提交
2126 2127
		ret = copy_to_iter(&veth, sizeof(veth), iter);
		if (ret != sizeof(veth) || !iov_iter_count(iter))
J
Jason Wang 已提交
2128 2129
			goto done;
	}
L
Linus Torvalds 已提交
2130

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

J
Jason Wang 已提交
2133
done:
2134 2135 2136 2137 2138 2139 2140
	/* 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 已提交
2141 2142 2143 2144

	return total;
}

J
Jason Wang 已提交
2145
static void *tun_ring_recv(struct tun_file *tfile, int noblock, int *err)
J
Jason Wang 已提交
2146 2147
{
	DECLARE_WAITQUEUE(wait, current);
J
Jason Wang 已提交
2148
	void *ptr = NULL;
J
Jason Wang 已提交
2149
	int error = 0;
J
Jason Wang 已提交
2150

J
Jason Wang 已提交
2151 2152
	ptr = ptr_ring_consume(&tfile->tx_ring);
	if (ptr)
J
Jason Wang 已提交
2153 2154
		goto out;
	if (noblock) {
J
Jason Wang 已提交
2155
		error = -EAGAIN;
J
Jason Wang 已提交
2156 2157 2158 2159 2160 2161 2162
		goto out;
	}

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

	while (1) {
J
Jason Wang 已提交
2163 2164
		ptr = ptr_ring_consume(&tfile->tx_ring);
		if (ptr)
J
Jason Wang 已提交
2165 2166
			break;
		if (signal_pending(current)) {
J
Jason Wang 已提交
2167
			error = -ERESTARTSYS;
J
Jason Wang 已提交
2168 2169 2170
			break;
		}
		if (tfile->socket.sk->sk_shutdown & RCV_SHUTDOWN) {
J
Jason Wang 已提交
2171
			error = -EFAULT;
J
Jason Wang 已提交
2172 2173 2174 2175 2176 2177 2178 2179 2180 2181
			break;
		}

		schedule();
	}

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

out:
J
Jason Wang 已提交
2182
	*err = error;
J
Jason Wang 已提交
2183
	return ptr;
J
Jason Wang 已提交
2184 2185
}

J
Jason Wang 已提交
2186
static ssize_t tun_do_read(struct tun_struct *tun, struct tun_file *tfile,
2187
			   struct iov_iter *to,
J
Jason Wang 已提交
2188
			   int noblock, void *ptr)
L
Linus Torvalds 已提交
2189
{
2190
	ssize_t ret;
J
Jason Wang 已提交
2191
	int err;
L
Linus Torvalds 已提交
2192

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

W
Wei Xu 已提交
2195
	if (!iov_iter_count(to)) {
J
Jason Wang 已提交
2196
		tun_ptr_free(ptr);
2197
		return 0;
W
Wei Xu 已提交
2198
	}
L
Linus Torvalds 已提交
2199

J
Jason Wang 已提交
2200
	if (!ptr) {
2201
		/* Read frames from ring */
J
Jason Wang 已提交
2202 2203
		ptr = tun_ring_recv(tfile, noblock, &err);
		if (!ptr)
2204 2205
			return err;
	}
H
Herbert Xu 已提交
2206

2207 2208
	if (tun_is_xdp_frame(ptr)) {
		struct xdp_frame *xdpf = tun_ptr_to_xdp(ptr);
J
Jason Wang 已提交
2209

2210
		ret = tun_put_user_xdp(tun, tfile, xdpf, to);
2211
		xdp_return_frame(xdpf);
J
Jason Wang 已提交
2212 2213 2214 2215 2216 2217 2218 2219 2220
	} 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 已提交
2221

2222 2223 2224
	return ret;
}

2225
static ssize_t tun_chr_read_iter(struct kiocb *iocb, struct iov_iter *to)
2226 2227 2228
{
	struct file *file = iocb->ki_filp;
	struct tun_file *tfile = file->private_data;
2229
	struct tun_struct *tun = tun_get(tfile);
2230
	ssize_t len = iov_iter_count(to), ret;
2231 2232 2233

	if (!tun)
		return -EBADFD;
2234
	ret = tun_do_read(tun, tfile, to, file->f_flags & O_NONBLOCK, NULL);
2235
	ret = min_t(ssize_t, ret, len);
Z
Zhi Yong Wu 已提交
2236 2237
	if (ret > 0)
		iocb->ki_pos = ret;
E
Eric W. Biederman 已提交
2238
	tun_put(tun);
L
Linus Torvalds 已提交
2239 2240 2241
	return ret;
}

2242
static void tun_prog_free(struct rcu_head *rcu)
2243
{
2244
	struct tun_prog *prog = container_of(rcu, struct tun_prog, rcu);
2245 2246 2247 2248 2249

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

J
Jason Wang 已提交
2250 2251
static int __tun_set_ebpf(struct tun_struct *tun,
			  struct tun_prog __rcu **prog_p,
2252
			  struct bpf_prog *prog)
2253
{
2254
	struct tun_prog *old, *new = NULL;
2255 2256 2257 2258 2259 2260 2261 2262

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

2263
	spin_lock_bh(&tun->lock);
2264
	old = rcu_dereference_protected(*prog_p,
2265
					lockdep_is_held(&tun->lock));
2266
	rcu_assign_pointer(*prog_p, new);
2267
	spin_unlock_bh(&tun->lock);
2268 2269

	if (old)
2270
		call_rcu(&old->rcu, tun_prog_free);
2271 2272 2273 2274

	return 0;
}

J
Jason Wang 已提交
2275 2276 2277 2278
static void tun_free_netdev(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

J
Jason Wang 已提交
2279
	BUG_ON(!(list_empty(&tun->disabled)));
2280
	free_percpu(tun->pcpu_stats);
J
Jason Wang 已提交
2281
	tun_flow_uninit(tun);
2282
	security_tun_dev_free_security(tun->security);
2283
	__tun_set_ebpf(tun, &tun->steering_prog, NULL);
2284
	__tun_set_ebpf(tun, &tun->filter_prog, NULL);
J
Jason Wang 已提交
2285 2286
}

L
Linus Torvalds 已提交
2287 2288 2289 2290
static void tun_setup(struct net_device *dev)
{
	struct tun_struct *tun = netdev_priv(dev);

2291 2292
	tun->owner = INVALID_UID;
	tun->group = INVALID_GID;
2293
	tun_default_link_ksettings(dev, &tun->link_ksettings);
L
Linus Torvalds 已提交
2294 2295

	dev->ethtool_ops = &tun_ethtool_ops;
2296 2297
	dev->needs_free_netdev = true;
	dev->priv_destructor = tun_free_netdev;
J
Jason Wang 已提交
2298 2299
	/* We prefer our own queue length */
	dev->tx_queue_len = TUN_READQ_SIZE;
L
Linus Torvalds 已提交
2300 2301
}

2302 2303 2304
/* Trivial set of netlink ops to allow deleting tun or tap
 * device with netlink.
 */
2305 2306
static int tun_validate(struct nlattr *tb[], struct nlattr *data[],
			struct netlink_ext_ack *extack)
2307 2308 2309 2310
{
	return -EINVAL;
}

2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364
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;
}

2365 2366 2367 2368 2369
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,
2370 2371
	.get_size       = tun_get_size,
	.fill_info      = tun_fill_info,
2372 2373
};

2374 2375
static void tun_sock_write_space(struct sock *sk)
{
J
Jason Wang 已提交
2376
	struct tun_file *tfile;
2377
	wait_queue_head_t *wqueue;
2378 2379 2380 2381

	if (!sock_writeable(sk))
		return;

2382
	if (!test_and_clear_bit(SOCKWQ_ASYNC_NOSPACE, &sk->sk_socket->flags))
2383 2384
		return;

2385 2386
	wqueue = sk_sleep(sk);
	if (wqueue && waitqueue_active(wqueue))
2387 2388
		wake_up_interruptible_sync_poll(wqueue, EPOLLOUT |
						EPOLLWRNORM | EPOLLWRBAND);
H
Herbert Xu 已提交
2389

J
Jason Wang 已提交
2390 2391
	tfile = container_of(sk, struct tun_file, sk);
	kill_fasync(&tfile->fasync, SIGIO, POLL_OUT);
2392 2393
}

2394
static int tun_sendmsg(struct socket *sock, struct msghdr *m, size_t total_len)
2395
{
J
Jason Wang 已提交
2396 2397
	int ret;
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2398
	struct tun_struct *tun = tun_get(tfile);
J
Jason Wang 已提交
2399 2400 2401

	if (!tun)
		return -EBADFD;
2402

A
Al Viro 已提交
2403
	ret = tun_get_user(tun, tfile, m->msg_control, &m->msg_iter,
J
Jason Wang 已提交
2404 2405
			   m->msg_flags & MSG_DONTWAIT,
			   m->msg_flags & MSG_MORE);
J
Jason Wang 已提交
2406 2407
	tun_put(tun);
	return ret;
2408 2409
}

2410
static int tun_recvmsg(struct socket *sock, struct msghdr *m, size_t total_len,
2411 2412
		       int flags)
{
J
Jason Wang 已提交
2413
	struct tun_file *tfile = container_of(sock, struct tun_file, socket);
2414
	struct tun_struct *tun = tun_get(tfile);
J
Jason Wang 已提交
2415
	void *ptr = m->msg_control;
2416
	int ret;
J
Jason Wang 已提交
2417

W
Wei Xu 已提交
2418 2419
	if (!tun) {
		ret = -EBADFD;
J
Jason Wang 已提交
2420
		goto out_free;
W
Wei Xu 已提交
2421
	}
J
Jason Wang 已提交
2422

2423
	if (flags & ~(MSG_DONTWAIT|MSG_TRUNC|MSG_ERRQUEUE)) {
2424
		ret = -EINVAL;
W
Wei Xu 已提交
2425
		goto out_put_tun;
2426
	}
2427 2428 2429 2430 2431
	if (flags & MSG_ERRQUEUE) {
		ret = sock_recv_errqueue(sock->sk, m, total_len,
					 SOL_PACKET, TUN_TX_TIMESTAMP);
		goto out;
	}
J
Jason Wang 已提交
2432
	ret = tun_do_read(tun, tfile, &m->msg_iter, flags & MSG_DONTWAIT, ptr);
2433
	if (ret > (ssize_t)total_len) {
2434 2435 2436
		m->msg_flags |= MSG_TRUNC;
		ret = flags & MSG_TRUNC ? ret : total_len;
	}
2437
out:
J
Jason Wang 已提交
2438
	tun_put(tun);
2439
	return ret;
W
Wei Xu 已提交
2440 2441 2442

out_put_tun:
	tun_put(tun);
J
Jason Wang 已提交
2443 2444
out_free:
	tun_ptr_free(ptr);
W
Wei Xu 已提交
2445
	return ret;
2446 2447
}

J
Jason Wang 已提交
2448 2449 2450
static int tun_ptr_peek_len(void *ptr)
{
	if (likely(ptr)) {
2451 2452
		if (tun_is_xdp_frame(ptr)) {
			struct xdp_frame *xdpf = tun_ptr_to_xdp(ptr);
J
Jason Wang 已提交
2453

2454
			return xdpf->len;
J
Jason Wang 已提交
2455 2456 2457 2458 2459 2460 2461
		}
		return __skb_array_len_with_tag(ptr);
	} else {
		return 0;
	}
}

J
Jason Wang 已提交
2462 2463 2464 2465 2466 2467
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;

2468
	tun = tun_get(tfile);
J
Jason Wang 已提交
2469 2470 2471
	if (!tun)
		return 0;

J
Jason Wang 已提交
2472
	ret = PTR_RING_PEEK_CALL(&tfile->tx_ring, tun_ptr_peek_len);
J
Jason Wang 已提交
2473 2474 2475 2476 2477
	tun_put(tun);

	return ret;
}

2478 2479
/* Ops structure to mimic raw sockets with tun */
static const struct proto_ops tun_socket_ops = {
J
Jason Wang 已提交
2480
	.peek_len = tun_peek_len,
2481 2482 2483 2484
	.sendmsg = tun_sendmsg,
	.recvmsg = tun_recvmsg,
};

2485 2486 2487
static struct proto tun_proto = {
	.name		= "tun",
	.owner		= THIS_MODULE,
J
Jason Wang 已提交
2488
	.obj_size	= sizeof(struct tun_file),
2489
};
2490

2491 2492
static int tun_flags(struct tun_struct *tun)
{
2493
	return tun->flags & (TUN_FEATURES | IFF_PERSIST | IFF_TUN | IFF_TAP);
2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506
}

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));
2507 2508 2509 2510
	return uid_valid(tun->owner)?
		sprintf(buf, "%u\n",
			from_kuid_munged(current_user_ns(), tun->owner)):
		sprintf(buf, "-1\n");
2511 2512 2513 2514 2515 2516
}

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));
2517 2518 2519 2520
	return gid_valid(tun->group) ?
		sprintf(buf, "%u\n",
			from_kgid_munged(current_user_ns(), tun->group)):
		sprintf(buf, "-1\n");
2521 2522 2523 2524 2525 2526
}

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

2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537
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
};

2538
static int tun_set_iff(struct net *net, struct file *file, struct ifreq *ifr)
L
Linus Torvalds 已提交
2539 2540
{
	struct tun_struct *tun;
J
Jason Wang 已提交
2541
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
2542 2543 2544
	struct net_device *dev;
	int err;

2545 2546 2547
	if (tfile->detached)
		return -EINVAL;

2548 2549 2550 2551 2552 2553 2554 2555 2556
	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;
	}

2557 2558
	dev = __dev_get_by_name(net, ifr->ifr_name);
	if (dev) {
2559 2560
		if (ifr->ifr_flags & IFF_TUN_EXCL)
			return -EBUSY;
2561 2562 2563 2564 2565 2566 2567
		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;

2568
		if (!!(ifr->ifr_flags & IFF_MULTI_QUEUE) !=
M
Michael S. Tsirkin 已提交
2569
		    !!(tun->flags & IFF_MULTI_QUEUE))
2570 2571
			return -EINVAL;

2572
		if (tun_not_capable(tun))
P
Paul Moore 已提交
2573
			return -EPERM;
2574
		err = security_tun_dev_open(tun->security);
P
Paul Moore 已提交
2575 2576 2577
		if (err < 0)
			return err;

2578 2579
		err = tun_attach(tun, file, ifr->ifr_flags & IFF_NOFILTER,
				 ifr->ifr_flags & IFF_NAPI);
2580 2581
		if (err < 0)
			return err;
J
Jason Wang 已提交
2582

M
Michael S. Tsirkin 已提交
2583
		if (tun->flags & IFF_MULTI_QUEUE &&
2584 2585 2586 2587
		    (tun->numqueues + tun->numdisabled > 1)) {
			/* One or more queue has already been attached, no need
			 * to initialize the device again.
			 */
2588
			netdev_state_change(dev);
2589 2590
			return 0;
		}
2591 2592 2593

		tun->flags = (tun->flags & ~TUN_FEATURES) |
			      (ifr->ifr_flags & TUN_FEATURES);
2594 2595 2596

		netdev_state_change(dev);
	} else {
L
Linus Torvalds 已提交
2597 2598
		char *name;
		unsigned long flags = 0;
2599 2600
		int queues = ifr->ifr_flags & IFF_MULTI_QUEUE ?
			     MAX_TAP_QUEUES : 1;
L
Linus Torvalds 已提交
2601

2602
		if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
2603
			return -EPERM;
P
Paul Moore 已提交
2604 2605 2606
		err = security_tun_dev_create();
		if (err < 0)
			return err;
2607

L
Linus Torvalds 已提交
2608 2609 2610
		/* Set dev type */
		if (ifr->ifr_flags & IFF_TUN) {
			/* TUN device */
M
Michael S. Tsirkin 已提交
2611
			flags |= IFF_TUN;
L
Linus Torvalds 已提交
2612 2613 2614
			name = "tun%d";
		} else if (ifr->ifr_flags & IFF_TAP) {
			/* TAP device */
M
Michael S. Tsirkin 已提交
2615
			flags |= IFF_TAP;
L
Linus Torvalds 已提交
2616
			name = "tap%d";
2617
		} else
2618
			return -EINVAL;
2619

L
Linus Torvalds 已提交
2620 2621 2622
		if (*ifr->ifr_name)
			name = ifr->ifr_name;

J
Jason Wang 已提交
2623
		dev = alloc_netdev_mqs(sizeof(struct tun_struct), name,
2624 2625
				       NET_NAME_UNKNOWN, tun_setup, queues,
				       queues);
2626

L
Linus Torvalds 已提交
2627 2628
		if (!dev)
			return -ENOMEM;
2629
		err = dev_get_valid_name(net, dev, name);
2630
		if (err < 0)
2631
			goto err_free_dev;
L
Linus Torvalds 已提交
2632

2633
		dev_net_set(dev, net);
2634
		dev->rtnl_link_ops = &tun_link_ops;
2635
		dev->ifindex = tfile->ifindex;
2636
		dev->sysfs_groups[0] = &tun_attr_group;
S
Stephen Hemminger 已提交
2637

L
Linus Torvalds 已提交
2638 2639 2640
		tun = netdev_priv(dev);
		tun->dev = dev;
		tun->flags = flags;
2641
		tun->txflt.count = 0;
2642
		tun->vnet_hdr_sz = sizeof(struct virtio_net_hdr);
2643

2644
		tun->align = NET_SKB_PAD;
J
Jason Wang 已提交
2645 2646
		tun->filter_attached = false;
		tun->sndbuf = tfile->socket.sk->sk_sndbuf;
J
Jason Wang 已提交
2647
		tun->rx_batched = 0;
2648
		RCU_INIT_POINTER(tun->steering_prog, NULL);
2649

2650 2651 2652 2653 2654 2655
		tun->pcpu_stats = netdev_alloc_pcpu_stats(struct tun_pcpu_stats);
		if (!tun->pcpu_stats) {
			err = -ENOMEM;
			goto err_free_dev;
		}

J
Jason Wang 已提交
2656 2657
		spin_lock_init(&tun->lock);

2658 2659
		err = security_tun_dev_alloc_security(&tun->security);
		if (err < 0)
2660
			goto err_free_stat;
P
Paul Moore 已提交
2661

L
Linus Torvalds 已提交
2662
		tun_net_init(dev);
2663
		tun_flow_init(tun);
J
Jason Wang 已提交
2664

2665
		dev->hw_features = NETIF_F_SG | NETIF_F_FRAGLIST |
J
Jason Wang 已提交
2666 2667
				   TUN_USER_FEATURES | NETIF_F_HW_VLAN_CTAG_TX |
				   NETIF_F_HW_VLAN_STAG_TX;
2668
		dev->features = dev->hw_features | NETIF_F_LLTX;
2669 2670 2671
		dev->vlan_features = dev->features &
				     ~(NETIF_F_HW_VLAN_CTAG_TX |
				       NETIF_F_HW_VLAN_STAG_TX);
2672

2673 2674 2675
		tun->flags = (tun->flags & ~TUN_FEATURES) |
			      (ifr->ifr_flags & TUN_FEATURES);

J
Jason Wang 已提交
2676
		INIT_LIST_HEAD(&tun->disabled);
2677
		err = tun_attach(tun, file, false, ifr->ifr_flags & IFF_NAPI);
2678
		if (err < 0)
2679
			goto err_free_flow;
2680

L
Linus Torvalds 已提交
2681 2682
		err = register_netdevice(tun->dev);
		if (err < 0)
2683
			goto err_detach;
L
Linus Torvalds 已提交
2684 2685
	}

2686 2687
	netif_carrier_on(tun->dev);

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

2690 2691 2692 2693
	/* Make sure persistent devices do not get stuck in
	 * xoff state.
	 */
	if (netif_running(tun->dev))
J
Jason Wang 已提交
2694
		netif_tx_wake_all_queues(tun->dev);
2695

L
Linus Torvalds 已提交
2696 2697 2698
	strcpy(ifr->ifr_name, tun->dev->name);
	return 0;

2699 2700
err_detach:
	tun_detach_all(dev);
2701 2702 2703
	/* register_netdevice() already called tun_free_netdev() */
	goto err_free_dev;

2704 2705 2706
err_free_flow:
	tun_flow_uninit(tun);
	security_tun_dev_free_security(tun->security);
2707 2708
err_free_stat:
	free_percpu(tun->pcpu_stats);
2709
err_free_dev:
L
Linus Torvalds 已提交
2710 2711 2712 2713
	free_netdev(dev);
	return err;
}

R
Rami Rosen 已提交
2714
static void tun_get_iff(struct net *net, struct tun_struct *tun,
2715
		       struct ifreq *ifr)
2716
{
2717
	tun_debug(KERN_INFO, tun, "tun_get_iff\n");
2718 2719 2720

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

2721
	ifr->ifr_flags = tun_flags(tun);
2722 2723 2724

}

2725 2726
/* This is like a cut-down ethtool ops, except done via tun fd so no
 * privs required. */
2727
static int set_offload(struct tun_struct *tun, unsigned long arg)
2728
{
2729
	netdev_features_t features = 0;
2730 2731

	if (arg & TUN_F_CSUM) {
2732
		features |= NETIF_F_HW_CSUM;
2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745
		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);
		}
2746 2747

		arg &= ~TUN_F_UFO;
2748 2749 2750 2751 2752 2753 2754
	}

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

2755
	tun->set_features = features;
2756 2757
	tun->dev->wanted_features &= ~TUN_USER_FEATURES;
	tun->dev->wanted_features |= features;
2758
	netdev_update_features(tun->dev);
2759 2760 2761 2762

	return 0;
}

J
Jason Wang 已提交
2763 2764 2765 2766 2767 2768
static void tun_detach_filter(struct tun_struct *tun, int n)
{
	int i;
	struct tun_file *tfile;

	for (i = 0; i < n; i++) {
2769
		tfile = rtnl_dereference(tun->tfiles[i]);
2770 2771 2772
		lock_sock(tfile->socket.sk);
		sk_detach_filter(tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2773 2774 2775 2776 2777 2778 2779 2780 2781 2782 2783
	}

	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++) {
2784
		tfile = rtnl_dereference(tun->tfiles[i]);
2785 2786 2787
		lock_sock(tfile->socket.sk);
		ret = sk_attach_filter(&tun->fprog, tfile->socket.sk);
		release_sock(tfile->socket.sk);
J
Jason Wang 已提交
2788 2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803
		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++) {
2804
		tfile = rtnl_dereference(tun->tfiles[i]);
J
Jason Wang 已提交
2805 2806 2807 2808
		tfile->socket.sk->sk_sndbuf = tun->sndbuf;
	}
}

2809 2810 2811 2812 2813 2814 2815 2816 2817
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 已提交
2818
		tun = tfile->detached;
2819
		if (!tun) {
2820
			ret = -EINVAL;
2821 2822 2823 2824 2825
			goto unlock;
		}
		ret = security_tun_dev_attach_queue(tun->security);
		if (ret < 0)
			goto unlock;
2826
		ret = tun_attach(tun, file, false, tun->flags & IFF_NAPI);
J
Jason Wang 已提交
2827
	} else if (ifr->ifr_flags & IFF_DETACH_QUEUE) {
2828
		tun = rtnl_dereference(tfile->tun);
M
Michael S. Tsirkin 已提交
2829
		if (!tun || !(tun->flags & IFF_MULTI_QUEUE) || tfile->detached)
J
Jason Wang 已提交
2830 2831 2832 2833
			ret = -EINVAL;
		else
			__tun_detach(tfile, false);
	} else
2834 2835
		ret = -EINVAL;

2836 2837 2838
	if (ret >= 0)
		netdev_state_change(tun->dev);

2839
unlock:
2840 2841 2842 2843
	rtnl_unlock();
	return ret;
}

2844 2845
static int tun_set_ebpf(struct tun_struct *tun, struct tun_prog **prog_p,
			void __user *data)
2846 2847 2848 2849 2850 2851 2852 2853 2854 2855 2856 2857 2858 2859 2860
{
	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);
	}

2861
	return __tun_set_ebpf(tun, prog_p, prog);
2862 2863
}

2864 2865
static long __tun_chr_ioctl(struct file *file, unsigned int cmd,
			    unsigned long arg, int ifreq_len)
L
Linus Torvalds 已提交
2866
{
E
Eric W. Biederman 已提交
2867
	struct tun_file *tfile = file->private_data;
2868
	struct net *net = sock_net(&tfile->sk);
E
Eric W. Biederman 已提交
2869
	struct tun_struct *tun;
L
Linus Torvalds 已提交
2870 2871
	void __user* argp = (void __user*)arg;
	struct ifreq ifr;
2872 2873
	kuid_t owner;
	kgid_t group;
2874
	int sndbuf;
2875
	int vnet_hdr_sz;
2876
	unsigned int ifindex;
M
Michael S. Tsirkin 已提交
2877
	int le;
2878
	int ret;
2879
	bool do_notify = false;
L
Linus Torvalds 已提交
2880

2881 2882
	if (cmd == TUNSETIFF || cmd == TUNSETQUEUE ||
	    (_IOC_TYPE(cmd) == SOCK_IOC_TYPE && cmd != SIOCGSKNS)) {
2883
		if (copy_from_user(&ifr, argp, ifreq_len))
L
Linus Torvalds 已提交
2884
			return -EFAULT;
D
David S. Miller 已提交
2885
	} else {
2886
		memset(&ifr, 0, sizeof(ifr));
D
David S. Miller 已提交
2887
	}
E
Eric W. Biederman 已提交
2888 2889 2890
	if (cmd == TUNGETFEATURES) {
		/* Currently this just means: "what IFF flags are valid?".
		 * This is needed because we never checked for invalid flags on
2891 2892 2893
		 * TUNSETIFF.
		 */
		return put_user(IFF_TUN | IFF_TAP | TUN_FEATURES,
E
Eric W. Biederman 已提交
2894
				(unsigned int __user*)argp);
2895
	} else if (cmd == TUNSETQUEUE) {
2896
		return tun_set_queue(file, &ifr);
2897 2898 2899 2900 2901
	} 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 已提交
2902

J
Jason Wang 已提交
2903
	ret = 0;
2904 2905
	rtnl_lock();

2906
	tun = tun_get(tfile);
2907 2908 2909 2910 2911
	if (cmd == TUNSETIFF) {
		ret = -EEXIST;
		if (tun)
			goto unlock;

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

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

2916 2917
		if (ret)
			goto unlock;
L
Linus Torvalds 已提交
2918

2919
		if (copy_to_user(argp, &ifr, ifreq_len))
2920 2921
			ret = -EFAULT;
		goto unlock;
L
Linus Torvalds 已提交
2922
	}
2923 2924 2925 2926 2927 2928 2929 2930 2931 2932 2933 2934 2935
	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 已提交
2936

2937
	ret = -EBADFD;
L
Linus Torvalds 已提交
2938
	if (!tun)
2939
		goto unlock;
L
Linus Torvalds 已提交
2940

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

E
Eric W. Biederman 已提交
2943
	ret = 0;
L
Linus Torvalds 已提交
2944
	switch (cmd) {
2945
	case TUNGETIFF:
R
Rami Rosen 已提交
2946
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
2947

2948 2949
		if (tfile->detached)
			ifr.ifr_flags |= IFF_DETACH_QUEUE;
2950 2951
		if (!tfile->socket.sk->sk_filter)
			ifr.ifr_flags |= IFF_NOFILTER;
2952

2953
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
2954
			ret = -EFAULT;
2955 2956
		break;

L
Linus Torvalds 已提交
2957 2958 2959
	case TUNSETNOCSUM:
		/* Disable/Enable checksum */

2960 2961
		/* [unimplemented] */
		tun_debug(KERN_INFO, tun, "ignored: set checksum %s\n",
2962
			  arg ? "disabled" : "enabled");
L
Linus Torvalds 已提交
2963 2964 2965
		break;

	case TUNSETPERSIST:
J
Jason Wang 已提交
2966 2967 2968
		/* Disable/Enable persist mode. Keep an extra reference to the
		 * module to prevent the module being unprobed.
		 */
M
Michael S. Tsirkin 已提交
2969 2970
		if (arg && !(tun->flags & IFF_PERSIST)) {
			tun->flags |= IFF_PERSIST;
J
Jason Wang 已提交
2971
			__module_get(THIS_MODULE);
2972
			do_notify = true;
J
Jason Wang 已提交
2973
		}
M
Michael S. Tsirkin 已提交
2974 2975
		if (!arg && (tun->flags & IFF_PERSIST)) {
			tun->flags &= ~IFF_PERSIST;
J
Jason Wang 已提交
2976
			module_put(THIS_MODULE);
2977
			do_notify = true;
J
Jason Wang 已提交
2978
		}
L
Linus Torvalds 已提交
2979

2980 2981
		tun_debug(KERN_INFO, tun, "persist %s\n",
			  arg ? "enabled" : "disabled");
L
Linus Torvalds 已提交
2982 2983 2984 2985
		break;

	case TUNSETOWNER:
		/* Set owner of the device */
2986 2987 2988 2989 2990 2991
		owner = make_kuid(current_user_ns(), arg);
		if (!uid_valid(owner)) {
			ret = -EINVAL;
			break;
		}
		tun->owner = owner;
2992
		do_notify = true;
2993
		tun_debug(KERN_INFO, tun, "owner set to %u\n",
2994
			  from_kuid(&init_user_ns, tun->owner));
L
Linus Torvalds 已提交
2995 2996
		break;

2997 2998
	case TUNSETGROUP:
		/* Set group of the device */
2999 3000 3001 3002 3003 3004
		group = make_kgid(current_user_ns(), arg);
		if (!gid_valid(group)) {
			ret = -EINVAL;
			break;
		}
		tun->group = group;
3005
		do_notify = true;
3006
		tun_debug(KERN_INFO, tun, "group set to %u\n",
3007
			  from_kgid(&init_user_ns, tun->group));
3008 3009
		break;

3010 3011 3012
	case TUNSETLINK:
		/* Only allow setting the type when the interface is down */
		if (tun->dev->flags & IFF_UP) {
3013 3014
			tun_debug(KERN_INFO, tun,
				  "Linktype set failed because interface is up\n");
3015
			ret = -EBUSY;
3016 3017
		} else {
			tun->dev->type = (int) arg;
3018 3019
			tun_debug(KERN_INFO, tun, "linktype set to %d\n",
				  tun->dev->type);
3020
			ret = 0;
3021
		}
E
Eric W. Biederman 已提交
3022
		break;
3023

L
Linus Torvalds 已提交
3024 3025 3026 3027 3028
#ifdef TUN_DEBUG
	case TUNSETDEBUG:
		tun->debug = arg;
		break;
#endif
3029
	case TUNSETOFFLOAD:
3030
		ret = set_offload(tun, arg);
E
Eric W. Biederman 已提交
3031
		break;
3032

3033 3034
	case TUNSETTXFILTER:
		/* Can be set only for TAPs */
E
Eric W. Biederman 已提交
3035
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3036
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
E
Eric W. Biederman 已提交
3037
			break;
H
Harvey Harrison 已提交
3038
		ret = update_filter(&tun->txflt, (void __user *)arg);
E
Eric W. Biederman 已提交
3039
		break;
L
Linus Torvalds 已提交
3040 3041

	case SIOCGIFHWADDR:
3042
		/* Get hw address */
3043 3044
		memcpy(ifr.ifr_hwaddr.sa_data, tun->dev->dev_addr, ETH_ALEN);
		ifr.ifr_hwaddr.sa_family = tun->dev->type;
3045
		if (copy_to_user(argp, &ifr, ifreq_len))
E
Eric W. Biederman 已提交
3046 3047
			ret = -EFAULT;
		break;
L
Linus Torvalds 已提交
3048 3049

	case SIOCSIFHWADDR:
3050
		/* Set hw address */
3051 3052
		tun_debug(KERN_DEBUG, tun, "set hw address: %pM\n",
			  ifr.ifr_hwaddr.sa_data);
3053 3054

		ret = dev_set_mac_address(tun->dev, &ifr.ifr_hwaddr);
E
Eric W. Biederman 已提交
3055
		break;
3056 3057

	case TUNGETSNDBUF:
J
Jason Wang 已提交
3058
		sndbuf = tfile->socket.sk->sk_sndbuf;
3059 3060 3061 3062 3063 3064 3065 3066 3067
		if (copy_to_user(argp, &sndbuf, sizeof(sndbuf)))
			ret = -EFAULT;
		break;

	case TUNSETSNDBUF:
		if (copy_from_user(&sndbuf, argp, sizeof(sndbuf))) {
			ret = -EFAULT;
			break;
		}
3068 3069 3070 3071
		if (sndbuf <= 0) {
			ret = -EINVAL;
			break;
		}
3072

J
Jason Wang 已提交
3073 3074
		tun->sndbuf = sndbuf;
		tun_set_sndbuf(tun);
3075 3076
		break;

3077 3078 3079 3080 3081 3082 3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094 3095
	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 已提交
3096 3097 3098 3099 3100 3101 3102 3103 3104 3105 3106 3107 3108 3109 3110 3111 3112
	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;

3113 3114 3115 3116 3117 3118 3119 3120
	case TUNGETVNETBE:
		ret = tun_get_vnet_be(tun, argp);
		break;

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

M
Michael S. Tsirkin 已提交
3121 3122 3123
	case TUNATTACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3124
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
3125 3126
			break;
		ret = -EFAULT;
J
Jason Wang 已提交
3127
		if (copy_from_user(&tun->fprog, argp, sizeof(tun->fprog)))
M
Michael S. Tsirkin 已提交
3128 3129
			break;

J
Jason Wang 已提交
3130
		ret = tun_attach_filter(tun);
M
Michael S. Tsirkin 已提交
3131 3132 3133 3134 3135
		break;

	case TUNDETACHFILTER:
		/* Can be set only for TAPs */
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3136
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
M
Michael S. Tsirkin 已提交
3137
			break;
J
Jason Wang 已提交
3138 3139
		ret = 0;
		tun_detach_filter(tun, tun->numqueues);
M
Michael S. Tsirkin 已提交
3140 3141
		break;

P
Pavel Emelyanov 已提交
3142 3143
	case TUNGETFILTER:
		ret = -EINVAL;
M
Michael S. Tsirkin 已提交
3144
		if ((tun->flags & TUN_TYPE_MASK) != IFF_TAP)
P
Pavel Emelyanov 已提交
3145 3146 3147 3148 3149 3150 3151
			break;
		ret = -EFAULT;
		if (copy_to_user(argp, &tun->fprog, sizeof(tun->fprog)))
			break;
		ret = 0;
		break;

3152
	case TUNSETSTEERINGEBPF:
3153
		ret = tun_set_ebpf(tun, &tun->steering_prog, argp);
3154 3155
		break;

3156 3157 3158 3159
	case TUNSETFILTEREBPF:
		ret = tun_set_ebpf(tun, &tun->filter_prog, argp);
		break;

L
Linus Torvalds 已提交
3160
	default:
E
Eric W. Biederman 已提交
3161 3162
		ret = -EINVAL;
		break;
3163
	}
L
Linus Torvalds 已提交
3164

3165 3166 3167
	if (do_notify)
		netdev_state_change(tun->dev);

3168 3169 3170 3171
unlock:
	rtnl_unlock();
	if (tun)
		tun_put(tun);
E
Eric W. Biederman 已提交
3172
	return ret;
L
Linus Torvalds 已提交
3173 3174
}

3175 3176 3177 3178 3179 3180 3181 3182 3183 3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196 3197 3198 3199 3200 3201 3202 3203 3204 3205 3206 3207 3208 3209
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 已提交
3210 3211
static int tun_chr_fasync(int fd, struct file *file, int on)
{
J
Jason Wang 已提交
3212
	struct tun_file *tfile = file->private_data;
L
Linus Torvalds 已提交
3213 3214
	int ret;

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

L
Linus Torvalds 已提交
3218
	if (on) {
3219
		__f_setown(file, task_pid(current), PIDTYPE_TGID, 0);
J
Jason Wang 已提交
3220
		tfile->flags |= TUN_FASYNC;
3221
	} else
J
Jason Wang 已提交
3222
		tfile->flags &= ~TUN_FASYNC;
J
Jonathan Corbet 已提交
3223 3224 3225
	ret = 0;
out:
	return ret;
L
Linus Torvalds 已提交
3226 3227 3228 3229
}

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

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

3235
	tfile = (struct tun_file *)sk_alloc(net, AF_UNSPEC, GFP_KERNEL,
3236
					    &tun_proto, 0);
E
Eric W. Biederman 已提交
3237 3238
	if (!tfile)
		return -ENOMEM;
3239 3240 3241 3242 3243
	if (ptr_ring_init(&tfile->tx_ring, 0, GFP_KERNEL)) {
		sk_free(&tfile->sk);
		return -ENOMEM;
	}

3244
	RCU_INIT_POINTER(tfile->tun, NULL);
J
Jason Wang 已提交
3245
	tfile->flags = 0;
3246
	tfile->ifindex = 0;
J
Jason Wang 已提交
3247 3248

	init_waitqueue_head(&tfile->wq.wait);
3249
	RCU_INIT_POINTER(tfile->socket.wq, &tfile->wq);
J
Jason Wang 已提交
3250 3251 3252 3253 3254 3255 3256 3257 3258

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

3262 3263
	sock_set_flag(&tfile->sk, SOCK_ZEROCOPY);

L
Linus Torvalds 已提交
3264 3265 3266 3267 3268
	return 0;
}

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

J
Jason Wang 已提交
3271
	tun_detach(tfile, true);
L
Linus Torvalds 已提交
3272 3273 3274 3275

	return 0;
}

3276
#ifdef CONFIG_PROC_FS
3277
static void tun_chr_show_fdinfo(struct seq_file *m, struct file *file)
3278
{
3279
	struct tun_file *tfile = file->private_data;
3280 3281 3282 3283 3284 3285
	struct tun_struct *tun;
	struct ifreq ifr;

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

	rtnl_lock();
3286
	tun = tun_get(tfile);
3287 3288 3289 3290 3291 3292 3293
	if (tun)
		tun_get_iff(current->nsproxy->net_ns, tun, &ifr);
	rtnl_unlock();

	if (tun)
		tun_put(tun);

3294
	seq_printf(m, "iff:\t%s\n", ifr.ifr_name);
3295 3296 3297
}
#endif

3298
static const struct file_operations tun_fops = {
3299
	.owner	= THIS_MODULE,
L
Linus Torvalds 已提交
3300
	.llseek = no_llseek,
3301
	.read_iter  = tun_chr_read_iter,
3302
	.write_iter = tun_chr_write_iter,
L
Linus Torvalds 已提交
3303
	.poll	= tun_chr_poll,
3304 3305 3306 3307
	.unlocked_ioctl	= tun_chr_ioctl,
#ifdef CONFIG_COMPAT
	.compat_ioctl = tun_chr_compat_ioctl,
#endif
L
Linus Torvalds 已提交
3308 3309
	.open	= tun_chr_open,
	.release = tun_chr_close,
3310 3311 3312 3313
	.fasync = tun_chr_fasync,
#ifdef CONFIG_PROC_FS
	.show_fdinfo = tun_chr_show_fdinfo,
#endif
L
Linus Torvalds 已提交
3314 3315 3316 3317 3318
};

static struct miscdevice tun_miscdev = {
	.minor = TUN_MINOR,
	.name = "tun",
3319
	.nodename = "net/tun",
L
Linus Torvalds 已提交
3320 3321 3322 3323 3324
	.fops = &tun_fops,
};

/* ethtool interface */

3325 3326
static void tun_default_link_ksettings(struct net_device *dev,
				       struct ethtool_link_ksettings *cmd)
3327 3328 3329 3330 3331 3332 3333 3334
{
	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;
3335 3336 3337 3338 3339 3340 3341 3342 3343 3344 3345 3346 3347 3348 3349 3350 3351
}

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 已提交
3352 3353 3354 3355 3356 3357 3358
	return 0;
}

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

3359 3360
	strlcpy(info->driver, DRV_NAME, sizeof(info->driver));
	strlcpy(info->version, DRV_VERSION, sizeof(info->version));
L
Linus Torvalds 已提交
3361 3362

	switch (tun->flags & TUN_TYPE_MASK) {
M
Michael S. Tsirkin 已提交
3363
	case IFF_TUN:
3364
		strlcpy(info->bus_info, "tun", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3365
		break;
M
Michael S. Tsirkin 已提交
3366
	case IFF_TAP:
3367
		strlcpy(info->bus_info, "tap", sizeof(info->bus_info));
L
Linus Torvalds 已提交
3368 3369 3370 3371 3372 3373 3374 3375 3376 3377 3378 3379 3380 3381 3382 3383 3384 3385 3386 3387 3388 3389
		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 已提交
3390 3391 3392 3393 3394 3395 3396 3397 3398 3399 3400 3401 3402 3403 3404 3405 3406 3407 3408 3409 3410 3411 3412
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;
}

3413
static const struct ethtool_ops tun_ethtool_ops = {
L
Linus Torvalds 已提交
3414 3415 3416
	.get_drvinfo	= tun_get_drvinfo,
	.get_msglevel	= tun_get_msglevel,
	.set_msglevel	= tun_set_msglevel,
3417
	.get_link	= ethtool_op_get_link,
3418
	.get_ts_info	= ethtool_op_get_ts_info,
J
Jason Wang 已提交
3419 3420
	.get_coalesce   = tun_get_coalesce,
	.set_coalesce   = tun_set_coalesce,
3421
	.get_link_ksettings = tun_get_link_ksettings,
3422
	.set_link_ksettings = tun_set_link_ksettings,
L
Linus Torvalds 已提交
3423 3424
};

J
Jason Wang 已提交
3425 3426 3427 3428
static int tun_queue_resize(struct tun_struct *tun)
{
	struct net_device *dev = tun->dev;
	struct tun_file *tfile;
3429
	struct ptr_ring **rings;
J
Jason Wang 已提交
3430 3431 3432
	int n = tun->numqueues + tun->numdisabled;
	int ret, i;

3433 3434
	rings = kmalloc_array(n, sizeof(*rings), GFP_KERNEL);
	if (!rings)
J
Jason Wang 已提交
3435 3436 3437 3438
		return -ENOMEM;

	for (i = 0; i < tun->numqueues; i++) {
		tfile = rtnl_dereference(tun->tfiles[i]);
3439
		rings[i] = &tfile->tx_ring;
J
Jason Wang 已提交
3440 3441
	}
	list_for_each_entry(tfile, &tun->disabled, next)
3442
		rings[i++] = &tfile->tx_ring;
J
Jason Wang 已提交
3443

3444 3445
	ret = ptr_ring_resize_multiple(rings, n,
				       dev->tx_queue_len, GFP_KERNEL,
J
Jason Wang 已提交
3446
				       tun_ptr_free);
J
Jason Wang 已提交
3447

3448
	kfree(rings);
J
Jason Wang 已提交
3449 3450 3451 3452 3453 3454 3455 3456 3457
	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);

3458 3459 3460
	if (dev->rtnl_link_ops != &tun_link_ops)
		return NOTIFY_DONE;

J
Jason Wang 已提交
3461 3462 3463 3464 3465 3466 3467 3468 3469 3470 3471 3472 3473 3474 3475
	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,
};
3476

L
Linus Torvalds 已提交
3477 3478 3479 3480
static int __init tun_init(void)
{
	int ret = 0;

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

3483
	ret = rtnl_link_register(&tun_link_ops);
3484
	if (ret) {
3485
		pr_err("Can't register link_ops\n");
3486
		goto err_linkops;
3487 3488
	}

L
Linus Torvalds 已提交
3489
	ret = misc_register(&tun_miscdev);
3490
	if (ret) {
3491
		pr_err("Can't register misc device %d\n", TUN_MINOR);
3492 3493
		goto err_misc;
	}
J
Jason Wang 已提交
3494

3495 3496 3497 3498 3499 3500
	ret = register_netdevice_notifier(&tun_notifier_block);
	if (ret) {
		pr_err("Can't register netdevice notifier\n");
		goto err_notifier;
	}

3501
	return  0;
3502 3503 3504

err_notifier:
	misc_deregister(&tun_miscdev);
3505
err_misc:
3506 3507
	rtnl_link_unregister(&tun_link_ops);
err_linkops:
L
Linus Torvalds 已提交
3508 3509 3510 3511 3512
	return ret;
}

static void tun_cleanup(void)
{
3513
	misc_deregister(&tun_miscdev);
3514
	rtnl_link_unregister(&tun_link_ops);
J
Jason Wang 已提交
3515
	unregister_netdevice_notifier(&tun_notifier_block);
L
Linus Torvalds 已提交
3516 3517
}

3518 3519 3520 3521 3522 3523
/* 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)
{
3524
	struct tun_file *tfile;
3525 3526
	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
3527 3528
	tfile = file->private_data;
	if (!tfile)
3529
		return ERR_PTR(-EBADFD);
J
Jason Wang 已提交
3530
	return &tfile->socket;
3531 3532 3533
}
EXPORT_SYMBOL_GPL(tun_get_socket);

3534
struct ptr_ring *tun_get_tx_ring(struct file *file)
J
Jason Wang 已提交
3535 3536 3537 3538 3539 3540 3541 3542
{
	struct tun_file *tfile;

	if (file->f_op != &tun_fops)
		return ERR_PTR(-EINVAL);
	tfile = file->private_data;
	if (!tfile)
		return ERR_PTR(-EBADFD);
3543
	return &tfile->tx_ring;
J
Jason Wang 已提交
3544
}
3545
EXPORT_SYMBOL_GPL(tun_get_tx_ring);
J
Jason Wang 已提交
3546

L
Linus Torvalds 已提交
3547 3548 3549 3550 3551 3552
module_init(tun_init);
module_exit(tun_cleanup);
MODULE_DESCRIPTION(DRV_DESCRIPTION);
MODULE_AUTHOR(DRV_COPYRIGHT);
MODULE_LICENSE("GPL");
MODULE_ALIAS_MISCDEV(TUN_MINOR);
3553
MODULE_ALIAS("devname:net/tun");