Plugged some memory leaks.

1 files changed, 10 insertions, 4 deletions

diff --git a/lib/ZNewLocU.c b/lib/ZNewLocU.c
index 59c6fbd..3361aba 100644
--- a/lib/ZNewLocU.c
+++ b/lib/ZNewLocU.c
@@ -104,6 +104,7 @@ Code_t ZNewLocateUser(user, nlocs, auth)
 	    if (retnotice.z_kind == SERVACK &&
 		!strcmp(retnotice.z_opcode,LOCATE_LOCATE)) {
 		    ack = 1;
+		    ZFreeNotice (&retnotice);
 		    continue;
 	    } 	
 
@@ -125,23 +126,28 @@ Code_t ZNewLocateUser(user, nlocs, auth)
 
 	    __locate_list = (ZLocations_t *)malloc((unsigned)__locate_num*
 						   sizeof(ZLocations_t));
-	    if (!__locate_list)
+	    if (!__locate_list) {
+		    ZFreeNotice (&retnotice);
 		    return (ENOMEM);
+	    }
 	
 	    for (ptr=retnotice.z_message, i=0;i<__locate_num;i++) {
 		    __locate_list[i].host = malloc((unsigned)strlen(ptr)+1);
-		    if (!__locate_list[i].host)
+		    if (!__locate_list[i].host) {
+		    nomem:
+			    ZFreeNotice (&retnotice);
 			    return (ENOMEM);
+		    }
 		    (void) strcpy(__locate_list[i].host, ptr);
 		    ptr += strlen(ptr)+1;
 		    __locate_list[i].time = malloc((unsigned)strlen(ptr)+1);
 		    if (!__locate_list[i].time)
-			    return (ENOMEM);
+			    goto nomem;
 		    (void) strcpy(__locate_list[i].time, ptr);
 		    ptr += strlen(ptr)+1;
 		    __locate_list[i].tty = malloc((unsigned)strlen(ptr)+1);
 		    if (!__locate_list[i].tty)
-			    return (ENOMEM);
+			    goto nomem;
 		    (void) strcpy(__locate_list[i].tty, ptr);
 		    ptr += strlen(ptr)+1;
 	    }