mssmt 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c2bf77bb8efa4e33fe73b3bd1b54e577e02097fadda8420924caf8df0ad9df21
-  data.tar.gz: a5715d5289504147a0b76f8c3935910df965cae05df2a284136735389f8daa27
+  metadata.gz: bc50f962065de3f0ceaeebaa7715ab5e32d03a882d7df11022e6e4bc90b459d1
+  data.tar.gz: 020cb3cfedb35c8c3a6cd123d9b772e666e9b9fe19f301c07acc6ca27a193ca6
 SHA512:
-  metadata.gz: 30e1c75109384039d6d87ed9702e7bfbe2848e499285460b67da66dd80668e32590ce409d5deedd64a08ed5cae6f558d40817c73275a271c0b92c6a0f1cab8e0
-  data.tar.gz: 6231535740549dfdb317e122945464ec92789c24479f164058ebdb06a3bfb2cabd688a0b6c0eca4c48881e9a891f5b72532a7609b4ec27edf132f9986f6ad558
+  metadata.gz: 468f823bbba1cff9549b5977a98270ec215e3ccf7b8057bcdd0642fd99da280233a0a821254fc2fb4c294d638ccaa6443cf04abd8a99fd4e61de6f2c51f169a1
+  data.tar.gz: 17f30d1180c82d4b4862c752b842e725e5d9527f1ad7f80b5e8eeb3e0d2573b127531afbfda474017103b97b21acbf3d1f1e891398686fdf06256dd3d246dc06

data/README.md

@@ -36,9 +36,9 @@ tree.insert(key, leaf)
 # Get root node
 root_node = tree.root_node
 # Root hash
-root_hash = tree.root_hash
+root_hash = tree.root_hash.unpack1('H*')
 # or
-root_node.node_hash
+root_node.node_hash.unpack1('H*')
 
 # Get leaf node in tree
 leaf = tree.get(key)

data/lib/mssmt/branch_node.rb

@@ -21,5 +21,10 @@ module MSSMT
           "#{left.node_hash}#{right.node_hash}#{[@sum].pack("Q>")}"
         )
     end
+
+    def ==(other)
+      return false unless other.is_a?(BranchNode)
+      node_hash == other.node_hash
+    end
   end
 end

data/lib/mssmt/leaf_node.rb

@@ -30,5 +30,10 @@ module MSSMT
     def empty?
       (value.nil? or value.empty?) && sum.zero?
     end
+
+    def ==(other)
+      return false unless other.is_a?(LeafNode)
+      node_hash == other.node_hash
+    end
   end
 end

data/lib/mssmt/proof.rb

@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+module MSSMT
+  # Merkle proof for MS-SMT.
+  class Proof
+    attr_reader :nodes
+
+    # Constructor
+    # @param [Array(MSSMT::BranchNode|MSSMT::LeafNode)] nodes Siblings that should be hashed with the leaf and
+    # its parents to arrive at the root of the MS-SMT.
+    def initialize(nodes)
+      @nodes = nodes
+    end
+
+    # Compresses a merkle proof by replacing its empty nodes with a bit vector.
+    # @return [MSSMT::CompressedProof]
+    def compress
+      bits = Array.new(nodes.length, false)
+      compact_nodes = []
+      nodes.each.each_with_index do |node, i|
+        # puts "#{node.node_hash}:#{Tree.empty_tree[Tree::MAX_LEVEL - 1].node_hash}"
+        if node.node_hash == Tree.empty_tree[Tree::MAX_LEVEL - 1].node_hash
+          bits[i] = true
+        else
+          compact_nodes << node
+        end
+      end
+      CompressedProof.new(compact_nodes, bits)
+    end
+
+    def ==(other)
+      return false unless other.is_a?(Proof)
+      nodes == other.nodes
+    end
+  end
+
+  # Compressed MS-SMT merkle proof.
+  # Since merkle proofs for a MS-SMT are always constant size (255 nodes),
+  # we replace its empty nodes by a bit vector.
+  class CompressedProof < Proof
+    attr_reader :bits
+
+    # Constructor
+    # @param [Array(MSSMT::BranchNode|MSSMT::LeafNode)] nodes Siblings that should be hashed with the leaf and
+    # its parents to arrive at the root of the MS-SMT.
+    # @param [Array] bits +bits+ determines whether a sibling node within a proof is part of the empty tree.
+    # This allows us to efficiently compress proofs by not including any pre-computed nodes.
+    def initialize(nodes, bits)
+      super(nodes)
+      @bits = bits
+    end
+
+    # Decompress a compressed merkle proof by replacing its bit vector with the empty nodes it represents.
+    # @return [MSSMT::Proof]
+    def decompress
+      count = 0
+      full_nodes = []
+      bits.each.with_index do |b, i|
+        if b
+          full_nodes << Tree.empty_tree[Tree::MAX_LEVEL - i]
+        else
+          full_nodes << nodes[count]
+          count += 1
+        end
+      end
+      Proof.new(full_nodes)
+    end
+  end
+end

data/lib/mssmt/tree.rb

@@ -10,10 +10,29 @@ module MSSMT
     # Index of the last bit for MS-SMT keys
     LAST_BIT_INDEX = MAX_LEVEL - 1
 
