mirror of
https://github.com/postgres/postgres.git
synced 2025-05-17 06:41:24 +03:00
96 lines
2.5 KiB
C
96 lines
2.5 KiB
C
/*-------------------------------------------------------------------------
|
|
*
|
|
* pipe.c
|
|
* pipe()
|
|
*
|
|
* Copyright (c) 1996-2007, 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
|
|
* $PostgreSQL: pgsql/src/port/pipe.c,v 1.12 2007/01/05 22:20:03 momjian Exp $
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
|
|
#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
|