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
print_help(const char * progname,int status)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
main(int argc,const char * argv[])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