]>
Commit | Line | Data |
---|---|---|
48925e37 | 1 | /* A network driver using virtio. |
296f96fc RR |
2 | * |
3 | * Copyright 2007 Rusty Russell <rusty@rustcorp.com.au> IBM Corporation | |
4 | * | |
5 | * This program is free software; you can redistribute it and/or modify | |
6 | * it under the terms of the GNU General Public License as published by | |
7 | * the Free Software Foundation; either version 2 of the License, or | |
8 | * (at your option) any later version. | |
9 | * | |
10 | * This program is distributed in the hope that it will be useful, | |
11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
13 | * GNU General Public License for more details. | |
14 | * | |
15 | * You should have received a copy of the GNU General Public License | |
adf8d3ff | 16 | * along with this program; if not, see <http://www.gnu.org/licenses/>. |
296f96fc RR |
17 | */ |
18 | //#define DEBUG | |
19 | #include <linux/netdevice.h> | |
20 | #include <linux/etherdevice.h> | |
a9ea3fc6 | 21 | #include <linux/ethtool.h> |
296f96fc RR |
22 | #include <linux/module.h> |
23 | #include <linux/virtio.h> | |
24 | #include <linux/virtio_net.h> | |
f600b690 | 25 | #include <linux/bpf.h> |
296f96fc | 26 | #include <linux/scatterlist.h> |
e918085a | 27 | #include <linux/if_vlan.h> |
5a0e3ad6 | 28 | #include <linux/slab.h> |
8de4b2f3 | 29 | #include <linux/cpu.h> |
ab7db917 | 30 | #include <linux/average.h> |
91815639 | 31 | #include <net/busy_poll.h> |
296f96fc | 32 | |
d34710e3 | 33 | static int napi_weight = NAPI_POLL_WEIGHT; |
6c0cd7c0 DL |
34 | module_param(napi_weight, int, 0444); |
35 | ||
eb939922 | 36 | static bool csum = true, gso = true; |
34a48579 RR |
37 | module_param(csum, bool, 0444); |
38 | module_param(gso, bool, 0444); | |
39 | ||
296f96fc | 40 | /* FIXME: MTU in config. */ |
5061de36 | 41 | #define GOOD_PACKET_LEN (ETH_HLEN + VLAN_HLEN + ETH_DATA_LEN) |
3f2c31d9 | 42 | #define GOOD_COPY_LEN 128 |
296f96fc | 43 | |
5377d758 JB |
44 | /* RX packet size EWMA. The average packet size is used to determine the packet |
45 | * buffer size when refilling RX rings. As the entire RX ring may be refilled | |
46 | * at once, the weight is chosen so that the EWMA will be insensitive to short- | |
47 | * term, transient changes in packet size. | |
ab7db917 | 48 | */ |
5377d758 | 49 | DECLARE_EWMA(pkt_len, 1, 64) |
ab7db917 MD |
50 | |
51 | /* Minimum alignment for mergeable packet buffers. */ | |
52 | #define MERGEABLE_BUFFER_ALIGN max(L1_CACHE_BYTES, 256) | |
53 | ||
66846048 | 54 | #define VIRTNET_DRIVER_VERSION "1.0.0" |
2a41f71d | 55 | |
3fa2a1df | 56 | struct virtnet_stats { |
83a27052 ED |
57 | struct u64_stats_sync tx_syncp; |
58 | struct u64_stats_sync rx_syncp; | |
3fa2a1df | 59 | u64 tx_bytes; |
60 | u64 tx_packets; | |
61 | ||
62 | u64 rx_bytes; | |
63 | u64 rx_packets; | |
64 | }; | |
65 | ||
e9d7417b JW |
66 | /* Internal representation of a send virtqueue */ |
67 | struct send_queue { | |
68 | /* Virtqueue associated with this send _queue */ | |
69 | struct virtqueue *vq; | |
70 | ||
71 | /* TX: fragments + linear part + virtio header */ | |
72 | struct scatterlist sg[MAX_SKB_FRAGS + 2]; | |
986a4f4d JW |
73 | |
74 | /* Name of the send queue: output.$index */ | |
75 | char name[40]; | |
e9d7417b JW |
76 | }; |
77 | ||
78 | /* Internal representation of a receive virtqueue */ | |
79 | struct receive_queue { | |
80 | /* Virtqueue associated with this receive_queue */ | |
81 | struct virtqueue *vq; | |
82 | ||
296f96fc RR |
83 | struct napi_struct napi; |
84 | ||
f600b690 JF |
85 | struct bpf_prog __rcu *xdp_prog; |
86 | ||
e9d7417b JW |
87 | /* Chain pages by the private ptr. */ |
88 | struct page *pages; | |
89 | ||
ab7db917 | 90 | /* Average packet length for mergeable receive buffers. */ |
5377d758 | 91 | struct ewma_pkt_len mrg_avg_pkt_len; |
ab7db917 | 92 | |
fb51879d MD |
93 | /* Page frag for packet buffer allocation. */ |
94 | struct page_frag alloc_frag; | |
95 | ||
e9d7417b JW |
96 | /* RX: fragments + linear part + virtio header */ |
97 | struct scatterlist sg[MAX_SKB_FRAGS + 2]; | |
986a4f4d JW |
98 | |
99 | /* Name of this receive queue: input.$index */ | |
100 | char name[40]; | |
e9d7417b JW |
101 | }; |
102 | ||
103 | struct virtnet_info { | |
104 | struct virtio_device *vdev; | |
105 | struct virtqueue *cvq; | |
106 | struct net_device *dev; | |
986a4f4d JW |
107 | struct send_queue *sq; |
108 | struct receive_queue *rq; | |
e9d7417b JW |
109 | unsigned int status; |
110 | ||
986a4f4d JW |
111 | /* Max # of queue pairs supported by the device */ |
112 | u16 max_queue_pairs; | |
113 | ||
114 | /* # of queue pairs currently used by the driver */ | |
115 | u16 curr_queue_pairs; | |
116 | ||
672aafd5 JF |
117 | /* # of XDP queue pairs currently used by the driver */ |
118 | u16 xdp_queue_pairs; | |
119 | ||
97402b96 HX |
120 | /* I like... big packets and I cannot lie! */ |
121 | bool big_packets; | |
122 | ||
3f2c31d9 MM |
123 | /* Host will merge rx buffers for big packets (shake it! shake it!) */ |
124 | bool mergeable_rx_bufs; | |
125 | ||
986a4f4d JW |
126 | /* Has control virtqueue */ |
127 | bool has_cvq; | |
128 | ||
e7428e95 MT |
129 | /* Host can handle any s/g split between our header and packet data */ |
130 | bool any_header_sg; | |
131 | ||
012873d0 MT |
132 | /* Packet virtio header size */ |
133 | u8 hdr_len; | |
134 | ||
3fa2a1df | 135 | /* Active statistics */ |
136 | struct virtnet_stats __percpu *stats; | |
137 | ||
3161e453 RR |
138 | /* Work struct for refilling if we run low on memory. */ |
139 | struct delayed_work refill; | |
140 | ||
586d17c5 JW |
141 | /* Work struct for config space updates */ |
142 | struct work_struct config_work; | |
143 | ||
986a4f4d JW |
144 | /* Does the affinity hint is set for virtqueues? */ |
145 | bool affinity_hint_set; | |
47be2479 | 146 | |
8017c279 SAS |
147 | /* CPU hotplug instances for online & dead */ |
148 | struct hlist_node node; | |
149 | struct hlist_node node_dead; | |
2ac46030 MT |
150 | |
151 | /* Control VQ buffers: protected by the rtnl lock */ | |
152 | struct virtio_net_ctrl_hdr ctrl_hdr; | |
153 | virtio_net_ctrl_ack ctrl_status; | |
a725ee3e | 154 | struct virtio_net_ctrl_mq ctrl_mq; |
2ac46030 MT |
155 | u8 ctrl_promisc; |
156 | u8 ctrl_allmulti; | |
a725ee3e | 157 | u16 ctrl_vid; |
16032be5 NA |
158 | |
159 | /* Ethtool settings */ | |
160 | u8 duplex; | |
161 | u32 speed; | |
296f96fc RR |
162 | }; |
163 | ||
9ab86bbc | 164 | struct padded_vnet_hdr { |
012873d0 | 165 | struct virtio_net_hdr_mrg_rxbuf hdr; |
9ab86bbc | 166 | /* |
012873d0 MT |
167 | * hdr is in a separate sg buffer, and data sg buffer shares same page |
168 | * with this header sg. This padding makes next sg 16 byte aligned | |
169 | * after the header. | |
9ab86bbc | 170 | */ |
012873d0 | 171 | char padding[4]; |
9ab86bbc SM |
172 | }; |
173 | ||
986a4f4d JW |
174 | /* Converting between virtqueue no. and kernel tx/rx queue no. |
175 | * 0:rx0 1:tx0 2:rx1 3:tx1 ... 2N:rxN 2N+1:txN 2N+2:cvq | |
176 | */ | |
177 | static int vq2txq(struct virtqueue *vq) | |
178 | { | |
9d0ca6ed | 179 | return (vq->index - 1) / 2; |
986a4f4d JW |
180 | } |
181 | ||
182 | static int txq2vq(int txq) | |
183 | { | |
184 | return txq * 2 + 1; | |
185 | } | |
186 | ||
187 | static int vq2rxq(struct virtqueue *vq) | |
188 | { | |
9d0ca6ed | 189 | return vq->index / 2; |
986a4f4d JW |
190 | } |
191 | ||
192 | static int rxq2vq(int rxq) | |
193 | { | |
194 | return rxq * 2; | |
195 | } | |
196 | ||
012873d0 | 197 | static inline struct virtio_net_hdr_mrg_rxbuf *skb_vnet_hdr(struct sk_buff *skb) |
296f96fc | 198 | { |
012873d0 | 199 | return (struct virtio_net_hdr_mrg_rxbuf *)skb->cb; |
296f96fc RR |
200 | } |
201 | ||
9ab86bbc SM |
202 | /* |
203 | * private is used to chain pages for big packets, put the whole | |
204 | * most recent used list in the beginning for reuse | |
205 | */ | |
e9d7417b | 206 | static void give_pages(struct receive_queue *rq, struct page *page) |
0a888fd1 | 207 | { |
9ab86bbc | 208 | struct page *end; |
0a888fd1 | 209 | |
e9d7417b | 210 | /* Find end of list, sew whole thing into vi->rq.pages. */ |
9ab86bbc | 211 | for (end = page; end->private; end = (struct page *)end->private); |
e9d7417b JW |
212 | end->private = (unsigned long)rq->pages; |
213 | rq->pages = page; | |
0a888fd1 MM |
214 | } |
215 | ||
e9d7417b | 216 | static struct page *get_a_page(struct receive_queue *rq, gfp_t gfp_mask) |
fb6813f4 | 217 | { |
e9d7417b | 218 | struct page *p = rq->pages; |
fb6813f4 | 219 | |
9ab86bbc | 220 | if (p) { |
e9d7417b | 221 | rq->pages = (struct page *)p->private; |
9ab86bbc SM |
222 | /* clear private here, it is used to chain pages */ |
223 | p->private = 0; | |
224 | } else | |
fb6813f4 RR |
225 | p = alloc_page(gfp_mask); |
226 | return p; | |
227 | } | |
228 | ||
e9d7417b | 229 | static void skb_xmit_done(struct virtqueue *vq) |
296f96fc | 230 | { |
e9d7417b | 231 | struct virtnet_info *vi = vq->vdev->priv; |
296f96fc | 232 | |
2cb9c6ba | 233 | /* Suppress further interrupts. */ |
e9d7417b | 234 | virtqueue_disable_cb(vq); |
11a3a154 | 235 | |
363f1514 | 236 | /* We were probably waiting for more output buffers. */ |
986a4f4d | 237 | netif_wake_subqueue(vi->dev, vq2txq(vq)); |
296f96fc RR |
238 | } |
239 | ||
ab7db917 MD |
240 | static unsigned int mergeable_ctx_to_buf_truesize(unsigned long mrg_ctx) |
241 | { | |
242 | unsigned int truesize = mrg_ctx & (MERGEABLE_BUFFER_ALIGN - 1); | |
243 | return (truesize + 1) * MERGEABLE_BUFFER_ALIGN; | |
244 | } | |
245 | ||
246 | static void *mergeable_ctx_to_buf_address(unsigned long mrg_ctx) | |
247 | { | |
248 | return (void *)(mrg_ctx & -MERGEABLE_BUFFER_ALIGN); | |
249 | ||
250 | } | |
251 | ||
252 | static unsigned long mergeable_buf_to_ctx(void *buf, unsigned int truesize) | |
253 | { | |
254 | unsigned int size = truesize / MERGEABLE_BUFFER_ALIGN; | |
255 | return (unsigned long)buf | (size - 1); | |
256 | } | |
257 | ||
3464645a | 258 | /* Called from bottom half context */ |
946fa564 MT |
259 | static struct sk_buff *page_to_skb(struct virtnet_info *vi, |
260 | struct receive_queue *rq, | |
2613af0e MD |
261 | struct page *page, unsigned int offset, |
262 | unsigned int len, unsigned int truesize) | |
9ab86bbc SM |
263 | { |
264 | struct sk_buff *skb; | |
012873d0 | 265 | struct virtio_net_hdr_mrg_rxbuf *hdr; |
2613af0e | 266 | unsigned int copy, hdr_len, hdr_padded_len; |
9ab86bbc | 267 | char *p; |
fb6813f4 | 268 | |
2613af0e | 269 | p = page_address(page) + offset; |
3f2c31d9 | 270 | |
9ab86bbc | 271 | /* copy small packet so we can reuse these pages for small data */ |
c67f5db8 | 272 | skb = napi_alloc_skb(&rq->napi, GOOD_COPY_LEN); |
9ab86bbc SM |
273 | if (unlikely(!skb)) |
274 | return NULL; | |
3f2c31d9 | 275 | |
9ab86bbc | 276 | hdr = skb_vnet_hdr(skb); |
3f2c31d9 | 277 | |
012873d0 MT |
278 | hdr_len = vi->hdr_len; |
279 | if (vi->mergeable_rx_bufs) | |
280 | hdr_padded_len = sizeof *hdr; | |
281 | else | |
2613af0e | 282 | hdr_padded_len = sizeof(struct padded_vnet_hdr); |
3f2c31d9 | 283 | |
9ab86bbc | 284 | memcpy(hdr, p, hdr_len); |
3f2c31d9 | 285 | |
9ab86bbc | 286 | len -= hdr_len; |
2613af0e MD |
287 | offset += hdr_padded_len; |
288 | p += hdr_padded_len; | |
3f2c31d9 | 289 | |
9ab86bbc SM |
290 | copy = len; |
291 | if (copy > skb_tailroom(skb)) | |
292 | copy = skb_tailroom(skb); | |
293 | memcpy(skb_put(skb, copy), p, copy); | |
3f2c31d9 | 294 | |
9ab86bbc SM |
295 | len -= copy; |
296 | offset += copy; | |
3f2c31d9 | 297 | |
2613af0e MD |
298 | if (vi->mergeable_rx_bufs) { |
299 | if (len) | |
300 | skb_add_rx_frag(skb, 0, page, offset, len, truesize); | |
301 | else | |
302 | put_page(page); | |
303 | return skb; | |
304 | } | |
305 | ||
e878d78b SL |
306 | /* |
307 | * Verify that we can indeed put this data into a skb. | |
308 | * This is here to handle cases when the device erroneously | |
309 | * tries to receive more than is possible. This is usually | |
310 | * the case of a broken device. | |
311 | */ | |
312 | if (unlikely(len > MAX_SKB_FRAGS * PAGE_SIZE)) { | |
be443899 | 313 | net_dbg_ratelimited("%s: too much data\n", skb->dev->name); |
e878d78b SL |
314 | dev_kfree_skb(skb); |
315 | return NULL; | |
316 | } | |
2613af0e | 317 | BUG_ON(offset >= PAGE_SIZE); |
9ab86bbc | 318 | while (len) { |
2613af0e MD |
319 | unsigned int frag_size = min((unsigned)PAGE_SIZE - offset, len); |
320 | skb_add_rx_frag(skb, skb_shinfo(skb)->nr_frags, page, offset, | |
321 | frag_size, truesize); | |
322 | len -= frag_size; | |
9ab86bbc SM |
323 | page = (struct page *)page->private; |
324 | offset = 0; | |
325 | } | |
3f2c31d9 | 326 | |
9ab86bbc | 327 | if (page) |
e9d7417b | 328 | give_pages(rq, page); |
3f2c31d9 | 329 | |
9ab86bbc SM |
330 | return skb; |
331 | } | |
3f2c31d9 | 332 | |
56434a01 JF |
333 | static void virtnet_xdp_xmit(struct virtnet_info *vi, |
334 | struct receive_queue *rq, | |
335 | struct send_queue *sq, | |
336 | struct xdp_buff *xdp) | |
337 | { | |
338 | struct page *page = virt_to_head_page(xdp->data); | |
339 | struct virtio_net_hdr_mrg_rxbuf *hdr; | |
340 | unsigned int num_sg, len; | |
341 | void *xdp_sent; | |
342 | int err; | |
343 | ||
344 | /* Free up any pending old buffers before queueing new ones. */ | |
345 | while ((xdp_sent = virtqueue_get_buf(sq->vq, &len)) != NULL) { | |
346 | struct page *sent_page = virt_to_head_page(xdp_sent); | |
347 | ||
348 | if (vi->mergeable_rx_bufs) | |
349 | put_page(sent_page); | |
350 | else | |
351 | give_pages(rq, sent_page); | |
352 | } | |
353 | ||
354 | /* Zero header and leave csum up to XDP layers */ | |
355 | hdr = xdp->data; | |
356 | memset(hdr, 0, vi->hdr_len); | |
357 | ||
358 | num_sg = 1; | |
359 | sg_init_one(sq->sg, xdp->data, xdp->data_end - xdp->data); | |
360 | err = virtqueue_add_outbuf(sq->vq, sq->sg, num_sg, | |
361 | xdp->data, GFP_ATOMIC); | |
362 | if (unlikely(err)) { | |
363 | if (vi->mergeable_rx_bufs) | |
364 | put_page(page); | |
365 | else | |
366 | give_pages(rq, page); | |
367 | return; // On error abort to avoid unnecessary kick | |
368 | } else if (!vi->mergeable_rx_bufs) { | |
369 | /* If not mergeable bufs must be big packets so cleanup pages */ | |
370 | give_pages(rq, (struct page *)page->private); | |
371 | page->private = 0; | |
372 | } | |
373 | ||
374 | virtqueue_kick(sq->vq); | |
375 | } | |
376 | ||
f600b690 | 377 | static u32 do_xdp_prog(struct virtnet_info *vi, |
56434a01 | 378 | struct receive_queue *rq, |
f600b690 JF |
379 | struct bpf_prog *xdp_prog, |
380 | struct page *page, int offset, int len) | |
381 | { | |
382 | int hdr_padded_len; | |
383 | struct xdp_buff xdp; | |
56434a01 | 384 | unsigned int qp; |
f600b690 JF |
385 | u32 act; |
386 | u8 *buf; | |
387 | ||
388 | buf = page_address(page) + offset; | |
389 | ||
390 | if (vi->mergeable_rx_bufs) | |
391 | hdr_padded_len = sizeof(struct virtio_net_hdr_mrg_rxbuf); | |
392 | else | |
393 | hdr_padded_len = sizeof(struct padded_vnet_hdr); | |
394 | ||
395 | xdp.data = buf + hdr_padded_len; | |
396 | xdp.data_end = xdp.data + (len - vi->hdr_len); | |
397 | ||
398 | act = bpf_prog_run_xdp(xdp_prog, &xdp); | |
399 | switch (act) { | |
400 | case XDP_PASS: | |
401 | return XDP_PASS; | |
56434a01 JF |
402 | case XDP_TX: |
403 | qp = vi->curr_queue_pairs - | |
404 | vi->xdp_queue_pairs + | |
405 | smp_processor_id(); | |
406 | xdp.data = buf + (vi->mergeable_rx_bufs ? 0 : 4); | |
407 | virtnet_xdp_xmit(vi, rq, &vi->sq[qp], &xdp); | |
408 | return XDP_TX; | |
f600b690 JF |
409 | default: |
410 | bpf_warn_invalid_xdp_action(act); | |
f600b690 JF |
411 | case XDP_ABORTED: |
412 | case XDP_DROP: | |
413 | return XDP_DROP; | |
414 | } | |
415 | } | |
416 | ||
012873d0 | 417 | static struct sk_buff *receive_small(struct virtnet_info *vi, void *buf, unsigned int len) |
f121159d MT |
418 | { |
419 | struct sk_buff * skb = buf; | |
420 | ||
012873d0 | 421 | len -= vi->hdr_len; |
f121159d MT |
422 | skb_trim(skb, len); |
423 | ||
424 | return skb; | |
425 | } | |
426 | ||
427 | static struct sk_buff *receive_big(struct net_device *dev, | |
946fa564 | 428 | struct virtnet_info *vi, |
f121159d MT |
429 | struct receive_queue *rq, |
430 | void *buf, | |
431 | unsigned int len) | |
432 | { | |
f600b690 | 433 | struct bpf_prog *xdp_prog; |
f121159d | 434 | struct page *page = buf; |
f600b690 | 435 | struct sk_buff *skb; |
f121159d | 436 | |
f600b690 JF |
437 | rcu_read_lock(); |
438 | xdp_prog = rcu_dereference(rq->xdp_prog); | |
439 | if (xdp_prog) { | |
440 | struct virtio_net_hdr_mrg_rxbuf *hdr = buf; | |
441 | u32 act; | |
442 | ||
b00f70b0 | 443 | if (unlikely(hdr->hdr.gso_type)) |
f600b690 | 444 | goto err_xdp; |
56434a01 JF |
445 | act = do_xdp_prog(vi, rq, xdp_prog, page, 0, len); |
446 | switch (act) { | |
447 | case XDP_PASS: | |
448 | break; | |
449 | case XDP_TX: | |
450 | rcu_read_unlock(); | |
451 | goto xdp_xmit; | |
452 | case XDP_DROP: | |
453 | default: | |
f600b690 | 454 | goto err_xdp; |
56434a01 | 455 | } |
f600b690 JF |
456 | } |
457 | rcu_read_unlock(); | |
458 | ||
459 | skb = page_to_skb(vi, rq, page, 0, len, PAGE_SIZE); | |
f121159d MT |
460 | if (unlikely(!skb)) |
461 | goto err; | |
462 | ||
463 | return skb; | |
464 | ||
f600b690 JF |
465 | err_xdp: |
466 | rcu_read_unlock(); | |
f121159d MT |
467 | err: |
468 | dev->stats.rx_dropped++; | |
469 | give_pages(rq, page); | |
56434a01 | 470 | xdp_xmit: |
f121159d MT |
471 | return NULL; |
472 | } | |
473 | ||
72979a6c JF |
474 | /* The conditions to enable XDP should preclude the underlying device from |
475 | * sending packets across multiple buffers (num_buf > 1). However per spec | |
476 | * it does not appear to be illegal to do so but rather just against convention. | |
477 | * So in order to avoid making a system unresponsive the packets are pushed | |
478 | * into a page and the XDP program is run. This will be extremely slow and we | |
479 | * push a warning to the user to fix this as soon as possible. Fixing this may | |
480 | * require resolving the underlying hardware to determine why multiple buffers | |
481 | * are being received or simply loading the XDP program in the ingress stack | |
482 | * after the skb is built because there is no advantage to running it here | |
483 | * anymore. | |
484 | */ | |
485 | static struct page *xdp_linearize_page(struct receive_queue *rq, | |
56a86f84 | 486 | u16 *num_buf, |
72979a6c JF |
487 | struct page *p, |
488 | int offset, | |
489 | unsigned int *len) | |
490 | { | |
491 | struct page *page = alloc_page(GFP_ATOMIC); | |
492 | unsigned int page_off = 0; | |
493 | ||
494 | if (!page) | |
495 | return NULL; | |
496 | ||
497 | memcpy(page_address(page) + page_off, page_address(p) + offset, *len); | |
498 | page_off += *len; | |
499 | ||
56a86f84 | 500 | while (--*num_buf) { |
72979a6c JF |
501 | unsigned int buflen; |
502 | unsigned long ctx; | |
503 | void *buf; | |
504 | int off; | |
505 | ||
506 | ctx = (unsigned long)virtqueue_get_buf(rq->vq, &buflen); | |
507 | if (unlikely(!ctx)) | |
508 | goto err_buf; | |
509 | ||
56a86f84 JW |
510 | buf = mergeable_ctx_to_buf_address(ctx); |
511 | p = virt_to_head_page(buf); | |
512 | off = buf - page_address(p); | |
513 | ||
72979a6c JF |
514 | /* guard against a misconfigured or uncooperative backend that |
515 | * is sending packet larger than the MTU. | |
516 | */ | |
56a86f84 JW |
517 | if ((page_off + buflen) > PAGE_SIZE) { |
518 | put_page(p); | |
72979a6c | 519 | goto err_buf; |
56a86f84 | 520 | } |
72979a6c JF |
521 | |
522 | memcpy(page_address(page) + page_off, | |
523 | page_address(p) + off, buflen); | |
524 | page_off += buflen; | |
56a86f84 | 525 | put_page(p); |
72979a6c JF |
526 | } |
527 | ||
528 | *len = page_off; | |
529 | return page; | |
530 | err_buf: | |
531 | __free_pages(page, 0); | |
532 | return NULL; | |
533 | } | |
534 | ||
8fc3b9e9 | 535 | static struct sk_buff *receive_mergeable(struct net_device *dev, |
fdd819b2 | 536 | struct virtnet_info *vi, |
8fc3b9e9 | 537 | struct receive_queue *rq, |
ab7db917 | 538 | unsigned long ctx, |
8fc3b9e9 | 539 | unsigned int len) |
9ab86bbc | 540 | { |
ab7db917 | 541 | void *buf = mergeable_ctx_to_buf_address(ctx); |
012873d0 MT |
542 | struct virtio_net_hdr_mrg_rxbuf *hdr = buf; |
543 | u16 num_buf = virtio16_to_cpu(vi->vdev, hdr->num_buffers); | |
8fc3b9e9 MT |
544 | struct page *page = virt_to_head_page(buf); |
545 | int offset = buf - page_address(page); | |
f600b690 JF |
546 | struct sk_buff *head_skb, *curr_skb; |
547 | struct bpf_prog *xdp_prog; | |
548 | unsigned int truesize; | |
549 | ||
56434a01 JF |
550 | head_skb = NULL; |
551 | ||
f600b690 JF |
552 | rcu_read_lock(); |
553 | xdp_prog = rcu_dereference(rq->xdp_prog); | |
554 | if (xdp_prog) { | |
72979a6c | 555 | struct page *xdp_page; |
f600b690 JF |
556 | u32 act; |
557 | ||
73b62bd0 | 558 | /* This happens when rx buffer size is underestimated */ |
f600b690 | 559 | if (unlikely(num_buf > 1)) { |
72979a6c | 560 | /* linearize data for XDP */ |
56a86f84 | 561 | xdp_page = xdp_linearize_page(rq, &num_buf, |
72979a6c JF |
562 | page, offset, &len); |
563 | if (!xdp_page) | |
564 | goto err_xdp; | |
565 | offset = 0; | |
566 | } else { | |
567 | xdp_page = page; | |
f600b690 JF |
568 | } |
569 | ||
570 | /* Transient failure which in theory could occur if | |
571 | * in-flight packets from before XDP was enabled reach | |
572 | * the receive path after XDP is loaded. In practice I | |
573 | * was not able to create this condition. | |
574 | */ | |
b00f70b0 | 575 | if (unlikely(hdr->hdr.gso_type)) |
f600b690 JF |
576 | goto err_xdp; |
577 | ||
275be061 | 578 | act = do_xdp_prog(vi, rq, xdp_prog, xdp_page, offset, len); |
56434a01 JF |
579 | switch (act) { |
580 | case XDP_PASS: | |
1830f893 JW |
581 | /* We can only create skb based on xdp_page. */ |
582 | if (unlikely(xdp_page != page)) { | |
583 | rcu_read_unlock(); | |
584 | put_page(page); | |
585 | head_skb = page_to_skb(vi, rq, xdp_page, | |
586 | 0, len, PAGE_SIZE); | |
5c33474d | 587 | ewma_pkt_len_add(&rq->mrg_avg_pkt_len, len); |
1830f893 JW |
588 | return head_skb; |
589 | } | |
56434a01 JF |
590 | break; |
591 | case XDP_TX: | |
5c33474d | 592 | ewma_pkt_len_add(&rq->mrg_avg_pkt_len, len); |
72979a6c JF |
593 | if (unlikely(xdp_page != page)) |
594 | goto err_xdp; | |
56434a01 JF |
595 | rcu_read_unlock(); |
596 | goto xdp_xmit; | |
597 | case XDP_DROP: | |
598 | default: | |
72979a6c JF |
599 | if (unlikely(xdp_page != page)) |
600 | __free_pages(xdp_page, 0); | |
5c33474d | 601 | ewma_pkt_len_add(&rq->mrg_avg_pkt_len, len); |
f600b690 | 602 | goto err_xdp; |
56434a01 | 603 | } |
f600b690 JF |
604 | } |
605 | rcu_read_unlock(); | |
ab7db917 | 606 | |
f600b690 JF |
607 | truesize = max(len, mergeable_ctx_to_buf_truesize(ctx)); |
608 | head_skb = page_to_skb(vi, rq, page, offset, len, truesize); | |
609 | curr_skb = head_skb; | |
9ab86bbc | 610 | |
8fc3b9e9 MT |
611 | if (unlikely(!curr_skb)) |
612 | goto err_skb; | |
9ab86bbc | 613 | while (--num_buf) { |
8fc3b9e9 MT |
614 | int num_skb_frags; |
615 | ||
ab7db917 MD |
616 | ctx = (unsigned long)virtqueue_get_buf(rq->vq, &len); |
617 | if (unlikely(!ctx)) { | |
8fc3b9e9 | 618 | pr_debug("%s: rx error: %d buffers out of %d missing\n", |
fdd819b2 | 619 | dev->name, num_buf, |
012873d0 MT |
620 | virtio16_to_cpu(vi->vdev, |
621 | hdr->num_buffers)); | |
8fc3b9e9 MT |
622 | dev->stats.rx_length_errors++; |
623 | goto err_buf; | |
3f2c31d9 | 624 | } |
8fc3b9e9 | 625 | |
ab7db917 | 626 | buf = mergeable_ctx_to_buf_address(ctx); |
8fc3b9e9 | 627 | page = virt_to_head_page(buf); |
8fc3b9e9 MT |
628 | |
629 | num_skb_frags = skb_shinfo(curr_skb)->nr_frags; | |
2613af0e MD |
630 | if (unlikely(num_skb_frags == MAX_SKB_FRAGS)) { |
631 | struct sk_buff *nskb = alloc_skb(0, GFP_ATOMIC); | |
8fc3b9e9 MT |
632 | |
633 | if (unlikely(!nskb)) | |
634 | goto err_skb; | |
2613af0e MD |
635 | if (curr_skb == head_skb) |
636 | skb_shinfo(curr_skb)->frag_list = nskb; | |
637 | else | |
638 | curr_skb->next = nskb; | |
639 | curr_skb = nskb; | |
640 | head_skb->truesize += nskb->truesize; | |
641 | num_skb_frags = 0; | |
642 | } | |
ab7db917 | 643 | truesize = max(len, mergeable_ctx_to_buf_truesize(ctx)); |
2613af0e MD |
644 | if (curr_skb != head_skb) { |
645 | head_skb->data_len += len; | |
646 | head_skb->len += len; | |
fb51879d | 647 | head_skb->truesize += truesize; |
2613af0e | 648 | } |
8fc3b9e9 | 649 | offset = buf - page_address(page); |
ba275241 JW |
650 | if (skb_can_coalesce(curr_skb, num_skb_frags, page, offset)) { |
651 | put_page(page); | |
652 | skb_coalesce_rx_frag(curr_skb, num_skb_frags - 1, | |
fb51879d | 653 | len, truesize); |
ba275241 JW |
654 | } else { |
655 | skb_add_rx_frag(curr_skb, num_skb_frags, page, | |
fb51879d | 656 | offset, len, truesize); |
ba275241 | 657 | } |
8fc3b9e9 MT |
658 | } |
659 | ||
5377d758 | 660 | ewma_pkt_len_add(&rq->mrg_avg_pkt_len, head_skb->len); |
8fc3b9e9 MT |
661 | return head_skb; |
662 | ||
f600b690 JF |
663 | err_xdp: |
664 | rcu_read_unlock(); | |
8fc3b9e9 MT |
665 | err_skb: |
666 | put_page(page); | |
667 | while (--num_buf) { | |
ab7db917 MD |
668 | ctx = (unsigned long)virtqueue_get_buf(rq->vq, &len); |
669 | if (unlikely(!ctx)) { | |
8fc3b9e9 MT |
670 | pr_debug("%s: rx error: %d buffers missing\n", |
671 | dev->name, num_buf); | |
672 | dev->stats.rx_length_errors++; | |
673 | break; | |
674 | } | |
ab7db917 | 675 | page = virt_to_head_page(mergeable_ctx_to_buf_address(ctx)); |
8fc3b9e9 | 676 | put_page(page); |
9ab86bbc | 677 | } |
8fc3b9e9 MT |
678 | err_buf: |
679 | dev->stats.rx_dropped++; | |
680 | dev_kfree_skb(head_skb); | |
56434a01 | 681 | xdp_xmit: |
8fc3b9e9 | 682 | return NULL; |
9ab86bbc SM |
683 | } |
684 | ||
946fa564 MT |
685 | static void receive_buf(struct virtnet_info *vi, struct receive_queue *rq, |
686 | void *buf, unsigned int len) | |
9ab86bbc | 687 | { |
e9d7417b | 688 | struct net_device *dev = vi->dev; |
58472a76 | 689 | struct virtnet_stats *stats = this_cpu_ptr(vi->stats); |
9ab86bbc | 690 | struct sk_buff *skb; |
012873d0 | 691 | struct virtio_net_hdr_mrg_rxbuf *hdr; |
3f2c31d9 | 692 | |
bcff3162 | 693 | if (unlikely(len < vi->hdr_len + ETH_HLEN)) { |
9ab86bbc SM |
694 | pr_debug("%s: short packet %i\n", dev->name, len); |
695 | dev->stats.rx_length_errors++; | |
ab7db917 MD |
696 | if (vi->mergeable_rx_bufs) { |
697 | unsigned long ctx = (unsigned long)buf; | |
698 | void *base = mergeable_ctx_to_buf_address(ctx); | |
699 | put_page(virt_to_head_page(base)); | |
700 | } else if (vi->big_packets) { | |
98bfd23c | 701 | give_pages(rq, buf); |
ab7db917 | 702 | } else { |
9ab86bbc | 703 | dev_kfree_skb(buf); |
ab7db917 | 704 | } |
9ab86bbc SM |
705 | return; |
706 | } | |
3f2c31d9 | 707 | |
f121159d | 708 | if (vi->mergeable_rx_bufs) |
fdd819b2 | 709 | skb = receive_mergeable(dev, vi, rq, (unsigned long)buf, len); |
f121159d | 710 | else if (vi->big_packets) |
946fa564 | 711 | skb = receive_big(dev, vi, rq, buf, len); |
f121159d | 712 | else |
012873d0 | 713 | skb = receive_small(vi, buf, len); |
f121159d MT |
714 | |
715 | if (unlikely(!skb)) | |
716 | return; | |
3f2c31d9 | 717 | |
9ab86bbc | 718 | hdr = skb_vnet_hdr(skb); |
3fa2a1df | 719 | |
83a27052 | 720 | u64_stats_update_begin(&stats->rx_syncp); |
3fa2a1df | 721 | stats->rx_bytes += skb->len; |
722 | stats->rx_packets++; | |
83a27052 | 723 | u64_stats_update_end(&stats->rx_syncp); |
296f96fc | 724 | |
e858fae2 | 725 | if (hdr->hdr.flags & VIRTIO_NET_HDR_F_DATA_VALID) |
10a8d94a | 726 | skb->ip_summed = CHECKSUM_UNNECESSARY; |
296f96fc | 727 | |
e858fae2 MR |
728 | if (virtio_net_hdr_to_skb(skb, &hdr->hdr, |
729 | virtio_is_little_endian(vi->vdev))) { | |
730 | net_warn_ratelimited("%s: bad gso: type: %u, size: %u\n", | |
731 | dev->name, hdr->hdr.gso_type, | |
732 | hdr->hdr.gso_size); | |
733 | goto frame_err; | |
296f96fc RR |
734 | } |
735 | ||
d1dc06dc MR |
736 | skb->protocol = eth_type_trans(skb, dev); |
737 | pr_debug("Receiving skb proto 0x%04x len %i type %i\n", | |
738 | ntohs(skb->protocol), skb->len, skb->pkt_type); | |
739 | ||
0fbd050a | 740 | napi_gro_receive(&rq->napi, skb); |
296f96fc RR |
741 | return; |
742 | ||
743 | frame_err: | |
744 | dev->stats.rx_frame_errors++; | |
296f96fc RR |
745 | dev_kfree_skb(skb); |
746 | } | |
747 | ||
946fa564 MT |
748 | static int add_recvbuf_small(struct virtnet_info *vi, struct receive_queue *rq, |
749 | gfp_t gfp) | |
296f96fc RR |
750 | { |
751 | struct sk_buff *skb; | |
012873d0 | 752 | struct virtio_net_hdr_mrg_rxbuf *hdr; |
9ab86bbc | 753 | int err; |
3f2c31d9 | 754 | |
5061de36 | 755 | skb = __netdev_alloc_skb_ip_align(vi->dev, GOOD_PACKET_LEN, gfp); |
9ab86bbc SM |
756 | if (unlikely(!skb)) |
757 | return -ENOMEM; | |
296f96fc | 758 | |
5061de36 | 759 | skb_put(skb, GOOD_PACKET_LEN); |
3f2c31d9 | 760 | |
9ab86bbc | 761 | hdr = skb_vnet_hdr(skb); |
547c890c | 762 | sg_init_table(rq->sg, 2); |
012873d0 | 763 | sg_set_buf(rq->sg, hdr, vi->hdr_len); |
e9d7417b | 764 | skb_to_sgvec(skb, rq->sg + 1, 0, skb->len); |
97402b96 | 765 | |
9dc7b9e4 | 766 | err = virtqueue_add_inbuf(rq->vq, rq->sg, 2, skb, gfp); |
9ab86bbc SM |
767 | if (err < 0) |
768 | dev_kfree_skb(skb); | |
97402b96 | 769 | |
9ab86bbc SM |
770 | return err; |
771 | } | |
97402b96 | 772 | |
012873d0 MT |
773 | static int add_recvbuf_big(struct virtnet_info *vi, struct receive_queue *rq, |
774 | gfp_t gfp) | |
9ab86bbc | 775 | { |
9ab86bbc SM |
776 | struct page *first, *list = NULL; |
777 | char *p; | |
778 | int i, err, offset; | |
779 | ||
a5835440 RR |
780 | sg_init_table(rq->sg, MAX_SKB_FRAGS + 2); |
781 | ||
e9d7417b | 782 | /* page in rq->sg[MAX_SKB_FRAGS + 1] is list tail */ |
9ab86bbc | 783 | for (i = MAX_SKB_FRAGS + 1; i > 1; --i) { |
e9d7417b | 784 | first = get_a_page(rq, gfp); |
9ab86bbc SM |
785 | if (!first) { |
786 | if (list) | |
e9d7417b | 787 | give_pages(rq, list); |
9ab86bbc | 788 | return -ENOMEM; |
97402b96 | 789 | } |
e9d7417b | 790 | sg_set_buf(&rq->sg[i], page_address(first), PAGE_SIZE); |
97402b96 | 791 | |
9ab86bbc SM |
792 | /* chain new page in list head to match sg */ |
793 | first->private = (unsigned long)list; | |
794 | list = first; | |
795 | } | |
296f96fc | 796 | |
e9d7417b | 797 | first = get_a_page(rq, gfp); |
9ab86bbc | 798 | if (!first) { |
e9d7417b | 799 | give_pages(rq, list); |
9ab86bbc SM |
800 | return -ENOMEM; |
801 | } | |
802 | p = page_address(first); | |
803 | ||
e9d7417b | 804 | /* rq->sg[0], rq->sg[1] share the same page */ |
012873d0 MT |
805 | /* a separated rq->sg[0] for header - required in case !any_header_sg */ |
806 | sg_set_buf(&rq->sg[0], p, vi->hdr_len); | |
9ab86bbc | 807 | |
e9d7417b | 808 | /* rq->sg[1] for data packet, from offset */ |
9ab86bbc | 809 | offset = sizeof(struct padded_vnet_hdr); |
e9d7417b | 810 | sg_set_buf(&rq->sg[1], p + offset, PAGE_SIZE - offset); |
9ab86bbc SM |
811 | |
812 | /* chain first in list head */ | |
813 | first->private = (unsigned long)list; | |
9dc7b9e4 RR |
814 | err = virtqueue_add_inbuf(rq->vq, rq->sg, MAX_SKB_FRAGS + 2, |
815 | first, gfp); | |
9ab86bbc | 816 | if (err < 0) |
e9d7417b | 817 | give_pages(rq, first); |
9ab86bbc SM |
818 | |
819 | return err; | |
296f96fc RR |
820 | } |
821 | ||
5377d758 | 822 | static unsigned int get_mergeable_buf_len(struct ewma_pkt_len *avg_pkt_len) |
3f2c31d9 | 823 | { |
ab7db917 | 824 | const size_t hdr_len = sizeof(struct virtio_net_hdr_mrg_rxbuf); |
fbf28d78 MD |
825 | unsigned int len; |
826 | ||
5377d758 | 827 | len = hdr_len + clamp_t(unsigned int, ewma_pkt_len_read(avg_pkt_len), |
fbf28d78 MD |
828 | GOOD_PACKET_LEN, PAGE_SIZE - hdr_len); |
829 | return ALIGN(len, MERGEABLE_BUFFER_ALIGN); | |
830 | } | |
831 | ||
832 | static int add_recvbuf_mergeable(struct receive_queue *rq, gfp_t gfp) | |
833 | { | |
fb51879d MD |
834 | struct page_frag *alloc_frag = &rq->alloc_frag; |
835 | char *buf; | |
ab7db917 | 836 | unsigned long ctx; |
3f2c31d9 | 837 | int err; |
fb51879d | 838 | unsigned int len, hole; |
3f2c31d9 | 839 | |
fbf28d78 | 840 | len = get_mergeable_buf_len(&rq->mrg_avg_pkt_len); |
ab7db917 | 841 | if (unlikely(!skb_page_frag_refill(len, alloc_frag, gfp))) |
9ab86bbc | 842 | return -ENOMEM; |
ab7db917 | 843 | |
fb51879d | 844 | buf = (char *)page_address(alloc_frag->page) + alloc_frag->offset; |
ab7db917 | 845 | ctx = mergeable_buf_to_ctx(buf, len); |
fb51879d | 846 | get_page(alloc_frag->page); |
fb51879d MD |
847 | alloc_frag->offset += len; |
848 | hole = alloc_frag->size - alloc_frag->offset; | |
ab7db917 MD |
849 | if (hole < len) { |
850 | /* To avoid internal fragmentation, if there is very likely not | |
851 | * enough space for another buffer, add the remaining space to | |
852 | * the current buffer. This extra space is not included in | |
853 | * the truesize stored in ctx. | |
854 | */ | |
fb51879d MD |
855 | len += hole; |
856 | alloc_frag->offset += hole; | |
857 | } | |
3f2c31d9 | 858 | |
fb51879d | 859 | sg_init_one(rq->sg, buf, len); |
ab7db917 | 860 | err = virtqueue_add_inbuf(rq->vq, rq->sg, 1, (void *)ctx, gfp); |
9ab86bbc | 861 | if (err < 0) |
2613af0e | 862 | put_page(virt_to_head_page(buf)); |
3f2c31d9 | 863 | |
9ab86bbc SM |
864 | return err; |
865 | } | |
3f2c31d9 | 866 | |
b2baed69 RR |
867 | /* |
868 | * Returns false if we couldn't fill entirely (OOM). | |
869 | * | |
870 | * Normally run in the receive path, but can also be run from ndo_open | |
871 | * before we're receiving packets, or from refill_work which is | |
872 | * careful to disable receiving (using napi_disable). | |
873 | */ | |
946fa564 MT |
874 | static bool try_fill_recv(struct virtnet_info *vi, struct receive_queue *rq, |
875 | gfp_t gfp) | |
9ab86bbc SM |
876 | { |
877 | int err; | |
1788f495 | 878 | bool oom; |
3f2c31d9 | 879 | |
fb51879d | 880 | gfp |= __GFP_COLD; |
9ab86bbc SM |
881 | do { |
882 | if (vi->mergeable_rx_bufs) | |
e9d7417b | 883 | err = add_recvbuf_mergeable(rq, gfp); |
9ab86bbc | 884 | else if (vi->big_packets) |
012873d0 | 885 | err = add_recvbuf_big(vi, rq, gfp); |
9ab86bbc | 886 | else |
946fa564 | 887 | err = add_recvbuf_small(vi, rq, gfp); |
3f2c31d9 | 888 | |
1788f495 | 889 | oom = err == -ENOMEM; |
9ed4cb07 | 890 | if (err) |
3f2c31d9 | 891 | break; |
b7dfde95 | 892 | } while (rq->vq->num_free); |
681daee2 | 893 | virtqueue_kick(rq->vq); |
3161e453 | 894 | return !oom; |
3f2c31d9 MM |
895 | } |
896 | ||
18445c4d | 897 | static void skb_recv_done(struct virtqueue *rvq) |
296f96fc RR |
898 | { |
899 | struct virtnet_info *vi = rvq->vdev->priv; | |
986a4f4d | 900 | struct receive_queue *rq = &vi->rq[vq2rxq(rvq)]; |
e9d7417b | 901 | |
18445c4d | 902 | /* Schedule NAPI, Suppress further interrupts if successful. */ |
e9d7417b | 903 | if (napi_schedule_prep(&rq->napi)) { |
1915a712 | 904 | virtqueue_disable_cb(rvq); |
e9d7417b | 905 | __napi_schedule(&rq->napi); |
18445c4d | 906 | } |
296f96fc RR |
907 | } |
908 | ||
e9d7417b | 909 | static void virtnet_napi_enable(struct receive_queue *rq) |
3e9d08ec | 910 | { |
e9d7417b | 911 | napi_enable(&rq->napi); |
3e9d08ec BR |
912 | |
913 | /* If all buffers were filled by other side before we napi_enabled, we | |
914 | * won't get another interrupt, so process any outstanding packets | |
915 | * now. virtnet_poll wants re-enable the queue, so we disable here. | |
916 | * We synchronize against interrupts via NAPI_STATE_SCHED */ | |
e9d7417b JW |
917 | if (napi_schedule_prep(&rq->napi)) { |
918 | virtqueue_disable_cb(rq->vq); | |
ec13ee80 | 919 | local_bh_disable(); |
e9d7417b | 920 | __napi_schedule(&rq->napi); |
ec13ee80 | 921 | local_bh_enable(); |
3e9d08ec BR |
922 | } |
923 | } | |
924 | ||
3161e453 RR |
925 | static void refill_work(struct work_struct *work) |
926 | { | |
e9d7417b JW |
927 | struct virtnet_info *vi = |
928 | container_of(work, struct virtnet_info, refill.work); | |
3161e453 | 929 | bool still_empty; |
986a4f4d JW |
930 | int i; |
931 | ||
55257d72 | 932 | for (i = 0; i < vi->curr_queue_pairs; i++) { |
986a4f4d | 933 | struct receive_queue *rq = &vi->rq[i]; |
3161e453 | 934 | |
986a4f4d | 935 | napi_disable(&rq->napi); |
946fa564 | 936 | still_empty = !try_fill_recv(vi, rq, GFP_KERNEL); |
986a4f4d | 937 | virtnet_napi_enable(rq); |
3161e453 | 938 | |
986a4f4d JW |
939 | /* In theory, this can happen: if we don't get any buffers in |
940 | * we will *never* try to fill again. | |
941 | */ | |
942 | if (still_empty) | |
943 | schedule_delayed_work(&vi->refill, HZ/2); | |
944 | } | |
3161e453 RR |
945 | } |
946 | ||
2ffa7598 | 947 | static int virtnet_receive(struct receive_queue *rq, int budget) |
296f96fc | 948 | { |
e9d7417b | 949 | struct virtnet_info *vi = rq->vq->vdev->priv; |
2ffa7598 | 950 | unsigned int len, received = 0; |
9ab86bbc | 951 | void *buf; |
296f96fc | 952 | |
296f96fc | 953 | while (received < budget && |
e9d7417b | 954 | (buf = virtqueue_get_buf(rq->vq, &len)) != NULL) { |
946fa564 | 955 | receive_buf(vi, rq, buf, len); |
296f96fc RR |
956 | received++; |
957 | } | |
958 | ||
be121f46 | 959 | if (rq->vq->num_free > virtqueue_get_vring_size(rq->vq) / 2) { |
946fa564 | 960 | if (!try_fill_recv(vi, rq, GFP_ATOMIC)) |
3b07e9ca | 961 | schedule_delayed_work(&vi->refill, 0); |
3161e453 | 962 | } |
296f96fc | 963 | |
2ffa7598 JW |
964 | return received; |
965 | } | |
966 | ||
967 | static int virtnet_poll(struct napi_struct *napi, int budget) | |
968 | { | |
969 | struct receive_queue *rq = | |
970 | container_of(napi, struct receive_queue, napi); | |
faadb05f | 971 | unsigned int r, received; |
2ffa7598 | 972 | |
faadb05f | 973 | received = virtnet_receive(rq, budget); |
2ffa7598 | 974 | |
8329d98e RR |
975 | /* Out of packets? */ |
976 | if (received < budget) { | |
cbdadbbf | 977 | r = virtqueue_enable_cb_prepare(rq->vq); |
0fbd050a | 978 | napi_complete_done(napi, received); |
cbdadbbf | 979 | if (unlikely(virtqueue_poll(rq->vq, r)) && |
8e95a202 | 980 | napi_schedule_prep(napi)) { |
e9d7417b | 981 | virtqueue_disable_cb(rq->vq); |
288379f0 | 982 | __napi_schedule(napi); |
4265f161 | 983 | } |
296f96fc RR |
984 | } |
985 | ||
986 | return received; | |
987 | } | |
988 | ||
91815639 JW |
989 | #ifdef CONFIG_NET_RX_BUSY_POLL |
990 | /* must be called with local_bh_disable()d */ | |
991 | static int virtnet_busy_poll(struct napi_struct *napi) | |
992 | { | |
993 | struct receive_queue *rq = | |
994 | container_of(napi, struct receive_queue, napi); | |
995 | struct virtnet_info *vi = rq->vq->vdev->priv; | |
996 | int r, received = 0, budget = 4; | |
997 | ||
998 | if (!(vi->status & VIRTIO_NET_S_LINK_UP)) | |
999 | return LL_FLUSH_FAILED; | |
1000 | ||
1001 | if (!napi_schedule_prep(napi)) | |
1002 | return LL_FLUSH_BUSY; | |
1003 | ||
1004 | virtqueue_disable_cb(rq->vq); | |
1005 | ||
1006 | again: | |
1007 | received += virtnet_receive(rq, budget); | |
1008 | ||
1009 | r = virtqueue_enable_cb_prepare(rq->vq); | |
1010 | clear_bit(NAPI_STATE_SCHED, &napi->state); | |
1011 | if (unlikely(virtqueue_poll(rq->vq, r)) && | |
1012 | napi_schedule_prep(napi)) { | |
1013 | virtqueue_disable_cb(rq->vq); | |
1014 | if (received < budget) { | |
1015 | budget -= received; | |
1016 | goto again; | |
1017 | } else { | |
1018 | __napi_schedule(napi); | |
1019 | } | |
1020 | } | |
1021 | ||
1022 | return received; | |
1023 | } | |
1024 | #endif /* CONFIG_NET_RX_BUSY_POLL */ | |
1025 | ||
986a4f4d JW |
1026 | static int virtnet_open(struct net_device *dev) |
1027 | { | |
1028 | struct virtnet_info *vi = netdev_priv(dev); | |
1029 | int i; | |
1030 | ||
e4166625 JW |
1031 | for (i = 0; i < vi->max_queue_pairs; i++) { |
1032 | if (i < vi->curr_queue_pairs) | |
1033 | /* Make sure we have some buffers: if oom use wq. */ | |
946fa564 | 1034 | if (!try_fill_recv(vi, &vi->rq[i], GFP_KERNEL)) |
e4166625 | 1035 | schedule_delayed_work(&vi->refill, 0); |
986a4f4d JW |
1036 | virtnet_napi_enable(&vi->rq[i]); |
1037 | } | |
1038 | ||
1039 | return 0; | |
1040 | } | |
1041 | ||
b7dfde95 | 1042 | static void free_old_xmit_skbs(struct send_queue *sq) |
296f96fc RR |
1043 | { |
1044 | struct sk_buff *skb; | |
6ee57bcc | 1045 | unsigned int len; |
e9d7417b | 1046 | struct virtnet_info *vi = sq->vq->vdev->priv; |
58472a76 | 1047 | struct virtnet_stats *stats = this_cpu_ptr(vi->stats); |
296f96fc | 1048 | |
e9d7417b | 1049 | while ((skb = virtqueue_get_buf(sq->vq, &len)) != NULL) { |
296f96fc | 1050 | pr_debug("Sent skb %p\n", skb); |
3fa2a1df | 1051 | |
83a27052 | 1052 | u64_stats_update_begin(&stats->tx_syncp); |
3fa2a1df | 1053 | stats->tx_bytes += skb->len; |
1054 | stats->tx_packets++; | |
83a27052 | 1055 | u64_stats_update_end(&stats->tx_syncp); |
3fa2a1df | 1056 | |
ed79bab8 | 1057 | dev_kfree_skb_any(skb); |
296f96fc RR |
1058 | } |
1059 | } | |
1060 | ||
e9d7417b | 1061 | static int xmit_skb(struct send_queue *sq, struct sk_buff *skb) |
296f96fc | 1062 | { |
012873d0 | 1063 | struct virtio_net_hdr_mrg_rxbuf *hdr; |
296f96fc | 1064 | const unsigned char *dest = ((struct ethhdr *)skb->data)->h_dest; |
e9d7417b | 1065 | struct virtnet_info *vi = sq->vq->vdev->priv; |
7bedc7dc | 1066 | unsigned num_sg; |
012873d0 | 1067 | unsigned hdr_len = vi->hdr_len; |
e7428e95 | 1068 | bool can_push; |
296f96fc | 1069 | |
e174961c | 1070 | pr_debug("%s: xmit %p %pM\n", vi->dev->name, skb, dest); |
e7428e95 MT |
1071 | |
1072 | can_push = vi->any_header_sg && | |
1073 | !((unsigned long)skb->data & (__alignof__(*hdr) - 1)) && | |
1074 | !skb_header_cloned(skb) && skb_headroom(skb) >= hdr_len; | |
1075 | /* Even if we can, don't push here yet as this would skew | |
1076 | * csum_start offset below. */ | |
1077 | if (can_push) | |
012873d0 | 1078 | hdr = (struct virtio_net_hdr_mrg_rxbuf *)(skb->data - hdr_len); |
e7428e95 MT |
1079 | else |
1080 | hdr = skb_vnet_hdr(skb); | |
296f96fc | 1081 | |
e858fae2 MR |
1082 | if (virtio_net_hdr_from_skb(skb, &hdr->hdr, |
1083 | virtio_is_little_endian(vi->vdev))) | |
1084 | BUG(); | |
296f96fc | 1085 | |
3f2c31d9 | 1086 | if (vi->mergeable_rx_bufs) |
012873d0 | 1087 | hdr->num_buffers = 0; |
3f2c31d9 | 1088 | |
547c890c | 1089 | sg_init_table(sq->sg, skb_shinfo(skb)->nr_frags + (can_push ? 1 : 2)); |
e7428e95 MT |
1090 | if (can_push) { |
1091 | __skb_push(skb, hdr_len); | |
1092 | num_sg = skb_to_sgvec(skb, sq->sg, 0, skb->len); | |
1093 | /* Pull header back to avoid skew in tx bytes calculations. */ | |
1094 | __skb_pull(skb, hdr_len); | |
1095 | } else { | |
1096 | sg_set_buf(sq->sg, hdr, hdr_len); | |
1097 | num_sg = skb_to_sgvec(skb, sq->sg + 1, 0, skb->len) + 1; | |
1098 | } | |
9dc7b9e4 | 1099 | return virtqueue_add_outbuf(sq->vq, sq->sg, num_sg, skb, GFP_ATOMIC); |
11a3a154 RR |
1100 | } |
1101 | ||
424efe9c | 1102 | static netdev_tx_t start_xmit(struct sk_buff *skb, struct net_device *dev) |
99ffc696 RR |
1103 | { |
1104 | struct virtnet_info *vi = netdev_priv(dev); | |
986a4f4d JW |
1105 | int qnum = skb_get_queue_mapping(skb); |
1106 | struct send_queue *sq = &vi->sq[qnum]; | |
9ed4cb07 | 1107 | int err; |
4b7fd2e6 MT |
1108 | struct netdev_queue *txq = netdev_get_tx_queue(dev, qnum); |
1109 | bool kick = !skb->xmit_more; | |
2cb9c6ba | 1110 | |
2cb9c6ba | 1111 | /* Free up any pending old buffers before queueing new ones. */ |
e9d7417b | 1112 | free_old_xmit_skbs(sq); |
99ffc696 | 1113 | |
074c3582 JK |
1114 | /* timestamp packet in software */ |
1115 | skb_tx_timestamp(skb); | |
1116 | ||
03f191ba | 1117 | /* Try to transmit */ |
b7dfde95 | 1118 | err = xmit_skb(sq, skb); |
48925e37 | 1119 | |
9ed4cb07 | 1120 | /* This should not happen! */ |
681daee2 | 1121 | if (unlikely(err)) { |
9ed4cb07 RR |
1122 | dev->stats.tx_fifo_errors++; |
1123 | if (net_ratelimit()) | |
1124 | dev_warn(&dev->dev, | |
b7dfde95 | 1125 | "Unexpected TXQ (%d) queue failure: %d\n", qnum, err); |
58eba97d | 1126 | dev->stats.tx_dropped++; |
85e94525 | 1127 | dev_kfree_skb_any(skb); |
58eba97d | 1128 | return NETDEV_TX_OK; |
296f96fc | 1129 | } |
03f191ba | 1130 | |
48925e37 RR |
1131 | /* Don't wait up for transmitted skbs to be freed. */ |
1132 | skb_orphan(skb); | |
1133 | nf_reset(skb); | |
1134 | ||
60302ff6 MT |
1135 | /* If running out of space, stop queue to avoid getting packets that we |
1136 | * are then unable to transmit. | |
1137 | * An alternative would be to force queuing layer to requeue the skb by | |
1138 | * returning NETDEV_TX_BUSY. However, NETDEV_TX_BUSY should not be | |
1139 | * returned in a normal path of operation: it means that driver is not | |
1140 | * maintaining the TX queue stop/start state properly, and causes | |
1141 | * the stack to do a non-trivial amount of useless work. | |
1142 | * Since most packets only take 1 or 2 ring slots, stopping the queue | |
1143 | * early means 16 slots are typically wasted. | |
d631b94e | 1144 | */ |
b7dfde95 | 1145 | if (sq->vq->num_free < 2+MAX_SKB_FRAGS) { |
986a4f4d | 1146 | netif_stop_subqueue(dev, qnum); |
e9d7417b | 1147 | if (unlikely(!virtqueue_enable_cb_delayed(sq->vq))) { |
48925e37 | 1148 | /* More just got used, free them then recheck. */ |
b7dfde95 LT |
1149 | free_old_xmit_skbs(sq); |
1150 | if (sq->vq->num_free >= 2+MAX_SKB_FRAGS) { | |
986a4f4d | 1151 | netif_start_subqueue(dev, qnum); |
e9d7417b | 1152 | virtqueue_disable_cb(sq->vq); |
48925e37 RR |
1153 | } |
1154 | } | |
99ffc696 | 1155 | } |
48925e37 | 1156 | |
4b7fd2e6 | 1157 | if (kick || netif_xmit_stopped(txq)) |
0b725a2c | 1158 | virtqueue_kick(sq->vq); |
296f96fc | 1159 | |
0b725a2c | 1160 | return NETDEV_TX_OK; |
c223a078 DM |
1161 | } |
1162 | ||
40cbfc37 AK |
1163 | /* |
1164 | * Send command via the control virtqueue and check status. Commands | |
1165 | * supported by the hypervisor, as indicated by feature bits, should | |
788a8b6d | 1166 | * never fail unless improperly formatted. |
40cbfc37 AK |
1167 | */ |
1168 | static bool virtnet_send_command(struct virtnet_info *vi, u8 class, u8 cmd, | |
d24bae32 | 1169 | struct scatterlist *out) |
40cbfc37 | 1170 | { |
f7bc9594 | 1171 | struct scatterlist *sgs[4], hdr, stat; |
d24bae32 | 1172 | unsigned out_num = 0, tmp; |
40cbfc37 AK |
1173 | |
1174 | /* Caller should know better */ | |
f7bc9594 | 1175 | BUG_ON(!virtio_has_feature(vi->vdev, VIRTIO_NET_F_CTRL_VQ)); |
40cbfc37 | 1176 | |
2ac46030 MT |
1177 | vi->ctrl_status = ~0; |
1178 | vi->ctrl_hdr.class = class; | |
1179 | vi->ctrl_hdr.cmd = cmd; | |
f7bc9594 | 1180 | /* Add header */ |
2ac46030 | 1181 | sg_init_one(&hdr, &vi->ctrl_hdr, sizeof(vi->ctrl_hdr)); |
f7bc9594 | 1182 | sgs[out_num++] = &hdr; |
40cbfc37 | 1183 | |
f7bc9594 RR |
1184 | if (out) |
1185 | sgs[out_num++] = out; | |
40cbfc37 | 1186 | |
f7bc9594 | 1187 | /* Add return status. */ |
2ac46030 | 1188 | sg_init_one(&stat, &vi->ctrl_status, sizeof(vi->ctrl_status)); |
d24bae32 | 1189 | sgs[out_num] = &stat; |
40cbfc37 | 1190 | |
d24bae32 | 1191 | BUG_ON(out_num + 1 > ARRAY_SIZE(sgs)); |
a7c58146 | 1192 | virtqueue_add_sgs(vi->cvq, sgs, out_num, 1, vi, GFP_ATOMIC); |
40cbfc37 | 1193 | |
67975901 | 1194 | if (unlikely(!virtqueue_kick(vi->cvq))) |
2ac46030 | 1195 | return vi->ctrl_status == VIRTIO_NET_OK; |
40cbfc37 AK |
1196 | |
1197 | /* Spin for a response, the kick causes an ioport write, trapping | |
1198 | * into the hypervisor, so the request should be handled immediately. | |
1199 | */ | |
047b9b94 HG |
1200 | while (!virtqueue_get_buf(vi->cvq, &tmp) && |
1201 | !virtqueue_is_broken(vi->cvq)) | |
40cbfc37 AK |
1202 | cpu_relax(); |
1203 | ||
2ac46030 | 1204 | return vi->ctrl_status == VIRTIO_NET_OK; |
40cbfc37 AK |
1205 | } |
1206 | ||
9c46f6d4 AW |
1207 | static int virtnet_set_mac_address(struct net_device *dev, void *p) |
1208 | { | |
1209 | struct virtnet_info *vi = netdev_priv(dev); | |
1210 | struct virtio_device *vdev = vi->vdev; | |
f2f2c8b4 | 1211 | int ret; |
e37e2ff3 | 1212 | struct sockaddr *addr; |
7e58d5ae | 1213 | struct scatterlist sg; |
9c46f6d4 | 1214 | |
e37e2ff3 AL |
1215 | addr = kmalloc(sizeof(*addr), GFP_KERNEL); |
1216 | if (!addr) | |
1217 | return -ENOMEM; | |
1218 | memcpy(addr, p, sizeof(*addr)); | |
1219 | ||
1220 | ret = eth_prepare_mac_addr_change(dev, addr); | |
f2f2c8b4 | 1221 | if (ret) |
e37e2ff3 | 1222 | goto out; |
9c46f6d4 | 1223 | |
7e58d5ae AK |
1224 | if (virtio_has_feature(vdev, VIRTIO_NET_F_CTRL_MAC_ADDR)) { |
1225 | sg_init_one(&sg, addr->sa_data, dev->addr_len); | |
1226 | if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_MAC, | |
d24bae32 | 1227 | VIRTIO_NET_CTRL_MAC_ADDR_SET, &sg)) { |
7e58d5ae AK |
1228 | dev_warn(&vdev->dev, |
1229 | "Failed to set mac address by vq command.\n"); | |
e37e2ff3 AL |
1230 | ret = -EINVAL; |
1231 | goto out; | |
7e58d5ae | 1232 | } |
7e93a02f MT |
1233 | } else if (virtio_has_feature(vdev, VIRTIO_NET_F_MAC) && |
1234 | !virtio_has_feature(vdev, VIRTIO_F_VERSION_1)) { | |
855e0c52 RR |
1235 | unsigned int i; |
1236 | ||
1237 | /* Naturally, this has an atomicity problem. */ | |
1238 | for (i = 0; i < dev->addr_len; i++) | |
1239 | virtio_cwrite8(vdev, | |
1240 | offsetof(struct virtio_net_config, mac) + | |
1241 | i, addr->sa_data[i]); | |
7e58d5ae AK |
1242 | } |
1243 | ||
1244 | eth_commit_mac_addr_change(dev, p); | |
e37e2ff3 | 1245 | ret = 0; |
9c46f6d4 | 1246 | |
e37e2ff3 AL |
1247 | out: |
1248 | kfree(addr); | |
1249 | return ret; | |
9c46f6d4 AW |
1250 | } |
1251 | ||
3fa2a1df | 1252 | static struct rtnl_link_stats64 *virtnet_stats(struct net_device *dev, |
1253 | struct rtnl_link_stats64 *tot) | |
1254 | { | |
1255 | struct virtnet_info *vi = netdev_priv(dev); | |
1256 | int cpu; | |
1257 | unsigned int start; | |
1258 | ||
1259 | for_each_possible_cpu(cpu) { | |
58472a76 | 1260 | struct virtnet_stats *stats = per_cpu_ptr(vi->stats, cpu); |
3fa2a1df | 1261 | u64 tpackets, tbytes, rpackets, rbytes; |
1262 | ||
1263 | do { | |
57a7744e | 1264 | start = u64_stats_fetch_begin_irq(&stats->tx_syncp); |
3fa2a1df | 1265 | tpackets = stats->tx_packets; |
1266 | tbytes = stats->tx_bytes; | |
57a7744e | 1267 | } while (u64_stats_fetch_retry_irq(&stats->tx_syncp, start)); |
83a27052 ED |
1268 | |
1269 | do { | |
57a7744e | 1270 | start = u64_stats_fetch_begin_irq(&stats->rx_syncp); |
3fa2a1df | 1271 | rpackets = stats->rx_packets; |
1272 | rbytes = stats->rx_bytes; | |
57a7744e | 1273 | } while (u64_stats_fetch_retry_irq(&stats->rx_syncp, start)); |
3fa2a1df | 1274 | |
1275 | tot->rx_packets += rpackets; | |
1276 | tot->tx_packets += tpackets; | |
1277 | tot->rx_bytes += rbytes; | |
1278 | tot->tx_bytes += tbytes; | |
1279 | } | |
1280 | ||
1281 | tot->tx_dropped = dev->stats.tx_dropped; | |
021ac8d3 | 1282 | tot->tx_fifo_errors = dev->stats.tx_fifo_errors; |
3fa2a1df | 1283 | tot->rx_dropped = dev->stats.rx_dropped; |
1284 | tot->rx_length_errors = dev->stats.rx_length_errors; | |
1285 | tot->rx_frame_errors = dev->stats.rx_frame_errors; | |
1286 | ||
1287 | return tot; | |
1288 | } | |
1289 | ||
da74e89d AS |
1290 | #ifdef CONFIG_NET_POLL_CONTROLLER |
1291 | static void virtnet_netpoll(struct net_device *dev) | |
1292 | { | |
1293 | struct virtnet_info *vi = netdev_priv(dev); | |
986a4f4d | 1294 | int i; |
da74e89d | 1295 | |
986a4f4d JW |
1296 | for (i = 0; i < vi->curr_queue_pairs; i++) |
1297 | napi_schedule(&vi->rq[i].napi); | |
da74e89d AS |
1298 | } |
1299 | #endif | |
1300 | ||
586d17c5 JW |
1301 | static void virtnet_ack_link_announce(struct virtnet_info *vi) |
1302 | { | |
1303 | rtnl_lock(); | |
1304 | if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_ANNOUNCE, | |
d24bae32 | 1305 | VIRTIO_NET_CTRL_ANNOUNCE_ACK, NULL)) |
586d17c5 JW |
1306 | dev_warn(&vi->dev->dev, "Failed to ack link announce.\n"); |
1307 | rtnl_unlock(); | |
1308 | } | |
1309 | ||
986a4f4d JW |
1310 | static int virtnet_set_queues(struct virtnet_info *vi, u16 queue_pairs) |
1311 | { | |
1312 | struct scatterlist sg; | |
986a4f4d JW |
1313 | struct net_device *dev = vi->dev; |
1314 | ||
1315 | if (!vi->has_cvq || !virtio_has_feature(vi->vdev, VIRTIO_NET_F_MQ)) | |
1316 | return 0; | |
1317 | ||
a725ee3e AL |
1318 | vi->ctrl_mq.virtqueue_pairs = cpu_to_virtio16(vi->vdev, queue_pairs); |
1319 | sg_init_one(&sg, &vi->ctrl_mq, sizeof(vi->ctrl_mq)); | |
986a4f4d JW |
1320 | |
1321 | if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_MQ, | |
d24bae32 | 1322 | VIRTIO_NET_CTRL_MQ_VQ_PAIRS_SET, &sg)) { |
986a4f4d JW |
1323 | dev_warn(&dev->dev, "Fail to set num of queue pairs to %d\n", |
1324 | queue_pairs); | |
1325 | return -EINVAL; | |
55257d72 | 1326 | } else { |
986a4f4d | 1327 | vi->curr_queue_pairs = queue_pairs; |
35ed159b JW |
1328 | /* virtnet_open() will refill when device is going to up. */ |
1329 | if (dev->flags & IFF_UP) | |
1330 | schedule_delayed_work(&vi->refill, 0); | |
55257d72 | 1331 | } |
986a4f4d JW |
1332 | |
1333 | return 0; | |
1334 | } | |
1335 | ||
296f96fc RR |
1336 | static int virtnet_close(struct net_device *dev) |
1337 | { | |
1338 | struct virtnet_info *vi = netdev_priv(dev); | |
986a4f4d | 1339 | int i; |
296f96fc | 1340 | |
b2baed69 RR |
1341 | /* Make sure refill_work doesn't re-enable napi! */ |
1342 | cancel_delayed_work_sync(&vi->refill); | |
986a4f4d JW |
1343 | |
1344 | for (i = 0; i < vi->max_queue_pairs; i++) | |
1345 | napi_disable(&vi->rq[i].napi); | |
296f96fc | 1346 | |
296f96fc RR |
1347 | return 0; |
1348 | } | |
1349 | ||
2af7698e AW |
1350 | static void virtnet_set_rx_mode(struct net_device *dev) |
1351 | { | |
1352 | struct virtnet_info *vi = netdev_priv(dev); | |
f565a7c2 | 1353 | struct scatterlist sg[2]; |
f565a7c2 | 1354 | struct virtio_net_ctrl_mac *mac_data; |
ccffad25 | 1355 | struct netdev_hw_addr *ha; |
32e7bfc4 | 1356 | int uc_count; |
4cd24eaf | 1357 | int mc_count; |
f565a7c2 AW |
1358 | void *buf; |
1359 | int i; | |
2af7698e | 1360 | |
788a8b6d | 1361 | /* We can't dynamically set ndo_set_rx_mode, so return gracefully */ |
2af7698e AW |
1362 | if (!virtio_has_feature(vi->vdev, VIRTIO_NET_F_CTRL_RX)) |
1363 | return; | |
1364 | ||
2ac46030 MT |
1365 | vi->ctrl_promisc = ((dev->flags & IFF_PROMISC) != 0); |
1366 | vi->ctrl_allmulti = ((dev->flags & IFF_ALLMULTI) != 0); | |
2af7698e | 1367 | |
2ac46030 | 1368 | sg_init_one(sg, &vi->ctrl_promisc, sizeof(vi->ctrl_promisc)); |
2af7698e AW |
1369 | |
1370 | if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_RX, | |
d24bae32 | 1371 | VIRTIO_NET_CTRL_RX_PROMISC, sg)) |
2af7698e | 1372 | dev_warn(&dev->dev, "Failed to %sable promisc mode.\n", |
2ac46030 | 1373 | vi->ctrl_promisc ? "en" : "dis"); |
2af7698e | 1374 | |
2ac46030 | 1375 | sg_init_one(sg, &vi->ctrl_allmulti, sizeof(vi->ctrl_allmulti)); |
2af7698e AW |
1376 | |
1377 | if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_RX, | |
d24bae32 | 1378 | VIRTIO_NET_CTRL_RX_ALLMULTI, sg)) |
2af7698e | 1379 | dev_warn(&dev->dev, "Failed to %sable allmulti mode.\n", |
2ac46030 | 1380 | vi->ctrl_allmulti ? "en" : "dis"); |
f565a7c2 | 1381 | |
32e7bfc4 | 1382 | uc_count = netdev_uc_count(dev); |
4cd24eaf | 1383 | mc_count = netdev_mc_count(dev); |
f565a7c2 | 1384 | /* MAC filter - use one buffer for both lists */ |
4cd24eaf JP |
1385 | buf = kzalloc(((uc_count + mc_count) * ETH_ALEN) + |
1386 | (2 * sizeof(mac_data->entries)), GFP_ATOMIC); | |
1387 | mac_data = buf; | |
e68ed8f0 | 1388 | if (!buf) |
f565a7c2 | 1389 | return; |
f565a7c2 | 1390 | |
23e258e1 AW |
1391 | sg_init_table(sg, 2); |
1392 | ||
f565a7c2 | 1393 | /* Store the unicast list and count in the front of the buffer */ |
fdd819b2 | 1394 | mac_data->entries = cpu_to_virtio32(vi->vdev, uc_count); |
ccffad25 | 1395 | i = 0; |
32e7bfc4 | 1396 | netdev_for_each_uc_addr(ha, dev) |
ccffad25 | 1397 | memcpy(&mac_data->macs[i++][0], ha->addr, ETH_ALEN); |
f565a7c2 AW |
1398 | |
1399 | sg_set_buf(&sg[0], mac_data, | |
32e7bfc4 | 1400 | sizeof(mac_data->entries) + (uc_count * ETH_ALEN)); |
f565a7c2 AW |
1401 | |
1402 | /* multicast list and count fill the end */ | |
32e7bfc4 | 1403 | mac_data = (void *)&mac_data->macs[uc_count][0]; |
f565a7c2 | 1404 | |
fdd819b2 | 1405 | mac_data->entries = cpu_to_virtio32(vi->vdev, mc_count); |
567ec874 | 1406 | i = 0; |
22bedad3 JP |
1407 | netdev_for_each_mc_addr(ha, dev) |
1408 | memcpy(&mac_data->macs[i++][0], ha->addr, ETH_ALEN); | |
f565a7c2 AW |
1409 | |
1410 | sg_set_buf(&sg[1], mac_data, | |
4cd24eaf | 1411 | sizeof(mac_data->entries) + (mc_count * ETH_ALEN)); |
f565a7c2 AW |
1412 | |
1413 | if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_MAC, | |
d24bae32 | 1414 | VIRTIO_NET_CTRL_MAC_TABLE_SET, sg)) |
99e872ae | 1415 | dev_warn(&dev->dev, "Failed to set MAC filter table.\n"); |
f565a7c2 AW |
1416 | |
1417 | kfree(buf); | |
2af7698e AW |
1418 | } |
1419 | ||
80d5c368 PM |
1420 | static int virtnet_vlan_rx_add_vid(struct net_device *dev, |
1421 | __be16 proto, u16 vid) | |
0bde9569 AW |
1422 | { |
1423 | struct virtnet_info *vi = netdev_priv(dev); | |
1424 | struct scatterlist sg; | |
1425 | ||
a725ee3e AL |
1426 | vi->ctrl_vid = vid; |
1427 | sg_init_one(&sg, &vi->ctrl_vid, sizeof(vi->ctrl_vid)); | |
0bde9569 AW |
1428 | |
1429 | if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_VLAN, | |
d24bae32 | 1430 | VIRTIO_NET_CTRL_VLAN_ADD, &sg)) |
0bde9569 | 1431 | dev_warn(&dev->dev, "Failed to add VLAN ID %d.\n", vid); |
8e586137 | 1432 | return 0; |
0bde9569 AW |
1433 | } |
1434 | ||
80d5c368 PM |
1435 | static int virtnet_vlan_rx_kill_vid(struct net_device *dev, |
1436 | __be16 proto, u16 vid) | |
0bde9569 AW |
1437 | { |
1438 | struct virtnet_info *vi = netdev_priv(dev); | |
1439 | struct scatterlist sg; | |
1440 | ||
a725ee3e AL |
1441 | vi->ctrl_vid = vid; |
1442 | sg_init_one(&sg, &vi->ctrl_vid, sizeof(vi->ctrl_vid)); | |
0bde9569 AW |
1443 | |
1444 | if (!virtnet_send_command(vi, VIRTIO_NET_CTRL_VLAN, | |
d24bae32 | 1445 | VIRTIO_NET_CTRL_VLAN_DEL, &sg)) |
0bde9569 | 1446 | dev_warn(&dev->dev, "Failed to kill VLAN ID %d.\n", vid); |
8e586137 | 1447 | return 0; |
0bde9569 AW |
1448 | } |
1449 | ||
8898c21c | 1450 | static void virtnet_clean_affinity(struct virtnet_info *vi, long hcpu) |
986a4f4d JW |
1451 | { |
1452 | int i; | |
1453 | ||
8898c21c WG |
1454 | if (vi->affinity_hint_set) { |
1455 | for (i = 0; i < vi->max_queue_pairs; i++) { | |
47be2479 WG |
1456 | virtqueue_set_affinity(vi->rq[i].vq, -1); |
1457 | virtqueue_set_affinity(vi->sq[i].vq, -1); | |
1458 | } | |
1459 | ||
8898c21c WG |
1460 | vi->affinity_hint_set = false; |
1461 | } | |
8898c21c | 1462 | } |
47be2479 | 1463 | |
8898c21c WG |
1464 | static void virtnet_set_affinity(struct virtnet_info *vi) |
1465 | { | |
1466 | int i; | |
1467 | int cpu; | |
986a4f4d JW |
1468 | |
1469 | /* In multiqueue mode, when the number of cpu is equal to the number of | |
1470 | * queue pairs, we let the queue pairs to be private to one cpu by | |
1471 | * setting the affinity hint to eliminate the contention. | |
1472 | */ | |
8898c21c WG |
1473 | if (vi->curr_queue_pairs == 1 || |
1474 | vi->max_queue_pairs != num_online_cpus()) { | |
1475 | virtnet_clean_affinity(vi, -1); | |
1476 | return; | |
986a4f4d JW |
1477 | } |
1478 | ||
8898c21c WG |
1479 | i = 0; |
1480 | for_each_online_cpu(cpu) { | |
986a4f4d JW |
1481 | virtqueue_set_affinity(vi->rq[i].vq, cpu); |
1482 | virtqueue_set_affinity(vi->sq[i].vq, cpu); | |
9bb8ca86 | 1483 | netif_set_xps_queue(vi->dev, cpumask_of(cpu), i); |
8898c21c | 1484 | i++; |
986a4f4d JW |
1485 | } |
1486 | ||
8898c21c | 1487 | vi->affinity_hint_set = true; |
986a4f4d JW |
1488 | } |
1489 | ||
8017c279 | 1490 | static int virtnet_cpu_online(unsigned int cpu, struct hlist_node *node) |
8de4b2f3 | 1491 | { |
8017c279 SAS |
1492 | struct virtnet_info *vi = hlist_entry_safe(node, struct virtnet_info, |
1493 | node); | |
1494 | virtnet_set_affinity(vi); | |
1495 | return 0; | |
1496 | } | |
8de4b2f3 | 1497 | |
8017c279 SAS |
1498 | static int virtnet_cpu_dead(unsigned int cpu, struct hlist_node *node) |
1499 | { | |
1500 | struct virtnet_info *vi = hlist_entry_safe(node, struct virtnet_info, | |
1501 | node_dead); | |
1502 | virtnet_set_affinity(vi); | |
1503 | return 0; | |
1504 | } | |
3ab098df | 1505 | |
8017c279 SAS |
1506 | static int virtnet_cpu_down_prep(unsigned int cpu, struct hlist_node *node) |
1507 | { | |
1508 | struct virtnet_info *vi = hlist_entry_safe(node, struct virtnet_info, | |
1509 | node); | |
1510 | ||
1511 | virtnet_clean_affinity(vi, cpu); | |
1512 | return 0; | |
1513 | } | |
1514 | ||
1515 | static enum cpuhp_state virtionet_online; | |
1516 | ||
1517 | static int virtnet_cpu_notif_add(struct virtnet_info *vi) | |
1518 | { | |
1519 | int ret; | |
1520 | ||
1521 | ret = cpuhp_state_add_instance_nocalls(virtionet_online, &vi->node); | |
1522 | if (ret) | |
1523 | return ret; | |
1524 | ret = cpuhp_state_add_instance_nocalls(CPUHP_VIRT_NET_DEAD, | |
1525 | &vi->node_dead); | |
1526 | if (!ret) | |
1527 | return ret; | |
1528 | cpuhp_state_remove_instance_nocalls(virtionet_online, &vi->node); | |
1529 | return ret; | |
1530 | } | |
1531 | ||
1532 | static void virtnet_cpu_notif_remove(struct virtnet_info *vi) | |
1533 | { | |
1534 | cpuhp_state_remove_instance_nocalls(virtionet_online, &vi->node); | |
1535 | cpuhp_state_remove_instance_nocalls(CPUHP_VIRT_NET_DEAD, | |
1536 | &vi->node_dead); | |
986a4f4d JW |
1537 | } |
1538 | ||
8f9f4668 RJ |
1539 | static void virtnet_get_ringparam(struct net_device *dev, |
1540 | struct ethtool_ringparam *ring) | |
1541 | { | |
1542 | struct virtnet_info *vi = netdev_priv(dev); | |
1543 | ||
986a4f4d JW |
1544 | ring->rx_max_pending = virtqueue_get_vring_size(vi->rq[0].vq); |
1545 | ring->tx_max_pending = virtqueue_get_vring_size(vi->sq[0].vq); | |
8f9f4668 RJ |
1546 | ring->rx_pending = ring->rx_max_pending; |
1547 | ring->tx_pending = ring->tx_max_pending; | |
8f9f4668 RJ |
1548 | } |
1549 | ||
66846048 RJ |
1550 | |
1551 | static void virtnet_get_drvinfo(struct net_device *dev, | |
1552 | struct ethtool_drvinfo *info) | |
1553 | { | |
1554 | struct virtnet_info *vi = netdev_priv(dev); | |
1555 | struct virtio_device *vdev = vi->vdev; | |
1556 | ||
1557 | strlcpy(info->driver, KBUILD_MODNAME, sizeof(info->driver)); | |
1558 | strlcpy(info->version, VIRTNET_DRIVER_VERSION, sizeof(info->version)); | |
1559 | strlcpy(info->bus_info, virtio_bus_name(vdev), sizeof(info->bus_info)); | |
1560 | ||
1561 | } | |
1562 | ||
d73bcd2c JW |
1563 | /* TODO: Eliminate OOO packets during switching */ |
1564 | static int virtnet_set_channels(struct net_device *dev, | |
1565 | struct ethtool_channels *channels) | |
1566 | { | |
1567 | struct virtnet_info *vi = netdev_priv(dev); | |
1568 | u16 queue_pairs = channels->combined_count; | |
1569 | int err; | |
1570 | ||
1571 | /* We don't support separate rx/tx channels. | |
1572 | * We don't allow setting 'other' channels. | |
1573 | */ | |
1574 | if (channels->rx_count || channels->tx_count || channels->other_count) | |
1575 | return -EINVAL; | |
1576 | ||
c18e9cd6 | 1577 | if (queue_pairs > vi->max_queue_pairs || queue_pairs == 0) |
d73bcd2c JW |
1578 | return -EINVAL; |
1579 | ||
f600b690 JF |
1580 | /* For now we don't support modifying channels while XDP is loaded |
1581 | * also when XDP is loaded all RX queues have XDP programs so we only | |
1582 | * need to check a single RX queue. | |
1583 | */ | |
1584 | if (vi->rq[0].xdp_prog) | |
1585 | return -EINVAL; | |
1586 | ||
47be2479 | 1587 | get_online_cpus(); |
d73bcd2c JW |
1588 | err = virtnet_set_queues(vi, queue_pairs); |
1589 | if (!err) { | |
1590 | netif_set_real_num_tx_queues(dev, queue_pairs); | |
1591 | netif_set_real_num_rx_queues(dev, queue_pairs); | |
1592 | ||
8898c21c | 1593 | virtnet_set_affinity(vi); |
d73bcd2c | 1594 | } |
47be2479 | 1595 | put_online_cpus(); |
d73bcd2c JW |
1596 | |
1597 | return err; | |
1598 | } | |
1599 | ||
1600 | static void virtnet_get_channels(struct net_device *dev, | |
1601 | struct ethtool_channels *channels) | |
1602 | { | |
1603 | struct virtnet_info *vi = netdev_priv(dev); | |
1604 | ||
1605 | channels->combined_count = vi->curr_queue_pairs; | |
1606 | channels->max_combined = vi->max_queue_pairs; | |
1607 | channels->max_other = 0; | |
1608 | channels->rx_count = 0; | |
1609 | channels->tx_count = 0; | |
1610 | channels->other_count = 0; | |
1611 | } | |
1612 | ||
16032be5 NA |
1613 | /* Check if the user is trying to change anything besides speed/duplex */ |
1614 | static bool virtnet_validate_ethtool_cmd(const struct ethtool_cmd *cmd) | |
1615 | { | |
1616 | struct ethtool_cmd diff1 = *cmd; | |
1617 | struct ethtool_cmd diff2 = {}; | |
1618 | ||
0cf3ace9 NA |
1619 | /* cmd is always set so we need to clear it, validate the port type |
1620 | * and also without autonegotiation we can ignore advertising | |
1621 | */ | |
16032be5 | 1622 | ethtool_cmd_speed_set(&diff1, 0); |
0cf3ace9 | 1623 | diff2.port = PORT_OTHER; |
16032be5 NA |
1624 | diff1.advertising = 0; |
1625 | diff1.duplex = 0; | |
16032be5 NA |
1626 | diff1.cmd = 0; |
1627 | ||
1628 | return !memcmp(&diff1, &diff2, sizeof(diff1)); | |
1629 | } | |
1630 | ||
1631 | static int virtnet_set_settings(struct net_device *dev, struct ethtool_cmd *cmd) | |
1632 | { | |
1633 | struct virtnet_info *vi = netdev_priv(dev); | |
1634 | u32 speed; | |
1635 | ||
1636 | speed = ethtool_cmd_speed(cmd); | |
1637 | /* don't allow custom speed and duplex */ | |
1638 | if (!ethtool_validate_speed(speed) || | |
1639 | !ethtool_validate_duplex(cmd->duplex) || | |
1640 | !virtnet_validate_ethtool_cmd(cmd)) | |
1641 | return -EINVAL; | |
1642 | vi->speed = speed; | |
1643 | vi->duplex = cmd->duplex; | |
1644 | ||
1645 | return 0; | |
1646 | } | |
1647 | ||
1648 | static int virtnet_get_settings(struct net_device *dev, struct ethtool_cmd *cmd) | |
1649 | { | |
1650 | struct virtnet_info *vi = netdev_priv(dev); | |
1651 | ||
1652 | ethtool_cmd_speed_set(cmd, vi->speed); | |
1653 | cmd->duplex = vi->duplex; | |
1654 | cmd->port = PORT_OTHER; | |
1655 | ||
1656 | return 0; | |
1657 | } | |
1658 | ||
1659 | static void virtnet_init_settings(struct net_device *dev) | |
1660 | { | |
1661 | struct virtnet_info *vi = netdev_priv(dev); | |
1662 | ||
1663 | vi->speed = SPEED_UNKNOWN; | |
1664 | vi->duplex = DUPLEX_UNKNOWN; | |
1665 | } | |
1666 | ||
0fc0b732 | 1667 | static const struct ethtool_ops virtnet_ethtool_ops = { |
66846048 | 1668 | .get_drvinfo = virtnet_get_drvinfo, |
9f4d26d0 | 1669 | .get_link = ethtool_op_get_link, |
8f9f4668 | 1670 | .get_ringparam = virtnet_get_ringparam, |
d73bcd2c JW |
1671 | .set_channels = virtnet_set_channels, |
1672 | .get_channels = virtnet_get_channels, | |
074c3582 | 1673 | .get_ts_info = ethtool_op_get_ts_info, |
16032be5 NA |
1674 | .get_settings = virtnet_get_settings, |
1675 | .set_settings = virtnet_set_settings, | |
a9ea3fc6 HX |
1676 | }; |
1677 | ||
f600b690 JF |
1678 | static int virtnet_xdp_set(struct net_device *dev, struct bpf_prog *prog) |
1679 | { | |
1680 | unsigned long int max_sz = PAGE_SIZE - sizeof(struct padded_vnet_hdr); | |
1681 | struct virtnet_info *vi = netdev_priv(dev); | |
1682 | struct bpf_prog *old_prog; | |
672aafd5 JF |
1683 | u16 xdp_qp = 0, curr_qp; |
1684 | int i, err; | |
f600b690 JF |
1685 | |
1686 | if (virtio_has_feature(vi->vdev, VIRTIO_NET_F_GUEST_TSO4) || | |
92502fe8 JW |
1687 | virtio_has_feature(vi->vdev, VIRTIO_NET_F_GUEST_TSO6) || |
1688 | virtio_has_feature(vi->vdev, VIRTIO_NET_F_GUEST_ECN) || | |
1689 | virtio_has_feature(vi->vdev, VIRTIO_NET_F_GUEST_UFO)) { | |
f600b690 JF |
1690 | netdev_warn(dev, "can't set XDP while host is implementing LRO, disable LRO first\n"); |
1691 | return -EOPNOTSUPP; | |
1692 | } | |
1693 | ||
1694 | if (vi->mergeable_rx_bufs && !vi->any_header_sg) { | |
1695 | netdev_warn(dev, "XDP expects header/data in single page, any_header_sg required\n"); | |
1696 | return -EINVAL; | |
1697 | } | |
1698 | ||
1699 | if (dev->mtu > max_sz) { | |
1700 | netdev_warn(dev, "XDP requires MTU less than %lu\n", max_sz); | |
1701 | return -EINVAL; | |
1702 | } | |
1703 | ||
672aafd5 JF |
1704 | curr_qp = vi->curr_queue_pairs - vi->xdp_queue_pairs; |
1705 | if (prog) | |
1706 | xdp_qp = nr_cpu_ids; | |
1707 | ||
1708 | /* XDP requires extra queues for XDP_TX */ | |
1709 | if (curr_qp + xdp_qp > vi->max_queue_pairs) { | |
1710 | netdev_warn(dev, "request %i queues but max is %i\n", | |
1711 | curr_qp + xdp_qp, vi->max_queue_pairs); | |
1712 | return -ENOMEM; | |
1713 | } | |
1714 | ||
1715 | err = virtnet_set_queues(vi, curr_qp + xdp_qp); | |
1716 | if (err) { | |
1717 | dev_warn(&dev->dev, "XDP Device queue allocation failure.\n"); | |
1718 | return err; | |
1719 | } | |
1720 | ||
f600b690 JF |
1721 | if (prog) { |
1722 | prog = bpf_prog_add(prog, vi->max_queue_pairs - 1); | |
672aafd5 JF |
1723 | if (IS_ERR(prog)) { |
1724 | virtnet_set_queues(vi, curr_qp); | |
f600b690 | 1725 | return PTR_ERR(prog); |
672aafd5 | 1726 | } |
f600b690 JF |
1727 | } |
1728 | ||
672aafd5 JF |
1729 | vi->xdp_queue_pairs = xdp_qp; |
1730 | netif_set_real_num_rx_queues(dev, curr_qp + xdp_qp); | |
1731 | ||
f600b690 JF |
1732 | for (i = 0; i < vi->max_queue_pairs; i++) { |
1733 | old_prog = rtnl_dereference(vi->rq[i].xdp_prog); | |
1734 | rcu_assign_pointer(vi->rq[i].xdp_prog, prog); | |
1735 | if (old_prog) | |
1736 | bpf_prog_put(old_prog); | |
1737 | } | |
1738 | ||
1739 | return 0; | |
1740 | } | |
1741 | ||
1742 | static bool virtnet_xdp_query(struct net_device *dev) | |
1743 | { | |
1744 | struct virtnet_info *vi = netdev_priv(dev); | |
1745 | int i; | |
1746 | ||
1747 | for (i = 0; i < vi->max_queue_pairs; i++) { | |
1748 | if (vi->rq[i].xdp_prog) | |
1749 | return true; | |
1750 | } | |
1751 | return false; | |
1752 | } | |
1753 | ||
1754 | static int virtnet_xdp(struct net_device *dev, struct netdev_xdp *xdp) | |
1755 | { | |
1756 | switch (xdp->command) { | |
1757 | case XDP_SETUP_PROG: | |
1758 | return virtnet_xdp_set(dev, xdp->prog); | |
1759 | case XDP_QUERY_PROG: | |
1760 | xdp->prog_attached = virtnet_xdp_query(dev); | |
1761 | return 0; | |
1762 | default: | |
1763 | return -EINVAL; | |
1764 | } | |
1765 | } | |
1766 | ||
76288b4e SH |
1767 | static const struct net_device_ops virtnet_netdev = { |
1768 | .ndo_open = virtnet_open, | |
1769 | .ndo_stop = virtnet_close, | |
1770 | .ndo_start_xmit = start_xmit, | |
1771 | .ndo_validate_addr = eth_validate_addr, | |
9c46f6d4 | 1772 | .ndo_set_mac_address = virtnet_set_mac_address, |
2af7698e | 1773 | .ndo_set_rx_mode = virtnet_set_rx_mode, |
3fa2a1df | 1774 | .ndo_get_stats64 = virtnet_stats, |
1824a989 AW |
1775 | .ndo_vlan_rx_add_vid = virtnet_vlan_rx_add_vid, |
1776 | .ndo_vlan_rx_kill_vid = virtnet_vlan_rx_kill_vid, | |
76288b4e SH |
1777 | #ifdef CONFIG_NET_POLL_CONTROLLER |
1778 | .ndo_poll_controller = virtnet_netpoll, | |
1779 | #endif | |
91815639 JW |
1780 | #ifdef CONFIG_NET_RX_BUSY_POLL |
1781 | .ndo_busy_poll = virtnet_busy_poll, | |
1782 | #endif | |
f600b690 | 1783 | .ndo_xdp = virtnet_xdp, |
76288b4e SH |
1784 | }; |
1785 | ||
586d17c5 | 1786 | static void virtnet_config_changed_work(struct work_struct *work) |
9f4d26d0 | 1787 | { |
586d17c5 JW |
1788 | struct virtnet_info *vi = |
1789 | container_of(work, struct virtnet_info, config_work); | |
9f4d26d0 MM |
1790 | u16 v; |
1791 | ||
855e0c52 RR |
1792 | if (virtio_cread_feature(vi->vdev, VIRTIO_NET_F_STATUS, |
1793 | struct virtio_net_config, status, &v) < 0) | |
507613bf | 1794 | return; |
586d17c5 JW |
1795 | |
1796 | if (v & VIRTIO_NET_S_ANNOUNCE) { | |
ee89bab1 | 1797 | netdev_notify_peers(vi->dev); |
586d17c5 JW |
1798 | virtnet_ack_link_announce(vi); |
1799 | } | |
9f4d26d0 MM |
1800 | |
1801 | /* Ignore unknown (future) status bits */ | |
1802 | v &= VIRTIO_NET_S_LINK_UP; | |
1803 | ||
1804 | if (vi->status == v) | |
507613bf | 1805 | return; |
9f4d26d0 MM |
1806 | |
1807 | vi->status = v; | |
1808 | ||
1809 | if (vi->status & VIRTIO_NET_S_LINK_UP) { | |
1810 | netif_carrier_on(vi->dev); | |
986a4f4d | 1811 | netif_tx_wake_all_queues(vi->dev); |
9f4d26d0 MM |
1812 | } else { |
1813 | netif_carrier_off(vi->dev); | |
986a4f4d | 1814 | netif_tx_stop_all_queues(vi->dev); |
9f4d26d0 MM |
1815 | } |
1816 | } | |
1817 | ||
1818 | static void virtnet_config_changed(struct virtio_device *vdev) | |
1819 | { | |
1820 | struct virtnet_info *vi = vdev->priv; | |
1821 | ||
3b07e9ca | 1822 | schedule_work(&vi->config_work); |
9f4d26d0 MM |
1823 | } |
1824 | ||
986a4f4d JW |
1825 | static void virtnet_free_queues(struct virtnet_info *vi) |
1826 | { | |
d4fb84ee AV |
1827 | int i; |
1828 | ||
ab3971b1 JW |
1829 | for (i = 0; i < vi->max_queue_pairs; i++) { |
1830 | napi_hash_del(&vi->rq[i].napi); | |
d4fb84ee | 1831 | netif_napi_del(&vi->rq[i].napi); |
ab3971b1 | 1832 | } |
d4fb84ee | 1833 | |
963abe5c ED |
1834 | /* We called napi_hash_del() before netif_napi_del(), |
1835 | * we need to respect an RCU grace period before freeing vi->rq | |
1836 | */ | |
1837 | synchronize_net(); | |
1838 | ||
986a4f4d JW |
1839 | kfree(vi->rq); |
1840 | kfree(vi->sq); | |
1841 | } | |
1842 | ||
1843 | static void free_receive_bufs(struct virtnet_info *vi) | |
1844 | { | |
f600b690 | 1845 | struct bpf_prog *old_prog; |
986a4f4d JW |
1846 | int i; |
1847 | ||
f600b690 | 1848 | rtnl_lock(); |
986a4f4d JW |
1849 | for (i = 0; i < vi->max_queue_pairs; i++) { |
1850 | while (vi->rq[i].pages) | |
1851 | __free_pages(get_a_page(&vi->rq[i], GFP_KERNEL), 0); | |
f600b690 JF |
1852 | |
1853 | old_prog = rtnl_dereference(vi->rq[i].xdp_prog); | |
1854 | RCU_INIT_POINTER(vi->rq[i].xdp_prog, NULL); | |
1855 | if (old_prog) | |
1856 | bpf_prog_put(old_prog); | |
986a4f4d | 1857 | } |
f600b690 | 1858 | rtnl_unlock(); |
986a4f4d JW |
1859 | } |
1860 | ||
fb51879d MD |
1861 | static void free_receive_page_frags(struct virtnet_info *vi) |
1862 | { | |
1863 | int i; | |
1864 | for (i = 0; i < vi->max_queue_pairs; i++) | |
1865 | if (vi->rq[i].alloc_frag.page) | |
1866 | put_page(vi->rq[i].alloc_frag.page); | |
1867 | } | |
1868 | ||
56434a01 JF |
1869 | static bool is_xdp_queue(struct virtnet_info *vi, int q) |
1870 | { | |
1871 | if (q < (vi->curr_queue_pairs - vi->xdp_queue_pairs)) | |
1872 | return false; | |
1873 | else if (q < vi->curr_queue_pairs) | |
1874 | return true; | |
1875 | else | |
1876 | return false; | |
1877 | } | |
1878 | ||
986a4f4d JW |
1879 | static void free_unused_bufs(struct virtnet_info *vi) |
1880 | { | |
1881 | void *buf; | |
1882 | int i; | |
1883 | ||
1884 | for (i = 0; i < vi->max_queue_pairs; i++) { | |
1885 | struct virtqueue *vq = vi->sq[i].vq; | |
56434a01 JF |
1886 | while ((buf = virtqueue_detach_unused_buf(vq)) != NULL) { |
1887 | if (!is_xdp_queue(vi, i)) | |
1888 | dev_kfree_skb(buf); | |
1889 | else | |
1890 | put_page(virt_to_head_page(buf)); | |
1891 | } | |
986a4f4d JW |
1892 | } |
1893 | ||
1894 | for (i = 0; i < vi->max_queue_pairs; i++) { | |
1895 | struct virtqueue *vq = vi->rq[i].vq; | |
1896 | ||
1897 | while ((buf = virtqueue_detach_unused_buf(vq)) != NULL) { | |
ab7db917 MD |
1898 | if (vi->mergeable_rx_bufs) { |
1899 | unsigned long ctx = (unsigned long)buf; | |
1900 | void *base = mergeable_ctx_to_buf_address(ctx); | |
1901 | put_page(virt_to_head_page(base)); | |
1902 | } else if (vi->big_packets) { | |
fa9fac17 | 1903 | give_pages(&vi->rq[i], buf); |
ab7db917 | 1904 | } else { |
986a4f4d | 1905 | dev_kfree_skb(buf); |
ab7db917 | 1906 | } |
986a4f4d | 1907 | } |
986a4f4d JW |
1908 | } |
1909 | } | |
1910 | ||
e9d7417b JW |
1911 | static void virtnet_del_vqs(struct virtnet_info *vi) |
1912 | { | |
1913 | struct virtio_device *vdev = vi->vdev; | |
1914 | ||
8898c21c | 1915 | virtnet_clean_affinity(vi, -1); |
986a4f4d | 1916 | |
e9d7417b | 1917 | vdev->config->del_vqs(vdev); |
986a4f4d JW |
1918 | |
1919 | virtnet_free_queues(vi); | |
e9d7417b JW |
1920 | } |
1921 | ||
986a4f4d | 1922 | static int virtnet_find_vqs(struct virtnet_info *vi) |
3f9c10b0 | 1923 | { |
986a4f4d JW |
1924 | vq_callback_t **callbacks; |
1925 | struct virtqueue **vqs; | |
1926 | int ret = -ENOMEM; | |
1927 | int i, total_vqs; | |
1928 | const char **names; | |
1929 | ||
1930 | /* We expect 1 RX virtqueue followed by 1 TX virtqueue, followed by | |
1931 | * possible N-1 RX/TX queue pairs used in multiqueue mode, followed by | |
1932 | * possible control vq. | |
1933 | */ | |
1934 | total_vqs = vi->max_queue_pairs * 2 + | |
1935 | virtio_has_feature(vi->vdev, VIRTIO_NET_F_CTRL_VQ); | |
1936 | ||
1937 | /* Allocate space for find_vqs parameters */ | |
1938 | vqs = kzalloc(total_vqs * sizeof(*vqs), GFP_KERNEL); | |
1939 | if (!vqs) | |
1940 | goto err_vq; | |
1941 | callbacks = kmalloc(total_vqs * sizeof(*callbacks), GFP_KERNEL); | |
1942 | if (!callbacks) | |
1943 | goto err_callback; | |
1944 | names = kmalloc(total_vqs * sizeof(*names), GFP_KERNEL); | |
1945 | if (!names) | |
1946 | goto err_names; | |
1947 | ||
1948 | /* Parameters for control virtqueue, if any */ | |
1949 | if (vi->has_cvq) { | |
1950 | callbacks[total_vqs - 1] = NULL; | |
1951 | names[total_vqs - 1] = "control"; | |
1952 | } | |
3f9c10b0 | 1953 | |
986a4f4d JW |
1954 | /* Allocate/initialize parameters for send/receive virtqueues */ |
1955 | for (i = 0; i < vi->max_queue_pairs; i++) { | |
1956 | callbacks[rxq2vq(i)] = skb_recv_done; | |
1957 | callbacks[txq2vq(i)] = skb_xmit_done; | |
1958 | sprintf(vi->rq[i].name, "input.%d", i); | |
1959 | sprintf(vi->sq[i].name, "output.%d", i); | |
1960 | names[rxq2vq(i)] = vi->rq[i].name; | |
1961 | names[txq2vq(i)] = vi->sq[i].name; | |
1962 | } | |
3f9c10b0 | 1963 | |
986a4f4d JW |
1964 | ret = vi->vdev->config->find_vqs(vi->vdev, total_vqs, vqs, callbacks, |
1965 | names); | |
1966 | if (ret) | |
1967 | goto err_find; | |
3f9c10b0 | 1968 | |
986a4f4d JW |
1969 | if (vi->has_cvq) { |
1970 | vi->cvq = vqs[total_vqs - 1]; | |
3f9c10b0 | 1971 | if (virtio_has_feature(vi->vdev, VIRTIO_NET_F_CTRL_VLAN)) |
f646968f | 1972 | vi->dev->features |= NETIF_F_HW_VLAN_CTAG_FILTER; |
3f9c10b0 | 1973 | } |
986a4f4d JW |
1974 | |
1975 | for (i = 0; i < vi->max_queue_pairs; i++) { | |
1976 | vi->rq[i].vq = vqs[rxq2vq(i)]; | |
1977 | vi->sq[i].vq = vqs[txq2vq(i)]; | |
1978 | } | |
1979 | ||
1980 | kfree(names); | |
1981 | kfree(callbacks); | |
1982 | kfree(vqs); | |
1983 | ||
3f9c10b0 | 1984 | return 0; |
986a4f4d JW |
1985 | |
1986 | err_find: | |
1987 | kfree(names); | |
1988 | err_names: | |
1989 | kfree(callbacks); | |
1990 | err_callback: | |
1991 | kfree(vqs); | |
1992 | err_vq: | |
1993 | return ret; | |
1994 | } | |
1995 | ||
1996 | static int virtnet_alloc_queues(struct virtnet_info *vi) | |
1997 | { | |
1998 | int i; | |
1999 | ||
2000 | vi->sq = kzalloc(sizeof(*vi->sq) * vi->max_queue_pairs, GFP_KERNEL); | |
2001 | if (!vi->sq) | |
2002 | goto err_sq; | |
2003 | vi->rq = kzalloc(sizeof(*vi->rq) * vi->max_queue_pairs, GFP_KERNEL); | |
008d4278 | 2004 | if (!vi->rq) |
986a4f4d JW |
2005 | goto err_rq; |
2006 | ||
2007 | INIT_DELAYED_WORK(&vi->refill, refill_work); | |
2008 | for (i = 0; i < vi->max_queue_pairs; i++) { | |
2009 | vi->rq[i].pages = NULL; | |
2010 | netif_napi_add(vi->dev, &vi->rq[i].napi, virtnet_poll, | |
2011 | napi_weight); | |
2012 | ||
2013 | sg_init_table(vi->rq[i].sg, ARRAY_SIZE(vi->rq[i].sg)); | |
5377d758 | 2014 | ewma_pkt_len_init(&vi->rq[i].mrg_avg_pkt_len); |
986a4f4d JW |
2015 | sg_init_table(vi->sq[i].sg, ARRAY_SIZE(vi->sq[i].sg)); |
2016 | } | |
2017 | ||
2018 | return 0; | |
2019 | ||
2020 | err_rq: | |
2021 | kfree(vi->sq); | |
2022 | err_sq: | |
2023 | return -ENOMEM; | |
2024 | } | |
2025 | ||
2026 | static int init_vqs(struct virtnet_info *vi) | |
2027 | { | |
2028 | int ret; | |
2029 | ||
2030 | /* Allocate send & receive queues */ | |
2031 | ret = virtnet_alloc_queues(vi); | |
2032 | if (ret) | |
2033 | goto err; | |
2034 | ||
2035 | ret = virtnet_find_vqs(vi); | |
2036 | if (ret) | |
2037 | goto err_free; | |
2038 | ||
47be2479 | 2039 | get_online_cpus(); |
8898c21c | 2040 | virtnet_set_affinity(vi); |
47be2479 WG |
2041 | put_online_cpus(); |
2042 | ||
986a4f4d JW |
2043 | return 0; |
2044 | ||
2045 | err_free: | |
2046 | virtnet_free_queues(vi); | |
2047 | err: | |
2048 | return ret; | |
3f9c10b0 AS |
2049 | } |
2050 | ||
fbf28d78 MD |
2051 | #ifdef CONFIG_SYSFS |
2052 | static ssize_t mergeable_rx_buffer_size_show(struct netdev_rx_queue *queue, | |
2053 | struct rx_queue_attribute *attribute, char *buf) | |
2054 | { | |
2055 | struct virtnet_info *vi = netdev_priv(queue->dev); | |
2056 | unsigned int queue_index = get_netdev_rx_queue_index(queue); | |
5377d758 | 2057 | struct ewma_pkt_len *avg; |
fbf28d78 MD |
2058 | |
2059 | BUG_ON(queue_index >= vi->max_queue_pairs); | |
2060 | avg = &vi->rq[queue_index].mrg_avg_pkt_len; | |
2061 | return sprintf(buf, "%u\n", get_mergeable_buf_len(avg)); | |
2062 | } | |
2063 | ||
2064 | static struct rx_queue_attribute mergeable_rx_buffer_size_attribute = | |
2065 | __ATTR_RO(mergeable_rx_buffer_size); | |
2066 | ||
2067 | static struct attribute *virtio_net_mrg_rx_attrs[] = { | |
2068 | &mergeable_rx_buffer_size_attribute.attr, | |
2069 | NULL | |
2070 | }; | |
2071 | ||
2072 | static const struct attribute_group virtio_net_mrg_rx_group = { | |
2073 | .name = "virtio_net", | |
2074 | .attrs = virtio_net_mrg_rx_attrs | |
2075 | }; | |
2076 | #endif | |
2077 | ||
892d6eb1 JW |
2078 | static bool virtnet_fail_on_feature(struct virtio_device *vdev, |
2079 | unsigned int fbit, | |
2080 | const char *fname, const char *dname) | |
2081 | { | |
2082 | if (!virtio_has_feature(vdev, fbit)) | |
2083 | return false; | |
2084 | ||
2085 | dev_err(&vdev->dev, "device advertises feature %s but not %s", | |
2086 | fname, dname); | |
2087 | ||
2088 | return true; | |
2089 | } | |
2090 | ||
2091 | #define VIRTNET_FAIL_ON(vdev, fbit, dbit) \ | |
2092 | virtnet_fail_on_feature(vdev, fbit, #fbit, dbit) | |
2093 | ||
2094 | static bool virtnet_validate_features(struct virtio_device *vdev) | |
2095 | { | |
2096 | if (!virtio_has_feature(vdev, VIRTIO_NET_F_CTRL_VQ) && | |
2097 | (VIRTNET_FAIL_ON(vdev, VIRTIO_NET_F_CTRL_RX, | |
2098 | "VIRTIO_NET_F_CTRL_VQ") || | |
2099 | VIRTNET_FAIL_ON(vdev, VIRTIO_NET_F_CTRL_VLAN, | |
2100 | "VIRTIO_NET_F_CTRL_VQ") || | |
2101 | VIRTNET_FAIL_ON(vdev, VIRTIO_NET_F_GUEST_ANNOUNCE, | |
2102 | "VIRTIO_NET_F_CTRL_VQ") || | |
2103 | VIRTNET_FAIL_ON(vdev, VIRTIO_NET_F_MQ, "VIRTIO_NET_F_CTRL_VQ") || | |
2104 | VIRTNET_FAIL_ON(vdev, VIRTIO_NET_F_CTRL_MAC_ADDR, | |
2105 | "VIRTIO_NET_F_CTRL_VQ"))) { | |
2106 | return false; | |
2107 | } | |
2108 | ||
2109 | return true; | |
2110 | } | |
2111 | ||
d0c2c997 JW |
2112 | #define MIN_MTU ETH_MIN_MTU |
2113 | #define MAX_MTU ETH_MAX_MTU | |
2114 | ||
296f96fc RR |
2115 | static int virtnet_probe(struct virtio_device *vdev) |
2116 | { | |
986a4f4d | 2117 | int i, err; |
296f96fc RR |
2118 | struct net_device *dev; |
2119 | struct virtnet_info *vi; | |
986a4f4d | 2120 | u16 max_queue_pairs; |
14de9d11 | 2121 | int mtu; |
986a4f4d | 2122 | |
6ba42248 MT |
2123 | if (!vdev->config->get) { |
2124 | dev_err(&vdev->dev, "%s failure: config access disabled\n", | |
2125 | __func__); | |
2126 | return -EINVAL; | |
2127 | } | |
2128 | ||
892d6eb1 JW |
2129 | if (!virtnet_validate_features(vdev)) |
2130 | return -EINVAL; | |
2131 | ||
986a4f4d | 2132 | /* Find if host supports multiqueue virtio_net device */ |
855e0c52 RR |
2133 | err = virtio_cread_feature(vdev, VIRTIO_NET_F_MQ, |
2134 | struct virtio_net_config, | |
2135 | max_virtqueue_pairs, &max_queue_pairs); | |
986a4f4d JW |
2136 | |
2137 | /* We need at least 2 queue's */ | |
2138 | if (err || max_queue_pairs < VIRTIO_NET_CTRL_MQ_VQ_PAIRS_MIN || | |
2139 | max_queue_pairs > VIRTIO_NET_CTRL_MQ_VQ_PAIRS_MAX || | |
2140 | !virtio_has_feature(vdev, VIRTIO_NET_F_CTRL_VQ)) | |
2141 | max_queue_pairs = 1; | |
296f96fc RR |
2142 | |
2143 | /* Allocate ourselves a network device with room for our info */ | |
986a4f4d | 2144 | dev = alloc_etherdev_mq(sizeof(struct virtnet_info), max_queue_pairs); |
296f96fc RR |
2145 | if (!dev) |
2146 | return -ENOMEM; | |
2147 | ||
2148 | /* Set up network device as normal. */ | |
f2f2c8b4 | 2149 | dev->priv_flags |= IFF_UNICAST_FLT | IFF_LIVE_ADDR_CHANGE; |
76288b4e | 2150 | dev->netdev_ops = &virtnet_netdev; |
296f96fc | 2151 | dev->features = NETIF_F_HIGHDMA; |
3fa2a1df | 2152 | |
7ad24ea4 | 2153 | dev->ethtool_ops = &virtnet_ethtool_ops; |
296f96fc RR |
2154 | SET_NETDEV_DEV(dev, &vdev->dev); |
2155 | ||
2156 | /* Do we support "hardware" checksums? */ | |
98e778c9 | 2157 | if (virtio_has_feature(vdev, VIRTIO_NET_F_CSUM)) { |
296f96fc | 2158 | /* This opens up the world of extra features. */ |
48900cb6 | 2159 | dev->hw_features |= NETIF_F_HW_CSUM | NETIF_F_SG; |
98e778c9 | 2160 | if (csum) |
48900cb6 | 2161 | dev->features |= NETIF_F_HW_CSUM | NETIF_F_SG; |
98e778c9 MM |
2162 | |
2163 | if (virtio_has_feature(vdev, VIRTIO_NET_F_GSO)) { | |
e3e3c423 | 2164 | dev->hw_features |= NETIF_F_TSO | NETIF_F_UFO |
34a48579 RR |
2165 | | NETIF_F_TSO_ECN | NETIF_F_TSO6; |
2166 | } | |
5539ae96 | 2167 | /* Individual feature bits: what can host handle? */ |
98e778c9 MM |
2168 | if (virtio_has_feature(vdev, VIRTIO_NET_F_HOST_TSO4)) |
2169 | dev->hw_features |= NETIF_F_TSO; | |
2170 | if (virtio_has_feature(vdev, VIRTIO_NET_F_HOST_TSO6)) | |
2171 | dev->hw_features |= NETIF_F_TSO6; | |
2172 | if (virtio_has_feature(vdev, VIRTIO_NET_F_HOST_ECN)) | |
2173 | dev->hw_features |= NETIF_F_TSO_ECN; | |
e3e3c423 VY |
2174 | if (virtio_has_feature(vdev, VIRTIO_NET_F_HOST_UFO)) |
2175 | dev->hw_features |= NETIF_F_UFO; | |
98e778c9 | 2176 | |
41f2f127 JW |
2177 | dev->features |= NETIF_F_GSO_ROBUST; |
2178 | ||
98e778c9 | 2179 | if (gso) |
e3e3c423 | 2180 | dev->features |= dev->hw_features & (NETIF_F_ALL_TSO|NETIF_F_UFO); |
98e778c9 | 2181 | /* (!csum && gso) case will be fixed by register_netdev() */ |
296f96fc | 2182 | } |
4f49129b TH |
2183 | if (virtio_has_feature(vdev, VIRTIO_NET_F_GUEST_CSUM)) |
2184 | dev->features |= NETIF_F_RXCSUM; | |
296f96fc | 2185 | |
4fda8302 JW |
2186 | dev->vlan_features = dev->features; |
2187 | ||
d0c2c997 JW |
2188 | /* MTU range: 68 - 65535 */ |
2189 | dev->min_mtu = MIN_MTU; | |
2190 | dev->max_mtu = MAX_MTU; | |
2191 | ||
296f96fc | 2192 | /* Configuration may specify what MAC to use. Otherwise random. */ |
855e0c52 RR |
2193 | if (virtio_has_feature(vdev, VIRTIO_NET_F_MAC)) |
2194 | virtio_cread_bytes(vdev, | |
2195 | offsetof(struct virtio_net_config, mac), | |
2196 | dev->dev_addr, dev->addr_len); | |
2197 | else | |
f2cedb63 | 2198 | eth_hw_addr_random(dev); |
296f96fc RR |
2199 | |
2200 | /* Set up our device-specific information */ | |
2201 | vi = netdev_priv(dev); | |
296f96fc RR |
2202 | vi->dev = dev; |
2203 | vi->vdev = vdev; | |
d9d5dcc8 | 2204 | vdev->priv = vi; |
3fa2a1df | 2205 | vi->stats = alloc_percpu(struct virtnet_stats); |
2206 | err = -ENOMEM; | |
2207 | if (vi->stats == NULL) | |
2208 | goto free; | |
2209 | ||
827da44c JS |
2210 | for_each_possible_cpu(i) { |
2211 | struct virtnet_stats *virtnet_stats; | |
2212 | virtnet_stats = per_cpu_ptr(vi->stats, i); | |
2213 | u64_stats_init(&virtnet_stats->tx_syncp); | |
2214 | u64_stats_init(&virtnet_stats->rx_syncp); | |
2215 | } | |
2216 | ||
586d17c5 | 2217 | INIT_WORK(&vi->config_work, virtnet_config_changed_work); |
296f96fc | 2218 | |
97402b96 | 2219 | /* If we can receive ANY GSO packets, we must allocate large ones. */ |
8e95a202 JP |
2220 | if (virtio_has_feature(vdev, VIRTIO_NET_F_GUEST_TSO4) || |
2221 | virtio_has_feature(vdev, VIRTIO_NET_F_GUEST_TSO6) || | |
e3e3c423 VY |
2222 | virtio_has_feature(vdev, VIRTIO_NET_F_GUEST_ECN) || |
2223 | virtio_has_feature(vdev, VIRTIO_NET_F_GUEST_UFO)) | |
97402b96 HX |
2224 | vi->big_packets = true; |
2225 | ||
3f2c31d9 MM |
2226 | if (virtio_has_feature(vdev, VIRTIO_NET_F_MRG_RXBUF)) |
2227 | vi->mergeable_rx_bufs = true; | |
2228 | ||
d04302b3 MT |
2229 | if (virtio_has_feature(vdev, VIRTIO_NET_F_MRG_RXBUF) || |
2230 | virtio_has_feature(vdev, VIRTIO_F_VERSION_1)) | |
012873d0 MT |
2231 | vi->hdr_len = sizeof(struct virtio_net_hdr_mrg_rxbuf); |
2232 | else | |
2233 | vi->hdr_len = sizeof(struct virtio_net_hdr); | |
2234 | ||
75993300 MT |
2235 | if (virtio_has_feature(vdev, VIRTIO_F_ANY_LAYOUT) || |
2236 | virtio_has_feature(vdev, VIRTIO_F_VERSION_1)) | |
e7428e95 MT |
2237 | vi->any_header_sg = true; |
2238 | ||
986a4f4d JW |
2239 | if (virtio_has_feature(vdev, VIRTIO_NET_F_CTRL_VQ)) |
2240 | vi->has_cvq = true; | |
2241 | ||
14de9d11 AC |
2242 | if (virtio_has_feature(vdev, VIRTIO_NET_F_MTU)) { |
2243 | mtu = virtio_cread16(vdev, | |
2244 | offsetof(struct virtio_net_config, | |
2245 | mtu)); | |
93a205ee | 2246 | if (mtu < dev->min_mtu) { |
14de9d11 | 2247 | __virtio_clear_bit(vdev, VIRTIO_NET_F_MTU); |
93a205ee | 2248 | } else { |
d0c2c997 | 2249 | dev->mtu = mtu; |
93a205ee AC |
2250 | dev->max_mtu = mtu; |
2251 | } | |
14de9d11 AC |
2252 | } |
2253 | ||
012873d0 MT |
2254 | if (vi->any_header_sg) |
2255 | dev->needed_headroom = vi->hdr_len; | |
6ebbc1a6 | 2256 | |
44900010 JW |
2257 | /* Enable multiqueue by default */ |
2258 | if (num_online_cpus() >= max_queue_pairs) | |
2259 | vi->curr_queue_pairs = max_queue_pairs; | |
2260 | else | |
2261 | vi->curr_queue_pairs = num_online_cpus(); | |
986a4f4d JW |
2262 | vi->max_queue_pairs = max_queue_pairs; |
2263 | ||
2264 | /* Allocate/initialize the rx/tx queues, and invoke find_vqs */ | |
3f9c10b0 | 2265 | err = init_vqs(vi); |
d2a7ddda | 2266 | if (err) |
9bb8ca86 | 2267 | goto free_stats; |
296f96fc | 2268 | |
fbf28d78 MD |
2269 | #ifdef CONFIG_SYSFS |
2270 | if (vi->mergeable_rx_bufs) | |
2271 | dev->sysfs_rx_queue_group = &virtio_net_mrg_rx_group; | |
2272 | #endif | |
0f13b66b ZYW |
2273 | netif_set_real_num_tx_queues(dev, vi->curr_queue_pairs); |
2274 | netif_set_real_num_rx_queues(dev, vi->curr_queue_pairs); | |
986a4f4d | 2275 | |
16032be5 NA |
2276 | virtnet_init_settings(dev); |
2277 | ||
296f96fc RR |
2278 | err = register_netdev(dev); |
2279 | if (err) { | |
2280 | pr_debug("virtio_net: registering device failed\n"); | |
d2a7ddda | 2281 | goto free_vqs; |
296f96fc | 2282 | } |
b3369c1f | 2283 | |
4baf1e33 MT |
2284 | virtio_device_ready(vdev); |
2285 | ||
8017c279 | 2286 | err = virtnet_cpu_notif_add(vi); |
8de4b2f3 WG |
2287 | if (err) { |
2288 | pr_debug("virtio_net: registering cpu notifier failed\n"); | |
f00e35e2 | 2289 | goto free_unregister_netdev; |
8de4b2f3 WG |
2290 | } |
2291 | ||
a220871b JW |
2292 | rtnl_lock(); |
2293 | virtnet_set_queues(vi, vi->curr_queue_pairs); | |
2294 | rtnl_unlock(); | |
44900010 | 2295 | |
167c25e4 JW |
2296 | /* Assume link up if device can't report link status, |
2297 | otherwise get link status from config. */ | |
2298 | if (virtio_has_feature(vi->vdev, VIRTIO_NET_F_STATUS)) { | |
2299 | netif_carrier_off(dev); | |
3b07e9ca | 2300 | schedule_work(&vi->config_work); |
167c25e4 JW |
2301 | } else { |
2302 | vi->status = VIRTIO_NET_S_LINK_UP; | |
2303 | netif_carrier_on(dev); | |
2304 | } | |
9f4d26d0 | 2305 | |
986a4f4d JW |
2306 | pr_debug("virtnet: registered device %s with %d RX and TX vq's\n", |
2307 | dev->name, max_queue_pairs); | |
2308 | ||
296f96fc RR |
2309 | return 0; |
2310 | ||
f00e35e2 | 2311 | free_unregister_netdev: |
02465555 MT |
2312 | vi->vdev->config->reset(vdev); |
2313 | ||
b3369c1f | 2314 | unregister_netdev(dev); |
d2a7ddda | 2315 | free_vqs: |
986a4f4d | 2316 | cancel_delayed_work_sync(&vi->refill); |
fb51879d | 2317 | free_receive_page_frags(vi); |
e9d7417b | 2318 | virtnet_del_vqs(vi); |
3fa2a1df | 2319 | free_stats: |
2320 | free_percpu(vi->stats); | |
296f96fc RR |
2321 | free: |
2322 | free_netdev(dev); | |
2323 | return err; | |
2324 | } | |
2325 | ||
04486ed0 | 2326 | static void remove_vq_common(struct virtnet_info *vi) |
296f96fc | 2327 | { |
04486ed0 | 2328 | vi->vdev->config->reset(vi->vdev); |
830a8a97 SM |
2329 | |
2330 | /* Free unused buffers in both send and recv, if any. */ | |
9ab86bbc | 2331 | free_unused_bufs(vi); |
fb6813f4 | 2332 | |
986a4f4d | 2333 | free_receive_bufs(vi); |
d2a7ddda | 2334 | |
fb51879d MD |
2335 | free_receive_page_frags(vi); |
2336 | ||
986a4f4d | 2337 | virtnet_del_vqs(vi); |
04486ed0 AS |
2338 | } |
2339 | ||
8cc085d6 | 2340 | static void virtnet_remove(struct virtio_device *vdev) |
04486ed0 AS |
2341 | { |
2342 | struct virtnet_info *vi = vdev->priv; | |
2343 | ||
8017c279 | 2344 | virtnet_cpu_notif_remove(vi); |
8de4b2f3 | 2345 | |
102a2786 MT |
2346 | /* Make sure no work handler is accessing the device. */ |
2347 | flush_work(&vi->config_work); | |
586d17c5 | 2348 | |
04486ed0 AS |
2349 | unregister_netdev(vi->dev); |
2350 | ||
2351 | remove_vq_common(vi); | |
fb6813f4 | 2352 | |
2e66f55b | 2353 | free_percpu(vi->stats); |
74b2553f | 2354 | free_netdev(vi->dev); |
296f96fc RR |
2355 | } |
2356 | ||
89107000 | 2357 | #ifdef CONFIG_PM_SLEEP |
0741bcb5 AS |
2358 | static int virtnet_freeze(struct virtio_device *vdev) |
2359 | { | |
2360 | struct virtnet_info *vi = vdev->priv; | |
986a4f4d | 2361 | int i; |
0741bcb5 | 2362 | |
8017c279 | 2363 | virtnet_cpu_notif_remove(vi); |
ec9debbd | 2364 | |
102a2786 MT |
2365 | /* Make sure no work handler is accessing the device */ |
2366 | flush_work(&vi->config_work); | |
586d17c5 | 2367 | |
0741bcb5 AS |
2368 | netif_device_detach(vi->dev); |
2369 | cancel_delayed_work_sync(&vi->refill); | |
2370 | ||
91815639 | 2371 | if (netif_running(vi->dev)) { |
ab3971b1 | 2372 | for (i = 0; i < vi->max_queue_pairs; i++) |
986a4f4d | 2373 | napi_disable(&vi->rq[i].napi); |
91815639 | 2374 | } |
0741bcb5 AS |
2375 | |
2376 | remove_vq_common(vi); | |
2377 | ||
2378 | return 0; | |
2379 | } | |
2380 | ||
2381 | static int virtnet_restore(struct virtio_device *vdev) | |
2382 | { | |
2383 | struct virtnet_info *vi = vdev->priv; | |
986a4f4d | 2384 | int err, i; |
0741bcb5 AS |
2385 | |
2386 | err = init_vqs(vi); | |
2387 | if (err) | |
2388 | return err; | |
2389 | ||
e53fbd11 MT |
2390 | virtio_device_ready(vdev); |
2391 | ||
6cd4ce00 JW |
2392 | if (netif_running(vi->dev)) { |
2393 | for (i = 0; i < vi->curr_queue_pairs; i++) | |
946fa564 | 2394 | if (!try_fill_recv(vi, &vi->rq[i], GFP_KERNEL)) |
6cd4ce00 JW |
2395 | schedule_delayed_work(&vi->refill, 0); |
2396 | ||
986a4f4d JW |
2397 | for (i = 0; i < vi->max_queue_pairs; i++) |
2398 | virtnet_napi_enable(&vi->rq[i]); | |
6cd4ce00 | 2399 | } |
0741bcb5 AS |
2400 | |
2401 | netif_device_attach(vi->dev); | |
2402 | ||
35ed159b | 2403 | rtnl_lock(); |
986a4f4d | 2404 | virtnet_set_queues(vi, vi->curr_queue_pairs); |
35ed159b | 2405 | rtnl_unlock(); |
986a4f4d | 2406 | |
8017c279 | 2407 | err = virtnet_cpu_notif_add(vi); |
ec9debbd JW |
2408 | if (err) |
2409 | return err; | |
2410 | ||
0741bcb5 AS |
2411 | return 0; |
2412 | } | |
2413 | #endif | |
2414 | ||
296f96fc RR |
2415 | static struct virtio_device_id id_table[] = { |
2416 | { VIRTIO_ID_NET, VIRTIO_DEV_ANY_ID }, | |
2417 | { 0 }, | |
2418 | }; | |
2419 | ||
f3358507 MT |
2420 | #define VIRTNET_FEATURES \ |
2421 | VIRTIO_NET_F_CSUM, VIRTIO_NET_F_GUEST_CSUM, \ | |
2422 | VIRTIO_NET_F_MAC, \ | |
2423 | VIRTIO_NET_F_HOST_TSO4, VIRTIO_NET_F_HOST_UFO, VIRTIO_NET_F_HOST_TSO6, \ | |
2424 | VIRTIO_NET_F_HOST_ECN, VIRTIO_NET_F_GUEST_TSO4, VIRTIO_NET_F_GUEST_TSO6, \ | |
2425 | VIRTIO_NET_F_GUEST_ECN, VIRTIO_NET_F_GUEST_UFO, \ | |
2426 | VIRTIO_NET_F_MRG_RXBUF, VIRTIO_NET_F_STATUS, VIRTIO_NET_F_CTRL_VQ, \ | |
2427 | VIRTIO_NET_F_CTRL_RX, VIRTIO_NET_F_CTRL_VLAN, \ | |
2428 | VIRTIO_NET_F_GUEST_ANNOUNCE, VIRTIO_NET_F_MQ, \ | |
2429 | VIRTIO_NET_F_CTRL_MAC_ADDR, \ | |
2430 | VIRTIO_NET_F_MTU | |
2431 | ||
c45a6816 | 2432 | static unsigned int features[] = { |
f3358507 MT |
2433 | VIRTNET_FEATURES, |
2434 | }; | |
2435 | ||
2436 | static unsigned int features_legacy[] = { | |
2437 | VIRTNET_FEATURES, | |
2438 | VIRTIO_NET_F_GSO, | |
e7428e95 | 2439 | VIRTIO_F_ANY_LAYOUT, |
c45a6816 RR |
2440 | }; |
2441 | ||
22402529 | 2442 | static struct virtio_driver virtio_net_driver = { |
c45a6816 RR |
2443 | .feature_table = features, |
2444 | .feature_table_size = ARRAY_SIZE(features), | |
f3358507 MT |
2445 | .feature_table_legacy = features_legacy, |
2446 | .feature_table_size_legacy = ARRAY_SIZE(features_legacy), | |
296f96fc RR |
2447 | .driver.name = KBUILD_MODNAME, |
2448 | .driver.owner = THIS_MODULE, | |
2449 | .id_table = id_table, | |
2450 | .probe = virtnet_probe, | |
8cc085d6 | 2451 | .remove = virtnet_remove, |
9f4d26d0 | 2452 | .config_changed = virtnet_config_changed, |
89107000 | 2453 | #ifdef CONFIG_PM_SLEEP |
0741bcb5 AS |
2454 | .freeze = virtnet_freeze, |
2455 | .restore = virtnet_restore, | |
2456 | #endif | |
296f96fc RR |
2457 | }; |
2458 | ||
8017c279 SAS |
2459 | static __init int virtio_net_driver_init(void) |
2460 | { | |
2461 | int ret; | |
2462 | ||
2463 | ret = cpuhp_setup_state_multi(CPUHP_AP_ONLINE_DYN, "AP_VIRT_NET_ONLINE", | |
2464 | virtnet_cpu_online, | |
2465 | virtnet_cpu_down_prep); | |
2466 | if (ret < 0) | |
2467 | goto out; | |
2468 | virtionet_online = ret; | |
2469 | ret = cpuhp_setup_state_multi(CPUHP_VIRT_NET_DEAD, "VIRT_NET_DEAD", | |
2470 | NULL, virtnet_cpu_dead); | |
2471 | if (ret) | |
2472 | goto err_dead; | |
2473 | ||
2474 | ret = register_virtio_driver(&virtio_net_driver); | |
2475 | if (ret) | |
2476 | goto err_virtio; | |
2477 | return 0; | |
2478 | err_virtio: | |
2479 | cpuhp_remove_multi_state(CPUHP_VIRT_NET_DEAD); | |
2480 | err_dead: | |
2481 | cpuhp_remove_multi_state(virtionet_online); | |
2482 | out: | |
2483 | return ret; | |
2484 | } | |
2485 | module_init(virtio_net_driver_init); | |
2486 | ||
2487 | static __exit void virtio_net_driver_exit(void) | |
2488 | { | |
2489 | cpuhp_remove_multi_state(CPUHP_VIRT_NET_DEAD); | |
2490 | cpuhp_remove_multi_state(virtionet_online); | |
2491 | unregister_virtio_driver(&virtio_net_driver); | |
2492 | } | |
2493 | module_exit(virtio_net_driver_exit); | |
296f96fc RR |
2494 | |
2495 | MODULE_DEVICE_TABLE(virtio, id_table); | |
2496 | MODULE_DESCRIPTION("Virtio network driver"); | |
2497 | MODULE_LICENSE("GPL"); |