Hardened Addresses checkbox

[perso/Immae/Projets/Cryptomonnaies/BIP39.git] / src / js / index.js

diff --git a/src/js/index.js b/src/js/index.js
index 6f0da65d391a29e58f3dcdc597fc3511039b7e40..68f1cdaa17e0b6213d057db7184aa263e8fa3f29 100644 (file)
--- a/src/js/index.js
+++ b/src/js/index.js
@@ -12,6 +12,7 @@
     var showPrivKey = true;
 
     var phraseChangeTimeoutEvent = null;
+    var rootKeyChangedTimeoutEvent = null;
 
     var DOM = {};
     DOM.network = $(".network");
@@ -34,6 +35,7 @@
     DOM.bip44account = $("#bip44 .account");
     DOM.bip44change = $("#bip44 .change");
     DOM.strength = $(".strength");
+    DOM.hardenedAddresses = $(".hardened-addresses");
     DOM.addresses = $(".addresses");
     DOM.rowsToAdd = $(".rows-to-add");
     DOM.more = $(".more");
@@ -50,12 +52,14 @@
         DOM.passphrase.on("input", delayedPhraseChanged);
         DOM.generate.on("click", generateClicked);
         DOM.more.on("click", showMore);
-        DOM.bip32path.on("input", delayedPhraseChanged);
-        DOM.bip44purpose.on("input", delayedPhraseChanged);
-        DOM.bip44coin.on("input", delayedPhraseChanged);
-        DOM.bip44account.on("input", delayedPhraseChanged);
-        DOM.bip44change.on("input", delayedPhraseChanged);
-        DOM.tab.on("click", delayedPhraseChanged);
+        DOM.rootKey.on("input", delayedRootKeyChanged);
+        DOM.bip32path.on("input", calcForDerivationPath);
+        DOM.bip44purpose.on("input", calcForDerivationPath);
+        DOM.bip44coin.on("input", calcForDerivationPath);
+        DOM.bip44account.on("input", calcForDerivationPath);
+        DOM.bip44change.on("input", calcForDerivationPath);
+        DOM.tab.on("shown.bs.tab", calcForDerivationPath);
+        DOM.hardenedAddresses.on("change", calcForDerivationPath);
         DOM.indexToggle.on("click", toggleIndexes);
         DOM.addressToggle.on("click", toggleAddresses);
         DOM.privateKeyToggle.on("click", togglePrivateKeys);
@@ -70,7 +74,7 @@
     function networkChanged(e) {
         var network = e.target.value;
         networks[network].onSelect();
-        delayedPhraseChanged();
+        displayBip32Info();
     }
 
     function delayedPhraseChanged() {
@@ -87,12 +91,57 @@
         hideValidationError();
         // Get the mnemonic phrase
         var phrase = DOM.phrase.val();
-        var passphrase = DOM.passphrase.val();
         var errorText = findPhraseErrors(phrase);
         if (errorText) {
             showValidationError(errorText);
             return;
         }
+        // Calculate and display
+        var passphrase = DOM.passphrase.val();
+        calcBip32RootKeyFromSeed(phrase, passphrase);
+        calcForDerivationPath();
+        hidePending();
+    }
+
+    function delayedRootKeyChanged() {
+        // Warn if there is an existing mnemonic or passphrase.
+        if (DOM.phrase.val().length > 0 || DOM.passphrase.val().length > 0) {
+            if (!confirm("This will clear existing mnemonic and passphrase")) {
+                DOM.rootKey.val(bip32RootKey);
+                return
+            }
+        }
+        hideValidationError();
+        showPending();
+        // Clear existing mnemonic and passphrase
+        DOM.phrase.val("");
+        DOM.passphrase.val("");
+        seed = null;
+        if (rootKeyChangedTimeoutEvent != null) {
+            clearTimeout(rootKeyChangedTimeoutEvent);
+        }
+        rootKeyChangedTimeoutEvent = setTimeout(rootKeyChanged, 400);
+    }
+
+    function rootKeyChanged() {
+        showPending();
+        hideValidationError();
+        // Validate the root key TODO
+        var rootKeyBase58 = DOM.rootKey.val();
+        var errorText = validateRootKey(rootKeyBase58);
+        if (errorText) {
+            showValidationError(errorText);
+            return;
+        }
+        // Calculate and display
+        calcBip32RootKeyFromBase58(rootKeyBase58);
+        calcForDerivationPath();
+        hidePending();
+    }
+
+    function calcForDerivationPath() {
+        showPending();
+        hideValidationError();
         // Get the derivation path
         var derivationPath = getDerivationPath();
         var errorText = findDerivationPathErrors(derivationPath);
@@ -100,8 +149,7 @@
             showValidationError(errorText);
             return;
         }
