src/TortoisePlink/LoginDialog.cpp

   1 // TortoiseSVN - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003, 2013 - 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 HINSTANCE g_hmodThisDll;
  24 HWND g_hwndMain;
  25
  26 class LoginDialog
  27 {
  28 public:
  29    LoginDialog(const std::string& prompt);
  30
  31    static bool DoLoginDialog(std::string& password, const std::string& prompt);
  32
  33 private:
  34    bool myOK;
  35    HWND _hdlg;
  36
  37    std::string  myPassword;
  38    std::string  myPrompt;
  39
  40    void CreateModule(void);
  41    void RetrieveValues();
  42
  43    std::string GetPassword();
  44
  45    friend BOOL CALLBACK LoginDialogProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam);
  46 };
  47
  48
  49 BOOL DoLoginDialog(char* password, int maxlen, const char* prompt)
  50 {
  51    g_hmodThisDll = GetModuleHandle(0);
  52    g_hwndMain = GetParentHwnd();
  53    std::string passwordstr;
  54    BOOL res = LoginDialog::DoLoginDialog(passwordstr, prompt);
  55    if (res)
  56       strncpy(password, passwordstr.c_str(), maxlen);
  57    return res;
  58 }
  59
  60
  61 BOOL CALLBACK LoginDialogProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
  62 {
  63    if (uMsg == WM_INITDIALOG)
  64    {
  65       LoginDialog* pDlg = (LoginDialog*) lParam;
  66       pDlg->_hdlg = hwndDlg;
  67       SetWindowLongPtr(hwndDlg, GWLP_USERDATA, lParam);
  68       // Set prompt text
  69       SendDlgItemMessage(hwndDlg, IDC_LOGIN_PROMPT, WM_SETTEXT,
  70                          pDlg->myPrompt.length(), (LPARAM) pDlg->myPrompt.c_str());
  71       // Make sure edit control has the focus
  72       //SendDlgItemMessage(hwndDlg, IDC_LOGIN_PASSWORD, WM_SETFOCUS, 0, 0);
  73       if (GetDlgCtrlID((HWND) wParam) != IDC_LOGIN_PASSWORD)
  74       {
  75          SetFocus(GetDlgItem(hwndDlg, IDC_LOGIN_PASSWORD));
  76          return FALSE;
  77       }
  78       return TRUE;
  79    }
  80    else if (uMsg == WM_COMMAND && LOWORD(wParam) == IDCANCEL && HIWORD(wParam) == BN_CLICKED)
  81    {
  82       LoginDialog* pDlg = (LoginDialog*) GetWindowLongPtr(hwndDlg, GWLP_USERDATA);
  83       pDlg->myOK = false;
  84       EndDialog(hwndDlg, IDCANCEL);
  85       return 1;
  86    }
  87    else if (uMsg == WM_COMMAND && LOWORD(wParam) == IDOK && HIWORD(wParam) == BN_CLICKED)
  88    {
  89       LoginDialog* pDlg = (LoginDialog*) GetWindowLongPtr(hwndDlg, GWLP_USERDATA);
  90       pDlg->myOK = true;
  91       pDlg->RetrieveValues();
  92       EndDialog(hwndDlg, IDOK);
  93       return 1;
  94    }
  95    return 0;
  96 }
  97
  98 LoginDialog::LoginDialog(const std::string& prompt)
  99 {
 100    myPrompt = prompt;
 101 }
 102
 103 void LoginDialog::CreateModule(void)
 104 {
 105    DialogBoxParam(g_hmodThisDll, MAKEINTRESOURCE(IDD_LOGIN), g_hwndMain,
 106                   (DLGPROC)(LoginDialogProc), (LPARAM)this);
 107 }
 108
 109
 110 bool LoginDialog::DoLoginDialog(std::string& password, const std::string& prompt)
 111 {
 112    LoginDialog *pDlg = new LoginDialog(prompt);
 113
 114    pDlg->CreateModule();
 115
 116    bool ret = pDlg->myOK;
 117    password = pDlg->myPassword;
 118
 119    delete pDlg;
 120
 121    return ret;
 122 }
 123
 124
 125 std::string LoginDialog::GetPassword()
 126 {
 127    char szTxt[256];
 128    SendDlgItemMessage(_hdlg, IDC_LOGIN_PASSWORD, WM_GETTEXT, sizeof(szTxt), (LPARAM)szTxt);
 129    std::string strText = szTxt;
 130    return strText;
 131 }
 132
 133 void LoginDialog::RetrieveValues()
 134 {
 135    myPassword = GetPassword();
 136 }
 137
 138
 139 BOOL IsWinNT()
 140 {
 141    OSVERSIONINFO vi;
 142    vi.dwOSVersionInfoSize = sizeof(vi);
 143    if (GetVersionEx(&vi))
 144    {
 145       if (vi.dwPlatformId == VER_PLATFORM_WIN32_NT)
 146       {
 147          return TRUE;
 148       }
 149    }
 150    return FALSE;
 151 }
 152
 153 HWND GetParentHwnd()
 154 {
 155    if (IsWinNT())
 156    {
 157       return GetDesktopWindow();
 158    }
 159    else
 160    {
 161       return GetForegroundWindow();
 162    }
 163 }