2430 lines
		
	
	
	
		
			78 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			2430 lines
		
	
	
	
		
			78 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /*
 | |
| This file is part of Telegram Desktop,
 | |
| the official desktop version of Telegram messaging app, see https://telegram.org
 | |
| 
 | |
| Telegram Desktop is free software: you can redistribute it and/or modify
 | |
| it under the terms of the GNU General Public License as published by
 | |
| the Free Software Foundation, either version 3 of the License, or
 | |
| (at your option) any later version.
 | |
| 
 | |
| It is distributed in the hope that it will be useful,
 | |
| but WITHOUT ANY WARRANTY; without even the implied warranty of
 | |
| MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 | |
| GNU General Public License for more details.
 | |
| 
 | |
| Full license: https://github.com/telegramdesktop/tdesktop/blob/master/LICENSE
 | |
| Copyright (c) 2014 John Preston, https://desktop.telegram.org
 | |
| */
 | |
| #include "stdafx.h"
 | |
| #include "pspecific.h"
 | |
| 
 | |
| #include "lang.h"
 | |
| #include "application.h"
 | |
| #include "mainwidget.h"
 | |
| 
 | |
| #include <Shobjidl.h>
 | |
| #include <dbghelp.h>
 | |
| #include <shellapi.h>
 | |
| #include <Strsafe.h>
 | |
| #include <shlobj.h>
 | |
| #include <Windowsx.h>
 | |
| 
 | |
| #include <qpa/qplatformnativeinterface.h>
 | |
| 
 | |
| #include <dwmapi.h>
 | |
| 
 | |
| #define min(a, b) ((a) < (b) ? (a) : (b))
 | |
| #define max(a, b) ((a) < (b) ? (b) : (a))
 | |
| 
 | |
| #ifndef DCX_USESTYLE
 | |
| #define DCX_USESTYLE 0x00010000
 | |
| #endif
 | |
| 
 | |
| #ifndef WM_NCPOINTERUPDATE
 | |
| #define WM_NCPOINTERUPDATE              0x0241
 | |
| #define WM_NCPOINTERDOWN                0x0242
 | |
| #define WM_NCPOINTERUP                  0x0243
 | |
| #endif
 | |
| 
 | |
| #include <gdiplus.h>
 | |
| #pragma comment (lib,"Gdiplus.lib")
 | |
| #pragma comment (lib,"Msimg32.lib")
 | |
| 
 | |
