/* -*- mode: c; c-basic-offset: 4; indent-tabs-mode: nil -*- */

/* tests/resolve/addrinfo-test.c */

/*

 * Copyright 2004 by the Massachusetts Institute of Technology.

 * All Rights Reserved.

 *

 * Export of this software from the United States of America may

 *   require a specific license from the United States Government.

 *   It is the responsibility of any person or organization contemplating

 *   export to obtain such a license before exporting.

 *

 * WITHIN THAT CONSTRAINT, permission to use, copy, modify, and

 * distribute this software and its documentation for any purpose and

 * without fee is hereby granted, provided that the above copyright

 * notice appear in all copies and that both that copyright notice and

 * this permission notice appear in supporting documentation, and that

 * the name of M.I.T. not be used in advertising or publicity pertaining

 * to distribution of the software without specific, written prior

 * permission.  Furthermore if you modify this software you must label

 * your software as modified software and not distribute it in such a

 * fashion that it might be confused with the original M.I.T. software.

 * M.I.T. makes no representations about the suitability of

 * this software for any purpose.  It is provided "as is" without express

 * or implied warranty.

 */

/*

 * A simple program to test the functionality of the getaddrinfo function.

 *

 * Usage:

 *   addrinfo-test [-t|-u|-R|-I] [-d|-s|-r] [-p port] [-P] [hostname]

 *

 *   When invoked with no arguments, NULL is used for the node name,

 *   which (at least with a non-null "port") means a socket address

 *   is desired that can be used with connect() or bind() (depending

 *   on whether "-P" is given).

 */

#include <k5-platform.h>

#include <sys/types.h>

#include <sys/socket.h>

#include <netdb.h>

#include <netinet/in.h> /* needed for IPPROTO_* on NetBSD */

#ifdef USE_FAKE_ADDRINFO

#include "fake-addrinfo.h"

#endif

static const char *protoname (int p) {

    static char buf[30];

#define X(N) if (p == IPPROTO_ ## N) return #N

    X(TCP);

    X(UDP);

    X(ICMP);

#ifdef IPPROTO_IPV6

    X(IPV6);

#endif

#ifdef IPPROTO_GRE

    X(GRE);

#endif

#ifdef IPPROTO_NONE

    X(NONE);

#endif

    X(RAW);

#ifdef IPPROTO_COMP

    X(COMP);

#endif

    snprintf(buf, sizeof(buf), " %-2d", p);

    return buf;

}

static const char *socktypename (int t) {

    static char buf[30];

    switch (t) {

    case SOCK_DGRAM: return "DGRAM";

    case SOCK_STREAM: return "STREAM";

    case SOCK_RAW: return "RAW";

    case SOCK_RDM: return "RDM";

    case SOCK_SEQPACKET: return "SEQPACKET";

    }

    snprintf(buf, sizeof(buf), " %-2d", t);

    return buf;

}

static char *whoami;

static void usage () {

    fprintf(stderr,

            "usage:\n"

            "\t%s [ options ] [host]\n"

            "options:\n"

            "\t-t\tspecify protocol IPPROTO_TCP\n"

            "\t-u\tspecify protocol IPPROTO_UDP\n"

            "\t-R\tspecify protocol IPPROTO_RAW\n"

            "\t-I\tspecify protocol IPPROTO_ICMP\n"

            "\n"

            "\t-d\tspecify socket type SOCK_DGRAM\n"

            "\t-s\tspecify socket type SOCK_STREAM\n"

            "\t-r\tspecify socket type SOCK_RAW\n"

            "\n"

            "\t-4\tspecify address family AF_INET\n"

#ifdef AF_INET6

            "\t-6\tspecify address family AF_INET6\n"

#endif

            "\n"

            "\t-p P\tspecify port P (service name or port number)\n"

            "\t-N\thostname is numeric, skip DNS query\n"

            "\t-n\tservice/port is numeric (sets AI_NUMERICSERV)\n"

            "\t-P\tset AI_PASSIVE\n"

            "\n"

            "default: protocol 0, socket type 0, address family 0, null port\n"

            ,

            whoami);

    /* [ -t | -u | -R | -I ] [ -d | -s | -r ] [ -p port ] */

    exit (1);

}

static const char *familyname (int f) {

    static char buf[30];

    switch (f) {

    default:

        snprintf(buf, sizeof(buf), "AF %d", f);

        return buf;

    case AF_INET: return "AF_INET";

#ifdef AF_INET6

    case AF_INET6: return "AF_INET6";

#endif

    }

}

#define eaistr(X) (X == EAI_SYSTEM ? strerror(errno) : gai_strerror(X))

int main (int argc, char *argv[])

