virtio_net.c 73.7 KB
Newer Older
1
/* A network driver using virtio.
R
Rusty Russell 已提交
2 3 4 5 6 7 8 9 10 11 12 13 14 15
 *
 * Copyright 2007 Rusty Russell <rusty@rustcorp.com.au> IBM Corporation
 *
 * 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.
 *
 * You should have received a copy of the GNU General Public License
16
 * along with this program; if not, see <http://www.gnu.org/licenses/>.
R
Rusty Russell 已提交
17 18 19 20
 */
//#define DEBUG
#include <linux/netdevice.h>
#include <linux/etherdevice.h>
21
#include <linux/ethtool.h>
R
Rusty Russell 已提交
22 23 24
#include <linux/module.h>
#include <linux/virtio.h>
#include <linux/virtio_net.h>
J
John Fastabend 已提交
25
#include <linux/bpf.h>
26
#include <linux/bpf_trace.h>
R
Rusty Russell 已提交
27
#include <linux/scatterlist.h>
28
#include <linux/if_vlan.h>
29
#include <linux/slab.h>
30
#include <linux/cpu.h>
31
#include <linux/average.h>
J
Jason Wang 已提交
32
#include <linux/filter.h>
33
#include <net/route.h>
34
#include <net/xdp.h>
R
Rusty Russell 已提交
35

36
static int napi_weight = NAPI_POLL_WEIGHT;
37 38
module_param(napi_weight, int, 0444);

W
Willem de Bruijn 已提交
39
static bool csum = true, gso = true, napi_tx;
R
Rusty Russell 已提交
40 41
module_param(csum, bool, 0444);
module_param(gso, bool, 0444);
W
Willem de Bruijn 已提交
42
module_param(napi_tx, bool, 0644);
R
Rusty Russell 已提交
43

R
Rusty Russell 已提交
44
/* FIXME: MTU in config. */
45
#define GOOD_PACKET_LEN (ETH_HLEN + VLAN_HLEN + ETH_DATA_LEN)
46
#define GOOD_COPY_LEN	128
R
Rusty Russell 已提交
47

48 49
#define VIRTNET_RX_PAD (NET_IP_ALIGN + NET_SKB_PAD)

50 51 52
/* Amount of XDP headroom to prepend to packets for use by xdp_adjust_head */
#define VIRTIO_XDP_HEADROOM 256

J
Johannes Berg 已提交
53 54 55 56
/* RX packet size EWMA. The average packet size is used to determine the packet
 * buffer size when refilling RX rings. As the entire RX ring may be refilled
 * at once, the weight is chosen so that the EWMA will be insensitive to short-
 * term, transient changes in packet size.
57
 */
58
DECLARE_EWMA(pkt_len, 0, 64)
59

60
#define VIRTNET_DRIVER_VERSION "1.0.0"
61

62 63 64 65 66 67
static const unsigned long guest_offloads[] = {
	VIRTIO_NET_F_GUEST_TSO4,
	VIRTIO_NET_F_GUEST_TSO6,
	VIRTIO_NET_F_GUEST_ECN,
	VIRTIO_NET_F_GUEST_UFO
};
68

69
struct virtnet_stats {
70 71
	struct u64_stats_sync tx_syncp;
	struct u64_stats_sync rx_syncp;
72 73 74 75 76 77 78
	u64 tx_bytes;
	u64 tx_packets;

	u64 rx_bytes;
	u64 rx_packets;
};

79 80 81 82 83 84 85
/* Internal representation of a send virtqueue */
struct send_queue {
	/* Virtqueue associated with this send _queue */
	struct virtqueue *vq;

	/* TX: fragments + linear part + virtio header */
	struct scatterlist sg[MAX_SKB_FRAGS + 2];
J
Jason Wang 已提交
86 87 88

	/* Name of the send queue: output.$index */
	char name[40];
W
Willem de Bruijn 已提交
89 90

	struct napi_struct napi;
91 92 93 94 95 96 97
};

/* Internal representation of a receive virtqueue */
struct receive_queue {
	/* Virtqueue associated with this receive_queue */
	struct virtqueue *vq;

R
Rusty Russell 已提交
98 99
	struct napi_struct napi;

J
John Fastabend 已提交
100 101
	struct bpf_prog __rcu *xdp_prog;

102 103 104
	/* Chain pages by the private ptr. */
	struct page *pages;

105
	/* Average packet length for mergeable receive buffers. */
J
Johannes Berg 已提交
106
	struct ewma_pkt_len mrg_avg_pkt_len;
107

108 109 110
	/* Page frag for packet buffer allocation. */
	struct page_frag alloc_frag;

111 112
	/* RX: fragments + linear part + virtio header */
	struct scatterlist sg[MAX_SKB_FRAGS + 2];
J
Jason Wang 已提交
113

114 115 116
	/* Min single buffer size for mergeable buffers case. */
	unsigned int min_buf_len;

J
Jason Wang 已提交
117 118
	/* Name of this receive queue: input.$index */
	char name[40];
119 120

	struct xdp_rxq_info xdp_rxq;
121 122 123 124 125 126
};

struct virtnet_info {
	struct virtio_device *vdev;
	struct virtqueue *cvq;
	struct net_device *dev;
J
Jason Wang 已提交
127 128
	struct send_queue *sq;
	struct receive_queue *rq;
129 130
	unsigned int status;

J
Jason Wang 已提交
131 132 133 134 135 136
	/* Max # of queue pairs supported by the device */
	u16 max_queue_pairs;

	/* # of queue pairs currently used by the driver */
	u16 curr_queue_pairs;

137 138 139
	/* # of XDP queue pairs currently used by the driver */
	u16 xdp_queue_pairs;

140 141 142
	/* I like... big packets and I cannot lie! */
	bool big_packets;

143 144 145
	/* Host will merge rx buffers for big packets (shake it! shake it!) */
	bool mergeable_rx_bufs;

J
Jason Wang 已提交
146 147 148
	/* Has control virtqueue */
	bool has_cvq;

149 150 151
	/* Host can handle any s/g split between our header and packet data */
	bool any_header_sg;

152 153 154
	/* Packet virtio header size */
	u8 hdr_len;

155 156 157
	/* Active statistics */
	struct virtnet_stats __percpu *stats;

158 159 160
	/* Work struct for refilling if we run low on memory. */
	struct delayed_work refill;

161 162 163
	/* Work struct for config space updates */
	struct work_struct config_work;

J
Jason Wang 已提交
164 165
	/* Does the affinity hint is set for virtqueues? */
	bool affinity_hint_set;
166

167 168 169
	/* CPU hotplug instances for online & dead */
	struct hlist_node node;
	struct hlist_node node_dead;
170 171 172 173

	/* Control VQ buffers: protected by the rtnl lock */
	struct virtio_net_ctrl_hdr ctrl_hdr;
	virtio_net_ctrl_ack ctrl_status;
174
	struct virtio_net_ctrl_mq ctrl_mq;
175 176
	u8 ctrl_promisc;
	u8 ctrl_allmulti;
177
	u16 ctrl_vid;
178
	u64 ctrl_offloads;
179 180 181 182

	/* Ethtool settings */
	u8 duplex;
	u32 speed;
183 184

	unsigned long guest_offloads;
R
Rusty Russell 已提交
185 186
};

187
struct padded_vnet_hdr {
188
	struct virtio_net_hdr_mrg_rxbuf hdr;
189
	/*
190 191 192
	 * hdr is in a separate sg buffer, and data sg buffer shares same page
	 * with this header sg. This padding makes next sg 16 byte aligned
	 * after the header.
193
	 */
194
	char padding[4];
195 196
};

J
Jason Wang 已提交
197 198 199 200 201
/* Converting between virtqueue no. and kernel tx/rx queue no.
 * 0:rx0 1:tx0 2:rx1 3:tx1 ... 2N:rxN 2N+1:txN 2N+2:cvq
 */
static int vq2txq(struct virtqueue *vq)
{
202
	return (vq->index - 1) / 2;
J
Jason Wang 已提交
203 204 205 206 207 208 209 210 211
}

static int txq2vq(int txq)
{
	return txq * 2 + 1;
}

static int vq2rxq(struct virtqueue *vq)
{
212
	return vq->index / 2;
J
Jason Wang 已提交
213 214 215 216 217 218 219
}

static int rxq2vq(int rxq)
{
	return rxq * 2;
}

220
static inline struct virtio_net_hdr_mrg_rxbuf *skb_vnet_hdr(struct sk_buff *skb)
R
Rusty Russell 已提交
221
{
222
	return (struct virtio_net_hdr_mrg_rxbuf *)skb->cb;
R
Rusty Russell 已提交
223 224
}

225 226 227 228
/*
 * private is used to chain pages for big packets, put the whole
 * most recent used list in the beginning for reuse
 */
229
static void give_pages(struct receive_queue *rq, struct page *page)
230
{
231
	struct page *end;
232

233
	/* Find end of list, sew whole thing into vi->rq.pages. */
234
	for (end = page; end->private; end = (struct page *)end->private);
235 236
	end->private = (unsigned long)rq->pages;
	rq->pages = page;
237 238
}

239
static struct page *get_a_page(struct receive_queue *rq, gfp_t gfp_mask)
240
{
241
	struct page *p = rq->pages;
242

243
	if (p) {
244
		rq->pages = (struct page *)p->private;
245 246 247
		/* clear private here, it is used to chain pages */
		p->private = 0;
	} else
248 249 250 251
		p = alloc_page(gfp_mask);
	return p;
}

252 253 254 255 256 257 258 259 260 261 262 263 264 265 266
static void virtqueue_napi_schedule(struct napi_struct *napi,
				    struct virtqueue *vq)
{
	if (napi_schedule_prep(napi)) {
		virtqueue_disable_cb(vq);
		__napi_schedule(napi);
	}
}

static void virtqueue_napi_complete(struct napi_struct *napi,
				    struct virtqueue *vq, int processed)
{
	int opaque;

	opaque = virtqueue_enable_cb_prepare(vq);
267 268 269 270 271 272
	if (napi_complete_done(napi, processed)) {
		if (unlikely(virtqueue_poll(vq, opaque)))
			virtqueue_napi_schedule(napi, vq);
	} else {
		virtqueue_disable_cb(vq);
	}
273 274
}

275
static void skb_xmit_done(struct virtqueue *vq)
R
Rusty Russell 已提交
276
{
277
	struct virtnet_info *vi = vq->vdev->priv;
W
Willem de Bruijn 已提交
278
	struct napi_struct *napi = &vi->sq[vq2txq(vq)].napi;
R
Rusty Russell 已提交
279

280
	/* Suppress further interrupts. */
281
	virtqueue_disable_cb(vq);
282

W
Willem de Bruijn 已提交
283 284 285 286 287
	if (napi->weight)
		virtqueue_napi_schedule(napi, vq);
	else
		/* We were probably waiting for more output buffers. */
		netif_wake_subqueue(vi->dev, vq2txq(vq));
R
Rusty Russell 已提交
288 289
}

290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306
#define MRG_CTX_HEADER_SHIFT 22
static void *mergeable_len_to_ctx(unsigned int truesize,
				  unsigned int headroom)
{
	return (void *)(unsigned long)((headroom << MRG_CTX_HEADER_SHIFT) | truesize);
}

static unsigned int mergeable_ctx_to_headroom(void *mrg_ctx)
{
	return (unsigned long)mrg_ctx >> MRG_CTX_HEADER_SHIFT;
}

static unsigned int mergeable_ctx_to_truesize(void *mrg_ctx)
{
	return (unsigned long)mrg_ctx & ((1 << MRG_CTX_HEADER_SHIFT) - 1);
}

307
/* Called from bottom half context */
M
Michael S. Tsirkin 已提交
308 309
static struct sk_buff *page_to_skb(struct virtnet_info *vi,
				   struct receive_queue *rq,
310 311
				   struct page *page, unsigned int offset,
				   unsigned int len, unsigned int truesize)
312 313
{
	struct sk_buff *skb;
314
	struct virtio_net_hdr_mrg_rxbuf *hdr;
315
	unsigned int copy, hdr_len, hdr_padded_len;
316
	char *p;
317

318
	p = page_address(page) + offset;
319

320
	/* copy small packet so we can reuse these pages for small data */
321
	skb = napi_alloc_skb(&rq->napi, GOOD_COPY_LEN);
322 323
	if (unlikely(!skb))
		return NULL;
324

325
	hdr = skb_vnet_hdr(skb);
326

327 328
	hdr_len = vi->hdr_len;
	if (vi->mergeable_rx_bufs)
329
		hdr_padded_len = sizeof(*hdr);
330
	else
331
		hdr_padded_len = sizeof(struct padded_vnet_hdr);
332

333
	memcpy(hdr, p, hdr_len);
334

335
	len -= hdr_len;
336 337
	offset += hdr_padded_len;
	p += hdr_padded_len;
338

339 340 341
	copy = len;
	if (copy > skb_tailroom(skb))
		copy = skb_tailroom(skb);
342
	skb_put_data(skb, p, copy);
343

344 345
	len -= copy;
	offset += copy;
346

347 348 349 350 351 352 353 354
	if (vi->mergeable_rx_bufs) {
		if (len)
			skb_add_rx_frag(skb, 0, page, offset, len, truesize);
		else
			put_page(page);
		return skb;
	}

355 356 357 358 359 360 361
	/*
	 * Verify that we can indeed put this data into a skb.
	 * This is here to handle cases when the device erroneously
	 * tries to receive more than is possible. This is usually
	 * the case of a broken device.
	 */
	if (unlikely(len > MAX_SKB_FRAGS * PAGE_SIZE)) {
362
		net_dbg_ratelimited("%s: too much data\n", skb->dev->name);
363 364 365
		dev_kfree_skb(skb);
		return NULL;
	}
366
	BUG_ON(offset >= PAGE_SIZE);
367
	while (len) {
368 369 370 371
		unsigned int frag_size = min((unsigned)PAGE_SIZE - offset, len);
		skb_add_rx_frag(skb, skb_shinfo(skb)->nr_frags, page, offset,
				frag_size, truesize);
		len -= frag_size;
372 373 374
		page = (struct page *)page->private;
		offset = 0;
	}
375

376
	if (page)
377
		give_pages(rq, page);
378

379 380
	return skb;
}
381

J
Jason Wang 已提交
382 383 384 385 386 387 388 389 390 391 392 393 394 395
static void virtnet_xdp_flush(struct net_device *dev)
{
	struct virtnet_info *vi = netdev_priv(dev);
	struct send_queue *sq;
	unsigned int qp;

	qp = vi->curr_queue_pairs - vi->xdp_queue_pairs + smp_processor_id();
	sq = &vi->sq[qp];

	virtqueue_kick(sq->vq);
}

static bool __virtnet_xdp_xmit(struct virtnet_info *vi,
			       struct xdp_buff *xdp)
J
John Fastabend 已提交
396 397
{
	struct virtio_net_hdr_mrg_rxbuf *hdr;
398
	unsigned int len;
399 400
	struct send_queue *sq;
	unsigned int qp;
J
John Fastabend 已提交
401 402 403
	void *xdp_sent;
	int err;

404 405 406
	qp = vi->curr_queue_pairs - vi->xdp_queue_pairs + smp_processor_id();
	sq = &vi->sq[qp];

J
John Fastabend 已提交
407 408
	/* Free up any pending old buffers before queueing new ones. */
	while ((xdp_sent = virtqueue_get_buf(sq->vq, &len)) != NULL) {
409
		struct page *sent_page = virt_to_head_page(xdp_sent);
410

411
		put_page(sent_page);
412
	}
J
John Fastabend 已提交
413

414 415 416 417
	xdp->data -= vi->hdr_len;
	/* Zero header and leave csum up to XDP layers */
	hdr = xdp->data;
	memset(hdr, 0, vi->hdr_len);
418

419
	sg_init_one(sq->sg, xdp->data, xdp->data_end - xdp->data);
420

421
	err = virtqueue_add_outbuf(sq->vq, sq->sg, 1, xdp->data, GFP_ATOMIC);
J
John Fastabend 已提交
422
	if (unlikely(err)) {
423
		struct page *page = virt_to_head_page(xdp->data);
424

425
		put_page(page);
426
		return false;
J
John Fastabend 已提交
427 428
	}

429
	return true;
J
John Fastabend 已提交
430 431
}

