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

   1 //
   2 // CleanStep.cs
   3 //
   4 // Author:
   5 //   Jb Evain (jbevain@gmail.com)
   6 //
   7 // (C) 2006 Jb Evain
   8 //
   9 // Permission is hereby granted, free of charge, to any person obtaining
  10 // a copy of this software and associated documentation files (the
  11 // "Software"), to deal in the Software without restriction, including
  12 // without limitation the rights to use, copy, modify, merge, publish,
  13 // distribute, sublicense, and/or sell copies of the Software, and to
  14 // permit persons to whom the Software is furnished to do so, subject to
  15 // the following conditions:
  16 //
  17 // The above copyright notice and this permission notice shall be
  18 // included in all copies or substantial portions of the Software.
  19 //
  20 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  21 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  22 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  23 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  24 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  25 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  26 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  27 //
  28
  29 using System.Collections;
  30
  31 using Mono.Cecil;
  32
  33 namespace Mono.Linker.Steps {
  34
  35         public class CleanStep : BaseStep {
  36
  37                 protected override void ProcessAssembly (AssemblyDefinition assembly)
  38                 {
  39                         if (Annotations.GetAction (assembly) == AssemblyAction.Link)
  40                                 CleanAssembly (assembly);
  41                 }
  42
  43                 static void CleanAssembly (AssemblyDefinition asm)
  44                 {
  45                         foreach (TypeDefinition type in asm.MainModule.Types)
  46                                 CleanType (type);
  47                 }
  48
  49                 static bool CheckType (ModuleDefinition module, TypeReference reference)
  50                 {
  51                         TypeSpecification spec = reference as TypeSpecification;
  52                         if (spec != null)
  53                                 return CheckType (module, spec.ElementType);
  54
  55                         TypeDefinition type = reference as TypeDefinition;
  56                         if (type == null)
  57                                 return true;
  58
  59                         return module.Types.Contains (type);
  60                 }
  61
  62                 static void CleanType (TypeDefinition type)
  63                 {
  64                         if (type.HasProperties)
  65                                 CleanProperties (type);
  66                         if (type.HasEvents)
  67                                 CleanEvents (type);
  68                 }
  69
  70                 static MethodDefinition CheckMethod (TypeDefinition type, MethodDefinition method)
  71                 {
  72                         if (method == null)
  73                                 return null;
  74
  75                         return type.Methods.Contains (method) ? method : null;
  76                 }
  77
  78                 static void CleanEvents (TypeDefinition type)
  79                 {
  80                         var events = type.Events;
  81
  82                         for (int i = 0; i < events.Count; i++) {
  83                                 var evt = events [i];
  84                                 evt.AddMethod = CheckMethod (type, evt.AddMethod);
  85                                 evt.InvokeMethod = CheckMethod (type, evt.InvokeMethod);
  86                                 evt.RemoveMethod = CheckMethod (type, evt.RemoveMethod);
  87
  88                                 if (!IsEventUsed (evt))
  89                                         events.RemoveAt (i--);
  90                         }
  91                 }
  92
  93                 static bool IsEventUsed (EventDefinition evt)
  94                 {
  95                         return evt.AddMethod != null || evt.InvokeMethod != null || evt.RemoveMethod != null;
  96                 }
  97
  98                 static void CleanProperties (TypeDefinition type)
  99                 {
 100                         var properties = type.Properties;
 101
 102                         for (int i = 0; i < properties.Count; i++) {
 103                                 var prop = properties [i];
 104                                 prop.GetMethod = CheckMethod (type, prop.GetMethod);
 105                                 prop.SetMethod = CheckMethod (type, prop.SetMethod);
 106
 107                                 if (!IsPropertyUsed (prop))
 108                                         properties.RemoveAt (i--);
 109                         }
 110                 }
 111
 112                 static bool IsPropertyUsed (PropertyDefinition prop)
 113                 {
 114                         return prop.GetMethod != null || prop.SetMethod != null;
 115                 }
 116         }
 117 }