interpreter.c File Reference

#include "diskpart.h"
#include "diskpart_msg.h"

Include dependency graph for interpreter.c:

Go to the source code of this file.

Functions
EXIT_CODE	InterpretCmd (int argc, LPWSTR *argv)
EXIT_CODE	InterpretScript (_In_ PWSTR input_line)
VOID	InterpretMain (VOID)

Variables
COMMAND	cmds []

Function Documentation

InterpretCmd()

EXIT_CODE InterpretCmd	(	int	argc,
		LPWSTR *	argv
	)

Definition at line 108 of file interpreter.c.

{
    PCOMMAND cmdptr;
    PCOMMAND cmdptr1 = NULL;
    PCOMMAND cmdptr2 = NULL;
    PCOMMAND cmdptr3 = NULL;
 
    /* If no args provided */
    if (argc < 1)
        return EXIT_SUCCESS;
 
    /* First, determine if the user wants to exit
       or to use a comment */
    if (_wcsicmp(argv[0], L"exit") == 0)
        return EXIT_EXIT;
 
    if (_wcsicmp(argv[0], L"rem") == 0)
        return EXIT_SUCCESS;
 
    /* Scan internal command table */
    for (cmdptr = cmds; cmdptr->cmd1; cmdptr++)
    {
        if ((cmdptr1 == NULL) &&
            (cmdptr->cmd1 != NULL) && (_wcsicmp(argv[0], cmdptr->cmd1) == 0))
            cmdptr1 = cmdptr;
 
        if ((cmdptr2 == NULL) &&
            (argc >= 2) &&
            (cmdptr->cmd1 != NULL) && (_wcsicmp(argv[0], cmdptr->cmd1) == 0) &&
            (cmdptr->cmd2 != NULL) && (_wcsicmp(argv[1], cmdptr->cmd2) == 0))
            cmdptr2 = cmdptr;
 
        if ((cmdptr3 == NULL) &&
            (argc >= 3) &&
            (cmdptr->cmd1 != NULL) && (_wcsicmp(argv[0], cmdptr->cmd1) == 0) &&
            (cmdptr->cmd2 != NULL) && (_wcsicmp(argv[1], cmdptr->cmd2) == 0) &&
            (cmdptr->cmd3 != NULL) && (_wcsicmp(argv[2], cmdptr->cmd3) == 0))
            cmdptr3 = cmdptr;
    }
 
    if (cmdptr3 != NULL)
    {
        if (cmdptr3->func == NULL)
            return HelpCommand(cmdptr3);
        else
            return cmdptr3->func(argc, argv);
    }
    else if (cmdptr2 != NULL)
    {
        if (cmdptr2->func == NULL)
            return HelpCommand(cmdptr2);
        else
            return cmdptr2->func(argc, argv);
    }
    else if (cmdptr1 != NULL)
    {
        if (cmdptr1->func == NULL)
            return HelpCommand(cmdptr1);
        else
            return cmdptr1->func(argc, argv);
    }
 
    HelpCommandList();
 
    return EXIT_SUCCESS;
}

Referenced by InterpretMain(), and InterpretScript().

InterpretMain()

VOID InterpretMain

(

VOID

)

Definition at line 232 of file interpreter.c.

{
    WCHAR input_line[MAX_STRING_SIZE];
    LPWSTR args_vector[MAX_ARGS_COUNT];
    INT args_count = 0;
    BOOL bWhiteSpace = TRUE;
    BOOL bQuote = FALSE;
    EXIT_CODE ExitCode = EXIT_SUCCESS;
    LPWSTR ptr;
 
    while (ExitCode != EXIT_EXIT)
    {
        args_count = 0;
        memset(args_vector, 0, sizeof(args_vector));
 
        /* Shown just before the input where the user places commands */
        ConResPuts(StdOut, IDS_APP_PROMPT);
 
        /* Get input from the user. */
        fgetws(input_line, MAX_STRING_SIZE, stdin);
 
        bQuote = FALSE;
        ptr = input_line;
        while (*ptr != 0)
        {
            if (*ptr == L'"')
                bQuote = !bQuote;
 
            if ((iswspace(*ptr) && (bQuote == FALSE))|| *ptr == L'\n')
            {
                *ptr = 0;
                bWhiteSpace = TRUE;
            }
            else
            {
                if ((bWhiteSpace != FALSE) && (bQuote == FALSE) && (args_count < MAX_ARGS_COUNT))
                {
                    args_vector[args_count] = ptr;
                    args_count++;
                }
                bWhiteSpace = FALSE;
            }
            ptr++;
        }
 
        /* Send the string to find the command */
        ExitCode = InterpretCmd(args_count, args_vector);
    }
}

Referenced by wmain().

InterpretScript()

EXIT_CODE InterpretScript

(

_In_ PWSTR

input_line

)

Definition at line 183 of file interpreter.c.

{
    PWSTR args_vector[MAX_ARGS_COUNT];
    INT args_count = 0;
    BOOL bWhiteSpace = TRUE;
    BOOL bQuote = FALSE;
    PWSTR ptr;
 
    memset(args_vector, 0, sizeof(args_vector));
 
    bQuote = FALSE;
    ptr = input_line;
    while (*ptr != 0)
    {
        if (*ptr == L'"')
            bQuote = !bQuote;
 
        if ((iswspace(*ptr) && (bQuote == FALSE))|| *ptr == L'\n')
        {
            *ptr = 0;
            bWhiteSpace = TRUE;
        }
        else
        {
            if ((bWhiteSpace != FALSE) && (bQuote == FALSE) && (args_count < MAX_ARGS_COUNT))
            {
                args_vector[args_count] = ptr;
                args_count++;
            }
 
            bWhiteSpace = FALSE;
        }
 
        ptr++;
    }
 
    /* sends the string to find the command */
    return InterpretCmd(args_count, args_vector);
}

Variable Documentation

cmds

COMMAND cmds[]

Definition at line 14 of file interpreter.c.

Referenced by InterpretCmd().