pssl.h

Go to the documentation of this file.

/*
 * pssl.h
 *
 * Secure Sockets Layer channel interface class.
 *
 * Portable Windows Library
 *
 * Copyright (c) 1993-2002 Equivalence Pty. Ltd.
 *
 * The contents of this file are subject to the Mozilla Public License
 * Version 1.0 (the "License"); you may not use this file except in
 * compliance with the License. You may obtain a copy of the License at
 * http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
 * the License for the specific language governing rights and limitations
 * under the License.
 *
 * The Original Code is Portable Windows Library.
 *
 * The Initial Developer of the Original Code is Equivalence Pty. Ltd.
 *
 * Contributor(s): ______________________________________.
 *
 * $Revision: 30209 $
 * $Author: rjongbloed $
 * $Date: 2013-07-31 14:51:00 +1000 (Wed, 31 Jul 2013) $
 */

#ifndef PTLIB_PSSL_H
#define PTLIB_PSSL_H

#ifdef P_USE_PRAGMA
#pragma interface
#endif

#include <ptlib/sockets.h>


struct ssl_st;
struct ssl_ctx_st;
struct x509_st;
struct X509_name_st;
struct evp_pkey_st;
struct dh_st;
struct aes_key_st;
struct SHAstate_st;

enum PSSLFileTypes {
  PSSLFileTypePEM,
  PSSLFileTypeASN1,
  PSSLFileTypeDEFAULT
};


typedef PNotifierTemplate<bool> PSSLPasswordNotifier;
#define PDECLARE_SSLPasswordNotifier(cls, fn) PDECLARE_NOTIFIER2(PString, cls, fn, bool)


class PSSLPrivateKey : public PObject
{
  PCLASSINFO(PSSLPrivateKey, PObject);
  public:
    PSSLPrivateKey();

    PSSLPrivateKey(
      unsigned modulus,   
      void (*callback)(int,int,void *) = NULL,  
      void *cb_arg = NULL                       
    );

    PSSLPrivateKey(
      const PFilePath & keyFile,  
      PSSLFileTypes fileType = PSSLFileTypeDEFAULT  
    );

    PSSLPrivateKey(
      const BYTE * keyData,   
      PINDEX keySize          
    );

    PSSLPrivateKey(
      const PBYTEArray & keyData  
    );

    PSSLPrivateKey(
      const PSSLPrivateKey & privKey
    );
    PSSLPrivateKey(
      evp_pkey_st * privKey,
      bool duplicate = true
    );

    PSSLPrivateKey & operator=(
      const PSSLPrivateKey & privKay
    );
    PSSLPrivateKey & operator=(
      evp_pkey_st * privKay
    );

    ~PSSLPrivateKey();

    operator evp_pkey_st *() const { return m_pkey; }

    void Attach(evp_pkey_st * key);

    PBoolean Create(
      unsigned modulus,   
      void (*callback)(int,int,void *) = NULL,  
      void *cb_arg = NULL                       
    );

    bool IsValid() const { return m_pkey != NULL; }

    bool SetData(
      const PBYTEArray & data
    );

    PBYTEArray GetData() const;

    PString AsString() const;

    bool Parse(
      const PString & keyStr
    );

    PBoolean Load(
      const PFilePath & keyFile,  
      PSSLFileTypes fileType = PSSLFileTypeDEFAULT,  
      const PSSLPasswordNotifier & notifier = PSSLPasswordNotifier()  
    );

    PBoolean Save(
      const PFilePath & keyFile,  
      PBoolean append = false,        
      PSSLFileTypes fileType = PSSLFileTypeDEFAULT  
    );


  protected:
    void FreePrivateKey();
    evp_pkey_st * m_pkey;
};


class PSSLCertificate : public PObject
{
    PCLASSINFO(PSSLCertificate, PObject);
  public:
    PSSLCertificate();

    PSSLCertificate(
      const PFilePath & certFile, 
      PSSLFileTypes fileType = PSSLFileTypeDEFAULT  
    );

    PSSLCertificate(
      const BYTE * certData,  
      PINDEX certSize        
    );

    PSSLCertificate(
      const PBYTEArray & certData  
    );

