WebAssembly Edit

What is WebAssembly (Wasm)?

As described on https://webassembly.org/

WebAssembly (abbreviated Wasm) is a binary instruction format for a stack-based virtual machine. Wasm is designed as a portable target for compilation of high-level languages like C/C++/Rust, enabling deployment on the web for client and server applications.

Overview

OPA is able to compile Rego policies into executable Wasm modules that can be evaluated with different inputs and external data. This is not running the OPA server in Wasm, nor is this just cross-compiled Golang code. The compiled Wasm module is a planned evaluation path for the source policy and query.

Current Status

The core language is supported fully but there are a number of built-in functions that are not, and probably won’t be natively supported in Wasm (e.g., http.send). Built-in functions that are not natively supported can be implemented in the host environment (e.g., JavaScript).

Compiling Policies

You can compile Rego policies into Wasm modules using the:

For example, the opa build command below compiles the example.rego file with the query data.example.allow = true and writes the output to policy.wasm. See opa build --help for more details.

opa build -d example.rego -o policy.wasm 'data.example.allow = true'

Note: The query must be specified at compile-time and cannot be changed without recompiling the binary!

Using Compiled Policies

JavaScript SDK

There is a JavaScript SDK available that simplifies the process of loading and evaluating compiled policies. If you want to evaluate Rego policies inside JavaScript we recommend you use the JavaScript SDK.

See https://github.com/open-policy-agent/npm-opa-wasm for more details.

There is an example NodeJS application located here.

From Scratch

If you want to integrate Wasm compiled policies into a language or runtime that does not have SDK support, read this section.

Instantiating the Wasm Module

Before you can evaluate Wasm compiled policies you need to instantiate the Wasm module produced by the compilation process described earlier on this page.

To load the compiled Wasm module refer the documentation for the Wasm runtime that you are using. At a high-level you must provide a memory buffer and a set of import functions. The memory buffer is a contiguous, mutable byte-array that allows you to pass data to the policy and receive output from the policy. The import functions are dependencies of the compiled policies.

Exports

The primary exported functions for interacting with policy modules are:

Function SignatureDescription
int32 eval(ctx_addr)Evaluates the loaded policy with the provided evaluation context. The return value is reserved for future use.
value_addr builtins(void)Returns the address of a mapping of built-in function names to numeric identifiers that are required by the policy.
ctx_addr opa_eval_ctx_new(void)Returns the address of a newly allocated evaluation context.
void opa_eval_ctx_set_input(ctx_addr, value_addr)Set the input value to use during evaluation. This must be called before each eval() call. If the input value is not set before evaluation, references to the input document result produce no results (i.e., they are undefined.)
void opa_eval_ctx_set_data(ctx_addr, value_addr)Set the data value to use during evalutaion. This should be called before each eval() call. If the data value is not set before evalutaion, references to base data documents produce no results (i.e., they are undefined.)
value_addr opa_eval_ctx_get_result(ctx_addr)Get the result set produced by the evaluation process.
addr opa_malloc(int32 size)Allocates size bytes in the shared memory and returns the starting address.
value_addr opa_json_parse(str_addr, size)Parses the JSON serialized value starting at str_addr of size bytes and returns the address of the parsed value. The parsed value may refer to a null, boolean, number, string, array, or object value.
str_addr opa_json_dump(value_addr)Dumps the value referred to by value_addr to a null-terminated JSON serialized string and returns the address of the start of the string.
void opa_heap_ptr_set(addr)Set the heap pointer for the next evaluation.
addr opa_heap_ptr_get(void)Get the current heap pointer.
void opa_heap_top_set(addr)Set the heap top for the next evaluation.
addr opa_heap_top_get(void)Get the current heap top.

The addresses passed and returned by the policy modules are 32-bit integer offsets into the shared memory region. The value_addr parameters and return values refer to OPA value data structures: null, boolean, number, string, array, object, and set.

Imports

Policy modules require the following function imports at instantiation-time:

