xref: /linux-6.15/include/linux/uacce.h (revision 82d00a93)
1 /* SPDX-License-Identifier: GPL-2.0-or-later */
2 #ifndef _LINUX_UACCE_H
3 #define _LINUX_UACCE_H
4 
5 #include <linux/cdev.h>
6 #include <uapi/misc/uacce/uacce.h>
7 
8 #define UACCE_NAME		"uacce"
9 #define UACCE_MAX_REGION	2
10 #define UACCE_MAX_NAME_SIZE	64
11 
12 struct uacce_queue;
13 struct uacce_device;
14 
15 /**
16  * struct uacce_qfile_region - structure of queue file region
17  * @type: type of the region
18  */
19 struct uacce_qfile_region {
20 	enum uacce_qfrt type;
21 };
22 
23 /**
24  * struct uacce_ops - uacce device operations
25  * @get_available_instances:  get available instances left of the device
26  * @get_queue: get a queue from the device
27  * @put_queue: free a queue to the device
28  * @start_queue: make the queue start work after get_queue
29  * @stop_queue: make the queue stop work before put_queue
30  * @is_q_updated: check whether the task is finished
31  * @mmap: mmap addresses of queue to user space
32  * @ioctl: ioctl for user space users of the queue
33  */
34 struct uacce_ops {
35 	int (*get_available_instances)(struct uacce_device *uacce);
36 	int (*get_queue)(struct uacce_device *uacce, unsigned long arg,
37 			 struct uacce_queue *q);
38 	void (*put_queue)(struct uacce_queue *q);
39 	int (*start_queue)(struct uacce_queue *q);
40 	void (*stop_queue)(struct uacce_queue *q);
41 	int (*is_q_updated)(struct uacce_queue *q);
42 	int (*mmap)(struct uacce_queue *q, struct vm_area_struct *vma,
43 		    struct uacce_qfile_region *qfr);
44 	long (*ioctl)(struct uacce_queue *q, unsigned int cmd,
45 		      unsigned long arg);
46 };
47 
48 /**
49  * struct uacce_interface - interface required for uacce_register()
50  * @name: the uacce device name.  Will show up in sysfs
51  * @flags: uacce device attributes
52  * @ops: pointer to the struct uacce_ops
53  */
54 struct uacce_interface {
55 	char name[UACCE_MAX_NAME_SIZE];
56 	unsigned int flags;
57 	const struct uacce_ops *ops;
58 };
59 
60 enum uacce_q_state {
61 	UACCE_Q_ZOMBIE = 0,
62 	UACCE_Q_INIT,
63 	UACCE_Q_STARTED,
64 };
65 
66 /**
67  * struct uacce_queue
68  * @uacce: pointer to uacce
69  * @priv: private pointer
70  * @wait: wait queue head
71  * @list: index into uacce_mm
72  * @uacce_mm: the corresponding mm
73  * @qfrs: pointer of qfr regions
74  * @state: queue state machine
75  */
76 struct uacce_queue {
77 	struct uacce_device *uacce;
78 	void *priv;
79 	wait_queue_head_t wait;
80 	struct list_head list;
81 	struct uacce_mm *uacce_mm;
82 	struct uacce_qfile_region *qfrs[UACCE_MAX_REGION];
83 	enum uacce_q_state state;
84 };
85 
86 /**
87  * struct uacce_device
88  * @algs: supported algorithms
89  * @api_ver: api version
90  * @ops: pointer to the struct uacce_ops
91  * @qf_pg_num: page numbers of the queue file regions
92  * @parent: pointer to the parent device
93  * @is_vf: whether virtual function
94  * @flags: uacce attributes
95  * @dev_id: id of the uacce device
96  * @cdev: cdev of the uacce
97  * @dev: dev of the uacce
98  * @priv: private pointer of the uacce
99  * @mm_list: list head of uacce_mm->list
100  * @mm_lock: lock for mm_list
101  * @inode: core vfs
102  */
103 struct uacce_device {
104 	const char *algs;
105 	const char *api_ver;
106 	const struct uacce_ops *ops;
107 	unsigned long qf_pg_num[UACCE_MAX_REGION];
108 	struct device *parent;
109 	bool is_vf;
110 	u32 flags;
111 	u32 dev_id;
112 	struct cdev *cdev;
113 	struct device dev;
114 	void *priv;
115 	struct list_head mm_list;
116 	struct mutex mm_lock;
117 	struct inode *inode;
118 };
119 
120 /**
121  * struct uacce_mm - keep track of queues bound to a process
122  * @list: index into uacce_device
123  * @queues: list of queues
124  * @mm: the mm struct
125  * @lock: protects the list of queues
126  * @pasid: pasid of the uacce_mm
127  * @handle: iommu_sva handle return from iommu_sva_bind_device
128  */
129 struct uacce_mm {
130 	struct list_head list;
131 	struct list_head queues;
132 	struct mm_struct *mm;
133 	struct mutex lock;
134 	int pasid;
135 	struct iommu_sva *handle;
136 };
137 
138 #if IS_ENABLED(CONFIG_UACCE)
139 
140 struct uacce_device *uacce_alloc(struct device *parent,
141 				 struct uacce_interface *interface);
142 int uacce_register(struct uacce_device *uacce);
143 void uacce_remove(struct uacce_device *uacce);
144 
145 #else /* CONFIG_UACCE */
146 
147 static inline
148 struct uacce_device *uacce_alloc(struct device *parent,
149 				 struct uacce_interface *interface)
150 {
151 	return ERR_PTR(-ENODEV);
152 }
153 
154 static inline int uacce_register(struct uacce_device *uacce)
155 {
156 	return -EINVAL;
157 }
158 
159 static inline void uacce_remove(struct uacce_device *uacce) {}
160 
161 #endif /* CONFIG_UACCE */
162 
163 #endif /* _LINUX_UACCE_H */
164