1 files changed, 63 insertions, 0 deletions
diff --git a/src/DotNetOpenAuth.Test/OpenId/DiffieHellmanTests.cs b/src/DotNetOpenAuth.Test/OpenId/DiffieHellmanTests.cs
new file mode 100644
index 0000000..a5ad752
--- /dev/null
+++ b/src/DotNetOpenAuth.Test/OpenId/DiffieHellmanTests.cs
@@ -0,0 +1,63 @@
+//-----------------------------------------------------------------------
+// <copyright file="DiffieHellmanTests.cs" company="Jason Alexander, Andrew Arnott">
+//     Copyright (c) Jason Alexander, Andrew Arnott. All rights reserved.
+// </copyright>
+//-----------------------------------------------------------------------
+
+namespace DotNetOpenAuth.Test.OpenId {
+	using System;
+	using System.IO;
+	using DotNetOpenAuth.OpenId;
+	using DotNetOpenAuth.OpenId.Messages;
+	using Microsoft.VisualStudio.TestTools.UnitTesting;
+	using Org.Mentalis.Security.Cryptography;
+
+	[TestClass]
+	public class DiffieHellmanTests {
+		[TestMethod]
+		public void Test() {
+			string s1 = Test1();
+			string s2 = Test1();
+
+			Assert.AreNotEqual(s1, s2, "Secret keys should NOT be the same.");
+		}
+
+		[TestMethod]
+		public void TestPublic() {
+			TextReader reader = new StringReader(TestSupport.LoadEmbeddedFile("dhpriv.txt"));
+
+			try {
+				string line;
+				while ((line = reader.ReadLine()) != null) {
+					string[] parts = line.Trim().Split(' ');
+					byte[] x = Convert.FromBase64String(parts[0]);
+					DiffieHellmanManaged dh = new DiffieHellmanManaged(AssociateDiffieHellmanRequest.DefaultMod, AssociateDiffieHellmanRequest.DefaultGen, x);
+					byte[] pub = dh.CreateKeyExchange();
+					byte[] y = Convert.FromBase64String(parts[1]);
+
+					if (y[0] == 0 && y[1] <= 127)
+						y.CopyTo(y, 1);
+
+					Assert.AreEqual(
+						Convert.ToBase64String(y),
+						Convert.ToBase64String(DiffieHellmanUtilities.EnsurePositive(pub)),
+						line);
+				}
+			} finally {
+				reader.Close();
+			}
+		}
+
+		private static string Test1() {
+			DiffieHellman dh1 = new DiffieHellmanManaged();
+			DiffieHellman dh2 = new DiffieHellmanManaged();
+
+			string secret1 = Convert.ToBase64String(dh1.DecryptKeyExchange(dh2.CreateKeyExchange()));
+			string secret2 = Convert.ToBase64String(dh2.DecryptKeyExchange(dh1.CreateKeyExchange()));
+
+			Assert.AreEqual(secret1, secret2, "Secret keys do not match for some reason.");
+
+			return secret1;
+		}
+	}
+}