noip.c (fixup_real_ip_socket): Support for temporary fixups.
[preload-hacks] / noip.c
diff --git a/noip.c b/noip.c
index ec66340..2a09c16 100644 (file)
--- a/noip.c
+++ b/noip.c
@@ -62,19 +62,21 @@ enum { UNUSED, STALE, USED };               /* Unix socket status values */
 enum { WANT_FRESH, WANT_EXISTING };    /* Socket address dispositions */
 enum { DENY, ALLOW };                  /* ACL verdicts */
 
-static int address_families[] = { AF_INET, -1 };
+static int address_families[] = { AF_INET, AF_INET6, -1 };
 
 #define ADDRBUFSZ 64
 
 /* Address representations. */
 typedef union ipaddr {
   struct in_addr v4;
+  struct in6_addr v6;
 } ipaddr;
 
 /* Convenient socket address hacking. */
 typedef union address {
   struct sockaddr sa;
   struct sockaddr_in sin;
+  struct sockaddr_in6 sin6;
 } address;
 
 /* Access control list nodes */
@@ -91,7 +93,7 @@ typedef struct full_ipaddr {
   int af;
   ipaddr addr;
 } full_ipaddr;
-#define MAX_LOCAL_IPADDRS 16
+#define MAX_LOCAL_IPADDRS 64
 static full_ipaddr local_ipaddrs[MAX_LOCAL_IPADDRS];
 static int n_local_ipaddrs;
 
@@ -144,6 +146,7 @@ static void import(void)
 /* Socket address casts */
 #define SA(sa) ((struct sockaddr *)(sa))
 #define SIN(sa) ((struct sockaddr_in *)(sa))
+#define SIN6(sa) ((struct sockaddr_in6 *)(sa))
 #define SUN(sa) ((struct sockaddr_un *)(sa))
 
 /* Raw bytes */
@@ -202,6 +205,7 @@ static int family_known_p(int af)
 {
   switch (af) {
     case AF_INET:
+    case AF_INET6:
       return (1);
     default:
       return (0);
@@ -213,6 +217,7 @@ static socklen_t family_socklen(int af)
 {
   switch (af) {
     case AF_INET: return (sizeof(struct sockaddr_in));
+    case AF_INET6: return (sizeof(struct sockaddr_in6));
     default: abort();
   }
 }
@@ -222,6 +227,7 @@ static int address_width(int af)
 {
   switch (af) {
     case AF_INET: return 32;
+    case AF_INET6: return 128;
     default: abort();
   }
 }
@@ -238,6 +244,23 @@ static int common_prefix_length(int af, const ipaddr *a, const ipaddr *b)
       if ((aa&m) == 0 && (bb&m) == m) return (32 - simple_mask_length(m));
       else return (-1);
     } break;
+    case AF_INET6: {
+      const uint8_t *aa = a->v6.s6_addr, *bb = b->v6.s6_addr;
+      unsigned m;
+      unsigned n;
+      int i;
+
+      for (i = 0; i < 16 && aa[i] == bb[i]; i++);
+      n = 8*i;
+      if (i < 16) {
+       m = aa[i]^bb[i];
+       if ((aa[i]&m) != 0 || (bb[i]&m) != m) return (-1);
+       n += 8 - simple_mask_length(m);
+       for (i++; i < 16; i++)
+         if (aa[i] || bb[i] != 0xff) return (-1);
+      }
+      return (n);
+    } break;
     default:
       abort();
   }
@@ -248,6 +271,7 @@ static int port_from_sockaddr(const struct sockaddr *sa)
 {
   switch (sa->sa_family) {
     case AF_INET: return (ntohs(SIN(sa)->sin_port));
+    case AF_INET6: return (ntohs(SIN6(sa)->sin6_port));
     default: abort();
   }
 }
@@ -257,14 +281,17 @@ static void port_to_sockaddr(struct sockaddr *sa, int port)
 {
   switch (sa->sa_family) {
     case AF_INET: SIN(sa)->sin_port = htons(port); break;
+    case AF_INET6: SIN6(sa)->sin6_port = htons(port); break;
     default: abort();
   }
 }
+
 /* Extract the address part from SA and store it in A. */
 static void ipaddr_from_sockaddr(ipaddr *a, const struct sockaddr *sa)
 {
   switch (sa->sa_family) {
     case AF_INET: a->v4 = SIN(sa)->sin_addr; break;
+    case AF_INET6: a->v6 = SIN6(sa)->sin6_addr; break;
     default: abort();
   }
 }
@@ -279,6 +306,7 @@ static int ipaddr_equal_p(int af, const ipaddr *a, const ipaddr *b)
 {
   switch (af) {
     case AF_INET: return (a->v4.s_addr == b->v4.s_addr);
+    case AF_INET6: return (memcmp(a->v6.s6_addr, b->v6.s6_addr, 16) == 0);
     default: abort();
   }
 }
