archive
/
windows-server-2003
mirror of https://github.com/selfrender/Windows-Server-2003
2
0
Fork 0
Code Issues Projects Releases Wiki Activity
Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
4 Commits
1 Branch
0 Tags
1.5 GiB
 
 
 
 
 
 
Branch: master
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'master'
${ noResults }
windows-server-2003/shell/cpls/system/profile.cpp

4284 lines
113 KiB
Raw Permalink Blame History

//*************************************************************
//
// Profile.c - User Profile tab
//
// Microsoft Confidential
// Copyright (c) Microsoft Corporation 1996
// All rights reserved
//
//*************************************************************
#include <nt.h>
#include <ntrtl.h>
#include <nturtl.h>
#include "sysdm.h"
#include <shellapi.h>
#include <shlobj.h>
#include <shlobjp.h>
#include <userenv.h>
#include <userenvp.h>
#include <getuser.h>
#include <objsel.h>
#include <strsafe.h>
#define TEMP_PROFILE TEXT("Temp profile (sysdm.cpl)")
#define USER_CRED_LOCATION TEXT("Application Data\\Microsoft\\Credentials")
#define PROFILE_MAPPING TEXT("Software\\Microsoft\\Windows NT\\CurrentVersion\\ProfileList")
#define WINLOGON_KEY TEXT("Software\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon")
#define SYSTEM_POLICIES_KEY TEXT("Software\\Policies\\Microsoft\\Windows\\System")
#define READONLY_RUP TEXT("ReadOnlyProfile")
#define PROFILE_LOCALONLY TEXT("LocalProfile")
#define PRF_USERSID 1
//
// Globals
//
const TCHAR c_szNTUserIni[] = TEXT("ntuser.ini");
#define PROFILE_GENERAL_SECTION TEXT("General")
#define PROFILE_EXCLUSION_LIST TEXT("ExclusionList")
DWORD g_dwProfileSize;
//
// Help ID's
//
// IDH_USERPROFILE + 20 is used in the OpenUserBrowser routine.
//
DWORD aUserProfileHelpIds[] = {
IDC_UP_LISTVIEW, (IDH_USERPROFILE + 0),
IDC_UP_DELETE, (IDH_USERPROFILE + 1),
IDC_UP_TYPE, (IDH_USERPROFILE + 2),
IDC_UP_COPY, (IDH_USERPROFILE + 3),
IDC_UP_ICON, (DWORD) -1,
IDC_UP_TEXT, (DWORD) -1,
// Change Type dialog
IDC_UPTYPE_LOCAL, (IDH_USERPROFILE + 4),
IDC_UPTYPE_FLOAT, (IDH_USERPROFILE + 5),
// removed IDC_UPTYPE_SLOW, IDC_UPTYPE_SLOW_TEXT
IDC_UPTYPE_GROUP, (IDH_USERPROFILE + 12),
// Copy To dialog
IDC_COPY_PROFILE, (IDH_USERPROFILE + 7),
IDC_COPY_PATH, (IDH_USERPROFILE + 7),
IDC_COPY_BROWSE, (IDH_USERPROFILE + 8),
IDC_COPY_USER, (IDH_USERPROFILE + 9),
IDC_COPY_CHANGE, (IDH_USERPROFILE + 10),
IDC_COPY_GROUP, (IDH_USERPROFILE + 9),
0, 0
};
//
// Local function proto-types
//
BOOL InitUserProfileDlg (HWND hDlg, LPARAM lParam);
BOOL FillListView (HWND hDlg, BOOL bAdmin);
BOOL RecurseDirectory (LPTSTR lpDir, UINT cchBuffer, LPTSTR lpExcludeList);
VOID UPSave (HWND hDlg);
VOID UPCleanUp (HWND hDlg);
BOOL IsProfileInUse (LPTSTR lpSid);
void UPDoItemChanged(HWND hDlg, int idCtl);
void UPDeleteProfile(HWND hDlg);
void UPChangeType(HWND hDlg);
void UPOnLink(WPARAM wParam);
INT_PTR APIENTRY ChangeTypeDlgProc (HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam);
void UPCopyProfile(HWND hDlg);
INT_PTR APIENTRY UPCopyDlgProc (HWND hDlg, UINT uMsg,WPARAM wParam, LPARAM lParam);
BOOL UPCreateProfile (HWND hDlg, LPUPCOPYINFO lpUPCopyInfo, LPTSTR lpDest, PSECURITY_DESCRIPTOR pSecDesc);
VOID UPDisplayErrorMessage(HWND hWnd, UINT uiSystemError, LPTSTR szMsgPrefix);
BOOL ApplyHiveSecurity(LPTSTR lpHiveName, PSID pSid);
BOOL CheckSemicolon (LPTSTR lpDir, UINT cchBuffer);
LPTSTR ConvertExclusionList (LPCTSTR lpSourceDir, LPCTSTR lpExclusionList);
BOOL ReadExclusionList(HKEY hKeyUser, LPTSTR szExcludeList, DWORD cchExcludeList);
BOOL ReadExclusionListFromIniFile(LPCTSTR lpSourceDir, LPTSTR szExcludeList, DWORD cchExcludeList);
HRESULT UPGetUserSelection(HWND hDlg, LPUSERDETAILS lpUserDetails);
BOOL ConfirmDirectory(HWND hDlg, LPTSTR szDir);
HRESULT CopyNonHiveNtUserFiles(LPCTSTR lpSrcDir, LPCTSTR lpDestDir);
LPTSTR CheckSlashEx(LPTSTR lpDir, UINT cchBuffer, UINT* pcchRemain );
//*************************************************************
//
// UserProfileDlgProc()
//
// Purpose: Dialog box procedure for profile tab
//
// Parameters: hDlg - handle to the dialog box
// uMsg - window message
// wParam - wParam
// lParam - lParam
//
// Return: TRUE if message was processed
// FALSE if not
//
// Comments:
//
// History: Date Author Comment
// 10/11/95 ericflo Created
//
//*************************************************************
INT_PTR APIENTRY UserProfileDlgProc (HWND hDlg, UINT uMsg,
WPARAM wParam, LPARAM lParam)
{
switch (uMsg) {
case WM_INITDIALOG:
if (!InitUserProfileDlg (hDlg, lParam)) {
EndDialog(hDlg, FALSE);
}
return TRUE;
case WM_NOTIFY:
switch (((NMHDR FAR*)lParam)->code)
{
case LVN_ITEMCHANGED:
UPDoItemChanged(hDlg, (int) wParam);
break;
case LVN_ITEMACTIVATE:
PostMessage (hDlg, WM_COMMAND, IDC_UP_TYPE, 0);
break;
case LVN_COLUMNCLICK:
break;
case NM_CLICK:
case NM_RETURN:
UPOnLink(wParam);
break;
default:
return FALSE;
}
break;
case WM_DESTROY:
UPCleanUp (hDlg);
break;
case WM_COMMAND:
switch (LOWORD(wParam)) {
case IDC_UP_DELETE:
UPDeleteProfile(hDlg);
break;
case IDC_UP_TYPE:
UPChangeType(hDlg);
break;
case IDC_UP_COPY:
UPCopyProfile(hDlg);
break;
case IDOK:
UPSave(hDlg);
EndDialog(hDlg, FALSE);
break;
case IDCANCEL:
EndDialog(hDlg, FALSE);
break;
default:
break;
}
break;
case WM_HELP: // F1
WinHelp((HWND)((LPHELPINFO) lParam)->hItemHandle, HELP_FILE, HELP_WM_HELP,
(DWORD_PTR) (LPSTR) aUserProfileHelpIds);
break;
case WM_CONTEXTMENU: // right mouse click
WinHelp((HWND) wParam, HELP_FILE, HELP_CONTEXTMENU,
(DWORD_PTR) (LPSTR) aUserProfileHelpIds);
return (TRUE);
}
return (FALSE);
}
//*************************************************************
//
// InitUserProfileDlg()
//
// Purpose: Initializes the User Profiles page
//
// Parameters: hDlg - Dialog box handle
//
// Return: TRUE if successful
// FALSE if an error occurs
//
// Comments:
//
// History: Date Author Comment
// 1/26/96 ericflo Created
//
//*************************************************************
BOOL InitUserProfileDlg (HWND hDlg, LPARAM lParam)
{
TCHAR szHeaderName[30];
LV_COLUMN col;
RECT rc;
HWND hLV;
INT iTotal = 0, iCurrent;
HWND hwndTemp;
BOOL bAdmin;
HCURSOR hOldCursor;
hOldCursor = SetCursor (LoadCursor(NULL, IDC_WAIT));
hLV = GetDlgItem(hDlg, IDC_UP_LISTVIEW);
// Set extended LV style for whole line selection
SendMessage(hLV, LVM_SETEXTENDEDLISTVIEWSTYLE, 0, LVS_EX_FULLROWSELECT);
//
// Insert Columns
//
GetClientRect (hLV, &rc);
LoadString(hInstance, IDS_UP_NAME, szHeaderName, ARRAYSIZE(szHeaderName));
col.mask = LVCF_FMT | LVCF_TEXT | LVCF_SUBITEM | LVCF_WIDTH;
col.fmt = LVCFMT_LEFT;
iCurrent = (int)(rc.right * .40);
iTotal += iCurrent;
col.cx = iCurrent;
col.pszText = szHeaderName;
col.iSubItem = 0;
ListView_InsertColumn (hLV, 0, &col);
LoadString(hInstance, IDS_UP_SIZE, szHeaderName, ARRAYSIZE(szHeaderName));
iCurrent = (int)(rc.right * .15);
iTotal += iCurrent;
col.cx = iCurrent;
col.fmt = LVCFMT_RIGHT;
col.iSubItem = 1;
ListView_InsertColumn (hLV, 1, &col);
LoadString(hInstance, IDS_UP_TYPE, szHeaderName, ARRAYSIZE(szHeaderName));
col.iSubItem = 2;
iCurrent = (int)(rc.right * .15);
iTotal += iCurrent;
col.cx = iCurrent;
col.fmt = LVCFMT_LEFT;
ListView_InsertColumn (hLV, 2, &col);
LoadString(hInstance, IDS_UP_STATUS, szHeaderName, ARRAYSIZE(szHeaderName));
col.iSubItem = 3;
iCurrent = (int)(rc.right * .15);
iTotal += iCurrent;
col.cx = iCurrent;
col.fmt = LVCFMT_LEFT;
ListView_InsertColumn (hLV, 3, &col);
LoadString(hInstance, IDS_UP_MOD, szHeaderName, ARRAYSIZE(szHeaderName));
col.iSubItem = 4;
col.cx = rc.right - iTotal - GetSystemMetrics(SM_CYHSCROLL);
col.fmt = LVCFMT_LEFT;
ListView_InsertColumn (hLV, 4, &col);
bAdmin = IsUserAnAdmin();
if (!bAdmin) {
POINT pt;
//
// If the user is not an admin, then we hide the
// delete and copy to buttons
//
hwndTemp = GetDlgItem (hDlg, IDC_UP_DELETE);
GetWindowRect (hwndTemp, &rc);
EnableWindow (hwndTemp, FALSE);
ShowWindow (hwndTemp, SW_HIDE);
hwndTemp = GetDlgItem (hDlg, IDC_UP_COPY);
EnableWindow (hwndTemp, FALSE);
ShowWindow (hwndTemp, SW_HIDE);
//
// Move the Change Type button over
//
pt.x = rc.left;
pt.y = rc.top;
ScreenToClient (hDlg, &pt);
SetWindowPos (GetDlgItem (hDlg, IDC_UP_TYPE),
HWND_TOP, pt.x, pt.y, 0, 0,
SWP_NOSIZE | SWP_NOZORDER);
}
if (IsOS(OS_ANYSERVER))
{
//
// Adjust the text displayed for the Users & Passwords link.
// There is no Control Panel applet per se on Server. The functionality
// is accessed through MMC.
//
TCHAR szTitle[80];
if (0 < LoadString(hInstance, IDS_UP_UPLINK_SERVER, szTitle, ARRAYSIZE(szTitle)))
{
SetWindowText(GetDlgItem(hDlg, IDC_UP_UPLINK), szTitle);
}
}
//
// Fill the user accounts
//
if (!FillListView (hDlg, bAdmin)) {
SetCursor(hOldCursor);
return FALSE;
}
SetCursor(hOldCursor);
return TRUE;
}
//*************************************************************
//
// AddUser()
//
// Purpose: Adds a new user to the listview
//
//
// Parameters: hDlg - handle to the dialog box
// lpSid - Sid (text form)
//
//
// Return: TRUE if successful
// FALSE if an error occurs
//
// Comments:
//
// History: Date Author Comment
// 1/26/96 ericflo Created
//
//*************************************************************
BOOL AddUser (HWND hDlg, LPTSTR lpSid, DWORD dwFlags)
{
LONG Error;
HKEY hKeyPolicy, hKeyProfile;
TCHAR szBuffer[2*MAX_PATH];
TCHAR szBuffer2[MAX_PATH];
TCHAR szTemp[100];
TCHAR szTemp2[100];
DWORD dwBufSize, dwBuf2Size;
PSID pSid;
DWORD dwSize, dwProfileFlags;
SID_NAME_USE SidName;
LV_ITEM item;
INT iItem;
HWND hwndTemp;
HKEY hkeyUser;
HANDLE hFile;
WIN32_FIND_DATA fd;
LPTSTR lpEnd;
SYSTEMTIME systime;
FILETIME ftLocal;
TCHAR szProfileSize[20];
INT iTypeID, iStatusID;
DWORD dwProfileType, dwProfileStatus, dwSysProfileType=0;
LPUSERINFO lpUserInfo;
BOOL bCentralAvailable = FALSE, bAccountUnknown;
TCHAR szExcludeList[2*MAX_PATH+1];
LPTSTR lpExcludeList = NULL;
UINT cchRemaining;
HRESULT hr;
DWORD dwFileAttr;
BOOL bGotTime = FALSE;
//
// Open the user's info
//
hr = StringCchPrintf(szBuffer, ARRAYSIZE(szBuffer), TEXT("%s\\%s"), PROFILE_MAPPING, lpSid);
if (FAILED(hr)) {
return FALSE;
}
Error = RegOpenKeyEx(HKEY_LOCAL_MACHINE,
szBuffer,
0,
KEY_READ,
&hkeyUser);
if (Error != ERROR_SUCCESS) {
return FALSE;
}
//
// If PI_HIDEPROFILE flag is set then do not display
// the user in system applet
//
dwSize = sizeof(DWORD);
Error = RegQueryValueEx (hkeyUser,
TEXT("Flags"),
NULL,
NULL,
(LPBYTE) &dwProfileFlags,
&dwSize);
if (Error == ERROR_SUCCESS && (dwProfileFlags & PI_HIDEPROFILE)) {
RegCloseKey (hkeyUser);
return FALSE;
}
//
// Query for the user's central profile location
//
dwSize = sizeof(szBuffer2);
Error = RegQueryValueEx (hkeyUser,
TEXT("CentralProfile"),
NULL,
NULL,
(LPBYTE) szBuffer2,
&dwSize);
if ((Error == ERROR_SUCCESS) && (szBuffer2[0] != TEXT('\0'))) {
bCentralAvailable = TRUE;
}
//
// Query for the user's local profile
//
dwSize = sizeof(szBuffer2);
Error = RegQueryValueEx (hkeyUser,
TEXT("ProfileImagePath"),
NULL,
NULL,
(LPBYTE) szBuffer2,
&dwSize);
if (Error != ERROR_SUCCESS) {
RegCloseKey (hkeyUser);
return FALSE;
}
//
// Profile paths need to be expanded
//
dwSize = ExpandEnvironmentStrings (szBuffer2, szBuffer, ARRAYSIZE(szBuffer));
if (!dwSize || dwSize > ARRAYSIZE(szBuffer)) {
RegCloseKey (hkeyUser);
return FALSE;
}
if (FAILED(StringCchCopy(szBuffer2, ARRAYSIZE(szBuffer2), szBuffer))) {
RegCloseKey (hkeyUser);
return FALSE;
}
//
// Check if this directory exists
//
dwFileAttr = GetFileAttributes(szBuffer);
if (dwFileAttr != INVALID_FILE_ATTRIBUTES)
{
//
// Directory exist, get the timestamp of the user hive and profile size
//
lpEnd = CheckSlashEx(szBuffer, ARRAYSIZE(szBuffer), &cchRemaining);
if (!lpEnd)
{
RegCloseKey(hkeyUser);
return FALSE;
}
if (FAILED(StringCchCopy(lpEnd, cchRemaining, TEXT("ntuser.man")))) {
RegCloseKey (hkeyUser);
return FALSE;
}
ZeroMemory(&fd, sizeof(fd));
hFile = FindFirstFile (szBuffer, &fd);
if (hFile != INVALID_HANDLE_VALUE)
{
FindClose (hFile);
bGotTime = TRUE;
}
else
{
if (FAILED(StringCchCopy(lpEnd, cchRemaining, TEXT("ntuser.dat")))) {
RegCloseKey (hkeyUser);
return FALSE;
}
hFile = FindFirstFile (szBuffer, &fd);
if (hFile != INVALID_HANDLE_VALUE)
{
FindClose(hFile);
bGotTime = TRUE;
}
}
*lpEnd = TEXT('\0');
//
// Get the exclusionlist from the registry or ini files
//
if (dwFlags & PRF_USERSID)
ReadExclusionList(NULL, szExcludeList, ARRAYSIZE(szExcludeList));
else
ReadExclusionListFromIniFile(szBuffer, szExcludeList, ARRAYSIZE(szExcludeList));
//
// Convert the exclusionlist read from the registry to a Null terminated list
// readable by recursedirectory.
//
if (szExcludeList[0] != TEXT('\0'))
lpExcludeList = ConvertExclusionList (szBuffer, szExcludeList);
else
lpExcludeList = NULL;
//
// Get the profile size
//
g_dwProfileSize = 0;
*lpEnd = TEXT('\0');
if (!RecurseDirectory (szBuffer, ARRAYSIZE(szBuffer), lpExcludeList)) {
g_dwProfileSize = (DWORD) (-1); // Something is wrong, set the size to -1
}
if (lpExcludeList) {
LocalFree (lpExcludeList);
lpExcludeList = NULL;
}
}
else
{
//
// We cannot get the attributes of the profile directory, it may not exist, or
// we don't have access permission to it, set the profile size to -1 to indicate
// this error.
//
g_dwProfileSize = (DWORD) (-1);
}
if (g_dwProfileSize == (DWORD) (-1))
{
if (FAILED(StringCchCopy(szProfileSize, ARRAYSIZE(szProfileSize), TEXT("-1"))))
{
RegCloseKey(hkeyUser);
return FALSE;
}
}
else if (!StrFormatByteSize((LONGLONG)g_dwProfileSize, szProfileSize, ARRAYSIZE(szProfileSize))) {
RegCloseKey (hkeyUser);
return FALSE;
}
//
// check out the state information to determine profile status.
// If the user is not logged on currently then report the status
// of last session.
//
dwSize = sizeof(DWORD);
Error = RegQueryValueEx (hkeyUser,
TEXT("State"),
NULL,
NULL,
(LPBYTE) &dwSysProfileType,
&dwSize);
//
// The constants being used come from
// windows\gina\userenv\profile\profile.h
//
if (dwSysProfileType & 0x00008000) {
dwProfileStatus = USERINFO_BACKUP;
}
else if (dwSysProfileType & 0x00000001) {
dwProfileStatus = USERINFO_MANDATORY;
}
else if ((dwSysProfileType & 0x00000800) ||
(dwSysProfileType & 0x00000080)) {
dwProfileStatus = USERINFO_TEMP;
}
else if (dwSysProfileType & 0x00000010) {
dwProfileStatus = USERINFO_FLOATING;
}
else {
dwProfileStatus = USERINFO_LOCAL;
}
if (dwSysProfileType & 0x00000001) {
dwProfileType = USERINFO_MANDATORY;
}
else {
if (bCentralAvailable) {
dwProfileType = USERINFO_FLOATING;
}
else {
dwProfileType = USERINFO_LOCAL;
}
// Check User Preference
HKEY hkeyPreference;
if (RegOpenKeyEx(hkeyUser,
TEXT("Preference"),
0,
KEY_READ,
&hkeyPreference) == ERROR_SUCCESS)
{
DWORD dwType;
dwSize = sizeof(dwProfileType);
Error = RegQueryValueEx (hkeyPreference,
TEXT("UserPreference"),
NULL,
&dwType,
(LPBYTE) &dwProfileType,
&dwSize);
RegCloseKey(hkeyPreference);
}
// Check User Preference in .bak if exist
if (FAILED(StringCchPrintf(szBuffer,
ARRAYSIZE(szBuffer),
TEXT("%s\\%s.bak\\Preference"),
PROFILE_MAPPING,
lpSid))) {
RegCloseKey (hkeyUser);
return FALSE;
}
Error = RegOpenKeyEx(HKEY_LOCAL_MACHINE,
szBuffer,
0,
KEY_READ,
&hKeyProfile);
if (Error == ERROR_SUCCESS) {
dwSize = sizeof(dwProfileType);
RegQueryValueEx(hKeyProfile,
TEXT("UserPreference"),
NULL,
NULL,
(LPBYTE) &dwProfileType,
&dwSize);
RegCloseKey (hKeyProfile);
}
// Check machine policy for disabling roaming profile
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,
SYSTEM_POLICIES_KEY,
0, KEY_READ,
&hKeyPolicy) == ERROR_SUCCESS) {
DWORD dwTmpVal;
dwSize = sizeof(dwTmpVal);
Error = RegQueryValueEx(hKeyPolicy,
PROFILE_LOCALONLY,
NULL, NULL,
(LPBYTE) &dwTmpVal,
&dwSize);
RegCloseKey (hKeyPolicy);
if (Error == ERROR_SUCCESS && dwTmpVal == 1) {
dwProfileType = USERINFO_LOCAL;
}
}
if (dwProfileType == USERINFO_FLOATING) {
BOOL bReadOnly = FALSE;
HKEY hSubKey;
//
// Check for a roaming profile security/read only preference
//
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, WINLOGON_KEY, 0, KEY_READ,
&hSubKey) == ERROR_SUCCESS) {
dwSize = sizeof(bReadOnly);
RegQueryValueEx(hSubKey, READONLY_RUP, NULL, NULL,
(LPBYTE) &bReadOnly, &dwSize);
RegCloseKey(hSubKey);
}
//
// Check for a roaming profile security/read only policy
//
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, SYSTEM_POLICIES_KEY, 0, KEY_READ,
&hSubKey) == ERROR_SUCCESS) {
dwSize = sizeof(bReadOnly);
RegQueryValueEx(hSubKey, READONLY_RUP, NULL, NULL,
(LPBYTE) &bReadOnly, &dwSize);
RegCloseKey(hSubKey);
}
if (bReadOnly) {
dwProfileType = USERINFO_READONLY;
}
}
}
switch (dwProfileStatus) {
case USERINFO_MANDATORY:
iStatusID = IDS_UP_MANDATORY;
break;
case USERINFO_BACKUP:
iStatusID = IDS_UP_BACKUP;
break;
case USERINFO_TEMP:
iStatusID = IDS_UP_TEMP;
break;
case USERINFO_FLOATING:
iStatusID = IDS_UP_FLOATING;
break;
default:
iStatusID = IDS_UP_LOCAL;
break;
}
switch (dwProfileType) {
case USERINFO_MANDATORY:
iTypeID = IDS_UP_MANDATORY;
break;
case USERINFO_READONLY:
iTypeID = IDS_UP_READONLY;
break;
case USERINFO_FLOATING:
iTypeID = IDS_UP_FLOATING;
break;
default:
iTypeID = IDS_UP_LOCAL;
break;
}
//
// Get the friendly display name
//
Error = RegQueryValueEx (hkeyUser, TEXT("Sid"), NULL, NULL, NULL, &dwSize);
if (Error != ERROR_SUCCESS) {
RegCloseKey (hkeyUser);
return FALSE;
}
pSid = LocalAlloc (LPTR, dwSize);
if (!pSid) {
RegCloseKey (hkeyUser);
return FALSE;
}
Error = RegQueryValueEx (hkeyUser,
TEXT("Sid"),
NULL,
NULL,
(LPBYTE)pSid,
&dwSize);
if (Error != ERROR_SUCCESS) {
RegCloseKey (hkeyUser);
LocalFree (pSid);
return FALSE;
}
//
// Get the friendly names
//
szTemp[0] = TEXT('\0');
dwBufSize = ARRAYSIZE(szTemp);
dwBuf2Size = ARRAYSIZE(szTemp2);
if (!LookupAccountSid (NULL, pSid, szTemp, &dwBufSize,
szTemp2, &dwBuf2Size, &SidName)) {
//
// Unknown account
//
LoadString (hInstance, IDS_UP_ACCUNKNOWN, szBuffer, ARRAYSIZE(szBuffer));
bAccountUnknown = TRUE;
} else {
if (szTemp[0] != TEXT('\0')) {
//
// Build the display name
//
if (FAILED(StringCchPrintf(szBuffer, ARRAYSIZE(szBuffer), TEXT("%s\\%s"), szTemp2, szTemp))) {
LoadString (hInstance, IDS_UP_ACCUNKNOWN, szBuffer, ARRAYSIZE(szBuffer));
bAccountUnknown = TRUE;
}
else {
bAccountUnknown = FALSE;
}
} else {
//
// Account deleted.
//
LoadString (hInstance, IDS_UP_ACCDELETED, szBuffer, ARRAYSIZE(szBuffer));
bAccountUnknown = TRUE;
}
}
//
// Free the sid
//
LocalFree (pSid);
//
// Allocate a UserInfo structure
//
dwSize = lstrlen (lpSid) + 1;
dwBuf2Size = lstrlen (szBuffer2) + 1;
dwBufSize = lstrlen (szBuffer) + 1;
lpUserInfo = (LPUSERINFO) LocalAlloc(LPTR, (sizeof(USERINFO) +
dwSize * sizeof(TCHAR) +
dwBuf2Size * sizeof(TCHAR) +
dwBufSize * sizeof(TCHAR)));
if (!lpUserInfo) {
RegCloseKey (hkeyUser);
return FALSE;
}
lpUserInfo->dwFlags = (bCentralAvailable ? USERINFO_FLAG_CENTRAL_AVAILABLE : 0);
lpUserInfo->dwFlags |= (bAccountUnknown ? USERINFO_FLAG_ACCOUNT_UNKNOWN : 0);
lpUserInfo->lpSid = (LPTSTR)((LPBYTE)lpUserInfo + sizeof(USERINFO));
lpUserInfo->lpProfile = (LPTSTR) (lpUserInfo->lpSid + dwSize);
lpUserInfo->lpUserName = (LPTSTR) (lpUserInfo->lpProfile + dwBuf2Size);
if (FAILED(StringCchCopy(lpUserInfo->lpSid, dwSize, lpSid))) {
ASSERT (FALSE);
}
if (FAILED(StringCchCopy(lpUserInfo->lpProfile, dwBuf2Size, szBuffer2))) {
ASSERT (FALSE);
}
lpUserInfo->dwProfileType = dwProfileType;
lpUserInfo->dwProfileStatus = dwProfileStatus;
if (FAILED(StringCchCopy(lpUserInfo->lpUserName, dwBufSize, szBuffer))) {
ASSERT (FALSE);
}
//
// Add the item to the listview
//
hwndTemp = GetDlgItem (hDlg, IDC_UP_LISTVIEW);
item.mask = LVIF_TEXT | LVIF_PARAM;
item.iItem = 0;
item.iSubItem = 0;
item.pszText = szBuffer;
item.lParam = (LPARAM) lpUserInfo;
iItem = ListView_InsertItem (hwndTemp, &item);
//
// Add the profile size
//
item.mask = LVIF_TEXT;
item.iItem = iItem;
item.iSubItem = 1;
item.pszText = szProfileSize;
SendMessage (hwndTemp, LVM_SETITEMTEXT, iItem, (LPARAM) &item);
//
// Add the profile type
//
LoadString (hInstance, iTypeID, szTemp, ARRAYSIZE(szTemp));
item.mask = LVIF_TEXT;
item.iItem = iItem;
item.iSubItem = 2;
item.pszText = szTemp;
SendMessage (hwndTemp, LVM_SETITEMTEXT, iItem, (LPARAM) &item);
//
// Add the profile status
//
LoadString (hInstance, iStatusID, szTemp, ARRAYSIZE(szTemp));
item.mask = LVIF_TEXT;
item.iItem = iItem;
item.iSubItem = 3;
item.pszText = szTemp;
SendMessage (hwndTemp, LVM_SETITEMTEXT, iItem, (LPARAM) &item);
if (bGotTime)
{
//
// Convert it to local time
//
if (!FileTimeToLocalFileTime(&fd.ftLastAccessTime, &ftLocal)) {
ftLocal = fd.ftLastAccessTime;
}
//
// Add the time/date stamp
//
FileTimeToSystemTime (&ftLocal, &systime);
GetDateFormat (LOCALE_USER_DEFAULT, DATE_SHORTDATE, &systime,
NULL, szBuffer, MAX_PATH);
}
item.mask = LVIF_TEXT;
item.iItem = iItem;
item.iSubItem = 4;
item.pszText = bGotTime ? szBuffer : TEXT("");
SendMessage (hwndTemp, LVM_SETITEMTEXT, iItem, (LPARAM) &item);
RegCloseKey (hkeyUser);
return TRUE;
}
//*************************************************************
//
// FillListView()
//
// Purpose: Fills the listview with either all the profiles
// or just the current user profile depending if
// the user has Admin privilages
//
// Parameters: hDlg - Dialog box handle
// bAdmin - User an admin
//
//
// Return: TRUE if successful
// FALSE if an error occurs
//
// Comments:
//
// History: Date Author Comment
// 1/26/96 ericflo Created
//
//*************************************************************
BOOL FillListView (HWND hDlg, BOOL bAdmin)
{
LV_ITEM item;
BOOL bRetVal = FALSE;
LPTSTR lpUserSid;
lpUserSid = GetSidString();
//
// If the current user has admin privilages, then
// he/she can see all the profiles on this machine,
// otherwise the user only gets their profile.
//
if (bAdmin) {
DWORD SubKeyIndex = 0;
TCHAR SubKeyName[100];
DWORD cchSubKeySize;
HKEY hkeyProfiles;
LONG Error;
//
// Open the profile list
//
Error = RegOpenKeyEx(HKEY_LOCAL_MACHINE,
PROFILE_MAPPING,
0,
KEY_READ,
&hkeyProfiles);
if (Error != ERROR_SUCCESS) {
if (lpUserSid)
DeleteSidString (lpUserSid);
return FALSE;
}
cchSubKeySize = ARRAYSIZE(SubKeyName);
while (TRUE) {
//
// Get the next sub-key name
//
Error = RegEnumKey(hkeyProfiles, SubKeyIndex, SubKeyName, cchSubKeySize);
if (Error != ERROR_SUCCESS) {
if (Error == ERROR_NO_MORE_ITEMS) {
//
// Successful end of enumeration
//
Error = ERROR_SUCCESS;
}
break;
}
if ((lpUserSid) && (lstrcmp(SubKeyName, lpUserSid) == 0))
AddUser (hDlg, SubKeyName, PRF_USERSID);
else
AddUser (hDlg, SubKeyName, 0);
//
// Go enumerate the next sub-key
//
SubKeyIndex ++;
}
//
// Close the registry
//
RegCloseKey (hkeyProfiles);
bRetVal = ((Error == ERROR_SUCCESS) ? TRUE : FALSE);
} else {
//
// The current user doesn't have admin privilages
//
if (lpUserSid) {
AddUser (hDlg, lpUserSid, PRF_USERSID);
bRetVal = TRUE;
}
}
if (bRetVal) {
//
// Select the first item
//
item.mask = LVIF_STATE;
item.iItem = 0;
item.iSubItem = 0;
item.state = LVIS_SELECTED | LVIS_FOCUSED;
item.stateMask = LVIS_SELECTED | LVIS_FOCUSED;
SendDlgItemMessage (hDlg, IDC_UP_LISTVIEW,
LVM_SETITEMSTATE, 0, (LPARAM) &item);
}
if (lpUserSid)
DeleteSidString (lpUserSid);
return (bRetVal);
}
//*************************************************************
//
// CheckSemicolon()
//
// Purpose: Checks for an ending semicolon and adds one if
// it is missing.
//
// Parameters: lpDir - directory
// cchSize - buffer size in character
//
// Return: TRUE - success
// FALSE - Insufficient buffer space
//
// Comments:
//
// History: Date Author Comment
// 6/19/95 ericlfo Created
//
//*************************************************************
BOOL CheckSemicolon (LPTSTR lpDir, UINT cchBuffer)
{
UINT cchDir = lstrlen(lpDir);
LPTSTR lpEnd;
lpEnd = lpDir + cchDir;
if (*(lpEnd - 1) != TEXT(';')) {
if (cchDir + 1 >= cchBuffer) {
return FALSE; // No space to put ;, should never happen
}
*lpEnd = TEXT(';');
lpEnd++;
*lpEnd = TEXT('\0');
}
return TRUE;
}
//*************************************************************
//
// ReadExclusionList()
//
// Purpose: Reads the exclusionlist for the user from the
// hkcu part of the registry
//
// Parameters: hKeyUser - Key to user's hive, NULL if HKCU
// szExclusionList - Buffer for exclusionList to be read into
// cchExcludeList - Size of the buffer
//
// Return: FALSE on error
//
//*************************************************************
BOOL ReadExclusionList(HKEY hKeyUser, LPTSTR szExcludeList, DWORD cchExcludeList)
{
TCHAR szExcludeList2[MAX_PATH];
TCHAR szExcludeList1[MAX_PATH];
HKEY hKey = NULL;
DWORD dwSize;
HRESULT hr;
//
// Check for a list of directories to exclude both user preferences
// and user policy
//
szExcludeList1[0] = TEXT('\0');
if (RegOpenKeyEx (hKeyUser ? hKeyUser : HKEY_CURRENT_USER,
WINLOGON_KEY, 0,
KEY_READ, &hKey) == ERROR_SUCCESS) {
dwSize = sizeof(szExcludeList1);
if (RegQueryValueEx (hKey,
TEXT("ExcludeProfileDirs"),
NULL,
NULL,
(LPBYTE) szExcludeList1,
&dwSize) != ERROR_SUCCESS) {
// ignore user exclusion list
szExcludeList1[0] = TEXT('\0');
}
RegCloseKey (hKey);
}
szExcludeList2[0] = TEXT('\0');
if (RegOpenKeyEx (HKEY_CURRENT_USER,
SYSTEM_POLICIES_KEY,
0, KEY_READ, &hKey) == ERROR_SUCCESS) {
dwSize = sizeof(szExcludeList2);
if (RegQueryValueEx (hKey,
TEXT("ExcludeProfileDirs"),
NULL,
NULL,
(LPBYTE) szExcludeList2,
&dwSize) != ERROR_SUCCESS) {
// ignore policy exclusion list
szExcludeList2[0] = TEXT('\0');
}
RegCloseKey (hKey);
}
//
// Merge the user preferences and policy together
//
szExcludeList[0] = TEXT('\0');
if (szExcludeList1[0] != TEXT('\0')) {
hr = StringCchCopy(szExcludeList, cchExcludeList, szExcludeList1);
if (FAILED(hr)) {
szExcludeList[0] = TEXT('\0');
return FALSE;
}
if (!CheckSemicolon(szExcludeList, cchExcludeList)) {
szExcludeList[0] = TEXT('\0');
return FALSE;
}
}
if (szExcludeList2[0] != TEXT('\0')) {
hr = StringCchCat(szExcludeList, cchExcludeList, szExcludeList2);
if (FAILED(hr)) {
szExcludeList[0] = TEXT('\0');
return FALSE;
}
}
return TRUE;
}
//*************************************************************
//
// ReadExclusionListFromIniFile()
//
// Purpose: Reads the exclusionlist for the user from the
// ntuser.ini from the local profile
//
// Parameters:
// (in) lpSourceDir - Profile Directory
// szExclusionList - Buffer for exclusionList to be read into
// cchExcludeList - Size of the buffer
//
// Return: FALSE on error
//
//
//*************************************************************
BOOL ReadExclusionListFromIniFile(LPCTSTR lpSourceDir, LPTSTR szExcludeList, DWORD cchExcludeList)
{
TCHAR szNTUserIni[MAX_PATH];
LPTSTR lpEnd;
//
// Get the exclusion list from the cache
//
szExcludeList[0] = TEXT('\0');
if (FAILED(StringCchCopy(szNTUserIni, ARRAYSIZE(szNTUserIni), lpSourceDir))) {
return FALSE;
}
lpEnd = CheckSlashEx(szNTUserIni, ARRAYSIZE(szNTUserIni), NULL);
if (!lpEnd)
{
return FALSE;
}
if (FAILED(StringCchCat(szNTUserIni, ARRAYSIZE(szNTUserIni), c_szNTUserIni))) {
return FALSE;
}
GetPrivateProfileString (PROFILE_GENERAL_SECTION,
PROFILE_EXCLUSION_LIST,
TEXT(""),
szExcludeList,
cchExcludeList,
szNTUserIni);
return TRUE;
}
//*************************************************************
//
// ConvertExclusionList()
//
// Purpose: Converts the semi-colon profile relative exclusion
// list to fully qualified null terminated exclusion
// list
//
// Parameters: lpSourceDir - Profile root directory
// lpExclusionList - List of directories to exclude
//
// Return: List if successful
// NULL if an error occurs
//
//*************************************************************
LPTSTR ConvertExclusionList (LPCTSTR lpSourceDir, LPCTSTR lpExclusionList)
{
LPTSTR lpExcludeList = NULL, lpInsert, lpEnd, lpTempList;
LPCTSTR lpTemp, lpDir;
TCHAR szTemp[MAX_PATH];
DWORD dwSize = 2; // double null terminator
DWORD dwStrLen;
UINT cchRemaining;
HRESULT hr;
//
// Setup a temp buffer to work with
//
hr = StringCchCopy(szTemp, ARRAYSIZE(szTemp), lpSourceDir);
if (FAILED(hr)) {
return NULL;
}
lpEnd = CheckSlashEx(szTemp, ARRAYSIZE(szTemp), &cchRemaining);
if (!lpEnd)
{
return NULL;
}
//
// Loop through the list
//
lpTemp = lpDir = lpExclusionList;
while (*lpTemp) {
//
// Look for the semicolon separator
//
while (*lpTemp && ((*lpTemp) != TEXT(';'))) {
lpTemp++;
}
//
// Remove any leading spaces
//
while (*lpDir == TEXT(' ')) {
lpDir++;
}
//
// Check whether the entry is empty
//
if (lpDir == lpTemp) {
// If we are at the end of the exclusion list, we're done
if (!*lpTemp) {
goto Exit;
}
//
// Prep for the next entry
//
lpTemp++;
lpDir = lpTemp;
continue;
}
//
// Put the directory name on the temp buffer
//
*lpEnd = TEXT('\0');
hr = StringCchCatN(lpEnd, cchRemaining, lpDir, (int)(lpTemp - lpDir));
if (FAILED(hr)) {
LocalFree (lpExcludeList);
lpExcludeList = NULL;
goto Exit;
}
//
// Add the string to the exclusion list
//
if (lpExcludeList) {
dwStrLen = lstrlen (szTemp) + 1;
dwSize += dwStrLen;
lpTempList = (LPTSTR) LocalReAlloc (lpExcludeList, dwSize * sizeof(TCHAR),
LMEM_MOVEABLE | LMEM_ZEROINIT);
if (!lpTempList) {
LocalFree (lpExcludeList);
lpExcludeList = NULL;
goto Exit;
}
lpExcludeList = lpTempList;
lpInsert = lpExcludeList + dwSize - dwStrLen - 1;
hr = StringCchCopy(lpInsert, dwStrLen, szTemp);
if (FAILED(hr)) {
LocalFree (lpExcludeList);
lpExcludeList = NULL;
goto Exit;
}
} else {
dwSize += lstrlen (szTemp);
lpExcludeList = (LPTSTR) LocalAlloc (LPTR, dwSize * sizeof(TCHAR));
if (!lpExcludeList) {
goto Exit;
}
hr = StringCchCopy(lpExcludeList, dwSize, szTemp);
if (FAILED(hr)) {
LocalFree (lpExcludeList);
lpExcludeList = NULL;
goto Exit;
}
}
//
// If we are at the end of the exclusion list, we're done
//
if (!(*lpTemp)) {
goto Exit;
}
//
// Prep for the next entry
//
lpTemp++;
lpDir = lpTemp;
}
Exit:
return lpExcludeList;
}
//*************************************************************
//
// RecurseDirectory()
//
// Purpose: Recurses through the subdirectories counting the size.
//
// Parameters: lpDir - Directory
// cchBuffer - Buffer size in char
// lpExcludeList - Null-termed list of dirs to be skipped (optional)
//
// Return: TRUE if successful
// FALSE if an error occurs
//
// Comments:
//
// History: Date Author Comment
// 1/30/96 ericflo Created
//
// Notes:
// The buffer size expected is MAX_PATH+4 for some internal processing
// We should fix this to be better post Win 2K.
//*************************************************************
BOOL RecurseDirectory (LPTSTR lpDir, UINT cchBuffer, LPTSTR lpExcludeList)
{
HANDLE hFile = INVALID_HANDLE_VALUE;
WIN32_FIND_DATA fd;
LPTSTR lpEnd, lpTemp;
BOOL bResult = TRUE;
BOOL bSkip;
UINT cchRemaining;
HRESULT hr;
//
// Setup the ending pointer
//
lpEnd = CheckSlashEx(lpDir, cchBuffer, &cchRemaining);
if (!lpEnd)
{
bResult = FALSE;
goto RecurseDir_Exit;
}
//
// Append *.* to the source directory
//
hr = StringCchCopy(lpEnd, cchRemaining, TEXT("*.*"));
if (FAILED(hr)) {
bResult = FALSE;
goto RecurseDir_Exit;
}
//
// Search through the source directory
//
hFile = FindFirstFile(lpDir, &fd);
if (hFile == INVALID_HANDLE_VALUE) {
if ( (GetLastError() == ERROR_FILE_NOT_FOUND) ||
(GetLastError() == ERROR_PATH_NOT_FOUND) ) {
//
// bResult is already initialized to TRUE, so
// just fall through.
//
} else {
bResult = FALSE;
}
goto RecurseDir_Exit;
}
do {
//
// Append the file / directory name to the working buffer
//
// skip the file if the path > MAX_PATH
if ((UINT)(1+lstrlen(fd.cFileName)+lstrlen(lpDir)+lstrlen(TEXT("\\*.*"))) >= cchBuffer) {
continue;
}
hr = StringCchCopy(lpEnd, cchRemaining, fd.cFileName);
if (FAILED(hr)) {
bResult = FALSE;
goto RecurseDir_Exit;
}
if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
//
// Check for "." and ".."
//
if (!lstrcmpi(fd.cFileName, TEXT("."))) {
continue;
}
if (!lstrcmpi(fd.cFileName, TEXT(".."))) {
continue;
}
//
// Check for reparse point
//
if (fd.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
continue;
}
//
// Check if this directory should be excluded
//
if (lpExcludeList) {
bSkip = FALSE;
lpTemp = lpExcludeList;
while (*lpTemp) {
if (lstrcmpi (lpTemp, lpDir) == 0) {
bSkip = TRUE;
break;
}
lpTemp += lstrlen (lpTemp) + 1;
}
if (bSkip)
continue;
}
//
// Found a directory.
//
// 1) Change into that subdirectory on the source drive.
// 2) Recurse down that tree.
// 3) Back up one level.
//
//
// Recurse the subdirectory
//
if (!RecurseDirectory(lpDir, cchBuffer, lpExcludeList)) {
bResult = FALSE;
goto RecurseDir_Exit;
}
} else {
//
// Found a file, add the filesize
//
g_dwProfileSize += fd.nFileSizeLow;
}
//
// Find the next entry
//
} while (FindNextFile(hFile, &fd));
RecurseDir_Exit:
//
// Remove the file / directory name appended above
//
if (lpEnd)
{
*lpEnd = TEXT('\0');
}
//
// Close the search handle
//
if (hFile != INVALID_HANDLE_VALUE) {
FindClose(hFile);
}
return bResult;
}
//*************************************************************
//
// UPCleanUp()
//
// Purpose: Free's resources for this dialog box
//
// Parameters: hDlg - Dialog box handle
//
// Return: void
//
// Comments:
//
// History: Date Author Comment
// 1/31/96 ericflo Created
//
//*************************************************************
VOID UPCleanUp (HWND hDlg)
{
int i, n;
HWND hwndTemp;
LPUSERINFO lpUserInfo;
LV_ITEM item;
//
// Free memory used for the listview items
//
hwndTemp = GetDlgItem (hDlg, IDC_UP_LISTVIEW);
n = (int)SendMessage (hwndTemp, LVM_GETITEMCOUNT, 0, 0L);
item.mask = LVIF_PARAM;
item.iSubItem = 0;
for (i = 0; i < n; i++) {
item.iItem = i;
if (SendMessage (hwndTemp, LVM_GETITEM, 0, (LPARAM) &item)) {
lpUserInfo = (LPUSERINFO) item.lParam;
} else {
lpUserInfo = NULL;
}
LocalFree (lpUserInfo);
}
}
//*************************************************************
//
// UPSave()
//
// Purpose: Saves the settings
//
// Parameters: hDlg - Dialog box handle
//
// Return: void
//
// Comments:
//
// History: Date Author Comment
// 1/31/96 ericflo Created
//
//*************************************************************
VOID UPSave (HWND hDlg)
{
int i, n;
HWND hwndTemp;
LPUSERINFO lpUserInfo;
LV_ITEM item;
TCHAR szBuffer[MAX_PATH];
HKEY hkeyUser;
LONG Error;
//
// Save type info
//
hwndTemp = GetDlgItem (hDlg, IDC_UP_LISTVIEW);
n = (int)SendMessage (hwndTemp, LVM_GETITEMCOUNT, 0, 0L);
item.mask = LVIF_PARAM;
item.iSubItem = 0;
for (i = 0; i < n; i++) {
item.iItem = i;
if (SendMessage (hwndTemp, LVM_GETITEM, 0, (LPARAM) &item)) {
lpUserInfo = (LPUSERINFO) item.lParam;
} else {
lpUserInfo = NULL;
}
if (lpUserInfo) {
if (lpUserInfo->dwFlags & USERINFO_FLAG_DIRTY) {
lpUserInfo->dwFlags &= ~USERINFO_FLAG_DIRTY;
if (FAILED(StringCchPrintf(szBuffer,
ARRAYSIZE(szBuffer),
TEXT("%s\\%s\\Preference"),
PROFILE_MAPPING,
lpUserInfo->lpSid))) {
continue;
}
Error = RegOpenKeyEx(HKEY_LOCAL_MACHINE,
szBuffer,
0,
KEY_SET_VALUE,
&hkeyUser);
if (Error != ERROR_SUCCESS) {
continue;
}
RegSetValueEx (hkeyUser,
TEXT("UserPreference"),
0,
REG_DWORD,
(LPBYTE) &lpUserInfo->dwProfileType,
sizeof(DWORD));
RegCloseKey(hkeyUser);
}
}
}
}
//*************************************************************
//
// IsProfileInUse()
//
// Purpose: Determines if the given profile is currently in use
//
// Parameters: lpSid - Sid (text) to test
//
// Return: TRUE if in use
// FALSE if not
//
// Comments:
//
// History: Date Author Comment
// 2/7/96 ericflo Created
//
//*************************************************************
BOOL IsProfileInUse (LPTSTR lpSid)
{
LONG lResult;
HKEY hkeyProfile;
lResult = RegOpenKeyEx (HKEY_USERS, lpSid, 0, KEY_READ, &hkeyProfile);
if (lResult == ERROR_SUCCESS) {
RegCloseKey (hkeyProfile);
return TRUE;
}
return FALSE;
}
void UPDoItemChanged(HWND hDlg, int idCtl)
{
int selection;
HWND hwndTemp;
LPUSERINFO lpUserInfo;
LV_ITEM item;
hwndTemp = GetDlgItem (hDlg, idCtl);
selection = GetSelectedItem (hwndTemp);
if (selection != -1)
{
item.mask = LVIF_PARAM;
item.iItem = selection;
item.iSubItem = 0;
if (SendMessage (hwndTemp, LVM_GETITEM, 0, (LPARAM) &item)) {
lpUserInfo = (LPUSERINFO) item.lParam;
} else {
lpUserInfo = NULL;
}
if (lpUserInfo) {
//
// Set the "Delete" button state
//
if (IsProfileInUse(lpUserInfo->lpSid)) {
EnableWindow (GetDlgItem (hDlg, IDC_UP_DELETE), FALSE);
} else {
EnableWindow (GetDlgItem (hDlg, IDC_UP_DELETE), TRUE);
}
//
// Set the "Change Type" button state
//
OSVERSIONINFOEXW version;
version.dwOSVersionInfoSize = sizeof(version);
if (GetVersionEx((LPOSVERSIONINFO)&version) &&
(version.wSuiteMask & VER_SUITE_PERSONAL)) {
ShowWindow(GetDlgItem (hDlg, IDC_UP_TYPE), SW_HIDE);
}
else if((lpUserInfo->dwProfileType == USERINFO_MANDATORY) ||
(lpUserInfo->dwProfileType == USERINFO_BACKUP)) {
EnableWindow (GetDlgItem (hDlg, IDC_UP_TYPE), FALSE);
} else {
EnableWindow (GetDlgItem (hDlg, IDC_UP_TYPE), TRUE);
}
//
// Set the "Copy To" button state
//
if ((lpUserInfo->dwFlags & USERINFO_FLAG_ACCOUNT_UNKNOWN) ||
IsProfileInUse(lpUserInfo->lpSid)) {
EnableWindow (GetDlgItem (hDlg, IDC_UP_COPY), FALSE);
} else {
EnableWindow (GetDlgItem (hDlg, IDC_UP_COPY), TRUE);
}
}
}
else {
//
// If nothing is selected set all buttons to inactive
//
EnableWindow (GetDlgItem (hDlg, IDC_UP_DELETE), FALSE);
EnableWindow (GetDlgItem (hDlg, IDC_UP_TYPE), FALSE);
EnableWindow (GetDlgItem (hDlg, IDC_UP_COPY), FALSE);
}
}
//*************************************************************
void
UPOnLink(WPARAM wParam)
{
switch(wParam)
{
case IDC_UP_UPLINK:
{
SHELLEXECUTEINFO execinfo = {0};
execinfo.cbSize = sizeof(execinfo);
execinfo.nShow = SW_SHOW;
execinfo.lpVerb = TEXT("open");
execinfo.lpFile = TEXT("control");
execinfo.lpParameters = TEXT("userpasswords");
ShellExecuteEx(&execinfo);
}
break;
default:
break;
}
}
//*************************************************************
//
// UPDeleteProfile()
//
// Purpose: Deletes a user's profile
//
// Parameters: hDlg - Dialog box handle
//
// Return:
//
// Comments:
//
// History: Date Author Comment
// 2/8/96 ericflo Created
//
//*************************************************************
void UPDeleteProfile(HWND hDlg)
{
int selection;
HWND hwndTemp;
LPUSERINFO lpUserInfo;
LV_ITEM item;
TCHAR szName[100];
TCHAR szBuffer1[100];
TCHAR szBuffer2[200];
HCURSOR hOldCursor;
//
// Get the selected profile
//
hwndTemp = GetDlgItem (hDlg, IDC_UP_LISTVIEW);
selection = GetSelectedItem (hwndTemp);
if (selection == -1) {
return;
}
item.mask = LVIF_PARAM;
item.iItem = selection;
item.iSubItem = 0;
if (SendMessage (hwndTemp, LVM_GETITEM, 0, (LPARAM) &item)) {
lpUserInfo = (LPUSERINFO) item.lParam;
} else {
lpUserInfo = NULL;
}
if (!lpUserInfo) {
return;
}
//
// Confirm that the user really wants to delete the profile
//
szBuffer1[0] = TEXT('\0');
ListView_GetItemText (hwndTemp, selection, 0, szName, ARRAYSIZE(szName));
LoadString (hInstance, IDS_UP_CONFIRM, szBuffer1, ARRAYSIZE(szBuffer1));
if (FAILED(StringCchPrintf(szBuffer2, ARRAYSIZE(szBuffer2), szBuffer1, szName))) {
return;
}
LoadString (hInstance, IDS_UP_CONFIRMTITLE, szBuffer1, ARRAYSIZE(szBuffer1));
if (MessageBox (hDlg, szBuffer2, szBuffer1,
MB_ICONQUESTION | MB_DEFBUTTON2| MB_YESNO) == IDNO) {
return;
}
//
// Delete the profile and remove the entry from the listview
//
hOldCursor = SetCursor (LoadCursor(NULL, IDC_WAIT));
if (!DeleteProfile (lpUserInfo->lpSid, NULL, NULL)) {
TCHAR szMsg[MAX_PATH];
szMsg[0] = TEXT('\0');
LoadString (hInstance, IDS_UP_DELETE_ERROR, szMsg, ARRAYSIZE(szMsg));
UPDisplayErrorMessage(hDlg, GetLastError(), szMsg);
}
if (ListView_DeleteItem(hwndTemp, selection)) {
LocalFree (lpUserInfo);
}
//
// Select another item
//
if (selection > 0) {
selection--;
}
item.mask = LVIF_STATE;
item.iItem = selection;
item.iSubItem = 0;
item.state = LVIS_SELECTED | LVIS_FOCUSED;
item.stateMask = LVIS_SELECTED | LVIS_FOCUSED;
SendDlgItemMessage (hDlg, IDC_UP_LISTVIEW,
LVM_SETITEMSTATE, selection, (LPARAM) &item);
SetCursor(hOldCursor);
}
//*************************************************************
//
// UPChangeType()
//
// Purpose: Display the "Change Type" dialog box
//
// Parameters: hDlg - Dialog box handle
//
// Return: void
//
// Comments:
//
// History: Date Author Comment
// 2/09/96 ericflo Created
//
//*************************************************************
void UPChangeType(HWND hDlg)
{
int selection, iTypeID;
HWND hwndTemp;
LPUSERINFO lpUserInfo;
LV_ITEM item;
TCHAR szType[100];
//
// Get the selected profile
//
hwndTemp = GetDlgItem (hDlg, IDC_UP_LISTVIEW);
selection = GetSelectedItem (hwndTemp);
if (selection == -1) {
return;
}
item.mask = LVIF_PARAM;
item.iItem = selection;
item.iSubItem = 0;
if (SendMessage (hwndTemp, LVM_GETITEM, 0, (LPARAM) &item)) {
lpUserInfo = (LPUSERINFO) item.lParam;
} else {
lpUserInfo = NULL;
}
if (!lpUserInfo) {
return;
}
// if profile type is mandatory then don't display change type dialog box.
// Although change type button is disabled for mandatory profile this function
// can be still executed with double clicking on profile item.
if (lpUserInfo->dwProfileType == USERINFO_MANDATORY ||
lpUserInfo->dwProfileType == USERINFO_BACKUP) {
return;
}
//
// Display the Change Type dialog
//
if (!DialogBoxParam (hInstance, MAKEINTRESOURCE(IDD_UP_TYPE), hDlg,
ChangeTypeDlgProc, (LPARAM)lpUserInfo)) {
return;
}
//
// Activate the Apply button
//
PropSheet_Changed(GetParent(hDlg), hDlg);
//
// Mark this item as 'dirty' so it will be saved
//
lpUserInfo->dwFlags |= USERINFO_FLAG_DIRTY;
//
// Fix the 'Type' field in the display
//
switch (lpUserInfo->dwProfileType) {
case USERINFO_MANDATORY:
iTypeID = IDS_UP_MANDATORY;
break;
case USERINFO_READONLY:
iTypeID = IDS_UP_READONLY;
break;
case USERINFO_FLOATING:
iTypeID = IDS_UP_FLOATING;
break;
default:
iTypeID = IDS_UP_LOCAL;
break;
}
LoadString (hInstance, iTypeID, szType, ARRAYSIZE(szType));
item.mask = LVIF_TEXT;
item.iItem = selection;
item.iSubItem = 2;
item.pszText = szType;
SendMessage (hwndTemp, LVM_SETITEMTEXT, selection, (LPARAM) &item);
}
//*************************************************************
//
// ChangeTypeDlgProc()
//
// Purpose: Dialog box procedure for changing the profile type
//
// Parameters: hDlg - handle to the dialog box
// uMsg - window message
// wParam - wParam
// lParam - lParam
//
// Return: TRUE if message was processed
// FALSE if not
//
// Comments:
//
// History: Date Author Comment
// 2/9/96 ericflo Created
//
//*************************************************************
INT_PTR APIENTRY ChangeTypeDlgProc (HWND hDlg, UINT uMsg,
WPARAM wParam, LPARAM lParam)
{
LPUSERINFO lpUserInfo;
HKEY hKeyPolicy;
switch (uMsg) {
case WM_INITDIALOG:
lpUserInfo = (LPUSERINFO) lParam;
if (!lpUserInfo) {
EndDialog(hDlg, FALSE);
break;
}
else {
TCHAR szMsg[MAX_PATH], szMsg1[MAX_PATH];
szMsg[0] = TEXT('\0');
LoadString (hInstance, IDS_UP_CHANGETYPEMSG, szMsg, ARRAYSIZE(szMsg));
StringCchPrintf (szMsg1, ARRAYSIZE(szMsg1), szMsg, lpUserInfo->lpUserName);
SetDlgItemText (hDlg, IDC_UPTYPE_GROUP, szMsg1);
}
SetWindowLongPtr (hDlg, GWLP_USERDATA, (LPARAM) lpUserInfo);
if (lpUserInfo->dwFlags & USERINFO_FLAG_CENTRAL_AVAILABLE) {
if (lpUserInfo->dwProfileType == USERINFO_LOCAL) {
CheckRadioButton (hDlg, IDC_UPTYPE_LOCAL, IDC_UPTYPE_FLOAT,
IDC_UPTYPE_LOCAL);
if (lpUserInfo->dwProfileStatus == USERINFO_TEMP) {
EnableWindow (GetDlgItem(hDlg, IDC_UPTYPE_FLOAT), FALSE);
EnableWindow (GetDlgItem(hDlg, IDOK), FALSE);
SetDefButton(hDlg, IDCANCEL);
}
else if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,
SYSTEM_POLICIES_KEY,
0, KEY_READ,
&hKeyPolicy) == ERROR_SUCCESS) {
DWORD dwTmpVal, dwSize;
dwSize = sizeof(dwTmpVal);
RegQueryValueEx(hKeyPolicy,
PROFILE_LOCALONLY,
NULL, NULL,
(LPBYTE) &dwTmpVal,
&dwSize);
RegCloseKey (hKeyPolicy);
if (dwTmpVal == 1) {
EnableWindow (GetDlgItem(hDlg, IDC_UPTYPE_FLOAT), FALSE);
EnableWindow (GetDlgItem(hDlg, IDOK), FALSE);
SetDefButton(hDlg, IDCANCEL);
}
}
}
else if (lpUserInfo->dwProfileStatus == USERINFO_TEMP) {
CheckRadioButton (hDlg, IDC_UPTYPE_LOCAL, IDC_UPTYPE_FLOAT,
IDC_UPTYPE_FLOAT);
EnableWindow (GetDlgItem(hDlg, IDC_UPTYPE_LOCAL), FALSE);
EnableWindow (GetDlgItem(hDlg, IDOK), FALSE);
SetDefButton(hDlg, IDCANCEL);
}
else { // ProfileType is USERINFO_FLOATING
CheckRadioButton (hDlg, IDC_UPTYPE_LOCAL, IDC_UPTYPE_FLOAT,
IDC_UPTYPE_FLOAT);
}
} else {
CheckRadioButton (hDlg, IDC_UPTYPE_LOCAL, IDC_UPTYPE_FLOAT,
IDC_UPTYPE_LOCAL);
EnableWindow (GetDlgItem(hDlg, IDC_UPTYPE_FLOAT), FALSE);
EnableWindow (GetDlgItem(hDlg, IDOK), FALSE);
SetDefButton(hDlg, IDCANCEL);
}
return TRUE;
case WM_COMMAND:
switch (LOWORD(wParam)) {
case IDOK:
lpUserInfo = (LPUSERINFO) GetWindowLongPtr(hDlg, GWLP_USERDATA);
if (!lpUserInfo) {
EndDialog (hDlg, FALSE);
break;
}
//
// Determine what the user wants
//
if (IsDlgButtonChecked(hDlg, IDC_UPTYPE_LOCAL)) {
lpUserInfo->dwProfileType = USERINFO_LOCAL;
} else {
BOOL bReadOnly = FALSE;
HKEY hSubKey;
DWORD dwSize;
//
// Check for a roaming profile security/read only preference
//
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, WINLOGON_KEY, 0, KEY_READ,
&hSubKey) == ERROR_SUCCESS) {
dwSize = sizeof(bReadOnly);
RegQueryValueEx(hSubKey, READONLY_RUP, NULL, NULL,
(LPBYTE) &bReadOnly, &dwSize);
RegCloseKey(hSubKey);
}
//
// Check for a roaming profile security/read only policy
//
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, SYSTEM_POLICIES_KEY, 0, KEY_READ,
&hSubKey) == ERROR_SUCCESS) {
dwSize = sizeof(bReadOnly);
RegQueryValueEx(hSubKey, READONLY_RUP, NULL, NULL,
(LPBYTE) &bReadOnly, &dwSize);
RegCloseKey(hSubKey);
}
if (bReadOnly) {
lpUserInfo->dwProfileType = USERINFO_READONLY;
}
else {
lpUserInfo->dwProfileType = USERINFO_FLOATING;
}
}
EndDialog(hDlg, TRUE);
break;
case IDCANCEL:
EndDialog(hDlg, FALSE);
break;
default:
break;
}
break;
case WM_HELP: // F1
WinHelp((HWND)((LPHELPINFO) lParam)->hItemHandle, HELP_FILE, HELP_WM_HELP,
(DWORD_PTR) (LPSTR) aUserProfileHelpIds);
break;
case WM_CONTEXTMENU: // right mouse click
WinHelp((HWND) wParam, HELP_FILE, HELP_CONTEXTMENU,
(DWORD_PTR) (LPSTR) aUserProfileHelpIds);
return (TRUE);
}
return (FALSE);
}
//*************************************************************
//
// UPInitCopyDlg()
//
// Purpose: Initializes the copy profile dialog
//
// Parameters: hDlg - Dialog box handle
// lParam - lParam (lpUserInfo)
//
// Return: TRUE if successful
// FALSE if an error occurs
//
// Comments:
//
// History: Date Author Comment
// 2/26/96 ericflo Created
//
//*************************************************************
BOOL UPInitCopyDlg (HWND hDlg, LPARAM lParam)
{
LPUSERINFO lpUserInfo;
LPUPCOPYINFO lpUPCopyInfo;
HKEY hKey;
LONG lResult;
TCHAR szBuffer[MAX_PATH];
TCHAR szTemp[100];
TCHAR szTemp2[100];
DWORD dwTempSize = 100, dwTemp2Size = 100;
PSID pSid;
DWORD dwSize;
SID_NAME_USE SidName;
HRESULT hr;
lpUserInfo = (LPUSERINFO) lParam;
if (!lpUserInfo) {
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
return FALSE;
}
//
// Create a CopyInfo structure
//
lpUPCopyInfo = (LPUPCOPYINFO) LocalAlloc(LPTR, sizeof(UPCOPYINFO));
if (!lpUPCopyInfo) {
return FALSE;
}
lpUPCopyInfo->dwFlags = 0;
lpUPCopyInfo->lpUserInfo = lpUserInfo;
lpUPCopyInfo->bDefaultSecurity = TRUE;
//
// Get the user's sid
//
hr = StringCchPrintf(szBuffer, ARRAYSIZE(szBuffer), TEXT("%s\\%s"), PROFILE_MAPPING, lpUserInfo->lpSid);
if (FAILED(hr)) {
LocalFree (lpUPCopyInfo);
SetLastError(HRESULT_CODE(hr));
return FALSE;
}
lResult = RegOpenKeyEx (HKEY_LOCAL_MACHINE,
szBuffer,
0,
KEY_READ,
&hKey);
if (lResult != ERROR_SUCCESS) {
LocalFree (lpUPCopyInfo);
SetLastError(lResult);
return FALSE;
}
//
// Query for the sid size
//
dwSize = 0;
lResult = RegQueryValueEx (hKey,
TEXT("Sid"),
NULL,
NULL,
NULL,
&dwSize);
if (lResult != ERROR_SUCCESS) {
RegCloseKey (hKey);
LocalFree (lpUPCopyInfo);
SetLastError(lResult);
return FALSE;
}
//
// Actually get the sid
//
pSid = LocalAlloc (LPTR, dwSize);
if (!pSid) {
RegCloseKey (hKey);
LocalFree (lpUPCopyInfo);
SetLastError(ERROR_OUTOFMEMORY);
return FALSE;
}
lResult = RegQueryValueEx (hKey,
TEXT("Sid"),
NULL,
NULL,
(LPBYTE) pSid,
&dwSize);
if (lResult != ERROR_SUCCESS) {
RegCloseKey (hKey);
LocalFree (pSid);
LocalFree (lpUPCopyInfo);
SetLastError(lResult);
return FALSE;
}
lpUPCopyInfo->pSid = pSid;
RegCloseKey (hKey);
//
// Get the friendly name
//
if (!LookupAccountSid (NULL, pSid, szTemp, &dwTempSize,
szTemp2, &dwTemp2Size, &SidName)) {
LocalFree (pSid);
LocalFree (lpUPCopyInfo);
return FALSE;
}
//
// Display nothing in the edit control till user sets it
// explicitly
//
szBuffer[0] = TEXT('\0');
SetDlgItemText (hDlg, IDC_COPY_USER, szBuffer);
//
// Save the copyinfo structure in the extra words
//
SetWindowLongPtr (hDlg, GWLP_USERDATA, (LPARAM) lpUPCopyInfo);
EnableWindow (GetDlgItem(hDlg, IDOK), FALSE);
SetDefButton(hDlg, IDCANCEL);
return TRUE;
}
//*************************************************************
//
// UPCopyProfile()
//
// Purpose: Displays the copy profile dialog box
//
// Parameters: hDlg - Dialog box handle
//
// Return: void
//
// Comments:
//
// History: Date Author Comment
// 2/13/96 ericflo Created
//
//*************************************************************
void UPCopyProfile(HWND hDlg)
{
int selection, iTypeID;
HWND hwndTemp;
LPUSERINFO lpUserInfo;
LV_ITEM item;
//
// Get the selected profile
//
hwndTemp = GetDlgItem (hDlg, IDC_UP_LISTVIEW);
selection = GetSelectedItem (hwndTemp);
if (selection == -1) {
return;
}
item.mask = LVIF_PARAM;
item.iItem = selection;
item.iSubItem = 0;
if (SendMessage (hwndTemp, LVM_GETITEM, 0, (LPARAM) &item)) {
lpUserInfo = (LPUSERINFO) item.lParam;
} else {
lpUserInfo = NULL;
}
if (!lpUserInfo) {
return;
}
//
// Display the Copy Profile dialog
//
if (!DialogBoxParam (hInstance, MAKEINTRESOURCE(IDD_UP_COPY), hDlg,
UPCopyDlgProc, (LPARAM)lpUserInfo)) {
return;
}
}
//*************************************************************
//
// UPCopyDlgProc()
//
// Purpose: Dialog box procedure for copying a profile
//
// Parameters: hDlg - handle to the dialog box
// uMsg - window message
// wParam - wParam
// lParam - lParam
//
// Return: TRUE if message was processed
// FALSE if not
//
// Comments:
//
// History: Date Author Comment
// 2/13/96 ericflo Created
//
//*************************************************************
INT_PTR APIENTRY UPCopyDlgProc (HWND hDlg, UINT uMsg,
WPARAM wParam, LPARAM lParam)
{
LPUPCOPYINFO lpUPCopyInfo;
WIN32_FILE_ATTRIBUTE_DATA fad;
switch (uMsg) {
case WM_INITDIALOG:
if (!UPInitCopyDlg(hDlg, lParam)) {
UPDisplayErrorMessage(hDlg, GetLastError(), NULL);
EndDialog(hDlg, FALSE);
}
return TRUE;
case WM_COMMAND:
switch (LOWORD(wParam)) {
case IDOK:
{
TCHAR szDir[MAX_PATH];
DWORD cchDir;
TCHAR szTemp[MAX_PATH];
HCURSOR hOldCursor;
lpUPCopyInfo = (LPUPCOPYINFO) GetWindowLongPtr(hDlg, GWLP_USERDATA);
if (!lpUPCopyInfo) {
EndDialog (hDlg, FALSE);
break;
}
if (!GetDlgItemText (hDlg, IDC_COPY_PATH, szTemp, ARRAYSIZE(szTemp))) {
break;
}
cchDir = ExpandEnvironmentStrings (szTemp, szDir, ARRAYSIZE(szDir));
if (!cchDir || cchDir > MAX_PATH) {
break;
}
//
// If the directory already exists "Warn the user"
//
if (GetFileAttributesEx (szDir, GetFileExInfoStandard, &fad)) {
if (!ConfirmDirectory(hDlg, szDir))
break;
//
// If it is just a file, delete it
//
// If it is a directory and if the user has decided the
// user that needs to have permissions, then delete the directory.
// and set ACLs using this Sid.
//
// Otherwise if the directory doesn't exist use the default ACLs.
// if it does exist, use the current ACLs
//
if (!(fad.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
SetFileAttributes(szDir, FILE_ATTRIBUTE_NORMAL);
DeleteFile(szDir);
}
else if (!(lpUPCopyInfo->bDefaultSecurity)) {
if (!Delnode(szDir)) {
WCHAR szTitle[100], szMsgFmt[100], szMessage[600];
if (!LoadString (hInstance, IDS_UP_ERRORTITLE, szTitle, ARRAYSIZE(szTitle)))
break;
if (!LoadString (hInstance, IDS_UP_DELNODE_ERROR, szMsgFmt, ARRAYSIZE(szMsgFmt)))
break;
StringCchPrintf(szMessage, ARRAYSIZE(szMessage), szMsgFmt, szDir);
MessageBox(hDlg, szMessage, szTitle, MB_OK | MB_ICONSTOP);
break;
}
}
}
hOldCursor = SetCursor (LoadCursor(NULL, IDC_WAIT));
if (!UPCreateProfile (hDlg, lpUPCopyInfo, szDir, NULL)) {
SetCursor(hOldCursor);
break;
}
LocalFree (lpUPCopyInfo->pSid);
LocalFree (lpUPCopyInfo);
SetCursor(hOldCursor);
EndDialog(hDlg, TRUE);
}
break;
case IDCANCEL:
lpUPCopyInfo = (LPUPCOPYINFO) GetWindowLongPtr(hDlg, GWLP_USERDATA);
if (lpUPCopyInfo) {
LocalFree (lpUPCopyInfo->pSid);
LocalFree(lpUPCopyInfo);
}
EndDialog(hDlg, FALSE);
break;
case IDC_COPY_BROWSE:
{
BROWSEINFO BrowseInfo;
TCHAR szBuffer[MAX_PATH];
LPITEMIDLIST pidl;
LoadString(hInstance, IDS_UP_DIRPICK, szBuffer, ARRAYSIZE(szBuffer));
BrowseInfo.hwndOwner = hDlg;
BrowseInfo.pidlRoot = NULL;
BrowseInfo.pszDisplayName = szBuffer;
BrowseInfo.lpszTitle = szBuffer;
BrowseInfo.ulFlags = BIF_RETURNONLYFSDIRS;
BrowseInfo.lpfn = NULL;
BrowseInfo.lParam = 0;
pidl = SHBrowseForFolder (&BrowseInfo);
if (pidl) {
SHGetPathFromIDList(pidl, szBuffer);
SHFree (pidl);
SetDlgItemText (hDlg, IDC_COPY_PATH, szBuffer);
SetFocus (GetDlgItem(hDlg, IDOK));
}
}
break;
case IDC_COPY_PATH:
if (HIWORD(wParam) == EN_UPDATE) {
if (SendDlgItemMessage(hDlg, IDC_COPY_PATH,
EM_LINELENGTH, 0, 0)) {
EnableWindow (GetDlgItem(hDlg, IDOK), TRUE);
SetDefButton(hDlg, IDOK);
} else {
EnableWindow (GetDlgItem(hDlg, IDOK), FALSE);
SetDefButton(hDlg, IDCANCEL);
}
}
break;
case IDC_COPY_PROFILE:
SetFocus (GetDlgItem(hDlg, IDC_COPY_PATH));
break;
case IDC_COPY_CHANGE:
{
LPUSERDETAILS lpUserDetails;
DWORD dwSize = 1024;
TCHAR szUserName[200];
PSID pNewSid;
lpUPCopyInfo = (LPUPCOPYINFO) GetWindowLongPtr(hDlg, GWLP_USERDATA);
if (!lpUPCopyInfo) {
EndDialog (hDlg, FALSE);
break;
}
lpUserDetails = (LPUSERDETAILS) LocalAlloc (LPTR, dwSize);
if (!lpUserDetails) {
break;
}
if (UPGetUserSelection(hDlg, lpUserDetails) != S_OK) {
LocalFree(lpUserDetails);
break;
}
// Save our new sid
//
lpUPCopyInfo->bDefaultSecurity = FALSE; // at this point user has selected a Sid
LocalFree (lpUPCopyInfo->pSid);
lpUPCopyInfo->pSid = lpUserDetails->psidUser;
// Update the edit control
StringCchCopy(szUserName, ARRAYSIZE(szUserName), lpUserDetails->pszDomainName);
StringCchCat(szUserName, ARRAYSIZE(szUserName), TEXT("\\"));
StringCchCat(szUserName, ARRAYSIZE(szUserName), lpUserDetails->pszAccountName);
SetDlgItemText (hDlg, IDC_COPY_USER, szUserName);
LocalFree(lpUserDetails->pszDomainName);
LocalFree(lpUserDetails->pszAccountName);
// Cleanup
LocalFree (lpUserDetails);
}
break;
default:
break;
}
break;
case WM_HELP: // F1
WinHelp((HWND)((LPHELPINFO) lParam)->hItemHandle, HELP_FILE, HELP_WM_HELP,
(DWORD_PTR) (LPSTR) aUserProfileHelpIds);
break;
case WM_CONTEXTMENU: // right mouse click
WinHelp((HWND) wParam, HELP_FILE, HELP_CONTEXTMENU,
(DWORD_PTR) (LPSTR) aUserProfileHelpIds);
return (TRUE);
}
return (FALSE);
}
//*************************************************************
//
// ConfirmDirectory()
//
// Purpose: Confirms the directory selected
//
// Parameters:
// hDlg - handle to the parent Dialogbox
// szDir - Direcory selected by user that exists
//
// Return: TRUE if it is confirmed
// FALSE if not
//
// Comments:
//
// History: Date Author Comment
// 6/24/99 ushaji Created
//
//*************************************************************
BOOL ConfirmDirectory(HWND hDlg, LPTSTR szDir)
{
LPTSTR szMsgTemplate=NULL, szMsg1=NULL, szMsg=NULL;
BOOL bRetVal = FALSE;
szMsgTemplate = (LPTSTR) LocalAlloc(LPTR, sizeof(TCHAR)*MAX_PATH);
if (!szMsgTemplate)
goto Exit;
szMsg1 = (LPTSTR) LocalAlloc(LPTR, sizeof(TCHAR)*MAX_PATH);
if (!szMsg1)
goto Exit;
szMsg = (LPTSTR) LocalAlloc(LPTR, sizeof(TCHAR)*(500+MAX_PATH));
if (!szMsg)
goto Exit;
if (!LoadString (hInstance, IDS_UP_CONFIRMCOPYMSG, szMsgTemplate, MAX_PATH)) {
goto Exit;
}
StringCchPrintf (szMsg, 500+MAX_PATH, szMsgTemplate, szDir);
if (!LoadString (hInstance, IDS_UP_CONFIRMCOPYTITLE, szMsg1, MAX_PATH)) {
goto Exit;
}
if (MessageBox(hDlg, szMsg, szMsg1,
MB_YESNO | MB_ICONQUESTION | MB_DEFBUTTON2) == IDYES)
bRetVal = TRUE;
Exit:
LocalFree(szMsgTemplate);
LocalFree(szMsg);
LocalFree(szMsg1);
return bRetVal;
}
//*************************************************************
//
// UPGetUserSelection()
//
// Purpose: Shows a UI and gets the user to select a
// username
//
// Parameters: hDlg - Parent window handle
// lpUserDetails - Pointer to an allocated user details
// structure which gets filled up if required
//
//
// Return: TRUE if successful
// FALSE if an error occurs or user chooses cancel
//
// Comments:
//
// History: Date Author Comment
// 4/14/99 ushaji adapted from Rahulth
//
//*************************************************************
HRESULT UPGetUserSelection(HWND hDlg, LPUSERDETAILS lpUserDetails)
{
PCWSTR apwszAttribs[] = {L"ObjectSid"};
DWORD dwError = ERROR_SUCCESS;
HRESULT hr = S_FALSE;
IDsObjectPicker * pDsObjectPicker = NULL;
DSOP_INIT_INFO InitInfo;
const ULONG cbNumScopes = 3;
DSOP_SCOPE_INIT_INFO ascopes[cbNumScopes];
IDataObject * pdo = NULL;
STGMEDIUM stgmedium = {TYMED_HGLOBAL, NULL};
UINT cf = 0;
PDS_SELECTION_LIST pDsSelList = NULL;
FORMATETC formatetc = {
(CLIPFORMAT)cf,
NULL,
DVASPECT_CONTENT,
-1,
TYMED_HGLOBAL
};
PDS_SELECTION pDsSelection = NULL;
BOOL bAllocatedStgMedium = FALSE;
SAFEARRAY * pVariantArr = NULL;
PSID pSid = NULL;
SID_NAME_USE eUse;
DWORD dwNameLen, dwDomLen, dwSize;
ZeroMemory(lpUserDetails, sizeof(USERDETAILS));
// This code is not uninitializing COM.
hr = CoInitialize(NULL);
if (FAILED(hr))
{
goto Exit;
}
hr = CoCreateInstance(CLSID_DsObjectPicker, NULL, CLSCTX_INPROC_SERVER, IID_IDsObjectPicker, (void **) &pDsObjectPicker);
if (FAILED(hr))
{
goto Exit;
}
//Initialize the scopes
ZeroMemory (ascopes, cbNumScopes * sizeof (DSOP_SCOPE_INIT_INFO));
ascopes[0].cbSize = sizeof (DSOP_SCOPE_INIT_INFO);
ascopes[0].flType = DSOP_SCOPE_TYPE_ENTERPRISE_DOMAIN;
ascopes[0].flScope = DSOP_SCOPE_FLAG_STARTING_SCOPE;
ascopes[0].FilterFlags.Uplevel.flBothModes = DSOP_FILTER_USERS |
DSOP_FILTER_BUILTIN_GROUPS |
DSOP_FILTER_WELL_KNOWN_PRINCIPALS |
DSOP_FILTER_UNIVERSAL_GROUPS_DL |
DSOP_FILTER_UNIVERSAL_GROUPS_SE |
DSOP_FILTER_GLOBAL_GROUPS_DL |
DSOP_FILTER_GLOBAL_GROUPS_SE |
DSOP_FILTER_DOMAIN_LOCAL_GROUPS_DL |
DSOP_FILTER_DOMAIN_LOCAL_GROUPS_SE;
ascopes[1].cbSize = sizeof (DSOP_SCOPE_INIT_INFO);
ascopes[1].flType = DSOP_SCOPE_TYPE_GLOBAL_CATALOG |
DSOP_SCOPE_TYPE_EXTERNAL_UPLEVEL_DOMAIN;
ascopes[1].FilterFlags.Uplevel.flBothModes =
ascopes[0].FilterFlags.Uplevel.flBothModes;
ascopes[2].cbSize = sizeof (DSOP_SCOPE_INIT_INFO);
ascopes[2].flType = DSOP_SCOPE_TYPE_TARGET_COMPUTER |
DSOP_SCOPE_TYPE_DOWNLEVEL_JOINED_DOMAIN |
DSOP_SCOPE_TYPE_EXTERNAL_DOWNLEVEL_DOMAIN;
ascopes[2].FilterFlags.flDownlevel = DSOP_DOWNLEVEL_FILTER_USERS |
DSOP_DOWNLEVEL_FILTER_LOCAL_GROUPS |
DSOP_DOWNLEVEL_FILTER_GLOBAL_GROUPS |
DSOP_DOWNLEVEL_FILTER_WORLD |
DSOP_DOWNLEVEL_FILTER_AUTHENTICATED_USER |
DSOP_DOWNLEVEL_FILTER_DIALUP |
DSOP_DOWNLEVEL_FILTER_INTERACTIVE |
DSOP_DOWNLEVEL_FILTER_NETWORK;
//Populate the InitInfo structure that is used to initialize the object
//picker.
ZeroMemory (&InitInfo, sizeof (InitInfo));
InitInfo.cbSize = sizeof (InitInfo);
InitInfo.cDsScopeInfos = cbNumScopes;
InitInfo.aDsScopeInfos = ascopes;
InitInfo.cAttributesToFetch = 1;
InitInfo.apwzAttributeNames = apwszAttribs;
hr = pDsObjectPicker->Initialize (&InitInfo);
if (FAILED(hr))
{
goto Exit;
}
hr = pDsObjectPicker->InvokeDialog (hDlg, &pdo);
if (FAILED(hr))
{
goto Exit;
}
if (S_FALSE == hr)
{ //the user hit cancel
goto Exit;
}
//if we are here, the user chose, OK, so find out what group was chosen
cf = RegisterClipboardFormat (CFSTR_DSOP_DS_SELECTION_LIST);
if (0 == cf)
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto Exit;
}
//set the clipformat for the formatetc structure
formatetc.cfFormat = (CLIPFORMAT)cf;
hr = pdo->GetData (&formatetc, &stgmedium);
if (FAILED(hr))
{
goto Exit;
}
bAllocatedStgMedium = TRUE;
pDsSelList = (PDS_SELECTION_LIST) GlobalLock(stgmedium.hGlobal);
if (NULL == pDsSelList)
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto Exit;
}
if (!pDsSelList->cItems) //some item must have been selected
{
hr = E_UNEXPECTED;
goto Exit;
}
pDsSelection = &(pDsSelList->aDsSelection[0]);
//we must get the ObjectSid attribute, otherwise we fail
if (! (VT_ARRAY & pDsSelection->pvarFetchedAttributes->vt))
{
hr = E_UNEXPECTED;
goto Exit;
}
pVariantArr = pDsSelection->pvarFetchedAttributes->parray;
pSid = (PSID) pVariantArr->pvData;
//store away the string representation of this sid
dwSize = GetLengthSid (pSid);
lpUserDetails->psidUser = LocalAlloc (LPTR, dwSize);
if (!lpUserDetails->psidUser) {
hr = E_OUTOFMEMORY;
goto Exit;
}
if (!CopySid (dwSize, lpUserDetails->psidUser, pSid)) {
hr = HRESULT_FROM_WIN32(GetLastError());
goto Exit;
}
lpUserDetails->pszDomainName = (LPTSTR) LocalAlloc (LPTR, MAX_PATH*sizeof(TCHAR));
lpUserDetails->pszAccountName = (LPTSTR) LocalAlloc (LPTR, MAX_PATH*sizeof(TCHAR));
if ((!lpUserDetails->pszDomainName) || (!lpUserDetails->pszAccountName)) {
hr = E_OUTOFMEMORY;
goto Exit;
}
dwNameLen = dwDomLen = MAX_PATH;
if (!LookupAccountSid (NULL, pSid, lpUserDetails->pszAccountName, &dwNameLen,
lpUserDetails->pszDomainName, &dwDomLen,
&eUse))
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto Exit;
}
hr = S_OK;
Exit:
if (pDsSelList)
GlobalUnlock (stgmedium.hGlobal);
if (bAllocatedStgMedium)
ReleaseStgMedium (&stgmedium);
if (pdo)
pdo->Release();
if (pDsObjectPicker)
pDsObjectPicker->Release ();
if (hr != S_OK) {
LocalFree(lpUserDetails->psidUser);
LocalFree(lpUserDetails->pszDomainName);
LocalFree(lpUserDetails->pszAccountName);
}
return hr;
}
//*************************************************************
//
// UPCreateProfile()
//
// Purpose: Creates a copy of the specified profile with
// the correct security.
//
// Parameters: lpUPCopyInfo - Copy Dialog information
// lpDest - Destination directory
// pNewSecDesc - New security descriptor
//
// Return: TRUE if successful
// FALSE if an error occurs
//
// Comments:
//
// History: Date Author Comment
// 2/13/96 ericflo Created
//
//*************************************************************
BOOL UPCreateProfile (HWND hDlg, LPUPCOPYINFO lpUPCopyInfo, LPTSTR lpDest,
PSECURITY_DESCRIPTOR pNewSecDesc)
{
HKEY RootKey, hKey;
SECURITY_DESCRIPTOR sd;
SECURITY_ATTRIBUTES sa;
SID_IDENTIFIER_AUTHORITY authNT = SECURITY_NT_AUTHORITY;
PACL pAcl = NULL;
PSID psidUser = NULL, psidSystem = NULL, psidAdmin = NULL;
DWORD cbAcl, AceIndex, dwSize;
ACE_HEADER * lpAceHeader;
HANDLE hFile;
BOOL bMandatory = FALSE;
TCHAR szTempPath[MAX_PATH];
TCHAR szBuffer[MAX_PATH];
TCHAR szHive[MAX_PATH];
BOOL bRetVal = FALSE;
HKEY hKeyProfile;
LONG Error;
LPTSTR lpEnd;
TCHAR szExcludeList[2 * MAX_PATH + 1];
DWORD dwErr = 0;
BOOL bSecurityFailed = TRUE;
UINT cchRemaining;
HRESULT hr;
// Getting the previous last error so that we can set the last error to this in the end.
dwErr = GetLastError();
if (!lpDest || !(*lpDest)) {
SetLastError(ERROR_INVALID_PARAMETER);
return FALSE;
}
//
// Create the security descriptor
//
//
// User Sid
//
psidUser = lpUPCopyInfo->pSid;
//
// Get the system sid
//
if (!AllocateAndInitializeSid(&authNT, 1, SECURITY_LOCAL_SYSTEM_RID,
0, 0, 0, 0, 0, 0, 0, &psidSystem)) {
dwErr = 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)) {
dwErr = GetLastError();
goto Exit;
}
//
// Allocate space for the ACL
//
cbAcl = (2 * GetLengthSid (psidUser)) + (2 * GetLengthSid (psidSystem)) +
(2 * GetLengthSid (psidAdmin)) + sizeof(ACL) +
(6 * (sizeof(ACCESS_ALLOWED_ACE) - sizeof(DWORD)));
pAcl = (PACL) LocalAlloc(LPTR, cbAcl);
if (!pAcl) {
dwErr = GetLastError();
goto Exit;
}
if (!InitializeAcl(pAcl, cbAcl, ACL_REVISION)) {
dwErr = GetLastError();
goto Exit;
}
//
// Add Aces for User, System, and Admin. Non-inheritable ACEs first
//
AceIndex = 0;
if (!AddAccessAllowedAce(pAcl, ACL_REVISION, FILE_ALL_ACCESS, psidUser)) {
dwErr = GetLastError();
goto Exit;
}
AceIndex++;
if (!AddAccessAllowedAce(pAcl, ACL_REVISION, FILE_ALL_ACCESS, psidSystem)) {
dwErr = GetLastError();
goto Exit;
}
AceIndex++;
if (!AddAccessAllowedAce(pAcl, ACL_REVISION, FILE_ALL_ACCESS, psidAdmin)) {
dwErr = GetLastError();
goto Exit;
}
//
// Now the inheritable ACEs
//
AceIndex++;
if (!AddAccessAllowedAce(pAcl, ACL_REVISION, GENERIC_ALL, psidUser)) {
dwErr = GetLastError();
goto Exit;
}
if (!GetAce(pAcl, AceIndex, (void **)&lpAceHeader)) {
dwErr = GetLastError();
goto Exit;
}
lpAceHeader->AceFlags |= (OBJECT_INHERIT_ACE | CONTAINER_INHERIT_ACE | INHERIT_ONLY_ACE);
AceIndex++;
if (!AddAccessAllowedAce(pAcl, ACL_REVISION, GENERIC_ALL, psidSystem)) {
dwErr = GetLastError();
goto Exit;
}
if (!GetAce(pAcl, AceIndex, (void **)&lpAceHeader)) {
dwErr = GetLastError();
goto Exit;
}
lpAceHeader->AceFlags |= (OBJECT_INHERIT_ACE | CONTAINER_INHERIT_ACE | INHERIT_ONLY_ACE);
AceIndex++;
if (!AddAccessAllowedAce(pAcl, ACL_REVISION, GENERIC_ALL, psidAdmin)) {
dwErr = GetLastError();
goto Exit;
}
if (!GetAce(pAcl, AceIndex, (void **)&lpAceHeader)) {
dwErr = GetLastError();
goto Exit;
}
lpAceHeader->AceFlags |= (OBJECT_INHERIT_ACE | CONTAINER_INHERIT_ACE | INHERIT_ONLY_ACE);
//
// Put together the security descriptor
//
if (!InitializeSecurityDescriptor(&sd, SECURITY_DESCRIPTOR_REVISION)) {
dwErr = GetLastError();
goto Exit;
}
if (!SetSecurityDescriptorDacl(&sd, TRUE, pAcl, FALSE)) {
dwErr = GetLastError();
goto Exit;
}
//
// Add the security descriptor to the sa structure
//
sa.nLength = sizeof(sa);
sa.lpSecurityDescriptor = &sd;
sa.bInheritHandle = FALSE;
//
// Create the destination directory
//
dwErr = SHCreateDirectoryEx (hDlg, lpDest, &sa);
if (ERROR_SUCCESS != dwErr && ERROR_ALREADY_EXISTS != dwErr) {
goto Exit;
}
//
// Save/copy the user's profile to a temp file
//
dwSize = GetTempPath(ARRAYSIZE(szTempPath), szTempPath);
if (!dwSize || dwSize >= MAX_PATH) {
dwErr = dwSize ? ERROR_INSUFFICIENT_BUFFER : GetLastError();
goto Exit;
}
//
// GetTempFileName is lame in the sense it does not take output
// buffer length - so potentially it can overflow the buffer.
// So we want to ensure a 20 [3(prefix)+3(suffix)+1(null)+some hex string) char
// buffer left for file name generation to be safe
//
if (dwSize > ARRAYSIZE(szTempPath) - 20) {
dwErr = ERROR_INSUFFICIENT_BUFFER;
goto Exit;
}
if (!GetTempFileName (szTempPath, TEXT("TMP"), 0, szBuffer)) {
dwErr = GetLastError();
goto Exit;
}
DeleteFile (szBuffer);
//
// Determine if we are working with a mandatory profile
//
hr = StringCchCopy(szHive, ARRAYSIZE(szHive), lpUPCopyInfo->lpUserInfo->lpProfile);
if (FAILED(hr)) {
dwErr = HRESULT_CODE(hr);
goto Exit;
}
lpEnd = CheckSlashEx(szHive, ARRAYSIZE(szHive), &cchRemaining);
if (!lpEnd)
{
dwErr = ERROR_INSUFFICIENT_BUFFER;
goto Exit;
}
hr = StringCchCopy(lpEnd, cchRemaining, TEXT("ntuser.man"));
if (FAILED(hr)) {
dwErr = HRESULT_CODE(hr);
goto Exit;
}
hFile = CreateFile(szHive, GENERIC_READ, FILE_SHARE_READ, NULL,
OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if (hFile != INVALID_HANDLE_VALUE) {
CloseHandle (hFile);
bMandatory = TRUE;
}
//
// Test if the requested profile is in use.
//
if (IsProfileInUse (lpUPCopyInfo->lpUserInfo->lpSid)) {
Error = RegOpenKeyEx (HKEY_USERS, lpUPCopyInfo->lpUserInfo->lpSid, 0,
KEY_READ, &hKeyProfile);
if (Error != ERROR_SUCCESS) {
dwErr = Error;
goto Exit;
}
Error = MyRegSaveKey (hKeyProfile, szBuffer);
RegCloseKey (hKeyProfile);
if (Error != ERROR_SUCCESS) {
DeleteFile (szBuffer);
dwErr = Error;
goto Exit;
}
} else {
if (!bMandatory) {
hr = StringCchCopy(lpEnd, cchRemaining, TEXT("ntuser.dat"));
if (FAILED(hr)) {
dwErr = HRESULT_CODE(hr);
goto Exit;
}
}
if (!CopyFile(szHive, szBuffer, FALSE)) {
dwErr = GetLastError();
goto Exit;
}
}
//
// Apply security to the hive
//
Error = MyRegLoadKey (HKEY_USERS, TEMP_PROFILE, szBuffer);
if (Error != ERROR_SUCCESS) {
DeleteFile (szBuffer);
dwErr = Error;
goto Exit;
}
bRetVal = ApplyHiveSecurity(TEMP_PROFILE, psidUser);
//
// Query for the user's exclusion list
//
if (bRetVal) {
//
// if it has come till here, we have managed to set the ACLs correctly
//
bSecurityFailed = FALSE;
//
// Open the root of the user's profile
//
if (RegOpenKeyEx(HKEY_USERS, TEMP_PROFILE, 0, KEY_READ, &RootKey) == ERROR_SUCCESS) {
//
// Check for a list of directories to exclude both user preferences
// and user policy
//
szExcludeList[0] = TEXT('\0');
if (!ReadExclusionList(RootKey, szExcludeList, ARRAYSIZE(szExcludeList))) {
szExcludeList[0] = TEXT('\0');
}
// Always exclude USER_CRED_LOCATION
if (szExcludeList[0] != TEXT('\0')) {
if (!CheckSemicolon(szExcludeList, ARRAYSIZE(szExcludeList))) {
szExcludeList[0] = TEXT('\0');
}
}
StringCchCat(szExcludeList, ARRAYSIZE(szExcludeList), USER_CRED_LOCATION);
RegCloseKey (RootKey);
}
}
else {
dwErr = GetLastError();
}
//
// Unload the hive first then check for ApplyHiveSecurity failure
//
Error = MyRegUnLoadKey(HKEY_USERS, TEMP_PROFILE);
if (Error != ERROR_SUCCESS) {
DeleteFile (szBuffer);
StringCchCat(szBuffer, ARRAYSIZE(szBuffer), TEXT(".log"));
DeleteFile (szBuffer);
dwErr = Error;
goto Exit;
}
if (bSecurityFailed) {
DeleteFile (szBuffer);
StringCchCat(szBuffer, ARRAYSIZE(szBuffer), TEXT(".log"));
DeleteFile (szBuffer);
goto Exit; // dwErr already set from ApplyHiveSecurity
}
//
// Copy the profile without the hive
//
bRetVal = CopyProfileDirectoryEx (lpUPCopyInfo->lpUserInfo->lpProfile,
lpDest,
CPD_IGNOREHIVE |
CPD_COPYIFDIFFERENT |
CPD_SYNCHRONIZE |
CPD_USEEXCLUSIONLIST |
CPD_IGNORESECURITY |
CPD_DELDESTEXCLUSIONS,
NULL,
(szExcludeList[0] != TEXT('\0')) ?
szExcludeList : NULL);
if (!bRetVal) {
dwErr = GetLastError();
DeleteFile(szBuffer);
StringCchCat(szBuffer, ARRAYSIZE(szBuffer), TEXT(".log"));
DeleteFile (szBuffer);
goto Exit;
}
//
// Now copy the hive
//
hr = StringCchCopy(szHive, ARRAYSIZE(szHive), lpDest);
if (FAILED(hr))
{
dwErr = HRESULT_CODE(hr);
bRetVal = FALSE;
}
else
{
lpEnd = CheckSlashEx(szHive, ARRAYSIZE(szHive), &cchRemaining);
if (!lpEnd)
{
dwErr = ERROR_INSUFFICIENT_BUFFER;
bRetVal = FALSE;
}
else
{
hr = StringCchCopy (lpEnd, cchRemaining, bMandatory ? TEXT("ntuser.man") : TEXT("ntuser.dat"));
if (FAILED(hr)) {
dwErr = HRESULT_CODE(hr);
bRetVal = FALSE;
}
}
}
//
// Setting the file attributes first
//
if (bRetVal) {
SetFileAttributes (szHive, FILE_ATTRIBUTE_NORMAL);
bRetVal = CopyFile (szBuffer, szHive, FALSE);
if (!bRetVal) {
dwErr = GetLastError();
}
}
if (!bRetVal) {
TCHAR szMsg[MAX_PATH], szMsgTemplate[MAX_PATH];
szMsg[0] = szMsgTemplate[0] = TEXT('\0');
LoadString (hInstance, IDS_UP_COPYHIVE_ERROR, szMsgTemplate, ARRAYSIZE(szMsgTemplate));
StringCchPrintf(szMsg, ARRAYSIZE(szMsg), szMsgTemplate, szHive);
UPDisplayErrorMessage(hDlg, dwErr, szMsg);
}
//
// Delete the temp file (and log file)
//
DeleteFile (szBuffer);
StringCchCat(szBuffer, ARRAYSIZE(szBuffer), TEXT(".log"));
DeleteFile (szBuffer);
//
// Since we use CPD_IGNOREHIVE, all "ntuser*.*" file will be skipped, including "ntuser.pol" and
// "ntuser.ini" etc, now copy those files also, but we need to skip the hive files
//
if (bRetVal)
{
hr = CopyNonHiveNtUserFiles(lpUPCopyInfo->lpUserInfo->lpProfile, lpDest);
if (FAILED(hr))
{
bRetVal = FALSE;
dwErr = HRESULT_CODE(hr);
}
}
Exit:
//
// Free the sids and acl
//
if (psidSystem) {
FreeSid(psidSystem);
}
if (psidAdmin) {
FreeSid(psidAdmin);
}
LocalFree (pAcl);
if ((!bRetVal) && (bSecurityFailed)) {
TCHAR szMsg[MAX_PATH];
szMsg[0] = TEXT('\0');
LoadString (hInstance, IDS_UP_SETSECURITY_ERROR, szMsg, ARRAYSIZE(szMsg));
UPDisplayErrorMessage(hDlg, dwErr, szMsg);
}
SetLastError(dwErr);
return (bRetVal);
}
//*************************************************************
//
// UPDisplayErrorMessage()
//
// Purpose: Display an error message
//
// Parameters: hWnd - parent window handle
// uiSystemError - Error code
//
// Return: void
//
// Comments:
//
// History: Date Author Comment
// 2/14/96 ericflo Created
//
//*************************************************************
VOID UPDisplayErrorMessage(HWND hWnd, UINT uiSystemError, LPTSTR szMsgPrefix)
{
TCHAR szMessage[MAX_PATH];
TCHAR szTitle[100];
LPTSTR lpEnd;
if (szMsgPrefix) {
StringCchCopy(szMessage, ARRAYSIZE(szMessage), szMsgPrefix);
}
else {
szMessage[0] = TEXT('\0');
}
lpEnd = szMessage+lstrlen(szMessage);
//
// retrieve the string matching the Win32 system error
//
FormatMessage(
FORMAT_MESSAGE_IGNORE_INSERTS | FORMAT_MESSAGE_FROM_SYSTEM,
NULL,
uiSystemError,
MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL),
lpEnd,
MAX_PATH-lstrlen(szMessage),
NULL);
//
// display a message box with this error
//
LoadString (hInstance, IDS_UP_ERRORTITLE, szTitle, ARRAYSIZE(szTitle));
MessageBox(hWnd, szMessage, szTitle, MB_OK | MB_ICONSTOP);
return;
}
//*************************************************************
//
// ApplySecurityToRegistryTree()
//
// Purpose: Applies the passed security descriptor to the passed
// key and all its descendants. Only the parts of
// the descriptor inddicated in the security
// info value are actually applied to each registry key.
//
// Parameters: RootKey - Registry key
// pSD - Security Descriptor
//
// Return: ERROR_SUCCESS if successful
//
// Comments:
//
// History: Date Author Comment
// 7/19/95 ericflo Created
//
//*************************************************************
DWORD ApplySecurityToRegistryTree(HKEY RootKey, PSECURITY_DESCRIPTOR pSD)
{
DWORD Error;
DWORD SubKeyIndex;
LPTSTR SubKeyName;
HKEY SubKey;
DWORD cchSubKeySize = MAX_PATH + 1;
//
// First apply security
//
RegSetKeySecurity(RootKey, DACL_SECURITY_INFORMATION, pSD);
//
// Open each sub-key and apply security to its sub-tree
//
SubKeyIndex = 0;
SubKeyName = (LPTSTR) LocalAlloc (LPTR, cchSubKeySize * sizeof(TCHAR));
if (!SubKeyName) {
return GetLastError();
}
while (TRUE) {
//
// Get the next sub-key name
//
Error = RegEnumKey(RootKey, SubKeyIndex, SubKeyName, cchSubKeySize);
if (Error != ERROR_SUCCESS) {
if (Error == ERROR_NO_MORE_ITEMS) {
//
// Successful end of enumeration
//
Error = ERROR_SUCCESS;
}
break;
}
//
// Open the sub-key
//
Error = RegOpenKeyEx(RootKey,
SubKeyName,
0,
WRITE_DAC | KEY_ENUMERATE_SUB_KEYS | READ_CONTROL,
&SubKey);
if (Error == ERROR_SUCCESS) {
//
// Apply security to the sub-tree
//
ApplySecurityToRegistryTree(SubKey, pSD);
//
// We're finished with the sub-key
//
RegCloseKey(SubKey);
}
//
// Go enumerate the next sub-key
//
SubKeyIndex ++;
}
LocalFree (SubKeyName);
return Error;
}
//*************************************************************
//
// ApplyHiveSecurity()
//
// Purpose: Initializes the new user hive created by copying
// the default hive.
//
// Parameters: lpHiveName - Name of hive in HKEY_USERS
// pSid - Sid (used by CreateNewUser)
//
// Return: TRUE if successful
// FALSE if an error occurs
//
// Comments:
//
// History: Date Author Comment
// 7/18/95 ericflo Created
//
//*************************************************************
BOOL ApplyHiveSecurity(LPTSTR lpHiveName, PSID pSid)
{
DWORD Error;
HKEY RootKey;
SECURITY_DESCRIPTOR sd;
SID_IDENTIFIER_AUTHORITY authNT = SECURITY_NT_AUTHORITY;
PACL pAcl = NULL;
PSID psidUser = pSid, psidSystem = NULL, psidAdmin = NULL;
DWORD cbAcl, AceIndex;
ACE_HEADER * lpAceHeader;
BOOL bRetVal = FALSE;
//
// Get the system sid
//
if (!AllocateAndInitializeSid(&authNT, 1, SECURITY_LOCAL_SYSTEM_RID,
0, 0, 0, 0, 0, 0, 0, &psidSystem)) {
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)) {
goto Exit;
}
//
// Allocate space for the ACL
//
cbAcl = (2 * GetLengthSid (psidUser)) + (2 * GetLengthSid (psidSystem)) +
(2 * GetLengthSid (psidAdmin)) + sizeof(ACL) +
(6 * (sizeof(ACCESS_ALLOWED_ACE) - sizeof(DWORD)));
pAcl = (PACL) LocalAlloc(LPTR, cbAcl);
if (!pAcl) {
goto Exit;
}
if (!InitializeAcl(pAcl, cbAcl, ACL_REVISION)) {
goto Exit;
}
//
// Add Aces for User, System, and Admin. Non-inheritable ACEs first
//
AceIndex = 0;
if (!AddAccessAllowedAce(pAcl, ACL_REVISION, KEY_ALL_ACCESS, psidUser)) {
goto Exit;
}
AceIndex++;
if (!AddAccessAllowedAce(pAcl, ACL_REVISION, KEY_ALL_ACCESS, psidSystem)) {
goto Exit;
}
AceIndex++;
if (!AddAccessAllowedAce(pAcl, ACL_REVISION, KEY_ALL_ACCESS, psidAdmin)) {
goto Exit;
}
//
// Now the inheritable ACEs
//
AceIndex++;
if (!AddAccessAllowedAce(pAcl, ACL_REVISION, GENERIC_ALL, psidUser)) {
goto Exit;
}
if (!GetAce(pAcl, AceIndex, (void **)&lpAceHeader)) {
goto Exit;
}
lpAceHeader->AceFlags |= (OBJECT_INHERIT_ACE | CONTAINER_INHERIT_ACE | INHERIT_ONLY_ACE);
AceIndex++;
if (!AddAccessAllowedAce(pAcl, ACL_REVISION, GENERIC_ALL, psidSystem)) {
goto Exit;
}
if (!GetAce(pAcl, AceIndex, (void **)&lpAceHeader)) {
goto Exit;
}
lpAceHeader->AceFlags |= (OBJECT_INHERIT_ACE | CONTAINER_INHERIT_ACE | INHERIT_ONLY_ACE);
AceIndex++;
if (!AddAccessAllowedAce(pAcl, ACL_REVISION, GENERIC_ALL, psidAdmin)) {
goto Exit;
}
if (!GetAce(pAcl, AceIndex, (void **)&lpAceHeader)) {
goto Exit;
}
lpAceHeader->AceFlags |= (OBJECT_INHERIT_ACE | CONTAINER_INHERIT_ACE | INHERIT_ONLY_ACE);
//
// Put together the security descriptor
//
if (!InitializeSecurityDescriptor(&sd, SECURITY_DESCRIPTOR_REVISION)) {
goto Exit;
}
if (!SetSecurityDescriptorDacl(&sd, TRUE, pAcl, FALSE)) {
goto Exit;
}
//
// Open the root of the user's profile
//
Error = RegOpenKeyEx(HKEY_USERS,
lpHiveName,
0,
WRITE_DAC | KEY_ENUMERATE_SUB_KEYS | READ_CONTROL,
&RootKey);
if (Error == ERROR_SUCCESS) {
//
// Set the security descriptor on the entire tree
//
Error = ApplySecurityToRegistryTree(RootKey, &sd);
if (Error == ERROR_SUCCESS) {
bRetVal = TRUE;
}
RegFlushKey (RootKey);
RegCloseKey(RootKey);
}
Exit:
//
// Free the sids and acl
//
if (psidSystem) {
FreeSid(psidSystem);
}
if (psidAdmin) {
FreeSid(psidAdmin);
}
LocalFree (pAcl);
return(bRetVal);
}
HRESULT CopyNonHiveNtUserFiles(LPCTSTR lpSrcDir, LPCTSTR lpDestDir)
{
HRESULT hr = E_FAIL;
TCHAR lpSrcBuffer[MAX_PATH];
TCHAR lpDestBuffer[MAX_PATH];
LPTSTR lpSrcEnd = NULL;
UINT cchSrcEnd = 0;
LPTSTR lpDestEnd = NULL;
UINT cchDestEnd = 0;
HANDLE hFile = INVALID_HANDLE_VALUE;
WIN32_FIND_DATA fd;
//
// Prepare the buffers
//
hr = StringCchCopy(lpSrcBuffer, ARRAYSIZE(lpSrcBuffer), lpSrcDir);
if (FAILED(hr))
{
goto Exit;
}
lpSrcEnd = CheckSlashEx(lpSrcBuffer, ARRAYSIZE(lpSrcBuffer), &cchSrcEnd);
if (!lpSrcEnd)
{
hr = HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER);
goto Exit;
}
hr = StringCchCopy(lpDestBuffer, ARRAYSIZE(lpDestBuffer), lpDestDir);
if (FAILED(hr))
{
goto Exit;
}
lpDestEnd = CheckSlashEx(lpDestBuffer, ARRAYSIZE(lpDestBuffer), &cchDestEnd);
if (!lpDestEnd)
{
hr = HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER);
goto Exit;
}
//
// Search "ntuser*.*"
//
hr = StringCchCopy(lpSrcEnd, cchSrcEnd, TEXT("ntuser*.*"));
if (FAILED(hr))
{
goto Exit;
}
hFile = FindFirstFile(lpSrcBuffer, &fd);
if (hFile != INVALID_HANDLE_VALUE)
{
do
{
//
// Skip hive files
//
if (CompareString(LOCALE_INVARIANT, NORM_IGNORECASE, fd.cFileName, -1, TEXT("ntuser.man"), -1) == CSTR_EQUAL ||
CompareString(LOCALE_INVARIANT, NORM_IGNORECASE, fd.cFileName, -1, TEXT("ntuser.dat"), -1) == CSTR_EQUAL ||
CompareString(LOCALE_INVARIANT, NORM_IGNORECASE, fd.cFileName, -1, TEXT("ntuser.tmp"), -1) == CSTR_EQUAL ||
CompareString(LOCALE_INVARIANT, NORM_IGNORECASE, fd.cFileName, -1, TEXT("ntuser.man.log"), -1) == CSTR_EQUAL ||
CompareString(LOCALE_INVARIANT, NORM_IGNORECASE, fd.cFileName, -1, TEXT("ntuser.dat.log"), -1) == CSTR_EQUAL )
{
continue;
}
//
// Construct file names
//
hr = StringCchCopy (lpSrcEnd, cchSrcEnd, fd.cFileName);
if (FAILED(hr))
{
goto Exit;
}
hr = StringCchCopy (lpDestEnd, cchDestEnd, fd.cFileName);
if (FAILED(hr))
{
goto Exit;
}
//
// Clear the dest file attribute in case it exists
//
SetFileAttributes (lpDestBuffer, FILE_ATTRIBUTE_NORMAL);
//
// Now copy the file
//
if(!CopyFile(lpSrcBuffer, lpDestBuffer, FALSE))
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto Exit;
}
} while (FindNextFile(hFile, &fd));
FindClose(hFile);
hFile = INVALID_HANDLE_VALUE;
}
hr = S_OK;
Exit:
if (hFile != INVALID_HANDLE_VALUE)
FindClose(hFile);
return hr;
}
//*************************************************************
//
// CheckSlashEx()
//
// Purpose: Checks for an ending slash and adds one if
// it is missing. It will take the buffer size
// to make it safe (not overflow the buffer).
//
// Parameters: lpDir - directory
// cchBuffer - buffer size
// pcchRemain - buffer remained after patch '\'
// can be NULL if not needed.
//
// Return: Pointer to the end of the string, NULL for
// overflowed buffer.
//
// Comments:
//
// History: Date Author Comment
// 06/19/95 ericflo Created
// 02/11/02 mingzhu Make it safe
//
//*************************************************************
LPTSTR CheckSlashEx(LPTSTR lpDir, UINT cchBuffer, UINT* pcchRemain )
{
LPTSTR lpEnd = NULL;
UINT cchLen = lstrlen(lpDir);
if (cchLen >= cchBuffer - 1) // Overflowed or full buffer
{
//DmAssert(cchLen == cchBuffer - 1); // Should never happen
if (pcchRemain)
*pcchRemain = 0;
lpEnd = NULL;
}
else
{
lpEnd = lpDir + cchLen;
if (pcchRemain)
*pcchRemain = cchBuffer - 1 - cchLen;
if (*(lpEnd - 1) != TEXT('\\'))
{
*lpEnd = TEXT('\\');
lpEnd++;
*lpEnd = TEXT('\0');
if (pcchRemain)
(*pcchRemain) --;
}
}
return lpEnd;
}