cdricms/vnidrop
1
1
Fork 0
mirror of https://github.com/sudosylabs/vnidrop.git synced 2026-08-12 13:39:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix(core): harden protected identity startup

Browse Source
This commit is contained in:
Hammed Abass
2026-08-09 20:25:16 +02:00
parent 429987785e
commit eefedd0cb0
25 changed files with 1351 additions and 616 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
shared/src/commonMain/kotlin/com/vnidrop/app/core/CoreRepository.kt
View File

@@ -31,10 +31,12 @@ import uniffi.vnidrop.TransferMetadata
import uniffi.vnidrop.TransferAccessMode
import uniffi.vnidrop.VnidropCore
import uniffi.vnidrop.clearInactiveTransferCache
import uniffi.vnidrop.defaultCoreLimits
import uniffi.vnidrop.defaultCoreNetworkConfig
class CoreRepository(
class CoreRepository internal constructor(
private val dispatcher: CoroutineDispatcher = Dispatchers.IO,
private val coreFactory: CoreFactory = ProtectedCoreFactory,
) : CoreGateway {
private val _state = MutableStateFlow(CoreState())
override val state: StateFlow<CoreState> = _state.asStateFlow()
@@ -79,7 +81,7 @@ class CoreRepository(
_state.update { it.copy(isInitialized = false, status = null) }
core = null
disposeCore(previousCore)
core = VnidropCore.initializeWithNetworkConfig(appDataDir, sink, relaySettings.toNative())
core = createCore(appDataDir, relaySettings)
currentAppDataDir = appDataDir
currentRelaySettings = relaySettings
refreshSnapshot(requireCore())
@@ -190,7 +192,7 @@ class CoreRepository(
try {
reclaimed = clearInactiveTransferCache(appDataDir)
} finally {
core = VnidropCore.initializeWithNetworkConfig(appDataDir, sink, relaySettings.toNative())
core = createCore(appDataDir, relaySettings)
refreshSnapshot(requireCore())
_state.update { it.copy(isInitialized = true) }
}
@@ -207,6 +209,9 @@ class CoreRepository(
}
}
private fun createCore(appDataDir: String, relaySettings: RelaySettings): VnidropCore =
coreFactory.create(appDataDir, sink, relaySettings)
override suspend fun receivedArtifacts(): Result<List<ReceivedArtifactModel>> = runCore { activeCore ->
activeCore.listReceivedArtifacts().map { artifact ->
ReceivedArtifactModel(
@@ -330,6 +335,23 @@ class CoreRepository(
}
}
internal fun interface CoreFactory {
fun create(appDataDir: String, eventSink: CoreEventSink, relaySettings: RelaySettings): VnidropCore
}
private object ProtectedCoreFactory : CoreFactory {
override fun create(
appDataDir: String,
eventSink: CoreEventSink,
relaySettings: RelaySettings,
): VnidropCore = VnidropCore.initializeWithExperimentalSavedDevices(
appDataDir,
eventSink,
defaultCoreLimits(),
relaySettings.toNative(),
)
}
private fun RelaySettings.toNative(): CoreNetworkConfig = when (mode) {
RelayMode.Automatic -> defaultCoreNetworkConfig()
RelayMode.StrictCustom -> CoreNetworkConfig(