src/WINNT/client_creds/credstab.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 #include <winsock2.h>
  11 #include <ws2tcpip.h>
  12
  13 extern "C" {
  14 #include <afsconfig.h>
  15 #include <afs/param.h>
  16 #include <roken.h>
  17 }
  18
  19 #include "afscreds.h"
  20
  21
  22 /*
  23  * PROTOTYPES _________________________________________________________________
  24  *
  25  */
  26
  27 void Creds_OnUpdate (HWND hDlg);
  28 void Creds_OnCheckRemind (HWND hDlg);
  29 void Creds_OnClickObtain (HWND hDlg);
  30 void Creds_OnClickDestroy (HWND hDlg);
  31
  32 BOOL CALLBACK NewCreds_DlgProc (HWND hDlg, UINT msg, WPARAM wp, LPARAM lp);
  33 void NewCreds_OnInitDialog (HWND hDlg);
  34 void NewCreds_OnEnable (HWND hDlg);
  35 BOOL NewCreds_OnOK (HWND hDlg);
  36 void NewCreds_OnCancel (HWND hDlg);
  37
  38
  39 /*
  40  * ROUTINES ___________________________________________________________________
  41  *
  42  */
  43
  44 BOOL CALLBACK Creds_DlgProc (HWND hDlg, UINT msg, WPARAM wp, LPARAM lp)
  45 {
  46    switch (msg)
  47       {
  48       case WM_INITDIALOG:
  49          RECT rTab;
  50          GetClientRect (GetParent(hDlg), &rTab);
  51          TabCtrl_AdjustRect (GetParent (hDlg), FALSE, &rTab);
  52          SetWindowPos (hDlg, NULL, rTab.left, rTab.top, 0, 0, SWP_NOSIZE | SWP_NOACTIVATE | SWP_NOZORDER);
  53
  54          SetWindowLongPtr (hDlg, DWLP_USER, lp);
  55          Creds_OnUpdate (hDlg);
  56          break;
  57
  58       case WM_COMMAND:
  59          switch (LOWORD(wp))
  60             {
  61             case IDC_CREDS_REMIND:
  62                Creds_OnCheckRemind (hDlg);
  63                break;
  64
  65             case IDC_CREDS_OBTAIN:
  66                Creds_OnClickObtain (hDlg);
  67                break;
  68
  69             case IDC_CREDS_DESTROY:
  70                Creds_OnClickDestroy (hDlg);
  71                break;
  72
  73             case IDHELP:
  74                Creds_DlgProc (hDlg, WM_HELP, 0, 0);
  75                break;
  76             }
  77          break;
  78       }
  79
  80    return FALSE;
  81 }
  82
  83
  84 void Creds_OnCheckRemind (HWND hDlg)
  85 {
  86    LPTSTR pszCell = (LPTSTR)GetWindowLongPtr (hDlg, DWLP_USER);
  87    lock_ObtainMutex(&g.credsLock);
  88    size_t iCreds;
  89    for (iCreds = 0; iCreds < g.cCreds; ++iCreds)
  90       {
  91       if (!lstrcmpi (g.aCreds[ iCreds ].szCell, pszCell))
  92          break;
  93       }
  94
  95    if (iCreds != g.cCreds)
  96       {
  97       g.aCreds[ iCreds ].fRemind = IsDlgButtonChecked (hDlg, IDC_CREDS_REMIND);
  98       SaveRemind (iCreds);
  99       }
 100    lock_ReleaseMutex(&g.credsLock);
 101 }
 102
 103
 104 void Creds_OnUpdate (HWND hDlg)
 105 {
 106    LPTSTR pszCell = (LPTSTR)GetWindowLongPtr (hDlg, DWLP_USER);
 107    if (!pszCell || !*pszCell)
 108       {
 109       BOOL fRunning = IsServiceRunning();
 110       ShowWindow (GetDlgItem (hDlg, IDC_RUNNING), fRunning);
 111       ShowWindow (GetDlgItem (hDlg, IDC_STOPPED), !fRunning);
 112       ShowWindow (GetDlgItem (hDlg, IDC_CREDS_OBTAIN), fRunning);
 113       return;
 114       }
 115
 116    lock_ObtainMutex(&g.credsLock);
 117    size_t iCreds;
 118    for (iCreds = 0; iCreds < g.cCreds; ++iCreds)
 119       {
 120       if (!lstrcmpi (g.aCreds[ iCreds ].szCell, pszCell))
 121          break;
 122       }
 123
 124    TCHAR szGateway[cchRESOURCE] = TEXT("");
 125    if (!g.fIsWinNT)
 126       GetGatewayName (szGateway);
 127
 128    if (!szGateway[0])
 129       {
 130       SetDlgItemText (hDlg, IDC_CREDS_CELL, pszCell);
 131       }
 132    else
 133       {
 134       TCHAR szCell[ cchRESOURCE ];
 135       TCHAR szFormat[ cchRESOURCE ];
 136       GetString (szFormat, IDS_CELL_GATEWAY);
 137       wsprintf (szCell, szFormat, pszCell, szGateway);
 138       SetDlgItemText (hDlg, IDC_CREDS_CELL, szCell);
 139       }
 140
 141    if (iCreds == g.cCreds)
 142       {
 143       TCHAR szText[cchRESOURCE];
 144       GetString (szText, IDS_NO_CREDS);
 145       SetDlgItemText (hDlg, IDC_CREDS_INFO, szText);
 146       }
 147    else
 148       {
 149       // FormatString(%t) expects a date in GMT, not the local time zone...
 150       //
 151       FILETIME ftLocal;
 152       SystemTimeToFileTime (&g.aCreds[ iCreds ].stExpires, &ftLocal);
 153
 154       FILETIME ftGMT;
 155       LocalFileTimeToFileTime (&ftLocal, &ftGMT);
 156
 157       SYSTEMTIME stGMT;
 158       FileTimeToSystemTime (&ftGMT, &stGMT);
 159
 160       SYSTEMTIME stNow;
 161       GetLocalTime (&stNow);
 162
 163       FILETIME ftNow;
 164       SystemTimeToFileTime (&stNow, &ftNow);
 165
 166       LONGLONG llNow = (((LONGLONG)ftNow.dwHighDateTime) << 32) + (LONGLONG)(ftNow.dwLowDateTime);
 167       LONGLONG llExpires = (((LONGLONG)ftLocal.dwHighDateTime) << 32) + (LONGLONG)(ftLocal.dwLowDateTime);
 168
 169       llNow /= c100ns1SECOND;
 170       llExpires /= c100ns1SECOND;
 171
 172       LPTSTR pszCreds = NULL;
 173       if (llExpires <= (llNow + (LONGLONG)cminREMIND_WARN * csec1MINUTE))
 174           pszCreds = FormatString (IDS_CREDS_EXPIRED, TEXT("%s"), g.aCreds[ iCreds ].szUser);
 175
 176       if (!pszCreds || !pszCreds[0])
 177           pszCreds = FormatString (IDS_CREDS, TEXT("%s%t"), g.aCreds[ iCreds ].szUser, &stGMT);
 178       SetDlgItemText (hDlg, IDC_CREDS_INFO, pszCreds);
 179       FreeString (pszCreds);
 180       }
 181
 182    lock_ReleaseMutex(&g.credsLock);
 183    CheckDlgButton (hDlg, IDC_CREDS_REMIND, (iCreds == g.cCreds) ? FALSE : g.aCreds[iCreds].fRemind);
 184
 185    EnableWindow (GetDlgItem (hDlg, IDC_CREDS_OBTAIN), IsServiceRunning());
 186    EnableWindow (GetDlgItem (hDlg, IDC_CREDS_REMIND), (iCreds != g.cCreds));
 187    EnableWindow (GetDlgItem (hDlg, IDC_CREDS_DESTROY), (iCreds != g.cCreds));
 188 }
 189
 190
 191 void Creds_OnClickObtain (HWND hDlg)
 192 {
 193    LPTSTR pszCell = (LPTSTR)GetWindowLongPtr (hDlg, DWLP_USER);
 194
 195    InterlockedIncrement (&g.fShowingMessage);
 196    ShowObtainCreds (FALSE, pszCell);
 197 }
 198
 199
 200 void Creds_OnClickDestroy (HWND hDlg)
 201 {
 202    LPTSTR pszCell = (LPTSTR)GetWindowLongPtr (hDlg, DWLP_USER);
 203    if (pszCell && *pszCell)
 204       {
 205       DestroyCurrentCredentials (pszCell);
 206       Main_RepopulateTabs (FALSE);
 207       Creds_OnUpdate (hDlg);
 208       }
 209 }
 210
 211
 212 struct _obtaincreds {
 213     DWORD type;
 214     HWND  parent;
 215     char * cell;
 216 };
 217
 218 void ObtainCredsThread(void * data)
 219 {
 220     struct _obtaincreds * oc = (struct _obtaincreds *)data;
 221
 222     ModalDialogParam (oc->type, oc->parent, (DLGPROC)NewCreds_DlgProc, (LPARAM)oc->cell);
 223     free(oc->cell);
 224     free(oc);
 225 }
 226
 227 void ShowObtainCreds (BOOL fExpiring, LPTSTR pszCell)
 228 {
 229     struct _obtaincreds * oc = (struct _obtaincreds *)malloc(sizeof(struct _obtaincreds));
 230     if ( !oc )
 231         return;
 232     oc->parent = (IsWindowVisible (g.hMain)) ? g.hMain : NULL;
 233     oc->type = fExpiring ? IDD_NEWCREDS_EXPIRE : IDD_NEWCREDS;
 234     oc->cell = strdup(pszCell);
 235
 236     HANDLE thread = 0;
 237     ULONG  threadID = 123;
 238
 239     thread = CreateThread(NULL, 0, (PTHREAD_START_ROUTINE)ObtainCredsThread,
 240                                     oc, 0, &threadID);
 241     if (thread != NULL)
 242         CloseHandle(thread);
 243     else {
 244         free(oc->cell);
 245         free(oc);
 246     }
 247 }
 248
 249
 250 BOOL CALLBACK NewCreds_DlgProc (HWND hDlg, UINT msg, WPARAM wp, LPARAM lp)
 251 {
 252    switch (msg)
 253       {
 254       case WM_INITDIALOG:
 255          SetWindowLongPtr (hDlg, DWLP_USER, lp);
 256          NewCreds_OnInitDialog (hDlg);
 257          break;
 258
 259       case WM_DESTROY:
 260          InterlockedDecrement (&g.fShowingMessage);
 261          break;
 262
 263       case WM_COMMAND:
 264          switch (LOWORD(wp))
 265             {
 266             case IDOK:
 267                if (NewCreds_OnOK (hDlg))
 268                   EndDialog (hDlg, IDOK);
 269                break;
 270
 271             case IDCANCEL:
 272                NewCreds_OnCancel (hDlg);
 273                EndDialog (hDlg, IDCANCEL);
 274                break;
 275
 276             case IDC_NEWCREDS_USER:
 277             case IDC_NEWCREDS_PASSWORD:
 278                NewCreds_OnEnable (hDlg);
 279                break;
 280
 281             case IDHELP:
 282                NewCreds_DlgProc (hDlg, WM_HELP, 0, 0);
 283                break;
 284             }
 285          break;
 286       }
 287    return FALSE;
 288 }
 289
 290
 291 void NewCreds_OnInitDialog (HWND hDlg)
 292 {
 293    LPTSTR pszCell = (LPTSTR)GetWindowLongPtr (hDlg, DWLP_USER);
 294    if (!pszCell)
 295       pszCell = TEXT("");
 296
 297    if (GetDlgItem (hDlg, IDC_NEWCREDS_TITLE))
 298       {
 299       TCHAR szText[ cchRESOURCE ];
 300       GetDlgItemText (hDlg, IDC_NEWCREDS_TITLE, szText, cchRESOURCE);
 301       LPTSTR pszText = FormatString (szText, TEXT("%s"), pszCell);
 302       SetDlgItemText (hDlg, IDC_NEWCREDS_TITLE, pszText);
 303       FreeString (pszText);
 304       }
 305
 306    if (pszCell && *pszCell)
 307       {
 308       SetDlgItemText (hDlg, IDC_NEWCREDS_CELL, pszCell);
 309       }
 310    else
 311       {
 312       TCHAR szCell[ cchRESOURCE ] = TEXT("");
 313       (void)GetDefaultCell (szCell);
 314       SetDlgItemText (hDlg, IDC_NEWCREDS_CELL, szCell);
 315       }
 316
 317    lock_ObtainMutex(&g.credsLock);
 318    size_t iCreds;
 319    for (iCreds = 0; iCreds < g.cCreds; ++iCreds)
 320       {
 321       if (*pszCell && !lstrcmpi (g.aCreds[ iCreds ].szCell, pszCell))
 322          break;
 323       }
 324    if ((iCreds == g.cCreds) || (!g.aCreds[ iCreds ].szUser[0]))
 325       {
 326       PostMessage (hDlg, WM_NEXTDLGCTL, (WPARAM)GetDlgItem (hDlg, IDC_NEWCREDS_USER), TRUE);
 327       }
 328    else // (we have a valid username already)
 329       {
 330       SetDlgItemText (hDlg, IDC_NEWCREDS_USER, g.aCreds[ iCreds ].szUser);
 331       PostMessage (hDlg, WM_NEXTDLGCTL, (WPARAM)GetDlgItem (hDlg, IDC_NEWCREDS_PASSWORD), TRUE);
 332       }
 333    lock_ReleaseMutex(&g.credsLock);
 334
 335    NewCreds_OnEnable (hDlg);
 336    SetForegroundWindow(hDlg);
 337    KillTimer (g.hMain, ID_SERVICE_TIMER);
 338 }
 339
 340
 341 void NewCreds_OnEnable (HWND hDlg)
 342 {
 343     BOOL fEnable = TRUE;
 344
 345     TCHAR szUser[ cchRESOURCE ];
 346     GetDlgItemText (hDlg, IDC_NEWCREDS_USER, szUser, cchRESOURCE);
 347     if (!szUser[0])
 348         fEnable = FALSE;
 349
 350     if ( !strchr(szUser, '@') ) {
 351         TCHAR szPassword[ cchRESOURCE ];
 352         GetDlgItemText (hDlg, IDC_NEWCREDS_PASSWORD, szPassword, cchRESOURCE);
 353         if (!szPassword[0])
 354             fEnable = FALSE;
 355     }
 356     EnableWindow (GetDlgItem (hDlg, IDOK), fEnable);
 357 }
 358
 359
 360 BOOL NewCreds_OnOK (HWND hDlg)
 361 {
 362    TCHAR szCell[ cchRESOURCE ];
 363    GetDlgItemText (hDlg, IDC_NEWCREDS_CELL, szCell, cchRESOURCE);
 364
 365    TCHAR szUser[ cchRESOURCE ];
 366    GetDlgItemText (hDlg, IDC_NEWCREDS_USER, szUser, cchRESOURCE);
 367
 368    TCHAR szPassword[ cchRESOURCE ];
 369    GetDlgItemText (hDlg, IDC_NEWCREDS_PASSWORD, szPassword, cchRESOURCE);
 370
 371    int rc;
 372    if ((rc = ObtainNewCredentials (szCell, szUser, szPassword, FALSE)) != 0)
 373       {
 374       EnableWindow (GetDlgItem (hDlg, IDOK), TRUE);
 375       EnableWindow (GetDlgItem (hDlg, IDCANCEL), TRUE);
 376       EnableWindow (GetDlgItem (hDlg, IDC_NEWCREDS_CELL), TRUE);
 377       EnableWindow (GetDlgItem (hDlg, IDC_NEWCREDS_USER), TRUE);
 378       EnableWindow (GetDlgItem (hDlg, IDC_NEWCREDS_PASSWORD), TRUE);
 379       return FALSE;
 380       }
 381
 382    Main_RepopulateTabs (FALSE);
 383    return TRUE;
 384 }
 385
 386
 387 void NewCreds_OnCancel (HWND hDlg)
 388 {
 389    TCHAR szText[ cchRESOURCE ] = "";
 390    LPTSTR pszCell = NULL;
 391
 392    if (GetDlgItem (hDlg, IDC_NEWCREDS_CELL))
 393    {
 394        GetDlgItemText (hDlg, IDC_NEWCREDS_CELL, szText, cchRESOURCE);
 395        if ( szText[0] )
 396            pszCell = szText;
 397    }
 398
 399    if ( !pszCell )
 400        pszCell = (LPTSTR)GetWindowLongPtr (hDlg, DWLP_USER);
 401    if (pszCell)
 402       {
 403       HWND hTab = GetDlgItem (g.hMain, IDC_TABS);
 404       LPTSTR pszTab = (LPTSTR)GetTabParam (hTab, TabCtrl_GetCurSel(hTab));
 405       HWND hChildDlg = NULL;
 406
 407       lock_ObtainMutex(&g.credsLock);
 408       for (size_t iCreds = 0; iCreds < g.cCreds; ++iCreds)
 409          {
 410          if (!lstrcmpi (g.aCreds[ iCreds ].szCell, pszCell))
 411             {
 412             g.aCreds[ iCreds ].fRemind = FALSE;
 413             SaveRemind (iCreds);
 414
 415             // Check the active tab, and fix its checkbox if necessary
 416             //
 417             if (pszTab && HIWORD(pszTab) && (!lstrcmpi (pszTab, pszCell)))
 418                {
 419                hChildDlg = GetTabChild (hTab);
 420                }
 421             }
 422          }
 423       lock_ReleaseMutex(&g.credsLock);
 424
 425       if (hChildDlg)
 426          CheckDlgButton (hChildDlg, IDC_CREDS_REMIND, FALSE);
 427       }
 428 }
 429