First usable version of the client. No command completion and similar nifty

features yet, but it works.

3 files changed, 243 insertions, 17 deletions

diff --git a/client/client.c b/client/client.c
index 260e043..50335e4 100644
--- a/client/client.c
+++ b/client/client.c
@@ -9,7 +9,13 @@
 #include <stdio.h>
 #include <string.h>
 #include <stdlib.h>
+#include <fcntl.h>
 #include <unistd.h>
+#include <errno.h>
+#include <sys/socket.h>
+#include <sys/un.h>
+#include <sys/time.h>
+#include <sys/types.h>
 #include <readline/readline.h>
 #include <readline/history.h>
 
@@ -17,12 +23,25 @@
 #include "lib/resource.h"
 #include "client/client.h"
 
-static char *opt_list = "";
+static char *opt_list = "s:v";
+static int verbose;
+
+static char *server_path = PATH_CONTROL_SOCKET_DIR "/bird.ctl";
+static int server_fd;
+static int server_reply;
+static byte server_read_buf[4096];
+static byte *server_read_pos = server_read_buf;
+
+static int input_initialized;
+static int input_hidden;
+static int input_hidden_end;
+
+/*** Parsing of arguments ***/
 
 static void
 usage(void)
 {
-  fprintf(stderr, "Usage: birdc\n");
+  fprintf(stderr, "Usage: birdc [-s <control-socket>] [-v]\n");
   exit(1);
 }
 
@@ -34,6 +53,12 @@ parse_args(int argc, char **argv)
   while ((c = getopt(argc, argv, opt_list)) >= 0)
     switch (c)
       {
+      case 's':
+	server_path = optarg;
+	break;
+      case 'v':
+	verbose++;
+	break;
       default:
 	usage();
       }
@@ -41,19 +66,211 @@ parse_args(int argc, char **argv)
     usage();
 }
 
