/* This file is part of Telegram Desktop, the official desktop application for the Telegram messaging service. For license and copyright information please follow this link: https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL */ #include "data/data_document.h" #include "data/data_session.h" #include "data/data_document_good_thumbnail.h" #include "lang/lang_keys.h" #include "inline_bots/inline_bot_layout_item.h" #include "mainwidget.h" #include "core/file_utilities.h" #include "core/media_active_cache.h" #include "core/mime_type.h" #include "media/audio/media_audio.h" #include "media/player/media_player_instance.h" #include "storage/localstorage.h" #include "platform/platform_specific.h" #include "history/history.h" #include "history/history_item.h" #include "window/window_controller.h" #include "storage/cache/storage_cache_database.h" #include "boxes/confirm_box.h" #include "ui/image/image.h" #include "ui/image/image_source.h" #include "mainwindow.h" #include "core/application.h" #include "media/streaming/media_streaming_loader_mtproto.h" #include "media/streaming/media_streaming_loader_local.h" namespace { constexpr auto kMemoryForCache = 32 * 1024 * 1024; using FilePathResolve = DocumentData::FilePathResolve; Core::MediaActiveCache &ActiveCache() { static auto Instance = Core::MediaActiveCache( kMemoryForCache, [](DocumentData *document) { document->unload(); }); return Instance; } int64 ComputeUsage(StickerData *sticker) { return (sticker != nullptr && sticker->image != nullptr) ? sticker->image->width() * sticker->image->height() * 4 : 0; } QString JoinStringList(const QStringList &list, const QString &separator) { const auto count = list.size(); if (!count) { return QString(); } auto result = QString(); auto fullsize = separator.size() * (count - 1); for (const auto &string : list) { fullsize += string.size(); } result.reserve(fullsize); result.append(list[0]); for (auto i = 1; i != count; ++i) { result.append(separator).append(list[i]); } return result; } void LaunchWithWarning(const QString &name, HistoryItem *item) { const auto warn = [&] { if (!Data::IsExecutableName(name)) { return false; } else if (!Auth().settings().exeLaunchWarning()) { return false; } else if (item && item->history()->peer->isVerified()) { return false; } return true; }(); if (!warn) { File::Launch(name); return; } const auto extension = '.' + Data::FileExtension(name); const auto callback = [=](bool checked) { if (checked) { Auth().settings().setExeLaunchWarning(false); Auth().saveSettingsDelayed(); } File::Launch(name); }; Ui::show(Box( lng_launch_exe_warning( lt_extension, textcmdStartSemibold() + extension + textcmdStopSemibold()), lang(lng_launch_exe_dont_ask), lang(lng_launch_exe_sure), callback)); } } // namespace bool fileIsImage(const QString &name, const QString &mime) { QString lowermime = mime.toLower(), namelower = name.toLower(); if (lowermime.startsWith(qstr("image/"))) { return true; } else if (namelower.endsWith(qstr(".bmp")) || namelower.endsWith(qstr(".jpg")) || namelower.endsWith(qstr(".jpeg")) || namelower.endsWith(qstr(".gif")) || namelower.endsWith(qstr(".webp")) || namelower.endsWith(qstr(".tga")) || namelower.endsWith(qstr(".tiff")) || namelower.endsWith(qstr(".tif")) || namelower.endsWith(qstr(".psd")) || namelower.endsWith(qstr(".png"))) { return true; } return false; } QString FileNameUnsafe( const QString &title, const QString &filter, const QString &prefix, QString name, bool savingAs, const QDir &dir) { #ifdef Q_OS_WIN name = name.replace(QRegularExpression(qsl("[\\\\\\/\\:\\*\\?\\\"\\<\\>\\|]")), qsl("_")); #elif defined Q_OS_MAC name = name.replace(QRegularExpression(qsl("[\\:]")), qsl("_")); #elif defined Q_OS_LINUX name = name.replace(QRegularExpression(qsl("[\\/]")), qsl("_")); #endif if (Global::AskDownloadPath() || savingAs) { if (!name.isEmpty() && name.at(0) == QChar::fromLatin1('.')) { name = filedialogDefaultName(prefix, name); } else if (dir.path() != qsl(".")) { QString path = dir.absolutePath(); if (path != cDialogLastPath()) { cSetDialogLastPath(path); Local::writeUserSettings(); } } // check if extension of filename is present in filter // it should be in first filter section on the first place // place it there, if it is not QString ext = QFileInfo(name).suffix(), fil = filter, sep = qsl(";;"); if (!ext.isEmpty()) { if (QRegularExpression(qsl("^[a-zA-Z_0-9]+$")).match(ext).hasMatch()) { QStringList filters = filter.split(sep); if (filters.size() > 1) { const auto &first = filters.at(0); int32 start = first.indexOf(qsl("(*.")); if (start >= 0) { if (!QRegularExpression(qsl("\\(\\*\\.") + ext + qsl("[\\)\\s]"), QRegularExpression::CaseInsensitiveOption).match(first).hasMatch()) { QRegularExpressionMatch m = QRegularExpression(qsl(" \\*\\.") + ext + qsl("[\\)\\s]"), QRegularExpression::CaseInsensitiveOption).match(first); if (m.hasMatch() && m.capturedStart() > start + 3) { int32 oldpos = m.capturedStart(), oldend = m.capturedEnd(); fil = first.mid(0, start + 3) + ext + qsl(" *.") + first.mid(start + 3, oldpos - start - 3) + first.mid(oldend - 1) + sep + JoinStringList(filters.mid(1), sep); } else { fil = first.mid(0, start + 3) + ext + qsl(" *.") + first.mid(start + 3) + sep + JoinStringList(filters.mid(1), sep); } } } else { fil = QString(); } } else { fil = QString(); } } else { fil = QString(); } } return filedialogGetSaveFile(name, title, fil, name) ? name : QString(); } QString path; if (Global::DownloadPath().isEmpty()) { path = File::DefaultDownloadPath(); } else if (Global::DownloadPath() == qsl("tmp")) { path = cTempDir(); } else { path = Global::DownloadPath(); } if (name.isEmpty()) name = qsl(".unknown"); if (name.at(0) == QChar::fromLatin1('.')) { if (!QDir().exists(path)) QDir().mkpath(path); return filedialogDefaultName(prefix, name, path); } if (dir.path() != qsl(".")) { path = dir.absolutePath() + '/'; } QString nameStart, extension; int32 extPos = name.lastIndexOf('.'); if (extPos >= 0) { nameStart = name.mid(0, extPos); extension = name.mid(extPos); } else { nameStart = name; } QString nameBase = path + nameStart; name = nameBase + extension; for (int i = 0; QFileInfo(name).exists(); ++i) { name = nameBase + QString(" (%1)").arg(i + 2) + extension; } if (!QDir().exists(path)) QDir().mkpath(path); return name; } QString FileNameForSave( const QString &title, const QString &filter, const QString &prefix, QString name, bool savingAs, const QDir &dir) { const auto result = FileNameUnsafe( title, filter, prefix, name, savingAs, dir); #ifdef Q_OS_WIN const auto lower = result.trimmed().toLower(); const auto kBadExtensions = { qstr(".lnk"), qstr(".scf") }; const auto kMaskExtension = qsl(".download"); for (const auto extension : kBadExtensions) { if (lower.endsWith(extension)) { return result + kMaskExtension; } } #endif // Q_OS_WIN return result; } QString documentSaveFilename(const DocumentData *data, bool forceSavingAs = false, const QString already = QString(), const QDir &dir = QDir()) { auto alreadySavingFilename = data->loadingFilePath(); if (!alreadySavingFilename.isEmpty()) { return alreadySavingFilename; } QString name, filter, caption, prefix; const auto mimeType = Core::MimeTypeForName(data->mimeString()); QStringList p = mimeType.globPatterns(); QString pattern = p.isEmpty() ? QString() : p.front(); if (data->isVoiceMessage()) { auto mp3 = data->hasMimeType(qstr("audio/mp3")); name = already.isEmpty() ? (mp3 ? qsl(".mp3") : qsl(".ogg")) : already; filter = mp3 ? qsl("MP3 Audio (*.mp3);;") : qsl("OGG Opus Audio (*.ogg);;"); filter += FileDialog::AllFilesFilter(); caption = lang(lng_save_audio); prefix = qsl("audio"); } else if (data->isVideoFile()) { name = already.isEmpty() ? data->filename() : already; if (name.isEmpty()) { name = pattern.isEmpty() ? qsl(".mov") : pattern.replace('*', QString()); } if (pattern.isEmpty()) { filter = qsl("MOV Video (*.mov);;") + FileDialog::AllFilesFilter(); } else { filter = mimeType.filterString() + qsl(";;") + FileDialog::AllFilesFilter(); } caption = lang(lng_save_video); prefix = qsl("video"); } else { name = already.isEmpty() ? data->filename() : already; if (name.isEmpty()) { name = pattern.isEmpty() ? qsl(".unknown") : pattern.replace('*', QString()); } if (pattern.isEmpty()) { filter = QString(); } else { filter = mimeType.filterString() + qsl(";;") + FileDialog::AllFilesFilter(); } caption = lang(data->isAudioFile() ? lng_save_audio_file : lng_save_file); prefix = qsl("doc"); } return FileNameForSave(caption, filter, prefix, name, forceSavingAs, dir); } void DocumentOpenClickHandler::Open( Data::FileOrigin origin, not_null data, HistoryItem *context) { if (!data->date) { return; } const auto openFile = [&] { const auto &location = data->location(true); if (data->size < App::kImageSizeLimit && location.accessEnable()) { const auto guard = gsl::finally([&] { location.accessDisable(); }); if (QImageReader(location.name()).canRead()) { Core::App().showDocument(data, context); return; } } LaunchWithWarning(location.name(), context); }; const auto &location = data->location(true); if (data->isTheme() && !location.isEmpty() && location.accessEnable()) { Core::App().showDocument(data, context); location.accessDisable(); } else if (data->canBePlayed()) { if (data->isAudioFile() || data->isVoiceMessage() || data->isVideoMessage()) { const auto msgId = context ? context->fullId() : FullMsgId(); Media::Player::instance()->playPause({ data, msgId }); } else if (context && data->isAnimation()) { data->owner().requestAnimationPlayInline(context); } else { Core::App().showDocument(data, context); } } else if (!location.isEmpty() || (data->loaded() && !data->filepath( FilePathResolve::SaveFromDataSilent).isEmpty())) { openFile(); } else if (data->status == FileReady || data->status == FileDownloadFailed) { DocumentSaveClickHandler::Save(origin, data); } } void DocumentOpenClickHandler::onClickImpl() const { Open(context(), document(), getActionItem()); } void DocumentSaveClickHandler::Save( Data::FileOrigin origin, not_null data, Mode mode) { if (!data->date) { return; } auto savename = QString(); if (mode != Mode::ToCacheOrFile || !data->saveToCache()) { const auto filepath = data->filepath(FilePathResolve::Checked); if (mode != Mode::ToNewFile && (!filepath.isEmpty() || !data->filepath( FilePathResolve::SaveFromData).isEmpty())) { return; } const auto fileinfo = QFileInfo(filepath); const auto filedir = filepath.isEmpty() ? QDir() : fileinfo.dir(); const auto filename = filepath.isEmpty() ? QString() : fileinfo.fileName(); savename = documentSaveFilename( data, (mode == Mode::ToNewFile), filename, filedir); if (savename.isEmpty()) { return; } } data->save(origin, savename); } void DocumentSaveClickHandler::onClickImpl() const { Save(context(), document()); } void DocumentCancelClickHandler::onClickImpl() const { const auto data = document(); if (!data->date) return; if (data->uploading()) { if (const auto item = App::histItemById(context())) { App::main()->cancelUploadLayer(item); } } else { data->cancel(); } } void DocumentOpenWithClickHandler::Open( Data::FileOrigin origin, not_null data) { if (!data->date) { return; } if (data->loaded()) { const auto path = data->filepath( FilePathResolve::SaveFromDataSilent); if (!path.isEmpty()) { File::OpenWith(path, QCursor::pos()); return; } } DocumentSaveClickHandler::Save( origin, data, DocumentSaveClickHandler::Mode::ToFile); } void DocumentOpenWithClickHandler::onClickImpl() const { Open(context(), document()); } Data::FileOrigin StickerData::setOrigin() const { return set.match([&](const MTPDinputStickerSetID &data) { return Data::FileOrigin( Data::FileOriginStickerSet(data.vid.v, data.vaccess_hash.v)); }, [&](const auto &) { return Data::FileOrigin(); }); } VoiceData::~VoiceData() { if (!waveform.isEmpty() && waveform[0] == -1 && waveform.size() > int32(sizeof(TaskId))) { TaskId taskId = 0; memcpy(&taskId, waveform.constData() + 1, sizeof(taskId)); Local::cancelTask(taskId); } } DocumentData::DocumentData(not_null owner, DocumentId id) : id(id) , _owner(owner) { } Data::Session &DocumentData::owner() const { return *_owner; } AuthSession &DocumentData::session() const { return _owner->session(); } void DocumentData::setattributes( const QVector &attributes) { _isImage = false; _supportsStreaming = SupportsStreaming::Unknown; for (const auto &attribute : attributes) { attribute.match([&](const MTPDdocumentAttributeImageSize & data) { dimensions = QSize(data.vw.v, data.vh.v); }, [&](const MTPDdocumentAttributeAnimated & data) { if (type == FileDocument || type == StickerDocument || type == VideoDocument) { type = AnimatedDocument; _additional = nullptr; } }, [&](const MTPDdocumentAttributeSticker & data) { if (type == FileDocument) { type = StickerDocument; _additional = std::make_unique(); } if (sticker()) { sticker()->alt = qs(data.valt); if (sticker()->set.type() != mtpc_inputStickerSetID || data.vstickerset.type() == mtpc_inputStickerSetID) { sticker()->set = data.vstickerset; } } }, [&](const MTPDdocumentAttributeVideo & data) { if (type == FileDocument) { type = data.is_round_message() ? RoundVideoDocument : VideoDocument; } _duration = data.vduration.v; setMaybeSupportsStreaming(data.is_supports_streaming()); dimensions = QSize(data.vw.v, data.vh.v); }, [&](const MTPDdocumentAttributeAudio & data) { if (type == FileDocument) { if (data.is_voice()) { type = VoiceDocument; _additional = std::make_unique(); } else { type = SongDocument; _additional = std::make_unique(); } } if (const auto voiceData = voice()) { voiceData->duration = data.vduration.v; voiceData->waveform = documentWaveformDecode( qba(data.vwaveform)); voiceData->wavemax = voiceData->waveform.empty() ? uchar(0) : *ranges::max_element(voiceData->waveform); } else if (const auto songData = song()) { songData->duration = data.vduration.v; songData->title = qs(data.vtitle); songData->performer = qs(data.vperformer); } }, [&](const MTPDdocumentAttributeFilename & data) { _filename = qs(data.vfile_name); // We don't want LTR/RTL mark/embedding/override/isolate chars // in filenames, because they introduce a security issue, when // an executable "Fil[x]gepj.exe" may look like "Filexe.jpeg". QChar controls[] = { 0x200E, // LTR Mark 0x200F, // RTL Mark 0x202A, // LTR Embedding 0x202B, // RTL Embedding 0x202D, // LTR Override 0x202E, // RTL Override 0x2066, // LTR Isolate 0x2067, // RTL Isolate }; for (const auto ch : controls) { _filename = std::move(_filename).replace(ch, "_"); } }, [&](const MTPDdocumentAttributeHasStickers &data) { }); } if (type == StickerDocument) { if (dimensions.width() <= 0 || dimensions.height() <= 0 || dimensions.width() > StickerMaxSize || dimensions.height() > StickerMaxSize || !saveToCache()) { type = FileDocument; _additional = nullptr; } } validateGoodThumbnail(); if (isAudioFile() || (isAnimation() && !isVideoMessage()) || isVoiceMessage()) { setMaybeSupportsStreaming(true); } } bool DocumentData::checkWallPaperProperties() { if (type == WallPaperDocument) { return true; } if (type != FileDocument || !_thumbnail || !dimensions.width() || !dimensions.height() || dimensions.width() > Storage::kMaxWallPaperDimension || dimensions.height() > Storage::kMaxWallPaperDimension || size > Storage::kMaxWallPaperInMemory) { return false; } type = WallPaperDocument; validateGoodThumbnail(); return true; } void DocumentData::updateThumbnails( ImagePtr thumbnailInline, ImagePtr thumbnail) { if (thumbnailInline && !_thumbnailInline) { _thumbnailInline = thumbnailInline; } if (thumbnail && (!_thumbnail || (sticker() && (_thumbnail->width() < thumbnail->width() || _thumbnail->height() < thumbnail->height())))) { _thumbnail = thumbnail; } } bool DocumentData::isWallPaper() const { return (type == WallPaperDocument); } bool DocumentData::isPatternWallPaper() const { return isWallPaper() && hasMimeType(qstr("image/png")); } bool DocumentData::hasThumbnail() const { return !_thumbnail->isNull(); } Image *DocumentData::thumbnailInline() const { return _thumbnailInline ? _thumbnailInline.get() : nullptr; } Image *DocumentData::thumbnail() const { return _thumbnail ? _thumbnail.get() : nullptr; } void DocumentData::loadThumbnail(Data::FileOrigin origin) { if (_thumbnail && !_thumbnail->loaded()) { _thumbnail->load(origin); } } Storage::Cache::Key DocumentData::goodThumbnailCacheKey() const { return Data::DocumentThumbCacheKey(_dc, id); } Image *DocumentData::goodThumbnail() const { return _goodThumbnail.get(); } void DocumentData::validateGoodThumbnail() { if (!isVideoFile() && !isAnimation() && !isWallPaper()) { _goodThumbnail = nullptr; } else if (!_goodThumbnail && hasRemoteLocation()) { _goodThumbnail = std::make_unique( std::make_unique(this)); } } void DocumentData::refreshGoodThumbnail() { if (_goodThumbnail && hasRemoteLocation()) { replaceGoodThumbnail(std::make_unique(this)); } } void DocumentData::replaceGoodThumbnail( std::unique_ptr &&source) { _goodThumbnail->replaceSource(std::move(source)); } void DocumentData::setGoodThumbnailOnUpload( QImage &&image, QByteArray &&bytes) { Expects(uploadingData != nullptr); if (image.isNull()) { return; } _goodThumbnail = std::make_unique( std::make_unique( QString(), std::move(bytes), "JPG", std::move(image))); } bool DocumentData::saveToCache() const { return (type == StickerDocument && size < Storage::kMaxStickerInMemory) || (isAnimation() && size < Storage::kMaxAnimationInMemory) || (isVoiceMessage() && size < Storage::kMaxVoiceInMemory) || (type == WallPaperDocument); } void DocumentData::unload() { // Forget thumb only when image cache limit exceeds. // // Also, you can't unload() images that you don't own // from the destructor, because they're already destroyed. // //_thumbnailInline->unload(); //_thumbnail->unload(); if (sticker()) { if (sticker()->image) { ActiveCache().decrement(ComputeUsage(sticker())); sticker()->image = nullptr; } } _replyPreview.clear(); if (!_data.isEmpty()) { ActiveCache().decrement(_data.size()); _data.clear(); } } void DocumentData::automaticLoad( Data::FileOrigin origin, const HistoryItem *item) { if (status != FileReady || loaded() || cancelled()) { return; } else if (!item && type != StickerDocument && !isAnimation()) { return; } const auto toCache = saveToCache(); if (!toCache && Global::AskDownloadPath()) { // We need a filename, but we're supposed to ask user for it. // No automatic download in this case. return; } const auto filename = toCache ? QString() : documentSaveFilename(this); const auto shouldLoadFromCloud = !Data::IsExecutableName(filename) && (item ? Data::AutoDownload::Should( Auth().settings().autoDownload(), item->history()->peer, this) : Data::AutoDownload::Should( Auth().settings().autoDownload(), this)); const auto loadFromCloud = shouldLoadFromCloud ? LoadFromCloudOrLocal : LoadFromLocalOnly; save( origin, filename, loadFromCloud, true); } void DocumentData::automaticLoadSettingsChanged() { if (!cancelled() || status != FileReady || loaded()) { return; } _loader = nullptr; } bool DocumentData::loaded(FilePathResolve resolve) const { if (loading() && _loader->finished()) { if (_loader->cancelled()) { destroyLoader(CancelledMtpFileLoader); } else { auto that = const_cast(this); that->_location = FileLocation(_loader->fileName()); ActiveCache().decrement(that->_data.size()); that->_data = _loader->bytes(); ActiveCache().increment(that->_data.size()); if (that->sticker() && !that->sticker()->image && !_loader->imageData().isNull()) { that->sticker()->image = std::make_unique( std::make_unique( QString(), _data, _loader->imageFormat(), _loader->imageData())); ActiveCache().increment(ComputeUsage(that->sticker())); } that->refreshGoodThumbnail(); destroyLoader(); if (!that->_data.isEmpty() || that->getStickerLarge()) { ActiveCache().up(that); } } _owner->notifyDocumentLayoutChanged(this); } return !data().isEmpty() || !filepath(resolve).isEmpty(); } void DocumentData::destroyLoader(mtpFileLoader *newValue) const { const auto loader = std::exchange(_loader, newValue); if (cancelled()) { loader->cancel(); } loader->stop(); delete loader; } bool DocumentData::loading() const { return _loader && !cancelled(); } QString DocumentData::loadingFilePath() const { return loading() ? _loader->fileName() : QString(); } bool DocumentData::displayLoading() const { return loading() ? (!_loader->loadingLocal() || !_loader->autoLoading()) : (uploading() && !waitingForAlbum()); } float64 DocumentData::progress() const { if (uploading()) { if (uploadingData->size > 0) { const auto result = float64(uploadingData->offset) / uploadingData->size; return snap(result, 0., 1.); } return 0.; } return loading() ? _loader->currentProgress() : (loaded() ? 1. : 0.); } int32 DocumentData::loadOffset() const { return loading() ? _loader->currentOffset() : 0; } bool DocumentData::uploading() const { return (uploadingData != nullptr); } void DocumentData::setWaitingForAlbum() { if (uploading()) { uploadingData->waitingForAlbum = true; } } bool DocumentData::waitingForAlbum() const { return uploading() && uploadingData->waitingForAlbum; } void DocumentData::save( Data::FileOrigin origin, const QString &toFile, LoadFromCloudSetting fromCloud, bool autoLoading) { if (loaded(FilePathResolve::Checked)) { auto &l = location(true); if (!toFile.isEmpty()) { if (!_data.isEmpty()) { QFile f(toFile); f.open(QIODevice::WriteOnly); f.write(_data); f.close(); setLocation(FileLocation(toFile)); Local::writeFileLocation(mediaKey(), FileLocation(toFile)); } else if (l.accessEnable()) { const auto &alreadyName = l.name(); if (alreadyName != toFile) { QFile(toFile).remove(); QFile(alreadyName).copy(toFile); } l.accessDisable(); } } return; } if (cancelled()) { _loader = nullptr; } if (_loader) { if (!_loader->setFileName(toFile)) { cancel(); // changes _actionOnLoad _loader = nullptr; } } if (_loader) { if (fromCloud == LoadFromCloudOrLocal) { _loader->permitLoadFromCloud(); } } else { status = FileReady; if (hasWebLocation()) { _loader = new mtpFileLoader( _urlLocation, size, fromCloud, autoLoading, cacheTag()); } else if (!_access && !_url.isEmpty()) { _loader = new webFileLoader( _url, toFile, fromCloud, autoLoading, cacheTag()); } else { _loader = new mtpFileLoader( StorageFileLocation( _dc, session().userId(), MTP_inputDocumentFileLocation( MTP_long(id), MTP_long(_access), MTP_bytes(_fileReference), MTP_string(QString()))), origin, locationType(), toFile, size, (saveToCache() ? LoadToCacheAsWell : LoadToFileOnly), fromCloud, autoLoading, cacheTag()); } _loader->connect(_loader, SIGNAL(progress(FileLoader*)), App::main(), SLOT(documentLoadProgress(FileLoader*))); _loader->connect(_loader, SIGNAL(failed(FileLoader*,bool)), App::main(), SLOT(documentLoadFailed(FileLoader*,bool))); } if (loading()) { _loader->start(); } _owner->notifyDocumentLayoutChanged(this); } void DocumentData::cancel() { if (!loading()) { return; } destroyLoader(CancelledMtpFileLoader); _owner->notifyDocumentLayoutChanged(this); App::main()->documentLoadProgress(this); } bool DocumentData::cancelled() const { return (_loader == CancelledMtpFileLoader); } VoiceWaveform documentWaveformDecode(const QByteArray &encoded5bit) { auto bitsCount = static_cast(encoded5bit.size() * 8); auto valuesCount = bitsCount / 5; if (!valuesCount) { return VoiceWaveform(); } // Read each 5 bit of encoded5bit as 0-31 unsigned char. // We count the index of the byte in which the desired 5-bit sequence starts. // And then we read a uint16 starting from that byte to guarantee to get all of those 5 bits. // // BUT! if it is the last byte we have, we're not allowed to read a uint16 starting with it. // Because it will be an overflow (we'll access one byte after the available memory). // We see, that only the last 5 bits could start in the last available byte and be problematic. // So we read in a general way all the entries in a general way except the last one. auto result = VoiceWaveform(valuesCount, 0); auto bitsData = encoded5bit.constData(); for (auto i = 0, l = valuesCount - 1; i != l; ++i) { auto byteIndex = (i * 5) / 8; auto bitShift = (i * 5) % 8; auto value = *reinterpret_cast(bitsData + byteIndex); result[i] = static_cast((value >> bitShift) & 0x1F); } auto lastByteIndex = ((valuesCount - 1) * 5) / 8; auto lastBitShift = ((valuesCount - 1) * 5) % 8; auto lastValue = (lastByteIndex == encoded5bit.size() - 1) ? static_cast(*reinterpret_cast(bitsData + lastByteIndex)) : *reinterpret_cast(bitsData + lastByteIndex); result[valuesCount - 1] = static_cast((lastValue >> lastBitShift) & 0x1F); return result; } QByteArray documentWaveformEncode5bit(const VoiceWaveform &waveform) { auto bitsCount = waveform.size() * 5; auto bytesCount = (bitsCount + 7) / 8; auto result = QByteArray(bytesCount + 1, 0); auto bitsData = result.data(); // Write each 0-31 unsigned char as 5 bit to result. // We reserve one extra byte to be able to dereference any of required bytes // as a uint16 without overflowing, even the byte with index "bytesCount - 1". for (auto i = 0, l = waveform.size(); i < l; ++i) { auto byteIndex = (i * 5) / 8; auto bitShift = (i * 5) % 8; auto value = (static_cast(waveform[i]) & 0x1F) << bitShift; *reinterpret_cast(bitsData + byteIndex) |= value; } result.resize(bytesCount); return result; } QByteArray DocumentData::data() const { if (!_data.isEmpty()) { ActiveCache().up(const_cast(this)); } return _data; } const FileLocation &DocumentData::location(bool check) const { if (check && !_location.check()) { const_cast(this)->_location = Local::readFileLocation(mediaKey()); } return _location; } void DocumentData::setLocation(const FileLocation &loc) { if (loc.check()) { _location = loc; } } QString DocumentData::filepath(FilePathResolve resolve) const { bool check = (resolve != FilePathResolve::Cached); QString result = (check && _location.name().isEmpty()) ? QString() : location(check).name(); bool saveFromData = result.isEmpty() && !data().isEmpty(); if (saveFromData) { if (resolve != FilePathResolve::SaveFromData && resolve != FilePathResolve::SaveFromDataSilent) { saveFromData = false; } else if (resolve == FilePathResolve::SaveFromDataSilent && Global::AskDownloadPath()) { saveFromData = false; } } if (saveFromData) { QString filename = documentSaveFilename(this); if (!filename.isEmpty()) { QFile f(filename); if (f.open(QIODevice::WriteOnly)) { if (f.write(data()) == data().size()) { f.close(); const_cast(this)->_location = FileLocation(filename); Local::writeFileLocation(mediaKey(), _location); result = filename; } } } } return result; } bool DocumentData::isStickerSetInstalled() const { Expects(sticker() != nullptr); const auto &set = sticker()->set; const auto &sets = _owner->stickerSets(); switch (set.type()) { case mtpc_inputStickerSetID: { auto it = sets.constFind(set.c_inputStickerSetID().vid.v); return (it != sets.cend()) && !(it->flags & MTPDstickerSet::Flag::f_archived) && (it->flags & MTPDstickerSet::Flag::f_installed_date); } break; case mtpc_inputStickerSetShortName: { auto name = qs(set.c_inputStickerSetShortName().vshort_name).toLower(); for (auto it = sets.cbegin(), e = sets.cend(); it != e; ++it) { if (it->shortName.toLower() == name) { return !(it->flags & MTPDstickerSet::Flag::f_archived) && (it->flags & MTPDstickerSet::Flag::f_installed_date); } } } break; } return false; } Image *DocumentData::getReplyPreview(Data::FileOrigin origin) { if (!_thumbnail) { return nullptr; } else if (_replyPreview && (_replyPreview.good() || !_thumbnail->loaded())) { return _replyPreview.image(); } const auto option = isVideoMessage() ? Images::Option::Circled : Images::Option::None; if (_thumbnail->loaded()) { _replyPreview.prepare( _thumbnail.get(), origin, option); } else { _thumbnail->load(origin); if (_thumbnailInline) { _replyPreview.prepare( _thumbnailInline.get(), origin, option | Images::Option::Blurred); } } return _replyPreview.image(); } StickerData *DocumentData::sticker() const { return (type == StickerDocument) ? static_cast(_additional.get()) : nullptr; } void DocumentData::checkStickerLarge() { const auto data = sticker(); if (!data) return; automaticLoad(stickerSetOrigin(), nullptr); if (!data->image && loaded()) { if (_data.isEmpty()) { const auto &loc = location(true); if (loc.accessEnable()) { data->image = std::make_unique( std::make_unique(loc.name())); loc.accessDisable(); } } else { auto format = QByteArray(); auto image = App::readImage(_data, &format, false); data->image = std::make_unique( std::make_unique( QString(), _data, format, std::move(image))); } if (const auto usage = ComputeUsage(data)) { ActiveCache().increment(usage); ActiveCache().up(this); } } } void DocumentData::checkStickerSmall() { if (thumbnailEnoughForSticker()) { _thumbnail->load(stickerSetOrigin()); } else { checkStickerLarge(); } } Image *DocumentData::getStickerLarge() { checkStickerLarge(); if (const auto data = sticker()) { return data->image.get(); } return nullptr; } Image *DocumentData::getStickerSmall() { if (thumbnailEnoughForSticker()) { return _thumbnail->isNull() ? nullptr : _thumbnail.get(); } else if (const auto data = sticker()) { return data->image.get(); } return nullptr; } Data::FileOrigin DocumentData::stickerSetOrigin() const { if (const auto data = sticker()) { if (const auto result = data->setOrigin()) { return result; } else if (Stickers::IsFaved(this)) { return Data::FileOriginStickerSet(Stickers::FavedSetId, 0); } } return Data::FileOrigin(); } Data::FileOrigin DocumentData::stickerOrGifOrigin() const { return (sticker() ? stickerSetOrigin() : isGifv() ? Data::FileOriginSavedGifs() : Data::FileOrigin()); } SongData *DocumentData::song() { return isSong() ? static_cast(_additional.get()) : nullptr; } const SongData *DocumentData::song() const { return const_cast(this)->song(); } VoiceData *DocumentData::voice() { return isVoiceMessage() ? static_cast(_additional.get()) : nullptr; } const VoiceData *DocumentData::voice() const { return const_cast(this)->voice(); } bool DocumentData::hasRemoteLocation() const { return (_dc != 0 && _access != 0); } bool DocumentData::canBeStreamed() const { // For now video messages are not streamed. return hasRemoteLocation() && supportsStreaming() && !isVideoMessage(); } bool DocumentData::canBePlayed() const { return !_inappPlaybackFailed && (isAnimation() || isVideoFile() || isAudioFile() || isVoiceMessage()) && (loaded() || canBeStreamed()); } void DocumentData::setInappPlaybackFailed() { _inappPlaybackFailed = true; } bool DocumentData::inappPlaybackFailed() const { return _inappPlaybackFailed; } auto DocumentData::createStreamingLoader(Data::FileOrigin origin) const -> std::unique_ptr { const auto &location = this->location(true); if (!data().isEmpty()) { return Media::Streaming::MakeBytesLoader(data()); } else if (!location.isEmpty() && location.accessEnable()) { auto result = Media::Streaming::MakeFileLoader(location.name()); location.accessDisable(); return result; } return hasRemoteLocation() ? std::make_unique( &session().api(), StorageFileLocation( _dc, session().userId(), MTP_inputDocumentFileLocation( MTP_long(id), MTP_long(_access), MTP_bytes(_fileReference), MTP_string(QString()))), size, origin) : nullptr; } bool DocumentData::hasWebLocation() const { return !_urlLocation.url().isEmpty(); } bool DocumentData::isNull() const { return !hasRemoteLocation() && !hasWebLocation() && _url.isEmpty(); } MTPInputDocument DocumentData::mtpInput() const { if (_access) { return MTP_inputDocument( MTP_long(id), MTP_long(_access), MTP_bytes(_fileReference)); } return MTP_inputDocumentEmpty(); } QByteArray DocumentData::fileReference() const { return _fileReference; } void DocumentData::refreshFileReference(const QByteArray &value) { _fileReference = value; _thumbnail->refreshFileReference(value); if (const auto data = sticker()) { data->loc.refreshFileReference(value); } } void DocumentData::refreshStickerThumbFileReference() { if (const auto data = sticker()) { if (_thumbnail->loading()) { data->loc.refreshFileReference( _thumbnail->location().fileReference()); } } } QString DocumentData::filename() const { return _filename; } QString DocumentData::mimeString() const { return _mimeString; } bool DocumentData::hasMimeType(QLatin1String mime) const { return !_mimeString.compare(mime, Qt::CaseInsensitive); } void DocumentData::setMimeString(const QString &mime) { _mimeString = mime; } MediaKey DocumentData::mediaKey() const { return ::mediaKey(locationType(), _dc, id); } Storage::Cache::Key DocumentData::cacheKey() const { if (hasWebLocation()) { return Data::WebDocumentCacheKey(_urlLocation); } else if (!_access && !_url.isEmpty()) { return Data::UrlCacheKey(_url); } else { return Data::DocumentCacheKey(_dc, id); } } uint8 DocumentData::cacheTag() const { if (type == StickerDocument) { return Data::kStickerCacheTag; } else if (isVoiceMessage()) { return Data::kVoiceMessageCacheTag; } else if (isVideoMessage()) { return Data::kVideoMessageCacheTag; } else if (isAnimation()) { return Data::kAnimationCacheTag; } else if (type == WallPaperDocument) { return Data::kImageCacheTag; } return 0; } QString DocumentData::composeNameString() const { if (auto songData = song()) { return ComposeNameString( _filename, songData->title, songData->performer); } return ComposeNameString(_filename, QString(), QString()); } LocationType DocumentData::locationType() const { return isVoiceMessage() ? AudioFileLocation : isVideoFile() ? VideoFileLocation : DocumentFileLocation; } bool DocumentData::isVoiceMessage() const { return (type == VoiceDocument); } bool DocumentData::isVideoMessage() const { return (type == RoundVideoDocument); } bool DocumentData::isAnimation() const { return (type == AnimatedDocument) || isVideoMessage() || (hasMimeType(qstr("image/gif")) && !_inappPlaybackFailed); } bool DocumentData::isGifv() const { return (type == AnimatedDocument) && hasMimeType(qstr("video/mp4")); } bool DocumentData::isTheme() const { return _filename.endsWith( qstr(".tdesktop-theme"), Qt::CaseInsensitive) || _filename.endsWith( qstr(".tdesktop-palette"), Qt::CaseInsensitive); } bool DocumentData::isSong() const { return (type == SongDocument); } bool DocumentData::isAudioFile() const { if (isVoiceMessage()) { return false; } else if (isSong()) { return true; } const auto prefix = qstr("audio/"); if (!_mimeString.startsWith(prefix, Qt::CaseInsensitive)) { if (_filename.endsWith(qstr(".opus"), Qt::CaseInsensitive)) { return true; } return false; } const auto left = _mimeString.midRef(prefix.size()).toString(); const auto types = { qstr("x-wav"), qstr("wav"), qstr("mp4") }; return ranges::find(types, left) != end(types); } bool DocumentData::isSharedMediaMusic() const { if (const auto songData = song()) { return (songData->duration > 0); } return false; } bool DocumentData::isVideoFile() const { return (type == VideoDocument); } TimeId DocumentData::getDuration() const { if (const auto song = this->song()) { return std::max(song->duration, 0); } else if (const auto voice = this->voice()) { return std::max(voice->duration, 0); } else if (isAnimation() || isVideoFile()) { return std::max(_duration, 0); } return -1; } bool DocumentData::isImage() const { return _isImage; } bool DocumentData::supportsStreaming() const { return (_supportsStreaming == SupportsStreaming::MaybeYes); } void DocumentData::setNotSupportsStreaming() { _supportsStreaming = SupportsStreaming::No; } void DocumentData::setMaybeSupportsStreaming(bool supports) { if (_supportsStreaming == SupportsStreaming::No) { return; } _supportsStreaming = supports ? SupportsStreaming::MaybeYes : SupportsStreaming::MaybeNo; } void DocumentData::recountIsImage() { _isImage = !isAnimation() && !isVideoFile() && fileIsImage(filename(), mimeString()); } bool DocumentData::thumbnailEnoughForSticker() const { return !_thumbnail->isNull() && ((_thumbnail->width() >= 128) || (_thumbnail->height() >= 128)); } void DocumentData::setRemoteLocation( int32 dc, uint64 access, const QByteArray &fileReference) { _fileReference = fileReference; if (_dc != dc || _access != access) { _dc = dc; _access = access; if (!isNull()) { if (_location.check()) { Local::writeFileLocation(mediaKey(), _location); } else { _location = Local::readFileLocation(mediaKey()); } } } validateGoodThumbnail(); } void DocumentData::setContentUrl(const QString &url) { _url = url; } void DocumentData::setWebLocation(const WebFileLocation &location) { _urlLocation = location; } void DocumentData::collectLocalData(not_null local) { if (local == this) { return; } _owner->cache().copyIfEmpty(local->cacheKey(), cacheKey()); if (!local->_data.isEmpty()) { ActiveCache().decrement(_data.size()); _data = local->_data; ActiveCache().increment(_data.size()); if (!_data.isEmpty()) { ActiveCache().up(this); } } if (!local->_location.isEmpty()) { _location = local->_location; Local::writeFileLocation(mediaKey(), _location); } } DocumentData::~DocumentData() { if (loading()) { destroyLoader(); } unload(); ActiveCache().remove(this); } QString DocumentData::ComposeNameString( const QString &filename, const QString &songTitle, const QString &songPerformer) { if (songTitle.isEmpty() && songPerformer.isEmpty()) { return filename.isEmpty() ? qsl("Unknown File") : filename; } if (songPerformer.isEmpty()) { return songTitle; } auto trackTitle = (songTitle.isEmpty() ? qsl("Unknown Track") : songTitle); return songPerformer + QString::fromUtf8(" \xe2\x80\x93 ") + trackTitle; } namespace Data { QString FileExtension(const QString &filepath) { const auto reversed = ranges::view::reverse(filepath); const auto last = ranges::find_first_of(reversed, ".\\/"); if (last == reversed.end() || *last != '.') { return QString(); } return QString(last.base(), last - reversed.begin()); } bool IsValidMediaFile(const QString &filepath) { static const auto kExtensions = [] { const auto list = qsl("\ 16svx 2sf 3g2 3gp 8svx aac aaf aif aifc aiff amr amv ape asf ast au aup \ avchd avi brstm bwf cam cdda cust dat divx drc dsh dsf dts dtshd dtsma \ dvr-ms dwd evo f4a f4b f4p f4v fla flac flr flv gif gifv gsf gsm gym iff \ ifo it jam la ly m1v m2p m2ts m2v m4a m4p m4v mcf mid mk3d mka mks mkv mng \ mov mp1 mp2 mp3 mp4 minipsf mod mpc mpe mpeg mpg mpv mscz mt2 mus mxf mxl \ niff nsf nsv off ofr ofs ogg ogv opus ots pac ps psf psf2 psflib ptb qsf \ qt ra raw rka rm rmj rmvb roq s3m shn sib sid smi smp sol spc spx ssf svi \ swa swf tak ts tta txm usf vgm vob voc vox vqf wav webm wma wmv wrap wtv \ wv xm xml ym yuv").split(' '); return base::flat_set(list.begin(), list.end()); }(); return ranges::binary_search( kExtensions, FileExtension(filepath).toLower()); } bool IsExecutableName(const QString &filepath) { static const auto kExtensions = [] { const auto joined = #ifdef Q_OS_MAC qsl("\ action app bin command csh osx workflow terminal url caction mpkg pkg xhtm \ webarchive"); #elif defined Q_OS_LINUX // Q_OS_MAC qsl("bin csh deb desktop ksh out pet pkg pup rpm run shar slp"); #else // Q_OS_MAC || Q_OS_LINUX qsl("\ ad ade adp app application appref-ms asp asx bas bat bin cer cfg chi chm \ cmd cnt com cpl crt csh der diagcab dll drv eml exe fon fxp gadget grp hlp \ hpj hta htt inf ini ins inx isp isu its jar jnlp job js jse ksh lnk local \ mad maf mag mam manifest maq mar mas mat mau mav maw mcf mda mdb mde mdt \ mdw mdz mht mhtml mmc mof msc msg msh msh1 msh2 msh1xml msh2xml mshxml msi \ msp mst ops osd paf pcd pif pl plg prf prg ps1 ps2 ps1xml ps2xml psc1 psc2 \ pst reg rgs scf scr sct search-ms settingcontent-ms shb shs slk sys tmp \ u3p url vb vbe vbp vbs vbscript vdx vsmacros vsd vsdm vsdx vss vssm vssx \ vst vstm vstx vsw vsx vtx website ws wsc wsf wsh xbap xll xnk"); #endif // !Q_OS_MAC && !Q_OS_LINUX const auto list = joined.split(' '); return base::flat_set(list.begin(), list.end()); }(); return ranges::binary_search( kExtensions, FileExtension(filepath).toLower()); } base::binary_guard ReadImageAsync( not_null document, FnMut postprocess, FnMut done) { auto result = base::binary_guard(); crl::async([ bytes = document->data(), path = document->filepath(), postprocess = std::move(postprocess), guard = result.make_guard(), callback = std::move(done) ]() mutable { auto format = QByteArray(); if (bytes.isEmpty()) { QFile f(path); if (f.size() <= App::kImageSizeLimit && f.open(QIODevice::ReadOnly)) { bytes = f.readAll(); } } auto image = bytes.isEmpty() ? QImage() : App::readImage(bytes, &format, false, nullptr); if (postprocess) { image = postprocess(std::move(image)); } crl::on_main(std::move(guard), [ image = std::move(image), callback = std::move(callback) ]() mutable { callback(std::move(image)); }); }); return result; } //void HandleUnsupportedMedia( // not_null document, // FullMsgId contextId) { // using Error = ::Media::Streaming::Error; // // document->setInappPlaybackFailed(); // const auto filepath = document->filepath( // DocumentData::FilePathResolve::SaveFromData); // if (filepath.isEmpty()) { // const auto save = [=] { // Ui::hideLayer(); // DocumentSaveClickHandler::Save( // (contextId ? contextId : Data::FileOrigin()), // document, // App::histItemById(contextId)); // }; // Ui::show(Box( // lang(lng_player_cant_stream), // lang(lng_player_download), // lang(lng_cancel), // save)); // } else if (IsValidMediaFile(filepath)) { // File::Launch(filepath); // } //} } // namespace Data