precomp.h File Reference

#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <windef.h>
#include <winbase.h>
#include <winreg.h>
#include <wincon.h>
#include <winuser.h>
#include <iphlpapi.h>
#include <iphlpapi_undoc.h>
#include <ndk/rtlfuncs.h>
#include <conutils.h>
#include <netsh.h>
#include <netsh_undoc.h>
#include "resource.h"

Include dependency graph for precomp.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	_DLL_LIST_ENTRY
struct	_HELPER_ENTRY
struct	_COMMAND_ENTRY
struct	_COMMAND_GROUP
struct	_CONTEXT_ENTRY

Macros
#define	WIN32_NO_STATUS
#define	MAX_STRING_SIZE   1024
#define	MAX_ARGS_COUNT   256
#define	REG_NETSH_PATH   L"Software\\Microsoft\\NetSh"

Typedefs
typedef struct _DLL_LIST_ENTRY	DLL_LIST_ENTRY
typedef struct _DLL_LIST_ENTRY *	PDLL_LIST_ENTRY
typedef struct _HELPER_ENTRY	HELPER_ENTRY
typedef struct _HELPER_ENTRY *	PHELPER_ENTRY
typedef struct _COMMAND_ENTRY	COMMAND_ENTRY
typedef struct _COMMAND_ENTRY *	PCOMMAND_ENTRY
typedef struct _COMMAND_GROUP	COMMAND_GROUP
typedef struct _COMMAND_GROUP *	PCOMMAND_GROUP
typedef struct _CONTEXT_ENTRY	CONTEXT_ENTRY
typedef struct _CONTEXT_ENTRY *	PCONTEXT_ENTRY

Functions
VOID	InitAliases (VOID)
VOID	DestroyAliases (VOID)
DWORD WINAPI	AliasCommand (LPCWSTR pwszMachine, LPWSTR *argv, DWORD dwCurrentIndex, DWORD dwArgCount, DWORD dwFlags, LPCVOID pvData, BOOL *pbDone)
DWORD WINAPI	ShowAliasCommand (LPCWSTR pwszMachine, LPWSTR *argv, DWORD dwCurrentIndex, DWORD dwArgCount, DWORD dwFlags, LPCVOID pvData, BOOL *pbDone)
DWORD WINAPI	UnaliasCommand (LPCWSTR pwszMachine, LPWSTR *argv, DWORD dwCurrentIndex, DWORD dwArgCount, DWORD dwFlags, LPCVOID pvData, BOOL *pbDone)
BOOL	CreateRootContext (VOID)
VOID	CleanupContext (VOID)
PCONTEXT_ENTRY	FindContextByGuid (_In_ const GUID *pGuid)
VOID	PrintCommandHelp (_In_ PCONTEXT_ENTRY pContext, _In_ PCOMMAND_GROUP pGroup, _In_ PCOMMAND_ENTRY pCommand)
VOID	PrintGroupHelp (_In_ PCONTEXT_ENTRY pContext, _In_ LPWSTR pszGroupName, _In_ BOOL bRecurse)
VOID	PrintContextHelp (_In_ PCONTEXT_ENTRY pContext)
DWORD	CreateRootHelper (VOID)
VOID	LoadHelpers (VOID)
VOID	UnloadHelpers (VOID)
PHELPER_ENTRY	FindHelper (_In_ const GUID *pguidHelper, _In_ PHELPER_ENTRY pHelper)
DWORD WINAPI	AddHelperCommand (LPCWSTR pwszMachine, LPWSTR *ppwcArguments, DWORD dwCurrentIndex, DWORD dwArgCount, DWORD dwFlags, LPCVOID pvData, BOOL *pbDone)
DWORD WINAPI	DeleteHelperCommand (LPCWSTR pwszMachine, LPWSTR *ppwcArguments, DWORD dwCurrentIndex, DWORD dwArgCount, DWORD dwFlags, LPCVOID pvData, BOOL *pbDone)
DWORD WINAPI	ShowHelperCommand (LPCWSTR pwszMachine, PWSTR *ppwcArguments, DWORD dwCurrentIndex, DWORD dwArgCount, DWORD dwFlags, LPCVOID pvData, BOOL *pbDone)
DWORD	InterpretLine (_In_ LPWSTR pszFileName)
VOID	InterpretInteractive (VOID)
DWORD	RunScript (_In_ LPCWSTR filename)
LPWSTR	MergeStrings (_In_ LPWSTR pszStringArray[], _In_ UINT nCount)
BOOL	CheckOsVersion (_In_ PNS_OSVERSIONCHECK pfnOsVersionCheck)
HRESULT	GetWmiVersionInfo (VOID)

