1 files changed, 133 insertions, 0 deletions
diff --git a/extra/yassl/taocrypt/include/modes.hpp b/extra/yassl/taocrypt/include/modes.hpp
new file mode 100644
index 00000000000..2a21ad46b76
--- /dev/null
+++ b/extra/yassl/taocrypt/include/modes.hpp
@@ -0,0 +1,133 @@
+/* modes.hpp                                
+ *
+ * Copyright (C) 2003 Sawtooth Consulting Ltd.
+ *
+ * This file is part of yaSSL.
+ *
+ * yaSSL 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.
+ *
+ * yaSSL 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 this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
+ */
+
+/* modes.hpp provides ECB and CBC modes for block cipher encryption/decryption
+*/
+
+
+#ifndef TAO_CRYPT_MODES_HPP
+#define TAO_CRYPT_MODES_HPP
+
+#include <string.h>
+#include "misc.hpp"
+
+namespace TaoCrypt {
+
+
+enum Mode { ECB, CBC };
+
+
+// BlockCipher abstraction
+template<CipherDir DIR, class T, Mode MODE>
+class BlockCipher {
+public:
+    BlockCipher() : cipher_(DIR, MODE) {}
+
+    void Process(byte* c, const byte* p, word32 sz) 
+            { cipher_.Process(c, p, sz); }
+    void SetKey(const byte* k, word32 sz)   
+            { cipher_.SetKey(k, sz, DIR); }
+    void SetKey(const byte* k, word32 sz, const byte* iv)   
+            { cipher_.SetKey(k, sz, DIR); cipher_.SetIV(iv); }
+private:
+    T cipher_;
+
+    BlockCipher(const BlockCipher&);            // hide copy
+    BlockCipher& operator=(const BlockCipher&); // and assign
+};
+
+
+// Mode Base for block ciphers, static size
+template<int BLOCK_SIZE>
+class Mode_BASE {
+public:
+    Mode_BASE() {}
+    virtual ~Mode_BASE() {}
+
+    virtual void ProcessAndXorBlock(const byte*, const byte*, byte*) const = 0;
+
+    void ECB_Process(byte*, const byte*, word32);
+    void CBC_Encrypt(byte*, const byte*, word32);
+    void CBC_Decrypt(byte*, const byte*, word32);
+
+    void SetIV(const byte* iv) { memcpy(reg_, iv, BLOCK_SIZE); }
+private:
+    byte reg_[BLOCK_SIZE];
+    byte tmp_[BLOCK_SIZE];
+
+    Mode_BASE(const Mode_BASE&);            // hide copy
+    Mode_BASE& operator=(const Mode_BASE&); // and assign
+};
+
+
+// ECB Process blocks
+template<int BLOCK_SIZE>
+void Mode_BASE<BLOCK_SIZE>::ECB_Process(byte* out, const byte* in, word32 sz)
+{
+    word32 blocks = sz / BLOCK_SIZE;
+
+    while (blocks--) {
+        ProcessAndXorBlock(in, 0, out);
+        out += BLOCK_SIZE;
+        in  += BLOCK_SIZE;
+    }
+}
+
+
+// CBC Encrypt
+template<int BLOCK_SIZE>
+void Mode_BASE<BLOCK_SIZE>::CBC_Encrypt(byte* out, const byte* in, word32 sz)
+{
+    word32 blocks = sz / BLOCK_SIZE;
+
+    while (blocks--) {
+        xorbuf(reg_, in, BLOCK_SIZE);
+        ProcessAndXorBlock(reg_, 0, reg_);
+        memcpy(out, reg_, BLOCK_SIZE);
+        out += BLOCK_SIZE;
+        in  += BLOCK_SIZE;
+    }
+}
+
+
+// CBC Decrypt
+template<int BLOCK_SIZE>
+void Mode_BASE<BLOCK_SIZE>::CBC_Decrypt(byte* out, const byte* in, word32 sz)
+{
+    word32 blocks = sz / BLOCK_SIZE;
+    byte   hold[BLOCK_SIZE];
+
+    while (blocks--) {
+        memcpy(tmp_, in, BLOCK_SIZE);
+        ProcessAndXorBlock(tmp_, 0, out);
+        xorbuf(out,  reg_, BLOCK_SIZE);
+        memcpy(hold, reg_,   BLOCK_SIZE); // swap reg_ and tmp_
+        memcpy(reg_,   tmp_, BLOCK_SIZE);
+        memcpy(tmp_, hold, BLOCK_SIZE);
+        out += BLOCK_SIZE;
+        in  += BLOCK_SIZE;
+    }
+}
+
+
+} // namespace
+
+#endif  // TAO_CRYPT_MODES_HPP