]> git.cameronkatri.com Git - bsdgames-darwin.git/blobdiff - fortune/strfile/strfile.c
strfile: Check that input/output filenames don't exceed the buffer size
[bsdgames-darwin.git] / fortune / strfile / strfile.c
index 6961512b024907c41331ca2c90fb1ca305f57a0a..a0598505139b81a410ab12275f1b9c5bb51133fc 100644 (file)
@@ -1,4 +1,4 @@
-/*     $NetBSD: strfile.c,v 1.26 2008/07/20 01:03:21 lukem Exp $       */
+/*     $NetBSD: strfile.c,v 1.39 2020/04/29 20:45:05 nia Exp $ */
 
 /*-
  * Copyright (c) 1989, 1993
  * SUCH DAMAGE.
  */
 
+#if HAVE_NBTOOL_CONFIG_H
+#include "nbtool_config.h"
+#endif
+
 #ifdef __NetBSD__
 #include <sys/cdefs.h>
 #ifndef lint
@@ -43,45 +47,27 @@ __COPYRIGHT("@(#) Copyright (c) 1989, 1993\
 #if 0
 static char sccsid[] = "@(#)strfile.c  8.1 (Berkeley) 5/31/93";
 #else
-__RCSID("$NetBSD: strfile.c,v 1.26 2008/07/20 01:03:21 lukem Exp $");
+__RCSID("$NetBSD: strfile.c,v 1.39 2020/04/29 20:45:05 nia Exp $");
 #endif
 #endif /* not lint */
 #endif /* __NetBSD__ */
 
-/* n.b.: this file is used at build-time - i.e. during build.sh. */
-
-# include      <sys/types.h>
-# include      <sys/param.h>
-# include      <ctype.h>
-# include      <stdio.h>
-# include      <stdlib.h>
-# include      <string.h>
-# include      <time.h>
-# include      <unistd.h>
+#include <sys/types.h>
+#include <sys/param.h>
+#include <ctype.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <time.h>
+#include <unistd.h>
+#include <inttypes.h>
+#include <err.h>
 
-# ifndef u_int32_t
-# define u_int32_t     unsigned int
-# endif
-# include      "strfile.h"
-
-# ifndef MAXPATHLEN
-# define       MAXPATHLEN      1024
-# endif        /* MAXPATHLEN */
-
-u_int32_t
-h2nl(u_int32_t h)
-{
-        unsigned char c[4];
-        u_int32_t rv;
+#include "strfile.h"
 
-        c[0] = (h >> 24) & 0xff;
-        c[1] = (h >> 16) & 0xff;
-        c[2] = (h >>  8) & 0xff;
-        c[3] = (h >>  0) & 0xff;
-        memcpy(&rv, c, sizeof rv);
-
-        return (rv);
-}
+#ifndef MAXPATHLEN
+#define        MAXPATHLEN      1024
+#endif /* MAXPATHLEN */
 
 /*
  *     This program takes a file composed of strings separated by
@@ -106,9 +92,6 @@ h2nl(u_int32_t h)
  *     Added ordering options.
  */
 
-# define       TRUE    1
-# define       FALSE   0
-
 # define       STORING_PTRS    (Oflag || Rflag)
 # define       CHUNKSIZE       512
 
@@ -118,7 +101,7 @@ h2nl(u_int32_t h)
                        else if (((sz) + 1) % CHUNKSIZE == 0) \
                                ptr = realloc(ptr, ((sz) + CHUNKSIZE) * sizeof *ptr); \
                        if (ptr == NULL) \
-                               die("out of space"); \
+                               err(1, "out of space"); \
                } while (0)
 
 typedef struct {
@@ -126,45 +109,34 @@ typedef struct {
        off_t   pos;
 } STR;
 
-char   *Infile         = NULL,         /* input file name */
-       Outfile[MAXPATHLEN] = "",       /* output file name */
-       Delimch         = '%';          /* delimiting character */
+static char *Infile = NULL;            /* input file name */
+static char Outfile[MAXPATHLEN] = "";  /* output file name */
+static char Delimch = '%';             /* delimiting character */
 
-int    Sflag           = FALSE;        /* silent run flag */
-int    Oflag           = FALSE;        /* ordering flag */
-int    Iflag           = FALSE;        /* ignore case flag */
-int    Rflag           = FALSE;        /* randomize order flag */
-int    Xflag           = FALSE;        /* set rotated bit */
-long   Num_pts         = 0;            /* number of pointers/strings */
+static int Sflag       = 0;            /* silent run flag */
+static int Oflag       = 0;            /* ordering flag */
+static int Iflag       = 0;            /* ignore case flag */
+static int Rflag       = 0;            /* randomize order flag */
+static int Xflag       = 0;            /* set rotated bit */
+static long Num_pts    = 0;            /* number of pointers/strings */
 
-off_t  *Seekpts;
+static off_t *Seekpts;
 
-FILE   *Sort_1, *Sort_2;               /* pointers for sorting */
+static FILE *Sort_1, *Sort_2;          /* pointers for sorting */
 
-STRFILE        Tbl;                            /* statistics table */
+static STRFILE Tbl;                    /* statistics table */
 
-STR    *Firstch;                       /* first chars of each string */
+static STR *Firstch;                   /* first chars of each string */
 
-#ifdef __GNUC__
-#define NORETURN       __dead
-#else
-#define NORETURN
-#endif
 
-#ifndef __dead /* not NetBSD, presumably */
-#define __dead ;
-#endif
-
-void   add_offset(FILE *, off_t);
-int    cmp_str(const void *, const void *);
-void   die(const char *) NORETURN;
-void   dieperror(const char *, char *) NORETURN;
-void   do_order(void);
-void   fwrite_be_offt(off_t, FILE *);
-void   getargs(int, char *[]);
-int    main(int, char *[]);
-void   randomize(void);
-void   usage(void) NORETURN;
+static uint32_t h2nl(uint32_t h);
+static void getargs(int argc, char **argv);
+static void usage(void) __dead;
+static void add_offset(FILE *fp, off_t off);
+static void do_order(void);
+static int cmp_str(const void *vp1, const void *vp2);
+static void randomize(void);
+static void fwrite_be_offt(off_t off, FILE *f);
 
 
 /*
@@ -177,29 +149,28 @@ void      usage(void) NORETURN;
  *     and then seek back to the beginning to write in the table.
  */
 int
-main(ac, av)
-       int     ac;
-       char    *av[];
+main(int ac, char **av)
 {
        char            *sp, dc;
        FILE            *inf, *outf;
-       off_t           last_off, length, pos, *p;
-       int             first, cnt;
+       off_t           last_off, length, pos;
+       int             first;
        char            *nsp;
        STR             *fp;
        static char     string[257];
+       long            i;
 
        /* sanity test */
-       if (sizeof(u_int32_t) != 4)
-               die("sizeof(unsigned int) != 4");
+       if (sizeof(uint32_t) != 4)
+               errx(1, "sizeof(uint32_t) != 4");
 
        getargs(ac, av);                /* evalute arguments */
        dc = Delimch;
        if ((inf = fopen(Infile, "r")) == NULL)
-               dieperror("open `%s'", Infile);
+               err(1, "open `%s'", Infile);
 
        if ((outf = fopen(Outfile, "w")) == NULL)
-               dieperror("open `%s'", Outfile);
+               err(1, "open `%s'", Outfile);
        if (!STORING_PTRS)
                (void) fseek(outf, sizeof Tbl, SEEK_SET);
 
@@ -239,7 +210,7 @@ main(ac, av)
                        else
                                fp->first = *nsp;
                        fp->pos = Seekpts[Num_pts - 1];
-                       first = FALSE;
+                       first = 0;
                }
        } while (sp != NULL);
 