{

    struct addrinfo *ap, *ap2;

    int err, numerichost = 0, numericserv = 0;

    char *hname, *port = 0, *sep;

    struct addrinfo hints;

    whoami = strrchr(argv[0], '/');

    if (whoami == 0)

        whoami = argv[0];

    else

        whoami = whoami+1;

    memset(&hints, 0, sizeof(hints));

    hints.ai_flags = 0;

    hints.ai_socktype = 0;

    hname = 0;

    hints.ai_family = 0;

    if (argc == 1)

        usage ();

    while (++argv, --argc > 0) {

        char *arg;

        arg = *argv;

        if (*arg != '-')

            hname = arg;

        else if (arg[1] == 0 || arg[2] != 0)

            usage ();

        else

            switch (arg[1]) {

            case 'u':

                hints.ai_protocol = IPPROTO_UDP;

                break;

            case 't':

                hints.ai_protocol = IPPROTO_TCP;

                break;

            case 'R':

                hints.ai_protocol = IPPROTO_RAW;

                break;

            case 'I':

                hints.ai_protocol = IPPROTO_ICMP;

                break;

            case 'd':

                hints.ai_socktype = SOCK_DGRAM;

                break;

            case 's':

                hints.ai_socktype = SOCK_STREAM;

                break;

            case 'r':

                hints.ai_socktype = SOCK_RAW;

                break;

            case 'p':

                if (argv[1] == 0 || argv[1][0] == 0 || argv[1][0] == '-')

                    usage ();

                port = argv[1];

                argc--, argv++;

                break;

            case '4':

                hints.ai_family = AF_INET;

                break;

#ifdef AF_INET6

            case '6':

                hints.ai_family = AF_INET6;

                break;

#endif

            case 'N':

                numerichost = 1;

                break;

            case 'n':

                numericserv = 1;

                break;

            case 'P':

                hints.ai_flags |= AI_PASSIVE;

                break;

            default:

                usage ();

            }

    }

    if (hname && !numerichost)

        hints.ai_flags |= AI_CANONNAME;

    if (numerichost) {

#ifdef AI_NUMERICHOST

        hints.ai_flags |= AI_NUMERICHOST;

#else

        fprintf(stderr, "AI_NUMERICHOST not defined on this platform\n");

        exit(1);

#endif

    }

    if (numericserv) {

#ifdef AI_NUMERICSERV

        hints.ai_flags |= AI_NUMERICSERV;

#else

        fprintf(stderr, "AI_NUMERICSERV not defined on this platform\n");

        exit(1);

#endif

    }

    printf("getaddrinfo(hostname %s, service %s,\n"

           "            hints { ",

           hname ? hname : "(null)", port ? port : "(null)");

    sep = "";

#define Z(FLAG) if (hints.ai_flags & AI_##FLAG) printf("%s%s", sep, #FLAG), sep = "|"

    Z(CANONNAME);

    Z(PASSIVE);

#ifdef AI_NUMERICHOST

    Z(NUMERICHOST);

#endif

#ifdef AI_NUMERICSERV

    Z(NUMERICSERV);

#endif

    if (sep[0] == 0)

        printf ("no-flags");

    if (hints.ai_family)

        printf(" %s", familyname(hints.ai_family));

    if (hints.ai_socktype)

        printf(" SOCK_%s", socktypename(hints.ai_socktype));

    if (hints.ai_protocol)

        printf(" IPPROTO_%s", protoname(hints.ai_protocol));

    printf(" }):\n");

    err = getaddrinfo(hname, port, &hints, &ap);

    if (err) {

        printf("\terror => %s\n", eaistr(err));

        return 1;

    }

    for (ap2 = ap; ap2; ap2 = ap2->ai_next) {

        char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];

        /* If we don't do this, even AIX's own getnameinfo will reject

           the sockaddr structures.  The sa_len field doesn't get set

           either, on AIX, but getnameinfo won't complain.  */

        if (ap2->ai_addr->sa_family == 0) {

            printf("BAD: sa_family zero! fixing...\n");

            ap2->ai_addr->sa_family = ap2->ai_family;

        } else if (ap2->ai_addr->sa_family != ap2->ai_family) {

            printf("BAD: sa_family != ai_family! fixing...\n");

            ap2->ai_addr->sa_family = ap2->ai_family;

        }

        if (getnameinfo(ap2->ai_addr, ap2->ai_addrlen, hbuf, sizeof(hbuf),

                        pbuf, sizeof(pbuf), NI_NUMERICHOST | NI_NUMERICSERV)) {

            strlcpy(hbuf, "...", sizeof(hbuf));

            strlcpy(pbuf, "...", sizeof(pbuf));

        }

        printf("%p:\n"

               "\tfamily = %s\tproto = %-4s\tsocktype = %s\n",

               (void *) ap2, familyname(ap2->ai_family),

               protoname (ap2->ai_protocol),

               socktypename (ap2->ai_socktype));

        if (ap2->ai_canonname) {

            if (ap2->ai_canonname[0])

                printf("\tcanonname = %s\n", ap2->ai_canonname);

            else

                printf("BAD: ai_canonname is set but empty!\n");

        } else if (ap2 == ap && (hints.ai_flags & AI_CANONNAME)) {

            printf("BAD: first ai_canonname is null!\n");

        }

        printf("\taddr = %-28s\tport = %s\n", hbuf, pbuf);

        err = getnameinfo(ap2->ai_addr, ap2->ai_addrlen, hbuf, sizeof (hbuf),

                          pbuf, sizeof(pbuf), NI_NAMEREQD);

        if (err)

            printf("\tgetnameinfo(NI_NAMEREQD): %s\n", eaistr(err));

        else

            printf("\tgetnameinfo => %s, %s\n", hbuf, pbuf);

    }

    freeaddrinfo(ap);

    return 0;

}