new copyright notice
[unix-history] / usr / src / bin / cat / cat.c
index 0d2ec46..2155190 100644 (file)
-static char *sccsid = "@(#)cat.c       4.1 (Berkeley) %G%";
 /*
 /*
- * Concatenate files.
+ * Copyright (c) 1989 The Regents of the University of California.
+ * All rights reserved.
+ *
+ * This code is derived from software contributed to Berkeley by
+ * Kevin Fall.
+ *
+ * %sccs.include.redist.c%
  */
 
  */
 
-#include <stdio.h>
-#include <sys/types.h>
+#ifndef lint
+char copyright[] =
+"@(#) Copyright (c) 1989 The Regents of the University of California.\n\
+ All rights reserved.\n";
+#endif /* not lint */
+
+#ifndef lint
+static char sccsid[] = "@(#)cat.c      5.11 (Berkeley) %G%";
+#endif /* not lint */
+
+#include <sys/param.h>
 #include <sys/stat.h>
 #include <sys/stat.h>
+#include <sys/file.h>
+#include <stdio.h>
+#include <ctype.h>
 
 
-char   stdbuf[BUFSIZ];
+extern int errno;
+int bflag, eflag, nflag, sflag, tflag, vflag;
+int rval;
+char *filename;
 
 main(argc, argv)
 
 main(argc, argv)
