iota-ruby 1.1.8-java

data/lib/iota/crypto/curl_ruby.rb

@@ -0,0 +1,70 @@
+module IOTA
+  module Crypto
+    class RubyCurl
+      NUMBER_OF_ROUNDS = 81
+      HASH_LENGTH = 243
+      STATE_LENGTH = 3 * HASH_LENGTH
+      TRUTH_TABLE = [1, 0, -1, 1, -1, 0, -1, 1, 0]
+
+      def initialize(rounds = nil)
+        @rounds = rounds || NUMBER_OF_ROUNDS
+        reset
+      end
+
+      def reset
+        @state = [0] * STATE_LENGTH
+      end
+
+      def absorb(trits)
+        length  = trits.length
+        offset  = 0
+
+        while offset < length
+          start = offset
+          stop  = [start + HASH_LENGTH, length].min
+
+          @state[0...stop-start] = trits.slice(start, stop-start)
+          transform
+
+          offset += HASH_LENGTH
+        end
+      end
+
+      def squeeze(trits)
+        trits[0...HASH_LENGTH] = @state.slice(0, HASH_LENGTH)
+        transform
+      end
+
+      def transform
+        previousState  = @state.slice(0, @state.length)
+        newState   = @state.slice(0, @state.length)
+
+        index = 0
+        round = 0
+        while round < @rounds
+          previousTrit = previousState[index].to_i
+
+          pos = 0
+          while true
+            index += (index < 365) ? 364 : -365
+            newTrit = previousState[index].to_i
+            newState[pos] = TRUTH_TABLE[previousTrit + (3 * newTrit) + 4]
+            previousTrit = newTrit
+            pos += 1
+            break if pos >= STATE_LENGTH
+          end
+
+          previousState = newState
+          newState = newState.slice(0, newState.length)
+          round += 1
+        end
+
+        @state = newState
+      end
+
+      def version
+        "Ruby"
+      end
+    end
+  end
+end

data/lib/iota/crypto/hmac.rb

@@ -0,0 +1,27 @@
+module IOTA
+  module Crypto
+    class Hmac
+      ROUNDS = 27
+
+      def initialize(key)
+        @key = Converter.trits(key)
+      end
+
+      def addHMAC(bundle)
+        curl = Curl.new(ROUNDS)
+        (0...bundle.bundle.length).step(1) do |i|
+          if bundle.bundle[i].value > 0
+            bundleHashTrits = Converter.trits(bundle.bundle[i].bundle)
+            hmac = Array.new(243, 0)
+            curl.reset
+            curl.absorb(@key)
+            curl.absorb(bundleHashTrits)
+            curl.squeeze(hmac)
+            hmacTrytes = Converter.trytes(hmac)
+            bundle.bundle[i].signatureMessageFragment = hmacTrytes + bundle.bundle[i].signatureMessageFragment[81...2187]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/iota/crypto/kerl.rb

