chiark / gitweb /
fixes
[inn-innduct.git] / backends / innduct.c
index 091b198..bc954f4 100644 (file)
@@ -190,6 +190,7 @@ perl -ne 'print if m/-8\<-/..m/-\>8-/; print "\f" if m/-\^L-/' backends/innduct.
 #include <glob.h>
 #include <time.h>
 #include <math.h>
+#include <ctype.h>
 
 #include <oop.h>
 #include <oop-read.h>
@@ -218,7 +219,7 @@ perl -ne 'print if m/-8\<-/..m/-\>8-/; print "\f" if m/-\^L-/' backends/innduct.
      int count;                                        \
    } T##List
 
-#define NODE(n) (assert((void*)&(n)->list_node == &(n)), &(n)->list_node)
+#define NODE(n) (assert((void*)&(n)->list_node == (n)), &(n)->list_node)
 
 #define LIST_CHECKCANHAVENODE(l,n) \
   ((void)((n) == ((l).u.for_type))) /* just for the type check */
@@ -244,7 +245,7 @@ perl -ne 'print if m/-8\<-/..m/-\>8-/; print "\f" if m/-\^L-/' backends/innduct.
 #define LIST_REMHEAD(l) LIST_REMSOMEHOW((l),list_remhead)
 #define LIST_REMTAIL(l) LIST_REMSOMEHOW((l),list_remtail)
 
-#define LIST_INIT(l) (list_new(&(l).u.li))
+#define LIST_INIT(l) ((l).count=0, list_new(&(l).u.li))
 #define LIST_HEAD(l) ((typeof((l).u.for_type))(list_head((struct list*)&(l))))
 #define LIST_NEXT(n) ((typeof(n))list_succ(NODE((n))))
 #define LIST_BACK(n) ((typeof(n))list_pred(NODE((n))))
