1// SPDX-License-Identifier: GPL-2.0 AND MIT
2/*
3 * Copyright © 2023 Intel Corporation
4 */
5#include <drm/ttm/ttm_tt.h>
6
7#include "ttm_kunit_helpers.h"
8
9static struct ttm_tt *ttm_tt_simple_create(struct ttm_buffer_object *bo,
10 uint32_t page_flags)
11{
12 struct ttm_tt *tt;
13
14 tt = kzalloc(size: sizeof(*tt), GFP_KERNEL);
15 ttm_tt_init(ttm: tt, bo, page_flags, caching: ttm_cached, extra_pages: 0);
16
17 return tt;
18}
19
20static void ttm_tt_simple_destroy(struct ttm_device *bdev, struct ttm_tt *ttm)
21{
22 kfree(objp: ttm);
23}
24
25static void dummy_ttm_bo_destroy(struct ttm_buffer_object *bo)
26{
27}
28
29struct ttm_device_funcs ttm_dev_funcs = {
30 .ttm_tt_create = ttm_tt_simple_create,
31 .ttm_tt_destroy = ttm_tt_simple_destroy,
32};
33EXPORT_SYMBOL_GPL(ttm_dev_funcs);
34
35int ttm_device_kunit_init(struct ttm_test_devices *priv,
36 struct ttm_device *ttm,
37 bool use_dma_alloc,
38 bool use_dma32)
39{
40 struct drm_device *drm = priv->drm;
41 int err;
42
43 err = ttm_device_init(bdev: ttm, funcs: &ttm_dev_funcs, dev: drm->dev,
44 mapping: drm->anon_inode->i_mapping,
45 vma_manager: drm->vma_offset_manager,
46 use_dma_alloc, use_dma32);
47
48 return err;
49}
50EXPORT_SYMBOL_GPL(ttm_device_kunit_init);
51
52struct ttm_buffer_object *ttm_bo_kunit_init(struct kunit *test,
53 struct ttm_test_devices *devs,
54 size_t size)
55{
56 struct drm_gem_object gem_obj = { };
57 struct ttm_buffer_object *bo;
58 int err;
59
60 bo = kunit_kzalloc(test, size: sizeof(*bo), GFP_KERNEL);
61 KUNIT_ASSERT_NOT_NULL(test, bo);
62
63 bo->base = gem_obj;
64 err = drm_gem_object_init(dev: devs->drm, obj: &bo->base, size);
65 KUNIT_ASSERT_EQ(test, err, 0);
66
67 bo->bdev = devs->ttm_dev;
68 bo->destroy = dummy_ttm_bo_destroy;
69
70 kref_init(kref: &bo->kref);
71
72 return bo;
73}
74EXPORT_SYMBOL_GPL(ttm_bo_kunit_init);
75
76struct ttm_place *ttm_place_kunit_init(struct kunit *test,
77 uint32_t mem_type, uint32_t flags)
78{
79 struct ttm_place *place;
80
81 place = kunit_kzalloc(test, size: sizeof(*place), GFP_KERNEL);
82 KUNIT_ASSERT_NOT_NULL(test, place);
83
84 place->mem_type = mem_type;
85 place->flags = flags;
86
87 return place;
88}
89EXPORT_SYMBOL_GPL(ttm_place_kunit_init);
90
91struct ttm_test_devices *ttm_test_devices_basic(struct kunit *test)
92{
93 struct ttm_test_devices *devs;
94
95 devs = kunit_kzalloc(test, size: sizeof(*devs), GFP_KERNEL);
96 KUNIT_ASSERT_NOT_NULL(test, devs);
97
98 devs->dev = drm_kunit_helper_alloc_device(test);
99 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, devs->dev);
100
101 devs->drm = __drm_kunit_helper_alloc_drm_device(test, dev: devs->dev,
102 size: sizeof(*devs->drm), offset: 0,
103 features: DRIVER_GEM);
104 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, devs->drm);
105
106 return devs;
107}
108EXPORT_SYMBOL_GPL(ttm_test_devices_basic);
109
110struct ttm_test_devices *ttm_test_devices_all(struct kunit *test)
111{
112 struct ttm_test_devices *devs;
113 struct ttm_device *ttm_dev;
114 int err;
115
116 devs = ttm_test_devices_basic(test);
117
118 ttm_dev = kunit_kzalloc(test, size: sizeof(*ttm_dev), GFP_KERNEL);
119 KUNIT_ASSERT_NOT_NULL(test, ttm_dev);
120
121 err = ttm_device_kunit_init(devs, ttm_dev, false, false);
122 KUNIT_ASSERT_EQ(test, err, 0);
123
124 devs->ttm_dev = ttm_dev;
125
126 return devs;
127}
128EXPORT_SYMBOL_GPL(ttm_test_devices_all);
129
130void ttm_test_devices_put(struct kunit *test, struct ttm_test_devices *devs)
131{
132 if (devs->ttm_dev)
133 ttm_device_fini(bdev: devs->ttm_dev);
134
135 drm_kunit_helper_free_device(test, dev: devs->dev);
136}
137EXPORT_SYMBOL_GPL(ttm_test_devices_put);
138
139int ttm_test_devices_init(struct kunit *test)
140{
141 struct ttm_test_devices *priv;
142
143 priv = kunit_kzalloc(test, size: sizeof(*priv), GFP_KERNEL);
144 KUNIT_ASSERT_NOT_NULL(test, priv);
145
146 priv = ttm_test_devices_basic(test);
147 test->priv = priv;
148
149 return 0;
150}
151EXPORT_SYMBOL_GPL(ttm_test_devices_init);
152
153void ttm_test_devices_fini(struct kunit *test)
154{
155 ttm_test_devices_put(test, test->priv);
156}
157EXPORT_SYMBOL_GPL(ttm_test_devices_fini);
158
159MODULE_LICENSE("GPL");
160

source code of linux/drivers/gpu/drm/ttm/tests/ttm_kunit_helpers.c