lib/lightning/onion/packet.rb in lightning-onion-0.2.3 vs lib/lightning/onion/packet.rb in lightning-onion-0.2.4

@@ -1,22 +1,25 @@
 # frozen_string_literal: true
 
 module Lightning
   module Onion
     class Packet
+      include Lightning::Onion::Sphinx
+
       attr_accessor :version, :public_key, :routing_info, :hmac
+
       def initialize(version, public_key, routing_info, hmac)
         @version = version
         @public_key = public_key
         @routing_info = routing_info
-        raise "invalid size #{routing_info.size}" unless routing_info.size == 1300 * 2
+        raise "invalid size #{routing_info.size}" unless routing_info.size == MAX_HOPS * HOP_LENGTH * 2
         @hmac = hmac
       end
 
       def self.parse(payload)
         version, public_key, rest = payload.unpack('aH66a*')
-        routing_info = rest[0...20 * 65].bth
-        hmac = rest[20 * 65..-1].bth
+        routing_info = rest[0...MAX_HOPS * HOP_LENGTH].bth
+        hmac = rest[MAX_HOPS * HOP_LENGTH..-1].bth
         new(version, public_key, routing_info, hmac)
       end
 
       def to_payload
         payload = +''