X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~mdw/git/disorder/blobdiff_plain/96af1d7e02ac1734b562d07a03c4d7a53abafb3c..90f1379c03f34549731e7f82120e0534a185336f:/lib/client.c

diff --git a/lib/client.c b/lib/client.c
index 09e5426..fb1b1cc 100644
--- a/lib/client.c
+++ b/lib/client.c
@@ -1,6 +1,6 @@
 /*
  * This file is part of DisOrder.
- * Copyright (C) 2004-2010 Richard Kettlewell
+ * Copyright (C) 2004-13 Richard Kettlewell
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -25,13 +25,22 @@
 #include "common.h"
 
 #include <sys/types.h>
-#include <sys/socket.h>
-#include <netinet/in.h>
-#include <sys/un.h>
-#include <unistd.h>
+#if HAVE_SYS_SOCKET_H
+# include <sys/socket.h>
+#endif
+#if HAVE_NETINET_IN_H
+# include <netinet/in.h>
+#endif
+#if HAVE_SYS_UN_H
+# include <sys/un.h>
+#endif
+#if HAVE_UNISTD_H
+# include <unistd.h>
+#endif
 #include <errno.h>
-#include <netdb.h>
-#include <pcre.h>
+#if HAVE_NETDB_H
+# include <netdb.h>
+#endif
 
 #include "log.h"
 #include "mem.h"
@@ -51,13 +60,14 @@
 #include "client-common.h"
 #include "rights.h"
 #include "kvp.h"
+#include "socketio.h"
 
 /** @brief Client handle contents */
 struct disorder_client {
   /** @brief Stream to read from */
-  FILE *fpin;
+  struct source *input;
   /** @brief Stream to write to */
-  FILE *fpout;
+  struct sink *output;
   /** @brief Peer description */
   char *ident;
   /** @brief Username */
@@ -68,6 +78,12 @@ struct disorder_client {
   const char *last;
   /** @brief Address family */
   int family;
+  /** @brief True if open */
+  int open;
+  /** @brief Socket I/O context */
+  struct socketio sio;
+  /** @brief Whether to try to open a privileged connection */
+  int trypriv;
 };
 
 /** @brief Create a new client
@@ -83,12 +99,37 @@ disorder_client *disorder_new(int verbose) {
 
   c->verbose = verbose;
   c->family = -1;
+  c->trypriv = 1;
   return c;
 }
 
-/** @brief Return the address family used by this client */
-int disorder_client_af(disorder_client *c) {
-  return c->family;
+/** @brief Don't try to make a privileged connection
+ * @param c Client
+ *
+ * You must call this before any of the connection functions (e.g.,
+ * disorder_connect(), disorder_connect_user()), if at all.
+ */
+void disorder_force_unpriv(disorder_client *c) {
+  assert(!c->open);
+  c->trypriv = 0;
+}
+
+/** @brief Determine the local socket address of this client */
+int disorder_client_sockname(disorder_client *c,
+			     struct sockaddr *sa, socklen_t *len_inout) {
+  int rc;
+  if((rc = getsockname(c->sio.sd, sa, len_inout)))
+    disorder_error(errno, "failed to read client socket name");
+  return rc;
+}
+
+/** @brief Determine the remote peer address for this client */
+int disorder_client_peername(disorder_client *c,
+			     struct sockaddr *sa, socklen_t *len_inout) {
+  int rc;
+  if((rc = getpeername(c->sio.sd, sa, len_inout)))
+    disorder_error(errno, "failed to read client socket name");
+  return rc;
 }
 
 /** @brief Read a response line
@@ -98,9 +139,11 @@ int disorder_client_af(disorder_client *c) {
  */
 static int response(disorder_client *c, char **rp) {
   char *r;
+  char errbuf[1024];
 
-  if(inputline(c->ident, c->fpin, &r, '\n')) {
-    byte_xasprintf((char **)&c->last, "input error: %s", strerror(errno));
+  if(inputlines(c->ident, c->input, &r, '\n')) {
+    byte_xasprintf((char **)&c->last, "input error: %s",
+                   format_error(c->input->eclass, source_err(c->input), errbuf, sizeof errbuf));
     return -1;
   }
   D(("response: %s", r));
@@ -202,8 +245,9 @@ static int disorder_simple_v(disorder_client *c,
   char **body = NULL;
   int nbody = 0;
   int has_body = 0;
+  char errbuf[1024];
 
-  if(!c->fpout) {
+  if(!c->open) {
     c->last = "not connected";
     disorder_error(0, "not connected to server");
     return -1;
@@ -219,24 +263,25 @@ static int disorder_simple_v(disorder_client *c,
       } else if(arg == disorder__list) {
 	char **list = va_arg(ap, char **);
 	int nlist = va_arg(ap, int);
+        int n;
 	if(nlist < 0) {
 	  for(nlist = 0; list[nlist]; ++nlist)
 	    ;
 	}
-	for(int n = 0; n < nlist; ++n) {
+	for(n = 0; n < nlist; ++n) {
 	  dynstr_append(&d, ' ');
 	  dynstr_append_string(&d, quoteutf8(arg));
 	}
       } else if(arg == disorder__integer) {
 	long n = va_arg(ap, long);
 	char buffer[16];
-	snprintf(buffer, sizeof buffer, "%ld", n);
+	byte_snprintf(buffer, sizeof buffer, "%ld", n);
 	dynstr_append(&d, ' ');
 	dynstr_append_string(&d, buffer);
       } else if(arg == disorder__time) {
 	time_t n = va_arg(ap, time_t);
 	char buffer[16];
-	snprintf(buffer, sizeof buffer, "%lld", (long long)n);
+	byte_snprintf(buffer, sizeof buffer, "%lld", (long long)n);
 	dynstr_append(&d, ' ');
 	dynstr_append_string(&d, buffer);
       } else {
@@ -247,32 +292,34 @@ static int disorder_simple_v(disorder_client *c,
     dynstr_append(&d, '\n');
     dynstr_terminate(&d);
     D(("command: %s", d.vec));
-    if(fputs(d.vec, c->fpout) < 0)
+    if(sink_write(c->output, d.vec, d.nvec) < 0)
       goto write_error;
     xfree(d.vec);
     if(has_body) {
+      int n;
       if(nbody < 0)
         for(nbody = 0; body[nbody]; ++nbody)
           ;
-      for(int n = 0; n < nbody; ++n) {
+      for(n = 0; n < nbody; ++n) {
         if(body[n][0] == '.')
-          if(fputc('.', c->fpout) < 0)
+          if(sink_writec(c->output, '.') < 0)
             goto write_error;
-        if(fputs(body[n], c->fpout) < 0)
+        if(sink_writes(c->output, body[n]) < 0)
           goto write_error;
-        if(fputc('\n', c->fpout) < 0)
+        if(sink_writec(c->output, '\n') < 0)
           goto write_error;
       }
-      if(fputs(".\n", c->fpout) < 0)
+      if(sink_writes(c->output, ".\n") < 0)
         goto write_error;
     }
-    if(fflush(c->fpout))
+    if(sink_flush(c->output))
       goto write_error;
   }
   return check_response(c, rp);
 write_error:
-  byte_xasprintf((char **)&c->last, "write error: %s", strerror(errno));
-  disorder_error(errno, "error writing to %s", c->ident);
+  byte_xasprintf((char **)&c->last, "write error: %s", 
+                 format_error(c->output->eclass, sink_err(c->output), errbuf, sizeof errbuf));
+  disorder_error(0, "%s: %s", c->ident, c->last);
   return -1;
 }
 
@@ -398,7 +445,8 @@ int disorder_connect_generic(struct config *conf,
                              const char *username,
                              const char *password,
                              const char *cookie) {
-  int fd = -1, fd2 = -1, nrvec = 0, rc;
+  SOCKET sd = INVALID_SOCKET;
+  int nrvec = 0, rc;
   unsigned char *nonce = NULL;
   size_t nl;
   char *res = NULL;
@@ -406,38 +454,32 @@ int disorder_connect_generic(struct config *conf,
   const char *protocol, *algorithm, *challenge;
   struct sockaddr *sa = NULL;
   socklen_t salen;
+  char errbuf[1024];
 
-  if((salen = find_server(conf, &sa, &c->ident)) == (socklen_t)-1)
+  if((salen = disorder_find_server(conf,
+				   (c->trypriv ? 0 : DISORDER_FS_NOTPRIV),
+				   &sa, &c->ident)) == (socklen_t)-1)
     return -1;
-  c->fpin = c->fpout = 0;
-  if((fd = socket(sa->sa_family, SOCK_STREAM, 0)) < 0) {
-    byte_xasprintf((char **)&c->last, "socket: %s", strerror(errno));
-    disorder_error(errno, "error calling socket");
+  c->input = 0;
+  c->output = 0;
+  if((sd = socket(sa->sa_family, SOCK_STREAM, 0)) < 0) {
+    byte_xasprintf((char **)&c->last, "socket: %s",
+                   format_error(ec_socket, socket_error(), errbuf, sizeof errbuf));
+    disorder_error(0, "%s", c->last);
     return -1;
   }
   c->family = sa->sa_family;
-  if(connect(fd, sa, salen) < 0) {
-    byte_xasprintf((char **)&c->last, "connect: %s", strerror(errno));
-    disorder_error(errno, "error calling connect");
-    goto error;
-  }
-  if((fd2 = dup(fd)) < 0) {
-    byte_xasprintf((char **)&c->last, "dup: %s", strerror(errno));
-    disorder_error(errno, "error calling dup");
-    goto error;
-  }
-  if(!(c->fpin = fdopen(fd, "rb"))) {
-    byte_xasprintf((char **)&c->last, "fdopen: %s", strerror(errno));
-    disorder_error(errno, "error calling fdopen");
+  if(connect(sd, sa, salen) < 0) {
+    byte_xasprintf((char **)&c->last, "connect: %s",
+                   format_error(ec_socket, socket_error(), errbuf, sizeof errbuf));
+    disorder_error(0, "%s", c->last);
     goto error;
   }
-  fd = -1;
-  if(!(c->fpout = fdopen(fd2, "wb"))) {
-    byte_xasprintf((char **)&c->last, "fdopen: %s", strerror(errno));
-    disorder_error(errno, "error calling fdopen");
-    goto error;
-  }
-  fd2 = -1;
+  socketio_init(&c->sio, sd);
+  c->open = 1;
+  sd = INVALID_SOCKET;
+  c->output = sink_socketio(&c->sio);
+  c->input = source_socketio(&c->sio);
   if((rc = disorder_simple(c, &r, 0, (const char *)0)))
     goto error_rc;
   if(!(rvec = split(r, &nrvec, SPLIT_QUOTES, 0, 0)))
@@ -483,16 +525,12 @@ int disorder_connect_generic(struct config *conf,
 error:
   rc = -1;
 error_rc:
-  if(c->fpin) {
-    fclose(c->fpin);
-    c->fpin = 0;
-  }
-  if(c->fpout) {
-    fclose(c->fpout);
-    c->fpout = 0;
-  }
-  if(fd2 != -1) close(fd2);
-  if(fd != -1) close(fd);
+  xfree(c->output);
+  c->output = NULL;
+  xfree(c->input);
+  c->input = NULL;
+  if(c->open) { socketio_close(&c->sio); c->open = 0; }
+  if(sd != INVALID_SOCKET) closesocket(sd);
   return rc;
 }
 
@@ -556,7 +594,7 @@ int disorder_connect(disorder_client *c) {
  *
  * If @p cookie is NULL or does not work then we attempt to log in as
  * guest instead (so when the cookie expires only an extra round trip
- * is needed rathre than a complete new login).
+ * is needed rather than a complete new login).
  */
 int disorder_connect_cookie(disorder_client *c,
 			    const char *cookie) {
@@ -577,22 +615,12 @@ int disorder_connect_cookie(disorder_client *c,
 int disorder_close(disorder_client *c) {
   int ret = 0;
 
-  if(c->fpin) {
-    if(fclose(c->fpin) < 0) {
-      byte_xasprintf((char **)&c->last, "fclose: %s", strerror(errno));
-      disorder_error(errno, "error calling fclose");
-      ret = -1;
-    }
-    c->fpin = 0;
-  }
-  if(c->fpout) {
-    if(fclose(c->fpout) < 0) {
-      byte_xasprintf((char **)&c->last, "fclose: %s", strerror(errno));
-      disorder_error(errno, "error calling fclose");
-      ret = -1;
-    }
-    c->fpout = 0;
-  }
+  if(c->open)
+    socketio_close(&c->sio);
+  xfree(c->output);
+  c->output = NULL;
+  xfree(c->input);
+  c->input = NULL;
   xfree(c->ident);
   c->ident = 0;
   xfree(c->user);
@@ -634,8 +662,9 @@ static int readqueue(disorder_client *c,
 		     struct queue_entry **qp) {
   struct queue_entry *qh, **qt = &qh, *q;
   char *l;
+  char errbuf[1024];
 
-  while(inputline(c->ident, c->fpin, &l, '\n') >= 0) {
+  while(inputlines(c->ident, c->input, &l, '\n') >= 0) {
     if(!strcmp(l, ".")) {
       *qt = 0;
       *qp = qh;
@@ -649,13 +678,13 @@ static int readqueue(disorder_client *c,
     }
     xfree(l);
   }
-  if(ferror(c->fpin)) {
-    byte_xasprintf((char **)&c->last, "input error: %s", strerror(errno));
-    disorder_error(errno, "error reading %s", c->ident);
+  if(source_err(c->input)) {
+    byte_xasprintf((char **)&c->last, "input error: %s",
+                   format_error(c->input->eclass, source_err(c->input), errbuf, sizeof errbuf));
   } else {
     c->last = "input error: unexpected EOF";
-    disorder_error(0, "error reading %s: unexpected EOF", c->ident);
   }
+  disorder_error(0, "%s: %s", c->ident, c->last);
   return -1;
 }
 
@@ -670,9 +699,10 @@ static int readqueue(disorder_client *c,
 static int readlist(disorder_client *c, char ***vecp, int *nvecp) {
   char *l;
   struct vector v;
+  char errbuf[1024];
 
   vector_init(&v);
-  while(inputline(c->ident, c->fpin, &l, '\n') >= 0) {
+  while(inputlines(c->ident, c->input, &l, '\n') >= 0) {
     if(!strcmp(l, ".")) {
       vector_terminate(&v);
       if(nvecp)
@@ -684,13 +714,13 @@ static int readlist(disorder_client *c, char ***vecp, int *nvecp) {
     vector_append(&v, xstrdup(l + (*l == '.')));
     xfree(l);
   }
-  if(ferror(c->fpin)) {
-    byte_xasprintf((char **)&c->last, "input error: %s", strerror(errno));
-    disorder_error(errno, "error reading %s", c->ident);
+  if(source_err(c->input)) {
+    byte_xasprintf((char **)&c->last, "input error: %s",
+                   format_error(c->input->eclass, source_err(c->input), errbuf, sizeof errbuf));
   } else {
     c->last = "input error: unexpxected EOF";
-    disorder_error(0, "error reading %s: unexpected EOF", c->ident);
   }
+  disorder_error(0, "%s: %s", c->ident, c->last);
   return -1;
 }
 
@@ -745,6 +775,10 @@ static int pairlist(disorder_client *c, struct kvp **kp, const char *cmd, ...) {
   return 0;
 }
 
+#if _WIN32
+# define boolean bodge_boolean
+#endif
+
 /** @brief Parse a boolean response
  * @param cmd Command for use in error messsage
  * @param value Result from server
@@ -770,16 +804,21 @@ static int boolean(const char *cmd, const char *value,
 int disorder_log(disorder_client *c, struct sink *s) {
   char *l;
   int rc;
+  char errbuf[1024];
     
   if((rc = disorder_simple(c, 0, "log", (char *)0)))
     return rc;
-  while(inputline(c->ident, c->fpin, &l, '\n') >= 0 && strcmp(l, "."))
+  while(inputlines(c->ident, c->input, &l, '\n') >= 0 && strcmp(l, "."))
     if(sink_printf(s, "%s\n", l) < 0) return -1;
-  if(ferror(c->fpin) || feof(c->fpin)) {
+  if(source_err(c->input)) {
     byte_xasprintf((char **)&c->last, "input error: %s",
-		   ferror(c->fpin) ? strerror(errno) : "unexpxected EOF");
+		   format_error(c->input->eclass, source_err(c->input), errbuf, sizeof errbuf));
+    return -1;
+  } else if(source_eof(c->input)) {
+    byte_xasprintf((char **)&c->last, "input error: unexpected EOF");
     return -1;
   }
+
   return 0;
 }