NetBSD/lib/libc/gen/shquote.c

182 lines
4.8 KiB
C

/* $NetBSD: shquote.c,v 1.5 2003/07/26 19:24:44 salo Exp $ */
/*
* Copyright (c) 2001 Christopher G. Demetriou
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software developed for the
* NetBSD Project. See http://www.NetBSD.org/ for
* information about NetBSD.
* 4. The name of the author may not be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
* <<Id: LICENSE,v 1.2 2000/06/14 15:57:33 cgd Exp>>
*/
/*
* Define SHQUOTE_USE_MULTIBYTE if you want shquote() to handle multibyte
* characters using mbrtowc().
*
* Please DO NOT rip this #ifdef out of the code. It's also here to help
* portability.
*/
#undef SHQUOTE_USE_MULTIBYTE
#include <stdlib.h>
#include <string.h>
#ifdef SHQUOTE_USE_MULTIBYTE
#include <limits.h>
#include <stdio.h>
#include <wchar.h>
#endif
/*
* shquote():
*
* Requotes arguments so that they'll be interpreted properly by the
* shell (/bin/sh).
*
* Wraps single quotes around the string, and replaces single quotes
* in the string with the sequence:
* '\''
*
* Returns the number of characters required to hold the resulting quoted
* argument.
*
* The buffer supplied is filled in and NUL-terminated. If 'bufsize'
* indicates that the buffer is too short to hold the output string, the
* first (bufsize - 1) bytes of quoted argument are filled in and the
* buffer is NUL-terminated.
*
* Changes could be made to optimize the length of strings output by this
* function:
*
* * if there are no metacharacters or whitespace in the input,
* the output could be the input string.
*/
#ifdef SHQUOTE_USE_MULTIBYTE
#define XLATE_OUTCH(x) wcrtomb(outch, (x), &mbso)
#define XLATE_INCH() \
do { \
n = mbrtowc(&c, arg, MB_CUR_MAX, &mbsi); \
} while (/*LINTED const cond*/0)
#else
#define XLATE_OUTCH(x) (outch[0] = (x), 1)
#define XLATE_INCH() \
do { \
n = ((c = *arg) != '\0') ? 1 : 0; \
} while (/*LINTED const cond*/0)
#endif
#define PUT(x) \
do { \
outchlen = XLATE_OUTCH(x); \
if (outchlen == (size_t)-1) \
goto bad; \
rv += outchlen; \
if (bufsize != 0) { \
if (bufsize < outchlen || \
(bufsize == outchlen && \
outch[outchlen - 1] != '\0')) { \
*buf = '\0'; \
bufsize = 0; \
} else { \
memcpy(buf, outch, outchlen); \
buf += outchlen; \
bufsize -= outchlen; \
} \
} \
} while (/*LINTED const cond*/0)
size_t
shquote(const char *arg, char *buf, size_t bufsize)
{
#ifdef SHQUOTE_USE_MULTIBYTE
char outch[MB_LEN_MAX];
mbstate_t mbsi, mbso;
wchar_t c, lastc;
size_t outchlen;
#else
char outch[1];
char c, lastc;
size_t outchlen;
#endif
size_t rv;
int n;
rv = 0;
lastc = 0;
#ifdef SHQUOTE_USE_MULTIBYTE
memset(&mbsi, 0, sizeof mbsi);
memset(&mbso, 0, sizeof mbso);
#endif
if (*arg != '\'')
PUT('\'');
for (;;) {
XLATE_INCH();
if (n == (size_t)-1)
goto bad;
if (n <= 0)
break;
arg += n;
lastc = c;
if (c == '\'') {
if (rv != 0)
PUT('\'');
PUT('\\');
PUT('\'');
for (;;) {
XLATE_INCH();
if (n <= 0 || c != '\'')
break;
PUT('\\');
PUT('\'');
arg += n;
}
if (n > 0)
PUT('\'');
} else
PUT(c);
}
if (lastc != '\'')
PUT('\'');
/* Put multibyte or NUL terminator, but don't count the NUL. */
PUT('\0');
rv--;
return rv;
bad:
/* A multibyte character encoding or decoding error occurred. */
return (size_t)-1;
}