interpreter.c File Reference

#include "precomp.h"
#include <debug.h>

Include dependency graph for interpreter.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Typedefs
typedef enum _INTERPRETER_STATE	INTERPRETER_STATE

Enumerations
enum	_INTERPRETER_STATE {   STATE_ANALYZE , STATE_COMMAND , STATE_GROUP , STATE_CONTEXT ,   STATE_PARENT_CONTEXT , STATE_DONE }

Functions
static PCOMMAND_ENTRY	GetGroupCommand (PWSTR pszCommand, PCOMMAND_GROUP pGroup)
static PCOMMAND_ENTRY	GetContextCommand (PWSTR pszCommand, PCONTEXT_ENTRY pContext)
static PCOMMAND_GROUP	GetContextGroup (PWSTR pszGroup, PCONTEXT_ENTRY pContext)
static PCONTEXT_ENTRY	GetContextSubContext (PWSTR pszContext, PCONTEXT_ENTRY pContext)
static DWORD	InterpretCommand (_In_ LPWSTR *argv, _In_ DWORD dwArgCount, _Inout_ PBOOL bDone)
DWORD	InterpretLine (_In_ LPWSTR pszInputLine)
VOID	PrintPrompt (_In_ PCONTEXT_ENTRY pContext)
VOID	InterpretInteractive (VOID)

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 12 of file interpreter.c.

Typedef Documentation

INTERPRETER_STATE

typedef enum _INTERPRETER_STATE INTERPRETER_STATE

Enumeration Type Documentation

_INTERPRETER_STATE

enum _INTERPRETER_STATE

Enumerator
STATE_ANALYZE
STATE_COMMAND
STATE_GROUP
STATE_CONTEXT
STATE_PARENT_CONTEXT
STATE_DONE

Definition at line 15 of file interpreter.c.

{
    STATE_ANALYZE,
    STATE_COMMAND,
    STATE_GROUP,
    STATE_CONTEXT,
    STATE_PARENT_CONTEXT,
    STATE_DONE
} INTERPRETER_STATE;

Function Documentation

GetContextCommand()

static PCOMMAND_ENTRY GetContextCommand ( PWSTR  pszCommand, PCONTEXT_ENTRY  pContext  )

static

Definition at line 56 of file interpreter.c.

{
    PCOMMAND_ENTRY pCommand;
 
    DPRINT("GetContextCommand(%S %p)\n", pszCommand, pContext);
 
    if (pszCommand == NULL)
        return NULL;
 
    pCommand = pContext->pCommandListHead;
    while (pCommand)
    {
        if (MatchToken(pszCommand, pCommand->pwszCmdToken))
            return pCommand;
 
        pCommand = pCommand->pNext;
    }
 
    return NULL;
}

Referenced by InterpretCommand().

GetContextGroup()

static PCOMMAND_GROUP GetContextGroup ( PWSTR  pszGroup, PCONTEXT_ENTRY  pContext  )

static

Definition at line 82 of file interpreter.c.

{
    PCOMMAND_GROUP pGroup;
 
    DPRINT("GetContextGroup(%S %p)\n", pszGroup, pContext);
 
    if (pszGroup == NULL)
        return NULL;
 
    pGroup = pContext->pGroupListHead;
    while (pGroup)
    {
        if (MatchToken(pszGroup, pGroup->pwszCmdGroupToken))
            return pGroup;
 
        pGroup = pGroup->pNext;
    }
 
    return NULL;
}

Referenced by InterpretCommand().

GetContextSubContext()

static PCONTEXT_ENTRY GetContextSubContext ( PWSTR  pszContext, PCONTEXT_ENTRY  pContext  )

static

Definition at line 108 of file interpreter.c.

{
    PCONTEXT_ENTRY pSubContext;
 
    DPRINT("GetContextSubContext(%S %p)\n", pszContext, pContext);
 
    if (pszContext == NULL)
        return NULL;
 
    pSubContext = pContext->pSubContextHead;
    while (pSubContext)
    {
        if (MatchToken(pszContext, pSubContext->pszContextName))
            return pSubContext;
 
        pSubContext = pSubContext->pNext;
    }
 
    return NULL;
}

