mcs/tools/linker/Mono.Linker.Steps/SweepStep.cs

   1 //
   2 // SweepStep.cs
   3 //
   4 // Author:
   5 //   Jb Evain (jbevain@gmail.com)
   6 //
   7 // (C) 2006 Jb Evain
   8 // (C) 2007 Novell, Inc.
   9 //
  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:
  17 //
  18 // The above copyright notice and this permission notice shall be
  19 // included in all copies or substantial portions of the Software.
  20 //
  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.
  28 //
  29
  30 using System.Collections;
  31 using System.Collections.Generic;
  32 using Mono.Cecil;
  33
  34 namespace Mono.Linker.Steps {
  35
  36         public class SweepStep : BaseStep {
  37
  38                 AssemblyDefinition [] assemblies;
  39
  40                 protected override void Process ()
  41                 {
  42                         assemblies = Context.GetAssemblies ();
  43                         foreach (var assembly in assemblies)
  44                                 SweepAssembly (assembly);
  45                 }
  46
  47                 void SweepAssembly (AssemblyDefinition assembly)
  48                 {
  49                         if (Annotations.GetAction (assembly) != AssemblyAction.Link)
  50                                 return;
  51
  52                         if (!IsMarkedAssembly (assembly)) {
  53                                 RemoveAssembly (assembly);
  54                                 return;
  55                         }
  56
  57                         var types = assembly.MainModule.Types;
  58                         var cloned_types = new List<TypeDefinition> (types);
  59
  60                         types.Clear ();
  61
  62                         foreach (TypeDefinition type in cloned_types) {
  63                                 if (Annotations.IsMarked (type)) {
  64                                         SweepType (type);
  65                                         types.Add (type);
  66                                         continue;
  67                                 }
  68                         }
  69                 }
  70
  71                 bool IsMarkedAssembly (AssemblyDefinition assembly)
  72                 {
  73                         return Annotations.IsMarked (assembly.MainModule);
  74                 }
  75
  76                 void RemoveAssembly (AssemblyDefinition assembly)
  77                 {
  78                         Annotations.SetAction (assembly, AssemblyAction.Delete);
  79
  80                         SweepReferences (assembly);
  81                 }
  82
  83                 void SweepReferences (AssemblyDefinition target)
  84                 {
  85                         foreach (var assembly in assemblies)
  86                                 SweepReferences (assembly, target);
  87                 }
  88
  89                 void SweepReferences (AssemblyDefinition assembly, AssemblyDefinition target)
  90                 {
  91                         var references = assembly.MainModule.AssemblyReferences;
  92                         for (int i = 0; i < references.Count; i++) {
  93                                 var reference = references [i];
  94                                 if (!AreSameReference (reference, target.Name))
  95                                         continue;
  96
  97                                 references.RemoveAt (i);
  98                                 return;
  99                         }
 100                 }
 101
 102                 static ICollection Clone (ICollection collection)
 103                 {
 104                         return new ArrayList (collection);
 105                 }
 106
 107                 void SweepType (TypeDefinition type)
 108                 {
 109                         if (type.HasFields)
 110                                 SweepCollection (type.Fields);
 111
 112                         if (type.HasMethods)
 113                                 SweepCollection (type.Methods);
 114
 115                         if (type.HasNestedTypes)
 116                                 SweepNestedTypes (type);
 117                 }
 118
 119                 void SweepNestedTypes (TypeDefinition type)
 120                 {
 121                         for (int i = 0; i < type.NestedTypes.Count; i++) {
 122                                 var nested = type.NestedTypes [i];
 123                                 if (Annotations.IsMarked (nested)) {
 124                                         SweepType (nested);
 125                                 } else {
 126                                         //type.NestedTypes.RemoveAt (i--);
 127                                 }
 128                         }
 129                 }
 130
 131                 void SweepCollection (IList list)
 132                 {
 133                         for (int i = 0; i < list.Count; i++)
 134                                 if (!Annotations.IsMarked ((IMetadataTokenProvider) list [i]))
 135                                         list.RemoveAt (i--);
 136                 }
 137
 138                 static bool AreSameReference (AssemblyNameReference a, AssemblyNameReference b)
 139                 {
 140                         if (a == b)
 141                                 return true;
 142
 143                         if (a.Name != b.Name)
 144                                 return false;
 145
 146                         if (a.Version != b.Version)
 147                                 return false;
 148
 149                         return true;
 150                 }
 151         }
 152 }