da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#pragma prototyped

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#include <ast.h>

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#include <ccode.h>

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#include <ctype.h>

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chinchar*

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chinfmtquote(const char* as, const char* qb, const char* qe, size_t n, int flags)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin{

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin register unsigned char* s = (unsigned char*)as;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin register unsigned char* e = s + n;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin register char* b;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin register int c;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin register int escaped;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin register int spaced;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin register int doublequote;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin register int singlequote;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin int shell;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin char* f;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin char* buf;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c = 4 * (n + 1);

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (qb)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c += strlen((char*)qb);

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (qe)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c += strlen((char*)qe);

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin b = buf = fmtbuf(c);

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin shell = 0;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin doublequote = 0;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin singlequote = 0;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (qb)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (qb[0] == '$' && qb[1] == '\'' && qb[2] == 0)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin shell = 1;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else if ((flags & FMT_SHELL) && qb[1] == 0)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (qb[0] == '"')

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin doublequote = 1;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else if (qb[0] == '\'')

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin singlequote = 1;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin while (*b = *qb++)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin b++;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else if (flags & FMT_SHELL)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin doublequote = 1;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin f = b;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin escaped = spaced = !!(flags & FMT_ALWAYS);

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin while (s < e)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if ((c = mbsize(s)) > 1)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin while (c-- && s < e)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = *s++;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c = *s++;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (!(flags & FMT_ESCAPED) && (iscntrl(c) || !isprint(c) || c == '\\'))

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin escaped = 1;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = '\\';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin switch (c)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin case CC_bel:

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c = 'a';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin break;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin case '\b':

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c = 'b';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin break;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin case '\f':

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c = 'f';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin break;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin case '\n':

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c = 'n';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin break;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin case '\r':

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c = 'r';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin break;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin case '\t':

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c = 't';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin break;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin case CC_vt:

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c = 'v';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin break;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin case CC_esc:

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c = 'E';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin break;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin case '\\':

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin break;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin default:

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (!(flags & FMT_WIDE) || !(c & 0200))

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = '0' + ((c >> 6) & 07);

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = '0' + ((c >> 3) & 07);

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c = '0' + (c & 07);

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin b--;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin break;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else if (c == '\\')

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin escaped = 1;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = c;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (*s)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c = *s++;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else if (qe && strchr(qe, c))

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (singlequote && c == '\'')

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin spaced = 1;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = '\'';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = '\\';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = '\'';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c = '\'';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin escaped = 1;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = '\\';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else if (c == '$' || c == '`')

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (c == '$' && (flags & FMT_PARAM) && (*s == '{' || *s == '('))

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (singlequote || shell)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin escaped = 1;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = '\'';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = c;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = *s++;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (shell)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin spaced = 1;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = '$';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c = '\'';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin escaped = 1;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = c;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin c = *s++;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else if (doublequote)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = '\\';

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else if (singlequote || (flags & FMT_SHELL))

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin spaced = 1;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else if (!spaced && !escaped && (isspace(c) || ((flags & FMT_SHELL) || shell) && (strchr("\";~&|()<>[]*?", c) || c == '#' && (b == f || isspace(*(b - 1))))))

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin spaced = 1;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b++ = c;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (qb)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (!escaped)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin buf += shell + !spaced;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (qe && (escaped || spaced))

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin while (*b = *qe++)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin b++;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *b = 0;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin return buf;

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin}

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chinchar*

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chinfmtnesq(const char* as, const char* qs, size_t n)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin{

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin return fmtquote(as, NiL, qs, n, 0);

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin}

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chinchar*

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chinfmtesq(const char* as, const char* qs)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin{

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin return fmtquote(as, NiL, qs, strlen((char*)as), 0);

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin}

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chinchar*

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chinfmtesc(const char* as)

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin{

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin return fmtquote(as, NiL, NiL, strlen((char*)as), 0);

da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin}