1 | // SPDX-License-Identifier: GPL-2.0 |
2 | /* |
3 | * Kunit test for drm_probe_helper functions |
4 | */ |
5 | |
6 | #include <drm/drm_atomic_state_helper.h> |
7 | #include <drm/drm_connector.h> |
8 | #include <drm/drm_device.h> |
9 | #include <drm/drm_drv.h> |
10 | #include <drm/drm_kunit_helpers.h> |
11 | #include <drm/drm_mode.h> |
12 | #include <drm/drm_modes.h> |
13 | #include <drm/drm_modeset_helper_vtables.h> |
14 | #include <drm/drm_probe_helper.h> |
15 | |
16 | #include <kunit/test.h> |
17 | |
18 | struct drm_probe_helper_test_priv { |
19 | struct drm_device *drm; |
20 | struct device *dev; |
21 | struct drm_connector connector; |
22 | }; |
23 | |
24 | static const struct drm_connector_helper_funcs drm_probe_helper_connector_helper_funcs = { |
25 | }; |
26 | |
27 | static const struct drm_connector_funcs drm_probe_helper_connector_funcs = { |
28 | .atomic_destroy_state = drm_atomic_helper_connector_destroy_state, |
29 | .atomic_duplicate_state = drm_atomic_helper_connector_duplicate_state, |
30 | .reset = drm_atomic_helper_connector_reset, |
31 | }; |
32 | |
33 | static int drm_probe_helper_test_init(struct kunit *test) |
34 | { |
35 | struct drm_probe_helper_test_priv *priv; |
36 | struct drm_connector *connector; |
37 | int ret; |
38 | |
39 | priv = kunit_kzalloc(test, size: sizeof(*priv), GFP_KERNEL); |
40 | KUNIT_ASSERT_NOT_NULL(test, priv); |
41 | test->priv = priv; |
42 | |
43 | priv->dev = drm_kunit_helper_alloc_device(test); |
44 | KUNIT_ASSERT_NOT_ERR_OR_NULL(test, priv->dev); |
45 | |
46 | priv->drm = __drm_kunit_helper_alloc_drm_device(test, dev: priv->dev, |
47 | size: sizeof(*priv->drm), offset: 0, |
48 | features: DRIVER_MODESET | DRIVER_ATOMIC); |
49 | KUNIT_ASSERT_NOT_ERR_OR_NULL(test, priv->drm); |
50 | |
51 | connector = &priv->connector; |
52 | ret = drmm_connector_init(dev: priv->drm, connector, |
53 | funcs: &drm_probe_helper_connector_funcs, |
54 | DRM_MODE_CONNECTOR_Unknown, |
55 | NULL); |
56 | KUNIT_ASSERT_EQ(test, ret, 0); |
57 | |
58 | drm_connector_helper_add(connector, funcs: &drm_probe_helper_connector_helper_funcs); |
59 | |
60 | return 0; |
61 | } |
62 | |
63 | typedef struct drm_display_mode *(*expected_mode_func_t)(struct drm_device *); |
64 | |
65 | struct drm_connector_helper_tv_get_modes_test { |
66 | const char *name; |
67 | unsigned int supported_tv_modes; |
68 | enum drm_connector_tv_mode default_mode; |
69 | bool cmdline; |
70 | enum drm_connector_tv_mode cmdline_mode; |
71 | expected_mode_func_t *expected_modes; |
72 | unsigned int num_expected_modes; |
73 | }; |
74 | |
75 | #define _TV_MODE_TEST(_name, _supported, _default, _cmdline, _cmdline_mode, ...) \ |
76 | { \ |
77 | .name = _name, \ |
78 | .supported_tv_modes = _supported, \ |
79 | .default_mode = _default, \ |
80 | .cmdline = _cmdline, \ |
81 | .cmdline_mode = _cmdline_mode, \ |
82 | .expected_modes = (expected_mode_func_t[]) { __VA_ARGS__ }, \ |
83 | .num_expected_modes = sizeof((expected_mode_func_t[]) { __VA_ARGS__ }) / \ |
84 | (sizeof(expected_mode_func_t)), \ |
85 | } |
86 | |
87 | #define TV_MODE_TEST(_name, _supported, _default, ...) \ |
88 | _TV_MODE_TEST(_name, _supported, _default, false, 0, __VA_ARGS__) |
89 | |
90 | #define TV_MODE_TEST_CMDLINE(_name, _supported, _default, _cmdline, ...) \ |
91 | _TV_MODE_TEST(_name, _supported, _default, true, _cmdline, __VA_ARGS__) |
92 | |
93 | static void |
94 | drm_test_connector_helper_tv_get_modes_check(struct kunit *test) |
95 | { |
96 | const struct drm_connector_helper_tv_get_modes_test *params = test->param_value; |
97 | struct drm_probe_helper_test_priv *priv = test->priv; |
98 | struct drm_connector *connector = &priv->connector; |
99 | struct drm_cmdline_mode *cmdline = &connector->cmdline_mode; |
100 | struct drm_display_mode *mode; |
101 | const struct drm_display_mode *expected; |
102 | size_t len; |
103 | int ret; |
104 | |
105 | if (params->cmdline) { |
106 | cmdline->tv_mode_specified = true; |
107 | cmdline->tv_mode = params->cmdline_mode; |
108 | } |
109 | |
110 | ret = drm_mode_create_tv_properties(dev: priv->drm, supported_tv_modes: params->supported_tv_modes); |
111 | KUNIT_ASSERT_EQ(test, ret, 0); |
112 | |
113 | drm_object_attach_property(obj: &connector->base, |
114 | property: priv->drm->mode_config.tv_mode_property, |
115 | init_val: params->default_mode); |
116 | |
117 | mutex_lock(&priv->drm->mode_config.mutex); |
118 | |
119 | ret = drm_connector_helper_tv_get_modes(connector); |
120 | KUNIT_EXPECT_EQ(test, ret, params->num_expected_modes); |
121 | |
122 | len = 0; |
123 | list_for_each_entry(mode, &connector->probed_modes, head) |
124 | len++; |
125 | KUNIT_EXPECT_EQ(test, len, params->num_expected_modes); |
126 | |
127 | if (params->num_expected_modes >= 1) { |
128 | mode = list_first_entry_or_null(&connector->probed_modes, |
129 | struct drm_display_mode, head); |
130 | KUNIT_ASSERT_NOT_NULL(test, mode); |
131 | |
132 | expected = params->expected_modes[0](priv->drm); |
133 | KUNIT_ASSERT_NOT_NULL(test, expected); |
134 | |
135 | KUNIT_EXPECT_TRUE(test, drm_mode_equal(mode, expected)); |
136 | KUNIT_EXPECT_TRUE(test, mode->type & DRM_MODE_TYPE_PREFERRED); |
137 | } |
138 | |
139 | if (params->num_expected_modes >= 2) { |
140 | mode = list_next_entry(mode, head); |
141 | KUNIT_ASSERT_NOT_NULL(test, mode); |
142 | |
143 | expected = params->expected_modes[1](priv->drm); |
144 | KUNIT_ASSERT_NOT_NULL(test, expected); |
145 | |
146 | KUNIT_EXPECT_TRUE(test, drm_mode_equal(mode, expected)); |
147 | KUNIT_EXPECT_FALSE(test, mode->type & DRM_MODE_TYPE_PREFERRED); |
148 | } |
149 | |
150 | mutex_unlock(lock: &priv->drm->mode_config.mutex); |
151 | } |
152 | |
153 | static const |
154 | struct drm_connector_helper_tv_get_modes_test drm_connector_helper_tv_get_modes_tests[] = { |
155 | { .name = "None" }, |
156 | TV_MODE_TEST("PAL" , |
157 | BIT(DRM_MODE_TV_MODE_PAL), |
158 | DRM_MODE_TV_MODE_PAL, |
159 | drm_mode_analog_pal_576i), |
160 | TV_MODE_TEST("NTSC" , |
161 | BIT(DRM_MODE_TV_MODE_NTSC), |
162 | DRM_MODE_TV_MODE_NTSC, |
163 | drm_mode_analog_ntsc_480i), |
164 | TV_MODE_TEST("Both, NTSC Default" , |
165 | BIT(DRM_MODE_TV_MODE_NTSC) | BIT(DRM_MODE_TV_MODE_PAL), |
166 | DRM_MODE_TV_MODE_NTSC, |
167 | drm_mode_analog_ntsc_480i, drm_mode_analog_pal_576i), |
168 | TV_MODE_TEST("Both, PAL Default" , |
169 | BIT(DRM_MODE_TV_MODE_NTSC) | BIT(DRM_MODE_TV_MODE_PAL), |
170 | DRM_MODE_TV_MODE_PAL, |
171 | drm_mode_analog_pal_576i, drm_mode_analog_ntsc_480i), |
172 | TV_MODE_TEST_CMDLINE("Both, NTSC Default, with PAL on command-line" , |
173 | BIT(DRM_MODE_TV_MODE_NTSC) | BIT(DRM_MODE_TV_MODE_PAL), |
174 | DRM_MODE_TV_MODE_NTSC, |
175 | DRM_MODE_TV_MODE_PAL, |
176 | drm_mode_analog_pal_576i, drm_mode_analog_ntsc_480i), |
177 | TV_MODE_TEST_CMDLINE("Both, PAL Default, with NTSC on command-line" , |
178 | BIT(DRM_MODE_TV_MODE_NTSC) | BIT(DRM_MODE_TV_MODE_PAL), |
179 | DRM_MODE_TV_MODE_PAL, |
180 | DRM_MODE_TV_MODE_NTSC, |
181 | drm_mode_analog_ntsc_480i, drm_mode_analog_pal_576i), |
182 | }; |
183 | |
184 | static void |
185 | drm_connector_helper_tv_get_modes_desc(const struct drm_connector_helper_tv_get_modes_test *t, |
186 | char *desc) |
187 | { |
188 | sprintf(buf: desc, fmt: "%s" , t->name); |
189 | } |
190 | |
191 | KUNIT_ARRAY_PARAM(drm_connector_helper_tv_get_modes, |
192 | drm_connector_helper_tv_get_modes_tests, |
193 | drm_connector_helper_tv_get_modes_desc); |
194 | |
195 | static struct kunit_case drm_test_connector_helper_tv_get_modes_tests[] = { |
196 | KUNIT_CASE_PARAM(drm_test_connector_helper_tv_get_modes_check, |
197 | drm_connector_helper_tv_get_modes_gen_params), |
198 | { } |
199 | }; |
200 | |
201 | static struct kunit_suite drm_test_connector_helper_tv_get_modes_suite = { |
202 | .name = "drm_connector_helper_tv_get_modes" , |
203 | .init = drm_probe_helper_test_init, |
204 | .test_cases = drm_test_connector_helper_tv_get_modes_tests, |
205 | }; |
206 | |
207 | kunit_test_suite(drm_test_connector_helper_tv_get_modes_suite); |
208 | |
209 | MODULE_AUTHOR("Maxime Ripard <maxime@cerno.tech>" ); |
210 | MODULE_LICENSE("GPL" ); |
211 | |