i915/selftests/mock_region.c

0001 // SPDX-License-Identifier: MIT
0002 /*
0003  * Copyright © 2019-2021 Intel Corporation
0004  */
0005
0006 #include <drm/ttm/ttm_placement.h>
0007 #include <linux/scatterlist.h>
0008
0009 #include "gem/i915_gem_region.h"
0010 #include "intel_memory_region.h"
0011 #include "intel_region_ttm.h"
0012
0013 #include "mock_region.h"
0014
0015 static void mock_region_put_pages(struct drm_i915_gem_object *obj,
0016                   struct sg_table *pages)
0017 {
0018     i915_refct_sgt_put(obj->mm.rsgt);
0019     obj->mm.rsgt = NULL;
0020     intel_region_ttm_resource_free(obj->mm.region, obj->mm.res);
0021 }
0022
0023 static int mock_region_get_pages(struct drm_i915_gem_object *obj)
0024 {
0025     struct sg_table *pages;
0026     int err;
0027
0028     obj->mm.res = intel_region_ttm_resource_alloc(obj->mm.region,
0029                               obj->bo_offset,
0030                               obj->base.size,
0031                               obj->flags);
0032     if (IS_ERR(obj->mm.res))
0033         return PTR_ERR(obj->mm.res);
0034
0035     obj->mm.rsgt = intel_region_ttm_resource_to_rsgt(obj->mm.region,
0036                              obj->mm.res,
0037                              obj->mm.region->min_page_size);
0038     if (IS_ERR(obj->mm.rsgt)) {
0039         err = PTR_ERR(obj->mm.rsgt);
0040         goto err_free_resource;
0041     }
0042
0043     pages = &obj->mm.rsgt->table;
0044     __i915_gem_object_set_pages(obj, pages, i915_sg_dma_sizes(pages->sgl));
0045
0046     return 0;
0047
0048 err_free_resource:
0049     intel_region_ttm_resource_free(obj->mm.region, obj->mm.res);
0050     return err;
0051 }
0052
0053 static const struct drm_i915_gem_object_ops mock_region_obj_ops = {
0054     .name = "mock-region",
0055     .get_pages = mock_region_get_pages,
0056     .put_pages = mock_region_put_pages,
0057     .release = i915_gem_object_release_memory_region,
0058 };
0059
0060 static int mock_object_init(struct intel_memory_region *mem,
0061                 struct drm_i915_gem_object *obj,
0062                 resource_size_t offset,
0063                 resource_size_t size,
0064                 resource_size_t page_size,
0065                 unsigned int flags)
0066 {
0067     static struct lock_class_key lock_class;
0068     struct drm_i915_private *i915 = mem->i915;
0069
0070     if (size > resource_size(&mem->region))
0071         return -E2BIG;
0072
0073     drm_gem_private_object_init(&i915->drm, &obj->base, size);
0074     i915_gem_object_init(obj, &mock_region_obj_ops, &lock_class, flags);
0075
0076     obj->bo_offset = offset;
0077
0078     obj->read_domains = I915_GEM_DOMAIN_CPU | I915_GEM_DOMAIN_GTT;
0079
0080     i915_gem_object_set_cache_coherency(obj, I915_CACHE_NONE);
0081
0082     i915_gem_object_init_memory_region(obj, mem);
0083
0084     return 0;
0085 }
0086
0087 static int mock_region_fini(struct intel_memory_region *mem)
0088 {
0089     struct drm_i915_private *i915 = mem->i915;
0090     int instance = mem->instance;
0091     int ret;
0092
0093     ret = intel_region_ttm_fini(mem);
0094     ida_free(&i915->selftest.mock_region_instances, instance);
0095
0096     return ret;
0097 }
0098
0099 static const struct intel_memory_region_ops mock_region_ops = {
0100     .init = intel_region_ttm_init,
0101     .release = mock_region_fini,
0102     .init_object = mock_object_init,
0103 };
0104
0105 struct intel_memory_region *
0106 mock_region_create(struct drm_i915_private *i915,
0107            resource_size_t start,
0108            resource_size_t size,
0109            resource_size_t min_page_size,
0110            resource_size_t io_start,
0111            resource_size_t io_size)
0112 {
0113     int instance = ida_alloc_max(&i915->selftest.mock_region_instances,
0114                      TTM_NUM_MEM_TYPES - TTM_PL_PRIV - 1,
0115                      GFP_KERNEL);
0116
0117     if (instance < 0)
0118         return ERR_PTR(instance);
0119
0120     return intel_memory_region_create(i915, start, size, min_page_size,
0121                       io_start, io_size,
0122                       INTEL_MEMORY_MOCK, instance,
0123                       &mock_region_ops);
0124 }