/* SPDX-License-Identifier: GPL-2.0-or-later */
#ifndef _LINUX_UACCE_H
#define _LINUX_UACCE_H

#include <linux/cdev.h>
#include <uapi/misc/uacce/uacce.h>

#define UACCE_NAME  "uacce"
#define UACCE_MAX_REGION 2
#define UACCE_MAX_NAME_SIZE 64
#define UACCE_MAX_ERR_THRESHOLD 65535

struct uacce_queue;
struct uacce_device;

/**
 * struct uacce_qfile_region - structure of queue file region
 * @type: type of the region
 */
struct uacce_qfile_region {
 enum uacce_qfrt type;
};

/**
 * struct uacce_ops - uacce device operations
 * @get_available_instances:  get available instances left of the device
 * @get_queue: get a queue from the device
 * @put_queue: free a queue to the device
 * @start_queue: make the queue start work after get_queue
 * @stop_queue: make the queue stop work before put_queue
 * @is_q_updated: check whether the task is finished
 * @mmap: mmap addresses of queue to user space
 * @ioctl: ioctl for user space users of the queue
 * @get_isolate_state: get the device state after set the isolate strategy
 * @isolate_err_threshold_write: stored the isolate error threshold to the device
 * @isolate_err_threshold_read: read the isolate error threshold value from the device
 */
struct uacce_ops {
 int (*get_available_instances)(struct uacce_device *uacce);
 int (*get_queue)(struct uacce_device *uacce, unsigned long arg,
    struct uacce_queue *q);
 void (*put_queue)(struct uacce_queue *q);
 int (*start_queue)(struct uacce_queue *q);
 void (*stop_queue)(struct uacce_queue *q);
 int (*is_q_updated)(struct uacce_queue *q);
 int (*mmap)(struct uacce_queue *q, struct vm_area_struct *vma,
      struct uacce_qfile_region *qfr);
 long (*ioctl)(struct uacce_queue *q, unsigned int cmd,
        unsigned long arg);
 enum uacce_dev_state (*get_isolate_state)(struct uacce_device *uacce);
 int (*isolate_err_threshold_write)(struct uacce_device *uacce, u32 num);
 u32 (*isolate_err_threshold_read)(struct uacce_device *uacce);
};

/**
 * struct uacce_interface - interface required for uacce_register()
 * @name: the uacce device name.  Will show up in sysfs
 * @flags: uacce device attributes
 * @ops: pointer to the struct uacce_ops
 */
struct uacce_interface {
 char name[UACCE_MAX_NAME_SIZE];
 unsigned int flags;
 const struct uacce_ops *ops;
};

enum uacce_dev_state {
 UACCE_DEV_NORMAL,
 UACCE_DEV_ISOLATE,
};

enum uacce_q_state {
 UACCE_Q_ZOMBIE = 0,
 UACCE_Q_INIT,
 UACCE_Q_STARTED,
};

/**
 * struct uacce_queue
 * @uacce: pointer to uacce
 * @priv: private pointer
 * @wait: wait queue head
 * @list: index into uacce queues list
 * @qfrs: pointer of qfr regions
 * @mutex: protects queue state
 * @state: queue state machine
 * @pasid: pasid associated to the mm
 * @handle: iommu_sva handle returned by iommu_sva_bind_device()
 * @mapping: user space mapping of the queue
 */
struct uacce_queue {
 struct uacce_device *uacce;
 void *priv;
 wait_queue_head_t wait;
 struct list_head list;
 struct uacce_qfile_region *qfrs[UACCE_MAX_REGION];
 struct mutex mutex;
 enum uacce_q_state state;
 u32 pasid;
 struct iommu_sva *handle;
 struct address_space *mapping;
};

/**
 * struct uacce_device
 * @algs: supported algorithms
 * @api_ver: api version
 * @ops: pointer to the struct uacce_ops
 * @qf_pg_num: page numbers of the queue file regions
 * @parent: pointer to the parent device
 * @is_vf: whether virtual function
 * @flags: uacce attributes
 * @dev_id: id of the uacce device
 * @cdev: cdev of the uacce
 * @dev: dev of the uacce
 * @mutex: protects uacce operation
 * @priv: private pointer of the uacce
 * @queues: list of queues
 */
struct uacce_device {
 const char *algs;
 const char *api_ver;
 const struct uacce_ops *ops;
 unsigned long qf_pg_num[UACCE_MAX_REGION];
 struct device *parent;
 bool is_vf;
 u32 flags;
 u32 dev_id;
 struct cdev *cdev;
 struct device dev;
 struct mutex mutex;
 void *priv;
 struct list_head queues;
};

#if IS_ENABLED(CONFIG_UACCE)

struct uacce_device *uacce_alloc(struct device *parent,
     struct uacce_interface *interface);
int uacce_register(struct uacce_device *uacce);
void uacce_remove(struct uacce_device *uacce);

#else /* CONFIG_UACCE */

static inline
struct uacce_device *uacce_alloc(struct device *parent,
     struct uacce_interface *interface)
{
 return ERR_PTR(-ENODEV);
}

static inline int uacce_register(struct uacce_device *uacce)
{
 return -EINVAL;
}

static inline void uacce_remove(struct uacce_device *uacce) {}

#endif /* CONFIG_UACCE */

#endif /* _LINUX_UACCE_H */