bitcoinrb 0.1.3 → 0.1.4

data/lib/bitcoin/wallet/base.rb

@@ -0,0 +1,84 @@
+require 'leveldb'
+module Bitcoin
+  module Wallet
+
+    class Base
+
+      attr_accessor :wallet_id
+      attr_reader :db
+
+      DEFAULT_PATH_PREFIX = "#{Bitcoin.base_dir}/db/wallet"
+
+      # Create new wallet. If wallet already exist, throw error.
+      # The wallet generates a seed using SecureRandom and store to db at initialization.
+      # @param [String] wallet_id new wallet id.
+      # @param [String] path_prefix wallet file path prefix.
+      # @return [Bitcoin::Wallet::Base] the wallet
+      def self.create(wallet_id = 1, path_prefix = DEFAULT_PATH_PREFIX)
+        raise ArgumentError, "wallet_id : #{wallet_id} already exist." if self.exist?(wallet_id, path_prefix)
+        w = self.new(wallet_id, path_prefix)
+        # generate seed
+        raise RuntimeError, 'the seed already exist.' if w.db.registered_master?
+        master = Bitcoin::Wallet::MasterKey.generate
+        w.db.register_master_key(master)
+        w
+      end
+
+      # load wallet with specified +wallet_id+
+      # @return [Bitcoin::Wallet::Base] the wallet
+      def self.load(wallet_id, path_prefix = DEFAULT_PATH_PREFIX)
+        raise ArgumentError, "wallet_id : #{wallet_id} dose not exist." unless self.exist?(wallet_id, path_prefix)
+        self.new(wallet_id, path_prefix)
+      end
+
+      # get account list based on BIP-44
+      def accounts
+        db.accounts.map{|raw| Account.parse_from_payload(raw)}
+      end
+
+      def create_account(purpose = Account::PURPOSE_TYPE[:legacy], index = 0, name)
+        account = Account.new(purpose, index, name)
+        account.wallet = self
+        account.init
+        account
+      end
+
+      # close database wallet
+      def close
+        db.close
+      end
+
+      # get master key
+      # @return [Bitcoin::Wallet::MasterKey]
+      def master_key
+        db.master_key
+      end
+
+      # encrypt wallet
+      # @param [String] passphrase the wallet passphrase
+      def encrypt(passphrase)
+
+      end
+
+      # decrypt wallet
+      # @param [String] passphrase the wallet passphrase
+      def decrypt(passphrase)
+
+      end
+
+      private
+
+      def initialize(wallet_id, path_prefix)
+        @db = Bitcoin::Wallet::DB.new("#{path_prefix}_#{wallet_id}")
+        @wallet_id = wallet_id
+      end
+
+      def self.exist?(wallet_id, path_prefix)
+        path = "#{path_prefix}_#{wallet_id}"
+        Dir.exist?(path)
+      end
+
+    end
+
+  end
+end

data/lib/bitcoin/wallet/db.rb

@@ -0,0 +1,57 @@
+module Bitcoin
+  module Wallet
+
+    class DB
+
+      KEY_PREFIX = {
+          account: 'a',       # key: account index, value: Account raw data.
+          master: 'm',          # value : wallet seed.
+      }
+
+      attr_reader :level_db
+      attr_accessor :master_key
+
+      def initialize(path = "#{Bitcoin.base_dir}/db/wallet")
+        FileUtils.mkdir_p(path)
+        @level_db = ::LevelDB::DB.new(path)
+      end
+
+      # close database
+      def close
+        level_db.close
+      end
+
+      # get accounts raw data.
+      def accounts
+        from = KEY_PREFIX[:account] + '00000000'
+        to = KEY_PREFIX[:account] + 'ffffffff'
+        level_db.each(from: from, to: to).map { |k, v| v}
+      end
+
+      def save_account(account)
+        level_db.batch do
+          key = KEY_PREFIX[:account] + account.index.to_s(16).rjust(8, '0')
+          level_db.put(key, account.to_payload)
+        end
+      end
+
+      # get master_key
+      def master_key
+        @master_key ||= Bitcoin::Wallet::MasterKey.parse_from_payload(level_db.get(KEY_PREFIX[:master]))
+      end
+
+      # save seed
+      # @param [Bitcoin::Wallet::MasterKey] master a master key.
+      def register_master_key(master)
+        level_db.put(KEY_PREFIX[:master], master.to_payload)
+        @master_key = master
+      end
+
+      # whether master key registered.
+      def registered_master?
+        !level_db.get(KEY_PREFIX[:master]).nil?
+      end
+
+    end
+  end
+end

data/lib/bitcoin/wallet/master_key.rb

