5 // Jb Evain (jbevain@gmail.com)
8 // (C) 2007 Novell, Inc.
10 // Permission is hereby granted, free of charge, to any person obtaining
11 // a copy of this software and associated documentation files (the
12 // "Software"), to deal in the Software without restriction, including
13 // without limitation the rights to use, copy, modify, merge, publish,
14 // distribute, sublicense, and/or sell copies of the Software, and to
15 // permit persons to whom the Software is furnished to do so, subject to
16 // the following conditions:
18 // The above copyright notice and this permission notice shall be
19 // included in all copies or substantial portions of the Software.
21 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
22 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
23 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
24 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
25 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
26 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
27 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
30 using System.Collections;
34 namespace Mono.Linker.Steps {
36 public class SweepStep : IStep {
38 AssemblyDefinition [] assemblies;
40 public void Process (LinkContext context)
42 assemblies = context.GetAssemblies ();
43 foreach (var assembly in assemblies)
44 SweepAssembly (assembly);
47 void SweepAssembly (AssemblyDefinition assembly)
49 if (Annotations.GetAction (assembly) != AssemblyAction.Link)
52 if (!IsMarkedAssembly (assembly)) {
53 RemoveAssembly (assembly);
57 var types = assembly.MainModule.Types;
58 var cloned_types = Clone (types);
62 foreach (TypeDefinition type in cloned_types) {
63 if (Annotations.IsMarked (type)) {
69 SweepReferences (assembly, type);
73 static bool IsMarkedAssembly (AssemblyDefinition assembly)
75 return Annotations.IsMarked (assembly.MainModule);
78 void RemoveAssembly (AssemblyDefinition assembly)
80 Annotations.SetAction (assembly, AssemblyAction.Delete);
82 SweepReferences (assembly);
85 void SweepReferences (AssemblyDefinition target)
87 foreach (var assembly in assemblies)
88 SweepReferences (assembly, target);
91 void SweepReferences (AssemblyDefinition assembly, AssemblyDefinition target)
93 var references = assembly.MainModule.AssemblyReferences;
94 for (int i = 0; i < references.Count; i++) {
95 var reference = references [i];
96 if (!AreSameReference (reference, target.Name))
99 references.RemoveAt (i);
104 static ICollection Clone (ICollection collection)
106 return new ArrayList (collection);
109 void SweepReferences (AssemblyDefinition assembly, TypeDefinition type)
111 foreach (AssemblyDefinition asm in assemblies) {
112 ModuleDefinition module = asm.MainModule;
113 if (!module.TypeReferences.Contains (type))
116 TypeReference typeRef = module.TypeReferences [type.FullName];
117 if (AssemblyMatch (assembly, typeRef)) {
118 SweepMemberReferences (module, typeRef);
119 module.TypeReferences.Remove (typeRef);
124 static void SweepMemberReferences (ModuleDefinition module, TypeReference reference)
126 var references = module.MemberReferences;
128 for (int i = 0; i < references.Count; i++) {
129 if (references [i].DeclaringType == reference)
130 references.RemoveAt (i--);
134 static bool AssemblyMatch (AssemblyDefinition assembly, TypeReference type)
136 AssemblyNameReference reference = type.Scope as AssemblyNameReference;
137 if (reference == null)
140 return AreSameReference (assembly.Name, reference);
143 static void SweepType (TypeDefinition type)
146 SweepCollection (type.Fields);
148 if (type.HasConstructors)
149 SweepCollection (type.Constructors);
152 SweepCollection (type.Methods);
155 static void SweepCollection (IList list)
157 for (int i = 0; i < list.Count; i++)
158 if (!Annotations.IsMarked ((IAnnotationProvider) list [i]))
162 static bool AreSameReference (AssemblyNameReference a, AssemblyNameReference b)
167 if (a.Name != b.Name)
170 if (a.Version != b.Version)