#include <drm/drm_crtc_helper.h>
 #include <drm/drm_damage_helper.h>
 #include <drm/drm_fourcc.h>
+#include <drm/drm_gem_atomic_helper.h>
 #include <drm/drm_gem_framebuffer_helper.h>
 #include <drm/drm_gem_shmem_helper.h>
 #include <drm/drm_modeset_helper_vtables.h>
        return 0;
 }
 
-static int udl_handle_damage(struct drm_framebuffer *fb, int x, int y,
-                            int width, int height)
+static int udl_handle_damage(struct drm_framebuffer *fb, const struct dma_buf_map *map,
+                            int x, int y, int width, int height)
 {
        struct drm_device *dev = fb->dev;
        struct dma_buf_attachment *import_attach = fb->obj[0]->import_attach;
+       void *vaddr = map->vaddr; /* TODO: Use mapping abstraction properly */
        int i, ret, tmp_ret;
        char *cmd;
        struct urb *urb;
        struct drm_rect clip;
        int log_bpp;
-       struct dma_buf_map map;
-       void *vaddr;
 
        ret = udl_log_cpp(fb->format->cpp[0]);
        if (ret < 0)
                        return ret;
        }
 
-       ret = drm_gem_shmem_vmap(fb->obj[0], &map);
-       if (ret) {
-               DRM_ERROR("failed to vmap fb\n");
-               goto out_dma_buf_end_cpu_access;
-       }
-       vaddr = map.vaddr; /* TODO: Use mapping abstraction properly */
-
        urb = udl_get_urb(dev);
        if (!urb) {
                ret = -ENOMEM;
-               goto out_drm_gem_shmem_vunmap;
+               goto out_dma_buf_end_cpu_access;
        }
        cmd = urb->transfer_buffer;
 
                                       &cmd, byte_offset, dev_byte_offset,
                                       byte_width);
                if (ret)
-                       goto out_drm_gem_shmem_vunmap;
+                       goto out_dma_buf_end_cpu_access;
        }
 
        if (cmd > (char *)urb->transfer_buffer) {
 
        ret = 0;
 
-out_drm_gem_shmem_vunmap:
-       drm_gem_shmem_vunmap(fb->obj[0], &map);
 out_dma_buf_end_cpu_access:
        if (import_attach) {
                tmp_ret = dma_buf_end_cpu_access(import_attach->dmabuf,
        struct drm_framebuffer *fb = plane_state->fb;
        struct udl_device *udl = to_udl(dev);
        struct drm_display_mode *mode = &crtc_state->mode;
+       struct drm_shadow_plane_state *shadow_plane_state = to_drm_shadow_plane_state(plane_state);
        char *buf;
        char *wrptr;
        int color_depth = UDL_COLOR_DEPTH_16BPP;
 
        udl->mode_buf_len = wrptr - buf;
 
-       udl_handle_damage(fb, 0, 0, fb->width, fb->height);
+       udl_handle_damage(fb, &shadow_plane_state->map[0], 0, 0, fb->width, fb->height);
 
        if (!crtc_state->mode_changed)
                return;
                               struct drm_plane_state *old_plane_state)
 {
        struct drm_plane_state *state = pipe->plane.state;
+       struct drm_shadow_plane_state *shadow_plane_state = to_drm_shadow_plane_state(state);
        struct drm_framebuffer *fb = state->fb;
        struct drm_rect rect;
 
                return;
 
        if (drm_atomic_helper_damage_merged(old_plane_state, state, &rect))
-               udl_handle_damage(fb, rect.x1, rect.y1, rect.x2 - rect.x1,
-                                 rect.y2 - rect.y1);
+               udl_handle_damage(fb, &shadow_plane_state->map[0], rect.x1, rect.y1,
+                                 rect.x2 - rect.x1, rect.y2 - rect.y1);
 }
 
-static const
-struct drm_simple_display_pipe_funcs udl_simple_display_pipe_funcs = {
+static const struct drm_simple_display_pipe_funcs udl_simple_display_pipe_funcs = {
        .mode_valid = udl_simple_display_pipe_mode_valid,
        .enable = udl_simple_display_pipe_enable,
        .disable = udl_simple_display_pipe_disable,
        .update = udl_simple_display_pipe_update,
-       .prepare_fb = drm_gem_fb_simple_display_pipe_prepare_fb,
+       DRM_GEM_SIMPLE_DISPLAY_PIPE_SHADOW_PLANE_FUNCS,
 };
 
 /*