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