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