feature: add ngx.req.get_original_addr.

Author: zhuizhuhaomeng

lib/resty/core/request.lua

@@ -42,6 +42,8 @@ local ngx_lua_ffi_req_start_time
 if subsystem == "stream" then
  ffi.cdef[[
  double ngx_stream_lua_ffi_req_start_time(ngx_stream_lua_request_t *r);
+ int ngx_stream_lua_ffi_req_dst_addr(ngx_stream_lua_request_t *r, char *buf,
+ size_t *buf_size, char *errbuf, size_t *errbuf_size);
  ]]
 
  ngx_lua_ffi_req_start_time = C.ngx_stream_lua_ffi_req_start_time
@@ -66,6 +68,24 @@ end
 
 
 if subsystem == "stream" then
+ local ERR_BUF_SIZE = 256
+
+ local buf = get_string_buf(128, 1)
+ local buf_sizep = ffi_new("size_t[1]")
+
+ function ngx.req.get_original_addr()
+ local r = get_request()
+ local err = get_string_buf(ERR_BUF_SIZE)
+ local errlenp = get_size_ptr()
+ buf_sizep[0] = 128
+ local rc = C.ngx_stream_lua_ffi_req_dst_addr(r, buf, buf_sizep, err, errlenp)
+ if tonumber(rc) ~= 0 then
+ return nil, tostring(err, errlenp[0])
+ end
+
+ return ffi_str(buf, buf_sizep[0])
+ end
+
  return _M
 end
 

t/process-type-hup.t

@@ -86,3 +86,4 @@ init_worker_by_lua(nginx.conf:48):6: process type: privileged
 --- no_error_log
 [error]
 --- skip_nginx: 4: < 1.11.2
+--- wait: 0.2

t/stream/request.t

@@ -37,3 +37,19 @@ qr/\[TRACE\s+\d+ content_by_lua\(nginx\.conf:\d+\):3 loop\]/
 [error]
 bad argument type
 stitch
+
+
+
+=== TEST 2: get dst address
+--- stream_server_config
+ content_by_lua_block {
+ local dst_addr = ngx.req.get_original_addr()
+ ngx.say("origin addr: ", dst_addr)
+ }
+--- stream_response eval
+qr/127.0.0.1:\d+/
+--- no_error_log
+[error]
+[alert]
+[crit]
+[crit2]