Variables
PCONTEXT_ENTRY	pRootContext
PCONTEXT_ENTRY	pCurrentContext
PHELPER_ENTRY	pHelperListHead
HMODULE	g_hModule
PWSTR	g_pszMachine
BOOL	g_bOnline
UINT	VersionInfoArchitecture
UINT	VersionInfoOsProductSuite
UINT	VersionInfoOsType
WCHAR	VersionInfoVersion [MAX_PATH]
WCHAR	VersionInfoBuildNumber [MAX_PATH]
WCHAR	VersionInfoServicePackMajorVersion [MAX_PATH]
WCHAR	VersionInfoServicePackMinorVersion [MAX_PATH]

Macro Definition Documentation

MAX_ARGS_COUNT

#define MAX_ARGS_COUNT   256

Definition at line 37 of file precomp.h.

MAX_STRING_SIZE

#define MAX_STRING_SIZE   1024

Definition at line 36 of file precomp.h.

REG_NETSH_PATH

#define REG_NETSH_PATH   L"Software\\Microsoft\\NetSh"

Definition at line 39 of file precomp.h.

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 16 of file precomp.h.

Typedef Documentation

COMMAND_ENTRY

typedef struct _COMMAND_ENTRY COMMAND_ENTRY

COMMAND_GROUP

typedef struct _COMMAND_GROUP COMMAND_GROUP

CONTEXT_ENTRY

typedef struct _CONTEXT_ENTRY CONTEXT_ENTRY

DLL_LIST_ENTRY

typedef struct _DLL_LIST_ENTRY DLL_LIST_ENTRY

HELPER_ENTRY

typedef struct _HELPER_ENTRY HELPER_ENTRY

PCOMMAND_ENTRY

typedef struct _COMMAND_ENTRY * PCOMMAND_ENTRY

PCOMMAND_GROUP

typedef struct _COMMAND_GROUP * PCOMMAND_GROUP

PCONTEXT_ENTRY

typedef struct _CONTEXT_ENTRY * PCONTEXT_ENTRY

PDLL_LIST_ENTRY

typedef struct _DLL_LIST_ENTRY * PDLL_LIST_ENTRY

PHELPER_ENTRY

typedef struct _HELPER_ENTRY * PHELPER_ENTRY

Function Documentation

AddHelperCommand()

DWORD WINAPI AddHelperCommand	(	LPCWSTR	pwszMachine,
		LPWSTR *	ppwcArguments,
		DWORD	dwCurrentIndex,
		DWORD	dwArgCount,
		DWORD	dwFlags,
		LPCVOID	pvData,
		BOOL *	pbDone
	)

Definition at line 461 of file helper.c.

{
    DWORD dwError = ERROR_SUCCESS;
 
    DPRINT("AddHelperCommand()\n");
 
    if (dwArgCount == 2)
    {
//        ConResPrintf(StdErr, IDS_INVALID_SYNTAX);
//        ConResPrintf(StdErr, IDS_HLP_ADD_HELPER_EX);
        return 1;
    }
 
    dwError = LoadHelperDll(ppwcArguments[2], TRUE);
    if (dwError != ERROR_SUCCESS)
        return dwError;
 
    StartHelpers(pHelperListHead);
 
    return ERROR_SUCCESS;
}

Referenced by CreateRootContext().

AliasCommand()

DWORD WINAPI AliasCommand	(	LPCWSTR	pwszMachine,
		LPWSTR *	argv,
		DWORD	dwCurrentIndex,
		DWORD	dwArgCount,
		DWORD	dwFlags,
		LPCVOID	pvData,
		BOOL *	pbDone
	)

Definition at line 123 of file alias.c.

{
    PALIAS_ENTRY pAliasEntry = NULL;
    PWSTR pszAlias;
 
    DPRINT("AliasCommand(dwCurrentIndex %lu dwArgCount %lu)\n",
           dwCurrentIndex, dwArgCount);
 
    /* Show aliases */
    if (dwArgCount - dwCurrentIndex == 0)
    {
        ShowAliases();
        return ERROR_SUCCESS;
    }
 
    if (dwArgCount - dwCurrentIndex == 1)
    {
        ShowAlias(argv[dwCurrentIndex]);
        return ERROR_SUCCESS;
    }
 
 
    /* TODO: Check builtin commands */
 
 
    pAliasEntry = GetAliasEntry(argv[dwCurrentIndex]);
    if (pAliasEntry)
    {
        pszAlias = MergeStrings(&argv[dwCurrentIndex + 1], dwArgCount - dwCurrentIndex - 1);
        DPRINT("Alias: %S\n", pszAlias);
        if (pszAlias == NULL)
            return ERROR_NOT_ENOUGH_MEMORY;
 
        if (pAliasEntry->pszAlias)
            HeapFree(GetProcessHeap(), 0, pAliasEntry->pszAlias);
 
        pAliasEntry->pszAlias = pszAlias;
    }
    else
    {
        pAliasEntry = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(ALIAS_ENTRY));
        if (pAliasEntry == NULL)
            return ERROR_NOT_ENOUGH_MEMORY;
 
        pAliasEntry->pszAliasName = HeapAlloc(GetProcessHeap(), 0,
                                              (wcslen(argv[dwCurrentIndex]) + 1) * sizeof(WCHAR));
        if (pAliasEntry->pszAliasName == NULL)
        {
            HeapFree(GetProcessHeap(), 0, pAliasEntry);
            return ERROR_NOT_ENOUGH_MEMORY;
        }
 
        wcscpy(pAliasEntry->pszAliasName, argv[dwCurrentIndex]);
        DPRINT("AliasName: %S\n", pAliasEntry->pszAliasName);
 
        pAliasEntry->pszAlias = MergeStrings(&argv[dwCurrentIndex + 1], dwArgCount - dwCurrentIndex - 1);
        DPRINT("Alias: %S\n", pAliasEntry->pszAlias);
        if (pAliasEntry->pszAlias == NULL)
        {
            HeapFree(GetProcessHeap(), 0, pAliasEntry->pszAliasName);
            HeapFree(GetProcessHeap(), 0, pAliasEntry);
            return ERROR_NOT_ENOUGH_MEMORY;
        }
 
        if (AliasListHead == NULL)
        {
            AliasListHead = pAliasEntry;
            AliasListTail = pAliasEntry;
        }
        else
        {
            pAliasEntry->pPrev = AliasListTail;
            AliasListTail->pNext = pAliasEntry;
            AliasListTail = pAliasEntry;
        }
    }
 
    DPRINT("Done\n");
 
    return ERROR_OKAY;
}