-    attr_reader :empty_tree, :store
+    attr_reader :store
+
+    def self.empty_tree
+      @empty_tree ||= build_empty_tree
+    end
+
+    # Generate empty tree.
+    # @return [Array]
+    def self.build_empty_tree
+      tree = Array.new(MSSMT::Tree::MAX_LEVEL + 1)
+      tree[MSSMT::Tree::MAX_LEVEL] = MSSMT::LeafNode.empty_leaf
+      MSSMT::Tree::MAX_LEVEL.times do |i|
+        branch =
+          MSSMT::BranchNode.new(
+            tree[MSSMT::Tree::MAX_LEVEL - i],
+            tree[MSSMT::Tree::MAX_LEVEL - i]
+          )
+        tree[MSSMT::Tree::MAX_LEVEL - (i + 1)] = branch
+      end
+      tree.freeze
+    end
 
     def initialize(store: MSSMT::Store::DefaultStore.new)
-      @empty_tree = build_empty_tree
       @store = store
     end
 
@@ -26,7 +45,7 @@ module MSSMT
     # Get root node in tree.
     # @return [MSSMT::BranchNode]
     def root_node
-      store.root.nil? ? empty_tree[0] : store.root
+      store.root.nil? ? Tree.empty_tree[0] : store.root
     end
 
     # Insert a leaf node at the given key within the MS-SMT.
@@ -49,10 +68,10 @@ module MSSMT
       root =
         walk_up(key, leaf, siblings) do |i, _, _, parent|
           prev_parent = prev_parents[MSSMT::Tree::MAX_LEVEL - 1 - i]
-          unless prev_parent == empty_tree[i].node_hash
+          unless prev_parent == Tree.empty_tree[i].node_hash
             store.delete_branch(prev_parent)
           end
-          unless parent.node_hash == empty_tree[i].node_hash
+          unless parent.node_hash == Tree.empty_tree[i].node_hash
             store.insert_branch(parent)
           end
         end
@@ -76,19 +95,19 @@ module MSSMT
 
     # Generate a merkle proof for the leaf node found at the given +key+.
     # @param [String] key key with hex format.
-    # @return [Array] merkle proof
+    # @return [MSSMT::Proof] merkle proof
     def merkle_proof(key)
       proof = Array.new(MAX_LEVEL)
       walk_down(key) { |i, _, sibling, _| proof[MAX_LEVEL - 1 - i] = sibling }
-      proof
+      MSSMT::Proof.new(proof)
     end
 
     # Verify whether a merkle proof for the leaf found at the given key is valid.
     # @param [String] key key with hex format.
     # @param [MSSMT::LeafNode] leaf leaf node.
-    # @param [Array] proof merkle proof.
+    # @param [MSSMT::Proof] proof merkle proof.
     def valid_merkle_proof?(key, leaf, proof)
-      root_hash == walk_up(key, leaf, proof).node_hash
+      root_hash == walk_up(key, leaf, proof.nodes).node_hash
     end
 
     private
@@ -110,7 +129,7 @@ module MSSMT
     end
 
     def get_node(height, key)
-      return empty_tree[height] if empty_tree[height].node_hash == key
+      return Tree.empty_tree[height] if Tree.empty_tree[height].node_hash == key
       store.branches[key] || store.leaves[key]
     end
 
@@ -152,21 +171,5 @@ module MSSMT
       value = [key].pack("H*")[idx / 8].ord
       (value >> (idx % 8)) & 1
     end
-
-    # Generate empty tree.
-    # @return [Array]
-    def build_empty_tree
-      tree = Array.new(MSSMT::Tree::MAX_LEVEL + 1)
-      tree[MSSMT::Tree::MAX_LEVEL] = MSSMT::LeafNode.empty_leaf
-      MSSMT::Tree::MAX_LEVEL.times do |i|
-        branch =
-          MSSMT::BranchNode.new(
-            tree[MSSMT::Tree::MAX_LEVEL - i],
-            tree[MSSMT::Tree::MAX_LEVEL - i]
-          )
-        tree[MSSMT::Tree::MAX_LEVEL - (i + 1)] = branch
-      end
-      tree
-    end
   end
 end

data/lib/mssmt/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module MSSMT
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/mssmt.rb

@@ -12,4 +12,6 @@ module MSSMT
   autoload :LeafNode, "mssmt/leaf_node"
   autoload :BranchNode, "mssmt/branch_node"
   autoload :Tree, "mssmt/tree"
+  autoload :Proof, "mssmt/proof"
+  autoload :CompressedProof, "mssmt/proof"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mssmt
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - azuchi
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-10-17 00:00:00.000000000 Z
+date: 2022-10-26 00:00:00.000000000 Z
 dependencies: []
 description: Ruby implementation of Merkle Sum Sparse Merkle Tree
 email:
@@ -33,6 +33,7 @@ files:
 - lib/mssmt.rb
 - lib/mssmt/branch_node.rb
 - lib/mssmt/leaf_node.rb
+- lib/mssmt/proof.rb
 - lib/mssmt/store.rb
 - lib/mssmt/store/default_store.rb
 - lib/mssmt/tree.rb