Class: Bitcoin::Protocol::PartialMerkleTree

Inherits:
Object
  • Object
show all
Defined in:
lib/bitcoin/protocol/partial_merkle_tree.rb

Defined Under Namespace

Classes: Node

Constant Summary collapse

BIT_MASK = 
[0x01, 0x02, 0x04, 0x08, 0x10, 0x20, 0x40, 0x80]

Instance Method Summary collapse

Constructor Details

#initialize(total_txs, hashes, flags) ⇒ PartialMerkleTree

Returns a new instance of PartialMerkleTree.



6
7
8
9
10
 
# File 'lib/bitcoin/protocol/partial_merkle_tree.rb', line 6

def initialize(total_txs, hashes, flags)
  @total_txs, @flags = total_txs, flags
  @hashes = hashes.map{|h| h.reverse_hth }
  @visit_idx = 0
end

Instance Method Details

#build_tree ⇒ Object 



16
17
18
19
20
21
22
23
24
 
# File 'lib/bitcoin/protocol/partial_merkle_tree.rb', line 16

def build_tree
  lay = @leaves = @total_txs.times.map{ Node.new(nil, nil, nil) }
  while lay.size > 1
    lay = lay.each_slice(2).map do |left, right|
      Node.new(nil, left, right)
    end
  end
  return lay[0]
end

#current_flag ⇒ Object 



26
27
28
 
# File 'lib/bitcoin/protocol/partial_merkle_tree.rb', line 26

def current_flag
  @flags[@visit_idx / 8].ord & BIT_MASK[@visit_idx % 8] == 0
end

#root ⇒ Object 



30
31
32
 
# File 'lib/bitcoin/protocol/partial_merkle_tree.rb', line 30

def root
  @root ||= build_tree
end

#set_value(node = root) ⇒ Object 



34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
 
# File 'lib/bitcoin/protocol/partial_merkle_tree.rb', line 34

def set_value(node = root)
  if current_flag || (node.left.nil? && node.right.nil?)
    node.value = @hashes.shift
    return
  end

  if node.left
    @visit_idx += 1
    set_value(node.left)
  end
  if node.right
    @visit_idx += 1
    set_value(node.right)
  end

  right = node.right || node.left
  node.value = Bitcoin.bitcoin_mrkl(node.left.value, right.value)

  return
end

#tx_hashes ⇒ Object 



12
13
14
 
# File 'lib/bitcoin/protocol/partial_merkle_tree.rb', line 12

def tx_hashes
  @leaves.reject{|n| n.value.nil? }.map{|n| n.value }
end

#valid_tree?(mrkl_root_hash) ⇒ Boolean

Returns:

  • (Boolean) 


55
56
57
58
59
60
 
# File 'lib/bitcoin/protocol/partial_merkle_tree.rb', line 55

def valid_tree?(mrkl_root_hash)
  return false unless @hashes.empty?
  return false if ((@visit_idx + 1)/8.0).ceil != @flags.length
  return false if mrkl_root_hash != root.value
  return true
end