/*	$OpenBSD: globtest.c,v 1.4 2019/01/25 00:19:26 millert Exp $	*/

/*
 * Public domain, 2008, Todd C. Miller <millert@openbsd.org>
 */

#include <err.h>
#include <glob.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define MAX_RESULTS	256

struct gl_entry {
	int flags;
	int nresults;
	char pattern[1024];
	char *results[MAX_RESULTS];
	mode_t modes[MAX_RESULTS];
};

int test_glob(struct gl_entry *);

int
main(int argc, char **argv)
{
	FILE *fp = stdin;
	char *buf, *cp;
	int errors = 0, lineno, mode;
	struct gl_entry entry;
	size_t len;

	if (argc > 1) {
		if ((fp = fopen(argv[1], "r")) == NULL)
			err(1, "%s", argv[1]);
	}

	/*
	 * Read in test file, which is formatted thusly:
	 *
	 * [pattern] <flags>
	 * result1 [mode]
	 * result2 [mode]
	 * result3 [mode]
	 * ...
	 *
	 */
	lineno = 0;
	memset(&entry, 0, sizeof(entry));
	while ((buf = fgetln(fp, &len)) != NULL) {
		lineno++;
		if (buf[len - 1] != '\n')
			errx(1, "missing newline at EOF");
		buf[--len] = '\0';
		if (len == 0)
			continue; /* blank line */

		if (buf[0] == '[') {
			/* check previous pattern */
			if (entry.pattern[0])
				errors += test_glob(&entry);

			/* start new entry */
			if ((cp = strrchr(buf + 1, ']')) == NULL)
				errx(1, "invalid entry on line %d", lineno);
			len = cp - buf - 1;
			if (len >= sizeof(entry.pattern))
				errx(1, "pattern too big on line %d", lineno);
			memcpy(entry.pattern, buf + 1, len);
			entry.pattern[len] = '\0';

			buf = cp + 2;
			if (*buf++ != '<')
				errx(1, "invalid entry on line %d", lineno);
			if ((cp = strchr(buf, '>')) == NULL)
				errx(1, "invalid entry on line %d", lineno);
			entry.flags = (int)strtol(buf, &cp, 0);
			if (*cp != '>' || entry.flags < 0 || entry.flags > 0x4000)
				errx(1, "invalid flags: %s", buf);
			entry.nresults = 0;
			continue;
		}
		if (!entry.pattern[0])
			errx(1, "missing entry on line %d", lineno);

		if (entry.nresults + 1 > MAX_RESULTS) {
			errx(1, "too many results for %s, max %d",
			    entry.pattern, MAX_RESULTS);
		}
		if ((cp = strchr(buf, ' ')) != NULL) {
			*cp++ = '\0';
			mode = strtol(cp, NULL, 8);
		} else
			mode = -1;
		entry.modes[entry.nresults] = (mode_t)mode;
		entry.results[entry.nresults++] = strdup(buf);
	}
	if (entry.pattern[0])
		errors += test_glob(&entry); /* test last pattern */
	exit(errors);
}

int test_glob(struct gl_entry *entry)
{
	glob_t gl;
	int i = 0;

	if (glob(entry->pattern, entry->flags, NULL, &gl) != 0)
		errx(1, "glob failed: %s", entry->pattern);

	if (gl.gl_matchc != entry->nresults)
		goto mismatch;

	for (i = 0; i < gl.gl_matchc; i++) {
		if (strcmp(gl.gl_pathv[i], entry->results[i]) != 0)
			goto mismatch;
		if ((entry->flags & GLOB_KEEPSTAT) != 0) {
			if (entry->modes[i] == -1 ||
			    gl.gl_statv[i] == NULL ||
			    entry->modes[i] != gl.gl_statv[i]->st_mode)
			goto badmode;
		}
		free(entry->results[i]);
	}
	return (0);
 badmode:
	warnx("mismatch mode for pattern %s, flags 0x%x, file \"%s\" "
	    "(found %07o, expected %07o)", entry->pattern, entry->flags,
	    gl.gl_pathv[i], gl.gl_statv[i] ? gl.gl_statv[i]->st_mode : 0,
	    entry->modes[i]);
	goto cleanup;
 mismatch:
	warnx("mismatch for pattern %s, flags 0x%x "
	    "(found \"%s\", expected \"%s\")", entry->pattern, entry->flags,
	    gl.gl_pathv[i], entry->results[i]);
 cleanup:
	while (i < gl.gl_matchc) {
		free(entry->results[i++]);
	}
	return (1);
}