struct vm_area_struct *vma)
 {
        struct drm_device *dev = obj->dev;
+       int ret;
 
        /* Check for valid size. */
        if (obj_size < vma->vm_end - vma->vm_start)
                return -EINVAL;
 
-       if (obj->funcs && obj->funcs->vm_ops)
-               vma->vm_ops = obj->funcs->vm_ops;
-       else if (dev->driver->gem_vm_ops)
-               vma->vm_ops = dev->driver->gem_vm_ops;
-       else
-               return -EINVAL;
+       if (obj->funcs && obj->funcs->mmap) {
+               ret = obj->funcs->mmap(obj, vma);
+               if (ret)
+                       return ret;
+               WARN_ON(!(vma->vm_flags & VM_DONTEXPAND));
+       } else {
+               if (obj->funcs && obj->funcs->vm_ops)
+                       vma->vm_ops = obj->funcs->vm_ops;
+               else if (dev->driver->gem_vm_ops)
+                       vma->vm_ops = dev->driver->gem_vm_ops;
+               else
+                       return -EINVAL;
+
+               vma->vm_flags |= VM_IO | VM_PFNMAP | VM_DONTEXPAND | VM_DONTDUMP;
+               vma->vm_page_prot = pgprot_writecombine(vm_get_page_prot(vma->vm_flags));
+               vma->vm_page_prot = pgprot_decrypted(vma->vm_page_prot);
+       }
 
-       vma->vm_flags |= VM_IO | VM_PFNMAP | VM_DONTEXPAND | VM_DONTDUMP;
        vma->vm_private_data = obj;
-       vma->vm_page_prot = pgprot_writecombine(vm_get_page_prot(vma->vm_flags));
-       vma->vm_page_prot = pgprot_decrypted(vma->vm_page_prot);
 
        /* Take a ref for this mapping of the object, so that the fault
         * handler can dereference the mmap offset's pointer to the object.
 
        struct file *fil;
        int ret;
 
+       if (obj->funcs && obj->funcs->mmap) {
+               ret = obj->funcs->mmap(obj, vma);
+               if (ret)
+                       return ret;
+               vma->vm_private_data = obj;
+               drm_gem_object_get(obj);
+               return 0;
+       }
+
        priv = kzalloc(sizeof(*priv), GFP_KERNEL);
        fil = kzalloc(sizeof(*fil), GFP_KERNEL);
        if (!priv || !fil) {
 
         */
        void (*vunmap)(struct drm_gem_object *obj, void *vaddr);
 
+       /**
+        * @mmap:
+        *
+        * Handle mmap() of the gem object, setup vma accordingly.
+        *
+        * This callback is optional.
+        *
+        * The callback is used by by both drm_gem_mmap_obj() and
+        * drm_gem_prime_mmap().  When @mmap is present @vm_ops is not
+        * used, the @mmap callback must set vma->vm_ops instead.
+        *
+        */
+       int (*mmap)(struct drm_gem_object *obj, struct vm_area_struct *vma);
+
        /**
         * @vm_ops:
         *