set.c File Reference

#include "precomp.h"

Include dependency graph for set.c:

Go to the source code of this file.

Macros
#define	ENV_BUFFER_SIZE   1024
#define	PARSE_IDENT(ident, identlen, p)

Functions
static BOOL	seta_eval (LPCTSTR expr)
static LPCTSTR	skip_ws (LPCTSTR p)
static LPTSTR	GetQuotedString (TCHAR *p)
INT	cmd_set (LPTSTR param)
static INT	ident_len (LPCTSTR p)
static INT	seta_identval (LPCTSTR ident)
static BOOL	calc (INT *lval, TCHAR op, INT rval)
static BOOL	seta_stmt (LPCTSTR *p_, INT *result)
static BOOL	seta_unaryTerm (LPCTSTR *p_, INT *result)
static BOOL	seta_mulTerm (LPCTSTR *p_, INT *result)
static BOOL	seta_ltorTerm (LPCTSTR *p_, INT *result, LPCTSTR ops, BOOL(*subTerm)(LPCTSTR *, INT *))
static BOOL	seta_addTerm (LPCTSTR *p_, INT *result)
static BOOL	seta_logShiftTerm (LPCTSTR *p_, INT *result)
static BOOL	seta_bitAndTerm (LPCTSTR *p_, INT *result)
static BOOL	seta_bitExclOrTerm (LPCTSTR *p_, INT *result)
static BOOL	seta_bitOrTerm (LPCTSTR *p_, INT *result)
static BOOL	seta_expr (LPCTSTR *p_, INT *result)
static BOOL	seta_assignment (LPCTSTR *p_, INT *result)

Macro Definition Documentation

ENV_BUFFER_SIZE

#define ENV_BUFFER_SIZE   1024

Definition at line 42 of file set.c.

PARSE_IDENT

#define PARSE_IDENT	(		ident,
			identlen,
			p
	)

Value:

do { \
    identlen = ident_len(p); \
    ident = (LPTSTR)alloca((identlen + 1) * sizeof(TCHAR)); \
    memmove(ident, p, identlen * sizeof(TCHAR)); \
    ident[identlen] = 0; \
    p += identlen; \
} while (0)

Definition at line 271 of file set.c.

Function Documentation

calc()

static BOOL calc ( INT *  lval, TCHAR  op, INT  rval  )

static

Definition at line 291 of file set.c.

{
    switch (op)
    {
    case '*':
        *lval *= rval;
        break;

    case '/':
    {
        if (rval == 0)
        {
            // FIXME: Localize
            ConErrPuts(_T("Division by zero error.\n"));
            nErrorLevel = 0x400023D1; // 1073750993;
            return FALSE;
        }
        *lval /= rval;
        break;
    }

    case '%':
    {
        if (rval == 0)
        {
            // FIXME: Localize
            ConErrPuts(_T("Division by zero error.\n"));
            nErrorLevel = 0x400023D1; // 1073750993;
            return FALSE;
        }
        *lval %= rval;
        break;
    }

    case '+':
        *lval += rval;
        break;
    case '-':
        *lval -= rval;
        break;
    case '&':
        *lval &= rval;
        break;
    case '^':
        *lval ^= rval;
        break;
    case '|':
        *lval |= rval;
        break;

    default:
        ConErrResPuts(STRING_INVALID_OPERAND);
        nErrorLevel = 0x400023CE; // 1073750990;
        return FALSE;
    }
    return TRUE;
}

Referenced by seta_assignment(), and seta_ltorTerm().

cmd_set()

INT cmd_set

(

LPTSTR

param

)

Definition at line 72 of file set.c.