J
Jason Wang 已提交
432 433 434 435 436 437 438 439 440 441
static int virtnet_xdp_xmit(struct net_device *dev, struct xdp_buff *xdp)
{
	struct virtnet_info *vi = netdev_priv(dev);
	bool sent = __virtnet_xdp_xmit(vi, xdp);

	if (!sent)
		return -ENOSPC;
	return 0;
}

442 443 444 445 446
static unsigned int virtnet_get_headroom(struct virtnet_info *vi)
{
	return vi->xdp_queue_pairs ? VIRTIO_XDP_HEADROOM : 0;
}

447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509
/* We copy the packet for XDP in the following cases:
 *
 * 1) Packet is scattered across multiple rx buffers.
 * 2) Headroom space is insufficient.
 *
 * This is inefficient but it's a temporary condition that
 * we hit right after XDP is enabled and until queue is refilled
 * with large buffers with sufficient headroom - so it should affect
 * at most queue size packets.
 * Afterwards, the conditions to enable
 * XDP should preclude the underlying device from sending packets
 * across multiple buffers (num_buf > 1), and we make sure buffers
 * have enough headroom.
 */
static struct page *xdp_linearize_page(struct receive_queue *rq,
				       u16 *num_buf,
				       struct page *p,
				       int offset,
				       int page_off,
				       unsigned int *len)
{
	struct page *page = alloc_page(GFP_ATOMIC);

	if (!page)
		return NULL;

	memcpy(page_address(page) + page_off, page_address(p) + offset, *len);
	page_off += *len;

	while (--*num_buf) {
		unsigned int buflen;
		void *buf;
		int off;

		buf = virtqueue_get_buf(rq->vq, &buflen);
		if (unlikely(!buf))
			goto err_buf;

		p = virt_to_head_page(buf);
		off = buf - page_address(p);

		/* guard against a misconfigured or uncooperative backend that
		 * is sending packet larger than the MTU.
		 */
		if ((page_off + buflen) > PAGE_SIZE) {
			put_page(p);
			goto err_buf;
		}

		memcpy(page_address(page) + page_off,
		       page_address(p) + off, buflen);
		page_off += buflen;
		put_page(p);
	}

	/* Headroom does not contribute to packet length */
	*len = page_off - VIRTIO_XDP_HEADROOM;
	return page;
err_buf:
	__free_pages(page, 0);
	return NULL;
}

510 511 512
static struct sk_buff *receive_small(struct net_device *dev,
				     struct virtnet_info *vi,
				     struct receive_queue *rq,
513
				     void *buf, void *ctx,
J
Jason Wang 已提交
514 515
				     unsigned int len,
				     bool *xdp_xmit)
516
{
517
	struct sk_buff *skb;
518
	struct bpf_prog *xdp_prog;
519
	unsigned int xdp_headroom = (unsigned long)ctx;
520 521 522 523
	unsigned int header_offset = VIRTNET_RX_PAD + xdp_headroom;
	unsigned int headroom = vi->hdr_len + header_offset;
	unsigned int buflen = SKB_DATA_ALIGN(GOOD_PACKET_LEN + headroom) +
			      SKB_DATA_ALIGN(sizeof(struct skb_shared_info));
524
	struct page *page = virt_to_head_page(buf);
J
Jason Wang 已提交
525
	unsigned int delta = 0, err;
526
	struct page *xdp_page;
527
	len -= vi->hdr_len;
528

529 530 531
	rcu_read_lock();
	xdp_prog = rcu_dereference(rq->xdp_prog);
	if (xdp_prog) {
532
		struct virtio_net_hdr_mrg_rxbuf *hdr = buf + header_offset;
533
		struct xdp_buff xdp;
534
		void *orig_data;
535 536 537 538
		u32 act;

		if (unlikely(hdr->hdr.gso_type || hdr->hdr.flags))
			goto err_xdp;
539

540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560
		if (unlikely(xdp_headroom < virtnet_get_headroom(vi))) {
			int offset = buf - page_address(page) + header_offset;
			unsigned int tlen = len + vi->hdr_len;
			u16 num_buf = 1;

			xdp_headroom = virtnet_get_headroom(vi);
			header_offset = VIRTNET_RX_PAD + xdp_headroom;
			headroom = vi->hdr_len + header_offset;
			buflen = SKB_DATA_ALIGN(GOOD_PACKET_LEN + headroom) +
				 SKB_DATA_ALIGN(sizeof(struct skb_shared_info));
			xdp_page = xdp_linearize_page(rq, &num_buf, page,
						      offset, header_offset,
						      &tlen);
			if (!xdp_page)
				goto err_xdp;

			buf = page_address(xdp_page);
			put_page(page);
			page = xdp_page;
		}

561 562
		xdp.data_hard_start = buf + VIRTNET_RX_PAD + vi->hdr_len;
		xdp.data = xdp.data_hard_start + xdp_headroom;
563
		xdp_set_data_meta_invalid(&xdp);
564
		xdp.data_end = xdp.data + len;
565
		xdp.rxq = &rq->xdp_rxq;
566
		orig_data = xdp.data;
567 568
		act = bpf_prog_run_xdp(xdp_prog, &xdp);

569 570
		switch (act) {
		case XDP_PASS:
571
			/* Recalculate length in case bpf program changed it */
572
			delta = orig_data - xdp.data;
573 574
			break;
		case XDP_TX:
J
Jason Wang 已提交
575
			if (unlikely(!__virtnet_xdp_xmit(vi, &xdp)))
576
				trace_xdp_exception(vi->dev, xdp_prog, act);
J
Jason Wang 已提交
577 578 579 580 581 582 583 584
			else
				*xdp_xmit = true;
			rcu_read_unlock();
			goto xdp_xmit;
		case XDP_REDIRECT:
			err = xdp_do_redirect(dev, &xdp, xdp_prog);
			if (!err)
				*xdp_xmit = true;
585 586 587
			rcu_read_unlock();
			goto xdp_xmit;
		default:
588 589 590 591
			bpf_warn_invalid_xdp_action(act);
		case XDP_ABORTED:
			trace_xdp_exception(vi->dev, xdp_prog, act);
		case XDP_DROP:
592 593 594 595 596
			goto err_xdp;
		}
	}
	rcu_read_unlock();

597 598
	skb = build_skb(buf, buflen);
	if (!skb) {
599
		put_page(page);
600 601 602 603 604 605 606 607 608 609
		goto err;
	}
	skb_reserve(skb, headroom - delta);
	skb_put(skb, len + delta);
	if (!delta) {
		buf += header_offset;
		memcpy(skb_vnet_hdr(skb), buf, vi->hdr_len);
	} /* keep zeroed vnet hdr since packet was changed by bpf */

err:
610
	return skb;
611 612 613 614

err_xdp:
	rcu_read_unlock();
	dev->stats.rx_dropped++;
615
	put_page(page);
616 617
xdp_xmit:
	return NULL;
618 619 620
}

static struct sk_buff *receive_big(struct net_device *dev,
M
Michael S. Tsirkin 已提交
621
				   struct virtnet_info *vi,
622 623 624 625 626
				   struct receive_queue *rq,
				   void *buf,
				   unsigned int len)
{
	struct page *page = buf;
627
	struct sk_buff *skb = page_to_skb(vi, rq, page, 0, len, PAGE_SIZE);
J
John Fastabend 已提交
628

629 630 631 632 633 634 635 636 637 638 639
	if (unlikely(!skb))
		goto err;

	return skb;

err:
	dev->stats.rx_dropped++;
	give_pages(rq, page);
	return NULL;
}

640
static struct sk_buff *receive_mergeable(struct net_device *dev,
M
Michael S. Tsirkin 已提交
641
					 struct virtnet_info *vi,
642
					 struct receive_queue *rq,
643 644
					 void *buf,
					 void *ctx,
J
Jason Wang 已提交
645 646
					 unsigned int len,
					 bool *xdp_xmit)
647
{
648 649
	struct virtio_net_hdr_mrg_rxbuf *hdr = buf;
	u16 num_buf = virtio16_to_cpu(vi->vdev, hdr->num_buffers);
650 651
	struct page *page = virt_to_head_page(buf);
	int offset = buf - page_address(page);
J
John Fastabend 已提交
652 653 654
	struct sk_buff *head_skb, *curr_skb;
	struct bpf_prog *xdp_prog;
	unsigned int truesize;
655
	unsigned int headroom = mergeable_ctx_to_headroom(ctx);
J
Jason Wang 已提交
656
	int err;
J
John Fastabend 已提交
657

J
John Fastabend 已提交
658 659
	head_skb = NULL;

J
John Fastabend 已提交
660 661 662
	rcu_read_lock();
	xdp_prog = rcu_dereference(rq->xdp_prog);
	if (xdp_prog) {
663
		struct page *xdp_page;
664 665
		struct xdp_buff xdp;
		void *data;
J
John Fastabend 已提交
666 667
		u32 act;

668
		/* This happens when rx buffer size is underestimated */
669 670
		if (unlikely(num_buf > 1 ||
			     headroom < virtnet_get_headroom(vi))) {
671
			/* linearize data for XDP */
672
			xdp_page = xdp_linearize_page(rq, &num_buf,
673 674 675
						      page, offset,
						      VIRTIO_XDP_HEADROOM,
						      &len);
676 677
			if (!xdp_page)
				goto err_xdp;
678
			offset = VIRTIO_XDP_HEADROOM;
679 680
		} else {
			xdp_page = page;
J
John Fastabend 已提交
681 682 683 684 685 686 687
		}

		/* Transient failure which in theory could occur if
		 * in-flight packets from before XDP was enabled reach
		 * the receive path after XDP is loaded. In practice I
		 * was not able to create this condition.
		 */
688
		if (unlikely(hdr->hdr.gso_type))
J
John Fastabend 已提交
689 690
			goto err_xdp;

691 692 693
		/* Allow consuming headroom but reserve enough space to push
		 * the descriptor on if we get an XDP_TX return code.
		 */
694
		data = page_address(xdp_page) + offset;
695
		xdp.data_hard_start = data - VIRTIO_XDP_HEADROOM + vi->hdr_len;
696
		xdp.data = data + vi->hdr_len;
697
		xdp_set_data_meta_invalid(&xdp);
698
		xdp.data_end = xdp.data + (len - vi->hdr_len);
699 700
		xdp.rxq = &rq->xdp_rxq;

701 702
		act = bpf_prog_run_xdp(xdp_prog, &xdp);

703 704 705
		if (act != XDP_PASS)
			ewma_pkt_len_add(&rq->mrg_avg_pkt_len, len);

J
John Fastabend 已提交
706 707
		switch (act) {
		case XDP_PASS:
708 709 710 711 712 713 714
			/* recalculate offset to account for any header
			 * adjustments. Note other cases do not build an
			 * skb and avoid using offset
			 */
			offset = xdp.data -
					page_address(xdp_page) - vi->hdr_len;

715 716 717 718 719
			/* We can only create skb based on xdp_page. */
			if (unlikely(xdp_page != page)) {
				rcu_read_unlock();
				put_page(page);
				head_skb = page_to_skb(vi, rq, xdp_page,
720
						       offset, len, PAGE_SIZE);
721 722
				return head_skb;
			}
J
John Fastabend 已提交
723 724
			break;
		case XDP_TX:
J
Jason Wang 已提交
725
			if (unlikely(!__virtnet_xdp_xmit(vi, &xdp)))
726
				trace_xdp_exception(vi->dev, xdp_prog, act);
J
Jason Wang 已提交
727 728
			else
				*xdp_xmit = true;
729 730
			if (unlikely(xdp_page != page))
				goto err_xdp;
J
John Fastabend 已提交
731 732
			rcu_read_unlock();
			goto xdp_xmit;
J
Jason Wang 已提交
733 734
		case XDP_REDIRECT:
			err = xdp_do_redirect(dev, &xdp, xdp_prog);
735
			if (!err)
J
Jason Wang 已提交
736 737 738
				*xdp_xmit = true;
			rcu_read_unlock();
			goto xdp_xmit;
J
John Fastabend 已提交
739
		default:
740 741 742 743
			bpf_warn_invalid_xdp_action(act);
		case XDP_ABORTED:
			trace_xdp_exception(vi->dev, xdp_prog, act);
		case XDP_DROP:
744 745
			if (unlikely(xdp_page != page))
				__free_pages(xdp_page, 0);
J
John Fastabend 已提交
746
			goto err_xdp;
J
John Fastabend 已提交
747
		}
J
John Fastabend 已提交
748 749
	}
	rcu_read_unlock();
750

751 752
	truesize = mergeable_ctx_to_truesize(ctx);
	if (unlikely(len > truesize)) {
753
		pr_debug("%s: rx error: len %u exceeds truesize %lu\n",
754 755 756 757
			 dev->name, len, (unsigned long)ctx);
		dev->stats.rx_length_errors++;
		goto err_skb;
	}
758

J
John Fastabend 已提交
759 760
	head_skb = page_to_skb(vi, rq, page, offset, len, truesize);
	curr_skb = head_skb;
761

762 763
	if (unlikely(!curr_skb))
		goto err_skb;
764
	while (--num_buf) {
765 766
		int num_skb_frags;

767
		buf = virtqueue_get_buf_ctx(rq->vq, &len, &ctx);
768
		if (unlikely(!buf)) {
769
			pr_debug("%s: rx error: %d buffers out of %d missing\n",
M
Michael S. Tsirkin 已提交
770
				 dev->name, num_buf,
771 772
				 virtio16_to_cpu(vi->vdev,
						 hdr->num_buffers));
773 774
			dev->stats.rx_length_errors++;
			goto err_buf;
775
		}
776 777

		page = virt_to_head_page(buf);
778 779 780

		truesize = mergeable_ctx_to_truesize(ctx);
		if (unlikely(len > truesize)) {
781
			pr_debug("%s: rx error: len %u exceeds truesize %lu\n",
782 783 784 785
				 dev->name, len, (unsigned long)ctx);
			dev->stats.rx_length_errors++;
			goto err_skb;
		}
786 787

		num_skb_frags = skb_shinfo(curr_skb)->nr_frags;
788 789
		if (unlikely(num_skb_frags == MAX_SKB_FRAGS)) {
			struct sk_buff *nskb = alloc_skb(0, GFP_ATOMIC);
790 791 792

			if (unlikely(!nskb))
				goto err_skb;
793 794 795 796 797 798 799 800 801 802 803
			if (curr_skb == head_skb)
				skb_shinfo(curr_skb)->frag_list = nskb;
			else
				curr_skb->next = nskb;
			curr_skb = nskb;
			head_skb->truesize += nskb->truesize;
			num_skb_frags = 0;
		}
		if (curr_skb != head_skb) {
			head_skb->data_len += len;
			head_skb->len += len;
804
			head_skb->truesize += truesize;
805
		}
806
		offset = buf - page_address(page);
807 808 809
		if (skb_can_coalesce(curr_skb, num_skb_frags, page, offset)) {
			put_page(page);
			skb_coalesce_rx_frag(curr_skb, num_skb_frags - 1,
810
					     len, truesize);
811 812
		} else {
			skb_add_rx_frag(curr_skb, num_skb_frags, page,
813
					offset, len, truesize);
814
		}
815 816
	}

J
Johannes Berg 已提交
817
	ewma_pkt_len_add(&rq->mrg_avg_pkt_len, head_skb->len);
818 819
	return head_skb;

J
John Fastabend 已提交
820 821
err_xdp:
	rcu_read_unlock();
822 823 824
err_skb:
	put_page(page);
	while (--num_buf) {
825 826
		buf = virtqueue_get_buf(rq->vq, &len);
		if (unlikely(!buf)) {
827 828 829 830 831
			pr_debug("%s: rx error: %d buffers missing\n",
				 dev->name, num_buf);
			dev->stats.rx_length_errors++;
			break;
		}
832
		page = virt_to_head_page(buf);
833
		put_page(page);
834
	}
835 836 837
err_buf:
	dev->stats.rx_dropped++;
	dev_kfree_skb(head_skb);
J
John Fastabend 已提交
838
xdp_xmit:
839
	return NULL;
840 841
}

