[libvirt] [PATCH 1/3] util: identity: Harden virIdentitySetCurrent()

Peter Krempa pkrempa at redhat.com
Wed Mar 25 08:35:06 UTC 2015


Don't unref the old identity unless we set the new one correctly and
unref the new one on failure to set it so that we don't leak any
references or use invalid pointers.
---
 src/util/viridentity.c | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/util/viridentity.c b/src/util/viridentity.c
index 6f3baee..9b8ba4a 100644
--- a/src/util/viridentity.c
+++ b/src/util/viridentity.c
@@ -111,15 +111,17 @@ int virIdentitySetCurrent(virIdentityPtr ident)
         return -1;

     old = virThreadLocalGet(&virIdentityCurrent);
-    virObjectUnref(old);

     if (virThreadLocalSet(&virIdentityCurrent,
                           virObjectRef(ident)) < 0) {
         virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                        _("Unable to set thread local identity"));
+        virObjectUnref(ident);
         return -1;
     }

+    virObjectUnref(old);
+
     return 0;
 }

-- 
2.2.2




More information about the libvir-list mailing list