From 313acacedfb6a8e1e513c1d0978b6856bb3563a6 Mon Sep 17 00:00:00 2001
From: Niels Andriesse <andriesseniels@gmail.com>
Date: Wed, 21 Aug 2019 09:50:03 +1000
Subject: [PATCH] Guard against invalid seeds

---
 .../thoughtcrime/securesms/loki/SeedActivity.kt   | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/src/org/thoughtcrime/securesms/loki/SeedActivity.kt b/src/org/thoughtcrime/securesms/loki/SeedActivity.kt
index 4a3387c8c4..b9475ea207 100644
--- a/src/org/thoughtcrime/securesms/loki/SeedActivity.kt
+++ b/src/org/thoughtcrime/securesms/loki/SeedActivity.kt
@@ -76,7 +76,13 @@ class SeedActivity : BaseActionBarActivity() {
 
     // region Updating
     private fun updateSeed() {
-        seed = Curve25519.getInstance("best").generateSeed(16)
+        val seed = Curve25519.getInstance(Curve25519.BEST).generateSeed(16)
+        try {
+            IdentityKeyUtil.generateIdentityKeyPair(this, seed + seed)
+        } catch (exception: Exception) {
+            return updateSeed()
+        }
+        this.seed = seed
     }
 
     private fun updateUI() {
@@ -138,9 +144,12 @@ class SeedActivity : BaseActionBarActivity() {
                 }
             }
         }
-        IdentityKeyUtil.save(this, IdentityKeyUtil.lokiSeedKey, Hex.toStringCondensed(seed))
+        val hexEncodedSeed = Hex.toStringCondensed(seed)
+        IdentityKeyUtil.save(this, IdentityKeyUtil.lokiSeedKey, hexEncodedSeed)
         if (seed.count() == 16) seed = seed + seed
-        IdentityKeyUtil.generateIdentityKeyPair(this, seed)
+        if (mode == Mode.Restore) {
+            IdentityKeyUtil.generateIdentityKeyPair(this, seed)
+        }
         val keyPair = IdentityKeyUtil.getIdentityKeyPair(this)
         val publicKey = keyPair.publicKey
         val hexEncodedPublicKey = keyPair.hexEncodedPublicKey