J
Jason Wang 已提交
842
static int receive_buf(struct virtnet_info *vi, struct receive_queue *rq,
J
Jason Wang 已提交
843
		       void *buf, unsigned int len, void **ctx, bool *xdp_xmit)
844
{
845
	struct net_device *dev = vi->dev;
846
	struct sk_buff *skb;
847
	struct virtio_net_hdr_mrg_rxbuf *hdr;
J
Jason Wang 已提交
848
	int ret;
849

850
	if (unlikely(len < vi->hdr_len + ETH_HLEN)) {
851 852
		pr_debug("%s: short packet %i\n", dev->name, len);
		dev->stats.rx_length_errors++;
853
		if (vi->mergeable_rx_bufs) {
854
			put_page(virt_to_head_page(buf));
855
		} else if (vi->big_packets) {
856
			give_pages(rq, buf);
857
		} else {
858
			put_page(virt_to_head_page(buf));
859
		}
J
Jason Wang 已提交
860
		return 0;
861
	}
862

863
	if (vi->mergeable_rx_bufs)
J
Jason Wang 已提交
864
		skb = receive_mergeable(dev, vi, rq, buf, ctx, len, xdp_xmit);
865
	else if (vi->big_packets)
M
Michael S. Tsirkin 已提交
866
		skb = receive_big(dev, vi, rq, buf, len);
867
	else
J
Jason Wang 已提交
868
		skb = receive_small(dev, vi, rq, buf, ctx, len, xdp_xmit);
869 870

	if (unlikely(!skb))
J
Jason Wang 已提交
871
		return 0;
872

873
	hdr = skb_vnet_hdr(skb);
874

J
Jason Wang 已提交
875
	ret = skb->len;
R
Rusty Russell 已提交
876

877
	if (hdr->hdr.flags & VIRTIO_NET_HDR_F_DATA_VALID)
878
		skb->ip_summed = CHECKSUM_UNNECESSARY;
R
Rusty Russell 已提交
879

880 881 882 883 884 885
	if (virtio_net_hdr_to_skb(skb, &hdr->hdr,
				  virtio_is_little_endian(vi->vdev))) {
		net_warn_ratelimited("%s: bad gso: type: %u, size: %u\n",
				     dev->name, hdr->hdr.gso_type,
				     hdr->hdr.gso_size);
		goto frame_err;
R
Rusty Russell 已提交
886 887
	}

888 889 890 891
	skb->protocol = eth_type_trans(skb, dev);
	pr_debug("Receiving skb proto 0x%04x len %i type %i\n",
		 ntohs(skb->protocol), skb->len, skb->pkt_type);

E
Eric Dumazet 已提交
892
	napi_gro_receive(&rq->napi, skb);
J
Jason Wang 已提交
893
	return ret;
R
Rusty Russell 已提交
894 895 896 897

frame_err:
	dev->stats.rx_frame_errors++;
	dev_kfree_skb(skb);
J
Jason Wang 已提交
898
	return 0;
R
Rusty Russell 已提交
899 900
}

901 902 903 904 905
/* Unlike mergeable buffers, all buffers are allocated to the
 * same size, except for the headroom. For this reason we do
 * not need to use  mergeable_len_to_ctx here - it is enough
 * to store the headroom as the context ignoring the truesize.
 */
M
Michael S. Tsirkin 已提交
906 907
static int add_recvbuf_small(struct virtnet_info *vi, struct receive_queue *rq,
			     gfp_t gfp)
R
Rusty Russell 已提交
908
{
909 910
	struct page_frag *alloc_frag = &rq->alloc_frag;
	char *buf;
911
	unsigned int xdp_headroom = virtnet_get_headroom(vi);
912
	void *ctx = (void *)(unsigned long)xdp_headroom;
913
	int len = vi->hdr_len + VIRTNET_RX_PAD + GOOD_PACKET_LEN + xdp_headroom;
914
	int err;
915

916 917 918
	len = SKB_DATA_ALIGN(len) +
	      SKB_DATA_ALIGN(sizeof(struct skb_shared_info));
	if (unlikely(!skb_page_frag_refill(len, alloc_frag, gfp)))
919
		return -ENOMEM;
R
Rusty Russell 已提交
920

921 922 923 924 925
	buf = (char *)page_address(alloc_frag->page) + alloc_frag->offset;
	get_page(alloc_frag->page);
	alloc_frag->offset += len;
	sg_init_one(rq->sg, buf + VIRTNET_RX_PAD + xdp_headroom,
		    vi->hdr_len + GOOD_PACKET_LEN);
926
	err = virtqueue_add_inbuf_ctx(rq->vq, rq->sg, 1, buf, ctx, gfp);
927
	if (err < 0)
928
		put_page(virt_to_head_page(buf));
929 930
	return err;
}
931

932 933
static int add_recvbuf_big(struct virtnet_info *vi, struct receive_queue *rq,
			   gfp_t gfp)
934 935 936 937 938
{
	struct page *first, *list = NULL;
	char *p;
	int i, err, offset;

939 940
	sg_init_table(rq->sg, MAX_SKB_FRAGS + 2);

941
	/* page in rq->sg[MAX_SKB_FRAGS + 1] is list tail */
942
	for (i = MAX_SKB_FRAGS + 1; i > 1; --i) {
943
		first = get_a_page(rq, gfp);
944 945
		if (!first) {
			if (list)
946
				give_pages(rq, list);
947
			return -ENOMEM;
948
		}
949
		sg_set_buf(&rq->sg[i], page_address(first), PAGE_SIZE);
950

951 952 953 954
		/* chain new page in list head to match sg */
		first->private = (unsigned long)list;
		list = first;
	}
R
Rusty Russell 已提交
955

956
	first = get_a_page(rq, gfp);
957
	if (!first) {
958
		give_pages(rq, list);
959 960 961 962
		return -ENOMEM;
	}
	p = page_address(first);

963
	/* rq->sg[0], rq->sg[1] share the same page */
964 965
	/* a separated rq->sg[0] for header - required in case !any_header_sg */
	sg_set_buf(&rq->sg[0], p, vi->hdr_len);
966

967
	/* rq->sg[1] for data packet, from offset */
968
	offset = sizeof(struct padded_vnet_hdr);
969
	sg_set_buf(&rq->sg[1], p + offset, PAGE_SIZE - offset);
970 971 972

	/* chain first in list head */
	first->private = (unsigned long)list;
973 974
	err = virtqueue_add_inbuf(rq->vq, rq->sg, MAX_SKB_FRAGS + 2,
				  first, gfp);
975
	if (err < 0)
976
		give_pages(rq, first);
977 978

	return err;
R
Rusty Russell 已提交
979 980
}

981 982
static unsigned int get_mergeable_buf_len(struct receive_queue *rq,
					  struct ewma_pkt_len *avg_pkt_len)
983
{
984
	const size_t hdr_len = sizeof(struct virtio_net_hdr_mrg_rxbuf);
985 986
	unsigned int len;

J
Johannes Berg 已提交
987
	len = hdr_len + clamp_t(unsigned int, ewma_pkt_len_read(avg_pkt_len),
988
				rq->min_buf_len, PAGE_SIZE - hdr_len);
989
	return ALIGN(len, L1_CACHE_BYTES);
990 991
}

992 993
static int add_recvbuf_mergeable(struct virtnet_info *vi,
				 struct receive_queue *rq, gfp_t gfp)
994
{
995
	struct page_frag *alloc_frag = &rq->alloc_frag;
996
	unsigned int headroom = virtnet_get_headroom(vi);
997
	char *buf;
998
	void *ctx;
999
	int err;
1000
	unsigned int len, hole;
1001

1002
	len = get_mergeable_buf_len(rq, &rq->mrg_avg_pkt_len);
1003
	if (unlikely(!skb_page_frag_refill(len + headroom, alloc_frag, gfp)))
1004
		return -ENOMEM;
1005

1006
	buf = (char *)page_address(alloc_frag->page) + alloc_frag->offset;
1007
	buf += headroom; /* advance address leaving hole at front of pkt */
1008
	get_page(alloc_frag->page);
1009
	alloc_frag->offset += len + headroom;
1010
	hole = alloc_frag->size - alloc_frag->offset;
1011
	if (hole < len + headroom) {
1012 1013
		/* To avoid internal fragmentation, if there is very likely not
		 * enough space for another buffer, add the remaining space to
1014
		 * the current buffer.
1015
		 */
1016 1017 1018
		len += hole;
		alloc_frag->offset += hole;
	}
1019

1020
	sg_init_one(rq->sg, buf, len);
1021
	ctx = mergeable_len_to_ctx(len, headroom);
1022
	err = virtqueue_add_inbuf_ctx(rq->vq, rq->sg, 1, buf, ctx, gfp);
1023
	if (err < 0)
1024
		put_page(virt_to_head_page(buf));
1025

1026 1027
	return err;
}
1028

1029 1030 1031 1032 1033 1034 1035
/*
 * Returns false if we couldn't fill entirely (OOM).
 *
 * Normally run in the receive path, but can also be run from ndo_open
 * before we're receiving packets, or from refill_work which is
 * careful to disable receiving (using napi_disable).
 */
M
Michael S. Tsirkin 已提交
1036 1037
static bool try_fill_recv(struct virtnet_info *vi, struct receive_queue *rq,
			  gfp_t gfp)
1038 1039
{
	int err;
1040
	bool oom;
1041

1042 1043
	do {
		if (vi->mergeable_rx_bufs)
1044
			err = add_recvbuf_mergeable(vi, rq, gfp);
1045
		else if (vi->big_packets)
1046
			err = add_recvbuf_big(vi, rq, gfp);
1047
		else
M
Michael S. Tsirkin 已提交
1048
			err = add_recvbuf_small(vi, rq, gfp);
1049

1050
		oom = err == -ENOMEM;
1051
		if (err)
1052
			break;
1053
	} while (rq->vq->num_free);
1054
	virtqueue_kick(rq->vq);
1055
	return !oom;
1056 1057
}

1058
static void skb_recv_done(struct virtqueue *rvq)
R
Rusty Russell 已提交
1059 1060
{
	struct virtnet_info *vi = rvq->vdev->priv;
J
Jason Wang 已提交
1061
	struct receive_queue *rq = &vi->rq[vq2rxq(rvq)];
1062

1063
	virtqueue_napi_schedule(&rq->napi, rvq);
R
Rusty Russell 已提交
1064 1065
}

1066
static void virtnet_napi_enable(struct virtqueue *vq, struct napi_struct *napi)
1067
{
1068
	napi_enable(napi);
1069 1070

	/* If all buffers were filled by other side before we napi_enabled, we
1071 1072 1073 1074 1075 1076
	 * won't get another interrupt, so process any outstanding packets now.
	 * Call local_bh_enable after to trigger softIRQ processing.
	 */
	local_bh_disable();
	virtqueue_napi_schedule(napi, vq);
	local_bh_enable();
1077 1078
}

W
Willem de Bruijn 已提交
1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096
static void virtnet_napi_tx_enable(struct virtnet_info *vi,
				   struct virtqueue *vq,
				   struct napi_struct *napi)
{
	if (!napi->weight)
		return;

	/* Tx napi touches cachelines on the cpu handling tx interrupts. Only
	 * enable the feature if this is likely affine with the transmit path.
	 */
	if (!vi->affinity_hint_set) {
		napi->weight = 0;
		return;
	}

	return virtnet_napi_enable(vq, napi);
}

1097 1098 1099 1100 1101 1102
static void virtnet_napi_tx_disable(struct napi_struct *napi)
{
	if (napi->weight)
		napi_disable(napi);
}

1103 1104
static void refill_work(struct work_struct *work)
{
1105 1106
	struct virtnet_info *vi =
		container_of(work, struct virtnet_info, refill.work);
1107
	bool still_empty;
J
Jason Wang 已提交
1108 1109
	int i;

1110
	for (i = 0; i < vi->curr_queue_pairs; i++) {
J
Jason Wang 已提交
1111
		struct receive_queue *rq = &vi->rq[i];
1112

J
Jason Wang 已提交
1113
		napi_disable(&rq->napi);
M
Michael S. Tsirkin 已提交
1114
		still_empty = !try_fill_recv(vi, rq, GFP_KERNEL);
1115
		virtnet_napi_enable(rq->vq, &rq->napi);
1116

J
Jason Wang 已提交
1117 1118 1119 1120 1121 1122
		/* In theory, this can happen: if we don't get any buffers in
		 * we will *never* try to fill again.
		 */
		if (still_empty)
			schedule_delayed_work(&vi->refill, HZ/2);
	}
1123 1124
}

J
Jason Wang 已提交
1125
static int virtnet_receive(struct receive_queue *rq, int budget, bool *xdp_xmit)
R
Rusty Russell 已提交
1126
{
1127
	struct virtnet_info *vi = rq->vq->vdev->priv;
J
Jason Wang 已提交
1128
	unsigned int len, received = 0, bytes = 0;
1129
	void *buf;
J
Jason Wang 已提交
1130
	struct virtnet_stats *stats = this_cpu_ptr(vi->stats);
R
Rusty Russell 已提交
1131

1132
	if (!vi->big_packets || vi->mergeable_rx_bufs) {
1133 1134 1135 1136
		void *ctx;

		while (received < budget &&
		       (buf = virtqueue_get_buf_ctx(rq->vq, &len, &ctx))) {
J
Jason Wang 已提交
1137
			bytes += receive_buf(vi, rq, buf, len, ctx, xdp_xmit);
1138 1139 1140 1141 1142
			received++;
		}
	} else {
		while (received < budget &&
		       (buf = virtqueue_get_buf(rq->vq, &len)) != NULL) {
J
Jason Wang 已提交
1143
			bytes += receive_buf(vi, rq, buf, len, NULL, xdp_xmit);
1144 1145
			received++;
		}
R
Rusty Russell 已提交
1146 1147
	}

1148
	if (rq->vq->num_free > virtqueue_get_vring_size(rq->vq) / 2) {
M
Michael S. Tsirkin 已提交
1149
		if (!try_fill_recv(vi, rq, GFP_ATOMIC))
1150
			schedule_delayed_work(&vi->refill, 0);
1151
	}
R
Rusty Russell 已提交
1152

J
Jason Wang 已提交
1153 1154 1155 1156 1157
	u64_stats_update_begin(&stats->rx_syncp);
	stats->rx_bytes += bytes;
	stats->rx_packets += received;
	u64_stats_update_end(&stats->rx_syncp);

1158 1159 1160
	return received;
}