Referenced by InterpretCommand().

GetGroupCommand()

static PCOMMAND_ENTRY GetGroupCommand ( PWSTR  pszCommand, PCOMMAND_GROUP  pGroup  )

static

Definition at line 30 of file interpreter.c.

{
    PCOMMAND_ENTRY pCommand;
 
    DPRINT("GetGroupCommand(%S %p)\n", pszCommand, pGroup);
 
    if (pszCommand == NULL)
        return NULL;
 
    pCommand = pGroup->pCommandListHead;
    while (pCommand)
    {
        if (MatchToken(pszCommand, pCommand->pwszCmdToken))
            return pCommand;
 
        pCommand = pCommand->pNext;
    }
 
    return NULL;
}

Referenced by InterpretCommand().

InterpretCommand()

static DWORD InterpretCommand ( _In_ LPWSTR *  argv, _In_ DWORD  dwArgCount, _Inout_ PBOOL  bDone  )

static

Definition at line 134 of file interpreter.c.

{
    PCONTEXT_ENTRY pTempContext, pTempSubContext = NULL;
    PCOMMAND_GROUP pGroup = NULL;
    PCOMMAND_ENTRY pCommand = NULL;
    INTERPRETER_STATE State = STATE_ANALYZE;
    DWORD dwArgIndex = 0;
    DWORD dwError = ERROR_SUCCESS;
 
    /* If no args provided */
    if (dwArgCount == 0)
        return ERROR_SUCCESS;
 
    if (pCurrentContext == NULL)
        pCurrentContext = pRootContext;
 
    /* Check for help keywords */
    if ((dwArgCount == 1) &&
        ((_wcsicmp(argv[0], L"?") == 0) || (_wcsicmp(argv[0], L"help") == 0)))
    {
        PrintContextHelp(pCurrentContext);
        return ERROR_SUCCESS;
    }
 
    pTempContext = pCurrentContext;
    while (dwArgIndex < dwArgCount)
    {
        switch (State)
        {
            case STATE_ANALYZE:
                DPRINT("STATE_ANALYZE\n");
 
                pCommand = GetContextCommand(argv[dwArgIndex], pTempContext);
                if (pCommand != NULL)
                {
                    State = STATE_COMMAND;
                    break;
                }
 
                pGroup = GetContextGroup(argv[dwArgIndex], pTempContext);
                if (pGroup != NULL)
                {
                    State = STATE_GROUP;
                    break;
                }
 
                pTempSubContext = GetContextSubContext(argv[dwArgIndex], pTempContext);
                if (pTempSubContext != NULL)
                {
                    State = STATE_CONTEXT;
                    break;
                }
 
                State = STATE_PARENT_CONTEXT;
                break;
 
            case STATE_COMMAND:
                DPRINT("STATE_COMMAND\n");
 
                /* Check for help keywords */
                if (((dwArgIndex + 1) == (dwArgCount - 1)) &&
                    ((_wcsicmp(argv[dwArgIndex + 1], L"?") == 0) || (_wcsicmp(argv[dwArgIndex + 1], L"help") == 0)))
                {
                    PrintCommandHelp(pTempContext, pGroup, pCommand);
                    State = STATE_DONE;
                    break;
                }
 
                if (pCommand->pfnCmdHandler != NULL)
                {
                    dwArgIndex++;
                    dwError = pCommand->pfnCmdHandler(pszMachine, argv, dwArgIndex, dwArgCount, 0, NULL, bDone);
                    if (dwError != ERROR_SUCCESS)
                    {
                        if (dwError == ERROR_SHOW_USAGE)
                        {
                            PrintCommandHelp(pTempContext, pGroup, pCommand);
                            dwError = ERROR_SUPPRESS_OUTPUT;
                        }
                    }
                    else
                    {
                        /* Execute the commands following a pushd command */
                        if ((_wcsicmp(pCommand->pwszCmdToken, L"pushd") == 0) &&
                            (dwArgIndex < dwArgCount))
                        {
                            State = STATE_ANALYZE;
                            break;
                        }
                    }
                }
 
                State = STATE_DONE;
                break;
 
            case STATE_GROUP:
                DPRINT("STATE_GROUP\n");
 
                /* Check for group without command */
                if (dwArgIndex == (dwArgCount - 1))
                {
                    PrintGroupHelp(pTempContext, pGroup->pwszCmdGroupToken, TRUE);
                    State = STATE_DONE;
                    break;
                }
 
                /* Check for help keywords */
                if (((dwArgIndex + 1) <= (dwArgCount - 1)) &&
                    ((_wcsicmp(argv[dwArgIndex + 1], L"?") == 0) || (_wcsicmp(argv[dwArgIndex + 1], L"help") == 0)))
                {
                    PrintGroupHelp(pTempContext, pGroup->pwszCmdGroupToken, TRUE);
                    State = STATE_DONE;
                    break;
                }
 
                dwArgIndex++;
                pCommand = GetGroupCommand(argv[dwArgIndex], pGroup);
                if (pCommand != NULL)
                {
                    State = STATE_COMMAND;
                    break;
                }
 
                dwError = ERROR_CMD_NOT_FOUND;
                State = STATE_DONE;
                break;
 
            case STATE_CONTEXT:
                DPRINT("STATE_CONTEXT\n");
 
                if (pTempSubContext == pCurrentContext)
                {
                    if (dwArgIndex != (dwArgCount - 1))
                        dwError = ERROR_CMD_NOT_FOUND;
 
                    State = STATE_DONE;
                    break;
                }
 
                if (dwArgIndex == (dwArgCount - 1))
                {
                    DPRINT("Set current context\n");
                    pCurrentContext = pTempSubContext;
                    if (pCurrentContext->pfnConnectFn)
                        dwError = pCurrentContext->pfnConnectFn(pszMachine);
                    State = STATE_DONE;
                    break;
                }
 
                /* Check for help keywords */
                if (((dwArgIndex + 1) <= (dwArgCount - 1)) &&
                    ((_wcsicmp(argv[dwArgIndex + 1], L"?") == 0) || (_wcsicmp(argv[dwArgIndex + 1], L"help") == 0)))
                {
                    PrintContextHelp(pTempSubContext);
                    State = STATE_DONE;
                    break;
                }
 
                DPRINT("Change temorary context\n");
                pTempContext = pTempSubContext;
                State = STATE_ANALYZE;
                dwArgIndex++;
                break;
 
            case STATE_PARENT_CONTEXT:
                DPRINT("STATE_PARENT_CONTEXT\n");
 
                if (pTempContext->pParentContext == NULL)
                {
                    dwError = ERROR_CMD_NOT_FOUND;
                    State = STATE_DONE;
                    break;
                }
 
                DPRINT("Change temorary context\n");
                pTempContext = pTempContext->pParentContext;
                State = STATE_ANALYZE;
                dwArgIndex = 0;
                break;
 
            case STATE_DONE:
                DPRINT("STATE_DONE dwError %ld\n", dwError);
                return dwError;
        }
    }
 
    /* Done */
    return ERROR_SUCCESS;
} 

