tools/tuner/Mono.Tuner/AdjustVisibility.cs

   1 //
   2 // AdjustVisibilityStep.cs
   3 //
   4 // Author:
   5 //   Jb Evain (jbevain@novell.com)
   6 //
   7 // (C) 2007 Novell, Inc.
   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;
  30 using System.Collections;
  31
  32 using Mono.Linker;
  33 using Mono.Linker.Steps;
  34
  35 using Mono.Cecil;
  36
  37 namespace Mono.Tuner {
  38
  39         public class AdjustVisibility : BaseStep {
  40
  41                 protected override void ProcessAssembly (AssemblyDefinition assembly)
  42                 {
  43                         if (Annotations.GetAction (assembly) != AssemblyAction.Link)
  44                                 return;
  45
  46                         ProcessTypes (assembly.MainModule.Types);
  47                 }
  48
  49                 static void ProcessTypes (ICollection types)
  50                 {
  51                         foreach (TypeDefinition type in types)
  52                                 ProcessType (type);
  53                 }
  54
  55                 static void ProcessType (TypeDefinition type)
  56                 {
  57                         if (!IsPublic (type))
  58                                 return;
  59
  60                         if (!IsMarkedAsPublic (type)) {
  61                                 SetInternalVisibility (type);
  62                                 return;
  63                         }
  64
  65                         if (type.IsEnum)
  66                                 return;
  67
  68                         ProcessFields (type.Fields);
  69                         ProcessMethods (type.Constructors);
  70                         ProcessMethods (type.Methods);
  71                 }
  72
  73                 static bool IsPublic (TypeDefinition type)
  74                 {
  75                         return type.DeclaringType == null ? type.IsPublic : type.IsNestedPublic;
  76                 }
  77
  78                 static void SetInternalVisibility (TypeDefinition type)
  79                 {
  80                         type.Attributes &= ~TypeAttributes.VisibilityMask;
  81                         if (type.DeclaringType == null)
  82                                 type.Attributes |= TypeAttributes.NotPublic;
  83                         else
  84                                 type.Attributes |= TypeAttributes.NestedAssembly;
  85
  86                         TunerAnnotations.Internalized (type);
  87                 }
  88
  89                 static void ProcessMethods (ICollection methods)
  90                 {
  91                         foreach (MethodDefinition method in methods)
  92                                 ProcessMethod (method);
  93                 }
  94
  95                 static void ProcessMethod (MethodDefinition method)
  96                 {
  97                         if (IsMarkedAsPublic (method))
  98                                 return;
  99
 100                         if (method.IsPublic)
 101                                 SetInternalVisibility (method);
 102                         else if (method.IsFamily || method.IsFamilyOrAssembly)
 103                                 SetProtectedAndInternalVisibility (method);
 104                 }
 105
 106                 static void SetInternalVisibility (MethodDefinition method)
 107                 {
 108                         method.Attributes &= ~MethodAttributes.MemberAccessMask;
 109                         method.Attributes |= MethodAttributes.Assem;
 110
 111                         TunerAnnotations.Internalized (method);
 112                 }
 113
 114                 static void SetProtectedAndInternalVisibility (MethodDefinition method)
 115                 {
 116                         method.Attributes &= ~MethodAttributes.MemberAccessMask;
 117                         method.Attributes |= MethodAttributes.FamANDAssem;
 118
 119                         TunerAnnotations.Internalized (method);
 120                 }
 121
 122                 static bool IsMarkedAsPublic (IAnnotationProvider provider)
 123                 {
 124                         return Annotations.IsPublic (provider);
 125                 }
 126
 127                 static void ProcessFields (FieldDefinitionCollection fields)
 128                 {
 129                         foreach (FieldDefinition field in fields)
 130                                 ProcessField (field);
 131                 }
 132
 133                 static void ProcessField (FieldDefinition field)
 134                 {
 135                         if (IsMarkedAsPublic (field))
 136                                 return;
 137
 138                         if (field.IsPublic)
 139                                 SetInternalVisibility (field);
 140                         else if (field.IsFamily || field.IsFamilyOrAssembly)
 141                                 SetProtectedAndInternalVisibility (field);
 142                 }
 143
 144                 static void SetInternalVisibility (FieldDefinition field)
 145                 {
 146                         field.Attributes &= ~FieldAttributes.FieldAccessMask;
 147                         field.Attributes |= FieldAttributes.Assembly;
 148
 149                         TunerAnnotations.Internalized (field);
 150                 }
 151
 152                 static void SetProtectedAndInternalVisibility (FieldDefinition field)
 153                 {
 154                         field.Attributes &= ~FieldAttributes.FieldAccessMask;
 155                         field.Attributes |= FieldAttributes.FamANDAssem;
 156
 157                         TunerAnnotations.Internalized (field);
 158                 }
 159         }
 160 }