porffor 0.16.0-fe07da0f4 → 0.17.0-9c7bd8098

package/compiler/cyclone.js

@@ -0,0 +1,535 @@
+// cyclone: wasm partial constant evaluator (it is fast and dangerous hence "cyclone")
+import { signedLEB128, ieee754_binary64, read_ieee754_binary64, read_signedLEB128 } from './encoding.js';
+import { Opcodes, Valtype } from './wasmSpec.js';
+
+const number = (n, valtype = valtypeBinary) => {
+  switch (valtype) {
+    case Valtype.i32: return [ Opcodes.i32_const, ...signedLEB128(n) ];
+    case Valtype.i64: return [ Opcodes.i64_const, ...signedLEB128(n) ];
+    case Valtype.f64: return [ Opcodes.f64_const, ...ieee754_binary64(n) ];
+  }
+};
+
+const f64ToI32Op = {
+  [Opcodes.f64_eq]: Opcodes.i32_eq,
+  [Opcodes.f64_ne]: Opcodes.i32_ne,
+  [Opcodes.f64_lt]: Opcodes.i32_lt_s,
+  [Opcodes.f64_le]: Opcodes.i32_le_s,
+  [Opcodes.f64_gt]: Opcodes.i32_gt_s,
+  [Opcodes.f64_ge]: Opcodes.i32_ge_s,
+  [Opcodes.f64_add]: Opcodes.i32_add,
+  [Opcodes.f64_sub]: Opcodes.i32_sub,
+  [Opcodes.f64_mul]: Opcodes.i32_mul,
+  [Opcodes.f64_div]: Opcodes.i32_div_s,
+};
+
+export default wasm => {
+  let stack = []; // """""stack"""""
+  for (let i = 0; i < wasm.length; i++) {
+    let op = wasm[i];
+    if (!op) continue;
+
+    // op = [ ...op.filter(x => x != null && x <= 0xff) ];
+    op = [ ...op ];
+    wasm[i] = op;
+
+    let opcode = op[0];
+    if (opcode === 0xfc) { // multibyte op
+      opcode = (opcode << 8) + op[1];
+    }
+
+    const push = val => {
+      stack.push({ val, op });
+    };
+
+    const pop = () => {
+      const popped = stack.pop();
+
+      // remove the op
+      wasm.splice(wasm.indexOf(popped.op), 1);
+      i--;
+
+      return popped.val;
+    };
+
+    const pop2 = () => [ pop(), pop() ];
+
+    const bool = v => v ? 1 : 0;
+
+    const replaceOp = newOp => {
+      op.splice(0, op.length, ...newOp);
+    };
+    const replaceVal = (val, valtype) => replaceOp(number(val, valtype));
+
+    const empty = () => {
+      stack = [];
+    };
+
+    switch (opcode) {
+      case Opcodes.i32_const: {
+        const n = read_signedLEB128(op.slice(1));
+        push(n);
+        break;
+      }
+      case Opcodes.f64_const: {
+        const n = read_ieee754_binary64(op.slice(1));
+        push(n);
+        break;
+      }
+
+      case Opcodes.i32_eqz: {
+        if (stack.length < 1) { empty(); break; };
+        const v = bool(pop() === 0);
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.i32_eq: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = bool(a === b);
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.i32_ne: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = bool(a !== b);
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.i32_lt_s: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = bool(a < b);
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.i32_le_s: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = bool(a <= b);
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.i32_gt_s: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = bool(a > b);
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.i32_ge_s: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = bool(a >= b);
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+
+      case Opcodes.i32_add: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = a + b;
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.i32_sub: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = a - b;
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.i32_mul: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = a * b;
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.i32_div_s: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = a / b;
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.i32_rem_s: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = a % b; // not rem but good enough
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+
+      case Opcodes.i32_and: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = a & b;
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.i32_or: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = a | b;
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.i32_xor: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = a ^ b;
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.i32_shl: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = a << b;
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.i32_shr_s: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = a >> b;
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.i32_shr_u: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = a >>> b;
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+
+      case Opcodes.f64_eq: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = bool(a === b);
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.f64_ne: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = bool(a !== b);
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.f64_lt: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = bool(a < b);
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.f64_le: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = bool(a <= b);
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.f64_gt: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = bool(a > b);
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+      case Opcodes.f64_ge: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = bool(a >= b);
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+
+      case Opcodes.f64_abs: {
+        if (stack.length < 1) { empty(); break; };
+        const v = Math.abs(pop());
+
+        replaceVal(v, Valtype.f64);
+        push(v);
+        break;
+      }
+      case Opcodes.f64_neg: {
+        if (stack.length < 1) { empty(); break; };
+        const v = -pop();
+
+        replaceVal(v, Valtype.f64);
+        push(v);
+        break;
+      }
+
+      case Opcodes.f64_ceil: {
+        if (stack.length < 1) { empty(); break; };
+        const v = Math.ceil(pop());
+
+        replaceVal(v, Valtype.f64);
+        push(v);
+        break;
+      }
+      case Opcodes.f64_floor: {
+        if (stack.length < 1) { empty(); break; };
+        const v = Math.floor(pop());
+
+        replaceVal(v, Valtype.f64);
+        push(v);
+        break;
+      }
+      case Opcodes.f64_trunc: {
+        if (stack.length < 1) { empty(); break; };
+        const v = Math.trunc(pop());
+
+        replaceVal(v, Valtype.f64);
+        push(v);
+        break;
+      }
+      case Opcodes.f64_nearest: {
+        if (stack.length < 1) { empty(); break; };
+        const v = Math.round(pop());
+
+        replaceVal(v, Valtype.f64);
+        push(v);
+        break;
+      }
+
+      case Opcodes.f64_sqrt: {
+        if (stack.length < 1) { empty(); break; };
+        const v = Math.sqrt(pop());
+
+        replaceVal(v, Valtype.f64);
+        push(v);
+        break;
+      }
+
+      case Opcodes.f64_add: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = a + b;
+
+        replaceVal(v, Valtype.f64);
+        push(v);
+        break;
+      }
+      case Opcodes.f64_sub: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = a - b;
+
+        replaceVal(v, Valtype.f64);
+        push(v);
+        break;
+      }
+      case Opcodes.f64_mul: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = a * b;
+
+        replaceVal(v, Valtype.f64);
+        push(v);
+        break;
+      }
+      case Opcodes.f64_div: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = a / b;
+
+        replaceVal(v, Valtype.f64);
+        push(v);
+        break;
+      }
+
+      case Opcodes.f64_min: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = Math.min(a, b);
+
+        replaceVal(v, Valtype.f64);
+        push(v);
+        break;
+      }
+      case Opcodes.f64_max: {
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = Math.max(a, b);
+
+        replaceVal(v, Valtype.f64);
+        push(v);
+        break;
+      }
+
+      case Opcodes.f64_copysign: { // ?
+        if (stack.length < 2) { empty(); break; };
+        const [ b, a ] = pop2();
+        const v = Math.abs(a) * (b > 0 ? 1 : -1);
+
+        replaceVal(v, Valtype.f64);
+        push(v);
+        break;
+      }
+
+      case Opcodes.f64_convert_i32_u:
+      case Opcodes.f64_convert_i32_s: {
+        if (stack.length < 1) { empty(); break; };
+        const v = pop();
+
+        replaceVal(v, Valtype.f64);
+        push(v);
+        break;
+      }
+
+      case 0xfc02: { // i32_trunc_sat_f64_s
+        if (stack.length < 1) { empty(); break; }
+        const v = pop();
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+
+      case 0xfc03: { // i32_trunc_sat_f64_u
+        if (stack.length < 1) { empty(); break; }
+        const v = pop();
+
+        replaceVal(v, Valtype.i32);
+        push(v);
+        break;
+      }
+
+      default: {
+        empty();
+        break;
+      }
+    }
+
+    // this does, eg:
+    // local.get 7 ;; $i (i32)
+    // f64.convert_i32_s
+    // f64.const 1
+    // f64.add
+    // i32.trunc_sat_f64_s <--
+    // local.set 7 ;; $i (i32)
+    // ->
+    // local.get 7 ;; $i (i32)
+    // i32.const 1
+    // i32.add
+    // local.set 7 ;; $i (i32)
+    if (i >= 2 &&
+      ((opcode >= 0xa0 && opcode <= 0xa3) || // main f64 math op
+      (opcode >= 0x61 && opcode <= 0x66)) // main f64 eq op
+    ) {
+      const o2 = wasm[i - 1][0];
+      if (o2 === Opcodes.f64_const) { // f64.const
+        const o3 = wasm[i - 2][0];
+        if (o3 === Opcodes.f64_convert_i32_s || o3 === Opcodes.f64_convert_i32_u) {
+          // remove now unneeded i32 -> f64 convert
+          wasm.splice(i - 2, 1);
+          i--;
+
+          // convert f64.const -> i32.const
+          const n = read_ieee754_binary64(wasm[i - 1].slice(1));
+          wasm.splice(i - 1, 1, number(n, Valtype.i32));
+
+          // convert math op from f64 to i32
+          wasm[i][0] = f64ToI32Op[wasm[i][0]];
+
+          const nextOp = wasm[i + 1];
+          if (nextOp && opcode >= 0xa0 && opcode <= 0xa3) {
+            if (nextOp[0] === 0xfc && (nextOp[1] === 0x02 || nextOp[1] === 0x03)) {
+              // remove optional unneeded f64 -> i32 convert after
+              wasm.splice(i + 1, 1);
+            } else {
+              // add now needed i32 -> f64 convert after
+              wasm.splice(i + 1, Opcodes.i32_trunc_sat_f64_s);
+            }
+          }
+        }
+      }
+    }
+
+    if ((opcode === 0xfc02 || opcode === 0xfc03) && i >= 3) { // i32.trunc_sat_f64_s/u
+      const o2 = wasm[i - 1][0];
+      if (
+        (o2 >= 0xa0 && o2 <= 0xa3) || // main f64 math op
+        (o2 >= 0x61 && o2 <= 0x66) // main f64 eq op
+      ) {
+        const o3 = wasm[i - 2][0];
+        if (o3 === Opcodes.f64_const) { // f64.const
+          const o4 = wasm[i - 3][0];
+          if (o4 === Opcodes.f64_convert_i32_s || o4 === Opcodes.f64_convert_i32_u) {
+            // remove now unneeded i32 -> f64 convert
+            wasm.splice(i - 3, 1);
+            i--;
+          } else {
+            // add now needed f64 -> i32 convert prior
+            wasm.splice(i - 2, 0, Opcodes.i32_trunc_sat_f64_s);
+            i++;
+          }
+
+          // convert f64.const -> i32.const
+          const n = read_ieee754_binary64(wasm[i - 2].slice(1));
+          wasm.splice(i - 2, 1, number(n, Valtype.i32));
+
+          // convert math op from f64 to i32
+          wasm[i - 1][0] = f64ToI32Op[wasm[i - 1][0]];
+
+          // remove this now unneeded f64 -> i32 convert
+          wasm.splice(i, 1);
+        }
+      }
+    }
+  }
+};

