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         struct sg_table *pages;
  26         int err;
  27
  28         obj->mm.res = intel_region_ttm_resource_alloc(obj->mm.region,
  29                                                       obj->bo_offset,
  30                                                       obj->base.size,
  31                                                       obj->flags);
  32         if (IS_ERR(obj->mm.res))
  33                 return PTR_ERR(obj->mm.res);
  34
  35         obj->mm.rsgt = intel_region_ttm_resource_to_rsgt(obj->mm.region,
  36                                                          obj->mm.res,
  37                                                          obj->mm.region->min_page_size);
  38         if (IS_ERR(obj->mm.rsgt)) {
  39                 err = PTR_ERR(obj->mm.rsgt);
  40                 goto err_free_resource;
  41         }
  42
  43         pages = &obj->mm.rsgt->table;
  44         __i915_gem_object_set_pages(obj, pages);
  45
  46         return 0;
  47
  48 err_free_resource:
  49         intel_region_ttm_resource_free(obj->mm.region, obj->mm.res);
  50         return err;
  51 }
  52
  53 static const struct drm_i915_gem_object_ops mock_region_obj_ops = {
  54         .name = "mock-region",
  55         .get_pages = mock_region_get_pages,
  56         .put_pages = mock_region_put_pages,
  57         .release = i915_gem_object_release_memory_region,
  58 };
  59
  60 static int mock_object_init(struct intel_memory_region *mem,
  61                             struct drm_i915_gem_object *obj,
  62                             resource_size_t offset,
  63                             resource_size_t size,
  64                             resource_size_t page_size,
  65                             unsigned int flags)
  66 {
  67         static struct lock_class_key lock_class;
  68         struct drm_i915_private *i915 = mem->i915;
  69
  70         if (size > resource_size(&mem->region))
  71                 return -E2BIG;
  72
  73         drm_gem_private_object_init(&i915->drm, &obj->base, size);
  74         i915_gem_object_init(obj, &mock_region_obj_ops, &lock_class, flags);
  75
  76         obj->bo_offset = offset;
  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                    resource_size_t io_size)
 112 {
 113         int instance = ida_alloc_max(&i915->selftest.mock_region_instances,
 114                                      TTM_NUM_MEM_TYPES - TTM_PL_PRIV - 1,
 115                                      GFP_KERNEL);
 116
 117         if (instance < 0)
 118                 return ERR_PTR(instance);
 119
 120         return intel_memory_region_create(i915, start, size, min_page_size,
 121                                           io_start, io_size,
 122                                           INTEL_MEMORY_MOCK, instance,
 123                                           &mock_region_ops);
 124 }