SslContext.h

Go to the documentation of this file.

// SslContext.h

// Declares the cSslContext class that holds everything a single SSL context needs to function





#pragma once

#include "mbedtls/ssl.h"
#include "../ByteBuffer.h"





// fwd:
class cCtrDrbgContext;
class cSslConfig;





class cSslContext abstract
{
public:
    cSslContext(void);

    virtual ~cSslContext();

    int Initialize(std::shared_ptr<const cSslConfig> a_Config);

    int Initialize(bool a_IsClient);

    bool IsValid(void) const { return m_IsValid; }

    void SetExpectedPeerName(const AString & a_ExpectedPeerName);

    int WritePlain(const void * a_Data, size_t a_NumBytes);

    int ReadPlain(void * a_Data, size_t a_MaxBytes);

    int Handshake(void);

    bool HasHandshaken(void) const { return m_HasHandshaken; }

    int NotifyClose(void);

protected:

    std::shared_ptr<const cSslConfig> m_Config;

    mbedtls_ssl_context m_Ssl;

    bool m_IsValid;

    bool m_HasHandshaken;

    static int ReceiveEncrypted(void * a_This, unsigned char * a_Buffer, size_t a_NumBytes)
    {
        return (static_cast<cSslContext *>(a_This))->ReceiveEncrypted(a_Buffer, a_NumBytes);
    }

    static int SendEncrypted(void * a_This, const unsigned char * a_Buffer, size_t a_NumBytes)
    {
        return (static_cast<cSslContext *>(a_This))->SendEncrypted(a_Buffer, a_NumBytes);
    }

    virtual int ReceiveEncrypted(unsigned char * a_Buffer, size_t a_NumBytes) = 0;

    virtual int SendEncrypted(const unsigned char * a_Buffer, size_t a_NumBytes) = 0;
} ;