Referenced by CreateRootContext().

CheckOsVersion()

BOOL CheckOsVersion

(

_In_ PNS_OSVERSIONCHECK

pfnOsVersionCheck

)

Definition at line 35 of file wmi.c.

{
    DPRINT("CheckOsVersion(%p)\n", pfnOsVersionCheck);
 
    if (pfnOsVersionCheck == NULL)
        return TRUE;
 
    return pfnOsVersionCheck(VersionInfoOsType,
                             VersionInfoOsProductSuite,
                             VersionInfoVersion,
                             VersionInfoBuildNumber,
                             VersionInfoServicePackMajorVersion,
                             VersionInfoServicePackMinorVersion,
                             VersionInfoArchitecture,
                             0);
}

Referenced by InterpretCommand(), PrintCommandHelp(), PrintContext(), PrintShortGroupCommands(), and PrintSubcontexts().

CleanupContext()

VOID CleanupContext

(

VOID

)

Definition at line 1016 of file context.c.

{
    /* Delete the context stack */
 
}

Referenced by wmain().

CreateRootContext()

BOOL CreateRootContext

(

VOID

)

Definition at line 810 of file context.c.

{
    PCOMMAND_GROUP pGroup;
 
    pRootContext = AddContext(NULL, L"netsh", NULL);
    DPRINT("pRootContext: %p\n", pRootContext);
    if (pRootContext == NULL)
        return FALSE;
 
    pRootContext->hModule = g_hModule;
 
    AddContextCommand(pRootContext, L"..",      UpCommand,      IDS_HLP_UP,      IDS_HLP_UP_EX, 0, NULL);
    AddContextCommand(pRootContext, L"?",       NULL,           IDS_HLP_HELP,    IDS_HLP_HELP_EX, 0, NULL);
    AddContextCommand(pRootContext, L"abort",   AbortCommand,   IDS_HLP_ABORT,   IDS_HLP_ABORT_EX, 0, NULL);
    AddContextCommand(pRootContext, L"alias",   AliasCommand,   IDS_HLP_ALIAS,   IDS_HLP_ALIAS_EX, 0, NULL);
    AddContextCommand(pRootContext, L"bye",     ExitCommand,    IDS_HLP_EXIT,    IDS_HLP_EXIT_EX, 0, NULL);
    AddContextCommand(pRootContext, L"commit",  CommitCommand,  IDS_HLP_COMMIT,  IDS_HLP_COMMIT_EX, 0, NULL);
    AddContextCommand(pRootContext, L"dump",    DumpCommand,    IDS_HLP_DUMP,    IDS_HLP_DUMP_EX, 0, NULL);
    AddContextCommand(pRootContext, L"exec",    ExecCommand,    IDS_HLP_EXEC,    IDS_HLP_EXEC_EX, 0, NULL);
    AddContextCommand(pRootContext, L"exit",    ExitCommand,    IDS_HLP_EXIT,    IDS_HLP_EXIT_EX, 0, NULL);
    AddContextCommand(pRootContext, L"help",    NULL,           IDS_HLP_HELP,    IDS_HLP_HELP_EX, 0, NULL);
    AddContextCommand(pRootContext, L"offline", OfflineCommand, IDS_HLP_OFFLINE, IDS_HLP_OFFLINE_EX, 0, NULL);
    AddContextCommand(pRootContext, L"online",  OnlineCommand,  IDS_HLP_ONLINE,  IDS_HLP_ONLINE_EX, 0, NULL);
    AddContextCommand(pRootContext, L"popd",    PopdCommand,    IDS_HLP_POPD,    IDS_HLP_POPD_EX, 0, NULL);
    AddContextCommand(pRootContext, L"pushd",   PushdCommand,   IDS_HLP_PUSHD,   IDS_HLP_PUSHD_EX, 0, NULL);
    AddContextCommand(pRootContext, L"quit",    ExitCommand,    IDS_HLP_EXIT,    IDS_HLP_EXIT_EX, 0, NULL);
    AddContextCommand(pRootContext, L"unalias", UnaliasCommand, IDS_HLP_UNALIAS, IDS_HLP_UNALIAS_EX, 0, NULL);
 
    pGroup = AddCommandGroup(pRootContext, L"add", IDS_HLP_GROUP_ADD, 0, NULL);
    if (pGroup)
    {
        AddGroupCommand(pGroup, L"helper", AddHelperCommand, IDS_HLP_ADD_HELPER, IDS_HLP_ADD_HELPER_EX, CMD_FLAG_LOCAL, NULL);
    }
 
    pGroup = AddCommandGroup(pRootContext, L"delete", IDS_HLP_GROUP_DELETE, 0, NULL);
    if (pGroup)
    {
        AddGroupCommand(pGroup, L"helper", DeleteHelperCommand, IDS_HLP_DEL_HELPER, IDS_HLP_DEL_HELPER_EX, CMD_FLAG_LOCAL, NULL);
    }
 
    pGroup = AddCommandGroup(pRootContext, L"set", IDS_HLP_GROUP_SET, 0, NULL);
    if (pGroup)
    {
        AddGroupCommand(pGroup, L"machine", SetMachineCommand, IDS_HLP_SET_MACHINE, IDS_HLP_SET_MACHINE_EX, CMD_FLAG_ONLINE, NULL);
        AddGroupCommand(pGroup, L"mode",    SetModeCommand,    IDS_HLP_SET_MODE,    IDS_HLP_SET_MODE_EX, 0, NULL);
    }
 
    pGroup = AddCommandGroup(pRootContext, L"show", IDS_HLP_GROUP_SHOW, 0, NULL);
    if (pGroup)
    {
        AddGroupCommand(pGroup, L"alias",  ShowAliasCommand,  IDS_HLP_SHOW_ALIAS,  IDS_HLP_SHOW_ALIAS_EX, 0, NULL);
        AddGroupCommand(pGroup, L"helper", ShowHelperCommand, IDS_HLP_SHOW_HELPER, IDS_HLP_SHOW_HELPER_EX, 0, NULL);
        AddGroupCommand(pGroup, L"mode",   ShowModeCommand,   IDS_HLP_SHOW_MODE,   IDS_HLP_SHOW_MODE_EX, 0, NULL);
    }
 
    pCurrentContext = pRootContext;
 
    return TRUE;
}

