RepoMirrors/tdesktop
1
0
Fork 0
mirror of https://github.com/telegramdesktop/tdesktop synced 2025-04-04 23:40:58 +00:00
Code Issues Projects Releases Wiki Activity

Use QSaveFile to write sensitive settings / data.

Browse Source
This commit is contained in:
John Preston 2020-02-24 18:54:16 +04:00
parent c2f58d3ab5
commit 60612635ef
1 changed files with 55 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

99
Telegram/SourceFiles/storage/localstorage.cpp
View File

@ -41,6 +41,7 @@ https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
#include "facades.h" #include "facades.h"
#include <QtCore/QBuffer> #include <QtCore/QBuffer>
#include <QtCore/QSaveFile>
#include <QtCore/QtEndian> #include <QtCore/QtEndian>
#include <QtCore/QDirIterator> #include <QtCore/QDirIterator>
@ -118,10 +119,18 @@ inline constexpr auto is_flag_type(FileOption) { return true; };
bool keyAlreadyUsed(QString &name, FileOptions options = FileOption::User | FileOption::Safe) { bool keyAlreadyUsed(QString &name, FileOptions options = FileOption::User | FileOption::Safe) {
name += '0'; name += '0';
if (QFileInfo(name).exists()) return true; if (QFileInfo(name).exists()) {
return true;
}
if (options & (FileOption::Safe)) { if (options & (FileOption::Safe)) {
name[name.size() - 1] = '1'; name[name.size() - 1] = '1';
return QFileInfo(name).exists(); if (QFileInfo(name).exists()) {
return true;
}
name[name.size() - 1] = 's';
if (QFileInfo(name).exists()) {
return true;
}
} }
return false; return false;
} }
@ -160,6 +169,8 @@ void clearKey(const FileKey &key, FileOptions options = FileOption::User | FileO
if (options & FileOption::Safe) { if (options & FileOption::Safe) {
name[name.size() - 1] = '1'; name[name.size() - 1] = '1';
QFile::remove(name); QFile::remove(name);
name[name.size() - 1] = 's';
QFile::remove(name);
} }
} }
@ -240,10 +251,12 @@ struct EncryptedDescriptor {
}; };
struct FileWriteDescriptor { struct FileWriteDescriptor {
FileWriteDescriptor(const FileKey &key, FileOptions options = FileOption::User | FileOption::Safe) { FileWriteDescriptor(const FileKey &key, FileOptions options = FileOption::User | FileOption::Safe)
: file((options & FileOption::Safe) ? (QFileDevice&)saveFile : plainFile) {
init(toFilePart(key), options); init(toFilePart(key), options);
} }
FileWriteDescriptor(const QString &name, FileOptions options = FileOption::User | FileOption::Safe) { FileWriteDescriptor(const QString &name, FileOptions options = FileOption::User | FileOption::Safe)
: file((options & FileOption::Safe) ? (QFileDevice&)saveFile : plainFile) {
init(name, options); init(name, options);
} }
void init(const QString &name, FileOptions options) { void init(const QString &name, FileOptions options) {
@ -253,29 +266,13 @@ struct FileWriteDescriptor {
if (!_working()) return; if (!_working()) return;
} }
// detect order of read attempts and file version const auto base = ((options & FileOption::User) ? _userBasePath : _basePath) + name;
QString toWrite[2];
toWrite[0] = ((options & FileOption::User) ? _userBasePath : _basePath) + name + '0';
if (options & FileOption::Safe) { if (options & FileOption::Safe) {
toWrite[1] = ((options & FileOption::User) ? _userBasePath : _basePath) + name + '1'; toDelete = base;
QFileInfo toWrite0(toWrite[0]); saveFile.setFileName(base + 's');
QFileInfo toWrite1(toWrite[1]); } else {
if (toWrite0.exists()) { plainFile.setFileName(base + '0');
if (toWrite1.exists()) {
QDateTime mod0 = toWrite0.lastModified(), mod1 = toWrite1.lastModified();
if (mod0 > mod1) {
qSwap(toWrite[0], toWrite[1]);
}
} else {
qSwap(toWrite[0], toWrite[1]);
}
toDelete = toWrite[1];
} else if (toWrite1.exists()) {
toDelete = toWrite[1];
}
} }
file.setFileName(toWrite[0]);
if (file.open(QIODevice::WriteOnly)) { if (file.open(QIODevice::WriteOnly)) {
file.write(tdfMagic, tdfMagicLen); file.write(tdfMagic, tdfMagicLen);
qint32 version = AppVersion; qint32 version = AppVersion;
@ -330,17 +327,21 @@ struct FileWriteDescriptor {
md5.feed(&version, sizeof(version)); md5.feed(&version, sizeof(version));
md5.feed(tdfMagic, tdfMagicLen); md5.feed(tdfMagic, tdfMagicLen);
file.write((const char*)md5.result(), 0x10); file.write((const char*)md5.result(), 0x10);
file.flush();
#ifndef Q_OS_WIN if (saveFile.isOpen()) {
fsync(file.handle()); saveFile.commit();
#endif // Q_OS_WIN } else {
file.close(); plainFile.close();
}
if (!toDelete.isEmpty()) { if (!toDelete.isEmpty()) {
QFile::remove(toDelete); QFile::remove(toDelete + '0');
QFile::remove(toDelete + '1');
} }
} }
QFile file; QFile plainFile;
QSaveFile saveFile;
QFileDevice &file;
QDataStream stream; QDataStream stream;
QString toDelete; QString toDelete;
@ -360,25 +361,35 @@ bool readFile(FileReadDescriptor &result, const QString &name, FileOptions optio
if (!_working()) return false; if (!_working()) return false;
} }
const auto base = ((options & FileOption::User) ? _userBasePath : _basePath) + name;
// detect order of read attempts // detect order of read attempts
QString toTry[2]; QString toTry[2];
toTry[0] = ((options & FileOption::User) ? _userBasePath : _basePath) + name + '0';
if (options & FileOption::Safe) { if (options & FileOption::Safe) {
QFileInfo toTry0(toTry[0]); const auto modern = base + 's';
if (toTry0.exists()) { if (QFileInfo(modern).exists()) {
toTry[1] = ((options & FileOption::User) ? _userBasePath : _basePath) + name + '1'; toTry[0] = modern;
QFileInfo toTry1(toTry[1]); } else {
if (toTry1.exists()) { // Legacy way.
QDateTime mod0 = toTry0.lastModified(), mod1 = toTry1.lastModified(); toTry[0] = base + '0';
if (mod0 < mod1) { QFileInfo toTry0(toTry[0]);
qSwap(toTry[0], toTry[1]); if (toTry0.exists()) {
toTry[1] = ((options & FileOption::User) ? _userBasePath : _basePath) + name + '1';
QFileInfo toTry1(toTry[1]);
if (toTry1.exists()) {
QDateTime mod0 = toTry0.lastModified(), mod1 = toTry1.lastModified();
if (mod0 < mod1) {
qSwap(toTry[0], toTry[1]);
}
} else {
toTry[1] = QString();
} }
} else { } else {
toTry[1] = QString(); toTry[0][toTry[0].size() - 1] = '1';
} }
} else {
toTry[0][toTry[0].size() - 1] = '1';
} }
} else {
toTry[0] = base + '0';
} }
for (int32 i = 0; i < 2; ++i) { for (int32 i = 0; i < 2; ++i) {
QString fname(toTry[i]); QString fname(toTry[i]);