Referenced by InterpretInteractive(), and InterpretLine().

InterpretInteractive()

VOID InterpretInteractive

(

VOID

)

Definition at line 405 of file interpreter.c.

{
    WCHAR input_line[MAX_STRING_SIZE];
    LPWSTR args_vector[MAX_ARGS_COUNT];
    DWORD dwArgCount = 0, i, len;
    BOOL bWhiteSpace = TRUE;
    BOOL bDone = FALSE;
    BOOL bInQuotes = FALSE;
    LPWSTR ptr, pStartQuote, pEndQuote;
    DWORD dwError = ERROR_SUCCESS;
 
    for (;;)
    {
        dwArgCount = 0;
        memset(args_vector, 0, sizeof(args_vector));
 
        /* Shown just before the input where the user places commands */
        if (pszMachine)
            ConPrintf(StdOut, L"[%s] ", pszMachine);
        PrintPrompt(pCurrentContext);
        ConPuts(StdOut, L">");
 
        /* Get input from the user. */
        fgetws(input_line, MAX_STRING_SIZE, stdin);
 
        ptr = input_line;
        while (*ptr != 0)
        {
            if (*ptr == L'\"')
                bInQuotes = (bInQuotes) ? FALSE : TRUE;
 
            if ((iswspace(*ptr) && (bInQuotes == FALSE)) || *ptr == L'\n')
            {
                *ptr = UNICODE_NULL;
                bWhiteSpace = TRUE;
            }
            else
            {
                if ((bWhiteSpace != FALSE) && (dwArgCount < MAX_ARGS_COUNT))
                {
                    args_vector[dwArgCount] = ptr;
                    dwArgCount++;
                }
                bWhiteSpace = FALSE;
            }
            ptr++;
        }
 
        /* Remove quotation marks */
        for (i = 0; i < dwArgCount; i++)
        {
            pStartQuote = wcschr(args_vector[i], L'\"');
            if (pStartQuote)
            {
                pEndQuote = wcschr(pStartQuote + 1, L'\"');
                if (pEndQuote)
                {
                    len = pEndQuote - pStartQuote;
                    memmove(pStartQuote, pStartQuote + 1, (len - 1) * sizeof(WCHAR));
                    *(pEndQuote - 1) = UNICODE_NULL;
                }
            }
        }
 
        dwError = InterpretCommand(args_vector, dwArgCount, &bDone);
        if ((dwError != ERROR_SUCCESS) && (dwError != ERROR_SUPPRESS_OUTPUT))
        {
            PWSTR pszCommandString = MergeStrings(args_vector, dwArgCount);
            PrintError(NULL, dwError, pszCommandString);
            HeapFree(GetProcessHeap(), 0, pszCommandString);
        }
        if (dwError != ERROR_SUPPRESS_OUTPUT)
            ConPuts(StdOut, L"\n");
 
        if (bDone)
            break;
    }
}

