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