{
    INT retval = 0;
    LPTSTR p;
    LPTSTR lpEnv;
    LPTSTR lpOutput;

    if (!_tcsncmp(param, _T("/?"), 2))
    {
        ConOutResPaging(TRUE,STRING_SET_HELP);
        return 0;
    }

    param = (LPTSTR)skip_ws(param);

    /* If no parameters, show the environment */
    if (param[0] == _T('\0'))
    {
        lpEnv = (LPTSTR)GetEnvironmentStrings();
        if (lpEnv)
        {
            lpOutput = lpEnv;
            while (*lpOutput)
            {
                /* Do not display the special '=X:' environment variables */
                if (*lpOutput != _T('='))
                {
                    ConOutPuts(lpOutput);
                    ConOutChar(_T('\n'));
                }
                lpOutput += _tcslen(lpOutput) + 1;
            }
            FreeEnvironmentStrings(lpEnv);
        }

        retval = 0;
        goto Quit;
    }

    /* The /A does *NOT* have to be followed by a whitespace */
    if (!_tcsnicmp(param, _T("/A"), 2))
    {
        BOOL Success;

        /* Save error level since seta_eval() modifies it, as
         * we need to set it later according to specific rules. */
        INT nOldErrorLevel = nErrorLevel;

        StripQuotes(param);
        Success = seta_eval(skip_ws(param + 2));
        if (!Success)
        {
#if 0
            /* Might seem random but this is what windows xp does -- This is a message ID */
            retval = 9165;
#endif
            retval = nErrorLevel;
            nErrorLevel = nOldErrorLevel;
        }
        else
        {
            retval = 0;
        }
        goto Quit;
    }

    if (!_tcsnicmp(param, _T("/P"), 2))
    {
        TCHAR value[1023];
        param = GetQuotedString((LPTSTR)skip_ws(param + 2));
        p = _tcschr(param, _T('='));
        if (!p)
        {
            ConErrResPuts(STRING_SYNTAX_COMMAND_INCORRECT);
            retval = 1;
            goto Quit;
        }

        *p++ = _T('\0');
        ConOutPrintf(_T("%s"), GetQuotedString(p));
        ConInString(value, ARRAYSIZE(value));

        if (!*value || !SetEnvironmentVariable(param, value))
        {
            retval = 1;
            goto Quit;
        }
        retval = 0;
        goto Quit;
    }

    param = GetQuotedString(param);

    p = _tcschr(param, _T('='));
    if (p)
    {
        /* Set or remove the environment variable */
        if (p == param)
        {
            /* Handle set =val case */
            ConErrResPuts(STRING_SYNTAX_COMMAND_INCORRECT);
            retval = 1;
            goto Quit;
        }

        *p++ = _T('\0');
        if (!SetEnvironmentVariable(param, *p ? p : NULL))
        {
            retval = 1;
            goto Quit;
        }
    }
    else
    {
        /* Display all the environment variables with the given prefix */
        LPTSTR pOrgParam = param;
        BOOLEAN bFound = FALSE;
        BOOLEAN bRestoreSpace;

        /*
         * Trim the prefix from "special" characters (only when displaying the
         * environment variables), so that e.g. "SET ,; ,;FOO" will display all
         * the variables starting by "FOO".
         * The SET command allows as well to set an environment variable whose name
         * actually contains these characters (e.g. "SET ,; ,;FOO=42"); however,
         * by trimming the characters, doing "SET ,; ,;FOO" would not allow seeing
         * such variables.
         * Thus, we also save a pointer to the original variable name prefix, that
         * we will look it up as well below.
         */
        while (_istspace(*param) || *param == _T(',') || *param == _T(';'))
            ++param;

        /* Just remove the very last space, if present */
        p = _tcsrchr(param, _T(' '));
        bRestoreSpace = (p != NULL);
        if (!p)
            p = param + _tcslen(param);
        *p = _T('\0');

        lpEnv = GetEnvironmentStrings();
        if (lpEnv)
        {
            lpOutput = lpEnv;
            while (*lpOutput)
            {
                /* Look up for both the original and truncated variable name prefix */
                if (!_tcsnicmp(lpOutput, pOrgParam, p - pOrgParam) ||
                    !_tcsnicmp(lpOutput, param, p - param))
                {
                    ConOutPuts(lpOutput);
                    ConOutChar(_T('\n'));
                    bFound = TRUE;
                }
                lpOutput += _tcslen(lpOutput) + 1;
            }
            FreeEnvironmentStrings(lpEnv);
        }

        /* Restore the truncated space for correctly
         * displaying the error message, if any. */
        if (bRestoreSpace)
            *p = _T(' ');

        if (!bFound)
        {
            ConErrResPrintf(STRING_SET_ENV_ERROR, param);
            retval = 1;
            goto Quit;
        }
    }

Quit:
    if (BatType != CMD_TYPE)
    {
        if (retval != 0)
            nErrorLevel = retval;
    }
    else
    {
        nErrorLevel = retval;
    }

    return retval;
}

GetQuotedString()

