/*

* TLS v1.0 and v1.2 PRFs

* (C) 2004-2010 Jack Lloyd

*

* Botan is released under the Simplified BSD License (see license.txt)

*/

#ifndef BOTAN_TLS_PRF_H_

#define BOTAN_TLS_PRF_H_

#include <botan/kdf.h>

#include <botan/mac.h>

namespace Botan {

/**

* PRF used in TLS 1.0/1.1

*/

class TLS_PRF final : public KDF

   {

   public:

      std::string name() const override { return "TLS-PRF"; }

      KDF* clone() const override { return new TLS_PRF; }

      size_t kdf(uint8_t key[], size_t key_len,

                 const uint8_t secret[], size_t secret_len,

                 const uint8_t salt[], size_t salt_len,

                 const uint8_t label[], size_t label_len) const override;

      TLS_PRF(std::unique_ptr<MessageAuthenticationCode> hmac_md5,

              std::unique_ptr<MessageAuthenticationCode> hmac_sha1) :

         m_hmac_md5(std::move(hmac_md5)),

         m_hmac_sha1(std::move(hmac_sha1))

         {}

      TLS_PRF();

   private:

      std::unique_ptr<MessageAuthenticationCode> m_hmac_md5;

      std::unique_ptr<MessageAuthenticationCode> m_hmac_sha1;

   };

/**

* PRF used in TLS 1.2

*/

class TLS_12_PRF final : public KDF

   {

   public:

      std::string name() const override { return "TLS-12-PRF(" + m_mac->name() + ")"; }

      KDF* clone() const override { return new TLS_12_PRF(m_mac->clone()); }

      size_t kdf(uint8_t key[], size_t key_len,

                 const uint8_t secret[], size_t secret_len,

                 const uint8_t salt[], size_t salt_len,

                 const uint8_t label[], size_t label_len) const override;

      /**

      * @param mac MAC algorithm to use

      */

      explicit TLS_12_PRF(MessageAuthenticationCode* mac) : m_mac(mac) {}

   private:

      std::unique_ptr<MessageAuthenticationCode> m_mac;

   };

}

#endif