/* This file is part of the Hesiod library.
 *
 *	$Source: /afs/sipb.mit.edu/project/sipb-athena/repository/src/hesiod/hestest.c,v $
 *	$Author: svalente $
 *	$Athena: hestest.c,v 1.5 88/08/07 22:00:44 treese Locked $
 *	$Log: hestest.c,v $
 *	Revision 1.3  1997/01/12 05:41:15  svalente
 *	Ported to HP/UX.
 *
 *	Revision 1.2  1996/06/01 18:46:32  ghudson
 *	Add reentrant interfaces.
 *
 *	Revision 1.1  1996/03/12 18:34:12  ghudson
 *	Initial revision
 *
 * Revision 1.1  1995/03/18  07:04:51  ghudson
 * Initial revision
 *
 *
 * Copyright 1988 by the Massachusetts Institute of Technology.  See the
 * file <mit-copyright.h> for copying and distribution information.
 */

#include "mit-copyright.h"

#ifndef lint
static char rcsid_hestest_c[] = "$Header: /afs/sipb.mit.edu/project/sipb-athena/repository/src/hesiod/hestest.c,v 1.3 1997/01/12 05:41:15 svalente Exp $";
#endif

#include <sys/types.h>
#include <stdio.h>
#include <string.h>
#include <netdb.h>
#include <pwd.h>
#include <netinet/in.h>
#include "hesiod.h"

char *word_end(char *s)
{
    while (*s && !isspace(*s))
	s++;
    return s;
}

char *find_word(char *s)
{
    while (isspace(*s))
	s++;
    return s;
}

char *get_word(char *p, char *buf)
{
    char *q = word_end(p);

    strncpy(buf, p, q - p);
    buf[q - p] = 0;
    return q;
}

char *get_field(char *p, int delim, char *buf)
{
    char *q = strchr(p, delim);

    if (q) {
	strncpy(buf, p, q - p);
	buf[q - p] = 0;
	return q + 1;
    } else {
	strcpy(buf, p);
	return NULL;
    }
}

int compare_vector(char **vector, char *spec)
{
    char field[100];

    for (; *vector; vector++) {
	spec = get_field(spec, '\\', field);
	if ((!spec && vector[1]) || strcmp(*vector, field) != 0)
	    return -1;
    }
    return (spec) ? -1 : 0;
}

int compare_pwnam(struct passwd *pw, char *spec)
{
    char field[100];

    spec = get_field(spec, ':', field);
    if (!spec || strcmp(pw->pw_name, field) != 0)
	return -1;
    spec = get_field(spec, ':', field);
    if (!spec || strcmp(pw->pw_passwd, field) != 0)
	return -1;
    spec = get_field(spec, ':', field);
    if (pw->pw_uid != atoi(field))
	return -1;
    spec = get_field(spec, ':', field);
    if (pw->pw_gid != atoi(field))
	return -1;
    spec = get_field(spec, ':', field);
    if (!spec || strcmp(pw->pw_gecos, field) != 0)
	return -1;
    spec = get_field(spec, ':', field);
    if (!spec || strcmp(pw->pw_dir, field) != 0)
	return -1;
    spec = get_field(spec, ':', field);
    if (spec || strcmp(pw->pw_shell, field) != 0)
	return -1;
    return 0;
}

int compare_serv(struct servent *serv, char *spec)
{
    char field[100], **aliases;

    spec = get_field(spec, ':', field);
    if (!spec || strcmp(serv->s_name, field) != 0)
	return -1;
    spec = get_field(spec, ':', field);
    if (!spec || strcmp(serv->s_proto, field) != 0)
	return -1;
    spec = get_field(spec, ':', field);
    if (serv->s_port != htons(atoi(field)))
	return -1;
    for (aliases = serv->s_aliases; *aliases; aliases++) {
	spec = get_field(spec, '\\', field);
	if ((!spec && aliases[1]) || strcmp(*aliases, field) != 0)
	    return -1;
    }
    return (spec) ? -1 : 0;
}