static LPTSTR GetQuotedString ( TCHAR *  p )

static

Definition at line 58 of file set.c.

{
    TCHAR *end;
    if (*p == _T('"'))
    {
        p = (LPTSTR)skip_ws(p + 1);
        /* If a matching quote is found, truncate the string */
        end = _tcsrchr(p, _T('"'));
        if (end)
            *end = _T('\0');
    }
    return p;
}

Referenced by cmd_set().

ident_len()

static INT ident_len ( LPCTSTR  p )

static

Definition at line 259 of file set.c.

{
    LPCTSTR p2 = p;
    if (__iscsymf(*p))
    {
        ++p2;
        while (__iscsym(*p2))
            ++p2;
    }
    return (INT)(p2-p);
}

Referenced by cc_token(), and try_parse_ccval().

seta_addTerm()

static BOOL seta_addTerm ( LPCTSTR *  p_, INT *  result  )

static

Definition at line 491 of file set.c.

{
    return seta_ltorTerm(p_, result, _T("*/%"), seta_mulTerm);
}

Referenced by seta_logShiftTerm().

seta_assignment()

static BOOL seta_assignment ( LPCTSTR *  p_, INT *  result  )

static

Definition at line 579 of file set.c.

{
    LPCTSTR p = *p_;
    LPTSTR ident;
    TCHAR op = 0;
    INT identlen, exprval;

    PARSE_IDENT(ident, identlen, p);
    if (identlen)
    {
        p = skip_ws(p);

        /* Handle = assignment */
        if (*p == _T('='))
        {
            op = *p;
            p = skip_ws(p + 1);
        }
        /* Handle *= /= %= += -= &= ^= |= assignments */
        else if (_tcschr(_T("*/%+-&^|"), *p))
        {
            op = *p;

            /* Find the '=', there may be some spaces before it */
            p = skip_ws(p + 1);
            if (*p != _T('='))
            {
                op = 0;
                goto evaluate;
            }

            /* Skip it */
            p = skip_ws(p + 1);
        }
        /* Handle <<= >>= assignments */
        else if (_tcschr(_T("<>"), *p))
        {
            op = *p;

            /* Check whether the next non-whitespace character is the same operator */
            p = skip_ws(p + 1);
            if (*p != op)
            {
                op = 0;
                goto evaluate;
            }

            /* Find the '=', there may be some spaces before it */
            p = skip_ws(p + 1);
            if (*p != _T('='))
            {
                op = 0;
                goto evaluate;
            }

            /* Skip it */
            p = skip_ws(p + 1);
        }
    }

evaluate:
    /* Allow to chain multiple assignments, such as: a=b=1 */
    if (ident && op)
    {
        INT identval;
        LPTSTR buf;

        if (!seta_assignment(&p, &exprval))
            return FALSE;

        identval = seta_identval(ident);

        switch (op)
        {
        /* Handle = assignment */
        case '=':
            identval = exprval;
            break;

        /* Handle <<= assignment */
        case '<':
        {
            /* Shift left has to be a positive number, 0-31 otherwise 0 is returned,
             * which differs from the compiler (for example gcc) so being explicit. */
            if (exprval < 0 || exprval >= (8 * sizeof(identval)))
                identval = 0;
            else
                identval <<= exprval;
            break;
        }

        /* Handle >>= assignment */
        case '>':
            identval >>= exprval;
            break;

        /* Other assignments */
        default:
            if (!calc(&identval, op, exprval))
                return FALSE;
        }

        buf = (LPTSTR)alloca(32 * sizeof(TCHAR));
        _sntprintf(buf, 32, _T("%i"), identval);
        SetEnvironmentVariable(ident, buf); // TODO FIXME - check return value
        exprval = identval;
    }
    else
    {
        /* Restore p in case we found an identifier but not an operator */
        p = *p_;
        if (!seta_expr(&p, &exprval))
            return FALSE;
    }

    *result = exprval;
    *p_ = p;
    return TRUE;
}

Referenced by seta_stmt().

seta_bitAndTerm()

static BOOL seta_bitAndTerm ( LPCTSTR *  p_, INT *  result  )

static

Definition at line 503 of file set.c.

