DwmAuthPKCrypto.hh

Go to the documentation of this file.

//===========================================================================
// @(#) $DwmPath: dwm/libDwmAuth/tags/libDwmAuth-0.3.6/include/DwmAuthPKCrypto.hh 11185 $
// @(#) $Id: DwmAuthPKCrypto.hh 11185 2020-09-19 03:32:25Z dwm $
//===========================================================================
//  Copyright (c) Daniel W. McRobb 2016
//  All rights reserved.
//
//  Redistribution and use in source and binary forms, with or without
//  modification, are permitted provided that the following conditions
//  are met:
//
//  1. Redistributions of source code must retain the above copyright
//     notice, this list of conditions and the following disclaimer.
//  2. Redistributions in binary form must reproduce the above copyright
//     notice, this list of conditions and the following disclaimer in the
//     documentation and/or other materials provided with the distribution.
//  3. The names of the authors and copyright holders may not be used to
//     endorse or promote products derived from this software without
//     specific prior written permission.
//
//  IN NO EVENT SHALL DANIEL W. MCROBB BE LIABLE TO ANY PARTY FOR
//  DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES,
//  INCLUDING LOST PROFITS, ARISING OUT OF THE USE OF THIS SOFTWARE,
//  EVEN IF DANIEL W. MCROBB HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH
//  DAMAGE.
//
//  THE SOFTWARE PROVIDED HEREIN IS ON AN "AS IS" BASIS, AND
//  DANIEL W. MCROBB HAS NO OBLIGATION TO PROVIDE MAINTENANCE, SUPPORT,
//  UPDATES, ENHANCEMENTS, OR MODIFICATIONS. DANIEL W. MCROBB MAKES NO
//  REPRESENTATIONS AND EXTENDS NO WARRANTIES OF ANY KIND, EITHER
//  IMPLIED OR EXPRESS, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
//  WARRANTIES OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE,
//  OR THAT THE USE OF THIS SOFTWARE WILL NOT INFRINGE ANY PATENT,
//  TRADEMARK OR OTHER RIGHTS.
//===========================================================================

//---------------------------------------------------------------------------
//---------------------------------------------------------------------------

#ifndef _DWMAUTHPKCRYPTO_HH_
#define _DWMAUTHPKCRYPTO_HH_

#include <string>
#include <utility>

#include <cryptopp/cryptlib.h>
#include <cryptopp/aes.h>
#include <cryptopp/filters.h>
#include <cryptopp/gcm.h>
#include <cryptopp/osrng.h>
#include <cryptopp/rsa.h>

#include "DwmDescriptorIOCapable.hh"
#include "DwmFileIOCapable.hh"
#include "DwmStreamIOCapable.hh"
#include "DwmStreamedLengthCapable.hh"

namespace Dwm {

  namespace Auth {

    //------------------------------------------------------------------------
    //------------------------------------------------------------------------
    class PKCrypto
    {
    public:
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      class PublicKey
        : public CryptoPP::RSA::PublicKey,
          public DescriptorIOCapable, public FileIOCapable,
          public StreamIOCapable, public StreamedLengthCapable
      {
      public:
        PublicKey()
            : CryptoPP::RSA::PublicKey()
        {}
        
        PublicKey(const CryptoPP::RSA::PublicKey & key)
            : CryptoPP::RSA::PublicKey(key)
        {}

        bool IsValid() const;
        
        std::istream & Read(std::istream & is) override;
        ssize_t Read(int fd) override;
        size_t Read(FILE * f) override;
        
        std::ostream & Write(std::ostream & os) const override;
        ssize_t Write(int fd) const override;
        size_t Write(FILE *f) const override;
        uint32_t StreamedLength() const override;
        
        std::string ToString() const;
        std::string ToBase64String() const;
        bool FromString(const std::string & s);
        bool FromBase64String(const std::string & s);
        
        //--------------------------------------------------------------------
        //--------------------------------------------------------------------
        bool operator == (const PublicKey & pk) const;
      };
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      class PrivateKey
        : public CryptoPP::RSA::PrivateKey,
          public DescriptorIOCapable, public FileIOCapable,
          public StreamIOCapable, public StreamedLengthCapable
      {
      public:
        PrivateKey()
            : CryptoPP::RSA::PrivateKey()
        {}
        
        PrivateKey(const CryptoPP::RSA::PrivateKey & key)
            : CryptoPP::RSA::PrivateKey(key)
        {}

        bool IsValid() const;

        std::istream & Read(std::istream & is) override;
        ssize_t Read(int fd) override;
        size_t Read(FILE * f) override;
        
        std::ostream & Write(std::ostream & os) const override;
        ssize_t Write(int fd) const override;
        size_t Write(FILE *f) const override;
        uint32_t StreamedLength() const override;
        
        std::string ToString() const;
        std::string ToBase64String() const;
        bool FromString(const std::string & s);
        bool FromBase64String(const std::string & s);

        //--------------------------------------------------------------------
        //--------------------------------------------------------------------
        bool operator == (const PrivateKey & pk) const;
      };
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      class KeyPair
        : public std::pair<PrivateKey,PublicKey>
      {
      public:
        //--------------------------------------------------------------------
        //--------------------------------------------------------------------
        KeyPair()
            : std::pair<PrivateKey,PublicKey>()
        {}
        
        //--------------------------------------------------------------------
        //--------------------------------------------------------------------
        KeyPair(const PrivateKey & privKey, const PublicKey & publicKey)
            : std::pair<PrivateKey,PublicKey>(privKey,publicKey)
        {}

        //--------------------------------------------------------------------
        //--------------------------------------------------------------------
        KeyPair & operator = (const KeyPair & keys)
        {
          if (this != &keys) {
            this->first = keys.first;
            this->second = keys.second;
          }
          return *this;
        }
        
        //--------------------------------------------------------------------
        //--------------------------------------------------------------------
        const PrivateKey & Private() const
        {
          return first;
        }
        
        //--------------------------------------------------------------------
        //--------------------------------------------------------------------
        PrivateKey & Private()
        {
          return first;
        }
        
        //--------------------------------------------------------------------
        //--------------------------------------------------------------------
        const PublicKey & Public() const
        {
          return second;
        }
        
        //--------------------------------------------------------------------
        //--------------------------------------------------------------------
        PublicKey & Public()
        {
          return second;
        }

        bool Load(const std::string & privKeyPath, std::string & name);
      };
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      static KeyPair CreateKeyPair(unsigned int keySize);
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      static std::string
      Encrypt(const PublicKey & publicKey, const std::string & plainstr);
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      static std::string
      Decrypt(const PrivateKey & privateKey, const std::string & cipherstr);

    };
    
  }  // namespace Auth

}  // namespace Dwm

#endif  // _DWMAUTHPKCRYPTO_HH_