Make sure XQueryTree results are freed

Fixes ArcticaProject/nx-libs#585

1 files changed, 17 insertions, 15 deletions

diff --git a/nx-X11/programs/Xserver/hw/nxagent/Events.c b/nx-X11/programs/Xserver/hw/nxagent/Events.c
index e360d21f6..4e08cd2c2 100644
--- a/nx-X11/programs/Xserver/hw/nxagent/Events.c
+++ b/nx-X11/programs/Xserver/hw/nxagent/Events.c
@@ -426,13 +426,12 @@ void nxagentRemoteWindowsTree(Window window, int level)
   int i, j;
   unsigned long rootWin, parentWin;
   unsigned int numChildren;
-  unsigned long *childList;
+  unsigned long *childList = NULL;
 
   if (!XQueryTree(nxagentDisplay, window, &rootWin, &parentWin, &childList,
                       &numChildren))
   {
     fprintf(stderr, "nxagentRemoteWindowsTree - XQueryTree failed.\n");
-
     return;
   }
 
@@ -3185,7 +3184,7 @@ int nxagentCheckWindowConfiguration(XConfigureEvent* X)
     #endif
   }
 
-  if (result && nchildren_return)
+  if (children_return)
   {
     XFree(children_return);
   }
@@ -3540,6 +3539,12 @@ int nxagentHandleReparentNotify(XEvent* X)
         result = XQueryTree(nxagentDisplay, w, &root_return,
                                 &parent_return, &children_return, &nchildren_return);
 
+        if (children_return)
+        {
+          XFree(children_return);
+          children_return = NULL;
+        }
+
         if (!result)
         {
           #ifdef WARNING
@@ -3548,11 +3553,6 @@ int nxagentHandleReparentNotify(XEvent* X)
 
           break;
         }
-
-        if (result && children_return)
-        {
-          XFree(children_return);
-        }
       }
 
       if (w && !nxagentWindowPtr(w))
@@ -3581,9 +3581,10 @@ int nxagentHandleReparentNotify(XEvent* X)
           #endif
         }
 
-        if (result && nchildren_return)
+        if (children_return)
         {
           XFree(children_return);
+          children_return = NULL;
         }
       }
       else
@@ -3647,17 +3648,18 @@ int nxagentHandleReparentNotify(XEvent* X)
       {
         result = XQueryTree(nxagentDisplay, w, &rootReturn, &parentReturn,
                                 &childrenReturn, &nchildrenReturn);
-    
-        if (parentReturn == rootReturn || parentReturn == 0 || result == 0)
+
+        if (childrenReturn)
         {
-          break;
+          XFree(childrenReturn);
+          childrenReturn = NULL;
         }
 
-        if (result == 1 && childrenReturn != NULL)
+        if (parentReturn == rootReturn || parentReturn == 0 || result == 0)
         {
-          XFree(childrenReturn);
+          break;
         }
-    
+
         w = parentReturn;
       }
       while (True);