16ae1554aSColin Percival /*-
26ae1554aSColin Percival * Copyright (c) 1986, 1993
36ae1554aSColin Percival * The Regents of the University of California. All rights reserved.
46ae1554aSColin Percival *
56ae1554aSColin Percival * This code is derived from software contributed to Berkeley by
66ae1554aSColin Percival * Ken Arnold.
76ae1554aSColin Percival *
86ae1554aSColin Percival * Redistribution and use in source and binary forms, with or without
96ae1554aSColin Percival * modification, are permitted provided that the following conditions
106ae1554aSColin Percival * are met:
116ae1554aSColin Percival * 1. Redistributions of source code must retain the above copyright
126ae1554aSColin Percival * notice, this list of conditions and the following disclaimer.
136ae1554aSColin Percival * 2. Redistributions in binary form must reproduce the above copyright
146ae1554aSColin Percival * notice, this list of conditions and the following disclaimer in the
156ae1554aSColin Percival * documentation and/or other materials provided with the distribution.
166ae1554aSColin Percival * 3. Neither the name of the University nor the names of its contributors
176ae1554aSColin Percival * may be used to endorse or promote products derived from this software
186ae1554aSColin Percival * without specific prior written permission.
196ae1554aSColin Percival *
206ae1554aSColin Percival * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
216ae1554aSColin Percival * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
226ae1554aSColin Percival * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
236ae1554aSColin Percival * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
246ae1554aSColin Percival * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
256ae1554aSColin Percival * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
266ae1554aSColin Percival * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
276ae1554aSColin Percival * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
286ae1554aSColin Percival * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
296ae1554aSColin Percival * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
306ae1554aSColin Percival * SUCH DAMAGE.
316ae1554aSColin Percival */
326ae1554aSColin Percival
336ae1554aSColin Percival #include <sys/stat.h>
346ae1554aSColin Percival #include <sys/endian.h>
356ae1554aSColin Percival
366ae1554aSColin Percival #include <assert.h>
376ae1554aSColin Percival #include <ctype.h>
386ae1554aSColin Percival #include <dirent.h>
396ae1554aSColin Percival #include <fcntl.h>
406ae1554aSColin Percival #include <locale.h>
416ae1554aSColin Percival #include <regex.h>
426ae1554aSColin Percival #include <stdbool.h>
436ae1554aSColin Percival #include <stdio.h>
446ae1554aSColin Percival #include <stdlib.h>
456ae1554aSColin Percival #include <string.h>
466ae1554aSColin Percival #include <time.h>
476ae1554aSColin Percival #include <unistd.h>
486ae1554aSColin Percival
496ae1554aSColin Percival #include "strfile.h"
506ae1554aSColin Percival #include "pathnames.h"
516ae1554aSColin Percival
526ae1554aSColin Percival #define TRUE true
536ae1554aSColin Percival #define FALSE false
546ae1554aSColin Percival
556ae1554aSColin Percival #define MINW 6 /* minimum wait if desired */
566ae1554aSColin Percival #define CPERS 20 /* # of chars for each sec */
576ae1554aSColin Percival #define SLEN 160 /* # of chars in short fortune */
586ae1554aSColin Percival
596ae1554aSColin Percival #define POS_UNKNOWN ((uint32_t) -1) /* pos for file unknown */
606ae1554aSColin Percival #define NO_PROB (-1) /* no prob specified for file */
616ae1554aSColin Percival
626ae1554aSColin Percival #ifdef DEBUG
636ae1554aSColin Percival #define DPRINTF(l,x) { if (Debug >= l) fprintf x; }
646ae1554aSColin Percival #undef NDEBUG
656ae1554aSColin Percival #else
666ae1554aSColin Percival #define DPRINTF(l,x)
676ae1554aSColin Percival #define NDEBUG 1
686ae1554aSColin Percival #endif
696ae1554aSColin Percival
706ae1554aSColin Percival typedef struct fd {
716ae1554aSColin Percival int percent;
726ae1554aSColin Percival int fd, datfd;
736ae1554aSColin Percival uint32_t pos;
746ae1554aSColin Percival FILE *inf;
756ae1554aSColin Percival const char *name;
766ae1554aSColin Percival const char *path;
776ae1554aSColin Percival char *datfile, *posfile;
786ae1554aSColin Percival bool read_tbl;
796ae1554aSColin Percival bool was_pos_file;
806ae1554aSColin Percival STRFILE tbl;
816ae1554aSColin Percival int num_children;
826ae1554aSColin Percival struct fd *child, *parent;
836ae1554aSColin Percival struct fd *next, *prev;
846ae1554aSColin Percival } FILEDESC;
856ae1554aSColin Percival
866ae1554aSColin Percival static bool Found_one; /* did we find a match? */
876ae1554aSColin Percival static bool Find_files = FALSE; /* just find a list of proper fortune files */
886ae1554aSColin Percival static bool Fortunes_only = FALSE; /* check only "fortunes" files */
896ae1554aSColin Percival static bool Wait = FALSE; /* wait desired after fortune */
906ae1554aSColin Percival static bool Short_only = FALSE; /* short fortune desired */
916ae1554aSColin Percival static bool Long_only = FALSE; /* long fortune desired */
926ae1554aSColin Percival static bool Offend = FALSE; /* offensive fortunes only */
936ae1554aSColin Percival static bool All_forts = FALSE; /* any fortune allowed */
946ae1554aSColin Percival static bool Equal_probs = FALSE; /* scatter un-allocted prob equally */
956ae1554aSColin Percival static bool Match = FALSE; /* dump fortunes matching a pattern */
966ae1554aSColin Percival static bool WriteToDisk = false; /* use files on disk to save state */
976ae1554aSColin Percival #ifdef DEBUG
986ae1554aSColin Percival static int Debug = 0; /* print debug messages */
996ae1554aSColin Percival #endif
1006ae1554aSColin Percival
1016ae1554aSColin Percival static char *Fortbuf = NULL; /* fortune buffer for -m */
1026ae1554aSColin Percival
1036ae1554aSColin Percival static int Fort_len = 0;
1046ae1554aSColin Percival
1056ae1554aSColin Percival static off_t Seekpts[2]; /* seek pointers to fortunes */
1066ae1554aSColin Percival
1076ae1554aSColin Percival static FILEDESC *File_list = NULL, /* Head of file list */
1086ae1554aSColin Percival *File_tail = NULL; /* Tail of file list */
1096ae1554aSColin Percival static FILEDESC *Fortfile; /* Fortune file to use */
1106ae1554aSColin Percival
1116ae1554aSColin Percival static STRFILE Noprob_tbl; /* sum of data for all no prob files */
1126ae1554aSColin Percival
1136ae1554aSColin Percival static const char *Fortune_path;
1146ae1554aSColin Percival static char **Fortune_path_arr;
1156ae1554aSColin Percival
1166ae1554aSColin Percival static int add_dir(FILEDESC *);
1176ae1554aSColin Percival static int add_file(int, const char *, const char *, FILEDESC **,
1186ae1554aSColin Percival FILEDESC **, FILEDESC *);
1196ae1554aSColin Percival static void all_forts(FILEDESC *, char *);
1206ae1554aSColin Percival static char *copy(const char *, u_int);
1216ae1554aSColin Percival static void display(FILEDESC *);
1226ae1554aSColin Percival static void do_free(void *);
1236ae1554aSColin Percival static void *do_malloc(u_int);
1246ae1554aSColin Percival static int form_file_list(char **, int);
1256ae1554aSColin Percival static int fortlen(void);
1266ae1554aSColin Percival static void get_fort(void);
1276ae1554aSColin Percival static void get_pos(FILEDESC *);
1286ae1554aSColin Percival static void get_tbl(FILEDESC *);
1296ae1554aSColin Percival static void getargs(int, char *[]);
1306ae1554aSColin Percival static void getpath(void);
1316ae1554aSColin Percival static void init_prob(void);
1326ae1554aSColin Percival static int is_dir(const char *);
1336ae1554aSColin Percival static int is_fortfile(const char *, char **, char **, int);
1346ae1554aSColin Percival static int is_off_name(const char *);
1356ae1554aSColin Percival static int max(int, int);
1366ae1554aSColin Percival static FILEDESC *new_fp(void);
1376ae1554aSColin Percival static char *off_name(const char *);
1386ae1554aSColin Percival static void open_dat(FILEDESC *);
1396ae1554aSColin Percival static void open_fp(FILEDESC *);
1406ae1554aSColin Percival static FILEDESC *pick_child(FILEDESC *);
1416ae1554aSColin Percival static void print_file_list(void);
1426ae1554aSColin Percival static void print_list(FILEDESC *, int);
1436ae1554aSColin Percival static void sum_noprobs(FILEDESC *);
1446ae1554aSColin Percival static void sum_tbl(STRFILE *, STRFILE *);
1456ae1554aSColin Percival static void usage(void);
1466ae1554aSColin Percival static void zero_tbl(STRFILE *);
1476ae1554aSColin Percival
1486ae1554aSColin Percival static char *conv_pat(char *);
1496ae1554aSColin Percival static int find_matches(void);
1506ae1554aSColin Percival static void matches_in_list(FILEDESC *);
1516ae1554aSColin Percival static int maxlen_in_list(FILEDESC *);
1526ae1554aSColin Percival
1536ae1554aSColin Percival static regex_t Re_pat;
1546ae1554aSColin Percival
1556ae1554aSColin Percival int
main(int argc,char * argv[])1566ae1554aSColin Percival main(int argc, char *argv[])
1576ae1554aSColin Percival {
1586ae1554aSColin Percival int fd;
1596ae1554aSColin Percival
1606ae1554aSColin Percival if (getenv("FORTUNE_SAVESTATE") != NULL)
1616ae1554aSColin Percival WriteToDisk = true;
1626ae1554aSColin Percival
1636ae1554aSColin Percival (void) setlocale(LC_ALL, "");
1646ae1554aSColin Percival
1656ae1554aSColin Percival getpath();
1666ae1554aSColin Percival getargs(argc, argv);
1676ae1554aSColin Percival
1686ae1554aSColin Percival if (Match)
1696ae1554aSColin Percival exit(find_matches() != 0);
1706ae1554aSColin Percival
1716ae1554aSColin Percival init_prob();
1726ae1554aSColin Percival do {
1736ae1554aSColin Percival get_fort();
1746ae1554aSColin Percival } while ((Short_only && fortlen() > SLEN) ||
1756ae1554aSColin Percival (Long_only && fortlen() <= SLEN));
1766ae1554aSColin Percival
1776ae1554aSColin Percival display(Fortfile);
1786ae1554aSColin Percival
1796ae1554aSColin Percival if (WriteToDisk) {
1806ae1554aSColin Percival if ((fd = creat(Fortfile->posfile, 0666)) < 0) {
1816ae1554aSColin Percival perror(Fortfile->posfile);
1826ae1554aSColin Percival exit(1);
1836ae1554aSColin Percival }
1846ae1554aSColin Percival /*
1856ae1554aSColin Percival * if we can, we exclusive lock, but since it isn't very
1866ae1554aSColin Percival * important, we just punt if we don't have easy locking
1876ae1554aSColin Percival * available.
1886ae1554aSColin Percival */
1896ae1554aSColin Percival flock(fd, LOCK_EX);
1906ae1554aSColin Percival write(fd, (char *) &Fortfile->pos, sizeof Fortfile->pos);
1916ae1554aSColin Percival if (!Fortfile->was_pos_file)
1926ae1554aSColin Percival chmod(Fortfile->path, 0666);
1936ae1554aSColin Percival flock(fd, LOCK_UN);
1946ae1554aSColin Percival }
1956ae1554aSColin Percival if (Wait) {
1966ae1554aSColin Percival if (Fort_len == 0)
1976ae1554aSColin Percival (void) fortlen();
1986ae1554aSColin Percival sleep((unsigned int) max(Fort_len / CPERS, MINW));
1996ae1554aSColin Percival }
2006ae1554aSColin Percival
2016ae1554aSColin Percival exit(0);
2026ae1554aSColin Percival }
2036ae1554aSColin Percival
2046ae1554aSColin Percival static void
display(FILEDESC * fp)2056ae1554aSColin Percival display(FILEDESC *fp)
2066ae1554aSColin Percival {
2076ae1554aSColin Percival char *p;
2086ae1554aSColin Percival unsigned char ch;
2096ae1554aSColin Percival char line[BUFSIZ];
2106ae1554aSColin Percival
2116ae1554aSColin Percival open_fp(fp);
2126ae1554aSColin Percival fseeko(fp->inf, Seekpts[0], SEEK_SET);
2136ae1554aSColin Percival for (Fort_len = 0; fgets(line, sizeof line, fp->inf) != NULL &&
2146ae1554aSColin Percival !STR_ENDSTRING(line, fp->tbl); Fort_len++) {
2156ae1554aSColin Percival if (fp->tbl.str_flags & STR_ROTATED)
2166ae1554aSColin Percival for (p = line; (ch = *p) != '\0'; ++p) {
2176ae1554aSColin Percival if (isascii(ch)) {
2186ae1554aSColin Percival if (isupper(ch))
2196ae1554aSColin Percival *p = 'A' + (ch - 'A' + 13) % 26;
2206ae1554aSColin Percival else if (islower(ch))
2216ae1554aSColin Percival *p = 'a' + (ch - 'a' + 13) % 26;
2226ae1554aSColin Percival }
2236ae1554aSColin Percival }
2246ae1554aSColin Percival if (fp->tbl.str_flags & STR_COMMENTS
2256ae1554aSColin Percival && line[0] == fp->tbl.str_delim
2266ae1554aSColin Percival && line[1] == fp->tbl.str_delim)
2276ae1554aSColin Percival continue;
2286ae1554aSColin Percival fputs(line, stdout);
2296ae1554aSColin Percival }
2306ae1554aSColin Percival (void) fflush(stdout);
2316ae1554aSColin Percival }
2326ae1554aSColin Percival
2336ae1554aSColin Percival /*
2346ae1554aSColin Percival * fortlen:
2356ae1554aSColin Percival * Return the length of the fortune.
2366ae1554aSColin Percival */
2376ae1554aSColin Percival static int
fortlen(void)2386ae1554aSColin Percival fortlen(void)
2396ae1554aSColin Percival {
2406ae1554aSColin Percival int nchar;
2416ae1554aSColin Percival char line[BUFSIZ];
2426ae1554aSColin Percival
2436ae1554aSColin Percival if (!(Fortfile->tbl.str_flags & (STR_RANDOM | STR_ORDERED)))
2446ae1554aSColin Percival nchar = (int)(Seekpts[1] - Seekpts[0]);
2456ae1554aSColin Percival else {
2466ae1554aSColin Percival open_fp(Fortfile);
2476ae1554aSColin Percival fseeko(Fortfile->inf, Seekpts[0], SEEK_SET);
2486ae1554aSColin Percival nchar = 0;
2496ae1554aSColin Percival while (fgets(line, sizeof line, Fortfile->inf) != NULL &&
2506ae1554aSColin Percival !STR_ENDSTRING(line, Fortfile->tbl))
2516ae1554aSColin Percival nchar += strlen(line);
2526ae1554aSColin Percival }
2536ae1554aSColin Percival Fort_len = nchar;
2546ae1554aSColin Percival
2556ae1554aSColin Percival return (nchar);
2566ae1554aSColin Percival }
2576ae1554aSColin Percival
2586ae1554aSColin Percival /*
2596ae1554aSColin Percival * This routine evaluates the arguments on the command line
2606ae1554aSColin Percival */
2616ae1554aSColin Percival static void
getargs(int argc,char * argv[])2626ae1554aSColin Percival getargs(int argc, char *argv[])
2636ae1554aSColin Percival {
2646ae1554aSColin Percival int ignore_case;
2656ae1554aSColin Percival char *pat;
2666ae1554aSColin Percival int ch;
2676ae1554aSColin Percival
2686ae1554aSColin Percival ignore_case = FALSE;
2696ae1554aSColin Percival pat = NULL;
2706ae1554aSColin Percival
2716ae1554aSColin Percival #ifdef DEBUG
2726ae1554aSColin Percival while ((ch = getopt(argc, argv, "aDefilm:osw")) != -1)
2736ae1554aSColin Percival #else
2746ae1554aSColin Percival while ((ch = getopt(argc, argv, "aefilm:osw")) != -1)
2756ae1554aSColin Percival #endif /* DEBUG */
2766ae1554aSColin Percival switch(ch) {
2776ae1554aSColin Percival case 'a': /* any fortune */
2786904ec03SPedro F. Giffuni All_forts = TRUE;
2796ae1554aSColin Percival break;
2806ae1554aSColin Percival #ifdef DEBUG
2816ae1554aSColin Percival case 'D':
2826ae1554aSColin Percival Debug++;
2836ae1554aSColin Percival break;
2846ae1554aSColin Percival #endif /* DEBUG */
2856904ec03SPedro F. Giffuni case 'e': /* scatter un-allocted prob equally */
2866904ec03SPedro F. Giffuni Equal_probs = TRUE;
2876ae1554aSColin Percival break;
2886ae1554aSColin Percival case 'f': /* find fortune files */
2896904ec03SPedro F. Giffuni Find_files = TRUE;
2906ae1554aSColin Percival break;
2916ae1554aSColin Percival case 'l': /* long ones only */
2926904ec03SPedro F. Giffuni Long_only = TRUE;
2936ae1554aSColin Percival Short_only = FALSE;
2946ae1554aSColin Percival break;
2956ae1554aSColin Percival case 'o': /* offensive ones only */
2966904ec03SPedro F. Giffuni Offend = TRUE;
2976ae1554aSColin Percival break;
2986ae1554aSColin Percival case 's': /* short ones only */
2996904ec03SPedro F. Giffuni Short_only = TRUE;
3006ae1554aSColin Percival Long_only = FALSE;
3016ae1554aSColin Percival break;
3026ae1554aSColin Percival case 'w': /* give time to read */
3036904ec03SPedro F. Giffuni Wait = TRUE;
3046ae1554aSColin Percival break;
3056ae1554aSColin Percival case 'm': /* dump out the fortunes */
3066904ec03SPedro F. Giffuni Match = TRUE;
3076ae1554aSColin Percival pat = optarg;
3086ae1554aSColin Percival break;
3096ae1554aSColin Percival case 'i': /* case-insensitive match */
3106ae1554aSColin Percival ignore_case++;
3116ae1554aSColin Percival break;
3126ae1554aSColin Percival case '?':
3136ae1554aSColin Percival default:
3146ae1554aSColin Percival usage();
3156ae1554aSColin Percival }
3166ae1554aSColin Percival argc -= optind;
3176ae1554aSColin Percival argv += optind;
3186ae1554aSColin Percival
3196ae1554aSColin Percival if (!form_file_list(argv, argc))
3206ae1554aSColin Percival exit(1); /* errors printed through form_file_list() */
3216ae1554aSColin Percival if (Find_files) {
3226ae1554aSColin Percival print_file_list();
3236ae1554aSColin Percival exit(0);
3246ae1554aSColin Percival }
3256ae1554aSColin Percival #ifdef DEBUG
3266ae1554aSColin Percival else if (Debug >= 1)
3276ae1554aSColin Percival print_file_list();
3286ae1554aSColin Percival #endif /* DEBUG */
3296ae1554aSColin Percival
3306ae1554aSColin Percival if (pat != NULL) {
3316ae1554aSColin Percival int error;
3326ae1554aSColin Percival
3336ae1554aSColin Percival if (ignore_case)
3346ae1554aSColin Percival pat = conv_pat(pat);
3356ae1554aSColin Percival error = regcomp(&Re_pat, pat, REG_BASIC);
3366ae1554aSColin Percival if (error) {
3376ae1554aSColin Percival fprintf(stderr, "regcomp(%s) fails\n", pat);
3386ae1554aSColin Percival exit(1);
3396ae1554aSColin Percival }
3406ae1554aSColin Percival }
3416ae1554aSColin Percival }
3426ae1554aSColin Percival
3436ae1554aSColin Percival /*
3446ae1554aSColin Percival * form_file_list:
3456ae1554aSColin Percival * Form the file list from the file specifications.
3466ae1554aSColin Percival */
3476ae1554aSColin Percival static int
form_file_list(char ** files,int file_cnt)3486ae1554aSColin Percival form_file_list(char **files, int file_cnt)
3496ae1554aSColin Percival {
3506ae1554aSColin Percival int i, percent;
3516ae1554aSColin Percival char *sp;
3526ae1554aSColin Percival char **pstr;
3536ae1554aSColin Percival
3546ae1554aSColin Percival if (file_cnt == 0) {
3556ae1554aSColin Percival if (Find_files) {
3566ae1554aSColin Percival Fortunes_only = TRUE;
3576ae1554aSColin Percival pstr = Fortune_path_arr;
3586ae1554aSColin Percival i = 0;
3596ae1554aSColin Percival while (*pstr) {
3606ae1554aSColin Percival i += add_file(NO_PROB, *pstr++, NULL,
3616ae1554aSColin Percival &File_list, &File_tail, NULL);
3626ae1554aSColin Percival }
3636ae1554aSColin Percival Fortunes_only = FALSE;
3646ae1554aSColin Percival if (!i) {
3656ae1554aSColin Percival fprintf(stderr, "No fortunes found in %s.\n",
3666ae1554aSColin Percival Fortune_path);
3676ae1554aSColin Percival }
3686ae1554aSColin Percival return (i != 0);
3696ae1554aSColin Percival } else {
3706ae1554aSColin Percival pstr = Fortune_path_arr;
3716ae1554aSColin Percival i = 0;
3726ae1554aSColin Percival while (*pstr) {
3736ae1554aSColin Percival i += add_file(NO_PROB, "fortunes", *pstr++,
3746ae1554aSColin Percival &File_list, &File_tail, NULL);
3756ae1554aSColin Percival }
3766ae1554aSColin Percival if (!i) {
3776ae1554aSColin Percival fprintf(stderr, "No fortunes found in %s.\n",
3786ae1554aSColin Percival Fortune_path);
3796ae1554aSColin Percival }
3806ae1554aSColin Percival return (i != 0);
3816ae1554aSColin Percival }
3826ae1554aSColin Percival }
3836ae1554aSColin Percival for (i = 0; i < file_cnt; i++) {
3846ae1554aSColin Percival percent = NO_PROB;
3856ae1554aSColin Percival if (!isdigit((unsigned char)files[i][0]))
3866ae1554aSColin Percival sp = files[i];
3876ae1554aSColin Percival else {
3886ae1554aSColin Percival percent = 0;
389777c9f5aSMark Johnston for (sp = files[i]; isdigit((unsigned char)*sp); sp++) {
3906ae1554aSColin Percival percent = percent * 10 + *sp - '0';
3916ae1554aSColin Percival if (percent > 100) {
3926ae1554aSColin Percival fprintf(stderr, "percentages must be <= 100\n");
3936ae1554aSColin Percival return (FALSE);
3946ae1554aSColin Percival }
395777c9f5aSMark Johnston }
3966ae1554aSColin Percival if (*sp == '.') {
3976ae1554aSColin Percival fprintf(stderr, "percentages must be integers\n");
3986ae1554aSColin Percival return (FALSE);
3996ae1554aSColin Percival }
4006ae1554aSColin Percival /*
4016ae1554aSColin Percival * If the number isn't followed by a '%', then
4026ae1554aSColin Percival * it was not a percentage, just the first part
4036ae1554aSColin Percival * of a file name which starts with digits.
4046ae1554aSColin Percival */
4056ae1554aSColin Percival if (*sp != '%') {
4066ae1554aSColin Percival percent = NO_PROB;
4076ae1554aSColin Percival sp = files[i];
4086ae1554aSColin Percival }
4096ae1554aSColin Percival else if (*++sp == '\0') {
4106ae1554aSColin Percival if (++i >= file_cnt) {
4116ae1554aSColin Percival fprintf(stderr, "percentages must precede files\n");
4126ae1554aSColin Percival return (FALSE);
4136ae1554aSColin Percival }
4146ae1554aSColin Percival sp = files[i];
4156ae1554aSColin Percival }
4166ae1554aSColin Percival }
4176ae1554aSColin Percival if (strcmp(sp, "all") == 0) {
4186ae1554aSColin Percival pstr = Fortune_path_arr;
4196ae1554aSColin Percival i = 0;
4206ae1554aSColin Percival while (*pstr) {
4216ae1554aSColin Percival i += add_file(NO_PROB, *pstr++, NULL,
4226ae1554aSColin Percival &File_list, &File_tail, NULL);
4236ae1554aSColin Percival }
4246ae1554aSColin Percival if (!i) {
4256ae1554aSColin Percival fprintf(stderr, "No fortunes found in %s.\n",
4266ae1554aSColin Percival Fortune_path);
4276ae1554aSColin Percival return (FALSE);
4286ae1554aSColin Percival }
4296ae1554aSColin Percival } else if (!add_file(percent, sp, NULL, &File_list,
4306ae1554aSColin Percival &File_tail, NULL)) {
4316ae1554aSColin Percival return (FALSE);
4326ae1554aSColin Percival }
4336ae1554aSColin Percival }
4346ae1554aSColin Percival
4356ae1554aSColin Percival return (TRUE);
4366ae1554aSColin Percival }
4376ae1554aSColin Percival
4386ae1554aSColin Percival /*
4396ae1554aSColin Percival * add_file:
4406ae1554aSColin Percival * Add a file to the file list.
4416ae1554aSColin Percival */
4426ae1554aSColin Percival static int
add_file(int percent,const char * file,const char * dir,FILEDESC ** head,FILEDESC ** tail,FILEDESC * parent)4436ae1554aSColin Percival add_file(int percent, const char *file, const char *dir, FILEDESC **head,
4446ae1554aSColin Percival FILEDESC **tail, FILEDESC *parent)
4456ae1554aSColin Percival {
4466ae1554aSColin Percival FILEDESC *fp;
4476ae1554aSColin Percival int fd;
4486ae1554aSColin Percival const char *path;
4496ae1554aSColin Percival char *tpath, *offensive;
4506ae1554aSColin Percival bool was_malloc;
4516ae1554aSColin Percival bool isdir;
4526ae1554aSColin Percival
4536ae1554aSColin Percival if (dir == NULL) {
4546ae1554aSColin Percival path = file;
4556ae1554aSColin Percival tpath = NULL;
4566ae1554aSColin Percival was_malloc = FALSE;
4576ae1554aSColin Percival }
4586ae1554aSColin Percival else {
4596ae1554aSColin Percival tpath = do_malloc((unsigned int)(strlen(dir) + strlen(file) + 2));
4606ae1554aSColin Percival strcat(strcat(strcpy(tpath, dir), "/"), file);
4616ae1554aSColin Percival path = tpath;
4626ae1554aSColin Percival was_malloc = TRUE;
4636ae1554aSColin Percival }
4646ae1554aSColin Percival if ((isdir = is_dir(path)) && parent != NULL) {
4656ae1554aSColin Percival if (was_malloc)
4666ae1554aSColin Percival free(tpath);
4676ae1554aSColin Percival return (FALSE); /* don't recurse */
4686ae1554aSColin Percival }
4696ae1554aSColin Percival offensive = NULL;
4706ae1554aSColin Percival if (!isdir && parent == NULL && (All_forts || Offend) &&
4716ae1554aSColin Percival !is_off_name(path)) {
4726ae1554aSColin Percival offensive = off_name(path);
4736ae1554aSColin Percival if (Offend) {
4746ae1554aSColin Percival if (was_malloc)
4756ae1554aSColin Percival free(tpath);
4766ae1554aSColin Percival path = tpath = offensive;
4776ae1554aSColin Percival offensive = NULL;
4786ae1554aSColin Percival was_malloc = TRUE;
4796ae1554aSColin Percival DPRINTF(1, (stderr, "\ttrying \"%s\"\n", path));
4806ae1554aSColin Percival file = off_name(file);
4816ae1554aSColin Percival }
4826ae1554aSColin Percival }
4836ae1554aSColin Percival
4846ae1554aSColin Percival DPRINTF(1, (stderr, "adding file \"%s\"\n", path));
4856ae1554aSColin Percival over:
4866ae1554aSColin Percival if ((fd = open(path, O_RDONLY)) < 0) {
4876ae1554aSColin Percival /*
4886ae1554aSColin Percival * This is a sneak. If the user said -a, and if the
4896ae1554aSColin Percival * file we're given isn't a file, we check to see if
4906ae1554aSColin Percival * there is a -o version. If there is, we treat it as
4916ae1554aSColin Percival * if *that* were the file given. We only do this for
4926ae1554aSColin Percival * individual files -- if we're scanning a directory,
4936ae1554aSColin Percival * we'll pick up the -o file anyway.
4946ae1554aSColin Percival */
4956ae1554aSColin Percival if (All_forts && offensive != NULL) {
4966ae1554aSColin Percival if (was_malloc)
4976ae1554aSColin Percival free(tpath);
4986ae1554aSColin Percival path = tpath = offensive;
4996ae1554aSColin Percival offensive = NULL;
5006ae1554aSColin Percival was_malloc = TRUE;
5016ae1554aSColin Percival DPRINTF(1, (stderr, "\ttrying \"%s\"\n", path));
5026ae1554aSColin Percival file = off_name(file);
5036ae1554aSColin Percival goto over;
5046ae1554aSColin Percival }
5056ae1554aSColin Percival if (dir == NULL && file[0] != '/') {
5066ae1554aSColin Percival int i = 0;
5076ae1554aSColin Percival char **pstr = Fortune_path_arr;
5086ae1554aSColin Percival
5096ae1554aSColin Percival while (*pstr) {
5106ae1554aSColin Percival i += add_file(percent, file, *pstr++,
5116ae1554aSColin Percival head, tail, parent);
5126ae1554aSColin Percival }
5136ae1554aSColin Percival if (!i) {
5146ae1554aSColin Percival fprintf(stderr, "No '%s' found in %s.\n",
5156ae1554aSColin Percival file, Fortune_path);
5166ae1554aSColin Percival }
5176ae1554aSColin Percival return (i != 0);
5186ae1554aSColin Percival }
5196ae1554aSColin Percival /*
5206ae1554aSColin Percival if (parent == NULL)
5216ae1554aSColin Percival perror(path);
5226ae1554aSColin Percival */
5236ae1554aSColin Percival if (was_malloc)
5246ae1554aSColin Percival free(tpath);
5256ae1554aSColin Percival return (FALSE);
5266ae1554aSColin Percival }
5276ae1554aSColin Percival
5286ae1554aSColin Percival DPRINTF(2, (stderr, "path = \"%s\"\n", path));
5296ae1554aSColin Percival
5306ae1554aSColin Percival fp = new_fp();
5316ae1554aSColin Percival fp->fd = fd;
5326ae1554aSColin Percival fp->percent = percent;
5336ae1554aSColin Percival fp->name = file;
5346ae1554aSColin Percival fp->path = path;
5356ae1554aSColin Percival fp->parent = parent;
5366ae1554aSColin Percival
5376ae1554aSColin Percival if ((isdir && !add_dir(fp)) ||
5386ae1554aSColin Percival (!isdir &&
5396ae1554aSColin Percival !is_fortfile(path, &fp->datfile, &fp->posfile, (parent != NULL))))
5406ae1554aSColin Percival {
5416ae1554aSColin Percival if (parent == NULL)
5426ae1554aSColin Percival fprintf(stderr,
5436ae1554aSColin Percival "fortune:%s not a fortune file or directory\n",
5446ae1554aSColin Percival path);
5456ae1554aSColin Percival if (was_malloc)
5466ae1554aSColin Percival free(tpath);
5476ae1554aSColin Percival do_free(fp->datfile);
5486ae1554aSColin Percival do_free(fp->posfile);
5496ae1554aSColin Percival free(fp);
5506ae1554aSColin Percival do_free(offensive);
5516ae1554aSColin Percival return (FALSE);
5526ae1554aSColin Percival }
5536ae1554aSColin Percival /*
5546ae1554aSColin Percival * If the user said -a, we need to make this node a pointer to
5556ae1554aSColin Percival * both files, if there are two. We don't need to do this if
5566ae1554aSColin Percival * we are scanning a directory, since the scan will pick up the
5576ae1554aSColin Percival * -o file anyway.
5586ae1554aSColin Percival */
5596ae1554aSColin Percival if (All_forts && parent == NULL && !is_off_name(path))
5606ae1554aSColin Percival all_forts(fp, offensive);
5616ae1554aSColin Percival if (*head == NULL)
5626ae1554aSColin Percival *head = *tail = fp;
5636ae1554aSColin Percival else if (fp->percent == NO_PROB) {
5646ae1554aSColin Percival (*tail)->next = fp;
5656ae1554aSColin Percival fp->prev = *tail;
5666ae1554aSColin Percival *tail = fp;
5676ae1554aSColin Percival }
5686ae1554aSColin Percival else {
5696ae1554aSColin Percival (*head)->prev = fp;
5706ae1554aSColin Percival fp->next = *head;
5716ae1554aSColin Percival *head = fp;
5726ae1554aSColin Percival }
5736ae1554aSColin Percival if (WriteToDisk)
5746ae1554aSColin Percival fp->was_pos_file = (access(fp->posfile, W_OK) >= 0);
5756ae1554aSColin Percival
5766ae1554aSColin Percival return (TRUE);
5776ae1554aSColin Percival }
5786ae1554aSColin Percival
5796ae1554aSColin Percival /*
5806ae1554aSColin Percival * new_fp:
5816ae1554aSColin Percival * Return a pointer to an initialized new FILEDESC.
5826ae1554aSColin Percival */
5836ae1554aSColin Percival static FILEDESC *
new_fp(void)5846ae1554aSColin Percival new_fp(void)
5856ae1554aSColin Percival {
5866ae1554aSColin Percival FILEDESC *fp;
5876ae1554aSColin Percival
5886ae1554aSColin Percival fp = do_malloc(sizeof(*fp));
5896ae1554aSColin Percival fp->datfd = -1;
5906ae1554aSColin Percival fp->pos = POS_UNKNOWN;
5916ae1554aSColin Percival fp->inf = NULL;
5926ae1554aSColin Percival fp->fd = -1;
5936ae1554aSColin Percival fp->percent = NO_PROB;
5946ae1554aSColin Percival fp->read_tbl = FALSE;
5956ae1554aSColin Percival fp->next = NULL;
5966ae1554aSColin Percival fp->prev = NULL;
5976ae1554aSColin Percival fp->child = NULL;
5986ae1554aSColin Percival fp->parent = NULL;
5996ae1554aSColin Percival fp->datfile = NULL;
6006ae1554aSColin Percival fp->posfile = NULL;
6016ae1554aSColin Percival
6026ae1554aSColin Percival return (fp);
6036ae1554aSColin Percival }
6046ae1554aSColin Percival
6056ae1554aSColin Percival /*
6066ae1554aSColin Percival * off_name:
6076ae1554aSColin Percival * Return a pointer to the offensive version of a file of this name.
6086ae1554aSColin Percival */
6096ae1554aSColin Percival static char *
off_name(const char * file)6106ae1554aSColin Percival off_name(const char *file)
6116ae1554aSColin Percival {
6126ae1554aSColin Percival char *new;
6136ae1554aSColin Percival
6146ae1554aSColin Percival new = copy(file, (unsigned int) (strlen(file) + 2));
6156ae1554aSColin Percival
6166ae1554aSColin Percival return (strcat(new, "-o"));
6176ae1554aSColin Percival }
6186ae1554aSColin Percival
6196ae1554aSColin Percival /*
6206ae1554aSColin Percival * is_off_name:
6216ae1554aSColin Percival * Is the file an offensive-style name?
6226ae1554aSColin Percival */
6236ae1554aSColin Percival static int
is_off_name(const char * file)6246ae1554aSColin Percival is_off_name(const char *file)
6256ae1554aSColin Percival {
6266ae1554aSColin Percival int len;
6276ae1554aSColin Percival
6286ae1554aSColin Percival len = strlen(file);
6296ae1554aSColin Percival
6306ae1554aSColin Percival return (len >= 3 && file[len - 2] == '-' && file[len - 1] == 'o');
6316ae1554aSColin Percival }
6326ae1554aSColin Percival
6336ae1554aSColin Percival /*
6346ae1554aSColin Percival * all_forts:
6356ae1554aSColin Percival * Modify a FILEDESC element to be the parent of two children if
6366ae1554aSColin Percival * there are two children to be a parent of.
6376ae1554aSColin Percival */
6386ae1554aSColin Percival static void
all_forts(FILEDESC * fp,char * offensive)6396ae1554aSColin Percival all_forts(FILEDESC *fp, char *offensive)
6406ae1554aSColin Percival {
6416ae1554aSColin Percival char *sp;
6426ae1554aSColin Percival FILEDESC *scene, *obscene;
6436ae1554aSColin Percival int fd;
6446ae1554aSColin Percival char *datfile, *posfile;
6456ae1554aSColin Percival
6466ae1554aSColin Percival if (fp->child != NULL) /* this is a directory, not a file */
6476ae1554aSColin Percival return;
6486ae1554aSColin Percival if (!is_fortfile(offensive, &datfile, &posfile, FALSE))
6496ae1554aSColin Percival return;
6506ae1554aSColin Percival if ((fd = open(offensive, O_RDONLY)) < 0)
6516ae1554aSColin Percival return;
6526ae1554aSColin Percival DPRINTF(1, (stderr, "adding \"%s\" because of -a\n", offensive));
6536ae1554aSColin Percival scene = new_fp();
6546ae1554aSColin Percival obscene = new_fp();
6556ae1554aSColin Percival *scene = *fp;
6566ae1554aSColin Percival
6576ae1554aSColin Percival fp->num_children = 2;
6586ae1554aSColin Percival fp->child = scene;
6596ae1554aSColin Percival scene->next = obscene;
6606ae1554aSColin Percival obscene->next = NULL;
6616ae1554aSColin Percival scene->child = obscene->child = NULL;
6626ae1554aSColin Percival scene->parent = obscene->parent = fp;
6636ae1554aSColin Percival
6646ae1554aSColin Percival fp->fd = -1;
6656ae1554aSColin Percival scene->percent = obscene->percent = NO_PROB;
6666ae1554aSColin Percival
6676ae1554aSColin Percival obscene->fd = fd;
6686ae1554aSColin Percival obscene->inf = NULL;
6696ae1554aSColin Percival obscene->path = offensive;
6706ae1554aSColin Percival if ((sp = strrchr(offensive, '/')) == NULL)
6716ae1554aSColin Percival obscene->name = offensive;
6726ae1554aSColin Percival else
6736ae1554aSColin Percival obscene->name = ++sp;
6746ae1554aSColin Percival obscene->datfile = datfile;
6756ae1554aSColin Percival obscene->posfile = posfile;
6766ae1554aSColin Percival obscene->read_tbl = false;
6776ae1554aSColin Percival if (WriteToDisk)
6786ae1554aSColin Percival obscene->was_pos_file = (access(obscene->posfile, W_OK) >= 0);
6796ae1554aSColin Percival }
6806ae1554aSColin Percival
6816ae1554aSColin Percival /*
6826ae1554aSColin Percival * add_dir:
6836ae1554aSColin Percival * Add the contents of an entire directory.
6846ae1554aSColin Percival */
6856ae1554aSColin Percival static int
add_dir(FILEDESC * fp)6866ae1554aSColin Percival add_dir(FILEDESC *fp)
6876ae1554aSColin Percival {
6886ae1554aSColin Percival DIR *dir;
6896ae1554aSColin Percival struct dirent *dirent;
6906ae1554aSColin Percival FILEDESC *tailp;
6916ae1554aSColin Percival char *name;
6926ae1554aSColin Percival
6936ae1554aSColin Percival (void) close(fp->fd);
6946ae1554aSColin Percival fp->fd = -1;
6956ae1554aSColin Percival if ((dir = opendir(fp->path)) == NULL) {
6966ae1554aSColin Percival perror(fp->path);
6976ae1554aSColin Percival return (FALSE);
6986ae1554aSColin Percival }
6996ae1554aSColin Percival tailp = NULL;
7006ae1554aSColin Percival DPRINTF(1, (stderr, "adding dir \"%s\"\n", fp->path));
7016ae1554aSColin Percival fp->num_children = 0;
7026ae1554aSColin Percival while ((dirent = readdir(dir)) != NULL) {
7036ae1554aSColin Percival if (dirent->d_namlen == 0)
7046ae1554aSColin Percival continue;
7056ae1554aSColin Percival name = copy(dirent->d_name, dirent->d_namlen);
7066ae1554aSColin Percival if (add_file(NO_PROB, name, fp->path, &fp->child, &tailp, fp))
7076ae1554aSColin Percival fp->num_children++;
7086ae1554aSColin Percival else
7096ae1554aSColin Percival free(name);
7106ae1554aSColin Percival }
7116ae1554aSColin Percival if (fp->num_children == 0) {
7126ae1554aSColin Percival (void) fprintf(stderr,
7136ae1554aSColin Percival "fortune: %s: No fortune files in directory.\n", fp->path);
7146ae1554aSColin Percival return (FALSE);
7156ae1554aSColin Percival }
7166ae1554aSColin Percival
7176ae1554aSColin Percival return (TRUE);
7186ae1554aSColin Percival }
7196ae1554aSColin Percival
7206ae1554aSColin Percival /*
7216ae1554aSColin Percival * is_dir:
7226ae1554aSColin Percival * Return TRUE if the file is a directory, FALSE otherwise.
7236ae1554aSColin Percival */
7246ae1554aSColin Percival static int
is_dir(const char * file)7256ae1554aSColin Percival is_dir(const char *file)
7266ae1554aSColin Percival {
7276ae1554aSColin Percival struct stat sbuf;
7286ae1554aSColin Percival
7296ae1554aSColin Percival if (stat(file, &sbuf) < 0)
7306ae1554aSColin Percival return (FALSE);
7316ae1554aSColin Percival
7326ae1554aSColin Percival return (sbuf.st_mode & S_IFDIR);
7336ae1554aSColin Percival }
7346ae1554aSColin Percival
7356ae1554aSColin Percival /*
7366ae1554aSColin Percival * is_fortfile:
7376ae1554aSColin Percival * Return TRUE if the file is a fortune database file. We try and
7386ae1554aSColin Percival * exclude files without reading them if possible to avoid
7396ae1554aSColin Percival * overhead. Files which start with ".", or which have "illegal"
7406ae1554aSColin Percival * suffixes, as contained in suflist[], are ruled out.
7416ae1554aSColin Percival */
7426ae1554aSColin Percival /* ARGSUSED */
7436ae1554aSColin Percival static int
is_fortfile(const char * file,char ** datp,char ** posp,int check_for_offend)7446ae1554aSColin Percival is_fortfile(const char *file, char **datp, char **posp, int check_for_offend)
7456ae1554aSColin Percival {
7466ae1554aSColin Percival int i;
7476ae1554aSColin Percival const char *sp;
7486ae1554aSColin Percival char *datfile;
7496ae1554aSColin Percival static const char *suflist[] = {
7506ae1554aSColin Percival /* list of "illegal" suffixes" */
7516ae1554aSColin Percival "dat", "pos", "c", "h", "p", "i", "f",
7526ae1554aSColin Percival "pas", "ftn", "ins.c", "ins,pas",
7536ae1554aSColin Percival "ins.ftn", "sml",
7546ae1554aSColin Percival NULL
7556ae1554aSColin Percival };
7566ae1554aSColin Percival
7576ae1554aSColin Percival DPRINTF(2, (stderr, "is_fortfile(%s) returns ", file));
7586ae1554aSColin Percival
7596ae1554aSColin Percival /*
7606ae1554aSColin Percival * Preclude any -o files for offendable people, and any non -o
7616ae1554aSColin Percival * files for completely offensive people.
7626ae1554aSColin Percival */
7636ae1554aSColin Percival if (check_for_offend && !All_forts) {
7646ae1554aSColin Percival i = strlen(file);
7656ae1554aSColin Percival if (Offend ^ (file[i - 2] == '-' && file[i - 1] == 'o')) {
7666ae1554aSColin Percival DPRINTF(2, (stderr, "FALSE (offending file)\n"));
7676ae1554aSColin Percival return (FALSE);
7686ae1554aSColin Percival }
7696ae1554aSColin Percival }
7706ae1554aSColin Percival
7716ae1554aSColin Percival if ((sp = strrchr(file, '/')) == NULL)
7726ae1554aSColin Percival sp = file;
7736ae1554aSColin Percival else
7746ae1554aSColin Percival sp++;
7756ae1554aSColin Percival if (*sp == '.') {
7766ae1554aSColin Percival DPRINTF(2, (stderr, "FALSE (file starts with '.')\n"));
7776ae1554aSColin Percival return (FALSE);
7786ae1554aSColin Percival }
7796ae1554aSColin Percival if (Fortunes_only && strncmp(sp, "fortunes", 8) != 0) {
7806ae1554aSColin Percival DPRINTF(2, (stderr, "FALSE (check fortunes only)\n"));
7816ae1554aSColin Percival return (FALSE);
7826ae1554aSColin Percival }
7836ae1554aSColin Percival if ((sp = strrchr(sp, '.')) != NULL) {
7846ae1554aSColin Percival sp++;
7856ae1554aSColin Percival for (i = 0; suflist[i] != NULL; i++)
7866ae1554aSColin Percival if (strcmp(sp, suflist[i]) == 0) {
7876ae1554aSColin Percival DPRINTF(2, (stderr, "FALSE (file has suffix \".%s\")\n", sp));
7886ae1554aSColin Percival return (FALSE);
7896ae1554aSColin Percival }
7906ae1554aSColin Percival }
7916ae1554aSColin Percival
7926ae1554aSColin Percival datfile = copy(file, (unsigned int) (strlen(file) + 4)); /* +4 for ".dat" */
7936ae1554aSColin Percival strcat(datfile, ".dat");
7946ae1554aSColin Percival if (access(datfile, R_OK) < 0) {
7956ae1554aSColin Percival DPRINTF(2, (stderr, "FALSE (no readable \".dat\" file)\n"));
7966ae1554aSColin Percival free(datfile);
7976ae1554aSColin Percival return (FALSE);
7986ae1554aSColin Percival }
7996ae1554aSColin Percival if (datp != NULL)
8006ae1554aSColin Percival *datp = datfile;
8016ae1554aSColin Percival else
8026ae1554aSColin Percival free(datfile);
8036ae1554aSColin Percival if (posp != NULL) {
8046ae1554aSColin Percival if (WriteToDisk) {
8056ae1554aSColin Percival *posp = copy(file, (unsigned int) (strlen(file) + 4)); /* +4 for ".dat" */
8066ae1554aSColin Percival strcat(*posp, ".pos");
8076ae1554aSColin Percival }
8086ae1554aSColin Percival else {
8096ae1554aSColin Percival *posp = NULL;
8106ae1554aSColin Percival }
8116ae1554aSColin Percival }
8126ae1554aSColin Percival DPRINTF(2, (stderr, "TRUE\n"));
8136ae1554aSColin Percival
8146ae1554aSColin Percival return (TRUE);
8156ae1554aSColin Percival }
8166ae1554aSColin Percival
8176ae1554aSColin Percival /*
8186ae1554aSColin Percival * copy:
8196ae1554aSColin Percival * Return a malloc()'ed copy of the string
8206ae1554aSColin Percival */
8216ae1554aSColin Percival static char *
copy(const char * str,unsigned int len)8226ae1554aSColin Percival copy(const char *str, unsigned int len)
8236ae1554aSColin Percival {
8246ae1554aSColin Percival char *new, *sp;
8256ae1554aSColin Percival
8266ae1554aSColin Percival new = do_malloc(len + 1);
8276ae1554aSColin Percival sp = new;
8286ae1554aSColin Percival do {
8296ae1554aSColin Percival *sp++ = *str;
8306ae1554aSColin Percival } while (*str++);
8316ae1554aSColin Percival
8326ae1554aSColin Percival return (new);
8336ae1554aSColin Percival }
8346ae1554aSColin Percival
8356ae1554aSColin Percival /*
8366ae1554aSColin Percival * do_malloc:
8376ae1554aSColin Percival * Do a malloc, checking for NULL return.
8386ae1554aSColin Percival */
8396ae1554aSColin Percival static void *
do_malloc(unsigned int size)8406ae1554aSColin Percival do_malloc(unsigned int size)
8416ae1554aSColin Percival {
8426ae1554aSColin Percival void *new;
8436ae1554aSColin Percival
8446ae1554aSColin Percival if ((new = malloc(size)) == NULL) {
8456ae1554aSColin Percival (void) fprintf(stderr, "fortune: out of memory.\n");
8466ae1554aSColin Percival exit(1);
8476ae1554aSColin Percival }
8486ae1554aSColin Percival
8496ae1554aSColin Percival return (new);
8506ae1554aSColin Percival }
8516ae1554aSColin Percival
8526ae1554aSColin Percival /*
8536ae1554aSColin Percival * do_free:
8546ae1554aSColin Percival * Free malloc'ed space, if any.
8556ae1554aSColin Percival */
8566ae1554aSColin Percival static void
do_free(void * ptr)8576ae1554aSColin Percival do_free(void *ptr)
8586ae1554aSColin Percival {
8596ae1554aSColin Percival if (ptr != NULL)
8606ae1554aSColin Percival free(ptr);
8616ae1554aSColin Percival }
8626ae1554aSColin Percival
8636ae1554aSColin Percival /*
8646ae1554aSColin Percival * init_prob:
8656ae1554aSColin Percival * Initialize the fortune probabilities.
8666ae1554aSColin Percival */
8676ae1554aSColin Percival static void
init_prob(void)8686ae1554aSColin Percival init_prob(void)
8696ae1554aSColin Percival {
8706ae1554aSColin Percival FILEDESC *fp, *last = NULL;
8716ae1554aSColin Percival int percent, num_noprob, frac;
8726ae1554aSColin Percival
8736ae1554aSColin Percival /*
8746ae1554aSColin Percival * Distribute the residual probability (if any) across all
8756ae1554aSColin Percival * files with unspecified probability (i.e., probability of 0)
8766ae1554aSColin Percival * (if any).
8776ae1554aSColin Percival */
8786ae1554aSColin Percival
8796ae1554aSColin Percival percent = 0;
8806ae1554aSColin Percival num_noprob = 0;
8816ae1554aSColin Percival for (fp = File_tail; fp != NULL; fp = fp->prev)
8826ae1554aSColin Percival if (fp->percent == NO_PROB) {
8836ae1554aSColin Percival num_noprob++;
8846ae1554aSColin Percival if (Equal_probs)
8856ae1554aSColin Percival last = fp;
8866ae1554aSColin Percival } else
8876ae1554aSColin Percival percent += fp->percent;
8886ae1554aSColin Percival DPRINTF(1, (stderr, "summing probabilities:%d%% with %d NO_PROB's",
8896ae1554aSColin Percival percent, num_noprob));
8906ae1554aSColin Percival if (percent > 100) {
8916ae1554aSColin Percival (void) fprintf(stderr,
8926ae1554aSColin Percival "fortune: probabilities sum to %d%% > 100%%!\n", percent);
8936ae1554aSColin Percival exit(1);
8946ae1554aSColin Percival } else if (percent < 100 && num_noprob == 0) {
8956ae1554aSColin Percival (void) fprintf(stderr,
8966ae1554aSColin Percival "fortune: no place to put residual probability (%d%% < 100%%)\n",
8976ae1554aSColin Percival percent);
8986ae1554aSColin Percival exit(1);
8996ae1554aSColin Percival } else if (percent == 100 && num_noprob != 0) {
9006ae1554aSColin Percival (void) fprintf(stderr,
9016ae1554aSColin Percival "fortune: no probability left to put in residual files (100%%)\n");
9026ae1554aSColin Percival exit(1);
9036ae1554aSColin Percival }
9046ae1554aSColin Percival percent = 100 - percent;
9056ae1554aSColin Percival if (Equal_probs) {
9066ae1554aSColin Percival if (num_noprob != 0) {
9076ae1554aSColin Percival if (num_noprob > 1) {
9086ae1554aSColin Percival frac = percent / num_noprob;
9096ae1554aSColin Percival DPRINTF(1, (stderr, ", frac = %d%%", frac));
9106ae1554aSColin Percival for (fp = File_tail; fp != last; fp = fp->prev)
9116ae1554aSColin Percival if (fp->percent == NO_PROB) {
9126ae1554aSColin Percival fp->percent = frac;
9136ae1554aSColin Percival percent -= frac;
9146ae1554aSColin Percival }
9156ae1554aSColin Percival }
9166ae1554aSColin Percival last->percent = percent;
9176ae1554aSColin Percival DPRINTF(1, (stderr, ", residual = %d%%", percent));
9186ae1554aSColin Percival }
9196ae1554aSColin Percival else
9206ae1554aSColin Percival DPRINTF(1, (stderr,
9216ae1554aSColin Percival ", %d%% distributed over remaining fortunes\n",
9226ae1554aSColin Percival percent));
9236ae1554aSColin Percival }
9246ae1554aSColin Percival DPRINTF(1, (stderr, "\n"));
9256ae1554aSColin Percival
9266ae1554aSColin Percival #ifdef DEBUG
9276ae1554aSColin Percival if (Debug >= 1)
9286ae1554aSColin Percival print_file_list();
9296ae1554aSColin Percival #endif
9306ae1554aSColin Percival }
9316ae1554aSColin Percival
9326ae1554aSColin Percival /*
9336ae1554aSColin Percival * get_fort:
9346ae1554aSColin Percival * Get the fortune data file's seek pointer for the next fortune.
9356ae1554aSColin Percival */
9366ae1554aSColin Percival static void
get_fort(void)9376ae1554aSColin Percival get_fort(void)
9386ae1554aSColin Percival {
9396ae1554aSColin Percival FILEDESC *fp;
9406ae1554aSColin Percival int choice;
9416ae1554aSColin Percival
9426ae1554aSColin Percival if (File_list->next == NULL || File_list->percent == NO_PROB)
9436ae1554aSColin Percival fp = File_list;
9446ae1554aSColin Percival else {
9456ae1554aSColin Percival choice = arc4random_uniform(100);
9466ae1554aSColin Percival DPRINTF(1, (stderr, "choice = %d\n", choice));
947db695788SJohn Baldwin for (fp = File_list; fp->percent != NO_PROB; fp = fp->next) {
9486ae1554aSColin Percival if (choice < fp->percent)
9496ae1554aSColin Percival break;
9506ae1554aSColin Percival else {
9516ae1554aSColin Percival choice -= fp->percent;
9526ae1554aSColin Percival DPRINTF(1, (stderr,
9536ae1554aSColin Percival " skip \"%s\", %d%% (choice = %d)\n",
9546ae1554aSColin Percival fp->name, fp->percent, choice));
9556ae1554aSColin Percival }
956db695788SJohn Baldwin }
9576ae1554aSColin Percival DPRINTF(1, (stderr,
9586ae1554aSColin Percival "using \"%s\", %d%% (choice = %d)\n",
9596ae1554aSColin Percival fp->name, fp->percent, choice));
9606ae1554aSColin Percival }
9616ae1554aSColin Percival if (fp->percent != NO_PROB)
9626ae1554aSColin Percival get_tbl(fp);
9636ae1554aSColin Percival else {
9646ae1554aSColin Percival if (fp->next != NULL) {
9656ae1554aSColin Percival sum_noprobs(fp);
9666ae1554aSColin Percival choice = arc4random_uniform(Noprob_tbl.str_numstr);
9676ae1554aSColin Percival DPRINTF(1, (stderr, "choice = %d (of %u) \n", choice,
9686ae1554aSColin Percival Noprob_tbl.str_numstr));
9696ae1554aSColin Percival while ((unsigned int)choice >= fp->tbl.str_numstr) {
9706ae1554aSColin Percival choice -= fp->tbl.str_numstr;
9716ae1554aSColin Percival fp = fp->next;
9726ae1554aSColin Percival DPRINTF(1, (stderr,
9736ae1554aSColin Percival " skip \"%s\", %u (choice = %d)\n",
9746ae1554aSColin Percival fp->name, fp->tbl.str_numstr,
9756ae1554aSColin Percival choice));
9766ae1554aSColin Percival }
9776ae1554aSColin Percival DPRINTF(1, (stderr, "using \"%s\", %u\n", fp->name,
9786ae1554aSColin Percival fp->tbl.str_numstr));
9796ae1554aSColin Percival }
9806ae1554aSColin Percival get_tbl(fp);
9816ae1554aSColin Percival }
9826ae1554aSColin Percival if (fp->child != NULL) {
9836ae1554aSColin Percival DPRINTF(1, (stderr, "picking child\n"));
9846ae1554aSColin Percival fp = pick_child(fp);
9856ae1554aSColin Percival }
9866ae1554aSColin Percival Fortfile = fp;
9876ae1554aSColin Percival get_pos(fp);
9886ae1554aSColin Percival open_dat(fp);
9896ae1554aSColin Percival lseek(fp->datfd,
9906ae1554aSColin Percival (off_t) (sizeof fp->tbl + fp->pos * sizeof Seekpts[0]), SEEK_SET);
9916ae1554aSColin Percival read(fp->datfd, Seekpts, sizeof Seekpts);
9926ae1554aSColin Percival Seekpts[0] = be64toh(Seekpts[0]);
9936ae1554aSColin Percival Seekpts[1] = be64toh(Seekpts[1]);
9946ae1554aSColin Percival }
9956ae1554aSColin Percival
9966ae1554aSColin Percival /*
9976ae1554aSColin Percival * pick_child
9986ae1554aSColin Percival * Pick a child from a chosen parent.
9996ae1554aSColin Percival */
10006ae1554aSColin Percival static FILEDESC *
pick_child(FILEDESC * parent)10016ae1554aSColin Percival pick_child(FILEDESC *parent)
10026ae1554aSColin Percival {
10036ae1554aSColin Percival FILEDESC *fp;
10046ae1554aSColin Percival int choice;
10056ae1554aSColin Percival
10066ae1554aSColin Percival if (Equal_probs) {
10076ae1554aSColin Percival choice = arc4random_uniform(parent->num_children);
10086ae1554aSColin Percival DPRINTF(1, (stderr, " choice = %d (of %d)\n",
10096ae1554aSColin Percival choice, parent->num_children));
10106ae1554aSColin Percival for (fp = parent->child; choice--; fp = fp->next)
10116ae1554aSColin Percival continue;
10126ae1554aSColin Percival DPRINTF(1, (stderr, " using %s\n", fp->name));
10136ae1554aSColin Percival return (fp);
10146ae1554aSColin Percival }
10156ae1554aSColin Percival else {
10166ae1554aSColin Percival get_tbl(parent);
10176ae1554aSColin Percival choice = arc4random_uniform(parent->tbl.str_numstr);
10186ae1554aSColin Percival DPRINTF(1, (stderr, " choice = %d (of %u)\n",
10196ae1554aSColin Percival choice, parent->tbl.str_numstr));
10206ae1554aSColin Percival for (fp = parent->child; (unsigned)choice >= fp->tbl.str_numstr;
10216ae1554aSColin Percival fp = fp->next) {
10226ae1554aSColin Percival choice -= fp->tbl.str_numstr;
10236ae1554aSColin Percival DPRINTF(1, (stderr, "\tskip %s, %u (choice = %d)\n",
10246ae1554aSColin Percival fp->name, fp->tbl.str_numstr, choice));
10256ae1554aSColin Percival }
10266ae1554aSColin Percival DPRINTF(1, (stderr, " using %s, %u\n", fp->name,
10276ae1554aSColin Percival fp->tbl.str_numstr));
10286ae1554aSColin Percival return (fp);
10296ae1554aSColin Percival }
10306ae1554aSColin Percival }
10316ae1554aSColin Percival
10326ae1554aSColin Percival /*
10336ae1554aSColin Percival * sum_noprobs:
10346ae1554aSColin Percival * Sum up all the noprob probabilities, starting with fp.
10356ae1554aSColin Percival */
10366ae1554aSColin Percival static void
sum_noprobs(FILEDESC * fp)10376ae1554aSColin Percival sum_noprobs(FILEDESC *fp)
10386ae1554aSColin Percival {
10396ae1554aSColin Percival static bool did_noprobs = FALSE;
10406ae1554aSColin Percival
10416ae1554aSColin Percival if (did_noprobs)
10426ae1554aSColin Percival return;
10436ae1554aSColin Percival zero_tbl(&Noprob_tbl);
10446ae1554aSColin Percival while (fp != NULL) {
10456ae1554aSColin Percival get_tbl(fp);
10466ae1554aSColin Percival sum_tbl(&Noprob_tbl, &fp->tbl);
10476ae1554aSColin Percival fp = fp->next;
10486ae1554aSColin Percival }
10496ae1554aSColin Percival did_noprobs = TRUE;
10506ae1554aSColin Percival }
10516ae1554aSColin Percival
10526ae1554aSColin Percival static int
max(int i,int j)10536ae1554aSColin Percival max(int i, int j)
10546ae1554aSColin Percival {
10556ae1554aSColin Percival return (i >= j ? i : j);
10566ae1554aSColin Percival }
10576ae1554aSColin Percival
10586ae1554aSColin Percival /*
10596ae1554aSColin Percival * open_fp:
10606ae1554aSColin Percival * Assocatiate a FILE * with the given FILEDESC.
10616ae1554aSColin Percival */
10626ae1554aSColin Percival static void
open_fp(FILEDESC * fp)10636ae1554aSColin Percival open_fp(FILEDESC *fp)
10646ae1554aSColin Percival {
10656ae1554aSColin Percival if (fp->inf == NULL && (fp->inf = fdopen(fp->fd, "r")) == NULL) {
10666ae1554aSColin Percival perror(fp->path);
10676ae1554aSColin Percival exit(1);
10686ae1554aSColin Percival }
10696ae1554aSColin Percival }
10706ae1554aSColin Percival
10716ae1554aSColin Percival /*
10726ae1554aSColin Percival * open_dat:
10736ae1554aSColin Percival * Open up the dat file if we need to.
10746ae1554aSColin Percival */
10756ae1554aSColin Percival static void
open_dat(FILEDESC * fp)10766ae1554aSColin Percival open_dat(FILEDESC *fp)
10776ae1554aSColin Percival {
10786ae1554aSColin Percival if (fp->datfd < 0 && (fp->datfd = open(fp->datfile, O_RDONLY)) < 0) {
10796ae1554aSColin Percival perror(fp->datfile);
10806ae1554aSColin Percival exit(1);
10816ae1554aSColin Percival }
10826ae1554aSColin Percival }
10836ae1554aSColin Percival
10846ae1554aSColin Percival /*
10856ae1554aSColin Percival * get_pos:
10866ae1554aSColin Percival * Get the position from the pos file, if there is one. If not,
10876ae1554aSColin Percival * return a random number.
10886ae1554aSColin Percival */
10896ae1554aSColin Percival static void
get_pos(FILEDESC * fp)10906ae1554aSColin Percival get_pos(FILEDESC *fp)
10916ae1554aSColin Percival {
10926ae1554aSColin Percival int fd;
10936ae1554aSColin Percival
10946ae1554aSColin Percival assert(fp->read_tbl);
10956ae1554aSColin Percival if (fp->pos == POS_UNKNOWN) {
10966ae1554aSColin Percival if (WriteToDisk) {
10976ae1554aSColin Percival if ((fd = open(fp->posfile, O_RDONLY)) < 0 ||
10986ae1554aSColin Percival read(fd, &fp->pos, sizeof fp->pos) != sizeof fp->pos)
10996ae1554aSColin Percival fp->pos = arc4random_uniform(fp->tbl.str_numstr);
11006ae1554aSColin Percival else if (fp->pos >= fp->tbl.str_numstr)
11016ae1554aSColin Percival fp->pos %= fp->tbl.str_numstr;
11026ae1554aSColin Percival if (fd >= 0)
11036ae1554aSColin Percival close(fd);
11046ae1554aSColin Percival }
11056ae1554aSColin Percival else
11066ae1554aSColin Percival fp->pos = arc4random_uniform(fp->tbl.str_numstr);
11076ae1554aSColin Percival }
11086ae1554aSColin Percival if (++(fp->pos) >= fp->tbl.str_numstr)
11096ae1554aSColin Percival fp->pos -= fp->tbl.str_numstr;
11106ae1554aSColin Percival DPRINTF(1, (stderr, "pos for %s is %ld\n", fp->name, (long)fp->pos));
11116ae1554aSColin Percival }
11126ae1554aSColin Percival
11136ae1554aSColin Percival /*
11146ae1554aSColin Percival * get_tbl:
11156ae1554aSColin Percival * Get the tbl data file the datfile.
11166ae1554aSColin Percival */
11176ae1554aSColin Percival static void
get_tbl(FILEDESC * fp)11186ae1554aSColin Percival get_tbl(FILEDESC *fp)
11196ae1554aSColin Percival {
11206ae1554aSColin Percival int fd;
11216ae1554aSColin Percival FILEDESC *child;
11226ae1554aSColin Percival
11236ae1554aSColin Percival if (fp->read_tbl)
11246ae1554aSColin Percival return;
11256ae1554aSColin Percival if (fp->child == NULL) {
11266ae1554aSColin Percival if ((fd = open(fp->datfile, O_RDONLY)) < 0) {
11276ae1554aSColin Percival perror(fp->datfile);
11286ae1554aSColin Percival exit(1);
11296ae1554aSColin Percival }
11306ae1554aSColin Percival if (read(fd, (char *) &fp->tbl, sizeof fp->tbl) != sizeof fp->tbl) {
11316ae1554aSColin Percival (void)fprintf(stderr,
11326ae1554aSColin Percival "fortune: %s corrupted\n", fp->path);
11336ae1554aSColin Percival exit(1);
11346ae1554aSColin Percival }
11356ae1554aSColin Percival /* fp->tbl.str_version = be32toh(fp->tbl.str_version); */
11366ae1554aSColin Percival fp->tbl.str_numstr = be32toh(fp->tbl.str_numstr);
11376ae1554aSColin Percival fp->tbl.str_longlen = be32toh(fp->tbl.str_longlen);
11386ae1554aSColin Percival fp->tbl.str_shortlen = be32toh(fp->tbl.str_shortlen);
11396ae1554aSColin Percival fp->tbl.str_flags = be32toh(fp->tbl.str_flags);
11406ae1554aSColin Percival (void) close(fd);
11416ae1554aSColin Percival }
11426ae1554aSColin Percival else {
11436ae1554aSColin Percival zero_tbl(&fp->tbl);
11446ae1554aSColin Percival for (child = fp->child; child != NULL; child = child->next) {
11456ae1554aSColin Percival get_tbl(child);
11466ae1554aSColin Percival sum_tbl(&fp->tbl, &child->tbl);
11476ae1554aSColin Percival }
11486ae1554aSColin Percival }
11496ae1554aSColin Percival fp->read_tbl = TRUE;
11506ae1554aSColin Percival }
11516ae1554aSColin Percival
11526ae1554aSColin Percival /*
11536ae1554aSColin Percival * zero_tbl:
11546ae1554aSColin Percival * Zero out the fields we care about in a tbl structure.
11556ae1554aSColin Percival */
11566ae1554aSColin Percival static void
zero_tbl(STRFILE * tp)11576ae1554aSColin Percival zero_tbl(STRFILE *tp)
11586ae1554aSColin Percival {
11596ae1554aSColin Percival tp->str_numstr = 0;
11606ae1554aSColin Percival tp->str_longlen = 0;
11616ae1554aSColin Percival tp->str_shortlen = ~0;
11626ae1554aSColin Percival }
11636ae1554aSColin Percival
11646ae1554aSColin Percival /*
11656ae1554aSColin Percival * sum_tbl:
11666ae1554aSColin Percival * Merge the tbl data of t2 into t1.
11676ae1554aSColin Percival */
11686ae1554aSColin Percival static void
sum_tbl(STRFILE * t1,STRFILE * t2)11696ae1554aSColin Percival sum_tbl(STRFILE *t1, STRFILE *t2)
11706ae1554aSColin Percival {
11716ae1554aSColin Percival t1->str_numstr += t2->str_numstr;
11726ae1554aSColin Percival if (t1->str_longlen < t2->str_longlen)
11736ae1554aSColin Percival t1->str_longlen = t2->str_longlen;
11746ae1554aSColin Percival if (t1->str_shortlen > t2->str_shortlen)
11756ae1554aSColin Percival t1->str_shortlen = t2->str_shortlen;
11766ae1554aSColin Percival }
11776ae1554aSColin Percival
11786ae1554aSColin Percival #define STR(str) ((str) == NULL ? "NULL" : (str))
11796ae1554aSColin Percival
11806ae1554aSColin Percival /*
11816ae1554aSColin Percival * print_file_list:
11826ae1554aSColin Percival * Print out the file list
11836ae1554aSColin Percival */
11846ae1554aSColin Percival static void
print_file_list(void)11856ae1554aSColin Percival print_file_list(void)
11866ae1554aSColin Percival {
11876ae1554aSColin Percival print_list(File_list, 0);
11886ae1554aSColin Percival }
11896ae1554aSColin Percival
11906ae1554aSColin Percival /*
11916ae1554aSColin Percival * print_list:
11926ae1554aSColin Percival * Print out the actual list, recursively.
11936ae1554aSColin Percival */
11946ae1554aSColin Percival static void
print_list(FILEDESC * list,int lev)11956ae1554aSColin Percival print_list(FILEDESC *list, int lev)
11966ae1554aSColin Percival {
11976ae1554aSColin Percival while (list != NULL) {
11986ae1554aSColin Percival fprintf(stderr, "%*s", lev * 4, "");
11996ae1554aSColin Percival if (list->percent == NO_PROB)
12006ae1554aSColin Percival fprintf(stderr, "___%%");
12016ae1554aSColin Percival else
12026ae1554aSColin Percival fprintf(stderr, "%3d%%", list->percent);
12036ae1554aSColin Percival fprintf(stderr, " %s", STR(list->name));
12046ae1554aSColin Percival DPRINTF(1, (stderr, " (%s, %s, %s)", STR(list->path),
12056ae1554aSColin Percival STR(list->datfile), STR(list->posfile)));
12066ae1554aSColin Percival fprintf(stderr, "\n");
12076ae1554aSColin Percival if (list->child != NULL)
12086ae1554aSColin Percival print_list(list->child, lev + 1);
12096ae1554aSColin Percival list = list->next;
12106ae1554aSColin Percival }
12116ae1554aSColin Percival }
12126ae1554aSColin Percival
12136ae1554aSColin Percival /*
12146ae1554aSColin Percival * conv_pat:
12156ae1554aSColin Percival * Convert the pattern to an ignore-case equivalent.
12166ae1554aSColin Percival */
12176ae1554aSColin Percival static char *
conv_pat(char * orig)12186ae1554aSColin Percival conv_pat(char *orig)
12196ae1554aSColin Percival {
12206ae1554aSColin Percival char *sp;
12216ae1554aSColin Percival unsigned int cnt;
12226ae1554aSColin Percival char *new;
12236ae1554aSColin Percival
12246ae1554aSColin Percival cnt = 1; /* allow for '\0' */
12256ae1554aSColin Percival for (sp = orig; *sp != '\0'; sp++)
12266ae1554aSColin Percival if (isalpha((unsigned char)*sp))
12276ae1554aSColin Percival cnt += 4;
12286ae1554aSColin Percival else
12296ae1554aSColin Percival cnt++;
12306ae1554aSColin Percival if ((new = malloc(cnt)) == NULL) {
12316ae1554aSColin Percival fprintf(stderr, "pattern too long for ignoring case\n");
12326ae1554aSColin Percival exit(1);
12336ae1554aSColin Percival }
12346ae1554aSColin Percival
12356ae1554aSColin Percival for (sp = new; *orig != '\0'; orig++) {
12366ae1554aSColin Percival if (islower((unsigned char)*orig)) {
12376ae1554aSColin Percival *sp++ = '[';
12386ae1554aSColin Percival *sp++ = *orig;
12396ae1554aSColin Percival *sp++ = toupper((unsigned char)*orig);
12406ae1554aSColin Percival *sp++ = ']';
12416ae1554aSColin Percival }
12426ae1554aSColin Percival else if (isupper((unsigned char)*orig)) {
12436ae1554aSColin Percival *sp++ = '[';
12446ae1554aSColin Percival *sp++ = *orig;
12456ae1554aSColin Percival *sp++ = tolower((unsigned char)*orig);
12466ae1554aSColin Percival *sp++ = ']';
12476ae1554aSColin Percival }
12486ae1554aSColin Percival else
12496ae1554aSColin Percival *sp++ = *orig;
12506ae1554aSColin Percival }
12516ae1554aSColin Percival *sp = '\0';
12526ae1554aSColin Percival
12536ae1554aSColin Percival return (new);
12546ae1554aSColin Percival }
12556ae1554aSColin Percival
12566ae1554aSColin Percival /*
12576ae1554aSColin Percival * find_matches:
12586ae1554aSColin Percival * Find all the fortunes which match the pattern we've been given.
12596ae1554aSColin Percival */
12606ae1554aSColin Percival static int
find_matches(void)12616ae1554aSColin Percival find_matches(void)
12626ae1554aSColin Percival {
12636ae1554aSColin Percival Fort_len = maxlen_in_list(File_list);
12646ae1554aSColin Percival DPRINTF(2, (stderr, "Maximum length is %d\n", Fort_len));
12656ae1554aSColin Percival /* extra length, "%\n" is appended */
12666ae1554aSColin Percival Fortbuf = do_malloc((unsigned int) Fort_len + 10);
12676ae1554aSColin Percival
12686ae1554aSColin Percival Found_one = FALSE;
12696ae1554aSColin Percival matches_in_list(File_list);
12706ae1554aSColin Percival
12716ae1554aSColin Percival return (Found_one);
12726ae1554aSColin Percival }
12736ae1554aSColin Percival
12746ae1554aSColin Percival /*
12756ae1554aSColin Percival * maxlen_in_list
12766ae1554aSColin Percival * Return the maximum fortune len in the file list.
12776ae1554aSColin Percival */
12786ae1554aSColin Percival static int
maxlen_in_list(FILEDESC * list)12796ae1554aSColin Percival maxlen_in_list(FILEDESC *list)
12806ae1554aSColin Percival {
12816ae1554aSColin Percival FILEDESC *fp;
12826ae1554aSColin Percival int len, maxlen;
12836ae1554aSColin Percival
12846ae1554aSColin Percival maxlen = 0;
12856ae1554aSColin Percival for (fp = list; fp != NULL; fp = fp->next) {
12866ae1554aSColin Percival if (fp->child != NULL) {
12876ae1554aSColin Percival if ((len = maxlen_in_list(fp->child)) > maxlen)
12886ae1554aSColin Percival maxlen = len;
12896ae1554aSColin Percival }
12906ae1554aSColin Percival else {
12916ae1554aSColin Percival get_tbl(fp);
12926ae1554aSColin Percival if (fp->tbl.str_longlen > (unsigned int)maxlen)
12936ae1554aSColin Percival maxlen = fp->tbl.str_longlen;
12946ae1554aSColin Percival }
12956ae1554aSColin Percival }
12966ae1554aSColin Percival
12976ae1554aSColin Percival return (maxlen);
12986ae1554aSColin Percival }
12996ae1554aSColin Percival
13006ae1554aSColin Percival /*
13016ae1554aSColin Percival * matches_in_list
13026ae1554aSColin Percival * Print out the matches from the files in the list.
13036ae1554aSColin Percival */
13046ae1554aSColin Percival static void
matches_in_list(FILEDESC * list)13056ae1554aSColin Percival matches_in_list(FILEDESC *list)
13066ae1554aSColin Percival {
13076ae1554aSColin Percival char *sp, *p;
13086ae1554aSColin Percival FILEDESC *fp;
13096ae1554aSColin Percival int in_file;
13106ae1554aSColin Percival unsigned char ch;
13116ae1554aSColin Percival
13126ae1554aSColin Percival for (fp = list; fp != NULL; fp = fp->next) {
13136ae1554aSColin Percival if (fp->child != NULL) {
13146ae1554aSColin Percival matches_in_list(fp->child);
13156ae1554aSColin Percival continue;
13166ae1554aSColin Percival }
13176ae1554aSColin Percival DPRINTF(1, (stderr, "searching in %s\n", fp->path));
13186ae1554aSColin Percival open_fp(fp);
13196ae1554aSColin Percival sp = Fortbuf;
13206ae1554aSColin Percival in_file = FALSE;
13216ae1554aSColin Percival while (fgets(sp, Fort_len, fp->inf) != NULL)
13226ae1554aSColin Percival if (fp->tbl.str_flags & STR_COMMENTS
13236ae1554aSColin Percival && sp[0] == fp->tbl.str_delim
13246ae1554aSColin Percival && sp[1] == fp->tbl.str_delim)
13256ae1554aSColin Percival continue;
13266ae1554aSColin Percival else if (!STR_ENDSTRING(sp, fp->tbl))
13276ae1554aSColin Percival sp += strlen(sp);
13286ae1554aSColin Percival else {
13296ae1554aSColin Percival *sp = '\0';
13306ae1554aSColin Percival if (fp->tbl.str_flags & STR_ROTATED)
13316ae1554aSColin Percival for (p = Fortbuf; (ch = *p) != '\0'; ++p) {
13326ae1554aSColin Percival if (isascii(ch)) {
13336ae1554aSColin Percival if (isupper(ch))
13346ae1554aSColin Percival *p = 'A' + (ch - 'A' + 13) % 26;
13356ae1554aSColin Percival else if (islower(ch))
13366ae1554aSColin Percival *p = 'a' + (ch - 'a' + 13) % 26;
13376ae1554aSColin Percival }
13386ae1554aSColin Percival }
13396ae1554aSColin Percival if (regexec(&Re_pat, Fortbuf, 0, NULL, 0) != REG_NOMATCH) {
13406ae1554aSColin Percival printf("%c%c", fp->tbl.str_delim,
13416ae1554aSColin Percival fp->tbl.str_delim);
13426ae1554aSColin Percival if (!in_file) {
13436ae1554aSColin Percival printf(" (%s)", fp->name);
13446ae1554aSColin Percival Found_one = TRUE;
13456ae1554aSColin Percival in_file = TRUE;
13466ae1554aSColin Percival }
13476ae1554aSColin Percival putchar('\n');
13486ae1554aSColin Percival (void) fwrite(Fortbuf, 1, (sp - Fortbuf), stdout);
13496ae1554aSColin Percival }
13506ae1554aSColin Percival sp = Fortbuf;
13516ae1554aSColin Percival }
13526ae1554aSColin Percival }
13536ae1554aSColin Percival }
13546ae1554aSColin Percival
13556ae1554aSColin Percival static void
usage(void)13566ae1554aSColin Percival usage(void)
13576ae1554aSColin Percival {
13586ae1554aSColin Percival (void) fprintf(stderr, "fortune [-a");
13596ae1554aSColin Percival #ifdef DEBUG
13606ae1554aSColin Percival (void) fprintf(stderr, "D");
13616ae1554aSColin Percival #endif /* DEBUG */
13626ae1554aSColin Percival (void) fprintf(stderr, "efilosw]");
13636ae1554aSColin Percival (void) fprintf(stderr, " [-m pattern]");
13646ae1554aSColin Percival (void) fprintf(stderr, " [[N%%] file/directory/all]\n");
13656ae1554aSColin Percival exit(1);
13666ae1554aSColin Percival }
13676ae1554aSColin Percival
13686ae1554aSColin Percival /*
13696ae1554aSColin Percival * getpath
1370d93a2f5dSCeri Davies * Set up file search path from environment var FORTUNE_PATH;
13716ae1554aSColin Percival * if not set, use the compiled in FORTDIR.
13726ae1554aSColin Percival */
13736ae1554aSColin Percival
13746ae1554aSColin Percival static void
getpath(void)13756ae1554aSColin Percival getpath(void)
13766ae1554aSColin Percival {
13776ae1554aSColin Percival int nstr, foundenv;
13786ae1554aSColin Percival char *pch, **ppch, *str, *path;
13796ae1554aSColin Percival
13806ae1554aSColin Percival foundenv = 1;
13816ae1554aSColin Percival Fortune_path = getenv("FORTUNE_PATH");
13826ae1554aSColin Percival if (Fortune_path == NULL) {
13836ae1554aSColin Percival Fortune_path = FORTDIR;
13846ae1554aSColin Percival foundenv = 0;
13856ae1554aSColin Percival }
13866ae1554aSColin Percival path = strdup(Fortune_path);
13876ae1554aSColin Percival
13886ae1554aSColin Percival for (nstr = 2, pch = path; *pch != '\0'; pch++) {
13896ae1554aSColin Percival if (*pch == ':')
13906ae1554aSColin Percival nstr++;
13916ae1554aSColin Percival }
13926ae1554aSColin Percival
13936ae1554aSColin Percival ppch = Fortune_path_arr = (char **)calloc(nstr, sizeof(char *));
13946ae1554aSColin Percival
13956ae1554aSColin Percival nstr = 0;
13966ae1554aSColin Percival str = strtok(path, ":");
13976ae1554aSColin Percival while (str) {
13986ae1554aSColin Percival if (is_dir(str)) {
13996ae1554aSColin Percival nstr++;
14006ae1554aSColin Percival *ppch++ = str;
14016ae1554aSColin Percival }
14026ae1554aSColin Percival str = strtok(NULL, ":");
14036ae1554aSColin Percival }
14046ae1554aSColin Percival
14056ae1554aSColin Percival if (nstr == 0) {
14066ae1554aSColin Percival if (foundenv == 1) {
14076ae1554aSColin Percival fprintf(stderr,
14086ae1554aSColin Percival "fortune: FORTUNE_PATH: None of the specified "
14096ae1554aSColin Percival "directories found.\n");
14106ae1554aSColin Percival exit(1);
14116ae1554aSColin Percival }
14126ae1554aSColin Percival free(path);
14136ae1554aSColin Percival Fortune_path_arr[0] = strdup(FORTDIR);
14146ae1554aSColin Percival }
14156ae1554aSColin Percival }
1416