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