cSSLChannel.h

Go to the documentation of this file.

//
//
#ifndef _INC_cSSLChannel_H
#define _INC_cSSLChannel_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "cSSLMsg.h"
#include "cSSLSession.h"
#include "../Net/cProtocol.h"
#include "../Cipher/cCipherBase.h"
#include "GrayCore/include/cTimeInt.h"
#include "GrayCore/include/cArray.h"
#include "GrayCore/include/cUnitTestDecl.h"
#include "GrayCore/include/HResult.h"
#include "GrayCore/include/cNewPtr.h"
 
namespace GrayLib
{
        UNITTEST2_PREDEF(cSSLChannel);
 
        //***************************************************
        //** TODO
 
        class GRAYLIB_LINK cSSLChannel
        {
 
        public:
                bool m_bFinished;
                cSSLMsgHandHello m_Hello;                       
                cSSLSession m_Neg;                                      
                cSSLSession m_Cur;                                      
                cNewPtr<cCipherBase> m_pCrypt;          
 
        public:
                cSSLChannel();
                virtual ~cSSLChannel();
 
                void ChangeCipherSpec();
 
                UNITTEST2_PREDEF(cSSLChannel);
        };
 
        class GRAYLIB_LINK CSSLClient : public cSSLChannel
        {
        public:
                cStream* m_pStream;             
 
        public:
                CSSLClient(cStream* pStream);
                virtual ~CSSLClient();
 
                HRESULT SendClientHello(int iVerBest, void* pRandomNumber, cArrayVal<SSL_CipherSuite_t>* pSupport, void* pSessionIdResume);
                HRESULT SendCertificate();
                HRESULT SendClientKeyExchange(bool bCertificateVerify);
                HRESULT SendChangeCipherSpec();
 
                HRESULT OnServerHello();
                HRESULT OnCertificate();
                HRESULT OnServerHelloDone();
                HRESULT OnChangeCipherSpec();
                HRESULT OnFinished();
 
                cStreamOutput* get_SSLOut() const;
                cStreamInput* get_SSLInp() const;
        };
 
        class GRAYLIB_LINK cSSLServerFactory;
 
        class GRAYLIB_LINK cSSLServerStream : public cRefBase, public cSSLChannel, public cProtocolStream
        {
 
        public:
                cProtocolStreamPtr m_pProtocol2;        
 
        protected:
 
                HRESULT OnHand_ClientHello(const BYTE* pHandshakeData, DWORD dwDataLength);
 
                HRESULT OnAlert(const cSSLMsgX& msg);
                HRESULT OnChangeCipherSpec(const cSSLMsgX& msg);
                HRESULT OnHandshake(const cSSLMsgX& msg);
                HRESULT OnApplication(const cSSLMsgX& msg);
 
                virtual HRESULT ProcessStreamPacket() override;
 
        public:
                cSSLServerStream(cSSLServerFactory* pFactory, IUnkObject* pServerConnection, cStreamInput* pStreamInp, cStreamOutput* pStreamOut);
                virtual ~cSSLServerStream();
 
                IUNKNOWN_DISAMBIG(cRefBase);
        };
 
        typedef cRefPtr<cSSLServerStream> cSSLServerStreamPtr;
 
        class GRAYLIB_LINK cSSLServerFactory : public IProtocolFactory
        {
 
        protected:
                cProtocolFactories m_Factories;         
 
        public:
                cSSLServerFactory();
                cSSLServerFactory(IProtocolFactory* pProtocol2);
 
                virtual const char* get_ProtocolName() const override
                {
                        return "SSL";
                }
                virtual HRESULT TestProtocol(const BYTE* pszPrefixData, size_t iPrefixLen) const override;
                virtual cProtocolStreamPtr CreateProtocolStream(IUnkObject* pServerConnection, cStreamInput* pStreamInp, cStreamOutput* pStreamOut) override
                {
                        return new cSSLServerStream(this, pServerConnection, pStreamInp, pStreamOut);
                }
        };
}
#endif