{
    LPCTSTR p = *p_;
    INT lval;

    /* Evaluate the left-hand side */
    if (!seta_logShiftTerm(&p, &lval))
        return FALSE;

    /* Handle << >> operators */
    while (*p && _tcschr(_T("<>"), *p))
    {
        INT rval;
        TCHAR op = *p;

        /* Check whether the next non-whitespace character is the same operator */
        p = skip_ws(p + 1);
        if (*p != op)
            break;

        /* Skip it */
        p = skip_ws(p + 1);

        /* Evaluate the immediate right-hand side */
        if (!seta_logShiftTerm(&p, &rval))
            return FALSE;

        /* This becomes the new left-hand side for the next iteration */
        switch (op)
        {
        case '<':
        {
            /* Shift left has to be a positive number, 0-31 otherwise 0 is returned,
             * which differs from the compiler (for example gcc) so being explicit. */
            if (rval < 0 || rval >= (8 * sizeof(lval)))
                lval = 0;
            else
                lval <<= rval;
            break;
        }

        case '>':
            lval >>= rval;
            break;

        default:
            ConErrResPuts(STRING_INVALID_OPERAND);
            nErrorLevel = 0x400023CE; // 1073750990;
            return FALSE;
        }
    }

    *result = lval;
    *p_ = p;
    return TRUE;
}

Referenced by seta_bitExclOrTerm().

seta_bitExclOrTerm()

static BOOL seta_bitExclOrTerm ( LPCTSTR *  p_, INT *  result  )

static

Definition at line 561 of file set.c.

{
    return seta_ltorTerm(p_, result, _T("&"), seta_bitAndTerm);
}

Referenced by seta_bitOrTerm().

seta_bitOrTerm()

static BOOL seta_bitOrTerm ( LPCTSTR *  p_, INT *  result  )

static

Definition at line 567 of file set.c.

{
    return seta_ltorTerm(p_, result, _T("^"), seta_bitExclOrTerm);
}

Referenced by seta_expr().

seta_eval()

static BOOL seta_eval ( LPCTSTR  expr )

static

Definition at line 723 of file set.c.

{
    INT rval;

    if (!*p)
    {
        ConErrResPuts(STRING_SYNTAX_COMMAND_INCORRECT);
        nErrorLevel = 1;
        return FALSE;
    }
    if (!seta_stmt(&p, &rval))
        return FALSE;

    /* If unparsed data remains, fail and bail out */
    if (*p)
    {
        ConErrResPuts(STRING_SYNTAX_COMMAND_INCORRECT); // Actually syntax error / missing operand.
        nErrorLevel = 0x400023CE; // 1073750990;
        return FALSE;
    }

    /* Echo the result of the evaluation only in interactive (non-batch) mode */
    if (!bc)
        ConOutPrintf(_T("%i"), rval);

    return TRUE;
}

Referenced by cmd_set().

seta_expr()

static BOOL seta_expr ( LPCTSTR *  p_, INT *  result  )

static

Definition at line 573 of file set.c.

{
    return seta_ltorTerm(p_, result, _T("|"), seta_bitOrTerm);
}

Referenced by seta_assignment().

seta_identval()

static INT seta_identval ( LPCTSTR  ident )

static

Definition at line 281 of file set.c.

{
    LPCTSTR identVal = GetEnvVarOrSpecial(ident);
    if (!identVal)
        return 0;
    else
        return _tcstol(identVal, NULL, 0);
}

Referenced by seta_assignment(), and seta_unaryTerm().

seta_logShiftTerm()

static BOOL seta_logShiftTerm ( LPCTSTR *  p_, INT *  result  )

static

Definition at line 497 of file set.c.

{
    return seta_ltorTerm(p_, result, _T("+-"), seta_addTerm);
}

Referenced by seta_bitAndTerm().

seta_ltorTerm()

static BOOL seta_ltorTerm ( LPCTSTR *  p_, INT *  result, LPCTSTR  ops, BOOL(*)(LPCTSTR *, INT *)  subTerm  )

static

Definition at line 460 of file set.c.

{
    LPCTSTR p = *p_;
    INT lval;

    /* Evaluate the left-hand side */
    if (!subTerm(&p, &lval))
        return FALSE;

    while (*p && _tcschr(ops, *p))
    {
        INT rval;
        TCHAR op = *p;

        p = skip_ws(p + 1);

        /* Evaluate the immediate right-hand side */
        if (!subTerm(&p, &rval))
            return FALSE;

        /* This becomes the new left-hand side for the next iteration */
        if (!calc(&lval, op, rval))
            return FALSE;
    }

    *result = lval;
    *p_ = p;
    return TRUE;
}

