module_instance.hpp

Go to the documentation of this file.

#ifndef KAGOME_CORE_RUNTIME_WAVM_IMPL_MODULE_INSTANCE_HPP
#define KAGOME_CORE_RUNTIME_WAVM_IMPL_MODULE_INSTANCE_HPP

#include "runtime/module_instance.hpp"

#include <string_view>

#include <WAVM/Runtime/Runtime.h>
#include <optional>

#include "log/logger.hpp"
#include "runtime/ptr_size.hpp"

namespace WAVM {
  namespace Runtime {
    struct Instance;
  }  // namespace Runtime
}  // namespace WAVM

namespace kagome::runtime::wavm {

  class CompartmentWrapper;

  class ModuleInstanceImpl
      : public ModuleInstance,
        public std::enable_shared_from_this<ModuleInstanceImpl> {
   public:
    enum class Error {
      FUNC_NOT_FOUND = 1,
      WRONG_ARG_COUNT,
      EXECUTION_ERROR,
      WRONG_RETURN_TYPE
    };

    ModuleInstanceImpl(
        InstanceEnvironment &&env,
        WAVM::Runtime::GCPointer<WAVM::Runtime::Instance> instance,
        WAVM::Runtime::ModuleRef module,
        std::shared_ptr<const CompartmentWrapper> compartment,
        const common::Hash256 &code_hash);

    const common::Hash256 &getCodeHash() const override {
      return code_hash_;
    }

    outcome::result<PtrSize> callExportFunction(
        std::string_view name, common::BufferView encoded_args) const override;

    outcome::result<std::optional<WasmValue>> getGlobal(
        std::string_view name) const override;

    void forDataSegment(DataSegmentProcessor const &callback) const override;

    InstanceEnvironment const &getEnvironment() const override;
    outcome::result<void> resetEnvironment() override;

   private:
    InstanceEnvironment env_;
    WAVM::Runtime::GCPointer<WAVM::Runtime::Instance> instance_;
    WAVM::Runtime::ModuleRef module_;
    std::shared_ptr<const CompartmentWrapper> compartment_;
    common::Hash256 code_hash_;
    log::Logger logger_;
  };

}  // namespace kagome::runtime::wavm

OUTCOME_HPP_DECLARE_ERROR(kagome::runtime::wavm, ModuleInstanceImpl::Error)

#endif  // KAGOME_CORE_RUNTIME_WAVM_IMPL_MODULE_INSTANCE_HPP