On Fri, Aug 18, 2017 at 04:36:06PM +0800, Wei Wang wrote: > On 08/18/2017 10:28 AM, Michael S. Tsirkin wrote: > > On Thu, Aug 17, 2017 at 11:26:56AM +0800, Wei Wang wrote: > > > Add a new vq to report hints of guest free pages to the host. > > > > > > Signed-off-by: Wei Wang <wei.w.wang@xxxxxxxxx> > > > Signed-off-by: Liang Li <liang.z.li@xxxxxxxxx> > > > --- > > > drivers/virtio/virtio_balloon.c | 167 +++++++++++++++++++++++++++++++----- > > > include/uapi/linux/virtio_balloon.h | 1 + > > > 2 files changed, 147 insertions(+), 21 deletions(-) > > > > > > diff --git a/drivers/virtio/virtio_balloon.c b/drivers/virtio/virtio_balloon.c > > > index 72041b4..e6755bc 100644 > > > --- a/drivers/virtio/virtio_balloon.c > > > +++ b/drivers/virtio/virtio_balloon.c > > > @@ -54,11 +54,12 @@ static struct vfsmount *balloon_mnt; > > > struct virtio_balloon { > > > struct virtio_device *vdev; > > > - struct virtqueue *inflate_vq, *deflate_vq, *stats_vq; > > > + struct virtqueue *inflate_vq, *deflate_vq, *stats_vq, *free_page_vq; > > > /* The balloon servicing is delegated to a freezable workqueue. */ > > > struct work_struct update_balloon_stats_work; > > > struct work_struct update_balloon_size_work; > > > + struct work_struct report_free_page_work; > > > /* Prevent updating balloon when it is being canceled. */ > > > spinlock_t stop_update_lock; > > > @@ -90,6 +91,13 @@ struct virtio_balloon { > > > /* Memory statistics */ > > > struct virtio_balloon_stat stats[VIRTIO_BALLOON_S_NR]; > > > + /* > > > + * Used by the device and driver to signal each other. > > > + * device->driver: start the free page report. > > > + * driver->device: end the free page report. > > > + */ > > > + __virtio32 report_free_page_signal; > > > + > > > /* To register callback in oom notifier call chain */ > > > struct notifier_block nb; > > > }; > > > @@ -174,6 +182,17 @@ static void send_balloon_page_sg(struct virtio_balloon *vb, > > > } while (unlikely(ret == -ENOSPC)); > > > } > > > +static void send_free_page_sg(struct virtqueue *vq, void *addr, uint32_t size) > > > +{ > > > + unsigned int len; > > > + > > > + add_one_sg(vq, addr, size); > > > + virtqueue_kick(vq); > > > + /* Release entries if there are */ > > > + while (virtqueue_get_buf(vq, &len)) > > > + ; > > > +} > > > + > > > /* > > > * Send balloon pages in sgs to host. The balloon pages are recorded in the > > > * page xbitmap. Each bit in the bitmap corresponds to a page of PAGE_SIZE. > > > @@ -511,42 +530,143 @@ static void update_balloon_size_func(struct work_struct *work) > > > queue_work(system_freezable_wq, work); > > > } > > > +static void virtio_balloon_send_free_pages(void *opaque, unsigned long pfn, > > > + unsigned long nr_pages) > > > +{ > > > + struct virtio_balloon *vb = (struct virtio_balloon *)opaque; > > > + void *addr = (void *)pfn_to_kaddr(pfn); > > > + uint32_t len = nr_pages << PAGE_SHIFT; > > > + > > > + send_free_page_sg(vb->free_page_vq, addr, len); > > > +} > > > + > > > +static void report_free_page_completion(struct virtio_balloon *vb) > > > +{ > > > + struct virtqueue *vq = vb->free_page_vq; > > > + struct scatterlist sg; > > > + unsigned int len; > > > + int ret; > > > + > > > + sg_init_one(&sg, &vb->report_free_page_signal, sizeof(__virtio32)); > > > +retry: > > > + ret = virtqueue_add_outbuf(vq, &sg, 1, vb, GFP_KERNEL); > > > + virtqueue_kick(vq); > > > + if (unlikely(ret == -ENOSPC)) { > > > + wait_event(vb->acked, virtqueue_get_buf(vq, &len)); > > > + goto retry; > > > + } > > > +} > > So the annoying thing here is that once this starts going, > > it will keep sending free pages from the list even if > > host is no longer interested. There should be a way > > for host to tell guest "stop" or "start from the beginning". > > This can be achieved via two output signal buf here: > signal_buf_start: filled with VIRTIO_BALLOON_F_FREE_PAGE_REPORT_START > signal_buf_end: filled with VIRTIO_BALLOON_F_FREE_PAGE_REPORT_END > > The device holds both, and can put one of them to the vq and notify. Do you mean device writes start and end in the buf? then it's an inbuf not an outbuf. > > > > > > It's the result of using same vq for guest to host and > > host to guest communication, and I think it's not a great idea. > > I'd reuse stats vq for host to guest requests maybe. > > > > > As we discussed before, we can't have a vq interleave the report of stats > and free pages. > The vq will be locked when one command is in use. So, when live migration > starts, the > periodically reported stats will be delayed. > Would this be OK? Or would you > like to have > one host to guest vq, and multiple host to guest vqs? That is, > > - host to guest: > CMD_VQ > > - guest to host: > STATS_REPORT_VQ > FREE_PAGE_VQ > > > Best, > Wei > Point is stats report vq is also host to guest. So I think it can be combined with CMD VQ. If it's too hard a separate vq isn't too bad though. -- MST