__getmainargs.c File Reference

#include <apitest.h>
#include <stdio.h>
#include <string.h>

Include dependency graph for __getmainargs.c:

Go to the source code of this file.

Macros
#define	winetest_ok_str(x, y)    winetest_ok(strcmp(x, y) == 0, "Wrong string. Expected '%s', got '%s'\n", y, x)
#define	winetest_ok_wstr(x, y)    winetest_ok(wcscmp(x, y) == 0, "Wrong string. Expected '%s', got '%s'\n", wine_dbgstr_w(y), wine_dbgstr_w(x))
#define	ok_argsA   (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : ok_argsA_imp
#define	ok_argsW   (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : ok_argsW_imp

Functions
const char **	__p__acmdln (void)
void	__getmainargs (int *argc, char ***argv, char ***env, int expand_wildcards, int *new_mode)
const wchar_t **	__p__wcmdln (void)
void	__wgetmainargs (int *argc, wchar_t ***wargv, wchar_t ***wenv, int expand_wildcards, int *new_mode)
void	ok_argsA_imp (const char *input_args, const char *arg1, const char *arg2, const char *arg3)
void	ok_argsW_imp (const wchar_t *input_args, const wchar_t *arg1, const wchar_t *arg2, const wchar_t *arg3)
	START_TEST (__getmainargs)

Macro Definition Documentation

ok_argsA

#define ok_argsA   (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : ok_argsA_imp

Definition at line 23 of file __getmainargs.c.

ok_argsW

#define ok_argsW   (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : ok_argsW_imp

Definition at line 24 of file __getmainargs.c.

winetest_ok_str

#define winetest_ok_str	(		x,
			y
	)		winetest_ok(strcmp(x, y) == 0, "Wrong string. Expected '%s', got '%s'\n", y, x)

Definition at line 19 of file __getmainargs.c.

winetest_ok_wstr

#define winetest_ok_wstr	(		x,
			y
	)		winetest_ok(wcscmp(x, y) == 0, "Wrong string. Expected '%s', got '%s'\n", wine_dbgstr_w(y), wine_dbgstr_w(x))

Definition at line 21 of file __getmainargs.c.

Function Documentation

__getmainargs()

void __getmainargs	(	int *	argc,
		char ***	argv,
		char ***	env,
		int	expand_wildcards,
		int *	new_mode
	)

Definition at line 182 of file getargs.c.

