Add logic for loading Airbitz data from disk

Author: swansontec

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ## Unreleased
 
+- added: `EdgeContextOptions.airbitzSupport`, for loading legacy Airbitz data from disk.
 - fixed: Export the `EdgeObjectTemplate` type.
 - fixed: TypeScript v5 compatibility.
 

src/core/login/airbitz-stashes.ts

@@ -0,0 +1,111 @@
+import { asCodec, asObject, asOptional, asString, Cleaner } from 'cleaners'
+import { justFolders, navigateDisklet } from 'disklet'
+
+import { fixUsername } from '../../client-side'
+import { asBase32, asEdgeBox, asEdgeSnrp } from '../../types/server-cleaners'
+import { EdgeIo } from '../../types/types'
+import { base58, utf8 } from '../../util/encoding'
+import { makeJsonFile } from '../../util/file-helpers'
+import { userIdSnrp } from '../scrypt/scrypt-selectors'
+import { LoginStash } from './login-stash'
+
+/**
+ * Reads legacy Airbitz login stashes from disk.
+ */
+export async function loadAirbitzStashes(
+  io: EdgeIo,
+  avoidUsernames: Set<string>
+): Promise<LoginStash[]> {
+  const out: LoginStash[] = []
+
+  const paths = await io.disklet.list('Accounts').then(justFolders)
+  for (const path of paths) {
+    const folder = navigateDisklet(io.disklet, path)
+    const [
+      carePackage,
+      loginPackage,
+      otp,
+      pin2Key,
+      recovery2Key,
+      usernameJson
+    ] = await Promise.all([
+      await carePackageFile.load(folder, 'CarePackage.json'),
+      await loginPackageFile.load(folder, 'LoginPackage.json'),
+      await otpFile.load(folder, 'OtpKey.json'),
+      await pin2KeyFile.load(folder, 'Pin2Key.json'),
+      await recovery2KeyFile.load(folder, 'Recovery2Key.json'),
+      await usernameFile.load(folder, 'UserName.json')
+    ])
+
+    if (usernameJson == null) continue
+    const username = fixUsername(usernameJson.userName)
+    if (avoidUsernames.has(username)) continue
+    const userId = await io.scrypt(
+      utf8.parse(username),
+      userIdSnrp.salt_hex,
+      userIdSnrp.n,
+      userIdSnrp.r,
+      userIdSnrp.p,
+      32
+    )
+
+    // Assemble a modern stash object:
+    const stash: LoginStash = {
+      appId: '',
+      loginId: userId,
+      pendingVouchers: [],
+      username
+    }
+    if (carePackage != null && loginPackage != null) {
+      stash.passwordKeySnrp = carePackage.SNRP2
+      stash.passwordBox = loginPackage.EMK_LP2
+      stash.syncKeyBox = loginPackage.ESyncKey
+      stash.passwordAuthBox = loginPackage.ELP1
+    }
+    if (otp != null) {
+      stash.otpKey = otp.TOTP
+    }
+    if (pin2Key != null) {
+      stash.pin2Key = pin2Key.pin2Key
+    }
+    if (recovery2Key != null) {
+      stash.recovery2Key = recovery2Key.recovery2Key
+    }
+
+    out.push(stash)
+  }
+
+  return out
+}
+
+/**
+ * A string of base58-encoded binary data.
+ */
+const asBase58: Cleaner<Uint8Array> = asCodec(
+  raw => base58.parse(asString(raw)),
+  clean => base58.stringify(clean)
+)
+
+const carePackageFile = makeJsonFile(
+  asObject({
+    SNRP2: asEdgeSnrp, // passwordKeySnrp
+    SNRP3: asOptional(asEdgeSnrp), // recoveryKeySnrp
+    SNRP4: asOptional(asEdgeSnrp), // questionKeySnrp
+    ERQ: asOptional(asEdgeBox) // questionBox
+  })
+)
+
+const loginPackageFile = makeJsonFile(
+  asObject({
+    EMK_LP2: asEdgeBox, // passwordBox
+    EMK_LRA3: asOptional(asEdgeBox), // recoveryBox
+
+    ESyncKey: asEdgeBox, // syncKeyBox
+    ELP1: asEdgeBox // passwordAuthBox
+  })
+)
+
+const otpFile = makeJsonFile(asObject({ TOTP: asBase32 }))
+const pin2KeyFile = makeJsonFile(asObject({ pin2Key: asBase58 }))
+const recovery2KeyFile = makeJsonFile(asObject({ recovery2Key: asBase58 }))
+const usernameFile = makeJsonFile(asObject({ userName: asString }))

