feature: implemented the FFI interface for the 'ngx_req.add_header()' API.

Signed-off-by: Thibault Charbonnier <[email protected]>

Author: spacewander

src/ngx_http_lua_headers.c

@@ -955,11 +955,15 @@ ngx_http_lua_ffi_set_resp_header(ngx_http_request_t *r, const u_char *key_data,
 
 
 int
-ngx_http_lua_ffi_req_header_set_single_value(ngx_http_request_t *r,
-    const u_char *key, size_t key_len, const u_char *value, size_t value_len)
+ngx_http_lua_ffi_req_set_header(ngx_http_request_t *r, const u_char *key,
+    size_t key_len, const u_char *value, size_t value_len,
+    ngx_http_lua_ffi_str_t *mvals, size_t mvals_len, int override,
+    char **errmsg)
 {
-    ngx_str_t                    k;
-    ngx_str_t                    v;
+    u_char                      *p;
+    size_t                       len;
+    ngx_uint_t                   i;
+    ngx_str_t                    k, v;
 
     if (r->connection->fd == (ngx_socket_t) -1) {  /* fake request */
         return NGX_HTTP_LUA_FFI_BAD_CONTEXT;
@@ -971,35 +975,71 @@ ngx_http_lua_ffi_req_header_set_single_value(ngx_http_request_t *r,
 
     k.data = ngx_palloc(r->pool, key_len + 1);
     if (k.data == NULL) {
-        return NGX_ERROR;
+        goto nomem;
     }
+
     ngx_memcpy(k.data, key, key_len);
     k.data[key_len] = '\0';
-
     k.len = key_len;
 
-    if (value_len == 0) {
+    if (mvals) {
+        if (mvals_len > 0) {
+            for (i = 0; i < mvals_len; i++) {
+                p = mvals[i].data;
+                len = mvals[i].len;
+
+                v.data = ngx_palloc(r->pool, len + 1);
+                if (v.data == NULL) {
+                    goto nomem;
+                }
+
+                ngx_memcpy(v.data, p, len);
+                v.data[len] = '\0';
+                v.len = len;
+
+                if (ngx_http_lua_set_input_header(r, k, v, override)
+                    != NGX_OK)
+                {
+                    goto failed;
+                }
+            }
+
+            return NGX_OK;
+        }
+
         v.data = NULL;
         v.len = 0;
 
-    } else {
+    } else if (value) {
         v.data = ngx_palloc(r->pool, value_len + 1);
         if (v.data == NULL) {
-            return NGX_ERROR;
+            goto nomem;
         }
+
         ngx_memcpy(v.data, value, value_len);
         v.data[value_len] = '\0';
-    }
+        v.len = value_len;
 
-    v.len = value_len;
+    } else {
+        v.data = NULL;
+        v.len = 0;
+    }
 
-    if (ngx_http_lua_set_input_header(r, k, v, 1 /* override */)
-        != NGX_OK)
-    {
-        return NGX_ERROR;
+    if (ngx_http_lua_set_input_header(r, k, v, override) != NGX_OK) {
+        goto failed;
     }
 
     return NGX_OK;
+
+nomem:
+
+    *errmsg = "no memory";
+    return NGX_ERROR;
+
+failed:
+
+    *errmsg = "failed to set header";
+    return NGX_ERROR;
 }