Restore list command.

2025-11-04 09:26:05 +00:00 · 2007-09-26 18:32:17 +00:00
parent 2aacbe764c
commit 8d01984182
5 changed files with 106 additions and 90 deletions
--- a/4
+++ b/4
@@ -1,4 +1,4 @@
-# $Id: Makefile,v 1.5 2007-09-26 13:43:14 nicm Exp $
+# $Id: Makefile,v 1.6 2007-09-26 18:32:16 nicm Exp $

 .SUFFIXES: .c .o .y .h
 .PHONY: clean
@@ -18,7 +18,7 @@ META?= \002 # C-b

 SRCS= tmux.c server.c server-msg.c server-fn.c buffer.c buffer-poll.c \
      xmalloc.c xmalloc-debug.c input.c screen.c window.c session.c local.c \
-      log.c client.c client-msg.c client-cmd.c op.c
+      log.c client.c client-msg.c client-cmd.c op.c op-list.c

 YACC= yacc -d

--- a/client.c
+++ b/client.c
@@ -1,4 +1,4 @@
-/* $Id: client.c,v 1.2 2007-09-26 18:12:19 nicm Exp $ */
+/* $Id: client.c,v 1.3 2007-09-26 18:32:16 nicm Exp $ */

 /*
 * Copyright (c) 2007 Nicholas Marriott <nicm@users.sourceforge.net>
@@ -36,7 +36,7 @@ void	client_handle_winch(struct client_ctx *);
 int	client_process_local(struct client_ctx *, const char **);

 int
-client_init(char *path, struct client_ctx *cctx, int ws)
+client_init(char *path, struct client_ctx *cctx, int start_server)
 {
 	struct sockaddr_un	sa;
 	struct stat		sb;
@@ -52,7 +52,7 @@ client_init(char *path, struct client_ctx *cctx, int ws)
 	retries = 0;
 retry:
 	if (stat(path, &sb) != 0) {
-		if (errno != ENOENT) {
+		if (!start_server || errno != ENOENT) {
 			log_warn("%s", path);
 			return (-1);
 		}
@@ -66,7 +66,7 @@ retry:
 		return (-1);
 	}

-	if (ws) {
+	if (start_server) {
 		if (!isatty(STDIN_FILENO)) {
 			log_warnx("stdin is not a tty");
 			return (-1);
@@ -96,7 +96,7 @@ retry:
 	}
 	if (connect(
 	    cctx->srv_fd, (struct sockaddr *) &sa, SUN_LEN(&sa)) == -1) {
-		if (errno == ECONNREFUSED && retries < 5) {
+		if (start_server && errno == ECONNREFUSED && retries < 5) {
 			if (unlink(path) != 0) {
 				log_warn("%s: unlink", path);
 				return (-1);
--- a/op-list.c
+++ b/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");
 		}
--- a/tmux.c
+++ b/tmux.c
@@ -1,4 +1,4 @@
-/* $Id: tmux.c,v 1.10 2007-09-26 14:08:16 nicm Exp $ */
+/* $Id: tmux.c,v 1.11 2007-09-26 18:32:16 nicm Exp $ */

 /*
 * Copyright (c) 2007 Nicholas Marriott <nicm@users.sourceforge.net>
@@ -43,7 +43,7 @@ struct op {
 	int		(*fn)(char *, int, char **);
 };
 struct op op_table[] = {
-//	{ "list", op_list },
+	{ "list", op_list },
 	{ "new", op_new },
 	{ "attach", op_attach }
 };
--- a/tmux.h
+++ b/tmux.h
@@ -1,4 +1,4 @@
-/* $Id: tmux.h,v 1.13 2007-09-26 18:09:23 nicm Exp $ */
+/* $Id: tmux.h,v 1.14 2007-09-26 18:32:17 nicm Exp $ */

 /*
 * Copyright (c) 2007 Nicholas Marriott <nicm@users.sourceforge.net>
@@ -442,6 +442,9 @@ void		 sigreset(void);
 int	 op_new(char *, int, char **);
 int	 op_attach(char *, int, char **);

+/* op-list.c */
+int	 op_list(char *, int, char **);
+
 /* client.c */
 int	 client_init(char *, struct client_ctx *, int);
 int	 client_main(struct client_ctx *);