if (!i915_gem_object_has_struct_page(obj))
                return -ENODEV;
 
-       ret = i915_gem_object_lock_interruptible(obj, NULL);
-       if (ret)
-               return ret;
+       assert_object_held(obj);
 
        ret = i915_gem_object_wait(obj,
                                   I915_WAIT_INTERRUPTIBLE,
                                   MAX_SCHEDULE_TIMEOUT);
        if (ret)
-               goto err_unlock;
+               return ret;
 
        ret = i915_gem_object_pin_pages(obj);
        if (ret)
-               goto err_unlock;
+               return ret;
 
        if (obj->cache_coherent & I915_BO_CACHE_COHERENT_FOR_READ ||
            !static_cpu_has(X86_FEATURE_CLFLUSH)) {
 
 err_unpin:
        i915_gem_object_unpin_pages(obj);
-err_unlock:
-       i915_gem_object_unlock(obj);
        return ret;
 }
 
        if (!i915_gem_object_has_struct_page(obj))
                return -ENODEV;
 
-       ret = i915_gem_object_lock_interruptible(obj, NULL);
-       if (ret)
-               return ret;
+       assert_object_held(obj);
 
        ret = i915_gem_object_wait(obj,
                                   I915_WAIT_INTERRUPTIBLE |
                                   I915_WAIT_ALL,
                                   MAX_SCHEDULE_TIMEOUT);
        if (ret)
-               goto err_unlock;
+               return ret;
 
        ret = i915_gem_object_pin_pages(obj);
        if (ret)
-               goto err_unlock;
+               return ret;
 
        if (obj->cache_coherent & I915_BO_CACHE_COHERENT_FOR_WRITE ||
            !static_cpu_has(X86_FEATURE_CLFLUSH)) {
 
 err_unpin:
        i915_gem_object_unpin_pages(obj);
-err_unlock:
-       i915_gem_object_unlock(obj);
        return ret;
 }
 
 
        vaddr = unmask_page(cache->vaddr);
        if (cache->vaddr & KMAP) {
+               struct drm_i915_gem_object *obj =
+                       (struct drm_i915_gem_object *)cache->node.mm;
                if (cache->vaddr & CLFLUSH_AFTER)
                        mb();
 
                kunmap_atomic(vaddr);
-               i915_gem_object_finish_access((struct drm_i915_gem_object *)cache->node.mm);
+               i915_gem_object_finish_access(obj);
+               i915_gem_object_unlock(obj);
        } else {
                struct i915_ggtt *ggtt = cache_to_ggtt(cache);
 
                unsigned int flushes;
                int err;
 
-               err = i915_gem_object_prepare_write(obj, &flushes);
+               err = i915_gem_object_lock_interruptible(obj, NULL);
                if (err)
                        return ERR_PTR(err);
 
+               err = i915_gem_object_prepare_write(obj, &flushes);
+               if (err) {
+                       i915_gem_object_unlock(obj);
+                       return ERR_PTR(err);
+               }
+
                BUILD_BUG_ON(KMAP & CLFLUSH_FLAGS);
                BUILD_BUG_ON((KMAP | CLFLUSH_FLAGS) & PAGE_MASK);
 
 
 i915_gem_object_finish_access(struct drm_i915_gem_object *obj)
 {
        i915_gem_object_unpin_pages(obj);
-       i915_gem_object_unlock(obj);
 }
 
 static inline struct intel_engine_cs *
 
        unsigned long n;
        int err;
 
+       i915_gem_object_lock(obj, NULL);
        err = i915_gem_object_prepare_read(obj, &needs_flush);
        if (err)
-               return err;
+               goto err_unlock;
 
        for (n = 0; n < obj->base.size >> PAGE_SHIFT; ++n) {
                u32 *ptr = kmap_atomic(i915_gem_object_get_page(obj, n));
        }
 
        i915_gem_object_finish_access(obj);
+err_unlock:
+       i915_gem_object_unlock(obj);
 
        return err;
 }
 
        u32 *cpu;
        int err;
 
