2004-01-13 Zoltan Varga <vargaz@freemail.hu>
[mono.git] / mcs / class / corlib / System.Reflection.Emit / AssemblyBuilder.cs
index efc070517a5569ea81241a36fb448935dc59c2da..c7487443bd4a90ba24afac2268ed778a8afd3e8b 100755 (executable)
@@ -18,20 +18,47 @@ using System.Runtime.CompilerServices;
 using System.Collections;
 using System.Runtime.InteropServices;
 using System.Security.Cryptography;
+using System.Security.Permissions;
 
 using Mono.Security;
 using Mono.Security.Cryptography;
 
 namespace System.Reflection.Emit {
 
+       internal struct RefEmitPermissionSet {
+               public SecurityAction action;
+               public string pset;
+
+               public RefEmitPermissionSet (SecurityAction action, string pset) {
+                       this.action = action;
+                       this.pset = pset;
+               }
+       }
+
        internal struct MonoResource {
                public byte[] data;
                public string name;
                public string filename;
                public ResourceAttributes attrs;
+               public int offset;
+       }
+
+       internal struct MonoWin32Resource {
+               public int res_type;
+               public int res_id;
+               public int lang_id;
+               public byte[] data;
+
+               public MonoWin32Resource (int res_type, int res_id, int lang_id, byte[] data) {
+                       this.res_type = res_type;
+                       this.res_id = res_id;
+                       this.lang_id = lang_id;
+                       this.data = data;
+               }
        }
 
        public sealed class AssemblyBuilder : Assembly {
+               #region Sync with reflection.h
                private IntPtr dynamic_assembly;
                private MethodInfo entry_point;
                private ModuleBuilder[] modules;
@@ -47,12 +74,17 @@ namespace System.Reflection.Emit {
                PEFileKinds pekind = PEFileKinds.Dll;
                bool delay_sign;
                uint access;
+               Module[] loaded_modules;
+               MonoWin32Resource[] win32_resources;
+               #endregion
                internal Type corlib_object_type = typeof (System.Object);
                internal Type corlib_value_type = typeof (System.ValueType);
                internal Type corlib_enum_type = typeof (System.Enum);
                internal Type corlib_void_type = typeof (void);
                ArrayList resource_writers = null;
+               Win32VersionResource version_res;
                bool created;
+               bool is_module_only;
 
                [MethodImplAttribute(MethodImplOptions.InternalCall)]
                private static extern void basic_init (AssemblyBuilder ab);
@@ -95,6 +127,15 @@ namespace System.Reflection.Emit {
                        }
                }
 
+#if NET_1_1
+               /* This is to keep signature compatibility with MS.NET */
+               public override string ImageRuntimeVersion {
+                       get {
+                               return base.ImageRuntimeVersion;
+                       }
+               }
+#endif
+
                public void AddResourceFile (string name, string fileName)
                {
                        AddResourceFile (name, fileName, ResourceAttributes.Public);
@@ -207,18 +248,45 @@ namespace System.Reflection.Emit {
 
                        ModuleBuilder r = new ModuleBuilder (this, name, fileName, emitSymbolInfo, transient);
 
+                       if ((modules != null) && is_module_only)
+                               throw new InvalidOperationException ("A module-only assembly can only contain one module.");
+
                        if (modules != null) {
                                ModuleBuilder[] new_modules = new ModuleBuilder [modules.Length + 1];
                                System.Array.Copy(modules, new_modules, modules.Length);
-                               new_modules [modules.Length] = r;
                                modules = new_modules;
                        } else {
                                modules = new ModuleBuilder [1];
-                               modules [0] = r;
                        }
+                       modules [modules.Length - 1] = r;
                        return r;
                }
 
+               [MethodImplAttribute(MethodImplOptions.InternalCall)]
+               private extern Module InternalAddModule (string fileName);
+
+               /*
+                * Mono extension to support /addmodule in mcs.
+                */
+               internal Module AddModule (string fileName)
+               {
+                       if (fileName == null)
+                               throw new ArgumentNullException (fileName);
+
+                       Module m = InternalAddModule (fileName);
+
+                       if (loaded_modules != null) {
+                               Module[] new_modules = new Module [loaded_modules.Length + 1];
+                               System.Array.Copy (loaded_modules, new_modules, loaded_modules.Length);
+                               loaded_modules = new_modules;
+                       } else {
+                               loaded_modules = new Module [1];
+                       }
+                       loaded_modules [loaded_modules.Length - 1] = m;
+
+                       return m;
+               }
+
                public IResourceWriter DefineResource (string name, string description, string fileName)
                {
                        return DefineResource (name, description, fileName, ResourceAttributes.Public);
@@ -238,16 +306,35 @@ namespace System.Reflection.Emit {
                        return writer;
                }
 
+               private void AddUnmanagedResource (Win32Resource res) {
+                       MemoryStream ms = new MemoryStream ();
+                       res.WriteTo (ms);
+
+                       if (win32_resources != null) {
+                               MonoWin32Resource[] new_res = new MonoWin32Resource [win32_resources.Length + 1];
+                               System.Array.Copy (win32_resources, new_res, win32_resources.Length);
+                               win32_resources = new_res;
+                       }
+                       else
+                               win32_resources = new MonoWin32Resource [1];
+
+                       win32_resources [win32_resources.Length - 1] = new MonoWin32Resource (res.Type.Id, res.Name.Id, res.Language, ms.ToArray ());
+               }
+
                [MonoTODO]
                public void DefineUnmanagedResource (byte[] resource)
                {
                        if (resource == null)
                                throw new ArgumentNullException ("resource");
 
+                       /*
+                        * The format of the argument byte array is not documented
+                        * so this method is impossible to implement.
+                        */
+
                        throw new NotImplementedException ();
                }
 
-               [MonoTODO]
                public void DefineUnmanagedResource (string resourceFileName)
                {
                        if (resourceFileName == null)
@@ -257,20 +344,106 @@ namespace System.Reflection.Emit {
                        if (!File.Exists (resourceFileName) || Directory.Exists (resourceFileName))
                                throw new FileNotFoundException ("File '" + resourceFileName + "' does not exists or is a directory.");
 
-                       throw new NotImplementedException ();
+                       using (FileStream fs = new FileStream (resourceFileName, FileMode.Open)) {
+                               Win32ResFileReader reader = new Win32ResFileReader (fs);
+
+                               foreach (Win32EncodedResource res in reader.ReadResources ()) {
+                                       if (res.Name.IsName || res.Type.IsName)
+                                               throw new InvalidOperationException ("resource files with named resources or non-default resource types are not supported.");
+
+                                       AddUnmanagedResource (res);
+                               }
+                       }
                }
 
-               [MonoTODO]
                public void DefineVersionInfoResource ()
                {
-                       throw new NotImplementedException ();
+                       if (version_res != null)
+                               throw new ArgumentException ("Native resource has already been defined.");                      
+
+                       version_res = new Win32VersionResource (1, 0);
+
+                       if (cattrs != null) {
+                               foreach (CustomAttributeBuilder cb in cattrs) {
+                                       string attrname = cb.Ctor.ReflectedType.FullName;
+
+                                       if (attrname == "System.Reflection.AssemblyProductAttribute")
+                                               version_res.ProductName = cb.string_arg ();
+                                       else if (attrname == "System.Reflection.AssemblyCompanyAttribute")
+                                               version_res.CompanyName = cb.string_arg ();
+                                       else if (attrname == "System.Reflection.AssemblyCopyrightAttribute")
+                                               version_res.LegalCopyright = cb.string_arg ();
+                                       else if (attrname == "System.Reflection.AssemblyTrademarkAttribute")
+                                               version_res.LegalTrademarks = cb.string_arg ();
+                                       else if (attrname == "System.Reflection.AssemblyCultureAttribute")
+                                               version_res.FileLanguage = new CultureInfo (cb.string_arg ()).LCID;
+                                       else if (attrname == "System.Reflection.AssemblyFileVersionAttribute")
+                                               version_res.FileVersion = cb.string_arg ();
+                                       else if (attrname == "System.Reflection.AssemblyInformationalVersionAttribute")
+                                               version_res.ProductVersion = cb.string_arg ();
+                                       else if (attrname == "System.Reflection.AssemblyTitleAttribute")
+                                               version_res.FileDescription = cb.string_arg ();
+                                       else if (attrname == "System.Reflection.AssemblyDescriptionAttribute")
+                                               version_res.Comments = cb.string_arg ();
+                               }
+                       }
                }
 
-               [MonoTODO]
                public void DefineVersionInfoResource (string product, string productVersion,
                                                       string company, string copyright, string trademark)
                {
-                       throw new NotImplementedException ();
+                       if (version_res != null)
+                               throw new ArgumentException ("Native resource has already been defined.");
+
+                       /*
+                        * We can only create the resource later, when the file name and
+                        * the binary version is known.
+                        */
+
+                       version_res = new Win32VersionResource (1, 0);
+                       version_res.ProductName = product;
+                       version_res.ProductVersion = productVersion;
+                       version_res.CompanyName = company;
+                       version_res.LegalCopyright = copyright;
+                       version_res.LegalTrademarks = trademark;
+               }
+
+               /* 
+                * Mono extension to support /win32icon in mcs
+                */
+               internal void DefineIconResource (string iconFileName)
+               {
+                       if (iconFileName == null)
+                               throw new ArgumentNullException ("iconFileName");
+                       if (iconFileName == String.Empty)
+                               throw new ArgumentException ("iconFileName");
+                       if (!File.Exists (iconFileName) || Directory.Exists (iconFileName))
+                               throw new FileNotFoundException ("File '" + iconFileName + "' does not exists or is a directory.");
+
+                       using (FileStream fs = new FileStream (iconFileName, FileMode.Open)) {
+                               Win32IconFileReader reader = new Win32IconFileReader (fs);
+                               
+                               ICONDIRENTRY[] entries = reader.ReadIcons ();
+
+                               Win32IconResource[] icons = new Win32IconResource [entries.Length];
+                               for (int i = 0; i < entries.Length; ++i) {
+                                       icons [i] = new Win32IconResource (i + 1, 0, entries [i]);
+                                       AddUnmanagedResource (icons [i]);
+                               }
+
+                               Win32GroupIconResource group = new Win32GroupIconResource (1, 0, icons);
+                               AddUnmanagedResource (group);
+                       }
+               }
+
+               private void DefineVersionInfoResourceImpl (string fileName) {
+                       // Add missing info
+                       if (version_res.FileVersion == "0.0.0.0")
+                               version_res.FileVersion = version;
+                       version_res.InternalName = Path.GetFileNameWithoutExtension (fileName);
+                       version_res.OriginalFilename = fileName;
+
+                       AddUnmanagedResource (version_res);
                }
 
                public ModuleBuilder GetDynamicModule (string name)
@@ -297,10 +470,6 @@ namespace System.Reflection.Emit {
                        throw not_supported ();
                }
 
-               public override FileStream[] GetFiles() {
-                       throw not_supported ();
-               }
-
                public override FileStream[] GetFiles(bool getResourceModules) {
                        throw not_supported ();
                }
@@ -332,6 +501,20 @@ namespace System.Reflection.Emit {
                        }
                }
 
+               /*
+                * Mono extension. If this is set, the assembly can only contain one
+                * module, access should be Save, and the saved image will not contain an
+                * assembly manifest.
+                */
+               internal bool IsModuleOnly {
+                       get {
+                               return is_module_only;
+                       }
+                       set {
+                               is_module_only = value;
+                       }
+               }
+
                public void Save (string assemblyFileName)
                {
                        if (resource_writers != null) {
@@ -343,14 +526,20 @@ namespace System.Reflection.Emit {
 
                        // Create a main module if not already created
                        ModuleBuilder mainModule = null;
-                       foreach (ModuleBuilder module in modules)
-                               if (module.FullyQualifiedName == assemblyFileName)
-                                       mainModule = module;
+                       if (modules != null) {
+                               foreach (ModuleBuilder module in modules)
+                                       if (module.FullyQualifiedName == assemblyFileName)
+                                               mainModule = module;
+                       }
                        if (mainModule == null)
                                mainModule = DefineDynamicModule ("RefEmit_OnDiskManifestModule", assemblyFileName);
 
-                       mainModule.IsMain = true;
+                       if (!is_module_only)
+                               mainModule.IsMain = true;
 
+                       if (version_res != null)
+                               DefineVersionInfoResourceImpl (assemblyFileName);
+                       
                        foreach (ModuleBuilder module in modules)
                                if (module != mainModule)
                                        module.Save ();
@@ -388,16 +577,10 @@ namespace System.Reflection.Emit {
                        int len, pos;
                        Mono.Security.StrongName sn;
                        if (attrname == "System.Reflection.AssemblyVersionAttribute") {
-                               data = customBuilder.Data;
-                               pos = 2;
-                               len = CustomAttributeBuilder.decode_len (data, pos, out pos);
-                               version = CustomAttributeBuilder.string_from_bytes (data, pos, len);
+                               version = create_assembly_version (customBuilder.string_arg ());
                                return;
                        } else if (attrname == "System.Reflection.AssemblyKeyFileAttribute") {
-                               data = customBuilder.Data;
-                               pos = 2;
-                               len = CustomAttributeBuilder.decode_len (data, pos, out pos);
-                               string keyfile_name = CustomAttributeBuilder.string_from_bytes (data, pos, len);
+                               string keyfile_name = customBuilder.string_arg ();
                                if (keyfile_name == String.Empty)
                                        return;
                                using (FileStream fs = new FileStream (keyfile_name, FileMode.Open)) {
@@ -412,10 +595,7 @@ namespace System.Reflection.Emit {
                                }
                                return;
                        } else if (attrname == "System.Reflection.AssemblyKeyNameAttribute") {
-                               data = customBuilder.Data;
-                               pos = 2;
-                               len = CustomAttributeBuilder.decode_len (data, pos, out pos);
-                               string key_name = CustomAttributeBuilder.string_from_bytes (data, pos, len);
+                               string key_name = customBuilder.string_arg ();
                                if (key_name == String.Empty)
                                        return;
                                CspParameters csparam = new CspParameters ();
@@ -425,10 +605,7 @@ namespace System.Reflection.Emit {
                                public_key = sn.PublicKey;
                                return;
                        } else if (attrname == "System.Reflection.AssemblyCultureAttribute") {
-                               data = customBuilder.Data;
-                               pos = 2;
-                               len = CustomAttributeBuilder.decode_len (data, pos, out pos);
-                               culture = CustomAttributeBuilder.string_from_bytes (data, pos, len);
+                               culture = customBuilder.string_arg ();
                        } else if (attrname == "System.Reflection.AssemblyAlgorithmIdAttribute") {
                                data = customBuilder.Data;
                                pos = 2;
@@ -459,7 +636,7 @@ namespace System.Reflection.Emit {
                                cattrs [0] = customBuilder;
                        }
                }
-               public void SetCustomAttribute( ConstructorInfo con, byte[] binaryAttribute) {
+               public void SetCustomAttribute ( ConstructorInfo con, byte[] binaryAttribute) {
                        if (con == null)
                                throw new ArgumentNullException ("con");
                        if (binaryAttribute == null)
@@ -527,9 +704,39 @@ namespace System.Reflection.Emit {
                        }
                }
 
-               // Same as in Mono.Security/Mono>Security.Cryptography/CryptoConvert.cs
+               private String create_assembly_version (String version) {
+                       String[] parts = version.Split ('.');
+                       int[] ver = new int [4] { 0, 0, 0, 0 };
 
+                       if ((parts.Length < 0) || (parts.Length > 4))
+                               throw new ArgumentException ("The version specified '" + version + "' is invalid");
 
+                       for (int i = 0; i < parts.Length; ++i) {
+                               if (parts [i] == "*") {
+                                       DateTime now = DateTime.Now;
 
+                                       if (i == 2) {
+                                               ver [2] = (now - new DateTime (2000, 1, 1)).Days;
+                                               if (parts.Length == 3)
+                                                       ver [3] = (now.Second + (now.Minute * 60) + (now.Hour * 3600)) / 2;
+                                       }
+                                       else
+                                               if (i == 3)
+                                                       ver [3] = (now.Second + (now.Minute * 60) + (now.Hour * 3600)) / 2;
+                                       else
+                                               throw new ArgumentException ("The version specified '" + version + "' is invalid");
+                               }
+                               else {
+                                       try {
+                                               ver [i] = Int32.Parse (parts [i]);
+                                       }
+                                       catch (FormatException) {
+                                               throw new ArgumentException ("The version specified '" + version + "' is invalid");
+                                       }
+                               }
+                       }
+
+                       return ver [0] + "." + ver [1] + "." + ver [2] + "." + ver [3];
+               }
        }
 }