mssmt 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bc50f962065de3f0ceaeebaa7715ab5e32d03a882d7df11022e6e4bc90b459d1
-  data.tar.gz: 020cb3cfedb35c8c3a6cd123d9b772e666e9b9fe19f301c07acc6ca27a193ca6
+  metadata.gz: e99668bb044c1670cee5d94e301f8113321c04ce9ac06b697fcfa8eabf1668f9
+  data.tar.gz: 7928cf9abd46c85c505dc770c2cbcce33e82ce98c2f7f772872792366bcd6eaa
 SHA512:
-  metadata.gz: 468f823bbba1cff9549b5977a98270ec215e3ccf7b8057bcdd0642fd99da280233a0a821254fc2fb4c294d638ccaa6443cf04abd8a99fd4e61de6f2c51f169a1
-  data.tar.gz: 17f30d1180c82d4b4862c752b842e725e5d9527f1ad7f80b5e8eeb3e0d2573b127531afbfda474017103b97b21acbf3d1f1e891398686fdf06256dd3d246dc06
+  metadata.gz: 1b0b1648a73caed3a29c764c688b8077dea1beaa65da3772429470fc1ca1eab352b11128538e0e3b4daa2e8c578452360a562a08bd69bebe3bebadd228070f74
+  data.tar.gz: 653ad3b9270105b25a9dfea42414918cc0640cdc9359a7c9c01c712ee503f6689d81ee74796510099c3827fae53209fd2a7325392a36541f64b39447482d6e30

data/lib/mssmt/branch_node.rb

@@ -16,15 +16,19 @@ module MSSMT
       @left = left
       @right = right
       @sum = left.sum + right.sum
+      warn("sum:#{@sum} cause overflow.") if @sum > 0xffffffffffffffff # TODO
+      @sum = (@sum & 0xffffffffffffffff)
       @node_hash =
         Digest::SHA256.digest(
           "#{left.node_hash}#{right.node_hash}#{[@sum].pack("Q>")}"
         )
     end
 
+    # Check whether same branch|computed node or not.
+    # @return [Boolean]
     def ==(other)
-      return false unless other.is_a?(BranchNode)
-      node_hash == other.node_hash
+      return false unless [BranchNode, ComputedNode].include?(other.class)
+      node_hash == other.node_hash && sum == other.sum
     end
   end
 end

data/lib/mssmt/computed_node.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module MSSMT
+  # Node within a MS-SMT that has already had its node_hash and sum computed, i.e., its preimage is not available.
+  class ComputedNode
+    attr_reader :node_hash, :sum
+
+    # Constructor
+    # @param [String] node_hash node hash with binary fomat.
+    # @param [Integer] sum
+    def initialize(node_hash, sum)
+      @node_hash = node_hash
+      warn("sum: #{sum} cause overflow.") if sum > 0xffffffffffffffff # TODO
+      @sum = sum
+    end
+
+    def ==(other)
+      return false unless [BranchNode, ComputedNode].include?(other.class)
+      node_hash == other.node_hash && sum == other.sum
+    end
+  end
+end

data/lib/mssmt/leaf_node.rb

@@ -10,7 +10,8 @@ module MSSMT
     # @param [Integer] sum integer value associated with the value
     def initialize(value, sum)
       @value = value
-      @sum = sum
+      warn("sum: #{sum} cause overflow.") if sum > 0xffffffffffffffff # TODO
+      @sum = sum & 0xffffffffffffffff
     end
 
     # Generate empty leaf node.

data/lib/mssmt/proof.rb

@@ -17,8 +17,7 @@ module MSSMT
       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
+        if node.node_hash == Tree.empty_tree[Tree::MAX_LEVEL - i].node_hash
           bits[i] = true
         else
           compact_nodes << node
@@ -49,6 +48,21 @@ module MSSMT
       @bits = bits
     end
 
+    # Decode compressed proof.
+    # @param [String] compressed proof with binary fomat.
+    # @return [MSSMT::CompressedProof]
+    def self.decode(data)
+      buf = data.is_a?(StringIO) ? data : StringIO.new(data)
+      nodes_len = buf.read(2).unpack1("n")
+      nodes =
+        nodes_len.times.map do
+          ComputedNode.new(buf.read(32), buf.read(8).unpack1("Q>"))
+        end
+      bytes = buf.read(MSSMT::Tree::MAX_LEVEL / 8)
+      bits = unpack_bits(bytes.unpack("C*"))
+      CompressedProof.new(nodes, bits)
+    end
+
     # Decompress a compressed merkle proof by replacing its bit vector with the empty nodes it represents.
     # @return [MSSMT::Proof]
     def decompress
@@ -64,5 +78,48 @@ module MSSMT
       end
       Proof.new(full_nodes)
     end
+
+    # Encode the compressed proof.
+    # @return [String] encoded string.
+    def encode
+      buf = [nodes.length].pack("n")
+      nodes.each do |node|
+        buf << node.node_hash
+        buf << [node.sum].pack("Q>")
+      end
+      buf << pack_bits.pack("C*")
+      buf
+    end
+
+    def ==(other)
+      return false unless other.is_a?(CompressedProof)
+      bits == other.bits && nodes == other.nodes
+    end
+
+    private
+
+    def pack_bits
+      bytes = Array.new((bits.length + 8 - 1) / 8, 0)
+      bits.each_with_index do |b, i|
+        next unless b
+        byte_index = i / 8
+        bit_index = i % 8
+        bytes[byte_index] |= (1 << bit_index)
+      end
+      bytes
+    end
+
+    def self.unpack_bits(bytes)
+      bit_len = bytes.length * 8
+      bits = Array.new(bit_len, false)
+      bit_len.times do |i|
+        byte_index = i / 8
+        bit_index = i % 8
+        bits[i] = ((bytes[byte_index] >> bit_index) & 1) == 1
+      end
+      bits
+    end
+
+    private_class_method :unpack_bits
   end
 end

data/lib/mssmt/version.rb

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

data/lib/mssmt.rb

@@ -11,6 +11,7 @@ module MSSMT
   autoload :Store, "mssmt/store"
   autoload :LeafNode, "mssmt/leaf_node"
   autoload :BranchNode, "mssmt/branch_node"
+  autoload :ComputedNode, "mssmt/computed_node"
   autoload :Tree, "mssmt/tree"
   autoload :Proof, "mssmt/proof"
   autoload :CompressedProof, "mssmt/proof"

metadata

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