1 files changed, 104 insertions, 0 deletions
diff --git a/js/src/wasm/WasmBuiltinModule.h b/js/src/wasm/WasmBuiltinModule.h
new file mode 100644
index 0000000000..42faffec73
--- /dev/null
+++ b/js/src/wasm/WasmBuiltinModule.h
@@ -0,0 +1,104 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*-
+ * vim: set ts=8 sts=2 et sw=2 tw=80:
+ *
+ * Copyright 2021 Mozilla Foundation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#ifndef wasm_builtin_module_h
+#define wasm_builtin_module_h
+
+#include "mozilla/EnumeratedArray.h"
+#include "mozilla/Maybe.h"
+#include "mozilla/Span.h"
+
+#include "wasm/WasmBuiltins.h"
+#include "wasm/WasmCompileArgs.h"
+#include "wasm/WasmConstants.h"
+#include "wasm/WasmSerialize.h"
+#include "wasm/WasmTypeDecls.h"
+#include "wasm/WasmTypeDef.h"
+
+namespace js {
+namespace wasm {
+
+struct MOZ_STACK_CLASS BuiltinModuleInstances {
+  explicit BuiltinModuleInstances(JSContext* cx)
+      : selfTest(cx), intGemm(cx), jsString(cx) {}
+
+  Rooted<JSObject*> selfTest;
+  Rooted<JSObject*> intGemm;
+  Rooted<JSObject*> jsString;
+
+  MutableHandle<JSObject*> operator[](BuiltinModuleId module) {
+    switch (module) {
+      case BuiltinModuleId::SelfTest: {
+        return &selfTest;
+      }
+      case BuiltinModuleId::IntGemm: {
+        return &intGemm;
+      }
+      case BuiltinModuleId::JSString: {
+        return &jsString;
+      }
+      default: {
+        MOZ_CRASH();
+      }
+    }
+  }
+};
+
+// An builtin module func is a natively implemented function that may be
+// compiled into a 'builtin module', which may be instantiated with a provided
+// memory yielding an exported WebAssembly function wrapping the builtin module.
+struct BuiltinModuleFunc {
+  // The name of the func as it is exported
+  const char* exportName;
+  // The params taken by the func.
+  mozilla::Span<const ValType> params;
+  // The optional result returned by the func.
+  mozilla::Maybe<const ValType> result;
+  // The signature of the builtin that implements the func
+  const SymbolicAddressSignature& signature;
+  // Whether this function takes a pointer to the memory base as a hidden final
+  // parameter.
+  bool usesMemory;
+
+  // Allocate a FuncType for this func, returning false for OOM
+  bool funcType(FuncType* type) const;
+
+  // Get the BuiltinModuleFunc for an BuiltinModuleFuncId. BuiltinModuleFuncId
+  // must be validated.
+  static const BuiltinModuleFunc& getFromId(BuiltinModuleFuncId id);
+};
+
+Maybe<BuiltinModuleId> ImportMatchesBuiltinModule(
+    mozilla::Span<const char> importName, BuiltinModuleIds enabledBuiltins);
+Maybe<const BuiltinModuleFunc*> ImportMatchesBuiltinModuleFunc(
+    mozilla::Span<const char> importName, BuiltinModuleId module);
+
+// Compile and return the builtin module for a particular
+// builtin module.
+bool CompileBuiltinModule(JSContext* cx, BuiltinModuleId module,
+                          MutableHandle<WasmModuleObject*> result);
+
+// Compile, instantiate and return the builtin module instance for a particular
+// builtin module.
+bool InstantiateBuiltinModule(JSContext* cx, BuiltinModuleId module,
+                              MutableHandle<JSObject*> result);
+
+}  // namespace wasm
+}  // namespace js
+
+#endif  // wasm_builtin_module_h