Initial commit: client source

source/EterBase/cipher.h

@@ -0,0 +1,62 @@
+#ifndef __CIPHER_H__
+#define __CIPHER_H__
+
+#ifdef _IMPROVED_PACKET_ENCRYPTION_
+
+#pragma warning(push)
+#pragma warning(disable: 4100 4127 4189 4231 4512 4706)
+#include <cryptopp/cryptlib.h>
+#pragma warning(pop)
+// Forward declaration
+class KeyAgreement;
+
+//THEMIDA
+// Communication channel encryption handler.
+class Cipher {
+ public:
+  explicit Cipher();
+  ~Cipher();
+
+  void CleanUp();
+
+  // Returns agreed value length in CryptoPP::bytes, or zero on failure.
+  size_t Prepare(void* buffer, size_t* length);
+  // Try to activate cipher algorithm with agreement data received from peer.
+  bool Activate(bool polarity, size_t agreed_length,
+                const void* buffer, size_t length);
+
+  // Encrypts the given block of data. (no padding required)
+  void Encrypt(void* buffer, size_t length) {
+    assert(activated_);
+    if (!activated_) {
+      return;
+    }
+    encoder_->ProcessData((CryptoPP::byte*)buffer, (const CryptoPP::byte*)buffer, length);
+  }
+  // Decrypts the given block of data. (no padding required)
+  void Decrypt(void* buffer, size_t length) {
+    assert(activated_);
+    if (!activated_) {
+      return;
+    }
+    decoder_->ProcessData((CryptoPP::byte*)buffer, (const CryptoPP::byte*)buffer, length);
+  }
+
+  bool activated() const { return activated_; }
+  
+  void set_activated(bool value) { activated_ = value; }
+
+ private:
+  bool SetUp(bool polarity);
+
+  bool activated_;
+
+  CryptoPP::SymmetricCipher* encoder_;
+  CryptoPP::SymmetricCipher* decoder_;
+
+  KeyAgreement* key_agreement_;
+};
+
+#endif // _IMPROVED_PACKET_ENCRYPTION_
+
+#endif // __CIPHER_H__