#ifndef _NV_VGPU_IOCTL_H_
#define _NV_VGPU_IOCTL_H_

#include "nvidia-vgpu-vfio.h"

#define NV_VGPU_IOCTL_MAGIC 'N'

#define NV_VGPU_IOCTL_GET_INFO \
	_IOWR(NV_VGPU_IOCTL_MAGIC, 0x01, struct nv_vgpu_ioctl_get_info)

#define NV_VGPU_IOCTL_MAP_BAR \
	_IOWR(NV_VGPU_IOCTL_MAGIC, 0x02, struct nv_vgpu_ioctl_map_bar)

#define NV_VGPU_IOCTL_UNMAP_BAR \
	_IOWR(NV_VGPU_IOCTL_MAGIC, 0x03, struct nv_vgpu_ioctl_unmap_bar)

#define NV_VGPU_IOCTL_ENABLE_IRQ \
	_IOWR(NV_VGPU_IOCTL_MAGIC, 0x04, struct nv_vgpu_ioctl_irq)

#define NV_VGPU_IOCTL_DISABLE_IRQ \
	_IOWR(NV_VGPU_IOCTL_MAGIC, 0x05, struct nv_vgpu_ioctl_irq)

#define NV_VGPU_IOCTL_GSP_RPC \
	_IOWR(NV_VGPU_IOCTL_MAGIC, 0x06, struct nv_vgpu_ioctl_gsp_rpc)

#define NV_VGPU_IOCTL_MIGRATE_SAVE \
	_IOWR(NV_VGPU_IOCTL_MAGIC, 0x07, struct nv_vgpu_ioctl_migrate)

#define NV_VGPU_IOCTL_MIGRATE_RESTORE \
	_IOWR(NV_VGPU_IOCTL_MAGIC, 0x08, struct nv_vgpu_ioctl_migrate)

#define NV_VGPU_IOCTL_SET_GSP_FIRMWARE \
	_IOWR(NV_VGPU_IOCTL_MAGIC, 0x09, struct nv_vgpu_ioctl_firmware)

#define NV_VGPU_IOCTL_REGISTER_VGPU_TYPE \
	_IOWR(NV_VGPU_IOCTL_MAGIC, 0x0A, struct nv_vgpu_ioctl_register_vgpu_type)

/* vGPU type info structure (shared between main.c and ioctl.c) */
struct nv_vgpu_type_info {
	char name[64];
	char description[128];
	u64 vdev_id;
	u64 pdev_id;
	u32 max_instances;
	u32 vram_mb;	/* emulated framebuffer size; 0 = 512 MiB default */
};

struct nv_vgpu_ioctl_get_info {
	u32 vgpu_type;
	u32 vgpu_id;
	u32 gpu_instance_id;
	u32 compute_instance_id;
	u32 state;
	u64 fb_base;
	u64 fb_size;
	u64 bar0_config_base;
	u32 num_bars;
	struct {
		u32 index;
		u64 gpa;
		u64 size;
		u64 iova;
		bool mapped;
		bool trapped;
	} bars[NV_VGPU_MAX_REGIONS];
};

struct nv_vgpu_ioctl_map_bar {
	u32 bar_index;
	u64 gpa;
	u64 size;
	u64 iova;
	u32 flags;
};

struct nv_vgpu_ioctl_unmap_bar {
	u32 bar_index;
	u64 iova;
	u64 size;
};

struct nv_vgpu_ioctl_irq {
	u32 irq_index;
	int fd;
};

struct nv_vgpu_ioctl_gsp_rpc {
	u32 type;
	u32 flags;
	u64 data_ptr;
	u32 data_size;
	u32 result;
};

struct nv_vgpu_ioctl_migrate {
	u64 data_ptr;
	u32 data_size;
	u32 flags;
	int result;
};

struct nv_vgpu_ioctl_firmware {
	u32 chip_id;
	u64 firmware_ptr;
	u32 firmware_size;
	u32 firmware_type;
};

struct nv_vgpu_ioctl_register_vgpu_type {
	char name[64];
	char description[128];
	u64 vdev_id;
	u64 pdev_id;
	u32 max_instances;
	u32 vram_mb;
};

int nv_vgpu_ioctl_get_info(struct nv_vgpu_device *vgpu, void __user *arg);
int nv_vgpu_ioctl_map_bar(struct nv_vgpu_device *vgpu, void __user *arg);
int nv_vgpu_ioctl_unmap_bar(struct nv_vgpu_device *vgpu, void __user *arg);
int nv_vgpu_ioctl_enable_irq(struct nv_vgpu_device *vgpu, void __user *arg);
int nv_vgpu_ioctl_disable_irq(struct nv_vgpu_device *vgpu, void __user *arg);
int nv_vgpu_ioctl_gsp_rpc(struct nv_vgpu_device *vgpu, void __user *arg);
int nv_vgpu_ioctl_migrate_save(struct nv_vgpu_device *vgpu, void __user *arg);
int nv_vgpu_ioctl_migrate_restore(struct nv_vgpu_device *vgpu, void __user *arg);
int nv_vgpu_ioctl_register_vgpu_type(void __user *arg);

#endif