Referenced by wmain().

InterpretLine()

DWORD InterpretLine

(

_In_ LPWSTR

pszInputLine

)

Definition at line 333 of file interpreter.c.

{
    LPWSTR args_vector[MAX_ARGS_COUNT];
    DWORD dwArgCount = 0, i, len;
    BOOL bWhiteSpace = TRUE;
    BOOL bDone = FALSE;
    BOOL bInQuotes = FALSE;
    LPWSTR ptr, pStartQuote, pEndQuote;
 
    memset(args_vector, 0, sizeof(args_vector));
 
    ptr = pszInputLine;
    while (*ptr != 0)
    {
        if (*ptr == L'\"')
            bInQuotes = (bInQuotes) ? FALSE : TRUE;
 
        if ((iswspace(*ptr) && (bInQuotes == FALSE)) || *ptr == L'\n')
        {
            *ptr = 0;
            bWhiteSpace = TRUE;
        }
        else
        {
            if ((bWhiteSpace != FALSE) && (dwArgCount < MAX_ARGS_COUNT))
            {
                args_vector[dwArgCount] = ptr;
                dwArgCount++;
            }
 
            bWhiteSpace = FALSE;
        }
 
        ptr++;
    }
 
    /* Remove quotation marks */
    for (i = 0; i < dwArgCount; i++)
    {
        pStartQuote = wcschr(args_vector[i], L'\"');
        if (pStartQuote)
        {
            pEndQuote = wcschr(pStartQuote + 1, L'\"');
            if (pEndQuote)
            {
                len = pEndQuote - pStartQuote;
                memmove(pStartQuote, pStartQuote + 1, (len - 1) * sizeof(WCHAR));
                *(pEndQuote - 1) = UNICODE_NULL;
            }
        }
    }
 
    return InterpretCommand(args_vector, dwArgCount, &bDone);
}

Referenced by RunScript(), and wmain().

PrintPrompt()

VOID PrintPrompt

(

_In_ PCONTEXT_ENTRY

pContext

)

Definition at line 391 of file interpreter.c.

{
    if (pContext != pRootContext)
    {
        PrintPrompt(pContext->pParentContext);
        ConPuts(StdOut, L" ");
    }
 
    ConPuts(StdOut, pContext->pszContextName);
}

Referenced by Batch(), ExecuteCommandWithEcho(), InterpretInteractive(), PrintPrompt(), ReadCommand(), and ReadLine().