summaryrefslogtreecommitdiff
path: root/src/asm2wasm-main.cpp
blob: 7c4d9305248b17f4c4f9470ad265ec67d4056350 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
//
// asm2wasm console tool
//

#include "asm2wasm.h"

using namespace cashew;
using namespace wasm;

int main(int argc, char **argv) {
  debug = getenv("ASM2WASM_DEBUG") ? getenv("ASM2WASM_DEBUG")[0] - '0' : 0;

  char *infile = argv[1];

  if (debug) std::cerr << "loading '" << infile << "'...\n";
  FILE *f = fopen(argv[1], "r");
  assert(f);
  fseek(f, 0, SEEK_END);
  int size = ftell(f);
  char *input = new char[size+1];
  rewind(f);
  int num = fread(input, 1, size, f);
  // On Windows, ftell() gives the byte position (\r\n counts as two bytes), but when
  // reading, fread() returns the number of characters read (\r\n is read as one char \n, and counted as one),
  // so return value of fread can be less than size reported by ftell, and that is normal.
  assert((num > 0 || size == 0) && num <= size);
  fclose(f);
  input[num] = 0;

  // emcc --separate-asm modules look like
  //
  //    Module["asm"] = (function(global, env, buffer) {
  //      ..
  //    });
  //
  // we need to clean that up.
  if (*input == 'M') {
    while (*input != 'f') {
      input++;
      num--;
    }
    char *end = input + num - 1;
    while (*end != '}') {
      *end = 0;
      end--;
    }
  }

  if (debug) std::cerr << "parsing...\n";
  cashew::Parser<Ref, DotZeroValueBuilder> builder;
  Ref asmjs = builder.parseToplevel(input);

  if (debug) std::cerr << "wasming...\n";
  Module wasm;
  wasm.memory.initial = wasm.memory.max = 16*1024*1024; // we would normally receive this from the compiler
  Asm2WasmBuilder asm2wasm(wasm);
  asm2wasm.processAsm(asmjs);

  if (debug) std::cerr << "optimizing...\n";
  asm2wasm.optimize();

  if (debug) std::cerr << "printing...\n";
  std::cout << wasm;

  if (debug) std::cerr << "done.\n";
}