?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
No empty .Rs/.Re
[netbsd-mini2440.git] / external / bsd / bind / dist / bin / win32 / BINDInstall / BINDInstallDlg.cpp
blob483df2b410071a2a6588d4994fa9b5acc9e4afd8
1 /*
2 * Portions Copyright (C) 2004-2009 Internet Systems Consortium, Inc. ("ISC")
3 * Portions Copyright (C) 2001, 2003 Internet Software Consortium.
4 *
5 * Permission to use, copy, modify, and/or distribute this software for any
6 * purpose with or without fee is hereby granted, provided that the above
7 * copyright notice and this permission notice appear in all copies.
8 *
9 * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
10 * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
11 * AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
12 * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
13 * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
14 * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
15 * PERFORMANCE OF THIS SOFTWARE.
16 */
17
18 /* Id: BINDInstallDlg.cpp,v 1.46 2009/12/04 21:59:23 marka Exp */
19
20 /*
21 * Copyright (c) 1999-2000 by Nortel Networks Corporation
22 *
23 * Permission to use, copy, modify, and distribute this software for any
24 * purpose with or without fee is hereby granted, provided that the above
25 * copyright notice and this permission notice appear in all copies.
26 *
27 * THE SOFTWARE IS PROVIDED "AS IS" AND NORTEL NETWORKS DISCLAIMS
28 * ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES
29 * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL NORTEL NETWORKS
30 * BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES
31 * OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
32 * WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION,
33 * ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
34 * SOFTWARE.
35 */
36
37 /*
38 * Define this to make a standalone installer that will copy msvcrt.dll
39 * and/or msvcrtd.dll during the install
40 */
41 // #define BINARIES_INSTALL
42
43 /*
44 * msvcrt.dll is the release c-runtime library for MSVC. msvcrtd.dll is the debug
45 * c-runtime library for MSVC. If you have debug binaries you want to have DEBUG_BINARIES
46 * defined. If you have release binaries you want to have RELEASE_BINARIES defined.
47 * If you have both, then define them both.
48 * Of course, you need msvcrt[d].dll present to install it!
49 */
50 #ifdef BINARIES_INSTALL
51 // # define DEBUG_BINARIES
52 // # define RELEASE_BINARIES
53 #endif
54
55 #include "stdafx.h"
56 #include "BINDInstall.h"
57 #include "BINDInstallDlg.h"
58 #include "DirBrowse.h"
59 #include <winsvc.h>
60 #include <named/ntservice.h>
61 #include <isc/bind_registry.h>
62 #include <isc/ntgroups.h>
63 #include <direct.h>
64 #include "AccountInfo.h"
65 #include "versioninfo.h"
66
67 #include <config.h>
68
69 #define MAX_GROUPS 100
70 #define MAX_PRIVS 50
71
72 #define LOCAL_SERVICE "NT AUTHORITY\\LocalService"
73
74 #ifdef _DEBUG
75 #define new DEBUG_NEW
76 #undef THIS_FILE
77 static char THIS_FILE[] = __FILE__;
78 #endif
79
80 typedef struct _xexception
81 {
82 _xexception(UINT string, ...);
83
84 CString resString;
85 } Exception;
86
87 _xexception::_xexception(UINT string, ...)
88 {
89 CString format;
90 va_list va;
91
92 format.LoadString(string);
93
94 va_start(va, string);
95 resString.FormatV(format, va);
96 va_end(va);
97 }
98
99 typedef struct _filedata {
100 enum FileDestinations {TargetDir, BinDir, EtcDir, WinSystem};
101 enum FileImportance {Trivial, Normal, Critical};
102
103 char *filename;
104 int destination;
105 int importance;
106 BOOL checkVer;
107 BOOL withTools;
108 } FileData;
109
110 const FileData installFiles[] =
111 {
112 #ifdef BINARIES_INSTALL
113 # ifdef DEBUG_BINARIES
114 {"msvcrtd.dll", FileData::WinSystem, FileData::Critical, TRUE, TRUE},
115 # endif
116 # ifdef RELEASE_BINARIES
117 {"msvcrt.dll", FileData::WinSystem, FileData::Critical, TRUE, TRUE},
118 # endif
119 #endif
120 #if _MSC_VER < 1400
121 #if _MSC_VER >= 1310
122 {"mfc71.dll", FileData::WinSystem, FileData::Critical, TRUE, TRUE},
123 {"msvcr71.dll", FileData::WinSystem, FileData::Critical, TRUE, TRUE},
124 #elif _MSC_VER > 1200 && _MSC_VER < 1310
125 {"mfc70.dll", FileData::WinSystem, FileData::Critical, TRUE, TRUE},
126 {"msvcr70.dll", FileData::WinSystem, FileData::Critical, TRUE, TRUE},
127 #endif
128 #endif
129 {"bindevt.dll", FileData::BinDir, FileData::Normal, FALSE, TRUE},
130 {"libbind9.dll", FileData::BinDir, FileData::Critical, FALSE, TRUE},
131 {"libisc.dll", FileData::BinDir, FileData::Critical, FALSE, TRUE},
132 {"libisccfg.dll", FileData::BinDir, FileData::Critical, FALSE, TRUE},
133 {"libisccc.dll", FileData::BinDir, FileData::Critical, FALSE, TRUE},
134 {"libdns.dll", FileData::BinDir, FileData::Critical, FALSE, TRUE},
135 {"liblwres.dll", FileData::BinDir, FileData::Critical, FALSE, TRUE},
136 {"libeay32.dll", FileData::BinDir, FileData::Critical, FALSE, TRUE},
137 #ifdef HAVE_LIBXML2
138 {"libxml2.dll", FileData::BinDir, FileData::Critical, FALSE, TRUE},
139 #endif
140 {"named.exe", FileData::BinDir, FileData::Critical, FALSE, FALSE},
141 {"nsupdate.exe", FileData::BinDir, FileData::Normal, FALSE, TRUE},
142 {"BINDInstall.exe", FileData::BinDir, FileData::Normal, FALSE, TRUE},
143 {"rndc.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
144 {"dig.exe", FileData::BinDir, FileData::Normal, FALSE, TRUE},
145 {"host.exe", FileData::BinDir, FileData::Normal, FALSE, TRUE},
146 {"nslookup.exe", FileData::BinDir, FileData::Normal, FALSE, TRUE},
147 {"arpaname.exe", FileData::BinDir, FileData::Normal, FALSE, TRUE},
148 {"nsec3hash.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
149 {"genrandom.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
150 {"rndc-confgen.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
151 {"ddns-confgen.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
152 {"dnssec-keygen.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
153 {"dnssec-signzone.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
154 {"dnssec-dsfromkey.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
155 {"dnssec-keyfromlabel.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
156 {"dnssec-revoke.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
157 {"named-checkconf.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
158 {"named-checkzone.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
159 {"named-compilezone.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
160 {"named-journalprint.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
161 {"pkcs11-destroy.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
162 {"pkcs11-keygen.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
163 {"pkcs11-list.exe", FileData::BinDir, FileData::Normal, FALSE, FALSE},
164 {"readme1st.txt", FileData::BinDir, FileData::Trivial, FALSE, TRUE},
165 {NULL, -1, -1}
166 };
167
168 /////////////////////////////////////////////////////////////////////////////
169 // CBINDInstallDlg dialog
170
171 CBINDInstallDlg::CBINDInstallDlg(CWnd* pParent /*=NULL*/)
172 : CDialog(CBINDInstallDlg::IDD, pParent) {
173 char buf[MAX_PATH];
174
175 //{{AFX_DATA_INIT(CBINDInstallDlg)
176 m_targetDir = _T("");
177 m_version = _T("");
178 m_toolsOnly = FALSE;
179 m_autoStart = FALSE;
180 m_keepFiles = FALSE;
181 m_current = _T("");
182 m_startOnInstall = FALSE;
183 m_accountName = _T("");
184 m_accountPassword = _T("");
185 m_accountName = _T("");
186 //}}AFX_DATA_INIT
187 // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
188 m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
189
190 GetSystemDirectory(buf, MAX_PATH);
191 m_winSysDir = buf;
192 m_defaultDir = buf;
193 m_defaultDir += "\\dns";
194 m_installed = FALSE;
195 m_accountExists = FALSE;
196 m_accountUsed = FALSE;
197 m_serviceExists = TRUE;
198 GetCurrentServiceAccountName();
199 m_currentAccount = m_accountName;
200 if (m_accountName == "") {
201 m_accountName = "named";
202 }
203 }
204
205 void CBINDInstallDlg::DoDataExchange(CDataExchange* pDX) {
206 CDialog::DoDataExchange(pDX);
207 //{{AFX_DATA_MAP(CBINDInstallDlg)
208 DDX_Text(pDX, IDC_TARGETDIR, m_targetDir);
209 DDX_Text(pDX, IDC_VERSION, m_version);
210 DDX_Text(pDX, IDC_ACCOUNT_NAME, m_accountName);
211 DDX_Text(pDX, IDC_ACCOUNT_PASSWORD, m_accountPassword);
212 DDX_Text(pDX, IDC_ACCOUNT_PASSWORD_CONFIRM, m_accountPasswordConfirm);
213 DDX_Check(pDX, IDC_TOOLS_ONLY, m_toolsOnly);
214 DDX_Check(pDX, IDC_AUTO_START, m_autoStart);
215 DDX_Check(pDX, IDC_KEEP_FILES, m_keepFiles);
216 DDX_Text(pDX, IDC_CURRENT, m_current);
217 DDX_Check(pDX, IDC_START, m_startOnInstall);
218 //}}AFX_DATA_MAP
219 }
220
221 BEGIN_MESSAGE_MAP(CBINDInstallDlg, CDialog)
222 //{{AFX_MSG_MAP(CBINDInstallDlg)
223 ON_WM_PAINT()
224 ON_WM_QUERYDRAGICON()
225 ON_BN_CLICKED(IDC_BROWSE, OnBrowse)
226 ON_BN_CLICKED(IDC_INSTALL, OnInstall)
227 ON_BN_CLICKED(IDC_EXIT, OnExit)
228 ON_BN_CLICKED(IDC_UNINSTALL, OnUninstall)
229 //}}AFX_MSG_MAP
230 END_MESSAGE_MAP()
231
232 /////////////////////////////////////////////////////////////////////////////
233 // CBINDInstallDlg message handlers
234
235 BOOL CBINDInstallDlg::OnInitDialog() {
236 CDialog::OnInitDialog();
237
238 // Set the icon for this dialog. The framework does this automatically
239 // when the application's main window is not a dialog
240 SetIcon(m_hIcon, TRUE); // Set big icon
241 SetIcon(m_hIcon, FALSE); // Set small icon
242
243 char filename[MAX_PATH];
244 char dirname[MAX_PATH];
245 char *fptr = &filename[0];
246 GetModuleFileName(NULL, filename, MAX_PATH);
247 char *dptr = strrchr(filename,'\\');
248 size_t index = dptr - fptr;
249 strncpy(dirname, filename, index);
250 dirname[index] = '\0';
251 CString Dirname(dirname);
252 m_currentDir = Dirname;
253
254 CVersionInfo bindInst(filename);
255 if(bindInst.IsValid())
256 m_version.Format(IDS_VERSION, bindInst.GetFileVersionString());
257 else
258 m_version.LoadString(IDS_NO_VERSION);
259
260 DWORD dwBufLen = MAX_PATH;
261 char buf[MAX_PATH];
262 HKEY hKey;
263
264 m_startOnInstall = CheckBINDService();
265
266 /* See if we are installed already */
267 if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, BIND_SUBKEY, 0, KEY_READ, &hKey)
268 == ERROR_SUCCESS) {
269 m_installed = TRUE;
270 memset(buf, 0, MAX_PATH);
271 // Get the install directory
272 if (RegQueryValueEx(hKey, "InstallDir", NULL, NULL, (LPBYTE)buf,
273 &dwBufLen) == ERROR_SUCCESS)
274 if (strcmp(buf, ""))
275 m_defaultDir = buf;
276
277 RegCloseKey(hKey);
278 }
279 m_targetDir = m_defaultDir;
280
281 // Set checkbox defaults
282 m_autoStart = TRUE;
283 m_keepFiles = TRUE;
284
285 UpdateData(FALSE);
286
287 return (TRUE); /* return(TRUE) unless you set the focus to a control */
288 }
289
290 /*
291 * If you add a minimize button to your dialog, you will need the code below
292 * to draw the icon. For MFC applications using the document/view model,
293 * this is automatically done for you by the framework.
294 */
295
296 void CBINDInstallDlg::OnPaint() {
297 if (IsIconic()) {
298 CPaintDC dc(this); // device context for painting
299
300 SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
301
302 // Center icon in client rectangle
303 int cxIcon = GetSystemMetrics(SM_CXICON);
304 int cyIcon = GetSystemMetrics(SM_CYICON);
305 CRect rect;
306 GetClientRect(&rect);
307 int x = (rect.Width() - cxIcon + 1) / 2;
308 int y = (rect.Height() - cyIcon + 1) / 2;
309
310 // Draw the icon
311 dc.DrawIcon(x, y, m_hIcon);
312 }
313 else {
314 CDialog::OnPaint();
315 }
316 }
317
318 // The system calls this to obtain the cursor to display while the user drags
319 // the minimized window.
320 HCURSOR CBINDInstallDlg::OnQueryDragIcon() {
321 return((HCURSOR)m_hIcon);
322 }
323
324 void CBINDInstallDlg::OnBrowse() {
325
326 CDirBrowse browse;
327
328 if (browse.DoModal() == IDOK) {
329 //m_targetDir = browse.m_selectedDir;
330 UpdateData(FALSE);
331 }
332 }
333
334 /*
335 * User pressed the exit button
336 */
337 void CBINDInstallDlg::OnExit() {
338 EndDialog(0);
339 }
340
341 /*
342 * User pressed the uninstall button. Make it go.
343 */
344 void CBINDInstallDlg::OnUninstall() {
345 UpdateData();
346
347 if (MsgBox(IDS_UNINSTALL, MB_YESNO) == IDYES) {
348 if (CheckBINDService())
349 StopBINDService();
350
351 SC_HANDLE hSCManager = OpenSCManager(NULL, NULL,
352 SC_MANAGER_ALL_ACCESS);
353 if (!hSCManager) {
354 MsgBox(IDS_ERR_OPEN_SCM, GetErrMessage());
355 return;
356 }
357
358 SC_HANDLE hService = OpenService(hSCManager, BIND_SERVICE_NAME,
359 SERVICE_ALL_ACCESS);
360 if (!hService && GetLastError() != ERROR_SERVICE_DOES_NOT_EXIST){
361 MsgBox(IDS_ERR_OPEN_SERVICE, GetErrMessage());
362 return;
363 }
364
365 SERVICE_STATUS ss;
366 QueryServiceStatus(hService, &ss);
367 if (ss.dwCurrentState == SERVICE_RUNNING) {
368 BOOL rc = ControlService(hService,
369 SERVICE_CONTROL_STOP, &ss);
370 if (rc == FALSE || ss.dwCurrentState != SERVICE_STOPPED) {
371 MsgBox(IDS_ERR_STOP_SERVICE, GetErrMessage());
372 return;
373 }
374
375 }
376 CloseServiceHandle(hService);
377 CloseServiceHandle(hSCManager);
378
379 // Directories
380 m_etcDir = m_targetDir + "\\etc";
381 m_binDir = m_targetDir + "\\bin";
382
383 UninstallTags();
384 UnregisterMessages(TRUE);
385 UnregisterService(TRUE);
386 DeleteFiles(TRUE);
387 if (m_keepFiles == FALSE)
388 RemoveDirs(TRUE);
389 else
390 GetDlgItem(IDC_CREATE_DIR)->SetWindowText("Not Removed");
391
392
393 // Delete registry keys for named
394 RegDeleteKey(HKEY_LOCAL_MACHINE, BIND_SESSION_SUBKEY);
395 RegDeleteKey(HKEY_LOCAL_MACHINE, BIND_SUBKEY);
396 RegDeleteKey(HKEY_LOCAL_MACHINE, BIND_UNINSTALL_SUBKEY);
397
398 ProgramGroup(FALSE);
399
400 SetCurrent(IDS_UNINSTALL_DONE);
401 MsgBox(IDS_UNINSTALL_DONE);
402 }
403 }
404
405 /*
406 * User pressed the install button. Make it go.
407 */
408 void CBINDInstallDlg::OnInstall() {
409 #if _MSC_VER >= 1400
410 char Vcredist_x86[MAX_PATH];
411 #endif
412 BOOL success = FALSE;
413 int oldlen;
414
415 if (CheckBINDService())
416 StopBINDService();
417
418 InstallTags();
419
420 UpdateData();
421
422 if (!m_toolsOnly && m_accountName != LOCAL_SERVICE) {
423 /*
424 * Check that the Passwords entered match.
425 */
426 if (m_accountPassword != m_accountPasswordConfirm) {
427 MsgBox(IDS_ERR_PASSWORD);
428 return;
429 }
430
431 /*
432 * Check that there is not leading / trailing whitespace.
433 * This is for compatibility with the standard password dialog.
434 * Passwords really should be treated as opaque blobs.
435 */
436 oldlen = m_accountPassword.GetLength();
437 m_accountPassword.TrimLeft();
438 m_accountPassword.TrimRight();
439 if (m_accountPassword.GetLength() != oldlen) {
440 MsgBox(IDS_ERR_WHITESPACE);
441 return;
442 }
443
444 /*
445 * Check the entered account name.
446 */
447 if (ValidateServiceAccount() == FALSE)
448 return;
449
450 /*
451 * For Registration we need to know if account was changed.
452 */
453 if (m_accountName != m_currentAccount)
454 m_accountUsed = FALSE;
455
456 if (m_accountUsed == FALSE && m_serviceExists == FALSE)
457 {
458 /*
459 * Check that the Password is not null.
460 */
461 if (m_accountPassword.GetLength() == 0) {
462 MsgBox(IDS_ERR_NULLPASSWORD);
463 return;
464 }
465 }
466 } else if (m_accountName == LOCAL_SERVICE) {
467 /* The LocalService always exists. */
468 m_accountExists = TRUE;
469 if (m_accountName != m_currentAccount)
470 m_accountUsed = FALSE;
471 }
472
473 /* Directories */
474 m_etcDir = m_targetDir + "\\etc";
475 m_binDir = m_targetDir + "\\bin";
476
477 if (m_defaultDir != m_targetDir) {
478 if (GetFileAttributes(m_targetDir) != 0xFFFFFFFF)
479 {
480 int install = MsgBox(IDS_DIREXIST,
481 MB_YESNO | MB_ICONQUESTION, m_targetDir);
482 if (install == IDNO)
483 return;
484 }
485 else {
486 int createDir = MsgBox(IDS_CREATEDIR,
487 MB_YESNO | MB_ICONQUESTION, m_targetDir);
488 if (createDir == IDNO)
489 return;
490 }
491 }
492
493 if (!m_toolsOnly) {
494 if (m_accountExists == FALSE) {
495 success = CreateServiceAccount(m_accountName.GetBuffer(30),
496 m_accountPassword.GetBuffer(30));
497 if (success == FALSE) {
498 MsgBox(IDS_CREATEACCOUNT_FAILED);
499 return;
500 }
501 m_accountExists = TRUE;
502 }
503 }
504
505 ProgramGroup(FALSE);
506
507 #if _MSC_VER >= 1400
508 /*
509 * Install Visual Studio libraries. As per:
510 * http://blogs.msdn.com/astebner/archive/2006/08/23/715755.aspx
511 *
512 * Vcredist_x86.exe /q:a /c:"msiexec /i vcredist.msi /qn /l*v %temp%\vcredist_x86.log"
513 */
514 /*system(".\\Vcredist_x86.exe /q:a /c:\"msiexec /i vcredist.msi /qn /l*v %temp%\vcredist_x86.log\"");*/
515
516 /*
517 * Enclose full path to Vcredist_x86.exe in quotes as
518 * m_currentDir may contain spaces.
519 */
520 sprintf(Vcredist_x86, "\"%s\\Vcredist_x86.exe\"",
521 (LPCTSTR) m_currentDir);
522 system(Vcredist_x86);
523 #endif
524 try {
525 CreateDirs();
526 CopyFiles();
527 if (!m_toolsOnly)
528 RegisterService();
529 RegisterMessages();
530
531 HKEY hKey;
532
533 /* Create a new key for named */
534 SetCurrent(IDS_CREATE_KEY);
535 if (RegCreateKey(HKEY_LOCAL_MACHINE, BIND_SUBKEY,
536 &hKey) == ERROR_SUCCESS) {
537 // Get the install directory
538 RegSetValueEx(hKey, "InstallDir", 0, REG_SZ,
539 (LPBYTE)(LPCTSTR)m_targetDir,
540 m_targetDir.GetLength());
541 RegCloseKey(hKey);
542 }
543
544
545 SetCurrent(IDS_ADD_REMOVE);
546 if (RegCreateKey(HKEY_LOCAL_MACHINE, BIND_UNINSTALL_SUBKEY,
547 &hKey) == ERROR_SUCCESS) {
548 CString buf(BIND_DISPLAY_NAME);
549
550 RegSetValueEx(hKey, "DisplayName", 0, REG_SZ,
551 (LPBYTE)(LPCTSTR)buf, buf.GetLength());
552
553 buf.Format("%s\\BINDInstall.exe", m_binDir);
554 RegSetValueEx(hKey, "UninstallString", 0, REG_SZ,
555 (LPBYTE)(LPCTSTR)buf, buf.GetLength());
556 RegCloseKey(hKey);
557 }
558
559 ProgramGroup(FALSE);
560
561 if (m_startOnInstall)
562 StartBINDService();
563 }
564 catch(Exception e) {
565 MessageBox(e.resString);
566 SetCurrent(IDS_CLEANUP);
567 FailedInstall();
568 MsgBox(IDS_FAIL);
569 return;
570 }
571 catch(DWORD dw) {
572 CString msg;
573 msg.Format("A fatal error occured\n(%s)", GetErrMessage(dw));
574 MessageBox(msg);
575 SetCurrent(IDS_CLEANUP);
576 FailedInstall();
577 MsgBox(IDS_FAIL);
578 return;
579 }
580
581 SetCurrent(IDS_INSTALL_DONE);
582 MsgBox(IDS_SUCCESS);
583 }
584
585 /*
586 * Methods to do the work
587 */
588 void CBINDInstallDlg::CreateDirs() {
589 /* s'OK if the directories already exist */
590 SetCurrent(IDS_CREATE_DIR, m_targetDir);
591 if (!CreateDirectory(m_targetDir, NULL) && GetLastError() != ERROR_ALREADY_EXISTS)
592 throw(Exception(IDS_ERR_CREATE_DIR, m_targetDir, GetErrMessage()));
593
594 SetCurrent(IDS_CREATE_DIR, m_etcDir);
595 if (!CreateDirectory(m_etcDir, NULL) && GetLastError() != ERROR_ALREADY_EXISTS)
596 throw(Exception(IDS_ERR_CREATE_DIR, m_etcDir, GetErrMessage()));
597
598 SetCurrent(IDS_CREATE_DIR, m_binDir);
599 if (!CreateDirectory(m_binDir, NULL) && GetLastError() != ERROR_ALREADY_EXISTS)
600 throw(Exception(IDS_ERR_CREATE_DIR, m_binDir, GetErrMessage()));
601
602 SetItemStatus(IDC_CREATE_DIR);
603 }
604
605 void CBINDInstallDlg::RemoveDirs(BOOL uninstall) {
606 if (!m_keepFiles) {
607 SetCurrent(IDS_REMOVE_DIR, m_binDir);
608 // Check for existence then remove if present
609 if (GetFileAttributes(m_binDir) != 0xFFFFFFFF)
610 RemoveDirectory(m_binDir);
611
612 SetCurrent(IDS_REMOVE_DIR, m_etcDir);
613 if (GetFileAttributes(m_etcDir) != 0xFFFFFFFF)
614 RemoveDirectory(m_etcDir);
615
616 SetCurrent(IDS_REMOVE_DIR, m_targetDir);
617 if (GetFileAttributes(m_targetDir) != 0xFFFFFFFF)
618 RemoveDirectory(m_targetDir);
619 }
620
621 if (uninstall)
622 SetItemStatus(IDC_CREATE_DIR, TRUE);
623 }
624
625 void CBINDInstallDlg::CopyFiles() {
626 CString destFile;
627
628 for (int i = 0; installFiles[i].filename; i++) {
629 if (m_toolsOnly && !installFiles[i].withTools)
630 continue;
631 SetCurrent(IDS_COPY_FILE, installFiles[i].filename);
632
633 destFile = DestDir(installFiles[i].destination) + "\\" +
634 installFiles[i].filename;
635 CString filespec = m_currentDir + "\\" + installFiles[i].filename;
636 CVersionInfo bindFile(destFile);
637
638 CVersionInfo origFile(filespec);
639 if (!origFile.IsValid() && installFiles[i].checkVer) {
640 if (MsgBox(IDS_FILE_BAD, MB_YESNO,
641 installFiles[i].filename) == IDNO)
642 throw(Exception(IDS_ERR_COPY_FILE,
643 installFiles[i].filename,
644 GetErrMessage()));
645 }
646
647 try {
648 /*
649 * Ignore Version checking. We need to make sure that all files get copied regardless
650 * of whether or not they are earlier or later versions since we cannot guarantee
651 * that we have either backward or forward compatibility between versions.
652 */
653 bindFile.CopyFileNoVersion(origFile);
654 }
655 catch(...) {
656 if (installFiles[i].importance != FileData::Trivial) {
657 if (installFiles[i].importance ==
658 FileData::Critical ||
659 MsgBox(IDS_ERR_NONCRIT_FILE, MB_YESNO,
660 installFiles[i].filename,
661 GetErrMessage()) == IDNO)
662 {
663 SetItemStatus(IDC_COPY_FILE, FALSE);
664 throw(Exception(IDS_ERR_COPY_FILE,
665 installFiles[i].filename,
666 GetErrMessage()));
667 }
668 }
669 }
670 }
671
672 SetItemStatus(IDC_COPY_FILE);
673 }
674
675 void CBINDInstallDlg::DeleteFiles(BOOL uninstall) {
676 CString destFile;
677
678 for (int i = 0; installFiles[i].filename; i++) {
679 if (installFiles[i].checkVer)
680 continue;
681
682 destFile = DestDir(installFiles[i].destination) + "\\" +
683 installFiles[i].filename;
684
685 if (uninstall)
686 SetCurrent(IDS_DELETE_FILE, installFiles[i].filename);
687
688 DeleteFile(destFile);
689 }
690
691 if (!m_keepFiles) {
692 WIN32_FIND_DATA findData;
693 CString file = m_etcDir + "\\*.*";
694 BOOL rc;
695 HANDLE hFile;
696
697 hFile = FindFirstFile(file, &findData);
698 rc = hFile != INVALID_HANDLE_VALUE;
699
700 while (rc == TRUE) {
701 if (strcmp(findData.cFileName, ".") &&
702 strcmp(findData.cFileName, "..")) {
703 file = m_etcDir + "\\" + findData.cFileName;
704 SetCurrent(IDS_DELETE_FILE, file);
705 DeleteFile(file);
706 }
707 rc = FindNextFile(hFile, &findData);
708 }
709 FindClose(hFile);
710 }
711
712 if (uninstall)
713 SetItemStatus(IDC_COPY_FILE, TRUE);
714 }
715
716 /*
717 * Get the service account name out of the registry, if any
718 */
719 void
720 CBINDInstallDlg::GetCurrentServiceAccountName() {
721 HKEY hKey;
722 BOOL keyFound = FALSE;
723 char accountName[MAX_PATH];
724 DWORD nameLen = MAX_PATH;
725 CString Tmp;
726 m_accountUsed = FALSE;
727
728 memset(accountName, 0, nameLen);
729 if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, BIND_SERVICE_SUBKEY, 0, KEY_READ,
730 &hKey) == ERROR_SUCCESS) {
731 keyFound = TRUE;
732 }
733 else {
734 m_serviceExists = FALSE;
735 }
736
737 if (keyFound == TRUE) {
738 /* Get the named service account, if one was specified */
739 if (RegQueryValueEx(hKey, "ObjectName", NULL, NULL,
740 (LPBYTE)accountName, &nameLen) != ERROR_SUCCESS)
741 keyFound = FALSE;
742 }
743
744 RegCloseKey(hKey);
745 if (keyFound == FALSE)
746 m_accountName = "";
747 else if (!strcmp(accountName, LOCAL_SERVICE)) {
748 m_accountName = LOCAL_SERVICE;
749 m_accountUsed = TRUE;
750 } else {
751 /*
752 * LocalSystem is not a regular account and is equivalent
753 * to no account but with lots of privileges
754 */
755 Tmp = accountName;
756 if (Tmp == ".\\LocalSystem")
757 m_accountName = "";
758 /* Found account strip any ".\" from it */
759 if (Tmp.Left(2) == ".\\") {
760 m_accountName = Tmp.Mid(2);
761 m_accountUsed = TRUE;
762 }
763 }
764 }
765
766 BOOL
767 CBINDInstallDlg::ValidateServiceAccount() {
768 wchar_t *PrivList[MAX_PRIVS];
769 unsigned int PrivCount = 0;
770 char *Groups[MAX_GROUPS];
771 unsigned int totalGroups = 0;
772 int status;
773 char *name;
774
775 name = m_accountName.GetBuffer(30);
776
777 status = GetAccountPrivileges(name, PrivList, &PrivCount,
778 Groups, &totalGroups, MAX_GROUPS);
779 if (status == RTN_NOACCOUNT) {
780 m_accountExists = FALSE;
781 /* We need to do this in case an account was previously used */
782 m_accountUsed = FALSE;
783 return (TRUE);
784 }
785 if (status != RTN_OK) {
786 MsgBox(IDS_ERR_BADACCOUNT);
787 return (FALSE);
788 }
789
790 m_accountExists = TRUE;
791 if (PrivCount > 1) {
792 if (MsgBox(IDS_ERR_TOOPRIVED, MB_YESNO) == IDYES)
793 return (FALSE);
794 else
795 return (TRUE);
796 }
797
798 /* See if we have the correct privilege */
799 if (wcscmp(PrivList[0], SE_SERVICE_LOGON_PRIV) != 0) {
800 MsgBox(IDS_ERR_WRONGPRIV, PrivList[0]);
801 return (FALSE);
802 }
803 return (TRUE);
804 }
805
806 void
807 CBINDInstallDlg::RegisterService() {
808 SC_HANDLE hSCManager;
809 SC_HANDLE hService;
810 CString StartName;
811
812 if (m_accountName == LOCAL_SERVICE)
813 StartName = LOCAL_SERVICE;
814 else
815 StartName = ".\\" + m_accountName;
816 /*
817 * We need to change the service rather than create it
818 * if the service already exists. Do nothing if we are already
819 * using that account
820 */
821 if (m_serviceExists == TRUE) {
822 if (m_accountUsed == FALSE) {
823 UpdateService(StartName);
824 SetItemStatus(IDC_REG_SERVICE);
825 return;
826 } else {
827 SetItemStatus(IDC_REG_SERVICE);
828 return;
829 }
830 }
831
832 SetCurrent(IDS_OPEN_SCM);
833 hSCManager= OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
834 if (!hSCManager)
835 throw(Exception(IDS_ERR_OPEN_SCM, GetErrMessage()));
836
837 DWORD dwStart = SERVICE_DEMAND_START;
838 if (m_autoStart)
839 dwStart = SERVICE_AUTO_START;
840
841 DWORD dwServiceType = SERVICE_WIN32_OWN_PROCESS;
842
843 CString namedLoc;
844 namedLoc.Format("%s\\bin\\named.exe", m_targetDir);
845
846 SetCurrent(IDS_CREATE_SERVICE);
847 hService = CreateService(hSCManager, BIND_SERVICE_NAME,
848 BIND_DISPLAY_NAME, SERVICE_ALL_ACCESS, dwServiceType, dwStart,
849 SERVICE_ERROR_NORMAL, namedLoc, NULL, NULL, NULL, StartName,
850 m_accountPassword);
851
852 if (!hService && GetLastError() != ERROR_SERVICE_EXISTS)
853 throw(Exception(IDS_ERR_CREATE_SERVICE, GetErrMessage()));
854
855 if (hService)
856 CloseServiceHandle(hService);
857
858 if (hSCManager)
859 CloseServiceHandle(hSCManager);
860
861 SetItemStatus(IDC_REG_SERVICE);
862 }
863
864 void
865 CBINDInstallDlg::UpdateService(CString StartName) {
866 SC_HANDLE hSCManager;
867 SC_HANDLE hService;
868
869 if(m_toolsOnly)
870 return;
871
872 SetCurrent(IDS_OPEN_SCM);
873 hSCManager= OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
874 if (!hSCManager) {
875 MsgBox(IDS_ERR_OPEN_SCM, GetErrMessage());
876 return;
877 }
878
879 DWORD dwStart = SERVICE_DEMAND_START;
880 if (m_autoStart)
881 dwStart = SERVICE_AUTO_START;
882
883 DWORD dwServiceType = SERVICE_WIN32_OWN_PROCESS;
884
885 CString namedLoc;
886 namedLoc.Format("%s\\bin\\named.exe", m_targetDir);
887
888 SetCurrent(IDS_OPEN_SERVICE);
889 hService = OpenService(hSCManager, BIND_SERVICE_NAME,
890 SERVICE_CHANGE_CONFIG);
891 if (!hService)
892 {
893 MsgBox(IDS_ERR_OPEN_SERVICE, GetErrMessage());
894 if (hSCManager)
895 CloseServiceHandle(hSCManager);
896 return;
897 } else {
898 if (ChangeServiceConfig(hService, dwServiceType, dwStart,
899 SERVICE_ERROR_NORMAL, namedLoc, NULL, NULL, NULL,
900 StartName, m_accountPassword, BIND_DISPLAY_NAME)
901 != TRUE) {
902 DWORD err = GetLastError();
903 MsgBox(IDS_ERR_UPDATE_SERVICE, GetErrMessage());
904 }
905 }
906
907 if (hService)
908 CloseServiceHandle(hService);
909
910 if (hSCManager)
911 CloseServiceHandle(hSCManager);
912
913 SetItemStatus(IDC_REG_SERVICE);
914 }
915
916 void CBINDInstallDlg::UnregisterService(BOOL uninstall) {
917 BOOL rc = FALSE;
918 SC_HANDLE hSCManager;
919 SC_HANDLE hService;
920
921 while(1) {
922 SetCurrent(IDS_OPEN_SCM);
923 hSCManager= OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
924 if (!hSCManager && uninstall == TRUE) {
925 MsgBox(IDS_ERR_OPEN_SCM, GetErrMessage());
926 break;
927 }
928
929 SetCurrent(IDS_OPEN_SERVICE);
930 hService = OpenService(hSCManager, BIND_SERVICE_NAME,
931 STANDARD_RIGHTS_REQUIRED);
932 if (!hService && uninstall == TRUE)
933 {
934 if (GetLastError() != ERROR_SERVICE_DOES_NOT_EXIST) {
935 MsgBox(IDS_ERR_OPEN_SERVICE, GetErrMessage());
936 break;
937 }
938 }
939 else {
940 SetCurrent(IDS_REMOVE_SERVICE);
941 if (!DeleteService(hService) && uninstall == TRUE) {
942 DWORD err = GetLastError();
943 if (err != ERROR_SERVICE_MARKED_FOR_DELETE &&
944 err != ERROR_SERVICE_DOES_NOT_EXIST) {
945 MsgBox(IDS_ERR_REMOVE_SERVICE, GetErrMessage());
946 break;
947 }
948 }
949 }
950
951 rc = TRUE;
952 break;
953 }
954
955 if (hService)
956 CloseServiceHandle(hService);
957
958 if (hSCManager)
959 CloseServiceHandle(hSCManager);
960
961 if (uninstall)
962 SetItemStatus(IDC_REG_SERVICE, rc);
963 }
964
965 void CBINDInstallDlg::RegisterMessages() {
966 HKEY hKey;
967 DWORD dwData;
968 char pszMsgDLL[MAX_PATH];
969
970 sprintf(pszMsgDLL, "%s\\%s", (LPCTSTR)m_binDir, "bindevt.dll");
971
972 SetCurrent(IDS_REGISTER_MESSAGES);
973 /* Create a new key for named */
974 if (RegCreateKey(HKEY_LOCAL_MACHINE, BIND_MESSAGE_SUBKEY, &hKey)
975 != ERROR_SUCCESS)
976 throw(Exception(IDS_ERR_CREATE_KEY, GetErrMessage()));
977
978 /* Add the Event-ID message-file name to the subkey. */
979 if (RegSetValueEx(hKey, "EventMessageFile", 0, REG_EXPAND_SZ,
980 (LPBYTE)pszMsgDLL, (DWORD)(strlen(pszMsgDLL) + 1)) != ERROR_SUCCESS)
981 throw(Exception(IDS_ERR_SET_VALUE, GetErrMessage()));
982
983 /* Set the supported types flags and addit to the subkey. */
984 dwData = EVENTLOG_ERROR_TYPE | EVENTLOG_WARNING_TYPE | EVENTLOG_INFORMATION_TYPE;
985 if (RegSetValueEx(hKey, "TypesSupported", 0, REG_DWORD,
986 (LPBYTE)&dwData, sizeof(DWORD)) != ERROR_SUCCESS)
987 throw(Exception(IDS_ERR_SET_VALUE, GetErrMessage()));
988
989 RegCloseKey(hKey);
990
991 SetItemStatus(IDC_REG_MESSAGE);
992 }
993
994 void CBINDInstallDlg::UnregisterMessages(BOOL uninstall) {
995 BOOL rc = FALSE;
996 HKEY hKey = NULL;
997
998 while(1) {
999 SetCurrent(IDS_UNREGISTER_MESSAGES);
1000 /* Open key for Application Event Log */
1001 if (RegOpenKey(HKEY_LOCAL_MACHINE, EVENTLOG_APP_SUBKEY, &hKey)
1002 != ERROR_SUCCESS)
1003 break;
1004
1005 /* Remove named from the list of messages sources */
1006 if (RegDeleteKey(hKey, BIND_MESSAGE_NAME) != ERROR_SUCCESS)
1007 break;
1008
1009 rc = TRUE;
1010 break;
1011 }
1012
1013 if (hKey)
1014 RegCloseKey(hKey);
1015
1016 if (uninstall)
1017 SetItemStatus(IDC_REG_MESSAGE, rc);
1018 }
1019
1020 /*
1021 * Install failed - clean up quietly
1022 */
1023 void CBINDInstallDlg::FailedInstall() {
1024 UnregisterMessages(FALSE);
1025 UnregisterService(FALSE);
1026 DeleteFiles(FALSE);
1027 RemoveDirs(FALSE);
1028 }
1029
1030 /*
1031 * Set the checklist tags for install
1032 */
1033 void CBINDInstallDlg::InstallTags() {
1034 CString tag;
1035
1036 tag.LoadString(IDS_INSTALL_FILE);
1037 GetDlgItem(IDC_COPY_TAG)->SetWindowText(tag);
1038 GetDlgItem(IDC_COPY_FILE)->SetWindowText("");
1039
1040 tag.LoadString(IDS_INSTALL_DIR);
1041 GetDlgItem(IDC_DIR_TAG)->SetWindowText(tag);
1042 GetDlgItem(IDC_CREATE_DIR)->SetWindowText("");
1043 GetDlgItem(IDC_REG_SERVICE)->SetWindowText("");
1044
1045 tag.LoadString(IDS_INSTALL_SERVICE);
1046 GetDlgItem(IDC_SERVICE_TAG)->SetWindowText(tag);
1047
1048 tag.LoadString(IDS_INSTALL_MESSAGE);
1049 GetDlgItem(IDC_MESSAGE_TAG)->SetWindowText(tag);
1050 GetDlgItem(IDC_REG_MESSAGE)->SetWindowText("");
1051 }
1052
1053 /*
1054 * Set the checklist tags for uninstall
1055 */
1056 void CBINDInstallDlg::UninstallTags() {
1057 CString tag;
1058
1059 tag.LoadString(IDS_UNINSTALL_FILES);
1060 GetDlgItem(IDC_COPY_TAG)->SetWindowText(tag);
1061 GetDlgItem(IDC_COPY_FILE)->SetWindowText("");
1062
1063 tag.LoadString(IDS_UNINSTALL_DIR);
1064 GetDlgItem(IDC_DIR_TAG)->SetWindowText(tag);
1065 GetDlgItem(IDC_CREATE_DIR)->SetWindowText("");
1066
1067 tag.LoadString(IDS_UNINSTALL_SERVICE);
1068 GetDlgItem(IDC_SERVICE_TAG)->SetWindowText(tag);
1069 GetDlgItem(IDC_REG_SERVICE)->SetWindowText("");
1070
1071 tag.LoadString(IDS_UNINSTALL_MESSAGE);
1072 GetDlgItem(IDC_MESSAGE_TAG)->SetWindowText(tag);
1073 GetDlgItem(IDC_REG_MESSAGE)->SetWindowText("");
1074 }
1075
1076 void CBINDInstallDlg::SetItemStatus(UINT nID, BOOL bSuccess) {
1077 GetDlgItem(nID)->SetWindowText(bSuccess == TRUE ? "Done" : "Failed");
1078 }
1079
1080
1081 /*
1082 * Set the text in the current operation field - use a string table string
1083 */
1084 void CBINDInstallDlg::SetCurrent(int id, ...) {
1085 CString format;
1086 va_list va;
1087 char buf[128];
1088
1089 format.LoadString(id);
1090 memset(buf, 0, 128);
1091
1092 va_start(va, id);
1093 vsprintf(buf, format, va);
1094 va_end(va);
1095
1096 m_current.Format("%s", buf);
1097 UpdateData(FALSE);
1098 }
1099
1100 /*
1101 * Stop the BIND service
1102 */
1103 void CBINDInstallDlg::StopBINDService() {
1104 SERVICE_STATUS svcStatus;
1105
1106 SetCurrent(IDS_STOP_SERVICE);
1107
1108 SC_HANDLE hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
1109 if (!hSCManager) {
1110 MsgBox(IDS_ERR_OPEN_SCM, GetErrMessage());
1111 }
1112
1113 SC_HANDLE hBINDSvc = OpenService(hSCManager, BIND_SERVICE_NAME,
1114 SERVICE_ALL_ACCESS);
1115 if (!hBINDSvc) {
1116 MsgBox(IDS_ERR_OPEN_SERVICE, GetErrMessage());
1117 }
1118
1119 BOOL rc = ControlService(hBINDSvc, SERVICE_CONTROL_STOP, &svcStatus);
1120 }
1121
1122 /*
1123 * Start the BIND service
1124 */
1125 void CBINDInstallDlg::StartBINDService() {
1126 SetCurrent(IDS_START_SERVICE);
1127
1128 SC_HANDLE hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
1129 if (!hSCManager) {
1130 MsgBox(IDS_ERR_OPEN_SCM, GetErrMessage());
1131 }
1132
1133 SC_HANDLE hBINDSvc = OpenService(hSCManager, BIND_SERVICE_NAME,
1134 SERVICE_ALL_ACCESS);
1135 if (!hBINDSvc) {
1136 MsgBox(IDS_ERR_OPEN_SERVICE, GetErrMessage());
1137 }
1138 BOOL rc = StartService(hBINDSvc, 0, NULL);
1139 }
1140
1141 /*
1142 * Check to see if the BIND service is running or not
1143 */
1144 BOOL CBINDInstallDlg::CheckBINDService() {
1145 SERVICE_STATUS svcStatus;
1146
1147 SC_HANDLE hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
1148 if (hSCManager) {
1149 SC_HANDLE hBINDSvc = OpenService(hSCManager, BIND_SERVICE_NAME,
1150 SERVICE_ALL_ACCESS);
1151 if (hBINDSvc) {
1152 BOOL rc = ControlService(hBINDSvc,
1153 SERVICE_CONTROL_INTERROGATE, &svcStatus);
1154 if (!rc)
1155 DWORD err = GetLastError();
1156
1157 return (svcStatus.dwCurrentState == SERVICE_RUNNING);
1158 }
1159 }
1160 return (FALSE);
1161 }
1162
1163 /*
1164 * Display message boxes with variable args, using string table strings
1165 * for the format specifiers
1166 */
1167 int CBINDInstallDlg::MsgBox(int id, ...) {
1168 CString format;
1169 va_list va;
1170 char buf[BUFSIZ];
1171
1172 format.LoadString(id);
1173 memset(buf, 0, BUFSIZ);
1174
1175 va_start(va, id);
1176 vsprintf(buf, format, va);
1177 va_end(va);
1178
1179 return (MessageBox(buf));
1180 }
1181
1182 int CBINDInstallDlg::MsgBox(int id, UINT type, ...) {
1183 CString format;
1184 va_list va;
1185 char buf[BUFSIZ];
1186
1187 format.LoadString(id);
1188 memset(buf, 0, BUFSIZ);
1189
1190 va_start(va, type);
1191 vsprintf(buf, format, va);
1192 va_end(va);
1193
1194 return(MessageBox(buf, NULL, type));
1195 }
1196
1197 /*
1198 * Call GetLastError(), retrieve the message associated with the error
1199 */
1200 CString CBINDInstallDlg::GetErrMessage(DWORD err) {
1201 LPVOID msgBuf;
1202 static char buf[BUFSIZ];
1203
1204 DWORD len = FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
1205 NULL, err == -1 ? GetLastError() : err, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR) &msgBuf, 0, NULL );
1206
1207
1208 strcpy(buf, (LPTSTR)msgBuf);
1209 LocalFree(msgBuf);
1210 /* Strip off the period and the \n */
1211 buf[len - 3] = 0;
1212 return(buf);
1213 }
1214
1215 void CBINDInstallDlg::ProgramGroup(BOOL create) {
1216 TCHAR path[MAX_PATH], commonPath[MAX_PATH], fileloc[MAX_PATH], linkpath[MAX_PATH];
1217 HRESULT hres;
1218 IShellLink *psl = NULL;
1219 LPMALLOC pMalloc = NULL;
1220 ITEMIDLIST *itemList = NULL;
1221
1222 HRESULT hr = SHGetMalloc(&pMalloc);
1223 if (hr != NOERROR) {
1224 MessageBox("Could not get a handle to Shell memory object");
1225 return;
1226 }
1227
1228 hr = SHGetSpecialFolderLocation(m_hWnd, CSIDL_COMMON_PROGRAMS, &itemList);
1229 if (hr != NOERROR) {
1230 MessageBox("Could not get a handle to the Common Programs folder");
1231 if (itemList) {
1232 pMalloc->Free(itemList);
1233 }
1234 return;
1235 }
1236
1237 hr = SHGetPathFromIDList(itemList, commonPath);
1238 pMalloc->Free(itemList);
1239
1240 if (create) {
1241 sprintf(path, "%s\\ISC", commonPath);
1242 CreateDirectory(path, NULL);
1243
1244 sprintf(path, "%s\\ISC\\BIND", commonPath);
1245 CreateDirectory(path, NULL);
1246
1247 hres = CoInitialize(NULL);
1248
1249 if (SUCCEEDED(hres)) {
1250 // Get a pointer to the IShellLink interface.
1251 hres = CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLink, (LPVOID *)&psl);
1252 if (SUCCEEDED(hres))
1253 {
1254 IPersistFile* ppf;
1255 sprintf(linkpath, "%s\\BINDCtrl.lnk", path);
1256 sprintf(fileloc, "%s\\BINDCtrl.exe", m_binDir);
1257
1258 psl->SetPath(fileloc);
1259 psl->SetDescription("BIND Control Panel");
1260
1261 hres = psl->QueryInterface(IID_IPersistFile, (void **)&ppf);
1262 if (SUCCEEDED(hres)) {
1263 WCHAR wsz[MAX_PATH];
1264
1265 MultiByteToWideChar(CP_ACP, 0, linkpath, -1, wsz, MAX_PATH);
1266 hres = ppf->Save(wsz, TRUE);
1267 ppf->Release();
1268 }
1269
1270 if (GetFileAttributes("readme.txt") != -1) {
1271 sprintf(fileloc, "%s\\Readme.txt", m_targetDir);
1272 sprintf(linkpath, "%s\\Readme.lnk", path);
1273
1274 psl->SetPath(fileloc);
1275 psl->SetDescription("BIND Readme");
1276
1277 hres = psl->QueryInterface(IID_IPersistFile, (void **)&ppf);
1278 if (SUCCEEDED(hres)) {
1279 WCHAR wsz[MAX_PATH];
1280
1281 MultiByteToWideChar(CP_ACP, 0, linkpath, -1, wsz, MAX_PATH);
1282 hres = ppf->Save(wsz, TRUE);
1283 ppf->Release();
1284 }
1285 psl->Release();
1286 }
1287 }
1288 CoUninitialize();
1289 }
1290 }
1291 else {
1292 TCHAR filename[MAX_PATH];
1293 WIN32_FIND_DATA fd;
1294
1295 sprintf(path, "%s\\ISC\\BIND", commonPath);
1296
1297 sprintf(filename, "%s\\*.*", path);
1298 HANDLE hFind = FindFirstFile(filename, &fd);
1299 if (hFind != INVALID_HANDLE_VALUE) {
1300 do {
1301 if (strcmp(fd.cFileName, ".") && strcmp(fd.cFileName, "..")) {
1302 sprintf(filename, "%s\\%s", path, fd.cFileName);
1303 DeleteFile(filename);
1304 }
1305 } while (FindNextFile(hFind, &fd));
1306 FindClose(hFind);
1307 }
1308 RemoveDirectory(path);
1309 sprintf(path, "%s\\ISC", commonPath);
1310 RemoveDirectory(path);
1311 }
1312 }
1313
1314 CString CBINDInstallDlg::DestDir(int destination) {
1315 switch(destination) {
1316 case FileData::TargetDir:
1317 return m_targetDir;
1318 case FileData::BinDir:
1319 return m_binDir;
1320 case FileData::EtcDir:
1321 return m_etcDir;
1322 case FileData::WinSystem:
1323 return m_winSysDir;
1324 }
1325 return("");
1326 }
NetBSD on the FriendlyARM MINI2440