    PSSLCertificate(
      const PString & certString  
    );

    PSSLCertificate(
      const PSSLCertificate & cert
    );
    PSSLCertificate(
      x509_st * cert,
      bool duplicate = true
    );

    PSSLCertificate & operator=(
      const PSSLCertificate & cert
    );
    PSSLCertificate & operator=(
      x509_st * cert
    );

    ~PSSLCertificate();

    operator x509_st *() const { return m_certificate; }

    void Attach(x509_st * cert);

    bool IsValid() const { return m_certificate != NULL; }

    
    PBoolean CreateRoot(
      const PString & subject,    
      const PSSLPrivateKey & key  
    );

    bool SetData(
      const PBYTEArray & data
    );

    PBYTEArray GetData() const;

    PString AsString() const;

    bool Parse(
      const PString & certStr
    );

    PBoolean Load(
      const PFilePath & certFile, 
      PSSLFileTypes fileType = PSSLFileTypeDEFAULT  
    );

    PBoolean Save(
      const PFilePath & keyFile,  
      PBoolean append = false,        
      PSSLFileTypes fileType = PSSLFileTypeDEFAULT  
    );

    class X509_Name : public PObject {
        PCLASSINFO(X509_Name, PObject);
      public:
        X509_Name(X509_name_st * name = NULL) : m_name(name) { }

        Comparison Compare(const PObject & other) const;
        void PrintOn(ostream & strm) const;

        bool IsValid() const { return m_name != NULL; }

        PString GetCommonName() const;
        PString GetNID(int id) const;
        PString AsString(
          int indent = -1  // Negative means single line
        ) const;

      protected:
        X509_name_st * m_name;
    };

    bool GetIssuerName(X509_Name & name) const;

    bool GetSubjectName(X509_Name & name) const;
    PString GetSubjectName() const;

    PString GetSubjectAltName() const;

    virtual void PrintOn(ostream & strm) const { strm << GetSubjectName(); }

  protected:
    void FreeCertificate();
    x509_st * m_certificate;
};


class PSSLDiffieHellman : public PObject
{
  PCLASSINFO(PSSLDiffieHellman, PObject);
  public:
    PSSLDiffieHellman();

    PSSLDiffieHellman(
      const PFilePath & dhFile, 
      PSSLFileTypes fileType = PSSLFileTypeDEFAULT  
    );

    PSSLDiffieHellman(
      const BYTE * pData, 
      PINDEX pSize,       
      const BYTE * gData, 
      PINDEX gSize        
    );

    PSSLDiffieHellman(
      const PSSLDiffieHellman & dh
    );

    PSSLDiffieHellman & operator=(
      const PSSLDiffieHellman & dh
    );

    ~PSSLDiffieHellman();

    operator dh_st *() const { return dh; }

    PBoolean Load(
      const PFilePath & dhFile, 
      PSSLFileTypes fileType = PSSLFileTypeDEFAULT  
    );

  protected:
    dh_st * dh;
};


#ifdef P_SSL_AES

class PAESContext : public PObject
{
  PCLASSINFO(PAESContext, PObject);
  public:
    PAESContext();
    PAESContext(bool encrypt, const void * key, PINDEX numBits);
    ~PAESContext();

    void SetEncrypt(const void * key, PINDEX numBits);
    void SetDecrypt(const void * key, PINDEX numBits);

    void Encrypt(const void * in, void * out);
    void Decrypt(const void * in, void * out);

  protected:
    aes_key_st * m_key;
};
#endif // P_SSL_AES


class PSHA1Context : public PObject
{
  PCLASSINFO(PSHA1Context, PObject);
  public:
    PSHA1Context();
    ~PSHA1Context();

    enum { BlockSize = 64 };

    void Update(const void * data, PINDEX length);
    void Update(const PString & str) { Update((const char *)str, str.GetLength()); }

    typedef BYTE Digest[20];
    void Finalise(Digest result);

    static void Process(const void * data, PINDEX length, Digest result);
    static void Process(const PString & str, Digest result) { Process((const char *)str, str.GetLength(), result); }

  protected:
    SHAstate_st * m_context;
};


