1 // SPDX-License-Identifier: GPL-2.0
2 /*
3 * Copyright (c) 2022 Maxime Ripard <mripard@kernel.org>
4 */
5
6 #include <kunit/test.h>
7
8 #include <drm/drm_connector.h>
9 #include <drm/drm_edid.h>
10 #include <drm/drm_drv.h>
11 #include <drm/drm_kunit_helpers.h>
12 #include <drm/drm_modes.h>
13 #include <drm/drm_modeset_helper_vtables.h>
14 #include <drm/drm_probe_helper.h>
15
16 struct drm_client_modeset_test_priv {
17 struct drm_device *drm;
18 struct device *dev;
19 struct drm_connector connector;
20 };
21
drm_client_modeset_connector_get_modes(struct drm_connector * connector)22 static int drm_client_modeset_connector_get_modes(struct drm_connector *connector)
23 {
24 struct drm_display_mode *mode;
25 int count;
26
27 count = drm_add_modes_noedid(connector, 1920, 1200);
28
29 mode = drm_mode_analog_ntsc_480i(connector->dev);
30 if (!mode)
31 return count;
32
33 drm_mode_probed_add(connector, mode);
34 count += 1;
35
36 mode = drm_mode_analog_pal_576i(connector->dev);
37 if (!mode)
38 return count;
39
40 drm_mode_probed_add(connector, mode);
41 count += 1;
42
43 return count;
44 }
45
46 static const struct drm_connector_helper_funcs drm_client_modeset_connector_helper_funcs = {
47 .get_modes = drm_client_modeset_connector_get_modes,
48 };
49
50 static const struct drm_connector_funcs drm_client_modeset_connector_funcs = {
51 };
52
drm_client_modeset_test_init(struct kunit * test)53 static int drm_client_modeset_test_init(struct kunit *test)
54 {
55 struct drm_client_modeset_test_priv *priv;
56 int ret;
57
58 priv = kunit_kzalloc(test, sizeof(*priv), GFP_KERNEL);
59 KUNIT_ASSERT_NOT_NULL(test, priv);
60
61 test->priv = priv;
62
63 priv->dev = drm_kunit_helper_alloc_device(test);
64 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, priv->dev);
65
66 priv->drm = __drm_kunit_helper_alloc_drm_device(test, priv->dev,
67 sizeof(*priv->drm), 0,
68 DRIVER_MODESET);
69 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, priv->drm);
70
71 ret = drmm_connector_init(priv->drm, &priv->connector,
72 &drm_client_modeset_connector_funcs,
73 DRM_MODE_CONNECTOR_Unknown,
74 NULL);
75 KUNIT_ASSERT_EQ(test, ret, 0);
76
77 drm_connector_helper_add(&priv->connector, &drm_client_modeset_connector_helper_funcs);
78
79 priv->connector.interlace_allowed = true;
80 priv->connector.doublescan_allowed = true;
81
82 return 0;
83 }
84
drm_client_modeset_test_exit(struct kunit * test)85 static void drm_client_modeset_test_exit(struct kunit *test)
86 {
87 struct drm_client_modeset_test_priv *priv = test->priv;
88
89 drm_kunit_helper_free_device(test, priv->dev);
90 }
91
drm_test_pick_cmdline_res_1920_1080_60(struct kunit * test)92 static void drm_test_pick_cmdline_res_1920_1080_60(struct kunit *test)
93 {
94 struct drm_client_modeset_test_priv *priv = test->priv;
95 struct drm_device *drm = priv->drm;
96 struct drm_connector *connector = &priv->connector;
97 struct drm_cmdline_mode *cmdline_mode = &connector->cmdline_mode;
98 struct drm_display_mode *expected_mode, *mode;
99 const char *cmdline = "1920x1080@60";
100 int ret;
101
102 expected_mode = drm_mode_find_dmt(priv->drm, 1920, 1080, 60, false);
103 KUNIT_ASSERT_NOT_NULL(test, expected_mode);
104
105 KUNIT_ASSERT_TRUE(test,
106 drm_mode_parse_command_line_for_connector(cmdline,
107 connector,
108 cmdline_mode));
109
110 mutex_lock(&drm->mode_config.mutex);
111 ret = drm_helper_probe_single_connector_modes(connector, 1920, 1080);
112 mutex_unlock(&drm->mode_config.mutex);
113 KUNIT_ASSERT_GT(test, ret, 0);
114
115 mode = drm_connector_pick_cmdline_mode(connector);
116 KUNIT_ASSERT_NOT_NULL(test, mode);
117
118 KUNIT_EXPECT_TRUE(test, drm_mode_equal(expected_mode, mode));
119 }
120
121 struct drm_connector_pick_cmdline_mode_test {
122 const char *cmdline;
123 struct drm_display_mode *(*func)(struct drm_device *drm);
124 };
125
126 #define TEST_CMDLINE(_cmdline, _fn) \
127 { \
128 .cmdline = _cmdline, \
129 .func = _fn, \
130 }
131
drm_test_pick_cmdline_named(struct kunit * test)132 static void drm_test_pick_cmdline_named(struct kunit *test)
133 {
134 const struct drm_connector_pick_cmdline_mode_test *params = test->param_value;
135 struct drm_client_modeset_test_priv *priv = test->priv;
136 struct drm_device *drm = priv->drm;
137 struct drm_connector *connector = &priv->connector;
138 struct drm_cmdline_mode *cmdline_mode = &connector->cmdline_mode;
139 const struct drm_display_mode *expected_mode, *mode;
140 const char *cmdline = params->cmdline;
141 int ret;
142
143 KUNIT_ASSERT_TRUE(test,
144 drm_mode_parse_command_line_for_connector(cmdline,
145 connector,
146 cmdline_mode));
147
148 mutex_lock(&drm->mode_config.mutex);
149 ret = drm_helper_probe_single_connector_modes(connector, 1920, 1080);
150 mutex_unlock(&drm->mode_config.mutex);
151 KUNIT_ASSERT_GT(test, ret, 0);
152
153 mode = drm_connector_pick_cmdline_mode(connector);
154 KUNIT_ASSERT_NOT_NULL(test, mode);
155
156 expected_mode = params->func(drm);
157 KUNIT_ASSERT_NOT_NULL(test, expected_mode);
158
159 KUNIT_EXPECT_TRUE(test, drm_mode_equal(expected_mode, mode));
160 }
161
162 static const
163 struct drm_connector_pick_cmdline_mode_test drm_connector_pick_cmdline_mode_tests[] = {
164 TEST_CMDLINE("NTSC", drm_mode_analog_ntsc_480i),
165 TEST_CMDLINE("NTSC-J", drm_mode_analog_ntsc_480i),
166 TEST_CMDLINE("PAL", drm_mode_analog_pal_576i),
167 TEST_CMDLINE("PAL-M", drm_mode_analog_ntsc_480i),
168 };
169
170 static void
drm_connector_pick_cmdline_mode_desc(const struct drm_connector_pick_cmdline_mode_test * t,char * desc)171 drm_connector_pick_cmdline_mode_desc(const struct drm_connector_pick_cmdline_mode_test *t,
172 char *desc)
173 {
174 sprintf(desc, "%s", t->cmdline);
175 }
176
177 KUNIT_ARRAY_PARAM(drm_connector_pick_cmdline_mode,
178 drm_connector_pick_cmdline_mode_tests,
179 drm_connector_pick_cmdline_mode_desc);
180
181 static struct kunit_case drm_test_pick_cmdline_tests[] = {
182 KUNIT_CASE(drm_test_pick_cmdline_res_1920_1080_60),
183 KUNIT_CASE_PARAM(drm_test_pick_cmdline_named,
184 drm_connector_pick_cmdline_mode_gen_params),
185 {}
186 };
187
188 static struct kunit_suite drm_test_pick_cmdline_test_suite = {
189 .name = "drm_test_pick_cmdline",
190 .init = drm_client_modeset_test_init,
191 .exit = drm_client_modeset_test_exit,
192 .test_cases = drm_test_pick_cmdline_tests
193 };
194
195 kunit_test_suite(drm_test_pick_cmdline_test_suite);
196
197 /*
198 * This file is included directly by drm_client_modeset.c so we can't
199 * use any MODULE_* macro here.
200 */
201