main.c revision 305188
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 305188 2016-09-01 07:53:59Z 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 return (1); 1612#endif 1613 failure_start(file, line, "Could not create directory %s", dirname); 1614 failure_finish(NULL); 1615 return(0); 1616} 1617 1618/* Create a file with the specified contents and report any failures. */ 1619int 1620assertion_make_file(const char *file, int line, 1621 const char *path, int mode, int csize, const void *contents) 1622{ 1623#if defined(_WIN32) && !defined(__CYGWIN__) 1624 /* TODO: Rework this to set file mode as well. */ 1625 FILE *f; 1626 (void)mode; /* UNUSED */ 1627 assertion_count(file, line); 1628 f = fopen(path, "wb"); 1629 if (f == NULL) { 1630 failure_start(file, line, "Could not create file %s", path); 1631 failure_finish(NULL); 1632 return (0); 1633 } 1634 if (contents != NULL) { 1635 size_t wsize; 1636 1637 if (csize < 0) 1638 wsize = strlen(contents); 1639 else 1640 wsize = (size_t)csize; 1641 if (wsize != fwrite(contents, 1, wsize, f)) { 1642 fclose(f); 1643 failure_start(file, line, 1644 "Could not write file %s", path); 1645 failure_finish(NULL); 1646 return (0); 1647 } 1648 } 1649 fclose(f); 1650 return (1); 1651#else 1652 int fd; 1653 assertion_count(file, line); 1654 fd = open(path, O_CREAT | O_WRONLY, mode >= 0 ? mode : 0644); 1655 if (fd < 0) { 1656 failure_start(file, line, "Could not create %s", path); 1657 failure_finish(NULL); 1658 return (0); 1659 } 1660 if (contents != NULL) { 1661 ssize_t wsize; 1662 1663 if (csize < 0) 1664 wsize = (ssize_t)strlen(contents); 1665 else 1666 wsize = (ssize_t)csize; 1667 if (wsize != write(fd, contents, wsize)) { 1668 close(fd); 1669 failure_start(file, line, 1670 "Could not write to %s", path); 1671 failure_finish(NULL); 1672 return (0); 1673 } 1674 } 1675 close(fd); 1676 return (1); 1677#endif 1678} 1679 1680/* Create a hardlink and report any failures. */ 1681int 1682assertion_make_hardlink(const char *file, int line, 1683 const char *newpath, const char *linkto) 1684{ 1685 int succeeded; 1686 1687 assertion_count(file, line); 1688#if defined(_WIN32) && !defined(__CYGWIN__) 1689 succeeded = my_CreateHardLinkA(newpath, linkto); 1690#elif HAVE_LINK 1691 succeeded = !link(linkto, newpath); 1692#else 1693 succeeded = 0; 1694#endif 1695 if (succeeded) 1696 return (1); 1697 failure_start(file, line, "Could not create hardlink"); 1698 logprintf(" New link: %s\n", newpath); 1699 logprintf(" Old name: %s\n", linkto); 1700 failure_finish(NULL); 1701 return(0); 1702} 1703 1704/* Create a symlink and report any failures. */ 1705int 1706assertion_make_symlink(const char *file, int line, 1707 const char *newpath, const char *linkto) 1708{ 1709#if defined(_WIN32) && !defined(__CYGWIN__) 1710 int targetIsDir = 0; /* TODO: Fix this */ 1711 assertion_count(file, line); 1712 if (my_CreateSymbolicLinkA(newpath, linkto, targetIsDir)) 1713 return (1); 1714#elif HAVE_SYMLINK 1715 assertion_count(file, line); 1716 if (0 == symlink(linkto, newpath)) 1717 return (1); 1718#endif 1719 failure_start(file, line, "Could not create symlink"); 1720 logprintf(" New link: %s\n", newpath); 1721 logprintf(" Old name: %s\n", linkto); 1722 failure_finish(NULL); 1723 return(0); 1724} 1725 1726/* Set umask, report failures. */ 1727int 1728assertion_umask(const char *file, int line, int mask) 1729{ 1730 assertion_count(file, line); 1731 (void)file; /* UNUSED */ 1732 (void)line; /* UNUSED */ 1733 umask(mask); 1734 return (1); 1735} 1736 1737/* Set times, report failures. */ 1738int 1739assertion_utimes(const char *file, int line, 1740 const char *pathname, long at, long at_nsec, long mt, long mt_nsec) 1741{ 1742 int r; 1743 1744#if defined(_WIN32) && !defined(__CYGWIN__) 1745#define WINTIME(sec, nsec) ((Int32x32To64(sec, 10000000) + EPOC_TIME)\ 1746 + (((nsec)/1000)*10)) 1747 HANDLE h; 1748 ULARGE_INTEGER wintm; 1749 FILETIME fatime, fmtime; 1750 FILETIME *pat, *pmt; 1751 1752 assertion_count(file, line); 1753 h = CreateFileA(pathname,GENERIC_READ | GENERIC_WRITE, 1754 FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 1755 FILE_FLAG_BACKUP_SEMANTICS, NULL); 1756 if (h == INVALID_HANDLE_VALUE) { 1757 failure_start(file, line, "Can't access %s\n", pathname); 1758 failure_finish(NULL); 1759 return (0); 1760 } 1761 1762 if (at > 0 || at_nsec > 0) { 1763 wintm.QuadPart = WINTIME(at, at_nsec); 1764 fatime.dwLowDateTime = wintm.LowPart; 1765 fatime.dwHighDateTime = wintm.HighPart; 1766 pat = &fatime; 1767 } else 1768 pat = NULL; 1769 if (mt > 0 || mt_nsec > 0) { 1770 wintm.QuadPart = WINTIME(mt, mt_nsec); 1771 fmtime.dwLowDateTime = wintm.LowPart; 1772 fmtime.dwHighDateTime = wintm.HighPart; 1773 pmt = &fmtime; 1774 } else 1775 pmt = NULL; 1776 if (pat != NULL || pmt != NULL) 1777 r = SetFileTime(h, NULL, pat, pmt); 1778 else 1779 r = 1; 1780 CloseHandle(h); 1781 if (r == 0) { 1782 failure_start(file, line, "Can't SetFileTime %s\n", pathname); 1783 failure_finish(NULL); 1784 return (0); 1785 } 1786 return (1); 1787#else /* defined(_WIN32) && !defined(__CYGWIN__) */ 1788 struct stat st; 1789 struct timeval times[2]; 1790 1791#if !defined(__FreeBSD__) 1792 mt_nsec = at_nsec = 0; /* Generic POSIX only has whole seconds. */ 1793#endif 1794 if (mt == 0 && mt_nsec == 0 && at == 0 && at_nsec == 0) 1795 return (1); 1796 1797 r = lstat(pathname, &st); 1798 if (r < 0) { 1799 failure_start(file, line, "Can't stat %s\n", pathname); 1800 failure_finish(NULL); 1801 return (0); 1802 } 1803 1804 if (mt == 0 && mt_nsec == 0) { 1805 mt = st.st_mtime; 1806#if defined(__FreeBSD__) 1807 mt_nsec = st.st_mtimespec.tv_nsec; 1808 /* FreeBSD generally only stores to microsecond res, so round. */ 1809 mt_nsec = (mt_nsec / 1000) * 1000; 1810#endif 1811 } 1812 if (at == 0 && at_nsec == 0) { 1813 at = st.st_atime; 1814#if defined(__FreeBSD__) 1815 at_nsec = st.st_atimespec.tv_nsec; 1816 /* FreeBSD generally only stores to microsecond res, so round. */ 1817 at_nsec = (at_nsec / 1000) * 1000; 1818#endif 1819 } 1820 1821 times[1].tv_sec = mt; 1822 times[1].tv_usec = mt_nsec / 1000; 1823 1824 times[0].tv_sec = at; 1825 times[0].tv_usec = at_nsec / 1000; 1826 1827#ifdef HAVE_LUTIMES 1828 r = lutimes(pathname, times); 1829#else 1830 r = utimes(pathname, times); 1831#endif 1832 if (r < 0) { 1833 failure_start(file, line, "Can't utimes %s\n", pathname); 1834 failure_finish(NULL); 1835 return (0); 1836 } 1837 return (1); 1838#endif /* defined(_WIN32) && !defined(__CYGWIN__) */ 1839} 1840 1841/* Set nodump, report failures. */ 1842int 1843assertion_nodump(const char *file, int line, const char *pathname) 1844{ 1845#if defined(HAVE_STRUCT_STAT_ST_FLAGS) && defined(UF_NODUMP) 1846 int r; 1847 1848 assertion_count(file, line); 1849 r = chflags(pathname, UF_NODUMP); 1850 if (r < 0) { 1851 failure_start(file, line, "Can't set nodump %s\n", pathname); 1852 failure_finish(NULL); 1853 return (0); 1854 } 1855#elif defined(EXT2_IOC_GETFLAGS) && defined(HAVE_WORKING_EXT2_IOC_GETFLAGS)\ 1856 && defined(EXT2_NODUMP_FL) 1857 int fd, r, flags; 1858 1859 assertion_count(file, line); 1860 fd = open(pathname, O_RDONLY | O_NONBLOCK); 1861 if (fd < 0) { 1862 failure_start(file, line, "Can't open %s\n", pathname); 1863 failure_finish(NULL); 1864 return (0); 1865 } 1866 r = ioctl(fd, EXT2_IOC_GETFLAGS, &flags); 1867 if (r < 0) { 1868 failure_start(file, line, "Can't get flags %s\n", pathname); 1869 failure_finish(NULL); 1870 return (0); 1871 } 1872 flags |= EXT2_NODUMP_FL; 1873 r = ioctl(fd, EXT2_IOC_SETFLAGS, &flags); 1874 if (r < 0) { 1875 failure_start(file, line, "Can't set nodump %s\n", pathname); 1876 failure_finish(NULL); 1877 return (0); 1878 } 1879 close(fd); 1880#else 1881 (void)pathname; /* UNUSED */ 1882 assertion_count(file, line); 1883#endif 1884 return (1); 1885} 1886 1887/* 1888 * 1889 * UTILITIES for use by tests. 1890 * 1891 */ 1892 1893/* 1894 * Check whether platform supports symlinks. This is intended 1895 * for tests to use in deciding whether to bother testing symlink 1896 * support; if the platform doesn't support symlinks, there's no point 1897 * in checking whether the program being tested can create them. 1898 * 1899 * Note that the first time this test is called, we actually go out to 1900 * disk to create and verify a symlink. This is necessary because 1901 * symlink support is actually a property of a particular filesystem 1902 * and can thus vary between directories on a single system. After 1903 * the first call, this returns the cached result from memory, so it's 1904 * safe to call it as often as you wish. 1905 */ 1906int 1907canSymlink(void) 1908{ 1909 /* Remember the test result */ 1910 static int value = 0, tested = 0; 1911 if (tested) 1912 return (value); 1913 1914 ++tested; 1915 assertion_make_file(__FILE__, __LINE__, "canSymlink.0", 0644, 1, "a"); 1916 /* Note: Cygwin has its own symlink() emulation that does not 1917 * use the Win32 CreateSymbolicLink() function. */ 1918#if defined(_WIN32) && !defined(__CYGWIN__) 1919 value = my_CreateSymbolicLinkA("canSymlink.1", "canSymlink.0", 0) 1920 && is_symlink(__FILE__, __LINE__, "canSymlink.1", "canSymlink.0"); 1921#elif HAVE_SYMLINK 1922 value = (0 == symlink("canSymlink.0", "canSymlink.1")) 1923 && is_symlink(__FILE__, __LINE__, "canSymlink.1","canSymlink.0"); 1924#endif 1925 return (value); 1926} 1927 1928/* Platform-dependent options for hiding the output of a subcommand. */ 1929#if defined(_WIN32) && !defined(__CYGWIN__) 1930static const char *redirectArgs = ">NUL 2>NUL"; /* Win32 cmd.exe */ 1931#else 1932static const char *redirectArgs = ">/dev/null 2>/dev/null"; /* POSIX 'sh' */ 1933#endif 1934/* 1935 * Can this platform run the bzip2 program? 1936 */ 1937int 1938canBzip2(void) 1939{ 1940 static int tested = 0, value = 0; 1941 if (!tested) { 1942 tested = 1; 1943 if (systemf("bzip2 -d -V %s", redirectArgs) == 0) 1944 value = 1; 1945 } 1946 return (value); 1947} 1948 1949/* 1950 * Can this platform run the grzip program? 1951 */ 1952int 1953canGrzip(void) 1954{ 1955 static int tested = 0, value = 0; 1956 if (!tested) { 1957 tested = 1; 1958 if (systemf("grzip -V %s", redirectArgs) == 0) 1959 value = 1; 1960 } 1961 return (value); 1962} 1963 1964/* 1965 * Can this platform run the gzip program? 1966 */ 1967int 1968canGzip(void) 1969{ 1970 static int tested = 0, value = 0; 1971 if (!tested) { 1972 tested = 1; 1973 if (systemf("gzip -V %s", redirectArgs) == 0) 1974 value = 1; 1975 } 1976 return (value); 1977} 1978 1979/* 1980 * Can this platform run the lrzip program? 1981 */ 1982int 1983canRunCommand(const char *cmd) 1984{ 1985 static int tested = 0, value = 0; 1986 if (!tested) { 1987 tested = 1; 1988 if (systemf("%s %s", cmd, redirectArgs) == 0) 1989 value = 1; 1990 } 1991 return (value); 1992} 1993 1994int 1995canLrzip(void) 1996{ 1997 static int tested = 0, value = 0; 1998 if (!tested) { 1999 tested = 1; 2000 if (systemf("lrzip -V %s", redirectArgs) == 0) 2001 value = 1; 2002 } 2003 return (value); 2004} 2005 2006/* 2007 * Can this platform run the lz4 program? 2008 */ 2009int 2010canLz4(void) 2011{ 2012 static int tested = 0, value = 0; 2013 if (!tested) { 2014 tested = 1; 2015 if (systemf("lz4 -V %s", redirectArgs) == 0) 2016 value = 1; 2017 } 2018 return (value); 2019} 2020 2021/* 2022 * Can this platform run the lzip program? 2023 */ 2024int 2025canLzip(void) 2026{ 2027 static int tested = 0, value = 0; 2028 if (!tested) { 2029 tested = 1; 2030 if (systemf("lzip -V %s", redirectArgs) == 0) 2031 value = 1; 2032 } 2033 return (value); 2034} 2035 2036/* 2037 * Can this platform run the lzma program? 2038 */ 2039int 2040canLzma(void) 2041{ 2042 static int tested = 0, value = 0; 2043 if (!tested) { 2044 tested = 1; 2045 if (systemf("lzma -V %s", redirectArgs) == 0) 2046 value = 1; 2047 } 2048 return (value); 2049} 2050 2051/* 2052 * Can this platform run the lzop program? 2053 */ 2054int 2055canLzop(void) 2056{ 2057 static int tested = 0, value = 0; 2058 if (!tested) { 2059 tested = 1; 2060 if (systemf("lzop -V %s", redirectArgs) == 0) 2061 value = 1; 2062 } 2063 return (value); 2064} 2065 2066/* 2067 * Can this platform run the xz program? 2068 */ 2069int 2070canXz(void) 2071{ 2072 static int tested = 0, value = 0; 2073 if (!tested) { 2074 tested = 1; 2075 if (systemf("xz -V %s", redirectArgs) == 0) 2076 value = 1; 2077 } 2078 return (value); 2079} 2080 2081/* 2082 * Can this filesystem handle nodump flags. 2083 */ 2084#if defined(HAVE_STRUCT_STAT_ST_FLAGS) && defined(UF_NODUMP) 2085 2086int 2087canNodump(void) 2088{ 2089 const char *path = "cannodumptest"; 2090 struct stat sb; 2091 2092 assertion_make_file(__FILE__, __LINE__, path, 0644, 0, NULL); 2093 if (chflags(path, UF_NODUMP) < 0) 2094 return (0); 2095 if (stat(path, &sb) < 0) 2096 return (0); 2097 if (sb.st_flags & UF_NODUMP) 2098 return (1); 2099 return (0); 2100} 2101 2102#elif defined(EXT2_IOC_GETFLAGS) && defined(HAVE_WORKING_EXT2_IOC_GETFLAGS)\ 2103 && defined(EXT2_NODUMP_FL) 2104 2105int 2106canNodump(void) 2107{ 2108 const char *path = "cannodumptest"; 2109 int fd, r, flags; 2110 2111 assertion_make_file(__FILE__, __LINE__, path, 0644, 0, NULL); 2112 fd = open(path, O_RDONLY | O_NONBLOCK); 2113 if (fd < 0) 2114 return (0); 2115 r = ioctl(fd, EXT2_IOC_GETFLAGS, &flags); 2116 if (r < 0) 2117 return (0); 2118 flags |= EXT2_NODUMP_FL; 2119 r = ioctl(fd, EXT2_IOC_SETFLAGS, &flags); 2120 if (r < 0) 2121 return (0); 2122 close(fd); 2123 fd = open(path, O_RDONLY | O_NONBLOCK); 2124 if (fd < 0) 2125 return (0); 2126 r = ioctl(fd, EXT2_IOC_GETFLAGS, &flags); 2127 if (r < 0) 2128 return (0); 2129 close(fd); 2130 if (flags & EXT2_NODUMP_FL) 2131 return (1); 2132 return (0); 2133} 2134 2135#else 2136 2137int 2138canNodump() 2139{ 2140 return (0); 2141} 2142 2143#endif 2144 2145/* 2146 * Sleep as needed; useful for verifying disk timestamp changes by 2147 * ensuring that the wall-clock time has actually changed before we 2148 * go back to re-read something from disk. 2149 */ 2150void 2151sleepUntilAfter(time_t t) 2152{ 2153 while (t >= time(NULL)) 2154#if defined(_WIN32) && !defined(__CYGWIN__) 2155 Sleep(500); 2156#else 2157 sleep(1); 2158#endif 2159} 2160 2161/* 2162 * Call standard system() call, but build up the command line using 2163 * sprintf() conventions. 2164 */ 2165int 2166systemf(const char *fmt, ...) 2167{ 2168 char buff[8192]; 2169 va_list ap; 2170 int r; 2171 2172 va_start(ap, fmt); 2173 vsprintf(buff, fmt, ap); 2174 if (verbosity > VERBOSITY_FULL) 2175 logprintf("Cmd: %s\n", buff); 2176 r = system(buff); 2177 va_end(ap); 2178 return (r); 2179} 2180 2181/* 2182 * Slurp a file into memory for ease of comparison and testing. 2183 * Returns size of file in 'sizep' if non-NULL, null-terminates 2184 * data in memory for ease of use. 2185 */ 2186char * 2187slurpfile(size_t * sizep, const char *fmt, ...) 2188{ 2189 char filename[8192]; 2190 struct stat st; 2191 va_list ap; 2192 char *p; 2193 ssize_t bytes_read; 2194 FILE *f; 2195 int r; 2196 2197 va_start(ap, fmt); 2198 vsprintf(filename, fmt, ap); 2199 va_end(ap); 2200 2201 f = fopen(filename, "rb"); 2202 if (f == NULL) { 2203 /* Note: No error; non-existent file is okay here. */ 2204 return (NULL); 2205 } 2206 r = fstat(fileno(f), &st); 2207 if (r != 0) { 2208 logprintf("Can't stat file %s\n", filename); 2209 fclose(f); 2210 return (NULL); 2211 } 2212 p = malloc((size_t)st.st_size + 1); 2213 if (p == NULL) { 2214 logprintf("Can't allocate %ld bytes of memory to read file %s\n", 2215 (long int)st.st_size, filename); 2216 fclose(f); 2217 return (NULL); 2218 } 2219 bytes_read = fread(p, 1, (size_t)st.st_size, f); 2220 if (bytes_read < st.st_size) { 2221 logprintf("Can't read file %s\n", filename); 2222 fclose(f); 2223 free(p); 2224 return (NULL); 2225 } 2226 p[st.st_size] = '\0'; 2227 if (sizep != NULL) 2228 *sizep = (size_t)st.st_size; 2229 fclose(f); 2230 return (p); 2231} 2232 2233/* 2234 * Slurp a file into memory for ease of comparison and testing. 2235 * Returns size of file in 'sizep' if non-NULL, null-terminates 2236 * data in memory for ease of use. 2237 */ 2238void 2239dumpfile(const char *filename, void *data, size_t len) 2240{ 2241 ssize_t bytes_written; 2242 FILE *f; 2243 2244 f = fopen(filename, "wb"); 2245 if (f == NULL) { 2246 logprintf("Can't open file %s for writing\n", filename); 2247 return; 2248 } 2249 bytes_written = fwrite(data, 1, len, f); 2250 if (bytes_written < (ssize_t)len) 2251 logprintf("Can't write file %s\n", filename); 2252 fclose(f); 2253} 2254 2255/* Read a uuencoded file from the reference directory, decode, and 2256 * write the result into the current directory. */ 2257#define VALID_UUDECODE(c) (c >= 32 && c <= 96) 2258#define UUDECODE(c) (((c) - 0x20) & 0x3f) 2259void 2260extract_reference_file(const char *name) 2261{ 2262 char buff[1024]; 2263 FILE *in, *out; 2264 2265 sprintf(buff, "%s/%s.uu", refdir, name); 2266 in = fopen(buff, "r"); 2267 failure("Couldn't open reference file %s", buff); 2268 assert(in != NULL); 2269 if (in == NULL) 2270 return; 2271 /* Read up to and including the 'begin' line. */ 2272 for (;;) { 2273 if (fgets(buff, sizeof(buff), in) == NULL) { 2274 /* TODO: This is a failure. */ 2275 return; 2276 } 2277 if (memcmp(buff, "begin ", 6) == 0) 2278 break; 2279 } 2280 /* Now, decode the rest and write it. */ 2281 out = fopen(name, "wb"); 2282 while (fgets(buff, sizeof(buff), in) != NULL) { 2283 char *p = buff; 2284 int bytes; 2285 2286 if (memcmp(buff, "end", 3) == 0) 2287 break; 2288 2289 bytes = UUDECODE(*p++); 2290 while (bytes > 0) { 2291 int n = 0; 2292 /* Write out 1-3 bytes from that. */ 2293 if (bytes > 0) { 2294 assert(VALID_UUDECODE(p[0])); 2295 assert(VALID_UUDECODE(p[1])); 2296 n = UUDECODE(*p++) << 18; 2297 n |= UUDECODE(*p++) << 12; 2298 fputc(n >> 16, out); 2299 --bytes; 2300 } 2301 if (bytes > 0) { 2302 assert(VALID_UUDECODE(p[0])); 2303 n |= UUDECODE(*p++) << 6; 2304 fputc((n >> 8) & 0xFF, out); 2305 --bytes; 2306 } 2307 if (bytes > 0) { 2308 assert(VALID_UUDECODE(p[0])); 2309 n |= UUDECODE(*p++); 2310 fputc(n & 0xFF, out); 2311 --bytes; 2312 } 2313 } 2314 } 2315 fclose(out); 2316 fclose(in); 2317} 2318 2319void 2320copy_reference_file(const char *name) 2321{ 2322 char buff[1024]; 2323 FILE *in, *out; 2324 size_t rbytes; 2325 2326 sprintf(buff, "%s/%s", refdir, name); 2327 in = fopen(buff, "rb"); 2328 failure("Couldn't open reference file %s", buff); 2329 assert(in != NULL); 2330 if (in == NULL) 2331 return; 2332 /* Now, decode the rest and write it. */ 2333 /* Not a lot of error checking here; the input better be right. */ 2334 out = fopen(name, "wb"); 2335 while ((rbytes = fread(buff, 1, sizeof(buff), in)) > 0) { 2336 if (fwrite(buff, 1, rbytes, out) != rbytes) { 2337 logprintf("Error: fwrite\n"); 2338 break; 2339 } 2340 } 2341 fclose(out); 2342 fclose(in); 2343} 2344 2345int 2346is_LargeInode(const char *file) 2347{ 2348#if defined(_WIN32) && !defined(__CYGWIN__) 2349 BY_HANDLE_FILE_INFORMATION bhfi; 2350 int r; 2351 2352 r = my_GetFileInformationByName(file, &bhfi); 2353 if (r != 0) 2354 return (0); 2355 return (bhfi.nFileIndexHigh & 0x0000FFFFUL); 2356#else 2357 struct stat st; 2358 int64_t ino; 2359 2360 if (stat(file, &st) < 0) 2361 return (0); 2362 ino = (int64_t)st.st_ino; 2363 return (ino > 0xffffffff); 2364#endif 2365} 2366 2367void 2368extract_reference_files(const char **names) 2369{ 2370 while (names && *names) 2371 extract_reference_file(*names++); 2372} 2373 2374/* 2375 * 2376 * TEST management 2377 * 2378 */ 2379 2380/* 2381 * "list.h" is simply created by "grep DEFINE_TEST test_*.c"; it has 2382 * a line like 2383 * DEFINE_TEST(test_function) 2384 * for each test. 2385 */ 2386 2387/* Use "list.h" to declare all of the test functions. */ 2388#undef DEFINE_TEST 2389#define DEFINE_TEST(name) void name(void); 2390#include "list.h" 2391 2392/* Use "list.h" to create a list of all tests (functions and names). */ 2393#undef DEFINE_TEST 2394#define DEFINE_TEST(n) { n, #n, 0 }, 2395struct test_list_t tests[] = { 2396 #include "list.h" 2397}; 2398 2399/* 2400 * Summarize repeated failures in the just-completed test. 2401 */ 2402static void 2403test_summarize(int failed, int skips_num) 2404{ 2405 unsigned int i; 2406 2407 switch (verbosity) { 2408 case VERBOSITY_SUMMARY_ONLY: 2409 printf(failed ? "E" : "."); 2410 fflush(stdout); 2411 break; 2412 case VERBOSITY_PASSFAIL: 2413 printf(failed ? "FAIL\n" : skips_num ? "ok (S)\n" : "ok\n"); 2414 break; 2415 } 2416 2417 log_console = (verbosity == VERBOSITY_LIGHT_REPORT); 2418 2419 for (i = 0; i < sizeof(failed_lines)/sizeof(failed_lines[0]); i++) { 2420 if (failed_lines[i].count > 1 && !failed_lines[i].skip) 2421 logprintf("%s:%d: Summary: Failed %d times\n", 2422 failed_filename, i, failed_lines[i].count); 2423 } 2424 /* Clear the failure history for the next file. */ 2425 failed_filename = NULL; 2426 memset(failed_lines, 0, sizeof(failed_lines)); 2427} 2428 2429/* 2430 * Actually run a single test, with appropriate setup and cleanup. 2431 */ 2432static int 2433test_run(int i, const char *tmpdir) 2434{ 2435 char workdir[1024]; 2436 char logfilename[64]; 2437 int failures_before = failures; 2438 int skips_before = skips; 2439 int oldumask; 2440 2441 switch (verbosity) { 2442 case VERBOSITY_SUMMARY_ONLY: /* No per-test reports at all */ 2443 break; 2444 case VERBOSITY_PASSFAIL: /* rest of line will include ok/FAIL marker */ 2445 printf("%3d: %-64s", i, tests[i].name); 2446 fflush(stdout); 2447 break; 2448 default: /* Title of test, details will follow */ 2449 printf("%3d: %s\n", i, tests[i].name); 2450 } 2451 2452 /* Chdir to the top-level work directory. */ 2453 if (!assertChdir(tmpdir)) { 2454 fprintf(stderr, 2455 "ERROR: Can't chdir to top work dir %s\n", tmpdir); 2456 exit(1); 2457 } 2458 /* Create a log file for this test. */ 2459 sprintf(logfilename, "%s.log", tests[i].name); 2460 logfile = fopen(logfilename, "w"); 2461 fprintf(logfile, "%s\n\n", tests[i].name); 2462 /* Chdir() to a work dir for this specific test. */ 2463 snprintf(workdir, sizeof(workdir), "%s/%s", tmpdir, tests[i].name); 2464 testworkdir = workdir; 2465 if (!assertMakeDir(testworkdir, 0755) 2466 || !assertChdir(testworkdir)) { 2467 fprintf(stderr, 2468 "ERROR: Can't chdir to work dir %s\n", testworkdir); 2469 exit(1); 2470 } 2471 /* Explicitly reset the locale before each test. */ 2472 setlocale(LC_ALL, "C"); 2473 /* Record the umask before we run the test. */ 2474 umask(oldumask = umask(0)); 2475 /* 2476 * Run the actual test. 2477 */ 2478 (*tests[i].func)(); 2479 /* 2480 * Clean up and report afterwards. 2481 */ 2482 testworkdir = NULL; 2483 /* Restore umask */ 2484 umask(oldumask); 2485 /* Reset locale. */ 2486 setlocale(LC_ALL, "C"); 2487 /* Reset directory. */ 2488 if (!assertChdir(tmpdir)) { 2489 fprintf(stderr, "ERROR: Couldn't chdir to temp dir %s\n", 2490 tmpdir); 2491 exit(1); 2492 } 2493 /* Report per-test summaries. */ 2494 tests[i].failures = failures - failures_before; 2495 test_summarize(tests[i].failures, skips - skips_before); 2496 /* Close the per-test log file. */ 2497 fclose(logfile); 2498 logfile = NULL; 2499 /* If there were no failures, we can remove the work dir and logfile. */ 2500 if (tests[i].failures == 0) { 2501 if (!keep_temp_files && assertChdir(tmpdir)) { 2502#if defined(_WIN32) && !defined(__CYGWIN__) 2503 /* Make sure not to leave empty directories. 2504 * Sometimes a processing of closing files used by tests 2505 * is not done, then rmdir will be failed and it will 2506 * leave a empty test directory. So we should wait a few 2507 * seconds and retry rmdir. */ 2508 int r, t; 2509 for (t = 0; t < 10; t++) { 2510 if (t > 0) 2511 Sleep(1000); 2512 r = systemf("rmdir /S /Q %s", tests[i].name); 2513 if (r == 0) 2514 break; 2515 } 2516 systemf("del %s", logfilename); 2517#else 2518 systemf("rm -rf %s", tests[i].name); 2519 systemf("rm %s", logfilename); 2520#endif 2521 } 2522 } 2523 /* Return appropriate status. */ 2524 return (tests[i].failures); 2525} 2526 2527/* 2528 * 2529 * 2530 * MAIN and support routines. 2531 * 2532 * 2533 */ 2534 2535static void 2536usage(const char *program) 2537{ 2538 static const int limit = sizeof(tests) / sizeof(tests[0]); 2539 int i; 2540 2541 printf("Usage: %s [options] <test> <test> ...\n", program); 2542 printf("Default is to run all tests.\n"); 2543 printf("Otherwise, specify the numbers of the tests you wish to run.\n"); 2544 printf("Options:\n"); 2545 printf(" -d Dump core after any failure, for debugging.\n"); 2546 printf(" -k Keep all temp files.\n"); 2547 printf(" Default: temp files for successful tests deleted.\n"); 2548#ifdef PROGRAM 2549 printf(" -p <path> Path to executable to be tested.\n"); 2550 printf(" Default: path taken from " ENVBASE " environment variable.\n"); 2551#endif 2552 printf(" -q Quiet.\n"); 2553 printf(" -r <dir> Path to dir containing reference files.\n"); 2554 printf(" Default: Current directory.\n"); 2555 printf(" -u Keep running specifies tests until one fails.\n"); 2556 printf(" -v Verbose.\n"); 2557 printf("Available tests:\n"); 2558 for (i = 0; i < limit; i++) 2559 printf(" %d: %s\n", i, tests[i].name); 2560 exit(1); 2561} 2562 2563static char * 2564get_refdir(const char *d) 2565{ 2566 size_t tried_size, buff_size; 2567 char *buff, *tried, *pwd = NULL, *p = NULL; 2568 2569#ifdef PATH_MAX 2570 buff_size = PATH_MAX; 2571#else 2572 buff_size = 8192; 2573#endif 2574 buff = calloc(buff_size, 1); 2575 if (buff == NULL) { 2576 fprintf(stderr, "Unable to allocate memory\n"); 2577 exit(1); 2578 } 2579 2580 /* Allocate a buffer to hold the various directories we checked. */ 2581 tried_size = buff_size * 2; 2582 tried = calloc(tried_size, 1); 2583 if (tried == NULL) { 2584 fprintf(stderr, "Unable to allocate memory\n"); 2585 exit(1); 2586 } 2587 2588 /* If a dir was specified, try that */ 2589 if (d != NULL) { 2590 pwd = NULL; 2591 snprintf(buff, buff_size, "%s", d); 2592 p = slurpfile(NULL, "%s/%s", buff, KNOWNREF); 2593 if (p != NULL) goto success; 2594 strncat(tried, buff, tried_size - strlen(tried) - 1); 2595 strncat(tried, "\n", tried_size - strlen(tried) - 1); 2596 goto failure; 2597 } 2598 2599 /* Get the current dir. */ 2600#ifdef PATH_MAX 2601 pwd = getcwd(NULL, PATH_MAX);/* Solaris getcwd needs the size. */ 2602#else 2603 pwd = getcwd(NULL, 0); 2604#endif 2605 while (pwd[strlen(pwd) - 1] == '\n') 2606 pwd[strlen(pwd) - 1] = '\0'; 2607 2608 /* Look for a known file. */ 2609 snprintf(buff, buff_size, "%s", pwd); 2610 p = slurpfile(NULL, "%s/%s", buff, KNOWNREF); 2611 if (p != NULL) goto success; 2612 strncat(tried, buff, tried_size - strlen(tried) - 1); 2613 strncat(tried, "\n", tried_size - strlen(tried) - 1); 2614 2615 snprintf(buff, buff_size, "%s/test", pwd); 2616 p = slurpfile(NULL, "%s/%s", buff, KNOWNREF); 2617 if (p != NULL) goto success; 2618 strncat(tried, buff, tried_size - strlen(tried) - 1); 2619 strncat(tried, "\n", tried_size - strlen(tried) - 1); 2620 2621#if defined(LIBRARY) 2622 snprintf(buff, buff_size, "%s/%s/test", pwd, LIBRARY); 2623#else 2624 snprintf(buff, buff_size, "%s/%s/test", pwd, PROGRAM); 2625#endif 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(PROGRAM_ALIAS) 2632 snprintf(buff, buff_size, "%s/%s/test", pwd, PROGRAM_ALIAS); 2633 p = slurpfile(NULL, "%s/%s", buff, KNOWNREF); 2634 if (p != NULL) goto success; 2635 strncat(tried, buff, tried_size - strlen(tried) - 1); 2636 strncat(tried, "\n", tried_size - strlen(tried) - 1); 2637#endif 2638 2639 if (memcmp(pwd, "/usr/obj", 8) == 0) { 2640 snprintf(buff, buff_size, "%s", pwd + 8); 2641 p = slurpfile(NULL, "%s/%s", buff, KNOWNREF); 2642 if (p != NULL) goto success; 2643 strncat(tried, buff, tried_size - strlen(tried) - 1); 2644 strncat(tried, "\n", tried_size - strlen(tried) - 1); 2645 2646 snprintf(buff, buff_size, "%s/test", pwd + 8); 2647 p = slurpfile(NULL, "%s/%s", buff, KNOWNREF); 2648 if (p != NULL) goto success; 2649 strncat(tried, buff, tried_size - strlen(tried) - 1); 2650 strncat(tried, "\n", tried_size - strlen(tried) - 1); 2651 } 2652 2653failure: 2654 printf("Unable to locate known reference file %s\n", KNOWNREF); 2655 printf(" Checked following directories:\n%s\n", tried); 2656 printf("Use -r option to specify full path to reference directory\n"); 2657#if defined(_WIN32) && !defined(__CYGWIN__) && defined(_DEBUG) 2658 DebugBreak(); 2659#endif 2660 exit(1); 2661 2662success: 2663 free(p); 2664 free(pwd); 2665 free(tried); 2666 2667 /* Copy result into a fresh buffer to reduce memory usage. */ 2668 p = strdup(buff); 2669 free(buff); 2670 return p; 2671} 2672 2673int 2674main(int argc, char **argv) 2675{ 2676 static const int limit = sizeof(tests) / sizeof(tests[0]); 2677 int test_set[sizeof(tests) / sizeof(tests[0])]; 2678 int i = 0, j = 0, tests_run = 0, tests_failed = 0, option; 2679 time_t now; 2680 char *refdir_alloc = NULL; 2681 const char *progname; 2682 char **saved_argv; 2683 const char *tmp, *option_arg, *p; 2684 char tmpdir[256], *pwd, *testprogdir, *tmp2 = NULL, *vlevel = NULL; 2685 char tmpdir_timestamp[256]; 2686 2687 (void)argc; /* UNUSED */ 2688 2689 /* Get the current dir. */ 2690#ifdef PATH_MAX 2691 pwd = getcwd(NULL, PATH_MAX);/* Solaris getcwd needs the size. */ 2692#else 2693 pwd = getcwd(NULL, 0); 2694#endif 2695 while (pwd[strlen(pwd) - 1] == '\n') 2696 pwd[strlen(pwd) - 1] = '\0'; 2697 2698#if defined(HAVE__CrtSetReportMode) && !defined(__WATCOMC__) 2699 /* To stop to run the default invalid parameter handler. */ 2700 _set_invalid_parameter_handler(invalid_parameter_handler); 2701 /* Disable annoying assertion message box. */ 2702 _CrtSetReportMode(_CRT_ASSERT, 0); 2703#endif 2704 2705 /* 2706 * Name of this program, used to build root of our temp directory 2707 * tree. 2708 */ 2709 progname = p = argv[0]; 2710 if ((testprogdir = (char *)malloc(strlen(progname) + 1)) == NULL) 2711 { 2712 fprintf(stderr, "ERROR: Out of memory."); 2713 exit(1); 2714 } 2715 strcpy(testprogdir, progname); 2716 while (*p != '\0') { 2717 /* Support \ or / dir separators for Windows compat. */ 2718 if (*p == '/' || *p == '\\') 2719 { 2720 progname = p + 1; 2721 i = j; 2722 } 2723 ++p; 2724 j++; 2725 } 2726 testprogdir[i] = '\0'; 2727#if defined(_WIN32) && !defined(__CYGWIN__) 2728 if (testprogdir[0] != '/' && testprogdir[0] != '\\' && 2729 !(((testprogdir[0] >= 'a' && testprogdir[0] <= 'z') || 2730 (testprogdir[0] >= 'A' && testprogdir[0] <= 'Z')) && 2731 testprogdir[1] == ':' && 2732 (testprogdir[2] == '/' || testprogdir[2] == '\\'))) 2733#else 2734 if (testprogdir[0] != '/') 2735#endif 2736 { 2737 /* Fixup path for relative directories. */ 2738 if ((testprogdir = (char *)realloc(testprogdir, 2739 strlen(pwd) + 1 + strlen(testprogdir) + 1)) == NULL) 2740 { 2741 fprintf(stderr, "ERROR: Out of memory."); 2742 exit(1); 2743 } 2744 memmove(testprogdir + strlen(pwd) + 1, testprogdir, 2745 strlen(testprogdir) + 1); 2746 memcpy(testprogdir, pwd, strlen(pwd)); 2747 testprogdir[strlen(pwd)] = '/'; 2748 } 2749 2750#ifdef PROGRAM 2751 /* Get the target program from environment, if available. */ 2752 testprogfile = getenv(ENVBASE); 2753#endif 2754 2755 if (getenv("TMPDIR") != NULL) 2756 tmp = getenv("TMPDIR"); 2757 else if (getenv("TMP") != NULL) 2758 tmp = getenv("TMP"); 2759 else if (getenv("TEMP") != NULL) 2760 tmp = getenv("TEMP"); 2761 else if (getenv("TEMPDIR") != NULL) 2762 tmp = getenv("TEMPDIR"); 2763 else 2764 tmp = "/tmp"; 2765 2766 /* Allow -d to be controlled through the environment. */ 2767 if (getenv(ENVBASE "_DEBUG") != NULL) 2768 dump_on_failure = 1; 2769 2770 /* Allow -v to be controlled through the environment. */ 2771 if (getenv("_VERBOSITY_LEVEL") != NULL) 2772 { 2773 vlevel = getenv("_VERBOSITY_LEVEL"); 2774 verbosity = atoi(vlevel); 2775 if (verbosity < VERBOSITY_SUMMARY_ONLY || verbosity > VERBOSITY_FULL) 2776 { 2777 /* Unsupported verbosity levels are silently ignored */ 2778 vlevel = NULL; 2779 verbosity = VERBOSITY_PASSFAIL; 2780 } 2781 } 2782 2783 /* Get the directory holding test files from environment. */ 2784 refdir = getenv(ENVBASE "_TEST_FILES"); 2785 2786 /* 2787 * Parse options, without using getopt(), which isn't available 2788 * on all platforms. 2789 */ 2790 ++argv; /* Skip program name */ 2791 while (*argv != NULL) { 2792 if (**argv != '-') 2793 break; 2794 p = *argv++; 2795 ++p; /* Skip '-' */ 2796 while (*p != '\0') { 2797 option = *p++; 2798 option_arg = NULL; 2799 /* If 'opt' takes an argument, parse that. */ 2800 if (option == 'p' || option == 'r') { 2801 if (*p != '\0') 2802 option_arg = p; 2803 else if (*argv == NULL) { 2804 fprintf(stderr, 2805 "Option -%c requires argument.\n", 2806 option); 2807 usage(progname); 2808 } else 2809 option_arg = *argv++; 2810 p = ""; /* End of this option word. */ 2811 } 2812 2813 /* Now, handle the option. */ 2814 switch (option) { 2815 case 'd': 2816 dump_on_failure = 1; 2817 break; 2818 case 'k': 2819 keep_temp_files = 1; 2820 break; 2821 case 'p': 2822#ifdef PROGRAM 2823 testprogfile = option_arg; 2824#else 2825 fprintf(stderr, "-p option not permitted\n"); 2826 usage(progname); 2827#endif 2828 break; 2829 case 'q': 2830 if (!vlevel) 2831 verbosity--; 2832 break; 2833 case 'r': 2834 refdir = option_arg; 2835 break; 2836 case 'u': 2837 until_failure++; 2838 break; 2839 case 'v': 2840 if (!vlevel) 2841 verbosity++; 2842 break; 2843 default: 2844 fprintf(stderr, "Unrecognized option '%c'\n", 2845 option); 2846 usage(progname); 2847 } 2848 } 2849 } 2850 2851 /* 2852 * Sanity-check that our options make sense. 2853 */ 2854#ifdef PROGRAM 2855 if (testprogfile == NULL) 2856 { 2857 if ((tmp2 = (char *)malloc(strlen(testprogdir) + 1 + 2858 strlen(PROGRAM) + 1)) == NULL) 2859 { 2860 fprintf(stderr, "ERROR: Out of memory."); 2861 exit(1); 2862 } 2863 strcpy(tmp2, testprogdir); 2864 strcat(tmp2, "/"); 2865 strcat(tmp2, PROGRAM); 2866 testprogfile = tmp2; 2867 } 2868 2869 { 2870 char *testprg; 2871#if defined(_WIN32) && !defined(__CYGWIN__) 2872 /* Command.com sometimes rejects '/' separators. */ 2873 testprg = strdup(testprogfile); 2874 for (i = 0; testprg[i] != '\0'; i++) { 2875 if (testprg[i] == '/') 2876 testprg[i] = '\\'; 2877 } 2878 testprogfile = testprg; 2879#endif 2880 /* Quote the name that gets put into shell command lines. */ 2881 testprg = malloc(strlen(testprogfile) + 3); 2882 strcpy(testprg, "\""); 2883 strcat(testprg, testprogfile); 2884 strcat(testprg, "\""); 2885 testprog = testprg; 2886 } 2887#endif 2888 2889#if !defined(_WIN32) && defined(SIGPIPE) 2890 { /* Ignore SIGPIPE signals */ 2891 struct sigaction sa; 2892 sa.sa_handler = SIG_IGN; 2893 sigemptyset(&sa.sa_mask); 2894 sa.sa_flags = 0; 2895 sigaction(SIGPIPE, &sa, NULL); 2896 } 2897#endif 2898 2899 /* 2900 * Create a temp directory for the following tests. 2901 * Include the time the tests started as part of the name, 2902 * to make it easier to track the results of multiple tests. 2903 */ 2904 now = time(NULL); 2905 for (i = 0; ; i++) { 2906 strftime(tmpdir_timestamp, sizeof(tmpdir_timestamp), 2907 "%Y-%m-%dT%H.%M.%S", 2908 localtime(&now)); 2909 sprintf(tmpdir, "%s/%s.%s-%03d", tmp, progname, 2910 tmpdir_timestamp, i); 2911 if (assertMakeDir(tmpdir,0755)) 2912 break; 2913 if (i >= 999) { 2914 fprintf(stderr, 2915 "ERROR: Unable to create temp directory %s\n", 2916 tmpdir); 2917 exit(1); 2918 } 2919 } 2920 2921 /* 2922 * If the user didn't specify a directory for locating 2923 * reference files, try to find the reference files in 2924 * the "usual places." 2925 */ 2926 refdir = refdir_alloc = get_refdir(refdir); 2927 2928 /* 2929 * Banner with basic information. 2930 */ 2931 printf("\n"); 2932 printf("If tests fail or crash, details will be in:\n"); 2933 printf(" %s\n", tmpdir); 2934 printf("\n"); 2935 if (verbosity > VERBOSITY_SUMMARY_ONLY) { 2936 printf("Reference files will be read from: %s\n", refdir); 2937#ifdef PROGRAM 2938 printf("Running tests on: %s\n", testprog); 2939#endif 2940 printf("Exercising: "); 2941 fflush(stdout); 2942 printf("%s\n", EXTRA_VERSION); 2943 } else { 2944 printf("Running "); 2945 fflush(stdout); 2946 } 2947 2948 /* 2949 * Run some or all of the individual tests. 2950 */ 2951 saved_argv = argv; 2952 do { 2953 argv = saved_argv; 2954 do { 2955 int test_num; 2956 2957 test_num = get_test_set(test_set, limit, *argv, tests); 2958 if (test_num < 0) { 2959 printf("*** INVALID Test %s\n", *argv); 2960 free(refdir_alloc); 2961 free(testprogdir); 2962 usage(progname); 2963 return (1); 2964 } 2965 for (i = 0; i < test_num; i++) { 2966 tests_run++; 2967 if (test_run(test_set[i], tmpdir)) { 2968 tests_failed++; 2969 if (until_failure) 2970 goto finish; 2971 } 2972 } 2973 if (*argv != NULL) 2974 argv++; 2975 } while (*argv != NULL); 2976 } while (until_failure); 2977 2978finish: 2979 /* Must be freed after all tests run */ 2980 free(tmp2); 2981 free(testprogdir); 2982 free(pwd); 2983 2984 /* 2985 * Report summary statistics. 2986 */ 2987 if (verbosity > VERBOSITY_SUMMARY_ONLY) { 2988 printf("\n"); 2989 printf("Totals:\n"); 2990 printf(" Tests run: %8d\n", tests_run); 2991 printf(" Tests failed: %8d\n", tests_failed); 2992 printf(" Assertions checked:%8d\n", assertions); 2993 printf(" Assertions failed: %8d\n", failures); 2994 printf(" Skips reported: %8d\n", skips); 2995 } 2996 if (failures) { 2997 printf("\n"); 2998 printf("Failing tests:\n"); 2999 for (i = 0; i < limit; ++i) { 3000 if (tests[i].failures) 3001 printf(" %d: %s (%d failures)\n", i, 3002 tests[i].name, tests[i].failures); 3003 } 3004 printf("\n"); 3005 printf("Details for failing tests: %s\n", tmpdir); 3006 printf("\n"); 3007 } else { 3008 if (verbosity == VERBOSITY_SUMMARY_ONLY) 3009 printf("\n"); 3010 printf("%d tests passed, no failures\n", tests_run); 3011 } 3012 3013 free(refdir_alloc); 3014 3015 /* If the final tmpdir is empty, we can remove it. */ 3016 /* This should be the usual case when all tests succeed. */ 3017 assertChdir(".."); 3018 rmdir(tmpdir); 3019 3020 return (tests_failed ? 1 : 0); 3021} 3022