1*543adbedSBen Gras /*
2*543adbedSBen Gras * Copyright (c) 2003-2009 Tim Kientzle
3*543adbedSBen Gras * All rights reserved.
4*543adbedSBen Gras *
5*543adbedSBen Gras * Redistribution and use in source and binary forms, with or without
6*543adbedSBen Gras * modification, are permitted provided that the following conditions
7*543adbedSBen Gras * are met:
8*543adbedSBen Gras * 1. Redistributions of source code must retain the above copyright
9*543adbedSBen Gras * notice, this list of conditions and the following disclaimer.
10*543adbedSBen Gras * 2. Redistributions in binary form must reproduce the above copyright
11*543adbedSBen Gras * notice, this list of conditions and the following disclaimer in the
12*543adbedSBen Gras * documentation and/or other materials provided with the distribution.
13*543adbedSBen Gras *
14*543adbedSBen Gras * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
15*543adbedSBen Gras * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
16*543adbedSBen Gras * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
17*543adbedSBen Gras * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
18*543adbedSBen Gras * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
19*543adbedSBen Gras * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
20*543adbedSBen Gras * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
21*543adbedSBen Gras * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22*543adbedSBen Gras * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
23*543adbedSBen Gras * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24*543adbedSBen Gras */
25*543adbedSBen Gras
26*543adbedSBen Gras #include "test.h"
27*543adbedSBen Gras #include <errno.h>
28*543adbedSBen Gras #include <locale.h>
29*543adbedSBen Gras #include <stdarg.h>
30*543adbedSBen Gras #include <time.h>
31*543adbedSBen Gras
32*543adbedSBen Gras /*
33*543adbedSBen Gras * This same file is used pretty much verbatim for all test harnesses.
34*543adbedSBen Gras *
35*543adbedSBen Gras * The next few lines are the only differences.
36*543adbedSBen Gras * TODO: Move this into a separate configuration header, have all test
37*543adbedSBen Gras * suites share one copy of this file.
38*543adbedSBen Gras */
39*543adbedSBen Gras __FBSDID("$FreeBSD: src/usr.bin/tar/test/main.c,v 1.6 2008/11/05 06:40:53 kientzle Exp $");
40*543adbedSBen Gras #define KNOWNREF "test_patterns_2.tar.uu"
41*543adbedSBen Gras #define ENVBASE "BSDTAR" /* Prefix for environment variables. */
42*543adbedSBen Gras #define PROGRAM "bsdtar" /* Name of program being tested. */
43*543adbedSBen Gras #undef LIBRARY /* Not testing a library. */
44*543adbedSBen Gras #undef EXTRA_DUMP /* How to dump extra data */
45*543adbedSBen Gras /* How to generate extra version info. */
46*543adbedSBen Gras #define EXTRA_VERSION (systemf("%s --version", testprog) ? "" : "")
47*543adbedSBen Gras
48*543adbedSBen Gras /*
49*543adbedSBen Gras *
50*543adbedSBen Gras * Windows support routines
51*543adbedSBen Gras *
52*543adbedSBen Gras * Note: Configuration is a tricky issue. Using HAVE_* feature macros
53*543adbedSBen Gras * in the test harness is dangerous because they cover up
54*543adbedSBen Gras * configuration errors. The classic example of this is omitting a
55*543adbedSBen Gras * configure check. If libarchive and libarchive_test both look for
56*543adbedSBen Gras * the same feature macro, such errors are hard to detect. Platform
57*543adbedSBen Gras * macros (e.g., _WIN32 or __GNUC__) are a little better, but can
58*543adbedSBen Gras * easily lead to very messy code. It's best to limit yourself
59*543adbedSBen Gras * to only the most generic programming techniques in the test harness
60*543adbedSBen Gras * and thus avoid conditionals altogether. Where that's not possible,
61*543adbedSBen Gras * try to minimize conditionals by grouping platform-specific tests in
62*543adbedSBen Gras * one place (e.g., test_acl_freebsd) or by adding new assert()
63*543adbedSBen Gras * functions (e.g., assertMakeHardlink()) to cover up platform
64*543adbedSBen Gras * differences. Platform-specific coding in libarchive_test is often
65*543adbedSBen Gras * a symptom that some capability is missing from libarchive itself.
66*543adbedSBen Gras */
67*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
68*543adbedSBen Gras #include <io.h>
69*543adbedSBen Gras #include <windows.h>
70*543adbedSBen Gras #ifndef F_OK
71*543adbedSBen Gras #define F_OK (0)
72*543adbedSBen Gras #endif
73*543adbedSBen Gras #ifndef S_ISDIR
74*543adbedSBen Gras #define S_ISDIR(m) ((m) & _S_IFDIR)
75*543adbedSBen Gras #endif
76*543adbedSBen Gras #ifndef S_ISREG
77*543adbedSBen Gras #define S_ISREG(m) ((m) & _S_IFREG)
78*543adbedSBen Gras #endif
79*543adbedSBen Gras #if !defined(__BORLANDC__)
80*543adbedSBen Gras #define access _access
81*543adbedSBen Gras #undef chdir
82*543adbedSBen Gras #define chdir _chdir
83*543adbedSBen Gras #endif
84*543adbedSBen Gras #ifndef fileno
85*543adbedSBen Gras #define fileno _fileno
86*543adbedSBen Gras #endif
87*543adbedSBen Gras /*#define fstat _fstat64*/
88*543adbedSBen Gras #if !defined(__BORLANDC__)
89*543adbedSBen Gras #define getcwd _getcwd
90*543adbedSBen Gras #endif
91*543adbedSBen Gras #define lstat stat
92*543adbedSBen Gras /*#define lstat _stat64*/
93*543adbedSBen Gras /*#define stat _stat64*/
94*543adbedSBen Gras #define rmdir _rmdir
95*543adbedSBen Gras #if !defined(__BORLANDC__)
96*543adbedSBen Gras #define strdup _strdup
97*543adbedSBen Gras #define umask _umask
98*543adbedSBen Gras #endif
99*543adbedSBen Gras #define int64_t __int64
100*543adbedSBen Gras #endif
101*543adbedSBen Gras
102*543adbedSBen Gras #if defined(HAVE__CrtSetReportMode)
103*543adbedSBen Gras # include <crtdbg.h>
104*543adbedSBen Gras #endif
105*543adbedSBen Gras
106*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
GetFunctionKernel32(const char * name)107*543adbedSBen Gras void *GetFunctionKernel32(const char *name)
108*543adbedSBen Gras {
109*543adbedSBen Gras static HINSTANCE lib;
110*543adbedSBen Gras static int set;
111*543adbedSBen Gras if (!set) {
112*543adbedSBen Gras set = 1;
113*543adbedSBen Gras lib = LoadLibrary("kernel32.dll");
114*543adbedSBen Gras }
115*543adbedSBen Gras if (lib == NULL) {
116*543adbedSBen Gras fprintf(stderr, "Can't load kernel32.dll?!\n");
117*543adbedSBen Gras exit(1);
118*543adbedSBen Gras }
119*543adbedSBen Gras return (void *)GetProcAddress(lib, name);
120*543adbedSBen Gras }
121*543adbedSBen Gras
122*543adbedSBen Gras static int
my_CreateSymbolicLinkA(const char * linkname,const char * target,int flags)123*543adbedSBen Gras my_CreateSymbolicLinkA(const char *linkname, const char *target, int flags)
124*543adbedSBen Gras {
125*543adbedSBen Gras static BOOLEAN (WINAPI *f)(LPCSTR, LPCSTR, DWORD);
126*543adbedSBen Gras static int set;
127*543adbedSBen Gras if (!set) {
128*543adbedSBen Gras set = 1;
129*543adbedSBen Gras f = GetFunctionKernel32("CreateSymbolicLinkA");
130*543adbedSBen Gras }
131*543adbedSBen Gras return f == NULL ? 0 : (*f)(linkname, target, flags);
132*543adbedSBen Gras }
133*543adbedSBen Gras
134*543adbedSBen Gras static int
my_CreateHardLinkA(const char * linkname,const char * target)135*543adbedSBen Gras my_CreateHardLinkA(const char *linkname, const char *target)
136*543adbedSBen Gras {
137*543adbedSBen Gras static BOOLEAN (WINAPI *f)(LPCSTR, LPCSTR, LPSECURITY_ATTRIBUTES);
138*543adbedSBen Gras static int set;
139*543adbedSBen Gras if (!set) {
140*543adbedSBen Gras set = 1;
141*543adbedSBen Gras f = GetFunctionKernel32("CreateHardLinkA");
142*543adbedSBen Gras }
143*543adbedSBen Gras return f == NULL ? 0 : (*f)(linkname, target, NULL);
144*543adbedSBen Gras }
145*543adbedSBen Gras
146*543adbedSBen Gras int
my_GetFileInformationByName(const char * path,BY_HANDLE_FILE_INFORMATION * bhfi)147*543adbedSBen Gras my_GetFileInformationByName(const char *path, BY_HANDLE_FILE_INFORMATION *bhfi)
148*543adbedSBen Gras {
149*543adbedSBen Gras HANDLE h;
150*543adbedSBen Gras int r;
151*543adbedSBen Gras
152*543adbedSBen Gras memset(bhfi, 0, sizeof(*bhfi));
153*543adbedSBen Gras h = CreateFile(path, FILE_READ_ATTRIBUTES, 0, NULL,
154*543adbedSBen Gras OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
155*543adbedSBen Gras if (h == INVALID_HANDLE_VALUE)
156*543adbedSBen Gras return (0);
157*543adbedSBen Gras r = GetFileInformationByHandle(h, bhfi);
158*543adbedSBen Gras CloseHandle(h);
159*543adbedSBen Gras return (r);
160*543adbedSBen Gras }
161*543adbedSBen Gras #endif
162*543adbedSBen Gras
163*543adbedSBen Gras #if defined(HAVE__CrtSetReportMode)
164*543adbedSBen Gras static void
invalid_parameter_handler(const wchar_t * expression,const wchar_t * function,const wchar_t * file,unsigned int line,uintptr_t pReserved)165*543adbedSBen Gras invalid_parameter_handler(const wchar_t * expression,
166*543adbedSBen Gras const wchar_t * function, const wchar_t * file,
167*543adbedSBen Gras unsigned int line, uintptr_t pReserved)
168*543adbedSBen Gras {
169*543adbedSBen Gras /* nop */
170*543adbedSBen Gras }
171*543adbedSBen Gras #endif
172*543adbedSBen Gras
173*543adbedSBen Gras /*
174*543adbedSBen Gras *
175*543adbedSBen Gras * OPTIONS FLAGS
176*543adbedSBen Gras *
177*543adbedSBen Gras */
178*543adbedSBen Gras
179*543adbedSBen Gras /* Enable core dump on failure. */
180*543adbedSBen Gras static int dump_on_failure = 0;
181*543adbedSBen Gras /* Default is to remove temp dirs and log data for successful tests. */
182*543adbedSBen Gras static int keep_temp_files = 0;
183*543adbedSBen Gras /* Default is to just report pass/fail for each test. */
184*543adbedSBen Gras static int verbosity = 0;
185*543adbedSBen Gras #define VERBOSITY_SUMMARY_ONLY -1 /* -q */
186*543adbedSBen Gras #define VERBOSITY_PASSFAIL 0 /* Default */
187*543adbedSBen Gras #define VERBOSITY_LIGHT_REPORT 1 /* -v */
188*543adbedSBen Gras #define VERBOSITY_FULL 2 /* -vv */
189*543adbedSBen Gras /* A few places generate even more output for verbosity > VERBOSITY_FULL,
190*543adbedSBen Gras * mostly for debugging the test harness itself. */
191*543adbedSBen Gras /* Cumulative count of assertion failures. */
192*543adbedSBen Gras static int failures = 0;
193*543adbedSBen Gras /* Cumulative count of reported skips. */
194*543adbedSBen Gras static int skips = 0;
195*543adbedSBen Gras /* Cumulative count of assertions checked. */
196*543adbedSBen Gras static int assertions = 0;
197*543adbedSBen Gras
198*543adbedSBen Gras /* Directory where uuencoded reference files can be found. */
199*543adbedSBen Gras static const char *refdir;
200*543adbedSBen Gras
201*543adbedSBen Gras /*
202*543adbedSBen Gras * Report log information selectively to console and/or disk log.
203*543adbedSBen Gras */
204*543adbedSBen Gras static int log_console = 0;
205*543adbedSBen Gras static FILE *logfile;
206*543adbedSBen Gras static void
vlogprintf(const char * fmt,va_list ap)207*543adbedSBen Gras vlogprintf(const char *fmt, va_list ap)
208*543adbedSBen Gras {
209*543adbedSBen Gras #ifdef va_copy
210*543adbedSBen Gras va_list lfap;
211*543adbedSBen Gras va_copy(lfap, ap);
212*543adbedSBen Gras #endif
213*543adbedSBen Gras if (log_console)
214*543adbedSBen Gras vfprintf(stdout, fmt, ap);
215*543adbedSBen Gras if (logfile != NULL)
216*543adbedSBen Gras #ifdef va_copy
217*543adbedSBen Gras vfprintf(logfile, fmt, lfap);
218*543adbedSBen Gras va_end(lfap);
219*543adbedSBen Gras #else
220*543adbedSBen Gras vfprintf(logfile, fmt, ap);
221*543adbedSBen Gras #endif
222*543adbedSBen Gras }
223*543adbedSBen Gras
224*543adbedSBen Gras static void
logprintf(const char * fmt,...)225*543adbedSBen Gras logprintf(const char *fmt, ...)
226*543adbedSBen Gras {
227*543adbedSBen Gras va_list ap;
228*543adbedSBen Gras va_start(ap, fmt);
229*543adbedSBen Gras vlogprintf(fmt, ap);
230*543adbedSBen Gras va_end(ap);
231*543adbedSBen Gras }
232*543adbedSBen Gras
233*543adbedSBen Gras /* Set up a message to display only if next assertion fails. */
234*543adbedSBen Gras static char msgbuff[4096];
235*543adbedSBen Gras static const char *msg, *nextmsg;
236*543adbedSBen Gras void
failure(const char * fmt,...)237*543adbedSBen Gras failure(const char *fmt, ...)
238*543adbedSBen Gras {
239*543adbedSBen Gras va_list ap;
240*543adbedSBen Gras va_start(ap, fmt);
241*543adbedSBen Gras vsprintf(msgbuff, fmt, ap);
242*543adbedSBen Gras va_end(ap);
243*543adbedSBen Gras nextmsg = msgbuff;
244*543adbedSBen Gras }
245*543adbedSBen Gras
246*543adbedSBen Gras /*
247*543adbedSBen Gras * Copy arguments into file-local variables.
248*543adbedSBen Gras * This was added to permit vararg assert() functions without needing
249*543adbedSBen Gras * variadic wrapper macros. Turns out that the vararg capability is almost
250*543adbedSBen Gras * never used, so almost all of the vararg assertions can be simplified
251*543adbedSBen Gras * by removing the vararg capability and reworking the wrapper macro to
252*543adbedSBen Gras * pass __FILE__, __LINE__ directly into the function instead of using
253*543adbedSBen Gras * this hook. I suspect this machinery is used so rarely that we
254*543adbedSBen Gras * would be better off just removing it entirely. That would simplify
255*543adbedSBen Gras * the code here noticably.
256*543adbedSBen Gras */
257*543adbedSBen Gras static const char *test_filename;
258*543adbedSBen Gras static int test_line;
259*543adbedSBen Gras static void *test_extra;
assertion_setup(const char * filename,int line)260*543adbedSBen Gras void assertion_setup(const char *filename, int line)
261*543adbedSBen Gras {
262*543adbedSBen Gras test_filename = filename;
263*543adbedSBen Gras test_line = line;
264*543adbedSBen Gras }
265*543adbedSBen Gras
266*543adbedSBen Gras /* Called at the beginning of each assert() function. */
267*543adbedSBen Gras static void
assertion_count(const char * file,int line)268*543adbedSBen Gras assertion_count(const char *file, int line)
269*543adbedSBen Gras {
270*543adbedSBen Gras (void)file; /* UNUSED */
271*543adbedSBen Gras (void)line; /* UNUSED */
272*543adbedSBen Gras ++assertions;
273*543adbedSBen Gras /* Proper handling of "failure()" message. */
274*543adbedSBen Gras msg = nextmsg;
275*543adbedSBen Gras nextmsg = NULL;
276*543adbedSBen Gras /* Uncomment to print file:line after every assertion.
277*543adbedSBen Gras * Verbose, but occasionally useful in tracking down crashes. */
278*543adbedSBen Gras /* printf("Checked %s:%d\n", file, line); */
279*543adbedSBen Gras }
280*543adbedSBen Gras
281*543adbedSBen Gras /*
282*543adbedSBen Gras * For each test source file, we remember how many times each
283*543adbedSBen Gras * assertion was reported. Cleared before each new test,
284*543adbedSBen Gras * used by test_summarize().
285*543adbedSBen Gras */
286*543adbedSBen Gras static struct line {
287*543adbedSBen Gras int count;
288*543adbedSBen Gras int skip;
289*543adbedSBen Gras } failed_lines[10000];
290*543adbedSBen Gras
291*543adbedSBen Gras /* Count this failure, setup up log destination and handle initial report. */
292*543adbedSBen Gras static void
failure_start(const char * filename,int line,const char * fmt,...)293*543adbedSBen Gras failure_start(const char *filename, int line, const char *fmt, ...)
294*543adbedSBen Gras {
295*543adbedSBen Gras va_list ap;
296*543adbedSBen Gras
297*543adbedSBen Gras /* Record another failure for this line. */
298*543adbedSBen Gras ++failures;
299*543adbedSBen Gras /* test_filename = filename; */
300*543adbedSBen Gras failed_lines[line].count++;
301*543adbedSBen Gras
302*543adbedSBen Gras /* Determine whether to log header to console. */
303*543adbedSBen Gras switch (verbosity) {
304*543adbedSBen Gras case VERBOSITY_FULL:
305*543adbedSBen Gras log_console = 1;
306*543adbedSBen Gras break;
307*543adbedSBen Gras case VERBOSITY_LIGHT_REPORT:
308*543adbedSBen Gras log_console = (failed_lines[line].count < 2);
309*543adbedSBen Gras break;
310*543adbedSBen Gras default:
311*543adbedSBen Gras log_console = 0;
312*543adbedSBen Gras }
313*543adbedSBen Gras
314*543adbedSBen Gras /* Log file:line header for this failure */
315*543adbedSBen Gras va_start(ap, fmt);
316*543adbedSBen Gras #if _MSC_VER
317*543adbedSBen Gras logprintf("%s(%d): ", filename, line);
318*543adbedSBen Gras #else
319*543adbedSBen Gras logprintf("%s:%d: ", filename, line);
320*543adbedSBen Gras #endif
321*543adbedSBen Gras vlogprintf(fmt, ap);
322*543adbedSBen Gras va_end(ap);
323*543adbedSBen Gras logprintf("\n");
324*543adbedSBen Gras
325*543adbedSBen Gras if (msg != NULL && msg[0] != '\0') {
326*543adbedSBen Gras logprintf(" Description: %s\n", msg);
327*543adbedSBen Gras msg = NULL;
328*543adbedSBen Gras }
329*543adbedSBen Gras
330*543adbedSBen Gras /* Determine whether to log details to console. */
331*543adbedSBen Gras if (verbosity == VERBOSITY_LIGHT_REPORT)
332*543adbedSBen Gras log_console = 0;
333*543adbedSBen Gras }
334*543adbedSBen Gras
335*543adbedSBen Gras /* Complete reporting of failed tests. */
336*543adbedSBen Gras /*
337*543adbedSBen Gras * The 'extra' hook here is used by libarchive to include libarchive
338*543adbedSBen Gras * error messages with assertion failures. It could also be used
339*543adbedSBen Gras * to add strerror() output, for example. Just define the EXTRA_DUMP()
340*543adbedSBen Gras * macro appropriately.
341*543adbedSBen Gras */
342*543adbedSBen Gras static void
failure_finish(void * extra)343*543adbedSBen Gras failure_finish(void *extra)
344*543adbedSBen Gras {
345*543adbedSBen Gras (void)extra; /* UNUSED (maybe) */
346*543adbedSBen Gras #ifdef EXTRA_DUMP
347*543adbedSBen Gras if (extra != NULL)
348*543adbedSBen Gras logprintf(" detail: %s\n", EXTRA_DUMP(extra));
349*543adbedSBen Gras #endif
350*543adbedSBen Gras
351*543adbedSBen Gras if (dump_on_failure) {
352*543adbedSBen Gras fprintf(stderr,
353*543adbedSBen Gras " *** forcing core dump so failure can be debugged ***\n");
354*543adbedSBen Gras *(char *)(NULL) = 0;
355*543adbedSBen Gras exit(1);
356*543adbedSBen Gras }
357*543adbedSBen Gras }
358*543adbedSBen Gras
359*543adbedSBen Gras /* Inform user that we're skipping some checks. */
360*543adbedSBen Gras void
test_skipping(const char * fmt,...)361*543adbedSBen Gras test_skipping(const char *fmt, ...)
362*543adbedSBen Gras {
363*543adbedSBen Gras char buff[1024];
364*543adbedSBen Gras va_list ap;
365*543adbedSBen Gras
366*543adbedSBen Gras va_start(ap, fmt);
367*543adbedSBen Gras vsprintf(buff, fmt, ap);
368*543adbedSBen Gras va_end(ap);
369*543adbedSBen Gras /* failure_start() isn't quite right, but is awfully convenient. */
370*543adbedSBen Gras failure_start(test_filename, test_line, "SKIPPING: %s", buff);
371*543adbedSBen Gras --failures; /* Undo failures++ in failure_start() */
372*543adbedSBen Gras /* Don't failure_finish() here. */
373*543adbedSBen Gras /* Mark as skip, so doesn't count as failed test. */
374*543adbedSBen Gras failed_lines[test_line].skip = 1;
375*543adbedSBen Gras ++skips;
376*543adbedSBen Gras }
377*543adbedSBen Gras
378*543adbedSBen Gras /*
379*543adbedSBen Gras *
380*543adbedSBen Gras * ASSERTIONS
381*543adbedSBen Gras *
382*543adbedSBen Gras */
383*543adbedSBen Gras
384*543adbedSBen Gras /* Generic assert() just displays the failed condition. */
385*543adbedSBen Gras int
assertion_assert(const char * file,int line,int value,const char * condition,void * extra)386*543adbedSBen Gras assertion_assert(const char *file, int line, int value,
387*543adbedSBen Gras const char *condition, void *extra)
388*543adbedSBen Gras {
389*543adbedSBen Gras assertion_count(file, line);
390*543adbedSBen Gras if (!value) {
391*543adbedSBen Gras failure_start(file, line, "Assertion failed: %s", condition);
392*543adbedSBen Gras failure_finish(extra);
393*543adbedSBen Gras }
394*543adbedSBen Gras return (value);
395*543adbedSBen Gras }
396*543adbedSBen Gras
397*543adbedSBen Gras /* chdir() and report any errors */
398*543adbedSBen Gras int
assertion_chdir(const char * file,int line,const char * pathname)399*543adbedSBen Gras assertion_chdir(const char *file, int line, const char *pathname)
400*543adbedSBen Gras {
401*543adbedSBen Gras assertion_count(file, line);
402*543adbedSBen Gras if (chdir(pathname) == 0)
403*543adbedSBen Gras return (1);
404*543adbedSBen Gras failure_start(file, line, "chdir(\"%s\")", pathname);
405*543adbedSBen Gras failure_finish(NULL);
406*543adbedSBen Gras return (0);
407*543adbedSBen Gras
408*543adbedSBen Gras }
409*543adbedSBen Gras
410*543adbedSBen Gras /* Verify two integers are equal. */
411*543adbedSBen Gras int
assertion_equal_int(const char * file,int line,long long v1,const char * e1,long long v2,const char * e2,void * extra)412*543adbedSBen Gras assertion_equal_int(const char *file, int line,
413*543adbedSBen Gras long long v1, const char *e1, long long v2, const char *e2, void *extra)
414*543adbedSBen Gras {
415*543adbedSBen Gras assertion_count(file, line);
416*543adbedSBen Gras if (v1 == v2)
417*543adbedSBen Gras return (1);
418*543adbedSBen Gras failure_start(file, line, "%s != %s", e1, e2);
419*543adbedSBen Gras logprintf(" %s=%lld (0x%llx, 0%llo)\n", e1, v1, v1, v1);
420*543adbedSBen Gras logprintf(" %s=%lld (0x%llx, 0%llo)\n", e2, v2, v2, v2);
421*543adbedSBen Gras failure_finish(extra);
422*543adbedSBen Gras return (0);
423*543adbedSBen Gras }
424*543adbedSBen Gras
strdump(const char * e,const char * p)425*543adbedSBen Gras static void strdump(const char *e, const char *p)
426*543adbedSBen Gras {
427*543adbedSBen Gras const char *q = p;
428*543adbedSBen Gras
429*543adbedSBen Gras logprintf(" %s = ", e);
430*543adbedSBen Gras if (p == NULL) {
431*543adbedSBen Gras logprintf("NULL");
432*543adbedSBen Gras return;
433*543adbedSBen Gras }
434*543adbedSBen Gras logprintf("\"");
435*543adbedSBen Gras while (*p != '\0') {
436*543adbedSBen Gras unsigned int c = 0xff & *p++;
437*543adbedSBen Gras switch (c) {
438*543adbedSBen Gras case '\a': printf("\a"); break;
439*543adbedSBen Gras case '\b': printf("\b"); break;
440*543adbedSBen Gras case '\n': printf("\n"); break;
441*543adbedSBen Gras case '\r': printf("\r"); break;
442*543adbedSBen Gras default:
443*543adbedSBen Gras if (c >= 32 && c < 127)
444*543adbedSBen Gras logprintf("%c", c);
445*543adbedSBen Gras else
446*543adbedSBen Gras logprintf("\\x%02X", c);
447*543adbedSBen Gras }
448*543adbedSBen Gras }
449*543adbedSBen Gras logprintf("\"");
450*543adbedSBen Gras logprintf(" (length %d)\n", q == NULL ? -1 : (int)strlen(q));
451*543adbedSBen Gras }
452*543adbedSBen Gras
453*543adbedSBen Gras /* Verify two strings are equal, dump them if not. */
454*543adbedSBen Gras int
assertion_equal_string(const char * file,int line,const char * v1,const char * e1,const char * v2,const char * e2,void * extra)455*543adbedSBen Gras assertion_equal_string(const char *file, int line,
456*543adbedSBen Gras const char *v1, const char *e1,
457*543adbedSBen Gras const char *v2, const char *e2,
458*543adbedSBen Gras void *extra)
459*543adbedSBen Gras {
460*543adbedSBen Gras assertion_count(file, line);
461*543adbedSBen Gras if (v1 == v2 || (v1 != NULL && v2 != NULL && strcmp(v1, v2) == 0))
462*543adbedSBen Gras return (1);
463*543adbedSBen Gras failure_start(file, line, "%s != %s", e1, e2);
464*543adbedSBen Gras strdump(e1, v1);
465*543adbedSBen Gras strdump(e2, v2);
466*543adbedSBen Gras failure_finish(extra);
467*543adbedSBen Gras return (0);
468*543adbedSBen Gras }
469*543adbedSBen Gras
470*543adbedSBen Gras static void
wcsdump(const char * e,const wchar_t * w)471*543adbedSBen Gras wcsdump(const char *e, const wchar_t *w)
472*543adbedSBen Gras {
473*543adbedSBen Gras logprintf(" %s = ", e);
474*543adbedSBen Gras if (w == NULL) {
475*543adbedSBen Gras logprintf("(null)");
476*543adbedSBen Gras return;
477*543adbedSBen Gras }
478*543adbedSBen Gras logprintf("\"");
479*543adbedSBen Gras while (*w != L'\0') {
480*543adbedSBen Gras unsigned int c = *w++;
481*543adbedSBen Gras if (c >= 32 && c < 127)
482*543adbedSBen Gras logprintf("%c", c);
483*543adbedSBen Gras else if (c < 256)
484*543adbedSBen Gras logprintf("\\x%02X", c);
485*543adbedSBen Gras else if (c < 0x10000)
486*543adbedSBen Gras logprintf("\\u%04X", c);
487*543adbedSBen Gras else
488*543adbedSBen Gras logprintf("\\U%08X", c);
489*543adbedSBen Gras }
490*543adbedSBen Gras logprintf("\"\n");
491*543adbedSBen Gras }
492*543adbedSBen Gras
493*543adbedSBen Gras #ifndef HAVE_WCSCMP
494*543adbedSBen Gras static int
wcscmp(const wchar_t * s1,const wchar_t * s2)495*543adbedSBen Gras wcscmp(const wchar_t *s1, const wchar_t *s2)
496*543adbedSBen Gras {
497*543adbedSBen Gras
498*543adbedSBen Gras while (*s1 == *s2++) {
499*543adbedSBen Gras if (*s1++ == L'\0')
500*543adbedSBen Gras return 0;
501*543adbedSBen Gras }
502*543adbedSBen Gras if (*s1 > *--s2)
503*543adbedSBen Gras return 1;
504*543adbedSBen Gras else
505*543adbedSBen Gras return -1;
506*543adbedSBen Gras }
507*543adbedSBen Gras #endif
508*543adbedSBen Gras
509*543adbedSBen Gras /* Verify that two wide strings are equal, dump them if not. */
510*543adbedSBen Gras int
assertion_equal_wstring(const char * file,int line,const wchar_t * v1,const char * e1,const wchar_t * v2,const char * e2,void * extra)511*543adbedSBen Gras assertion_equal_wstring(const char *file, int line,
512*543adbedSBen Gras const wchar_t *v1, const char *e1,
513*543adbedSBen Gras const wchar_t *v2, const char *e2,
514*543adbedSBen Gras void *extra)
515*543adbedSBen Gras {
516*543adbedSBen Gras assertion_count(file, line);
517*543adbedSBen Gras if (v1 == v2 || wcscmp(v1, v2) == 0)
518*543adbedSBen Gras return (1);
519*543adbedSBen Gras failure_start(file, line, "%s != %s", e1, e2);
520*543adbedSBen Gras wcsdump(e1, v1);
521*543adbedSBen Gras wcsdump(e2, v2);
522*543adbedSBen Gras failure_finish(extra);
523*543adbedSBen Gras return (0);
524*543adbedSBen Gras }
525*543adbedSBen Gras
526*543adbedSBen Gras /*
527*543adbedSBen Gras * Pretty standard hexdump routine. As a bonus, if ref != NULL, then
528*543adbedSBen Gras * any bytes in p that differ from ref will be highlighted with '_'
529*543adbedSBen Gras * before and after the hex value.
530*543adbedSBen Gras */
531*543adbedSBen Gras static void
hexdump(const char * p,const char * ref,size_t l,size_t offset)532*543adbedSBen Gras hexdump(const char *p, const char *ref, size_t l, size_t offset)
533*543adbedSBen Gras {
534*543adbedSBen Gras size_t i, j;
535*543adbedSBen Gras char sep;
536*543adbedSBen Gras
537*543adbedSBen Gras if (p == NULL) {
538*543adbedSBen Gras logprintf("(null)\n");
539*543adbedSBen Gras return;
540*543adbedSBen Gras }
541*543adbedSBen Gras for(i=0; i < l; i+=16) {
542*543adbedSBen Gras logprintf("%04x", (unsigned)(i + offset));
543*543adbedSBen Gras sep = ' ';
544*543adbedSBen Gras for (j = 0; j < 16 && i + j < l; j++) {
545*543adbedSBen Gras if (ref != NULL && p[i + j] != ref[i + j])
546*543adbedSBen Gras sep = '_';
547*543adbedSBen Gras logprintf("%c%02x", sep, 0xff & (int)p[i+j]);
548*543adbedSBen Gras if (ref != NULL && p[i + j] == ref[i + j])
549*543adbedSBen Gras sep = ' ';
550*543adbedSBen Gras }
551*543adbedSBen Gras for (; j < 16; j++) {
552*543adbedSBen Gras logprintf("%c ", sep);
553*543adbedSBen Gras sep = ' ';
554*543adbedSBen Gras }
555*543adbedSBen Gras logprintf("%c", sep);
556*543adbedSBen Gras for (j=0; j < 16 && i + j < l; j++) {
557*543adbedSBen Gras int c = p[i + j];
558*543adbedSBen Gras if (c >= ' ' && c <= 126)
559*543adbedSBen Gras logprintf("%c", c);
560*543adbedSBen Gras else
561*543adbedSBen Gras logprintf(".");
562*543adbedSBen Gras }
563*543adbedSBen Gras logprintf("\n");
564*543adbedSBen Gras }
565*543adbedSBen Gras }
566*543adbedSBen Gras
567*543adbedSBen Gras /* Verify that two blocks of memory are the same, display the first
568*543adbedSBen Gras * block of differences if they're not. */
569*543adbedSBen Gras int
assertion_equal_mem(const char * file,int line,const void * _v1,const char * e1,const void * _v2,const char * e2,size_t l,const char * ld,void * extra)570*543adbedSBen Gras assertion_equal_mem(const char *file, int line,
571*543adbedSBen Gras const void *_v1, const char *e1,
572*543adbedSBen Gras const void *_v2, const char *e2,
573*543adbedSBen Gras size_t l, const char *ld, void *extra)
574*543adbedSBen Gras {
575*543adbedSBen Gras const char *v1 = (const char *)_v1;
576*543adbedSBen Gras const char *v2 = (const char *)_v2;
577*543adbedSBen Gras size_t offset;
578*543adbedSBen Gras
579*543adbedSBen Gras assertion_count(file, line);
580*543adbedSBen Gras if (v1 == v2 || (v1 != NULL && v2 != NULL && memcmp(v1, v2, l) == 0))
581*543adbedSBen Gras return (1);
582*543adbedSBen Gras
583*543adbedSBen Gras failure_start(file, line, "%s != %s", e1, e2);
584*543adbedSBen Gras logprintf(" size %s = %d\n", ld, (int)l);
585*543adbedSBen Gras /* Dump 48 bytes (3 lines) so that the first difference is
586*543adbedSBen Gras * in the second line. */
587*543adbedSBen Gras offset = 0;
588*543adbedSBen Gras while (l > 64 && memcmp(v1, v2, 32) == 0) {
589*543adbedSBen Gras /* Two lines agree, so step forward one line. */
590*543adbedSBen Gras v1 += 16;
591*543adbedSBen Gras v2 += 16;
592*543adbedSBen Gras l -= 16;
593*543adbedSBen Gras offset += 16;
594*543adbedSBen Gras }
595*543adbedSBen Gras logprintf(" Dump of %s\n", e1);
596*543adbedSBen Gras hexdump(v1, v2, l < 64 ? l : 64, offset);
597*543adbedSBen Gras logprintf(" Dump of %s\n", e2);
598*543adbedSBen Gras hexdump(v2, v1, l < 64 ? l : 64, offset);
599*543adbedSBen Gras logprintf("\n");
600*543adbedSBen Gras failure_finish(extra);
601*543adbedSBen Gras return (0);
602*543adbedSBen Gras }
603*543adbedSBen Gras
604*543adbedSBen Gras /* Verify that the named file exists and is empty. */
605*543adbedSBen Gras int
assertion_empty_file(const char * f1fmt,...)606*543adbedSBen Gras assertion_empty_file(const char *f1fmt, ...)
607*543adbedSBen Gras {
608*543adbedSBen Gras char buff[1024];
609*543adbedSBen Gras char f1[1024];
610*543adbedSBen Gras struct stat st;
611*543adbedSBen Gras va_list ap;
612*543adbedSBen Gras ssize_t s;
613*543adbedSBen Gras FILE *f;
614*543adbedSBen Gras
615*543adbedSBen Gras assertion_count(test_filename, test_line);
616*543adbedSBen Gras va_start(ap, f1fmt);
617*543adbedSBen Gras vsprintf(f1, f1fmt, ap);
618*543adbedSBen Gras va_end(ap);
619*543adbedSBen Gras
620*543adbedSBen Gras if (stat(f1, &st) != 0) {
621*543adbedSBen Gras failure_start(test_filename, test_line, "Stat failed: %s", f1);
622*543adbedSBen Gras failure_finish(NULL);
623*543adbedSBen Gras return (0);
624*543adbedSBen Gras }
625*543adbedSBen Gras if (st.st_size == 0)
626*543adbedSBen Gras return (1);
627*543adbedSBen Gras
628*543adbedSBen Gras failure_start(test_filename, test_line, "File should be empty: %s", f1);
629*543adbedSBen Gras logprintf(" File size: %d\n", (int)st.st_size);
630*543adbedSBen Gras logprintf(" Contents:\n");
631*543adbedSBen Gras f = fopen(f1, "rb");
632*543adbedSBen Gras if (f == NULL) {
633*543adbedSBen Gras logprintf(" Unable to open %s\n", f1);
634*543adbedSBen Gras } else {
635*543adbedSBen Gras s = ((off_t)sizeof(buff) < st.st_size) ?
636*543adbedSBen Gras (ssize_t)sizeof(buff) : (ssize_t)st.st_size;
637*543adbedSBen Gras s = fread(buff, 1, s, f);
638*543adbedSBen Gras hexdump(buff, NULL, s, 0);
639*543adbedSBen Gras fclose(f);
640*543adbedSBen Gras }
641*543adbedSBen Gras failure_finish(NULL);
642*543adbedSBen Gras return (0);
643*543adbedSBen Gras }
644*543adbedSBen Gras
645*543adbedSBen Gras /* Verify that the named file exists and is not empty. */
646*543adbedSBen Gras int
assertion_non_empty_file(const char * f1fmt,...)647*543adbedSBen Gras assertion_non_empty_file(const char *f1fmt, ...)
648*543adbedSBen Gras {
649*543adbedSBen Gras char f1[1024];
650*543adbedSBen Gras struct stat st;
651*543adbedSBen Gras va_list ap;
652*543adbedSBen Gras
653*543adbedSBen Gras assertion_count(test_filename, test_line);
654*543adbedSBen Gras va_start(ap, f1fmt);
655*543adbedSBen Gras vsprintf(f1, f1fmt, ap);
656*543adbedSBen Gras va_end(ap);
657*543adbedSBen Gras
658*543adbedSBen Gras if (stat(f1, &st) != 0) {
659*543adbedSBen Gras failure_start(test_filename, test_line, "Stat failed: %s", f1);
660*543adbedSBen Gras failure_finish(NULL);
661*543adbedSBen Gras return (0);
662*543adbedSBen Gras }
663*543adbedSBen Gras if (st.st_size == 0) {
664*543adbedSBen Gras failure_start(test_filename, test_line, "File empty: %s", f1);
665*543adbedSBen Gras failure_finish(NULL);
666*543adbedSBen Gras return (0);
667*543adbedSBen Gras }
668*543adbedSBen Gras return (1);
669*543adbedSBen Gras }
670*543adbedSBen Gras
671*543adbedSBen Gras /* Verify that two files have the same contents. */
672*543adbedSBen Gras /* TODO: hexdump the first bytes that actually differ. */
673*543adbedSBen Gras int
assertion_equal_file(const char * fn1,const char * f2pattern,...)674*543adbedSBen Gras assertion_equal_file(const char *fn1, const char *f2pattern, ...)
675*543adbedSBen Gras {
676*543adbedSBen Gras char fn2[1024];
677*543adbedSBen Gras va_list ap;
678*543adbedSBen Gras char buff1[1024];
679*543adbedSBen Gras char buff2[1024];
680*543adbedSBen Gras FILE *f1, *f2;
681*543adbedSBen Gras int n1, n2;
682*543adbedSBen Gras
683*543adbedSBen Gras assertion_count(test_filename, test_line);
684*543adbedSBen Gras va_start(ap, f2pattern);
685*543adbedSBen Gras vsprintf(fn2, f2pattern, ap);
686*543adbedSBen Gras va_end(ap);
687*543adbedSBen Gras
688*543adbedSBen Gras f1 = fopen(fn1, "rb");
689*543adbedSBen Gras f2 = fopen(fn2, "rb");
690*543adbedSBen Gras for (;;) {
691*543adbedSBen Gras n1 = fread(buff1, 1, sizeof(buff1), f1);
692*543adbedSBen Gras n2 = fread(buff2, 1, sizeof(buff2), f2);
693*543adbedSBen Gras if (n1 != n2)
694*543adbedSBen Gras break;
695*543adbedSBen Gras if (n1 == 0 && n2 == 0) {
696*543adbedSBen Gras fclose(f1);
697*543adbedSBen Gras fclose(f2);
698*543adbedSBen Gras return (1);
699*543adbedSBen Gras }
700*543adbedSBen Gras if (memcmp(buff1, buff2, n1) != 0)
701*543adbedSBen Gras break;
702*543adbedSBen Gras }
703*543adbedSBen Gras fclose(f1);
704*543adbedSBen Gras fclose(f2);
705*543adbedSBen Gras failure_start(test_filename, test_line, "Files not identical");
706*543adbedSBen Gras logprintf(" file1=\"%s\"\n", fn1);
707*543adbedSBen Gras logprintf(" file2=\"%s\"\n", fn2);
708*543adbedSBen Gras failure_finish(test_extra);
709*543adbedSBen Gras return (0);
710*543adbedSBen Gras }
711*543adbedSBen Gras
712*543adbedSBen Gras /* Verify that the named file does exist. */
713*543adbedSBen Gras int
assertion_file_exists(const char * fpattern,...)714*543adbedSBen Gras assertion_file_exists(const char *fpattern, ...)
715*543adbedSBen Gras {
716*543adbedSBen Gras char f[1024];
717*543adbedSBen Gras va_list ap;
718*543adbedSBen Gras
719*543adbedSBen Gras assertion_count(test_filename, test_line);
720*543adbedSBen Gras va_start(ap, fpattern);
721*543adbedSBen Gras vsprintf(f, fpattern, ap);
722*543adbedSBen Gras va_end(ap);
723*543adbedSBen Gras
724*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
725*543adbedSBen Gras if (!_access(f, 0))
726*543adbedSBen Gras return (1);
727*543adbedSBen Gras #else
728*543adbedSBen Gras if (!access(f, F_OK))
729*543adbedSBen Gras return (1);
730*543adbedSBen Gras #endif
731*543adbedSBen Gras failure_start(test_filename, test_line, "File should exist: %s", f);
732*543adbedSBen Gras failure_finish(test_extra);
733*543adbedSBen Gras return (0);
734*543adbedSBen Gras }
735*543adbedSBen Gras
736*543adbedSBen Gras /* Verify that the named file doesn't exist. */
737*543adbedSBen Gras int
assertion_file_not_exists(const char * fpattern,...)738*543adbedSBen Gras assertion_file_not_exists(const char *fpattern, ...)
739*543adbedSBen Gras {
740*543adbedSBen Gras char f[1024];
741*543adbedSBen Gras va_list ap;
742*543adbedSBen Gras
743*543adbedSBen Gras assertion_count(test_filename, test_line);
744*543adbedSBen Gras va_start(ap, fpattern);
745*543adbedSBen Gras vsprintf(f, fpattern, ap);
746*543adbedSBen Gras va_end(ap);
747*543adbedSBen Gras
748*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
749*543adbedSBen Gras if (_access(f, 0))
750*543adbedSBen Gras return (1);
751*543adbedSBen Gras #else
752*543adbedSBen Gras if (access(f, F_OK))
753*543adbedSBen Gras return (1);
754*543adbedSBen Gras #endif
755*543adbedSBen Gras failure_start(test_filename, test_line, "File should not exist: %s", f);
756*543adbedSBen Gras failure_finish(test_extra);
757*543adbedSBen Gras return (0);
758*543adbedSBen Gras }
759*543adbedSBen Gras
760*543adbedSBen Gras /* Compare the contents of a file to a block of memory. */
761*543adbedSBen Gras int
assertion_file_contents(const void * buff,int s,const char * fpattern,...)762*543adbedSBen Gras assertion_file_contents(const void *buff, int s, const char *fpattern, ...)
763*543adbedSBen Gras {
764*543adbedSBen Gras char fn[1024];
765*543adbedSBen Gras va_list ap;
766*543adbedSBen Gras char *contents;
767*543adbedSBen Gras FILE *f;
768*543adbedSBen Gras int n;
769*543adbedSBen Gras
770*543adbedSBen Gras assertion_count(test_filename, test_line);
771*543adbedSBen Gras va_start(ap, fpattern);
772*543adbedSBen Gras vsprintf(fn, fpattern, ap);
773*543adbedSBen Gras va_end(ap);
774*543adbedSBen Gras
775*543adbedSBen Gras f = fopen(fn, "rb");
776*543adbedSBen Gras if (f == NULL) {
777*543adbedSBen Gras failure_start(test_filename, test_line,
778*543adbedSBen Gras "File should exist: %s", fn);
779*543adbedSBen Gras failure_finish(test_extra);
780*543adbedSBen Gras return (0);
781*543adbedSBen Gras }
782*543adbedSBen Gras contents = malloc(s * 2);
783*543adbedSBen Gras n = fread(contents, 1, s * 2, f);
784*543adbedSBen Gras fclose(f);
785*543adbedSBen Gras if (n == s && memcmp(buff, contents, s) == 0) {
786*543adbedSBen Gras free(contents);
787*543adbedSBen Gras return (1);
788*543adbedSBen Gras }
789*543adbedSBen Gras failure_start(test_filename, test_line, "File contents don't match");
790*543adbedSBen Gras logprintf(" file=\"%s\"\n", fn);
791*543adbedSBen Gras if (n > 0)
792*543adbedSBen Gras hexdump(contents, buff, n > 512 ? 512 : n, 0);
793*543adbedSBen Gras else {
794*543adbedSBen Gras logprintf(" File empty, contents should be:\n");
795*543adbedSBen Gras hexdump(buff, NULL, s > 512 ? 512 : s, 0);
796*543adbedSBen Gras }
797*543adbedSBen Gras failure_finish(test_extra);
798*543adbedSBen Gras free(contents);
799*543adbedSBen Gras return (0);
800*543adbedSBen Gras }
801*543adbedSBen Gras
802*543adbedSBen Gras /* Check the contents of a text file, being tolerant of line endings. */
803*543adbedSBen Gras int
assertion_text_file_contents(const char * buff,const char * fn)804*543adbedSBen Gras assertion_text_file_contents(const char *buff, const char *fn)
805*543adbedSBen Gras {
806*543adbedSBen Gras char *contents;
807*543adbedSBen Gras const char *btxt, *ftxt;
808*543adbedSBen Gras FILE *f;
809*543adbedSBen Gras int n, s;
810*543adbedSBen Gras
811*543adbedSBen Gras assertion_count(test_filename, test_line);
812*543adbedSBen Gras f = fopen(fn, "r");
813*543adbedSBen Gras s = strlen(buff);
814*543adbedSBen Gras contents = malloc(s * 2 + 128);
815*543adbedSBen Gras n = fread(contents, 1, s * 2 + 128 - 1, f);
816*543adbedSBen Gras if (n >= 0)
817*543adbedSBen Gras contents[n] = '\0';
818*543adbedSBen Gras fclose(f);
819*543adbedSBen Gras /* Compare texts. */
820*543adbedSBen Gras btxt = buff;
821*543adbedSBen Gras ftxt = (const char *)contents;
822*543adbedSBen Gras while (*btxt != '\0' && *ftxt != '\0') {
823*543adbedSBen Gras if (*btxt == *ftxt) {
824*543adbedSBen Gras ++btxt;
825*543adbedSBen Gras ++ftxt;
826*543adbedSBen Gras continue;
827*543adbedSBen Gras }
828*543adbedSBen Gras if (btxt[0] == '\n' && ftxt[0] == '\r' && ftxt[1] == '\n') {
829*543adbedSBen Gras /* Pass over different new line characters. */
830*543adbedSBen Gras ++btxt;
831*543adbedSBen Gras ftxt += 2;
832*543adbedSBen Gras continue;
833*543adbedSBen Gras }
834*543adbedSBen Gras break;
835*543adbedSBen Gras }
836*543adbedSBen Gras if (*btxt == '\0' && *ftxt == '\0') {
837*543adbedSBen Gras free(contents);
838*543adbedSBen Gras return (1);
839*543adbedSBen Gras }
840*543adbedSBen Gras failure_start(test_filename, test_line, "Contents don't match");
841*543adbedSBen Gras logprintf(" file=\"%s\"\n", fn);
842*543adbedSBen Gras if (n > 0)
843*543adbedSBen Gras hexdump(contents, buff, n, 0);
844*543adbedSBen Gras else {
845*543adbedSBen Gras logprintf(" File empty, contents should be:\n");
846*543adbedSBen Gras hexdump(buff, NULL, s, 0);
847*543adbedSBen Gras }
848*543adbedSBen Gras failure_finish(test_extra);
849*543adbedSBen Gras free(contents);
850*543adbedSBen Gras return (0);
851*543adbedSBen Gras }
852*543adbedSBen Gras
853*543adbedSBen Gras /* Verify that a text file contains the specified lines, regardless of order */
854*543adbedSBen Gras /* This could be more efficient if we sorted both sets of lines, etc, but
855*543adbedSBen Gras * since this is used only for testing and only ever deals with a dozen or so
856*543adbedSBen Gras * lines at a time, this relatively crude approach is just fine. */
857*543adbedSBen Gras int
assertion_file_contains_lines_any_order(const char * file,int line,const char * pathname,const char * lines[])858*543adbedSBen Gras assertion_file_contains_lines_any_order(const char *file, int line,
859*543adbedSBen Gras const char *pathname, const char *lines[])
860*543adbedSBen Gras {
861*543adbedSBen Gras char *buff;
862*543adbedSBen Gras size_t buff_size;
863*543adbedSBen Gras size_t expected_count, actual_count, i, j;
864*543adbedSBen Gras char **expected;
865*543adbedSBen Gras char *p, **actual;
866*543adbedSBen Gras char c;
867*543adbedSBen Gras int expected_failure = 0, actual_failure = 0;
868*543adbedSBen Gras
869*543adbedSBen Gras assertion_count(file, line);
870*543adbedSBen Gras
871*543adbedSBen Gras buff = slurpfile(&buff_size, "%s", pathname);
872*543adbedSBen Gras if (buff == NULL) {
873*543adbedSBen Gras failure_start(pathname, line, "Can't read file: %s", pathname);
874*543adbedSBen Gras failure_finish(NULL);
875*543adbedSBen Gras return (0);
876*543adbedSBen Gras }
877*543adbedSBen Gras
878*543adbedSBen Gras // Make a copy of the provided lines and count up the expected file size.
879*543adbedSBen Gras expected_count = 0;
880*543adbedSBen Gras for (i = 0; lines[i] != NULL; ++i) {
881*543adbedSBen Gras }
882*543adbedSBen Gras expected_count = i;
883*543adbedSBen Gras expected = malloc(sizeof(char *) * expected_count);
884*543adbedSBen Gras for (i = 0; lines[i] != NULL; ++i) {
885*543adbedSBen Gras expected[i] = strdup(lines[i]);
886*543adbedSBen Gras }
887*543adbedSBen Gras
888*543adbedSBen Gras // Break the file into lines
889*543adbedSBen Gras actual_count = 0;
890*543adbedSBen Gras for (c = '\0', p = buff; p < buff + buff_size; ++p) {
891*543adbedSBen Gras if (*p == '\x0d' || *p == '\x0a')
892*543adbedSBen Gras *p = '\0';
893*543adbedSBen Gras if (c == '\0' && *p != '\0')
894*543adbedSBen Gras ++actual_count;
895*543adbedSBen Gras c = *p;
896*543adbedSBen Gras }
897*543adbedSBen Gras actual = malloc(sizeof(char *) * actual_count);
898*543adbedSBen Gras for (j = 0, p = buff; p < buff + buff_size; p += 1 + strlen(p)) {
899*543adbedSBen Gras if (*p != '\0') {
900*543adbedSBen Gras actual[j] = p;
901*543adbedSBen Gras ++j;
902*543adbedSBen Gras }
903*543adbedSBen Gras }
904*543adbedSBen Gras
905*543adbedSBen Gras // Erase matching lines from both lists
906*543adbedSBen Gras for (i = 0; i < expected_count; ++i) {
907*543adbedSBen Gras if (expected[i] == NULL)
908*543adbedSBen Gras continue;
909*543adbedSBen Gras for (j = 0; j < actual_count; ++j) {
910*543adbedSBen Gras if (actual[j] == NULL)
911*543adbedSBen Gras continue;
912*543adbedSBen Gras if (strcmp(expected[i], actual[j]) == 0) {
913*543adbedSBen Gras free(expected[i]);
914*543adbedSBen Gras expected[i] = NULL;
915*543adbedSBen Gras actual[j] = NULL;
916*543adbedSBen Gras break;
917*543adbedSBen Gras }
918*543adbedSBen Gras }
919*543adbedSBen Gras }
920*543adbedSBen Gras
921*543adbedSBen Gras // If there's anything left, it's a failure
922*543adbedSBen Gras for (i = 0; i < expected_count; ++i) {
923*543adbedSBen Gras if (expected[i] != NULL)
924*543adbedSBen Gras ++expected_failure;
925*543adbedSBen Gras }
926*543adbedSBen Gras for (j = 0; j < actual_count; ++j) {
927*543adbedSBen Gras if (actual[j] != NULL)
928*543adbedSBen Gras ++actual_failure;
929*543adbedSBen Gras }
930*543adbedSBen Gras if (expected_failure == 0 && actual_failure == 0) {
931*543adbedSBen Gras free(buff);
932*543adbedSBen Gras free(expected);
933*543adbedSBen Gras free(actual);
934*543adbedSBen Gras return (1);
935*543adbedSBen Gras }
936*543adbedSBen Gras failure_start(file, line, "File doesn't match: %s", pathname);
937*543adbedSBen Gras for (i = 0; i < expected_count; ++i) {
938*543adbedSBen Gras if (expected[i] != NULL) {
939*543adbedSBen Gras logprintf(" Expected but not present: %s\n", expected[i]);
940*543adbedSBen Gras free(expected[i]);
941*543adbedSBen Gras }
942*543adbedSBen Gras }
943*543adbedSBen Gras for (j = 0; j < actual_count; ++j) {
944*543adbedSBen Gras if (actual[j] != NULL)
945*543adbedSBen Gras logprintf(" Present but not expected: %s\n", actual[j]);
946*543adbedSBen Gras }
947*543adbedSBen Gras failure_finish(NULL);
948*543adbedSBen Gras free(buff);
949*543adbedSBen Gras free(expected);
950*543adbedSBen Gras free(actual);
951*543adbedSBen Gras return (0);
952*543adbedSBen Gras }
953*543adbedSBen Gras
954*543adbedSBen Gras /* Test that two paths point to the same file. */
955*543adbedSBen Gras /* As a side-effect, asserts that both files exist. */
956*543adbedSBen Gras static int
is_hardlink(const char * file,int line,const char * path1,const char * path2)957*543adbedSBen Gras is_hardlink(const char *file, int line,
958*543adbedSBen Gras const char *path1, const char *path2)
959*543adbedSBen Gras {
960*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
961*543adbedSBen Gras BY_HANDLE_FILE_INFORMATION bhfi1, bhfi2;
962*543adbedSBen Gras int r;
963*543adbedSBen Gras
964*543adbedSBen Gras assertion_count(file, line);
965*543adbedSBen Gras r = my_GetFileInformationByName(path1, &bhfi1);
966*543adbedSBen Gras if (r == 0) {
967*543adbedSBen Gras failure_start(file, line, "File %s can't be inspected?", path1);
968*543adbedSBen Gras failure_finish(NULL);
969*543adbedSBen Gras return (0);
970*543adbedSBen Gras }
971*543adbedSBen Gras r = my_GetFileInformationByName(path2, &bhfi2);
972*543adbedSBen Gras if (r == 0) {
973*543adbedSBen Gras failure_start(file, line, "File %s can't be inspected?", path2);
974*543adbedSBen Gras failure_finish(NULL);
975*543adbedSBen Gras return (0);
976*543adbedSBen Gras }
977*543adbedSBen Gras return (bhfi1.dwVolumeSerialNumber == bhfi2.dwVolumeSerialNumber
978*543adbedSBen Gras && bhfi1.nFileIndexHigh == bhfi2.nFileIndexHigh
979*543adbedSBen Gras && bhfi1.nFileIndexLow == bhfi2.nFileIndexLow);
980*543adbedSBen Gras #else
981*543adbedSBen Gras struct stat st1, st2;
982*543adbedSBen Gras int r;
983*543adbedSBen Gras
984*543adbedSBen Gras assertion_count(file, line);
985*543adbedSBen Gras r = lstat(path1, &st1);
986*543adbedSBen Gras if (r != 0) {
987*543adbedSBen Gras failure_start(file, line, "File should exist: %s", path1);
988*543adbedSBen Gras failure_finish(NULL);
989*543adbedSBen Gras return (0);
990*543adbedSBen Gras }
991*543adbedSBen Gras r = lstat(path2, &st2);
992*543adbedSBen Gras if (r != 0) {
993*543adbedSBen Gras failure_start(file, line, "File should exist: %s", path2);
994*543adbedSBen Gras failure_finish(NULL);
995*543adbedSBen Gras return (0);
996*543adbedSBen Gras }
997*543adbedSBen Gras return (st1.st_ino == st2.st_ino && st1.st_dev == st2.st_dev);
998*543adbedSBen Gras #endif
999*543adbedSBen Gras }
1000*543adbedSBen Gras
1001*543adbedSBen Gras int
assertion_is_hardlink(const char * file,int line,const char * path1,const char * path2)1002*543adbedSBen Gras assertion_is_hardlink(const char *file, int line,
1003*543adbedSBen Gras const char *path1, const char *path2)
1004*543adbedSBen Gras {
1005*543adbedSBen Gras if (is_hardlink(file, line, path1, path2))
1006*543adbedSBen Gras return (1);
1007*543adbedSBen Gras failure_start(file, line,
1008*543adbedSBen Gras "Files %s and %s are not hardlinked", path1, path2);
1009*543adbedSBen Gras failure_finish(NULL);
1010*543adbedSBen Gras return (0);
1011*543adbedSBen Gras }
1012*543adbedSBen Gras
1013*543adbedSBen Gras int
assertion_is_not_hardlink(const char * file,int line,const char * path1,const char * path2)1014*543adbedSBen Gras assertion_is_not_hardlink(const char *file, int line,
1015*543adbedSBen Gras const char *path1, const char *path2)
1016*543adbedSBen Gras {
1017*543adbedSBen Gras if (!is_hardlink(file, line, path1, path2))
1018*543adbedSBen Gras return (1);
1019*543adbedSBen Gras failure_start(file, line,
1020*543adbedSBen Gras "Files %s and %s should not be hardlinked", path1, path2);
1021*543adbedSBen Gras failure_finish(NULL);
1022*543adbedSBen Gras return (0);
1023*543adbedSBen Gras }
1024*543adbedSBen Gras
1025*543adbedSBen Gras /* Verify a/b/mtime of 'pathname'. */
1026*543adbedSBen Gras /* If 'recent', verify that it's within last 10 seconds. */
1027*543adbedSBen Gras static int
assertion_file_time(const char * file,int line,const char * pathname,long t,long nsec,char type,int recent)1028*543adbedSBen Gras assertion_file_time(const char *file, int line,
1029*543adbedSBen Gras const char *pathname, long t, long nsec, char type, int recent)
1030*543adbedSBen Gras {
1031*543adbedSBen Gras long long filet, filet_nsec;
1032*543adbedSBen Gras int r;
1033*543adbedSBen Gras
1034*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
1035*543adbedSBen Gras #define EPOC_TIME (116444736000000000ULL)
1036*543adbedSBen Gras FILETIME ftime, fbirthtime, fatime, fmtime;
1037*543adbedSBen Gras ULARGE_INTEGER wintm;
1038*543adbedSBen Gras HANDLE h;
1039*543adbedSBen Gras ftime.dwLowDateTime = 0;
1040*543adbedSBen Gras ftime.dwHighDateTime = 0;
1041*543adbedSBen Gras
1042*543adbedSBen Gras assertion_count(file, line);
1043*543adbedSBen Gras h = CreateFile(pathname, FILE_READ_ATTRIBUTES, 0, NULL,
1044*543adbedSBen Gras OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
1045*543adbedSBen Gras if (h == INVALID_HANDLE_VALUE) {
1046*543adbedSBen Gras failure_start(file, line, "Can't access %s\n", pathname);
1047*543adbedSBen Gras failure_finish(NULL);
1048*543adbedSBen Gras return (0);
1049*543adbedSBen Gras }
1050*543adbedSBen Gras r = GetFileTime(h, &fbirthtime, &fatime, &fmtime);
1051*543adbedSBen Gras switch (type) {
1052*543adbedSBen Gras case 'a': ftime = fatime; break;
1053*543adbedSBen Gras case 'b': ftime = fbirthtime; break;
1054*543adbedSBen Gras case 'm': ftime = fmtime; break;
1055*543adbedSBen Gras }
1056*543adbedSBen Gras CloseHandle(h);
1057*543adbedSBen Gras if (r == 0) {
1058*543adbedSBen Gras failure_start(file, line, "Can't GetFileTime %s\n", pathname);
1059*543adbedSBen Gras failure_finish(NULL);
1060*543adbedSBen Gras return (0);
1061*543adbedSBen Gras }
1062*543adbedSBen Gras wintm.LowPart = ftime.dwLowDateTime;
1063*543adbedSBen Gras wintm.HighPart = ftime.dwHighDateTime;
1064*543adbedSBen Gras filet = (wintm.QuadPart - EPOC_TIME) / 10000000;
1065*543adbedSBen Gras filet_nsec = ((wintm.QuadPart - EPOC_TIME) % 10000000) * 100;
1066*543adbedSBen Gras nsec = (nsec / 100) * 100; /* Round the request */
1067*543adbedSBen Gras #else
1068*543adbedSBen Gras struct stat st;
1069*543adbedSBen Gras
1070*543adbedSBen Gras assertion_count(file, line);
1071*543adbedSBen Gras r = lstat(pathname, &st);
1072*543adbedSBen Gras if (r != 0) {
1073*543adbedSBen Gras failure_start(file, line, "Can't stat %s\n", pathname);
1074*543adbedSBen Gras failure_finish(NULL);
1075*543adbedSBen Gras return (0);
1076*543adbedSBen Gras }
1077*543adbedSBen Gras switch (type) {
1078*543adbedSBen Gras case 'a': filet = st.st_atime; break;
1079*543adbedSBen Gras case 'm': filet = st.st_mtime; break;
1080*543adbedSBen Gras case 'b': filet = 0; break;
1081*543adbedSBen Gras default: fprintf(stderr, "INTERNAL: Bad type %c for file time", type);
1082*543adbedSBen Gras exit(1);
1083*543adbedSBen Gras }
1084*543adbedSBen Gras #if defined(__FreeBSD__)
1085*543adbedSBen Gras switch (type) {
1086*543adbedSBen Gras case 'a': filet_nsec = st.st_atimespec.tv_nsec; break;
1087*543adbedSBen Gras case 'b': filet = st.st_birthtime;
1088*543adbedSBen Gras filet_nsec = st.st_birthtimespec.tv_nsec; break;
1089*543adbedSBen Gras case 'm': filet_nsec = st.st_mtimespec.tv_nsec; break;
1090*543adbedSBen Gras default: fprintf(stderr, "INTERNAL: Bad type %c for file time", type);
1091*543adbedSBen Gras exit(1);
1092*543adbedSBen Gras }
1093*543adbedSBen Gras /* FreeBSD generally only stores to microsecond res, so round. */
1094*543adbedSBen Gras filet_nsec = (filet_nsec / 1000) * 1000;
1095*543adbedSBen Gras nsec = (nsec / 1000) * 1000;
1096*543adbedSBen Gras #else
1097*543adbedSBen Gras filet_nsec = nsec = 0; /* Generic POSIX only has whole seconds. */
1098*543adbedSBen Gras if (type == 'b') return (1); /* Generic POSIX doesn't have birthtime */
1099*543adbedSBen Gras #if defined(__HAIKU__)
1100*543adbedSBen Gras if (type == 'a') return (1); /* Haiku doesn't have atime. */
1101*543adbedSBen Gras #endif
1102*543adbedSBen Gras #endif
1103*543adbedSBen Gras #endif
1104*543adbedSBen Gras if (recent) {
1105*543adbedSBen Gras /* Check that requested time is up-to-date. */
1106*543adbedSBen Gras time_t now = time(NULL);
1107*543adbedSBen Gras if (filet < now - 10 || filet > now + 1) {
1108*543adbedSBen Gras failure_start(file, line,
1109*543adbedSBen Gras "File %s has %ctime %ld, %ld seconds ago\n",
1110*543adbedSBen Gras pathname, type, filet, now - filet);
1111*543adbedSBen Gras failure_finish(NULL);
1112*543adbedSBen Gras return (0);
1113*543adbedSBen Gras }
1114*543adbedSBen Gras } else if (filet != t || filet_nsec != nsec) {
1115*543adbedSBen Gras failure_start(file, line,
1116*543adbedSBen Gras "File %s has %ctime %ld.%09ld, expected %ld.%09ld",
1117*543adbedSBen Gras pathname, type, filet, filet_nsec, t, nsec);
1118*543adbedSBen Gras failure_finish(NULL);
1119*543adbedSBen Gras return (0);
1120*543adbedSBen Gras }
1121*543adbedSBen Gras return (1);
1122*543adbedSBen Gras }
1123*543adbedSBen Gras
1124*543adbedSBen Gras /* Verify atime of 'pathname'. */
1125*543adbedSBen Gras int
assertion_file_atime(const char * file,int line,const char * pathname,long t,long nsec)1126*543adbedSBen Gras assertion_file_atime(const char *file, int line,
1127*543adbedSBen Gras const char *pathname, long t, long nsec)
1128*543adbedSBen Gras {
1129*543adbedSBen Gras return assertion_file_time(file, line, pathname, t, nsec, 'a', 0);
1130*543adbedSBen Gras }
1131*543adbedSBen Gras
1132*543adbedSBen Gras /* Verify atime of 'pathname' is up-to-date. */
1133*543adbedSBen Gras int
assertion_file_atime_recent(const char * file,int line,const char * pathname)1134*543adbedSBen Gras assertion_file_atime_recent(const char *file, int line, const char *pathname)
1135*543adbedSBen Gras {
1136*543adbedSBen Gras return assertion_file_time(file, line, pathname, 0, 0, 'a', 1);
1137*543adbedSBen Gras }
1138*543adbedSBen Gras
1139*543adbedSBen Gras /* Verify birthtime of 'pathname'. */
1140*543adbedSBen Gras int
assertion_file_birthtime(const char * file,int line,const char * pathname,long t,long nsec)1141*543adbedSBen Gras assertion_file_birthtime(const char *file, int line,
1142*543adbedSBen Gras const char *pathname, long t, long nsec)
1143*543adbedSBen Gras {
1144*543adbedSBen Gras return assertion_file_time(file, line, pathname, t, nsec, 'b', 0);
1145*543adbedSBen Gras }
1146*543adbedSBen Gras
1147*543adbedSBen Gras /* Verify birthtime of 'pathname' is up-to-date. */
1148*543adbedSBen Gras int
assertion_file_birthtime_recent(const char * file,int line,const char * pathname)1149*543adbedSBen Gras assertion_file_birthtime_recent(const char *file, int line,
1150*543adbedSBen Gras const char *pathname)
1151*543adbedSBen Gras {
1152*543adbedSBen Gras return assertion_file_time(file, line, pathname, 0, 0, 'b', 1);
1153*543adbedSBen Gras }
1154*543adbedSBen Gras
1155*543adbedSBen Gras /* Verify mtime of 'pathname'. */
1156*543adbedSBen Gras int
assertion_file_mtime(const char * file,int line,const char * pathname,long t,long nsec)1157*543adbedSBen Gras assertion_file_mtime(const char *file, int line,
1158*543adbedSBen Gras const char *pathname, long t, long nsec)
1159*543adbedSBen Gras {
1160*543adbedSBen Gras return assertion_file_time(file, line, pathname, t, nsec, 'm', 0);
1161*543adbedSBen Gras }
1162*543adbedSBen Gras
1163*543adbedSBen Gras /* Verify mtime of 'pathname' is up-to-date. */
1164*543adbedSBen Gras int
assertion_file_mtime_recent(const char * file,int line,const char * pathname)1165*543adbedSBen Gras assertion_file_mtime_recent(const char *file, int line, const char *pathname)
1166*543adbedSBen Gras {
1167*543adbedSBen Gras return assertion_file_time(file, line, pathname, 0, 0, 'm', 1);
1168*543adbedSBen Gras }
1169*543adbedSBen Gras
1170*543adbedSBen Gras /* Verify number of links to 'pathname'. */
1171*543adbedSBen Gras int
assertion_file_nlinks(const char * file,int line,const char * pathname,int nlinks)1172*543adbedSBen Gras assertion_file_nlinks(const char *file, int line,
1173*543adbedSBen Gras const char *pathname, int nlinks)
1174*543adbedSBen Gras {
1175*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
1176*543adbedSBen Gras BY_HANDLE_FILE_INFORMATION bhfi;
1177*543adbedSBen Gras int r;
1178*543adbedSBen Gras
1179*543adbedSBen Gras assertion_count(file, line);
1180*543adbedSBen Gras r = my_GetFileInformationByName(pathname, &bhfi);
1181*543adbedSBen Gras if (r != 0 && bhfi.nNumberOfLinks == (DWORD)nlinks)
1182*543adbedSBen Gras return (1);
1183*543adbedSBen Gras failure_start(file, line, "File %s has %d links, expected %d",
1184*543adbedSBen Gras pathname, bhfi.nNumberOfLinks, nlinks);
1185*543adbedSBen Gras failure_finish(NULL);
1186*543adbedSBen Gras return (0);
1187*543adbedSBen Gras #else
1188*543adbedSBen Gras struct stat st;
1189*543adbedSBen Gras int r;
1190*543adbedSBen Gras
1191*543adbedSBen Gras assertion_count(file, line);
1192*543adbedSBen Gras r = lstat(pathname, &st);
1193*543adbedSBen Gras if (r == 0 && st.st_nlink == nlinks)
1194*543adbedSBen Gras return (1);
1195*543adbedSBen Gras failure_start(file, line, "File %s has %d links, expected %d",
1196*543adbedSBen Gras pathname, st.st_nlink, nlinks);
1197*543adbedSBen Gras failure_finish(NULL);
1198*543adbedSBen Gras return (0);
1199*543adbedSBen Gras #endif
1200*543adbedSBen Gras }
1201*543adbedSBen Gras
1202*543adbedSBen Gras /* Verify size of 'pathname'. */
1203*543adbedSBen Gras int
assertion_file_size(const char * file,int line,const char * pathname,long size)1204*543adbedSBen Gras assertion_file_size(const char *file, int line, const char *pathname, long size)
1205*543adbedSBen Gras {
1206*543adbedSBen Gras int64_t filesize;
1207*543adbedSBen Gras int r;
1208*543adbedSBen Gras
1209*543adbedSBen Gras assertion_count(file, line);
1210*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
1211*543adbedSBen Gras {
1212*543adbedSBen Gras BY_HANDLE_FILE_INFORMATION bhfi;
1213*543adbedSBen Gras r = !my_GetFileInformationByName(pathname, &bhfi);
1214*543adbedSBen Gras filesize = ((int64_t)bhfi.nFileSizeHigh << 32) + bhfi.nFileSizeLow;
1215*543adbedSBen Gras }
1216*543adbedSBen Gras #else
1217*543adbedSBen Gras {
1218*543adbedSBen Gras struct stat st;
1219*543adbedSBen Gras r = lstat(pathname, &st);
1220*543adbedSBen Gras filesize = st.st_size;
1221*543adbedSBen Gras }
1222*543adbedSBen Gras #endif
1223*543adbedSBen Gras if (r == 0 && filesize == size)
1224*543adbedSBen Gras return (1);
1225*543adbedSBen Gras failure_start(file, line, "File %s has size %ld, expected %ld",
1226*543adbedSBen Gras pathname, (long)filesize, (long)size);
1227*543adbedSBen Gras failure_finish(NULL);
1228*543adbedSBen Gras return (0);
1229*543adbedSBen Gras }
1230*543adbedSBen Gras
1231*543adbedSBen Gras /* Assert that 'pathname' is a dir. If mode >= 0, verify that too. */
1232*543adbedSBen Gras int
assertion_is_dir(const char * file,int line,const char * pathname,int mode)1233*543adbedSBen Gras assertion_is_dir(const char *file, int line, const char *pathname, int mode)
1234*543adbedSBen Gras {
1235*543adbedSBen Gras struct stat st;
1236*543adbedSBen Gras int r;
1237*543adbedSBen Gras
1238*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
1239*543adbedSBen Gras (void)mode; /* UNUSED */
1240*543adbedSBen Gras #endif
1241*543adbedSBen Gras assertion_count(file, line);
1242*543adbedSBen Gras r = lstat(pathname, &st);
1243*543adbedSBen Gras if (r != 0) {
1244*543adbedSBen Gras failure_start(file, line, "Dir should exist: %s", pathname);
1245*543adbedSBen Gras failure_finish(NULL);
1246*543adbedSBen Gras return (0);
1247*543adbedSBen Gras }
1248*543adbedSBen Gras if (!S_ISDIR(st.st_mode)) {
1249*543adbedSBen Gras failure_start(file, line, "%s is not a dir", pathname);
1250*543adbedSBen Gras failure_finish(NULL);
1251*543adbedSBen Gras return (0);
1252*543adbedSBen Gras }
1253*543adbedSBen Gras #if !defined(_WIN32) || defined(__CYGWIN__)
1254*543adbedSBen Gras /* Windows doesn't handle permissions the same way as POSIX,
1255*543adbedSBen Gras * so just ignore the mode tests. */
1256*543adbedSBen Gras /* TODO: Can we do better here? */
1257*543adbedSBen Gras if (mode >= 0 && mode != (st.st_mode & 07777)) {
1258*543adbedSBen Gras failure_start(file, line, "Dir %s has wrong mode", pathname);
1259*543adbedSBen Gras logprintf(" Expected: 0%3o\n", mode);
1260*543adbedSBen Gras logprintf(" Found: 0%3o\n", st.st_mode & 07777);
1261*543adbedSBen Gras failure_finish(NULL);
1262*543adbedSBen Gras return (0);
1263*543adbedSBen Gras }
1264*543adbedSBen Gras #endif
1265*543adbedSBen Gras return (1);
1266*543adbedSBen Gras }
1267*543adbedSBen Gras
1268*543adbedSBen Gras /* Verify that 'pathname' is a regular file. If 'mode' is >= 0,
1269*543adbedSBen Gras * verify that too. */
1270*543adbedSBen Gras int
assertion_is_reg(const char * file,int line,const char * pathname,int mode)1271*543adbedSBen Gras assertion_is_reg(const char *file, int line, const char *pathname, int mode)
1272*543adbedSBen Gras {
1273*543adbedSBen Gras struct stat st;
1274*543adbedSBen Gras int r;
1275*543adbedSBen Gras
1276*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
1277*543adbedSBen Gras (void)mode; /* UNUSED */
1278*543adbedSBen Gras #endif
1279*543adbedSBen Gras assertion_count(file, line);
1280*543adbedSBen Gras r = lstat(pathname, &st);
1281*543adbedSBen Gras if (r != 0 || !S_ISREG(st.st_mode)) {
1282*543adbedSBen Gras failure_start(file, line, "File should exist: %s", pathname);
1283*543adbedSBen Gras failure_finish(NULL);
1284*543adbedSBen Gras return (0);
1285*543adbedSBen Gras }
1286*543adbedSBen Gras #if !defined(_WIN32) || defined(__CYGWIN__)
1287*543adbedSBen Gras /* Windows doesn't handle permissions the same way as POSIX,
1288*543adbedSBen Gras * so just ignore the mode tests. */
1289*543adbedSBen Gras /* TODO: Can we do better here? */
1290*543adbedSBen Gras if (mode >= 0 && mode != (st.st_mode & 07777)) {
1291*543adbedSBen Gras failure_start(file, line, "File %s has wrong mode", pathname);
1292*543adbedSBen Gras logprintf(" Expected: 0%3o\n", mode);
1293*543adbedSBen Gras logprintf(" Found: 0%3o\n", st.st_mode & 07777);
1294*543adbedSBen Gras failure_finish(NULL);
1295*543adbedSBen Gras return (0);
1296*543adbedSBen Gras }
1297*543adbedSBen Gras #endif
1298*543adbedSBen Gras return (1);
1299*543adbedSBen Gras }
1300*543adbedSBen Gras
1301*543adbedSBen Gras /* Check whether 'pathname' is a symbolic link. If 'contents' is
1302*543adbedSBen Gras * non-NULL, verify that the symlink has those contents. */
1303*543adbedSBen Gras static int
is_symlink(const char * file,int line,const char * pathname,const char * contents)1304*543adbedSBen Gras is_symlink(const char *file, int line,
1305*543adbedSBen Gras const char *pathname, const char *contents)
1306*543adbedSBen Gras {
1307*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
1308*543adbedSBen Gras (void)pathname; /* UNUSED */
1309*543adbedSBen Gras (void)contents; /* UNUSED */
1310*543adbedSBen Gras assertion_count(file, line);
1311*543adbedSBen Gras /* Windows sort-of has real symlinks, but they're only usable
1312*543adbedSBen Gras * by privileged users and are crippled even then, so there's
1313*543adbedSBen Gras * really not much point in bothering with this. */
1314*543adbedSBen Gras return (0);
1315*543adbedSBen Gras #else
1316*543adbedSBen Gras char buff[300];
1317*543adbedSBen Gras struct stat st;
1318*543adbedSBen Gras ssize_t linklen;
1319*543adbedSBen Gras int r;
1320*543adbedSBen Gras
1321*543adbedSBen Gras assertion_count(file, line);
1322*543adbedSBen Gras r = lstat(pathname, &st);
1323*543adbedSBen Gras if (r != 0) {
1324*543adbedSBen Gras failure_start(file, line,
1325*543adbedSBen Gras "Symlink should exist: %s", pathname);
1326*543adbedSBen Gras failure_finish(NULL);
1327*543adbedSBen Gras return (0);
1328*543adbedSBen Gras }
1329*543adbedSBen Gras if (!S_ISLNK(st.st_mode))
1330*543adbedSBen Gras return (0);
1331*543adbedSBen Gras if (contents == NULL)
1332*543adbedSBen Gras return (1);
1333*543adbedSBen Gras linklen = readlink(pathname, buff, sizeof(buff));
1334*543adbedSBen Gras if (linklen < 0) {
1335*543adbedSBen Gras failure_start(file, line, "Can't read symlink %s", pathname);
1336*543adbedSBen Gras failure_finish(NULL);
1337*543adbedSBen Gras return (0);
1338*543adbedSBen Gras }
1339*543adbedSBen Gras buff[linklen] = '\0';
1340*543adbedSBen Gras if (strcmp(buff, contents) != 0)
1341*543adbedSBen Gras return (0);
1342*543adbedSBen Gras return (1);
1343*543adbedSBen Gras #endif
1344*543adbedSBen Gras }
1345*543adbedSBen Gras
1346*543adbedSBen Gras /* Assert that path is a symlink that (optionally) contains contents. */
1347*543adbedSBen Gras int
assertion_is_symlink(const char * file,int line,const char * path,const char * contents)1348*543adbedSBen Gras assertion_is_symlink(const char *file, int line,
1349*543adbedSBen Gras const char *path, const char *contents)
1350*543adbedSBen Gras {
1351*543adbedSBen Gras if (is_symlink(file, line, path, contents))
1352*543adbedSBen Gras return (1);
1353*543adbedSBen Gras if (contents)
1354*543adbedSBen Gras failure_start(file, line, "File %s is not a symlink to %s",
1355*543adbedSBen Gras path, contents);
1356*543adbedSBen Gras else
1357*543adbedSBen Gras failure_start(file, line, "File %s is not a symlink", path);
1358*543adbedSBen Gras failure_finish(NULL);
1359*543adbedSBen Gras return (0);
1360*543adbedSBen Gras }
1361*543adbedSBen Gras
1362*543adbedSBen Gras
1363*543adbedSBen Gras /* Create a directory and report any errors. */
1364*543adbedSBen Gras int
assertion_make_dir(const char * file,int line,const char * dirname,int mode)1365*543adbedSBen Gras assertion_make_dir(const char *file, int line, const char *dirname, int mode)
1366*543adbedSBen Gras {
1367*543adbedSBen Gras assertion_count(file, line);
1368*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
1369*543adbedSBen Gras (void)mode; /* UNUSED */
1370*543adbedSBen Gras if (0 == _mkdir(dirname))
1371*543adbedSBen Gras return (1);
1372*543adbedSBen Gras #else
1373*543adbedSBen Gras if (0 == mkdir(dirname, mode))
1374*543adbedSBen Gras return (1);
1375*543adbedSBen Gras #endif
1376*543adbedSBen Gras failure_start(file, line, "Could not create directory %s", dirname);
1377*543adbedSBen Gras failure_finish(NULL);
1378*543adbedSBen Gras return(0);
1379*543adbedSBen Gras }
1380*543adbedSBen Gras
1381*543adbedSBen Gras /* Create a file with the specified contents and report any failures. */
1382*543adbedSBen Gras int
assertion_make_file(const char * file,int line,const char * path,int mode,const char * contents)1383*543adbedSBen Gras assertion_make_file(const char *file, int line,
1384*543adbedSBen Gras const char *path, int mode, const char *contents)
1385*543adbedSBen Gras {
1386*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
1387*543adbedSBen Gras /* TODO: Rework this to set file mode as well. */
1388*543adbedSBen Gras FILE *f;
1389*543adbedSBen Gras (void)mode; /* UNUSED */
1390*543adbedSBen Gras assertion_count(file, line);
1391*543adbedSBen Gras f = fopen(path, "wb");
1392*543adbedSBen Gras if (f == NULL) {
1393*543adbedSBen Gras failure_start(file, line, "Could not create file %s", path);
1394*543adbedSBen Gras failure_finish(NULL);
1395*543adbedSBen Gras return (0);
1396*543adbedSBen Gras }
1397*543adbedSBen Gras if (contents != NULL) {
1398*543adbedSBen Gras if (strlen(contents)
1399*543adbedSBen Gras != fwrite(contents, 1, strlen(contents), f)) {
1400*543adbedSBen Gras fclose(f);
1401*543adbedSBen Gras failure_start(file, line,
1402*543adbedSBen Gras "Could not write file %s", path);
1403*543adbedSBen Gras failure_finish(NULL);
1404*543adbedSBen Gras return (0);
1405*543adbedSBen Gras }
1406*543adbedSBen Gras }
1407*543adbedSBen Gras fclose(f);
1408*543adbedSBen Gras return (1);
1409*543adbedSBen Gras #else
1410*543adbedSBen Gras int fd;
1411*543adbedSBen Gras assertion_count(file, line);
1412*543adbedSBen Gras fd = open(path, O_CREAT | O_WRONLY, mode >= 0 ? mode : 0644);
1413*543adbedSBen Gras if (fd < 0) {
1414*543adbedSBen Gras failure_start(file, line, "Could not create %s", path);
1415*543adbedSBen Gras failure_finish(NULL);
1416*543adbedSBen Gras return (0);
1417*543adbedSBen Gras }
1418*543adbedSBen Gras if (contents != NULL) {
1419*543adbedSBen Gras if ((ssize_t)strlen(contents)
1420*543adbedSBen Gras != write(fd, contents, strlen(contents))) {
1421*543adbedSBen Gras close(fd);
1422*543adbedSBen Gras failure_start(file, line, "Could not write to %s", path);
1423*543adbedSBen Gras failure_finish(NULL);
1424*543adbedSBen Gras return (0);
1425*543adbedSBen Gras }
1426*543adbedSBen Gras }
1427*543adbedSBen Gras close(fd);
1428*543adbedSBen Gras return (1);
1429*543adbedSBen Gras #endif
1430*543adbedSBen Gras }
1431*543adbedSBen Gras
1432*543adbedSBen Gras /* Create a hardlink and report any failures. */
1433*543adbedSBen Gras int
assertion_make_hardlink(const char * file,int line,const char * newpath,const char * linkto)1434*543adbedSBen Gras assertion_make_hardlink(const char *file, int line,
1435*543adbedSBen Gras const char *newpath, const char *linkto)
1436*543adbedSBen Gras {
1437*543adbedSBen Gras int succeeded;
1438*543adbedSBen Gras
1439*543adbedSBen Gras assertion_count(file, line);
1440*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
1441*543adbedSBen Gras succeeded = my_CreateHardLinkA(newpath, linkto);
1442*543adbedSBen Gras #elif HAVE_LINK
1443*543adbedSBen Gras succeeded = !link(linkto, newpath);
1444*543adbedSBen Gras #else
1445*543adbedSBen Gras succeeded = 0;
1446*543adbedSBen Gras #endif
1447*543adbedSBen Gras if (succeeded)
1448*543adbedSBen Gras return (1);
1449*543adbedSBen Gras failure_start(file, line, "Could not create hardlink");
1450*543adbedSBen Gras logprintf(" New link: %s\n", newpath);
1451*543adbedSBen Gras logprintf(" Old name: %s\n", linkto);
1452*543adbedSBen Gras failure_finish(NULL);
1453*543adbedSBen Gras return(0);
1454*543adbedSBen Gras }
1455*543adbedSBen Gras
1456*543adbedSBen Gras /* Create a symlink and report any failures. */
1457*543adbedSBen Gras int
assertion_make_symlink(const char * file,int line,const char * newpath,const char * linkto)1458*543adbedSBen Gras assertion_make_symlink(const char *file, int line,
1459*543adbedSBen Gras const char *newpath, const char *linkto)
1460*543adbedSBen Gras {
1461*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
1462*543adbedSBen Gras int targetIsDir = 0; /* TODO: Fix this */
1463*543adbedSBen Gras assertion_count(file, line);
1464*543adbedSBen Gras if (my_CreateSymbolicLinkA(newpath, linkto, targetIsDir))
1465*543adbedSBen Gras return (1);
1466*543adbedSBen Gras #elif HAVE_SYMLINK
1467*543adbedSBen Gras assertion_count(file, line);
1468*543adbedSBen Gras if (0 == symlink(linkto, newpath))
1469*543adbedSBen Gras return (1);
1470*543adbedSBen Gras #endif
1471*543adbedSBen Gras failure_start(file, line, "Could not create symlink");
1472*543adbedSBen Gras logprintf(" New link: %s\n", newpath);
1473*543adbedSBen Gras logprintf(" Old name: %s\n", linkto);
1474*543adbedSBen Gras failure_finish(NULL);
1475*543adbedSBen Gras return(0);
1476*543adbedSBen Gras }
1477*543adbedSBen Gras
1478*543adbedSBen Gras /* Set umask, report failures. */
1479*543adbedSBen Gras int
assertion_umask(const char * file,int line,int mask)1480*543adbedSBen Gras assertion_umask(const char *file, int line, int mask)
1481*543adbedSBen Gras {
1482*543adbedSBen Gras assertion_count(file, line);
1483*543adbedSBen Gras (void)file; /* UNUSED */
1484*543adbedSBen Gras (void)line; /* UNUSED */
1485*543adbedSBen Gras umask(mask);
1486*543adbedSBen Gras return (1);
1487*543adbedSBen Gras }
1488*543adbedSBen Gras
1489*543adbedSBen Gras /*
1490*543adbedSBen Gras *
1491*543adbedSBen Gras * UTILITIES for use by tests.
1492*543adbedSBen Gras *
1493*543adbedSBen Gras */
1494*543adbedSBen Gras
1495*543adbedSBen Gras /*
1496*543adbedSBen Gras * Check whether platform supports symlinks. This is intended
1497*543adbedSBen Gras * for tests to use in deciding whether to bother testing symlink
1498*543adbedSBen Gras * support; if the platform doesn't support symlinks, there's no point
1499*543adbedSBen Gras * in checking whether the program being tested can create them.
1500*543adbedSBen Gras *
1501*543adbedSBen Gras * Note that the first time this test is called, we actually go out to
1502*543adbedSBen Gras * disk to create and verify a symlink. This is necessary because
1503*543adbedSBen Gras * symlink support is actually a property of a particular filesystem
1504*543adbedSBen Gras * and can thus vary between directories on a single system. After
1505*543adbedSBen Gras * the first call, this returns the cached result from memory, so it's
1506*543adbedSBen Gras * safe to call it as often as you wish.
1507*543adbedSBen Gras */
1508*543adbedSBen Gras int
canSymlink(void)1509*543adbedSBen Gras canSymlink(void)
1510*543adbedSBen Gras {
1511*543adbedSBen Gras /* Remember the test result */
1512*543adbedSBen Gras static int value = 0, tested = 0;
1513*543adbedSBen Gras if (tested)
1514*543adbedSBen Gras return (value);
1515*543adbedSBen Gras
1516*543adbedSBen Gras ++tested;
1517*543adbedSBen Gras assertion_make_file(__FILE__, __LINE__, "canSymlink.0", 0644, "a");
1518*543adbedSBen Gras /* Note: Cygwin has its own symlink() emulation that does not
1519*543adbedSBen Gras * use the Win32 CreateSymbolicLink() function. */
1520*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
1521*543adbedSBen Gras value = my_CreateSymbolicLinkA("canSymlink.1", "canSymlink.0", 0)
1522*543adbedSBen Gras && is_symlink(__FILE__, __LINE__, "canSymlink.1", "canSymlink.0");
1523*543adbedSBen Gras #elif HAVE_SYMLINK
1524*543adbedSBen Gras value = (0 == symlink("canSymlink.0", "canSymlink.1"))
1525*543adbedSBen Gras && is_symlink(__FILE__, __LINE__, "canSymlink.1","canSymlink.0");
1526*543adbedSBen Gras #endif
1527*543adbedSBen Gras return (value);
1528*543adbedSBen Gras }
1529*543adbedSBen Gras
1530*543adbedSBen Gras /*
1531*543adbedSBen Gras * Can this platform run the gzip program?
1532*543adbedSBen Gras */
1533*543adbedSBen Gras /* Platform-dependent options for hiding the output of a subcommand. */
1534*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
1535*543adbedSBen Gras static const char *redirectArgs = ">NUL 2>NUL"; /* Win32 cmd.exe */
1536*543adbedSBen Gras #else
1537*543adbedSBen Gras static const char *redirectArgs = ">/dev/null 2>/dev/null"; /* POSIX 'sh' */
1538*543adbedSBen Gras #endif
1539*543adbedSBen Gras int
canGzip(void)1540*543adbedSBen Gras canGzip(void)
1541*543adbedSBen Gras {
1542*543adbedSBen Gras static int tested = 0, value = 0;
1543*543adbedSBen Gras if (!tested) {
1544*543adbedSBen Gras tested = 1;
1545*543adbedSBen Gras if (systemf("gzip -V %s", redirectArgs) == 0)
1546*543adbedSBen Gras value = 1;
1547*543adbedSBen Gras }
1548*543adbedSBen Gras return (value);
1549*543adbedSBen Gras }
1550*543adbedSBen Gras
1551*543adbedSBen Gras /*
1552*543adbedSBen Gras * Can this platform run the gunzip program?
1553*543adbedSBen Gras */
1554*543adbedSBen Gras int
canGunzip(void)1555*543adbedSBen Gras canGunzip(void)
1556*543adbedSBen Gras {
1557*543adbedSBen Gras static int tested = 0, value = 0;
1558*543adbedSBen Gras if (!tested) {
1559*543adbedSBen Gras tested = 1;
1560*543adbedSBen Gras if (systemf("gunzip -V %s", redirectArgs) == 0)
1561*543adbedSBen Gras value = 1;
1562*543adbedSBen Gras }
1563*543adbedSBen Gras return (value);
1564*543adbedSBen Gras }
1565*543adbedSBen Gras
1566*543adbedSBen Gras /*
1567*543adbedSBen Gras * Sleep as needed; useful for verifying disk timestamp changes by
1568*543adbedSBen Gras * ensuring that the wall-clock time has actually changed before we
1569*543adbedSBen Gras * go back to re-read something from disk.
1570*543adbedSBen Gras */
1571*543adbedSBen Gras void
sleepUntilAfter(time_t t)1572*543adbedSBen Gras sleepUntilAfter(time_t t)
1573*543adbedSBen Gras {
1574*543adbedSBen Gras while (t >= time(NULL))
1575*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
1576*543adbedSBen Gras Sleep(500);
1577*543adbedSBen Gras #else
1578*543adbedSBen Gras sleep(1);
1579*543adbedSBen Gras #endif
1580*543adbedSBen Gras }
1581*543adbedSBen Gras
1582*543adbedSBen Gras /*
1583*543adbedSBen Gras * Call standard system() call, but build up the command line using
1584*543adbedSBen Gras * sprintf() conventions.
1585*543adbedSBen Gras */
1586*543adbedSBen Gras int
systemf(const char * fmt,...)1587*543adbedSBen Gras systemf(const char *fmt, ...)
1588*543adbedSBen Gras {
1589*543adbedSBen Gras char buff[8192];
1590*543adbedSBen Gras va_list ap;
1591*543adbedSBen Gras int r;
1592*543adbedSBen Gras
1593*543adbedSBen Gras va_start(ap, fmt);
1594*543adbedSBen Gras vsprintf(buff, fmt, ap);
1595*543adbedSBen Gras if (verbosity > VERBOSITY_FULL)
1596*543adbedSBen Gras logprintf("Cmd: %s\n", buff);
1597*543adbedSBen Gras r = system(buff);
1598*543adbedSBen Gras va_end(ap);
1599*543adbedSBen Gras return (r);
1600*543adbedSBen Gras }
1601*543adbedSBen Gras
1602*543adbedSBen Gras /*
1603*543adbedSBen Gras * Slurp a file into memory for ease of comparison and testing.
1604*543adbedSBen Gras * Returns size of file in 'sizep' if non-NULL, null-terminates
1605*543adbedSBen Gras * data in memory for ease of use.
1606*543adbedSBen Gras */
1607*543adbedSBen Gras char *
slurpfile(size_t * sizep,const char * fmt,...)1608*543adbedSBen Gras slurpfile(size_t * sizep, const char *fmt, ...)
1609*543adbedSBen Gras {
1610*543adbedSBen Gras char filename[8192];
1611*543adbedSBen Gras struct stat st;
1612*543adbedSBen Gras va_list ap;
1613*543adbedSBen Gras char *p;
1614*543adbedSBen Gras ssize_t bytes_read;
1615*543adbedSBen Gras FILE *f;
1616*543adbedSBen Gras int r;
1617*543adbedSBen Gras
1618*543adbedSBen Gras va_start(ap, fmt);
1619*543adbedSBen Gras vsprintf(filename, fmt, ap);
1620*543adbedSBen Gras va_end(ap);
1621*543adbedSBen Gras
1622*543adbedSBen Gras f = fopen(filename, "rb");
1623*543adbedSBen Gras if (f == NULL) {
1624*543adbedSBen Gras /* Note: No error; non-existent file is okay here. */
1625*543adbedSBen Gras return (NULL);
1626*543adbedSBen Gras }
1627*543adbedSBen Gras r = fstat(fileno(f), &st);
1628*543adbedSBen Gras if (r != 0) {
1629*543adbedSBen Gras logprintf("Can't stat file %s\n", filename);
1630*543adbedSBen Gras fclose(f);
1631*543adbedSBen Gras return (NULL);
1632*543adbedSBen Gras }
1633*543adbedSBen Gras p = malloc((size_t)st.st_size + 1);
1634*543adbedSBen Gras if (p == NULL) {
1635*543adbedSBen Gras logprintf("Can't allocate %ld bytes of memory to read file %s\n",
1636*543adbedSBen Gras (long int)st.st_size, filename);
1637*543adbedSBen Gras fclose(f);
1638*543adbedSBen Gras return (NULL);
1639*543adbedSBen Gras }
1640*543adbedSBen Gras bytes_read = fread(p, 1, (size_t)st.st_size, f);
1641*543adbedSBen Gras if (bytes_read < st.st_size) {
1642*543adbedSBen Gras logprintf("Can't read file %s\n", filename);
1643*543adbedSBen Gras fclose(f);
1644*543adbedSBen Gras free(p);
1645*543adbedSBen Gras return (NULL);
1646*543adbedSBen Gras }
1647*543adbedSBen Gras p[st.st_size] = '\0';
1648*543adbedSBen Gras if (sizep != NULL)
1649*543adbedSBen Gras *sizep = (size_t)st.st_size;
1650*543adbedSBen Gras fclose(f);
1651*543adbedSBen Gras return (p);
1652*543adbedSBen Gras }
1653*543adbedSBen Gras
1654*543adbedSBen Gras /* Read a uuencoded file from the reference directory, decode, and
1655*543adbedSBen Gras * write the result into the current directory. */
1656*543adbedSBen Gras #define UUDECODE(c) (((c) - 0x20) & 0x3f)
1657*543adbedSBen Gras void
extract_reference_file(const char * name)1658*543adbedSBen Gras extract_reference_file(const char *name)
1659*543adbedSBen Gras {
1660*543adbedSBen Gras char buff[1024];
1661*543adbedSBen Gras FILE *in, *out;
1662*543adbedSBen Gras
1663*543adbedSBen Gras sprintf(buff, "%s/%s.uu", refdir, name);
1664*543adbedSBen Gras in = fopen(buff, "r");
1665*543adbedSBen Gras failure("Couldn't open reference file %s", buff);
1666*543adbedSBen Gras assert(in != NULL);
1667*543adbedSBen Gras if (in == NULL)
1668*543adbedSBen Gras return;
1669*543adbedSBen Gras /* Read up to and including the 'begin' line. */
1670*543adbedSBen Gras for (;;) {
1671*543adbedSBen Gras if (fgets(buff, sizeof(buff), in) == NULL) {
1672*543adbedSBen Gras /* TODO: This is a failure. */
1673*543adbedSBen Gras return;
1674*543adbedSBen Gras }
1675*543adbedSBen Gras if (memcmp(buff, "begin ", 6) == 0)
1676*543adbedSBen Gras break;
1677*543adbedSBen Gras }
1678*543adbedSBen Gras /* Now, decode the rest and write it. */
1679*543adbedSBen Gras /* Not a lot of error checking here; the input better be right. */
1680*543adbedSBen Gras out = fopen(name, "wb");
1681*543adbedSBen Gras while (fgets(buff, sizeof(buff), in) != NULL) {
1682*543adbedSBen Gras char *p = buff;
1683*543adbedSBen Gras int bytes;
1684*543adbedSBen Gras
1685*543adbedSBen Gras if (memcmp(buff, "end", 3) == 0)
1686*543adbedSBen Gras break;
1687*543adbedSBen Gras
1688*543adbedSBen Gras bytes = UUDECODE(*p++);
1689*543adbedSBen Gras while (bytes > 0) {
1690*543adbedSBen Gras int n = 0;
1691*543adbedSBen Gras /* Write out 1-3 bytes from that. */
1692*543adbedSBen Gras if (bytes > 0) {
1693*543adbedSBen Gras n = UUDECODE(*p++) << 18;
1694*543adbedSBen Gras n |= UUDECODE(*p++) << 12;
1695*543adbedSBen Gras fputc(n >> 16, out);
1696*543adbedSBen Gras --bytes;
1697*543adbedSBen Gras }
1698*543adbedSBen Gras if (bytes > 0) {
1699*543adbedSBen Gras n |= UUDECODE(*p++) << 6;
1700*543adbedSBen Gras fputc((n >> 8) & 0xFF, out);
1701*543adbedSBen Gras --bytes;
1702*543adbedSBen Gras }
1703*543adbedSBen Gras if (bytes > 0) {
1704*543adbedSBen Gras n |= UUDECODE(*p++);
1705*543adbedSBen Gras fputc(n & 0xFF, out);
1706*543adbedSBen Gras --bytes;
1707*543adbedSBen Gras }
1708*543adbedSBen Gras }
1709*543adbedSBen Gras }
1710*543adbedSBen Gras fclose(out);
1711*543adbedSBen Gras fclose(in);
1712*543adbedSBen Gras }
1713*543adbedSBen Gras
1714*543adbedSBen Gras /*
1715*543adbedSBen Gras *
1716*543adbedSBen Gras * TEST management
1717*543adbedSBen Gras *
1718*543adbedSBen Gras */
1719*543adbedSBen Gras
1720*543adbedSBen Gras /*
1721*543adbedSBen Gras * "list.h" is simply created by "grep DEFINE_TEST test_*.c"; it has
1722*543adbedSBen Gras * a line like
1723*543adbedSBen Gras * DEFINE_TEST(test_function)
1724*543adbedSBen Gras * for each test.
1725*543adbedSBen Gras */
1726*543adbedSBen Gras
1727*543adbedSBen Gras /* Use "list.h" to declare all of the test functions. */
1728*543adbedSBen Gras #undef DEFINE_TEST
1729*543adbedSBen Gras #define DEFINE_TEST(name) void name(void);
1730*543adbedSBen Gras #include "list.h"
1731*543adbedSBen Gras
1732*543adbedSBen Gras /* Use "list.h" to create a list of all tests (functions and names). */
1733*543adbedSBen Gras #undef DEFINE_TEST
1734*543adbedSBen Gras #define DEFINE_TEST(n) { n, #n, 0 },
1735*543adbedSBen Gras struct { void (*func)(void); const char *name; int failures; } tests[] = {
1736*543adbedSBen Gras #include "list.h"
1737*543adbedSBen Gras };
1738*543adbedSBen Gras
1739*543adbedSBen Gras /*
1740*543adbedSBen Gras * Summarize repeated failures in the just-completed test.
1741*543adbedSBen Gras */
1742*543adbedSBen Gras static void
test_summarize(const char * filename,int failed)1743*543adbedSBen Gras test_summarize(const char *filename, int failed)
1744*543adbedSBen Gras {
1745*543adbedSBen Gras unsigned int i;
1746*543adbedSBen Gras
1747*543adbedSBen Gras switch (verbosity) {
1748*543adbedSBen Gras case VERBOSITY_SUMMARY_ONLY:
1749*543adbedSBen Gras printf(failed ? "E" : ".");
1750*543adbedSBen Gras fflush(stdout);
1751*543adbedSBen Gras break;
1752*543adbedSBen Gras case VERBOSITY_PASSFAIL:
1753*543adbedSBen Gras printf(failed ? "FAIL\n" : "ok\n");
1754*543adbedSBen Gras break;
1755*543adbedSBen Gras }
1756*543adbedSBen Gras
1757*543adbedSBen Gras log_console = (verbosity == VERBOSITY_LIGHT_REPORT);
1758*543adbedSBen Gras
1759*543adbedSBen Gras for (i = 0; i < sizeof(failed_lines)/sizeof(failed_lines[0]); i++) {
1760*543adbedSBen Gras if (failed_lines[i].count > 1 && !failed_lines[i].skip)
1761*543adbedSBen Gras logprintf("%s:%d: Summary: Failed %d times\n",
1762*543adbedSBen Gras filename, i, failed_lines[i].count);
1763*543adbedSBen Gras }
1764*543adbedSBen Gras /* Clear the failure history for the next file. */
1765*543adbedSBen Gras memset(failed_lines, 0, sizeof(failed_lines));
1766*543adbedSBen Gras }
1767*543adbedSBen Gras
1768*543adbedSBen Gras /*
1769*543adbedSBen Gras * Actually run a single test, with appropriate setup and cleanup.
1770*543adbedSBen Gras */
1771*543adbedSBen Gras static int
test_run(int i,const char * tmpdir)1772*543adbedSBen Gras test_run(int i, const char *tmpdir)
1773*543adbedSBen Gras {
1774*543adbedSBen Gras char logfilename[64];
1775*543adbedSBen Gras int failures_before = failures;
1776*543adbedSBen Gras int oldumask;
1777*543adbedSBen Gras
1778*543adbedSBen Gras switch (verbosity) {
1779*543adbedSBen Gras case VERBOSITY_SUMMARY_ONLY: /* No per-test reports at all */
1780*543adbedSBen Gras break;
1781*543adbedSBen Gras case VERBOSITY_PASSFAIL: /* rest of line will include ok/FAIL marker */
1782*543adbedSBen Gras printf("%3d: %-50s", i, tests[i].name);
1783*543adbedSBen Gras fflush(stdout);
1784*543adbedSBen Gras break;
1785*543adbedSBen Gras default: /* Title of test, details will follow */
1786*543adbedSBen Gras printf("%3d: %s\n", i, tests[i].name);
1787*543adbedSBen Gras }
1788*543adbedSBen Gras
1789*543adbedSBen Gras /* Chdir to the top-level work directory. */
1790*543adbedSBen Gras if (!assertChdir(tmpdir)) {
1791*543adbedSBen Gras fprintf(stderr,
1792*543adbedSBen Gras "ERROR: Can't chdir to top work dir %s\n", tmpdir);
1793*543adbedSBen Gras exit(1);
1794*543adbedSBen Gras }
1795*543adbedSBen Gras /* Create a log file for this test. */
1796*543adbedSBen Gras sprintf(logfilename, "%s.log", tests[i].name);
1797*543adbedSBen Gras logfile = fopen(logfilename, "w");
1798*543adbedSBen Gras fprintf(logfile, "%s\n\n", tests[i].name);
1799*543adbedSBen Gras /* Chdir() to a work dir for this specific test. */
1800*543adbedSBen Gras if (!assertMakeDir(tests[i].name, 0755)
1801*543adbedSBen Gras || !assertChdir(tests[i].name)) {
1802*543adbedSBen Gras fprintf(stderr,
1803*543adbedSBen Gras "ERROR: Can't chdir to work dir %s/%s\n",
1804*543adbedSBen Gras tmpdir, tests[i].name);
1805*543adbedSBen Gras exit(1);
1806*543adbedSBen Gras }
1807*543adbedSBen Gras /* Explicitly reset the locale before each test. */
1808*543adbedSBen Gras setlocale(LC_ALL, "C");
1809*543adbedSBen Gras /* Record the umask before we run the test. */
1810*543adbedSBen Gras umask(oldumask = umask(0));
1811*543adbedSBen Gras /*
1812*543adbedSBen Gras * Run the actual test.
1813*543adbedSBen Gras */
1814*543adbedSBen Gras (*tests[i].func)();
1815*543adbedSBen Gras /*
1816*543adbedSBen Gras * Clean up and report afterwards.
1817*543adbedSBen Gras */
1818*543adbedSBen Gras /* Restore umask */
1819*543adbedSBen Gras umask(oldumask);
1820*543adbedSBen Gras /* Reset locale. */
1821*543adbedSBen Gras setlocale(LC_ALL, "C");
1822*543adbedSBen Gras /* Reset directory. */
1823*543adbedSBen Gras if (!assertChdir(tmpdir)) {
1824*543adbedSBen Gras fprintf(stderr, "ERROR: Couldn't chdir to temp dir %s\n",
1825*543adbedSBen Gras tmpdir);
1826*543adbedSBen Gras exit(1);
1827*543adbedSBen Gras }
1828*543adbedSBen Gras /* Report per-test summaries. */
1829*543adbedSBen Gras tests[i].failures = failures - failures_before;
1830*543adbedSBen Gras test_summarize(test_filename, tests[i].failures);
1831*543adbedSBen Gras /* Close the per-test log file. */
1832*543adbedSBen Gras fclose(logfile);
1833*543adbedSBen Gras logfile = NULL;
1834*543adbedSBen Gras /* If there were no failures, we can remove the work dir and logfile. */
1835*543adbedSBen Gras if (tests[i].failures == 0) {
1836*543adbedSBen Gras if (!keep_temp_files && assertChdir(tmpdir)) {
1837*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
1838*543adbedSBen Gras /* Make sure not to leave empty directories.
1839*543adbedSBen Gras * Sometimes a processing of closing files used by tests
1840*543adbedSBen Gras * is not done, then rmdir will be failed and it will
1841*543adbedSBen Gras * leave a empty test directory. So we should wait a few
1842*543adbedSBen Gras * seconds and retry rmdir. */
1843*543adbedSBen Gras int r, t;
1844*543adbedSBen Gras for (t = 0; t < 10; t++) {
1845*543adbedSBen Gras if (t > 0)
1846*543adbedSBen Gras Sleep(1000);
1847*543adbedSBen Gras r = systemf("rmdir /S /Q %s", tests[i].name);
1848*543adbedSBen Gras if (r == 0)
1849*543adbedSBen Gras break;
1850*543adbedSBen Gras }
1851*543adbedSBen Gras systemf("del %s", logfilename);
1852*543adbedSBen Gras #else
1853*543adbedSBen Gras systemf("rm -rf %s", tests[i].name);
1854*543adbedSBen Gras systemf("rm %s", logfilename);
1855*543adbedSBen Gras #endif
1856*543adbedSBen Gras }
1857*543adbedSBen Gras }
1858*543adbedSBen Gras /* Return appropriate status. */
1859*543adbedSBen Gras return (tests[i].failures);
1860*543adbedSBen Gras }
1861*543adbedSBen Gras
1862*543adbedSBen Gras /*
1863*543adbedSBen Gras *
1864*543adbedSBen Gras *
1865*543adbedSBen Gras * MAIN and support routines.
1866*543adbedSBen Gras *
1867*543adbedSBen Gras *
1868*543adbedSBen Gras */
1869*543adbedSBen Gras
1870*543adbedSBen Gras static void
usage(const char * program)1871*543adbedSBen Gras usage(const char *program)
1872*543adbedSBen Gras {
1873*543adbedSBen Gras static const int limit = sizeof(tests) / sizeof(tests[0]);
1874*543adbedSBen Gras int i;
1875*543adbedSBen Gras
1876*543adbedSBen Gras printf("Usage: %s [options] <test> <test> ...\n", program);
1877*543adbedSBen Gras printf("Default is to run all tests.\n");
1878*543adbedSBen Gras printf("Otherwise, specify the numbers of the tests you wish to run.\n");
1879*543adbedSBen Gras printf("Options:\n");
1880*543adbedSBen Gras printf(" -d Dump core after any failure, for debugging.\n");
1881*543adbedSBen Gras printf(" -k Keep all temp files.\n");
1882*543adbedSBen Gras printf(" Default: temp files for successful tests deleted.\n");
1883*543adbedSBen Gras #ifdef PROGRAM
1884*543adbedSBen Gras printf(" -p <path> Path to executable to be tested.\n");
1885*543adbedSBen Gras printf(" Default: path taken from " ENVBASE " environment variable.\n");
1886*543adbedSBen Gras #endif
1887*543adbedSBen Gras printf(" -q Quiet.\n");
1888*543adbedSBen Gras printf(" -r <dir> Path to dir containing reference files.\n");
1889*543adbedSBen Gras printf(" Default: Current directory.\n");
1890*543adbedSBen Gras printf(" -v Verbose.\n");
1891*543adbedSBen Gras printf("Available tests:\n");
1892*543adbedSBen Gras for (i = 0; i < limit; i++)
1893*543adbedSBen Gras printf(" %d: %s\n", i, tests[i].name);
1894*543adbedSBen Gras exit(1);
1895*543adbedSBen Gras }
1896*543adbedSBen Gras
1897*543adbedSBen Gras static char *
get_refdir(const char * d)1898*543adbedSBen Gras get_refdir(const char *d)
1899*543adbedSBen Gras {
1900*543adbedSBen Gras char tried[512] = { '\0' };
1901*543adbedSBen Gras char buff[128];
1902*543adbedSBen Gras char *pwd, *p;
1903*543adbedSBen Gras
1904*543adbedSBen Gras /* If a dir was specified, try that */
1905*543adbedSBen Gras if (d != NULL) {
1906*543adbedSBen Gras pwd = NULL;
1907*543adbedSBen Gras snprintf(buff, sizeof(buff), "%s", d);
1908*543adbedSBen Gras p = slurpfile(NULL, "%s/%s", buff, KNOWNREF);
1909*543adbedSBen Gras if (p != NULL) goto success;
1910*543adbedSBen Gras strncat(tried, buff, sizeof(tried) - strlen(tried) - 1);
1911*543adbedSBen Gras strncat(tried, "\n", sizeof(tried) - strlen(tried) - 1);
1912*543adbedSBen Gras goto failure;
1913*543adbedSBen Gras }
1914*543adbedSBen Gras
1915*543adbedSBen Gras /* Get the current dir. */
1916*543adbedSBen Gras pwd = getcwd(NULL, 0);
1917*543adbedSBen Gras while (pwd[strlen(pwd) - 1] == '\n')
1918*543adbedSBen Gras pwd[strlen(pwd) - 1] = '\0';
1919*543adbedSBen Gras
1920*543adbedSBen Gras /* Look for a known file. */
1921*543adbedSBen Gras snprintf(buff, sizeof(buff), "%s", pwd);
1922*543adbedSBen Gras p = slurpfile(NULL, "%s/%s", buff, KNOWNREF);
1923*543adbedSBen Gras if (p != NULL) goto success;
1924*543adbedSBen Gras strncat(tried, buff, sizeof(tried) - strlen(tried) - 1);
1925*543adbedSBen Gras strncat(tried, "\n", sizeof(tried) - strlen(tried) - 1);
1926*543adbedSBen Gras
1927*543adbedSBen Gras snprintf(buff, sizeof(buff), "%s/test", pwd);
1928*543adbedSBen Gras p = slurpfile(NULL, "%s/%s", buff, KNOWNREF);
1929*543adbedSBen Gras if (p != NULL) goto success;
1930*543adbedSBen Gras strncat(tried, buff, sizeof(tried) - strlen(tried) - 1);
1931*543adbedSBen Gras strncat(tried, "\n", sizeof(tried) - strlen(tried) - 1);
1932*543adbedSBen Gras
1933*543adbedSBen Gras #if defined(LIBRARY)
1934*543adbedSBen Gras snprintf(buff, sizeof(buff), "%s/%s/test", pwd, LIBRARY);
1935*543adbedSBen Gras #else
1936*543adbedSBen Gras snprintf(buff, sizeof(buff), "%s/%s/test", pwd, PROGRAM);
1937*543adbedSBen Gras #endif
1938*543adbedSBen Gras p = slurpfile(NULL, "%s/%s", buff, KNOWNREF);
1939*543adbedSBen Gras if (p != NULL) goto success;
1940*543adbedSBen Gras strncat(tried, buff, sizeof(tried) - strlen(tried) - 1);
1941*543adbedSBen Gras strncat(tried, "\n", sizeof(tried) - strlen(tried) - 1);
1942*543adbedSBen Gras
1943*543adbedSBen Gras if (memcmp(pwd, "/usr/obj", 8) == 0) {
1944*543adbedSBen Gras snprintf(buff, sizeof(buff), "%s", pwd + 8);
1945*543adbedSBen Gras p = slurpfile(NULL, "%s/%s", buff, KNOWNREF);
1946*543adbedSBen Gras if (p != NULL) goto success;
1947*543adbedSBen Gras strncat(tried, buff, sizeof(tried) - strlen(tried) - 1);
1948*543adbedSBen Gras strncat(tried, "\n", sizeof(tried) - strlen(tried) - 1);
1949*543adbedSBen Gras
1950*543adbedSBen Gras snprintf(buff, sizeof(buff), "%s/test", pwd + 8);
1951*543adbedSBen Gras p = slurpfile(NULL, "%s/%s", buff, KNOWNREF);
1952*543adbedSBen Gras if (p != NULL) goto success;
1953*543adbedSBen Gras strncat(tried, buff, sizeof(tried) - strlen(tried) - 1);
1954*543adbedSBen Gras strncat(tried, "\n", sizeof(tried) - strlen(tried) - 1);
1955*543adbedSBen Gras }
1956*543adbedSBen Gras
1957*543adbedSBen Gras failure:
1958*543adbedSBen Gras printf("Unable to locate known reference file %s\n", KNOWNREF);
1959*543adbedSBen Gras printf(" Checked following directories:\n%s\n", tried);
1960*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__) && defined(_DEBUG)
1961*543adbedSBen Gras DebugBreak();
1962*543adbedSBen Gras #endif
1963*543adbedSBen Gras exit(1);
1964*543adbedSBen Gras
1965*543adbedSBen Gras success:
1966*543adbedSBen Gras free(p);
1967*543adbedSBen Gras free(pwd);
1968*543adbedSBen Gras return strdup(buff);
1969*543adbedSBen Gras }
1970*543adbedSBen Gras
1971*543adbedSBen Gras int
main(int argc,char ** argv)1972*543adbedSBen Gras main(int argc, char **argv)
1973*543adbedSBen Gras {
1974*543adbedSBen Gras static const int limit = sizeof(tests) / sizeof(tests[0]);
1975*543adbedSBen Gras int i, tests_run = 0, tests_failed = 0, option;
1976*543adbedSBen Gras time_t now;
1977*543adbedSBen Gras char *refdir_alloc = NULL;
1978*543adbedSBen Gras const char *progname;
1979*543adbedSBen Gras const char *tmp, *option_arg, *p;
1980*543adbedSBen Gras char tmpdir[256];
1981*543adbedSBen Gras char tmpdir_timestamp[256];
1982*543adbedSBen Gras
1983*543adbedSBen Gras (void)argc; /* UNUSED */
1984*543adbedSBen Gras
1985*543adbedSBen Gras #if defined(HAVE__CrtSetReportMode)
1986*543adbedSBen Gras /* To stop to run the default invalid parameter handler. */
1987*543adbedSBen Gras _set_invalid_parameter_handler(invalid_parameter_handler);
1988*543adbedSBen Gras /* Disable annoying assertion message box. */
1989*543adbedSBen Gras _CrtSetReportMode(_CRT_ASSERT, 0);
1990*543adbedSBen Gras #endif
1991*543adbedSBen Gras
1992*543adbedSBen Gras /*
1993*543adbedSBen Gras * Name of this program, used to build root of our temp directory
1994*543adbedSBen Gras * tree.
1995*543adbedSBen Gras */
1996*543adbedSBen Gras progname = p = argv[0];
1997*543adbedSBen Gras while (*p != '\0') {
1998*543adbedSBen Gras /* Support \ or / dir separators for Windows compat. */
1999*543adbedSBen Gras if (*p == '/' || *p == '\\')
2000*543adbedSBen Gras progname = p + 1;
2001*543adbedSBen Gras ++p;
2002*543adbedSBen Gras }
2003*543adbedSBen Gras
2004*543adbedSBen Gras #ifdef PROGRAM
2005*543adbedSBen Gras /* Get the target program from environment, if available. */
2006*543adbedSBen Gras testprogfile = getenv(ENVBASE);
2007*543adbedSBen Gras #endif
2008*543adbedSBen Gras
2009*543adbedSBen Gras if (getenv("TMPDIR") != NULL)
2010*543adbedSBen Gras tmp = getenv("TMPDIR");
2011*543adbedSBen Gras else if (getenv("TMP") != NULL)
2012*543adbedSBen Gras tmp = getenv("TMP");
2013*543adbedSBen Gras else if (getenv("TEMP") != NULL)
2014*543adbedSBen Gras tmp = getenv("TEMP");
2015*543adbedSBen Gras else if (getenv("TEMPDIR") != NULL)
2016*543adbedSBen Gras tmp = getenv("TEMPDIR");
2017*543adbedSBen Gras else
2018*543adbedSBen Gras tmp = "/tmp";
2019*543adbedSBen Gras
2020*543adbedSBen Gras /* Allow -d to be controlled through the environment. */
2021*543adbedSBen Gras if (getenv(ENVBASE "_DEBUG") != NULL)
2022*543adbedSBen Gras dump_on_failure = 1;
2023*543adbedSBen Gras
2024*543adbedSBen Gras /* Get the directory holding test files from environment. */
2025*543adbedSBen Gras refdir = getenv(ENVBASE "_TEST_FILES");
2026*543adbedSBen Gras
2027*543adbedSBen Gras /*
2028*543adbedSBen Gras * Parse options, without using getopt(), which isn't available
2029*543adbedSBen Gras * on all platforms.
2030*543adbedSBen Gras */
2031*543adbedSBen Gras ++argv; /* Skip program name */
2032*543adbedSBen Gras while (*argv != NULL) {
2033*543adbedSBen Gras if (**argv != '-')
2034*543adbedSBen Gras break;
2035*543adbedSBen Gras p = *argv++;
2036*543adbedSBen Gras ++p; /* Skip '-' */
2037*543adbedSBen Gras while (*p != '\0') {
2038*543adbedSBen Gras option = *p++;
2039*543adbedSBen Gras option_arg = NULL;
2040*543adbedSBen Gras /* If 'opt' takes an argument, parse that. */
2041*543adbedSBen Gras if (option == 'p' || option == 'r') {
2042*543adbedSBen Gras if (*p != '\0')
2043*543adbedSBen Gras option_arg = p;
2044*543adbedSBen Gras else if (*argv == NULL) {
2045*543adbedSBen Gras fprintf(stderr,
2046*543adbedSBen Gras "Option -%c requires argument.\n",
2047*543adbedSBen Gras option);
2048*543adbedSBen Gras usage(progname);
2049*543adbedSBen Gras } else
2050*543adbedSBen Gras option_arg = *argv++;
2051*543adbedSBen Gras p = ""; /* End of this option word. */
2052*543adbedSBen Gras }
2053*543adbedSBen Gras
2054*543adbedSBen Gras /* Now, handle the option. */
2055*543adbedSBen Gras switch (option) {
2056*543adbedSBen Gras case 'd':
2057*543adbedSBen Gras dump_on_failure = 1;
2058*543adbedSBen Gras break;
2059*543adbedSBen Gras case 'k':
2060*543adbedSBen Gras keep_temp_files = 1;
2061*543adbedSBen Gras break;
2062*543adbedSBen Gras case 'p':
2063*543adbedSBen Gras #ifdef PROGRAM
2064*543adbedSBen Gras testprogfile = option_arg;
2065*543adbedSBen Gras #else
2066*543adbedSBen Gras fprintf(stderr, "-p option not permitted\n");
2067*543adbedSBen Gras usage(progname);
2068*543adbedSBen Gras #endif
2069*543adbedSBen Gras break;
2070*543adbedSBen Gras case 'q':
2071*543adbedSBen Gras verbosity--;
2072*543adbedSBen Gras break;
2073*543adbedSBen Gras case 'r':
2074*543adbedSBen Gras refdir = option_arg;
2075*543adbedSBen Gras break;
2076*543adbedSBen Gras case 'v':
2077*543adbedSBen Gras verbosity++;
2078*543adbedSBen Gras break;
2079*543adbedSBen Gras default:
2080*543adbedSBen Gras fprintf(stderr, "Unrecognized option '%c'\n",
2081*543adbedSBen Gras option);
2082*543adbedSBen Gras usage(progname);
2083*543adbedSBen Gras }
2084*543adbedSBen Gras }
2085*543adbedSBen Gras }
2086*543adbedSBen Gras
2087*543adbedSBen Gras /*
2088*543adbedSBen Gras * Sanity-check that our options make sense.
2089*543adbedSBen Gras */
2090*543adbedSBen Gras #ifdef PROGRAM
2091*543adbedSBen Gras if (testprogfile == NULL) {
2092*543adbedSBen Gras fprintf(stderr, "Program executable required\n");
2093*543adbedSBen Gras usage(progname);
2094*543adbedSBen Gras }
2095*543adbedSBen Gras
2096*543adbedSBen Gras {
2097*543adbedSBen Gras char *testprg;
2098*543adbedSBen Gras #if defined(_WIN32) && !defined(__CYGWIN__)
2099*543adbedSBen Gras /* Command.com sometimes rejects '/' separators. */
2100*543adbedSBen Gras testprg = strdup(testprogfile);
2101*543adbedSBen Gras for (i = 0; testprg[i] != '\0'; i++) {
2102*543adbedSBen Gras if (testprg[i] == '/')
2103*543adbedSBen Gras testprg[i] = '\\';
2104*543adbedSBen Gras }
2105*543adbedSBen Gras testprogfile = testprg;
2106*543adbedSBen Gras #endif
2107*543adbedSBen Gras /* Quote the name that gets put into shell command lines. */
2108*543adbedSBen Gras testprg = malloc(strlen(testprogfile) + 3);
2109*543adbedSBen Gras strcpy(testprg, "\"");
2110*543adbedSBen Gras strcat(testprg, testprogfile);
2111*543adbedSBen Gras strcat(testprg, "\"");
2112*543adbedSBen Gras testprog = testprg;
2113*543adbedSBen Gras }
2114*543adbedSBen Gras #endif
2115*543adbedSBen Gras
2116*543adbedSBen Gras /*
2117*543adbedSBen Gras * Create a temp directory for the following tests.
2118*543adbedSBen Gras * Include the time the tests started as part of the name,
2119*543adbedSBen Gras * to make it easier to track the results of multiple tests.
2120*543adbedSBen Gras */
2121*543adbedSBen Gras now = time(NULL);
2122*543adbedSBen Gras for (i = 0; ; i++) {
2123*543adbedSBen Gras strftime(tmpdir_timestamp, sizeof(tmpdir_timestamp),
2124*543adbedSBen Gras "%Y-%m-%dT%H.%M.%S",
2125*543adbedSBen Gras localtime(&now));
2126*543adbedSBen Gras sprintf(tmpdir, "%s/%s.%s-%03d", tmp, progname,
2127*543adbedSBen Gras tmpdir_timestamp, i);
2128*543adbedSBen Gras if (assertMakeDir(tmpdir,0755))
2129*543adbedSBen Gras break;
2130*543adbedSBen Gras if (i >= 999) {
2131*543adbedSBen Gras fprintf(stderr,
2132*543adbedSBen Gras "ERROR: Unable to create temp directory %s\n",
2133*543adbedSBen Gras tmpdir);
2134*543adbedSBen Gras exit(1);
2135*543adbedSBen Gras }
2136*543adbedSBen Gras }
2137*543adbedSBen Gras
2138*543adbedSBen Gras /*
2139*543adbedSBen Gras * If the user didn't specify a directory for locating
2140*543adbedSBen Gras * reference files, try to find the reference files in
2141*543adbedSBen Gras * the "usual places."
2142*543adbedSBen Gras */
2143*543adbedSBen Gras refdir = refdir_alloc = get_refdir(refdir);
2144*543adbedSBen Gras
2145*543adbedSBen Gras /*
2146*543adbedSBen Gras * Banner with basic information.
2147*543adbedSBen Gras */
2148*543adbedSBen Gras printf("\n");
2149*543adbedSBen Gras printf("If tests fail or crash, details will be in:\n");
2150*543adbedSBen Gras printf(" %s\n", tmpdir);
2151*543adbedSBen Gras printf("\n");
2152*543adbedSBen Gras if (verbosity > VERBOSITY_SUMMARY_ONLY) {
2153*543adbedSBen Gras printf("Reference files will be read from: %s\n", refdir);
2154*543adbedSBen Gras #ifdef PROGRAM
2155*543adbedSBen Gras printf("Running tests on: %s\n", testprog);
2156*543adbedSBen Gras #endif
2157*543adbedSBen Gras printf("Exercising: ");
2158*543adbedSBen Gras fflush(stdout);
2159*543adbedSBen Gras printf("%s\n", EXTRA_VERSION);
2160*543adbedSBen Gras } else {
2161*543adbedSBen Gras printf("Running ");
2162*543adbedSBen Gras fflush(stdout);
2163*543adbedSBen Gras }
2164*543adbedSBen Gras
2165*543adbedSBen Gras /*
2166*543adbedSBen Gras * Run some or all of the individual tests.
2167*543adbedSBen Gras */
2168*543adbedSBen Gras if (*argv == NULL) {
2169*543adbedSBen Gras /* Default: Run all tests. */
2170*543adbedSBen Gras for (i = 0; i < limit; i++) {
2171*543adbedSBen Gras if (test_run(i, tmpdir))
2172*543adbedSBen Gras tests_failed++;
2173*543adbedSBen Gras tests_run++;
2174*543adbedSBen Gras }
2175*543adbedSBen Gras } else {
2176*543adbedSBen Gras while (*(argv) != NULL) {
2177*543adbedSBen Gras if (**argv >= '0' && **argv <= '9') {
2178*543adbedSBen Gras i = atoi(*argv);
2179*543adbedSBen Gras if (i < 0 || i >= limit) {
2180*543adbedSBen Gras printf("*** INVALID Test %s\n", *argv);
2181*543adbedSBen Gras free(refdir_alloc);
2182*543adbedSBen Gras usage(progname);
2183*543adbedSBen Gras /* usage() never returns */
2184*543adbedSBen Gras }
2185*543adbedSBen Gras } else {
2186*543adbedSBen Gras for (i = 0; i < limit; ++i) {
2187*543adbedSBen Gras if (strcmp(*argv, tests[i].name) == 0)
2188*543adbedSBen Gras break;
2189*543adbedSBen Gras }
2190*543adbedSBen Gras if (i >= limit) {
2191*543adbedSBen Gras printf("*** INVALID Test ``%s''\n",
2192*543adbedSBen Gras *argv);
2193*543adbedSBen Gras free(refdir_alloc);
2194*543adbedSBen Gras usage(progname);
2195*543adbedSBen Gras /* usage() never returns */
2196*543adbedSBen Gras }
2197*543adbedSBen Gras }
2198*543adbedSBen Gras if (test_run(i, tmpdir))
2199*543adbedSBen Gras tests_failed++;
2200*543adbedSBen Gras tests_run++;
2201*543adbedSBen Gras argv++;
2202*543adbedSBen Gras }
2203*543adbedSBen Gras }
2204*543adbedSBen Gras
2205*543adbedSBen Gras /*
2206*543adbedSBen Gras * Report summary statistics.
2207*543adbedSBen Gras */
2208*543adbedSBen Gras if (verbosity > VERBOSITY_SUMMARY_ONLY) {
2209*543adbedSBen Gras printf("\n");
2210*543adbedSBen Gras printf("Totals:\n");
2211*543adbedSBen Gras printf(" Tests run: %8d\n", tests_run);
2212*543adbedSBen Gras printf(" Tests failed: %8d\n", tests_failed);
2213*543adbedSBen Gras printf(" Assertions checked:%8d\n", assertions);
2214*543adbedSBen Gras printf(" Assertions failed: %8d\n", failures);
2215*543adbedSBen Gras printf(" Skips reported: %8d\n", skips);
2216*543adbedSBen Gras }
2217*543adbedSBen Gras if (failures) {
2218*543adbedSBen Gras printf("\n");
2219*543adbedSBen Gras printf("Failing tests:\n");
2220*543adbedSBen Gras for (i = 0; i < limit; ++i) {
2221*543adbedSBen Gras if (tests[i].failures)
2222*543adbedSBen Gras printf(" %d: %s (%d failures)\n", i,
2223*543adbedSBen Gras tests[i].name, tests[i].failures);
2224*543adbedSBen Gras }
2225*543adbedSBen Gras printf("\n");
2226*543adbedSBen Gras printf("Details for failing tests: %s\n", tmpdir);
2227*543adbedSBen Gras printf("\n");
2228*543adbedSBen Gras } else {
2229*543adbedSBen Gras if (verbosity == VERBOSITY_SUMMARY_ONLY)
2230*543adbedSBen Gras printf("\n");
2231*543adbedSBen Gras printf("%d tests passed, no failures\n", tests_run);
2232*543adbedSBen Gras }
2233*543adbedSBen Gras
2234*543adbedSBen Gras free(refdir_alloc);
2235*543adbedSBen Gras
2236*543adbedSBen Gras /* If the final tmpdir is empty, we can remove it. */
2237*543adbedSBen Gras /* This should be the usual case when all tests succeed. */
2238*543adbedSBen Gras assertChdir("..");
2239*543adbedSBen Gras rmdir(tmpdir);
2240*543adbedSBen Gras
2241*543adbedSBen Gras return (tests_failed ? 1 : 0);
2242*543adbedSBen Gras }
2243