164287048SMartin Matuska /* 264287048SMartin Matuska * Copyright (c) 2003-2009 Tim Kientzle 364287048SMartin Matuska * All rights reserved. 464287048SMartin Matuska * 564287048SMartin Matuska * Redistribution and use in source and binary forms, with or without 664287048SMartin Matuska * modification, are permitted provided that the following conditions 764287048SMartin Matuska * are met: 864287048SMartin Matuska * 1. Redistributions of source code must retain the above copyright 964287048SMartin Matuska * notice, this list of conditions and the following disclaimer. 1064287048SMartin Matuska * 2. Redistributions in binary form must reproduce the above copyright 1164287048SMartin Matuska * notice, this list of conditions and the following disclaimer in the 1264287048SMartin Matuska * documentation and/or other materials provided with the distribution. 1364287048SMartin Matuska * 1464287048SMartin Matuska * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR 1564287048SMartin Matuska * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 1664287048SMartin Matuska * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 1764287048SMartin Matuska * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT, 1864287048SMartin Matuska * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 1964287048SMartin Matuska * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 2064287048SMartin Matuska * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 2164287048SMartin Matuska * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 2264287048SMartin Matuska * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 2364287048SMartin Matuska * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 2464287048SMartin Matuska */ 2564287048SMartin Matuska 2664287048SMartin Matuska #include "test.h" 2764287048SMartin Matuska #include "test_utils.h" 2864287048SMartin Matuska #ifdef HAVE_SYS_IOCTL_H 2964287048SMartin Matuska #include <sys/ioctl.h> 3064287048SMartin Matuska #endif 3164287048SMartin Matuska #ifdef HAVE_SYS_TIME_H 3264287048SMartin Matuska #include <sys/time.h> 3364287048SMartin Matuska #endif 3464287048SMartin Matuska #include <errno.h> 3564287048SMartin Matuska #ifdef HAVE_ICONV_H 3664287048SMartin Matuska #include <iconv.h> 3764287048SMartin Matuska #endif 3864287048SMartin Matuska /* 3964287048SMartin Matuska * Some Linux distributions have both linux/ext2_fs.h and ext2fs/ext2_fs.h. 4064287048SMartin Matuska * As the include guards don't agree, the order of include is important. 4164287048SMartin Matuska */ 4264287048SMartin Matuska #ifdef HAVE_LINUX_EXT2_FS_H 4364287048SMartin Matuska #include <linux/ext2_fs.h> /* for Linux file flags */ 4464287048SMartin Matuska #endif 4564287048SMartin Matuska #if defined(HAVE_EXT2FS_EXT2_FS_H) && !defined(__CYGWIN__) 4664287048SMartin Matuska #include <ext2fs/ext2_fs.h> /* Linux file flags, broken on Cygwin */ 4764287048SMartin Matuska #endif 4864287048SMartin Matuska #ifdef HAVE_LINUX_FS_H 4964287048SMartin Matuska #include <linux/fs.h> 5064287048SMartin Matuska #endif 5164287048SMartin Matuska #include <limits.h> 5264287048SMartin Matuska #include <locale.h> 5364287048SMartin Matuska #ifdef HAVE_SIGNAL_H 5464287048SMartin Matuska #include <signal.h> 5564287048SMartin Matuska #endif 5664287048SMartin Matuska #include <stdarg.h> 5764287048SMartin Matuska #include <time.h> 5864287048SMartin Matuska 594657548dSMartin Matuska #ifdef HAVE_SIGNAL_H 604657548dSMartin Matuska #endif 6164287048SMartin Matuska #ifdef HAVE_ACL_LIBACL_H 6264287048SMartin Matuska #include <acl/libacl.h> 6364287048SMartin Matuska #endif 6464287048SMartin Matuska #ifdef HAVE_SYS_TYPES_H 6564287048SMartin Matuska #include <sys/types.h> 6664287048SMartin Matuska #endif 6764287048SMartin Matuska #ifdef HAVE_SYS_ACL_H 6864287048SMartin Matuska #include <sys/acl.h> 6964287048SMartin Matuska #endif 70e46d4714SMartin Matuska #ifdef HAVE_SYS_EA_H 71e46d4714SMartin Matuska #include <sys/ea.h> 72e46d4714SMartin Matuska #endif 73e46d4714SMartin Matuska #ifdef HAVE_SYS_EXTATTR_H 74e46d4714SMartin Matuska #include <sys/extattr.h> 75e46d4714SMartin Matuska #endif 76e46d4714SMartin Matuska #if HAVE_SYS_XATTR_H 77e46d4714SMartin Matuska #include <sys/xattr.h> 78e46d4714SMartin Matuska #elif HAVE_ATTR_XATTR_H 79e46d4714SMartin Matuska #include <attr/xattr.h> 80e46d4714SMartin Matuska #endif 814657548dSMartin Matuska #ifdef HAVE_SYS_RICHACL_H 824657548dSMartin Matuska #include <sys/richacl.h> 834657548dSMartin Matuska #endif 844657548dSMartin Matuska #if HAVE_MEMBERSHIP_H 8564287048SMartin Matuska #include <membership.h> 8664287048SMartin Matuska #endif 8764287048SMartin Matuska 8864287048SMartin Matuska /* 8964287048SMartin Matuska * 9064287048SMartin Matuska * Windows support routines 9164287048SMartin Matuska * 9264287048SMartin Matuska * Note: Configuration is a tricky issue. Using HAVE_* feature macros 9364287048SMartin Matuska * in the test harness is dangerous because they cover up 9464287048SMartin Matuska * configuration errors. The classic example of this is omitting a 9564287048SMartin Matuska * configure check. If libarchive and libarchive_test both look for 9664287048SMartin Matuska * the same feature macro, such errors are hard to detect. Platform 9764287048SMartin Matuska * macros (e.g., _WIN32 or __GNUC__) are a little better, but can 9864287048SMartin Matuska * easily lead to very messy code. It's best to limit yourself 9964287048SMartin Matuska * to only the most generic programming techniques in the test harness 10064287048SMartin Matuska * and thus avoid conditionals altogether. Where that's not possible, 10164287048SMartin Matuska * try to minimize conditionals by grouping platform-specific tests in 10264287048SMartin Matuska * one place (e.g., test_acl_freebsd) or by adding new assert() 10364287048SMartin Matuska * functions (e.g., assertMakeHardlink()) to cover up platform 10464287048SMartin Matuska * differences. Platform-specific coding in libarchive_test is often 10564287048SMartin Matuska * a symptom that some capability is missing from libarchive itself. 10664287048SMartin Matuska */ 10764287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 10864287048SMartin Matuska #include <io.h> 10964287048SMartin Matuska #include <direct.h> 11064287048SMartin Matuska #include <windows.h> 11164287048SMartin Matuska #ifndef F_OK 11264287048SMartin Matuska #define F_OK (0) 11364287048SMartin Matuska #endif 11464287048SMartin Matuska #ifndef S_ISDIR 11564287048SMartin Matuska #define S_ISDIR(m) ((m) & _S_IFDIR) 11664287048SMartin Matuska #endif 11764287048SMartin Matuska #ifndef S_ISREG 11864287048SMartin Matuska #define S_ISREG(m) ((m) & _S_IFREG) 11964287048SMartin Matuska #endif 12064287048SMartin Matuska #if !defined(__BORLANDC__) 12164287048SMartin Matuska #define access _access 12264287048SMartin Matuska #undef chdir 12364287048SMartin Matuska #define chdir _chdir 12464287048SMartin Matuska #endif 12564287048SMartin Matuska #ifndef fileno 12664287048SMartin Matuska #define fileno _fileno 12764287048SMartin Matuska #endif 12864287048SMartin Matuska /*#define fstat _fstat64*/ 12964287048SMartin Matuska #if !defined(__BORLANDC__) 13064287048SMartin Matuska #define getcwd _getcwd 13164287048SMartin Matuska #endif 13264287048SMartin Matuska #define lstat stat 13364287048SMartin Matuska /*#define lstat _stat64*/ 13464287048SMartin Matuska /*#define stat _stat64*/ 13564287048SMartin Matuska #define rmdir _rmdir 13664287048SMartin Matuska #if !defined(__BORLANDC__) 13764287048SMartin Matuska #define strdup _strdup 13864287048SMartin Matuska #define umask _umask 13964287048SMartin Matuska #endif 14064287048SMartin Matuska #define int64_t __int64 14164287048SMartin Matuska #endif 14264287048SMartin Matuska 14364287048SMartin Matuska #if defined(HAVE__CrtSetReportMode) 14464287048SMartin Matuska # include <crtdbg.h> 14564287048SMartin Matuska #endif 14664287048SMartin Matuska 14764287048SMartin Matuska mode_t umasked(mode_t expected_mode) 14864287048SMartin Matuska { 14964287048SMartin Matuska mode_t mode = umask(0); 15064287048SMartin Matuska umask(mode); 15164287048SMartin Matuska return expected_mode & ~mode; 15264287048SMartin Matuska } 15364287048SMartin Matuska 15464287048SMartin Matuska /* Path to working directory for current test */ 15564287048SMartin Matuska const char *testworkdir; 15664287048SMartin Matuska #ifdef PROGRAM 15764287048SMartin Matuska /* Pathname of exe to be tested. */ 15864287048SMartin Matuska const char *testprogfile; 15964287048SMartin Matuska /* Name of exe to use in printf-formatted command strings. */ 16064287048SMartin Matuska /* On Windows, this includes leading/trailing quotes. */ 16164287048SMartin Matuska const char *testprog; 16264287048SMartin Matuska #endif 16364287048SMartin Matuska 16464287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 16564287048SMartin Matuska static void *GetFunctionKernel32(const char *); 16664287048SMartin Matuska static int my_CreateSymbolicLinkA(const char *, const char *, int); 16764287048SMartin Matuska static int my_CreateHardLinkA(const char *, const char *); 16864287048SMartin Matuska static int my_GetFileInformationByName(const char *, 16964287048SMartin Matuska BY_HANDLE_FILE_INFORMATION *); 17064287048SMartin Matuska 17152c2bb75SMartin Matuska typedef struct _REPARSE_DATA_BUFFER { 17252c2bb75SMartin Matuska ULONG ReparseTag; 17352c2bb75SMartin Matuska USHORT ReparseDataLength; 17452c2bb75SMartin Matuska USHORT Reserved; 17552c2bb75SMartin Matuska union { 17652c2bb75SMartin Matuska struct { 17752c2bb75SMartin Matuska USHORT SubstituteNameOffset; 17852c2bb75SMartin Matuska USHORT SubstituteNameLength; 17952c2bb75SMartin Matuska USHORT PrintNameOffset; 18052c2bb75SMartin Matuska USHORT PrintNameLength; 18152c2bb75SMartin Matuska ULONG Flags; 18252c2bb75SMartin Matuska WCHAR PathBuffer[1]; 18352c2bb75SMartin Matuska } SymbolicLinkReparseBuffer; 18452c2bb75SMartin Matuska struct { 18552c2bb75SMartin Matuska USHORT SubstituteNameOffset; 18652c2bb75SMartin Matuska USHORT SubstituteNameLength; 18752c2bb75SMartin Matuska USHORT PrintNameOffset; 18852c2bb75SMartin Matuska USHORT PrintNameLength; 18952c2bb75SMartin Matuska WCHAR PathBuffer[1]; 19052c2bb75SMartin Matuska } MountPointReparseBuffer; 19152c2bb75SMartin Matuska struct { 19252c2bb75SMartin Matuska UCHAR DataBuffer[1]; 19352c2bb75SMartin Matuska } GenericReparseBuffer; 19452c2bb75SMartin Matuska } DUMMYUNIONNAME; 19552c2bb75SMartin Matuska } REPARSE_DATA_BUFFER, *PREPARSE_DATA_BUFFER; 19652c2bb75SMartin Matuska 19764287048SMartin Matuska static void * 19864287048SMartin Matuska GetFunctionKernel32(const char *name) 19964287048SMartin Matuska { 20064287048SMartin Matuska static HINSTANCE lib; 20164287048SMartin Matuska static int set; 20264287048SMartin Matuska if (!set) { 20364287048SMartin Matuska set = 1; 20464287048SMartin Matuska lib = LoadLibrary("kernel32.dll"); 20564287048SMartin Matuska } 20664287048SMartin Matuska if (lib == NULL) { 20764287048SMartin Matuska fprintf(stderr, "Can't load kernel32.dll?!\n"); 20864287048SMartin Matuska exit(1); 20964287048SMartin Matuska } 21064287048SMartin Matuska return (void *)GetProcAddress(lib, name); 21164287048SMartin Matuska } 21264287048SMartin Matuska 21364287048SMartin Matuska static int 21452c2bb75SMartin Matuska my_CreateSymbolicLinkA(const char *linkname, const char *target, 21552c2bb75SMartin Matuska int targetIsDir) 21664287048SMartin Matuska { 21764287048SMartin Matuska static BOOLEAN (WINAPI *f)(LPCSTR, LPCSTR, DWORD); 21852c2bb75SMartin Matuska DWORD attrs; 21964287048SMartin Matuska static int set; 22052c2bb75SMartin Matuska int ret, tmpflags, llen, tlen; 22152c2bb75SMartin Matuska int flags = 0; 22252c2bb75SMartin Matuska char *src, *tgt, *p; 22364287048SMartin Matuska if (!set) { 22464287048SMartin Matuska set = 1; 22564287048SMartin Matuska f = GetFunctionKernel32("CreateSymbolicLinkA"); 22664287048SMartin Matuska } 22752c2bb75SMartin Matuska if (f == NULL) 22852c2bb75SMartin Matuska return (0); 22952c2bb75SMartin Matuska 23052c2bb75SMartin Matuska tlen = strlen(target); 23152c2bb75SMartin Matuska llen = strlen(linkname); 23252c2bb75SMartin Matuska 23352c2bb75SMartin Matuska if (tlen == 0 || llen == 0) 23452c2bb75SMartin Matuska return (0); 23552c2bb75SMartin Matuska 23652c2bb75SMartin Matuska tgt = malloc((tlen + 1) * sizeof(char)); 23752c2bb75SMartin Matuska if (tgt == NULL) 23852c2bb75SMartin Matuska return (0); 23952c2bb75SMartin Matuska src = malloc((llen + 1) * sizeof(char)); 24052c2bb75SMartin Matuska if (src == NULL) { 24152c2bb75SMartin Matuska free(tgt); 24252c2bb75SMartin Matuska return (0); 24352c2bb75SMartin Matuska } 24452c2bb75SMartin Matuska 24552c2bb75SMartin Matuska /* 24652c2bb75SMartin Matuska * Translate slashes to backslashes 24752c2bb75SMartin Matuska */ 24852c2bb75SMartin Matuska p = src; 24952c2bb75SMartin Matuska while(*linkname != '\0') { 25052c2bb75SMartin Matuska if (*linkname == '/') 25152c2bb75SMartin Matuska *p = '\\'; 25252c2bb75SMartin Matuska else 25352c2bb75SMartin Matuska *p = *linkname; 25452c2bb75SMartin Matuska linkname++; 25552c2bb75SMartin Matuska p++; 25652c2bb75SMartin Matuska } 25752c2bb75SMartin Matuska *p = '\0'; 25852c2bb75SMartin Matuska 25952c2bb75SMartin Matuska p = tgt; 26052c2bb75SMartin Matuska while(*target != '\0') { 26152c2bb75SMartin Matuska if (*target == '/') 26252c2bb75SMartin Matuska *p = '\\'; 26352c2bb75SMartin Matuska else 26452c2bb75SMartin Matuska *p = *target; 26552c2bb75SMartin Matuska target++; 26652c2bb75SMartin Matuska p++; 26752c2bb75SMartin Matuska } 26852c2bb75SMartin Matuska *p = '\0'; 26952c2bb75SMartin Matuska 27052c2bb75SMartin Matuska /* 27152c2bb75SMartin Matuska * Each test has to specify if a file or a directory symlink 27252c2bb75SMartin Matuska * should be created. 27352c2bb75SMartin Matuska */ 27452c2bb75SMartin Matuska if (targetIsDir) { 27552c2bb75SMartin Matuska #if defined(SYMBOLIC_LINK_FLAG_DIRECTORY) 27652c2bb75SMartin Matuska flags |= SYMBOLIC_LINK_FLAG_DIRECTORY; 27752c2bb75SMartin Matuska #else 27852c2bb75SMartin Matuska flags |= 0x1; 27952c2bb75SMartin Matuska #endif 28052c2bb75SMartin Matuska } 28152c2bb75SMartin Matuska 28252c2bb75SMartin Matuska #if defined(SYMBOLIC_LINK_FLAG_ALLOW_UNPRIVILEGED_CREATE) 28352c2bb75SMartin Matuska tmpflags = flags | SYMBOLIC_LINK_FLAG_ALLOW_UNPRIVILEGED_CREATE; 28452c2bb75SMartin Matuska #else 28552c2bb75SMartin Matuska tmpflags = flags | 0x2; 28652c2bb75SMartin Matuska #endif 28752c2bb75SMartin Matuska /* 28852c2bb75SMartin Matuska * Windows won't overwrite existing links 28952c2bb75SMartin Matuska */ 29052c2bb75SMartin Matuska attrs = GetFileAttributesA(linkname); 29152c2bb75SMartin Matuska if (attrs != INVALID_FILE_ATTRIBUTES) { 29252c2bb75SMartin Matuska if (attrs & FILE_ATTRIBUTE_DIRECTORY) 29352c2bb75SMartin Matuska RemoveDirectoryA(linkname); 29452c2bb75SMartin Matuska else 29552c2bb75SMartin Matuska DeleteFileA(linkname); 29652c2bb75SMartin Matuska } 29752c2bb75SMartin Matuska 29852c2bb75SMartin Matuska ret = (*f)(src, tgt, tmpflags); 29952c2bb75SMartin Matuska /* 30052c2bb75SMartin Matuska * Prior to Windows 10 the SYMBOLIC_LINK_FLAG_ALLOW_UNPRIVILEGED_CREATE 301f057565eSMartin Matuska * is not understood 30252c2bb75SMartin Matuska */ 30352c2bb75SMartin Matuska if (!ret) 30452c2bb75SMartin Matuska ret = (*f)(src, tgt, flags); 30552c2bb75SMartin Matuska 30652c2bb75SMartin Matuska free(src); 30752c2bb75SMartin Matuska free(tgt); 30852c2bb75SMartin Matuska return (ret); 30964287048SMartin Matuska } 31064287048SMartin Matuska 31164287048SMartin Matuska static int 31264287048SMartin Matuska my_CreateHardLinkA(const char *linkname, const char *target) 31364287048SMartin Matuska { 31464287048SMartin Matuska static BOOLEAN (WINAPI *f)(LPCSTR, LPCSTR, LPSECURITY_ATTRIBUTES); 31564287048SMartin Matuska static int set; 31664287048SMartin Matuska if (!set) { 31764287048SMartin Matuska set = 1; 31864287048SMartin Matuska f = GetFunctionKernel32("CreateHardLinkA"); 31964287048SMartin Matuska } 32064287048SMartin Matuska return f == NULL ? 0 : (*f)(linkname, target, NULL); 32164287048SMartin Matuska } 32264287048SMartin Matuska 32364287048SMartin Matuska static int 32464287048SMartin Matuska my_GetFileInformationByName(const char *path, BY_HANDLE_FILE_INFORMATION *bhfi) 32564287048SMartin Matuska { 32664287048SMartin Matuska HANDLE h; 32764287048SMartin Matuska int r; 32864287048SMartin Matuska 32964287048SMartin Matuska memset(bhfi, 0, sizeof(*bhfi)); 33064287048SMartin Matuska h = CreateFile(path, FILE_READ_ATTRIBUTES, 0, NULL, 33164287048SMartin Matuska OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL); 33264287048SMartin Matuska if (h == INVALID_HANDLE_VALUE) 33364287048SMartin Matuska return (0); 33464287048SMartin Matuska r = GetFileInformationByHandle(h, bhfi); 33564287048SMartin Matuska CloseHandle(h); 33664287048SMartin Matuska return (r); 33764287048SMartin Matuska } 33864287048SMartin Matuska #endif 33964287048SMartin Matuska 34064287048SMartin Matuska #if defined(HAVE__CrtSetReportMode) && !defined(__WATCOMC__) 34164287048SMartin Matuska static void 34264287048SMartin Matuska invalid_parameter_handler(const wchar_t * expression, 34364287048SMartin Matuska const wchar_t * function, const wchar_t * file, 34464287048SMartin Matuska unsigned int line, uintptr_t pReserved) 34564287048SMartin Matuska { 34664287048SMartin Matuska /* nop */ 34764287048SMartin Matuska // Silence unused-parameter compiler warnings. 34864287048SMartin Matuska (void)expression; 34964287048SMartin Matuska (void)function; 35064287048SMartin Matuska (void)file; 35164287048SMartin Matuska (void)line; 35264287048SMartin Matuska (void)pReserved; 35364287048SMartin Matuska } 35464287048SMartin Matuska #endif 35564287048SMartin Matuska 35664287048SMartin Matuska /* 35764287048SMartin Matuska * 35864287048SMartin Matuska * OPTIONS FLAGS 35964287048SMartin Matuska * 36064287048SMartin Matuska */ 36164287048SMartin Matuska 36264287048SMartin Matuska /* Enable core dump on failure. */ 36364287048SMartin Matuska static int dump_on_failure = 0; 36464287048SMartin Matuska /* Default is to remove temp dirs and log data for successful tests. */ 36564287048SMartin Matuska static int keep_temp_files = 0; 36664287048SMartin Matuska /* Default is to run the specified tests once and report errors. */ 36764287048SMartin Matuska static int until_failure = 0; 36864287048SMartin Matuska /* Default is to just report pass/fail for each test. */ 36964287048SMartin Matuska static int verbosity = 0; 37064287048SMartin Matuska #define VERBOSITY_SUMMARY_ONLY -1 /* -q */ 37164287048SMartin Matuska #define VERBOSITY_PASSFAIL 0 /* Default */ 37264287048SMartin Matuska #define VERBOSITY_LIGHT_REPORT 1 /* -v */ 37364287048SMartin Matuska #define VERBOSITY_FULL 2 /* -vv */ 37464287048SMartin Matuska /* A few places generate even more output for verbosity > VERBOSITY_FULL, 37564287048SMartin Matuska * mostly for debugging the test harness itself. */ 37664287048SMartin Matuska /* Cumulative count of assertion failures. */ 37764287048SMartin Matuska static int failures = 0; 37864287048SMartin Matuska /* Cumulative count of reported skips. */ 37964287048SMartin Matuska static int skips = 0; 38064287048SMartin Matuska /* Cumulative count of assertions checked. */ 38164287048SMartin Matuska static int assertions = 0; 38264287048SMartin Matuska 38364287048SMartin Matuska /* Directory where uuencoded reference files can be found. */ 38464287048SMartin Matuska static const char *refdir; 38564287048SMartin Matuska 38664287048SMartin Matuska /* 38764287048SMartin Matuska * Report log information selectively to console and/or disk log. 38864287048SMartin Matuska */ 38964287048SMartin Matuska static int log_console = 0; 39064287048SMartin Matuska static FILE *logfile; 391f9762417SMartin Matuska static void __LA_PRINTFLIKE(1, 0) 39264287048SMartin Matuska vlogprintf(const char *fmt, va_list ap) 39364287048SMartin Matuska { 39464287048SMartin Matuska #ifdef va_copy 39564287048SMartin Matuska va_list lfap; 39664287048SMartin Matuska va_copy(lfap, ap); 39764287048SMartin Matuska #endif 39864287048SMartin Matuska if (log_console) 39964287048SMartin Matuska vfprintf(stdout, fmt, ap); 40064287048SMartin Matuska if (logfile != NULL) 40164287048SMartin Matuska #ifdef va_copy 40264287048SMartin Matuska vfprintf(logfile, fmt, lfap); 40364287048SMartin Matuska va_end(lfap); 40464287048SMartin Matuska #else 40564287048SMartin Matuska vfprintf(logfile, fmt, ap); 40664287048SMartin Matuska #endif 40764287048SMartin Matuska } 40864287048SMartin Matuska 409f9762417SMartin Matuska static void __LA_PRINTFLIKE(1, 2) 41064287048SMartin Matuska logprintf(const char *fmt, ...) 41164287048SMartin Matuska { 41264287048SMartin Matuska va_list ap; 41364287048SMartin Matuska va_start(ap, fmt); 41464287048SMartin Matuska vlogprintf(fmt, ap); 41564287048SMartin Matuska va_end(ap); 41664287048SMartin Matuska } 41764287048SMartin Matuska 41864287048SMartin Matuska /* Set up a message to display only if next assertion fails. */ 41964287048SMartin Matuska static char msgbuff[4096]; 42064287048SMartin Matuska static const char *msg, *nextmsg; 42164287048SMartin Matuska void 42264287048SMartin Matuska failure(const char *fmt, ...) 42364287048SMartin Matuska { 42464287048SMartin Matuska va_list ap; 42564287048SMartin Matuska if (fmt == NULL) { 42664287048SMartin Matuska nextmsg = NULL; 42764287048SMartin Matuska } else { 42864287048SMartin Matuska va_start(ap, fmt); 42964287048SMartin Matuska vsprintf(msgbuff, fmt, ap); 43064287048SMartin Matuska va_end(ap); 43164287048SMartin Matuska nextmsg = msgbuff; 43264287048SMartin Matuska } 43364287048SMartin Matuska } 43464287048SMartin Matuska 43564287048SMartin Matuska /* 43664287048SMartin Matuska * Copy arguments into file-local variables. 43764287048SMartin Matuska * This was added to permit vararg assert() functions without needing 43864287048SMartin Matuska * variadic wrapper macros. Turns out that the vararg capability is almost 43964287048SMartin Matuska * never used, so almost all of the vararg assertions can be simplified 44064287048SMartin Matuska * by removing the vararg capability and reworking the wrapper macro to 44164287048SMartin Matuska * pass __FILE__, __LINE__ directly into the function instead of using 44264287048SMartin Matuska * this hook. I suspect this machinery is used so rarely that we 44364287048SMartin Matuska * would be better off just removing it entirely. That would simplify 44464287048SMartin Matuska * the code here noticeably. 44564287048SMartin Matuska */ 44664287048SMartin Matuska static const char *skipping_filename; 44764287048SMartin Matuska static int skipping_line; 44864287048SMartin Matuska void skipping_setup(const char *filename, int line) 44964287048SMartin Matuska { 45064287048SMartin Matuska skipping_filename = filename; 45164287048SMartin Matuska skipping_line = line; 45264287048SMartin Matuska } 45364287048SMartin Matuska 45464287048SMartin Matuska /* Called at the beginning of each assert() function. */ 45564287048SMartin Matuska static void 45664287048SMartin Matuska assertion_count(const char *file, int line) 45764287048SMartin Matuska { 45864287048SMartin Matuska (void)file; /* UNUSED */ 45964287048SMartin Matuska (void)line; /* UNUSED */ 46064287048SMartin Matuska ++assertions; 46164287048SMartin Matuska /* Proper handling of "failure()" message. */ 46264287048SMartin Matuska msg = nextmsg; 46364287048SMartin Matuska nextmsg = NULL; 46464287048SMartin Matuska /* Uncomment to print file:line after every assertion. 46564287048SMartin Matuska * Verbose, but occasionally useful in tracking down crashes. */ 46664287048SMartin Matuska /* printf("Checked %s:%d\n", file, line); */ 46764287048SMartin Matuska } 46864287048SMartin Matuska 46964287048SMartin Matuska /* 47064287048SMartin Matuska * For each test source file, we remember how many times each 47164287048SMartin Matuska * assertion was reported. Cleared before each new test, 47264287048SMartin Matuska * used by test_summarize(). 47364287048SMartin Matuska */ 47464287048SMartin Matuska static struct line { 47564287048SMartin Matuska int count; 47664287048SMartin Matuska int skip; 47764287048SMartin Matuska } failed_lines[10000]; 4780f8c7ba2SKyle Evans static const char *failed_filename; 47964287048SMartin Matuska 48064287048SMartin Matuska /* Count this failure, setup up log destination and handle initial report. */ 481f9762417SMartin Matuska static void __LA_PRINTFLIKE(3, 4) 48264287048SMartin Matuska failure_start(const char *filename, int line, const char *fmt, ...) 48364287048SMartin Matuska { 48464287048SMartin Matuska va_list ap; 48564287048SMartin Matuska 48664287048SMartin Matuska /* Record another failure for this line. */ 48764287048SMartin Matuska ++failures; 48864287048SMartin Matuska failed_filename = filename; 48964287048SMartin Matuska failed_lines[line].count++; 49064287048SMartin Matuska 49164287048SMartin Matuska /* Determine whether to log header to console. */ 49264287048SMartin Matuska switch (verbosity) { 49364287048SMartin Matuska case VERBOSITY_LIGHT_REPORT: 49464287048SMartin Matuska log_console = (failed_lines[line].count < 2); 49564287048SMartin Matuska break; 49664287048SMartin Matuska default: 49764287048SMartin Matuska log_console = (verbosity >= VERBOSITY_FULL); 49864287048SMartin Matuska } 49964287048SMartin Matuska 50064287048SMartin Matuska /* Log file:line header for this failure */ 50164287048SMartin Matuska va_start(ap, fmt); 50264287048SMartin Matuska #if _MSC_VER 50364287048SMartin Matuska logprintf("%s(%d): ", filename, line); 50464287048SMartin Matuska #else 50564287048SMartin Matuska logprintf("%s:%d: ", filename, line); 50664287048SMartin Matuska #endif 50764287048SMartin Matuska vlogprintf(fmt, ap); 50864287048SMartin Matuska va_end(ap); 50964287048SMartin Matuska logprintf("\n"); 51064287048SMartin Matuska 51164287048SMartin Matuska if (msg != NULL && msg[0] != '\0') { 51264287048SMartin Matuska logprintf(" Description: %s\n", msg); 51364287048SMartin Matuska msg = NULL; 51464287048SMartin Matuska } 51564287048SMartin Matuska 51664287048SMartin Matuska /* Determine whether to log details to console. */ 51764287048SMartin Matuska if (verbosity == VERBOSITY_LIGHT_REPORT) 51864287048SMartin Matuska log_console = 0; 51964287048SMartin Matuska } 52064287048SMartin Matuska 52164287048SMartin Matuska /* Complete reporting of failed tests. */ 52264287048SMartin Matuska /* 52364287048SMartin Matuska * The 'extra' hook here is used by libarchive to include libarchive 52464287048SMartin Matuska * error messages with assertion failures. It could also be used 52564287048SMartin Matuska * to add strerror() output, for example. Just define the EXTRA_DUMP() 52664287048SMartin Matuska * macro appropriately. 52764287048SMartin Matuska */ 52864287048SMartin Matuska static void 52964287048SMartin Matuska failure_finish(void *extra) 53064287048SMartin Matuska { 53164287048SMartin Matuska (void)extra; /* UNUSED (maybe) */ 53264287048SMartin Matuska #ifdef EXTRA_DUMP 53364287048SMartin Matuska if (extra != NULL) { 53464287048SMartin Matuska logprintf(" errno: %d\n", EXTRA_ERRNO(extra)); 53564287048SMartin Matuska logprintf(" detail: %s\n", EXTRA_DUMP(extra)); 53664287048SMartin Matuska } 53764287048SMartin Matuska #endif 53864287048SMartin Matuska 53964287048SMartin Matuska if (dump_on_failure) { 54064287048SMartin Matuska fprintf(stderr, 54164287048SMartin Matuska " *** forcing core dump so failure can be debugged ***\n"); 54264287048SMartin Matuska abort(); 54364287048SMartin Matuska } 54464287048SMartin Matuska } 54564287048SMartin Matuska 54664287048SMartin Matuska /* Inform user that we're skipping some checks. */ 54764287048SMartin Matuska void 54864287048SMartin Matuska test_skipping(const char *fmt, ...) 54964287048SMartin Matuska { 55064287048SMartin Matuska char buff[1024]; 55164287048SMartin Matuska va_list ap; 55264287048SMartin Matuska 55364287048SMartin Matuska va_start(ap, fmt); 55464287048SMartin Matuska vsprintf(buff, fmt, ap); 55564287048SMartin Matuska va_end(ap); 55664287048SMartin Matuska /* Use failure() message if set. */ 55764287048SMartin Matuska msg = nextmsg; 55864287048SMartin Matuska nextmsg = NULL; 55964287048SMartin Matuska /* failure_start() isn't quite right, but is awfully convenient. */ 56064287048SMartin Matuska failure_start(skipping_filename, skipping_line, "SKIPPING: %s", buff); 56164287048SMartin Matuska --failures; /* Undo failures++ in failure_start() */ 56264287048SMartin Matuska /* Don't failure_finish() here. */ 56364287048SMartin Matuska /* Mark as skip, so doesn't count as failed test. */ 56464287048SMartin Matuska failed_lines[skipping_line].skip = 1; 56564287048SMartin Matuska ++skips; 56664287048SMartin Matuska } 56764287048SMartin Matuska 56864287048SMartin Matuska /* 56964287048SMartin Matuska * 57064287048SMartin Matuska * ASSERTIONS 57164287048SMartin Matuska * 57264287048SMartin Matuska */ 57364287048SMartin Matuska 57464287048SMartin Matuska /* Generic assert() just displays the failed condition. */ 57564287048SMartin Matuska int 57664287048SMartin Matuska assertion_assert(const char *file, int line, int value, 57764287048SMartin Matuska const char *condition, void *extra) 57864287048SMartin Matuska { 57964287048SMartin Matuska assertion_count(file, line); 58064287048SMartin Matuska if (!value) { 58164287048SMartin Matuska failure_start(file, line, "Assertion failed: %s", condition); 58264287048SMartin Matuska failure_finish(extra); 58364287048SMartin Matuska } 58464287048SMartin Matuska return (value); 58564287048SMartin Matuska } 58664287048SMartin Matuska 58764287048SMartin Matuska /* chdir() and report any errors */ 58864287048SMartin Matuska int 58964287048SMartin Matuska assertion_chdir(const char *file, int line, const char *pathname) 59064287048SMartin Matuska { 59164287048SMartin Matuska assertion_count(file, line); 59264287048SMartin Matuska if (chdir(pathname) == 0) 59364287048SMartin Matuska return (1); 59464287048SMartin Matuska failure_start(file, line, "chdir(\"%s\")", pathname); 59564287048SMartin Matuska failure_finish(NULL); 59664287048SMartin Matuska return (0); 59764287048SMartin Matuska 59864287048SMartin Matuska } 59964287048SMartin Matuska 600c3afd20fSMartin Matuska /* change file/directory permissions and errors if it fails */ 601c3afd20fSMartin Matuska int 602c3afd20fSMartin Matuska assertion_chmod(const char *file, int line, const char *pathname, int mode) 603c3afd20fSMartin Matuska { 604c3afd20fSMartin Matuska assertion_count(file, line); 605c3afd20fSMartin Matuska if (chmod(pathname, mode) == 0) 606c3afd20fSMartin Matuska return (1); 607c3afd20fSMartin Matuska failure_start(file, line, "chmod(\"%s\", %4.o)", pathname, mode); 608c3afd20fSMartin Matuska failure_finish(NULL); 609c3afd20fSMartin Matuska return (0); 610c3afd20fSMartin Matuska 611c3afd20fSMartin Matuska } 612c3afd20fSMartin Matuska 61364287048SMartin Matuska /* Verify two integers are equal. */ 61464287048SMartin Matuska int 61564287048SMartin Matuska assertion_equal_int(const char *file, int line, 61664287048SMartin Matuska long long v1, const char *e1, long long v2, const char *e2, void *extra) 61764287048SMartin Matuska { 61864287048SMartin Matuska assertion_count(file, line); 61964287048SMartin Matuska if (v1 == v2) 62064287048SMartin Matuska return (1); 62164287048SMartin Matuska failure_start(file, line, "%s != %s", e1, e2); 62264287048SMartin Matuska logprintf(" %s=%lld (0x%llx, 0%llo)\n", e1, v1, v1, v1); 62364287048SMartin Matuska logprintf(" %s=%lld (0x%llx, 0%llo)\n", e2, v2, v2, v2); 62464287048SMartin Matuska failure_finish(extra); 62564287048SMartin Matuska return (0); 62664287048SMartin Matuska } 62764287048SMartin Matuska 62864287048SMartin Matuska /* 62964287048SMartin Matuska * Utility to convert a single UTF-8 sequence. 63064287048SMartin Matuska */ 63164287048SMartin Matuska static int 63264287048SMartin Matuska _utf8_to_unicode(uint32_t *pwc, const char *s, size_t n) 63364287048SMartin Matuska { 63464287048SMartin Matuska static const char utf8_count[256] = { 63564287048SMartin Matuska 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 00 - 0F */ 63664287048SMartin Matuska 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 10 - 1F */ 63764287048SMartin Matuska 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 20 - 2F */ 63864287048SMartin Matuska 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 30 - 3F */ 63964287048SMartin Matuska 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 40 - 4F */ 64064287048SMartin Matuska 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 50 - 5F */ 64164287048SMartin Matuska 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 60 - 6F */ 64264287048SMartin Matuska 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,/* 70 - 7F */ 64364287048SMartin Matuska 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,/* 80 - 8F */ 64464287048SMartin Matuska 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,/* 90 - 9F */ 64564287048SMartin Matuska 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,/* A0 - AF */ 64664287048SMartin Matuska 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,/* B0 - BF */ 64764287048SMartin Matuska 0, 0, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,/* C0 - CF */ 64864287048SMartin Matuska 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,/* D0 - DF */ 64964287048SMartin Matuska 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,/* E0 - EF */ 65064287048SMartin Matuska 4, 4, 4, 4, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 /* F0 - FF */ 65164287048SMartin Matuska }; 65264287048SMartin Matuska int ch; 65364287048SMartin Matuska int cnt; 65464287048SMartin Matuska uint32_t wc; 65564287048SMartin Matuska 65664287048SMartin Matuska *pwc = 0; 65764287048SMartin Matuska 65864287048SMartin Matuska /* Sanity check. */ 65964287048SMartin Matuska if (n == 0) 66064287048SMartin Matuska return (0); 66164287048SMartin Matuska /* 66264287048SMartin Matuska * Decode 1-4 bytes depending on the value of the first byte. 66364287048SMartin Matuska */ 66464287048SMartin Matuska ch = (unsigned char)*s; 66564287048SMartin Matuska if (ch == 0) 66664287048SMartin Matuska return (0); /* Standard: return 0 for end-of-string. */ 66764287048SMartin Matuska cnt = utf8_count[ch]; 66864287048SMartin Matuska 66964287048SMartin Matuska /* Invalid sequence or there are not plenty bytes. */ 67064287048SMartin Matuska if (n < (size_t)cnt) 67164287048SMartin Matuska return (-1); 67264287048SMartin Matuska 67364287048SMartin Matuska /* Make a Unicode code point from a single UTF-8 sequence. */ 67464287048SMartin Matuska switch (cnt) { 67564287048SMartin Matuska case 1: /* 1 byte sequence. */ 67664287048SMartin Matuska *pwc = ch & 0x7f; 67764287048SMartin Matuska return (cnt); 67864287048SMartin Matuska case 2: /* 2 bytes sequence. */ 67964287048SMartin Matuska if ((s[1] & 0xc0) != 0x80) return (-1); 68064287048SMartin Matuska *pwc = ((ch & 0x1f) << 6) | (s[1] & 0x3f); 68164287048SMartin Matuska return (cnt); 68264287048SMartin Matuska case 3: /* 3 bytes sequence. */ 68364287048SMartin Matuska if ((s[1] & 0xc0) != 0x80) return (-1); 68464287048SMartin Matuska if ((s[2] & 0xc0) != 0x80) return (-1); 68564287048SMartin Matuska wc = ((ch & 0x0f) << 12) 68664287048SMartin Matuska | ((s[1] & 0x3f) << 6) 68764287048SMartin Matuska | (s[2] & 0x3f); 68864287048SMartin Matuska if (wc < 0x800) 68964287048SMartin Matuska return (-1);/* Overlong sequence. */ 69064287048SMartin Matuska break; 69164287048SMartin Matuska case 4: /* 4 bytes sequence. */ 69264287048SMartin Matuska if (n < 4) 69364287048SMartin Matuska return (-1); 69464287048SMartin Matuska if ((s[1] & 0xc0) != 0x80) return (-1); 69564287048SMartin Matuska if ((s[2] & 0xc0) != 0x80) return (-1); 69664287048SMartin Matuska if ((s[3] & 0xc0) != 0x80) return (-1); 69764287048SMartin Matuska wc = ((ch & 0x07) << 18) 69864287048SMartin Matuska | ((s[1] & 0x3f) << 12) 69964287048SMartin Matuska | ((s[2] & 0x3f) << 6) 70064287048SMartin Matuska | (s[3] & 0x3f); 70164287048SMartin Matuska if (wc < 0x10000) 70264287048SMartin Matuska return (-1);/* Overlong sequence. */ 70364287048SMartin Matuska break; 70464287048SMartin Matuska default: 70564287048SMartin Matuska return (-1); 70664287048SMartin Matuska } 70764287048SMartin Matuska 70864287048SMartin Matuska /* The code point larger than 0x10FFFF is not legal 70964287048SMartin Matuska * Unicode values. */ 71064287048SMartin Matuska if (wc > 0x10FFFF) 71164287048SMartin Matuska return (-1); 71264287048SMartin Matuska /* Correctly gets a Unicode, returns used bytes. */ 71364287048SMartin Matuska *pwc = wc; 71464287048SMartin Matuska return (cnt); 71564287048SMartin Matuska } 71664287048SMartin Matuska 71764287048SMartin Matuska static void strdump(const char *e, const char *p, int ewidth, int utf8) 71864287048SMartin Matuska { 71964287048SMartin Matuska const char *q = p; 72064287048SMartin Matuska 72164287048SMartin Matuska logprintf(" %*s = ", ewidth, e); 72264287048SMartin Matuska if (p == NULL) { 72364287048SMartin Matuska logprintf("NULL\n"); 72464287048SMartin Matuska return; 72564287048SMartin Matuska } 72664287048SMartin Matuska logprintf("\""); 72764287048SMartin Matuska while (*p != '\0') { 72864287048SMartin Matuska unsigned int c = 0xff & *p++; 72964287048SMartin Matuska switch (c) { 73064287048SMartin Matuska case '\a': logprintf("\\a"); break; 73164287048SMartin Matuska case '\b': logprintf("\\b"); break; 73264287048SMartin Matuska case '\n': logprintf("\\n"); break; 73364287048SMartin Matuska case '\r': logprintf("\\r"); break; 73464287048SMartin Matuska default: 73564287048SMartin Matuska if (c >= 32 && c < 127) 73664287048SMartin Matuska logprintf("%c", c); 73764287048SMartin Matuska else 73864287048SMartin Matuska logprintf("\\x%02X", c); 73964287048SMartin Matuska } 74064287048SMartin Matuska } 74164287048SMartin Matuska logprintf("\""); 74264287048SMartin Matuska logprintf(" (length %d)", q == NULL ? -1 : (int)strlen(q)); 74364287048SMartin Matuska 74464287048SMartin Matuska /* 74564287048SMartin Matuska * If the current string is UTF-8, dump its code points. 74664287048SMartin Matuska */ 74764287048SMartin Matuska if (utf8) { 74864287048SMartin Matuska size_t len; 74964287048SMartin Matuska uint32_t uc; 75064287048SMartin Matuska int n; 75164287048SMartin Matuska int cnt = 0; 75264287048SMartin Matuska 75364287048SMartin Matuska p = q; 75464287048SMartin Matuska len = strlen(p); 75564287048SMartin Matuska logprintf(" ["); 75664287048SMartin Matuska while ((n = _utf8_to_unicode(&uc, p, len)) > 0) { 75764287048SMartin Matuska if (p != q) 75864287048SMartin Matuska logprintf(" "); 75964287048SMartin Matuska logprintf("%04X", uc); 76064287048SMartin Matuska p += n; 76164287048SMartin Matuska len -= n; 76264287048SMartin Matuska cnt++; 76364287048SMartin Matuska } 76464287048SMartin Matuska logprintf("]"); 76564287048SMartin Matuska logprintf(" (count %d", cnt); 76664287048SMartin Matuska if (n < 0) { 767f9762417SMartin Matuska logprintf(",unknown %zu bytes", len); 76864287048SMartin Matuska } 76964287048SMartin Matuska logprintf(")"); 77064287048SMartin Matuska 77164287048SMartin Matuska } 77264287048SMartin Matuska logprintf("\n"); 77364287048SMartin Matuska } 77464287048SMartin Matuska 77564287048SMartin Matuska /* Verify two strings are equal, dump them if not. */ 77664287048SMartin Matuska int 77764287048SMartin Matuska assertion_equal_string(const char *file, int line, 77864287048SMartin Matuska const char *v1, const char *e1, 77964287048SMartin Matuska const char *v2, const char *e2, 78064287048SMartin Matuska void *extra, int utf8) 78164287048SMartin Matuska { 78264287048SMartin Matuska int l1, l2; 78364287048SMartin Matuska 78464287048SMartin Matuska assertion_count(file, line); 78564287048SMartin Matuska if (v1 == v2 || (v1 != NULL && v2 != NULL && strcmp(v1, v2) == 0)) 78664287048SMartin Matuska return (1); 78764287048SMartin Matuska failure_start(file, line, "%s != %s", e1, e2); 78864287048SMartin Matuska l1 = (int)strlen(e1); 78964287048SMartin Matuska l2 = (int)strlen(e2); 79064287048SMartin Matuska if (l1 < l2) 79164287048SMartin Matuska l1 = l2; 79264287048SMartin Matuska strdump(e1, v1, l1, utf8); 79364287048SMartin Matuska strdump(e2, v2, l1, utf8); 79464287048SMartin Matuska failure_finish(extra); 79564287048SMartin Matuska return (0); 79664287048SMartin Matuska } 79764287048SMartin Matuska 79864287048SMartin Matuska static void 79964287048SMartin Matuska wcsdump(const char *e, const wchar_t *w) 80064287048SMartin Matuska { 80164287048SMartin Matuska logprintf(" %s = ", e); 80264287048SMartin Matuska if (w == NULL) { 80364287048SMartin Matuska logprintf("(null)"); 80464287048SMartin Matuska return; 80564287048SMartin Matuska } 80664287048SMartin Matuska logprintf("\""); 80764287048SMartin Matuska while (*w != L'\0') { 80864287048SMartin Matuska unsigned int c = *w++; 80964287048SMartin Matuska if (c >= 32 && c < 127) 81064287048SMartin Matuska logprintf("%c", c); 81164287048SMartin Matuska else if (c < 256) 81264287048SMartin Matuska logprintf("\\x%02X", c); 81364287048SMartin Matuska else if (c < 0x10000) 81464287048SMartin Matuska logprintf("\\u%04X", c); 81564287048SMartin Matuska else 81664287048SMartin Matuska logprintf("\\U%08X", c); 81764287048SMartin Matuska } 81864287048SMartin Matuska logprintf("\"\n"); 81964287048SMartin Matuska } 82064287048SMartin Matuska 82164287048SMartin Matuska #ifndef HAVE_WCSCMP 82264287048SMartin Matuska static int 82364287048SMartin Matuska wcscmp(const wchar_t *s1, const wchar_t *s2) 82464287048SMartin Matuska { 82564287048SMartin Matuska 82664287048SMartin Matuska while (*s1 == *s2++) { 82764287048SMartin Matuska if (*s1++ == L'\0') 82864287048SMartin Matuska return 0; 82964287048SMartin Matuska } 83064287048SMartin Matuska if (*s1 > *--s2) 83164287048SMartin Matuska return 1; 83264287048SMartin Matuska else 83364287048SMartin Matuska return -1; 83464287048SMartin Matuska } 83564287048SMartin Matuska #endif 83664287048SMartin Matuska 83764287048SMartin Matuska /* Verify that two wide strings are equal, dump them if not. */ 83864287048SMartin Matuska int 83964287048SMartin Matuska assertion_equal_wstring(const char *file, int line, 84064287048SMartin Matuska const wchar_t *v1, const char *e1, 84164287048SMartin Matuska const wchar_t *v2, const char *e2, 84264287048SMartin Matuska void *extra) 84364287048SMartin Matuska { 84464287048SMartin Matuska assertion_count(file, line); 84564287048SMartin Matuska if (v1 == v2) 84664287048SMartin Matuska return (1); 84764287048SMartin Matuska if (v1 != NULL && v2 != NULL && wcscmp(v1, v2) == 0) 84864287048SMartin Matuska return (1); 84964287048SMartin Matuska failure_start(file, line, "%s != %s", e1, e2); 85064287048SMartin Matuska wcsdump(e1, v1); 85164287048SMartin Matuska wcsdump(e2, v2); 85264287048SMartin Matuska failure_finish(extra); 85364287048SMartin Matuska return (0); 85464287048SMartin Matuska } 85564287048SMartin Matuska 85664287048SMartin Matuska /* 85764287048SMartin Matuska * Pretty standard hexdump routine. As a bonus, if ref != NULL, then 85864287048SMartin Matuska * any bytes in p that differ from ref will be highlighted with '_' 85964287048SMartin Matuska * before and after the hex value. 86064287048SMartin Matuska */ 86164287048SMartin Matuska static void 86264287048SMartin Matuska hexdump(const char *p, const char *ref, size_t l, size_t offset) 86364287048SMartin Matuska { 86464287048SMartin Matuska size_t i, j; 86564287048SMartin Matuska char sep; 86664287048SMartin Matuska 86764287048SMartin Matuska if (p == NULL) { 86864287048SMartin Matuska logprintf("(null)\n"); 86964287048SMartin Matuska return; 87064287048SMartin Matuska } 87164287048SMartin Matuska for(i=0; i < l; i+=16) { 87264287048SMartin Matuska logprintf("%04x", (unsigned)(i + offset)); 87364287048SMartin Matuska sep = ' '; 87464287048SMartin Matuska for (j = 0; j < 16 && i + j < l; j++) { 87564287048SMartin Matuska if (ref != NULL && p[i + j] != ref[i + j]) 87664287048SMartin Matuska sep = '_'; 87764287048SMartin Matuska logprintf("%c%02x", sep, 0xff & (int)p[i+j]); 87864287048SMartin Matuska if (ref != NULL && p[i + j] == ref[i + j]) 87964287048SMartin Matuska sep = ' '; 88064287048SMartin Matuska } 88164287048SMartin Matuska for (; j < 16; j++) { 88264287048SMartin Matuska logprintf("%c ", sep); 88364287048SMartin Matuska sep = ' '; 88464287048SMartin Matuska } 88564287048SMartin Matuska logprintf("%c", sep); 88664287048SMartin Matuska for (j=0; j < 16 && i + j < l; j++) { 88764287048SMartin Matuska int c = p[i + j]; 88864287048SMartin Matuska if (c >= ' ' && c <= 126) 88964287048SMartin Matuska logprintf("%c", c); 89064287048SMartin Matuska else 89164287048SMartin Matuska logprintf("."); 89264287048SMartin Matuska } 89364287048SMartin Matuska logprintf("\n"); 89464287048SMartin Matuska } 89564287048SMartin Matuska } 89664287048SMartin Matuska 89764287048SMartin Matuska /* Verify that two blocks of memory are the same, display the first 89864287048SMartin Matuska * block of differences if they're not. */ 89964287048SMartin Matuska int 90064287048SMartin Matuska assertion_equal_mem(const char *file, int line, 90164287048SMartin Matuska const void *_v1, const char *e1, 90264287048SMartin Matuska const void *_v2, const char *e2, 90364287048SMartin Matuska size_t l, const char *ld, void *extra) 90464287048SMartin Matuska { 90564287048SMartin Matuska const char *v1 = (const char *)_v1; 90664287048SMartin Matuska const char *v2 = (const char *)_v2; 90764287048SMartin Matuska size_t offset; 90864287048SMartin Matuska 90964287048SMartin Matuska assertion_count(file, line); 91064287048SMartin Matuska if (v1 == v2 || (v1 != NULL && v2 != NULL && memcmp(v1, v2, l) == 0)) 91164287048SMartin Matuska return (1); 91264287048SMartin Matuska if (v1 == NULL || v2 == NULL) 91364287048SMartin Matuska return (0); 91464287048SMartin Matuska 91564287048SMartin Matuska failure_start(file, line, "%s != %s", e1, e2); 91664287048SMartin Matuska logprintf(" size %s = %d\n", ld, (int)l); 91764287048SMartin Matuska /* Dump 48 bytes (3 lines) so that the first difference is 91864287048SMartin Matuska * in the second line. */ 91964287048SMartin Matuska offset = 0; 92064287048SMartin Matuska while (l > 64 && memcmp(v1, v2, 32) == 0) { 92164287048SMartin Matuska /* Two lines agree, so step forward one line. */ 92264287048SMartin Matuska v1 += 16; 92364287048SMartin Matuska v2 += 16; 92464287048SMartin Matuska l -= 16; 92564287048SMartin Matuska offset += 16; 92664287048SMartin Matuska } 92764287048SMartin Matuska logprintf(" Dump of %s\n", e1); 92864287048SMartin Matuska hexdump(v1, v2, l < 128 ? l : 128, offset); 92964287048SMartin Matuska logprintf(" Dump of %s\n", e2); 93064287048SMartin Matuska hexdump(v2, v1, l < 128 ? l : 128, offset); 93164287048SMartin Matuska logprintf("\n"); 93264287048SMartin Matuska failure_finish(extra); 93364287048SMartin Matuska return (0); 93464287048SMartin Matuska } 93564287048SMartin Matuska 93664287048SMartin Matuska /* Verify that a block of memory is filled with the specified byte. */ 93764287048SMartin Matuska int 93864287048SMartin Matuska assertion_memory_filled_with(const char *file, int line, 93964287048SMartin Matuska const void *_v1, const char *vd, 94064287048SMartin Matuska size_t l, const char *ld, 94164287048SMartin Matuska char b, const char *bd, void *extra) 94264287048SMartin Matuska { 94364287048SMartin Matuska const char *v1 = (const char *)_v1; 94464287048SMartin Matuska size_t c = 0; 94564287048SMartin Matuska size_t i; 94664287048SMartin Matuska (void)ld; /* UNUSED */ 94764287048SMartin Matuska 94864287048SMartin Matuska assertion_count(file, line); 94964287048SMartin Matuska 95064287048SMartin Matuska for (i = 0; i < l; ++i) { 95164287048SMartin Matuska if (v1[i] == b) { 95264287048SMartin Matuska ++c; 95364287048SMartin Matuska } 95464287048SMartin Matuska } 95564287048SMartin Matuska if (c == l) 95664287048SMartin Matuska return (1); 95764287048SMartin Matuska 95864287048SMartin Matuska failure_start(file, line, "%s (size %d) not filled with %s", vd, (int)l, bd); 95964287048SMartin Matuska logprintf(" Only %d bytes were correct\n", (int)c); 96064287048SMartin Matuska failure_finish(extra); 96164287048SMartin Matuska return (0); 96264287048SMartin Matuska } 96364287048SMartin Matuska 96464287048SMartin Matuska /* Verify that the named file exists and is empty. */ 96564287048SMartin Matuska int 96664287048SMartin Matuska assertion_empty_file(const char *filename, int line, const char *f1) 96764287048SMartin Matuska { 96864287048SMartin Matuska char buff[1024]; 96964287048SMartin Matuska struct stat st; 97064287048SMartin Matuska ssize_t s; 97164287048SMartin Matuska FILE *f; 97264287048SMartin Matuska 97364287048SMartin Matuska assertion_count(filename, line); 97464287048SMartin Matuska 97564287048SMartin Matuska if (stat(f1, &st) != 0) { 97664287048SMartin Matuska failure_start(filename, line, "Stat failed: %s", f1); 97764287048SMartin Matuska failure_finish(NULL); 97864287048SMartin Matuska return (0); 97964287048SMartin Matuska } 98064287048SMartin Matuska if (st.st_size == 0) 98164287048SMartin Matuska return (1); 98264287048SMartin Matuska 98364287048SMartin Matuska failure_start(filename, line, "File should be empty: %s", f1); 98464287048SMartin Matuska logprintf(" File size: %d\n", (int)st.st_size); 98564287048SMartin Matuska logprintf(" Contents:\n"); 98664287048SMartin Matuska f = fopen(f1, "rb"); 98764287048SMartin Matuska if (f == NULL) { 98864287048SMartin Matuska logprintf(" Unable to open %s\n", f1); 98964287048SMartin Matuska } else { 99064287048SMartin Matuska s = ((off_t)sizeof(buff) < st.st_size) ? 99164287048SMartin Matuska (ssize_t)sizeof(buff) : (ssize_t)st.st_size; 99264287048SMartin Matuska s = fread(buff, 1, s, f); 99364287048SMartin Matuska hexdump(buff, NULL, s, 0); 99464287048SMartin Matuska fclose(f); 99564287048SMartin Matuska } 99664287048SMartin Matuska failure_finish(NULL); 99764287048SMartin Matuska return (0); 99864287048SMartin Matuska } 99964287048SMartin Matuska 100064287048SMartin Matuska /* Verify that the named file exists and is not empty. */ 100164287048SMartin Matuska int 100264287048SMartin Matuska assertion_non_empty_file(const char *filename, int line, const char *f1) 100364287048SMartin Matuska { 100464287048SMartin Matuska struct stat st; 100564287048SMartin Matuska 100664287048SMartin Matuska assertion_count(filename, line); 100764287048SMartin Matuska 100864287048SMartin Matuska if (stat(f1, &st) != 0) { 100964287048SMartin Matuska failure_start(filename, line, "Stat failed: %s", f1); 101064287048SMartin Matuska failure_finish(NULL); 101164287048SMartin Matuska return (0); 101264287048SMartin Matuska } 101364287048SMartin Matuska if (st.st_size == 0) { 101464287048SMartin Matuska failure_start(filename, line, "File empty: %s", f1); 101564287048SMartin Matuska failure_finish(NULL); 101664287048SMartin Matuska return (0); 101764287048SMartin Matuska } 101864287048SMartin Matuska return (1); 101964287048SMartin Matuska } 102064287048SMartin Matuska 102164287048SMartin Matuska /* Verify that two files have the same contents. */ 102264287048SMartin Matuska /* TODO: hexdump the first bytes that actually differ. */ 102364287048SMartin Matuska int 102464287048SMartin Matuska assertion_equal_file(const char *filename, int line, const char *fn1, const char *fn2) 102564287048SMartin Matuska { 102664287048SMartin Matuska char buff1[1024]; 102764287048SMartin Matuska char buff2[1024]; 102864287048SMartin Matuska FILE *f1, *f2; 102964287048SMartin Matuska int n1, n2; 103064287048SMartin Matuska 103164287048SMartin Matuska assertion_count(filename, line); 103264287048SMartin Matuska 103364287048SMartin Matuska f1 = fopen(fn1, "rb"); 103464287048SMartin Matuska f2 = fopen(fn2, "rb"); 103564287048SMartin Matuska if (f1 == NULL || f2 == NULL) { 103664287048SMartin Matuska if (f1) fclose(f1); 103764287048SMartin Matuska if (f2) fclose(f2); 103864287048SMartin Matuska return (0); 103964287048SMartin Matuska } 104064287048SMartin Matuska for (;;) { 104164287048SMartin Matuska n1 = (int)fread(buff1, 1, sizeof(buff1), f1); 104264287048SMartin Matuska n2 = (int)fread(buff2, 1, sizeof(buff2), f2); 104364287048SMartin Matuska if (n1 != n2) 104464287048SMartin Matuska break; 104564287048SMartin Matuska if (n1 == 0 && n2 == 0) { 104664287048SMartin Matuska fclose(f1); 104764287048SMartin Matuska fclose(f2); 104864287048SMartin Matuska return (1); 104964287048SMartin Matuska } 105064287048SMartin Matuska if (memcmp(buff1, buff2, n1) != 0) 105164287048SMartin Matuska break; 105264287048SMartin Matuska } 105364287048SMartin Matuska fclose(f1); 105464287048SMartin Matuska fclose(f2); 105564287048SMartin Matuska failure_start(filename, line, "Files not identical"); 105664287048SMartin Matuska logprintf(" file1=\"%s\"\n", fn1); 105764287048SMartin Matuska logprintf(" file2=\"%s\"\n", fn2); 105864287048SMartin Matuska failure_finish(NULL); 105964287048SMartin Matuska return (0); 106064287048SMartin Matuska } 106164287048SMartin Matuska 106264287048SMartin Matuska /* Verify that the named file does exist. */ 106364287048SMartin Matuska int 106464287048SMartin Matuska assertion_file_exists(const char *filename, int line, const char *f) 106564287048SMartin Matuska { 106664287048SMartin Matuska assertion_count(filename, line); 106764287048SMartin Matuska 106864287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 106964287048SMartin Matuska if (!_access(f, 0)) 107064287048SMartin Matuska return (1); 107164287048SMartin Matuska #else 107264287048SMartin Matuska if (!access(f, F_OK)) 107364287048SMartin Matuska return (1); 107464287048SMartin Matuska #endif 107564287048SMartin Matuska failure_start(filename, line, "File should exist: %s", f); 107664287048SMartin Matuska failure_finish(NULL); 107764287048SMartin Matuska return (0); 107864287048SMartin Matuska } 107964287048SMartin Matuska 108064287048SMartin Matuska /* Verify that the named file doesn't exist. */ 108164287048SMartin Matuska int 108264287048SMartin Matuska assertion_file_not_exists(const char *filename, int line, const char *f) 108364287048SMartin Matuska { 108464287048SMartin Matuska assertion_count(filename, line); 108564287048SMartin Matuska 108664287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 108764287048SMartin Matuska if (_access(f, 0)) 108864287048SMartin Matuska return (1); 108964287048SMartin Matuska #else 109064287048SMartin Matuska if (access(f, F_OK)) 109164287048SMartin Matuska return (1); 109264287048SMartin Matuska #endif 109364287048SMartin Matuska failure_start(filename, line, "File should not exist: %s", f); 109464287048SMartin Matuska failure_finish(NULL); 109564287048SMartin Matuska return (0); 109664287048SMartin Matuska } 109764287048SMartin Matuska 109864287048SMartin Matuska /* Compare the contents of a file to a block of memory. */ 109964287048SMartin Matuska int 110064287048SMartin Matuska assertion_file_contents(const char *filename, int line, const void *buff, int s, const char *fn) 110164287048SMartin Matuska { 110264287048SMartin Matuska char *contents; 110364287048SMartin Matuska FILE *f; 110464287048SMartin Matuska int n; 110564287048SMartin Matuska 110664287048SMartin Matuska assertion_count(filename, line); 110764287048SMartin Matuska 110864287048SMartin Matuska f = fopen(fn, "rb"); 110964287048SMartin Matuska if (f == NULL) { 111064287048SMartin Matuska failure_start(filename, line, 111164287048SMartin Matuska "File should exist: %s", fn); 111264287048SMartin Matuska failure_finish(NULL); 111364287048SMartin Matuska return (0); 111464287048SMartin Matuska } 111564287048SMartin Matuska contents = malloc(s * 2); 111664287048SMartin Matuska n = (int)fread(contents, 1, s * 2, f); 111764287048SMartin Matuska fclose(f); 111864287048SMartin Matuska if (n == s && memcmp(buff, contents, s) == 0) { 111964287048SMartin Matuska free(contents); 112064287048SMartin Matuska return (1); 112164287048SMartin Matuska } 112264287048SMartin Matuska failure_start(filename, line, "File contents don't match"); 112364287048SMartin Matuska logprintf(" file=\"%s\"\n", fn); 112464287048SMartin Matuska if (n > 0) 112564287048SMartin Matuska hexdump(contents, buff, n > 512 ? 512 : n, 0); 112664287048SMartin Matuska else { 112764287048SMartin Matuska logprintf(" File empty, contents should be:\n"); 112864287048SMartin Matuska hexdump(buff, NULL, s > 512 ? 512 : s, 0); 112964287048SMartin Matuska } 113064287048SMartin Matuska failure_finish(NULL); 113164287048SMartin Matuska free(contents); 113264287048SMartin Matuska return (0); 113364287048SMartin Matuska } 113464287048SMartin Matuska 113564287048SMartin Matuska /* Check the contents of a text file, being tolerant of line endings. */ 113664287048SMartin Matuska int 113764287048SMartin Matuska assertion_text_file_contents(const char *filename, int line, const char *buff, const char *fn) 113864287048SMartin Matuska { 113964287048SMartin Matuska char *contents; 114064287048SMartin Matuska const char *btxt, *ftxt; 114164287048SMartin Matuska FILE *f; 114264287048SMartin Matuska int n, s; 114364287048SMartin Matuska 114464287048SMartin Matuska assertion_count(filename, line); 114564287048SMartin Matuska f = fopen(fn, "r"); 114664287048SMartin Matuska if (f == NULL) { 114764287048SMartin Matuska failure_start(filename, line, 114864287048SMartin Matuska "File doesn't exist: %s", fn); 114964287048SMartin Matuska failure_finish(NULL); 115064287048SMartin Matuska return (0); 115164287048SMartin Matuska } 115264287048SMartin Matuska s = (int)strlen(buff); 115364287048SMartin Matuska contents = malloc(s * 2 + 128); 115464287048SMartin Matuska n = (int)fread(contents, 1, s * 2 + 128 - 1, f); 115564287048SMartin Matuska if (n >= 0) 115664287048SMartin Matuska contents[n] = '\0'; 115764287048SMartin Matuska fclose(f); 115864287048SMartin Matuska /* Compare texts. */ 115964287048SMartin Matuska btxt = buff; 116064287048SMartin Matuska ftxt = (const char *)contents; 116164287048SMartin Matuska while (*btxt != '\0' && *ftxt != '\0') { 116264287048SMartin Matuska if (*btxt == *ftxt) { 116364287048SMartin Matuska ++btxt; 116464287048SMartin Matuska ++ftxt; 116564287048SMartin Matuska continue; 116664287048SMartin Matuska } 116764287048SMartin Matuska if (btxt[0] == '\n' && ftxt[0] == '\r' && ftxt[1] == '\n') { 116864287048SMartin Matuska /* Pass over different new line characters. */ 116964287048SMartin Matuska ++btxt; 117064287048SMartin Matuska ftxt += 2; 117164287048SMartin Matuska continue; 117264287048SMartin Matuska } 117364287048SMartin Matuska break; 117464287048SMartin Matuska } 117564287048SMartin Matuska if (*btxt == '\0' && *ftxt == '\0') { 117664287048SMartin Matuska free(contents); 117764287048SMartin Matuska return (1); 117864287048SMartin Matuska } 117964287048SMartin Matuska failure_start(filename, line, "Contents don't match"); 118064287048SMartin Matuska logprintf(" file=\"%s\"\n", fn); 118164287048SMartin Matuska if (n > 0) { 118264287048SMartin Matuska hexdump(contents, buff, n, 0); 1183f9762417SMartin Matuska logprintf(" expected\n"); 118464287048SMartin Matuska hexdump(buff, contents, s, 0); 118564287048SMartin Matuska } else { 118664287048SMartin Matuska logprintf(" File empty, contents should be:\n"); 118764287048SMartin Matuska hexdump(buff, NULL, s, 0); 118864287048SMartin Matuska } 118964287048SMartin Matuska failure_finish(NULL); 119064287048SMartin Matuska free(contents); 119164287048SMartin Matuska return (0); 119264287048SMartin Matuska } 119364287048SMartin Matuska 119464287048SMartin Matuska /* Verify that a text file contains the specified lines, regardless of order */ 119564287048SMartin Matuska /* This could be more efficient if we sorted both sets of lines, etc, but 119664287048SMartin Matuska * since this is used only for testing and only ever deals with a dozen or so 119764287048SMartin Matuska * lines at a time, this relatively crude approach is just fine. */ 119864287048SMartin Matuska int 119964287048SMartin Matuska assertion_file_contains_lines_any_order(const char *file, int line, 120064287048SMartin Matuska const char *pathname, const char *lines[]) 120164287048SMartin Matuska { 120264287048SMartin Matuska char *buff; 120364287048SMartin Matuska size_t buff_size; 120464287048SMartin Matuska size_t expected_count, actual_count, i, j; 120564287048SMartin Matuska char **expected = NULL; 120664287048SMartin Matuska char *p, **actual = NULL; 120764287048SMartin Matuska char c; 120864287048SMartin Matuska int expected_failure = 0, actual_failure = 0; 120964287048SMartin Matuska 121064287048SMartin Matuska assertion_count(file, line); 121164287048SMartin Matuska 121264287048SMartin Matuska buff = slurpfile(&buff_size, "%s", pathname); 121364287048SMartin Matuska if (buff == NULL) { 121464287048SMartin Matuska failure_start(pathname, line, "Can't read file: %s", pathname); 121564287048SMartin Matuska failure_finish(NULL); 121664287048SMartin Matuska return (0); 121764287048SMartin Matuska } 121864287048SMartin Matuska 121964287048SMartin Matuska /* Make a copy of the provided lines and count up the expected 122064287048SMartin Matuska * file size. */ 122164287048SMartin Matuska for (i = 0; lines[i] != NULL; ++i) { 122264287048SMartin Matuska } 122364287048SMartin Matuska expected_count = i; 122464287048SMartin Matuska if (expected_count) { 122564287048SMartin Matuska expected = malloc(sizeof(char *) * expected_count); 122664287048SMartin Matuska if (expected == NULL) { 122764287048SMartin Matuska failure_start(pathname, line, "Can't allocate memory"); 122864287048SMartin Matuska failure_finish(NULL); 122964287048SMartin Matuska free(expected); 1230a8fc61d5SMartin Matuska free(buff); 123164287048SMartin Matuska return (0); 123264287048SMartin Matuska } 123364287048SMartin Matuska for (i = 0; lines[i] != NULL; ++i) { 123464287048SMartin Matuska expected[i] = strdup(lines[i]); 123564287048SMartin Matuska } 123664287048SMartin Matuska } 123764287048SMartin Matuska 123864287048SMartin Matuska /* Break the file into lines */ 123964287048SMartin Matuska actual_count = 0; 124064287048SMartin Matuska for (c = '\0', p = buff; p < buff + buff_size; ++p) { 124164287048SMartin Matuska if (*p == '\x0d' || *p == '\x0a') 124264287048SMartin Matuska *p = '\0'; 124364287048SMartin Matuska if (c == '\0' && *p != '\0') 124464287048SMartin Matuska ++actual_count; 124564287048SMartin Matuska c = *p; 124664287048SMartin Matuska } 124764287048SMartin Matuska if (actual_count) { 124864287048SMartin Matuska actual = calloc(sizeof(char *), actual_count); 124964287048SMartin Matuska if (actual == NULL) { 125064287048SMartin Matuska failure_start(pathname, line, "Can't allocate memory"); 125164287048SMartin Matuska failure_finish(NULL); 125264287048SMartin Matuska free(expected); 1253a8fc61d5SMartin Matuska free(buff); 125464287048SMartin Matuska return (0); 125564287048SMartin Matuska } 125664287048SMartin Matuska for (j = 0, p = buff; p < buff + buff_size; 125764287048SMartin Matuska p += 1 + strlen(p)) { 125864287048SMartin Matuska if (*p != '\0') { 125964287048SMartin Matuska actual[j] = p; 126064287048SMartin Matuska ++j; 126164287048SMartin Matuska } 126264287048SMartin Matuska } 126364287048SMartin Matuska } 126464287048SMartin Matuska 126564287048SMartin Matuska /* Erase matching lines from both lists */ 126664287048SMartin Matuska for (i = 0; i < expected_count; ++i) { 126764287048SMartin Matuska if (expected[i] == NULL) 126864287048SMartin Matuska continue; 126964287048SMartin Matuska for (j = 0; j < actual_count; ++j) { 127064287048SMartin Matuska if (actual[j] == NULL) 127164287048SMartin Matuska continue; 127264287048SMartin Matuska if (strcmp(expected[i], actual[j]) == 0) { 127364287048SMartin Matuska free(expected[i]); 127464287048SMartin Matuska expected[i] = NULL; 127564287048SMartin Matuska actual[j] = NULL; 127664287048SMartin Matuska break; 127764287048SMartin Matuska } 127864287048SMartin Matuska } 127964287048SMartin Matuska } 128064287048SMartin Matuska 128164287048SMartin Matuska /* If there's anything left, it's a failure */ 128264287048SMartin Matuska for (i = 0; i < expected_count; ++i) { 128364287048SMartin Matuska if (expected[i] != NULL) 128464287048SMartin Matuska ++expected_failure; 128564287048SMartin Matuska } 128664287048SMartin Matuska for (j = 0; j < actual_count; ++j) { 128764287048SMartin Matuska if (actual[j] != NULL) 128864287048SMartin Matuska ++actual_failure; 128964287048SMartin Matuska } 129064287048SMartin Matuska if (expected_failure == 0 && actual_failure == 0) { 129164287048SMartin Matuska free(buff); 129264287048SMartin Matuska free(expected); 129364287048SMartin Matuska free(actual); 129464287048SMartin Matuska return (1); 129564287048SMartin Matuska } 129664287048SMartin Matuska failure_start(file, line, "File doesn't match: %s", pathname); 129764287048SMartin Matuska for (i = 0; i < expected_count; ++i) { 129864287048SMartin Matuska if (expected[i] != NULL) { 129964287048SMartin Matuska logprintf(" Expected but not present: %s\n", expected[i]); 130064287048SMartin Matuska free(expected[i]); 130164287048SMartin Matuska } 130264287048SMartin Matuska } 130364287048SMartin Matuska for (j = 0; j < actual_count; ++j) { 130464287048SMartin Matuska if (actual[j] != NULL) 130564287048SMartin Matuska logprintf(" Present but not expected: %s\n", actual[j]); 130664287048SMartin Matuska } 130764287048SMartin Matuska failure_finish(NULL); 130864287048SMartin Matuska free(buff); 130964287048SMartin Matuska free(expected); 131064287048SMartin Matuska free(actual); 131164287048SMartin Matuska return (0); 131264287048SMartin Matuska } 131364287048SMartin Matuska 131464287048SMartin Matuska /* Verify that a text file does not contains the specified strings */ 131564287048SMartin Matuska int 131664287048SMartin Matuska assertion_file_contains_no_invalid_strings(const char *file, int line, 131764287048SMartin Matuska const char *pathname, const char *strings[]) 131864287048SMartin Matuska { 131964287048SMartin Matuska char *buff; 132064287048SMartin Matuska int i; 132164287048SMartin Matuska 132264287048SMartin Matuska buff = slurpfile(NULL, "%s", pathname); 132364287048SMartin Matuska if (buff == NULL) { 132464287048SMartin Matuska failure_start(file, line, "Can't read file: %s", pathname); 132564287048SMartin Matuska failure_finish(NULL); 132664287048SMartin Matuska return (0); 132764287048SMartin Matuska } 132864287048SMartin Matuska 132964287048SMartin Matuska for (i = 0; strings[i] != NULL; ++i) { 133064287048SMartin Matuska if (strstr(buff, strings[i]) != NULL) { 133164287048SMartin Matuska failure_start(file, line, "Invalid string in %s: %s", pathname, 133264287048SMartin Matuska strings[i]); 133364287048SMartin Matuska failure_finish(NULL); 133464287048SMartin Matuska free(buff); 133564287048SMartin Matuska return(0); 133664287048SMartin Matuska } 133764287048SMartin Matuska } 133864287048SMartin Matuska 133964287048SMartin Matuska free(buff); 134064287048SMartin Matuska return (0); 134164287048SMartin Matuska } 134264287048SMartin Matuska 134364287048SMartin Matuska /* Test that two paths point to the same file. */ 134464287048SMartin Matuska /* As a side-effect, asserts that both files exist. */ 134564287048SMartin Matuska static int 134664287048SMartin Matuska is_hardlink(const char *file, int line, 134764287048SMartin Matuska const char *path1, const char *path2) 134864287048SMartin Matuska { 134964287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 135064287048SMartin Matuska BY_HANDLE_FILE_INFORMATION bhfi1, bhfi2; 135164287048SMartin Matuska int r; 135264287048SMartin Matuska 135364287048SMartin Matuska assertion_count(file, line); 135464287048SMartin Matuska r = my_GetFileInformationByName(path1, &bhfi1); 135564287048SMartin Matuska if (r == 0) { 135664287048SMartin Matuska failure_start(file, line, "File %s can't be inspected?", path1); 135764287048SMartin Matuska failure_finish(NULL); 135864287048SMartin Matuska return (0); 135964287048SMartin Matuska } 136064287048SMartin Matuska r = my_GetFileInformationByName(path2, &bhfi2); 136164287048SMartin Matuska if (r == 0) { 136264287048SMartin Matuska failure_start(file, line, "File %s can't be inspected?", path2); 136364287048SMartin Matuska failure_finish(NULL); 136464287048SMartin Matuska return (0); 136564287048SMartin Matuska } 136664287048SMartin Matuska return (bhfi1.dwVolumeSerialNumber == bhfi2.dwVolumeSerialNumber 136764287048SMartin Matuska && bhfi1.nFileIndexHigh == bhfi2.nFileIndexHigh 136864287048SMartin Matuska && bhfi1.nFileIndexLow == bhfi2.nFileIndexLow); 136964287048SMartin Matuska #else 137064287048SMartin Matuska struct stat st1, st2; 137164287048SMartin Matuska int r; 137264287048SMartin Matuska 137364287048SMartin Matuska assertion_count(file, line); 137464287048SMartin Matuska r = lstat(path1, &st1); 137564287048SMartin Matuska if (r != 0) { 137664287048SMartin Matuska failure_start(file, line, "File should exist: %s", path1); 137764287048SMartin Matuska failure_finish(NULL); 137864287048SMartin Matuska return (0); 137964287048SMartin Matuska } 138064287048SMartin Matuska r = lstat(path2, &st2); 138164287048SMartin Matuska if (r != 0) { 138264287048SMartin Matuska failure_start(file, line, "File should exist: %s", path2); 138364287048SMartin Matuska failure_finish(NULL); 138464287048SMartin Matuska return (0); 138564287048SMartin Matuska } 138664287048SMartin Matuska return (st1.st_ino == st2.st_ino && st1.st_dev == st2.st_dev); 138764287048SMartin Matuska #endif 138864287048SMartin Matuska } 138964287048SMartin Matuska 139064287048SMartin Matuska int 139164287048SMartin Matuska assertion_is_hardlink(const char *file, int line, 139264287048SMartin Matuska const char *path1, const char *path2) 139364287048SMartin Matuska { 139464287048SMartin Matuska if (is_hardlink(file, line, path1, path2)) 139564287048SMartin Matuska return (1); 139664287048SMartin Matuska failure_start(file, line, 139764287048SMartin Matuska "Files %s and %s are not hardlinked", path1, path2); 139864287048SMartin Matuska failure_finish(NULL); 139964287048SMartin Matuska return (0); 140064287048SMartin Matuska } 140164287048SMartin Matuska 140264287048SMartin Matuska int 140364287048SMartin Matuska assertion_is_not_hardlink(const char *file, int line, 140464287048SMartin Matuska const char *path1, const char *path2) 140564287048SMartin Matuska { 140664287048SMartin Matuska if (!is_hardlink(file, line, path1, path2)) 140764287048SMartin Matuska return (1); 140864287048SMartin Matuska failure_start(file, line, 140964287048SMartin Matuska "Files %s and %s should not be hardlinked", path1, path2); 141064287048SMartin Matuska failure_finish(NULL); 141164287048SMartin Matuska return (0); 141264287048SMartin Matuska } 141364287048SMartin Matuska 141464287048SMartin Matuska /* Verify a/b/mtime of 'pathname'. */ 141564287048SMartin Matuska /* If 'recent', verify that it's within last 10 seconds. */ 141664287048SMartin Matuska static int 141764287048SMartin Matuska assertion_file_time(const char *file, int line, 141864287048SMartin Matuska const char *pathname, long t, long nsec, char type, int recent) 141964287048SMartin Matuska { 142064287048SMartin Matuska long long filet, filet_nsec; 142164287048SMartin Matuska int r; 142264287048SMartin Matuska 142364287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 142464287048SMartin Matuska #define EPOC_TIME (116444736000000000ULL) 142564287048SMartin Matuska FILETIME fxtime, fbirthtime, fatime, fmtime; 142664287048SMartin Matuska ULARGE_INTEGER wintm; 142764287048SMartin Matuska HANDLE h; 142864287048SMartin Matuska fxtime.dwLowDateTime = 0; 142964287048SMartin Matuska fxtime.dwHighDateTime = 0; 143064287048SMartin Matuska 143164287048SMartin Matuska assertion_count(file, line); 143264287048SMartin Matuska /* Note: FILE_FLAG_BACKUP_SEMANTICS applies to open 143364287048SMartin Matuska * a directory file. If not, CreateFile() will fail when 143464287048SMartin Matuska * the pathname is a directory. */ 143564287048SMartin Matuska h = CreateFile(pathname, FILE_READ_ATTRIBUTES, 0, NULL, 143664287048SMartin Matuska OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL); 143764287048SMartin Matuska if (h == INVALID_HANDLE_VALUE) { 143864287048SMartin Matuska failure_start(file, line, "Can't access %s\n", pathname); 143964287048SMartin Matuska failure_finish(NULL); 144064287048SMartin Matuska return (0); 144164287048SMartin Matuska } 144264287048SMartin Matuska r = GetFileTime(h, &fbirthtime, &fatime, &fmtime); 144364287048SMartin Matuska switch (type) { 144464287048SMartin Matuska case 'a': fxtime = fatime; break; 144564287048SMartin Matuska case 'b': fxtime = fbirthtime; break; 144664287048SMartin Matuska case 'm': fxtime = fmtime; break; 144764287048SMartin Matuska } 144864287048SMartin Matuska CloseHandle(h); 144964287048SMartin Matuska if (r == 0) { 145064287048SMartin Matuska failure_start(file, line, "Can't GetFileTime %s\n", pathname); 145164287048SMartin Matuska failure_finish(NULL); 145264287048SMartin Matuska return (0); 145364287048SMartin Matuska } 145464287048SMartin Matuska wintm.LowPart = fxtime.dwLowDateTime; 145564287048SMartin Matuska wintm.HighPart = fxtime.dwHighDateTime; 145664287048SMartin Matuska filet = (wintm.QuadPart - EPOC_TIME) / 10000000; 145764287048SMartin Matuska filet_nsec = ((wintm.QuadPart - EPOC_TIME) % 10000000) * 100; 145864287048SMartin Matuska nsec = (nsec / 100) * 100; /* Round the request */ 145964287048SMartin Matuska #else 146064287048SMartin Matuska struct stat st; 146164287048SMartin Matuska 146264287048SMartin Matuska assertion_count(file, line); 146364287048SMartin Matuska r = lstat(pathname, &st); 146464287048SMartin Matuska if (r != 0) { 146564287048SMartin Matuska failure_start(file, line, "Can't stat %s\n", pathname); 146664287048SMartin Matuska failure_finish(NULL); 146764287048SMartin Matuska return (0); 146864287048SMartin Matuska } 146964287048SMartin Matuska switch (type) { 147064287048SMartin Matuska case 'a': filet = st.st_atime; break; 147164287048SMartin Matuska case 'm': filet = st.st_mtime; break; 147264287048SMartin Matuska case 'b': filet = 0; break; 147364287048SMartin Matuska default: fprintf(stderr, "INTERNAL: Bad type %c for file time", type); 147464287048SMartin Matuska exit(1); 147564287048SMartin Matuska } 147664287048SMartin Matuska #if defined(__FreeBSD__) 147764287048SMartin Matuska switch (type) { 147864287048SMartin Matuska case 'a': filet_nsec = st.st_atimespec.tv_nsec; break; 147964287048SMartin Matuska case 'b': filet = st.st_birthtime; 148064287048SMartin Matuska /* FreeBSD filesystems that don't support birthtime 148164287048SMartin Matuska * (e.g., UFS1) always return -1 here. */ 148264287048SMartin Matuska if (filet == -1) { 148364287048SMartin Matuska return (1); 148464287048SMartin Matuska } 148564287048SMartin Matuska filet_nsec = st.st_birthtimespec.tv_nsec; break; 148664287048SMartin Matuska case 'm': filet_nsec = st.st_mtimespec.tv_nsec; break; 148764287048SMartin Matuska default: fprintf(stderr, "INTERNAL: Bad type %c for file time", type); 148864287048SMartin Matuska exit(1); 148964287048SMartin Matuska } 149064287048SMartin Matuska /* FreeBSD generally only stores to microsecond res, so round. */ 149164287048SMartin Matuska filet_nsec = (filet_nsec / 1000) * 1000; 149264287048SMartin Matuska nsec = (nsec / 1000) * 1000; 149364287048SMartin Matuska #else 149464287048SMartin Matuska filet_nsec = nsec = 0; /* Generic POSIX only has whole seconds. */ 149564287048SMartin Matuska if (type == 'b') return (1); /* Generic POSIX doesn't have birthtime */ 149664287048SMartin Matuska #if defined(__HAIKU__) 149764287048SMartin Matuska if (type == 'a') return (1); /* Haiku doesn't have atime. */ 149864287048SMartin Matuska #endif 149964287048SMartin Matuska #endif 150064287048SMartin Matuska #endif 150164287048SMartin Matuska if (recent) { 150264287048SMartin Matuska /* Check that requested time is up-to-date. */ 150364287048SMartin Matuska time_t now = time(NULL); 150464287048SMartin Matuska if (filet < now - 10 || filet > now + 1) { 150564287048SMartin Matuska failure_start(file, line, 150664287048SMartin Matuska "File %s has %ctime %lld, %lld seconds ago\n", 150764287048SMartin Matuska pathname, type, filet, now - filet); 150864287048SMartin Matuska failure_finish(NULL); 150964287048SMartin Matuska return (0); 151064287048SMartin Matuska } 151164287048SMartin Matuska } else if (filet != t || filet_nsec != nsec) { 151264287048SMartin Matuska failure_start(file, line, 1513f9762417SMartin Matuska "File %s has %ctime %lld.%09lld, expected %ld.%09ld", 151464287048SMartin Matuska pathname, type, filet, filet_nsec, t, nsec); 151564287048SMartin Matuska failure_finish(NULL); 151664287048SMartin Matuska return (0); 151764287048SMartin Matuska } 151864287048SMartin Matuska return (1); 151964287048SMartin Matuska } 152064287048SMartin Matuska 152164287048SMartin Matuska /* Verify atime of 'pathname'. */ 152264287048SMartin Matuska int 152364287048SMartin Matuska assertion_file_atime(const char *file, int line, 152464287048SMartin Matuska const char *pathname, long t, long nsec) 152564287048SMartin Matuska { 152664287048SMartin Matuska return assertion_file_time(file, line, pathname, t, nsec, 'a', 0); 152764287048SMartin Matuska } 152864287048SMartin Matuska 152964287048SMartin Matuska /* Verify atime of 'pathname' is up-to-date. */ 153064287048SMartin Matuska int 153164287048SMartin Matuska assertion_file_atime_recent(const char *file, int line, const char *pathname) 153264287048SMartin Matuska { 153364287048SMartin Matuska return assertion_file_time(file, line, pathname, 0, 0, 'a', 1); 153464287048SMartin Matuska } 153564287048SMartin Matuska 153664287048SMartin Matuska /* Verify birthtime of 'pathname'. */ 153764287048SMartin Matuska int 153864287048SMartin Matuska assertion_file_birthtime(const char *file, int line, 153964287048SMartin Matuska const char *pathname, long t, long nsec) 154064287048SMartin Matuska { 154164287048SMartin Matuska return assertion_file_time(file, line, pathname, t, nsec, 'b', 0); 154264287048SMartin Matuska } 154364287048SMartin Matuska 154464287048SMartin Matuska /* Verify birthtime of 'pathname' is up-to-date. */ 154564287048SMartin Matuska int 154664287048SMartin Matuska assertion_file_birthtime_recent(const char *file, int line, 154764287048SMartin Matuska const char *pathname) 154864287048SMartin Matuska { 154964287048SMartin Matuska return assertion_file_time(file, line, pathname, 0, 0, 'b', 1); 155064287048SMartin Matuska } 155164287048SMartin Matuska 155264287048SMartin Matuska /* Verify mode of 'pathname'. */ 155364287048SMartin Matuska int 155464287048SMartin Matuska assertion_file_mode(const char *file, int line, const char *pathname, int expected_mode) 155564287048SMartin Matuska { 155664287048SMartin Matuska int mode; 155764287048SMartin Matuska int r; 155864287048SMartin Matuska 155964287048SMartin Matuska assertion_count(file, line); 156064287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 156164287048SMartin Matuska failure_start(file, line, "assertFileMode not yet implemented for Windows"); 156264287048SMartin Matuska (void)mode; /* UNUSED */ 156364287048SMartin Matuska (void)r; /* UNUSED */ 156464287048SMartin Matuska (void)pathname; /* UNUSED */ 156564287048SMartin Matuska (void)expected_mode; /* UNUSED */ 156664287048SMartin Matuska #else 156764287048SMartin Matuska { 156864287048SMartin Matuska struct stat st; 156964287048SMartin Matuska r = lstat(pathname, &st); 157064287048SMartin Matuska mode = (int)(st.st_mode & 0777); 157164287048SMartin Matuska } 157264287048SMartin Matuska if (r == 0 && mode == expected_mode) 157364287048SMartin Matuska return (1); 157464287048SMartin Matuska failure_start(file, line, "File %s has mode %o, expected %o", 157564287048SMartin Matuska pathname, mode, expected_mode); 157664287048SMartin Matuska #endif 157764287048SMartin Matuska failure_finish(NULL); 157864287048SMartin Matuska return (0); 157964287048SMartin Matuska } 158064287048SMartin Matuska 158164287048SMartin Matuska /* Verify mtime of 'pathname'. */ 158264287048SMartin Matuska int 158364287048SMartin Matuska assertion_file_mtime(const char *file, int line, 158464287048SMartin Matuska const char *pathname, long t, long nsec) 158564287048SMartin Matuska { 158664287048SMartin Matuska return assertion_file_time(file, line, pathname, t, nsec, 'm', 0); 158764287048SMartin Matuska } 158864287048SMartin Matuska 158964287048SMartin Matuska /* Verify mtime of 'pathname' is up-to-date. */ 159064287048SMartin Matuska int 159164287048SMartin Matuska assertion_file_mtime_recent(const char *file, int line, const char *pathname) 159264287048SMartin Matuska { 159364287048SMartin Matuska return assertion_file_time(file, line, pathname, 0, 0, 'm', 1); 159464287048SMartin Matuska } 159564287048SMartin Matuska 159664287048SMartin Matuska /* Verify number of links to 'pathname'. */ 159764287048SMartin Matuska int 159864287048SMartin Matuska assertion_file_nlinks(const char *file, int line, 159964287048SMartin Matuska const char *pathname, int nlinks) 160064287048SMartin Matuska { 160164287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 160264287048SMartin Matuska BY_HANDLE_FILE_INFORMATION bhfi; 160364287048SMartin Matuska int r; 160464287048SMartin Matuska 160564287048SMartin Matuska assertion_count(file, line); 160664287048SMartin Matuska r = my_GetFileInformationByName(pathname, &bhfi); 160764287048SMartin Matuska if (r != 0 && bhfi.nNumberOfLinks == (DWORD)nlinks) 160864287048SMartin Matuska return (1); 1609f9762417SMartin Matuska failure_start(file, line, "File %s has %jd links, expected %d", 1610f9762417SMartin Matuska pathname, (intmax_t)bhfi.nNumberOfLinks, nlinks); 161164287048SMartin Matuska failure_finish(NULL); 161264287048SMartin Matuska return (0); 161364287048SMartin Matuska #else 161464287048SMartin Matuska struct stat st; 161564287048SMartin Matuska int r; 161664287048SMartin Matuska 161764287048SMartin Matuska assertion_count(file, line); 161864287048SMartin Matuska r = lstat(pathname, &st); 161964287048SMartin Matuska if (r == 0 && (int)st.st_nlink == nlinks) 162064287048SMartin Matuska return (1); 1621f9762417SMartin Matuska failure_start(file, line, "File %s has %jd links, expected %d", 1622f9762417SMartin Matuska pathname, (intmax_t)st.st_nlink, nlinks); 162364287048SMartin Matuska failure_finish(NULL); 162464287048SMartin Matuska return (0); 162564287048SMartin Matuska #endif 162664287048SMartin Matuska } 162764287048SMartin Matuska 162864287048SMartin Matuska /* Verify size of 'pathname'. */ 162964287048SMartin Matuska int 163064287048SMartin Matuska assertion_file_size(const char *file, int line, const char *pathname, long size) 163164287048SMartin Matuska { 163264287048SMartin Matuska int64_t filesize; 163364287048SMartin Matuska int r; 163464287048SMartin Matuska 163564287048SMartin Matuska assertion_count(file, line); 163664287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 163764287048SMartin Matuska { 163864287048SMartin Matuska BY_HANDLE_FILE_INFORMATION bhfi; 163964287048SMartin Matuska r = !my_GetFileInformationByName(pathname, &bhfi); 164064287048SMartin Matuska filesize = ((int64_t)bhfi.nFileSizeHigh << 32) + bhfi.nFileSizeLow; 164164287048SMartin Matuska } 164264287048SMartin Matuska #else 164364287048SMartin Matuska { 164464287048SMartin Matuska struct stat st; 164564287048SMartin Matuska r = lstat(pathname, &st); 164664287048SMartin Matuska filesize = st.st_size; 164764287048SMartin Matuska } 164864287048SMartin Matuska #endif 164964287048SMartin Matuska if (r == 0 && filesize == size) 165064287048SMartin Matuska return (1); 165164287048SMartin Matuska failure_start(file, line, "File %s has size %ld, expected %ld", 165264287048SMartin Matuska pathname, (long)filesize, (long)size); 165364287048SMartin Matuska failure_finish(NULL); 165464287048SMartin Matuska return (0); 165564287048SMartin Matuska } 165664287048SMartin Matuska 165764287048SMartin Matuska /* Assert that 'pathname' is a dir. If mode >= 0, verify that too. */ 165864287048SMartin Matuska int 165964287048SMartin Matuska assertion_is_dir(const char *file, int line, const char *pathname, int mode) 166064287048SMartin Matuska { 166164287048SMartin Matuska struct stat st; 166264287048SMartin Matuska int r; 166364287048SMartin Matuska 166464287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 166564287048SMartin Matuska (void)mode; /* UNUSED */ 166664287048SMartin Matuska #endif 166764287048SMartin Matuska assertion_count(file, line); 166864287048SMartin Matuska r = lstat(pathname, &st); 166964287048SMartin Matuska if (r != 0) { 167064287048SMartin Matuska failure_start(file, line, "Dir should exist: %s", pathname); 167164287048SMartin Matuska failure_finish(NULL); 167264287048SMartin Matuska return (0); 167364287048SMartin Matuska } 167464287048SMartin Matuska if (!S_ISDIR(st.st_mode)) { 167564287048SMartin Matuska failure_start(file, line, "%s is not a dir", pathname); 167664287048SMartin Matuska failure_finish(NULL); 167764287048SMartin Matuska return (0); 167864287048SMartin Matuska } 167964287048SMartin Matuska #if !defined(_WIN32) || defined(__CYGWIN__) 168064287048SMartin Matuska /* Windows doesn't handle permissions the same way as POSIX, 168164287048SMartin Matuska * so just ignore the mode tests. */ 168264287048SMartin Matuska /* TODO: Can we do better here? */ 168364287048SMartin Matuska if (mode >= 0 && (mode_t)mode != (st.st_mode & 07777)) { 168464287048SMartin Matuska failure_start(file, line, "Dir %s has wrong mode", pathname); 168564287048SMartin Matuska logprintf(" Expected: 0%3o\n", mode); 168664287048SMartin Matuska logprintf(" Found: 0%3o\n", st.st_mode & 07777); 168764287048SMartin Matuska failure_finish(NULL); 168864287048SMartin Matuska return (0); 168964287048SMartin Matuska } 169064287048SMartin Matuska #endif 169164287048SMartin Matuska return (1); 169264287048SMartin Matuska } 169364287048SMartin Matuska 169464287048SMartin Matuska /* Verify that 'pathname' is a regular file. If 'mode' is >= 0, 169564287048SMartin Matuska * verify that too. */ 169664287048SMartin Matuska int 169764287048SMartin Matuska assertion_is_reg(const char *file, int line, const char *pathname, int mode) 169864287048SMartin Matuska { 169964287048SMartin Matuska struct stat st; 170064287048SMartin Matuska int r; 170164287048SMartin Matuska 170264287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 170364287048SMartin Matuska (void)mode; /* UNUSED */ 170464287048SMartin Matuska #endif 170564287048SMartin Matuska assertion_count(file, line); 170664287048SMartin Matuska r = lstat(pathname, &st); 170764287048SMartin Matuska if (r != 0 || !S_ISREG(st.st_mode)) { 170864287048SMartin Matuska failure_start(file, line, "File should exist: %s", pathname); 170964287048SMartin Matuska failure_finish(NULL); 171064287048SMartin Matuska return (0); 171164287048SMartin Matuska } 171264287048SMartin Matuska #if !defined(_WIN32) || defined(__CYGWIN__) 171364287048SMartin Matuska /* Windows doesn't handle permissions the same way as POSIX, 171464287048SMartin Matuska * so just ignore the mode tests. */ 171564287048SMartin Matuska /* TODO: Can we do better here? */ 171664287048SMartin Matuska if (mode >= 0 && (mode_t)mode != (st.st_mode & 07777)) { 171764287048SMartin Matuska failure_start(file, line, "File %s has wrong mode", pathname); 171864287048SMartin Matuska logprintf(" Expected: 0%3o\n", mode); 171964287048SMartin Matuska logprintf(" Found: 0%3o\n", st.st_mode & 07777); 172064287048SMartin Matuska failure_finish(NULL); 172164287048SMartin Matuska return (0); 172264287048SMartin Matuska } 172364287048SMartin Matuska #endif 172464287048SMartin Matuska return (1); 172564287048SMartin Matuska } 172664287048SMartin Matuska 172752c2bb75SMartin Matuska /* 172852c2bb75SMartin Matuska * Check whether 'pathname' is a symbolic link. If 'contents' is 172952c2bb75SMartin Matuska * non-NULL, verify that the symlink has those contents. 173052c2bb75SMartin Matuska * 173152c2bb75SMartin Matuska * On platforms with directory symlinks, set isdir to 0 to test for a file 173252c2bb75SMartin Matuska * symlink and to 1 to test for a directory symlink. On other platforms 173352c2bb75SMartin Matuska * the variable is ignored. 173452c2bb75SMartin Matuska */ 173564287048SMartin Matuska static int 173664287048SMartin Matuska is_symlink(const char *file, int line, 173752c2bb75SMartin Matuska const char *pathname, const char *contents, int isdir) 173864287048SMartin Matuska { 173964287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 174052c2bb75SMartin Matuska HANDLE h; 174152c2bb75SMartin Matuska DWORD inbytes; 174252c2bb75SMartin Matuska REPARSE_DATA_BUFFER *buf; 174352c2bb75SMartin Matuska BY_HANDLE_FILE_INFORMATION st; 174452c2bb75SMartin Matuska size_t len, len2; 174552c2bb75SMartin Matuska wchar_t *linknamew, *contentsw; 174652c2bb75SMartin Matuska const char *p; 174752c2bb75SMartin Matuska char *s, *pn; 174852c2bb75SMartin Matuska int ret = 0; 174952c2bb75SMartin Matuska BYTE *indata; 175052c2bb75SMartin Matuska const DWORD flag = FILE_FLAG_BACKUP_SEMANTICS | 175152c2bb75SMartin Matuska FILE_FLAG_OPEN_REPARSE_POINT; 175252c2bb75SMartin Matuska 175352c2bb75SMartin Matuska /* Replace slashes with backslashes in pathname */ 175452c2bb75SMartin Matuska pn = malloc((strlen(pathname) + 1) * sizeof(char)); 175552c2bb75SMartin Matuska p = pathname; 175652c2bb75SMartin Matuska s = pn; 175752c2bb75SMartin Matuska while(*p != '\0') { 175852c2bb75SMartin Matuska if(*p == '/') 175952c2bb75SMartin Matuska *s = '\\'; 176052c2bb75SMartin Matuska else 176152c2bb75SMartin Matuska *s = *p; 176252c2bb75SMartin Matuska p++; 176352c2bb75SMartin Matuska s++; 176452c2bb75SMartin Matuska } 176552c2bb75SMartin Matuska *s = '\0'; 176652c2bb75SMartin Matuska 176752c2bb75SMartin Matuska h = CreateFileA(pn, 0, FILE_SHARE_READ, NULL, OPEN_EXISTING, 176852c2bb75SMartin Matuska flag, NULL); 176952c2bb75SMartin Matuska free(pn); 177052c2bb75SMartin Matuska if (h == INVALID_HANDLE_VALUE) { 177152c2bb75SMartin Matuska failure_start(file, line, "Can't access %s\n", pathname); 177252c2bb75SMartin Matuska failure_finish(NULL); 177364287048SMartin Matuska return (0); 177452c2bb75SMartin Matuska } 177552c2bb75SMartin Matuska ret = GetFileInformationByHandle(h, &st); 177652c2bb75SMartin Matuska if (ret == 0) { 177752c2bb75SMartin Matuska failure_start(file, line, 177852c2bb75SMartin Matuska "Can't stat: %s", pathname); 177952c2bb75SMartin Matuska failure_finish(NULL); 178052c2bb75SMartin Matuska } else if ((st.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) == 0) { 178152c2bb75SMartin Matuska failure_start(file, line, 178252c2bb75SMartin Matuska "Not a symlink: %s", pathname); 178352c2bb75SMartin Matuska failure_finish(NULL); 178452c2bb75SMartin Matuska ret = 0; 178552c2bb75SMartin Matuska } 178652c2bb75SMartin Matuska if (isdir && ((st.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == 0)) { 178752c2bb75SMartin Matuska failure_start(file, line, 178852c2bb75SMartin Matuska "Not a directory symlink: %s", pathname); 178952c2bb75SMartin Matuska failure_finish(NULL); 179052c2bb75SMartin Matuska ret = 0; 179152c2bb75SMartin Matuska } 179252c2bb75SMartin Matuska if (!isdir && 179352c2bb75SMartin Matuska ((st.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0)) { 179452c2bb75SMartin Matuska failure_start(file, line, 179552c2bb75SMartin Matuska "Not a file symlink: %s", pathname); 179652c2bb75SMartin Matuska failure_finish(NULL); 179752c2bb75SMartin Matuska ret = 0; 179852c2bb75SMartin Matuska } 179952c2bb75SMartin Matuska if (ret == 0) { 180052c2bb75SMartin Matuska CloseHandle(h); 180152c2bb75SMartin Matuska return (0); 180252c2bb75SMartin Matuska } 180352c2bb75SMartin Matuska 180452c2bb75SMartin Matuska indata = malloc(MAXIMUM_REPARSE_DATA_BUFFER_SIZE); 180552c2bb75SMartin Matuska ret = DeviceIoControl(h, FSCTL_GET_REPARSE_POINT, NULL, 0, indata, 180652c2bb75SMartin Matuska 1024, &inbytes, NULL); 180752c2bb75SMartin Matuska CloseHandle(h); 180852c2bb75SMartin Matuska if (ret == 0) { 180952c2bb75SMartin Matuska free(indata); 181052c2bb75SMartin Matuska failure_start(file, line, 181152c2bb75SMartin Matuska "Could not retrieve symlink target: %s", pathname); 181252c2bb75SMartin Matuska failure_finish(NULL); 181352c2bb75SMartin Matuska return (0); 181452c2bb75SMartin Matuska } 181552c2bb75SMartin Matuska 181652c2bb75SMartin Matuska buf = (REPARSE_DATA_BUFFER *) indata; 181752c2bb75SMartin Matuska if (buf->ReparseTag != IO_REPARSE_TAG_SYMLINK) { 181852c2bb75SMartin Matuska free(indata); 181952c2bb75SMartin Matuska /* File is not a symbolic link */ 182052c2bb75SMartin Matuska failure_start(file, line, 182152c2bb75SMartin Matuska "Not a symlink: %s", pathname); 182252c2bb75SMartin Matuska failure_finish(NULL); 182352c2bb75SMartin Matuska return (0); 182452c2bb75SMartin Matuska } 182552c2bb75SMartin Matuska 182652c2bb75SMartin Matuska if (contents == NULL) { 182752c2bb75SMartin Matuska free(indata); 182852c2bb75SMartin Matuska return (1); 182952c2bb75SMartin Matuska } 183052c2bb75SMartin Matuska 183152c2bb75SMartin Matuska len = buf->SymbolicLinkReparseBuffer.SubstituteNameLength; 183252c2bb75SMartin Matuska 183352c2bb75SMartin Matuska linknamew = malloc(len + sizeof(wchar_t)); 183452c2bb75SMartin Matuska if (linknamew == NULL) { 183552c2bb75SMartin Matuska free(indata); 183652c2bb75SMartin Matuska return (0); 183752c2bb75SMartin Matuska } 183852c2bb75SMartin Matuska 183952c2bb75SMartin Matuska memcpy(linknamew, &((BYTE *)buf->SymbolicLinkReparseBuffer.PathBuffer) 184052c2bb75SMartin Matuska [buf->SymbolicLinkReparseBuffer.SubstituteNameOffset], len); 184152c2bb75SMartin Matuska free(indata); 184252c2bb75SMartin Matuska 184352c2bb75SMartin Matuska linknamew[len / sizeof(wchar_t)] = L'\0'; 184452c2bb75SMartin Matuska 184552c2bb75SMartin Matuska contentsw = malloc(len + sizeof(wchar_t)); 184652c2bb75SMartin Matuska if (contentsw == NULL) { 184752c2bb75SMartin Matuska free(linknamew); 184852c2bb75SMartin Matuska return (0); 184952c2bb75SMartin Matuska } 185052c2bb75SMartin Matuska 185152c2bb75SMartin Matuska len2 = mbsrtowcs(contentsw, &contents, (len + sizeof(wchar_t) 185252c2bb75SMartin Matuska / sizeof(wchar_t)), NULL); 185352c2bb75SMartin Matuska 185452c2bb75SMartin Matuska if (len2 > 0 && wcscmp(linknamew, contentsw) != 0) 185552c2bb75SMartin Matuska ret = 1; 185652c2bb75SMartin Matuska 185752c2bb75SMartin Matuska free(linknamew); 185852c2bb75SMartin Matuska free(contentsw); 185952c2bb75SMartin Matuska return (ret); 186064287048SMartin Matuska #else 186164287048SMartin Matuska char buff[300]; 186264287048SMartin Matuska struct stat st; 186364287048SMartin Matuska ssize_t linklen; 186464287048SMartin Matuska int r; 186564287048SMartin Matuska 186652c2bb75SMartin Matuska (void)isdir; /* UNUSED */ 186764287048SMartin Matuska assertion_count(file, line); 186864287048SMartin Matuska r = lstat(pathname, &st); 186964287048SMartin Matuska if (r != 0) { 187064287048SMartin Matuska failure_start(file, line, 187164287048SMartin Matuska "Symlink should exist: %s", pathname); 187264287048SMartin Matuska failure_finish(NULL); 187364287048SMartin Matuska return (0); 187464287048SMartin Matuska } 187564287048SMartin Matuska if (!S_ISLNK(st.st_mode)) 187664287048SMartin Matuska return (0); 187764287048SMartin Matuska if (contents == NULL) 187864287048SMartin Matuska return (1); 187979085fd3SMartin Matuska linklen = readlink(pathname, buff, sizeof(buff) - 1); 188064287048SMartin Matuska if (linklen < 0) { 188164287048SMartin Matuska failure_start(file, line, "Can't read symlink %s", pathname); 188264287048SMartin Matuska failure_finish(NULL); 188364287048SMartin Matuska return (0); 188464287048SMartin Matuska } 188564287048SMartin Matuska buff[linklen] = '\0'; 188664287048SMartin Matuska if (strcmp(buff, contents) != 0) 188764287048SMartin Matuska return (0); 188864287048SMartin Matuska return (1); 188964287048SMartin Matuska #endif 189064287048SMartin Matuska } 189164287048SMartin Matuska 189264287048SMartin Matuska /* Assert that path is a symlink that (optionally) contains contents. */ 189364287048SMartin Matuska int 189464287048SMartin Matuska assertion_is_symlink(const char *file, int line, 189552c2bb75SMartin Matuska const char *path, const char *contents, int isdir) 189664287048SMartin Matuska { 189752c2bb75SMartin Matuska if (is_symlink(file, line, path, contents, isdir)) 189864287048SMartin Matuska return (1); 189964287048SMartin Matuska if (contents) 190064287048SMartin Matuska failure_start(file, line, "File %s is not a symlink to %s", 190164287048SMartin Matuska path, contents); 190264287048SMartin Matuska else 190364287048SMartin Matuska failure_start(file, line, "File %s is not a symlink", path); 190464287048SMartin Matuska failure_finish(NULL); 190564287048SMartin Matuska return (0); 190664287048SMartin Matuska } 190764287048SMartin Matuska 190864287048SMartin Matuska 190964287048SMartin Matuska /* Create a directory and report any errors. */ 191064287048SMartin Matuska int 191164287048SMartin Matuska assertion_make_dir(const char *file, int line, const char *dirname, int mode) 191264287048SMartin Matuska { 191364287048SMartin Matuska assertion_count(file, line); 191464287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 191564287048SMartin Matuska (void)mode; /* UNUSED */ 191664287048SMartin Matuska if (0 == _mkdir(dirname)) 191764287048SMartin Matuska return (1); 191864287048SMartin Matuska #else 191964287048SMartin Matuska if (0 == mkdir(dirname, mode)) { 192064287048SMartin Matuska if (0 == chmod(dirname, mode)) { 192164287048SMartin Matuska assertion_file_mode(file, line, dirname, mode); 192264287048SMartin Matuska return (1); 192364287048SMartin Matuska } 192464287048SMartin Matuska } 192564287048SMartin Matuska #endif 192664287048SMartin Matuska failure_start(file, line, "Could not create directory %s", dirname); 192764287048SMartin Matuska failure_finish(NULL); 192864287048SMartin Matuska return(0); 192964287048SMartin Matuska } 193064287048SMartin Matuska 193164287048SMartin Matuska /* Create a file with the specified contents and report any failures. */ 193264287048SMartin Matuska int 193364287048SMartin Matuska assertion_make_file(const char *file, int line, 193464287048SMartin Matuska const char *path, int mode, int csize, const void *contents) 193564287048SMartin Matuska { 193664287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 193764287048SMartin Matuska /* TODO: Rework this to set file mode as well. */ 193864287048SMartin Matuska FILE *f; 193964287048SMartin Matuska (void)mode; /* UNUSED */ 194064287048SMartin Matuska assertion_count(file, line); 194164287048SMartin Matuska f = fopen(path, "wb"); 194264287048SMartin Matuska if (f == NULL) { 194364287048SMartin Matuska failure_start(file, line, "Could not create file %s", path); 194464287048SMartin Matuska failure_finish(NULL); 194564287048SMartin Matuska return (0); 194664287048SMartin Matuska } 194764287048SMartin Matuska if (contents != NULL) { 194864287048SMartin Matuska size_t wsize; 194964287048SMartin Matuska 195064287048SMartin Matuska if (csize < 0) 195164287048SMartin Matuska wsize = strlen(contents); 195264287048SMartin Matuska else 195364287048SMartin Matuska wsize = (size_t)csize; 195464287048SMartin Matuska if (wsize != fwrite(contents, 1, wsize, f)) { 195564287048SMartin Matuska fclose(f); 195664287048SMartin Matuska failure_start(file, line, 195764287048SMartin Matuska "Could not write file %s", path); 195864287048SMartin Matuska failure_finish(NULL); 195964287048SMartin Matuska return (0); 196064287048SMartin Matuska } 196164287048SMartin Matuska } 196264287048SMartin Matuska fclose(f); 196364287048SMartin Matuska return (1); 196464287048SMartin Matuska #else 196564287048SMartin Matuska int fd; 196664287048SMartin Matuska assertion_count(file, line); 196764287048SMartin Matuska fd = open(path, O_CREAT | O_WRONLY, mode >= 0 ? mode : 0644); 196864287048SMartin Matuska if (fd < 0) { 196964287048SMartin Matuska failure_start(file, line, "Could not create %s", path); 197064287048SMartin Matuska failure_finish(NULL); 197164287048SMartin Matuska return (0); 197264287048SMartin Matuska } 197364287048SMartin Matuska if (0 != chmod(path, mode)) { 197464287048SMartin Matuska failure_start(file, line, "Could not chmod %s", path); 197564287048SMartin Matuska failure_finish(NULL); 197664287048SMartin Matuska close(fd); 197764287048SMartin Matuska return (0); 197864287048SMartin Matuska } 197964287048SMartin Matuska if (contents != NULL) { 198064287048SMartin Matuska ssize_t wsize; 198164287048SMartin Matuska 198264287048SMartin Matuska if (csize < 0) 198364287048SMartin Matuska wsize = (ssize_t)strlen(contents); 198464287048SMartin Matuska else 198564287048SMartin Matuska wsize = (ssize_t)csize; 198664287048SMartin Matuska if (wsize != write(fd, contents, wsize)) { 198764287048SMartin Matuska close(fd); 198864287048SMartin Matuska failure_start(file, line, 198964287048SMartin Matuska "Could not write to %s", path); 199064287048SMartin Matuska failure_finish(NULL); 199164287048SMartin Matuska close(fd); 199264287048SMartin Matuska return (0); 199364287048SMartin Matuska } 199464287048SMartin Matuska } 199564287048SMartin Matuska close(fd); 199664287048SMartin Matuska assertion_file_mode(file, line, path, mode); 199764287048SMartin Matuska return (1); 199864287048SMartin Matuska #endif 199964287048SMartin Matuska } 200064287048SMartin Matuska 200164287048SMartin Matuska /* Create a hardlink and report any failures. */ 200264287048SMartin Matuska int 200364287048SMartin Matuska assertion_make_hardlink(const char *file, int line, 200464287048SMartin Matuska const char *newpath, const char *linkto) 200564287048SMartin Matuska { 200664287048SMartin Matuska int succeeded; 200764287048SMartin Matuska 200864287048SMartin Matuska assertion_count(file, line); 200964287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 201064287048SMartin Matuska succeeded = my_CreateHardLinkA(newpath, linkto); 201164287048SMartin Matuska #elif HAVE_LINK 201264287048SMartin Matuska succeeded = !link(linkto, newpath); 201364287048SMartin Matuska #else 201464287048SMartin Matuska succeeded = 0; 201564287048SMartin Matuska #endif 201664287048SMartin Matuska if (succeeded) 201764287048SMartin Matuska return (1); 201864287048SMartin Matuska failure_start(file, line, "Could not create hardlink"); 201964287048SMartin Matuska logprintf(" New link: %s\n", newpath); 202064287048SMartin Matuska logprintf(" Old name: %s\n", linkto); 202164287048SMartin Matuska failure_finish(NULL); 202264287048SMartin Matuska return(0); 202364287048SMartin Matuska } 202464287048SMartin Matuska 202552c2bb75SMartin Matuska /* 202652c2bb75SMartin Matuska * Create a symlink and report any failures. 202752c2bb75SMartin Matuska * 202852c2bb75SMartin Matuska * Windows symlinks need to know if the target is a directory. 202952c2bb75SMartin Matuska */ 203064287048SMartin Matuska int 203164287048SMartin Matuska assertion_make_symlink(const char *file, int line, 203252c2bb75SMartin Matuska const char *newpath, const char *linkto, int targetIsDir) 203364287048SMartin Matuska { 203464287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 203564287048SMartin Matuska assertion_count(file, line); 203664287048SMartin Matuska if (my_CreateSymbolicLinkA(newpath, linkto, targetIsDir)) 203764287048SMartin Matuska return (1); 203864287048SMartin Matuska #elif HAVE_SYMLINK 203952c2bb75SMartin Matuska (void)targetIsDir; /* UNUSED */ 204064287048SMartin Matuska assertion_count(file, line); 204164287048SMartin Matuska if (0 == symlink(linkto, newpath)) 204264287048SMartin Matuska return (1); 204352c2bb75SMartin Matuska #else 204452c2bb75SMartin Matuska (void)targetIsDir; /* UNUSED */ 204564287048SMartin Matuska #endif 204664287048SMartin Matuska failure_start(file, line, "Could not create symlink"); 204764287048SMartin Matuska logprintf(" New link: %s\n", newpath); 204864287048SMartin Matuska logprintf(" Old name: %s\n", linkto); 204964287048SMartin Matuska failure_finish(NULL); 205064287048SMartin Matuska return(0); 205164287048SMartin Matuska } 205264287048SMartin Matuska 205364287048SMartin Matuska /* Set umask, report failures. */ 205464287048SMartin Matuska int 205564287048SMartin Matuska assertion_umask(const char *file, int line, int mask) 205664287048SMartin Matuska { 205764287048SMartin Matuska assertion_count(file, line); 205864287048SMartin Matuska (void)file; /* UNUSED */ 205964287048SMartin Matuska (void)line; /* UNUSED */ 206064287048SMartin Matuska umask(mask); 206164287048SMartin Matuska return (1); 206264287048SMartin Matuska } 206364287048SMartin Matuska 206464287048SMartin Matuska /* Set times, report failures. */ 206564287048SMartin Matuska int 206664287048SMartin Matuska assertion_utimes(const char *file, int line, 206764287048SMartin Matuska const char *pathname, long at, long at_nsec, long mt, long mt_nsec) 206864287048SMartin Matuska { 206964287048SMartin Matuska int r; 207064287048SMartin Matuska 207164287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 207264287048SMartin Matuska #define WINTIME(sec, nsec) ((Int32x32To64(sec, 10000000) + EPOC_TIME)\ 207364287048SMartin Matuska + (((nsec)/1000)*10)) 207464287048SMartin Matuska HANDLE h; 207564287048SMartin Matuska ULARGE_INTEGER wintm; 207664287048SMartin Matuska FILETIME fatime, fmtime; 207764287048SMartin Matuska FILETIME *pat, *pmt; 207864287048SMartin Matuska 207964287048SMartin Matuska assertion_count(file, line); 208064287048SMartin Matuska h = CreateFileA(pathname,GENERIC_READ | GENERIC_WRITE, 208164287048SMartin Matuska FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 208264287048SMartin Matuska FILE_FLAG_BACKUP_SEMANTICS, NULL); 208364287048SMartin Matuska if (h == INVALID_HANDLE_VALUE) { 208464287048SMartin Matuska failure_start(file, line, "Can't access %s\n", pathname); 208564287048SMartin Matuska failure_finish(NULL); 208664287048SMartin Matuska return (0); 208764287048SMartin Matuska } 208864287048SMartin Matuska 208964287048SMartin Matuska if (at > 0 || at_nsec > 0) { 209064287048SMartin Matuska wintm.QuadPart = WINTIME(at, at_nsec); 209164287048SMartin Matuska fatime.dwLowDateTime = wintm.LowPart; 209264287048SMartin Matuska fatime.dwHighDateTime = wintm.HighPart; 209364287048SMartin Matuska pat = &fatime; 209464287048SMartin Matuska } else 209564287048SMartin Matuska pat = NULL; 209664287048SMartin Matuska if (mt > 0 || mt_nsec > 0) { 209764287048SMartin Matuska wintm.QuadPart = WINTIME(mt, mt_nsec); 209864287048SMartin Matuska fmtime.dwLowDateTime = wintm.LowPart; 209964287048SMartin Matuska fmtime.dwHighDateTime = wintm.HighPart; 210064287048SMartin Matuska pmt = &fmtime; 210164287048SMartin Matuska } else 210264287048SMartin Matuska pmt = NULL; 210364287048SMartin Matuska if (pat != NULL || pmt != NULL) 210464287048SMartin Matuska r = SetFileTime(h, NULL, pat, pmt); 210564287048SMartin Matuska else 210664287048SMartin Matuska r = 1; 210764287048SMartin Matuska CloseHandle(h); 210864287048SMartin Matuska if (r == 0) { 210964287048SMartin Matuska failure_start(file, line, "Can't SetFileTime %s\n", pathname); 211064287048SMartin Matuska failure_finish(NULL); 211164287048SMartin Matuska return (0); 211264287048SMartin Matuska } 211364287048SMartin Matuska return (1); 211464287048SMartin Matuska #else /* defined(_WIN32) && !defined(__CYGWIN__) */ 211564287048SMartin Matuska struct stat st; 211664287048SMartin Matuska struct timeval times[2]; 211764287048SMartin Matuska 211864287048SMartin Matuska #if !defined(__FreeBSD__) 211964287048SMartin Matuska mt_nsec = at_nsec = 0; /* Generic POSIX only has whole seconds. */ 212064287048SMartin Matuska #endif 212164287048SMartin Matuska if (mt == 0 && mt_nsec == 0 && at == 0 && at_nsec == 0) 212264287048SMartin Matuska return (1); 212364287048SMartin Matuska 212464287048SMartin Matuska r = lstat(pathname, &st); 212564287048SMartin Matuska if (r < 0) { 212664287048SMartin Matuska failure_start(file, line, "Can't stat %s\n", pathname); 212764287048SMartin Matuska failure_finish(NULL); 212864287048SMartin Matuska return (0); 212964287048SMartin Matuska } 213064287048SMartin Matuska 213164287048SMartin Matuska if (mt == 0 && mt_nsec == 0) { 213264287048SMartin Matuska mt = st.st_mtime; 213364287048SMartin Matuska #if defined(__FreeBSD__) 213464287048SMartin Matuska mt_nsec = st.st_mtimespec.tv_nsec; 213564287048SMartin Matuska /* FreeBSD generally only stores to microsecond res, so round. */ 213664287048SMartin Matuska mt_nsec = (mt_nsec / 1000) * 1000; 213764287048SMartin Matuska #endif 213864287048SMartin Matuska } 213964287048SMartin Matuska if (at == 0 && at_nsec == 0) { 214064287048SMartin Matuska at = st.st_atime; 214164287048SMartin Matuska #if defined(__FreeBSD__) 214264287048SMartin Matuska at_nsec = st.st_atimespec.tv_nsec; 214364287048SMartin Matuska /* FreeBSD generally only stores to microsecond res, so round. */ 214464287048SMartin Matuska at_nsec = (at_nsec / 1000) * 1000; 214564287048SMartin Matuska #endif 214664287048SMartin Matuska } 214764287048SMartin Matuska 214864287048SMartin Matuska times[1].tv_sec = mt; 214964287048SMartin Matuska times[1].tv_usec = mt_nsec / 1000; 215064287048SMartin Matuska 215164287048SMartin Matuska times[0].tv_sec = at; 215264287048SMartin Matuska times[0].tv_usec = at_nsec / 1000; 215364287048SMartin Matuska 215464287048SMartin Matuska #ifdef HAVE_LUTIMES 215564287048SMartin Matuska r = lutimes(pathname, times); 215664287048SMartin Matuska #else 215764287048SMartin Matuska r = utimes(pathname, times); 215864287048SMartin Matuska #endif 215964287048SMartin Matuska if (r < 0) { 216064287048SMartin Matuska failure_start(file, line, "Can't utimes %s\n", pathname); 216164287048SMartin Matuska failure_finish(NULL); 216264287048SMartin Matuska return (0); 216364287048SMartin Matuska } 216464287048SMartin Matuska return (1); 216564287048SMartin Matuska #endif /* defined(_WIN32) && !defined(__CYGWIN__) */ 216664287048SMartin Matuska } 216764287048SMartin Matuska 216864287048SMartin Matuska /* Compare file flags */ 216964287048SMartin Matuska int 217064287048SMartin Matuska assertion_compare_fflags(const char *file, int line, const char *patha, 217164287048SMartin Matuska const char *pathb, int nomatch) 217264287048SMartin Matuska { 217364287048SMartin Matuska #if defined(HAVE_STRUCT_STAT_ST_FLAGS) && defined(UF_NODUMP) 217464287048SMartin Matuska struct stat sa, sb; 217564287048SMartin Matuska 217664287048SMartin Matuska assertion_count(file, line); 217764287048SMartin Matuska 217864287048SMartin Matuska if (stat(patha, &sa) < 0) 217964287048SMartin Matuska return (0); 218064287048SMartin Matuska if (stat(pathb, &sb) < 0) 218164287048SMartin Matuska return (0); 218264287048SMartin Matuska if (!nomatch && sa.st_flags != sb.st_flags) { 218364287048SMartin Matuska failure_start(file, line, "File flags should be identical: " 218464287048SMartin Matuska "%s=%#010x %s=%#010x", patha, sa.st_flags, pathb, 218564287048SMartin Matuska sb.st_flags); 218664287048SMartin Matuska failure_finish(NULL); 218764287048SMartin Matuska return (0); 218864287048SMartin Matuska } 218964287048SMartin Matuska if (nomatch && sa.st_flags == sb.st_flags) { 219064287048SMartin Matuska failure_start(file, line, "File flags should be different: " 219164287048SMartin Matuska "%s=%#010x %s=%#010x", patha, sa.st_flags, pathb, 219264287048SMartin Matuska sb.st_flags); 219364287048SMartin Matuska failure_finish(NULL); 219464287048SMartin Matuska return (0); 219564287048SMartin Matuska } 219664287048SMartin Matuska #elif (defined(FS_IOC_GETFLAGS) && defined(HAVE_WORKING_FS_IOC_GETFLAGS) && \ 219764287048SMartin Matuska defined(FS_NODUMP_FL)) || \ 219864287048SMartin Matuska (defined(EXT2_IOC_GETFLAGS) && defined(HAVE_WORKING_EXT2_IOC_GETFLAGS) \ 219964287048SMartin Matuska && defined(EXT2_NODUMP_FL)) 220064287048SMartin Matuska int fd, r, flagsa, flagsb; 220164287048SMartin Matuska 220264287048SMartin Matuska assertion_count(file, line); 220364287048SMartin Matuska fd = open(patha, O_RDONLY | O_NONBLOCK); 220464287048SMartin Matuska if (fd < 0) { 220564287048SMartin Matuska failure_start(file, line, "Can't open %s\n", patha); 220664287048SMartin Matuska failure_finish(NULL); 220764287048SMartin Matuska return (0); 220864287048SMartin Matuska } 220964287048SMartin Matuska r = ioctl(fd, 221064287048SMartin Matuska #ifdef FS_IOC_GETFLAGS 221164287048SMartin Matuska FS_IOC_GETFLAGS, 221264287048SMartin Matuska #else 221364287048SMartin Matuska EXT2_IOC_GETFLAGS, 221464287048SMartin Matuska #endif 221564287048SMartin Matuska &flagsa); 221664287048SMartin Matuska close(fd); 221764287048SMartin Matuska if (r < 0) { 221864287048SMartin Matuska failure_start(file, line, "Can't get flags %s\n", patha); 221964287048SMartin Matuska failure_finish(NULL); 222064287048SMartin Matuska return (0); 222164287048SMartin Matuska } 222264287048SMartin Matuska fd = open(pathb, O_RDONLY | O_NONBLOCK); 222364287048SMartin Matuska if (fd < 0) { 222464287048SMartin Matuska failure_start(file, line, "Can't open %s\n", pathb); 222564287048SMartin Matuska failure_finish(NULL); 222664287048SMartin Matuska return (0); 222764287048SMartin Matuska } 222864287048SMartin Matuska r = ioctl(fd, 222964287048SMartin Matuska #ifdef FS_IOC_GETFLAGS 223064287048SMartin Matuska FS_IOC_GETFLAGS, 223164287048SMartin Matuska #else 223264287048SMartin Matuska EXT2_IOC_GETFLAGS, 223364287048SMartin Matuska #endif 223464287048SMartin Matuska &flagsb); 223564287048SMartin Matuska close(fd); 223664287048SMartin Matuska if (r < 0) { 223764287048SMartin Matuska failure_start(file, line, "Can't get flags %s\n", pathb); 223864287048SMartin Matuska failure_finish(NULL); 223964287048SMartin Matuska return (0); 224064287048SMartin Matuska } 224164287048SMartin Matuska if (!nomatch && flagsa != flagsb) { 224264287048SMartin Matuska failure_start(file, line, "File flags should be identical: " 224364287048SMartin Matuska "%s=%#010x %s=%#010x", patha, flagsa, pathb, flagsb); 224464287048SMartin Matuska failure_finish(NULL); 224564287048SMartin Matuska return (0); 224664287048SMartin Matuska } 224764287048SMartin Matuska if (nomatch && flagsa == flagsb) { 224864287048SMartin Matuska failure_start(file, line, "File flags should be different: " 224964287048SMartin Matuska "%s=%#010x %s=%#010x", patha, flagsa, pathb, flagsb); 225064287048SMartin Matuska failure_finish(NULL); 225164287048SMartin Matuska return (0); 225264287048SMartin Matuska } 225364287048SMartin Matuska #else 225464287048SMartin Matuska (void)patha; /* UNUSED */ 225564287048SMartin Matuska (void)pathb; /* UNUSED */ 225664287048SMartin Matuska (void)nomatch; /* UNUSED */ 225764287048SMartin Matuska assertion_count(file, line); 225864287048SMartin Matuska #endif 225964287048SMartin Matuska return (1); 226064287048SMartin Matuska } 226164287048SMartin Matuska 226264287048SMartin Matuska /* Set nodump, report failures. */ 226364287048SMartin Matuska int 226464287048SMartin Matuska assertion_set_nodump(const char *file, int line, const char *pathname) 226564287048SMartin Matuska { 226664287048SMartin Matuska #if defined(HAVE_STRUCT_STAT_ST_FLAGS) && defined(UF_NODUMP) 226764287048SMartin Matuska int r; 226864287048SMartin Matuska 226964287048SMartin Matuska assertion_count(file, line); 227064287048SMartin Matuska r = chflags(pathname, UF_NODUMP); 227164287048SMartin Matuska if (r < 0) { 227264287048SMartin Matuska failure_start(file, line, "Can't set nodump %s\n", pathname); 227364287048SMartin Matuska failure_finish(NULL); 227464287048SMartin Matuska return (0); 227564287048SMartin Matuska } 227664287048SMartin Matuska #elif (defined(FS_IOC_GETFLAGS) && defined(HAVE_WORKING_FS_IOC_GETFLAGS) && \ 227764287048SMartin Matuska defined(FS_NODUMP_FL)) || \ 227864287048SMartin Matuska (defined(EXT2_IOC_GETFLAGS) && defined(HAVE_WORKING_EXT2_IOC_GETFLAGS) \ 227964287048SMartin Matuska && defined(EXT2_NODUMP_FL)) 228064287048SMartin Matuska int fd, r, flags; 228164287048SMartin Matuska 228264287048SMartin Matuska assertion_count(file, line); 228364287048SMartin Matuska fd = open(pathname, O_RDONLY | O_NONBLOCK); 228464287048SMartin Matuska if (fd < 0) { 228564287048SMartin Matuska failure_start(file, line, "Can't open %s\n", pathname); 228664287048SMartin Matuska failure_finish(NULL); 228764287048SMartin Matuska return (0); 228864287048SMartin Matuska } 228964287048SMartin Matuska r = ioctl(fd, 229064287048SMartin Matuska #ifdef FS_IOC_GETFLAGS 229164287048SMartin Matuska FS_IOC_GETFLAGS, 229264287048SMartin Matuska #else 229364287048SMartin Matuska EXT2_IOC_GETFLAGS, 229464287048SMartin Matuska #endif 229564287048SMartin Matuska &flags); 229664287048SMartin Matuska if (r < 0) { 229764287048SMartin Matuska failure_start(file, line, "Can't get flags %s\n", pathname); 229864287048SMartin Matuska failure_finish(NULL); 229964287048SMartin Matuska return (0); 230064287048SMartin Matuska } 230164287048SMartin Matuska #ifdef FS_NODUMP_FL 230264287048SMartin Matuska flags |= FS_NODUMP_FL; 230364287048SMartin Matuska #else 230464287048SMartin Matuska flags |= EXT2_NODUMP_FL; 230564287048SMartin Matuska #endif 230664287048SMartin Matuska 230764287048SMartin Matuska r = ioctl(fd, 230864287048SMartin Matuska #ifdef FS_IOC_SETFLAGS 230964287048SMartin Matuska FS_IOC_SETFLAGS, 231064287048SMartin Matuska #else 231164287048SMartin Matuska EXT2_IOC_SETFLAGS, 231264287048SMartin Matuska #endif 231364287048SMartin Matuska &flags); 231464287048SMartin Matuska if (r < 0) { 231564287048SMartin Matuska failure_start(file, line, "Can't set nodump %s\n", pathname); 231664287048SMartin Matuska failure_finish(NULL); 231764287048SMartin Matuska return (0); 231864287048SMartin Matuska } 231964287048SMartin Matuska close(fd); 232064287048SMartin Matuska #else 232164287048SMartin Matuska (void)pathname; /* UNUSED */ 232264287048SMartin Matuska assertion_count(file, line); 232364287048SMartin Matuska #endif 232464287048SMartin Matuska return (1); 232564287048SMartin Matuska } 232664287048SMartin Matuska 232764287048SMartin Matuska #ifdef PROGRAM 232864287048SMartin Matuska static void assert_version_id(char **qq, size_t *ss) 232964287048SMartin Matuska { 233064287048SMartin Matuska char *q = *qq; 233164287048SMartin Matuska size_t s = *ss; 233264287048SMartin Matuska 233364287048SMartin Matuska /* Version number is a series of digits and periods. */ 233464287048SMartin Matuska while (s > 0 && (*q == '.' || (*q >= '0' && *q <= '9'))) { 233564287048SMartin Matuska ++q; 233664287048SMartin Matuska --s; 233764287048SMartin Matuska } 233864287048SMartin Matuska 233964287048SMartin Matuska if (q[0] == 'd' && q[1] == 'e' && q[2] == 'v') { 234064287048SMartin Matuska q += 3; 234164287048SMartin Matuska s -= 3; 234264287048SMartin Matuska } 234364287048SMartin Matuska 234464287048SMartin Matuska /* Skip a single trailing a,b,c, or d. */ 234564287048SMartin Matuska if (*q == 'a' || *q == 'b' || *q == 'c' || *q == 'd') 234664287048SMartin Matuska ++q; 234764287048SMartin Matuska 234864287048SMartin Matuska /* Version number terminated by space. */ 234964287048SMartin Matuska failure("No space after version: ``%s''", q); 235064287048SMartin Matuska assert(s > 1); 235164287048SMartin Matuska failure("No space after version: ``%s''", q); 235264287048SMartin Matuska assert(*q == ' '); 235364287048SMartin Matuska 235464287048SMartin Matuska ++q; --s; 235564287048SMartin Matuska 235664287048SMartin Matuska *qq = q; 235764287048SMartin Matuska *ss = s; 235864287048SMartin Matuska } 235964287048SMartin Matuska 236064287048SMartin Matuska 236164287048SMartin Matuska /* 236264287048SMartin Matuska * Check program version 236364287048SMartin Matuska */ 236464287048SMartin Matuska void assertVersion(const char *prog, const char *base) 236564287048SMartin Matuska { 236664287048SMartin Matuska int r; 236764287048SMartin Matuska char *p, *q; 236864287048SMartin Matuska size_t s; 2369f1951fd7SMartin Matuska size_t prog_len = strlen(base); 237064287048SMartin Matuska 237164287048SMartin Matuska r = systemf("%s --version >version.stdout 2>version.stderr", prog); 237264287048SMartin Matuska if (r != 0) 237364287048SMartin Matuska r = systemf("%s -W version >version.stdout 2>version.stderr", 237464287048SMartin Matuska prog); 237564287048SMartin Matuska 237664287048SMartin Matuska failure("Unable to run either %s --version or %s -W version", 237764287048SMartin Matuska prog, prog); 237864287048SMartin Matuska if (!assert(r == 0)) 237964287048SMartin Matuska return; 238064287048SMartin Matuska 238164287048SMartin Matuska /* --version should generate nothing to stdout. */ 238264287048SMartin Matuska assertEmptyFile("version.stderr"); 238364287048SMartin Matuska 238464287048SMartin Matuska /* Verify format of version message. */ 238564287048SMartin Matuska q = p = slurpfile(&s, "version.stdout"); 238664287048SMartin Matuska 238764287048SMartin Matuska /* Version message should start with name of program, then space. */ 238864287048SMartin Matuska assert(s > prog_len + 1); 238964287048SMartin Matuska 239064287048SMartin Matuska failure("Version must start with '%s': ``%s''", base, p); 239164287048SMartin Matuska if (!assertEqualMem(q, base, prog_len)) { 239264287048SMartin Matuska free(p); 239364287048SMartin Matuska return; 239464287048SMartin Matuska } 239564287048SMartin Matuska 239664287048SMartin Matuska q += prog_len; s -= prog_len; 239764287048SMartin Matuska 239864287048SMartin Matuska assert(*q == ' '); 239964287048SMartin Matuska q++; s--; 240064287048SMartin Matuska 240164287048SMartin Matuska assert_version_id(&q, &s); 240264287048SMartin Matuska 240364287048SMartin Matuska /* Separator. */ 240464287048SMartin Matuska failure("No `-' between program name and versions: ``%s''", p); 240564287048SMartin Matuska assertEqualMem(q, "- ", 2); 240664287048SMartin Matuska q += 2; s -= 2; 240764287048SMartin Matuska 240864287048SMartin Matuska failure("Not long enough for libarchive version: ``%s''", p); 240964287048SMartin Matuska assert(s > 11); 241064287048SMartin Matuska 241164287048SMartin Matuska failure("Libarchive version must start with `libarchive': ``%s''", p); 241264287048SMartin Matuska assertEqualMem(q, "libarchive ", 11); 241364287048SMartin Matuska 241464287048SMartin Matuska q += 11; s -= 11; 241564287048SMartin Matuska 241664287048SMartin Matuska assert_version_id(&q, &s); 241764287048SMartin Matuska 241864287048SMartin Matuska /* Skip arbitrary third-party version numbers. */ 241964287048SMartin Matuska while (s > 0 && (*q == ' ' || *q == '-' || *q == '/' || *q == '.' || 2420482a8150SMartin Matuska isalnum((unsigned char)*q))) { 242164287048SMartin Matuska ++q; 242264287048SMartin Matuska --s; 242364287048SMartin Matuska } 242464287048SMartin Matuska 242564287048SMartin Matuska /* All terminated by end-of-line. */ 242664287048SMartin Matuska assert(s >= 1); 242764287048SMartin Matuska 242864287048SMartin Matuska /* Skip an optional CR character (e.g., Windows) */ 242964287048SMartin Matuska failure("Version output must end with \\n or \\r\\n"); 243064287048SMartin Matuska 243164287048SMartin Matuska if (*q == '\r') { ++q; --s; } 243264287048SMartin Matuska assertEqualMem(q, "\n", 1); 243364287048SMartin Matuska 243464287048SMartin Matuska free(p); 243564287048SMartin Matuska } 243664287048SMartin Matuska #endif /* PROGRAM */ 243764287048SMartin Matuska 243864287048SMartin Matuska /* 243964287048SMartin Matuska * 244064287048SMartin Matuska * UTILITIES for use by tests. 244164287048SMartin Matuska * 244264287048SMartin Matuska */ 244364287048SMartin Matuska 244464287048SMartin Matuska /* 244564287048SMartin Matuska * Check whether platform supports symlinks. This is intended 244664287048SMartin Matuska * for tests to use in deciding whether to bother testing symlink 244764287048SMartin Matuska * support; if the platform doesn't support symlinks, there's no point 244864287048SMartin Matuska * in checking whether the program being tested can create them. 244964287048SMartin Matuska * 245064287048SMartin Matuska * Note that the first time this test is called, we actually go out to 245164287048SMartin Matuska * disk to create and verify a symlink. This is necessary because 245264287048SMartin Matuska * symlink support is actually a property of a particular filesystem 245364287048SMartin Matuska * and can thus vary between directories on a single system. After 245464287048SMartin Matuska * the first call, this returns the cached result from memory, so it's 245564287048SMartin Matuska * safe to call it as often as you wish. 245664287048SMartin Matuska */ 245764287048SMartin Matuska int 245864287048SMartin Matuska canSymlink(void) 245964287048SMartin Matuska { 246064287048SMartin Matuska /* Remember the test result */ 246164287048SMartin Matuska static int value = 0, tested = 0; 246264287048SMartin Matuska if (tested) 246364287048SMartin Matuska return (value); 246464287048SMartin Matuska 246564287048SMartin Matuska ++tested; 246664287048SMartin Matuska assertion_make_file(__FILE__, __LINE__, "canSymlink.0", 0644, 1, "a"); 246764287048SMartin Matuska /* Note: Cygwin has its own symlink() emulation that does not 246864287048SMartin Matuska * use the Win32 CreateSymbolicLink() function. */ 246964287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 247064287048SMartin Matuska value = my_CreateSymbolicLinkA("canSymlink.1", "canSymlink.0", 0) 247152c2bb75SMartin Matuska && is_symlink(__FILE__, __LINE__, "canSymlink.1", "canSymlink.0", 247252c2bb75SMartin Matuska 0); 247364287048SMartin Matuska #elif HAVE_SYMLINK 247464287048SMartin Matuska value = (0 == symlink("canSymlink.0", "canSymlink.1")) 247552c2bb75SMartin Matuska && is_symlink(__FILE__, __LINE__, "canSymlink.1","canSymlink.0", 247652c2bb75SMartin Matuska 0); 247764287048SMartin Matuska #endif 247864287048SMartin Matuska return (value); 247964287048SMartin Matuska } 248064287048SMartin Matuska 248164287048SMartin Matuska /* Platform-dependent options for hiding the output of a subcommand. */ 248264287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 248364287048SMartin Matuska static const char *redirectArgs = ">NUL 2>NUL"; /* Win32 cmd.exe */ 248464287048SMartin Matuska #else 248564287048SMartin Matuska static const char *redirectArgs = ">/dev/null 2>/dev/null"; /* POSIX 'sh' */ 248664287048SMartin Matuska #endif 248764287048SMartin Matuska /* 248864287048SMartin Matuska * Can this platform run the bzip2 program? 248964287048SMartin Matuska */ 249064287048SMartin Matuska int 249164287048SMartin Matuska canBzip2(void) 249264287048SMartin Matuska { 249364287048SMartin Matuska static int tested = 0, value = 0; 249464287048SMartin Matuska if (!tested) { 249564287048SMartin Matuska tested = 1; 2496f9762417SMartin Matuska if (systemf("bzip2 --help %s", redirectArgs) == 0) 249764287048SMartin Matuska value = 1; 249864287048SMartin Matuska } 249964287048SMartin Matuska return (value); 250064287048SMartin Matuska } 250164287048SMartin Matuska 250264287048SMartin Matuska /* 250364287048SMartin Matuska * Can this platform run the grzip program? 250464287048SMartin Matuska */ 250564287048SMartin Matuska int 250664287048SMartin Matuska canGrzip(void) 250764287048SMartin Matuska { 250864287048SMartin Matuska static int tested = 0, value = 0; 250964287048SMartin Matuska if (!tested) { 251064287048SMartin Matuska tested = 1; 251164287048SMartin Matuska if (systemf("grzip -V %s", redirectArgs) == 0) 251264287048SMartin Matuska value = 1; 251364287048SMartin Matuska } 251464287048SMartin Matuska return (value); 251564287048SMartin Matuska } 251664287048SMartin Matuska 251764287048SMartin Matuska /* 251864287048SMartin Matuska * Can this platform run the gzip program? 251964287048SMartin Matuska */ 252064287048SMartin Matuska int 252164287048SMartin Matuska canGzip(void) 252264287048SMartin Matuska { 252364287048SMartin Matuska static int tested = 0, value = 0; 252464287048SMartin Matuska if (!tested) { 252564287048SMartin Matuska tested = 1; 2526f9762417SMartin Matuska if (systemf("gzip --help %s", redirectArgs) == 0) 252764287048SMartin Matuska value = 1; 252864287048SMartin Matuska } 252964287048SMartin Matuska return (value); 253064287048SMartin Matuska } 253164287048SMartin Matuska 253264287048SMartin Matuska /* 253364287048SMartin Matuska * Can this platform run the lrzip program? 253464287048SMartin Matuska */ 253564287048SMartin Matuska int 253664287048SMartin Matuska canRunCommand(const char *cmd) 253764287048SMartin Matuska { 253864287048SMartin Matuska static int tested = 0, value = 0; 253964287048SMartin Matuska if (!tested) { 254064287048SMartin Matuska tested = 1; 254164287048SMartin Matuska if (systemf("%s %s", cmd, redirectArgs) == 0) 254264287048SMartin Matuska value = 1; 254364287048SMartin Matuska } 254464287048SMartin Matuska return (value); 254564287048SMartin Matuska } 254664287048SMartin Matuska 254764287048SMartin Matuska int 254864287048SMartin Matuska canLrzip(void) 254964287048SMartin Matuska { 255064287048SMartin Matuska static int tested = 0, value = 0; 255164287048SMartin Matuska if (!tested) { 255264287048SMartin Matuska tested = 1; 255364287048SMartin Matuska if (systemf("lrzip -V %s", redirectArgs) == 0) 255464287048SMartin Matuska value = 1; 255564287048SMartin Matuska } 255664287048SMartin Matuska return (value); 255764287048SMartin Matuska } 255864287048SMartin Matuska 255964287048SMartin Matuska /* 256064287048SMartin Matuska * Can this platform run the lz4 program? 256164287048SMartin Matuska */ 256264287048SMartin Matuska int 256364287048SMartin Matuska canLz4(void) 256464287048SMartin Matuska { 256564287048SMartin Matuska static int tested = 0, value = 0; 256664287048SMartin Matuska if (!tested) { 256764287048SMartin Matuska tested = 1; 2568f9762417SMartin Matuska if (systemf("lz4 --help %s", redirectArgs) == 0) 256964287048SMartin Matuska value = 1; 257064287048SMartin Matuska } 257164287048SMartin Matuska return (value); 257264287048SMartin Matuska } 257364287048SMartin Matuska 257464287048SMartin Matuska /* 25755c831a5bSMartin Matuska * Can this platform run the zstd program? 25765c831a5bSMartin Matuska */ 25775c831a5bSMartin Matuska int 25785c831a5bSMartin Matuska canZstd(void) 25795c831a5bSMartin Matuska { 25805c831a5bSMartin Matuska static int tested = 0, value = 0; 25815c831a5bSMartin Matuska if (!tested) { 25825c831a5bSMartin Matuska tested = 1; 2583f9762417SMartin Matuska if (systemf("zstd --help %s", redirectArgs) == 0) 25845c831a5bSMartin Matuska value = 1; 25855c831a5bSMartin Matuska } 25865c831a5bSMartin Matuska return (value); 25875c831a5bSMartin Matuska } 25885c831a5bSMartin Matuska 25895c831a5bSMartin Matuska /* 259064287048SMartin Matuska * Can this platform run the lzip program? 259164287048SMartin Matuska */ 259264287048SMartin Matuska int 259364287048SMartin Matuska canLzip(void) 259464287048SMartin Matuska { 259564287048SMartin Matuska static int tested = 0, value = 0; 259664287048SMartin Matuska if (!tested) { 259764287048SMartin Matuska tested = 1; 2598f9762417SMartin Matuska if (systemf("lzip --help %s", redirectArgs) == 0) 259964287048SMartin Matuska value = 1; 260064287048SMartin Matuska } 260164287048SMartin Matuska return (value); 260264287048SMartin Matuska } 260364287048SMartin Matuska 260464287048SMartin Matuska /* 260564287048SMartin Matuska * Can this platform run the lzma program? 260664287048SMartin Matuska */ 260764287048SMartin Matuska int 260864287048SMartin Matuska canLzma(void) 260964287048SMartin Matuska { 261064287048SMartin Matuska static int tested = 0, value = 0; 261164287048SMartin Matuska if (!tested) { 261264287048SMartin Matuska tested = 1; 2613f9762417SMartin Matuska if (systemf("lzma %s", redirectArgs) == 0) 261464287048SMartin Matuska value = 1; 261564287048SMartin Matuska } 261664287048SMartin Matuska return (value); 261764287048SMartin Matuska } 261864287048SMartin Matuska 261964287048SMartin Matuska /* 262064287048SMartin Matuska * Can this platform run the lzop program? 262164287048SMartin Matuska */ 262264287048SMartin Matuska int 262364287048SMartin Matuska canLzop(void) 262464287048SMartin Matuska { 262564287048SMartin Matuska static int tested = 0, value = 0; 262664287048SMartin Matuska if (!tested) { 262764287048SMartin Matuska tested = 1; 2628f9762417SMartin Matuska if (systemf("lzop --help %s", redirectArgs) == 0) 262964287048SMartin Matuska value = 1; 263064287048SMartin Matuska } 263164287048SMartin Matuska return (value); 263264287048SMartin Matuska } 263364287048SMartin Matuska 263464287048SMartin Matuska /* 263564287048SMartin Matuska * Can this platform run the xz program? 263664287048SMartin Matuska */ 263764287048SMartin Matuska int 263864287048SMartin Matuska canXz(void) 263964287048SMartin Matuska { 264064287048SMartin Matuska static int tested = 0, value = 0; 264164287048SMartin Matuska if (!tested) { 264264287048SMartin Matuska tested = 1; 2643f9762417SMartin Matuska if (systemf("xz --help %s", redirectArgs) == 0) 264464287048SMartin Matuska value = 1; 264564287048SMartin Matuska } 264664287048SMartin Matuska return (value); 264764287048SMartin Matuska } 264864287048SMartin Matuska 264964287048SMartin Matuska /* 265064287048SMartin Matuska * Can this filesystem handle nodump flags. 265164287048SMartin Matuska */ 265264287048SMartin Matuska int 265364287048SMartin Matuska canNodump(void) 265464287048SMartin Matuska { 265564287048SMartin Matuska #if defined(HAVE_STRUCT_STAT_ST_FLAGS) && defined(UF_NODUMP) 265664287048SMartin Matuska const char *path = "cannodumptest"; 265764287048SMartin Matuska struct stat sb; 265864287048SMartin Matuska 265964287048SMartin Matuska assertion_make_file(__FILE__, __LINE__, path, 0644, 0, NULL); 266064287048SMartin Matuska if (chflags(path, UF_NODUMP) < 0) 266164287048SMartin Matuska return (0); 266264287048SMartin Matuska if (stat(path, &sb) < 0) 266364287048SMartin Matuska return (0); 266464287048SMartin Matuska if (sb.st_flags & UF_NODUMP) 266564287048SMartin Matuska return (1); 266664287048SMartin Matuska #elif (defined(FS_IOC_GETFLAGS) && defined(HAVE_WORKING_FS_IOC_GETFLAGS) \ 266764287048SMartin Matuska && defined(FS_NODUMP_FL)) || \ 266864287048SMartin Matuska (defined(EXT2_IOC_GETFLAGS) && defined(HAVE_WORKING_EXT2_IOC_GETFLAGS) \ 266964287048SMartin Matuska && defined(EXT2_NODUMP_FL)) 267064287048SMartin Matuska const char *path = "cannodumptest"; 267164287048SMartin Matuska int fd, r, flags; 267264287048SMartin Matuska 267364287048SMartin Matuska assertion_make_file(__FILE__, __LINE__, path, 0644, 0, NULL); 267464287048SMartin Matuska fd = open(path, O_RDONLY | O_NONBLOCK); 267564287048SMartin Matuska if (fd < 0) 267664287048SMartin Matuska return (0); 267764287048SMartin Matuska r = ioctl(fd, 267864287048SMartin Matuska #ifdef FS_IOC_GETFLAGS 267964287048SMartin Matuska FS_IOC_GETFLAGS, 268064287048SMartin Matuska #else 268164287048SMartin Matuska EXT2_IOC_GETFLAGS, 268264287048SMartin Matuska #endif 268364287048SMartin Matuska &flags); 268464287048SMartin Matuska if (r < 0) 268564287048SMartin Matuska return (0); 268664287048SMartin Matuska #ifdef FS_NODUMP_FL 268764287048SMartin Matuska flags |= FS_NODUMP_FL; 268864287048SMartin Matuska #else 268964287048SMartin Matuska flags |= EXT2_NODUMP_FL; 269064287048SMartin Matuska #endif 269164287048SMartin Matuska r = ioctl(fd, 269264287048SMartin Matuska #ifdef FS_IOC_SETFLAGS 269364287048SMartin Matuska FS_IOC_SETFLAGS, 269464287048SMartin Matuska #else 269564287048SMartin Matuska EXT2_IOC_SETFLAGS, 269664287048SMartin Matuska #endif 269764287048SMartin Matuska &flags); 269864287048SMartin Matuska if (r < 0) 269964287048SMartin Matuska return (0); 270064287048SMartin Matuska close(fd); 270164287048SMartin Matuska fd = open(path, O_RDONLY | O_NONBLOCK); 270264287048SMartin Matuska if (fd < 0) 270364287048SMartin Matuska return (0); 270464287048SMartin Matuska r = ioctl(fd, 270564287048SMartin Matuska #ifdef FS_IOC_GETFLAGS 270664287048SMartin Matuska FS_IOC_GETFLAGS, 270764287048SMartin Matuska #else 270864287048SMartin Matuska EXT2_IOC_GETFLAGS, 270964287048SMartin Matuska #endif 271064287048SMartin Matuska &flags); 271164287048SMartin Matuska if (r < 0) 271264287048SMartin Matuska return (0); 271364287048SMartin Matuska close(fd); 271464287048SMartin Matuska #ifdef FS_NODUMP_FL 271564287048SMartin Matuska if (flags & FS_NODUMP_FL) 271664287048SMartin Matuska #else 271764287048SMartin Matuska if (flags & EXT2_NODUMP_FL) 271864287048SMartin Matuska #endif 271964287048SMartin Matuska return (1); 272064287048SMartin Matuska #endif 272164287048SMartin Matuska return (0); 272264287048SMartin Matuska } 272364287048SMartin Matuska 272443f9e382SMartin Matuska /* Get extended attribute value from a path */ 272543f9e382SMartin Matuska void * 2726e46d4714SMartin Matuska getXattr(const char *path, const char *name, size_t *sizep) 2727e46d4714SMartin Matuska { 2728e46d4714SMartin Matuska void *value = NULL; 2729e46d4714SMartin Matuska #if ARCHIVE_XATTR_SUPPORT 2730e46d4714SMartin Matuska ssize_t size; 2731e46d4714SMartin Matuska #if ARCHIVE_XATTR_LINUX 2732e46d4714SMartin Matuska size = lgetxattr(path, name, NULL, 0); 2733e46d4714SMartin Matuska #elif ARCHIVE_XATTR_DARWIN 2734e46d4714SMartin Matuska size = getxattr(path, name, NULL, 0, 0, XATTR_NOFOLLOW); 2735e46d4714SMartin Matuska #elif ARCHIVE_XATTR_AIX 2736e46d4714SMartin Matuska size = lgetea(path, name, NULL, 0); 2737e46d4714SMartin Matuska #elif ARCHIVE_XATTR_FREEBSD 2738e46d4714SMartin Matuska size = extattr_get_link(path, EXTATTR_NAMESPACE_USER, name + 5, 2739e46d4714SMartin Matuska NULL, 0); 2740e46d4714SMartin Matuska #endif 2741e46d4714SMartin Matuska 2742e46d4714SMartin Matuska if (size >= 0) { 2743e46d4714SMartin Matuska value = malloc(size); 2744e46d4714SMartin Matuska #if ARCHIVE_XATTR_LINUX 2745e46d4714SMartin Matuska size = lgetxattr(path, name, value, size); 2746e46d4714SMartin Matuska #elif ARCHIVE_XATTR_DARWIN 2747e46d4714SMartin Matuska size = getxattr(path, name, value, size, 0, XATTR_NOFOLLOW); 2748e46d4714SMartin Matuska #elif ARCHIVE_XATTR_AIX 2749e46d4714SMartin Matuska size = lgetea(path, name, value, size); 2750e46d4714SMartin Matuska #elif ARCHIVE_XATTR_FREEBSD 2751e46d4714SMartin Matuska size = extattr_get_link(path, EXTATTR_NAMESPACE_USER, name + 5, 2752e46d4714SMartin Matuska value, size); 2753e46d4714SMartin Matuska #endif 2754e46d4714SMartin Matuska if (size < 0) { 2755e46d4714SMartin Matuska free(value); 2756e46d4714SMartin Matuska value = NULL; 2757e46d4714SMartin Matuska } 2758e46d4714SMartin Matuska } 2759e46d4714SMartin Matuska if (size < 0) 2760e46d4714SMartin Matuska *sizep = 0; 2761e46d4714SMartin Matuska else 2762e46d4714SMartin Matuska *sizep = (size_t)size; 2763e46d4714SMartin Matuska #else /* !ARCHIVE_XATTR_SUPPORT */ 2764e46d4714SMartin Matuska (void)path; /* UNUSED */ 2765e46d4714SMartin Matuska (void)name; /* UNUSED */ 2766e46d4714SMartin Matuska *sizep = 0; 2767e46d4714SMartin Matuska #endif /* !ARCHIVE_XATTR_SUPPORT */ 2768e46d4714SMartin Matuska return (value); 2769e46d4714SMartin Matuska } 2770e46d4714SMartin Matuska 2771e46d4714SMartin Matuska /* 2772e46d4714SMartin Matuska * Set extended attribute on a path 2773e46d4714SMartin Matuska * Returns 0 on error, 1 on success 2774e46d4714SMartin Matuska */ 2775e46d4714SMartin Matuska int 2776e46d4714SMartin Matuska setXattr(const char *path, const char *name, const void *value, size_t size) 2777e46d4714SMartin Matuska { 2778e46d4714SMartin Matuska #if ARCHIVE_XATTR_SUPPORT 2779e46d4714SMartin Matuska #if ARCHIVE_XATTR_LINUX 2780e46d4714SMartin Matuska if (lsetxattr(path, name, value, size, 0) == 0) 2781e46d4714SMartin Matuska #elif ARCHIVE_XATTR_DARWIN 2782e46d4714SMartin Matuska if (setxattr(path, name, value, size, 0, XATTR_NOFOLLOW) == 0) 2783e46d4714SMartin Matuska #elif ARCHIVE_XATTR_AIX 2784e46d4714SMartin Matuska if (lsetea(path, name, value, size, 0) == 0) 2785e46d4714SMartin Matuska #elif ARCHIVE_XATTR_FREEBSD 2786e46d4714SMartin Matuska if (extattr_set_link(path, EXTATTR_NAMESPACE_USER, name + 5, value, 2787e46d4714SMartin Matuska size) > -1) 2788e46d4714SMartin Matuska #else 2789e46d4714SMartin Matuska if (0) 2790e46d4714SMartin Matuska #endif 2791e46d4714SMartin Matuska return (1); 2792e46d4714SMartin Matuska #else /* !ARCHIVE_XATTR_SUPPORT */ 2793e46d4714SMartin Matuska (void)path; /* UNUSED */ 2794e46d4714SMartin Matuska (void)name; /* UNUSED */ 2795e46d4714SMartin Matuska (void)value; /* UNUSED */ 2796e46d4714SMartin Matuska (void)size; /* UNUSED */ 2797e46d4714SMartin Matuska #endif /* !ARCHIVE_XATTR_SUPPORT */ 2798e46d4714SMartin Matuska return (0); 2799e46d4714SMartin Matuska } 2800e46d4714SMartin Matuska 28014657548dSMartin Matuska #if ARCHIVE_ACL_SUNOS 280264287048SMartin Matuska /* Fetch ACLs on Solaris using acl() or facl() */ 280364287048SMartin Matuska void * 280464287048SMartin Matuska sunacl_get(int cmd, int *aclcnt, int fd, const char *path) 280564287048SMartin Matuska { 280664287048SMartin Matuska int cnt, cntcmd; 280764287048SMartin Matuska size_t size; 280864287048SMartin Matuska void *aclp; 280964287048SMartin Matuska 281064287048SMartin Matuska if (cmd == GETACL) { 281164287048SMartin Matuska cntcmd = GETACLCNT; 281264287048SMartin Matuska size = sizeof(aclent_t); 281364287048SMartin Matuska } 28144657548dSMartin Matuska #if ARCHIVE_ACL_SUNOS_NFS4 281564287048SMartin Matuska else if (cmd == ACE_GETACL) { 281664287048SMartin Matuska cntcmd = ACE_GETACLCNT; 281764287048SMartin Matuska size = sizeof(ace_t); 281864287048SMartin Matuska } 281964287048SMartin Matuska #endif 282064287048SMartin Matuska else { 282164287048SMartin Matuska errno = EINVAL; 282264287048SMartin Matuska *aclcnt = -1; 282364287048SMartin Matuska return (NULL); 282464287048SMartin Matuska } 282564287048SMartin Matuska 282664287048SMartin Matuska aclp = NULL; 282764287048SMartin Matuska cnt = -2; 282864287048SMartin Matuska while (cnt == -2 || (cnt == -1 && errno == ENOSPC)) { 282964287048SMartin Matuska if (path != NULL) 283064287048SMartin Matuska cnt = acl(path, cntcmd, 0, NULL); 283164287048SMartin Matuska else 283264287048SMartin Matuska cnt = facl(fd, cntcmd, 0, NULL); 283364287048SMartin Matuska 283464287048SMartin Matuska if (cnt > 0) { 283564287048SMartin Matuska if (aclp == NULL) 283664287048SMartin Matuska aclp = malloc(cnt * size); 283764287048SMartin Matuska else 283864287048SMartin Matuska aclp = realloc(NULL, cnt * size); 283964287048SMartin Matuska if (aclp != NULL) { 284064287048SMartin Matuska if (path != NULL) 284164287048SMartin Matuska cnt = acl(path, cmd, cnt, aclp); 284264287048SMartin Matuska else 284364287048SMartin Matuska cnt = facl(fd, cmd, cnt, aclp); 284464287048SMartin Matuska } 284564287048SMartin Matuska } else { 284664287048SMartin Matuska free(aclp); 284764287048SMartin Matuska aclp = NULL; 284864287048SMartin Matuska break; 284964287048SMartin Matuska } 285064287048SMartin Matuska } 285164287048SMartin Matuska 285264287048SMartin Matuska *aclcnt = cnt; 285364287048SMartin Matuska return (aclp); 285464287048SMartin Matuska } 28554657548dSMartin Matuska #endif /* ARCHIVE_ACL_SUNOS */ 285664287048SMartin Matuska 285764287048SMartin Matuska /* 285864287048SMartin Matuska * Set test ACLs on a path 285964287048SMartin Matuska * Return values: 286064287048SMartin Matuska * 0: error setting ACLs 286164287048SMartin Matuska * ARCHIVE_TEST_ACL_TYPE_POSIX1E: POSIX.1E ACLs have been set 286264287048SMartin Matuska * ARCHIVE_TEST_ACL_TYPE_NFS4: NFSv4 or extended ACLs have been set 286364287048SMartin Matuska */ 286464287048SMartin Matuska int 286564287048SMartin Matuska setTestAcl(const char *path) 286664287048SMartin Matuska { 28674657548dSMartin Matuska #if ARCHIVE_ACL_SUPPORT 286864287048SMartin Matuska int r = 1; 28694657548dSMartin Matuska #if ARCHIVE_ACL_LIBACL || ARCHIVE_ACL_FREEBSD || ARCHIVE_ACL_DARWIN 287064287048SMartin Matuska acl_t acl; 287164287048SMartin Matuska #endif 28724657548dSMartin Matuska #if ARCHIVE_ACL_LIBRICHACL 28734657548dSMartin Matuska struct richacl *richacl; 28744657548dSMartin Matuska #endif 28754657548dSMartin Matuska #if ARCHIVE_ACL_LIBACL || ARCHIVE_ACL_FREEBSD 287664287048SMartin Matuska const char *acltext_posix1e = "user:1:rw-," 287764287048SMartin Matuska "group:15:r-x," 287864287048SMartin Matuska "user::rwx," 287964287048SMartin Matuska "group::rwx," 288064287048SMartin Matuska "other::r-x," 288164287048SMartin Matuska "mask::rwx"; 28824657548dSMartin Matuska #elif ARCHIVE_ACL_SUNOS /* Solaris POSIX.1e */ 288364287048SMartin Matuska aclent_t aclp_posix1e[] = { 288464287048SMartin Matuska { USER_OBJ, -1, 4 | 2 | 1 }, 288564287048SMartin Matuska { USER, 1, 4 | 2 }, 288664287048SMartin Matuska { GROUP_OBJ, -1, 4 | 2 | 1 }, 288764287048SMartin Matuska { GROUP, 15, 4 | 1 }, 288864287048SMartin Matuska { CLASS_OBJ, -1, 4 | 2 | 1 }, 288964287048SMartin Matuska { OTHER_OBJ, -1, 4 | 2 | 1 } 289064287048SMartin Matuska }; 289164287048SMartin Matuska #endif 28924657548dSMartin Matuska #if ARCHIVE_ACL_FREEBSD /* FreeBSD NFS4 */ 289364287048SMartin Matuska const char *acltext_nfs4 = "user:1:rwpaRcs::allow:1," 289464287048SMartin Matuska "group:15:rxaRcs::allow:15," 289564287048SMartin Matuska "owner@:rwpxaARWcCos::allow," 289664287048SMartin Matuska "group@:rwpxaRcs::allow," 289764287048SMartin Matuska "everyone@:rxaRcs::allow"; 28984657548dSMartin Matuska #elif ARCHIVE_ACL_LIBRICHACL 28994657548dSMartin Matuska const char *acltext_nfs4 = "owner:rwpxaARWcCoS::mask," 29004657548dSMartin Matuska "group:rwpxaRcS::mask," 29014657548dSMartin Matuska "other:rxaRcS::mask," 29024657548dSMartin Matuska "user:1:rwpaRcS::allow," 29034657548dSMartin Matuska "group:15:rxaRcS::allow," 29044657548dSMartin Matuska "owner@:rwpxaARWcCoS::allow," 29054657548dSMartin Matuska "group@:rwpxaRcS::allow," 29064657548dSMartin Matuska "everyone@:rxaRcS::allow"; 29074657548dSMartin Matuska #elif ARCHIVE_ACL_SUNOS_NFS4 /* Solaris NFS4 */ 290864287048SMartin Matuska ace_t aclp_nfs4[] = { 290964287048SMartin Matuska { 1, ACE_READ_DATA | ACE_WRITE_DATA | ACE_APPEND_DATA | 291064287048SMartin Matuska ACE_READ_ATTRIBUTES | ACE_READ_NAMED_ATTRS | ACE_READ_ACL | 291164287048SMartin Matuska ACE_SYNCHRONIZE, 0, ACE_ACCESS_ALLOWED_ACE_TYPE }, 291264287048SMartin Matuska { 15, ACE_READ_DATA | ACE_EXECUTE | ACE_READ_ATTRIBUTES | 291364287048SMartin Matuska ACE_READ_NAMED_ATTRS | ACE_READ_ACL | ACE_SYNCHRONIZE, 291464287048SMartin Matuska ACE_IDENTIFIER_GROUP, ACE_ACCESS_ALLOWED_ACE_TYPE }, 291564287048SMartin Matuska { -1, ACE_READ_DATA | ACE_WRITE_DATA | ACE_APPEND_DATA | 291664287048SMartin Matuska ACE_EXECUTE | ACE_READ_ATTRIBUTES | ACE_WRITE_ATTRIBUTES | 291764287048SMartin Matuska ACE_READ_NAMED_ATTRS | ACE_WRITE_NAMED_ATTRS | 291864287048SMartin Matuska ACE_READ_ACL | ACE_WRITE_ACL | ACE_WRITE_OWNER | ACE_SYNCHRONIZE, 291964287048SMartin Matuska ACE_OWNER, ACE_ACCESS_ALLOWED_ACE_TYPE }, 292064287048SMartin Matuska { -1, ACE_READ_DATA | ACE_WRITE_DATA | ACE_APPEND_DATA | 292164287048SMartin Matuska ACE_EXECUTE | ACE_READ_ATTRIBUTES | ACE_READ_NAMED_ATTRS | 292264287048SMartin Matuska ACE_READ_ACL | ACE_SYNCHRONIZE, ACE_GROUP | ACE_IDENTIFIER_GROUP, 292364287048SMartin Matuska ACE_ACCESS_ALLOWED_ACE_TYPE }, 292464287048SMartin Matuska { -1, ACE_READ_DATA | ACE_EXECUTE | ACE_READ_ATTRIBUTES | 292564287048SMartin Matuska ACE_READ_NAMED_ATTRS | ACE_READ_ACL | ACE_SYNCHRONIZE, 292664287048SMartin Matuska ACE_EVERYONE, ACE_ACCESS_ALLOWED_ACE_TYPE } 292764287048SMartin Matuska }; 29284657548dSMartin Matuska #elif ARCHIVE_ACL_DARWIN /* Mac OS X */ 292964287048SMartin Matuska acl_entry_t aclent; 293064287048SMartin Matuska acl_permset_t permset; 293164287048SMartin Matuska const uid_t uid = 1; 293264287048SMartin Matuska uuid_t uuid; 293364287048SMartin Matuska int i; 293464287048SMartin Matuska const acl_perm_t acl_perms[] = { 293564287048SMartin Matuska ACL_READ_DATA, 293664287048SMartin Matuska ACL_WRITE_DATA, 293764287048SMartin Matuska ACL_APPEND_DATA, 293864287048SMartin Matuska ACL_EXECUTE, 293964287048SMartin Matuska ACL_READ_ATTRIBUTES, 294064287048SMartin Matuska ACL_READ_EXTATTRIBUTES, 294164287048SMartin Matuska ACL_READ_SECURITY, 294264287048SMartin Matuska #if HAVE_DECL_ACL_SYNCHRONIZE 294364287048SMartin Matuska ACL_SYNCHRONIZE 294464287048SMartin Matuska #endif 294564287048SMartin Matuska }; 29464657548dSMartin Matuska #endif /* ARCHIVE_ACL_DARWIN */ 294764287048SMartin Matuska 29484657548dSMartin Matuska #if ARCHIVE_ACL_FREEBSD 294964287048SMartin Matuska acl = acl_from_text(acltext_nfs4); 295064287048SMartin Matuska failure("acl_from_text() error: %s", strerror(errno)); 295164287048SMartin Matuska if (assert(acl != NULL) == 0) 295264287048SMartin Matuska return (0); 29534657548dSMartin Matuska #elif ARCHIVE_ACL_LIBRICHACL 29544657548dSMartin Matuska richacl = richacl_from_text(acltext_nfs4, NULL, NULL); 29554657548dSMartin Matuska failure("richacl_from_text() error: %s", strerror(errno)); 29564657548dSMartin Matuska if (assert(richacl != NULL) == 0) 29574657548dSMartin Matuska return (0); 29584657548dSMartin Matuska #elif ARCHIVE_ACL_DARWIN 295964287048SMartin Matuska acl = acl_init(1); 296064287048SMartin Matuska failure("acl_init() error: %s", strerror(errno)); 296164287048SMartin Matuska if (assert(acl != NULL) == 0) 296264287048SMartin Matuska return (0); 296364287048SMartin Matuska r = acl_create_entry(&acl, &aclent); 296464287048SMartin Matuska failure("acl_create_entry() error: %s", strerror(errno)); 296564287048SMartin Matuska if (assertEqualInt(r, 0) == 0) 296664287048SMartin Matuska goto testacl_free; 296764287048SMartin Matuska r = acl_set_tag_type(aclent, ACL_EXTENDED_ALLOW); 296864287048SMartin Matuska failure("acl_set_tag_type() error: %s", strerror(errno)); 296964287048SMartin Matuska if (assertEqualInt(r, 0) == 0) 297064287048SMartin Matuska goto testacl_free; 297164287048SMartin Matuska r = acl_get_permset(aclent, &permset); 297264287048SMartin Matuska failure("acl_get_permset() error: %s", strerror(errno)); 297364287048SMartin Matuska if (assertEqualInt(r, 0) == 0) 297464287048SMartin Matuska goto testacl_free; 297564287048SMartin Matuska for (i = 0; i < (int)(sizeof(acl_perms) / sizeof(acl_perms[0])); i++) { 297664287048SMartin Matuska r = acl_add_perm(permset, acl_perms[i]); 297764287048SMartin Matuska failure("acl_add_perm() error: %s", strerror(errno)); 297864287048SMartin Matuska if (assertEqualInt(r, 0) == 0) 297964287048SMartin Matuska goto testacl_free; 298064287048SMartin Matuska } 298164287048SMartin Matuska r = acl_set_permset(aclent, permset); 298264287048SMartin Matuska failure("acl_set_permset() error: %s", strerror(errno)); 298364287048SMartin Matuska if (assertEqualInt(r, 0) == 0) 298464287048SMartin Matuska goto testacl_free; 29854657548dSMartin Matuska r = mbr_uid_to_uuid(uid, uuid); 29864657548dSMartin Matuska failure("mbr_uid_to_uuid() error: %s", strerror(errno)); 298764287048SMartin Matuska if (assertEqualInt(r, 0) == 0) 298864287048SMartin Matuska goto testacl_free; 298964287048SMartin Matuska r = acl_set_qualifier(aclent, uuid); 299064287048SMartin Matuska failure("acl_set_qualifier() error: %s", strerror(errno)); 299164287048SMartin Matuska if (assertEqualInt(r, 0) == 0) 299264287048SMartin Matuska goto testacl_free; 29934657548dSMartin Matuska #endif /* ARCHIVE_ACL_DARWIN */ 299464287048SMartin Matuska 29954657548dSMartin Matuska #if ARCHIVE_ACL_NFS4 29964657548dSMartin Matuska #if ARCHIVE_ACL_FREEBSD 299764287048SMartin Matuska r = acl_set_file(path, ACL_TYPE_NFS4, acl); 299864287048SMartin Matuska acl_free(acl); 29994657548dSMartin Matuska #elif ARCHIVE_ACL_LIBRICHACL 30004657548dSMartin Matuska r = richacl_set_file(path, richacl); 30014657548dSMartin Matuska richacl_free(richacl); 30024657548dSMartin Matuska #elif ARCHIVE_ACL_SUNOS_NFS4 300364287048SMartin Matuska r = acl(path, ACE_SETACL, 300464287048SMartin Matuska (int)(sizeof(aclp_nfs4)/sizeof(aclp_nfs4[0])), aclp_nfs4); 30054657548dSMartin Matuska #elif ARCHIVE_ACL_DARWIN 300664287048SMartin Matuska r = acl_set_file(path, ACL_TYPE_EXTENDED, acl); 300764287048SMartin Matuska acl_free(acl); 300864287048SMartin Matuska #endif 300964287048SMartin Matuska if (r == 0) 301064287048SMartin Matuska return (ARCHIVE_TEST_ACL_TYPE_NFS4); 30114657548dSMartin Matuska #endif /* ARCHIVE_ACL_NFS4 */ 301264287048SMartin Matuska 30134657548dSMartin Matuska #if ARCHIVE_ACL_POSIX1E 30144657548dSMartin Matuska #if ARCHIVE_ACL_FREEBSD || ARCHIVE_ACL_LIBACL 301564287048SMartin Matuska acl = acl_from_text(acltext_posix1e); 301664287048SMartin Matuska failure("acl_from_text() error: %s", strerror(errno)); 301764287048SMartin Matuska if (assert(acl != NULL) == 0) 301864287048SMartin Matuska return (0); 301964287048SMartin Matuska 302064287048SMartin Matuska r = acl_set_file(path, ACL_TYPE_ACCESS, acl); 302164287048SMartin Matuska acl_free(acl); 30224657548dSMartin Matuska #elif ARCHIVE_ACL_SUNOS 302364287048SMartin Matuska r = acl(path, SETACL, 302464287048SMartin Matuska (int)(sizeof(aclp_posix1e)/sizeof(aclp_posix1e[0])), aclp_posix1e); 302564287048SMartin Matuska #endif 302664287048SMartin Matuska if (r == 0) 302764287048SMartin Matuska return (ARCHIVE_TEST_ACL_TYPE_POSIX1E); 302864287048SMartin Matuska else 302964287048SMartin Matuska return (0); 30304657548dSMartin Matuska #endif /* ARCHIVE_ACL_POSIX1E */ 30314657548dSMartin Matuska #if ARCHIVE_ACL_DARWIN 303264287048SMartin Matuska testacl_free: 303364287048SMartin Matuska acl_free(acl); 303464287048SMartin Matuska #endif 30354657548dSMartin Matuska #endif /* ARCHIVE_ACL_SUPPORT */ 303664287048SMartin Matuska (void)path; /* UNUSED */ 303764287048SMartin Matuska return (0); 303864287048SMartin Matuska } 303964287048SMartin Matuska 304064287048SMartin Matuska /* 304164287048SMartin Matuska * Sleep as needed; useful for verifying disk timestamp changes by 304264287048SMartin Matuska * ensuring that the wall-clock time has actually changed before we 304364287048SMartin Matuska * go back to re-read something from disk. 304464287048SMartin Matuska */ 304564287048SMartin Matuska void 304664287048SMartin Matuska sleepUntilAfter(time_t t) 304764287048SMartin Matuska { 304864287048SMartin Matuska while (t >= time(NULL)) 304964287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 305064287048SMartin Matuska Sleep(500); 305164287048SMartin Matuska #else 305264287048SMartin Matuska sleep(1); 305364287048SMartin Matuska #endif 305464287048SMartin Matuska } 305564287048SMartin Matuska 305664287048SMartin Matuska /* 305764287048SMartin Matuska * Call standard system() call, but build up the command line using 305864287048SMartin Matuska * sprintf() conventions. 305964287048SMartin Matuska */ 306064287048SMartin Matuska int 306164287048SMartin Matuska systemf(const char *fmt, ...) 306264287048SMartin Matuska { 306364287048SMartin Matuska char buff[8192]; 306464287048SMartin Matuska va_list ap; 306564287048SMartin Matuska int r; 306664287048SMartin Matuska 306764287048SMartin Matuska va_start(ap, fmt); 306864287048SMartin Matuska vsprintf(buff, fmt, ap); 306964287048SMartin Matuska if (verbosity > VERBOSITY_FULL) 307064287048SMartin Matuska logprintf("Cmd: %s\n", buff); 307164287048SMartin Matuska r = system(buff); 307264287048SMartin Matuska va_end(ap); 307364287048SMartin Matuska return (r); 307464287048SMartin Matuska } 307564287048SMartin Matuska 307664287048SMartin Matuska /* 307764287048SMartin Matuska * Slurp a file into memory for ease of comparison and testing. 307864287048SMartin Matuska * Returns size of file in 'sizep' if non-NULL, null-terminates 307964287048SMartin Matuska * data in memory for ease of use. 308064287048SMartin Matuska */ 308164287048SMartin Matuska char * 308264287048SMartin Matuska slurpfile(size_t * sizep, const char *fmt, ...) 308364287048SMartin Matuska { 308464287048SMartin Matuska char filename[8192]; 308564287048SMartin Matuska struct stat st; 308664287048SMartin Matuska va_list ap; 308764287048SMartin Matuska char *p; 308864287048SMartin Matuska ssize_t bytes_read; 308964287048SMartin Matuska FILE *f; 309064287048SMartin Matuska int r; 309164287048SMartin Matuska 309264287048SMartin Matuska va_start(ap, fmt); 309364287048SMartin Matuska vsprintf(filename, fmt, ap); 309464287048SMartin Matuska va_end(ap); 309564287048SMartin Matuska 309664287048SMartin Matuska f = fopen(filename, "rb"); 309764287048SMartin Matuska if (f == NULL) { 309864287048SMartin Matuska /* Note: No error; non-existent file is okay here. */ 309964287048SMartin Matuska return (NULL); 310064287048SMartin Matuska } 310164287048SMartin Matuska r = fstat(fileno(f), &st); 310264287048SMartin Matuska if (r != 0) { 310364287048SMartin Matuska logprintf("Can't stat file %s\n", filename); 310464287048SMartin Matuska fclose(f); 310564287048SMartin Matuska return (NULL); 310664287048SMartin Matuska } 310764287048SMartin Matuska p = malloc((size_t)st.st_size + 1); 310864287048SMartin Matuska if (p == NULL) { 310964287048SMartin Matuska logprintf("Can't allocate %ld bytes of memory to read file %s\n", 311064287048SMartin Matuska (long int)st.st_size, filename); 311164287048SMartin Matuska fclose(f); 311264287048SMartin Matuska return (NULL); 311364287048SMartin Matuska } 311464287048SMartin Matuska bytes_read = fread(p, 1, (size_t)st.st_size, f); 311564287048SMartin Matuska if (bytes_read < st.st_size) { 311664287048SMartin Matuska logprintf("Can't read file %s\n", filename); 311764287048SMartin Matuska fclose(f); 311864287048SMartin Matuska free(p); 311964287048SMartin Matuska return (NULL); 312064287048SMartin Matuska } 312164287048SMartin Matuska p[st.st_size] = '\0'; 312264287048SMartin Matuska if (sizep != NULL) 312364287048SMartin Matuska *sizep = (size_t)st.st_size; 312464287048SMartin Matuska fclose(f); 312564287048SMartin Matuska return (p); 312664287048SMartin Matuska } 312764287048SMartin Matuska 312864287048SMartin Matuska /* 312964287048SMartin Matuska * Slurp a file into memory for ease of comparison and testing. 313064287048SMartin Matuska * Returns size of file in 'sizep' if non-NULL, null-terminates 313164287048SMartin Matuska * data in memory for ease of use. 313264287048SMartin Matuska */ 313364287048SMartin Matuska void 313464287048SMartin Matuska dumpfile(const char *filename, void *data, size_t len) 313564287048SMartin Matuska { 313664287048SMartin Matuska ssize_t bytes_written; 313764287048SMartin Matuska FILE *f; 313864287048SMartin Matuska 313964287048SMartin Matuska f = fopen(filename, "wb"); 314064287048SMartin Matuska if (f == NULL) { 314164287048SMartin Matuska logprintf("Can't open file %s for writing\n", filename); 314264287048SMartin Matuska return; 314364287048SMartin Matuska } 314464287048SMartin Matuska bytes_written = fwrite(data, 1, len, f); 314564287048SMartin Matuska if (bytes_written < (ssize_t)len) 314664287048SMartin Matuska logprintf("Can't write file %s\n", filename); 314764287048SMartin Matuska fclose(f); 314864287048SMartin Matuska } 314964287048SMartin Matuska 315064287048SMartin Matuska /* Read a uuencoded file from the reference directory, decode, and 315164287048SMartin Matuska * write the result into the current directory. */ 315264287048SMartin Matuska #define VALID_UUDECODE(c) (c >= 32 && c <= 96) 315364287048SMartin Matuska #define UUDECODE(c) (((c) - 0x20) & 0x3f) 315464287048SMartin Matuska void 315564287048SMartin Matuska extract_reference_file(const char *name) 315664287048SMartin Matuska { 315764287048SMartin Matuska char buff[1024]; 315864287048SMartin Matuska FILE *in, *out; 315964287048SMartin Matuska 316064287048SMartin Matuska sprintf(buff, "%s/%s.uu", refdir, name); 316164287048SMartin Matuska in = fopen(buff, "r"); 316264287048SMartin Matuska failure("Couldn't open reference file %s", buff); 316364287048SMartin Matuska assert(in != NULL); 316464287048SMartin Matuska if (in == NULL) 316564287048SMartin Matuska return; 316664287048SMartin Matuska /* Read up to and including the 'begin' line. */ 316764287048SMartin Matuska for (;;) { 316864287048SMartin Matuska if (fgets(buff, sizeof(buff), in) == NULL) { 316964287048SMartin Matuska /* TODO: This is a failure. */ 317064287048SMartin Matuska return; 317164287048SMartin Matuska } 317264287048SMartin Matuska if (memcmp(buff, "begin ", 6) == 0) 317364287048SMartin Matuska break; 317464287048SMartin Matuska } 317564287048SMartin Matuska /* Now, decode the rest and write it. */ 317664287048SMartin Matuska out = fopen(name, "wb"); 317764287048SMartin Matuska while (fgets(buff, sizeof(buff), in) != NULL) { 317864287048SMartin Matuska char *p = buff; 317964287048SMartin Matuska int bytes; 318064287048SMartin Matuska 318164287048SMartin Matuska if (memcmp(buff, "end", 3) == 0) 318264287048SMartin Matuska break; 318364287048SMartin Matuska 318464287048SMartin Matuska bytes = UUDECODE(*p++); 318564287048SMartin Matuska while (bytes > 0) { 318664287048SMartin Matuska int n = 0; 318764287048SMartin Matuska /* Write out 1-3 bytes from that. */ 318864287048SMartin Matuska if (bytes > 0) { 318964287048SMartin Matuska assert(VALID_UUDECODE(p[0])); 319064287048SMartin Matuska assert(VALID_UUDECODE(p[1])); 319164287048SMartin Matuska n = UUDECODE(*p++) << 18; 319264287048SMartin Matuska n |= UUDECODE(*p++) << 12; 319364287048SMartin Matuska fputc(n >> 16, out); 319464287048SMartin Matuska --bytes; 319564287048SMartin Matuska } 319664287048SMartin Matuska if (bytes > 0) { 319764287048SMartin Matuska assert(VALID_UUDECODE(p[0])); 319864287048SMartin Matuska n |= UUDECODE(*p++) << 6; 319964287048SMartin Matuska fputc((n >> 8) & 0xFF, out); 320064287048SMartin Matuska --bytes; 320164287048SMartin Matuska } 320264287048SMartin Matuska if (bytes > 0) { 320364287048SMartin Matuska assert(VALID_UUDECODE(p[0])); 320464287048SMartin Matuska n |= UUDECODE(*p++); 320564287048SMartin Matuska fputc(n & 0xFF, out); 320664287048SMartin Matuska --bytes; 320764287048SMartin Matuska } 320864287048SMartin Matuska } 320964287048SMartin Matuska } 321064287048SMartin Matuska fclose(out); 321164287048SMartin Matuska fclose(in); 321264287048SMartin Matuska } 321364287048SMartin Matuska 321464287048SMartin Matuska void 321564287048SMartin Matuska copy_reference_file(const char *name) 321664287048SMartin Matuska { 321764287048SMartin Matuska char buff[1024]; 321864287048SMartin Matuska FILE *in, *out; 321964287048SMartin Matuska size_t rbytes; 322064287048SMartin Matuska 322164287048SMartin Matuska sprintf(buff, "%s/%s", refdir, name); 322264287048SMartin Matuska in = fopen(buff, "rb"); 322364287048SMartin Matuska failure("Couldn't open reference file %s", buff); 322464287048SMartin Matuska assert(in != NULL); 322564287048SMartin Matuska if (in == NULL) 322664287048SMartin Matuska return; 322764287048SMartin Matuska /* Now, decode the rest and write it. */ 322864287048SMartin Matuska /* Not a lot of error checking here; the input better be right. */ 322964287048SMartin Matuska out = fopen(name, "wb"); 323064287048SMartin Matuska while ((rbytes = fread(buff, 1, sizeof(buff), in)) > 0) { 323164287048SMartin Matuska if (fwrite(buff, 1, rbytes, out) != rbytes) { 323264287048SMartin Matuska logprintf("Error: fwrite\n"); 323364287048SMartin Matuska break; 323464287048SMartin Matuska } 323564287048SMartin Matuska } 323664287048SMartin Matuska fclose(out); 323764287048SMartin Matuska fclose(in); 323864287048SMartin Matuska } 323964287048SMartin Matuska 324064287048SMartin Matuska int 324164287048SMartin Matuska is_LargeInode(const char *file) 324264287048SMartin Matuska { 324364287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 324464287048SMartin Matuska BY_HANDLE_FILE_INFORMATION bhfi; 324564287048SMartin Matuska int r; 324664287048SMartin Matuska 324764287048SMartin Matuska r = my_GetFileInformationByName(file, &bhfi); 324864287048SMartin Matuska if (r != 0) 324964287048SMartin Matuska return (0); 325064287048SMartin Matuska return (bhfi.nFileIndexHigh & 0x0000FFFFUL); 325164287048SMartin Matuska #else 325264287048SMartin Matuska struct stat st; 325364287048SMartin Matuska int64_t ino; 325464287048SMartin Matuska 325564287048SMartin Matuska if (stat(file, &st) < 0) 325664287048SMartin Matuska return (0); 325764287048SMartin Matuska ino = (int64_t)st.st_ino; 325864287048SMartin Matuska return (ino > 0xffffffff); 325964287048SMartin Matuska #endif 326064287048SMartin Matuska } 326164287048SMartin Matuska 326264287048SMartin Matuska void 326364287048SMartin Matuska extract_reference_files(const char **names) 326464287048SMartin Matuska { 326564287048SMartin Matuska while (names && *names) 326664287048SMartin Matuska extract_reference_file(*names++); 326764287048SMartin Matuska } 326864287048SMartin Matuska 326964287048SMartin Matuska #ifndef PROGRAM 327064287048SMartin Matuska /* Set ACLs */ 327164287048SMartin Matuska int 327264287048SMartin Matuska assertion_entry_set_acls(const char *file, int line, struct archive_entry *ae, 327364287048SMartin Matuska struct archive_test_acl_t *acls, int n) 327464287048SMartin Matuska { 327564287048SMartin Matuska int i, r, ret; 327664287048SMartin Matuska 327764287048SMartin Matuska assertion_count(file, line); 327864287048SMartin Matuska 327964287048SMartin Matuska ret = 0; 328064287048SMartin Matuska archive_entry_acl_clear(ae); 328164287048SMartin Matuska for (i = 0; i < n; i++) { 328264287048SMartin Matuska r = archive_entry_acl_add_entry(ae, 328364287048SMartin Matuska acls[i].type, acls[i].permset, acls[i].tag, 328464287048SMartin Matuska acls[i].qual, acls[i].name); 328564287048SMartin Matuska if (r != 0) { 328664287048SMartin Matuska ret = 1; 3287f9762417SMartin Matuska failure_start(file, line, "type=%#010x, " 328864287048SMartin Matuska "permset=%#010x, tag=%d, qual=%d name=%s", 328964287048SMartin Matuska acls[i].type, acls[i].permset, acls[i].tag, 329064287048SMartin Matuska acls[i].qual, acls[i].name); 329164287048SMartin Matuska failure_finish(NULL); 329264287048SMartin Matuska } 329364287048SMartin Matuska } 329464287048SMartin Matuska 329564287048SMartin Matuska return (ret); 329664287048SMartin Matuska } 329764287048SMartin Matuska 329864287048SMartin Matuska static int 329964287048SMartin Matuska archive_test_acl_match(struct archive_test_acl_t *acl, int type, int permset, 330064287048SMartin Matuska int tag, int qual, const char *name) 330164287048SMartin Matuska { 330264287048SMartin Matuska if (type != acl->type) 330364287048SMartin Matuska return (0); 330464287048SMartin Matuska if (permset != acl->permset) 330564287048SMartin Matuska return (0); 330664287048SMartin Matuska if (tag != acl->tag) 330764287048SMartin Matuska return (0); 330864287048SMartin Matuska if (tag == ARCHIVE_ENTRY_ACL_USER_OBJ) 330964287048SMartin Matuska return (1); 331064287048SMartin Matuska if (tag == ARCHIVE_ENTRY_ACL_GROUP_OBJ) 331164287048SMartin Matuska return (1); 331264287048SMartin Matuska if (tag == ARCHIVE_ENTRY_ACL_EVERYONE) 331364287048SMartin Matuska return (1); 331464287048SMartin Matuska if (tag == ARCHIVE_ENTRY_ACL_OTHER) 331564287048SMartin Matuska return (1); 331664287048SMartin Matuska if (qual != acl->qual) 331764287048SMartin Matuska return (0); 331864287048SMartin Matuska if (name == NULL) { 331964287048SMartin Matuska if (acl->name == NULL || acl->name[0] == '\0') 332064287048SMartin Matuska return (1); 332164287048SMartin Matuska return (0); 332264287048SMartin Matuska } 332364287048SMartin Matuska if (acl->name == NULL) { 332464287048SMartin Matuska if (name[0] == '\0') 332564287048SMartin Matuska return (1); 332664287048SMartin Matuska return (0); 332764287048SMartin Matuska } 332864287048SMartin Matuska return (0 == strcmp(name, acl->name)); 332964287048SMartin Matuska } 333064287048SMartin Matuska 333164287048SMartin Matuska /* Compare ACLs */ 333264287048SMartin Matuska int 333364287048SMartin Matuska assertion_entry_compare_acls(const char *file, int line, 333464287048SMartin Matuska struct archive_entry *ae, struct archive_test_acl_t *acls, int cnt, 333564287048SMartin Matuska int want_type, int mode) 333664287048SMartin Matuska { 333764287048SMartin Matuska int *marker; 333864287048SMartin Matuska int i, r, n, ret; 333964287048SMartin Matuska int type, permset, tag, qual; 334064287048SMartin Matuska int matched; 334164287048SMartin Matuska const char *name; 334264287048SMartin Matuska 334364287048SMartin Matuska assertion_count(file, line); 334464287048SMartin Matuska 334564287048SMartin Matuska ret = 0; 334664287048SMartin Matuska n = 0; 334764287048SMartin Matuska marker = malloc(sizeof(marker[0]) * cnt); 334864287048SMartin Matuska 334964287048SMartin Matuska for (i = 0; i < cnt; i++) { 335064287048SMartin Matuska if ((acls[i].type & want_type) != 0) { 335164287048SMartin Matuska marker[n] = i; 335264287048SMartin Matuska n++; 335364287048SMartin Matuska } 335464287048SMartin Matuska } 335564287048SMartin Matuska 335664287048SMartin Matuska if (n == 0) { 335764287048SMartin Matuska failure_start(file, line, "No ACL's to compare, type mask: %d", 335864287048SMartin Matuska want_type); 335964287048SMartin Matuska return (1); 336064287048SMartin Matuska } 336164287048SMartin Matuska 336264287048SMartin Matuska while (0 == (r = archive_entry_acl_next(ae, want_type, 336364287048SMartin Matuska &type, &permset, &tag, &qual, &name))) { 336464287048SMartin Matuska for (i = 0, matched = 0; i < n && !matched; i++) { 336564287048SMartin Matuska if (archive_test_acl_match(&acls[marker[i]], type, 336664287048SMartin Matuska permset, tag, qual, name)) { 336764287048SMartin Matuska /* We found a match; remove it. */ 336864287048SMartin Matuska marker[i] = marker[n - 1]; 336964287048SMartin Matuska n--; 337064287048SMartin Matuska matched = 1; 337164287048SMartin Matuska } 337264287048SMartin Matuska } 337364287048SMartin Matuska if (type == ARCHIVE_ENTRY_ACL_TYPE_ACCESS 337464287048SMartin Matuska && tag == ARCHIVE_ENTRY_ACL_USER_OBJ) { 337564287048SMartin Matuska if (!matched) { 337664287048SMartin Matuska failure_start(file, line, "No match for " 337764287048SMartin Matuska "user_obj perm"); 337864287048SMartin Matuska failure_finish(NULL); 337964287048SMartin Matuska ret = 1; 338064287048SMartin Matuska } 338164287048SMartin Matuska if ((permset << 6) != (mode & 0700)) { 338264287048SMartin Matuska failure_start(file, line, "USER_OBJ permset " 338364287048SMartin Matuska "(%02o) != user mode (%02o)", permset, 338464287048SMartin Matuska 07 & (mode >> 6)); 338564287048SMartin Matuska failure_finish(NULL); 338664287048SMartin Matuska ret = 1; 338764287048SMartin Matuska } 338864287048SMartin Matuska } else if (type == ARCHIVE_ENTRY_ACL_TYPE_ACCESS 338964287048SMartin Matuska && tag == ARCHIVE_ENTRY_ACL_GROUP_OBJ) { 339064287048SMartin Matuska if (!matched) { 339164287048SMartin Matuska failure_start(file, line, "No match for " 339264287048SMartin Matuska "group_obj perm"); 339364287048SMartin Matuska failure_finish(NULL); 339464287048SMartin Matuska ret = 1; 339564287048SMartin Matuska } 339664287048SMartin Matuska if ((permset << 3) != (mode & 0070)) { 339764287048SMartin Matuska failure_start(file, line, "GROUP_OBJ permset " 339864287048SMartin Matuska "(%02o) != group mode (%02o)", permset, 339964287048SMartin Matuska 07 & (mode >> 3)); 340064287048SMartin Matuska failure_finish(NULL); 340164287048SMartin Matuska ret = 1; 340264287048SMartin Matuska } 340364287048SMartin Matuska } else if (type == ARCHIVE_ENTRY_ACL_TYPE_ACCESS 340464287048SMartin Matuska && tag == ARCHIVE_ENTRY_ACL_OTHER) { 340564287048SMartin Matuska if (!matched) { 340664287048SMartin Matuska failure_start(file, line, "No match for " 340764287048SMartin Matuska "other perm"); 340864287048SMartin Matuska failure_finish(NULL); 340964287048SMartin Matuska ret = 1; 341064287048SMartin Matuska } 341164287048SMartin Matuska if ((permset << 0) != (mode & 0007)) { 341264287048SMartin Matuska failure_start(file, line, "OTHER permset " 341364287048SMartin Matuska "(%02o) != other mode (%02o)", permset, 341464287048SMartin Matuska mode & 07); 341564287048SMartin Matuska failure_finish(NULL); 341664287048SMartin Matuska ret = 1; 341764287048SMartin Matuska } 341864287048SMartin Matuska } else if (matched != 1) { 341964287048SMartin Matuska failure_start(file, line, "Could not find match for " 342064287048SMartin Matuska "ACL (type=%#010x,permset=%#010x,tag=%d,qual=%d," 342164287048SMartin Matuska "name=``%s'')", type, permset, tag, qual, name); 342264287048SMartin Matuska failure_finish(NULL); 342364287048SMartin Matuska ret = 1; 342464287048SMartin Matuska } 342564287048SMartin Matuska } 342664287048SMartin Matuska if (r != ARCHIVE_EOF) { 342764287048SMartin Matuska failure_start(file, line, "Should not exit before EOF"); 342864287048SMartin Matuska failure_finish(NULL); 342964287048SMartin Matuska ret = 1; 343064287048SMartin Matuska } 343164287048SMartin Matuska if ((want_type & ARCHIVE_ENTRY_ACL_TYPE_ACCESS) != 0 && 343264287048SMartin Matuska (mode_t)(mode & 0777) != (archive_entry_mode(ae) & 0777)) { 343364287048SMartin Matuska failure_start(file, line, "Mode (%02o) and entry mode (%02o) " 343464287048SMartin Matuska "mismatch", mode, archive_entry_mode(ae)); 343564287048SMartin Matuska failure_finish(NULL); 343664287048SMartin Matuska ret = 1; 343764287048SMartin Matuska } 343864287048SMartin Matuska if (n != 0) { 343964287048SMartin Matuska failure_start(file, line, "Could not find match for ACL " 344064287048SMartin Matuska "(type=%#010x,permset=%#010x,tag=%d,qual=%d,name=``%s'')", 344164287048SMartin Matuska acls[marker[0]].type, acls[marker[0]].permset, 344264287048SMartin Matuska acls[marker[0]].tag, acls[marker[0]].qual, 344364287048SMartin Matuska acls[marker[0]].name); 344464287048SMartin Matuska failure_finish(NULL); 344564287048SMartin Matuska ret = 1; 344664287048SMartin Matuska /* Number of ACLs not matched should == 0 */ 344764287048SMartin Matuska } 344864287048SMartin Matuska free(marker); 344964287048SMartin Matuska return (ret); 345064287048SMartin Matuska } 345164287048SMartin Matuska #endif /* !defined(PROGRAM) */ 345264287048SMartin Matuska 345364287048SMartin Matuska /* 345464287048SMartin Matuska * 345564287048SMartin Matuska * TEST management 345664287048SMartin Matuska * 345764287048SMartin Matuska */ 345864287048SMartin Matuska 345964287048SMartin Matuska /* 346064287048SMartin Matuska * "list.h" is simply created by "grep DEFINE_TEST test_*.c"; it has 346164287048SMartin Matuska * a line like 346264287048SMartin Matuska * DEFINE_TEST(test_function) 346364287048SMartin Matuska * for each test. 346464287048SMartin Matuska */ 346564287048SMartin Matuska 346664287048SMartin Matuska /* Use "list.h" to declare all of the test functions. */ 346764287048SMartin Matuska #undef DEFINE_TEST 346864287048SMartin Matuska #define DEFINE_TEST(name) void name(void); 346964287048SMartin Matuska #include "list.h" 347064287048SMartin Matuska 347164287048SMartin Matuska /* Use "list.h" to create a list of all tests (functions and names). */ 347264287048SMartin Matuska #undef DEFINE_TEST 347364287048SMartin Matuska #define DEFINE_TEST(n) { n, #n, 0 }, 34740f8c7ba2SKyle Evans static struct test_list_t tests[] = { 347564287048SMartin Matuska #include "list.h" 347664287048SMartin Matuska }; 347764287048SMartin Matuska 347864287048SMartin Matuska /* 347964287048SMartin Matuska * Summarize repeated failures in the just-completed test. 348064287048SMartin Matuska */ 348164287048SMartin Matuska static void 348264287048SMartin Matuska test_summarize(int failed, int skips_num) 348364287048SMartin Matuska { 348464287048SMartin Matuska unsigned int i; 348564287048SMartin Matuska 348664287048SMartin Matuska switch (verbosity) { 348764287048SMartin Matuska case VERBOSITY_SUMMARY_ONLY: 348864287048SMartin Matuska printf(failed ? "E" : "."); 348964287048SMartin Matuska fflush(stdout); 349064287048SMartin Matuska break; 349164287048SMartin Matuska case VERBOSITY_PASSFAIL: 349264287048SMartin Matuska printf(failed ? "FAIL\n" : skips_num ? "ok (S)\n" : "ok\n"); 349364287048SMartin Matuska break; 349464287048SMartin Matuska } 349564287048SMartin Matuska 349664287048SMartin Matuska log_console = (verbosity == VERBOSITY_LIGHT_REPORT); 349764287048SMartin Matuska 349864287048SMartin Matuska for (i = 0; i < sizeof(failed_lines)/sizeof(failed_lines[0]); i++) { 349964287048SMartin Matuska if (failed_lines[i].count > 1 && !failed_lines[i].skip) 350064287048SMartin Matuska logprintf("%s:%d: Summary: Failed %d times\n", 350164287048SMartin Matuska failed_filename, i, failed_lines[i].count); 350264287048SMartin Matuska } 350364287048SMartin Matuska /* Clear the failure history for the next file. */ 350464287048SMartin Matuska failed_filename = NULL; 350564287048SMartin Matuska memset(failed_lines, 0, sizeof(failed_lines)); 350664287048SMartin Matuska } 350764287048SMartin Matuska 350864287048SMartin Matuska /* 350964287048SMartin Matuska * Actually run a single test, with appropriate setup and cleanup. 351064287048SMartin Matuska */ 351164287048SMartin Matuska static int 351264287048SMartin Matuska test_run(int i, const char *tmpdir) 351364287048SMartin Matuska { 3514b1c91e4bSMartin Matuska #ifdef PATH_MAX 3515f9762417SMartin Matuska char workdir[PATH_MAX * 2]; 3516b1c91e4bSMartin Matuska #else 3517f9762417SMartin Matuska char workdir[1024 * 2]; 3518b1c91e4bSMartin Matuska #endif 351964287048SMartin Matuska char logfilename[64]; 352064287048SMartin Matuska int failures_before = failures; 352164287048SMartin Matuska int skips_before = skips; 352264287048SMartin Matuska int oldumask; 352364287048SMartin Matuska 352464287048SMartin Matuska switch (verbosity) { 352564287048SMartin Matuska case VERBOSITY_SUMMARY_ONLY: /* No per-test reports at all */ 352664287048SMartin Matuska break; 352764287048SMartin Matuska case VERBOSITY_PASSFAIL: /* rest of line will include ok/FAIL marker */ 352864287048SMartin Matuska printf("%3d: %-64s", i, tests[i].name); 352964287048SMartin Matuska fflush(stdout); 353064287048SMartin Matuska break; 353164287048SMartin Matuska default: /* Title of test, details will follow */ 353264287048SMartin Matuska printf("%3d: %s\n", i, tests[i].name); 353364287048SMartin Matuska } 353464287048SMartin Matuska 353564287048SMartin Matuska /* Chdir to the top-level work directory. */ 353664287048SMartin Matuska if (!assertChdir(tmpdir)) { 353764287048SMartin Matuska fprintf(stderr, 353864287048SMartin Matuska "ERROR: Can't chdir to top work dir %s\n", tmpdir); 353964287048SMartin Matuska exit(1); 354064287048SMartin Matuska } 354164287048SMartin Matuska /* Create a log file for this test. */ 354264287048SMartin Matuska sprintf(logfilename, "%s.log", tests[i].name); 354364287048SMartin Matuska logfile = fopen(logfilename, "w"); 354464287048SMartin Matuska fprintf(logfile, "%s\n\n", tests[i].name); 354564287048SMartin Matuska /* Chdir() to a work dir for this specific test. */ 354664287048SMartin Matuska snprintf(workdir, sizeof(workdir), "%s/%s", tmpdir, tests[i].name); 354764287048SMartin Matuska testworkdir = workdir; 354864287048SMartin Matuska if (!assertMakeDir(testworkdir, 0755) 354964287048SMartin Matuska || !assertChdir(testworkdir)) { 355064287048SMartin Matuska fprintf(stderr, 355164287048SMartin Matuska "ERROR: Can't chdir to work dir %s\n", testworkdir); 355264287048SMartin Matuska exit(1); 355364287048SMartin Matuska } 355464287048SMartin Matuska /* Explicitly reset the locale before each test. */ 355564287048SMartin Matuska setlocale(LC_ALL, "C"); 355664287048SMartin Matuska /* Record the umask before we run the test. */ 355764287048SMartin Matuska umask(oldumask = umask(0)); 355864287048SMartin Matuska /* 355964287048SMartin Matuska * Run the actual test. 356064287048SMartin Matuska */ 356164287048SMartin Matuska (*tests[i].func)(); 356264287048SMartin Matuska /* 356364287048SMartin Matuska * Clean up and report afterwards. 356464287048SMartin Matuska */ 356564287048SMartin Matuska testworkdir = NULL; 356664287048SMartin Matuska /* Restore umask */ 356764287048SMartin Matuska umask(oldumask); 356864287048SMartin Matuska /* Reset locale. */ 356964287048SMartin Matuska setlocale(LC_ALL, "C"); 357064287048SMartin Matuska /* Reset directory. */ 357164287048SMartin Matuska if (!assertChdir(tmpdir)) { 357264287048SMartin Matuska fprintf(stderr, "ERROR: Couldn't chdir to temp dir %s\n", 357364287048SMartin Matuska tmpdir); 357464287048SMartin Matuska exit(1); 357564287048SMartin Matuska } 357664287048SMartin Matuska /* Report per-test summaries. */ 357764287048SMartin Matuska tests[i].failures = failures - failures_before; 357864287048SMartin Matuska test_summarize(tests[i].failures, skips - skips_before); 357964287048SMartin Matuska /* Close the per-test log file. */ 358064287048SMartin Matuska fclose(logfile); 358164287048SMartin Matuska logfile = NULL; 358264287048SMartin Matuska /* If there were no failures, we can remove the work dir and logfile. */ 358364287048SMartin Matuska if (tests[i].failures == 0) { 358464287048SMartin Matuska if (!keep_temp_files && assertChdir(tmpdir)) { 358564287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 358664287048SMartin Matuska /* Make sure not to leave empty directories. 358764287048SMartin Matuska * Sometimes a processing of closing files used by tests 358864287048SMartin Matuska * is not done, then rmdir will be failed and it will 358964287048SMartin Matuska * leave a empty test directory. So we should wait a few 359064287048SMartin Matuska * seconds and retry rmdir. */ 359164287048SMartin Matuska int r, t; 359264287048SMartin Matuska for (t = 0; t < 10; t++) { 359364287048SMartin Matuska if (t > 0) 359464287048SMartin Matuska Sleep(1000); 359564287048SMartin Matuska r = systemf("rmdir /S /Q %s", tests[i].name); 359664287048SMartin Matuska if (r == 0) 359764287048SMartin Matuska break; 359864287048SMartin Matuska } 359964287048SMartin Matuska systemf("del %s", logfilename); 360064287048SMartin Matuska #else 360164287048SMartin Matuska systemf("rm -rf %s", tests[i].name); 360264287048SMartin Matuska systemf("rm %s", logfilename); 360364287048SMartin Matuska #endif 360464287048SMartin Matuska } 360564287048SMartin Matuska } 360664287048SMartin Matuska /* Return appropriate status. */ 360764287048SMartin Matuska return (tests[i].failures); 360864287048SMartin Matuska } 360964287048SMartin Matuska 361064287048SMartin Matuska /* 361164287048SMartin Matuska * 361264287048SMartin Matuska * 361364287048SMartin Matuska * MAIN and support routines. 361464287048SMartin Matuska * 361564287048SMartin Matuska * 361664287048SMartin Matuska */ 361764287048SMartin Matuska 361864287048SMartin Matuska static void 361964287048SMartin Matuska usage(const char *program) 362064287048SMartin Matuska { 362164287048SMartin Matuska static const int limit = sizeof(tests) / sizeof(tests[0]); 362264287048SMartin Matuska int i; 362364287048SMartin Matuska 362464287048SMartin Matuska printf("Usage: %s [options] <test> <test> ...\n", program); 362564287048SMartin Matuska printf("Default is to run all tests.\n"); 362664287048SMartin Matuska printf("Otherwise, specify the numbers of the tests you wish to run.\n"); 362764287048SMartin Matuska printf("Options:\n"); 362864287048SMartin Matuska printf(" -d Dump core after any failure, for debugging.\n"); 362964287048SMartin Matuska printf(" -k Keep all temp files.\n"); 363064287048SMartin Matuska printf(" Default: temp files for successful tests deleted.\n"); 363164287048SMartin Matuska #ifdef PROGRAM 363264287048SMartin Matuska printf(" -p <path> Path to executable to be tested.\n"); 363364287048SMartin Matuska printf(" Default: path taken from " ENVBASE " environment variable.\n"); 363464287048SMartin Matuska #endif 363564287048SMartin Matuska printf(" -q Quiet.\n"); 363664287048SMartin Matuska printf(" -r <dir> Path to dir containing reference files.\n"); 363764287048SMartin Matuska printf(" Default: Current directory.\n"); 363864287048SMartin Matuska printf(" -u Keep running specifies tests until one fails.\n"); 363964287048SMartin Matuska printf(" -v Verbose.\n"); 364064287048SMartin Matuska printf("Available tests:\n"); 364164287048SMartin Matuska for (i = 0; i < limit; i++) 364264287048SMartin Matuska printf(" %d: %s\n", i, tests[i].name); 364364287048SMartin Matuska exit(1); 364464287048SMartin Matuska } 364564287048SMartin Matuska 364664287048SMartin Matuska static char * 364764287048SMartin Matuska get_refdir(const char *d) 364864287048SMartin Matuska { 364964287048SMartin Matuska size_t tried_size, buff_size; 365064287048SMartin Matuska char *buff, *tried, *pwd = NULL, *p = NULL; 365164287048SMartin Matuska 365264287048SMartin Matuska #ifdef PATH_MAX 365364287048SMartin Matuska buff_size = PATH_MAX; 365464287048SMartin Matuska #else 365564287048SMartin Matuska buff_size = 8192; 365664287048SMartin Matuska #endif 365764287048SMartin Matuska buff = calloc(buff_size, 1); 365864287048SMartin Matuska if (buff == NULL) { 365964287048SMartin Matuska fprintf(stderr, "Unable to allocate memory\n"); 366064287048SMartin Matuska exit(1); 366164287048SMartin Matuska } 366264287048SMartin Matuska 366364287048SMartin Matuska /* Allocate a buffer to hold the various directories we checked. */ 366464287048SMartin Matuska tried_size = buff_size * 2; 366564287048SMartin Matuska tried = calloc(tried_size, 1); 366664287048SMartin Matuska if (tried == NULL) { 366764287048SMartin Matuska fprintf(stderr, "Unable to allocate memory\n"); 366864287048SMartin Matuska exit(1); 366964287048SMartin Matuska } 367064287048SMartin Matuska 367164287048SMartin Matuska /* If a dir was specified, try that */ 367264287048SMartin Matuska if (d != NULL) { 367364287048SMartin Matuska pwd = NULL; 367464287048SMartin Matuska snprintf(buff, buff_size, "%s", d); 367564287048SMartin Matuska p = slurpfile(NULL, "%s/%s", buff, KNOWNREF); 367664287048SMartin Matuska if (p != NULL) goto success; 367764287048SMartin Matuska strncat(tried, buff, tried_size - strlen(tried) - 1); 367864287048SMartin Matuska strncat(tried, "\n", tried_size - strlen(tried) - 1); 367964287048SMartin Matuska goto failure; 368064287048SMartin Matuska } 368164287048SMartin Matuska 368264287048SMartin Matuska /* Get the current dir. */ 3683*ddce862aSMartin Matuska #if defined(PATH_MAX) && !defined(__GLIBC__) 368464287048SMartin Matuska pwd = getcwd(NULL, PATH_MAX);/* Solaris getcwd needs the size. */ 368564287048SMartin Matuska #else 368664287048SMartin Matuska pwd = getcwd(NULL, 0); 368764287048SMartin Matuska #endif 368864287048SMartin Matuska while (pwd[strlen(pwd) - 1] == '\n') 368964287048SMartin Matuska pwd[strlen(pwd) - 1] = '\0'; 369064287048SMartin Matuska 369164287048SMartin Matuska /* Look for a known file. */ 369264287048SMartin Matuska snprintf(buff, buff_size, "%s", pwd); 369364287048SMartin Matuska p = slurpfile(NULL, "%s/%s", buff, KNOWNREF); 369464287048SMartin Matuska if (p != NULL) goto success; 369564287048SMartin Matuska strncat(tried, buff, tried_size - strlen(tried) - 1); 369664287048SMartin Matuska strncat(tried, "\n", tried_size - strlen(tried) - 1); 369764287048SMartin Matuska 369864287048SMartin Matuska snprintf(buff, buff_size, "%s/test", pwd); 369964287048SMartin Matuska p = slurpfile(NULL, "%s/%s", buff, KNOWNREF); 370064287048SMartin Matuska if (p != NULL) goto success; 370164287048SMartin Matuska strncat(tried, buff, tried_size - strlen(tried) - 1); 370264287048SMartin Matuska strncat(tried, "\n", tried_size - strlen(tried) - 1); 370364287048SMartin Matuska 370464287048SMartin Matuska #if defined(LIBRARY) 370564287048SMartin Matuska snprintf(buff, buff_size, "%s/%s/test", pwd, LIBRARY); 370664287048SMartin Matuska #else 370764287048SMartin Matuska snprintf(buff, buff_size, "%s/%s/test", pwd, PROGRAM); 370864287048SMartin Matuska #endif 370964287048SMartin Matuska p = slurpfile(NULL, "%s/%s", buff, KNOWNREF); 371064287048SMartin Matuska if (p != NULL) goto success; 371164287048SMartin Matuska strncat(tried, buff, tried_size - strlen(tried) - 1); 371264287048SMartin Matuska strncat(tried, "\n", tried_size - strlen(tried) - 1); 371364287048SMartin Matuska 371464287048SMartin Matuska #if defined(PROGRAM_ALIAS) 371564287048SMartin Matuska snprintf(buff, buff_size, "%s/%s/test", pwd, PROGRAM_ALIAS); 371664287048SMartin Matuska p = slurpfile(NULL, "%s/%s", buff, KNOWNREF); 371764287048SMartin Matuska if (p != NULL) goto success; 371864287048SMartin Matuska strncat(tried, buff, tried_size - strlen(tried) - 1); 371964287048SMartin Matuska strncat(tried, "\n", tried_size - strlen(tried) - 1); 372064287048SMartin Matuska #endif 372164287048SMartin Matuska 372264287048SMartin Matuska if (memcmp(pwd, "/usr/obj", 8) == 0) { 372364287048SMartin Matuska snprintf(buff, buff_size, "%s", pwd + 8); 372464287048SMartin Matuska p = slurpfile(NULL, "%s/%s", buff, KNOWNREF); 372564287048SMartin Matuska if (p != NULL) goto success; 372664287048SMartin Matuska strncat(tried, buff, tried_size - strlen(tried) - 1); 372764287048SMartin Matuska strncat(tried, "\n", tried_size - strlen(tried) - 1); 372864287048SMartin Matuska 372964287048SMartin Matuska snprintf(buff, buff_size, "%s/test", pwd + 8); 373064287048SMartin Matuska p = slurpfile(NULL, "%s/%s", buff, KNOWNREF); 373164287048SMartin Matuska if (p != NULL) goto success; 373264287048SMartin Matuska strncat(tried, buff, tried_size - strlen(tried) - 1); 373364287048SMartin Matuska strncat(tried, "\n", tried_size - strlen(tried) - 1); 373464287048SMartin Matuska } 373564287048SMartin Matuska 373664287048SMartin Matuska failure: 373764287048SMartin Matuska printf("Unable to locate known reference file %s\n", KNOWNREF); 373864287048SMartin Matuska printf(" Checked following directories:\n%s\n", tried); 373964287048SMartin Matuska printf("Use -r option to specify full path to reference directory\n"); 374064287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) && defined(_DEBUG) 374164287048SMartin Matuska DebugBreak(); 374264287048SMartin Matuska #endif 374364287048SMartin Matuska exit(1); 374464287048SMartin Matuska 374564287048SMartin Matuska success: 374664287048SMartin Matuska free(p); 374764287048SMartin Matuska free(pwd); 374864287048SMartin Matuska free(tried); 374964287048SMartin Matuska 375064287048SMartin Matuska /* Copy result into a fresh buffer to reduce memory usage. */ 375164287048SMartin Matuska p = strdup(buff); 375264287048SMartin Matuska free(buff); 375364287048SMartin Matuska return p; 375464287048SMartin Matuska } 375564287048SMartin Matuska 375664287048SMartin Matuska int 375764287048SMartin Matuska main(int argc, char **argv) 375864287048SMartin Matuska { 375964287048SMartin Matuska static const int limit = sizeof(tests) / sizeof(tests[0]); 376064287048SMartin Matuska int test_set[sizeof(tests) / sizeof(tests[0])]; 376164287048SMartin Matuska int i = 0, j = 0, tests_run = 0, tests_failed = 0, option; 376264287048SMartin Matuska time_t now; 376364287048SMartin Matuska char *refdir_alloc = NULL; 376464287048SMartin Matuska const char *progname; 376564287048SMartin Matuska char **saved_argv; 376664287048SMartin Matuska const char *tmp, *option_arg, *p; 3767b1c91e4bSMartin Matuska #ifdef PATH_MAX 3768b1c91e4bSMartin Matuska char tmpdir[PATH_MAX]; 3769b1c91e4bSMartin Matuska #else 3770b1c91e4bSMartin Matuska char tmpdir[256]; 3771b1c91e4bSMartin Matuska #endif 3772b1c91e4bSMartin Matuska char *pwd, *testprogdir, *tmp2 = NULL, *vlevel = NULL; 3773b1c91e4bSMartin Matuska char tmpdir_timestamp[32]; 377464287048SMartin Matuska 377564287048SMartin Matuska (void)argc; /* UNUSED */ 377664287048SMartin Matuska 377764287048SMartin Matuska /* Get the current dir. */ 3778*ddce862aSMartin Matuska #if defined(PATH_MAX) && !defined(__GLIBC__) 377964287048SMartin Matuska pwd = getcwd(NULL, PATH_MAX);/* Solaris getcwd needs the size. */ 378064287048SMartin Matuska #else 378164287048SMartin Matuska pwd = getcwd(NULL, 0); 378264287048SMartin Matuska #endif 378364287048SMartin Matuska while (pwd[strlen(pwd) - 1] == '\n') 378464287048SMartin Matuska pwd[strlen(pwd) - 1] = '\0'; 378564287048SMartin Matuska 378664287048SMartin Matuska #if defined(HAVE__CrtSetReportMode) && !defined(__WATCOMC__) 378764287048SMartin Matuska /* To stop to run the default invalid parameter handler. */ 378864287048SMartin Matuska _set_invalid_parameter_handler(invalid_parameter_handler); 378964287048SMartin Matuska /* Disable annoying assertion message box. */ 379064287048SMartin Matuska _CrtSetReportMode(_CRT_ASSERT, 0); 379164287048SMartin Matuska #endif 379264287048SMartin Matuska 379364287048SMartin Matuska /* 379464287048SMartin Matuska * Name of this program, used to build root of our temp directory 379564287048SMartin Matuska * tree. 379664287048SMartin Matuska */ 379764287048SMartin Matuska progname = p = argv[0]; 379864287048SMartin Matuska if ((testprogdir = (char *)malloc(strlen(progname) + 1)) == NULL) 379964287048SMartin Matuska { 380064287048SMartin Matuska fprintf(stderr, "ERROR: Out of memory."); 380164287048SMartin Matuska exit(1); 380264287048SMartin Matuska } 380364287048SMartin Matuska strcpy(testprogdir, progname); 380464287048SMartin Matuska while (*p != '\0') { 380564287048SMartin Matuska /* Support \ or / dir separators for Windows compat. */ 380664287048SMartin Matuska if (*p == '/' || *p == '\\') 380764287048SMartin Matuska { 380864287048SMartin Matuska progname = p + 1; 380964287048SMartin Matuska i = j; 381064287048SMartin Matuska } 381164287048SMartin Matuska ++p; 381264287048SMartin Matuska j++; 381364287048SMartin Matuska } 381464287048SMartin Matuska testprogdir[i] = '\0'; 381564287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 381664287048SMartin Matuska if (testprogdir[0] != '/' && testprogdir[0] != '\\' && 381764287048SMartin Matuska !(((testprogdir[0] >= 'a' && testprogdir[0] <= 'z') || 381864287048SMartin Matuska (testprogdir[0] >= 'A' && testprogdir[0] <= 'Z')) && 381964287048SMartin Matuska testprogdir[1] == ':' && 382064287048SMartin Matuska (testprogdir[2] == '/' || testprogdir[2] == '\\'))) 382164287048SMartin Matuska #else 382264287048SMartin Matuska if (testprogdir[0] != '/') 382364287048SMartin Matuska #endif 382464287048SMartin Matuska { 382564287048SMartin Matuska /* Fixup path for relative directories. */ 382664287048SMartin Matuska if ((testprogdir = (char *)realloc(testprogdir, 382764287048SMartin Matuska strlen(pwd) + 1 + strlen(testprogdir) + 1)) == NULL) 382864287048SMartin Matuska { 382964287048SMartin Matuska fprintf(stderr, "ERROR: Out of memory."); 383064287048SMartin Matuska exit(1); 383164287048SMartin Matuska } 383264287048SMartin Matuska memmove(testprogdir + strlen(pwd) + 1, testprogdir, 383364287048SMartin Matuska strlen(testprogdir) + 1); 383464287048SMartin Matuska memcpy(testprogdir, pwd, strlen(pwd)); 383564287048SMartin Matuska testprogdir[strlen(pwd)] = '/'; 383664287048SMartin Matuska } 383764287048SMartin Matuska 383864287048SMartin Matuska #ifdef PROGRAM 383964287048SMartin Matuska /* Get the target program from environment, if available. */ 384064287048SMartin Matuska testprogfile = getenv(ENVBASE); 384164287048SMartin Matuska #endif 384264287048SMartin Matuska 384364287048SMartin Matuska if (getenv("TMPDIR") != NULL) 384464287048SMartin Matuska tmp = getenv("TMPDIR"); 384564287048SMartin Matuska else if (getenv("TMP") != NULL) 384664287048SMartin Matuska tmp = getenv("TMP"); 384764287048SMartin Matuska else if (getenv("TEMP") != NULL) 384864287048SMartin Matuska tmp = getenv("TEMP"); 384964287048SMartin Matuska else if (getenv("TEMPDIR") != NULL) 385064287048SMartin Matuska tmp = getenv("TEMPDIR"); 385164287048SMartin Matuska else 385264287048SMartin Matuska tmp = "/tmp"; 385364287048SMartin Matuska 385464287048SMartin Matuska /* Allow -d to be controlled through the environment. */ 385564287048SMartin Matuska if (getenv(ENVBASE "_DEBUG") != NULL) 385664287048SMartin Matuska dump_on_failure = 1; 385764287048SMartin Matuska 385864287048SMartin Matuska /* Allow -v to be controlled through the environment. */ 385964287048SMartin Matuska if (getenv("_VERBOSITY_LEVEL") != NULL) 386064287048SMartin Matuska { 386164287048SMartin Matuska vlevel = getenv("_VERBOSITY_LEVEL"); 386264287048SMartin Matuska verbosity = atoi(vlevel); 386364287048SMartin Matuska if (verbosity < VERBOSITY_SUMMARY_ONLY || verbosity > VERBOSITY_FULL) 386464287048SMartin Matuska { 386564287048SMartin Matuska /* Unsupported verbosity levels are silently ignored */ 386664287048SMartin Matuska vlevel = NULL; 386764287048SMartin Matuska verbosity = VERBOSITY_PASSFAIL; 386864287048SMartin Matuska } 386964287048SMartin Matuska } 387064287048SMartin Matuska 387164287048SMartin Matuska /* Get the directory holding test files from environment. */ 387264287048SMartin Matuska refdir = getenv(ENVBASE "_TEST_FILES"); 387364287048SMartin Matuska 387464287048SMartin Matuska /* 387564287048SMartin Matuska * Parse options, without using getopt(), which isn't available 387664287048SMartin Matuska * on all platforms. 387764287048SMartin Matuska */ 387864287048SMartin Matuska ++argv; /* Skip program name */ 387964287048SMartin Matuska while (*argv != NULL) { 388064287048SMartin Matuska if (**argv != '-') 388164287048SMartin Matuska break; 388264287048SMartin Matuska p = *argv++; 388364287048SMartin Matuska ++p; /* Skip '-' */ 388464287048SMartin Matuska while (*p != '\0') { 388564287048SMartin Matuska option = *p++; 388664287048SMartin Matuska option_arg = NULL; 388764287048SMartin Matuska /* If 'opt' takes an argument, parse that. */ 388864287048SMartin Matuska if (option == 'p' || option == 'r') { 388964287048SMartin Matuska if (*p != '\0') 389064287048SMartin Matuska option_arg = p; 389164287048SMartin Matuska else if (*argv == NULL) { 389264287048SMartin Matuska fprintf(stderr, 389364287048SMartin Matuska "Option -%c requires argument.\n", 389464287048SMartin Matuska option); 389564287048SMartin Matuska usage(progname); 389664287048SMartin Matuska } else 389764287048SMartin Matuska option_arg = *argv++; 389864287048SMartin Matuska p = ""; /* End of this option word. */ 389964287048SMartin Matuska } 390064287048SMartin Matuska 390164287048SMartin Matuska /* Now, handle the option. */ 390264287048SMartin Matuska switch (option) { 390364287048SMartin Matuska case 'd': 390464287048SMartin Matuska dump_on_failure = 1; 390564287048SMartin Matuska break; 390664287048SMartin Matuska case 'k': 390764287048SMartin Matuska keep_temp_files = 1; 390864287048SMartin Matuska break; 390964287048SMartin Matuska case 'p': 391064287048SMartin Matuska #ifdef PROGRAM 391164287048SMartin Matuska testprogfile = option_arg; 391264287048SMartin Matuska #else 391364287048SMartin Matuska fprintf(stderr, "-p option not permitted\n"); 391464287048SMartin Matuska usage(progname); 391564287048SMartin Matuska #endif 391664287048SMartin Matuska break; 391764287048SMartin Matuska case 'q': 391864287048SMartin Matuska if (!vlevel) 391964287048SMartin Matuska verbosity--; 392064287048SMartin Matuska break; 392164287048SMartin Matuska case 'r': 392264287048SMartin Matuska refdir = option_arg; 392364287048SMartin Matuska break; 392464287048SMartin Matuska case 'u': 392564287048SMartin Matuska until_failure++; 392664287048SMartin Matuska break; 392764287048SMartin Matuska case 'v': 392864287048SMartin Matuska if (!vlevel) 392964287048SMartin Matuska verbosity++; 393064287048SMartin Matuska break; 393164287048SMartin Matuska default: 393264287048SMartin Matuska fprintf(stderr, "Unrecognized option '%c'\n", 393364287048SMartin Matuska option); 393464287048SMartin Matuska usage(progname); 393564287048SMartin Matuska } 393664287048SMartin Matuska } 393764287048SMartin Matuska } 393864287048SMartin Matuska 393964287048SMartin Matuska /* 394064287048SMartin Matuska * Sanity-check that our options make sense. 394164287048SMartin Matuska */ 394264287048SMartin Matuska #ifdef PROGRAM 394364287048SMartin Matuska if (testprogfile == NULL) 394464287048SMartin Matuska { 394564287048SMartin Matuska if ((tmp2 = (char *)malloc(strlen(testprogdir) + 1 + 394664287048SMartin Matuska strlen(PROGRAM) + 1)) == NULL) 394764287048SMartin Matuska { 394864287048SMartin Matuska fprintf(stderr, "ERROR: Out of memory."); 394964287048SMartin Matuska exit(1); 395064287048SMartin Matuska } 395164287048SMartin Matuska strcpy(tmp2, testprogdir); 395264287048SMartin Matuska strcat(tmp2, "/"); 395364287048SMartin Matuska strcat(tmp2, PROGRAM); 395464287048SMartin Matuska testprogfile = tmp2; 395564287048SMartin Matuska } 395664287048SMartin Matuska 395764287048SMartin Matuska { 395864287048SMartin Matuska char *testprg; 395964287048SMartin Matuska #if defined(_WIN32) && !defined(__CYGWIN__) 396064287048SMartin Matuska /* Command.com sometimes rejects '/' separators. */ 396164287048SMartin Matuska testprg = strdup(testprogfile); 396264287048SMartin Matuska for (i = 0; testprg[i] != '\0'; i++) { 396364287048SMartin Matuska if (testprg[i] == '/') 396464287048SMartin Matuska testprg[i] = '\\'; 396564287048SMartin Matuska } 396664287048SMartin Matuska testprogfile = testprg; 396764287048SMartin Matuska #endif 396864287048SMartin Matuska /* Quote the name that gets put into shell command lines. */ 396964287048SMartin Matuska testprg = malloc(strlen(testprogfile) + 3); 397064287048SMartin Matuska strcpy(testprg, "\""); 397164287048SMartin Matuska strcat(testprg, testprogfile); 397264287048SMartin Matuska strcat(testprg, "\""); 397364287048SMartin Matuska testprog = testprg; 397464287048SMartin Matuska } 397564287048SMartin Matuska #endif 397664287048SMartin Matuska 397764287048SMartin Matuska #if !defined(_WIN32) && defined(SIGPIPE) 397864287048SMartin Matuska { /* Ignore SIGPIPE signals */ 397964287048SMartin Matuska struct sigaction sa; 398064287048SMartin Matuska sa.sa_handler = SIG_IGN; 398164287048SMartin Matuska sigemptyset(&sa.sa_mask); 398264287048SMartin Matuska sa.sa_flags = 0; 398364287048SMartin Matuska sigaction(SIGPIPE, &sa, NULL); 398464287048SMartin Matuska } 398564287048SMartin Matuska #endif 398664287048SMartin Matuska 398764287048SMartin Matuska /* 398864287048SMartin Matuska * Create a temp directory for the following tests. 398964287048SMartin Matuska * Include the time the tests started as part of the name, 399064287048SMartin Matuska * to make it easier to track the results of multiple tests. 399164287048SMartin Matuska */ 399264287048SMartin Matuska now = time(NULL); 399364287048SMartin Matuska for (i = 0; ; i++) { 399464287048SMartin Matuska strftime(tmpdir_timestamp, sizeof(tmpdir_timestamp), 399564287048SMartin Matuska "%Y-%m-%dT%H.%M.%S", 399664287048SMartin Matuska localtime(&now)); 3997b1c91e4bSMartin Matuska if ((strlen(tmp) + 1 + strlen(progname) + 1 + 3998b1c91e4bSMartin Matuska strlen(tmpdir_timestamp) + 1 + 3) > 3999b1c91e4bSMartin Matuska (sizeof(tmpdir) / sizeof(char))) { 4000b1c91e4bSMartin Matuska fprintf(stderr, 4001b1c91e4bSMartin Matuska "ERROR: Temp directory pathname too long\n"); 4002b1c91e4bSMartin Matuska exit(1); 4003b1c91e4bSMartin Matuska } 4004b1c91e4bSMartin Matuska snprintf(tmpdir, sizeof(tmpdir), "%s/%s.%s-%03d", tmp, 4005b1c91e4bSMartin Matuska progname, tmpdir_timestamp, i); 400664287048SMartin Matuska if (assertMakeDir(tmpdir,0755)) 400764287048SMartin Matuska break; 400864287048SMartin Matuska if (i >= 999) { 400964287048SMartin Matuska fprintf(stderr, 401064287048SMartin Matuska "ERROR: Unable to create temp directory %s\n", 401164287048SMartin Matuska tmpdir); 401264287048SMartin Matuska exit(1); 401364287048SMartin Matuska } 401464287048SMartin Matuska } 401564287048SMartin Matuska 401664287048SMartin Matuska /* 401764287048SMartin Matuska * If the user didn't specify a directory for locating 401864287048SMartin Matuska * reference files, try to find the reference files in 401964287048SMartin Matuska * the "usual places." 402064287048SMartin Matuska */ 402164287048SMartin Matuska refdir = refdir_alloc = get_refdir(refdir); 402264287048SMartin Matuska 402364287048SMartin Matuska /* 402464287048SMartin Matuska * Banner with basic information. 402564287048SMartin Matuska */ 402664287048SMartin Matuska printf("\n"); 402764287048SMartin Matuska printf("If tests fail or crash, details will be in:\n"); 402864287048SMartin Matuska printf(" %s\n", tmpdir); 402964287048SMartin Matuska printf("\n"); 403064287048SMartin Matuska if (verbosity > VERBOSITY_SUMMARY_ONLY) { 403164287048SMartin Matuska printf("Reference files will be read from: %s\n", refdir); 403264287048SMartin Matuska #ifdef PROGRAM 403364287048SMartin Matuska printf("Running tests on: %s\n", testprog); 403464287048SMartin Matuska #endif 403564287048SMartin Matuska printf("Exercising: "); 403664287048SMartin Matuska fflush(stdout); 403764287048SMartin Matuska printf("%s\n", EXTRA_VERSION); 403864287048SMartin Matuska } else { 403964287048SMartin Matuska printf("Running "); 404064287048SMartin Matuska fflush(stdout); 404164287048SMartin Matuska } 404264287048SMartin Matuska 404364287048SMartin Matuska /* 404464287048SMartin Matuska * Run some or all of the individual tests. 404564287048SMartin Matuska */ 404664287048SMartin Matuska saved_argv = argv; 404764287048SMartin Matuska do { 404864287048SMartin Matuska argv = saved_argv; 404964287048SMartin Matuska do { 405064287048SMartin Matuska int test_num; 405164287048SMartin Matuska 405264287048SMartin Matuska test_num = get_test_set(test_set, limit, *argv, tests); 405364287048SMartin Matuska if (test_num < 0) { 405464287048SMartin Matuska printf("*** INVALID Test %s\n", *argv); 405564287048SMartin Matuska free(refdir_alloc); 405664287048SMartin Matuska free(testprogdir); 405764287048SMartin Matuska usage(progname); 405864287048SMartin Matuska return (1); 405964287048SMartin Matuska } 406064287048SMartin Matuska for (i = 0; i < test_num; i++) { 406164287048SMartin Matuska tests_run++; 406264287048SMartin Matuska if (test_run(test_set[i], tmpdir)) { 406364287048SMartin Matuska tests_failed++; 406464287048SMartin Matuska if (until_failure) 406564287048SMartin Matuska goto finish; 406664287048SMartin Matuska } 406764287048SMartin Matuska } 406864287048SMartin Matuska if (*argv != NULL) 406964287048SMartin Matuska argv++; 407064287048SMartin Matuska } while (*argv != NULL); 407164287048SMartin Matuska } while (until_failure); 407264287048SMartin Matuska 407364287048SMartin Matuska finish: 407464287048SMartin Matuska /* Must be freed after all tests run */ 407564287048SMartin Matuska free(tmp2); 407664287048SMartin Matuska free(testprogdir); 407764287048SMartin Matuska free(pwd); 407864287048SMartin Matuska 407964287048SMartin Matuska /* 408064287048SMartin Matuska * Report summary statistics. 408164287048SMartin Matuska */ 408264287048SMartin Matuska if (verbosity > VERBOSITY_SUMMARY_ONLY) { 408364287048SMartin Matuska printf("\n"); 408464287048SMartin Matuska printf("Totals:\n"); 408564287048SMartin Matuska printf(" Tests run: %8d\n", tests_run); 408664287048SMartin Matuska printf(" Tests failed: %8d\n", tests_failed); 408764287048SMartin Matuska printf(" Assertions checked:%8d\n", assertions); 408864287048SMartin Matuska printf(" Assertions failed: %8d\n", failures); 408964287048SMartin Matuska printf(" Skips reported: %8d\n", skips); 409064287048SMartin Matuska } 409164287048SMartin Matuska if (failures) { 409264287048SMartin Matuska printf("\n"); 409364287048SMartin Matuska printf("Failing tests:\n"); 409464287048SMartin Matuska for (i = 0; i < limit; ++i) { 409564287048SMartin Matuska if (tests[i].failures) 409664287048SMartin Matuska printf(" %d: %s (%d failures)\n", i, 409764287048SMartin Matuska tests[i].name, tests[i].failures); 409864287048SMartin Matuska } 409964287048SMartin Matuska printf("\n"); 410064287048SMartin Matuska printf("Details for failing tests: %s\n", tmpdir); 410164287048SMartin Matuska printf("\n"); 410264287048SMartin Matuska } else { 410364287048SMartin Matuska if (verbosity == VERBOSITY_SUMMARY_ONLY) 410464287048SMartin Matuska printf("\n"); 410564287048SMartin Matuska printf("%d tests passed, no failures\n", tests_run); 410664287048SMartin Matuska } 410764287048SMartin Matuska 410864287048SMartin Matuska free(refdir_alloc); 410964287048SMartin Matuska 411064287048SMartin Matuska /* If the final tmpdir is empty, we can remove it. */ 411164287048SMartin Matuska /* This should be the usual case when all tests succeed. */ 411264287048SMartin Matuska assertChdir(".."); 411364287048SMartin Matuska rmdir(tmpdir); 411464287048SMartin Matuska 411564287048SMartin Matuska return (tests_failed ? 1 : 0); 411664287048SMartin Matuska } 4117