codeowners update

[mono.git] / samples / embed / teste.c

diff --git a/samples/embed/teste.c b/samples/embed/teste.c
index ee7b84227d7990428ca328ee1508f5bb8ca317a9..6f727a00e1bb0a012ff04ae8ad912981715fbcac 100644 (file)
--- a/samples/embed/teste.c
+++ b/samples/embed/teste.c
@@ -1,9 +1,12 @@
 #include <mono/jit/jit.h>
+#include <mono/metadata/environment.h>
+#include <mono/utils/mono-publib.h>
+#include <stdlib.h>
 
 /*
  * Very simple mono embedding example.
  * Compile with: 
- *     gcc -o teste teste.c `pkg-config --cflags --libs mono` -lm
+ *     gcc -o teste teste.c `pkg-config --cflags --libs mono-2` -lm
  *     mcs test.cs
  * Run with:
  *     ./teste test.exe
@@ -14,18 +17,50 @@ gimme () {
        return mono_string_new (mono_domain_get (), "All your monos are belong to us!");
 }
 
+static void main_function (MonoDomain *domain, const char *file, int argc, char** argv)
+{
+       MonoAssembly *assembly;
+
+       assembly = mono_domain_assembly_open (domain, file);
+       if (!assembly)
+               exit (2);
+       /*
+        * mono_jit_exec() will run the Main() method in the assembly.
+        * The return value needs to be looked up from
+        * System.Environment.ExitCode.
+        */
+       mono_jit_exec (domain, assembly, argc, argv);
+}
+
+static int malloc_count = 0;
+
+static void* custom_malloc(size_t bytes)
+{
+       ++malloc_count;
+       return malloc(bytes);
+}
+
 int 
 main(int argc, char* argv[]) {
        MonoDomain *domain;
-       MonoAssembly *assembly;
        const char *file;
        int retval;
-
+       
        if (argc < 2){
-               fprintf (stderr, "Please provide an assembly to load");
+               fprintf (stderr, "Please provide an assembly to load\n");
                return 1;
        }
        file = argv [1];
+
+       MonoAllocatorVTable mem_vtable = {custom_malloc};
+       mono_set_allocator_vtable (&mem_vtable);
+
+       /*
+        * Load the default Mono configuration file, this is needed
+        * if you are planning on using the dllmaps defined on the
+        * system configuration
+        */
+       mono_config_parse (NULL);
        /*
         * mono_jit_init() creates a domain: each assembly is
         * loaded and run in a MonoDomain.
@@ -35,16 +70,16 @@ main(int argc, char* argv[]) {
         * We add our special internal call, so that C# code
         * can call us back.
         */
-       mono_add_internal_call ("Mono::gimme", gimme);
-       assembly = mono_domain_assembly_open (domain, file);
-       if (!assembly)
-               return 2;
-       /*
-        * mono_jit_exec() will run the Main() method in the assembly
-        * and return the value.
-        */
-       retval = mono_jit_exec (domain, assembly, argc - 1, argv + 1);
+       mono_add_internal_call ("MonoEmbed::gimme", gimme);
+
+       main_function (domain, file, argc - 1, argv + 1);
+       
+       retval = mono_environment_exitcode_get ();
+       
        mono_jit_cleanup (domain);
+
+       fprintf (stdout, "custom malloc calls = %d\n", malloc_count);
+
        return retval;
 }