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