1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175
static void free_old_xmit_skbs(struct send_queue *sq)
{
	struct sk_buff *skb;
	unsigned int len;
	struct virtnet_info *vi = sq->vq->vdev->priv;
	struct virtnet_stats *stats = this_cpu_ptr(vi->stats);
	unsigned int packets = 0;
	unsigned int bytes = 0;

	while ((skb = virtqueue_get_buf(sq->vq, &len)) != NULL) {
		pr_debug("Sent skb %p\n", skb);

		bytes += skb->len;
		packets++;

1176
		dev_consume_skb_any(skb);
1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190
	}

	/* Avoid overhead when no packets have been processed
	 * happens when called speculatively from start_xmit.
	 */
	if (!packets)
		return;

	u64_stats_update_begin(&stats->tx_syncp);
	stats->tx_bytes += bytes;
	stats->tx_packets += packets;
	u64_stats_update_end(&stats->tx_syncp);
}

1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209
static void virtnet_poll_cleantx(struct receive_queue *rq)
{
	struct virtnet_info *vi = rq->vq->vdev->priv;
	unsigned int index = vq2rxq(rq->vq);
	struct send_queue *sq = &vi->sq[index];
	struct netdev_queue *txq = netdev_get_tx_queue(vi->dev, index);

	if (!sq->napi.weight)
		return;

	if (__netif_tx_trylock(txq)) {
		free_old_xmit_skbs(sq);
		__netif_tx_unlock(txq);
	}

	if (sq->vq->num_free >= 2 + MAX_SKB_FRAGS)
		netif_tx_wake_queue(txq);
}

1210 1211 1212 1213
static int virtnet_poll(struct napi_struct *napi, int budget)
{
	struct receive_queue *rq =
		container_of(napi, struct receive_queue, napi);
1214
	unsigned int received;
J
Jason Wang 已提交
1215
	bool xdp_xmit = false;
1216

1217 1218
	virtnet_poll_cleantx(rq);

J
Jason Wang 已提交
1219
	received = virtnet_receive(rq, budget, &xdp_xmit);
1220

1221
	/* Out of packets? */
1222 1223
	if (received < budget)
		virtqueue_napi_complete(napi, rq->vq, received);
R
Rusty Russell 已提交
1224

J
Jason Wang 已提交
1225 1226 1227
	if (xdp_xmit)
		xdp_do_flush_map();

R
Rusty Russell 已提交
1228 1229 1230
	return received;
}

J
Jason Wang 已提交
1231 1232 1233
static int virtnet_open(struct net_device *dev)
{
	struct virtnet_info *vi = netdev_priv(dev);
1234
	int i, err;
J
Jason Wang 已提交
1235

1236 1237 1238
	for (i = 0; i < vi->max_queue_pairs; i++) {
		if (i < vi->curr_queue_pairs)
			/* Make sure we have some buffers: if oom use wq. */
M
Michael S. Tsirkin 已提交
1239
			if (!try_fill_recv(vi, &vi->rq[i], GFP_KERNEL))
1240
				schedule_delayed_work(&vi->refill, 0);
1241 1242 1243 1244 1245

		err = xdp_rxq_info_reg(&vi->rq[i].xdp_rxq, dev, i);
		if (err < 0)
			return err;

1246
		virtnet_napi_enable(vi->rq[i].vq, &vi->rq[i].napi);
W
Willem de Bruijn 已提交
1247
		virtnet_napi_tx_enable(vi, vi->sq[i].vq, &vi->sq[i].napi);
J
Jason Wang 已提交
1248 1249 1250 1251 1252
	}

	return 0;
}

W
Willem de Bruijn 已提交
1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270
static int virtnet_poll_tx(struct napi_struct *napi, int budget)
{
	struct send_queue *sq = container_of(napi, struct send_queue, napi);
	struct virtnet_info *vi = sq->vq->vdev->priv;
	struct netdev_queue *txq = netdev_get_tx_queue(vi->dev, vq2txq(sq->vq));

	__netif_tx_lock(txq, raw_smp_processor_id());
	free_old_xmit_skbs(sq);
	__netif_tx_unlock(txq);

	virtqueue_napi_complete(napi, sq->vq, 0);

	if (sq->vq->num_free >= 2 + MAX_SKB_FRAGS)
		netif_tx_wake_queue(txq);

	return 0;
}

1271
static int xmit_skb(struct send_queue *sq, struct sk_buff *skb)
R
Rusty Russell 已提交
1272
{
1273
	struct virtio_net_hdr_mrg_rxbuf *hdr;
R
Rusty Russell 已提交
1274
	const unsigned char *dest = ((struct ethhdr *)skb->data)->h_dest;
1275
	struct virtnet_info *vi = sq->vq->vdev->priv;
1276
	int num_sg;
1277
	unsigned hdr_len = vi->hdr_len;
1278
	bool can_push;
R
Rusty Russell 已提交
1279

J
Johannes Berg 已提交
1280
	pr_debug("%s: xmit %p %pM\n", vi->dev->name, skb, dest);
1281 1282 1283 1284 1285 1286 1287

	can_push = vi->any_header_sg &&
		!((unsigned long)skb->data & (__alignof__(*hdr) - 1)) &&
		!skb_header_cloned(skb) && skb_headroom(skb) >= hdr_len;
	/* Even if we can, don't push here yet as this would skew
	 * csum_start offset below. */
	if (can_push)
1288
		hdr = (struct virtio_net_hdr_mrg_rxbuf *)(skb->data - hdr_len);
1289 1290
	else
		hdr = skb_vnet_hdr(skb);
R
Rusty Russell 已提交
1291

1292
	if (virtio_net_hdr_from_skb(skb, &hdr->hdr,
1293
				    virtio_is_little_endian(vi->vdev), false))
1294
		BUG();
R
Rusty Russell 已提交
1295

1296
	if (vi->mergeable_rx_bufs)
1297
		hdr->num_buffers = 0;
1298

1299
	sg_init_table(sq->sg, skb_shinfo(skb)->nr_frags + (can_push ? 1 : 2));
1300 1301 1302
	if (can_push) {
		__skb_push(skb, hdr_len);
		num_sg = skb_to_sgvec(skb, sq->sg, 0, skb->len);
1303 1304
		if (unlikely(num_sg < 0))
			return num_sg;
1305 1306 1307 1308
		/* Pull header back to avoid skew in tx bytes calculations. */
		__skb_pull(skb, hdr_len);
	} else {
		sg_set_buf(sq->sg, hdr, hdr_len);
1309 1310 1311 1312
		num_sg = skb_to_sgvec(skb, sq->sg + 1, 0, skb->len);
		if (unlikely(num_sg < 0))
			return num_sg;
		num_sg++;
1313
	}
1314
	return virtqueue_add_outbuf(sq->vq, sq->sg, num_sg, skb, GFP_ATOMIC);
1315 1316
}

1317
static netdev_tx_t start_xmit(struct sk_buff *skb, struct net_device *dev)
1318 1319
{
	struct virtnet_info *vi = netdev_priv(dev);
J
Jason Wang 已提交
1320 1321
	int qnum = skb_get_queue_mapping(skb);
	struct send_queue *sq = &vi->sq[qnum];
1322
	int err;
1323 1324
	struct netdev_queue *txq = netdev_get_tx_queue(dev, qnum);
	bool kick = !skb->xmit_more;
W
Willem de Bruijn 已提交
1325
	bool use_napi = sq->napi.weight;
1326 1327

	/* Free up any pending old buffers before queueing new ones. */
1328
	free_old_xmit_skbs(sq);
1329

1330 1331 1332
	if (use_napi && kick)
		virtqueue_enable_cb_delayed(sq->vq);

1333 1334 1335
	/* timestamp packet in software */
	skb_tx_timestamp(skb);

1336
	/* Try to transmit */
1337
	err = xmit_skb(sq, skb);
1338

1339
	/* This should not happen! */
1340
	if (unlikely(err)) {
1341 1342 1343
		dev->stats.tx_fifo_errors++;
		if (net_ratelimit())
			dev_warn(&dev->dev,
1344
				 "Unexpected TXQ (%d) queue failure: %d\n", qnum, err);
1345
		dev->stats.tx_dropped++;
1346
		dev_kfree_skb_any(skb);
1347
		return NETDEV_TX_OK;
R
Rusty Russell 已提交
1348
	}
1349

1350
	/* Don't wait up for transmitted skbs to be freed. */
W
Willem de Bruijn 已提交
1351 1352 1353 1354
	if (!use_napi) {
		skb_orphan(skb);
		nf_reset(skb);
	}
1355

1356 1357 1358 1359 1360 1361 1362 1363 1364
	/* If running out of space, stop queue to avoid getting packets that we
	 * are then unable to transmit.
	 * An alternative would be to force queuing layer to requeue the skb by
	 * returning NETDEV_TX_BUSY. However, NETDEV_TX_BUSY should not be
	 * returned in a normal path of operation: it means that driver is not
	 * maintaining the TX queue stop/start state properly, and causes
	 * the stack to do a non-trivial amount of useless work.
	 * Since most packets only take 1 or 2 ring slots, stopping the queue
	 * early means 16 slots are typically wasted.
1365
	 */
1366
	if (sq->vq->num_free < 2+MAX_SKB_FRAGS) {
J
Jason Wang 已提交
1367
		netif_stop_subqueue(dev, qnum);
W
Willem de Bruijn 已提交
1368 1369
		if (!use_napi &&
		    unlikely(!virtqueue_enable_cb_delayed(sq->vq))) {
1370
			/* More just got used, free them then recheck. */
1371 1372
			free_old_xmit_skbs(sq);
			if (sq->vq->num_free >= 2+MAX_SKB_FRAGS) {
J
Jason Wang 已提交
1373
				netif_start_subqueue(dev, qnum);
1374
				virtqueue_disable_cb(sq->vq);
1375 1376
			}
		}
1377
	}
1378

1379
	if (kick || netif_xmit_stopped(txq))
1380
		virtqueue_kick(sq->vq);
R
Rusty Russell 已提交
1381

1382
	return NETDEV_TX_OK;
1383 1384
}

1385 1386 1387
/*
 * Send command via the control virtqueue and check status.  Commands
 * supported by the hypervisor, as indicated by feature bits, should
S
stephen hemminger 已提交
1388
 * never fail unless improperly formatted.
1389 1390
 */
static bool virtnet_send_command(struct virtnet_info *vi, u8 class, u8 cmd,
1391
				 struct scatterlist *out)
1392
{
1393
	struct scatterlist *sgs[4], hdr, stat;
1394
	unsigned out_num = 0, tmp;
1395 1396

	/* Caller should know better */
1397
	BUG_ON(!virtio_has_feature(vi->vdev, VIRTIO_NET_F_CTRL_VQ));
1398

1399 1400 1401
	vi->ctrl_status = ~0;
	vi->ctrl_hdr.class = class;
	vi->ctrl_hdr.cmd = cmd;
1402
	/* Add header */
1403
	sg_init_one(&hdr, &vi->ctrl_hdr, sizeof(vi->ctrl_hdr));
1404
	sgs[out_num++] = &hdr;
1405

1406 1407
	if (out)
		sgs[out_num++] = out;
1408

1409
	/* Add return status. */
1410
	sg_init_one(&stat, &vi->ctrl_status, sizeof(vi->ctrl_status));
1411
	sgs[out_num] = &stat;
1412

1413
	BUG_ON(out_num + 1 > ARRAY_SIZE(sgs));
1414
	virtqueue_add_sgs(vi->cvq, sgs, out_num, 1, vi, GFP_ATOMIC);
1415

1416
	if (unlikely(!virtqueue_kick(vi->cvq)))
1417
		return vi->ctrl_status == VIRTIO_NET_OK;
1418 1419 1420 1421

	/* Spin for a response, the kick causes an ioport write, trapping
	 * into the hypervisor, so the request should be handled immediately.
	 */
1422 1423
	while (!virtqueue_get_buf(vi->cvq, &tmp) &&
	       !virtqueue_is_broken(vi->cvq))
1424 1425
		cpu_relax();

1426
	return vi->ctrl_status == VIRTIO_NET_OK;
1427 1428
}

1429 1430 1431 1432
static int virtnet_set_mac_address(struct net_device *dev, void *p)
{
	struct virtnet_info *vi = netdev_priv(dev);
	struct virtio_device *vdev = vi->vdev;
1433
	int ret;
1434
	struct sockaddr *addr;
1435
	struct scatterlist sg;
1436

1437
	addr = kmemdup(p, sizeof(*addr), GFP_KERNEL);
1438 1439 1440 1441
	if (!addr)
		return -ENOMEM;

	ret = eth_prepare_mac_addr_change(dev, addr);
1442
	if (ret)
1443
		goto out;
1444

1445 1446 1447
	if (virtio_has_feature(vdev, VIRTIO_NET_F_CTRL_MAC_ADDR)) {
		sg_init_one(&sg, addr->sa_data, dev->addr_len);
		if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_MAC,
1448
					  VIRTIO_NET_CTRL_MAC_ADDR_SET, &sg)) {
1449 1450
			dev_warn(&vdev->dev,
				 "Failed to set mac address by vq command.\n");
1451 1452
			ret = -EINVAL;
			goto out;
1453
		}
1454 1455
	} else if (virtio_has_feature(vdev, VIRTIO_NET_F_MAC) &&
		   !virtio_has_feature(vdev, VIRTIO_F_VERSION_1)) {
1456 1457 1458 1459 1460 1461 1462
		unsigned int i;

		/* Naturally, this has an atomicity problem. */
		for (i = 0; i < dev->addr_len; i++)
			virtio_cwrite8(vdev,
				       offsetof(struct virtio_net_config, mac) +
				       i, addr->sa_data[i]);
1463 1464 1465
	}

	eth_commit_mac_addr_change(dev, p);
1466
	ret = 0;
1467

1468 1469 1470
out:
	kfree(addr);
	return ret;
1471 1472
}

1473 1474
static void virtnet_stats(struct net_device *dev,
			  struct rtnl_link_stats64 *tot)
1475 1476 1477 1478 1479 1480
{
	struct virtnet_info *vi = netdev_priv(dev);
	int cpu;
	unsigned int start;

	for_each_possible_cpu(cpu) {
E
Eric Dumazet 已提交
1481
		struct virtnet_stats *stats = per_cpu_ptr(vi->stats, cpu);
1482 1483 1484
		u64 tpackets, tbytes, rpackets, rbytes;

		do {
1485
			start = u64_stats_fetch_begin_irq(&stats->tx_syncp);
1486 1487
			tpackets = stats->tx_packets;
			tbytes   = stats->tx_bytes;
1488
		} while (u64_stats_fetch_retry_irq(&stats->tx_syncp, start));
1489 1490

		do {
1491
			start = u64_stats_fetch_begin_irq(&stats->rx_syncp);
1492 1493
			rpackets = stats->rx_packets;
			rbytes   = stats->rx_bytes;
1494
		} while (u64_stats_fetch_retry_irq(&stats->rx_syncp, start));
1495 1496 1497 1498 1499 1500 1501 1502

		tot->rx_packets += rpackets;
		tot->tx_packets += tpackets;
		tot->rx_bytes   += rbytes;
		tot->tx_bytes   += tbytes;
	}

	tot->tx_dropped = dev->stats.tx_dropped;
1503
	tot->tx_fifo_errors = dev->stats.tx_fifo_errors;
1504 1505 1506 1507 1508
	tot->rx_dropped = dev->stats.rx_dropped;
	tot->rx_length_errors = dev->stats.rx_length_errors;
	tot->rx_frame_errors = dev->stats.rx_frame_errors;
}

