2010-02-03 Sebastien Pouliot <sebastien@ximian.com>
[mono.git] / mcs / class / PEAPI / Metadata.cs
index c77ae5d9b650fdf484389ed14b6f525c679558db..49ed4dc52331de7a14edb01a82f4c05dc2464089 100644 (file)
@@ -2,6 +2,7 @@ using System;
 using System.IO;
 using System.Collections;
 using System.Text;
+using System.Reflection;
 
 namespace PEAPI {
 
@@ -28,7 +29,7 @@ namespace PEAPI {
        /// <summary>
        /// Attributes for this assembly
        /// </summary>
-       public enum AssemAttr { EnableJITCompileTracking = 0x8000, 
+       public enum AssemAttr { Retargetable = 0x100, EnableJITCompileTracking = 0x8000, 
                DisableJITCompileOptimizer = 0x4000}
 
        /// <summary>
@@ -71,7 +72,7 @@ namespace PEAPI {
                Family, FamOrAssem, Public, Static = 0x0010, PublicStatic = 0x16, 
                Final = 0x0020, PublicStaticFinal = 0x36, Virtual = 0x0040, 
                PrivateVirtual, PublicVirtual = 0x0046, HideBySig = 0x0080, 
-               NewSlot = 0x0100, Abstract = 0x0400, SpecialName = 0x0800,
+               NewSlot = 0x0100, Strict = 0x200, Abstract = 0x0400, SpecialName = 0x0800,
                RTSpecialName = 0x1000, SpecialRTSpecialName = 0x1800, 
                HasSecurity = 0x4000, RequireSecObject = 0x8000}
 
@@ -241,6 +242,8 @@ namespace PEAPI {
                protected bool done = false;
                protected MDTable tabIx;
                protected bool sortTable = false;
+               //Temporary hack.. 
+               private bool has_custom_attrs = false;
 
                internal MetaDataElement() { }
 
@@ -253,6 +256,11 @@ namespace PEAPI {
                        }
                }
 
+               public bool HasCustomAttr {
+                       get { return has_custom_attrs; }
+                       set { has_custom_attrs = value; }
+               }
+
                internal virtual uint GetCodedIx(CIx code) { return 0; }
 
                /// <summary>
@@ -463,9 +471,14 @@ namespace PEAPI {
 
                internal sealed override void BuildTables(MetaData md) 
                {
+                       md.AddToTable(MDTable.CustomAttribute, this);
+                       if (byteVal == null) {
+                               valIx = 0;
+                               return;
+                       }
+
                        BinaryWriter bw = new BinaryWriter(new MemoryStream());
                        bw.Write(byteVal);
-                       md.AddToTable(MDTable.CustomAttribute, this);
                        MemoryStream str = (MemoryStream)bw.BaseStream;
                        valIx = md.AddToBlobHeap(str.ToArray());
                }
@@ -489,19 +502,17 @@ namespace PEAPI {
        /// Descriptor for security permissions for a class or a method
        /// </summary>
 
-       public class DeclSecurity : MetaDataElement {
+       public abstract class BaseDeclSecurity : MetaDataElement {
 
                ushort action;
                MetaDataElement parent;
                uint permissionIx;
-               byte [] byteVal;
 
-               internal DeclSecurity(MetaDataElement paren, ushort act, byte [] val)        
+               internal BaseDeclSecurity(MetaDataElement paren, ushort act)
                {
                        parent = paren;
                        action = act;
                        tabIx = MDTable.DeclSecurity;
-                       byteVal = val;
                }
 
                internal override uint SortKey() 
@@ -518,15 +529,18 @@ namespace PEAPI {
                internal sealed override void BuildTables(MetaData md) 
                {
                        if (done) return;
+
                        BinaryWriter bw = new BinaryWriter (new MemoryStream ());
-                       bw.Write (byteVal);
                        md.AddToTable (MDTable.DeclSecurity, this);
                        MemoryStream str = (MemoryStream)bw.BaseStream;
+                       WriteSig (bw);
                        permissionIx = md.AddToBlobHeap(str.ToArray());
 
                        done = true;
                }
 
+               internal abstract void WriteSig (BinaryWriter bw);
+
                internal sealed override void Write(FileImage output) 
                {
                        output.Write(action);
@@ -536,6 +550,154 @@ namespace PEAPI {
 
        }
 
+       public class DeclSecurity : BaseDeclSecurity {
+
+               byte [] byteVal;
+
+               internal DeclSecurity(MetaDataElement paren, ushort act, byte [] val)        
+                       : base (paren, act)
+               {
+                       byteVal = val;
+               }
+
+               internal override void WriteSig (BinaryWriter bw)
+               {
+                       bw.Write (byteVal);
+               }
+
+       }
+
+       public class DeclSecurity_20 : BaseDeclSecurity {
+
+               PermissionSet ps;
+
+               internal DeclSecurity_20 (MetaDataElement paren, ushort act, PermissionSet ps)
+                       : base (paren, act)
+               {
+                        this.ps = ps;
+               }
+
+               internal override void WriteSig (BinaryWriter bw)
+               {
+                       ps.Write (bw);
+               }
+       }
+
+       public class PermissionMember {
+
+               MemberTypes member_type;
+               PEAPI.Type type;
+               string name;
+               object value;
+
+               public PermissionMember (MemberTypes member_type, PEAPI.Type type, string name, object value)
+               {
+                       this.member_type = member_type;
+                       this.type = type;
+                       this.name = name;
+                       this.value = value;
+               }
+
+               public void Write (BinaryWriter bw)
+               {
+                       byte [] b;
+
+                       if (member_type == MemberTypes.Field)
+                               bw.Write ((byte) 0x53);
+                       else
+                               //Property
+                               bw.Write ((byte) 0x54);
+
+                       if (type is PrimitiveType) {
+                               bw.Write (type.GetTypeIndex ());
+                       } else {
+                               //must be enum
+                               bw.Write ((byte) 0x55); //ENUM
+
+                               b = Encoding.UTF8.GetBytes (((ClassRef) type).TypeName ());
+                               MetaData.CompressNum ((uint) b.Length, (MemoryStream) bw.BaseStream);
+                               bw.Write (b);
+                       }
+                       
+                       b = Encoding.UTF8.GetBytes (name);
+                       MetaData.CompressNum ((uint) b.Length, (MemoryStream) bw.BaseStream);
+                       bw.Write (b);
+
+                       ((Constant) value).Write (bw);
+               }
+
+       }
+
+       public class Permission
+       {
+               PEAPI.Type type;
+
+               //PermissionMembers
+               ArrayList members;
+               string name;
+
+               public Permission (PEAPI.Type type, string name)
+               {
+                       this.type = type;
+                       this.name = name;
+               }
+
+               public void AddMember (PEAPI.PermissionMember member)
+               {
+                       if (members == null)
+                               members = new ArrayList ();
+
+                       members.Add (member);
+               }
+
+               public void Write (BinaryWriter bw)
+               {
+                       byte [] b = Encoding.UTF8.GetBytes (name);
+                       MetaData.CompressNum ((uint) b.Length, (MemoryStream) bw.BaseStream);
+                       bw.Write (b);
+
+                       BinaryWriter perm_writer = new BinaryWriter (new MemoryStream (), Encoding.Unicode);
+                       MemoryStream str = (MemoryStream) perm_writer.BaseStream;
+
+                       MetaData.CompressNum ((uint) members.Count, str);//number of params
+                       foreach (PermissionMember member in members)
+                               member.Write (perm_writer);
+
+                       bw.Write ((byte) str.Length); //(optional) parameters length
+                       bw.Write (str.ToArray ());
+               }
+       }
+
+       public class PermissionSet 
+       {
+               PEAPI.SecurityAction sec_action;
+               ArrayList permissions;
+               PEAPI.PermissionSet ps;
+
+               public PermissionSet (PEAPI.SecurityAction sec_action)
+               {
+                       this.sec_action = sec_action;
+               }
+
+               public void AddPermission (PEAPI.Permission perm)
+               {
+                       if (permissions == null)
+                               permissions = new ArrayList ();
+
+                       permissions.Add (perm);
+               }
+
+               public void Write (BinaryWriter bw)
+               {
+                       bw.Write ((byte) 0x2e);
+                       MetaData.CompressNum ((uint) permissions.Count, (MemoryStream) bw.BaseStream);
+
+                       foreach (Permission perm in permissions)
+                               perm.Write (bw);
+               }
+
+       }
+
        /**************************************************************************/  
        /// <summary>
        /// Descriptor for layout information for a field
@@ -1178,6 +1340,10 @@ namespace PEAPI {
                        tabIx = MDTable.Param;
                }
 
+               public bool HasMarshalInfo {
+                       get { return marshalInfo != null; }
+               }
+
                /// <summary>
                /// Add a default value to this parameter
                /// </summary>
@@ -1538,9 +1704,14 @@ namespace PEAPI {
                /// <param name="pars">parameters</param>
                /// <returns>a descriptor for this new method</returns>
                public MethodDef AddMethod(string name, Type retType, Param[] pars) 
+               {
+                       return AddMethod (name, new Param (ParamAttr.Default, "", retType), pars);
+               }
+
+               public MethodDef AddMethod (string name, Param ret_param, Param [] pars) 
                {
                        // Console.WriteLine("Adding method " + name + " to class " + this.name);
-                       MethodDef meth = new MethodDef(metaData,name,retType, pars);
+                       MethodDef meth = new MethodDef(metaData,name, ret_param, pars);
                        methods.Add(meth);
                        return meth;
                }
@@ -1555,9 +1726,9 @@ namespace PEAPI {
                /// <param name="pars">parameters</param>
                /// <returns>a descriptor for this new method</returns>
                public MethodDef AddMethod(MethAttr mAtts, ImplAttr iAtts, string name, 
-                               Type retType, Param[] pars) {
+                               Param ret_param, Param [] pars) {
                        // Console.WriteLine("Adding method " + name + " to class " + this.name);
-                       MethodDef meth = new MethodDef(metaData,mAtts,iAtts,name,retType,pars);
+                       MethodDef meth = new MethodDef (metaData, mAtts, iAtts, name, ret_param, pars);
                        methods.Add(meth);
                        return meth;
                }
@@ -3650,7 +3821,6 @@ namespace PEAPI {
                // private static readonly byte LocalSigByte = 0x7;
                uint parIx = 0, textOffset = 0;
                private CallConv callConv = CallConv.Default;
-               private Type retType;
                private int gen_param_count;
 
                MetaData metaData;
@@ -3667,21 +3837,23 @@ namespace PEAPI {
                ImplMap pinvokeImpl;
                Param ret_param;
 
-
-               internal MethodDef(MetaData md, string name, Type retType, Param[] pars) : base (name)
+               internal MethodDef (MetaData md, string name, Param ret_param, Param [] pars)
+                       : this (md, 0, 0, name, ret_param, pars)
                {
-                       this.retType = retType;
-                       metaData = md;
-                       parList = pars;
-                       if (parList != null) numPars = parList.Length;
-                       tabIx = MDTable.Method;
                }
 
                internal MethodDef (MetaData md, MethAttr mAttrSet, ImplAttr iAttrSet, string name, 
-                               Type retType, Param [] pars) : this (md, name, retType, pars)
+                               Param ret_param, Param [] pars) 
+                       : base (name)
                {
                        methFlags = (ushort)mAttrSet;
                        implFlags = (ushort)iAttrSet;
+                       this.ret_param = ret_param;
+                       metaData = md;
+                       parList = pars;
+                       if (parList != null) 
+                               numPars = parList.Length;
+                       tabIx = MDTable.Method;
                }
 
                internal Param[] GetPars() 
@@ -3767,7 +3939,6 @@ namespace PEAPI {
                /* Add Marshal info for return type */
                public void AddRetTypeMarshallInfo (NativeType marshallType) 
                {
-                       ret_param = new Param (ParamAttr.HasFieldMarshal, "", retType);
                        ret_param.AddMarshallInfo (marshallType);
                }
 
@@ -3802,7 +3973,7 @@ namespace PEAPI {
                        for (int i=0; i < numPars; i++) {
                                pars[i] = parList[i].GetParType();
                        }
-                       varArgSig = new MethodRef(this,name,retType,pars,true,optPars, 0);
+                       varArgSig = new MethodRef (this, name, ret_param.GetParType (), pars, true, optPars, 0);
 
                        if (varArgSigList == null)
                                varArgSigList = new ArrayList ();
@@ -3816,9 +3987,9 @@ namespace PEAPI {
                        if ((callConv & CallConv.Generic) == CallConv.Generic)
                                MetaData.CompressNum ((uint) gen_param_count, sig);
                        MetaData.CompressNum((uint)numPars,sig);
-                       if (ret_param != null)
-                               ret_param.seqNo = 0;
-                       retType.TypeSig(sig);
+
+                       ret_param.seqNo = 0;
+                       ret_param.TypeSig (sig);
                        for (ushort i=0; i < numPars; i++) {
                                parList[i].seqNo = (ushort)(i+1);
                                parList[i].TypeSig(sig);
@@ -3847,7 +4018,7 @@ namespace PEAPI {
                        nameIx = md.AddToStringsHeap(name);
                        sigIx = GetSigIx(md);
                        parIx = md.TableIndex(MDTable.Param);
-                       if (ret_param != null) {
+                       if (ret_param.HasMarshalInfo || ret_param.HasCustomAttr) {
                                md.AddToTable(MDTable.Param, ret_param);
                                ret_param.BuildTables(md);
                        }
@@ -4031,6 +4202,8 @@ namespace PEAPI {
                public static readonly NativeType VariantBool = new NativeType(0x25);
                public static readonly NativeType FuncPtr = new NativeType(0x26);
                public static readonly NativeType AsAny = new NativeType(0x28);
+               public static readonly NativeType LPStruct = new NativeType(0x2b);
+               public static readonly NativeType Error = new NativeType(0x2d);
 
                protected byte typeIndex;
 
@@ -4372,6 +4545,11 @@ namespace PEAPI {
                        tabIx = MDTable.AssemblyRef;
                }
 
+               public void AddAssemblyAttr (AssemAttr aa)
+               {
+                       flags |= (uint)aa;
+               }
+
                /// <summary>
                /// Add version information about this external assembly
                /// </summary>
@@ -4946,7 +5124,7 @@ namespace PEAPI {
                        cattr_list.Add (cattr);
                }
 
-               internal void AddDeclSecurity (DeclSecurity decl_sec)
+               internal void AddDeclSecurity (BaseDeclSecurity decl_sec)
                {
                        if (declsec_list == null)
                                declsec_list = new ArrayList ();
@@ -5233,7 +5411,7 @@ namespace PEAPI {
                        }
 
                        if (declsec_list != null) {
-                               foreach (DeclSecurity decl_sec in declsec_list)
+                               foreach (BaseDeclSecurity decl_sec in declsec_list)
                                        decl_sec.BuildTables (this);
                        }