Fixed a couple of bugs in the RP sample's custom store.

1 files changed, 9 insertions, 2 deletions

diff --git a/samples/OpenIdRelyingPartyWebForms/Code/CustomStore.cs b/samples/OpenIdRelyingPartyWebForms/Code/CustomStore.cs
index 12008ca..3ab6292 100644
--- a/samples/OpenIdRelyingPartyWebForms/Code/CustomStore.cs
+++ b/samples/OpenIdRelyingPartyWebForms/Code/CustomStore.cs
@@ -1,8 +1,15 @@
-namespace OpenIdRelyingPartyWebForms.Code {
+//-----------------------------------------------------------------------
+// <copyright file="CustomStore.cs" company="Andrew Arnott">
+//     Copyright (c) Andrew Arnott. All rights reserved.
+// </copyright>
+//-----------------------------------------------------------------------
+
+namespace OpenIdRelyingPartyWebForms.Code {
 	using System;
 	using System.Collections.Generic;
 	using System.Data;
 	using System.Globalization;
+	using System.Linq;
 	using DotNetOpenAuth;
 	using DotNetOpenAuth.Configuration;
 	using DotNetOpenAuth.Messaging.Bindings;
@@ -94,7 +101,7 @@
 				yield break;
 			}
 
-			foreach (CustomStoreDataSet.CryptoKeyRow row in view) {
+			foreach (CustomStoreDataSet.CryptoKeyRow row in view.Cast<DataRowView>().Select(rv => rv.Row)) {
 				yield return new KeyValuePair<string, CryptoKey>(row.Handle, new CryptoKey(row.Secret, row.ExpiresUtc));
 			}
 		}