xref: /reactos/dll/cpl/sysdm/userprofile.c (revision 94a413ae) 
1 /*
2  * PROJECT:     ReactOS System Control Panel Applet
3  * LICENSE:     GPL - See COPYING in the top level directory
4  * FILE:        dll/cpl/sysdm/userprofile.c
5  * PURPOSE:     Computer settings for networking
6  * COPYRIGHT:   Copyright Thomas Weidenmueller <w3seek@reactos.org>
7  *              Copyright 2006 Ged Murphy <gedmurphy@gmail.com>
8  *
9  */
10 
11 #include "precomp.h"
12 #include <sddl.h>
13 #include <winnls.h>
14 
15 #include <debug.h>
16 
17 typedef struct _PROFILEDATA
18 {
19     DWORD dwRefCount;
20     DWORD dwState;
21     BOOL bUnknownProfile;
22     PWSTR pszFullName;
23     PWSTR pszProfilePath;
24 } PROFILEDATA, *PPROFILEDATA;
25 
26 
27 static
28 BOOL
29 OnProfileTypeInit(
30     _In_ HWND hwndDlg,
31     _In_ PPROFILEDATA pProfileData)
32 {
33     PWSTR pszRawBuffer = NULL, pszCookedBuffer = NULL;
34     INT nLength;
35 
36     nLength = LoadStringW(hApplet, IDS_USERPROFILE_TYPE_TEXT, (PWSTR)&pszRawBuffer, 0);
37     pszRawBuffer = NULL;
38     if (nLength == 0)
39         return FALSE;
40 
41     pszRawBuffer = HeapAlloc(GetProcessHeap(), 0, (nLength + 1) * sizeof(WCHAR));
42     if (pszRawBuffer == NULL)
43         return FALSE;
44 
45     LoadStringW(hApplet, IDS_USERPROFILE_TYPE_TEXT, pszRawBuffer, nLength + 1);
46 
47     pszCookedBuffer = HeapAlloc(GetProcessHeap(), 0, (nLength + wcslen(pProfileData->pszFullName) + 1) * sizeof(WCHAR));
48     if (pszCookedBuffer == NULL)
49         goto done;
50 
51     swprintf(pszCookedBuffer, pszRawBuffer, pProfileData->pszFullName);
52 
53     /* Set the full text */
54     SetDlgItemText(hwndDlg, IDC_USERPROFILE_TYPE_TEXT, pszCookedBuffer);
55 
56     /* FIXME: Right now, we support local user profiles only! */
57     EnableWindow(GetDlgItem(hwndDlg, IDC_USERPROFILE_TYPE_ROAMING), FALSE);
58     Button_SetCheck(GetDlgItem(hwndDlg, IDC_USERPROFILE_TYPE_LOCAL), BST_CHECKED);
59     EnableWindow(GetDlgItem(hwndDlg, IDOK), FALSE);
60 
61 done:
62     if (pszCookedBuffer != NULL)
63         HeapFree(GetProcessHeap(), 0, pszCookedBuffer);
64 
65     if (pszRawBuffer != NULL)
66         HeapFree(GetProcessHeap(), 0, pszRawBuffer);
67 
68     return TRUE;
69 }
70 
71 
72 static
73 INT_PTR
74 CALLBACK
75 UserProfileTypeDlgProc(
76     _In_ HWND hwndDlg,
77     _In_ UINT uMsg,
78     _In_ WPARAM wParam,
79     _In_ LPARAM lParam)
80 {
81     switch (uMsg)
82     {
83         case WM_INITDIALOG:
84             OnProfileTypeInit(hwndDlg, (PPROFILEDATA)lParam);
85             return TRUE;
86 
87         case WM_DESTROY:
88             break;
89 
90         case WM_COMMAND:
91             switch (LOWORD(wParam))
92             {
93                 case IDOK:
94                 case IDCANCEL:
95                     EndDialog(hwndDlg,
96                               LOWORD(wParam));
97                     return TRUE;
98             }
99             break;
100     }
101 
102     return FALSE;
103 }
104 
105 
106 static
107 BOOL
108 ChangeUserProfileType(
109     _In_ HWND hwndDlg)
110 {
111     HWND hwndListView;
112     LVITEM Item;
113     INT iSelected;
114 
115     DPRINT("ChangeUserProfileType(%p)\n", hwndDlg);
116 
117     hwndListView = GetDlgItem(hwndDlg, IDC_USERPROFILE_LIST);
118     if (hwndListView == NULL)
119         return FALSE;
120 
121     iSelected = ListView_GetNextItem(hwndListView, -1, LVNI_SELECTED);
122     if (iSelected == -1)
123         return FALSE;
124 
125     ZeroMemory(&Item, sizeof(LVITEM));
126     Item.mask = LVIF_PARAM;
127     Item.iItem = iSelected;
128     if (!ListView_GetItem(hwndListView, &Item))
129         return FALSE;
130 
131     if (Item.lParam == 0)
132         return FALSE;
133 
134     if (DialogBoxParam(hApplet,
135                        MAKEINTRESOURCE(IDD_USERPROFILE_TYPE),
136                        hwndDlg,
137                        UserProfileTypeDlgProc,
138                        (LPARAM)Item.lParam) == IDOK)
139     {
140         /* FIXME: Update the profile list view */
141         return TRUE;
142     }
143 
144     return FALSE;
145 }
146 
147 
148 static
149 BOOL
150 DeleteUserProfile(
151     _In_ HWND hwndDlg)
152 {
153     WCHAR szTitle[64], szRawText[128], szCookedText[256];
154     HWND hwndListView;
155     LVITEM Item;
156     INT iSelected;
157     PPROFILEDATA pProfileData;
158 
159     DPRINT("DeleteUserProfile()\n");
160 
161     hwndListView = GetDlgItem(hwndDlg, IDC_USERPROFILE_LIST);
162     if (hwndListView == NULL)
163         return FALSE;
164 
165     iSelected = ListView_GetNextItem(hwndListView, -1, LVNI_SELECTED);
166     if (iSelected == -1)
167         return FALSE;
168 
169     ZeroMemory(&Item, sizeof(LVITEM));
170     Item.mask = LVIF_PARAM;
171     Item.iItem = iSelected;
172     if (!ListView_GetItem(hwndListView, &Item))
173         return FALSE;
174 
175     if (Item.lParam == 0)
176         return FALSE;
177 
178     pProfileData = (PPROFILEDATA)Item.lParam;
179     if (pProfileData->dwRefCount != 0)
180         return FALSE;
181 
182     LoadStringW(hApplet, IDS_USERPROFILE_CONFIRM_DELETE_TITLE, szTitle, ARRAYSIZE(szTitle));
183     LoadStringW(hApplet, IDS_USERPROFILE_CONFIRM_DELETE, szRawText, ARRAYSIZE(szRawText));
184     swprintf(szCookedText, szRawText, pProfileData->pszFullName);
185 
186     if (MessageBoxW(hwndDlg,
187                     szCookedText,
188                     szTitle,
189                     MB_ICONQUESTION | MB_YESNO) == IDYES)
190     {
191         /* FIXME: Delete the profile here! */
192         return TRUE;
193     }
194 
195     return FALSE;
196 }
197 
198 
199 static
200 INT_PTR
201 CALLBACK
202 CopyUserProfileDlgProc(
203     _In_ HWND hwndDlg,
204     _In_ UINT uMsg,
205     _In_ WPARAM wParam,
206     _In_ LPARAM lParam)
207 {
208     switch (uMsg)
209     {
210         case WM_INITDIALOG:
211             return TRUE;
212 
213         case WM_DESTROY:
214             break;
215 
216         case WM_COMMAND:
217             switch (LOWORD(wParam))
218             {
219                 case IDOK:
220                 case IDCANCEL:
221                     EndDialog(hwndDlg,
222                               LOWORD(wParam));
223                     return TRUE;
224             }
225             break;
226     }
227 
228     return FALSE;
229 }
230 
231 
232 static
233 BOOL
234 CopyUserProfile(
235     _In_ HWND hwndDlg)
236 {
237     HWND hwndListView;
238     LVITEM Item;
239     INT iSelected;
240 
241     DPRINT("CopyUserProfile()\n");
242 
243     hwndListView = GetDlgItem(hwndDlg, IDC_USERPROFILE_LIST);
244     if (hwndListView == NULL)
245         return FALSE;
246 
247     iSelected = ListView_GetNextItem(hwndListView, -1, LVNI_SELECTED);
248     if (iSelected == -1)
249         return FALSE;
250 
251     ZeroMemory(&Item, sizeof(LVITEM));
252     Item.mask = LVIF_PARAM;
253     Item.iItem = iSelected;
254     if (!ListView_GetItem(hwndListView, &Item))
255         return FALSE;
256 
257     if (Item.lParam == 0)
258         return FALSE;
259 
260     if (DialogBoxParam(hApplet,
261                        MAKEINTRESOURCE(IDD_USERPROFILE_COPY),
262                        hwndDlg,
263                        CopyUserProfileDlgProc,
264                        (LPARAM)Item.lParam) == IDOK)
265     {
266         /* FIXME: Update the profile list view */
267         return TRUE;
268     }
269 
270     return FALSE;
271 }
272 
273 
274 static VOID
275 SetListViewColumns(
276     _In_ HWND hwndListView)
277 {
278     LV_COLUMN column;
279     RECT rect;
280     TCHAR szStr[32];
281 
282     GetClientRect(hwndListView, &rect);
283 
284     SendMessage(hwndListView, LVM_SETEXTENDEDLISTVIEWSTYLE, 0, LVS_EX_FULLROWSELECT);
285 
286     memset(&column, 0x00, sizeof(column));
287     column.mask = LVCF_FMT | LVCF_WIDTH | LVCF_SUBITEM | LVCF_TEXT;
288     column.fmt = LVCFMT_LEFT;
289     column.cx = (INT)((rect.right - rect.left) * 0.40);
290     column.iSubItem = 0;
291     LoadString(hApplet, IDS_USERPROFILE_NAME, szStr, ARRAYSIZE(szStr));
292     column.pszText = szStr;
293     (void)ListView_InsertColumn(hwndListView, 0, &column);
294 
295     column.fmt = LVCFMT_RIGHT;
296     column.cx = (INT)((rect.right - rect.left) * 0.15);
297     column.iSubItem = 1;
298     LoadString(hApplet, IDS_USERPROFILE_SIZE, szStr, ARRAYSIZE(szStr));
299     column.pszText = szStr;
300     (void)ListView_InsertColumn(hwndListView, 1, &column);
301 
302     column.fmt = LVCFMT_LEFT;
303     column.cx = (INT)((rect.right - rect.left) * 0.15);
304     column.iSubItem = 2;
305     LoadString(hApplet, IDS_USERPROFILE_TYPE, szStr, ARRAYSIZE(szStr));
306     column.pszText = szStr;
307     (void)ListView_InsertColumn(hwndListView, 2, &column);
308 
309     column.fmt = LVCFMT_LEFT;
310     column.cx = (INT)((rect.right - rect.left) * 0.15);
311     column.iSubItem = 3;
312     LoadString(hApplet, IDS_USERPROFILE_STATUS, szStr, ARRAYSIZE(szStr));
313     column.pszText = szStr;
314     (void)ListView_InsertColumn(hwndListView, 3, &column);
315 
316     column.fmt = LVCFMT_LEFT;
317     column.cx = (INT)((rect.right - rect.left) * 0.15) - GetSystemMetrics(SM_CYHSCROLL);
318     column.iSubItem = 4;
319     LoadString(hApplet, IDS_USERPROFILE_MODIFIED, szStr, ARRAYSIZE(szStr));
320     column.pszText = szStr;
321     (void)ListView_InsertColumn(hwndListView, 4, &column);
322 }
323 
324 
325 static
326 BOOL
327 GetProfileSize(
328     _In_ PWSTR pszProfilePath,
329     _Inout_ PULONGLONG pullProfileSize)
330 {
331     HANDLE hFile = INVALID_HANDLE_VALUE;
332     WIN32_FIND_DATA FindData;
333     DWORD dwProfilePathLength;
334     ULARGE_INTEGER Size;
335     BOOL bResult = TRUE;
336 
337     dwProfilePathLength = wcslen(pszProfilePath);
338 
339     wcscat(pszProfilePath, L"\\*.*");
340 
341     hFile = FindFirstFileW(pszProfilePath, &FindData);
342     if (hFile == INVALID_HANDLE_VALUE)
343     {
344         if ((GetLastError() != ERROR_FILE_NOT_FOUND) &&
345             (GetLastError() != ERROR_PATH_NOT_FOUND))
346             bResult = FALSE;
347 
348         goto done;
349     }
350 
351     do
352     {
353         if (FindData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
354         {
355             if ((_wcsicmp(FindData.cFileName, L".") == 0) ||
356                 (_wcsicmp(FindData.cFileName, L"..") == 0))
357                 continue;
358 
359             pszProfilePath[dwProfilePathLength + 1] = UNICODE_NULL;
360             wcscat(pszProfilePath, FindData.cFileName);
361 
362             if (!GetProfileSize(pszProfilePath, pullProfileSize))
363             {
364                 bResult = FALSE;
365                 goto done;
366             }
367         }
368         else
369         {
370             Size.u.LowPart = FindData.nFileSizeLow;
371             Size.u.HighPart = FindData.nFileSizeHigh;
372             *pullProfileSize += Size.QuadPart;
373         }
374     }
375     while (FindNextFile(hFile, &FindData));
376 
377 done:
378     pszProfilePath[dwProfilePathLength] = UNICODE_NULL;
379 
380     if (hFile != INVALID_HANDLE_VALUE)
381         FindClose(hFile);
382 
383     return bResult;
384 }
385 
386 
387 static
388 BOOL
389 GetProfileName(
390     _In_ PSID pProfileSid,
391     _In_ DWORD dwNameBufferSize,
392     _Out_ PWSTR pszNameBuffer,
393     _Out_ PBOOL pbUnknownProfile)
394 {
395     WCHAR szAccountName[128], szDomainName[128];
396     DWORD dwAccountNameSize, dwDomainNameSize;
397     SID_NAME_USE Use;
398 
399     dwAccountNameSize = ARRAYSIZE(szAccountName);
400     dwDomainNameSize = ARRAYSIZE(szDomainName);
401     if (!LookupAccountSidW(NULL,
402                            pProfileSid,
403                            szAccountName,
404                            &dwAccountNameSize,
405                            szDomainName,
406                            &dwDomainNameSize,
407                            &Use))
408     {
409         /* Unknown account */
410         LoadStringW(hApplet, IDS_USERPROFILE_ACCOUNT_UNKNOWN, pszNameBuffer, dwNameBufferSize);
411         *pbUnknownProfile = TRUE;
412     }
413     else
414     {
415         /* Show only the user accounts */
416         if (Use != SidTypeUser)
417             return FALSE;
418 
419         if (szAccountName[0] == UNICODE_NULL)
420         {
421             /* Deleted account */
422             LoadStringW(hApplet, IDS_USERPROFILE_ACCOUNT_DELETED, pszNameBuffer, dwNameBufferSize);
423         }
424         else
425         {
426             /* Normal account */
427             wsprintf(pszNameBuffer, L"%s\\%s", szDomainName, szAccountName);
428         }
429         *pbUnknownProfile = FALSE;
430     }
431 
432     return TRUE;
433 }
434 
435 
436 static VOID
437 AddUserProfile(
438     _In_ HWND hwndListView,
439     _In_ PSID pProfileSid,
440     _In_ HKEY hProfileKey)
441 {
442     WCHAR szTempProfilePath[MAX_PATH], szProfilePath[MAX_PATH];
443     WCHAR szNameBuffer[256];
444     PPROFILEDATA pProfileData = NULL;
445     DWORD dwProfileData, dwSize, dwType, dwState = 0, dwRefCount = 0;
446     DWORD dwProfilePathLength;
447     PWSTR ptr;
448     INT nId, iItem;
449     LV_ITEM lvi;
450     WIN32_FIND_DATA FindData;
451     HANDLE hFile;
452     SYSTEMTIME SystemTime;
453     ULONGLONG ullProfileSize;
454     BOOL bUnknownProfile;
455     DWORD dwError;
456 
457     /* Get the profile path */
458     dwSize = MAX_PATH * sizeof(WCHAR);
459     dwError = RegQueryValueExW(hProfileKey,
460                                L"ProfileImagePath",
461                                NULL,
462                                &dwType,
463                                (LPBYTE)szTempProfilePath,
464                                &dwSize);
465     if (dwError != ERROR_SUCCESS)
466         return;
467 
468     /* Expand it */
469     ExpandEnvironmentStringsW(szTempProfilePath,
470                               szProfilePath,
471                               MAX_PATH);
472 
473     /* Check if the profile path exists */
474     hFile = FindFirstFileW(szProfilePath, &FindData);
475     if (hFile == INVALID_HANDLE_VALUE)
476         return;
477 
478     FindClose(hFile);
479 
480     /* Get the length of the profile path */
481     dwProfilePathLength = wcslen(szProfilePath);
482 
483     /* Check for the ntuser.dat file */
484     wcscat(szProfilePath, L"\\ntuser.dat");
485     hFile = FindFirstFileW(szProfilePath, &FindData);
486     if (hFile == INVALID_HANDLE_VALUE)
487         return;
488 
489     FindClose(hFile);
490     szProfilePath[dwProfilePathLength] = UNICODE_NULL;
491 
492     /* Get the profile size */
493     ullProfileSize = 0ULL;
494     GetProfileSize(szProfilePath, &ullProfileSize);
495 
496     /* Get the profile name */
497     if (!GetProfileName(pProfileSid,
498                         ARRAYSIZE(szNameBuffer),
499                         szNameBuffer,
500                         &bUnknownProfile))
501         return;
502 
503     /* Get the profile state value */
504     dwSize = sizeof(dwState);
505     if (RegQueryValueExW(hProfileKey,
506                          L"State",
507                          NULL,
508                          &dwType,
509                          (LPBYTE)&dwState,
510                          &dwSize) != ERROR_SUCCESS)
511     {
512         dwState = 0;
513     }
514 
515     /* Get the profile reference counter */
516     dwSize = sizeof(dwRefCount);
517     if (RegQueryValueExW(hProfileKey,
518                          L"RefCount",
519                          NULL,
520                          &dwType,
521                          (LPBYTE)&dwRefCount,
522                          &dwSize) != ERROR_SUCCESS)
523     {
524         dwRefCount = 0;
525     }
526 
527     /* Create and fill the profile data entry */
528     dwProfileData = sizeof(PROFILEDATA) +
529                     ((wcslen(szNameBuffer) + 1) * sizeof(WCHAR)) +
530                     ((wcslen(szProfilePath) + 1) * sizeof(WCHAR));
531     pProfileData = HeapAlloc(GetProcessHeap(),
532                              HEAP_ZERO_MEMORY,
533                              dwProfileData);
534     if (pProfileData == NULL)
535         return;
536 
537     pProfileData->dwRefCount = dwRefCount;
538     pProfileData->dwState = dwState;
539     pProfileData->bUnknownProfile = bUnknownProfile;
540 
541     ptr = (PWSTR)((ULONG_PTR)pProfileData + sizeof(PROFILEDATA));
542     pProfileData->pszFullName = ptr;
543 
544     wcscpy(pProfileData->pszFullName, szNameBuffer);
545 
546     ptr = (PWSTR)((ULONG_PTR)ptr + ((wcslen(pProfileData->pszFullName) + 1) * sizeof(WCHAR)));
547     pProfileData->pszProfilePath = ptr;
548     wcscpy(pProfileData->pszProfilePath, szProfilePath);
549 
550     /* Add the profile and set its name */
551     memset(&lvi, 0x00, sizeof(lvi));
552     lvi.mask = LVIF_TEXT | LVIF_STATE | LVIF_PARAM;
553     lvi.pszText = pProfileData->pszFullName;
554     lvi.state = 0;
555     lvi.lParam = (LPARAM)pProfileData;
556     iItem = ListView_InsertItem(hwndListView, &lvi);
557 
558     /* Set the profile size */
559     StrFormatByteSizeW(ullProfileSize, szNameBuffer, ARRAYSIZE(szNameBuffer) - 1);
560     ListView_SetItemText(hwndListView, iItem, 1, szNameBuffer);
561 
562     /* Set the profile type */
563     if (dwState & 0x0010) // PROFILE_UPDATE_CENTRAL
564         nId = IDS_USERPROFILE_ROAMING;
565     else
566         nId = IDS_USERPROFILE_LOCAL;
567 
568     LoadStringW(hApplet, nId, szNameBuffer, ARRAYSIZE(szNameBuffer));
569 
570     ListView_SetItemText(hwndListView, iItem, 2, szNameBuffer);
571 
572     /* FIXME: Set the profile status */
573     if (dwState & 0x0001) // PROFILE_MANDATORY
574         nId = IDS_USERPROFILE_MANDATORY;
575     else if (dwState & 0x0010) // PROFILE_UPDATE_CENTRAL
576         nId = IDS_USERPROFILE_ROAMING;
577     else
578         nId = IDS_USERPROFILE_LOCAL;
579 
580     LoadStringW(hApplet, nId, szNameBuffer, ARRAYSIZE(szNameBuffer));
581 
582     ListView_SetItemText(hwndListView, iItem, 3, szNameBuffer);
583 
584     /* Set the profile modified time */
585     FileTimeToSystemTime(&FindData.ftLastWriteTime,
586                          &SystemTime);
587 
588     GetDateFormatW(LOCALE_USER_DEFAULT,
589                    DATE_SHORTDATE,
590                    &SystemTime,
591                    NULL,
592                    szNameBuffer,
593                    ARRAYSIZE(szNameBuffer));
594 
595     ListView_SetItemText(hwndListView, iItem, 4, szNameBuffer);
596 }
597 
598 
599 static VOID
600 UpdateButtonState(
601     _In_ HWND hwndDlg,
602     _In_ HWND hwndListView)
603 {
604     LVITEM Item;
605     INT iSelected;
606     BOOL bChange = FALSE;
607     BOOL bCopy = FALSE;
608     BOOL bDelete = FALSE;
609     PPROFILEDATA pProfileData;
610 
611     if (ListView_GetSelectedCount(hwndListView) != 0)
612     {
613         iSelected = ListView_GetNextItem(hwndListView, -1, LVNI_SELECTED);
614         if (iSelected != -1)
615         {
616             ZeroMemory(&Item, sizeof(LVITEM));
617             Item.mask = LVIF_PARAM;
618             Item.iItem = iSelected;
619             if (ListView_GetItem(hwndListView, &Item))
620             {
621                 if (Item.lParam != 0)
622                 {
623                     pProfileData = (PPROFILEDATA)Item.lParam;
624 
625                     if (pProfileData->bUnknownProfile)
626                     {
627                         bDelete = TRUE;
628                         bCopy = FALSE;
629                     }
630                     else
631                     {
632                         bDelete = (pProfileData->dwRefCount == 0);
633                         bCopy = (pProfileData->dwRefCount == 0);
634                     }
635                 }
636             }
637 
638             bChange = TRUE;
639         }
640     }
641 
642     EnableWindow(GetDlgItem(hwndDlg, IDC_USERPROFILE_CHANGE), bChange);
643     EnableWindow(GetDlgItem(hwndDlg, IDC_USERPROFILE_DELETE), bDelete);
644     EnableWindow(GetDlgItem(hwndDlg, IDC_USERPROFILE_COPY), bCopy);
645 }
646 
647 
648 static VOID
649 AddUserProfiles(
650     _In_ HWND hwndDlg,
651     _In_ HWND hwndListView,
652     _In_ BOOL bAdmin)
653 {
654     HKEY hKeyUserProfiles = INVALID_HANDLE_VALUE;
655     HKEY hProfileKey;
656     DWORD dwIndex;
657     WCHAR szProfileSid[64];
658     DWORD dwSidLength;
659     FILETIME ftLastWrite;
660     DWORD dwSize;
661     HANDLE hToken = NULL;
662     PTOKEN_USER pTokenUser = NULL;
663     PSID pProfileSid;
664     PWSTR pszProfileSid;
665 
666     if (!OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken))
667         return;
668 
669     if (GetTokenInformation(hToken, TokenUser, NULL, 0, &dwSize) ||
670         GetLastError() != ERROR_INSUFFICIENT_BUFFER)
671     {
672         goto done;
673     }
674 
675     pTokenUser = HeapAlloc(GetProcessHeap(), 0, dwSize);
676     if (pTokenUser == NULL)
677         goto done;
678 
679     if (!GetTokenInformation(hToken, TokenUser, pTokenUser, dwSize, &dwSize))
680         goto done;
681 
682     if (RegOpenKeyExW(HKEY_LOCAL_MACHINE,
683                       L"Software\\Microsoft\\Windows NT\\CurrentVersion\\ProfileList",
684                       0,
685                       KEY_READ,
686                       &hKeyUserProfiles))
687         goto done;
688 
689     if (bAdmin)
690     {
691         for (dwIndex = 0; ; dwIndex++)
692         {
693             dwSidLength = ARRAYSIZE(szProfileSid);
694             if (RegEnumKeyExW(hKeyUserProfiles,
695                               dwIndex,
696                               szProfileSid,
697                               &dwSidLength,
698                               NULL,
699                               NULL,
700                               NULL,
701                               &ftLastWrite))
702                 break;
703 
704             if (RegOpenKeyExW(hKeyUserProfiles,
705                               szProfileSid,
706                               0,
707                               KEY_READ,
708                               &hProfileKey) == ERROR_SUCCESS)
709             {
710                 if (ConvertStringSidToSid(szProfileSid, &pProfileSid))
711                 {
712                     AddUserProfile(hwndListView,
713                                    pProfileSid,
714                                    hProfileKey);
715                     LocalFree(pProfileSid);
716                 }
717 
718                 RegCloseKey(hProfileKey);
719             }
720         }
721     }
722     else
723     {
724         if (ConvertSidToStringSidW(pTokenUser->User.Sid, &pszProfileSid))
725         {
726             if (RegOpenKeyExW(hKeyUserProfiles,
727                               pszProfileSid,
728                               0,
729                               KEY_READ,
730                               &hProfileKey) == ERROR_SUCCESS)
731             {
732                 AddUserProfile(hwndListView,
733                                pTokenUser->User.Sid,
734                                hProfileKey);
735                 RegCloseKey(hProfileKey);
736             }
737 
738             LocalFree(pszProfileSid);
739         }
740     }
741 
742     if (ListView_GetItemCount(hwndListView) != 0)
743         ListView_SetItemState(hwndListView, 0, LVIS_SELECTED | LVIS_FOCUSED, LVIS_SELECTED | LVIS_FOCUSED);
744 
745     UpdateButtonState(hwndDlg, hwndListView);
746 
747 done:
748     if (hKeyUserProfiles != INVALID_HANDLE_VALUE)
749         RegCloseKey(hKeyUserProfiles);
750 
751     if (pTokenUser != NULL)
752         HeapFree(GetProcessHeap(), 0, pTokenUser);
753 
754     if (hToken != NULL)
755         CloseHandle(hToken);
756 }
757 
758 
759 static VOID
760 OnInitUserProfileDialog(HWND hwndDlg)
761 {
762     BOOL bAdmin;
763 
764     bAdmin = IsUserAdmin();
765 
766     /* Initialize the list view control */
767     SetListViewColumns(GetDlgItem(hwndDlg, IDC_USERPROFILE_LIST));
768 
769     /* Hide the delete and copy buttons for non-admins */
770     ShowWindow(GetDlgItem(hwndDlg, IDC_USERPROFILE_DELETE), bAdmin ? SW_SHOW : SW_HIDE);
771     ShowWindow(GetDlgItem(hwndDlg, IDC_USERPROFILE_COPY), bAdmin ? SW_SHOW : SW_HIDE);
772 
773     /* Add the profiles to the list view */
774     AddUserProfiles(hwndDlg, GetDlgItem(hwndDlg, IDC_USERPROFILE_LIST), bAdmin);
775 }
776 
777 static
778 VOID
779 OnNotify(
780     _In_ HWND hwndDlg,
781     _In_ NMHDR *nmhdr)
782 {
783     LPNMLISTVIEW pNMLV;
784 
785     if (nmhdr->idFrom == IDC_USERACCOUNT_LINK && nmhdr->code == NM_CLICK)
786     {
787         ShellExecuteW(hwndDlg, NULL, L"usrmgr.cpl", NULL, NULL, 0);
788     }
789     else if (nmhdr->idFrom == IDC_USERPROFILE_LIST)
790     {
791         switch(nmhdr->code)
792         {
793             case LVN_ITEMCHANGED:
794                 UpdateButtonState(hwndDlg, nmhdr->hwndFrom);
795                 break;
796 
797             case NM_DBLCLK:
798                 ChangeUserProfileType(hwndDlg);
799                 break;
800 
801             case LVN_DELETEITEM:
802                 pNMLV = (LPNMLISTVIEW)nmhdr;
803                 if (pNMLV->lParam != 0)
804                     HeapFree(GetProcessHeap(), 0, (LPVOID)pNMLV->lParam);
805                 break;
806         }
807     }
808 }
809 
810 
811 /* Property page dialog callback */
812 INT_PTR CALLBACK
813 UserProfileDlgProc(HWND hwndDlg,
814                    UINT uMsg,
815                    WPARAM wParam,
816                    LPARAM lParam)
817 {
818     switch (uMsg)
819     {
820         case WM_INITDIALOG:
821             OnInitUserProfileDialog(hwndDlg);
822             return TRUE;
823 
824         case WM_COMMAND:
825             switch (LOWORD(wParam))
826             {
827                 case IDOK:
828                 case IDCANCEL:
829                     EndDialog(hwndDlg,
830                               LOWORD(wParam));
831                     return TRUE;
832 
833                 case IDC_USERPROFILE_CHANGE:
834                     ChangeUserProfileType(hwndDlg);
835                     break;
836 
837                 case IDC_USERPROFILE_DELETE:
838                     DeleteUserProfile(hwndDlg);
839                     break;
840 
841                 case IDC_USERPROFILE_COPY:
842                     CopyUserProfile(hwndDlg);
843                     break;
844             }
845             break;
846 
847         case WM_NOTIFY:
848             OnNotify(hwndDlg, (NMHDR *)lParam);
849             break;
850     }
851 
852     return FALSE;
853 }
854