@@ -631,8 +632,9 @@ static int xwaitpid(pid_t *pid, const char *what) {
   int r= kill(*pid, SIGKILL);
   if (r) sysdie("cannot kill %s child", what);
 
-  pid_t got= waitpid(*pid, &status, WNOHANG);
+  pid_t got= waitpid(*pid, &status, 0);
   if (got==-1) sysdie("cannot reap %s child", what);
+  if (got==0) die("cannot reap %s child", what);
 
   *pid= 0;
 
@@ -1122,10 +1124,7 @@ static int connecting_fdpass_sock;
 
 static void connect_attempt_discard(void) {
   if (connecting_child) {
-    int r= kill(connecting_child, SIGTERM);
-    if (r) syswarn("failed to kill connecting child");
     int status= xwaitpid(&connecting_child, "connect");
-
     if (!(WIFEXITED(status) ||
          (WIFSIGNALED(status) && WTERMSIG(status) == SIGKILL)))
       report_child_status("connect", status);
@@ -1137,9 +1136,15 @@ static void connect_attempt_discard(void) {
 }
 
 #define PREP_DECL_MSG_CMSG(msg)                        \
+  char msgbyte= 0;                             \
+  struct iovec msgiov;                         \
+  msgiov.iov_base= &msgbyte;                   \
+  msgiov.iov_len= 1;                           \
   struct msghdr msg;                           \
   memset(&msg,0,sizeof(msg));                  \
-  char msg##cbuf[CMSG_SPACE(sizeof(fd))];      \
+  char msg##cbuf[CMSG_SPACE(sizeof(int))];     \
+  msg.msg_iov= &msgiov;                                \
+  msg.msg_iovlen= 1;                           \
   msg.msg_control= msg##cbuf;                  \
   msg.msg_controllen= sizeof(msg##cbuf);
 
@@ -1148,42 +1153,45 @@ static void *connchild_event(oop_source *lp, int fd, oop_event e, void *u) {
 
   assert(fd == connecting_fdpass_sock);
 
+  PREP_DECL_MSG_CMSG(msg);
+  
+  ssize_t rs= recvmsg(fd, &msg, 0);
+  if (rs<0) {
+    if (isewouldblock(errno)) return OOP_CONTINUE;
+    syswarn("failed to read socket from connecting child");
+    goto x;
+  }
+
   conn= xmalloc(sizeof(*conn));
   memset(conn,0,sizeof(*conn));
+  LIST_INIT(conn->waiting);
+  LIST_INIT(conn->priority);
+  LIST_INIT(conn->sent);
 
-  PREP_DECL_MSG_CMSG(msg);
   struct cmsghdr *h= 0;
-  ssize_t rs= recvmsg(fd, &msg, MSG_DONTWAIT);
   if (rs >= 0) h= CMSG_FIRSTHDR(&msg);
   if (!h) {
-    int status;
-    pid_t got= waitpid(connecting_child, &status, WNOHANG);
-    if (got != -1) {
-      assert(got==connecting_child);
-      connecting_child= 0;
-      if (WIFEXITED(status)) {
-       if (WEXITSTATUS(status) != 0 &&
-           WEXITSTATUS(status) != CONNCHILD_ESTATUS_STREAM &&
-           WEXITSTATUS(status) != CONNCHILD_ESTATUS_NOSTREAM)
-         /* child already reported the problem */;
-       else
-         warn("connect: connection child exited code %d but no cmsg",
+    int status= xwaitpid(&connecting_child, "connect child (broken)");
+
+    if (WIFEXITED(status)) {
+      if (WEXITSTATUS(status) != 0 &&
+         WEXITSTATUS(status) != CONNCHILD_ESTATUS_STREAM &&
+         WEXITSTATUS(status) != CONNCHILD_ESTATUS_NOSTREAM)
+       /* child already reported the problem */;
+      else {
+       if (e == OOP_EXCEPTION)
+         warn("connect: connection child exited code %d but"
+              " unexpected exception on fdpass socket",
               WEXITSTATUS(status));
-      } else if (WIFSIGNALED(status) && WTERMSIG(status) == SIGALRM) {
-       warn("connect: connection attempt timed out");
-      } else {
-       report_child_status("connect", status);
+       else
+         warn("connect: connection child exited code %d but"
+              " no cmsg (rs=%d)",
+              WEXITSTATUS(status), (int)rs);
       }
+    } else if (WIFSIGNALED(status) && WTERMSIG(status) == SIGALRM) {
+      warn("connect: connection attempt timed out");
     } else {
-      /* child is still running apparently, report the socket problem */
-      if (rs < 0)
-       syswarn("connect: read from fdpass socket failed");
-      else if (e == OOP_EXCEPTION)
-       warn("connect: unexpected exception on fdpass socket");
-      else if (!rs)
-       warn("connect: unexpected EOF on fdpass socket");
-      else
-       fatal("connect: unexpected lack of cmsg from child");
+      report_child_status("connect", status);
     }
     goto x;
   }
@@ -1219,9 +1227,6 @@ static void *connchild_event(oop_source *lp, int fd, oop_event e, void *u) {
   }
 
   /* Phew! */
-  LIST_INIT(conn->waiting);
-  LIST_INIT(conn->priority);
-  LIST_INIT(conn->sent);
   conn->max_queue= conn->stream ? max_queue_per_conn : 1;
 
   loop->on_fd(loop, conn->fd, OOP_EXCEPTION, conn_exception, conn);
@@ -1255,7 +1260,7 @@ static void connect_start(void) {
   assert(!connecting_child);
   assert(!connecting_fdpass_sock);
 
-  notice("starting connection attempt");
+  info("starting connection attempt");
 
   int socks[2];
   int r= socketpair(AF_UNIX, SOCK_STREAM, 0, socks);
@@ -1272,13 +1277,26 @@ static void connect_start(void) {
 
     alarm(connection_setup_timeout);
     if (NNTPconnect((char*)remote_host, port, &cn_from, &cn_to, buf) < 0) {
-      if (buf[0]) fatal("connect: rejected: %s", sanitise(buf));
-      else sysfatal("connect: connection attempt failed");
+      int l= strlen(buf);
+      int stripped=0;
+      while (l>0) {
+       unsigned char c= buf[l-1];
+       if (!isspace(c)) break;
+       if (c=='\n' || c=='\r') stripped=1;
+       --l;
+      }
+      if (!buf[0]) {
+       sysfatal("connect: connection attempt failed");
+      } else {
+       buf[l]= 0;
+       fatal("connect: %s: %s", stripped ? "rejected" : "failed",
+             sanitise(buf));
+      }
     }
     if (NNTPsendpassword((char*)remote_host, cn_from, cn_to) < 0)
       sysfatal("connect: authentication failed");
     if (try_stream) {
-      if (fputs("MODE STREAM\r\n", cn_to) ||
+      if (fputs("MODE STREAM\r\n", cn_to)==EOF ||
          fflush(cn_to))
        sysfatal("connect: could not send MODE STREAM");
       buf[sizeof(buf)-1]= 0;
@@ -1290,7 +1308,7 @@ static void connect_start(void) {
       }
       int l= strlen(buf);
       assert(l>=1);
-      if (buf[-1]!='\n')
+      if (buf[l-1]!='\n')
        fatal("connect: response to MODE STREAM is too long: %.100s...",
              sanitise(buf));
       l--;  if (l>0 && buf[l-1]=='\r') l--;
@@ -1325,13 +1343,15 @@ static void connect_start(void) {
 
     msg.msg_controllen= cmsg->cmsg_len;
     r= sendmsg(socks[1], &msg, 0);
-    if (r) sysdie("sendmsg failed for new connection");
+    if (r<0) sysdie("sendmsg failed for new connection");
+    if (r!=1) die("sendmsg for new connection gave wrong result %d",r);
 
     _exit(exitstatus);
   }
 
   xclose(socks[1], "connecting fdpass child's socket",0);
   connecting_fdpass_sock= socks[0];
+  xsetnonblock(connecting_fdpass_sock, 1);
   on_fd_read_except(connecting_fdpass_sock, connchild_event);
 }
 
@@ -1365,6 +1385,7 @@ static void check_assign_articles(void) {
       if (!inqueue) use->since_activity= 0; /* reset idle counter */
       while (spare>0) {
        Article *art= LIST_REMHEAD(queue);
+       if (!art) break;
        LIST_ADDTAIL(use->waiting, art);
        spare--;
       }
@@ -3062,7 +3083,7 @@ static const Option innduct_options[]= {
 {'q',"quiet-multiple",   0,       &quiet_multiple,           op_setint, 1   },
 {0,"no-daemon",          0,       &become_daemon,            op_setint, 0   },
 {0,"no-streaming",       0,       &try_stream,               op_setint, 0   },
-{0,"inndconf",           "F",     &inndconffile,             op_string      },
+{'C',"inndconf",         "F",     &inndconffile,             op_string      },
 {'P',"port",             "PORT",  &port,                     op_integer     },
 {0,"help",               0,       0,                         help           },
 
@@ -3127,6 +3148,9 @@ int main(int argc, char **argv) {
 
   /* defaults */
 
+  int r= innconf_read(inndconffile);
+  if (!r) badusage("could not read inn.conf (more info on stderr)");
+
   if (!remote_host) remote_host= sitename;
 
   if (nocheck_thresh < 0 || nocheck_thresh > 100)
@@ -3149,7 +3173,6 @@ int main(int argc, char **argv) {
   max_bad_data_ratio *= 0.01;
 
   if (!feedfile) {
-    innconf_read(inndconffile);
     feedfile= xasprintf("%s/%s",innconf->pathoutgoing,sitename);
   } else if (!feedfile[0]) {
     badusage("feed filename must be nonempty");
@@ -3228,7 +3251,7 @@ int main(int argc, char **argv) {
 
   /* let's go */
 
-  void *r= oop_sys_run(sysloop);
-  assert(r == OOP_ERROR);
+  void *run= oop_sys_run(sysloop);
+  assert(run == OOP_ERROR);
   sysdie("event loop failed");
 }