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 void CleanType (TypeDefinition type)
  50                 {
  51                         if (type.HasProperties)
  52                                 CleanProperties (type);
  53                         if (type.HasEvents)
  54                                 CleanEvents (type);
  55
  56                         if (type.HasNestedTypes)
  57                                 foreach (var nested in type.NestedTypes)
  58                                         CleanType (nested);
  59                 }
  60
  61                 static MethodDefinition CheckMethod (TypeDefinition type, MethodDefinition method)
  62                 {
  63                         if (method == null)
  64                                 return null;
  65
  66                         return type.Methods.Contains (method) ? method : null;
  67                 }
  68
  69                 static void CleanEvents (TypeDefinition type)
  70                 {
  71                         var events = type.Events;
  72
  73                         for (int i = 0; i < events.Count; i++) {
  74                                 var evt = events [i];
  75                                 evt.AddMethod = CheckMethod (type, evt.AddMethod);
  76                                 evt.InvokeMethod = CheckMethod (type, evt.InvokeMethod);
  77                                 evt.RemoveMethod = CheckMethod (type, evt.RemoveMethod);
  78
  79                                 if (!IsEventUsed (evt))
  80                                         events.RemoveAt (i--);
  81                         }
  82                 }
  83
  84                 static bool IsEventUsed (EventDefinition evt)
  85                 {
  86                         return evt.AddMethod != null || evt.InvokeMethod != null || evt.RemoveMethod != null;
  87                 }
  88
  89                 static void CleanProperties (TypeDefinition type)
  90                 {
  91                         var properties = type.Properties;
  92
  93                         for (int i = 0; i < properties.Count; i++) {
  94                                 var prop = properties [i];
  95                                 prop.GetMethod = CheckMethod (type, prop.GetMethod);
  96                                 prop.SetMethod = CheckMethod (type, prop.SetMethod);
  97
  98                                 if (!IsPropertyUsed (prop))
  99                                         properties.RemoveAt (i--);
 100                         }
 101                 }
 102
 103                 static bool IsPropertyUsed (PropertyDefinition prop)
 104                 {
 105                         return prop.GetMethod != null || prop.SetMethod != null;
 106                 }
 107         }
 108 }