wasm3/platforms/app/main.c

#include <stdio.h>
#include <stdlib.h>
#include <time.h>

#include "m3.h"
#include "m3_api_wasi.h"
#include "m3_env.h"

#define FATAL(msg, ...) { printf("Error: [Fatal] " msg "\n", ##__VA_ARGS__); goto _onfatal; }

int  main  (int i_argc, const char * i_argv [])
{
//    m3_PrintM3Info ();
    
    M3Result result = c_m3Err_none;

    IM3Runtime env = NULL;

    if (i_argc < 3) FATAL("not enough arguments");

    u8* wasm = NULL;
    u32 fsize = 0;

    //printf("Loading WebAssembly...\n");

    FILE* f = fopen (i_argv[1], "rb");
    if (f)
    {
        fseek (f, 0, SEEK_END);
        fsize = ftell(f);
        fseek (f, 0, SEEK_SET);

        if (fsize > 1000000) FATAL("file is too big");

        wasm = (u8*) malloc(fsize);
        fread (wasm, 1, fsize, f);
        fclose (f);
    } else {
    	FATAL("cannot open file");
    }

    IM3Module module;
    result = m3_ParseModule (& module, wasm, fsize);
    if (result) FATAL("m3_ParseModule: %s", result);

    // TODO: Detect stack exhaustion
    env = m3_NewRuntime (8*1024);
    if (!env) FATAL("m3_NewRuntime");

    result = m3_LoadModule (env, module);
    if (result) FATAL("m3_LoadModule: %s", result);

    result = m3_LinkWASI (module);
    if (result) FATAL("m3_LinkWASI: %s", result);

    IM3Function func;
    result = m3_FindFunction (& func, env, "__post_instantiate");

    if (not result)
    {
        result = m3_Call (func);
        if (result) FATAL("__post_instantiate failed: %s", result);
    }
    else m3_IgnoreErrorInfo (env);

    const char* fname = i_argv[2];

    result = m3_FindFunction (& func, env, fname);
    if (result) FATAL("m3_FindFunction: %s", result);

    //printf("Running...\n");

    if (!strcmp(fname, "_main") || !strcmp(fname, "main")) {
        i_argc -= 2;
        i_argv += 2;
        result = m3_CallMain (func, i_argc, i_argv);
        if (result) FATAL("m3_CallMain: %s", result);

        m3stack_t stack = (m3stack_t)(env->stack);
        return stack[0];
    } else {
        i_argc -= 3;
        i_argv += 3;
        result = m3_CallWithArgs (func, i_argc, i_argv);
    }

    m3_FreeRuntime (env);

    free (wasm);

_onfatal:
    if (result) {
        printf ("Error: %s", result);
        if (env)
        {
            M3ErrorInfo info = m3_GetErrorInfo (env);
            printf (" (%s)", info.message);
        }
    }

    printf ("\n");

    return 0;
}