blob: a014ca042390aeebc58bab7eec878c576b474985 [file] [log] [blame]
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001/* Copyright (C) 2009 Red Hat, Inc.
2 * Author: Michael S. Tsirkin <mst@redhat.com>
3 *
4 * This work is licensed under the terms of the GNU GPL, version 2.
5 *
6 * virtio-net server in host kernel.
7 */
8
9#include <linux/compat.h>
10#include <linux/eventfd.h>
11#include <linux/vhost.h>
12#include <linux/virtio_net.h>
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +000013#include <linux/miscdevice.h>
14#include <linux/module.h>
Michael S. Tsirkinbab632d2011-07-18 03:48:46 +000015#include <linux/moduleparam.h>
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +000016#include <linux/mutex.h>
17#include <linux/workqueue.h>
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +000018#include <linux/file.h>
Tejun Heo5a0e3ad2010-03-24 17:04:11 +090019#include <linux/slab.h>
Ingo Molnare6017572017-02-01 16:36:40 +010020#include <linux/sched/clock.h>
Ingo Molnar174cd4b2017-02-02 19:15:33 +010021#include <linux/sched/signal.h>
Michael S. Tsirkin23cc5a92013-01-23 21:46:47 +010022#include <linux/vmalloc.h>
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +000023
24#include <linux/net.h>
25#include <linux/if_packet.h>
26#include <linux/if_arp.h>
27#include <linux/if_tun.h>
Arnd Bergmann501c7742010-02-18 05:46:50 +000028#include <linux/if_macvlan.h>
Sainath Grandhi635b8c82017-02-10 16:03:47 -080029#include <linux/if_tap.h>
Basil Gorc53cff5e2012-05-03 22:55:23 +000030#include <linux/if_vlan.h>
Jason Wangc67df112017-05-17 12:14:45 +080031#include <linux/skb_array.h>
32#include <linux/skbuff.h>
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +000033
34#include <net/sock.h>
Jesper Dangaard Brouer1ffcbc82018-04-17 16:45:47 +020035#include <net/xdp.h>
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +000036
37#include "vhost.h"
38
Michael S. Tsirkinf9611c42012-12-06 14:56:00 +020039static int experimental_zcopytx = 1;
Michael S. Tsirkinbab632d2011-07-18 03:48:46 +000040module_param(experimental_zcopytx, int, 0444);
Michael S. Tsirkinf9611c42012-12-06 14:56:00 +020041MODULE_PARM_DESC(experimental_zcopytx, "Enable Zero Copy TX;"
42 " 1 -Enable; 0 - Disable");
Michael S. Tsirkinbab632d2011-07-18 03:48:46 +000043
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +000044/* Max number of bytes transferred before requeueing the job.
45 * Using this limit prevents one virtqueue from starving others. */
46#define VHOST_NET_WEIGHT 0x80000
47
haibinzhang(张海斌)a2ac9992018-04-09 07:22:17 +000048/* Max number of packets transferred before requeueing the job.
Paolo Abenidb688c22018-04-24 10:34:36 +020049 * Using this limit prevents one virtqueue from starving others with small
50 * pkts.
51 */
52#define VHOST_NET_PKT_WEIGHT 256
haibinzhang(张海斌)a2ac9992018-04-09 07:22:17 +000053
Michael S. Tsirkinbab632d2011-07-18 03:48:46 +000054/* MAX number of TX used buffers for outstanding zerocopy */
55#define VHOST_MAX_PEND 128
56#define VHOST_GOODCOPY_LEN 256
57
Michael S. Tsirkineaae8132012-11-01 09:16:51 +000058/*
59 * For transmit, used buffer len is unused; we override it to track buffer
60 * status internally; used for zerocopy tx only.
61 */
62/* Lower device DMA failed */
Michael S. Tsirkinbf995732014-10-24 11:49:27 +030063#define VHOST_DMA_FAILED_LEN ((__force __virtio32)3)
Michael S. Tsirkineaae8132012-11-01 09:16:51 +000064/* Lower device DMA done */
Michael S. Tsirkinbf995732014-10-24 11:49:27 +030065#define VHOST_DMA_DONE_LEN ((__force __virtio32)2)
Michael S. Tsirkineaae8132012-11-01 09:16:51 +000066/* Lower device DMA in progress */
Michael S. Tsirkinbf995732014-10-24 11:49:27 +030067#define VHOST_DMA_IN_PROGRESS ((__force __virtio32)1)
Michael S. Tsirkineaae8132012-11-01 09:16:51 +000068/* Buffer unused */
Michael S. Tsirkinbf995732014-10-24 11:49:27 +030069#define VHOST_DMA_CLEAR_LEN ((__force __virtio32)0)
Michael S. Tsirkineaae8132012-11-01 09:16:51 +000070
Michael S. Tsirkinbf995732014-10-24 11:49:27 +030071#define VHOST_DMA_IS_DONE(len) ((__force u32)(len) >= (__force u32)VHOST_DMA_DONE_LEN)
Michael S. Tsirkineaae8132012-11-01 09:16:51 +000072
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +000073enum {
Asias He8570a6e2013-05-06 16:38:20 +080074 VHOST_NET_FEATURES = VHOST_FEATURES |
75 (1ULL << VHOST_NET_F_VIRTIO_NET_HDR) |
Jason Wang6b1e6cc2016-06-23 02:04:32 -040076 (1ULL << VIRTIO_NET_F_MRG_RXBUF) |
77 (1ULL << VIRTIO_F_IOMMU_PLATFORM)
Asias He8570a6e2013-05-06 16:38:20 +080078};
79
80enum {
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +000081 VHOST_NET_VQ_RX = 0,
82 VHOST_NET_VQ_TX = 1,
83 VHOST_NET_VQ_MAX = 2,
84};
85
Asias Hefe729a52013-05-06 16:38:24 +080086struct vhost_net_ubuf_ref {
Michael S. Tsirkin0ad8b482014-02-13 11:42:05 +020087 /* refcount follows semantics similar to kref:
88 * 0: object is released
89 * 1: no outstanding ubufs
90 * >1: outstanding ubufs
91 */
92 atomic_t refcount;
Asias He28394002013-04-27 15:07:46 +080093 wait_queue_head_t wait;
94 struct vhost_virtqueue *vq;
95};
96
Jason Wangc67df112017-05-17 12:14:45 +080097#define VHOST_RX_BATCH 64
98struct vhost_net_buf {
Jason Wang5990a302018-01-04 11:14:27 +080099 void **queue;
Jason Wangc67df112017-05-17 12:14:45 +0800100 int tail;
101 int head;
102};
103
Asias He3ab2e422013-04-27 11:16:48 +0800104struct vhost_net_virtqueue {
105 struct vhost_virtqueue vq;
Michael S. Tsirkin81f95a52013-04-28 15:51:40 +0300106 size_t vhost_hlen;
107 size_t sock_hlen;
Asias He28394002013-04-27 15:07:46 +0800108 /* vhost zerocopy support fields below: */
109 /* last used idx for outstanding DMA zerocopy buffers */
110 int upend_idx;
Jason Wangf5a49412018-05-29 14:18:19 +0800111 /* For TX, first used idx for DMA done zerocopy buffers
112 * For RX, number of batched heads
113 */
Asias He28394002013-04-27 15:07:46 +0800114 int done_idx;
115 /* an array of userspace buffers info */
116 struct ubuf_info *ubuf_info;
117 /* Reference counting for outstanding ubufs.
118 * Protected by vq mutex. Writers must also take device mutex. */
Asias Hefe729a52013-05-06 16:38:24 +0800119 struct vhost_net_ubuf_ref *ubufs;
Jason Wang5990a302018-01-04 11:14:27 +0800120 struct ptr_ring *rx_ring;
Jason Wangc67df112017-05-17 12:14:45 +0800121 struct vhost_net_buf rxq;
Asias He3ab2e422013-04-27 11:16:48 +0800122};
123
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000124struct vhost_net {
125 struct vhost_dev dev;
Asias He3ab2e422013-04-27 11:16:48 +0800126 struct vhost_net_virtqueue vqs[VHOST_NET_VQ_MAX];
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000127 struct vhost_poll poll[VHOST_NET_VQ_MAX];
Michael S. Tsirkineaae8132012-11-01 09:16:51 +0000128 /* Number of TX recently submitted.
129 * Protected by tx vq lock. */
130 unsigned tx_packets;
131 /* Number of times zerocopy TX recently failed.
132 * Protected by tx vq lock. */
133 unsigned tx_zcopy_err;
Michael S. Tsirkin1280c272012-12-04 00:17:14 +0200134 /* Flush in progress. Protected by tx vq lock. */
135 bool tx_flush;
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000136};
137
Asias Hefe729a52013-05-06 16:38:24 +0800138static unsigned vhost_net_zcopy_mask __read_mostly;
Asias He28394002013-04-27 15:07:46 +0800139
Jason Wangc67df112017-05-17 12:14:45 +0800140static void *vhost_net_buf_get_ptr(struct vhost_net_buf *rxq)
141{
142 if (rxq->tail != rxq->head)
143 return rxq->queue[rxq->head];
144 else
145 return NULL;
146}
147
148static int vhost_net_buf_get_size(struct vhost_net_buf *rxq)
149{
150 return rxq->tail - rxq->head;
151}
152
153static int vhost_net_buf_is_empty(struct vhost_net_buf *rxq)
154{
155 return rxq->tail == rxq->head;
156}
157
158static void *vhost_net_buf_consume(struct vhost_net_buf *rxq)
159{
160 void *ret = vhost_net_buf_get_ptr(rxq);
161 ++rxq->head;
162 return ret;
163}
164
165static int vhost_net_buf_produce(struct vhost_net_virtqueue *nvq)
166{
167 struct vhost_net_buf *rxq = &nvq->rxq;
168
169 rxq->head = 0;
Jason Wang5990a302018-01-04 11:14:27 +0800170 rxq->tail = ptr_ring_consume_batched(nvq->rx_ring, rxq->queue,
Jason Wangc67df112017-05-17 12:14:45 +0800171 VHOST_RX_BATCH);
172 return rxq->tail;
173}
174
175static void vhost_net_buf_unproduce(struct vhost_net_virtqueue *nvq)
176{
177 struct vhost_net_buf *rxq = &nvq->rxq;
178
Jason Wang5990a302018-01-04 11:14:27 +0800179 if (nvq->rx_ring && !vhost_net_buf_is_empty(rxq)) {
180 ptr_ring_unconsume(nvq->rx_ring, rxq->queue + rxq->head,
181 vhost_net_buf_get_size(rxq),
Jason Wang3a403072018-03-09 14:50:34 +0800182 tun_ptr_free);
Jason Wangc67df112017-05-17 12:14:45 +0800183 rxq->head = rxq->tail = 0;
184 }
185}
186
Jason Wangfc72d1d2018-01-04 11:14:28 +0800187static int vhost_net_buf_peek_len(void *ptr)
188{
Jesper Dangaard Brouer1ffcbc82018-04-17 16:45:47 +0200189 if (tun_is_xdp_frame(ptr)) {
190 struct xdp_frame *xdpf = tun_ptr_to_xdp(ptr);
Jason Wangfc72d1d2018-01-04 11:14:28 +0800191
Jesper Dangaard Brouer1ffcbc82018-04-17 16:45:47 +0200192 return xdpf->len;
Jason Wangfc72d1d2018-01-04 11:14:28 +0800193 }
194
195 return __skb_array_len_with_tag(ptr);
196}
197
Jason Wangc67df112017-05-17 12:14:45 +0800198static int vhost_net_buf_peek(struct vhost_net_virtqueue *nvq)
199{
200 struct vhost_net_buf *rxq = &nvq->rxq;
201
202 if (!vhost_net_buf_is_empty(rxq))
203 goto out;
204
205 if (!vhost_net_buf_produce(nvq))
206 return 0;
207
208out:
Jason Wangfc72d1d2018-01-04 11:14:28 +0800209 return vhost_net_buf_peek_len(vhost_net_buf_get_ptr(rxq));
Jason Wangc67df112017-05-17 12:14:45 +0800210}
211
212static void vhost_net_buf_init(struct vhost_net_buf *rxq)
213{
214 rxq->head = rxq->tail = 0;
215}
216
Asias Hefe729a52013-05-06 16:38:24 +0800217static void vhost_net_enable_zcopy(int vq)
Asias He28394002013-04-27 15:07:46 +0800218{
Asias Hefe729a52013-05-06 16:38:24 +0800219 vhost_net_zcopy_mask |= 0x1 << vq;
Asias He28394002013-04-27 15:07:46 +0800220}
221
Asias Hefe729a52013-05-06 16:38:24 +0800222static struct vhost_net_ubuf_ref *
223vhost_net_ubuf_alloc(struct vhost_virtqueue *vq, bool zcopy)
Asias He28394002013-04-27 15:07:46 +0800224{
Asias Hefe729a52013-05-06 16:38:24 +0800225 struct vhost_net_ubuf_ref *ubufs;
Asias He28394002013-04-27 15:07:46 +0800226 /* No zero copy backend? Nothing to count. */
227 if (!zcopy)
228 return NULL;
229 ubufs = kmalloc(sizeof(*ubufs), GFP_KERNEL);
230 if (!ubufs)
231 return ERR_PTR(-ENOMEM);
Michael S. Tsirkin0ad8b482014-02-13 11:42:05 +0200232 atomic_set(&ubufs->refcount, 1);
Asias He28394002013-04-27 15:07:46 +0800233 init_waitqueue_head(&ubufs->wait);
234 ubufs->vq = vq;
235 return ubufs;
236}
237
Michael S. Tsirkin0ad8b482014-02-13 11:42:05 +0200238static int vhost_net_ubuf_put(struct vhost_net_ubuf_ref *ubufs)
Asias He28394002013-04-27 15:07:46 +0800239{
Michael S. Tsirkin0ad8b482014-02-13 11:42:05 +0200240 int r = atomic_sub_return(1, &ubufs->refcount);
241 if (unlikely(!r))
242 wake_up(&ubufs->wait);
243 return r;
Asias He28394002013-04-27 15:07:46 +0800244}
245
Asias Hefe729a52013-05-06 16:38:24 +0800246static void vhost_net_ubuf_put_and_wait(struct vhost_net_ubuf_ref *ubufs)
Asias He28394002013-04-27 15:07:46 +0800247{
Michael S. Tsirkin0ad8b482014-02-13 11:42:05 +0200248 vhost_net_ubuf_put(ubufs);
249 wait_event(ubufs->wait, !atomic_read(&ubufs->refcount));
Michael S. Tsirkinc38e39c2013-06-25 17:29:46 +0300250}
251
252static void vhost_net_ubuf_put_wait_and_free(struct vhost_net_ubuf_ref *ubufs)
253{
254 vhost_net_ubuf_put_and_wait(ubufs);
Asias He28394002013-04-27 15:07:46 +0800255 kfree(ubufs);
256}
257
Asias Heb1ad8492013-05-06 11:16:00 +0800258static void vhost_net_clear_ubuf_info(struct vhost_net *n)
259{
Asias Heb1ad8492013-05-06 11:16:00 +0800260 int i;
261
Michael S. Tsirkin288cfe72013-06-06 15:20:46 +0300262 for (i = 0; i < VHOST_NET_VQ_MAX; ++i) {
263 kfree(n->vqs[i].ubuf_info);
264 n->vqs[i].ubuf_info = NULL;
Asias Heb1ad8492013-05-06 11:16:00 +0800265 }
266}
267
Asias He0a1febf2013-06-05 21:17:38 +0800268static int vhost_net_set_ubuf_info(struct vhost_net *n)
Asias He28394002013-04-27 15:07:46 +0800269{
270 bool zcopy;
271 int i;
272
Michael S. Tsirkin288cfe72013-06-06 15:20:46 +0300273 for (i = 0; i < VHOST_NET_VQ_MAX; ++i) {
Asias Hefe729a52013-05-06 16:38:24 +0800274 zcopy = vhost_net_zcopy_mask & (0x1 << i);
Asias He28394002013-04-27 15:07:46 +0800275 if (!zcopy)
276 continue;
Kees Cook6da2ec52018-06-12 13:55:00 -0700277 n->vqs[i].ubuf_info =
278 kmalloc_array(UIO_MAXIOV,
279 sizeof(*n->vqs[i].ubuf_info),
280 GFP_KERNEL);
Asias He28394002013-04-27 15:07:46 +0800281 if (!n->vqs[i].ubuf_info)
282 goto err;
283 }
284 return 0;
285
286err:
Michael S. Tsirkin288cfe72013-06-06 15:20:46 +0300287 vhost_net_clear_ubuf_info(n);
Asias He28394002013-04-27 15:07:46 +0800288 return -ENOMEM;
289}
290
Asias He0a1febf2013-06-05 21:17:38 +0800291static void vhost_net_vq_reset(struct vhost_net *n)
Asias He28394002013-04-27 15:07:46 +0800292{
293 int i;
294
Michael S. Tsirkin288cfe72013-06-06 15:20:46 +0300295 vhost_net_clear_ubuf_info(n);
296
Asias He28394002013-04-27 15:07:46 +0800297 for (i = 0; i < VHOST_NET_VQ_MAX; i++) {
298 n->vqs[i].done_idx = 0;
299 n->vqs[i].upend_idx = 0;
300 n->vqs[i].ubufs = NULL;
Michael S. Tsirkin81f95a52013-04-28 15:51:40 +0300301 n->vqs[i].vhost_hlen = 0;
302 n->vqs[i].sock_hlen = 0;
Jason Wangc67df112017-05-17 12:14:45 +0800303 vhost_net_buf_init(&n->vqs[i].rxq);
Asias He28394002013-04-27 15:07:46 +0800304 }
305
306}
307
Michael S. Tsirkineaae8132012-11-01 09:16:51 +0000308static void vhost_net_tx_packet(struct vhost_net *net)
309{
310 ++net->tx_packets;
311 if (net->tx_packets < 1024)
312 return;
313 net->tx_packets = 0;
314 net->tx_zcopy_err = 0;
315}
316
317static void vhost_net_tx_err(struct vhost_net *net)
318{
319 ++net->tx_zcopy_err;
320}
321
322static bool vhost_net_tx_select_zcopy(struct vhost_net *net)
323{
Michael S. Tsirkin1280c272012-12-04 00:17:14 +0200324 /* TX flush waits for outstanding DMAs to be done.
325 * Don't start new DMAs.
326 */
327 return !net->tx_flush &&
328 net->tx_packets / 64 >= net->tx_zcopy_err;
Michael S. Tsirkineaae8132012-11-01 09:16:51 +0000329}
330
Michael S. Tsirkinbab632d2011-07-18 03:48:46 +0000331static bool vhost_sock_zcopy(struct socket *sock)
332{
333 return unlikely(experimental_zcopytx) &&
334 sock_flag(sock->sk, SOCK_ZEROCOPY);
335}
336
Michael S. Tsirkinb2116162012-11-01 09:16:46 +0000337/* In case of DMA done not in order in lower device driver for some reason.
338 * upend_idx is used to track end of used idx, done_idx is used to track head
339 * of used idx. Once lower device DMA done contiguously, we will signal KVM
340 * guest used idx.
341 */
Jason Wang094afe72013-09-02 16:40:56 +0800342static void vhost_zerocopy_signal_used(struct vhost_net *net,
343 struct vhost_virtqueue *vq)
Michael S. Tsirkinb2116162012-11-01 09:16:46 +0000344{
Asias He28394002013-04-27 15:07:46 +0800345 struct vhost_net_virtqueue *nvq =
346 container_of(vq, struct vhost_net_virtqueue, vq);
Jason Wangc92112a2013-09-02 16:40:57 +0800347 int i, add;
Michael S. Tsirkinb2116162012-11-01 09:16:46 +0000348 int j = 0;
349
Asias He28394002013-04-27 15:07:46 +0800350 for (i = nvq->done_idx; i != nvq->upend_idx; i = (i + 1) % UIO_MAXIOV) {
Michael S. Tsirkineaae8132012-11-01 09:16:51 +0000351 if (vq->heads[i].len == VHOST_DMA_FAILED_LEN)
352 vhost_net_tx_err(net);
Michael S. Tsirkinb2116162012-11-01 09:16:46 +0000353 if (VHOST_DMA_IS_DONE(vq->heads[i].len)) {
354 vq->heads[i].len = VHOST_DMA_CLEAR_LEN;
Michael S. Tsirkinb2116162012-11-01 09:16:46 +0000355 ++j;
356 } else
357 break;
358 }
Jason Wangc92112a2013-09-02 16:40:57 +0800359 while (j) {
360 add = min(UIO_MAXIOV - nvq->done_idx, j);
361 vhost_add_used_and_signal_n(vq->dev, vq,
362 &vq->heads[nvq->done_idx], add);
363 nvq->done_idx = (nvq->done_idx + add) % UIO_MAXIOV;
364 j -= add;
365 }
Michael S. Tsirkinb2116162012-11-01 09:16:46 +0000366}
367
Michael S. Tsirkineaae8132012-11-01 09:16:51 +0000368static void vhost_zerocopy_callback(struct ubuf_info *ubuf, bool success)
Michael S. Tsirkinb2116162012-11-01 09:16:46 +0000369{
Asias Hefe729a52013-05-06 16:38:24 +0800370 struct vhost_net_ubuf_ref *ubufs = ubuf->ctx;
Michael S. Tsirkinb2116162012-11-01 09:16:46 +0000371 struct vhost_virtqueue *vq = ubufs->vq;
Michael S. Tsirkin0ad8b482014-02-13 11:42:05 +0200372 int cnt;
Michael S. Tsirkinb2116162012-11-01 09:16:46 +0000373
Michael S. Tsirkinb0c057c2014-02-13 11:45:11 +0200374 rcu_read_lock_bh();
375
Jason Wang19c73b32013-09-02 16:41:00 +0800376 /* set len to mark this desc buffers done DMA */
377 vq->heads[ubuf->desc].len = success ?
378 VHOST_DMA_DONE_LEN : VHOST_DMA_FAILED_LEN;
Michael S. Tsirkin0ad8b482014-02-13 11:42:05 +0200379 cnt = vhost_net_ubuf_put(ubufs);
Jason Wang19c73b32013-09-02 16:41:00 +0800380
Michael S. Tsirkin24eb21a2012-11-01 09:16:55 +0000381 /*
382 * Trigger polling thread if guest stopped submitting new buffers:
Michael S. Tsirkin0ad8b482014-02-13 11:42:05 +0200383 * in this case, the refcount after decrement will eventually reach 1.
Michael S. Tsirkin24eb21a2012-11-01 09:16:55 +0000384 * We also trigger polling periodically after each 16 packets
385 * (the value 16 here is more or less arbitrary, it's tuned to trigger
386 * less than 10% of times).
387 */
Michael S. Tsirkin0ad8b482014-02-13 11:42:05 +0200388 if (cnt <= 1 || !(cnt % 16))
Michael S. Tsirkin24eb21a2012-11-01 09:16:55 +0000389 vhost_poll_queue(&vq->poll);
Michael S. Tsirkinb0c057c2014-02-13 11:45:11 +0200390
391 rcu_read_unlock_bh();
Michael S. Tsirkinb2116162012-11-01 09:16:46 +0000392}
393
Jason Wang03088132016-03-04 06:24:53 -0500394static inline unsigned long busy_clock(void)
395{
396 return local_clock() >> 10;
397}
398
Toshiaki Makita027b1762018-07-03 16:31:32 +0900399static bool vhost_can_busy_poll(unsigned long endtime)
Jason Wang03088132016-03-04 06:24:53 -0500400{
Toshiaki Makita027b1762018-07-03 16:31:32 +0900401 return likely(!need_resched() && !time_after(busy_clock(), endtime) &&
402 !signal_pending(current));
Jason Wang03088132016-03-04 06:24:53 -0500403}
404
Jason Wang8241a1e2016-06-01 01:56:33 -0400405static void vhost_net_disable_vq(struct vhost_net *n,
406 struct vhost_virtqueue *vq)
407{
408 struct vhost_net_virtqueue *nvq =
409 container_of(vq, struct vhost_net_virtqueue, vq);
410 struct vhost_poll *poll = n->poll + (nvq - n->vqs);
411 if (!vq->private_data)
412 return;
413 vhost_poll_stop(poll);
414}
415
416static int vhost_net_enable_vq(struct vhost_net *n,
417 struct vhost_virtqueue *vq)
418{
419 struct vhost_net_virtqueue *nvq =
420 container_of(vq, struct vhost_net_virtqueue, vq);
421 struct vhost_poll *poll = n->poll + (nvq - n->vqs);
422 struct socket *sock;
423
424 sock = vq->private_data;
425 if (!sock)
426 return 0;
427
428 return vhost_poll_start(poll, sock->file);
429}
430
Jason Wang03088132016-03-04 06:24:53 -0500431static int vhost_net_tx_get_vq_desc(struct vhost_net *net,
432 struct vhost_virtqueue *vq,
Toshiaki Makita027b1762018-07-03 16:31:32 +0900433 unsigned int *out_num, unsigned int *in_num,
434 bool *busyloop_intr)
Jason Wang03088132016-03-04 06:24:53 -0500435{
436 unsigned long uninitialized_var(endtime);
437 int r = vhost_get_vq_desc(vq, vq->iov, ARRAY_SIZE(vq->iov),
Jason Wang6b1e6cc2016-06-23 02:04:32 -0400438 out_num, in_num, NULL, NULL);
Jason Wang03088132016-03-04 06:24:53 -0500439
440 if (r == vq->num && vq->busyloop_timeout) {
441 preempt_disable();
442 endtime = busy_clock() + vq->busyloop_timeout;
Toshiaki Makita027b1762018-07-03 16:31:32 +0900443 while (vhost_can_busy_poll(endtime)) {
444 if (vhost_has_work(vq->dev)) {
445 *busyloop_intr = true;
446 break;
447 }
448 if (!vhost_vq_avail_empty(vq->dev, vq))
449 break;
Christian Borntraegerf2f09a42016-10-25 11:03:14 +0200450 cpu_relax();
Toshiaki Makita027b1762018-07-03 16:31:32 +0900451 }
Jason Wang03088132016-03-04 06:24:53 -0500452 preempt_enable();
453 r = vhost_get_vq_desc(vq, vq->iov, ARRAY_SIZE(vq->iov),
Jason Wang6b1e6cc2016-06-23 02:04:32 -0400454 out_num, in_num, NULL, NULL);
Jason Wang03088132016-03-04 06:24:53 -0500455 }
456
457 return r;
458}
459
Jason Wang0ed005c2017-01-18 15:02:02 +0800460static bool vhost_exceeds_maxpend(struct vhost_net *net)
461{
462 struct vhost_net_virtqueue *nvq = &net->vqs[VHOST_NET_VQ_TX];
463 struct vhost_virtqueue *vq = &nvq->vq;
464
Willem de Bruijn1e6f7452017-10-06 13:22:31 -0400465 return (nvq->upend_idx + UIO_MAXIOV - nvq->done_idx) % UIO_MAXIOV >
466 min_t(unsigned int, VHOST_MAX_PEND, vq->num >> 2);
Jason Wang0ed005c2017-01-18 15:02:02 +0800467}
468
Jason Wangb0d0ea52018-07-20 08:15:14 +0800469static size_t init_iov_iter(struct vhost_virtqueue *vq, struct iov_iter *iter,
470 size_t hdr_size, int out)
471{
472 /* Skip header. TODO: support TSO. */
473 size_t len = iov_length(vq->iov, out);
474
475 iov_iter_init(iter, WRITE, vq->iov, out, len);
476 iov_iter_advance(iter, hdr_size);
477
478 return iov_iter_count(iter);
479}
480
Jason Wang272f35c2018-07-20 08:15:15 +0800481static bool vhost_exceeds_weight(int pkts, int total_len)
482{
483 return total_len >= VHOST_NET_WEIGHT ||
484 pkts >= VHOST_NET_PKT_WEIGHT;
485}
486
Jason Wanga2a91a12018-07-20 08:15:16 +0800487static int get_tx_bufs(struct vhost_net *net,
488 struct vhost_net_virtqueue *nvq,
489 struct msghdr *msg,
490 unsigned int *out, unsigned int *in,
491 size_t *len, bool *busyloop_intr)
492{
493 struct vhost_virtqueue *vq = &nvq->vq;
494 int ret;
495
496 ret = vhost_net_tx_get_vq_desc(net, vq, out, in, busyloop_intr);
497 if (ret < 0 || ret == vq->num)
498 return ret;
499
500 if (*in) {
501 vq_err(vq, "Unexpected descriptor format for TX: out %d, int %d\n",
502 *out, *in);
503 return -EFAULT;
504 }
505
506 /* Sanity check */
507 *len = init_iov_iter(vq, &msg->msg_iter, nvq->vhost_hlen, *out);
508 if (*len == 0) {
509 vq_err(vq, "Unexpected header len for TX: %zd expected %zd\n",
510 *len, nvq->vhost_hlen);
511 return -EFAULT;
512 }
513
514 return ret;
515}
516
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000517/* Expects to be always run from workqueue - which acts as
518 * read-size critical section for our kind of RCU. */
519static void handle_tx(struct vhost_net *net)
520{
Asias He28394002013-04-27 15:07:46 +0800521 struct vhost_net_virtqueue *nvq = &net->vqs[VHOST_NET_VQ_TX];
Michael S. Tsirkin81f95a52013-04-28 15:51:40 +0300522 struct vhost_virtqueue *vq = &nvq->vq;
Al Viro98a527a2014-12-10 15:00:58 -0500523 unsigned out, in;
Michael S. Tsirkind5675bd2010-06-24 16:59:59 +0300524 int head;
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000525 struct msghdr msg = {
526 .msg_name = NULL,
527 .msg_namelen = 0,
528 .msg_control = NULL,
529 .msg_controllen = 0,
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000530 .msg_flags = MSG_DONTWAIT,
531 };
532 size_t len, total_len = 0;
Jason Wang70181d512013-04-10 20:50:48 +0000533 int err;
Arnd Bergmann28457ee2010-03-09 19:24:45 +0100534 struct socket *sock;
Asias Hefe729a52013-05-06 16:38:24 +0800535 struct vhost_net_ubuf_ref *uninitialized_var(ubufs);
Michael S. Tsirkincedb9bd2012-12-06 17:00:18 +0200536 bool zcopy, zcopy_used;
haibinzhang(张海斌)a2ac9992018-04-09 07:22:17 +0000537 int sent_pkts = 0;
Arnd Bergmann28457ee2010-03-09 19:24:45 +0100538
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000539 mutex_lock(&vq->mutex);
Asias He2e26af72013-05-07 14:54:33 +0800540 sock = vq->private_data;
541 if (!sock)
542 goto out;
543
Jason Wang6b1e6cc2016-06-23 02:04:32 -0400544 if (!vq_iotlb_prefetch(vq))
545 goto out;
546
Michael S. Tsirkin8ea8cf82011-05-20 02:10:54 +0300547 vhost_disable_notify(&net->dev, vq);
Jason Wangfeb88922017-11-13 11:45:34 +0800548 vhost_net_disable_vq(net, vq);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000549
Asias He28394002013-04-27 15:07:46 +0800550 zcopy = nvq->ubufs;
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000551
552 for (;;) {
Toshiaki Makita027b1762018-07-03 16:31:32 +0900553 bool busyloop_intr;
554
Michael S. Tsirkinbab632d2011-07-18 03:48:46 +0000555 /* Release DMAs done buffers first */
556 if (zcopy)
Michael S. Tsirkineaae8132012-11-01 09:16:51 +0000557 vhost_zerocopy_signal_used(net, vq);
Michael S. Tsirkinbab632d2011-07-18 03:48:46 +0000558
Toshiaki Makita027b1762018-07-03 16:31:32 +0900559 busyloop_intr = false;
Jason Wanga2a91a12018-07-20 08:15:16 +0800560 head = get_tx_bufs(net, nvq, &msg, &out, &in, &len,
561 &busyloop_intr);
Michael S. Tsirkind5675bd2010-06-24 16:59:59 +0300562 /* On error, stop handling until the next kick. */
Michael S. Tsirkin7b3384f2010-07-01 18:40:12 +0300563 if (unlikely(head < 0))
Michael S. Tsirkind5675bd2010-06-24 16:59:59 +0300564 break;
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000565 /* Nothing new? Wait for eventfd to tell us they refilled. */
566 if (head == vq->num) {
Toshiaki Makita027b1762018-07-03 16:31:32 +0900567 if (unlikely(busyloop_intr)) {
568 vhost_poll_queue(&vq->poll);
569 } else if (unlikely(vhost_enable_notify(&net->dev, vq))) {
Michael S. Tsirkin8ea8cf82011-05-20 02:10:54 +0300570 vhost_disable_notify(&net->dev, vq);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000571 continue;
572 }
573 break;
574 }
Jason Wangce21a022013-09-02 16:40:59 +0800575
576 zcopy_used = zcopy && len >= VHOST_GOODCOPY_LEN
Willem de Bruijn1e6f7452017-10-06 13:22:31 -0400577 && !vhost_exceeds_maxpend(net)
Jason Wangce21a022013-09-02 16:40:59 +0800578 && vhost_net_tx_select_zcopy(net);
Michael S. Tsirkincedb9bd2012-12-06 17:00:18 +0200579
Michael S. Tsirkinbab632d2011-07-18 03:48:46 +0000580 /* use msg_control to pass vhost zerocopy ubuf info to skb */
Michael S. Tsirkincedb9bd2012-12-06 17:00:18 +0200581 if (zcopy_used) {
Jason Wangce21a022013-09-02 16:40:59 +0800582 struct ubuf_info *ubuf;
583 ubuf = nvq->ubuf_info + nvq->upend_idx;
Michael S. Tsirkinbab632d2011-07-18 03:48:46 +0000584
Michael S. Tsirkin8b386942014-10-24 14:19:48 +0300585 vq->heads[nvq->upend_idx].id = cpu_to_vhost32(vq, head);
Jason Wangce21a022013-09-02 16:40:59 +0800586 vq->heads[nvq->upend_idx].len = VHOST_DMA_IN_PROGRESS;
587 ubuf->callback = vhost_zerocopy_callback;
588 ubuf->ctx = nvq->ubufs;
589 ubuf->desc = nvq->upend_idx;
Eric Dumazetc1d1b432017-08-31 16:48:22 -0700590 refcount_set(&ubuf->refcnt, 1);
Jason Wangce21a022013-09-02 16:40:59 +0800591 msg.msg_control = ubuf;
592 msg.msg_controllen = sizeof(ubuf);
593 ubufs = nvq->ubufs;
Michael S. Tsirkin0ad8b482014-02-13 11:42:05 +0200594 atomic_inc(&ubufs->refcount);
Asias He28394002013-04-27 15:07:46 +0800595 nvq->upend_idx = (nvq->upend_idx + 1) % UIO_MAXIOV;
Jason Wangce21a022013-09-02 16:40:59 +0800596 } else {
Jason Wang4364d5f2013-06-05 15:40:46 +0800597 msg.msg_control = NULL;
Jason Wangce21a022013-09-02 16:40:59 +0800598 ubufs = NULL;
599 }
Jason Wang0ed005c2017-01-18 15:02:02 +0800600 total_len += len;
601 if (total_len < VHOST_NET_WEIGHT &&
602 !vhost_vq_avail_empty(&net->dev, vq) &&
603 likely(!vhost_exceeds_maxpend(net))) {
604 msg.msg_flags |= MSG_MORE;
605 } else {
606 msg.msg_flags &= ~MSG_MORE;
607 }
608
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000609 /* TODO: Check specific error and bomb out unless ENOBUFS? */
Ying Xue1b784142015-03-02 15:37:48 +0800610 err = sock->ops->sendmsg(sock, &msg, len);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000611 if (unlikely(err < 0)) {
Michael S. Tsirkincedb9bd2012-12-06 17:00:18 +0200612 if (zcopy_used) {
Jason Wangce21a022013-09-02 16:40:59 +0800613 vhost_net_ubuf_put(ubufs);
Asias He28394002013-04-27 15:07:46 +0800614 nvq->upend_idx = ((unsigned)nvq->upend_idx - 1)
615 % UIO_MAXIOV;
Michael S. Tsirkinbab632d2011-07-18 03:48:46 +0000616 }
David Stevens8dd014a2010-07-27 18:52:21 +0300617 vhost_discard_vq_desc(vq, 1);
Jason Wangfeb88922017-11-13 11:45:34 +0800618 vhost_net_enable_vq(net, vq);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000619 break;
620 }
621 if (err != len)
Michael S. Tsirkin95c0ec62010-06-24 17:10:25 +0300622 pr_debug("Truncated TX packet: "
623 " len %d != %zd\n", err, len);
Michael S. Tsirkincedb9bd2012-12-06 17:00:18 +0200624 if (!zcopy_used)
Michael S. Tsirkinbab632d2011-07-18 03:48:46 +0000625 vhost_add_used_and_signal(&net->dev, vq, head, 0);
Jason Wangc8fb2172012-05-02 11:42:41 +0800626 else
Michael S. Tsirkineaae8132012-11-01 09:16:51 +0000627 vhost_zerocopy_signal_used(net, vq);
Michael S. Tsirkineaae8132012-11-01 09:16:51 +0000628 vhost_net_tx_packet(net);
Jason Wang272f35c2018-07-20 08:15:15 +0800629 if (unlikely(vhost_exceeds_weight(++sent_pkts, total_len))) {
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000630 vhost_poll_queue(&vq->poll);
631 break;
632 }
633 }
Asias He2e26af72013-05-07 14:54:33 +0800634out:
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000635 mutex_unlock(&vq->mutex);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000636}
637
Jason Wangc67df112017-05-17 12:14:45 +0800638static int peek_head_len(struct vhost_net_virtqueue *rvq, struct sock *sk)
David Stevens8dd014a2010-07-27 18:52:21 +0300639{
640 struct sk_buff *head;
641 int len = 0;
Jason Wang783e3982011-01-17 16:11:17 +0800642 unsigned long flags;
David Stevens8dd014a2010-07-27 18:52:21 +0300643
Jason Wang5990a302018-01-04 11:14:27 +0800644 if (rvq->rx_ring)
Jason Wangc67df112017-05-17 12:14:45 +0800645 return vhost_net_buf_peek(rvq);
Jason Wang1576d982016-06-30 14:45:36 +0800646
Jason Wang783e3982011-01-17 16:11:17 +0800647 spin_lock_irqsave(&sk->sk_receive_queue.lock, flags);
David Stevens8dd014a2010-07-27 18:52:21 +0300648 head = skb_peek(&sk->sk_receive_queue);
Basil Gorc53cff5e2012-05-03 22:55:23 +0000649 if (likely(head)) {
David Stevens8dd014a2010-07-27 18:52:21 +0300650 len = head->len;
Jiri Pirkodf8a39d2015-01-13 17:13:44 +0100651 if (skb_vlan_tag_present(head))
Basil Gorc53cff5e2012-05-03 22:55:23 +0000652 len += VLAN_HLEN;
653 }
654
Jason Wang783e3982011-01-17 16:11:17 +0800655 spin_unlock_irqrestore(&sk->sk_receive_queue.lock, flags);
David Stevens8dd014a2010-07-27 18:52:21 +0300656 return len;
657}
658
Jason Wang1576d982016-06-30 14:45:36 +0800659static int sk_has_rx_data(struct sock *sk)
660{
661 struct socket *sock = sk->sk_socket;
662
663 if (sock->ops->peek_len)
664 return sock->ops->peek_len(sock);
665
666 return skb_queue_empty(&sk->sk_receive_queue);
667}
668
Jason Wangf5a49412018-05-29 14:18:19 +0800669static void vhost_rx_signal_used(struct vhost_net_virtqueue *nvq)
670{
671 struct vhost_virtqueue *vq = &nvq->vq;
672 struct vhost_dev *dev = vq->dev;
673
674 if (!nvq->done_idx)
675 return;
676
677 vhost_add_used_and_signal_n(dev, vq, vq->heads, nvq->done_idx);
678 nvq->done_idx = 0;
679}
680
Toshiaki Makitabe294a52018-07-03 16:31:33 +0900681static int vhost_net_rx_peek_head_len(struct vhost_net *net, struct sock *sk,
682 bool *busyloop_intr)
Jason Wang03088132016-03-04 06:24:53 -0500683{
Toshiaki Makita28b9b332018-07-03 16:31:31 +0900684 struct vhost_net_virtqueue *rnvq = &net->vqs[VHOST_NET_VQ_RX];
685 struct vhost_net_virtqueue *tnvq = &net->vqs[VHOST_NET_VQ_TX];
Toshiaki Makita6369fec2018-07-03 16:31:34 +0900686 struct vhost_virtqueue *rvq = &rnvq->vq;
Toshiaki Makita28b9b332018-07-03 16:31:31 +0900687 struct vhost_virtqueue *tvq = &tnvq->vq;
Jason Wang03088132016-03-04 06:24:53 -0500688 unsigned long uninitialized_var(endtime);
Toshiaki Makita28b9b332018-07-03 16:31:31 +0900689 int len = peek_head_len(rnvq, sk);
Jason Wang03088132016-03-04 06:24:53 -0500690
Toshiaki Makita28b9b332018-07-03 16:31:31 +0900691 if (!len && tvq->busyloop_timeout) {
Jason Wangf5a49412018-05-29 14:18:19 +0800692 /* Flush batched heads first */
Toshiaki Makita28b9b332018-07-03 16:31:31 +0900693 vhost_rx_signal_used(rnvq);
Jason Wang03088132016-03-04 06:24:53 -0500694 /* Both tx vq and rx socket were polled here */
Toshiaki Makita28b9b332018-07-03 16:31:31 +0900695 mutex_lock_nested(&tvq->mutex, 1);
696 vhost_disable_notify(&net->dev, tvq);
Jason Wang03088132016-03-04 06:24:53 -0500697
698 preempt_disable();
Toshiaki Makita28b9b332018-07-03 16:31:31 +0900699 endtime = busy_clock() + tvq->busyloop_timeout;
Jason Wang03088132016-03-04 06:24:53 -0500700
Toshiaki Makitabe294a52018-07-03 16:31:33 +0900701 while (vhost_can_busy_poll(endtime)) {
702 if (vhost_has_work(&net->dev)) {
703 *busyloop_intr = true;
704 break;
705 }
Toshiaki Makita6369fec2018-07-03 16:31:34 +0900706 if ((sk_has_rx_data(sk) &&
707 !vhost_vq_avail_empty(&net->dev, rvq)) ||
Toshiaki Makitabe294a52018-07-03 16:31:33 +0900708 !vhost_vq_avail_empty(&net->dev, tvq))
709 break;
Christian Borntraegerf2f09a42016-10-25 11:03:14 +0200710 cpu_relax();
Toshiaki Makitabe294a52018-07-03 16:31:33 +0900711 }
Jason Wang03088132016-03-04 06:24:53 -0500712
713 preempt_enable();
714
Toshiaki Makita28b9b332018-07-03 16:31:31 +0900715 if (!vhost_vq_avail_empty(&net->dev, tvq)) {
716 vhost_poll_queue(&tvq->poll);
717 } else if (unlikely(vhost_enable_notify(&net->dev, tvq))) {
718 vhost_disable_notify(&net->dev, tvq);
719 vhost_poll_queue(&tvq->poll);
Jason Wang8b949be2017-09-05 09:22:05 +0800720 }
721
Toshiaki Makita28b9b332018-07-03 16:31:31 +0900722 mutex_unlock(&tvq->mutex);
Jason Wang03088132016-03-04 06:24:53 -0500723
Toshiaki Makita28b9b332018-07-03 16:31:31 +0900724 len = peek_head_len(rnvq, sk);
Jason Wang03088132016-03-04 06:24:53 -0500725 }
726
727 return len;
728}
729
David Stevens8dd014a2010-07-27 18:52:21 +0300730/* This is a multi-buffer version of vhost_get_desc, that works if
731 * vq has read descriptors only.
732 * @vq - the relevant virtqueue
733 * @datalen - data length we'll be reading
734 * @iovcount - returned count of io vectors we fill
735 * @log - vhost log
736 * @log_num - log offset
Jason Wang94249362011-01-17 16:11:08 +0800737 * @quota - headcount quota, 1 for big buffer
David Stevens8dd014a2010-07-27 18:52:21 +0300738 * returns number of buffer heads allocated, negative on error
739 */
740static int get_rx_bufs(struct vhost_virtqueue *vq,
741 struct vring_used_elem *heads,
742 int datalen,
743 unsigned *iovcount,
744 struct vhost_log *log,
Jason Wang94249362011-01-17 16:11:08 +0800745 unsigned *log_num,
746 unsigned int quota)
David Stevens8dd014a2010-07-27 18:52:21 +0300747{
748 unsigned int out, in;
749 int seg = 0;
750 int headcount = 0;
751 unsigned d;
752 int r, nlogs = 0;
Michael S. Tsirkin8b386942014-10-24 14:19:48 +0300753 /* len is always initialized before use since we are always called with
754 * datalen > 0.
755 */
756 u32 uninitialized_var(len);
David Stevens8dd014a2010-07-27 18:52:21 +0300757
Jason Wang94249362011-01-17 16:11:08 +0800758 while (datalen > 0 && headcount < quota) {
Jason Wange0e9b402010-09-14 23:53:05 +0800759 if (unlikely(seg >= UIO_MAXIOV)) {
David Stevens8dd014a2010-07-27 18:52:21 +0300760 r = -ENOBUFS;
761 goto err;
762 }
Michael S. Tsirkin47283be2014-06-05 15:20:27 +0300763 r = vhost_get_vq_desc(vq, vq->iov + seg,
David Stevens8dd014a2010-07-27 18:52:21 +0300764 ARRAY_SIZE(vq->iov) - seg, &out,
765 &in, log, log_num);
Michael S. Tsirkina39ee442014-03-27 12:53:37 +0200766 if (unlikely(r < 0))
767 goto err;
768
769 d = r;
David Stevens8dd014a2010-07-27 18:52:21 +0300770 if (d == vq->num) {
771 r = 0;
772 goto err;
773 }
774 if (unlikely(out || in <= 0)) {
775 vq_err(vq, "unexpected descriptor format for RX: "
776 "out %d, in %d\n", out, in);
777 r = -EINVAL;
778 goto err;
779 }
780 if (unlikely(log)) {
781 nlogs += *log_num;
782 log += *log_num;
783 }
Michael S. Tsirkin8b386942014-10-24 14:19:48 +0300784 heads[headcount].id = cpu_to_vhost32(vq, d);
785 len = iov_length(vq->iov + seg, in);
786 heads[headcount].len = cpu_to_vhost32(vq, len);
787 datalen -= len;
David Stevens8dd014a2010-07-27 18:52:21 +0300788 ++headcount;
789 seg += in;
790 }
Michael S. Tsirkin99975cc2015-01-07 10:51:00 +0200791 heads[headcount - 1].len = cpu_to_vhost32(vq, len + datalen);
David Stevens8dd014a2010-07-27 18:52:21 +0300792 *iovcount = seg;
793 if (unlikely(log))
794 *log_num = nlogs;
Michael S. Tsirkind8316f32014-03-27 12:00:26 +0200795
796 /* Detect overrun */
797 if (unlikely(datalen > 0)) {
798 r = UIO_MAXIOV + 1;
799 goto err;
800 }
David Stevens8dd014a2010-07-27 18:52:21 +0300801 return headcount;
802err:
803 vhost_discard_vq_desc(vq, headcount);
804 return r;
805}
806
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000807/* Expects to be always run from workqueue - which acts as
808 * read-size critical section for our kind of RCU. */
Jason Wang94249362011-01-17 16:11:08 +0800809static void handle_rx(struct vhost_net *net)
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000810{
Michael S. Tsirkin81f95a52013-04-28 15:51:40 +0300811 struct vhost_net_virtqueue *nvq = &net->vqs[VHOST_NET_VQ_RX];
812 struct vhost_virtqueue *vq = &nvq->vq;
David Stevens8dd014a2010-07-27 18:52:21 +0300813 unsigned uninitialized_var(in), log;
814 struct vhost_log *vq_log;
815 struct msghdr msg = {
816 .msg_name = NULL,
817 .msg_namelen = 0,
818 .msg_control = NULL, /* FIXME: get and handle RX aux data. */
819 .msg_controllen = 0,
David Stevens8dd014a2010-07-27 18:52:21 +0300820 .msg_flags = MSG_DONTWAIT,
821 };
Jason Wang0960b642015-02-15 16:35:17 +0800822 struct virtio_net_hdr hdr = {
823 .flags = 0,
824 .gso_type = VIRTIO_NET_HDR_GSO_NONE
David Stevens8dd014a2010-07-27 18:52:21 +0300825 };
David Stevens8dd014a2010-07-27 18:52:21 +0300826 size_t total_len = 0;
Michael S. Tsirkin910a5782012-10-24 20:37:51 +0200827 int err, mergeable;
Jason Wangf5a49412018-05-29 14:18:19 +0800828 s16 headcount;
David Stevens8dd014a2010-07-27 18:52:21 +0300829 size_t vhost_hlen, sock_hlen;
830 size_t vhost_len, sock_len;
Toshiaki Makitabe294a52018-07-03 16:31:33 +0900831 bool busyloop_intr = false;
Asias He2e26af72013-05-07 14:54:33 +0800832 struct socket *sock;
Al Viroba7438ae2014-12-10 15:51:28 -0500833 struct iov_iter fixup;
Jason Wang0960b642015-02-15 16:35:17 +0800834 __virtio16 num_buffers;
Paolo Abenidb688c22018-04-24 10:34:36 +0200835 int recv_pkts = 0;
David Stevens8dd014a2010-07-27 18:52:21 +0300836
Jason Wangaaa31492018-03-26 16:10:23 +0800837 mutex_lock_nested(&vq->mutex, 0);
Asias He2e26af72013-05-07 14:54:33 +0800838 sock = vq->private_data;
839 if (!sock)
840 goto out;
Jason Wang6b1e6cc2016-06-23 02:04:32 -0400841
842 if (!vq_iotlb_prefetch(vq))
843 goto out;
844
Michael S. Tsirkin8ea8cf82011-05-20 02:10:54 +0300845 vhost_disable_notify(&net->dev, vq);
Jason Wang8241a1e2016-06-01 01:56:33 -0400846 vhost_net_disable_vq(net, vq);
Asias He2e26af72013-05-07 14:54:33 +0800847
Michael S. Tsirkin81f95a52013-04-28 15:51:40 +0300848 vhost_hlen = nvq->vhost_hlen;
849 sock_hlen = nvq->sock_hlen;
David Stevens8dd014a2010-07-27 18:52:21 +0300850
Michael S. Tsirkinea16c512014-06-05 15:20:23 +0300851 vq_log = unlikely(vhost_has_feature(vq, VHOST_F_LOG_ALL)) ?
David Stevens8dd014a2010-07-27 18:52:21 +0300852 vq->log : NULL;
Michael S. Tsirkinea16c512014-06-05 15:20:23 +0300853 mergeable = vhost_has_feature(vq, VIRTIO_NET_F_MRG_RXBUF);
David Stevens8dd014a2010-07-27 18:52:21 +0300854
Toshiaki Makitabe294a52018-07-03 16:31:33 +0900855 while ((sock_len = vhost_net_rx_peek_head_len(net, sock->sk,
856 &busyloop_intr))) {
David Stevens8dd014a2010-07-27 18:52:21 +0300857 sock_len += sock_hlen;
858 vhost_len = sock_len + vhost_hlen;
Jason Wangf5a49412018-05-29 14:18:19 +0800859 headcount = get_rx_bufs(vq, vq->heads + nvq->done_idx,
860 vhost_len, &in, vq_log, &log,
Jason Wang94249362011-01-17 16:11:08 +0800861 likely(mergeable) ? UIO_MAXIOV : 1);
David Stevens8dd014a2010-07-27 18:52:21 +0300862 /* On error, stop handling until the next kick. */
863 if (unlikely(headcount < 0))
Jason Wang8241a1e2016-06-01 01:56:33 -0400864 goto out;
David Stevens8dd014a2010-07-27 18:52:21 +0300865 /* OK, now we need to know about added descriptors. */
866 if (!headcount) {
Toshiaki Makita6369fec2018-07-03 16:31:34 +0900867 if (unlikely(busyloop_intr)) {
868 vhost_poll_queue(&vq->poll);
869 } else if (unlikely(vhost_enable_notify(&net->dev, vq))) {
David Stevens8dd014a2010-07-27 18:52:21 +0300870 /* They have slipped one in as we were
871 * doing that: check again. */
Michael S. Tsirkin8ea8cf82011-05-20 02:10:54 +0300872 vhost_disable_notify(&net->dev, vq);
David Stevens8dd014a2010-07-27 18:52:21 +0300873 continue;
874 }
875 /* Nothing new? Wait for eventfd to tell us
876 * they refilled. */
Jason Wang8241a1e2016-06-01 01:56:33 -0400877 goto out;
David Stevens8dd014a2010-07-27 18:52:21 +0300878 }
Toshiaki Makita6369fec2018-07-03 16:31:34 +0900879 busyloop_intr = false;
Jason Wang5990a302018-01-04 11:14:27 +0800880 if (nvq->rx_ring)
Wei Xu6e474082017-12-01 05:10:36 -0500881 msg.msg_control = vhost_net_buf_consume(&nvq->rxq);
882 /* On overrun, truncate and discard */
883 if (unlikely(headcount > UIO_MAXIOV)) {
884 iov_iter_init(&msg.msg_iter, READ, vq->iov, 1, 1);
885 err = sock->ops->recvmsg(sock, &msg,
886 1, MSG_DONTWAIT | MSG_TRUNC);
887 pr_debug("Discarded rx packet: len %zd\n", sock_len);
888 continue;
889 }
David Stevens8dd014a2010-07-27 18:52:21 +0300890 /* We don't need to be notified again. */
Al Viroba7438ae2014-12-10 15:51:28 -0500891 iov_iter_init(&msg.msg_iter, READ, vq->iov, in, vhost_len);
892 fixup = msg.msg_iter;
893 if (unlikely((vhost_hlen))) {
894 /* We will supply the header ourselves
895 * TODO: support TSO.
896 */
897 iov_iter_advance(&msg.msg_iter, vhost_hlen);
Al Viroba7438ae2014-12-10 15:51:28 -0500898 }
Ying Xue1b784142015-03-02 15:37:48 +0800899 err = sock->ops->recvmsg(sock, &msg,
David Stevens8dd014a2010-07-27 18:52:21 +0300900 sock_len, MSG_DONTWAIT | MSG_TRUNC);
901 /* Userspace might have consumed the packet meanwhile:
902 * it's not supposed to do this usually, but might be hard
903 * to prevent. Discard data we got (if any) and keep going. */
904 if (unlikely(err != sock_len)) {
905 pr_debug("Discarded rx packet: "
906 " len %d, expected %zd\n", err, sock_len);
907 vhost_discard_vq_desc(vq, headcount);
908 continue;
909 }
Al Viroba7438ae2014-12-10 15:51:28 -0500910 /* Supply virtio_net_hdr if VHOST_NET_F_VIRTIO_NET_HDR */
Michael S. Tsirkin4c5a8442015-02-25 15:19:28 +0100911 if (unlikely(vhost_hlen)) {
912 if (copy_to_iter(&hdr, sizeof(hdr),
913 &fixup) != sizeof(hdr)) {
914 vq_err(vq, "Unable to write vnet_hdr "
915 "at addr %p\n", vq->iov->iov_base);
Jason Wang8241a1e2016-06-01 01:56:33 -0400916 goto out;
Michael S. Tsirkin4c5a8442015-02-25 15:19:28 +0100917 }
918 } else {
919 /* Header came from socket; we'll need to patch
920 * ->num_buffers over if VIRTIO_NET_F_MRG_RXBUF
921 */
922 iov_iter_advance(&fixup, sizeof(hdr));
David Stevens8dd014a2010-07-27 18:52:21 +0300923 }
924 /* TODO: Should check and handle checksum. */
Michael S. Tsirkin5201aa42015-02-03 11:07:06 +0200925
Jason Wang0960b642015-02-15 16:35:17 +0800926 num_buffers = cpu_to_vhost16(vq, headcount);
Jason Wangcfbdab92011-01-17 16:10:59 +0800927 if (likely(mergeable) &&
Michael S. Tsirkin0d79a492015-02-25 15:20:01 +0100928 copy_to_iter(&num_buffers, sizeof num_buffers,
929 &fixup) != sizeof num_buffers) {
David Stevens8dd014a2010-07-27 18:52:21 +0300930 vq_err(vq, "Failed num_buffers write");
931 vhost_discard_vq_desc(vq, headcount);
Jason Wang8241a1e2016-06-01 01:56:33 -0400932 goto out;
David Stevens8dd014a2010-07-27 18:52:21 +0300933 }
Jason Wangf5a49412018-05-29 14:18:19 +0800934 nvq->done_idx += headcount;
935 if (nvq->done_idx > VHOST_RX_BATCH)
936 vhost_rx_signal_used(nvq);
David Stevens8dd014a2010-07-27 18:52:21 +0300937 if (unlikely(vq_log))
938 vhost_log_write(vq, vq_log, log, vhost_len);
939 total_len += vhost_len;
Jason Wang272f35c2018-07-20 08:15:15 +0800940 if (unlikely(vhost_exceeds_weight(++recv_pkts, total_len))) {
David Stevens8dd014a2010-07-27 18:52:21 +0300941 vhost_poll_queue(&vq->poll);
Jason Wang8241a1e2016-06-01 01:56:33 -0400942 goto out;
David Stevens8dd014a2010-07-27 18:52:21 +0300943 }
944 }
Toshiaki Makitabe294a52018-07-03 16:31:33 +0900945 if (unlikely(busyloop_intr))
946 vhost_poll_queue(&vq->poll);
947 else
948 vhost_net_enable_vq(net, vq);
Asias He2e26af72013-05-07 14:54:33 +0800949out:
Jason Wangf5a49412018-05-29 14:18:19 +0800950 vhost_rx_signal_used(nvq);
David Stevens8dd014a2010-07-27 18:52:21 +0300951 mutex_unlock(&vq->mutex);
David Stevens8dd014a2010-07-27 18:52:21 +0300952}
953
Tejun Heoc23f34452010-06-02 20:40:00 +0200954static void handle_tx_kick(struct vhost_work *work)
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000955{
Tejun Heoc23f34452010-06-02 20:40:00 +0200956 struct vhost_virtqueue *vq = container_of(work, struct vhost_virtqueue,
957 poll.work);
958 struct vhost_net *net = container_of(vq->dev, struct vhost_net, dev);
959
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000960 handle_tx(net);
961}
962
Tejun Heoc23f34452010-06-02 20:40:00 +0200963static void handle_rx_kick(struct vhost_work *work)
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000964{
Tejun Heoc23f34452010-06-02 20:40:00 +0200965 struct vhost_virtqueue *vq = container_of(work, struct vhost_virtqueue,
966 poll.work);
967 struct vhost_net *net = container_of(vq->dev, struct vhost_net, dev);
968
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000969 handle_rx(net);
970}
971
Tejun Heoc23f34452010-06-02 20:40:00 +0200972static void handle_tx_net(struct vhost_work *work)
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000973{
Tejun Heoc23f34452010-06-02 20:40:00 +0200974 struct vhost_net *net = container_of(work, struct vhost_net,
975 poll[VHOST_NET_VQ_TX].work);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000976 handle_tx(net);
977}
978
Tejun Heoc23f34452010-06-02 20:40:00 +0200979static void handle_rx_net(struct vhost_work *work)
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000980{
Tejun Heoc23f34452010-06-02 20:40:00 +0200981 struct vhost_net *net = container_of(work, struct vhost_net,
982 poll[VHOST_NET_VQ_RX].work);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +0000983 handle_rx(net);
984}
985
986static int vhost_net_open(struct inode *inode, struct file *f)
987{
Michael S. Tsirkin23cc5a92013-01-23 21:46:47 +0100988 struct vhost_net *n;
Tejun Heoc23f34452010-06-02 20:40:00 +0200989 struct vhost_dev *dev;
Asias He3ab2e422013-04-27 11:16:48 +0800990 struct vhost_virtqueue **vqs;
Jason Wang5990a302018-01-04 11:14:27 +0800991 void **queue;
Zhi Yong Wu59566b6e2013-12-07 04:13:03 +0800992 int i;
Tejun Heoc23f34452010-06-02 20:40:00 +0200993
Michal Hockodcda9b02017-07-12 14:36:45 -0700994 n = kvmalloc(sizeof *n, GFP_KERNEL | __GFP_RETRY_MAYFAIL);
Michal Hocko6c5ab652017-05-08 15:57:15 -0700995 if (!n)
996 return -ENOMEM;
Kees Cook6da2ec52018-06-12 13:55:00 -0700997 vqs = kmalloc_array(VHOST_NET_VQ_MAX, sizeof(*vqs), GFP_KERNEL);
Asias He3ab2e422013-04-27 11:16:48 +0800998 if (!vqs) {
Romain Francoised04257b2014-06-12 10:42:34 +0200999 kvfree(n);
Asias He3ab2e422013-04-27 11:16:48 +08001000 return -ENOMEM;
1001 }
Tejun Heoc23f34452010-06-02 20:40:00 +02001002
Jason Wang5990a302018-01-04 11:14:27 +08001003 queue = kmalloc_array(VHOST_RX_BATCH, sizeof(void *),
Jason Wangc67df112017-05-17 12:14:45 +08001004 GFP_KERNEL);
1005 if (!queue) {
1006 kfree(vqs);
1007 kvfree(n);
1008 return -ENOMEM;
1009 }
1010 n->vqs[VHOST_NET_VQ_RX].rxq.queue = queue;
1011
Tejun Heoc23f34452010-06-02 20:40:00 +02001012 dev = &n->dev;
Asias He3ab2e422013-04-27 11:16:48 +08001013 vqs[VHOST_NET_VQ_TX] = &n->vqs[VHOST_NET_VQ_TX].vq;
1014 vqs[VHOST_NET_VQ_RX] = &n->vqs[VHOST_NET_VQ_RX].vq;
1015 n->vqs[VHOST_NET_VQ_TX].vq.handle_kick = handle_tx_kick;
1016 n->vqs[VHOST_NET_VQ_RX].vq.handle_kick = handle_rx_kick;
Asias He28394002013-04-27 15:07:46 +08001017 for (i = 0; i < VHOST_NET_VQ_MAX; i++) {
1018 n->vqs[i].ubufs = NULL;
1019 n->vqs[i].ubuf_info = NULL;
1020 n->vqs[i].upend_idx = 0;
1021 n->vqs[i].done_idx = 0;
Michael S. Tsirkin81f95a52013-04-28 15:51:40 +03001022 n->vqs[i].vhost_hlen = 0;
1023 n->vqs[i].sock_hlen = 0;
Alexander Potapenkoab7e34b2018-03-09 14:50:32 +08001024 n->vqs[i].rx_ring = NULL;
Jason Wangc67df112017-05-17 12:14:45 +08001025 vhost_net_buf_init(&n->vqs[i].rxq);
Asias He28394002013-04-27 15:07:46 +08001026 }
Zhi Yong Wu59566b6e2013-12-07 04:13:03 +08001027 vhost_dev_init(dev, vqs, VHOST_NET_VQ_MAX);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001028
Linus Torvaldsa9a08842018-02-11 14:34:03 -08001029 vhost_poll_init(n->poll + VHOST_NET_VQ_TX, handle_tx_net, EPOLLOUT, dev);
1030 vhost_poll_init(n->poll + VHOST_NET_VQ_RX, handle_rx_net, EPOLLIN, dev);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001031
1032 f->private_data = n;
1033
1034 return 0;
1035}
1036
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001037static struct socket *vhost_net_stop_vq(struct vhost_net *n,
1038 struct vhost_virtqueue *vq)
1039{
1040 struct socket *sock;
Jason Wangc67df112017-05-17 12:14:45 +08001041 struct vhost_net_virtqueue *nvq =
1042 container_of(vq, struct vhost_net_virtqueue, vq);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001043
1044 mutex_lock(&vq->mutex);
Asias He22fa90c2013-05-07 14:54:36 +08001045 sock = vq->private_data;
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001046 vhost_net_disable_vq(n, vq);
Asias He22fa90c2013-05-07 14:54:36 +08001047 vq->private_data = NULL;
Jason Wangc67df112017-05-17 12:14:45 +08001048 vhost_net_buf_unproduce(nvq);
Jason Wang303fd71b2018-03-09 14:50:33 +08001049 nvq->rx_ring = NULL;
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001050 mutex_unlock(&vq->mutex);
1051 return sock;
1052}
1053
1054static void vhost_net_stop(struct vhost_net *n, struct socket **tx_sock,
1055 struct socket **rx_sock)
1056{
Asias He3ab2e422013-04-27 11:16:48 +08001057 *tx_sock = vhost_net_stop_vq(n, &n->vqs[VHOST_NET_VQ_TX].vq);
1058 *rx_sock = vhost_net_stop_vq(n, &n->vqs[VHOST_NET_VQ_RX].vq);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001059}
1060
1061static void vhost_net_flush_vq(struct vhost_net *n, int index)
1062{
1063 vhost_poll_flush(n->poll + index);
Asias He3ab2e422013-04-27 11:16:48 +08001064 vhost_poll_flush(&n->vqs[index].vq.poll);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001065}
1066
1067static void vhost_net_flush(struct vhost_net *n)
1068{
1069 vhost_net_flush_vq(n, VHOST_NET_VQ_TX);
1070 vhost_net_flush_vq(n, VHOST_NET_VQ_RX);
Asias He28394002013-04-27 15:07:46 +08001071 if (n->vqs[VHOST_NET_VQ_TX].ubufs) {
Asias He3ab2e422013-04-27 11:16:48 +08001072 mutex_lock(&n->vqs[VHOST_NET_VQ_TX].vq.mutex);
Michael S. Tsirkin1280c272012-12-04 00:17:14 +02001073 n->tx_flush = true;
Asias He3ab2e422013-04-27 11:16:48 +08001074 mutex_unlock(&n->vqs[VHOST_NET_VQ_TX].vq.mutex);
Michael S. Tsirkin1280c272012-12-04 00:17:14 +02001075 /* Wait for all lower device DMAs done. */
Asias Hefe729a52013-05-06 16:38:24 +08001076 vhost_net_ubuf_put_and_wait(n->vqs[VHOST_NET_VQ_TX].ubufs);
Asias He3ab2e422013-04-27 11:16:48 +08001077 mutex_lock(&n->vqs[VHOST_NET_VQ_TX].vq.mutex);
Michael S. Tsirkin1280c272012-12-04 00:17:14 +02001078 n->tx_flush = false;
Michael S. Tsirkin0ad8b482014-02-13 11:42:05 +02001079 atomic_set(&n->vqs[VHOST_NET_VQ_TX].ubufs->refcount, 1);
Asias He3ab2e422013-04-27 11:16:48 +08001080 mutex_unlock(&n->vqs[VHOST_NET_VQ_TX].vq.mutex);
Michael S. Tsirkin1280c272012-12-04 00:17:14 +02001081 }
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001082}
1083
1084static int vhost_net_release(struct inode *inode, struct file *f)
1085{
1086 struct vhost_net *n = f->private_data;
1087 struct socket *tx_sock;
1088 struct socket *rx_sock;
1089
1090 vhost_net_stop(n, &tx_sock, &rx_sock);
1091 vhost_net_flush(n);
Michael S. Tsirkinb2116162012-11-01 09:16:46 +00001092 vhost_dev_stop(&n->dev);
夷则(Caspar)f6f93f72017-12-25 00:08:58 +08001093 vhost_dev_cleanup(&n->dev);
Michael S. Tsirkin81f95a52013-04-28 15:51:40 +03001094 vhost_net_vq_reset(n);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001095 if (tx_sock)
Al Viro09aaacf2014-03-05 20:39:00 -05001096 sockfd_put(tx_sock);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001097 if (rx_sock)
Al Viro09aaacf2014-03-05 20:39:00 -05001098 sockfd_put(rx_sock);
Michael S. Tsirkinb0c057c2014-02-13 11:45:11 +02001099 /* Make sure no callbacks are outstanding */
1100 synchronize_rcu_bh();
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001101 /* We do an extra flush before freeing memory,
1102 * since jobs can re-queue themselves. */
1103 vhost_net_flush(n);
Jason Wangc67df112017-05-17 12:14:45 +08001104 kfree(n->vqs[VHOST_NET_VQ_RX].rxq.queue);
Asias He3ab2e422013-04-27 11:16:48 +08001105 kfree(n->dev.vqs);
Romain Francoised04257b2014-06-12 10:42:34 +02001106 kvfree(n);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001107 return 0;
1108}
1109
1110static struct socket *get_raw_socket(int fd)
1111{
1112 struct {
1113 struct sockaddr_ll sa;
1114 char buf[MAX_ADDR_LEN];
1115 } uaddr;
Denys Vlasenko9b2c45d2018-02-12 20:00:20 +01001116 int r;
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001117 struct socket *sock = sockfd_lookup(fd, &r);
Krishna Kumard47effe2011-03-01 17:06:37 +05301118
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001119 if (!sock)
1120 return ERR_PTR(-ENOTSOCK);
1121
1122 /* Parameter checking */
1123 if (sock->sk->sk_type != SOCK_RAW) {
1124 r = -ESOCKTNOSUPPORT;
1125 goto err;
1126 }
1127
Denys Vlasenko9b2c45d2018-02-12 20:00:20 +01001128 r = sock->ops->getname(sock, (struct sockaddr *)&uaddr.sa, 0);
1129 if (r < 0)
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001130 goto err;
1131
1132 if (uaddr.sa.sll_family != AF_PACKET) {
1133 r = -EPFNOSUPPORT;
1134 goto err;
1135 }
1136 return sock;
1137err:
Al Viro09aaacf2014-03-05 20:39:00 -05001138 sockfd_put(sock);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001139 return ERR_PTR(r);
1140}
1141
Jason Wang5990a302018-01-04 11:14:27 +08001142static struct ptr_ring *get_tap_ptr_ring(int fd)
Jason Wangc67df112017-05-17 12:14:45 +08001143{
Jason Wang5990a302018-01-04 11:14:27 +08001144 struct ptr_ring *ring;
Jason Wangc67df112017-05-17 12:14:45 +08001145 struct file *file = fget(fd);
1146
1147 if (!file)
1148 return NULL;
Jason Wang5990a302018-01-04 11:14:27 +08001149 ring = tun_get_tx_ring(file);
1150 if (!IS_ERR(ring))
Jason Wangc67df112017-05-17 12:14:45 +08001151 goto out;
Jason Wang5990a302018-01-04 11:14:27 +08001152 ring = tap_get_ptr_ring(file);
1153 if (!IS_ERR(ring))
Jason Wangc67df112017-05-17 12:14:45 +08001154 goto out;
Jason Wang5990a302018-01-04 11:14:27 +08001155 ring = NULL;
Jason Wangc67df112017-05-17 12:14:45 +08001156out:
1157 fput(file);
Jason Wang5990a302018-01-04 11:14:27 +08001158 return ring;
Jason Wangc67df112017-05-17 12:14:45 +08001159}
1160
Arnd Bergmann501c7742010-02-18 05:46:50 +00001161static struct socket *get_tap_socket(int fd)
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001162{
1163 struct file *file = fget(fd);
1164 struct socket *sock;
Krishna Kumard47effe2011-03-01 17:06:37 +05301165
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001166 if (!file)
1167 return ERR_PTR(-EBADF);
1168 sock = tun_get_socket(file);
Arnd Bergmann501c7742010-02-18 05:46:50 +00001169 if (!IS_ERR(sock))
1170 return sock;
Sainath Grandhi635b8c82017-02-10 16:03:47 -08001171 sock = tap_get_socket(file);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001172 if (IS_ERR(sock))
1173 fput(file);
1174 return sock;
1175}
1176
1177static struct socket *get_socket(int fd)
1178{
1179 struct socket *sock;
Krishna Kumard47effe2011-03-01 17:06:37 +05301180
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001181 /* special case to disable backend */
1182 if (fd == -1)
1183 return NULL;
1184 sock = get_raw_socket(fd);
1185 if (!IS_ERR(sock))
1186 return sock;
Arnd Bergmann501c7742010-02-18 05:46:50 +00001187 sock = get_tap_socket(fd);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001188 if (!IS_ERR(sock))
1189 return sock;
1190 return ERR_PTR(-ENOTSOCK);
1191}
1192
1193static long vhost_net_set_backend(struct vhost_net *n, unsigned index, int fd)
1194{
1195 struct socket *sock, *oldsock;
1196 struct vhost_virtqueue *vq;
Asias He28394002013-04-27 15:07:46 +08001197 struct vhost_net_virtqueue *nvq;
Asias Hefe729a52013-05-06 16:38:24 +08001198 struct vhost_net_ubuf_ref *ubufs, *oldubufs = NULL;
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001199 int r;
1200
1201 mutex_lock(&n->dev.mutex);
1202 r = vhost_dev_check_owner(&n->dev);
1203 if (r)
1204 goto err;
1205
1206 if (index >= VHOST_NET_VQ_MAX) {
1207 r = -ENOBUFS;
1208 goto err;
1209 }
Asias He3ab2e422013-04-27 11:16:48 +08001210 vq = &n->vqs[index].vq;
Asias He28394002013-04-27 15:07:46 +08001211 nvq = &n->vqs[index];
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001212 mutex_lock(&vq->mutex);
1213
1214 /* Verify that ring has been setup correctly. */
1215 if (!vhost_vq_access_ok(vq)) {
1216 r = -EFAULT;
Jeff Dike1dace8c2010-03-04 16:10:14 -05001217 goto err_vq;
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001218 }
1219 sock = get_socket(fd);
1220 if (IS_ERR(sock)) {
1221 r = PTR_ERR(sock);
Jeff Dike1dace8c2010-03-04 16:10:14 -05001222 goto err_vq;
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001223 }
1224
1225 /* start polling new socket */
Asias He22fa90c2013-05-07 14:54:36 +08001226 oldsock = vq->private_data;
David S. Miller11fe8832010-07-20 18:25:24 -07001227 if (sock != oldsock) {
Asias Hefe729a52013-05-06 16:38:24 +08001228 ubufs = vhost_net_ubuf_alloc(vq,
1229 sock && vhost_sock_zcopy(sock));
Michael S. Tsirkinbab632d2011-07-18 03:48:46 +00001230 if (IS_ERR(ubufs)) {
1231 r = PTR_ERR(ubufs);
1232 goto err_ubufs;
1233 }
Jason Wang692a9982013-01-28 01:05:17 +00001234
Krishna Kumard47effe2011-03-01 17:06:37 +05301235 vhost_net_disable_vq(n, vq);
Asias He22fa90c2013-05-07 14:54:36 +08001236 vq->private_data = sock;
Jason Wangc67df112017-05-17 12:14:45 +08001237 vhost_net_buf_unproduce(nvq);
Greg Kurz80f7d032016-02-16 15:59:44 +01001238 r = vhost_vq_init_access(vq);
Jason Wangf59281d2011-06-21 18:04:27 +08001239 if (r)
Jason Wang692a9982013-01-28 01:05:17 +00001240 goto err_used;
Jason Wang2b8b3282013-01-28 01:05:18 +00001241 r = vhost_net_enable_vq(n, vq);
1242 if (r)
1243 goto err_used;
Jason Wang303fd71b2018-03-09 14:50:33 +08001244 if (index == VHOST_NET_VQ_RX)
1245 nvq->rx_ring = get_tap_ptr_ring(fd);
Jason Wang692a9982013-01-28 01:05:17 +00001246
Asias He28394002013-04-27 15:07:46 +08001247 oldubufs = nvq->ubufs;
1248 nvq->ubufs = ubufs;
Michael S. Tsirkin64e9a9b2012-12-03 07:31:51 +00001249
1250 n->tx_packets = 0;
1251 n->tx_zcopy_err = 0;
Michael S. Tsirkin1280c272012-12-04 00:17:14 +02001252 n->tx_flush = false;
Jeff Dikedd1f4072010-03-04 16:10:14 -05001253 }
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001254
Michael S. Tsirkin1680e902010-07-15 15:19:12 +03001255 mutex_unlock(&vq->mutex);
1256
Michael S. Tsirkinc047e5f2011-07-20 13:41:31 +03001257 if (oldubufs) {
Michael S. Tsirkinc38e39c2013-06-25 17:29:46 +03001258 vhost_net_ubuf_put_wait_and_free(oldubufs);
Michael S. Tsirkinc047e5f2011-07-20 13:41:31 +03001259 mutex_lock(&vq->mutex);
Michael S. Tsirkineaae8132012-11-01 09:16:51 +00001260 vhost_zerocopy_signal_used(n, vq);
Michael S. Tsirkinc047e5f2011-07-20 13:41:31 +03001261 mutex_unlock(&vq->mutex);
1262 }
Michael S. Tsirkinbab632d2011-07-18 03:48:46 +00001263
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001264 if (oldsock) {
1265 vhost_net_flush_vq(n, index);
Al Viro09aaacf2014-03-05 20:39:00 -05001266 sockfd_put(oldsock);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001267 }
Jeff Dike1dace8c2010-03-04 16:10:14 -05001268
Michael S. Tsirkin1680e902010-07-15 15:19:12 +03001269 mutex_unlock(&n->dev.mutex);
1270 return 0;
1271
Jason Wang692a9982013-01-28 01:05:17 +00001272err_used:
Asias He22fa90c2013-05-07 14:54:36 +08001273 vq->private_data = oldsock;
Jason Wang692a9982013-01-28 01:05:17 +00001274 vhost_net_enable_vq(n, vq);
1275 if (ubufs)
Michael S. Tsirkinc38e39c2013-06-25 17:29:46 +03001276 vhost_net_ubuf_put_wait_and_free(ubufs);
Michael S. Tsirkinbab632d2011-07-18 03:48:46 +00001277err_ubufs:
Jason Wangb8f1f652018-06-21 13:11:31 +08001278 if (sock)
1279 sockfd_put(sock);
Jeff Dike1dace8c2010-03-04 16:10:14 -05001280err_vq:
1281 mutex_unlock(&vq->mutex);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001282err:
1283 mutex_unlock(&n->dev.mutex);
1284 return r;
1285}
1286
1287static long vhost_net_reset_owner(struct vhost_net *n)
1288{
1289 struct socket *tx_sock = NULL;
1290 struct socket *rx_sock = NULL;
1291 long err;
Jason Wanga9709d62016-06-23 02:04:31 -04001292 struct vhost_umem *umem;
Krishna Kumard47effe2011-03-01 17:06:37 +05301293
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001294 mutex_lock(&n->dev.mutex);
1295 err = vhost_dev_check_owner(&n->dev);
1296 if (err)
1297 goto done;
Jason Wanga9709d62016-06-23 02:04:31 -04001298 umem = vhost_dev_reset_owner_prepare();
1299 if (!umem) {
Michael S. Tsirkin150b9e52013-04-28 17:12:08 +03001300 err = -ENOMEM;
1301 goto done;
1302 }
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001303 vhost_net_stop(n, &tx_sock, &rx_sock);
1304 vhost_net_flush(n);
Jason Wang4cd87952018-01-25 22:03:52 +08001305 vhost_dev_stop(&n->dev);
Jason Wanga9709d62016-06-23 02:04:31 -04001306 vhost_dev_reset_owner(&n->dev, umem);
Michael S. Tsirkin81f95a52013-04-28 15:51:40 +03001307 vhost_net_vq_reset(n);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001308done:
1309 mutex_unlock(&n->dev.mutex);
1310 if (tx_sock)
Al Viro09aaacf2014-03-05 20:39:00 -05001311 sockfd_put(tx_sock);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001312 if (rx_sock)
Al Viro09aaacf2014-03-05 20:39:00 -05001313 sockfd_put(rx_sock);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001314 return err;
1315}
1316
1317static int vhost_net_set_features(struct vhost_net *n, u64 features)
1318{
David Stevens8dd014a2010-07-27 18:52:21 +03001319 size_t vhost_hlen, sock_hlen, hdr_len;
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001320 int i;
David Stevens8dd014a2010-07-27 18:52:21 +03001321
Michael S. Tsirkine4fca7d2014-10-24 14:23:52 +03001322 hdr_len = (features & ((1ULL << VIRTIO_NET_F_MRG_RXBUF) |
1323 (1ULL << VIRTIO_F_VERSION_1))) ?
David Stevens8dd014a2010-07-27 18:52:21 +03001324 sizeof(struct virtio_net_hdr_mrg_rxbuf) :
1325 sizeof(struct virtio_net_hdr);
1326 if (features & (1 << VHOST_NET_F_VIRTIO_NET_HDR)) {
1327 /* vhost provides vnet_hdr */
1328 vhost_hlen = hdr_len;
1329 sock_hlen = 0;
1330 } else {
1331 /* socket provides vnet_hdr */
1332 vhost_hlen = 0;
1333 sock_hlen = hdr_len;
1334 }
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001335 mutex_lock(&n->dev.mutex);
1336 if ((features & (1 << VHOST_F_LOG_ALL)) &&
Jason Wang6b1e6cc2016-06-23 02:04:32 -04001337 !vhost_log_access_ok(&n->dev))
1338 goto out_unlock;
1339
1340 if ((features & (1ULL << VIRTIO_F_IOMMU_PLATFORM))) {
1341 if (vhost_init_device_iotlb(&n->dev, true))
1342 goto out_unlock;
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001343 }
Jason Wang6b1e6cc2016-06-23 02:04:32 -04001344
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001345 for (i = 0; i < VHOST_NET_VQ_MAX; ++i) {
Asias He3ab2e422013-04-27 11:16:48 +08001346 mutex_lock(&n->vqs[i].vq.mutex);
Michael S. Tsirkinea16c512014-06-05 15:20:23 +03001347 n->vqs[i].vq.acked_features = features;
Michael S. Tsirkin81f95a52013-04-28 15:51:40 +03001348 n->vqs[i].vhost_hlen = vhost_hlen;
1349 n->vqs[i].sock_hlen = sock_hlen;
Asias He3ab2e422013-04-27 11:16:48 +08001350 mutex_unlock(&n->vqs[i].vq.mutex);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001351 }
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001352 mutex_unlock(&n->dev.mutex);
1353 return 0;
Jason Wang6b1e6cc2016-06-23 02:04:32 -04001354
1355out_unlock:
1356 mutex_unlock(&n->dev.mutex);
1357 return -EFAULT;
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001358}
1359
Asias Heb1ad8492013-05-06 11:16:00 +08001360static long vhost_net_set_owner(struct vhost_net *n)
1361{
1362 int r;
1363
1364 mutex_lock(&n->dev.mutex);
Michael S. Tsirkin05c05352013-06-06 15:20:39 +03001365 if (vhost_dev_has_owner(&n->dev)) {
1366 r = -EBUSY;
1367 goto out;
1368 }
Asias Heb1ad8492013-05-06 11:16:00 +08001369 r = vhost_net_set_ubuf_info(n);
1370 if (r)
1371 goto out;
1372 r = vhost_dev_set_owner(&n->dev);
1373 if (r)
1374 vhost_net_clear_ubuf_info(n);
1375 vhost_net_flush(n);
1376out:
1377 mutex_unlock(&n->dev.mutex);
1378 return r;
1379}
1380
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001381static long vhost_net_ioctl(struct file *f, unsigned int ioctl,
1382 unsigned long arg)
1383{
1384 struct vhost_net *n = f->private_data;
1385 void __user *argp = (void __user *)arg;
1386 u64 __user *featurep = argp;
1387 struct vhost_vring_file backend;
1388 u64 features;
1389 int r;
Krishna Kumard47effe2011-03-01 17:06:37 +05301390
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001391 switch (ioctl) {
1392 case VHOST_NET_SET_BACKEND:
Takuya Yoshikawad3553a52010-05-27 19:01:58 +09001393 if (copy_from_user(&backend, argp, sizeof backend))
1394 return -EFAULT;
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001395 return vhost_net_set_backend(n, backend.index, backend.fd);
1396 case VHOST_GET_FEATURES:
Stefan Hajnoczi0dd05a32012-07-21 06:55:36 +00001397 features = VHOST_NET_FEATURES;
Takuya Yoshikawad3553a52010-05-27 19:01:58 +09001398 if (copy_to_user(featurep, &features, sizeof features))
1399 return -EFAULT;
1400 return 0;
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001401 case VHOST_SET_FEATURES:
Takuya Yoshikawad3553a52010-05-27 19:01:58 +09001402 if (copy_from_user(&features, featurep, sizeof features))
1403 return -EFAULT;
Stefan Hajnoczi0dd05a32012-07-21 06:55:36 +00001404 if (features & ~VHOST_NET_FEATURES)
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001405 return -EOPNOTSUPP;
1406 return vhost_net_set_features(n, features);
1407 case VHOST_RESET_OWNER:
1408 return vhost_net_reset_owner(n);
Asias Heb1ad8492013-05-06 11:16:00 +08001409 case VHOST_SET_OWNER:
1410 return vhost_net_set_owner(n);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001411 default:
1412 mutex_lock(&n->dev.mutex);
Michael S. Tsirkin935cdee2012-12-06 14:03:34 +02001413 r = vhost_dev_ioctl(&n->dev, ioctl, argp);
1414 if (r == -ENOIOCTLCMD)
1415 r = vhost_vring_ioctl(&n->dev, ioctl, argp);
1416 else
1417 vhost_net_flush(n);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001418 mutex_unlock(&n->dev.mutex);
1419 return r;
1420 }
1421}
1422
1423#ifdef CONFIG_COMPAT
1424static long vhost_net_compat_ioctl(struct file *f, unsigned int ioctl,
1425 unsigned long arg)
1426{
1427 return vhost_net_ioctl(f, ioctl, (unsigned long)compat_ptr(arg));
1428}
1429#endif
1430
Jason Wang6b1e6cc2016-06-23 02:04:32 -04001431static ssize_t vhost_net_chr_read_iter(struct kiocb *iocb, struct iov_iter *to)
1432{
1433 struct file *file = iocb->ki_filp;
1434 struct vhost_net *n = file->private_data;
1435 struct vhost_dev *dev = &n->dev;
1436 int noblock = file->f_flags & O_NONBLOCK;
1437
1438 return vhost_chr_read_iter(dev, to, noblock);
1439}
1440
1441static ssize_t vhost_net_chr_write_iter(struct kiocb *iocb,
1442 struct iov_iter *from)
1443{
1444 struct file *file = iocb->ki_filp;
1445 struct vhost_net *n = file->private_data;
1446 struct vhost_dev *dev = &n->dev;
1447
1448 return vhost_chr_write_iter(dev, from);
1449}
1450
Al Viroafc9a422017-07-03 06:39:46 -04001451static __poll_t vhost_net_chr_poll(struct file *file, poll_table *wait)
Jason Wang6b1e6cc2016-06-23 02:04:32 -04001452{
1453 struct vhost_net *n = file->private_data;
1454 struct vhost_dev *dev = &n->dev;
1455
1456 return vhost_chr_poll(file, dev, wait);
1457}
1458
Tobias Klauser373a83a2010-05-17 15:12:49 +02001459static const struct file_operations vhost_net_fops = {
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001460 .owner = THIS_MODULE,
1461 .release = vhost_net_release,
Jason Wang6b1e6cc2016-06-23 02:04:32 -04001462 .read_iter = vhost_net_chr_read_iter,
1463 .write_iter = vhost_net_chr_write_iter,
1464 .poll = vhost_net_chr_poll,
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001465 .unlocked_ioctl = vhost_net_ioctl,
1466#ifdef CONFIG_COMPAT
1467 .compat_ioctl = vhost_net_compat_ioctl,
1468#endif
1469 .open = vhost_net_open,
Arnd Bergmann6038f372010-08-15 18:52:59 +02001470 .llseek = noop_llseek,
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001471};
1472
1473static struct miscdevice vhost_net_misc = {
stephen hemminger7c7c7f02012-01-11 19:30:38 +00001474 .minor = VHOST_NET_MINOR,
1475 .name = "vhost-net",
1476 .fops = &vhost_net_fops,
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001477};
1478
Christoph Hellwiga8d37822010-04-13 14:11:25 -04001479static int vhost_net_init(void)
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001480{
Michael S. Tsirkinbab632d2011-07-18 03:48:46 +00001481 if (experimental_zcopytx)
Asias Hefe729a52013-05-06 16:38:24 +08001482 vhost_net_enable_zcopy(VHOST_NET_VQ_TX);
Tejun Heoc23f34452010-06-02 20:40:00 +02001483 return misc_register(&vhost_net_misc);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001484}
1485module_init(vhost_net_init);
1486
Christoph Hellwiga8d37822010-04-13 14:11:25 -04001487static void vhost_net_exit(void)
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001488{
1489 misc_deregister(&vhost_net_misc);
Michael S. Tsirkin3a4d5c92010-01-14 06:17:27 +00001490}
1491module_exit(vhost_net_exit);
1492
1493MODULE_VERSION("0.0.1");
1494MODULE_LICENSE("GPL v2");
1495MODULE_AUTHOR("Michael S. Tsirkin");
1496MODULE_DESCRIPTION("Host kernel accelerator for virtio net");
stephen hemminger7c7c7f02012-01-11 19:30:38 +00001497MODULE_ALIAS_MISCDEV(VHOST_NET_MINOR);
1498MODULE_ALIAS("devname:vhost-net");