1 files changed, 45 insertions, 0 deletions
diff --git a/kernel/include/memory.h b/kernel/include/memory.h
new file mode 100644
index 0000000..14a35e9
--- /dev/null
+++ b/kernel/include/memory.h
@@ -0,0 +1,45 @@
+#ifndef _NV_VGPU_MEMORY_H_
+#define _NV_VGPU_MEMORY_H_
+
+#include "nvidia-vgpu-vfio.h"
+
+#define NV_VGPU_MAX_MAPPINGS 1024
+
+struct nv_vgpu_mapping {
+	u64 gpa;
+	u64 iova;
+	u64 size;
+	struct page **pages;
+	int page_count;
+	bool pinned;
+	bool mapped;
+	struct list_head list;
+};
+
+struct nv_vgpu_mm_ctx {
+	struct nv_vgpu_device *vgpu;
+	struct mutex lock;
+	struct list_head mappings;
+	u32 mapping_count;
+	struct iommu_domain *iommu_domain;
+	struct dma_iommu_mapping *dma_mapping;
+	bool use_simple;
+};
+
+int nv_vgpu_memory_init(struct nv_vgpu_device *vgpu);
+void nv_vgpu_memory_cleanup(struct nv_vgpu_device *vgpu);
+
+int nv_vgpu_memory_map(struct nv_vgpu_device *vgpu, u64 gpa, u64 size, u64 *iova);
+int nv_vgpu_memory_unmap(struct nv_vgpu_device *vgpu, u64 iova, u64 size);
+
+int nv_vgpu_memory_pin(struct nv_vgpu_device *vgpu, u64 gpa, u64 size);
+int nv_vgpu_memory_unpin(struct nv_vgpu_device *vgpu, u64 gpa, u64 size);
+
+int nv_vgpu_memory_mmap(struct nv_vgpu_device *vgpu, struct vm_area_struct *vma);
+
+static inline struct nv_vgpu_mm_ctx *vgpu_to_mm_ctx(struct nv_vgpu_device *vgpu)
+{
+	return vgpu->mm_ctx;
+}
+
+#endif
+\ No newline at end of file