252 lines
8 KiB
C
252 lines
8 KiB
C
|
//***************************************************************************
|
||
|
|
// Copyright (c) Microsoft Corporation
|
||
|
|
//
|
||
|
|
// Module Name:
|
||
|
|
// GETMAC.H
|
||
|
|
//
|
||
|
|
// Abstract:
|
||
|
|
// Contains function prototypes and macros.
|
||
|
|
//
|
||
|
|
// Author:
|
||
|
|
// Vasundhara .G
|
||
|
|
//
|
||
|
|
// Revision History:
|
||
|
|
// Vasundhara .G 26-sep-2k : Created It.
|
||
|
|
// Vasundhara .G 31-oct-2k : Modified.
|
||
|
|
// Added macros and some #defines.
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
#ifndef __GETMAC_H
|
||
|
|
#define __GETMAC_H
|
||
|
|
|
||
|
|
// constants / defines / enumerations
|
||
|
|
|
||
|
|
#define MAX_STRING 256
|
||
|
|
#define MAX_OPTIONS 7
|
||
|
|
#define MAX_COLUMNS 5
|
||
|
|
#define USAGE_END 37
|
||
|
|
|
||
|
|
//Command line parser index
|
||
|
|
#define CMD_PARSE_SERVER 0
|
||
|
|
#define CMD_PARSE_USER 1
|
||
|
|
#define CMD_PARSE_PWD 2
|
||
|
|
#define CMD_PARSE_FMT 3
|
||
|
|
#define CMD_PARSE_USG 4
|
||
|
|
#define CMD_PARSE_HRD 5
|
||
|
|
#define CMD_PARSE_VER 6
|
||
|
|
|
||
|
|
//show results index
|
||
|
|
#define SH_RES_HOST 0
|
||
|
|
#define SH_RES_CON 1
|
||
|
|
#define SH_RES_TYPE 2
|
||
|
|
#define SH_RES_MAC 3
|
||
|
|
#define SH_RES_TRAN 4
|
||
|
|
|
||
|
|
//wmi registry key value
|
||
|
|
#define WMI_HKEY_CLASSES_ROOT 2147483648
|
||
|
|
#define WMI_HKEY_CURRENT_USER 2147483649
|
||
|
|
#define WMI_HKEY_LOCAL_MACHINE 2147483650
|
||
|
|
#define WMI_HKEY_USERS 2147483651
|
||
|
|
#define WMI_HKEY_CURRENT_CONFIG 2147482652
|
||
|
|
|
||
|
|
// Error constants
|
||
|
|
#define ERROR_USER_WITH_NOSERVER GetResString( IDS_USER_NMACHINE )
|
||
|
|
#define ERROR_SERVER_WITH_NOPASSWORD GetResString( IDS_SERVER_NPASSWORD )
|
||
|
|
#define ERROR_NULL_SERVER GetResString( IDS_NULL_SERVER )
|
||
|
|
#define ERROR_NULL_USER GetResString( IDS_NULL_USER )
|
||
|
|
#define ERROR_INVALID_HEADER_OPTION GetResString( IDS_INVALID_OPTIONS )
|
||
|
|
#define ERROR_TYPE_REQUEST GetResString( IDS_TYPE_REQUEST )
|
||
|
|
#define ERROR_STRING GetResString( IDS_ERROR_STRING )
|
||
|
|
#define ERROR_VERSION_MISMATCH GetResString( IDS_ERROR_VERSION_MISMATCH )
|
||
|
|
#define ERROR_NOT_RESPONDING GetResString( IDS_NOT_RESPONDING )
|
||
|
|
#define ERROR_NO_MACHINE GetResString( IDS_NO_MACHINE )
|
||
|
|
#define ERROR_INVALID_MACHINE GetResString( IDS_INVALID_MACHINE )
|
||
|
|
#define ERROR_WKST_NOT_FOUND GetResString( IDS_WKST_NOT_FOUND )
|
||
|
|
|
||
|
|
//warning message
|
||
|
|
#define IGNORE_LOCALCREDENTIALS GetResString( IDS_IGNORE_LOCALCREDENTIALS )
|
||
|
|
#define WARNING_STRING GetResString( IDS_WARNING_STRING )
|
||
|
|
|
||
|
|
//info message
|
||
|
|
#define NO_NETWORK_ADAPTERS GetResString( IDS_NO_NETWORK_ADAPTERS )
|
||
|
|
#define NO_NETWOK_PROTOCOLS GetResString( IDS_NO_NETWOK_PROTOCOLS )
|
||
|
|
|
||
|
|
//show results column length
|
||
|
|
#define HOST_NAME_WIDTH AsLong(GetResString( IDS_HOST_NAME_WIDTH ),10 )
|
||
|
|
#define CONN_NAME_WIDTH AsLong(GetResString( IDS_CONN_NAME_WIDTH ),10 )
|
||
|
|
#define ADAPT_TYPE_WIDTH AsLong(GetResString( IDS_ADAPT_TYPE_WIDTH ),10 )
|
||
|
|
#define MAC_ADDR_WIDTH AsLong(GetResString( IDS_MAC_ADDR_WIDTH ),10 )
|
||
|
|
#define TRANS_NAME_WIDTH AsLong(GetResString( IDS_TRANS_NAME_WIDTH ),10 )
|
||
|
|
|
||
|
|
//output headers strings
|
||
|
|
#define RES_HOST_NAME GetResString( RES_HOST )
|
||
|
|
#define RES_CONNECTION_NAME GetResString( RES_CONNECTION )
|
||
|
|
#define RES_ADAPTER_TYPE GetResString( RES_ADAPTER )
|
||
|
|
#define RES_MAC_ADDRESS GetResString( RES_ADDRESS )
|
||
|
|
#define RES_TRANS_NAME GetResString( RES_TRANSNAME )
|
||
|
|
|
||
|
|
//general
|
||
|
|
#define NOT_AVAILABLE GetResString( IDS_NOT_AVAILABLE )
|
||
|
|
#define DISABLED GetResString( IDS_DISABLED )
|
||
|
|
#define HYPHEN_STRING GetResString( IDS_HYPHEN_STRING )
|
||
|
|
#define COLON_STRING GetResString( IDS_COLON_STRING )
|
||
|
|
#define NEW_LINE GetResString( IDS_NEW_LINE )
|
||
|
|
#define FORMAT_TYPES GetResString( IDS_FORMAT_TYPES )
|
||
|
|
|
||
|
|
//registry key names
|
||
|
|
#define DEFAULT_ADDRESS _T( "000000000000" )
|
||
|
|
#define CONNECTION_KEYPATH _T( "SYSTEM\\CurrentControlSet\\Control\\Network\\{4D36E972-E325-11CE-BFC1-08002BE10318}\\" )
|
||
|
|
#define TRANSPORT_KEYPATH _T( "SYSTEM\\CurrentControlSet\\Services\\" )
|
||
|
|
#define LINKAGE _T( "\\Linkage" )
|
||
|
|
#define ROUTE _T( "route" )
|
||
|
|
#define EXPORT _T( "Export" )
|
||
|
|
#define CONNECTION_STRING _T( "\\Connection" )
|
||
|
|
#define REG_NAME _T( "name" )
|
||
|
|
#define NETBIOS _T( "NetBIOS" )
|
||
|
|
|
||
|
|
|
||
|
|
//command line options
|
||
|
|
#define CMDOPTION_SERVER _T( "s" )
|
||
|
|
#define CMDOPTION_USER _T( "u" )
|
||
|
|
#define CMDOPTION_PASSWORD _T( "p" )
|
||
|
|
#define CMDOPTION_FORMAT _T( "fo" )
|
||
|
|
#define CMDOPTION_USAGE _T( "?" )
|
||
|
|
#define CMDOPTION_HEADER _T( "nh" )
|
||
|
|
#define CMDOPTION_VERBOSE _T( "v" )
|
||
|
|
#define TOKEN_BACKSLASH2 _T( "\\\\" )
|
||
|
|
#define TOKEN_BACKSLASH3 _T( "\\\\\\" )
|
||
|
|
|
||
|
|
//wmi classes and property names
|
||
|
|
#define HYPHEN_CHAR L'-'
|
||
|
|
#define COLON_CHAR L':'
|
||
|
|
#define BACK_SLASHS L"\\\\"
|
||
|
|
#define BACK_SLASH L"\\"
|
||
|
|
#define NETCONNECTION_STATUS L"NetConnectionStatus"
|
||
|
|
#define ADAPTER_MACADDR L"MACAddress"
|
||
|
|
#define NETCONNECTION_ID L"NetConnectionID"
|
||
|
|
#define DEVICE_ID L"DeviceID"
|
||
|
|
#define NAME L"Name"
|
||
|
|
#define HOST_NAME L"SystemName"
|
||
|
|
#define SETTING_ID L"SettingID"
|
||
|
|
#define NETWORK_ADAPTER_CLASS L"Win32_NetworkAdapter"
|
||
|
|
#define NETWORK_ADAPTER_CONFIG_CLASS L"Win32_NetworkAdapterConfiguration"
|
||
|
|
#define NETWORK_PROTOCOL L"Win32_NetworkProtocol"
|
||
|
|
#define CLASS_CIMV2_Win32_OperatingSystem L"Win32_OperatingSystem"
|
||
|
|
#define QUERY_LANGUAGE L"WQL"
|
||
|
|
#define ASSOCIATOR_QUERY _T( "ASSOCIATORS OF {Win32_NetworkAdapter.DeviceID=\"%s\"} WHERE ResultClass=Win32_NetworkAdapterConfiguration" )
|
||
|
|
|
||
|
|
#define SUCCESS 0
|
||
|
|
#define FAILURE 1
|
||
|
|
|
||
|
|
#define WMI_NAMESPACE_CIMV2 L"root\\cimv2"
|
||
|
|
#define WMI_NAMESPACE_DEFAULT L"root\\default"
|
||
|
|
#define WMI_CLAUSE_AND L"AND"
|
||
|
|
#define WMI_CLAUSE_OR L"OR"
|
||
|
|
#define WMI_CLAUSE_WHERE L"WHERE"
|
||
|
|
#define WMI_REGISTRY L"StdRegProv"
|
||
|
|
#define WMI_REGISTRY_M_STRINGVALUE L"GetStringValue"
|
||
|
|
#define WMI_REGISTRY_M_MSTRINGVALUE L"GetMultiStringValue"
|
||
|
|
#define WMI_REGISTRY_IN_HDEFKEY L"hDefKey"
|
||
|
|
#define WMI_REGISTRY_IN_SUBKEY L"sSubKeyName"
|
||
|
|
#define WMI_REGISTRY_IN_VALUENAME L"sValueName"
|
||
|
|
#define WMI_REGISTRY_OUT_VALUE L"sValue"
|
||
|
|
#define WMI_REGISTRY_OUT_RETURNVALUE L"ReturnValue"
|
||
|
|
#define CAPTION L"Caption"
|
||
|
|
|
||
|
|
|
||
|
|
//macro for freeing, deleting, releasing memory which has been allocated using calloc
|
||
|
|
|
||
|
|
#define FREESTRING( pStr ) \
|
||
|
|
if ( pStr ) \
|
||
|
|
{ \
|
||
|
|
free( pStr ); \
|
||
|
|
pStr = NULL; \
|
||
|
|
}
|
||
|
|
|
||
|
|
#define DELETESTRING( pStr ) \
|
||
|
|
if ( pStr ) \
|
||
|
|
{ \
|
||
|
|
delete[] pStr; \
|
||
|
|
pStr = NULL; \
|
||
|
|
}
|
||
|
|
|
||
|
|
#define SAFERELEASE( pIObj ) \
|
||
|
|
if ( pIObj != NULL ) \
|
||
|
|
{ \
|
||
|
|
pIObj->Release(); \
|
||
|
|
pIObj = NULL; \
|
||
|
|
}
|
||
|
|
|
||
|
|
#define SAFEBSTRRELEASE( pIObj ) \
|
||
|
|
if ( pIObj != NULL ) \
|
||
|
|
{ \
|
||
|
|
SysFreeString( pIObj ); \
|
||
|
|
pIObj = NULL; \
|
||
|
|
}
|
||
|
|
|
||
|
|
#define ONFAILTHROWERROR(hResult) \
|
||
|
|
if (FAILED(hResult)) \
|
||
|
|
{ \
|
||
|
|
_com_issue_error(hResult); \
|
||
|
|
}
|
||
|
|
|
||
|
|
#define SAFE_RELEASE( interfacepointer ) \
|
||
|
|
if ( (interfacepointer) != NULL ) \
|
||
|
|
{ \
|
||
|
|
(interfacepointer)->Release(); \
|
||
|
|
(interfacepointer) = NULL; \
|
||
|
|
} \
|
||
|
|
1
|
||
|
|
|
||
|
|
#define SAFE_EXECUTE( statement ) \
|
||
|
|
hRes = statement; \
|
||
|
|
if ( FAILED( hRes ) ) \
|
||
|
|
{ \
|
||
|
|
_com_issue_error( hRes ); \
|
||
|
|
} \
|
||
|
|
1
|
||
|
|
|
||
|
|
//function prototype
|
||
|
|
|
||
|
|
BOOL ConnectWmi( IWbemLocator *pLocator,
|
||
|
|
IWbemServices **ppServices,
|
||
|
|
LPCWSTR pwszServer,
|
||
|
|
LPCWSTR pwszUser,
|
||
|
|
LPCWSTR pwszPassword,
|
||
|
|
COAUTHIDENTITY **ppAuthIdentity,
|
||
|
|
BOOL bCheckWithNullPwd = FALSE,
|
||
|
|
LPCWSTR pwszNamespace = WMI_NAMESPACE_CIMV2,
|
||
|
|
HRESULT *phRes = NULL,
|
||
|
|
BOOL *pbLocalSystem = NULL );
|
||
|
|
|
||
|
|
BOOL ConnectWmiEx( IWbemLocator *pLocator,
|
||
|
|
IWbemServices **ppServices,
|
||
|
|
LPCWSTR pwszServer,
|
||
|
|
CHString &strUserName,
|
||
|
|
CHString &strPassword,
|
||
|
|
COAUTHIDENTITY **ppAuthIdentity,
|
||
|
|
BOOL bNeedPassword = FALSE,
|
||
|
|
LPCWSTR pszNamespace = WMI_NAMESPACE_CIMV2,
|
||
|
|
BOOL *pbLocalSystem = NULL );
|
||
|
|
|
||
|
|
BOOL IsValidServerEx( LPCWSTR pwszServer,
|
||
|
|
BOOL &bLocalSystem );
|
||
|
|
|
||
|
|
HRESULT SetInterfaceSecurity( IUnknown *pInterface,
|
||
|
|
COAUTHIDENTITY *pAuthIdentity );
|
||
|
|
|
||
|
|
VOID WINAPI WbemFreeAuthIdentity( COAUTHIDENTITY **ppAuthIdentity );
|
||
|
|
|
||
|
|
VOID WMISaveError( HRESULT hResError );
|
||
|
|
|
||
|
|
DWORD GetTargetVersionEx( IWbemServices* pWbemServices,
|
||
|
|
COAUTHIDENTITY* pAuthIdentity );
|
||
|
|
|
||
|
|
// inline functions
|
||
|
|
inline VOID WMISaveError( _com_error &e )
|
||
|
|
{
|
||
|
|
WMISaveError( e.Error() );
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif // __GETMAC_H
|