1 files changed, 131 insertions, 0 deletions
diff --git a/indra/llmessage/tests/llxorcipher_test.cpp b/indra/llmessage/tests/llxorcipher_test.cpp
new file mode 100644
index 0000000000..9acac53604
--- /dev/null
+++ b/indra/llmessage/tests/llxorcipher_test.cpp
@@ -0,0 +1,131 @@
+/**
+ * @file llxorcipher_tut.cpp
+ * @author Adroit
+ * @date 2007-03
+ * @brief llxorcipher, llnullcipher test cases.
+ *
+ * $LicenseInfo:firstyear=2007&license=viewerlgpl$
+ * Second Life Viewer Source Code
+ * Copyright (C) 2010, Linden Research, Inc.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation;
+ * version 2.1 of the License only.
+ *
+ * This library 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
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+ *
+ * Linden Research, Inc., 945 Battery Street, San Francisco, CA  94111  USA
+ * $/LicenseInfo$
+ */
+
+#include "linden_common.h"
+#include "lltut.h"
+#include "llxorcipher.h"
+#include "llnullcipher.h"
+
+namespace tut
+{
+    struct cipher
+    {
+    };
+    typedef test_group<cipher> cipher_t;
+    typedef cipher_t::object cipher_object_t;
+    tut::cipher_t tut_cipher("cipher");
+
+    //encrypt->decrypt
+    template<> template<>
+    void cipher_object_t::test<1>()
+    {
+        const U32 len = 3;
+        const U8 pad[] = "abc";
+        const char str[] = "SecondLife";
+        const S32 str_len = sizeof(str);
+        U8 encrypted[str_len];
+        U8 decrypted[str_len];
+        LLXORCipher xorCipher(pad, len);
+        LLXORCipher xorCipher1(pad, len);
+
+        U32 length = xorCipher.requiredEncryptionSpace(50);
+        ensure("requiredEncryptionSpace() function failed", (length == 50));
+
+        U32 lenEncrypted = xorCipher.encrypt((U8 *) str, str_len, encrypted, str_len);
+        ensure("Encryption failed", (lenEncrypted == str_len));
+        U32 lenDecrypted = xorCipher1.decrypt(encrypted, str_len, decrypted, str_len);
+        ensure("Decryption failed", (lenDecrypted == str_len));
+        ensure_memory_matches("LLXORCipher Encrypt/Decrypt failed", str, str_len, decrypted, lenDecrypted);
+    }
+
+    // operator=
+    template<> template<>
+    void cipher_object_t::test<2>()
+    {
+        const U8 pad[] = "ABCDEFGHIJKLMNOPQ"; // pad len longer than data to be ciphered
+        const U32 pad_len = sizeof(pad);
+        const U8 pad1[] = "SecondLife";
+        const U32 pad_len1 = sizeof(pad1);
+        const char str[] = "To Be Ciphered";
+        const S32 str_len = sizeof(str);
+        U8 encrypted[str_len];
+        U8 decrypted[str_len];
+
+        LLXORCipher xorCipher(pad, pad_len);
+        LLXORCipher xorCipher1(pad1, pad_len1);
+
+        xorCipher.encrypt((U8 *) str, str_len, encrypted, str_len);
+        // make xorCipher1 same as xorCipher..so that xorCipher1 can decrypt what was
+        // encrypted using xorCipher
+        xorCipher1 = xorCipher;
+        U32 lenDecrypted = xorCipher1.decrypt(encrypted, str_len, decrypted, str_len);
+        ensure_memory_matches("LLXORCipher operator= failed", str, str_len, decrypted, lenDecrypted);
+    }
+
+    //in place encrypt->decrypt
+    template<> template<>
+    void cipher_object_t::test<3>()
+    {
+        U32 padNum = 0x12349087;
+        const U8* pad = (U8*) &padNum;
+        const U32 pad_len = sizeof(U32);
+        char str[] = "To Be Ciphered a long string.........!!!.";
+        char str1[] = "To Be Ciphered a long string.........!!!."; // same as str
+        const S32 str_len = sizeof(str);
+
+        LLXORCipher xorCipher(pad, pad_len);
+        LLXORCipher xorCipher1(pad, pad_len);
+        xorCipher.encrypt((U8 *) str, str_len);
+        // it should not be the same as original data!
+        ensure("LLXORCipher: In Place encrypt failed", memcmp(str, str1, str_len) != 0);
+        xorCipher1.decrypt((U8 *) str, str_len);
+        // it should not be the same as original data!
+        ensure_memory_matches("LLXORCipher: In Place decrypt failed", str, str_len, str1, str_len);
+    }
+
+    //LLNullCipher encrypt->decrypt
+    template<> template<>
+    void cipher_object_t::test<4>()
+    {
+        const char str[] = "SecondLife";
+        const S32 str_len = sizeof(str);
+        U8 encrypted[str_len];
+        U8 decrypted[str_len];
+        LLNullCipher nullCipher;
+        LLNullCipher nullCipher1;
+
+        U32 length = nullCipher.requiredEncryptionSpace(50);
+        ensure("LLNullCipher::requiredEncryptionSpace() function failed", (length == 50));
+
+        U32 len1 = nullCipher.encrypt((U8 *) str, str_len, encrypted, str_len);
+        ensure_memory_matches("LLNullCipher - Source transformed during encryption.", encrypted, len1, str, str_len);
+
+        U32 len2 = nullCipher1.decrypt(encrypted, str_len, decrypted, str_len);
+        ensure_memory_matches("LLNullCipher - Decryption failed", decrypted, len2, str, str_len);
+    }
+}