Git-Mirrors/keepassxc
1
0
Fork 0
mirror of https://github.com/keepassxreboot/keepassxc.git synced 2024-09-20 08:05:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Additional layer of protection for unsafe saves

Browse Source 
* Attempt to restore database, if that fails retain the temporary file and tell the user where it is located
This commit is contained in:
Jonathan White 2019-04-07 09:21:58 -04:00
parent 3b0b5d85e9
commit 791b796c23
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/core/Database.cpp
View File

@ -251,11 +251,15 @@ bool Database::save(const QString& filePath, QString* error, bool atomic, bool b
tempFile.setAutoRemove(false);
setFilePath(filePath);
return true;
} else {
// restore the database from the backup
if (backup) {
restoreDatabase(filePath);
} else if (!backup || !restoreDatabase(filePath)) {
// Failed to copy new database in place, and
// failed to restore from backup or backups disabled
tempFile.setAutoRemove(false);
if (error) {
*error = tr("%1\nBackup database located at %2").arg(tempFile.errorString(), tempFile.fileName());
}
markAsModified();
return false;
}
}
@ -338,8 +342,12 @@ bool Database::restoreDatabase(const QString& filePath)
auto match = re.match(filePath);
auto backupFilePath = match.captured(1) + ".old" + match.captured(2);
QFile::remove(filePath);
return QFile::copy(backupFilePath, filePath);
// Only try to restore if the backup file actually exists
if (QFile::exists(backupFilePath)) {
QFile::remove(filePath);
return QFile::copy(backupFilePath, filePath);
}
return false;
}
bool Database::isReadOnly() const