xref: /openbsd/usr.bin/vis/vis.c (revision d7259957)
1 /*	$OpenBSD: vis.c,v 1.22 2022/12/04 23:50:50 cheloha Exp $	*/
2 /*	$NetBSD: vis.c,v 1.4 1994/12/20 16:13:03 jtc Exp $	*/
3 
4 /*-
5  * Copyright (c) 1989, 1993
6  *	The Regents of the University of California.  All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 3. Neither the name of the University nor the names of its contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30  * SUCH DAMAGE.
31  */
32 
33 #include <stdio.h>
34 #include <string.h>
35 #include <stdlib.h>
36 #include <unistd.h>
37 #include <limits.h>
38 #include <err.h>
39 #include <vis.h>
40 
41 int eflags, fold, foldwidth=80, none, markeol;
42 
43 #ifdef DEBUG
44 int debug;
45 #endif
46 
47 int foldit(char *, int, int);
48 void process(FILE *);
49 __dead void usage(void);
50 
51 int
main(int argc,char * argv[])52 main(int argc, char *argv[])
53 {
54 	const char *errstr;
55 	FILE *fp;
56 	int ch;
57 
58 	if (pledge("stdio rpath", NULL) == -1)
59 		err(1, "pledge");
60 
61 	while ((ch = getopt(argc, argv, "anwctsobfF:ld")) != -1)
62 		switch(ch) {
63 		case 'a':
64 			eflags |= VIS_ALL;
65 			break;
66 		case 'n':
67 			none = 1;
68 			break;
69 		case 'w':
70 			eflags |= VIS_WHITE;
71 			break;
72 		case 'c':
73 			eflags |= VIS_CSTYLE;
74 			break;
75 		case 't':
76 			eflags |= VIS_TAB;
77 			break;
78 		case 's':
79 			eflags |= VIS_SAFE;
80 			break;
81 		case 'o':
82 			eflags |= VIS_OCTAL;
83 			break;
84 		case 'b':
85 			eflags |= VIS_NOSLASH;
86 			break;
87 		case 'F':
88 			foldwidth = strtonum(optarg, 1, INT_MAX, &errstr);
89 			if (errstr)
90 				errx(1, "%s: %s", optarg, errstr);
91 			if (foldwidth < 5)
92 				errx(1, "can't fold lines to less than 5 cols");
93 			/*FALLTHROUGH*/
94 		case 'f':
95 			fold = 1;	/* fold output lines to 80 cols */
96 			break;		/* using hidden newline */
97 		case 'l':
98 			markeol = 1;	/* mark end of line with \$ */
99 			break;
100 #ifdef DEBUG
101 		case 'd':
102 			debug = 1;
103 			break;
104 #endif
105 		default:
106 			usage();
107 		}
108 	argc -= optind;
109 	argv += optind;
110 
111 	if (*argv)
112 		while (*argv) {
113 			if ((fp=fopen(*argv, "r")) != NULL) {
114 				process(fp);
115 				fclose(fp);
116 			} else
117 				warn("%s", *argv);
118 			argv++;
119 		}
120 	else {
121 		if (pledge("stdio", NULL) == -1)
122 			err(1, "pledge");
123 		process(stdin);
124 	}
125 	exit(0);
126 }
127 
128 void
process(FILE * fp)129 process(FILE *fp)
130 {
131 	static int col = 0;
132 	char *cp = "\0";
133 	int c, rachar;
134 	char buff[5];
135 
136 	cp++;			/* so *(cp-1) starts out != '\n' */
137 	c = getc(fp);
138 	while (c != EOF) {
139 		rachar = getc(fp);
140 		if (none) {
141 			cp = buff;
142 			*cp++ = c;
143 			if (c == '\\')
144 				*cp++ = '\\';
145 			*cp = '\0';
146 		} else if (markeol && c == '\n') {
147 			cp = buff;
148 			if ((eflags & VIS_NOSLASH) == 0)
149 				*cp++ = '\\';
150 			*cp++ = '$';
151 			*cp++ = '\n';
152 			*cp = '\0';
153 		} else
154 			(void) vis(buff, (char)c, eflags, (char)rachar);
155 
156 		cp = buff;
157 		if (fold) {
158 #ifdef DEBUG
159 			if (debug)
160 				printf("<%02d,", col);
161 #endif
162 			col = foldit(cp, col, foldwidth);
163 #ifdef DEBUG
164 			if (debug)
165 				printf("%02d>", col);
166 #endif
167 		}
168 		do {
169 			putchar(*cp);
170 		} while (*++cp);
171 		c = rachar;
172 	}
173 	/*
174 	 * terminate partial line with a hidden newline
175 	 */
176 	if (fold && *(cp-1) != '\n')
177 		printf("\\\n");
178 }
179 
180 __dead void
usage(void)181 usage(void)
182 {
183 	extern char *__progname;
184 
185 	fprintf(stderr, "usage: %s [-abcflnostw] [-F foldwidth] [file ...]\n",
186 	    __progname);
187 	exit(1);
188 }
189