Cleanup part II: split up client stuff and ops. More to come.

1 files changed, 126 insertions, 0 deletions

diff --git a/op-list.c b/op-list.c
new file mode 100644
index 00000000..e2f78efe
--- /dev/null
+++ b/op-list.c
@@ -0,0 +1,126 @@
+/* $Id: op-list.c,v 1.1 2007-09-26 13:43:15 nicm Exp $ */
+
+/*
+ * Copyright (c) 2007 Nicholas Marriott <nicm@users.sourceforge.net>
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
+ * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
+ * 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;
+
+	/* 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);
+	}
+
+	/* Main loop. */
+	for (;;) {
+		/* Set up pollfd. */
+		pfd.fd = server_fd;
+		pfd.events = POLLIN;
+		if (BUFFER_USED(server_out) > 0)
+			pfd.events |= POLLOUT;
+
+		/* Do the poll. */
+		if (poll(&pfd, 1, INFTIM) == -1) {
+			if (errno == EAGAIN || errno == EINTR)
+				continue;
+			err(1, "poll");
+		}
+
+		/* Read/write from sockets. */
+		if (buffer_poll(&pfd, server_in, server_out) != 0)
+			errx(1, "lost server"); 
+
+		/* 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);
+		
+		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);
+				*strchr(tim, '\n') = '\0';
+				printf("%s: %u windows (created %s)\n",
+				    se.name, se.windows, tim);
+			}
+			exit(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); 
+				} else {
+					printf("%u: %s (%s)\n",
+					    we.idx, we.name, we.tty);
+				}
+			}
+			exit(0);
+		default:
+			fatalx("unexpected message");
+		}
+	}
+}