1 files changed, 66 insertions, 0 deletions
diff --git a/extensions/spellcheck/hunspell/glue/RLBoxHunspell.h b/extensions/spellcheck/hunspell/glue/RLBoxHunspell.h
new file mode 100644
index 0000000000..2e5a11d936
--- /dev/null
+++ b/extensions/spellcheck/hunspell/glue/RLBoxHunspell.h
@@ -0,0 +1,66 @@
+/* -*- Mode: C++; tab-width: 20; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#ifndef EXTENSIONS_SPELLCHECK_HUNSPELL_GLUE_RLBOXHUNSPELL_H_
+#define EXTENSIONS_SPELLCHECK_HUNSPELL_GLUE_RLBOXHUNSPELL_H_
+
+#include "RLBoxHunspellTypes.h"
+
+// Load general firefox configuration of RLBox
+#include "mozilla/rlbox/rlbox_config.h"
+
+#ifdef MOZ_WASM_SANDBOXING_HUNSPELL
+// Include the generated header file so that we are able to resolve the symbols
+// in the wasm binary
+#  include "rlbox.wasm.h"
+#  define RLBOX_USE_STATIC_CALLS() rlbox_wasm2c_sandbox_lookup_symbol
+#  include "mozilla/rlbox/rlbox_wasm2c_sandbox.hpp"
+#else
+#  define RLBOX_USE_STATIC_CALLS() rlbox_noop_sandbox_lookup_symbol
+#  include "mozilla/rlbox/rlbox_noop_sandbox.hpp"
+#endif
+
+#include "mozilla/rlbox/rlbox.hpp"
+
+#include <hunspell.h>
+#include "mozHunspellRLBoxGlue.h"
+
+class RLBoxHunspell {
+ public:
+  static RLBoxHunspell* Create(const nsCString& affpath,
+                               const nsCString& dpath);
+
+  ~RLBoxHunspell();
+
+  int spell(const std::string& stdWord);
+  const std::string& get_dict_encoding() const;
+
+  std::vector<std::string> suggest(const std::string& word);
+
+ private:
+  struct RLBoxDeleter {
+    void operator()(rlbox_sandbox_hunspell* sandbox) {
+      sandbox->destroy_sandbox();
+      delete sandbox;
+    }
+  };
+
+  RLBoxHunspell(
+      mozilla::UniquePtr<rlbox_sandbox_hunspell, RLBoxDeleter> aSandbox,
+      const nsCString& affpath, const nsCString& dpath);
+
+  mozilla::UniquePtr<rlbox_sandbox_hunspell, RLBoxDeleter> mSandbox;
+  sandbox_callback_hunspell<hunspell_create_filemgr_t*> mCreateFilemgr;
+  sandbox_callback_hunspell<hunspell_get_line_t*> mGetLine;
+  sandbox_callback_hunspell<hunspell_get_line_num_t*> mGetLineNum;
+  sandbox_callback_hunspell<hunspell_destruct_filemgr_t*> mDestructFilemgr;
+  sandbox_callback_hunspell<hunspell_ToUpperCase_t*> mHunspellToUpperCase;
+  sandbox_callback_hunspell<hunspell_ToLowerCase_t*> mHunspellToLowerCase;
+  sandbox_callback_hunspell<hunspell_get_current_cs_t*> mHunspellGetCurrentCS;
+  tainted_hunspell<Hunhandle*> mHandle;
+  std::string mDicEncoding;
+};
+
+#endif