src/TortoiseProc/GravatarPictureBox.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2013-2017 - 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 #include "stdafx.h"
  20 #include "GravatarPictureBox.h"
  21 #include "Picture.h"
  22 #include "MessageBox.h"
  23 #include "UnicodeUtils.h"
  24 #include "Git.h"
  25 #include "LoglistCommonResource.h"
  26 #include "resource.h"
  27 #include <WinCrypt.h>
  28 #include "SmartHandle.h"
  29
  30 typedef CComCritSecLock<CComCriticalSection> CAutoLocker;
  31
  32 static CString CalcMD5(CString text)
  33 {
  34         HCRYPTPROV hProv = 0;
  35         if (!CryptAcquireContext(&hProv, nullptr, nullptr, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
  36                 return L"";
  37         SCOPE_EXIT { CryptReleaseContext(hProv, 0); };
  38
  39         HCRYPTHASH hHash = 0;
  40         if (!CryptCreateHash(hProv, CALG_MD5, 0, 0, &hHash))
  41                 return L"";
  42         SCOPE_EXIT { CryptDestroyHash(hHash); };
  43
  44         CStringA textA = CUnicodeUtils::GetUTF8(text);
  45         if (!CryptHashData(hHash, (LPBYTE)(LPCSTR)textA, textA.GetLength(), 0))
  46                 return L"";
  47
  48         CString hash;
  49         BYTE rgbHash[16];
  50         DWORD cbHash = _countof(rgbHash);
  51         if (!CryptGetHashParam(hHash, HP_HASHVAL, rgbHash, &cbHash, 0))
  52                 return L"";
  53
  54         for (DWORD i = 0; i < cbHash; i++)
  55         {
  56                 BYTE hi = rgbHash[i] >> 4;
  57                 BYTE lo = rgbHash[i]  & 0xf;
  58                 hash.AppendChar(hi + (hi > 9 ? 87 : 48));
  59                 hash.AppendChar(lo + (lo > 9 ? 87 : 48));
  60         }
  61
  62         return hash;
  63 }
  64
  65 BEGIN_MESSAGE_MAP(CGravatar, CStatic)
  66         ON_WM_PAINT()
  67 END_MESSAGE_MAP()
  68
  69 CGravatar::CGravatar()
  70         : CStatic()
  71         , m_gravatarEvent(INVALID_HANDLE_VALUE)
  72         , m_gravatarThread(nullptr)
  73         , m_gravatarExit(nullptr)
  74         , m_bEnableGravatar(false)
  75 {
  76         m_gravatarLock.Init();
  77 }
  78
  79 CGravatar::~CGravatar()
  80 {
  81         SafeTerminateGravatarThread();
  82         m_gravatarLock.Term();
  83         if (m_gravatarEvent)
  84                 CloseHandle(m_gravatarEvent);
  85 }
  86
  87 void CGravatar::Init()
  88 {
  89         if (m_bEnableGravatar)
  90         {
  91                 if (m_gravatarEvent == INVALID_HANDLE_VALUE)
  92                         m_gravatarEvent = ::CreateEvent(nullptr, FALSE, TRUE, nullptr);
  93                 if (m_gravatarThread == nullptr)
  94                 {
  95                         m_gravatarExit = new bool(false);
  96                         m_gravatarThread = AfxBeginThread([](LPVOID lpVoid) -> UINT { reinterpret_cast<CGravatar*>(lpVoid)->GravatarThread(); return 0; }, this, THREAD_PRIORITY_BELOW_NORMAL);
  97                         if (m_gravatarThread == nullptr)
  98                         {
  99                                 CMessageBox::Show(GetSafeHwnd(), IDS_ERR_THREADSTARTFAILED, IDS_APPNAME, MB_OK | MB_ICONERROR);
 100                                 delete m_gravatarExit;
 101                                 m_gravatarExit = nullptr;
 102                                 return;
 103                         }
 104                 }
 105         }
 106 }
 107
 108 void CGravatar::LoadGravatar(CString email)
 109 {
 110         if (m_gravatarThread == nullptr)
 111                 return;
 112
 113         if (email.IsEmpty())
 114         {
 115                 m_gravatarLock.Lock();
 116                 m_email.Empty();
 117                 if (!m_filename.IsEmpty())
 118                 {
 119                         m_filename.Empty();
 120                         m_gravatarLock.Unlock();
 121                         Invalidate();
 122                 }
 123                 else
 124                         m_gravatarLock.Unlock();
 125                 return;
 126         }
 127
 128         email.Trim();
 129         email.MakeLower();
 130         m_gravatarLock.Lock();
 131         bool diff = m_email != email;
 132         m_email = email;
 133         m_gravatarLock.Unlock();
 134         if (diff)
 135                 ::SetEvent(m_gravatarEvent);
 136 }
 137
 138 void CGravatar::GravatarThread()
 139 {
 140         bool *gravatarExit = m_gravatarExit;
 141         SCOPE_EXIT { delete gravatarExit; };
 142         CString gravatarBaseUrl = CRegString(L"Software\\TortoiseGit\\GravatarUrl", L"http://www.gravatar.com/avatar/%HASH%?d=identicon");
 143
 144         CString hostname;
 145         CString baseUrlPath;
 146         URL_COMPONENTS urlComponents = {0};
 147         urlComponents.dwStructSize = sizeof(urlComponents);
 148         urlComponents.lpszHostName = hostname.GetBuffer(INTERNET_MAX_HOST_NAME_LENGTH);
 149         urlComponents.dwHostNameLength = INTERNET_MAX_HOST_NAME_LENGTH;
 150         urlComponents.lpszUrlPath = baseUrlPath.GetBuffer(INTERNET_MAX_PATH_LENGTH);
 151         urlComponents.dwUrlPathLength = INTERNET_MAX_PATH_LENGTH;
 152         if (!InternetCrackUrl(gravatarBaseUrl, gravatarBaseUrl.GetLength(), 0, &urlComponents))
 153         {
 154                 m_filename.Empty();
 155                 return;
 156         }
 157         hostname.ReleaseBuffer();
 158         baseUrlPath.ReleaseBuffer();
 159
 160         HINTERNET hOpenHandle = InternetOpen(L"TortoiseGit", INTERNET_OPEN_TYPE_PRECONFIG, nullptr, nullptr, 0);
 161         SCOPE_EXIT { InternetCloseHandle(hOpenHandle); };
 162         bool isHttps = urlComponents.nScheme == INTERNET_SCHEME_HTTPS;
 163         HINTERNET hConnectHandle = InternetConnect(hOpenHandle, hostname, urlComponents.nPort, nullptr, nullptr, isHttps ? INTERNET_SCHEME_HTTP : urlComponents.nScheme, 0, 0);
 164         if (!hConnectHandle)
 165         {
 166                 m_filename.Empty();
 167                 return;
 168         }
 169         SCOPE_EXIT { InternetCloseHandle(hConnectHandle); };
 170
 171         while (!*gravatarExit)
 172         {
 173                 ::WaitForSingleObject(m_gravatarEvent, INFINITE);
 174                 while (!*gravatarExit)
 175                 {
 176                         m_gravatarLock.Lock();
 177                         CString email = m_email;
 178                         m_gravatarLock.Unlock();
 179                         if (email.IsEmpty())
 180                                 break;
 181
 182                         Sleep(500);
 183                         if (*gravatarExit)
 184                                 break;
 185                         m_gravatarLock.Lock();
 186                         bool diff = email != m_email;
 187                         m_gravatarLock.Unlock();
 188                         if (diff)
 189                                 continue;
 190
 191                         CString md5 = CalcMD5(email);
 192                         if (md5.IsEmpty())
 193                                 continue;
 194
 195                         CString gravatarUrl = baseUrlPath;
 196                         gravatarUrl.Replace(L"%HASH%", md5);
 197                         CString tempFile;
 198                         GetTempPath(tempFile);
 199                         tempFile += md5;
 200                         if (PathFileExists(tempFile))
 201                         {
 202                                 CAutoLocker lock(m_gravatarLock);
 203                                 if (m_email == email)
 204                                 {
 205                                         m_filename = tempFile;
 206                                         m_email.Empty();
 207                                 }
 208                                 else
 209                                         m_filename.Empty();
 210                         }
 211                         else
 212                         {
 213                                 BOOL ret = DownloadToFile(gravatarExit, hConnectHandle, isHttps, gravatarUrl, tempFile);
 214                                 if (ret)
 215                                 {
 216                                         DeleteFile(tempFile);
 217                                         if (*gravatarExit)
 218                                                 break;
 219                                         CAutoLocker lock(m_gravatarLock);
 220                                         m_filename.Empty();
 221                                 }
 222                                 if (*gravatarExit)
 223                                         break;
 224                                 CAutoLocker lock(m_gravatarLock);
 225                                 if (m_email == email && !ret)
 226                                 {
 227                                         CAutoFile hFile = CreateFile(tempFile, FILE_READ_ATTRIBUTES | FILE_WRITE_ATTRIBUTES, FILE_SHARE_READ, nullptr, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, nullptr);
 228                                         FILETIME creationTime = {};
 229                                         GetFileTime(hFile, &creationTime, nullptr, nullptr);
 230                                         uint64_t delta = 7 * 24 * 60 * 60 * 10000000LL;
 231                                         DWORD low = creationTime.dwLowDateTime;
 232                                         creationTime.dwLowDateTime += delta & 0xffffffff;
 233                                         creationTime.dwHighDateTime += delta >> 32;
 234                                         if (creationTime.dwLowDateTime < low)
 235                                                 creationTime.dwHighDateTime++;
 236                                         SetFileTime(hFile, &creationTime, nullptr, nullptr);
 237                                         m_filename = tempFile;
 238                                         m_email.Empty();
 239                                 }
 240                                 else
 241                                         m_filename.Empty();
 242                         }
 243                         if (*gravatarExit)
 244                                 break;
 245                         Invalidate();
 246                 }
 247         }
 248 }
 249
 250 int CGravatar::DownloadToFile(bool* gravatarExit, const HINTERNET hConnectHandle, bool isHttps, const CString& urlpath, const CString& dest)
 251 {
 252         HINTERNET hResourceHandle = HttpOpenRequest(hConnectHandle, nullptr, urlpath, nullptr, nullptr, nullptr, INTERNET_FLAG_KEEP_CONNECTION | (isHttps ? INTERNET_FLAG_SECURE : 0), 0);
 253         if (!hResourceHandle)
 254                 return -1;
 255
 256         SCOPE_EXIT { InternetCloseHandle(hResourceHandle); };
 257 resend:
 258         if (*gravatarExit)
 259                 return (int)INET_E_DOWNLOAD_FAILURE;
 260
 261         BOOL httpsendrequest = HttpSendRequest(hResourceHandle, nullptr, 0, nullptr, 0);
 262
 263         DWORD dwError = InternetErrorDlg(GetSafeHwnd(), hResourceHandle, ERROR_SUCCESS, FLAGS_ERROR_UI_FILTER_FOR_ERRORS | FLAGS_ERROR_UI_FLAGS_CHANGE_OPTIONS | FLAGS_ERROR_UI_FLAGS_GENERATE_DATA, nullptr);
 264
 265         if (dwError == ERROR_INTERNET_FORCE_RETRY)
 266                 goto resend;
 267         else if (!httpsendrequest || *gravatarExit)
 268                 return (int)INET_E_DOWNLOAD_FAILURE;
 269
 270         DWORD statusCode = 0;
 271         DWORD length = sizeof(statusCode);
 272         if (!HttpQueryInfo(hResourceHandle, HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER, (LPVOID)&statusCode, &length, nullptr) || statusCode != 200)
 273         {
 274                 if (statusCode == 404)
 275                         return ERROR_FILE_NOT_FOUND;
 276                 else if (statusCode == 403)
 277                         return ERROR_ACCESS_DENIED;
 278                 return (int)INET_E_DOWNLOAD_FAILURE;
 279         }
 280
 281         CFile destinationFile;
 282         if (!destinationFile.Open(dest, CFile::modeCreate | CFile::modeWrite))
 283                 return ERROR_ACCESS_DENIED;
 284
 285         DWORD downloadedSum = 0; // sum of bytes downloaded so far
 286         while (!*gravatarExit)
 287         {
 288                 DWORD size; // size of the data available
 289                 if (!InternetQueryDataAvailable(hResourceHandle, &size, 0, 0))
 290                         return (int)INET_E_DOWNLOAD_FAILURE;
 291
 292                 DWORD downloaded; // size of the downloaded data
 293                 auto buff = std::make_unique<TCHAR[]>(size + 1);
 294                 if (!InternetReadFile(hResourceHandle, (LPVOID)buff.get(), size, &downloaded))
 295                         return (int)INET_E_DOWNLOAD_FAILURE;
 296
 297                 if (downloaded == 0)
 298                         break;
 299
 300                 buff[downloaded] = '\0';
 301                 destinationFile.Write(buff.get(), downloaded);
 302
 303                 downloadedSum += downloaded;
 304         }
 305         destinationFile.Close();
 306         if (downloadedSum == 0)
 307                 return (int)INET_E_DOWNLOAD_FAILURE;
 308
 309         return 0;
 310 }
 311
 312 void CGravatar::SafeTerminateGravatarThread()
 313 {
 314         if (m_gravatarExit != nullptr)
 315                 *m_gravatarExit = true;
 316         if (m_gravatarThread)
 317         {
 318                 ::SetEvent(m_gravatarEvent);
 319                 ::WaitForSingleObject(m_gravatarThread, 1000);
 320                 m_gravatarThread = nullptr;
 321         }
 322 }
 323
 324 void CGravatar::OnPaint()
 325 {
 326         CPaintDC dc(this);
 327         RECT rect;
 328         GetClientRect(&rect);
 329         dc.FillSolidRect(&rect, GetSysColor(COLOR_BTNFACE));
 330         m_gravatarLock.Lock();
 331         CString filename = m_filename;
 332         m_gravatarLock.Unlock();
 333         if (filename.IsEmpty()) return;
 334         CPicture picture;
 335         picture.Load(filename.GetString());
 336         picture.Show(dc, rect);
 337 }