Somewhat cleaner PSK handling

1 files changed, 4 insertions, 3 deletions

diff --git a/src/tls/tls_messages.h b/src/tls/tls_messages.h
index b29aac600..97d72a0cb 100644
--- a/src/tls/tls_messages.h
+++ b/src/tls/tls_messages.h
@@ -214,7 +214,9 @@ class Client_Key_Exchange : public Handshake_Message
          { return pre_master; }
 
       SecureVector<byte> pre_master_secret(RandomNumberGenerator& rng,
-                                           const Handshake_State* state);
+                                           const Handshake_State* state,
+                                           Credentials_Manager& creds,
+                                           const Policy& policy);
 
       Client_Key_Exchange(Record_Writer& output,
                           Handshake_State* state,
@@ -223,8 +225,7 @@ class Client_Key_Exchange : public Handshake_Message
                           RandomNumberGenerator& rng);
 
       Client_Key_Exchange(const MemoryRegion<byte>& buf,
-                          const Handshake_State* state,
-                          Credentials_Manager& creds);
+                          const Handshake_State* state);
 
    private:
       MemoryVector<byte> serialize() const { return key_material; }