/* SPDX-License-Identifier: GPL-2.0 */
/* Copyright(c) 2023 Advanced Micro Devices, Inc. */

#ifndef _DIRTY_H_
#define _DIRTY_H_

struct pds_vfio_region {
 unsigned long *host_seq;
 unsigned long *host_ack;
 u64 bmp_bytes;
 u64 size;
 u64 start;
 u64 page_size;
 struct pds_lm_sg_elem *sgl;
 dma_addr_t sgl_addr;
 u32 dev_bmp_offset_start_byte;
 u16 num_sge;
};

struct pds_vfio_dirty {
 struct pds_vfio_region *regions;
 u8 num_regions;
 bool is_enabled;
};

struct pds_vfio_pci_device;

bool pds_vfio_dirty_is_enabled(struct pds_vfio_pci_device *pds_vfio);
void pds_vfio_dirty_set_enabled(struct pds_vfio_pci_device *pds_vfio);
void pds_vfio_dirty_set_disabled(struct pds_vfio_pci_device *pds_vfio);
void pds_vfio_dirty_disable(struct pds_vfio_pci_device *pds_vfio,
       bool send_cmd);

int pds_vfio_dma_logging_report(struct vfio_device *vdev, unsigned long iova,
    unsigned long length,
    struct iova_bitmap *dirty);
int pds_vfio_dma_logging_start(struct vfio_device *vdev,
          struct rb_root_cached *ranges, u32 nnodes,
          u64 *page_size);
int pds_vfio_dma_logging_stop(struct vfio_device *vdev);
#endif /* _DIRTY_H_ */