noise_handshake.hpp

Go to the documentation of this file.

#pragma once
 
 
#include <esphome/api/bytes.hpp>
#include <esphome/api/crypto/cipher_state.hpp>
#include <esphome/api/crypto/primitives.hpp>
#include <esphome/api/crypto/symmetric_state.hpp>
 
namespace esphome::api::noise {
 
inline constexpr auto protocol_name = "Noise_NNpsk0_25519_ChaChaPoly_SHA256";
 
class NoiseHandshake {
public:
    enum class Role { Initiator, Responder };
 
    void initialize(Role role, const SymmetricKey& psk, ByteView prologue);
 
    ByteBuffer write_message(ByteView payload = ByteView{});
 
    bool read_message(ByteView message, ByteBuffer& out_payload);
 
    [[nodiscard]] bool is_complete() const noexcept {
        return complete_;
    }
 
    void split(CipherState& send, CipherState& recv);
 
    [[nodiscard]] const Hash& handshake_hash() const noexcept {
        return symmetric_.handshake_hash();
    }
 
    void set_ephemeral_for_testing(const PrivateKey& private_key);
 
private:
    void ensure_ephemeral();
 
    Role role_ = Role::Initiator;
    SymmetricKey psk_{};
    SymmetricState symmetric_;
    PrivateKey ephemeral_private_{};
    PublicKey ephemeral_public_{};
    PublicKey remote_ephemeral_{};
    bool have_ephemeral_ = false;
    bool complete_ = false;
};
 
}  // namespace esphome::api::noise