@@ -0,0 +1,100 @@
+module Bitcoin
+  module Wallet
+
+    # HD Wallet master seed
+    class MasterKey
+      extend Bitcoin::Util
+      include Bitcoin::Util
+
+      attr_reader :seed
+      attr_accessor :salt
+      attr_accessor :encrypted
+      attr_accessor :mnemonic # ephemeral data existing only at initialization
+
+      def initialize(seed, salt: '', encrypted: false, mnemonic: nil)
+        @mnemonic = mnemonic
+        @seed = seed
+        @encrypted = encrypted
+        @salt = salt
+      end
+
+      # generate new master key.
+      # @return Bitcoin::Wallet::MasterKey
+      def self.generate
+        entropy = SecureRandom.hex(16)
+        mnemonic = Bitcoin::Mnemonic.new('english')
+        self.recover_from_words(mnemonic.to_mnemonic(entropy))
+      end
+
+      # recover master key from mnemonic word list.
+      # @param [Array] words the mnemonic word list.
+      # @return Bitcoin::Wallet::MasterKey
+      def self.recover_from_words(words)
+        mnemonic = Bitcoin::Mnemonic.new('english')
+        seed = mnemonic.to_seed(words)
+        self.new(seed, mnemonic: words)
+      end
+
+      # parse master key raw data
+      # @param [String] payload raw data
+      # @return [Bitcoin::Wallet::MasterKey]
+      def self.parse_from_payload(payload)
+        flag, payload = unpack_var_int(payload)
+        raise 'encrypted flag is invalid.' unless [0, 1].include?(flag)
+        salt, payload = unpack_var_string(payload)
+        seed, payload = unpack_var_string(payload)
+        self.new(seed.bth, salt: salt.bth, encrypted: flag == 1)
+      end
+
+      # generate payload with following format
+      # [encrypted(false:0, true:1)][salt(var str)][seed(var str)]
+      def to_payload
+        flg = encrypted ? 1 : 0
+        pack_var_int(flg) << [salt, seed].map{|v|pack_var_string(v.htb)}.join
+      end
+
+      # get master key
+      # @return [Bitcoin::ExtKey] the master key
+      def key
+        raise 'seed is encrypted. please decrypt the seed.' if encrypted
+        Bitcoin::ExtKey.generate_master(seed)
+      end
+
+      # encrypt seed
+      def encrypt(passphrase)
+        raise 'seed already encrypted.' if encrypted
+        @salt = SecureRandom.hex(16)
+        enc = OpenSSL::Cipher.new('AES-256-CBC')
+        enc.encrypt
+        enc.key, enc.iv = key_iv(enc, passphrase)
+        encrypted_data = ''
+        encrypted_data << enc.update(seed)
+        encrypted_data << enc.final
+        @seed = encrypted_data
+        @encrypted = true
+      end
+
+      # decrypt seed
+      def decrypt(passphrase)
+        raise 'seed is not encrypted.' unless encrypted
+        dec = OpenSSL::Cipher.new('AES-256-CBC')
+        dec.decrypt
+        dec.key, dec.iv = key_iv(dec, passphrase)
+        decrypted_data = ''
+        decrypted_data << dec.update(seed)
+        decrypted_data << dec.final
+        @seed = decrypted_data
+        @encrypted = false
+        @salt = ''
+      end
+
+      private
+
+      def key_iv(enc, passphrase)
+        key_iv = OpenSSL::PKCS5.pbkdf2_hmac_sha1(passphrase, salt, 2000, enc.key_len + enc.iv_len)
+        [key_iv[0, enc.key_len], key_iv[enc.key_len, enc.iv_len]]
+      end
+
+    end
+  end
+end

data/lib/bitcoin/wallet.rb

@@ -0,0 +1,8 @@
+module Bitcoin
+  module Wallet
+    autoload :Base, 'bitcoin/wallet/base'
+    autoload :Account, 'bitcoin/wallet/account'
+    autoload :DB, 'bitcoin/wallet/db'
+    autoload :MasterKey, 'bitcoin/wallet/master_key'
+  end
+end

data/lib/bitcoin.rb

@@ -5,6 +5,7 @@ require 'securerandom'
 require 'json'
 require 'bech32'
 require 'ffi'
+require 'tmpdir'
 require_relative 'openassets'
 
 module Bitcoin
@@ -17,6 +18,7 @@ module Bitcoin
   autoload :BlockHeader, 'bitcoin/block_header'
   autoload :Tx, 'bitcoin/tx'
   autoload :Script, 'bitcoin/script/script'
+  autoload :Multisig, 'bitcoin/script/multisig'
   autoload :ScriptInterpreter, 'bitcoin/script/script_interpreter'
   autoload :ScriptError, 'bitcoin/script/script_error'
   autoload :TxChecker, 'bitcoin/script/tx_checker'
@@ -35,6 +37,8 @@ module Bitcoin
   autoload :ValidationState, 'bitcoin/validation'
   autoload :Network, 'bitcoin/network'
   autoload :Store, 'bitcoin/store'
+  autoload :RPC, 'bitcoin/rpc'
+  autoload :Wallet, 'bitcoin/wallet'
 
   require_relative 'bitcoin/constants'
 