1509 1510 1511 1512
#ifdef CONFIG_NET_POLL_CONTROLLER
static void virtnet_netpoll(struct net_device *dev)
{
	struct virtnet_info *vi = netdev_priv(dev);
J
Jason Wang 已提交
1513
	int i;
1514

J
Jason Wang 已提交
1515 1516
	for (i = 0; i < vi->curr_queue_pairs; i++)
		napi_schedule(&vi->rq[i].napi);
1517 1518 1519
}
#endif

1520 1521 1522 1523
static void virtnet_ack_link_announce(struct virtnet_info *vi)
{
	rtnl_lock();
	if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_ANNOUNCE,
1524
				  VIRTIO_NET_CTRL_ANNOUNCE_ACK, NULL))
1525 1526 1527 1528
		dev_warn(&vi->dev->dev, "Failed to ack link announce.\n");
	rtnl_unlock();
}

1529
static int _virtnet_set_queues(struct virtnet_info *vi, u16 queue_pairs)
J
Jason Wang 已提交
1530 1531 1532 1533 1534 1535 1536
{
	struct scatterlist sg;
	struct net_device *dev = vi->dev;

	if (!vi->has_cvq || !virtio_has_feature(vi->vdev, VIRTIO_NET_F_MQ))
		return 0;

1537 1538
	vi->ctrl_mq.virtqueue_pairs = cpu_to_virtio16(vi->vdev, queue_pairs);
	sg_init_one(&sg, &vi->ctrl_mq, sizeof(vi->ctrl_mq));
J
Jason Wang 已提交
1539 1540

	if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_MQ,
1541
				  VIRTIO_NET_CTRL_MQ_VQ_PAIRS_SET, &sg)) {
J
Jason Wang 已提交
1542 1543 1544
		dev_warn(&dev->dev, "Fail to set num of queue pairs to %d\n",
			 queue_pairs);
		return -EINVAL;
1545
	} else {
J
Jason Wang 已提交
1546
		vi->curr_queue_pairs = queue_pairs;
1547 1548 1549
		/* virtnet_open() will refill when device is going to up. */
		if (dev->flags & IFF_UP)
			schedule_delayed_work(&vi->refill, 0);
1550
	}
J
Jason Wang 已提交
1551 1552 1553 1554

	return 0;
}

1555 1556 1557 1558 1559 1560 1561 1562 1563 1564
static int virtnet_set_queues(struct virtnet_info *vi, u16 queue_pairs)
{
	int err;

	rtnl_lock();
	err = _virtnet_set_queues(vi, queue_pairs);
	rtnl_unlock();
	return err;
}

R
Rusty Russell 已提交
1565 1566 1567
static int virtnet_close(struct net_device *dev)
{
	struct virtnet_info *vi = netdev_priv(dev);
J
Jason Wang 已提交
1568
	int i;
R
Rusty Russell 已提交
1569

1570 1571
	/* Make sure refill_work doesn't re-enable napi! */
	cancel_delayed_work_sync(&vi->refill);
J
Jason Wang 已提交
1572

W
Willem de Bruijn 已提交
1573
	for (i = 0; i < vi->max_queue_pairs; i++) {
1574
		xdp_rxq_info_unreg(&vi->rq[i].xdp_rxq);
J
Jason Wang 已提交
1575
		napi_disable(&vi->rq[i].napi);
1576
		virtnet_napi_tx_disable(&vi->sq[i].napi);
W
Willem de Bruijn 已提交
1577
	}
R
Rusty Russell 已提交
1578 1579 1580 1581

	return 0;
}

1582 1583 1584
static void virtnet_set_rx_mode(struct net_device *dev)
{
	struct virtnet_info *vi = netdev_priv(dev);
1585 1586
	struct scatterlist sg[2];
	struct virtio_net_ctrl_mac *mac_data;
J
Jiri Pirko 已提交
1587
	struct netdev_hw_addr *ha;
1588
	int uc_count;
1589
	int mc_count;
1590 1591
	void *buf;
	int i;
1592

S
stephen hemminger 已提交
1593
	/* We can't dynamically set ndo_set_rx_mode, so return gracefully */
1594 1595 1596
	if (!virtio_has_feature(vi->vdev, VIRTIO_NET_F_CTRL_RX))
		return;

1597 1598
	vi->ctrl_promisc = ((dev->flags & IFF_PROMISC) != 0);
	vi->ctrl_allmulti = ((dev->flags & IFF_ALLMULTI) != 0);
1599

1600
	sg_init_one(sg, &vi->ctrl_promisc, sizeof(vi->ctrl_promisc));
1601 1602

	if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_RX,
1603
				  VIRTIO_NET_CTRL_RX_PROMISC, sg))
1604
		dev_warn(&dev->dev, "Failed to %sable promisc mode.\n",
1605
			 vi->ctrl_promisc ? "en" : "dis");
1606

1607
	sg_init_one(sg, &vi->ctrl_allmulti, sizeof(vi->ctrl_allmulti));
1608 1609

	if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_RX,
1610
				  VIRTIO_NET_CTRL_RX_ALLMULTI, sg))
1611
		dev_warn(&dev->dev, "Failed to %sable allmulti mode.\n",
1612
			 vi->ctrl_allmulti ? "en" : "dis");
1613

1614
	uc_count = netdev_uc_count(dev);
1615
	mc_count = netdev_mc_count(dev);
1616
	/* MAC filter - use one buffer for both lists */
1617 1618 1619
	buf = kzalloc(((uc_count + mc_count) * ETH_ALEN) +
		      (2 * sizeof(mac_data->entries)), GFP_ATOMIC);
	mac_data = buf;
1620
	if (!buf)
1621 1622
		return;

1623 1624
	sg_init_table(sg, 2);

1625
	/* Store the unicast list and count in the front of the buffer */
M
Michael S. Tsirkin 已提交
1626
	mac_data->entries = cpu_to_virtio32(vi->vdev, uc_count);
J
Jiri Pirko 已提交
1627
	i = 0;
1628
	netdev_for_each_uc_addr(ha, dev)
J
Jiri Pirko 已提交
1629
		memcpy(&mac_data->macs[i++][0], ha->addr, ETH_ALEN);
1630 1631

	sg_set_buf(&sg[0], mac_data,
1632
		   sizeof(mac_data->entries) + (uc_count * ETH_ALEN));
1633 1634

	/* multicast list and count fill the end */
1635
	mac_data = (void *)&mac_data->macs[uc_count][0];
1636

M
Michael S. Tsirkin 已提交
1637
	mac_data->entries = cpu_to_virtio32(vi->vdev, mc_count);
1638
	i = 0;
1639 1640
	netdev_for_each_mc_addr(ha, dev)
		memcpy(&mac_data->macs[i++][0], ha->addr, ETH_ALEN);
1641 1642

	sg_set_buf(&sg[1], mac_data,
1643
		   sizeof(mac_data->entries) + (mc_count * ETH_ALEN));
1644 1645

	if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_MAC,
1646
				  VIRTIO_NET_CTRL_MAC_TABLE_SET, sg))
1647
		dev_warn(&dev->dev, "Failed to set MAC filter table.\n");
1648 1649

	kfree(buf);
1650 1651
}

1652 1653
static int virtnet_vlan_rx_add_vid(struct net_device *dev,
				   __be16 proto, u16 vid)
1654 1655 1656 1657
{
	struct virtnet_info *vi = netdev_priv(dev);
	struct scatterlist sg;

1658 1659
	vi->ctrl_vid = vid;
	sg_init_one(&sg, &vi->ctrl_vid, sizeof(vi->ctrl_vid));
1660 1661

	if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_VLAN,
1662
				  VIRTIO_NET_CTRL_VLAN_ADD, &sg))
1663
		dev_warn(&dev->dev, "Failed to add VLAN ID %d.\n", vid);
1664
	return 0;
1665 1666
}

1667 1668
static int virtnet_vlan_rx_kill_vid(struct net_device *dev,
				    __be16 proto, u16 vid)
1669 1670 1671 1672
{
	struct virtnet_info *vi = netdev_priv(dev);
	struct scatterlist sg;

1673 1674
	vi->ctrl_vid = vid;
	sg_init_one(&sg, &vi->ctrl_vid, sizeof(vi->ctrl_vid));
1675 1676

	if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_VLAN,
1677
				  VIRTIO_NET_CTRL_VLAN_DEL, &sg))
1678
		dev_warn(&dev->dev, "Failed to kill VLAN ID %d.\n", vid);
1679
	return 0;
1680 1681
}

1682
static void virtnet_clean_affinity(struct virtnet_info *vi, long hcpu)
J
Jason Wang 已提交
1683 1684 1685
{
	int i;

1686 1687
	if (vi->affinity_hint_set) {
		for (i = 0; i < vi->max_queue_pairs; i++) {
1688 1689 1690 1691
			virtqueue_set_affinity(vi->rq[i].vq, -1);
			virtqueue_set_affinity(vi->sq[i].vq, -1);
		}

1692 1693 1694
		vi->affinity_hint_set = false;
	}
}
1695

1696 1697 1698 1699
static void virtnet_set_affinity(struct virtnet_info *vi)
{
	int i;
	int cpu;
J
Jason Wang 已提交
1700 1701 1702 1703 1704

	/* In multiqueue mode, when the number of cpu is equal to the number of
	 * queue pairs, we let the queue pairs to be private to one cpu by
	 * setting the affinity hint to eliminate the contention.
	 */
1705 1706 1707 1708
	if (vi->curr_queue_pairs == 1 ||
	    vi->max_queue_pairs != num_online_cpus()) {
		virtnet_clean_affinity(vi, -1);
		return;
J
Jason Wang 已提交
1709 1710
	}

1711 1712
	i = 0;
	for_each_online_cpu(cpu) {
J
Jason Wang 已提交
1713 1714
		virtqueue_set_affinity(vi->rq[i].vq, cpu);
		virtqueue_set_affinity(vi->sq[i].vq, cpu);
1715
		netif_set_xps_queue(vi->dev, cpumask_of(cpu), i);
1716
		i++;
J
Jason Wang 已提交
1717 1718
	}

1719
	vi->affinity_hint_set = true;
J
Jason Wang 已提交
1720 1721
}

1722
static int virtnet_cpu_online(unsigned int cpu, struct hlist_node *node)
1723
{
1724 1725 1726 1727 1728
	struct virtnet_info *vi = hlist_entry_safe(node, struct virtnet_info,
						   node);
	virtnet_set_affinity(vi);
	return 0;
}
1729

1730 1731 1732 1733 1734 1735 1736
static int virtnet_cpu_dead(unsigned int cpu, struct hlist_node *node)
{
	struct virtnet_info *vi = hlist_entry_safe(node, struct virtnet_info,
						   node_dead);
	virtnet_set_affinity(vi);
	return 0;
}
1737

1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768
static int virtnet_cpu_down_prep(unsigned int cpu, struct hlist_node *node)
{
	struct virtnet_info *vi = hlist_entry_safe(node, struct virtnet_info,
						   node);

	virtnet_clean_affinity(vi, cpu);
	return 0;
}

static enum cpuhp_state virtionet_online;

static int virtnet_cpu_notif_add(struct virtnet_info *vi)
{
	int ret;

	ret = cpuhp_state_add_instance_nocalls(virtionet_online, &vi->node);
	if (ret)
		return ret;
	ret = cpuhp_state_add_instance_nocalls(CPUHP_VIRT_NET_DEAD,
					       &vi->node_dead);
	if (!ret)
		return ret;
	cpuhp_state_remove_instance_nocalls(virtionet_online, &vi->node);
	return ret;
}

static void virtnet_cpu_notif_remove(struct virtnet_info *vi)
{
	cpuhp_state_remove_instance_nocalls(virtionet_online, &vi->node);
	cpuhp_state_remove_instance_nocalls(CPUHP_VIRT_NET_DEAD,
					    &vi->node_dead);
J
Jason Wang 已提交
1769 1770
}

R
Rick Jones 已提交
1771 1772 1773 1774 1775
static void virtnet_get_ringparam(struct net_device *dev,
				struct ethtool_ringparam *ring)
{
	struct virtnet_info *vi = netdev_priv(dev);

J
Jason Wang 已提交
1776 1777
	ring->rx_max_pending = virtqueue_get_vring_size(vi->rq[0].vq);
	ring->tx_max_pending = virtqueue_get_vring_size(vi->sq[0].vq);
R
Rick Jones 已提交
1778 1779 1780 1781
	ring->rx_pending = ring->rx_max_pending;
	ring->tx_pending = ring->tx_max_pending;
}

1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794

static void virtnet_get_drvinfo(struct net_device *dev,
				struct ethtool_drvinfo *info)
{
	struct virtnet_info *vi = netdev_priv(dev);
	struct virtio_device *vdev = vi->vdev;

	strlcpy(info->driver, KBUILD_MODNAME, sizeof(info->driver));
	strlcpy(info->version, VIRTNET_DRIVER_VERSION, sizeof(info->version));
	strlcpy(info->bus_info, virtio_bus_name(vdev), sizeof(info->bus_info));

}

1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808
/* TODO: Eliminate OOO packets during switching */
static int virtnet_set_channels(struct net_device *dev,
				struct ethtool_channels *channels)
{
	struct virtnet_info *vi = netdev_priv(dev);
	u16 queue_pairs = channels->combined_count;
	int err;

	/* We don't support separate rx/tx channels.
	 * We don't allow setting 'other' channels.
	 */
	if (channels->rx_count || channels->tx_count || channels->other_count)
		return -EINVAL;

1809
	if (queue_pairs > vi->max_queue_pairs || queue_pairs == 0)
1810 1811
		return -EINVAL;

J
John Fastabend 已提交
1812 1813 1814 1815 1816 1817 1818
	/* For now we don't support modifying channels while XDP is loaded
	 * also when XDP is loaded all RX queues have XDP programs so we only
	 * need to check a single RX queue.
	 */
	if (vi->rq[0].xdp_prog)
		return -EINVAL;

1819
	get_online_cpus();
1820
	err = _virtnet_set_queues(vi, queue_pairs);
1821 1822 1823 1824
	if (!err) {
		netif_set_real_num_tx_queues(dev, queue_pairs);
		netif_set_real_num_rx_queues(dev, queue_pairs);

1825
		virtnet_set_affinity(vi);
1826
	}
1827
	put_online_cpus();
1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844

	return err;
}

static void virtnet_get_channels(struct net_device *dev,
				 struct ethtool_channels *channels)
{
	struct virtnet_info *vi = netdev_priv(dev);

	channels->combined_count = vi->curr_queue_pairs;
	channels->max_combined = vi->max_queue_pairs;
	channels->max_other = 0;
	channels->rx_count = 0;
	channels->tx_count = 0;
	channels->other_count = 0;
}

1845
/* Check if the user is trying to change anything besides speed/duplex */
1846 1847
static bool
virtnet_validate_ethtool_cmd(const struct ethtool_link_ksettings *cmd)
1848
{
1849 1850
	struct ethtool_link_ksettings diff1 = *cmd;
	struct ethtool_link_ksettings diff2 = {};
1851

1852 1853 1854
	/* cmd is always set so we need to clear it, validate the port type
	 * and also without autonegotiation we can ignore advertising
	 */
1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868
	diff1.base.speed = 0;
	diff2.base.port = PORT_OTHER;
	ethtool_link_ksettings_zero_link_mode(&diff1, advertising);
	diff1.base.duplex = 0;
	diff1.base.cmd = 0;
	diff1.base.link_mode_masks_nwords = 0;

	return !memcmp(&diff1.base, &diff2.base, sizeof(diff1.base)) &&
		bitmap_empty(diff1.link_modes.supported,
			     __ETHTOOL_LINK_MODE_MASK_NBITS) &&
		bitmap_empty(diff1.link_modes.advertising,
			     __ETHTOOL_LINK_MODE_MASK_NBITS) &&
		bitmap_empty(diff1.link_modes.lp_advertising,
			     __ETHTOOL_LINK_MODE_MASK_NBITS);
1869 1870
}

