main.c revision 307138
1/*
2 * Copyright (c) 2003-2009 Tim Kientzle
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
15 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
16 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
17 * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
18 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
19 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
20 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
21 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
23 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#include "test.h"
27#include "test_utils.h"
28#ifdef HAVE_SYS_IOCTL_H
29#include <sys/ioctl.h>
30#endif
31#ifdef HAVE_SYS_TIME_H
32#include <sys/time.h>
33#endif
34#include <errno.h>
35#ifdef HAVE_ICONV_H
36#include <iconv.h>
37#endif
38/*
39 * Some Linux distributions have both linux/ext2_fs.h and ext2fs/ext2_fs.h.
40 * As the include guards don't agree, the order of include is important.
41 */
42#ifdef HAVE_LINUX_EXT2_FS_H
43#include <linux/ext2_fs.h>      /* for Linux file flags */
44#endif
45#if defined(HAVE_EXT2FS_EXT2_FS_H) && !defined(__CYGWIN__)
46#include <ext2fs/ext2_fs.h>     /* Linux file flags, broken on Cygwin */
47#endif
48#include <limits.h>
49#include <locale.h>
50#ifdef HAVE_SIGNAL_H
51#include <signal.h>
52#endif
53#include <stdarg.h>
54#include <time.h>
55
56/*
57 * This same file is used pretty much verbatim for all test harnesses.
58 *
59 * The next few lines are the only differences.
60 * TODO: Move this into a separate configuration header, have all test
61 * suites share one copy of this file.
62 */
63__FBSDID("$FreeBSD: stable/11/contrib/libarchive/libarchive/test/main.c 307138 2016-10-12 10:28:22Z mm $");
64#define KNOWNREF	"test_compat_gtar_1.tar.uu"
65#define	ENVBASE "LIBARCHIVE" /* Prefix for environment variables. */
66#undef	PROGRAM              /* Testing a library, not a program. */
67#define	LIBRARY	"libarchive"
68#define	EXTRA_DUMP(x)	archive_error_string((struct archive *)(x))
69#define	EXTRA_ERRNO(x)	archive_errno((struct archive *)(x))
70#define	EXTRA_VERSION	archive_version_details()
71
72/*
73 *
74 * Windows support routines
75 *
76 * Note: Configuration is a tricky issue.  Using HAVE_* feature macros
77 * in the test harness is dangerous because they cover up
78 * configuration errors.  The classic example of this is omitting a
79 * configure check.  If libarchive and libarchive_test both look for
80 * the same feature macro, such errors are hard to detect.  Platform
81 * macros (e.g., _WIN32 or __GNUC__) are a little better, but can
82 * easily lead to very messy code.  It's best to limit yourself
83 * to only the most generic programming techniques in the test harness
84 * and thus avoid conditionals altogether.  Where that's not possible,
85 * try to minimize conditionals by grouping platform-specific tests in
86 * one place (e.g., test_acl_freebsd) or by adding new assert()
87 * functions (e.g., assertMakeHardlink()) to cover up platform
88 * differences.  Platform-specific coding in libarchive_test is often
89 * a symptom that some capability is missing from libarchive itself.
90 */
91#if defined(_WIN32) && !defined(__CYGWIN__)
92#include <io.h>
93#include <direct.h>
94#include <windows.h>
95#ifndef F_OK
96#define F_OK (0)
97#endif
98#ifndef S_ISDIR
99#define S_ISDIR(m)  ((m) & _S_IFDIR)
100#endif
101#ifndef S_ISREG
102#define S_ISREG(m)  ((m) & _S_IFREG)
103#endif
104#if !defined(__BORLANDC__)
105#define access _access
106#undef chdir
107#define chdir _chdir
108#endif
109#ifndef fileno
110#define fileno _fileno
111#endif
112/*#define fstat _fstat64*/
113#if !defined(__BORLANDC__)
114#define getcwd _getcwd
115#endif
116#define lstat stat
117/*#define lstat _stat64*/
118/*#define stat _stat64*/
119#define rmdir _rmdir
120#if !defined(__BORLANDC__)
121#define strdup _strdup
122#define umask _umask
123#endif
124#define int64_t __int64
125#endif
126
127#if defined(HAVE__CrtSetReportMode)
128# include <crtdbg.h>
129#endif
130
131/* Path to working directory for current test */
132const char *testworkdir;
133#ifdef PROGRAM
134/* Pathname of exe to be tested. */
135const char *testprogfile;
136/* Name of exe to use in printf-formatted command strings. */
137/* On Windows, this includes leading/trailing quotes. */
138const char *testprog;
139#endif
140
141#if defined(_WIN32) && !defined(__CYGWIN__)
142static void	*GetFunctionKernel32(const char *);
143static int	 my_CreateSymbolicLinkA(const char *, const char *, int);
144static int	 my_CreateHardLinkA(const char *, const char *);
145static int	 my_GetFileInformationByName(const char *,
146		     BY_HANDLE_FILE_INFORMATION *);
147
148static void *
149GetFunctionKernel32(const char *name)
150{
151	static HINSTANCE lib;
152	static int set;
153	if (!set) {
154		set = 1;
155		lib = LoadLibrary("kernel32.dll");
156	}
157	if (lib == NULL) {
158		fprintf(stderr, "Can't load kernel32.dll?!\n");
159		exit(1);
160	}
161	return (void *)GetProcAddress(lib, name);
162}
163
164static int
165my_CreateSymbolicLinkA(const char *linkname, const char *target, int flags)
166{
167	static BOOLEAN (WINAPI *f)(LPCSTR, LPCSTR, DWORD);
168	static int set;
169	if (!set) {
170		set = 1;
171		f = GetFunctionKernel32("CreateSymbolicLinkA");
172	}
173	return f == NULL ? 0 : (*f)(linkname, target, flags);
174}
175
176static int
177my_CreateHardLinkA(const char *linkname, const char *target)
178{
179	static BOOLEAN (WINAPI *f)(LPCSTR, LPCSTR, LPSECURITY_ATTRIBUTES);
180	static int set;
181	if (!set) {
182		set = 1;
183		f = GetFunctionKernel32("CreateHardLinkA");
184	}
185	return f == NULL ? 0 : (*f)(linkname, target, NULL);
186}
187
188static int
189my_GetFileInformationByName(const char *path, BY_HANDLE_FILE_INFORMATION *bhfi)
190{
191	HANDLE h;
192	int r;
193
194	memset(bhfi, 0, sizeof(*bhfi));
195	h = CreateFile(path, FILE_READ_ATTRIBUTES, 0, NULL,
196		OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
197	if (h == INVALID_HANDLE_VALUE)
198		return (0);
199	r = GetFileInformationByHandle(h, bhfi);
200	CloseHandle(h);
201	return (r);
202}
203#endif
204
205#if defined(HAVE__CrtSetReportMode) && !defined(__WATCOMC__)
206static void
207invalid_parameter_handler(const wchar_t * expression,
208    const wchar_t * function, const wchar_t * file,
209    unsigned int line, uintptr_t pReserved)
210{
211	/* nop */
212}
213#endif
214
215/*
216 *
217 * OPTIONS FLAGS
218 *
219 */
220
221/* Enable core dump on failure. */
222static int dump_on_failure = 0;
223/* Default is to remove temp dirs and log data for successful tests. */
224static int keep_temp_files = 0;
225/* Default is to run the specified tests once and report errors. */
226static int until_failure = 0;
227/* Default is to just report pass/fail for each test. */
228static int verbosity = 0;
229#define	VERBOSITY_SUMMARY_ONLY -1 /* -q */
230#define VERBOSITY_PASSFAIL 0   /* Default */
231#define VERBOSITY_LIGHT_REPORT 1 /* -v */
232#define VERBOSITY_FULL 2 /* -vv */
233/* A few places generate even more output for verbosity > VERBOSITY_FULL,
234 * mostly for debugging the test harness itself. */
235/* Cumulative count of assertion failures. */
236static int failures = 0;
237/* Cumulative count of reported skips. */
238static int skips = 0;
239/* Cumulative count of assertions checked. */
240static int assertions = 0;
241
242/* Directory where uuencoded reference files can be found. */
243static const char *refdir;
244
245/*
246 * Report log information selectively to console and/or disk log.
247 */
248static int log_console = 0;
249static FILE *logfile;
250static void
251vlogprintf(const char *fmt, va_list ap)
252{
253#ifdef va_copy
254	va_list lfap;
255	va_copy(lfap, ap);
256#endif
257	if (log_console)
258		vfprintf(stdout, fmt, ap);
259	if (logfile != NULL)
260#ifdef va_copy
261		vfprintf(logfile, fmt, lfap);
262	va_end(lfap);
263#else
264		vfprintf(logfile, fmt, ap);
265#endif
266}
267
268static void
269logprintf(const char *fmt, ...)
270{
271	va_list ap;
272	va_start(ap, fmt);
273	vlogprintf(fmt, ap);
274	va_end(ap);
275}
276
277/* Set up a message to display only if next assertion fails. */
278static char msgbuff[4096];
279static const char *msg, *nextmsg;
280void
281failure(const char *fmt, ...)
282{
283	va_list ap;
284	if (fmt == NULL) {
285		nextmsg = NULL;
286	} else {
287		va_start(ap, fmt);
288		vsprintf(msgbuff, fmt, ap);
289		va_end(ap);
290		nextmsg = msgbuff;
291	}
292}
293
294/*
295 * Copy arguments into file-local variables.
296 * This was added to permit vararg assert() functions without needing
297 * variadic wrapper macros.  Turns out that the vararg capability is almost
298 * never used, so almost all of the vararg assertions can be simplified
299 * by removing the vararg capability and reworking the wrapper macro to
300 * pass __FILE__, __LINE__ directly into the function instead of using
301 * this hook.  I suspect this machinery is used so rarely that we
302 * would be better off just removing it entirely.  That would simplify
303 * the code here noticeably.
304 */
305static const char *skipping_filename;
306static int skipping_line;
307void skipping_setup(const char *filename, int line)
308{
309	skipping_filename = filename;
310	skipping_line = line;
311}
312
313/* Called at the beginning of each assert() function. */
314static void
315assertion_count(const char *file, int line)
316{
317	(void)file; /* UNUSED */
318	(void)line; /* UNUSED */
319	++assertions;
320	/* Proper handling of "failure()" message. */
321	msg = nextmsg;
322	nextmsg = NULL;
323	/* Uncomment to print file:line after every assertion.
324	 * Verbose, but occasionally useful in tracking down crashes. */
325	/* printf("Checked %s:%d\n", file, line); */
326}
327
328/*
329 * For each test source file, we remember how many times each
330 * assertion was reported.  Cleared before each new test,
331 * used by test_summarize().
332 */
333static struct line {
334	int count;
335	int skip;
336}  failed_lines[10000];
337const char *failed_filename;
338
339/* Count this failure, setup up log destination and handle initial report. */
340static void
341failure_start(const char *filename, int line, const char *fmt, ...)
342{
343	va_list ap;
344
345	/* Record another failure for this line. */
346	++failures;
347	failed_filename = filename;
348	failed_lines[line].count++;
349
350	/* Determine whether to log header to console. */
351	switch (verbosity) {
352	case VERBOSITY_LIGHT_REPORT:
353		log_console = (failed_lines[line].count < 2);
354		break;
355	default:
356		log_console = (verbosity >= VERBOSITY_FULL);
357	}
358
359	/* Log file:line header for this failure */
360	va_start(ap, fmt);
361#if _MSC_VER
362	logprintf("%s(%d): ", filename, line);
363#else
364	logprintf("%s:%d: ", filename, line);
365#endif
366	vlogprintf(fmt, ap);
367	va_end(ap);
368	logprintf("\n");
369
370	if (msg != NULL && msg[0] != '\0') {
371		logprintf("   Description: %s\n", msg);
372		msg = NULL;
373	}
374
375	/* Determine whether to log details to console. */
376	if (verbosity == VERBOSITY_LIGHT_REPORT)
377		log_console = 0;
378}
379
380/* Complete reporting of failed tests. */
381/*
382 * The 'extra' hook here is used by libarchive to include libarchive
383 * error messages with assertion failures.  It could also be used
384 * to add strerror() output, for example.  Just define the EXTRA_DUMP()
385 * macro appropriately.
386 */
387static void
388failure_finish(void *extra)
389{
390	(void)extra; /* UNUSED (maybe) */
391#ifdef EXTRA_DUMP
392	if (extra != NULL) {
393		logprintf("    errno: %d\n", EXTRA_ERRNO(extra));
394		logprintf("   detail: %s\n", EXTRA_DUMP(extra));
395	}
396#endif
397
398	if (dump_on_failure) {
399		fprintf(stderr,
400		    " *** forcing core dump so failure can be debugged ***\n");
401		abort();
402	}
403}
404
405/* Inform user that we're skipping some checks. */
406void
407test_skipping(const char *fmt, ...)
408{
409	char buff[1024];
410	va_list ap;
411
412	va_start(ap, fmt);
413	vsprintf(buff, fmt, ap);
414	va_end(ap);
415	/* Use failure() message if set. */
416	msg = nextmsg;
417	nextmsg = NULL;
418	/* failure_start() isn't quite right, but is awfully convenient. */
419	failure_start(skipping_filename, skipping_line, "SKIPPING: %s", buff);
420	--failures; /* Undo failures++ in failure_start() */
421	/* Don't failure_finish() here. */
422	/* Mark as skip, so doesn't count as failed test. */
423	failed_lines[skipping_line].skip = 1;
424	++skips;
425}
426
427/*
428 *
429 * ASSERTIONS
430 *
431 */
432
433/* Generic assert() just displays the failed condition. */
434int
435assertion_assert(const char *file, int line, int value,
436    const char *condition, void *extra)
437{
438	assertion_count(file, line);
439	if (!value) {
440		failure_start(file, line, "Assertion failed: %s", condition);
441		failure_finish(extra);
442	}
443	return (value);
444}
445
446/* chdir() and report any errors */
447int
448assertion_chdir(const char *file, int line, const char *pathname)
449{
450	assertion_count(file, line);
451	if (chdir(pathname) == 0)
452		return (1);
453	failure_start(file, line, "chdir(\"%s\")", pathname);
454	failure_finish(NULL);
455	return (0);
456
457}
458
459/* Verify two integers are equal. */
460int
461assertion_equal_int(const char *file, int line,
462    long long v1, const char *e1, long long v2, const char *e2, void *extra)
463{
464	assertion_count(file, line);
465	if (v1 == v2)
466		return (1);
467	failure_start(file, line, "%s != %s", e1, e2);
468	logprintf("      %s=%lld (0x%llx, 0%llo)\n", e1, v1, v1, v1);
469	logprintf("      %s=%lld (0x%llx, 0%llo)\n", e2, v2, v2, v2);
470	failure_finish(extra);
471	return (0);
472}
473
474/*
475 * Utility to convert a single UTF-8 sequence.
476 */
477static int
478_utf8_to_unicode(uint32_t *pwc, const char *s, size_t n)
479{
480	static const char utf8_count[256] = {
481		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 00 - 0F */
482		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 10 - 1F */
483		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 20 - 2F */
484		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 30 - 3F */
485		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 40 - 4F */
486		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 50 - 5F */
487		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 60 - 6F */
488		 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 70 - 7F */
489		 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,/* 80 - 8F */
490		 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,/* 90 - 9F */
491		 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,/* A0 - AF */
492		 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,/* B0 - BF */
493		 0, 0, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,/* C0 - CF */
494		 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,/* D0 - DF */
495		 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,/* E0 - EF */
496		 4, 4, 4, 4, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 /* F0 - FF */
497	};
498	int ch;
499	int cnt;
500	uint32_t wc;
501
502	*pwc = 0;
503
504	/* Sanity check. */
505	if (n == 0)
506		return (0);
507	/*
508	 * Decode 1-4 bytes depending on the value of the first byte.
509	 */
510	ch = (unsigned char)*s;
511	if (ch == 0)
512		return (0); /* Standard:  return 0 for end-of-string. */
513	cnt = utf8_count[ch];
514
515	/* Invalide sequence or there are not plenty bytes. */
516	if (n < (size_t)cnt)
517		return (-1);
518
519	/* Make a Unicode code point from a single UTF-8 sequence. */
520	switch (cnt) {
521	case 1:	/* 1 byte sequence. */
522		*pwc = ch & 0x7f;
523		return (cnt);
524	case 2:	/* 2 bytes sequence. */
525		if ((s[1] & 0xc0) != 0x80) return (-1);
526		*pwc = ((ch & 0x1f) << 6) | (s[1] & 0x3f);
527		return (cnt);
528	case 3:	/* 3 bytes sequence. */
529		if ((s[1] & 0xc0) != 0x80) return (-1);
530		if ((s[2] & 0xc0) != 0x80) return (-1);
531		wc = ((ch & 0x0f) << 12)
532		    | ((s[1] & 0x3f) << 6)
533		    | (s[2] & 0x3f);
534		if (wc < 0x800)
535			return (-1);/* Overlong sequence. */
536		break;
537	case 4:	/* 4 bytes sequence. */
538		if (n < 4)
539			return (-1);
540		if ((s[1] & 0xc0) != 0x80) return (-1);
541		if ((s[2] & 0xc0) != 0x80) return (-1);
542		if ((s[3] & 0xc0) != 0x80) return (-1);
543		wc = ((ch & 0x07) << 18)
544		    | ((s[1] & 0x3f) << 12)
545		    | ((s[2] & 0x3f) << 6)
546		    | (s[3] & 0x3f);
547		if (wc < 0x10000)
548			return (-1);/* Overlong sequence. */
549		break;
550	default:
551		return (-1);
552	}
553
554	/* The code point larger than 0x10FFFF is not leagal
555	 * Unicode values. */
556	if (wc > 0x10FFFF)
557		return (-1);
558	/* Correctly gets a Unicode, returns used bytes. */
559	*pwc = wc;
560	return (cnt);
561}
562
563static void strdump(const char *e, const char *p, int ewidth, int utf8)
564{
565	const char *q = p;
566
567	logprintf("      %*s = ", ewidth, e);
568	if (p == NULL) {
569		logprintf("NULL\n");
570		return;
571	}
572	logprintf("\"");
573	while (*p != '\0') {
574		unsigned int c = 0xff & *p++;
575		switch (c) {
576		case '\a': logprintf("\\a"); break;
577		case '\b': logprintf("\\b"); break;
578		case '\n': logprintf("\\n"); break;
579		case '\r': logprintf("\\r"); break;
580		default:
581			if (c >= 32 && c < 127)
582				logprintf("%c", c);
583			else
584				logprintf("\\x%02X", c);
585		}
586	}
587	logprintf("\"");
588	logprintf(" (length %d)", q == NULL ? -1 : (int)strlen(q));
589
590	/*
591	 * If the current string is UTF-8, dump its code points.
592	 */
593	if (utf8) {
594		size_t len;
595		uint32_t uc;
596		int n;
597		int cnt = 0;
598
599		p = q;
600		len = strlen(p);
601		logprintf(" [");
602		while ((n = _utf8_to_unicode(&uc, p, len)) > 0) {
603			if (p != q)
604				logprintf(" ");
605			logprintf("%04X", uc);
606			p += n;
607			len -= n;
608			cnt++;
609		}
610		logprintf("]");
611		logprintf(" (count %d", cnt);
612		if (n < 0) {
613			logprintf(",unknown %d bytes", len);
614		}
615		logprintf(")");
616
617	}
618	logprintf("\n");
619}
620
621/* Verify two strings are equal, dump them if not. */
622int
623assertion_equal_string(const char *file, int line,
624    const char *v1, const char *e1,
625    const char *v2, const char *e2,
626    void *extra, int utf8)
627{
628	int l1, l2;
629
630	assertion_count(file, line);
631	if (v1 == v2 || (v1 != NULL && v2 != NULL && strcmp(v1, v2) == 0))
632		return (1);
633	failure_start(file, line, "%s != %s", e1, e2);
634	l1 = (int)strlen(e1);
635	l2 = (int)strlen(e2);
636	if (l1 < l2)
637		l1 = l2;
638	strdump(e1, v1, l1, utf8);
639	strdump(e2, v2, l1, utf8);
640	failure_finish(extra);
641	return (0);
642}
643
644static void
645wcsdump(const char *e, const wchar_t *w)
646{
647	logprintf("      %s = ", e);
648	if (w == NULL) {
649		logprintf("(null)");
650		return;
651	}
652	logprintf("\"");
653	while (*w != L'\0') {
654		unsigned int c = *w++;
655		if (c >= 32 && c < 127)
656			logprintf("%c", c);
657		else if (c < 256)
658			logprintf("\\x%02X", c);
659		else if (c < 0x10000)
660			logprintf("\\u%04X", c);
661		else
662			logprintf("\\U%08X", c);
663	}
664	logprintf("\"\n");
665}
666
667#ifndef HAVE_WCSCMP
668static int
669wcscmp(const wchar_t *s1, const wchar_t *s2)
670{
671
672	while (*s1 == *s2++) {
673		if (*s1++ == L'\0')
674			return 0;
675	}
676	if (*s1 > *--s2)
677		return 1;
678	else
679		return -1;
680}
681#endif
682
683/* Verify that two wide strings are equal, dump them if not. */
684int
685assertion_equal_wstring(const char *file, int line,
686    const wchar_t *v1, const char *e1,
687    const wchar_t *v2, const char *e2,
688    void *extra)
689{
690	assertion_count(file, line);
691	if (v1 == v2)
692		return (1);
693	if (v1 != NULL && v2 != NULL && wcscmp(v1, v2) == 0)
694		return (1);
695	failure_start(file, line, "%s != %s", e1, e2);
696	wcsdump(e1, v1);
697	wcsdump(e2, v2);
698	failure_finish(extra);
699	return (0);
700}
701
702/*
703 * Pretty standard hexdump routine.  As a bonus, if ref != NULL, then
704 * any bytes in p that differ from ref will be highlighted with '_'
705 * before and after the hex value.
706 */
707static void
708hexdump(const char *p, const char *ref, size_t l, size_t offset)
709{
710	size_t i, j;
711	char sep;
712
713	if (p == NULL) {
714		logprintf("(null)\n");
715		return;
716	}
717	for(i=0; i < l; i+=16) {
718		logprintf("%04x", (unsigned)(i + offset));
719		sep = ' ';
720		for (j = 0; j < 16 && i + j < l; j++) {
721			if (ref != NULL && p[i + j] != ref[i + j])
722				sep = '_';
723			logprintf("%c%02x", sep, 0xff & (int)p[i+j]);
724			if (ref != NULL && p[i + j] == ref[i + j])
725				sep = ' ';
726		}
727		for (; j < 16; j++) {
728			logprintf("%c  ", sep);
729			sep = ' ';
730		}
731		logprintf("%c", sep);
732		for (j=0; j < 16 && i + j < l; j++) {
733			int c = p[i + j];
734			if (c >= ' ' && c <= 126)
735				logprintf("%c", c);
736			else
737				logprintf(".");
738		}
739		logprintf("\n");
740	}
741}
742
743/* Verify that two blocks of memory are the same, display the first
744 * block of differences if they're not. */
745int
746assertion_equal_mem(const char *file, int line,
747    const void *_v1, const char *e1,
748    const void *_v2, const char *e2,
749    size_t l, const char *ld, void *extra)
750{
751	const char *v1 = (const char *)_v1;
752	const char *v2 = (const char *)_v2;
753	size_t offset;
754
755	assertion_count(file, line);
756	if (v1 == v2 || (v1 != NULL && v2 != NULL && memcmp(v1, v2, l) == 0))
757		return (1);
758	if (v1 == NULL || v2 == NULL)
759		return (0);
760
761	failure_start(file, line, "%s != %s", e1, e2);
762	logprintf("      size %s = %d\n", ld, (int)l);
763	/* Dump 48 bytes (3 lines) so that the first difference is
764	 * in the second line. */
765	offset = 0;
766	while (l > 64 && memcmp(v1, v2, 32) == 0) {
767		/* Two lines agree, so step forward one line. */
768		v1 += 16;
769		v2 += 16;
770		l -= 16;
771		offset += 16;
772	}
773	logprintf("      Dump of %s\n", e1);
774	hexdump(v1, v2, l < 128 ? l : 128, offset);
775	logprintf("      Dump of %s\n", e2);
776	hexdump(v2, v1, l < 128 ? l : 128, offset);
777	logprintf("\n");
778	failure_finish(extra);
779	return (0);
780}
781
782/* Verify that a block of memory is filled with the specified byte. */
783int
784assertion_memory_filled_with(const char *file, int line,
785    const void *_v1, const char *vd,
786    size_t l, const char *ld,
787    char b, const char *bd, void *extra)
788{
789	const char *v1 = (const char *)_v1;
790	size_t c = 0;
791	size_t i;
792	(void)ld; /* UNUSED */
793
794	assertion_count(file, line);
795
796	for (i = 0; i < l; ++i) {
797		if (v1[i] == b) {
798			++c;
799		}
800	}
801	if (c == l)
802		return (1);
803
804	failure_start(file, line, "%s (size %d) not filled with %s", vd, (int)l, bd);
805	logprintf("   Only %d bytes were correct\n", (int)c);
806	failure_finish(extra);
807	return (0);
808}
809
810/* Verify that the named file exists and is empty. */
811int
812assertion_empty_file(const char *filename, int line, const char *f1)
813{
814	char buff[1024];
815	struct stat st;
816	ssize_t s;
817	FILE *f;
818
819	assertion_count(filename, line);
820
821	if (stat(f1, &st) != 0) {
822		failure_start(filename, line, "Stat failed: %s", f1);
823		failure_finish(NULL);
824		return (0);
825	}
826	if (st.st_size == 0)
827		return (1);
828
829	failure_start(filename, line, "File should be empty: %s", f1);
830	logprintf("    File size: %d\n", (int)st.st_size);
831	logprintf("    Contents:\n");
832	f = fopen(f1, "rb");
833	if (f == NULL) {
834		logprintf("    Unable to open %s\n", f1);
835	} else {
836		s = ((off_t)sizeof(buff) < st.st_size) ?
837		    (ssize_t)sizeof(buff) : (ssize_t)st.st_size;
838		s = fread(buff, 1, s, f);
839		hexdump(buff, NULL, s, 0);
840		fclose(f);
841	}
842	failure_finish(NULL);
843	return (0);
844}
845
846/* Verify that the named file exists and is not empty. */
847int
848assertion_non_empty_file(const char *filename, int line, const char *f1)
849{
850	struct stat st;
851
852	assertion_count(filename, line);
853
854	if (stat(f1, &st) != 0) {
855		failure_start(filename, line, "Stat failed: %s", f1);
856		failure_finish(NULL);
857		return (0);
858	}
859	if (st.st_size == 0) {
860		failure_start(filename, line, "File empty: %s", f1);
861		failure_finish(NULL);
862		return (0);
863	}
864	return (1);
865}
866
867/* Verify that two files have the same contents. */
868/* TODO: hexdump the first bytes that actually differ. */
869int
870assertion_equal_file(const char *filename, int line, const char *fn1, const char *fn2)
871{
872	char buff1[1024];
873	char buff2[1024];
874	FILE *f1, *f2;
875	int n1, n2;
876
877	assertion_count(filename, line);
878
879	f1 = fopen(fn1, "rb");
880	f2 = fopen(fn2, "rb");
881	if (f1 == NULL || f2 == NULL) {
882		if (f1) fclose(f1);
883		if (f2) fclose(f2);
884		return (0);
885	}
886	for (;;) {
887		n1 = (int)fread(buff1, 1, sizeof(buff1), f1);
888		n2 = (int)fread(buff2, 1, sizeof(buff2), f2);
889		if (n1 != n2)
890			break;
891		if (n1 == 0 && n2 == 0) {
892			fclose(f1);
893			fclose(f2);
894			return (1);
895		}
896		if (memcmp(buff1, buff2, n1) != 0)
897			break;
898	}
899	fclose(f1);
900	fclose(f2);
901	failure_start(filename, line, "Files not identical");
902	logprintf("  file1=\"%s\"\n", fn1);
903	logprintf("  file2=\"%s\"\n", fn2);
904	failure_finish(NULL);
905	return (0);
906}
907
908/* Verify that the named file does exist. */
909int
910assertion_file_exists(const char *filename, int line, const char *f)
911{
912	assertion_count(filename, line);
913
914#if defined(_WIN32) && !defined(__CYGWIN__)
915	if (!_access(f, 0))
916		return (1);
917#else
918	if (!access(f, F_OK))
919		return (1);
920#endif
921	failure_start(filename, line, "File should exist: %s", f);
922	failure_finish(NULL);
923	return (0);
924}
925
926/* Verify that the named file doesn't exist. */
927int
928assertion_file_not_exists(const char *filename, int line, const char *f)
929{
930	assertion_count(filename, line);
931
932#if defined(_WIN32) && !defined(__CYGWIN__)
933	if (_access(f, 0))
934		return (1);
935#else
936	if (access(f, F_OK))
937		return (1);
938#endif
939	failure_start(filename, line, "File should not exist: %s", f);
940	failure_finish(NULL);
941	return (0);
942}
943
944/* Compare the contents of a file to a block of memory. */
945int
946assertion_file_contents(const char *filename, int line, const void *buff, int s, const char *fn)
947{
948	char *contents;
949	FILE *f;
950	int n;
951
952	assertion_count(filename, line);
953
954	f = fopen(fn, "rb");
955	if (f == NULL) {
956		failure_start(filename, line,
957		    "File should exist: %s", fn);
958		failure_finish(NULL);
959		return (0);
960	}
961	contents = malloc(s * 2);
962	n = (int)fread(contents, 1, s * 2, f);
963	fclose(f);
964	if (n == s && memcmp(buff, contents, s) == 0) {
965		free(contents);
966		return (1);
967	}
968	failure_start(filename, line, "File contents don't match");
969	logprintf("  file=\"%s\"\n", fn);
970	if (n > 0)
971		hexdump(contents, buff, n > 512 ? 512 : n, 0);
972	else {
973		logprintf("  File empty, contents should be:\n");
974		hexdump(buff, NULL, s > 512 ? 512 : s, 0);
975	}
976	failure_finish(NULL);
977	free(contents);
978	return (0);
979}
980
981/* Check the contents of a text file, being tolerant of line endings. */
982int
983assertion_text_file_contents(const char *filename, int line, const char *buff, const char *fn)
984{
985	char *contents;
986	const char *btxt, *ftxt;
987	FILE *f;
988	int n, s;
989
990	assertion_count(filename, line);
991	f = fopen(fn, "r");
992	if (f == NULL) {
993		failure_start(filename, line,
994		    "File doesn't exist: %s", fn);
995		failure_finish(NULL);
996		return (0);
997	}
998	s = (int)strlen(buff);
999	contents = malloc(s * 2 + 128);
1000	n = (int)fread(contents, 1, s * 2 + 128 - 1, f);
1001	if (n >= 0)
1002		contents[n] = '\0';
1003	fclose(f);
1004	/* Compare texts. */
1005	btxt = buff;
1006	ftxt = (const char *)contents;
1007	while (*btxt != '\0' && *ftxt != '\0') {
1008		if (*btxt == *ftxt) {
1009			++btxt;
1010			++ftxt;
1011			continue;
1012		}
1013		if (btxt[0] == '\n' && ftxt[0] == '\r' && ftxt[1] == '\n') {
1014			/* Pass over different new line characters. */
1015			++btxt;
1016			ftxt += 2;
1017			continue;
1018		}
1019		break;
1020	}
1021	if (*btxt == '\0' && *ftxt == '\0') {
1022		free(contents);
1023		return (1);
1024	}
1025	failure_start(filename, line, "Contents don't match");
1026	logprintf("  file=\"%s\"\n", fn);
1027	if (n > 0) {
1028		hexdump(contents, buff, n, 0);
1029		logprintf("  expected\n", fn);
1030		hexdump(buff, contents, s, 0);
1031	} else {
1032		logprintf("  File empty, contents should be:\n");
1033		hexdump(buff, NULL, s, 0);
1034	}
1035	failure_finish(NULL);
1036	free(contents);
1037	return (0);
1038}
1039
1040/* Verify that a text file contains the specified lines, regardless of order */
1041/* This could be more efficient if we sorted both sets of lines, etc, but
1042 * since this is used only for testing and only ever deals with a dozen or so
1043 * lines at a time, this relatively crude approach is just fine. */
1044int
1045assertion_file_contains_lines_any_order(const char *file, int line,
1046    const char *pathname, const char *lines[])
1047{
1048	char *buff;
1049	size_t buff_size;
1050	size_t expected_count, actual_count, i, j;
1051	char **expected = NULL;
1052	char *p, **actual = NULL;
1053	char c;
1054	int expected_failure = 0, actual_failure = 0;
1055
1056	assertion_count(file, line);
1057
1058	buff = slurpfile(&buff_size, "%s", pathname);
1059	if (buff == NULL) {
1060		failure_start(pathname, line, "Can't read file: %s", pathname);
1061		failure_finish(NULL);
1062		return (0);
1063	}
1064
1065	/* Make a copy of the provided lines and count up the expected
1066	 * file size. */
1067	for (i = 0; lines[i] != NULL; ++i) {
1068	}
1069	expected_count = i;
1070	if (expected_count) {
1071		expected = malloc(sizeof(char *) * expected_count);
1072		if (expected == NULL) {
1073			failure_start(pathname, line, "Can't allocate memory");
1074			failure_finish(NULL);
1075			free(expected);
1076			return (0);
1077		}
1078		for (i = 0; lines[i] != NULL; ++i) {
1079			expected[i] = strdup(lines[i]);
1080		}
1081	}
1082
1083	/* Break the file into lines */
1084	actual_count = 0;
1085	for (c = '\0', p = buff; p < buff + buff_size; ++p) {
1086		if (*p == '\x0d' || *p == '\x0a')
1087			*p = '\0';
1088		if (c == '\0' && *p != '\0')
1089			++actual_count;
1090		c = *p;
1091	}
1092	if (actual_count) {
1093		actual = calloc(sizeof(char *), actual_count);
1094		if (actual == NULL) {
1095			failure_start(pathname, line, "Can't allocate memory");
1096			failure_finish(NULL);
1097			free(expected);
1098			return (0);
1099		}
1100		for (j = 0, p = buff; p < buff + buff_size;
1101		    p += 1 + strlen(p)) {
1102			if (*p != '\0') {
1103				actual[j] = p;
1104				++j;
1105			}
1106		}
1107	}
1108
1109	/* Erase matching lines from both lists */
1110	for (i = 0; i < expected_count; ++i) {
1111		if (expected[i] == NULL)
1112			continue;
1113		for (j = 0; j < actual_count; ++j) {
1114			if (actual[j] == NULL)
1115				continue;
1116			if (strcmp(expected[i], actual[j]) == 0) {
1117				free(expected[i]);
1118				expected[i] = NULL;
1119				actual[j] = NULL;
1120				break;
1121			}
1122		}
1123	}
1124
1125	/* If there's anything left, it's a failure */
1126	for (i = 0; i < expected_count; ++i) {
1127		if (expected[i] != NULL)
1128			++expected_failure;
1129	}
1130	for (j = 0; j < actual_count; ++j) {
1131		if (actual[j] != NULL)
1132			++actual_failure;
1133	}
1134	if (expected_failure == 0 && actual_failure == 0) {
1135		free(buff);
1136		free(expected);
1137		free(actual);
1138		return (1);
1139	}
1140	failure_start(file, line, "File doesn't match: %s", pathname);
1141	for (i = 0; i < expected_count; ++i) {
1142		if (expected[i] != NULL) {
1143			logprintf("  Expected but not present: %s\n", expected[i]);
1144			free(expected[i]);
1145		}
1146	}
1147	for (j = 0; j < actual_count; ++j) {
1148		if (actual[j] != NULL)
1149			logprintf("  Present but not expected: %s\n", actual[j]);
1150	}
1151	failure_finish(NULL);
1152	free(buff);
1153	free(expected);
1154	free(actual);
1155	return (0);
1156}
1157
1158/* Test that two paths point to the same file. */
1159/* As a side-effect, asserts that both files exist. */
1160static int
1161is_hardlink(const char *file, int line,
1162    const char *path1, const char *path2)
1163{
1164#if defined(_WIN32) && !defined(__CYGWIN__)
1165	BY_HANDLE_FILE_INFORMATION bhfi1, bhfi2;
1166	int r;
1167
1168	assertion_count(file, line);
1169	r = my_GetFileInformationByName(path1, &bhfi1);
1170	if (r == 0) {
1171		failure_start(file, line, "File %s can't be inspected?", path1);
1172		failure_finish(NULL);
1173		return (0);
1174	}
1175	r = my_GetFileInformationByName(path2, &bhfi2);
1176	if (r == 0) {
1177		failure_start(file, line, "File %s can't be inspected?", path2);
1178		failure_finish(NULL);
1179		return (0);
1180	}
1181	return (bhfi1.dwVolumeSerialNumber == bhfi2.dwVolumeSerialNumber
1182		&& bhfi1.nFileIndexHigh == bhfi2.nFileIndexHigh
1183		&& bhfi1.nFileIndexLow == bhfi2.nFileIndexLow);
1184#else
1185	struct stat st1, st2;
1186	int r;
1187
1188	assertion_count(file, line);
1189	r = lstat(path1, &st1);
1190	if (r != 0) {
1191		failure_start(file, line, "File should exist: %s", path1);
1192		failure_finish(NULL);
1193		return (0);
1194	}
1195	r = lstat(path2, &st2);
1196	if (r != 0) {
1197		failure_start(file, line, "File should exist: %s", path2);
1198		failure_finish(NULL);
1199		return (0);
1200	}
1201	return (st1.st_ino == st2.st_ino && st1.st_dev == st2.st_dev);
1202#endif
1203}
1204
1205int
1206assertion_is_hardlink(const char *file, int line,
1207    const char *path1, const char *path2)
1208{
1209	if (is_hardlink(file, line, path1, path2))
1210		return (1);
1211	failure_start(file, line,
1212	    "Files %s and %s are not hardlinked", path1, path2);
1213	failure_finish(NULL);
1214	return (0);
1215}
1216
1217int
1218assertion_is_not_hardlink(const char *file, int line,
1219    const char *path1, const char *path2)
1220{
1221	if (!is_hardlink(file, line, path1, path2))
1222		return (1);
1223	failure_start(file, line,
1224	    "Files %s and %s should not be hardlinked", path1, path2);
1225	failure_finish(NULL);
1226	return (0);
1227}
1228
1229/* Verify a/b/mtime of 'pathname'. */
1230/* If 'recent', verify that it's within last 10 seconds. */
1231static int
1232assertion_file_time(const char *file, int line,
1233    const char *pathname, long t, long nsec, char type, int recent)
1234{
1235	long long filet, filet_nsec;
1236	int r;
1237
1238#if defined(_WIN32) && !defined(__CYGWIN__)
1239#define EPOC_TIME	(116444736000000000ULL)
1240	FILETIME fxtime, fbirthtime, fatime, fmtime;
1241	ULARGE_INTEGER wintm;
1242	HANDLE h;
1243	fxtime.dwLowDateTime = 0;
1244	fxtime.dwHighDateTime = 0;
1245
1246	assertion_count(file, line);
1247	/* Note: FILE_FLAG_BACKUP_SEMANTICS applies to open
1248	 * a directory file. If not, CreateFile() will fail when
1249	 * the pathname is a directory. */
1250	h = CreateFile(pathname, FILE_READ_ATTRIBUTES, 0, NULL,
1251	    OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
1252	if (h == INVALID_HANDLE_VALUE) {
1253		failure_start(file, line, "Can't access %s\n", pathname);
1254		failure_finish(NULL);
1255		return (0);
1256	}
1257	r = GetFileTime(h, &fbirthtime, &fatime, &fmtime);
1258	switch (type) {
1259	case 'a': fxtime = fatime; break;
1260	case 'b': fxtime = fbirthtime; break;
1261	case 'm': fxtime = fmtime; break;
1262	}
1263	CloseHandle(h);
1264	if (r == 0) {
1265		failure_start(file, line, "Can't GetFileTime %s\n", pathname);
1266		failure_finish(NULL);
1267		return (0);
1268	}
1269	wintm.LowPart = fxtime.dwLowDateTime;
1270	wintm.HighPart = fxtime.dwHighDateTime;
1271	filet = (wintm.QuadPart - EPOC_TIME) / 10000000;
1272	filet_nsec = ((wintm.QuadPart - EPOC_TIME) % 10000000) * 100;
1273	nsec = (nsec / 100) * 100; /* Round the request */
1274#else
1275	struct stat st;
1276
1277	assertion_count(file, line);
1278	r = lstat(pathname, &st);
1279	if (r != 0) {
1280		failure_start(file, line, "Can't stat %s\n", pathname);
1281		failure_finish(NULL);
1282		return (0);
1283	}
1284	switch (type) {
1285	case 'a': filet = st.st_atime; break;
1286	case 'm': filet = st.st_mtime; break;
1287	case 'b': filet = 0; break;
1288	default: fprintf(stderr, "INTERNAL: Bad type %c for file time", type);
1289		exit(1);
1290	}
1291#if defined(__FreeBSD__)
1292	switch (type) {
1293	case 'a': filet_nsec = st.st_atimespec.tv_nsec; break;
1294	case 'b': filet = st.st_birthtime;
1295		/* FreeBSD filesystems that don't support birthtime
1296		 * (e.g., UFS1) always return -1 here. */
1297		if (filet == -1) {
1298			return (1);
1299		}
1300		filet_nsec = st.st_birthtimespec.tv_nsec; break;
1301	case 'm': filet_nsec = st.st_mtimespec.tv_nsec; break;
1302	default: fprintf(stderr, "INTERNAL: Bad type %c for file time", type);
1303		exit(1);
1304	}
1305	/* FreeBSD generally only stores to microsecond res, so round. */
1306	filet_nsec = (filet_nsec / 1000) * 1000;
1307	nsec = (nsec / 1000) * 1000;
1308#else
1309	filet_nsec = nsec = 0;	/* Generic POSIX only has whole seconds. */
1310	if (type == 'b') return (1); /* Generic POSIX doesn't have birthtime */
1311#if defined(__HAIKU__)
1312	if (type == 'a') return (1); /* Haiku doesn't have atime. */
1313#endif
1314#endif
1315#endif
1316	if (recent) {
1317		/* Check that requested time is up-to-date. */
1318		time_t now = time(NULL);
1319		if (filet < now - 10 || filet > now + 1) {
1320			failure_start(file, line,
1321			    "File %s has %ctime %lld, %lld seconds ago\n",
1322			    pathname, type, filet, now - filet);
1323			failure_finish(NULL);
1324			return (0);
1325		}
1326	} else if (filet != t || filet_nsec != nsec) {
1327		failure_start(file, line,
1328		    "File %s has %ctime %lld.%09lld, expected %lld.%09lld",
1329		    pathname, type, filet, filet_nsec, t, nsec);
1330		failure_finish(NULL);
1331		return (0);
1332	}
1333	return (1);
1334}
1335
1336/* Verify atime of 'pathname'. */
1337int
1338assertion_file_atime(const char *file, int line,
1339    const char *pathname, long t, long nsec)
1340{
1341	return assertion_file_time(file, line, pathname, t, nsec, 'a', 0);
1342}
1343
1344/* Verify atime of 'pathname' is up-to-date. */
1345int
1346assertion_file_atime_recent(const char *file, int line, const char *pathname)
1347{
1348	return assertion_file_time(file, line, pathname, 0, 0, 'a', 1);
1349}
1350
1351/* Verify birthtime of 'pathname'. */
1352int
1353assertion_file_birthtime(const char *file, int line,
1354    const char *pathname, long t, long nsec)
1355{
1356	return assertion_file_time(file, line, pathname, t, nsec, 'b', 0);
1357}
1358
1359/* Verify birthtime of 'pathname' is up-to-date. */
1360int
1361assertion_file_birthtime_recent(const char *file, int line,
1362    const char *pathname)
1363{
1364	return assertion_file_time(file, line, pathname, 0, 0, 'b', 1);
1365}
1366
1367/* Verify mtime of 'pathname'. */
1368int
1369assertion_file_mtime(const char *file, int line,
1370    const char *pathname, long t, long nsec)
1371{
1372	return assertion_file_time(file, line, pathname, t, nsec, 'm', 0);
1373}
1374
1375/* Verify mtime of 'pathname' is up-to-date. */
1376int
1377assertion_file_mtime_recent(const char *file, int line, const char *pathname)
1378{
1379	return assertion_file_time(file, line, pathname, 0, 0, 'm', 1);
1380}
1381
1382/* Verify number of links to 'pathname'. */
1383int
1384assertion_file_nlinks(const char *file, int line,
1385    const char *pathname, int nlinks)
1386{
1387#if defined(_WIN32) && !defined(__CYGWIN__)
1388	BY_HANDLE_FILE_INFORMATION bhfi;
1389	int r;
1390
1391	assertion_count(file, line);
1392	r = my_GetFileInformationByName(pathname, &bhfi);
1393	if (r != 0 && bhfi.nNumberOfLinks == (DWORD)nlinks)
1394		return (1);
1395	failure_start(file, line, "File %s has %d links, expected %d",
1396	    pathname, bhfi.nNumberOfLinks, nlinks);
1397	failure_finish(NULL);
1398	return (0);
1399#else
1400	struct stat st;
1401	int r;
1402
1403	assertion_count(file, line);
1404	r = lstat(pathname, &st);
1405	if (r == 0 && (int)st.st_nlink == nlinks)
1406			return (1);
1407	failure_start(file, line, "File %s has %d links, expected %d",
1408	    pathname, st.st_nlink, nlinks);
1409	failure_finish(NULL);
1410	return (0);
1411#endif
1412}
1413
1414/* Verify size of 'pathname'. */
1415int
1416assertion_file_size(const char *file, int line, const char *pathname, long size)
1417{
1418	int64_t filesize;
1419	int r;
1420
1421	assertion_count(file, line);
1422#if defined(_WIN32) && !defined(__CYGWIN__)
1423	{
1424		BY_HANDLE_FILE_INFORMATION bhfi;
1425		r = !my_GetFileInformationByName(pathname, &bhfi);
1426		filesize = ((int64_t)bhfi.nFileSizeHigh << 32) + bhfi.nFileSizeLow;
1427	}
1428#else
1429	{
1430		struct stat st;
1431		r = lstat(pathname, &st);
1432		filesize = st.st_size;
1433	}
1434#endif
1435	if (r == 0 && filesize == size)
1436			return (1);
1437	failure_start(file, line, "File %s has size %ld, expected %ld",
1438	    pathname, (long)filesize, (long)size);
1439	failure_finish(NULL);
1440	return (0);
1441}
1442
1443/* Verify mode of 'pathname'. */
1444int
1445assertion_file_mode(const char *file, int line, const char *pathname, int expected_mode)
1446{
1447	int mode;
1448	int r;
1449
1450	assertion_count(file, line);
1451#if defined(_WIN32) && !defined(__CYGWIN__)
1452	failure_start(file, line, "assertFileMode not yet implemented for Windows");
1453#else
1454	{
1455		struct stat st;
1456		r = lstat(pathname, &st);
1457		mode = (int)(st.st_mode & 0777);
1458	}
1459	if (r == 0 && mode == expected_mode)
1460			return (1);
1461	failure_start(file, line, "File %s has mode %o, expected %o",
1462	    pathname, mode, expected_mode);
1463#endif
1464	failure_finish(NULL);
1465	return (0);
1466}
1467
1468/* Assert that 'pathname' is a dir.  If mode >= 0, verify that too. */
1469int
1470assertion_is_dir(const char *file, int line, const char *pathname, int mode)
1471{
1472	struct stat st;
1473	int r;
1474
1475#if defined(_WIN32) && !defined(__CYGWIN__)
1476	(void)mode; /* UNUSED */
1477#endif
1478	assertion_count(file, line);
1479	r = lstat(pathname, &st);
1480	if (r != 0) {
1481		failure_start(file, line, "Dir should exist: %s", pathname);
1482		failure_finish(NULL);
1483		return (0);
1484	}
1485	if (!S_ISDIR(st.st_mode)) {
1486		failure_start(file, line, "%s is not a dir", pathname);
1487		failure_finish(NULL);
1488		return (0);
1489	}
1490#if !defined(_WIN32) || defined(__CYGWIN__)
1491	/* Windows doesn't handle permissions the same way as POSIX,
1492	 * so just ignore the mode tests. */
1493	/* TODO: Can we do better here? */
1494	if (mode >= 0 && (mode_t)mode != (st.st_mode & 07777)) {
1495		failure_start(file, line, "Dir %s has wrong mode", pathname);
1496		logprintf("  Expected: 0%3o\n", mode);
1497		logprintf("  Found: 0%3o\n", st.st_mode & 07777);
1498		failure_finish(NULL);
1499		return (0);
1500	}
1501#endif
1502	return (1);
1503}
1504
1505/* Verify that 'pathname' is a regular file.  If 'mode' is >= 0,
1506 * verify that too. */
1507int
1508assertion_is_reg(const char *file, int line, const char *pathname, int mode)
1509{
1510	struct stat st;
1511	int r;
1512
1513#if defined(_WIN32) && !defined(__CYGWIN__)
1514	(void)mode; /* UNUSED */
1515#endif
1516	assertion_count(file, line);
1517	r = lstat(pathname, &st);
1518	if (r != 0 || !S_ISREG(st.st_mode)) {
1519		failure_start(file, line, "File should exist: %s", pathname);
1520		failure_finish(NULL);
1521		return (0);
1522	}
1523#if !defined(_WIN32) || defined(__CYGWIN__)
1524	/* Windows doesn't handle permissions the same way as POSIX,
1525	 * so just ignore the mode tests. */
1526	/* TODO: Can we do better here? */
1527	if (mode >= 0 && (mode_t)mode != (st.st_mode & 07777)) {
1528		failure_start(file, line, "File %s has wrong mode", pathname);
1529		logprintf("  Expected: 0%3o\n", mode);
1530		logprintf("  Found: 0%3o\n", st.st_mode & 07777);
1531		failure_finish(NULL);
1532		return (0);
1533	}
1534#endif
1535	return (1);
1536}
1537
1538/* Check whether 'pathname' is a symbolic link.  If 'contents' is
1539 * non-NULL, verify that the symlink has those contents. */
1540static int
1541is_symlink(const char *file, int line,
1542    const char *pathname, const char *contents)
1543{
1544#if defined(_WIN32) && !defined(__CYGWIN__)
1545	(void)pathname; /* UNUSED */
1546	(void)contents; /* UNUSED */
1547	assertion_count(file, line);
1548	/* Windows sort-of has real symlinks, but they're only usable
1549	 * by privileged users and are crippled even then, so there's
1550	 * really not much point in bothering with this. */
1551	return (0);
1552#else
1553	char buff[300];
1554	struct stat st;
1555	ssize_t linklen;
1556	int r;
1557
1558	assertion_count(file, line);
1559	r = lstat(pathname, &st);
1560	if (r != 0) {
1561		failure_start(file, line,
1562		    "Symlink should exist: %s", pathname);
1563		failure_finish(NULL);
1564		return (0);
1565	}
1566	if (!S_ISLNK(st.st_mode))
1567		return (0);
1568	if (contents == NULL)
1569		return (1);
1570	linklen = readlink(pathname, buff, sizeof(buff));
1571	if (linklen < 0) {
1572		failure_start(file, line, "Can't read symlink %s", pathname);
1573		failure_finish(NULL);
1574		return (0);
1575	}
1576	buff[linklen] = '\0';
1577	if (strcmp(buff, contents) != 0)
1578		return (0);
1579	return (1);
1580#endif
1581}
1582
1583/* Assert that path is a symlink that (optionally) contains contents. */
1584int
1585assertion_is_symlink(const char *file, int line,
1586    const char *path, const char *contents)
1587{
1588	if (is_symlink(file, line, path, contents))
1589		return (1);
1590	if (contents)
1591		failure_start(file, line, "File %s is not a symlink to %s",
1592		    path, contents);
1593	else
1594		failure_start(file, line, "File %s is not a symlink", path);
1595	failure_finish(NULL);
1596	return (0);
1597}
1598
1599
1600/* Create a directory and report any errors. */
1601int
1602assertion_make_dir(const char *file, int line, const char *dirname, int mode)
1603{
1604	assertion_count(file, line);
1605#if defined(_WIN32) && !defined(__CYGWIN__)
1606	(void)mode; /* UNUSED */
1607	if (0 == _mkdir(dirname))
1608		return (1);
1609#else
1610	if (0 == mkdir(dirname, mode)) {
1611		if (0 == chmod(dirname, mode)) {
1612			assertion_file_mode(file, line, dirname, mode);
1613			return (1);
1614		}
1615	}
1616#endif
1617	failure_start(file, line, "Could not create directory %s", dirname);
1618	failure_finish(NULL);
1619	return(0);
1620}
1621
1622/* Create a file with the specified contents and report any failures. */
1623int
1624assertion_make_file(const char *file, int line,
1625    const char *path, int mode, int csize, const void *contents)
1626{
1627#if defined(_WIN32) && !defined(__CYGWIN__)
1628	/* TODO: Rework this to set file mode as well. */
1629	FILE *f;
1630	(void)mode; /* UNUSED */
1631	assertion_count(file, line);
1632	f = fopen(path, "wb");
1633	if (f == NULL) {
1634		failure_start(file, line, "Could not create file %s", path);
1635		failure_finish(NULL);
1636		return (0);
1637	}
1638	if (contents != NULL) {
1639		size_t wsize;
1640
1641		if (csize < 0)
1642			wsize = strlen(contents);
1643		else
1644			wsize = (size_t)csize;
1645		if (wsize != fwrite(contents, 1, wsize, f)) {
1646			fclose(f);
1647			failure_start(file, line,
1648			    "Could not write file %s", path);
1649			failure_finish(NULL);
1650			return (0);
1651		}
1652	}
1653	fclose(f);
1654	return (1);
1655#else
1656	int fd;
1657	assertion_count(file, line);
1658	fd = open(path, O_CREAT | O_WRONLY, mode >= 0 ? mode : 0644);
1659	if (fd < 0) {
1660		failure_start(file, line, "Could not create %s", path);
1661		failure_finish(NULL);
1662		return (0);
1663	}
1664	if (0 != chmod(path, mode)) {
1665		failure_start(file, line, "Could not chmod %s", path);
1666		failure_finish(NULL);
1667		return (0);
1668	}
1669	if (contents != NULL) {
1670		ssize_t wsize;
1671
1672		if (csize < 0)
1673			wsize = (ssize_t)strlen(contents);
1674		else
1675			wsize = (ssize_t)csize;
1676		if (wsize != write(fd, contents, wsize)) {
1677			close(fd);
1678			failure_start(file, line,
1679			    "Could not write to %s", path);
1680			failure_finish(NULL);
1681			return (0);
1682		}
1683	}
1684	close(fd);
1685	assertion_file_mode(file, line, path, mode);
1686	return (1);
1687#endif
1688}
1689
1690/* Create a hardlink and report any failures. */
1691int
1692assertion_make_hardlink(const char *file, int line,
1693    const char *newpath, const char *linkto)
1694{
1695	int succeeded;
1696
1697	assertion_count(file, line);
1698#if defined(_WIN32) && !defined(__CYGWIN__)
1699	succeeded = my_CreateHardLinkA(newpath, linkto);
1700#elif HAVE_LINK
1701	succeeded = !link(linkto, newpath);
1702#else
1703	succeeded = 0;
1704#endif
1705	if (succeeded)
1706		return (1);
1707	failure_start(file, line, "Could not create hardlink");
1708	logprintf("   New link: %s\n", newpath);
1709	logprintf("   Old name: %s\n", linkto);
1710	failure_finish(NULL);
1711	return(0);
1712}
1713
1714/* Create a symlink and report any failures. */
1715int
1716assertion_make_symlink(const char *file, int line,
1717    const char *newpath, const char *linkto)
1718{
1719#if defined(_WIN32) && !defined(__CYGWIN__)
1720	int targetIsDir = 0;  /* TODO: Fix this */
1721	assertion_count(file, line);
1722	if (my_CreateSymbolicLinkA(newpath, linkto, targetIsDir))
1723		return (1);
1724#elif HAVE_SYMLINK
1725	assertion_count(file, line);
1726	if (0 == symlink(linkto, newpath))
1727		return (1);
1728#endif
1729	failure_start(file, line, "Could not create symlink");
1730	logprintf("   New link: %s\n", newpath);
1731	logprintf("   Old name: %s\n", linkto);
1732	failure_finish(NULL);
1733	return(0);
1734}
1735
1736/* Set umask, report failures. */
1737int
1738assertion_umask(const char *file, int line, int mask)
1739{
1740	assertion_count(file, line);
1741	(void)file; /* UNUSED */
1742	(void)line; /* UNUSED */
1743	umask(mask);
1744	return (1);
1745}
1746
1747/* Set times, report failures. */
1748int
1749assertion_utimes(const char *file, int line,
1750    const char *pathname, long at, long at_nsec, long mt, long mt_nsec)
1751{
1752	int r;
1753
1754#if defined(_WIN32) && !defined(__CYGWIN__)
1755#define WINTIME(sec, nsec) ((Int32x32To64(sec, 10000000) + EPOC_TIME)\
1756	 + (((nsec)/1000)*10))
1757	HANDLE h;
1758	ULARGE_INTEGER wintm;
1759	FILETIME fatime, fmtime;
1760	FILETIME *pat, *pmt;
1761
1762	assertion_count(file, line);
1763	h = CreateFileA(pathname,GENERIC_READ | GENERIC_WRITE,
1764		    FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING,
1765		    FILE_FLAG_BACKUP_SEMANTICS, NULL);
1766	if (h == INVALID_HANDLE_VALUE) {
1767		failure_start(file, line, "Can't access %s\n", pathname);
1768		failure_finish(NULL);
1769		return (0);
1770	}
1771
1772	if (at > 0 || at_nsec > 0) {
1773		wintm.QuadPart = WINTIME(at, at_nsec);
1774		fatime.dwLowDateTime = wintm.LowPart;
1775		fatime.dwHighDateTime = wintm.HighPart;
1776		pat = &fatime;
1777	} else
1778		pat = NULL;
1779	if (mt > 0 || mt_nsec > 0) {
1780		wintm.QuadPart = WINTIME(mt, mt_nsec);
1781		fmtime.dwLowDateTime = wintm.LowPart;
1782		fmtime.dwHighDateTime = wintm.HighPart;
1783		pmt = &fmtime;
1784	} else
1785		pmt = NULL;
1786	if (pat != NULL || pmt != NULL)
1787		r = SetFileTime(h, NULL, pat, pmt);
1788	else
1789		r = 1;
1790	CloseHandle(h);
1791	if (r == 0) {
1792		failure_start(file, line, "Can't SetFileTime %s\n", pathname);
1793		failure_finish(NULL);
1794		return (0);
1795	}
1796	return (1);
1797#else /* defined(_WIN32) && !defined(__CYGWIN__) */
1798	struct stat st;
1799	struct timeval times[2];
1800
1801#if !defined(__FreeBSD__)
1802	mt_nsec = at_nsec = 0;	/* Generic POSIX only has whole seconds. */
1803#endif
1804	if (mt == 0 && mt_nsec == 0 && at == 0 && at_nsec == 0)
1805		return (1);
1806
1807	r = lstat(pathname, &st);
1808	if (r < 0) {
1809		failure_start(file, line, "Can't stat %s\n", pathname);
1810		failure_finish(NULL);
1811		return (0);
1812	}
1813
1814	if (mt == 0 && mt_nsec == 0) {
1815		mt = st.st_mtime;
1816#if defined(__FreeBSD__)
1817		mt_nsec = st.st_mtimespec.tv_nsec;
1818		/* FreeBSD generally only stores to microsecond res, so round. */
1819		mt_nsec = (mt_nsec / 1000) * 1000;
1820#endif
1821	}
1822	if (at == 0 && at_nsec == 0) {
1823		at = st.st_atime;
1824#if defined(__FreeBSD__)
1825		at_nsec = st.st_atimespec.tv_nsec;
1826		/* FreeBSD generally only stores to microsecond res, so round. */
1827		at_nsec = (at_nsec / 1000) * 1000;
1828#endif
1829	}
1830
1831	times[1].tv_sec = mt;
1832	times[1].tv_usec = mt_nsec / 1000;
1833
1834	times[0].tv_sec = at;
1835	times[0].tv_usec = at_nsec / 1000;
1836
1837#ifdef HAVE_LUTIMES
1838	r = lutimes(pathname, times);
1839#else
1840	r = utimes(pathname, times);
1841#endif
1842	if (r < 0) {
1843		failure_start(file, line, "Can't utimes %s\n", pathname);
1844		failure_finish(NULL);
1845		return (0);
1846	}
1847	return (1);
1848#endif /* defined(_WIN32) && !defined(__CYGWIN__) */
1849}
1850
1851/* Set nodump, report failures. */
1852int
1853assertion_nodump(const char *file, int line, const char *pathname)
1854{
1855#if defined(HAVE_STRUCT_STAT_ST_FLAGS) && defined(UF_NODUMP)
1856	int r;
1857
1858	assertion_count(file, line);
1859	r = chflags(pathname, UF_NODUMP);
1860	if (r < 0) {
1861		failure_start(file, line, "Can't set nodump %s\n", pathname);
1862		failure_finish(NULL);
1863		return (0);
1864	}
1865#elif defined(EXT2_IOC_GETFLAGS) && defined(HAVE_WORKING_EXT2_IOC_GETFLAGS)\
1866	 && defined(EXT2_NODUMP_FL)
1867	int fd, r, flags;
1868
1869	assertion_count(file, line);
1870	fd = open(pathname, O_RDONLY | O_NONBLOCK);
1871	if (fd < 0) {
1872		failure_start(file, line, "Can't open %s\n", pathname);
1873		failure_finish(NULL);
1874		return (0);
1875	}
1876	r = ioctl(fd, EXT2_IOC_GETFLAGS, &flags);
1877	if (r < 0) {
1878		failure_start(file, line, "Can't get flags %s\n", pathname);
1879		failure_finish(NULL);
1880		return (0);
1881	}
1882	flags |= EXT2_NODUMP_FL;
1883	r = ioctl(fd, EXT2_IOC_SETFLAGS, &flags);
1884	if (r < 0) {
1885		failure_start(file, line, "Can't set nodump %s\n", pathname);
1886		failure_finish(NULL);
1887		return (0);
1888	}
1889	close(fd);
1890#else
1891	(void)pathname; /* UNUSED */
1892	assertion_count(file, line);
1893#endif
1894	return (1);
1895}
1896
1897/*
1898 *
1899 *  UTILITIES for use by tests.
1900 *
1901 */
1902
1903/*
1904 * Check whether platform supports symlinks.  This is intended
1905 * for tests to use in deciding whether to bother testing symlink
1906 * support; if the platform doesn't support symlinks, there's no point
1907 * in checking whether the program being tested can create them.
1908 *
1909 * Note that the first time this test is called, we actually go out to
1910 * disk to create and verify a symlink.  This is necessary because
1911 * symlink support is actually a property of a particular filesystem
1912 * and can thus vary between directories on a single system.  After
1913 * the first call, this returns the cached result from memory, so it's
1914 * safe to call it as often as you wish.
1915 */
1916int
1917canSymlink(void)
1918{
1919	/* Remember the test result */
1920	static int value = 0, tested = 0;
1921	if (tested)
1922		return (value);
1923
1924	++tested;
1925	assertion_make_file(__FILE__, __LINE__, "canSymlink.0", 0644, 1, "a");
1926	/* Note: Cygwin has its own symlink() emulation that does not
1927	 * use the Win32 CreateSymbolicLink() function. */
1928#if defined(_WIN32) && !defined(__CYGWIN__)
1929	value = my_CreateSymbolicLinkA("canSymlink.1", "canSymlink.0", 0)
1930	    && is_symlink(__FILE__, __LINE__, "canSymlink.1", "canSymlink.0");
1931#elif HAVE_SYMLINK
1932	value = (0 == symlink("canSymlink.0", "canSymlink.1"))
1933	    && is_symlink(__FILE__, __LINE__, "canSymlink.1","canSymlink.0");
1934#endif
1935	return (value);
1936}
1937
1938/* Platform-dependent options for hiding the output of a subcommand. */
1939#if defined(_WIN32) && !defined(__CYGWIN__)
1940static const char *redirectArgs = ">NUL 2>NUL"; /* Win32 cmd.exe */
1941#else
1942static const char *redirectArgs = ">/dev/null 2>/dev/null"; /* POSIX 'sh' */
1943#endif
1944/*
1945 * Can this platform run the bzip2 program?
1946 */
1947int
1948canBzip2(void)
1949{
1950	static int tested = 0, value = 0;
1951	if (!tested) {
1952		tested = 1;
1953		if (systemf("bzip2 -d -V %s", redirectArgs) == 0)
1954			value = 1;
1955	}
1956	return (value);
1957}
1958
1959/*
1960 * Can this platform run the grzip program?
1961 */
1962int
1963canGrzip(void)
1964{
1965	static int tested = 0, value = 0;
1966	if (!tested) {
1967		tested = 1;
1968		if (systemf("grzip -V %s", redirectArgs) == 0)
1969			value = 1;
1970	}
1971	return (value);
1972}
1973
1974/*
1975 * Can this platform run the gzip program?
1976 */
1977int
1978canGzip(void)
1979{
1980	static int tested = 0, value = 0;
1981	if (!tested) {
1982		tested = 1;
1983		if (systemf("gzip -V %s", redirectArgs) == 0)
1984			value = 1;
1985	}
1986	return (value);
1987}
1988
1989/*
1990 * Can this platform run the lrzip program?
1991 */
1992int
1993canRunCommand(const char *cmd)
1994{
1995  static int tested = 0, value = 0;
1996  if (!tested) {
1997    tested = 1;
1998    if (systemf("%s %s", cmd, redirectArgs) == 0)
1999      value = 1;
2000  }
2001  return (value);
2002}
2003
2004int
2005canLrzip(void)
2006{
2007	static int tested = 0, value = 0;
2008	if (!tested) {
2009		tested = 1;
2010		if (systemf("lrzip -V %s", redirectArgs) == 0)
2011			value = 1;
2012	}
2013	return (value);
2014}
2015
2016/*
2017 * Can this platform run the lz4 program?
2018 */
2019int
2020canLz4(void)
2021{
2022	static int tested = 0, value = 0;
2023	if (!tested) {
2024		tested = 1;
2025		if (systemf("lz4 -V %s", redirectArgs) == 0)
2026			value = 1;
2027	}
2028	return (value);
2029}
2030
2031/*
2032 * Can this platform run the lzip program?
2033 */
2034int
2035canLzip(void)
2036{
2037	static int tested = 0, value = 0;
2038	if (!tested) {
2039		tested = 1;
2040		if (systemf("lzip -V %s", redirectArgs) == 0)
2041			value = 1;
2042	}
2043	return (value);
2044}
2045
2046/*
2047 * Can this platform run the lzma program?
2048 */
2049int
2050canLzma(void)
2051{
2052	static int tested = 0, value = 0;
2053	if (!tested) {
2054		tested = 1;
2055		if (systemf("lzma -V %s", redirectArgs) == 0)
2056			value = 1;
2057	}
2058	return (value);
2059}
2060
2061/*
2062 * Can this platform run the lzop program?
2063 */
2064int
2065canLzop(void)
2066{
2067	static int tested = 0, value = 0;
2068	if (!tested) {
2069		tested = 1;
2070		if (systemf("lzop -V %s", redirectArgs) == 0)
2071			value = 1;
2072	}
2073	return (value);
2074}
2075
2076/*
2077 * Can this platform run the xz program?
2078 */
2079int
2080canXz(void)
2081{
2082	static int tested = 0, value = 0;
2083	if (!tested) {
2084		tested = 1;
2085		if (systemf("xz -V %s", redirectArgs) == 0)
2086			value = 1;
2087	}
2088	return (value);
2089}
2090
2091/*
2092 * Can this filesystem handle nodump flags.
2093 */
2094#if defined(HAVE_STRUCT_STAT_ST_FLAGS) && defined(UF_NODUMP)
2095
2096int
2097canNodump(void)
2098{
2099	const char *path = "cannodumptest";
2100	struct stat sb;
2101
2102	assertion_make_file(__FILE__, __LINE__, path, 0644, 0, NULL);
2103	if (chflags(path, UF_NODUMP) < 0)
2104		return (0);
2105	if (stat(path, &sb) < 0)
2106		return (0);
2107	if (sb.st_flags & UF_NODUMP)
2108		return (1);
2109	return (0);
2110}
2111
2112#elif defined(EXT2_IOC_GETFLAGS) && defined(HAVE_WORKING_EXT2_IOC_GETFLAGS)\
2113	 && defined(EXT2_NODUMP_FL)
2114
2115int
2116canNodump(void)
2117{
2118	const char *path = "cannodumptest";
2119	int fd, r, flags;
2120
2121	assertion_make_file(__FILE__, __LINE__, path, 0644, 0, NULL);
2122	fd = open(path, O_RDONLY | O_NONBLOCK);
2123	if (fd < 0)
2124		return (0);
2125	r = ioctl(fd, EXT2_IOC_GETFLAGS, &flags);
2126	if (r < 0)
2127		return (0);
2128	flags |= EXT2_NODUMP_FL;
2129	r = ioctl(fd, EXT2_IOC_SETFLAGS, &flags);
2130	if (r < 0)
2131		return (0);
2132	close(fd);
2133	fd = open(path, O_RDONLY | O_NONBLOCK);
2134	if (fd < 0)
2135		return (0);
2136	r = ioctl(fd, EXT2_IOC_GETFLAGS, &flags);
2137	if (r < 0)
2138		return (0);
2139	close(fd);
2140	if (flags & EXT2_NODUMP_FL)
2141		return (1);
2142	return (0);
2143}
2144
2145#else
2146
2147int
2148canNodump()
2149{
2150	return (0);
2151}
2152
2153#endif
2154
2155/*
2156 * Sleep as needed; useful for verifying disk timestamp changes by
2157 * ensuring that the wall-clock time has actually changed before we
2158 * go back to re-read something from disk.
2159 */
2160void
2161sleepUntilAfter(time_t t)
2162{
2163	while (t >= time(NULL))
2164#if defined(_WIN32) && !defined(__CYGWIN__)
2165		Sleep(500);
2166#else
2167		sleep(1);
2168#endif
2169}
2170
2171/*
2172 * Call standard system() call, but build up the command line using
2173 * sprintf() conventions.
2174 */
2175int
2176systemf(const char *fmt, ...)
2177{
2178	char buff[8192];
2179	va_list ap;
2180	int r;
2181
2182	va_start(ap, fmt);
2183	vsprintf(buff, fmt, ap);
2184	if (verbosity > VERBOSITY_FULL)
2185		logprintf("Cmd: %s\n", buff);
2186	r = system(buff);
2187	va_end(ap);
2188	return (r);
2189}
2190
2191/*
2192 * Slurp a file into memory for ease of comparison and testing.
2193 * Returns size of file in 'sizep' if non-NULL, null-terminates
2194 * data in memory for ease of use.
2195 */
2196char *
2197slurpfile(size_t * sizep, const char *fmt, ...)
2198{
2199	char filename[8192];
2200	struct stat st;
2201	va_list ap;
2202	char *p;
2203	ssize_t bytes_read;
2204	FILE *f;
2205	int r;
2206
2207	va_start(ap, fmt);
2208	vsprintf(filename, fmt, ap);
2209	va_end(ap);
2210
2211	f = fopen(filename, "rb");
2212	if (f == NULL) {
2213		/* Note: No error; non-existent file is okay here. */
2214		return (NULL);
2215	}
2216	r = fstat(fileno(f), &st);
2217	if (r != 0) {
2218		logprintf("Can't stat file %s\n", filename);
2219		fclose(f);
2220		return (NULL);
2221	}
2222	p = malloc((size_t)st.st_size + 1);
2223	if (p == NULL) {
2224		logprintf("Can't allocate %ld bytes of memory to read file %s\n",
2225		    (long int)st.st_size, filename);
2226		fclose(f);
2227		return (NULL);
2228	}
2229	bytes_read = fread(p, 1, (size_t)st.st_size, f);
2230	if (bytes_read < st.st_size) {
2231		logprintf("Can't read file %s\n", filename);
2232		fclose(f);
2233		free(p);
2234		return (NULL);
2235	}
2236	p[st.st_size] = '\0';
2237	if (sizep != NULL)
2238		*sizep = (size_t)st.st_size;
2239	fclose(f);
2240	return (p);
2241}
2242
2243/*
2244 * Slurp a file into memory for ease of comparison and testing.
2245 * Returns size of file in 'sizep' if non-NULL, null-terminates
2246 * data in memory for ease of use.
2247 */
2248void
2249dumpfile(const char *filename, void *data, size_t len)
2250{
2251	ssize_t bytes_written;
2252	FILE *f;
2253
2254	f = fopen(filename, "wb");
2255	if (f == NULL) {
2256		logprintf("Can't open file %s for writing\n", filename);
2257		return;
2258	}
2259	bytes_written = fwrite(data, 1, len, f);
2260	if (bytes_written < (ssize_t)len)
2261		logprintf("Can't write file %s\n", filename);
2262	fclose(f);
2263}
2264
2265/* Read a uuencoded file from the reference directory, decode, and
2266 * write the result into the current directory. */
2267#define VALID_UUDECODE(c) (c >= 32 && c <= 96)
2268#define	UUDECODE(c) (((c) - 0x20) & 0x3f)
2269void
2270extract_reference_file(const char *name)
2271{
2272	char buff[1024];
2273	FILE *in, *out;
2274
2275	sprintf(buff, "%s/%s.uu", refdir, name);
2276	in = fopen(buff, "r");
2277	failure("Couldn't open reference file %s", buff);
2278	assert(in != NULL);
2279	if (in == NULL)
2280		return;
2281	/* Read up to and including the 'begin' line. */
2282	for (;;) {
2283		if (fgets(buff, sizeof(buff), in) == NULL) {
2284			/* TODO: This is a failure. */
2285			return;
2286		}
2287		if (memcmp(buff, "begin ", 6) == 0)
2288			break;
2289	}
2290	/* Now, decode the rest and write it. */
2291	out = fopen(name, "wb");
2292	while (fgets(buff, sizeof(buff), in) != NULL) {
2293		char *p = buff;
2294		int bytes;
2295
2296		if (memcmp(buff, "end", 3) == 0)
2297			break;
2298
2299		bytes = UUDECODE(*p++);
2300		while (bytes > 0) {
2301			int n = 0;
2302			/* Write out 1-3 bytes from that. */
2303			if (bytes > 0) {
2304				assert(VALID_UUDECODE(p[0]));
2305				assert(VALID_UUDECODE(p[1]));
2306				n = UUDECODE(*p++) << 18;
2307				n |= UUDECODE(*p++) << 12;
2308				fputc(n >> 16, out);
2309				--bytes;
2310			}
2311			if (bytes > 0) {
2312				assert(VALID_UUDECODE(p[0]));
2313				n |= UUDECODE(*p++) << 6;
2314				fputc((n >> 8) & 0xFF, out);
2315				--bytes;
2316			}
2317			if (bytes > 0) {
2318				assert(VALID_UUDECODE(p[0]));
2319				n |= UUDECODE(*p++);
2320				fputc(n & 0xFF, out);
2321				--bytes;
2322			}
2323		}
2324	}
2325	fclose(out);
2326	fclose(in);
2327}
2328
2329void
2330copy_reference_file(const char *name)
2331{
2332	char buff[1024];
2333	FILE *in, *out;
2334	size_t rbytes;
2335
2336	sprintf(buff, "%s/%s", refdir, name);
2337	in = fopen(buff, "rb");
2338	failure("Couldn't open reference file %s", buff);
2339	assert(in != NULL);
2340	if (in == NULL)
2341		return;
2342	/* Now, decode the rest and write it. */
2343	/* Not a lot of error checking here; the input better be right. */
2344	out = fopen(name, "wb");
2345	while ((rbytes = fread(buff, 1, sizeof(buff), in)) > 0) {
2346		if (fwrite(buff, 1, rbytes, out) != rbytes) {
2347			logprintf("Error: fwrite\n");
2348			break;
2349		}
2350	}
2351	fclose(out);
2352	fclose(in);
2353}
2354
2355int
2356is_LargeInode(const char *file)
2357{
2358#if defined(_WIN32) && !defined(__CYGWIN__)
2359	BY_HANDLE_FILE_INFORMATION bhfi;
2360	int r;
2361
2362	r = my_GetFileInformationByName(file, &bhfi);
2363	if (r != 0)
2364		return (0);
2365	return (bhfi.nFileIndexHigh & 0x0000FFFFUL);
2366#else
2367	struct stat st;
2368	int64_t ino;
2369
2370	if (stat(file, &st) < 0)
2371		return (0);
2372	ino = (int64_t)st.st_ino;
2373	return (ino > 0xffffffff);
2374#endif
2375}
2376
2377void
2378extract_reference_files(const char **names)
2379{
2380	while (names && *names)
2381		extract_reference_file(*names++);
2382}
2383
2384/*
2385 *
2386 * TEST management
2387 *
2388 */
2389
2390/*
2391 * "list.h" is simply created by "grep DEFINE_TEST test_*.c"; it has
2392 * a line like
2393 *      DEFINE_TEST(test_function)
2394 * for each test.
2395 */
2396
2397/* Use "list.h" to declare all of the test functions. */
2398#undef DEFINE_TEST
2399#define	DEFINE_TEST(name) void name(void);
2400#include "list.h"
2401
2402/* Use "list.h" to create a list of all tests (functions and names). */
2403#undef DEFINE_TEST
2404#define	DEFINE_TEST(n) { n, #n, 0 },
2405struct test_list_t tests[] = {
2406	#include "list.h"
2407};
2408
2409/*
2410 * Summarize repeated failures in the just-completed test.
2411 */
2412static void
2413test_summarize(int failed, int skips_num)
2414{
2415	unsigned int i;
2416
2417	switch (verbosity) {
2418	case VERBOSITY_SUMMARY_ONLY:
2419		printf(failed ? "E" : ".");
2420		fflush(stdout);
2421		break;
2422	case VERBOSITY_PASSFAIL:
2423		printf(failed ? "FAIL\n" : skips_num ? "ok (S)\n" : "ok\n");
2424		break;
2425	}
2426
2427	log_console = (verbosity == VERBOSITY_LIGHT_REPORT);
2428
2429	for (i = 0; i < sizeof(failed_lines)/sizeof(failed_lines[0]); i++) {
2430		if (failed_lines[i].count > 1 && !failed_lines[i].skip)
2431			logprintf("%s:%d: Summary: Failed %d times\n",
2432			    failed_filename, i, failed_lines[i].count);
2433	}
2434	/* Clear the failure history for the next file. */
2435	failed_filename = NULL;
2436	memset(failed_lines, 0, sizeof(failed_lines));
2437}
2438
2439/*
2440 * Actually run a single test, with appropriate setup and cleanup.
2441 */
2442static int
2443test_run(int i, const char *tmpdir)
2444{
2445	char workdir[1024];
2446	char logfilename[64];
2447	int failures_before = failures;
2448	int skips_before = skips;
2449	int oldumask;
2450
2451	switch (verbosity) {
2452	case VERBOSITY_SUMMARY_ONLY: /* No per-test reports at all */
2453		break;
2454	case VERBOSITY_PASSFAIL: /* rest of line will include ok/FAIL marker */
2455		printf("%3d: %-64s", i, tests[i].name);
2456		fflush(stdout);
2457		break;
2458	default: /* Title of test, details will follow */
2459		printf("%3d: %s\n", i, tests[i].name);
2460	}
2461
2462	/* Chdir to the top-level work directory. */
2463	if (!assertChdir(tmpdir)) {
2464		fprintf(stderr,
2465		    "ERROR: Can't chdir to top work dir %s\n", tmpdir);
2466		exit(1);
2467	}
2468	/* Create a log file for this test. */
2469	sprintf(logfilename, "%s.log", tests[i].name);
2470	logfile = fopen(logfilename, "w");
2471	fprintf(logfile, "%s\n\n", tests[i].name);
2472	/* Chdir() to a work dir for this specific test. */
2473	snprintf(workdir, sizeof(workdir), "%s/%s", tmpdir, tests[i].name);
2474	testworkdir = workdir;
2475	if (!assertMakeDir(testworkdir, 0755)
2476	    || !assertChdir(testworkdir)) {
2477		fprintf(stderr,
2478		    "ERROR: Can't chdir to work dir %s\n", testworkdir);
2479		exit(1);
2480	}
2481	/* Explicitly reset the locale before each test. */
2482	setlocale(LC_ALL, "C");
2483	/* Record the umask before we run the test. */
2484	umask(oldumask = umask(0));
2485	/*
2486	 * Run the actual test.
2487	 */
2488	(*tests[i].func)();
2489	/*
2490	 * Clean up and report afterwards.
2491	 */
2492	testworkdir = NULL;
2493	/* Restore umask */
2494	umask(oldumask);
2495	/* Reset locale. */
2496	setlocale(LC_ALL, "C");
2497	/* Reset directory. */
2498	if (!assertChdir(tmpdir)) {
2499		fprintf(stderr, "ERROR: Couldn't chdir to temp dir %s\n",
2500		    tmpdir);
2501		exit(1);
2502	}
2503	/* Report per-test summaries. */
2504	tests[i].failures = failures - failures_before;
2505	test_summarize(tests[i].failures, skips - skips_before);
2506	/* Close the per-test log file. */
2507	fclose(logfile);
2508	logfile = NULL;
2509	/* If there were no failures, we can remove the work dir and logfile. */
2510	if (tests[i].failures == 0) {
2511		if (!keep_temp_files && assertChdir(tmpdir)) {
2512#if defined(_WIN32) && !defined(__CYGWIN__)
2513			/* Make sure not to leave empty directories.
2514			 * Sometimes a processing of closing files used by tests
2515			 * is not done, then rmdir will be failed and it will
2516			 * leave a empty test directory. So we should wait a few
2517			 * seconds and retry rmdir. */
2518			int r, t;
2519			for (t = 0; t < 10; t++) {
2520				if (t > 0)
2521					Sleep(1000);
2522				r = systemf("rmdir /S /Q %s", tests[i].name);
2523				if (r == 0)
2524					break;
2525			}
2526			systemf("del %s", logfilename);
2527#else
2528			systemf("rm -rf %s", tests[i].name);
2529			systemf("rm %s", logfilename);
2530#endif
2531		}
2532	}
2533	/* Return appropriate status. */
2534	return (tests[i].failures);
2535}
2536
2537/*
2538 *
2539 *
2540 * MAIN and support routines.
2541 *
2542 *
2543 */
2544
2545static void
2546usage(const char *program)
2547{
2548	static const int limit = sizeof(tests) / sizeof(tests[0]);
2549	int i;
2550
2551	printf("Usage: %s [options] <test> <test> ...\n", program);
2552	printf("Default is to run all tests.\n");
2553	printf("Otherwise, specify the numbers of the tests you wish to run.\n");
2554	printf("Options:\n");
2555	printf("  -d  Dump core after any failure, for debugging.\n");
2556	printf("  -k  Keep all temp files.\n");
2557	printf("      Default: temp files for successful tests deleted.\n");
2558#ifdef PROGRAM
2559	printf("  -p <path>  Path to executable to be tested.\n");
2560	printf("      Default: path taken from " ENVBASE " environment variable.\n");
2561#endif
2562	printf("  -q  Quiet.\n");
2563	printf("  -r <dir>   Path to dir containing reference files.\n");
2564	printf("      Default: Current directory.\n");
2565	printf("  -u  Keep running specifies tests until one fails.\n");
2566	printf("  -v  Verbose.\n");
2567	printf("Available tests:\n");
2568	for (i = 0; i < limit; i++)
2569		printf("  %d: %s\n", i, tests[i].name);
2570	exit(1);
2571}
2572
2573static char *
2574get_refdir(const char *d)
2575{
2576	size_t tried_size, buff_size;
2577	char *buff, *tried, *pwd = NULL, *p = NULL;
2578
2579#ifdef PATH_MAX
2580	buff_size = PATH_MAX;
2581#else
2582	buff_size = 8192;
2583#endif
2584	buff = calloc(buff_size, 1);
2585	if (buff == NULL) {
2586		fprintf(stderr, "Unable to allocate memory\n");
2587		exit(1);
2588	}
2589
2590	/* Allocate a buffer to hold the various directories we checked. */
2591	tried_size = buff_size * 2;
2592	tried = calloc(tried_size, 1);
2593	if (tried == NULL) {
2594		fprintf(stderr, "Unable to allocate memory\n");
2595		exit(1);
2596	}
2597
2598	/* If a dir was specified, try that */
2599	if (d != NULL) {
2600		pwd = NULL;
2601		snprintf(buff, buff_size, "%s", d);
2602		p = slurpfile(NULL, "%s/%s", buff, KNOWNREF);
2603		if (p != NULL) goto success;
2604		strncat(tried, buff, tried_size - strlen(tried) - 1);
2605		strncat(tried, "\n", tried_size - strlen(tried) - 1);
2606		goto failure;
2607	}
2608
2609	/* Get the current dir. */
2610#ifdef PATH_MAX
2611	pwd = getcwd(NULL, PATH_MAX);/* Solaris getcwd needs the size. */
2612#else
2613	pwd = getcwd(NULL, 0);
2614#endif
2615	while (pwd[strlen(pwd) - 1] == '\n')
2616		pwd[strlen(pwd) - 1] = '\0';
2617
2618	/* Look for a known file. */
2619	snprintf(buff, buff_size, "%s", pwd);
2620	p = slurpfile(NULL, "%s/%s", buff, KNOWNREF);
2621	if (p != NULL) goto success;
2622	strncat(tried, buff, tried_size - strlen(tried) - 1);
2623	strncat(tried, "\n", tried_size - strlen(tried) - 1);
2624
2625	snprintf(buff, buff_size, "%s/test", pwd);
2626	p = slurpfile(NULL, "%s/%s", buff, KNOWNREF);
2627	if (p != NULL) goto success;
2628	strncat(tried, buff, tried_size - strlen(tried) - 1);
2629	strncat(tried, "\n", tried_size - strlen(tried) - 1);
2630
2631#if defined(LIBRARY)
2632	snprintf(buff, buff_size, "%s/%s/test", pwd, LIBRARY);
2633#else
2634	snprintf(buff, buff_size, "%s/%s/test", pwd, PROGRAM);
2635#endif
2636	p = slurpfile(NULL, "%s/%s", buff, KNOWNREF);
2637	if (p != NULL) goto success;
2638	strncat(tried, buff, tried_size - strlen(tried) - 1);
2639	strncat(tried, "\n", tried_size - strlen(tried) - 1);
2640
2641#if defined(PROGRAM_ALIAS)
2642	snprintf(buff, buff_size, "%s/%s/test", pwd, PROGRAM_ALIAS);
2643	p = slurpfile(NULL, "%s/%s", buff, KNOWNREF);
2644	if (p != NULL) goto success;
2645	strncat(tried, buff, tried_size - strlen(tried) - 1);
2646	strncat(tried, "\n", tried_size - strlen(tried) - 1);
2647#endif
2648
2649	if (memcmp(pwd, "/usr/obj", 8) == 0) {
2650		snprintf(buff, buff_size, "%s", pwd + 8);
2651		p = slurpfile(NULL, "%s/%s", buff, KNOWNREF);
2652		if (p != NULL) goto success;
2653		strncat(tried, buff, tried_size - strlen(tried) - 1);
2654		strncat(tried, "\n", tried_size - strlen(tried) - 1);
2655
2656		snprintf(buff, buff_size, "%s/test", pwd + 8);
2657		p = slurpfile(NULL, "%s/%s", buff, KNOWNREF);
2658		if (p != NULL) goto success;
2659		strncat(tried, buff, tried_size - strlen(tried) - 1);
2660		strncat(tried, "\n", tried_size - strlen(tried) - 1);
2661	}
2662
2663failure:
2664	printf("Unable to locate known reference file %s\n", KNOWNREF);
2665	printf("  Checked following directories:\n%s\n", tried);
2666	printf("Use -r option to specify full path to reference directory\n");
2667#if defined(_WIN32) && !defined(__CYGWIN__) && defined(_DEBUG)
2668	DebugBreak();
2669#endif
2670	exit(1);
2671
2672success:
2673	free(p);
2674	free(pwd);
2675	free(tried);
2676
2677	/* Copy result into a fresh buffer to reduce memory usage. */
2678	p = strdup(buff);
2679	free(buff);
2680	return p;
2681}
2682
2683int
2684main(int argc, char **argv)
2685{
2686	static const int limit = sizeof(tests) / sizeof(tests[0]);
2687	int test_set[sizeof(tests) / sizeof(tests[0])];
2688	int i = 0, j = 0, tests_run = 0, tests_failed = 0, option;
2689	time_t now;
2690	char *refdir_alloc = NULL;
2691	const char *progname;
2692	char **saved_argv;
2693	const char *tmp, *option_arg, *p;
2694	char tmpdir[256], *pwd, *testprogdir, *tmp2 = NULL, *vlevel = NULL;
2695	char tmpdir_timestamp[256];
2696
2697	(void)argc; /* UNUSED */
2698
2699	/* Get the current dir. */
2700#ifdef PATH_MAX
2701	pwd = getcwd(NULL, PATH_MAX);/* Solaris getcwd needs the size. */
2702#else
2703	pwd = getcwd(NULL, 0);
2704#endif
2705	while (pwd[strlen(pwd) - 1] == '\n')
2706		pwd[strlen(pwd) - 1] = '\0';
2707
2708#if defined(HAVE__CrtSetReportMode) && !defined(__WATCOMC__)
2709	/* To stop to run the default invalid parameter handler. */
2710	_set_invalid_parameter_handler(invalid_parameter_handler);
2711	/* Disable annoying assertion message box. */
2712	_CrtSetReportMode(_CRT_ASSERT, 0);
2713#endif
2714
2715	/*
2716	 * Name of this program, used to build root of our temp directory
2717	 * tree.
2718	 */
2719	progname = p = argv[0];
2720	if ((testprogdir = (char *)malloc(strlen(progname) + 1)) == NULL)
2721	{
2722		fprintf(stderr, "ERROR: Out of memory.");
2723		exit(1);
2724	}
2725	strcpy(testprogdir, progname);
2726	while (*p != '\0') {
2727		/* Support \ or / dir separators for Windows compat. */
2728		if (*p == '/' || *p == '\\')
2729		{
2730			progname = p + 1;
2731			i = j;
2732		}
2733		++p;
2734		j++;
2735	}
2736	testprogdir[i] = '\0';
2737#if defined(_WIN32) && !defined(__CYGWIN__)
2738	if (testprogdir[0] != '/' && testprogdir[0] != '\\' &&
2739	    !(((testprogdir[0] >= 'a' && testprogdir[0] <= 'z') ||
2740	       (testprogdir[0] >= 'A' && testprogdir[0] <= 'Z')) &&
2741		testprogdir[1] == ':' &&
2742		(testprogdir[2] == '/' || testprogdir[2] == '\\')))
2743#else
2744	if (testprogdir[0] != '/')
2745#endif
2746	{
2747		/* Fixup path for relative directories. */
2748		if ((testprogdir = (char *)realloc(testprogdir,
2749			strlen(pwd) + 1 + strlen(testprogdir) + 1)) == NULL)
2750		{
2751			fprintf(stderr, "ERROR: Out of memory.");
2752			exit(1);
2753		}
2754		memmove(testprogdir + strlen(pwd) + 1, testprogdir,
2755		    strlen(testprogdir) + 1);
2756		memcpy(testprogdir, pwd, strlen(pwd));
2757		testprogdir[strlen(pwd)] = '/';
2758	}
2759
2760#ifdef PROGRAM
2761	/* Get the target program from environment, if available. */
2762	testprogfile = getenv(ENVBASE);
2763#endif
2764
2765	if (getenv("TMPDIR") != NULL)
2766		tmp = getenv("TMPDIR");
2767	else if (getenv("TMP") != NULL)
2768		tmp = getenv("TMP");
2769	else if (getenv("TEMP") != NULL)
2770		tmp = getenv("TEMP");
2771	else if (getenv("TEMPDIR") != NULL)
2772		tmp = getenv("TEMPDIR");
2773	else
2774		tmp = "/tmp";
2775
2776	/* Allow -d to be controlled through the environment. */
2777	if (getenv(ENVBASE "_DEBUG") != NULL)
2778		dump_on_failure = 1;
2779
2780	/* Allow -v to be controlled through the environment. */
2781	if (getenv("_VERBOSITY_LEVEL") != NULL)
2782	{
2783		vlevel = getenv("_VERBOSITY_LEVEL");
2784		verbosity = atoi(vlevel);
2785		if (verbosity < VERBOSITY_SUMMARY_ONLY || verbosity > VERBOSITY_FULL)
2786		{
2787			/* Unsupported verbosity levels are silently ignored */
2788			vlevel = NULL;
2789			verbosity = VERBOSITY_PASSFAIL;
2790		}
2791	}
2792
2793	/* Get the directory holding test files from environment. */
2794	refdir = getenv(ENVBASE "_TEST_FILES");
2795
2796	/*
2797	 * Parse options, without using getopt(), which isn't available
2798	 * on all platforms.
2799	 */
2800	++argv; /* Skip program name */
2801	while (*argv != NULL) {
2802		if (**argv != '-')
2803			break;
2804		p = *argv++;
2805		++p; /* Skip '-' */
2806		while (*p != '\0') {
2807			option = *p++;
2808			option_arg = NULL;
2809			/* If 'opt' takes an argument, parse that. */
2810			if (option == 'p' || option == 'r') {
2811				if (*p != '\0')
2812					option_arg = p;
2813				else if (*argv == NULL) {
2814					fprintf(stderr,
2815					    "Option -%c requires argument.\n",
2816					    option);
2817					usage(progname);
2818				} else
2819					option_arg = *argv++;
2820				p = ""; /* End of this option word. */
2821			}
2822
2823			/* Now, handle the option. */
2824			switch (option) {
2825			case 'd':
2826				dump_on_failure = 1;
2827				break;
2828			case 'k':
2829				keep_temp_files = 1;
2830				break;
2831			case 'p':
2832#ifdef PROGRAM
2833				testprogfile = option_arg;
2834#else
2835				fprintf(stderr, "-p option not permitted\n");
2836				usage(progname);
2837#endif
2838				break;
2839			case 'q':
2840				if (!vlevel)
2841					verbosity--;
2842				break;
2843			case 'r':
2844				refdir = option_arg;
2845				break;
2846			case 'u':
2847				until_failure++;
2848				break;
2849			case 'v':
2850				if (!vlevel)
2851					verbosity++;
2852				break;
2853			default:
2854				fprintf(stderr, "Unrecognized option '%c'\n",
2855				    option);
2856				usage(progname);
2857			}
2858		}
2859	}
2860
2861	/*
2862	 * Sanity-check that our options make sense.
2863	 */
2864#ifdef PROGRAM
2865	if (testprogfile == NULL)
2866	{
2867		if ((tmp2 = (char *)malloc(strlen(testprogdir) + 1 +
2868			strlen(PROGRAM) + 1)) == NULL)
2869		{
2870			fprintf(stderr, "ERROR: Out of memory.");
2871			exit(1);
2872		}
2873		strcpy(tmp2, testprogdir);
2874		strcat(tmp2, "/");
2875		strcat(tmp2, PROGRAM);
2876		testprogfile = tmp2;
2877	}
2878
2879	{
2880		char *testprg;
2881#if defined(_WIN32) && !defined(__CYGWIN__)
2882		/* Command.com sometimes rejects '/' separators. */
2883		testprg = strdup(testprogfile);
2884		for (i = 0; testprg[i] != '\0'; i++) {
2885			if (testprg[i] == '/')
2886				testprg[i] = '\\';
2887		}
2888		testprogfile = testprg;
2889#endif
2890		/* Quote the name that gets put into shell command lines. */
2891		testprg = malloc(strlen(testprogfile) + 3);
2892		strcpy(testprg, "\"");
2893		strcat(testprg, testprogfile);
2894		strcat(testprg, "\"");
2895		testprog = testprg;
2896	}
2897#endif
2898
2899#if !defined(_WIN32) && defined(SIGPIPE)
2900	{   /* Ignore SIGPIPE signals */
2901		struct sigaction sa;
2902		sa.sa_handler = SIG_IGN;
2903		sigemptyset(&sa.sa_mask);
2904		sa.sa_flags = 0;
2905		sigaction(SIGPIPE, &sa, NULL);
2906	}
2907#endif
2908
2909	/*
2910	 * Create a temp directory for the following tests.
2911	 * Include the time the tests started as part of the name,
2912	 * to make it easier to track the results of multiple tests.
2913	 */
2914	now = time(NULL);
2915	for (i = 0; ; i++) {
2916		strftime(tmpdir_timestamp, sizeof(tmpdir_timestamp),
2917		    "%Y-%m-%dT%H.%M.%S",
2918		    localtime(&now));
2919		sprintf(tmpdir, "%s/%s.%s-%03d", tmp, progname,
2920		    tmpdir_timestamp, i);
2921		if (assertMakeDir(tmpdir,0755))
2922			break;
2923		if (i >= 999) {
2924			fprintf(stderr,
2925			    "ERROR: Unable to create temp directory %s\n",
2926			    tmpdir);
2927			exit(1);
2928		}
2929	}
2930
2931	/*
2932	 * If the user didn't specify a directory for locating
2933	 * reference files, try to find the reference files in
2934	 * the "usual places."
2935	 */
2936	refdir = refdir_alloc = get_refdir(refdir);
2937
2938	/*
2939	 * Banner with basic information.
2940	 */
2941	printf("\n");
2942	printf("If tests fail or crash, details will be in:\n");
2943	printf("   %s\n", tmpdir);
2944	printf("\n");
2945	if (verbosity > VERBOSITY_SUMMARY_ONLY) {
2946		printf("Reference files will be read from: %s\n", refdir);
2947#ifdef PROGRAM
2948		printf("Running tests on: %s\n", testprog);
2949#endif
2950		printf("Exercising: ");
2951		fflush(stdout);
2952		printf("%s\n", EXTRA_VERSION);
2953	} else {
2954		printf("Running ");
2955		fflush(stdout);
2956	}
2957
2958	/*
2959	 * Run some or all of the individual tests.
2960	 */
2961	saved_argv = argv;
2962	do {
2963		argv = saved_argv;
2964		do {
2965			int test_num;
2966
2967			test_num = get_test_set(test_set, limit, *argv, tests);
2968			if (test_num < 0) {
2969				printf("*** INVALID Test %s\n", *argv);
2970				free(refdir_alloc);
2971				free(testprogdir);
2972				usage(progname);
2973				return (1);
2974			}
2975			for (i = 0; i < test_num; i++) {
2976				tests_run++;
2977				if (test_run(test_set[i], tmpdir)) {
2978					tests_failed++;
2979					if (until_failure)
2980						goto finish;
2981				}
2982			}
2983			if (*argv != NULL)
2984				argv++;
2985		} while (*argv != NULL);
2986	} while (until_failure);
2987
2988finish:
2989	/* Must be freed after all tests run */
2990	free(tmp2);
2991	free(testprogdir);
2992	free(pwd);
2993
2994	/*
2995	 * Report summary statistics.
2996	 */
2997	if (verbosity > VERBOSITY_SUMMARY_ONLY) {
2998		printf("\n");
2999		printf("Totals:\n");
3000		printf("  Tests run:         %8d\n", tests_run);
3001		printf("  Tests failed:      %8d\n", tests_failed);
3002		printf("  Assertions checked:%8d\n", assertions);
3003		printf("  Assertions failed: %8d\n", failures);
3004		printf("  Skips reported:    %8d\n", skips);
3005	}
3006	if (failures) {
3007		printf("\n");
3008		printf("Failing tests:\n");
3009		for (i = 0; i < limit; ++i) {
3010			if (tests[i].failures)
3011				printf("  %d: %s (%d failures)\n", i,
3012				    tests[i].name, tests[i].failures);
3013		}
3014		printf("\n");
3015		printf("Details for failing tests: %s\n", tmpdir);
3016		printf("\n");
3017	} else {
3018		if (verbosity == VERBOSITY_SUMMARY_ONLY)
3019			printf("\n");
3020		printf("%d tests passed, no failures\n", tests_run);
3021	}
3022
3023	free(refdir_alloc);
3024
3025	/* If the final tmpdir is empty, we can remove it. */
3026	/* This should be the usual case when all tests succeed. */
3027	assertChdir("..");
3028	rmdir(tmpdir);
3029
3030	return (tests_failed ? 1 : 0);
3031}
3032