?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
2010-04-01 Jb Evain <jbevain@novell.com>
[mcs.git] / ilasm / codegen / CodeGen.cs
blobb0b1803c18060a442523c60a46fda4794cb477ae
1 //
2 // Mono.ILASM.CodeGen.cs
3 //
4 // Author(s):
5 // Sergey Chaban (serge@wildwestsoftware.com)
6 // Jackson Harper (Jackson@LatitudeGeo.com)
7 //
8 // (C) Sergey Chaban
9 // (C) 2003 Jackson Harper, All rights reserved
10 //
11
12 using PEAPI;
13 using System;
14 using System.IO;
15 using System.Collections;
16 using System.Reflection;
17 using System.Reflection.Emit;
18 using System.Text;
19 using System.Security;
20
21 using SSPermissionSet = System.Security.PermissionSet;
22 using MIPermissionSet = Mono.ILASM.PermissionSet;
23
24 using MIAssembly = Mono.ILASM.Assembly;
25
26 namespace Mono.ILASM {
27
28 public class CodeGen {
29
30 private PEFile pefile;
31 private ExternAssembly current_assemblyref;
32 private ExternModule current_moduleref;
33 private string current_namespace;
34 private TypeDef current_typedef;
35 private MethodDef current_methoddef;
36 private ArrayList typedef_stack;
37 private int typedef_stack_top;
38 private SymbolWriter symwriter;
39 private ICustomAttrTarget current_customattrtarget;
40 private IDeclSecurityTarget current_declsectarget;
41 private PEAPI.NativeType current_field_native_type;
42
43 private MIAssembly this_assembly;
44
45 private TypeManager type_manager;
46 private ExternTable extern_table;
47 private Hashtable global_field_table;
48 private Hashtable global_method_table;
49 private Hashtable global_methodref_table;
50 private Hashtable global_fieldref_table;
51 private Hashtable data_table;
52 private FileRef file_ref;
53 private ArrayList manifestResources;
54 private Hashtable typeref_table;
55
56 private ArrayList defcont_list;
57
58 private int sub_system;
59 private int cor_flags;
60 private long image_base;
61 private long stack_reserve;
62
63 private string output_file;
64 private string debug_file;
65 private bool is_dll;
66 private bool entry_point;
67
68 private Module this_module;
69
70 public CodeGen (string output_file, bool is_dll, bool debugging_info)
71 {
72 this.output_file = output_file;
73 this.is_dll = is_dll;
74
75 if (debugging_info)
76 symwriter = new SymbolWriter (output_file);
77
78 type_manager = new TypeManager (this);
79 extern_table = new ExternTable ();
80 typedef_stack = new ArrayList ();
81 typedef_stack_top = 0;
82 global_field_table = new Hashtable ();
83 global_method_table = new Hashtable ();
84
85 data_table = new Hashtable ();
86
87 defcont_list = new ArrayList ();
88
89 sub_system = -1;
90 cor_flags = -1;
91 image_base = -1;
92 stack_reserve = -1;
93 entry_point = false;
94 this_module = null;
95 }
96
97 public PEFile PEFile {
98 get { return pefile; }
99 }
100
101 public SymbolWriter SymbolWriter {
102 get { return symwriter; }
103 }
104
105 public string CurrentNameSpace {
106 get { return current_namespace; }
107 set { current_namespace = value; }
108 }
109
110 public TypeDef CurrentTypeDef {
111 get { return current_typedef; }
112 }
113
114 public MethodDef CurrentMethodDef {
115 get { return current_methoddef; }
116 }
117
118 public ExternAssembly CurrentAssemblyRef {
119 get { return current_assemblyref; }
120 }
121
122 public ExternModule CurrentModuleRef {
123 get { return current_moduleref; }
124 }
125
126 public ICustomAttrTarget CurrentCustomAttrTarget {
127 get { return current_customattrtarget; }
128 set { current_customattrtarget = value; }
129 }
130
131 public IDeclSecurityTarget CurrentDeclSecurityTarget {
132 get { return current_declsectarget; }
133 set { current_declsectarget = value; }
134 }
135
136 public ExternTable ExternTable {
137 get { return extern_table; }
138 }
139
140 public TypeManager TypeManager {
141 get { return type_manager; }
142 }
143
144 public bool HasEntryPoint {
145 get { return entry_point; }
146 set {
147 /* if (!value) error: unsetting entrypoint ? */
148 if (entry_point)
149 Report.Error ("Multiple .entrypoint declarations.");
150 entry_point = value;
151 }
152 }
153
154 public TypeRef GetTypeRef (string name)
155 {
156 TypeRef tr = null;
157
158 if (typeref_table == null)
159 typeref_table = new Hashtable ();
160 else
161 tr = typeref_table [name] as TypeRef;
162
163 if (tr == null) {
164 tr = new TypeRef (name, false, null);
165 typeref_table [name] = tr;
166 }
167
168 return tr;
169 }
170
171 public GlobalMethodRef GetGlobalMethodRef (BaseTypeRef ret_type, PEAPI.CallConv call_conv,
172 string name, BaseTypeRef[] param, int gen_param_count)
173 {
174 string key = MethodDef.CreateSignature (ret_type, call_conv, name, param, gen_param_count, true);
175
176 GlobalMethodRef methref = null;
177
178 if (global_methodref_table == null)
179 global_methodref_table = new Hashtable ();
180 else
181 methref = (GlobalMethodRef) global_methodref_table [key];
182
183 if (methref == null) {
184 methref = new GlobalMethodRef (ret_type, call_conv, name, param, gen_param_count);
185 global_methodref_table [key] = methref;
186 }
187
188 return methref;
189 }
190
191 public GlobalFieldRef GetGlobalFieldRef (BaseTypeRef ret_type, string name)
192 {
193 string key = ret_type.FullName + name;
194
195 GlobalFieldRef fieldref = null;
196
197 if (global_fieldref_table == null)
198 global_fieldref_table = new Hashtable ();
199 else
200 fieldref = (GlobalFieldRef) global_fieldref_table [key];
201
202 if (fieldref == null) {
203 fieldref = new GlobalFieldRef (ret_type, name);
204 global_fieldref_table [key] = fieldref;
205 }
206
207 return fieldref;
208 }
209
210 public void SetSubSystem (int sub_system)
211 {
212 this.sub_system = sub_system;
213 }
214
215 public void SetCorFlags (int cor_flags)
216 {
217 this.cor_flags = cor_flags;
218 }
219
220 public void SetImageBase (long image_base)
221 {
222 this.image_base = image_base;
223 }
224
225 public void SetStackReserve (long stack_reserve)
226 {
227 this.stack_reserve = stack_reserve;
228 }
229
230 public void SetThisAssembly (string name, PEAPI.AssemAttr attr)
231 {
232 if (this_assembly != null && this_assembly.Name != name)
233 Report.Error ("Multiple assembly declarations");
234
235 this_assembly = new Assembly (name);
236 this_assembly.SetAssemblyAttr (attr);
237 if (name != "mscorlib")
238 ExternTable.AddCorlib ();
239 }
240
241 public void SetModuleName (string module_name)
242 {
243 this_module = new Module (module_name);
244 CurrentCustomAttrTarget = this_module;
245 }
246
247 public void SetFileRef (FileRef file_ref)
248 {
249 this.file_ref = file_ref;
250 }
251
252 public MIAssembly ThisAssembly {
253 get { return this_assembly; }
254 }
255
256 public bool IsThisAssembly (string name)
257 {
258 return (this_assembly != null && name == this_assembly.Name);
259 }
260
261 public Module ThisModule {
262 get { return this_module; }
263 }
264
265 public bool IsThisModule (string name)
266 {
267 return (this_module != null && name == this_module.Name);
268 }
269
270 public void BeginSourceFile (string name)
271 {
272 if (symwriter != null)
273 symwriter.BeginSourceFile (name);
274 }
275
276 public void EndSourceFile ()
277 {
278 if (symwriter != null)
279 symwriter.EndSourceFile ();
280 }
281
282 public void BeginTypeDef (TypeAttr attr, string name, BaseClassRef parent,
283 ArrayList impl_list, Location location, GenericParameters gen_params)
284 {
285 TypeDef outer = null;
286 string cache_name = CacheName (name);
287 if (typedef_stack_top > 0) {
288 StringBuilder sb = new StringBuilder ();
289
290 for (int i = 0; i < typedef_stack_top; i++){
291 outer = (TypeDef) typedef_stack [i];
292 if (i == 0)
293 /* Use FullName for outermost class to get the
294 namespace also */
295 sb.Append (outer.FullName);
296 else
297 sb.Append (outer.Name);
298 sb.Append ("/");
299 }
300 sb.Append (name);
301 cache_name = sb.ToString ();
302 }
303
304 TypeDef typedef = type_manager[cache_name];
305
306 if (typedef != null) {
307 // Class head is allready defined, we are just reopening the class
308 current_customattrtarget = current_typedef = typedef;
309 current_declsectarget = typedef;
310 typedef_stack.Add (current_typedef);
311 typedef_stack_top++;
312 return;
313 }
314
315 typedef = new TypeDef (attr, current_namespace,
316 name, parent, impl_list, location, gen_params, outer);
317
318 type_manager[cache_name] = typedef;
319 current_customattrtarget = current_typedef = typedef;
320 current_declsectarget = typedef;
321 typedef_stack.Add (typedef);
322 typedef_stack_top++;
323 }
324
325 public void AddFieldMarshalInfo (PEAPI.NativeType native_type)
326 {
327 current_field_native_type = native_type;
328 }
329
330 public void AddFieldDef (FieldDef fielddef)
331 {
332 if (current_field_native_type != null) {
333 fielddef.AddMarshalInfo (current_field_native_type);
334 current_field_native_type = null;
335 }
336
337 if (current_typedef != null) {
338 current_typedef.AddFieldDef (fielddef);
339 } else {
340 global_field_table.Add (
341 new DictionaryEntry (fielddef.Name, fielddef.Type.FullName),
342 fielddef);
343 }
344 }
345
346 public void AddDataDef (DataDef datadef)
347 {
348 if (data_table [datadef.Name] != null)
349 Report.Error ("Duplicate global label '" + datadef.Name + "'");
350 data_table [datadef.Name] = datadef;
351 }
352
353 public void AddManifestResource (ManifestResource mr)
354 {
355 if (manifestResources == null)
356 manifestResources = new ArrayList ();
357 manifestResources.Add (mr);
358 }
359
360 public PEAPI.DataConstant GetDataConst (string name)
361 {
362 DataDef def = (DataDef) data_table [name];
363 if (def == null)
364 return null;
365
366 return (DataConstant) def.PeapiConstant;
367 }
368
369 public void BeginMethodDef (MethodDef methoddef)
370 {
371 if (current_typedef != null) {
372 current_typedef.AddMethodDef (methoddef);
373 } else {
374 global_method_table.Add (methoddef.Signature,
375 methoddef);
376 }
377
378 current_customattrtarget = current_methoddef = methoddef;
379 current_declsectarget = methoddef;
380 }
381
382 public void EndMethodDef (Location location)
383 {
384 if (symwriter != null)
385 symwriter.EndMethod (location);
386
387 current_methoddef = null;
388 }
389
390 public void EndTypeDef ()
391 {
392 typedef_stack_top--;
393 typedef_stack.RemoveAt (typedef_stack_top);
394
395 if (typedef_stack_top > 0)
396 current_typedef = (TypeDef) typedef_stack [typedef_stack_top-1];
397 else
398 current_typedef = null;
399
400 }
401
402 public void BeginAssemblyRef (string name, AssemblyName asmb_name, PEAPI.AssemAttr attr)
403 {
404 current_customattrtarget = current_assemblyref = ExternTable.AddAssembly (name, asmb_name, attr);
405 current_declsectarget = current_assemblyref;
406 }
407
408 public void EndAssemblyRef ()
409 {
410 current_assemblyref = null;
411 current_customattrtarget = null;
412 current_declsectarget = null;
413 }
414
415 public void AddToDefineContentsList (TypeDef typedef)
416 {
417 defcont_list.Add (typedef);
418 }
419
420 public void AddPermission (PEAPI.SecurityAction sec_action, object perm)
421 {
422 if (CurrentDeclSecurityTarget == null)
423 return;
424
425 AddPermission (sec_action, perm, CurrentDeclSecurityTarget.DeclSecurity);
426 }
427
428 private void AddPermission (PEAPI.SecurityAction sec_action, object perm, DeclSecurity decl_sec)
429 {
430 SSPermissionSet ps = perm as SSPermissionSet;
431 if (ps != null) {
432 decl_sec.AddPermissionSet (sec_action, ps);
433 return;
434 }
435
436 IPermission iper = perm as IPermission;
437 if (iper != null) {
438 decl_sec.AddPermission (sec_action, iper);
439 return;
440 }
441
442 MIPermissionSet ps20 = perm as MIPermissionSet;
443 if (ps20 != null) {
444 decl_sec.AddPermissionSet (sec_action, ps20);
445 return;
446 }
447 }
448
449 public void Write ()
450 {
451 FileStream out_stream = null;
452
453 try {
454 if (ThisModule == null)
455 this_module = new Module (Path.GetFileName (output_file));
456
457 out_stream = new FileStream (output_file, FileMode.Create, FileAccess.Write);
458 pefile = new PEFile (ThisAssembly != null ? ThisAssembly.Name : null, ThisModule.Name, is_dll, ThisAssembly != null, null, out_stream);
459 PEAPI.Assembly asmb = pefile.GetThisAssembly ();
460
461 ThisModule.PeapiModule = pefile.GetThisModule ();
462
463 if (file_ref != null)
464 file_ref.Resolve (this);
465
466 extern_table.Resolve (this);
467 type_manager.DefineAll ();
468
469 if (manifestResources != null) {
470 foreach (ManifestResource mr in manifestResources)
471 pefile.AddManifestResource (mr);
472 }
473
474 foreach (FieldDef fielddef in global_field_table.Values) {
475 fielddef.Define (this);
476 }
477
478 foreach (MethodDef methoddef in global_method_table.Values) {
479 methoddef.Define (this);
480 }
481
482 foreach (TypeDef typedef in defcont_list) {
483 typedef.DefineContents (this);
484 }
485
486 if (ThisAssembly != null)
487 ThisAssembly.Resolve (this, pefile.GetThisAssembly ());
488 ThisModule.Resolve (this, pefile.GetThisModule ());
489
490 if (sub_system != -1)
491 pefile.SetSubSystem ((PEAPI.SubSystem) sub_system);
492 if (cor_flags != -1)
493 pefile.SetCorFlags (cor_flags);
494 if (stack_reserve != -1)
495 pefile.SetStackReserve (stack_reserve);
496
497 pefile.WritePEFile ();
498
499 if (symwriter != null) {
500 Guid guid = pefile.GetThisModule ().Guid;
501 symwriter.Write (guid);
502 }
503 } finally {
504 if (out_stream != null)
505 out_stream.Close ();
506 }
507 }
508
509 public PEAPI.Method ResolveMethod (string signature)
510 {
511 MethodDef methoddef = (MethodDef) global_method_table[signature];
512 if (methoddef == null)
513 Report.Error ("Unable to resolve global method : " + signature);
514
515 return methoddef.Resolve (this);
516 }
517
518 public PEAPI.Method ResolveVarargMethod (string sig_only_required_params, string sig_with_optional_params,
519 CodeGen code_gen, PEAPI.Type[] opt)
520 {
521 MethodDef methoddef = (MethodDef) global_method_table [sig_only_required_params];
522 if (methoddef == null)
523 Report.Error ("Unable to resolve global method : " + sig_only_required_params);
524
525 methoddef.Resolve (code_gen);
526 return methoddef.GetVarargSig (opt, sig_with_optional_params);
527 }
528
529 public PEAPI.Field ResolveField (string name, string type_name)
530 {
531 FieldDef fielddef = (FieldDef) global_field_table[new DictionaryEntry (name, type_name)];
532 if (fielddef == null)
533 Report.Error (String.Format ("Unable to resolve global field : {0} {1}", type_name, name));
534
535 return fielddef.Resolve (this);
536 }
537
538 private string CacheName (string name)
539 {
540 if (current_namespace == null ||
541 current_namespace == String.Empty)
542 return name;
543
544 return current_namespace + "." + name;
545 }
546 }
547
548 }
549
git-svn mirror of mcs