Referenced by wmain().

CreateRootHelper()

DWORD CreateRootHelper

(

VOID

)

Definition at line 245 of file helper.c.

{
    PHELPER_ENTRY pHelper;
 
    pHelper = (PHELPER_ENTRY)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(HELPER_ENTRY));
    if (pHelper == NULL)
        return ERROR_OUTOFMEMORY;
 
    /* FIXME: More to initialize here? */
 
    pHelperListHead = pHelper;
    pHelperListTail = pHelper;
 
    return ERROR_SUCCESS;
}

Referenced by wmain().

DeleteHelperCommand()

DWORD WINAPI DeleteHelperCommand	(	LPCWSTR	pwszMachine,
		LPWSTR *	ppwcArguments,
		DWORD	dwCurrentIndex,
		DWORD	dwArgCount,
		DWORD	dwFlags,
		LPCVOID	pvData,
		BOOL *	pbDone
	)

Definition at line 493 of file helper.c.

{
    PDLL_LIST_ENTRY pEntry;
    HKEY hKey;
    DWORD dwError;
 
    DPRINT("DeleteHelper()\n");
 
    if (dwArgCount == 2)
    {
//        ConResPrintf(StdErr, IDS_INVALID_SYNTAX);
//        ConResPrintf(StdErr, IDS_HLP_DEL_HELPER_EX);
        return 1;
    }
 
    pEntry = pDllListHead;
    while (pEntry != NULL)
    {
        if (wcscmp(pEntry->pszShortName, ppwcArguments[2]) == 0)
        {
            DPRINT1("remove %S\n", pEntry->pszShortName);
 
            if (pEntry->pPrev != NULL)
                pEntry->pPrev->pNext = pEntry->pNext;
            if (pEntry->pNext != NULL)
                pEntry->pNext->pPrev = pEntry->pPrev;
            if (pDllListTail == pEntry)
                pDllListTail = pEntry->pPrev;
            if (pDllListHead == pEntry)
                pDllListHead = pEntry->pNext;
            pEntry->pPrev = NULL;
            pEntry->pNext = NULL;
 
            dwError = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
                                    REG_NETSH_PATH,
                                    0,
                                    KEY_WRITE,
                                    &hKey);
            if (dwError == ERROR_SUCCESS)
            {
                RegDeleteValue(hKey, pEntry->pszValueName);
                RegCloseKey(hKey);
            }
 
            FreeHelperDll(pEntry);
 
            return 1;
        }
 
        pEntry = pEntry->pNext;
    }
 
    return ERROR_SUCCESS;
}

