2003-11-06 Zoltan Varga <vargaz@freemail.hu>
[mono.git] / mcs / class / corlib / System.Reflection.Emit / AssemblyBuilder.cs
index 71990fb4d69744bc7db3d4aed66983eeb41d7649..144b7133a1d33efb30e17300851641970cd1c1a7 100755 (executable)
@@ -16,6 +16,11 @@ using System.Runtime.Serialization;
 using System.Globalization;
 using System.Runtime.CompilerServices;
 using System.Collections;
+using System.Runtime.InteropServices;
+using System.Security.Cryptography;
+
+using Mono.Security;
+using Mono.Security.Cryptography;
 
 namespace System.Reflection.Emit {
 
@@ -34,18 +39,20 @@ namespace System.Reflection.Emit {
                private string dir;
                private CustomAttributeBuilder[] cattrs;
                private MonoResource[] resources;
-               string keyfile;
+               byte[] public_key;
                string version;
                string culture;
                uint algid;
                uint flags;
                PEFileKinds pekind = PEFileKinds.Dll;
                bool delay_sign;
+               uint access;
                internal Type corlib_object_type = typeof (System.Object);
                internal Type corlib_value_type = typeof (System.ValueType);
                internal Type corlib_enum_type = typeof (System.Enum);
-               private int[] table_indexes;
-               internal ArrayList methods;
+               internal Type corlib_void_type = typeof (void);
+               ArrayList resource_writers = null;
+               bool created;
 
                [MethodImplAttribute(MethodImplOptions.InternalCall)]
                private static extern void basic_init (AssemblyBuilder ab);
@@ -53,29 +60,23 @@ namespace System.Reflection.Emit {
                internal AssemblyBuilder (AssemblyName n, string directory, AssemblyBuilderAccess access) {
                        name = n.Name;
                        dir = directory;
-                       basic_init (this);
-               }
+                       this.access = (uint)access;
 
-               internal int get_next_table_index (object obj, int table, bool inc) {
-                       if (table_indexes == null) {
-                               table_indexes = new int [64];
-                               for (int i=0; i < 64; ++i)
-                                       table_indexes [i] = 1;
-                               /* allow room for .<Module> in TypeDef table */
-                               table_indexes [0x02] = 2;
+                       /* Set defaults from n */
+                       if (n.CultureInfo != null) {
+                               culture = n.CultureInfo.Name;
                        }
-                       // Console.WriteLine ("getindex for table "+table.ToString()+" got "+table_indexes [table].ToString());
-                       if (inc) {
-                               if ((table == 0x06) && (methods != null))
-                                       methods.Add (obj);
-                               return table_indexes [table]++;
+                       Version v = n.Version;
+                       if (v != null) {
+                               version = v.ToString ();
                        }
-                       return table_indexes [table];
+
+                       basic_init (this);
                }
 
                public override string CodeBase {
                        get {
-                               return null;
+                               throw not_supported ();
                        }
                }
                
@@ -87,7 +88,7 @@ namespace System.Reflection.Emit {
 
                public override string Location {
                        get {
-                               return null;
+                               throw not_supported ();
                        }
                }
 
@@ -98,6 +99,18 @@ namespace System.Reflection.Emit {
 
                public void AddResourceFile (string name, string fileName, ResourceAttributes attribute)
                {
+                       AddResourceFile (name, fileName, attribute, true);
+               }
+
+               private void AddResourceFile (string name, string fileName, ResourceAttributes attribute, bool fileNeedsToExists)
+               {
+                       check_name_and_filename (name, fileName, fileNeedsToExists);
+
+                       // Resource files are created/searched under the assembly storage
+                       // directory
+                       if (dir != null)
+                               fileName = Path.Combine (dir, fileName);
+
                        if (resources != null) {
                                MonoResource[] new_r = new MonoResource [resources.Length + 1];
                                System.Array.Copy(resources, new_r, resources.Length);
@@ -139,25 +152,57 @@ namespace System.Reflection.Emit {
                        }
                }
 
+               internal void EmbedResource (string name, byte[] blob, ResourceAttributes attribute)
+               {
+                       if (resources != null) {
+                               MonoResource[] new_r = new MonoResource [resources.Length + 1];
+                               System.Array.Copy(resources, new_r, resources.Length);
+                               resources = new_r;
+                       } else {
+                               resources = new MonoResource [1];
+                       }
+                       int p = resources.Length - 1;
+                       resources [p].name = name;
+                       resources [p].attrs = attribute;
+                       resources [p].data = blob;
+               }
+
                public ModuleBuilder DefineDynamicModule (string name)
                {
-                       return DefineDynamicModule (name, name, false);
+                       return DefineDynamicModule (name, name, false, true);
                }
 
                public ModuleBuilder DefineDynamicModule (string name, bool emitSymbolInfo)
                {
-                       return DefineDynamicModule (name, name, emitSymbolInfo);
+                       return DefineDynamicModule (name, name, emitSymbolInfo, true);
                }
 
                public ModuleBuilder DefineDynamicModule(string name, string fileName)
                {
-                       return DefineDynamicModule (name, fileName, false);
+                       return DefineDynamicModule (name, fileName, false, false);
                }
 
                public ModuleBuilder DefineDynamicModule (string name, string fileName,
                                                          bool emitSymbolInfo)
                {
-                       ModuleBuilder r = new ModuleBuilder (this, name, fileName, emitSymbolInfo);
+                       return DefineDynamicModule (name, fileName, emitSymbolInfo, false);
+               }
+
+               public ModuleBuilder DefineDynamicModule (string name, string fileName,
+                                                         bool emitSymbolInfo, bool transient)
+               {
+                       check_name_and_filename (name, fileName, false);
+
+                       if (!transient) {
+                               if (Path.GetExtension (fileName) == String.Empty)
+                                       throw new ArgumentException ("Module file name '" + fileName + "' must have file extension.");
+                               if (!IsSave)
+                                       throw new NotSupportedException ("Persistable modules are not supported in a dynamic assembly created with AssemblyBuilderAccess.Run");
+                               if (created)
+                                       throw new InvalidOperationException ("Assembly was already saved.");
+                       }
+
+                       ModuleBuilder r = new ModuleBuilder (this, name, fileName, emitSymbolInfo, transient);
 
                        if (modules != null) {
                                ModuleBuilder[] new_modules = new ModuleBuilder [modules.Length + 1];
@@ -179,97 +224,132 @@ namespace System.Reflection.Emit {
                public IResourceWriter DefineResource (string name, string description,
                                                       string fileName, ResourceAttributes attribute)
                {
-                       return null;
+                       IResourceWriter writer;
+
+                       // description seems to be ignored
+                       AddResourceFile (name, fileName, attribute, false);
+                       writer = new ResourceWriter (fileName);
+                       if (resource_writers == null)
+                               resource_writers = new ArrayList ();
+                       resource_writers.Add (writer);
+                       return writer;
                }
 
+               [MonoTODO]
                public void DefineUnmanagedResource (byte[] resource)
                {
+                       if (resource == null)
+                               throw new ArgumentNullException ("resource");
+
+                       throw new NotImplementedException ();
                }
 
+               [MonoTODO]
                public void DefineUnmanagedResource (string resourceFileName)
                {
+                       if (resourceFileName == null)
+                               throw new ArgumentNullException ("resourceFileName");
+                       if (resourceFileName == String.Empty)
+                               throw new ArgumentException ("resourceFileName");
+                       if (!File.Exists (resourceFileName) || Directory.Exists (resourceFileName))
+                               throw new FileNotFoundException ("File '" + resourceFileName + "' does not exists or is a directory.");
+
+                       throw new NotImplementedException ();
                }
 
+               [MonoTODO]
                public void DefineVersionInfoResource ()
                {
+                       throw new NotImplementedException ();
                }
 
+               [MonoTODO]
                public void DefineVersionInfoResource (string product, string productVersion,
                                                       string company, string copyright, string trademark)
                {
+                       throw new NotImplementedException ();
                }
 
                public ModuleBuilder GetDynamicModule (string name)
                {
+                       if (name == null)
+                               throw new ArgumentNullException ("name");
+                       if (name == "")
+                               throw new ArgumentException ("Name can't be null");
+
+                       if (modules != null)
+                               for (int i = 0; i < modules.Length; ++i)
+                                       if (modules [i].name == name)
+                                               return modules [i];
                        return null;
                }
 
                public override Type[] GetExportedTypes ()
                {
-                       return null;
+                       throw not_supported ();
                }
 
                public override FileStream GetFile (string name)
                {
-                       return null;
+                       throw not_supported ();
                }
 
-               /*public virtual FileStream[] GetFiles() {
-                       return null;
+               public override FileStream[] GetFiles() {
+                       throw not_supported ();
                }
-               public override FileStream[] GetFiles(bool getResourceModules) {
-                       return null;
-               }*/
 
-               /*public virtual ManifestResourceInfo GetManifestResourceInfo(string resourceName)
-                 {
-                       return null;
-               }
-               public virtual string[] GetManifestResourceNames() {
-                       return null;
+               public override FileStream[] GetFiles(bool getResourceModules) {
+                       throw not_supported ();
                }
-               public virtual Stream GetManifestResourceStream(string name) {
-                       return null;
+
+               public override ManifestResourceInfo GetManifestResourceInfo(string resourceName) {
+                       throw not_supported ();
                }
-               public virtual Stream GetManifestResourceStream(Type type, string name) {
-                       return null;
-               }*/
 
-               [MethodImplAttribute(MethodImplOptions.InternalCall)]
-               private static extern int getUSIndex (AssemblyBuilder ab, string str);
+               public override string[] GetManifestResourceNames() {
+                       throw not_supported ();
+               }
 
-               [MethodImplAttribute(MethodImplOptions.InternalCall)]
-               private static extern int getToken (AssemblyBuilder ab, MemberInfo member);
+               public override Stream GetManifestResourceStream(string name) {
+                       throw not_supported ();
+               }
+               public override Stream GetManifestResourceStream(Type type, string name) {
+                       throw not_supported ();
+               }
 
-               internal int GetToken (string str) {
-                       return getUSIndex (this, str);
+               internal bool IsSave {
+                       get {
+                               return access != (uint)AssemblyBuilderAccess.Run;
+                       }
                }
-               
-               internal int GetToken (MemberInfo member) {
-                       return getToken (this, member);
+
+               internal string AssemblyDir {
+                       get {
+                               return dir;
+                       }
                }
-               
-               [MethodImplAttribute(MethodImplOptions.InternalCall)]
-               private static extern int getDataChunk (AssemblyBuilder ab, byte[] buf, int offset);
 
                public void Save (string assemblyFileName)
                {
-                       byte[] buf = new byte [65536];
-                       FileStream file;
-                       int count, offset;
-
-                       if (dir != null) {
-                               assemblyFileName = String.Format ("{0}{1}{2}", dir, System.IO.Path.DirectorySeparatorChar, assemblyFileName);
+                       if (resource_writers != null) {
+                               foreach (IResourceWriter writer in resource_writers) {
+                                       writer.Generate ();
+                                       writer.Close ();
+                               }
                        }
 
-                       file = new FileStream (assemblyFileName, FileMode.Create, FileAccess.Write);
+                       // Create a main module if not already created
+                       ModuleBuilder mainModule = null;
+                       foreach (ModuleBuilder module in modules)
+                               if (module.FullyQualifiedName == assemblyFileName)
+                                       mainModule = module;
+                       if (mainModule == null)
+                               mainModule = DefineDynamicModule ("RefEmit_OnDiskManifestModule", assemblyFileName);
 
-                       offset = 0;
-                       while ((count = getDataChunk (this, buf, offset)) != 0) {
-                               file.Write (buf, 0, count);
-                               offset += count;
-                       }
-                       file.Close ();
+                       mainModule.IsMain = true;
+
+                       foreach (ModuleBuilder module in modules)
+                               module.Save ();
                }
 
                public void SetEntryPoint (MethodInfo entryMethod)
@@ -279,14 +359,24 @@ namespace System.Reflection.Emit {
 
                public void SetEntryPoint (MethodInfo entryMethod, PEFileKinds fileKind)
                {
+                       if (entryMethod == null)
+                               throw new ArgumentNullException ("entryMethod");
+                       if (entryMethod.DeclaringType.Assembly != this)
+                               throw new InvalidOperationException ("Entry method is not defined in the same assembly.");
+
                        entry_point = entryMethod;
                        pekind = fileKind;
                }
 
-               public void SetCustomAttribute( CustomAttributeBuilder customBuilder) {
+               public void SetCustomAttribute( CustomAttributeBuilder customBuilder) 
+               {
+                       if (customBuilder == null)
+                               throw new ArgumentNullException ("customBuilder");
+
                        string attrname = customBuilder.Ctor.ReflectedType.FullName;
                        byte[] data;
                        int len, pos;
+                       Mono.Security.StrongName sn;
                        if (attrname == "System.Reflection.AssemblyVersionAttribute") {
                                data = customBuilder.Data;
                                pos = 2;
@@ -297,7 +387,33 @@ namespace System.Reflection.Emit {
                                data = customBuilder.Data;
                                pos = 2;
                                len = CustomAttributeBuilder.decode_len (data, pos, out pos);
-                               keyfile = CustomAttributeBuilder.string_from_bytes (data, pos, len);
+                               string keyfile_name = CustomAttributeBuilder.string_from_bytes (data, pos, len);
+                               if (keyfile_name == String.Empty)
+                                       return;
+                               using (FileStream fs = new FileStream (keyfile_name, FileMode.Open)) {
+                                       byte[] snkeypair = new byte [fs.Length];
+                                       fs.Read (snkeypair, 0, snkeypair.Length);
+
+                                       // this will import public or private/public keys
+                                       RSA rsa = CryptoConvert.FromCapiKeyBlob (snkeypair);
+                                       // and export only the public part
+                                       sn = new Mono.Security.StrongName (rsa);
+                                       public_key = sn.PublicKey;
+                               }
+                               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);
+                               if (key_name == String.Empty)
+                                       return;
+                               CspParameters csparam = new CspParameters ();
+                               csparam.KeyContainerName = key_name;
+                               RSA rsacsp = new RSACryptoServiceProvider (csparam);
+                               sn = new Mono.Security.StrongName (rsacsp);
+                               public_key = sn.PublicKey;
+                               return;
                        } else if (attrname == "System.Reflection.AssemblyCultureAttribute") {
                                data = customBuilder.Data;
                                pos = 2;
@@ -334,6 +450,11 @@ namespace System.Reflection.Emit {
                        }
                }
                public void SetCustomAttribute( ConstructorInfo con, byte[] binaryAttribute) {
+                       if (con == null)
+                               throw new ArgumentNullException ("con");
+                       if (binaryAttribute == null)
+                               throw new ArgumentNullException ("binaryAttribute");
+
                        SetCustomAttribute (new CustomAttributeBuilder (con, binaryAttribute));
                }
 
@@ -342,5 +463,63 @@ namespace System.Reflection.Emit {
                        this.corlib_value_type = corlib_value_type;
                        this.corlib_enum_type = corlib_enum_type;
                }
+
+               public void SetCorlibTypeBuilders (Type corlib_object_type, Type corlib_value_type, Type corlib_enum_type, Type corlib_void_type)
+               {
+                       SetCorlibTypeBuilders (corlib_object_type, corlib_value_type, corlib_enum_type);
+                       this.corlib_void_type = corlib_void_type;
+               }
+
+               private Exception not_supported () {
+                       // Strange message but this is what MS.NET prints...
+                       return new NotSupportedException ("The invoked member is not supported in a dynamic module.");
+               }
+
+               private void check_name_and_filename (string name, string fileName,
+                                                                                         bool fileNeedsToExists) {
+                       if (name == null)
+                               throw new ArgumentNullException ("name");
+                       if (fileName == null)
+                               throw new ArgumentNullException ("fileName");
+                       if (name == "")
+                               throw new ArgumentException ("name cannot be empty", "name");
+                       if (fileName == "")
+                               throw new ArgumentException ("fileName cannot be empty", "fileName");
+                       if (Path.GetFileName (fileName) != fileName)
+                               throw new ArgumentException ("fileName '" + fileName + "' must not include a path.");
+
+                       // Resource files are created/searched under the assembly storage
+                       // directory
+                       string fullFileName = fileName;
+                       if (dir != null)
+                               fullFileName = Path.Combine (dir, fileName);
+
+                       if (fileNeedsToExists && !File.Exists (fullFileName))
+                               throw new FileNotFoundException ("Could not find file '" + fileName + "'");
+
+                       if (resources != null) {
+                               for (int i = 0; i < resources.Length; ++i) {
+                                       if (resources [i].filename == fullFileName)
+                                               throw new ArgumentException ("Duplicate file name '" + fileName + "'");
+                                       if (resources [i].name == name)
+                                               throw new ArgumentException ("Duplicate name '" + name + "'");
+                               }
+                       }
+
+                       if (modules != null) {
+                               for (int i = 0; i < modules.Length; ++i) {
+                                       // Use fileName instead of fullFileName here
+                                       if (!modules [i].IsTransient () && (modules [i].FileName == fileName))
+                                               throw new ArgumentException ("Duplicate file name '" + fileName + "'");
+                                       if (modules [i].Name == name)
+                                               throw new ArgumentException ("Duplicate name '" + name + "'");
+                               }
+                       }
+               }
+
+               // Same as in Mono.Security/Mono>Security.Cryptography/CryptoConvert.cs
+
+
+
        }
 }