9872cbeb1312568ee825c3814bd3c59654e6c016
[mono.git] / mcs / tools / linker / Mono.Linker.Steps / ResolveFromAssemblyStep.cs
1 //
2 // ResolveFromAssemblyStep.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 using Mono.Cecil;
31
32 namespace Mono.Linker.Steps {
33
34         public class ResolveFromAssemblyStep : ResolveStep {
35
36                 AssemblyDefinition _assembly;
37                 string _file;
38
39                 public ResolveFromAssemblyStep (string assembly)
40                 {
41                         _file = assembly;
42                 }
43
44                 public ResolveFromAssemblyStep (AssemblyDefinition assembly)
45                 {
46                         _assembly = assembly;
47                 }
48
49                 protected override void Process ()
50                 {
51                         if (_assembly != null)
52                                 Context.Resolver.CacheAssembly (_assembly);
53
54                         AssemblyDefinition assembly = _assembly ?? Context.Resolve (_file);
55
56                         switch (assembly.MainModule.Kind) {
57                         case ModuleKind.Dll:
58                                 ProcessLibrary (Context, assembly);
59                                 return;
60                         default:
61                                 ProcessExecutable (assembly);
62                                 return;
63                         }
64                 }
65
66                 public static void ProcessLibrary (LinkContext context, AssemblyDefinition assembly)
67                 {
68                         context.Annotations.SetAction (assembly, AssemblyAction.Copy);
69
70                         foreach (TypeDefinition type in assembly.MainModule.Types)
71                                 MarkType (context, type);
72                 }
73
74                 static void MarkType (LinkContext context, TypeDefinition type)
75                 {
76                         context.Annotations.Mark (type);
77
78                         if (type.HasFields)
79                                 MarkFields (context, type.Fields);
80                         if (type.HasMethods)
81                                 MarkMethods (context, type.Methods);
82                         if (type.HasNestedTypes)
83                                 foreach (var nested in type.NestedTypes)
84                                         MarkType (context, nested);
85                 }
86
87                 void ProcessExecutable (AssemblyDefinition assembly)
88                 {
89                         Annotations.SetAction (assembly, AssemblyAction.Link);
90
91                         Annotations.Mark (assembly.EntryPoint.DeclaringType);
92                         MarkMethod (Context, assembly.EntryPoint, MethodAction.Parse);
93                 }
94
95                 static void MarkFields (LinkContext context, ICollection fields)
96                 {
97                         foreach (FieldDefinition field in fields)
98                                 context.Annotations.Mark (field);
99                 }
100
101                 static void MarkMethods (LinkContext context, ICollection methods)
102                 {
103                         foreach (MethodDefinition method in methods)
104                                 MarkMethod (context, method, MethodAction.ForceParse);
105                 }
106
107                 static void MarkMethod (LinkContext context, MethodDefinition method, MethodAction action)
108                 {
109                         context.Annotations.Mark (method);
110                         context.Annotations.SetAction (method, action);
111                 }
112         }
113 }