1 /* 2 * sasearch.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 <divsufsort.h> 54 #include "lfs.h" 55 56 57 static 58 void 59 print_help(const char *progname, int status) { 60 fprintf(stderr, 61 "sasearch, a simple SA-based full-text search tool, version %s\n", 62 divsufsort_version()); 63 fprintf(stderr, "usage: %s PATTERN FILE SAFILE\n\n", progname); 64 exit(status); 65 } 66 67 int 68 main(int argc, const char *argv[]) { 69 FILE *fp; 70 const char *P; 71 sauchar_t *T; 72 saidx_t *SA; 73 LFS_OFF_T n; 74 size_t Psize; 75 saidx_t i, size, left; 76 77 if((argc == 1) || 78 (strcmp(argv[1], "-h") == 0) || 79 (strcmp(argv[1], "--help") == 0)) { print_help(argv[0], EXIT_SUCCESS); } 80 if(argc != 4) { print_help(argv[0], EXIT_FAILURE); } 81 82 P = argv[1]; 83 Psize = strlen(P); 84 85 /* Open a file for reading. */ 86 #if HAVE_FOPEN_S 87 if(fopen_s(&fp, argv[2], "rb") != 0) { 88 #else 89 if((fp = LFS_FOPEN(argv[2], "rb")) == NULL) { 90 #endif 91 fprintf(stderr, "%s: Cannot open file `%s': ", argv[0], argv[2]); 92 perror(NULL); 93 exit(EXIT_FAILURE); 94 } 95 96 /* Get the file size. */ 97 if(LFS_FSEEK(fp, 0, SEEK_END) == 0) { 98 n = LFS_FTELL(fp); 99 rewind(fp); 100 if(n < 0) { 101 fprintf(stderr, "%s: Cannot ftell `%s': ", argv[0], argv[2]); 102 perror(NULL); 103 exit(EXIT_FAILURE); 104 } 105 } else { 106 fprintf(stderr, "%s: Cannot fseek `%s': ", argv[0], argv[2]); 107 perror(NULL); 108 exit(EXIT_FAILURE); 109 } 110 111 /* Allocate 5n bytes of memory. */ 112 T = (sauchar_t *)malloc((size_t)n * sizeof(sauchar_t)); 113 SA = (saidx_t *)malloc((size_t)n * sizeof(saidx_t)); 114 if((T == NULL) || (SA == NULL)) { 115 fprintf(stderr, "%s: Cannot allocate memory.\n", argv[0]); 116 exit(EXIT_FAILURE); 117 } 118 119 /* Read n bytes of data. */ 120 if(fread(T, sizeof(sauchar_t), (size_t)n, fp) != (size_t)n) { 121 fprintf(stderr, "%s: %s `%s': ", 122 argv[0], 123 (ferror(fp) || !feof(fp)) ? "Cannot read from" : "Unexpected EOF in", 124 argv[2]); 125 perror(NULL); 126 exit(EXIT_FAILURE); 127 } 128 fclose(fp); 129 130 /* Open the SA file for reading. */ 131 #if HAVE_FOPEN_S 132 if(fopen_s(&fp, argv[3], "rb") != 0) { 133 #else 134 if((fp = LFS_FOPEN(argv[3], "rb")) == NULL) { 135 #endif 136 fprintf(stderr, "%s: Cannot open file `%s': ", argv[0], argv[3]); 137 perror(NULL); 138 exit(EXIT_FAILURE); 139 } 140 141 /* Read n * sizeof(saidx_t) bytes of data. */ 142 if(fread(SA, sizeof(saidx_t), (size_t)n, fp) != (size_t)n) { 143 fprintf(stderr, "%s: %s `%s': ", 144 argv[0], 145 (ferror(fp) || !feof(fp)) ? "Cannot read from" : "Unexpected EOF in", 146 argv[3]); 147 perror(NULL); 148 exit(EXIT_FAILURE); 149 } 150 fclose(fp); 151 152 /* Search and print */ 153 size = sa_search(T, (saidx_t)n, 154 (const sauchar_t *)P, (saidx_t)Psize, 155 SA, (saidx_t)n, &left); 156 for(i = 0; i < size; ++i) { 157 fprintf(stdout, "%" PRIdSAIDX_T "\n", SA[left + i]); 158 } 159 160 /* Deallocate memory. */ 161 free(SA); 162 free(T); 163 164 return 0; 165 } 166
