dtlsClient.hpp

Go to the documentation of this file.

#include <array>
#include <cstdint>
#include <iostream>

#include <openssl/bio.h>
#include <openssl/conf.h>
#include <openssl/dh.h>
#include <openssl/engine.h>
#include <openssl/err.h>
#include <openssl/ssl.h>

#include "IPv4_5TupleL2Ident.hpp"
#include "mbuf.hpp"
#include "stateMachine.hpp"

namespace DTLS_Client {

struct dtlsClient {
    SSL *ssl;
    BIO *wbio;
    BIO *rbio;
    uint32_t localIP;
    uint32_t remoteIP;
    uint16_t localPort;
    uint16_t remotePort;
    uint16_t counter;
    std::array<uint8_t, 6> localMac;
    std::array<uint8_t, 6> remoteMac;
};

struct States {
    static constexpr StateID DOWN = 0;
    static constexpr StateID HANDSHAKE = 1;
    static constexpr StateID ESTABLISHED = 2;
    static constexpr StateID RUN_TEARDOWN = 3;
    static constexpr StateID DELETED = 4;
};

SSL_CTX *createCTX();

void configStateMachine(StateMachine<IPv4_5TupleL2Ident<mbuf>, mbuf> &sm);

StateMachine<IPv4_5TupleL2Ident<mbuf>, mbuf>::State createStateData(SSL_CTX *ctx,
    uint32_t localIP, uint32_t remoteIP, uint16_t localPort, uint16_t remotePort,
    std::array<uint8_t, 6> localMac, std::array<uint8_t, 6> remoteMac);

/*
 * The following functions are the state functions, as they should be registered
 * in the StateMachine<>
 */

void initHandshake(StateMachine<IPv4_5TupleL2Ident<mbuf>, mbuf>::State &state, mbuf *,
    StateMachine<IPv4_5TupleL2Ident<mbuf>, mbuf>::FunIface &funIface);

void runHandshake(StateMachine<IPv4_5TupleL2Ident<mbuf>, mbuf>::State &state, mbuf *,
    StateMachine<IPv4_5TupleL2Ident<mbuf>, mbuf>::FunIface &funIface);

void sendData(StateMachine<IPv4_5TupleL2Ident<mbuf>, mbuf>::State &state, mbuf *,
    StateMachine<IPv4_5TupleL2Ident<mbuf>, mbuf>::FunIface &funIface);

void runTeardown(StateMachine<IPv4_5TupleL2Ident<mbuf>, mbuf>::State &state, mbuf *,
    StateMachine<IPv4_5TupleL2Ident<mbuf>, mbuf>::FunIface &funIface);

}; // namespace DTLS_Client

extern "C" {
void *DtlsClient_init(uint32_t dstIP, uint16_t dstPort, uint8_t srcMac[6], uint8_t dstMac[6]);

void *DtlsClient_connect(void *obj, struct rte_mbuf **inPkts, unsigned int inCount,
    unsigned int *sendCount, unsigned int *freeCount, uint32_t srcIP, uint16_t srcPort);

void DtlsClient_getPkts(void *obj, struct rte_mbuf **sendPkts, struct rte_mbuf **freePkts);

void *DtlsClient_process(void *obj, struct rte_mbuf **inPkts, unsigned int inCount,
    unsigned int *sendCount, unsigned int *freeCount);

void DtlsClient_free(void *obj);
}