xref: /freebsd/bin/cat/cat.c (revision cafefe8c1b76269ef8cc8d4e7d499ad906d663de)
14b88c807SRodney W. Grimes /*
24b88c807SRodney W. Grimes  * Copyright (c) 1989, 1993
34b88c807SRodney W. Grimes  *	The Regents of the University of California.  All rights reserved.
44b88c807SRodney W. Grimes  *
54b88c807SRodney W. Grimes  * This code is derived from software contributed to Berkeley by
64b88c807SRodney W. Grimes  * Kevin Fall.
74b88c807SRodney W. Grimes  *
84b88c807SRodney W. Grimes  * Redistribution and use in source and binary forms, with or without
94b88c807SRodney W. Grimes  * modification, are permitted provided that the following conditions
104b88c807SRodney W. Grimes  * are met:
114b88c807SRodney W. Grimes  * 1. Redistributions of source code must retain the above copyright
124b88c807SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer.
134b88c807SRodney W. Grimes  * 2. Redistributions in binary form must reproduce the above copyright
144b88c807SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer in the
154b88c807SRodney W. Grimes  *    documentation and/or other materials provided with the distribution.
164b88c807SRodney W. Grimes  * 3. All advertising materials mentioning features or use of this software
174b88c807SRodney W. Grimes  *    must display the following acknowledgement:
184b88c807SRodney W. Grimes  *	This product includes software developed by the University of
194b88c807SRodney W. Grimes  *	California, Berkeley and its contributors.
204b88c807SRodney W. Grimes  * 4. Neither the name of the University nor the names of its contributors
214b88c807SRodney W. Grimes  *    may be used to endorse or promote products derived from this software
224b88c807SRodney W. Grimes  *    without specific prior written permission.
234b88c807SRodney W. Grimes  *
244b88c807SRodney W. Grimes  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
254b88c807SRodney W. Grimes  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
264b88c807SRodney W. Grimes  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
274b88c807SRodney W. Grimes  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
284b88c807SRodney W. Grimes  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
294b88c807SRodney W. Grimes  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
304b88c807SRodney W. Grimes  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
314b88c807SRodney W. Grimes  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
324b88c807SRodney W. Grimes  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
334b88c807SRodney W. Grimes  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
344b88c807SRodney W. Grimes  * SUCH DAMAGE.
354b88c807SRodney W. Grimes  */
364b88c807SRodney W. Grimes 
374b88c807SRodney W. Grimes #ifndef lint
38890acb95SSteve Price static char const copyright[] =
394b88c807SRodney W. Grimes "@(#) Copyright (c) 1989, 1993\n\
404b88c807SRodney W. Grimes 	The Regents of the University of California.  All rights reserved.\n";
414b88c807SRodney W. Grimes #endif /* not lint */
424b88c807SRodney W. Grimes 
434b88c807SRodney W. Grimes #ifndef lint
444c95995fSPhilippe Charnier #if 0
4512f93eb9SPhilippe Charnier static char sccsid[] = "@(#)cat.c	8.2 (Berkeley) 4/27/95";
464c95995fSPhilippe Charnier #endif
474c95995fSPhilippe Charnier static const char rcsid[] =
482a456239SPeter Wemm   "$FreeBSD$";
494b88c807SRodney W. Grimes #endif /* not lint */
504b88c807SRodney W. Grimes 
514b88c807SRodney W. Grimes #include <sys/param.h>
524b88c807SRodney W. Grimes #include <sys/stat.h>
534b88c807SRodney W. Grimes 
544b88c807SRodney W. Grimes #include <ctype.h>
554b88c807SRodney W. Grimes #include <err.h>
564b88c807SRodney W. Grimes #include <fcntl.h>
573043192bSAndrey A. Chernov #include <locale.h>
584b88c807SRodney W. Grimes #include <stdio.h>
594b88c807SRodney W. Grimes #include <stdlib.h>
60cafefe8cSDima Dorfman #include <string.h>
614b88c807SRodney W. Grimes #include <unistd.h>
624b88c807SRodney W. Grimes 
634b88c807SRodney W. Grimes int bflag, eflag, nflag, sflag, tflag, vflag;
644b88c807SRodney W. Grimes int rval;
65a5da0999SWarner Losh const char *filename;
664b88c807SRodney W. Grimes 
674b88c807SRodney W. Grimes void cook_args __P((char *argv[]));
684b88c807SRodney W. Grimes void cook_buf __P((FILE *));
69d1762d1fSWarner Losh int main __P((int argc, char *argv[]));
704b88c807SRodney W. Grimes void raw_args __P((char *argv[]));
714b88c807SRodney W. Grimes void raw_cat __P((int));
724b88c807SRodney W. Grimes 
734b88c807SRodney W. Grimes int
74d1762d1fSWarner Losh main(argc, argv)
75d1762d1fSWarner Losh 	int argc;
76d1762d1fSWarner Losh 	char *argv[];
774b88c807SRodney W. Grimes {
784b88c807SRodney W. Grimes 	int ch;
794b88c807SRodney W. Grimes 
803043192bSAndrey A. Chernov 	setlocale(LC_CTYPE, "");
813043192bSAndrey A. Chernov 
8293ef08afSWarner Losh 	while ((ch = getopt(argc, argv, "benstuv")) != -1)
834b88c807SRodney W. Grimes 		switch (ch) {
844b88c807SRodney W. Grimes 		case 'b':
854b88c807SRodney W. Grimes 			bflag = nflag = 1;	/* -b implies -n */
864b88c807SRodney W. Grimes 			break;
874b88c807SRodney W. Grimes 		case 'e':
884b88c807SRodney W. Grimes 			eflag = vflag = 1;	/* -e implies -v */
894b88c807SRodney W. Grimes 			break;
904b88c807SRodney W. Grimes 		case 'n':
914b88c807SRodney W. Grimes 			nflag = 1;
924b88c807SRodney W. Grimes 			break;
934b88c807SRodney W. Grimes 		case 's':
944b88c807SRodney W. Grimes 			sflag = 1;
954b88c807SRodney W. Grimes 			break;
964b88c807SRodney W. Grimes 		case 't':
974b88c807SRodney W. Grimes 			tflag = vflag = 1;	/* -t implies -v */
984b88c807SRodney W. Grimes 			break;
994b88c807SRodney W. Grimes 		case 'u':
1002192b407SJeroen Ruigrok van der Werven 			setbuf(stdout, NULL);
1014b88c807SRodney W. Grimes 			break;
1024b88c807SRodney W. Grimes 		case 'v':
1034b88c807SRodney W. Grimes 			vflag = 1;
1044b88c807SRodney W. Grimes 			break;
1058d72a3d7SWarner Losh 		default:
1064b88c807SRodney W. Grimes 			(void)fprintf(stderr,
1074b88c807SRodney W. Grimes 			    "usage: cat [-benstuv] [-] [file ...]\n");
1084b88c807SRodney W. Grimes 			exit(1);
1094b88c807SRodney W. Grimes 		}
1104b88c807SRodney W. Grimes 	argv += optind;
1114b88c807SRodney W. Grimes 
1124b88c807SRodney W. Grimes 	if (bflag || eflag || nflag || sflag || tflag || vflag)
1134b88c807SRodney W. Grimes 		cook_args(argv);
1144b88c807SRodney W. Grimes 	else
1154b88c807SRodney W. Grimes 		raw_args(argv);
1164b88c807SRodney W. Grimes 	if (fclose(stdout))
1174b88c807SRodney W. Grimes 		err(1, "stdout");
1184b88c807SRodney W. Grimes 	exit(rval);
1194b88c807SRodney W. Grimes }
1204b88c807SRodney W. Grimes 
1214b88c807SRodney W. Grimes void
1224b88c807SRodney W. Grimes cook_args(argv)
1234b88c807SRodney W. Grimes 	char **argv;
1244b88c807SRodney W. Grimes {
1254b88c807SRodney W. Grimes 	register FILE *fp;
1264b88c807SRodney W. Grimes 
1274b88c807SRodney W. Grimes 	fp = stdin;
1284b88c807SRodney W. Grimes 	filename = "stdin";
1294b88c807SRodney W. Grimes 	do {
1304b88c807SRodney W. Grimes 		if (*argv) {
1314b88c807SRodney W. Grimes 			if (!strcmp(*argv, "-"))
1324b88c807SRodney W. Grimes 				fp = stdin;
1334b88c807SRodney W. Grimes 			else if ((fp = fopen(*argv, "r")) == NULL) {
1344b88c807SRodney W. Grimes 				warn("%s", *argv);
135001aff9fSBruce Evans 				rval = 1;
1364b88c807SRodney W. Grimes 				++argv;
1374b88c807SRodney W. Grimes 				continue;
1384b88c807SRodney W. Grimes 			}
1394b88c807SRodney W. Grimes 			filename = *argv++;
1404b88c807SRodney W. Grimes 		}
1414b88c807SRodney W. Grimes 		cook_buf(fp);
1424b88c807SRodney W. Grimes 		if (fp != stdin)
1434b88c807SRodney W. Grimes 			(void)fclose(fp);
1444b88c807SRodney W. Grimes 	} while (*argv);
1454b88c807SRodney W. Grimes }
1464b88c807SRodney W. Grimes 
1474b88c807SRodney W. Grimes void
1484b88c807SRodney W. Grimes cook_buf(fp)
1494b88c807SRodney W. Grimes 	register FILE *fp;
1504b88c807SRodney W. Grimes {
1514b88c807SRodney W. Grimes 	register int ch, gobble, line, prev;
1524b88c807SRodney W. Grimes 
1534b88c807SRodney W. Grimes 	line = gobble = 0;
1544b88c807SRodney W. Grimes 	for (prev = '\n'; (ch = getc(fp)) != EOF; prev = ch) {
1554b88c807SRodney W. Grimes 		if (prev == '\n') {
1564b88c807SRodney W. Grimes 			if (ch == '\n') {
1574b88c807SRodney W. Grimes 				if (sflag) {
1584b88c807SRodney W. Grimes 					if (!gobble && putchar(ch) == EOF)
1594b88c807SRodney W. Grimes 						break;
1604b88c807SRodney W. Grimes 					gobble = 1;
1614b88c807SRodney W. Grimes 					continue;
1624b88c807SRodney W. Grimes 				}
1634b88c807SRodney W. Grimes 				if (nflag && !bflag) {
1644b88c807SRodney W. Grimes 					(void)fprintf(stdout, "%6d\t", ++line);
1654b88c807SRodney W. Grimes 					if (ferror(stdout))
1664b88c807SRodney W. Grimes 						break;
1674b88c807SRodney W. Grimes 				}
1684b88c807SRodney W. Grimes 			} else if (nflag) {
1694b88c807SRodney W. Grimes 				(void)fprintf(stdout, "%6d\t", ++line);
1704b88c807SRodney W. Grimes 				if (ferror(stdout))
1714b88c807SRodney W. Grimes 					break;
1724b88c807SRodney W. Grimes 			}
1734b88c807SRodney W. Grimes 		}
1744b88c807SRodney W. Grimes 		gobble = 0;
1754b88c807SRodney W. Grimes 		if (ch == '\n') {
1764b88c807SRodney W. Grimes 			if (eflag)
1774b88c807SRodney W. Grimes 				if (putchar('$') == EOF)
1784b88c807SRodney W. Grimes 					break;
1794b88c807SRodney W. Grimes 		} else if (ch == '\t') {
1804b88c807SRodney W. Grimes 			if (tflag) {
1814b88c807SRodney W. Grimes 				if (putchar('^') == EOF || putchar('I') == EOF)
1824b88c807SRodney W. Grimes 					break;
1834b88c807SRodney W. Grimes 				continue;
1844b88c807SRodney W. Grimes 			}
1854b88c807SRodney W. Grimes 		} else if (vflag) {
1863043192bSAndrey A. Chernov 			if (!isascii(ch) && !isprint(ch)) {
1874b88c807SRodney W. Grimes 				if (putchar('M') == EOF || putchar('-') == EOF)
1884b88c807SRodney W. Grimes 					break;
1894b88c807SRodney W. Grimes 				ch = toascii(ch);
1904b88c807SRodney W. Grimes 			}
1914b88c807SRodney W. Grimes 			if (iscntrl(ch)) {
1924b88c807SRodney W. Grimes 				if (putchar('^') == EOF ||
1934b88c807SRodney W. Grimes 				    putchar(ch == '\177' ? '?' :
1944b88c807SRodney W. Grimes 				    ch | 0100) == EOF)
1954b88c807SRodney W. Grimes 					break;
1964b88c807SRodney W. Grimes 				continue;
1974b88c807SRodney W. Grimes 			}
1984b88c807SRodney W. Grimes 		}
1994b88c807SRodney W. Grimes 		if (putchar(ch) == EOF)
2004b88c807SRodney W. Grimes 			break;
2014b88c807SRodney W. Grimes 	}
2024b88c807SRodney W. Grimes 	if (ferror(fp)) {
2034b88c807SRodney W. Grimes 		warn("%s", filename);
204001aff9fSBruce Evans 		rval = 1;
2054b88c807SRodney W. Grimes 		clearerr(fp);
2064b88c807SRodney W. Grimes 	}
2074b88c807SRodney W. Grimes 	if (ferror(stdout))
2084b88c807SRodney W. Grimes 		err(1, "stdout");
2094b88c807SRodney W. Grimes }
2104b88c807SRodney W. Grimes 
2114b88c807SRodney W. Grimes void
2124b88c807SRodney W. Grimes raw_args(argv)
2134b88c807SRodney W. Grimes 	char **argv;
2144b88c807SRodney W. Grimes {
2154b88c807SRodney W. Grimes 	register int fd;
2164b88c807SRodney W. Grimes 
2174b88c807SRodney W. Grimes 	fd = fileno(stdin);
2184b88c807SRodney W. Grimes 	filename = "stdin";
2194b88c807SRodney W. Grimes 	do {
2204b88c807SRodney W. Grimes 		if (*argv) {
2214b88c807SRodney W. Grimes 			if (!strcmp(*argv, "-"))
2224b88c807SRodney W. Grimes 				fd = fileno(stdin);
2234b88c807SRodney W. Grimes 			else if ((fd = open(*argv, O_RDONLY, 0)) < 0) {
2244b88c807SRodney W. Grimes 				warn("%s", *argv);
225001aff9fSBruce Evans 				rval = 1;
2264b88c807SRodney W. Grimes 				++argv;
2274b88c807SRodney W. Grimes 				continue;
2284b88c807SRodney W. Grimes 			}
2294b88c807SRodney W. Grimes 			filename = *argv++;
2304b88c807SRodney W. Grimes 		}
2314b88c807SRodney W. Grimes 		raw_cat(fd);
2324b88c807SRodney W. Grimes 		if (fd != fileno(stdin))
2334b88c807SRodney W. Grimes 			(void)close(fd);
2344b88c807SRodney W. Grimes 	} while (*argv);
2354b88c807SRodney W. Grimes }
2364b88c807SRodney W. Grimes 
2374b88c807SRodney W. Grimes void
2384b88c807SRodney W. Grimes raw_cat(rfd)
2394b88c807SRodney W. Grimes 	register int rfd;
2404b88c807SRodney W. Grimes {
241a5da0999SWarner Losh 	register int off, wfd;
242a5da0999SWarner Losh 	ssize_t nr, nw;
243a5da0999SWarner Losh 	static size_t bsize;
2444b88c807SRodney W. Grimes 	static char *buf;
2454b88c807SRodney W. Grimes 	struct stat sbuf;
2464b88c807SRodney W. Grimes 
2474b88c807SRodney W. Grimes 	wfd = fileno(stdout);
2484b88c807SRodney W. Grimes 	if (buf == NULL) {
2494b88c807SRodney W. Grimes 		if (fstat(wfd, &sbuf))
2504b88c807SRodney W. Grimes 			err(1, "%s", filename);
2514b88c807SRodney W. Grimes 		bsize = MAX(sbuf.st_blksize, 1024);
252d1762d1fSWarner Losh 		if ((buf = malloc(bsize)) == NULL)
2532192b407SJeroen Ruigrok van der Werven 			err(1, "buffer");
2544b88c807SRodney W. Grimes 	}
2554b88c807SRodney W. Grimes 	while ((nr = read(rfd, buf, bsize)) > 0)
2564b88c807SRodney W. Grimes 		for (off = 0; nr; nr -= nw, off += nw)
257a5da0999SWarner Losh 			if ((nw = write(wfd, buf + off, (size_t)nr)) < 0)
2584b88c807SRodney W. Grimes 				err(1, "stdout");
259001aff9fSBruce Evans 	if (nr < 0) {
2604b88c807SRodney W. Grimes 		warn("%s", filename);
261001aff9fSBruce Evans 		rval = 1;
262001aff9fSBruce Evans 	}
2634b88c807SRodney W. Grimes }
264