forked from thebaselab/codeapp
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
c: Run WASM binaries in Web worker thebaselab#612
- Loading branch information
1 parent
358e1b5
commit e9c6152
Showing
8 changed files
with
268 additions
and
7 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,59 @@ | ||
| // | ||
| // WASMService.swift | ||
| // Code | ||
| // | ||
| // Created by Ken Chung on 14/03/2024. | ||
| // | ||
|
|
||
| import GCDWebServers | ||
|
|
||
| class WASMService { | ||
| static let PORT = 20233 | ||
| private let webServer = GCDWebServer() | ||
|
|
||
| init() { | ||
| let basePath = "/" | ||
| let directoryPath = Resources.wasmHTML.deletingLastPathComponent().path + "/" | ||
| webServer.addHandler( | ||
| match: { requestMethod, requestURL, requestHeaders, urlPath, urlQuery in | ||
| if requestMethod != "GET" { | ||
| return nil | ||
| } | ||
| if !urlPath.hasPrefix(basePath) { | ||
| return nil | ||
| } | ||
| return GCDWebServerRequest( | ||
| method: requestMethod, url: requestURL, headers: requestHeaders, path: urlPath, | ||
| query: urlQuery) | ||
| }, | ||
| processBlock: { request in | ||
| let filePath = | ||
| directoryPath | ||
| + GCDWebServerNormalizePath(String(request.path.dropFirst(basePath.count))) | ||
| var isDirectory: ObjCBool = false | ||
| guard FileManager.default.fileExists(atPath: filePath, isDirectory: &isDirectory), | ||
| !isDirectory.boolValue | ||
| else { | ||
| return GCDWebServerResponse( | ||
| statusCode: GCDWebServerClientErrorHTTPStatusCode.httpStatusCode_NotFound | ||
| .rawValue) | ||
| } | ||
| let response = GCDWebServerFileResponse( | ||
| file: filePath, byteRange: request.byteRange) | ||
| response?.setValue("bytes", forAdditionalHeader: "Accept-Ranges") | ||
| response?.setValue("same-origin", forAdditionalHeader: "Cross-Origin-Opener-Policy") | ||
| response?.setValue( | ||
| "require-corp", forAdditionalHeader: "Cross-Origin-Embedder-Policy") | ||
| response?.setValue( | ||
| "same-origin", forAdditionalHeader: "Cross-Origin-Resource-Policy") | ||
| response?.cacheControlMaxAge = 10 | ||
| return response | ||
| }) | ||
|
|
||
| try? webServer.start(options: [ | ||
| GCDWebServerOption_AutomaticallySuspendInBackground: true, | ||
| GCDWebServerOption_BindToLocalhost: true, | ||
| GCDWebServerOption_Port: WASMService.PORT, | ||
| ]) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,46 @@ | ||
| <!DOCTYPE html> | ||
| <html> | ||
| <head> | ||
| <title>Code - WebAssembly Execution</title> | ||
| </head> | ||
| <body> | ||
| <script> | ||
| var messageHandler = (e) => { | ||
| const action = e.data[0]; | ||
|
|
||
| switch (action) { | ||
| case "prompt": | ||
| const int32 = new Int32Array(e.data[2]);; | ||
| const str = prompt(e.data[1]); | ||
| for (var i=0, strLen=str.length; i<strLen; i++) { | ||
| Atomics.store(int32, i, str.charCodeAt(i)); | ||
| } | ||
| Atomics.notify(int32, 0, 1); | ||
| break; | ||
| case "print": | ||
| window.webkit.messageHandlers.aShell.postMessage('print:' + e.data[1]); | ||
| break; | ||
| case "println": | ||
| window.webkit.messageHandlers.aShell.postMessage('print:' + e.data[1] + "\n"); | ||
| break; | ||
| case "print_error": | ||
| window.webkit.messageHandlers.aShell.postMessage('print_error:' + e.data[1]); | ||
| break; | ||
| default: | ||
| console.error(`${action} not hanlded`); | ||
| } | ||
| }; | ||
|
|
||
| async function executeWebAssembly(bufferString, args, cwd, tty, env) { | ||
| const worker = new Worker("worker.js"); | ||
| worker.onmessage = messageHandler; | ||
| const sab = new SharedArrayBuffer(Int32Array.BYTES_PER_ELEMENT * 8192); | ||
| const int32 = new Int32Array(sab); | ||
| worker.postMessage([bufferString, args, cwd, tty, env, sab]); | ||
| await Atomics.waitAsync(int32, 0, 0, 120000).value; | ||
| worker.terminate(); | ||
| return int32[0]; | ||
| } | ||
| </script> | ||
| </body> | ||
| </html> |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,139 @@ | ||
| importScripts("require.js"); | ||
|
|
||
| // make the "require" function available to all | ||
| Tarp.require({expose: true}); | ||
| // Have a global variable: | ||
| var window = {}; | ||
| var global = {}; | ||
| window.global = window; | ||
| // and a Buffer variable | ||
| var Buffer = require('buffer').Buffer; | ||
| var process = require('process'); | ||
|
|
||
| // Functions to deal with WebAssembly: | ||
| // These should load a wasm program: http://andrewsweeney.net/post/llvm-to-wasm/ | ||
| /* Array of bytes to base64 string decoding */ | ||
| // Modules for @wasmer: | ||
| const WASI = require('@wasmer/wasi/lib').WASI; | ||
| const browserBindings = require('@wasmer/wasi/lib/bindings/browser').default; | ||
| const WasmFs = require('@wasmer/wasmfs').WasmFs; | ||
| const lowerI64Imports = require("@wasmer/wasm-transformer").lowerI64Imports | ||
|
|
||
| function b64ToUint6 (nChr) { | ||
|
|
||
| return nChr > 64 && nChr < 91 ? | ||
| nChr - 65 | ||
| : nChr > 96 && nChr < 123 ? | ||
| nChr - 71 | ||
| : nChr > 47 && nChr < 58 ? | ||
| nChr + 4 | ||
| : nChr === 43 ? | ||
| 62 | ||
| : nChr === 47 ? | ||
| 63 | ||
| : | ||
| 0; | ||
|
|
||
| } | ||
|
|
||
| function base64DecToArr (sBase64, nBlockSize) { | ||
| var | ||
| sB64Enc = sBase64.replace(/[^A-Za-z0-9\+\/]/g, ""), nInLen = sB64Enc.length, | ||
| nOutLen = nBlockSize ? Math.ceil((nInLen * 3 + 1 >>> 2) / nBlockSize) * nBlockSize : nInLen * 3 + 1 >>> 2, aBytes = new Uint8Array(nOutLen); | ||
|
|
||
| for (var nMod3, nMod4, nUint24 = 0, nOutIdx = 0, nInIdx = 0; nInIdx < nInLen; nInIdx++) { | ||
| nMod4 = nInIdx & 3; | ||
| nUint24 |= b64ToUint6(sB64Enc.charCodeAt(nInIdx)) << 18 - 6 * nMod4; | ||
| if (nMod4 === 3 || nInLen - nInIdx === 1) { | ||
| for (nMod3 = 0; nMod3 < 3 && nOutIdx < nOutLen; nMod3++, nOutIdx++) { | ||
| aBytes[nOutIdx] = nUint24 >>> (16 >>> nMod3 & 24) & 255; | ||
| } | ||
| nUint24 = 0; | ||
| } | ||
| } | ||
| return aBytes; | ||
| } | ||
|
|
||
| // bufferString: program in base64 format | ||
| // args: arguments (argv[argc]) | ||
| // stdinBuffer: standard input | ||
| // cwd: current working directory | ||
| function executeWebAssembly(bufferString, args, cwd, tty, env) { | ||
| // Input: base64 encoded binary wasm file | ||
| // if (!('WebAssembly' in window)) { | ||
| // window.webkit.messageHandlers.aShell.postMessage('WebAssembly not supported'); | ||
| // return; | ||
| // } | ||
|
|
||
| var arrayBuffer = base64DecToArr(bufferString); | ||
| const loweredWasmBytes = lowerI64Imports(arrayBuffer); | ||
| var errorMessage = ''; | ||
| var errorCode = 0; | ||
| // window.standardInput = ''; | ||
| // TODO: link with other libraries/frameworks? impossible, I guess. | ||
| // TODO: keyboard input (directly from onkeypress) | ||
| try { | ||
| const wasmFs = new WasmFs(); // local file system. Used less often. | ||
| let wasi = new WASI({ | ||
| preopens: {'.': cwd, '/': '/'}, | ||
| args: args, | ||
| env: env, | ||
| bindings: { | ||
| ...browserBindings, | ||
| fs: wasmFs.fs, | ||
| } | ||
| }) | ||
| wasi.args = args | ||
| if (tty != 1) { | ||
| wasi.bindings.isTTY = (fd) => false; | ||
| } | ||
| const module = new WebAssembly.Module(loweredWasmBytes); | ||
| const instance = new WebAssembly.Instance(module, wasi.getImports(module)); | ||
| wasi.start(instance); | ||
| } | ||
| catch (error) { | ||
| // window.webkit.messageHandlers.aShell.postMessage('WebAssembly error: ' + error); | ||
| // WASI returns an error even in some cases where things went well. | ||
| // We find the type of the error, and return the appropriate error message | ||
| if (error.code === 'undefined') { | ||
| errorCode = 1; | ||
| errorMessage = '\nwasm: ' + error + '\n'; | ||
| } else if (error.code != null) { | ||
| // Numerical error code. Send the return code back to Swift. | ||
| errorCode = error.code; | ||
| } else { | ||
| errorCode = 1; | ||
| } | ||
| } | ||
| return [errorCode, errorMessage]; | ||
| } | ||
|
|
||
| println = (m) => { | ||
| postMessage(["println", m]); | ||
| }; | ||
|
|
||
| print_error = (m) => { | ||
| postMessage(["print_error", m]); | ||
| }; | ||
|
|
||
| prompt = (p) => { | ||
| const sab = new SharedArrayBuffer(Int32Array.BYTES_PER_ELEMENT * 8192); | ||
| const int32 = new Int32Array(sab); | ||
| postMessage(["prompt", p, sab]); | ||
| Atomics.wait(int32, 0, 0); | ||
| const firstZeroIndex = int32.indexOf(0); | ||
| return String.fromCharCode.apply(null, int32.subarray(0, firstZeroIndex)); | ||
| } | ||
|
|
||
| onmessage = (e) => { | ||
| const int32 = new Int32Array(e.data[5]); | ||
|
|
||
| executeWebAssembly(e.data[0], e.data[1], e.data[2], e.data[3], e.data[4]); | ||
|
|
||
| // Finishing execution | ||
| Atomics.store(int32, 0, 1); | ||
| Atomics.notify(int32, 0, 1); | ||
| }; | ||
|
|
||
| console.log = println; | ||
| console.error = print_error; |