Add an example that uses the Coraza waf.

- handle getting well known request/response headers.
- encode source/destination ports as 8 bytes little endian.
- better handle the case when jaxrs resource responds without a resource

Signed-off-by: Hiram Chirino <hiram@hiramchirino.com>

Author: chirino

README.md

@@ -10,7 +10,7 @@ To build the project, you need to have Maven installed. You can build the projec
 mvn clean install
 ```
 
-## Quarkus Example 
+## Quarkus Example
 
 The `quarkus-proxy-wasm-example` directory contains a simple example of how to use the proxy-wasm Java host with Quarkus.
 
@@ -35,7 +35,7 @@ public class Resources {
 }
 ```
 
-The `WasmPlugin` annotation is used to specify the name of the plugin to be used for filtering. 
+The `WasmPlugin` annotation is used to specify the name of the plugin to be used for filtering.
 
 ```java
 package io.roastedroot.proxywasm.jaxrs.example;

pom.xml

@@ -234,6 +234,7 @@
       <modules>
         <module>quarkus-proxy-wasm</module>
         <module>quarkus-proxy-wasm-example</module>
+        <module>quarkus-x-corazawaf-example</module>
       </modules>
     </profile>
   </profiles>

proxy-wasm-java-host/src/main/java/io/roastedroot/proxywasm/Helpers.java

@@ -1,6 +1,8 @@
 package io.roastedroot.proxywasm;
 
 import com.dylibso.chicory.runtime.HostFunction;
+import java.nio.ByteBuffer;
+import java.nio.ByteOrder;
 import java.nio.charset.StandardCharsets;
 import java.time.Duration;
 import java.util.ArrayList;
@@ -45,20 +47,31 @@ public static byte[] bytes(Duration value) {
     }
 
     public static byte[] bytes(int value) {
-        // TODO: test to check byte order
-        return new byte[] {
-            (byte) (value >> 24), (byte) (value >> 16), (byte) (value >> 8), (byte) value
-        };
+        return ByteBuffer.allocate(Integer.BYTES)
+                .order(ByteOrder.LITTLE_ENDIAN)
+                .putInt(value)
+                .array();
+    }
+
+    public static byte[] bytes(long value) {
+        return ByteBuffer.allocate(Long.BYTES)
+                .order(ByteOrder.LITTLE_ENDIAN)
+                .putLong(value)
+                .array();
     }
 
     public static int int32(byte[] bytes) {
         if (bytes == null || bytes.length != 4) {
             throw new IllegalArgumentException("Byte array must be exactly 4 bytes long");
         }
-        return ((bytes[0] & 0xFF) << 24)
-                | ((bytes[1] & 0xFF) << 16)
-                | ((bytes[2] & 0xFF) << 8)
-                | (bytes[3] & 0xFF);
+        return ByteBuffer.wrap(bytes).order(ByteOrder.LITTLE_ENDIAN).getInt();
+    }
+
+    public static long int64(byte[] bytes) {
+        if (bytes == null || bytes.length != 8) {
+            throw new IllegalArgumentException("Byte array must be exactly 8 bytes long");
+        }
+        return ByteBuffer.wrap(bytes).order(ByteOrder.LITTLE_ENDIAN).getLong();
     }
 
     public static String string(byte[] value) {

proxy-wasm-java-host/src/main/java/io/roastedroot/proxywasm/LogHandler.java

@@ -4,6 +4,14 @@ public interface LogHandler {
 
     LogHandler DEFAULT = new LogHandler() {};
 
+    LogHandler SYSTEM =
+            new LogHandler() {
+                @Override
+                public void log(LogLevel level, String message) throws WasmException {
+                    System.out.println(level + ": " + message);
+                }
+            };
+
     default void log(LogLevel level, String message) throws WasmException {}
 
     default LogLevel getLogLevel() throws WasmException {

proxy-wasm-java-host/src/main/java/io/roastedroot/proxywasm/SplitProxyMap.java

@@ -0,0 +1,60 @@
+package io.roastedroot.proxywasm;
+
+import java.util.Map;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
+import java.util.stream.StreamSupport;
+
+/**
+ * A ProxyMap implementation that chains to another ProxyMap instance.
+ */
+public class SplitProxyMap implements ProxyMap {
+
+    private final ProxyMap primary;
+    private final ProxyMap secondary;
+
+    public SplitProxyMap(ProxyMap primary, ProxyMap secondary) {
+        this.primary = primary;
+        this.secondary = secondary;
+    }
+
+    @Override
+    public int size() {
+        return primary.size() + secondary.size();
+    }
+
+    @Override
+    public void add(String key, String value) {
+        secondary.add(key, value);
+    }
+
+    @Override
+    public void put(String key, String value) {
+        if (primary.get(key) != null) {
+            primary.put(key, value);
+        }
+        secondary.put(key, value);
+    }
+
+    @Override
+    public Iterable<? extends Map.Entry<String, String>> entries() {
+        return Stream.concat(
+                        StreamSupport.stream(primary.entries().spliterator(), false),
+                        StreamSupport.stream(secondary.entries().spliterator(), false))
+                .collect(Collectors.toList());
+    }
+
+    @Override
+    public String get(String key) {
+        String value = primary.get(key);
+        if (value != null) {
+            return value;
+        }
+        return secondary.get(key);
+    }
+
+    @Override
+    public void remove(String key) {
+        secondary.remove(key);
+    }
+}

proxy-wasm-java-host/src/main/java/io/roastedroot/proxywasm/WellKnownHeaders.java

@@ -10,4 +10,5 @@ private WellKnownHeaders() {}
     public static final String AUTHORITY = ":authority";
     public static final String PATH = ":path";
     public static final String METHOD = ":method";
+    public static final String STATUS = ":status";
 }

proxy-wasm-java-host/src/main/java/io/roastedroot/proxywasm/plugin/HttpContext.java

@@ -79,12 +79,16 @@ public byte[] getHttpRequestBody() {
         return httpRequestBody;
     }
 
+    public void setHttpRequestBody(byte[] httpRequestBody) {
+        this.httpRequestBody = httpRequestBody;
+    }
+
     public byte[] getHttpResponseBody() {
         return httpResponseBody;
     }
 
-    public void setHttpRequestBody(byte[] httpRequestBody) {
-        this.httpRequestBody = httpRequestBody;
+    public void setHttpResponseBody(byte[] httpResponseBody) {
+        this.httpResponseBody = httpResponseBody;
     }
 
     public byte[] getGrpcReceiveBuffer() {
@@ -111,10 +115,6 @@ public void setDownStreamData(byte[] downStreamData) {
         this.downStreamData = downStreamData;
     }
 
-    public void setHttpResponseBody(byte[] httpResponseBody) {
-        this.httpResponseBody = httpResponseBody;
-    }
-
     public SendResponse getSendResponse() {
         return sendResponse;
     }

proxy-wasm-java-host/src/main/java/io/roastedroot/proxywasm/plugin/HttpRequestAdaptor.java

@@ -12,11 +12,13 @@ public interface HttpRequestAdaptor {
 
     String remoteAddress();
 
-    String remotePort();
+    int remotePort();
 
     String localAddress();
 
-    String localPort();
+    int localPort();
+
+    String protocol();
 
     ProxyMap getHttpRequestHeaders();
 

proxy-wasm-java-host/src/main/java/io/roastedroot/proxywasm/plugin/Plugin.java

@@ -29,7 +29,6 @@
 import io.roastedroot.proxywasm.WasmException;
 import io.roastedroot.proxywasm.WasmResult;
 import java.net.URI;
-import java.net.URISyntaxException;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -418,11 +417,8 @@ public int httpCall(
 
             URI uri = null;
             try {
-                uri =
-                        URI.create(
-                                new URI(scheme, null, authority, connectPort, null, null, null)
-                                        + path);
-            } catch (IllegalArgumentException | URISyntaxException e) {
+                uri = URI.create(scheme + "://" + authority + path);
+            } catch (IllegalArgumentException e) {
                 throw new WasmException(WasmResult.BAD_ARGUMENT);
             }
 

proxy-wasm-java-host/src/test/go-examples/unit_tester/http_call.go

@@ -31,7 +31,7 @@ type httpCallTests struct {
 }
 
 func (
-	p *pluginContext) httpCallTests(contextID uint32) types.HttpContext {
+p *pluginContext) httpCallTests(contextID uint32) types.HttpContext {
 	return &httpCallTests{
 		DefaultHttpContext: types.DefaultHttpContext{},
 		contextID:          contextID,
@@ -42,12 +42,8 @@ func (
 
 func (ctx *httpCallTests) OnHttpRequestHeaders(int, bool) types.Action {
 	proxywasm.LogDebug("OnHttpRequestHeaders")
-	var err error
-	ctx.headers, err = proxywasm.GetHttpRequestHeaders()
-	if err != nil {
-		proxywasm.LogCriticalf("failed to get request headers: %v", err)
-	}
 
+	ctx.headers = nil
 	method, err := proxywasm.GetProperty([]string{"request", "method"})
 	if err != nil {
 		proxywasm.LogCriticalf("failed to get request method: %v", err)
@@ -63,6 +59,11 @@ func (ctx *httpCallTests) OnHttpRequestHeaders(int, bool) types.Action {
 	path := gjson.Get(ctx.pluginContext.config, "path").Str
 	ctx.headers = append(ctx.headers, [2]string{":path", path})
 
+	requestHeaders, err := proxywasm.GetHttpRequestHeaders()
+	if err != nil {
+		proxywasm.LogCriticalf("failed to get request headers: %v", err)
+	}
+	ctx.headers = append(ctx.headers, requestHeaders...)
 	return types.ActionContinue
 }