On Mon, 10 Feb 2020 04:49:54 -0500
Yan Zhao <yan.y.zhao@xxxxxxxxx> wrote:
> On Sat, Feb 08, 2020 at 03:42:31AM +0800, Kirti Wankhede wrote:
> > VFIO_IOMMU_DIRTY_PAGES ioctl performs three operations:
> > - Start pinned and unpinned pages tracking while migration is active
> > - Stop pinned and unpinned dirty pages tracking. This is also used to
> > stop dirty pages tracking if migration failed or cancelled.
> > - Get dirty pages bitmap. This ioctl returns bitmap of dirty pages, its
> > user space application responsibility to copy content of dirty pages
> > from source to destination during migration.
> >
> > To prevent DoS attack, memory for bitmap is allocated per vfio_dma
> > structure. Bitmap size is calculated considering smallest supported page
> > size. Bitmap is allocated when dirty logging is enabled for those
> > vfio_dmas whose vpfn list is not empty or whole range is mapped, in
> > case of pass-through device.
> >
> > There could be multiple option as to when bitmap should be populated:
> > * Polulate bitmap for already pinned pages when bitmap is allocated for
> > a vfio_dma with the smallest supported page size. Updates bitmap from
> > page pinning and unpinning functions. When user application queries
> > bitmap, check if requested page size is same as page size used to
> > populated bitmap. If it is equal, copy bitmap. But if not equal,
> > re-populated bitmap according to requested page size and then copy to
> > user.
> > Pros: Bitmap gets populated on the fly after dirty tracking has
> > started.
> > Cons: If requested page size is different than smallest supported
> > page size, then bitmap has to be re-populated again, with
> > additional overhead of allocating bitmap memory again for
> > re-population of bitmap.
> >
> > * Populate bitmap when bitmap is queried by user application.
> > Pros: Bitmap is populated with requested page size. This eliminates
> > the need to re-populate bitmap if requested page size is
> > different than smallest supported pages size.
> > Cons: There is one time processing time, when bitmap is queried.
> >
> > I prefer later option with simple logic and to eliminate over-head of
> > bitmap repopulation in case of differnt page sizes. Later option is
> > implemented in this patch.
> >
> > Signed-off-by: Kirti Wankhede <kwankhede@xxxxxxxxxx>
> > Reviewed-by: Neo Jia <cjia@xxxxxxxxxx>
> > ---
> > drivers/vfio/vfio_iommu_type1.c | 299 ++++++++++++++++++++++++++++++++++++++--
> > 1 file changed, 287 insertions(+), 12 deletions(-)
> >
> > diff --git a/drivers/vfio/vfio_iommu_type1.c b/drivers/vfio/vfio_iommu_type1.c
> > index d386461e5d11..df358dc1c85b 100644
> > --- a/drivers/vfio/vfio_iommu_type1.c
> > +++ b/drivers/vfio/vfio_iommu_type1.c
[snip]
> > @@ -830,6 +924,113 @@ static unsigned long vfio_pgsize_bitmap(struct vfio_iommu *iommu)
> > return bitmap;
> > }
> >
> > +static int vfio_iova_dirty_bitmap(struct vfio_iommu *iommu, dma_addr_t iova,
> > + size_t size, uint64_t pgsize,
> > + unsigned char __user *bitmap)
> > +{
> > + struct vfio_dma *dma;
> > + dma_addr_t i = iova, iova_limit;
> > + unsigned int bsize, nbits = 0, l = 0;
> > + unsigned long pgshift = __ffs(pgsize);
> > +
> > + while ((dma = vfio_find_dma(iommu, i, pgsize))) {
> > + int ret, j;
> > + unsigned int npages = 0, shift = 0;
> > + unsigned char temp = 0;
> > +
> > + /* mark all pages dirty if all pages are pinned and mapped. */
> > + if (dma->iommu_mapped) {
> > + iova_limit = min(dma->iova + dma->size, iova + size);
> > + npages = iova_limit/pgsize;
> > + bitmap_set(dma->bitmap, 0, npages);
> for pass-through devices, it's not good to always return all pinned pages as
> dirty. could it also call vfio_pin_pages to track dirty pages? or any
> other interface provided to do that?
See patch 7/7. Thanks,
Alex
