Restore list command.

op-list.c

@@ -1,4 +1,4 @@
-/* $Id: op-list.c,v 1.1 2007-09-26 13:43:15 nicm Exp $ */
+/* $Id: op-list.c,v 1.2 2007-09-26 18:32:16 nicm Exp $ */
 
 /*
  * Copyright (c) 2007 Nicholas Marriott <nicm@users.sourceforge.net>
@@ -16,109 +16,122 @@
  * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  */
 
-/* List sessions or windows. */
-void
-op_list(char *name)
-{
-	struct sessions_data	 sd;
-	struct windows_data	 wd;
-	struct pollfd	 	 pfd;
-	struct hdr		 hdr;
+#include <sys/types.h>
 
-	/* Send query data. */
-	if (*name == '\0') {
-		hdr.type = MSG_SESSIONS;
-		hdr.size = sizeof sd;
-		buffer_write(server_out, &hdr, sizeof hdr);
-		buffer_write(server_out, &sd, hdr.size);
-	} else {
-		hdr.type = MSG_WINDOWS;
-		hdr.size = sizeof wd;
-		buffer_write(server_out, &hdr, sizeof hdr);
-		strlcpy(wd.name, name, sizeof wd.name);
-		buffer_write(server_out, &wd, hdr.size);
+#include <errno.h>
+#include <getopt.h>
+#include <string.h>
+
+#include "tmux.h"
+
+int
+op_list(char *path, int argc, char **argv)
+{
+	struct client_ctx	cctx;
+	char			name[MAXNAMELEN], *tim;
+	int			opt;
+	struct sessions_data	sdata;
+	struct sessions_entry	sent;
+	struct windows_data	wdata;
+	struct windows_entry	went;
+	struct pollfd		pfd;
+	struct hdr		hdr;
+
+	*name = '\0';
+	optind = 1;
+	while ((opt = getopt(argc, argv, "n:?")) != EOF) {
+		switch (opt) {
+		case 'n':
+			if (strlcpy(name, optarg, sizeof name) >= sizeof name) {
+				log_warnx("%s: session name too long", optarg);
+				return (1);
+			}
+			break;
+		case '?':
+		default:
+			return (usage("list [-n session]"));
+		}
+	}	
+	argc -= optind;
+	argv += optind;			
+	if (argc != 0)
+		return (usage("list [-n session]"));
+
+	if (client_init(path, &cctx, 0) != 0)
+		return (1);
+ 
+	if (*name == '\0')
+		client_write_server(&cctx, MSG_SESSIONS, &sdata, sizeof sdata);
+	else {
+		strlcpy(wdata.name, name, sizeof wdata.name);
+		client_write_server(&cctx, MSG_WINDOWS, &wdata, sizeof wdata);
 	}
 
-	/* Main loop. */
 	for (;;) {
-		/* Set up pollfd. */
-		pfd.fd = server_fd;
+		pfd.fd = cctx.srv_fd;
 		pfd.events = POLLIN;
-		if (BUFFER_USED(server_out) > 0)
+		if (BUFFER_USED(cctx.srv_out) > 0)
 			pfd.events |= POLLOUT;
 
-		/* Do the poll. */
 		if (poll(&pfd, 1, INFTIM) == -1) {
 			if (errno == EAGAIN || errno == EINTR)
 				continue;
-			err(1, "poll");
+			log_warn("poll");
+			return (-1);
 		}
 
-		/* Read/write from sockets. */
-		if (buffer_poll(&pfd, server_in, server_out) != 0)
-			errx(1, "lost server"); 
+		if (buffer_poll(&pfd, cctx.srv_in, cctx.srv_out) != 0) {
+			log_warnx("lost server");
+			return (-1);
+		}
 
-		/* Process data. */
-		process_list(name);
-	}
-}
-
-void
-op_list_process(const char *name)
-{
-	struct sessions_data	 sd;
-	struct sessions_entry	 se;
-	struct windows_data	 wd;
-	struct windows_entry	 we;
-	struct hdr		 hdr;
-	char		        *tim;
-	
-	for (;;) {
-		if (BUFFER_USED(server_in) < sizeof hdr)
-			break;
-		memcpy(&hdr, BUFFER_OUT(server_in), sizeof hdr);
-		if (BUFFER_USED(server_in) < (sizeof hdr) + hdr.size)
-			break;
-		buffer_remove(server_in, sizeof hdr);
+		if (BUFFER_USED(cctx.srv_in) < sizeof hdr)
+			continue;
+		memcpy(&hdr, BUFFER_OUT(cctx.srv_in), sizeof hdr);
+		if (BUFFER_USED(cctx.srv_in) < (sizeof hdr) + hdr.size)
+			continue;
+		buffer_remove(cctx.srv_in, sizeof hdr);
 		
 		switch (hdr.type) {
 		case MSG_SESSIONS:
-			if (hdr.size < sizeof sd)
-				errx(1, "bad MSG_SESSIONS size");
-			buffer_read(server_in, &sd, sizeof sd);
-			hdr.size -= sizeof sd; 
-			if (sd.sessions == 0 && hdr.size == 0)
-				exit(0);
-			if (hdr.size < sd.sessions * sizeof se)
-				errx(1, "bad MSG_SESSIONS size");
-			while (sd.sessions-- > 0) {
-				buffer_read(server_in, &se, sizeof se);
-				tim = ctime(&se.tim);
+			if (hdr.size < sizeof sdata)
+				fatalx("bad MSG_SESSIONS size");
+			buffer_read(cctx.srv_in, &sdata, sizeof sdata);
+			hdr.size -= sizeof sdata; 
+			if (sdata.sessions == 0 && hdr.size == 0)
+				return (0);
+			if (hdr.size < sdata.sessions * sizeof sent)
+				fatalx("bad MSG_SESSIONS size");
+			while (sdata.sessions-- > 0) {
+				buffer_read(cctx.srv_in, &sent, sizeof sent);
+				tim = ctime(&sent.tim);
 				*strchr(tim, '\n') = '\0';
 				printf("%s: %u windows (created %s)\n",
-				    se.name, se.windows, tim);
+				    sent.name, sent.windows, tim);
 			}
-			exit(0);
+			return (0);
 		case MSG_WINDOWS:
-			if (hdr.size < sizeof wd)
-				errx(1, "bad MSG_WINDOWS size");
-			buffer_read(server_in, &wd, sizeof wd);
-			hdr.size -= sizeof wd; 
-			if (wd.windows == 0 && hdr.size == 0)
-				errx(1, "session not found: %s", name);
-			if (hdr.size < wd.windows * sizeof we)
-				errx(1, "bad MSG_WINDOWS size");
-			while (wd.windows-- > 0) {
-				buffer_read(server_in, &we, sizeof we);
-				if (*we.title != '\0') {
-					printf("%u: %s \"%s\" (%s)\n",
-					    we.idx, we.name, we.title, we.tty); 
+			if (hdr.size < sizeof wdata)
+				fatalx("bad MSG_WINDOWS size");
+			buffer_read(cctx.srv_in, &wdata, sizeof wdata);
+			hdr.size -= sizeof wdata; 
+			if (wdata.windows == 0 && hdr.size == 0) {
+				log_warnx("session not found: %s", name);
+				return (1);
+			}
+			if (hdr.size < wdata.windows * sizeof went)
+				fatalx("bad MSG_WINDOWS size");
+			while (wdata.windows-- > 0) {
+				buffer_read(cctx.srv_in, &went, sizeof went);
+				if (*went.title != '\0') {
+					printf("%u: %s \"%s\" (%s)\n", went.idx,
+					    went.name, went.title, went.tty);
 				} else {
 					printf("%u: %s (%s)\n",
-					    we.idx, we.name, we.tty);
+					    went.idx, went.name, went.tty);
 				}
 			}
-			exit(0);
+			return (0);
 		default:
 			fatalx("unexpected message");
 		}