package/compiler/decompile.js

@@ -4,6 +4,8 @@ import { read_ieee754_binary64, read_signedLEB128, read_unsignedLEB128 } from '.
 const inv = (obj, keyMap = x => x) => Object.keys(obj).reduce((acc, x) => { acc[keyMap(obj[x])] = x; return acc; }, {});
 const invOpcodes = inv(Opcodes);
 const invValtype = inv(Valtype);
+globalThis.invOpcodes = invOpcodes;
+globalThis.invValtype = invValtype;
 
 export default (wasm, name = '', ind = 0, locals = {}, params = [], returns = [], funcs = [], globals = {}, exceptions = []) => {
   const invLocals = inv(locals, x => x.idx);
@@ -91,7 +93,7 @@ export default (wasm, name = '', ind = 0, locals = {}, params = [], returns = []
       if (callFunc) out += ` ;; $${callFunc.name} ${makeSignature(callFunc.params, callFunc.returns)}`;
       if (globalThis.importFuncs && inst[1] < importFuncs.length) {
         const importFunc = importFuncs[inst[1]];
-        out += ` ;; import ${importFunc.name} ${makeSignature(new Array(importFunc.params).fill(valtypeBinary), new Array(importFunc.returns).fill(valtypeBinary),)}`;
+        out += ` ;; import ${importFunc.name} ${makeSignature(typeof importFunc.params === 'object' ? importFunc.params : new Array(importFunc.params).fill(valtypeBinary), new Array(importFunc.returns).fill(valtypeBinary),)}`;
       }
     }