* Copyright (C) 2009 Roberto De Ioris <roberto@unbit.it>
  * Copyright (C) 2009 Jaya Kumar <jayakumar.lkml@gmail.com>
  * Copyright (C) 2009 Bernie Thompson <bernie@plugable.com>
-
  */
 
-#include <linux/dma-buf.h>
-
 #include <drm/drm_atomic_helper.h>
 #include <drm/drm_crtc_helper.h>
 #include <drm/drm_damage_helper.h>
                             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;
+       int i, ret;
        char *cmd;
        struct urb *urb;
        struct drm_rect clip;
        else if ((clip.x2 > fb->width) || (clip.y2 > fb->height))
                return -EINVAL;
 
-       if (import_attach) {
-               ret = dma_buf_begin_cpu_access(import_attach->dmabuf,
-                                              DMA_FROM_DEVICE);
-               if (ret)
-                       return ret;
-       }
+       ret = drm_gem_fb_begin_cpu_access(fb, DMA_FROM_DEVICE);
+       if (ret)
+               return ret;
 
        urb = udl_get_urb(dev);
        if (!urb) {
                ret = -ENOMEM;
-               goto out_dma_buf_end_cpu_access;
+               goto out_drm_gem_fb_end_cpu_access;
        }
        cmd = urb->transfer_buffer;
 
                                       &cmd, byte_offset, dev_byte_offset,
                                       byte_width);
                if (ret)
-                       goto out_dma_buf_end_cpu_access;
+                       goto out_drm_gem_fb_end_cpu_access;
        }
 
        if (cmd > (char *)urb->transfer_buffer) {
 
        ret = 0;
 
-out_dma_buf_end_cpu_access:
-       if (import_attach) {
-               tmp_ret = dma_buf_end_cpu_access(import_attach->dmabuf,
-                                                DMA_FROM_DEVICE);
-               if (tmp_ret && !ret)
-                       ret = tmp_ret; /* only update ret if not set yet */
-       }
-
+out_drm_gem_fb_end_cpu_access:
+       drm_gem_fb_end_cpu_access(fb, DMA_FROM_DEVICE);
        return ret;
 }