113 lines
2.5 KiB
C++
113 lines
2.5 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
|
|
*/
|
|
#pragma once
|
|
|
|
#include "storage/storage_account.h"
|
|
|
|
#include <QtCore/QBuffer>
|
|
|
|
namespace Storage {
|
|
namespace details {
|
|
|
|
[[nodiscard]] QString ToFilePart(FileKey val);
|
|
[[nodiscard]] bool KeyAlreadyUsed(QString &name);
|
|
[[nodiscard]] FileKey GenerateKey(const QString &basePath);
|
|
void ClearKey(const FileKey &key, const QString &basePath);
|
|
|
|
[[nodiscard]] bool CheckStreamStatus(QDataStream &stream);
|
|
[[nodiscard]] MTP::AuthKeyPtr CreateLocalKey(
|
|
const QByteArray &passcode,
|
|
const QByteArray &salt);
|
|
[[nodiscard]] MTP::AuthKeyPtr CreateLegacyLocalKey(
|
|
const QByteArray &passcode,
|
|
const QByteArray &salt);
|
|
|
|
struct FileReadDescriptor final {
|
|
~FileReadDescriptor();
|
|
|
|
int32 version = 0;
|
|
QByteArray data;
|
|
QBuffer buffer;
|
|
QDataStream stream;
|
|
};
|
|
|
|
struct EncryptedDescriptor final {
|
|
EncryptedDescriptor();
|
|
explicit EncryptedDescriptor(uint32 size);
|
|
~EncryptedDescriptor();
|
|
|
|
void finish();
|
|
|
|
QByteArray data;
|
|
QBuffer buffer;
|
|
QDataStream stream;
|
|
};
|
|
|
|
[[nodiscard]] QByteArray PrepareEncrypted(
|
|
EncryptedDescriptor &data,
|
|
const MTP::AuthKeyPtr &key);
|
|
|
|
class FileWriteDescriptor final {
|
|
public:
|
|
FileWriteDescriptor(
|
|
const FileKey &key,
|
|
const QString &basePath);
|
|
FileWriteDescriptor(
|
|
const QString &name,
|
|
const QString &basePath);
|
|
~FileWriteDescriptor();
|
|
|
|
void writeData(const QByteArray &data);
|
|
void writeEncrypted(
|
|
EncryptedDescriptor &data,
|
|
const MTP::AuthKeyPtr &key);
|
|
|
|
private:
|
|
void init(const QString &name);
|
|
[[nodiscard]] QString path(char postfix) const;
|
|
template <typename File>
|
|
[[nodiscard]] bool open(File &file, char postfix);
|
|
[[nodiscard]] bool writeHeader(QFileDevice &file);
|
|
void writeFooter(QFileDevice &file);
|
|
void finish();
|
|
|
|
const QString _basePath;
|
|
QBuffer _buffer;
|
|
QDataStream _stream;
|
|
QByteArray _safeData;
|
|
QString _base;
|
|
HashMd5 _md5;
|
|
int _fullSize = 0;
|
|
|
|
};
|
|
|
|
bool ReadFile(
|
|
FileReadDescriptor &result,
|
|
const QString &name,
|
|
const QString &basePath);
|
|
|
|
bool DecryptLocal(
|
|
EncryptedDescriptor &result,
|
|
const QByteArray &encrypted,
|
|
const MTP::AuthKeyPtr &key);
|
|
|
|
bool ReadEncryptedFile(
|
|
FileReadDescriptor &result,
|
|
const QString &name,
|
|
const QString &basePath,
|
|
const MTP::AuthKeyPtr &key);
|
|
|
|
bool ReadEncryptedFile(
|
|
FileReadDescriptor &result,
|
|
const FileKey &fkey,
|
|
const QString &basePath,
|
|
const MTP::AuthKeyPtr &key);
|
|
|
|
} // namespace details
|
|
} // namespace Storage
|