init

Telegram/SourceFiles/storage/details/storage_file_utilities.h

@@ -0,0 +1,113 @@
+/*
+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,
+		bool sync = false);
+	FileWriteDescriptor(
+		const QString &name,
+		const QString &basePath,
+		bool sync = false);
+	~FileWriteDescriptor();
+
+	void writeData(const QByteArray &data);
+	void writeEncrypted(
+		EncryptedDescriptor &data,
+		const MTP::AuthKeyPtr &key);
+
+private:
+	void init(const QString &name);
+	void finish();
+
+	const QString _basePath;
+	QBuffer _buffer;
+	QDataStream _stream;
+	QByteArray _safeData;
+	QString _base;
+	HashMd5 _md5;
+	int _fullSize = 0;
+	bool _sync = false;
+
+};
+
+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);
+
+void Sync();
+void Finish();
+
+} // namespace details
+} // namespace Storage