1
2
3#include "vkms_drv.h"
4#include <linux/crc32.h>
5#include <drm/drm_atomic.h>
6#include <drm/drm_atomic_helper.h>
7#include <drm/drm_gem_framebuffer_helper.h>
8
9
10
11
12
13
14
15
16
17
18static uint32_t compute_crc(void *vaddr_out, struct vkms_crc_data *crc_out)
19{
20 int i, j, src_offset;
21 int x_src = crc_out->src.x1 >> 16;
22 int y_src = crc_out->src.y1 >> 16;
23 int h_src = drm_rect_height(&crc_out->src) >> 16;
24 int w_src = drm_rect_width(&crc_out->src) >> 16;
25 u32 crc = 0;
26
27 for (i = y_src; i < y_src + h_src; ++i) {
28 for (j = x_src; j < x_src + w_src; ++j) {
29 src_offset = crc_out->offset
30 + (i * crc_out->pitch)
31 + (j * crc_out->cpp);
32
33 memset(vaddr_out + src_offset + 24, 0, 8);
34 crc = crc32_le(crc, vaddr_out + src_offset,
35 sizeof(u32));
36 }
37 }
38
39 return crc;
40}
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57static void blend(void *vaddr_dst, void *vaddr_src,
58 struct vkms_crc_data *crc_dst,
59 struct vkms_crc_data *crc_src)
60{
61 int i, j, j_dst, i_dst;
62 int offset_src, offset_dst;
63
64 int x_src = crc_src->src.x1 >> 16;
65 int y_src = crc_src->src.y1 >> 16;
66
67 int x_dst = crc_src->dst.x1;
68 int y_dst = crc_src->dst.y1;
69 int h_dst = drm_rect_height(&crc_src->dst);
70 int w_dst = drm_rect_width(&crc_src->dst);
71
72 int y_limit = y_src + h_dst;
73 int x_limit = x_src + w_dst;
74
75 for (i = y_src, i_dst = y_dst; i < y_limit; ++i) {
76 for (j = x_src, j_dst = x_dst; j < x_limit; ++j) {
77 offset_dst = crc_dst->offset
78 + (i_dst * crc_dst->pitch)
79 + (j_dst++ * crc_dst->cpp);
80 offset_src = crc_src->offset
81 + (i * crc_src->pitch)
82 + (j * crc_src->cpp);
83
84 memcpy(vaddr_dst + offset_dst,
85 vaddr_src + offset_src, sizeof(u32));
86 }
87 i_dst++;
88 }
89}
90
91static void compose_cursor(struct vkms_crc_data *cursor_crc,
92 struct vkms_crc_data *primary_crc, void *vaddr_out)
93{
94 struct drm_gem_object *cursor_obj;
95 struct vkms_gem_object *cursor_vkms_obj;
96
97 cursor_obj = drm_gem_fb_get_obj(&cursor_crc->fb, 0);
98 cursor_vkms_obj = drm_gem_to_vkms_gem(cursor_obj);
99
100 mutex_lock(&cursor_vkms_obj->pages_lock);
101 if (!cursor_vkms_obj->vaddr) {
102 DRM_WARN("cursor plane vaddr is NULL");
103 goto out;
104 }
105
106 blend(vaddr_out, cursor_vkms_obj->vaddr, primary_crc, cursor_crc);
107
108out:
109 mutex_unlock(&cursor_vkms_obj->pages_lock);
110}
111
112static uint32_t _vkms_get_crc(struct vkms_crc_data *primary_crc,
113 struct vkms_crc_data *cursor_crc)
114{
115 struct drm_framebuffer *fb = &primary_crc->fb;
116 struct drm_gem_object *gem_obj = drm_gem_fb_get_obj(fb, 0);
117 struct vkms_gem_object *vkms_obj = drm_gem_to_vkms_gem(gem_obj);
118 void *vaddr_out = kzalloc(vkms_obj->gem.size, GFP_KERNEL);
119 u32 crc = 0;
120
121 if (!vaddr_out) {
122 DRM_ERROR("Failed to allocate memory for output frame.");
123 return 0;
124 }
125
126 mutex_lock(&vkms_obj->pages_lock);
127 if (WARN_ON(!vkms_obj->vaddr)) {
128 mutex_unlock(&vkms_obj->pages_lock);
129 kfree(vaddr_out);
130 return crc;
131 }
132
133 memcpy(vaddr_out, vkms_obj->vaddr, vkms_obj->gem.size);
134 mutex_unlock(&vkms_obj->pages_lock);
135
136 if (cursor_crc)
137 compose_cursor(cursor_crc, primary_crc, vaddr_out);
138
139 crc = compute_crc(vaddr_out, primary_crc);
140
141 kfree(vaddr_out);
142
143 return crc;
144}
145
146
147
148
149
150
151
152
153
154
155void vkms_crc_work_handle(struct work_struct *work)
156{
157 struct vkms_crtc_state *crtc_state = container_of(work,
158 struct vkms_crtc_state,
159 crc_work);
160 struct drm_crtc *crtc = crtc_state->base.crtc;
161 struct vkms_output *out = drm_crtc_to_vkms_output(crtc);
162 struct vkms_device *vdev = container_of(out, struct vkms_device,
163 output);
164 struct vkms_crc_data *primary_crc = NULL;
165 struct vkms_crc_data *cursor_crc = NULL;
166 struct drm_plane *plane;
167 u32 crc32 = 0;
168 u64 frame_start, frame_end;
169 unsigned long flags;
170
171 spin_lock_irqsave(&out->state_lock, flags);
172 frame_start = crtc_state->frame_start;
173 frame_end = crtc_state->frame_end;
174 spin_unlock_irqrestore(&out->state_lock, flags);
175
176
177 if (!frame_start || frame_start == frame_end)
178 goto out;
179
180 drm_for_each_plane(plane, &vdev->drm) {
181 struct vkms_plane_state *vplane_state;
182 struct vkms_crc_data *crc_data;
183
184 vplane_state = to_vkms_plane_state(plane->state);
185 crc_data = vplane_state->crc_data;
186
187 if (drm_framebuffer_read_refcount(&crc_data->fb) == 0)
188 continue;
189
190 if (plane->type == DRM_PLANE_TYPE_PRIMARY)
191 primary_crc = crc_data;
192 else
193 cursor_crc = crc_data;
194 }
195
196 if (primary_crc)
197 crc32 = _vkms_get_crc(primary_crc, cursor_crc);
198
199 frame_end = drm_crtc_accurate_vblank_count(crtc);
200
201
202
203
204 while (frame_start <= frame_end)
205 drm_crtc_add_crc_entry(crtc, true, frame_start++, &crc32);
206
207out:
208
209 spin_lock_irqsave(&out->state_lock, flags);
210 crtc_state->frame_end = frame_end;
211 crtc_state->frame_start = 0;
212 spin_unlock_irqrestore(&out->state_lock, flags);
213}
214
215static const char * const pipe_crc_sources[] = {"auto"};
216
217const char *const *vkms_get_crc_sources(struct drm_crtc *crtc,
218 size_t *count)
219{
220 *count = ARRAY_SIZE(pipe_crc_sources);
221 return pipe_crc_sources;
222}
223
224static int vkms_crc_parse_source(const char *src_name, bool *enabled)
225{
226 int ret = 0;
227
228 if (!src_name) {
229 *enabled = false;
230 } else if (strcmp(src_name, "auto") == 0) {
231 *enabled = true;
232 } else {
233 *enabled = false;
234 ret = -EINVAL;
235 }
236
237 return ret;
238}
239
240int vkms_verify_crc_source(struct drm_crtc *crtc, const char *src_name,
241 size_t *values_cnt)
242{
243 bool enabled;
244
245 if (vkms_crc_parse_source(src_name, &enabled) < 0) {
246 DRM_DEBUG_DRIVER("unknown source %s\n", src_name);
247 return -EINVAL;
248 }
249
250 *values_cnt = 1;
251
252 return 0;
253}
254
255int vkms_set_crc_source(struct drm_crtc *crtc, const char *src_name)
256{
257 struct vkms_output *out = drm_crtc_to_vkms_output(crtc);
258 bool enabled = false;
259 unsigned long flags;
260 int ret = 0;
261
262 ret = vkms_crc_parse_source(src_name, &enabled);
263
264
265 flush_workqueue(out->crc_workq);
266
267 spin_lock_irqsave(&out->lock, flags);
268 out->crc_enabled = enabled;
269 spin_unlock_irqrestore(&out->lock, flags);
270
271 return ret;
272}
273