[runtime] Fix mkbundle compilation on OSX
[mono.git] / mcs / tools / mkbundle / mkbundle.cs
index a76e9f88c92a1440897c007e92937f4d16b89311..6e3a3e0de4bdda47067e661ff5053594aa27e4fb 100755 (executable)
@@ -40,6 +40,8 @@ class MakeBundle {
        static string output = "a.out";
        static string object_out = null;
        static List<string> link_paths = new List<string> ();
+       static List<string> aot_paths = new List<string> ();
+       static List<string> aot_names = new List<string> ();
        static Dictionary<string,string> libraries = new Dictionary<string,string> ();
        static bool autodeps = false;
        static bool keeptemp = false;
@@ -63,6 +65,10 @@ class MakeBundle {
        static bool custom_mode = true;
        static string embedded_options = null;
        static string runtime = null;
+       static bool aot_compile = false;
+       static string aot_args = "static";
+       static string aot_mode = "";
+       static string aot_runtime = null;
        static string sdk_path = null;
        static string lib_path = null;
        static Dictionary<string,string> environment = new Dictionary<string,string>();
@@ -135,7 +141,7 @@ class MakeBundle {
                                        return 1;
                                }
                                if (sdk_path != null || runtime != null)
-                                       Error ("You can not specify one of --runtime, --sdk or --cross");
+                                       Error ("You can only specify one of --runtime, --sdk or --cross");
                                custom_mode = false;
                                autodeps = true;
                                cross_target = args [++i];
@@ -213,7 +219,7 @@ class MakeBundle {
                                autodeps = true;
                                sdk_path = args [++i];
                                if (cross_target != null || runtime != null)
-                                       Error ("You can not specify one of --runtime, --sdk or --cross");
+                                       Error ("You can only specify one of --runtime, --sdk or --cross");
                                break;
                        case "--runtime":
                                if (i+1 == top){
@@ -351,11 +357,33 @@ class MakeBundle {
                        case "--bundled-header":
                                bundled_header = true;
                                break;
+                       case "--aot-runtime":
+                               aot_runtime = args [++i];
+                               break;
+                       case "--aot-mode":
+                               if (i+1 == top) {
+                                       Console.WriteLine ("Need string of aot mode");
+                                       return 1;
+                               }
+                               aot_mode = args [++i];
+                               aot_compile = true;
+                               break;
+                       case "--aot-args":
+                               if (i+1 == top) {
+                                       Console.WriteLine ("AOT arguments are passed as a comma-delimited list");
+                                       return 1;
+                               }
+                               if (args [i + 1].Contains ("outfile")) {
+                                       Console.WriteLine ("Per-aot-output arguments (ex: outfile, llvm-outfile) cannot be given");
+                                       return 1;
+                               }
+                               aot_args = String.Format("static,{0}", args [++i]);
+                               aot_compile = true;
+                               break;
                        default:
                                sources.Add (args [i]);
                                break;
                        }
-
                }
                // Modern bundling starts here
                if (!custom_mode){
@@ -411,6 +439,10 @@ class MakeBundle {
                foreach (string file in assemblies)
                        if (!QueueAssembly (files, file))
                                return 1;
+
+               if (aot_compile)
+                       AotCompile (files);
+
                if (custom_mode)
                        GenerateBundles (files);
                else 
@@ -537,7 +569,6 @@ class MakeBundle {
                                        Console.WriteLine ("At {0:x} with input {1}", package.Position, fileStream.Length);
                                fileStream.CopyTo (package);
                                package.Position = package.Position + (align - (package.Position % align));
-
                                return (int) ret;
                        }
                }
@@ -546,7 +577,6 @@ class MakeBundle {
                {
                        var p = package.Position;
                        var size = AddFile (fname);
-                       
                        locations [entry] = Tuple.Create(p, size);
                }
 
@@ -719,6 +749,7 @@ void          mono_register_config_for_assembly (const char* assembly_name, cons
                        } else {
                                tc.WriteLine ("#include <mono/metadata/mono-config.h>");
                                tc.WriteLine ("#include <mono/metadata/assembly.h>\n");
+                               tc.WriteLine ("#include <mono/jit/jit.h>\n");
                        }
 
                        if (compress) {
@@ -859,6 +890,7 @@ void          mono_register_config_for_assembly (const char* assembly_name, cons
                        }
                        ts.Close ();
 
+                       // Managed assemblies baked in
                        if (compress)
                                tc.WriteLine ("\nstatic const CompressedAssembly *compressed [] = {");
                        else
@@ -868,6 +900,37 @@ void          mono_register_config_for_assembly (const char* assembly_name, cons
                                tc.WriteLine ("\t&{0},", c);
                        }
                        tc.WriteLine ("\tNULL\n};\n");
+
+
+                       // AOT baked in plus loader
+                       foreach (string asm in aot_names){
+                               tc.WriteLine ("\textern const void *mono_aot_module_{0}_info;", asm);
+                       }
+
+                       tc.WriteLine ("\nstatic void install_aot_modules (void) {\n");
+                       foreach (string asm in aot_names){
+                               tc.WriteLine ("\tmono_aot_register_module (mono_aot_module_{0}_info);\n", asm);
+                       }
+
+                       string enum_aot_mode;
+                       switch (aot_mode) {
+                       case "full": 
+                               enum_aot_mode = "MONO_AOT_MODE_FULL";
+                               break;
+                       case "llvmonly": 
+                               enum_aot_mode = "MONO_AOT_MODE_LLVMONLY";
+                               break;
+                       case "": 
+                               enum_aot_mode = "MONO_AOT_MODE_NORMAL";
+                               break;
+                       default:
+                               throw new Exception ("Unsupported AOT mode");
+                       }
+                       tc.WriteLine ("\tmono_jit_set_aot_mode ({0});", enum_aot_mode);
+
+                       tc.WriteLine ("\n}\n");
+
+
                        tc.WriteLine ("static char *image_name = \"{0}\";", prog);
 
                        if (ctor_func != null) {
@@ -1013,15 +1076,20 @@ void          mono_register_config_for_assembly (const char* assembly_name, cons
                                        debugging = "-ggdb";
                                if (static_link)
                                {
+                                       string platform_libs;
                                        string smonolib;
-                                       if (style == "osx")
+                                       if (style == "osx") {
                                                smonolib = "`pkg-config --variable=libdir mono-2`/libmono-2.0.a ";
-                                       else
+                                               platform_libs = "-liconv -framework Foundation ";
+                                       } else {
                                                smonolib = "-Wl,-Bstatic -lmono-2.0 -Wl,-Bdynamic ";
+                                               platform_libs = "";
+                                       }
+
                                        cmd = String.Format("{4} -o '{2}' -Wall `pkg-config --cflags mono-2` {0} {3} " +
-                                               "`pkg-config --libs-only-L mono-2` " + smonolib +
-                                               "`pkg-config --libs-only-l mono-2 | sed -e \"s/\\-lmono-2.0 //\"` {1}",
-                                               temp_c, temp_o, output, zlib, cc);
+                                               "`pkg-config --libs-only-L mono-2` {5} {6} " + platform_libs +
+                                               "`pkg-config --libs-only-l mono-2 | sed -e \"s/\\-lmono-2.0 //\"` {1} -g ",
+                                               temp_c, temp_o, output, zlib, cc, smonolib, String.Join (" ", aot_paths));
                                }
                                else
                                {
@@ -1110,6 +1178,14 @@ void          mono_register_config_for_assembly (const char* assembly_name, cons
                }
 
                if (error != null) {
+                       Console.Error.WriteLine ("Failure to load i18n assemblies, the following directories were searched for the assemblies:");
+                       foreach (var path in link_paths){
+                               Console.Error.WriteLine ("   Path: " + path);
+                       }
+                       if (custom_mode){
+                               Console.WriteLine ("In Custom mode, you need to provide the directory to lookup assemblies from using -L");
+                       }
+                       
                        Error ("Couldn't load one or more of the i18n assemblies: " + error);
                        Environment.Exit (1);
                }
@@ -1118,10 +1194,10 @@ void          mono_register_config_for_assembly (const char* assembly_name, cons
        
        static readonly Universe universe = new Universe ();
        static readonly Dictionary<string, string> loaded_assemblies = new Dictionary<string, string> ();
-       static readonly string resourcePathSeparator = (Path.DirectorySeparatorChar == '\\') ? $"\\{Path.DirectorySeparatorChar}" : $"{Path.DirectorySeparatorChar}";
 
        public static string GetAssemblyName (string path)
        {
+               string resourcePathSeparator = style == "windows" ? "\\\\" : "/";
                string name = Path.GetFileName (path);
 
                // A bit of a hack to support satellite assemblies. They all share the same name but
@@ -1161,9 +1237,18 @@ void          mono_register_config_for_assembly (const char* assembly_name, cons
                        return true;
                try {
                        Assembly a = universe.LoadFile (path);
+                       if (a == null) {
+                               Error ("Unable to to load assembly `{0}'", path);
+                               return false;
+                       }
 
                        foreach (AssemblyName an in a.GetReferencedAssemblies ()) {
                                a = LoadAssembly (an.Name);
+                               if (a == null) {
+                                       Error ("Unable to load assembly `{0}' referenced by `{1}'", an.Name, path);
+                                       return false;
+                               }
+
                                if (!QueueAssembly (files, a.CodeBase))
                                        return false;
                        }
@@ -1183,7 +1268,12 @@ void          mono_register_config_for_assembly (const char* assembly_name, cons
                Assembly a = null;
                
                try {
+                       if (!quiet)
+                               Console.WriteLine ("Attempting to load assembly: {0}", assembly);
                        a = universe.LoadFile (assembly);
+                       if (!quiet)
+                               Console.WriteLine ("Assembly {0} loaded successfully.", assembly);
+                       
                } catch (FileNotFoundException){
                        Error ($"Cannot find assembly `{assembly}'");
                } catch (IKVM.Reflection.BadImageFormatException f) {
@@ -1206,6 +1296,8 @@ void          mono_register_config_for_assembly (const char* assembly_name, cons
                string total_log = "";
                foreach (string dir in link_paths){
                        string full_path = Path.Combine (dir, assembly);
+                       if (!quiet)
+                               Console.WriteLine ("Attempting to load assembly from: " + full_path);
                        if (!assembly.EndsWith (".dll") && !assembly.EndsWith (".exe"))
                                full_path += ".dll";
                        
@@ -1307,6 +1399,42 @@ void          mono_register_config_for_assembly (const char* assembly_name, cons
                }
        }
 
+
+       static string EncodeAotSymbol (string symbol)
+       {
+               var sb = new StringBuilder ();
+               /* This mimics what the aot-compiler does */
+               foreach (var b in System.Text.Encoding.UTF8.GetBytes (symbol)) {
+                       char c = (char) b;
+                       if ((c >= '0' && c <= '9') ||
+                               (c >= 'a' && c <= 'z') ||
+                               (c >= 'A' && c <= 'Z')) {
+                               sb.Append (c);
+                               continue;
+                       }
+                       sb.Append ('_');
+               }
+               return sb.ToString ();
+       }
+
+       static void AotCompile (List<string> files)
+       {
+               if (aot_runtime == null)
+                       aot_runtime = runtime;
+
+               foreach (var fileName in files) {
+                       string path = LocateFile (new Uri (fileName).LocalPath);
+                       string outPath = String.Format ("{0}.aot_out", path);
+                       aot_paths.Add (outPath);
+                       var name = System.Reflection.Assembly.LoadFrom(path).GetName().Name;
+                       aot_names.Add (EncodeAotSymbol (name));
+                       var aot_mode_string = "";
+                       if (aot_mode != null)
+                               aot_mode_string = "," + aot_mode;
+                       Execute (String.Format ("{0} --aot={1},outfile={2}{3} {4}", aot_runtime, aot_args, outPath, aot_mode_string, path));
+               }
+       }
+
        static void Execute (string cmdLine)
        {
                if (IsUnix) {