zt-tls-handshake.h

Go to the documentation of this file.

// zt-tls-handshake.h
 
#ifndef ZT_TLS_HANDSHAKE_H
#define ZT_TLS_HANDSHAKE_H
 
#include "ns3/object.h"
#include "ns3/node.h"
#include <map>
#include <string>
#include <functional>
 
namespace ns3 {
 
class ZtTlsHandshake : public Object {
public:
  static TypeId GetTypeId();
 
  ZtTlsHandshake();
 
  void StartHandshake(Ptr<Node> client, Ptr<Node> server, uint32_t clientId, uint32_t serverId);
 
  bool HasSession(uint32_t peerId) const;
 
  std::string GetSessionKey(uint32_t peerId) const;
 
  void SetExternalLogger(std::function<void(std::string)> logger);
 
  void SetPolicyValidator(std::function<bool(uint32_t, std::string)> validator);
 
private:
  void Log(const std::string& msg) const;
 
  std::map<uint32_t, std::string> m_sessionKeys;               
  std::function<void(std::string)> m_logger;                   
  std::function<bool(uint32_t, std::string)> m_policyValidator; 
};
 
} // namespace ns3
 
#endif // ZT_TLS_HANDSHAKE_H