xref: /freebsd/contrib/libc-vis/vis.c (revision 28f6c2f292806bf31230a959bc4b19d7081669a7)
1 /*	$NetBSD: vis.c,v 1.74 2017/11/27 16:37:21 christos Exp $	*/
2 
3 /*-
4  * Copyright (c) 1989, 1993
5  *	The Regents of the University of California.  All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  * 3. Neither the name of the University nor the names of its contributors
16  *    may be used to endorse or promote products derived from this software
17  *    without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29  * SUCH DAMAGE.
30  */
31 
32 /*-
33  * Copyright (c) 1999, 2005 The NetBSD Foundation, Inc.
34  * All rights reserved.
35  *
36  * Redistribution and use in source and binary forms, with or without
37  * modification, are permitted provided that the following conditions
38  * are met:
39  * 1. Redistributions of source code must retain the above copyright
40  *    notice, this list of conditions and the following disclaimer.
41  * 2. Redistributions in binary form must reproduce the above copyright
42  *    notice, this list of conditions and the following disclaimer in the
43  *    documentation and/or other materials provided with the distribution.
44  *
45  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
46  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
47  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
48  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
49  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
50  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
51  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
52  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
53  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
54  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
55  * POSSIBILITY OF SUCH DAMAGE.
56  */
57 
58 #include <sys/cdefs.h>
59 #if defined(LIBC_SCCS) && !defined(lint)
60 __RCSID("$NetBSD: vis.c,v 1.74 2017/11/27 16:37:21 christos Exp $");
61 #endif /* LIBC_SCCS and not lint */
62 #ifdef __FBSDID
63 __FBSDID("$FreeBSD$");
64 #define	_DIAGASSERT(x)	assert(x)
65 #endif
66 
67 #include "namespace.h"
68 #include <sys/types.h>
69 #include <sys/param.h>
70 
71 #include <assert.h>
72 #include <vis.h>
73 #include <errno.h>
74 #include <stdlib.h>
75 #include <wchar.h>
76 #include <wctype.h>
77 
78 #ifdef __weak_alias
79 __weak_alias(strvisx,_strvisx)
80 #endif
81 
82 #if !HAVE_VIS || !HAVE_SVIS
83 #include <ctype.h>
84 #include <limits.h>
85 #include <stdio.h>
86 #include <string.h>
87 
88 /*
89  * The reason for going through the trouble to deal with character encodings
90  * in vis(3), is that we use this to safe encode output of commands. This
91  * safe encoding varies depending on the character set. For example if we
92  * display ps output in French, we don't want to display French characters
93  * as M-foo.
94  */
95 
96 static wchar_t *do_svis(wchar_t *, wint_t, int, wint_t, const wchar_t *);
97 
98 #undef BELL
99 #define BELL L'\a'
100 
101 #if defined(LC_C_LOCALE)
102 #define iscgraph(c)      isgraph_l(c, LC_C_LOCALE)
103 #else
104 /* Keep it simple for now, no locale stuff */
105 #define iscgraph(c)	isgraph(c)
106 #ifdef notyet
107 #include <locale.h>
108 static int
109 iscgraph(int c) {
110 	int rv;
111 	char *ol;
112 
113 	ol = setlocale(LC_CTYPE, "C");
114 	rv = isgraph(c);
115 	if (ol)
116 		setlocale(LC_CTYPE, ol);
117 	return rv;
118 }
119 #endif
120 #endif
121 
122 #define ISGRAPH(flags, c) \
123     (((flags) & VIS_NOLOCALE) ? iscgraph(c) : iswgraph(c))
124 
125 #define iswoctal(c)	(((u_char)(c)) >= L'0' && ((u_char)(c)) <= L'7')
126 #define iswwhite(c)	(c == L' ' || c == L'\t' || c == L'\n')
127 #define iswsafe(c)	(c == L'\b' || c == BELL || c == L'\r')
128 #define xtoa(c)		L"0123456789abcdef"[c]
129 #define XTOA(c)		L"0123456789ABCDEF"[c]
130 
131 #define MAXEXTRAS	30
132 
133 static const wchar_t char_shell[] = L"'`\";&<>()|{}]\\$!^~";
134 static const wchar_t char_glob[] = L"*?[#";
135 
136 #if !HAVE_NBTOOL_CONFIG_H
137 #ifndef __NetBSD__
138 /*
139  * On NetBSD MB_LEN_MAX is currently 32 which does not fit on any integer
140  * integral type and it is probably wrong, since currently the maximum
141  * number of bytes and character needs is 6. Until this is fixed, the
142  * loops below are using sizeof(uint64_t) - 1 instead of MB_LEN_MAX, and
143  * the assertion is commented out.
144  */
145 #ifdef __FreeBSD__
146 /*
147  * On FreeBSD including <sys/systm.h> for CTASSERT only works in kernel
148  * mode.
149  */
150 #ifndef CTASSERT
151 #define CTASSERT(x)             _CTASSERT(x, __LINE__)
152 #define _CTASSERT(x, y)         __CTASSERT(x, y)
153 #define __CTASSERT(x, y)        typedef char __assert ## y[(x) ? 1 : -1]
154 #endif
155 #endif /* __FreeBSD__ */
156 CTASSERT(MB_LEN_MAX <= sizeof(uint64_t));
157 #endif /* !__NetBSD__ */
158 #endif
159 
160 /*
161  * This is do_hvis, for HTTP style (RFC 1808)
162  */
163 static wchar_t *
164 do_hvis(wchar_t *dst, wint_t c, int flags, wint_t nextc, const wchar_t *extra)
165 {
166 	if (iswalnum(c)
167 	    /* safe */
168 	    || c == L'$' || c == L'-' || c == L'_' || c == L'.' || c == L'+'
169 	    /* extra */
170 	    || c == L'!' || c == L'*' || c == L'\'' || c == L'(' || c == L')'
171 	    || c == L',')
172 		dst = do_svis(dst, c, flags, nextc, extra);
173 	else {
174 		*dst++ = L'%';
175 		*dst++ = xtoa(((unsigned int)c >> 4) & 0xf);
176 		*dst++ = xtoa((unsigned int)c & 0xf);
177 	}
178 
179 	return dst;
180 }
181 
182 /*
183  * This is do_mvis, for Quoted-Printable MIME (RFC 2045)
184  * NB: No handling of long lines or CRLF.
185  */
186 static wchar_t *
187 do_mvis(wchar_t *dst, wint_t c, int flags, wint_t nextc, const wchar_t *extra)
188 {
189 	if ((c != L'\n') &&
190 	    /* Space at the end of the line */
191 	    ((iswspace(c) && (nextc == L'\r' || nextc == L'\n')) ||
192 	    /* Out of range */
193 	    (!iswspace(c) && (c < 33 || (c > 60 && c < 62) || c > 126)) ||
194 	    /* Specific char to be escaped */
195 	    wcschr(L"#$@[\\]^`{|}~", c) != NULL)) {
196 		*dst++ = L'=';
197 		*dst++ = XTOA(((unsigned int)c >> 4) & 0xf);
198 		*dst++ = XTOA((unsigned int)c & 0xf);
199 	} else
200 		dst = do_svis(dst, c, flags, nextc, extra);
201 	return dst;
202 }
203 
204 /*
205  * Output single byte of multibyte character.
206  */
207 static wchar_t *
208 do_mbyte(wchar_t *dst, wint_t c, int flags, wint_t nextc, int iswextra)
209 {
210 	if (flags & VIS_CSTYLE) {
211 		switch (c) {
212 		case L'\n':
213 			*dst++ = L'\\'; *dst++ = L'n';
214 			return dst;
215 		case L'\r':
216 			*dst++ = L'\\'; *dst++ = L'r';
217 			return dst;
218 		case L'\b':
219 			*dst++ = L'\\'; *dst++ = L'b';
220 			return dst;
221 		case BELL:
222 			*dst++ = L'\\'; *dst++ = L'a';
223 			return dst;
224 		case L'\v':
225 			*dst++ = L'\\'; *dst++ = L'v';
226 			return dst;
227 		case L'\t':
228 			*dst++ = L'\\'; *dst++ = L't';
229 			return dst;
230 		case L'\f':
231 			*dst++ = L'\\'; *dst++ = L'f';
232 			return dst;
233 		case L' ':
234 			*dst++ = L'\\'; *dst++ = L's';
235 			return dst;
236 		case L'\0':
237 			*dst++ = L'\\'; *dst++ = L'0';
238 			if (iswoctal(nextc)) {
239 				*dst++ = L'0';
240 				*dst++ = L'0';
241 			}
242 			return dst;
243 		/* We cannot encode these characters in VIS_CSTYLE
244 		 * because they special meaning */
245 		case L'n':
246 		case L'r':
247 		case L'b':
248 		case L'a':
249 		case L'v':
250 		case L't':
251 		case L'f':
252 		case L's':
253 		case L'0':
254 		case L'M':
255 		case L'^':
256 		case L'$': /* vis(1) -l */
257 			break;
258 		default:
259 			if (ISGRAPH(flags, c) && !iswoctal(c)) {
260 				*dst++ = L'\\';
261 				*dst++ = c;
262 				return dst;
263 			}
264 		}
265 	}
266 	if (iswextra || ((c & 0177) == L' ') || (flags & VIS_OCTAL)) {
267 		*dst++ = L'\\';
268 		*dst++ = (u_char)(((u_int32_t)(u_char)c >> 6) & 03) + L'0';
269 		*dst++ = (u_char)(((u_int32_t)(u_char)c >> 3) & 07) + L'0';
270 		*dst++ =			     (c	      & 07) + L'0';
271 	} else {
272 		if ((flags & VIS_NOSLASH) == 0)
273 			*dst++ = L'\\';
274 
275 		if (c & 0200) {
276 			c &= 0177;
277 			*dst++ = L'M';
278 		}
279 
280 		if (iswcntrl(c)) {
281 			*dst++ = L'^';
282 			if (c == 0177)
283 				*dst++ = L'?';
284 			else
285 				*dst++ = c + L'@';
286 		} else {
287 			*dst++ = L'-';
288 			*dst++ = c;
289 		}
290 	}
291 
292 	return dst;
293 }
294 
295 /*
296  * This is do_vis, the central code of vis.
297  * dst:	      Pointer to the destination buffer
298  * c:	      Character to encode
299  * flags:     Flags word
300  * nextc:     The character following 'c'
301  * extra:     Pointer to the list of extra characters to be
302  *	      backslash-protected.
303  */
304 static wchar_t *
305 do_svis(wchar_t *dst, wint_t c, int flags, wint_t nextc, const wchar_t *extra)
306 {
307 	int iswextra, i, shft;
308 	uint64_t bmsk, wmsk;
309 
310 	iswextra = wcschr(extra, c) != NULL;
311 	if (!iswextra && (ISGRAPH(flags, c) || iswwhite(c) ||
312 	    ((flags & VIS_SAFE) && iswsafe(c)))) {
313 		*dst++ = c;
314 		return dst;
315 	}
316 
317 	/* See comment in istrsenvisx() output loop, below. */
318 	wmsk = 0;
319 	for (i = sizeof(wmsk) - 1; i >= 0; i--) {
320 		shft = i * NBBY;
321 		bmsk = (uint64_t)0xffLL << shft;
322 		wmsk |= bmsk;
323 		if ((c & wmsk) || i == 0)
324 			dst = do_mbyte(dst, (wint_t)(
325 			    (uint64_t)(c & bmsk) >> shft),
326 			    flags, nextc, iswextra);
327 	}
328 
329 	return dst;
330 }
331 
332 typedef wchar_t *(*visfun_t)(wchar_t *, wint_t, int, wint_t, const wchar_t *);
333 
334 /*
335  * Return the appropriate encoding function depending on the flags given.
336  */
337 static visfun_t
338 getvisfun(int flags)
339 {
340 	if (flags & VIS_HTTPSTYLE)
341 		return do_hvis;
342 	if (flags & VIS_MIMESTYLE)
343 		return do_mvis;
344 	return do_svis;
345 }
346 
347 /*
348  * Expand list of extra characters to not visually encode.
349  */
350 static wchar_t *
351 makeextralist(int flags, const char *src)
352 {
353 	wchar_t *dst, *d;
354 	size_t len;
355 	const wchar_t *s;
356 	mbstate_t mbstate;
357 
358 	bzero(&mbstate, sizeof(mbstate));
359 	len = strlen(src);
360 	if ((dst = calloc(len + MAXEXTRAS, sizeof(*dst))) == NULL)
361 		return NULL;
362 
363 	if ((flags & VIS_NOLOCALE) || mbsrtowcs(dst, &src, len, &mbstate) == (size_t)-1) {
364 		size_t i;
365 		for (i = 0; i < len; i++)
366 			dst[i] = (wchar_t)(u_char)src[i];
367 		d = dst + len;
368 	} else
369 		d = dst + wcslen(dst);
370 
371 	if (flags & VIS_GLOB)
372 		for (s = char_glob; *s; *d++ = *s++)
373 			continue;
374 
375 	if (flags & VIS_SHELL)
376 		for (s = char_shell; *s; *d++ = *s++)
377 			continue;
378 
379 	if (flags & VIS_SP) *d++ = L' ';
380 	if (flags & VIS_TAB) *d++ = L'\t';
381 	if (flags & VIS_NL) *d++ = L'\n';
382 	if (flags & VIS_DQ) *d++ = L'"';
383 	if ((flags & VIS_NOSLASH) == 0) *d++ = L'\\';
384 	*d = L'\0';
385 
386 	return dst;
387 }
388 
389 /*
390  * istrsenvisx()
391  * 	The main internal function.
392  *	All user-visible functions call this one.
393  */
394 static int
395 istrsenvisx(char **mbdstp, size_t *dlen, const char *mbsrc, size_t mblength,
396     int flags, const char *mbextra, int *cerr_ptr)
397 {
398 	char mbbuf[MB_CUR_MAX];
399 	wchar_t *dst, *src, *pdst, *psrc, *start, *extra;
400 	size_t len, olen;
401 	uint64_t bmsk, wmsk;
402 	wint_t c;
403 	visfun_t f;
404 	int clen = 0, cerr, error = -1, i, shft;
405 	char *mbdst, *mbwrite, *mdst;
406 	ssize_t mbslength;
407 	size_t maxolen;
408 	mbstate_t mbstate;
409 
410 	_DIAGASSERT(mbdstp != NULL);
411 	_DIAGASSERT(mbsrc != NULL || mblength == 0);
412 	_DIAGASSERT(mbextra != NULL);
413 
414 	mbslength = (ssize_t)mblength;
415 	/*
416 	 * When inputing a single character, must also read in the
417 	 * next character for nextc, the look-ahead character.
418 	 */
419 	if (mbslength == 1)
420 		mbslength++;
421 
422 	/*
423 	 * Input (mbsrc) is a char string considered to be multibyte
424 	 * characters.  The input loop will read this string pulling
425 	 * one character, possibly multiple bytes, from mbsrc and
426 	 * converting each to wchar_t in src.
427 	 *
428 	 * The vis conversion will be done using the wide char
429 	 * wchar_t string.
430 	 *
431 	 * This will then be converted back to a multibyte string to
432 	 * return to the caller.
433 	 */
434 
435 	/* Allocate space for the wide char strings */
436 	psrc = pdst = extra = NULL;
437 	mdst = NULL;
438 	if ((psrc = calloc(mbslength + 1, sizeof(*psrc))) == NULL)
439 		return -1;
440 	if ((pdst = calloc((16 * mbslength) + 1, sizeof(*pdst))) == NULL)
441 		goto out;
442 	if (*mbdstp == NULL) {
443 		if ((mdst = calloc((16 * mbslength) + 1, sizeof(*mdst))) == NULL)
444 			goto out;
445 		*mbdstp = mdst;
446 	}
447 
448 	mbdst = *mbdstp;
449 	dst = pdst;
450 	src = psrc;
451 
452 	if (flags & VIS_NOLOCALE) {
453 		/* Do one byte at a time conversion */
454 		cerr = 1;
455 	} else {
456 		/* Use caller's multibyte conversion error flag. */
457 		cerr = cerr_ptr ? *cerr_ptr : 0;
458 	}
459 
460 	/*
461 	 * Input loop.
462 	 * Handle up to mblength characters (not bytes).  We do not
463 	 * stop at NULs because we may be processing a block of data
464 	 * that includes NULs.
465 	 */
466 	bzero(&mbstate, sizeof(mbstate));
467 	while (mbslength > 0) {
468 		/* Convert one multibyte character to wchar_t. */
469 		if (!cerr)
470 			clen = mbrtowc(src, mbsrc, MIN(mbslength, MB_LEN_MAX),
471 			    &mbstate);
472 		if (cerr || clen < 0) {
473 			/* Conversion error, process as a byte instead. */
474 			*src = (wint_t)(u_char)*mbsrc;
475 			clen = 1;
476 			cerr = 1;
477 		}
478 		if (clen == 0) {
479 			/*
480 			 * NUL in input gives 0 return value. process
481 			 * as single NUL byte and keep going.
482 			 */
483 			clen = 1;
484 		}
485 		/* Advance buffer character pointer. */
486 		src++;
487 		/* Advance input pointer by number of bytes read. */
488 		mbsrc += clen;
489 		/* Decrement input byte count. */
490 		mbslength -= clen;
491 	}
492 	len = src - psrc;
493 	src = psrc;
494 
495 	/*
496 	 * In the single character input case, we will have actually
497 	 * processed two characters, c and nextc.  Reset len back to
498 	 * just a single character.
499 	 */
500 	if (mblength < len)
501 		len = mblength;
502 
503 	/* Convert extra argument to list of characters for this mode. */
504 	extra = makeextralist(flags, mbextra);
505 	if (!extra) {
506 		if (dlen && *dlen == 0) {
507 			errno = ENOSPC;
508 			goto out;
509 		}
510 		*mbdst = '\0';	/* can't create extra, return "" */
511 		error = 0;
512 		goto out;
513 	}
514 
515 	/* Look up which processing function to call. */
516 	f = getvisfun(flags);
517 
518 	/*
519 	 * Main processing loop.
520 	 * Call do_Xvis processing function one character at a time
521 	 * with next character available for look-ahead.
522 	 */
523 	for (start = dst; len > 0; len--) {
524 		c = *src++;
525 		dst = (*f)(dst, c, flags, len >= 1 ? *src : L'\0', extra);
526 		if (dst == NULL) {
527 			errno = ENOSPC;
528 			goto out;
529 		}
530 	}
531 
532 	/* Terminate the string in the buffer. */
533 	*dst = L'\0';
534 
535 	/*
536 	 * Output loop.
537 	 * Convert wchar_t string back to multibyte output string.
538 	 * If we have hit a multi-byte conversion error on input,
539 	 * output byte-by-byte here.  Else use wctomb().
540 	 */
541 	len = wcslen(start);
542 	maxolen = dlen ? *dlen : (wcslen(start) * MB_LEN_MAX + 1);
543 	olen = 0;
544 	bzero(&mbstate, sizeof(mbstate));
545 	for (dst = start; len > 0; len--) {
546 		if (!cerr) {
547 			/*
548 			 * If we have at least MB_CUR_MAX bytes in the buffer,
549 			 * we'll just do the conversion in-place into mbdst.  We
550 			 * need to be a little more conservative when we get to
551 			 * the end of the buffer, as we may not have MB_CUR_MAX
552 			 * bytes but we may not need it.
553 			 */
554 			if (maxolen - olen > MB_CUR_MAX)
555 				mbwrite = mbdst;
556 			else
557 				mbwrite = mbbuf;
558 			clen = wcrtomb(mbwrite, *dst, &mbstate);
559 			if (clen > 0 && mbwrite != mbdst) {
560 				/*
561 				 * Don't break past our output limit, noting
562 				 * that maxolen includes the nul terminator so
563 				 * we can't write past maxolen - 1 here.
564 				 */
565 				if (olen + clen >= maxolen) {
566 					errno = ENOSPC;
567 					goto out;
568 				}
569 
570 				memcpy(mbdst, mbwrite, clen);
571 			}
572 		}
573 		if (cerr || clen < 0) {
574 			/*
575 			 * Conversion error, process as a byte(s) instead.
576 			 * Examine each byte and higher-order bytes for
577 			 * data.  E.g.,
578 			 *	0x000000000000a264 -> a2 64
579 			 *	0x000000001f00a264 -> 1f 00 a2 64
580 			 */
581 			clen = 0;
582 			wmsk = 0;
583 			for (i = sizeof(wmsk) - 1; i >= 0; i--) {
584 				shft = i * NBBY;
585 				bmsk = (uint64_t)0xffLL << shft;
586 				wmsk |= bmsk;
587 				if ((*dst & wmsk) || i == 0) {
588 					if (olen + clen + 1 >= maxolen) {
589 						errno = ENOSPC;
590 						goto out;
591 					}
592 
593 					mbdst[clen++] = (char)(
594 					    (uint64_t)(*dst & bmsk) >>
595 					    shft);
596 				}
597 			}
598 			cerr = 1;
599 		}
600 
601 		/*
602 		 * We'll be dereferencing mbdst[clen] after this to write the
603 		 * nul terminator; the above paths should have checked for a
604 		 * possible overflow already.
605 		 */
606 		assert(olen + clen < maxolen);
607 
608 		/* Advance output pointer by number of bytes written. */
609 		mbdst += clen;
610 		/* Advance buffer character pointer. */
611 		dst++;
612 		/* Incrment output character count. */
613 		olen += clen;
614 	}
615 
616 	/* Terminate the output string. */
617 	*mbdst = '\0';
618 
619 	if (flags & VIS_NOLOCALE) {
620 		/* Pass conversion error flag out. */
621 		if (cerr_ptr)
622 			*cerr_ptr = cerr;
623 	}
624 
625 	free(extra);
626 	free(pdst);
627 	free(psrc);
628 
629 	return (int)olen;
630 out:
631 	free(extra);
632 	free(pdst);
633 	free(psrc);
634 	free(mdst);
635 	return error;
636 }
637 
638 static int
639 istrsenvisxl(char **mbdstp, size_t *dlen, const char *mbsrc,
640     int flags, const char *mbextra, int *cerr_ptr)
641 {
642 	return istrsenvisx(mbdstp, dlen, mbsrc,
643 	    mbsrc != NULL ? strlen(mbsrc) : 0, flags, mbextra, cerr_ptr);
644 }
645 
646 #endif
647 
648 #if !HAVE_SVIS
649 /*
650  *	The "svis" variants all take an "extra" arg that is a pointer
651  *	to a NUL-terminated list of characters to be encoded, too.
652  *	These functions are useful e. g. to encode strings in such a
653  *	way so that they are not interpreted by a shell.
654  */
655 
656 char *
657 svis(char *mbdst, int c, int flags, int nextc, const char *mbextra)
658 {
659 	char cc[2];
660 	int ret;
661 
662 	cc[0] = c;
663 	cc[1] = nextc;
664 
665 	ret = istrsenvisx(&mbdst, NULL, cc, 1, flags, mbextra, NULL);
666 	if (ret < 0)
667 		return NULL;
668 	return mbdst + ret;
669 }
670 
671 char *
672 snvis(char *mbdst, size_t dlen, int c, int flags, int nextc, const char *mbextra)
673 {
674 	char cc[2];
675 	int ret;
676 
677 	cc[0] = c;
678 	cc[1] = nextc;
679 
680 	ret = istrsenvisx(&mbdst, &dlen, cc, 1, flags, mbextra, NULL);
681 	if (ret < 0)
682 		return NULL;
683 	return mbdst + ret;
684 }
685 
686 int
687 strsvis(char *mbdst, const char *mbsrc, int flags, const char *mbextra)
688 {
689 	return istrsenvisxl(&mbdst, NULL, mbsrc, flags, mbextra, NULL);
690 }
691 
692 int
693 strsnvis(char *mbdst, size_t dlen, const char *mbsrc, int flags, const char *mbextra)
694 {
695 	return istrsenvisxl(&mbdst, &dlen, mbsrc, flags, mbextra, NULL);
696 }
697 
698 int
699 strsvisx(char *mbdst, const char *mbsrc, size_t len, int flags, const char *mbextra)
700 {
701 	return istrsenvisx(&mbdst, NULL, mbsrc, len, flags, mbextra, NULL);
702 }
703 
704 int
705 strsnvisx(char *mbdst, size_t dlen, const char *mbsrc, size_t len, int flags,
706     const char *mbextra)
707 {
708 	return istrsenvisx(&mbdst, &dlen, mbsrc, len, flags, mbextra, NULL);
709 }
710 
711 int
712 strsenvisx(char *mbdst, size_t dlen, const char *mbsrc, size_t len, int flags,
713     const char *mbextra, int *cerr_ptr)
714 {
715 	return istrsenvisx(&mbdst, &dlen, mbsrc, len, flags, mbextra, cerr_ptr);
716 }
717 #endif
718 
719 #if !HAVE_VIS
720 /*
721  * vis - visually encode characters
722  */
723 char *
724 vis(char *mbdst, int c, int flags, int nextc)
725 {
726 	char cc[2];
727 	int ret;
728 
729 	cc[0] = c;
730 	cc[1] = nextc;
731 
732 	ret = istrsenvisx(&mbdst, NULL, cc, 1, flags, "", NULL);
733 	if (ret < 0)
734 		return NULL;
735 	return mbdst + ret;
736 }
737 
738 char *
739 nvis(char *mbdst, size_t dlen, int c, int flags, int nextc)
740 {
741 	char cc[2];
742 	int ret;
743 
744 	cc[0] = c;
745 	cc[1] = nextc;
746 
747 	ret = istrsenvisx(&mbdst, &dlen, cc, 1, flags, "", NULL);
748 	if (ret < 0)
749 		return NULL;
750 	return mbdst + ret;
751 }
752 
753 /*
754  * strvis - visually encode characters from src into dst
755  *
756  *	Dst must be 4 times the size of src to account for possible
757  *	expansion.  The length of dst, not including the trailing NULL,
758  *	is returned.
759  */
760 
761 int
762 strvis(char *mbdst, const char *mbsrc, int flags)
763 {
764 	return istrsenvisxl(&mbdst, NULL, mbsrc, flags, "", NULL);
765 }
766 
767 int
768 strnvis(char *mbdst, size_t dlen, const char *mbsrc, int flags)
769 {
770 	return istrsenvisxl(&mbdst, &dlen, mbsrc, flags, "", NULL);
771 }
772 
773 int
774 stravis(char **mbdstp, const char *mbsrc, int flags)
775 {
776 	*mbdstp = NULL;
777 	return istrsenvisxl(mbdstp, NULL, mbsrc, flags, "", NULL);
778 }
779 
780 /*
781  * strvisx - visually encode characters from src into dst
782  *
783  *	Dst must be 4 times the size of src to account for possible
784  *	expansion.  The length of dst, not including the trailing NULL,
785  *	is returned.
786  *
787  *	Strvisx encodes exactly len characters from src into dst.
788  *	This is useful for encoding a block of data.
789  */
790 
791 int
792 strvisx(char *mbdst, const char *mbsrc, size_t len, int flags)
793 {
794 	return istrsenvisx(&mbdst, NULL, mbsrc, len, flags, "", NULL);
795 }
796 
797 int
798 strnvisx(char *mbdst, size_t dlen, const char *mbsrc, size_t len, int flags)
799 {
800 	return istrsenvisx(&mbdst, &dlen, mbsrc, len, flags, "", NULL);
801 }
802 
803 int
804 strenvisx(char *mbdst, size_t dlen, const char *mbsrc, size_t len, int flags,
805     int *cerr_ptr)
806 {
807 	return istrsenvisx(&mbdst, &dlen, mbsrc, len, flags, "", cerr_ptr);
808 }
809 #endif
810