mir/kotatogram-desktop
2
0
Fork 0
mirror of https://github.com/kotatogram/kotatogram-desktop synced 2025-09-04 00:25:17 +00:00
Code Issues Releases Wiki Activity

Limit image size the app tries to read.

Browse Source 
This will prevent some OOM crashes.
This commit is contained in:
John Preston
2020-03-23 15:27:07 +04:00
parent d5b8fc703e
commit 2c0b852dad
1 changed files with 24 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
Telegram/SourceFiles/app.cpp
View File

@@ -56,6 +56,9 @@ https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
#endif // OS_MAC_OLD
namespace {
constexpr auto kImageAreaLimit = 6'016 * 3'384;
App::LaunchState _launchState = App::Launched;
HistoryView::Element *hoveredItem = nullptr,
@@ -331,6 +334,13 @@ namespace App {
reader.setAutoTransform(true);
#endif // OS_MAC_OLD
if (animated) *animated = reader.supportsAnimation() && reader.imageCount() > 1;
if (!reader.canRead()) {
return QImage();
}
const auto imageSize = reader.size();
if (imageSize.width() * imageSize.height() > kImageAreaLimit) {
return QImage();
}
QByteArray fmt = reader.format();
if (!fmt.isEmpty()) *format = fmt;
if (!reader.read(&result)) {