| namespace {
 | |
| 	bool frameless = true;
 | |
| 	bool useDWM = false;
 | |
| 	bool useTheme = false;
 | |
| 	bool useOpenAs = false;
 | |
| 	bool themeInited = false;
 | |
| 	bool finished = true;
 | |
| 	int menuShown = 0, menuHidden = 0;
 | |
| 	int dleft = 0, dtop = 0;
 | |
| 	QMargins simpleMargins, margins;
 | |
| 	HICON bigIcon = 0, smallIcon = 0, overlayIcon = 0;
 | |
| 
 | |
| 	UINT tbCreatedMsgId = 0;
 | |
| 	ITaskbarList3 *tbListInterface = 0;
 | |
| 
 | |
| 	HWND createTaskbarHider() {
 | |
| 		HINSTANCE appinst = (HINSTANCE)GetModuleHandle(0);
 | |
| 		HWND hWnd = 0;
 | |
| 		
 | |
| 		QString cn = QString("TelegramTaskbarHider");
 | |
| 		LPCWSTR _cn = (LPCWSTR)cn.utf16();
 | |
| 		WNDCLASSEX wc;
 | |
| 
 | |
| 		wc.cbSize        = sizeof(wc);
 | |
| 		wc.style         = 0;
 | |
| 		wc.lpfnWndProc   = DefWindowProc;
 | |
| 		wc.cbClsExtra    = 0;
 | |
| 		wc.cbWndExtra    = 0;
 | |
| 		wc.hInstance     = appinst;
 | |
| 		wc.hIcon         = 0;
 | |
| 		wc.hCursor       = 0;
 | |
| 		wc.hbrBackground = 0;
 | |
| 		wc.lpszMenuName  = NULL;
 | |
| 		wc.lpszClassName = _cn;
 | |
| 		wc.hIconSm       = 0;
 | |
| 		if (!RegisterClassEx(&wc)) {
 | |
| 			DEBUG_LOG(("Application Error: could not register taskbar hider window class, error: %1").arg(GetLastError()));
 | |
| 			return hWnd;
 | |
| 		}
 | |
| 		
 | |
| 		hWnd = CreateWindowEx(WS_EX_TOOLWINDOW, _cn, 0, WS_POPUP, 0, 0, 0, 0, 0, 0, appinst, 0);
 | |
| 		if (!hWnd) {
 | |
| 			DEBUG_LOG(("Application Error: could not create taskbar hider window class, error: %1").arg(GetLastError()));
 | |
| 			return hWnd;
 | |
| 		}
 | |
| 		return hWnd;
 | |
| 	}
 | |
| 
 | |
| 	enum {
 | |
| 		_PsShadowMoved = 0x01,
 | |
| 		_PsShadowResized = 0x02,
 | |
| 		_PsShadowShown = 0x04,
 | |
| 		_PsShadowHidden = 0x08,
 | |
| 		_PsShadowActivate = 0x10,
 | |
| 	};
 | |
| 
 | |
| 	enum {
 | |
| 		_PsInitHor = 0x01,
 | |
| 		_PsInitVer = 0x02,
 | |
| 	};
 | |
| 
 | |
| 	int32 _psSize = 0;
 | |
| 	class _PsShadowWindows {
 | |
| 	public:
 | |
| 
 | |
| 		_PsShadowWindows() : screenDC(0), max_w(0), max_h(0), _x(0), _y(0), _w(0), _h(0), hidden(true), r(0), g(0), b(0), noKeyColor(RGB(255, 255, 255)) {
 | |
| 			for (int i = 0; i < 4; ++i) {
 | |
| 				dcs[i] = 0;
 | |
| 				bitmaps[i] = 0;
 | |
| 				hwnds[i] = 0;
 | |
| 			}
 | |
| 		}
 | |
| 		
 | |
| 		void setColor(QColor c) {
 | |
| 			r = c.red();
 | |
| 			g = c.green();
 | |
| 			b = c.blue();
 | |
| 			
 | |
| 			if (!hwnds[0]) return;
 | |
| 			Gdiplus::SolidBrush brush(Gdiplus::Color(_alphas[0], r, g, b));
 | |
| 			for (int i = 0; i < 4; ++i) {
 | |
| 				Gdiplus::Graphics graphics(dcs[i]);
 | |
| 				graphics.SetCompositingMode(Gdiplus::CompositingModeSourceCopy);
 | |
| 				if ((i % 2) && _h || !(i % 2) && _w) {
 | |
| 					graphics.FillRectangle(&brush, 0, 0, (i % 2) ? _size : _w, (i % 2) ? _h : _size);
 | |
| 				}
 | |
| 			}
 | |
| 			initCorners();
 | |
| 
 | |
| 			_x = _y = _w = _h = 0;
 | |
| 			update(_PsShadowMoved | _PsShadowResized);
 | |
| 		}
 | |
| 
 | |
| 		bool init(QColor c) {
 | |
| 			style::rect topLeft = st::wndShadow;
 | |
| 			_fullsize = topLeft.width();
 | |
| 			_shift = st::wndShadowShift;
 | |
| 			QImage cornersImage(_fullsize, _fullsize, QImage::Format_ARGB32_Premultiplied);
 | |
| 			{
 | |
| 				QPainter p(&cornersImage);
 | |
| 				p.drawPixmap(QPoint(0, 0), App::sprite(), topLeft);
 | |
| 			}
 | |
| 			uchar *bits = cornersImage.bits();
 | |
| 			if (bits) {
 | |
| 				for (
 | |
| 					quint32 *p = (quint32*)bits, *end = (quint32*)(bits + cornersImage.byteCount());
 | |
| 					p < end;
 | |
| 				++p
 | |
| 					) {
 | |
| 					*p = (*p ^ 0x00ffffff) << 24;
 | |
| 				}
 | |
| 			}
 | |
| 
 | |
| 			_metaSize = _fullsize + 2 * _shift;
 | |
| 			_alphas.reserve(_metaSize);
 | |
| 			_colors.reserve(_metaSize * _metaSize);
 | |
| 			for (int32 j = 0; j < _metaSize; ++j) {
 | |
| 				for (int32 i = 0; i < _metaSize; ++i) {
 | |
| 					_colors.push_back((i < 2 * _shift || j < 2 * _shift) ? 1 : qMax(BYTE(1), BYTE(cornersImage.pixel(QPoint(i - 2 * _shift, j - 2 * _shift)) >> 24)));
 | |
| 				}
 | |
| 			}
 | |
| 			uchar prev = 0;
 | |
| 			for (int32 i = 0; i < _metaSize; ++i) {
 | |
| 				uchar a = _colors[(_metaSize - 1) * _metaSize + i];
 | |
| 				if (a < prev) break;
 | |
| 
 | |
| 				_alphas.push_back(a);
 | |
| 				prev = a;
 | |
| 			}
 | |
| 			_psSize = _size = _alphas.size() - 2 * _shift;
 | |
| 
 | |
| 			setColor(c);
 | |
| 
 | |
| 			Gdiplus::GdiplusStartupInput gdiplusStartupInput;
 | |
| 			ULONG_PTR gdiplusToken;
 | |
| 			Gdiplus::Status gdiRes = Gdiplus::GdiplusStartup(&gdiplusToken, &gdiplusStartupInput, NULL);
 | |
| 	
 | |
| 			if (gdiRes != Gdiplus::Ok) {
 | |
| 				DEBUG_LOG(("Application Error: could not init GDI+, error: %1").arg((int)gdiRes));
 | |
| 				return false;
 | |
| 			}
 | |
| 			blend.AlphaFormat = AC_SRC_ALPHA;
 | |
| 			blend.SourceConstantAlpha = 255;
 | |
| 			blend.BlendFlags = 0;
 | |
| 			blend.BlendOp = AC_SRC_OVER;
 | |
| 
 | |
| 			screenDC = GetDC(0);
 | |
| 			if (!screenDC) return false;
 | |
| 
 | |
| 			QRect avail(QDesktopWidget().availableGeometry());
 | |
| 			max_w = avail.width();
 | |
| 			if (max_w < st::wndMinWidth) max_w = st::wndMinWidth;
 | |
| 			max_h = avail.height();
 | |
| 			if (max_h < st::wndMinHeight) max_h = st::wndMinHeight;
 | |
| 
 | |
| 			HINSTANCE appinst = (HINSTANCE)GetModuleHandle(0);
 | |
| 		
 | |
| 			for (int i = 0; i < 4; ++i) {
 | |
| 				QString cn = QString("TelegramShadow%1").arg(i);
 | |
| 				LPCWSTR _cn = (LPCWSTR)cn.utf16();
 | |
| 				WNDCLASSEX wc;
 | |
| 
 | |
| 				wc.cbSize        = sizeof(wc);
 | |
| 				wc.style         = 0;
 | |
| 				wc.lpfnWndProc   = wndProc;
 | |
| 				wc.cbClsExtra    = 0;
 | |
| 				wc.cbWndExtra    = 0;
 | |
| 				wc.hInstance     = appinst;
 | |
| 				wc.hIcon         = 0;
 | |
| 				wc.hCursor       = 0;
 | |
| 				wc.hbrBackground = 0;
 | |
| 				wc.lpszMenuName  = NULL;
 | |
| 				wc.lpszClassName = _cn;
 | |
| 				wc.hIconSm       = 0;
 | |
| 				if (!RegisterClassEx(&wc)) {
 | |
| 					DEBUG_LOG(("Application Error: could not register shadow window class %1, error: %2").arg(i).arg(GetLastError()));
 | |
| 					destroy();
 | |
| 					return false;
 | |
| 				}
 | |
| 
 | |
| 				hwnds[i] = CreateWindowEx(WS_EX_LAYERED | WS_EX_TOOLWINDOW, _cn, 0, WS_POPUP, 0, 0, 0, 0, 0, 0, appinst, 0);
 | |
| 				if (!hwnds[i]) {
 | |
| 					DEBUG_LOG(("Application Error: could not create shadow window class %1, error: %2").arg(i).arg(GetLastError()));
 | |
| 					destroy();
 | |
| 					return false;
 | |
| 				}
 | |
| 
 | |
| 				dcs[i] = CreateCompatibleDC(screenDC);
 | |
| 				if (!dcs[i]) {
 | |
| 					DEBUG_LOG(("Application Error: could not create dc for shadow window class %1, error: %2").arg(i).arg(GetLastError()));
 | |
| 					destroy();
 | |
| 					return false;
 | |
| 				}
 | |
| 
 | |
| 				bitmaps[i] = CreateCompatibleBitmap(screenDC, (i % 2) ? _size : max_w, (i % 2) ? max_h : _size);
 | |
| 				if (!bitmaps[i]) {
 | |
| 					DEBUG_LOG(("Application Error: could not create bitmap for shadow window class %1, error: %2").arg(i).arg(GetLastError()));
 | |
| 					destroy();
 | |
| 					return false;
 | |
| 				}
 | |
| 
 | |
| 				SelectObject(dcs[i], bitmaps[i]);
 | |
| 			}
 | |
| 
 | |
| 			initCorners();
 | |
| 			return true;
 | |
| 		}
 | |
| 
 | |
| 		void initCorners(int directions = (_PsInitHor | _PsInitVer)) {
 | |
| 			bool hor = (directions & _PsInitHor), ver = (directions & _PsInitVer);
 | |
| 			Gdiplus::Graphics graphics0(dcs[0]), graphics1(dcs[1]), graphics2(dcs[2]), graphics3(dcs[3]);
 | |
| 			graphics0.SetCompositingMode(Gdiplus::CompositingModeSourceCopy);
 | |
| 			graphics1.SetCompositingMode(Gdiplus::CompositingModeSourceCopy);
 | |
| 			graphics2.SetCompositingMode(Gdiplus::CompositingModeSourceCopy);
 | |
| 			graphics3.SetCompositingMode(Gdiplus::CompositingModeSourceCopy);
 | |
| 
 | |
| 			Gdiplus::SolidBrush brush(Gdiplus::Color(_alphas[0], r, g, b));
 | |
| 			if (hor) graphics0.FillRectangle(&brush, 0, 0, _fullsize - (_size - _shift), 2 * _shift);
 | |
| 
 | |
| 			if (ver) {
 | |
| 				graphics1.FillRectangle(&brush, 0, 0, _size, 2 * _shift);
 | |
| 				graphics3.FillRectangle(&brush, 0, 0, _size, 2 * _shift);
 | |
| 				graphics1.FillRectangle(&brush, _size - _shift, 2 * _shift, _shift, _fullsize);
 | |
| 				graphics3.FillRectangle(&brush, 0, 2 * _shift, _shift, _fullsize);
 | |
| 			}
 | |
| 
 | |
| 			if (hor) {
 | |
| 				for (int j = 2 * _shift; j < _size; ++j) {
 | |
| 					for (int k = 0; k < _fullsize - (_size - _shift); ++k) {
 | |
| 						brush.SetColor(Gdiplus::Color(_colors[j * _metaSize + k + (_size + _shift)], r, g, b));
 | |
| 						graphics0.FillRectangle(&brush, k, j, 1, 1);
 | |
| 						graphics2.FillRectangle(&brush, k, _size - (j - 2 * _shift) - 1, 1, 1);
 | |
| 					}
 | |
| 				}
 | |
| 				for (int j = _size; j < _size + 2 * _shift; ++j) {
 | |
| 					for (int k = 0; k < _fullsize - (_size - _shift); ++k) {
 | |
| 						brush.SetColor(Gdiplus::Color(_colors[j * _metaSize + k + (_size + _shift)], r, g, b));
 | |
| 						graphics2.FillRectangle(&brush, k, _size - (j - 2 * _shift) - 1, 1, 1);
 | |
| 					}
 | |
| 				}
 | |
| 			}
 | |
| 			if (ver) {
 | |
| 				for (int j = 2 * _shift; j < _fullsize + 2 * _shift; ++j) {
 | |
| 					for (int k = _shift; k < _size; ++k) {
 | |
| 						brush.SetColor(Gdiplus::Color(_colors[j * _metaSize + (k + _shift)], r, g, b));
 | |
| 						graphics1.FillRectangle(&brush, _size - k - 1, j, 1, 1);
 | |
| 						graphics3.FillRectangle(&brush, k, j, 1, 1);
 | |
| 					}
 | |
| 				}
 | |
| 			}
 | |
| 		}
 | |
| 		void verCorners(int h, Gdiplus::Graphics *pgraphics1, Gdiplus::Graphics *pgraphics3) {
 | |
| 			Gdiplus::SolidBrush brush(Gdiplus::Color(_alphas[0], r, g, b));
 | |
| 			pgraphics1->FillRectangle(&brush, _size - _shift, h - _fullsize, _shift, _fullsize);
 | |
| 			pgraphics3->FillRectangle(&brush, 0, h - _fullsize, _shift, _fullsize);
 | |
| 			for (int j = 0; j < _fullsize; ++j) {
 | |
| 				for (int k = _shift; k < _size; ++k) {
 | |
| 					brush.SetColor(Gdiplus::Color(_colors[(j + 2 * _shift) * _metaSize + k + _shift], r, g, b));
 | |
| 					pgraphics1->FillRectangle(&brush, _size - k - 1, h - j - 1, 1, 1);
 | |
| 					pgraphics3->FillRectangle(&brush, k, h - j - 1, 1, 1);
 | |
| 				}
 | |
| 			}
 | |
| 		}
 | |
| 		void horCorners(int w, Gdiplus::Graphics *pgraphics0, Gdiplus::Graphics *pgraphics2) {
 | |
| 			Gdiplus::SolidBrush brush(Gdiplus::Color(_alphas[0], r, g, b));
 | |
| 			pgraphics0->FillRectangle(&brush, w - 2 * _size - (_fullsize - (_size - _shift)), 0, _fullsize - (_size - _shift), 2 * _shift);
 | |
| 			for (int j = 2 * _shift; j < _size; ++j) {
 | |
| 				for (int k = 0; k < _fullsize - (_size - _shift); ++k) {
 | |
| 					brush.SetColor(Gdiplus::Color(_colors[j * _metaSize + k + (_size + _shift)], r, g, b));
 | |
| 					pgraphics0->FillRectangle(&brush, w - 2 * _size - k - 1, j, 1, 1);
 | |
| 					pgraphics2->FillRectangle(&brush, w - 2 * _size - k - 1, _size - (j - 2 * _shift) - 1, 1, 1);
 | |
| 				}
 | |
| 			}
 | |
| 			for (int j = _size; j < _size + 2 * _shift; ++j) {
 | |
| 				for (int k = 0; k < _fullsize - (_size - _shift); ++k) {
 | |
| 					brush.SetColor(Gdiplus::Color(_colors[j * _metaSize + k + (_size + _shift)], r, g, b));
 | |
| 					pgraphics2->FillRectangle(&brush, w - 2 * _size - k - 1, _size - (j - 2 * _shift) - 1, 1, 1);
 | |
| 				}
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		void update(int changes, WINDOWPOS *pos = 0) {
 | |
| 			HWND hwnd = Application::wnd() ? Application::wnd()->psHwnd() : 0;
 | |
| 			if (!hwnd || !hwnds[0]) return;
 | |
| 
 | |
| 			if (changes == _PsShadowActivate) {
 | |
| 				for (int i = 0; i < 4; ++i) {
 | |
| 					SetWindowPos(hwnds[i], hwnd, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE | SWP_NOACTIVATE);
 | |
| 				}
 | |
| 				return;
 | |
| 			}
 | |
| 
 | |
| 			if (changes & _PsShadowHidden) {
 | |
| 				if (!hidden) {
 | |
| 					for (int i = 0; i < 4; ++i) {
 | |
| 						hidden = true;
 | |
| 						ShowWindow(hwnds[i], SW_HIDE);
 | |
| 					}
 | |
| 				}
 | |
| 				return;
 | |
| 			}
 | |
| 			if (!Application::wnd()->psPosInited()) return;
 | |
| 
 | |
| 			int x = _x, y = _y, w = _w, h = _h;
 | |
| 			if (pos && (!(pos->flags & SWP_NOMOVE) || !(pos->flags & SWP_NOSIZE) || !(pos->flags & SWP_NOREPOSITION))) {
 | |
| 				if (!(pos->flags & SWP_NOMOVE)) {
 | |
| 					x = pos->x - _size;
 | |
| 					y = pos->y - _size;
 | |
| 				} else if (pos->flags & SWP_NOSIZE) {
 | |
| 					for (int i = 0; i < 4; ++i) {
 | |
| 						SetWindowPos(hwnds[i], hwnd, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE | SWP_NOACTIVATE);
 | |
| 					}
 | |
| 					return;
 | |
| 				}
 | |
| 				if (!(pos->flags & SWP_NOSIZE)) {
 | |
| 					w = pos->cx + 2 * _size;
 | |
| 					h = pos->cy + 2 * _size;
 | |
| 				}
 | |
| 			} else {
 | |
| 				RECT r;
 | |
| 				GetWindowRect(hwnd, &r);
 | |
| 				x = r.left - _size;
 | |
| 				y = r.top - _size;
 | |
| 				w = r.right + _size - x;
 | |
| 				h = r.bottom + _size - y;
 | |
| 			}
 | |
| 			if (h < 2 * _fullsize + 2 * _shift) {
 | |
| 				h = 2 * _fullsize + 2 * _shift;
 | |
| 			}
 | |
| 			if (w < 2 * (_fullsize + _shift)) {
 | |
| 				w = 2 * (_fullsize + _shift);
 | |
| 			}
 | |
| 
 | |
| 			if (w != _w) {
 | |
| 				int from = (_w > 2 * (_fullsize + _shift)) ? (_w - _size - _fullsize - _shift) : (_fullsize - (_size - _shift));
 | |
| 				int to = w - _size - _fullsize - _shift;
 | |
| 				if (w > max_w) {
 | |
| 					from = _fullsize - (_size - _shift);
 | |
| 					max_w *= 2;
 | |
| 					for (int i = 0; i < 4; i += 2) {
 | |
| 						DeleteObject(bitmaps[i]); 
 | |
| 						bitmaps[i] = CreateCompatibleBitmap(screenDC, max_w, _size);
 | |
| 						SelectObject(dcs[i], bitmaps[i]);
 | |
| 					}
 | |
| 					initCorners(_PsInitHor);
 | |
| 				}
 | |
| 				Gdiplus::Graphics graphics0(dcs[0]), graphics2(dcs[2]);
 | |
| 				graphics0.SetCompositingMode(Gdiplus::CompositingModeSourceCopy);
 | |
| 				graphics2.SetCompositingMode(Gdiplus::CompositingModeSourceCopy);
 | |
| 				Gdiplus::SolidBrush brush(Gdiplus::Color(_alphas[0], r, g, b));
 | |
| 				if (to > from) {
 | |
| 					graphics0.FillRectangle(&brush, from, 0, to - from, 2 * _shift);
 | |
| 					for (int i = 2 * _shift; i < _size; ++i) {
 | |
| 						Gdiplus::Pen pen(Gdiplus::Color(_alphas[i], r, g, b));
 | |
| 						graphics0.DrawLine(&pen, from, i, to, i);
 | |
| 						graphics2.DrawLine(&pen, from, _size - (i - 2 * _shift) - 1, to, _size - (i - 2 * _shift) - 1);
 | |
| 					}
 | |
| 					for (int i = _size; i < _size + 2 * _shift; ++i) {
 | |
| 						Gdiplus::Pen pen(Gdiplus::Color(_alphas[i], r, g, b));
 | |
| 						graphics2.DrawLine(&pen, from, _size - (i - 2 * _shift) - 1, to, _size - (i - 2 * _shift) - 1);
 | |
| 					}
 | |
| 				}
 | |
| 				if (_w > w) {
 | |
| 					graphics0.FillRectangle(&brush, w - _size - _fullsize - _shift, 0, _fullsize - (_size - _shift), _size);
 | |
| 					graphics2.FillRectangle(&brush, w - _size - _fullsize - _shift, 0, _fullsize - (_size - _shift), _size);
 | |
| 				}
 | |
| 				horCorners(w, &graphics0, &graphics2);
 | |
| 				POINT p0 = { x + _size, y }, p2 = { x + _size, y + h - _size }, f = { 0, 0 };
 | |
| 				SIZE s = { w - 2 * _size, _size };
 | |
| 				updateWindow(0, &p0, &s);
 | |
| 				updateWindow(2, &p2, &s);
 | |
| 			} else if (x != _x || y != _y) {
 | |
| 				POINT p0 = { x + _size, y }, p2 = { x + _size, y + h - _size };
 | |
| 				updateWindow(0, &p0);
 | |
| 				updateWindow(2, &p2);
 | |
| 			} else if (h != _h) {
 | |
| 				POINT p2 = { x + _size, y + h - _size };
 | |
| 				updateWindow(2, &p2);
 | |
| 			}
 | |
| 
 | |
| 			if (h != _h) {
 | |
| 				int from = (_h > 2 * _fullsize + 2 * _shift) ? (_h - _fullsize) : (_fullsize + 2 * _shift);
 | |
| 				int to = h - _fullsize;
 | |
| 				if (h > max_h) {
 | |
| 					from = (_fullsize + 2 * _shift);
 | |
| 					max_h *= 2;
 | |
| 					for (int i = 1; i < 4; i += 2) {
 | |
| 						DeleteObject(bitmaps[i]);
 | |
| 						bitmaps[i] = CreateCompatibleBitmap(dcs[i], _size, max_h);
 | |
| 						SelectObject(dcs[i], bitmaps[i]);
 | |
| 					}
 | |
| 					initCorners(_PsInitVer);
 | |
| 				}
 | |
| 				Gdiplus::Graphics graphics1(dcs[1]), graphics3(dcs[3]);
 | |
| 				graphics1.SetCompositingMode(Gdiplus::CompositingModeSourceCopy);
 | |
| 				graphics3.SetCompositingMode(Gdiplus::CompositingModeSourceCopy);
 | |
| 
 | |
| 				Gdiplus::SolidBrush brush(Gdiplus::Color(_alphas[0], r, g, b));
 | |
| 				if (to > from) {
 | |
| 					graphics1.FillRectangle(&brush, _size - _shift, from, _shift, to - from);
 | |
| 					graphics3.FillRectangle(&brush, 0, from, _shift, to - from);
 | |
| 					for (int i = 2 * _shift; i < _size + _shift; ++i) {
 | |
| 						Gdiplus::Pen pen(Gdiplus::Color(_alphas[i], r, g, b));
 | |
| 						graphics1.DrawLine(&pen, _size + _shift - i - 1, from, _size + _shift - i - 1, to);
 | |
| 						graphics3.DrawLine(&pen, i - _shift, from, i - _shift, to);
 | |
| 					}
 | |
| 				}
 | |
| 				if (_h > h) {
 | |
| 					graphics1.FillRectangle(&brush, 0, h - _fullsize, _size, _fullsize);
 | |
| 					graphics3.FillRectangle(&brush, 0, h - _fullsize, _size, _fullsize);
 | |
| 				}
 | |
| 				verCorners(h, &graphics1, &graphics3);
 | |
| 
 | |
| 				POINT p1 = {x + w - _size, y}, p3 = {x, y}, f = {0, 0};
 | |
| 				SIZE s = { _size, h };
 | |
| 				updateWindow(1, &p1, &s);
 | |
| 				updateWindow(3, &p3, &s);
 | |
| 			} else if (x != _x || y != _y) {
 | |
| 				POINT p1 = { x + w - _size, y }, p3 = { x, y };
 | |
| 				updateWindow(1, &p1);
 | |
| 				updateWindow(3, &p3);
 | |
| 			} else if (w != _w) {
 | |
| 				POINT p1 = { x + w - _size, y };
 | |
| 				updateWindow(1, &p1);
 | |
| 			}
 | |
| 			_x = x;
 | |
| 			_y = y;
 | |
| 			_w = w;
 | |
| 			_h = h;
 | |
| 	
 | |
| 			if (hidden && (changes & _PsShadowShown)) {
 | |
| 				for (int i = 0; i < 4; ++i) {
 | |
| 					SetWindowPos(hwnds[i], hwnd, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE | SWP_SHOWWINDOW | SWP_NOACTIVATE);
 | |
| 				}
 | |
| 				hidden = false;
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		void updateWindow(int i, POINT *p, SIZE *s = 0) {
 | |
| 			static POINT f = {0, 0};
 | |
| 			if (s) {
 | |
| 				UpdateLayeredWindow(hwnds[i], (s ? screenDC : 0), p, s, (s ? dcs[i] : 0), (s ? (&f) : 0), noKeyColor, &blend, ULW_ALPHA);
 | |
| 			} else {
 | |
| 				SetWindowPos(hwnds[i], 0, p->x, p->y, 0, 0, SWP_NOACTIVATE | SWP_NOSIZE | SWP_NOZORDER);
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		void destroy() {
 | |
| 			for (int i = 0; i < 4; ++i) {
 | |
| 				if (dcs[i]) DeleteDC(dcs[i]);
 | |
| 				if (bitmaps[i]) DeleteObject(bitmaps[i]);
 | |
| 				if (hwnds[i]) DestroyWindow(hwnds[i]);
 | |
| 				dcs[i] = 0;
 | |
| 				bitmaps[i] = 0;
 | |
| 				hwnds[i] = 0;
 | |
| 			}
 | |
| 			if (screenDC) ReleaseDC(0, screenDC);
 | |
| 		}
 | |
| 
 | |
| 	private:
 | |
| 
 | |
| 		int _x, _y, _w, _h;
 | |
| 		int _metaSize, _fullsize, _size, _shift;
 | |
| 		QVector<BYTE> _alphas, _colors;
 | |
| 
 | |
| 		bool hidden;
 | |
| 
 | |
| 		HWND hwnds[4];
 | |
| 		HDC dcs[4], screenDC;
 | |
| 		HBITMAP bitmaps[4];
 | |
| 		int max_w, max_h;
 | |
| 		BLENDFUNCTION blend;
 | |
| 
 | |
| 		BYTE r, g, b;
 | |
| 		COLORREF noKeyColor;
 | |
| 
 | |
| 		static LRESULT CALLBACK _PsShadowWindows::wndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam);
 | |
| 
 | |
| 	};
 | |
| 	_PsShadowWindows _psShadowWindows;
 | |
| 
 | |
| 	LRESULT CALLBACK _PsShadowWindows::wndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {
 | |
| 		if (finished) return DefWindowProc(hwnd, msg, wParam, lParam);
 | |
| 
 | |
| 		int i;
 | |
| 		for (i = 0; i < 4; ++i) {
 | |
| 			if (_psShadowWindows.hwnds[i] && hwnd == _psShadowWindows.hwnds[i]) {
 | |
| 				break;
 | |
| 			}
 | |
| 		}
 | |
| 		if (i == 4) return DefWindowProc(hwnd, msg, wParam, lParam);
 | |
| 
 | |
| 		switch (msg) {
 | |
| 			case WM_CLOSE:
 | |
| 				Application::wnd()->close();
 | |
| 			break;
 | |
| 			case WM_NCHITTEST: {
 | |
| 				int32 xPos = GET_X_LPARAM(lParam), yPos = GET_Y_LPARAM(lParam);
 | |
| 				switch (i) {
 | |
| 				case 0: return HTTOP;
 | |
| 				case 1: return (yPos < _psShadowWindows._y + _psSize) ? HTTOPRIGHT : ((yPos >= _psShadowWindows._y + _psShadowWindows._h - _psSize) ? HTBOTTOMRIGHT : HTRIGHT);
 | |
| 				case 2: return HTBOTTOM;
 | |
| 				case 3: return (yPos < _psShadowWindows._y + _psSize) ? HTTOPLEFT : ((yPos >= _psShadowWindows._y + _psShadowWindows._h - _psSize) ? HTBOTTOMLEFT : HTLEFT);
 | |
| 				}
 | |
| 				return HTTRANSPARENT;
 | |
| 			} break;
 | |
| 
 | |
| 			case WM_NCACTIVATE: return DefWindowProc(hwnd, msg, wParam, lParam);
 | |
| 			case WM_NCLBUTTONDOWN:
 | |
| 			case WM_NCLBUTTONUP:
 | |
| 			case WM_NCLBUTTONDBLCLK:
 | |
| 			case WM_NCMBUTTONDOWN:
 | |
| 			case WM_NCMBUTTONUP:
 | |
| 			case WM_NCMBUTTONDBLCLK:
 | |
| 			case WM_NCRBUTTONDOWN:
 | |
| 			case WM_NCRBUTTONUP:
 | |
| 			case WM_NCRBUTTONDBLCLK:
 | |
| 			case WM_NCXBUTTONDOWN:
 | |
| 			case WM_NCXBUTTONUP:
 | |
| 			case WM_NCXBUTTONDBLCLK:
 | |
| 			case WM_NCMOUSEHOVER:
 | |
| 			case WM_NCMOUSELEAVE:
 | |
| 			case WM_NCMOUSEMOVE:
 | |
| 			case WM_NCPOINTERUPDATE:
 | |
| 			case WM_NCPOINTERDOWN:
 | |
| 			case WM_NCPOINTERUP:
 | |
| 				if (App::wnd() && App::wnd()->psHwnd()) {
 | |
| 					if (msg == WM_NCLBUTTONDOWN) {
 | |
| 						::SetForegroundWindow(App::wnd()->psHwnd());
 | |
| 					}
 | |
| 					LRESULT res = SendMessage(App::wnd()->psHwnd(), msg, wParam, lParam);
 | |
| 					return res;
 | |
| 				}
 | |
| 				return 0;
 | |
| 			break;
 | |
| 			case WM_ACTIVATE:
 | |
| 				if (App::wnd() && App::wnd()->psHwnd() && wParam == WA_ACTIVE) {
 | |
| 					if ((HWND)lParam != App::wnd()->psHwnd()) {
 | |
| 						::SetForegroundWindow(hwnd);
 | |
| 						::SetWindowPos(App::wnd()->psHwnd(), hwnd, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE);
 | |
| 					}
 | |
| 				}
 | |
| 				return DefWindowProc(hwnd, msg, wParam, lParam);
 | |
| 			break;
 | |
| 			default:
 | |
| 				return DefWindowProc(hwnd, msg, wParam, lParam);
 | |
| 		}
 | |
| 		return 0;
 | |
| 	}
 | |
| 
 | |
| 	QColor _shActive(0, 0, 0), _shInactive(0, 0, 0);
 | |
| 
 | |
| 	typedef BOOL (FAR STDAPICALLTYPE *f_dwmDefWindowProc)(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam, _Out_ LRESULT *plResult);
 | |
| 	f_dwmDefWindowProc dwmDefWindowProc;
 | |
| 
 | |
| 	typedef HRESULT (FAR STDAPICALLTYPE *f_dwmSetWindowAttribute)(HWND hWnd, DWORD dwAttribute, _In_ LPCVOID pvAttribute, DWORD cbAttribute);
 | |
| 	f_dwmSetWindowAttribute dwmSetWindowAttribute;
 | |
| 	
 | |
| 	typedef HRESULT (FAR STDAPICALLTYPE *f_dwmExtendFrameIntoClientArea)(HWND hWnd, const MARGINS *pMarInset);
 | |
| 	f_dwmExtendFrameIntoClientArea dwmExtendFrameIntoClientArea;
 | |
| 
 | |
| 	typedef HRESULT (FAR STDAPICALLTYPE *f_setWindowTheme)(HWND hWnd, LPCWSTR pszSubAppName, LPCWSTR pszSubIdList);
 | |
| 	f_setWindowTheme setWindowTheme;
 | |
| 
 | |
| 	typedef HRESULT (FAR STDAPICALLTYPE *f_openAs_RunDLL)(HWND hWnd, HINSTANCE hInstance, LPCWSTR lpszCmdLine, int nCmdShow);
 | |
| 	f_openAs_RunDLL openAs_RunDLL;
 | |
| 
 | |
| 	typedef HRESULT (FAR STDAPICALLTYPE *f_shOpenWithDialog)(HWND hwndParent, const OPENASINFO *poainfo);
 | |
| 	f_shOpenWithDialog shOpenWithDialog;
 | |
| 
 | |
| 	template <typename TFunction>
 | |
| 	bool loadFunction(HINSTANCE dll, LPCSTR name, TFunction &func) {
 | |
| 		if (!dll) return false;
 | |
| 
 | |
| 		func = (TFunction)GetProcAddress(dll, name);
 | |
| 		return !!func;
 | |
| 	}
 | |
| 
 | |
| 	class _PsInitializer {
 | |
| 	public:
 | |
| 		_PsInitializer() {
 | |
| 			setupDWM();
 | |
| 			useDWM = true;
 | |
| 			frameless = !useDWM;
 | |
| 
 | |
| 			setupUx();
 | |
| 			setupOpenAs();
 | |
| 		}
 | |
| 		void setupDWM() {
 | |
| 			HINSTANCE procId = LoadLibrary(L"DWMAPI.DLL");
 | |
| 
 | |
| 			if (!loadFunction(procId, "DwmDefWindowProc", dwmDefWindowProc)) return;
 | |
| 			if (!loadFunction(procId, "DwmSetWindowAttribute", dwmSetWindowAttribute)) return;
 | |
| 			if (!loadFunction(procId, "DwmExtendFrameIntoClientArea", dwmExtendFrameIntoClientArea)) return;
 | |
| 			useDWM = true;
 | |
| 		}
 | |
| 		void setupUx() {
 | |
| 			HINSTANCE procId = LoadLibrary(L"UXTHEME.DLL");
 | |
| 
 | |
| 			if (!loadFunction(procId, "SetWindowTheme", setWindowTheme)) return;
 | |
| 			useTheme = true;
 | |
| 		}
 | |
| 		void setupOpenAs() {
 | |
| 			HINSTANCE procId = LoadLibrary(L"SHELL32.DLL");
 | |
| 
 | |
| 			if (!loadFunction(procId, "SHOpenWithDialog", shOpenWithDialog) && !loadFunction(procId, "OpenAs_RunDLLW", openAs_RunDLL)) return;
 | |
| 			useOpenAs = true;
 | |
| 		}
 | |
| 	};
 | |
| 	_PsInitializer _psInitializer;
 | |
| 
 | |
| 	class _PsEventFilter : public QAbstractNativeEventFilter {
 | |
| 	public:
 | |
| 		_PsEventFilter() {
 | |
| 		}
 | |
| 
 | |
| 		bool nativeEventFilter(const QByteArray &eventType, void *message, long *result) {
 | |
| 			Window *wnd = Application::wnd();
 | |
| 			if (!wnd) return false;
 | |
| 
 | |
| 			MSG *msg = (MSG*)message;
 | |
| 			if (msg->message == WM_ENDSESSION) {
 | |
| 				App::quit();
 | |
| 				return false;
 | |
| 			}
 | |
| 			if (msg->hwnd == wnd->psHwnd() || msg->hwnd && !wnd->psHwnd()) {
 | |
| 				return mainWindowEvent(msg->hwnd, msg->message, msg->wParam, msg->lParam, (LRESULT*)result);
 | |
| 			}
 | |
| 			return false;
 | |
| 		}
 | |
| 
 | |
| 		bool mainWindowEvent(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam, LRESULT *result) {
 | |
| 			if (tbCreatedMsgId && msg == tbCreatedMsgId) {
 | |
| 				if (CoCreateInstance(CLSID_TaskbarList, NULL, CLSCTX_ALL, IID_ITaskbarList3, (void**)&tbListInterface) != S_OK) {
 | |
| 					tbListInterface = 0;
 | |
| 				}
 | |
| 			}
 | |
| 			switch (msg) {
 | |
| 
 | |
| 			case WM_DESTROY: {
 | |
| 				App::quit();
 | |
| 			} return false;
 | |
| 
 | |
| 			case WM_ACTIVATE: {
 | |
| 				if (LOWORD(wParam) == WA_CLICKACTIVE) {
 | |
| 					App::wnd()->inactivePress(true);
 | |
| 				}
 | |
| 				Application::wnd()->psUpdateMargins();
 | |
| 				if (LOWORD(wParam) != WA_INACTIVE) {
 | |
| 					_psShadowWindows.setColor(_shActive);
 | |
| 					_psShadowWindows.update(_PsShadowActivate);
 | |
| 				} else {
 | |
| 					_psShadowWindows.setColor(_shInactive);
 | |
| 				}
 | |
| 				QTimer::singleShot(0, Application::wnd(), SLOT(updateCounter()));
 | |
| 				Application::wnd()->update();
 | |
| 			} return false;
 | |
| 				
 | |
| 			case WM_NCPAINT: if (QSysInfo::WindowsVersion >= QSysInfo::WV_WINDOWS8) return false; *result = 0; return true;
 | |
| 
 | |
| 			case WM_NCCALCSIZE: if (!useDWM) return false; {
 | |
| 				if (wParam == TRUE) {
 | |
| 					LPNCCALCSIZE_PARAMS params = (LPNCCALCSIZE_PARAMS)lParam;
 | |
| 					params->rgrc[0].left += margins.left() - simpleMargins.left();
 | |
| 					params->rgrc[0].top += margins.top() - simpleMargins.top();
 | |
| 					params->rgrc[0].right -= margins.right() - simpleMargins.right();
 | |
| 					params->rgrc[0].bottom -= margins.bottom() - simpleMargins.bottom();
 | |
| 				} else if (wParam == FALSE) {
 | |
| 					LPRECT rect = (LPRECT)lParam;
 | |
| 
 | |
| 					rect->left += margins.left() - simpleMargins.left();
 | |
| 					rect->top += margins.top() - simpleMargins.top();
 | |
| 					rect->right += margins.right() - simpleMargins.right();
 | |
| 					rect->bottom += margins.bottom() - simpleMargins.bottom();
 | |
| 				}
 | |
| 				*result = 0;
 | |
| 			} return true;
 | |
| 
 | |
| 			case WM_NCACTIVATE: {
 | |
| 				Application::wnd()->psUpdateMargins();
 | |
| 				*result = LRESULT(TRUE);
 | |
| 				Application::wnd()->repaint();
 | |
| 			} return true;
 | |
| 
 | |
| 			case WM_WINDOWPOSCHANGING:
 | |
| 			case WM_WINDOWPOSCHANGED: {
 | |
| 				_psShadowWindows.update(_PsShadowMoved | _PsShadowResized, (WINDOWPOS*)lParam);
 | |
| 			} return false;
 | |
| 
 | |
| 			case WM_SIZE: {
 | |
| 				if (App::wnd()) {
 | |
| 					if (wParam == SIZE_MAXIMIZED || wParam == SIZE_RESTORED || wParam == SIZE_MINIMIZED) {
 | |
| 						if (wParam != SIZE_RESTORED || App::wnd()->windowState() != Qt::WindowNoState) {
 | |
| 							Qt::WindowState state = Qt::WindowNoState;
 | |
| 							if (wParam == SIZE_MAXIMIZED) {
 | |
| 								state = Qt::WindowMaximized;
 | |
| 							} else if (wParam == SIZE_MINIMIZED) {
 | |
| 								state = Qt::WindowMinimized;
 | |
| 							}
 | |
| 							emit App::wnd()->windowHandle()->windowStateChanged(state);
 | |
| 						} else {
 | |
| 							App::wnd()->psUpdatedPosition();
 | |
| 						}
 | |
| 						int changes = (wParam == SIZE_MINIMIZED || wParam == SIZE_MAXIMIZED) ? _PsShadowHidden : (_PsShadowResized | _PsShadowShown);
 | |
| 						_psShadowWindows.update(changes);
 | |
| 					}
 | |
| 				}
 | |
| 			} return false;
 | |
| 
 | |
| 			case WM_SHOWWINDOW: {
 | |
| 				LONG style = GetWindowLong(hWnd, GWL_STYLE);
 | |
| 				int changes = _PsShadowResized | ((wParam && !(style & (WS_MAXIMIZE | WS_MINIMIZE))) ? _PsShadowShown : _PsShadowHidden);
 | |
| 				_psShadowWindows.update(changes);
 | |
| 			} return false;
 | |
| 
 | |
| 			case WM_MOVE: {
 | |
| 				_psShadowWindows.update(_PsShadowMoved);
 | |
| 				App::wnd()->psUpdatedPosition();
 | |
| 			} return false;
 | |
| 
 | |
| 			case WM_NCHITTEST: {
 | |
| 				POINTS p = MAKEPOINTS(lParam);
 | |
| 				RECT r;
 | |
| 				GetWindowRect(hWnd, &r);
 | |
| 				HitTestType res = Application::wnd()->hitTest(QPoint(p.x - r.left + dleft, p.y - r.top + dtop));
 | |
| 				switch (res) {
 | |
| 					case HitTestClient:
 | |
| 					case HitTestSysButton:   *result = HTCLIENT; break;
 | |
| 					case HitTestIcon:        *result = HTCAPTION; break;
 | |
| 					case HitTestCaption:     *result = HTCAPTION; break;
 | |
| 					case HitTestTop:         *result = HTTOP; break;
 | |
| 					case HitTestTopRight:    *result = HTTOPRIGHT; break;
 | |
| 					case HitTestRight:       *result = HTRIGHT; break;
 | |
| 					case HitTestBottomRight: *result = HTBOTTOMRIGHT; break;
 | |
| 					case HitTestBottom:      *result = HTBOTTOM; break;
 | |
| 					case HitTestBottomLeft:  *result = HTBOTTOMLEFT; break;
 | |
| 					case HitTestLeft:        *result = HTLEFT; break;
 | |
| 					case HitTestTopLeft:     *result = HTTOPLEFT; break;
 | |
| 					case HitTestNone: 
 | |
| 					default:                 *result = HTTRANSPARENT; break;
 | |
| 				};
 | |
| 			} return true;
 | |
| 
 | |
| 			case WM_NCRBUTTONUP: {
 | |
| 				SendMessage(hWnd, WM_SYSCOMMAND, SC_MOUSEMENU, lParam);
 | |
| 			} return true;
 | |
| 
 | |
| 			case WM_NCLBUTTONDOWN: {
 | |
| 				POINTS p = MAKEPOINTS(lParam);
 | |
| 				RECT r;
 | |
| 				GetWindowRect(hWnd, &r);
 | |
| 				HitTestType res = Application::wnd()->hitTest(QPoint(p.x - r.left + dleft, p.y - r.top + dtop));
 | |
| 				switch (res) {
 | |
| 					case HitTestIcon: 
 | |
| 						if (menuHidden && getms() < menuHidden + 10) {
 | |
| 							menuHidden = 0;
 | |
| 							if (getms() < menuShown + GetDoubleClickTime()) {
 | |
| 								Application::wnd()->close();
 | |
| 							}
 | |
| 						} else {
 | |
| 							QRect icon = Application::wnd()->iconRect();
 | |
| 							p.x = r.left - dleft + icon.left();
 | |
| 							p.y = r.top - dtop + icon.top() + icon.height();
 | |
| 							Application::wnd()->psUpdateSysMenu(Application::wnd()->windowHandle()->windowState());
 | |
| 							menuShown = getms();
 | |
| 							menuHidden = 0;
 | |
| 							TrackPopupMenu(Application::wnd()->psMenu(), TPM_LEFTALIGN | TPM_TOPALIGN | TPM_LEFTBUTTON, p.x, p.y, 0, hWnd, 0);
 | |
| 							menuHidden = getms();
 | |
| 						}
 | |
| 					return true;
 | |
| 				};
 | |
| 			} return false;
 | |
| 
 | |
| 			case WM_NCLBUTTONDBLCLK: {
 | |
| 				POINTS p = MAKEPOINTS(lParam);
 | |
| 				RECT r;
 | |
| 				GetWindowRect(hWnd, &r);
 | |
| 				HitTestType res = Application::wnd()->hitTest(QPoint(p.x - r.left + dleft, p.y - r.top + dtop));
 | |
| 				switch (res) {
 | |
| 					case HitTestIcon: Application::wnd()->close(); return true;
 | |
| 				};
 | |
| 			} return false;
 | |
| 
 | |
| 			case WM_SYSCOMMAND: {
 | |
| 				if (wParam == SC_MOUSEMENU) {
 | |
| 					POINTS p = MAKEPOINTS(lParam);
 | |
| 					Application::wnd()->psUpdateSysMenu(Application::wnd()->windowHandle()->windowState());
 | |
| 					TrackPopupMenu(Application::wnd()->psMenu(), TPM_LEFTALIGN | TPM_TOPALIGN | TPM_LEFTBUTTON, p.x, p.y, 0, hWnd, 0);
 | |
| 				}
 | |
| 			} return false;
 | |
| 
 | |
| 			case WM_COMMAND: {
 | |
| 				if (HIWORD(wParam)) return false;
 | |
| 				int cmd = LOWORD(wParam);
 | |
| 				switch (cmd) {
 | |
| 					case SC_CLOSE: Application::wnd()->close(); return true;
 | |
| 					case SC_MINIMIZE: Application::wnd()->setWindowState(Qt::WindowMinimized); return true;
 | |
| 					case SC_MAXIMIZE: Application::wnd()->setWindowState(Qt::WindowMaximized); return true;
 | |
| 					case SC_RESTORE: Application::wnd()->setWindowState(Qt::WindowNoState); return true;
 | |
| 				}
 | |
| 			} return true;
 | |
| 
 | |
| 			}
 | |
| 			return false;
 | |
| 		}
 | |
| 	};
 | |
| 	_PsEventFilter *_psEventFilter = 0;
 | |
| 
 | |
| };
 | |
| 
 | |
| PsMainWindow::PsMainWindow(QWidget *parent) : QMainWindow(parent), ps_hWnd(0), ps_menu(0), icon256(qsl(":/gui/art/icon256.png")), iconbig256(qsl(":/gui/art/iconbig256.png")), wndIcon(QPixmap::fromImage(icon256, Qt::ColorOnly)),
 | |
| 	ps_iconBig(0), ps_iconSmall(0), ps_iconOverlay(0), trayIcon(0), trayIconMenu(0), posInited(false), ps_tbHider_hWnd(createTaskbarHider()), psIdle(false) {
 | |
| 	tbCreatedMsgId = RegisterWindowMessage(L"TaskbarButtonCreated");
 | |
| 	connect(&psIdleTimer, SIGNAL(timeout()), this, SLOT(psIdleTimeout()));
 | |
| 	psIdleTimer.setSingleShot(false);
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psNotIdle() const {
 | |
| 	psIdleTimer.stop();
 | |
| 	if (psIdle) {
 | |
| 		psIdle = false;
 | |
| 		if (App::main()) App::main()->setOnline();
 | |
| 		if (App::wnd()) App::wnd()->checkHistoryActivation();
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psIdleTimeout() {
 | |
| 	LASTINPUTINFO lii;
 | |
| 	lii.cbSize = sizeof(LASTINPUTINFO);
 | |
| 	BOOL res = GetLastInputInfo(&lii);
 | |
| 	if (res) {
 | |
| 		uint64 ticks = GetTickCount();
 | |
| 		if (lii.dwTime >= ticks - IdleMsecs) {
 | |
| 			psNotIdle();
 | |
| 		}
 | |
| 	} else { // error {
 | |
| 		psNotIdle();
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psShowTrayMenu() {
 | |
| 	trayIconMenu->popup(QCursor::pos());
 | |
| }
 | |
| 
 | |
| bool PsMainWindow::psIsActive(int state) const {
 | |
| 	if (state < 0) state = this->windowState();
 | |
| 	return isActiveWindow() && isVisible() && !(state & Qt::WindowMinimized) && !psIdle;
 | |
| }
 | |
| 
 | |
| bool PsMainWindow::psIsOnline(int windowState) const {
 | |
| 	if (windowState < 0) windowState = this->windowState();
 | |
| 	if (windowState & Qt::WindowMinimized) {
 | |
| 		return false;
 | |
| 	} else if (!isVisible()) {
 | |
| 		return false;
 | |
| 	}
 | |
| 	LASTINPUTINFO lii;
 | |
| 	lii.cbSize = sizeof(LASTINPUTINFO);
 | |
| 	BOOL res = GetLastInputInfo(&lii);
 | |
| 	if (res) {
 | |
| 		uint64 ticks = GetTickCount();
 | |
| 		if (lii.dwTime < ticks - IdleMsecs) {
 | |
| 			if (!psIdle) {
 | |
| 				psIdle = true;
 | |
| 				psIdleTimer.start(900);
 | |
| 			}
 | |
| 			return false;
 | |
| 		} else {
 | |
| 			psNotIdle();
 | |
| 		}
 | |
| 	} else { // error
 | |
| 		psNotIdle();
 | |
| 	}
 | |
| 	return true;
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psRefreshTaskbarIcon() {
 | |
| 	QWidget *w = new QWidget(this);
 | |
| 	w->setWindowFlags(Qt::Tool | Qt::FramelessWindowHint);
 | |
| 	w->setGeometry(x() + 1, y() + 1, 1, 1);
 | |
| 	QPalette p(w->palette());
 | |
| 	p.setColor(QPalette::Background, st::titleBG->c);
 | |
| 	QWindow *wnd = w->windowHandle();
 | |
| 	w->setPalette(p);
 | |
| 	w->show();
 | |
| 	w->activateWindow();
 | |
| 	delete w;
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psUpdateWorkmode() {
 | |
| 	switch (cWorkMode()) {
 | |
| 	case dbiwmWindowAndTray: {
 | |
| 		setupTrayIcon();
 | |
| 		HWND psOwner = (HWND)GetWindowLong(ps_hWnd, GWL_HWNDPARENT);
 | |
| 		if (psOwner) {
 | |
| 			SetWindowLong(ps_hWnd, GWL_HWNDPARENT, 0);
 | |
| 			psRefreshTaskbarIcon();
 | |
| 		}
 | |
| 	} break;
 | |
| 
 | |
| 	case dbiwmTrayOnly: {
 | |
| 		setupTrayIcon();
 | |
| 		HWND psOwner = (HWND)GetWindowLong(ps_hWnd, GWL_HWNDPARENT);
 | |
| 		if (!psOwner) {
 | |
| 			SetWindowLong(ps_hWnd, GWL_HWNDPARENT, (LONG)ps_tbHider_hWnd);
 | |
| 		}
 | |
| 	} break;
 | |
| 
 | |
| 	case dbiwmWindowOnly: {
 | |
| 		if (trayIcon) {
 | |
| 			trayIcon->setContextMenu(0);
 | |
| 			trayIcon->deleteLater();
 | |
| 		}
 | |
| 		trayIcon = 0;
 | |
| 
 | |
| 		HWND psOwner = (HWND)GetWindowLong(ps_hWnd, GWL_HWNDPARENT);
 | |
| 		if (psOwner) {
 | |
| 			SetWindowLong(ps_hWnd, GWL_HWNDPARENT, 0);
 | |
| 			psRefreshTaskbarIcon();
 | |
| 		}
 | |
| 	} break;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| HICON qt_pixmapToWinHICON(const QPixmap &);
 | |
| static HICON _qt_createHIcon(const QIcon &icon, int xSize, int ySize) {
 | |
|     if (!icon.isNull()) {
 | |
|         const QPixmap pm = icon.pixmap(icon.actualSize(QSize(xSize, ySize)));
 | |
|         if (!pm.isNull())
 | |
|             return qt_pixmapToWinHICON(pm);
 | |
|     }
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psUpdateCounter() {
 | |
| 	int32 counter = App::histories().unreadFull;
 | |
| 	style::color bg = (App::histories().unreadMuted < counter) ? st::counterBG : st::counterMuteBG;
 | |
| 	QIcon iconSmall, iconBig;
 | |
| 	iconSmall.addPixmap(QPixmap::fromImage(iconWithCounter(16, counter, bg, true), Qt::ColorOnly));
 | |
| 	iconSmall.addPixmap(QPixmap::fromImage(iconWithCounter(32, counter, bg, true), Qt::ColorOnly));
 | |
| 	iconBig.addPixmap(QPixmap::fromImage(iconWithCounter(32, tbListInterface ? 0 : counter, bg, false), Qt::ColorOnly));
 | |
| 	iconBig.addPixmap(QPixmap::fromImage(iconWithCounter(64, tbListInterface ? 0 : counter, bg, false), Qt::ColorOnly));
 | |
| 	if (trayIcon) {
 | |
| 		trayIcon->setIcon(iconSmall);
 | |
| 	}
 | |
| 
 | |
| 	setWindowTitle((counter > 0) ? qsl("Telegram (%1)").arg(counter) : qsl("Telegram"));
 | |
| 	psDestroyIcons();
 | |
|     ps_iconSmall = _qt_createHIcon(iconSmall, GetSystemMetrics(SM_CXSMICON), GetSystemMetrics(SM_CYSMICON));
 | |
|     ps_iconBig = _qt_createHIcon(iconBig, GetSystemMetrics(SM_CXICON), GetSystemMetrics(SM_CYICON));
 | |
|     SendMessage(ps_hWnd, WM_SETICON, 0, (LPARAM)ps_iconSmall);
 | |
| 	SendMessage(ps_hWnd, WM_SETICON, 1, (LPARAM)(ps_iconBig ? ps_iconBig : ps_iconSmall));
 | |
| 	if (tbListInterface) {
 | |
| 		if (counter > 0) {
 | |
| 			QIcon iconOverlay;
 | |
| 			iconOverlay.addPixmap(QPixmap::fromImage(iconWithCounter(-16, counter, bg, false), Qt::ColorOnly));
 | |
| 			iconOverlay.addPixmap(QPixmap::fromImage(iconWithCounter(-32, counter, bg, false), Qt::ColorOnly));
 | |
| 			ps_iconOverlay = _qt_createHIcon(iconOverlay, GetSystemMetrics(SM_CXSMICON), GetSystemMetrics(SM_CYSMICON));
 | |
| 		}
 | |
| 		QString description = counter > 0 ? QString("%1 unread messages").arg(counter) : qsl("No unread messages");
 | |
| 		static WCHAR descriptionArr[1024];
 | |
| 		description.toWCharArray(descriptionArr);
 | |
| 		tbListInterface->SetOverlayIcon(ps_hWnd, ps_iconOverlay, descriptionArr);
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psUpdateDelegate() {
 | |
| }
 | |
| 
 | |
| namespace {
 | |
| 	HMONITOR enumMonitor = 0;
 | |
| 	RECT enumMonitorWork;
 | |
| 
 | |
| 	BOOL CALLBACK _monitorEnumProc(
 | |
| 	  _In_  HMONITOR hMonitor,
 | |
| 	  _In_  HDC hdcMonitor,
 | |
| 	  _In_  LPRECT lprcMonitor,
 | |
| 	  _In_  LPARAM dwData
 | |
| 	) {
 | |
| 		MONITORINFOEX info;
 | |
| 		info.cbSize = sizeof(info);
 | |
| 		GetMonitorInfo(hMonitor, &info);
 | |
| 		if (dwData == hashCrc32(info.szDevice, sizeof(info.szDevice))) {
 | |
| 			enumMonitor = hMonitor;
 | |
| 			enumMonitorWork = info.rcWork;
 | |
| 			return FALSE;
 | |
| 		}
 | |
| 		return TRUE;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psInitSize() {
 | |
| 	setMinimumWidth(st::wndMinWidth);
 | |
| 	setMinimumHeight(st::wndMinHeight);
 | |
| 
 | |
| 	TWindowPos pos(cWindowPos());
 | |
| 	if (cDebug()) { // temp while design
 | |
| 		pos.w = st::wndDefWidth;
 | |
| 		pos.h = st::wndDefHeight;
 | |
| 	}
 | |
| 	QRect avail(QDesktopWidget().availableGeometry());
 | |
| 	bool maximized = false;
 | |
| 	QRect geom(avail.x() + (avail.width() - st::wndDefWidth) / 2, avail.y() + (avail.height() - st::wndDefHeight) / 2, st::wndDefWidth, st::wndDefHeight);
 | |
| 	if (pos.w && pos.h) {
 | |
| 		if (pos.y < 0) pos.y = 0;
 | |
| 		enumMonitor = 0;
 | |
| 		EnumDisplayMonitors(0, 0, &_monitorEnumProc, pos.moncrc);
 | |
| 		if (enumMonitor) {
 | |
| 			int32 w = enumMonitorWork.right - enumMonitorWork.left, h = enumMonitorWork.bottom - enumMonitorWork.top;
 | |
| 			if (w >= st::wndMinWidth && h >= st::wndMinHeight) {
 | |
| 				if (pos.w > w) pos.w = w;
 | |
| 				if (pos.h > h) pos.h = h;
 | |
| 				pos.x += enumMonitorWork.left;
 | |
| 				pos.y += enumMonitorWork.top;
 | |
| 				if (pos.x < enumMonitorWork.right - 10 && pos.y < enumMonitorWork.bottom - 10) {
 | |
| 					geom = QRect(pos.x, pos.y, pos.w, pos.h);
 | |
| 				}
 | |
| 			}
 | |
| 		}
 | |
| 		maximized = pos.maximized;
 | |
| 	}
 | |
| 	setGeometry(geom);
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psInitFrameless() {
 | |
| 	psUpdatedPositionTimer.setSingleShot(true);
 | |
| 	connect(&psUpdatedPositionTimer, SIGNAL(timeout()), this, SLOT(psSavePosition()));
 | |
| 
 | |
| 	QPlatformNativeInterface *i = QGuiApplication::platformNativeInterface();
 | |
|     ps_hWnd = static_cast<HWND>(i->nativeResourceForWindow(QByteArrayLiteral("handle"), windowHandle()));
 | |
| 
 | |
| 	if (!ps_hWnd) return;
 | |
| 
 | |
| 	if (frameless) {
 | |
| 		setWindowFlags(Qt::FramelessWindowHint);
 | |
| 	}
 | |
| 
 | |
| //	RegisterApplicationRestart(NULL, 0);
 | |
| 
 | |
| 	psInitSysMenu();
 | |
| 	connect(windowHandle(), SIGNAL(windowStateChanged(Qt::WindowState)), this, SLOT(psStateChanged(Qt::WindowState)));
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psSavePosition(Qt::WindowState state) {
 | |
| 	if (state == Qt::WindowActive) state = windowHandle()->windowState();
 | |
| 	if (state == Qt::WindowMinimized || !posInited) return;
 | |
| 
 | |
| 	TWindowPos pos(cWindowPos()), curPos = pos;
 | |
| 
 | |
| 	if (state == Qt::WindowMaximized) {
 | |
| 		curPos.maximized = 1;
 | |
| 	} else {
 | |
| 		RECT w;
 | |
| 		GetWindowRect(ps_hWnd, &w);
 | |
| 		curPos.x = w.left;
 | |
| 		curPos.y = w.top;
 | |
| 		curPos.w = w.right - w.left;
 | |
| 		curPos.h = w.bottom - w.top;
 | |
| 		curPos.maximized = 0;
 | |
| 	}
 | |
| 
 | |
| 	HMONITOR hMonitor = MonitorFromWindow(ps_hWnd, MONITOR_DEFAULTTONEAREST);
 | |
| 	if (hMonitor) {
 | |
| 		MONITORINFOEX info;
 | |
| 		info.cbSize = sizeof(info);
 | |
| 		GetMonitorInfo(hMonitor, &info);
 | |
| 		if (!curPos.maximized) {
 | |
| 			curPos.x -= info.rcWork.left;
 | |
| 			curPos.y -= info.rcWork.top;
 | |
| 		}
 | |
| 		curPos.moncrc = hashCrc32(info.szDevice, sizeof(info.szDevice));
 | |
| 	}
 | |
| 
 | |
| 	if (curPos.w >= st::wndMinWidth && curPos.h >= st::wndMinHeight) {
 | |
| 		if (curPos.x != pos.x || curPos.y != pos.y || curPos.w != pos.w || curPos.h != pos.h || curPos.moncrc != pos.moncrc || curPos.maximized != pos.maximized) {
 | |
| 			cSetWindowPos(curPos);
 | |
| 			App::writeConfig();
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psUpdatedPosition() {
 | |
| 	psUpdatedPositionTimer.start(SaveWindowPositionTimeout);
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psStateChanged(Qt::WindowState state) {
 | |
| 	psUpdateSysMenu(state);
 | |
| 	psUpdateMargins();
 | |
| 	if (state == Qt::WindowMinimized && GetWindowLong(ps_hWnd, GWL_HWNDPARENT)) {
 | |
| 		App::wnd()->minimizeToTray();
 | |
| 	}
 | |
| 	psSavePosition(state);
 | |
| }
 | |
| 
 | |
| Q_DECLARE_METATYPE(QMargins);
 | |
| void PsMainWindow::psFirstShow() {
 | |
| 	_psShadowWindows.init(_shActive);
 | |
| 	finished = false;
 | |
| 
 | |
| 	psUpdateMargins();
 | |
| 
 | |
| 	_psShadowWindows.update(_PsShadowHidden);
 | |
| 	bool showShadows = true;
 | |
| 
 | |
| 	show();
 | |
| 	if (cWindowPos().maximized) {
 | |
| 		setWindowState(Qt::WindowMaximized);
 | |
| 	}
 | |
| 
 | |
| 	if (cFromAutoStart()) {
 | |
| 		if (cStartMinimized()) {
 | |
| 			setWindowState(Qt::WindowMinimized);
 | |
| 			if (cWorkMode() == dbiwmTrayOnly || cWorkMode() == dbiwmWindowAndTray) {
 | |
| 				hide();
 | |
| 			} else {
 | |
| 				show();
 | |
| 			}
 | |
| 			showShadows = false;
 | |
| 		} else {
 | |
| 			show();
 | |
| 		}
 | |
| 	} else {
 | |
| 		show();
 | |
| 	}
 | |
| 	posInited = true;
 | |
| 	if (showShadows) {
 | |
| 		_psShadowWindows.update(_PsShadowMoved | _PsShadowResized | _PsShadowShown);
 | |
| 	}
 | |
| }
 | |
| 
 | |
| bool PsMainWindow::psHandleTitle() {
 | |
| 	return useDWM;
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psInitSysMenu() {
 | |
| 	Qt::WindowStates states = windowState();
 | |
| 	ps_menu = GetSystemMenu(ps_hWnd, FALSE);
 | |
| 	psUpdateSysMenu(windowHandle()->windowState());
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psUpdateSysMenu(Qt::WindowState state) {
 | |
| 	if (!ps_menu) return;
 | |
| 
 | |
| 	int menuToDisable = SC_RESTORE;
 | |
| 	if (state == Qt::WindowMaximized) {
 | |
| 		menuToDisable = SC_MAXIMIZE;
 | |
| 	} else if (state == Qt::WindowMinimized) {
 | |
| 		menuToDisable = SC_MINIMIZE;
 | |
| 	}
 | |
| 	int itemCount = GetMenuItemCount(ps_menu);
 | |
| 	for (int i = 0; i < itemCount; ++i) {
 | |
| 		MENUITEMINFO itemInfo = {0};
 | |
| 		itemInfo.cbSize = sizeof(itemInfo);
 | |
| 		itemInfo.fMask = MIIM_TYPE | MIIM_STATE | MIIM_ID;
 | |
| 		if (GetMenuItemInfo(ps_menu, i, TRUE, &itemInfo)) {
 | |
| 			if (itemInfo.fType & MFT_SEPARATOR) {
 | |
| 				continue;
 | |
| 			}
 | |
| 			if (itemInfo.wID && !(itemInfo.fState & MFS_DEFAULT)) {
 | |
| 				UINT fOldState = itemInfo.fState, fState = itemInfo.fState & ~MFS_DISABLED;
 | |
| 				if (itemInfo.wID == SC_CLOSE) {
 | |
| 					fState |= MFS_DEFAULT;
 | |
| 				} else if (itemInfo.wID == menuToDisable || (itemInfo.wID != SC_MINIMIZE && itemInfo.wID != SC_MAXIMIZE && itemInfo.wID != SC_RESTORE)) {
 | |
| 					fState |= MFS_DISABLED;
 | |
| 				}
 | |
| 				itemInfo.fMask = MIIM_STATE;
 | |
| 				itemInfo.fState = fState;
 | |
| 				if (!SetMenuItemInfo(ps_menu, i, TRUE, &itemInfo)) {
 | |
| 					DEBUG_LOG(("PS Error: could not set state %1 to menu item %2, old state %3, error %4").arg(fState).arg(itemInfo.wID).arg(fOldState).arg(GetLastError()));
 | |
| 					DestroyMenu(ps_menu);
 | |
| 					ps_menu = 0;
 | |
| 					break;
 | |
| 				}
 | |
| 			}
 | |
| 		} else {
 | |
| 			DEBUG_LOG(("PS Error: could not get state, menu item %1 of %2, error %3").arg(i).arg(itemCount).arg(GetLastError()));
 | |
| 			DestroyMenu(ps_menu);
 | |
| 			ps_menu = 0;
 | |
| 			break;
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psUpdateMargins() {
 | |
| 	if (!useDWM) return;
 | |
| 
 | |
| 	RECT r, a;
 | |
| 
 | |
| 	GetClientRect(ps_hWnd, &r);
 | |
| 	a = r;
 | |
| 
 | |
| 	LONG style = GetWindowLong(ps_hWnd, GWL_STYLE), styleEx = GetWindowLong(ps_hWnd, GWL_EXSTYLE);
 | |
| 	AdjustWindowRectEx(&a, style, false, styleEx);
 | |
| 	simpleMargins = QMargins(a.left - r.left, a.top - r.top, r.right - a.right, r.bottom - a.bottom);
 | |
| 	if (style & WS_MAXIMIZE) {
 | |
| 		RECT w, m;
 | |
| 		GetWindowRect(ps_hWnd, &w);
 | |
| 		m = w;
 | |
| 		
 | |
| 		HMONITOR hMonitor = MonitorFromRect(&w, MONITOR_DEFAULTTONEAREST);
 | |
| 		if (hMonitor) {
 | |
| 			MONITORINFO mi;
 | |
| 			mi.cbSize = sizeof(mi);
 | |
| 			GetMonitorInfo(hMonitor, &mi);
 | |
| 			m = mi.rcWork;
 | |
| 		}
 | |
| 
 | |
| 		dleft = w.left - m.left;
 | |
| 		dtop = w.top - m.top;
 | |
| 
 | |
| 		margins.setLeft(simpleMargins.left() - w.left + m.left);
 | |
| 		margins.setRight(simpleMargins.right() - m.right + w.right);
 | |
| 		margins.setBottom(simpleMargins.bottom() - m.bottom + w.bottom);
 | |
| 		margins.setTop(simpleMargins.top() - w.top + m.top);
 | |
| 	} else {
 | |
| 		margins = simpleMargins;
 | |
| 		dleft = dtop = 0;
 | |
| 	}
 | |
| 
 | |
| 	QPlatformNativeInterface *i = QGuiApplication::platformNativeInterface();
 | |
| 	i->setWindowProperty(windowHandle()->handle(), "WindowsCustomMargins", QVariant::fromValue<QMargins>(margins));
 | |
| 	if (!themeInited) {
 | |
| 		themeInited = true;
 | |
| 		if (useTheme) {
 | |
| 			if (QSysInfo::WindowsVersion < QSysInfo::WV_WINDOWS8) {
 | |
| 				setWindowTheme(ps_hWnd, L" ", L" ");
 | |
| 				QApplication::setStyle(QStyleFactory::create("Windows"));
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psFlash() {
 | |
| 	if (GetForegroundWindow() == ps_hWnd) return;
 | |
| 
 | |
| 	FLASHWINFO info;
 | |
| 	info.cbSize = sizeof(info);
 | |
| 	info.hwnd = ps_hWnd;
 | |
| 	info.dwFlags = FLASHW_ALL;
 | |
| 	info.dwTimeout = 0;
 | |
| 	info.uCount = 1;
 | |
| 	FlashWindowEx(&info);
 | |
| }
 | |
| 
 | |
| HWND PsMainWindow::psHwnd() const {
 | |
| 	return ps_hWnd;
 | |
| }
 | |
| 
 | |
| HMENU PsMainWindow::psMenu() const {
 | |
| 	return ps_menu;
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psDestroyIcons() {
 | |
| 	if (ps_iconBig) {
 | |
|         DestroyIcon(ps_iconBig);
 | |
|         ps_iconBig = 0;
 | |
|     }
 | |
|     if (ps_iconSmall) {
 | |
|         DestroyIcon(ps_iconSmall);
 | |
|         ps_iconSmall = 0;
 | |
|     }
 | |
| 	if (ps_iconOverlay) {
 | |
| 		DestroyIcon(ps_iconOverlay);
 | |
| 		ps_iconOverlay = 0;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| PsMainWindow::~PsMainWindow() {
 | |
| 	finished = true;
 | |
| 	if (ps_menu) DestroyMenu(ps_menu);
 | |
| 	psDestroyIcons();
 | |
| 	_psShadowWindows.destroy();
 | |
| 	if (ps_tbHider_hWnd) DestroyWindow(ps_tbHider_hWnd);
 | |
| }
 | |
| 
 | |
| namespace {
 | |
| 	QRect _monitorRect;
 | |
| 	uint64 _monitorLastGot = 0;
 | |
| }
 | |
| 
 | |
| QRect psDesktopRect() {
 | |
| 	uint64 tnow = getms();
 | |
| 	if (tnow > _monitorLastGot + 1000 || 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) {
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psActivateNotify(NotifyWindow *w) {
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psClearNotifies(PeerId peerId) {
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psNotifyShown(NotifyWindow *w) {
 | |
| }
 | |
| 
 | |
| void PsMainWindow::psPlatformNotify(HistoryItem *item) {
 | |
| }
 | |
| 
 | |
| PsApplication::PsApplication(int &argc, char **argv) : QApplication(argc, argv) {
 | |
| }
 | |
| 
 | |
| void PsApplication::psInstallEventFilter() {
 | |
| 	delete _psEventFilter;
 | |
| 	_psEventFilter = new _PsEventFilter();
 | |
| 	installNativeEventFilter(_psEventFilter);
 | |
| }
 | |
| 
 | |
| PsApplication::~PsApplication() {
 | |
| 	delete _psEventFilter;
 | |
| 	_psEventFilter = 0;
 | |
| }
 | |
| 
 | |
| PsUpdateDownloader::PsUpdateDownloader(QThread *thread, const MTPDhelp_appUpdate &update) : already(0), reply(0), full(0) {
 | |
| 	updateUrl = qs(update.vurl);
 | |
| 	moveToThread(thread);
 | |
| 	manager.moveToThread(thread);
 | |
| 	App::setProxySettings(manager);
 | |
| 
 | |
| 	connect(thread, SIGNAL(started()), this, SLOT(start()));
 | |
| 	initOutput();
 | |
| }
 | |
| 
 | |
| PsUpdateDownloader::PsUpdateDownloader(QThread *thread, const QString &url) : already(0), reply(0), full(0) {
 | |
| 	updateUrl = url;
 | |
| 	moveToThread(thread);
 | |
| 	manager.moveToThread(thread);
 | |
| 	App::setProxySettings(manager);
 | |
| 
 | |
| 	connect(thread, SIGNAL(started()), this, SLOT(start()));
 | |
| 	initOutput();
 | |
| }
 | |
| 
 | |
| void PsUpdateDownloader::initOutput() {
 | |
| 	QString fileName;
 | |
| 	QRegularExpressionMatch m = QRegularExpression(qsl("/([^/\\?]+)(\\?|$)")).match(updateUrl);
 | |
| 	if (m.hasMatch()) {
 | |
| 		fileName = m.captured(1).replace(QRegularExpression(qsl("[^a-zA-Z0-9_\\-]")), QString());
 | |
| 	}
 | |
| 	if (fileName.isEmpty()) {
 | |
| 		fileName = qsl("tupdate-%1").arg(rand());
 | |
| 	}
 | |
| 	QString dirStr = cWorkingDir() + qsl("tupdates/");
 | |
| 	fileName = dirStr + fileName;
 | |
| 	QFileInfo file(fileName);
 | |
| 
 | |
| 	QDir dir(dirStr);
 | |
| 	if (dir.exists()) {
 | |
| 		QFileInfoList all = dir.entryInfoList(QDir::Files);
 | |
| 		for (QFileInfoList::iterator i = all.begin(), e = all.end(); i != e; ++i) {
 | |
| 			if (i->absoluteFilePath() != file.absoluteFilePath()) {
 | |
| 				QFile::remove(i->absoluteFilePath());
 | |
| 			}
 | |
| 		}
 | |
| 	} else {
 | |
| 		dir.mkdir(dir.absolutePath());
 | |
| 	}
 | |
| 	outputFile.setFileName(fileName);
 | |
| 	if (file.exists()) {
 | |
| 		uint64 fullSize = file.size();
 | |
| 		if (fullSize < INT_MAX) {
 | |
| 			int32 goodSize = (int32)fullSize;
 | |
| 			if (goodSize % UpdateChunk) {
 | |
| 				goodSize = goodSize - (goodSize % UpdateChunk);
 | |
| 				if (goodSize) {
 | |
| 					if (outputFile.open(QIODevice::ReadOnly)) {
 | |
| 						QByteArray goodData = outputFile.readAll().mid(0, goodSize);
 | |
| 						outputFile.close();
 | |
| 						if (outputFile.open(QIODevice::WriteOnly)) {
 | |
| 							outputFile.write(goodData);
 | |
| 							outputFile.close();
 | |
| 							
 | |
| 							QMutexLocker lock(&mutex);
 | |
| 							already = goodSize;
 | |
| 						}
 | |
| 					}
 | |
| 				}
 | |
| 			} else {
 | |
| 				QMutexLocker lock(&mutex);
 | |
| 				already = goodSize;
 | |
| 			}
 | |
| 		}
 | |
| 		if (!already) {
 | |
| 			QFile::remove(fileName);
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void PsUpdateDownloader::start() {
 | |
| 	sendRequest();
 | |
| }
 | |
| 
 | |
| void PsUpdateDownloader::sendRequest() {
 | |
| 	QNetworkRequest req(updateUrl);
 | |
| 	QByteArray rangeHeaderValue = "bytes=" + QByteArray::number(already) + "-";// + QByteArray::number(already + cUpdateChunk() - 1); 
 | |
| 	req.setRawHeader("Range", rangeHeaderValue);
 | |
| 	req.setAttribute(QNetworkRequest::HttpPipeliningAllowedAttribute, true);
 | |
| 	if (reply) reply->deleteLater();
 | |
| 	reply = manager.get(req);
 | |
| 	connect(reply, SIGNAL(downloadProgress(qint64,qint64)), this, SLOT(partFinished(qint64,qint64)));
 | |
| 	connect(reply, SIGNAL(error(QNetworkReply::NetworkError)), this, SLOT(partFailed(QNetworkReply::NetworkError)));
 | |
| 	connect(reply, SIGNAL(metaDataChanged()), this, SLOT(partMetaGot()));
 | |
| }
 | |
| 
 | |
| void PsUpdateDownloader::partMetaGot() {
 | |
| 	typedef QList<QNetworkReply::RawHeaderPair> Pairs;
 | |
| 	Pairs pairs = reply->rawHeaderPairs();
 | |
| 	for (Pairs::iterator i = pairs.begin(), e = pairs.end(); i != e; ++i) {
 | |
| 		if (QString::fromUtf8(i->first).toLower() == "content-range") {
 | |
| 			QRegularExpressionMatch m = QRegularExpression(qsl("/(\\d+)([^\\d]|$)")).match(QString::fromUtf8(i->second));
 | |
| 			if (m.hasMatch()) {
 | |
| 				{
 | |
| 					QMutexLocker lock(&mutex);
 | |
| 					full = m.captured(1).toInt();
 | |
| 				}
 | |
| 				emit App::app()->updateDownloading(already, full);
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| int32 PsUpdateDownloader::ready() {
 | |
| 	QMutexLocker lock(&mutex);
 | |
| 	return already;
 | |
| }
 | |
| 
 | |
| int32 PsUpdateDownloader::size() {
 | |
| 	QMutexLocker lock(&mutex);
 | |
| 	return full;
 | |
| }
 | |
| 
 | |
| void PsUpdateDownloader::partFinished(qint64 got, qint64 total) {
 | |
| 	if (!reply) return;
 | |
| 
 | |
| 	QVariant statusCode = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);
 | |
|     if (statusCode.isValid()) {
 | |
| 	    int status = statusCode.toInt();
 | |
| 		if (status != 200 && status != 206 && status != 416) {
 | |
| 			LOG(("Update Error: Bad HTTP status received in partFinished(): %1").arg(status));
 | |
| 			return fatalFail();
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	if (!already && !full) {
 | |
| 		QMutexLocker lock(&mutex);
 | |
| 		full = total;
 | |
| 	}
 | |
| 	DEBUG_LOG(("Update Info: part %1 of %2").arg(got).arg(total));
 | |
| 
 | |
| 	if (!outputFile.isOpen()) {
 | |
| 		if (!outputFile.open(QIODevice::Append)) {
 | |
| 			LOG(("Update Error: Could not open output file '%1' for appending").arg(outputFile.fileName()));
 | |
| 			return fatalFail();
 | |
| 		}
 | |
| 	}
 | |
| 	QByteArray r = reply->readAll();
 | |
| 	if (!r.isEmpty()) {
 | |
| 		outputFile.write(r);
 | |
| 
 | |
| 		QMutexLocker lock(&mutex);
 | |
| 		already += r.size();
 | |
| 	}
 | |
| 	if (got >= total) {
 | |
| 		reply->deleteLater();
 | |
| 		reply = 0;
 | |
| 		outputFile.close();
 | |
| 		unpackUpdate();
 | |
| 	} else {
 | |
| 		emit App::app()->updateDownloading(already, full);
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void PsUpdateDownloader::partFailed(QNetworkReply::NetworkError e) {
 | |
| 	if (!reply) return;
 | |
| 
 | |
| 	QVariant statusCode = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);
 | |
| 	reply->deleteLater();
 | |
| 	reply = 0;
 | |
|     if (statusCode.isValid()) {
 | |
| 	    int status = statusCode.toInt();
 | |
| 		if (status == 416) { // Requested range not satisfiable
 | |
| 			outputFile.close();
 | |
| 			unpackUpdate();
 | |
| 			return;
 | |
| 		}
 | |
| 	}
 | |
| 	LOG(("Update Error: failed to download part starting from %1, error %2").arg(already).arg(e));
 | |
| 	emit App::app()->updateFailed();
 | |
| }
 | |
| 
 | |
| void PsUpdateDownloader::deleteDir(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);
 | |
| }
 | |
| 
 | |
| void PsUpdateDownloader::fatalFail() {
 | |
| 	clearAll();
 | |
| 	emit App::app()->updateFailed();
 | |
| }
 | |
| 
 | |
| void PsUpdateDownloader::clearAll() {
 | |
| 	deleteDir(cWorkingDir() + qsl("tupdates"));
 | |
| }
 | |
| 
 | |
| void PsUpdateDownloader::unpackUpdate() {
 | |
| 	QByteArray packed;
 | |
| 	if (!outputFile.open(QIODevice::ReadOnly)) {
 | |
| 		LOG(("Update Error: cant read updates file!"));
 | |
| 		return fatalFail();
 | |
| 	}
 | |
| 
 | |
| 	const int32 hSigLen = 128, hShaLen = 20, hPropsLen = LZMA_PROPS_SIZE, hOriginalSizeLen = sizeof(int32), hSize = hSigLen + hShaLen + hPropsLen + hOriginalSizeLen; // header
 | |
| 
 | |
| 	QByteArray compressed = outputFile.readAll();
 | |
| 	int32 compressedLen = compressed.size() - hSize;
 | |
| 	if (compressedLen <= 0) {
 | |
| 		LOG(("Update Error: bad compressed size: %1").arg(compressed.size()));
 | |
| 		return fatalFail();
 | |
| 	}
 | |
| 	outputFile.close();
 | |
| 
 | |
| 	QString tempDirPath = cWorkingDir() + qsl("tupdates/temp"), readyDirPath = cWorkingDir() + qsl("tupdates/ready");
 | |
| 	deleteDir(tempDirPath);
 | |
| 	deleteDir(readyDirPath);
 | |
| 
 | |
| 	QDir tempDir(tempDirPath), readyDir(readyDirPath);
 | |
| 	if (tempDir.exists() || readyDir.exists()) {
 | |
| 		LOG(("Update Error: cant clear tupdates/temp or tupdates/ready dir!"));
 | |
| 		return fatalFail();
 | |
| 	}
 | |
| 
 | |
| 	uchar sha1Buffer[20];
 | |
| 	bool goodSha1 = !memcmp(compressed.constData() + hSigLen, hashSha1(compressed.constData() + hSigLen + hShaLen, compressedLen + hPropsLen + hOriginalSizeLen, sha1Buffer), hShaLen);
 | |
| 	if (!goodSha1) {
 | |
| 		LOG(("Update Error: bad SHA1 hash of update file!"));
 | |
| 		return fatalFail();
 | |
| 	}
 | |
| 
 | |
| 	RSA *pbKey = PEM_read_bio_RSAPublicKey(BIO_new_mem_buf(const_cast<char*>(UpdatesPublicKey), -1), 0, 0, 0);
 | |
| 	if (!pbKey) {
 | |
| 		LOG(("Update Error: cant read public rsa key!"));
 | |
| 		return fatalFail();
 | |
| 	}
 | |
| 	if (RSA_verify(NID_sha1, (const uchar*)(compressed.constData() + hSigLen), hShaLen, (const uchar*)(compressed.constData()), hSigLen, pbKey) != 1) { // verify signature
 | |
| 		RSA_free(pbKey);
 | |
| 		LOG(("Update Error: bad RSA signature of update file!"));
 | |
| 		return fatalFail();
 | |
| 	}
 | |
| 	RSA_free(pbKey);
 | |
| 
 | |
| 	QByteArray uncompressed;
 | |
| 
 | |
| 	int32 uncompressedLen;
 | |
| 	memcpy(&uncompressedLen, compressed.constData() + hSigLen + hShaLen + hPropsLen, hOriginalSizeLen);
 | |
| 	uncompressed.resize(uncompressedLen);
 | |
| 
 | |
| 	size_t resultLen = uncompressed.size();
 | |
| 	SizeT srcLen = compressedLen;
 | |
| 	int uncompressRes = LzmaUncompress((uchar*)uncompressed.data(), &resultLen, (const uchar*)(compressed.constData() + hSize), &srcLen, (const uchar*)(compressed.constData() + hSigLen + hShaLen), LZMA_PROPS_SIZE);
 | |
| 	if (uncompressRes != SZ_OK) {
 | |
| 		LOG(("Update Error: could not uncompress lzma, code: %1").arg(uncompressRes));
 | |
| 		return fatalFail();
 | |
| 	}
 | |
| 
 | |
| 	tempDir.mkdir(tempDir.absolutePath());
 | |
| 
 | |
| 	quint32 version;
 | |
| 	{
 | |
| 		QBuffer buffer(&uncompressed);
 | |
| 		buffer.open(QIODevice::ReadOnly);
 | |
| 		QDataStream stream(&buffer);
 | |
| 		stream.setVersion(QDataStream::Qt_5_1);
 | |
| 
 | |
| 		stream >> version;
 | |
| 		if (stream.status() != QDataStream::Ok) {
 | |
| 			LOG(("Update Error: cant read version from downloaded stream, status: %1").arg(stream.status()));
 | |
| 			return fatalFail();
 | |
| 		}
 | |
| 		if (version <= AppVersion) {
 | |
| 			LOG(("Update Error: downloaded version %1 is not greater, than mine %2").arg(version).arg(AppVersion));
 | |
| 			return fatalFail();
 | |
| 		}
 | |
| 
 | |
| 		quint32 filesCount;
 | |
| 		stream >> filesCount;
 | |
| 		if (stream.status() != QDataStream::Ok) {
 | |
| 			LOG(("Update Error: cant read files count from downloaded stream, status: %1").arg(stream.status()));
 | |
| 			return fatalFail();
 | |
| 		}
 | |
| 		if (!filesCount) {
 | |
| 			LOG(("Update Error: update is empty!"));
 | |
| 			return fatalFail();
 | |
| 		}
 | |
| 		for (int32 i = 0; i < filesCount; ++i) {
 | |
| 			QString relativeName;
 | |
| 			quint32 fileSize;
 | |
| 			QByteArray fileInnerData;
 | |
| 
 | |
| 			stream >> relativeName >> fileSize >> fileInnerData;
 | |
| 			if (stream.status() != QDataStream::Ok) {
 | |
| 				LOG(("Update Error: cant read file from downloaded stream, status: %1").arg(stream.status()));
 | |
| 				return fatalFail();
 | |
| 			}
 | |
| 			if (fileSize != fileInnerData.size()) {
 | |
| 				LOG(("Update Error: bad file size %1 not matching data size %2").arg(fileSize).arg(fileInnerData.size()));
 | |
| 				return fatalFail();
 | |
| 			}
 | |
| 
 | |
| 			QFile f(tempDirPath + '/' + relativeName);
 | |
| 			if (!f.open(QIODevice::WriteOnly)) {
 | |
| 				LOG(("Update Error: cant open file '%1' for writing").arg(tempDirPath + '/' + relativeName));
 | |
| 				return fatalFail();
 | |
| 			}
 | |
| 			if (f.write(fileInnerData) != fileSize) {
 | |
| 				f.close();
 | |
| 				LOG(("Update Error: cant write file '%1'").arg(tempDirPath + '/' + relativeName));
 | |
| 				return fatalFail();
 | |
| 			}
 | |
| 			f.close();
 | |
| 		}
 | |
| 
 | |
| 		// create tdata/version file
 | |
| 		tempDir.mkdir(QDir(tempDirPath + qsl("/tdata")).absolutePath());
 | |
| 		std::wstring versionString = ((version % 1000) ? QString("%1.%2.%3").arg(int(version / 1000000)).arg(int((version % 1000000) / 1000)).arg(int(version % 1000)) : QString("%1.%2").arg(int(version / 1000000)).arg(int((version % 1000000) / 1000))).toStdWString();
 | |
| 		DWORD versionNum = DWORD(version), versionLen = DWORD(versionString.size() * sizeof(WCHAR));
 | |
| 		WCHAR versionStr[32];
 | |
| 		memcpy(versionStr, versionString.c_str(), versionLen);
 | |
| 
 | |
| 		QFile fVersion(tempDirPath + qsl("/tdata/version"));		
 | |
| 		if (!fVersion.open(QIODevice::WriteOnly)) {
 | |
| 			LOG(("Update Error: cant write version file '%1'").arg(tempDirPath + qsl("/version")));
 | |
| 			return fatalFail();
 | |
| 		}
 | |
| 		fVersion.write((const char*)&versionNum, sizeof(DWORD));
 | |
| 		fVersion.write((const char*)&versionLen, sizeof(DWORD));
 | |
| 		fVersion.write((const char*)&versionStr[0], versionLen);
 | |
| 		fVersion.close();
 | |
| 	}
 | |
| 	
 | |
| 	if (!tempDir.rename(tempDir.absolutePath(), readyDir.absolutePath())) {
 | |
| 		LOG(("Update Error: cant rename temp dir '%1' to ready dir '%2'").arg(tempDir.absolutePath()).arg(readyDir.absolutePath()));
 | |
| 		return fatalFail();
 | |
| 	}
 | |
| 	deleteDir(tempDirPath);
 | |
| 	outputFile.remove();
 | |
| 
 | |
| 	emit App::app()->updateReady();
 | |
| }
 | |
| 
 | |
| PsUpdateDownloader::~PsUpdateDownloader() {
 | |
| 	delete reply;
 | |
| 	reply = 0;
 | |
| }
 | |
| 
 | |
| 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);
 | |
| 					return FALSE;
 | |
| 				}
 | |
| 			}
 | |
| 		}
 | |
| 		return TRUE;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void psActivateProcess(uint64 pid) {
 | |
| 	if (pid) {
 | |
| 		::EnumWindows((WNDENUMPROC)_ActivateProcess, (LPARAM)&pid);
 | |
| 	}
 | |
| }
 | |
| 
 | |
| QString psCurrentCountry() {
 | |
| 	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);
 | |
| 		return len ? QString::fromStdWString(std::wstring(wstrCountry)) : QString::fromLatin1(DefaultCountry);
 | |
| 	}
 | |
| 	return QString::fromLatin1(DefaultCountry);
 | |
| }
 | |
| 
 | |
| namespace {
 | |
| 	QString langById(int lngId) {
 | |
| 		int primary = lngId & 0xFF;
 | |
| 		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::fromLatin1(DefaultLanguage);
 | |
| 	}
 | |
| }
 | |
| 
 | |
| QString psCurrentLanguage() {
 | |
| 	int chCount = GetLocaleInfo(LOCALE_USER_DEFAULT, LOCALE_SNAME, 0, 0);
 | |
| 	if (chCount && chCount < 128) {
 | |
| 		WCHAR wstrLocale[128];
 | |
| 		int len = GetLocaleInfo(LOCALE_USER_DEFAULT, LOCALE_SNAME, wstrLocale, chCount);
 | |
| 		if (!len) return QString::fromLatin1(DefaultLanguage);
 | |
| 		QString locale = QString::fromStdWString(std::wstring(wstrLocale));
 | |
| 		QRegularExpressionMatch m = QRegularExpression("(^|[^a-z])([a-z]{2})-").match(locale);
 | |
| 		if (m.hasMatch()) {
 | |
| 			return m.captured(2);
 | |
| 		}
 | |
| 	}
 | |
| 	chCount = GetLocaleInfo(LOCALE_USER_DEFAULT, LOCALE_ILANGUAGE, 0, 0);
 | |
| 	if (chCount && chCount < 128) {
 | |
| 		WCHAR wstrLocale[128];
 | |
| 		int len = GetLocaleInfo(LOCALE_USER_DEFAULT, LOCALE_ILANGUAGE, wstrLocale, chCount), lngId = 0;
 | |
| 		if (len < 5) return QString::fromLatin1(DefaultLanguage);
 | |
| 
 | |
| 		for (int i = 0; i < 4; ++i) {
 | |
| 			WCHAR ch = wstrLocale[i];
 | |
| 			lngId *= 16;
 | |
| 			if (ch >= WCHAR('0') && ch <= WCHAR('9')) {
 | |
| 				lngId += (ch - WCHAR('0'));
 | |
| 			} else if (ch >= WCHAR('A') && ch <= WCHAR('F')) {
 | |
| 				lngId += (10 + ch - WCHAR('A'));
 | |
| 			} else {
 | |
| 				return QString::fromLatin1(DefaultLanguage);
 | |
| 			}
 | |
| 		}
 | |
| 		return langById(lngId);
 | |
| 	}
 | |
| 	return QString::fromLatin1(DefaultLanguage);
 | |
| }
 | |
| 
 | |
| 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)));
 | |
| 		return appData.absolutePath() + '/' + QString::fromWCharArray(AppName) + '/';
 | |
| 	}
 | |
| 	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() + '/' + QString::fromWCharArray(AppNameOld) + '/';
 | |
| 	}
 | |
| 	return QString();
 | |
| }
 | |
| 
 | |
| QString psDownloadPath() {
 | |
| 	return QStandardPaths::writableLocation(QStandardPaths::DownloadLocation) + '/' + QString::fromWCharArray(AppName) + '/';
 | |
| }
 | |
| 
 | |
| QString psCurrentExeDirectory(int argc, char *argv[]) {
 | |
| 	LPWSTR *args;
 | |
| 	int argsCount;
 | |
| 	args = CommandLineToArgvW(GetCommandLine(), &argsCount);
 | |
| 	if (args) {
 | |
| 		QFileInfo info(QDir::fromNativeSeparators(QString::fromWCharArray(args[0])));
 | |
| 		if (info.isFile()) {
 | |
| 			return info.absoluteDir().absolutePath() + '/';
 | |
| 		}
 | |
| 		LocalFree(args);
 | |
| 	}
 | |
| 	return QString();
 | |
| }
 | |
| 
 | |
| QString psCurrentExeName(int argc, char *argv[]) {
 | |
| 	LPWSTR *args;
 | |
| 	int argsCount;
 | |
| 	args = CommandLineToArgvW(GetCommandLine(), &argsCount);
 | |
| 	if (args) {
 | |
| 		QFileInfo info(QDir::fromNativeSeparators(QString::fromWCharArray(args[0])));
 | |
| 		if (info.isFile()) {
 | |
| 			return info.fileName();
 | |
| 		}
 | |
| 		LocalFree(args);
 | |
| 	}
 | |
| 	return QString();
 | |
| }
 | |
| 
 | |
| void psDoCleanup() {
 | |
| 	try {
 | |
| 		psAutoStart(false, true);
 | |
| 		psSendToMenu(false, true);
 | |
| 	} catch (...) {
 | |
| 	}
 | |
| }
 | |
| 
 | |
| 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 = QString::fromStdWString(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;
 | |
| }
 | |
| 
 | |
| bool psCheckReadyUpdate() {
 | |
| 	QString readyPath = cWorkingDir() + qsl("tupdates/ready");
 | |
| 	if (!QDir(readyPath).exists()) {
 | |
| 		return false;
 | |
| 	}
 | |
| 
 | |
| 	// check ready version
 | |
| 	QString versionPath = readyPath + qsl("/tdata/version");
 | |
| 	{
 | |
| 		QFile fVersion(versionPath);
 | |
| 		if (!fVersion.open(QIODevice::ReadOnly)) {
 | |
| 			LOG(("Update Error: cant read version file '%1'").arg(versionPath));
 | |
| 			PsUpdateDownloader::clearAll();
 | |
| 			return false;
 | |
| 		}
 | |
| 		DWORD versionNum;
 | |
| 		if (fVersion.read((char*)&versionNum, sizeof(DWORD)) != sizeof(DWORD)) {
 | |
| 			LOG(("Update Error: cant read version from file '%1'").arg(versionPath));
 | |
| 			PsUpdateDownloader::clearAll();
 | |
| 			return false;
 | |
| 		}
 | |
| 		fVersion.close();
 | |
| 		if (versionNum <= AppVersion) {
 | |
| 			LOG(("Update Error: cant install version %1 having version %2").arg(versionNum).arg(AppVersion));
 | |
| 			PsUpdateDownloader::clearAll();
 | |
| 			return false;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	QString curUpdater = (cExeDir() + qsl("Updater.exe"));
 | |
| 	QFileInfo updater(cWorkingDir() + qsl("tupdates/ready/Updater.exe"));
 | |
| 	if (!updater.exists()) {
 | |
| 		QFileInfo current(curUpdater);
 | |
| 		if (!current.exists()) {
 | |
| 			PsUpdateDownloader::clearAll();
 | |
| 			return false;
 | |
| 		}
 | |
| 		if (CopyFile(current.absoluteFilePath().toStdWString().c_str(), updater.absoluteFilePath().toStdWString().c_str(), TRUE) == FALSE) {
 | |
| 			PsUpdateDownloader::clearAll();
 | |
| 			return false;
 | |
| 		}
 | |
| 	}
 | |
| 	if (CopyFile(updater.absoluteFilePath().toStdWString().c_str(), curUpdater.toStdWString().c_str(), FALSE) == FALSE) {
 | |
| 		DWORD errorCode = GetLastError();
 | |
| 		if (errorCode == ERROR_ACCESS_DENIED) { // we are in write-protected dir, like Program Files
 | |
| 			cSetWriteProtected(true);
 | |
| 			return true;
 | |
| 		} else {
 | |
| 			PsUpdateDownloader::clearAll();
 | |
| 			return false;
 | |
| 		}
 | |
| 	}
 | |
| 	if (DeleteFile(updater.absoluteFilePath().toStdWString().c_str()) == FALSE) {
 | |
| 		PsUpdateDownloader::clearAll();
 | |
| 		return false;
 | |
| 	}
 | |
| 	return true;
 | |
| }
 | |
| 
 | |
| void psPostprocessFile(const QString &name) {
 | |
| 	std::wstring zoneFile = QDir::toNativeSeparators(name).toStdWString() + L":Zone.Identifier";
 | |
| 	HANDLE f = CreateFile(zoneFile.c_str(), GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, 0, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
 | |
| 	if (f == INVALID_HANDLE_VALUE) { // :(
 | |
| 		return;
 | |
| 	}
 | |
| 
 | |
| 	const char data[] = "[ZoneTransfer]\r\nZoneId=3\r\n";
 | |
| 
 | |
| 	DWORD written = 0;
 | |
| 	BOOL result = WriteFile(f, data, sizeof(data), &written, NULL);
 | |
| 	CloseHandle(f);
 | |
| 
 | |
| 	if (!result || written != sizeof(data)) { // :(
 | |
| 		return;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void psOpenFile(const QString &name, bool openWith) {
 | |
| 	std::wstring wname = QDir::toNativeSeparators(name).toStdWString();
 | |
| 
 | |
| 	if (openWith && useOpenAs) {
 | |
| 		if (shOpenWithDialog) {
 | |
| 			OPENASINFO info;
 | |
| 			info.oaifInFlags = OAIF_ALLOW_REGISTRATION | OAIF_REGISTER_EXT | OAIF_EXEC;
 | |
| 			info.pcszClass = NULL;
 | |
| 			info.pcszFile = wname.c_str();
 | |
| 			shOpenWithDialog(0, &info);
 | |
| 		} else {
 | |
| 			openAs_RunDLL(0, 0, wname.c_str(), SW_SHOWNORMAL);
 | |
| 		}
 | |
| 	} else {
 | |
| 		ShellExecute(0, L"open", wname.c_str(), 0, 0, SW_SHOWNORMAL);
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void psShowInFolder(const QString &name) {
 | |
| 	QString nameEscaped = QDir::toNativeSeparators(name).replace('"', qsl("\"\""));
 | |
| 	ShellExecute(0, 0, qsl("explorer").toStdWString().c_str(), (qsl("/select,") + nameEscaped).toStdWString().c_str(), 0, SW_SHOWNORMAL);
 | |
| }
 | |
| 
 | |
| void psStart() {
 | |
| }
 | |
| 
 | |
| void psFinish() {
 | |
| }
 | |
| 
 | |
| namespace {
 | |
| 	void _psLogError(const char *str, LSTATUS code) {
 | |
| 		WCHAR errMsg[2048];
 | |
| 		LPTSTR errorText = NULL, errorTextDefault = L"(Unknown error)";
 | |
| 		FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_IGNORE_INSERTS, NULL, code, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR)&errorText, 0, 0);
 | |
| 		if (!errorText) {
 | |
| 			errorText = errorTextDefault;
 | |
| 		}
 | |
| 		LOG((str).arg(code).arg(QString::fromStdWString(errorText)));
 | |
| 		if (errorText != errorTextDefault) {
 | |
| 			LocalFree(errorText);
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	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;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void psRegisterCustomScheme() {
 | |
| 	DEBUG_LOG(("App Info: Checking custom scheme 'tg'.."));
 | |
| 
 | |
| 	HKEY rkey;
 | |
| 	QString exe = QDir::toNativeSeparators(QDir(cExeDir()).absolutePath() + '/' + QString::fromWCharArray(AppFile) + qsl(".exe"));
 | |
| 
 | |
| 	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;
 | |
| }
 | |
| 
 | |
| void psExecUpdater() {
 | |
| 	QString targs = qsl("-update");
 | |
| 	if (cFromAutoStart()) targs += qsl(" -autostart");
 | |
| 	if (cDebug()) targs += qsl(" -debug");
 | |
| 	if (cWriteProtected()) targs += qsl(" -writeprotected \"") + cExeDir() + '"';
 | |
| 
 | |
| 	QString updaterPath = cWriteProtected() ? (cWorkingDir() + qsl("tupdates/ready/Updater.exe")) : (cExeDir() + qsl("Updater.exe"));
 | |
| 
 | |
| 	QString updater(QDir::toNativeSeparators(updaterPath)), wdir(QDir::toNativeSeparators(cWorkingDir()));
 | |
| 
 | |
| 	DEBUG_LOG(("Application Info: executing %1 %2").arg(cExeDir() + "Updater.exe").arg(targs));
 | |
| 	HINSTANCE r = ShellExecute(0, cWriteProtected() ? L"runas" : 0, updater.toStdWString().c_str(), targs.toStdWString().c_str(), wdir.isEmpty() ? 0 : wdir.toStdWString().c_str(), SW_SHOWNORMAL);
 | |
| 	if (long(r) < 32) {
 | |
| 		DEBUG_LOG(("Application Error: failed to execute %1, working directory: '%2', result: %3").arg(updater).arg(wdir).arg(long(r)));
 | |
| 		QString readyPath = cWorkingDir() + qsl("tupdates/ready");
 | |
| 		PsUpdateDownloader::deleteDir(readyPath);
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void psExecTelegram() {
 | |
| 	QString targs = qsl("-noupdate");
 | |
| 	if (cRestartingToSettings()) targs += qsl(" -tosettings");
 | |
| 	if (cFromAutoStart()) targs += qsl(" -autostart");
 | |
| 	if (cDebug()) targs += qsl(" -debug");
 | |
| 	if (cDataFile() != (cTestMode() ? qsl("data_test") : qsl("data"))) targs += qsl(" -key \"") + cDataFile() + '"';
 | |
| 
 | |
| 	QString telegram(QDir::toNativeSeparators(cExeDir() + QString::fromWCharArray(AppFile) + qsl(".exe"))), wdir(QDir::toNativeSeparators(cWorkingDir()));
 | |
| 
 | |
| 	DEBUG_LOG(("Application Info: executing %1 %2").arg(cExeDir() + QString::fromWCharArray(AppFile) + qsl(".exe")).arg(targs));
 | |
| 	HINSTANCE r = ShellExecute(0, 0, telegram.toStdWString().c_str(), targs.toStdWString().c_str(), wdir.isEmpty() ? 0 : wdir.toStdWString().c_str(), SW_SHOWNORMAL);
 | |
| 	if (long(r) < 32) {
 | |
| 		DEBUG_LOG(("Application Error: failed to execute %1, working directory: '%2', result: %3").arg(telegram).arg(wdir).arg(long(r)));
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void _manageAppLnk(bool create, bool silent, int path_csidl, const wchar_t *args, const wchar_t *description) {
 | |
| 	WCHAR startupFolder[MAX_PATH];
 | |
| 	HRESULT hres = SHGetFolderPath(0, path_csidl, 0, SHGFP_TYPE_CURRENT, startupFolder);
 | |
| 	if (SUCCEEDED(hres)) {
 | |
| 		QString lnk = QString::fromWCharArray(startupFolder) + '\\' + QString::fromWCharArray(AppFile) + qsl(".lnk");
 | |
| 		if (create) {
 | |
| 			IShellLink* psl;
 | |
| 			hres = CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLink, (LPVOID*)&psl);
 | |
| 			if (SUCCEEDED(hres)) {
 | |
| 				IPersistFile* ppf;
 | |
| 
 | |
| 				QString exe = QDir::toNativeSeparators(QDir(cExeDir()).absolutePath() + '/' + QString::fromWCharArray(AppFile) + qsl(".exe")), dir = QDir::toNativeSeparators(QDir(cWorkingDir()).absolutePath());
 | |
| 				psl->SetArguments(args);
 | |
| 				psl->SetPath(exe.toStdWString().c_str());
 | |
| 				psl->SetWorkingDirectory(dir.toStdWString().c_str());
 | |
| 				psl->SetDescription(description);
 | |
| 
 | |
| 				hres = psl->QueryInterface(IID_IPersistFile, (LPVOID*)&ppf);
 | |
| 
 | |
| 				if (SUCCEEDED(hres)) {
 | |
| 					hres = ppf->Save(lnk.toStdWString().c_str(), TRUE);
 | |
| 					ppf->Release();
 | |
| 				} else {
 | |
| 					if (!silent) LOG(("App Error: could not create interface IID_IPersistFile %1").arg(hres));
 | |
| 				}
 | |
| 				psl->Release();
 | |
| 			} else {
 | |
| 				if (!silent) LOG(("App Error: could not create instance of IID_IShellLink %1").arg(hres));
 | |
| 			}
 | |
| 		} else {
 | |
| 			QFile::remove(lnk);
 | |
| 		}
 | |
| 	} else {
 | |
| 		if (!silent) LOG(("App Error: could not get CSIDL %1 folder %2").arg(path_csidl).arg(hres));
 | |
| 	}
 | |
| }
 | |
| 
 | |
| 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);
 | |
| 	widget->event(&e);
 | |
| 	if (!wm) widget->setAttribute(Qt::WA_Mapped, false);
 | |
| 	if (!wv) widget->setAttribute(Qt::WA_WState_Visible, false);
 | |
| }
 | |
| 
 | |
| #ifdef _NEED_WIN_GENERATE_DUMP
 | |
| static const WCHAR *_programName = AppName; // folder in APPDATA, if current path is unavailable for writing
 | |
| static const WCHAR *_exeName = L"Telegram.exe";
 | |
| 
 | |
| LPTOP_LEVEL_EXCEPTION_FILTER _oldWndExceptionFilter = 0;
 | |
| 
 | |
| typedef BOOL (FAR STDAPICALLTYPE *t_miniDumpWriteDump)(
 | |
|     _In_ HANDLE hProcess,
 | |
|     _In_ DWORD ProcessId,
 | |
|     _In_ HANDLE hFile,
 | |
|     _In_ MINIDUMP_TYPE DumpType,
 | |
|     _In_opt_ PMINIDUMP_EXCEPTION_INFORMATION ExceptionParam,
 | |
|     _In_opt_ PMINIDUMP_USER_STREAM_INFORMATION UserStreamParam,
 | |
|     _In_opt_ PMINIDUMP_CALLBACK_INFORMATION CallbackParam
 | |
| );
 | |
| t_miniDumpWriteDump miniDumpWriteDump = 0;
 | |
| 
 | |
| HANDLE _generateDumpFileAtPath(const WCHAR *path) {
 | |
| 	static const int maxFileLen = MAX_PATH * 10;
 | |
| 
 | |
| 	WCHAR szPath[maxFileLen];
 | |
| 	wsprintf(szPath, L"%stdumps\\", path);
 | |
| 
 | |
|     if (!CreateDirectory(szPath, NULL)) {
 | |
| 		DWORD errCode = GetLastError();
 | |
| 		if (errCode && errCode != ERROR_ALREADY_EXISTS) {
 | |
| 			return 0;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
|     WCHAR szFileName[maxFileLen];
 | |
| 	WCHAR szExeName[maxFileLen];
 | |
| 
 | |
| 	wcscpy_s(szExeName, _exeName);
 | |
| 	WCHAR *dotFrom = wcschr(szExeName, WCHAR(L'.'));
 | |
| 	if (dotFrom) {
 | |
| 		wsprintf(dotFrom, L"");
 | |
| 	}
 | |
| 
 | |
|     SYSTEMTIME stLocalTime;
 | |
| 
 | |
|     GetLocalTime(&stLocalTime);
 | |
| 
 | |
|     wsprintf(szFileName, L"%s%s-%s-%04d%02d%02d-%02d%02d%02d-%ld-%ld.dmp", 
 | |
|              szPath, szExeName, AppVersionStr, 
 | |
|              stLocalTime.wYear, stLocalTime.wMonth, stLocalTime.wDay, 
 | |
|              stLocalTime.wHour, stLocalTime.wMinute, stLocalTime.wSecond, 
 | |
|              GetCurrentProcessId(), GetCurrentThreadId());
 | |
|     return CreateFile(szFileName, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_WRITE|FILE_SHARE_READ, 0, CREATE_ALWAYS, 0, 0);
 | |
| }
 | |
| 
 | |
| void _generateDump(EXCEPTION_POINTERS* pExceptionPointers) {
 | |
| 	static const int maxFileLen = MAX_PATH * 10;
 | |
| 
 | |
| 	HMODULE hDll = LoadLibrary(L"DBGHELP.DLL");
 | |
| 	if (!hDll) return;
 | |
| 
 | |
| 	miniDumpWriteDump = (t_miniDumpWriteDump)GetProcAddress(hDll, "MiniDumpWriteDump");
 | |
| 	if (!miniDumpWriteDump) return;
 | |
| 
 | |
| 	HANDLE hDumpFile = 0;
 | |
| 
 | |
| 	WCHAR szPath[maxFileLen];
 | |
| 	DWORD len = GetModuleFileName(GetModuleHandle(0), szPath, maxFileLen);
 | |
| 	if (!len) return;
 | |
| 
 | |
| 	WCHAR *pathEnd = szPath  + len;
 | |
| 
 | |
| 	if (!_wcsicmp(pathEnd - wcslen(_exeName), _exeName)) {
 | |
| 		wsprintf(pathEnd - wcslen(_exeName), L"");
 | |
| 		hDumpFile = _generateDumpFileAtPath(szPath);
 | |
| 	}
 | |
| 	if (!hDumpFile || hDumpFile == INVALID_HANDLE_VALUE) {
 | |
| 		WCHAR wstrPath[maxFileLen];
 | |
| 		DWORD wstrPathLen;
 | |
| 		if (wstrPathLen = GetEnvironmentVariable(L"APPDATA", wstrPath, maxFileLen)) {
 | |
| 			wsprintf(wstrPath + wstrPathLen, L"\\%s\\", _programName);
 | |
| 			hDumpFile = _generateDumpFileAtPath(wstrPath);
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	if (!hDumpFile || hDumpFile == INVALID_HANDLE_VALUE) {
 | |
| 		return;
 | |
| 	}
 | |
| 
 | |
| 	MINIDUMP_EXCEPTION_INFORMATION ExpParam = {0};
 | |
|     ExpParam.ThreadId = GetCurrentThreadId();
 | |
|     ExpParam.ExceptionPointers = pExceptionPointers;
 | |
|     ExpParam.ClientPointers = TRUE;
 | |
| 
 | |
|     miniDumpWriteDump(GetCurrentProcess(), GetCurrentProcessId(), hDumpFile, MiniDumpWithDataSegs, &ExpParam, NULL, NULL);
 | |
| }
 | |
| 
 | |
| LONG CALLBACK _exceptionFilter(EXCEPTION_POINTERS* pExceptionPointers) {
 | |
| 	_generateDump(pExceptionPointers);
 | |
|     return _oldWndExceptionFilter ? (*_oldWndExceptionFilter)(pExceptionPointers) : EXCEPTION_CONTINUE_SEARCH;
 | |
| }
 | |
| #endif
 | 
