Free/close windows properly.

2 files changed, 20 insertions, 4 deletions

diff --git a/server.c b/server.c
index 5b24858a..0ee924ca 100644
--- a/server.c
+++ b/server.c
@@ -1,4 +1,4 @@
-/* $Id: server.c,v 1.31 2007-10-23 10:25:03 nicm Exp $ */
+/* $Id: server.c,v 1.32 2007-10-24 11:05:59 nicm Exp $ */
 
 /*
  * Copyright (c) 2007 Nicholas Marriott <nicm@users.sourceforge.net>
@@ -123,6 +123,7 @@ server_main(char *srv_path, int srv_fd)
 {
 	struct pollfd	*pfds, *pfd;
 	int		 nfds;
+	u_int		 i;
 
 	siginit();
 
@@ -174,6 +175,18 @@ server_main(char *srv_path, int srv_fd)
 		server_handle_clients(&pfd);
 	}
 
+	for (i = 0; i < ARRAY_LENGTH(&sessions); i++) {
+		if (ARRAY_ITEM(&sessions, i) != NULL)
+			session_destroy(ARRAY_ITEM(&sessions, i));
+	}
+	ARRAY_FREE(&sessions);
+
+	for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
+		if (ARRAY_ITEM(&clients, i) != NULL)
+			server_lost_client(ARRAY_ITEM(&clients, i));
+	}
+	ARRAY_FREE(&clients);
+
 	key_bindings_free();
 
 	close(srv_fd);
diff --git a/session.c b/session.c
index f2ad1cf4..c55542e4 100644
--- a/session.c
+++ b/session.c
@@ -1,4 +1,4 @@
-/* $Id: session.c,v 1.23 2007-10-19 22:16:53 nicm Exp $ */
+/* $Id: session.c,v 1.24 2007-10-24 11:05:59 nicm Exp $ */
 
 /*
  * Copyright (c) 2007 Nicholas Marriott <nicm@users.sourceforge.net>
@@ -124,8 +124,11 @@ session_destroy(struct session *s)
 	while (!ARRAY_EMPTY(&sessions) && ARRAY_LAST(&sessions) == NULL)
 		ARRAY_TRUNC(&sessions, 1);
 	
-	while (!ARRAY_EMPTY(&s->windows))
-		window_remove(&s->windows, ARRAY_FIRST(&s->windows));
+	for (i = 0; i < ARRAY_LENGTH(&s->windows); i++) {
+		if (ARRAY_ITEM(&s->windows, i) != NULL)
+			window_remove(&s->windows, ARRAY_ITEM(&s->windows, i));
+	}
+	ARRAY_FREE(&s->windows);
 
 	xfree(s->name);
 	xfree(s);