Referenced by seta_addTerm(), seta_bitExclOrTerm(), seta_bitOrTerm(), seta_expr(), and seta_logShiftTerm().

seta_mulTerm()

static BOOL seta_mulTerm ( LPCTSTR *  p_, INT *  result  )

static

Definition at line 416 of file set.c.

{
    LPCTSTR p = *p_;
    TCHAR op = 0;
    INT rval;

    if (_tcschr(_T("!~-+"), *p))
    {
        op = *p;
        p = skip_ws(p + 1);

        if (!seta_mulTerm(&p, &rval))
            return FALSE;

        switch (op)
        {
        case '!':
            rval = !rval;
            break;
        case '~':
            rval = ~rval;
            break;
        case '-':
            rval = -rval;
            break;
#if 0
        case '+':
            rval = rval;
            break;
#endif
        }
    }
    else
    {
        if (!seta_unaryTerm(&p, &rval))
            return FALSE;
    }

    *result = rval;
    *p_ = p;
    return TRUE;
}

Referenced by seta_addTerm().

seta_stmt()

static BOOL seta_stmt ( LPCTSTR *  p_, INT *  result  )

static

Definition at line 700 of file set.c.

{
    LPCTSTR p = *p_;
    INT rval;

    if (!seta_assignment(&p, &rval))
        return FALSE;

    /* Loop over each statement */
    while (*p == _T(','))
    {
        p = skip_ws(p + 1);

        if (!seta_assignment(&p, &rval))
            return FALSE;
    }

    *result = rval;
    *p_ = p;
    return TRUE;
}

Referenced by seta_eval(), and seta_unaryTerm().

seta_unaryTerm()

static BOOL seta_unaryTerm ( LPCTSTR *  p_, INT *  result  )

static

Definition at line 353 of file set.c.

{
    LPCTSTR p = *p_;
    INT rval;

    if (*p == _T('('))
    {
        p = skip_ws(p + 1);
        if (!seta_stmt(&p, &rval))
            return FALSE;
        if (*p++ != _T(')'))
        {
            ConErrResPuts(STRING_EXPECTED_CLOSE_PAREN);
            nErrorLevel = 0x400023CC; // 1073750988;
            return FALSE;
        }
        *result = rval;
    }
    else if (_istdigit(*p))
    {
        errno = 0;
        rval = _tcstol(p, (LPTSTR*)&p, 0);

        /* Check for overflow / underflow */
        if (errno == ERANGE)
        {
            // FIXME: Localize
            ConErrPuts(_T("Invalid number. Numbers are limited to 32-bits of precision.\n"));
            nErrorLevel = 0x400023D0; // 1073750992;
            return FALSE;
        }
        /*
         * _tcstol() stopped at the first non-digit character. If it's not a whitespace,
         * or if it's the start of a possible identifier, this means the number being
         * interpreted was invalid.
         */
        else if (*p && !_istspace(*p) && __iscsymf(*p))
        {
            // FIXME: Localize
            ConErrPuts(_T("Invalid number. Numeric constants are either decimal (42), hexadecimal (0x2A), or octal (052).\n"));
            nErrorLevel = 0x400023CF; // 1073750991;
            return FALSE;
        }
        *result = rval;
    }
    else if (__iscsymf(*p))
    {
        LPTSTR ident;
        INT identlen;
        PARSE_IDENT(ident, identlen, p);
        *result = seta_identval(ident);
    }
    else
    {
        ConErrResPuts(STRING_EXPECTED_NUMBER_OR_VARIABLE);
        nErrorLevel = 0x400023CD; // 1073750989;
        return FALSE;
    }
    *p_ = skip_ws(p);
    return TRUE;
}

Referenced by seta_mulTerm().

skip_ws()

static LPCTSTR skip_ws ( LPCTSTR  p )

static

Definition at line 48 of file set.c.

{
    while (*p && *p <= _T(' '))
        ++p;
    return p;
}

Referenced by cmd_set(), findend(), GetQuotedString(), import_file(), process_preprocessor(), seta_assignment(), seta_bitAndTerm(), seta_ltorTerm(), seta_mulTerm(), seta_stmt(), seta_unaryTerm(), and tokenize().