We don't need the "rpath" promise if there are no file arguments.  We
drop this promise in other utilities that default to stdin in the
no-file case.

If we consolidate raw_args() and cook_args() into a single function,
cat_file(), we can drop the "rpath" promise in main(), in a single
place in cat.c.

The alternative is to do pledge("stdio", NULL) in two places in cat.c.
I'm not a fan of that.

ok?

Index: cat.c
===================================================================
RCS file: /cvs/src/bin/cat/cat.c,v
retrieving revision 1.32
diff -u -p -r1.32 cat.c
--- cat.c       24 Oct 2021 21:24:21 -0000      1.32
+++ cat.c       13 Dec 2021 17:51:39 -0000
@@ -50,9 +50,8 @@
 int bflag, eflag, nflag, sflag, tflag, vflag;
 int rval;
 
-void cook_args(char *argv[]);
+void cat_file(const char *);
 void cook_buf(FILE *, const char *);
-void raw_args(char *argv[]);
 void raw_cat(int, const char *);
 
 int
@@ -94,38 +93,58 @@ main(int argc, char *argv[])
        }
        argv += optind;
 
-       if (bflag || eflag || nflag || sflag || tflag || vflag)
-               cook_args(argv);
-       else
-               raw_args(argv);
+       if (*argv == NULL) {
+               if (pledge("stdio", NULL) == -1)
+                       err(1, "pledge");
+               cat_file(NULL);
+       } else {
+               for (; *argv != NULL; argv++)
+                       cat_file(*argv);
+       }
        if (fclose(stdout))
                err(1, "stdout");
        return rval;
 }
 
 void
-cook_args(char **argv)
+cat_file(const char *path)
 {
+       const char *name;
        FILE *fp;
+       int fd;
 
-       if (*argv == NULL) {
-               cook_buf(stdin, "stdin");
-               return;
-       }
-
-       for (; *argv != NULL; argv++) {
-               if (!strcmp(*argv, "-")) {
-                       cook_buf(stdin, "stdin");
-                       clearerr(stdin);
-                       continue;
+       if (bflag || eflag || nflag || sflag || tflag || vflag) {
+               if (path == NULL || strcmp(path, "-") == 0) {
+                       name = "(stdin)";
+                       fp = stdin;
+               } else {
+                       name = path;
+                       if ((fp = fopen(name, "r")) == NULL) {
+                               warn("%s", name);
+                               rval = 1;
+                               return;
+                       }
                }
-               if ((fp = fopen(*argv, "r")) == NULL) {
-                       warn("%s", *argv);
-                       rval = 1;
-                       continue;
+               cook_buf(fp, name);
+               if (fp == stdin)
+                       clearerr(stdin);
+               else
+                       fclose(fp);
+       } else {
+               if (path == NULL || strcmp(path, "-") == 0) {
+                       name = "(stdin)";
+                       fd = STDIN_FILENO;
+               } else {
+                       name = path;
+                       if ((fd = open(name, O_RDONLY)) == -1) {
+                               warn("%s", name);
+                               rval = 1;
+                               return;
+                       }
                }
-               cook_buf(fp, *argv);
-               fclose(fp);
+               raw_cat(fd, name);
+               if (fd != STDIN_FILENO)
+                       close(fd);
        }
 }
 
@@ -191,31 +210,6 @@ cook_buf(FILE *fp, const char *filename)
        }
        if (ferror(stdout))
                err(1, "stdout");
-}
-
-void
-raw_args(char **argv)
-{
-       int fd;
-
-       if (*argv == NULL) {
-               raw_cat(fileno(stdin), "stdin");
-               return;
-       }
-
-       for (; *argv != NULL; argv++) {
-               if (!strcmp(*argv, "-")) {
-                       raw_cat(fileno(stdin), "stdin");
-                       continue;
-               }
-               if ((fd = open(*argv, O_RDONLY)) == -1) {
-                       warn("%s", *argv);
-                       rval = 1;
-                       continue;
-               }
-               raw_cat(fd, *argv);
-               close(fd);
-       }
 }
 
 void

Reply via email to