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

#ifndef _LM_H_
#define _LM_H_

#include <linux/fs.h>
#include <linux/mutex.h>
#include <linux/scatterlist.h>
#include <linux/types.h>

#include <linux/pds/pds_common.h>
#include <linux/pds/pds_adminq.h>

struct pds_vfio_lm_file {
        struct file *filep;
        struct mutex lock;      /* protect live migration data file */
        u64 size;               /* Size with valid data */
        u64 alloc_size;         /* Total allocated size. Always >= len */
        void *page_mem;         /* memory allocated for pages */
        struct page **pages;    /* Backing pages for file */
        unsigned long long npages;
        struct sg_table sg_table;       /* SG table for backing pages */
        struct pds_lm_sg_elem *sgl;     /* DMA mapping */
        dma_addr_t sgl_addr;
        u16 num_sge;
        struct scatterlist *last_offset_sg;     /* Iterator */
        unsigned int sg_last_entry;
        unsigned long last_offset;
        bool disabled;
};

struct pds_vfio_pci_device;

struct file *
pds_vfio_step_device_state_locked(struct pds_vfio_pci_device *pds_vfio,
                                  enum vfio_device_mig_state next);

void pds_vfio_put_save_file(struct pds_vfio_pci_device *pds_vfio);
void pds_vfio_put_restore_file(struct pds_vfio_pci_device *pds_vfio);

#endif /* _LM_H_ */