feat: add request header (#42)

Author: spacewander

src/http/ngx_http_wasm_api.c

@@ -936,13 +936,20 @@ proxy_add_header_map_value(int32_t type, int32_t key_data, int32_t key_size,
     must_get_memory(key, log, key_data, key_size);
     must_get_memory(val, log, data, size);
 
-    if (type == PROXY_MAP_TYPE_HTTP_RESPONSE_HEADERS) {
+    switch (type) {
+    case PROXY_MAP_TYPE_HTTP_REQUEST_HEADERS:
+        rc = ngx_http_wasm_set_req_header(r, key, key_size, val, size, 0);
+        break;
+
+    case PROXY_MAP_TYPE_HTTP_RESPONSE_HEADERS:
         rc = ngx_http_wasm_set_resp_header(r, key, key_size, 0, val, size, 0);
-        if (rc != NGX_OK) {
-            return PROXY_RESULT_BAD_ARGUMENT;
-        }
+        break;
+
+    default:
+        return PROXY_RESULT_BAD_ARGUMENT;
     }
 
+
     return PROXY_RESULT_OK;
 }
 

t/http_req_header.t

@@ -160,3 +160,41 @@ location /t {
 }
 --- response_body
 bar
+
+
+
+=== TEST 9: add header
+--- config
+location /t {
+    access_by_lua_block {
+        local wasm = require("resty.proxy-wasm")
+        local plugin = assert(wasm.load("plugin", "t/testdata/http_header/main.go.wasm"))
+        local ctx = assert(wasm.on_configure(plugin, 'req_hdr_add'))
+        assert(wasm.on_http_request_headers(ctx))
+    }
+    content_by_lua_block {
+        ngx.say(ngx.req.get_headers()["foo"])
+    }
+}
+--- more_headers
+FOO: foo
+--- response_body
+foobar
+
+
+
+=== TEST 10: add header, missing
+--- config
+location /t {
+    access_by_lua_block {
+        local wasm = require("resty.proxy-wasm")
+        local plugin = assert(wasm.load("plugin", "t/testdata/http_header/main.go.wasm"))
+        local ctx = assert(wasm.on_configure(plugin, 'req_hdr_add'))
+        assert(wasm.on_http_request_headers(ctx))
+    }
+    content_by_lua_block {
+        ngx.say(ngx.req.get_headers()["foo"])
+    }
+}
+--- response_body
+bar

t/testdata/http_header/main.go

@@ -68,6 +68,9 @@ func (ctx *httpContext) OnHttpRequestHeaders(numHeaders int, endOfStream bool) t
 
 	case "req_hdr_set":
 		proxywasm.ReplaceHttpRequestHeader("foo", "bar")
+
+	case "req_hdr_add":
+		proxywasm.AddHttpRequestHeader("foo", "bar")
 	}
 
 	return types.ActionContinue