data/lib/openassets/payload.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module OpenAssets
 
   MARKER = "\x4f\x41"
@@ -34,9 +36,11 @@ module OpenAssets
 
     # generate binary payload
     def to_payload
-      payload = MARKER << VERSION
+      payload = String.new
+      payload << MARKER
+      payload << VERSION
       payload << Bitcoin.pack_var_int(quantities.size) << quantities.map{|q| LEB128.encode_unsigned(q).read }.join
-      payload << Bitcoin.pack_var_int(metadata.length) << metadata.bytes.map{|b|b.to_s(16)}.join.htb
+      payload << Bitcoin.pack_var_int(metadata.length) << metadata.bytes.map{|b| sprintf("%02x", b)}.join.htb
       payload
     end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: bitcoinrb
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - azuchi
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-10-03 00:00:00.000000000 Z
+date: 2017-12-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ecdsa
@@ -58,14 +58,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.0.1
+        version: 1.0.3
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.0.1
+        version: 1.0.3
 - !ruby/object:Gem::Dependency
   name: daemon-spawn
   requirement: !ruby/object:Gem::Requirement
@@ -109,7 +109,21 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: parallel
+  name: leb128
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+- !ruby/object:Gem::Dependency
+  name: leveldb-ruby
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -123,19 +137,47 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: leb128
+  name: eventmachine_httpserver
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 1.0.0
+        version: '0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 1.0.0
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rest-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: iniparse
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -213,6 +255,7 @@ files:
 - Rakefile
 - bin/console
 - bin/setup
+- bitcoinrb.conf.sample
 - bitcoinrb.gemspec
 - exe/bitcoinrb-cli
 - exe/bitcoinrbd
@@ -249,6 +292,7 @@ files:
 - lib/bitcoin/message/inventory.rb
 - lib/bitcoin/message/mem_pool.rb
 - lib/bitcoin/message/merkle_block.rb
+- lib/bitcoin/message/network_addr.rb
 - lib/bitcoin/message/not_found.rb
 - lib/bitcoin/message/ping.rb
 - lib/bitcoin/message/pong.rb
@@ -273,10 +317,15 @@ files:
 - lib/bitcoin/network/peer_discovery.rb
 - lib/bitcoin/network/pool.rb
 - lib/bitcoin/node.rb
+- lib/bitcoin/node/cli.rb
+- lib/bitcoin/node/configuration.rb
 - lib/bitcoin/node/spv.rb
-- lib/bitcoin/node/spv_block_chain.rb
 - lib/bitcoin/opcodes.rb
 - lib/bitcoin/out_point.rb
+- lib/bitcoin/rpc.rb
+- lib/bitcoin/rpc/http_server.rb
+- lib/bitcoin/rpc/request_handler.rb
+- lib/bitcoin/script/multisig.rb
 - lib/bitcoin/script/script.rb
 - lib/bitcoin/script/script_error.rb
 - lib/bitcoin/script/script_interpreter.rb
@@ -286,13 +335,21 @@ files:
 - lib/bitcoin/secp256k1/native.rb
 - lib/bitcoin/secp256k1/ruby.rb
 - lib/bitcoin/store.rb
-- lib/bitcoin/store/spv_block_store.rb
+- lib/bitcoin/store/chain_entry.rb
+- lib/bitcoin/store/db.rb
+- lib/bitcoin/store/db/level_db.rb
+- lib/bitcoin/store/spv_chain.rb
 - lib/bitcoin/tx.rb
 - lib/bitcoin/tx_in.rb
 - lib/bitcoin/tx_out.rb
 - lib/bitcoin/util.rb
 - lib/bitcoin/validation.rb
 - lib/bitcoin/version.rb
+- lib/bitcoin/wallet.rb
+- lib/bitcoin/wallet/account.rb
+- lib/bitcoin/wallet/base.rb
+- lib/bitcoin/wallet/db.rb
+- lib/bitcoin/wallet/master_key.rb
 - lib/openassets.rb
 - lib/openassets/marker_output.rb
 - lib/openassets/payload.rb
@@ -316,7 +373,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.6.13
+rubygems_version: 2.6.12
 signing_key: 
 specification_version: 4
 summary: "[WIP]The implementation of Bitcoin Protocol for Ruby."

data/lib/bitcoin/node/spv_block_chain.rb

@@ -1,15 +0,0 @@
-module Bitcoin
-  module Node
-
-    class SPVBlockChain
-
-      attr_reader :block_store
-
-      def initialize(block_store = Bitcoin::Store::SPVBlockStore.new)
-        @block_store = block_store
-      end
-
-    end
-
-  end
-end

data/lib/bitcoin/store/spv_block_store.rb

@@ -1,11 +0,0 @@
-module Bitcoin
-
-  module Store
-
-    class SPVBlockStore
-
-    end
-
-  end
-
-end