NamespaceNameParamsResultDescription
envopa_abort(addr)voidCalled if an internal error occurs. The addr refers to a null-terminated string in the shared memory buffer.
envopa_builtin0(builtin_id, ctx)addrCalled to dispatch the built-in function identified by the builtin_id. The ctx parameter reserved for future use. The result addr must refer to a value in the shared-memory buffer. The function accepts 0 arguments.
envopa_builtin1(builtin_id, ctx, _1)addrSame as previous except the function accepts 1 argument.
envopa_builtin2(builtin_id, ctx, _1, _2)addrSame as previous except the function accepts 2 arguments.
envopa_builtin3(builtin_id, ctx, _1, _2, _3)addrSame as previous except the function accepts 3 arguments.
envopa_builtin4(builtin_id, ctx, _1, _2, _3, _4)addrSame as previous except the function accepts 4 arguments.

The policy module also requires a shared memory buffer named env.memory.

Memory Buffer

A shared memory buffer must be provided as an import for the policy module with the name env.memory. The buffer must be large enough to accommodate the input, provided data, and result of evaluation.

Built-in Functions

After instantiating the policy module, call the exported builtins function to receive a mapping of built-in functions required during evaluation. The result maps required built-in function names to the identifiers supplied to the built-in function callbacks (e.g., opa_builtin0, opa_builtin1, etc.)

For example:

const memory = new WebAssembly.Memory({ initial: 5 });
const policy_module = await WebAssembly.instantiate(byte_buffer, /* import object */);
const addr = policy_module.instance.exports.builtins();
const str_addr =  policy_module.instance.exports.opa_json_dump(addr);
const builtin_map = deserialize_null_terminated_JSON_string(memory, str_addr);

The built-in function mapping will contain all of the built-in functions that may be required during evaluation. For example, the following query refers to the http.send built-in function which is not included in the policy module:

result := http.send({"method": "get", "url": "https://example.com/api/lookup/12345"})

If this query was compiled to Wasm the built-in map would contain a single element:

{
    "http.send": 0
}

When the evaluation runs, the opa_builtin1 callback would invoked with builtin_id set to 0.

Evaluation

Once instantiated, the policy module is ready to be evaluated. Use the opa_eval_ctx_new exported function to create an evaluation context. Use the opa_eval_ctx_set_input and opa_eval_ctx_set_data exported functions to specify the values of the input and base data documents to use during evaluation.

To evaluate, call to the exported eval function with the eval context address as the only parameter.

Input

The (optional) input document for a policy can be provided by loading a JSON string into the shared memory buffer. Use the opa_malloc exported function to allocate a buffer the size of the JSON string and copy the contents in at the returned address. After the raw string is loaded into memory you will need to call the opa_json_parse exported method to get an address to the parsed input document for use in evaluations. Set the address via the opa_eval_ctx_set_input exported functoin supplying the evaluation context address and parsed input document address.

External Data

External data can be loaded for use in evaluation. Similar to the input this is done by loading a JSON string into the shared memory buffer. Use opa_malloc and opa_json_parse followed by opa_eval_ctx_set_data to set the address on the evaluation context.

After loading the external data use the opa_heap_ptr_get and opa_heap_top_get exported methods to save the current point in the heap before evaluation. After evaluation these should be reset by calling opa_heap_ptr_set and opa_heap_top_set to ensure that evaluation restarts back at the saved data and re-uses heap space. This is particularly important if re-evaluating many times with the same data.

Results

After evaluation results can be retrieved via the exported opa_eval_ctx_get_result function. Pass in the evaluation context address. The return value is an address in the shared memory buffer to the structured result. To access the JSON result use the opa_json_dump exported function to retrieve a pointer in shared memory to a null terminated JSON string.

The result of evaluation is the set variable bindings that satisfy the expressions in the query. For example, the query x = 1; y = 2; y > x would produce the following result set:

[
    {
        "x": 1,
        "y": 2
    }
]

Sets are represented as JSON arrays.