/*      $OpenBSD: netdb.c,v 1.2 2011/05/01 04:27:07 guenther Exp $      */

/*
 * Public domain, 2004, Otto Moerbeek <otto@drijf.net>
 */

#include <err.h>
#include <netdb.h>
#include <stdarg.h>
#include <string.h>

int ret = 0;

void
checkp(int n, struct protoent *p, int proto, const char *name, ...)
{
        va_list va;
        char *a;
        int i;

        if (p == NULL) {
                warnx("%d proto struct is NULL", n);
                ret = 1;
                return;
        }
        if (p->p_proto != proto) {
                warnx("%d proto num mismatch %d %d", n, p->p_proto, proto);
                ret = 1;
        }
        if (strcmp(p->p_name, name) != 0) {
                warnx("%d proto name mismatch %s %s", n, p->p_name, name);
                ret = 1;
        }
        va_start(va, name);
        a = va_arg(va, char *);
        i = 0;
        while (a != NULL) {
                if (strcmp(p->p_aliases[i], a) != 0) {
                        warnx("%d proto alias mismatch %s %s", n,
                            p->p_aliases[i], a);
                        ret = 1;
                }
                i++;
                a = va_arg(va, char *);
        }
        if (p->p_aliases[i] != NULL) {
                        warnx("%d proto alias list does not end with NULL", n);
                        ret = 1;
        }
        va_end(va);
}


void
checks(int n, struct servent *s, int port, const char *proto,
    const char *name, ...)
{
        va_list va;
        char *a;
        int i;

        if (s == NULL) {
                warnx("%d serv struct is NULL", n);
                ret = 1;
                return;
        }
        if (s->s_port != ntohs(port)) {
                warnx("%d port num mismatch %d %d", n, s->s_port, port);        
                ret = 1;
        }
        if (strcmp(s->s_name, name) != 0) {
                warnx("%d serv name mismatch %s %s", n, s->s_name, name);
                ret = 1;
        }
        if (strcmp(s->s_proto, proto) != 0) {
                warnx("%d serv proto mismatch %s %s", n, s->s_proto, proto);
                ret = 1;
        }
        va_start(va, name);
        a = va_arg(va, char *);
        i = 0;
        while (a != NULL) {
                if (strcmp(s->s_aliases[i], a) != 0) {
                        warnx("%d serv alias mismatch %s %s", n,
                            s->s_aliases[i], a);
                        ret = 1;
                }
                i++;
                a = va_arg(va, char *);
        }
        if (s->s_aliases[i] != NULL) {
                        warnx("%d serv alias list does not end with NULL", n);
                        ret = 1;
        }
        va_end(va);
}

int
main()
{
        struct protoent *p;
        struct protoent protoent;
        struct protoent_data protoent_data;
        struct servent *s;
        struct servent servent;
        struct servent_data servent_data;
        int r;

        p = getprotobynumber(35);
        checkp(1, p, 35, "idpr", "IDPR", (char *)NULL);

        p = getprotobyname("igp");
        checkp(2, p, 9, "igp", "IGP", (char *) NULL);

        p = getprotobyname("RDP");
        checkp(3, p, 27, "rdp", "RDP", (char *) NULL);

        p = getprotobyname("vrrp");
        checkp(4, p, 112, "carp", "CARP", "vrrp", (char *) NULL);

        p = getprotobyname("nonexistent");
        if (p != NULL)
                err(1, "nonexistent proto found");

        memset(&protoent_data, 0, sizeof(protoent_data));
        r = getprotobynumber_r(35, &protoent, &protoent_data);
        if (r != 0) 
                err(1, "proto not found");

        checkp(5, &protoent, 35, "idpr", "IDPR", (char *)NULL);

        r = getprotobyname_r("vrrp", &protoent, &protoent_data);
        if (r != 0) 
                err(1, "proto not found");
        checkp(6, &protoent, 112, "carp", "CARP", "vrrp", (char *) NULL);

        r = getprotobyname_r("nonexistent", &protoent, &protoent_data);
        if (r != -1)
                err(1, "nonexistent proto found");

        r = getprotobyname_r("", &protoent, &protoent_data);
        if (r != -1)
                err(1, "nonexistent proto found");


        r = getprotobynumber_r(256, &protoent, &protoent_data);
        if (r != -1)
                err(1, "nonexistent proto found");

        s = getservbyname("zip", NULL);
        checks(7, s, 6, "ddp", "zip", (char *)NULL);

        s = getservbyname("nicname", "tcp");
        checks(8, s, 43, "tcp", "whois", "nicname", (char *)NULL);

        s = getservbyport(htons(42), "tcp");
        checks(9, s, 42, "tcp", "nameserver", "name", (char *)NULL);

        memset(&servent_data, 0, sizeof(servent_data));
        r = getservbyname_r("zip", "ddp", &servent, &servent_data);
        if (r != 0) 
                err(1, "servent not found");
        checks(10, &servent, 6, "ddp", "zip", (char *)NULL);

        r = getservbyport_r(htons(520), NULL, &servent, &servent_data);
        if (r != 0) 
                err(1, "servent not found");
        checks(11, &servent, 520, "udp", "route", "router", "routed", (char *)NULL);

        r = getservbyname_r("nonexistent", NULL, &servent, &servent_data);
        if (r != -1) 
                err(1, "nonexiststent servent found");

        r = getservbyport_r(htons(50000), NULL, &servent, &servent_data);
        if (r != -1) 
                err(1, "nonexistent servent found");

        r = getservbyport_r(htons(520), "tcp", &servent, &servent_data);
        if (r != -1) 
                err(1, "nonexistent servent found");

        return ret;
}