mirror of https://github.com/procxx/kepka.git
120 lines
3.4 KiB
C++
120 lines
3.4 KiB
C++
//
|
|
// This file is part of Kepka,
|
|
// an unofficial desktop version of Telegram messaging app,
|
|
// see https://github.com/procxx/kepka
|
|
//
|
|
// Kepka 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.
|
|
//
|
|
// In addition, as a special exception, the copyright holders give permission
|
|
// to link the code of portions of this program with the OpenSSL library.
|
|
//
|
|
// Full license: https://github.com/procxx/kepka/blob/master/LICENSE
|
|
// Copyright (c) 2014-2017 John Preston, https://desktop.telegram.org
|
|
// Copyright (c) 2017- Kepka Contributors, https://github.com/procxx
|
|
//
|
|
#pragma once
|
|
|
|
#include "mtproto/auth_key.h"
|
|
#include "ui/images.h"
|
|
|
|
namespace Serialize {
|
|
|
|
inline int stringSize(const QString &str) {
|
|
return sizeof(quint32) + str.size() * sizeof(ushort);
|
|
}
|
|
|
|
inline int bytearraySize(const QByteArray &arr) {
|
|
return sizeof(quint32) + arr.size();
|
|
}
|
|
|
|
inline int bytesSize(base::const_byte_span bytes) {
|
|
return sizeof(quint32) + bytes.size();
|
|
}
|
|
|
|
struct ReadBytesVectorWrap {
|
|
base::byte_vector &bytes;
|
|
};
|
|
|
|
inline ReadBytesVectorWrap bytes(base::byte_vector &bytes) {
|
|
return ReadBytesVectorWrap{bytes};
|
|
}
|
|
|
|
// Compatible with QDataStream &operator>>(QDataStream &, QByteArray &);
|
|
inline QDataStream &operator>>(QDataStream &stream, ReadBytesVectorWrap data) {
|
|
auto &bytes = data.bytes;
|
|
bytes.clear();
|
|
quint32 len;
|
|
stream >> len;
|
|
if (stream.status() != QDataStream::Ok || len == 0xFFFFFFFF) {
|
|
return stream;
|
|
}
|
|
|
|
constexpr auto kStep = quint32(1024 * 1024);
|
|
for (auto allocated = quint32(0); allocated < len;) {
|
|
auto blockSize = qMin(kStep, len - allocated);
|
|
bytes.resize(allocated + blockSize);
|
|
if (stream.readRawData(reinterpret_cast<char *>(bytes.data()) + allocated, blockSize) != blockSize) {
|
|
bytes.clear();
|
|
stream.setStatus(QDataStream::ReadPastEnd);
|
|
return stream;
|
|
}
|
|
allocated += blockSize;
|
|
}
|
|
|
|
return stream;
|
|
}
|
|
|
|
struct WriteBytesWrap {
|
|
base::const_byte_span bytes;
|
|
};
|
|
|
|
inline WriteBytesWrap bytes(base::const_byte_span bytes) {
|
|
return WriteBytesWrap{bytes};
|
|
}
|
|
|
|
inline QDataStream &operator<<(QDataStream &stream, WriteBytesWrap data) {
|
|
auto bytes = data.bytes;
|
|
if (bytes.empty()) {
|
|
stream << quint32(0xFFFFFFFF);
|
|
} else {
|
|
auto size = quint32(bytes.size());
|
|
stream << size;
|
|
stream.writeRawData(reinterpret_cast<const char *>(bytes.data()), size);
|
|
}
|
|
return stream;
|
|
}
|
|
|
|
inline QDataStream &operator<<(QDataStream &stream, ReadBytesVectorWrap data) {
|
|
return stream << WriteBytesWrap{data.bytes};
|
|
}
|
|
|
|
inline int dateTimeSize() {
|
|
return (sizeof(qint64) + sizeof(quint32) + sizeof(qint8));
|
|
}
|
|
|
|
void writeStorageImageLocation(QDataStream &stream, const StorageImageLocation &loc);
|
|
StorageImageLocation readStorageImageLocation(QDataStream &stream);
|
|
int storageImageLocationSize();
|
|
|
|
template <typename T> inline T read(QDataStream &stream) {
|
|
auto result = T();
|
|
stream >> result;
|
|
return result;
|
|
}
|
|
|
|
template <> inline MTP::AuthKey::Data read<MTP::AuthKey::Data>(QDataStream &stream) {
|
|
auto result = MTP::AuthKey::Data();
|
|
stream.readRawData(reinterpret_cast<char *>(result.data()), result.size());
|
|
return result;
|
|
}
|
|
|
|
} // namespace Serialize
|