-char **argv;
+       int argc;
+       char **argv;
 {
 {
-       int fflg = 0;
-       register FILE *fi;
-       register c;
-       int dev, ino = -1;
-       struct stat statb;
+       extern int optind;
+       int ch;
+       char *strerror();
 
 
-       setbuf(stdout, stdbuf);
-       for( ; argc>1 && argv[1][0]=='-'; argc--,argv++) {
-               switch(argv[1][1]) {
-               case 0:
+       while ((ch = getopt(argc, argv, "benstuv")) != EOF)
+               switch (ch) {
+               case 'b':
+                       bflag = nflag = 1;      /* -b implies -n */
+                       break;
+               case 'e':
+                       eflag = vflag = 1;      /* -e implies -v */
+                       break;
+               case 'n':
+                       nflag = 1;
+                       break;
+               case 's':
+                       sflag = 1;
+                       break;
+               case 't':
+                       tflag = vflag = 1;      /* -t implies -v */
                        break;
                case 'u':
                        setbuf(stdout, (char *)NULL);
                        break;
                case 'u':
                        setbuf(stdout, (char *)NULL);
-                       continue;
+                       break;
+               case 'v':
+                       vflag = 1;
+                       break;
+               case '?':
+                       (void)fprintf(stderr,
+                           "usage: cat [-benstuv] [-] [file ...]\n");
+                       exit(1);
+               }
+       argv += optind;
+
+       if (bflag || eflag || nflag || sflag || tflag || vflag)
+               cook_args(argv);
+       else
+               raw_args(argv);
+       exit(rval);
+}
+
+cook_args(argv)
+       char **argv;
+{
+       register FILE *fp;
+
+       fp = stdin;
+       filename = "-";
+       do {
+               if (*argv) {
+                       if (!strcmp(*argv, "-"))
+                               fp = stdin;
+                       else if (!(fp = fopen(*argv, "r"))) {
+                               (void)fprintf(stderr, 
+                                   "cat: %s: %s\n", *argv, strerror(errno));
+                               rval = 1;
+                               ++argv;
+                               continue;
+                       }
+                       filename = *argv++;
                }
                }
-               break;
+               cook_buf(fp);
+               if (fp != stdin)
+                       (void)fclose(fp);
+       } while (*argv);
+}
+
+cook_buf(fp)
+       register FILE *fp;
+{
+       register int ch, gobble, line, prev;
+
+       line = gobble = 0;
+       for (prev = '\n'; (ch = getc(fp)) != EOF; prev = ch) {
+               if (prev == '\n') {
+                       if (ch == '\n') {
+                               if (sflag) {
+                                       if (!gobble && putchar(ch) == EOF)
+                                               break;
+                                       gobble = 1;
+                                       continue;
+                               }
+                               if (nflag && !bflag) {
+                                       (void)fprintf(stdout, "%6d\t", ++line);
+                                       if (ferror(stdout))
+                                               break;
+                               }
+                       } else if (nflag) {
+                               (void)fprintf(stdout, "%6d\t", ++line);
+                               if (ferror(stdout))
+                                       break;
+                       }
+               }
+               gobble = 0;
+               if (ch == '\n') {
+                       if (eflag)
+                               if (putchar('$') == EOF)
+                                       break;
+               } else if (ch == '\t') {
+                       if (tflag) {
+                               if (putchar('^') == EOF || putchar('I') == EOF)
+                                       break;
+                               continue;
+                       }
+               } else if (vflag) {
+                       if (!isascii(ch)) {
+                               if (putchar('M') == EOF || putchar('-') == EOF)
+                                       break;
+                               ch = toascii(ch);
+                       }
+                       if (iscntrl(ch)) {
+                               if (putchar('^') == EOF ||
+                                   putchar(ch == '\177' ? '?' :
+                                   ch | 0100) == EOF)
+                                       break;
+                               continue;
+                       }
+               }
+               if (putchar(ch) == EOF)
+                       break;
        }
        }
-       fstat(fileno(stdout), &statb);
-       statb.st_mode &= S_IFMT;
-       if (statb.st_mode!=S_IFCHR && statb.st_mode!=S_IFBLK) {
-               dev = statb.st_dev;
-               ino = statb.st_ino;
+       if (ferror(fp)) {
+               (void)fprintf(stderr, "cat: %s: read error\n", filename);
+               rval = 1;
        }
        }
-       if (argc < 2) {
-               argc = 2;
-               fflg++;
+       if (ferror(stdout)) {
+               clearerr(stdout);
+               (void)fprintf(stderr, "cat: stdout: write error\n");
+               rval = 1;
        }
        }
-       while (--argc > 0) {
-               if (fflg || (*++argv)[0]=='-' && (*argv)[1]=='\0')
-                       fi = stdin;
-               else {
-                       if ((fi = fopen(*argv, "r")) == NULL) {
-                               fprintf(stderr, "cat: can't open %s\n", *argv);
+}
+
+raw_args(argv)
+       char **argv;
+{
+       register int fd;
+
+       fd = fileno(stdin);
+       filename = "-";
+       do {
+               if (*argv) {
+                       if (!strcmp(*argv, "-"))
+                               fd = fileno(stdin);
+                       else if ((fd = open(*argv, O_RDONLY, 0)) < 0) {
+                               (void)fprintf(stderr, "cat: %s: %s\n",
+                                   *argv, strerror(errno));
+                               rval = 1;
+                               ++argv;
                                continue;
                        }
                                continue;
                        }
+                       filename = *argv++;
+               }
+               rval |= raw_cat(fd);
+               if (fd != fileno(stdin))
+                       (void)close(fd);
+       } while (*argv);
+}
+
+raw_cat(fd)
+       register int fd;
+{
+       register int nr, nw, off;
+       static int bsize;
+       static char *buf;
+       struct stat sbuf;
+       char *malloc(), *strerror();
+
+       if (!buf) {
+               if (fstat(fileno(stdout), &sbuf)) {
+                       (void)fprintf(stderr, "cat: %s: %s\n", filename,
+                           strerror(errno));
+                       return(1);
+               }
+               bsize = MAX(sbuf.st_blksize, 1024);
+               if (!(buf = malloc((u_int)bsize))) {
+                       (void)fprintf(stderr, "cat: %s: no memory.\n",
+                           filename);
+                       return(1);
                }
                }
-               fstat(fileno(fi), &statb);
-               if (statb.st_dev==dev && statb.st_ino==ino) {
-                       fprintf(stderr, "cat: input %s is output\n",
-                          fflg?"-": *argv);
-                       fclose(fi);
-                       continue;
+       }
+       while ((nr = read(fd, buf, bsize)) > 0)
+               for (off = 0; off < nr;) {
+                       if ((nw = write(fileno(stdout), buf + off, nr)) < 0) {
+                               perror("cat: stdout");
+                               return(1);
+                       }
+                       off += nw;
                }
                }
-               while ((c = getc(fi)) != EOF)
-                       putchar(c);
-               if (fi!=stdin)
-                       fclose(fi);
+       if (nr < 0) {
+               (void)fprintf(stderr, "cat: %s: %s\n", filename,
+                   strerror(errno));
+               return(1);
        }
        return(0);
 }
        }
        return(0);
 }