test: abnormal input for header API (#74)

Author: spacewander

t/http_req_header.t

@@ -389,3 +389,48 @@ location /t {
 qr/get request scheme: \S+/
 --- grep_error_log_out
 get request scheme: http,
+
+
+
+=== TEST 21: get header, abnormal
+--- config
+location /t {
+    content_by_lua_block {
+        local json = require("cjson")
+        local wasm = require("resty.proxy-wasm")
+        local plugin = assert(wasm.load("plugin", "t/testdata/http_header/main.go.wasm"))
+        local conf = {action = 'req_hdr_get_abnormal'}
+        for _, e in ipairs({
+            ""
+        }) do
+            conf.header = e
+            local ctx = assert(wasm.on_configure(plugin, json.encode(conf)))
+            assert(wasm.on_http_request_headers(ctx))
+        end
+    }
+}
+--- more_headers
+X-API: foo
+--- error_log
+error status returned by host: not found
+
+
+
+=== TEST 22: set header, abnormal
+--- config
+location /t {
+    content_by_lua_block {
+        local json = require("cjson")
+        local wasm = require("resty.proxy-wasm")
+        local plugin = assert(wasm.load("plugin", "t/testdata/http_header/main.go.wasm"))
+        local conf = {action = 'req_hdr_set_abnormal'}
+        for _, e in ipairs({
+            {"", ""}
+        }) do
+            conf.header = e[1]
+            conf.value = e[2]
+            local ctx = assert(wasm.on_configure(plugin, json.encode(conf)))
+            assert(wasm.on_http_request_headers(ctx))
+        end
+    }
+}

t/http_resp_header.t

@@ -253,3 +253,48 @@ qr/get response header \S+ \S+/
 get response header content-type text/plain,
 get response header content-length 0,
 get response header connection keep-alive,
+
+
+
+=== TEST 14: get header, abnormal
+--- config
+location /t {
+    return 200;
+    header_filter_by_lua_block {
+        local json = require("cjson")
+        local wasm = require("resty.proxy-wasm")
+        local plugin = assert(wasm.load("plugin", "t/testdata/http_header/main.go.wasm"))
+        local conf = {action = 'resp_hdr_get_abnormal'}
+        for _, e in ipairs({
+            ""
+        }) do
+            conf.header = e
+            local ctx = assert(wasm.on_configure(plugin, json.encode(conf)))
+            assert(wasm.on_http_response_headers(ctx))
+        end
+    }
+}
+--- error_log
+error status returned by host: not found
+
+
+
+=== TEST 15: set header, abnormal
+--- config
+location /t {
+    return 200;
+    header_filter_by_lua_block {
+        local json = require("cjson")
+        local wasm = require("resty.proxy-wasm")
+        local plugin = assert(wasm.load("plugin", "t/testdata/http_header/main.go.wasm"))
+        local conf = {action = 'resp_hdr_set_abnormal'}
+        for _, e in ipairs({
+            {"", ""}
+        }) do
+            conf.header = e[1]
+            conf.value = e[2]
+            local ctx = assert(wasm.on_configure(plugin, json.encode(conf)))
+            assert(wasm.on_http_response_headers(ctx))
+        end
+    }
+}

t/testdata/http_header/main.go

@@ -3,6 +3,7 @@ package main
 import (
 	"github.com/tetratelabs/proxy-wasm-go-sdk/proxywasm"
 	"github.com/tetratelabs/proxy-wasm-go-sdk/proxywasm/types"
+	"github.com/valyala/fastjson"
 )
 
 func main() {
@@ -37,7 +38,22 @@ func (ctx *httpContext) OnHttpRequestHeaders(numHeaders int, endOfStream bool) t
 		return types.ActionContinue
 	}
 
-	action := string(data)
+	var action string
+	var conf *fastjson.Value
+	if data[0] == '{' {
+		var p fastjson.Parser
+		conf, err = p.ParseBytes(data)
+		if err != nil {
+			proxywasm.LogErrorf("erorr decoding plugin configuration: %v", err)
+			return types.ActionContinue
+		}
+
+		action = string(conf.GetStringBytes("action"))
+
+	} else {
+		action = string(data)
+	}
+
 	switch action {
 	case "req_hdr_get_all":
 		hdrs, err := proxywasm.GetHttpRequestHeaders()
@@ -66,6 +82,15 @@ func (ctx *httpContext) OnHttpRequestHeaders(numHeaders int, endOfStream bool) t
 		}
 		proxywasm.LogWarnf("get request header: %v", res)
 
+	case "req_hdr_get_abnormal":
+		hdr := string(conf.GetStringBytes("header"))
+		res, err := proxywasm.GetHttpRequestHeader(hdr)
+		if err != nil {
+			proxywasm.LogErrorf("error get request header: %v", err)
+			return types.ActionContinue
+		}
+		proxywasm.LogWarnf("get request header: %v", res)
+
 	case "req_hdr_set":
 		proxywasm.ReplaceHttpRequestHeader("foo", "bar")
 
@@ -75,6 +100,15 @@ func (ctx *httpContext) OnHttpRequestHeaders(numHeaders int, endOfStream bool) t
 	case "req_hdr_del":
 		proxywasm.RemoveHttpRequestHeader("foo")
 
+	case "req_hdr_set_abnormal":
+		hdr := string(conf.GetStringBytes("header"))
+		v := string(conf.GetStringBytes("value"))
+		err := proxywasm.ReplaceHttpRequestHeader(hdr, v)
+		if err != nil {
+			proxywasm.LogErrorf("error set request header: %v", err)
+			return types.ActionContinue
+		}
+
 	case "req_path_get":
 		res, err := proxywasm.GetHttpRequestHeader(":path")
 		if err != nil {
@@ -110,7 +144,22 @@ func (ctx *httpContext) OnHttpResponseHeaders(numHeaders int, endOfStream bool)
 		return types.ActionContinue
 	}
 
-	action := string(data)
+	var action string
+	var conf *fastjson.Value
+	if data[0] == '{' {
+		var p fastjson.Parser
+		conf, err = p.ParseBytes(data)
+		if err != nil {
+			proxywasm.LogErrorf("erorr decoding plugin configuration: %v", err)
+			return types.ActionContinue
+		}
+
+		action = string(conf.GetStringBytes("action"))
+
+	} else {
+		action = string(data)
+	}
+
 	if action == "resp_hdr_get_all" {
 		hdrs, err := proxywasm.GetHttpResponseHeaders()
 		if err != nil {
@@ -140,6 +189,15 @@ func (ctx *httpContext) OnHttpResponseHeaders(numHeaders int, endOfStream bool)
 		proxywasm.RemoveHttpResponseHeader("add")
 	case "resp_hdr_del_miss":
 		proxywasm.RemoveHttpResponseHeader("aa")
+	case "resp_hdr_set_abnormal":
+		hdr := string(conf.GetStringBytes("header"))
+		v := string(conf.GetStringBytes("value"))
+		err := proxywasm.ReplaceHttpResponseHeader(hdr, v)
+		if err != nil {
+			proxywasm.LogErrorf("error set request header: %v", err)
+			return types.ActionContinue
+		}
+
 	case "resp_hdr_get":
 		res, err := proxywasm.GetHttpResponseHeader("add")
 		if err != nil {
@@ -162,6 +220,14 @@ func (ctx *httpContext) OnHttpResponseHeaders(numHeaders int, endOfStream bool)
 			return types.ActionContinue
 		}
 		proxywasm.LogWarnf("get response header: %v", res)
+	case "resp_hdr_get_abnormal":
+		hdr := string(conf.GetStringBytes("header"))
+		res, err := proxywasm.GetHttpResponseHeader(hdr)
+		if err != nil {
+			proxywasm.LogErrorf("error get response header: %v", err)
+			return types.ActionContinue
+		}
+		proxywasm.LogWarnf("get response header: %v", res)
 	}
 
 	return types.ActionContinue