feature: add reuseport for binding local port for udp cosocket.

Author: zhuizhuhaomeng

src/ngx_stream_lua_socket_tcp.c

@@ -999,7 +999,7 @@ ngx_stream_lua_socket_tcp_bind(lua_State *L)
     luaL_checktype(L, 1, LUA_TTABLE);
 
     text = (u_char *) luaL_checklstring(L, 2, &len);
-    local = ngx_stream_lua_parse_addr(L, text, len);
+    local = ngx_stream_lua_parse_addr(L, text, len, 0);
     if (local == NULL) {
         lua_pushnil(L);
         lua_pushfstring(L, "bad address");

src/ngx_stream_lua_socket_udp.c

@@ -262,11 +262,12 @@ ngx_stream_lua_socket_udp_bind(lua_State *L)
     u_char                 *text;
     size_t                  len;
     ngx_addr_t             *local;
+    ngx_int_t               port = 0;
 
     ngx_stream_lua_request_t    *r;
 
     n = lua_gettop(L);
-    if (n != 2) {
+    if (n != 2 && n != 3) {
         return luaL_error(L, "expecting 2 arguments, but got %d",
                           lua_gettop(L));
     }
@@ -286,7 +287,13 @@ ngx_stream_lua_socket_udp_bind(lua_State *L)
     luaL_checktype(L, 1, LUA_TTABLE);
 
     text = (u_char *) luaL_checklstring(L, 2, &len);
-    local = ngx_stream_lua_parse_addr(L, text, len);
+    if (n == 3) {
+        port = luaL_checkinteger(L, 3);
+        if (port <= 0 || port > 65535)
+            return luaL_error(L, "bad port number: %d", port);
+    }
+
+    local = ngx_stream_lua_parse_addr(L, text, len, port);
     if (local == NULL) {
         lua_pushnil(L);
         lua_pushfstring(L, "bad address");
@@ -499,6 +506,7 @@ ngx_stream_lua_socket_udp_setpeername(lua_State *L)
         u->resolved->socklen = url.addrs[0].socklen;
         u->resolved->naddrs = 1;
         u->resolved->host = url.addrs[0].name;
+        u->resolved->port = ngx_inet_get_port(url.addrs[0].sockaddr);
 
     } else {
         u->resolved->host = host;
@@ -1640,7 +1648,7 @@ ngx_stream_lua_udp_connect_set_transparent(ngx_stream_lua_udp_connection_t *uc,
 
 #endif /* SO_BINDANY */
 
-return NGX_OK;
+    return NGX_OK;
 }
 #endif
 
@@ -1742,7 +1750,7 @@ ngx_stream_lua_udp_connect(ngx_stream_lua_socket_udp_upstream_t *u)
 #endif
 
 #if (NGX_HAVE_IP_BIND_ADDRESS_NO_PORT || NGX_LINUX)
-        port = u->resolved->port;
+        port = ngx_inet_get_port(local->sockaddr);
 #endif
 
 #if (NGX_HAVE_IP_BIND_ADDRESS_NO_PORT)

src/ngx_stream_lua_socket_udp.h

@@ -44,7 +44,7 @@ typedef struct {
     struct sockaddr          *sockaddr;
     socklen_t                 socklen;
 #if (NGX_HAVE_TRANSPARENT_PROXY)
-    unsigned                   transparent:1;
+    unsigned                  transparent:1;
 #endif
     ngx_str_t                 server;
     ngx_log_t                 log;

src/ngx_stream_lua_util.c

@@ -3701,7 +3701,8 @@ ngx_stream_lua_set_sa_restart(ngx_log_t *log)
 
 
 ngx_addr_t *
-ngx_stream_lua_parse_addr(lua_State *L, u_char *text, size_t len)
+ngx_stream_lua_parse_addr(lua_State *L, u_char *text, size_t len,
+    in_port_t port)
 {
     ngx_addr_t           *addr;
     size_t                socklen;
@@ -3718,19 +3719,26 @@ ngx_stream_lua_parse_addr(lua_State *L, u_char *text, size_t len)
     ngx_memzero(&inaddr6, (sizeof(struct in6_addr)));
 #endif
 
-    inaddr = ngx_inet_addr(text, len);
-    if (inaddr != INADDR_NONE) {
-        family = AF_INET;
+    if (len == 7 && memcmp(text, "0.0.0.0", 7) == 0) {
+        inaddr = INADDR_NONE;
         socklen = sizeof(struct sockaddr_in);
+        family = AF_INET;
+
+    } else {
+        inaddr = ngx_inet_addr(text, len);
+        if (inaddr != INADDR_NONE) {
+            family = AF_INET;
+            socklen = sizeof(struct sockaddr_in);
 #if (NGX_HAVE_INET6)
 
-    } else if (ngx_inet6_addr(text, len, inaddr6.s6_addr) == NGX_OK) {
-        family = AF_INET6;
-        socklen = sizeof(struct sockaddr_in6);
+        } else if (ngx_inet6_addr(text, len, inaddr6.s6_addr) == NGX_OK) {
+            family = AF_INET6;
+            socklen = sizeof(struct sockaddr_in6);
 #endif
 
-    } else {
-        return NULL;
+        } else {
+            return NULL;
+        }
     }
 
     addr = lua_newuserdata(L, sizeof(ngx_addr_t) + socklen + len);
@@ -3756,6 +3764,7 @@ ngx_stream_lua_parse_addr(lua_State *L, u_char *text, size_t len)
         break;
     }
 
+    ngx_inet_set_port(addr->sockaddr, port);
     addr->name.data = (u_char *) addr->sockaddr + socklen;
     addr->name.len = len;
     ngx_memcpy(addr->name.data, text, len);

src/ngx_stream_lua_util.h

@@ -198,7 +198,8 @@ ngx_int_t ngx_stream_lua_open_and_stat_file(u_char *name,
 ngx_chain_t *ngx_stream_lua_chain_get_free_buf(ngx_log_t *log, ngx_pool_t *p,
     ngx_chain_t **free, size_t len);
 
-ngx_addr_t *ngx_stream_lua_parse_addr(lua_State *L, u_char *text, size_t len);
+ngx_addr_t *ngx_stream_lua_parse_addr(lua_State *L, u_char *text, size_t len,
+    in_port_t port);
 
 
 static ngx_inline void