@@ -277,12 +248,12 @@ main(ac, av)
        Tbl.str_flags = h2nl(Tbl.str_flags);
        (void) fwrite((char *) &Tbl, sizeof Tbl, 1, outf);
        if (STORING_PTRS) {
-               for (p = Seekpts, cnt = Num_pts; cnt--; ++p)
-                       fwrite_be_offt(*p, outf);
+               for (i = 0; i < Num_pts; i++)
+                       fwrite_be_offt(Seekpts[i], outf);
        }
        fflush(outf);
        if (ferror(outf))
-               dieperror("fwrite %s", Outfile);
+               err(1, "fwrite %s", Outfile);
        (void) fclose(outf);
        exit(0);
 }
@@ -290,14 +261,13 @@ main(ac, av)
 /*
  *     This routine evaluates arguments from the command line
  */
-void
-getargs(argc, argv)
-       int     argc;
-       char    **argv;
+static void
+getargs(int argc, char **argv)
 {
        int     ch;
        extern  int optind;
        extern  char *optarg;
+       size_t  len;
 
        while ((ch = getopt(argc, argv, "c:iorsx")) != -1)
                switch(ch) {
@@ -331,44 +301,36 @@ getargs(argc, argv)
 
        if (*argv) {
                Infile = *argv;
-               if (*++argv)
-                       (void) strcpy(Outfile, *argv);
+               if (*++argv) {
+                       len = strlen(*argv);
+                       if (len >= sizeof(Outfile)) {
+                               puts("Bad output filename");
+                               usage();
+                       }
+                       (void) memcpy(Outfile, *argv, len + 1);
+               }
        }
        if (!Infile) {
                puts("No input file name");
                usage();
        }
        if (*Outfile == '\0') {
+               len = strlen(Infile) + sizeof(".dat");
+               if (len > sizeof(Outfile)) {
+                       puts("Bad input filename");
+                       usage();
+               }
                (void) strcpy(Outfile, Infile);
                (void) strcat(Outfile, ".dat");
        }
 }
 
-void
-usage()
+static void
+usage(void)
 {
        (void) fprintf(stderr,
-           "strfile [-iorsx] [-c char] sourcefile [datafile]\n");
-       exit(1);
-}
-
-void
-die(str)
-       const char *str;
-{
-       fprintf(stderr, "strfile: %s\n", str);
-       exit(1);
-}
-
-void
-dieperror(fmt, file)
-       const char *fmt;
-       char *file;
-{
-       fprintf(stderr, "strfile: ");
-       fprintf(stderr, fmt, file);
-       fprintf(stderr, ": ");
-       perror(NULL);
+           "Usage: %s [-iorsx] [-c char] sourcefile [datafile]\n",
+           getprogname());
        exit(1);
 }
 
