1 /* SPDX-License-Identifier: GPL-2.0 OR MIT */
2 /*
3  * Copyright 2021 VMware, Inc.
4  *
5  * Permission is hereby granted, free of charge, to any person
6  * obtaining a copy of this software and associated documentation
7  * files (the "Software"), to deal in the Software without
8  * restriction, including without limitation the rights to use, copy,
9  * modify, merge, publish, distribute, sublicense, and/or sell copies
10  * of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be
14  * included in all copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
17  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
19  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
20  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
21  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
22  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
23  * SOFTWARE.
24  *
25  */
26 
27 #include "vmwgfx_drv.h"
28 
29 #include "drm/drm_prime.h"
30 #include "drm/drm_gem_ttm_helper.h"
31 
32 /**
33  * vmw_buffer_object - Convert a struct ttm_buffer_object to a struct
34  * vmw_buffer_object.
35  *
36  * @bo: Pointer to the TTM buffer object.
37  * Return: Pointer to the struct vmw_buffer_object embedding the
38  * TTM buffer object.
39  */
40 static struct vmw_buffer_object *
41 vmw_buffer_object(struct ttm_buffer_object *bo)
42 {
43 	return container_of(bo, struct vmw_buffer_object, base);
44 }
45 
46 static void vmw_gem_object_free(struct drm_gem_object *gobj)
47 {
48 	struct ttm_buffer_object *bo = drm_gem_ttm_of_gem(gobj);
49 	if (bo) {
50 		ttm_bo_put(bo);
51 	}
52 }
53 
54 static int vmw_gem_object_open(struct drm_gem_object *obj,
55 			       struct drm_file *file_priv)
56 {
57 	return 0;
58 }
59 
60 static void vmw_gem_object_close(struct drm_gem_object *obj,
61 				 struct drm_file *file_priv)
62 {
63 }
64 
65 static int vmw_gem_pin_private(struct drm_gem_object *obj, bool do_pin)
66 {
67 	struct ttm_buffer_object *bo = drm_gem_ttm_of_gem(obj);
68 	struct vmw_buffer_object *vbo = vmw_buffer_object(bo);
69 	int ret;
70 
71 	ret = ttm_bo_reserve(bo, false, false, NULL);
72 	if (unlikely(ret != 0))
73 		goto err;
74 
75 	vmw_bo_pin_reserved(vbo, do_pin);
76 
77 	ttm_bo_unreserve(bo);
78 
79 err:
80 	return ret;
81 }
82 
83 
84 static int vmw_gem_object_pin(struct drm_gem_object *obj)
85 {
86 	return vmw_gem_pin_private(obj, true);
87 }
88 
89 static void vmw_gem_object_unpin(struct drm_gem_object *obj)
90 {
91 	vmw_gem_pin_private(obj, false);
92 }
93 
94 static struct sg_table *vmw_gem_object_get_sg_table(struct drm_gem_object *obj)
95 {
96 	struct ttm_buffer_object *bo = drm_gem_ttm_of_gem(obj);
97 	struct vmw_ttm_tt *vmw_tt =
98 		container_of(bo->ttm, struct vmw_ttm_tt, dma_ttm);
99 
100 	if (vmw_tt->vsgt.sgt)
101 		return vmw_tt->vsgt.sgt;
102 
103 	return drm_prime_pages_to_sg(obj->dev, vmw_tt->dma_ttm.pages, vmw_tt->dma_ttm.num_pages);
104 }
105 
106 
107 static const struct drm_gem_object_funcs vmw_gem_object_funcs = {
108 	.free = vmw_gem_object_free,
109 	.open = vmw_gem_object_open,
110 	.close = vmw_gem_object_close,
111 	.print_info = drm_gem_ttm_print_info,
112 	.pin = vmw_gem_object_pin,
113 	.unpin = vmw_gem_object_unpin,
114 	.get_sg_table = vmw_gem_object_get_sg_table,
115 	.vmap = drm_gem_ttm_vmap,
116 	.vunmap = drm_gem_ttm_vunmap,
117 	.mmap = drm_gem_ttm_mmap,
118 };
119 
120 /**
121  * vmw_gem_destroy - vmw buffer object destructor
122  *
123  * @bo: Pointer to the embedded struct ttm_buffer_object
124  */
125 void vmw_gem_destroy(struct ttm_buffer_object *bo)
126 {
127 	struct vmw_buffer_object *vbo = vmw_buffer_object(bo);
128 
129 	WARN_ON(vbo->dirty);
130 	WARN_ON(!RB_EMPTY_ROOT(&vbo->res_tree));
131 	vmw_bo_unmap(vbo);
132 	drm_gem_object_release(&vbo->base.base);
133 	kfree(vbo);
134 }
135 
136 int vmw_gem_object_create_with_handle(struct vmw_private *dev_priv,
137 				      struct drm_file *filp,
138 				      uint32_t size,
139 				      uint32_t *handle,
140 				      struct vmw_buffer_object **p_vbo)
141 {
142 	int ret;
143 
144 	ret = vmw_bo_create(dev_priv, size,
145 			    (dev_priv->has_mob) ?
146 				    &vmw_sys_placement :
147 				    &vmw_vram_sys_placement,
148 			    true, false, &vmw_gem_destroy, p_vbo);
149 
150 	(*p_vbo)->base.base.funcs = &vmw_gem_object_funcs;
151 	if (ret != 0)
152 		goto out_no_bo;
153 
154 	ret = drm_gem_handle_create(filp, &(*p_vbo)->base.base, handle);
155 	/* drop reference from allocate - handle holds it now */
156 	drm_gem_object_put(&(*p_vbo)->base.base);
157 out_no_bo:
158 	return ret;
159 }
160 
161 
162 int vmw_gem_object_create_ioctl(struct drm_device *dev, void *data,
163 				struct drm_file *filp)
164 {
165 	struct vmw_private *dev_priv = vmw_priv(dev);
166 	union drm_vmw_alloc_dmabuf_arg *arg =
167 	    (union drm_vmw_alloc_dmabuf_arg *)data;
168 	struct drm_vmw_alloc_dmabuf_req *req = &arg->req;
169 	struct drm_vmw_dmabuf_rep *rep = &arg->rep;
170 	struct vmw_buffer_object *vbo;
171 	uint32_t handle;
172 	int ret;
173 
174 	ret = vmw_gem_object_create_with_handle(dev_priv, filp,
175 						req->size, &handle, &vbo);
176 	if (ret)
177 		goto out_no_bo;
178 
179 	rep->handle = handle;
180 	rep->map_handle = drm_vma_node_offset_addr(&vbo->base.base.vma_node);
181 	rep->cur_gmr_id = handle;
182 	rep->cur_gmr_offset = 0;
183 out_no_bo:
184 	return ret;
185 }
186 
187 #if defined(CONFIG_DEBUG_FS)
188 
189 static void vmw_bo_print_info(int id, struct vmw_buffer_object *bo, struct seq_file *m)
190 {
191 	const char *placement;
192 	const char *type;
193 
194 	switch (bo->base.resource->mem_type) {
195 	case TTM_PL_SYSTEM:
196 		placement = " CPU";
197 		break;
198 	case VMW_PL_GMR:
199 		placement = " GMR";
200 		break;
201 	case VMW_PL_MOB:
202 		placement = " MOB";
203 		break;
204 	case VMW_PL_SYSTEM:
205 		placement = "VCPU";
206 		break;
207 	case TTM_PL_VRAM:
208 		placement = "VRAM";
209 		break;
210 	default:
211 		placement = "None";
212 		break;
213 	}
214 
215 	switch (bo->base.type) {
216 	case ttm_bo_type_device:
217 		type = "device";
218 		break;
219 	case ttm_bo_type_kernel:
220 		type = "kernel";
221 		break;
222 	case ttm_bo_type_sg:
223 		type = "sg    ";
224 		break;
225 	default:
226 		type = "none  ";
227 		break;
228 	}
229 
230 	seq_printf(m, "\t\t0x%08x: %12zu bytes %s, type = %s",
231 		   id, bo->base.base.size, placement, type);
232 	seq_printf(m, ", priority = %u, pin_count = %u, GEM refs = %d, TTM refs = %d",
233 		   bo->base.priority,
234 		   bo->base.pin_count,
235 		   kref_read(&bo->base.base.refcount),
236 		   kref_read(&bo->base.kref));
237 	seq_puts(m, "\n");
238 }
239 
240 static int vmw_debugfs_gem_info_show(struct seq_file *m, void *unused)
241 {
242 	struct vmw_private *vdev = (struct vmw_private *)m->private;
243 	struct drm_device *dev = &vdev->drm;
244 	struct drm_file *file;
245 	int r;
246 
247 	r = mutex_lock_interruptible(&dev->filelist_mutex);
248 	if (r)
249 		return r;
250 
251 	list_for_each_entry(file, &dev->filelist, lhead) {
252 		struct task_struct *task;
253 		struct drm_gem_object *gobj;
254 		int id;
255 
256 		/*
257 		 * Although we have a valid reference on file->pid, that does
258 		 * not guarantee that the task_struct who called get_pid() is
259 		 * still alive (e.g. get_pid(current) => fork() => exit()).
260 		 * Therefore, we need to protect this ->comm access using RCU.
261 		 */
262 		rcu_read_lock();
263 		task = pid_task(file->pid, PIDTYPE_PID);
264 		seq_printf(m, "pid %8d command %s:\n", pid_nr(file->pid),
265 			   task ? task->comm : "<unknown>");
266 		rcu_read_unlock();
267 
268 		spin_lock(&file->table_lock);
269 		idr_for_each_entry(&file->object_idr, gobj, id) {
270 			struct vmw_buffer_object *bo = gem_to_vmw_bo(gobj);
271 
272 			vmw_bo_print_info(id, bo, m);
273 		}
274 		spin_unlock(&file->table_lock);
275 	}
276 
277 	mutex_unlock(&dev->filelist_mutex);
278 	return 0;
279 }
280 
281 DEFINE_SHOW_ATTRIBUTE(vmw_debugfs_gem_info);
282 
283 #endif
284 
285 void vmw_debugfs_gem_init(struct vmw_private *vdev)
286 {
287 #if defined(CONFIG_DEBUG_FS)
288 	struct drm_minor *minor = vdev->drm.primary;
289 	struct dentry *root = minor->debugfs_root;
290 
291 	debugfs_create_file("vmwgfx_gem_info", 0444, root, vdev,
292 			    &vmw_debugfs_gem_info_fops);
293 #endif
294 }
295