Referenced by CreateRootContext().

DestroyAliases()

VOID DestroyAliases

(

VOID

)

Definition at line 103 of file alias.c.

{
    PALIAS_ENTRY pAliasEntry;
 
    while (AliasListHead != NULL)
    {
        pAliasEntry = AliasListHead;
        AliasListHead = AliasListHead->pNext;
 
        HeapFree(GetProcessHeap(), 0, pAliasEntry->pszAliasName);
        HeapFree(GetProcessHeap(), 0, pAliasEntry->pszAlias);
        HeapFree(GetProcessHeap(), 0, pAliasEntry);
    }
 
    AliasListTail = NULL;
}

Referenced by wmain().

FindContextByGuid()

PCONTEXT_ENTRY FindContextByGuid

(

_In_ const GUID *

pGuid

)

FindHelper()

PHELPER_ENTRY FindHelper	(	_In_ const GUID *	pguidHelper,
		_In_ PHELPER_ENTRY	pHelper
	)

Definition at line 360 of file helper.c.

{
    PHELPER_ENTRY pFoundHelper;
 
//    pHelper = pHelperListHead;
    while (pHelper != NULL)
    {
        if (IsEqualGUID(pguidHelper, &pHelper->Attributes.guidHelper))
            return pHelper;
 
        pFoundHelper = FindHelper(pguidHelper, pHelper->pSubHelperHead);
        if (pFoundHelper)
            return pFoundHelper;
 
        pHelper = pHelper->pNext;
    }
 
    return NULL;
}

Referenced by FindHelper(), RegisterContext(), and RegisterHelper().

GetWmiVersionInfo()

HRESULT GetWmiVersionInfo

(

VOID

)

Definition at line 246 of file wmi.c.

{
    BSTR Path = NULL;
    IWbemLocator *Locator = NULL;
    IWbemServices *Services = NULL;
    HRESULT hr;
 
    CoInitialize(NULL);
 
    CoInitializeSecurity(NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_DEFAULT,
                         RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE, NULL);
    hr = CoCreateInstance(&CLSID_WbemLocator, NULL, CLSCTX_INPROC_SERVER,
                          &IID_IWbemLocator, (void **)&Locator);
    if (hr != S_OK)
    {
        DPRINT1("Can't create instance of WbemLocator\n");
        return hr;
    }
 
    Path = SysAllocString(L"ROOT\\CIMV2");
    if (Path == NULL)
    {
        hr = E_OUTOFMEMORY;
        goto done;
    }
 
    hr = IWbemLocator_ConnectServer(Locator, Path, NULL, NULL, NULL, 0, NULL, NULL, &Services);
    if (hr != S_OK)
    {
        DPRINT1("Failed to get IWbemServices interface %08x\n", hr);
        goto done;
    }
 
    hr = CoSetProxyBlanket((IUnknown *)Services, RPC_C_AUTHN_WINNT, RPC_C_AUTHZ_NONE, NULL,
                           RPC_C_AUTHN_LEVEL_CALL, RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE);
    if (hr != S_OK)
    {
        DPRINT1("Failed to set proxy blanket %08x\n", hr);
        goto done;
    }
 
    hr = QueryOperatingSystemInfo(Services);
    if (hr == S_OK)
        hr = QueryProcessorInfo(Services);
 
done:
    if (Path)
        SysFreeString(Path);
 
    if (Services)
        IWbemServices_Release(Services);
 
    if (Locator)
        IWbemLocator_Release(Locator);
 
    CoUninitialize();
 
    return hr;
}

Referenced by wmain().

InitAliases()

VOID InitAliases

(

VOID

)

Definition at line 95 of file alias.c.

{
    AliasListHead = NULL;
    AliasListTail = NULL;
}

Referenced by wmain().

InterpretInteractive()

VOID InterpretInteractive

(

VOID

)

Definition at line 453 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 (g_pszMachine)
            ConPrintf(StdOut, L"[%s] ", g_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

pszFileName

)

Definition at line 381 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().

LoadHelpers()

VOID LoadHelpers

(

VOID

)

Definition at line 263 of file helper.c.