class PSSLContext : public PObject
{
    PCLASSINFO(PSSLContext, PObject);
  public:
    enum Method {
      SSLv23,
      SSLv3,
      TLSv1
    };

    PSSLContext(
      const void * sessionId = NULL,  
      PINDEX idSize = 0               
    );
    PSSLContext(
      Method method,                  
      const void * sessionId = NULL,  
      PINDEX idSize = 0               
    );

    ~PSSLContext();

    operator ssl_ctx_st *() const { return m_context; }

    bool SetVerifyLocations(
      const PFilePath & caFile, 
      const PDirectory & caDir  
    );

    bool SetVerifyCertificate(
      const PSSLCertificate & cert
    );

    P_DECLARE_ENUM(VerifyMode,
      VerifyNone,
      VerifyPeer,
      VerifyPeerMandatory
    );

    void SetVerifyMode(
      VerifyMode mode,    
      unsigned depth = 9  
    );

    VerifyMode GetVerifyMode() const;

    bool AddClientCA(
      const PSSLCertificate & certificate
    );
    bool AddClientCA(
      const PList<PSSLCertificate> & certificates
    );

    bool UseCertificate(
      const PSSLCertificate & certificate
    );

    bool UsePrivateKey(
      const PSSLPrivateKey & key
    );

    bool UseDiffieHellman(
      const PSSLDiffieHellman & dh
    );

    bool SetCipherList(
      const PString & ciphers   
    );

    bool SetCredentials(
      const PString & authority,    
      const PString & certificate,  
      const PString & privateKey,   
      bool create = false           
    );

    void SetPasswordNotifier(
      const PSSLPasswordNotifier & notifier   
    );

  protected:
    void Construct(Method method, const void * sessionId, PINDEX idSize);

    ssl_ctx_st * m_context;
    PSSLPasswordNotifier m_passwordNotifier;
};


class PSSLChannel : public PIndirectChannel
{
  PCLASSINFO(PSSLChannel, PIndirectChannel)
  public:
    PSSLChannel(
      PSSLContext * context = NULL,   
      PBoolean autoDeleteContext = false  
    );
    PSSLChannel(
      PSSLContext & context           
    );

    ~PSSLChannel();

    // Overrides from PChannel
    virtual PBoolean Read(void * buf, PINDEX len);
    virtual PBoolean Write(const void * buf, PINDEX len);
    virtual PBoolean Close();
    virtual PBoolean Shutdown(ShutdownValue) { return true; }
    virtual PString GetErrorText(ErrorGroup group = NumErrorGroups) const;
    virtual PBoolean ConvertOSError(P_INT_PTR error, ErrorGroup group = LastGeneralError);

    // New functions
    PBoolean Accept();

    PBoolean Accept(
      PChannel & channel  
    );

    PBoolean Accept(
      PChannel * channel,     
      PBoolean autoDelete = true  
    );


    PBoolean Connect();

    PBoolean Connect(
      PChannel & channel  
    );

    PBoolean Connect(
      PChannel * channel,     
      PBoolean autoDelete = true  
    );

    bool AddClientCA(
      const PSSLCertificate & certificate
    );
    bool AddClientCA(
      const PList<PSSLCertificate> & certificates
    );

    PBoolean UseCertificate(
      const PSSLCertificate & certificate
    );

    PBoolean UsePrivateKey(
      const PSSLPrivateKey & key
    );

    PString GetCipherList() const;

    typedef PSSLContext::VerifyMode VerifyMode;

    void SetVerifyMode(
      VerifyMode mode
    );

    bool GetPeerCertificate(
      PSSLCertificate & certificate,
      PString * error = NULL
    );

    PSSLContext * GetContext() const { return m_context; }

    virtual PBoolean RawSSLRead(void * buf, PINDEX & len);

  protected:
    void Construct(PSSLContext * ctx, PBoolean autoDel);

    virtual PBoolean OnOpen();

  protected:
    PSSLContext * m_context;
    bool          m_autoDeleteContext;
    ssl_st      * m_ssl;
};

#endif // PTLIB_PSSL_H


// End Of File ///////////////////////////////////////////////////////////////