src/SshAskPass/SshAskPass.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2008-2013 - TortoiseGit
   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 Foundation,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19
  20 // SshAskPass.cpp : Defines the entry point for the application.
  21 //
  22
  23 #include "stdafx.h"
  24 #include "SshAskPass.h"
  25 #include <stdio.h>
  26 #include "propsys.h"
  27 #include "PropKey.h"
  28 #include <atlbase.h>
  29
  30 #include <commctrl.h>
  31 #pragma comment(lib, "comctl32.lib")
  32 #pragma comment(linker, "\"/manifestdependency:type='win32' name='Microsoft.Windows.Common-Controls' version='6.0.0.0' processorArchitecture='*' publicKeyToken='6595b64144ccf1df' language='*'\"")
  33
  34 #define MAX_LOADSTRING 100
  35
  36 // Global Variables:
  37 HINSTANCE hInst;                                                                // current instance
  38 TCHAR szTitle[MAX_LOADSTRING];                                  // The title bar text
  39 TCHAR szWindowClass[MAX_LOADSTRING];                    // the main window class name
  40
  41 // Forward declarations of functions included in this code module:
  42 INT_PTR CALLBACK        About(HWND, UINT, WPARAM, LPARAM);
  43
  44 TCHAR g_Promptphrase[] = _T("Enter your OpenSSH passphrase:");
  45 TCHAR *g_Prompt = NULL;
  46
  47 TCHAR g_PassWord[MAX_LOADSTRING];
  48
  49 int APIENTRY _tWinMain(HINSTANCE        /*hInstance*/,
  50                                          HINSTANCE              /*hPrevInstance*/,
  51                                          LPTSTR                 lpCmdLine,
  52                                          int                    /*nCmdShow*/)
  53 {
  54         SetDllDirectory(L"");
  55
  56         InitCommonControls();
  57
  58         if( _tcslen(lpCmdLine) == 0 )
  59         {
  60                 g_Prompt = g_Promptphrase;
  61         }
  62         else
  63         {
  64                 g_Prompt = lpCmdLine;
  65         }
  66
  67         TCHAR *yesno=_T("(yes/no)");
  68         size_t lens = _tcslen(yesno);
  69         TCHAR *p = lpCmdLine;
  70         BOOL bYesNo=FALSE;
  71
  72         while(*p)
  73         {
  74                 if (_tcsncicmp(p, yesno, lens) == 0)
  75                 {
  76                         bYesNo = TRUE;
  77                         break;
  78                 }
  79                 p++;
  80         }
  81
  82         if(bYesNo)
  83         {
  84                 if (::MessageBox(NULL, lpCmdLine, _T("TortoiseGit - git CLI stdin wrapper"), MB_YESNO|MB_ICONQUESTION) == IDYES)
  85                 {
  86                         _tprintf(_T("yes"));
  87                 }
  88                 else
  89                 {
  90                         _tprintf(_T("no"));
  91                 }
  92                 return 0;
  93         }
  94         else
  95         {
  96                 if(DialogBox(hInst, MAKEINTRESOURCE(IDD_ASK_PASSWORD), NULL, About) == IDOK)
  97                 {
  98                         _tprintf(_T("%s\n"), g_PassWord);
  99                         return 0;
 100                 }
 101                 _tprintf(_T("\n"));
 102                 return -1;
 103         }
 104 }
 105
 106 void MarkWindowAsUnpinnable(HWND hWnd)
 107 {
 108         typedef HRESULT (WINAPI *SHGPSFW) (HWND hwnd,REFIID riid,void** ppv);
 109
 110         HMODULE hShell = AtlLoadSystemLibraryUsingFullPath(_T("Shell32.dll"));
 111
 112         if (hShell) {
 113                 SHGPSFW pfnSHGPSFW = (SHGPSFW)::GetProcAddress(hShell, "SHGetPropertyStoreForWindow");
 114                 if (pfnSHGPSFW) {
 115                         IPropertyStore *pps;
 116                         HRESULT hr = pfnSHGPSFW(hWnd, IID_PPV_ARGS(&pps));
 117                         if (SUCCEEDED(hr)) {
 118                                 PROPVARIANT var;
 119                                 var.vt = VT_BOOL;
 120                                 var.boolVal = VARIANT_TRUE;
 121                                 hr = pps->SetValue(PKEY_AppUserModel_PreventPinning, var);
 122                                 pps->Release();
 123                         }
 124                 }
 125                 FreeLibrary(hShell);
 126         }
 127 }
 128
 129 // Message handler for password box.
 130 INT_PTR CALLBACK About(HWND hDlg, UINT message, WPARAM wParam, LPARAM /*lParam*/)
 131 {
 132         switch (message)
 133         {
 134         case WM_INITDIALOG:
 135                 {
 136                         MarkWindowAsUnpinnable(hDlg);
 137                         RECT rect;
 138                         ::GetWindowRect(hDlg,&rect);
 139                         DWORD dwWidth = GetSystemMetrics(SM_CXSCREEN);
 140                         DWORD dwHeight = GetSystemMetrics(SM_CYSCREEN);
 141
 142                         DWORD x,y;
 143                         x=(dwWidth - (rect.right-rect.left))/2;
 144                         y=(dwHeight - (rect.bottom-rect.top))/2;
 145
 146                         ::MoveWindow(hDlg,x,y,rect.right-rect.left,rect.bottom-rect.top,TRUE);
 147                         HWND title=::GetDlgItem(hDlg,IDC_STATIC_TITLE);
 148                         ::SetWindowText(title,g_Prompt);
 149
 150                         TCHAR *pass =_T("pass");
 151                         size_t passlens = _tcslen(pass);
 152                         TCHAR *p = g_Prompt;
 153                         bool password = false;
 154                         while (*p)
 155                         {
 156                                 if (_tcsncicmp(p, pass, passlens) == 0)
 157                                 {
 158                                         password = true;
 159                                         break;
 160                                 }
 161                                 p++;
 162                         }
 163                         if (!password)
 164                                 SendMessage(::GetDlgItem(hDlg, IDC_PASSWORD), EM_SETPASSWORDCHAR, 0, 0);
 165                 }
 166                 return (INT_PTR)TRUE;
 167
 168         case WM_COMMAND:
 169
 170                 if (LOWORD(wParam) == IDOK || LOWORD(wParam) == IDCANCEL)
 171                 {
 172                         if( LOWORD(wParam) == IDOK )
 173                         {
 174                                 HWND password = ::GetDlgItem(hDlg,IDC_PASSWORD);
 175                                 ::GetWindowText(password,g_PassWord,MAX_LOADSTRING);
 176                         }
 177                         EndDialog(hDlg, LOWORD(wParam));
 178                         return (INT_PTR)TRUE;
 179                 }
 180                 break;
 181         }
 182         return (INT_PTR)FALSE;
 183 }