1871 1872
static int virtnet_set_link_ksettings(struct net_device *dev,
				      const struct ethtool_link_ksettings *cmd)
1873 1874 1875 1876
{
	struct virtnet_info *vi = netdev_priv(dev);
	u32 speed;

1877
	speed = cmd->base.speed;
1878 1879
	/* don't allow custom speed and duplex */
	if (!ethtool_validate_speed(speed) ||
1880
	    !ethtool_validate_duplex(cmd->base.duplex) ||
1881 1882 1883
	    !virtnet_validate_ethtool_cmd(cmd))
		return -EINVAL;
	vi->speed = speed;
1884
	vi->duplex = cmd->base.duplex;
1885 1886 1887 1888

	return 0;
}

1889 1890
static int virtnet_get_link_ksettings(struct net_device *dev,
				      struct ethtool_link_ksettings *cmd)
1891 1892 1893
{
	struct virtnet_info *vi = netdev_priv(dev);

1894 1895 1896
	cmd->base.speed = vi->speed;
	cmd->base.duplex = vi->duplex;
	cmd->base.port = PORT_OTHER;
1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908

	return 0;
}

static void virtnet_init_settings(struct net_device *dev)
{
	struct virtnet_info *vi = netdev_priv(dev);

	vi->speed = SPEED_UNKNOWN;
	vi->duplex = DUPLEX_UNKNOWN;
}

1909 1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925 1926
static void virtnet_update_settings(struct virtnet_info *vi)
{
	u32 speed;
	u8 duplex;

	if (!virtio_has_feature(vi->vdev, VIRTIO_NET_F_SPEED_DUPLEX))
		return;

	speed = virtio_cread32(vi->vdev, offsetof(struct virtio_net_config,
						  speed));
	if (ethtool_validate_speed(speed))
		vi->speed = speed;
	duplex = virtio_cread8(vi->vdev, offsetof(struct virtio_net_config,
						  duplex));
	if (ethtool_validate_duplex(duplex))
		vi->duplex = duplex;
}

1927
static const struct ethtool_ops virtnet_ethtool_ops = {
1928
	.get_drvinfo = virtnet_get_drvinfo,
1929
	.get_link = ethtool_op_get_link,
R
Rick Jones 已提交
1930
	.get_ringparam = virtnet_get_ringparam,
1931 1932
	.set_channels = virtnet_set_channels,
	.get_channels = virtnet_get_channels,
1933
	.get_ts_info = ethtool_op_get_ts_info,
1934 1935
	.get_link_ksettings = virtnet_get_link_ksettings,
	.set_link_ksettings = virtnet_set_link_ksettings,
1936 1937
};

1938 1939 1940 1941 1942 1943 1944 1945 1946
static void virtnet_freeze_down(struct virtio_device *vdev)
{
	struct virtnet_info *vi = vdev->priv;
	int i;

	/* Make sure no work handler is accessing the device */
	flush_work(&vi->config_work);

	netif_device_detach(vi->dev);
1947
	netif_tx_disable(vi->dev);
1948 1949 1950
	cancel_delayed_work_sync(&vi->refill);

	if (netif_running(vi->dev)) {
W
Willem de Bruijn 已提交
1951
		for (i = 0; i < vi->max_queue_pairs; i++) {
1952
			napi_disable(&vi->rq[i].napi);
1953
			virtnet_napi_tx_disable(&vi->sq[i].napi);
W
Willem de Bruijn 已提交
1954
		}
1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975
	}
}

static int init_vqs(struct virtnet_info *vi);

static int virtnet_restore_up(struct virtio_device *vdev)
{
	struct virtnet_info *vi = vdev->priv;
	int err, i;

	err = init_vqs(vi);
	if (err)
		return err;

	virtio_device_ready(vdev);

	if (netif_running(vi->dev)) {
		for (i = 0; i < vi->curr_queue_pairs; i++)
			if (!try_fill_recv(vi, &vi->rq[i], GFP_KERNEL))
				schedule_delayed_work(&vi->refill, 0);

W
Willem de Bruijn 已提交
1976
		for (i = 0; i < vi->max_queue_pairs; i++) {
1977
			virtnet_napi_enable(vi->rq[i].vq, &vi->rq[i].napi);
W
Willem de Bruijn 已提交
1978 1979 1980
			virtnet_napi_tx_enable(vi, vi->sq[i].vq,
					       &vi->sq[i].napi);
		}
1981 1982 1983 1984 1985 1986
	}

	netif_device_attach(vi->dev);
	return err;
}

1987 1988 1989 1990 1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027
static int virtnet_set_guest_offloads(struct virtnet_info *vi, u64 offloads)
{
	struct scatterlist sg;
	vi->ctrl_offloads = cpu_to_virtio64(vi->vdev, offloads);

	sg_init_one(&sg, &vi->ctrl_offloads, sizeof(vi->ctrl_offloads));

	if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_GUEST_OFFLOADS,
				  VIRTIO_NET_CTRL_GUEST_OFFLOADS_SET, &sg)) {
		dev_warn(&vi->dev->dev, "Fail to set guest offload. \n");
		return -EINVAL;
	}

	return 0;
}

static int virtnet_clear_guest_offloads(struct virtnet_info *vi)
{
	u64 offloads = 0;

	if (!vi->guest_offloads)
		return 0;

	if (virtio_has_feature(vi->vdev, VIRTIO_NET_F_GUEST_CSUM))
		offloads = 1ULL << VIRTIO_NET_F_GUEST_CSUM;

	return virtnet_set_guest_offloads(vi, offloads);
}

static int virtnet_restore_guest_offloads(struct virtnet_info *vi)
{
	u64 offloads = vi->guest_offloads;

	if (!vi->guest_offloads)
		return 0;
	if (virtio_has_feature(vi->vdev, VIRTIO_NET_F_GUEST_CSUM))
		offloads |= 1ULL << VIRTIO_NET_F_GUEST_CSUM;

	return virtnet_set_guest_offloads(vi, offloads);
}

2028 2029
static int virtnet_xdp_set(struct net_device *dev, struct bpf_prog *prog,
			   struct netlink_ext_ack *extack)
J
John Fastabend 已提交
2030 2031 2032 2033
{
	unsigned long int max_sz = PAGE_SIZE - sizeof(struct padded_vnet_hdr);
	struct virtnet_info *vi = netdev_priv(dev);
	struct bpf_prog *old_prog;
2034
	u16 xdp_qp = 0, curr_qp;
2035
	int i, err;
J
John Fastabend 已提交
2036

2037 2038 2039 2040 2041
	if (!virtio_has_feature(vi->vdev, VIRTIO_NET_F_CTRL_GUEST_OFFLOADS)
	    && (virtio_has_feature(vi->vdev, VIRTIO_NET_F_GUEST_TSO4) ||
	        virtio_has_feature(vi->vdev, VIRTIO_NET_F_GUEST_TSO6) ||
	        virtio_has_feature(vi->vdev, VIRTIO_NET_F_GUEST_ECN) ||
		virtio_has_feature(vi->vdev, VIRTIO_NET_F_GUEST_UFO))) {
2042
		NL_SET_ERR_MSG_MOD(extack, "Can't set XDP while host is implementing LRO, disable LRO first");
J
John Fastabend 已提交
2043 2044 2045 2046
		return -EOPNOTSUPP;
	}

	if (vi->mergeable_rx_bufs && !vi->any_header_sg) {
2047
		NL_SET_ERR_MSG_MOD(extack, "XDP expects header/data in single page, any_header_sg required");
J
John Fastabend 已提交
2048 2049 2050 2051
		return -EINVAL;
	}

	if (dev->mtu > max_sz) {
2052
		NL_SET_ERR_MSG_MOD(extack, "MTU too large to enable XDP");
J
John Fastabend 已提交
2053 2054 2055 2056
		netdev_warn(dev, "XDP requires MTU less than %lu\n", max_sz);
		return -EINVAL;
	}

2057 2058 2059 2060 2061 2062
	curr_qp = vi->curr_queue_pairs - vi->xdp_queue_pairs;
	if (prog)
		xdp_qp = nr_cpu_ids;

	/* XDP requires extra queues for XDP_TX */
	if (curr_qp + xdp_qp > vi->max_queue_pairs) {
2063
		NL_SET_ERR_MSG_MOD(extack, "Too few free TX rings available");
2064 2065 2066 2067 2068
		netdev_warn(dev, "request %i queues but max is %i\n",
			    curr_qp + xdp_qp, vi->max_queue_pairs);
		return -ENOMEM;
	}

2069 2070 2071 2072 2073 2074
	if (prog) {
		prog = bpf_prog_add(prog, vi->max_queue_pairs - 1);
		if (IS_ERR(prog))
			return PTR_ERR(prog);
	}

2075 2076 2077
	/* Make sure NAPI is not using any XDP TX queues for RX. */
	for (i = 0; i < vi->max_queue_pairs; i++)
		napi_disable(&vi->rq[i].napi);
J
John Fastabend 已提交
2078

2079
	netif_set_real_num_rx_queues(dev, curr_qp + xdp_qp);
2080 2081 2082 2083
	err = _virtnet_set_queues(vi, curr_qp + xdp_qp);
	if (err)
		goto err;
	vi->xdp_queue_pairs = xdp_qp;
2084

J
John Fastabend 已提交
2085 2086 2087
	for (i = 0; i < vi->max_queue_pairs; i++) {
		old_prog = rtnl_dereference(vi->rq[i].xdp_prog);
		rcu_assign_pointer(vi->rq[i].xdp_prog, prog);
2088 2089 2090 2091 2092 2093
		if (i == 0) {
			if (!old_prog)
				virtnet_clear_guest_offloads(vi);
			if (!prog)
				virtnet_restore_guest_offloads(vi);
		}
J
John Fastabend 已提交
2094 2095
		if (old_prog)
			bpf_prog_put(old_prog);
2096
		virtnet_napi_enable(vi->rq[i].vq, &vi->rq[i].napi);
J
John Fastabend 已提交
2097 2098 2099
	}

	return 0;
2100

2101 2102 2103
err:
	for (i = 0; i < vi->max_queue_pairs; i++)
		virtnet_napi_enable(vi->rq[i].vq, &vi->rq[i].napi);
2104 2105 2106
	if (prog)
		bpf_prog_sub(prog, vi->max_queue_pairs - 1);
	return err;
J
John Fastabend 已提交
2107 2108
}

2109
static u32 virtnet_xdp_query(struct net_device *dev)
J
John Fastabend 已提交
2110 2111
{
	struct virtnet_info *vi = netdev_priv(dev);
2112
	const struct bpf_prog *xdp_prog;
J
John Fastabend 已提交
2113 2114 2115
	int i;

	for (i = 0; i < vi->max_queue_pairs; i++) {
2116 2117 2118
		xdp_prog = rtnl_dereference(vi->rq[i].xdp_prog);
		if (xdp_prog)
			return xdp_prog->aux->id;
J
John Fastabend 已提交
2119
	}
2120
	return 0;
J
John Fastabend 已提交
2121 2122
}

2123
static int virtnet_xdp(struct net_device *dev, struct netdev_bpf *xdp)
J
John Fastabend 已提交
2124 2125 2126
{
	switch (xdp->command) {
	case XDP_SETUP_PROG:
2127
		return virtnet_xdp_set(dev, xdp->prog, xdp->extack);
J
John Fastabend 已提交
2128
	case XDP_QUERY_PROG:
2129 2130
		xdp->prog_id = virtnet_xdp_query(dev);
		xdp->prog_attached = !!xdp->prog_id;
J
John Fastabend 已提交
2131 2132 2133 2134 2135 2136
		return 0;
	default:
		return -EINVAL;
	}
}

2137 2138 2139 2140 2141
static const struct net_device_ops virtnet_netdev = {
	.ndo_open            = virtnet_open,
	.ndo_stop   	     = virtnet_close,
	.ndo_start_xmit      = start_xmit,
	.ndo_validate_addr   = eth_validate_addr,
2142
	.ndo_set_mac_address = virtnet_set_mac_address,
2143
	.ndo_set_rx_mode     = virtnet_set_rx_mode,
2144
	.ndo_get_stats64     = virtnet_stats,
2145 2146
	.ndo_vlan_rx_add_vid = virtnet_vlan_rx_add_vid,
	.ndo_vlan_rx_kill_vid = virtnet_vlan_rx_kill_vid,
2147 2148
#ifdef CONFIG_NET_POLL_CONTROLLER
	.ndo_poll_controller = virtnet_netpoll,
J
Jason Wang 已提交
2149
#endif
2150
	.ndo_bpf		= virtnet_xdp,
J
Jason Wang 已提交
2151 2152
	.ndo_xdp_xmit		= virtnet_xdp_xmit,
	.ndo_xdp_flush		= virtnet_xdp_flush,
2153
	.ndo_features_check	= passthru_features_check,
2154 2155
};

2156
static void virtnet_config_changed_work(struct work_struct *work)
2157
{
2158 2159
	struct virtnet_info *vi =
		container_of(work, struct virtnet_info, config_work);
2160 2161
	u16 v;

2162 2163
	if (virtio_cread_feature(vi->vdev, VIRTIO_NET_F_STATUS,
				 struct virtio_net_config, status, &v) < 0)
M
Michael S. Tsirkin 已提交
2164
		return;
2165 2166

	if (v & VIRTIO_NET_S_ANNOUNCE) {
2167
		netdev_notify_peers(vi->dev);
2168 2169
		virtnet_ack_link_announce(vi);
	}
2170 2171 2172 2173 2174

	/* Ignore unknown (future) status bits */
	v &= VIRTIO_NET_S_LINK_UP;

	if (vi->status == v)
M
Michael S. Tsirkin 已提交
2175
		return;
2176 2177 2178 2179

	vi->status = v;

	if (vi->status & VIRTIO_NET_S_LINK_UP) {
2180
		virtnet_update_settings(vi);
2181
		netif_carrier_on(vi->dev);
J
Jason Wang 已提交
2182
		netif_tx_wake_all_queues(vi->dev);
2183 2184
	} else {
		netif_carrier_off(vi->dev);
J
Jason Wang 已提交
2185
		netif_tx_stop_all_queues(vi->dev);
2186 2187 2188 2189 2190 2191 2192
	}
}

static void virtnet_config_changed(struct virtio_device *vdev)
{
	struct virtnet_info *vi = vdev->priv;

2193
	schedule_work(&vi->config_work);
2194 2195
}

J
Jason Wang 已提交
2196 2197
static void virtnet_free_queues(struct virtnet_info *vi)
{
2198 2199
	int i;

2200 2201
	for (i = 0; i < vi->max_queue_pairs; i++) {
		napi_hash_del(&vi->rq[i].napi);
2202
		netif_napi_del(&vi->rq[i].napi);
W
Willem de Bruijn 已提交
2203
		netif_napi_del(&vi->sq[i].napi);
2204
	}
2205

2206 2207 2208 2209 2210
	/* We called napi_hash_del() before netif_napi_del(),
	 * we need to respect an RCU grace period before freeing vi->rq
	 */
	synchronize_net();

J
Jason Wang 已提交
2211 2212 2213 2214
	kfree(vi->rq);
	kfree(vi->sq);
}

