add windows versions of socketpair / pipe

1 files changed, 119 insertions, 8 deletions

diff --git a/tests/compat/pipe2.c b/tests/compat/pipe2.c
index e2aaa28..7600783 100644
--- a/tests/compat/pipe2.c
+++ b/tests/compat/pipe2.c
@@ -1,12 +1,110 @@
 /*
  * Public domain
+ *
+ * pipe2/pipe/socketpair emulation
+ * Brent Cook <bcook@openbsd.org>
  */
 
+#include <errno.h>
 #include <fcntl.h>
 #include <unistd.h>
 #include <sys/socket.h>
+
 #undef socketpair
 
+#ifdef _WIN32
+
+static int setfd(int fd, int flag)
+{
+        int rc = -1;
+        if (flag & FD_CLOEXEC) {
+                HANDLE h = (HANDLE)_get_osfhandle(fd);
+                if (h != NULL)
+                        rc = SetHandleInformation(h, HANDLE_FLAG_INHERIT, 0) == 0 ? -1 : 0;
+        }
+        return rc;
+}
+
+static int setfl(int fd, int flag)
+{
+        int rc = -1;
+        if (flag & O_NONBLOCK) {
+                long mode = 1;
+                rc = ioctlsocket(fd, FIONBIO, &mode);
+        }
+        return rc;
+}
+
+int socketpair(int domain, int type, int protocol, int socket_vector[2])
+{
+        if (domain != AF_UNIX || !(type & SOCK_STREAM) || protocol != PF_UNSPEC)
+                return -1;
+
+        socket_vector[0] = -1;
+        socket_vector[1] = -1;
+
+        int listener = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+        if (listener == -1) {
+                return -1;
+        }
+
+        struct sockaddr_in addr = {
+                .sin_family = AF_INET,
+                .sin_addr.s_addr = htonl(INADDR_LOOPBACK),
+                .sin_port = 0,
+        };
+
+        int yes = 1, e;
+        if (setsockopt(listener, SOL_SOCKET, SO_REUSEADDR,
+                        (void *)&yes, sizeof yes) == -1)
+                goto err;
+
+        if (bind(listener, (struct sockaddr *)&addr, sizeof addr) != 0)
+                goto err;
+
+        memset(&addr, 0, sizeof addr);
+        socklen_t addrlen = sizeof addr;
+        if (getsockname(listener, (struct sockaddr *)&addr, &addrlen) != 0)
+                goto err;
+
+        addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
+        addr.sin_family = AF_INET;
+
+        if (listen(listener, 1) != 0)
+                goto err;
+
+        socket_vector[0] = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, 0);
+        if (socket_vector[0] == -1)
+                goto err;
+
+        if (connect(socket_vector[0], (struct sockaddr *)&addr, sizeof addr) != 0)
+                goto err;
+
+        socket_vector[1] = accept(listener, NULL, NULL);
+        if (socket_vector[1] == -1)
+                goto err;
+
+        closesocket(listener);
+        return 0;
+
+err:
+        e = WSAGetLastError();
+        closesocket(listener);
+        closesocket(socket_vector[0]);
+        closesocket(socket_vector[1]);
+        WSASetLastError(e);
+        socket_vector[0] = -1;
+        socket_vector[1] = -1;
+        return -1;
+}
+
+int pipe(int fildes[2])
+{
+        return socketpair(AF_UNIX, SOCK_STREAM | SOCK_NONBLOCK, PF_UNSPEC, fildes);
+}
+
+#else
+
 static int setfd(int fd, int flag)
 {
         int flags = fcntl(fd, F_GETFD);
@@ -20,18 +118,25 @@ static int setfl(int fd, int flag)
         flags |= flag;
         return fcntl(fd, F_SETFL, flags);
 }
+#endif
 
 int pipe2(int fildes[2], int flags)
 {
         int rc = pipe(fildes);
         if (rc == 0) {
                 if (flags & O_NONBLOCK) {
-                        setfl(fildes[0], O_NONBLOCK);
-                        setfl(fildes[1], O_NONBLOCK);
+                        rc |= setfl(fildes[0], O_NONBLOCK);
+                        rc |= setfl(fildes[1], O_NONBLOCK);
                 }
                 if (flags & O_CLOEXEC) {
-                        setfd(fildes[0], FD_CLOEXEC);
-                        setfd(fildes[1], FD_CLOEXEC);
+                        rc |= setfd(fildes[0], FD_CLOEXEC);
+                        rc |= setfd(fildes[1], FD_CLOEXEC);
+                }
+                if (rc != 0) {
+                        int e = errno;
+                        close(fildes[0]);
+                        close(fildes[1]);
+                        errno = e;
                 }
         }
         return rc;
@@ -44,12 +149,18 @@ int bsd_socketpair(int domain, int type, int protocol, int socket_vector[2])
         int rc = socketpair(domain, type, protocol, socket_vector);
         if (rc == 0) {
                 if (flags & SOCK_NONBLOCK) {
-                        setfl(socket_vector[0], O_NONBLOCK);
-                        setfl(socket_vector[1], O_NONBLOCK);
+                        rc |= setfl(socket_vector[0], O_NONBLOCK);
+                        rc |= setfl(socket_vector[1], O_NONBLOCK);
                 }
                 if (flags & SOCK_CLOEXEC) {
-                        setfd(socket_vector[0], FD_CLOEXEC);
-                        setfd(socket_vector[1], FD_CLOEXEC);
+                        rc |= setfd(socket_vector[0], FD_CLOEXEC);
+                        rc |= setfd(socket_vector[1], FD_CLOEXEC);
+                }
+                if (rc != 0) {
+                        int e = errno;
+                        close(socket_vector[0]);
+                        close(socket_vector[1]);
+                        errno = e;
                 }
         }
         return rc;