SecureSocket.h

Go to the documentation of this file.

/*  SecureSocket.h - SSL communication through the Schannel API
    Copyright (C) 2001-2004 Mark Weaver
    Written by Mark Weaver <mark@npsl.co.uk>

    Part of the Open-Win32 library.
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Library General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.

    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    Library General Public License for more details.

    You should have received a copy of the GNU Library General Public
    License along with this library; if not, write to the
    Free Software Foundation, Inc., 59 Temple Place - Suite 330,
    Boston, MA  02111-1307, USA.
*/

#ifndef OW32_SecureSocket_h
#define OW32_SecureSocket_h

#include <OW32/Socket.h>
#include <OW32/auto_array_ptr.h>
#include <OW32/OW32Libs.h>

// Open Win32 namespace
namespace OW32
{

#ifdef _MSC_VER
#pragma warning(disable: 4251)
#endif

class OW32_LIB_EXPORT CSecureSocket : 
    public CSocket
{
private:
    CSecureSocket(const CSecureSocket& );
    CSecureSocket& operator= (const CSecureSocket& );
    void initialise();

protected:
    // The handshake loop
    virtual SECURITY_STATUS handshakeLoop(bool fDoInitialRead)=0;

public:
    CSecureSocket();
    CSecureSocket(SOCKET s);
    ~CSecureSocket();

    // Handshake, authentication, that sort of thing.
    virtual bool negotiate()=0;

    virtual int send(const char* buf, int len);
    virtual int recv(char* buf, int len);

    void setCredentials(CredHandle hCreds) { m_hCreds = hCreds; }
    void setRequireClientAuth(bool bRequireClientAuth) { m_bRequireClientAuth = bRequireClientAuth; }

    static SECURITY_STATUS createCredentialsFromCertificate(CredHandle* phCreds, PCCERT_CONTEXT pCertContext,
            DWORD dwDirection, DWORD dwEnabledProtocols = 0);

    SECURITY_STATUS getRemoteCert(PCCERT_CONTEXT* pRemoteCertContext);

    void freeCredentials();

protected:
    SECURITY_STATUS querySizes();

    SecPkgContext_StreamSizes m_Sizes;
    CtxtHandle m_hContext;
    CredHandle m_hCreds;

    auto_array_ptr<char> m_Extra;
    int m_ExtraCount;
    int m_ExtraDecrypted,m_ExtraDecryptedPos;

    auto_array_ptr<char> m_SendBuf;

    bool m_bRequireClientAuth;
    bool m_bHandshakeDone;
    bool m_ownCredentials;
};

#ifdef _MSC_VER
#pragma warning(default: 4251)
#endif

} // namespace OW32

#endif // OW32_SecureSocket_h

---