-        // Calculate and display
-        calcBip32Seed(phrase, passphrase, derivationPath);
+        calcBip32ExtendedKey(derivationPath);
         displayBip32Info();
         hidePending();
     }
@@ -148,9 +196,16 @@
         return words;
     }
 
-    function calcBip32Seed(phrase, passphrase, path) {
+    function calcBip32RootKeyFromSeed(phrase, passphrase) {
         seed = mnemonic.toSeed(phrase, passphrase);
         bip32RootKey = bitcoin.HDNode.fromSeedHex(seed, network);
+    }
+
+    function calcBip32RootKeyFromBase58(rootKeyBase58) {
+        bip32RootKey = bitcoin.HDNode.fromBase58(rootKeyBase58);
+    }
+
+    function calcBip32ExtendedKey(path) {
         bip32ExtendedKey = bip32RootKey;
         // Derive the key from the path
         var pathBits = path.split("/");
@@ -195,8 +250,16 @@
                 proper.push(part.toLowerCase());
             }
         }
-        // TODO some levenstein on the words
         var properPhrase = proper.join(' ');
+        // Check each word
+        for (var i=0; i<proper.length; i++) {
+            var word = proper[i];
+            if (WORDLISTS["english"].indexOf(word) == -1) {
+                console.log("Finding closest match to " + word);
+                var nearestWord = findNearestWord(word);
+                return word + " not in wordlist, did you mean " + nearestWord + "?";
+            }
+        }
         // Check the words are valid
         var isValid = mnemonic.check(properPhrase);
         if (!isValid) {
@@ -205,6 +268,16 @@
         return false;
     }
 
+    function validateRootKey(rootKeyBase58) {
+        try {
+            bitcoin.HDNode.fromBase58(rootKeyBase58);
+        }
+        catch (e) {
+            return "Invalid root key";
+        }
+        return "";
+    }
+
     function getDerivationPath() {
         if (DOM.bip44tab.hasClass("active")) {
             var purpose = parseIntNoNaN(DOM.bip44purpose.val(), 44);
@@ -291,16 +364,27 @@
 
     function TableRow(index) {
 
+        var useHardenedAddresses = DOM.hardenedAddresses.prop("checked");
+
         function init() {
             calculateValues();
         }
 
         function calculateValues() {
             setTimeout(function() {
-                var key = bip32ExtendedKey.derive(index);
+                var key = "";
+                if (useHardenedAddresses) {
+                    key = bip32ExtendedKey.deriveHardened(index);
+                }
+                else {
+                    key = bip32ExtendedKey.derive(index);
+                }
                 var address = key.getAddress().toString();
                 var privkey = key.privKey.toWIF(network);
                 var indexText = getDerivationPath() + "/" + index;
+                if (useHardenedAddresses) {
+                    indexText = indexText + "'";
+                }
                 addAddressToList(indexText, address, privkey);
             }, 50)
         }
@@ -389,6 +473,21 @@
             .show();
     }
 
+    function findNearestWord(word) {
+        var words = WORDLISTS["english"];
+        var minDistance = 99;
+        var closestWord = words[0];
+        for (var i=0; i<words.length; i++) {
+            var comparedTo = words[i];
+            var distance = Levenshtein.get(word, comparedTo);
+            if (distance < minDistance) {
+                closestWord = comparedTo;
+                minDistance = distance;
+            }
+        }
+        return closestWord;
+    }
+
     function hidePending() {
         DOM.feedback
             .text("")