{
   int i, doexpand, slashesAdded, escapedQuote, inQuotes, bufferIndex, anyLetter;
   size_t len;
   char* buffer;
 
   /* missing threading init */
 
   i = 0;
   doexpand = expand_wildcards;
   escapedQuote = FALSE;
   anyLetter = FALSE;
   slashesAdded = 0;
   inQuotes = 0;
   bufferIndex = 0;
 
   if (__argv && _environ)
   {
      *argv = __argv;
      *env = _environ;
      *argc = __argc;
      return;
   }
 
   __argc = 0;
 
   len = strlen(_acmdln);
   buffer = malloc(sizeof(char) * len);
 
   // Reference: https://msdn.microsoft.com/en-us/library/a1y7w461.aspx
   while (TRUE)
   {
      // Arguments are delimited by white space, which is either a space or a tab.
      if (i >= len || ((_acmdln[i] == ' ' || _acmdln[i] == '\t') && !inQuotes))
      {
         // Handle the case when empty spaces are in the end of the cmdline
         if (anyLetter)
         {
            aexpand(strndup(buffer, bufferIndex), doexpand);
         }
         // Copy the last element from buffer and quit the loop
         if (i >= len)
         {
            break;
         }
 
         while (_acmdln[i] == ' ' || _acmdln[i] == '\t')
            ++i;
         anyLetter = FALSE;
         bufferIndex = 0;
         slashesAdded = 0;
         escapedQuote = FALSE;
         continue;
      }
 
      anyLetter = TRUE;
 
      if (_acmdln[i] == '\\')
      {
         buffer[bufferIndex++] = _acmdln[i];
         ++slashesAdded;
         ++i;
         escapedQuote = FALSE;
         continue;
      }
 
      if (_acmdln[i] == '\"')
      {
         if (slashesAdded > 0)
         {
            if (slashesAdded % 2 == 0)
            {
               // If an even number of backslashes is followed by a double quotation mark, then one backslash (\‍)
               // is placed in the argv array for every pair of backslashes (\\‍), and the double quotation mark (")
               // is interpreted as a string delimiter.
               bufferIndex -= slashesAdded / 2;
            }
            else
            {
               // If an odd number of backslashes is followed by a double quotation mark, then one backslash (\‍)
               // is placed in the argv array for every pair of backslashes (\\‍) and the double quotation mark is
               // interpreted as an escape sequence by the remaining backslash, causing a literal double quotation mark (")
               // to be placed in argv.
               bufferIndex -= slashesAdded / 2 + 1;
               buffer[bufferIndex++] = '\"';
               slashesAdded = 0;
               escapedQuote = TRUE;
               ++i;
               continue;
            }
            slashesAdded = 0;
         }
         else if (!inQuotes && i > 0 && _acmdln[i - 1] == '\"' && !escapedQuote)
         {
            buffer[bufferIndex++] = '\"';
            ++i;
            escapedQuote = TRUE;
            continue;
         }
         slashesAdded = 0;
         escapedQuote = FALSE;
         inQuotes = !inQuotes;
         doexpand = inQuotes ? FALSE : expand_wildcards;
         ++i;
         continue;
      }
 
      buffer[bufferIndex++] = _acmdln[i];
      slashesAdded = 0;
      escapedQuote = FALSE;
      ++i;
   }
 
   /* Free the temporary buffer. */
   free(buffer);
 
   *argc = __argc;
   if (__argv == NULL)
   {
      __argv = (char**)malloc(sizeof(char*));
      __argv[0] = 0;
   }
   *argv = __argv;
   *env  = _environ;
 
   _pgmptr = malloc(MAX_PATH * sizeof(char));
   if (_pgmptr)
   {
      if (!GetModuleFileNameA(NULL, _pgmptr, MAX_PATH))
        _pgmptr[0] = '\0';
      else
        _pgmptr[MAX_PATH - 1] = '\0';
   }
   else
   {
      _pgmptr = _strdup(__argv[0]);
   }
 
   HeapValidate(GetProcessHeap(), 0, NULL);
 
   // if (new_mode) _set_new_mode(*new_mode);
}

Referenced by ok_argsA_imp().

__p__acmdln()

const char ** __p__acmdln

(

void

)

Definition at line 368 of file environ.c.

{
    return &_acmdln;
}

Referenced by ok_argsA_imp(), and Test__acmdln().

__p__wcmdln()

const wchar_t ** __p__wcmdln

(

void

)

Definition at line 376 of file environ.c.

{
    return &_wcmdln;
}

Referenced by ok_argsW_imp(), and Test__wcmdln().

__wgetmainargs()

void __wgetmainargs	(	int *	argc,
		wchar_t ***	wargv,
		wchar_t ***	wenv,
		int	expand_wildcards,
		int *	new_mode
	)

Definition at line 328 of file getargs.c.