@@ -0,0 +1,82 @@
+module IOTA
+  module Crypto
+    class Kerl
+      BIT_HASH_LENGTH = 384
+      HASH_LENGTH = Curl::HASH_LENGTH
+
+      def initialize
+        reset
+      end
+
+      def reset
+        unless RUBY_PLATFORM =~ /java/
+          require 'digest/sha3'
+          @hasher = Digest::SHA3.new(BIT_HASH_LENGTH)
+        else
+          require "iota/crypto/sha3_ruby"
+          @hasher = Digest::RubySHA3.new(BIT_HASH_LENGTH)
+        end
+      end
+
+      def absorb(trits, offset = 0, length = nil)
+        pad = trits.length % HASH_LENGTH != 0 ? trits.length % HASH_LENGTH : HASH_LENGTH
+        trits.concat([0] * (HASH_LENGTH - pad))
+
+        length = trits.length if length.nil?
+
+        if length % HASH_LENGTH != 0 || length == 0
+          raise StandardError,  "Illegal length provided"
+        end
+
+        while offset < length
+          limit = [offset + HASH_LENGTH, length].min
+
+          # If we're copying over a full chunk, zero last trit
+          trits[limit - 1] = 0 if limit - offset == HASH_LENGTH
+
+          signed_bytes = Converter.convertToBytes(trits[offset...limit])
+
+          # Convert signed bytes into their equivalent unsigned representation
+          # In order to use Python's built-in bytes type
+          unsigned_bytes = signed_bytes.map{ |b| Converter.convertSign(b) }.pack('c*').force_encoding('UTF-8')
+
+          @hasher.update(unsigned_bytes)
+
+          offset += HASH_LENGTH
+        end
+      end
+
+      def squeeze(trits, offset = 0, length = nil)
+        pad = trits.length % HASH_LENGTH != 0 ? trits.length % HASH_LENGTH : HASH_LENGTH
+        trits.concat([0] * (HASH_LENGTH - pad))
+
+        length = trits.length > 0 ? trits.length : HASH_LENGTH if length.nil?
+
+        if length % HASH_LENGTH != 0 || length == 0
+          raise StandardError,  "Illegal length provided"
+        end
+
+        while offset < length
+          unsigned_hash =  @hasher.digest
+
+          signed_hash = unsigned_hash.bytes.map { |b| Converter.convertSign(b) }
+
+          trits_from_hash = Converter.convertToTrits(signed_hash)
+          trits_from_hash[HASH_LENGTH - 1] = 0
+
+          limit = [HASH_LENGTH, length - offset].min
+
+          trits[offset...offset+limit] = trits_from_hash[0...limit]
+
+          flipped_bytes = unsigned_hash.bytes.map{ |b| Converter.convertSign(~b)}.pack('c*').force_encoding('UTF-8')
+
+          # Reset internal state before feeding back in
+          reset
+          @hasher.update(flipped_bytes)
+
+          offset += HASH_LENGTH
+        end
+      end
+    end
+  end
+end

data/lib/iota/crypto/pow_provider.rb

@@ -0,0 +1,27 @@
+require 'ffi'
+
+module IOTA
+  module Crypto
+    class PowProvider
+      extend FFI::Library
+
+      ccurl_version = '0.3.0'
+      lib_root = File.join(File.dirname(__FILE__), '../../../ext/pow')
+      if FFI::Platform.windows?
+        libccurl_path = "#{lib_root}/ccurl-#{ccurl_version}.dll"
+      elsif FFI::Platform.mac?
+        libccurl_path = "#{lib_root}/libccurl-#{ccurl_version}.dylib"
+      else
+        libccurl_path = "#{lib_root}/libccurl-#{ccurl_version}.so"
+      end
+
+      ffi_lib libccurl_path
+
+      attach_function :ccurl_pow, [ :string, :int ], :string
+      attach_function :ccurl_digest_transaction, [ :string ], :string
+
+      alias_method :pow, :ccurl_pow
+      alias_method :digest, :ccurl_digest_transaction
+    end
+  end
+end

data/lib/iota/crypto/private_key.rb

