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