-static char *
-get_command(void)
-{
-  static char *cmd_buffer;
+/*** Input ***/
+
+static void server_send(char *);
+static void io_loop(int);
 
-  if (cmd_buffer)
-    free(cmd_buffer);
-  cmd_buffer = readline("bird> ");
+static void
+got_line(char *cmd_buffer)
+{
   if (!cmd_buffer)
-    exit(0);
+    {
+      cleanup();
+      exit(0);
+    }
   if (cmd_buffer[0])
-    add_history(cmd_buffer);
-  return cmd_buffer;
+    {
+      add_history(cmd_buffer);
+      /* FIXME: Builtin commands: exit, ... */
+      server_send(cmd_buffer);
+      input_hidden = -1;
+      io_loop(0);
+      input_hidden = 0;
+    }
+  free(cmd_buffer);
+}
+
+static void
+input_init(void)
+{
+  rl_readline_name = "birdc";
+  rl_callback_handler_install("bird> ", got_line);
+  input_initialized = 1;
+  if (fcntl(0, F_SETFL, O_NONBLOCK) < 0)
+    die("fcntl: %m");
+}
+
+static void
+input_hide(void)
+{
+  if (input_hidden)
+    return;
+  if (rl_line_buffer)
+    {
+      input_hidden_end = rl_end;
+      rl_end = 0;
+      rl_expand_prompt("");
+      rl_redisplay();
+      input_hidden = 1;
+    }
+}
+
+static void
+input_reveal(void)
+{
+  if (input_hidden <= 0)
+    return;
+  rl_end = input_hidden_end;
+  rl_expand_prompt("bird> ");
+  rl_forced_update_display();
+  input_hidden = 0;
+}
+
+void
+cleanup(void)
+{
+  if (input_initialized)
+    {
+      input_initialized = 0;
+      input_hide();
+      rl_callback_handler_remove();
+    }
+}
+
+/*** Communication with server ***/
+
+static void
+server_connect(void)
+{
+  struct sockaddr_un sa;
+
+  server_fd = socket(AF_UNIX, SOCK_STREAM, 0);
+  if (server_fd < 0)
+    die("Cannot create socket: %m");
+  bzero(&sa, sizeof(sa));
+  sa.sun_family = AF_UNIX;
+  strcpy(sa.sun_path, server_path);
+  if (connect(server_fd, (struct sockaddr *) &sa, sizeof(struct sockaddr)) < 0)
+    die("Unable to connect to server control socket (%s): %m", server_path);
+  if (fcntl(server_fd, F_SETFL, O_NONBLOCK) < 0)
+    die("fcntl: %m");
+}
+
+static void
+server_got_reply(char *x)
+{
+  int code;
+
+  input_hide();
+  if (*x == '+')			/* Async reply */
+    printf(">>> %s\n", x+1);
+  else if (x[0] == ' ')			/* Continuation */
+    printf("%s%s\n", x+1, verbose ? "     " : "");
+  else if (strlen(x) > 4 &&
+	   sscanf(x, "%d", &code) == 1 && code >= 0 && code < 10000 &&
+	   (x[4] == ' ' || x[4] == '-'))
+    {
+      if (code)
+	printf("%s\n", verbose ? x : x+5);
+      if (x[4] == ' ')
+	server_reply = code;
+    }
+  else
+    printf("??? <%s>\n", x);
+}
+
+static void
+server_read(void)
+{
+  int c;
+  byte *start, *p;
+
+  c = read(server_fd, server_read_pos, server_read_buf + sizeof(server_read_buf) - server_read_pos);
+  if (!c)
+    die("Connection closed by server.");
+  if (c < 0)
+    die("Server read error: %m");
+  start = server_read_buf;
+  p = server_read_pos;
+  server_read_pos += c;
+  while (p < server_read_pos)
+    if (*p++ == '\n')
+      {
+	p[-1] = 0;
+	server_got_reply(start);
+	start = p;
+      }
+  if (start != server_read_buf)
+    {
+      int l = server_read_pos - start;
+      memmove(server_read_buf, start, l);
+      server_read_pos = server_read_buf + l;
+    }
+  else if (server_read_pos == server_read_buf + sizeof(server_read_buf))
+    {
+      strcpy(server_read_buf, "?<too-long>");
+      server_read_pos = server_read_buf + 11;
+    }
+}
+
+static fd_set select_fds;
+
+static void
+io_loop(int mode)
+{
+  server_reply = -1;
+  while (mode || server_reply < 0)
+    {
+      FD_SET(server_fd, &select_fds);
+      if (mode)
+	FD_SET(0, &select_fds);
+      select(server_fd+1, &select_fds, NULL, NULL, NULL);
+      if (FD_ISSET(server_fd, &select_fds))
+	{
+	  server_read();
+	  if (mode)
+	    input_reveal();
+	}
+      if (FD_ISSET(0, &select_fds))
+	rl_callback_read_char();
+    }
+  input_reveal();
+}
+
+static void
+server_send(char *cmd)
+{
+  int l = strlen(cmd);
+  byte *z = alloca(l + 1);
+
+  memcpy(z, cmd, l);
+  z[l++] = '\n';
+  while (l)
+    {
+      int cnt = write(server_fd, z, l);
+      if (cnt < 0)
+	{
+	  if (errno == -EAGAIN)
+	    {
+	      fd_set set;
+	      FD_ZERO(&set);
+	      do
+		{
+		  FD_SET(server_fd, &set);
+		  select(server_fd+1, NULL, &set, NULL, NULL);
+		}
+	      while (!FD_ISSET(server_fd, &set));
+	    }
+	  else
+	    die("Server write error: %m");
+	}
+      else
+	{
+	  l -= cnt;
+	  z += cnt;
+	}
+    }
 }
 
 int
@@ -65,10 +282,11 @@ main(int argc, char **argv)
 #endif
 
   parse_args(argc, argv);
+  server_connect();
+  io_loop(0);
 
-  for(;;)
-    {
-      char *c = get_command();
-      puts(c);
-    }
+  input_init();
+
+  io_loop(1);
+  return 0;
 }
diff --git a/client/client.h b/client/client.h
index 2e1e050..f0edeeb 100644
--- a/client/client.h
+++ b/client/client.h
@@ -5,3 +5,7 @@
  *
  *	Can be freely distributed and used under the terms of the GNU GPL.
  */
+
+/* client.c */
+
+void cleanup(void);
diff --git a/client/util.c b/client/util.c
index 65a1fb2..7699faf 100644
--- a/client/util.c
+++ b/client/util.c
@@ -16,12 +16,15 @@
 
 /* Client versions of logging functions */
 
+/* FIXME: Use bsprintf, so that %m works */
+
 void
 bug(char *msg, ...)
 {
   va_list args;
 
   va_start(args, msg);
+  cleanup();
   fputs("Internal error: ", stderr);
   vfprintf(stderr, msg, args);
   fputc('\n', stderr);
@@ -34,6 +37,7 @@ die(char *msg, ...)
   va_list args;
 
   va_start(args, msg);
+  cleanup();
   vfprintf(stderr, msg, args);
   fputc('\n', stderr);
   exit(1);