1
// TortoiseGit - a Windows shell extension for easy version control
3 // Copyright (C) 2013-2017 - TortoiseGit
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.
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.
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.
20 #include "GravatarPictureBox.h"
22 #include "MessageBox.h"
23 #include "UnicodeUtils.h"
25 #include "LoglistCommonResource.h"
28 #include "SmartHandle.h"
30 typedef CComCritSecLock
<CComCriticalSection
> CAutoLocker
;
32 static CString
CalcMD5(CString text
)
35 if (!CryptAcquireContext(&hProv
, nullptr, nullptr, PROV_RSA_FULL
, CRYPT_VERIFYCONTEXT
))
37 SCOPE_EXIT
{ CryptReleaseContext(hProv
, 0); };
40 if (!CryptCreateHash(hProv
, CALG_MD5
, 0, 0, &hHash
))
42 SCOPE_EXIT
{ CryptDestroyHash(hHash
); };
44 CStringA textA
= CUnicodeUtils::GetUTF8(text
);
45 if (!CryptHashData(hHash
, (LPBYTE
)(LPCSTR
)textA
, textA
.GetLength(), 0))
50 DWORD cbHash
= _countof(rgbHash
);
51 if (!CryptGetHashParam(hHash
, HP_HASHVAL
, rgbHash
, &cbHash
, 0))
54 for (DWORD i
= 0; i
< cbHash
; i
++)
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));
65 BEGIN_MESSAGE_MAP(CGravatar
, CStatic
)
69 CGravatar::CGravatar()
71 , m_gravatarEvent(INVALID_HANDLE_VALUE
)
72 , m_gravatarThread(nullptr)
73 , m_gravatarExit(nullptr)
74 , m_bEnableGravatar(false)
76 m_gravatarLock
.Init();
79 CGravatar::~CGravatar()
81 SafeTerminateGravatarThread();
82 m_gravatarLock
.Term();
84 CloseHandle(m_gravatarEvent
);
87 void CGravatar::Init()
89 if (m_bEnableGravatar
)
91 if (m_gravatarEvent
== INVALID_HANDLE_VALUE
)
92 m_gravatarEvent
= ::CreateEvent(nullptr, FALSE
, TRUE
, nullptr);
93 if (m_gravatarThread
== nullptr)
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)
99 CMessageBox::Show(GetSafeHwnd(), IDS_ERR_THREADSTARTFAILED
, IDS_APPNAME
, MB_OK
| MB_ICONERROR
);
100 delete m_gravatarExit
;
101 m_gravatarExit
= nullptr;
108 void CGravatar::LoadGravatar(CString email
)
110 if (m_gravatarThread
== nullptr)
115 m_gravatarLock
.Lock();
117 if (!m_filename
.IsEmpty())
120 m_gravatarLock
.Unlock();
124 m_gravatarLock
.Unlock();
130 m_gravatarLock
.Lock();
131 bool diff
= m_email
!= email
;
133 m_gravatarLock
.Unlock();
135 ::SetEvent(m_gravatarEvent
);
138 void CGravatar::GravatarThread()
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");
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
))
157 hostname
.ReleaseBuffer();
158 baseUrlPath
.ReleaseBuffer();
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);
169 SCOPE_EXIT
{ InternetCloseHandle(hConnectHandle
); };
171 while (!*gravatarExit
)
173 ::WaitForSingleObject(m_gravatarEvent
, INFINITE
);
174 while (!*gravatarExit
)
176 m_gravatarLock
.Lock();
177 CString email
= m_email
;
178 m_gravatarLock
.Unlock();
185 m_gravatarLock
.Lock();
186 bool diff
= email
!= m_email
;
187 m_gravatarLock
.Unlock();
191 CString md5
= CalcMD5(email
);
195 CString gravatarUrl
= baseUrlPath
;
196 gravatarUrl
.Replace(L
"%HASH%", md5
);
198 GetTempPath(tempFile
);
200 if (PathFileExists(tempFile
))
202 CAutoLocker
lock(m_gravatarLock
);
203 if (m_email
== email
)
205 m_filename
= tempFile
;
213 BOOL ret
= DownloadToFile(gravatarExit
, hConnectHandle
, isHttps
, gravatarUrl
, tempFile
);
216 DeleteFile(tempFile
);
219 CAutoLocker
lock(m_gravatarLock
);
224 CAutoLocker
lock(m_gravatarLock
);
225 if (m_email
== email
&& !ret
)
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
;
250 int CGravatar::DownloadToFile(bool* gravatarExit
, const HINTERNET hConnectHandle
, bool isHttps
, const CString
& urlpath
, const CString
& dest
)
252 HINTERNET hResourceHandle
= HttpOpenRequest(hConnectHandle
, nullptr, urlpath
, nullptr, nullptr, nullptr, INTERNET_FLAG_KEEP_CONNECTION
| (isHttps
? INTERNET_FLAG_SECURE
: 0), 0);
253 if (!hResourceHandle
)
256 SCOPE_EXIT
{ InternetCloseHandle(hResourceHandle
); };
259 return (int)INET_E_DOWNLOAD_FAILURE
;
261 BOOL httpsendrequest
= HttpSendRequest(hResourceHandle
, nullptr, 0, nullptr, 0);
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);
265 if (dwError
== ERROR_INTERNET_FORCE_RETRY
)
267 else if (!httpsendrequest
|| *gravatarExit
)
268 return (int)INET_E_DOWNLOAD_FAILURE
;
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)
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
;
281 CFile destinationFile
;
282 if (!destinationFile
.Open(dest
, CFile::modeCreate
| CFile::modeWrite
))
283 return ERROR_ACCESS_DENIED
;
285 DWORD downloadedSum
= 0; // sum of bytes downloaded so far
286 while (!*gravatarExit
)
288 DWORD size
; // size of the data available
289 if (!InternetQueryDataAvailable(hResourceHandle
, &size
, 0, 0))
290 return (int)INET_E_DOWNLOAD_FAILURE
;
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
;
300 buff
[downloaded
] = '\0';
301 destinationFile
.Write(buff
.get(), downloaded
);
303 downloadedSum
+= downloaded
;
305 destinationFile
.Close();
306 if (downloadedSum
== 0)
307 return (int)INET_E_DOWNLOAD_FAILURE
;
312 void CGravatar::SafeTerminateGravatarThread()
314 if (m_gravatarExit
!= nullptr)
315 *m_gravatarExit
= true;
316 if (m_gravatarThread
)
318 ::SetEvent(m_gravatarEvent
);
319 ::WaitForSingleObject(m_gravatarThread
, 1000);
320 m_gravatarThread
= nullptr;
324 void CGravatar::OnPaint()
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;
335 picture
.Load(filename
.GetString());
336 picture
.Show(dc
, rect
);