porffor 0.14.0-7bef6473d → 0.14.0-7f4307d22

package/compiler/wrap.js

@@ -1,22 +1,27 @@
+import { encodeVector, encodeLocal } from './encoding.js';
+import { importedFuncs } from './builtins.js';
 import compile from './index.js';
 import decompile from './decompile.js';
-import { encodeVector, encodeLocal } from './encoding.js';
 import { TYPES } from './types.js';
 import { log } from './log.js';
 import Prefs from './prefs.js';
 
 const bold = x => `\u001b[1m${x}\u001b[0m`;
 
-const porfToJSValue = (memory, funcs, value, type) => {
+const porfToJSValue = ({ memory, funcs, pages }, value, type) => {
   switch (type) {
     case TYPES.boolean: return Boolean(value);
     case TYPES.undefined: return undefined;
     case TYPES.object: return value === 0 ? null : {};
 
     case TYPES.function: {
-      // wasm func index, including all imports
-      const func = funcs.find(x => (x.originalIndex ?? x.index) === value);
-      // if (!func) return value;
+      let func;
+      if (value < 0) {
+        func = importedFuncs[value + importedFuncs.length];
+      } else {
+        func = funcs.find(x => ((x.originalIndex ?? x.index) - importedFuncs.length) === value);
+      }
+
       if (!func) return function () {};
 
       // make fake empty func for repl/etc
@@ -36,9 +41,22 @@ const porfToJSValue = (memory, funcs, value, type) => {
     case TYPES.array: {
       const length = (new Int32Array(memory.buffer, value, 1))[0];
 
-      // have to slice because of memory alignment (?)
-      const buf = memory.buffer.slice(value + 4, value + 4 + 8 * length);
-      return Array.from(new Float64Array(buf, 0, length));
+      const out = [];
+      for (let i = 0; i < length; i++) {
+        const offset = value + 4 + (i * 9);
+
+        // have to slice because of memory alignment (?)
+        const v = (new Float64Array(memory.buffer.slice(offset, offset + 8), 0, 1))[0];
+        const t = (new Uint8Array(memory.buffer, offset + 8, 1))[0];
+
+        // console.log(`reading value at index ${i}...`)
+        // console.log('  memory:', Array.from(new Uint8Array(memory.buffer, offset, 9)).map(x => x.toString(16).padStart(2, '0')).join(' '));
+        // console.log('  read:', { value: v, type: t }, '\n');
+
+        out.push(porfToJSValue({ memory, funcs, pages }, v, t));
+      }
+
+      return out;
     }
 
     case TYPES.date: {
@@ -61,12 +79,24 @@ const porfToJSValue = (memory, funcs, value, type) => {
         // console.log('  memory:', Array.from(new Uint8Array(memory.buffer, offset, 9)).map(x => x.toString(16).padStart(2, '0')).join(' '));
         // console.log('  read:', { value: v, type: t }, '\n');
 
-        out.add(porfToJSValue(memory, funcs, v, t));
+        out.add(porfToJSValue({ memory, funcs, pages }, v, t));
       }
 
       return out;
     }
 
+    case TYPES.symbol: {
+      const descStore = pages.get('bytestring: __Porffor_symbol_descStore/ptr').ind * pageSize;
+      const offset = descStore + 4 + ((value - 1) * 9);
+
+      const v = (new Float64Array(memory.buffer.slice(offset, offset + 8), 0, 1))[0];
+      const t = (new Uint8Array(memory.buffer, offset + 8, 1))[0];
+
+      const desc = porfToJSValue({ memory, funcs, pages }, v, t);
+
+      return Symbol(desc);
+    }
+
     default: return value;
   }
 };
@@ -86,40 +116,12 @@ export default async (source, flags = [ 'module' ], customImports = {}, print =
   times.push(performance.now() - t1);
   if (Prefs.profileCompiler) console.log(bold(`compiled in ${times[0].toFixed(2)}ms`));
 
-  const t2 = performance.now();
-
-  let instance;
-  try {
-    let wasmEngine = WebAssembly;
-    if (Prefs.asur) {
-      log.warning('wrap', 'using our !experimental! asur wasm engine instead of host to run');
-      wasmEngine = await import('../asur/index.js');
-    }
-
-    0, { instance } = await wasmEngine.instantiate(wasm, {
-      '': {
-        p: valtype === 'i64' ? i => print(Number(i).toString()) : i => print(i.toString()),
-        c: valtype === 'i64' ? i => print(String.fromCharCode(Number(i))) : i => print(String.fromCharCode(i)),
-        t: () => performance.now(),
-        u: () => performance.timeOrigin,
-        y: () => {},
-        z: () => {},
-        ...customImports
-      }
-    });
-  } catch (e) {
-    // only backtrace for runner, not test262/etc
-    if (!process.argv[1].includes('/runner')) throw e;
-
-    const funcInd = parseInt(e.message.match(/function #([0-9]+) /)?.[1]);
-    const blobOffset = parseInt(e.message.split('@')?.[1]);
-
-    if (!funcInd) throw e;
+  const backtrace = (funcInd, blobOffset) => {
+    if (funcInd == null || blobOffset == null) return false;
 
     // convert blob offset -> function wasm offset.
     // this is not good code and is somewhat duplicated
     // I just want it to work for debugging, I don't care about perf/yes
-
     const func = funcs.find(x => x.index === funcInd);
     const locals = Object.values(func.locals).sort((a, b) => a.idx - b.idx).slice(func.params.length).sort((a, b) => a.idx - b.idx);
 
@@ -137,7 +139,7 @@ export default async (source, flags = [ 'module' ], customImports = {}, print =
 
     if (typeCount !== 0) localDecl.push(encodeLocal(typeCount, lastType));
 
-    const toFind = encodeVector(localDecl).concat(func.wasm.flat().filter(x => x != null && x <= 0xff).slice(0, 40));
+    const toFind = encodeVector(localDecl).concat(func.wasm.flat().filter(x => x != null && x <= 0xff).slice(0, 60));
 
     let i = 0;
     for (; i < wasm.length; i++) {
@@ -152,7 +154,7 @@ export default async (source, flags = [ 'module' ], customImports = {}, print =
       if (!mismatch) break;
     }
 
-    if (i === wasm.length) throw e;
+    if (i === wasm.length) return false;
 
     const offset = (blobOffset - i) + encodeVector(localDecl).length;
 
@@ -163,7 +165,7 @@ export default async (source, flags = [ 'module' ], customImports = {}, print =
       if (cumLen === offset) break;
     }
 
-    if (cumLen !== offset) throw e;
+    if (cumLen !== offset) return false;
 
     i -= 1;
 
@@ -188,6 +190,38 @@ export default async (source, flags = [ 'module' ], customImports = {}, print =
     console.log(decomp.join('\n'));
     console.log('\x1B[90m...\x1B[0m\n');
 
+    return true;
+  };
+
+  const t2 = performance.now();
+
+  let instance;
+  try {
+    let wasmEngine = WebAssembly;
+    if (Prefs.asur) {
+      log.warning('wrap', 'using our !experimental! asur wasm engine instead of host to run');
+      wasmEngine = await import('../asur/index.js');
+    }
+
+    0, { instance } = await wasmEngine.instantiate(wasm, {
+      '': {
+        p: valtype === 'i64' ? i => print(Number(i).toString()) : i => print(i.toString()),
+        c: valtype === 'i64' ? i => print(String.fromCharCode(Number(i))) : i => print(String.fromCharCode(i)),
+        t: () => performance.now(),
+        u: () => performance.timeOrigin,
+        y: () => {},
+        z: () => {},
+        ...customImports
+      }
+    });
+  } catch (e) {
+    // only backtrace for runner, not test262/etc
+    if (!process.argv[1].includes('/runner')) throw e;
+
+    const funcInd = parseInt(e.message.match(/function #([0-9]+) /)?.[1]);
+    const blobOffset = parseInt(e.message.split('@')?.[1]);
+
+    backtrace(funcInd, blobOffset);
     throw e;
   }
 
@@ -195,6 +229,7 @@ export default async (source, flags = [ 'module' ], customImports = {}, print =
   if (Prefs.profileCompiler) console.log(`instantiated in ${times[1].toFixed(2)}ms`);
 
   const exports = {};
+  const rawValues = process.argv.includes('-i');
 
   const exceptTag = instance.exports['0'], memory = instance.exports['$'];
   for (const x in instance.exports) {
@@ -213,10 +248,11 @@ export default async (source, flags = [ 'module' ], customImports = {}, print =
     exports[func.name] = function() {
       try {
         const ret = exp.apply(this, arguments);
-
         if (ret == null) return undefined;
 
-        return porfToJSValue(memory, funcs, ret[0], ret[1])
+        if (rawValues) return { value: ret[0], type: ret[1], js: porfToJSValue({ memory, funcs, pages }, ret[0], ret[1]) };
+
+        return porfToJSValue({ memory, funcs, pages }, ret[0], ret[1]);
       } catch (e) {
         if (e.is && e.is(exceptTag)) {
           const exceptId = e.getArg(exceptTag, 0);
@@ -231,6 +267,17 @@ export default async (source, flags = [ 'module' ], customImports = {}, print =
           throw new constructor(exception.message);
         }
 
+        if (e instanceof WebAssembly.RuntimeError) {
+          // only backtrace for runner, not test262/etc
+          if (!process.argv[1].includes('/runner')) throw e;
+
+          const match = e.stack.match(/wasm-function\[([0-9]+)\]:([0-9a-z]+)/) ?? [];
+          const funcInd = parseInt(match[1]);
+          const blobOffset = parseInt(match[2]);
+
+          backtrace(funcInd, blobOffset);
+        }
+
         throw e;
       }
     };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "porffor",
   "description": "a basic experimental wip aot optimizing js -> wasm engine/compiler/runtime in js",
-  "version": "0.14.0-7bef6473d",
+  "version": "0.14.0-7f4307d22",
   "author": "CanadaHonk",
   "license": "MIT",
   "scripts": {

package/runner/index.js

@@ -25,7 +25,7 @@ if (process.argv.includes('--help')) {
   console.log(`Usage: \x1B[1mporf [command] path/to/script.js [...prefs] [...args]\x1B[0m`);
 
   // commands
-  console.log(`\n\u001b[4mCommands\x1B[0m`);
+  console.log(`\n\x1B[1mCommands:\x1B[0m`);
   for (const [ cmd, [ color, desc ] ] of Object.entries({
     run: [ 34, 'Run a JS file' ],
     wasm: [ 34, 'Compile a JS file to a Wasm binary\n' ],
@@ -60,8 +60,10 @@ if (process.argv.includes('--help')) {
 
 let file = process.argv.slice(2).find(x => x[0] !== '-');
 if (['run', 'wasm', 'native', 'c', 'profile', 'debug', 'debug-wasm'].includes(file)) {
+  // remove this arg
+  process.argv.splice(process.argv.indexOf(file), 1);
+
   if (file === 'profile') {
-    process.argv.splice(process.argv.indexOf(file), 1);
     await import('./profiler.js');
     await new Promise(() => {}); // do nothing for the rest of this file
   }
@@ -75,12 +77,11 @@ if (['run', 'wasm', 'native', 'c', 'profile', 'debug', 'debug-wasm'].includes(fi
   }
 
   if (file === 'debug') {
-    process.argv.splice(process.argv.indexOf(file), 1);
     await import('./debug.js');
     await new Promise(() => {}); // do nothing for the rest of this file
   }
 
-  file = process.argv.slice(process.argv.indexOf(file) + 1).find(x => x[0] !== '-');
+  file = process.argv.slice(2).find(x => x[0] !== '-');
 
   const nonOptOutFile = process.argv.slice(process.argv.indexOf(file) + 1).find(x => x[0] !== '-');
   if (nonOptOutFile) {

package/runner/repl.js

@@ -1,6 +1,11 @@
+import { TYPE_NAMES } from '../compiler/types.js';
 import compile from '../compiler/wrap.js';
 import version from './version.js';
 
+import util from 'node:util';
+
+process.argv.push('--no-opt-unused');
+
 let repl;
 try {
   // try importing node:repl
@@ -86,10 +91,21 @@ const run = async (source, _context, _filename, callback, run = true) => {
     lastPages = [...pages.keys()];
   }
 
-  const ret = run ? exports.main() : undefined;
-  callback(null, ret);
+  let ret = run ? exports.main() : undefined;
+  let value, type;
+  if (ret?.type != null) {
+    value = ret.value;
+    type = ret.type;
+    ret = ret.js;
+  }
+
+  console.log(util.inspect(ret, false, 2, true), (value != null ? `\x1B[34m\x1B[3m(value: ${value}, type: ${TYPE_NAMES[type]})\x1B[0m` : ''));
+
+  // callback(null, ret);
 
   prev = prev + ';\n' + source.trim();
+
+  callback();
 };
 
 const replServer = repl.start({ prompt: '> ', eval: run });