diff options
Diffstat (limited to 'src/backend/port/pipe.c')
-rw-r--r-- | src/backend/port/pipe.c | 95 |
1 files changed, 95 insertions, 0 deletions
diff --git a/src/backend/port/pipe.c b/src/backend/port/pipe.c new file mode 100644 index 00000000000..eeed3fc2e11 --- /dev/null +++ b/src/backend/port/pipe.c @@ -0,0 +1,95 @@ +/*------------------------------------------------------------------------- + * + * pipe.c + * pipe() + * + * Copyright (c) 1996-2011, PostgreSQL Global Development Group + * + * This is a replacement version of pipe for Win32 which allows + * returned handles to be used in select(). Note that read/write calls + * must be replaced with recv/send. + * + * IDENTIFICATION + * src/backend/port/pipe.c + * + *------------------------------------------------------------------------- + */ + +#include "postgres.h" + +#ifdef WIN32 +int +pgpipe(int handles[2]) +{ + SOCKET s; + struct sockaddr_in serv_addr; + int len = sizeof(serv_addr); + + handles[0] = handles[1] = INVALID_SOCKET; + + if ((s = socket(AF_INET, SOCK_STREAM, 0)) == INVALID_SOCKET) + { + ereport(LOG, (errmsg_internal("pgpipe failed to create socket: %ui", WSAGetLastError()))); + return -1; + } + + memset((void *) &serv_addr, 0, sizeof(serv_addr)); + serv_addr.sin_family = AF_INET; + serv_addr.sin_port = htons(0); + serv_addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK); + if (bind(s, (SOCKADDR *) & serv_addr, len) == SOCKET_ERROR) + { + ereport(LOG, (errmsg_internal("pgpipe failed to bind: %ui", WSAGetLastError()))); + closesocket(s); + return -1; + } + if (listen(s, 1) == SOCKET_ERROR) + { + ereport(LOG, (errmsg_internal("pgpipe failed to listen: %ui", WSAGetLastError()))); + closesocket(s); + return -1; + } + if (getsockname(s, (SOCKADDR *) & serv_addr, &len) == SOCKET_ERROR) + { + ereport(LOG, (errmsg_internal("pgpipe failed to getsockname: %ui", WSAGetLastError()))); + closesocket(s); + return -1; + } + if ((handles[1] = socket(PF_INET, SOCK_STREAM, 0)) == INVALID_SOCKET) + { + ereport(LOG, (errmsg_internal("pgpipe failed to create socket 2: %ui", WSAGetLastError()))); + closesocket(s); + return -1; + } + + if (connect(handles[1], (SOCKADDR *) & serv_addr, len) == SOCKET_ERROR) + { + ereport(LOG, (errmsg_internal("pgpipe failed to connect socket: %ui", WSAGetLastError()))); + closesocket(s); + return -1; + } + if ((handles[0] = accept(s, (SOCKADDR *) & serv_addr, &len)) == INVALID_SOCKET) + { + ereport(LOG, (errmsg_internal("pgpipe failed to accept socket: %ui", WSAGetLastError()))); + closesocket(handles[1]); + handles[1] = INVALID_SOCKET; + closesocket(s); + return -1; + } + closesocket(s); + return 0; +} + + +int +piperead(int s, char *buf, int len) +{ + int ret = recv(s, buf, len, 0); + + if (ret < 0 && WSAGetLastError() == WSAECONNRESET) + /* EOF on the pipe! (win32 socket based implementation) */ + ret = 0; + return ret; +} + +#endif |