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 #include <afs/afskfw.h>
  14 #include "ipaddrchg.h"
  15 }
  16
  17 #include "afscreds.h"
  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 #ifdef USE_MS2MIT
 212                               KFW_import_windows_lsa();
 213 #endif /* USE_MS2MIT */
 214                                           KFW_AFS_renew_tokens_for_all_cells();
 215                                                   }
 216
 217                       CloseServiceHandle (hService);
 218                   }
 219
 220                   CloseServiceHandle (hManager);
 221               }
 222               KFW_AFS_wait_for_service_start();
 223               ObtainTokensFromUserIfNeeded(g.hMain);
 224           }
 225           break;
 226       }
 227
 228    return FALSE;
 229 }
 230
 231
 232 void Main_Show (BOOL fShow)
 233 {
 234    if (fShow)
 235       {
 236       ShowWindow (g.hMain, SW_SHOW);
 237       SetWindowPos (g.hMain, HWND_TOP, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE);
 238       SetForegroundWindow (g.hMain);
 239       }
 240    else
 241       {
 242       SetWindowPos (g.hMain, HWND_BOTTOM, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE | SWP_HIDEWINDOW);
 243       }
 244 }
 245
 246
 247 void Main_OnInitDialog (HWND hDlg)
 248 {
 249     if (!g.fIsWinNT)
 250     {
 251         TCHAR szTitle[256];
 252         GetString (szTitle, IDS_TITLE_95);
 253         SetWindowText (hDlg, szTitle);
 254     }
 255
 256     TCHAR szVersion[256];
 257     DWORD dwPatch = 0;
 258     TCHAR szUser[256];
 259     GetString (szVersion, IDS_UNKNOWN);
 260     GetString (szUser, IDS_UNKNOWN);
 261
 262     HKEY hk;
 263     if (RegOpenKey (HKEY_LOCAL_MACHINE, AFSREG_CLT_SW_VERSION_SUBKEY, &hk) == 0)
 264     {
 265         DWORD dwSize = sizeof(szVersion);
 266         DWORD dwType = REG_SZ;
 267         RegQueryValueEx (hk, REGVAL_AFS_VERSION, NULL, &dwType, (PBYTE)szVersion, &dwSize);
 268
 269         dwSize = sizeof(dwPatch);
 270         dwType = REG_DWORD;
 271         RegQueryValueEx (hk, REGVAL_AFS_PATCH, NULL, &dwType, (PBYTE)&dwPatch, &dwSize);
 272         RegCloseKey (hk);
 273     }
 274
 275     /* We should probably be using GetUserNameEx() for this */
 276     BOOL fFoundUserName = FALSE;
 277     if (RegOpenKey (HKEY_CURRENT_USER, TEXT("Software\\Microsoft\\Windows\\CurrentVersion\\Explorer"), &hk) == 0)
 278     {
 279         DWORD dwSize = sizeof(szUser);
 280         DWORD dwType = REG_SZ;
 281         if (RegQueryValueEx (hk, TEXT("Logon User Name"), NULL, &dwType, (PBYTE)szUser, &dwSize) == 0)
 282             fFoundUserName = TRUE;
 283         RegCloseKey (hk);
 284     }
 285     if (!fFoundUserName )
 286     {
 287         if (RegOpenKey (HKEY_LOCAL_MACHINE, TEXT("Software\\Microsoft\\Windows NT\\CurrentVersion\\Explorer"), &hk) == 0)
 288         {
 289             DWORD dwSize = sizeof(szUser);
 290             DWORD dwType = REG_SZ;
 291             if (RegQueryValueEx (hk, TEXT("Logon User Name"), NULL, &dwType, (PBYTE)szUser, &dwSize) == 0)
 292                 fFoundUserName = TRUE;
 293             RegCloseKey (hk);
 294         }
 295     }
 296     if (!fFoundUserName )
 297     {
 298         if (RegOpenKey (HKEY_LOCAL_MACHINE, TEXT("Software\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon"), &hk) == 0)
 299         {
 300             DWORD dwSize = sizeof(szUser);
 301             DWORD dwType = REG_SZ;
 302             if (RegQueryValueEx (hk, TEXT("DefaultUserName"), NULL, &dwType, (PBYTE)szUser, &dwSize) == 0)
 303                 fFoundUserName = TRUE;
 304             RegCloseKey (hk);
 305         }
 306     }
 307     if (!fFoundUserName)
 308     {
 309         if (RegOpenKey (HKEY_LOCAL_MACHINE, TEXT("Network\\Logon"), &hk) == 0)
 310         {
 311             DWORD dwSize = sizeof(szUser);
 312             DWORD dwType = REG_SZ;
 313             if (RegQueryValueEx (hk, TEXT("UserName"), NULL, &dwType, (PBYTE)szUser, &dwSize) == 0)
 314                 fFoundUserName = TRUE;
 315             RegCloseKey (hk);
 316         }
 317     }
 318
 319     TCHAR szSource[ cchRESOURCE ];
 320     TCHAR szTarget[ cchRESOURCE ];
 321
 322     GetString (szSource, (dwPatch) ? IDS_TITLE_VERSION : IDS_TITLE_VERSION_NOPATCH);
 323     wsprintf (szTarget, szSource, szVersion, dwPatch);
 324     SetDlgItemText (hDlg, IDC_TITLE_VERSION, szTarget);
 325
 326     GetDlgItemText (hDlg, IDC_TITLE_NT, szSource, cchRESOURCE);
 327     wsprintf (szTarget, szSource, szUser);
 328     SetDlgItemText (hDlg, IDC_TITLE_NT, szTarget);
 329 }
 330
 331
 332 void Main_OnCheckMenuRemind (void)
 333 {
 334    BOOL fRemind = FALSE;
 335    lock_ObtainMutex(&g.credsLock);
 336    for (size_t iCreds = 0; iCreds < g.cCreds; ++iCreds)
 337       {
 338       if (g.aCreds[ iCreds ].fRemind)
 339          fRemind = TRUE;
 340       }
 341
 342    fRemind = !fRemind;
 343    for (iCreds = 0; iCreds < g.cCreds; ++iCreds)
 344       {
 345       if (g.aCreds[ iCreds ].fRemind != fRemind)
 346          {
 347          g.aCreds[ iCreds ].fRemind = fRemind;
 348          SaveRemind (iCreds);
 349          }
 350       }
 351    lock_ReleaseMutex(&g.credsLock);
 352
 353    // Check the active tab, and fix its checkbox if necessary
 354    //
 355    HWND hTab = GetDlgItem (g.hMain, IDC_TABS);
 356    LPTSTR pszTab = (LPTSTR)GetTabParam (hTab, TabCtrl_GetCurSel(hTab));
 357    if (ISCELLTAB(pszTab) && (*pszTab))
 358       {
 359       HWND hDlg = GetTabChild (hTab);
 360       if (hDlg)
 361          CheckDlgButton (hDlg, IDC_CREDS_REMIND, fRemind);
 362       }
 363
 364    // Make sure the reminder timer is going
 365    //
 366    Main_EnableRemindTimer (fRemind);
 367 }
 368
 369
 370 void Main_OnRemindTimer (void)
 371 {
 372    Main_RepopulateTabs (TRUE);
 373
 374    // See if anything is close to expiring; if so, display a warning
 375    // dialog. Make sure we never display a warning more than once.
 376    //
 377    size_t iExpired;
 378    if ((iExpired = Main_FindExpiredCreds()) != -1)
 379       {
 380       if (InterlockedIncrement (&g.fShowingMessage) != 1)
 381          InterlockedDecrement (&g.fShowingMessage);
 382       else
 383          ShowObtainCreds (TRUE, g.aCreds[ iExpired ].szCell);
 384       }
 385 }
 386
 387
 388 void Main_OnMouseOver (void)
 389 {
 390    if ((GetTickCount() - g.tickLastRetest) > cmsecMOUSEOVER)
 391       {
 392       Main_RepopulateTabs (TRUE);
 393       }
 394 }
 395
 396
 397 void Main_OnSelectTab (void)
 398 {
 399    HWND hTab = GetDlgItem (g.hMain, IDC_TABS);
 400    size_t iTab = TabCtrl_GetCurSel (hTab);
 401
 402    HWND hDlgOld = GetTabChild (hTab);
 403
 404    HWND hDlgNew;
 405    if ((hDlgNew = Main_CreateTabDialog (hTab, iTab)) != NULL)
 406       ShowWindow (hDlgNew, SW_SHOW);
 407
 408    if (hDlgOld)
 409       DestroyWindow (hDlgOld);
 410 }
 411
 412
 413 void Main_OnCheckTerminate (void)
 414 {
 415     HKEY hk;
 416
 417     if (RegOpenKey (HKEY_CURRENT_USER, AFSREG_USER_OPENAFS_SUBKEY, &hk) == 0)
 418     {
 419         DWORD dwSize = sizeof(g.fStartup);
 420         DWORD dwType = REG_DWORD;
 421         RegQueryValueEx (hk, TEXT("ShowTrayIcon"), NULL, &dwType, (PBYTE)&g.fStartup, &dwSize);
 422         RegCloseKey (hk);
 423     }
 424     else if (RegOpenKey (HKEY_LOCAL_MACHINE, AFSREG_CLT_OPENAFS_SUBKEY, &hk) == 0)
 425     {
 426       DWORD dwSize = sizeof(g.fStartup);
 427       DWORD dwType = REG_DWORD;
 428       RegQueryValueEx (hk, TEXT("ShowTrayIcon"), NULL, &dwType, (PBYTE)&g.fStartup, &dwSize);
 429       RegCloseKey (hk);
 430     }
 431
 432     Shortcut_FixStartup (cszSHORTCUT_NAME, g.fStartup);
 433
 434     if (!g.fStartup)
 435         Quit();
 436 }
 437
 438
 439 HWND Main_CreateTabDialog (HWND hTab, size_t iTab)
 440 {
 441    HWND hDlg = NULL;
 442    LPTSTR psz = NULL;
 443
 444    TC_ITEM Item;
 445    memset (&Item, 0x00, sizeof(Item));
 446    Item.mask = TCIF_PARAM;
 447    if (TabCtrl_GetItem (hTab, iTab, &Item))
 448       {
 449       psz = (LPTSTR)(Item.lParam);
 450       }
 451
 452    if (psz == dwTABPARAM_ADVANCED)    // Advanced tab
 453       {
 454       hDlg = ModelessDialog (IDD_TAB_ADVANCED, hTab, (DLGPROC)Advanced_DlgProc);
 455       }
 456    else if (psz == dwTABPARAM_MOUNT)  // Mount Points tab
 457       {
 458       hDlg = ModelessDialog (IDD_TAB_MOUNT, hTab, (DLGPROC)Mount_DlgProc);
 459       }
 460    else if (ISCELLTAB(psz) && !*psz)  // No Creds tab
 461       {
 462       hDlg = ModelessDialogParam (IDD_TAB_NOCREDS, hTab, (DLGPROC)Creds_DlgProc, (LPARAM)psz);
 463       }
 464    else if (ISCELLTAB(psz) && *psz)   // Creds tab for a particular cell
 465       {
 466       hDlg = ModelessDialogParam (IDD_TAB_CREDS, hTab, (DLGPROC)Creds_DlgProc, (LPARAM)psz);
 467       }
 468
 469    return hDlg;
 470 }
 471
 472
 473 void Main_RepopulateTabs (BOOL fDestroyInvalid)
 474 {
 475    static BOOL fInHere = FALSE;
 476    if (!fInHere)
 477       {
 478       fInHere = TRUE;
 479
 480       if (IsWindowVisible (g.hMain))
 481          fDestroyInvalid = FALSE;
 482
 483       // First we'll have to look around and see what credentials we currently
 484       // have. This call just updates g.aCreds[]; it doesn't do anything else.
 485       //
 486       (void)GetCurrentCredentials();
 487
 488       // We want one tab on the main dialog for each g.aCredentials entry,
 489       // and one tab for Advanced.
 490       //
 491       HWND hTab = GetDlgItem (g.hMain, IDC_TABS);
 492
 493       // Generate a list of the lParams we'll be giving tabs...
 494       //
 495       LPTSTR *aTabs = NULL;
 496       size_t cTabs = 0;
 497       size_t iTabOut = 0;
 498
 499       size_t nCreds = 0;
 500       lock_ObtainMutex(&g.credsLock);
 501       for (size_t iCreds = 0; iCreds < g.cCreds; ++iCreds)
 502          {
 503          if (g.aCreds[ iCreds ].szCell[0])
 504             ++nCreds;
 505          }
 506       if (!nCreds)
 507          {
 508          fDestroyInvalid = TRUE;
 509          }
 510
 511       if (!fDestroyInvalid)
 512          {
 513          int nTabs = TabCtrl_GetItemCount(hTab);
 514          for (int iTab = 0; iTab < nTabs; ++iTab)
 515             {
 516             LPTSTR pszTab = (LPTSTR)GetTabParam (hTab, iTab);
 517             if (ISCELLTAB(pszTab) && (*pszTab))
 518                {
 519                if (REALLOC (aTabs, cTabs, 1+iTabOut, cREALLOC_TABS))
 520                   aTabs[ iTabOut++ ] = CloneString(pszTab);
 521                }
 522             }
 523          }
 524
 525       if (nCreds == 0)
 526          {
 527          if (REALLOC (aTabs, cTabs, 1+iTabOut, cREALLOC_TABS))
 528             aTabs[ iTabOut++ ] = CloneString (TEXT(""));
 529          }
 530       else for (iCreds = 0; iCreds < g.cCreds; ++iCreds)
 531          {
 532          if (g.aCreds[ iCreds ].szCell[0])
 533             {
 534             for (size_t ii = 0; ii < iTabOut; ++ii)
 535                {
 536                if (!ISCELLTAB (aTabs[ii]))
 537                   continue;
 538                if (!lstrcmpi (g.aCreds[ iCreds ].szCell, aTabs[ ii ]))
 539                   break;
 540                }
 541             if (ii == iTabOut)
 542                {
 543                if (REALLOC (aTabs, cTabs, 1+iTabOut, cREALLOC_TABS))
 544                   aTabs[ iTabOut++ ] = CloneString (g.aCreds[ iCreds ].szCell);
 545                }
 546             }
 547          }
 548       lock_ReleaseMutex(&g.credsLock);
 549
 550       if (REALLOC (aTabs, cTabs, 1+iTabOut, cREALLOC_TABS))
 551          aTabs[ iTabOut++ ] = dwTABPARAM_MOUNT;
 552
 553       if (g.fIsWinNT)
 554          {
 555          if (REALLOC (aTabs, cTabs, 1+iTabOut, cREALLOC_TABS))
 556             aTabs[ iTabOut++ ] = dwTABPARAM_ADVANCED;
 557          }
 558
 559       // Now erase the current tabs, and re-add new ones. Remember which tab is
 560       // currently selected, so we can try to go back to it later.
 561       //
 562       int iTabSel = 0;
 563       if (TabCtrl_GetItemCount(hTab))
 564          {
 565          LPTSTR pszTabSel = (LPTSTR)GetTabParam (hTab, TabCtrl_GetCurSel(hTab));
 566          for (size_t iSel = 0; iSel < iTabOut; ++iSel)
 567             {
 568             if ((!ISCELLTAB(pszTabSel))  && (!ISCELLTAB(aTabs[iSel])) && (pszTabSel == aTabs[iSel]))
 569                iTabSel = iSel;
 570             else if (ISCELLTAB(pszTabSel) && ISCELLTAB(aTabs[iSel]) && !lstrcmpi (pszTabSel, aTabs[iSel]))
 571                iTabSel = iSel;
 572             }
 573          }
 574
 575       int nTabs = TabCtrl_GetItemCount(hTab);
 576       for (int iTab = 0; iTab < nTabs; ++iTab)
 577          {
 578          LPTSTR pszTab = (LPTSTR)GetTabParam (hTab, iTab);
 579          if (ISCELLTAB(pszTab))
 580             FreeString (pszTab);
 581          }
 582       TabCtrl_DeleteAllItems (hTab);
 583
 584       for (size_t ii = 0; ii < iTabOut; ++ii)
 585          {
 586          TCHAR szTitle[cchRESOURCE];
 587          if (aTabs[ii] == dwTABPARAM_ADVANCED)
 588             GetString (szTitle, IDS_ADVANCED);
 589          else if (aTabs[ii] == dwTABPARAM_MOUNT)
 590             GetString (szTitle, IDS_MOUNT);
 591          else if ((nCreds <= 1) || (aTabs[ii][0] == TEXT('\0')))
 592             GetString (szTitle, IDS_CREDENTIALS);
 593          else
 594             lstrcpy (szTitle, aTabs[ii]);
 595
 596          TC_ITEM Item;
 597          memset (&Item, 0x00, sizeof(Item));
 598          Item.mask = TCIF_PARAM | TCIF_TEXT;
 599          Item.pszText = szTitle;
 600          Item.cchTextMax = cchRESOURCE;
 601          Item.lParam = (LPARAM)(aTabs[ii]);
 602
 603          TabCtrl_InsertItem (hTab, ii, &Item);
 604          }
 605
 606       if (aTabs != NULL)
 607          Free (aTabs);
 608
 609       TabCtrl_SetCurSel (hTab, iTabSel);
 610       Main_OnSelectTab ();
 611
 612       fInHere = FALSE;
 613       }
 614 }
 615
 616
 617 void Main_EnableRemindTimer (BOOL fEnable)
 618 {
 619    static BOOL bEnabled = FALSE;
 620
 621    if ( fEnable == FALSE && bEnabled == TRUE ) {
 622        KillTimer (g.hMain, ID_REMIND_TIMER);
 623        bEnabled = FALSE;
 624    } else if ( fEnable == TRUE && bEnabled == FALSE ) {
 625       SetTimer (g.hMain, ID_REMIND_TIMER, (ULONG)cmsec1MINUTE * cminREMIND_TEST, NULL);
 626       bEnabled = TRUE;
 627    }
 628 }
 629
 630
 631 size_t Main_FindExpiredCreds (void)
 632 {
 633    size_t retval = (size_t) -1;
 634    lock_ObtainMutex(&g.expirationCheckLock);
 635    if ( KFW_is_available() )
 636        KFW_AFS_renew_expiring_tokens();
 637    lock_ObtainMutex(&g.credsLock);
 638    for (size_t iCreds = 0; iCreds < g.cCreds; ++iCreds)
 639       {
 640       if (!g.aCreds[ iCreds ].szCell[0])
 641          continue;
 642       if (!g.aCreds[ iCreds ].fRemind)
 643          continue;
 644
 645       SYSTEMTIME stNow;
 646       GetLocalTime (&stNow);
 647
 648       FILETIME ftNow;
 649       SystemTimeToFileTime (&stNow, &ftNow);
 650
 651       FILETIME ftExpires;
 652       SystemTimeToFileTime (&g.aCreds[ iCreds ].stExpires, &ftExpires);
 653
 654       LONGLONG llNow = (((LONGLONG)ftNow.dwHighDateTime) << 32) + (LONGLONG)(ftNow.dwLowDateTime);
 655       LONGLONG llExpires = (((LONGLONG)ftExpires.dwHighDateTime) << 32) + (LONGLONG)(ftExpires.dwLowDateTime);
 656
 657       llNow /= c100ns1SECOND;
 658       llExpires /= c100ns1SECOND;
 659
 660       if (llExpires <= (llNow + (LONGLONG)cminREMIND_WARN * csec1MINUTE))
 661          {
 662          if ( KFW_is_available() &&
 663               KFW_AFS_renew_token_for_cell(g.aCreds[ iCreds ].szCell) )
 664              continue;
 665          retval = (size_t) iCreds;
 666          break;
 667          }
 668       }
 669
 670    lock_ReleaseMutex(&g.credsLock);
 671    lock_ReleaseMutex(&g.expirationCheckLock);
 672
 673    return retval;
 674 }
 675
 676
 677 BOOL CALLBACK Terminate_DlgProc (HWND hDlg, UINT msg, WPARAM wp, LPARAM lp)
 678 {
 679    switch (msg)
 680       {
 681       case WM_INITDIALOG:
 682          Terminate_OnInitDialog (hDlg);
 683          break;
 684
 685       case WM_COMMAND:
 686          switch (LOWORD(wp))
 687             {
 688             case IDOK:
 689                Terminate_OnOK (hDlg);
 690                break;
 691
 692             case IDCANCEL:
 693                EndDialog (hDlg, IDCANCEL);
 694                break;
 695             }
 696          break;
 697       }
 698
 699    return FALSE;
 700 }
 701
 702
 703 void Terminate_OnInitDialog (HWND hDlg)
 704 {
 705    BOOL fPersistent = IsServicePersistent();
 706
 707    CheckDlgButton (hDlg, IDC_STARTUP, g.fStartup);
 708    CheckDlgButton (hDlg, IDC_LEAVE, fPersistent);
 709    CheckDlgButton (hDlg, IDC_STOP, !fPersistent);
 710 }
 711
 712
 713 void Terminate_OnOK (HWND hDlg)
 714 {
 715    if (IsServiceRunning())
 716       {
 717       if (IsDlgButtonChecked (hDlg, IDC_STOP))
 718          {
 719          SC_HANDLE hManager;
 720              if ((hManager = OpenSCManager (NULL, NULL,
 721                                             SC_MANAGER_CONNECT |
 722                                             SC_MANAGER_ENUMERATE_SERVICE |
 723                                             SC_MANAGER_QUERY_LOCK_STATUS)) != NULL)
 724             {
 725             SC_HANDLE hService;
 726             if ((hService = OpenService (hManager, TEXT("TransarcAFSDaemon"),
 727                                          SERVICE_QUERY_STATUS | SERVICE_START)) != NULL)
 728                {
 729                SERVICE_STATUS Status;
 730                ControlService (hService, SERVICE_CONTROL_STOP, &Status);
 731
 732                CloseServiceHandle (hService);
 733                }
 734
 735             CloseServiceHandle (hManager);
 736             }
 737          }
 738       }
 739
 740    g.fStartup = IsDlgButtonChecked (hDlg, IDC_STARTUP);
 741
 742     HKEY hk;
 743     if (RegCreateKey (HKEY_CURRENT_USER, AFSREG_USER_OPENAFS_SUBKEY, &hk) == 0)
 744     {
 745         DWORD dwSize = sizeof(g.fStartup);
 746         DWORD dwType = REG_DWORD;
 747         RegSetValueEx (hk, TEXT("ShowTrayIcon"), NULL, dwType, (PBYTE)&g.fStartup, dwSize);
 748         RegCloseKey (hk);
 749     }
 750
 751    Shortcut_FixStartup (cszSHORTCUT_NAME, g.fStartup);
 752
 753    Quit();
 754    EndDialog (hDlg, IDOK);
 755 }
 756