cipher.h

Go to the documentation of this file.

// ================================================================
// Description: Cipher class.
// Copyright:   Copyright (c) 2012 by Joe Linoff
// Version:     1.2
// Author:      Joe Linoff
//
// LICENSE
//   The cipher package is free software; you can redistribute it and/or
//   modify it under the terms of the GNU General Public License as
//   published by the Free Software Foundation; either version 2 of the
//   License, or (at your option) any later version.
//       
//   The cipher package 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
//   General Public License for more details. You should have received
//   a copy of the GNU General Public License along with the change
//   tool; if not, write to the Free Software Foundation, Inc., 59
//   Temple Place, Suite 330, Boston, MA 02111-1307 USA.
// ================================================================
#ifndef cipher_h
#define cipher_h

#include <string>
#include <vector>
#include <utility> // pair

#define CIPHER_DEFAULT_CIPHER "aes-256-cbc"
#define CIPHER_DEFAULT_DIGEST "sha1"
#define CIPHER_DEFAULT_COUNT  1

class Cipher
{
public:
  typedef unsigned int uint;
  typedef unsigned char uchar;
  typedef uchar aes_key_t[32];
  typedef uchar aes_iv_t[32];
  typedef uchar aes_salt_t[8];
  typedef std::pair<uchar*,uint> kv1_t;
public:
  Cipher();
  
  Cipher(const std::string& cipher,
    const std::string& digest,
    uint count,
    bool embed=true);
  
  ~Cipher();
public:
  std::string encrypt(const std::string& plaintext,
            const std::string& pass="",
            const std::string& salt="");
  
  void encrypt_file(const std::string& ifn,
          const std::string& ofn,
          const std::string& pass="",
          const std::string& salt="");
public:
  std::string decrypt(const std::string& ciphertext,
            const std::string& pass="",
            const std::string& salt="");
  
  void decrypt_file(const std::string& ifn,
          const std::string& ofn,
          const std::string& pass="",
          const std::string& salt="");
public:
  std::string encode_base64(uchar* ciphertext,
             uint   ciphertext_len) const;
  
  kv1_t encode_cipher(const std::string& plaintext) const;
  
  kv1_t decode_base64(const std::string& mimetext) const;
  
  std::string decode_cipher(uchar* ciphertext,
             uint   ciphertext_len) const;
public:
  std::string file_read(const std::string& fn) const;
  void file_write(const std::string& fn,
        const std::string& data,
        bool nl=false) const;
public:
  void debug(bool b=true) {m_debug=b;}
  bool debug() const {return m_debug;}
private:
  void set_salt(const std::string& salt);
  void init(const std::string& pass);
  
private:
  std::string m_pass;
  std::string m_cipher;
  std::string m_digest;
  aes_salt_t  m_salt;
  aes_key_t   m_key;
  aes_iv_t    m_iv;
  uint        m_count;
  bool        m_embed;
  bool        m_debug;
};

#endif