SslSocket.h

/*NB_REVISION*/
 
/*NB_COPYRIGHT*/
 
#ifndef _SSL_SOCKET_H_
#define _SSL_SOCKET_H_
 
#ifdef NB_SSL_SUPPORTED
 
#include <buffers.h>
#include <netinterface.h>
 
#include <crypto/wolfssl/ssl.h>
#include <crypto/NetBurner/CryptoSocket.h>
#include <crypto/ssl.h>
 
// Time stuff
#define SHUTDOWN_RETRY_INTERVAL_IMMEDIATE \
    (1)   // Because of our asynchronous connections, we will generally have to try at least twice to completely
          // shutdown with WolfSSL. However, the time between these calls can be extremely short.
#define SHUTDOWN_RETRY_INTERVAL (TICKS_PER_SECOND)
#define CONNECT_RETRY_INTERVAL (TICKS_PER_SECOND)
 
class SslSocket : public CryptoSocket
{
   public:
    SslSocket();
    ~SslSocket();
 
    uint16_t InitSocket(int tcpFd, WOLFSSL_CTX *ctx, const char *commonName, uint32_t sockFlags = 0, int verifyPeer = -1);
 
    void CleanupTcpClose();
 
    // Called from wolf ssl loop mostly to handle notifications and
    // time out stuff for all async process except read
    void ProcessAsyncStuff() override;
    int CheckSocketRecv() override;
    uint32_t SocketRead(char *buf, uint32_t len) override;
    uint32_t SocketWrite(const char *buf, uint32_t len) override;
    SocketHasData_t SocketHasData() override;
 
    // Need to add a GetNewSocket() for each derived type
    static SslSocket *GetNewSocket(int tcpFd, WOLFSSL_CTX *ctx, const char *commonName, uint32_t sockFlags = 0, int verifyPeer = -1);
 
    inline WOLFSSL *GetWolfSsl() { return (WOLFSSL *)m_wolfCtx; }
 
    static bool SocketsAvail();
    static int FreeSockets();
 
   private:
    int ticketLen;
 
    void WriteUnwrittenData() override;
    virtual bool PendOnHandshake() { return (m_error == SSL_ERROR_HANDSHAKE_INCOMPLETE); }
 
    static int DoTicketCallback(WOLFSSL *ssl, const unsigned char *ticket, int ticketLen, void* pvSock);
 
    static SslSocket *FindNextEmptySocket();
    static SslSocket gSslSockets[];
    static Pool      gSslSocketPool;
    friend void SSL_DumpSSLSockets();
    friend void  SSL_dumpJSONSockets(int fd);
   public:
    inline static const Pool *GetSocketPool() { return &gSslSocketPool; }
};
 
//extern SslSocket gSslSockets[];
 
#endif /* NB_SSL_SUPPORTED */
#endif