@@ -295,6 +323,19 @@ static int sockaddr_in_range_p(const struct sockaddr *sa,
       return (ntohl(a->v4.s_addr) <= addr &&
              addr <= ntohl(b->v4.s_addr));
     } break;
+    case AF_INET6: {
+      const uint8_t *ss = SIN6(sa)->sin6_addr.s6_addr;
+      const uint8_t *aa = a->v6.s6_addr, *bb = b->v6.s6_addr;
+      int h = 1, l = 1;
+      int i;
+
+      for (i = 0; h && l && i < 16; i++, ss++, aa++, bb++) {
+       if (*ss < *aa || *bb < *ss) return (0);
+       if (*aa < *ss) l = 0;
+       if (*ss < *bb) h = 0;
+      }
+      return (1);
+    } break;
     default:
       abort();
   }
@@ -311,6 +352,15 @@ static void wildcard_address(int af, struct sockaddr *sa)
       sin->sin_port = 0;
       sin->sin_addr.s_addr = INADDR_ANY;
     } break;
+    case AF_INET6: {
+      struct sockaddr_in6 *sin6 = SIN6(sa);
+      memset(sin6, 0, sizeof(*sin6));
+      sin6->sin6_family = AF_INET6;
+      sin6->sin6_port = 0;
+      sin6->sin6_addr = in6addr_any;
+      sin6->sin6_scope_id = 0;
+      sin6->sin6_flowinfo = 0;
+    } break;
     default:
       abort();
   }
@@ -329,6 +379,16 @@ static void mask_address(int af, ipaddr *a, int plen, int highp)
       if (highp) addr |= ~mask;
       a->v4.s_addr = htonl(addr & 0xffffffff);
     } break;
+    case AF_INET6: {
+      int i = plen/8;
+      unsigned m = (0xff << (8 - plen%8)) & 0xff;
+      unsigned s = highp ? 0xff : 0;
+      if (m) {
+       a->v6.s6_addr[i] = (a->v6.s6_addr[i] & m) | (s & ~m);
+       i++;
+      }
+      for (; i < 16; i++) a->v6.s6_addr[i] = s;
+    } break;
     default:
       abort();
   }
@@ -384,7 +444,7 @@ static char *present_sockaddr(const struct sockaddr *sa, socklen_t len,
       }
       WANT(1); PUTC(0);
     } break;
