archive
/
windows-xp
mirror of https://github.com/tongzx/nt5src
2
0
Fork 0
Code Issues Projects Releases Wiki Activity
Source code of Windows XP (NT5)
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.
6 Commits
1 Branch
0 Tags
2.0 GiB
 
 
 
 
 
 
Branch: master
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'master'
${ noResults }
windows-xp/Source/XPSP1/NT/base/fs/dfs/setup/dfssetup.cxx

2517 lines
61 KiB
Raw Permalink Blame History

//+----------------------------------------------------------------------------
//
// Copyright (C) 1995, Microsoft Corporation
//
// File: dfssetup.cxx
//
// Contents: Code to setup Dfs on a machine.
//
// Note that this code can be built as an exe or as a DLL. To
// switch between the two, simply edit the following fields in
// the sources file:
// TARGETTYPE=[PROGRAM | DYNLINK]
// UMTYPE=[console | windows]
// Delete the following two lines from sources to build an exe.
// DLLDEF=obj\*\dfssetup.def
// DLLENTRY=_DllMainCRTStartup
// DLLBASE=@$(BASEDIR)\PUBLIC\SDK\LIB\coffbase.txt,dfssetup
//
// Classes: None
//
// Functions:
//
// History: 12-28-95 Milans Created
//
//-----------------------------------------------------------------------------
extern "C" {
#include <nt.h>
#include <ntrtl.h>
#include <nturtl.h>
#include <stdlib.h>
}
#include <windows.h>
#include <rpc.h> // For UuidCreate
#include <winldap.h>
#include <dsgetdc.h>
#include <lm.h>
#include <dfsstr.h>
#include <dfsmsrv.h>
#include <debug.h> // Needed for debug printing
#include <dfsm.hxx> // Needed for volume types
#include <lmerrext.h>
#include "registry.hxx" // Helper functions...
#include "setupsvc.hxx"
#include "config.hxx" // Config UI functions
//
// Until we get the schema right in the DS, we have to set our own SD on
// certain objects
//
#include <aclapi.h>
#include <permit.h>
DECLARE_DEBUG(DfsSetup)
DECLARE_INFOLEVEL(DfsSetup)
#if DBG == 1
#define dprintf(x) DfsSetupInlineDebugOut x
# else
#define dprintf(x)
#endif
#define MAX_NETBIOS_NAME_LEN 16+1
extern DWORD
RemoveDfs(void);
BOOLEAN
DfsCheckForOldDfsService();
BOOLEAN
DfsIsDfsServiceRunning();
VOID
Usage();
DWORD
SetupDfsRoot(
LPWSTR wszDfsRootShare);
DWORD
SetupFTDfs(
IN LPWSTR wszRootShare,
IN LPWSTR wszFTDfsName);
DWORD
InitializeVolumeObjectStorage();
DWORD
CreateVolumeObject(
IN LPWSTR wszObjectName,
IN LPWSTR pwszEntryPath,
IN LPWSTR pwszServer,
IN LPWSTR pwszShare,
IN LPWSTR pwszComment,
OUT GUID *guidVolume);
DWORD
CreateFTRootVolumeInfo(
LPWSTR wszObjectName,
LPWSTR wszFTDfsConfigDN,
LPWSTR wszDomainName,
LPWSTR wszDfsName,
LPWSTR wszServerName,
LPWSTR wszShareName,
LPWSTR wszSharePath,
LPWSTR wszDCName,
BOOLEAN fNewFTDfs);
DWORD
StoreLvolInfo(
IN GUID *pVolumeID,
IN PWSTR pwszStorageID,
IN PWSTR pwszShareName,
IN PWSTR pwszEntryPath,
IN ULONG ulVolumeType);
DWORD
GetDomainAndComputerName(
OUT LPWSTR wszDomain OPTIONAL,
OUT LPWSTR wszComputer OPTIONAL);
DWORD
GetSharePath(
IN LPWSTR wszShareName,
OUT LPWSTR wszSharePath);
DWORD
TeardownFtDfs(
IN LPWSTR wszRootShare,
IN LPWSTR wszFTDfsName);
//+----------------------------------------------------------------------------
//
// Function: main
//
// Synopsis: Entry point in case you want to build this as an exe.
// Configures all Dfs components on a machine.
//
// Arguments: [argc] --
// [argv] --
//
// Returns: Nothing
//
//-----------------------------------------------------------------------------
void _cdecl
main(int argc, char *argv[])
{
DWORD dwErr = ERROR_SUCCESS;
DWORD cbRootLen;
BOOL fRootSetup = FALSE;
BOOLEAN OldDfsService = FALSE;
WCHAR wszDfsRootShare[ MAX_PATH ];
//
// Figure out the type of machine we are installing on - client or root
//
if (argc != 1 && argc != 3) {
Usage();
return;
}
if (argc == 3) {
fRootSetup = TRUE;
}
if ((dwErr == ERROR_SUCCESS) && fRootSetup) {
if (_stricmp( argv[1], "-root" ) != 0) {
Usage();
return;
}
cbRootLen = strlen(argv[2]);
mbstowcs( wszDfsRootShare, argv[2], cbRootLen + 1 );
dprintf((DEB_ERROR,"Setting up Dfs Root...\n"));
} else {
dprintf((DEB_ERROR,"Setting up Dfs Server...\n"));
}
//
// Configure the Dfs Driver
//
if (dwErr == ERROR_SUCCESS) {
dwErr = ConfigDfs();
if (dwErr == ERROR_SUCCESS) {
dwErr = ConfigDfsService();
if (dwErr != ERROR_SUCCESS) {
dprintf((
DEB_ERROR, "Win32 error configuring Dfs Service %d\n",
dwErr));
(void)RemoveDfs();
} else {
dprintf((DEB_ERROR,"Successfully configured Dfs...\n") );
}
} else {
dprintf((DEB_ERROR,"Error %d configuring Dfs driver!\n", dwErr));
}
}
//
// If this is a root setup, configure the necessary information
//
if (dwErr == ERROR_SUCCESS && fRootSetup) {
dwErr = SetupDfsRoot( wszDfsRootShare );
}
}
//+----------------------------------------------------------------------------
//
// Function: SetupDfsRoot
//
// Synopsis: Does necessary setup to make this Dfs a root of the Dfs.
//
// Arguments: [wszDfsRootShare] -- The share to use as the Dfs root.
//
// Returns: Win32 error from configuring the root storages etc.
//
//-----------------------------------------------------------------------------
DWORD
SetupDfsRoot(
LPWSTR wszDfsRootShare)
{
DWORD dwErr = ERROR_SUCCESS;
GUID guid;
WCHAR wszDfsRootPath[ MAX_PATH ];
WCHAR wszComputerName[ MAX_NETBIOS_NAME_LEN ];
PWCHAR wszDfsRootPrefix = NULL;
dwErr = GetDomainAndComputerName( NULL, wszComputerName );
//
// Figure out the share path for the Root Dfs share
//
if (dwErr == ERROR_SUCCESS) {
dwErr = GetSharePath( wszDfsRootShare, wszDfsRootPath );
}
//
// We have all the info we need now. Lets get to work....
//
// 1. Initialize the volume object section in the registry.
//
// 2. Initialize (ie, create if necessary) the storage used for the
// Dfs root.
//
// 3. Create the root volume object.
//
// 4. Configure the root volume as a local volume by updating the
// LocalVolumes section in the registry.
//
//
// Initialize the Dfs Manager Volume Object Store
//
if (dwErr == ERROR_SUCCESS) {
dwErr = InitializeVolumeObjectStorage();
}
if (dwErr == ERROR_SUCCESS) {
dprintf((DEB_ERROR,
"Setting [%ws] as Dfs storage root...\n", wszDfsRootPath));
dwErr = DfsInitGlobals( wszComputerName, DFS_MANAGER_SERVER );
if (dwErr == ERROR_SUCCESS) {
wszDfsRootPrefix = new WCHAR[1 +
wcslen(wszComputerName) +
1 +
wcslen(wszDfsRootShare) +
1];
if (wszDfsRootPrefix == NULL) {
dwErr = ERROR_OUTOFMEMORY;
}
}
if (dwErr == ERROR_SUCCESS) {
wszDfsRootPrefix[0] = UNICODE_PATH_SEP;
wcscpy( &wszDfsRootPrefix[1], wszComputerName );
wcscat( wszDfsRootPrefix, UNICODE_PATH_SEP_STR );
wcscat( wszDfsRootPrefix, wszDfsRootShare );
dwErr = CreateVolumeObject(
DOMAIN_ROOT_VOL, // Name of volume object
wszDfsRootPrefix, // EntryPath of volume
wszComputerName, // Server name
wszDfsRootShare, // Share name
L"Dfs Root Volume", // Comment
&guid); // Id of created volume
}
if (dwErr == ERROR_SUCCESS) {
dwErr = StoreLvolInfo(
&guid,
wszDfsRootPath,
wszDfsRootShare,
wszDfsRootPrefix,
DFS_VOL_TYPE_DFS | DFS_VOL_TYPE_REFERRAL_SVC);
}
if (wszDfsRootPrefix != NULL)
delete [] wszDfsRootPrefix;
if (dwErr != ERROR_SUCCESS)
dwErr = ERROR_INVALID_FUNCTION;
if (dwErr == ERROR_SUCCESS) {
DWORD dwErr;
CRegKey cregVolumesDir( HKEY_LOCAL_MACHINE, &dwErr, VOLUMES_DIR );
if (dwErr == ERROR_SUCCESS) {
CRegSZ cregRootShare(
cregVolumesDir,
ROOT_SHARE_VALUE_NAME,
wszDfsRootShare );
dwErr = cregRootShare.QueryErrorStatus();
}
}
}
return( dwErr );
}
//+----------------------------------------------------------------------------
//
// Function: SetupFTDfs
//
// Synopsis: Main exported function
//
// Arguments: [argc] --
// [argv] --
//
// Returns: Nothing
//
//-----------------------------------------------------------------------------
DWORD
SetupFTDfs(
IN LPWSTR wszRootShare,
IN LPWSTR wszFTDfsName)
{
DWORD dwErr = ERROR_SUCCESS;
WCHAR wszDomainName[MAX_PATH+1];
WCHAR wszComputerName[MAX_PATH+1];
WCHAR wszRootSharePath[MAX_PATH+1];
WCHAR wszDfsConfigDN[MAX_PATH+1];
WCHAR wszSDDfsConfigDN[MAX_PATH+1];
WCHAR wszConfigurationDN[ MAX_PATH+1 ];
WCHAR wszServerShare[MAX_PATH+1];
PDOMAIN_CONTROLLER_INFO pDCInfo = NULL;
LDAP *pldap = NULL;
PLDAPMessage pMsg = NULL;
LDAPModW ldapModClass, ldapModCN, ldapModPkt, ldapModPktGuid, ldapModServer;
LDAP_BERVAL ldapPkt, ldapPktGuid;
PLDAP_BERVAL rgModPktVals[2];
PLDAP_BERVAL rgModPktGuidVals[2];
LPWSTR rgModClassVals[2];
LPWSTR rgModCNVals[2];
LPWSTR rgModServerVals[5];
LPWSTR rgAttrs[5];
PLDAPModW rgldapMods[6];
BOOLEAN fNewFTDfs;
//
// We need some information before we start:
//
// 1. Our Domain name
//
// 2. Our Computer name
//
// 3. The share path of wszRootShare
//
// 4. The DN of the Configuration OU of our domain
//
dwErr = GetDomainAndComputerName( wszDomainName, wszComputerName );
if (dwErr != ERROR_SUCCESS) {
dprintf((
DEB_ERROR,
"Win32 Error %d getting Domain/Computer name\n", dwErr));
goto Cleanup;
}
dwErr = GetSharePath( wszRootShare, wszRootSharePath );
if (dwErr != ERROR_SUCCESS) {
dprintf((
DEB_ERROR,
"Win32 Error %d getting share path for %ws\n",
dwErr, wszRootShare));
goto Cleanup;
}
dwErr = DsGetDcName(
NULL, // Computer to remote to
NULL, // Domain - use our own
NULL, // Domain Guid
NULL, // Site Guid
DS_DIRECTORY_SERVICE_REQUIRED | // Flags
DS_PDC_REQUIRED,
&pDCInfo); // Return info
if (dwErr != ERROR_SUCCESS) {
dprintf((
DEB_ERROR,
"DsGetDcName failed with Win32 Error %d\n", dwErr));
goto Cleanup;
}
pldap = ldap_initW(&pDCInfo->DomainControllerAddress[2], LDAP_PORT);
if (pldap == NULL) {
dprintf((DEB_ERROR, "ldap_init failed\n"));
goto Cleanup;
}
dwErr = ldap_set_option(pldap, LDAP_OPT_AREC_EXCLUSIVE, LDAP_OPT_ON);
if (dwErr != LDAP_SUCCESS) {
dprintf((
DEB_ERROR,
"ldap_set_option failed with ldap error %d\n", dwErr));
pldap = NULL;
goto Cleanup;
}
dwErr = ldap_bind_s(pldap, NULL, NULL, LDAP_AUTH_SSPI);
if (dwErr != LDAP_SUCCESS) {
dprintf((
DEB_ERROR,
"ldap_bind_s failed with ldap error %d\n", dwErr));
pldap = NULL;
goto Cleanup;
}
rgAttrs[0] = L"defaultnamingContext";
rgAttrs[1] = NULL;
dwErr = ldap_search_sW(
pldap,
L"",
LDAP_SCOPE_BASE,
L"(objectClass=*)",
rgAttrs,
0,
&pMsg);
if (dwErr == LDAP_SUCCESS) {
PLDAPMessage pEntry = NULL;
PWCHAR *rgszNamingContexts = NULL;
DWORD i, cNamingContexts;
BOOLEAN fFoundCfg;
if ((pEntry = ldap_first_entry(pldap, pMsg)) != NULL &&
(rgszNamingContexts = ldap_get_valuesW(pldap, pEntry, rgAttrs[0])) != NULL &&
(cNamingContexts = ldap_count_valuesW(rgszNamingContexts)) > 0) {
wcscpy( wszConfigurationDN, *rgszNamingContexts );
fFoundCfg = TRUE;
if (!fFoundCfg) {
dwErr = ERROR_UNEXP_NET_ERR;
}
} else {
dwErr = ERROR_UNEXP_NET_ERR;
}
if (pEntry != NULL)
ldap_msgfree( pEntry );
if (rgszNamingContexts != NULL)
ldap_value_freeW( rgszNamingContexts );
}
if (dwErr != ERROR_SUCCESS) {
dprintf((DEB_ERROR, "Unable to find Configuration naming context\n"));
goto Cleanup;
}
//
// Great, we have all the parameters now, so configure the DS
//
//
// See if the DfsConfiguration object exists; if not, create it.
//
wsprintf(
wszDfsConfigDN,
L"CN=Dfs-Configuration,CN=System,%ws",
wszConfigurationDN);
rgAttrs[0] = L"cn";
rgAttrs[1] = NULL;
dwErr = ldap_search_sW(
pldap,
wszDfsConfigDN,
LDAP_SCOPE_BASE,
L"(objectClass=*)",
rgAttrs,
0,
&pMsg);
if (dwErr == LDAP_SUCCESS)
ldap_msgfree( pMsg );
if (dwErr == LDAP_NO_SUCH_OBJECT) {
rgModClassVals[0] = L"dfsConfiguration";
rgModClassVals[1] = NULL;
ldapModClass.mod_op = LDAP_MOD_ADD;
ldapModClass.mod_type = L"objectClass";
ldapModClass.mod_vals.modv_strvals = rgModClassVals;
rgModCNVals[0] = L"Dfs-Configuration";
rgModCNVals[1] = NULL;
ldapModCN.mod_op = LDAP_MOD_ADD;
ldapModCN.mod_type = L"cn";
ldapModCN.mod_vals.modv_strvals = rgModCNVals;
rgldapMods[0] = &ldapModClass;
rgldapMods[1] = &ldapModCN;
rgldapMods[2] = NULL;
dwErr = ldap_add_sW(
pldap,
wszDfsConfigDN,
rgldapMods);
}
if (dwErr != LDAP_SUCCESS) {
dprintf((
DEB_ERROR,
"1:ldap_add_s for %ws failed with LDAP error %d\n",
wszDfsConfigDN, dwErr ));
goto Cleanup;
}
//
// Check to see if we are joining an FTDfs or creating a new one.
//
wsprintf(
wszDfsConfigDN,
L"CN=%ws,CN=Dfs-Configuration,CN=System,%ws",
wszFTDfsName,
wszConfigurationDN);
wsprintf(
wszServerShare,
L"\\\\%ws\\%ws",
wszComputerName,
wszRootShare);
rgAttrs[0] = L"remoteServerName";
rgAttrs[1] = NULL;
dwErr = ldap_search_sW(
pldap,
wszDfsConfigDN,
LDAP_SCOPE_BASE,
L"(objectClass=*)",
rgAttrs,
0,
&pMsg);
if (dwErr == ERROR_SUCCESS) {
//
// We are joining an existing FT Dfs. Append our server\share to it
//
LDAPMessage *pmsgServers;
PWCHAR *rgServers, *rgNewServers;
DWORD cServers;
fNewFTDfs = FALSE;
pmsgServers = ldap_first_entry(pldap, pMsg);
if (pmsgServers != NULL) {
rgServers = ldap_get_valuesW(
pldap,
pmsgServers,
L"remoteServerName");
if (rgServers != NULL) {
cServers = ldap_count_valuesW( rgServers );
rgNewServers = new LPWSTR [ cServers + 2 ];
if (rgNewServers != NULL) {
CopyMemory( rgNewServers, rgServers, cServers * sizeof(rgServers[0]) );
rgNewServers[cServers] = wszServerShare;
rgNewServers[cServers+1] = NULL;
ldapModServer.mod_op = LDAP_MOD_REPLACE;
ldapModServer.mod_type = L"remoteServerName";
ldapModServer.mod_vals.modv_strvals = rgNewServers;
rgldapMods[0] = &ldapModServer;
rgldapMods[1] = NULL;
dwErr = ldap_modify_sW(pldap, wszDfsConfigDN, rgldapMods);
if (LDAP_ATTRIBUTE_OR_VALUE_EXISTS == dwErr)
dwErr = ERROR_SUCCESS;
delete [] rgNewServers;
} else {
dwErr = ERROR_OUTOFMEMORY;
}
ldap_value_freeW( rgServers );
} else {
dwErr = ERROR_OUTOFMEMORY;
}
ldap_msgfree( pmsgServers );
} else {
dwErr = ERROR_OUTOFMEMORY;
}
} else if (dwErr == LDAP_NO_SUCH_OBJECT) {
GUID idPkt;
DWORD dwPktVersion = 1;
//
// We are creating a new FTDfs, create a container to hold the Dfs
// configuration for it.
//
fNewFTDfs = TRUE;
//
// Generate the class and CN attributes
//
rgModClassVals[0] = L"ftDfs";
rgModClassVals[1] = NULL;
ldapModClass.mod_op = LDAP_MOD_ADD;
ldapModClass.mod_type = L"objectClass";
ldapModClass.mod_vals.modv_strvals = rgModClassVals;
rgModCNVals[0] = wszFTDfsName;
rgModCNVals[1] = NULL;
ldapModCN.mod_op = LDAP_MOD_ADD;
ldapModCN.mod_type = L"cn";
ldapModCN.mod_vals.modv_strvals = rgModCNVals;
//
// Generate the null PKT attribute
//
ldapPkt.bv_len = sizeof(DWORD);
ldapPkt.bv_val = (PCHAR) &dwPktVersion;
rgModPktVals[0] = &ldapPkt;
rgModPktVals[1] = NULL;
ldapModPkt.mod_op = LDAP_MOD_ADD | LDAP_MOD_BVALUES;
ldapModPkt.mod_type = L"pKT";
ldapModPkt.mod_vals.modv_bvals = rgModPktVals;
//
// Generate a PKT Guid attribute
//
UuidCreate( &idPkt );
ldapPktGuid.bv_len = sizeof(GUID);
ldapPktGuid.bv_val = (PCHAR) &idPkt;
rgModPktGuidVals[0] = &ldapPktGuid;
rgModPktGuidVals[1] = NULL;
ldapModPktGuid.mod_op = LDAP_MOD_ADD | LDAP_MOD_BVALUES;
ldapModPktGuid.mod_type = L"pKTGuid";
ldapModPktGuid.mod_vals.modv_bvals = rgModPktGuidVals;
//
// Generate a Remote-Server-Name attribute
//
rgModServerVals[0] = wszServerShare;
rgModServerVals[1] = NULL;
ldapModServer.mod_op = LDAP_MOD_ADD;
ldapModServer.mod_type = L"remoteServerName";
ldapModServer.mod_vals.modv_strvals = rgModServerVals;
//
// Assemble all the LDAPMod structures
//
rgldapMods[0] = &ldapModClass;
rgldapMods[1] = &ldapModCN;
rgldapMods[2] = &ldapModPkt;
rgldapMods[3] = &ldapModPktGuid;
rgldapMods[4] = &ldapModServer;
rgldapMods[5] = NULL;
//
// Create the Dfs metadata object.
//
dwErr = ldap_add_sW( pldap, wszDfsConfigDN, rgldapMods );
if (dwErr == ERROR_SUCCESS) {
dprintf((
DEB_ERROR,
"2:ldap_add_s worked for %ws with ldap error %d\n",
wszDfsConfigDN, dwErr));
}
}
if (dwErr != LDAP_SUCCESS) {
dprintf((
DEB_ERROR,
"2:ldap_add_s failed for %ws with ldap error %d\n",
wszDfsConfigDN, dwErr));
goto Cleanup;
}
//
// Finally, create the root volume object
//
dwErr = CreateFTRootVolumeInfo(
DOMAIN_ROOT_VOL,
wszDfsConfigDN,
wszDomainName,
wszFTDfsName,
wszComputerName,
wszRootShare,
wszRootSharePath,
&pDCInfo->DomainControllerAddress[2],
fNewFTDfs);
if (dwErr != LDAP_SUCCESS) {
dprintf((
DEB_ERROR, "CreateVolumeObject failed with error %d\n", dwErr));
} else {
dprintf((
DEB_ERROR, "Successfully created FT-Dfs Configuration!\n"));
}
Cleanup:
if (pDCInfo != NULL)
NetApiBufferFree( pDCInfo );
if (pldap != NULL)
ldap_unbind( pldap );
return( dwErr );
}
//+----------------------------------------------------------------------------
//
// Function: RemoveDfsRoot
//
// Synopsis: Removes the Dfs Root config info.
//
// Arguments: None
//
// Returns: Win32 error from registry actions
//
//-----------------------------------------------------------------------------
DWORD
RemoveDfsRoot()
{
DWORD dwErr;
CRegKey cregVolumesDir( HKEY_LOCAL_MACHINE, &dwErr, VOLUMES_DIR );
CRegKey *pcregLV = NULL;
if (dwErr != ERROR_SUCCESS) { // Unable to open volumes dir
return(dwErr);
}
pcregLV = new CRegKey( // Open local volumes section
HKEY_LOCAL_MACHINE,
REG_KEY_LOCAL_VOLUMES);
if (pcregLV != NULL) {
dwErr = pcregLV->QueryErrorStatus();
} else {
dwErr = ERROR_OUTOFMEMORY;
}
if (dwErr == ERROR_SUCCESS) {
dwErr = pcregLV->Delete(); // Delete local volumes
delete pcregLV;
pcregLV = NULL;
if (dwErr == ERROR_SUCCESS) {
//
// Recreate an empty local volumes key
//
pcregLV = new CRegKey( HKEY_LOCAL_MACHINE, REG_KEY_LOCAL_VOLUMES);
cregVolumesDir.Delete(); // Delete volumes dir
}
}
if (pcregLV != NULL) {
delete pcregLV;
}
return( dwErr );
}
//+----------------------------------------------------------------------------
//
// Function: DfsSetupDfs
//
// Synopsis: Entry point in case you want to build this as a DLL. This
// function is suitable for being called from a setup .inf
// file.
//
// Arguments: [cArgs] -- Count of args
// [lpszArgs] -- Array of args
// [lpszTextOut] -- On return, points to a global buffer
// containing the null string. This is required by the
// .inf file
//
// Returns: TRUE.
//
//-----------------------------------------------------------------------------
LPSTR szReturn = "";
extern "C" BOOL
DfsSetupDfs(
DWORD cArgs,
LPSTR lpszArgs[],
LPSTR *lpszTextOut)
{
int argc;
LPSTR *argv;
argv = (LPSTR *) malloc( (cArgs+1) * sizeof(LPSTR) );
if (argv == NULL) {
return( FALSE );
}
argv[0] = "DfsSetup";
for (argc = 1; argc <= (int) cArgs; argc++) {
argv[ argc ] = lpszArgs[ argc-1 ];
}
main( argc, argv );
free( argv );
*lpszTextOut = szReturn;
return( TRUE );
}
//+----------------------------------------------------------------------------
//
// Function: DfsSetupDfsRoot
//
// Synopsis: Entry point for setting up just the root part of Dfs in
// case you want to build this as a DLL. This function is
// suitable for being called from a setup .inf file.
//
// Arguments: [cArgs] -- Count of args
// [lpszArgs] -- Array of args
// [lpszTextOut] -- On return, points to a global buffer
// containing the null string. This is required by the
// .inf file
//
// Returns: TRUE.
//
//-----------------------------------------------------------------------------
BOOLEAN
DfsSetupDfsRoot(
DWORD cArgs,
LPSTR szArgs[],
LPSTR *szTextOut)
{
if (cArgs == 1) {
DWORD dwErr;
WCHAR wszDfsRootShare[MAX_PATH];
mbstowcs( wszDfsRootShare, szArgs[0], strlen(szArgs[0]) + 1);
dwErr = SetupDfsRoot( wszDfsRootShare );
}
*szTextOut = szReturn;
return( TRUE );
}
//+----------------------------------------------------------------------------
//
// Function: DfsSetupGetConfig
//
// Synopsis: Reads the current configuration from the registry.
//
// Arguments: [pcfg] -- The DFS_CONFIGURATION info to fill
//
// Returns: TRUE if successfully read the config info, FALSE otherwise
//
//-----------------------------------------------------------------------------
BOOLEAN
DfsSetupGetConfig(
HWND hwnd,
DFS_CONFIGURATION *pcfg)
{
DWORD dwErr;
CRegKey cregVolumesDir( HKEY_LOCAL_MACHINE, &dwErr, VOLUMES_DIR );
ULONG cbBuffer;
WCHAR wszErr[128];
ZeroMemory( pcfg, sizeof(DFS_CONFIGURATION) );
if (dwErr == ERROR_ACCESS_DENIED) {
MessageBox(
hwnd,
L"Insufficient priviledge",
DFS_COMPONENT_NAME,
MB_OK | MB_ICONSTOP);
return( FALSE );
} else if (dwErr != ERROR_SUCCESS && dwErr != ERROR_FILE_NOT_FOUND) {
swprintf(wszErr, L"Unexpected error %08lx accessing registry", dwErr);
MessageBox( hwnd, wszErr, DFS_COMPONENT_NAME, MB_OK | MB_ICONSTOP);
return( FALSE );
}
if (dwErr == ERROR_SUCCESS) {
pcfg->fHostsDfs = TRUE;
CRegSZ cregRootShare( cregVolumesDir, ROOT_SHARE_VALUE_NAME );
dwErr = cregRootShare.QueryErrorStatus();
if (dwErr == ERROR_SUCCESS) {
cbBuffer = sizeof(pcfg->szRootShare);
dwErr = cregRootShare.GetString( pcfg->szRootShare, &cbBuffer );
}
if (dwErr == ERROR_SUCCESS) {
DWORD dwErrFTDfs;
CRegSZ cregFTDfs( cregVolumesDir, FTDFS_VALUE_NAME );
dwErrFTDfs = cregFTDfs.QueryErrorStatus();
if (dwErrFTDfs == ERROR_SUCCESS) {
pcfg->fFTDfs = TRUE;
cbBuffer = sizeof(pcfg->szFTDfs);
dwErr = cregFTDfs.GetString(pcfg->szFTDfs, &cbBuffer);
}
}
if (dwErr != ERROR_SUCCESS) {
swprintf(wszErr, L"Error %08lx accessing registry", dwErr);
MessageBox( hwnd, wszErr, DFS_COMPONENT_NAME, MB_OK | MB_ICONSTOP);
return( FALSE );
}
}
return( TRUE );
}
//+----------------------------------------------------------------------------
//
// Function: DfsSetupConfigure
//
// Synopsis: Presents the dialog box for configuring the Dfs root.
//
// Arguments: [cArgs] -- Count of args
// [lpszArgs] -- Array of args
// [lpszTextOut] -- On return, points to a global buffer
// containing the name of the share selected by the
// user to serve as the root of the Dfs. If the user
// hits Cancel, or decides not to setup a Dfs root,
// this string is set to point to the NULL string.
//
// Returns: TRUE iff machine should be rebooted
//
//-----------------------------------------------------------------------------
LPSTR szCancel = "Cancel";
LPSTR szReboot = "Reboot";
BOOLEAN
DfsSetupConfigure(
DWORD cArgs,
LPSTR szArgs[],
LPSTR *szTextOut)
{
DWORD dwErr = 0;
HWND hwnd;
DFS_CONFIGURATION dfscfg, dfsNewCfg;
CHAR *pchUnused;
BOOLEAN fRtn=FALSE;
*szTextOut = szCancel; // Default return result
ASSERT( cArgs == 1 );
hwnd = (HWND) LongToHandle( strtoul( szArgs[0], &pchUnused, 16 ) );
//
// Read the current configuration
//
HCURSOR hCursor=LoadCursor(NULL,IDC_WAIT);
if (!DfsSetupGetConfig( hwnd, &dfscfg )) {
return fRtn;
}
//
// Next, pop up a dialog to allow the user to change the configuration
//
dfsNewCfg = dfscfg;
// Prompt the user for what flavour of Dfs to create
if (ConfigureDfs( hwnd, &dfsNewCfg ) == TRUE)
{
// Prompt the user for a local machine share
if (ConfigDfsShare(hwnd, &dfsNewCfg) == TRUE)
{
// Set the Hourglass
if (hCursor)
SetCursor(hCursor);
//
// User hit OK. Figure out what changed, if anything.
//
if (dfscfg.fHostsDfs == FALSE && dfsNewCfg.fHostsDfs == FALSE) {
//
// No change...
//
NOTHING;
} else if (dfscfg.fHostsDfs == FALSE && dfsNewCfg.fHostsDfs == TRUE) {
//
// We want to host a new Dfs, so call SetupDfsRoot or
// SetupFTDfsRoot depending on what kind of Dfs Host the user
// wants to setup
//
if (dfsNewCfg.fFTDfs) {
dwErr = SetupFTDfs( dfsNewCfg.szRootShare, dfsNewCfg.szFTDfs );
} else {
dwErr = SetupDfsRoot( dfsNewCfg.szRootShare );
}
*szTextOut = szReboot;
if (dwErr == 0) {
CRegKey RebootKey(HKEY_LOCAL_MACHINE, REBOOT_KEY, KEY_ALL_ACCESS,
NULL, REG_OPTION_VOLATILE);
dwErr = RebootKey.QueryErrorStatus();
if (dwErr) {
dprintf((DEB_ERROR,"DfsStopHosting could not add reboot key\n"));
}
else {
CRegDWORD rvReboot((const CRegKey &) RebootKey, REG_VALUE_REBOOT, 1);
dwErr = rvReboot.QueryErrorStatus();
}
}
} else if (dfscfg.fHostsDfs == TRUE && dfsNewCfg.fHostsDfs == FALSE) {
//
// We want to delete our Dfs root
//
if (MessageBox(
hwnd,
L"Are you sure you want to delete the Dfs\n"
L"rooted at this machine?",
DFS_COMPONENT_NAME,
MB_ICONQUESTION | MB_YESNO) == IDYES) {
dwErr = RemoveDfsRoot();
*szTextOut = dwErr == ERROR_SUCCESS ? szReboot : szCancel;
if (dwErr == 0) {
CRegKey RebootKey(HKEY_LOCAL_MACHINE, REBOOT_KEY, KEY_ALL_ACCESS,
NULL, REG_OPTION_VOLATILE);
dwErr = RebootKey.QueryErrorStatus();
if (dwErr) {
dprintf((DEB_ERROR,"DfsStopHosting could not add reboot key\n"));
}
else {
CRegDWORD rvReboot((const CRegKey &) RebootKey, REG_VALUE_REBOOT, 1);
dwErr = rvReboot.QueryErrorStatus();
}
}
}
} else if (dfscfg.fHostsDfs == TRUE && dfsNewCfg.fHostsDfs == TRUE) {
//
// User might have changed the root share, or simply changed
// status with respect to FTDfs
//
if ((_wcsicmp( dfscfg.szRootShare, dfsNewCfg.szRootShare )) != 0 ||
dfscfg.fFTDfs != dfsNewCfg.fFTDfs) {
dwErr = RemoveDfsRoot();
if (dwErr == ERROR_SUCCESS) {
if (dfsNewCfg.fFTDfs) {
dwErr = SetupFTDfs(
dfsNewCfg.szRootShare,
dfsNewCfg.szFTDfs);
} else {
dwErr = SetupDfsRoot( dfsNewCfg.szRootShare );
}
}
*szTextOut = szReboot;
if (dwErr == 0) {
CRegKey RebootKey(HKEY_LOCAL_MACHINE, REBOOT_KEY, KEY_ALL_ACCESS,
NULL, REG_OPTION_VOLATILE);
dwErr = RebootKey.QueryErrorStatus();
if (dwErr) {
dprintf((DEB_ERROR,"DfsStopHosting could not add reboot key\n"));
}
else {
CRegDWORD rvReboot((const CRegKey &) RebootKey, REG_VALUE_REBOOT, 1);
dwErr = rvReboot.QueryErrorStatus();
}
}
}
//
// Raid: 455295 Put in code to handle FTDfs name change
//
}
//
// Only request a reboot if everything has gone fine so far.
// MariusB and JonN 8/20/97
//
if (!dwErr)
fRtn=TRUE;
}
}
return fRtn;
}
//+----------------------------------------------------------------------------
//
// Function: Usage
//
// Synopsis: Prints out the usage message in case you want to build this
// as an .exe
//
// Arguments:
//
// Returns:
//
//-----------------------------------------------------------------------------
VOID
Usage()
{
dprintf((DEB_ERROR,"Usage: dfssetup -root <path name> -share <root share>\n"));
dprintf((DEB_ERROR," <path name> is the name of an empty directory to use\n"));
dprintf((DEB_ERROR," as the root storage for Dfs\n"));
dprintf((DEB_ERROR," <root share> is the share name to access <path name>\n"));
}
//+----------------------------------------------------------------------------
//
// Function: InitializeVolumeObjectStorage
//
// Synopsis: Initializes the Dfs Manager Volume Object store.
//
// Arguments: None
//
// Returns: DWORD from registry operations.
//
//-----------------------------------------------------------------------------
DWORD
InitializeVolumeObjectStorage()
{
DWORD dwErr;
CRegKey *pcregVolumeObjectStore = NULL;
pcregVolumeObjectStore = new CRegKey(HKEY_LOCAL_MACHINE, VOLUMES_DIR );
if (pcregVolumeObjectStore != NULL) {
dwErr = pcregVolumeObjectStore->QueryErrorStatus();
if ( dwErr == ERROR_SUCCESS ) {
dwErr = pcregVolumeObjectStore->Delete();
if (dwErr == ERROR_SUCCESS) {
delete pcregVolumeObjectStore;
pcregVolumeObjectStore = new CRegKey(
HKEY_LOCAL_MACHINE,
VOLUMES_DIR );
if (pcregVolumeObjectStore != NULL) {
dwErr = pcregVolumeObjectStore->QueryErrorStatus();
} else {
dwErr = ERROR_OUTOFMEMORY;
}
}
}
} else {
dwErr = ERROR_OUTOFMEMORY;
}
if (dwErr == ERROR_SUCCESS) {
dprintf((DEB_ERROR,"Successfully inited Dfs Manager Volume Storage...\n"));
}
if (pcregVolumeObjectStore != NULL) {
delete pcregVolumeObjectStore;
}
return(dwErr);
}
//+----------------------------------------------------------------------------
//
// Function: CreateVolumeObject
//
// Synopsis: Creates a volume object to bootstrap the Dfs namespace.
//
// Arguments: [pwszObjectName] -- The name of the volume object, relative
// to VOLUMES_DIR
// [pwszEntryPath] -- EntryPath of the volume.
// [pwszServer] -- Name of server used to access this Dfs volume
// [pwszShare] -- Name of share used to access this Dfs volume
// [pwszComment] -- Comment to stamp on the volume object.
// [guidVolume] -- ID of newly create dfs volume
//
// Returns:
//
//-----------------------------------------------------------------------------
DWORD
CreateVolumeObject(
LPWSTR wszObjectName,
LPWSTR pwszEntryPath,
LPWSTR pwszServer,
LPWSTR pwszShare,
LPWSTR pwszComment,
GUID *guidVolume)
{
DWORD dwErr;
WCHAR wszFullObject[ MAX_PATH ];
//
// First, compute the full object name, storage ids, and machine name.
//
wcscpy( wszFullObject, VOLUMES_DIR );
wcscat( wszFullObject, wszObjectName );
//
// Next, get a guid for this volume
//
UuidCreate( guidVolume );
//
// Lastly, create this volume object
//
dwErr = DfsManagerCreateVolumeObject(
wszFullObject,
pwszEntryPath,
pwszServer,
pwszShare,
pwszComment,
guidVolume);
if (dwErr == ERROR_SUCCESS) {
dprintf((DEB_ERROR,"Successfully inited Dfs Manager Volume [%ws]...\n", pwszEntryPath));
}
return(dwErr);
}
//+----------------------------------------------------------------------------
//
// Function: CreateFTRootVolumeInfo
//
// Synopsis: Creates a Dfs volume object - used to bootstrap a Dfs by
// creating a root volume object
//
// Arguments: [wszObjectName] -- Name of volume object
// [wszFTDfsConfigDN] -- The DN of the FTDfs config object in DS
// [wszDomainName] -- Name of FTDfs domain
// [wszDfsName] -- Name of Dfs
// [wszServerName] -- Name of root server
// [wszShareName] -- Name of root share
// [wszDCName] -- DC to use
// [fNewFTDfs] -- If true, this is a new FTDfs
//
// Returns:
//
//-----------------------------------------------------------------------------
DWORD
CreateFTRootVolumeInfo(
LPWSTR wszObjectName,
LPWSTR wszFTDfsConfigDN,
LPWSTR wszDomainName,
LPWSTR wszDfsName,
LPWSTR wszServerName,
LPWSTR wszShareName,
LPWSTR wszSharePath,
LPWSTR wszDCName,
BOOLEAN fNewFTDfs)
{
DWORD dwErr = ERROR_SUCCESS;
WCHAR wszFullObjectName[ MAX_PATH ];
WCHAR wszDfsPrefix[ MAX_PATH ];
GUID idVolume;
static BOOLEAN fInited = FALSE;
wcscpy( wszDfsPrefix, UNICODE_PATH_SEP_STR );
wcscat( wszDfsPrefix, wszDomainName );
wcscat( wszDfsPrefix, UNICODE_PATH_SEP_STR );
wcscat( wszDfsPrefix, wszDfsName );
wcscpy( wszFullObjectName, LDAP_VOLUMES_DIR );
wcscat( wszFullObjectName, wszObjectName );
if (!fInited) {
//
// Create the volumes dir key that indicates that this machine is to
// be a root of an FTDfs
//
CRegKey cregVolumesDir( HKEY_LOCAL_MACHINE, VOLUMES_DIR );
dwErr = cregVolumesDir.QueryErrorStatus();
if (dwErr == ERROR_SUCCESS) {
CRegSZ cregRootShare(
cregVolumesDir,
ROOT_SHARE_VALUE_NAME,
wszShareName );
dwErr = cregRootShare.QueryErrorStatus();
if (dwErr == ERROR_SUCCESS) {
CRegSZ cregFTDfs(
cregVolumesDir,
FTDFS_VALUE_NAME,
wszDfsName);
CRegSZ cregFTDfsConfigDN(
cregVolumesDir,
FTDFS_DN_VALUE_NAME,
wszFTDfsConfigDN);
dwErr = cregFTDfs.QueryErrorStatus();
if (dwErr == ERROR_SUCCESS) {
dwErr = cregFTDfsConfigDN.QueryErrorStatus();
if (dwErr != ERROR_SUCCESS)
cregFTDfs.Delete();
}
}
}
if (dwErr == ERROR_SUCCESS) {
dwErr = DfsInitGlobals(
wszDfsName,
DFS_MANAGER_FTDFS);
}
if (dwErr == ERROR_SUCCESS)
fInited = TRUE;
}
DfsManagerSetDcName(wszDCName);
if (dwErr == ERROR_SUCCESS) {
if (fNewFTDfs) {
//
// Generate a Guid for the new Volume
//
UuidCreate( &idVolume );
dwErr = DfsManagerCreateVolumeObject(
wszFullObjectName,
wszDfsPrefix,
wszServerName,
wszShareName,
L"Root Volume",
&idVolume);
} else {
dwErr = DfsManagerAddService(
wszFullObjectName,
wszServerName,
wszShareName,
&idVolume);
}
}
if (dwErr == ERROR_SUCCESS) {
dwErr = StoreLvolInfo(
&idVolume,
wszSharePath,
wszShareName,
wszDfsPrefix,
DFS_VOL_TYPE_DFS | DFS_VOL_TYPE_REFERRAL_SVC);
}
return( dwErr );
}
//+----------------------------------------------------------------------------
//
// Function: StoreLvolInfo
//
// Synopsis: Stores information about a local volume in the registry.
//
// Arguments: [pVolumeID] -- Id of dfs volume
// [pwszStorageId] -- Storage used by dfs volume
// [pwszShareName] -- LM Share used to access dfs volume
// [pwszEntryPath] -- EntryPath of dfs volume
// [ulVolumeType] -- Type of dfs volume. See DFS_VOL_TYPE_xxx
//
// Returns: DWORD from registry operations.
//
//-----------------------------------------------------------------------------
extern VOID
GuidToString(
IN GUID *pID,
OUT PWSTR pwszID);
DWORD
StoreLvolInfo(
IN GUID *pVolumeID,
IN PWSTR pwszStorageID,
IN PWSTR pwszShareName,
IN PWSTR pwszEntryPath,
IN ULONG ulVolumeType)
{
DWORD dwErr;
WCHAR wszLvolKey[_MAX_PATH];
UNICODE_STRING ustrNtStorageId;
PWCHAR pwcGuid;
wcscpy(wszLvolKey, REG_KEY_LOCAL_VOLUMES);
wcscat(wszLvolKey, UNICODE_PATH_SEP_STR);
pwcGuid = wszLvolKey + wcslen(wszLvolKey);
ASSERT( *pwcGuid == UNICODE_NULL );
GuidToString( pVolumeID, pwcGuid );
if (!RtlDosPathNameToNtPathName_U(
pwszStorageID,
&ustrNtStorageId,
NULL,
NULL)) {
return(ERROR_OUTOFMEMORY);
} else {
ustrNtStorageId.Buffer[ustrNtStorageId.Length/sizeof(WCHAR)] = UNICODE_NULL;
}
CRegKey rkeyLvol(HKEY_LOCAL_MACHINE, wszLvolKey, KEY_WRITE, NULL, REG_OPTION_NON_VOLATILE);
dwErr = rkeyLvol.QueryErrorStatus();
if (dwErr != ERROR_SUCCESS) {
RtlFreeUnicodeString(&ustrNtStorageId);
return(dwErr);
}
CRegSZ rvEntryPath((const CRegKey &) rkeyLvol, REG_VALUE_ENTRY_PATH, pwszEntryPath);
CRegSZ rvShortEntryPath((const CRegKey &) rkeyLvol, REG_VALUE_SHORT_PATH, pwszEntryPath);
CRegDWORD rvEntryType((const CRegKey &) rkeyLvol, REG_VALUE_ENTRY_TYPE, ulVolumeType);
CRegSZ rvStorageId((const CRegKey &) rkeyLvol, REG_VALUE_STORAGE_ID, ustrNtStorageId.Buffer);
CRegSZ rvShareName((const CRegKey &) rkeyLvol, REG_VALUE_SHARE_NAME, pwszShareName);
RtlFreeUnicodeString(&ustrNtStorageId);
if (ERROR_SUCCESS != (dwErr = rvEntryPath.QueryErrorStatus()) ||
ERROR_SUCCESS != (dwErr = rvShortEntryPath.QueryErrorStatus()) ||
ERROR_SUCCESS != (dwErr = rvEntryType.QueryErrorStatus()) ||
ERROR_SUCCESS != (dwErr = rvStorageId.QueryErrorStatus()) ||
ERROR_SUCCESS != (dwErr = rvShareName.QueryErrorStatus())) {
rkeyLvol.Delete();
} else {
dprintf((DEB_ERROR,"Successfully stored local volume info for [%ws]\n", pwszEntryPath));
}
return(dwErr);
}
//+------------------------------------------------------------------
//
// Function: CheckForOldDfsService
//
// Synopsis:
//
// Effects: -none-
//
// Arguments: -none-
//
// Returns: TRUE if the old (pre -ds build) dfs service is installed
//
//
// History: 10-09-96 JimMcN Created
//
// Notes:
//
//
//-------------------------------------------------------------------
BOOLEAN DfsCheckForOldDfsService( )
{
DWORD dwErr = 0;
DWORD DfsVersion;
// Open Service Controller
CService cSvc;
if (!(dwErr = cSvc.Init()))
{
dwErr = cSvc._OpenService(L"Dfs", SERVICE_QUERY_STATUS);
cSvc._CloseService();
if (dwErr != 0) {
return(FALSE);
}
CRegKey cregDfsService( HKEY_LOCAL_MACHINE,
&dwErr,
L"System\\CurrentControlSet\\Services\\Dfs"
);
if (dwErr == ERROR_SUCCESS) {
CRegDWORD DfsNewService((const CRegKey &)cregDfsService,
L"DfsVersion", &DfsVersion);
dwErr = DfsNewService.QueryErrorStatus();
if (dwErr != 0) {
dprintf((DEB_ERROR,"CheckForOldDfsService Failed Newserv\n"));
return(TRUE);
}
if (DfsVersion < DFS_VERSION_NUMBER) {
return(TRUE);
}
return(FALSE);
}
return(FALSE);
}
return FALSE ;
}
//+----------------------------------------------------------------------------
//
// Function: GetDomainAndComputerName
//
// Synopsis: Retrieves the domain and computer name of the local machine
//
// Arguments: [wszDomain] -- On successful return, contains name of domain.
// If this parameter is NULL on entry, the domain name is
// not returned.
//
// [wszComputer] -- On successful return, contains name of
// computer. If this parameter is NULL on entry, the
// computer name is not returned.
//
// Returns: [ERROR_SUCCESS] -- Successfully returning names.
//
// Win32 Error from calling NetWkstaGetInfo
//
//-----------------------------------------------------------------------------
DWORD
GetDomainAndComputerName(
OUT LPWSTR wszDomain OPTIONAL,
OUT LPWSTR wszComputer OPTIONAL)
{
DWORD dwErr;
PWKSTA_INFO_100 wkstaInfo = NULL;
dwErr = NetWkstaGetInfo( NULL, 100, (LPBYTE *) &wkstaInfo );
if (dwErr == NERR_Success) {
if (wszDomain)
wcscpy(wszDomain, wkstaInfo->wki100_langroup);
if (wszComputer)
wcscpy(wszComputer, wkstaInfo->wki100_computername);
NetApiBufferFree( wkstaInfo );
}
return( dwErr );
}
//+----------------------------------------------------------------------------
//
// Function: GetSharePath
//
// Synopsis: Returns the share path for a share on the local machine
//
// Arguments: [wszShare] -- Name of share
//
// [wszPath] -- On return, share path of wszShare
//
// Returns: [ERROR_SUCCESS] -- Successfully returning share path
//
// Win32 error from NetShareGetInfo
//
//-----------------------------------------------------------------------------
DWORD
GetSharePath(
IN LPWSTR wszShare,
OUT LPWSTR wszPath)
{
DWORD dwErr;
PSHARE_INFO_2 pshi2;
dwErr = NetShareGetInfo(
NULL, // Server (local machine)
wszShare, // Share Name
2, // Level,
(LPBYTE *) &pshi2); // Buffer
if (dwErr == ERROR_SUCCESS) {
wcscpy( wszPath, pshi2->shi2_path );
NetApiBufferFree( pshi2 );
}
return( dwErr );
}
//+----------------------------------------------------------------------------
//
// Function: DfsIsServiceRunning
//
// Synopsis: Check for active dfs servvice..
//
// Arguments:
//
// Returns: [TRUE] -- if dfs service is active.
// FALSE otherwise.
//
//-----------------------------------------------------------------------------
BOOLEAN
DfsIsDfsServiceRunning()
{
DWORD dwErr;
CService cSvc;
if (!(dwErr = cSvc.Init()))
{
dwErr = cSvc._OpenService(L"Dfs", SERVICE_QUERY_STATUS);
cSvc._CloseService();
return(dwErr == 0);
}
return(FALSE);
}
//+----------------------------------------------------------------------------
//
// Function: DfsGetDfsName
//
// Synopsis: Returns the share for a local dfs
//
// Arguments: [DfsRootShare] -- Name of share
//
// Returns: [TRUE] -- Successfully returning share path
//
// FALSE otherwise.
//
//-----------------------------------------------------------------------------
BOOLEAN
DfsGetDfsName(
OUT LPWSTR DfsRootShare,
IN OUT ULONG *Length,
OUT BOOLEAN *IsFtDfs)
{
DWORD dwErr;
ULONG SaveLength = *Length;
// CService cSvc;
ULONG ul = MAX_PATH;
WCHAR wc[MAX_PATH];
*IsFtDfs = FALSE;
#if 0
if (!(dwErr = cSvc.Init()))
{
dwErr = cSvc._OpenService(L"Dfs", SERVICE_QUERY_STATUS);
cSvc._CloseService();
if (dwErr != 0) {
return(FALSE);
}
}
#endif
CRegKey VolKey(HKEY_LOCAL_MACHINE, &dwErr, VOLUMES_DIR, KEY_READ);
if (dwErr) {
dprintf((DEB_ERROR,"DfsGetDfsName Failed to open VOLUMES_DIR\n"));
return(FALSE);
}
CRegSZ RootShare((const CRegKey &)VolKey, ROOT_SHARE_VALUE_NAME);
dwErr = RootShare.QueryErrorStatus();
if (dwErr) {
dprintf((DEB_ERROR,"DfsGetDfsName Failed to get root share name\n"));
return(FALSE);
}
dwErr = RootShare.GetString(DfsRootShare, Length);
if (dwErr) {
dprintf((DEB_ERROR,"DfsGetDfsName Failed to get root share val\n"));
return(FALSE);
}
CRegSZ FtDfs((const CRegKey &)VolKey, FTDFS_VALUE_NAME);
dwErr = FtDfs.QueryErrorStatus();
dwErr = FtDfs.GetString(wc, &ul);
*IsFtDfs = (dwErr == 0);
if (*IsFtDfs) {
memset(DfsRootShare, 0, SaveLength);
memcpy(DfsRootShare, wc, ul);
*Length=ul; }
dprintf((DEB_ERROR,"DfsGetDfsName is %s FTDFS\n", *IsFtDfs?"":"NOT"));
return(TRUE);
}
//+----------------------------------------------------------------------------
//
// Function: DfsNeedReboot
//
// Synopsis: Returns true if reboot is required for completing configuration.
//
// Arguments:
//
// Returns: [TRUE] -- Reboot Needed.
//
// FALSE otherwise.
//
//-----------------------------------------------------------------------------
BOOLEAN
DfsNeedReboot()
{
DWORD dwErr;
DWORD RebootDW = 20;
WCHAR buffer[20];
CRegKey RebootKey(HKEY_LOCAL_MACHINE, &dwErr, REBOOT_KEY, KEY_READ);
if (dwErr = RebootKey.QueryErrorStatus()) {
return(FALSE);
}
CRegSZ RebootVal((const CRegKey &)RebootKey, REG_VALUE_REBOOT);
dwErr = RebootVal.QueryErrorStatus();
DWORD GetString( LPWSTR pwszData, ULONG *pcbData);
dwErr = RebootVal.GetString(buffer, &RebootDW);
dprintf((DEB_ERROR,"DfsNeedReboot is %s Needed\n", dwErr == 0?"":"NOT"));
return(dwErr == 0);
}
//+----------------------------------------------------------------------------
//
// Function: DfsStopHostingDfs
//
// Synopsis: Remove this machine from a machine/ft dfs.
//
// Arguments:
//
// Returns: ERROR_SUCCESS if no error
//
// setup error otherwise.
//
//-----------------------------------------------------------------------------
DWORD
DfsStopHostingDfs()
{
DWORD dwErr = 0;
BOOLEAN ftDfs = FALSE;
ULONG ul = MAX_PATH;
WCHAR FtDfsName[MAX_PATH];
WCHAR DfsRootShare[MAX_PATH];
WCHAR wszFullObjectName[MAX_PATH];
WCHAR wszDomainName[MAX_PATH];
WCHAR wszComputerName[MAX_PATH];
if (DfsGetDfsName(FtDfsName, &ul, &ftDfs)) {
if (ftDfs) {
wcscpy( wszFullObjectName, LDAP_VOLUMES_DIR );
wcscat( wszFullObjectName, DOMAIN_ROOT_VOL );
dwErr = DfsInitGlobals(
FtDfsName,
DFS_MANAGER_FTDFS);
if (dwErr) {
dprintf((DEB_ERROR,"DfsStopHostingDfs init globals fail.\n"));
return(dwErr);
}
CRegKey VolKey(HKEY_LOCAL_MACHINE, &dwErr, VOLUMES_DIR, KEY_READ);
if (dwErr) {
dprintf((DEB_ERROR,"DfsGetDfsName Failed to open VOLUMES_DIR\n"));
return(dwErr);
}
CRegSZ RootShare((const CRegKey &)VolKey, ROOT_SHARE_VALUE_NAME);
dwErr = RootShare.QueryErrorStatus();
if (dwErr) {
dprintf((DEB_ERROR,"DfsGetDfsName Failed to get root share name\n"));
return(dwErr);
}
ul = MAX_PATH;
dwErr = RootShare.GetString(DfsRootShare, &ul);
if (dwErr) {
dprintf((DEB_ERROR,"DfsGetDfsName Failed to get root share val\n"));
return(dwErr);
}
dwErr = GetDomainAndComputerName( wszDomainName, wszComputerName );
if (dwErr) {
dprintf((DEB_ERROR,"DfsGetDfsName Failed computer name\n"));
return(dwErr);
}
dwErr = DfsManagerRemoveService(wszFullObjectName, wszComputerName);
if (dwErr && dwErr != NERR_DfsCantRemoveLastServerShare) {
return(dwErr);
}
dwErr = TeardownFtDfs(DfsRootShare, FtDfsName);
}
if (dwErr)
return(dwErr);
dwErr = RemoveDfsRoot();
}
if (dwErr == 0) {
CRegKey RebootKey(HKEY_LOCAL_MACHINE, REBOOT_KEY, KEY_ALL_ACCESS, NULL, REG_OPTION_VOLATILE);
dwErr = RebootKey.QueryErrorStatus();
if (dwErr) {
dprintf((DEB_ERROR,"DfsStopHosting could not add reboot key\n"));
}
else {
CRegDWORD rvReboot((const CRegKey &) RebootKey, REG_VALUE_REBOOT, 1);
dwErr = rvReboot.QueryErrorStatus();
}
}
else {
dprintf((DEB_ERROR,"DfsStopHosting Dfs Failed to remove dfs root\n"));
}
return(dwErr);
}
//+----------------------------------------------------------------------------
//
// Function: TeardownFtDfs
//
// Synopsis: Removes FtDfs information from the ds.
//
// Arguments: wszRootShare the share hosting the dfs.
// wszFTDfsName the name of the FtDfs to remove from the ds.
//
//
// Returns: Nothing
//
//-----------------------------------------------------------------------------
DWORD
TeardownFtDfs(
IN LPWSTR wszRootShare,
IN LPWSTR wszFTDfsName)
{
DWORD dwErr = ERROR_SUCCESS;
DWORD i, j;
WCHAR wszDomainName[MAX_PATH];
WCHAR wszComputerName[MAX_PATH];
WCHAR wszRootSharePath[MAX_PATH];
WCHAR wszConfigurationDN[ MAX_PATH ];
WCHAR wszDfsConfigDN[ MAX_PATH ];
WCHAR wszServerShare[MAX_PATH];
PDOMAIN_CONTROLLER_INFO pDCInfo = NULL;
LDAP *pldap = NULL;
PLDAPMessage pMsg = NULL;
LDAPModW ldapModClass, ldapModCN, ldapModPkt, ldapModPktGuid, ldapModServer;
LDAP_BERVAL ldapPkt, ldapPktGuid;
PLDAP_BERVAL rgModPktVals[2];
PLDAP_BERVAL rgModPktGuidVals[2];
LPWSTR rgModClassVals[2];
LPWSTR rgModCNVals[2];
LPWSTR rgModServerVals[5];
LPWSTR rgAttrs[5];
PLDAPModW rgldapMods[6];
BOOLEAN fNewFTDfs;
//
// We need some information before we start:
//
// 1. Our Domain name
//
// 2. Our Computer name
//
// 3. The share path of wszRootShare
//
// 4. The DN of the Configuration OU of our domain
//
dwErr = GetDomainAndComputerName( wszDomainName, wszComputerName );
if (dwErr != ERROR_SUCCESS) {
dprintf((
DEB_ERROR,
"Win32 Error %d getting Domain/Computer name\n", dwErr));
goto Cleanup;
}
dwErr = GetSharePath( wszRootShare, wszRootSharePath );
if (dwErr != ERROR_SUCCESS) {
dprintf((
DEB_ERROR,
"Win32 Error %d getting share path for %ws\n",
dwErr, wszRootShare));
goto Cleanup;
}
dwErr = DsGetDcName(
NULL, // Computer to remote to
NULL, // Domain - use our own
NULL, // Domain Guid
NULL, // Site Guid
DS_DIRECTORY_SERVICE_REQUIRED | // Flags
DS_PDC_REQUIRED,
&pDCInfo); // Return info
if (dwErr != ERROR_SUCCESS) {
dprintf((
DEB_ERROR,
"DsGetDcName failed with Win32 Error %d\n", dwErr));
goto Cleanup;
}
pldap = ldap_initW(&pDCInfo->DomainControllerAddress[2], LDAP_PORT);
if (pldap == NULL) {
dprintf((DEB_ERROR, "ldap_init failed\n"));
goto Cleanup;
}
dwErr = ldap_set_option(pldap, LDAP_OPT_AREC_EXCLUSIVE, LDAP_OPT_ON);
if (dwErr != LDAP_SUCCESS) {
dprintf((
DEB_ERROR,
"ldap_set_option failed with ldap error %d\n", dwErr));
pldap = NULL;
goto Cleanup;
}
dwErr = ldap_bind_s(pldap, NULL, NULL, LDAP_AUTH_SSPI);
if (dwErr != LDAP_SUCCESS) {
dprintf((
DEB_ERROR,
"ldap_bind_s failed with ldap error %d\n", dwErr));
pldap = NULL;
goto Cleanup;
}
rgAttrs[0] = L"defaultnamingContext";
rgAttrs[1] = NULL;
dwErr = ldap_search_sW(
pldap,
L"",
LDAP_SCOPE_BASE,
L"(objectClass=*)",
rgAttrs,
0,
&pMsg);
if (dwErr == LDAP_SUCCESS) {
PLDAPMessage pEntry = NULL;
PWCHAR *rgszNamingContexts = NULL;
DWORD i, cNamingContexts;
BOOLEAN fFoundCfg;
if ((pEntry = ldap_first_entry(pldap, pMsg)) != NULL &&
(rgszNamingContexts = ldap_get_valuesW(pldap, pEntry, rgAttrs[0])) != NULL &&
(cNamingContexts = ldap_count_valuesW(rgszNamingContexts)) > 0) {
wcscpy( wszConfigurationDN, *rgszNamingContexts );
fFoundCfg = TRUE;
if (!fFoundCfg) {
dwErr = ERROR_UNEXP_NET_ERR;
}
} else {
dwErr = ERROR_UNEXP_NET_ERR;
}
if (rgszNamingContexts != NULL)
ldap_value_freeW( rgszNamingContexts );
ldap_msgfree( pMsg );
}
if (dwErr != ERROR_SUCCESS) {
dprintf((DEB_ERROR, "TD:Unable to find Configuration naming context\n"));
goto Cleanup;
}
//
// Great, we have all the parameters now, so configure the DS
//
//
// See if the DfsConfiguration object exists; if not, create it.
//
wsprintf(
wszDfsConfigDN,
L"CN=Dfs-Configuration,CN=System,%ws",
wszConfigurationDN);
rgAttrs[0] = L"cn";
rgAttrs[1] = NULL;
dwErr = ldap_search_sW(
pldap,
wszDfsConfigDN,
LDAP_SCOPE_BASE,
L"(objectClass=*)",
rgAttrs,
0,
&pMsg);
if (dwErr == LDAP_SUCCESS)
ldap_msgfree( pMsg );
if (dwErr != LDAP_SUCCESS) {
dprintf((
DEB_ERROR,
"3: ldap_search_sW for %ws failed with LDAP error %d\n",
wszDfsConfigDN, dwErr ));
goto Cleanup;
}
//
// Check to see if we are joining an FTDfs or creating a new one.
//
wsprintf(
wszDfsConfigDN,
L"CN=%ws,CN=Dfs-Configuration,CN=System,%ws",
wszFTDfsName,
wszConfigurationDN);
wsprintf(
wszServerShare,
L"\\\\%ws\\%ws",
wszComputerName,
wszRootShare);
rgAttrs[0] = L"remoteServerName";
rgAttrs[1] = NULL;
dwErr = ldap_search_sW(
pldap,
wszDfsConfigDN,
LDAP_SCOPE_BASE,
L"(objectClass=*)",
rgAttrs,
0,
&pMsg);
if (dwErr == ERROR_SUCCESS) {
//
// We found a Dfs object to modify/delete
//
LDAPMessage *pmsgServers;
PWCHAR *rgServers, *rgNewServers;
DWORD cServers;
fNewFTDfs = FALSE;
pmsgServers = ldap_first_entry(pldap, pMsg);
if (pmsgServers != NULL) {
rgServers = ldap_get_valuesW(
pldap,
pmsgServers,
L"remoteServerName");
if (rgServers != NULL) {
cServers = ldap_count_valuesW( rgServers );
rgNewServers = new LPWSTR [ cServers + 1 ];
if (rgNewServers != NULL) {
for (i = j = 0; i < cServers; i += 1) {
if (wcscmp(wszServerShare, rgServers[i]) == 0) {
continue;
}
rgNewServers[j++] = rgServers[i];
}
rgNewServers[j] = NULL;
if (j) {
ldapModServer.mod_op = LDAP_MOD_REPLACE;
ldapModServer.mod_type = L"remoteServerName";
ldapModServer.mod_vals.modv_strvals = rgNewServers;
rgldapMods[0] = &ldapModServer;
rgldapMods[1] = NULL;
dwErr = ldap_modify_sW(pldap, wszDfsConfigDN, rgldapMods);
} else {
//
// Delete the Dfs metadata object.
//
dwErr = ldap_delete_sW( pldap, wszDfsConfigDN);
}
delete [] rgNewServers;
} else {
dwErr = ERROR_OUTOFMEMORY;
}
ldap_value_freeW( rgServers );
} else {
dwErr = ERROR_OUTOFMEMORY;
}
} else {
dwErr = ERROR_OUTOFMEMORY;
}
ldap_msgfree( pMsg );
}
if (dwErr != LDAP_SUCCESS) {
dprintf((
DEB_ERROR,
"4: ldap_modify_s/ldap_delete_s failed for %ws with ldap error %d\n",
wszDfsConfigDN, dwErr));
goto Cleanup;
}
Cleanup:
if (pDCInfo != NULL)
NetApiBufferFree( pDCInfo );
if (pldap != NULL)
ldap_unbind( pldap );
return( dwErr );
}