Massive overhaul of networking code. Changes:
[mmh] / sbr / client.c
index a7ab764..ae6968f 100644 (file)
@@ -11,6 +11,7 @@
 
 #include <h/mh.h>
 #include <h/mts.h>
+#include <h/utils.h>
 #include <errno.h>
 #include <sys/socket.h>
 #include <netinet/in.h>
 # include <arpa/inet.h>
 #endif
 
-#ifdef HESIOD
-# include <hesiod.h>
-#endif
-
-#ifdef KPOP
-# include <krb.h>
-# include <ctype.h>
-#endif /* KPOP */
-
 #define        TRUE         1
 #define        FALSE        0
 
-#define        OOPS1      (-2)
-#define        OOPS2      (-3)
-
 #define        MAXARGS   1000
-#define        MAXNETS      5
-#define        MAXHOSTS    25
-
-struct addrent {
-    int a_addrtype;            /* assumes AF_INET for inet_netof () */
-    union {
-       int un_net;
-       char un_addr[14];
-    } un;
-};
-
-#define        a_net  un.un_net
-#define        a_addr un.un_addr
-
-static struct addrent *n1, *n2;
-static struct addrent nets[MAXNETS];
-static struct addrent *h1, *h2;
-static struct addrent hosts[MAXHOSTS];
-
-#ifdef KPOP
-static CREDENTIALS cred;
-static MSG_DAT msg_data;
-static KTEXT ticket = (KTEXT) NULL;
-static Key_schedule schedule;
-static char *kservice;                 /* "pop" if using kpop */
-char krb_realm[REALM_SZ];
-char *PrincipalHostname();
-#endif /* KPOP */
-
-#if !defined(h_addr)
-# define h_addr h_addr_list[0]
-#endif
-
-#define        inaddr_copy(hp,sin) \
-    memcpy(&((sin)->sin_addr), (hp)->h_addr, (hp)->h_length)
 
 /*
  * static prototypes
  */
-static int rcaux (struct servent *, struct hostent *, int, char *, int);
-static int getport (int, int, char *, int);
-static int inet (struct hostent *, int);
-struct hostent *gethostbystring (char *s);
 
 /* client's own static version of several nmh subroutines */
 static char **client_brkstring (char *, char *, char *);
 static int client_brkany (char, char *);
 static char **client_copyip (char **, char **, int);
 static char *client_getcpy (char *);
+static void client_freelist(char **);
 
 
 int
