1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Diagnostics.CodeAnalysis;
using System.Linq;
using Mono.Cecil;
namespace Mono.Linker
{
readonly struct LinkerAttributesInformation
{
readonly Dictionary<Type, List<Attribute>>? _linkerAttributes;
private LinkerAttributesInformation (Dictionary<Type, List<Attribute>>? cache)
{
this._linkerAttributes = cache;
}
public static LinkerAttributesInformation Create (LinkContext context, ICustomAttributeProvider provider)
{
Debug.Assert (context.CustomAttributes.HasAny (provider));
Dictionary<Type, List<Attribute>>? cache = null;
foreach (var customAttribute in context.CustomAttributes.GetCustomAttributes (provider)) {
var attributeType = customAttribute.AttributeType;
Attribute? attributeValue;
switch (attributeType.Name) {
case "RequiresUnreferencedCodeAttribute" when attributeType.Namespace == "System.Diagnostics.CodeAnalysis":
attributeValue = ProcessRequiresUnreferencedCodeAttribute (context, provider, customAttribute);
break;
case "DynamicDependencyAttribute" when attributeType.Namespace == "System.Diagnostics.CodeAnalysis":
attributeValue = DynamicDependency.ProcessAttribute (context, provider, customAttribute);
break;
case "RemoveAttributeInstancesAttribute":
if (provider is not TypeDefinition td)
continue;
// The attribute is never removed if it's explicitly preserved (e.g. via xml descriptor)
if (context.Annotations.TryGetPreserve (td, out TypePreserve preserve) && preserve != TypePreserve.Nothing)
continue;
attributeValue = BuildRemoveAttributeInstancesAttribute (context, td, customAttribute);
break;
default:
continue;
}
if (attributeValue == null)
continue;
if (cache == null)
cache = new Dictionary<Type, List<Attribute>> ();
Type attributeValueType = attributeValue.GetType ();
if (!cache.TryGetValue (attributeValueType, out var attributeList)) {
attributeList = new List<Attribute> ();
cache.Add (attributeValueType, attributeList);
}
attributeList.Add (attributeValue);
}
return new LinkerAttributesInformation (cache);
}
public bool HasAttribute<T> () where T : Attribute
{
return _linkerAttributes != null && _linkerAttributes.ContainsKey (typeof (T));
}
public IEnumerable<T> GetAttributes<T> () where T : Attribute
{
if (_linkerAttributes == null || !_linkerAttributes.TryGetValue (typeof (T), out var attributeList))
return Enumerable.Empty<T> ();
if (attributeList == null || attributeList.Count == 0)
throw new InvalidOperationException ("Unexpected list of attributes.");
return attributeList.Cast<T> ();
}
static Attribute? ProcessRequiresUnreferencedCodeAttribute (LinkContext context, ICustomAttributeProvider provider, CustomAttribute customAttribute)
{
if (!(provider is MethodDefinition || provider is TypeDefinition))
return null;
if (customAttribute.HasConstructorArguments && customAttribute.ConstructorArguments[0].Value is string message) {
var ruca = new RequiresUnreferencedCodeAttribute (message);
if (customAttribute.HasProperties) {
foreach (var prop in customAttribute.Properties) {
if (prop.Name == "Url") {
ruca.Url = prop.Argument.Value as string;
break;
}
}
}
return ruca;
}
context.LogWarning (
$"Attribute '{typeof (RequiresUnreferencedCodeAttribute).FullName}' doesn't have the required number of parameters specified.",
2028, (IMemberDefinition) provider);
return null;
}
static RemoveAttributeInstancesAttribute? BuildRemoveAttributeInstancesAttribute (LinkContext context, TypeDefinition attributeContext, CustomAttribute ca)
{
switch (ca.ConstructorArguments.Count) {
case 0:
return new RemoveAttributeInstancesAttribute ();
case 1:
// Argument is always boxed
return new RemoveAttributeInstancesAttribute ((CustomAttributeArgument) ca.ConstructorArguments[0].Value);
default:
context.LogWarning (
$"Attribute '{ca.AttributeType.GetDisplayName ()}' doesn't have the required number of arguments specified.",
2028, attributeContext);
return null;
};
}
}
}
|
