1033 lines
28 KiB
C++
1033 lines
28 KiB
C++
/*
|
|
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 "ui/image/image_location.h"
|
|
|
|
#include "ui/image/image.h"
|
|
#include "platform/platform_specific.h"
|
|
#include "storage/cache/storage_cache_types.h"
|
|
#include "storage/serialize_common.h"
|
|
#include "data/data_file_origin.h"
|
|
#include "base/overload.h"
|
|
#include "main/main_session.h"
|
|
|
|
#include <QtCore/QBuffer>
|
|
|
|
namespace {
|
|
|
|
constexpr auto kDocumentBaseCacheTag = 0x0000000000010000ULL;
|
|
constexpr auto kDocumentBaseCacheMask = 0x000000000000FF00ULL;
|
|
constexpr auto kPhotoBaseCacheTag = 0x0000000000020000ULL;
|
|
constexpr auto kPhotoBaseCacheMask = 0x000000000000FF00ULL;
|
|
|
|
constexpr auto kNonStorageLocationToken = quint8(0x10);
|
|
constexpr auto kLegacyInMessagePeerIdFlag = quint8(0x08);
|
|
constexpr auto kModernLocationFlag = quint8(0x20);
|
|
constexpr auto kInMessageFieldsFlag = quint8(0x40);
|
|
|
|
enum class NonStorageLocationType : quint8 {
|
|
Web,
|
|
Geo,
|
|
Url,
|
|
Memory,
|
|
};
|
|
|
|
MTPInputPeer GenerateInputPeer(
|
|
PeerId id,
|
|
uint64 accessHash,
|
|
PeerId inMessagePeerId,
|
|
int32 inMessageId,
|
|
UserId self) {
|
|
const auto bareId = [&] {
|
|
return peerToBareMTPInt(id);
|
|
};
|
|
if (inMessageId && peerIsUser(inMessagePeerId)) {
|
|
return MTP_inputPeerUserFromMessage(
|
|
GenerateInputPeer(id, accessHash, 0, 0, self),
|
|
MTP_int(inMessageId),
|
|
MTP_int(peerToUser(inMessagePeerId).bare)); // #TODO ids
|
|
} else if (inMessageId && peerIsChannel(inMessagePeerId)) {
|
|
return MTP_inputPeerChannelFromMessage(
|
|
GenerateInputPeer(id, accessHash, 0, 0, self),
|
|
MTP_int(inMessageId),
|
|
MTP_int(peerToChannel(inMessagePeerId).bare)); // #TODO ids
|
|
} else if (!id) {
|
|
return MTP_inputPeerEmpty();
|
|
} else if (id == peerFromUser(self)) {
|
|
return MTP_inputPeerSelf();
|
|
} else if (peerIsUser(id)) {
|
|
return MTP_inputPeerUser(bareId(), MTP_long(accessHash));
|
|
} else if (peerIsChat(id)) {
|
|
return MTP_inputPeerChat(bareId());
|
|
} else if (peerIsChannel(id)) {
|
|
return MTP_inputPeerChannel(bareId(), MTP_long(accessHash));
|
|
} else {
|
|
return MTP_inputPeerEmpty();
|
|
}
|
|
}
|
|
|
|
} // namespace
|
|
|
|
WebFileLocation WebFileLocation::Null;
|
|
|
|
StorageFileLocation::StorageFileLocation(
|
|
int32 dcId,
|
|
UserId self,
|
|
const MTPInputFileLocation &tl)
|
|
: _dcId(dcId) {
|
|
tl.match([&](const MTPDinputFileLocation &data) {
|
|
_type = Type::Legacy;
|
|
_volumeId = data.vvolume_id().v;
|
|
_localId = data.vlocal_id().v;
|
|
_accessHash = data.vsecret().v;
|
|
_fileReference = data.vfile_reference().v;
|
|
}, [&](const MTPDinputEncryptedFileLocation &data) {
|
|
_type = Type::Encrypted;
|
|
_id = data.vid().v;
|
|
_accessHash = data.vaccess_hash().v;
|
|
}, [&](const MTPDinputDocumentFileLocation &data) {
|
|
_type = Type::Document;
|
|
_id = data.vid().v;
|
|
_accessHash = data.vaccess_hash().v;
|
|
_fileReference = data.vfile_reference().v;
|
|
_sizeLetter = data.vthumb_size().v.isEmpty()
|
|
? uint8(0)
|
|
: uint8(data.vthumb_size().v[0]);
|
|
}, [&](const MTPDinputSecureFileLocation &data) {
|
|
_type = Type::Secure;
|
|
_id = data.vid().v;
|
|
_accessHash = data.vaccess_hash().v;
|
|
}, [&](const MTPDinputTakeoutFileLocation &data) {
|
|
_type = Type::Takeout;
|
|
}, [&](const MTPDinputPhotoFileLocation &data) {
|
|
_type = Type::Photo;
|
|
_id = data.vid().v;
|
|
_accessHash = data.vaccess_hash().v;
|
|
_fileReference = data.vfile_reference().v;
|
|
_sizeLetter = data.vthumb_size().v.isEmpty()
|
|
? char(0)
|
|
: data.vthumb_size().v[0];
|
|
}, [&](const MTPDinputPhotoLegacyFileLocation &data) {
|
|
_type = Type::Legacy;
|
|
_volumeId = data.vvolume_id().v;
|
|
_localId = data.vlocal_id().v;
|
|
_accessHash = data.vsecret().v;
|
|
_fileReference = data.vfile_reference().v;
|
|
}, [&](const MTPDinputPeerPhotoFileLocation &data) {
|
|
_type = Type::PeerPhoto;
|
|
const auto fillPeer = base::overload([&](
|
|
const MTPDinputPeerEmpty &data) {
|
|
_id = 0;
|
|
}, [&](const MTPDinputPeerSelf &data) {
|
|
_id = peerFromUser(self).value;
|
|
}, [&](const MTPDinputPeerChat &data) {
|
|
_id = peerFromChat(data.vchat_id()).value;
|
|
}, [&](const MTPDinputPeerUser &data) {
|
|
_id = peerFromUser(data.vuser_id()).value;
|
|
_accessHash = data.vaccess_hash().v;
|
|
}, [&](const MTPDinputPeerChannel &data) {
|
|
_id = peerFromChannel(data.vchannel_id()).value;
|
|
_accessHash = data.vaccess_hash().v;
|
|
});
|
|
data.vpeer().match(fillPeer, [&](
|
|
const MTPDinputPeerUserFromMessage &data) {
|
|
data.vpeer().match(fillPeer, [&](auto &&) {
|
|
// Bad data provided.
|
|
_id = _accessHash = 0;
|
|
});
|
|
_inMessagePeerId = peerFromUser(data.vuser_id());
|
|
_inMessageId = data.vmsg_id().v;
|
|
}, [&](const MTPDinputPeerChannelFromMessage &data) {
|
|
data.vpeer().match(fillPeer, [&](auto &&) {
|
|
// Bad data provided.
|
|
_id = _accessHash = 0;
|
|
});
|
|
_inMessagePeerId = peerFromChannel(data.vchannel_id());
|
|
_inMessageId = data.vmsg_id().v;
|
|
});
|
|
_volumeId = data.vphoto_id().v;
|
|
_sizeLetter = data.is_big() ? 'c' : 'a';
|
|
|
|
// _localId place is used in serialization.
|
|
Ensures(_localId == 0);
|
|
}, [&](const MTPDinputStickerSetThumb &data) {
|
|
_type = Type::StickerSetThumb;
|
|
data.vstickerset().match([&](const MTPDinputStickerSetEmpty &data) {
|
|
_id = 0;
|
|
}, [&](const MTPDinputStickerSetID &data) {
|
|
_id = data.vid().v;
|
|
_accessHash = data.vaccess_hash().v;
|
|
}, [&](const auto &data) {
|
|
Unexpected("InputStickerSet type in StorageFileLocation.");
|
|
});
|
|
_volumeId = 0;
|
|
_localId = data.vthumb_version().v;
|
|
}, [&](const MTPDinputGroupCallStream &data) {
|
|
_type = Type::GroupCallStream;
|
|
data.vcall().match([&](const MTPDinputGroupCall &data) {
|
|
_id = data.vid().v;
|
|
_accessHash = data.vaccess_hash().v;
|
|
});
|
|
_volumeId = data.vtime_ms().v;
|
|
_localId = data.vscale().v;
|
|
});
|
|
}
|
|
|
|
StorageFileLocation StorageFileLocation::convertToModernPeerPhoto(
|
|
uint64 id,
|
|
uint64 accessHash,
|
|
uint64 photoId) const {
|
|
if (_type != Type::Legacy && _type != Type::PeerPhoto) {
|
|
return *this;
|
|
} else if (!photoId) {
|
|
return StorageFileLocation();
|
|
}
|
|
|
|
auto result = *this;
|
|
result._type = Type::PeerPhoto;
|
|
result._id = id;
|
|
result._accessHash = accessHash;
|
|
result._sizeLetter = uint8('a');
|
|
result._volumeId = photoId;
|
|
result._localId = 0;
|
|
result._inMessagePeerId = 0;
|
|
result._inMessageId = 0;
|
|
return result;
|
|
}
|
|
|
|
int32 StorageFileLocation::dcId() const {
|
|
return _dcId;
|
|
}
|
|
|
|
uint64 StorageFileLocation::objectId() const {
|
|
return _id;
|
|
}
|
|
|
|
MTPInputFileLocation StorageFileLocation::tl(UserId self) const {
|
|
switch (_type) {
|
|
case Type::Legacy:
|
|
return MTP_inputFileLocation(
|
|
MTP_long(_volumeId),
|
|
MTP_int(_localId),
|
|
MTP_long(_accessHash),
|
|
MTP_bytes(_fileReference));
|
|
|
|
case Type::Encrypted:
|
|
return MTP_inputSecureFileLocation(
|
|
MTP_long(_id),
|
|
MTP_long(_accessHash));
|
|
|
|
case Type::Document:
|
|
return MTP_inputDocumentFileLocation(
|
|
MTP_long(_id),
|
|
MTP_long(_accessHash),
|
|
MTP_bytes(_fileReference),
|
|
MTP_string(_sizeLetter
|
|
? std::string(1, char(_sizeLetter))
|
|
: std::string()));
|
|
|
|
case Type::Secure:
|
|
return MTP_inputSecureFileLocation(
|
|
MTP_long(_id),
|
|
MTP_long(_accessHash));
|
|
|
|
case Type::Takeout:
|
|
return MTP_inputTakeoutFileLocation();
|
|
|
|
case Type::Photo:
|
|
return MTP_inputPhotoFileLocation(
|
|
MTP_long(_id),
|
|
MTP_long(_accessHash),
|
|
MTP_bytes(_fileReference),
|
|
MTP_string(std::string(1, char(_sizeLetter))));
|
|
|
|
case Type::PeerPhoto:
|
|
return MTP_inputPeerPhotoFileLocation(
|
|
MTP_flags((_sizeLetter == 'c')
|
|
? MTPDinputPeerPhotoFileLocation::Flag::f_big
|
|
: MTPDinputPeerPhotoFileLocation::Flag(0)),
|
|
GenerateInputPeer(
|
|
PeerId(_id),
|
|
_accessHash,
|
|
_inMessagePeerId,
|
|
_inMessageId,
|
|
self),
|
|
MTP_long(_volumeId));
|
|
|
|
case Type::StickerSetThumb:
|
|
return MTP_inputStickerSetThumb(
|
|
MTP_inputStickerSetID(MTP_long(_id), MTP_long(_accessHash)),
|
|
MTP_int(_localId));
|
|
|
|
case Type::GroupCallStream:
|
|
return MTP_inputGroupCallStream(
|
|
MTP_inputGroupCall(MTP_long(_id), MTP_long(_accessHash)),
|
|
MTP_long(_volumeId),
|
|
MTP_int(_localId));
|
|
|
|
}
|
|
Unexpected("Type in StorageFileLocation::tl.");
|
|
}
|
|
|
|
QByteArray StorageFileLocation::serialize() const {
|
|
auto result = QByteArray();
|
|
if (valid()) {
|
|
result.reserve(serializeSize());
|
|
auto buffer = QBuffer(&result);
|
|
buffer.open(QIODevice::WriteOnly);
|
|
auto stream = QDataStream(&buffer);
|
|
stream.setVersion(QDataStream::Qt_5_1);
|
|
|
|
Assert(!(quint8(_type) & kModernLocationFlag)
|
|
&& !(quint8(_type) & kInMessageFieldsFlag));
|
|
auto typeWithFlags = quint8(_type);
|
|
typeWithFlags |= kModernLocationFlag;
|
|
auto field1 = qint32(_localId);
|
|
auto field2 = qint32(0);
|
|
if (_inMessagePeerId != 0) {
|
|
Assert(field1 == 0);
|
|
typeWithFlags |= kInMessageFieldsFlag;
|
|
field1 = qint32(uint32(_inMessagePeerId.value >> 32));
|
|
field2 = qint32(uint32(_inMessagePeerId.value & 0xFFFFFFFFULL));
|
|
}
|
|
Assert(typeWithFlags != kNonStorageLocationToken);
|
|
stream
|
|
<< quint16(_dcId)
|
|
<< typeWithFlags
|
|
<< quint8(_sizeLetter)
|
|
<< field1
|
|
<< quint64(_id)
|
|
<< quint64(_accessHash)
|
|
<< quint64(_volumeId)
|
|
<< field2
|
|
<< qint32(_inMessageId)
|
|
<< _fileReference;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
int StorageFileLocation::serializeSize() const {
|
|
return valid()
|
|
? int(sizeof(uint64) * 5 + Serialize::bytearraySize(_fileReference))
|
|
: 0;
|
|
}
|
|
|
|
std::optional<StorageFileLocation> StorageFileLocation::FromSerialized(
|
|
const QByteArray &serialized) {
|
|
if (serialized.isEmpty()) {
|
|
return StorageFileLocation();
|
|
}
|
|
|
|
quint16 dcId = 0;
|
|
quint8 typeWithFlags = 0;
|
|
quint8 sizeLetter = 0;
|
|
qint32 field1 = 0;
|
|
quint64 id = 0;
|
|
quint64 accessHash = 0;
|
|
quint64 volumeId = 0;
|
|
qint32 field2 = 0;
|
|
qint32 inMessageId = 0;
|
|
QByteArray fileReference;
|
|
auto stream = QDataStream(serialized);
|
|
stream.setVersion(QDataStream::Qt_5_1);
|
|
stream
|
|
>> dcId
|
|
>> typeWithFlags;
|
|
if (typeWithFlags == kNonStorageLocationToken) {
|
|
return std::nullopt;
|
|
}
|
|
stream
|
|
>> sizeLetter
|
|
>> field1
|
|
>> id
|
|
>> accessHash
|
|
>> volumeId;
|
|
const auto modern = ((typeWithFlags & kModernLocationFlag) != 0);
|
|
const auto inMessageFields
|
|
= ((typeWithFlags & kInMessageFieldsFlag) != 0);
|
|
if (modern) {
|
|
stream >> field2 >> inMessageId;
|
|
typeWithFlags &= ~kModernLocationFlag;
|
|
if (inMessageFields) {
|
|
typeWithFlags &= ~kInMessageFieldsFlag;
|
|
}
|
|
} else if (typeWithFlags & kLegacyInMessagePeerIdFlag) {
|
|
typeWithFlags &= ~kLegacyInMessagePeerIdFlag;
|
|
stream >> field2 >> inMessageId;
|
|
}
|
|
stream >> fileReference;
|
|
|
|
auto result = StorageFileLocation();
|
|
result._dcId = dcId;
|
|
result._type = Type(typeWithFlags);
|
|
result._sizeLetter = sizeLetter;
|
|
result._accessHash = accessHash;
|
|
result._volumeId = volumeId;
|
|
result._inMessageId = inMessageId;
|
|
result._fileReference = fileReference;
|
|
|
|
if (modern) {
|
|
result._id = id;
|
|
if (inMessageFields) {
|
|
result._localId = 0;
|
|
result._inMessagePeerId = PeerId(
|
|
(uint64(uint32(field1)) << 32) | uint64(uint32(field2)));
|
|
} else {
|
|
result._localId = field1;
|
|
result._inMessagePeerId = 0;
|
|
}
|
|
} else {
|
|
result._id = (result._type == Type::PeerPhoto)
|
|
? DeserializePeerId(id).value
|
|
: id;
|
|
result._localId = (result._type == Type::PeerPhoto)
|
|
? 0
|
|
: field1;
|
|
result._inMessagePeerId = (field2 && result._type == Type::PeerPhoto)
|
|
? ((field2 > 0)
|
|
? peerFromUser(UserId(field2))
|
|
: peerFromChannel(ChannelId(-field2)))
|
|
: PeerId();
|
|
}
|
|
if (result._type == Type::StickerSetThumb && result._volumeId != 0) {
|
|
// Legacy field values that cannot be converted to modern.
|
|
// No information about thumb_version, which is required.
|
|
return std::nullopt;
|
|
}
|
|
|
|
return (stream.status() == QDataStream::Ok && result.valid())
|
|
? std::make_optional(result)
|
|
: std::nullopt;
|
|
}
|
|
|
|
StorageFileLocation::Type StorageFileLocation::type() const {
|
|
return _type;
|
|
}
|
|
|
|
bool StorageFileLocation::valid() const {
|
|
switch (_type) {
|
|
case Type::Legacy:
|
|
return (_dcId != 0) && (_volumeId != 0) && (_localId != 0);
|
|
|
|
case Type::Encrypted:
|
|
case Type::Secure:
|
|
case Type::Document:
|
|
return (_dcId != 0) && (_id != 0);
|
|
|
|
case Type::Photo:
|
|
return (_dcId != 0) && (_id != 0) && (_sizeLetter != 0);
|
|
|
|
case Type::Takeout:
|
|
return true;
|
|
|
|
case Type::PeerPhoto:
|
|
case Type::StickerSetThumb:
|
|
return (_dcId != 0) && (_id != 0);
|
|
|
|
case Type::GroupCallStream:
|
|
return (_dcId != 0) && (_id != 0) && (_volumeId != 0);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool StorageFileLocation::isLegacy() const {
|
|
return (_type == Type::Legacy);
|
|
}
|
|
|
|
bool StorageFileLocation::isDocumentThumbnail() const {
|
|
return (_type == Type::Document) && (_sizeLetter != 0);
|
|
}
|
|
|
|
Storage::Cache::Key StorageFileLocation::cacheKey() const {
|
|
using Key = Storage::Cache::Key;
|
|
|
|
// Skip '1' for legacy document cache keys.
|
|
// Skip '2' because it is used for good (fullsize) document thumbnails.
|
|
const auto shifted = ((uint64(_type) + 3) << 8);
|
|
const auto sliced = uint64(_dcId) & 0xFFULL;
|
|
switch (_type) {
|
|
case Type::Legacy:
|
|
case Type::PeerPhoto:
|
|
case Type::StickerSetThumb:
|
|
return Key{
|
|
shifted | sliced | (uint64(uint32(_localId)) << 16),
|
|
_volumeId };
|
|
|
|
case Type::Encrypted:
|
|
case Type::Secure:
|
|
return Key{ shifted | sliced, _id };
|
|
|
|
case Type::Document:
|
|
// Keep old cache keys for documents.
|
|
if (_sizeLetter == 0) {
|
|
return Data::DocumentCacheKey(_dcId, _id);
|
|
//return Key{ 0x100ULL | sliced, _id };
|
|
}
|
|
[[fallthrough]];
|
|
case Type::Photo:
|
|
return Key{ shifted | sliced | (uint64(_sizeLetter) << 16), _id };
|
|
|
|
case Type::Takeout:
|
|
return Key{ shifted, 0 };
|
|
|
|
case Type::GroupCallStream:
|
|
return Key{
|
|
(shifted
|
|
| sliced
|
|
| (uint32(_localId) << 16)
|
|
| (_volumeId << 20)),
|
|
_id };
|
|
}
|
|
return Key();
|
|
}
|
|
|
|
Storage::Cache::Key StorageFileLocation::bigFileBaseCacheKey() const {
|
|
switch (_type) {
|
|
case Type::Document: {
|
|
const auto high = kDocumentBaseCacheTag
|
|
| ((uint64(_dcId) << 16) & kDocumentBaseCacheMask)
|
|
| (_id >> 48);
|
|
const auto low = (_id << 16);
|
|
|
|
Ensures((low & 0xFFULL) == 0);
|
|
return Storage::Cache::Key{ high, low };
|
|
}
|
|
|
|
case Type::StickerSetThumb: {
|
|
const auto high = (uint64(uint32(_localId)) << 24)
|
|
| ((uint64(_type) + 1) << 16)
|
|
| ((uint64(_dcId) & 0xFFULL) << 8)
|
|
| (_volumeId >> 56);
|
|
const auto low = (_volumeId << 8);
|
|
|
|
Ensures((low & 0xFFULL) == 0);
|
|
return Storage::Cache::Key{ high, low };
|
|
}
|
|
|
|
case Type::Photo: {
|
|
const auto high = kPhotoBaseCacheTag
|
|
| ((uint64(_dcId) << 16) & kPhotoBaseCacheMask)
|
|
| (_id >> 48);
|
|
const auto low = (_id << 16);
|
|
|
|
Ensures((low & 0xFFULL) == 0);
|
|
return Storage::Cache::Key{ high, low };
|
|
}
|
|
|
|
case Type::Legacy:
|
|
case Type::PeerPhoto:
|
|
case Type::Encrypted:
|
|
case Type::Secure:
|
|
case Type::Takeout:
|
|
case Type::GroupCallStream:
|
|
Unexpected("Not implemented file location type.");
|
|
|
|
};
|
|
Unexpected("Invalid file location type.");
|
|
}
|
|
|
|
QByteArray StorageFileLocation::fileReference() const {
|
|
return _fileReference;
|
|
}
|
|
|
|
bool StorageFileLocation::refreshFileReference(
|
|
const Data::UpdatedFileReferences &updates) {
|
|
const auto i = (_type == Type::Document)
|
|
? updates.data.find(Data::DocumentFileLocationId{ _id })
|
|
: (_type == Type::Photo)
|
|
? updates.data.find(Data::PhotoFileLocationId{ _id })
|
|
: end(updates.data);
|
|
return (i != end(updates.data))
|
|
? refreshFileReference(i->second)
|
|
: false;
|
|
}
|
|
|
|
bool StorageFileLocation::refreshFileReference(const QByteArray &data) {
|
|
if (data.isEmpty() || _fileReference == data) {
|
|
return false;
|
|
}
|
|
_fileReference = data;
|
|
return true;
|
|
}
|
|
|
|
const StorageFileLocation &StorageFileLocation::Invalid() {
|
|
static auto result = StorageFileLocation();
|
|
return result;
|
|
}
|
|
|
|
bool operator==(const StorageFileLocation &a, const StorageFileLocation &b) {
|
|
const auto valid = a.valid();
|
|
if (valid != b.valid()) {
|
|
return false;
|
|
} else if (!valid) {
|
|
return true;
|
|
}
|
|
const auto type = a._type;
|
|
if (type != b._type) {
|
|
return false;
|
|
}
|
|
|
|
using Type = StorageFileLocation::Type;
|
|
switch (type) {
|
|
case Type::Legacy:
|
|
return (a._dcId == b._dcId)
|
|
&& (a._volumeId == b._volumeId)
|
|
&& (a._localId == b._localId);
|
|
|
|
case Type::Encrypted:
|
|
case Type::Secure:
|
|
return (a._dcId == b._dcId) && (a._id == b._id);
|
|
|
|
case Type::Photo:
|
|
case Type::Document:
|
|
return (a._dcId == b._dcId)
|
|
&& (a._id == b._id)
|
|
&& (a._sizeLetter == b._sizeLetter);
|
|
|
|
case Type::Takeout:
|
|
return true;
|
|
|
|
case Type::PeerPhoto:
|
|
return (a._dcId == b._dcId)
|
|
&& (a._volumeId == b._volumeId)
|
|
&& (a._localId == b._localId)
|
|
&& (a._id == b._id)
|
|
&& (a._sizeLetter == b._sizeLetter);
|
|
|
|
case Type::StickerSetThumb:
|
|
return (a._dcId == b._dcId)
|
|
&& (a._volumeId == b._volumeId)
|
|
&& (a._localId == b._localId)
|
|
&& (a._id == b._id);
|
|
|
|
case Type::GroupCallStream:
|
|
return (a._dcId == b._dcId)
|
|
&& (a._id == b._id)
|
|
&& (a._localId == b._localId);
|
|
};
|
|
Unexpected("Type in StorageFileLocation::operator==.");
|
|
}
|
|
|
|
bool operator<(const StorageFileLocation &a, const StorageFileLocation &b) {
|
|
const auto valid = a.valid();
|
|
if (valid != b.valid()) {
|
|
return !valid;
|
|
} else if (!valid) {
|
|
return false;
|
|
}
|
|
const auto type = a._type;
|
|
if (type != b._type) {
|
|
return (type < b._type);
|
|
}
|
|
|
|
using Type = StorageFileLocation::Type;
|
|
switch (type) {
|
|
case Type::Legacy:
|
|
return std::tie(a._localId, a._volumeId, a._dcId)
|
|
< std::tie(b._localId, b._volumeId, b._dcId);
|
|
|
|
case Type::Encrypted:
|
|
case Type::Secure:
|
|
return std::tie(a._id, a._dcId) < std::tie(b._id, b._dcId);
|
|
|
|
case Type::Photo:
|
|
case Type::Document:
|
|
return std::tie(a._id, a._dcId, a._sizeLetter)
|
|
< std::tie(b._id, b._dcId, b._sizeLetter);
|
|
|
|
case Type::Takeout:
|
|
return false;
|
|
|
|
case Type::PeerPhoto:
|
|
return std::tie(
|
|
a._id,
|
|
a._sizeLetter,
|
|
a._localId,
|
|
a._volumeId,
|
|
a._dcId)
|
|
< std::tie(
|
|
b._id,
|
|
b._sizeLetter,
|
|
b._localId,
|
|
b._volumeId,
|
|
b._dcId);
|
|
|
|
case Type::StickerSetThumb:
|
|
return std::tie(a._id, a._localId, a._volumeId, a._dcId)
|
|
< std::tie(b._id, b._localId, b._volumeId, b._dcId);
|
|
|
|
case Type::GroupCallStream:
|
|
return std::tie(a._id, a._localId, a._dcId)
|
|
< std::tie(b._id, b._localId, b._dcId);
|
|
};
|
|
Unexpected("Type in StorageFileLocation::operator==.");
|
|
}
|
|
|
|
InMemoryKey inMemoryKey(const StorageFileLocation &location) {
|
|
const auto key = location.cacheKey();
|
|
return { key.high, key.low };
|
|
}
|
|
|
|
InMemoryKey inMemoryKey(const WebFileLocation &location) {
|
|
auto result = InMemoryKey();
|
|
const auto &url = location.url();
|
|
const auto sha = hashSha1(url.data(), url.size());
|
|
bytes::copy(
|
|
bytes::object_as_span(&result),
|
|
bytes::make_span(sha).subspan(0, sizeof(result)));
|
|
return result;
|
|
}
|
|
|
|
InMemoryKey inMemoryKey(const GeoPointLocation &location) {
|
|
return InMemoryKey(
|
|
(uint64(std::round(std::abs(location.lat + 360.) * 1000000)) << 32)
|
|
| uint64(std::round(std::abs(location.lon + 360.) * 1000000)),
|
|
(uint64(location.width) << 32) | uint64(location.height));
|
|
}
|
|
|
|
InMemoryKey inMemoryKey(const PlainUrlLocation &location) {
|
|
auto result = InMemoryKey();
|
|
const auto &url = location.url;
|
|
const auto sha = hashSha1(url.data(), url.size() * sizeof(QChar));
|
|
bytes::copy(
|
|
bytes::object_as_span(&result),
|
|
bytes::make_span(sha).subspan(0, sizeof(result)));
|
|
return result;
|
|
}
|
|
|
|
InMemoryKey inMemoryKey(const InMemoryLocation &location) {
|
|
auto result = InMemoryKey();
|
|
const auto &data = location.bytes;
|
|
const auto sha = hashSha1(data.data(), data.size());
|
|
bytes::copy(
|
|
bytes::object_as_span(&result),
|
|
bytes::make_span(sha).subspan(0, sizeof(result)));
|
|
return result;
|
|
}
|
|
|
|
InMemoryKey inMemoryKey(const DownloadLocation &location) {
|
|
return v::match(location.data, [](const auto &data) {
|
|
return inMemoryKey(data);
|
|
});
|
|
}
|
|
|
|
StorageImageLocation::StorageImageLocation(
|
|
const StorageFileLocation &file,
|
|
int width,
|
|
int height)
|
|
: _file(file)
|
|
, _width(width)
|
|
, _height(height) {
|
|
}
|
|
|
|
QByteArray StorageImageLocation::serialize() const {
|
|
auto result = _file.serialize();
|
|
if (!result.isEmpty() || (_width > 0) || (_height > 0)) {
|
|
result.reserve(result.size() + 2 * sizeof(qint32));
|
|
auto buffer = QBuffer(&result);
|
|
buffer.open(QIODevice::Append);
|
|
auto stream = QDataStream(&buffer);
|
|
stream.setVersion(QDataStream::Qt_5_1);
|
|
stream << qint32(_width) << qint32(_height);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
int StorageImageLocation::serializeSize() const {
|
|
const auto partial = _file.serializeSize();
|
|
return (partial > 0 || _width > 0 || _height > 0)
|
|
? (partial + 2 * sizeof(qint32))
|
|
: 0;
|
|
}
|
|
|
|
std::optional<StorageImageLocation> StorageImageLocation::FromSerialized(
|
|
const QByteArray &serialized) {
|
|
if (const auto file = StorageFileLocation::FromSerialized(serialized)) {
|
|
const auto my = 2 * sizeof(qint32);
|
|
const auto full = serialized.size();
|
|
if (!full) {
|
|
return StorageImageLocation(*file, 0, 0);
|
|
} else if (full >= my) {
|
|
qint32 width = 0;
|
|
qint32 height = 0;
|
|
|
|
const auto dimensions = QByteArray::fromRawData(
|
|
serialized.data() + full - my, my);
|
|
auto stream = QDataStream(dimensions);
|
|
stream.setVersion(QDataStream::Qt_5_1);
|
|
stream >> width >> height;
|
|
|
|
return (stream.status() == QDataStream::Ok)
|
|
? StorageImageLocation(*file, width, height)
|
|
: std::optional<StorageImageLocation>();
|
|
}
|
|
}
|
|
return std::nullopt;
|
|
}
|
|
|
|
QByteArray DownloadLocation::serialize() const {
|
|
if (!valid() || v::is<StorageFileLocation>(data)) {
|
|
return v::get<StorageFileLocation>(data).serialize();
|
|
}
|
|
auto result = QByteArray();
|
|
auto buffer = QBuffer(&result);
|
|
buffer.open(QIODevice::WriteOnly);
|
|
auto stream = QDataStream(&buffer);
|
|
stream.setVersion(QDataStream::Qt_5_1);
|
|
stream << quint16(0) << kNonStorageLocationToken;
|
|
|
|
v::match(data, [&](const StorageFileLocation &data) {
|
|
Unexpected("Variant in DownloadLocation::serialize.");
|
|
}, [&](const WebFileLocation &data) {
|
|
stream
|
|
<< quint8(NonStorageLocationType::Web)
|
|
<< data.url()
|
|
<< quint64(data.accessHash());
|
|
}, [&](const GeoPointLocation &data) {
|
|
stream
|
|
<< quint8(NonStorageLocationType::Geo)
|
|
<< qreal(data.lat)
|
|
<< qreal(data.lon)
|
|
<< quint64(data.access)
|
|
<< qint32(data.width)
|
|
<< qint32(data.height)
|
|
<< qint32(data.zoom)
|
|
<< qint32(data.scale);
|
|
}, [&](const PlainUrlLocation &data) {
|
|
stream << quint8(NonStorageLocationType::Url) << data.url.toUtf8();
|
|
}, [&](const InMemoryLocation &data) {
|
|
stream << quint8(NonStorageLocationType::Memory) << data.bytes;
|
|
});
|
|
buffer.close();
|
|
return result;
|
|
}
|
|
|
|
int DownloadLocation::serializeSize() const {
|
|
if (!valid() || v::is<StorageFileLocation>(data)) {
|
|
return v::get<StorageFileLocation>(data).serializeSize();
|
|
}
|
|
auto result = sizeof(quint16) + sizeof(quint8) + sizeof(quint8);
|
|
v::match(data, [&](const StorageFileLocation &data) {
|
|
Unexpected("Variant in DownloadLocation::serializeSize.");
|
|
}, [&](const WebFileLocation &data) {
|
|
result += Serialize::bytearraySize(data.url()) + sizeof(quint64);
|
|
}, [&](const GeoPointLocation &data) {
|
|
result += 2 * sizeof(qreal) + sizeof(quint64) + 4 * sizeof(qint32);
|
|
}, [&](const PlainUrlLocation &data) {
|
|
result += Serialize::bytearraySize(data.url.toUtf8());
|
|
}, [&](const InMemoryLocation &data) {
|
|
result += Serialize::bytearraySize(data.bytes);
|
|
});
|
|
return result;
|
|
}
|
|
|
|
std::optional<DownloadLocation> DownloadLocation::FromSerialized(
|
|
const QByteArray &serialized) {
|
|
quint16 dcId = 0;
|
|
quint8 token = 0;
|
|
auto stream = QDataStream(serialized);
|
|
stream.setVersion(QDataStream::Qt_5_1);
|
|
stream >> dcId >> token;
|
|
if (dcId != 0 || token != kNonStorageLocationToken) {
|
|
const auto storage = StorageFileLocation::FromSerialized(serialized);
|
|
return storage
|
|
? std::make_optional(DownloadLocation{ *storage })
|
|
: std::nullopt;
|
|
}
|
|
quint8 type = 0;
|
|
stream >> type;
|
|
switch (NonStorageLocationType(type)) {
|
|
case NonStorageLocationType::Web: {
|
|
QByteArray url;
|
|
quint64 accessHash = 0;
|
|
stream >> url >> accessHash;
|
|
return (stream.status() == QDataStream::Ok)
|
|
? std::make_optional(
|
|
DownloadLocation{ WebFileLocation(url, accessHash) })
|
|
: std::nullopt;
|
|
} break;
|
|
|
|
case NonStorageLocationType::Geo: {
|
|
qreal lat = 0.;
|
|
qreal lon = 0.;
|
|
quint64 access = 0;
|
|
qint32 width = 0;
|
|
qint32 height = 0;
|
|
qint32 zoom = 0;
|
|
qint32 scale = 0;
|
|
stream >> lat >> lon >> access >> width >> height >> zoom >> scale;
|
|
return (stream.status() == QDataStream::Ok)
|
|
? std::make_optional(
|
|
DownloadLocation{ GeoPointLocation{
|
|
.lat = lat,
|
|
.lon = lon,
|
|
.access = access,
|
|
.width = width,
|
|
.height = height,
|
|
.zoom = zoom,
|
|
.scale = scale } })
|
|
: std::nullopt;
|
|
} break;
|
|
|
|
case NonStorageLocationType::Url: {
|
|
QByteArray utf;
|
|
stream >> utf;
|
|
const auto url = base::FromUtf8Safe(utf);
|
|
return (stream.status() == QDataStream::Ok)
|
|
? std::make_optional(DownloadLocation{ PlainUrlLocation{ url } })
|
|
: std::nullopt;
|
|
} break;
|
|
|
|
case NonStorageLocationType::Memory: {
|
|
QByteArray bytes;
|
|
stream >> bytes;
|
|
return (stream.status() == QDataStream::Ok)
|
|
? std::make_optional(
|
|
DownloadLocation{ InMemoryLocation{ bytes } })
|
|
: std::nullopt;
|
|
} break;
|
|
}
|
|
return std::nullopt;
|
|
}
|
|
|
|
DownloadLocation DownloadLocation::convertToModernPeerPhoto(
|
|
uint64 id,
|
|
uint64 accessHash,
|
|
uint64 photoId) const {
|
|
if (!v::is<StorageFileLocation>(data)) {
|
|
return *this;
|
|
}
|
|
auto &file = v::get<StorageFileLocation>(data);
|
|
return DownloadLocation{
|
|
file.convertToModernPeerPhoto(id, accessHash, photoId)
|
|
};
|
|
}
|
|
|
|
Storage::Cache::Key DownloadLocation::cacheKey() const {
|
|
return v::match(data, [](const GeoPointLocation &data) {
|
|
return Data::GeoPointCacheKey(data);
|
|
}, [](const StorageFileLocation &data) {
|
|
return data.valid()
|
|
? data.cacheKey()
|
|
: Storage::Cache::Key();
|
|
}, [](const WebFileLocation &data) {
|
|
return data.isNull()
|
|
? Storage::Cache::Key()
|
|
: Data::WebDocumentCacheKey(data);
|
|
}, [](const PlainUrlLocation &data) {
|
|
return data.url.isEmpty()
|
|
? Storage::Cache::Key()
|
|
: Data::UrlCacheKey(data.url);
|
|
}, [](const InMemoryLocation &data) {
|
|
return Storage::Cache::Key();
|
|
});
|
|
}
|
|
|
|
Storage::Cache::Key DownloadLocation::bigFileBaseCacheKey() const {
|
|
return v::is<StorageFileLocation>(data)
|
|
? v::get<StorageFileLocation>(data).bigFileBaseCacheKey()
|
|
: Storage::Cache::Key();
|
|
}
|
|
|
|
bool DownloadLocation::valid() const {
|
|
return v::match(data, [](const GeoPointLocation &data) {
|
|
return true;
|
|
}, [](const StorageFileLocation &data) {
|
|
return data.valid();
|
|
}, [](const WebFileLocation &data) {
|
|
return !data.isNull();
|
|
}, [](const PlainUrlLocation &data) {
|
|
return !data.url.isEmpty();
|
|
}, [](const InMemoryLocation &data) {
|
|
return !data.bytes.isEmpty();
|
|
});
|
|
}
|
|
|
|
bool DownloadLocation::isLegacy() const {
|
|
return v::is<StorageFileLocation>(data)
|
|
? v::get<StorageFileLocation>(data).isLegacy()
|
|
: false;
|
|
}
|
|
|
|
QByteArray DownloadLocation::fileReference() const {
|
|
if (!v::is<StorageFileLocation>(data)) {
|
|
return QByteArray();
|
|
}
|
|
return v::get<StorageFileLocation>(data).fileReference();
|
|
}
|
|
|
|
bool DownloadLocation::refreshFileReference(const QByteArray &data) {
|
|
if (!v::is<StorageFileLocation>(this->data)) {
|
|
return false;
|
|
}
|
|
auto &file = v::get<StorageFileLocation>(this->data);
|
|
return file.refreshFileReference(data);
|
|
}
|
|
|
|
bool DownloadLocation::refreshFileReference(
|
|
const Data::UpdatedFileReferences &updates) {
|
|
if (!v::is<StorageFileLocation>(data)) {
|
|
return false;
|
|
}
|
|
auto &file = v::get<StorageFileLocation>(data);
|
|
return file.refreshFileReference(updates);
|
|
}
|
|
|
|
ImageLocation::ImageLocation(
|
|
const DownloadLocation &file,
|
|
int width,
|
|
int height)
|
|
: _file(file)
|
|
, _width(width)
|
|
, _height(height) {
|
|
}
|
|
|
|
QByteArray ImageLocation::serialize() const {
|
|
auto result = _file.serialize();
|
|
if (!result.isEmpty() || (_width > 0) || (_height > 0)) {
|
|
result.reserve(result.size() + 2 * sizeof(qint32));
|
|
auto buffer = QBuffer(&result);
|
|
buffer.open(QIODevice::Append);
|
|
auto stream = QDataStream(&buffer);
|
|
stream.setVersion(QDataStream::Qt_5_1);
|
|
stream << qint32(_width) << qint32(_height);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
int ImageLocation::serializeSize() const {
|
|
const auto partial = _file.serializeSize();
|
|
return (partial > 0 || _width > 0 || _height > 0)
|
|
? (partial + 2 * sizeof(qint32))
|
|
: 0;
|
|
}
|
|
|
|
std::optional<ImageLocation> ImageLocation::FromSerialized(
|
|
const QByteArray &serialized) {
|
|
if (const auto file = DownloadLocation::FromSerialized(serialized)) {
|
|
const auto my = 2 * sizeof(qint32);
|
|
const auto full = serialized.size();
|
|
if (!full) {
|
|
return ImageLocation(*file, 0, 0);
|
|
} else if (full >= my) {
|
|
qint32 width = 0;
|
|
qint32 height = 0;
|
|
|
|
const auto dimensions = QByteArray::fromRawData(
|
|
serialized.data() + full - my, my);
|
|
auto stream = QDataStream(dimensions);
|
|
stream.setVersion(QDataStream::Qt_5_1);
|
|
stream >> width >> height;
|
|
|
|
return (stream.status() == QDataStream::Ok)
|
|
? ImageLocation(*file, width, height)
|
|
: std::optional<ImageLocation>();
|
|
}
|
|
}
|
|
return std::nullopt;
|
|
}
|