src/core/root.ts

@@ -9,6 +9,7 @@ import { Dispatch } from './actions'
 import { CLIENT_FILE_NAME, clientFile } from './context/client-file'
 import { INFO_CACHE_FILE_NAME, infoCacheFile } from './context/info-cache-file'
 import { filterLogs, LogBackend, makeLog } from './log/log'
+import { loadAirbitzStashes } from './login/airbitz-stashes'
 import { loadStashes } from './login/login-stash'
 import { PluginIos, watchPlugins } from './plugins/plugins-actions'
 import { RootOutput, rootPixie, RootProps } from './root-pixie'
@@ -33,15 +34,16 @@ export async function makeContext(
 ): Promise<EdgeContext> {
   const { io } = ios
   const {
+    airbitzSupport = false,
     apiKey,
     appId = '',
     authServer = 'https://login.edge.app/api',
-    infoServer,
-    syncServer,
     deviceDescription = null,
     hideKeys = false,
+    infoServer,
     plugins: pluginsInit = {},
-    skipBlockHeight = false
+    skipBlockHeight = false,
+    syncServer
   } = opts
   const infoServers =
     typeof infoServer === 'string'
@@ -82,19 +84,35 @@ export async function makeContext(
   })
   const log = makeLog(logBackend, 'edge-core')
 
+  // Load the login stashes from disk:
   let [clientInfo, infoCache = {}, stashes] = await Promise.all([
     clientFile.load(io.disklet, CLIENT_FILE_NAME),
     infoCacheFile.load(io.disklet, INFO_CACHE_FILE_NAME),
     loadStashes(io.disklet, log)
   ])
 
+  // Load legacy stashes from disk
+  if (airbitzSupport) {
+    // Edge will write modern files to disk at login time,
+    // but it won't delete the legacy Airbitz data.
+    // Once this happens, we need to ignore the legacy files
+    // and just use the new files:
+    const avoidUsernames = new Set<string>()
+    for (const { username } of stashes) {
+      if (username != null) avoidUsernames.add(username)
+    }
+
+    const airbitzStashes = await loadAirbitzStashes(io, avoidUsernames)
+    stashes.push(...airbitzStashes)
+  }
+
   // Save the clientId if we don't have one:
   if (clientInfo == null) {
     clientInfo = { clientId: io.random(16) }
     await clientFile.save(io.disklet, CLIENT_FILE_NAME, clientInfo)
   }
 
-  // Load the login stashes from disk:
+  // Write everything to redux:
   redux.dispatch({
     type: 'INIT',
     payload: {

src/react-native.tsx

@@ -36,6 +36,7 @@ export function MakeEdgeContext(props: EdgeContextProps): JSX.Element {
     onLog = defaultOnLog,
 
     // Inner context options:
+    airbitzSupport = false,
     apiKey = '',
     appId = '',
     authServer,
@@ -63,6 +64,7 @@ export function MakeEdgeContext(props: EdgeContextProps): JSX.Element {
           bridgifyLogBackend({ crashReporter, onLog }),
           pluginUris,
           {
+            airbitzSupport,
             apiKey,
             appId,
             authServer,

src/types/exports.ts

@@ -76,6 +76,7 @@ export interface EdgeContextProps extends CommonProps {
   onLog?: EdgeOnLog
 
   // EdgeContextOptions:
+  airbitzSupport?: boolean
   apiKey?: string
   appId?: string
   authServer?: string

src/types/types.ts

@@ -1683,6 +1683,9 @@ export interface EdgeContextOptions {
   path?: string // Only used on node.js
   plugins?: EdgeCorePluginsInit
 
+  /** True to load Airbitz user files from disk */
+  airbitzSupport?: boolean
+
   /**
    * True to skip updating the `EdgeCurrencyWallet.blockHeight` property.
    * This may improve performance by reducing bridge traffic,
@@ -1857,6 +1860,7 @@ export interface EdgeFakeWorldOptions {
 
 export interface EdgeFakeContextOptions {
   // EdgeContextOptions:
+  airbitzSupport?: boolean
   apiKey: string
   appId: string
   deviceDescription?: string