+       i915_gem_object_lock(ctx->obj, NULL);
        err = i915_gem_object_prepare_write(ctx->obj, &needs_clflush);
        if (err)
-               return err;
+               goto out;
 
        page = i915_gem_object_get_page(ctx->obj, offset >> PAGE_SHIFT);
        map = kmap_atomic(page);
        kunmap_atomic(map);
        i915_gem_object_finish_access(ctx->obj);
 
-       return 0;
+out:
+       i915_gem_object_unlock(ctx->obj);
+       return err;
 }
 
 static int cpu_get(struct context *ctx, unsigned long offset, u32 *v)
        u32 *cpu;
        int err;
 
+       i915_gem_object_lock(ctx->obj, NULL);
        err = i915_gem_object_prepare_read(ctx->obj, &needs_clflush);
        if (err)
-               return err;
+               goto out;
 
        page = i915_gem_object_get_page(ctx->obj, offset >> PAGE_SHIFT);
        map = kmap_atomic(page);
        kunmap_atomic(map);
        i915_gem_object_finish_access(ctx->obj);
 
-       return 0;
+out:
+       i915_gem_object_unlock(ctx->obj);
+       return err;
 }
 
 static int gtt_set(struct context *ctx, unsigned long offset, u32 v)
 
        unsigned int n, m, need_flush;
        int err;
 
+       i915_gem_object_lock(obj, NULL);
        err = i915_gem_object_prepare_write(obj, &need_flush);
        if (err)
-               return err;
+               goto out;
 
        for (n = 0; n < real_page_count(obj); n++) {
                u32 *map;
        i915_gem_object_finish_access(obj);
        obj->read_domains = I915_GEM_DOMAIN_GTT | I915_GEM_DOMAIN_CPU;
        obj->write_domain = 0;
-       return 0;
+out:
+       i915_gem_object_unlock(obj);
+       return err;
 }
 
 static noinline int cpu_check(struct drm_i915_gem_object *obj,
        unsigned int n, m, needs_flush;
        int err;
 
+       i915_gem_object_lock(obj, NULL);
        err = i915_gem_object_prepare_read(obj, &needs_flush);
        if (err)
-               return err;
+               goto out_unlock;
 
        for (n = 0; n < real_page_count(obj); n++) {
                u32 *map;
        }
 
        i915_gem_object_finish_access(obj);
+out_unlock:
+       i915_gem_object_unlock(obj);
        return err;
 }
 
 
        if (ret)
                goto err_unmap;
 
+       i915_gem_object_unlock(bb->obj);
        INIT_LIST_HEAD(&bb->list);
        list_add(&bb->list, &s->workload->shadow_bb);
 
 
        u64 remain;
        int ret;
 
-       ret = i915_gem_object_prepare_read(obj, &needs_clflush);
+       ret = i915_gem_object_lock_interruptible(obj, NULL);
        if (ret)
                return ret;
 
+       ret = i915_gem_object_prepare_read(obj, &needs_clflush);
+       if (ret) {
+               i915_gem_object_unlock(obj);
+               return ret;
+       }
+
        fence = i915_gem_object_lock_fence(obj);
        i915_gem_object_finish_access(obj);
+       i915_gem_object_unlock(obj);
+
        if (!fence)
                return -ENOMEM;
 
        u64 remain;
        int ret;
 
-       ret = i915_gem_object_prepare_write(obj, &needs_clflush);
+       ret = i915_gem_object_lock_interruptible(obj, NULL);
        if (ret)
                return ret;
 
+       ret = i915_gem_object_prepare_write(obj, &needs_clflush);
+       if (ret) {
+               i915_gem_object_unlock(obj);
+               return ret;
+       }
+
        fence = i915_gem_object_lock_fence(obj);
        i915_gem_object_finish_access(obj);
+       i915_gem_object_unlock(obj);
+
        if (!fence)
                return -ENOMEM;