windows-nt/Source/XPSP1/NT/admin/admt/workobj/regtrans.cpp
2020-09-26 16:20:57 +08:00

1176 lines
41 KiB
C++

/*---------------------------------------------------------------------------
File: RegTranslator.cpp
Comments: Routines for translating security on the registry keys and files
that form a user profile.
(c) Copyright 1999, Mission Critical Software, Inc., All Rights Reserved
Proprietary and confidential to Mission Critical Software, Inc.
REVISION LOG ENTRY
Revision By: Christy Boles
Revised on 05/12/99 11:11:46
---------------------------------------------------------------------------
*/
#include "stdafx.h"
#include "stargs.hpp"
#include "sd.hpp"
#include "SecObj.hpp"
#include "sidcache.hpp"
#include "sdstat.hpp"
#include "Common.hpp"
#include "UString.hpp"
#include "ErrDct.hpp"
#include "TReg.hpp"
#include "TxtSid.h"
#include "RegTrans.h"
#include <WinBase.h>
//#import "\bin\McsDctWorkerObjects.tlb"
#import "WorkObj.tlb"
#include "CommaLog.hpp"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
extern TErrorDct err;
#define LEN_SID 200
#define REGKEY_ADD_LIMIT 15
DWORD
TranslateRegHive(
HKEY hKeyRoot, // in - root of registry hive to translate
const LPWSTR keyName, // in - name of registry key
SecurityTranslatorArgs * stArgs, // in - translation settings
TSDRidCache * cache, // in - translation table
TSDResolveStats * stat, // in - stats on items modified
BOOL bWin2K // in - flag, whether the machine is Win2K
)
{
DWORD rc = 0;
// Translate the permissions on the root key
TRegSD sd(keyName,hKeyRoot);
if ( sd.HasDacl() )
{
TSD * pSD = sd.GetSecurity();
// if there are more than 15 aces in a registry entry, and we are re-ACLing in Add Mode,
// skip the entry. There seems to be a problem in Windows 2000 RC2 where if there are more than
// 30 or so ACEs, some of them never get resolved. If this happens to the Administrator account,
// (This may only happen if there are 30 or so ACEs alphabetically ahead of the Administrator account)
// The Administrator account never gets resolved, and the machine is effectively trashed.
if ( bWin2K && ( stArgs->TranslationMode() == ADD_SECURITY ) && (pSD->GetNumDaclAces() >= REGKEY_ADD_LIMIT) )
{
err.MsgWrite(0,DCT_MSG_SKIPPING_REGKEY_TRANSLATION_SDD,keyName,pSD->GetNumDaclAces(),REGKEY_ADD_LIMIT);
}
else
{
sd.ResolveSD(stArgs,stat,regkey ,NULL);
}
}
// Recursively process any subkeys
int n = 0;
FILETIME writeTime;
WCHAR name[MAX_PATH];
DWORD lenName = DIM(name);
WCHAR fullName[2000];
HKEY hKey;
do
{
lenName = DIM(name);
rc = RegEnumKeyEx(hKeyRoot,n,name,&lenName,NULL,NULL,NULL,&writeTime);
if ( rc && rc != ERROR_MORE_DATA )
break;
swprintf(fullName,L"%s\\%s",keyName,name);
// Open the subkey
rc = RegCreateKeyEx(hKeyRoot,name,0,L"",REG_OPTION_BACKUP_RESTORE,KEY_ALL_ACCESS | READ_CONTROL | ACCESS_SYSTEM_SECURITY,NULL,&hKey,NULL);
if (! rc )
{
// Process the subkey
TranslateRegHive(hKey,fullName,stArgs,cache,stat,bWin2K);
RegCloseKey(hKey);
}
else
{
if ( (rc != ERROR_FILE_NOT_FOUND) && (rc != ERROR_INVALID_HANDLE) )
{
err.SysMsgWrite(ErrS,rc,DCT_MSG_REG_KEY_OPEN_FAILED_SD,fullName,rc);
}
}
n++;
} while ( rc == ERROR_SUCCESS || rc == ERROR_MORE_DATA);
if ( rc != ERROR_NO_MORE_ITEMS && rc != ERROR_FILE_NOT_FOUND && rc != ERROR_INVALID_HANDLE )
{
err.SysMsgWrite(ErrS,rc,DCT_MSG_REGKEYENUM_FAILED_D,rc);
}
return rc;
}
DWORD
TranslateRegistry(
WCHAR const * computer, // in - computername to translate, or NULL
SecurityTranslatorArgs * stArgs, // in - translation settings
TSDRidCache * cache, // in - translation account mapping
TSDResolveStats * stat // in - stats for items examined and modified
)
{
DWORD rc = 0;
WCHAR comp[LEN_Computer];
if ( ! computer )
{
comp[0] = 0;
}
else
{
safecopy(comp,computer);
}
MCSDCTWORKEROBJECTSLib::IAccessCheckerPtr pAccess(__uuidof(MCSDCTWORKEROBJECTSLib::AccessChecker));
TRegKey hKey;
DWORD verMaj,verMin,verSP;
BOOL bWin2K = TRUE; // assume win2k unless we're sure it's not
// get the OS version - we need to know the OS version because Win2K can fail when registry keys
// have many entries
HRESULT hr = pAccess->raw_GetOsVersion(SysAllocString(comp),&verMaj,&verMin,&verSP);
if ( SUCCEEDED(hr) )
{
if ( verMaj < 5 )
bWin2K = FALSE;
}
err.MsgWrite(0,DCT_MSG_TRANSLATING_REGISTRY);
rc = hKey.Connect(HKEY_CLASSES_ROOT,computer);
if ( ! rc )
{
rc = TranslateRegHive(hKey.KeyGet(),L"HKEY_CLASSES_ROOT",stArgs,cache,stat,bWin2K);
hKey.Close();
}
rc = hKey.Connect(HKEY_LOCAL_MACHINE,computer);
if ( ! rc )
{
rc = TranslateRegHive(hKey.KeyGet(),L"HKEY_LOCAL_MACHINE",stArgs,cache,stat,bWin2K);
hKey.Close();
}
rc = hKey.Connect(HKEY_USERS,computer);
if (! rc )
{
rc = TranslateRegHive(hKey.KeyGet(),L"HKEY_USERS",stArgs,cache,stat,bWin2K);
hKey.Close();
}
rc = hKey.Connect(HKEY_PERFORMANCE_DATA,computer);
if ( ! rc )
{
rc = TranslateRegHive(hKey.KeyGet(),L"HKEY_PERFORMANCE_DATA",stArgs,cache,stat,bWin2K);
hKey.Close();
}
rc = hKey.Connect(HKEY_CURRENT_CONFIG,computer);
if ( ! rc )
{
rc = TranslateRegHive(hKey.KeyGet(),L"HKEY_CURRENT_CONFIG",stArgs,cache,stat,bWin2K);
hKey.Close();
}
rc = hKey.Connect(HKEY_DYN_DATA,computer);
if ( ! rc )
{
rc = TranslateRegHive(hKey.KeyGet(),L"HKEY_DYN_DATA",stArgs,cache,stat,bWin2K);
hKey.Close();
}
return rc;
}
DWORD
TranslateUserProfile(
WCHAR const * profileName, // in - name of file containing user profile
SecurityTranslatorArgs * stArgs, // in - translation settings
TSDRidCache * cache, // in - translation table
TSDResolveStats * stat, // in - stats on items modified
WCHAR * sSourceName, // in - Source account name
WCHAR * sSourceDomainName // in - Source domain name
)
{
DWORD rc = 0;
WCHAR oldName[MAX_PATH];
WCHAR newName[MAX_PATH];
WCHAR otherName[MAX_PATH];
HKEY hKey;
HRESULT hr = S_OK;
BOOL bWin2K = TRUE;
MCSDCTWORKEROBJECTSLib::IAccessCheckerPtr pAccess(__uuidof(MCSDCTWORKEROBJECTSLib::AccessChecker));
safecopy(oldName,profileName);
safecopy(newName,profileName);
UStrCpy(newName+UStrLen(newName),".temp");
safecopy(otherName,profileName);
UStrCpy(otherName + UStrLen(otherName),".premigration");
// check the OS version of the computer
// if UNC name is specified, get the computer name
if ( profileName[0] == L'\\' && profileName[1] == L'\\' )
{
bWin2K = TRUE; // if the profile is specified in UNC format (roaming profile) it can be used
// from multiple machines. There is no guarantee that the profile will not be loaded on a win2000 machine
}
else
{
DWORD verMaj;
DWORD verMin;
DWORD verSP;
HRESULT hr = pAccess->raw_GetOsVersion(SysAllocString(L""),&verMaj,&verMin,&verSP);
if ( SUCCEEDED(hr) )
{
if ( verMaj < 5 )
{
bWin2K = FALSE;
}
}
}
// Load the registry hive into the registry
rc = RegLoadKey(HKEY_USERS,L"OnePointTranslation",profileName);
if ( ! rc )
{
// Open the key
rc = RegOpenKeyEx(HKEY_USERS,L"OnePointTranslation",0,KEY_ALL_ACCESS | READ_CONTROL | ACCESS_SYSTEM_SECURITY,&hKey);
if ( ! rc )
{
// Process the registry hive
rc = TranslateRegHive(hKey,L"",stArgs,cache,stat,bWin2K);
// Unload the registry hive
if ( ! stArgs->NoChange() )
{
DeleteFile(newName);
hr = UpdateMappedDrives(sSourceName, sSourceDomainName, L"OnePointTranslation");
rc = RegSaveKey(hKey,newName,NULL);
}
else
{
rc = 0;
}
if ( rc )
{
err.SysMsgWrite(ErrS,rc,DCT_MSG_SAVE_HIVE_FAILED_SD,newName,rc);
}
RegCloseKey(hKey);
}
rc = RegUnLoadKey(HKEY_USERS,L"OnePointTranslation");
if ( rc )
{
err.SysMsgWrite(ErrE,rc,DCT_MSG_KEY_UNLOADKEY_FAILED_SD,profileName,rc);
}
}
else
{
err.SysMsgWrite(ErrS,rc,DCT_MSG_PROFILE_LOAD_FAILED_SD,profileName,rc);
}
if ( ! rc )
{
if (! stArgs->NoChange() )
{
// Switch out the filenames
if ( MoveFileEx(oldName,otherName,MOVEFILE_REPLACE_EXISTING) )
{
if ( ! MoveFileEx(newName,oldName,0) )
{
rc = GetLastError();
err.SysMsgWrite(ErrS,rc,DCT_MSG_RENAME_DIR_FAILED_SSD,newName,oldName,rc);
}
}
else
{
rc = GetLastError();
if ( rc == ERROR_ACCESS_DENIED )
{
// we do not have access to the directory
// temporarily grant ourselves access
// Set NTFS permissions for the results directory
WCHAR dirName[LEN_Path];
safecopy(dirName,oldName);
WCHAR * slash = wcsrchr(dirName,L'\\');
if ( slash )
{
(*slash) = 0;
}
TFileSD fsdDirBefore(dirName);
TFileSD fsdDirTemp(dirName);
TFileSD fsdDatBefore(oldName);
TFileSD fsdDatTemp(oldName);
TFileSD fsdNewBefore(newName);
TFileSD fsdNewTemp(newName);
BOOL dirChanged = FALSE;
BOOL datChanged = FALSE;
BOOL newChanged = FALSE;
// Temporarily reset the permissions on the directory and the appropriate files
if ( fsdDirTemp.GetSecurity() != NULL )
{
TACE ace(ACCESS_ALLOWED_ACE_TYPE,0,DACL_FULLCONTROL_MASK,
GetWellKnownSid(stArgs->IsLocalSystem() ? 7/*SYSTEM*/ : 1/*ADMINISTRATORS*/));
PACL acl = NULL;
fsdDirTemp.GetSecurity()->ACLAddAce(&acl,&ace,0);
if (acl)
{
fsdDirTemp.GetSecurity()->SetDacl(acl,TRUE);
fsdDirTemp.WriteSD();
dirChanged = TRUE;
}
}
if ( fsdDatTemp.GetSecurity() != NULL )
{
TACE ace(ACCESS_ALLOWED_ACE_TYPE,0,DACL_FULLCONTROL_MASK,
GetWellKnownSid(stArgs->IsLocalSystem() ? 7/*SYSTEM*/ : 1/*ADMINISTRATORS*/));
PACL acl = NULL;
fsdDatTemp.GetSecurity()->ACLAddAce(&acl,&ace,0);
if (acl)
{
fsdDatTemp.GetSecurity()->SetDacl(acl,TRUE);
fsdDatTemp.WriteSD();
datChanged = TRUE;
}
}
if ( fsdNewTemp.GetSecurity() != NULL )
{
TACE ace(ACCESS_ALLOWED_ACE_TYPE,0,DACL_FULLCONTROL_MASK,
GetWellKnownSid(stArgs->IsLocalSystem() ? 7/*SYSTEM*/ : 1/*ADMINISTRATORS*/));
PACL acl = NULL;
fsdNewTemp.GetSecurity()->ACLAddAce(&acl,&ace,0);
if (acl)
{
fsdNewTemp.GetSecurity()->SetDacl(acl,TRUE);
fsdNewTemp.WriteSD();
newChanged = TRUE;
}
}
rc = 0;
// Now retry the operations
if ( MoveFileEx(oldName,otherName,MOVEFILE_REPLACE_EXISTING) )
{
if ( ! MoveFileEx(newName,oldName,0) )
{
rc = GetLastError();
err.SysMsgWrite(ErrS,rc,DCT_MSG_RENAME_DIR_FAILED_SSD,newName,oldName,rc);
}
}
else
{
rc = GetLastError();
err.SysMsgWrite(ErrS,rc,DCT_MSG_RENAME_DIR_FAILED_SSD,oldName,otherName,rc);
}
// now that we're done, set the permissions back to what they were
if ( dirChanged )
{
fsdDirBefore.Changed(TRUE);
fsdDirBefore.WriteSD();
}
if ( datChanged )
{
fsdDatBefore.Changed(TRUE);
fsdDatBefore.WriteSD();
}
if ( newChanged )
{
fsdNewBefore.Changed(TRUE);
fsdNewBefore.WriteSD();
}
}
else
{
err.SysMsgWrite(ErrS,rc,DCT_MSG_RENAME_DIR_FAILED_SSD,oldName,otherName,rc);
}
}
}
}
return rc;
}
DWORD
CopyDirectoryTree(
WCHAR const * targetDirectory, // in - target to copy files/dirs to
WCHAR const * sourceDirectory // in - source directory to copy files/dirs from
)
{
DWORD rc = 0;
HANDLE hFind;
WIN32_FIND_DATA fDat;
WCHAR sourceWC[MAX_PATH];
WCHAR sourceFile[MAX_PATH];
WCHAR targetFile[MAX_PATH];
safecopy(sourceWC,sourceDirectory);
UStrCpy(sourceWC + UStrLen(sourceWC),L"\\*.*");
// Loop through the items in the source directory
hFind = FindFirstFile(sourceWC,&fDat);
if ( hFind != INVALID_HANDLE_VALUE )
{
do {
if ( UStrICmp(fDat.cFileName,L".") && UStrICmp(fDat.cFileName,L"..") )
{
//build the source and target filenames
swprintf(sourceFile,L"%s\\%s",sourceDirectory,fDat.cFileName);
swprintf(targetFile,L"%s\\%s",targetDirectory,fDat.cFileName);
TFileSD fileSD(sourceFile);
if ( fDat.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY )
{
// copy the directory, update its SD
SECURITY_ATTRIBUTES sAttr;
sAttr.nLength = (sizeof sAttr);
sAttr.bInheritHandle = FALSE;
sAttr.lpSecurityDescriptor = NULL;
if ( fileSD.GetSecurity() )
{
sAttr.lpSecurityDescriptor = fileSD.GetSecurity()->MakeRelSD();
}
if (! CreateDirectoryEx(sourceFile,targetFile,&sAttr) )
{
err.SysMsgWrite(ErrE,rc,DCT_MSG_COPY_DIR_FAILED_SSD,sourceFile,targetFile,rc);
}
else
{
// Recursively process the contents of the directory
CopyDirectoryTree(targetFile,sourceFile);
}
}
else
{
// Copy the file
if (! CopyFile(sourceFile,targetFile,TRUE) )
{
rc = GetLastError();
err.SysMsgWrite(ErrW,rc,DCT_MSG_COPY_FILE_FAILED_SSD,sourceFile,targetFile,rc);
}
else
{
// copy the security descriptor for the file
TFileSD targetSD(targetFile);
targetSD.CopyAccessData(&fileSD);
targetSD.WriteSD();
}
}
}
if (! FindNextFile(hFind,&fDat) )
{
rc = GetLastError();
}
} while ( ! rc );
if ( rc != ERROR_NO_MORE_FILES )
{
err.SysMsgWrite(ErrE,rc,DCT_MSG_FILE_ENUM_FAILED_SD,sourceDirectory,rc);
}
FindClose(hFind);
}
return rc;
}
DWORD
CreateNewProfileDirectory(
WCHAR const * oldDirectoryName, // in - directory name for old profile path
WCHAR const * newAccountName, // in - target account name for new profile directory
WCHAR * profileDirectory // out- directory name created for new profile directory
)
{
DWORD rc = 0;
WCHAR targetDir[MAX_PATH];
WCHAR targetDirWithSuffix[MAX_PATH];
WCHAR * slashNdx;
TFileSD fileSD(const_cast<WCHAR*>(oldDirectoryName));
SECURITY_ATTRIBUTES sAttr;
sAttr.nLength = (sizeof sAttr);
sAttr.bInheritHandle = FALSE;
sAttr.lpSecurityDescriptor = NULL;
if ( fileSD.GetSecurity() )
{
sAttr.lpSecurityDescriptor = fileSD.GetSecurity()->MakeRelSD();
}
safecopy(targetDir,oldDirectoryName);
if ( targetDir[UStrLen(targetDir)-1] == L'\\' )
targetDir[UStrLen(targetDir)-1] = 0;
slashNdx = wcsrchr(targetDir,L'\\');
if ( slashNdx )
{
UStrCpy(slashNdx+1,newAccountName);
// try to create the directory
if ( ! CreateDirectory(targetDir,NULL) )
{
rc = GetLastError();
int ndx = 0;
do {
if ( ndx >= 1000 ) // abort
{
rc = ERROR_ALREADY_EXISTS;
break;
}
if ( rc != ERROR_ALREADY_EXISTS )
break;
swprintf(targetDirWithSuffix,L"%s[%03d]",targetDir,ndx);
if ( ! CreateDirectoryEx(oldDirectoryName,targetDirWithSuffix,&sAttr) )
{
rc = GetLastError();
}
else
{
rc = 0;
}
ndx++;
} while ( rc == ERROR_ALREADY_EXISTS );
if (! rc )
UStrCpy(profileDirectory,targetDirWithSuffix);
}
else
{
UStrCpy(profileDirectory,targetDir);
}
}
else
{
rc = ERROR_INVALID_NAME;
}
return rc;
}
DWORD
UpdateProfilePermissions(
WCHAR const * path, // in - path for directory to update
SecurityTranslatorArgs * globalArgs, // in - path for overall job
TRidNode * pNode // in - account to translate
)
{
DWORD rc = 0;
SecurityTranslatorArgs localArgs;
TSDResolveStats stat(localArgs.Cache());
BOOL bUseMapFile = globalArgs->UsingMapFile();
// set-up the parameters for the translation
localArgs.Cache()->CopyDomainInfo(globalArgs->Cache());
localArgs.Cache()->ToSorted();
if (!bUseMapFile)
{
localArgs.SetUsingMapFile(FALSE);
localArgs.Cache()->InsertLast(pNode->GetAcctName(),pNode->SrcRid(),pNode->GetTargetAcctName(),pNode->TgtRid(),pNode->Type());
}
else
{
localArgs.SetUsingMapFile(TRUE);
localArgs.Cache()->InsertLastWithSid(pNode->GetAcctName(),pNode->GetSrcDomSid(),pNode->GetSrcDomName(),pNode->SrcRid(),
pNode->GetTargetAcctName(),pNode->GetTgtDomSid(),pNode->GetTgtDomName(),pNode->TgtRid(),pNode->Type());
}
localArgs.TranslateFiles(TRUE);
localArgs.SetTranslationMode(globalArgs->TranslationMode());
localArgs.SetWriteChanges(!globalArgs->NoChange());
localArgs.PathList()->AddPath(const_cast<WCHAR*>(path),0);
rc = ResolveAll(&localArgs,&stat);
return rc;
}
// if the specified node is a normal share, this attempts to convert it to a path
// using the administrative shares
void
BuildAdminPathForShare(
WCHAR const * sharePath, // in -
WCHAR * adminShare
)
{
// if all else fails, return the same name as specified in the node
UStrCpy(adminShare,sharePath);
SHARE_INFO_502 * shInfo = NULL;
DWORD rc = 0;
WCHAR shareName[LEN_Path];
WCHAR * slash = NULL;
WCHAR server[LEN_Path];
safecopy(server,sharePath);
// split out just the server name
slash = wcschr(server+3,L'\\');
if ( slash )
{
(*slash) = 0;
}
// now get just the share name
UStrCpy(shareName,sharePath + UStrLen(server) +1);
slash = wcschr(shareName,L'\\');
if ( slash )
*slash = 0;
rc = NetShareGetInfo(server,shareName,502,(LPBYTE*)&shInfo);
if ( ! rc )
{
if ( *shInfo->shi502_path )
{
// build the administrative path name for the share
UStrCpy(adminShare,server);
UStrCpy(adminShare + UStrLen(adminShare),L"\\");
UStrCpy(adminShare + UStrLen(adminShare),shInfo->shi502_path);
WCHAR * colon = wcschr(adminShare,L':');
if ( colon )
{
*colon = L'$';
UStrCpy(adminShare + UStrLen(adminShare),L"\\");
UStrCpy(adminShare + UStrLen(adminShare),slash+1);
}
else
{
// something went wrong -- revert to the given path
UStrCpy(adminShare,sharePath);
}
}
NetApiBufferFree(shInfo);
}
}
DWORD
CopyProfileDirectoryAndTranslate(
WCHAR const * directory, // in - directory path for profile
WCHAR * directoryOut, // out- new Profile Path (including environment variables)
TRidNode * pNode, // in - node for account being translated
SecurityTranslatorArgs * stArgs, // in - translation settings
TSDResolveStats * stat // in - stats on items modified
)
{
DWORD rc = 0;
WCHAR fullPath[MAX_PATH];
WCHAR targetPath[MAX_PATH];
WCHAR profileName[MAX_PATH];
WCHAR targetAcctName[MAX_PATH];
WCHAR sourceDomName[MAX_PATH];
HANDLE hFind;
WIN32_FIND_DATA fDat;
BOOL bTranslateDirOnly = FALSE;
rc = ExpandEnvironmentStrings(directory,fullPath,DIM(fullPath));
if ( !rc )
{
rc = GetLastError();
err.SysMsgWrite(ErrE,rc,DCT_MSG_EXPAND_STRINGS_FAILED_SD,directory,rc);
}
else
{
// Create a new directory for the target profile
// Get the account name for target user
wcscpy(targetAcctName, pNode->GetTargetAcctName());
if ( wcslen(targetAcctName) == 0 )
{
// if target user name not specified then use the source name.
wcscpy(targetAcctName, pNode->GetAcctName());
}
//stArgs->SetTranslationMode(ADD_SECURITY);
// We are changing our stratergy. We are not going to copy the profile directories anymore.
// we will be reACLing the directories and the Registry instead.
/*
rc = CreateNewProfileDirectory(fullPath, targetAcctName,targetPath);
if ( ! rc )
{
rc = CopyDirectoryTree(targetPath,fullPath);
}
*/
BuildAdminPathForShare(fullPath,targetPath);
wcscpy(sourceDomName, const_cast<WCHAR*>(stArgs->Cache()->GetSourceDomainName()));
//if we are using a sID mapping file, try to get the src domain name from this node's information
wcscpy(sourceDomName, pNode->GetSrcDomName());
// Look for profile files in the target directory
// look for NTUser.MAN
swprintf(profileName,L"%s\\NTUser.MAN",targetPath);
hFind = FindFirstFile(profileName,&fDat);
if ( hFind != INVALID_HANDLE_VALUE )
{
err.MsgWrite(0,DCT_MSG_TRANSLATING_NTUSER_MAN_S,targetAcctName);
rc = TranslateUserProfile(profileName,stArgs,stArgs->Cache(),stat, pNode->GetAcctName(), sourceDomName);
FindClose(hFind);
}
else
{
// check for NTUser.DAT
swprintf(profileName,L"%s\\NTUser.DAT",targetPath);
hFind = FindFirstFile(profileName,&fDat);
if ( hFind != INVALID_HANDLE_VALUE )
{
err.MsgWrite(0,DCT_MSG_TRANSLATING_NTUSER_BAT_S,targetAcctName);
rc = TranslateUserProfile(profileName,stArgs,stArgs->Cache(),stat,pNode->GetAcctName(), sourceDomName);
FindClose(hFind);
}
else
{
err.MsgWrite(ErrS,DCT_MSG_PROFILE_REGHIVE_NOT_FOUND_SS,targetAcctName,targetPath);
bTranslateDirOnly = TRUE; //set falg to atleast change permissins on the share dir
rc = 2; // File not found
}
}
if ((!rc) || (bTranslateDirOnly))
rc = UpdateProfilePermissions(targetPath,stArgs,pNode);
wcscpy(directoryOut, fullPath);
}
return rc;
}
DWORD
TranslateLocalProfiles(
SecurityTranslatorArgs * stArgs, // in - translation settings
TSDRidCache * cache, // in - translation table
TSDResolveStats * stat // in - stats on items modified
)
{
DWORD rc = 0;
WCHAR keyName[MAX_PATH];
DWORD lenKeyName = DIM(keyName);
TRegKey keyProfiles;
BOOL bUseMapFile = stArgs->UsingMapFile();
rc = keyProfiles.Open(L"SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\ProfileList",HKEY_LOCAL_MACHINE);
if ( ! rc )
{
// get the number of subkeys
// enumerate the subkeys
DWORD ndx;
DWORD nSubKeys = 0;
rc = RegQueryInfoKey(keyProfiles.KeyGet(),NULL,0,NULL,&nSubKeys,NULL,NULL,NULL,NULL,NULL,NULL,NULL);
if ( ! rc )
{
// construct a list containing the sub-keys
PSID * pSids = new PSID[nSubKeys];
for ( ndx = nSubKeys - 1 ; (long)ndx >= 0 ; ndx-- )
{
rc = keyProfiles.SubKeyEnum(ndx,keyName,lenKeyName);
if ( rc )
break;
pSids[ndx] = SidFromString(keyName);
}
if ( ! rc )
{
// process each profile
for ( ndx = 0 ; ndx < nSubKeys ; ndx++ )
{
do // once
{
if ( ! pSids[ndx] )
continue;
// see if this user needs to be translated
TRidNode * pNode = NULL;
if (!bUseMapFile)
pNode = (TRidNode*)cache->Lookup(pSids[ndx]);
else
pNode = (TRidNode*)cache->LookupWODomain(pSids[ndx]);
if ( pNode == (TRidNode *)-1 )
pNode = NULL;
if ( pNode && pNode->IsValidOnTgt() ) // need to translate this one
{
PSID pSidTgt = NULL;
WCHAR strSourceSid[200];
WCHAR strTargetSid[200];
DWORD dimSid = DIM(strSourceSid);
TRegKey srcKey;
TRegKey tgtKey;
DWORD disposition;
WCHAR keyPath[MAX_PATH];
WCHAR targetPath[MAX_PATH];
DWORD lenValue;
DWORD typeValue;
if (!bUseMapFile)
pSidTgt = cache->GetTgtSid(pSids[ndx]);
else
pSidTgt = cache->GetTgtSidWODomain(pSids[ndx]);
GetTextualSid(pSids[ndx],strSourceSid,&dimSid);
dimSid = DIM(strTargetSid);
GetTextualSid(pSidTgt,strTargetSid,&dimSid);
rc = srcKey.Open(strSourceSid,&keyProfiles);
if ( rc )
{
err.SysMsgWrite(ErrS,rc,DCT_MSG_PROFILE_ENTRY_OPEN_FAILED_SD,pNode->GetAcctName(),rc );
break;
}
if ( (stArgs->TranslationMode() == ADD_SECURITY) || (stArgs->TranslationMode() == REPLACE_SECURITY) )
{
// make a copy of this registry key, so the profile will refer to the new user
if ( ! stArgs->NoChange() )
{
rc = tgtKey.Create(strTargetSid,&keyProfiles,&disposition);
}
else
{
// We need to see if the key already exists or not and set the DISPOSITION accordingly.
rc = tgtKey.OpenRead(strTargetSid, &keyProfiles);
if ( rc )
{
disposition = REG_CREATED_NEW_KEY;
rc = 0;
}
tgtKey.Close();
}
if ( rc )
{
err.SysMsgWrite(ErrS,rc,DCT_MSG_PROFILE_CREATE_ENTRY_FAILED_SD,pNode->GetTargetAcctName(),rc);
break;
}
if ( disposition == REG_CREATED_NEW_KEY || (stArgs->TranslationMode() == REPLACE_SECURITY))
{
// copy the entries from the source key
if ( ! stArgs->NoChange() )
{
rc = tgtKey.HiveCopy(&srcKey);
}
else
{
rc = 0;
tgtKey = srcKey;
}
if ( rc )
{
err.SysMsgWrite(ErrS,rc,DCT_MSG_COPY_PROFILE_FAILED_SSD,pNode->GetAcctName(),pNode->GetTargetAcctName(),rc);
break;
}
// now get the profile path ...
lenValue = (sizeof keyPath);
rc = tgtKey.ValueGet(L"ProfileImagePath",(void *)keyPath,&lenValue,&typeValue);
if ( rc )
{
err.SysMsgWrite(ErrS,rc,DCT_MSG_GET_PROFILE_PATH_FAILED_SD,pNode->GetAcctName(),rc);
break;
}
//copy the profile directory and its contents, and translate the profile registry hive itself
rc = CopyProfileDirectoryAndTranslate(keyPath,targetPath,pNode,stArgs,stat);
if ( rc )
{
// Since the translation failed and we created the key we should delete it.
if ( disposition == REG_CREATED_NEW_KEY )
{
if ( ! stArgs->NoChange() )
keyProfiles.SubKeyDel(strTargetSid);
}
break;
}
// Update the ProfileImagePath key
if ( !stArgs->NoChange() )
rc = tgtKey.ValueSet(L"ProfileImagePath",(void*)targetPath,(1+UStrLen(targetPath)) * (sizeof WCHAR),typeValue);
else
rc = 0;
if ( rc )
{
err.SysMsgWrite(ErrS,rc,DCT_MSG_SET_PROFILE_PATH_FAILED_SD,pNode->GetTargetAcctName(),rc);
break;
}
// update the SID property
if ( !stArgs->NoChange() )
rc = tgtKey.ValueSet(L"Sid",(void*)pSidTgt,GetLengthSid(pSidTgt),REG_BINARY);
else
rc = 0;
if ( rc )
{
rc = GetLastError();
err.SysMsgWrite(ErrS,rc,DCT_MSG_UPDATE_PROFILE_SID_FAILED_SD,pNode->GetTargetAcctName(),rc);
break;
}
}
else
{
err.MsgWrite(ErrW,DCT_MSG_PROFILE_EXISTS_S,pNode->GetTargetAcctName());
break;
}
}
if ( stArgs->TranslationMode() != ADD_SECURITY )
{
// delete the old registry key
if ( ! stArgs->NoChange() )
rc = keyProfiles.SubKeyDel(strSourceSid);
else
rc = 0;
if ( rc )
{
err.SysMsgWrite(ErrS,rc,DCT_MSG_DELETE_PROFILE_FAILED_SD,pNode->GetAcctName(),rc);
break;
}
else
{
err.MsgWrite(0, DCT_MSG_DELETED_PROFILE_S, pNode->GetAcctName());
}
}
}
} while ( FALSE );
}
// clean up the list
for ( ndx = 0 ; ndx < nSubKeys ; ndx++ )
{
if ( pSids[ndx] )
FreeSid(pSids[ndx]);
pSids[ndx] = NULL;
}
delete [] pSids;
}
}
if ( rc && rc != ERROR_NO_MORE_ITEMS )
{
err.SysMsgWrite(ErrS,rc,DCT_MSG_ENUM_PROFILES_FAILED_D,rc);
}
}
else
{
err.SysMsgWrite(ErrS,rc,DCT_MSG_OPEN_PROFILELIST_FAILED_D,rc);
}
return rc;
}
DWORD
TranslateRemoteProfile(
WCHAR const * sourceProfilePath, // in - source profile path
WCHAR * targetProfilePath, // out- new profile path for target account
WCHAR const * sourceName, // in - name of source account
WCHAR const * targetName, // in - name of target account
WCHAR const * srcDomain, // in - source domain
WCHAR const * tgtDomain, // in - target domain
IIManageDB * pDb, // in - pointer to DB object
long lActionID, // in - action ID of this migration
PSID sourceSid, // in - source sid from MoveObj2K
BOOL bNoWriteChanges // in - No Change mode.
)
{
DWORD rc = 0;
BYTE srcSid[LEN_SID];
PSID tgtSid[LEN_SID];
SecurityTranslatorArgs stArgs;
TSDResolveStats stat(stArgs.Cache());
TRidNode * pNode = NULL;
WCHAR domain[LEN_Domain];
DWORD lenDomain = DIM(domain);
DWORD lenSid = DIM(srcSid);
DWORD srcRid=0;
DWORD tgtRid=0;
SID_NAME_USE snu;
IVarSetPtr pVs(__uuidof(VarSet));
IUnknown * pUnk = NULL;
HRESULT hr = S_OK;
WCHAR sActionInfo[MAX_PATH];
_bstr_t sSSam;
long lrid;
stArgs.Cache()->SetSourceAndTargetDomains(srcDomain,tgtDomain);
if ( stArgs.Cache()->IsInitialized() )
{
// Get the source account's rid
if (! LookupAccountName(stArgs.Cache()->GetSourceDCName(),sourceName,srcSid,&lenSid,domain,&lenDomain,&snu) )
{
rc = GetLastError();
}
else
{
if ( !UStrICmp(domain,srcDomain) )
{
PUCHAR pCount = GetSidSubAuthorityCount(srcSid);
if ( pCount )
{
DWORD nSub = (DWORD)(*pCount) - 1;
DWORD * pRid = GetSidSubAuthority(srcSid,nSub);
if ( pRid )
{
srcRid = *pRid;
}
}
}
}
//if we couldn't get the src Rid, we are likely doing an intra-forest migration.
//In this case we will lookup the src Rid in the Migrated Objects table
if (!srcRid)
{
CopySid(GetLengthSid(srcSid), srcSid , sourceSid);
hr = pVs->QueryInterface(IID_IUnknown, (void**)&pUnk);
if ( SUCCEEDED(hr) )
hr = pDb->raw_GetMigratedObjects(lActionID, &pUnk);
if ( SUCCEEDED(hr) )
{
long lCnt = pVs->get("MigratedObjects");
bool bFound = false;
for ( long l = 0; (l < lCnt) && (!bFound); l++)
{
wsprintf(sActionInfo, L"MigratedObjects.%d.%s", l, GET_STRING(DB_SourceSamName));
sSSam = pVs->get(sActionInfo);
if (_wcsicmp(sourceName, (WCHAR*)sSSam) == 0)
{
wsprintf(sActionInfo, L"MigratedObjects.%d.%s", l, GET_STRING(DB_SourceRid));
lrid = pVs->get(sActionInfo);
srcRid = (DWORD)lrid;
bFound = true;
}
}
}
}
lenSid = DIM(tgtSid);
lenDomain = DIM(domain);
// Get the target account's rid
if (! LookupAccountName(stArgs.Cache()->GetTargetDCName(),targetName,tgtSid,&lenSid,domain,&lenDomain,&snu) )
{
rc = GetLastError();
}
else
{
if ( !UStrICmp(domain,tgtDomain) )
{
PUCHAR pCount = GetSidSubAuthorityCount(tgtSid);
if ( pCount )
{
DWORD nSub = (DWORD)(*pCount) - 1;
DWORD * pRid = GetSidSubAuthority(tgtSid,nSub);
if ( pRid )
{
tgtRid = *pRid;
}
}
}
}
}
if ( ((srcRid && tgtRid) || !stArgs.NoChange()) && (!bNoWriteChanges) )
{
stArgs.Cache()->InsertLast(const_cast<WCHAR * const>(sourceName), srcRid, const_cast<WCHAR * const>(targetName), tgtRid);
pNode = (TRidNode*)stArgs.Cache()->Lookup(srcSid);
if ( pNode )
{
// Set up the security translation parameters
stArgs.SetTranslationMode(ADD_SECURITY);
stArgs.TranslateFiles(FALSE);
stArgs.TranslateUserProfiles(TRUE);
stArgs.SetWriteChanges(!bNoWriteChanges);
//copy the profile directory and its contents, and translate the profile registry hive itself
rc = CopyProfileDirectoryAndTranslate(sourceProfilePath,targetProfilePath,pNode,&stArgs,&stat);
}
}
return rc;
}
HRESULT UpdateMappedDrives(WCHAR * sSourceSam, WCHAR * sSourceDomain, WCHAR * sRegistryKey)
{
TRegKey reg;
TRegKey regDrive;
DWORD rc = 0;
WCHAR netKey[LEN_Path];
int len = LEN_Path;
int ndx = 0;
HRESULT hr = S_OK;
WCHAR sValue[LEN_Path];
WCHAR sAcct[LEN_Path];
WCHAR keyname[LEN_Path];
// Build the account name string that we need to check for
wsprintf(sAcct, L"%s\\%s", (WCHAR*) sSourceDomain, (WCHAR*) sSourceSam);
// Get the path to the Network subkey for this users profile.
wsprintf(netKey, L"%s\\%s", (WCHAR*) sRegistryKey, L"Network");
rc = reg.Open(netKey, HKEY_USERS);
if ( !rc )
{
while ( !reg.SubKeyEnum(ndx, keyname, len) )
{
rc = regDrive.Open(keyname, reg.KeyGet());
if ( !rc )
{
// Get the user name value that we need to check.
rc = regDrive.ValueGetStr(L"UserName", sValue, LEN_Path);
if ( !rc )
{
if ( !_wcsicmp(sAcct, sValue) )
{
// Found this account name in the mapped drive user name.so we will set the key to ""
regDrive.ValueSetStr(L"UserName", L"");
err.MsgWrite(0, DCT_MSG_RESET_MAPPED_CREDENTIAL_S, sValue);
}
}
else
hr = HRESULT_FROM_WIN32(GetLastError());
regDrive.Close();
}
else
hr = HRESULT_FROM_WIN32(GetLastError());
ndx++;
}
reg.Close();
}
else
hr = HRESULT_FROM_WIN32(GetLastError());
return hr;
}