1/* 2 * Copyright (c) 1995 Wolfram Schneider <wosch@FreeBSD.org>. Berlin. 3 * Copyright (c) 1989, 1993 4 * The Regents of the University of California. All rights reserved. 5 * 6 * This code is derived from software contributed to Berkeley by 7 * James A. Woods. 8 * 9 * Redistribution and use in source and binary forms, with or without 10 * modification, are permitted provided that the following conditions 11 * are met: 12 * 1. Redistributions of source code must retain the above copyright 13 * notice, this list of conditions and the following disclaimer. 14 * 2. Redistributions in binary form must reproduce the above copyright 15 * notice, this list of conditions and the following disclaimer in the 16 * documentation and/or other materials provided with the distribution. 17 * 3. All advertising materials mentioning features or use of this software 18 * must display the following acknowledgement: 19 * This product includes software developed by the University of 20 * California, Berkeley and its contributors. 21 * 4. Neither the name of the University nor the names of its contributors 22 * may be used to endorse or promote products derived from this software 23 * without specific prior written permission. 24 * 25 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 26 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 27 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 28 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 29 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 30 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 31 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 32 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 33 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 34 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 35 * SUCH DAMAGE. 36 * 37 * $FreeBSD: src/usr.bin/locate/locate/fastfind.c,v 1.14 2005/12/07 12:22:46 des Exp $ 38 */ 39 40 41#ifndef _LOCATE_STATISTIC_ 42#define _LOCATE_STATISTIC_ 43 44void 45statistic (fp, path_fcodes) 46 FILE *fp; /* open database */ 47 char *path_fcodes; /* for error message */ 48{ 49 register int lines, chars, size, big, zwerg; 50 register u_char *p, *s; 51 register int c; 52 int count, umlaut; 53 u_char bigram1[NBG], bigram2[NBG], path[MAXPATHLEN]; 54 55 for (c = 0, p = bigram1, s = bigram2; c < NBG; c++) { 56 p[c] = check_bigram_char(getc(fp)); 57 s[c] = check_bigram_char(getc(fp)); 58 } 59 60 lines = chars = big = zwerg = umlaut = 0; 61 size = NBG + NBG; 62 63 for (c = getc(fp), count = 0; c != EOF; size++) { 64 if (c == SWITCH) { 65 count += getwf(fp) - OFFSET; 66 size += sizeof(int); 67 zwerg++; 68 } else 69 count += c - OFFSET; 70 71 for (p = path + count; (c = getc(fp)) > SWITCH; size++) 72 if (c < PARITY) { 73 if (c == UMLAUT) { 74 c = getc(fp); 75 size++; 76 umlaut++; 77 } 78 p++; 79 } else { 80 /* bigram char */ 81 big++; 82 p += 2; 83 } 84 85 p++; 86 lines++; 87 chars += (p - path); 88 } 89 90 (void)printf("\nDatabase: %s\n", path_fcodes); 91 (void)printf("Compression: Front: %2.2f%%, ", 92 (size + big - (2 * NBG)) / (chars / (float)100)); 93 (void)printf("Bigram: %2.2f%%, ", (size - big) / (size / (float)100)); 94 (void)printf("Total: %2.2f%%\n", 95 (size - (2 * NBG)) / (chars / (float)100)); 96 (void)printf("Filenames: %d, ", lines); 97 (void)printf("Characters: %d, ", chars); 98 (void)printf("Database size: %d\n", size); 99 (void)printf("Bigram characters: %d, ", big); 100 (void)printf("Integers: %d, ", zwerg); 101 (void)printf("8-Bit characters: %d\n", umlaut); 102 103} 104#endif /* _LOCATE_STATISTIC_ */ 105 106extern char separator; 107 108void 109#ifdef FF_MMAP 110 111 112#ifdef FF_ICASE 113fastfind_mmap_icase 114#else 115fastfind_mmap 116#endif /* FF_ICASE */ 117(pathpart, paddr, len, database) 118 char *pathpart; /* search string */ 119 caddr_t paddr; /* mmap pointer */ 120 int len; /* length of database */ 121 char *database; /* for error message */ 122 123 124#else /* MMAP */ 125 126 127#ifdef FF_ICASE 128fastfind_icase 129#else 130fastfind 131#endif /* FF_ICASE */ 132 133(fp, pathpart, database) 134 FILE *fp; /* open database */ 135 char *pathpart; /* search string */ 136 char *database; /* for error message */ 137 138 139#endif /* MMAP */ 140 141{ 142 register u_char *p, *s, *patend, *q, *foundchar; 143 register int c, cc; 144 int count, found, globflag; 145 u_char *cutoff; 146 u_char bigram1[NBG], bigram2[NBG], path[MAXPATHLEN]; 147 148#ifdef FF_ICASE 149 /* use a lookup table for case insensitive search */ 150 u_char table[UCHAR_MAX + 1]; 151 152 tolower_word(pathpart); 153#endif /* FF_ICASE*/ 154 155 /* init bigram table */ 156#ifdef FF_MMAP 157 if (len < (2*NBG)) 158 errx(1, "database too small: %s", database); 159 160 for (c = 0, p = bigram1, s = bigram2; c < NBG; c++, len-= 2) { 161 p[c] = check_bigram_char(*paddr++); 162 s[c] = check_bigram_char(*paddr++); 163 } 164#else 165 for (c = 0, p = bigram1, s = bigram2; c < NBG; c++) { 166 p[c] = check_bigram_char(getc(fp)); 167 s[c] = check_bigram_char(getc(fp)); 168 } 169#endif /* FF_MMAP */ 170 171 /* find optimal (last) char for searching */ 172 for (p = pathpart; *p != '\0'; p++) 173 if (index(LOCATE_REG, *p) != NULL) 174 break; 175 176 if (*p == '\0') 177 globflag = 0; 178 else 179 globflag = 1; 180 181 p = pathpart; 182 patend = patprep(p); 183 cc = *patend; 184 185#ifdef FF_ICASE 186 /* set patend char to true */ 187 for (c = 0; c < UCHAR_MAX + 1; c++) 188 table[c] = 0; 189 190 table[TOLOWER(*patend)] = 1; 191 table[toupper(*patend)] = 1; 192#endif /* FF_ICASE */ 193 194 195 /* main loop */ 196 found = count = 0; 197 foundchar = 0; 198 199#ifdef FF_MMAP 200 c = (u_char)*paddr++; len--; 201 for (; len > 0; ) { 202#else 203 c = getc(fp); 204 for (; c != EOF; ) { 205#endif /* FF_MMAP */ 206 207 /* go forward or backward */ 208 if (c == SWITCH) { /* big step, an integer */ 209#ifdef FF_MMAP 210 count += getwm(paddr) - OFFSET; 211 len -= INTSIZE; paddr += INTSIZE; 212#else 213 count += getwf(fp) - OFFSET; 214#endif /* FF_MMAP */ 215 } else { /* slow step, =< 14 chars */ 216 count += c - OFFSET; 217 } 218 219#ifdef __APPLE__ 220 if (count < 0) { 221 errx(1, "Your locate database appears to be corrupt. " 222 "Run 'sudo /usr/libexec/locate.updatedb' to " 223 "regenerate the database."); 224 } 225#endif /* __APPLE__ */ 226 227 /* overlay old path */ 228 p = path + count; 229 foundchar = p - 1; 230 231#ifdef FF_MMAP 232 for (; len > 0;) { 233 c = (u_char)*paddr++; 234 len--; 235#else 236 for (;;) { 237 c = getc(fp); 238#endif /* FF_MMAP */ 239 /* 240 * == UMLAUT: 8 bit char followed 241 * <= SWITCH: offset 242 * >= PARITY: bigram 243 * rest: single ascii char 244 * 245 * offset < SWITCH < UMLAUT < ascii < PARITY < bigram 246 */ 247 if (c < PARITY) { 248 if (c <= UMLAUT) { 249 if (c == UMLAUT) { 250#ifdef FF_MMAP 251 c = (u_char)*paddr++; 252 len--; 253#else 254 c = getc(fp); 255#endif /* FF_MMAP */ 256 257 } else 258 break; /* SWITCH */ 259 } 260#ifdef FF_ICASE 261 if (table[c]) 262#else 263 if (c == cc) 264#endif /* FF_ICASE */ 265 foundchar = p; 266 *p++ = c; 267 } 268 else { 269 /* bigrams are parity-marked */ 270 TO7BIT(c); 271 272#ifndef FF_ICASE 273 if (bigram1[c] == cc || 274 bigram2[c] == cc) 275#else 276 277 if (table[bigram1[c]] || 278 table[bigram2[c]]) 279#endif /* FF_ICASE */ 280 foundchar = p + 1; 281 282 *p++ = bigram1[c]; 283 *p++ = bigram2[c]; 284 } 285 } 286 287 if (found) { /* previous line matched */ 288 cutoff = path; 289 *p-- = '\0'; 290 foundchar = p; 291 } else if (foundchar >= path + count) { /* a char matched */ 292 *p-- = '\0'; 293 cutoff = path + count; 294 } else /* nothing to do */ 295 continue; 296 297 found = 0; 298 for (s = foundchar; s >= cutoff; s--) { 299 if (*s == cc 300#ifdef FF_ICASE 301 || TOLOWER(*s) == cc 302#endif /* FF_ICASE */ 303 ) { /* fast first char check */ 304 for (p = patend - 1, q = s - 1; *p != '\0'; 305 p--, q--) 306 if (*q != *p 307#ifdef FF_ICASE 308 && TOLOWER(*q) != *p 309#endif /* FF_ICASE */ 310 ) 311 break; 312 if (*p == '\0') { /* fast match success */ 313 found = 1; 314 if (!globflag || 315#ifndef FF_ICASE 316 !fnmatch(pathpart, path, 0)) 317#else 318 !fnmatch(pathpart, path, 319 FNM_CASEFOLD)) 320#endif /* !FF_ICASE */ 321 { 322 if (f_silent) 323 counter++; 324 else if (f_limit) { 325 counter++; 326 if (f_limit >= counter) 327 (void)printf("%s%c",path,separator); 328 else 329 errx(0, "[show only %d lines]", counter - 1); 330 } else 331 (void)printf("%s%c",path,separator); 332 } 333 break; 334 } 335 } 336 } 337 } 338} 339