1 /* 2 * suftest.c for libdivsufsort 3 * Copyright (c) 2003-2008 Yuta Mori All Rights Reserved. 4 * 5 * Permission is hereby granted, free of charge, to any person 6 * obtaining a copy of this software and associated documentation 7 * files (the "Software"), to deal in the Software without 8 * restriction, including without limitation the rights to use, 9 * copy, modify, merge, publish, distribute, sublicense, and/or sell 10 * copies of the Software, and to permit persons to whom the 11 * Software is furnished to do so, subject to the following 12 * conditions: 13 * 14 * The above copyright notice and this permission notice shall be 15 * included in all copies or substantial portions of the Software. 16 * 17 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, 18 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES 19 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND 20 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT 21 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, 22 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING 23 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR 24 * OTHER DEALINGS IN THE SOFTWARE. 25 */ 26 27 #if HAVE_CONFIG_H 28 # include "config.h" 29 #endif 30 #include <stdio.h> 31 #if HAVE_STRING_H 32 # include <string.h> 33 #endif 34 #if HAVE_STDLIB_H 35 # include <stdlib.h> 36 #endif 37 #if HAVE_MEMORY_H 38 # include <memory.h> 39 #endif 40 #if HAVE_STDDEF_H 41 # include <stddef.h> 42 #endif 43 #if HAVE_STRINGS_H 44 # include <strings.h> 45 #endif 46 #if HAVE_SYS_TYPES_H 47 # include <sys/types.h> 48 #endif 49 #if HAVE_IO_H && HAVE_FCNTL_H 50 # include <io.h> 51 # include <fcntl.h> 52 #endif 53 #include <time.h> 54 #include <divsufsort.h> 55 #include "lfs.h" 56 57 58 static 59 void 60 print_help(const char *progname, int status) { 61 fprintf(stderr, 62 "suftest, a suffixsort tester, version %s.\n", 63 divsufsort_version()); 64 fprintf(stderr, "usage: %s FILE\n\n", progname); 65 exit(status); 66 } 67 68 int 69 main(int argc, const char *argv[]) { 70 FILE *fp; 71 const char *fname; 72 sauchar_t *T; 73 saidx_t *SA; 74 LFS_OFF_T n; 75 clock_t start, finish; 76 saint_t needclose = 1; 77 78 /* Check arguments. */ 79 if((argc == 1) || 80 (strcmp(argv[1], "-h") == 0) || 81 (strcmp(argv[1], "--help") == 0)) { print_help(argv[0], EXIT_SUCCESS); } 82 if(argc != 2) { print_help(argv[0], EXIT_FAILURE); } 83 84 /* Open a file for reading. */ 85 if(strcmp(argv[1], "-") != 0) { 86 #if HAVE_FOPEN_S 87 if(fopen_s(&fp, fname = argv[1], "rb") != 0) { 88 #else 89 if((fp = LFS_FOPEN(fname = argv[1], "rb")) == NULL) { 90 #endif 91 fprintf(stderr, "%s: Cannot open file `%s': ", argv[0], fname); 92 perror(NULL); 93 exit(EXIT_FAILURE); 94 } 95 } else { 96 #if HAVE__SETMODE && HAVE__FILENO 97 if(_setmode(_fileno(stdin), _O_BINARY) == -1) { 98 fprintf(stderr, "%s: Cannot set mode: ", argv[0]); 99 perror(NULL); 100 exit(EXIT_FAILURE); 101 } 102 #endif 103 fp = stdin; 104 fname = "stdin"; 105 needclose = 0; 106 } 107 108 /* Get the file size. */ 109 if(LFS_FSEEK(fp, 0, SEEK_END) == 0) { 110 n = LFS_FTELL(fp); 111 rewind(fp); 112 if(n < 0) { 113 fprintf(stderr, "%s: Cannot ftell `%s': ", argv[0], fname); 114 perror(NULL); 115 exit(EXIT_FAILURE); 116 } 117 if(0x7fffffff <= n) { 118 fprintf(stderr, "%s: Input file `%s' is too big.\n", argv[0], fname); 119 exit(EXIT_FAILURE); 120 } 121 } else { 122 fprintf(stderr, "%s: Cannot fseek `%s': ", argv[0], fname); 123 perror(NULL); 124 exit(EXIT_FAILURE); 125 } 126 127 /* Allocate 5n bytes of memory. */ 128 T = (sauchar_t *)malloc((size_t)n * sizeof(sauchar_t)); 129 SA = (saidx_t *)malloc((size_t)n * sizeof(saidx_t)); 130 if((T == NULL) || (SA == NULL)) { 131 fprintf(stderr, "%s: Cannot allocate memory.\n", argv[0]); 132 exit(EXIT_FAILURE); 133 } 134 135 /* Read n bytes of data. */ 136 if(fread(T, sizeof(sauchar_t), (size_t)n, fp) != (size_t)n) { 137 fprintf(stderr, "%s: %s `%s': ", 138 argv[0], 139 (ferror(fp) || !feof(fp)) ? "Cannot read from" : "Unexpected EOF in", 140 argv[1]); 141 perror(NULL); 142 exit(EXIT_FAILURE); 143 } 144 if(needclose & 1) { fclose(fp); } 145 146 /* Construct the suffix array. */ 147 fprintf(stderr, "%s: %" PRIdOFF_T " bytes ... ", fname, n); 148 start = clock(); 149 if(divsufsort(T, SA, (saidx_t)n) != 0) { 150 fprintf(stderr, "%s: Cannot allocate memory.\n", argv[0]); 151 exit(EXIT_FAILURE); 152 } 153 finish = clock(); 154 fprintf(stderr, "%.4f sec\n", (double)(finish - start) / (double)CLOCKS_PER_SEC); 155 156 /* Check the suffix array. */ 157 if(sufcheck(T, SA, (saidx_t)n, 1) != 0) { exit(EXIT_FAILURE); } 158 159 /* Deallocate memory. */ 160 free(SA); 161 free(T); 162 163 return 0; 164 } 165