@@ -376,10 +338,8 @@ dieperror(fmt, file)
  * add_offset:
  *     Add an offset to the list, or write it out, as appropriate.
  */
-void
-add_offset(fp, off)
-       FILE    *fp;
-       off_t   off;
+static void
+add_offset(FILE *fp, off_t off)
 {
 
        if (!STORING_PTRS) {
@@ -395,8 +355,8 @@ add_offset(fp, off)
  * do_order:
  *     Order the strings alphabetically (possibly ignoring case).
  */
-void
-do_order()
+static void
+do_order(void)
 {
        int     i;
        off_t   *lp;
@@ -415,9 +375,8 @@ do_order()
        Tbl.str_flags |= STR_ORDERED;
 }
 
-int
-cmp_str(vp1, vp2)
-       const void *vp1, *vp2;
+static int
+cmp_str(const void *vp1, const void *vp2)
 {
        const STR       *p1, *p2;
        int     c1, c2;
@@ -437,8 +396,8 @@ cmp_str(vp1, vp2)
        (void) fseek(Sort_1, p1->pos, SEEK_SET);
        (void) fseek(Sort_2, p2->pos, SEEK_SET);
 
-       n1 = FALSE;
-       n2 = FALSE;
+       n1 = 0;
+       n2 = 0;
        while (!isalnum(c1 = getc(Sort_1)) && c1 != '\0')
                SET_N(n1, c1);
        while (!isalnum(c2 = getc(Sort_2)) && c2 != '\0')
@@ -471,14 +430,14 @@ cmp_str(vp1, vp2)
  *     not to randomize across delimiter boundaries.  All
  *     randomization is done within each block.
  */
-void
-randomize()
+static void
+randomize(void)
 {
        int     cnt, i;
        off_t   tmp;
        off_t   *sp;
 
-       srandom((int)(time((time_t *) NULL) + getpid()));
+       srandom((int)(time(NULL) + getpid()));
 
        Tbl.str_flags |= STR_RANDOM;
        cnt = Tbl.str_numstr;
@@ -500,10 +459,8 @@ randomize()
  *     Write out the off paramater as a 64 bit big endian number
  */
 
-void
-fwrite_be_offt(off, f)
-       off_t    off;
-       FILE    *f;
+static void
+fwrite_be_offt(off_t off, FILE *f)
 {
        int             i;
        unsigned char   c[8];
@@ -514,3 +471,18 @@ fwrite_be_offt(off, f)
        }
        fwrite(c, sizeof(c), 1, f);
 }
+
+static uint32_t
+h2nl(uint32_t h)
+{
+        unsigned char c[4];
+        uint32_t rv;
+
+        c[0] = (h >> 24) & 0xff;
+        c[1] = (h >> 16) & 0xff;
+        c[2] = (h >>  8) & 0xff;
+        c[3] = (h >>  0) & 0xff;
+        memcpy(&rv, c, sizeof rv);
+
+        return (rv);
+}