{
   int i, doexpand, slashesAdded, escapedQuote, inQuotes, bufferIndex, anyLetter;
   size_t len;
   wchar_t* buffer;
 
   /* missing threading init */
 
   i = 0;
   doexpand = expand_wildcards;
   escapedQuote = FALSE;
   anyLetter = TRUE;
   slashesAdded = 0;
   inQuotes = 0;
   bufferIndex = 0;
 
   if (__wargv && __winitenv)
   {
      *wargv = __wargv;
      *wenv = __winitenv;
      *argc = __argc;
      return;
   }
 
   __argc = 0;
 
   len = wcslen(_wcmdln);
   buffer = malloc(sizeof(wchar_t) * len);
 
   // Reference: https://msdn.microsoft.com/en-us/library/a1y7w461.aspx
   while (TRUE)
   {
      // Arguments are delimited by white space, which is either a space or a tab.
      if (i >= len || ((_wcmdln[i] == ' ' || _wcmdln[i] == '\t') && !inQuotes))
      {
         // Handle the case when empty spaces are in the end of the cmdline
         if (anyLetter)
         {
            wexpand(wcsndup(buffer, bufferIndex), doexpand);
         }
         // Copy the last element from buffer and quit the loop
         if (i >= len)
         {
            break;
         }
 
         while (_wcmdln[i] == ' ' || _wcmdln[i] == '\t')
            ++i;
         anyLetter = FALSE;
         bufferIndex = 0;
         slashesAdded = 0;
         escapedQuote = FALSE;
         continue;
      }
 
      anyLetter = TRUE;
 
      if (_wcmdln[i] == '\\')
      {
         buffer[bufferIndex++] = _wcmdln[i];
         ++slashesAdded;
         ++i;
         escapedQuote = FALSE;
         continue;
      }
 
      if (_wcmdln[i] == '\"')
      {
         if (slashesAdded > 0)
         {
            if (slashesAdded % 2 == 0)
            {
               // If an even number of backslashes is followed by a double quotation mark, then one backslash (\‍)
               // is placed in the argv array for every pair of backslashes (\\‍), and the double quotation mark (")
               // is interpreted as a string delimiter.
               bufferIndex -= slashesAdded / 2;
            }
            else
            {
               // If an odd number of backslashes is followed by a double quotation mark, then one backslash (\‍)
               // is placed in the argv array for every pair of backslashes (\\‍) and the double quotation mark is
               // interpreted as an escape sequence by the remaining backslash, causing a literal double quotation mark (")
               // to be placed in argv.
               bufferIndex -= slashesAdded / 2 + 1;
               buffer[bufferIndex++] = '\"';
               slashesAdded = 0;
               escapedQuote = TRUE;
               ++i;
               continue;
            }
            slashesAdded = 0;
         }
         else if (!inQuotes && i > 0 && _wcmdln[i - 1] == '\"' && !escapedQuote)
         {
            buffer[bufferIndex++] = '\"';
            ++i;
            escapedQuote = TRUE;
            continue;
         }
         slashesAdded = 0;
         escapedQuote = FALSE;
         inQuotes = !inQuotes;
         doexpand = inQuotes ? FALSE : expand_wildcards;
         ++i;
         continue;
      }
 
      buffer[bufferIndex++] = _wcmdln[i];
      slashesAdded = 0;
      escapedQuote = FALSE;
      ++i;
   }
 
   /* Free the temporary buffer. */
   free(buffer);
 
   *argc = __argc;
   if (__wargv == NULL)
   {
      __wargv = (wchar_t**)malloc(sizeof(wchar_t*));
      __wargv[0] = 0;
   }
   *wargv = __wargv;
   *wenv = __winitenv;
 
   _wpgmptr = malloc(MAX_PATH * sizeof(wchar_t));
   if (_wpgmptr)
   {
      if (!GetModuleFileNameW(NULL, _wpgmptr, MAX_PATH))
        _wpgmptr[0] = '\0';
      else
        _wpgmptr[MAX_PATH - 1] = '\0';
   }
   else
   {
      _wpgmptr = _wcsdup(__wargv[0]);
   }
 
   HeapValidate(GetProcessHeap(), 0, NULL);
 
   // if (new_mode) _set_new_mode(*new_mode);
}

Referenced by ok_argsW_imp().

ok_argsA_imp()

void ok_argsA_imp	(	const char *	input_args,
		const char *	arg1,
		const char *	arg2,
		const char *	arg3
	)

Definition at line 28 of file __getmainargs.c.

{
    int argc = 0, mode = 0;
    int expect_count = arg3 == NULL ? (arg2 == NULL ? 2 : 3) : 4;
    char** argv, **env;
 
    /* Remove cached argv, setup our input as program argument. */
    *__p___argv() = NULL;
    *__p__acmdln() = input_args;
 
    /* Process the commandline stored in _acmdln */
    __getmainargs(&argc, &argv, &env, 0, &mode);
 
    winetest_ok(argc == expect_count, "Wrong value for argc, expected: %d, got: %d\n", expect_count, argc);
    if(argc != expect_count)
        return;
 
    winetest_ok_str(argv[0], "test.exe");
    winetest_ok_str(argv[1], arg1);
    if (expect_count > 2)
    {
        winetest_ok_str(argv[2], arg2);
        if (expect_count > 3)
            winetest_ok_str(argv[3], arg3);
    }
}

ok_argsW_imp()

