lib/noise/protocol.rb in noise-ruby-0.5.2 vs lib/noise/protocol.rb in noise-ruby-0.5.3

@@ -7,10 +7,11 @@
     attr_accessor :cipher_state_handshake
     attr_accessor :psks
     attr_reader :name, :cipher_fn, :hash_fn, :dh_fn, :hkdf_fn, :pattern
     attr_reader :handshake_state, :keypairs, :keypair_fn
     attr_reader :handshake_hash
+    attr_accessor :ck
 
     CIPHER = {
       'AESGCM': Noise::Functions::Cipher::AesGcm,
       'ChaChaPoly': Noise::Functions::Cipher::ChaChaPoly
     }.stringify_keys.freeze
@@ -64,9 +65,10 @@
       end
       @handshake_hash = @symmetric_state.handshake_hash
       @keypairs = @handshake_state.keypairs
 
       @handshake_state = nil
+      @ck = @symmetric_state.ck
       @symmetric_state = nil
       @cipher_state_handshake = nil
       @prologue = nil
       @initiator = nil
       @dh_fn = nil