xref: /freebsd/contrib/libc-vis/vis.c (revision 1a2f06d0f2905c9a18340b377cbbe772f2ca6844)
1ff88ef41SBrooks Davis /*	$NetBSD: vis.c,v 1.74 2017/11/27 16:37:21 christos Exp $	*/
28ccca122SBrooks Davis 
38ccca122SBrooks Davis /*-
48ccca122SBrooks Davis  * Copyright (c) 1989, 1993
58ccca122SBrooks Davis  *	The Regents of the University of California.  All rights reserved.
68ccca122SBrooks Davis  *
78ccca122SBrooks Davis  * Redistribution and use in source and binary forms, with or without
88ccca122SBrooks Davis  * modification, are permitted provided that the following conditions
98ccca122SBrooks Davis  * are met:
108ccca122SBrooks Davis  * 1. Redistributions of source code must retain the above copyright
118ccca122SBrooks Davis  *    notice, this list of conditions and the following disclaimer.
128ccca122SBrooks Davis  * 2. Redistributions in binary form must reproduce the above copyright
138ccca122SBrooks Davis  *    notice, this list of conditions and the following disclaimer in the
148ccca122SBrooks Davis  *    documentation and/or other materials provided with the distribution.
158ccca122SBrooks Davis  * 3. Neither the name of the University nor the names of its contributors
168ccca122SBrooks Davis  *    may be used to endorse or promote products derived from this software
178ccca122SBrooks Davis  *    without specific prior written permission.
188ccca122SBrooks Davis  *
198ccca122SBrooks Davis  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
208ccca122SBrooks Davis  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
218ccca122SBrooks Davis  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
228ccca122SBrooks Davis  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
238ccca122SBrooks Davis  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
248ccca122SBrooks Davis  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
258ccca122SBrooks Davis  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
268ccca122SBrooks Davis  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
278ccca122SBrooks Davis  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
288ccca122SBrooks Davis  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
298ccca122SBrooks Davis  * SUCH DAMAGE.
308ccca122SBrooks Davis  */
318ccca122SBrooks Davis 
328ccca122SBrooks Davis /*-
338ccca122SBrooks Davis  * Copyright (c) 1999, 2005 The NetBSD Foundation, Inc.
348ccca122SBrooks Davis  * All rights reserved.
358ccca122SBrooks Davis  *
368ccca122SBrooks Davis  * Redistribution and use in source and binary forms, with or without
378ccca122SBrooks Davis  * modification, are permitted provided that the following conditions
388ccca122SBrooks Davis  * are met:
398ccca122SBrooks Davis  * 1. Redistributions of source code must retain the above copyright
408ccca122SBrooks Davis  *    notice, this list of conditions and the following disclaimer.
418ccca122SBrooks Davis  * 2. Redistributions in binary form must reproduce the above copyright
428ccca122SBrooks Davis  *    notice, this list of conditions and the following disclaimer in the
438ccca122SBrooks Davis  *    documentation and/or other materials provided with the distribution.
448ccca122SBrooks Davis  *
458ccca122SBrooks Davis  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
468ccca122SBrooks Davis  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
478ccca122SBrooks Davis  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
488ccca122SBrooks Davis  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
498ccca122SBrooks Davis  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
508ccca122SBrooks Davis  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
518ccca122SBrooks Davis  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
528ccca122SBrooks Davis  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
538ccca122SBrooks Davis  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
548ccca122SBrooks Davis  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
558ccca122SBrooks Davis  * POSSIBILITY OF SUCH DAMAGE.
568ccca122SBrooks Davis  */
578ccca122SBrooks Davis 
588ccca122SBrooks Davis #include <sys/cdefs.h>
598ccca122SBrooks Davis #if defined(LIBC_SCCS) && !defined(lint)
60ff88ef41SBrooks Davis __RCSID("$NetBSD: vis.c,v 1.74 2017/11/27 16:37:21 christos Exp $");
618ccca122SBrooks Davis #endif /* LIBC_SCCS and not lint */
62778c12a6SBrooks Davis #ifdef __FBSDID
638ccca122SBrooks Davis __FBSDID("$FreeBSD$");
64778c12a6SBrooks Davis #define	_DIAGASSERT(x)	assert(x)
65778c12a6SBrooks Davis #endif
668ccca122SBrooks Davis 
678ccca122SBrooks Davis #include "namespace.h"
688ccca122SBrooks Davis #include <sys/types.h>
69778c12a6SBrooks Davis #include <sys/param.h>
708ccca122SBrooks Davis 
718ccca122SBrooks Davis #include <assert.h>
728ccca122SBrooks Davis #include <vis.h>
738ccca122SBrooks Davis #include <errno.h>
748ccca122SBrooks Davis #include <stdlib.h>
75778c12a6SBrooks Davis #include <wchar.h>
76778c12a6SBrooks Davis #include <wctype.h>
778ccca122SBrooks Davis 
788ccca122SBrooks Davis #ifdef __weak_alias
798ccca122SBrooks Davis __weak_alias(strvisx,_strvisx)
808ccca122SBrooks Davis #endif
818ccca122SBrooks Davis 
828ccca122SBrooks Davis #if !HAVE_VIS || !HAVE_SVIS
838ccca122SBrooks Davis #include <ctype.h>
848ccca122SBrooks Davis #include <limits.h>
858ccca122SBrooks Davis #include <stdio.h>
868ccca122SBrooks Davis #include <string.h>
878ccca122SBrooks Davis 
88778c12a6SBrooks Davis /*
89778c12a6SBrooks Davis  * The reason for going through the trouble to deal with character encodings
90778c12a6SBrooks Davis  * in vis(3), is that we use this to safe encode output of commands. This
91778c12a6SBrooks Davis  * safe encoding varies depending on the character set. For example if we
92778c12a6SBrooks Davis  * display ps output in French, we don't want to display French characters
93778c12a6SBrooks Davis  * as M-foo.
94778c12a6SBrooks Davis  */
95778c12a6SBrooks Davis 
96778c12a6SBrooks Davis static wchar_t *do_svis(wchar_t *, wint_t, int, wint_t, const wchar_t *);
978ccca122SBrooks Davis 
988ccca122SBrooks Davis #undef BELL
99778c12a6SBrooks Davis #define BELL L'\a'
1008ccca122SBrooks Davis 
1018dfeba04SBrooks Davis #if defined(LC_C_LOCALE)
1028dfeba04SBrooks Davis #define iscgraph(c)      isgraph_l(c, LC_C_LOCALE)
1038dfeba04SBrooks Davis #else
1048dfeba04SBrooks Davis /* Keep it simple for now, no locale stuff */
1058dfeba04SBrooks Davis #define iscgraph(c)	isgraph(c)
1068dfeba04SBrooks Davis #ifdef notyet
1078dfeba04SBrooks Davis #include <locale.h>
1088dfeba04SBrooks Davis static int
1098dfeba04SBrooks Davis iscgraph(int c) {
1108dfeba04SBrooks Davis 	int rv;
1118dfeba04SBrooks Davis 	char *ol;
1128dfeba04SBrooks Davis 
1138dfeba04SBrooks Davis 	ol = setlocale(LC_CTYPE, "C");
1148dfeba04SBrooks Davis 	rv = isgraph(c);
1158dfeba04SBrooks Davis 	if (ol)
1168dfeba04SBrooks Davis 		setlocale(LC_CTYPE, ol);
1178dfeba04SBrooks Davis 	return rv;
1188dfeba04SBrooks Davis }
1198dfeba04SBrooks Davis #endif
1208dfeba04SBrooks Davis #endif
1218dfeba04SBrooks Davis 
1228dfeba04SBrooks Davis #define ISGRAPH(flags, c) \
1238dfeba04SBrooks Davis     (((flags) & VIS_NOLOCALE) ? iscgraph(c) : iswgraph(c))
1248dfeba04SBrooks Davis 
125778c12a6SBrooks Davis #define iswoctal(c)	(((u_char)(c)) >= L'0' && ((u_char)(c)) <= L'7')
126778c12a6SBrooks Davis #define iswwhite(c)	(c == L' ' || c == L'\t' || c == L'\n')
127778c12a6SBrooks Davis #define iswsafe(c)	(c == L'\b' || c == BELL || c == L'\r')
128778c12a6SBrooks Davis #define xtoa(c)		L"0123456789abcdef"[c]
129778c12a6SBrooks Davis #define XTOA(c)		L"0123456789ABCDEF"[c]
1308ccca122SBrooks Davis 
1318dfeba04SBrooks Davis #define MAXEXTRAS	30
1328dfeba04SBrooks Davis 
1338dfeba04SBrooks Davis static const wchar_t char_shell[] = L"'`\";&<>()|{}]\\$!^~";
1348dfeba04SBrooks Davis static const wchar_t char_glob[] = L"*?[#";
1358ccca122SBrooks Davis 
136778c12a6SBrooks Davis #if !HAVE_NBTOOL_CONFIG_H
137778c12a6SBrooks Davis #ifndef __NetBSD__
138778c12a6SBrooks Davis /*
139778c12a6SBrooks Davis  * On NetBSD MB_LEN_MAX is currently 32 which does not fit on any integer
140778c12a6SBrooks Davis  * integral type and it is probably wrong, since currently the maximum
141778c12a6SBrooks Davis  * number of bytes and character needs is 6. Until this is fixed, the
142778c12a6SBrooks Davis  * loops below are using sizeof(uint64_t) - 1 instead of MB_LEN_MAX, and
143778c12a6SBrooks Davis  * the assertion is commented out.
144778c12a6SBrooks Davis  */
145778c12a6SBrooks Davis #ifdef __FreeBSD__
146778c12a6SBrooks Davis /*
147778c12a6SBrooks Davis  * On FreeBSD including <sys/systm.h> for CTASSERT only works in kernel
148778c12a6SBrooks Davis  * mode.
149778c12a6SBrooks Davis  */
150778c12a6SBrooks Davis #ifndef CTASSERT
151778c12a6SBrooks Davis #define CTASSERT(x)             _CTASSERT(x, __LINE__)
152778c12a6SBrooks Davis #define _CTASSERT(x, y)         __CTASSERT(x, y)
153778c12a6SBrooks Davis #define __CTASSERT(x, y)        typedef char __assert ## y[(x) ? 1 : -1]
154778c12a6SBrooks Davis #endif
155778c12a6SBrooks Davis #endif /* __FreeBSD__ */
156778c12a6SBrooks Davis CTASSERT(MB_LEN_MAX <= sizeof(uint64_t));
157778c12a6SBrooks Davis #endif /* !__NetBSD__ */
158778c12a6SBrooks Davis #endif
1598ccca122SBrooks Davis 
1608ccca122SBrooks Davis /*
1618ccca122SBrooks Davis  * This is do_hvis, for HTTP style (RFC 1808)
1628ccca122SBrooks Davis  */
163778c12a6SBrooks Davis static wchar_t *
164778c12a6SBrooks Davis do_hvis(wchar_t *dst, wint_t c, int flags, wint_t nextc, const wchar_t *extra)
1658ccca122SBrooks Davis {
166778c12a6SBrooks Davis 	if (iswalnum(c)
1678ccca122SBrooks Davis 	    /* safe */
168778c12a6SBrooks Davis 	    || c == L'$' || c == L'-' || c == L'_' || c == L'.' || c == L'+'
1698ccca122SBrooks Davis 	    /* extra */
170778c12a6SBrooks Davis 	    || c == L'!' || c == L'*' || c == L'\'' || c == L'(' || c == L')'
171778c12a6SBrooks Davis 	    || c == L',')
172778c12a6SBrooks Davis 		dst = do_svis(dst, c, flags, nextc, extra);
173778c12a6SBrooks Davis 	else {
174778c12a6SBrooks Davis 		*dst++ = L'%';
1758ccca122SBrooks Davis 		*dst++ = xtoa(((unsigned int)c >> 4) & 0xf);
1768ccca122SBrooks Davis 		*dst++ = xtoa((unsigned int)c & 0xf);
1778ccca122SBrooks Davis 	}
1788ccca122SBrooks Davis 
1798ccca122SBrooks Davis 	return dst;
1808ccca122SBrooks Davis }
1818ccca122SBrooks Davis 
1828ccca122SBrooks Davis /*
1838ccca122SBrooks Davis  * This is do_mvis, for Quoted-Printable MIME (RFC 2045)
1848ccca122SBrooks Davis  * NB: No handling of long lines or CRLF.
1858ccca122SBrooks Davis  */
186778c12a6SBrooks Davis static wchar_t *
187778c12a6SBrooks Davis do_mvis(wchar_t *dst, wint_t c, int flags, wint_t nextc, const wchar_t *extra)
1888ccca122SBrooks Davis {
189778c12a6SBrooks Davis 	if ((c != L'\n') &&
1908ccca122SBrooks Davis 	    /* Space at the end of the line */
191778c12a6SBrooks Davis 	    ((iswspace(c) && (nextc == L'\r' || nextc == L'\n')) ||
1928ccca122SBrooks Davis 	    /* Out of range */
193778c12a6SBrooks Davis 	    (!iswspace(c) && (c < 33 || (c > 60 && c < 62) || c > 126)) ||
1948ccca122SBrooks Davis 	    /* Specific char to be escaped */
195778c12a6SBrooks Davis 	    wcschr(L"#$@[\\]^`{|}~", c) != NULL)) {
196778c12a6SBrooks Davis 		*dst++ = L'=';
1978ccca122SBrooks Davis 		*dst++ = XTOA(((unsigned int)c >> 4) & 0xf);
1988ccca122SBrooks Davis 		*dst++ = XTOA((unsigned int)c & 0xf);
199778c12a6SBrooks Davis 	} else
200778c12a6SBrooks Davis 		dst = do_svis(dst, c, flags, nextc, extra);
201778c12a6SBrooks Davis 	return dst;
2028ccca122SBrooks Davis }
203778c12a6SBrooks Davis 
204778c12a6SBrooks Davis /*
205778c12a6SBrooks Davis  * Output single byte of multibyte character.
206778c12a6SBrooks Davis  */
207778c12a6SBrooks Davis static wchar_t *
208778c12a6SBrooks Davis do_mbyte(wchar_t *dst, wint_t c, int flags, wint_t nextc, int iswextra)
209778c12a6SBrooks Davis {
210778c12a6SBrooks Davis 	if (flags & VIS_CSTYLE) {
211778c12a6SBrooks Davis 		switch (c) {
212778c12a6SBrooks Davis 		case L'\n':
213778c12a6SBrooks Davis 			*dst++ = L'\\'; *dst++ = L'n';
214778c12a6SBrooks Davis 			return dst;
215778c12a6SBrooks Davis 		case L'\r':
216778c12a6SBrooks Davis 			*dst++ = L'\\'; *dst++ = L'r';
217778c12a6SBrooks Davis 			return dst;
218778c12a6SBrooks Davis 		case L'\b':
219778c12a6SBrooks Davis 			*dst++ = L'\\'; *dst++ = L'b';
220778c12a6SBrooks Davis 			return dst;
221778c12a6SBrooks Davis 		case BELL:
222778c12a6SBrooks Davis 			*dst++ = L'\\'; *dst++ = L'a';
223778c12a6SBrooks Davis 			return dst;
224778c12a6SBrooks Davis 		case L'\v':
225778c12a6SBrooks Davis 			*dst++ = L'\\'; *dst++ = L'v';
226778c12a6SBrooks Davis 			return dst;
227778c12a6SBrooks Davis 		case L'\t':
228778c12a6SBrooks Davis 			*dst++ = L'\\'; *dst++ = L't';
229778c12a6SBrooks Davis 			return dst;
230778c12a6SBrooks Davis 		case L'\f':
231778c12a6SBrooks Davis 			*dst++ = L'\\'; *dst++ = L'f';
232778c12a6SBrooks Davis 			return dst;
233778c12a6SBrooks Davis 		case L' ':
234778c12a6SBrooks Davis 			*dst++ = L'\\'; *dst++ = L's';
235778c12a6SBrooks Davis 			return dst;
236778c12a6SBrooks Davis 		case L'\0':
237778c12a6SBrooks Davis 			*dst++ = L'\\'; *dst++ = L'0';
238778c12a6SBrooks Davis 			if (iswoctal(nextc)) {
239778c12a6SBrooks Davis 				*dst++ = L'0';
240778c12a6SBrooks Davis 				*dst++ = L'0';
241778c12a6SBrooks Davis 			}
242778c12a6SBrooks Davis 			return dst;
2438dfeba04SBrooks Davis 		/* We cannot encode these characters in VIS_CSTYLE
2448dfeba04SBrooks Davis 		 * because they special meaning */
2458dfeba04SBrooks Davis 		case L'n':
2468dfeba04SBrooks Davis 		case L'r':
2478dfeba04SBrooks Davis 		case L'b':
2488dfeba04SBrooks Davis 		case L'a':
2498dfeba04SBrooks Davis 		case L'v':
2508dfeba04SBrooks Davis 		case L't':
2518dfeba04SBrooks Davis 		case L'f':
2528dfeba04SBrooks Davis 		case L's':
2538dfeba04SBrooks Davis 		case L'0':
2548dfeba04SBrooks Davis 		case L'M':
2558dfeba04SBrooks Davis 		case L'^':
2568dfeba04SBrooks Davis 		case L'$': /* vis(1) -l */
2578dfeba04SBrooks Davis 			break;
258778c12a6SBrooks Davis 		default:
2598dfeba04SBrooks Davis 			if (ISGRAPH(flags, c) && !iswoctal(c)) {
260778c12a6SBrooks Davis 				*dst++ = L'\\';
261778c12a6SBrooks Davis 				*dst++ = c;
262778c12a6SBrooks Davis 				return dst;
263778c12a6SBrooks Davis 			}
264778c12a6SBrooks Davis 		}
265778c12a6SBrooks Davis 	}
266778c12a6SBrooks Davis 	if (iswextra || ((c & 0177) == L' ') || (flags & VIS_OCTAL)) {
267778c12a6SBrooks Davis 		*dst++ = L'\\';
268778c12a6SBrooks Davis 		*dst++ = (u_char)(((u_int32_t)(u_char)c >> 6) & 03) + L'0';
269778c12a6SBrooks Davis 		*dst++ = (u_char)(((u_int32_t)(u_char)c >> 3) & 07) + L'0';
270778c12a6SBrooks Davis 		*dst++ =			     (c	      & 07) + L'0';
271778c12a6SBrooks Davis 	} else {
272778c12a6SBrooks Davis 		if ((flags & VIS_NOSLASH) == 0)
273778c12a6SBrooks Davis 			*dst++ = L'\\';
274778c12a6SBrooks Davis 
275778c12a6SBrooks Davis 		if (c & 0200) {
276778c12a6SBrooks Davis 			c &= 0177;
277778c12a6SBrooks Davis 			*dst++ = L'M';
278778c12a6SBrooks Davis 		}
279778c12a6SBrooks Davis 
280778c12a6SBrooks Davis 		if (iswcntrl(c)) {
281778c12a6SBrooks Davis 			*dst++ = L'^';
282778c12a6SBrooks Davis 			if (c == 0177)
283778c12a6SBrooks Davis 				*dst++ = L'?';
284778c12a6SBrooks Davis 			else
285778c12a6SBrooks Davis 				*dst++ = c + L'@';
286778c12a6SBrooks Davis 		} else {
287778c12a6SBrooks Davis 			*dst++ = L'-';
288778c12a6SBrooks Davis 			*dst++ = c;
289778c12a6SBrooks Davis 		}
290778c12a6SBrooks Davis 	}
291778c12a6SBrooks Davis 
2928ccca122SBrooks Davis 	return dst;
2938ccca122SBrooks Davis }
2948ccca122SBrooks Davis 
2958ccca122SBrooks Davis /*
2968ccca122SBrooks Davis  * This is do_vis, the central code of vis.
2978ccca122SBrooks Davis  * dst:	      Pointer to the destination buffer
2988ccca122SBrooks Davis  * c:	      Character to encode
299778c12a6SBrooks Davis  * flags:     Flags word
3008ccca122SBrooks Davis  * nextc:     The character following 'c'
3018ccca122SBrooks Davis  * extra:     Pointer to the list of extra characters to be
3028ccca122SBrooks Davis  *	      backslash-protected.
3038ccca122SBrooks Davis  */
304778c12a6SBrooks Davis static wchar_t *
305778c12a6SBrooks Davis do_svis(wchar_t *dst, wint_t c, int flags, wint_t nextc, const wchar_t *extra)
3068ccca122SBrooks Davis {
307778c12a6SBrooks Davis 	int iswextra, i, shft;
308778c12a6SBrooks Davis 	uint64_t bmsk, wmsk;
3098ccca122SBrooks Davis 
310778c12a6SBrooks Davis 	iswextra = wcschr(extra, c) != NULL;
3118dfeba04SBrooks Davis 	if (!iswextra && (ISGRAPH(flags, c) || iswwhite(c) ||
312778c12a6SBrooks Davis 	    ((flags & VIS_SAFE) && iswsafe(c)))) {
3138ccca122SBrooks Davis 		*dst++ = c;
3148ccca122SBrooks Davis 		return dst;
3158ccca122SBrooks Davis 	}
316778c12a6SBrooks Davis 
317778c12a6SBrooks Davis 	/* See comment in istrsenvisx() output loop, below. */
318778c12a6SBrooks Davis 	wmsk = 0;
319778c12a6SBrooks Davis 	for (i = sizeof(wmsk) - 1; i >= 0; i--) {
320778c12a6SBrooks Davis 		shft = i * NBBY;
321778c12a6SBrooks Davis 		bmsk = (uint64_t)0xffLL << shft;
322778c12a6SBrooks Davis 		wmsk |= bmsk;
323778c12a6SBrooks Davis 		if ((c & wmsk) || i == 0)
324778c12a6SBrooks Davis 			dst = do_mbyte(dst, (wint_t)(
325778c12a6SBrooks Davis 			    (uint64_t)(c & bmsk) >> shft),
326778c12a6SBrooks Davis 			    flags, nextc, iswextra);
3278ccca122SBrooks Davis 	}
3288ccca122SBrooks Davis 
3298ccca122SBrooks Davis 	return dst;
3308ccca122SBrooks Davis }
3318ccca122SBrooks Davis 
332778c12a6SBrooks Davis typedef wchar_t *(*visfun_t)(wchar_t *, wint_t, int, wint_t, const wchar_t *);
3338ccca122SBrooks Davis 
3348ccca122SBrooks Davis /*
3358ccca122SBrooks Davis  * Return the appropriate encoding function depending on the flags given.
3368ccca122SBrooks Davis  */
3378ccca122SBrooks Davis static visfun_t
338778c12a6SBrooks Davis getvisfun(int flags)
3398ccca122SBrooks Davis {
340778c12a6SBrooks Davis 	if (flags & VIS_HTTPSTYLE)
3418ccca122SBrooks Davis 		return do_hvis;
342778c12a6SBrooks Davis 	if (flags & VIS_MIMESTYLE)
3438ccca122SBrooks Davis 		return do_mvis;
3448ccca122SBrooks Davis 	return do_svis;
3458ccca122SBrooks Davis }
3468ccca122SBrooks Davis 
3478ccca122SBrooks Davis /*
348778c12a6SBrooks Davis  * Expand list of extra characters to not visually encode.
3498ccca122SBrooks Davis  */
350778c12a6SBrooks Davis static wchar_t *
351778c12a6SBrooks Davis makeextralist(int flags, const char *src)
3528ccca122SBrooks Davis {
353778c12a6SBrooks Davis 	wchar_t *dst, *d;
354778c12a6SBrooks Davis 	size_t len;
3558dfeba04SBrooks Davis 	const wchar_t *s;
356f0a6d7e2SBryan Drewery 	mbstate_t mbstate;
3578ccca122SBrooks Davis 
358f0a6d7e2SBryan Drewery 	bzero(&mbstate, sizeof(mbstate));
359778c12a6SBrooks Davis 	len = strlen(src);
360778c12a6SBrooks Davis 	if ((dst = calloc(len + MAXEXTRAS, sizeof(*dst))) == NULL)
3618ccca122SBrooks Davis 		return NULL;
362778c12a6SBrooks Davis 
363f0a6d7e2SBryan Drewery 	if ((flags & VIS_NOLOCALE) || mbsrtowcs(dst, &src, len, &mbstate) == (size_t)-1) {
364778c12a6SBrooks Davis 		size_t i;
365778c12a6SBrooks Davis 		for (i = 0; i < len; i++)
3668dfeba04SBrooks Davis 			dst[i] = (wchar_t)(u_char)src[i];
367778c12a6SBrooks Davis 		d = dst + len;
368778c12a6SBrooks Davis 	} else
369778c12a6SBrooks Davis 		d = dst + wcslen(dst);
370778c12a6SBrooks Davis 
3718dfeba04SBrooks Davis 	if (flags & VIS_GLOB)
3728dfeba04SBrooks Davis 		for (s = char_glob; *s; *d++ = *s++)
3738dfeba04SBrooks Davis 			continue;
3748dfeba04SBrooks Davis 
3758dfeba04SBrooks Davis 	if (flags & VIS_SHELL)
3768dfeba04SBrooks Davis 		for (s = char_shell; *s; *d++ = *s++)
3778dfeba04SBrooks Davis 			continue;
378778c12a6SBrooks Davis 
379778c12a6SBrooks Davis 	if (flags & VIS_SP) *d++ = L' ';
380778c12a6SBrooks Davis 	if (flags & VIS_TAB) *d++ = L'\t';
381778c12a6SBrooks Davis 	if (flags & VIS_NL) *d++ = L'\n';
382ff88ef41SBrooks Davis 	if (flags & VIS_DQ) *d++ = L'"';
383778c12a6SBrooks Davis 	if ((flags & VIS_NOSLASH) == 0) *d++ = L'\\';
384778c12a6SBrooks Davis 	*d = L'\0';
385778c12a6SBrooks Davis 
3868ccca122SBrooks Davis 	return dst;
3878ccca122SBrooks Davis }
3888ccca122SBrooks Davis 
3898ccca122SBrooks Davis /*
390778c12a6SBrooks Davis  * istrsenvisx()
391778c12a6SBrooks Davis  * 	The main internal function.
392778c12a6SBrooks Davis  *	All user-visible functions call this one.
3938ccca122SBrooks Davis  */
3948ccca122SBrooks Davis static int
3958dfeba04SBrooks Davis istrsenvisx(char **mbdstp, size_t *dlen, const char *mbsrc, size_t mblength,
396778c12a6SBrooks Davis     int flags, const char *mbextra, int *cerr_ptr)
3978ccca122SBrooks Davis {
398778c12a6SBrooks Davis 	wchar_t *dst, *src, *pdst, *psrc, *start, *extra;
399778c12a6SBrooks Davis 	size_t len, olen;
400778c12a6SBrooks Davis 	uint64_t bmsk, wmsk;
401778c12a6SBrooks Davis 	wint_t c;
4028ccca122SBrooks Davis 	visfun_t f;
4038dfeba04SBrooks Davis 	int clen = 0, cerr, error = -1, i, shft;
4048dfeba04SBrooks Davis 	char *mbdst, *mdst;
405778c12a6SBrooks Davis 	ssize_t mbslength, maxolen;
406f0a6d7e2SBryan Drewery 	mbstate_t mbstate;
4078ccca122SBrooks Davis 
4088dfeba04SBrooks Davis 	_DIAGASSERT(mbdstp != NULL);
409ac25e238SBrooks Davis 	_DIAGASSERT(mbsrc != NULL || mblength == 0);
410778c12a6SBrooks Davis 	_DIAGASSERT(mbextra != NULL);
411778c12a6SBrooks Davis 
412ff88ef41SBrooks Davis 	mbslength = (ssize_t)mblength;
413ff88ef41SBrooks Davis 	/*
414ff88ef41SBrooks Davis 	 * When inputing a single character, must also read in the
415ff88ef41SBrooks Davis 	 * next character for nextc, the look-ahead character.
416ff88ef41SBrooks Davis 	 */
417ff88ef41SBrooks Davis 	if (mbslength == 1)
418ff88ef41SBrooks Davis 		mbslength++;
419ff88ef41SBrooks Davis 
420778c12a6SBrooks Davis 	/*
421778c12a6SBrooks Davis 	 * Input (mbsrc) is a char string considered to be multibyte
422778c12a6SBrooks Davis 	 * characters.  The input loop will read this string pulling
423778c12a6SBrooks Davis 	 * one character, possibly multiple bytes, from mbsrc and
424778c12a6SBrooks Davis 	 * converting each to wchar_t in src.
425778c12a6SBrooks Davis 	 *
426778c12a6SBrooks Davis 	 * The vis conversion will be done using the wide char
427778c12a6SBrooks Davis 	 * wchar_t string.
428778c12a6SBrooks Davis 	 *
429778c12a6SBrooks Davis 	 * This will then be converted back to a multibyte string to
430778c12a6SBrooks Davis 	 * return to the caller.
431778c12a6SBrooks Davis 	 */
432778c12a6SBrooks Davis 
433778c12a6SBrooks Davis 	/* Allocate space for the wide char strings */
434778c12a6SBrooks Davis 	psrc = pdst = extra = NULL;
4358dfeba04SBrooks Davis 	mdst = NULL;
436ff88ef41SBrooks Davis 	if ((psrc = calloc(mbslength + 1, sizeof(*psrc))) == NULL)
4378ccca122SBrooks Davis 		return -1;
438ff88ef41SBrooks Davis 	if ((pdst = calloc((16 * mbslength) + 1, sizeof(*pdst))) == NULL)
439778c12a6SBrooks Davis 		goto out;
4408dfeba04SBrooks Davis 	if (*mbdstp == NULL) {
441ff88ef41SBrooks Davis 		if ((mdst = calloc((16 * mbslength) + 1, sizeof(*mdst))) == NULL)
4428dfeba04SBrooks Davis 			goto out;
4438dfeba04SBrooks Davis 		*mbdstp = mdst;
4448dfeba04SBrooks Davis 	}
4458dfeba04SBrooks Davis 
4468dfeba04SBrooks Davis 	mbdst = *mbdstp;
447778c12a6SBrooks Davis 	dst = pdst;
448778c12a6SBrooks Davis 	src = psrc;
449778c12a6SBrooks Davis 
4508dfeba04SBrooks Davis 	if (flags & VIS_NOLOCALE) {
4518dfeba04SBrooks Davis 		/* Do one byte at a time conversion */
4528dfeba04SBrooks Davis 		cerr = 1;
4538dfeba04SBrooks Davis 	} else {
454778c12a6SBrooks Davis 		/* Use caller's multibyte conversion error flag. */
4558dfeba04SBrooks Davis 		cerr = cerr_ptr ? *cerr_ptr : 0;
4568dfeba04SBrooks Davis 	}
457778c12a6SBrooks Davis 
458778c12a6SBrooks Davis 	/*
459778c12a6SBrooks Davis 	 * Input loop.
460778c12a6SBrooks Davis 	 * Handle up to mblength characters (not bytes).  We do not
461778c12a6SBrooks Davis 	 * stop at NULs because we may be processing a block of data
462778c12a6SBrooks Davis 	 * that includes NULs.
463778c12a6SBrooks Davis 	 */
464f0a6d7e2SBryan Drewery 	bzero(&mbstate, sizeof(mbstate));
465778c12a6SBrooks Davis 	while (mbslength > 0) {
466778c12a6SBrooks Davis 		/* Convert one multibyte character to wchar_t. */
467778c12a6SBrooks Davis 		if (!cerr)
468*1a2f06d0SAlex Richardson 			clen = mbrtowc(src, mbsrc, MIN(mbslength, MB_LEN_MAX),
469*1a2f06d0SAlex Richardson 			    &mbstate);
470778c12a6SBrooks Davis 		if (cerr || clen < 0) {
471778c12a6SBrooks Davis 			/* Conversion error, process as a byte instead. */
472778c12a6SBrooks Davis 			*src = (wint_t)(u_char)*mbsrc;
473778c12a6SBrooks Davis 			clen = 1;
474778c12a6SBrooks Davis 			cerr = 1;
4758ccca122SBrooks Davis 		}
476ff88ef41SBrooks Davis 		if (clen == 0) {
477778c12a6SBrooks Davis 			/*
478778c12a6SBrooks Davis 			 * NUL in input gives 0 return value. process
479778c12a6SBrooks Davis 			 * as single NUL byte and keep going.
480778c12a6SBrooks Davis 			 */
481778c12a6SBrooks Davis 			clen = 1;
482ff88ef41SBrooks Davis 		}
483778c12a6SBrooks Davis 		/* Advance buffer character pointer. */
484778c12a6SBrooks Davis 		src++;
485778c12a6SBrooks Davis 		/* Advance input pointer by number of bytes read. */
486778c12a6SBrooks Davis 		mbsrc += clen;
487778c12a6SBrooks Davis 		/* Decrement input byte count. */
488778c12a6SBrooks Davis 		mbslength -= clen;
4898ccca122SBrooks Davis 	}
490778c12a6SBrooks Davis 	len = src - psrc;
491778c12a6SBrooks Davis 	src = psrc;
492ff88ef41SBrooks Davis 
493778c12a6SBrooks Davis 	/*
494778c12a6SBrooks Davis 	 * In the single character input case, we will have actually
495778c12a6SBrooks Davis 	 * processed two characters, c and nextc.  Reset len back to
496778c12a6SBrooks Davis 	 * just a single character.
497778c12a6SBrooks Davis 	 */
498778c12a6SBrooks Davis 	if (mblength < len)
499778c12a6SBrooks Davis 		len = mblength;
500778c12a6SBrooks Davis 
501778c12a6SBrooks Davis 	/* Convert extra argument to list of characters for this mode. */
502778c12a6SBrooks Davis 	extra = makeextralist(flags, mbextra);
503778c12a6SBrooks Davis 	if (!extra) {
5048ccca122SBrooks Davis 		if (dlen && *dlen == 0) {
5058ccca122SBrooks Davis 			errno = ENOSPC;
506778c12a6SBrooks Davis 			goto out;
5078ccca122SBrooks Davis 		}
508778c12a6SBrooks Davis 		*mbdst = '\0';	/* can't create extra, return "" */
509778c12a6SBrooks Davis 		error = 0;
510778c12a6SBrooks Davis 		goto out;
5118ccca122SBrooks Davis 	}
5128ccca122SBrooks Davis 
513778c12a6SBrooks Davis 	/* Look up which processing function to call. */
514778c12a6SBrooks Davis 	f = getvisfun(flags);
5158ccca122SBrooks Davis 
516778c12a6SBrooks Davis 	/*
517778c12a6SBrooks Davis 	 * Main processing loop.
518778c12a6SBrooks Davis 	 * Call do_Xvis processing function one character at a time
519778c12a6SBrooks Davis 	 * with next character available for look-ahead.
520778c12a6SBrooks Davis 	 */
5218ccca122SBrooks Davis 	for (start = dst; len > 0; len--) {
5228ccca122SBrooks Davis 		c = *src++;
523778c12a6SBrooks Davis 		dst = (*f)(dst, c, flags, len >= 1 ? *src : L'\0', extra);
5248ccca122SBrooks Davis 		if (dst == NULL) {
5258ccca122SBrooks Davis 			errno = ENOSPC;
526778c12a6SBrooks Davis 			goto out;
5278ccca122SBrooks Davis 		}
5288ccca122SBrooks Davis 	}
529778c12a6SBrooks Davis 
530778c12a6SBrooks Davis 	/* Terminate the string in the buffer. */
531778c12a6SBrooks Davis 	*dst = L'\0';
532778c12a6SBrooks Davis 
533778c12a6SBrooks Davis 	/*
534778c12a6SBrooks Davis 	 * Output loop.
535778c12a6SBrooks Davis 	 * Convert wchar_t string back to multibyte output string.
536778c12a6SBrooks Davis 	 * If we have hit a multi-byte conversion error on input,
537778c12a6SBrooks Davis 	 * output byte-by-byte here.  Else use wctomb().
538778c12a6SBrooks Davis 	 */
539778c12a6SBrooks Davis 	len = wcslen(start);
540778c12a6SBrooks Davis 	maxolen = dlen ? *dlen : (wcslen(start) * MB_LEN_MAX + 1);
541778c12a6SBrooks Davis 	olen = 0;
542f0a6d7e2SBryan Drewery 	bzero(&mbstate, sizeof(mbstate));
543778c12a6SBrooks Davis 	for (dst = start; len > 0; len--) {
544778c12a6SBrooks Davis 		if (!cerr)
545f0a6d7e2SBryan Drewery 			clen = wcrtomb(mbdst, *dst, &mbstate);
546778c12a6SBrooks Davis 		if (cerr || clen < 0) {
547778c12a6SBrooks Davis 			/*
548778c12a6SBrooks Davis 			 * Conversion error, process as a byte(s) instead.
549778c12a6SBrooks Davis 			 * Examine each byte and higher-order bytes for
550778c12a6SBrooks Davis 			 * data.  E.g.,
551778c12a6SBrooks Davis 			 *	0x000000000000a264 -> a2 64
552778c12a6SBrooks Davis 			 *	0x000000001f00a264 -> 1f 00 a2 64
553778c12a6SBrooks Davis 			 */
554778c12a6SBrooks Davis 			clen = 0;
555778c12a6SBrooks Davis 			wmsk = 0;
556778c12a6SBrooks Davis 			for (i = sizeof(wmsk) - 1; i >= 0; i--) {
557778c12a6SBrooks Davis 				shft = i * NBBY;
558778c12a6SBrooks Davis 				bmsk = (uint64_t)0xffLL << shft;
559778c12a6SBrooks Davis 				wmsk |= bmsk;
560778c12a6SBrooks Davis 				if ((*dst & wmsk) || i == 0)
561778c12a6SBrooks Davis 					mbdst[clen++] = (char)(
562778c12a6SBrooks Davis 					    (uint64_t)(*dst & bmsk) >>
563778c12a6SBrooks Davis 					    shft);
5648ccca122SBrooks Davis 			}
565778c12a6SBrooks Davis 			cerr = 1;
566778c12a6SBrooks Davis 		}
567778c12a6SBrooks Davis 		/* If this character would exceed our output limit, stop. */
568778c12a6SBrooks Davis 		if (olen + clen > (size_t)maxolen)
569778c12a6SBrooks Davis 			break;
570778c12a6SBrooks Davis 		/* Advance output pointer by number of bytes written. */
571778c12a6SBrooks Davis 		mbdst += clen;
572778c12a6SBrooks Davis 		/* Advance buffer character pointer. */
573778c12a6SBrooks Davis 		dst++;
574778c12a6SBrooks Davis 		/* Incrment output character count. */
575778c12a6SBrooks Davis 		olen += clen;
576778c12a6SBrooks Davis 	}
577778c12a6SBrooks Davis 
578778c12a6SBrooks Davis 	/* Terminate the output string. */
579778c12a6SBrooks Davis 	*mbdst = '\0';
580778c12a6SBrooks Davis 
5818dfeba04SBrooks Davis 	if (flags & VIS_NOLOCALE) {
582778c12a6SBrooks Davis 		/* Pass conversion error flag out. */
583778c12a6SBrooks Davis 		if (cerr_ptr)
584778c12a6SBrooks Davis 			*cerr_ptr = cerr;
5858dfeba04SBrooks Davis 	}
586778c12a6SBrooks Davis 
587778c12a6SBrooks Davis 	free(extra);
588778c12a6SBrooks Davis 	free(pdst);
589778c12a6SBrooks Davis 	free(psrc);
590778c12a6SBrooks Davis 
591778c12a6SBrooks Davis 	return (int)olen;
592778c12a6SBrooks Davis out:
593778c12a6SBrooks Davis 	free(extra);
594778c12a6SBrooks Davis 	free(pdst);
595778c12a6SBrooks Davis 	free(psrc);
5968dfeba04SBrooks Davis 	free(mdst);
597778c12a6SBrooks Davis 	return error;
598778c12a6SBrooks Davis }
599ac25e238SBrooks Davis 
600ac25e238SBrooks Davis static int
6018dfeba04SBrooks Davis istrsenvisxl(char **mbdstp, size_t *dlen, const char *mbsrc,
602ac25e238SBrooks Davis     int flags, const char *mbextra, int *cerr_ptr)
603ac25e238SBrooks Davis {
6048dfeba04SBrooks Davis 	return istrsenvisx(mbdstp, dlen, mbsrc,
605ac25e238SBrooks Davis 	    mbsrc != NULL ? strlen(mbsrc) : 0, flags, mbextra, cerr_ptr);
606ac25e238SBrooks Davis }
607ac25e238SBrooks Davis 
608778c12a6SBrooks Davis #endif
609778c12a6SBrooks Davis 
610778c12a6SBrooks Davis #if !HAVE_SVIS
611778c12a6SBrooks Davis /*
612778c12a6SBrooks Davis  *	The "svis" variants all take an "extra" arg that is a pointer
613778c12a6SBrooks Davis  *	to a NUL-terminated list of characters to be encoded, too.
614778c12a6SBrooks Davis  *	These functions are useful e. g. to encode strings in such a
615778c12a6SBrooks Davis  *	way so that they are not interpreted by a shell.
616778c12a6SBrooks Davis  */
617778c12a6SBrooks Davis 
618778c12a6SBrooks Davis char *
619778c12a6SBrooks Davis svis(char *mbdst, int c, int flags, int nextc, const char *mbextra)
620778c12a6SBrooks Davis {
621778c12a6SBrooks Davis 	char cc[2];
622778c12a6SBrooks Davis 	int ret;
623778c12a6SBrooks Davis 
624778c12a6SBrooks Davis 	cc[0] = c;
625778c12a6SBrooks Davis 	cc[1] = nextc;
626778c12a6SBrooks Davis 
6278dfeba04SBrooks Davis 	ret = istrsenvisx(&mbdst, NULL, cc, 1, flags, mbextra, NULL);
628778c12a6SBrooks Davis 	if (ret < 0)
629778c12a6SBrooks Davis 		return NULL;
630778c12a6SBrooks Davis 	return mbdst + ret;
631778c12a6SBrooks Davis }
632778c12a6SBrooks Davis 
633778c12a6SBrooks Davis char *
634778c12a6SBrooks Davis snvis(char *mbdst, size_t dlen, int c, int flags, int nextc, const char *mbextra)
635778c12a6SBrooks Davis {
636778c12a6SBrooks Davis 	char cc[2];
637778c12a6SBrooks Davis 	int ret;
638778c12a6SBrooks Davis 
639778c12a6SBrooks Davis 	cc[0] = c;
640778c12a6SBrooks Davis 	cc[1] = nextc;
641778c12a6SBrooks Davis 
6428dfeba04SBrooks Davis 	ret = istrsenvisx(&mbdst, &dlen, cc, 1, flags, mbextra, NULL);
643778c12a6SBrooks Davis 	if (ret < 0)
644778c12a6SBrooks Davis 		return NULL;
645778c12a6SBrooks Davis 	return mbdst + ret;
6468ccca122SBrooks Davis }
6478ccca122SBrooks Davis 
6488ccca122SBrooks Davis int
649778c12a6SBrooks Davis strsvis(char *mbdst, const char *mbsrc, int flags, const char *mbextra)
6508ccca122SBrooks Davis {
6518dfeba04SBrooks Davis 	return istrsenvisxl(&mbdst, NULL, mbsrc, flags, mbextra, NULL);
6528ccca122SBrooks Davis }
6538ccca122SBrooks Davis 
6548ccca122SBrooks Davis int
655778c12a6SBrooks Davis strsnvis(char *mbdst, size_t dlen, const char *mbsrc, int flags, const char *mbextra)
6568ccca122SBrooks Davis {
6578dfeba04SBrooks Davis 	return istrsenvisxl(&mbdst, &dlen, mbsrc, flags, mbextra, NULL);
658778c12a6SBrooks Davis }
659778c12a6SBrooks Davis 
660778c12a6SBrooks Davis int
661778c12a6SBrooks Davis strsvisx(char *mbdst, const char *mbsrc, size_t len, int flags, const char *mbextra)
662778c12a6SBrooks Davis {
6638dfeba04SBrooks Davis 	return istrsenvisx(&mbdst, NULL, mbsrc, len, flags, mbextra, NULL);
664778c12a6SBrooks Davis }
665778c12a6SBrooks Davis 
666778c12a6SBrooks Davis int
667778c12a6SBrooks Davis strsnvisx(char *mbdst, size_t dlen, const char *mbsrc, size_t len, int flags,
668778c12a6SBrooks Davis     const char *mbextra)
669778c12a6SBrooks Davis {
6708dfeba04SBrooks Davis 	return istrsenvisx(&mbdst, &dlen, mbsrc, len, flags, mbextra, NULL);
671778c12a6SBrooks Davis }
672778c12a6SBrooks Davis 
673778c12a6SBrooks Davis int
674778c12a6SBrooks Davis strsenvisx(char *mbdst, size_t dlen, const char *mbsrc, size_t len, int flags,
675778c12a6SBrooks Davis     const char *mbextra, int *cerr_ptr)
676778c12a6SBrooks Davis {
6778dfeba04SBrooks Davis 	return istrsenvisx(&mbdst, &dlen, mbsrc, len, flags, mbextra, cerr_ptr);
6788ccca122SBrooks Davis }
6798ccca122SBrooks Davis #endif
6808ccca122SBrooks Davis 
6818ccca122SBrooks Davis #if !HAVE_VIS
6828ccca122SBrooks Davis /*
6838ccca122SBrooks Davis  * vis - visually encode characters
6848ccca122SBrooks Davis  */
685778c12a6SBrooks Davis char *
686778c12a6SBrooks Davis vis(char *mbdst, int c, int flags, int nextc)
6878ccca122SBrooks Davis {
688778c12a6SBrooks Davis 	char cc[2];
689778c12a6SBrooks Davis 	int ret;
6908ccca122SBrooks Davis 
691778c12a6SBrooks Davis 	cc[0] = c;
692778c12a6SBrooks Davis 	cc[1] = nextc;
6938ccca122SBrooks Davis 
6948dfeba04SBrooks Davis 	ret = istrsenvisx(&mbdst, NULL, cc, 1, flags, "", NULL);
695778c12a6SBrooks Davis 	if (ret < 0)
6968ccca122SBrooks Davis 		return NULL;
697778c12a6SBrooks Davis 	return mbdst + ret;
6988ccca122SBrooks Davis }
6998ccca122SBrooks Davis 
7008ccca122SBrooks Davis char *
701778c12a6SBrooks Davis nvis(char *mbdst, size_t dlen, int c, int flags, int nextc)
7028ccca122SBrooks Davis {
703778c12a6SBrooks Davis 	char cc[2];
704778c12a6SBrooks Davis 	int ret;
7058ccca122SBrooks Davis 
706778c12a6SBrooks Davis 	cc[0] = c;
707778c12a6SBrooks Davis 	cc[1] = nextc;
7088ccca122SBrooks Davis 
7098dfeba04SBrooks Davis 	ret = istrsenvisx(&mbdst, &dlen, cc, 1, flags, "", NULL);
710778c12a6SBrooks Davis 	if (ret < 0)
711778c12a6SBrooks Davis 		return NULL;
712778c12a6SBrooks Davis 	return mbdst + ret;
713778c12a6SBrooks Davis }
7148ccca122SBrooks Davis 
7158ccca122SBrooks Davis /*
716778c12a6SBrooks Davis  * strvis - visually encode characters from src into dst
717778c12a6SBrooks Davis  *
718778c12a6SBrooks Davis  *	Dst must be 4 times the size of src to account for possible
719778c12a6SBrooks Davis  *	expansion.  The length of dst, not including the trailing NULL,
720778c12a6SBrooks Davis  *	is returned.
721778c12a6SBrooks Davis  */
722778c12a6SBrooks Davis 
723778c12a6SBrooks Davis int
724778c12a6SBrooks Davis strvis(char *mbdst, const char *mbsrc, int flags)
725778c12a6SBrooks Davis {
7268dfeba04SBrooks Davis 	return istrsenvisxl(&mbdst, NULL, mbsrc, flags, "", NULL);
727778c12a6SBrooks Davis }
728778c12a6SBrooks Davis 
729778c12a6SBrooks Davis int
730778c12a6SBrooks Davis strnvis(char *mbdst, size_t dlen, const char *mbsrc, int flags)
731778c12a6SBrooks Davis {
7328dfeba04SBrooks Davis 	return istrsenvisxl(&mbdst, &dlen, mbsrc, flags, "", NULL);
7338dfeba04SBrooks Davis }
7348dfeba04SBrooks Davis 
7358dfeba04SBrooks Davis int
7368dfeba04SBrooks Davis stravis(char **mbdstp, const char *mbsrc, int flags)
7378dfeba04SBrooks Davis {
7388dfeba04SBrooks Davis 	*mbdstp = NULL;
7398dfeba04SBrooks Davis 	return istrsenvisxl(mbdstp, NULL, mbsrc, flags, "", NULL);
740778c12a6SBrooks Davis }
741778c12a6SBrooks Davis 
742778c12a6SBrooks Davis /*
743778c12a6SBrooks Davis  * strvisx - visually encode characters from src into dst
7448ccca122SBrooks Davis  *
7458ccca122SBrooks Davis  *	Dst must be 4 times the size of src to account for possible
7468ccca122SBrooks Davis  *	expansion.  The length of dst, not including the trailing NULL,
7478ccca122SBrooks Davis  *	is returned.
7488ccca122SBrooks Davis  *
749778c12a6SBrooks Davis  *	Strvisx encodes exactly len characters from src into dst.
7508ccca122SBrooks Davis  *	This is useful for encoding a block of data.
7518ccca122SBrooks Davis  */
7528ccca122SBrooks Davis 
753778c12a6SBrooks Davis int
754778c12a6SBrooks Davis strvisx(char *mbdst, const char *mbsrc, size_t len, int flags)
755778c12a6SBrooks Davis {
7568dfeba04SBrooks Davis 	return istrsenvisx(&mbdst, NULL, mbsrc, len, flags, "", NULL);
7578ccca122SBrooks Davis }
7588ccca122SBrooks Davis 
7598ccca122SBrooks Davis int
760778c12a6SBrooks Davis strnvisx(char *mbdst, size_t dlen, const char *mbsrc, size_t len, int flags)
7618ccca122SBrooks Davis {
7628dfeba04SBrooks Davis 	return istrsenvisx(&mbdst, &dlen, mbsrc, len, flags, "", NULL);
7638ccca122SBrooks Davis }
7648ccca122SBrooks Davis 
7658ccca122SBrooks Davis int
766778c12a6SBrooks Davis strenvisx(char *mbdst, size_t dlen, const char *mbsrc, size_t len, int flags,
767778c12a6SBrooks Davis     int *cerr_ptr)
7688ccca122SBrooks Davis {
7698dfeba04SBrooks Davis 	return istrsenvisx(&mbdst, &dlen, mbsrc, len, flags, "", cerr_ptr);
7708ccca122SBrooks Davis }
7718ccca122SBrooks Davis #endif
772