Fix missing passing of the IO object

KJ Tsanaktsidis · KJ Tsanaktsidis · commit 3fb1441bee88 · 2023-05-30T13:50:30.000+08:00
diff --git a/ext/socket/udpsocket.c b/ext/socket/udpsocket.c
@@ -88,6 +88,7 @@ udp_connect(VALUE sock, VALUE host, VALUE port)
 
     GetOpenFile(sock, fptr);
     arg.res = rsock_addrinfo(host, port, rsock_fd_family(fptr->fd), SOCK_DGRAM, 0);
+    arg.io = sock;
     ret = rb_ensure(udp_connect_internal, (VALUE)&arg,
                     rsock_freeaddrinfo, (VALUE)arg.res);
     if (!ret) rsock_sys_fail_host_port("connect(2)", host, port);
@@ -134,6 +135,7 @@ udp_bind(VALUE sock, VALUE host, VALUE port)
 
     GetOpenFile(sock, fptr);
     arg.res = rsock_addrinfo(host, port, rsock_fd_family(fptr->fd), SOCK_DGRAM, 0);
+    arg.io = sock;
     ret = rb_ensure(udp_bind_internal, (VALUE)&arg,
                     rsock_freeaddrinfo, (VALUE)arg.res);
     if (!ret) rsock_sys_fail_host_port("bind(2)", host, port);
diff --git a/ext/socket/unixsocket.c b/ext/socket/unixsocket.c
@@ -73,7 +73,7 @@ rsock_init_unixsock(VALUE io, VALUE path, int server)
     }
 
     rsock_init_sock(io, fd);
-    RB_IO_OPEN(io, fptr);
+    GetOpenFile(io, fptr);
 
     if (server) {
         status = bind(fd, (struct sockaddr*)&sockaddr, sockaddrlen);
@@ -83,6 +83,7 @@ rsock_init_unixsock(VALUE io, VALUE path, int server)
         struct unixsock_arg arg;
         arg.sockaddr = &sockaddr;
         arg.sockaddrlen = sockaddrlen;
+        arg.io = io;
 
         status = (int)rb_protect(unixsock_connect_internal, (VALUE)&arg, &error_tag);
 
