1 /*
2  * Copyright (C) 2015 Red Hat, Inc.
3  * All Rights Reserved.
4  *
5  * Permission is hereby granted, free of charge, to any person obtaining
6  * a copy of this software and associated documentation files (the
7  * "Software"), to deal in the Software without restriction, including
8  * without limitation the rights to use, copy, modify, merge, publish,
9  * distribute, sublicense, and/or sell copies of the Software, and to
10  * permit persons to whom the Software is furnished to do so, subject to
11  * the following conditions:
12  *
13  * The above copyright notice and this permission notice (including the
14  * next paragraph) shall be included in all copies or substantial
15  * portions of the Software.
16  *
17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
18  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
19  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
20  * IN NO EVENT SHALL THE COPYRIGHT OWNER(S) AND/OR ITS SUPPLIERS BE
21  * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
22  * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
23  * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
24  */
25 
26 #include <linux/virtio.h>
27 #include <linux/virtio_config.h>
28 #include <drm/drmP.h>
29 #include "virtgpu_drv.h"
30 
31 static int virtio_gpu_fbdev = 1;
32 
33 MODULE_PARM_DESC(fbdev, "Disable/Enable framebuffer device & console");
34 module_param_named(fbdev, virtio_gpu_fbdev, int, 0400);
35 
36 static void virtio_gpu_config_changed_work_func(struct work_struct *work)
37 {
38 	struct virtio_gpu_device *vgdev =
39 		container_of(work, struct virtio_gpu_device,
40 			     config_changed_work);
41 	u32 events_read, events_clear = 0;
42 
43 	/* read the config space */
44 	virtio_cread(vgdev->vdev, struct virtio_gpu_config,
45 		     events_read, &events_read);
46 	if (events_read & VIRTIO_GPU_EVENT_DISPLAY) {
47 		virtio_gpu_cmd_get_display_info(vgdev);
48 		drm_helper_hpd_irq_event(vgdev->ddev);
49 		events_clear |= VIRTIO_GPU_EVENT_DISPLAY;
50 	}
51 	virtio_cwrite(vgdev->vdev, struct virtio_gpu_config,
52 		      events_clear, &events_clear);
53 }
54 
55 static void virtio_gpu_ctx_id_get(struct virtio_gpu_device *vgdev,
56 				  uint32_t *resid)
57 {
58 	int handle;
59 
60 	idr_preload(GFP_KERNEL);
61 	spin_lock(&vgdev->ctx_id_idr_lock);
62 	handle = idr_alloc(&vgdev->ctx_id_idr, NULL, 1, 0, 0);
63 	spin_unlock(&vgdev->ctx_id_idr_lock);
64 	idr_preload_end();
65 	*resid = handle;
66 }
67 
68 static void virtio_gpu_ctx_id_put(struct virtio_gpu_device *vgdev, uint32_t id)
69 {
70 	spin_lock(&vgdev->ctx_id_idr_lock);
71 	idr_remove(&vgdev->ctx_id_idr, id);
72 	spin_unlock(&vgdev->ctx_id_idr_lock);
73 }
74 
75 static void virtio_gpu_context_create(struct virtio_gpu_device *vgdev,
76 				      uint32_t nlen, const char *name,
77 				      uint32_t *ctx_id)
78 {
79 	virtio_gpu_ctx_id_get(vgdev, ctx_id);
80 	virtio_gpu_cmd_context_create(vgdev, *ctx_id, nlen, name);
81 }
82 
83 static void virtio_gpu_context_destroy(struct virtio_gpu_device *vgdev,
84 				      uint32_t ctx_id)
85 {
86 	virtio_gpu_cmd_context_destroy(vgdev, ctx_id);
87 	virtio_gpu_ctx_id_put(vgdev, ctx_id);
88 }
89 
90 static void virtio_gpu_init_vq(struct virtio_gpu_queue *vgvq,
91 			       void (*work_func)(struct work_struct *work))
92 {
93 	spin_lock_init(&vgvq->qlock);
94 	init_waitqueue_head(&vgvq->ack_queue);
95 	INIT_WORK(&vgvq->dequeue_work, work_func);
96 }
97 
98 static void virtio_gpu_get_capsets(struct virtio_gpu_device *vgdev,
99 				   int num_capsets)
100 {
101 	int i, ret;
102 
103 	vgdev->capsets = kcalloc(num_capsets,
104 				 sizeof(struct virtio_gpu_drv_capset),
105 				 GFP_KERNEL);
106 	if (!vgdev->capsets) {
107 		DRM_ERROR("failed to allocate cap sets\n");
108 		return;
109 	}
110 	for (i = 0; i < num_capsets; i++) {
111 		virtio_gpu_cmd_get_capset_info(vgdev, i);
112 		ret = wait_event_timeout(vgdev->resp_wq,
113 					 vgdev->capsets[i].id > 0, 5 * HZ);
114 		if (ret == 0) {
115 			DRM_ERROR("timed out waiting for cap set %d\n", i);
116 			kfree(vgdev->capsets);
117 			vgdev->capsets = NULL;
118 			return;
119 		}
120 		DRM_INFO("cap set %d: id %d, max-version %d, max-size %d\n",
121 			 i, vgdev->capsets[i].id,
122 			 vgdev->capsets[i].max_version,
123 			 vgdev->capsets[i].max_size);
124 	}
125 	vgdev->num_capsets = num_capsets;
126 }
127 
128 int virtio_gpu_driver_load(struct drm_device *dev, unsigned long flags)
129 {
130 	static vq_callback_t *callbacks[] = {
131 		virtio_gpu_ctrl_ack, virtio_gpu_cursor_ack
132 	};
133 	static const char * const names[] = { "control", "cursor" };
134 
135 	struct virtio_gpu_device *vgdev;
136 	/* this will expand later */
137 	struct virtqueue *vqs[2];
138 	u32 num_scanouts, num_capsets;
139 	int ret;
140 
141 	if (!virtio_has_feature(dev->virtdev, VIRTIO_F_VERSION_1))
142 		return -ENODEV;
143 
144 	vgdev = kzalloc(sizeof(struct virtio_gpu_device), GFP_KERNEL);
145 	if (!vgdev)
146 		return -ENOMEM;
147 
148 	vgdev->ddev = dev;
149 	dev->dev_private = vgdev;
150 	vgdev->vdev = dev->virtdev;
151 	vgdev->dev = dev->dev;
152 
153 	spin_lock_init(&vgdev->display_info_lock);
154 	spin_lock_init(&vgdev->ctx_id_idr_lock);
155 	idr_init(&vgdev->ctx_id_idr);
156 	spin_lock_init(&vgdev->resource_idr_lock);
157 	idr_init(&vgdev->resource_idr);
158 	init_waitqueue_head(&vgdev->resp_wq);
159 	virtio_gpu_init_vq(&vgdev->ctrlq, virtio_gpu_dequeue_ctrl_func);
160 	virtio_gpu_init_vq(&vgdev->cursorq, virtio_gpu_dequeue_cursor_func);
161 
162 	vgdev->fence_drv.context = dma_fence_context_alloc(1);
163 	spin_lock_init(&vgdev->fence_drv.lock);
164 	INIT_LIST_HEAD(&vgdev->fence_drv.fences);
165 	INIT_LIST_HEAD(&vgdev->cap_cache);
166 	INIT_WORK(&vgdev->config_changed_work,
167 		  virtio_gpu_config_changed_work_func);
168 
169 	if (virtio_has_feature(vgdev->vdev, VIRTIO_GPU_F_VIRGL))
170 		vgdev->has_virgl_3d = true;
171 	DRM_INFO("virgl 3d acceleration %s\n",
172 		 vgdev->has_virgl_3d ? "enabled" : "not available");
173 
174 	ret = vgdev->vdev->config->find_vqs(vgdev->vdev, 2, vqs,
175 					    callbacks, names);
176 	if (ret) {
177 		DRM_ERROR("failed to find virt queues\n");
178 		goto err_vqs;
179 	}
180 	vgdev->ctrlq.vq = vqs[0];
181 	vgdev->cursorq.vq = vqs[1];
182 	ret = virtio_gpu_alloc_vbufs(vgdev);
183 	if (ret) {
184 		DRM_ERROR("failed to alloc vbufs\n");
185 		goto err_vbufs;
186 	}
187 
188 	ret = virtio_gpu_ttm_init(vgdev);
189 	if (ret) {
190 		DRM_ERROR("failed to init ttm %d\n", ret);
191 		goto err_ttm;
192 	}
193 
194 	/* get display info */
195 	virtio_cread(vgdev->vdev, struct virtio_gpu_config,
196 		     num_scanouts, &num_scanouts);
197 	vgdev->num_scanouts = min_t(uint32_t, num_scanouts,
198 				    VIRTIO_GPU_MAX_SCANOUTS);
199 	if (!vgdev->num_scanouts) {
200 		DRM_ERROR("num_scanouts is zero\n");
201 		ret = -EINVAL;
202 		goto err_scanouts;
203 	}
204 	DRM_INFO("number of scanouts: %d\n", num_scanouts);
205 
206 	virtio_cread(vgdev->vdev, struct virtio_gpu_config,
207 		     num_capsets, &num_capsets);
208 	DRM_INFO("number of cap sets: %d\n", num_capsets);
209 
210 	ret = virtio_gpu_modeset_init(vgdev);
211 	if (ret)
212 		goto err_modeset;
213 
214 	virtio_device_ready(vgdev->vdev);
215 	vgdev->vqs_ready = true;
216 
217 	if (num_capsets)
218 		virtio_gpu_get_capsets(vgdev, num_capsets);
219 	virtio_gpu_cmd_get_display_info(vgdev);
220 	wait_event_timeout(vgdev->resp_wq, !vgdev->display_info_pending,
221 			   5 * HZ);
222 	if (virtio_gpu_fbdev)
223 		virtio_gpu_fbdev_init(vgdev);
224 
225 	return 0;
226 
227 err_modeset:
228 err_scanouts:
229 	virtio_gpu_ttm_fini(vgdev);
230 err_ttm:
231 	virtio_gpu_free_vbufs(vgdev);
232 err_vbufs:
233 	vgdev->vdev->config->del_vqs(vgdev->vdev);
234 err_vqs:
235 	kfree(vgdev);
236 	return ret;
237 }
238 
239 static void virtio_gpu_cleanup_cap_cache(struct virtio_gpu_device *vgdev)
240 {
241 	struct virtio_gpu_drv_cap_cache *cache_ent, *tmp;
242 
243 	list_for_each_entry_safe(cache_ent, tmp, &vgdev->cap_cache, head) {
244 		kfree(cache_ent->caps_cache);
245 		kfree(cache_ent);
246 	}
247 }
248 
249 int virtio_gpu_driver_unload(struct drm_device *dev)
250 {
251 	struct virtio_gpu_device *vgdev = dev->dev_private;
252 
253 	vgdev->vqs_ready = false;
254 	flush_work(&vgdev->ctrlq.dequeue_work);
255 	flush_work(&vgdev->cursorq.dequeue_work);
256 	flush_work(&vgdev->config_changed_work);
257 	vgdev->vdev->config->del_vqs(vgdev->vdev);
258 
259 	virtio_gpu_modeset_fini(vgdev);
260 	virtio_gpu_ttm_fini(vgdev);
261 	virtio_gpu_free_vbufs(vgdev);
262 	virtio_gpu_cleanup_cap_cache(vgdev);
263 	kfree(vgdev->capsets);
264 	kfree(vgdev);
265 	return 0;
266 }
267 
268 int virtio_gpu_driver_open(struct drm_device *dev, struct drm_file *file)
269 {
270 	struct virtio_gpu_device *vgdev = dev->dev_private;
271 	struct virtio_gpu_fpriv *vfpriv;
272 	uint32_t id;
273 	char dbgname[64], tmpname[TASK_COMM_LEN];
274 
275 	/* can't create contexts without 3d renderer */
276 	if (!vgdev->has_virgl_3d)
277 		return 0;
278 
279 	get_task_comm(tmpname, current);
280 	snprintf(dbgname, sizeof(dbgname), "%s", tmpname);
281 	dbgname[63] = 0;
282 	/* allocate a virt GPU context for this opener */
283 	vfpriv = kzalloc(sizeof(*vfpriv), GFP_KERNEL);
284 	if (!vfpriv)
285 		return -ENOMEM;
286 
287 	virtio_gpu_context_create(vgdev, strlen(dbgname), dbgname, &id);
288 
289 	vfpriv->ctx_id = id;
290 	file->driver_priv = vfpriv;
291 	return 0;
292 }
293 
294 void virtio_gpu_driver_postclose(struct drm_device *dev, struct drm_file *file)
295 {
296 	struct virtio_gpu_device *vgdev = dev->dev_private;
297 	struct virtio_gpu_fpriv *vfpriv;
298 
299 	if (!vgdev->has_virgl_3d)
300 		return;
301 
302 	vfpriv = file->driver_priv;
303 
304 	virtio_gpu_context_destroy(vgdev, vfpriv->ctx_id);
305 	kfree(vfpriv);
306 	file->driver_priv = NULL;
307 }
308