1 /*- 2 * Copyright (c) 1986, 1993 3 * The Regents of the University of California. All rights reserved. 4 * 5 * This code is derived from software contributed to Berkeley by 6 * Ken Arnold. 7 * 8 * Redistribution and use in source and binary forms, with or without 9 * modification, are permitted provided that the following conditions 10 * are met: 11 * 1. Redistributions of source code must retain the above copyright 12 * notice, this list of conditions and the following disclaimer. 13 * 2. Redistributions in binary form must reproduce the above copyright 14 * notice, this list of conditions and the following disclaimer in the 15 * documentation and/or other materials provided with the distribution. 16 * 3. Neither the name of the University nor the names of its contributors 17 * may be used to endorse or promote products derived from this software 18 * without specific prior written permission. 19 * 20 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 23 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 24 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 25 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 26 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 27 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 30 * SUCH DAMAGE. 31 * 32 * @(#) Copyright (c) 1986, 1993 The Regents of the University of California. All rights reserved. 33 * @(#)fortune.c 8.1 (Berkeley) 5/31/93 34 * $FreeBSD: src/games/fortune/fortune/fortune.c,v 1.18.2.1 2001/07/02 00:35:27 dd Exp $ 35 */ 36 37 #include <sys/param.h> 38 #include <sys/stat.h> 39 #include <netinet/in.h> 40 41 #include <dirent.h> 42 #include <err.h> 43 #include <fcntl.h> 44 #include <assert.h> 45 #include <unistd.h> 46 #include <stdbool.h> 47 #include <stdio.h> 48 #include <ctype.h> 49 #include <stdlib.h> 50 #include <string.h> 51 #include <locale.h> 52 #include <time.h> 53 54 #include "strfile.h" 55 #include "pathnames.h" 56 57 #define MINW 6 /* minimum wait if desired */ 58 #define CPERS 20 /* # of chars for each sec */ 59 #define SLEN 160 /* # of chars in short fortune */ 60 61 #define POS_UNKNOWN ((long)-1) /* pos for file unknown */ 62 #define NO_PROB (-1) /* no prob specified for file */ 63 64 #ifdef DEBUG 65 #define DPRINTF(l,x) { if (Debug >= l) fprintf x; } 66 #undef NDEBUG 67 #else 68 #define DPRINTF(l,x) 69 #define NDEBUG 1 70 #endif 71 72 typedef struct fd { 73 int percent; 74 int fd, datfd; 75 long pos; 76 FILE *inf; 77 const char *name; 78 const char *path; 79 char *datfile, *posfile; 80 bool read_tbl; 81 bool was_pos_file; 82 STRFILE tbl; 83 int num_children; 84 struct fd *child, *parent; 85 struct fd *next, *prev; 86 } FILEDESC; 87 88 bool Found_one; /* did we find a match? */ 89 bool Find_files = false; /* just find a list of proper fortune files */ 90 bool Fortunes_only = false; /* check only "fortunes" files */ 91 bool Wait = false; /* wait desired after fortune */ 92 bool Short_only = false; /* short fortune desired */ 93 bool Long_only = false; /* long fortune desired */ 94 bool Offend = false; /* offensive fortunes only */ 95 bool All_forts = false; /* any fortune allowed */ 96 bool Equal_probs= false; /* scatter un-allocted prob equally */ 97 #ifndef NO_REGEX 98 bool Match = false; /* dump fortunes matching a pattern */ 99 #endif 100 #ifdef DEBUG 101 bool Debug = false; /* print debug messages */ 102 #endif 103 104 char *Fortbuf = NULL; /* fortune buffer for -m */ 105 106 int Fort_len = 0; 107 108 long Seekpts[2]; /* seek pointers to fortunes */ 109 110 FILEDESC *File_list = NULL, /* Head of file list */ 111 *File_tail = NULL; /* Tail of file list */ 112 FILEDESC *Fortfile; /* Fortune file to use */ 113 114 STRFILE Noprob_tbl; /* sum of data for all no prob files */ 115 116 bool add_dir(FILEDESC *); 117 bool add_file(int, 118 const char *, const char *, FILEDESC **, FILEDESC **, FILEDESC *); 119 void all_forts(FILEDESC *, char *); 120 char *copy(const char *, u_int); 121 void display(FILEDESC *); 122 void do_free(void *); 123 void *do_malloc(u_int); 124 bool form_file_list(char **, int); 125 int fortlen(void); 126 void get_fort(void); 127 void get_pos(FILEDESC *); 128 void get_tbl(FILEDESC *); 129 void getargs(int, char *[]); 130 void init_prob(void); 131 bool is_dir(const char *); 132 bool is_fortfile(const char *, char **, char **, int); 133 bool is_off_name(const char *); 134 int max(int, int); 135 FILEDESC *new_fp(void); 136 char *off_name(const char *); 137 void open_dat(FILEDESC *); 138 void open_fp(FILEDESC *); 139 FILEDESC *pick_child(FILEDESC *); 140 void print_file_list(void); 141 void print_list(FILEDESC *, int); 142 void sum_noprobs(FILEDESC *); 143 void sum_tbl(STRFILE *, STRFILE *); 144 void usage(void); 145 void zero_tbl(STRFILE *); 146 147 #ifndef NO_REGEX 148 char *conv_pat(char *); 149 int find_matches(void); 150 void matches_in_list(FILEDESC *); 151 int maxlen_in_list(FILEDESC *); 152 #endif 153 154 #ifndef NO_REGEX 155 #ifdef REGCMP 156 #define RE_COMP(p) (Re_pat = regcmp(p, NULL)) 157 #define BAD_COMP(f) ((f) == NULL) 158 #define RE_EXEC(p) regex(Re_pat, (p)) 159 160 char *Re_pat; 161 char *regcmp(), *regex(); 162 #else 163 #define RE_COMP(p) (p = re_comp(p)) 164 #define BAD_COMP(f) ((f) != NULL) 165 #define RE_EXEC(p) re_exec(p) 166 #endif 167 #endif 168 169 int 170 main(int ac, char *av[]) 171 { 172 #ifdef OK_TO_WRITE_DISK 173 int fd; 174 #endif /* OK_TO_WRITE_DISK */ 175 176 setlocale(LC_ALL, ""); 177 178 getargs(ac, av); 179 180 #ifndef NO_REGEX 181 if (Match) 182 exit(find_matches() != 0); 183 #endif 184 185 init_prob(); 186 srandomdev(); 187 do { 188 get_fort(); 189 } while ((Short_only && fortlen() > SLEN) || 190 (Long_only && fortlen() <= SLEN)); 191 192 display(Fortfile); 193 194 #ifdef OK_TO_WRITE_DISK 195 if ((fd = creat(Fortfile->posfile, 0666)) < 0) { 196 perror(Fortfile->posfile); 197 exit(1); 198 } 199 #ifdef LOCK_EX 200 /* 201 * if we can, we exclusive lock, but since it isn't very 202 * important, we just punt if we don't have easy locking 203 * available. 204 */ 205 flock(fd, LOCK_EX); 206 #endif /* LOCK_EX */ 207 write(fd, (char *) &Fortfile->pos, sizeof Fortfile->pos); 208 if (!Fortfile->was_pos_file) 209 chmod(Fortfile->path, 0666); 210 #ifdef LOCK_EX 211 flock(fd, LOCK_UN); 212 #endif /* LOCK_EX */ 213 #endif /* OK_TO_WRITE_DISK */ 214 if (Wait) { 215 if (Fort_len == 0) 216 fortlen(); 217 sleep((unsigned int)max(Fort_len / CPERS, MINW)); 218 } 219 220 exit(0); 221 } 222 223 void 224 display(FILEDESC *fp) 225 { 226 char *p; 227 unsigned char ch; 228 char line[BUFSIZ]; 229 230 open_fp(fp); 231 fseek(fp->inf, Seekpts[0], SEEK_SET); 232 for (Fort_len = 0; fgets(line, sizeof line, fp->inf) != NULL && 233 !STR_ENDSTRING(line, fp->tbl); Fort_len++) { 234 if (fp->tbl.str_flags & STR_ROTATED) 235 for (p = line; (ch = *p) != '\0'; ++p) { 236 if (isascii(ch)) { 237 if (isupper(ch)) 238 *p = 'A' + (ch - 'A' + 13) % 26; 239 else if (islower(ch)) 240 *p = 'a' + (ch - 'a' + 13) % 26; 241 } 242 } 243 if (fp->tbl.str_flags & STR_COMMENTS 244 && line[0] == fp->tbl.str_delim 245 && line[1] == fp->tbl.str_delim) 246 continue; 247 fputs(line, stdout); 248 } 249 fflush(stdout); 250 } 251 252 /* 253 * fortlen: 254 * Return the length of the fortune. 255 */ 256 int 257 fortlen(void) 258 { 259 int nchar; 260 char line[BUFSIZ]; 261 262 if (!(Fortfile->tbl.str_flags & (STR_RANDOM | STR_ORDERED))) 263 nchar = (Seekpts[1] - Seekpts[0] <= SLEN); 264 else { 265 open_fp(Fortfile); 266 fseek(Fortfile->inf, Seekpts[0], SEEK_SET); 267 nchar = 0; 268 while (fgets(line, sizeof line, Fortfile->inf) != NULL && 269 !STR_ENDSTRING(line, Fortfile->tbl)) 270 nchar += strlen(line); 271 } 272 Fort_len = nchar; 273 274 return (nchar); 275 } 276 277 /* 278 * This routine evaluates the arguments on the command line 279 */ 280 void 281 getargs(int argc, char **argv) 282 { 283 int ignore_case; 284 #ifndef NO_REGEX 285 char *pat; 286 #endif /* NO_REGEX */ 287 int ch; 288 289 ignore_case = false; 290 #ifndef NO_REGEX 291 pat = NULL; 292 #endif /* NO_REGEX */ 293 294 #ifdef DEBUG 295 while ((ch = getopt(argc, argv, "aDefilm:osw")) != -1) 296 #else 297 while ((ch = getopt(argc, argv, "aefilm:osw")) != -1) 298 #endif /* DEBUG */ 299 switch(ch) { 300 case 'a': /* any fortune */ 301 All_forts++; 302 break; 303 #ifdef DEBUG 304 case 'D': 305 Debug++; 306 break; 307 #endif /* DEBUG */ 308 case 'e': 309 Equal_probs++; /* scatter un-allocted prob equally */ 310 break; 311 case 'f': /* find fortune files */ 312 Find_files++; 313 break; 314 case 'l': /* long ones only */ 315 Long_only++; 316 Short_only = false; 317 break; 318 case 'o': /* offensive ones only */ 319 Offend++; 320 break; 321 case 's': /* short ones only */ 322 Short_only++; 323 Long_only = false; 324 break; 325 case 'w': /* give time to read */ 326 Wait++; 327 break; 328 #ifdef NO_REGEX 329 case 'i': /* case-insensitive match */ 330 case 'm': /* dump out the fortunes */ 331 fprintf(stderr, 332 "fortune: can't match fortunes on this system (Sorry)\n"); 333 exit(0); 334 #else /* NO_REGEX */ 335 case 'm': /* dump out the fortunes */ 336 Match++; 337 pat = optarg; 338 break; 339 case 'i': /* case-insensitive match */ 340 ignore_case++; 341 break; 342 #endif /* NO_REGEX */ 343 case '?': 344 default: 345 usage(); 346 } 347 argc -= optind; 348 argv += optind; 349 350 if (!form_file_list(argv, argc)) 351 exit(1); /* errors printed through form_file_list() */ 352 if (Find_files) { 353 print_file_list(); 354 exit(0); 355 } 356 #ifdef DEBUG 357 else if (Debug >= 1) 358 print_file_list(); 359 #endif /* DEBUG */ 360 361 #ifndef NO_REGEX 362 if (pat != NULL) { 363 if (ignore_case) 364 pat = conv_pat(pat); 365 if (BAD_COMP(RE_COMP(pat))) { 366 #ifndef REGCMP 367 fprintf(stderr, "%s\n", pat); 368 #else /* REGCMP */ 369 fprintf(stderr, "bad pattern: %s\n", pat); 370 #endif /* REGCMP */ 371 } 372 } 373 #endif /* NO_REGEX */ 374 } 375 376 /* 377 * form_file_list: 378 * Form the file list from the file specifications. 379 */ 380 bool 381 form_file_list(char **files, int file_cnt) 382 { 383 bool i; 384 int percent; 385 const char *sp; 386 387 if (file_cnt == 0) { 388 if (Find_files) { 389 Fortunes_only = true; 390 i = add_file(NO_PROB, FORTDIR, NULL, &File_list, 391 &File_tail, NULL); 392 Fortunes_only = false; 393 return (i); 394 } else 395 return (add_file(NO_PROB, "fortunes", FORTDIR, 396 &File_list, &File_tail, NULL)); 397 } 398 for (i = 0; i < file_cnt; i++) { 399 percent = NO_PROB; 400 if (!isdigit((unsigned char)files[i][0])) 401 sp = files[i]; 402 else { 403 percent = 0; 404 for (sp = files[i]; isdigit((unsigned char)*sp); sp++) 405 percent = percent * 10 + *sp - '0'; 406 if (percent > 100) { 407 fprintf(stderr, "percentages must be <= 100\n"); 408 return (false); 409 } 410 if (*sp == '.') { 411 fprintf(stderr, "percentages must be integers\n"); 412 return (false); 413 } 414 /* 415 * If the number isn't followed by a '%', then 416 * it was not a percentage, just the first part 417 * of a file name which starts with digits. 418 */ 419 if (*sp != '%') { 420 percent = NO_PROB; 421 sp = files[i]; 422 } 423 else if (*++sp == '\0') { 424 if (++i >= file_cnt) { 425 fprintf(stderr, "percentages must precede files\n"); 426 return (false); 427 } 428 sp = files[i]; 429 } 430 } 431 if (strcmp(sp, "all") == 0) 432 sp = FORTDIR; 433 if (!add_file(percent, sp, NULL, &File_list, &File_tail, NULL)) 434 return (false); 435 } 436 437 return (true); 438 } 439 440 /* 441 * add_file: 442 * Add a file to the file list. 443 */ 444 bool 445 add_file(int percent, const char *file, const char *dir, 446 FILEDESC **head, FILEDESC **tail, FILEDESC *parent) 447 { 448 FILEDESC *fp; 449 int fd; 450 const char *path; 451 char *tpath, *offensive; 452 bool was_malloc; 453 bool isdir; 454 455 if (dir == NULL) { 456 path = file; 457 tpath = NULL; 458 was_malloc = false; 459 } 460 else { 461 tpath = do_malloc((unsigned int)(strlen(dir) + strlen(file) + 2)); 462 strcat(strcat(strcpy(tpath, dir), "/"), file); 463 path = tpath; 464 was_malloc = true; 465 } 466 if ((isdir = is_dir(path)) && parent != NULL) { 467 if (was_malloc) 468 free(tpath); 469 return (false); /* don't recurse */ 470 } 471 offensive = NULL; 472 if (!isdir && parent == NULL && (All_forts || Offend) && 473 !is_off_name(path)) { 474 offensive = off_name(path); 475 if (Offend) { 476 if (was_malloc) 477 free(tpath); 478 path = offensive; 479 offensive = NULL; 480 was_malloc = true; 481 DPRINTF(1, (stderr, "\ttrying \"%s\"\n", path)); 482 file = off_name(file); 483 } 484 } 485 486 DPRINTF(1, (stderr, "adding file \"%s\"\n", path)); 487 over: 488 if ((fd = open(path, O_RDONLY)) < 0) { 489 /* 490 * This is a sneak. If the user said -a, and if the 491 * file we're given isn't a file, we check to see if 492 * there is a -o version. If there is, we treat it as 493 * if *that* were the file given. We only do this for 494 * individual files -- if we're scanning a directory, 495 * we'll pick up the -o file anyway. 496 */ 497 if (All_forts && offensive != NULL) { 498 if (was_malloc) 499 free(tpath); 500 path = offensive; 501 offensive = NULL; 502 was_malloc = true; 503 DPRINTF(1, (stderr, "\ttrying \"%s\"\n", path)); 504 file = off_name(file); 505 goto over; 506 } 507 if (dir == NULL && file[0] != '/') 508 return (add_file(percent, file, FORTDIR, head, tail, 509 parent)); 510 if (parent == NULL) 511 perror(path); 512 if (was_malloc) 513 free(tpath); 514 return (false); 515 } 516 517 DPRINTF(2, (stderr, "path = \"%s\"\n", path)); 518 519 fp = new_fp(); 520 fp->fd = fd; 521 fp->percent = percent; 522 fp->name = file; 523 fp->path = path; 524 fp->parent = parent; 525 526 if ((isdir && !add_dir(fp)) || 527 (!isdir && 528 !is_fortfile(path, &fp->datfile, &fp->posfile, (parent != NULL)))) 529 { 530 if (parent == NULL) 531 fprintf(stderr, 532 "fortune:%s not a fortune file or directory\n", 533 path); 534 if (was_malloc) 535 free(tpath); 536 do_free(fp->datfile); 537 do_free(fp->posfile); 538 free(fp); 539 do_free(offensive); 540 return (false); 541 } 542 /* 543 * If the user said -a, we need to make this node a pointer to 544 * both files, if there are two. We don't need to do this if 545 * we are scanning a directory, since the scan will pick up the 546 * -o file anyway. 547 */ 548 if (All_forts && parent == NULL && !is_off_name(path)) 549 all_forts(fp, offensive); 550 if (*head == NULL) 551 *head = *tail = fp; 552 else if (fp->percent == NO_PROB) { 553 (*tail)->next = fp; 554 fp->prev = *tail; 555 *tail = fp; 556 } 557 else { 558 (*head)->prev = fp; 559 fp->next = *head; 560 *head = fp; 561 } 562 #ifdef OK_TO_WRITE_DISK 563 fp->was_pos_file = (access(fp->posfile, W_OK) >= 0); 564 #endif /* OK_TO_WRITE_DISK */ 565 566 return (true); 567 } 568 569 /* 570 * new_fp: 571 * Return a pointer to an initialized new FILEDESC. 572 */ 573 FILEDESC * 574 new_fp(void) 575 { 576 FILEDESC *fp; 577 578 fp = (FILEDESC *)do_malloc(sizeof(*fp)); 579 fp->datfd = -1; 580 fp->pos = POS_UNKNOWN; 581 fp->inf = NULL; 582 fp->fd = -1; 583 fp->percent = NO_PROB; 584 fp->read_tbl = false; 585 fp->next = NULL; 586 fp->prev = NULL; 587 fp->child = NULL; 588 fp->parent = NULL; 589 fp->datfile = NULL; 590 fp->posfile = NULL; 591 592 return (fp); 593 } 594 595 /* 596 * off_name: 597 * Return a pointer to the offensive version of a file of this name. 598 */ 599 char * 600 off_name(const char *file) 601 { 602 char *new; 603 604 new = copy(file, (unsigned int)(strlen(file) + 2)); 605 606 return (strcat(new, "-o")); 607 } 608 609 /* 610 * is_off_name: 611 * Is the file an offensive-style name? 612 */ 613 bool 614 is_off_name(const char *file) 615 { 616 int len; 617 618 len = strlen(file); 619 620 return (len >= 3 && file[len - 2] == '-' && file[len - 1] == 'o'); 621 } 622 623 /* 624 * all_forts: 625 * Modify a FILEDESC element to be the parent of two children if 626 * there are two children to be a parent of. 627 */ 628 void 629 all_forts(FILEDESC *fp, char *offensive) 630 { 631 char *sp; 632 FILEDESC *scene, *obscene; 633 int fd; 634 char *datfile, *posfile; 635 636 if (fp->child != NULL) /* this is a directory, not a file */ 637 return; 638 if (!is_fortfile(offensive, &datfile, &posfile, false)) 639 return; 640 if ((fd = open(offensive, O_RDONLY)) < 0) 641 return; 642 DPRINTF(1, (stderr, "adding \"%s\" because of -a\n", offensive)); 643 scene = new_fp(); 644 obscene = new_fp(); 645 *scene = *fp; 646 647 fp->num_children = 2; 648 fp->child = scene; 649 scene->next = obscene; 650 obscene->next = NULL; 651 scene->child = obscene->child = NULL; 652 scene->parent = obscene->parent = fp; 653 654 fp->fd = -1; 655 scene->percent = obscene->percent = NO_PROB; 656 657 obscene->fd = fd; 658 obscene->inf = NULL; 659 obscene->path = offensive; 660 if ((sp = rindex(offensive, '/')) == NULL) 661 obscene->name = offensive; 662 else 663 obscene->name = ++sp; 664 obscene->datfile = datfile; 665 obscene->posfile = posfile; 666 obscene->read_tbl = false; 667 #ifdef OK_TO_WRITE_DISK 668 obscene->was_pos_file = (access(obscene->posfile, W_OK) >= 0); 669 #endif /* OK_TO_WRITE_DISK */ 670 } 671 672 /* 673 * add_dir: 674 * Add the contents of an entire directory. 675 */ 676 bool 677 add_dir(FILEDESC *fp) 678 { 679 DIR *dir; 680 struct dirent *dirent; 681 FILEDESC *tailp; 682 char *name; 683 684 close(fp->fd); 685 fp->fd = -1; 686 if ((dir = opendir(fp->path)) == NULL) { 687 perror(fp->path); 688 return (false); 689 } 690 tailp = NULL; 691 DPRINTF(1, (stderr, "adding dir \"%s\"\n", fp->path)); 692 fp->num_children = 0; 693 while ((dirent = readdir(dir)) != NULL) { 694 if ((name = strdup(dirent->d_name)) == NULL) 695 err(1, "strdup failed"); 696 if (add_file(NO_PROB, name, fp->path, &fp->child, &tailp, fp)) 697 fp->num_children++; 698 else 699 free(name); 700 } 701 if (fp->num_children == 0) { 702 fprintf(stderr, 703 "fortune: %s: No fortune files in directory.\n", fp->path); 704 return (false); 705 } 706 707 return (true); 708 } 709 710 /* 711 * is_dir: 712 * Return true if the file is a directory, false otherwise. 713 */ 714 bool 715 is_dir(const char *file) 716 { 717 struct stat sbuf; 718 719 if (stat(file, &sbuf) < 0) 720 return (false); 721 722 return (sbuf.st_mode & S_IFDIR); 723 } 724 725 /* 726 * is_fortfile: 727 * Return true if the file is a fortune database file. We try and 728 * exclude files without reading them if possible to avoid 729 * overhead. Files which start with ".", or which have "illegal" 730 * suffixes, as contained in suflist[], are ruled out. 731 */ 732 /* ARGSUSED */ 733 bool 734 is_fortfile(const char *file, char **datp, char **posp, int check_for_offend) 735 { 736 int i; 737 const char *sp; 738 char *datfile; 739 static const char *suflist[] = { 740 /* list of "illegal" suffixes" */ 741 "dat", "pos", "c", "h", "p", "i", "f", 742 "pas", "ftn", "ins.c", "ins,pas", 743 "ins.ftn", "sml", 744 NULL 745 }; 746 747 DPRINTF(2, (stderr, "is_fortfile(%s) returns ", file)); 748 749 /* 750 * Preclude any -o files for offendable people, and any non -o 751 * files for completely offensive people. 752 */ 753 if (check_for_offend && !All_forts) { 754 i = strlen(file); 755 if (Offend ^ (file[i - 2] == '-' && file[i - 1] == 'o')) { 756 DPRINTF(2, (stderr, "false (offending file)\n")); 757 return (false); 758 } 759 } 760 761 if ((sp = rindex(file, '/')) == NULL) 762 sp = file; 763 else 764 sp++; 765 if (*sp == '.') { 766 DPRINTF(2, (stderr, "false (file starts with '.')\n")); 767 return (false); 768 } 769 if (Fortunes_only && strncmp(sp, "fortunes", 8) != 0) { 770 DPRINTF(2, (stderr, "false (check fortunes only)\n")); 771 return (false); 772 } 773 if ((sp = rindex(sp, '.')) != NULL) { 774 sp++; 775 for (i = 0; suflist[i] != NULL; i++) 776 if (strcmp(sp, suflist[i]) == 0) { 777 DPRINTF(2, (stderr, "false (file has suffix \".%s\")\n", sp)); 778 return (false); 779 } 780 } 781 782 datfile = copy(file, (unsigned int)(strlen(file) + 4)); /* +4 for ".dat" */ 783 strcat(datfile, ".dat"); 784 if (access(datfile, R_OK) < 0) { 785 DPRINTF(2, (stderr, "false (no readable \".dat\" file)\n")); 786 #ifdef DEBUG 787 if (Debug < 2) 788 DPRINTF(0, (stderr, "Warning: file \"%s\" unreadable\n", datfile)); 789 #endif 790 free(datfile); 791 return (false); 792 } 793 if (datp != NULL) 794 *datp = datfile; 795 else 796 free(datfile); 797 #ifdef OK_TO_WRITE_DISK 798 if (posp != NULL) { 799 *posp = copy(file, (unsigned int)(strlen(file) + 4)); /* +4 for ".dat" */ 800 strcat(*posp, ".pos"); 801 } 802 #else 803 if (posp != NULL) { 804 *posp = NULL; 805 } 806 #endif /* OK_TO_WRITE_DISK */ 807 DPRINTF(2, (stderr, "true\n")); 808 809 return (true); 810 } 811 812 /* 813 * copy: 814 * Return a malloc()'ed copy of the string 815 */ 816 char * 817 copy(const char *str, unsigned int len) 818 { 819 char *new, *sp; 820 821 new = do_malloc(len + 1); 822 sp = new; 823 do { 824 *sp++ = *str; 825 } while (*str++); 826 827 return (new); 828 } 829 830 /* 831 * do_malloc: 832 * Do a malloc, checking for NULL return. 833 */ 834 void * 835 do_malloc(unsigned int size) 836 { 837 void *new; 838 839 if ((new = malloc(size)) == NULL) { 840 fprintf(stderr, "fortune: out of memory.\n"); 841 exit(1); 842 } 843 844 return (new); 845 } 846 847 /* 848 * do_free: 849 * Free malloc'ed space, if any. 850 */ 851 void 852 do_free(void *ptr) 853 { 854 if (ptr != NULL) 855 free(ptr); 856 } 857 858 /* 859 * init_prob: 860 * Initialize the fortune probabilities. 861 */ 862 void 863 init_prob(void) 864 { 865 FILEDESC *fp, *last; 866 int percent, num_noprob, frac; 867 868 last = NULL; 869 /* 870 * Distribute the residual probability (if any) across all 871 * files with unspecified probability (i.e., probability of 0) 872 * (if any). 873 */ 874 875 percent = 0; 876 num_noprob = 0; 877 for (fp = File_tail; fp != NULL; fp = fp->prev) 878 if (fp->percent == NO_PROB) { 879 num_noprob++; 880 if (Equal_probs) 881 last = fp; 882 } else 883 percent += fp->percent; 884 DPRINTF(1, (stderr, "summing probabilities:%d%% with %d NO_PROB's", 885 percent, num_noprob)); 886 if (percent > 100) { 887 fprintf(stderr, 888 "fortune: probabilities sum to %d%% > 100%%!\n", percent); 889 exit(1); 890 } else if (percent < 100 && num_noprob == 0) { 891 fprintf(stderr, 892 "fortune: no place to put residual probability (%d%% < 100%%)\n", 893 percent); 894 exit(1); 895 } else if (percent == 100 && num_noprob != 0) { 896 fprintf(stderr, 897 "fortune: no probability left to put in residual files (100%%)\n"); 898 exit(1); 899 } 900 percent = 100 - percent; 901 if (Equal_probs) { 902 if (num_noprob != 0) { 903 if (num_noprob > 1) { 904 frac = percent / num_noprob; 905 DPRINTF(1, (stderr, ", frac = %d%%", frac)); 906 for (fp = File_list; fp != last; fp = fp->next) 907 if (fp->percent == NO_PROB) { 908 fp->percent = frac; 909 percent -= frac; 910 } 911 } 912 last->percent = percent; 913 DPRINTF(1, (stderr, ", residual = %d%%", percent)); 914 } 915 } else { 916 DPRINTF(1, (stderr, 917 ", %d%% distributed over remaining fortunes\n", 918 percent)); 919 } 920 DPRINTF(1, (stderr, "\n")); 921 922 #ifdef DEBUG 923 if (Debug >= 1) 924 print_file_list(); 925 #endif 926 } 927 928 /* 929 * get_fort: 930 * Get the fortune data file's seek pointer for the next fortune. 931 */ 932 void 933 get_fort(void) 934 { 935 FILEDESC *fp; 936 int choice; 937 938 if (File_list->next == NULL || File_list->percent == NO_PROB) 939 fp = File_list; 940 else { 941 choice = random() % 100; 942 DPRINTF(1, (stderr, "choice = %d\n", choice)); 943 for (fp = File_list; fp->percent != NO_PROB; fp = fp->next) 944 if (choice < fp->percent) 945 break; 946 else { 947 choice -= fp->percent; 948 DPRINTF(1, (stderr, 949 " skip \"%s\", %d%% (choice = %d)\n", 950 fp->name, fp->percent, choice)); 951 } 952 DPRINTF(1, (stderr, "using \"%s\", %d%% (choice = %d)\n", 953 fp->name, fp->percent, choice)); 954 } 955 if (fp->percent != NO_PROB) 956 get_tbl(fp); 957 else { 958 if (fp->next != NULL) { 959 sum_noprobs(fp); 960 choice = random() % Noprob_tbl.str_numstr; 961 DPRINTF(1, (stderr, "choice = %d (of %ld) \n", choice, 962 Noprob_tbl.str_numstr)); 963 while ((unsigned int)choice >= fp->tbl.str_numstr) { 964 choice -= fp->tbl.str_numstr; 965 fp = fp->next; 966 DPRINTF(1, (stderr, 967 " skip \"%s\", %ld (choice = %d)\n", 968 fp->name, fp->tbl.str_numstr, 969 choice)); 970 } 971 DPRINTF(1, (stderr, "using \"%s\", %ld\n", fp->name, 972 fp->tbl.str_numstr)); 973 } 974 get_tbl(fp); 975 } 976 if (fp->child != NULL) { 977 DPRINTF(1, (stderr, "picking child\n")); 978 fp = pick_child(fp); 979 } 980 Fortfile = fp; 981 get_pos(fp); 982 open_dat(fp); 983 lseek(fp->datfd, 984 (off_t) (sizeof fp->tbl + fp->pos * sizeof Seekpts[0]), SEEK_SET); 985 read(fp->datfd, Seekpts, sizeof Seekpts); 986 Seekpts[0] = ntohl(Seekpts[0]); 987 Seekpts[1] = ntohl(Seekpts[1]); 988 } 989 990 /* 991 * pick_child 992 * Pick a child from a chosen parent. 993 */ 994 FILEDESC * 995 pick_child(FILEDESC *parent) 996 { 997 FILEDESC *fp; 998 int choice; 999 1000 if (Equal_probs) { 1001 choice = random() % parent->num_children; 1002 DPRINTF(1, (stderr, " choice = %d (of %d)\n", 1003 choice, parent->num_children)); 1004 for (fp = parent->child; choice--; fp = fp->next) 1005 continue; 1006 DPRINTF(1, (stderr, " using %s\n", fp->name)); 1007 return (fp); 1008 } 1009 else { 1010 get_tbl(parent); 1011 choice = random() % parent->tbl.str_numstr; 1012 DPRINTF(1, (stderr, " choice = %d (of %ld)\n", 1013 choice, parent->tbl.str_numstr)); 1014 for (fp = parent->child; (unsigned int)choice >= fp->tbl.str_numstr; 1015 fp = fp->next) { 1016 choice -= fp->tbl.str_numstr; 1017 DPRINTF(1, (stderr, "\tskip %s, %ld (choice = %d)\n", 1018 fp->name, fp->tbl.str_numstr, choice)); 1019 } 1020 DPRINTF(1, (stderr, " using %s, %ld\n", fp->name, 1021 fp->tbl.str_numstr)); 1022 return (fp); 1023 } 1024 } 1025 1026 /* 1027 * sum_noprobs: 1028 * Sum up all the noprob probabilities, starting with fp. 1029 */ 1030 void 1031 sum_noprobs(FILEDESC *fp) 1032 { 1033 static bool did_noprobs = false; 1034 1035 if (did_noprobs) 1036 return; 1037 zero_tbl(&Noprob_tbl); 1038 while (fp != NULL) { 1039 get_tbl(fp); 1040 sum_tbl(&Noprob_tbl, &fp->tbl); 1041 fp = fp->next; 1042 } 1043 did_noprobs = true; 1044 } 1045 1046 int 1047 max(int i, int j) 1048 { 1049 return (i >= j ? i : j); 1050 } 1051 1052 /* 1053 * open_fp: 1054 * Assocatiate a FILE * with the given FILEDESC. 1055 */ 1056 void 1057 open_fp(FILEDESC *fp) 1058 { 1059 if (fp->inf == NULL && (fp->inf = fdopen(fp->fd, "r")) == NULL) { 1060 perror(fp->path); 1061 exit(1); 1062 } 1063 } 1064 1065 /* 1066 * open_dat: 1067 * Open up the dat file if we need to. 1068 */ 1069 void 1070 open_dat(FILEDESC *fp) 1071 { 1072 if (fp->datfd < 0 && (fp->datfd = open(fp->datfile, O_RDONLY)) < 0) { 1073 perror(fp->datfile); 1074 exit(1); 1075 } 1076 } 1077 1078 /* 1079 * get_pos: 1080 * Get the position from the pos file, if there is one. If not, 1081 * return a random number. 1082 */ 1083 void 1084 get_pos(FILEDESC *fp) 1085 { 1086 #ifdef OK_TO_WRITE_DISK 1087 int fd; 1088 #endif /* OK_TO_WRITE_DISK */ 1089 1090 assert(fp->read_tbl); 1091 if (fp->pos == POS_UNKNOWN) { 1092 #ifdef OK_TO_WRITE_DISK 1093 if ((fd = open(fp->posfile, O_RDONLY)) < 0 || 1094 read(fd, &fp->pos, sizeof fp->pos) != sizeof fp->pos) 1095 fp->pos = random() % fp->tbl.str_numstr; 1096 else if (fp->pos >= fp->tbl.str_numstr) 1097 fp->pos %= fp->tbl.str_numstr; 1098 if (fd >= 0) 1099 close(fd); 1100 #else 1101 fp->pos = random() % fp->tbl.str_numstr; 1102 #endif /* OK_TO_WRITE_DISK */ 1103 } 1104 if ((unsigned int)++(fp->pos) >= fp->tbl.str_numstr) 1105 fp->pos -= fp->tbl.str_numstr; 1106 DPRINTF(1, (stderr, "pos for %s is %ld\n", fp->name, fp->pos)); 1107 } 1108 1109 /* 1110 * get_tbl: 1111 * Get the tbl data file the datfile. 1112 */ 1113 void 1114 get_tbl(FILEDESC *fp) 1115 { 1116 int fd; 1117 FILEDESC *child; 1118 1119 if (fp->read_tbl) 1120 return; 1121 if (fp->child == NULL) { 1122 if ((fd = open(fp->datfile, O_RDONLY)) < 0) { 1123 perror(fp->datfile); 1124 exit(1); 1125 } 1126 if (read(fd, (char *) &fp->tbl, sizeof fp->tbl) != sizeof fp->tbl) { 1127 fprintf(stderr, 1128 "fortune: %s corrupted\n", fp->path); 1129 exit(1); 1130 } 1131 /* fp->tbl.str_version = ntohl(fp->tbl.str_version); */ 1132 fp->tbl.str_numstr = ntohl(fp->tbl.str_numstr); 1133 fp->tbl.str_longlen = ntohl(fp->tbl.str_longlen); 1134 fp->tbl.str_shortlen = ntohl(fp->tbl.str_shortlen); 1135 fp->tbl.str_flags = ntohl(fp->tbl.str_flags); 1136 close(fd); 1137 } 1138 else { 1139 zero_tbl(&fp->tbl); 1140 for (child = fp->child; child != NULL; child = child->next) { 1141 get_tbl(child); 1142 sum_tbl(&fp->tbl, &child->tbl); 1143 } 1144 } 1145 fp->read_tbl = true; 1146 } 1147 1148 /* 1149 * zero_tbl: 1150 * Zero out the fields we care about in a tbl structure. 1151 */ 1152 void 1153 zero_tbl(STRFILE *tp) 1154 { 1155 tp->str_numstr = 0; 1156 tp->str_longlen = 0; 1157 tp->str_shortlen = ~((unsigned long)0); 1158 } 1159 1160 /* 1161 * sum_tbl: 1162 * Merge the tbl data of t2 into t1. 1163 */ 1164 void 1165 sum_tbl(STRFILE *t1, STRFILE *t2) 1166 { 1167 t1->str_numstr += t2->str_numstr; 1168 if (t1->str_longlen < t2->str_longlen) 1169 t1->str_longlen = t2->str_longlen; 1170 if (t1->str_shortlen > t2->str_shortlen) 1171 t1->str_shortlen = t2->str_shortlen; 1172 } 1173 1174 #define STR(str) ((str) == NULL ? "NULL" : (str)) 1175 1176 /* 1177 * print_file_list: 1178 * Print out the file list 1179 */ 1180 void 1181 print_file_list(void) 1182 { 1183 print_list(File_list, 0); 1184 } 1185 1186 /* 1187 * print_list: 1188 * Print out the actual list, recursively. 1189 */ 1190 void 1191 print_list(FILEDESC *list, int lev) 1192 { 1193 while (list != NULL) { 1194 fprintf(stderr, "%*s", lev * 4, ""); 1195 if (list->percent == NO_PROB) 1196 fprintf(stderr, "___%%"); 1197 else 1198 fprintf(stderr, "%3d%%", list->percent); 1199 fprintf(stderr, " %s", STR(list->name)); 1200 DPRINTF(1, (stderr, " (%s, %s, %s)", STR(list->path), 1201 STR(list->datfile), STR(list->posfile))); 1202 fprintf(stderr, "\n"); 1203 if (list->child != NULL) 1204 print_list(list->child, lev + 1); 1205 list = list->next; 1206 } 1207 } 1208 1209 #ifndef NO_REGEX 1210 /* 1211 * conv_pat: 1212 * Convert the pattern to an ignore-case equivalent. 1213 */ 1214 char * 1215 conv_pat(char *orig) 1216 { 1217 char *sp; 1218 unsigned int cnt; 1219 char *new; 1220 1221 cnt = 1; /* allow for '\0' */ 1222 for (sp = orig; *sp != '\0'; sp++) 1223 if (isalpha((unsigned char)*sp)) 1224 cnt += 4; 1225 else 1226 cnt++; 1227 if ((new = malloc(cnt)) == NULL) { 1228 fprintf(stderr, "pattern too long for ignoring case\n"); 1229 exit(1); 1230 } 1231 1232 for (sp = new; *orig != '\0'; orig++) { 1233 if (islower((unsigned char)*orig)) { 1234 *sp++ = '['; 1235 *sp++ = *orig; 1236 *sp++ = toupper((unsigned char)*orig); 1237 *sp++ = ']'; 1238 } 1239 else if (isupper((unsigned char)*orig)) { 1240 *sp++ = '['; 1241 *sp++ = *orig; 1242 *sp++ = tolower((unsigned char)*orig); 1243 *sp++ = ']'; 1244 } 1245 else 1246 *sp++ = *orig; 1247 } 1248 *sp = '\0'; 1249 1250 return (new); 1251 } 1252 1253 /* 1254 * find_matches: 1255 * Find all the fortunes which match the pattern we've been given. 1256 */ 1257 int 1258 find_matches(void) 1259 { 1260 Fort_len = maxlen_in_list(File_list); 1261 DPRINTF(2, (stderr, "Maximum length is %d\n", Fort_len)); 1262 /* extra length, "%\n" is appended */ 1263 Fortbuf = do_malloc((unsigned int)Fort_len + 10); 1264 1265 Found_one = false; 1266 matches_in_list(File_list); 1267 1268 return (Found_one); 1269 } 1270 1271 /* 1272 * maxlen_in_list 1273 * Return the maximum fortune len in the file list. 1274 */ 1275 int 1276 maxlen_in_list(FILEDESC *list) 1277 { 1278 FILEDESC *fp; 1279 int len, maxlen; 1280 1281 maxlen = 0; 1282 for (fp = list; fp != NULL; fp = fp->next) { 1283 if (fp->child != NULL) { 1284 if ((len = maxlen_in_list(fp->child)) > maxlen) 1285 maxlen = len; 1286 } 1287 else { 1288 get_tbl(fp); 1289 if (fp->tbl.str_longlen > (unsigned int)maxlen) 1290 maxlen = fp->tbl.str_longlen; 1291 } 1292 } 1293 1294 return (maxlen); 1295 } 1296 1297 /* 1298 * matches_in_list 1299 * Print out the matches from the files in the list. 1300 */ 1301 void 1302 matches_in_list(FILEDESC *list) 1303 { 1304 char *sp, *p; 1305 FILEDESC *fp; 1306 int in_file; 1307 unsigned char ch; 1308 1309 for (fp = list; fp != NULL; fp = fp->next) { 1310 if (fp->child != NULL) { 1311 matches_in_list(fp->child); 1312 continue; 1313 } 1314 DPRINTF(1, (stderr, "searching in %s\n", fp->path)); 1315 open_fp(fp); 1316 sp = Fortbuf; 1317 in_file = false; 1318 while (fgets(sp, Fort_len, fp->inf) != NULL) 1319 if (fp->tbl.str_flags & STR_COMMENTS 1320 && sp[0] == fp->tbl.str_delim 1321 && sp[1] == fp->tbl.str_delim) 1322 continue; 1323 else if (!STR_ENDSTRING(sp, fp->tbl)) 1324 sp += strlen(sp); 1325 else { 1326 *sp = '\0'; 1327 if (fp->tbl.str_flags & STR_ROTATED) 1328 for (p = Fortbuf; (ch = *p) != '\0'; ++p) { 1329 if (isascii(ch)) { 1330 if (isupper(ch)) 1331 *p = 'A' + (ch - 'A' + 13) % 26; 1332 else if (islower(ch)) 1333 *p = 'a' + (ch - 'a' + 13) % 26; 1334 } 1335 } 1336 if (RE_EXEC(Fortbuf)) { 1337 printf("%c%c", fp->tbl.str_delim, 1338 fp->tbl.str_delim); 1339 if (!in_file) { 1340 printf(" (%s)", fp->name); 1341 Found_one = true; 1342 in_file = true; 1343 } 1344 putchar('\n'); 1345 fwrite(Fortbuf, 1, (sp - Fortbuf), stdout); 1346 } 1347 sp = Fortbuf; 1348 } 1349 } 1350 } 1351 #endif /* NO_REGEX */ 1352 1353 void 1354 usage(void) 1355 { 1356 fprintf(stderr, "fortune [-a"); 1357 #ifdef DEBUG 1358 fprintf(stderr, "D"); 1359 #endif /* DEBUG */ 1360 fprintf(stderr, "f"); 1361 #ifndef NO_REGEX 1362 fprintf(stderr, "i"); 1363 #endif 1364 fprintf(stderr, "losw]"); 1365 #ifndef NO_REGEX 1366 fprintf(stderr, " [-m pattern]"); 1367 #endif 1368 fprintf(stderr, "[[#%%] file/directory/all]\n"); 1369 exit(1); 1370 } 1371