xref: /dragonfly/lib/libc/gen/vis.c (revision 984263bc)
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. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *	This product includes software developed by the University of
16  *	California, Berkeley and its contributors.
17  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  *
33  * $FreeBSD: src/lib/libc/gen/vis.c,v 1.5.8.2 2001/03/05 09:44:34 obrien Exp $
34  */
35 
36 #if defined(LIBC_SCCS) && !defined(lint)
37 static char sccsid[] = "@(#)vis.c	8.1 (Berkeley) 7/19/93";
38 #endif /* LIBC_SCCS and not lint */
39 
40 #include <sys/types.h>
41 #include <limits.h>
42 #include <ctype.h>
43 #include <stdio.h>
44 #include <vis.h>
45 
46 #define	isoctal(c)	(((u_char)(c)) >= '0' && ((u_char)(c)) <= '7')
47 
48 /*
49  * vis - visually encode characters
50  */
51 char *
52 vis(dst, c, flag, nextc)
53 	register char *dst;
54 	int c, nextc;
55 	register int flag;
56 {
57 	c = (unsigned char)c;
58 
59 	if (flag & VIS_HTTPSTYLE) {
60 		/* Described in RFC 1808 */
61 		if (!(isalnum(c) /* alpha-numeric */
62 		    /* safe */
63 		    || c == '$' || c == '-' || c == '_' || c == '.' || c == '+'
64 		    /* extra */
65 		    || c == '!' || c == '*' || c == '\'' || c == '('
66 		    || c == ')' || c == ',')) {
67 			*dst++ = '%';
68 			snprintf(dst, 4, (c < 16 ? "0%X" : "%X"), c);
69 			dst += 2;
70 			goto done;
71 		}
72 	}
73 
74 	if (isgraph(c) ||
75 	   ((flag & VIS_SP) == 0 && c == ' ') ||
76 	   ((flag & VIS_TAB) == 0 && c == '\t') ||
77 	   ((flag & VIS_NL) == 0 && c == '\n') ||
78 	   ((flag & VIS_SAFE) && (c == '\b' || c == '\007' || c == '\r'))) {
79 		*dst++ = c;
80 		if (c == '\\' && (flag & VIS_NOSLASH) == 0)
81 			*dst++ = '\\';
82 		*dst = '\0';
83 		return (dst);
84 	}
85 
86 	if (flag & VIS_CSTYLE) {
87 		switch(c) {
88 		case '\n':
89 			*dst++ = '\\';
90 			*dst++ = 'n';
91 			goto done;
92 		case '\r':
93 			*dst++ = '\\';
94 			*dst++ = 'r';
95 			goto done;
96 		case '\b':
97 			*dst++ = '\\';
98 			*dst++ = 'b';
99 			goto done;
100 #if __STDC__
101 		case '\a':
102 #else
103 		case '\007':
104 #endif
105 			*dst++ = '\\';
106 			*dst++ = 'a';
107 			goto done;
108 		case '\v':
109 			*dst++ = '\\';
110 			*dst++ = 'v';
111 			goto done;
112 		case '\t':
113 			*dst++ = '\\';
114 			*dst++ = 't';
115 			goto done;
116 		case '\f':
117 			*dst++ = '\\';
118 			*dst++ = 'f';
119 			goto done;
120 		case ' ':
121 			*dst++ = '\\';
122 			*dst++ = 's';
123 			goto done;
124 		case '\0':
125 			*dst++ = '\\';
126 			*dst++ = '0';
127 			if (isoctal(nextc)) {
128 				*dst++ = '0';
129 				*dst++ = '0';
130 			}
131 			goto done;
132 		}
133 	}
134 	if (((c & 0177) == ' ') || (flag & VIS_OCTAL)) {
135 		*dst++ = '\\';
136 		*dst++ = ((u_char)c >> 6 & 07) + '0';
137 		*dst++ = ((u_char)c >> 3 & 07) + '0';
138 		*dst++ = ((u_char)c & 07) + '0';
139 		goto done;
140 	}
141 	if ((flag & VIS_NOSLASH) == 0)
142 		*dst++ = '\\';
143 	if (c & 0200) {
144 		c &= 0177;
145 		*dst++ = 'M';
146 	}
147 	if (iscntrl(c)) {
148 		*dst++ = '^';
149 		if (c == 0177)
150 			*dst++ = '?';
151 		else
152 			*dst++ = c + '@';
153 	} else {
154 		*dst++ = '-';
155 		*dst++ = c;
156 	}
157 done:
158 	*dst = '\0';
159 	return (dst);
160 }
161 
162 /*
163  * strvis, strvisx - visually encode characters from src into dst
164  *
165  *	Dst must be 4 times the size of src to account for possible
166  *	expansion.  The length of dst, not including the trailing NULL,
167  *	is returned.
168  *
169  *	Strvisx encodes exactly len bytes from src into dst.
170  *	This is useful for encoding a block of data.
171  */
172 int
173 strvis(dst, src, flag)
174 	register char *dst;
175 	register const char *src;
176 	int flag;
177 {
178 	register char c;
179 	char *start;
180 
181 	for (start = dst; (c = *src); )
182 		dst = vis(dst, c, flag, *++src);
183 	*dst = '\0';
184 	return (dst - start);
185 }
186 
187 int
188 strvisx(dst, src, len, flag)
189 	register char *dst;
190 	register const char *src;
191 	register size_t len;
192 	int flag;
193 {
194 	int c;
195 	char *start;
196 
197 	for (start = dst; len > 1; len--) {
198 		c = *src;
199 		dst = vis(dst, c, flag, *++src);
200 	}
201 	if (len)
202 		dst = vis(dst, *src, flag, '\0');
203 	*dst = '\0';
204 
205 	return (dst - start);
206 }
207