int compare_office(struct hes_postoffice *office, char *spec)
{
    char field[100];

    spec = get_field(spec, ':', field);
    if (!spec || strcmp(office->po_type, field) != 0)
	return -1;
    spec = get_field(spec, ':', field);
    if (!spec || strcmp(office->po_host, field) != 0)
	return -1;
    spec = get_field(spec, ':', field);
    if (spec || strcmp(office->po_name, field) != 0)
	return -1;
    return 0;
}

void free_ptrs(char **ptrs)
{
    for (; *ptrs; ptrs++)
	free(*ptrs);
}

int main(argc, argv)
    int argc;
    char **argv;
{
    FILE *fp;
    char buf[1000], *p, *q, name[100], type[100], *vec[100], **vecp;
    char buf2[1000];
    int line, status;
    struct passwd pw, *pwp;
    struct servent serv, *servp;
    struct hes_postoffice office, *officep;

    if (argc != 2) {
	fprintf(stderr, "Usage: %s filename\n", argv[0]);
	exit(1);
    }

    fp = fopen(argv[1], "r");
    if (!fp) {
	fprintf(stderr, "Couldn't open %s for reading.\n", argv[1]);
	exit(1);
    }

    hes_init();
    line = 0;
    while (fgets(buf, sizeof(buf), fp)) {
	line++;

	/* Strip off trailing spaces (inefficiently). */
	while (isspace(buf[strlen(buf) - 1]))
	    buf[strlen(buf) - 1] = 0;

	/* Get the first word, discard comment lines and invalid lines. */
	q = word_end(p = find_word(buf));
	if (*p == '#' || !*p)
	    continue;
	if (!*q) {
	    fprintf(stderr, "Invalid test: %s\n", buf);
	    continue;
	}

	/* Test for hes_resolve and hes_resolve_r. */
	if (q - p == 7 && strncmp(p, "resolve", 7) == 0) {
	    q = get_word(find_word(q), name);
	    q = get_word(find_word(q), type);
	    p = find_word(q);
	    status = hes_resolve_r(name, type, vec, 100);
	    if (status != HES_ER_OK && !(*p == 'E' && p[1] == 0)) {
		printf("Line %d failed (hes_resolve_r error %d).\n",
		       line, status);
		continue;
	    }
	    if (status == HES_ER_OK && compare_vector(vec, p) < 0) {
		printf("Line %d failed in hes_resolve_r().\n", line);
		free_ptrs(vec);
		continue;
	    }
	    if (status == HES_ER_OK)
		free_ptrs(vec);
	    vecp = hes_resolve(name, type);
	    if (!vecp) {
		if (*p == 'E' && p[1] == 0) {
		    printf("Line %d passed (errors %d, %d).\n",
			   line, status, hes_error());
		} else {
		    printf("Line %d failed (hes_resolve error %d).\n",
			   line, hes_error());
		}
		continue;
	    }
	    if (compare_vector(vecp, p) < 0)
		printf("Line %d failed in hes_resolve().\n", line);
	    else
		printf("Line %d passed.\n", line);
	    free_ptrs(vecp);

	/* Test for hes_getpwnam and hes_getpwnam_r. */
	} else if (q - p == 8 && strncmp(p, "getpwnam", 8) == 0) {
	    q = get_word(find_word(q), name);
	    p = find_word(q);
	    status = hes_getpwnam_r(name, &pw, buf2, sizeof(buf2));
	    if (status != HES_ER_OK && !(*p == 'E' && p[1] == 0)) {
		printf("Line %d failed (hes_getpwnam_r error %d).\n",
		       line, status);
		continue;
	    }
	    if (status == HES_ER_OK && compare_pwnam(&pw, p) < 0) {
		printf("Line %d failed in hes_getpwnam_r().\n", line);
		continue;
	    }
	    pwp = hes_getpwnam(name);
	    if (!pwp) {
		if (*p == 'E' && p[1] == 0) {
		    printf("Line %d passed (errors %d, %d).\n",
			   line, status, hes_error());
		} else {
		    printf("Line %d failed (hes_getpwnam error %d).\n",
			   line, hes_error());
		}
		continue;
	    }
	    if (compare_pwnam(pwp, p) < 0)
		printf("Line %d failed in hes_getpwnam().\n", line);
	    else
		printf("Line %d passed.\n", line);

	/* Test for hes_getpwuid and hes_getpwuid_r. */
	} else if (q - p == 8 && strncmp(p, "getpwuid", 8) == 0) {
	    q = get_word(find_word(q), name);
	    p = find_word(q);
	    status = hes_getpwuid_r(atoi(name), &pw, buf2, sizeof(buf2));
	    if (status != HES_ER_OK && !(*p == 'E' && p[1] == 0)) {
		printf("Line %d failed (hes_getpwuid_r error %d).\n",
		       line, status);
		continue;
	    }
	    if (status == HES_ER_OK && compare_pwnam(&pw, p) < 0) {
		printf("Line %d failed in hes_getpwuid_r().\n", line);
		continue;
	    }
	    pwp = hes_getpwuid(atoi(name));
	    if (!pwp) {
		if (*p == 'E' && p[1] == 0) {
		    printf("Line %d passed (errors %d, %d).\n",
			   line, status, hes_error());
		} else {
		    printf("Line %d failed (hes_getpwuid error %d).\n",
			   line, hes_error());
		}
		continue;
	    }
	    if (compare_pwnam(pwp, p) < 0)
		printf("Line %d failed in hes_getpwuid().\n", line);
	    else
		printf("Line %d passed.\n", line);

	/* Test for hes_getservbyname and hes_getservbyname_r. */
	} else if (q - p == 13 && strncmp(p, "getservbyname", 13) == 0) {
	    q = get_word(find_word(q), name);
	    q = get_word(find_word(q), type);
	    p = find_word(q);
	    status = hes_getservbyname_r(name, type, &serv, buf2,
					 sizeof(buf2));
	    if (status != HES_ER_OK && !(*p == 'E' && p[1] == 0)) {
		printf("Line %d failed (hes_getservbyname_r error %d).\n",
		       line, status);
		continue;
	    }
	    if (status == HES_ER_OK && compare_serv(&serv, p) < 0) {
		printf("Line %d failed in hes_getservbyname_r.\n", line);
		continue;
	    }
	    servp = hes_getservbyname(name, type);
	    if (!servp) {
		if (*p == 'E' && p[1] == 0) {
		    printf("Line %d passed (errors %d, %d).\n",
			   line, status, hes_error());
		} else {
		    printf("Line %d failed (hes_getservbyname error %d).\n",
			   line, hes_error());
		}
		continue;
	    }
	    if (compare_serv(servp, p) < 0)
		printf("Line %d failed in hes_getservbyname().\n", line);
	    else
		printf("Line %d passed.\n", line);

	/* Test for hes_getmailhost and hes_getmailhost_r. */
	} else if (q - p == 11 && strncmp(p, "getmailhost", 11) == 0) {
	    q = get_word(find_word(q), name);
	    p = find_word(q);
	    status = hes_getmailhost_r(name, &office, buf2, sizeof(buf2));
	    if (status != HES_ER_OK && !(*p == 'E' && p[1] == 0)) {
		printf("Line %d failed (hes_getmailhost_r error %d).\n",
		       line, status);
		continue;
	    }
	    if (status == HES_ER_OK && compare_office(&office, p) < 0) {
		printf("Line %d failed in hes_getmailhost_r.\n", line);
		continue;
	    }
	    officep = hes_getmailhost(name);
	    if (!officep) {
		if (*p == 'E' && p[1] == 0) {
		    printf("Line %d passed (errors %d, %d).\n",
			   line, status, hes_error());
		} else {
		    printf("Line %d failed (hes_getmailhost error %d).\n",
			   line, hes_error());
		}
		continue;
	    }
	    if (compare_office(officep, p) < 0)
		printf("Line %d failed in hes_getmailhost().\n", line);
	    else
		printf("Line %d passed.\n", line);

	} else {
		printf("Line %d invalid.\n", line);
	}
    }

    return 0;
}