Support BIP32 HD Wallets

1 files changed, 152 insertions, 43 deletions

diff --git a/public/js/setup.coffee b/public/js/setup.coffee
index 7daf960f..fdba60df 100644
--- a/public/js/setup.coffee
+++ b/public/js/setup.coffee
@@ -1,16 +1,28 @@
 #= require ../js/jquery-1.8.2.min.js
 #= require ../js/jquery-ui.min.js
+#= require ../js/jquery.printElement.min.js
 #= require ../js/bootstrap.min.js
+#= require ../js/crypto-min.js
 #= require ../js/2.5.3-crypto-sha256.js
 #= require ../js/jsbn.js
 #= require ../js/jsbn2.js
 #= require ../js/check_address.js
+#= require ../js/bitcoinjs-min.js
+#= require ../js/sha512.js
+#= require ../js/modsqrt.js
+#= require ../js/rfc1751.js
+#= require ../js/bip32.js
+#= require ../js/secure_random.js
+#= require ../js/qrcode.js
+#= require ../js/aes.js
 
 g = exports ? this
 g.proceed = false
 
 $(->
+  $('[data-toggle=popover]').popover()
   $('#title').focus()
+
   $.getJSON("/js/rates.json", (data) ->
     currencies = Object.keys(data)
     currencies = currencies.sort()
@@ -21,8 +33,6 @@ $(->
     )
     $("#currency option[value='CAD']").attr('selected', 'selected')
 
-    user = $('#username').val()
-
     $('#currency').change(->
       $('#symbol option').remove()
       symbol = $(this).val()
@@ -37,54 +47,153 @@ $(->
           $("#symbol option[value='quadrigacx']").attr('selected', 'selected')
         when 'USD'
           $("#symbol option[value='bitstamp']").attr('selected', 'selected')
-
-    ).change()
-
-    units = ['BTC', 'mBTC', 'µBTC', 'satoshis']
-    $.each(units, (i, v) ->
-      $('#unit').append("<option value='#{v}'>#{v}</option>")
     )
 
-    $('#title').blur(->
-      if $(this).val() == ''
-        $(this).parent().addClass('has-error')
-      else
-          $(this).parent().removeClass('has-error')
+    user = $('#username').val()
+    $.getJSON("/#{user}.json", (data) ->
+      $('#title').val(data.title)
+      $('#logo').val(data.logo)
+      $('#bip32').val(data.bip32)
+      $('#address').val(data.address)
+      $("#symbol option[value='#{data.symbol}']").attr('selected', 'selected')
+      $('#commission').val(data.commission)
+      $('#unit').val(data.unit)
+      $('#currency').change()
+      $('#setup').fadeIn()
     )
+  )
+
+  units = ['BTC', 'mBTC', '&micro;BTC', 'satoshis']
+  $.each(units, (i, v) ->
+    $('#unit').append("<option value='#{v}'>#{v}</option>")
+  )
 
-    $('#address').blur(->
-      if check_address($(this).val())
+  $('#title').blur(->
+    if $(this).val() == '' and $('#logo').val() == ''
+      $(this).parent().addClass('has-error')
+    else
         $(this).parent().removeClass('has-error')
-      else
-        $(this).parent().addClass('has-error')
-    )
+  )
 
-    $('#confirm').blur(->
-      $('#confirm_error').remove() 
-      if $('#password').val() != $('#confirm').val()
-        $('#password, #confirm').parent().addClass('has-error')
-        $('#confirm').parent().after('<div id="confirm_error" class="alert alert-danger">Passwords don\'t match</div>')
-      else
-        $('#password, #confirm').parent().removeClass('has-error')
-    )
+  $('#bip32').blur(->
+    try
+      new BIP32($(this).val())
+      $(this).parent().removeClass('has-error')
+    catch
+      $(this).parent().addClass('has-error')
+  )
 
-    $('#setup').submit(->
-      $('.form-control').blur()
-      if $('.has-error').length > 0
-        $('.has-error').effect('shake', 500)
-        return false
-    )
+  $('#generate').click(->
+    rng = new SecureRandom()
+    bytes = new Array(256)
+    rng.nextBytes(bytes)
 
-    if user
-      $('#setup').attr('action', "/#{user}").attr('method', 'post')
-      $.getJSON("/#{user}.json", (data) ->
-        $('#title').val(data.title)
-        $('#address').val(data.address)
-        $('#logo').val(data.logo)
-        $("#symbol option[value='#{data.symbol}']").attr('selected', 'selected')
-        $('#commission').val(data.commission)
-        $('#unit').val(data.unit)
-        $('#setup').fadeIn()
-      )
+    hasher = new jsSHA(bytes.toString(), 'TEXT')   
+    rng.nextBytes(bytes)
+
+    I = hasher.getHMAC(bytes.toString(), "TEXT", "SHA-512", "HEX")
+    il = Crypto.util.hexToBytes(I.slice(0, 64))
+    ir = Crypto.util.hexToBytes(I.slice(64, 128))
+
+    key = new BIP32()
+    key.eckey = new Bitcoin.ECKey(il)
+    key.eckey.pub = key.eckey.getPubPoint()
+    key.eckey.setCompressed(true)
+    key.eckey.pubKeyHash = Bitcoin.Util.sha256ripe160(key.eckey.pub.getEncoded(true))
+    key.has_private_key = true
+
+    key.chain_code = ir
+    key.child_index = 0
+    key.parent_fingerprint = Bitcoin.Util.hexToBytes("00000000")
+    key.version = BITCOIN_MAINNET_PRIVATE
+    key.depth = 0
+
+    key.build_extended_public_key()
+    key.build_extended_private_key()
+
+    g.pubkey = key.extended_public_key_string()
+    g.privkey = key.extended_private_key_string()
+
+    $('#modal').modal()
+    $('#privkey').html(g.privkey)
+    $('#qr').html('')
+    new QRCode('qr', text: g.privkey, width: 260, height: 260)
+  )
+
+  $('#encryption-password').keyup((e) ->
+    return if e.keyCode is 9 or e.keyCode is 16
+    enc_privkey = CryptoJS.AES.encrypt(g.privkey, $(this).val()).toString()
+    enc_privkey = g.privkey if $(this).val() == ''
+    $('#privkey').html(enc_privkey)
+    $('#qr').html('')
+    new QRCode('qr', text: enc_privkey, width: 260, height: 260)
+
+    if $(this).val()
+      $('#status').html('(Encrypted)')
+    else
+      $('#status').html('(Unencrypted Plaintext)')
+  )
+
+  $('#confirm').blur(->
+    $('#confirm_error').remove() 
+    if $('#password').val() != $('#confirm').val()
+      $('#password, #confirm').parent().addClass('has-error')
+      $('#confirm').parent().after('<div id="confirm_error" class="alert alert-danger">Passwords don\'t match</div>')
+    else
+      $('#password, #confirm').parent().removeClass('has-error')
+  )
+
+  $('#encryption-password, #encryption-confirm').keyup((e) -> check_passwords(e) unless e.keyCode is 9)
+  $('#encryption-password, #encryption-confirm').blur((e) -> check_passwords(e))
+
+  check_passwords = (e) ->
+    if $('#encryption-password').val() != $('#encryption-confirm').val() or $('#encryption-password').val() is ''
+      $('#encryption-password, #encryption-confirm').parent().addClass('has-error')
+      $('#modal .alert').removeClass('alert-success').addClass('alert-danger')
+    else
+      $('#encryption-password, #encryption-confirm').parent().removeClass('has-error')
+      $('#modal .alert').removeClass('alert-danger').addClass('alert-success')
+
+  $('#print').click(->
+    $('#key-dialog').printElement(printMode: 'popup')
+  )
+
+  $('a[data-toggle="tab"]').on('shown.bs.tab', (e) ->
+    $('#encryption-password').focus()
+    switch e.target.attributes.href.value
+      when '#step1'
+        $('.modal-footer button:first').hide()
+        $('.modal-footer button:last').html('Next').off().on('click', -> $('a[data-toggle="tab"]:eq(1)').click())
+      when '#step2'
+        $('.modal-footer button:first').show().html('Back').off().on('click', -> $('a[data-toggle="tab"]:eq(0)').click())
+        $('.modal-footer button:last').html('Next').off().on('click', -> $('a[data-toggle="tab"]:eq(2)').click())
+      when '#step3'
+        $('.modal-footer button:first').show().html('Back').off().on('click', -> $('a[data-toggle="tab"]:eq(1)').click())
+        $('.modal-footer button:last').html('Got it!').off().on('click', -> 
+          $('#modal .form-control').blur()
+          if $('#modal .has-error').length > 0
+            $('a[data-toggle="tab"]:eq(1)').click()
+            $('#modal .has-error').effect('shake', 500)
+          else
+            g.update_pub_key = true
+            $('#modal').modal('hide')
+        )
+  )
+
+  $('.modal-footer button:last').on('click', -> $('a[data-toggle="tab"]:eq(1)').click())
+  $('#modal').on('shown.bs.modal', -> 
+    g.update_pub_key = false
+    $('#encryption-password, #encryption-confirm').val('').blur()
+    if $('.tab-pane:eq(2)').hasClass('active')
+      $('a[data-toggle="tab"]:eq(1)').click()
+  ).on('hidden.bs.modal', ->
+    $('#bip32').val(g.pubkey) if g.update_pub_key
+  )
+
+  $('#setup').submit(->
+    $('#setup .form-control').blur()
+    if $('#setup .has-error').length > 0
+      $('#setup .has-error').effect('shake', 500)
+      return false
   )
 )