1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
|
// SPDX-License-Identifier: GPL-2.0 or MIT
/*
* Copyright (c) 2025 Red Hat.
* Author: Jocelyn Falempe <jfalempe@redhat.com>
*
* KUNIT tests for drm panic
*/
#include <drm/drm_fourcc.h>
#include <drm/drm_panic.h>
#include <kunit/test.h>
#include <linux/units.h>
#include <linux/vmalloc.h>
/* Check the framebuffer color only if the panic colors are the default */
#if (CONFIG_DRM_PANIC_BACKGROUND_COLOR == 0 && \
CONFIG_DRM_PANIC_FOREGROUND_COLOR == 0xffffff)
static void drm_panic_check_color_byte(struct kunit *test, u8 b)
{
KUNIT_EXPECT_TRUE(test, (b == 0 || b == 0xff));
}
#else
static void drm_panic_check_color_byte(struct kunit *test, u8 b) {}
#endif
struct drm_test_mode {
const int width;
const int height;
const u32 format;
void (*draw_screen)(struct drm_scanout_buffer *sb);
const char *fname;
};
/*
* Run all tests for the 3 panic screens: user, kmsg and qr_code
*/
#define DRM_TEST_MODE_LIST(func) \
DRM_PANIC_TEST_MODE(1024, 768, DRM_FORMAT_XRGB8888, func) \
DRM_PANIC_TEST_MODE(300, 200, DRM_FORMAT_XRGB8888, func) \
DRM_PANIC_TEST_MODE(1920, 1080, DRM_FORMAT_XRGB8888, func) \
DRM_PANIC_TEST_MODE(1024, 768, DRM_FORMAT_RGB565, func) \
DRM_PANIC_TEST_MODE(1024, 768, DRM_FORMAT_RGB888, func) \
#define DRM_PANIC_TEST_MODE(w, h, f, name) { \
.width = w, \
.height = h, \
.format = f, \
.draw_screen = draw_panic_screen_##name, \
.fname = #name, \
}, \
static const struct drm_test_mode drm_test_modes_cases[] = {
DRM_TEST_MODE_LIST(user)
DRM_TEST_MODE_LIST(kmsg)
#if IS_ENABLED(CONFIG_DRM_PANIC_SCREEN_QR_CODE)
DRM_TEST_MODE_LIST(qr_code)
#endif
};
#undef DRM_PANIC_TEST_MODE
static int drm_test_panic_init(struct kunit *test)
{
struct drm_scanout_buffer *priv;
priv = kunit_kzalloc(test, sizeof(*priv), GFP_KERNEL);
KUNIT_ASSERT_NOT_NULL(test, priv);
test->priv = priv;
drm_panic_set_description("Kunit testing");
return 0;
}
/*
* Test drawing the panic screen, using a memory mapped framebuffer
* Set the whole buffer to 0xa5, and then check that all pixels have been
* written.
*/
static void drm_test_panic_screen_user_map(struct kunit *test)
{
struct drm_scanout_buffer *sb = test->priv;
const struct drm_test_mode *params = test->param_value;
char *fb;
int fb_size;
int i;
sb->format = drm_format_info(params->format);
fb_size = params->width * params->height * sb->format->cpp[0];
fb = vmalloc(fb_size);
KUNIT_ASSERT_NOT_NULL(test, fb);
memset(fb, 0xa5, fb_size);
iosys_map_set_vaddr(&sb->map[0], fb);
sb->width = params->width;
sb->height = params->height;
sb->pitch[0] = params->width * sb->format->cpp[0];
params->draw_screen(sb);
for (i = 0; i < fb_size; i++)
drm_panic_check_color_byte(test, fb[i]);
vfree(fb);
}
/*
* Test drawing the panic screen, using a list of pages framebuffer
* Set the whole buffer to 0xa5, and then check that all pixels have been
* written.
*/
static void drm_test_panic_screen_user_page(struct kunit *test)
{
struct drm_scanout_buffer *sb = test->priv;
const struct drm_test_mode *params = test->param_value;
int fb_size, p, i, npages;
struct page **pages;
u8 *vaddr;
sb->format = drm_format_info(params->format);
fb_size = params->width * params->height * sb->format->cpp[0];
npages = DIV_ROUND_UP(fb_size, PAGE_SIZE);
pages = kmalloc_objs(struct page *, npages);
KUNIT_ASSERT_NOT_NULL(test, pages);
for (p = 0; p < npages; p++) {
pages[p] = alloc_page(GFP_KERNEL);
if (!pages[p]) {
npages = p - 1;
KUNIT_FAIL(test, "Can't allocate page\n");
goto free_pages;
}
vaddr = kmap_local_page(pages[p]);
memset(vaddr, 0xa5, PAGE_SIZE);
kunmap_local(vaddr);
}
sb->pages = pages;
sb->width = params->width;
sb->height = params->height;
sb->pitch[0] = params->width * sb->format->cpp[0];
params->draw_screen(sb);
for (p = 0; p < npages; p++) {
int bytes_in_page = (p == npages - 1) ? fb_size - p * PAGE_SIZE : PAGE_SIZE;
vaddr = kmap_local_page(pages[p]);
for (i = 0; i < bytes_in_page; i++)
drm_panic_check_color_byte(test, vaddr[i]);
kunmap_local(vaddr);
}
free_pages:
for (p = 0; p < npages; p++)
__free_page(pages[p]);
kfree(pages);
}
static void drm_test_panic_set_pixel(struct drm_scanout_buffer *sb,
unsigned int x,
unsigned int y,
u32 color)
{
struct kunit *test = (struct kunit *)sb->private;
KUNIT_ASSERT_TRUE(test, x < sb->width && y < sb->height);
}
/*
* Test drawing the panic screen, using the set_pixel callback
* Check that all calls to set_pixel() are within the framebuffer
*/
static void drm_test_panic_screen_user_set_pixel(struct kunit *test)
{
struct drm_scanout_buffer *sb = test->priv;
const struct drm_test_mode *params = test->param_value;
sb->format = drm_format_info(params->format);
sb->set_pixel = drm_test_panic_set_pixel;
sb->width = params->width;
sb->height = params->height;
sb->private = test;
params->draw_screen(sb);
}
static void drm_test_panic_desc(const struct drm_test_mode *t, char *desc)
{
sprintf(desc, "Panic screen %s, mode: %d x %d \t%p4cc",
t->fname, t->width, t->height, &t->format);
}
KUNIT_ARRAY_PARAM(drm_test_panic_screen_user_map, drm_test_modes_cases, drm_test_panic_desc);
KUNIT_ARRAY_PARAM(drm_test_panic_screen_user_page, drm_test_modes_cases, drm_test_panic_desc);
KUNIT_ARRAY_PARAM(drm_test_panic_screen_user_set_pixel, drm_test_modes_cases, drm_test_panic_desc);
static struct kunit_case drm_panic_screen_user_test[] = {
KUNIT_CASE_PARAM(drm_test_panic_screen_user_map,
drm_test_panic_screen_user_map_gen_params),
KUNIT_CASE_PARAM(drm_test_panic_screen_user_page,
drm_test_panic_screen_user_page_gen_params),
KUNIT_CASE_PARAM(drm_test_panic_screen_user_set_pixel,
drm_test_panic_screen_user_set_pixel_gen_params),
{ }
};
static struct kunit_suite drm_panic_suite = {
.name = "drm_panic",
.init = drm_test_panic_init,
.test_cases = drm_panic_screen_user_test,
};
kunit_test_suite(drm_panic_suite);
|
