Plug potential memory and socket leak when polling multiple folders or when the upstream sends bogus message sizes. Patch by Sunil Shetye.

svn path=/branches/BRANCH_6-3/; revision=4527

1 files changed, 3 insertions, 1 deletions

diff --git a/driver.c b/driver.c
index 71bd72f4..4dffcb41 100644
--- a/driver.c
+++ b/driver.c
@@ -1373,7 +1373,8 @@ is restored."));
 		    if (count > INT_MAX/sizeof(int))
 		    {
 			report(stderr, GT_("bogus message count!"));
-			return(PS_PROTOCOL);
+			err = PS_PROTOCOL;
+			goto cleanUp;
 		    }
 
 		    /* 
@@ -1389,6 +1390,7 @@ is restored."));
 		    if (proto->getsizes &&
 			!(proto->getpartialsizes && NUM_NONZERO(ctl->fetchsizelimit)))
 		    {
+			xfree(msgsizes);
 			msgsizes = xmalloc(sizeof(int) * count);
 			for (i = 0; i < count; i++)
 			    msgsizes[i] = 0;