mirror of
https://github.com/extism/extism.git
synced 2026-01-13 07:47:56 -05:00
2a24d13c9b
- Uses `tracing` instead of `log` crate - Uses `tracing-subscriber` instead of `fern` - This allows us to automatically capture `log` events using `tracing-subscriber` - Breaking: Makes `extism::set_log_file` private and only used through the C API, Rust users should use `tracing-subscriber` to determine which filters/levels to log. - Adds `extism::set_log_callback` function to set a callback that can be used for custom logging from Rust. - Adds `bool extism_log_custom(const char *level)` and `extism_log_drain(void (*fn)(const char *s, size_t length)` to the C API to enable custom sinks in other SDKs
93 lines
2.3 KiB
C
93 lines
2.3 KiB
C
#include "../runtime/extism.h"
|
|
|
|
#include <assert.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/stat.h>
|
|
#include <unistd.h>
|
|
|
|
void log_handler(const char *line, uintptr_t length) {
|
|
fwrite(line, length, 1, stderr);
|
|
}
|
|
|
|
void hello_world(ExtismCurrentPlugin *plugin, const ExtismVal *inputs,
|
|
uint64_t n_inputs, ExtismVal *outputs, uint64_t n_outputs,
|
|
void *data) {
|
|
puts("Hello from C!");
|
|
puts(data);
|
|
|
|
ExtismSize ptr_offs = inputs[0].v.i64;
|
|
|
|
uint8_t *buf = extism_current_plugin_memory(plugin) + ptr_offs;
|
|
uint64_t length = extism_current_plugin_memory_length(plugin, ptr_offs);
|
|
fwrite(buf, length, 1, stdout);
|
|
fputc('\n', stdout);
|
|
outputs[0].v.i64 = inputs[0].v.i64;
|
|
}
|
|
|
|
void free_data(void *x) { puts("Freeing userdata"); }
|
|
|
|
uint8_t *read_file(const char *filename, size_t *len) {
|
|
|
|
FILE *fp = fopen(filename, "rb");
|
|
if (fp == NULL) {
|
|
return NULL;
|
|
}
|
|
fseek(fp, 0, SEEK_END);
|
|
size_t length = ftell(fp);
|
|
fseek(fp, 0, SEEK_SET);
|
|
|
|
uint8_t *data = malloc(length);
|
|
if (data == NULL) {
|
|
fclose(fp);
|
|
return NULL;
|
|
}
|
|
|
|
assert(fread(data, 1, length, fp) == length);
|
|
fclose(fp);
|
|
|
|
*len = length;
|
|
return data;
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
if (argc < 2) {
|
|
fputs("Not enough arguments\n", stderr);
|
|
exit(1);
|
|
}
|
|
|
|
extism_log_custom("extism=trace,cranelift=trace");
|
|
|
|
size_t len = 0;
|
|
uint8_t *data = read_file("../wasm/code-functions.wasm", &len);
|
|
ExtismValType inputs[] = {PTR};
|
|
ExtismValType outputs[] = {PTR};
|
|
ExtismFunction *f =
|
|
extism_function_new("hello_world", inputs, 1, outputs, 1, hello_world,
|
|
"Hello, again!", free_data);
|
|
|
|
char *errmsg = NULL;
|
|
ExtismPlugin *plugin = extism_plugin_new(
|
|
data, len, (const ExtismFunction **)&f, 1, true, &errmsg);
|
|
free(data);
|
|
if (plugin == NULL) {
|
|
puts(errmsg);
|
|
extism_plugin_new_error_free(errmsg);
|
|
exit(1);
|
|
}
|
|
|
|
assert(extism_plugin_call(plugin, "count_vowels", (uint8_t *)argv[1],
|
|
strlen(argv[1])) == 0);
|
|
ExtismSize out_len = extism_plugin_output_length(plugin);
|
|
const uint8_t *output = extism_plugin_output_data(plugin);
|
|
write(STDOUT_FILENO, output, out_len);
|
|
write(STDOUT_FILENO, "\n", 1);
|
|
extism_plugin_free(plugin);
|
|
extism_function_free(f);
|
|
extism_log_drain(log_handler);
|
|
return 0;
|
|
}
|