{
    PWSTR pszNameBuffer = NULL;
    PWSTR pszValueBuffer = NULL;
    HKEY hKey;
    DWORD dwValueCount, dwMaxNameLength, dwMaxValueLength;
    DWORD dwNameLength, dwValueLength, dwType;
    DWORD dwIndex, dwError;
 
    DPRINT("LoadHelpers()\n");
 
    dwError = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
                            REG_NETSH_PATH,
                            0,
                            KEY_READ,
                            &hKey);
    if (dwError != ERROR_SUCCESS)
        return;
 
    dwError = RegQueryInfoKeyW(hKey,
                               NULL,
                               NULL,
                               NULL,
                               NULL,
                               NULL,
                               NULL,
                               &dwValueCount,
                               &dwMaxNameLength,
                               &dwMaxValueLength,
                               NULL,
                               NULL);
    if (dwError != ERROR_SUCCESS)
        goto done;
 
    pszNameBuffer = HeapAlloc(GetProcessHeap(), 0,
                              (dwMaxNameLength + 1) * sizeof(WCHAR));
    if (pszNameBuffer == NULL)
        goto done;
 
    pszValueBuffer = HeapAlloc(GetProcessHeap(), 0,
                               dwMaxValueLength + sizeof(WCHAR));
    if (pszValueBuffer == NULL)
        goto done;
 
    for (dwIndex = 0; dwIndex < dwValueCount; dwIndex++)
    {
        dwNameLength = dwMaxNameLength + 1;
        dwValueLength = dwMaxValueLength + sizeof(WCHAR);
        dwError = RegEnumValueW(hKey,
                                dwIndex,
                                pszNameBuffer,
                                &dwNameLength,
                                NULL,
                                &dwType,
                                (PBYTE)pszValueBuffer,
                                &dwValueLength);
        if (dwError != ERROR_SUCCESS)
            break;
 
        DPRINT("Dll: %S --> %S  %lu\n", pszNameBuffer, pszValueBuffer, dwError);
        LoadHelperDll(pszValueBuffer, FALSE);
    }
 
done:
    if (pszValueBuffer)
        HeapFree(GetProcessHeap(), 0, pszValueBuffer);
 
    if (pszNameBuffer)
        HeapFree(GetProcessHeap(), 0, pszNameBuffer);
 
    RegCloseKey(hKey);
 
    StartHelpers(pHelperListHead);
}

Referenced by wmain().

MergeStrings()

LPWSTR MergeStrings	(	_In_ LPWSTR	pszStringArray[],
		_In_ UINT	nCount
	)

Definition at line 53 of file netsh.c.

{
    LPWSTR pszOutString = NULL;
    size_t nLength;
    UINT i;
 
    if ((pszStringArray == NULL) || (nCount == 0))
        return NULL;
 
    nLength = 0;
    for (i = 0; i < nCount; i++)
        nLength += wcslen(pszStringArray[i]);
 
    if (nLength > 0)
        nLength += nCount; /* Space characters and terminating zero */
 
    pszOutString = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, nLength * sizeof(WCHAR));
    if (pszOutString == NULL)
        return NULL;
 
    for (i = 0; i < nCount; i++)
    {
        if (i != 0)
            wcscat(pszOutString, L" ");
        wcscat(pszOutString, pszStringArray[i]);
    }
 
    return pszOutString;
}

Referenced by AliasCommand(), InterpretInteractive(), and wmain().

PrintCommandHelp()

VOID PrintCommandHelp	(	_In_ PCONTEXT_ENTRY	pContext,
		_In_ PCOMMAND_GROUP	pGroup,
		_In_ PCOMMAND_ENTRY	pCommand
	)

Definition at line 337 of file help.c.

{
    LPWSTR pszCommandBuffer;
    DWORD_PTR Args[2];
    DWORD dwLength = 1;
 
    DPRINT("PrintCommandHelp(%p %p %p)\n", pContext, pGroup, pCommand);
 
    if (!CheckOsVersion(pCommand->pfnOsVersionCheck))
        return;
 
    if (((pCommand->dwFlags & CMD_FLAG_LOCAL) && (g_pszMachine != NULL)) ||
        ((pCommand->dwFlags & CMD_FLAG_ONLINE) && (g_bOnline == FALSE)) ||
        (pCommand->dwFlags & CMD_FLAG_HIDDEN))
        return;
 
    dwLength += wcslen(pCommand->pwszCmdToken);
    if (pGroup)
        dwLength += (wcslen(pGroup->pwszCmdGroupToken) + 1);
 
    pszCommandBuffer = HeapAlloc(GetProcessHeap(), 0, dwLength * sizeof(WCHAR));
    if (pszCommandBuffer == NULL)
        return;
 
    _swprintf(pszCommandBuffer, L"%s%s%s",
              (pGroup) ? pGroup->pwszCmdGroupToken : L"",
              (pGroup) ? L" " : L"",
              pCommand->pwszCmdToken);
 
    Args[0] = (DWORD_PTR)pszCommandBuffer;
    Args[1] = (DWORD_PTR)NULL;
 
    ConResMsgPrintfExV(StdOut,
                       pContext->hModule,
                       FORMAT_MESSAGE_ARGUMENT_ARRAY,
                       pCommand->dwCmdHlpToken,
                       LANG_USER_DEFAULT,
                       (va_list *)&Args);
    ConPuts(StdOut, L"\n");
 
    HeapFree(GetProcessHeap(), 0, pszCommandBuffer);
}

