Merge pull request #12677 from NejcZdovc/hotfix/#12672-seed

Fixes recovery seed when you recover wallet with funds

Author: bsclifton

app/browser/api/ledger.js

@@ -1011,8 +1011,7 @@ const onWalletRecovery = (state, error, result) => {
     // convert buffer to Uint8Array
     let seed = result && result.getIn(['properties', 'wallet', 'keyinfo', 'seed'])
     if (seed) {
-      seed = new Uint8Array(Object.values(seed))
-      result = result.setIn(['properties', 'wallet', 'keyinfo', 'seed'], seed)
+      result = result.setIn(['properties', 'wallet', 'keyinfo', 'seed'], uintKeySeed(seed))
     }
 
     callback(error, result)
@@ -1637,12 +1636,29 @@ const onBraveryProperties = (state, error, result) => {
   }
 
   if (result) {
+    if (result.properties && result.properties.wallet && result.properties.wallet.keyinfo) {
+      result.properties.wallet.keyinfo.seed = uintKeySeed(result.properties.wallet.keyinfo.seed)
+    }
     muonWriter(statePath, result)
   }
 
   return state
 }
 
+const uintKeySeed = (currentSeed) => {
+  if (currentSeed == null) {
+    return currentSeed
+  }
+
+  try {
+    currentSeed.toJSON()
+    const seed = new Uint8Array(Object.values(currentSeed))
+    currentSeed = seed
+  } catch (err) { }
+
+  return currentSeed
+}
+
 const getBalance = (state) => {
   if (!client) return
 
@@ -1678,6 +1694,11 @@ const onCallback = (state, result, delayTime) => {
     return state
   }
 
+  const seed = result.getIn(['properties', 'wallet', 'keyinfo', 'seed'])
+  if (seed) {
+    result = result.setIn(['properties', 'wallet', 'keyinfo', 'seed'], uintKeySeed(seed))
+  }
+
   const regularResults = result.toJS()
 
   if (client && result.getIn(['properties', 'wallet'])) {
@@ -2591,7 +2612,8 @@ const getMethods = () => {
       observeTransactions,
       onWalletRecovery,
       getStateInfo,
-      lockInContributionAmount
+      lockInContributionAmount,
+      uintKeySeed
     }
   }
 

test/unit/app/browser/api/ledgerTest.js

@@ -1711,4 +1711,58 @@ describe('ledger api unit tests', function () {
       })
     })
   })
+
+  describe('uintKeySeed', function () {
+    const buff = Buffer.from([
+      32,
+      87,
+      30,
+      26,
+      223,
+      56,
+      224,
+      31,
+      213,
+      136,
+      248,
+      95,
+      136,
+      56,
+      250,
+      78,
+      179,
+      121,
+      255,
+      162,
+      195,
+      39,
+      143,
+      136,
+      18,
+      140,
+      49,
+      216,
+      221,
+      154,
+      78,
+      173
+    ])
+
+    const uint = new Uint8Array(Object.values(buff))
+
+    it('null case', function () {
+      const result = ledgerApi.uintKeySeed()
+      assert.equal(result, undefined)
+    })
+
+    it('seed is in the correct format', function () {
+      const result = ledgerApi.uintKeySeed(uint)
+      assert.deepStrictEqual(result, uint)
+    })
+
+    it('seed needs to be converted', function () {
+      const result = ledgerApi.uintKeySeed(buff)
+      assert.deepStrictEqual(result, uint)
+    })
+  })
 })