1/*
  2 * Copyright 2012 Red Hat Inc.
  3 *
  4 * Permission is hereby granted, free of charge, to any person obtaining a
  5 * copy of this software and associated documentation files (the
  6 * "Software"), to deal in the Software without restriction, including
  7 * without limitation the rights to use, copy, modify, merge, publish,
  8 * distribute, sub license, and/or sell copies of the Software, and to
  9 * permit persons to whom the Software is furnished to do so, subject to
 10 * the following conditions:
 11 *
 12 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 13 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 14 * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
 15 * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM,
 16 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
 17 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
 18 * USE OR OTHER DEALINGS IN THE SOFTWARE.
 19 *
 20 * The above copyright notice and this permission notice (including the
 21 * next paragraph) shall be included in all copies or substantial portions
 22 * of the Software.
 23 *
 24 */
 25/*
 26 * Authors: Dave Airlie <airlied@redhat.com>
 27 */
 28#include <drm/drmP.h>
 29#include <drm/ttm/ttm_page_alloc.h>
 30
 31#include "ast_drv.h"
 32
 33static inline struct ast_private *
 34ast_bdev(struct ttm_bo_device *bd)
 35{
 36	return container_of(bd, struct ast_private, ttm.bdev);
 37}
 38
 39static int
 40ast_ttm_mem_global_init(struct drm_global_reference *ref)
 41{
 42	return ttm_mem_global_init(ref->object);
 43}
 44
 45static void
 46ast_ttm_mem_global_release(struct drm_global_reference *ref)
 47{
 48	ttm_mem_global_release(ref->object);
 49}
 50
 51static int ast_ttm_global_init(struct ast_private *ast)
 52{
 53	struct drm_global_reference *global_ref;
 54	int r;
 55
 56	global_ref = &ast->ttm.mem_global_ref;
 57	global_ref->global_type = DRM_GLOBAL_TTM_MEM;
 58	global_ref->size = sizeof(struct ttm_mem_global);
 59	global_ref->init = &ast_ttm_mem_global_init;
 60	global_ref->release = &ast_ttm_mem_global_release;
 61	r = drm_global_item_ref(global_ref);
 62	if (r != 0) {
 63		DRM_ERROR("Failed setting up TTM memory accounting "
 64			  "subsystem.\n");
 65		return r;
 66	}
 67
 68	ast->ttm.bo_global_ref.mem_glob =
 69		ast->ttm.mem_global_ref.object;
 70	global_ref = &ast->ttm.bo_global_ref.ref;
 71	global_ref->global_type = DRM_GLOBAL_TTM_BO;
 72	global_ref->size = sizeof(struct ttm_bo_global);
 73	global_ref->init = &ttm_bo_global_init;
 74	global_ref->release = &ttm_bo_global_release;
 75	r = drm_global_item_ref(global_ref);
 76	if (r != 0) {
 77		DRM_ERROR("Failed setting up TTM BO subsystem.\n");
 78		drm_global_item_unref(&ast->ttm.mem_global_ref);
 79		return r;
 80	}
 81	return 0;
 82}
 83
 84static void
 85ast_ttm_global_release(struct ast_private *ast)
 86{
 87	if (ast->ttm.mem_global_ref.release == NULL)
 88		return;
 89
 90	drm_global_item_unref(&ast->ttm.bo_global_ref.ref);
 91	drm_global_item_unref(&ast->ttm.mem_global_ref);
 92	ast->ttm.mem_global_ref.release = NULL;
 93}
 94
 95
 96static void ast_bo_ttm_destroy(struct ttm_buffer_object *tbo)
 97{
 98	struct ast_bo *bo;
 99
100	bo = container_of(tbo, struct ast_bo, bo);
101
102	drm_gem_object_release(&bo->gem);
103	kfree(bo);
104}
105
106static bool ast_ttm_bo_is_ast_bo(struct ttm_buffer_object *bo)
107{
108	if (bo->destroy == &ast_bo_ttm_destroy)
109		return true;
110	return false;
111}
112
113static int
114ast_bo_init_mem_type(struct ttm_bo_device *bdev, uint32_t type,
115		     struct ttm_mem_type_manager *man)
116{
117	switch (type) {
118	case TTM_PL_SYSTEM:
119		man->flags = TTM_MEMTYPE_FLAG_MAPPABLE;
120		man->available_caching = TTM_PL_MASK_CACHING;
121		man->default_caching = TTM_PL_FLAG_CACHED;
122		break;
123	case TTM_PL_VRAM:
124		man->func = &ttm_bo_manager_func;
125		man->flags = TTM_MEMTYPE_FLAG_FIXED |
126			TTM_MEMTYPE_FLAG_MAPPABLE;
127		man->available_caching = TTM_PL_FLAG_UNCACHED |
128			TTM_PL_FLAG_WC;
129		man->default_caching = TTM_PL_FLAG_WC;
130		break;
131	default:
132		DRM_ERROR("Unsupported memory type %u\n", (unsigned)type);
133		return -EINVAL;
134	}
135	return 0;
136}
137
138static void
139ast_bo_evict_flags(struct ttm_buffer_object *bo, struct ttm_placement *pl)
140{
141	struct ast_bo *astbo = ast_bo(bo);
142
143	if (!ast_ttm_bo_is_ast_bo(bo))
144		return;
145
146	ast_ttm_placement(astbo, TTM_PL_FLAG_SYSTEM);
147	*pl = astbo->placement;
148}
149
150static int ast_bo_verify_access(struct ttm_buffer_object *bo, struct file *filp)
151{
152	struct ast_bo *astbo = ast_bo(bo);
153
154	return drm_vma_node_verify_access(&astbo->gem.vma_node,
155					  filp->private_data);
156}
157
158static int ast_ttm_io_mem_reserve(struct ttm_bo_device *bdev,
159				  struct ttm_mem_reg *mem)
160{
161	struct ttm_mem_type_manager *man = &bdev->man[mem->mem_type];
162	struct ast_private *ast = ast_bdev(bdev);
163
164	mem->bus.addr = NULL;
165	mem->bus.offset = 0;
166	mem->bus.size = mem->num_pages << PAGE_SHIFT;
167	mem->bus.base = 0;
168	mem->bus.is_iomem = false;
169	if (!(man->flags & TTM_MEMTYPE_FLAG_MAPPABLE))
170		return -EINVAL;
171	switch (mem->mem_type) {
172	case TTM_PL_SYSTEM:
173		/* system memory */
174		return 0;
175	case TTM_PL_VRAM:
176		mem->bus.offset = mem->start << PAGE_SHIFT;
177		mem->bus.base = pci_resource_start(ast->dev->pdev, 0);
178		mem->bus.is_iomem = true;
179		break;
180	default:
181		return -EINVAL;
182		break;
183	}
184	return 0;
185}
186
187static void ast_ttm_io_mem_free(struct ttm_bo_device *bdev, struct ttm_mem_reg *mem)
188{
189}
190
191static void ast_ttm_backend_destroy(struct ttm_tt *tt)
192{
193	ttm_tt_fini(tt);
194	kfree(tt);
195}
196
197static struct ttm_backend_func ast_tt_backend_func = {
198	.destroy = &ast_ttm_backend_destroy,
199};
200
201
202static struct ttm_tt *ast_ttm_tt_create(struct ttm_buffer_object *bo,
203					uint32_t page_flags)
204{
205	struct ttm_tt *tt;
206
207	tt = kzalloc(sizeof(struct ttm_tt), GFP_KERNEL);
208	if (tt == NULL)
209		return NULL;
210	tt->func = &ast_tt_backend_func;
211	if (ttm_tt_init(tt, bo, page_flags)) {
212		kfree(tt);
213		return NULL;
214	}
215	return tt;
216}
217
218struct ttm_bo_driver ast_bo_driver = {
219	.ttm_tt_create = ast_ttm_tt_create,
220	.init_mem_type = ast_bo_init_mem_type,
221	.eviction_valuable = ttm_bo_eviction_valuable,
222	.evict_flags = ast_bo_evict_flags,
223	.move = NULL,
224	.verify_access = ast_bo_verify_access,
225	.io_mem_reserve = &ast_ttm_io_mem_reserve,
226	.io_mem_free = &ast_ttm_io_mem_free,
227};
228
229int ast_mm_init(struct ast_private *ast)
230{
231	int ret;
232	struct drm_device *dev = ast->dev;
233	struct ttm_bo_device *bdev = &ast->ttm.bdev;
234
235	ret = ast_ttm_global_init(ast);
236	if (ret)
237		return ret;
238
239	ret = ttm_bo_device_init(&ast->ttm.bdev,
240				 ast->ttm.bo_global_ref.ref.object,
241				 &ast_bo_driver,
242				 dev->anon_inode->i_mapping,
243				 DRM_FILE_PAGE_OFFSET,
244				 true);
245	if (ret) {
246		DRM_ERROR("Error initialising bo driver; %d\n", ret);
247		return ret;
248	}
249
250	ret = ttm_bo_init_mm(bdev, TTM_PL_VRAM,
251			     ast->vram_size >> PAGE_SHIFT);
252	if (ret) {
253		DRM_ERROR("Failed ttm VRAM init: %d\n", ret);
254		return ret;
255	}
256
257	arch_io_reserve_memtype_wc(pci_resource_start(dev->pdev, 0),
258				   pci_resource_len(dev->pdev, 0));
259	ast->fb_mtrr = arch_phys_wc_add(pci_resource_start(dev->pdev, 0),
260					pci_resource_len(dev->pdev, 0));
261
262	return 0;
263}
264
265void ast_mm_fini(struct ast_private *ast)
266{
267	struct drm_device *dev = ast->dev;
268
269	ttm_bo_device_release(&ast->ttm.bdev);
270
271	ast_ttm_global_release(ast);
272
273	arch_phys_wc_del(ast->fb_mtrr);
274	arch_io_free_memtype_wc(pci_resource_start(dev->pdev, 0),
275				pci_resource_len(dev->pdev, 0));
276}
277
278void ast_ttm_placement(struct ast_bo *bo, int domain)
279{
280	u32 c = 0;
281	unsigned i;
282
283	bo->placement.placement = bo->placements;
284	bo->placement.busy_placement = bo->placements;
285	if (domain & TTM_PL_FLAG_VRAM)
286		bo->placements[c++].flags = TTM_PL_FLAG_WC | TTM_PL_FLAG_UNCACHED | TTM_PL_FLAG_VRAM;
287	if (domain & TTM_PL_FLAG_SYSTEM)
288		bo->placements[c++].flags = TTM_PL_FLAG_CACHED | TTM_PL_FLAG_SYSTEM;
289	if (!c)
290		bo->placements[c++].flags = TTM_PL_FLAG_CACHED | TTM_PL_FLAG_SYSTEM;
291	bo->placement.num_placement = c;
292	bo->placement.num_busy_placement = c;
293	for (i = 0; i < c; ++i) {
294		bo->placements[i].fpfn = 0;
295		bo->placements[i].lpfn = 0;
296	}
297}
298
299int ast_bo_create(struct drm_device *dev, int size, int align,
300		  uint32_t flags, struct ast_bo **pastbo)
301{
302	struct ast_private *ast = dev->dev_private;
303	struct ast_bo *astbo;
304	size_t acc_size;
305	int ret;
306
307	astbo = kzalloc(sizeof(struct ast_bo), GFP_KERNEL);
308	if (!astbo)
309		return -ENOMEM;
310
311	ret = drm_gem_object_init(dev, &astbo->gem, size);
312	if (ret)
313		goto error;
314
315	astbo->bo.bdev = &ast->ttm.bdev;
316
317	ast_ttm_placement(astbo, TTM_PL_FLAG_VRAM | TTM_PL_FLAG_SYSTEM);
318
319	acc_size = ttm_bo_dma_acc_size(&ast->ttm.bdev, size,
320				       sizeof(struct ast_bo));
321
322	ret = ttm_bo_init(&ast->ttm.bdev, &astbo->bo, size,
323			  ttm_bo_type_device, &astbo->placement,
324			  align >> PAGE_SHIFT, false, acc_size,
325			  NULL, NULL, ast_bo_ttm_destroy);
326	if (ret)
327		goto error;
328
329	*pastbo = astbo;
330	return 0;
331error:
332	kfree(astbo);
333	return ret;
334}
335
336static inline u64 ast_bo_gpu_offset(struct ast_bo *bo)
337{
338	return bo->bo.offset;
339}
340
341int ast_bo_pin(struct ast_bo *bo, u32 pl_flag, u64 *gpu_addr)
342{
343	struct ttm_operation_ctx ctx = { false, false };
344	int i, ret;
345
346	if (bo->pin_count) {
347		bo->pin_count++;
348		if (gpu_addr)
349			*gpu_addr = ast_bo_gpu_offset(bo);
350	}
351
352	ast_ttm_placement(bo, pl_flag);
353	for (i = 0; i < bo->placement.num_placement; i++)
354		bo->placements[i].flags |= TTM_PL_FLAG_NO_EVICT;
355	ret = ttm_bo_validate(&bo->bo, &bo->placement, &ctx);
356	if (ret)
357		return ret;
358
359	bo->pin_count = 1;
360	if (gpu_addr)
361		*gpu_addr = ast_bo_gpu_offset(bo);
362	return 0;
363}
364
365int ast_bo_unpin(struct ast_bo *bo)
366{
367	struct ttm_operation_ctx ctx = { false, false };
368	int i;
369	if (!bo->pin_count) {
370		DRM_ERROR("unpin bad %p\n", bo);
371		return 0;
372	}
373	bo->pin_count--;
374	if (bo->pin_count)
375		return 0;
376
377	for (i = 0; i < bo->placement.num_placement ; i++)
378		bo->placements[i].flags &= ~TTM_PL_FLAG_NO_EVICT;
379	return ttm_bo_validate(&bo->bo, &bo->placement, &ctx);
380}
381
382int ast_bo_push_sysram(struct ast_bo *bo)
383{
384	struct ttm_operation_ctx ctx = { false, false };
385	int i, ret;
386	if (!bo->pin_count) {
387		DRM_ERROR("unpin bad %p\n", bo);
388		return 0;
389	}
390	bo->pin_count--;
391	if (bo->pin_count)
392		return 0;
393
394	if (bo->kmap.virtual)
395		ttm_bo_kunmap(&bo->kmap);
396
397	ast_ttm_placement(bo, TTM_PL_FLAG_SYSTEM);
398	for (i = 0; i < bo->placement.num_placement ; i++)
399		bo->placements[i].flags |= TTM_PL_FLAG_NO_EVICT;
400
401	ret = ttm_bo_validate(&bo->bo, &bo->placement, &ctx);
402	if (ret) {
403		DRM_ERROR("pushing to VRAM failed\n");
404		return ret;
405	}
406	return 0;
407}
408
409int ast_mmap(struct file *filp, struct vm_area_struct *vma)
410{
411	struct drm_file *file_priv;
412	struct ast_private *ast;
413
414	if (unlikely(vma->vm_pgoff < DRM_FILE_PAGE_OFFSET))
415		return -EINVAL;
416
417	file_priv = filp->private_data;
418	ast = file_priv->minor->dev->dev_private;
419	return ttm_bo_mmap(filp, vma, &ast->ttm.bdev);
420}