feature: add ffi api ngx_stream_lua_ffi_socket_tcp_getfd.

Author: zhuizhuhaomeng

src/ngx_stream_lua_socket_tcp.c

@@ -271,6 +271,8 @@ static char ngx_stream_lua_ssl_session_metatable_key;
 #endif
 
 #define ngx_stream_lua_tcp_socket_metatable_literal_key "__tcp_cosocket_mt"
+#define ngx_stream_lua_raw_req_socket_metatable_literal_key \
+ "__tcp_raw_req_cosocket_mt"
 
 
 void
@@ -308,7 +310,7 @@ ngx_stream_lua_inject_socket_tcp_api(ngx_log_t *log, lua_State *L)
  /* {{{raw req socket object metatable */
  lua_pushlightuserdata(L, ngx_stream_lua_lightudata_mask(
  raw_req_socket_metatable_key));
- lua_createtable(L, 0 /* narr */, 9 /* nrec */);
+ lua_createtable(L, 0 /* narr */, 10 /* nrec */);
 
  lua_pushcfunction(L, ngx_stream_lua_socket_tcp_receive);
  lua_setfield(L, -2, "receive");
@@ -344,6 +346,12 @@ ngx_stream_lua_inject_socket_tcp_api(ngx_log_t *log, lua_State *L)
  lua_pushvalue(L, -1);
  lua_setfield(L, -2, "__index");
 
+ lua_rawset(L, LUA_REGISTRYINDEX);
+
+ lua_pushliteral(L, ngx_stream_lua_raw_req_socket_metatable_literal_key);
+ lua_pushlightuserdata(L, ngx_stream_lua_lightudata_mask(
+ raw_req_socket_metatable_key));
+ lua_rawget(L, LUA_REGISTRYINDEX);
  lua_rawset(L, LUA_REGISTRYINDEX);
  /* }}} */
 
@@ -6759,4 +6767,26 @@ ngx_stream_lua_cleanup_conn_pools(lua_State *L)
  lua_pop(L, 1);
 }
 
+
+int
+ngx_stream_lua_ffi_socket_tcp_getfd(ngx_stream_lua_request_t *r,
+ngx_stream_lua_socket_tcp_upstream_t *u, const char **errmsg)
+{
+ int fd;
+
+ *errmsg = NULL;
+
+ if (u == NULL || u->peer.connection == NULL) {
+ *errmsg = "closed";
+ return -1;
+ }
+
+ fd = u->peer.connection->fd;
+ if (fd == -1) {
+ *errmsg = "faked connection";
+ }
+
+ return fd;
+}
+
 /* vi:set ft=c ts=4 sw=4 et fdm=marker: */

t/058-tcp-socket.t

@@ -4,7 +4,7 @@ use Test::Nginx::Socket::Lua::Stream;
 
 repeat_each(2);
 
-plan tests => repeat_each() * 224;
+plan tests => repeat_each() * 227;
 
 our $HtmlDir = html_dir;
 
@@ -3601,3 +3601,69 @@ connected: 1
 setkeepalive: 1
 --- no_error_log
 [error]
+
+
+
+=== TEST 69: getfd()
+--- stream_server_config
+ content_by_lua_block {
+ local sock = ngx.socket.tcp()
+ local ok, err = sock:connect("127.0.0.1", $TEST_NGINX_SERVER_PORT)
+ if not ok then
+ ngx.say("failed to connect: ", err)
+ return
+ end
+
+ ngx.say("connected: ", ok)
+ ngx.say("fd: ", sock:getfd())
+
+ local req = "GET /foo HTTP/1.0\r\nHost: localhost\r\nConnection: close\r\n\r\n"
+ -- req = "OK"
+
+ local bytes, err = sock:send(req)
+ if not bytes then
+ ngx.say("failed to send request: ", err)
+ return
+ end
+
+ ngx.say("request sent: ", bytes)
+
+ while true do
+ local line, err, part = sock:receive()
+ if line then
+ ngx.say("received: ", line)
+
+ else
+ ngx.say("failed to receive a line: ", err, " [", part, "]")
+ break
+ end
+ end
+
+ ok, err = sock:close()
+ ngx.say("close: ", ok, " ", err)
+ }
+
+--- config
+ server_tokens off;
+
+ location /foo {
+ content_by_lua_block { ngx.say("foo") }
+ more_clear_headers Date;
+ }
+
+--- stream_response eval
+qr{connected: 1
+fd: \d+
+request sent: 57
+received: HTTP/1.1 200 OK
+received: Server: nginx
+received: Content-Type: text/plain
+received: Content-Length: 4
+received: Connection: close
+received: 
+received: foo
+failed to receive a line: closed \[\]
+close: 1 nil
+}
+--- no_error_log
+[error]

t/067-req-socket.t

@@ -1000,3 +1000,38 @@ bad argument #2 to '?' (bad max argument)
 send data after read side timeout
 --- error_log
 receiveany unexpected err: timeout
+
+
+
+=== TEST 17: getfd()
+--- stream_server_config
+ content_by_lua_block {
+ local sock, err = ngx.req.socket()
+ if sock then
+ ngx.say("got the request socket")
+ else
+ ngx.say("failed to get the request socket: ", err)
+ return
+ end
+
+ ngx.say("fd: ", sock:getfd())
+ for i = 1, 3 do
+ local data, err, part = sock:receive(5)
+ if data then
+ ngx.say("received: ", data)
+ else
+ ngx.say("failed to receive: ", err, " [", part, "]")
+ end
+ end
+ }
+--- stream_request chomp
+hello world! my
+--- stream_response eval
+qr/got the request socket
+fd: \d+
+received: hello
+received: worl
+received: d! my
+/
+--- no_error_log
+[error]