@@ -0,0 +1,80 @@
+module IOTA
+  module Crypto
+    class PrivateKey
+      HASH_LENGTH = Kerl::HASH_LENGTH
+
+      attr_reader :key
+
+      def initialize(seed, index, security)
+        key = []
+        offset = 0
+        buffer = []
+
+        (0...index).step(1) do |i|
+          # Treat ``seed`` like a really big number and add ``index``.
+          # Note that addition works a little bit differently in balanced ternary.
+          (0...seed.length).step(1) do |j|
+            seed[j] += 1
+
+            if seed[j] > 1
+              seed[j] = -1
+            else
+              break
+            end
+          end
+        end
+
+        kerl = Kerl.new
+        kerl.absorb(seed, 0, seed.length)
+        kerl.squeeze(seed, 0, seed.length)
+        kerl.reset
+        kerl.absorb(seed, 0, seed.length)
+
+        security.times do
+          (0...27).step(1) do |i|
+            kerl.squeeze(buffer, 0, seed.length)
+            (0...HASH_LENGTH).step(1) do |j|
+              key[offset] = buffer[j]
+              offset += 1
+            end
+          end
+        end
+
+        @key = key
+      end
+
+      def digests
+        digestsArray = []
+        buffer = []
+
+        (0...(@key.length / 6561).floor).step(1) do |i|
+          keyFragment = @key.slice(i * 6561, 6561)
+
+          (0...27).step(1) do |j|
+            buffer = keyFragment.slice(j * HASH_LENGTH, HASH_LENGTH);
+
+            (0...26).step(1) do |k|
+              kKerl = Kerl.new
+              kKerl.absorb(buffer, 0, buffer.length)
+              kKerl.squeeze(buffer, 0, HASH_LENGTH)
+            end
+
+            (0...HASH_LENGTH).step(1) do |k|
+              keyFragment[j * HASH_LENGTH + k] = buffer[k]
+            end
+          end
+
+          kerl = Kerl.new
+          kerl.absorb(keyFragment, 0, keyFragment.length)
+          kerl.squeeze(buffer, 0, HASH_LENGTH)
+
+          (0...HASH_LENGTH).step(1) do |j|
+            digestsArray[i * HASH_LENGTH + j] = buffer[j];
+          end
+        end
+
+        digestsArray
+      end
+    end
+  end
+end

data/lib/iota/crypto/sha3_ruby.rb

@@ -0,0 +1,122 @@
+require 'digest'
+
+module Digest
+  class RubySHA3 < Digest::Class
+    PILN = [10,  7, 11, 17, 18,  3,  5, 16,
+             8, 21, 24,  4, 15, 23, 19, 13,
+            12,  2, 20, 14, 22,  9,  6,  1]
+
+    ROTC = [ 1,  3,  6, 10, 15, 21, 28, 36,
+            45, 55,  2, 14, 27, 41, 56,  8,
+            25, 43, 62, 18, 39, 61, 20, 44]
+
+    RNDC = [0x0000000000000001, 0x0000000000008082, 0x800000000000808a,
+            0x8000000080008000, 0x000000000000808b, 0x0000000080000001,
+            0x8000000080008081, 0x8000000000008009, 0x000000000000008a,
+            0x0000000000000088, 0x0000000080008009, 0x000000008000000a,
+            0x000000008000808b, 0x800000000000008b, 0x8000000000008089,
+            0x8000000000008003, 0x8000000000008002, 0x8000000000000080,
+            0x000000000000800a, 0x800000008000000a, 0x8000000080008081,
+            0x8000000000008080, 0x0000000080000001, 0x8000000080008008]
+
+    def initialize hash_size = 512
+      @size = hash_size / 8
+      @buffer = ''
+    end
+
+    def << s
+      @buffer << s.unpack('C*').pack('C*')
+      self
+    end
+    alias update <<
+
+    def reset
+      # @buffer.clear # CHANGE: DO NOT CLEAR BUFFER AS WE NEED
+      self
+    end
+
+    def digest(data = nil)
+      if data
+        update(data)
+        value = finish
+      else
+        cloned = dup
+        value = cloned.finish
+      end
+      value
+    end
+
+    def hexdigest(data = nil)
+      value = digest(data)
+      value.unpack("H*").first
+    end
+
+    def inspect
+      "#<#{self.class}: #{hexdigest}>"
+    end
+
+    def finish
+      s = Array.new 25, 0
+      width = 200 - @size * 2
+
+      buffer_dup = @buffer.dup
+      buffer_dup << "\x01" << "\0" * (width - buffer_dup.size % width)
+      buffer_dup[-1] = (buffer_dup[-1].ord | 0x80).chr.unpack('C*').pack('C*')
+
+      0.step buffer_dup.size - 1, width do |j|
+        quads = buffer_dup[j, width].unpack 'Q*'
+        (width / 8).times do |i|
+          s[i] ^= quads[i]
+        end
+
+        keccak s
+      end
+
+      s.pack('Q*')[0, @size]
+    end
+
+    private
+    def keccak s
+      24.times.each_with_object [] do |round, a|
+        # Theta
+        5.times do |i|
+          a[i] = s[i] ^ s[i + 5] ^ s[i + 10] ^ s[i + 15] ^ s[i + 20]
+        end
+
+        5.times do |i|
+          t = a[(i + 4) % 5] ^ rotate(a[(i + 1) % 5], 1)
+          0.step 24, 5 do |j|
+            s[j + i] ^= t
+          end
+        end
+
+        # Rho Pi
+        t = s[1]
+        24.times do |i|
+          j = PILN[i]
+          a[0] = s[j]
+          s[j] = rotate t, ROTC[i]
+          t = a[0]
+        end
+
+        # Chi
+        0.step 24, 5 do |j|
+          5.times do |i|
+            a[i] = s[j + i]
+          end
+
+          5.times do |i|
+            s[j + i] ^= ~a[(i + 1) % 5] & a[(i + 2) % 5]
+          end
+        end
+
+        # Iota
+        s[0] ^= RNDC[round]
+      end
+    end
+
+    def rotate x, y
+      (x << y | x >> 64 - y) & (1 << 64) - 1
+    end
+  end
+end

