692 lines
20 KiB
C++
692 lines
20 KiB
C++
|
#include "pch.h"
|
||
|
#pragma hdrstop
|
||
|
#include "diag.h"
|
||
|
|
||
|
// Command strings
|
||
|
//
|
||
|
#define SZ_CMD_SHOW_BINDINGS L"showbindings"
|
||
|
#define SZ_CMD_SHOW_COMPONENTS L"showcomponents"
|
||
|
#define SZ_CMD_SHOW_STACK_TABLE L"showstacktable"
|
||
|
#define SZ_CMD_SHOW_LAN_ADAPTER_PNPIDS L"showlanpnpids"
|
||
|
#define SZ_CMD_ADD_COMPONENT L"addcomponent"
|
||
|
#define SZ_CMD_REMOVE_COMPONENT L"removecomponent"
|
||
|
#define SZ_CMD_UPDATE_COMPONENT L"updatecomponent"
|
||
|
#define SZ_CMD_REMOVE_REFS L"removerefs"
|
||
|
#define SZ_CMD_ENABLE_BINDING L"enablebinding"
|
||
|
#define SZ_CMD_DISABLE_BINDING L"disablebinding"
|
||
|
#define SZ_CMD_MOVE_BINDING L"movebinding"
|
||
|
#define SZ_CMD_WRITE_BINDINGS L"writebindings"
|
||
|
#define SZ_CMD_SET_WANORDER L"setwanorder"
|
||
|
#define SZ_CMD_FULL_DIAGNOSTIC L"full"
|
||
|
#define SZ_CMD_CLEANUP L"cleanup"
|
||
|
#define SZ_CMD_ADD_REMOVE_STRESS L"arstress"
|
||
|
|
||
|
#define SZ_CMD_SHOW_LAN_CONNECTIONS L"showlan"
|
||
|
#define SZ_CMD_SHOW_LAN_DETAILS L"showlandetails"
|
||
|
#define SZ_CMD_LAN_CHANGE_STATE L"changelanstate"
|
||
|
#define SZ_CMD_SHOW_ALL_DEVICES L"showalldevices"
|
||
|
|
||
|
#define SZ_PARAM_CONNECT L"connect"
|
||
|
#define SZ_PARAM_DISCONNECT L"disconnect"
|
||
|
|
||
|
// Parameter strings for SZ_CMD_SHOW_BINDINGS
|
||
|
//
|
||
|
#define SZ_PARAM_BELOW L"below"
|
||
|
#define SZ_PARAM_INVOLVING L"involving"
|
||
|
#define SZ_PARAM_UPPER L"upper"
|
||
|
#define SZ_PARAM_DISABLED L"disabled"
|
||
|
|
||
|
// Parameter strings for SZ_CMD_ADD_COMPONENT
|
||
|
//
|
||
|
#define SZ_PARAM_NET L"net"
|
||
|
#define SZ_PARAM_INFRARED L"irda"
|
||
|
#define SZ_PARAM_TRANS L"trans"
|
||
|
#define SZ_PARAM_CLIENT L"client"
|
||
|
#define SZ_PARAM_SERVICE L"service"
|
||
|
|
||
|
// Parameter strings for SZ_CMD_MOVE_BINDING
|
||
|
//
|
||
|
#define SZ_PARAM_BEFORE L"before"
|
||
|
#define SZ_PARAM_AFTER L"after"
|
||
|
|
||
|
// Parameter strings for SZ_CMD_SET_WANORDER
|
||
|
//
|
||
|
#define SZ_PARAM_FIRST L"first"
|
||
|
#define SZ_PARAM_LAST L"last"
|
||
|
|
||
|
// Parameter strings for SZ_CMD_FULL_DIAGNOSTIC
|
||
|
//
|
||
|
#define SZ_PARAM_LEAK_CHECK L"leakcheck"
|
||
|
|
||
|
// Aliases used when refering to components
|
||
|
//
|
||
|
#define SZ_ALIAS_ALL L"all"
|
||
|
|
||
|
VOID
|
||
|
Usage (
|
||
|
IN PCTSTR pszProgramName,
|
||
|
IN COMMAND Command)
|
||
|
{
|
||
|
switch (Command)
|
||
|
{
|
||
|
case CMD_SHOW_BINDINGS:
|
||
|
g_pDiagCtx->Printf (ttidNcDiag,
|
||
|
"\n"
|
||
|
"%S %S [below | involving | upper | disabled] [all | <infid> | <pnpid>]\n"
|
||
|
"\n"
|
||
|
" below : show bindings below component\n"
|
||
|
" involving : show bindings involving component\n"
|
||
|
" upper : show adapter's upper bindings\n"
|
||
|
" disabled : show disabled bindings\n"
|
||
|
"\n"
|
||
|
" to specify the component:\n"
|
||
|
" all : matches all components\n"
|
||
|
" <infid> : matches component with specified Inf Id\n"
|
||
|
" <pnpid> : matches component with specified PpP Id\n"
|
||
|
"\n",
|
||
|
pszProgramName,
|
||
|
SZ_CMD_SHOW_BINDINGS);
|
||
|
break;
|
||
|
|
||
|
case CMD_SHOW_COMPONENTS:
|
||
|
break;
|
||
|
|
||
|
case CMD_SHOW_STACK_TABLE:
|
||
|
break;
|
||
|
|
||
|
case CMD_SHOW_LAN_ADAPTER_PNPIDS:
|
||
|
break;
|
||
|
|
||
|
case CMD_ADD_COMPONENT:
|
||
|
g_pDiagCtx->Printf (ttidNcDiag,
|
||
|
"\n"
|
||
|
"%S %S <class> <infid>\n"
|
||
|
"\n"
|
||
|
" <class> : one of 'net', 'irda', 'trans', 'client', 'service'\n"
|
||
|
" <infid> : the Inf ID of the component to add\n"
|
||
|
"\n",
|
||
|
pszProgramName,
|
||
|
SZ_CMD_ADD_COMPONENT);
|
||
|
break;
|
||
|
|
||
|
case CMD_REMOVE_COMPONENT:
|
||
|
g_pDiagCtx->Printf (ttidNcDiag,
|
||
|
"\n"
|
||
|
"%S %S [<infid> | <pnpid>]\n"
|
||
|
"\n"
|
||
|
" <infid> : matches component with specified Inf Id\n"
|
||
|
" <pnpid> : matches component with specified PnP Id\n"
|
||
|
"\n",
|
||
|
pszProgramName,
|
||
|
SZ_CMD_REMOVE_COMPONENT);
|
||
|
break;
|
||
|
|
||
|
case CMD_UPDATE_COMPONENT:
|
||
|
g_pDiagCtx->Printf (ttidNcDiag,
|
||
|
"\n"
|
||
|
"%S %S [<infid> | <pnpid>]\n"
|
||
|
"\n"
|
||
|
" <infid> : matches component with specified Inf Id\n"
|
||
|
" <pnpid> : matches component with specified PnP Id\n"
|
||
|
"\n",
|
||
|
pszProgramName,
|
||
|
SZ_CMD_UPDATE_COMPONENT);
|
||
|
break;
|
||
|
|
||
|
case CMD_REMOVE_REFS:
|
||
|
g_pDiagCtx->Printf (ttidNcDiag,
|
||
|
"\n"
|
||
|
"%S %S <infid>\n"
|
||
|
"\n"
|
||
|
" <infid> : matches component with specified Inf Id\n"
|
||
|
"\n",
|
||
|
pszProgramName,
|
||
|
SZ_CMD_REMOVE_REFS);
|
||
|
break;
|
||
|
|
||
|
|
||
|
case CMD_ENABLE_BINDING:
|
||
|
case CMD_DISABLE_BINDING:
|
||
|
g_pDiagCtx->Printf (ttidNcDiag,
|
||
|
"\n"
|
||
|
"%S %S <bindpath>\n"
|
||
|
"\n"
|
||
|
" <bindpath> : bindpath as specified like output of 'showbindings'\n"
|
||
|
"\n",
|
||
|
pszProgramName,
|
||
|
(CMD_ENABLE_BINDING == Command)
|
||
|
? SZ_CMD_ENABLE_BINDING
|
||
|
: SZ_CMD_DISABLE_BINDING);
|
||
|
break;
|
||
|
|
||
|
case CMD_MOVE_BINDING:
|
||
|
g_pDiagCtx->Printf (ttidNcDiag,
|
||
|
"\n"
|
||
|
"%S %S <srcbindpath> [before | after] <dstbindpath>\n"
|
||
|
"\n"
|
||
|
" <srcbindpath> : bindpath as specified like output of 'showbindings'\n"
|
||
|
" <dstbindpath> : bindpath as specified like output of 'showbindings'\n"
|
||
|
" or 'null' to make srcbindpath first if 'before'\n"
|
||
|
" is specified, or last if 'after' is specified\n"
|
||
|
"\n",
|
||
|
pszProgramName,
|
||
|
SZ_CMD_MOVE_BINDING);
|
||
|
break;
|
||
|
|
||
|
case CMD_SET_WANORDER:
|
||
|
g_pDiagCtx->Printf (ttidNcDiag,
|
||
|
"\n"
|
||
|
"%S %S [first | last]\n"
|
||
|
"\n"
|
||
|
" first : order WAN adapters before LAN adapters\n"
|
||
|
" last : order WAN adapters after LAN adapters\n"
|
||
|
"\n",
|
||
|
pszProgramName,
|
||
|
SZ_CMD_SET_WANORDER);
|
||
|
break;
|
||
|
|
||
|
case CMD_FULL_DIAGNOSTIC:
|
||
|
break;
|
||
|
|
||
|
case CMD_ADD_REMOVE_STRESS:
|
||
|
break;
|
||
|
|
||
|
case CMD_SHOW_LAN_CONNECTIONS:
|
||
|
break;
|
||
|
|
||
|
case CMD_SHOW_ALL_DEVICES:
|
||
|
break;
|
||
|
|
||
|
case CMD_SHOW_LAN_DETAILS:
|
||
|
g_pDiagCtx->Printf (ttidNcDiag,
|
||
|
"\n"
|
||
|
"%S %S <lan connection name>\n"
|
||
|
"\n"
|
||
|
" <lan connection name> : friendly name of LAN connection\n"
|
||
|
" (note: use quotes if name contains spaces\n"
|
||
|
"\n",
|
||
|
pszProgramName,
|
||
|
SZ_CMD_SHOW_LAN_DETAILS);
|
||
|
break;
|
||
|
|
||
|
case CMD_LAN_CHANGE_STATE:
|
||
|
g_pDiagCtx->Printf (ttidNcDiag,
|
||
|
"\n"
|
||
|
"%S %S <lan connection name> [connect | disconnect]\n"
|
||
|
"\n"
|
||
|
" <lan connection name> : friendly name of LAN connection\n"
|
||
|
" (note: use quotes if name contains spaces\n\n"
|
||
|
" <connect> : connects the given lan connection\n"
|
||
|
" <disconnect> : disconnects the given lan connection\n"
|
||
|
"\n",
|
||
|
pszProgramName,
|
||
|
SZ_CMD_LAN_CHANGE_STATE);
|
||
|
break;
|
||
|
|
||
|
default:
|
||
|
g_pDiagCtx->Printf (ttidNcDiag,
|
||
|
"\n"
|
||
|
"Network Configuration Diagnostic\n"
|
||
|
" View, manipulate, or test network configuration.\n"
|
||
|
"\n"
|
||
|
"%S [options]\n"
|
||
|
" %-15S - Show the current bindings\n"
|
||
|
" %-15S - Show the currently installed components\n"
|
||
|
" %-15S - Show the current stack table\n"
|
||
|
" %-15S - Show PnP ids of LAN adapters\n"
|
||
|
" %-15S - Add a component\n"
|
||
|
" %-15S - Remove a component\n"
|
||
|
" %-15S - Update a component\n"
|
||
|
" %-15S - Enable a binding\n"
|
||
|
" %-15S - Disable a binding\n"
|
||
|
" %-15S - Change binding order\n"
|
||
|
" %-15S - Rewrite all bindings\n"
|
||
|
" %-15S - Change WAN adapter order\n"
|
||
|
" %-15S - Perform a full (non-destructive) diagnostic\n"
|
||
|
" %-15S - Perform Add/Remove component stress\n"
|
||
|
" %-15S - Show the current LAN connections\n"
|
||
|
" %-15S - Show details for a specific LAN connection\n"
|
||
|
" %-15S - Connects or disconnects a specific LAN connection\n"
|
||
|
" %-15S - Lists all devices\n"
|
||
|
"\n\n",
|
||
|
pszProgramName,
|
||
|
SZ_CMD_SHOW_BINDINGS,
|
||
|
SZ_CMD_SHOW_COMPONENTS,
|
||
|
SZ_CMD_SHOW_STACK_TABLE,
|
||
|
SZ_CMD_SHOW_LAN_ADAPTER_PNPIDS,
|
||
|
SZ_CMD_ADD_COMPONENT,
|
||
|
SZ_CMD_REMOVE_COMPONENT,
|
||
|
SZ_CMD_UPDATE_COMPONENT,
|
||
|
SZ_CMD_ENABLE_BINDING,
|
||
|
SZ_CMD_DISABLE_BINDING,
|
||
|
SZ_CMD_MOVE_BINDING,
|
||
|
SZ_CMD_WRITE_BINDINGS,
|
||
|
SZ_CMD_SET_WANORDER,
|
||
|
SZ_CMD_FULL_DIAGNOSTIC,
|
||
|
SZ_CMD_ADD_REMOVE_STRESS,
|
||
|
SZ_CMD_SHOW_LAN_CONNECTIONS,
|
||
|
SZ_CMD_SHOW_LAN_DETAILS,
|
||
|
SZ_CMD_LAN_CHANGE_STATE,
|
||
|
SZ_CMD_SHOW_ALL_DEVICES);
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#define NthArgIsPresent(_i) (_i < argc)
|
||
|
#define NthArgIs(_i, _sz) ((_i < argc) && (0 == _wcsicmp(argv[_i], _sz)))
|
||
|
|
||
|
EXTERN_C
|
||
|
VOID
|
||
|
__cdecl
|
||
|
wmain (
|
||
|
IN INT argc,
|
||
|
IN PCWSTR argv[])
|
||
|
{
|
||
|
CDiagContext DiagCtx;
|
||
|
DIAG_OPTIONS Options;
|
||
|
INT iArg;
|
||
|
|
||
|
DiagCtx.SetFlags (DF_SHOW_CONSOLE_OUTPUT);
|
||
|
g_pDiagCtx = &DiagCtx;
|
||
|
|
||
|
ZeroMemory (&Options, sizeof(Options));
|
||
|
Options.pDiagCtx = g_pDiagCtx;
|
||
|
Options.Command = CMD_INVALID;
|
||
|
|
||
|
if (argc < 2)
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
iArg = 1;
|
||
|
|
||
|
if (NthArgIs (iArg, SZ_CMD_SHOW_BINDINGS))
|
||
|
{
|
||
|
Options.Command = CMD_SHOW_BINDINGS;
|
||
|
iArg++;
|
||
|
|
||
|
if (NthArgIs (iArg, SZ_PARAM_BELOW))
|
||
|
{
|
||
|
Options.ShowBindParam = SHOW_BELOW;
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_PARAM_INVOLVING))
|
||
|
{
|
||
|
Options.ShowBindParam = SHOW_INVOLVING;
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_PARAM_UPPER))
|
||
|
{
|
||
|
Options.ShowBindParam = SHOW_UPPER;
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_PARAM_DISABLED))
|
||
|
{
|
||
|
Options.ShowBindParam = SHOW_DISABLED;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if (SHOW_DISABLED != Options.ShowBindParam)
|
||
|
{
|
||
|
iArg++;
|
||
|
if (NthArgIs (iArg, SZ_ALIAS_ALL))
|
||
|
{
|
||
|
Options.CompSpecifier.Type = CST_ALL;
|
||
|
}
|
||
|
else if (NthArgIsPresent (iArg))
|
||
|
{
|
||
|
Options.CompSpecifier.Type = CST_BY_NAME;
|
||
|
Options.CompSpecifier.pszInfOrPnpId = argv[iArg];
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_SHOW_COMPONENTS))
|
||
|
{
|
||
|
Options.Command = CMD_SHOW_COMPONENTS;
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_SHOW_STACK_TABLE))
|
||
|
{
|
||
|
Options.Command = CMD_SHOW_STACK_TABLE;
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_SHOW_LAN_ADAPTER_PNPIDS))
|
||
|
{
|
||
|
Options.Command = CMD_SHOW_LAN_ADAPTER_PNPIDS;
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_ADD_COMPONENT))
|
||
|
{
|
||
|
Options.Command = CMD_ADD_COMPONENT;
|
||
|
iArg++;
|
||
|
|
||
|
if (NthArgIs (iArg, SZ_PARAM_NET))
|
||
|
{
|
||
|
Options.ClassGuid = GUID_DEVCLASS_NET;
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_PARAM_INFRARED))
|
||
|
{
|
||
|
Options.ClassGuid = GUID_DEVCLASS_INFRARED;
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_PARAM_TRANS))
|
||
|
{
|
||
|
Options.ClassGuid = GUID_DEVCLASS_NETTRANS;
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_PARAM_CLIENT))
|
||
|
{
|
||
|
Options.ClassGuid = GUID_DEVCLASS_NETCLIENT;
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_PARAM_SERVICE))
|
||
|
{
|
||
|
Options.ClassGuid = GUID_DEVCLASS_NETSERVICE;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
iArg++;
|
||
|
if (NthArgIsPresent (iArg))
|
||
|
{
|
||
|
Options.pszInfId = argv[iArg];
|
||
|
|
||
|
iArg++;
|
||
|
if (NthArgIs (iArg, SZ_PARAM_LEAK_CHECK))
|
||
|
{
|
||
|
Options.fLeakCheck = TRUE;
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_REMOVE_COMPONENT))
|
||
|
{
|
||
|
Options.Command = CMD_REMOVE_COMPONENT;
|
||
|
|
||
|
iArg++;
|
||
|
if (NthArgIsPresent (iArg))
|
||
|
{
|
||
|
Options.pszInfId = argv[iArg];
|
||
|
|
||
|
iArg++;
|
||
|
if (NthArgIs (iArg, SZ_PARAM_LEAK_CHECK))
|
||
|
{
|
||
|
Options.fLeakCheck = TRUE;
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_UPDATE_COMPONENT))
|
||
|
{
|
||
|
Options.Command = CMD_UPDATE_COMPONENT;
|
||
|
|
||
|
iArg++;
|
||
|
if (NthArgIsPresent (iArg))
|
||
|
{
|
||
|
Options.pszInfId = argv[iArg];
|
||
|
|
||
|
iArg++;
|
||
|
if (NthArgIs (iArg, SZ_PARAM_LEAK_CHECK))
|
||
|
{
|
||
|
Options.fLeakCheck = TRUE;
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_REMOVE_REFS))
|
||
|
{
|
||
|
Options.Command = CMD_REMOVE_REFS;
|
||
|
|
||
|
iArg++;
|
||
|
if (NthArgIsPresent (iArg))
|
||
|
{
|
||
|
Options.pszInfId = argv[iArg];
|
||
|
|
||
|
iArg++;
|
||
|
if (NthArgIs (iArg, SZ_PARAM_LEAK_CHECK))
|
||
|
{
|
||
|
Options.fLeakCheck = TRUE;
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_ENABLE_BINDING))
|
||
|
{
|
||
|
Options.Command = CMD_ENABLE_BINDING;
|
||
|
iArg++;
|
||
|
if (NthArgIsPresent (iArg))
|
||
|
{
|
||
|
Options.pszBindPath = argv[iArg];
|
||
|
|
||
|
iArg++;
|
||
|
if (NthArgIs (iArg, SZ_PARAM_LEAK_CHECK))
|
||
|
{
|
||
|
Options.fLeakCheck = TRUE;
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_DISABLE_BINDING))
|
||
|
{
|
||
|
Options.Command = CMD_DISABLE_BINDING;
|
||
|
iArg++;
|
||
|
if (NthArgIsPresent (iArg))
|
||
|
{
|
||
|
Options.pszBindPath = argv[iArg];
|
||
|
|
||
|
iArg++;
|
||
|
if (NthArgIs (iArg, SZ_PARAM_LEAK_CHECK))
|
||
|
{
|
||
|
Options.fLeakCheck = TRUE;
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_MOVE_BINDING))
|
||
|
{
|
||
|
Options.Command = CMD_MOVE_BINDING;
|
||
|
iArg++;
|
||
|
if (NthArgIsPresent (iArg) &&
|
||
|
NthArgIsPresent (iArg+1) &&
|
||
|
NthArgIsPresent (iArg+2))
|
||
|
{
|
||
|
Options.pszBindPath = argv[iArg];
|
||
|
Options.pszOtherBindPath = argv[iArg+2];
|
||
|
|
||
|
if (NthArgIs (iArg+1, SZ_PARAM_BEFORE))
|
||
|
{
|
||
|
Options.fMoveBefore = TRUE;
|
||
|
}
|
||
|
else if (NthArgIs (iArg+1, SZ_PARAM_AFTER))
|
||
|
{
|
||
|
Options.fMoveBefore = FALSE;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_WRITE_BINDINGS))
|
||
|
{
|
||
|
Options.Command = CMD_WRITE_BINDINGS;
|
||
|
|
||
|
iArg++;
|
||
|
if (NthArgIs (iArg, SZ_PARAM_LEAK_CHECK))
|
||
|
{
|
||
|
Options.fLeakCheck = TRUE;
|
||
|
}
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_SET_WANORDER))
|
||
|
{
|
||
|
Options.Command = CMD_SET_WANORDER;
|
||
|
iArg++;
|
||
|
if (NthArgIs (iArg+1, SZ_PARAM_FIRST))
|
||
|
{
|
||
|
Options.fWanAdaptersFirst = TRUE;
|
||
|
}
|
||
|
else if (NthArgIs (iArg+1, SZ_PARAM_LAST))
|
||
|
{
|
||
|
Options.fWanAdaptersFirst = FALSE;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_FULL_DIAGNOSTIC))
|
||
|
{
|
||
|
Options.Command = CMD_FULL_DIAGNOSTIC;
|
||
|
|
||
|
iArg++;
|
||
|
if (NthArgIs (iArg, SZ_PARAM_LEAK_CHECK))
|
||
|
{
|
||
|
Options.fLeakCheck = TRUE;
|
||
|
}
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_CLEANUP))
|
||
|
{
|
||
|
Options.Command = CMD_CLEANUP;
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_ADD_REMOVE_STRESS))
|
||
|
{
|
||
|
Options.Command = CMD_ADD_REMOVE_STRESS;
|
||
|
|
||
|
iArg++;
|
||
|
if (NthArgIs (iArg, SZ_PARAM_LEAK_CHECK))
|
||
|
{
|
||
|
Options.fLeakCheck = TRUE;
|
||
|
}
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_SHOW_LAN_CONNECTIONS))
|
||
|
{
|
||
|
Options.Command = CMD_SHOW_LAN_CONNECTIONS;
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_SHOW_ALL_DEVICES))
|
||
|
{
|
||
|
Options.Command = CMD_SHOW_ALL_DEVICES;
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_SHOW_LAN_DETAILS))
|
||
|
{
|
||
|
Options.Command = CMD_SHOW_LAN_DETAILS;
|
||
|
iArg++;
|
||
|
|
||
|
if (NthArgIsPresent (iArg))
|
||
|
{
|
||
|
Options.szLanConnection = argv[iArg];
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
else if (NthArgIs (iArg, SZ_CMD_LAN_CHANGE_STATE))
|
||
|
{
|
||
|
Options.Command = CMD_LAN_CHANGE_STATE;
|
||
|
iArg++;
|
||
|
|
||
|
if (NthArgIsPresent (iArg))
|
||
|
{
|
||
|
Options.szLanConnection = argv[iArg];
|
||
|
|
||
|
if (NthArgIs (iArg + 1, SZ_PARAM_CONNECT))
|
||
|
{
|
||
|
Options.fConnect = TRUE;
|
||
|
}
|
||
|
else if (NthArgIs (iArg + 1, SZ_PARAM_DISCONNECT))
|
||
|
{
|
||
|
Options.fConnect = FALSE;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Usage (argv[0], Options.Command);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
// If we're doing leak checking, preload some DLLs that are delayloaded.
|
||
|
// If we don't do this, we see a buncn of "leaks" associated with
|
||
|
// the loader bringing in DLLs in the middle of our tests.
|
||
|
//
|
||
|
if (Options.fLeakCheck)
|
||
|
{
|
||
|
LoadLibraryW (L"mswsock.dll");
|
||
|
LoadLibraryW (L"comctl32.dll");
|
||
|
LoadLibraryW (L"comdlg32.dll");
|
||
|
LoadLibraryW (L"mprapi.dll");
|
||
|
LoadLibraryW (L"mswsock.dll");
|
||
|
LoadLibraryW (L"netapi32.dll");
|
||
|
LoadLibraryW (L"rtutils.dll");
|
||
|
LoadLibraryW (L"setupapi.dll");
|
||
|
LoadLibraryW (L"shell32.dll");
|
||
|
LoadLibraryW (L"userenv.dll");
|
||
|
LoadLibraryW (L"winspool.drv");
|
||
|
LoadLibraryW (L"ws2_32.dll");
|
||
|
}
|
||
|
|
||
|
HRESULT hr = CoInitializeEx (
|
||
|
NULL,
|
||
|
COINIT_DISABLE_OLE1DDE | COINIT_APARTMENTTHREADED);
|
||
|
|
||
|
if (FAILED(hr))
|
||
|
{
|
||
|
g_pDiagCtx->Printf (ttidNcDiag,
|
||
|
"Problem 0x%08x initializing COM library", hr);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
// Call each of the external command processors
|
||
|
//
|
||
|
NetCfgDiagFromCommandArgs (&Options);
|
||
|
NetManDiagFromCommandArgs (&Options);
|
||
|
|
||
|
g_pDiagCtx->Printf (ttidNcDiag, "\n");
|
||
|
}
|