*/
 void drm_plane_force_disable(struct drm_plane *plane)
 {
+       struct drm_framebuffer *old_fb = plane->fb;
        int ret;
 
-       if (!plane->fb)
+       if (!old_fb)
                return;
 
        ret = plane->funcs->disable_plane(plane);
        if (ret)
                DRM_ERROR("failed to disable plane with busy fb\n");
        /* disconnect the plane from the fb and crtc: */
-       __drm_framebuffer_unreference(plane->fb);
+       __drm_framebuffer_unreference(old_fb);
        plane->fb = NULL;
        plane->crtc = NULL;
 }
        }
 
        drm_modeset_lock_all(dev);
+       old_fb = plane->fb;
        ret = plane->funcs->update_plane(plane, crtc, fb,
                                         plane_req->crtc_x, plane_req->crtc_y,
                                         plane_req->crtc_w, plane_req->crtc_h,
                                         plane_req->src_x, plane_req->src_y,
                                         plane_req->src_w, plane_req->src_h);
        if (!ret) {
-               old_fb = plane->fb;
                plane->crtc = crtc;
                plane->fb = fb;
                fb = NULL;
+       } else {
+               old_fb = NULL;
        }
        drm_modeset_unlock_all(dev);
 
        ret = crtc->funcs->set_config(set);
        if (ret == 0) {
                crtc->primary->crtc = crtc;
-
-               /* crtc->fb must be updated by ->set_config, enforces this. */
-               WARN_ON(fb != crtc->primary->fb);
+               crtc->primary->fb = fb;
        }
 
        list_for_each_entry(tmp, &crtc->dev->mode_config.crtc_list, head) {
 
                .y2 = crtc->mode.vdisplay,
        };
        struct drm_connector **connector_list;
-       struct drm_framebuffer *tmpfb;
        int num_connectors, ret;
 
        if (!crtc->enabled) {
        set.num_connectors = num_connectors;
 
        /*
-        * set_config() adjusts crtc->primary->fb; however the DRM setplane
-        * code that called us expects to handle the framebuffer update and
-        * reference counting; save and restore the current fb before
-        * calling it.
-        *
-        * N.B., we call set_config() directly here rather than using
+        * We call set_config() directly here rather than using
         * drm_mode_set_config_internal.  We're reprogramming the same
         * connectors that were already in use, so we shouldn't need the extra
         * cross-CRTC fb refcounting to accomodate stealing connectors.
         * drm_mode_setplane() already handles the basic refcounting for the
         * framebuffers involved in this operation.
         */
-       tmpfb = plane->fb;
        ret = crtc->funcs->set_config(&set);
-       plane->fb = tmpfb;
 
        kfree(connector_list);
        return ret;