-client (char *args, char *protocol, char *service, int rproto,
-               char *response, int len_response)
+client (char *args, char *service, char *response, int len_response, int debug)
 {
-    int sd;
-    register char **ap;
-    char *arguments[MAXARGS];
-    register struct hostent *hp;
-    register struct servent *sp;
-#ifndef        HAVE_GETHOSTBYNAME
-    register struct netent *np;
-#endif
-
-#ifdef KPOP
-    char *cp;
-
-    kservice = service;
-    if (cp = strchr (service, '/')) {  /* "pop/kpop" */
-       *cp++ = '\0';           /* kservice = "pop" */
-       service = cp;           /* service  = "kpop" */
-    } else {
-       kservice = NULL;        /* not using KERBEROS */
-    }
-#endif /* KPOP */
-    
-
-    if ((sp = getservbyname (service, protocol)) == NULL) {
-#ifdef HESIOD
-       if ((sp = hes_getservbyname (service, protocol)) == NULL) {
-           snprintf (response, len_response, "%s/%s: unknown service", protocol, service);
-           return NOTOK;
-       }
-#else
-       snprintf (response, len_response, "%s/%s: unknown service", protocol, service);
-       return NOTOK;
-#endif
-    }
+    int sd, rc;
+    char **ap, *arguments[MAXARGS];
+    struct addrinfo hints, *res, *ai;
 
     ap = arguments;
     if (args != NULL && *args != 0) {
@@ -139,267 +59,76 @@ client (char *args, char *protocol, char *service, int rproto,
        *ap = NULL;
     }
 
-    n1 = nets;
-    n2 = nets + sizeof(nets) / sizeof(nets[0]);
-
-    h1 = hosts;
-    h2 = hosts + sizeof(hosts) / sizeof(hosts[0]);
+    memset(&hints, 0, sizeof(hints));
+    hints.ai_flags = AI_ADDRCONFIG;
+    hints.ai_family = PF_UNSPEC;
+    hints.ai_socktype = SOCK_STREAM;
 
     for (ap = arguments; *ap; ap++) {
-       if (**ap == '\01') {
-/*
- * the assumption here is that if the system doesn't have a
- * gethostbyname() function, it must not use DNS. So we need to look
- * into the /etc/hosts using gethostent(). There probablly aren't any
- * systems still like this, but you never know. On every system I have
- * access to, this section is ignored.
- */
-#ifndef        HAVE_GETHOSTBYNAME
-           if ((np = getnetbyname (*ap + 1))) {
-#ifdef HAVE_SETHOSTENT
-               sethostent (1);
-#endif /* HAVE_SETHOSTENT */
-               while ((hp = gethostent()))
-                   if (np->n_addrtype == hp->h_addrtype
-                           && inet (hp, np->n_net)) {
-                       switch (sd = rcaux (sp, hp, rproto, response, len_response)) {
-                           case NOTOK: 
-                               continue;
-                           case OOPS1: 
-                               break;
-                           case OOPS2: 
-                               return NOTOK;
-
-                           default: 
-                               return sd;
-                       }
-                       break;
-                   }
-           }
-#endif /* don't HAVE_GETHOSTBYNAME */
-           continue;
+
+       if (debug) {
+           fprintf(stderr, "Trying to connect to \"%s\" ...\n", *ap);
        }
 
-       if ((hp = gethostbystring (*ap))) {
-           switch (sd = rcaux (sp, hp, rproto, response, len_response)) {
-               case NOTOK: 
-               case OOPS1: 
-                   break;
-               case OOPS2: 
-                   return NOTOK;
+       rc = getaddrinfo(*ap, service, &hints, &res);
 
-               default: 
-                   return sd;
+       if (rc) {
+           if (debug) {
+               fprintf(stderr, "Lookup of \"%s\" failed: %s\n", *ap,
+                       gai_strerror(rc));
            }
            continue;
        }
-    }
 
-    strncpy (response, "no servers available", len_response);
-    return NOTOK;
-}
-
-
-static int
-rcaux (struct servent *sp, struct hostent *hp, int rproto,
-               char *response, int len_response)
-{
-    int sd;
-    struct in_addr in;
-    register struct addrent *ap;
-    struct sockaddr_in in_socket;
-    register struct sockaddr_in *isock = &in_socket;
-
-#ifdef KPOP
-    int rem;
-    struct hostent *hp2;
-#endif /* KPOP */
-
-    for (ap = nets; ap < n1; ap++)
-       if (ap->a_addrtype == hp->h_addrtype && inet (hp, ap->a_net))
-           return NOTOK;
-
-    for (ap = hosts; ap < h1; ap++)
-       if (ap->a_addrtype == hp->h_addrtype
-               && memcmp(ap->a_addr, hp->h_addr, hp->h_length) == 0)
-           return NOTOK;
-
-    if ((sd = getport (rproto, hp->h_addrtype, response, len_response)) == NOTOK)
-       return OOPS2;
-
-    memset (isock, 0, sizeof(*isock));
-    isock->sin_family = hp->h_addrtype;
-    inaddr_copy (hp, isock);
-    isock->sin_port = sp->s_port;
-
-    if (connect (sd, (struct sockaddr *) isock, sizeof(*isock)) == NOTOK)
-       switch (errno) {
-           case ENETDOWN: 
-           case ENETUNREACH: 
-               close (sd);
-               if (n1 < n2) {
-                   n1->a_addrtype = hp->h_addrtype;
-                   memcpy(&in, hp->h_addr, sizeof(in));
-                   n1->a_net = inet_netof (in);
-                   n1++;
-               }
-               return OOPS1;
-
-           case ETIMEDOUT: 
-           case ECONNREFUSED: 
-           default: 
-               close (sd);
-               if (h1 < h2) {
-                   h1->a_addrtype = hp->h_addrtype;
-                   memcpy(h1->a_addr, hp->h_addr, hp->h_length);
-                   h1++;
-               }
-               return NOTOK;
-       }
+       for (ai = res; ai != NULL; ai = ai->ai_next) {
+           if (debug) {
+               char address[NI_MAXHOST];
 
-#ifdef KPOP
-    if (kservice) {    /* "pop" */
-       char *instance;
+               rc = getnameinfo(ai->ai_addr, ai->ai_addrlen, address,
+                                sizeof(address), NULL, NULL, NI_NUMERICHOST);
 
-       if (( hp2 = gethostbyaddr( hp->h_addr, hp->h_length, hp->h_addrtype ))
-               == NULL ) {
-           return NOTOK;
-       }
-       if ((instance = strdup (hp2->h_name)) == NULL) {
-           close (sd);
-           strncpy (response, "Out of memory.", len_response);
-           return OOPS2;
-       }
-       ticket = (KTEXT) malloc (sizeof(KTEXT_ST));
-       rem = krb_sendauth (0L, sd, ticket, kservice, instance,
-                          (char *) krb_realmofhost (instance),
-                          (unsigned long) 0, &msg_data, &cred, schedule,
-                          (struct sockaddr_in *) NULL,
-                          (struct sockaddr_in *) NULL,
-                          "KPOPV0.1");
-       free (instance);
-       if (rem != KSUCCESS) {
-           close (sd);
-           strncpy (response, "Post office refused connection: ", len_response);
-           strncat (response, krb_err_txt[rem], len_response - strlen(response));
-           return OOPS2;
-       }
-    }
-#endif /* KPOP */
+               fprintf(stderr, "Connecting to %s...\n",
+                       rc ? "unknown" : address);
+           }
 
-    return sd;
-}
+           sd = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
 
+           if (sd < 0) {
+               if (debug)
+                   fprintf(stderr, "socket() failed: %s\n", strerror(errno));
+               continue;
+           }
 
-static int
-getport (int rproto, int addrtype, char *response, int len_response)
-{
-    int sd, port;
-    struct sockaddr_in in_socket, *isock;
-
-    isock = &in_socket;
-    if (rproto && addrtype != AF_INET) {
-       snprintf (response, len_response, "reserved ports not supported for af=%d", addrtype);
-       errno = ENOPROTOOPT;
-       return NOTOK;
-    }
+           if (connect(sd, ai->ai_addr, ai->ai_addrlen) == 0) {
+               freeaddrinfo(res);
+               client_freelist(ap);
+               return sd;
+           }
 
-    if ((sd = socket (AF_INET, SOCK_STREAM, 0)) == NOTOK) {
-       char *s;
+           if (debug) {
+               fprintf(stderr, "Connection failed: %s\n", strerror(errno));
+           }
 
-       if ((s = strerror (errno)))
-           snprintf (response, len_response, "unable to create socket: %s", s);
-       else
-           snprintf (response, len_response, "unable to create socket: unknown error");
-       return NOTOK;
-    }
-#ifdef KPOP
-    if (kservice)      /* "pop" */
-       return(sd);
-#endif /* KPOP */
-    if (!rproto)
-       return sd;
-
-    memset(isock, 0, sizeof(*isock));
-    isock->sin_family = addrtype;
-    for (port = IPPORT_RESERVED - 1;;) {
-       isock->sin_port = htons ((unsigned short) port);
-       if (bind (sd, (struct sockaddr *) isock, sizeof(*isock)) != NOTOK)
-           return sd;
-
-       switch (errno) {
-           char *s;
-
-           case EADDRINUSE: 
-           case EADDRNOTAVAIL: 
-               if (--port <= IPPORT_RESERVED / 2) {
-                   strncpy (response, "ports available", len_response);
-                   return NOTOK;
-               }
-               break;
-
-           default: 
-               if ((s = strerror (errno)))
-                   snprintf (response, len_response, "unable to bind socket: %s", s);
-               else
-                   snprintf (response, len_response, "unable to bind socket: unknown error");
-               return NOTOK;
+           close(sd);
        }
     }
-}
 
-
-static int
-inet (struct hostent *hp, int net)
-{
-    struct in_addr in;
-
-    memcpy(&in, hp->h_addr, sizeof(in));
-    return (inet_netof (in) == net);
+    freeaddrinfo(res);
+    client_freelist(ap);
+    strncpy (response, "no servers available", len_response);
+    return NOTOK;
 }
 
 
 /*
- * taken from ISODE's compat/internet.c
+ * Free a list of strings
  */
 
-static char *empty = NULL;
-
-#ifdef h_addr
-static char *addrs[2] = { NULL };
-#endif
-
-struct hostent *
-gethostbystring (char *s)
+static void
+client_freelist(char **list)
 {
-    register struct hostent *h;
-    static struct hostent hs;
-#ifdef DG
-    static struct in_addr iaddr;
-#else
-    static unsigned long iaddr;
-#endif
-
-    iaddr = inet_addr (s);
-#ifdef DG
-    if (iaddr.s_addr == NOTOK && strcmp (s, "255.255.255.255"))
-#else
-    if (((int) iaddr == NOTOK) && strcmp (s, "255.255.255.255"))
-#endif
-       return gethostbyname (s);
-
-    h = &hs;
-    h->h_name = s;
-    h->h_aliases = &empty;
-    h->h_addrtype = AF_INET;
-    h->h_length = sizeof(iaddr);
-#ifdef h_addr
-    h->h_addr_list = addrs;
-    memset(addrs, 0, sizeof(addrs));
-#endif
-    h->h_addr = (char *) &iaddr;
-
-    return h;
+    while (*list++ != NULL)
+       free(*list);
 }
 
 
@@ -474,8 +203,7 @@ client_getcpy (char *str)
     size_t len;
 
     len = strlen(str) + 1;
-    if (!(cp = malloc(len)))
-       return NULL;
+    cp = mh_xmalloc(len);
 
     memcpy (cp, str, len);
     return cp;