3312 lines
96 KiB
C
3312 lines
96 KiB
C
|
//*************************************************************
|
||
|
//
|
||
|
// SETUP.C - API's used by setup to create groups/items
|
||
|
//
|
||
|
// Microsoft Confidential
|
||
|
// Copyright (c) Microsoft Corporation 1995
|
||
|
// All rights reserved
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
#include <uenv.h>
|
||
|
#include <sddl.h> // ConvertStringSecurityDescriptorToSecurityDescriptor
|
||
|
|
||
|
|
||
|
// See ConvertStringSecurityDescriptorToSecurityDescriptor documentation
|
||
|
// for a description of the string security descriptor format.
|
||
|
//
|
||
|
// These ACLs are setup to allow
|
||
|
// System, Administrators, Creator-Owner: Full Control
|
||
|
// Power Users: Modify (RWXD)
|
||
|
// Users: Read (RX)
|
||
|
// Users: Write (folders only)
|
||
|
//
|
||
|
// The combination of "Users: Write (folders only)" and the Creator-Owner ACE
|
||
|
// means that restricted users can create subfolders and files, and have full
|
||
|
// control to files that they create, but they cannot modify or delete files
|
||
|
// created by someone else.
|
||
|
|
||
|
const TCHAR c_szCommonDocumentsACL[] = TEXT("D:P(A;CIOI;GA;;;SY)(A;CIOI;GA;;;BA)(A;CIOIIO;GA;;;CO)(A;CIOI;GRGWGXSD;;;PU)(A;CIOI;GRGX;;;BU)(A;CI;0x116;;;BU)");
|
||
|
const TCHAR c_szCommonAppDataACL[] = TEXT("D:P(A;CIOI;GA;;;SY)(A;CIOI;GA;;;BA)(A;CIOIIO;GA;;;CO)(A;CIOI;GRGWGXSD;;;PU)(A;CIOI;GRGX;;;BU)(A;CI;0x116;;;BU)");
|
||
|
|
||
|
|
||
|
BOOL PrependPath(LPCTSTR szFile, LPTSTR szResult);
|
||
|
BOOL CheckProfile (LPTSTR lpProfilesDir, LPTSTR lpProfileValue,
|
||
|
LPTSTR lpProfileName);
|
||
|
void HideSpecialProfiles(void);
|
||
|
void SetAclForSystemProfile(PSID pSidSystem, LPTSTR szExpandedProfilePath);
|
||
|
|
||
|
//*************************************************************
|
||
|
//
|
||
|
// CreateGroup()
|
||
|
//
|
||
|
// Purpose: Creates a program group (sub-directory)
|
||
|
//
|
||
|
// Parameters: lpGroupName - Name of group
|
||
|
// bCommonGroup - Common or Personal group
|
||
|
//
|
||
|
// Return: TRUE if successful
|
||
|
// FALSE if an error occurs
|
||
|
//
|
||
|
// Comments:
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 8/08/95 ericflo Created
|
||
|
// 3/29/00 AlexArm Split into CreateGroup
|
||
|
// and CreateGroupEx
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
BOOL WINAPI CreateGroup(LPCTSTR lpGroupName, BOOL bCommonGroup)
|
||
|
{
|
||
|
//
|
||
|
// Call CreateGroupEx with no name.
|
||
|
//
|
||
|
return CreateGroupEx( lpGroupName, bCommonGroup, NULL, 0 );
|
||
|
}
|
||
|
|
||
|
//*************************************************************
|
||
|
//
|
||
|
// CreateGroupEx()
|
||
|
//
|
||
|
// Purpose: Creates a program group (sub-directory) and sets
|
||
|
// the localized name for the program group
|
||
|
//
|
||
|
// Parameters: lpGroupName - Name of group
|
||
|
// bCommonGroup - Common or Personal group
|
||
|
// lpResourceModuleName - Name of the resource module.
|
||
|
// uResourceID - Resource ID for the MUI display name.
|
||
|
//
|
||
|
// Return: TRUE if successful
|
||
|
// FALSE if an error occurs
|
||
|
//
|
||
|
// Comments:
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 8/08/95 ericflo Created
|
||
|
// 3/29/00 AlexArm Split into CreateGroup
|
||
|
// and CreateGroupEx
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
BOOL WINAPI CreateGroupEx(LPCWSTR lpGroupName, BOOL bCommonGroup,
|
||
|
LPCWSTR lpResourceModuleName, UINT uResourceID)
|
||
|
{
|
||
|
TCHAR szDirectory[MAX_PATH];
|
||
|
LPTSTR lpEnd;
|
||
|
LPTSTR lpAdjustedGroupName;
|
||
|
int csidl;
|
||
|
PSHELL32_API pShell32Api;
|
||
|
DWORD dwResult;
|
||
|
|
||
|
//
|
||
|
// Validate parameters
|
||
|
//
|
||
|
|
||
|
if (!lpGroupName || !(*lpGroupName)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateGroupEx: Failing due to NULL group name.")));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateGroupEx: Entering with <%s>."), lpGroupName));
|
||
|
|
||
|
|
||
|
if ( ERROR_SUCCESS != LoadShell32Api( &pShell32Api ) ) {
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Extract the CSIDL (if any) from lpGroupName
|
||
|
//
|
||
|
|
||
|
csidl = ExtractCSIDL(lpGroupName, &lpAdjustedGroupName);
|
||
|
|
||
|
if (-1 != csidl)
|
||
|
{
|
||
|
//
|
||
|
// Use this csidl
|
||
|
// WARNING: if a CSIDL is provided, the bCommonGroup flag is meaningless
|
||
|
//
|
||
|
|
||
|
DebugMsg((DM_VERBOSE,
|
||
|
TEXT("CreateGroupEx: CSIDL = <0x%x> contained in lpGroupName replaces csidl."),
|
||
|
csidl));
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
//
|
||
|
// Default to CSIDL_..._PROGRAMS
|
||
|
//
|
||
|
csidl = bCommonGroup ? CSIDL_COMMON_PROGRAMS : CSIDL_PROGRAMS;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Get the programs directory
|
||
|
//
|
||
|
|
||
|
if (!GetSpecialFolderPath (csidl, szDirectory)) {
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Now append the requested directory
|
||
|
//
|
||
|
|
||
|
lpEnd = CheckSlash (szDirectory);
|
||
|
lstrcpy (lpEnd, lpAdjustedGroupName);
|
||
|
|
||
|
|
||
|
//
|
||
|
// Create the group (directory)
|
||
|
//
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateGroupEx: Calling CreatedNestedDirectory with <%s>"),
|
||
|
szDirectory));
|
||
|
|
||
|
if (!CreateNestedDirectory(szDirectory, NULL)) {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateGroupEx: CreatedNestedDirectory failed.")));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// If the localized name is specified, set it.
|
||
|
//
|
||
|
if (lpResourceModuleName != NULL) {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateGroupEx: Calling SHSetLocalizedName.")));
|
||
|
dwResult = pShell32Api->pfnShSetLocalizedName(szDirectory,
|
||
|
lpResourceModuleName, uResourceID);
|
||
|
if (dwResult != ERROR_SUCCESS) {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateGroupEx: SHSetLocalizedName failed <0x%x>."),
|
||
|
dwResult));
|
||
|
return FALSE;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Success
|
||
|
//
|
||
|
|
||
|
pShell32Api->pfnShChangeNotify (SHCNE_MKDIR, SHCNF_PATH, szDirectory, NULL);
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateGroupEx: Leaving successfully.")));
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
|
||
|
//*************************************************************
|
||
|
//
|
||
|
// DeleteGroup()
|
||
|
//
|
||
|
// Purpose: Deletes a program group (sub-directory)
|
||
|
//
|
||
|
// Parameters: lpGroupName - Name of group
|
||
|
// bCommonGroup - Common or Personal group
|
||
|
//
|
||
|
// Return: TRUE if successful
|
||
|
// FALSE if an error occurs
|
||
|
//
|
||
|
// Comments:
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 8/10/95 ericflo Created
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
BOOL WINAPI DeleteGroup(LPCTSTR lpGroupName, BOOL bCommonGroup)
|
||
|
{
|
||
|
TCHAR szDirectory[MAX_PATH];
|
||
|
LPTSTR lpEnd;
|
||
|
LPTSTR lpAdjustedGroupName;
|
||
|
int csidl;
|
||
|
PSHELL32_API pShell32Api;
|
||
|
|
||
|
//
|
||
|
// Validate parameters
|
||
|
//
|
||
|
|
||
|
if (!lpGroupName || !(*lpGroupName)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("DeleteGroup: Failing due to NULL group name.")));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("DeleteGroup: Entering with <%s>."), lpGroupName));
|
||
|
|
||
|
if (ERROR_SUCCESS != LoadShell32Api( &pShell32Api ) ) {
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Extract the CSIDL (if any) from lpGroupName
|
||
|
//
|
||
|
|
||
|
csidl = ExtractCSIDL(lpGroupName, &lpAdjustedGroupName);
|
||
|
|
||
|
if (-1 != csidl)
|
||
|
{
|
||
|
//
|
||
|
// Use this csidl
|
||
|
// WARNING: if a CSIDL is provided, the bCommonGroup flag is meaningless
|
||
|
//
|
||
|
|
||
|
DebugMsg((DM_VERBOSE,
|
||
|
TEXT("DeleteGroup: CSIDL = <0x%x> contained in lpGroupName replaces csidl."),
|
||
|
csidl));
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
//
|
||
|
// Default to CSIDL_..._PROGRAMS
|
||
|
//
|
||
|
|
||
|
csidl = bCommonGroup ? CSIDL_COMMON_PROGRAMS : CSIDL_PROGRAMS;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Get the programs directory
|
||
|
//
|
||
|
|
||
|
if (!GetSpecialFolderPath (csidl, szDirectory)) {
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Now append the requested directory
|
||
|
//
|
||
|
|
||
|
lpEnd = CheckSlash (szDirectory);
|
||
|
lstrcpy (lpEnd, lpAdjustedGroupName);
|
||
|
|
||
|
|
||
|
//
|
||
|
// Delete the group (directory)
|
||
|
//
|
||
|
|
||
|
if (!Delnode(szDirectory)) {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("DeleteGroup: Delnode failed.")));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Success
|
||
|
//
|
||
|
pShell32Api->pfnShChangeNotify (SHCNE_RMDIR, SHCNF_PATH, szDirectory, NULL);
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("DeleteGroup: Leaving successfully.")));
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
//*************************************************************
|
||
|
//
|
||
|
// CreateLinkFile()
|
||
|
//
|
||
|
// Purpose: Creates a link file in the specified directory
|
||
|
//
|
||
|
// Parameters: cidl - CSIDL_ of a special folder
|
||
|
// lpSubDirectory - Subdirectory of special folder
|
||
|
// lpFileName - File name of item
|
||
|
// lpCommandLine - Command line (including args)
|
||
|
// lpIconPath - Icon path (can be NULL)
|
||
|
// iIconIndex - Index of icon in icon path
|
||
|
// lpWorkingDir - Working directory
|
||
|
// wHotKey - Hot key
|
||
|
// iShowCmd - ShowWindow flag
|
||
|
// lpDescription - Description of the item
|
||
|
//
|
||
|
// Return: TRUE if successful
|
||
|
// FALSE if an error occurs
|
||
|
//
|
||
|
// Comments:
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 3/26/98 ericflo Created
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
BOOL WINAPI CreateLinkFile(INT csidl, LPCTSTR lpSubDirectory,
|
||
|
LPCTSTR lpFileName, LPCTSTR lpCommandLine,
|
||
|
LPCTSTR lpIconPath, int iIconIndex,
|
||
|
LPCTSTR lpWorkingDirectory, WORD wHotKey,
|
||
|
int iShowCmd, LPCTSTR lpDescription)
|
||
|
{
|
||
|
return CreateLinkFileEx(csidl, lpSubDirectory, lpFileName, lpCommandLine,
|
||
|
lpIconPath, iIconIndex, lpWorkingDirectory, wHotKey,
|
||
|
iShowCmd, lpDescription, NULL, 0);
|
||
|
}
|
||
|
//*************************************************************
|
||
|
//
|
||
|
// CreateLinkFileEx()
|
||
|
//
|
||
|
// Purpose: Creates a link file in the specified directory
|
||
|
//
|
||
|
// Parameters: cidl - CSIDL_ of a special folder
|
||
|
// lpSubDirectory - Subdirectory of special folder
|
||
|
// lpFileName - File name of item
|
||
|
// lpCommandLine - Command line (including args)
|
||
|
// lpIconPath - Icon path (can be NULL)
|
||
|
// iIconIndex - Index of icon in icon path
|
||
|
// lpWorkingDir - Working directory
|
||
|
// wHotKey - Hot key
|
||
|
// iShowCmd - ShowWindow flag
|
||
|
// lpDescription - Description of the item
|
||
|
// lpResourceModuleName - Name of the resource module.
|
||
|
// uResourceID - Resource ID for the MUI display name.
|
||
|
//
|
||
|
// Return: TRUE if successful
|
||
|
// FALSE if an error occurs
|
||
|
//
|
||
|
// Comments:
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 3/26/98 ericflo Created
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
|
||
|
BOOL WINAPI CreateLinkFileEx(INT csidl, LPCTSTR lpSubDirectory,
|
||
|
LPCTSTR lpFileName, LPCTSTR lpCommandLine,
|
||
|
LPCTSTR lpIconPath, int iIconIndex,
|
||
|
LPCTSTR lpWorkingDirectory, WORD wHotKey,
|
||
|
int iShowCmd, LPCTSTR lpDescription,
|
||
|
LPCWSTR lpResourceModuleName, UINT uResourceID)
|
||
|
{
|
||
|
TCHAR szItem[MAX_PATH];
|
||
|
TCHAR szArgs[MAX_PATH];
|
||
|
TCHAR szLinkName[MAX_PATH];
|
||
|
TCHAR szPath[MAX_PATH];
|
||
|
LPTSTR lpArgs, lpEnd;
|
||
|
IShellLink *psl;
|
||
|
IPersistFile *ppf;
|
||
|
BOOL bRetVal = FALSE;
|
||
|
HINSTANCE hInstOLE32 = NULL;
|
||
|
PFNCOCREATEINSTANCE pfnCoCreateInstance;
|
||
|
PFNCOINITIALIZE pfnCoInitialize;
|
||
|
PFNCOUNINITIALIZE pfnCoUninitialize;
|
||
|
LPTSTR lpAdjustedSubDir = NULL;
|
||
|
PSHELL32_API pShell32Api;
|
||
|
PSHLWAPI_API pShlwapiApi;
|
||
|
DWORD dwResult;
|
||
|
|
||
|
//
|
||
|
// Verbose output
|
||
|
//
|
||
|
|
||
|
#if DBG
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: Entering.")));
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: csidl = <0x%x>."), csidl));
|
||
|
if (lpSubDirectory && *lpSubDirectory) {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: lpSubDirectory = <%s>."), lpSubDirectory));
|
||
|
}
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: lpFileName = <%s>."), lpFileName));
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: lpCommandLine = <%s>."), lpCommandLine));
|
||
|
|
||
|
if (lpIconPath) {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: lpIconPath = <%s>."), lpIconPath));
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: iIconIndex = <%d>."), iIconIndex));
|
||
|
}
|
||
|
|
||
|
if (lpWorkingDirectory) {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: lpWorkingDirectory = <%s>."), lpWorkingDirectory));
|
||
|
} else {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: Null working directory. Setting to %%HOMEDRIVE%%%%HOMEPATH%%")));
|
||
|
}
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: wHotKey = <%d>."), wHotKey));
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: iShowCmd = <%d>."), iShowCmd));
|
||
|
|
||
|
if (lpDescription) {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: lpDescription = <%s>."), lpDescription));
|
||
|
} else {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: Null description.")));
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
//
|
||
|
// Load a few functions we need
|
||
|
//
|
||
|
|
||
|
hInstOLE32 = LoadLibrary (TEXT("ole32.dll"));
|
||
|
|
||
|
if (!hInstOLE32) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateLinkFileEx: Failed to load ole32 with %d."),
|
||
|
GetLastError()));
|
||
|
goto ExitNoFree;
|
||
|
}
|
||
|
|
||
|
|
||
|
pfnCoCreateInstance = (PFNCOCREATEINSTANCE)GetProcAddress (hInstOLE32,
|
||
|
"CoCreateInstance");
|
||
|
|
||
|
if (!pfnCoCreateInstance) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateLinkFileEx: Failed to find CoCreateInstance with %d."),
|
||
|
GetLastError()));
|
||
|
goto ExitNoFree;
|
||
|
}
|
||
|
|
||
|
pfnCoInitialize = (PFNCOINITIALIZE)GetProcAddress (hInstOLE32,
|
||
|
"CoInitialize");
|
||
|
|
||
|
if (!pfnCoInitialize) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateLinkFileEx: Failed to find CoInitialize with %d."),
|
||
|
GetLastError()));
|
||
|
goto ExitNoFree;
|
||
|
}
|
||
|
|
||
|
pfnCoUninitialize = (PFNCOUNINITIALIZE)GetProcAddress (hInstOLE32,
|
||
|
"CoUninitialize");
|
||
|
|
||
|
if (!pfnCoUninitialize) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateLinkFileEx: Failed to find CoUninitialize with %d."),
|
||
|
GetLastError()));
|
||
|
goto ExitNoFree;
|
||
|
}
|
||
|
|
||
|
if (ERROR_SUCCESS != LoadShell32Api( &pShell32Api ) ) {
|
||
|
goto ExitNoFree;
|
||
|
}
|
||
|
|
||
|
|
||
|
pShlwapiApi = LoadShlwapiApi();
|
||
|
|
||
|
if ( !pShlwapiApi ) {
|
||
|
goto ExitNoFree;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Get the special folder directory
|
||
|
// First check if there is a CSIDL in the subdirectory
|
||
|
//
|
||
|
|
||
|
if (lpSubDirectory && *lpSubDirectory) {
|
||
|
|
||
|
int csidl2 = ExtractCSIDL(lpSubDirectory, &lpAdjustedSubDir);
|
||
|
|
||
|
if (-1 != csidl2)
|
||
|
{
|
||
|
csidl = csidl2;
|
||
|
DebugMsg((DM_VERBOSE,
|
||
|
TEXT("CreateLinkFileEx: CSIDL = <0x%x> contained in lpSubDirectory replaces csidl."),
|
||
|
csidl));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
szLinkName[0] = TEXT('\0');
|
||
|
if (csidl && !GetSpecialFolderPath (csidl, szLinkName)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateLinkFileEx: Failed to get profiles directory.")));
|
||
|
goto ExitNoFree;
|
||
|
}
|
||
|
|
||
|
|
||
|
if (lpAdjustedSubDir && *lpAdjustedSubDir) {
|
||
|
|
||
|
if (szLinkName[0] != TEXT('\0')) {
|
||
|
lpEnd = CheckSlash (szLinkName);
|
||
|
|
||
|
} else {
|
||
|
lpEnd = szLinkName;
|
||
|
}
|
||
|
|
||
|
lstrcpy (lpEnd, lpAdjustedSubDir);
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Create the target directory
|
||
|
//
|
||
|
|
||
|
if (!CreateNestedDirectory(szLinkName, NULL)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateLinkFileEx: Failed to create subdirectory <%s> with %d"),
|
||
|
szLinkName, GetLastError()));
|
||
|
goto ExitNoFree;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Now tack on the filename and extension.
|
||
|
//
|
||
|
|
||
|
lpEnd = CheckSlash (szLinkName);
|
||
|
lstrcpy (lpEnd, lpFileName);
|
||
|
lstrcat (lpEnd, c_szLNK);
|
||
|
|
||
|
|
||
|
//
|
||
|
// Split the command line into the executable name
|
||
|
// and arguments.
|
||
|
//
|
||
|
|
||
|
lstrcpy (szItem, lpCommandLine);
|
||
|
|
||
|
lpArgs = pShlwapiApi->pfnPathGetArgs(szItem);
|
||
|
|
||
|
if (*lpArgs) {
|
||
|
lstrcpy (szArgs, lpArgs);
|
||
|
|
||
|
lpArgs--;
|
||
|
while (*lpArgs == TEXT(' ')) {
|
||
|
lpArgs--;
|
||
|
}
|
||
|
lpArgs++;
|
||
|
*lpArgs = TEXT('\0');
|
||
|
} else {
|
||
|
szArgs[0] = TEXT('\0');
|
||
|
}
|
||
|
|
||
|
pShlwapiApi->pfnPathUnquoteSpaces (szItem);
|
||
|
|
||
|
|
||
|
//
|
||
|
// Create an IShellLink object
|
||
|
//
|
||
|
|
||
|
pfnCoInitialize(NULL);
|
||
|
|
||
|
if (FAILED(pfnCoCreateInstance(&CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER,
|
||
|
&IID_IShellLink, (LPVOID*)&psl)))
|
||
|
{
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateLinkFileEx: Could not create instance of IShellLink .")));
|
||
|
goto ExitNoFree;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Query for IPersistFile
|
||
|
//
|
||
|
|
||
|
if (FAILED(psl->lpVtbl->QueryInterface(psl, &IID_IPersistFile, &ppf)))
|
||
|
{
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateLinkFileEx: QueryInterface of IShellLink failed.")));
|
||
|
goto ExitFreePSL;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Set the item information
|
||
|
//
|
||
|
|
||
|
if (lpDescription) {
|
||
|
psl->lpVtbl->SetDescription(psl, lpDescription);
|
||
|
}
|
||
|
|
||
|
PrependPath(szItem, szPath);
|
||
|
psl->lpVtbl->SetPath(psl, szPath);
|
||
|
|
||
|
|
||
|
psl->lpVtbl->SetArguments(psl, szArgs);
|
||
|
if (lpWorkingDirectory) {
|
||
|
psl->lpVtbl->SetWorkingDirectory(psl, lpWorkingDirectory);
|
||
|
} else {
|
||
|
psl->lpVtbl->SetWorkingDirectory(psl, TEXT("%HOMEDRIVE%%HOMEPATH%"));
|
||
|
}
|
||
|
|
||
|
PrependPath(lpIconPath, szPath);
|
||
|
psl->lpVtbl->SetIconLocation(psl, szPath, iIconIndex);
|
||
|
|
||
|
psl->lpVtbl->SetHotkey(psl, wHotKey);
|
||
|
psl->lpVtbl->SetShowCmd(psl, iShowCmd);
|
||
|
|
||
|
|
||
|
//
|
||
|
// If the localized name is specified, set it.
|
||
|
//
|
||
|
|
||
|
if (lpResourceModuleName != NULL) {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: Calling SHSetLocalizedName on link <%s>."), szLinkName));
|
||
|
ppf->lpVtbl->Save(ppf, szLinkName, TRUE);
|
||
|
dwResult = pShell32Api->pfnShSetLocalizedName(szLinkName,
|
||
|
lpResourceModuleName, uResourceID);
|
||
|
if (dwResult != ERROR_SUCCESS) {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: SHSetLocalizedName failed <0x%x>."),
|
||
|
dwResult));
|
||
|
goto ExitFreePSL;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Save the item to disk
|
||
|
//
|
||
|
|
||
|
bRetVal = SUCCEEDED(ppf->lpVtbl->Save(ppf, szLinkName, TRUE));
|
||
|
|
||
|
if (bRetVal) {
|
||
|
pShell32Api->pfnShChangeNotify (SHCNE_CREATE, SHCNF_PATH, szLinkName, NULL);
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Release the IPersistFile object
|
||
|
//
|
||
|
|
||
|
ppf->lpVtbl->Release(ppf);
|
||
|
|
||
|
|
||
|
ExitFreePSL:
|
||
|
|
||
|
//
|
||
|
// Release the IShellLink object
|
||
|
//
|
||
|
|
||
|
psl->lpVtbl->Release(psl);
|
||
|
|
||
|
pfnCoUninitialize();
|
||
|
|
||
|
ExitNoFree:
|
||
|
|
||
|
if (hInstOLE32) {
|
||
|
FreeLibrary (hInstOLE32);
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Finished.
|
||
|
//
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateLinkFileEx: Leaving with status of %d."), bRetVal));
|
||
|
|
||
|
return bRetVal;
|
||
|
}
|
||
|
|
||
|
|
||
|
//*************************************************************
|
||
|
//
|
||
|
// DeleteLinkFile()
|
||
|
//
|
||
|
// Purpose: Deletes the specified link file
|
||
|
//
|
||
|
// Parameters: csidl - CSIDL of a special folder
|
||
|
// lpSubDirectory - Subdirectory of special folder
|
||
|
// lpFileName - File name of item
|
||
|
// bDeleteSubDirectory - Delete the subdirectory if possible
|
||
|
//
|
||
|
// Return: TRUE if successful
|
||
|
// FALSE if an error occurs
|
||
|
//
|
||
|
// Comments:
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 3/26/98 ericflo Created
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
BOOL WINAPI DeleteLinkFile(INT csidl, LPCTSTR lpSubDirectory,
|
||
|
LPCTSTR lpFileName, BOOL bDeleteSubDirectory)
|
||
|
{
|
||
|
TCHAR szLinkName[MAX_PATH];
|
||
|
LPTSTR lpEnd;
|
||
|
LPTSTR lpAdjustedSubDir = NULL;
|
||
|
PSHELL32_API pShell32Api;
|
||
|
|
||
|
//
|
||
|
// Verbose output
|
||
|
//
|
||
|
|
||
|
#if DBG
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("DeleteLinkFile: Entering.")));
|
||
|
DebugMsg((DM_VERBOSE, TEXT("DeleteLinkFile: csidl = 0x%x."), csidl));
|
||
|
if (lpSubDirectory && *lpSubDirectory) {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("DeleteLinkFile: lpSubDirectory = <%s>."), lpSubDirectory));
|
||
|
}
|
||
|
DebugMsg((DM_VERBOSE, TEXT("DeleteLinkFile: lpFileName = <%s>."), lpFileName));
|
||
|
DebugMsg((DM_VERBOSE, TEXT("DeleteLinkFile: bDeleteSubDirectory = %d."), bDeleteSubDirectory));
|
||
|
|
||
|
#endif
|
||
|
|
||
|
if (ERROR_SUCCESS != LoadShell32Api( &pShell32Api ) ) {
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Get the special folder directory
|
||
|
// First check if there is a CSIDL in the subdirectory
|
||
|
//
|
||
|
|
||
|
if (lpSubDirectory && *lpSubDirectory) {
|
||
|
|
||
|
int csidl2 = ExtractCSIDL(lpSubDirectory, &lpAdjustedSubDir);
|
||
|
|
||
|
if (-1 != csidl2)
|
||
|
{
|
||
|
csidl = csidl2;
|
||
|
DebugMsg((DM_VERBOSE,
|
||
|
TEXT("CreateLinkFile: CSIDL = <0x%x> contained in lpSubDirectory replaces csidl."),
|
||
|
csidl));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
szLinkName[0] = TEXT('\0');
|
||
|
if (csidl && !GetSpecialFolderPath (csidl, szLinkName)) {
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
if (lpAdjustedSubDir && *lpAdjustedSubDir) {
|
||
|
if (szLinkName[0] != TEXT('\0')) {
|
||
|
lpEnd = CheckSlash (szLinkName);
|
||
|
} else {
|
||
|
lpEnd = szLinkName;
|
||
|
}
|
||
|
|
||
|
lstrcpy (lpEnd, lpAdjustedSubDir);
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Now tack on the filename and extension.
|
||
|
//
|
||
|
|
||
|
lpEnd = CheckSlash (szLinkName);
|
||
|
lstrcpy (lpEnd, lpFileName);
|
||
|
lstrcat (lpEnd, c_szLNK);
|
||
|
|
||
|
//
|
||
|
// Delete the file
|
||
|
//
|
||
|
|
||
|
if (!DeleteFile (szLinkName)) {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("DeleteLinkFile: Failed to delete <%s>. Error = %d"),
|
||
|
szLinkName, GetLastError()));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
pShell32Api->pfnShChangeNotify (SHCNE_DELETE, SHCNF_PATH, szLinkName, NULL);
|
||
|
|
||
|
//
|
||
|
// Delete the subdirectory if appropriate (and possible).
|
||
|
//
|
||
|
|
||
|
if (bDeleteSubDirectory) {
|
||
|
*(lpEnd-1) = TEXT('\0');
|
||
|
if (RemoveDirectory(szLinkName)) {
|
||
|
pShell32Api->pfnShChangeNotify (SHCNE_RMDIR, SHCNF_PATH, szLinkName, NULL);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Success
|
||
|
//
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("DeleteLinkFile: Leaving successfully.")));
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
//*************************************************************
|
||
|
//
|
||
|
// PrependPath()
|
||
|
//
|
||
|
// Purpose: Expands the given filename to have %systemroot%
|
||
|
// if appropriate
|
||
|
//
|
||
|
// Parameters: lpFile - File to check
|
||
|
// lpResult - Result buffer (MAX_PATH chars in size)
|
||
|
//
|
||
|
// Return: TRUE if successful
|
||
|
// FALSE if an error occurs
|
||
|
//
|
||
|
// Comments:
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 10/11/95 ericflo Created
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
BOOL PrependPath(LPCTSTR lpFile, LPTSTR lpResult)
|
||
|
{
|
||
|
TCHAR szReturn [MAX_PATH];
|
||
|
TCHAR szSysRoot[MAX_PATH];
|
||
|
LPTSTR lpFileName;
|
||
|
DWORD dwSysLen;
|
||
|
|
||
|
|
||
|
//
|
||
|
// Verbose Output
|
||
|
//
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("PrependPath: Entering with <%s>"),
|
||
|
lpFile ? lpFile : TEXT("NULL")));
|
||
|
|
||
|
|
||
|
if (!lpFile || !*lpFile) {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("PrependPath: lpFile is NULL, setting lpResult to a null string")));
|
||
|
*lpResult = TEXT('\0');
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Call SearchPath to find the filename
|
||
|
//
|
||
|
|
||
|
if (!SearchPath (NULL, lpFile, TEXT(".exe"), MAX_PATH, szReturn, &lpFileName)) {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("PrependPath: SearchPath failed with error %d. Using input string"), GetLastError()));
|
||
|
lstrcpy (lpResult, lpFile);
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
|
||
|
UnExpandSysRoot(szReturn, lpResult);
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("PrependPath: Leaving with <%s>"), lpResult));
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
|
||
|
//*************************************************************
|
||
|
//
|
||
|
// SetFilePermissions()
|
||
|
//
|
||
|
// Purpose: Sets the given permissions on a file or directory
|
||
|
//
|
||
|
// Parameters: lpFile - File to set security on
|
||
|
// pszSD - String security descriptor
|
||
|
//
|
||
|
// Return: (BOOL) TRUE if successful
|
||
|
// FALSE if an error occurs
|
||
|
//
|
||
|
// Comments:
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 01/25/2001 jeffreys Created
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
BOOL SetFilePermissions(LPCTSTR lpFile, LPCTSTR pszSD)
|
||
|
{
|
||
|
PSECURITY_DESCRIPTOR pSD;
|
||
|
BOOL bRetVal = FALSE;
|
||
|
|
||
|
if (ConvertStringSecurityDescriptorToSecurityDescriptor(pszSD, SDDL_REVISION, &pSD, NULL))
|
||
|
{
|
||
|
//
|
||
|
// Set the security
|
||
|
//
|
||
|
if (SetFileSecurity (lpFile, DACL_SECURITY_INFORMATION, pSD))
|
||
|
{
|
||
|
bRetVal = TRUE;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
DebugMsg((DM_WARNING, TEXT("SetFilePermissions: SetFileSecurity failed. Error = %d"), GetLastError()));
|
||
|
}
|
||
|
|
||
|
LocalFree(pSD);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
DebugMsg((DM_WARNING, TEXT("SetFilePermissions: ConvertStringSDToSD failed. Error = %d"), GetLastError()));
|
||
|
}
|
||
|
|
||
|
return bRetVal;
|
||
|
}
|
||
|
|
||
|
|
||
|
//*************************************************************
|
||
|
//
|
||
|
// ConvertCommonGroups()
|
||
|
//
|
||
|
// Purpose: Calls grpconv.exe to convert progman common groups
|
||
|
// to Explorer common groups, and create floppy links.
|
||
|
//
|
||
|
// NT 4 appended " (Common)" to the common groups. For
|
||
|
// NT 5, we are going to remove this tag.
|
||
|
//
|
||
|
// Parameters: none
|
||
|
//
|
||
|
// Return: TRUE if successful
|
||
|
// FALSE if an error occurs
|
||
|
// Comments:
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 10/1/95 ericflo Created
|
||
|
// 12/5/96 ericflo Remove common tag
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
BOOL ConvertCommonGroups (void)
|
||
|
{
|
||
|
STARTUPINFO si;
|
||
|
PROCESS_INFORMATION ProcessInformation;
|
||
|
BOOL Result;
|
||
|
TCHAR szBuffer[MAX_PATH];
|
||
|
TCHAR szBuffer2[MAX_PATH];
|
||
|
DWORD dwType, dwSize, dwConvert;
|
||
|
BOOL bRunGrpConv = TRUE;
|
||
|
LONG lResult;
|
||
|
HKEY hKey;
|
||
|
HANDLE hFile;
|
||
|
WIN32_FIND_DATA fd;
|
||
|
TCHAR szCommon[30] = {0};
|
||
|
UINT cchCommon, cchFileName;
|
||
|
LPTSTR lpTag, lpEnd, lpEnd2;
|
||
|
|
||
|
|
||
|
//
|
||
|
// Verbose output
|
||
|
//
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("ConvertCommonGroups: Entering.")));
|
||
|
|
||
|
|
||
|
//
|
||
|
// Check if we have run grpconv before.
|
||
|
//
|
||
|
|
||
|
lResult = RegOpenKeyEx (HKEY_LOCAL_MACHINE,
|
||
|
TEXT("Software\\Program Groups"),
|
||
|
0,
|
||
|
KEY_ALL_ACCESS,
|
||
|
&hKey);
|
||
|
|
||
|
if (lResult == ERROR_SUCCESS) {
|
||
|
|
||
|
dwSize = sizeof(dwConvert);
|
||
|
|
||
|
lResult = RegQueryValueEx (hKey,
|
||
|
TEXT("ConvertedToLinks"),
|
||
|
NULL,
|
||
|
&dwType,
|
||
|
(LPBYTE)&dwConvert,
|
||
|
&dwSize);
|
||
|
|
||
|
if (lResult == ERROR_SUCCESS) {
|
||
|
|
||
|
//
|
||
|
// If dwConvert is 1, then grpconv has run before.
|
||
|
// Don't run it again.
|
||
|
//
|
||
|
|
||
|
if (dwConvert) {
|
||
|
bRunGrpConv = FALSE;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Now set the value to prevent grpconv from running in the future
|
||
|
//
|
||
|
|
||
|
dwConvert = 1;
|
||
|
RegSetValueEx (hKey,
|
||
|
TEXT("ConvertedToLinks"),
|
||
|
0,
|
||
|
REG_DWORD,
|
||
|
(LPBYTE) &dwConvert,
|
||
|
sizeof(dwConvert));
|
||
|
|
||
|
|
||
|
RegCloseKey (hKey);
|
||
|
}
|
||
|
|
||
|
|
||
|
if (bRunGrpConv) {
|
||
|
|
||
|
//
|
||
|
// Initialize process startup info
|
||
|
//
|
||
|
|
||
|
si.cb = sizeof(STARTUPINFO);
|
||
|
si.lpReserved = NULL;
|
||
|
si.lpDesktop = NULL;
|
||
|
si.lpTitle = NULL;
|
||
|
si.dwFlags = 0;
|
||
|
si.lpReserved2 = NULL;
|
||
|
si.cbReserved2 = 0;
|
||
|
|
||
|
|
||
|
//
|
||
|
// Spawn grpconv
|
||
|
//
|
||
|
|
||
|
lstrcpy (szBuffer, TEXT("grpconv -n"));
|
||
|
|
||
|
Result = CreateProcess(
|
||
|
NULL,
|
||
|
szBuffer,
|
||
|
NULL,
|
||
|
NULL,
|
||
|
FALSE,
|
||
|
NORMAL_PRIORITY_CLASS,
|
||
|
NULL,
|
||
|
NULL,
|
||
|
&si,
|
||
|
&ProcessInformation
|
||
|
);
|
||
|
|
||
|
if (!Result) {
|
||
|
DebugMsg((DM_WARNING, TEXT("ConvertCommonGroups: grpconv failed to start due to error %d."), GetLastError()));
|
||
|
return FALSE;
|
||
|
|
||
|
} else {
|
||
|
|
||
|
//
|
||
|
// Wait for up to 2 minutes
|
||
|
//
|
||
|
|
||
|
WaitForSingleObject(ProcessInformation.hProcess, 120000);
|
||
|
|
||
|
//
|
||
|
// Close our handles to the process and thread
|
||
|
//
|
||
|
|
||
|
CloseHandle(ProcessInformation.hProcess);
|
||
|
CloseHandle(ProcessInformation.hThread);
|
||
|
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Loop through all the program groups in the All Users profile
|
||
|
// and remove the " (Common)" tag.
|
||
|
//
|
||
|
|
||
|
LoadString (g_hDllInstance, IDS_COMMON, szCommon, 30);
|
||
|
cchCommon = lstrlen (szCommon);
|
||
|
|
||
|
if (!GetSpecialFolderPath (CSIDL_COMMON_PROGRAMS, szBuffer2)) {
|
||
|
return FALSE;
|
||
|
}
|
||
|
lstrcpy (szBuffer, szBuffer2);
|
||
|
|
||
|
lpEnd = CheckSlash (szBuffer);
|
||
|
lpEnd2 = CheckSlash (szBuffer2);
|
||
|
|
||
|
lstrcpy (lpEnd, c_szStarDotStar);
|
||
|
|
||
|
hFile = FindFirstFile (szBuffer, &fd);
|
||
|
|
||
|
if (hFile != INVALID_HANDLE_VALUE) {
|
||
|
|
||
|
do {
|
||
|
|
||
|
if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
|
||
|
|
||
|
cchFileName = lstrlen (fd.cFileName);
|
||
|
|
||
|
if (cchFileName > cchCommon) {
|
||
|
lpTag = fd.cFileName + cchFileName - cchCommon;
|
||
|
|
||
|
if (!lstrcmpi(lpTag, szCommon)) {
|
||
|
|
||
|
lstrcpy (lpEnd, fd.cFileName);
|
||
|
*lpTag = TEXT('\0');
|
||
|
lstrcpy (lpEnd2, fd.cFileName);
|
||
|
|
||
|
if (MoveFileEx (szBuffer, szBuffer2, MOVEFILE_REPLACE_EXISTING)) {
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("ConvertCommonGroups: Successfully changed group name:")));
|
||
|
DebugMsg((DM_VERBOSE, TEXT("ConvertCommonGroups: Orginial: %s"), szBuffer));
|
||
|
DebugMsg((DM_VERBOSE, TEXT("ConvertCommonGroups: New: %s"), szBuffer2));
|
||
|
|
||
|
} else {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("ConvertCommonGroups: Failed to change group name with error %d."), GetLastError()));
|
||
|
DebugMsg((DM_VERBOSE, TEXT("ConvertCommonGroups: Orginial: %s"), szBuffer));
|
||
|
DebugMsg((DM_VERBOSE, TEXT("ConvertCommonGroups: New: %s"), szBuffer2));
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
} while (FindNextFile(hFile, &fd));
|
||
|
|
||
|
FindClose (hFile);
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Success
|
||
|
//
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("ConvertCommonGroups: Leaving Successfully.")));
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
//*************************************************************
|
||
|
//
|
||
|
// DetermineProfilesLocation()
|
||
|
//
|
||
|
// Purpose: Determines if the profiles directory
|
||
|
// should be in the old NT4 location or
|
||
|
// the new NT5 location
|
||
|
//
|
||
|
// Parameters: none
|
||
|
//
|
||
|
// Return: TRUE if successful
|
||
|
// FALSE if an error occurs
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
BOOL WINAPI DetermineProfilesLocation (BOOL bCleanInstall)
|
||
|
{
|
||
|
TCHAR szDirectory[MAX_PATH];
|
||
|
TCHAR szDest[MAX_PATH];
|
||
|
PTSTR szCurDest;
|
||
|
PCTSTR szLookAhead;
|
||
|
WIN32_FILE_ATTRIBUTE_DATA fad;
|
||
|
DWORD dwSize, dwDisp;
|
||
|
HKEY hKey;
|
||
|
LPTSTR lpEnd;
|
||
|
|
||
|
|
||
|
//
|
||
|
// Check for an unattended entry first
|
||
|
//
|
||
|
|
||
|
if (bCleanInstall) {
|
||
|
|
||
|
if (!ExpandEnvironmentStrings (TEXT("%SystemRoot%\\System32\\$winnt$.inf"), szDirectory,
|
||
|
ARRAYSIZE(szDirectory))) {
|
||
|
|
||
|
DebugMsg((DM_WARNING, TEXT("DetermineProfilesLocation: ExpandEnvironmentStrings failed with error %d"), GetLastError()));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
szDest[0] = TEXT('\0');
|
||
|
GetPrivateProfileString (TEXT("guiunattended"), TEXT("profilesdir"), TEXT(""),
|
||
|
szDest, MAX_PATH, szDirectory);
|
||
|
|
||
|
if (szDest[0] != TEXT('\0')) {
|
||
|
|
||
|
//
|
||
|
// Since $winnt$.inf is an INF, we must strip out %% pairs
|
||
|
//
|
||
|
|
||
|
szCurDest = szDest;
|
||
|
szLookAhead = szDest;
|
||
|
|
||
|
#ifdef UNICODE
|
||
|
while (*szLookAhead) {
|
||
|
if (szLookAhead[0] == L'%' && szLookAhead[1] == L'%') {
|
||
|
szLookAhead++; // pair of %%; skip one char
|
||
|
}
|
||
|
|
||
|
*szCurDest++ = *szLookAhead++;
|
||
|
}
|
||
|
#else
|
||
|
//
|
||
|
// This code path is not compiled so it has not been tested
|
||
|
//
|
||
|
|
||
|
#error Code written but not tested!
|
||
|
|
||
|
while (*szLookAhead) {
|
||
|
|
||
|
if (IsDbcsLeadByte (*szLookAhead)) {
|
||
|
|
||
|
*szCurDest++ = *szLookAhead++; // copy first half of byte pair
|
||
|
|
||
|
} else if (*szLookAhead == '%') {
|
||
|
|
||
|
if (!IsDbcsLeadByte (szLookAhead[1]) && szLookAhead[1] == '%') {
|
||
|
szLookAhead++; // pair of %%; skip one char
|
||
|
}
|
||
|
}
|
||
|
|
||
|
*szCurDest++ = *szLookAhead++;
|
||
|
}
|
||
|
|
||
|
#endif
|
||
|
|
||
|
*szCurDest = 0;
|
||
|
|
||
|
//
|
||
|
// The unattend profile directory exists. We need to set this
|
||
|
// path in the registry
|
||
|
//
|
||
|
|
||
|
if (RegCreateKeyEx (HKEY_LOCAL_MACHINE, PROFILE_LIST_PATH,
|
||
|
0, NULL, REG_OPTION_NON_VOLATILE,
|
||
|
KEY_WRITE, NULL, &hKey,
|
||
|
&dwDisp) == ERROR_SUCCESS) {
|
||
|
|
||
|
if (RegSetValueEx (hKey, PROFILES_DIRECTORY,
|
||
|
0, REG_EXPAND_SZ, (LPBYTE) szDest,
|
||
|
((lstrlen(szDest) + 1) * sizeof(TCHAR))) == ERROR_SUCCESS) {
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("DetermineProfilesLocation: Using unattend location %s for user profiles"), szDest));
|
||
|
}
|
||
|
|
||
|
RegCloseKey (hKey);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
} else {
|
||
|
|
||
|
//
|
||
|
// By default, the OS will try to use the new location for
|
||
|
// user profiles, but if we are doing an upgrade of a machine
|
||
|
// with profiles in the NT4 location, we want to continue
|
||
|
// to use that location.
|
||
|
//
|
||
|
// Build a test path to the old All Users directory on NT4
|
||
|
// to determine which location to use.
|
||
|
//
|
||
|
|
||
|
if (!ExpandEnvironmentStrings (NT4_PROFILES_DIRECTORY, szDirectory,
|
||
|
ARRAYSIZE(szDirectory))) {
|
||
|
|
||
|
DebugMsg((DM_WARNING, TEXT("DetermineProfilesLocation: ExpandEnvironmentStrings failed with error %d, setting the dir unexpanded"), GetLastError()));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
lpEnd = CheckSlash (szDirectory);
|
||
|
lstrcpy (lpEnd, ALL_USERS);
|
||
|
|
||
|
if (GetFileAttributesEx (szDirectory, GetFileExInfoStandard, &fad)) {
|
||
|
|
||
|
//
|
||
|
// An app was found that creates an "All Users" directory under NT4 profiles directory
|
||
|
// Check for Default User as well.
|
||
|
//
|
||
|
|
||
|
lstrcpy (lpEnd, DEFAULT_USER);
|
||
|
|
||
|
if (GetFileAttributesEx (szDirectory, GetFileExInfoStandard, &fad)) {
|
||
|
|
||
|
//
|
||
|
// The old profiles directory exists. We need to set this
|
||
|
// path in the registry
|
||
|
//
|
||
|
|
||
|
if (RegCreateKeyEx (HKEY_LOCAL_MACHINE, PROFILE_LIST_PATH,
|
||
|
0, NULL, REG_OPTION_NON_VOLATILE,
|
||
|
KEY_WRITE, NULL, &hKey,
|
||
|
&dwDisp) == ERROR_SUCCESS) {
|
||
|
|
||
|
if (RegSetValueEx (hKey, PROFILES_DIRECTORY,
|
||
|
0, REG_EXPAND_SZ, (LPBYTE) NT4_PROFILES_DIRECTORY,
|
||
|
((lstrlen(NT4_PROFILES_DIRECTORY) + 1) * sizeof(TCHAR))) == ERROR_SUCCESS) {
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("DetermineProfilesLocation: Using NT4 location for user profiles")));
|
||
|
}
|
||
|
|
||
|
RegCloseKey (hKey);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Check if the profiles directory exists.
|
||
|
//
|
||
|
|
||
|
dwSize = ARRAYSIZE(szDirectory);
|
||
|
if (!GetProfilesDirectoryEx(szDirectory, &dwSize, TRUE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("DetermineProfilesLocation: Failed to query profiles directory root.")));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
if (!CreateSecureAdminDirectory(szDirectory, OTHERSIDS_EVERYONE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("DetermineProfilesLocation: Failed to create profiles subdirectory <%s>. Error = %d."),
|
||
|
szDirectory, GetLastError()));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Decide where the Default User profile should be
|
||
|
//
|
||
|
|
||
|
if (!CheckProfile (szDirectory, DEFAULT_USER_PROFILE, DEFAULT_USER)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("DetermineProfilesLocation: Failed to check default user profile Error = %d."),
|
||
|
GetLastError()));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Check if the profiles\Default User directory exists
|
||
|
//
|
||
|
|
||
|
dwSize = ARRAYSIZE(szDirectory);
|
||
|
if (!GetDefaultUserProfileDirectoryEx(szDirectory, &dwSize, TRUE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("DetermineProfilesLocation: Failed to query default user profile directory.")));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
if (!CreateSecureAdminDirectory (szDirectory, OTHERSIDS_EVERYONE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("DetermineProfilesLocation: Failed to create Default User subdirectory <%s>. Error = %d."),
|
||
|
szDirectory, GetLastError()));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
SetFileAttributes (szDirectory, FILE_ATTRIBUTE_HIDDEN);
|
||
|
|
||
|
|
||
|
//
|
||
|
// Decide where the All Users profile should be
|
||
|
//
|
||
|
|
||
|
dwSize = ARRAYSIZE(szDirectory);
|
||
|
if (!GetProfilesDirectoryEx(szDirectory, &dwSize, TRUE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("DetermineProfilesLocation: Failed to query profiles directory root.")));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
if (!CheckProfile (szDirectory, ALL_USERS_PROFILE, ALL_USERS)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("DetermineProfilesLocation: Failed to check all users profile Error = %d."),
|
||
|
GetLastError()));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Check if the profiles\All Users directory exists
|
||
|
//
|
||
|
|
||
|
dwSize = ARRAYSIZE(szDirectory);
|
||
|
if (!GetAllUsersProfileDirectoryEx(szDirectory, &dwSize, TRUE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("DetermineProfilesLocation: Failed to query all users profile directory.")));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Give additional permissions for power users/everyone
|
||
|
//
|
||
|
|
||
|
if (!CreateSecureAdminDirectory (szDirectory, OTHERSIDS_POWERUSERS | OTHERSIDS_EVERYONE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("DetermineProfilesLocation: Failed to create All Users subdirectory <%s>. Error = %d."),
|
||
|
szDirectory, GetLastError()));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Hide some special profiles like NetworkService etc.
|
||
|
//
|
||
|
|
||
|
if (!bCleanInstall) {
|
||
|
HideSpecialProfiles();
|
||
|
}
|
||
|
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
//*************************************************************
|
||
|
//
|
||
|
// InitializeProfiles()
|
||
|
//
|
||
|
// Purpose: Confirms / Creates the profile, Default User,
|
||
|
// and All Users directories, and converts any
|
||
|
// existing common groups.
|
||
|
//
|
||
|
// Parameters:
|
||
|
//
|
||
|
// Return: TRUE if successful
|
||
|
// FALSE if an error occurs
|
||
|
//
|
||
|
// Comments: This should only be called by GUI mode setup!
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 8/08/95 ericflo Created
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
BOOL WINAPI InitializeProfiles (BOOL bGuiModeSetup)
|
||
|
{
|
||
|
TCHAR szDirectory[MAX_PATH];
|
||
|
TCHAR szSystemProfile[MAX_PATH];
|
||
|
TCHAR szTemp[MAX_PATH];
|
||
|
TCHAR szTemp2[MAX_PATH];
|
||
|
DWORD dwSize;
|
||
|
DWORD dwDisp;
|
||
|
LPTSTR lpEnd;
|
||
|
DWORD i;
|
||
|
HKEY hKey;
|
||
|
SECURITY_DESCRIPTOR sd;
|
||
|
SECURITY_ATTRIBUTES sa;
|
||
|
BOOL bRetVal = FALSE;
|
||
|
DWORD dwErr;
|
||
|
PSHELL32_API pShell32Api;
|
||
|
|
||
|
dwErr = GetLastError();
|
||
|
|
||
|
//
|
||
|
// Verbose output
|
||
|
//
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("InitializeProfiles: Entering.")));
|
||
|
|
||
|
|
||
|
|
||
|
//
|
||
|
// Create a named mutex that represents GUI mode setup running.
|
||
|
// This allows other processes that load userenv to detect that
|
||
|
// setup is running.
|
||
|
//
|
||
|
|
||
|
InitializeSecurityDescriptor( &sd, SECURITY_DESCRIPTOR_REVISION );
|
||
|
|
||
|
SetSecurityDescriptorDacl (
|
||
|
&sd,
|
||
|
TRUE, // Dacl present
|
||
|
NULL, // NULL Dacl
|
||
|
FALSE // Not defaulted
|
||
|
);
|
||
|
|
||
|
sa.lpSecurityDescriptor = &sd;
|
||
|
sa.bInheritHandle = FALSE;
|
||
|
sa.nLength = sizeof(sa);
|
||
|
|
||
|
if (!bGuiModeSetup) {
|
||
|
|
||
|
//
|
||
|
// block loading user profile
|
||
|
//
|
||
|
|
||
|
if (g_hProfileSetup) {
|
||
|
ResetEvent (g_hProfileSetup);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
dwErr = LoadShell32Api( &pShell32Api );
|
||
|
|
||
|
if ( dwErr != ERROR_SUCCESS ) {
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Set the USERPROFILE environment variable
|
||
|
//
|
||
|
|
||
|
ExpandEnvironmentStrings(SYSTEM_PROFILE_LOCATION, szSystemProfile, MAX_PATH);
|
||
|
|
||
|
//
|
||
|
// Requery for the default user profile directory (expanded version)
|
||
|
//
|
||
|
|
||
|
dwSize = ARRAYSIZE(szDirectory);
|
||
|
if (!GetDefaultUserProfileDirectoryEx(szDirectory, &dwSize, TRUE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("InitializeProfiles: Failed to query default user profile directory.")));
|
||
|
dwErr = GetLastError();
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Set the USERPROFILE environment variable
|
||
|
//
|
||
|
|
||
|
SetEnvironmentVariable (TEXT("USERPROFILE"), szDirectory);
|
||
|
|
||
|
//
|
||
|
// Create all the folders under Default User
|
||
|
//
|
||
|
|
||
|
lpEnd = CheckSlash (szDirectory);
|
||
|
|
||
|
|
||
|
//
|
||
|
// Loop through the shell folders
|
||
|
//
|
||
|
|
||
|
for (i=0; i < g_dwNumShellFolders; i++) {
|
||
|
|
||
|
lstrcpy (lpEnd, c_ShellFolders[i].szFolderLocation);
|
||
|
|
||
|
if (!CreateNestedDirectory(szDirectory, NULL)) {
|
||
|
dwErr = GetLastError();
|
||
|
DebugMsg((DM_WARNING, TEXT("InitializeProfiles: Failed to create the destination directory <%s>. Error = %d"),
|
||
|
szDirectory, dwErr));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
if (c_ShellFolders[i].iFolderResourceID != 0) {
|
||
|
// NOTE this returns an HRESULT
|
||
|
dwErr = pShell32Api->pfnShSetLocalizedName( szDirectory,
|
||
|
c_ShellFolders[i].lpFolderResourceDLL,
|
||
|
c_ShellFolders[i].iFolderResourceID );
|
||
|
if (FAILED(dwErr)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("InitializeProfiles: SHSetLocalizedName failed for directory <%s>. Error = %d"),
|
||
|
szDirectory, dwErr));
|
||
|
goto Exit;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (c_ShellFolders[i].bHidden) {
|
||
|
SetFileAttributes(szDirectory, GetFileAttributes(szDirectory) | FILE_ATTRIBUTE_HIDDEN);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Remove the %USERPROFILE%\Personal directory if it exists.
|
||
|
// Windows NT 4.0 had a Personal folder in the root of the
|
||
|
// user's profile. NT 5.0 renames this folder to My Documents
|
||
|
//
|
||
|
|
||
|
if (LoadString (g_hDllInstance, IDS_SH_PERSONAL2, szTemp, ARRAYSIZE(szTemp))) {
|
||
|
lstrcpy (lpEnd, szTemp);
|
||
|
RemoveDirectory(szDirectory);
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Migrate the Template Directory if it exists. Copy it from %systemroot%\shellnew
|
||
|
// to Templates directory under default user. Do the same for existing profiles.
|
||
|
//
|
||
|
|
||
|
if ((LoadString (g_hDllInstance, IDS_SH_TEMPLATES2, szTemp, ARRAYSIZE(szTemp))) &&
|
||
|
(ExpandEnvironmentStrings (szTemp, szTemp2, ARRAYSIZE(szTemp2))) &&
|
||
|
(LoadString (g_hDllInstance, IDS_SH_TEMPLATES, szTemp, ARRAYSIZE(szTemp)))) {
|
||
|
|
||
|
//
|
||
|
// if all of the above succeeded
|
||
|
//
|
||
|
|
||
|
lstrcpy (lpEnd, szTemp);
|
||
|
DebugMsg((DM_VERBOSE, TEXT("InitializeProfiles: Copying <%s> to %s. Error = %d"), szTemp2, szDirectory));
|
||
|
CopyProfileDirectory(szTemp2, szDirectory, CPD_IGNORECOPYERRORS | CPD_IGNOREHIVE);
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Remove %USERPROFILE%\Temp if it exists. The Temp directory
|
||
|
// will now be in the Local Settings folder
|
||
|
//
|
||
|
|
||
|
lstrcpy (lpEnd, TEXT("Temp"));
|
||
|
Delnode(szDirectory);
|
||
|
|
||
|
|
||
|
//
|
||
|
// Remove %USERPROFILE%\Temporary Internet Files if it exists. The
|
||
|
// Temporary Internet Files directory will now be in the Local Settings
|
||
|
// folder
|
||
|
//
|
||
|
|
||
|
if (LoadString (g_hDllInstance, IDS_TEMPINTERNETFILES, szTemp, ARRAYSIZE(szTemp))) {
|
||
|
lstrcpy (lpEnd, szTemp);
|
||
|
Delnode(szDirectory);
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Remove %USERPROFILE%\History if it exists. The History
|
||
|
// directory will now be in the Local Settings folder
|
||
|
//
|
||
|
|
||
|
if (LoadString (g_hDllInstance, IDS_HISTORY, szTemp, ARRAYSIZE(szTemp))) {
|
||
|
lstrcpy (lpEnd, szTemp);
|
||
|
Delnode(szDirectory);
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Set the User Shell Folder paths in the registry
|
||
|
//
|
||
|
|
||
|
lstrcpy (szTemp, TEXT(".Default"));
|
||
|
lpEnd = CheckSlash (szTemp);
|
||
|
lstrcpy(lpEnd, USER_SHELL_FOLDERS);
|
||
|
|
||
|
lstrcpy (szDirectory, TEXT("%USERPROFILE%"));
|
||
|
lpEnd = CheckSlash (szDirectory);
|
||
|
|
||
|
if (RegCreateKeyEx (HKEY_USERS, szTemp,
|
||
|
0, NULL, REG_OPTION_NON_VOLATILE,
|
||
|
KEY_READ | KEY_WRITE, NULL, &hKey,
|
||
|
&dwDisp) == ERROR_SUCCESS) {
|
||
|
|
||
|
for (i=0; i < g_dwNumShellFolders; i++) {
|
||
|
|
||
|
if (c_ShellFolders[i].bAddCSIDL) {
|
||
|
lstrcpy (lpEnd, c_ShellFolders[i].szFolderLocation);
|
||
|
|
||
|
RegSetValueEx (hKey, c_ShellFolders[i].lpFolderName,
|
||
|
0, REG_EXPAND_SZ, (LPBYTE) szDirectory,
|
||
|
((lstrlen(szDirectory) + 1) * sizeof(TCHAR)));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
RegCloseKey (hKey);
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Set the Shell Folder paths in the registry
|
||
|
//
|
||
|
|
||
|
lstrcpy (szTemp, TEXT(".Default"));
|
||
|
lpEnd = CheckSlash (szTemp);
|
||
|
lstrcpy(lpEnd, SHELL_FOLDERS);
|
||
|
|
||
|
dwSize = ARRAYSIZE(szDirectory);
|
||
|
if (!GetDefaultUserProfileDirectoryEx(szDirectory, &dwSize, TRUE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("InitializeProfiles: Failed to query default user profile directory.")));
|
||
|
dwErr = GetLastError();
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
lpEnd = CheckSlash (szDirectory);
|
||
|
|
||
|
if (RegCreateKeyEx (HKEY_USERS, szTemp,
|
||
|
0, NULL, REG_OPTION_NON_VOLATILE,
|
||
|
KEY_READ | KEY_WRITE, NULL, &hKey,
|
||
|
&dwDisp) == ERROR_SUCCESS) {
|
||
|
|
||
|
for (i=0; i < g_dwNumShellFolders; i++) {
|
||
|
|
||
|
if (c_ShellFolders[i].bAddCSIDL) {
|
||
|
lstrcpy (lpEnd, c_ShellFolders[i].szFolderLocation);
|
||
|
|
||
|
RegSetValueEx (hKey, c_ShellFolders[i].lpFolderName,
|
||
|
0, REG_SZ, (LPBYTE) szDirectory,
|
||
|
((lstrlen(szDirectory) + 1) * sizeof(TCHAR)));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
RegCloseKey (hKey);
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Set the per user TEMP and TMP environment variables
|
||
|
//
|
||
|
|
||
|
if (LoadString (g_hDllInstance, IDS_SH_TEMP,
|
||
|
szTemp, ARRAYSIZE(szTemp))) {
|
||
|
|
||
|
lstrcpy (szDirectory, TEXT("%USERPROFILE%"));
|
||
|
lpEnd = CheckSlash (szDirectory);
|
||
|
|
||
|
LoadString (g_hDllInstance, IDS_SH_LOCALSETTINGS,
|
||
|
lpEnd, ARRAYSIZE(szTemp)-(lstrlen(szDirectory)));
|
||
|
|
||
|
lpEnd = CheckSlash (szDirectory);
|
||
|
lstrcpy (lpEnd, szTemp);
|
||
|
|
||
|
|
||
|
if (RegCreateKeyEx (HKEY_USERS, TEXT(".Default\\Environment"),
|
||
|
0, NULL, REG_OPTION_NON_VOLATILE,
|
||
|
KEY_WRITE, NULL, &hKey,
|
||
|
&dwDisp) == ERROR_SUCCESS) {
|
||
|
|
||
|
RegSetValueEx (hKey, TEXT("TEMP"),
|
||
|
0, REG_EXPAND_SZ, (LPBYTE) szDirectory,
|
||
|
((lstrlen(szDirectory) + 1) * sizeof(TCHAR)));
|
||
|
|
||
|
RegSetValueEx (hKey, TEXT("TMP"),
|
||
|
0, REG_EXPAND_SZ, (LPBYTE) szDirectory,
|
||
|
((lstrlen(szDirectory) + 1) * sizeof(TCHAR)));
|
||
|
|
||
|
RegCloseKey (hKey);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Set the user preference exclusion list. This will
|
||
|
// prevent the Local Settings folder from roaming
|
||
|
//
|
||
|
|
||
|
if (LoadString (g_hDllInstance, IDS_EXCLUSIONLIST,
|
||
|
szDirectory, ARRAYSIZE(szDirectory))) {
|
||
|
|
||
|
lstrcpy (szTemp, TEXT(".Default"));
|
||
|
lpEnd = CheckSlash (szTemp);
|
||
|
lstrcpy(lpEnd, WINLOGON_KEY);
|
||
|
|
||
|
if (RegCreateKeyEx (HKEY_USERS, szTemp,
|
||
|
0, NULL, REG_OPTION_NON_VOLATILE,
|
||
|
KEY_READ | KEY_WRITE, NULL, &hKey,
|
||
|
&dwDisp) == ERROR_SUCCESS) {
|
||
|
|
||
|
RegSetValueEx (hKey, TEXT("ExcludeProfileDirs"),
|
||
|
0, REG_SZ, (LPBYTE) szDirectory,
|
||
|
((lstrlen(szDirectory) + 1) * sizeof(TCHAR)));
|
||
|
|
||
|
RegCloseKey (hKey);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Requery for the all users profile directory (expanded version)
|
||
|
//
|
||
|
|
||
|
dwSize = ARRAYSIZE(szDirectory);
|
||
|
if (!GetAllUsersProfileDirectoryEx(szDirectory, &dwSize, TRUE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("InitializeProfiles: Failed to query all users profile directory.")));
|
||
|
dwErr = GetLastError();
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Set the ALLUSERSPROFILE environment variable
|
||
|
//
|
||
|
|
||
|
SetEnvironmentVariable (TEXT("ALLUSERSPROFILE"), szDirectory);
|
||
|
|
||
|
|
||
|
//
|
||
|
// Create all the folders under All Users
|
||
|
//
|
||
|
|
||
|
|
||
|
lpEnd = CheckSlash (szDirectory);
|
||
|
|
||
|
|
||
|
//
|
||
|
// Loop through the shell folders
|
||
|
//
|
||
|
|
||
|
for (i=0; i < g_dwNumCommonShellFolders; i++) {
|
||
|
|
||
|
lstrcpy (lpEnd, c_CommonShellFolders[i].szFolderLocation);
|
||
|
|
||
|
if (!CreateNestedDirectory(szDirectory, NULL)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("InitializeProfiles: Failed to create the destination directory <%s>. Error = %d"),
|
||
|
szDirectory, GetLastError()));
|
||
|
dwErr = GetLastError();
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
if (c_CommonShellFolders[i].iFolderResourceID != 0) {
|
||
|
// NOTE this returns an HRESULT
|
||
|
dwErr = pShell32Api->pfnShSetLocalizedName( szDirectory,
|
||
|
c_CommonShellFolders[i].lpFolderResourceDLL,
|
||
|
c_CommonShellFolders[i].iFolderResourceID );
|
||
|
if (FAILED(dwErr)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("InitializeProfiles: SHSetLocalizedName failed for directory <%s>. Error = %d"),
|
||
|
szDirectory, dwErr));
|
||
|
goto Exit;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (c_CommonShellFolders[i].bHidden) {
|
||
|
SetFileAttributes(szDirectory, GetFileAttributes(szDirectory) | FILE_ATTRIBUTE_HIDDEN);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Unsecure the Documents and App Data folders in the All Users profile
|
||
|
//
|
||
|
|
||
|
if (LoadString (g_hDllInstance, IDS_SH_SHAREDDOCS, szTemp, ARRAYSIZE(szTemp))) {
|
||
|
lstrcpy (lpEnd, szTemp);
|
||
|
SetFilePermissions(szDirectory, c_szCommonDocumentsACL);
|
||
|
}
|
||
|
|
||
|
if (LoadString (g_hDllInstance, IDS_SH_APPDATA, szTemp, ARRAYSIZE(szTemp))) {
|
||
|
lstrcpy (lpEnd, szTemp);
|
||
|
SetFilePermissions(szDirectory, c_szCommonAppDataACL);
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Set the User Shell Folder paths in the registry
|
||
|
//
|
||
|
|
||
|
lstrcpy (szDirectory, TEXT("%ALLUSERSPROFILE%"));
|
||
|
lpEnd = CheckSlash (szDirectory);
|
||
|
|
||
|
if (RegCreateKeyEx (HKEY_LOCAL_MACHINE, USER_SHELL_FOLDERS,
|
||
|
0, NULL, REG_OPTION_NON_VOLATILE,
|
||
|
KEY_READ | KEY_WRITE, NULL, &hKey,
|
||
|
&dwDisp) == ERROR_SUCCESS) {
|
||
|
|
||
|
for (i=0; i < g_dwNumCommonShellFolders; i++) {
|
||
|
|
||
|
if (c_ShellFolders[i].bAddCSIDL) {
|
||
|
lstrcpy (lpEnd, c_CommonShellFolders[i].szFolderLocation);
|
||
|
|
||
|
RegSetValueEx (hKey, c_CommonShellFolders[i].lpFolderName,
|
||
|
0, REG_EXPAND_SZ, (LPBYTE) szDirectory,
|
||
|
((lstrlen(szDirectory) + 1) * sizeof(TCHAR)));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
RegCloseKey (hKey);
|
||
|
}
|
||
|
|
||
|
#if defined(_WIN64)
|
||
|
//
|
||
|
// On 64-bit NT, we need to create the user shell folders in
|
||
|
// the 32-bit view of the registry so that 32-bit app calls
|
||
|
// to SHGetFolderPath(...CSIDL_COMMON_APPDATA...) would succeed.
|
||
|
//
|
||
|
if (RegCreateKeyEx (HKEY_LOCAL_MACHINE, USER_SHELL_FOLDERS32,
|
||
|
0, NULL, REG_OPTION_NON_VOLATILE,
|
||
|
KEY_READ | KEY_WRITE, NULL, &hKey,
|
||
|
&dwDisp) == ERROR_SUCCESS) {
|
||
|
|
||
|
for (i=0; i < g_dwNumCommonShellFolders; i++) {
|
||
|
|
||
|
if (c_ShellFolders[i].bAddCSIDL) {
|
||
|
lstrcpy (lpEnd, c_CommonShellFolders[i].szFolderLocation);
|
||
|
|
||
|
RegSetValueEx (hKey, c_CommonShellFolders[i].lpFolderName,
|
||
|
0, REG_EXPAND_SZ, (LPBYTE) szDirectory,
|
||
|
((lstrlen(szDirectory) + 1) * sizeof(TCHAR)));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
RegCloseKey (hKey);
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
//
|
||
|
// Convert any Program Manager common groups
|
||
|
//
|
||
|
|
||
|
if (!ConvertCommonGroups()) {
|
||
|
DebugMsg((DM_WARNING, TEXT("InitializeProfiles: ConvertCommonGroups failed.")));
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Success
|
||
|
//
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("InitializeProfiles: Leaving successfully.")));
|
||
|
|
||
|
bRetVal = TRUE;
|
||
|
|
||
|
Exit:
|
||
|
|
||
|
if ((!bGuiModeSetup) && (g_hProfileSetup)) {
|
||
|
SetEvent (g_hProfileSetup);
|
||
|
}
|
||
|
|
||
|
SetLastError(dwErr);
|
||
|
return bRetVal;
|
||
|
}
|
||
|
|
||
|
//*************************************************************
|
||
|
//
|
||
|
// CheckProfile()
|
||
|
//
|
||
|
// Purpose: Checks and creates a storage location for either
|
||
|
// the Default User or All Users profile
|
||
|
//
|
||
|
// Parameters: LPTSTR lpProfilesDir - Root of the profiles
|
||
|
// LPTSTR lpProfileValue - Profile registry value name
|
||
|
// LPTSTR lpProfileName - Default profile name
|
||
|
//
|
||
|
//
|
||
|
// Return: TRUE if successful
|
||
|
// FALSE if an error occurs
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
BOOL CheckProfile (LPTSTR lpProfilesDir, LPTSTR lpProfileValue,
|
||
|
LPTSTR lpProfileName)
|
||
|
{
|
||
|
TCHAR szTemp[MAX_PATH];
|
||
|
TCHAR szTemp2[MAX_PATH];
|
||
|
TCHAR szName[MAX_PATH];
|
||
|
TCHAR szFormat[30];
|
||
|
DWORD dwSize, dwDisp, dwType;
|
||
|
LPTSTR lpEnd;
|
||
|
LONG lResult;
|
||
|
HKEY hKey;
|
||
|
INT iStrLen;
|
||
|
WIN32_FILE_ATTRIBUTE_DATA fad;
|
||
|
|
||
|
|
||
|
//
|
||
|
// Open the ProfileList key
|
||
|
//
|
||
|
|
||
|
lResult = RegCreateKeyEx (HKEY_LOCAL_MACHINE, PROFILE_LIST_PATH,
|
||
|
0, NULL, REG_OPTION_NON_VOLATILE,
|
||
|
KEY_READ | KEY_WRITE, NULL, &hKey, &dwDisp);
|
||
|
|
||
|
if (lResult != ERROR_SUCCESS) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CheckProfile: Failed to open profile list key with %d."),
|
||
|
lResult));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Check the registry to see if this folder is defined already
|
||
|
//
|
||
|
|
||
|
dwSize = sizeof(szTemp);
|
||
|
if (RegQueryValueEx (hKey, lpProfileValue, NULL, &dwType,
|
||
|
(LPBYTE) szTemp, &dwSize) == ERROR_SUCCESS) {
|
||
|
RegCloseKey (hKey);
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Generate the default name
|
||
|
//
|
||
|
|
||
|
lstrcpy (szTemp, lpProfilesDir);
|
||
|
lpEnd = CheckSlash (szTemp);
|
||
|
lstrcpy (lpEnd, lpProfileName);
|
||
|
lstrcpy (szName, lpProfileName);
|
||
|
|
||
|
|
||
|
//
|
||
|
// Check if this directory exists
|
||
|
//
|
||
|
|
||
|
ExpandEnvironmentStrings (szTemp, szTemp2, ARRAYSIZE(szTemp2));
|
||
|
|
||
|
if (GetFileAttributesEx (szTemp2, GetFileExInfoStandard, &fad)) {
|
||
|
|
||
|
if (fad.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
|
||
|
|
||
|
|
||
|
//
|
||
|
// Check if this directory is under the system root.
|
||
|
// If so, this is ok, we don't need to generate a unique
|
||
|
// name for this.
|
||
|
//
|
||
|
|
||
|
ExpandEnvironmentStrings (TEXT("%SystemRoot%"), szTemp,
|
||
|
ARRAYSIZE(szTemp));
|
||
|
|
||
|
iStrLen = lstrlen (szTemp);
|
||
|
|
||
|
if (CompareString (LOCALE_USER_DEFAULT, NORM_IGNORECASE,
|
||
|
szTemp, iStrLen, szTemp2, iStrLen) != CSTR_EQUAL) {
|
||
|
|
||
|
|
||
|
//
|
||
|
// The directory exists already. Use a new name of
|
||
|
// Profile Name (SystemDirectory)
|
||
|
//
|
||
|
// eg: Default User (WINNT)
|
||
|
//
|
||
|
|
||
|
lpEnd = szTemp + lstrlen(szTemp) - 1;
|
||
|
|
||
|
while ((lpEnd > szTemp) && ((*lpEnd) != TEXT('\\')))
|
||
|
lpEnd--;
|
||
|
|
||
|
if (*lpEnd == TEXT('\\')) {
|
||
|
lpEnd++;
|
||
|
}
|
||
|
|
||
|
LoadString (g_hDllInstance, IDS_PROFILE_FORMAT, szFormat,
|
||
|
ARRAYSIZE(szFormat));
|
||
|
wsprintf (szName, szFormat, lpProfileName, lpEnd);
|
||
|
|
||
|
//
|
||
|
// To prevent reusing the directories, delete it first..
|
||
|
//
|
||
|
|
||
|
lstrcpy (szTemp, lpProfilesDir);
|
||
|
lpEnd = CheckSlash (szTemp);
|
||
|
lstrcpy (lpEnd, szName);
|
||
|
|
||
|
if (ExpandEnvironmentStrings (szTemp, szTemp2, ARRAYSIZE(szTemp2))) {
|
||
|
DebugMsg((DM_VERBOSE, TEXT("InitializeProfiles: Delnoding directory.. %s"), szTemp2));
|
||
|
Delnode(szTemp2);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Save the profile name in the registry
|
||
|
//
|
||
|
|
||
|
RegSetValueEx (hKey, lpProfileValue,
|
||
|
0, REG_SZ, (LPBYTE) szName,
|
||
|
((lstrlen(szName) + 1) * sizeof(TCHAR)));
|
||
|
|
||
|
RegCloseKey (hKey);
|
||
|
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("InitializeProfiles: The %s profile is mapped to %s"),
|
||
|
lpProfileName, szName));
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
|
||
|
//*************************************************************
|
||
|
//
|
||
|
// CreateUserProfile()
|
||
|
//
|
||
|
// Purpose: Creates a new user profile, but does not load
|
||
|
// the hive.
|
||
|
//
|
||
|
// Parameters: pSid - SID pointer
|
||
|
// lpUserName - User name
|
||
|
// lpUserHive - Optional user hive
|
||
|
// lpProfileDir - Receives the new profile directory
|
||
|
// dwDirSize - Size of lpProfileDir
|
||
|
//
|
||
|
// Return: TRUE if successful
|
||
|
// FALSE if an error occurs
|
||
|
//
|
||
|
// Comments: If a user hive isn't specified the default user
|
||
|
// hive will be used.
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 9/12/95 ericflo Created
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
BOOL WINAPI CreateUserProfile (PSID pSid, LPCTSTR lpUserName, LPCTSTR lpUserHive,
|
||
|
LPTSTR lpProfileDir, DWORD dwDirSize)
|
||
|
{
|
||
|
return CreateUserProfileEx(pSid, lpUserName, lpUserHive, lpProfileDir, dwDirSize, TRUE);
|
||
|
}
|
||
|
|
||
|
//*************************************************************
|
||
|
//
|
||
|
// CreateUserProfileEx()
|
||
|
//
|
||
|
// Purpose: Creates a new user profile, but does not load
|
||
|
// the hive.
|
||
|
//
|
||
|
// Parameters: pSid - SID pointer
|
||
|
// lpUserName - User name
|
||
|
// lpUserHive - Optional user hive
|
||
|
// lpProfileDir - Receives the new profile directory
|
||
|
// dwDirSize - Size of lpProfileDir
|
||
|
// bWin9xUpg - Flag to say whether it is win9x upgrade
|
||
|
//
|
||
|
// Return: TRUE if successful
|
||
|
// FALSE if an error occurs
|
||
|
//
|
||
|
// Comments: If a user hive isn't specified the default user
|
||
|
// hive will be used.
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 9/12/95 ericflo Created
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
BOOL WINAPI CreateUserProfileEx (PSID pSid, LPCTSTR lpUserName, LPCTSTR lpUserHive,
|
||
|
LPTSTR lpProfileDir, DWORD dwDirSize, BOOL bWin9xUpg)
|
||
|
{
|
||
|
TCHAR szProfileDir[MAX_PATH];
|
||
|
TCHAR szExpProfileDir[MAX_PATH] = {0};
|
||
|
TCHAR szDirectory[MAX_PATH];
|
||
|
TCHAR LocalProfileKey[MAX_PATH];
|
||
|
UNICODE_STRING UnicodeString;
|
||
|
LPTSTR lpSidString, lpEnd, lpSave;
|
||
|
NTSTATUS NtStatus;
|
||
|
LONG lResult;
|
||
|
DWORD dwDisp;
|
||
|
DWORD dwError;
|
||
|
DWORD dwSize;
|
||
|
DWORD dwType;
|
||
|
HKEY hKey;
|
||
|
|
||
|
|
||
|
|
||
|
//
|
||
|
// Check parameters
|
||
|
//
|
||
|
|
||
|
if (!lpUserName || !lpUserName[0]) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: Null username.")));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
if (!pSid) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: Null SID.")));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Verbose output
|
||
|
//
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateUserProfile: Entering with <%s>."), lpUserName));
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateUserProfile: Entering with user hive of <%s>."),
|
||
|
lpUserHive ? lpUserHive : TEXT("NULL")));
|
||
|
|
||
|
|
||
|
//
|
||
|
// Convert the sid into text format
|
||
|
//
|
||
|
|
||
|
NtStatus = RtlConvertSidToUnicodeString(&UnicodeString, pSid, (BOOLEAN)TRUE);
|
||
|
|
||
|
if (!NT_SUCCESS(NtStatus)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: RtlConvertSidToUnicodeString failed, status = 0x%x"),
|
||
|
NtStatus));
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
lpSidString = UnicodeString.Buffer;
|
||
|
|
||
|
|
||
|
//
|
||
|
// Check if this user's profile exists already
|
||
|
//
|
||
|
|
||
|
lstrcpy(LocalProfileKey, PROFILE_LIST_PATH);
|
||
|
lstrcat(LocalProfileKey, TEXT("\\"));
|
||
|
lstrcat(LocalProfileKey, lpSidString);
|
||
|
|
||
|
szProfileDir[0] = TEXT('\0');
|
||
|
|
||
|
if (RegOpenKeyEx (HKEY_LOCAL_MACHINE, LocalProfileKey,
|
||
|
0, KEY_READ, &hKey) == ERROR_SUCCESS) {
|
||
|
|
||
|
dwSize = sizeof(szProfileDir);
|
||
|
RegQueryValueEx (hKey, PROFILE_IMAGE_VALUE_NAME, NULL,
|
||
|
&dwType, (LPBYTE) szProfileDir, &dwSize);
|
||
|
|
||
|
RegCloseKey (hKey);
|
||
|
}
|
||
|
|
||
|
|
||
|
if (szProfileDir[0] == TEXT('\0')) {
|
||
|
|
||
|
//
|
||
|
// Make the user's directory
|
||
|
//
|
||
|
|
||
|
dwSize = ARRAYSIZE(szProfileDir);
|
||
|
if (!GetProfilesDirectoryEx(szProfileDir, &dwSize, FALSE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: Failed to get profile root directory.")));
|
||
|
RtlFreeUnicodeString(&UnicodeString);
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
if (!ComputeLocalProfileName (NULL, lpUserName, szProfileDir, ARRAYSIZE(szProfileDir),
|
||
|
szExpProfileDir, ARRAYSIZE(szExpProfileDir), pSid, bWin9xUpg)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: Failed to create directory.")));
|
||
|
RtlFreeUnicodeString(&UnicodeString);
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Copy the default user profile into this directory
|
||
|
//
|
||
|
|
||
|
dwSize = ARRAYSIZE(szDirectory);
|
||
|
if (!GetDefaultUserProfileDirectoryEx(szDirectory, &dwSize, TRUE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: Failed to get default user profile.")));
|
||
|
RtlFreeUnicodeString(&UnicodeString);
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
|
||
|
if (lpUserHive) {
|
||
|
|
||
|
//
|
||
|
// Copy the default user profile without the hive.
|
||
|
//
|
||
|
|
||
|
if (!CopyProfileDirectory (szDirectory, szExpProfileDir, CPD_IGNORECOPYERRORS | CPD_IGNOREHIVE | CPD_IGNORELONGFILENAMES)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: CopyProfileDirectory failed with error %d."), GetLastError()));
|
||
|
RtlFreeUnicodeString(&UnicodeString);
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Now copy the hive
|
||
|
//
|
||
|
|
||
|
lstrcpy (szDirectory, szExpProfileDir);
|
||
|
lpEnd = CheckSlash (szDirectory);
|
||
|
lstrcpy (lpEnd, c_szNTUserDat);
|
||
|
|
||
|
if (lstrcmpi (lpUserHive, szDirectory)) {
|
||
|
|
||
|
if (!CopyFile (lpUserHive, szDirectory, FALSE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: Failed to copy user hive with error %d."), GetLastError()));
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: Source: %s."), lpUserHive));
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: Destination: %s."), szDirectory));
|
||
|
RtlFreeUnicodeString(&UnicodeString);
|
||
|
return FALSE;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
} else {
|
||
|
|
||
|
//
|
||
|
// Copy the default user profile and the hive.
|
||
|
//
|
||
|
|
||
|
if (!CopyProfileDirectory (szDirectory, szExpProfileDir, CPD_IGNORECOPYERRORS | CPD_IGNORELONGFILENAMES)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: CopyProfileDirectory failed with error %d."), GetLastError()));
|
||
|
RtlFreeUnicodeString(&UnicodeString);
|
||
|
return FALSE;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Save the user's profile in the registry.
|
||
|
//
|
||
|
|
||
|
lResult = RegCreateKeyEx(HKEY_LOCAL_MACHINE, LocalProfileKey, 0, 0, 0,
|
||
|
KEY_READ | KEY_WRITE, NULL, &hKey, &dwDisp);
|
||
|
|
||
|
if (lResult != ERROR_SUCCESS) {
|
||
|
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: Failed trying to create the local profile key <%s>, error = %d."), LocalProfileKey, lResult));
|
||
|
RtlFreeUnicodeString(&UnicodeString);
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
//
|
||
|
// Add the profile directory
|
||
|
//
|
||
|
|
||
|
lResult = RegSetValueEx(hKey, PROFILE_IMAGE_VALUE_NAME, 0,
|
||
|
REG_EXPAND_SZ,
|
||
|
(LPBYTE)szProfileDir,
|
||
|
sizeof(TCHAR)*(lstrlen(szProfileDir) + 1));
|
||
|
|
||
|
|
||
|
if (lResult != ERROR_SUCCESS) {
|
||
|
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: First RegSetValueEx failed, error = %d."), lResult));
|
||
|
RegCloseKey (hKey);
|
||
|
RtlFreeUnicodeString(&UnicodeString);
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Add the users's SID
|
||
|
//
|
||
|
|
||
|
lResult = RegSetValueEx(hKey, TEXT("Sid"), 0,
|
||
|
REG_BINARY, pSid, RtlLengthSid(pSid));
|
||
|
|
||
|
|
||
|
if (lResult != ERROR_SUCCESS) {
|
||
|
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: Second RegSetValueEx failed, error = %d."), lResult));
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Close the registry key
|
||
|
//
|
||
|
|
||
|
RegCloseKey (hKey);
|
||
|
|
||
|
} else {
|
||
|
|
||
|
//
|
||
|
// The user already has a profile, so just copy the hive if
|
||
|
// appropriate.
|
||
|
//
|
||
|
|
||
|
ExpandEnvironmentStrings (szProfileDir, szExpProfileDir,
|
||
|
ARRAYSIZE(szExpProfileDir));
|
||
|
|
||
|
if (lpUserHive) {
|
||
|
|
||
|
//
|
||
|
// Copy the hive
|
||
|
//
|
||
|
|
||
|
lstrcpy (szDirectory, szExpProfileDir);
|
||
|
lpEnd = CheckSlash (szDirectory);
|
||
|
lstrcpy (lpEnd, c_szNTUserDat);
|
||
|
|
||
|
SetFileAttributes (szDirectory, FILE_ATTRIBUTE_NORMAL);
|
||
|
|
||
|
if (lstrcmpi (lpUserHive, szDirectory)) {
|
||
|
if (!CopyFile (lpUserHive, szDirectory, FALSE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: Failed to copy user hive with error %d."), GetLastError()));
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: Source: %s."), lpUserHive));
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: Destination: %s."), szDirectory));
|
||
|
RtlFreeUnicodeString(&UnicodeString);
|
||
|
return FALSE;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Now load the hive temporary so the security can be fixed
|
||
|
//
|
||
|
|
||
|
lpEnd = CheckSlash (szExpProfileDir);
|
||
|
lpSave = lpEnd - 1;
|
||
|
lstrcpy (lpEnd, c_szNTUserDat);
|
||
|
|
||
|
lResult = MyRegLoadKey(HKEY_USERS, lpSidString, szExpProfileDir);
|
||
|
|
||
|
*lpSave = TEXT('\0');
|
||
|
|
||
|
if (lResult != ERROR_SUCCESS) {
|
||
|
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: Failed to load hive, error = %d."), lResult));
|
||
|
dwError = GetLastError();
|
||
|
DeleteProfileEx (lpSidString, szExpProfileDir, FALSE, HKEY_LOCAL_MACHINE, NULL);
|
||
|
RtlFreeUnicodeString(&UnicodeString);
|
||
|
SetLastError(dwError);
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
if (!SetupNewHive(NULL, lpSidString, pSid)) {
|
||
|
|
||
|
DebugMsg((DM_WARNING, TEXT("CreateUserProfile: SetupNewHive failed.")));
|
||
|
dwError = GetLastError();
|
||
|
MyRegUnLoadKey(HKEY_USERS, lpSidString);
|
||
|
DeleteProfileEx (lpSidString, szExpProfileDir, FALSE, HKEY_LOCAL_MACHINE, NULL);
|
||
|
RtlFreeUnicodeString(&UnicodeString);
|
||
|
SetLastError(dwError);
|
||
|
return FALSE;
|
||
|
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Unload the hive
|
||
|
//
|
||
|
|
||
|
MyRegUnLoadKey(HKEY_USERS, lpSidString);
|
||
|
|
||
|
|
||
|
//
|
||
|
// Free the sid string
|
||
|
//
|
||
|
|
||
|
RtlFreeUnicodeString(&UnicodeString);
|
||
|
|
||
|
|
||
|
//
|
||
|
// Save the profile path if appropriate
|
||
|
//
|
||
|
|
||
|
if (lpProfileDir) {
|
||
|
|
||
|
if ((DWORD)lstrlen(szExpProfileDir) < dwDirSize) {
|
||
|
lstrcpy (lpProfileDir, szExpProfileDir);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Success
|
||
|
//
|
||
|
|
||
|
DebugMsg((DM_VERBOSE, TEXT("CreateUserProfile: Leaving successfully.")));
|
||
|
|
||
|
return TRUE;
|
||
|
|
||
|
}
|
||
|
|
||
|
|
||
|
//*************************************************************************
|
||
|
//
|
||
|
// SecureUserProfiles()
|
||
|
//
|
||
|
// Routine Description :
|
||
|
// This function secures user profiles during FAT->NTFS conversion.
|
||
|
// The function loops through all profiles registered under current
|
||
|
// OS and sets the security for the corresponding profile directory
|
||
|
// and nested subdirs. Assumption is the function will be called
|
||
|
// only during FAT->NTFS conversion.
|
||
|
//
|
||
|
// Arguments :
|
||
|
// None.
|
||
|
//
|
||
|
// Return Value :
|
||
|
// None.
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 8/8/00 santanuc Created
|
||
|
//
|
||
|
//*************************************************************************
|
||
|
|
||
|
void WINAPI SecureUserProfiles(void)
|
||
|
{
|
||
|
SECURITY_DESCRIPTOR DirSd, FileSd;
|
||
|
SID_IDENTIFIER_AUTHORITY authNT = SECURITY_NT_AUTHORITY;
|
||
|
PACL pDirAcl = NULL, pFileAcl = NULL;
|
||
|
PSID pSidOwner=NULL, pSidAdmin = NULL, pSidSystem = NULL;
|
||
|
DWORD cbAcl, aceIndex;
|
||
|
HKEY hKeyProfilesList, hKeyProfile = NULL;
|
||
|
TCHAR szSIDName[MAX_PATH], szProfilePath[MAX_PATH], szExpandedProfilePath[MAX_PATH];
|
||
|
DWORD dwSIDNameSize, dwSize;
|
||
|
DWORD dwIndex;
|
||
|
LONG lResult;
|
||
|
FILETIME ft;
|
||
|
|
||
|
|
||
|
//
|
||
|
// Get the system sid
|
||
|
//
|
||
|
|
||
|
if (!AllocateAndInitializeSid(&authNT, 1, SECURITY_LOCAL_SYSTEM_RID,
|
||
|
0, 0, 0, 0, 0, 0, 0, &pSidSystem)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to initialize system sid. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Get the Admin sid
|
||
|
//
|
||
|
|
||
|
if (!AllocateAndInitializeSid(&authNT, 2, SECURITY_BUILTIN_DOMAIN_RID,
|
||
|
DOMAIN_ALIAS_RID_ADMINS, 0, 0,
|
||
|
0, 0, 0, 0, &pSidAdmin)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to initialize admin sid. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Open the ProfileList key
|
||
|
//
|
||
|
|
||
|
if (RegOpenKeyEx (HKEY_LOCAL_MACHINE,
|
||
|
PROFILE_LIST_PATH,
|
||
|
0, KEY_READ, &hKeyProfilesList) != ERROR_SUCCESS) {
|
||
|
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to open ProfileList key")));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Enumerate the profiles
|
||
|
//
|
||
|
|
||
|
dwIndex = 0;
|
||
|
dwSIDNameSize = ARRAYSIZE(szSIDName);
|
||
|
lResult = RegEnumKeyEx(hKeyProfilesList,
|
||
|
dwIndex,
|
||
|
szSIDName,
|
||
|
&dwSIDNameSize,
|
||
|
NULL, NULL, NULL, &ft);
|
||
|
|
||
|
|
||
|
while (lResult == ERROR_SUCCESS) {
|
||
|
|
||
|
if (RegOpenKeyEx (hKeyProfilesList,
|
||
|
szSIDName,
|
||
|
0, KEY_READ, &hKeyProfile) == ERROR_SUCCESS) {
|
||
|
|
||
|
dwSize = sizeof(szProfilePath);
|
||
|
if (RegQueryValueEx (hKeyProfile,
|
||
|
PROFILE_IMAGE_VALUE_NAME,
|
||
|
NULL, NULL,
|
||
|
(LPBYTE) szProfilePath,
|
||
|
&dwSize) != ERROR_SUCCESS) {
|
||
|
goto NextProfile;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Expand the profile image filename
|
||
|
//
|
||
|
|
||
|
ExpandEnvironmentStrings(szProfilePath, szExpandedProfilePath, MAX_PATH);
|
||
|
|
||
|
//
|
||
|
// Create the acl for the user profile dir
|
||
|
//
|
||
|
|
||
|
//
|
||
|
// Get the owner sid
|
||
|
//
|
||
|
|
||
|
if (AllocateAndInitSidFromString(szSIDName, &pSidOwner) != STATUS_SUCCESS) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to create owner sid."), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Allocate space for the Dir object ACL
|
||
|
//
|
||
|
|
||
|
cbAcl = (2 * GetLengthSid (pSidOwner)) +
|
||
|
(2 * GetLengthSid (pSidSystem)) +
|
||
|
(2 * GetLengthSid (pSidAdmin)) +
|
||
|
sizeof(ACL) +
|
||
|
(6 * (sizeof(ACCESS_ALLOWED_ACE) - sizeof(DWORD)));
|
||
|
|
||
|
|
||
|
pDirAcl = (PACL) GlobalAlloc(GMEM_FIXED, cbAcl);
|
||
|
if (!pDirAcl) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to allocate memory for acl. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
if (!InitializeAcl(pDirAcl, cbAcl, ACL_REVISION)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to initialize acl. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Allocate space for File object ACL
|
||
|
//
|
||
|
|
||
|
cbAcl = (GetLengthSid (pSidOwner)) +
|
||
|
(GetLengthSid (pSidSystem)) +
|
||
|
(GetLengthSid (pSidAdmin)) +
|
||
|
sizeof(ACL) +
|
||
|
(3 * (sizeof(ACCESS_ALLOWED_ACE) - sizeof(DWORD)));
|
||
|
|
||
|
|
||
|
pFileAcl = (PACL) GlobalAlloc(GMEM_FIXED, cbAcl);
|
||
|
if (!pFileAcl) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to allocate memory for acl. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
if (!InitializeAcl(pFileAcl, cbAcl, ACL_REVISION)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to initialize acl. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Add Aces. Non-inheritable ACEs first
|
||
|
//
|
||
|
|
||
|
aceIndex = 0;
|
||
|
if (!AddAccessAllowedAce(pDirAcl, ACL_REVISION, FILE_ALL_ACCESS, pSidOwner)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to add owner ace. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
if (!AddAccessAllowedAce(pFileAcl, ACL_REVISION, FILE_ALL_ACCESS, pSidOwner)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to add owner ace. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
aceIndex++;
|
||
|
if (!AddAccessAllowedAce(pDirAcl, ACL_REVISION, FILE_ALL_ACCESS, pSidSystem)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to add system ace. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
if (!AddAccessAllowedAce(pFileAcl, ACL_REVISION, FILE_ALL_ACCESS, pSidSystem)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to add system ace. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
aceIndex++;
|
||
|
if (!AddAccessAllowedAce(pDirAcl, ACL_REVISION, FILE_ALL_ACCESS, pSidAdmin)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to add builtin admin ace. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
if (!AddAccessAllowedAce(pFileAcl, ACL_REVISION, FILE_ALL_ACCESS, pSidAdmin)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to add builtin admin ace. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Now the Inheritable Aces.
|
||
|
//
|
||
|
|
||
|
aceIndex++;
|
||
|
if (!AddAccessAllowedAceEx(pDirAcl, ACL_REVISION, OBJECT_INHERIT_ACE | CONTAINER_INHERIT_ACE | INHERIT_ONLY_ACE, GENERIC_ALL, pSidOwner)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to add owner ace. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
aceIndex++;
|
||
|
if (!AddAccessAllowedAceEx(pDirAcl, ACL_REVISION, OBJECT_INHERIT_ACE | CONTAINER_INHERIT_ACE | INHERIT_ONLY_ACE, GENERIC_ALL, pSidSystem)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to add system ace. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
aceIndex++;
|
||
|
if (!AddAccessAllowedAceEx(pDirAcl, ACL_REVISION, OBJECT_INHERIT_ACE | CONTAINER_INHERIT_ACE | INHERIT_ONLY_ACE, GENERIC_ALL, pSidAdmin)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to add builtin admin ace. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Put together the security descriptor
|
||
|
//
|
||
|
|
||
|
if (!InitializeSecurityDescriptor(&DirSd, SECURITY_DESCRIPTOR_REVISION)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to initialize security descriptor. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
|
||
|
if (!SetSecurityDescriptorDacl(&DirSd, TRUE, pDirAcl, FALSE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to set security descriptor dacl. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
if (!InitializeSecurityDescriptor(&FileSd, SECURITY_DESCRIPTOR_REVISION)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to initialize security descriptor. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
|
||
|
if (!SetSecurityDescriptorDacl(&FileSd, TRUE, pFileAcl, FALSE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to set security descriptor dacl. Error = %d"), GetLastError()));
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Pass the profile path to SecureProfile for securing
|
||
|
// the profile dir and nested subdirs and files.
|
||
|
//
|
||
|
|
||
|
if (!SecureNestedDir (szExpandedProfilePath, &DirSd, &FileSd)) {
|
||
|
goto IssueError;
|
||
|
}
|
||
|
|
||
|
goto NextProfile;
|
||
|
|
||
|
IssueError:
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to secure %s Profile directory"), szExpandedProfilePath));
|
||
|
|
||
|
}
|
||
|
|
||
|
NextProfile:
|
||
|
|
||
|
// Free the allocated stuffs
|
||
|
|
||
|
if (hKeyProfile) {
|
||
|
RegCloseKey(hKeyProfile);
|
||
|
hKeyProfile = NULL;
|
||
|
}
|
||
|
|
||
|
if (pSidOwner) {
|
||
|
FreeSid(pSidOwner);
|
||
|
pSidOwner = NULL;
|
||
|
}
|
||
|
|
||
|
if (pDirAcl) {
|
||
|
GlobalFree (pDirAcl);
|
||
|
pDirAcl = NULL;
|
||
|
}
|
||
|
|
||
|
if (pFileAcl) {
|
||
|
GlobalFree (pFileAcl);
|
||
|
pFileAcl = NULL;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Reset for the next loop
|
||
|
//
|
||
|
|
||
|
dwIndex++;
|
||
|
dwSIDNameSize = ARRAYSIZE(szSIDName);
|
||
|
lResult = RegEnumKeyEx(hKeyProfilesList,
|
||
|
dwIndex,
|
||
|
szSIDName,
|
||
|
&dwSIDNameSize,
|
||
|
NULL, NULL, NULL, &ft);
|
||
|
}
|
||
|
|
||
|
RegCloseKey(hKeyProfilesList);
|
||
|
|
||
|
Exit:
|
||
|
|
||
|
if (pSidSystem) {
|
||
|
FreeSid(pSidSystem);
|
||
|
}
|
||
|
|
||
|
if (pSidAdmin) {
|
||
|
FreeSid(pSidAdmin);
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|
||
|
|
||
|
//*************************************************************************
|
||
|
//
|
||
|
// HideSpecialProfiles()
|
||
|
//
|
||
|
// Routine Description :
|
||
|
// This function secures special profiles for which PI_HIDEPROFILE
|
||
|
// flag is specifed, e.g. LocalService, NetworkService etc. except
|
||
|
// system account profile. This mark the profile dir as super hidden.
|
||
|
//
|
||
|
// Arguments :
|
||
|
// None.
|
||
|
//
|
||
|
// Return Value :
|
||
|
// None.
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 8/8/00 santanuc Created
|
||
|
//
|
||
|
//*************************************************************************
|
||
|
|
||
|
void HideSpecialProfiles(void)
|
||
|
{
|
||
|
HKEY hKeyProfilesList, hKeyProfile = NULL;
|
||
|
TCHAR szSIDName[MAX_PATH], szProfilePath[MAX_PATH], szExpandedProfilePath[MAX_PATH];
|
||
|
DWORD dwSIDNameSize, dwSize;
|
||
|
DWORD dwIndex, dwFlags;
|
||
|
LONG lResult;
|
||
|
FILETIME ft;
|
||
|
|
||
|
|
||
|
//
|
||
|
// Open the ProfileList key
|
||
|
//
|
||
|
|
||
|
if (RegOpenKeyEx (HKEY_LOCAL_MACHINE,
|
||
|
PROFILE_LIST_PATH,
|
||
|
0, KEY_READ, &hKeyProfilesList) != ERROR_SUCCESS) {
|
||
|
|
||
|
DebugMsg((DM_WARNING, TEXT("HideSpecialProfiles: Failed to open ProfileList key")));
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Enumerate the profiles
|
||
|
//
|
||
|
|
||
|
dwIndex = 0;
|
||
|
dwSIDNameSize = ARRAYSIZE(szSIDName);
|
||
|
lResult = RegEnumKeyEx(hKeyProfilesList,
|
||
|
dwIndex,
|
||
|
szSIDName,
|
||
|
&dwSIDNameSize,
|
||
|
NULL, NULL, NULL, &ft);
|
||
|
|
||
|
|
||
|
while (lResult == ERROR_SUCCESS) {
|
||
|
|
||
|
if (RegOpenKeyEx (hKeyProfilesList,
|
||
|
szSIDName,
|
||
|
0, KEY_READ, &hKeyProfile) == ERROR_SUCCESS) {
|
||
|
|
||
|
//
|
||
|
// Process only if PI_HIDEPROFILE flag is set
|
||
|
//
|
||
|
|
||
|
dwSize = sizeof(DWORD);
|
||
|
if (RegQueryValueEx (hKeyProfile,
|
||
|
PROFILE_FLAGS,
|
||
|
NULL, NULL,
|
||
|
(LPBYTE) &dwFlags,
|
||
|
&dwSize) != ERROR_SUCCESS) {
|
||
|
goto NextProfile;
|
||
|
}
|
||
|
|
||
|
if (!(dwFlags & PI_HIDEPROFILE)) {
|
||
|
goto NextProfile;
|
||
|
}
|
||
|
|
||
|
dwSize = sizeof(szProfilePath);
|
||
|
if (RegQueryValueEx (hKeyProfile,
|
||
|
PROFILE_IMAGE_VALUE_NAME,
|
||
|
NULL, NULL,
|
||
|
(LPBYTE) szProfilePath,
|
||
|
&dwSize) != ERROR_SUCCESS) {
|
||
|
goto NextProfile;
|
||
|
}
|
||
|
|
||
|
// Ignore profile for system account
|
||
|
|
||
|
if (lstrcmp(szProfilePath, SYSTEM_PROFILE_LOCATION) == 0) {
|
||
|
goto NextProfile;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Expand the profile image filename
|
||
|
//
|
||
|
|
||
|
ExpandEnvironmentStrings(szProfilePath, szExpandedProfilePath, MAX_PATH);
|
||
|
|
||
|
// Mark the profile hidden
|
||
|
SetFileAttributes(szExpandedProfilePath,
|
||
|
FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_SYSTEM |
|
||
|
GetFileAttributes(szExpandedProfilePath));
|
||
|
|
||
|
}
|
||
|
|
||
|
NextProfile:
|
||
|
|
||
|
// Free the allocated stuffs
|
||
|
|
||
|
if (hKeyProfile) {
|
||
|
RegCloseKey(hKeyProfile);
|
||
|
hKeyProfile = NULL;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Reset for the next loop
|
||
|
//
|
||
|
|
||
|
dwIndex++;
|
||
|
dwSIDNameSize = ARRAYSIZE(szSIDName);
|
||
|
lResult = RegEnumKeyEx(hKeyProfilesList,
|
||
|
dwIndex,
|
||
|
szSIDName,
|
||
|
&dwSIDNameSize,
|
||
|
NULL, NULL, NULL, &ft);
|
||
|
}
|
||
|
|
||
|
RegCloseKey(hKeyProfilesList);
|
||
|
|
||
|
}
|
||
|
|
||
|
|
||
|
//*************************************************************
|
||
|
//
|
||
|
// CopySystemProfile()
|
||
|
//
|
||
|
// Purpose: Create the system profile information under
|
||
|
// ProfileList entry.
|
||
|
// In case of upgrade copy system profile from older
|
||
|
// location to new location and delete the old system
|
||
|
// profile
|
||
|
//
|
||
|
// Parameters:
|
||
|
//
|
||
|
// Return: TRUE if successful
|
||
|
// FALSE if an error occurs. Call GetLastError()
|
||
|
//
|
||
|
// Comments: This should only be called by GUI mode setup!
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 03/13/01 santanuc Created
|
||
|
//
|
||
|
//*************************************************************
|
||
|
|
||
|
BOOL WINAPI CopySystemProfile(BOOL bCleanInstall)
|
||
|
{
|
||
|
HANDLE hToken = NULL;
|
||
|
LPTSTR SidString = NULL, lpEnd;
|
||
|
TCHAR szLocalProfileKey[MAX_PATH], szBuffer[MAX_PATH];
|
||
|
TCHAR szSrc[MAX_PATH], szDest[MAX_PATH];
|
||
|
LONG lResult;
|
||
|
HKEY hKey = NULL, hKeyShellFolders;
|
||
|
DWORD dwFlags, dwInternalFlags, dwRefCount;
|
||
|
DWORD dwDisp, dwSize, dwType, i;
|
||
|
BOOL bCopySystemProfile = TRUE, bCopyFromDefault = TRUE;
|
||
|
PSID pSystemSid = NULL;
|
||
|
DWORD dwErr = ERROR_SUCCESS;
|
||
|
|
||
|
//
|
||
|
// Set the Profile information for system account i.e sid s-1-5-18
|
||
|
//
|
||
|
|
||
|
if (!OpenProcessToken(GetCurrentProcess(), TOKEN_IMPERSONATE |
|
||
|
TOKEN_QUERY | TOKEN_DUPLICATE, &hToken)) {
|
||
|
dwErr = GetLastError();
|
||
|
DebugMsg((DM_WARNING, TEXT("CopySystemProfile: Fail to open system token. Error %d"), dwErr));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Get the Sid string
|
||
|
//
|
||
|
|
||
|
SidString = GetSidString(hToken);
|
||
|
if (!SidString) {
|
||
|
dwErr = ERROR_ACCESS_DENIED;
|
||
|
DebugMsg((DM_WARNING, TEXT("CopySystemProfile: Failed to get sid string for system account")));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
pSystemSid = GetUserSid(hToken);
|
||
|
if (!pSystemSid) {
|
||
|
dwErr = ERROR_ACCESS_DENIED;
|
||
|
DebugMsg((DM_WARNING, TEXT("CopySystemProfile: Failed to get sid for system account")));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Open the profile mapping
|
||
|
//
|
||
|
|
||
|
lstrcpy(szLocalProfileKey, PROFILE_LIST_PATH);
|
||
|
lpEnd = CheckSlash (szLocalProfileKey);
|
||
|
lstrcpy(lpEnd, SidString);
|
||
|
|
||
|
lResult = RegCreateKeyEx(HKEY_LOCAL_MACHINE, szLocalProfileKey, 0, 0, 0,
|
||
|
KEY_READ | KEY_WRITE, NULL, &hKey, &dwDisp);
|
||
|
|
||
|
if (lResult != ERROR_SUCCESS) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CopySystemProfile: Failed to open profile mapping key. Error %d"), lResult));
|
||
|
dwErr = lResult;
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Save the flags
|
||
|
//
|
||
|
|
||
|
dwFlags = PI_LITELOAD | PI_HIDEPROFILE;
|
||
|
lResult = RegSetValueEx (hKey,
|
||
|
PROFILE_FLAGS,
|
||
|
0,
|
||
|
REG_DWORD,
|
||
|
(LPBYTE) &dwFlags,
|
||
|
sizeof(DWORD));
|
||
|
|
||
|
if (lResult != ERROR_SUCCESS) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CopySystemProfile: Failed to save profile flags. Error %d"), lResult));
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Save the internal flags
|
||
|
//
|
||
|
|
||
|
dwInternalFlags = 0;
|
||
|
lResult = RegSetValueEx (hKey,
|
||
|
PROFILE_STATE,
|
||
|
0,
|
||
|
REG_DWORD,
|
||
|
(LPBYTE) &dwInternalFlags,
|
||
|
sizeof(DWORD));
|
||
|
|
||
|
if (lResult != ERROR_SUCCESS) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CopySystemProfile: Failed to save internal flags. Error %d"), lResult));
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Save the ref count
|
||
|
//
|
||
|
|
||
|
dwRefCount = 1;
|
||
|
lResult = RegSetValueEx (hKey,
|
||
|
PROFILE_REF_COUNT,
|
||
|
0,
|
||
|
REG_DWORD,
|
||
|
(LPBYTE) &dwRefCount,
|
||
|
sizeof(DWORD));
|
||
|
|
||
|
if (lResult != ERROR_SUCCESS) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CopySystemProfile: Failed to save profile ref count. Error %d"), lResult));
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Save the sid
|
||
|
//
|
||
|
|
||
|
lResult = RegSetValueEx (hKey,
|
||
|
TEXT("Sid"),
|
||
|
0,
|
||
|
REG_BINARY,
|
||
|
(LPBYTE) pSystemSid,
|
||
|
RtlLengthSid(pSystemSid));
|
||
|
|
||
|
if (lResult != ERROR_SUCCESS) {
|
||
|
DebugMsg((DM_WARNING, TEXT("CopySystemProfile: Failed to save profile system sid. Error %d"), lResult));
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Figure out whether any existing system profile exist or not
|
||
|
// in upgrade scenario
|
||
|
//
|
||
|
|
||
|
lstrcpy(szBuffer, SYSTEM_PROFILE_LOCATION);
|
||
|
ExpandEnvironmentStrings(szBuffer, szDest, MAX_PATH);
|
||
|
|
||
|
if (!bCleanInstall) {
|
||
|
dwSize = ARRAYSIZE(szBuffer) * sizeof(TCHAR);
|
||
|
lResult = RegQueryValueEx(hKey,
|
||
|
PROFILE_IMAGE_VALUE_NAME,
|
||
|
NULL,
|
||
|
&dwType,
|
||
|
(LPBYTE) szBuffer,
|
||
|
&dwSize);
|
||
|
if (ERROR_SUCCESS == lResult) {
|
||
|
if (lstrcmp(szBuffer, SYSTEM_PROFILE_LOCATION) == 0) {
|
||
|
bCopySystemProfile = FALSE;
|
||
|
}
|
||
|
else {
|
||
|
ExpandEnvironmentStrings(szBuffer, szSrc, MAX_PATH);
|
||
|
bCopyFromDefault = FALSE;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (bCopySystemProfile) {
|
||
|
|
||
|
if (!CreateSecureDirectory(NULL, szDest, pSystemSid, FALSE)) {
|
||
|
dwErr = GetLastError();
|
||
|
DebugMsg((DM_WARNING, TEXT("CopySystemProfile: Fail to create SystemProfile dir. Error %d"), dwErr));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
|
||
|
if (bCopyFromDefault) {
|
||
|
dwSize = ARRAYSIZE(szSrc);
|
||
|
if (!GetDefaultUserProfileDirectoryEx(szSrc, &dwSize, TRUE)) {
|
||
|
dwErr = GetLastError();
|
||
|
DebugMsg((DM_WARNING, TEXT("CopySystemProfile: Failed to get default user profile. Error %d"), dwErr));
|
||
|
goto Exit;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Copy the existing or Default user profile in System profile location
|
||
|
//
|
||
|
|
||
|
if (!CopyProfileDirectoryEx(szSrc, szDest, CPD_IGNOREHIVE | CPD_FORCECOPY | CPD_IGNORECOPYERRORS,
|
||
|
NULL, NULL)) {
|
||
|
dwErr = GetLastError();
|
||
|
DebugMsg((DM_WARNING, TEXT("CopySystemProfile: Failed to copy system profile. Error %d"), dwErr));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
if (!bCopyFromDefault) {
|
||
|
Delnode(szSrc);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
SetAclForSystemProfile(pSystemSid, szDest);
|
||
|
|
||
|
//
|
||
|
// Save local profile path
|
||
|
//
|
||
|
|
||
|
lstrcpy(szBuffer, SYSTEM_PROFILE_LOCATION);
|
||
|
lResult = RegSetValueEx (hKey,
|
||
|
PROFILE_IMAGE_VALUE_NAME,
|
||
|
0,
|
||
|
REG_EXPAND_SZ,
|
||
|
(LPBYTE) szBuffer,
|
||
|
((lstrlen(szBuffer) + 1) * sizeof(TCHAR)));
|
||
|
|
||
|
if (lResult != ERROR_SUCCESS) {
|
||
|
dwErr = lResult;
|
||
|
DebugMsg((DM_WARNING, TEXT("CopySystemProfile: Failed to save profile image path with error %d"), lResult));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Set the Shell Folder paths in the registry for system account
|
||
|
//
|
||
|
|
||
|
lstrcpy (szBuffer, TEXT(".Default"));
|
||
|
lpEnd = CheckSlash (szBuffer);
|
||
|
lstrcpy(lpEnd, SHELL_FOLDERS);
|
||
|
|
||
|
lpEnd = CheckSlash (szDest);
|
||
|
|
||
|
if (RegCreateKeyEx (HKEY_USERS, szBuffer,
|
||
|
0, NULL, REG_OPTION_NON_VOLATILE,
|
||
|
KEY_READ | KEY_WRITE, NULL, &hKeyShellFolders,
|
||
|
&dwDisp) == ERROR_SUCCESS) {
|
||
|
|
||
|
for (i=0; i < g_dwNumShellFolders; i++) {
|
||
|
|
||
|
if (c_ShellFolders[i].bAddCSIDL) {
|
||
|
lstrcpy (lpEnd, c_ShellFolders[i].szFolderLocation);
|
||
|
|
||
|
RegSetValueEx (hKeyShellFolders, c_ShellFolders[i].lpFolderName,
|
||
|
0, REG_SZ, (LPBYTE) szDest,
|
||
|
((lstrlen(szDest) + 1) * sizeof(TCHAR)));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
RegCloseKey (hKeyShellFolders);
|
||
|
}
|
||
|
|
||
|
Exit:
|
||
|
|
||
|
if (hToken) {
|
||
|
CloseHandle(hToken);
|
||
|
}
|
||
|
|
||
|
if (SidString) {
|
||
|
DeleteSidString(SidString);
|
||
|
}
|
||
|
|
||
|
if (pSystemSid) {
|
||
|
DeleteUserSid(pSystemSid);
|
||
|
}
|
||
|
|
||
|
if (hKey) {
|
||
|
RegCloseKey (hKey);
|
||
|
}
|
||
|
|
||
|
SetLastError(dwErr);
|
||
|
|
||
|
return(dwErr == ERROR_SUCCESS ? TRUE : FALSE);
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
//*************************************************************************
|
||
|
//
|
||
|
// SetAclForSystemProfile()
|
||
|
//
|
||
|
// Routine Description :
|
||
|
// This function secures dir/files of system profile i.e.
|
||
|
// %windir%\system32\config\systemprofile
|
||
|
//
|
||
|
// Arguments :
|
||
|
// pSidSystem - System sid
|
||
|
// szExpandedProfilePath - Expanded system profile location
|
||
|
//
|
||
|
// Return Value :
|
||
|
// None.
|
||
|
//
|
||
|
// History: Date Author Comment
|
||
|
// 04/06/01 santanuc Created
|
||
|
//
|
||
|
//*************************************************************************
|
||
|
|
||
|
void SetAclForSystemProfile(PSID pSidSystem, LPTSTR szExpandedProfilePath)
|
||
|
{
|
||
|
SECURITY_DESCRIPTOR DirSd, FileSd;
|
||
|
SID_IDENTIFIER_AUTHORITY authNT = SECURITY_NT_AUTHORITY;
|
||
|
PACL pDirAcl = NULL, pFileAcl = NULL;
|
||
|
PSID pSidAdmin = NULL;
|
||
|
DWORD cbAcl, aceIndex;
|
||
|
|
||
|
|
||
|
//
|
||
|
// Get the Admin sid
|
||
|
//
|
||
|
|
||
|
if (!AllocateAndInitializeSid(&authNT, 2, SECURITY_BUILTIN_DOMAIN_RID,
|
||
|
DOMAIN_ALIAS_RID_ADMINS, 0, 0,
|
||
|
0, 0, 0, 0, &pSidAdmin)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SecureUserProfiles: Failed to initialize admin sid. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Allocate space for the Dir object ACL
|
||
|
//
|
||
|
|
||
|
cbAcl = (2 * GetLengthSid (pSidSystem)) +
|
||
|
(2 * GetLengthSid (pSidAdmin)) +
|
||
|
sizeof(ACL) +
|
||
|
(4 * (sizeof(ACCESS_ALLOWED_ACE) - sizeof(DWORD)));
|
||
|
|
||
|
|
||
|
pDirAcl = (PACL) GlobalAlloc(GMEM_FIXED, cbAcl);
|
||
|
if (!pDirAcl) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SetAclForSystemProfile: Failed to allocate memory for acl. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
if (!InitializeAcl(pDirAcl, cbAcl, ACL_REVISION)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SetAclForSystemProfile: Failed to initialize acl. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Allocate space for File object ACL
|
||
|
//
|
||
|
|
||
|
cbAcl = (GetLengthSid (pSidSystem)) +
|
||
|
(GetLengthSid (pSidAdmin)) +
|
||
|
sizeof(ACL) +
|
||
|
(2 * (sizeof(ACCESS_ALLOWED_ACE) - sizeof(DWORD)));
|
||
|
|
||
|
|
||
|
pFileAcl = (PACL) GlobalAlloc(GMEM_FIXED, cbAcl);
|
||
|
if (!pFileAcl) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SetAclForSystemProfile: Failed to allocate memory for acl. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
if (!InitializeAcl(pFileAcl, cbAcl, ACL_REVISION)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SetAclForSystemProfile: Failed to initialize acl. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Add Aces. Non-inheritable ACEs first
|
||
|
//
|
||
|
|
||
|
aceIndex = 0;
|
||
|
if (!AddAccessAllowedAce(pDirAcl, ACL_REVISION, FILE_ALL_ACCESS, pSidSystem)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SetAclForSystemProfile: Failed to add system ace. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
if (!AddAccessAllowedAce(pFileAcl, ACL_REVISION, FILE_ALL_ACCESS, pSidSystem)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SetAclForSystemProfile: Failed to add system ace. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
aceIndex++;
|
||
|
if (!AddAccessAllowedAce(pDirAcl, ACL_REVISION, FILE_ALL_ACCESS, pSidAdmin)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SetAclForSystemProfile: Failed to add builtin admin ace. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
if (!AddAccessAllowedAce(pFileAcl, ACL_REVISION, FILE_ALL_ACCESS, pSidAdmin)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SetAclForSystemProfile: Failed to add builtin admin ace. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Now the Inheritable Aces.
|
||
|
//
|
||
|
|
||
|
aceIndex++;
|
||
|
if (!AddAccessAllowedAceEx(pDirAcl, ACL_REVISION, OBJECT_INHERIT_ACE | CONTAINER_INHERIT_ACE | INHERIT_ONLY_ACE, GENERIC_ALL, pSidSystem)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SetAclForSystemProfile: Failed to add system ace. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
aceIndex++;
|
||
|
if (!AddAccessAllowedAceEx(pDirAcl, ACL_REVISION, OBJECT_INHERIT_ACE | CONTAINER_INHERIT_ACE | INHERIT_ONLY_ACE, GENERIC_ALL, pSidAdmin)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SetAclForSystemProfile: Failed to add builtin admin ace. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
|
||
|
//
|
||
|
// Put together the security descriptor
|
||
|
//
|
||
|
|
||
|
if (!InitializeSecurityDescriptor(&DirSd, SECURITY_DESCRIPTOR_REVISION)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SetAclForSystemProfile: Failed to initialize security descriptor. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
|
||
|
if (!SetSecurityDescriptorDacl(&DirSd, TRUE, pDirAcl, FALSE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SetAclForSystemProfile: Failed to set security descriptor dacl. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
if (!InitializeSecurityDescriptor(&FileSd, SECURITY_DESCRIPTOR_REVISION)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SetAclForSystemProfile: Failed to initialize security descriptor. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
|
||
|
if (!SetSecurityDescriptorDacl(&FileSd, TRUE, pFileAcl, FALSE)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SetAclForSystemProfile: Failed to set security descriptor dacl. Error = %d"), GetLastError()));
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Pass the profile path to SecureProfile for securing
|
||
|
// the profile dir and nested subdirs and files.
|
||
|
//
|
||
|
|
||
|
if (!SecureNestedDir (szExpandedProfilePath, &DirSd, &FileSd)) {
|
||
|
DebugMsg((DM_WARNING, TEXT("SetAclForSystemProfile: Failed to secure %s Profile directory"), szExpandedProfilePath));
|
||
|
}
|
||
|
|
||
|
Exit:
|
||
|
|
||
|
if (pDirAcl) {
|
||
|
GlobalFree (pDirAcl);
|
||
|
pDirAcl = NULL;
|
||
|
}
|
||
|
|
||
|
if (pFileAcl) {
|
||
|
GlobalFree (pFileAcl);
|
||
|
pFileAcl = NULL;
|
||
|
}
|
||
|
|
||
|
if (pSidAdmin) {
|
||
|
FreeSid(pSidAdmin);
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|