drivers/gpu/drm/i915/selftests/mock_region.c

   1 // SPDX-License-Identifier: MIT
   2 /*
   3  * Copyright © 2019-2021 Intel Corporation
   4  */
   5
   6 #include <drm/ttm/ttm_placement.h>
   7 #include <linux/scatterlist.h>
   8
   9 #include "gem/i915_gem_region.h"
  10 #include "intel_memory_region.h"
  11 #include "intel_region_ttm.h"
  12
  13 #include "mock_region.h"
  14
  15 static void mock_region_put_pages(struct drm_i915_gem_object *obj,
  16                                   struct sg_table *pages)
  17 {
  18         i915_refct_sgt_put(obj->mm.rsgt);
  19         obj->mm.rsgt = NULL;
  20         intel_region_ttm_resource_free(obj->mm.region, obj->mm.res);
  21 }
  22
  23 static int mock_region_get_pages(struct drm_i915_gem_object *obj)
  24 {
  25         unsigned int flags;
  26         struct sg_table *pages;
  27         int err;
  28
  29         flags = 0;
  30         if (obj->flags & I915_BO_ALLOC_CONTIGUOUS)
  31                 flags |= TTM_PL_FLAG_CONTIGUOUS;
  32
  33         obj->mm.res = intel_region_ttm_resource_alloc(obj->mm.region,
  34                                                       obj->base.size,
  35                                                       flags);
  36         if (IS_ERR(obj->mm.res))
  37                 return PTR_ERR(obj->mm.res);
  38
  39         obj->mm.rsgt = intel_region_ttm_resource_to_rsgt(obj->mm.region,
  40                                                          obj->mm.res);
  41         if (IS_ERR(obj->mm.rsgt)) {
  42                 err = PTR_ERR(obj->mm.rsgt);
  43                 goto err_free_resource;
  44         }
  45
  46         pages = &obj->mm.rsgt->table;
  47         __i915_gem_object_set_pages(obj, pages, i915_sg_dma_sizes(pages->sgl));
  48
  49         return 0;
  50
  51 err_free_resource:
  52         intel_region_ttm_resource_free(obj->mm.region, obj->mm.res);
  53         return err;
  54 }
  55
  56 static const struct drm_i915_gem_object_ops mock_region_obj_ops = {
  57         .name = "mock-region",
  58         .get_pages = mock_region_get_pages,
  59         .put_pages = mock_region_put_pages,
  60         .release = i915_gem_object_release_memory_region,
  61 };
  62
  63 static int mock_object_init(struct intel_memory_region *mem,
  64                             struct drm_i915_gem_object *obj,
  65                             resource_size_t size,
  66                             resource_size_t page_size,
  67                             unsigned int flags)
  68 {
  69         static struct lock_class_key lock_class;
  70         struct drm_i915_private *i915 = mem->i915;
  71
  72         if (size > resource_size(&mem->region))
  73                 return -E2BIG;
  74
  75         drm_gem_private_object_init(&i915->drm, &obj->base, size);
  76         i915_gem_object_init(obj, &mock_region_obj_ops, &lock_class, flags);
  77
  78         obj->read_domains = I915_GEM_DOMAIN_CPU | I915_GEM_DOMAIN_GTT;
  79
  80         i915_gem_object_set_cache_coherency(obj, I915_CACHE_NONE);
  81
  82         i915_gem_object_init_memory_region(obj, mem);
  83
  84         return 0;
  85 }
  86
  87 static int mock_region_fini(struct intel_memory_region *mem)
  88 {
  89         struct drm_i915_private *i915 = mem->i915;
  90         int instance = mem->instance;
  91         int ret;
  92
  93         ret = intel_region_ttm_fini(mem);
  94         ida_free(&i915->selftest.mock_region_instances, instance);
  95
  96         return ret;
  97 }
  98
  99 static const struct intel_memory_region_ops mock_region_ops = {
 100         .init = intel_region_ttm_init,
 101         .release = mock_region_fini,
 102         .init_object = mock_object_init,
 103 };
 104
 105 struct intel_memory_region *
 106 mock_region_create(struct drm_i915_private *i915,
 107                    resource_size_t start,
 108                    resource_size_t size,
 109                    resource_size_t min_page_size,
 110                    resource_size_t io_start)
 111 {
 112         int instance = ida_alloc_max(&i915->selftest.mock_region_instances,
 113                                      TTM_NUM_MEM_TYPES - TTM_PL_PRIV - 1,
 114                                      GFP_KERNEL);
 115
 116         if (instance < 0)
 117                 return ERR_PTR(instance);
 118
 119         return intel_memory_region_create(i915, start, size, min_page_size,
 120                                           io_start, INTEL_MEMORY_MOCK, instance,
 121                                           &mock_region_ops);
 122 }