WXL
4 天以前 2cc85c64f1c64a2dbaeae276a3e2ca8420de76b7
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
/*
    MIT License http://www.opensource.org/licenses/mit-license.php
    Author Tobias Koppers @sokra
*/
 
"use strict";
 
const t = require("@webassemblyjs/ast");
const { decode } = require("@webassemblyjs/wasm-parser");
const EnvironmentNotSupportAsyncWarning = require("../EnvironmentNotSupportAsyncWarning");
const Parser = require("../Parser");
const StaticExportsDependency = require("../dependencies/StaticExportsDependency");
const WebAssemblyImportDependency = require("../dependencies/WebAssemblyImportDependency");
 
/** @typedef {import("./AsyncWebAssemblyModulesPlugin").AsyncWasmModuleClass} AsyncWasmModule */
/** @typedef {import("../Module").BuildInfo} BuildInfo */
/** @typedef {import("../Module").BuildMeta} BuildMeta */
/** @typedef {import("../NormalModule")} NormalModule */
/** @typedef {import("../Parser").ParserState} ParserState */
/** @typedef {import("../Parser").PreparsedAst} PreparsedAst */
 
const WASM_HEADER = Buffer.from([0x00, 0x61, 0x73, 0x6d]);
 
const decoderOpts = {
    ignoreCodeSection: true,
    ignoreDataSection: true,
 
    // this will avoid having to lookup with identifiers in the ModuleContext
    ignoreCustomNameSection: true
};
 
class WebAssemblyParser extends Parser {
    /**
     * Parses the provided source and updates the parser state.
     * @param {string | Buffer | PreparsedAst} source the source to parse
     * @param {ParserState} state the parser state
     * @returns {ParserState} the parser state
     */
    parse(source, state) {
        if (!Buffer.isBuffer(source)) {
            throw new Error("WebAssemblyParser input must be a Buffer");
        }
 
        const buildMeta = /** @type {BuildMeta} */ (state.module.buildMeta);
        buildMeta.exportsType = "namespace";
        buildMeta.async = true;
 
        EnvironmentNotSupportAsyncWarning.check(
            state.module,
            state.compilation.runtimeTemplate,
            "asyncWebAssembly"
        );
 
        // flag it as async module
        const buildInfo = /** @type {BuildInfo} */ (state.module.buildInfo);
 
        buildInfo.strict = true;
 
        if (/** @type {AsyncWasmModule} */ (state.module).phase === "source") {
            // For source phase, only validate magic header
            if (source.length < 4 || !source.subarray(0, 4).equals(WASM_HEADER)) {
                throw new Error(
                    "Source phase imports require valid WebAssembly modules. Invalid magic header (expected \\0asm)."
                );
            }
 
            // Source phase exports the WebAssembly.Module as default
            state.module.addDependency(
                new StaticExportsDependency(["default"], false)
            );
 
            // Skip full parsing - no exports/imports needed for source phase
            return state;
        }
 
        // parse it
        const program = decode(source, decoderOpts);
        const module = program.body[0];
        /** @type {string[]} */
        const exports = [];
 
        t.traverse(module, {
            ModuleExport({ node }) {
                exports.push(node.name);
            },
 
            ModuleImport({ node }) {
                const dep = new WebAssemblyImportDependency(
                    node.module,
                    node.name,
                    node.descr,
                    false
                );
 
                state.module.addDependency(dep);
            }
        });
 
        state.module.addDependency(new StaticExportsDependency(exports, false));
 
        return state;
    }
}
 
module.exports = WebAssemblyParser;