Referenced by InterpretCommand().

PrintContextHelp()

VOID PrintContextHelp

(

_In_ PCONTEXT_ENTRY

pContext

)

Definition at line 412 of file help.c.

{
    ConResPrintf(StdOut, IDS_HELP_HEADER);
    PrintContext(pContext);
    PrintSubcontexts(pContext);
    ConResPrintf(StdOut, IDS_HELP_FOOTER);
}

Referenced by InterpretCommand().

PrintGroupHelp()

VOID PrintGroupHelp	(	_In_ PCONTEXT_ENTRY	pContext,
		_In_ LPWSTR	pszGroupName,
		_In_ BOOL	bRecurse
	)

Definition at line 385 of file help.c.

{
    PCOMMAND_GROUP pGroup;
 
    if (bRecurse)
    {
        if (pContext != pRootContext)
            PrintGroupHelp(pContext->pParentContext, pszGroupName, bRecurse);
    }
 
    pGroup = pContext->pGroupListHead;
    while (pGroup != NULL)
    {
        if (_wcsicmp(pszGroupName, pGroup->pwszCmdGroupToken) == 0)
        {
            PrintCurrentContextHeader(pContext);
            PrintShortGroupCommands(pContext, pGroup);
        }
        pGroup = pGroup->pNext;
    }
}

Referenced by InterpretCommand(), and PrintGroupHelp().

RunScript()

DWORD RunScript

(

_In_ LPCWSTR

filename

)

Definition at line 22 of file netsh.c.

{
    WCHAR tmp_string[MAX_STRING_SIZE];
    FILE *script;
    DWORD dwError = ERROR_SUCCESS;
 
    /* Open the file for processing */
    script = _wfopen(filename, L"r");
    if (script == NULL)
    {
        ConResPrintf(StdErr, IDS_OPEN_FAILED, filename);
        return ERROR_FILE_NOT_FOUND;
    }
 
    /* Read and process the script */
    while (fgetws(tmp_string, MAX_STRING_SIZE, script) != NULL)
    {
        dwError = InterpretLine(tmp_string);
        if (dwError != ERROR_SUCCESS)
            break;
    }
 
    /* Close the file */
    fclose(script);
 
    return dwError;
}

Referenced by ExecCommand(), and wmain().

ShowAliasCommand()

DWORD WINAPI ShowAliasCommand	(	LPCWSTR	pwszMachine,
		LPWSTR *	argv,
		DWORD	dwCurrentIndex,
		DWORD	dwArgCount,
		DWORD	dwFlags,
		LPCVOID	pvData,
		BOOL *	pbDone
	)

Definition at line 215 of file alias.c.

{
    DPRINT("ShowAliasCommand()\n");
 
    ShowAliases();
 
    return ERROR_SUCCESS;
}

Referenced by CreateRootContext().

ShowHelperCommand()

DWORD WINAPI ShowHelperCommand	(	LPCWSTR	pwszMachine,
		PWSTR *	ppwcArguments,
		DWORD	dwCurrentIndex,
		DWORD	dwArgCount,
		DWORD	dwFlags,
		LPCVOID	pvData,
		BOOL *	pbDone
	)

Definition at line 614 of file helper.c.

{
    DPRINT("ShowHelperCommand()\n");
 
    ConPrintf(StdOut, L"Helper GUID                             DLL Name          Command\n");
    ConPrintf(StdOut, L"--------------------------------------  ----------------  --------\n");
 
    if (pRootContext == NULL)
        return ERROR_SUCCESS;
 
    PrintHelpers(pHelperListHead, 0);
 
    return ERROR_SUCCESS;
}

Referenced by CreateRootContext().

UnaliasCommand()

DWORD WINAPI UnaliasCommand	(	LPCWSTR	pwszMachine,
		LPWSTR *	argv,
		DWORD	dwCurrentIndex,
		DWORD	dwArgCount,
		DWORD	dwFlags,
		LPCVOID	pvData,
		BOOL *	pbDone
	)

Definition at line 234 of file alias.c.

