bitcoinrb 1.12.0 → 1.12.1

data/lib/bitcoin/store/spv_chain.rb

@@ -1,101 +0,0 @@
-module Bitcoin
-
-  module Store
-
-    KEY_PREFIX = {
-        entry: 'e',   # key: block hash, value: Bitcoin::Store::ChainEntry payload
-        height: 'h',  # key: block height, value: block hash.
-        best: 'B',    # value: best block hash.
-        next: 'n'     # key: block hash, value: A hash of the next block of the specified hash
-    }
-
-    class SPVChain
-
-      attr_reader :db
-      attr_reader :logger
-
-      def initialize(db = Bitcoin::Store::DB::LevelDB.new)
-        @db = db # TODO multiple db switch
-        @logger = Bitcoin::Logger.create(:debug)
-        initialize_block
-      end
-
-      # get latest block in the store.
-      # @return[Bitcoin::Store::ChainEntry]
-      def latest_block
-        hash = db.best_hash
-        return nil unless hash
-        find_entry_by_hash(hash)
-      end
-
-      # find block entry with the specified height.
-      def find_entry_by_height(height)
-        find_entry_by_hash(db.get_hash_from_height(height))
-      end
-
-      # find block entry with the specified hash
-      def find_entry_by_hash(hash)
-        payload = db.get_entry_payload_from_hash(hash)
-        return nil unless payload
-        ChainEntry.parse_from_payload(payload)
-      end
-
-      # append block header to chain.
-      # @param [Bitcoin::BlockHeader] header a block header.
-      # @return [Bitcoin::Store::ChainEntry] appended block header entry.
-      def append_header(header)
-        logger.info("append header #{header.block_id}")
-        raise "this header is invalid. #{header.block_hash}" unless header.valid?
-        best_block = latest_block
-        current_height = best_block.height
-        if best_block.block_hash == header.prev_hash
-          entry = Bitcoin::Store::ChainEntry.new(header, current_height + 1)
-          db.save_entry(entry)
-          entry
-        else
-          unless find_entry_by_hash(header.block_hash)
-            # TODO implements recovery process
-            raise "header's previous hash(#{header.prev_hash}) does not match current best block's(#{best_block.block_hash})."
-          end
-        end
-      end
-
-      # get next block hash for specified +hash+
-      # @param [String] hash the block hash(little endian)
-      # @return [String] the next block hash. If it does not exist yet, return nil.
-      def next_hash(hash)
-        db.next_hash(hash)
-      end
-
-      # get median time past for specified block +hash+
-      # @param [String] hash the block hash.
-      # @return [Integer] the median time past value.
-      def mtp(hash)
-        time = []
-        Bitcoin::MEDIAN_TIME_SPAN.times do
-          entry = find_entry_by_hash(hash)
-          break unless entry
-
-          time << entry.header.time
-          hash = entry.header.prev_hash
-        end
-        time.sort!
-        time[time.size / 2]
-      end
-
-      private
-
-      # if database is empty, put genesis block.
-      def initialize_block
-        unless latest_block
-          block = Bitcoin.chain_params.genesis_block
-          genesis = ChainEntry.new(block.header, 0)
-          db.save_entry(genesis)
-        end
-      end
-
-    end
-
-  end
-
-end

data/lib/bitcoin/store/utxo_db.rb

