wasi.hh

Go to the documentation of this file.

 
#ifndef WASMTIME_WASI_HH
#define WASMTIME_WASI_HH
 
#include <memory>
#include <string>
#include <vector>
#include <wasi.h>
#include <wasmtime/conf.h>
#include <wasmtime/helpers.hh>
 
#ifdef WASMTIME_FEATURE_WASI
 
namespace wasmtime {
 
class WasiConfig {
  WASMTIME_OWN_WRAPPER(WasiConfig, wasi_config);
 
  WasiConfig() : ptr(wasi_config_new()) {}
 
  void argv(const std::vector<std::string> &args) {
    std::vector<const char *> ptrs;
    ptrs.reserve(args.size());
    for (const auto &arg : args) {
      ptrs.push_back(arg.c_str());
    }
 
    wasi_config_set_argv(ptr.get(), args.size(), ptrs.data());
  }
 
  void inherit_argv() { wasi_config_inherit_argv(ptr.get()); }
 
  void env(const std::vector<std::pair<std::string, std::string>> &env) {
    std::vector<const char *> names;
    std::vector<const char *> values;
    for (const auto &[name, value] : env) {
      names.push_back(name.c_str());
      values.push_back(value.c_str());
    }
    wasi_config_set_env(ptr.get(), env.size(), names.data(), values.data());
  }
 
  void inherit_env() { wasi_config_inherit_env(ptr.get()); }
 
  [[nodiscard]] bool stdin_file(const std::string &path) {
    return wasi_config_set_stdin_file(ptr.get(), path.c_str());
  }
 
  void inherit_stdin() { wasi_config_inherit_stdin(ptr.get()); }
 
  [[nodiscard]] bool stdout_file(const std::string &path) {
    return wasi_config_set_stdout_file(ptr.get(), path.c_str());
  }
 
  void inherit_stdout() { wasi_config_inherit_stdout(ptr.get()); }
 
  [[nodiscard]] bool stderr_file(const std::string &path) {
    return wasi_config_set_stderr_file(ptr.get(), path.c_str());
  }
 
  void inherit_stderr() { wasi_config_inherit_stderr(ptr.get()); }
 
  [[nodiscard]] bool preopen_dir(const std::string &path,
                                 const std::string &guest_path,
                                 size_t dir_perms, size_t file_perms) {
    return wasi_config_preopen_dir(ptr.get(), path.c_str(), guest_path.c_str(),
                                   dir_perms, file_perms);
  }
 
  void inherit_network() { wasi_config_inherit_network(ptr.get()); }
 
  void allow_ip_name_lookup(bool enable) {
    wasi_config_allow_ip_name_lookup(ptr.get(), enable);
  }
};
 
} // namespace wasmtime
 
#endif // WASMTIME_FEATURE_WASI
 
#endif // WASMTIME_WASI_HH