mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1650 lines
33 KiB
1650 lines
33 KiB
/*++
|
|
|
|
Copyright (C) Microsoft Corporation, 1998 - 1999
|
|
|
|
Module Name:
|
|
|
|
location.cxx
|
|
|
|
Abstract:
|
|
|
|
This module provides all the functions for determining the
|
|
machines current physical location.
|
|
|
|
Author:
|
|
|
|
Steve Kiraly (SteveKi) 13-July-1998
|
|
|
|
Revision History:
|
|
|
|
Steve Kiraly (SteveKi) 13-July-1998 Genesis
|
|
|
|
--*/
|
|
|
|
#include <precomp.hxx>
|
|
#pragma hdrstop
|
|
|
|
#include "dsinterf.hxx"
|
|
#include "persist.hxx"
|
|
#include "physloc.hxx"
|
|
|
|
const IN_ADDR c_LoopBackAddress = { 127,0,0,1 };
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
TPhysicalLocation
|
|
|
|
Description:
|
|
|
|
TPhysicalLocation constructor.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
Nothing.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
TPhysicalLocation::
|
|
TPhysicalLocation(
|
|
VOID
|
|
) : m_fInitalized( FALSE ),
|
|
m_eDiscoveryType( kDiscoveryTypeUnknown ),
|
|
m_IpHlpApi( gszIpHlpApiLibrary ),
|
|
m_GetIpAddrTable( NULL ),
|
|
m_SecExt( gszSecurityLibrary ),
|
|
m_GetComputerObjectName( NULL ),
|
|
m_GetUserNameEx( NULL ),
|
|
m_NetApi( gszNetApiLibrary ),
|
|
m_DsAddressToSiteNames( NULL ),
|
|
m_NetApiBufferFree( NULL ),
|
|
m_WinSock( gszWinSockLibrary ),
|
|
m_inet_ntoa( NULL )
|
|
{
|
|
DBGMSG( DBG_TRACE, ( "TPhysicalLocation::ctor.\n" ) );
|
|
|
|
//
|
|
// Check if the Ip helper library was loaded.
|
|
//
|
|
if (m_IpHlpApi.bValid())
|
|
{
|
|
m_GetIpAddrTable = reinterpret_cast<pfGetIpAddrTable>( m_IpHlpApi.pfnGetProc( "GetIpAddrTable" ) );
|
|
}
|
|
|
|
//
|
|
// Check if the security library was loaded.
|
|
//
|
|
if (m_SecExt.bValid())
|
|
{
|
|
m_GetComputerObjectName = reinterpret_cast<pfGetComputerObjectName>( m_SecExt.pfnGetProc( "GetComputerObjectNameW" ) );
|
|
m_GetUserNameEx = reinterpret_cast<pfGetUserNameEx>( m_SecExt.pfnGetProc( "GetUserNameExW" ) );
|
|
}
|
|
|
|
//
|
|
// Check if the netapi library was loaded.
|
|
//
|
|
if (m_NetApi.bValid())
|
|
{
|
|
m_DsAddressToSiteNames = reinterpret_cast<pfDsAddressToSiteNames>( m_NetApi.pfnGetProc( "DsAddressToSiteNamesExW" ) );
|
|
m_NetApiBufferFree = reinterpret_cast<pfNetApiBufferFree>( m_NetApi.pfnGetProc( "NetApiBufferFree" ) );
|
|
}
|
|
|
|
//
|
|
// Check if the winsock library was loaded.
|
|
//
|
|
if (m_WinSock.bValid())
|
|
{
|
|
m_inet_ntoa = reinterpret_cast<LPFN_INET_NTOA>( m_WinSock.pfnGetProc( "inet_ntoa" ) );
|
|
}
|
|
|
|
//
|
|
// All the function pointers must be valid for this class to be in
|
|
// the initialized state.
|
|
//
|
|
if (m_GetIpAddrTable && m_GetComputerObjectName && m_GetUserNameEx && m_DsAddressToSiteNames && m_NetApiBufferFree && m_inet_ntoa)
|
|
{
|
|
m_fInitalized = TRUE;
|
|
}
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
TPhysicalLocation
|
|
|
|
Description:
|
|
|
|
TPhysicalLocation destructor.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
Nothing.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
TPhysicalLocation::
|
|
~TPhysicalLocation(
|
|
VOID
|
|
)
|
|
{
|
|
DBGMSG( DBG_TRACE, ( "TPhysicalLocation::dtor.\n" ) );
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
bValid
|
|
|
|
Description:
|
|
|
|
This routine indicates if the TPhysicalLocation class is
|
|
in a consistent state, i.e. usable.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
TRUE usable state, FALSE not usable.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
bValid(
|
|
VOID
|
|
) const
|
|
{
|
|
return m_fInitalized;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
Discover
|
|
|
|
Description:
|
|
|
|
The discover routine is where all the work is done. We attempt
|
|
to fetch the physical location string from various sources in a
|
|
pre defined order. If any of the steps completes the search is
|
|
terminated. The current order is to look in the HKLM policy key,
|
|
this machines location property in the DS
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
TRUE usable state, FALSE not usable.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
Discover(
|
|
VOID
|
|
)
|
|
{
|
|
DBGMSG( DBG_TRACE, ( "TPhysicalLocation::Discover.\n" ) );
|
|
|
|
//
|
|
// Invalidate the current location information.
|
|
//
|
|
Invalidate();
|
|
|
|
//
|
|
// Read the group policy setting if it exists.
|
|
//
|
|
if (ReadGroupPolicyLocationSetting( m_strLocation ))
|
|
{
|
|
m_eDiscoveryType = kDiscoveryTypePolicy;
|
|
}
|
|
|
|
//
|
|
// If a DS is available and the group policy did not have any
|
|
// location information then continue looking for location information.
|
|
//
|
|
if (m_Ds.bIsDsAvailable() && m_eDiscoveryType == kDiscoveryTypeUnknown)
|
|
{
|
|
if (ReadMachinesLocationProperty( m_strLocation ))
|
|
{
|
|
m_eDiscoveryType = kDiscoveryTypeMachine;
|
|
}
|
|
else if (ReadSubnetLocationProperty( m_strLocation ))
|
|
{
|
|
m_eDiscoveryType = kDiscoveryTypeSubnet;
|
|
}
|
|
else if (ReadSiteLocationProperty( m_strLocation ))
|
|
{
|
|
m_eDiscoveryType = kDiscoveryTypeSite;
|
|
}
|
|
}
|
|
|
|
return m_eDiscoveryType != kDiscoveryTypeUnknown;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
GetExact
|
|
|
|
Description:
|
|
|
|
Returns the excact location string found by a call to the discover method.
|
|
|
|
Arguments:
|
|
|
|
strLocation - reference to a string object where to return the string.
|
|
|
|
Return Value:
|
|
|
|
TRUE search string returned, FALSE error occurred.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
GetExact(
|
|
IN TString &strLocation
|
|
) const
|
|
{
|
|
DBGMSG( DBG_TRACE, ( "TPhysicalLocation::GetExact.\n" ) );
|
|
|
|
return strLocation.bUpdate( m_strLocation );
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
GetSearch
|
|
|
|
Description:
|
|
|
|
Returns a string that is valid for a search. It widens the scope if the
|
|
current location string is one fetched for this machine.
|
|
|
|
Arguments:
|
|
|
|
strLocation - reference to a string object where to return the search string.
|
|
|
|
Return Value:
|
|
|
|
TRUE search string returned, FALSE error occurred.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
GetSearch(
|
|
IN TString &strLocation
|
|
) const
|
|
{
|
|
DBGMSG( DBG_TRACE, ( "TPhysicalLocation::GetSearch.\n" ) );
|
|
|
|
TStatusB bStatus;
|
|
|
|
if (m_eDiscoveryType == kDiscoveryTypeMachine || m_eDiscoveryType == kDiscoveryTypePolicy)
|
|
{
|
|
bStatus DBGCHK = WidenScope( m_strLocation, 1, strLocation );
|
|
|
|
//
|
|
// If the scope could not be widen then use the current location string.
|
|
//
|
|
if (!bStatus)
|
|
{
|
|
bStatus DBGCHK = strLocation.bUpdate( m_strLocation );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
bStatus DBGCHK = strLocation.bUpdate( m_strLocation );
|
|
}
|
|
|
|
if( bStatus )
|
|
{
|
|
UINT uLen = strLocation.uLen();
|
|
if( uLen && gchSeparator != static_cast<LPCTSTR>(strLocation)[uLen-1] )
|
|
{
|
|
//
|
|
// Put the final slash after successfully
|
|
// widening the location scope
|
|
//
|
|
static const TCHAR szSepStr[] = { gchSeparator };
|
|
bStatus DBGCHK = strLocation.bCat( szSepStr );
|
|
}
|
|
}
|
|
|
|
return bStatus;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
Invalidate
|
|
|
|
Description:
|
|
|
|
Invalidates the location string. After this call the location string
|
|
is not valid until the disover method is called.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
Nothing.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
VOID
|
|
TPhysicalLocation::
|
|
Invalidate(
|
|
VOID
|
|
)
|
|
{
|
|
m_eDiscoveryType = kDiscoveryTypeUnknown;
|
|
m_strLocation.bUpdate( NULL );
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
ReadGroupPolicyLocationSetting
|
|
|
|
Description:
|
|
|
|
Reads the location string from the local registry that was written
|
|
by the group policy editor.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
TRUE the location string was read, FALSE error occured or string not available.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
ReadGroupPolicyLocationSetting(
|
|
IN OUT TString &strLocationX
|
|
)
|
|
{
|
|
DBGMSG( DBG_TRACE, ( "TPhysicalLocation::ReadGroupPolicyLocationSetting.\n" ) );
|
|
|
|
//
|
|
// Open the registry key where the physical location is stored
|
|
// by the group policy code.
|
|
//
|
|
TPersist Reg( gszGroupPolicyPhysicalLocationPath, TPersist::kOpen|TPersist::kRead, HKEY_LOCAL_MACHINE );
|
|
|
|
TStatusB bStatus;
|
|
|
|
//
|
|
// Was the key opened, i.e. does it exist?
|
|
//
|
|
bStatus DBGCHK = Reg.bValid();
|
|
|
|
if (bStatus)
|
|
{
|
|
TString strLocation;
|
|
|
|
//
|
|
// Read the location string from the registry.
|
|
//
|
|
bStatus DBGCHK = Reg.bRead( gszGroupPolicyPhysicalLocationKey, strLocation );
|
|
|
|
//
|
|
// If the string was read and it is not blank then make a local copy of the
|
|
// location string and we are done.
|
|
//
|
|
if (bStatus && !strLocation.bEmpty())
|
|
{
|
|
bStatus DBGCHK = strLocationX.bUpdate( strLocation );
|
|
}
|
|
}
|
|
|
|
return bStatus;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
ReadUserLocationProperty
|
|
|
|
Description:
|
|
|
|
Reads the location string from the current users object in the
|
|
DS.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
TRUE success, FALSE error occurred.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
ReadUserLocationProperty(
|
|
IN OUT TString &strLocation
|
|
)
|
|
{
|
|
DBGMSG( DBG_TRACE, ( "TPhysicalLocation::ReadUserLocationProperty.\n" ) );
|
|
|
|
TCHAR szName[INTERNET_MAX_HOST_NAME_LENGTH+1];
|
|
DWORD dwSize = COUNTOF( szName );
|
|
TStatusB bStatus;
|
|
|
|
bStatus DBGCHK = m_GetUserNameEx(NameFullyQualifiedDN, szName, &dwSize);
|
|
|
|
if (bStatus)
|
|
{
|
|
TString strUserPath;
|
|
TString strLDAPPrefix;
|
|
|
|
bStatus DBGCHK = m_Ds.GetLDAPPrefixPerUser( strLDAPPrefix ) &&
|
|
strUserPath.bCat( strLDAPPrefix ) &&
|
|
strUserPath.bCat( szName );
|
|
|
|
if (bStatus)
|
|
{
|
|
bStatus DBGCHK = m_Ds.ReadStringProperty( strUserPath, gszUserLocationPropertyName, strLocation );
|
|
}
|
|
}
|
|
|
|
return bStatus;
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
ReadMachinesLocationProperty
|
|
|
|
Description:
|
|
|
|
Reads the location string from the machine object in the DS.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
TRUE success, FALSE error occurred.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
ReadMachinesLocationProperty(
|
|
IN OUT TString &strLocation
|
|
)
|
|
{
|
|
DBGMSG( DBG_TRACE, ( "TPhysicalLocation::ReadMachinesLocationProperty.\n" ) );
|
|
|
|
LPTSTR pszName = NULL;
|
|
DWORD dwSize = 0;
|
|
TStatusB bStatus;
|
|
|
|
bStatus DBGCHK = m_GetComputerObjectName(NameFullyQualifiedDN, NULL, &dwSize);
|
|
|
|
if( bStatus && dwSize )
|
|
{
|
|
pszName = new TCHAR[dwSize+1];
|
|
|
|
if( pszName )
|
|
{
|
|
bStatus DBGCHK = m_GetComputerObjectName(NameFullyQualifiedDN, pszName, &dwSize);
|
|
|
|
if (bStatus)
|
|
{
|
|
TString strComputerPath;
|
|
TString strLDAPPrefix;
|
|
|
|
bStatus DBGCHK = m_Ds.GetLDAPPrefix( strLDAPPrefix ) &&
|
|
strComputerPath.bCat( strLDAPPrefix ) &&
|
|
strComputerPath.bCat( pszName );
|
|
|
|
if (bStatus)
|
|
{
|
|
bStatus DBGCHK = m_Ds.ReadStringProperty( strComputerPath, gszMachineLocationPropertyName, strLocation );
|
|
}
|
|
}
|
|
|
|
delete [] pszName;
|
|
}
|
|
}
|
|
|
|
if( bStatus )
|
|
{
|
|
DBGMSG( DBG_TRACE, ( "TPhysicalLocation::ReadMachinesLocationProperty " TSTR ".\n", pszName ) );
|
|
}
|
|
|
|
return bStatus;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
ReadSubnetLocationProperty
|
|
|
|
Description:
|
|
|
|
Reads the location string from the subnet objet for this
|
|
machine. This routine searches the list of IP address IP masks
|
|
and finds a subnet object in the DS that closly matches this
|
|
machine.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
TRUE success, FALSE error occurred.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
ReadSubnetLocationProperty(
|
|
IN OUT TString &strLocation
|
|
)
|
|
{
|
|
DBGMSG( DBG_TRACE, ( "TPhysicalLocation::ReadSubnetLocationProperty.\n" ) );
|
|
|
|
TString strSubnetPath;
|
|
TString strLDAPPrefix;
|
|
TSubnets Subnets;
|
|
TStatusB bStatus;
|
|
|
|
bStatus DBGCHK = GetSubnetObjectNames( Subnets );
|
|
|
|
if (bStatus)
|
|
{
|
|
for( UINT i = 0; i < Subnets.NumEntries(); i++ )
|
|
{
|
|
bStatus DBGCHK = m_Ds.GetLDAPPrefix( strLDAPPrefix ) &&
|
|
strSubnetPath.bUpdate( strLDAPPrefix ) &&
|
|
strSubnetPath.bCat( Subnets.Table(i) );
|
|
|
|
if (bStatus)
|
|
{
|
|
bStatus DBGCHK = m_Ds.ReadStringProperty( strSubnetPath, gszSubnetLocationPropertyName, strLocation );
|
|
|
|
if (bStatus && !strLocation.bEmpty())
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return bStatus;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
ReadSiteLocationProperty
|
|
|
|
Description:
|
|
|
|
Reads the location string from the site object. This routine
|
|
searches the list of IP address and finds a site nearest this
|
|
machines ip adress. All of the searching code actually occurrs
|
|
in the DSAddressToSiteNames api.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
TRUE success, FALSE error occurred.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
ReadSiteLocationProperty(
|
|
IN OUT TString &strLocation
|
|
)
|
|
{
|
|
DBGMSG( DBG_TRACE, ( "TPhysicalLocation::ReadSiteLocationProperty.\n" ) );
|
|
|
|
PMIB_IPADDRTABLE pAddrTable = NULL;
|
|
TStatusB bStatus;
|
|
TString strSiteName;
|
|
TString strSubnetName;
|
|
TString strSubnetPath;
|
|
IN_ADDR IpNetAddr;
|
|
|
|
//
|
|
// Get the DS name.
|
|
//
|
|
bStatus DBGCHK = GetIpAddressTable( &pAddrTable );
|
|
|
|
if (bStatus)
|
|
{
|
|
bStatus DBGNOCHK = FALSE;
|
|
|
|
//
|
|
// Look for a site that has a location string that corresponds
|
|
// to the the first ipaddress in the list of ip addresses.
|
|
//
|
|
for (UINT i = 0; i < pAddrTable->dwNumEntries; i++)
|
|
{
|
|
//
|
|
// Skip the loopback or unassigned interfaces
|
|
//
|
|
if ((pAddrTable->table[i].dwAddr != 0) && (pAddrTable->table[i].dwAddr != c_LoopBackAddress.s_addr))
|
|
{
|
|
IpNetAddr.s_addr = pAddrTable->table[i].dwAddr;
|
|
|
|
DBGMSG( DBG_TRACE, ( "IP Address %s\n", m_inet_ntoa( IpNetAddr ) ) );
|
|
|
|
//
|
|
// Attempt to translate this ip address to either a site path or subnet path.
|
|
//
|
|
bStatus DBGNOCHK = AddrToSite( pAddrTable->table[i].dwAddr, strSiteName, strSubnetName );
|
|
|
|
if (bStatus)
|
|
{
|
|
//
|
|
// If a near match subnet was found use this first.
|
|
//
|
|
if (!strSubnetName.bEmpty())
|
|
{
|
|
bStatus DBGCHK = GetSubnetLocationString( strSubnetName, strLocation );
|
|
}
|
|
|
|
//
|
|
// If the location string was not found on the subnet then attempt to
|
|
// get the location on the site object.
|
|
//
|
|
if (!strSiteName.bEmpty() && strLocation.bEmpty())
|
|
{
|
|
bStatus DBGCHK = GetSiteLocationString( strSiteName, strLocation );
|
|
}
|
|
|
|
//
|
|
// If a location string was found exit the loop, the first ipaddress wins.
|
|
//
|
|
if (strLocation.bEmpty())
|
|
{
|
|
bStatus DBGNOCHK = FALSE;
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
delete [] pAddrTable;
|
|
|
|
return bStatus;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
GetSubnetLocationString
|
|
|
|
Description:
|
|
|
|
Reads the location string from the subnet object. This routine
|
|
converts the subnet name to an adsi path then reads the location
|
|
property fron this subnet object.
|
|
|
|
Arguments:
|
|
|
|
pszSubnetName - pointer string which contains the subnet name.
|
|
strSiteLocation - reference to string object where to return the subnet
|
|
location string.
|
|
|
|
Return Value:
|
|
|
|
TRUE success, FALSE error occurred.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
GetSubnetLocationString(
|
|
IN LPCTSTR pszSubnetName,
|
|
IN OUT TString &strSubnetLocation
|
|
)
|
|
{
|
|
TString strSubnetPath;
|
|
TString strConfig;
|
|
TString strLDAPPrefix;
|
|
TStatusB bStatus;
|
|
TCHAR szSubnetName[MAX_PATH];
|
|
|
|
//
|
|
// Sites are located in the configuration container.
|
|
//
|
|
bStatus DBGCHK = m_Ds.GetConfigurationContainer( strConfig );
|
|
|
|
if (bStatus)
|
|
{
|
|
//
|
|
// Escape the / in the subnet name.
|
|
//
|
|
for ( LPTSTR p = szSubnetName; pszSubnetName && *pszSubnetName; )
|
|
{
|
|
if (*pszSubnetName == _T('/'))
|
|
{
|
|
*p++ = _T('\\');
|
|
}
|
|
|
|
*p++ = *pszSubnetName++;
|
|
}
|
|
|
|
*p = NULL;
|
|
|
|
//
|
|
// Build the site adsi object path.
|
|
//
|
|
bStatus DBGCHK = m_Ds.GetLDAPPrefix( strLDAPPrefix ) &&
|
|
strSubnetPath.bCat( strLDAPPrefix ) &&
|
|
strSubnetPath.bCat( gszCNEquals ) &&
|
|
strSubnetPath.bCat( szSubnetName ) &&
|
|
strSubnetPath.bCat( gszComma ) &&
|
|
strSubnetPath.bCat( gszSubnetContainter) &&
|
|
strSubnetPath.bCat( gszComma ) &&
|
|
strSubnetPath.bCat( strConfig );
|
|
|
|
DBGMSG( DBG_TRACE, ( "Subnet Path " TSTR ".\n", (LPCTSTR)strSubnetPath ) );
|
|
|
|
if (bStatus)
|
|
{
|
|
//
|
|
// Read the site location property.
|
|
//
|
|
bStatus DBGCHK = m_Ds.ReadStringProperty( strSubnetPath, gszSubnetLocationPropertyName, strSubnetLocation );
|
|
}
|
|
}
|
|
|
|
return bStatus;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
GetSiteLocationString
|
|
|
|
Description:
|
|
|
|
Reads the location string from the site object. This routine
|
|
converts the site name to an adsi path then reads the location
|
|
property fron this site object.
|
|
|
|
Arguments:
|
|
|
|
pszSiteName - pointer string which contains the site name.
|
|
strSiteLocation - reference to string object where to return the site
|
|
location string.
|
|
|
|
Return Value:
|
|
|
|
TRUE success, FALSE error occurred.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
GetSiteLocationString(
|
|
IN LPCTSTR pszSiteName,
|
|
IN OUT TString &strSiteLocation
|
|
)
|
|
{
|
|
TString strConfig;
|
|
TString strSitePath;
|
|
TString strLDAPPrefix;
|
|
TStatusB bStatus;
|
|
|
|
//
|
|
// Sites are located in the configuration container.
|
|
//
|
|
bStatus DBGCHK = m_Ds.GetConfigurationContainer( strConfig );
|
|
|
|
if (bStatus)
|
|
{
|
|
//
|
|
// Build the site adsi object path.
|
|
//
|
|
bStatus DBGCHK = m_Ds.GetLDAPPrefix( strLDAPPrefix ) &&
|
|
strSitePath.bCat( strLDAPPrefix ) &&
|
|
strSitePath.bCat( gszCNEquals ) &&
|
|
strSitePath.bCat( pszSiteName ) &&
|
|
strSitePath.bCat( gszComma ) &&
|
|
strSitePath.bCat( gszSitesContainter) &&
|
|
strSitePath.bCat( gszComma ) &&
|
|
strSitePath.bCat( strConfig );
|
|
|
|
DBGMSG( DBG_TRACE, ( "Site Path " TSTR ".\n", (LPCTSTR)strSitePath ) );
|
|
|
|
if (bStatus)
|
|
{
|
|
//
|
|
// Read the site location property.
|
|
//
|
|
bStatus DBGCHK = m_Ds.ReadStringProperty( strSitePath, gszSiteLocationPropertyName, strSiteLocation );
|
|
}
|
|
}
|
|
|
|
return bStatus;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
TPhysicalLocation::vTrimSlash
|
|
|
|
Description:
|
|
|
|
Removes trailing slashes from the given string
|
|
|
|
Arguments:
|
|
|
|
strLocation - string to shorten
|
|
|
|
Return Value:
|
|
|
|
None
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
VOID
|
|
TPhysicalLocation::
|
|
vTrimSlash (
|
|
IN OUT TString &strLocation
|
|
)
|
|
{
|
|
UINT uLen;
|
|
LPTSTR szTrimSlash;
|
|
|
|
uLen = strLocation.uLen();
|
|
|
|
while (uLen && *(static_cast<LPCTSTR>(strLocation)+uLen-1) == gchSeparator)
|
|
{
|
|
uLen--;
|
|
};
|
|
|
|
if (uLen)
|
|
{
|
|
szTrimSlash = new TCHAR[uLen+1];
|
|
|
|
if (szTrimSlash)
|
|
{
|
|
_tcsncpy (szTrimSlash, strLocation, uLen);
|
|
|
|
*(szTrimSlash+uLen) = 0;
|
|
|
|
strLocation.bUpdate (szTrimSlash);
|
|
|
|
delete [] szTrimSlash;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
strLocation.bUpdate (gszNULL);
|
|
}
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
AddrToSite
|
|
|
|
Description:
|
|
|
|
Converts a single ip address to a site name.
|
|
|
|
Arguments:
|
|
|
|
dwAddr - the IP address.
|
|
strSiteName - return the site name here.
|
|
strSubnetName - return the subnet name here.
|
|
|
|
Return Value:
|
|
|
|
TRUE success, FALSE error occurred.
|
|
|
|
Notes:
|
|
|
|
DsAddressToSiteNamesEx only returns the nearest subnet when there are multiple
|
|
sites defined in the DS.
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
AddrToSite(
|
|
IN DWORD dwAddr,
|
|
IN TString &strSiteName,
|
|
IN TString &strSubnetName
|
|
)
|
|
{
|
|
DWORD dwStatus = ERROR_SUCCESS;
|
|
SOCKET_ADDRESS SockAddr = {0};
|
|
SOCKADDR_IN SockAddrIn = {0};
|
|
PWSTR *ppSiteNames = NULL;
|
|
PWSTR *ppSubnetNames = NULL;
|
|
TStatusB bStatus;
|
|
|
|
//
|
|
// Create a socket from an IP address.
|
|
//
|
|
SockAddr.iSockaddrLength = sizeof(SOCKADDR_IN);
|
|
SockAddr.lpSockaddr = (LPSOCKADDR)&SockAddrIn;
|
|
|
|
SockAddrIn.sin_family = AF_INET;
|
|
SockAddrIn.sin_port = 0;
|
|
|
|
memcpy( &SockAddrIn.sin_addr, &dwAddr, sizeof(dwAddr) );
|
|
|
|
//
|
|
// If a no zero address was given then attempt to find a site
|
|
// that this address belongs to.
|
|
//
|
|
if (dwAddr)
|
|
{
|
|
dwStatus = m_DsAddressToSiteNames( NULL, 1, &SockAddr, &ppSiteNames, &ppSubnetNames );
|
|
|
|
if (dwStatus == ERROR_SUCCESS)
|
|
{
|
|
if (ppSiteNames)
|
|
{
|
|
DBGMSG( DBG_TRACE, ( "SiteName " TSTR "\n", DBGSTR( ppSiteNames[0] ) ) );
|
|
|
|
bStatus DBGCHK = strSiteName.bUpdate( ppSiteNames[0] );
|
|
|
|
m_NetApiBufferFree( ppSiteNames );
|
|
}
|
|
|
|
if (ppSubnetNames)
|
|
{
|
|
DBGMSG( DBG_TRACE, ( "SubnetName " TSTR "\n", DBGSTR( ppSubnetNames[0] ) ) );
|
|
|
|
bStatus DBGCHK = strSubnetName.bUpdate( ppSubnetNames[0] );
|
|
|
|
m_NetApiBufferFree( ppSubnetNames );
|
|
}
|
|
}
|
|
}
|
|
|
|
return dwStatus == ERROR_SUCCESS;
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
GetIpAddressTable
|
|
|
|
Description:
|
|
|
|
Returns an array of Ip address for this machine. The caller
|
|
must free the array using the delete operator.
|
|
|
|
Arguments:
|
|
|
|
ppAddrTable - Pointer were to return a pointer to the array of
|
|
ip addresses.
|
|
|
|
Return Value:
|
|
|
|
TRUE success, FALSE error occurred.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
GetIpAddressTable(
|
|
PMIB_IPADDRTABLE *ppAddrTable
|
|
)
|
|
{
|
|
SPLASSERT( ppAddrTable );
|
|
|
|
DWORD dwTableSize = 0;
|
|
PMIB_IPADDRTABLE pAddrTable = NULL;
|
|
DWORD dwStatus = ERROR_SUCCESS;
|
|
BOOL bRet = FALSE;
|
|
|
|
dwStatus = m_GetIpAddrTable(NULL, &dwTableSize, FALSE);
|
|
|
|
if (dwStatus == ERROR_INSUFFICIENT_BUFFER)
|
|
{
|
|
pAddrTable = (PMIB_IPADDRTABLE)new BYTE[dwTableSize];
|
|
|
|
if (pAddrTable)
|
|
{
|
|
dwStatus = m_GetIpAddrTable(pAddrTable, &dwTableSize, FALSE);
|
|
|
|
if (dwStatus == ERROR_SUCCESS)
|
|
{
|
|
*ppAddrTable = pAddrTable;
|
|
bRet = TRUE;
|
|
}
|
|
else
|
|
{
|
|
delete [] pAddrTable;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// new has failed
|
|
SetLastError(ERROR_OUTOFMEMORY);
|
|
}
|
|
}
|
|
|
|
return bRet;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
GetSubnetObjectNames
|
|
|
|
Description:
|
|
|
|
Returns a list of subnet objects for this machine. Since the machine
|
|
may have multiple ip address the same goes for the subnets.
|
|
|
|
Arguments:
|
|
|
|
Subnets - Refrence to subnet Class that will contain on return a
|
|
list of subnet objects.
|
|
|
|
Return Value:
|
|
|
|
TRUE subnet object returned, FALSE error occurred.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
GetSubnetObjectNames(
|
|
IN TSubnets &Subnets
|
|
)
|
|
{
|
|
TString strConfig;
|
|
TStatusB bStatus;
|
|
|
|
//
|
|
// Get the path to the configuration container.
|
|
//
|
|
bStatus DBGCHK = m_Ds.GetConfigurationContainer( strConfig );
|
|
|
|
if (bStatus)
|
|
{
|
|
TSubnets Subnets2;
|
|
TString strSubnetObject;
|
|
|
|
//
|
|
// Get the subnet names.
|
|
//
|
|
bStatus DBGCHK = GetSubnetNames( Subnets2 );
|
|
|
|
//
|
|
// Transform the subnet names into subnet object names.
|
|
//
|
|
for( UINT i = 0; i < Subnets2.NumEntries(); i++ )
|
|
{
|
|
bStatus DBGCHK = strSubnetObject.bUpdate( gszCNEquals ) &&
|
|
strSubnetObject.bCat( Subnets2.Table(i) ) &&
|
|
strSubnetObject.bCat( gszComma ) &&
|
|
strSubnetObject.bCat( gszSubnetContainter ) &&
|
|
strSubnetObject.bCat( gszComma ) &&
|
|
strSubnetObject.bCat( strConfig );
|
|
if (bStatus)
|
|
{
|
|
DBGMSG( DBG_TRACE, ( "SubnetObject " TSTR ".\n", (LPCTSTR)strSubnetObject ) );
|
|
|
|
//
|
|
// Add the subnet object entry.
|
|
//
|
|
if (!(bStatus DBGCHK = Subnets.AddEntry( strSubnetObject )))
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return bStatus;
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
GetSubnetNames
|
|
|
|
Description:
|
|
|
|
Return a list of subnet strings for this machine, a machine may
|
|
be on multiple subnet since they may have multiple ip addresses.
|
|
|
|
Arguments:
|
|
|
|
TSubnet & - refrence to class of Subnets.
|
|
|
|
Return Value:
|
|
|
|
TRUE subnets were returned, FALSE error occurred.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
GetSubnetNames(
|
|
IN TSubnets &Subnets
|
|
)
|
|
{
|
|
TStatusB bStatus;
|
|
PMIB_IPADDRTABLE pAddrTable = NULL;
|
|
TString strSubnet;
|
|
|
|
Subnets.ClearAll();
|
|
|
|
bStatus DBGCHK = GetIpAddressTable( &pAddrTable );
|
|
|
|
if (bStatus && pAddrTable )
|
|
{
|
|
for (UINT i = 0; i < pAddrTable->dwNumEntries; i++)
|
|
{
|
|
//
|
|
// Skip the loopback or unassigned interfaces
|
|
//
|
|
if((pAddrTable->table[i].dwAddr != 0) && (pAddrTable->table[i].dwAddr != c_LoopBackAddress.s_addr))
|
|
{
|
|
IN_ADDR IpSubnetAddr;
|
|
|
|
IpSubnetAddr.s_addr = pAddrTable->table[i].dwAddr & pAddrTable->table[i].dwMask;
|
|
|
|
LPSTR pszIpAddress = m_inet_ntoa( IpSubnetAddr );
|
|
|
|
//
|
|
// Format the subnet mask to match the way subnet objects are
|
|
// in the DS. XX.XX.XX.XX/SS where SS are the number of significant
|
|
// bit in the subnet mask.
|
|
//
|
|
bStatus DBGCHK = strSubnet.bFormat( _T("%S\\/%d"),
|
|
pszIpAddress ? pszIpAddress : "",
|
|
NumSetBits( pAddrTable->table[i].dwMask ) );
|
|
|
|
if (bStatus)
|
|
{
|
|
//
|
|
// Add the subnet entry to the list of subnets.
|
|
//
|
|
bStatus DBGCHK = Subnets.AddEntry( strSubnet );
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
delete [] pAddrTable;
|
|
|
|
return bStatus;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
WidenScope
|
|
|
|
Description:
|
|
|
|
This routine expands the scope of the specified physical location string.
|
|
The physical location string uses / as scope separators. For example
|
|
if the physical location string on a computer object is
|
|
"/Redmond/Main Campus/Building 26 North/Floor 1/Office 1438"
|
|
after calling this routine with a count of 1 the new physical location
|
|
string will be "/Redmond/Main Campus/Building 26 North/Floor 1"
|
|
Thus it has expanded the scope this computer object is in.
|
|
|
|
Arguments:
|
|
|
|
pszString - pointer to physical location string to widen
|
|
uCount - number of scopes to widen the scope by, must be 1 or greater
|
|
strString - refrence to string object where to return the new widened scope.
|
|
|
|
Return Value:
|
|
|
|
TRUE the scope was widened, FALSE error occurred.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::
|
|
WidenScope(
|
|
IN LPCTSTR pszString,
|
|
IN UINT uCount,
|
|
IN OUT TString &strString
|
|
) const
|
|
{
|
|
TStatusB bStatus;
|
|
bStatus DBGNOCHK = FALSE;
|
|
|
|
if (uCount && pszString && *pszString)
|
|
{
|
|
UINT uLen = _tcslen( pszString );
|
|
|
|
LPTSTR pszTemp = new TCHAR [uLen+1];
|
|
|
|
if (pszTemp)
|
|
{
|
|
_tcscpy( pszTemp, pszString );
|
|
|
|
//
|
|
// Search from the end of the string to the beginning
|
|
// counting the number of separators, terminate when the
|
|
// desired count is reached.
|
|
//
|
|
for( LPTSTR p = pszTemp + uLen - 1; p != pszTemp; p-- )
|
|
{
|
|
if (*p == gchSeparator)
|
|
{
|
|
if (!--uCount)
|
|
{
|
|
*p = NULL;
|
|
bStatus DBGCHK = strString.bUpdate( pszTemp );
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
delete [] pszTemp;
|
|
}
|
|
|
|
return bStatus;
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
NumSetBits
|
|
|
|
Description:
|
|
|
|
Determines the number of set bits in the specified value.
|
|
|
|
Arguments:
|
|
|
|
DWORD Value - Value to count set bits.
|
|
|
|
Return Value:
|
|
|
|
Number of set bits in the value.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
UINT
|
|
TPhysicalLocation::
|
|
NumSetBits(
|
|
IN DWORD Value
|
|
)
|
|
{
|
|
UINT Count = 0;
|
|
|
|
//
|
|
// Start with the high order bit set and shift the bit
|
|
// to the right. This routine could be made very generic
|
|
// if a pointer to the value and the size in bytes of the
|
|
// value was accepted.
|
|
//
|
|
for( DWORD i = 1 << ( sizeof(Value) * 8 - 1 ); i; i >>= 1 )
|
|
{
|
|
Count += (Value & i) ? 1 : 0;
|
|
}
|
|
|
|
return Count;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
TSubnets
|
|
|
|
Description:
|
|
|
|
Constructor.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
Nothing.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
TPhysicalLocation::TSubnets::
|
|
TSubnets(
|
|
VOID
|
|
) : m_uNumEntries( 0 ),
|
|
m_pstrTable( NULL )
|
|
{
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
TSubnets
|
|
|
|
Description:
|
|
|
|
Destructor.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
Nothing.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
TPhysicalLocation::TSubnets::
|
|
~TSubnets(
|
|
VOID
|
|
)
|
|
{
|
|
delete [] m_pstrTable;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
ClearAll
|
|
|
|
Description:
|
|
|
|
Release the all the subnet entries.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
Nothing.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
VOID
|
|
TPhysicalLocation::TSubnets::
|
|
ClearAll(
|
|
VOID
|
|
)
|
|
{
|
|
m_uNumEntries = 0;
|
|
delete [] m_pstrTable;
|
|
m_pstrTable = NULL;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
NumEntries
|
|
|
|
Description:
|
|
|
|
Return the total number of valid entries.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
Number of subnet entries.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
UINT
|
|
TPhysicalLocation::TSubnets::
|
|
NumEntries(
|
|
VOID
|
|
)
|
|
{
|
|
return m_uNumEntries;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
Table
|
|
|
|
Description:
|
|
|
|
Return the string using the speified index.
|
|
|
|
Arguments:
|
|
|
|
Index - Zero based index of which string to return.
|
|
|
|
Return Value:
|
|
|
|
Refrence to string.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
TString &
|
|
TPhysicalLocation::TSubnets::
|
|
Table(
|
|
UINT Index
|
|
)
|
|
{
|
|
SPLASSERT( m_pstrTable || Index < m_uNumEntries);
|
|
return m_pstrTable[Index];
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
|
|
AddEntry
|
|
|
|
Description:
|
|
|
|
Add a new subnet entry into the array of subnet strings.
|
|
|
|
Arguments:
|
|
|
|
pszNew - pointer to new subnet string to add.
|
|
|
|
Return Value:
|
|
|
|
TRUE new subnet string was added, FALSE error.
|
|
|
|
Notes:
|
|
|
|
--*/
|
|
BOOL
|
|
TPhysicalLocation::TSubnets::
|
|
AddEntry(
|
|
IN LPCTSTR pszNew
|
|
)
|
|
{
|
|
TStatusB bStatus;
|
|
|
|
//
|
|
// Assume failure.
|
|
//
|
|
bStatus DBGNOCHK = FALSE;
|
|
|
|
//
|
|
// Allocate the new array of strings. Yes this
|
|
// is not the most efficent code but it works.
|
|
// Maybe a string list would be a better
|
|
// choice if I had one or the time to implement one.
|
|
//
|
|
TString *pTemp = new TString [m_uNumEntries+1];
|
|
|
|
if (pTemp)
|
|
{
|
|
//
|
|
// Assume success this is needed when the first
|
|
// string is added.
|
|
//
|
|
bStatus DBGNOCHK = TRUE;
|
|
|
|
//
|
|
// Copy the strings to the new array of strings.
|
|
//
|
|
for( UINT i = 0; i < m_uNumEntries; i++ )
|
|
{
|
|
if (!(bStatus DBGCHK = pTemp[i].bUpdate( m_pstrTable[i] )))
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
|
|
//
|
|
// Add the new string.
|
|
//
|
|
if (bStatus)
|
|
{
|
|
bStatus DBGCHK = pTemp[i].bUpdate( pszNew );
|
|
|
|
if( bStatus )
|
|
{
|
|
delete [] m_pstrTable;
|
|
m_pstrTable = pTemp;
|
|
m_uNumEntries++;
|
|
}
|
|
}
|
|
|
|
//
|
|
// If something failed free the temp string array.
|
|
//
|
|
if( !bStatus )
|
|
{
|
|
delete [] pTemp;
|
|
}
|
|
}
|
|
|
|
return bStatus;
|
|
}
|
|
|
|
/*++
|
|
|
|
Name:
|
|
TPhysicalLocation::bLocationEnabled
|
|
|
|
Description:
|
|
Checks policy bit to determine if physical location support is enabled
|
|
|
|
|
|
Arguments:
|
|
None
|
|
|
|
Return Value:
|
|
TRUE if enabled, FALSE otherwise
|
|
|
|
Notes:
|
|
Support is disabled by default
|
|
|
|
--*/
|
|
|
|
BOOL
|
|
TPhysicalLocation::
|
|
bLocationEnabled(
|
|
VOID
|
|
)
|
|
{
|
|
TStatusB bStatus;
|
|
BOOL bRet = FALSE;
|
|
|
|
TPersist PersistPolicy( gszSpoolerPolicy, TPersist::kOpen|TPersist::kRead, HKEY_LOCAL_MACHINE );
|
|
|
|
if( VALID_OBJ( PersistPolicy ) )
|
|
{
|
|
bStatus DBGNOCHK = PersistPolicy.bRead( gszUsePhysicalLocation, bRet );
|
|
}
|
|
|
|
return bRet;
|
|
}
|
|
|