1 // SPDX-License-Identifier: GPL-2.0
2 /*
3 * Copyright (c) 2022 Maxime Ripard <[email protected]>
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_test_pick_cmdline_res_1920_1080_60(struct kunit * test)85 static void drm_test_pick_cmdline_res_1920_1080_60(struct kunit *test)
86 {
87 struct drm_client_modeset_test_priv *priv = test->priv;
88 struct drm_device *drm = priv->drm;
89 struct drm_connector *connector = &priv->connector;
90 struct drm_cmdline_mode *cmdline_mode = &connector->cmdline_mode;
91 struct drm_display_mode *expected_mode, *mode;
92 const char *cmdline = "1920x1080@60";
93 int ret;
94
95 expected_mode = drm_mode_find_dmt(priv->drm, 1920, 1080, 60, false);
96 KUNIT_ASSERT_NOT_NULL(test, expected_mode);
97
98 ret = drm_kunit_add_mode_destroy_action(test, expected_mode);
99 KUNIT_ASSERT_EQ(test, ret, 0);
100
101 KUNIT_ASSERT_TRUE(test,
102 drm_mode_parse_command_line_for_connector(cmdline,
103 connector,
104 cmdline_mode));
105
106 mutex_lock(&drm->mode_config.mutex);
107 ret = drm_helper_probe_single_connector_modes(connector, 1920, 1080);
108 mutex_unlock(&drm->mode_config.mutex);
109 KUNIT_ASSERT_GT(test, ret, 0);
110
111 mode = drm_connector_pick_cmdline_mode(connector);
112 KUNIT_ASSERT_NOT_NULL(test, mode);
113
114 KUNIT_EXPECT_TRUE(test, drm_mode_equal(expected_mode, mode));
115 }
116
117 struct drm_connector_pick_cmdline_mode_test {
118 const char *cmdline;
119 struct drm_display_mode *(*func)(struct drm_device *drm);
120 };
121
122 #define TEST_CMDLINE(_cmdline, _fn) \
123 { \
124 .cmdline = _cmdline, \
125 .func = _fn, \
126 }
127
drm_test_pick_cmdline_named(struct kunit * test)128 static void drm_test_pick_cmdline_named(struct kunit *test)
129 {
130 const struct drm_connector_pick_cmdline_mode_test *params = test->param_value;
131 struct drm_client_modeset_test_priv *priv = test->priv;
132 struct drm_device *drm = priv->drm;
133 struct drm_connector *connector = &priv->connector;
134 struct drm_cmdline_mode *cmdline_mode = &connector->cmdline_mode;
135 const struct drm_display_mode *expected_mode, *mode;
136 const char *cmdline = params->cmdline;
137 int ret;
138
139 KUNIT_ASSERT_TRUE(test,
140 drm_mode_parse_command_line_for_connector(cmdline,
141 connector,
142 cmdline_mode));
143
144 mutex_lock(&drm->mode_config.mutex);
145 ret = drm_helper_probe_single_connector_modes(connector, 1920, 1080);
146 mutex_unlock(&drm->mode_config.mutex);
147 KUNIT_ASSERT_GT(test, ret, 0);
148
149 mode = drm_connector_pick_cmdline_mode(connector);
150 KUNIT_ASSERT_NOT_NULL(test, mode);
151
152 expected_mode = params->func(drm);
153 KUNIT_ASSERT_NOT_NULL(test, expected_mode);
154
155 KUNIT_EXPECT_TRUE(test, drm_mode_equal(expected_mode, mode));
156 }
157
158 static const
159 struct drm_connector_pick_cmdline_mode_test drm_connector_pick_cmdline_mode_tests[] = {
160 TEST_CMDLINE("NTSC", drm_mode_analog_ntsc_480i),
161 TEST_CMDLINE("NTSC-J", drm_mode_analog_ntsc_480i),
162 TEST_CMDLINE("PAL", drm_mode_analog_pal_576i),
163 TEST_CMDLINE("PAL-M", drm_mode_analog_ntsc_480i),
164 };
165
166 static void
drm_connector_pick_cmdline_mode_desc(const struct drm_connector_pick_cmdline_mode_test * t,char * desc)167 drm_connector_pick_cmdline_mode_desc(const struct drm_connector_pick_cmdline_mode_test *t,
168 char *desc)
169 {
170 sprintf(desc, "%s", t->cmdline);
171 }
172
173 KUNIT_ARRAY_PARAM(drm_connector_pick_cmdline_mode,
174 drm_connector_pick_cmdline_mode_tests,
175 drm_connector_pick_cmdline_mode_desc);
176
177 static struct kunit_case drm_test_pick_cmdline_tests[] = {
178 KUNIT_CASE(drm_test_pick_cmdline_res_1920_1080_60),
179 KUNIT_CASE_PARAM(drm_test_pick_cmdline_named,
180 drm_connector_pick_cmdline_mode_gen_params),
181 {}
182 };
183
184 static struct kunit_suite drm_test_pick_cmdline_test_suite = {
185 .name = "drm_test_pick_cmdline",
186 .init = drm_client_modeset_test_init,
187 .test_cases = drm_test_pick_cmdline_tests
188 };
189
190 kunit_test_suite(drm_test_pick_cmdline_test_suite);
191
192 /*
193 * This file is included directly by drm_client_modeset.c so we can't
194 * use any MODULE_* macro here.
195 */
196