transport.hxx

// dawn/src/proto/wakaama/transport.hxx
//
// SPDX-License-Identifier: Apache-2.0
//
 
#pragma once
 
#include "internal.hxx"
 
#include <netinet/in.h>
#include <sys/socket.h>
#include <vector>
 
#ifdef CONFIG_DAWN_PROTO_WAKAAMA_DTLS_PSK
extern "C"
{
#  include <tinydtls/dtls.h>
}
#endif
 
namespace dawn
{
namespace wakaama_internal
{
class Transport
{
public:
  struct Connection
  {
    Connection *next;
    int sock;
    uint16_t securityInstanceId;
    const uint8_t *pskIdentity;
    size_t pskIdentityLen;
    const uint8_t *pskKey;
    size_t pskKeyLen;
    sockaddr_storage addr;
    socklen_t addrLen;
#ifdef CONFIG_DAWN_PROTO_WAKAAMA_DTLS_PSK
    session_t dtlsSession;
    dtls_context_t *dtlsContext;
    time_t lastSend;
    bool dtls;
#endif
  };
 
  explicit Transport(CProtoWakaama &owner);
 
  ~Transport();
 
  int initConnectionPool();
 
  int openSocket();
 
  int initDtls();
 
  void destroyDtls();
 
  void closeAllConnections();
 
  void thread();
 
  Connection *connectServer(uint16_t securityInstanceId);
 
  Connection *findConnection(const void *addr, size_t addrLen);
 
  void handlePacket(Connection *conn, uint8_t *buffer, size_t length);
 
  void closeConnection(Connection *conn);
 
private:
  bool receiveOne(uint8_t *buffer, size_t capacity);
 
  CProtoWakaama &owner;
  Connection *connections;
  std::vector<Connection *> connectionPool;
  int sockfd;
#ifdef CONFIG_DAWN_PROTO_WAKAAMA_DTLS_PSK
  dtls_context_t *dtlsContext;
  dtls_handler_t dtlsHandler;
#endif
};
 
} // namespace wakaama_internal
} // namespace dawn