2215
static void _free_receive_bufs(struct virtnet_info *vi)
J
Jason Wang 已提交
2216
{
J
John Fastabend 已提交
2217
	struct bpf_prog *old_prog;
J
Jason Wang 已提交
2218 2219 2220 2221 2222
	int i;

	for (i = 0; i < vi->max_queue_pairs; i++) {
		while (vi->rq[i].pages)
			__free_pages(get_a_page(&vi->rq[i], GFP_KERNEL), 0);
J
John Fastabend 已提交
2223 2224 2225 2226 2227

		old_prog = rtnl_dereference(vi->rq[i].xdp_prog);
		RCU_INIT_POINTER(vi->rq[i].xdp_prog, NULL);
		if (old_prog)
			bpf_prog_put(old_prog);
J
Jason Wang 已提交
2228
	}
2229 2230 2231 2232 2233 2234
}

static void free_receive_bufs(struct virtnet_info *vi)
{
	rtnl_lock();
	_free_receive_bufs(vi);
J
John Fastabend 已提交
2235
	rtnl_unlock();
J
Jason Wang 已提交
2236 2237
}

2238 2239 2240 2241 2242 2243 2244 2245
static void free_receive_page_frags(struct virtnet_info *vi)
{
	int i;
	for (i = 0; i < vi->max_queue_pairs; i++)
		if (vi->rq[i].alloc_frag.page)
			put_page(vi->rq[i].alloc_frag.page);
}

2246
static bool is_xdp_raw_buffer_queue(struct virtnet_info *vi, int q)
J
John Fastabend 已提交
2247 2248 2249 2250 2251 2252 2253 2254 2255
{
	if (q < (vi->curr_queue_pairs - vi->xdp_queue_pairs))
		return false;
	else if (q < vi->curr_queue_pairs)
		return true;
	else
		return false;
}

J
Jason Wang 已提交
2256 2257 2258 2259 2260 2261 2262
static void free_unused_bufs(struct virtnet_info *vi)
{
	void *buf;
	int i;

	for (i = 0; i < vi->max_queue_pairs; i++) {
		struct virtqueue *vq = vi->sq[i].vq;
J
John Fastabend 已提交
2263
		while ((buf = virtqueue_detach_unused_buf(vq)) != NULL) {
2264
			if (!is_xdp_raw_buffer_queue(vi, i))
J
John Fastabend 已提交
2265 2266 2267 2268
				dev_kfree_skb(buf);
			else
				put_page(virt_to_head_page(buf));
		}
J
Jason Wang 已提交
2269 2270 2271 2272 2273 2274
	}

	for (i = 0; i < vi->max_queue_pairs; i++) {
		struct virtqueue *vq = vi->rq[i].vq;

		while ((buf = virtqueue_detach_unused_buf(vq)) != NULL) {
2275
			if (vi->mergeable_rx_bufs) {
2276
				put_page(virt_to_head_page(buf));
2277
			} else if (vi->big_packets) {
2278
				give_pages(&vi->rq[i], buf);
2279
			} else {
2280
				put_page(virt_to_head_page(buf));
2281
			}
J
Jason Wang 已提交
2282 2283 2284 2285
		}
	}
}

2286 2287 2288 2289
static void virtnet_del_vqs(struct virtnet_info *vi)
{
	struct virtio_device *vdev = vi->vdev;

2290
	virtnet_clean_affinity(vi, -1);
J
Jason Wang 已提交
2291

2292
	vdev->config->del_vqs(vdev);
J
Jason Wang 已提交
2293 2294

	virtnet_free_queues(vi);
2295 2296
}

2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308
/* How large should a single buffer be so a queue full of these can fit at
 * least one full packet?
 * Logic below assumes the mergeable buffer header is used.
 */
static unsigned int mergeable_min_buf_len(struct virtnet_info *vi, struct virtqueue *vq)
{
	const unsigned int hdr_len = sizeof(struct virtio_net_hdr_mrg_rxbuf);
	unsigned int rq_size = virtqueue_get_vring_size(vq);
	unsigned int packet_len = vi->big_packets ? IP_MAX_MTU : vi->dev->max_mtu;
	unsigned int buf_len = hdr_len + ETH_HLEN + VLAN_HLEN + packet_len;
	unsigned int min_buf_len = DIV_ROUND_UP(buf_len, rq_size);

2309 2310
	return max(max(min_buf_len, hdr_len) - hdr_len,
		   (unsigned int)GOOD_PACKET_LEN);
2311 2312
}

J
Jason Wang 已提交
2313
static int virtnet_find_vqs(struct virtnet_info *vi)
2314
{
J
Jason Wang 已提交
2315 2316 2317 2318 2319
	vq_callback_t **callbacks;
	struct virtqueue **vqs;
	int ret = -ENOMEM;
	int i, total_vqs;
	const char **names;
2320
	bool *ctx;
J
Jason Wang 已提交
2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338

	/* We expect 1 RX virtqueue followed by 1 TX virtqueue, followed by
	 * possible N-1 RX/TX queue pairs used in multiqueue mode, followed by
	 * possible control vq.
	 */
	total_vqs = vi->max_queue_pairs * 2 +
		    virtio_has_feature(vi->vdev, VIRTIO_NET_F_CTRL_VQ);

	/* Allocate space for find_vqs parameters */
	vqs = kzalloc(total_vqs * sizeof(*vqs), GFP_KERNEL);
	if (!vqs)
		goto err_vq;
	callbacks = kmalloc(total_vqs * sizeof(*callbacks), GFP_KERNEL);
	if (!callbacks)
		goto err_callback;
	names = kmalloc(total_vqs * sizeof(*names), GFP_KERNEL);
	if (!names)
		goto err_names;
2339
	if (!vi->big_packets || vi->mergeable_rx_bufs) {
2340 2341 2342 2343 2344 2345
		ctx = kzalloc(total_vqs * sizeof(*ctx), GFP_KERNEL);
		if (!ctx)
			goto err_ctx;
	} else {
		ctx = NULL;
	}
J
Jason Wang 已提交
2346 2347 2348 2349 2350 2351

	/* Parameters for control virtqueue, if any */
	if (vi->has_cvq) {
		callbacks[total_vqs - 1] = NULL;
		names[total_vqs - 1] = "control";
	}
2352

J
Jason Wang 已提交
2353 2354 2355 2356 2357 2358 2359 2360
	/* Allocate/initialize parameters for send/receive virtqueues */
	for (i = 0; i < vi->max_queue_pairs; i++) {
		callbacks[rxq2vq(i)] = skb_recv_done;
		callbacks[txq2vq(i)] = skb_xmit_done;
		sprintf(vi->rq[i].name, "input.%d", i);
		sprintf(vi->sq[i].name, "output.%d", i);
		names[rxq2vq(i)] = vi->rq[i].name;
		names[txq2vq(i)] = vi->sq[i].name;
2361 2362
		if (ctx)
			ctx[rxq2vq(i)] = true;
J
Jason Wang 已提交
2363
	}
2364

J
Jason Wang 已提交
2365
	ret = vi->vdev->config->find_vqs(vi->vdev, total_vqs, vqs, callbacks,
2366
					 names, ctx, NULL);
J
Jason Wang 已提交
2367 2368
	if (ret)
		goto err_find;
2369

J
Jason Wang 已提交
2370 2371
	if (vi->has_cvq) {
		vi->cvq = vqs[total_vqs - 1];
2372
		if (virtio_has_feature(vi->vdev, VIRTIO_NET_F_CTRL_VLAN))
2373
			vi->dev->features |= NETIF_F_HW_VLAN_CTAG_FILTER;
2374
	}
J
Jason Wang 已提交
2375 2376 2377

	for (i = 0; i < vi->max_queue_pairs; i++) {
		vi->rq[i].vq = vqs[rxq2vq(i)];
2378
		vi->rq[i].min_buf_len = mergeable_min_buf_len(vi, vi->rq[i].vq);
J
Jason Wang 已提交
2379 2380 2381 2382 2383 2384
		vi->sq[i].vq = vqs[txq2vq(i)];
	}

	kfree(names);
	kfree(callbacks);
	kfree(vqs);
2385
	kfree(ctx);
J
Jason Wang 已提交
2386

2387
	return 0;
J
Jason Wang 已提交
2388 2389

err_find:
2390 2391
	kfree(ctx);
err_ctx:
J
Jason Wang 已提交
2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408
	kfree(names);
err_names:
	kfree(callbacks);
err_callback:
	kfree(vqs);
err_vq:
	return ret;
}

static int virtnet_alloc_queues(struct virtnet_info *vi)
{
	int i;

	vi->sq = kzalloc(sizeof(*vi->sq) * vi->max_queue_pairs, GFP_KERNEL);
	if (!vi->sq)
		goto err_sq;
	vi->rq = kzalloc(sizeof(*vi->rq) * vi->max_queue_pairs, GFP_KERNEL);
2409
	if (!vi->rq)
J
Jason Wang 已提交
2410 2411 2412 2413 2414 2415 2416
		goto err_rq;

	INIT_DELAYED_WORK(&vi->refill, refill_work);
	for (i = 0; i < vi->max_queue_pairs; i++) {
		vi->rq[i].pages = NULL;
		netif_napi_add(vi->dev, &vi->rq[i].napi, virtnet_poll,
			       napi_weight);
2417 2418
		netif_tx_napi_add(vi->dev, &vi->sq[i].napi, virtnet_poll_tx,
				  napi_tx ? napi_weight : 0);
J
Jason Wang 已提交
2419 2420

		sg_init_table(vi->rq[i].sg, ARRAY_SIZE(vi->rq[i].sg));
J
Johannes Berg 已提交
2421
		ewma_pkt_len_init(&vi->rq[i].mrg_avg_pkt_len);
J
Jason Wang 已提交
2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445
		sg_init_table(vi->sq[i].sg, ARRAY_SIZE(vi->sq[i].sg));
	}

	return 0;

err_rq:
	kfree(vi->sq);
err_sq:
	return -ENOMEM;
}

static int init_vqs(struct virtnet_info *vi)
{
	int ret;

	/* Allocate send & receive queues */
	ret = virtnet_alloc_queues(vi);
	if (ret)
		goto err;

	ret = virtnet_find_vqs(vi);
	if (ret)
		goto err_free;

2446
	get_online_cpus();
2447
	virtnet_set_affinity(vi);
2448 2449
	put_online_cpus();

J
Jason Wang 已提交
2450 2451 2452 2453 2454 2455
	return 0;

err_free:
	virtnet_free_queues(vi);
err:
	return ret;
2456 2457
}

2458 2459
#ifdef CONFIG_SYSFS
static ssize_t mergeable_rx_buffer_size_show(struct netdev_rx_queue *queue,
2460
		char *buf)
2461 2462 2463
{
	struct virtnet_info *vi = netdev_priv(queue->dev);
	unsigned int queue_index = get_netdev_rx_queue_index(queue);
J
Johannes Berg 已提交
2464
	struct ewma_pkt_len *avg;
2465 2466 2467

	BUG_ON(queue_index >= vi->max_queue_pairs);
	avg = &vi->rq[queue_index].mrg_avg_pkt_len;
2468 2469
	return sprintf(buf, "%u\n",
		       get_mergeable_buf_len(&vi->rq[queue_index], avg));
2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485
}

static struct rx_queue_attribute mergeable_rx_buffer_size_attribute =
	__ATTR_RO(mergeable_rx_buffer_size);

static struct attribute *virtio_net_mrg_rx_attrs[] = {
	&mergeable_rx_buffer_size_attribute.attr,
	NULL
};

static const struct attribute_group virtio_net_mrg_rx_group = {
	.name = "virtio_net",
	.attrs = virtio_net_mrg_rx_attrs
};
#endif

2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519
static bool virtnet_fail_on_feature(struct virtio_device *vdev,
				    unsigned int fbit,
				    const char *fname, const char *dname)
{
	if (!virtio_has_feature(vdev, fbit))
		return false;

	dev_err(&vdev->dev, "device advertises feature %s but not %s",
		fname, dname);

	return true;
}

