![[BACK]](/cvsweb/icons/back.gif){kind=icon}
Return to setup.c CVS log ![[TXT]](/cvsweb/icons/text.gif){kind=icon}
![[DIR]](/cvsweb/icons/dir.gif){kind=icon}
Up to [truecrypt] / truecrypt / setup|
Return to setup.c CVS log | Up to [truecrypt] / truecrypt / setup |
1.1 root 1: /* Copyright (C) 2004 TrueCrypt Team, truecrypt.org
2: This product uses components written by Paul Le Roux <[email protected]> */
3:
4: #include "TCdefs.h"
5: #include <SrRestorePtApi.h>
6:
7: #define MAX_PASSWORD
8:
9: #include "apidrvr.h"
10: #include "dlgcode.h"
11: #include "dismount.h"
12: #include "../common/resource.h"
13:
14: #include "resource.h"
15:
16: #include "dir.h"
17: #include "setup.h"
18:
19: #include <sys\types.h>
20: #include <sys\stat.h>
21:
22: #pragma warning( disable : 4201 )
23: #pragma warning( disable : 4115 )
24:
25: #include <shlobj.h>
26:
27: #pragma warning( default : 4201 )
28: #pragma warning( default : 4115 )
29:
30: char dlg_file_name[TC_MAX_PATH];
31: BOOL bUninstall = FALSE;
32: BOOL bDone = FALSE;
33:
34: HMODULE SystemRestoreDll = 0;
35:
36: BOOL
37: StatDeleteFile (char *lpszFile)
38: {
39: struct __stat64 st;
40:
41: if (_stat64 (lpszFile, &st) == 0)
42: return DeleteFile (lpszFile);
43: else
44: return TRUE;
45: }
46:
47: BOOL
48: StatRemoveDirectory (char *lpszDir)
49: {
50: struct __stat64 st;
51:
52: if (_stat64 (lpszDir, &st) == 0)
53: return RemoveDirectory (lpszDir);
54: else
55: return TRUE;
56: }
57:
58: HRESULT
59: CreateLink (char *lpszPathObj, char *lpszArguments,
60: char *lpszPathLink)
61: {
62: HRESULT hres;
63: IShellLink *psl;
64:
65: /* Get a pointer to the IShellLink interface. */
66: hres = CoCreateInstance (&CLSID_ShellLink, NULL,
67: CLSCTX_INPROC_SERVER, &IID_IShellLink, &psl);
68: if (SUCCEEDED (hres))
69: {
70: IPersistFile *ppf;
71:
72: /* Set the path to the shortcut target, and add the
73: description. */
74: psl->lpVtbl->SetPath (psl, lpszPathObj);
75: psl->lpVtbl->SetArguments (psl, lpszArguments);
76:
77: /* Query IShellLink for the IPersistFile interface for saving
78: the shortcut in persistent storage. */
79: hres = psl->lpVtbl->QueryInterface (psl, &IID_IPersistFile,
80: &ppf);
81:
82: if (SUCCEEDED (hres))
83: {
84: WORD wsz[TC_MAX_PATH];
85:
86: /* Ensure that the string is ANSI. */
87: MultiByteToWideChar (CP_ACP, 0, lpszPathLink, -1,
88: wsz, TC_MAX_PATH);
89:
90: /* Save the link by calling IPersistFile::Save. */
91: hres = ppf->lpVtbl->Save (ppf, wsz, TRUE);
92: ppf->lpVtbl->Release (ppf);
93: }
94: psl->lpVtbl->Release (psl);
95: }
96: return hres;
97: }
98:
99: void
100: GetProgramPath (HWND hwndDlg, char *path)
101: {
102: ITEMIDLIST *i;
103: HRESULT res;
104:
105: if (nCurrentOS == WIN_NT && IsDlgButtonChecked (hwndDlg, IDC_ALL_USERS))
106: res = SHGetSpecialFolderLocation (hwndDlg, CSIDL_COMMON_PROGRAMS, &i);
107: else
108: res = SHGetSpecialFolderLocation (hwndDlg, CSIDL_PROGRAMS, &i);
109:
110: SHGetPathFromIDList (i, path);
111: }
112:
113:
114: void
115: StatusMessage (HWND hwndDlg, char *head, char *txt)
116: {
117: char szTmp[TC_MAX_PATH];
118: sprintf (szTmp, head, txt);
119: SendMessage (GetDlgItem (hwndDlg, IDC_FILES), LB_ADDSTRING, 0, (LPARAM) szTmp);
120:
121: SendDlgItemMessage (hwndDlg, IDC_FILES, LB_SETTOPINDEX,
122: SendDlgItemMessage (hwndDlg, IDC_FILES, LB_GETCOUNT, 0, 0) - 1, 0);
123: }
124:
125: void
126: RegMessage (HWND hwndDlg, char *txt)
127: {
128: StatusMessage (hwndDlg, "Adding registry entry %s", txt);
129: }
130:
131: void
132: CopyMessage (HWND hwndDlg, char *txt)
133: {
134: StatusMessage (hwndDlg, "Copying %s", txt);
135: }
136:
137: void
138: RemoveMessage (HWND hwndDlg, char *txt)
139: {
140: StatusMessage (hwndDlg, "Removing %s", txt);
141: }
142:
143: void
144: ServiceMessage (HWND hwndDlg, char *txt)
145: {
146: StatusMessage (hwndDlg, "Service %s", txt);
147: }
148:
149: void
150: IconMessage (HWND hwndDlg, char *txt)
151: {
152: StatusMessage (hwndDlg, "Adding icon %s", txt);
153: }
154:
155: int CALLBACK
156: BrowseCallbackProc(HWND hwnd,UINT uMsg,LPARAM lp, LPARAM pData)
157: {
158: switch(uMsg) {
159: case BFFM_INITIALIZED:
160: {
161: /* WParam is TRUE since we are passing a path.
162: It would be FALSE if we were passing a pidl. */
163: SendMessage(hwnd,BFFM_SETSELECTION,TRUE,(LPARAM)pData);
164: break;
165: }
166:
167: case BFFM_SELCHANGED:
168: {
169: char szDir[TC_MAX_PATH];
170:
171: /* Set the status window to the currently selected path. */
172: if (SHGetPathFromIDList((LPITEMIDLIST) lp ,szDir))
173: {
174: SendMessage(hwnd,BFFM_SETSTATUSTEXT,0,(LPARAM)szDir);
175: }
176: break;
177: }
178:
179: default:
180: break;
181: }
182:
183: return 0;
184: }
185:
186: BOOL
187: BrowseFiles2 (HWND hwndDlg, char* lpszTitle, char* lpszFileName)
188: {
189: BROWSEINFO bi;
190: LPITEMIDLIST pidl;
191: LPMALLOC pMalloc;
192: BOOL bOK = FALSE;
193:
194: if (SUCCEEDED(SHGetMalloc(&pMalloc)))
195: {
196: ZeroMemory(&bi,sizeof(bi));
197: bi.hwndOwner = hwndDlg;
198: bi.pszDisplayName = 0;
199: bi.lpszTitle = lpszTitle;
200: bi.pidlRoot = 0;
201: bi.ulFlags = BIF_RETURNONLYFSDIRS | BIF_STATUSTEXT /*| BIF_EDITBOX*/;
202: bi.lpfn = BrowseCallbackProc;
203: bi.lParam = (LPARAM)lpszFileName;
204:
205: pidl = SHBrowseForFolder(&bi);
206: if (pidl!=NULL)
207: {
208: if (SHGetPathFromIDList(pidl,lpszFileName)==TRUE)
209: {
210: bOK = TRUE;
211: }
212:
213: pMalloc->lpVtbl->Free(pMalloc,pidl);
214: pMalloc->lpVtbl->Release(pMalloc);
215: }
216: }
217:
218: return bOK;
219: }
220:
221:
222: void
223: LoadLicense (HWND hwndDlg)
224: {
225: FILE *fp;
226:
227: fp = fopen ("Setup Files\\license.txt", "rb");
228:
229: if (fp == NULL)
230: return;
231: else
232: {
233: long x;
234:
235: fseek (fp, 0, SEEK_END);
236: x = ftell (fp);
237: rewind (fp);
238:
239: if (x > 0)
240: {
241: char *tmp = malloc (x + 1);
242: long z;
243:
244: if (tmp == NULL)
245: goto exit;
246: z = (long) fread (tmp, 1, x, fp);
247: if (z != x)
248: {
249: free (tmp);
250: goto exit;
251: }
252: else
253: {
254: int i;
255: tmp[x] = 0;
256:
257: //// Remove single CRLFs
258: //for (i = 0; i < x - 3; i++)
259: //{
260: // if (tmp[i] == 0xd && tmp[i+2] == 0xd)
261: // i += 4;
262:
263: // if (tmp[i] == 0xd && tmp[i+2] != 0xd)
264: // {
265: // tmp[i] = tmp[i+1] = ' ';
266: // }
267: //}
268:
269: SendMessage (GetDlgItem (hwndDlg, IDC_LICENSE), WM_SETFONT, (WPARAM) hFixedFont, (LPARAM) 0);
270: SetWindowText (GetDlgItem (hwndDlg, IDC_LICENSE), tmp);
271:
272: free (tmp);
273: }
274: }
275: }
276:
277: exit:
278: fclose (fp);
279: }
280:
281: BOOL
282: DoFilesInstall (HWND hwndDlg, char *szDestDir, BOOL bUninstallSupport)
283: {
284:
285:
286: char *szFiles[]=
287: {
288: "ATrueCrypt.exe", "ATrueCrypt Format.exe",
289: "Alicense.txt", "ATrueCrypt User Guide.pdf",
290: "WTrueCrypt Setup.exe", "STrueCryptService.exe", "Dtruecrypt.sys",
291: "Itruecrypt.vxd"
292: };
293:
294: char szTmp[TC_MAX_PATH];
295: BOOL bOK = TRUE;
296: int i;
297:
298: if (bUninstall == TRUE)
299: bUninstallSupport = FALSE;
300:
301: for (i = 0; i < sizeof (szFiles) / sizeof (szFiles[0]); i++)
302: {
303: BOOL bResult, bSlash;
304: char szDir[TC_MAX_PATH];
305: int x;
306:
307: if (bUninstallSupport == FALSE && strstr (szFiles[i], "TrueCrypt Setup") != 0)
308: continue;
309:
310: if (*szFiles[i] == 'A')
311: strcpy (szDir, szDestDir);
312: else if (*szFiles[i] == 'S')
313: GetSystemDirectory (szDir, sizeof (szDir));
314: else if (*szFiles[i] == 'I')
315: {
316: GetSystemDirectory (szDir, sizeof (szDir));
317:
318: x = strlen (szDestDir);
319: if (szDestDir[x - 1] == '\\')
320: bSlash = TRUE;
321: else
322: bSlash = FALSE;
323:
324: if (bSlash == FALSE)
325: strcat (szDir, "\\");
326:
327: strcat (szDir, "IOSUBSYS");
328: }
329: else if (*szFiles[i] == 'D')
330: {
331: GetSystemDirectory (szDir, sizeof (szDir));
332:
333: x = strlen (szDestDir);
334: if (szDestDir[x - 1] == '\\')
335: bSlash = TRUE;
336: else
337: bSlash = FALSE;
338:
339: if (bSlash == FALSE)
340: strcat (szDir, "\\");
341:
342: strcat (szDir, "Drivers");
343: }
344: else if (*szFiles[i] == 'W')
345: GetWindowsDirectory (szDir, sizeof (szDir));
346:
347: x = strlen (szDestDir);
348: if (szDestDir[x - 1] == '\\')
349: bSlash = TRUE;
350: else
351: bSlash = FALSE;
352:
353: if (bSlash == FALSE)
354: strcat (szDir, "\\");
355:
356: if ((*szFiles[i] == 'D' || *szFiles[i] == 'S') && nCurrentOS != WIN_NT)
357: continue;
358:
359: if (*szFiles[i] == 'I' && nCurrentOS == WIN_NT)
360: continue;
361:
362: sprintf (szTmp, "%s%s", szDir, szFiles[i] + 1);
363:
364: if (bUninstall == FALSE)
365: CopyMessage (hwndDlg, szTmp);
366: else
367: RemoveMessage (hwndDlg, szTmp);
368:
369: if (bUninstall == FALSE)
370: {
371: bResult = CopyFile (szFiles[i] + 1, szTmp, FALSE);
372: if (!bResult)
373: {
374: char s[256];
375: sprintf (s, "Setup Files\\%s", szFiles[i] + 1);
376: bResult = CopyFile (s, szTmp, FALSE);
377: }
378: }
379: else
380: {
381: bResult = StatDeleteFile (szTmp);
382: }
383:
384: if (bResult == FALSE)
385: {
386: LPVOID lpMsgBuf;
387: DWORD dwError = GetLastError ();
388: char szTmp2[700];
389:
390: FormatMessage (
391: FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
392: NULL,
393: dwError,
394: MAKELANGID (LANG_NEUTRAL, SUBLANG_DEFAULT), /* Default language */
395: (char *) &lpMsgBuf,
396: 0,
397: NULL
398: );
399:
400:
401: if (bUninstall == FALSE)
402: sprintf (szTmp2, "The installation of '%s' has failed. %s Do you want to continue with the Install?",
403: szTmp, lpMsgBuf);
404: else
405: sprintf (szTmp2, "The uninstallation of '%s' has failed. %s Do you want to continue with the Uninstall?",
406: szTmp, lpMsgBuf);
407:
408: LocalFree (lpMsgBuf);
409:
410: if (MessageBox (hwndDlg, szTmp2, lpszTitle, MB_YESNO | MB_ICONHAND) != IDYES)
411: return FALSE;
412: }
413:
414: }
415:
416: return bOK;
417: }
418:
419: BOOL
420: DoRegInstall (HWND hwndDlg, char *szDestDir, BOOL bInstallType, BOOL bUninstallSupport)
421: {
422: char szDir[TC_MAX_PATH], *key;
423: HKEY hkey = 0;
424: BOOL bSlash, bOK = FALSE;
425: DWORD dw;
426: int x;
427:
428: strcpy (szDir, szDestDir);
429: x = strlen (szDestDir);
430: if (szDestDir[x - 1] == '\\')
431: bSlash = TRUE;
432: else
433: bSlash = FALSE;
434:
435: if (bSlash == FALSE)
436: strcat (szDir, "\\");
437:
438: if (nCurrentOS == WIN_NT)
439: {
440: /* 9/9/99 FIX This code should no longer be needed as we use
441: the "services" api to install the driver now, rather than
442: setting the registry by hand */
443:
444: /* Install device driver */
445: key = "SYSTEM\\CurrentControlSet\\Services\\truecrypt";
446: RegMessage (hwndDlg, key);
447: if (RegCreateKeyEx (HKEY_LOCAL_MACHINE,
448: key,
449: 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkey, &dw) != ERROR_SUCCESS)
450: goto error;
451:
452: dw = 1;
453: if (RegSetValueEx (hkey, "Type", 0, REG_DWORD, (BYTE *) & dw, 4) != ERROR_SUCCESS)
454: goto error;
455:
456: dw = 1;
457: if (RegSetValueEx (hkey, "Start", 0, REG_DWORD, (BYTE *) & dw, 4) != ERROR_SUCCESS)
458: goto error;
459:
460: dw = 1;
461: if (RegSetValueEx (hkey, "ErrorControl", 0, REG_DWORD, (BYTE *) & dw, 4) != ERROR_SUCCESS)
462: goto error;
463:
464: if (RegSetValueEx (hkey, "Group", 0, REG_SZ, (BYTE *) "Primary disk", 13) != ERROR_SUCCESS)
465: goto error;
466:
467: RegCloseKey (hkey);
468: hkey = 0;
469: }
470:
471: if (bInstallType == TRUE)
472: {
473: char szTmp[TC_MAX_PATH];
474:
475: key = "SOFTWARE\\Classes\\TrueCryptVolume";
476: RegMessage (hwndDlg, key);
477: if (RegCreateKeyEx (HKEY_LOCAL_MACHINE,
478: key,
479: 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkey, &dw) != ERROR_SUCCESS)
480: goto error;
481:
482: sprintf (szTmp, "TrueCrypt Volume", szDir);
483: if (RegSetValueEx (hkey, "", 0, REG_SZ, (BYTE *) szTmp, strlen (szTmp) + 1) != ERROR_SUCCESS)
484: goto error;
485:
486: RegCloseKey (hkey);
487: hkey = 0;
488:
489: key = "SOFTWARE\\Classes\\TrueCryptVolume\\DefaultIcon";
490: RegMessage (hwndDlg, key);
491: if (RegCreateKeyEx (HKEY_LOCAL_MACHINE,
492: key,
493: 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkey, &dw) != ERROR_SUCCESS)
494: goto error;
495:
496: sprintf (szTmp, "%sTrueCrypt.exe,1", szDir);
497: if (RegSetValueEx (hkey, "", 0, REG_SZ, (BYTE *) szTmp, strlen (szTmp) + 1) != ERROR_SUCCESS)
498: goto error;
499:
500: RegCloseKey (hkey);
501: hkey = 0;
502:
503: key = "SOFTWARE\\Classes\\TrueCryptVolume\\Shell\\open\\command";
504: RegMessage (hwndDlg, key);
505: if (RegCreateKeyEx (HKEY_LOCAL_MACHINE,
506: key,
507: 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkey, &dw) != ERROR_SUCCESS)
508: goto error;
509:
510: sprintf (szTmp, "\"%sTrueCrypt.exe\" /v \"%%1\"", szDir );
511: if (RegSetValueEx (hkey, "", 0, REG_SZ, (BYTE *) szTmp, strlen (szTmp) + 1) != ERROR_SUCCESS)
512: goto error;
513:
514: RegCloseKey (hkey);
515: hkey = 0;
516:
517: key = "SOFTWARE\\Classes\\.tc";
518: RegMessage (hwndDlg, key);
519: if (RegCreateKeyEx (HKEY_LOCAL_MACHINE,
520: key,
521: 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkey, &dw) != ERROR_SUCCESS)
522: goto error;
523:
524: strcpy (szTmp, "TrueCryptVolume");
525: if (RegSetValueEx (hkey, "", 0, REG_SZ, (BYTE *) szTmp, strlen (szTmp) + 1) != ERROR_SUCCESS)
526: goto error;
527: }
528:
529:
530: if (bUninstallSupport == TRUE)
531: {
532: char szTmp[TC_MAX_PATH];
533:
534: key = "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\TrueCrypt";
535: RegMessage (hwndDlg, key);
536: if (RegCreateKeyEx (HKEY_LOCAL_MACHINE,
537: key,
538: 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkey, &dw) != ERROR_SUCCESS)
539: goto error;
540:
541: GetWindowsDirectory (szDir, sizeof (szDir));
542:
543: x = strlen (szDir);
544: if (szDir[x - 1] == '\\')
545: bSlash = TRUE;
546: else
547: bSlash = FALSE;
548:
549: if (bSlash == FALSE)
550: strcat (szDir, "\\");
551:
552: sprintf (szTmp, "%sTrueCrypt Setup.exe /u", szDir);
553: if (RegSetValueEx (hkey, "UninstallString", 0, REG_SZ, (BYTE *) szTmp, strlen (szTmp) + 1) != ERROR_SUCCESS)
554: goto error;
555:
556: strcpy (szTmp, "TrueCrypt");
557: if (RegSetValueEx (hkey, "DisplayName", 0, REG_SZ, (BYTE *) szTmp, strlen (szTmp) + 1) != ERROR_SUCCESS)
558: goto error;
559: }
560:
561: bOK = TRUE;
562:
563: error:
564: if (hkey != 0)
565: RegCloseKey (hkey);
566:
567: if (bOK == FALSE)
568: {
569: handleWin32Error (hwndDlg);
570: MessageBox (hwndDlg, "The installation of the registry entries has failed", lpszTitle, MB_ICONHAND);
571: }
572:
573: return bOK;
574: }
575:
576: BOOL
577: DoRegUninstall (HWND hwndDlg)
578: {
579: BOOL bOK = FALSE;
580:
581: StatusMessage (hwndDlg, "%s", "Removing registry entries");
582:
583: RegDeleteKey (HKEY_LOCAL_MACHINE, "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\TrueCrypt");
584: RegDeleteKey (HKEY_LOCAL_MACHINE, "SOFTWARE\\Classes\\TrueCryptVolume\\Shell\\open\\command");
585: RegDeleteKey (HKEY_LOCAL_MACHINE, "SOFTWARE\\Classes\\TrueCryptVolume\\Shell\\open");
586: RegDeleteKey (HKEY_LOCAL_MACHINE, "SOFTWARE\\Classes\\TrueCryptVolume\\Shell");
587: RegDeleteKey (HKEY_LOCAL_MACHINE, "SOFTWARE\\Classes\\TrueCryptVolume\\DefaultIcon");
588: RegDeleteKey (HKEY_LOCAL_MACHINE, "SOFTWARE\\Classes\\TrueCryptVolume");
589: RegDeleteKey (HKEY_CURRENT_USER, "SOFTWARE\\TrueCrypt");
590: if (RegDeleteKey (HKEY_LOCAL_MACHINE, "SOFTWARE\\Classes\\.tc") != ERROR_SUCCESS)
591: goto error;
592:
593: bOK = TRUE;
594:
595: error:
596:
597: if (bOK == FALSE && GetLastError ()!= ERROR_NO_TOKEN && GetLastError ()!= ERROR_FILE_NOT_FOUND
598: && GetLastError ()!= ERROR_PATH_NOT_FOUND)
599: {
600: handleWin32Error (hwndDlg);
601: MessageBox (hwndDlg, "The uninstallation of the registry entries has failed", lpszTitle, MB_ICONHAND);
602: }
603: else
604: bOK = TRUE;
605:
606: return bOK;
607:
608: }
609:
610: BOOL
611: DoServiceUninstall (HWND hwndDlg, char *lpszService)
612: {
613: SC_HANDLE hManager, hService = NULL;
614: BOOL bOK = FALSE, bRet;
615: SERVICE_STATUS status;
616: char szTmp[128];
617: int x;
618:
619: if (nCurrentOS != WIN_NT)
620: return TRUE;
621: else
622: memset (&status, 0, sizeof (status)); /* Keep VC6 quiet */
623:
624: hManager = OpenSCManager (NULL, NULL, SC_MANAGER_ALL_ACCESS);
625: if (hManager == NULL)
626: goto error;
627:
628: hService = OpenService (hManager, lpszService, SERVICE_ALL_ACCESS);
629: if (hService == NULL)
630: goto error;
631:
632: sprintf (szTmp, "stopping %s", lpszService);
633: ServiceMessage (hwndDlg, szTmp);
634:
635: #define WAIT_PERIOD 3
636:
637: for (x = 0; x < WAIT_PERIOD; x++)
638: {
639: bRet = QueryServiceStatus (hService, &status);
640: if (bRet != TRUE)
641: goto error;
642:
643: if (status.dwCurrentState != SERVICE_START_PENDING &&
644: status.dwCurrentState != SERVICE_STOP_PENDING &&
645: status.dwCurrentState != SERVICE_CONTINUE_PENDING)
646: break;
647:
648: Sleep (1000);
649: }
650:
651: if (status.dwCurrentState != SERVICE_STOPPED)
652: {
653: bRet = ControlService (hService, SERVICE_CONTROL_STOP, &status);
654: if (bRet == FALSE)
655: goto try_delete;
656:
657: for (x = 0; x < WAIT_PERIOD; x++)
658: {
659: bRet = QueryServiceStatus (hService, &status);
660: if (bRet != TRUE)
661: goto error;
662:
663: if (status.dwCurrentState != SERVICE_START_PENDING &&
664: status.dwCurrentState != SERVICE_STOP_PENDING &&
665: status.dwCurrentState != SERVICE_CONTINUE_PENDING)
666: break;
667:
668: Sleep (1000);
669: }
670:
671: if (status.dwCurrentState != SERVICE_STOPPED && status.dwCurrentState != SERVICE_STOP_PENDING)
672: goto error;
673: }
674:
675: try_delete:
676: sprintf (szTmp, "deleting %s", lpszService);
677: ServiceMessage (hwndDlg, szTmp);
678:
679: if (hService != NULL)
680: CloseServiceHandle (hService);
681:
682: if (hManager != NULL)
683: CloseServiceHandle (hManager);
684:
685: hManager = OpenSCManager (NULL, NULL, SC_MANAGER_ALL_ACCESS);
686: if (hManager == NULL)
687: goto error;
688:
689: hService = OpenService (hManager, lpszService, SERVICE_ALL_ACCESS);
690: if (hService == NULL)
691: goto error;
692:
693: bRet = DeleteService (hService);
694: if (bRet == FALSE)
695: goto error;
696:
697: bOK = TRUE;
698:
699: error:
700: if (bOK == FALSE && GetLastError ()!= ERROR_SERVICE_DOES_NOT_EXIST)
701: {
702: handleWin32Error (hwndDlg);
703: MessageBox (hwndDlg, "The uninstallation of the device driver has failed", lpszTitle, MB_ICONHAND);
704: }
705: else
706: bOK = TRUE;
707:
708: if (hService != NULL)
709: CloseServiceHandle (hService);
710:
711: if (hManager != NULL)
712: CloseServiceHandle (hManager);
713:
714: return bOK;
715: }
716:
717: BOOL
718: DoDriverUnload (HWND hwndDlg)
719: {
720: BOOL bOK = TRUE;
721: int status;
722:
723: status = DriverAttach ();
724: if (status != 0)
725: {
726: if (status == ERR_OS_ERROR && GetLastError ()!= ERROR_FILE_NOT_FOUND)
727: {
728: handleWin32Error (hwndDlg);
729: AbortProcess (IDS_NODRIVER);
730: }
731:
732: if (status != ERR_OS_ERROR)
733: {
734: handleError (NULL, status);
735: AbortProcess (IDS_NODRIVER);
736: }
737: }
738:
739: if (hDriver != INVALID_HANDLE_VALUE)
740: {
741: DWORD dwError;
742: int x;
743:
744:
745: if (nCurrentOS == WIN_NT)
746: {
747: ServiceMessage (hwndDlg, "dismounting any mounted volumes");
748:
749: if (UnmountAllVolumes (hwndDlg, &dwError, &x) == FALSE)
750: {
751: bOK = FALSE;
752: MessageBox (hwndDlg, "Volumes are still mounted; all volumes must be dismounted before uninstallation can continue", lpszTitle, MB_ICONHAND);
753: }
754: }
755: else
756: {
757:
758: MOUNT_LIST_STRUCT driver;
759: DWORD dwResult;
760: BOOL bResult;
761:
762: bResult = DeviceIoControl (hDriver, MOUNT_LIST, &driver, sizeof (driver), &driver,
763: sizeof (driver), &dwResult, NULL);
764:
765: if (bResult == TRUE)
766: {
767: if (driver.ulMountedDrives != 0)
768: {
769: bOK = FALSE;
770: MessageBox (hwndDlg, "Volumes are still mounted; all volumes must be dismounted before uninstallation can continue", lpszTitle, MB_ICONHAND);
771: }
772: }
773: else
774: {
775: bOK = FALSE;
776: handleWin32Error (hwndDlg);
777: }
778: }
779:
780: CloseHandle (hDriver);
781: hDriver = INVALID_HANDLE_VALUE;
782:
783: }
784:
785: return bOK;
786: }
787:
788: BOOL
789: DoServiceInstall (HWND hwndDlg)
790: {
791: BOOL bOK = FALSE;
792:
793: if (nCurrentOS != WIN_NT)
794: return TRUE;
795:
796: ServiceMessage (hwndDlg, "installing TrueCryptService");
797: ServiceMessage (hwndDlg, "starting TrueCryptService");
798:
799: if (CheckService ()== FALSE)
800: goto error;
801:
802: bOK = TRUE;
803:
804: error:
805: if (bOK == FALSE)
806: {
807: MessageBox ((HWND) hwndDlg, "The installation of the service has failed", lpszTitle, MB_ICONHAND);
808: }
809:
810: return bOK;
811: }
812:
813: BOOL
814: DoDriverInstall (HWND hwndDlg)
815: {
816: SC_HANDLE hManager, hService = NULL;
817: BOOL bOK = FALSE, bRet, bSlash;
818: char szDir[TC_MAX_PATH];
819: int x;
820:
821: if (nCurrentOS != WIN_NT)
822: return TRUE;
823:
824: hManager = OpenSCManager (NULL, NULL, SC_MANAGER_ALL_ACCESS);
825: if (hManager == NULL)
826: goto error;
827:
828: GetSystemDirectory (szDir, sizeof (szDir));
829:
830: x = strlen (szDir);
831: if (szDir[x - 1] == '\\')
832: bSlash = TRUE;
833: else
834: bSlash = FALSE;
835:
836: if (bSlash == FALSE)
837: strcat (szDir, "\\");
838:
839: strcat (szDir, "Drivers\\truecrypt.sys");
840:
841: ServiceMessage (hwndDlg, "installing TrueCrypt driver service");
842:
843: hService = CreateService (hManager, "truecrypt", "truecrypt",
844: SERVICE_ALL_ACCESS, SERVICE_KERNEL_DRIVER, SERVICE_AUTO_START, SERVICE_ERROR_NORMAL,
845: szDir, NULL, NULL, NULL, NULL, NULL
846: );
847: if (hService == NULL)
848: goto error;
849: else
850: CloseServiceHandle (hService);
851:
852: hService = OpenService (hManager, "truecrypt", SERVICE_ALL_ACCESS);
853: if (hService == NULL)
854: goto error;
855:
856: ServiceMessage (hwndDlg, "starting TrueCrypt driver service");
857:
858: bRet = StartService (hService, 0, NULL);
859: if (bRet == FALSE)
860: goto error;
861:
862: bOK = TRUE;
863:
864: error:
865: if (bOK == FALSE && GetLastError ()!= ERROR_SERVICE_ALREADY_RUNNING)
866: {
867: handleWin32Error (hwndDlg);
868: MessageBox (hwndDlg, "The installation of the device driver has failed", lpszTitle, MB_ICONHAND);
869: }
870: else
871: bOK = TRUE;
872:
873: if (hService != NULL)
874: CloseServiceHandle (hService);
875:
876: if (hManager != NULL)
877: CloseServiceHandle (hManager);
878:
879: return bOK;
880: }
881:
882: BOOL
883: DoShortcutsUninstall (HWND hwndDlg, char *szDestDir)
884: {
885: char szLinkDir[TC_MAX_PATH], szDir[TC_MAX_PATH];
886: char szTmp[TC_MAX_PATH], szTmp2[TC_MAX_PATH];
887: BOOL bSlash, bOK = FALSE;
888: HRESULT hOle;
889: int x;
890: BOOL allUsers = FALSE;
891:
892: hOle = OleInitialize (NULL);
893:
894: // User start menu
895: SHGetSpecialFolderPath (hwndDlg, szLinkDir, CSIDL_PROGRAMS, 0);
896: x = strlen (szLinkDir);
897: if (szLinkDir[x - 1] == '\\')
898: bSlash = TRUE;
899: else
900: bSlash = FALSE;
901:
902: if (bSlash == FALSE)
903: strcat (szLinkDir, "\\");
904:
905: strcat (szLinkDir, "TrueCrypt");
906:
907: // Global start menu
908: if (nCurrentOS == WIN_NT)
909: {
910: struct _stat st;
911: char path[TC_MAX_PATH];
912:
913: SHGetSpecialFolderPath (hwndDlg, path, CSIDL_COMMON_PROGRAMS, 0);
914: strcat (path, "\\TrueCrypt");
915:
916: if (_stat (path, &st) == 0)
917: {
918: strcpy (szLinkDir, path);
919: allUsers = TRUE;
920: }
921: }
922:
923: // Start menu entries
924: sprintf (szTmp2, "%s%s", szLinkDir, "\\TrueCrypt.lnk");
925: RemoveMessage (hwndDlg, szTmp2);
926: if (StatDeleteFile (szTmp2) == FALSE)
927: goto error;
928:
929: sprintf (szTmp2, "%s%s", szLinkDir, "\\TrueCrypt User's Guide.lnk");
930: RemoveMessage (hwndDlg, szTmp2);
931: if (StatDeleteFile (szTmp2) == FALSE)
932: goto error;
933:
934: GetWindowsDirectory (szDir, sizeof (szDir));
935: sprintf (szTmp2, "%s%s", szLinkDir, "\\Uninstall TrueCrypt.lnk");
936: RemoveMessage (hwndDlg, szTmp2);
937: if (StatDeleteFile (szTmp2) == FALSE)
938: goto error;
939:
940: // Start menu group
941: RemoveMessage ((HWND) hwndDlg, szLinkDir);
942: if (StatRemoveDirectory (szLinkDir) == FALSE)
943: {
944: handleWin32Error ((HWND) hwndDlg);
945: goto error;
946: }
947:
948: // Desktop icon
949:
950: if (allUsers)
951: SHGetSpecialFolderPath (hwndDlg, szLinkDir, CSIDL_COMMON_DESKTOPDIRECTORY, 0);
952: else
953: SHGetSpecialFolderPath (hwndDlg, szLinkDir, CSIDL_DESKTOPDIRECTORY, 0);
954:
955: sprintf (szTmp2, "%s%s", szLinkDir, "\\TrueCrypt.lnk");
956:
957: RemoveMessage (hwndDlg, szTmp2);
958: if (StatDeleteFile (szTmp2) == FALSE)
959: goto error;
960:
961: bOK = TRUE;
962:
963: error:
964: OleUninitialize ();
965:
966: return bOK;
967: }
968:
969: BOOL
970: DoShortcutsInstall (HWND hwndDlg, char *szDestDir, BOOL bProgGroup, BOOL bDesktopIcon)
971: {
972: char szLinkDir[TC_MAX_PATH], szDir[TC_MAX_PATH];
973: char szTmp[TC_MAX_PATH], szTmp2[TC_MAX_PATH];
974: BOOL bSlash, bOK = FALSE;
975: HRESULT hOle;
976: int x;
977:
978: if (bProgGroup == FALSE && bDesktopIcon == FALSE)
979: return TRUE;
980:
981: hOle = OleInitialize (NULL);
982:
983: GetProgramPath (hwndDlg, szLinkDir);
984:
985: x = strlen (szLinkDir);
986: if (szLinkDir[x - 1] == '\\')
987: bSlash = TRUE;
988: else
989: bSlash = FALSE;
990:
991: if (bSlash == FALSE)
992: strcat (szLinkDir, "\\");
993:
994: strcat (szLinkDir, "TrueCrypt");
995:
996: strcpy (szDir, szDestDir);
997: x = strlen (szDestDir);
998: if (szDestDir[x - 1] == '\\')
999: bSlash = TRUE;
1000: else
1001: bSlash = FALSE;
1002:
1003: if (bSlash == FALSE)
1004: strcat (szDir, "\\");
1005:
1006: if (bProgGroup)
1007: {
1008: if (mkfulldir (szLinkDir, TRUE) != 0)
1009: {
1010: char szTmp[TC_MAX_PATH];
1011: int x;
1012:
1013: //sprintf (szTmp, "The program folder '%s' does not exist. Do you want to create this folder?", szLinkDir);
1014: //x = MessageBox (hwndDlg, szTmp, lpszTitle, MB_ICONQUESTION | MB_YESNO);
1015: //if (x == IDNO)
1016: //{
1017: // goto error;
1018: //}
1019:
1020: if (mkfulldir (szLinkDir, FALSE) != 0)
1021: {
1022: handleWin32Error (hwndDlg);
1023: sprintf (szTmp, "The folder '%s' could not be created", szLinkDir);
1024: MessageBox (hwndDlg, szTmp, lpszTitle, MB_ICONHAND);
1025: goto error;
1026: }
1027: }
1028:
1029: sprintf (szTmp, "%s%s", szDir, "TrueCrypt.exe");
1030: sprintf (szTmp2, "%s%s", szLinkDir, "\\TrueCrypt.lnk");
1031:
1032: IconMessage (hwndDlg, szTmp2);
1033: if (CreateLink (szTmp, "", szTmp2) != S_OK)
1034: goto error;
1035:
1036:
1037: sprintf (szTmp, "%s%s", szDir, "TrueCrypt User Guide.pdf");
1038: sprintf (szTmp2, "%s%s", szLinkDir, "\\TrueCrypt User's Guide.lnk");
1039:
1040: IconMessage (hwndDlg, szTmp2);
1041: if (CreateLink (szTmp, "", szTmp2) != S_OK)
1042: goto error;
1043:
1044: GetWindowsDirectory (szDir, sizeof (szDir));
1045: x = strlen (szDir);
1046: if (szDir[x - 1] == '\\')
1047: bSlash = TRUE;
1048: else
1049: bSlash = FALSE;
1050:
1051: if (bSlash == FALSE)
1052: strcat (szDir, "\\");
1053:
1054: sprintf (szTmp, "%s%s", szDir, "TrueCrypt Setup.exe");
1055: sprintf (szTmp2, "%s%s", szLinkDir, "\\Uninstall TrueCrypt.lnk");
1056:
1057: IconMessage (hwndDlg, szTmp2);
1058: if (CreateLink (szTmp, "/u", szTmp2) != S_OK)
1059: goto error;
1060:
1061: }
1062:
1063: if (bDesktopIcon)
1064: {
1065: strcpy (szDir, szDestDir);
1066: x = strlen (szDestDir);
1067: if (szDestDir[x - 1] == '\\')
1068: bSlash = TRUE;
1069: else
1070: bSlash = FALSE;
1071:
1072: if (bSlash == FALSE)
1073: strcat (szDir, "\\");
1074:
1075: if (nCurrentOS == WIN_NT && IsDlgButtonChecked (hwndDlg, IDC_ALL_USERS))
1076: SHGetSpecialFolderPath (hwndDlg, szLinkDir, CSIDL_COMMON_DESKTOPDIRECTORY, 0);
1077: else
1078: SHGetSpecialFolderPath (hwndDlg, szLinkDir, CSIDL_DESKTOPDIRECTORY, 0);
1079:
1080: sprintf (szTmp, "%s%s", szDir, "TrueCrypt.exe");
1081: sprintf (szTmp2, "%s%s", szLinkDir, "\\TrueCrypt.lnk");
1082:
1083: IconMessage (hwndDlg, szTmp2);
1084:
1085: if (CreateLink (szTmp, "", szTmp2) != S_OK)
1086: goto error;
1087: }
1088:
1089: bOK = TRUE;
1090:
1091: error:
1092: OleUninitialize ();
1093:
1094: return bOK;
1095: }
1096:
1097:
1098: void
1099: RebootPrompt (HWND hwndDlg, BOOL bOK)
1100: {
1101: if (bOK == TRUE)
1102: {
1103: SetWindowText (GetDlgItem ((HWND) hwndDlg, IDOK), "E&xit");
1104:
1105: EnableWindow (GetDlgItem ((HWND) hwndDlg, IDCANCEL), FALSE);
1106:
1107: bDone = TRUE;
1108:
1109: if (nCurrentOS == WIN_NT)
1110: {
1111: if (bUninstall == FALSE)
1112: MessageBox ((HWND) hwndDlg, "TrueCrypt has been successfuly installed.", lpszTitle, MB_ICONINFORMATION);
1113: else
1114: MessageBox ((HWND) hwndDlg, "TrueCrypt has been successfuly uninstalled.", lpszTitle, MB_ICONINFORMATION);
1115: }
1116: else
1117: {
1118: int x;
1119:
1120: if (bUninstall == FALSE)
1121: x = MessageBox ((HWND) hwndDlg, "TrueCrypt has been successfuly installed.\nTo use TrueCrypt your system must be restarted.", lpszTitle, MB_ICONINFORMATION);
1122: else
1123: x = MessageBox ((HWND) hwndDlg, "TrueCrypt has been successfuly uninstalled.\nYour system must be restarted.", lpszTitle, MB_ICONINFORMATION);
1124: }
1125: }
1126: else
1127: {
1128: if (bUninstall == FALSE)
1129: MessageBox ((HWND) hwndDlg, "The installation has failed!", lpszTitle, MB_ICONHAND);
1130: else
1131: MessageBox ((HWND) hwndDlg, "The uninstall has failed!", lpszTitle, MB_ICONHAND);
1132: }
1133: }
1134:
1135: static void SetSystemRestorePoint (void *hwndDlg, BOOL finalize)
1136: {
1137: static RESTOREPOINTINFO RestPtInfo;
1138: static STATEMGRSTATUS SMgrStatus;
1139: static BOOL failed = FALSE;
1140: static BOOL (__stdcall *_SRSetRestorePoint)(PRESTOREPOINTINFO, PSTATEMGRSTATUS);
1141:
1142: if (!SystemRestoreDll) return;
1143:
1144: _SRSetRestorePoint = (BOOL (__stdcall *)(PRESTOREPOINTINFO, PSTATEMGRSTATUS))GetProcAddress (SystemRestoreDll,"SRSetRestorePointA");
1145: if (_SRSetRestorePoint == 0)
1146: {
1147: FreeLibrary (SystemRestoreDll);
1148: SystemRestoreDll = 0;
1149: return;
1150: }
1151:
1152: if (!finalize)
1153: {
1154: StatusMessage (hwndDlg, "%s", "Creating system restore point");
1155:
1156: // Initialize the RESTOREPOINTINFO structure
1157: RestPtInfo.dwEventType = BEGIN_SYSTEM_CHANGE;
1158:
1159: // Notify the system that changes are about to be made.
1160: // An application is to be installed.
1161: RestPtInfo.dwRestorePtType = APPLICATION_INSTALL;
1162:
1163: // Set RestPtInfo.llSequenceNumber.
1164: RestPtInfo.llSequenceNumber = 0;
1165:
1166: // String to be displayed by System Restore for this restore point.
1167: strcpy(RestPtInfo.szDescription, "TrueCrypt installation");
1168:
1169: // Notify the system that changes are to be made and that
1170: // the beginning of the restore point should be marked.
1171: if(!_SRSetRestorePoint(&RestPtInfo, &SMgrStatus))
1172: {
1173: StatusMessage (hwndDlg, "%s", "Failed to create System Restore point!");
1174: failed = TRUE;
1175: }
1176:
1177: return;
1178: }
1179:
1180: if (failed) return;
1181:
1182: StatusMessage (hwndDlg, "%s", "Closing system restore point");
1183:
1184: // The application performs some installation operations here.
1185:
1186: // Re-initialize the RESTOREPOINTINFO structure to notify the
1187: // system that the operation is finished.
1188: RestPtInfo.dwEventType = END_SYSTEM_CHANGE;
1189:
1190: // End the system change by returning the sequence number
1191: // received from the first call to SRSetRestorePoint.
1192: RestPtInfo.llSequenceNumber = SMgrStatus.llSequenceNumber;
1193:
1194: // Notify the system that the operation is done and that this
1195: // is the end of the restore point.
1196: if(!_SRSetRestorePoint(&RestPtInfo, &SMgrStatus))
1197: {
1198: StatusMessage (hwndDlg, "%s", "Closing system restore point failed!");
1199: }
1200: else
1201: StatusMessage (hwndDlg, "%s", "System restore point created");
1202:
1203: }
1204:
1205: void
1206: DoUninstall (void *hwndDlg)
1207: {
1208: BOOL bOK = TRUE;
1209:
1210: EnableWindow (GetDlgItem ((HWND) hwndDlg, IDOK), FALSE);
1211:
1212: WaitCursor ();
1213:
1214: SendMessage (GetDlgItem ((HWND) hwndDlg, IDC_FILES), LB_RESETCONTENT, 0, 0);
1215:
1216: if (DoDriverUnload (hwndDlg) == FALSE)
1217: {
1218: bOK = FALSE;
1219: }
1220: else if (DoServiceUninstall (hwndDlg, "TrueCryptService") == FALSE)
1221: {
1222: bOK = FALSE;
1223: }
1224: else if (DoServiceUninstall (hwndDlg, "truecrypt") == FALSE)
1225: {
1226: bOK = FALSE;
1227: }
1228: else if (DoFilesInstall ((HWND) hwndDlg, dlg_file_name, FALSE) == FALSE)
1229: {
1230: bOK = FALSE;
1231: }
1232: else if (DoRegUninstall ((HWND) hwndDlg) == FALSE)
1233: {
1234: bOK = FALSE;
1235: }
1236: else if (DoShortcutsUninstall (hwndDlg, dlg_file_name) == FALSE)
1237: {
1238: bOK = FALSE;
1239: }
1240: else
1241: {
1242: RemoveMessage ((HWND) hwndDlg, dlg_file_name);
1243: if (StatRemoveDirectory (dlg_file_name) == FALSE)
1244: {
1245: handleWin32Error ((HWND) hwndDlg);
1246: bOK = FALSE;
1247: }
1248: }
1249:
1250: NormalCursor ();
1251:
1252: EnableWindow (GetDlgItem ((HWND) hwndDlg, IDOK), TRUE);
1253:
1254: RebootPrompt (hwndDlg, bOK);
1255:
1256: }
1257:
1258: void
1259: DoInstall (void *hwndDlg)
1260: {
1261: BOOL bOK = TRUE;
1262:
1263: EnableWindow (GetDlgItem ((HWND) hwndDlg, IDOK), FALSE);
1264:
1265: WaitCursor ();
1266:
1267: SendMessage (GetDlgItem ((HWND) hwndDlg, IDC_FILES), LB_RESETCONTENT, 0, 0);
1268:
1269: if (IsButtonChecked (GetDlgItem ((HWND) hwndDlg, IDC_SYSTEM_RESTORE)))
1270: SetSystemRestorePoint (hwndDlg, FALSE);
1271:
1272: if (DoDriverUnload (hwndDlg) == FALSE)
1273: {
1274: bOK = FALSE;
1275: }
1276: else if (DoServiceUninstall (hwndDlg, "TrueCryptService") == FALSE)
1277: {
1278: bOK = FALSE;
1279: }
1280: else if (DoServiceUninstall (hwndDlg, "truecrypt") == FALSE)
1281: {
1282: bOK = FALSE;
1283: }
1284: else if (DoFilesInstall ((HWND) hwndDlg, dlg_file_name, IsButtonChecked (GetDlgItem ((HWND) hwndDlg, IDC_UNINSTALL))) == FALSE)
1285: {
1286: bOK = FALSE;
1287: }
1288: else if (DoRegInstall ((HWND) hwndDlg, dlg_file_name,
1289: IsButtonChecked (GetDlgItem ((HWND) hwndDlg, IDC_FILE_TYPE)),
1290: IsButtonChecked (GetDlgItem ((HWND) hwndDlg, IDC_UNINSTALL))) == FALSE)
1291: {
1292: bOK = FALSE;
1293: }
1294: else if (DoDriverInstall (hwndDlg) == FALSE)
1295: {
1296: bOK = FALSE;
1297: }
1298: else if (DoServiceInstall (hwndDlg) == FALSE)
1299: {
1300: bOK = FALSE;
1301: }
1302: else if (DoShortcutsInstall (hwndDlg, dlg_file_name,
1303: IsButtonChecked (GetDlgItem ((HWND) hwndDlg, IDC_PROG_GROUP)),
1304: IsButtonChecked (GetDlgItem ((HWND) hwndDlg, IDC_DESKTOP_ICON))) == FALSE)
1305: {
1306: bOK = FALSE;
1307: }
1308:
1309: if (IsButtonChecked (GetDlgItem ((HWND) hwndDlg, IDC_SYSTEM_RESTORE)))
1310: SetSystemRestorePoint (hwndDlg, TRUE);
1311:
1312: if (bOK)
1313: StatusMessage (hwndDlg, "%s", "Installation completed.");
1314:
1315: NormalCursor ();
1316:
1317: EnableWindow (GetDlgItem ((HWND) hwndDlg, IDOK), TRUE);
1318:
1319: RebootPrompt (hwndDlg, bOK);
1320: }
1321:
1322: BOOL
1323: IsAdmin (void)
1324: {
1325: HANDLE hAccessToken;
1326: UCHAR InfoBuffer[1024];
1327: PTOKEN_GROUPS ptgGroups = (PTOKEN_GROUPS) InfoBuffer;
1328: DWORD dwInfoBufferSize;
1329: PSID psidAdministrators;
1330: SID_IDENTIFIER_AUTHORITY siaNtAuthority = SECURITY_NT_AUTHORITY;
1331: BOOL bSuccess;
1332: UINT x;
1333:
1334: if (!OpenThreadToken (GetCurrentThread (), TOKEN_QUERY, TRUE,
1335: &hAccessToken))
1336: {
1337: if (GetLastError ()!= ERROR_NO_TOKEN)
1338: return FALSE;
1339:
1340: /* Retry against process token if no thread token exists */
1341: if (!OpenProcessToken (GetCurrentProcess (), TOKEN_QUERY,
1342: &hAccessToken))
1343: return FALSE;
1344: }
1345:
1346: bSuccess = GetTokenInformation (hAccessToken, TokenGroups, InfoBuffer,
1347: 1024, &dwInfoBufferSize);
1348:
1349: CloseHandle (hAccessToken);
1350:
1351: if (!bSuccess)
1352: return FALSE;
1353:
1354: if (!AllocateAndInitializeSid (&siaNtAuthority, 2,
1355: SECURITY_BUILTIN_DOMAIN_RID,
1356: DOMAIN_ALIAS_RID_ADMINS,
1357: 0, 0, 0, 0, 0, 0,
1358: &psidAdministrators))
1359: return FALSE;
1360:
1361: /* Assume that we don't find the admin SID. */
1362: bSuccess = FALSE;
1363:
1364: for (x = 0; x < ptgGroups->GroupCount; x++)
1365: {
1366: if (EqualSid (psidAdministrators, ptgGroups->Groups[x].Sid))
1367: {
1368: bSuccess = TRUE;
1369: break;
1370: }
1371:
1372: }
1373:
1374: FreeSid (psidAdministrators);
1375: return bSuccess;
1376: }
1377:
1378: BOOL WINAPI
1379: InstallDlgProc (HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM lParam)
1380: {
1381: WORD lw = LOWORD (wParam);
1382: if (lParam); /* remove warning */
1383:
1384: switch (msg)
1385: {
1386: case WM_INITDIALOG:
1387: SetDefaultUserFont (hwndDlg);
1388: InitDialog (hwndDlg);
1389:
1390: SetWindowText (GetDlgItem (hwndDlg, IDC_DESTINATION), "C:\\Program Files\\TrueCrypt");
1391:
1392: if (bUninstall == FALSE)
1393: {
1394: SendMessage (GetDlgItem (hwndDlg, IDC_FILES), LB_ADDSTRING, 0, (LPARAM) "By clicking 'Install', you accept the license agreement.");
1395:
1396: LoadLicense (hwndDlg);
1397: }
1398:
1399: SendMessage (GetDlgItem (hwndDlg, IDC_ALL_USERS), BM_SETCHECK, BST_CHECKED, 0);
1400: if (nCurrentOS != WIN_NT)
1401: EnableWindow (GetDlgItem ((HWND) hwndDlg, IDC_ALL_USERS), FALSE);
1402:
1403: SendMessage (GetDlgItem (hwndDlg, IDC_FILE_TYPE), BM_SETCHECK, BST_CHECKED, 0);
1404: SendMessage (GetDlgItem (hwndDlg, IDC_UNINSTALL), BM_SETCHECK, BST_CHECKED, 0);
1405: SendMessage (GetDlgItem (hwndDlg, IDC_PROG_GROUP), BM_SETCHECK, BST_CHECKED, 0);
1406: SendMessage (GetDlgItem (hwndDlg, IDC_DESKTOP_ICON), BM_SETCHECK, BST_CHECKED, 0);
1407:
1408: // System Restore
1409: SystemRestoreDll = LoadLibrary("srclient.dll");
1410:
1411: if (SystemRestoreDll != 0)
1412: SendMessage (GetDlgItem (hwndDlg, IDC_SYSTEM_RESTORE), BM_SETCHECK, BST_CHECKED, 0);
1413: else
1414: EnableWindow (GetDlgItem ((HWND) hwndDlg, IDC_SYSTEM_RESTORE), FALSE);
1415:
1416: SetWindowText (hwndDlg, lpszTitle);
1417: return 1;
1418:
1419: case WM_SYSCOMMAND:
1420: if (lw == IDC_ABOUT)
1421: {
1422: DialogBox (hInst, MAKEINTRESOURCE (IDD_ABOUT_DLG), hwndDlg, (DLGPROC) AboutDlgProc);
1423: return 1;
1424: }
1425: return 0;
1426:
1427: case WM_COMMAND:
1428: if (lw == IDOK)
1429: {
1430: char szDirname[TC_MAX_PATH];
1431:
1432: if (bDone == TRUE)
1433: {
1434: EndDialog (hwndDlg, IDOK);
1435: return 1;
1436: }
1437:
1438: GetWindowText (GetDlgItem (hwndDlg, IDC_DESTINATION), szDirname, sizeof (szDirname));
1439:
1440: if (bUninstall == FALSE)
1441: {
1442: if (mkfulldir (szDirname, TRUE) != 0)
1443: {
1444: char szTmp[TC_MAX_PATH];
1445: int x;
1446:
1447: //sprintf (szTmp, "The directory '%s' does not exist. Do you want to create this directory?", szDirname);
1448: //x = MessageBox (hwndDlg, szTmp, lpszTitle, MB_ICONQUESTION | MB_YESNO);
1449: //if (x == IDNO)
1450: //{
1451: // SetFocus (GetDlgItem (hwndDlg, IDC_DESTINATION));
1452: // return 1;
1453: //}
1454:
1455: if (mkfulldir (szDirname, FALSE) != 0)
1456: {
1457: handleWin32Error (hwndDlg);
1458: sprintf (szTmp, "The directory '%s' could not be created", szDirname);
1459: MessageBox (hwndDlg, szTmp, lpszTitle, MB_ICONHAND);
1460: return 1;
1461: }
1462:
1463: }
1464: }
1465:
1466: strcpy (dlg_file_name, szDirname);
1467:
1468: if (bUninstall == FALSE)
1469: _beginthread (DoInstall, 16384, (void *) hwndDlg);
1470: else
1471: _beginthread (DoUninstall, 16384, (void *) hwndDlg);
1472:
1473: return 1;
1474: }
1475:
1476: if (lw == IDCANCEL)
1477: {
1478: EndDialog (hwndDlg, IDCANCEL);
1479: return 1;
1480: }
1481:
1482: if (lw == IDC_BROWSE)
1483: {
1484: char szDirname[TC_MAX_PATH];
1485:
1486: GetWindowText (GetDlgItem (hwndDlg, IDC_DESTINATION), szDirname, sizeof (szDirname));
1487:
1488: if (BrowseFiles2 (hwndDlg, "Please select a folder", szDirname) == TRUE)
1489: SetWindowText (GetDlgItem (hwndDlg, IDC_DESTINATION), szDirname);
1490:
1491: return 1;
1492: }
1493:
1494: if (lw == IDC_DESTINATION && HIWORD (wParam) == EN_CHANGE && bDone == FALSE)
1495: {
1496: if (GetWindowTextLength (GetDlgItem (hwndDlg, IDC_DESTINATION)) <= 0)
1497: EnableWindow (GetDlgItem (hwndDlg, IDOK), FALSE);
1498: else
1499: EnableWindow (GetDlgItem (hwndDlg, IDOK), TRUE);
1500: return 1;
1501: }
1502:
1503: return 0;
1504:
1505: case WM_CLOSE:
1506: EndDialog (hwndDlg, IDCANCEL);
1507: return 1;
1508: }
1509:
1510: return 0;
1511: }
1512:
1513:
1514: int WINAPI
1515: WINMAIN (HINSTANCE hInstance, HINSTANCE hPrevInstance, char *lpszCommandLine,
1516: int nCmdShow)
1517: {
1518: if (nCmdShow && hPrevInstance); /* Remove unused parameter warning */
1519:
1520: lpszTitle = "TrueCrypt Setup";
1521:
1522: /* Call InitApp to initialize the common code */
1523: InitApp (hInstance);
1524:
1525: if (nCurrentOS == WIN_NT && IsAdmin ()!= TRUE)
1526: if (MessageBox (NULL, "To successfully install/uninstall TrueCrypt under Windows NT you must be running as an Administrator, "
1527: "do you still want to continue?", lpszTitle, MB_YESNO | MB_ICONQUESTION) != IDYES)
1528: return 0;
1529:
1530: if (lpszCommandLine[0] == '/' && (lpszCommandLine[1] == 'u' || lpszCommandLine[1] == 'U'))
1531: {
1532: bUninstall = TRUE;
1533: }
1534:
1535: if (bUninstall == FALSE)
1536: {
1537: /* Create the main dialog box */
1538: DialogBox (hInstance, MAKEINTRESOURCE (IDD_INSTALL), NULL, (DLGPROC) InstallDlgProc);
1539: }
1540: else
1541: {
1542: /* Create the main dialog box */
1543: DialogBox (hInstance, MAKEINTRESOURCE (IDD_UNINSTALL), NULL, (DLGPROC) InstallDlgProc);
1544: }
1545:
1546: /* Terminate */
1547: return 0;
1548: }
This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.