contrib/libc-vis/vis.c

   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         wchar_t *dst, *src, *pdst, *psrc, *start, *extra;
 399         size_t len, olen;
 400         uint64_t bmsk, wmsk;
 401         wint_t c;
 402         visfun_t f;
 403         int clen = 0, cerr, error = -1, i, shft;
 404         char *mbdst, *mdst;
 405         ssize_t mbslength, maxolen;
 406         mbstate_t mbstate;
 407
 408         _DIAGASSERT(mbdstp != NULL);
 409         _DIAGASSERT(mbsrc != NULL || mblength == 0);
 410         _DIAGASSERT(mbextra != NULL);
 411
 412         mbslength = (ssize_t)mblength;
 413         /*
 414          * When inputing a single character, must also read in the
 415          * next character for nextc, the look-ahead character.
 416          */
 417         if (mbslength == 1)
 418                 mbslength++;
 419
 420         /*
 421          * Input (mbsrc) is a char string considered to be multibyte
 422          * characters.  The input loop will read this string pulling
 423          * one character, possibly multiple bytes, from mbsrc and
 424          * converting each to wchar_t in src.
 425          *
 426          * The vis conversion will be done using the wide char
 427          * wchar_t string.
 428          *
 429          * This will then be converted back to a multibyte string to
 430          * return to the caller.
 431          */
 432
 433         /* Allocate space for the wide char strings */
 434         psrc = pdst = extra = NULL;
 435         mdst = NULL;
 436         if ((psrc = calloc(mbslength + 1, sizeof(*psrc))) == NULL)
 437                 return -1;
 438         if ((pdst = calloc((16 * mbslength) + 1, sizeof(*pdst))) == NULL)
 439                 goto out;
 440         if (*mbdstp == NULL) {
 441                 if ((mdst = calloc((16 * mbslength) + 1, sizeof(*mdst))) == NULL)
 442                         goto out;
 443                 *mbdstp = mdst;
 444         }
 445
 446         mbdst = *mbdstp;
 447         dst = pdst;
 448         src = psrc;
 449
 450         if (flags & VIS_NOLOCALE) {
 451                 /* Do one byte at a time conversion */
 452                 cerr = 1;
 453         } else {
 454                 /* Use caller's multibyte conversion error flag. */
 455                 cerr = cerr_ptr ? *cerr_ptr : 0;
 456         }
 457
 458         /*
 459          * Input loop.
 460          * Handle up to mblength characters (not bytes).  We do not
 461          * stop at NULs because we may be processing a block of data
 462          * that includes NULs.
 463          */
 464         bzero(&mbstate, sizeof(mbstate));
 465         while (mbslength > 0) {
 466                 /* Convert one multibyte character to wchar_t. */
 467                 if (!cerr)
 468                         clen = mbrtowc(src, mbsrc, MIN(mbslength, MB_LEN_MAX),
 469                             &mbstate);
 470                 if (cerr || clen < 0) {
 471                         /* Conversion error, process as a byte instead. */
 472                         *src = (wint_t)(u_char)*mbsrc;
 473                         clen = 1;
 474                         cerr = 1;
 475                 }
 476                 if (clen == 0) {
 477                         /*
 478                          * NUL in input gives 0 return value. process
 479                          * as single NUL byte and keep going.
 480                          */
 481                         clen = 1;
 482                 }
 483                 /* Advance buffer character pointer. */
 484                 src++;
 485                 /* Advance input pointer by number of bytes read. */
 486                 mbsrc += clen;
 487                 /* Decrement input byte count. */
 488                 mbslength -= clen;
 489         }
 490         len = src - psrc;
 491         src = psrc;
 492
 493         /*
 494          * In the single character input case, we will have actually
 495          * processed two characters, c and nextc.  Reset len back to
 496          * just a single character.
 497          */
 498         if (mblength < len)
 499                 len = mblength;
 500
 501         /* Convert extra argument to list of characters for this mode. */
 502         extra = makeextralist(flags, mbextra);
 503         if (!extra) {
 504                 if (dlen && *dlen == 0) {
 505                         errno = ENOSPC;
 506                         goto out;
 507                 }
 508                 *mbdst = '\0';  /* can't create extra, return "" */
 509                 error = 0;
 510                 goto out;
 511         }
 512
 513         /* Look up which processing function to call. */
 514         f = getvisfun(flags);
 515
 516         /*
 517          * Main processing loop.
 518          * Call do_Xvis processing function one character at a time
 519          * with next character available for look-ahead.
 520          */
 521         for (start = dst; len > 0; len--) {
 522                 c = *src++;
 523                 dst = (*f)(dst, c, flags, len >= 1 ? *src : L'\0', extra);
 524                 if (dst == NULL) {
 525                         errno = ENOSPC;
 526                         goto out;
 527                 }
 528         }
 529
 530         /* Terminate the string in the buffer. */
 531         *dst = L'\0';
 532
 533         /*
 534          * Output loop.
 535          * Convert wchar_t string back to multibyte output string.
 536          * If we have hit a multi-byte conversion error on input,
 537          * output byte-by-byte here.  Else use wctomb().
 538          */
 539         len = wcslen(start);
 540         maxolen = dlen ? *dlen : (wcslen(start) * MB_LEN_MAX + 1);
 541         olen = 0;
 542         bzero(&mbstate, sizeof(mbstate));
 543         for (dst = start; len > 0; len--) {
 544                 if (!cerr)
 545                         clen = wcrtomb(mbdst, *dst, &mbstate);
 546                 if (cerr || clen < 0) {
 547                         /*
 548                          * Conversion error, process as a byte(s) instead.
 549                          * Examine each byte and higher-order bytes for
 550                          * data.  E.g.,
 551                          *      0x000000000000a264 -> a2 64
 552                          *      0x000000001f00a264 -> 1f 00 a2 64
 553                          */
 554                         clen = 0;
 555                         wmsk = 0;
 556                         for (i = sizeof(wmsk) - 1; i >= 0; i--) {
 557                                 shft = i * NBBY;
 558                                 bmsk = (uint64_t)0xffLL << shft;
 559                                 wmsk |= bmsk;
 560                                 if ((*dst & wmsk) || i == 0)
 561                                         mbdst[clen++] = (char)(
 562                                             (uint64_t)(*dst & bmsk) >>
 563                                             shft);
 564                         }
 565                         cerr = 1;
 566                 }
 567                 /* If this character would exceed our output limit, stop. */
 568                 if (olen + clen > (size_t)maxolen)
 569                         break;
 570                 /* Advance output pointer by number of bytes written. */
 571                 mbdst += clen;
 572                 /* Advance buffer character pointer. */
 573                 dst++;
 574                 /* Incrment output character count. */
 575                 olen += clen;
 576         }
 577
 578         /* Terminate the output string. */
 579         *mbdst = '\0';
 580
 581         if (flags & VIS_NOLOCALE) {
 582                 /* Pass conversion error flag out. */
 583                 if (cerr_ptr)
 584                         *cerr_ptr = cerr;
 585         }
 586
 587         free(extra);
 588         free(pdst);
 589         free(psrc);
 590
 591         return (int)olen;
 592 out:
 593         free(extra);
 594         free(pdst);
 595         free(psrc);
 596         free(mdst);
 597         return error;
 598 }
 599
 600 static int
 601 istrsenvisxl(char **mbdstp, size_t *dlen, const char *mbsrc,
 602     int flags, const char *mbextra, int *cerr_ptr)
 603 {
 604         return istrsenvisx(mbdstp, dlen, mbsrc,
 605             mbsrc != NULL ? strlen(mbsrc) : 0, flags, mbextra, cerr_ptr);
 606 }
 607
 608 #endif
 609
 610 #if !HAVE_SVIS
 611 /*
 612  *      The "svis" variants all take an "extra" arg that is a pointer
 613  *      to a NUL-terminated list of characters to be encoded, too.
 614  *      These functions are useful e. g. to encode strings in such a
 615  *      way so that they are not interpreted by a shell.
 616  */
 617
 618 char *
 619 svis(char *mbdst, int c, int flags, int nextc, const char *mbextra)
 620 {
 621         char cc[2];
 622         int ret;
 623
 624         cc[0] = c;
 625         cc[1] = nextc;
 626
 627         ret = istrsenvisx(&mbdst, NULL, cc, 1, flags, mbextra, NULL);
 628         if (ret < 0)
 629                 return NULL;
 630         return mbdst + ret;
 631 }
 632
 633 char *
 634 snvis(char *mbdst, size_t dlen, int c, int flags, int nextc, const char *mbextra)
 635 {
 636         char cc[2];
 637         int ret;
 638
 639         cc[0] = c;
 640         cc[1] = nextc;
 641
 642         ret = istrsenvisx(&mbdst, &dlen, cc, 1, flags, mbextra, NULL);
 643         if (ret < 0)
 644                 return NULL;
 645         return mbdst + ret;
 646 }
 647
 648 int
 649 strsvis(char *mbdst, const char *mbsrc, int flags, const char *mbextra)
 650 {
 651         return istrsenvisxl(&mbdst, NULL, mbsrc, flags, mbextra, NULL);
 652 }
 653
 654 int
 655 strsnvis(char *mbdst, size_t dlen, const char *mbsrc, int flags, const char *mbextra)
 656 {
 657         return istrsenvisxl(&mbdst, &dlen, mbsrc, flags, mbextra, NULL);
 658 }
 659
 660 int
 661 strsvisx(char *mbdst, const char *mbsrc, size_t len, int flags, const char *mbextra)
 662 {
 663         return istrsenvisx(&mbdst, NULL, mbsrc, len, flags, mbextra, NULL);
 664 }
 665
 666 int
 667 strsnvisx(char *mbdst, size_t dlen, const char *mbsrc, size_t len, int flags,
 668     const char *mbextra)
 669 {
 670         return istrsenvisx(&mbdst, &dlen, mbsrc, len, flags, mbextra, NULL);
 671 }
 672
 673 int
 674 strsenvisx(char *mbdst, size_t dlen, const char *mbsrc, size_t len, int flags,
 675     const char *mbextra, int *cerr_ptr)
 676 {
 677         return istrsenvisx(&mbdst, &dlen, mbsrc, len, flags, mbextra, cerr_ptr);
 678 }
 679 #endif
 680
 681 #if !HAVE_VIS
 682 /*
 683  * vis - visually encode characters
 684  */
 685 char *
 686 vis(char *mbdst, int c, int flags, int nextc)
 687 {
 688         char cc[2];
 689         int ret;
 690
 691         cc[0] = c;
 692         cc[1] = nextc;
 693
 694         ret = istrsenvisx(&mbdst, NULL, cc, 1, flags, "", NULL);
 695         if (ret < 0)
 696                 return NULL;
 697         return mbdst + ret;
 698 }
 699
 700 char *
 701 nvis(char *mbdst, size_t dlen, int c, int flags, int nextc)
 702 {
 703         char cc[2];
 704         int ret;
 705
 706         cc[0] = c;
 707         cc[1] = nextc;
 708
 709         ret = istrsenvisx(&mbdst, &dlen, cc, 1, flags, "", NULL);
 710         if (ret < 0)
 711                 return NULL;
 712         return mbdst + ret;
 713 }
 714
 715 /*
 716  * strvis - visually encode characters from src into dst
 717  *
 718  *      Dst must be 4 times the size of src to account for possible
 719  *      expansion.  The length of dst, not including the trailing NULL,
 720  *      is returned.
 721  */
 722
 723 int
 724 strvis(char *mbdst, const char *mbsrc, int flags)
 725 {
 726         return istrsenvisxl(&mbdst, NULL, mbsrc, flags, "", NULL);
 727 }
 728
 729 int
 730 strnvis(char *mbdst, size_t dlen, const char *mbsrc, int flags)
 731 {
 732         return istrsenvisxl(&mbdst, &dlen, mbsrc, flags, "", NULL);
 733 }
 734
 735 int
 736 stravis(char **mbdstp, const char *mbsrc, int flags)
 737 {
 738         *mbdstp = NULL;
 739         return istrsenvisxl(mbdstp, NULL, mbsrc, flags, "", NULL);
 740 }
 741
 742 /*
 743  * strvisx - visually encode characters from src into dst
 744  *
 745  *      Dst must be 4 times the size of src to account for possible
 746  *      expansion.  The length of dst, not including the trailing NULL,
 747  *      is returned.
 748  *
 749  *      Strvisx encodes exactly len characters from src into dst.
 750  *      This is useful for encoding a block of data.
 751  */
 752
 753 int
 754 strvisx(char *mbdst, const char *mbsrc, size_t len, int flags)
 755 {
 756         return istrsenvisx(&mbdst, NULL, mbsrc, len, flags, "", NULL);
 757 }
 758
 759 int
 760 strnvisx(char *mbdst, size_t dlen, const char *mbsrc, size_t len, int flags)
 761 {
 762         return istrsenvisx(&mbdst, &dlen, mbsrc, len, flags, "", NULL);
 763 }
 764
 765 int
 766 strenvisx(char *mbdst, size_t dlen, const char *mbsrc, size_t len, int flags,
 767     int *cerr_ptr)
 768 {
 769         return istrsenvisx(&mbdst, &dlen, mbsrc, len, flags, "", cerr_ptr);
 770 }
 771 #endif