PairedDeviceStore.kt

1package com.peachpasswords.sync23import android.content.Context4import androidx.security.crypto.EncryptedSharedPreferences5import androidx.security.crypto.MasterKey6import dagger.hilt.android.qualifiers.ApplicationContext7import kotlinx.coroutines.channels.awaitClose8import kotlinx.coroutines.flow.Flow9import kotlinx.coroutines.flow.callbackFlow10import kotlinx.serialization.builtins.ListSerializer11import kotlinx.serialization.json.Json12import javax.inject.Inject13import javax.inject.Singleton1415/**16 * Persistent registry of paired devices, encrypted at rest with17 * EncryptedSharedPreferences (same construction as DuressConfigStore /18 * LicenseRepository, v2.1.36). The whole list lives under a single key as a19 * kotlinx.serialization JSON array of the REAL [PairedDevice] (sync package).20 *21 * Reconciliation with the C7 brief:22 *  - The brief invents a fresh `PairedDevice(id, name, publicKey: ByteArray,23 *    lastSeen, trustStatus, lastSyncAt)`. That type does NOT exist in this24 *    shape. The real [PairedDevice] (PairedDevice.kt) keys on `deviceId`, holds25 *    the public key as `longTermPublicKeyHex` (hex, not ByteArray), and tracks26 *    `pairedAt` / `lastSyncedAt` / `lastKnownIp`. This store reuses it verbatim;27 *    Base64 publicKey storage is therefore unnecessary — the hex field is the28 *    canonical wire form already produced by PairingFrames.frameToPairedDevice.29 *  - The brief's `TrustStatus` enum does not exist either. The real model uses a30 *    nullable `trustSource: String?` (e.g. "direct", "manifest"). Trust state is31 *    expressed there. See C7-INTEGRATION.md for the trust-status delta.32 *  - `updateLastSync` here maps to the real `lastSyncedAt` field. There is no33 *    `lastSeen` field on the real PairedDevice; [updateLastSeen] folds into34 *    `lastKnownIp` refresh + `lastSyncedAt` is left untouched. It is kept as a35 *    no-op-friendly hook so callers from the brief compile; see integration note.36 */37@Singleton38class PairedDeviceStore @Inject constructor(39    @ApplicationContext context: Context,40) : PairedDeviceRegistry {41    private val masterKey = MasterKey.Builder(context)42        .setKeyScheme(MasterKey.KeyScheme.AES256_GCM)43        .build()4445    private val prefs = EncryptedSharedPreferences.create(46        context,47        PREFS_FILE,48        masterKey,49        EncryptedSharedPreferences.PrefKeyEncryptionScheme.AES256_SIV,50        EncryptedSharedPreferences.PrefValueEncryptionScheme.AES256_GCM,51    )5253    private val listSerializer = ListSerializer(PairedDevice.serializer())54    private val revocationSerializer = ListSerializer(DeviceRevocation.serializer())5556    override fun getAll(): List<PairedDevice> {57        val raw = prefs.getString(KEY_DEVICES, null) ?: return emptyList()58        return runCatching { json.decodeFromString(listSerializer, raw) }.getOrDefault(emptyList())59    }6061    override fun get(id: String): PairedDevice? = getAll().firstOrNull { it.deviceId == id }6263    override fun getByPubkey(pubkeyHex: String): PairedDevice? =64        getAll().firstOrNull {65            it.longTermPublicKeyHex?.let { key -> CryptoCompare.hex(key, pubkeyHex) } == true66        }6768    /** Insert or replace by [PairedDevice.deviceId]. */69    override fun upsert(device: PairedDevice) {70        val next = getAll().filterNot { it.deviceId == device.deviceId } + device71        persist(next)72    }7374    override fun remove(deviceId: String) {75        persist(getAll().filterNot { it.deviceId == deviceId })76    }7778    // ── Device revocation (Part 5 §C) ────────────────────────────────────────7980    override fun listRevocations(): List<DeviceRevocation> {81        val raw = prefs.getString(KEY_REVOCATIONS, null) ?: return emptyList()82        return runCatching { json.decodeFromString(revocationSerializer, raw) }83            .getOrDefault(emptyList())84    }8586    /**87     * Revoke a device: record it, then drop it from the registry. The record88     * rides out in every manifest this device signs from now on. A stale entry89     * may survive in the pending-vouch queue; the confirm path re-checks90     * revocation before promoting anything out of it.91     */92    fun revoke(93        deviceId: String,94        revokedByDeviceId: String,95        nowMs: Long = System.currentTimeMillis(),96    ): DeviceRevocation? {97        val existing = listRevocations()98        if (existing.any { it.deviceId == deviceId }) return null99        val record = DeviceRevocation(100            deviceId = deviceId,101            longTermPublicKeyHex = get(deviceId)?.longTermPublicKeyHex ?: "0".repeat(64),102            revokedAt = nowMs,103            revokedByDeviceId = revokedByDeviceId,104        )105        persistRevocations(existing + record)106        remove(deviceId)107        return record108    }109110    /**111     * `revokedAt` is peer-supplied and is NOT used for ordering — revocation is112     * terminal, so an incoming record can never displace one already held.113     */114    override fun mergeRevocations(incoming: List<DeviceRevocation>): List<String> {115        val existing = listRevocations()116        val known = existing.map { it.deviceId }.toSet()117        val fresh = incoming.filterNot { known.contains(it.deviceId) }118        if (fresh.isEmpty()) return emptyList()119        persistRevocations((existing + fresh).takeLast(MAX_REVOCATIONS))120        return fresh.map { it.deviceId }121    }122123    /**124     * Clear a revocation. The ONLY legitimate caller is a direct, SAS-confirmed125     * re-pair — a physically present user on both devices. Never call this from126     * a path a remote peer can reach.127     */128    override fun clearRevocation(deviceId: String) {129        val existing = listRevocations()130        val filtered = existing.filterNot { it.deviceId == deviceId }131        if (filtered.size != existing.size) persistRevocations(filtered)132    }133134    fun clear() {135        prefs.edit().remove(KEY_DEVICES).apply()136    }137138    /**139     * Refresh the device's `lastKnownIp` (the closest real analogue to the140     * brief's `lastSeen`). No-op when the id is unknown or [ip] is null.141     */142    fun updateLastSeen(id: String, ip: String? = null) {143        val current = get(id) ?: return144        if (ip == null) return145        upsert(current.copy(lastKnownIp = ip))146    }147148    fun updateLastSync(id: String, at: Long = System.currentTimeMillis()) {149        val current = get(id) ?: return150        upsert(current.copy(lastSyncedAt = at))151    }152153    /**154     * Reconnect prefers each pair's relay over the global fallback. Rewrite all155     * established pairs when the user changes the mesh relay in Settings.156     */157    fun setRelayUrlForAll(relayUrl: String) {158        persist(withRelayUrl(getAll(), relayUrl))159    }160161    /**162     * Observe the full list. EncryptedSharedPreferences supports163     * OnSharedPreferenceChangeListener, so this emits on every persist —164     * including writes from other components in-process.165     */166    fun observeAll(): Flow<List<PairedDevice>> = callbackFlow {167        trySend(getAll())168        val listener = android.content.SharedPreferences.OnSharedPreferenceChangeListener { _, key ->169            if (key == KEY_DEVICES || key == null) trySend(getAll())170        }171        prefs.registerOnSharedPreferenceChangeListener(listener)172        awaitClose { prefs.unregisterOnSharedPreferenceChangeListener(listener) }173    }174175    private fun persistRevocations(records: List<DeviceRevocation>) {176        prefs.edit()177            .putString(KEY_REVOCATIONS, json.encodeToString(revocationSerializer, records))178            .apply()179    }180181    private fun persist(devices: List<PairedDevice>) {182        prefs.edit()183            .putString(KEY_DEVICES, json.encodeToString(listSerializer, devices))184            .apply()185    }186187    private companion object {188        const val PREFS_FILE = "peach_paired_devices"189        const val KEY_DEVICES = "paired_devices"190        const val KEY_REVOCATIONS = "device_revocations"191        /** Bound the list so a hostile peer cannot grow storage without limit. */192        const val MAX_REVOCATIONS = 500193        val json = Json { encodeDefaults = true; ignoreUnknownKeys = true }194    }195}196197internal fun withRelayUrl(198    devices: List<PairedDevice>,199    relayUrl: String,200): List<PairedDevice> = devices.map { it.copy(relayUrl = relayUrl) }