Dnspy/dnSpy/Roslyn/dnSpy.Roslyn/Compiler/FrameworkDetector.cs
2021-09-20 18:20:01 +02:00

129 lines
4.0 KiB
C#

/*
Copyright (C) 2014-2019 de4dot@gmail.com
This file is part of dnSpy
dnSpy 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 3 of the License, or
(at your option) any later version.
dnSpy 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 dnSpy. If not, see <http://www.gnu.org/licenses/>.
*/
using System;
using System.Diagnostics;
using System.IO;
using dnlib.DotNet;
namespace dnSpy.Roslyn.Compiler {
enum FrameworkKind {
Unknown,
DotNetFramework2,
DotNetFramework4,
DotNet,
Unity2,
Unity4,
}
static class FrameworkDetector {
public static FrameworkKind GetFrameworkKind(ModuleDef module) {
var corlib = module.CorLibTypes.AssemblyRef;
var moduleLocation = module.Location;
if (File.Exists(moduleLocation) && File.Exists(Path.Combine(Path.GetDirectoryName(moduleLocation)!, "UnityEngine.dll"))) {
Debug.Assert(corlib.Version.Major == 2 || corlib.Version.Major == 4);
return corlib.Version.Major == 2 ? FrameworkKind.Unity2 : FrameworkKind.Unity4;
}
var info = TryGetTargetFrameworkAttribute(module.Assembly);
if (info.framework is not null) {
if (info.framework == ".NETFramework") {
if (info.version.StartsWith("2.") || info.version.StartsWith("3."))
return FrameworkKind.DotNetFramework2;
if (info.version.StartsWith("4."))
return FrameworkKind.DotNetFramework4;
Debug.Fail("Unknown .NET Framework version");
return FrameworkKind.Unknown;
}
if (info.framework == ".NETCoreApp")
return FrameworkKind.DotNet;
return FrameworkKind.Unknown;
}
switch (corlib.FullName) {
case "mscorlib, Version=2.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089":
return FrameworkKind.DotNetFramework2;
case "mscorlib, Version=4.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089":
return FrameworkKind.DotNetFramework4;
}
switch (corlib.Name) {
case "System.Runtime":
if (corlib.Version >= new Version(4, 1, 0, 0))
return FrameworkKind.DotNet;
break;
}
return FrameworkKind.Unknown;
}
static (string framework, string version, string? profile) TryGetTargetFrameworkAttribute(AssemblyDef asm) {
var ca = asm?.CustomAttributes.Find("System.Runtime.Versioning.TargetFrameworkAttribute");
if (ca is null)
return default;
if (ca.ConstructorArguments.Count != 1)
return default;
var arg = ca.ConstructorArguments[0];
if (arg.Type.GetElementType() != ElementType.String)
return default;
var s = arg.Value as UTF8String;
if (UTF8String.IsNullOrEmpty(s))
return default;
string attrString = s;
// See corclr/src/mscorlib/src/System/Runtime/Versioning/BinaryCompatibility.cs
var values = attrString.Split(new char[] { ',' });
if (values.Length < 2 || values.Length > 3)
return default;
var framework = values[0].Trim();
if (framework.Length == 0)
return default;
string? versionStr = null;
string? profile = null;
for (int i = 1; i < values.Length; i++) {
var kvp = values[i].Split('=');
if (kvp.Length != 2)
return default;
var key = kvp[0].Trim();
var value = kvp[1].Trim();
if (key.Equals("Version", StringComparison.OrdinalIgnoreCase)) {
if (value.StartsWith("v", StringComparison.OrdinalIgnoreCase))
value = value.Substring(1);
versionStr = value;
if (!Version.TryParse(value, out var version))
return default;
}
else if (key.Equals("Profile", StringComparison.OrdinalIgnoreCase)) {
if (!string.IsNullOrEmpty(value))
profile = value;
}
}
if (versionStr is null || versionStr.Length == 0)
return default;
return (framework, versionStr, profile);
}
}
}