• Home
• fr.maif
• otoroshi_2.12
• 16.22.0
• source code
• functions.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

wasm.proxywasm.functions.scala Maven / Gradle / Ivy

package otoroshi.wasm.proxywasm

import akka.stream.Materializer
import io.otoroshi.wasm4s.scaladsl._
import org.extism.sdk.{ExtismCurrentPlugin, HostFunction, LibExtism}
import org.extism.sdk.wasmotoroshi._
import otoroshi.env.Env
import otoroshi.wasm._

import java.util.Optional
import java.util.concurrent.atomic.AtomicReference
import scala.concurrent.ExecutionContext

object ProxyWasmFunctions {

  //private def getCurrentVmData(): VmData = {
  //  WasmContextSlot.getCurrentContext() match {
  //    case Some(data: VmData) => data
  //    case _                  =>
  //      println("missing vm data")
  //      new RuntimeException("missing vm data").printStackTrace()
  //      throw new RuntimeException("missing vm data")
  //  }
  //}

  def build(
      state: ProxyWasmState,
      vmDataRef: AtomicReference[WasmVmData]
  )(implicit ec: ExecutionContext, env: Env, mat: Materializer): Seq[HostFunction[EnvUserData]] = {
    def getCurrentVmData(): VmData = {
      Option(vmDataRef.get()) match {
        case Some(data: VmData) => data
        case _                  =>
          println("missing vm data")
          new RuntimeException("missing vm data").printStackTrace()
          throw new RuntimeException("missing vm data")
      }
    }
    Seq(
      new HostFunction[EnvUserData](
        "proxy_log",
        parameters(3),
        parameters(1),
        (
            plugin: ExtismCurrentPlugin,
            params: Array[LibExtism.ExtismVal],
            returns: Array[LibExtism.ExtismVal],
            data: Optional[EnvUserData]
        ) => state.proxyLog(plugin, params(0).v.i32, params(1).v.i32, params(2).v.i32, getCurrentVmData()),
        Optional.empty[EnvUserData]()
      ).withNamespace("env"),
      new HostFunction[EnvUserData](
        "proxy_get_buffer_bytes",
        parameters(5),
        parameters(1),
        (
            plugin: ExtismCurrentPlugin,
            params: Array[LibExtism.ExtismVal],
            returns: Array[LibExtism.ExtismVal],
            data: Optional[EnvUserData]
        ) =>
          state.proxyGetBuffer(
            plugin,
            getCurrentVmData(),
            params(0).v.i32,
            params(1).v.i32,
            params(2).v.i32,
            params(3).v.i32,
            params(4).v.i32
          ),
        Optional.empty[EnvUserData]()
      ).withNamespace("env"),
      new HostFunction[EnvUserData](
        "proxy_set_effective_context",
        parameters(1),
        parameters(1),
        (
            plugin: ExtismCurrentPlugin,
            params: Array[LibExtism.ExtismVal],
            returns: Array[LibExtism.ExtismVal],
            data: Optional[EnvUserData]
        ) => state.proxySetEffectiveContext(plugin, params(0).v.i32),
        Optional.empty[EnvUserData]()
      ).withNamespace("env"),
      new HostFunction[EnvUserData](
        "proxy_get_header_map_pairs",
        parameters(3),
        parameters(1),
        (
            plugin: ExtismCurrentPlugin,
            params: Array[LibExtism.ExtismVal],
            returns: Array[LibExtism.ExtismVal],
            data: Optional[EnvUserData]
        ) =>
          state.proxyGetHeaderMapPairs(plugin, getCurrentVmData(), params(0).v.i32, params(1).v.i32, params(2).v.i32),
        Optional.empty[EnvUserData]()
      ).withNamespace("env"),
      new HostFunction[EnvUserData](
        "proxy_set_buffer_bytes",
        parameters(5),
        parameters(1),
        (
            plugin: ExtismCurrentPlugin,
            params: Array[LibExtism.ExtismVal],
            returns: Array[LibExtism.ExtismVal],
            data: Optional[EnvUserData]
        ) =>
          state.proxySetBuffer(
            plugin,
            getCurrentVmData(),
            params(0).v.i32,
            params(1).v.i32,
            params(2).v.i32,
            params(3).v.i32,
            params(4).v.i32
          ),
        Optional.empty[EnvUserData]()
      ).withNamespace("env"),
      new HostFunction[EnvUserData](
        "proxy_get_header_map_value",
        parameters(5),
        parameters(1),
        (
            plugin: ExtismCurrentPlugin,
            params: Array[LibExtism.ExtismVal],
            returns: Array[LibExtism.ExtismVal],
            data: Optional[EnvUserData]
        ) =>
          state.proxyGetHeaderMapValue(
            plugin,
            getCurrentVmData(),
            params(0).v.i32,
            params(1).v.i32,
            params(2).v.i32,
            params(3).v.i32,
            params(4).v.i32
          ),
        Optional.empty[EnvUserData]()
      ).withNamespace("env"),
      new HostFunction[EnvUserData](
        "proxy_get_property",
        parameters(4),
        parameters(1),
        (
            plugin: ExtismCurrentPlugin,
            params: Array[LibExtism.ExtismVal],
            returns: Array[LibExtism.ExtismVal],
            data: Optional[EnvUserData]
        ) =>
          state.proxyGetProperty(
            plugin,
            getCurrentVmData(),
            params(0).v.i32,
            params(1).v.i32,
            params(2).v.i32,
            params(3).v.i32
          ),
        Optional.empty[EnvUserData]()
      ).withNamespace("env"),
      new HostFunction[EnvUserData](
        "proxy_increment_metric",
        Seq(LibExtism.ExtismValType.I32, LibExtism.ExtismValType.I64).toArray,
        parameters(1),
        (
            plugin: ExtismCurrentPlugin,
            params: Array[LibExtism.ExtismVal],
            returns: Array[LibExtism.ExtismVal],
            data: Optional[EnvUserData]
        ) => state.proxyIncrementMetricValue(plugin, getCurrentVmData(), params(0).v.i32, params(1).v.i64),
        Optional.empty[EnvUserData]()
      ).withNamespace("env"),
      new HostFunction[EnvUserData](
        "proxy_define_metric",
        parameters(4),
        parameters(1),
        (
            plugin: ExtismCurrentPlugin,
            params: Array[LibExtism.ExtismVal],
            returns: Array[LibExtism.ExtismVal],
            data: Optional[EnvUserData]
        ) => state.proxyDefineMetric(plugin, params(0).v.i32, params(1).v.i32, params(2).v.i32, params(3).v.i32),
        Optional.empty[EnvUserData]()
      ).withNamespace("env"),
      new HostFunction[EnvUserData](
        "proxy_set_tick_period_milliseconds",
        parameters(1),
        parameters(1),
        (
            plugin: ExtismCurrentPlugin,
            params: Array[LibExtism.ExtismVal],
            returns: Array[LibExtism.ExtismVal],
            data: Optional[EnvUserData]
        ) => state.proxySetTickPeriodMilliseconds(getCurrentVmData(), params(0).v.i32),
        Optional.empty[EnvUserData]()
      ).withNamespace("env"),
      new HostFunction[EnvUserData](
        "proxy_replace_header_map_value",
        parameters(5),
        parameters(1),
        (
            plugin: ExtismCurrentPlugin,
            params: Array[LibExtism.ExtismVal],
            returns: Array[LibExtism.ExtismVal],
            data: Optional[EnvUserData]
        ) =>
          state.proxyReplaceHeaderMapValue(
            plugin,
            getCurrentVmData(),
            params(0).v.i32,
            params(1).v.i32,
            params(2).v.i32,
            params(3).v.i32,
            params(4).v.i32
          ),
        Optional.empty[EnvUserData]()
      ).withNamespace("env"),
      new HostFunction[EnvUserData](
        "proxy_send_local_response",
        parameters(8),
        parameters(1),
        (
            plugin: ExtismCurrentPlugin,
            params: Array[LibExtism.ExtismVal],
            returns: Array[LibExtism.ExtismVal],
            data: Optional[EnvUserData]
        ) =>
          state.proxySendHttpResponse(
            plugin,
            params(0).v.i32,
            params(1).v.i32,
            params(2).v.i32,
            params(3).v.i32,
            params(4).v.i32,
            params(5).v.i32,
            params(6).v.i32,
            params(7).v.i32,
            getCurrentVmData()
          ),
        Optional.empty[EnvUserData]()
      ).withNamespace("env")
    )
  }

  private def parameters(n: Int): Array[LibExtism.ExtismValType] = {
    (0 until n).map(_ => LibExtism.ExtismValType.I32).toArray
  }
}