CommandLine.c File Reference

#include <apitest.h>
#include <stdio.h>
#include <ndk/umtypes.h>
#include "./CmdLineUtil/CmdLineUtil.h"

Include dependency graph for CommandLine.c:

Go to the source code of this file.

Classes
struct	_TEST_CASE

Macros
#define	WIN32_NO_STATUS
#define	COUNT_OF(x)   (sizeof((x))/sizeof((x)[0]))
#define	SPACECHAR   L' '
#define	DQUOTECHAR   L'"'

Typedefs
typedef struct _TEST_CASE	TEST_CASE
typedef struct _TEST_CASE *	PTEST_CASE

Functions
LPWSTR	ExtractCmdLine (IN LPWSTR lpszCommandLine)
VOID	ExtractCmdLine_U (IN OUT PUNICODE_STRING pCommandLine_U)
static void	Test_CommandLine (IN ULONG TestNumber, IN PTEST_CASE TestCase)
	START_TEST (CommandLine)

Variables
static WCHAR	UtilityProgramDirectory [MAX_PATH]
static TEST_CASE	TestCases []

Macro Definition Documentation

COUNT_OF

#define COUNT_OF

(

x

)

(sizeof((x))/sizeof((x)[0]))

Definition at line 16 of file CommandLine.c.

DQUOTECHAR

#define DQUOTECHAR   L'"'

Definition at line 23 of file CommandLine.c.

SPACECHAR

#define SPACECHAR   L' '

