Fixed InternalsVisibleTo in DefaultUnresolvedAssembly.

Code was taken from the CSharpAssembly.

1 files changed, 45 insertions, 3 deletions

diff --git a/ICSharpCode.NRefactory/TypeSystem/Implementation/DefaultUnresolvedAssembly.cs b/ICSharpCode.NRefactory/TypeSystem/Implementation/DefaultUnresolvedAssembly.cs
index d1cbb4f7..de9834cc 100644
--- a/ICSharpCode.NRefactory/TypeSystem/Implementation/DefaultUnresolvedAssembly.cs
+++ b/ICSharpCode.NRefactory/TypeSystem/Implementation/DefaultUnresolvedAssembly.cs
@@ -332,12 +332,54 @@ namespace ICSharpCode.NRefactory.TypeSystem.Implementation
 			public ICompilation Compilation {
 				get { return compilation; }
 			}
-			
+	
 			public bool InternalsVisibleTo(IAssembly assembly)
 			{
-				return assembly == this;
+				if (this == assembly)
+					return true;
+				foreach (string shortName in GetInternalsVisibleTo()) {
+					if (assembly.AssemblyName == shortName)
+						return true;
+				}
+				return false;
 			}
-			
+
+			volatile string[] internalsVisibleTo;
+
+			string[] GetInternalsVisibleTo()
+			{
+				var result = this.internalsVisibleTo;
+				if (result != null) {
+					return result;
+				} else {
+					using (var busyLock = BusyManager.Enter(this)) {
+						Debug.Assert(busyLock.Success);
+						if (!busyLock.Success) {
+							return new string[0];
+						}
+						internalsVisibleTo = (
+							from attr in this.AssemblyAttributes
+							where attr.AttributeType.Name == "InternalsVisibleToAttribute"
+							&& attr.AttributeType.Namespace == "System.Runtime.CompilerServices"
+							&& attr.PositionalArguments.Count == 1
+							select GetShortName(attr.PositionalArguments.Single().ConstantValue as string)
+						).ToArray();
+					}
+					return internalsVisibleTo;
+				}
+			}
+
+			static string GetShortName(string fullAssemblyName)
+			{
+				if (fullAssemblyName == null)
+					return null;
+				int pos = fullAssemblyName.IndexOf(',');
+				if (pos < 0)
+					return fullAssemblyName;
+				else
+					return fullAssemblyName.Substring(0, pos);
+			}
+
 			public ITypeDefinition GetTypeDefinition(TopLevelTypeName topLevelTypeName)
 			{
 				IUnresolvedTypeDefinition td;