{
    PALIAS_ENTRY pAliasEntry = NULL;
 
    DPRINT("UnaliasCommand()\n");
 
    if (dwArgCount - dwCurrentIndex != 1)
        return ERROR_INVALID_SYNTAX;
 
    DPRINT("Alias %S\n", argv[dwCurrentIndex]);
 
    pAliasEntry = AliasListHead;
    while (pAliasEntry)
    {
        if (wcscmp(pAliasEntry->pszAliasName, argv[dwCurrentIndex]) == 0)
        {
            DPRINT("Alias found %S\n", argv[dwCurrentIndex]);
 
            if (pAliasEntry->pNext != NULL)
                pAliasEntry->pNext->pPrev = pAliasEntry->pPrev;
            else
                AliasListTail = pAliasEntry->pPrev;
 
            if (pAliasEntry->pPrev != NULL)
                pAliasEntry->pPrev->pNext = pAliasEntry->pNext;
            else
                AliasListHead = pAliasEntry->pNext;
 
            HeapFree(GetProcessHeap(), 0, pAliasEntry->pszAliasName);
            HeapFree(GetProcessHeap(), 0, pAliasEntry->pszAlias);
            HeapFree(GetProcessHeap(), 0, pAliasEntry);
 
            return ERROR_SUCCESS;
        }
 
        pAliasEntry = pAliasEntry->pNext;
    }
 
    ConResPrintf(StdOut, IDS_ALIAS_NOT_FOUND, argv[dwCurrentIndex]);
 
    return ERROR_SUCCESS;
}

Referenced by CreateRootContext().

UnloadHelpers()

VOID UnloadHelpers

(

VOID

)

Definition at line 340 of file helper.c.

{
    PDLL_LIST_ENTRY pEntry;
 
    while (pDllListHead != NULL)
    {
        pEntry = pDllListHead;
        pDllListHead = pEntry->pNext;
 
//        if (pEntry->Attributes.pfnStop)
//            pEntry->Attributes.pfnStop(0);
 
        FreeHelperDll(pEntry);
    }
 
    pDllListTail = NULL;
}

Referenced by wmain().

Variable Documentation

g_bOnline

BOOL g_bOnline

extern

Definition at line 33 of file context.c.

Referenced by ExitCommand(), InterpretCommand(), OfflineCommand(), OnlineCommand(), PrintCommandHelp(), PrintContext(), PrintShortGroupCommands(), PrintSubcontexts(), SetModeCommand(), and ShowModeCommand().

g_hModule

HMODULE g_hModule

extern

Definition at line 17 of file netsh.c.

g_pszMachine

PWSTR g_pszMachine

extern

Definition at line 32 of file context.c.

Referenced by InterpretCommand(), InterpretInteractive(), PrintCommandHelp(), PrintContext(), PrintShortGroupCommands(), PrintSubcontexts(), RegisterContext(), SetMachineCommand(), and wmain().

pCurrentContext

PCONTEXT_ENTRY pCurrentContext

extern

Definition at line 27 of file context.c.

Referenced by CreateRootContext(), DumpCommand(), InterpretCommand(), InterpretInteractive(), PopdCommand(), PrintCurrentContextHeader(), PushdCommand(), and UpCommand().

pHelperListHead

PHELPER_ENTRY pHelperListHead

extern

Definition at line 20 of file helper.c.

Referenced by AddHelperCommand(), CreateRootHelper(), LoadHelpers(), PrintHelpers(), RegisterContext(), RegisterHelper(), and ShowHelperCommand().

pRootContext

PCONTEXT_ENTRY pRootContext

extern

Definition at line 26 of file context.c.

Referenced by AbortCommand(), CommitCommand(), CreateRootContext(), ExitCommand(), FindContextByGuid(), GetContextFullName(), InterpretCommand(), OfflineCommand(), OnlineCommand(), PrintContext(), PrintGroupHelp(), PrintPrompt(), RegisterContext(), SetModeCommand(), ShowHelperCommand(), and UpCommand().

VersionInfoArchitecture

UINT VersionInfoArchitecture

extern

Definition at line 24 of file wmi.c.

Referenced by CheckOsVersion(), and QueryProcessorInfo().

VersionInfoBuildNumber

WCHAR VersionInfoBuildNumber[MAX_PATH]

extern

Definition at line 28 of file wmi.c.

Referenced by CheckOsVersion(), and QueryOperatingSystemInfo().

VersionInfoOsProductSuite

UINT VersionInfoOsProductSuite

extern

Definition at line 25 of file wmi.c.

Referenced by CheckOsVersion(), and QueryOperatingSystemInfo().

VersionInfoOsType

UINT VersionInfoOsType

extern

Definition at line 26 of file wmi.c.

Referenced by CheckOsVersion(), and QueryOperatingSystemInfo().

VersionInfoServicePackMajorVersion

WCHAR VersionInfoServicePackMajorVersion[MAX_PATH]

extern

Definition at line 29 of file wmi.c.

Referenced by CheckOsVersion(), and QueryOperatingSystemInfo().

VersionInfoServicePackMinorVersion

WCHAR VersionInfoServicePackMinorVersion[MAX_PATH]

extern

Definition at line 30 of file wmi.c.

Referenced by CheckOsVersion(), and QueryOperatingSystemInfo().

VersionInfoVersion

WCHAR VersionInfoVersion[MAX_PATH]

extern

Definition at line 27 of file wmi.c.

Referenced by CheckOsVersion(), and QueryOperatingSystemInfo().