opensslmfc.h

MiniCA V2.0版本源码。《小型CA系统V2.1含源码》发表以来

/*
Module : OpenSSLMfc.H
Purpose: Defines the interface for wrapper classes for the OpenSSL C variable types
Created: PJN / 24-05-2002
History: None

Copyright (c) 2002 - 2005 by PJ Naughter.  

All rights reserved.

Copyright / Usage Details:

You are allowed to include the source code in any product (commercial, shareware, freeware or otherwise) 
when your product is released in binary form. You are allowed to modify the source code in any way you want 
except you cannot modify the copyright details at the top of each module. If you want to distribute source 
code with your application, then you are only allowed to distribute versions released by the author. This is 
to maintain a single distribution point for the source code. 

*/


/////////////////////////////// Defines ///////////////////////////////////////

#ifndef __OPENSSLMFC_H__
#define __OPENSSLMFC_H__

#ifndef SOCKMFC_EXT_CLASS
#define SOCKMFC_EXT_CLASS
#endif



/////////////////////////////// Includes //////////////////////////////////////

#include <openssl\ssl.h>
#include <openssl\err.h>
#include "SocMFC.h"



/////////////////////////////// Classes ///////////////////////////////////////

//class which encapsulates the OpenSSL SSL_CTX* variable
class SOCKMFC_EXT_CLASS CSSLContext 
{
public:
//Constructors / Destructors
  CSSLContext();
  CSSLContext(SSL_CTX* pSSLContext);
  CSSLContext(const CSSLContext& SSLContext);
  ~CSSLContext();

//Methods
  void Close();
  void Attach(SSL_CTX* pSSLContext);
  SSL_CTX* Detach();

//operators
  operator SSL_CTX*() const;
  CSSLContext& operator=(const CSSLContext& SSLContext);

//Data
  SSL_CTX* m_pSSLContext; //The underlying OpenSSL context variable
};



//class which encapsulates the OpenSSL SSL* variable
class SOCKMFC_EXT_CLASS CSSL 
{
public:
//Constructors / Destructors
  CSSL();
  CSSL(SSL* pSSLContext);
  CSSL(const CSSL& ssl);
  ~CSSL();

//Methods
  void Close();
  void Attach(SSL* pSSL);
  SSL* Detach();

//operators
  operator SSL*() const;
  CSSL& operator=(const CSSL& ssl);

//Data
  SSL* m_pSSL; //The underlying OpenSSL "SSL" variable
};



//Class which encapsulates a SSL socket, in OpenSSL parlance, this is an SSL* struct (aka CSSL) doing file IO over a socket (aka CWSocket)
class SOCKMFC_EXT_CLASS CSSLSocket
{
public:
//Constructors / Destructors
  CSSLSocket();
  virtual ~CSSLSocket();

//General Methods
  BOOL    Create(CSSLContext& sslContext, CWSocket& socket);
  BOOL    Connect(LPCTSTR lpszHostAddress, UINT nHostPort);
  BOOL    ConnectViaSocks4(LPCTSTR lpszHostAddress, UINT nHostPort, LPCTSTR lpszSocksServer, UINT nSocksPort, DWORD dwConnectionTimeout = 5000);
  BOOL    ConnectViaSocks5(LPCTSTR lpszHostAddress, UINT nHostPort, LPCTSTR lpszSocksServer, UINT nSocksPort, LPCTSTR lpszUserName = NULL, LPCTSTR lpszPassword = NULL, DWORD dwConnectionTimeout = 5000, BOOL bUDP = FALSE);
  BOOL    ConnectViaHTTPProxy(LPCTSTR lpszHostAddress, UINT nHostPort, LPCTSTR lpszHTTPServer, UINT nHTTPProxyPort, CString& sProxyResponse, LPCTSTR lpszUserName = NULL, LPCTSTR pszPassword = NULL, DWORD dwConnectionTimeout = 5000, LPCTSTR lpszUserAgent = NULL);
  BOOL    Accept(DWORD dwSSLNegotiationTimeout);
  void    Close();
  int     Send(void* pBuffer, int nBuf);
  int     Receive(void* pBuffer, int nBuf);
  BOOL    IsReadible(DWORD dwTimeout);

//operators
  operator SSL*() const;
  operator CWSocket&() const;

protected:
  CSSL      m_SSL;
  CWSocket* m_pSocket;
};

#endif //__OPENSSLMFC_H__