#define VIRTNET_FAIL_ON(vdev, fbit, dbit)			\
	virtnet_fail_on_feature(vdev, fbit, #fbit, dbit)

static bool virtnet_validate_features(struct virtio_device *vdev)
{
	if (!virtio_has_feature(vdev, VIRTIO_NET_F_CTRL_VQ) &&
	    (VIRTNET_FAIL_ON(vdev, VIRTIO_NET_F_CTRL_RX,
			     "VIRTIO_NET_F_CTRL_VQ") ||
	     VIRTNET_FAIL_ON(vdev, VIRTIO_NET_F_CTRL_VLAN,
			     "VIRTIO_NET_F_CTRL_VQ") ||
	     VIRTNET_FAIL_ON(vdev, VIRTIO_NET_F_GUEST_ANNOUNCE,
			     "VIRTIO_NET_F_CTRL_VQ") ||
	     VIRTNET_FAIL_ON(vdev, VIRTIO_NET_F_MQ, "VIRTIO_NET_F_CTRL_VQ") ||
	     VIRTNET_FAIL_ON(vdev, VIRTIO_NET_F_CTRL_MAC_ADDR,
			     "VIRTIO_NET_F_CTRL_VQ"))) {
		return false;
	}

	return true;
}

2520 2521 2522
#define MIN_MTU ETH_MIN_MTU
#define MAX_MTU ETH_MAX_MTU

2523
static int virtnet_validate(struct virtio_device *vdev)
R
Rusty Russell 已提交
2524
{
2525 2526 2527 2528 2529 2530
	if (!vdev->config->get) {
		dev_err(&vdev->dev, "%s failure: config access disabled\n",
			__func__);
		return -EINVAL;
	}

2531 2532 2533
	if (!virtnet_validate_features(vdev))
		return -EINVAL;

2534 2535 2536 2537 2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552
	if (virtio_has_feature(vdev, VIRTIO_NET_F_MTU)) {
		int mtu = virtio_cread16(vdev,
					 offsetof(struct virtio_net_config,
						  mtu));
		if (mtu < MIN_MTU)
			__virtio_clear_bit(vdev, VIRTIO_NET_F_MTU);
	}

	return 0;
}

static int virtnet_probe(struct virtio_device *vdev)
{
	int i, err;
	struct net_device *dev;
	struct virtnet_info *vi;
	u16 max_queue_pairs;
	int mtu;

J
Jason Wang 已提交
2553
	/* Find if host supports multiqueue virtio_net device */
2554 2555 2556
	err = virtio_cread_feature(vdev, VIRTIO_NET_F_MQ,
				   struct virtio_net_config,
				   max_virtqueue_pairs, &max_queue_pairs);
J
Jason Wang 已提交
2557 2558 2559 2560 2561 2562

	/* We need at least 2 queue's */
	if (err || max_queue_pairs < VIRTIO_NET_CTRL_MQ_VQ_PAIRS_MIN ||
	    max_queue_pairs > VIRTIO_NET_CTRL_MQ_VQ_PAIRS_MAX ||
	    !virtio_has_feature(vdev, VIRTIO_NET_F_CTRL_VQ))
		max_queue_pairs = 1;
R
Rusty Russell 已提交
2563 2564

	/* Allocate ourselves a network device with room for our info */
J
Jason Wang 已提交
2565
	dev = alloc_etherdev_mq(sizeof(struct virtnet_info), max_queue_pairs);
R
Rusty Russell 已提交
2566 2567 2568 2569
	if (!dev)
		return -ENOMEM;

	/* Set up network device as normal. */
2570
	dev->priv_flags |= IFF_UNICAST_FLT | IFF_LIVE_ADDR_CHANGE;
2571
	dev->netdev_ops = &virtnet_netdev;
R
Rusty Russell 已提交
2572
	dev->features = NETIF_F_HIGHDMA;
2573

2574
	dev->ethtool_ops = &virtnet_ethtool_ops;
R
Rusty Russell 已提交
2575 2576 2577
	SET_NETDEV_DEV(dev, &vdev->dev);

	/* Do we support "hardware" checksums? */
2578
	if (virtio_has_feature(vdev, VIRTIO_NET_F_CSUM)) {
R
Rusty Russell 已提交
2579
		/* This opens up the world of extra features. */
J
Jason Wang 已提交
2580
		dev->hw_features |= NETIF_F_HW_CSUM | NETIF_F_SG;
2581
		if (csum)
J
Jason Wang 已提交
2582
			dev->features |= NETIF_F_HW_CSUM | NETIF_F_SG;
2583 2584

		if (virtio_has_feature(vdev, VIRTIO_NET_F_GSO)) {
2585
			dev->hw_features |= NETIF_F_TSO
R
Rusty Russell 已提交
2586 2587
				| NETIF_F_TSO_ECN | NETIF_F_TSO6;
		}
2588
		/* Individual feature bits: what can host handle? */
2589 2590 2591 2592 2593 2594 2595
		if (virtio_has_feature(vdev, VIRTIO_NET_F_HOST_TSO4))
			dev->hw_features |= NETIF_F_TSO;
		if (virtio_has_feature(vdev, VIRTIO_NET_F_HOST_TSO6))
			dev->hw_features |= NETIF_F_TSO6;
		if (virtio_has_feature(vdev, VIRTIO_NET_F_HOST_ECN))
			dev->hw_features |= NETIF_F_TSO_ECN;

2596 2597
		dev->features |= NETIF_F_GSO_ROBUST;

2598
		if (gso)
2599
			dev->features |= dev->hw_features & NETIF_F_ALL_TSO;
2600
		/* (!csum && gso) case will be fixed by register_netdev() */
R
Rusty Russell 已提交
2601
	}
2602 2603
	if (virtio_has_feature(vdev, VIRTIO_NET_F_GUEST_CSUM))
		dev->features |= NETIF_F_RXCSUM;
R
Rusty Russell 已提交
2604

2605 2606
	dev->vlan_features = dev->features;

2607 2608 2609 2610
	/* MTU range: 68 - 65535 */
	dev->min_mtu = MIN_MTU;
	dev->max_mtu = MAX_MTU;

R
Rusty Russell 已提交
2611
	/* Configuration may specify what MAC to use.  Otherwise random. */
2612 2613 2614 2615 2616
	if (virtio_has_feature(vdev, VIRTIO_NET_F_MAC))
		virtio_cread_bytes(vdev,
				   offsetof(struct virtio_net_config, mac),
				   dev->dev_addr, dev->addr_len);
	else
2617
		eth_hw_addr_random(dev);
R
Rusty Russell 已提交
2618 2619 2620 2621 2622

	/* Set up our device-specific information */
	vi = netdev_priv(dev);
	vi->dev = dev;
	vi->vdev = vdev;
2623
	vdev->priv = vi;
2624 2625 2626 2627 2628
	vi->stats = alloc_percpu(struct virtnet_stats);
	err = -ENOMEM;
	if (vi->stats == NULL)
		goto free;

2629 2630 2631 2632 2633 2634 2635
	for_each_possible_cpu(i) {
		struct virtnet_stats *virtnet_stats;
		virtnet_stats = per_cpu_ptr(vi->stats, i);
		u64_stats_init(&virtnet_stats->tx_syncp);
		u64_stats_init(&virtnet_stats->rx_syncp);
	}

2636
	INIT_WORK(&vi->config_work, virtnet_config_changed_work);
R
Rusty Russell 已提交
2637

2638
	/* If we can receive ANY GSO packets, we must allocate large ones. */
2639 2640
	if (virtio_has_feature(vdev, VIRTIO_NET_F_GUEST_TSO4) ||
	    virtio_has_feature(vdev, VIRTIO_NET_F_GUEST_TSO6) ||
2641 2642
	    virtio_has_feature(vdev, VIRTIO_NET_F_GUEST_ECN) ||
	    virtio_has_feature(vdev, VIRTIO_NET_F_GUEST_UFO))
2643 2644
		vi->big_packets = true;

2645 2646 2647
	if (virtio_has_feature(vdev, VIRTIO_NET_F_MRG_RXBUF))
		vi->mergeable_rx_bufs = true;

2648 2649
	if (virtio_has_feature(vdev, VIRTIO_NET_F_MRG_RXBUF) ||
	    virtio_has_feature(vdev, VIRTIO_F_VERSION_1))
2650 2651 2652 2653
		vi->hdr_len = sizeof(struct virtio_net_hdr_mrg_rxbuf);
	else
		vi->hdr_len = sizeof(struct virtio_net_hdr);

2654 2655
	if (virtio_has_feature(vdev, VIRTIO_F_ANY_LAYOUT) ||
	    virtio_has_feature(vdev, VIRTIO_F_VERSION_1))
2656 2657
		vi->any_header_sg = true;

J
Jason Wang 已提交
2658 2659 2660
	if (virtio_has_feature(vdev, VIRTIO_NET_F_CTRL_VQ))
		vi->has_cvq = true;

2661 2662 2663 2664
	if (virtio_has_feature(vdev, VIRTIO_NET_F_MTU)) {
		mtu = virtio_cread16(vdev,
				     offsetof(struct virtio_net_config,
					      mtu));
2665
		if (mtu < dev->min_mtu) {
2666 2667 2668 2669 2670 2671
			/* Should never trigger: MTU was previously validated
			 * in virtnet_validate.
			 */
			dev_err(&vdev->dev, "device MTU appears to have changed "
				"it is now %d < %d", mtu, dev->min_mtu);
			goto free_stats;
2672
		}
2673

2674 2675 2676
		dev->mtu = mtu;
		dev->max_mtu = mtu;

2677 2678 2679
		/* TODO: size buffers correctly in this case. */
		if (dev->mtu > ETH_DATA_LEN)
			vi->big_packets = true;
2680 2681
	}

2682 2683
	if (vi->any_header_sg)
		dev->needed_headroom = vi->hdr_len;
2684

2685 2686 2687 2688 2689
	/* Enable multiqueue by default */
	if (num_online_cpus() >= max_queue_pairs)
		vi->curr_queue_pairs = max_queue_pairs;
	else
		vi->curr_queue_pairs = num_online_cpus();
J
Jason Wang 已提交
2690 2691 2692
	vi->max_queue_pairs = max_queue_pairs;

	/* Allocate/initialize the rx/tx queues, and invoke find_vqs */
2693
	err = init_vqs(vi);
2694
	if (err)
2695
		goto free_stats;
R
Rusty Russell 已提交
2696

2697 2698 2699 2700
#ifdef CONFIG_SYSFS
	if (vi->mergeable_rx_bufs)
		dev->sysfs_rx_queue_group = &virtio_net_mrg_rx_group;
#endif
2701 2702
	netif_set_real_num_tx_queues(dev, vi->curr_queue_pairs);
	netif_set_real_num_rx_queues(dev, vi->curr_queue_pairs);
J
Jason Wang 已提交
2703

2704 2705
	virtnet_init_settings(dev);

R
Rusty Russell 已提交
2706 2707 2708
	err = register_netdev(dev);
	if (err) {
		pr_debug("virtio_net: registering device failed\n");
2709
		goto free_vqs;
R
Rusty Russell 已提交
2710
	}
2711

M
Michael S. Tsirkin 已提交
2712 2713
	virtio_device_ready(vdev);

2714
	err = virtnet_cpu_notif_add(vi);
2715 2716
	if (err) {
		pr_debug("virtio_net: registering cpu notifier failed\n");
2717
		goto free_unregister_netdev;
2718 2719
	}

2720
	virtnet_set_queues(vi, vi->curr_queue_pairs);
2721

J
Jason Wang 已提交
2722 2723 2724 2725
	/* Assume link up if device can't report link status,
	   otherwise get link status from config. */
	if (virtio_has_feature(vi->vdev, VIRTIO_NET_F_STATUS)) {
		netif_carrier_off(dev);
2726
		schedule_work(&vi->config_work);
J
Jason Wang 已提交
2727 2728
	} else {
		vi->status = VIRTIO_NET_S_LINK_UP;
2729
		virtnet_update_settings(vi);
J
Jason Wang 已提交
2730 2731
		netif_carrier_on(dev);
	}
2732

2733 2734 2735 2736
	for (i = 0; i < ARRAY_SIZE(guest_offloads); i++)
		if (virtio_has_feature(vi->vdev, guest_offloads[i]))
			set_bit(guest_offloads[i], &vi->guest_offloads);

J
Jason Wang 已提交
2737 2738 2739
	pr_debug("virtnet: registered device %s with %d RX and TX vq's\n",
		 dev->name, max_queue_pairs);

R
Rusty Russell 已提交
2740 2741
	return 0;

2742
free_unregister_netdev:
2743 2744
	vi->vdev->config->reset(vdev);

2745
	unregister_netdev(dev);
2746
free_vqs:
J
Jason Wang 已提交
2747
	cancel_delayed_work_sync(&vi->refill);
2748
	free_receive_page_frags(vi);
2749
	virtnet_del_vqs(vi);
2750 2751
free_stats:
	free_percpu(vi->stats);
R
Rusty Russell 已提交
2752 2753 2754 2755 2756
free:
	free_netdev(dev);
	return err;
}

2757
static void remove_vq_common(struct virtnet_info *vi)
R
Rusty Russell 已提交
2758
{
2759
	vi->vdev->config->reset(vi->vdev);
S
Shirley Ma 已提交
2760 2761

	/* Free unused buffers in both send and recv, if any. */
2762
	free_unused_bufs(vi);
2763

J
Jason Wang 已提交
2764
	free_receive_bufs(vi);
2765

2766 2767
	free_receive_page_frags(vi);

J
Jason Wang 已提交
2768
	virtnet_del_vqs(vi);
2769 2770
}

2771
static void virtnet_remove(struct virtio_device *vdev)
2772 2773 2774
{
	struct virtnet_info *vi = vdev->priv;

2775
	virtnet_cpu_notif_remove(vi);
2776

2777 2778
	/* Make sure no work handler is accessing the device. */
	flush_work(&vi->config_work);
2779

2780 2781 2782
	unregister_netdev(vi->dev);

	remove_vq_common(vi);
2783

2784
	free_percpu(vi->stats);
2785
	free_netdev(vi->dev);
R
Rusty Russell 已提交
2786 2787
}

2788
static __maybe_unused int virtnet_freeze(struct virtio_device *vdev)
2789 2790 2791
{
	struct virtnet_info *vi = vdev->priv;

2792
	virtnet_cpu_notif_remove(vi);
2793
	virtnet_freeze_down(vdev);
2794 2795 2796 2797 2798
	remove_vq_common(vi);

	return 0;
}

2799
static __maybe_unused int virtnet_restore(struct virtio_device *vdev)
2800 2801
{
	struct virtnet_info *vi = vdev->priv;
2802
	int err;
2803

2804
	err = virtnet_restore_up(vdev);
2805 2806
	if (err)
		return err;
J
Jason Wang 已提交
2807 2808
	virtnet_set_queues(vi, vi->curr_queue_pairs);

2809
	err = virtnet_cpu_notif_add(vi);
2810 2811 2812
	if (err)
		return err;

2813 2814 2815
	return 0;
}

R
Rusty Russell 已提交
2816 2817 2818 2819 2820
static struct virtio_device_id id_table[] = {
	{ VIRTIO_ID_NET, VIRTIO_DEV_ANY_ID },
	{ 0 },
};

2821 2822 2823 2824 2825 2826 2827 2828 2829 2830
#define VIRTNET_FEATURES \
	VIRTIO_NET_F_CSUM, VIRTIO_NET_F_GUEST_CSUM, \
	VIRTIO_NET_F_MAC, \
	VIRTIO_NET_F_HOST_TSO4, VIRTIO_NET_F_HOST_UFO, VIRTIO_NET_F_HOST_TSO6, \
	VIRTIO_NET_F_HOST_ECN, VIRTIO_NET_F_GUEST_TSO4, VIRTIO_NET_F_GUEST_TSO6, \
	VIRTIO_NET_F_GUEST_ECN, VIRTIO_NET_F_GUEST_UFO, \
	VIRTIO_NET_F_MRG_RXBUF, VIRTIO_NET_F_STATUS, VIRTIO_NET_F_CTRL_VQ, \
	VIRTIO_NET_F_CTRL_RX, VIRTIO_NET_F_CTRL_VLAN, \
	VIRTIO_NET_F_GUEST_ANNOUNCE, VIRTIO_NET_F_MQ, \
	VIRTIO_NET_F_CTRL_MAC_ADDR, \
2831 2832
	VIRTIO_NET_F_MTU, VIRTIO_NET_F_CTRL_GUEST_OFFLOADS, \
	VIRTIO_NET_F_SPEED_DUPLEX
2833

2834
static unsigned int features[] = {
2835 2836 2837 2838 2839 2840
	VIRTNET_FEATURES,
};

static unsigned int features_legacy[] = {
	VIRTNET_FEATURES,
	VIRTIO_NET_F_GSO,
2841
	VIRTIO_F_ANY_LAYOUT,
2842 2843
};

2844
static struct virtio_driver virtio_net_driver = {
2845 2846
	.feature_table = features,
	.feature_table_size = ARRAY_SIZE(features),
2847 2848
	.feature_table_legacy = features_legacy,
	.feature_table_size_legacy = ARRAY_SIZE(features_legacy),
R
Rusty Russell 已提交
2849 2850 2851
	.driver.name =	KBUILD_MODNAME,
	.driver.owner =	THIS_MODULE,
	.id_table =	id_table,
2852
	.validate =	virtnet_validate,
R
Rusty Russell 已提交
2853
	.probe =	virtnet_probe,
2854
	.remove =	virtnet_remove,
2855
	.config_changed = virtnet_config_changed,
2856
#ifdef CONFIG_PM_SLEEP
2857 2858 2859
	.freeze =	virtnet_freeze,
	.restore =	virtnet_restore,
#endif
R
Rusty Russell 已提交
2860 2861
};

2862 2863 2864 2865
static __init int virtio_net_driver_init(void)
{
	int ret;

T
Thomas Gleixner 已提交
2866
	ret = cpuhp_setup_state_multi(CPUHP_AP_ONLINE_DYN, "virtio/net:online",
2867 2868 2869 2870 2871
				      virtnet_cpu_online,
				      virtnet_cpu_down_prep);
	if (ret < 0)
		goto out;
	virtionet_online = ret;
T
Thomas Gleixner 已提交
2872
	ret = cpuhp_setup_state_multi(CPUHP_VIRT_NET_DEAD, "virtio/net:dead",
2873 2874 2875 2876 2877 2878 2879 2880 2881 2882 2883 2884 2885 2886 2887 2888 2889 2890 2891
				      NULL, virtnet_cpu_dead);
	if (ret)
		goto err_dead;

        ret = register_virtio_driver(&virtio_net_driver);
	if (ret)
		goto err_virtio;
	return 0;
err_virtio:
	cpuhp_remove_multi_state(CPUHP_VIRT_NET_DEAD);
err_dead:
	cpuhp_remove_multi_state(virtionet_online);
out:
	return ret;
}
module_init(virtio_net_driver_init);

static __exit void virtio_net_driver_exit(void)
{
A
Andrew Jones 已提交
2892
	unregister_virtio_driver(&virtio_net_driver);
2893 2894 2895 2896
	cpuhp_remove_multi_state(CPUHP_VIRT_NET_DEAD);
	cpuhp_remove_multi_state(virtionet_online);
}
module_exit(virtio_net_driver_exit);
R
Rusty Russell 已提交
2897 2898 2899 2900

MODULE_DEVICE_TABLE(virtio, id_table);
MODULE_DESCRIPTION("Virtio network driver");
MODULE_LICENSE("GPL");