void ok_argsW_imp	(	const wchar_t *	input_args,
		const wchar_t *	arg1,
		const wchar_t *	arg2,
		const wchar_t *	arg3
	)

Definition at line 56 of file __getmainargs.c.

{
    int argc = 0, mode = 0;
    int expect_count = arg3 == NULL ? (arg2 == NULL ? 2 : 3) : 4;
    wchar_t** argv, **env;
 
    /* Remove cached wargv, setup our input as program argument. */
    *__p___wargv() = NULL;
    *__p__wcmdln() = input_args;
 
    /* Process the commandline stored in _wcmdln */
    __wgetmainargs(&argc, &argv, &env, 0, &mode);
 
    winetest_ok(argc == expect_count, "Wrong value for argc, expected: %d, got: %d\n", expect_count, argc);
    if(argc != expect_count)
        return;
 
    winetest_ok_wstr(argv[0], L"test.exe");
    winetest_ok_wstr(argv[1], arg1);
    if (expect_count > 2)
    {
        winetest_ok_wstr(argv[2], arg2);
        if (expect_count > 3)
            winetest_ok_wstr(argv[3], arg3);
    }
}

START_TEST()

START_TEST

(

__getmainargs

)

Definition at line 83 of file __getmainargs.c.

{
    ok_argsA("test.exe \"a b c\" d e", "a b c", "d", "e");
    ok_argsA("test.exe \"ab\\\"c\" \"\\\\\" d", "ab\"c", "\\", "d");
    ok_argsA("test.exe a\\\\\\b d\"e f\"g h", "a\\\\\\b", "de fg", "h");
    ok_argsA("test.exe a\\\\\\\"b c d", "a\\\"b", "c", "d");
    ok_argsA("test.exe a\\\\\\\\\"b c\" d e", "a\\\\b c", "d", "e");
    ok_argsA("test.exe a b \"\"", "a", "b", "");
    ok_argsA("test.exe a \"\" b", "a", "", "b");
    ok_argsA("test.exe a \"b\"\" c", "a", "b\"", "c");
    ok_argsA("test.exe a \"b\\\"\" c", "a", "b\"", "c");
    ok_argsA("test.exe a  \"  b\\  \"\"  c", "a", "  b\\  \"", "c");
    ok_argsA("test.exe a  \"b\\  \"\"\"  c\" d", "a", "b\\  \"  c", "d");
    ok_argsA("test.exe a  \"b\\  \"\"\"  \"c  \"\"\"\" d", "a", "b\\  \"  c", "\" d");
    ok_argsA("test.exe a b c  ", "a", "b", "c");
    ok_argsA("test.exe \"a b c\"", "a b c", NULL, NULL);
 
    ok_argsW(L"test.exe \"a b c\" d e", L"a b c", L"d", L"e");
    ok_argsW(L"test.exe \"ab\\\"c\" \"\\\\\" d", L"ab\"c", L"\\", L"d");
    ok_argsW(L"test.exe a\\\\\\b d\"e f\"g h", L"a\\\\\\b", L"de fg", L"h");
    ok_argsW(L"test.exe a\\\\\\\"b c d", L"a\\\"b", L"c", L"d");
    ok_argsW(L"test.exe a\\\\\\\\\"b c\" d e", L"a\\\\b c", L"d", L"e");
    ok_argsW(L"test.exe a b \"\"", L"a", L"b", L"");
    ok_argsW(L"test.exe a \"\" b", L"a", L"", L"b");
    ok_argsW(L"test.exe a \"b\"\" c", L"a", L"b\"", L"c");
    ok_argsW(L"test.exe a \"b\\\"\" c", L"a", L"b\"", L"c");
    ok_argsW(L"test.exe a  \"  b\\  \"\"  c", L"a", L"  b\\  \"", L"c");
    ok_argsW(L"test.exe a  \"b\\  \"\"\"  c\" d", L"a", L"b\\  \"  c", L"d");
    ok_argsW(L"test.exe a  \"b\\  \"\"\"  \"c  \"\"\"\" d", L"a", L"b\\  \"  c", L"\" d");
    ok_argsW(L"test.exe a b c  ", L"a", L"b", L"c");
    ok_argsW(L"test.exe \"a b c\"", L"a b c", NULL, NULL);
}