data/lib/iota/crypto/signing.rb

@@ -0,0 +1,97 @@
+module IOTA
+  module Crypto
+    class Signing
+      HASH_LENGTH = Kerl::HASH_LENGTH
+
+      class << self
+        def address(digests)
+          addressTrits = []
+
+          kerl = Kerl.new
+
+          kerl.absorb(digests, 0, digests.length)
+          kerl.squeeze(addressTrits, 0, Kerl::HASH_LENGTH)
+
+          addressTrits
+        end
+
+        def digest(normalizedBundleFragment, signatureFragment)
+          buffer = []
+          kerl = Kerl.new
+
+          (0...27).step(1) do |i|
+            buffer = signatureFragment.slice(i * HASH_LENGTH, HASH_LENGTH)
+
+            j = normalizedBundleFragment[i] + 13
+
+            while j > 0
+              jKerl = Kerl.new
+              jKerl.absorb(buffer, 0, buffer.length)
+              jKerl.squeeze(buffer, 0, HASH_LENGTH)
+              j -= 1
+            end
+
+            kerl.absorb(buffer, 0, buffer.length)
+          end
+
+          kerl.squeeze(buffer, 0, HASH_LENGTH)
+          buffer
+        end
+
+        def signatureFragment(normalizedBundleFragment, keyFragment)
+          signatureFragment = keyFragment.slice(0, keyFragment.length)
+          hash = []
+
+          kerl = Kerl.new
+
+          (0...27).step(1) do |i|
+            hash = signatureFragment.slice(i * HASH_LENGTH, HASH_LENGTH)
+
+            (0...13-normalizedBundleFragment[i]).step(1) do |j|
+              kerl.reset
+              kerl.absorb(hash, 0, hash.length)
+              kerl.squeeze(hash, 0, HASH_LENGTH)
+            end
+
+            (0...HASH_LENGTH).step(1) do |j|
+              signatureFragment[i * HASH_LENGTH + j] = hash[j]
+            end
+          end
+
+          signatureFragment
+        end
+
+        def validateSignatures(expectedAddress, signatureFragments, bundleHash)
+          if !bundleHash
+            raise StandardError, "Invalid bundle hash provided"
+          end
+
+          bundle = Bundle.new
+
+          normalizedBundleFragments = []
+          normalizedBundleHash = bundle.normalizedBundle(bundleHash)
+
+          # Split hash into 3 fragments
+          (0...3).step(1) do |i|
+            normalizedBundleFragments[i] = normalizedBundleHash.slice(i * 27, 27)
+          end
+
+          # Get digests
+          digests = []
+          (0...signatureFragments.length).step(1) do |i|
+            digestBuffer = digest(normalizedBundleFragments[i % 3], Converter.trits(signatureFragments[i]))
+
+            (0...HASH_LENGTH).step(1) do |j|
+              digests[i * 243 + j] = digestBuffer[j]
+            end
+          end
+
+          addressTrits = address(digests)
+          address = Converter.trytes(addressTrits)
+
+          expectedAddress == address
+        end
+      end
+    end
+  end
+end