noip.c: Hack ioctl(2) as well.
[preload-hacks] / noip.c
diff --git a/noip.c b/noip.c
index ec66340..b9043f0 100644 (file)
--- a/noip.c
+++ b/noip.c
@@ -34,6 +34,7 @@
 #include <assert.h>
 #include <ctype.h>
 #include <errno.h>
+#include <stdarg.h>
 #include <stddef.h>
 #include <stdio.h>
 #include <stdlib.h>
@@ -62,19 +63,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 +94,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;
 
@@ -123,7 +126,8 @@ static aclnode *connect_real,  **connect_tail = &connect_real;
   _(recvfrom, ssize_t, (int, void *buf, size_t, int,                   \
                        struct sockaddr *from, socklen_t *fromlen))     \
   _(sendmsg, ssize_t, (int, const struct msghdr *, int))               \
-  _(recvmsg, ssize_t, (int, struct msghdr *, int))
+  _(recvmsg, ssize_t, (int, struct msghdr *, int))                     \
+  _(ioctl, int, (int, unsigned long, ...))
 
 /* Function pointers to set up. */
 #define DECL(imp, ret, args) static ret (*real_##imp) args;
@@ -144,6 +148,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 +207,7 @@ static int family_known_p(int af)
 {
   switch (af) {
     case AF_INET:
+    case AF_INET6:
       return (1);
     default:
       return (0);
@@ -213,6 +219,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 +229,7 @@ static int address_width(int af)
 {
   switch (af) {
     case AF_INET: return 32;
+    case AF_INET6: return 128;
     default: abort();
   }
 }
@@ -238,6 +246,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 +273,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 +283,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 +308,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 +325,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 +354,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 +381,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 +446,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 +469,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 +505,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 +528,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 +543,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 +680,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 +717,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 +765,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 +796,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 +1046,9 @@ static void parse_acl_line(char **pp, aclnode ***tail)
     }
     SKIPSPC;
     if (*p != ',') break;
-    p++;
+    if (*p) p++;
   }
+  *pp = p;
   return;
 
 bad:
@@ -1000,6 +1069,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 +1183,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 +1197,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 +1235,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 +1267,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;
@@ -1299,6 +1369,36 @@ int setsockopt(int sk, int lev, int opt, const void *p, socklen_t len)
   return (real_setsockopt(sk, lev, opt, p, len));
 }
 
+int ioctl(int fd, unsigned long op, ...)
+{
+  va_list ap;
+  void *arg;
+  int sk;
+  int rc;
+
+  va_start(ap, op);
+  arg = va_arg(ap, void *);
+
+  switch (op) {
+    case SIOCGIFADDR:
+    case SIOCGIFBRDADDR:
+    case SIOCGIFDSTADDR:
+    case SIOCGIFNETMASK:
+      PRESERVING_ERRNO({
+       if (fixup_real_ip_socket(fd, AF_INET, &sk)) goto real;
+      });
+      rc = real_ioctl(sk, op, arg);
+      PRESERVING_ERRNO({ close(sk); });
+      break;
+    default:
+    real:
+      rc = real_ioctl(fd, op, arg);
+      break;
+  }
+  va_end(ap);
+  return (rc);
+}
+
 /*----- Initialization ----------------------------------------------------*/
 
 /* Clean up the socket directory, deleting stale sockets. */
@@ -1377,13 +1477,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");