noise_handshake.h

Go to the documentation of this file.

#pragma once
#include "esphome/core/defines.h"
#ifdef USE_NOISE
#include <cstddef>
#include <cstdint>
 
#include <noise/protocol.h>
 
#include "noise.h"
 
namespace esphome::noise {
 
class NoiseResponderHandshake {
 public:
  // The ACTION_ prefix is macro-collision safety: SDK headers #define bare
  // names like READ/WRITE, and macros expand even inside an enum class.
  enum class Action : uint8_t { ACTION_READ, ACTION_WRITE, ACTION_SPLIT, ACTION_FAILED };
 
  NoiseResponderHandshake() = default;
  ~NoiseResponderHandshake();
  // Owns a raw noise-c handshake state; copying would double free it
  NoiseResponderHandshake(const NoiseResponderHandshake &) = delete;
  NoiseResponderHandshake &operator=(const NoiseResponderHandshake &) = delete;
 
  [[nodiscard]] int init(const psk_t &psk, const uint8_t *prologue, size_t prologue_len);
  [[nodiscard]] Action action() const;
  [[nodiscard]] int read_message(uint8_t *data, size_t len);
  [[nodiscard]] int write_message(uint8_t *out, size_t capacity, size_t &out_len);
  [[nodiscard]] int split(NoiseCipherState *&send_cipher, NoiseCipherState *&recv_cipher);
 
 protected:
  int fail_init_(int err);
 
  NoiseHandshakeState *handshake_{nullptr};
};
 
}  // namespace esphome::noise
#endif  // USE_NOISE