Extracts the command tail from the command line (deletes the program's name and keep the rest).

Definition at line 22 of file CommandLine.c.

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 10 of file CommandLine.c.

Typedef Documentation

PTEST_CASE

typedef struct _TEST_CASE * PTEST_CASE

TEST_CASE

typedef struct _TEST_CASE TEST_CASE

Function Documentation

ExtractCmdLine()

LPWSTR ExtractCmdLine

(

IN LPWSTR

lpszCommandLine

)

Definition at line 25 of file CommandLine.c.

{
    BOOL inDoubleQuote = FALSE;

    /*
     * Skip the program's name (the first token in the command line).
     * Handle quoted program's name.
     */
    if (lpszCommandLine)
    {
        while ( (*lpszCommandLine > SPACECHAR) ||
                (*lpszCommandLine && inDoubleQuote) )
        {
            if (*lpszCommandLine == DQUOTECHAR)
                inDoubleQuote = !inDoubleQuote;

            ++lpszCommandLine;
        }

        /* Skip all white spaces preceeding the second token. */
        while (*lpszCommandLine && (*lpszCommandLine <= SPACECHAR))
            ++lpszCommandLine;
    }

    return lpszCommandLine;
}

Referenced by Test_CommandLine().

ExtractCmdLine_U()

VOID ExtractCmdLine_U

(

IN OUT PUNICODE_STRING

pCommandLine_U

)

Definition at line 52 of file CommandLine.c.

{
    BOOL inDoubleQuote = FALSE;
    PWSTR lpszCommandLine;

    /*
     * Skip the program's name (the first token in the command line).
     * Handle quoted program's name.
     */
    if (pCommandLine_U && pCommandLine_U->Buffer && (pCommandLine_U->Length != 0))
    {
        lpszCommandLine = pCommandLine_U->Buffer;

        while ( (pCommandLine_U->Length > 0) &&
                ( (*lpszCommandLine > SPACECHAR) ||
                  (*lpszCommandLine && inDoubleQuote) ) )
        {
            if (*lpszCommandLine == DQUOTECHAR)
                inDoubleQuote = !inDoubleQuote;

            ++lpszCommandLine;
            pCommandLine_U->Length -= sizeof(WCHAR);
        }

        /* Skip all white spaces preceeding the second token. */
        while ((pCommandLine_U->Length > 0) && *lpszCommandLine && (*lpszCommandLine <= SPACECHAR))
        {
            ++lpszCommandLine;
            pCommandLine_U->Length -= sizeof(WCHAR);
        }

        pCommandLine_U->Buffer = lpszCommandLine;
    }

    return;
}

Referenced by Test_CommandLine().

START_TEST()

START_TEST

(

CommandLine

)

Definition at line 285 of file CommandLine.c.

{
    ULONG i;

    DWORD dwRet;
    LPWSTR p = NULL;


    /*
     * Initialize the UtilityProgramDirectory variable.
     */
    dwRet = GetModuleFileNameW(NULL, UtilityProgramDirectory, COUNT_OF(UtilityProgramDirectory));
    ok(dwRet != 0, "ERROR: Cannot retrieve the path to the current running process, last error %lu\n", GetLastError());
    if (dwRet == 0) return;

    /* Path : executable.exe or "executable.exe" or C:\path\executable.exe or "C:\path\executable.exe" */
    p = wcsrchr(UtilityProgramDirectory, L'\\');
    if (p && *p != 0)
        *++p = 0; /* Null-terminate there : C:\path\ or "C:\path\ */
    else
        UtilityProgramDirectory[0] = 0; /* Suppress the executable.exe name */

    wcscat(UtilityProgramDirectory, L"testdata\\CmdLineUtil.exe");

    /* Close the opened quote if needed. */
    if (UtilityProgramDirectory[0] == L'"') wcscat(UtilityProgramDirectory, L"\"");


    /*
     * Now launch the tests.
     */
    for (i = 0 ; i < COUNT_OF(TestCases) ; ++i)
    {
        Test_CommandLine(i, &TestCases[i]);
    }
}

Test_CommandLine()

static void Test_CommandLine ( IN ULONG  TestNumber, IN PTEST_CASE  TestCase  )

static

Definition at line 114 of file CommandLine.c.

{
    BOOL bRet;

    BOOL bWasntInQuotes = (UtilityProgramDirectory[0] != L'"');
    WCHAR CmdLine[MAX_PATH] = L"";
    STARTUPINFOW si;
    PROCESS_INFORMATION pi;

    ZeroMemory(&si, sizeof(si));
    ZeroMemory(&pi, sizeof(pi));
    si.cb = sizeof(si);


    /* Initialize the command line. */
    if (TestCase->bEncloseProgramNameInQuotes && bWasntInQuotes)
        wcscpy(CmdLine, L"\"");

    wcscat(CmdLine, UtilityProgramDirectory);

    if (TestCase->bEncloseProgramNameInQuotes && bWasntInQuotes)
        wcscat(CmdLine, L"\"");

    /* Add a separating space and copy the tested command line parameters. */
    wcscat(CmdLine, L" ");
    wcscat(CmdLine, TestCase->CmdLine);


    /*
     * Launch the utility program and wait till it's terminated.
     */
    bRet = CreateProcessW(NULL,
                          CmdLine,
                          NULL, NULL,
                          FALSE,
                          CREATE_UNICODE_ENVIRONMENT,
                          NULL, NULL,
                          &si, &pi);
    ok(bRet, "Test %lu - Failed to launch ' %S ', error = %lu.\n", TestNumber, CmdLine, GetLastError());

    if (bRet)
    {
        /* Wait until child process exits. */
        WaitForSingleObject(pi.hProcess, INFINITE);

        /* Close process and thread handles. */
        CloseHandle(pi.hThread);
        CloseHandle(pi.hProcess);
    }

    /*
     * Analyses the result.
     */
    {
        /* Open the data file. */
        HANDLE hFile = CreateFileW(DATAFILE,
                                   GENERIC_READ,
                                   0, NULL,
                                   OPEN_EXISTING,
                                   FILE_ATTRIBUTE_NORMAL,
                                   NULL);
        ok(hFile != INVALID_HANDLE_VALUE, "Test %lu - Failed to open the data file 'C:\\cmdline.dat', error = %lu.\n", TestNumber, GetLastError());

        if (hFile != INVALID_HANDLE_VALUE)
        {
            WCHAR BuffWinMain[MAX_PATH]; LPWSTR WinMainCmdLine = BuffWinMain;
            WCHAR BuffWin32[MAX_PATH]  ; LPWSTR Win32CmdLine   = BuffWin32  ;
            WCHAR BuffNT[0xffff /* Maximum USHORT size */];
            UNICODE_STRING NTCmdLine;

            DWORD dwSize, dwStringSize;

            /*
             * Format of the data file :
             *
             * [size_of_string 4 bytes][null_terminated_C_string]
             * [size_of_string 4 bytes][null_terminated_C_string]
             * [UNICODE_STRING_structure][string_buffer_of_UNICODE_STRING]
             */

            /* 1- Read the WinMain's command line. */
            dwStringSize = 0;

            ReadFile(hFile,
                     &dwStringSize,
                     sizeof(dwStringSize),
                     &dwSize,
                     NULL);

            dwStringSize = min(dwStringSize, sizeof(BuffWinMain));
            ReadFile(hFile,
                     WinMainCmdLine,
                     dwStringSize,
                     &dwSize,
                     NULL);
            *(LPWSTR)((ULONG_PTR)WinMainCmdLine + dwStringSize) = 0;

            /* 2- Read the Win32 mode command line. */
            dwStringSize = 0;

            ReadFile(hFile,
                     &dwStringSize,
                     sizeof(dwStringSize),
                     &dwSize,
                     NULL);

            dwStringSize = min(dwStringSize, sizeof(BuffWin32));
            ReadFile(hFile,
                     Win32CmdLine,
                     dwStringSize,
                     &dwSize,
                     NULL);
            *(LPWSTR)((ULONG_PTR)Win32CmdLine + dwStringSize) = 0;

            /* 3- Finally, read the UNICODE_STRING command line. */
            ReadFile(hFile,
                     &NTCmdLine,
                     sizeof(NTCmdLine),
                     &dwSize,
                     NULL);

            NTCmdLine.Buffer = BuffNT;
            ReadFile(hFile,
                     NTCmdLine.Buffer,
                     NTCmdLine.Length,
                     &dwSize,
                     NULL);

            /* Now close the file. */
            CloseHandle(hFile);

            /*
             * Remove the program's name in the Win32 and NT command lines.
             */
            Win32CmdLine = ExtractCmdLine(Win32CmdLine);
            ExtractCmdLine_U(&NTCmdLine);

            /* Print the results */
            /*
            *(LPWSTR)((ULONG_PTR)NTCmdLine.Buffer + NTCmdLine.Length) = 0;
            printf("WinMain cmdline = '%S'\n"
                   "Win32   cmdline = '%S'\n"
                   "NT      cmdline = '%S'\n"
                   "NT       length = %u\n",
                   WinMainCmdLine,
                   Win32CmdLine,
                   NTCmdLine.Buffer, NTCmdLine.Length);
            */

            /*
             * Now check the results.
             */
            dwStringSize = min(lstrlenW(WinMainCmdLine), lstrlenW(Win32CmdLine));
            ok(wcslen(WinMainCmdLine) == wcslen(Win32CmdLine), "Test %lu - WinMain and Win32 command lines do not have the same length !\n", TestNumber);
            ok(wcsncmp(WinMainCmdLine, Win32CmdLine, dwStringSize) == 0, "Test %lu - WinMain and Win32 command lines are different !\n", TestNumber);

            dwStringSize = min(lstrlenW(WinMainCmdLine), NTCmdLine.Length / sizeof(WCHAR));
            ok(wcsncmp(WinMainCmdLine, NTCmdLine.Buffer, dwStringSize) == 0, "Test %lu - WinMain and NT command lines are different !\n", TestNumber);

            dwStringSize = min(lstrlenW(Win32CmdLine), NTCmdLine.Length / sizeof(WCHAR));
            ok(wcsncmp(Win32CmdLine, NTCmdLine.Buffer, dwStringSize) == 0, "Test %lu - Win32 and NT command lines are different !\n", TestNumber);
        }
    }

    /*
     * Always delete the data file.
     */
    DeleteFileW(DATAFILE);
}

Referenced by START_TEST().

Variable Documentation

TestCases

TEST_CASE TestCases[]

static

Initial value:

=
{
    {L"", FALSE},
    {L"foo bar", FALSE},
    {L"\"foo bar\"", FALSE},
    {L"foo \"bar John\" Doe", FALSE},

    {L"", TRUE},
    {L"foo bar", TRUE},
    {L"\"foo bar\"", TRUE},
    {L"foo \"bar John\" Doe", TRUE},
}

Definition at line 101 of file CommandLine.c.

Referenced by RunTestCases(), and START_TEST().

UtilityProgramDirectory

WCHAR UtilityProgramDirectory[MAX_PATH]

static

Definition at line 92 of file CommandLine.c.

Referenced by START_TEST(), and Test_CommandLine().