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 		if (vgdev->has_edid)
48 			virtio_gpu_cmd_get_edids(vgdev);
49 		virtio_gpu_cmd_get_display_info(vgdev);
50 		drm_helper_hpd_irq_event(vgdev->ddev);
51 		events_clear |= VIRTIO_GPU_EVENT_DISPLAY;
52 	}
53 	virtio_cwrite(vgdev->vdev, struct virtio_gpu_config,
54 		      events_clear, &events_clear);
55 }
56 
57 static int virtio_gpu_context_create(struct virtio_gpu_device *vgdev,
58 				      uint32_t nlen, const char *name)
59 {
60 	int handle = ida_alloc(&vgdev->ctx_id_ida, GFP_KERNEL);
61 
62 	if (handle < 0)
63 		return handle;
64 	handle += 1;
65 	virtio_gpu_cmd_context_create(vgdev, handle, nlen, name);
66 	return handle;
67 }
68 
69 static void virtio_gpu_context_destroy(struct virtio_gpu_device *vgdev,
70 				      uint32_t ctx_id)
71 {
72 	virtio_gpu_cmd_context_destroy(vgdev, ctx_id);
73 	ida_free(&vgdev->ctx_id_ida, ctx_id - 1);
74 }
75 
76 static void virtio_gpu_init_vq(struct virtio_gpu_queue *vgvq,
77 			       void (*work_func)(struct work_struct *work))
78 {
79 	spin_lock_init(&vgvq->qlock);
80 	init_waitqueue_head(&vgvq->ack_queue);
81 	INIT_WORK(&vgvq->dequeue_work, work_func);
82 }
83 
84 static void virtio_gpu_get_capsets(struct virtio_gpu_device *vgdev,
85 				   int num_capsets)
86 {
87 	int i, ret;
88 
89 	vgdev->capsets = kcalloc(num_capsets,
90 				 sizeof(struct virtio_gpu_drv_capset),
91 				 GFP_KERNEL);
92 	if (!vgdev->capsets) {
93 		DRM_ERROR("failed to allocate cap sets\n");
94 		return;
95 	}
96 	for (i = 0; i < num_capsets; i++) {
97 		virtio_gpu_cmd_get_capset_info(vgdev, i);
98 		ret = wait_event_timeout(vgdev->resp_wq,
99 					 vgdev->capsets[i].id > 0, 5 * HZ);
100 		if (ret == 0) {
101 			DRM_ERROR("timed out waiting for cap set %d\n", i);
102 			kfree(vgdev->capsets);
103 			vgdev->capsets = NULL;
104 			return;
105 		}
106 		DRM_INFO("cap set %d: id %d, max-version %d, max-size %d\n",
107 			 i, vgdev->capsets[i].id,
108 			 vgdev->capsets[i].max_version,
109 			 vgdev->capsets[i].max_size);
110 	}
111 	vgdev->num_capsets = num_capsets;
112 }
113 
114 int virtio_gpu_driver_load(struct drm_device *dev, unsigned long flags)
115 {
116 	static vq_callback_t *callbacks[] = {
117 		virtio_gpu_ctrl_ack, virtio_gpu_cursor_ack
118 	};
119 	static const char * const names[] = { "control", "cursor" };
120 
121 	struct virtio_gpu_device *vgdev;
122 	/* this will expand later */
123 	struct virtqueue *vqs[2];
124 	u32 num_scanouts, num_capsets;
125 	int ret;
126 
127 	if (!virtio_has_feature(dev_to_virtio(dev->dev), VIRTIO_F_VERSION_1))
128 		return -ENODEV;
129 
130 	vgdev = kzalloc(sizeof(struct virtio_gpu_device), GFP_KERNEL);
131 	if (!vgdev)
132 		return -ENOMEM;
133 
134 	vgdev->ddev = dev;
135 	dev->dev_private = vgdev;
136 	vgdev->vdev = dev_to_virtio(dev->dev);
137 	vgdev->dev = dev->dev;
138 
139 	spin_lock_init(&vgdev->display_info_lock);
140 	ida_init(&vgdev->ctx_id_ida);
141 	ida_init(&vgdev->resource_ida);
142 	init_waitqueue_head(&vgdev->resp_wq);
143 	virtio_gpu_init_vq(&vgdev->ctrlq, virtio_gpu_dequeue_ctrl_func);
144 	virtio_gpu_init_vq(&vgdev->cursorq, virtio_gpu_dequeue_cursor_func);
145 
146 	vgdev->fence_drv.context = dma_fence_context_alloc(1);
147 	spin_lock_init(&vgdev->fence_drv.lock);
148 	INIT_LIST_HEAD(&vgdev->fence_drv.fences);
149 	INIT_LIST_HEAD(&vgdev->cap_cache);
150 	INIT_WORK(&vgdev->config_changed_work,
151 		  virtio_gpu_config_changed_work_func);
152 
153 #ifdef __LITTLE_ENDIAN
154 	if (virtio_has_feature(vgdev->vdev, VIRTIO_GPU_F_VIRGL))
155 		vgdev->has_virgl_3d = true;
156 	DRM_INFO("virgl 3d acceleration %s\n",
157 		 vgdev->has_virgl_3d ? "enabled" : "not supported by host");
158 #else
159 	DRM_INFO("virgl 3d acceleration not supported by guest\n");
160 #endif
161 	if (virtio_has_feature(vgdev->vdev, VIRTIO_GPU_F_EDID)) {
162 		vgdev->has_edid = true;
163 		DRM_INFO("EDID support available.\n");
164 	}
165 
166 	ret = virtio_find_vqs(vgdev->vdev, 2, vqs, callbacks, names, NULL);
167 	if (ret) {
168 		DRM_ERROR("failed to find virt queues\n");
169 		goto err_vqs;
170 	}
171 	vgdev->ctrlq.vq = vqs[0];
172 	vgdev->cursorq.vq = vqs[1];
173 	ret = virtio_gpu_alloc_vbufs(vgdev);
174 	if (ret) {
175 		DRM_ERROR("failed to alloc vbufs\n");
176 		goto err_vbufs;
177 	}
178 
179 	ret = virtio_gpu_ttm_init(vgdev);
180 	if (ret) {
181 		DRM_ERROR("failed to init ttm %d\n", ret);
182 		goto err_ttm;
183 	}
184 
185 	/* get display info */
186 	virtio_cread(vgdev->vdev, struct virtio_gpu_config,
187 		     num_scanouts, &num_scanouts);
188 	vgdev->num_scanouts = min_t(uint32_t, num_scanouts,
189 				    VIRTIO_GPU_MAX_SCANOUTS);
190 	if (!vgdev->num_scanouts) {
191 		DRM_ERROR("num_scanouts is zero\n");
192 		ret = -EINVAL;
193 		goto err_scanouts;
194 	}
195 	DRM_INFO("number of scanouts: %d\n", num_scanouts);
196 
197 	virtio_cread(vgdev->vdev, struct virtio_gpu_config,
198 		     num_capsets, &num_capsets);
199 	DRM_INFO("number of cap sets: %d\n", num_capsets);
200 
201 	ret = virtio_gpu_modeset_init(vgdev);
202 	if (ret)
203 		goto err_modeset;
204 
205 	virtio_device_ready(vgdev->vdev);
206 	vgdev->vqs_ready = true;
207 
208 	if (num_capsets)
209 		virtio_gpu_get_capsets(vgdev, num_capsets);
210 	if (vgdev->has_edid)
211 		virtio_gpu_cmd_get_edids(vgdev);
212 	virtio_gpu_cmd_get_display_info(vgdev);
213 	wait_event_timeout(vgdev->resp_wq, !vgdev->display_info_pending,
214 			   5 * HZ);
215 	if (virtio_gpu_fbdev)
216 		virtio_gpu_fbdev_init(vgdev);
217 
218 	return 0;
219 
220 err_modeset:
221 err_scanouts:
222 	virtio_gpu_ttm_fini(vgdev);
223 err_ttm:
224 	virtio_gpu_free_vbufs(vgdev);
225 err_vbufs:
226 	vgdev->vdev->config->del_vqs(vgdev->vdev);
227 err_vqs:
228 	kfree(vgdev);
229 	return ret;
230 }
231 
232 static void virtio_gpu_cleanup_cap_cache(struct virtio_gpu_device *vgdev)
233 {
234 	struct virtio_gpu_drv_cap_cache *cache_ent, *tmp;
235 
236 	list_for_each_entry_safe(cache_ent, tmp, &vgdev->cap_cache, head) {
237 		kfree(cache_ent->caps_cache);
238 		kfree(cache_ent);
239 	}
240 }
241 
242 void virtio_gpu_driver_unload(struct drm_device *dev)
243 {
244 	struct virtio_gpu_device *vgdev = dev->dev_private;
245 
246 	vgdev->vqs_ready = false;
247 	flush_work(&vgdev->ctrlq.dequeue_work);
248 	flush_work(&vgdev->cursorq.dequeue_work);
249 	flush_work(&vgdev->config_changed_work);
250 	vgdev->vdev->config->del_vqs(vgdev->vdev);
251 
252 	virtio_gpu_modeset_fini(vgdev);
253 	virtio_gpu_ttm_fini(vgdev);
254 	virtio_gpu_free_vbufs(vgdev);
255 	virtio_gpu_cleanup_cap_cache(vgdev);
256 	kfree(vgdev->capsets);
257 	kfree(vgdev);
258 }
259 
260 int virtio_gpu_driver_open(struct drm_device *dev, struct drm_file *file)
261 {
262 	struct virtio_gpu_device *vgdev = dev->dev_private;
263 	struct virtio_gpu_fpriv *vfpriv;
264 	int id;
265 	char dbgname[TASK_COMM_LEN];
266 
267 	/* can't create contexts without 3d renderer */
268 	if (!vgdev->has_virgl_3d)
269 		return 0;
270 
271 	/* allocate a virt GPU context for this opener */
272 	vfpriv = kzalloc(sizeof(*vfpriv), GFP_KERNEL);
273 	if (!vfpriv)
274 		return -ENOMEM;
275 
276 	get_task_comm(dbgname, current);
277 	id = virtio_gpu_context_create(vgdev, strlen(dbgname), dbgname);
278 	if (id < 0) {
279 		kfree(vfpriv);
280 		return id;
281 	}
282 
283 	vfpriv->ctx_id = id;
284 	file->driver_priv = vfpriv;
285 	return 0;
286 }
287 
288 void virtio_gpu_driver_postclose(struct drm_device *dev, struct drm_file *file)
289 {
290 	struct virtio_gpu_device *vgdev = dev->dev_private;
291 	struct virtio_gpu_fpriv *vfpriv;
292 
293 	if (!vgdev->has_virgl_3d)
294 		return;
295 
296 	vfpriv = file->driver_priv;
297 
298 	virtio_gpu_context_destroy(vgdev, vfpriv->ctx_id);
299 	kfree(vfpriv);
300 	file->driver_priv = NULL;
301 }
302