If any of the function calls in _msm_gem_kernel_new fail we need
to make sure to dereference the GEM object with the appropriate
function for the current locking state.

Signed-off-by: Jordan Crouse <jcro...@codeaurora.org>
---
 drivers/gpu/drm/msm/msm_gem.c | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/drivers/gpu/drm/msm/msm_gem.c b/drivers/gpu/drm/msm/msm_gem.c
index 00c795ced02c..4646e9e45fc2 100644
--- a/drivers/gpu/drm/msm/msm_gem.c
+++ b/drivers/gpu/drm/msm/msm_gem.c
@@ -1041,23 +1041,29 @@ static void *_msm_gem_kernel_new(struct drm_device 
*dev, uint32_t size,
 
        if (iova) {
                ret = msm_gem_get_iova(obj, aspace, iova);
-               if (ret) {
-                       drm_gem_object_put(obj);
-                       return ERR_PTR(ret);
-               }
+               if (ret)
+                       goto err;
        }
 
        vaddr = msm_gem_get_vaddr(obj);
        if (IS_ERR(vaddr)) {
                msm_gem_put_iova(obj, aspace);
-               drm_gem_object_put(obj);
-               return ERR_CAST(vaddr);
+               ret = PTR_ERR(vaddr);
+               goto err;
        }
 
        if (bo)
                *bo = obj;
 
        return vaddr;
+err:
+       if (locked)
+               drm_gem_object_put(obj);
+       else
+               drm_gem_object_put_unlocked(obj);
+
+       return ERR_PTR(ret);
+
 }
 
 void *msm_gem_kernel_new(struct drm_device *dev, uint32_t size,
-- 
2.18.0

_______________________________________________
dri-devel mailing list
dri-devel@lists.freedesktop.org
https://lists.freedesktop.org/mailman/listinfo/dri-devel

Reply via email to