rixxc/KeePassDX
1
0
Fork 0
mirror of https://github.com/Kunzisoft/KeePassDX.git synced 2025-12-04 15:49:33 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge database metadata

Browse Source
This commit is contained in:
J-Jamet
2022-01-15 21:03:40 +01:00
parent c26995779a
commit f9445de71f
4 changed files with 66 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/src/main/java/com/kunzisoft/keepass/database/action/AssignPasswordInDatabaseRunnable.kt
View File

@@ -43,7 +43,7 @@ open class AssignPasswordInDatabaseRunnable (
System.arraycopy(database.masterKey, 0, mBackupKey!!, 0, mBackupKey!!.size)
val uriInputStream = UriUtil.getUriInputStream(context.contentResolver, mMainCredential.keyFileUri)
database.retrieveMasterKey(mMainCredential.masterPassword, uriInputStream)
database.assignMasterKey(mMainCredential.masterPassword, uriInputStream)
} catch (e: Exception) {
erase(mBackupKey)
setError(e)

3
app/src/main/java/com/kunzisoft/keepass/database/element/Database.kt
View File

@@ -916,9 +916,10 @@ class Database {
}
@Throws(IOException::class)
fun retrieveMasterKey(key: String?, keyInputStream: InputStream?) {
fun assignMasterKey(key: String?, keyInputStream: InputStream?) {
mDatabaseKDB?.retrieveMasterKey(key, keyInputStream)
mDatabaseKDBX?.retrieveMasterKey(key, keyInputStream)
mDatabaseKDBX?.keyLastChanged = DateInstant()
}
fun rootCanContainsEntry(): Boolean {

5
app/src/main/java/com/kunzisoft/keepass/database/element/database/DatabaseKDBX.kt
View File

@@ -88,14 +88,13 @@ class DatabaseKDBX : DatabaseVersioned<UUID, UUID, GroupKDBX, EntryKDBX> {
var kdbxVersion = UnsignedInt(0)
var name = ""
var nameChanged = DateInstant()
// TODO change setting date
var settingsChanged = DateInstant()
var description = ""
var descriptionChanged = DateInstant()
var defaultUserName = ""
var defaultUserNameChanged = DateInstant()
// TODO change setting date
var settingsChanged = DateInstant()
// TODO last change date
var keyLastChanged = DateInstant()
var keyChangeRecDays: Long = -1
var keyChangeForceDays: Long = 1

104
app/src/main/java/com/kunzisoft/keepass/database/merge/DatabaseKDBXMerger.kt
View File

@@ -2,7 +2,6 @@ package com.kunzisoft.keepass.database.merge
import com.kunzisoft.keepass.database.action.node.NodeHandler
import com.kunzisoft.keepass.database.element.Attachment
import com.kunzisoft.keepass.database.element.DateInstant
import com.kunzisoft.keepass.database.element.database.DatabaseKDBX
import com.kunzisoft.keepass.database.element.entry.EntryKDBX
import com.kunzisoft.keepass.database.element.group.GroupKDBX
@@ -15,12 +14,28 @@ class DatabaseKDBXMerger(private var database: DatabaseKDBX) {
fun merge(databaseToMerge: DatabaseKDBX) {
// TODO database data
var nameChanged = DateInstant()
var settingsChanged = DateInstant()
var descriptionChanged = DateInstant()
var defaultUserNameChanged = DateInstant()
var keyLastChanged = DateInstant()
if (database.nameChanged.date.before(databaseToMerge.nameChanged.date)) {
database.name = databaseToMerge.name
database.nameChanged = databaseToMerge.nameChanged
}
if (database.descriptionChanged.date.before(databaseToMerge.descriptionChanged.date)) {
database.description = databaseToMerge.description
database.descriptionChanged = databaseToMerge.descriptionChanged
}
if (database.defaultUserNameChanged.date.before(databaseToMerge.defaultUserNameChanged.date)) {
database.defaultUserName = databaseToMerge.defaultUserName
database.defaultUserNameChanged = databaseToMerge.defaultUserNameChanged
}
if (database.keyLastChanged.date.before(databaseToMerge.keyLastChanged.date)) {
database.keyChangeRecDays = databaseToMerge.keyChangeRecDays
database.keyChangeForceDays = databaseToMerge.keyChangeForceDays
database.isKeyChangeForceOnce = databaseToMerge.isKeyChangeForceOnce
database.keyLastChanged = databaseToMerge.keyLastChanged
}
if (database.settingsChanged.date.before(databaseToMerge.settingsChanged.date)) {
// TODO settings
database.settingsChanged = databaseToMerge.settingsChanged
}
val databaseRootGroupId = database.rootGroup?.nodeId
val databaseRootGroupIdToMerge = databaseToMerge.rootGroup?.nodeId
@@ -40,7 +55,6 @@ class DatabaseKDBXMerger(private var database: DatabaseKDBX) {
}
}
// TODO fix concurrent modification exception
databaseToMerge.rootGroup?.doForEachChild(
object : NodeHandler<EntryKDBX>() {
override fun operate(node: EntryKDBX): Boolean {
@@ -80,21 +94,21 @@ class DatabaseKDBXMerger(private var database: DatabaseKDBX) {
}
}
private fun mergeEntry(entryToMerge: EntryKDBX, databaseToMerge: DatabaseKDBX) {
val entryId = entryToMerge.nodeId
val databaseEntryToMerge = databaseToMerge.getEntryById(entryId)
val databaseEntry = database.getEntryById(entryId)
private fun mergeEntry(nodeToMerge: EntryKDBX, databaseToMerge: DatabaseKDBX) {
val entryId = nodeToMerge.nodeId
val entryToMerge = databaseToMerge.getEntryById(entryId)
val entry = database.getEntryById(entryId)
val deletedObject = database.getDeletedObject(entryId)
if (databaseEntryToMerge != null) {
if (entryToMerge != null) {
// Retrieve parent in current database
var parentEntryToMerge: GroupKDBX? = null
databaseEntryToMerge.parent?.nodeId?.let {
entryToMerge.parent?.nodeId?.let {
parentEntryToMerge = database.getGroupById(it)
}
// Copy attachments in main pool
val newAttachments = mutableListOf<Attachment>()
databaseEntryToMerge.getAttachments(databaseToMerge.attachmentPool).forEach { attachment ->
entryToMerge.getAttachments(databaseToMerge.attachmentPool).forEach { attachment ->
val binarySize = attachment.binaryData.getSize()
val binaryData = database.buildNewBinaryAttachment(
isRAMSufficient.invoke(binarySize),
@@ -110,64 +124,68 @@ class DatabaseKDBXMerger(private var database: DatabaseKDBX) {
}
newAttachments.add(Attachment(attachment.name, binaryData))
}
databaseEntryToMerge.removeAttachments()
entryToMerge.removeAttachments()
newAttachments.forEach { newAttachment ->
databaseEntryToMerge.putAttachment(newAttachment, database.attachmentPool)
entryToMerge.putAttachment(newAttachment, database.attachmentPool)
}
if (databaseEntry == null) {
if (entry == null) {
// If it's a deleted object, but another instance was updated
// If entry parent to add exists and in current database
if (deletedObject == null
|| deletedObject.deletionTime.date
.before(databaseEntryToMerge.lastModificationTime.date)
.before(entryToMerge.lastModificationTime.date)
|| parentEntryToMerge != null) {
database.addEntryTo(databaseEntryToMerge, parentEntryToMerge)
database.addEntryTo(entryToMerge, parentEntryToMerge)
}
} else if (databaseEntry.lastModificationTime.date
.before(databaseEntryToMerge.lastModificationTime.date)
} else if (entry.lastModificationTime.date
.before(entryToMerge.lastModificationTime.date)
) {
if (parentEntryToMerge == databaseEntry.parent) {
databaseEntry.updateWith(databaseEntryToMerge)
if (parentEntryToMerge == entry.parent) {
entry.updateWith(entryToMerge)
} else {
// Update entry with databaseEntryToMerge and merge history
database.removeEntryFrom(databaseEntry, databaseEntry.parent)
database.removeEntryFrom(entry, entry.parent)
// TODO history =
if (parentEntryToMerge != null) {
database.addEntryTo(databaseEntryToMerge, parentEntryToMerge)
database.addEntryTo(entryToMerge, parentEntryToMerge)
}
}
}
}
}
private fun mergeGroup(node: GroupKDBX, databaseToMerge: DatabaseKDBX) {
val groupId = node.nodeId
val databaseGroupToMerge = databaseToMerge.getGroupById(groupId)
val databaseGroup = database.getGroupById(groupId)
private fun mergeGroup(nodeToMerge: GroupKDBX, databaseToMerge: DatabaseKDBX) {
val groupId = nodeToMerge.nodeId
val groupToMerge = databaseToMerge.getGroupById(groupId)
val group = database.getGroupById(groupId)
val deletedObject = database.getDeletedObject(groupId)
if (databaseGroupToMerge != null) {
if (groupToMerge != null) {
// Retrieve parent in current database
var parentGroup: GroupKDBX? = null
databaseGroupToMerge.parent?.nodeId?.let {
parentGroup = database.getGroupById(it)
var parentGroupToMerge: GroupKDBX? = null
groupToMerge.parent?.nodeId?.let {
parentGroupToMerge = database.getGroupById(it)
}
if (databaseGroup == null) {
if (group == null) {
// If group parent to add exists and in current database
if (deletedObject == null
|| deletedObject.deletionTime.date
.before(databaseGroupToMerge.lastModificationTime.date)
|| parentGroup != null) {
database.addGroupTo(databaseGroupToMerge, parentGroup)
.before(groupToMerge.lastModificationTime.date)
|| parentGroupToMerge != null) {
database.addGroupTo(groupToMerge, parentGroupToMerge)
}
} else if (databaseGroup.lastModificationTime.date
.before(databaseGroupToMerge.lastModificationTime.date)
} else if (group.lastModificationTime.date
.before(groupToMerge.lastModificationTime.date)
) {
database.removeGroupFrom(databaseGroup, databaseGroup.parent)
if (parentGroup != null) {
database.addGroupTo(databaseGroupToMerge, parentGroup)
if (parentGroupToMerge == group.parent) {
group.updateWith(groupToMerge)
} else {
database.removeGroupFrom(group, group.parent)
if (parentGroupToMerge != null) {
database.addGroupTo(groupToMerge, parentGroupToMerge)
}
}
}
}