DwmAESGCMOutBuffer.hh

Go to the documentation of this file.

//===========================================================================
// @(#) $DwmPath: dwm/libDwmAuth/tags/libDwmAuth-0.3.6/include/DwmAESGCMOutBuffer.hh 10837 $
// @(#) $Id: DwmAESGCMOutBuffer.hh 10837 2020-07-05 09:14:15Z dwm $
//===========================================================================
//  Copyright (c) Daniel W. McRobb 2020
//  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 _DWMAESGCMOUTBUFFER_HH_
#define _DWMAESGCMOUTBUFFER_HH_

#include <iostream>
#include <string>

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

namespace Dwm {

  namespace AESGCM {

    //------------------------------------------------------------------------
    //------------------------------------------------------------------------
    class OutBuffer
      : public std::streambuf
    {
    public:
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      OutBuffer(std::ostream & os, const std::string & key);

    protected:
      int_type overflow(int_type c) override;
      int sync() override;
      std::streamsize xsputn(const char *p, std::streamsize n) override;

    private:
      std::ostream                                              &_os;
      CryptoPP::byte                                             _key[16];
      CryptoPP::byte                                             _iv[16];
      CryptoPP::AutoSeededRandomPool                             _rng;
      CryptoPP::GCM<CryptoPP::AES>::Encryption                   _encryptor;
      std::unique_ptr<CryptoPP::AuthenticatedEncryptionFilter>   _aef;
      std::string                                                _msgbuf;
    };
    
  }  // namespace AESGCM

}  // namespace Dwm

#endif  // _DWMAESGCMOUTBUFFER_HH_