src/TortoisePlink/LoginDialog.cpp

   1 // TortoiseSVN - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003, 2013, 2018 - TortoiseSVN
   4
   5 // This program is free software; you can redistribute it and/or
   6 // modify it under the terms of the GNU General Public License
   7 // as published by the Free Software Foundation; either version 2
   8 // of the License, or (at your option) any later version.
   9
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14
  15 // You should have received a copy of the GNU General Public License
  16 // along with this program; if not, write to the Free Software
  17 // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  18
  19 #include "LoginDialog.h"
  20 #include "TortoisePlinkRes.h"
  21 #include <string>
  22
  23 #pragma comment(linker, "\"/manifestdependency:type='win32' name='Microsoft.Windows.Common-Controls' version='6.0.0.0' processorArchitecture='*' publicKeyToken='6595b64144ccf1df' language='*'\"")
  24
  25 HINSTANCE g_hmodThisDll;
  26 HWND g_hwndMain;
  27
  28 class LoginDialog
  29 {
  30 public:
  31    LoginDialog(const std::string& prompt);
  32
  33    static bool DoLoginDialog(std::string& password, const std::string& prompt);
  34
  35 private:
  36    bool myOK;
  37    HWND _hdlg;
  38
  39    std::string  myPassword;
  40    std::string  myPrompt;
  41
  42    void CreateModule(void);
  43    void RetrieveValues();
  44
  45    std::string GetPassword();
  46
  47    friend BOOL CALLBACK LoginDialogProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam);
  48 };
  49
  50
  51 BOOL DoLoginDialog(char* password, int maxlen, const char* prompt)
  52 {
  53    g_hmodThisDll = GetModuleHandle(0);
  54    g_hwndMain = GetParentHwnd();
  55    std::string passwordstr;
  56    BOOL res = LoginDialog::DoLoginDialog(passwordstr, prompt);
  57    if (res)
  58       strncpy(password, passwordstr.c_str(), maxlen);
  59    return res;
  60 }
  61
  62
  63 BOOL CALLBACK LoginDialogProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
  64 {
  65    if (uMsg == WM_INITDIALOG)
  66    {
  67       LoginDialog* pDlg = (LoginDialog*) lParam;
  68       pDlg->_hdlg = hwndDlg;
  69       SetWindowLongPtr(hwndDlg, GWLP_USERDATA, lParam);
  70       // Set prompt text
  71       SendDlgItemMessage(hwndDlg, IDC_LOGIN_PROMPT, WM_SETTEXT,
  72                          pDlg->myPrompt.length(), (LPARAM) pDlg->myPrompt.c_str());
  73       // Make sure edit control has the focus
  74       //SendDlgItemMessage(hwndDlg, IDC_LOGIN_PASSWORD, WM_SETFOCUS, 0, 0);
  75       if (GetDlgCtrlID((HWND) wParam) != IDC_LOGIN_PASSWORD)
  76       {
  77          SetFocus(GetDlgItem(hwndDlg, IDC_LOGIN_PASSWORD));
  78          return FALSE;
  79       }
  80       return TRUE;
  81    }
  82    else if (uMsg == WM_COMMAND && LOWORD(wParam) == IDCANCEL && HIWORD(wParam) == BN_CLICKED)
  83    {
  84       LoginDialog* pDlg = (LoginDialog*) GetWindowLongPtr(hwndDlg, GWLP_USERDATA);
  85       pDlg->myOK = false;
  86       EndDialog(hwndDlg, IDCANCEL);
  87       return 1;
  88    }
  89    else if (uMsg == WM_COMMAND && LOWORD(wParam) == IDOK && HIWORD(wParam) == BN_CLICKED)
  90    {
  91       LoginDialog* pDlg = (LoginDialog*) GetWindowLongPtr(hwndDlg, GWLP_USERDATA);
  92       pDlg->myOK = true;
  93       pDlg->RetrieveValues();
  94       EndDialog(hwndDlg, IDOK);
  95       return 1;
  96    }
  97    return 0;
  98 }
  99
 100 LoginDialog::LoginDialog(const std::string& prompt)
 101 {
 102    myPrompt = prompt;
 103 }
 104
 105 void LoginDialog::CreateModule(void)
 106 {
 107    DialogBoxParam(g_hmodThisDll, MAKEINTRESOURCE(IDD_LOGIN), g_hwndMain,
 108                   (DLGPROC)(LoginDialogProc), (LPARAM)this);
 109 }
 110
 111
 112 bool LoginDialog::DoLoginDialog(std::string& password, const std::string& prompt)
 113 {
 114    LoginDialog *pDlg = new LoginDialog(prompt);
 115
 116    pDlg->CreateModule();
 117
 118    bool ret = pDlg->myOK;
 119    password = pDlg->myPassword;
 120
 121    delete pDlg;
 122
 123    return ret;
 124 }
 125
 126
 127 std::string LoginDialog::GetPassword()
 128 {
 129    char szTxt[256];
 130    SendDlgItemMessage(_hdlg, IDC_LOGIN_PASSWORD, WM_GETTEXT, sizeof(szTxt), (LPARAM)szTxt);
 131    std::string strText = szTxt;
 132    return strText;
 133 }
 134
 135 void LoginDialog::RetrieveValues()
 136 {
 137    myPassword = GetPassword();
 138 }
 139
 140
 141 BOOL IsWinNT()
 142 {
 143    OSVERSIONINFO vi;
 144    vi.dwOSVersionInfoSize = sizeof(vi);
 145    if (GetVersionEx(&vi))
 146    {
 147       if (vi.dwPlatformId == VER_PLATFORM_WIN32_NT)
 148       {
 149          return TRUE;
 150       }
 151    }
 152    return FALSE;
 153 }
 154
 155 HWND GetParentHwnd()
 156 {
 157    if (IsWinNT())
 158    {
 159       return GetDesktopWindow();
 160    }
 161    else
 162    {
 163       return GetForegroundWindow();
 164    }
 165 }