3564fa73692fe6ac727252e4dfe05f6977773cac
[openafs.git] / src / WINNT / client_creds / window.cpp
1 /*
2  * Copyright 2000, International Business Machines Corporation and others.
3  * All Rights Reserved.
4  * 
5  * This software has been released under the terms of the IBM Public
6  * License.  For details, see the LICENSE file in the top-level source
7  * directory or online at http://www.openafs.org/dl/license10.html
8  */
9
10 extern "C" {
11 #include <afs/param.h>
12 #include <afs/stds.h>
13 }
14
15 #include "afscreds.h"
16 #include "afskfw.h"
17
18
19 /*
20  * DEFINITIONS ________________________________________________________________
21  *
22  */
23
24 #define ID_REMIND_TIMER      1000
25 #define ID_SERVICE_TIMER     1001
26
27 #define cREALLOC_TABS        4
28
29 #define dwTABPARAM_MOUNT     (LPTSTR)0
30 #define dwTABPARAM_ADVANCED  (LPTSTR)1
31 #define ISCELLTAB(_psz)      ((HIWORD((LONG)(_psz))) != 0)
32
33
34 /*
35  * PROTOTYPES _________________________________________________________________
36  *
37  */
38
39 void Main_OnInitDialog (HWND hDlg);
40 void Main_OnCheckMenuRemind (void);
41 void Main_OnRemindTimer (void);
42 void Main_OnMouseOver (void);
43 void Main_OnSelectTab (void);
44 void Main_OnCheckTerminate (void);
45 HWND Main_CreateTabDialog (HWND hTab, size_t iTab);
46
47 BOOL CALLBACK Terminate_DlgProc (HWND hDlg, UINT msg, WPARAM wp, LPARAM lp);
48 void Terminate_OnInitDialog (HWND hDlg);
49 void Terminate_OnOK (HWND hDlg);
50
51
52 /*
53  * ROUTINES ___________________________________________________________________
54  *
55  */
56
57 BOOL CALLBACK Main_DlgProc (HWND hDlg, UINT msg, WPARAM wp, LPARAM lp)
58 {
59    static UINT msgCheckTerminate = 0;
60    if (msgCheckTerminate == 0)
61       msgCheckTerminate = RegisterWindowMessage (TEXT("AfsCredsCheckTerminate"));
62
63    if (msg == msgCheckTerminate)
64       {
65       Main_OnCheckTerminate();
66       }
67    else switch (msg)
68       {
69       case WM_INITDIALOG:
70          g.hMain = hDlg;
71          Main_OnInitDialog (hDlg);
72          break;
73
74       case WM_DESTROY:
75          Creds_CloseLibraries();
76          ChangeTrayIcon (NIM_DELETE);
77          break;
78
79       case WM_ACTIVATEAPP:
80          if (wp)
81             {
82             Main_RepopulateTabs (FALSE);
83             }
84          break;
85
86       case WM_COMMAND:
87          switch (LOWORD(wp))
88             {
89             case IDOK:
90             case IDCANCEL:
91                Main_Show (FALSE);
92                break;
93
94             case M_ACTIVATE:
95                if (g.fIsWinNT || IsServiceRunning())
96                   {
97                   if (!lp) // Got here from "/show" parameter? switch tabs.
98                      {
99                      HWND hTab = GetDlgItem (g.hMain, IDC_TABS);
100                      TabCtrl_SetCurSel (hTab, 0);
101                      Main_OnSelectTab();
102                      }
103                   Main_Show (TRUE);
104                   }
105                else
106                   {
107                   Message (MB_ICONHAND, IDS_UNCONFIG_TITLE_95, IDS_UNCONFIG_DESC_95);
108                   }
109                break;
110
111             case M_TERMINATE:
112                if (g.fIsWinNT && IsServiceRunning())
113                   ModalDialog (IDD_TERMINATE, NULL, (DLGPROC)Terminate_DlgProc);
114                else if (g.fIsWinNT)
115                   ModalDialog (IDD_TERMINATE_SMALL, NULL, (DLGPROC)Terminate_DlgProc);
116                else // (!g.fIsWinNT)
117                   ModalDialog (IDD_TERMINATE_SMALL_95, NULL, (DLGPROC)Terminate_DlgProc);
118                break;
119
120             case M_TERMINATE_NOW:
121                Quit();
122                break;
123
124             case M_REMIND:
125                Main_OnCheckMenuRemind();
126                break;
127             }
128          break;
129
130       case WM_TIMER:
131          Main_OnRemindTimer();
132          break;
133
134       case WM_NOTIFY:
135          switch (((NMHDR*)lp)->code)
136             {
137             case TCN_SELCHANGE:
138                Main_OnSelectTab();
139                break;
140             }
141          break;
142
143       case WM_TRAYICON:
144          switch (lp)
145             {
146             case WM_LBUTTONDOWN:
147                if (IsServiceRunning() || !IsServiceConfigured())
148                   Main_Show (TRUE);
149                else if (!g.fIsWinNT)
150                   Message (MB_ICONHAND, IDS_UNCONFIG_TITLE_95, IDS_UNCONFIG_DESC_95);
151                else
152                   ShowStartupWizard();
153                break;
154
155             case WM_RBUTTONDOWN:
156                HMENU hm;
157                if ((hm = TaLocale_LoadMenu (MENU_TRAYICON)) != 0)
158                   {
159                   POINT pt;
160                   GetCursorPos(&pt);
161
162                   HMENU hmDummy = CreateMenu();
163                   InsertMenu (hmDummy, 0, MF_POPUP, (UINT)hm, NULL);
164
165                   BOOL fRemind = FALSE;
166                   lock_ObtainMutex(&g.credsLock);
167                   for (size_t iCreds = 0; iCreds < g.cCreds; ++iCreds)
168                      {
169                      if (g.aCreds[ iCreds ].fRemind)
170                         fRemind = TRUE;
171                      }
172                   lock_ReleaseMutex(&g.credsLock);
173                   CheckMenuItem (hm, M_REMIND, MF_BYCOMMAND | ((fRemind) ? MF_CHECKED : MF_UNCHECKED));
174
175                   TrackPopupMenu (GetSubMenu (hmDummy, 0),
176                                   TPM_RIGHTALIGN | TPM_RIGHTBUTTON,
177                                   pt.x, pt.y, NULL, hDlg, NULL);
178
179                   DestroyMenu (hmDummy);
180                   }
181                break;
182
183             case WM_MOUSEMOVE:
184                Main_OnMouseOver();
185                break;
186             }
187          break;
188       case WM_OBTAIN_TOKENS:
189           if ( InterlockedIncrement (&g.fShowingMessage) != 1 )
190               InterlockedDecrement (&g.fShowingMessage);
191           else
192               ShowObtainCreds (wp, (char *)lp);
193           GlobalFree((void *)lp);
194           break;
195
196       case WM_START_SERVICE:
197           {
198               SC_HANDLE hManager;
199               if ((hManager = OpenSCManager ( NULL, NULL, 
200                                               SC_MANAGER_CONNECT |
201                                               SC_MANAGER_ENUMERATE_SERVICE |
202                                               SC_MANAGER_QUERY_LOCK_STATUS)) != NULL)
203               {
204                   SC_HANDLE hService;
205                   if ((hService = OpenService ( hManager, TEXT("TransarcAFSDaemon"), 
206                                                 SERVICE_QUERY_STATUS | SERVICE_START)) != NULL)
207                   {
208                       if (StartService (hService, 0, 0))
209                           TestAndDoMapShare(SERVICE_START_PENDING);
210                                   if ( KFW_is_available() && KFW_AFS_wait_for_service_start() ) {
211                                           KFW_AFS_renew_tokens_for_all_cells();
212                                                   }
213
214                       CloseServiceHandle (hService);
215                   }
216
217                   CloseServiceHandle (hManager);
218               }
219               KFW_AFS_wait_for_service_start();
220               ObtainTokensFromUserIfNeeded(g.hMain);
221           }
222           break;
223       }
224
225    return FALSE;
226 }
227
228
229 void Main_Show (BOOL fShow)
230 {
231    if (fShow)
232       {
233       ShowWindow (g.hMain, SW_SHOW);
234       SetWindowPos (g.hMain, HWND_TOP, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE);
235       SetForegroundWindow (g.hMain);
236       }
237    else
238       {
239       SetWindowPos (g.hMain, HWND_BOTTOM, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE | SWP_HIDEWINDOW);
240       }
241 }
242
243
244 void Main_OnInitDialog (HWND hDlg)
245 {
246    if (!g.fIsWinNT)
247       {
248       TCHAR szTitle[256];
249       GetString (szTitle, IDS_TITLE_95);
250       SetWindowText (hDlg, szTitle);
251       }
252
253    TCHAR szVersion[256];
254    DWORD dwPatch = 0;
255    TCHAR szUser[256];
256    GetString (szVersion, IDS_UNKNOWN);
257    GetString (szUser, IDS_UNKNOWN);
258
259    HKEY hk;
260    if (RegOpenKey (HKEY_LOCAL_MACHINE, REGSTR_PATH_AFS, &hk) == 0)
261       {
262       DWORD dwSize = sizeof(szVersion);
263       DWORD dwType = REG_SZ;
264       RegQueryValueEx (hk, REGVAL_AFS_VERSION, NULL, &dwType, (PBYTE)szVersion, &dwSize);
265
266       dwSize = sizeof(dwPatch);
267       dwType = REG_DWORD;
268       RegQueryValueEx (hk, REGVAL_AFS_PATCH, NULL, &dwType, (PBYTE)&dwPatch, &dwSize);
269       RegCloseKey (hk);
270       }
271
272    BOOL fFoundUserName = FALSE;
273    if (RegOpenKey (HKEY_LOCAL_MACHINE, TEXT("Software\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon"), &hk) == 0)
274       {
275       DWORD dwSize = sizeof(szUser);
276       DWORD dwType = REG_SZ;
277       if (RegQueryValueEx (hk, TEXT("DefaultUserName"), NULL, &dwType, (PBYTE)szUser, &dwSize) == 0)
278          fFoundUserName = TRUE;
279       RegCloseKey (hk);
280       }
281    if (!fFoundUserName)
282       {
283       if (RegOpenKey (HKEY_LOCAL_MACHINE, TEXT("Network\\Logon"), &hk) == 0)
284          {
285          DWORD dwSize = sizeof(szUser);
286          DWORD dwType = REG_SZ;
287          if (RegQueryValueEx (hk, TEXT("UserName"), NULL, &dwType, (PBYTE)szUser, &dwSize) == 0)
288             fFoundUserName = TRUE;
289          RegCloseKey (hk);
290          }
291       }
292
293    TCHAR szSource[ cchRESOURCE ];
294    TCHAR szTarget[ cchRESOURCE ];
295
296    GetString (szSource, (dwPatch) ? IDS_TITLE_VERSION : IDS_TITLE_VERSION_NOPATCH);
297    wsprintf (szTarget, szSource, szVersion, dwPatch);
298    SetDlgItemText (hDlg, IDC_TITLE_VERSION, szTarget);
299
300    GetDlgItemText (hDlg, IDC_TITLE_NT, szSource, cchRESOURCE);
301    wsprintf (szTarget, szSource, szUser);
302    SetDlgItemText (hDlg, IDC_TITLE_NT, szTarget);
303 }
304
305
306 void Main_OnCheckMenuRemind (void)
307 {
308    BOOL fRemind = FALSE;
309    lock_ObtainMutex(&g.credsLock);
310    for (size_t iCreds = 0; iCreds < g.cCreds; ++iCreds)
311       {
312       if (g.aCreds[ iCreds ].fRemind)
313          fRemind = TRUE;
314       }
315
316    fRemind = !fRemind;
317    for (iCreds = 0; iCreds < g.cCreds; ++iCreds)
318       {
319       if (g.aCreds[ iCreds ].fRemind != fRemind)
320          {
321          g.aCreds[ iCreds ].fRemind = fRemind;
322          SaveRemind (iCreds);
323          }
324       }
325    lock_ReleaseMutex(&g.credsLock);
326
327    // Check the active tab, and fix its checkbox if necessary
328    //
329    HWND hTab = GetDlgItem (g.hMain, IDC_TABS);
330    LPTSTR pszTab = (LPTSTR)GetTabParam (hTab, TabCtrl_GetCurSel(hTab));
331    if (ISCELLTAB(pszTab) && (*pszTab))
332       {
333       HWND hDlg = GetTabChild (hTab);
334       if (hDlg)
335          CheckDlgButton (hDlg, IDC_CREDS_REMIND, fRemind);
336       }
337
338    // Make sure the reminder timer is going
339    //
340    Main_EnableRemindTimer (fRemind);
341 }
342
343
344 void Main_OnRemindTimer (void)
345 {
346    Main_RepopulateTabs (TRUE);
347
348    // See if anything is close to expiring; if so, display a warning
349    // dialog. Make sure we never display a warning more than once.
350    //
351    size_t iExpired;
352    if ((iExpired = Main_FindExpiredCreds()) != -1)
353       {
354       if (InterlockedIncrement (&g.fShowingMessage) != 1)
355          InterlockedDecrement (&g.fShowingMessage);
356       else
357          ShowObtainCreds (TRUE, g.aCreds[ iExpired ].szCell);
358       }
359 }
360
361
362 void Main_OnMouseOver (void)
363 {
364    if ((GetTickCount() - g.tickLastRetest) > cmsecMOUSEOVER)
365       {
366       Main_RepopulateTabs (TRUE);
367       }
368 }
369
370
371 void Main_OnSelectTab (void)
372 {
373    HWND hTab = GetDlgItem (g.hMain, IDC_TABS);
374    size_t iTab = TabCtrl_GetCurSel (hTab);
375
376    HWND hDlgOld = GetTabChild (hTab);
377
378    HWND hDlgNew;
379    if ((hDlgNew = Main_CreateTabDialog (hTab, iTab)) != NULL)
380       ShowWindow (hDlgNew, SW_SHOW);
381
382    if (hDlgOld)
383       DestroyWindow (hDlgOld);
384 }
385
386
387 void Main_OnCheckTerminate (void)
388 {
389    HKEY hk;
390    if (RegOpenKey (HKEY_LOCAL_MACHINE, TEXT("System\\CurrentControlSet\\Services\\TransarcAFSDaemon\\Parameters"), &hk) == 0)
391       {
392       DWORD dwSize = sizeof(g.fStartup);
393       DWORD dwType = REG_DWORD;
394       RegQueryValueEx (hk, TEXT("ShowTrayIcon"), NULL, &dwType, (PBYTE)&g.fStartup, &dwSize);
395       RegCloseKey (hk);
396       }
397
398    Shortcut_FixStartup (cszSHORTCUT_NAME, g.fStartup);
399
400    if (!g.fStartup)
401       Quit();
402 }
403
404
405 HWND Main_CreateTabDialog (HWND hTab, size_t iTab)
406 {
407    HWND hDlg = NULL;
408    LPTSTR psz = NULL;
409
410    TC_ITEM Item;
411    memset (&Item, 0x00, sizeof(Item));
412    Item.mask = TCIF_PARAM;
413    if (TabCtrl_GetItem (hTab, iTab, &Item))
414       {
415       psz = (LPTSTR)(Item.lParam);
416       }
417
418    if (psz == dwTABPARAM_ADVANCED)    // Advanced tab
419       {
420       hDlg = ModelessDialog (IDD_TAB_ADVANCED, hTab, (DLGPROC)Advanced_DlgProc);
421       }
422    else if (psz == dwTABPARAM_MOUNT)  // Mount Points tab
423       {
424       hDlg = ModelessDialog (IDD_TAB_MOUNT, hTab, (DLGPROC)Mount_DlgProc);
425       }
426    else if (ISCELLTAB(psz) && !*psz)  // No Creds tab
427       {
428       hDlg = ModelessDialogParam (IDD_TAB_NOCREDS, hTab, (DLGPROC)Creds_DlgProc, (LPARAM)psz);
429       }
430    else if (ISCELLTAB(psz) && *psz)   // Creds tab for a particular cell
431       {
432       hDlg = ModelessDialogParam (IDD_TAB_CREDS, hTab, (DLGPROC)Creds_DlgProc, (LPARAM)psz);
433       }
434
435    return hDlg;
436 }
437
438
439 void Main_RepopulateTabs (BOOL fDestroyInvalid)
440 {
441    static BOOL fInHere = FALSE;
442    if (!fInHere)
443       {
444       fInHere = TRUE;
445
446       if (IsWindowVisible (g.hMain))
447          fDestroyInvalid = FALSE;
448
449       // First we'll have to look around and see what credentials we currently
450       // have. This call just updates g.aCreds[]; it doesn't do anything else.
451       //
452       (void)GetCurrentCredentials();
453
454       // We want one tab on the main dialog for each g.aCredentials entry,
455       // and one tab for Advanced.
456       //
457       HWND hTab = GetDlgItem (g.hMain, IDC_TABS);
458
459       // Generate a list of the lParams we'll be giving tabs...
460       //
461       LPTSTR *aTabs = NULL;
462       size_t cTabs = 0;
463       size_t iTabOut = 0;
464
465       size_t nCreds = 0;
466       lock_ObtainMutex(&g.credsLock);
467       for (size_t iCreds = 0; iCreds < g.cCreds; ++iCreds)
468          {
469          if (g.aCreds[ iCreds ].szCell[0])
470             ++nCreds;
471          }
472       if (!nCreds)
473          {
474          fDestroyInvalid = TRUE;
475          }
476
477       if (!fDestroyInvalid)
478          {
479          int nTabs = TabCtrl_GetItemCount(hTab);
480          for (int iTab = 0; iTab < nTabs; ++iTab)
481             {
482             LPTSTR pszTab = (LPTSTR)GetTabParam (hTab, iTab);
483             if (ISCELLTAB(pszTab) && (*pszTab))
484                {
485                if (REALLOC (aTabs, cTabs, 1+iTabOut, cREALLOC_TABS))
486                   aTabs[ iTabOut++ ] = CloneString(pszTab);
487                }
488             }
489          }
490
491       if (nCreds == 0)
492          {
493          if (REALLOC (aTabs, cTabs, 1+iTabOut, cREALLOC_TABS))
494             aTabs[ iTabOut++ ] = CloneString (TEXT(""));
495          }
496       else for (iCreds = 0; iCreds < g.cCreds; ++iCreds)
497          {
498          if (g.aCreds[ iCreds ].szCell[0])
499             {
500             for (size_t ii = 0; ii < iTabOut; ++ii)
501                {
502                if (!ISCELLTAB (aTabs[ii]))
503                   continue;
504                if (!lstrcmpi (g.aCreds[ iCreds ].szCell, aTabs[ ii ]))
505                   break;
506                }
507             if (ii == iTabOut)
508                {
509                if (REALLOC (aTabs, cTabs, 1+iTabOut, cREALLOC_TABS))
510                   aTabs[ iTabOut++ ] = CloneString (g.aCreds[ iCreds ].szCell);
511                }
512             }
513          }
514       lock_ReleaseMutex(&g.credsLock);
515
516       if (REALLOC (aTabs, cTabs, 1+iTabOut, cREALLOC_TABS))
517          aTabs[ iTabOut++ ] = dwTABPARAM_MOUNT;
518
519       if (g.fIsWinNT)
520          {
521          if (REALLOC (aTabs, cTabs, 1+iTabOut, cREALLOC_TABS))
522             aTabs[ iTabOut++ ] = dwTABPARAM_ADVANCED;
523          }
524
525       // Now erase the current tabs, and re-add new ones. Remember which tab is
526       // currently selected, so we can try to go back to it later.
527       //
528       int iTabSel = 0;
529       if (TabCtrl_GetItemCount(hTab))
530          {
531          LPTSTR pszTabSel = (LPTSTR)GetTabParam (hTab, TabCtrl_GetCurSel(hTab));
532          for (size_t iSel = 0; iSel < iTabOut; ++iSel)
533             {
534             if ((!ISCELLTAB(pszTabSel))  && (!ISCELLTAB(aTabs[iSel])) && (pszTabSel == aTabs[iSel]))
535                iTabSel = iSel;
536             else if (ISCELLTAB(pszTabSel) && ISCELLTAB(aTabs[iSel]) && !lstrcmpi (pszTabSel, aTabs[iSel]))
537                iTabSel = iSel;
538             }
539          }
540
541       int nTabs = TabCtrl_GetItemCount(hTab);
542       for (int iTab = 0; iTab < nTabs; ++iTab)
543          {
544          LPTSTR pszTab = (LPTSTR)GetTabParam (hTab, iTab);
545          if (ISCELLTAB(pszTab))
546             FreeString (pszTab);
547          }
548       TabCtrl_DeleteAllItems (hTab);
549
550       for (size_t ii = 0; ii < iTabOut; ++ii)
551          {
552          TCHAR szTitle[cchRESOURCE];
553          if (aTabs[ii] == dwTABPARAM_ADVANCED)
554             GetString (szTitle, IDS_ADVANCED);
555          else if (aTabs[ii] == dwTABPARAM_MOUNT)
556             GetString (szTitle, IDS_MOUNT);
557          else if ((nCreds <= 1) || (aTabs[ii][0] == TEXT('\0')))
558             GetString (szTitle, IDS_CREDENTIALS);
559          else
560             lstrcpy (szTitle, aTabs[ii]);
561
562          TC_ITEM Item;
563          memset (&Item, 0x00, sizeof(Item));
564          Item.mask = TCIF_PARAM | TCIF_TEXT;
565          Item.pszText = szTitle;
566          Item.cchTextMax = cchRESOURCE;
567          Item.lParam = (LPARAM)(aTabs[ii]);
568
569          TabCtrl_InsertItem (hTab, ii, &Item);
570          }
571
572       if (aTabs != NULL)
573          Free (aTabs);
574
575       TabCtrl_SetCurSel (hTab, iTabSel);
576       Main_OnSelectTab ();
577
578       fInHere = FALSE;
579       }
580 }
581
582
583 void Main_EnableRemindTimer (BOOL fEnable)
584 {
585    static BOOL bEnabled = FALSE;
586
587    if ( fEnable == FALSE && bEnabled == TRUE ) {
588        KillTimer (g.hMain, ID_REMIND_TIMER);
589        bEnabled = FALSE;
590    } else if ( fEnable == TRUE && bEnabled == FALSE ) {
591       SetTimer (g.hMain, ID_REMIND_TIMER, (ULONG)cmsec1MINUTE * cminREMIND_TEST, NULL);
592       bEnabled = TRUE;
593    }
594 }
595
596
597 size_t Main_FindExpiredCreds (void)
598 {
599    size_t retval = (size_t) -1;
600    lock_ObtainMutex(&g.expirationCheckLock);
601    if ( KFW_is_available() )
602        KFW_AFS_renew_expiring_credentials();
603    lock_ObtainMutex(&g.credsLock);
604    for (size_t iCreds = 0; iCreds < g.cCreds; ++iCreds)
605       {
606       if (!g.aCreds[ iCreds ].szCell[0])
607          continue;
608       if (!g.aCreds[ iCreds ].fRemind)
609          continue;
610
611       SYSTEMTIME stNow;
612       GetLocalTime (&stNow);
613
614       FILETIME ftNow;
615       SystemTimeToFileTime (&stNow, &ftNow);
616
617       FILETIME ftExpires;
618       SystemTimeToFileTime (&g.aCreds[ iCreds ].stExpires, &ftExpires);
619
620       LONGLONG llNow = (((LONGLONG)ftNow.dwHighDateTime) << 32) + (LONGLONG)(ftNow.dwLowDateTime);
621       LONGLONG llExpires = (((LONGLONG)ftExpires.dwHighDateTime) << 32) + (LONGLONG)(ftExpires.dwLowDateTime);
622
623       llNow /= c100ns1SECOND;
624       llExpires /= c100ns1SECOND;
625
626       if (llExpires <= (llNow + (LONGLONG)cminREMIND_WARN * csec1MINUTE))
627          {
628          if ( KFW_is_available() &&
629               KFW_AFS_renew_token_for_cell(g.aCreds[ iCreds ].szCell) )
630              continue;
631          retval = (size_t) iCreds;
632          break;
633          }
634       }
635    
636    lock_ReleaseMutex(&g.credsLock);
637    lock_ReleaseMutex(&g.expirationCheckLock);
638
639    return retval;
640 }
641
642
643 BOOL CALLBACK Terminate_DlgProc (HWND hDlg, UINT msg, WPARAM wp, LPARAM lp)
644 {
645    switch (msg)
646       {
647       case WM_INITDIALOG:
648          Terminate_OnInitDialog (hDlg);
649          break;
650
651       case WM_COMMAND:
652          switch (LOWORD(wp))
653             {
654             case IDOK:
655                Terminate_OnOK (hDlg);
656                break;
657
658             case IDCANCEL:
659                EndDialog (hDlg, IDCANCEL);
660                break;
661             }
662          break;
663       }
664
665    return FALSE;
666 }
667
668
669 void Terminate_OnInitDialog (HWND hDlg)
670 {
671    BOOL fPersistent = IsServicePersistent();
672
673    CheckDlgButton (hDlg, IDC_STARTUP, g.fStartup);
674    CheckDlgButton (hDlg, IDC_LEAVE, fPersistent);
675    CheckDlgButton (hDlg, IDC_STOP, !fPersistent);
676 }
677
678
679 void Terminate_OnOK (HWND hDlg)
680 {
681    if (IsServiceRunning())
682       {
683       if (IsDlgButtonChecked (hDlg, IDC_STOP))
684          {
685          SC_HANDLE hManager;
686              if ((hManager = OpenSCManager (NULL, NULL, 
687                                             SC_MANAGER_CONNECT |
688                                             SC_MANAGER_ENUMERATE_SERVICE |
689                                             SC_MANAGER_QUERY_LOCK_STATUS)) != NULL)
690             {
691             SC_HANDLE hService;
692             if ((hService = OpenService (hManager, TEXT("TransarcAFSDaemon"), 
693                                          SERVICE_QUERY_STATUS | SERVICE_START)) != NULL)
694                {
695                SERVICE_STATUS Status;
696                ControlService (hService, SERVICE_CONTROL_STOP, &Status);
697
698                CloseServiceHandle (hService);
699                }
700
701             CloseServiceHandle (hManager);
702             }
703          }
704       }
705
706    g.fStartup = IsDlgButtonChecked (hDlg, IDC_STARTUP);
707
708    HKEY hk;
709    if (RegCreateKey (HKEY_LOCAL_MACHINE, TEXT("System\\CurrentControlSet\\Services\\TransarcAFSDaemon\\Parameters"), &hk) == 0)
710       {
711       DWORD dwSize = sizeof(g.fStartup);
712       DWORD dwType = REG_DWORD;
713       RegSetValueEx (hk, TEXT("ShowTrayIcon"), NULL, dwType, (PBYTE)&g.fStartup, dwSize);
714       RegCloseKey (hk);
715       }
716
717    Shortcut_FixStartup (cszSHORTCUT_NAME, g.fStartup);
718
719    Quit();
720    EndDialog (hDlg, IDOK);
721 }
722