@@ -1,226 +0,0 @@
-require 'leveldb-native'
-
-module Bitcoin
-  module Store
-    class UtxoDB
-
-      KEY_PREFIX = {
-        out_point: 'o',        # key: out_point(tx_hash and index), value: Utxo
-        script: 's',           # key: script_pubkey and out_point(tx_hash and index), value: Utxo
-        height: 'h',           # key: block_height and out_point, value: Utxo
-        tx_hash: 't',          # key: tx_hash of transaction, value: [block_height, tx_index]
-        block: 'b',            # key: block_height and tx_index, value: tx_hash
-        tx_payload: 'p',       # key: tx_hash, value: Tx
-      }
-
-      attr_reader :level_db, :logger
-
-      def initialize(path = "#{Bitcoin.base_dir}/db/utxo")
-        FileUtils.mkdir_p(path)
-        @level_db = ::LevelDBNative::DB.new(path)
-        @logger = Bitcoin::Logger.create(:debug)
-      end
-
-      def close
-        level_db.close
-      end
-
-      # Save payload of a transaction into db
-      #
-      # @param  [String] tx_hash
-      # @param  [String] tx_payload
-      def save_tx(tx_hash, tx_payload)
-        logger.info("UtxoDB#save_tx:#{[tx_hash, tx_payload]}")
-        level_db.batch do
-          # tx_hash -> [block_height, tx_index]
-          key = KEY_PREFIX[:tx_payload] + tx_hash
-          level_db.put(key, tx_payload)
-        end
-      end
-
-      # Save tx position (block height and index in the block) into db
-      # When node receives `header` message, node should call save_tx_position to store block height and its index.
-      #
-      # @param  [String] tx_hash
-      # @param  [Integer] block_height
-      # @param  [Integer] tx_index
-      def save_tx_position(tx_hash, block_height, tx_index)
-        logger.info("UtxoDB#save_tx_position:#{[tx_hash, block_height, tx_index]}")
-        level_db.batch do
-          # tx_hash -> [block_height, tx_index]
-          key = KEY_PREFIX[:tx_hash] + tx_hash
-          level_db.put(key, [block_height, tx_index].pack('N2').bth)
-
-          # block_hash and tx_index -> tx_hash
-          key = KEY_PREFIX[:block] + [block_height, tx_index].pack('N2').bth
-          level_db.put(key, tx_hash)
-        end
-      end
-
-      # Save utxo into db
-      #
-      # @param [Bitcoin::OutPoint] out_point
-      # @param [Double] value
-      # @param [Bitcoin::Script] script_pubkey
-      # @param [Integer] block_height
-      def save_utxo(out_point, value, script_pubkey, block_height=nil)
-        logger.info("UtxoDB#save_utxo:#{[out_point, value, script_pubkey, block_height]}")
-        level_db.batch do
-          utxo = Bitcoin::Wallet::Utxo.new(out_point.tx_hash, out_point.index, value, script_pubkey, block_height)
-          payload = utxo.to_payload
-
-          # out_point
-          key = KEY_PREFIX[:out_point] + out_point.to_hex
-          return if level_db.contains?(key)
-          level_db.put(key, payload)
-
-          # script_pubkey
-          if script_pubkey
-            key = KEY_PREFIX[:script] + script_pubkey.to_hex + out_point.to_hex
-            level_db.put(key, payload)
-          end
-
-          # height
-          unless block_height.nil?
-            key = KEY_PREFIX[:height] + [block_height].pack('N').bth + out_point.to_hex
-            level_db.put(key, payload)
-          end
-
-          utxo
-        end
-      end
-
-      # Get transaction stored via save_tx and save_tx_position
-      #
-      # @param  [string] tx_hash
-      # @return [block_height, tx_index, tx_payload]
-      def get_tx(tx_hash)
-        key = KEY_PREFIX[:tx_hash] + tx_hash
-        return [] unless level_db.contains?(key)
-        block_height, tx_index = level_db.get(key).htb.unpack('N2')
-        key = KEY_PREFIX[:tx_payload] + tx_hash
-        tx_payload = level_db.get(key)
-        [block_height, tx_index, tx_payload]
-      end
-
-      # Delete utxo from db
-      #
-      # @param  [Bitcoin::Outpoint] out_point
-      # @return [Bitcoin::Wallet::Utxo] 
-      def delete_utxo(out_point)
-        level_db.batch do
-          # [:out_point]
-          key = KEY_PREFIX[:out_point] + out_point.to_hex
-          return unless level_db.contains?(key)
-          utxo = Bitcoin::Wallet::Utxo.parse_from_payload(level_db.get(key))
-          level_db.delete(key)
-
-          # [:script]
-          if utxo.script_pubkey
-            key = KEY_PREFIX[:script] + utxo.script_pubkey.to_hex + out_point.to_hex
-            level_db.delete(key)
-          end
-
-          if utxo.block_height
-            # [:height]
-            key = KEY_PREFIX[:height] + [utxo.block_height].pack('N').bth + out_point.to_hex
-            level_db.delete(key)
-
-            # [:block]
-            key = KEY_PREFIX[:block] + [utxo.block_height, utxo.index].pack('N2').bth
-            level_db.delete(key)
-          end
-
-          # handles both [:tx_hash] and [:tx_payload]
-          if utxo.tx_hash
-            key = KEY_PREFIX[:tx_hash] + utxo.tx_hash
-            level_db.delete(key)
-
-            key = KEY_PREFIX[:tx_payload] + utxo.tx_hash
-            level_db.delete(key)
-          end
-
-          utxo
-        end
-      end
-
-      # Get utxo of the specified out point
-      #
-      # @param  [Bitcoin::Outpoint] out_point
-      # @return [Bitcoin::Wallet::Utxo]
-      def get_utxo(out_point)
-        level_db.batch do
-          key = KEY_PREFIX[:out_point] + out_point.to_hex
-          return unless level_db.contains?(key)
-          return Bitcoin::Wallet::Utxo.parse_from_payload(level_db.get(key))
-        end
-      end
-
-      # return [Bitcoin::Wallet::Utxo ...]
-      def list_unspent(current_block_height: 9999999, min: 0, max: 9999999, addresses: nil)
-        if addresses
-          list_unspent_by_addresses(current_block_height, min: min, max: max, addresses: addresses)
-        else
-          list_unspent_by_block_height(current_block_height, min: min, max: max)
-        end
-      end
-
-      # @param [Bitcoin::Wallet::Account]
-      # return [Bitcoin::Wallet::Utxo ...]
-      def list_unspent_in_account(account, current_block_height: 9999999, min: 0, max: 9999999)
-        return [] unless account
-
-        script_pubkeys = case account.purpose
-          when Bitcoin::Wallet::Account::PURPOSE_TYPE[:legacy]
-            account.watch_targets.map { |t| Bitcoin::Script.to_p2pkh(t).to_hex }
-          when Bitcoin::Wallet::Account::PURPOSE_TYPE[:nested_witness]
-            account.watch_targets.map { |t| Bitcoin::Script.to_p2wpkh(t).to_p2sh.to_hex }
-          when Bitcoin::Wallet::Account::PURPOSE_TYPE[:native_segwit]
-            account.watch_targets.map { |t| Bitcoin::Script.to_p2wpkh(t).to_hex }
-          end
-        list_unspent_by_script_pubkeys(current_block_height, min: min, max: max, script_pubkeys: script_pubkeys)
-      end
-
-      # @param [Bitcoin::Wallet::Account]
-      # return [Bitcoin::Wallet::Utxo ...]
-      def get_balance(account, current_block_height: 9999999, min: 0, max: 9999999)
-        list_unspent_in_account(account, current_block_height: current_block_height, min: min, max: max).sum { |u| u.value }
-      end
-
-      private
-
-      def utxos_between(from, to)
-        level_db.each(from: from, to: to).map { |k, v| Bitcoin::Wallet::Utxo.parse_from_payload(v) }
-      end
-
-      class ::Array
-        def with_height(min, max)
-          select { |u| u.block_height >= min && u.block_height <= max }
-        end
-      end
-
-      def list_unspent_by_block_height(current_block_height, min: 0, max: 9999999)
-        max_height = [current_block_height - min, 0].max
-        min_height = [current_block_height - max, 0].max
-        from = KEY_PREFIX[:height] + [min_height].pack('N').bth + '000000000000000000000000000000000000000000000000000000000000000000000000'
-        to = KEY_PREFIX[:height] + [max_height].pack('N').bth + 'ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff'
-        utxos_between(from, to)
-      end
-
-      def list_unspent_by_addresses(current_block_height, min: 0, max: 9999999, addresses: [])
-        script_pubkeys = addresses.map { |a| Bitcoin::Script.parse_from_addr(a).to_hex }
-        list_unspent_by_script_pubkeys(current_block_height, min: min, max: max, script_pubkeys: script_pubkeys)
-      end
-
-      def list_unspent_by_script_pubkeys(current_block_height, min: 0, max: 9999999, script_pubkeys: [])
-        max_height = current_block_height - min
-        min_height = current_block_height - max
-        script_pubkeys.map do |key|
-          from = KEY_PREFIX[:script] + key + '000000000000000000000000000000000000000000000000000000000000000000000000'
-          to = KEY_PREFIX[:script] + key + 'ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff'
-          utxos_between(from, to).with_height(min_height, max_height)
-        end.flatten
-      end
-    end
-  end
-end

data/lib/bitcoin/store.rb

@@ -1,12 +0,0 @@
-require 'leveldb-native'
-
-module Bitcoin
-  module Store
-
-    autoload :DB, 'bitcoin/store/db'
-    autoload :SPVChain, 'bitcoin/store/spv_chain'
-    autoload :ChainEntry, 'bitcoin/store/chain_entry'
-    autoload :UtxoDB, 'bitcoin/store/utxo_db'
-
-  end
-end