-    case AF_INET: {
+    case AF_INET: case AF_INET6: {
       char addrbuf[NI_MAXHOST], portbuf[NI_MAXSERV];
       int err = getnameinfo(sa, len,
                            addrbuf, sizeof(addrbuf),
@@ -407,7 +467,7 @@ nospace:
 
 /* Guess the family of a textual socket address. */
 static int guess_address_family(const char *p)
-  { return (AF_INET); }
+  { return (strchr(p, ':') ? AF_INET6 : AF_INET); }
 
 /* Parse a socket address P and write the result to SA. */
 static int parse_sockaddr(struct sockaddr *sa, const char *p)
@@ -443,7 +503,7 @@ static int parse_sockaddr(struct sockaddr *sa, const char *p)
 #ifdef DEBUG
 
 /* Write to standard error a description of the ACL node A. */
-static void dump_aclnode(aclnode *a)
+static void dump_aclnode(const aclnode *a)
 {
   char buf[ADDRBUFSZ];
   const char *p;
@@ -466,7 +526,7 @@ static void dump_aclnode(aclnode *a)
   fputc('\n', stderr);
 }
 
-static void dump_acl(aclnode *a)
+static void dump_acl(const aclnode *a)
 {
   int act = ALLOW;
 
@@ -481,7 +541,7 @@ static void dump_acl(aclnode *a)
 #endif
 
 /* Returns nonzero if the ACL A allows the socket address SA. */
-static int acl_allows_p(aclnode *a, const struct sockaddr *sa)
+static int acl_allows_p(const aclnode *a, const struct sockaddr *sa)
 {
   unsigned short port = port_from_sockaddr(sa);
   int act = ALLOW;
@@ -618,10 +678,10 @@ static int encode_inet_addr(struct sockaddr_un *sun,
 }
 
 /* Decode the Unix address SUN to an Internet address SIN.  If AF_HINT is
- * nonzero, an empty address (indicative of an unbound Unix-domain socket) of
- * the is translated to a wildcard Internet address of the appropriate
- * family.  Returns zero on success; -1 on failure (e.g., it wasn't one of
- * our addresses).
+ * nonzero, an empty address (indicative of an unbound Unix-domain socket) is
+ * translated to a wildcard Internet address of the appropriate family.
+ * Returns zero on success; -1 on failure (e.g., it wasn't one of our
+ * addresses).
  */
 static int decode_inet_addr(struct sockaddr *sa, int af_hint,
                            const struct sockaddr_un *sun,
@@ -655,9 +715,11 @@ static int decode_inet_addr(struct sockaddr *sa, int af_hint,
 
 /* SK is (or at least might be) a Unix-domain socket we created when an
  * Internet socket was asked for.  We've decided it should be an Internet
- * socket after all, with family AF_HINT, so convert it.
+ * socket after all, with family AF_HINT, so convert it.  If TMP is not null,
+ * then don't replace the existing descriptor: store the new socket in *TMP
+ * and return zero.
  */
-static int fixup_real_ip_socket(int sk, int af_hint)
+static int fixup_real_ip_socket(int sk, int af_hint, int *tmp)
 {
   int nsk;
   int type;
@@ -701,18 +763,22 @@ static int fixup_real_ip_socket(int sk, int af_hint)
 } while (0);
   OPTS(FIX)
 #undef FIX
-  if ((f = fcntl(sk, F_GETFL)) < 0 ||
-      (fd = fcntl(sk, F_GETFD)) < 0 ||
-      fcntl(nsk, F_SETFL, f) < 0 ||
-      dup2(nsk, sk) < 0) {
+  if (tmp)
+    *tmp = nsk;
+  else {
+    if ((f = fcntl(sk, F_GETFL)) < 0 ||
+       (fd = fcntl(sk, F_GETFD)) < 0 ||
+       fcntl(nsk, F_SETFL, f) < 0 ||
+       dup2(nsk, sk) < 0) {
+      close(nsk);
+      return (-1);
+    }
+    unlink(sun.sun_path);
     close(nsk);
-    return (-1);
-  }
-  unlink(sun.sun_path);
-  close(nsk);
-  if (fcntl(sk, F_SETFD, fd) < 0) {
-    perror("noip: fixup_real_ip_socket F_SETFD");
-    abort();
+    if (fcntl(sk, F_SETFD, fd) < 0) {
+      perror("noip: fixup_real_ip_socket F_SETFD");
+      abort();
+    }
   }
   return (0);
 }
@@ -728,7 +794,7 @@ static int do_implicit_bind(int sk, const struct sockaddr **sa,
   socklen_t mylen = sizeof(*sun);
 
   if (acl_allows_p(connect_real, *sa)) {
-    if (fixup_real_ip_socket(sk, (*sa)->sa_family)) return (-1);
+    if (fixup_real_ip_socket(sk, (*sa)->sa_family, 0)) return (-1);
   } else {
     if (real_getsockname(sk, SA(sun), &mylen) < 0) return (-1);
     if (sun->sun_family == AF_UNIX) {
@@ -978,8 +1044,9 @@ static void parse_acl_line(char **pp, aclnode ***tail)
     }
     SKIPSPC;
     if (*p != ',') break;
-    p++;
+    if (*p) p++;
   }
+  *pp = p;
   return;
 
 bad:
@@ -1000,6 +1067,7 @@ static void parse_autoports(char **pp)
   if (*p != '-') goto bad; p++;
   NEXTNUMBER(q, del); y = strtoul(q, 0, 0); RESCAN(del);
   minautoport = x; maxautoport = y;
+  *pp = p;
   return;
 
 bad:
@@ -1113,7 +1181,7 @@ int socket(int pf, int ty, int proto)
 
 int socketpair(int pf, int ty, int proto, int *sk)
 {
-  if (pf == PF_INET) {
+  if (family_known_p(pf)) {
     pf = PF_UNIX;
     proto = 0;
   }
@@ -1127,7 +1195,7 @@ int bind(int sk, const struct sockaddr *sa, socklen_t len)
   if (family_known_p(sa->sa_family)) {
     PRESERVING_ERRNO({
       if (acl_allows_p(bind_real, sa)) {
-       if (fixup_real_ip_socket(sk, sa->sa_family))
+       if (fixup_real_ip_socket(sk, sa->sa_family, 0))
          return (-1);
       } else {
        encode_inet_addr(&sun, sa, WANT_FRESH);
@@ -1165,7 +1233,7 @@ ssize_t sendto(int sk, const void *buf, size_t len, int flags,
 {
   struct sockaddr_un sun;
 
-  if (to && to->sa_family == AF_INET) {
+  if (to && family_known_p(to->sa_family)) {
     PRESERVING_ERRNO({
       do_implicit_bind(sk, &to, &tolen, &sun);
     });
@@ -1197,7 +1265,7 @@ ssize_t sendmsg(int sk, const struct msghdr *msg, int flags)
   const struct sockaddr *sa;
   struct msghdr mymsg;
 
-  if (msg->msg_name && SA(msg->msg_name)->sa_family == AF_INET) {
+  if (msg->msg_name && family_known_p(SA(msg->msg_name)->sa_family)) {
     PRESERVING_ERRNO({
       sa = SA(msg->msg_name);
       mymsg = *msg;
@@ -1377,13 +1445,13 @@ static void create_sockdir(void)
 {
   struct stat st;
 
-  if (stat(sockdir, &st)) {
+  if (lstat(sockdir, &st)) {
     if (errno == ENOENT) {
       if (mkdir(sockdir, 0700)) {
        perror("noip: creating socketdir");
        exit(127);
       }
-      if (!stat(sockdir, &st))
+      if (!lstat(sockdir, &st))
        goto check;
     }
     perror("noip: checking socketdir");