SshSocket.h

/*NB_REVISION*/
 
/*NB_COPYRIGHT*/
 
#ifndef _NB_SSH_SOCKET_H_
#define _NB_SSH_SOCKET_H_
 
#include <predef.h>
 
#ifdef NB_SSH_SUPPORTED
#include <nbrtos.h>
 
#include <crypto/NetBurner/CryptoSocket.h>
#include <crypto/wolfssl/wolfcrypt/sha256.h>
#include <crypto/wolfssl/wolfcrypt/coding.h>
#include <ssh/wolfssh/ssh.h>
 
class SshSocket : public CryptoSocket
{
   public:
    SshSocket();
    ~SshSocket();
 
    int InitSocket(int tcpFd, WOLFSSH_CTX *ctx, int sockFlags = 0);
 
    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 SshSocket *GetNewSocket(int tcpFd, WOLFSSH_CTX *ctx, int sockFlags = 0);
 
    WOLFSSH *GetWolfSsh() { return (WOLFSSH *)m_wolfCtx; }
 
   private:
    static SshSocket *FindNextEmptySocket();
    void WriteUnwrittenData() override;
 
    static SshSocket gSshSockets[];
    static Pool      gSshSocketPool;
   public:
    inline static const Pool *GetSocketPool() { return &gSshSocketPool; }
};
 
#endif   // NB_SSH_SUPPORTED
#endif   // _NB_SSH_SOCKET_H_