/* This file is part of Telegram Desktop, the official desktop application for the Telegram messaging service. For license and copyright information please follow this link: https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL */ #include "platform/win/specific_win.h" #include "platform/win/main_window_win.h" #include "platform/win/notifications_manager_win.h" #include "platform/win/windows_app_user_model_id.h" #include "platform/win/windows_dlls.h" #include "platform/win/windows_event_filter.h" #include "lang/lang_keys.h" #include "mainwindow.h" #include "mainwidget.h" #include "history/history_location_manager.h" #include "storage/localstorage.h" #include "core/crash_reports.h" #include #include #include #include #include #include #pragma warning(push) #pragma warning(disable:4091) #include #include #pragma warning(pop) #include #include #include #include #include #include #include #include #include #include #include #include #include #include #ifndef DCX_USESTYLE #define DCX_USESTYLE 0x00010000 #endif #ifndef WM_NCPOINTERUPDATE #define WM_NCPOINTERUPDATE 0x0241 #define WM_NCPOINTERDOWN 0x0242 #define WM_NCPOINTERUP 0x0243 #endif using namespace Microsoft::WRL; using namespace ABI::Windows::UI::Notifications; using namespace ABI::Windows::Data::Xml::Dom; using namespace Windows::Foundation; using namespace Platform; namespace { QStringList _initLogs; bool themeInited = false; bool finished = true; QMargins simpleMargins, margins; HICON bigIcon = 0, smallIcon = 0, overlayIcon = 0; class _PsInitializer { public: _PsInitializer() { Dlls::start(); } }; _PsInitializer _psInitializer; }; QAbstractNativeEventFilter *psNativeEventFilter() { return EventFilter::createInstance(); } void psDeleteDir(const QString &dir) { std::wstring wDir = QDir::toNativeSeparators(dir).toStdWString(); WCHAR path[4096]; memcpy(path, wDir.c_str(), (wDir.size() + 1) * sizeof(WCHAR)); path[wDir.size() + 1] = 0; SHFILEOPSTRUCT file_op = { NULL, FO_DELETE, path, L"", FOF_NOCONFIRMATION | FOF_NOERRORUI | FOF_SILENT, false, 0, L"" }; int res = SHFileOperation(&file_op); } namespace { BOOL CALLBACK _ActivateProcess(HWND hWnd, LPARAM lParam) { uint64 &processId(*(uint64*)lParam); DWORD dwProcessId; ::GetWindowThreadProcessId(hWnd, &dwProcessId); if ((uint64)dwProcessId == processId) { // found top-level window static const int32 nameBufSize = 1024; WCHAR nameBuf[nameBufSize]; int32 len = GetWindowText(hWnd, nameBuf, nameBufSize); if (len && len < nameBufSize) { if (QRegularExpression(qsl("^Telegram(\\s*\\(\\d+\\))?$")).match(QString::fromStdWString(nameBuf)).hasMatch()) { BOOL res = ::SetForegroundWindow(hWnd); ::SetFocus(hWnd); return FALSE; } } } return TRUE; } } namespace { TimeMs _lastUserAction = 0; } // namespace void psUserActionDone() { _lastUserAction = getms(true); EventFilter::getInstance()->setSessionLoggedOff(false); } bool psIdleSupported() { LASTINPUTINFO lii; lii.cbSize = sizeof(LASTINPUTINFO); return GetLastInputInfo(&lii); } TimeMs psIdleTime() { LASTINPUTINFO lii; lii.cbSize = sizeof(LASTINPUTINFO); return GetLastInputInfo(&lii) ? (GetTickCount() - lii.dwTime) : (getms(true) - _lastUserAction); } QStringList psInitLogs() { return _initLogs; } void psClearInitLogs() { _initLogs = QStringList(); } void psActivateProcess(uint64 pid) { if (pid) { ::EnumWindows((WNDENUMPROC)_ActivateProcess, (LPARAM)&pid); } } QString psAppDataPath() { static const int maxFileLen = MAX_PATH * 10; WCHAR wstrPath[maxFileLen]; if (GetEnvironmentVariable(L"APPDATA", wstrPath, maxFileLen)) { QDir appData(QString::fromStdWString(std::wstring(wstrPath))); #ifdef OS_WIN_STORE return appData.absolutePath() + qsl("/Telegram Desktop UWP/"); #else // OS_WIN_STORE return appData.absolutePath() + '/' + str_const_toString(AppName) + '/'; #endif // OS_WIN_STORE } return QString(); } QString psAppDataPathOld() { static const int maxFileLen = MAX_PATH * 10; WCHAR wstrPath[maxFileLen]; if (GetEnvironmentVariable(L"APPDATA", wstrPath, maxFileLen)) { QDir appData(QString::fromStdWString(std::wstring(wstrPath))); return appData.absolutePath() + '/' + str_const_toString(AppNameOld) + '/'; } return QString(); } QString psDownloadPath() { return QStandardPaths::writableLocation(QStandardPaths::DownloadLocation) + '/' + str_const_toString(AppName) + '/'; } void psDoCleanup() { try { psAutoStart(false, true); psSendToMenu(false, true); AppUserModelId::cleanupShortcut(); } catch (...) { } } namespace { QRect _monitorRect; TimeMs _monitorLastGot = 0; } // namespace QRect psDesktopRect() { auto tnow = getms(); if (tnow > _monitorLastGot + 1000LL || tnow < _monitorLastGot) { _monitorLastGot = tnow; HMONITOR hMonitor = MonitorFromWindow(App::wnd()->psHwnd(), MONITOR_DEFAULTTONEAREST); if (hMonitor) { MONITORINFOEX info; info.cbSize = sizeof(info); GetMonitorInfo(hMonitor, &info); _monitorRect = QRect(info.rcWork.left, info.rcWork.top, info.rcWork.right - info.rcWork.left, info.rcWork.bottom - info.rcWork.top); } else { _monitorRect = QApplication::desktop()->availableGeometry(App::wnd()); } } return _monitorRect; } void psShowOverAll(QWidget *w, bool canFocus) { } void psBringToBack(QWidget *w) { } int psCleanup() { __try { psDoCleanup(); } __except(EXCEPTION_EXECUTE_HANDLER) { return 0; } return 0; } void psDoFixPrevious() { try { static const int bufSize = 4096; DWORD checkType, checkSize = bufSize * 2; WCHAR checkStr[bufSize]; QString appId = str_const_toString(AppId); QString newKeyStr1 = QString("Software\\Wow6432Node\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\%1_is1").arg(appId); QString newKeyStr2 = QString("Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\%1_is1").arg(appId); QString oldKeyStr1 = QString("SOFTWARE\\Wow6432Node\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\%1_is1").arg(appId); QString oldKeyStr2 = QString("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\%1_is1").arg(appId); HKEY newKey1, newKey2, oldKey1, oldKey2; LSTATUS newKeyRes1 = RegOpenKeyEx(HKEY_CURRENT_USER, newKeyStr1.toStdWString().c_str(), 0, KEY_READ, &newKey1); LSTATUS newKeyRes2 = RegOpenKeyEx(HKEY_CURRENT_USER, newKeyStr2.toStdWString().c_str(), 0, KEY_READ, &newKey2); LSTATUS oldKeyRes1 = RegOpenKeyEx(HKEY_LOCAL_MACHINE, oldKeyStr1.toStdWString().c_str(), 0, KEY_READ, &oldKey1); LSTATUS oldKeyRes2 = RegOpenKeyEx(HKEY_LOCAL_MACHINE, oldKeyStr2.toStdWString().c_str(), 0, KEY_READ, &oldKey2); bool existNew1 = (newKeyRes1 == ERROR_SUCCESS) && (RegQueryValueEx(newKey1, L"InstallDate", 0, &checkType, (BYTE*)checkStr, &checkSize) == ERROR_SUCCESS); checkSize = bufSize * 2; bool existNew2 = (newKeyRes2 == ERROR_SUCCESS) && (RegQueryValueEx(newKey2, L"InstallDate", 0, &checkType, (BYTE*)checkStr, &checkSize) == ERROR_SUCCESS); checkSize = bufSize * 2; bool existOld1 = (oldKeyRes1 == ERROR_SUCCESS) && (RegQueryValueEx(oldKey1, L"InstallDate", 0, &checkType, (BYTE*)checkStr, &checkSize) == ERROR_SUCCESS); checkSize = bufSize * 2; bool existOld2 = (oldKeyRes2 == ERROR_SUCCESS) && (RegQueryValueEx(oldKey2, L"InstallDate", 0, &checkType, (BYTE*)checkStr, &checkSize) == ERROR_SUCCESS); checkSize = bufSize * 2; if (newKeyRes1 == ERROR_SUCCESS) RegCloseKey(newKey1); if (newKeyRes2 == ERROR_SUCCESS) RegCloseKey(newKey2); if (oldKeyRes1 == ERROR_SUCCESS) RegCloseKey(oldKey1); if (oldKeyRes2 == ERROR_SUCCESS) RegCloseKey(oldKey2); if (existNew1 || existNew2) { oldKeyRes1 = existOld1 ? RegDeleteKey(HKEY_LOCAL_MACHINE, oldKeyStr1.toStdWString().c_str()) : ERROR_SUCCESS; oldKeyRes2 = existOld2 ? RegDeleteKey(HKEY_LOCAL_MACHINE, oldKeyStr2.toStdWString().c_str()) : ERROR_SUCCESS; } QString userDesktopLnk, commonDesktopLnk; WCHAR userDesktopFolder[MAX_PATH], commonDesktopFolder[MAX_PATH]; HRESULT userDesktopRes = SHGetFolderPath(0, CSIDL_DESKTOPDIRECTORY, 0, SHGFP_TYPE_CURRENT, userDesktopFolder); HRESULT commonDesktopRes = SHGetFolderPath(0, CSIDL_COMMON_DESKTOPDIRECTORY, 0, SHGFP_TYPE_CURRENT, commonDesktopFolder); if (SUCCEEDED(userDesktopRes)) { userDesktopLnk = QString::fromWCharArray(userDesktopFolder) + "\\Telegram.lnk"; } if (SUCCEEDED(commonDesktopRes)) { commonDesktopLnk = QString::fromWCharArray(commonDesktopFolder) + "\\Telegram.lnk"; } QFile userDesktopFile(userDesktopLnk), commonDesktopFile(commonDesktopLnk); if (QFile::exists(userDesktopLnk) && QFile::exists(commonDesktopLnk) && userDesktopLnk != commonDesktopLnk) { bool removed = QFile::remove(commonDesktopLnk); } } catch (...) { } } int psFixPrevious() { __try { psDoFixPrevious(); } __except(EXCEPTION_EXECUTE_HANDLER) { return 0; } return 0; } namespace Platform { void start() { Dlls::init(); } void finish() { EventFilter::destroy(); } QString SystemCountry() { int chCount = GetLocaleInfo(LOCALE_USER_DEFAULT, LOCALE_SISO3166CTRYNAME, 0, 0); if (chCount && chCount < 128) { WCHAR wstrCountry[128]; int len = GetLocaleInfo(LOCALE_USER_DEFAULT, LOCALE_SISO3166CTRYNAME, wstrCountry, chCount); if (len) { return QString::fromStdWString(std::wstring(wstrCountry)); } } return QString(); } bool IsApplicationActive() { return static_cast(QApplication::instance())->activeWindow() != nullptr; } QString CurrentExecutablePath(int argc, char *argv[]) { WCHAR result[MAX_PATH + 1] = { 0 }; auto count = GetModuleFileName(nullptr, result, MAX_PATH + 1); if (count < MAX_PATH + 1) { auto info = QFileInfo(QDir::fromNativeSeparators(QString::fromWCharArray(result))); return info.absoluteFilePath(); } // Fallback to the first command line argument. auto argsCount = 0; if (auto args = CommandLineToArgvW(GetCommandLine(), &argsCount)) { auto info = QFileInfo(QDir::fromNativeSeparators(QString::fromWCharArray(args[0]))); LocalFree(args); return info.absoluteFilePath(); } return QString(); } namespace { QString GetLangCodeById(unsigned lngId) { auto primary = (lngId & 0xFFU); switch (primary) { case 0x36: return qsl("af"); case 0x1C: return qsl("sq"); case 0x5E: return qsl("am"); case 0x01: return qsl("ar"); case 0x2B: return qsl("hy"); case 0x4D: return qsl("as"); case 0x2C: return qsl("az"); case 0x45: return qsl("bn"); case 0x6D: return qsl("ba"); case 0x2D: return qsl("eu"); case 0x23: return qsl("be"); case 0x1A: if (lngId == LANG_CROATIAN) { return qsl("hr"); } else if (lngId == LANG_BOSNIAN_NEUTRAL || lngId == LANG_BOSNIAN) { return qsl("bs"); } return qsl("sr"); break; case 0x7E: return qsl("br"); case 0x02: return qsl("bg"); case 0x92: return qsl("ku"); case 0x03: return qsl("ca"); case 0x04: return qsl("zh"); case 0x83: return qsl("co"); case 0x05: return qsl("cs"); case 0x06: return qsl("da"); case 0x65: return qsl("dv"); case 0x13: return qsl("nl"); case 0x09: return qsl("en"); case 0x25: return qsl("et"); case 0x38: return qsl("fo"); case 0x0B: return qsl("fi"); case 0x0c: return qsl("fr"); case 0x62: return qsl("fy"); case 0x56: return qsl("gl"); case 0x37: return qsl("ka"); case 0x07: return qsl("de"); case 0x08: return qsl("el"); case 0x6F: return qsl("kl"); case 0x47: return qsl("gu"); case 0x68: return qsl("ha"); case 0x0D: return qsl("he"); case 0x39: return qsl("hi"); case 0x0E: return qsl("hu"); case 0x0F: return qsl("is"); case 0x70: return qsl("ig"); case 0x21: return qsl("id"); case 0x5D: return qsl("iu"); case 0x3C: return qsl("ga"); case 0x34: return qsl("xh"); case 0x35: return qsl("zu"); case 0x10: return qsl("it"); case 0x11: return qsl("ja"); case 0x4B: return qsl("kn"); case 0x3F: return qsl("kk"); case 0x53: return qsl("kh"); case 0x87: return qsl("rw"); case 0x12: return qsl("ko"); case 0x40: return qsl("ky"); case 0x54: return qsl("lo"); case 0x26: return qsl("lv"); case 0x27: return qsl("lt"); case 0x6E: return qsl("lb"); case 0x2F: return qsl("mk"); case 0x3E: return qsl("ms"); case 0x4C: return qsl("ml"); case 0x3A: return qsl("mt"); case 0x81: return qsl("mi"); case 0x4E: return qsl("mr"); case 0x50: return qsl("mn"); case 0x61: return qsl("ne"); case 0x14: return qsl("no"); case 0x82: return qsl("oc"); case 0x48: return qsl("or"); case 0x63: return qsl("ps"); case 0x29: return qsl("fa"); case 0x15: return qsl("pl"); case 0x16: return qsl("pt"); case 0x67: return qsl("ff"); case 0x46: return qsl("pa"); case 0x18: return qsl("ro"); case 0x17: return qsl("rm"); case 0x19: return qsl("ru"); case 0x3B: return qsl("se"); case 0x4F: return qsl("sa"); case 0x32: return qsl("tn"); case 0x59: return qsl("sd"); case 0x5B: return qsl("si"); case 0x1B: return qsl("sk"); case 0x24: return qsl("sl"); case 0x0A: return qsl("es"); case 0x41: return qsl("sw"); case 0x1D: return qsl("sv"); case 0x28: return qsl("tg"); case 0x49: return qsl("ta"); case 0x44: return qsl("tt"); case 0x4A: return qsl("te"); case 0x1E: return qsl("th"); case 0x51: return qsl("bo"); case 0x73: return qsl("ti"); case 0x1F: return qsl("tr"); case 0x42: return qsl("tk"); case 0x22: return qsl("uk"); case 0x20: return qsl("ur"); case 0x80: return qsl("ug"); case 0x43: return qsl("uz"); case 0x2A: return qsl("vi"); case 0x52: return qsl("cy"); case 0x88: return qsl("wo"); case 0x78: return qsl("ii"); case 0x6A: return qsl("yo"); } return QString(); } } // namespace QString SystemLanguage() { constexpr auto kMaxLanguageLength = 128; auto uiLanguageId = GetUserDefaultUILanguage(); auto uiLanguageLength = GetLocaleInfo(uiLanguageId, LOCALE_SNAME, nullptr, 0); if (uiLanguageLength > 0 && uiLanguageLength < kMaxLanguageLength) { WCHAR uiLanguageWideString[kMaxLanguageLength] = { 0 }; uiLanguageLength = GetLocaleInfo(uiLanguageId, LOCALE_SNAME, uiLanguageWideString, uiLanguageLength); if (uiLanguageLength <= 0) { return QString(); } return QString::fromWCharArray(uiLanguageWideString); } auto uiLanguageCodeLength = GetLocaleInfo(uiLanguageId, LOCALE_ILANGUAGE, nullptr, 0); if (uiLanguageCodeLength > 0 && uiLanguageCodeLength < kMaxLanguageLength) { WCHAR uiLanguageCodeWideString[kMaxLanguageLength] = { 0 }; uiLanguageCodeLength = GetLocaleInfo(uiLanguageId, LOCALE_ILANGUAGE, uiLanguageCodeWideString, uiLanguageCodeLength); if (uiLanguageCodeLength <= 0) { return QString(); } auto languageCode = 0U; for (auto i = 0; i != uiLanguageCodeLength; ++i) { auto ch = uiLanguageCodeWideString[i]; if (!ch) { break; } languageCode *= 0x10U; if (ch >= WCHAR('0') && ch <= WCHAR('9')) { languageCode += static_cast(int(ch) - int(WCHAR('0'))); } else if (ch >= WCHAR('A') && ch <= WCHAR('F')) { languageCode += static_cast(0x0A + int(ch) - int(WCHAR('A'))); } else { return QString(); } } return GetLangCodeById(languageCode); } return QString(); } } // namespace Platform namespace { void _psLogError(const char *str, LSTATUS code) { LPWSTR errorTextFormatted = nullptr; auto formatFlags = FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_IGNORE_INSERTS; FormatMessage( formatFlags, NULL, code, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR)&errorTextFormatted, 0, 0); auto errorText = errorTextFormatted ? errorTextFormatted : L"(Unknown error)"; LOG((str).arg(code).arg(QString::fromStdWString(errorText))); LocalFree(errorTextFormatted); } bool _psOpenRegKey(LPCWSTR key, PHKEY rkey) { DEBUG_LOG(("App Info: opening reg key %1...").arg(QString::fromStdWString(key))); LSTATUS status = RegOpenKeyEx(HKEY_CURRENT_USER, key, 0, KEY_QUERY_VALUE | KEY_WRITE, rkey); if (status != ERROR_SUCCESS) { if (status == ERROR_FILE_NOT_FOUND) { status = RegCreateKeyEx(HKEY_CURRENT_USER, key, 0, 0, REG_OPTION_NON_VOLATILE, KEY_QUERY_VALUE | KEY_WRITE, 0, rkey, 0); if (status != ERROR_SUCCESS) { QString msg = qsl("App Error: could not create '%1' registry key, error %2").arg(QString::fromStdWString(key)).arg(qsl("%1: %2")); _psLogError(msg.toUtf8().constData(), status); return false; } } else { QString msg = qsl("App Error: could not open '%1' registry key, error %2").arg(QString::fromStdWString(key)).arg(qsl("%1: %2")); _psLogError(msg.toUtf8().constData(), status); return false; } } return true; } bool _psSetKeyValue(HKEY rkey, LPCWSTR value, QString v) { static const int bufSize = 4096; DWORD defaultType, defaultSize = bufSize * 2; WCHAR defaultStr[bufSize] = { 0 }; if (RegQueryValueEx(rkey, value, 0, &defaultType, (BYTE*)defaultStr, &defaultSize) != ERROR_SUCCESS || defaultType != REG_SZ || defaultSize != (v.size() + 1) * 2 || QString::fromStdWString(defaultStr) != v) { WCHAR tmp[bufSize] = { 0 }; if (!v.isEmpty()) wsprintf(tmp, v.replace(QChar('%'), qsl("%%")).toStdWString().c_str()); LSTATUS status = RegSetValueEx(rkey, value, 0, REG_SZ, (BYTE*)tmp, (wcslen(tmp) + 1) * sizeof(WCHAR)); if (status != ERROR_SUCCESS) { QString msg = qsl("App Error: could not set %1, error %2").arg(value ? ('\'' + QString::fromStdWString(value) + '\'') : qsl("(Default)")).arg("%1: %2"); _psLogError(msg.toUtf8().constData(), status); return false; } } return true; } } namespace Platform { void RegisterCustomScheme() { if (cExeName().isEmpty()) { return; } #ifndef TDESKTOP_DISABLE_REGISTER_CUSTOM_SCHEME DEBUG_LOG(("App Info: Checking custom scheme 'tg'...")); HKEY rkey; QString exe = QDir::toNativeSeparators(cExeDir() + cExeName()); // Legacy URI scheme registration if (!_psOpenRegKey(L"Software\\Classes\\tg", &rkey)) return; if (!_psSetKeyValue(rkey, L"URL Protocol", QString())) return; if (!_psSetKeyValue(rkey, 0, qsl("URL:Telegram Link"))) return; if (!_psOpenRegKey(L"Software\\Classes\\tg\\DefaultIcon", &rkey)) return; if (!_psSetKeyValue(rkey, 0, '"' + exe + qsl(",1\""))) return; if (!_psOpenRegKey(L"Software\\Classes\\tg\\shell", &rkey)) return; if (!_psOpenRegKey(L"Software\\Classes\\tg\\shell\\open", &rkey)) return; if (!_psOpenRegKey(L"Software\\Classes\\tg\\shell\\open\\command", &rkey)) return; if (!_psSetKeyValue(rkey, 0, '"' + exe + qsl("\" -workdir \"") + cWorkingDir() + qsl("\" -- \"%1\""))) return; // URI scheme registration as Default Program - Windows Vista and above if (!_psOpenRegKey(L"Software\\Classes\\tdesktop.tg", &rkey)) return; if (!_psOpenRegKey(L"Software\\Classes\\tdesktop.tg\\DefaultIcon", &rkey)) return; if (!_psSetKeyValue(rkey, 0, '"' + exe + qsl(",1\""))) return; if (!_psOpenRegKey(L"Software\\Classes\\tdesktop.tg\\shell", &rkey)) return; if (!_psOpenRegKey(L"Software\\Classes\\tdesktop.tg\\shell\\open", &rkey)) return; if (!_psOpenRegKey(L"Software\\Classes\\tdesktop.tg\\shell\\open\\command", &rkey)) return; if (!_psSetKeyValue(rkey, 0, '"' + exe + qsl("\" -workdir \"") + cWorkingDir() + qsl("\" -- \"%1\""))) return; if (!_psOpenRegKey(L"Software\\TelegramDesktop", &rkey)) return; if (!_psOpenRegKey(L"Software\\TelegramDesktop\\Capabilities", &rkey)) return; if (!_psSetKeyValue(rkey, L"ApplicationName", qsl("Telegram Desktop"))) return; if (!_psSetKeyValue(rkey, L"ApplicationDescription", qsl("Telegram Desktop"))) return; if (!_psOpenRegKey(L"Software\\TelegramDesktop\\Capabilities\\UrlAssociations", &rkey)) return; if (!_psSetKeyValue(rkey, L"tg", qsl("tdesktop.tg"))) return; if (!_psOpenRegKey(L"Software\\RegisteredApplications", &rkey)) return; if (!_psSetKeyValue(rkey, L"Telegram Desktop", qsl("SOFTWARE\\TelegramDesktop\\Capabilities"))) return; #endif // !TDESKTOP_DISABLE_REGISTER_CUSTOM_SCHEME } } // namespace Platform void psNewVersion() { Platform::RegisterCustomScheme(); if (Local::oldSettingsVersion() < 8051) { AppUserModelId::checkPinned(); } if (Local::oldSettingsVersion() > 0 && Local::oldSettingsVersion() < 10021) { // Reset icons cache, because we've changed the application icon. if (Dlls::SHChangeNotify) { Dlls::SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, nullptr, nullptr); } } } void _manageAppLnk(bool create, bool silent, int path_csidl, const wchar_t *args, const wchar_t *description) { if (cExeName().isEmpty()) { return; } WCHAR startupFolder[MAX_PATH]; HRESULT hr = SHGetFolderPath(0, path_csidl, 0, SHGFP_TYPE_CURRENT, startupFolder); if (SUCCEEDED(hr)) { QString lnk = QString::fromWCharArray(startupFolder) + '\\' + str_const_toString(AppFile) + qsl(".lnk"); if (create) { ComPtr shellLink; hr = CoCreateInstance(CLSID_ShellLink, nullptr, CLSCTX_INPROC_SERVER, IID_PPV_ARGS(&shellLink)); if (SUCCEEDED(hr)) { ComPtr persistFile; QString exe = QDir::toNativeSeparators(cExeDir() + cExeName()), dir = QDir::toNativeSeparators(QDir(cWorkingDir()).absolutePath()); shellLink->SetArguments(args); shellLink->SetPath(exe.toStdWString().c_str()); shellLink->SetWorkingDirectory(dir.toStdWString().c_str()); shellLink->SetDescription(description); ComPtr propertyStore; hr = shellLink.As(&propertyStore); if (SUCCEEDED(hr)) { PROPVARIANT appIdPropVar; hr = InitPropVariantFromString(AppUserModelId::getId(), &appIdPropVar); if (SUCCEEDED(hr)) { hr = propertyStore->SetValue(AppUserModelId::getKey(), appIdPropVar); PropVariantClear(&appIdPropVar); if (SUCCEEDED(hr)) { hr = propertyStore->Commit(); } } } hr = shellLink.As(&persistFile); if (SUCCEEDED(hr)) { hr = persistFile->Save(lnk.toStdWString().c_str(), TRUE); } else { if (!silent) LOG(("App Error: could not create interface IID_IPersistFile %1").arg(hr)); } } else { if (!silent) LOG(("App Error: could not create instance of IID_IShellLink %1").arg(hr)); } } else { QFile::remove(lnk); } } else { if (!silent) LOG(("App Error: could not get CSIDL %1 folder %2").arg(path_csidl).arg(hr)); } } void psAutoStart(bool start, bool silent) { _manageAppLnk(start, silent, CSIDL_STARTUP, L"-autostart", L"Telegram autorun link.\nYou can disable autorun in Telegram settings."); } void psSendToMenu(bool send, bool silent) { _manageAppLnk(send, silent, CSIDL_SENDTO, L"-sendpath", L"Telegram send to link.\nYou can disable send to menu item in Telegram settings."); } void psUpdateOverlayed(TWidget *widget) { bool wm = widget->testAttribute(Qt::WA_Mapped), wv = widget->testAttribute(Qt::WA_WState_Visible); if (!wm) widget->setAttribute(Qt::WA_Mapped, true); if (!wv) widget->setAttribute(Qt::WA_WState_Visible, true); widget->update(); QEvent e(QEvent::UpdateRequest); QGuiApplication::sendEvent(widget, &e); if (!wm) widget->setAttribute(Qt::WA_Mapped, false); if (!wv) widget->setAttribute(Qt::WA_WState_Visible, false); } // Stack walk code is inspired by http://www.codeproject.com/Articles/11132/Walking-the-callstack static const int StackEntryMaxNameLength = MAX_SYM_NAME + 1; typedef BOOL(FAR STDAPICALLTYPE *t_SymCleanup)( _In_ HANDLE hProcess ); t_SymCleanup symCleanup = 0; typedef PVOID (FAR STDAPICALLTYPE *t_SymFunctionTableAccess64)( _In_ HANDLE hProcess, _In_ DWORD64 AddrBase ); t_SymFunctionTableAccess64 symFunctionTableAccess64 = 0; typedef BOOL (FAR STDAPICALLTYPE *t_SymGetLineFromAddr64)( _In_ HANDLE hProcess, _In_ DWORD64 dwAddr, _Out_ PDWORD pdwDisplacement, _Out_ PIMAGEHLP_LINEW64 Line ); t_SymGetLineFromAddr64 symGetLineFromAddr64 = 0; typedef DWORD64 (FAR STDAPICALLTYPE *t_SymGetModuleBase64)( _In_ HANDLE hProcess, _In_ DWORD64 qwAddr ); t_SymGetModuleBase64 symGetModuleBase64 = 0; typedef BOOL (FAR STDAPICALLTYPE *t_SymGetModuleInfo64)( _In_ HANDLE hProcess, _In_ DWORD64 qwAddr, _Out_ PIMAGEHLP_MODULEW64 ModuleInfo ); t_SymGetModuleInfo64 symGetModuleInfo64 = 0; typedef DWORD (FAR STDAPICALLTYPE *t_SymGetOptions)( VOID ); t_SymGetOptions symGetOptions = 0; typedef DWORD (FAR STDAPICALLTYPE *t_SymSetOptions)( _In_ DWORD SymOptions ); t_SymSetOptions symSetOptions = 0; typedef BOOL (FAR STDAPICALLTYPE *t_SymGetSymFromAddr64)( IN HANDLE hProcess, IN DWORD64 dwAddr, OUT PDWORD64 pdwDisplacement, OUT PIMAGEHLP_SYMBOL64 Symbol ); t_SymGetSymFromAddr64 symGetSymFromAddr64 = 0; typedef BOOL (FAR STDAPICALLTYPE *t_SymInitialize)( _In_ HANDLE hProcess, _In_opt_ PCWSTR UserSearchPath, _In_ BOOL fInvadeProcess ); t_SymInitialize symInitialize = 0; typedef DWORD64 (FAR STDAPICALLTYPE *t_SymLoadModule64)( _In_ HANDLE hProcess, _In_opt_ HANDLE hFile, _In_opt_ PCSTR ImageName, _In_opt_ PCSTR ModuleName, _In_ DWORD64 BaseOfDll, _In_ DWORD SizeOfDll ); t_SymLoadModule64 symLoadModule64; typedef BOOL (FAR STDAPICALLTYPE *t_StackWalk64)( _In_ DWORD MachineType, _In_ HANDLE hProcess, _In_ HANDLE hThread, _Inout_ LPSTACKFRAME64 StackFrame, _Inout_ PVOID ContextRecord, _In_opt_ PREAD_PROCESS_MEMORY_ROUTINE64 ReadMemoryRoutine, _In_opt_ PFUNCTION_TABLE_ACCESS_ROUTINE64 FunctionTableAccessRoutine, _In_opt_ PGET_MODULE_BASE_ROUTINE64 GetModuleBaseRoutine, _In_opt_ PTRANSLATE_ADDRESS_ROUTINE64 TranslateAddress ); t_StackWalk64 stackWalk64 = 0; typedef DWORD (FAR STDAPICALLTYPE *t_UnDecorateSymbolName)( PCSTR DecoratedName, PSTR UnDecoratedName, DWORD UndecoratedLength, DWORD Flags ); t_UnDecorateSymbolName unDecorateSymbolName = 0; typedef BOOL(FAR STDAPICALLTYPE *t_SymGetSearchPath)( _In_ HANDLE hProcess, _Out_writes_(SearchPathLength) PWSTR SearchPath, _In_ DWORD SearchPathLength ); t_SymGetSearchPath symGetSearchPath = 0; BOOL __stdcall ReadProcessMemoryRoutine64( _In_ HANDLE hProcess, _In_ DWORD64 qwBaseAddress, _Out_writes_bytes_(nSize) PVOID lpBuffer, _In_ DWORD nSize, _Out_ LPDWORD lpNumberOfBytesRead ) { SIZE_T st; BOOL bRet = ReadProcessMemory(hProcess, (LPVOID)qwBaseAddress, lpBuffer, nSize, &st); *lpNumberOfBytesRead = (DWORD)st; return bRet; } // **************************************** ToolHelp32 ************************ #define MAX_MODULE_NAME32 255 #define TH32CS_SNAPMODULE 0x00000008 #pragma pack( push, 8 ) typedef struct tagMODULEENTRY32 { DWORD dwSize; DWORD th32ModuleID; // This module DWORD th32ProcessID; // owning process DWORD GlblcntUsage; // Global usage count on the module DWORD ProccntUsage; // Module usage count in th32ProcessID's context BYTE * modBaseAddr; // Base address of module in th32ProcessID's context DWORD modBaseSize; // Size in bytes of module starting at modBaseAddr HMODULE hModule; // The hModule of this module in th32ProcessID's context char szModule[MAX_MODULE_NAME32 + 1]; char szExePath[MAX_PATH]; } MODULEENTRY32; typedef MODULEENTRY32 *PMODULEENTRY32; typedef MODULEENTRY32 *LPMODULEENTRY32; #pragma pack( pop ) typedef HANDLE (FAR STDAPICALLTYPE *t_CreateToolhelp32Snapshot)(DWORD dwFlags, DWORD th32ProcessID); t_CreateToolhelp32Snapshot createToolhelp32Snapshot = 0; typedef BOOL (FAR STDAPICALLTYPE *t_Module32First)(HANDLE hSnapshot, LPMODULEENTRY32 lpme); t_Module32First module32First = 0; typedef BOOL (FAR STDAPICALLTYPE *t_Module32Next)(HANDLE hSnapshot, LPMODULEENTRY32 lpme); t_Module32Next module32Next = 0; bool LoadDbgHelp(bool extended = false) { if (stackWalk64 && (!extended || symInitialize)) return true; HMODULE hDll = 0; WCHAR szTemp[4096]; if (GetModuleFileName(NULL, szTemp, 4096) > 0) { wcscat_s(szTemp, L".local"); if (GetFileAttributes(szTemp) == INVALID_FILE_ATTRIBUTES) { // ".local" file does not exist, so we can try to load the dbghelp.dll from the "Debugging Tools for Windows" if (GetEnvironmentVariable(L"ProgramFiles", szTemp, 4096) > 0) { wcscat_s(szTemp, L"\\Debugging Tools for Windows\\dbghelp.dll"); // now check if the file exists: if (GetFileAttributes(szTemp) != INVALID_FILE_ATTRIBUTES) { hDll = LoadLibrary(szTemp); } } // Still not found? Then try to load the 64-Bit version: if (!hDll && (GetEnvironmentVariable(L"ProgramFiles", szTemp, 4096) > 0)) { wcscat_s(szTemp, L"\\Debugging Tools for Windows 64-Bit\\dbghelp.dll"); if (GetFileAttributes(szTemp) != INVALID_FILE_ATTRIBUTES) { hDll = LoadLibrary(szTemp); } } } } if (!hDll) { hDll = LoadLibrary(L"DBGHELP.DLL"); } if (!hDll) return false; stackWalk64 = (t_StackWalk64)GetProcAddress(hDll, "StackWalk64"); symFunctionTableAccess64 = (t_SymFunctionTableAccess64)GetProcAddress(hDll, "SymFunctionTableAccess64"); symGetModuleBase64 = (t_SymGetModuleBase64)GetProcAddress(hDll, "SymGetModuleBase64"); if (!stackWalk64 || !symFunctionTableAccess64 || !symGetModuleBase64) { stackWalk64 = 0; return false; } if (extended) { HANDLE hProcess = GetCurrentProcess(); DWORD dwProcessId = GetCurrentProcessId(); symGetLineFromAddr64 = (t_SymGetLineFromAddr64)GetProcAddress(hDll, "SymGetLineFromAddrW64"); symGetModuleInfo64 = (t_SymGetModuleInfo64)GetProcAddress(hDll, "SymGetModuleInfoW64"); symGetSymFromAddr64 = (t_SymGetSymFromAddr64)GetProcAddress(hDll, "SymGetSymFromAddr64"); unDecorateSymbolName = (t_UnDecorateSymbolName)GetProcAddress(hDll, "UnDecorateSymbolName"); symInitialize = (t_SymInitialize)GetProcAddress(hDll, "SymInitializeW"); symCleanup = (t_SymCleanup)GetProcAddress(hDll, "SymCleanup"); symGetSearchPath = (t_SymGetSearchPath)GetProcAddress(hDll, "SymGetSearchPathW"); symGetOptions = (t_SymGetOptions)GetProcAddress(hDll, "SymGetOptions"); symSetOptions = (t_SymSetOptions)GetProcAddress(hDll, "SymSetOptions"); symLoadModule64 = (t_SymLoadModule64)GetProcAddress(hDll, "SymLoadModule64"); if (!symGetModuleInfo64 || !symGetLineFromAddr64 || !symGetSymFromAddr64 || !unDecorateSymbolName || !symInitialize || !symCleanup || !symGetOptions || !symSetOptions || !symLoadModule64) { symInitialize = 0; return false; } const size_t nSymPathLen = 10 * MAX_PATH; WCHAR szSymPath[nSymPathLen] = { 0 }; wcscat_s(szSymPath, nSymPathLen, L".;..;"); WCHAR szTemp[MAX_PATH + 1] = { 0 }; if (GetCurrentDirectory(MAX_PATH, szTemp) > 0) { wcscat_s(szSymPath, nSymPathLen, szTemp); wcscat_s(szSymPath, nSymPathLen, L";"); } if (GetModuleFileName(NULL, szTemp, MAX_PATH) > 0) { for (WCHAR *p = (szTemp + wcslen(szTemp) - 1); p >= szTemp; --p) { if ((*p == '\\') || (*p == '/') || (*p == ':')) { *p = 0; break; } } if (wcslen(szTemp) > 0) { wcscat_s(szSymPath, nSymPathLen, szTemp); wcscat_s(szSymPath, nSymPathLen, L";"); } } if (GetEnvironmentVariable(L"_NT_SYMBOL_PATH", szTemp, MAX_PATH) > 0) { wcscat_s(szSymPath, nSymPathLen, szTemp); wcscat_s(szSymPath, nSymPathLen, L";"); } if (GetEnvironmentVariable(L"_NT_ALTERNATE_SYMBOL_PATH", szTemp, MAX_PATH) > 0) { wcscat_s(szSymPath, nSymPathLen, szTemp); wcscat_s(szSymPath, nSymPathLen, L";"); } if (GetEnvironmentVariable(L"SYSTEMROOT", szTemp, MAX_PATH) > 0) { wcscat_s(szSymPath, nSymPathLen, szTemp); wcscat_s(szSymPath, nSymPathLen, L";"); // also add the "system32"-directory: wcscat_s(szTemp, MAX_PATH, L"\\system32"); wcscat_s(szSymPath, nSymPathLen, szTemp); wcscat_s(szSymPath, nSymPathLen, L";"); } if (GetEnvironmentVariable(L"SYSTEMDRIVE", szTemp, MAX_PATH) > 0) { wcscat_s(szSymPath, nSymPathLen, L"SRV*"); wcscat_s(szSymPath, nSymPathLen, szTemp); wcscat_s(szSymPath, nSymPathLen, L"\\websymbols*http://msdl.microsoft.com/download/symbols;"); } else { wcscat_s(szSymPath, nSymPathLen, L"SRV*c:\\websymbols*http://msdl.microsoft.com/download/symbols;"); } if (symInitialize(hProcess, szSymPath, FALSE) == FALSE) { symInitialize = 0; return false; } DWORD symOptions = symGetOptions(); symOptions |= SYMOPT_LOAD_LINES; symOptions |= SYMOPT_FAIL_CRITICAL_ERRORS; symOptions = symSetOptions(symOptions); const WCHAR *dllname[] = { L"kernel32.dll", L"tlhelp32.dll" }; HINSTANCE hToolhelp = NULL; HANDLE hSnap; MODULEENTRY32 me; me.dwSize = sizeof(me); BOOL keepGoing; size_t i; for (i = 0; i < (sizeof(dllname) / sizeof(dllname[0])); i++) { hToolhelp = LoadLibrary(dllname[i]); if (!hToolhelp) continue; createToolhelp32Snapshot = (t_CreateToolhelp32Snapshot)GetProcAddress(hToolhelp, "CreateToolhelp32Snapshot"); module32First = (t_Module32First)GetProcAddress(hToolhelp, "Module32First"); module32Next = (t_Module32Next)GetProcAddress(hToolhelp, "Module32Next"); if (createToolhelp32Snapshot && module32First && module32Next) { break; // found the functions! } FreeLibrary(hToolhelp); hToolhelp = NULL; } if (hToolhelp == NULL) { return false; } hSnap = createToolhelp32Snapshot(TH32CS_SNAPMODULE, dwProcessId); if (hSnap == (HANDLE)-1) return FALSE; keepGoing = !!module32First(hSnap, &me); int cnt = 0; while (keepGoing) { symLoadModule64(hProcess, 0, me.szExePath, me.szModule, (DWORD64)me.modBaseAddr, me.modBaseSize); ++cnt; keepGoing = !!module32Next(hSnap, &me); } CloseHandle(hSnap); FreeLibrary(hToolhelp); return (cnt > 0); } return true; } struct StackEntry { DWORD64 offset; // if 0, we have no valid entry CHAR name[StackEntryMaxNameLength]; CHAR undName[StackEntryMaxNameLength]; CHAR undFullName[StackEntryMaxNameLength]; DWORD64 offsetFromSmybol; DWORD offsetFromLine; DWORD lineNumber; WCHAR lineFileName[StackEntryMaxNameLength]; DWORD symType; LPCSTR symTypeString; WCHAR moduleName[StackEntryMaxNameLength]; DWORD64 baseOfImage; WCHAR loadedImageName[StackEntryMaxNameLength]; }; enum StackEntryType { StackEntryFirst, StackEntryNext, StackEntryLast, }; char GetModuleInfoData[2 * sizeof(IMAGEHLP_MODULEW64)]; BOOL _getModuleInfo(HANDLE hProcess, DWORD64 baseAddr, IMAGEHLP_MODULEW64 *pModuleInfo) { pModuleInfo->SizeOfStruct = sizeof(IMAGEHLP_MODULEW64); memcpy(GetModuleInfoData, pModuleInfo, sizeof(IMAGEHLP_MODULEW64)); if (symGetModuleInfo64(hProcess, baseAddr, (IMAGEHLP_MODULEW64*)GetModuleInfoData) != FALSE) { // only copy as much memory as is reserved... memcpy(pModuleInfo, GetModuleInfoData, sizeof(IMAGEHLP_MODULEW64)); pModuleInfo->SizeOfStruct = sizeof(IMAGEHLP_MODULEW64); return TRUE; } return FALSE; } void psWriteDump() { } void psWriteStackTrace() { #ifndef TDESKTOP_DISABLE_CRASH_REPORTS if (!LoadDbgHelp()) { CrashReports::dump() << "ERROR: Could not load dbghelp.dll!\n"; return; } HANDLE hThread = GetCurrentThread(), hProcess = GetCurrentProcess(); const CONTEXT *context = NULL; LPVOID pUserData = NULL; CONTEXT c; int frameNum; memset(&c, 0, sizeof(CONTEXT)); c.ContextFlags = CONTEXT_FULL; RtlCaptureContext(&c); // init STACKFRAME for first call STACKFRAME64 s; // in/out stackframe memset(&s, 0, sizeof(s)); DWORD imageType; #ifdef _M_IX86 // normally, call ImageNtHeader() and use machine info from PE header imageType = IMAGE_FILE_MACHINE_I386; s.AddrPC.Offset = c.Eip; s.AddrPC.Mode = AddrModeFlat; s.AddrFrame.Offset = c.Ebp; s.AddrFrame.Mode = AddrModeFlat; s.AddrStack.Offset = c.Esp; s.AddrStack.Mode = AddrModeFlat; #elif _M_X64 imageType = IMAGE_FILE_MACHINE_AMD64; s.AddrPC.Offset = c.Rip; s.AddrPC.Mode = AddrModeFlat; s.AddrFrame.Offset = c.Rsp; s.AddrFrame.Mode = AddrModeFlat; s.AddrStack.Offset = c.Rsp; s.AddrStack.Mode = AddrModeFlat; #elif _M_IA64 imageType = IMAGE_FILE_MACHINE_IA64; s.AddrPC.Offset = c.StIIP; s.AddrPC.Mode = AddrModeFlat; s.AddrFrame.Offset = c.IntSp; s.AddrFrame.Mode = AddrModeFlat; s.AddrBStore.Offset = c.RsBSP; s.AddrBStore.Mode = AddrModeFlat; s.AddrStack.Offset = c.IntSp; s.AddrStack.Mode = AddrModeFlat; #else #error "Platform not supported!" #endif for (frameNum = 0; frameNum < 1024; ++frameNum) { // get next stack frame (StackWalk64(), SymFunctionTableAccess64(), SymGetModuleBase64()) // if this returns ERROR_INVALID_ADDRESS (487) or ERROR_NOACCESS (998), you can // assume that either you are done, or that the stack is so hosed that the next // deeper frame could not be found. // CONTEXT need not to be suplied if imageTyp is IMAGE_FILE_MACHINE_I386! if (!stackWalk64(imageType, hProcess, hThread, &s, &c, ReadProcessMemoryRoutine64, symFunctionTableAccess64, symGetModuleBase64, NULL)) { CrashReports::dump() << "ERROR: Call to StackWalk64() failed!\n"; return; } if (s.AddrPC.Offset == s.AddrReturn.Offset) { CrashReports::dump() << s.AddrPC.Offset << "\n"; CrashReports::dump() << "ERROR: StackWalk64() endless callstack!"; return; } if (s.AddrPC.Offset != 0) { // we seem to have a valid PC CrashReports::dump() << s.AddrPC.Offset << "\n"; } if (s.AddrReturn.Offset == 0) { break; } } #endif // !TDESKTOP_DISABLE_CRASH_REPORTS } bool psLaunchMaps(const LocationCoords &coords) { return QDesktopServices::openUrl(qsl("bingmaps:?lvl=16&collection=point.%1_%2_Point").arg(coords.latAsString()).arg(coords.lonAsString())); }