include pipe2/socketpair compat for macOS

3 files changed, 69 insertions, 1 deletions

diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index f4277bd..0b2f7fa 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -346,7 +346,16 @@ add_test(timingsafe timingsafe)
 
 # tlstest
 if(NOT CMAKE_HOST_WIN32 AND NOT CMAKE_SYSTEM_NAME MATCHES "MINGW")
-add_executable(tlstest tlstest.c)
+
+set(TLSTEST_SRC tlstest.c)
+check_function_exists(pipe2 HAVE_PIPE2)
+if(HAVE_PIPE2)
+        add_definitions(-DHAVE_PIPE2)
+else()
+        set(TLSTEST_SRC ${TLSTEST_SRC} compat/pipe2.c)
+endif()
+
+add_executable(tlstest ${TLSTEST_SRC})
 target_link_libraries(tlstest ${TESTS_LIBS})
 if(NOT MSVC)
         add_test(tlstest ${CMAKE_CURRENT_SOURCE_DIR}/tlstest.sh)
diff --git a/tests/Makefile.am b/tests/Makefile.am
index ab1f2f1..57347d6 100644
--- a/tests/Makefile.am
+++ b/tests/Makefile.am
@@ -337,6 +337,9 @@ if !HOST_WIN
 TESTS += tlstest.sh
 check_PROGRAMS += tlstest
 tlstest_SOURCES = tlstest.c
+if !HAVE_PIPE2
+tlstest_SOURCES += compat/pipe2.c
+endif
 EXTRA_DIST += tlstest.sh tlstest.bat
 endif
 
diff --git a/tests/compat/pipe2.c b/tests/compat/pipe2.c
new file mode 100644
index 0000000..e2aaa28
--- /dev/null
+++ b/tests/compat/pipe2.c
@@ -0,0 +1,56 @@
+/*
+ * Public domain
+ */
+
+#include <fcntl.h>
+#include <unistd.h>
+#include <sys/socket.h>
+#undef socketpair
+
+static int setfd(int fd, int flag)
+{
+        int flags = fcntl(fd, F_GETFD);
+        flags |= flag;
+        return fcntl(fd, F_SETFD, flags);
+}
+
+static int setfl(int fd, int flag)
+{
+        int flags = fcntl(fd, F_GETFL);
+        flags |= flag;
+        return fcntl(fd, F_SETFL, flags);
+}
+
+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);
+                }
+                if (flags & O_CLOEXEC) {
+                        setfd(fildes[0], FD_CLOEXEC);
+                        setfd(fildes[1], FD_CLOEXEC);
+                }
+        }
+        return rc;
+}
+
+int bsd_socketpair(int domain, int type, int protocol, int socket_vector[2])
+{
+        int flags = type & ~0xf;
+        type &= 0xf;
+        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);
+                }
+                if (flags & SOCK_CLOEXEC) {
+                        setfd(socket_vector[0], FD_CLOEXEC);
+                        setfd(socket_vector[1], FD_CLOEXEC);
+                }
+        }
+        return rc;
+}