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