mirror of
https://github.com/extism/extism.git
synced 2026-01-15 16:57:57 -05:00
0f8954c203
EIP: https://github.com/extism/proposals/pull/8 This PR makes minor breaking changes to several SDKs, but not to runtime C API. The threadsafety updates in the Rust SDK are kind of specific to Rust, I'm not sure if it makes sense to add the locks to all the other SDKs at this point. For the most part the `Context` and `Plugin` types in the SDKs should be safe to use protected by a mutex but they aren't inherently threadsafe. That kind of locking should probably be done by the user. - Runtime - improve thread safety - reinstantiates less - fixes a potential resource exhaustion bug from re-instantiating using the same store too many times - Rust SDK - adds `Send` and `Sync` implementations for `Context` - adds test sharing a context between threads - adds `Plugin::call_map` to call a plugin and handle the output with the lock held - adds testing sharing an `Arc<Mutex<Plugin>>` between threads - adds `Plugin::create` and `Plugin::create_from_manifest` to create a plugin without a `Context` - Python - BREAKING - changes `Plugin` constructor to take `context` as an optional named argument, to update use `Plugin(data, context=context)` instead - Ruby - BREAKING - changes `Plugin` constructor to take `context` as an optional named argument, to update use `Plugin.new(data, context=context)` instead - Go - adds `NewPlugin` and `NewPluginFromManifest` functions - Node - BREAKING - changes `Plugin` constructor to take `context` as an optional named argument, to update use `new Plugin(data, wasi, config, host, context)` instead of `new Plugin(context, data, wasi, functions, config)` (most people are probably using `context.plugin` instead of the Plugin constructor anyway) - OCaml - BREAKING - changes `Plugin.create` and `Plugin.of_manifest` to take `context` as an optional named argument, to update `Plugin.create ~context data` and `Plugin.of_manifest ~context data` instead - Haskell - adds `createPlugin` and `createPluginFromManifest` functions - Elixir - adds `Plugin.new` to make a plugin without going through `Context.new_plugin` - Java - adds new `Plugin` constructors without a `Context` argument - C++ - BREAKING - Updates `Plugin` constructor to take an optional context as the last argument, instead of requiring it to be the first argument - Use `Plugin(wasm, wasi, functions, ctx)` instead of `Plugin(ctx, wasm, wasi, functions)` - Zig - Adds `Plugin.create` and `Plugin.createWithManifest` to create plugins in their own context. --------- Co-authored-by: zach <zach@dylib.so> Co-authored-by: Benjamin Eckel <bhelx@simst.im>
93 lines
2.4 KiB
C++
93 lines
2.4 KiB
C++
#include "../extism.hpp"
|
|
|
|
#include <fstream>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
std::vector<uint8_t> read(const char *filename) {
|
|
std::ifstream file(filename, std::ios::binary);
|
|
return std::vector<uint8_t>((std::istreambuf_iterator<char>(file)),
|
|
std::istreambuf_iterator<char>());
|
|
}
|
|
|
|
const std::string code = "../../wasm/code.wasm";
|
|
|
|
namespace {
|
|
using namespace extism;
|
|
|
|
TEST(Context, Basic) {
|
|
Context context;
|
|
ASSERT_NE(context.pointer, nullptr);
|
|
}
|
|
|
|
TEST(Plugin, Manifest) {
|
|
Manifest manifest = Manifest::path(code);
|
|
manifest.set_config("a", "1");
|
|
|
|
ASSERT_NO_THROW(Plugin plugin(manifest));
|
|
Plugin plugin(manifest);
|
|
|
|
Buffer buf = plugin.call("count_vowels", "this is a test");
|
|
ASSERT_EQ((std::string)buf, "{\"count\": 4}");
|
|
}
|
|
|
|
TEST(Plugin, BadManifest) {
|
|
Manifest manifest;
|
|
ASSERT_THROW(Plugin plugin(manifest), Error);
|
|
}
|
|
|
|
TEST(Plugin, Bytes) {
|
|
Context context;
|
|
auto wasm = read(code.c_str());
|
|
ASSERT_NO_THROW(Plugin plugin = context.plugin(wasm));
|
|
Plugin plugin = context.plugin(wasm);
|
|
|
|
Buffer buf = plugin.call("count_vowels", "this is another test");
|
|
ASSERT_EQ(buf.string(), "{\"count\": 6}");
|
|
}
|
|
|
|
TEST(Plugin, UpdateConfig) {
|
|
Context context;
|
|
auto wasm = read(code.c_str());
|
|
Plugin plugin = context.plugin(wasm);
|
|
|
|
Config config;
|
|
config["abc"] = "123";
|
|
ASSERT_NO_THROW(plugin.config(config));
|
|
}
|
|
|
|
TEST(Plugin, FunctionExists) {
|
|
Context context;
|
|
auto wasm = read(code.c_str());
|
|
Plugin plugin = context.plugin(wasm);
|
|
|
|
ASSERT_FALSE(plugin.function_exists("bad_function"));
|
|
ASSERT_TRUE(plugin.function_exists("count_vowels"));
|
|
}
|
|
|
|
TEST(Plugin, HostFunction) {
|
|
auto wasm = read("../../wasm/code-functions.wasm");
|
|
auto t = std::vector<ValType>{ValType::I64};
|
|
Function hello_world =
|
|
Function("hello_world", t, t,
|
|
[](CurrentPlugin plugin, const std::vector<Val> ¶ms,
|
|
std::vector<Val> &results, void *user_data) {
|
|
auto offs = plugin.alloc(4);
|
|
memcpy(plugin.memory() + offs, "test", 4);
|
|
results[0].v.i64 = (int64_t)offs;
|
|
});
|
|
auto functions = std::vector<Function>{
|
|
hello_world,
|
|
};
|
|
Plugin plugin(wasm, true, functions);
|
|
auto buf = plugin.call("count_vowels", "aaa");
|
|
ASSERT_EQ(buf.length, 4);
|
|
ASSERT_EQ((std::string)buf, "test");
|
|
}
|
|
|
|
}; // namespace
|
|
|
|
int main(int argc, char **argv) {
|
|
testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
} |