12345678910111213141516171819202122232425import { HKDF } from "@stablelib/hkdf";26import { SHA256 } from "@stablelib/sha256";27import type { SyncVaultAdapter } from "./sync-coordinator";28import { getOrCreateDeviceId } from "./device-identity";29import {30 computeVaultId,31 deriveSyncIdentityKeypair,32 deriveSyncSigningKeypair,33 hexToPubkey,34 pubkeyToHex,35} from "../shared/sync-identity";36import {37 buildSignedDeviceManifest,38 verifyDeviceManifest,39} from "../shared/device-manifest";40import { constantTimeEqualHex } from "../shared/constant-time";41import {42 applyMergedRemoteEntries,43 getDeletedEntries,44 getRootKeyHex,45 getEntries,46 getEntryById,47 getVaultVersion,48 isUnlocked,49} from "./vault";50import {51 findPairedDeviceById,52 findPairedDeviceByPubkey,53 isDeviceRevoked,54 listDeviceRevocations,55 listPairedDevices,56 mergeDeviceRevocations,57 removePairedDeviceById,58 upsertPairedDevice,59} from "./paired-devices";60import {61 removePendingVouchedDevice,62} from "./pending-vouched-devices";63import {64 bytesToHex as rendezvousBytesToHex,65 deriveMeshRendezvousSecret,66 normalizeRelayUrl,67} from "../shared/rendezvous";68import { recordSyncAlerts } from "./sync-alerts";69import { SYNC_LIMITS, clampSyncTimestamp } from "../shared/sync-limits";70import type { VaultEntry } from "../shared/types";71import type {72 ManifestDeviceEntry,73 PairedDevice,74 SignedDeviceManifest,75} from "../shared/sync-types";76777879const ENTRY_KEY_INFO_SUFFIX = "sync-entry-v1";80const ENTRY_AAD = "peach-sync-entry-v1";8182export class VaultLockedError extends Error {83 constructor() {84 super("vault_locked");85 this.name = "VaultLockedError";86 }87}8889export function createSyncVaultAdapter(90 context: { deviceName?: string } = {},91): SyncVaultAdapter {92 return {93 async computeVaultId() {94 const master = requireMasterKey();95 return computeVaultId(master);96 },9798 async getOurDeviceId() {99 return getOrCreateDeviceId();100 },101102 async getLocalVersion() {103 return getVaultVersion();104 },105106 async buildEntryManifest() {107 requireUnlocked();108 const manifest: Record<string, number> = {};109 for (const entry of getEntries()) {110 manifest[entry.id] = entry.modifiedAt;111 }112 return manifest;113 },114115 async buildTombstoneManifest() {116 117 118 119 120 121 122 123 124 125 126 requireUnlocked();127 const manifest: Record<string, number> = {};128 for (const entry of getDeletedEntries()) {129 if (typeof entry.deletedAt === "number") {130 manifest[entry.id] = entry.deletedAt;131 }132 }133 return manifest;134 },135136 async getEncryptedEntries(entryIds) {137 requireUnlocked();138 const masterKey = requireMasterKey();139 const out: Record<string, string> = {};140 for (const id of entryIds) {141 const entry = getEntryById(id);142 if (!entry) continue;143 const key = await deriveEntryKey(masterKey, id);144 out[id] = await encryptEntry(key, entry);145 }146 return out;147 },148149 async applyEntries(entries) {150 requireUnlocked();151 const entryList = Object.entries(entries);152 153 if (entryList.length > SYNC_LIMITS.MAX_ENTRY_COUNT) {154 throw new Error(155 `sync entry count ${entryList.length} exceeds limit ${SYNC_LIMITS.MAX_ENTRY_COUNT}`,156 );157 }158 const masterKey = requireMasterKey();159 const decrypted: VaultEntry[] = [];160 for (const [id, envelope] of entryList) {161 162 163 if (envelope.length > (SYNC_LIMITS.MAX_ENTRY_PLAINTEXT_BYTES * 4) / 3 + 64) {164 continue; 165 }166 const key = await deriveEntryKey(masterKey, id);167 const entry = await decryptEntryOrNull(key, envelope);168 if (!entry) continue;169 170 if (JSON.stringify(entry).length > SYNC_LIMITS.MAX_ENTRY_PLAINTEXT_BYTES) {171 continue;172 }173 174 175 if (entry.id !== id) continue;176 decrypted.push(entry);177 }178 if (decrypted.length > 0) {179 await applyMergedRemoteEntries(decrypted, {}, context);180 }181 },182183 async applyTombstones(tombstones) {184 requireUnlocked();185 const tombstoneList = Object.entries(tombstones);186 187 if (tombstoneList.length > SYNC_LIMITS.MAX_TOMBSTONE_COUNT) {188 throw new Error(189 `sync tombstone count ${tombstoneList.length} exceeds limit ${SYNC_LIMITS.MAX_TOMBSTONE_COUNT}`,190 );191 }192 const cleaned: Record<string, number> = {};193 for (const [id, ts] of tombstoneList) {194 if (typeof ts === "number" && Number.isFinite(ts) && ts > 0) {195 196 197 198 cleaned[id] = clampSyncTimestamp(ts);199 }200 }201 if (Object.keys(cleaned).length > 0) {202 await applyMergedRemoteEntries([], cleaned, context);203 }204 },205206 async buildOutgoingDeviceManifest() {207 requireUnlocked();208 const masterKey = requireMasterKey();209 const issuerDeviceId = await getOrCreateDeviceId();210 const signing = await deriveSyncSigningKeypair(masterKey, issuerDeviceId);211 const revoked = await listDeviceRevocations();212 const revokedIds = new Set(revoked.map((r) => r.deviceId));213 const devices = (await listPairedDevices())214 .filter((d) => !revokedIds.has(d.deviceId))215 .map(toManifestEntry)216 .filter((d): d is ManifestDeviceEntry => d !== null);217 return buildSignedDeviceManifest(218 issuerDeviceId,219 devices,220 signing.privateKey,221 { revoked },222 );223 },224225 async applyIncomingDeviceManifest(manifest, authenticatedRemoteStaticPublicKey) {226 requireUnlocked();227 await applyVerifiedPeerManifest(228 manifest,229 authenticatedRemoteStaticPublicKey,230 );231 },232 };233}234235function toManifestEntry(device: PairedDevice): ManifestDeviceEntry | null {236 if (237 !device.deviceId ||238 !device.deviceName ||239 !device.deviceType ||240 !device.longTermPublicKeyHex ||241 !/^[0-9a-f]{64}$/.test(device.longTermPublicKeyHex) ||242 !device.signingPublicKeyHex ||243 !/^[0-9a-f]{64}$/.test(device.signingPublicKeyHex)244 ) {245 return null;246 }247 return {248 deviceId: device.deviceId,249 deviceName: device.deviceName,250 deviceType: device.deviceType,251 longTermPublicKeyHex: device.longTermPublicKeyHex,252 signingPublicKeyHex: device.signingPublicKeyHex,253 wsPort: device.wsPort,254 relayUrl: device.relayUrl,255 pairedAt: device.pairedAt,256 };257}258259260261262263264265266267268269270271272273async function applyVerifiedPeerManifest(274 manifest: SignedDeviceManifest,275 authenticatedRemoteStaticPublicKey: Uint8Array,276): Promise<void> {277 const issuerId = manifest?.body?.issuerDeviceId;278 if (typeof issuerId !== "string" || !issuerId) { return; }279280 if (authenticatedRemoteStaticPublicKey.length !== 32) {281 return;282 }283 const authenticatedPubkeyHex = pubkeyToHex(authenticatedRemoteStaticPublicKey);284 const issuer = await findPairedDeviceByPubkey(authenticatedPubkeyHex);285 if (286 !issuer ||287 issuer.deviceId !== issuerId ||288 (289 issuer.trustSource !== "direct" &&290 issuer.trustSource !== "vouched"291 ) ||292 !issuer.signingPublicKeyHex ||293 !/^[0-9a-f]{64}$/.test(issuer.signingPublicKeyHex)294 ) {295 return;296 }297298 let body: Awaited<ReturnType<typeof verifyDeviceManifest>>;299 try {300 body = await verifyDeviceManifest(301 manifest,302 hexToPubkey(issuer!.signingPublicKeyHex!),303 );304 } catch {305 306 307 308 309 310 const masterForIssuer = requireMasterKey();311 try {312 const derived = await deriveSyncSigningKeypair(313 masterForIssuer,314 issuer.deviceId,315 );316 body = await verifyDeviceManifest(manifest, derived.publicKeyRaw);317 } catch {318 return;319 } finally {320 masterForIssuer.fill(0);321 }322 }323 if (body.issuerDeviceId !== issuer.deviceId) {324 return;325 }326327 const ourDeviceId = await getOrCreateDeviceId();328 const masterKey = requireMasterKey();329 const localKeys = deriveSyncIdentityKeypair(masterKey, ourDeviceId);330331 try {332 333 334 335 336 const newlyRevoked = await mergeDeviceRevocations(337 body.revoked.filter((r) => r.deviceId !== ourDeviceId),338 );339 for (const deviceId of newlyRevoked) {340 await removePairedDeviceById(deviceId);341 await removePendingVouchedDevice(deviceId);342 void recordSyncAlerts([343 {344 type: "DEVICE_REVOKED",345 entryId: "",346 entryTitle: deviceId,347 deviceName: issuer.deviceName ?? "paired device",348 timestamp: Date.now(),349 },350 ]);351 }352353 for (const entry of body.devices) {354 if (entry.deviceId === ourDeviceId || entry.deviceId === issuer.deviceId) {355 continue;356 }357 358 359 360 361 362 if (await isDeviceRevoked(entry.deviceId, entry.longTermPublicKeyHex)) {363 continue;364 }365 const existing = await findPairedDeviceById(entry.deviceId);366 const meshRoute = deriveManifestMeshRoute(367 masterKey,368 localKeys.publicKey,369 entry,370 );371372 373 374 375 376 377 if (!existing) {378 if (!meshRoute) continue;379 await upsertPairedDevice({380 deviceId: entry.deviceId,381 deviceName: entry.deviceName,382 deviceType: entry.deviceType,383 longTermPublicKeyHex: entry.longTermPublicKeyHex.toLowerCase(),384 signingPublicKeyHex: entry.signingPublicKeyHex.toLowerCase(),385 wsPort: entry.wsPort,386 rendezvousSecretHex: meshRoute.rendezvousSecretHex,387 relayUrl: meshRoute.relayUrl,388 pairedAt: entry.pairedAt,389 trustSource: "vouched",390 });391 await removePendingVouchedDevice(entry.deviceId);392 continue;393 }394395 396 397 398 399 if (400 existing.longTermPublicKeyHex &&401 !constantTimeEqualHex(402 existing.longTermPublicKeyHex,403 entry.longTermPublicKeyHex,404 )405 ) {406 407 408 409 410 411 const knownName = existing.deviceName ?? entry.deviceName;412 notifyPopup({413 type: "DEVICE_KEY_ROTATED",414 payload: { deviceName: knownName, deviceId: entry.deviceId },415 });416 void recordSyncAlerts([417 {418 type: "DEVICE_KEY_ROTATED",419 entryId: "",420 entryTitle: knownName,421 deviceName: knownName,422 timestamp: Date.now(),423 },424 ]);425 }426427 428 429 430 431 await upsertPairedDevice({432 ...existing,433 deviceId: entry.deviceId,434 deviceName: existing.deviceName ?? entry.deviceName,435 deviceType: entry.deviceType,436 longTermPublicKeyHex: entry.longTermPublicKeyHex.toLowerCase(),437 signingPublicKeyHex: entry.signingPublicKeyHex.toLowerCase(),438 pairedAt: existing.pairedAt ?? entry.pairedAt,439 trustSource: existing.trustSource ?? "direct",440 wsPort: existing.wsPort ?? entry.wsPort,441 rendezvousSecretHex:442 existing.trustSource === "vouched" && meshRoute443 ? meshRoute.rendezvousSecretHex444 : existing.rendezvousSecretHex,445 relayUrl:446 existing.trustSource === "vouched" && meshRoute447 ? meshRoute.relayUrl448 : existing.relayUrl,449 lastSyncedAt: existing.lastSyncedAt,450 });451 }452 } finally {453 masterKey.fill(0);454 localKeys.secretKey.fill(0);455 }456}457458function deriveManifestMeshRoute(459 masterKey: Uint8Array,460 localPublicKey: Uint8Array,461 entry: ManifestDeviceEntry,462): { rendezvousSecretHex: string; relayUrl: string } | null {463 try {464 if (!entry.relayUrl) return null;465 const relayUrl = normalizeRelayUrl(entry.relayUrl);466 const secret = deriveMeshRendezvousSecret(467 masterKey,468 localPublicKey,469 hexToPubkey(entry.longTermPublicKeyHex),470 );471 try {472 return {473 rendezvousSecretHex: rendezvousBytesToHex(secret),474 relayUrl,475 };476 } finally {477 secret.fill(0);478 }479 } catch {480 return null;481 }482}483484function notifyPopup(msg: { type: string; payload?: unknown }): void {485 try {486 void chrome.runtime.sendMessage(msg).catch(() => undefined);487 } catch {488 489 }490}491492function requireUnlocked(): void {493 if (!isUnlocked()) throw new VaultLockedError();494}495496497498499500501502function requireMasterKey(): Uint8Array {503 if (!isUnlocked()) throw new VaultLockedError();504 const hex = getRootKeyHex();505 if (!hex) throw new VaultLockedError();506 return hexToBytes(hex);507}508509async function deriveEntryKey(masterKey: Uint8Array, entryId: string): Promise<CryptoKey> {510 511 512 513 514 const info = new TextEncoder().encode(`${entryId}\x00${ENTRY_KEY_INFO_SUFFIX}`);515 const raw = new HKDF(SHA256, masterKey, new Uint8Array(0), info).expand(32);516 return crypto.subtle.importKey(517 "raw",518 raw.buffer as ArrayBuffer,519 { name: "AES-GCM" },520 false,521 ["encrypt", "decrypt"],522 );523}524525async function encryptEntry(key: CryptoKey, entry: VaultEntry): Promise<string> {526 const plaintext = new TextEncoder().encode(JSON.stringify(entry));527 const nonce = crypto.getRandomValues(new Uint8Array(12));528 const aad = new TextEncoder().encode(ENTRY_AAD);529 const ct = await crypto.subtle.encrypt(530 {531 name: "AES-GCM",532 iv: nonce.buffer as ArrayBuffer,533 additionalData: aad.buffer as ArrayBuffer,534 tagLength: 128,535 },536 key,537 plaintext.buffer as ArrayBuffer,538 );539 const ctBytes = new Uint8Array(ct);540 const envelope = new Uint8Array(nonce.length + ctBytes.length);541 envelope.set(nonce, 0);542 envelope.set(ctBytes, nonce.length);543 return bytesToBase64(envelope);544}545546async function decryptEntryOrNull(key: CryptoKey, envelopeB64: string): Promise<VaultEntry | null> {547 try {548 const bytes = base64ToBytes(envelopeB64);549 if (bytes.length < 12 + 16) return null;550 const nonce = bytes.subarray(0, 12);551 const ct = bytes.subarray(12);552 const aad = new TextEncoder().encode(ENTRY_AAD);553 const pt = await crypto.subtle.decrypt(554 {555 name: "AES-GCM",556 iv: nonce.buffer.slice(nonce.byteOffset, nonce.byteOffset + nonce.byteLength) as ArrayBuffer,557 additionalData: aad.buffer as ArrayBuffer,558 tagLength: 128,559 },560 key,561 ct.buffer.slice(ct.byteOffset, ct.byteOffset + ct.byteLength) as ArrayBuffer,562 );563 const json = new TextDecoder().decode(pt);564 const parsed = JSON.parse(json);565 if (!parsed || typeof parsed !== "object" || typeof parsed.id !== "string") return null;566 return parsed as VaultEntry;567 } catch {568 return null;569 }570}571572function hexToBytes(hex: string): Uint8Array {573 if (hex.length % 2 !== 0) throw new Error("hex length must be even");574 const out = new Uint8Array(hex.length / 2);575 for (let i = 0; i < out.length; i++) {576 out[i] = parseInt(hex.slice(i * 2, i * 2 + 2), 16);577 }578 return out;579}580581function bytesToBase64(bytes: Uint8Array): string {582 let bin = "";583 for (let i = 0; i < bytes.length; i++) bin += String.fromCharCode(bytes[i]);584 return btoa(bin);585}586587function base64ToBytes(b64: string): Uint8Array {588 const bin = atob(b64);589 const out = new Uint8Array(bin.length);590 for (let i = 0; i < bin.length; i++) out[i] = bin.charCodeAt(i);591 return out;592}