PostgreSQL Source Code  git master
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros
pqcomm.c File Reference
#include "postgres.h"
#include <signal.h>
#include <fcntl.h>
#include <grp.h>
#include <unistd.h>
#include <sys/file.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <sys/time.h>
#include <netdb.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include "common/ip.h"
#include "libpq/libpq.h"
#include "miscadmin.h"
#include "storage/ipc.h"
#include "utils/guc.h"
#include "utils/memutils.h"
Include dependency graph for pqcomm.c:

Go to the source code of this file.

Macros

#define PQ_SEND_BUFFER_SIZE   8192
 
#define PQ_RECV_BUFFER_SIZE   8192
 

Functions

static void socket_comm_reset (void)
 
static void socket_close (int code, Datum arg)
 
static void socket_set_nonblocking (bool nonblocking)
 
static int socket_flush (void)
 
static int socket_flush_if_writable (void)
 
static bool socket_is_send_pending (void)
 
static int socket_putmessage (char msgtype, const char *s, size_t len)
 
static void socket_putmessage_noblock (char msgtype, const char *s, size_t len)
 
static void socket_startcopyout (void)
 
static void socket_endcopyout (bool errorAbort)
 
static int internal_putbytes (const char *s, size_t len)
 
static int internal_flush (void)
 
void pq_init (void)
 
int StreamServerPort (int family, char *hostName, unsigned short portNumber, char *unixSocketDir, pgsocket ListenSocket[], int MaxListen)
 
int StreamConnection (pgsocket server_fd, Port *port)
 
void StreamClose (pgsocket sock)
 
void TouchSocketFiles (void)
 
void RemoveSocketFiles (void)
 
static int pq_recvbuf (void)
 
int pq_getbyte (void)
 
int pq_peekbyte (void)
 
int pq_getbyte_if_available (unsigned char *c)
 
int pq_getbytes (char *s, size_t len)
 
static int pq_discardbytes (size_t len)
 
int pq_getstring (StringInfo s)
 
void pq_startmsgread (void)
 
void pq_endmsgread (void)
 
bool pq_is_reading_msg (void)
 
int pq_getmessage (StringInfo s, int maxlen)
 
int pq_putbytes (const char *s, size_t len)
 
int pq_getkeepalivesidle (Port *port)
 
int pq_setkeepalivesidle (int idle, Port *port)
 
int pq_getkeepalivesinterval (Port *port)
 
int pq_setkeepalivesinterval (int interval, Port *port)
 
int pq_getkeepalivescount (Port *port)
 
int pq_setkeepalivescount (int count, Port *port)
 

Variables

int Unix_socket_permissions
 
char * Unix_socket_group
 
static Listsock_paths = NIL
 
static char * PqSendBuffer
 
static int PqSendBufferSize
 
static int PqSendPointer
 
static int PqSendStart
 
static char PqRecvBuffer [PQ_RECV_BUFFER_SIZE]
 
static int PqRecvPointer
 
static int PqRecvLength
 
static bool PqCommBusy
 
static bool PqCommReadingMsg
 
static bool DoingCopyOut
 
static PQcommMethods PqCommSocketMethods
 
PQcommMethodsPqCommMethods = &PqCommSocketMethods
 
WaitEventSetFeBeWaitSet
 

Macro Definition Documentation

#define PQ_RECV_BUFFER_SIZE   8192

Definition at line 116 of file pqcomm.c.

Referenced by pq_recvbuf().

#define PQ_SEND_BUFFER_SIZE   8192

Definition at line 115 of file pqcomm.c.

Referenced by pq_init(), and StreamConnection().

Function Documentation

static int internal_flush ( void  )
static

Definition at line 1369 of file pqcomm.c.

References ClientConnectionLost, COMMERROR, EAGAIN, EINTR, ereport, errcode_for_socket_access(), errmsg(), EWOULDBLOCK, InterruptPending, MyProcPort, PqSendBuffer, PqSendPointer, PqSendStart, and secure_write().

Referenced by internal_putbytes(), socket_flush(), and socket_flush_if_writable().

1370 {
1371  static int last_reported_send_errno = 0;
1372 
1373  char *bufptr = PqSendBuffer + PqSendStart;
1374  char *bufend = PqSendBuffer + PqSendPointer;
1375 
1376  while (bufptr < bufend)
1377  {
1378  int r;
1379 
1380  r = secure_write(MyProcPort, bufptr, bufend - bufptr);
1381 
1382  if (r <= 0)
1383  {
1384  if (errno == EINTR)
1385  continue; /* Ok if we were interrupted */
1386 
1387  /*
1388  * Ok if no data writable without blocking, and the socket is in
1389  * non-blocking mode.
1390  */
1391  if (errno == EAGAIN ||
1392  errno == EWOULDBLOCK)
1393  {
1394  return 0;
1395  }
1396 
1397  /*
1398  * Careful: an ereport() that tries to write to the client would
1399  * cause recursion to here, leading to stack overflow and core
1400  * dump! This message must go *only* to the postmaster log.
1401  *
1402  * If a client disconnects while we're in the midst of output, we
1403  * might write quite a bit of data before we get to a safe query
1404  * abort point. So, suppress duplicate log messages.
1405  */
1406  if (errno != last_reported_send_errno)
1407  {
1408  last_reported_send_errno = errno;
1411  errmsg("could not send data to client: %m")));
1412  }
1413 
1414  /*
1415  * We drop the buffered data anyway so that processing can
1416  * continue, even though we'll probably quit soon. We also set a
1417  * flag that'll cause the next CHECK_FOR_INTERRUPTS to terminate
1418  * the connection.
1419  */
1420  PqSendStart = PqSendPointer = 0;
1422  InterruptPending = 1;
1423  return EOF;
1424  }
1425 
1426  last_reported_send_errno = 0; /* reset after any successful send */
1427  bufptr += r;
1428  PqSendStart += r;
1429  }
1430 
1431  PqSendStart = PqSendPointer = 0;
1432  return 0;
1433 }
#define EWOULDBLOCK
Definition: win32.h:301
struct Port * MyProcPort
Definition: globals.c:40
static char * PqSendBuffer
Definition: pqcomm.c:118
static int PqSendStart
Definition: pqcomm.c:121
#define EAGAIN
Definition: win32.h:293
volatile bool ClientConnectionLost
Definition: globals.c:32
static int PqSendPointer
Definition: pqcomm.c:120
#define COMMERROR
Definition: elog.h:30
ssize_t secure_write(Port *port, void *ptr, size_t len)
Definition: be-secure.c:240
#define ereport(elevel, rest)
Definition: elog.h:122
int errcode_for_socket_access(void)
Definition: elog.c:669
#define EINTR
Definition: win32.h:295
volatile bool InterruptPending
Definition: globals.c:29
int errmsg(const char *fmt,...)
Definition: elog.c:797
static int internal_putbytes ( const char *  s,
size_t  len 
)
static

Definition at line 1316 of file pqcomm.c.

References internal_flush(), PqSendBuffer, PqSendBufferSize, PqSendPointer, and socket_set_nonblocking().

Referenced by pq_putbytes(), and socket_putmessage().

1317 {
1318  size_t amount;
1319 
1320  while (len > 0)
1321  {
1322  /* If buffer is full, then flush it out */
1324  {
1325  socket_set_nonblocking(false);
1326  if (internal_flush())
1327  return EOF;
1328  }
1329  amount = PqSendBufferSize - PqSendPointer;
1330  if (amount > len)
1331  amount = len;
1332  memcpy(PqSendBuffer + PqSendPointer, s, amount);
1333  PqSendPointer += amount;
1334  s += amount;
1335  len -= amount;
1336  }
1337  return 0;
1338 }
static char * PqSendBuffer
Definition: pqcomm.c:118
static int internal_flush(void)
Definition: pqcomm.c:1369
static void socket_set_nonblocking(bool nonblocking)
Definition: pqcomm.c:869
static int PqSendPointer
Definition: pqcomm.c:120
static int PqSendBufferSize
Definition: pqcomm.c:119
static int pq_discardbytes ( size_t  len)
static

Definition at line 1075 of file pqcomm.c.

References Assert, pq_recvbuf(), PqCommReadingMsg, PqRecvLength, and PqRecvPointer.

Referenced by pq_getmessage().

1076 {
1077  size_t amount;
1078 
1080 
1081  while (len > 0)
1082  {
1083  while (PqRecvPointer >= PqRecvLength)
1084  {
1085  if (pq_recvbuf()) /* If nothing in buffer, then recv some */
1086  return EOF; /* Failed to recv data */
1087  }
1088  amount = PqRecvLength - PqRecvPointer;
1089  if (amount > len)
1090  amount = len;
1091  PqRecvPointer += amount;
1092  len -= amount;
1093  }
1094  return 0;
1095 }
static int PqRecvLength
Definition: pqcomm.c:125
static int pq_recvbuf(void)
Definition: pqcomm.c:886
#define Assert(condition)
Definition: c.h:670
static int PqRecvPointer
Definition: pqcomm.c:124
static bool PqCommReadingMsg
Definition: pqcomm.c:131
void pq_endmsgread ( void  )

Definition at line 1181 of file pqcomm.c.

References Assert, and PqCommReadingMsg.

Referenced by CopyFrom(), ProcessRepliesIfAny(), ProcessStartupPacket(), and SocketBackend().

1182 {
1184 
1185  PqCommReadingMsg = false;
1186 }
#define Assert(condition)
Definition: c.h:670
static bool PqCommReadingMsg
Definition: pqcomm.c:131
int pq_getbyte ( void  )

Definition at line 947 of file pqcomm.c.

References Assert, pq_recvbuf(), PqCommReadingMsg, PqRecvBuffer, PqRecvLength, and PqRecvPointer.

Referenced by CopyGetData(), recv_password_packet(), and SocketBackend().

948 {
950 
951  while (PqRecvPointer >= PqRecvLength)
952  {
953  if (pq_recvbuf()) /* If nothing in buffer, then recv some */
954  return EOF; /* Failed to recv data */
955  }
956  return (unsigned char) PqRecvBuffer[PqRecvPointer++];
957 }
static char PqRecvBuffer[PQ_RECV_BUFFER_SIZE]
Definition: pqcomm.c:123
static int PqRecvLength
Definition: pqcomm.c:125
static int pq_recvbuf(void)
Definition: pqcomm.c:886
#define Assert(condition)
Definition: c.h:670
static int PqRecvPointer
Definition: pqcomm.c:124
static bool PqCommReadingMsg
Definition: pqcomm.c:131
int pq_getbyte_if_available ( unsigned char *  c)

Definition at line 987 of file pqcomm.c.

References Assert, COMMERROR, EAGAIN, EINTR, ereport, errcode_for_socket_access(), errmsg(), EWOULDBLOCK, MyProcPort, PqCommReadingMsg, PqRecvBuffer, PqRecvLength, PqRecvPointer, secure_read(), and socket_set_nonblocking().

Referenced by ProcessRepliesIfAny().

988 {
989  int r;
990 
992 
994  {
996  return 1;
997  }
998 
999  /* Put the socket into non-blocking mode */
1000  socket_set_nonblocking(true);
1001 
1002  r = secure_read(MyProcPort, c, 1);
1003  if (r < 0)
1004  {
1005  /*
1006  * Ok if no data available without blocking or interrupted (though
1007  * EINTR really shouldn't happen with a non-blocking socket). Report
1008  * other errors.
1009  */
1010  if (errno == EAGAIN || errno == EWOULDBLOCK || errno == EINTR)
1011  r = 0;
1012  else
1013  {
1014  /*
1015  * Careful: an ereport() that tries to write to the client would
1016  * cause recursion to here, leading to stack overflow and core
1017  * dump! This message must go *only* to the postmaster log.
1018  */
1021  errmsg("could not receive data from client: %m")));
1022  r = EOF;
1023  }
1024  }
1025  else if (r == 0)
1026  {
1027  /* EOF detected */
1028  r = EOF;
1029  }
1030 
1031  return r;
1032 }
#define EWOULDBLOCK
Definition: win32.h:301
static char PqRecvBuffer[PQ_RECV_BUFFER_SIZE]
Definition: pqcomm.c:123
struct Port * MyProcPort
Definition: globals.c:40
ssize_t secure_read(Port *port, void *ptr, size_t len)
Definition: be-secure.c:137
#define EAGAIN
Definition: win32.h:293
static void socket_set_nonblocking(bool nonblocking)
Definition: pqcomm.c:869
static int PqRecvLength
Definition: pqcomm.c:125
char * c
#define COMMERROR
Definition: elog.h:30
#define ereport(elevel, rest)
Definition: elog.h:122
int errcode_for_socket_access(void)
Definition: elog.c:669
#define EINTR
Definition: win32.h:295
#define Assert(condition)
Definition: c.h:670
int errmsg(const char *fmt,...)
Definition: elog.c:797
static int PqRecvPointer
Definition: pqcomm.c:124
static bool PqCommReadingMsg
Definition: pqcomm.c:131
int pq_getbytes ( char *  s,
size_t  len 
)

Definition at line 1041 of file pqcomm.c.

References Assert, pq_recvbuf(), PqCommReadingMsg, PqRecvBuffer, PqRecvLength, and PqRecvPointer.

Referenced by CopyGetData(), GetOldFunctionMessage(), pq_getmessage(), and ProcessStartupPacket().

1042 {
1043  size_t amount;
1044 
1046 
1047  while (len > 0)
1048  {
1049  while (PqRecvPointer >= PqRecvLength)
1050  {
1051  if (pq_recvbuf()) /* If nothing in buffer, then recv some */
1052  return EOF; /* Failed to recv data */
1053  }
1054  amount = PqRecvLength - PqRecvPointer;
1055  if (amount > len)
1056  amount = len;
1057  memcpy(s, PqRecvBuffer + PqRecvPointer, amount);
1058  PqRecvPointer += amount;
1059  s += amount;
1060  len -= amount;
1061  }
1062  return 0;
1063 }
static char PqRecvBuffer[PQ_RECV_BUFFER_SIZE]
Definition: pqcomm.c:123
static int PqRecvLength
Definition: pqcomm.c:125
static int pq_recvbuf(void)
Definition: pqcomm.c:886
#define Assert(condition)
Definition: c.h:670
static int PqRecvPointer
Definition: pqcomm.c:124
static bool PqCommReadingMsg
Definition: pqcomm.c:131
int pq_getkeepalivescount ( Port port)

Definition at line 1823 of file pqcomm.c.

References SockAddr::addr, Port::default_keepalives_count, elog, IS_AF_UNIX, Port::keepalives_count, Port::laddr, LOG, NULL, and Port::sock.

Referenced by pq_setkeepalivescount(), and show_tcp_keepalives_count().

1824 {
1825 #ifdef TCP_KEEPCNT
1826  if (port == NULL || IS_AF_UNIX(port->laddr.addr.ss_family))
1827  return 0;
1828 
1829  if (port->keepalives_count != 0)
1830  return port->keepalives_count;
1831 
1832  if (port->default_keepalives_count == 0)
1833  {
1834  ACCEPT_TYPE_ARG3 size = sizeof(port->default_keepalives_count);
1835 
1836  if (getsockopt(port->sock, IPPROTO_TCP, TCP_KEEPCNT,
1837  (char *) &port->default_keepalives_count,
1838  &size) < 0)
1839  {
1840  elog(LOG, "getsockopt(TCP_KEEPCNT) failed: %m");
1841  port->default_keepalives_count = -1; /* don't know */
1842  }
1843  }
1844 
1845  return port->default_keepalives_count;
1846 #else
1847  return 0;
1848 #endif
1849 }
struct sockaddr_storage addr
Definition: pqcomm.h:64
#define LOG
Definition: elog.h:26
int default_keepalives_count
Definition: libpq-be.h:164
pgsocket sock
Definition: libpq-be.h:120
int keepalives_count
Definition: libpq-be.h:167
#define IS_AF_UNIX(fam)
Definition: ip.h:24
SockAddr laddr
Definition: libpq-be.h:123
#define NULL
Definition: c.h:226
#define elog
Definition: elog.h:219
int pq_getkeepalivesidle ( Port port)

Definition at line 1643 of file pqcomm.c.

References SockAddr::addr, Port::default_keepalives_idle, elog, IS_AF_UNIX, Port::keepalives_idle, Port::laddr, LOG, NULL, and Port::sock.

Referenced by pq_setkeepalivesidle(), and show_tcp_keepalives_idle().

1644 {
1645 #if defined(TCP_KEEPIDLE) || defined(TCP_KEEPALIVE) || defined(WIN32)
1646  if (port == NULL || IS_AF_UNIX(port->laddr.addr.ss_family))
1647  return 0;
1648 
1649  if (port->keepalives_idle != 0)
1650  return port->keepalives_idle;
1651 
1652  if (port->default_keepalives_idle == 0)
1653  {
1654 #ifndef WIN32
1655  ACCEPT_TYPE_ARG3 size = sizeof(port->default_keepalives_idle);
1656 
1657 #ifdef TCP_KEEPIDLE
1658  if (getsockopt(port->sock, IPPROTO_TCP, TCP_KEEPIDLE,
1659  (char *) &port->default_keepalives_idle,
1660  &size) < 0)
1661  {
1662  elog(LOG, "getsockopt(TCP_KEEPIDLE) failed: %m");
1663  port->default_keepalives_idle = -1; /* don't know */
1664  }
1665 #else
1666  if (getsockopt(port->sock, IPPROTO_TCP, TCP_KEEPALIVE,
1667  (char *) &port->default_keepalives_idle,
1668  &size) < 0)
1669  {
1670  elog(LOG, "getsockopt(TCP_KEEPALIVE) failed: %m");
1671  port->default_keepalives_idle = -1; /* don't know */
1672  }
1673 #endif /* TCP_KEEPIDLE */
1674 #else /* WIN32 */
1675  /* We can't get the defaults on Windows, so return "don't know" */
1676  port->default_keepalives_idle = -1;
1677 #endif /* WIN32 */
1678  }
1679 
1680  return port->default_keepalives_idle;
1681 #else
1682  return 0;
1683 #endif
1684 }
int keepalives_idle
Definition: libpq-be.h:165
struct sockaddr_storage addr
Definition: pqcomm.h:64
#define LOG
Definition: elog.h:26
pgsocket sock
Definition: libpq-be.h:120
#define IS_AF_UNIX(fam)
Definition: ip.h:24
SockAddr laddr
Definition: libpq-be.h:123
int default_keepalives_idle
Definition: libpq-be.h:162
#define NULL
Definition: c.h:226
#define elog
Definition: elog.h:219
int pq_getkeepalivesinterval ( Port port)

Definition at line 1742 of file pqcomm.c.

References SockAddr::addr, Port::default_keepalives_interval, elog, IS_AF_UNIX, Port::keepalives_interval, Port::laddr, LOG, NULL, and Port::sock.

Referenced by pq_setkeepalivesinterval(), and show_tcp_keepalives_interval().

1743 {
1744 #if defined(TCP_KEEPINTVL) || defined(SIO_KEEPALIVE_VALS)
1745  if (port == NULL || IS_AF_UNIX(port->laddr.addr.ss_family))
1746  return 0;
1747 
1748  if (port->keepalives_interval != 0)
1749  return port->keepalives_interval;
1750 
1751  if (port->default_keepalives_interval == 0)
1752  {
1753 #ifndef WIN32
1754  ACCEPT_TYPE_ARG3 size = sizeof(port->default_keepalives_interval);
1755 
1756  if (getsockopt(port->sock, IPPROTO_TCP, TCP_KEEPINTVL,
1757  (char *) &port->default_keepalives_interval,
1758  &size) < 0)
1759  {
1760  elog(LOG, "getsockopt(TCP_KEEPINTVL) failed: %m");
1761  port->default_keepalives_interval = -1; /* don't know */
1762  }
1763 #else
1764  /* We can't get the defaults on Windows, so return "don't know" */
1765  port->default_keepalives_interval = -1;
1766 #endif /* WIN32 */
1767  }
1768 
1769  return port->default_keepalives_interval;
1770 #else
1771  return 0;
1772 #endif
1773 }
struct sockaddr_storage addr
Definition: pqcomm.h:64
#define LOG
Definition: elog.h:26
int default_keepalives_interval
Definition: libpq-be.h:163
pgsocket sock
Definition: libpq-be.h:120
#define IS_AF_UNIX(fam)
Definition: ip.h:24
SockAddr laddr
Definition: libpq-be.h:123
#define NULL
Definition: c.h:226
#define elog
Definition: elog.h:219
int keepalives_interval
Definition: libpq-be.h:166
int pq_getmessage ( StringInfo  s,
int  maxlen 
)

Definition at line 1219 of file pqcomm.c.

References Assert, COMMERROR, StringInfoData::data, enlargeStringInfo(), ereport, errcode(), errmsg(), StringInfoData::len, PG_CATCH, PG_END_TRY, PG_RE_THROW, PG_TRY, pq_discardbytes(), pq_getbytes(), PqCommReadingMsg, and resetStringInfo().

Referenced by CopyGetData(), ProcessRepliesIfAny(), recv_password_packet(), and SocketBackend().

1220 {
1221  int32 len;
1222 
1224 
1225  resetStringInfo(s);
1226 
1227  /* Read message length word */
1228  if (pq_getbytes((char *) &len, 4) == EOF)
1229  {
1231  (errcode(ERRCODE_PROTOCOL_VIOLATION),
1232  errmsg("unexpected EOF within message length word")));
1233  return EOF;
1234  }
1235 
1236  len = ntohl(len);
1237 
1238  if (len < 4 ||
1239  (maxlen > 0 && len > maxlen))
1240  {
1242  (errcode(ERRCODE_PROTOCOL_VIOLATION),
1243  errmsg("invalid message length")));
1244  return EOF;
1245  }
1246 
1247  len -= 4; /* discount length itself */
1248 
1249  if (len > 0)
1250  {
1251  /*
1252  * Allocate space for message. If we run out of room (ridiculously
1253  * large message), we will elog(ERROR), but we want to discard the
1254  * message body so as not to lose communication sync.
1255  */
1256  PG_TRY();
1257  {
1258  enlargeStringInfo(s, len);
1259  }
1260  PG_CATCH();
1261  {
1262  if (pq_discardbytes(len) == EOF)
1264  (errcode(ERRCODE_PROTOCOL_VIOLATION),
1265  errmsg("incomplete message from client")));
1266 
1267  /* we discarded the rest of the message so we're back in sync. */
1268  PqCommReadingMsg = false;
1269  PG_RE_THROW();
1270  }
1271  PG_END_TRY();
1272 
1273  /* And grab the message */
1274  if (pq_getbytes(s->data, len) == EOF)
1275  {
1277  (errcode(ERRCODE_PROTOCOL_VIOLATION),
1278  errmsg("incomplete message from client")));
1279  return EOF;
1280  }
1281  s->len = len;
1282  /* Place a trailing null per StringInfo convention */
1283  s->data[len] = '\0';
1284  }
1285 
1286  /* finished reading the message. */
1287  PqCommReadingMsg = false;
1288 
1289  return 0;
1290 }
static int pq_discardbytes(size_t len)
Definition: pqcomm.c:1075
int errcode(int sqlerrcode)
Definition: elog.c:575
signed int int32
Definition: c.h:253
int pq_getbytes(char *s, size_t len)
Definition: pqcomm.c:1041
#define COMMERROR
Definition: elog.h:30
void enlargeStringInfo(StringInfo str, int needed)
Definition: stringinfo.c:277
void resetStringInfo(StringInfo str)
Definition: stringinfo.c:94
#define ereport(elevel, rest)
Definition: elog.h:122
#define PG_CATCH()
Definition: elog.h:293
#define Assert(condition)
Definition: c.h:670
#define PG_RE_THROW()
Definition: elog.h:314
int errmsg(const char *fmt,...)
Definition: elog.c:797
static bool PqCommReadingMsg
Definition: pqcomm.c:131
#define PG_TRY()
Definition: elog.h:284
#define PG_END_TRY()
Definition: elog.h:300
int pq_getstring ( StringInfo  s)

Definition at line 1113 of file pqcomm.c.

References appendBinaryStringInfo(), Assert, i, pq_recvbuf(), PqCommReadingMsg, PqRecvBuffer, PqRecvLength, PqRecvPointer, and resetStringInfo().

Referenced by GetOldFunctionMessage(), and SocketBackend().

1114 {
1115  int i;
1116 
1118 
1119  resetStringInfo(s);
1120 
1121  /* Read until we get the terminating '\0' */
1122  for (;;)
1123  {
1124  while (PqRecvPointer >= PqRecvLength)
1125  {
1126  if (pq_recvbuf()) /* If nothing in buffer, then recv some */
1127  return EOF; /* Failed to recv data */
1128  }
1129 
1130  for (i = PqRecvPointer; i < PqRecvLength; i++)
1131  {
1132  if (PqRecvBuffer[i] == '\0')
1133  {
1134  /* include the '\0' in the copy */
1136  i - PqRecvPointer + 1);
1137  PqRecvPointer = i + 1; /* advance past \0 */
1138  return 0;
1139  }
1140  }
1141 
1142  /* If we're here we haven't got the \0 in the buffer yet. */
1144  PqRecvLength - PqRecvPointer);
1146  }
1147 }
static char PqRecvBuffer[PQ_RECV_BUFFER_SIZE]
Definition: pqcomm.c:123
static int PqRecvLength
Definition: pqcomm.c:125
static int pq_recvbuf(void)
Definition: pqcomm.c:886
void resetStringInfo(StringInfo str)
Definition: stringinfo.c:94
#define Assert(condition)
Definition: c.h:670
int i
static int PqRecvPointer
Definition: pqcomm.c:124
static bool PqCommReadingMsg
Definition: pqcomm.c:131
void appendBinaryStringInfo(StringInfo str, const char *data, int datalen)
Definition: stringinfo.c:240
void pq_init ( void  )

Definition at line 175 of file pqcomm.c.

References AddWaitEventToSet(), COMMERROR, CreateWaitEventSet(), DoingCopyOut, ereport, errmsg(), MemoryContextAlloc(), MyLatch, MyProcPort, NULL, on_proc_exit(), pg_set_noblock(), PQ_SEND_BUFFER_SIZE, PqCommBusy, PqCommReadingMsg, PqRecvLength, PqRecvPointer, PqSendBuffer, PqSendBufferSize, PqSendPointer, PqSendStart, Port::sock, socket_close(), TopMemoryContext, WL_LATCH_SET, WL_POSTMASTER_DEATH, and WL_SOCKET_WRITEABLE.

Referenced by BackendInitialize().

176 {
177  /* initialize state variables */
181  PqCommBusy = false;
182  PqCommReadingMsg = false;
183  DoingCopyOut = false;
184 
185  /* set up process-exit hook to close the socket */
187 
188  /*
189  * In backends (as soon as forked) we operate the underlying socket in
190  * nonblocking mode and use latches to implement blocking semantics if
191  * needed. That allows us to provide safely interruptible reads and
192  * writes.
193  *
194  * Use COMMERROR on failure, because ERROR would try to send the error to
195  * the client, which might require changing the mode again, leading to
196  * infinite recursion.
197  */
198 #ifndef WIN32
201  (errmsg("could not set socket to nonblocking mode: %m")));
202 #endif
203 
206  NULL, NULL);
209 }
#define WL_SOCKET_WRITEABLE
Definition: latch.h:126
static bool DoingCopyOut
Definition: pqcomm.c:132
struct Port * MyProcPort
Definition: globals.c:40
int AddWaitEventToSet(WaitEventSet *set, uint32 events, pgsocket fd, Latch *latch, void *user_data)
Definition: latch.c:613
void on_proc_exit(pg_on_exit_callback function, Datum arg)
Definition: ipc.c:292
static char * PqSendBuffer
Definition: pqcomm.c:118
static int PqSendStart
Definition: pqcomm.c:121
WaitEventSet * FeBeWaitSet
Definition: pqcomm.c:167
pgsocket sock
Definition: libpq-be.h:120
WaitEventSet * CreateWaitEventSet(MemoryContext context, int nevents)
Definition: latch.c:485
static int PqRecvLength
Definition: pqcomm.c:125
static int PqSendPointer
Definition: pqcomm.c:120
#define COMMERROR
Definition: elog.h:30
#define ereport(elevel, rest)
Definition: elog.h:122
MemoryContext TopMemoryContext
Definition: mcxt.c:43
static int PqSendBufferSize
Definition: pqcomm.c:119
static bool PqCommBusy
Definition: pqcomm.c:130
#define WL_POSTMASTER_DEATH
Definition: latch.h:128
#define NULL
Definition: c.h:226
bool pg_set_noblock(pgsocket sock)
Definition: noblock.c:21
static void socket_close(int code, Datum arg)
Definition: pqcomm.c:237
int errmsg(const char *fmt,...)
Definition: elog.c:797
void * MemoryContextAlloc(MemoryContext context, Size size)
Definition: mcxt.c:749
struct Latch * MyLatch
Definition: globals.c:51
static int PqRecvPointer
Definition: pqcomm.c:124
static bool PqCommReadingMsg
Definition: pqcomm.c:131
#define WL_LATCH_SET
Definition: latch.h:124
#define PQ_SEND_BUFFER_SIZE
Definition: pqcomm.c:115
bool pq_is_reading_msg ( void  )

Definition at line 1197 of file pqcomm.c.

References PqCommReadingMsg.

Referenced by PostgresMain().

1198 {
1199  return PqCommReadingMsg;
1200 }
static bool PqCommReadingMsg
Definition: pqcomm.c:131
int pq_peekbyte ( void  )

Definition at line 966 of file pqcomm.c.

References Assert, pq_recvbuf(), PqCommReadingMsg, PqRecvBuffer, PqRecvLength, and PqRecvPointer.

Referenced by recv_password_packet().

967 {
969 
970  while (PqRecvPointer >= PqRecvLength)
971  {
972  if (pq_recvbuf()) /* If nothing in buffer, then recv some */
973  return EOF; /* Failed to recv data */
974  }
975  return (unsigned char) PqRecvBuffer[PqRecvPointer];
976 }
static char PqRecvBuffer[PQ_RECV_BUFFER_SIZE]
Definition: pqcomm.c:123
static int PqRecvLength
Definition: pqcomm.c:125
static int pq_recvbuf(void)
Definition: pqcomm.c:886
#define Assert(condition)
Definition: c.h:670
static int PqRecvPointer
Definition: pqcomm.c:124
static bool PqCommReadingMsg
Definition: pqcomm.c:131
int pq_putbytes ( const char *  s,
size_t  len 
)

Definition at line 1300 of file pqcomm.c.

References Assert, DoingCopyOut, internal_putbytes(), and PqCommBusy.

Referenced by CopySendEndOfRow(), and socket_endcopyout().

1301 {
1302  int res;
1303 
1304  /* Should only be called by old-style COPY OUT */
1306  /* No-op if reentrant call */
1307  if (PqCommBusy)
1308  return 0;
1309  PqCommBusy = true;
1310  res = internal_putbytes(s, len);
1311  PqCommBusy = false;
1312  return res;
1313 }
static bool DoingCopyOut
Definition: pqcomm.c:132
static int internal_putbytes(const char *s, size_t len)
Definition: pqcomm.c:1316
static bool PqCommBusy
Definition: pqcomm.c:130
#define Assert(condition)
Definition: c.h:670
static int pq_recvbuf ( void  )
static

Definition at line 886 of file pqcomm.c.

References COMMERROR, EINTR, ereport, errcode_for_socket_access(), errmsg(), memmove, MyProcPort, PQ_RECV_BUFFER_SIZE, PqRecvBuffer, PqRecvLength, PqRecvPointer, secure_read(), and socket_set_nonblocking().

Referenced by pq_discardbytes(), pq_getbyte(), pq_getbytes(), pq_getstring(), and pq_peekbyte().

887 {
888  if (PqRecvPointer > 0)
889  {
891  {
892  /* still some unread data, left-justify it in the buffer */
896  PqRecvPointer = 0;
897  }
898  else
900  }
901 
902  /* Ensure that we're in blocking mode */
903  socket_set_nonblocking(false);
904 
905  /* Can fill buffer from PqRecvLength and upwards */
906  for (;;)
907  {
908  int r;
909 
912 
913  if (r < 0)
914  {
915  if (errno == EINTR)
916  continue; /* Ok if interrupted */
917 
918  /*
919  * Careful: an ereport() that tries to write to the client would
920  * cause recursion to here, leading to stack overflow and core
921  * dump! This message must go *only* to the postmaster log.
922  */
925  errmsg("could not receive data from client: %m")));
926  return EOF;
927  }
928  if (r == 0)
929  {
930  /*
931  * EOF detected. We used to write a log message here, but it's
932  * better to expect the ultimate caller to do that.
933  */
934  return EOF;
935  }
936  /* r contains number of bytes read, so just incr length */
937  PqRecvLength += r;
938  return 0;
939  }
940 }
static char PqRecvBuffer[PQ_RECV_BUFFER_SIZE]
Definition: pqcomm.c:123
struct Port * MyProcPort
Definition: globals.c:40
ssize_t secure_read(Port *port, void *ptr, size_t len)
Definition: be-secure.c:137
static void socket_set_nonblocking(bool nonblocking)
Definition: pqcomm.c:869
static int PqRecvLength
Definition: pqcomm.c:125
#define memmove(d, s, c)
Definition: c.h:1057
#define COMMERROR
Definition: elog.h:30
#define ereport(elevel, rest)
Definition: elog.h:122
int errcode_for_socket_access(void)
Definition: elog.c:669
#define EINTR
Definition: win32.h:295
int errmsg(const char *fmt,...)
Definition: elog.c:797
#define PQ_RECV_BUFFER_SIZE
Definition: pqcomm.c:116
static int PqRecvPointer
Definition: pqcomm.c:124
int pq_setkeepalivescount ( int  count,
Port port 
)

Definition at line 1852 of file pqcomm.c.

References SockAddr::addr, Port::default_keepalives_count, elog, IS_AF_UNIX, Port::keepalives_count, Port::laddr, LOG, NULL, pq_getkeepalivescount(), Port::sock, STATUS_ERROR, and STATUS_OK.

Referenced by assign_tcp_keepalives_count(), and StreamConnection().

1853 {
1854  if (port == NULL || IS_AF_UNIX(port->laddr.addr.ss_family))
1855  return STATUS_OK;
1856 
1857 #ifdef TCP_KEEPCNT
1858  if (count == port->keepalives_count)
1859  return STATUS_OK;
1860 
1861  if (port->default_keepalives_count <= 0)
1862  {
1863  if (pq_getkeepalivescount(port) < 0)
1864  {
1865  if (count == 0)
1866  return STATUS_OK; /* default is set but unknown */
1867  else
1868  return STATUS_ERROR;
1869  }
1870  }
1871 
1872  if (count == 0)
1873  count = port->default_keepalives_count;
1874 
1875  if (setsockopt(port->sock, IPPROTO_TCP, TCP_KEEPCNT,
1876  (char *) &count, sizeof(count)) < 0)
1877  {
1878  elog(LOG, "setsockopt(TCP_KEEPCNT) failed: %m");
1879  return STATUS_ERROR;
1880  }
1881 
1882  port->keepalives_count = count;
1883 #else
1884  if (count != 0)
1885  {
1886  elog(LOG, "setsockopt(TCP_KEEPCNT) not supported");
1887  return STATUS_ERROR;
1888  }
1889 #endif
1890 
1891  return STATUS_OK;
1892 }
struct sockaddr_storage addr
Definition: pqcomm.h:64
#define STATUS_ERROR
Definition: c.h:971
#define LOG
Definition: elog.h:26
int default_keepalives_count
Definition: libpq-be.h:164
pgsocket sock
Definition: libpq-be.h:120
int keepalives_count
Definition: libpq-be.h:167
#define IS_AF_UNIX(fam)
Definition: ip.h:24
#define STATUS_OK
Definition: c.h:970
SockAddr laddr
Definition: libpq-be.h:123
#define NULL
Definition: c.h:226
int pq_getkeepalivescount(Port *port)
Definition: pqcomm.c:1823
#define elog
Definition: elog.h:219
int pq_setkeepalivesidle ( int  idle,
Port port 
)

Definition at line 1687 of file pqcomm.c.

References SockAddr::addr, Port::default_keepalives_idle, elog, IS_AF_UNIX, Port::keepalives_idle, Port::keepalives_interval, Port::laddr, LOG, NULL, pq_getkeepalivesidle(), Port::sock, STATUS_ERROR, and STATUS_OK.

Referenced by assign_tcp_keepalives_idle(), and StreamConnection().

1688 {
1689  if (port == NULL || IS_AF_UNIX(port->laddr.addr.ss_family))
1690  return STATUS_OK;
1691 
1692 #if defined(TCP_KEEPIDLE) || defined(TCP_KEEPALIVE) || defined(SIO_KEEPALIVE_VALS)
1693  if (idle == port->keepalives_idle)
1694  return STATUS_OK;
1695 
1696 #ifndef WIN32
1697  if (port->default_keepalives_idle <= 0)
1698  {
1699  if (pq_getkeepalivesidle(port) < 0)
1700  {
1701  if (idle == 0)
1702  return STATUS_OK; /* default is set but unknown */
1703  else
1704  return STATUS_ERROR;
1705  }
1706  }
1707 
1708  if (idle == 0)
1709  idle = port->default_keepalives_idle;
1710 
1711 #ifdef TCP_KEEPIDLE
1712  if (setsockopt(port->sock, IPPROTO_TCP, TCP_KEEPIDLE,
1713  (char *) &idle, sizeof(idle)) < 0)
1714  {
1715  elog(LOG, "setsockopt(TCP_KEEPIDLE) failed: %m");
1716  return STATUS_ERROR;
1717  }
1718 #else
1719  if (setsockopt(port->sock, IPPROTO_TCP, TCP_KEEPALIVE,
1720  (char *) &idle, sizeof(idle)) < 0)
1721  {
1722  elog(LOG, "setsockopt(TCP_KEEPALIVE) failed: %m");
1723  return STATUS_ERROR;
1724  }
1725 #endif
1726 
1727  port->keepalives_idle = idle;
1728 #else /* WIN32 */
1729  return pq_setkeepaliveswin32(port, idle, port->keepalives_interval);
1730 #endif
1731 #else /* TCP_KEEPIDLE || SIO_KEEPALIVE_VALS */
1732  if (idle != 0)
1733  {
1734  elog(LOG, "setting the keepalive idle time is not supported");
1735  return STATUS_ERROR;
1736  }
1737 #endif
1738  return STATUS_OK;
1739 }
int keepalives_idle
Definition: libpq-be.h:165
struct sockaddr_storage addr
Definition: pqcomm.h:64
#define STATUS_ERROR
Definition: c.h:971
#define LOG
Definition: elog.h:26
pgsocket sock
Definition: libpq-be.h:120
#define IS_AF_UNIX(fam)
Definition: ip.h:24
#define STATUS_OK
Definition: c.h:970
SockAddr laddr
Definition: libpq-be.h:123
int default_keepalives_idle
Definition: libpq-be.h:162
#define NULL
Definition: c.h:226
int pq_getkeepalivesidle(Port *port)
Definition: pqcomm.c:1643
#define elog
Definition: elog.h:219
int keepalives_interval
Definition: libpq-be.h:166
int pq_setkeepalivesinterval ( int  interval,
Port port 
)

Definition at line 1776 of file pqcomm.c.

References SockAddr::addr, Port::default_keepalives_interval, elog, IS_AF_UNIX, Port::keepalives_idle, Port::keepalives_interval, Port::laddr, LOG, NULL, pq_getkeepalivesinterval(), Port::sock, STATUS_ERROR, and STATUS_OK.

Referenced by assign_tcp_keepalives_interval(), and StreamConnection().

1777 {
1778  if (port == NULL || IS_AF_UNIX(port->laddr.addr.ss_family))
1779  return STATUS_OK;
1780 
1781 #if defined(TCP_KEEPINTVL) || defined (SIO_KEEPALIVE_VALS)
1782  if (interval == port->keepalives_interval)
1783  return STATUS_OK;
1784 
1785 #ifndef WIN32
1786  if (port->default_keepalives_interval <= 0)
1787  {
1788  if (pq_getkeepalivesinterval(port) < 0)
1789  {
1790  if (interval == 0)
1791  return STATUS_OK; /* default is set but unknown */
1792  else
1793  return STATUS_ERROR;
1794  }
1795  }
1796 
1797  if (interval == 0)
1799 
1800  if (setsockopt(port->sock, IPPROTO_TCP, TCP_KEEPINTVL,
1801  (char *) &interval, sizeof(interval)) < 0)
1802  {
1803  elog(LOG, "setsockopt(TCP_KEEPINTVL) failed: %m");
1804  return STATUS_ERROR;
1805  }
1806 
1807  port->keepalives_interval = interval;
1808 #else /* WIN32 */
1809  return pq_setkeepaliveswin32(port, port->keepalives_idle, interval);
1810 #endif
1811 #else
1812  if (interval != 0)
1813  {
1814  elog(LOG, "setsockopt(TCP_KEEPINTVL) not supported");
1815  return STATUS_ERROR;
1816  }
1817 #endif
1818 
1819  return STATUS_OK;
1820 }
int keepalives_idle
Definition: libpq-be.h:165
int pq_getkeepalivesinterval(Port *port)
Definition: pqcomm.c:1742
struct sockaddr_storage addr
Definition: pqcomm.h:64
#define STATUS_ERROR
Definition: c.h:971
#define LOG
Definition: elog.h:26
int default_keepalives_interval
Definition: libpq-be.h:163
pgsocket sock
Definition: libpq-be.h:120
#define IS_AF_UNIX(fam)
Definition: ip.h:24
#define STATUS_OK
Definition: c.h:970
SockAddr laddr
Definition: libpq-be.h:123
#define NULL
Definition: c.h:226
#define elog
Definition: elog.h:219
int keepalives_interval
Definition: libpq-be.h:166
void pq_startmsgread ( void  )

Definition at line 1157 of file pqcomm.c.

References ereport, errcode(), errmsg(), FATAL, and PqCommReadingMsg.

Referenced by CopyGetData(), ProcessRepliesIfAny(), ProcessStartupPacket(), ReceiveCopyBegin(), recv_password_packet(), and SocketBackend().

1158 {
1159  /*
1160  * There shouldn't be a read active already, but let's check just to be
1161  * sure.
1162  */
1163  if (PqCommReadingMsg)
1164  ereport(FATAL,
1165  (errcode(ERRCODE_PROTOCOL_VIOLATION),
1166  errmsg("terminating connection because protocol synchronization was lost")));
1167 
1168  PqCommReadingMsg = true;
1169 }
int errcode(int sqlerrcode)
Definition: elog.c:575
#define FATAL
Definition: elog.h:52
#define ereport(elevel, rest)
Definition: elog.h:122
int errmsg(const char *fmt,...)
Definition: elog.c:797
static bool PqCommReadingMsg
Definition: pqcomm.c:131
void RemoveSocketFiles ( void  )

Definition at line 836 of file pqcomm.c.

References lfirst, NIL, and unlink().

Referenced by CloseServerPorts().

837 {
838  ListCell *l;
839 
840  /* Loop through all created sockets... */
841  foreach(l, sock_paths)
842  {
843  char *sock_path = (char *) lfirst(l);
844 
845  /* Ignore any error. */
846  (void) unlink(sock_path);
847  }
848  /* Since we're about to exit, no need to reclaim storage */
849  sock_paths = NIL;
850 }
#define NIL
Definition: pg_list.h:69
static List * sock_paths
Definition: pqcomm.c:106
int unlink(const char *filename)
#define lfirst(lc)
Definition: pg_list.h:106
static void socket_close ( int  code,
Datum  arg 
)
static

Definition at line 237 of file pqcomm.c.

References free, Port::gss, MyProcPort, NULL, PGINVALID_SOCKET, secure_close(), and Port::sock.

Referenced by pq_init().

238 {
239  /* Nothing to do in a standalone backend, where MyProcPort is NULL. */
240  if (MyProcPort != NULL)
241  {
242 #if defined(ENABLE_GSS) || defined(ENABLE_SSPI)
243 #ifdef ENABLE_GSS
244  OM_uint32 min_s;
245 
246  /*
247  * Shutdown GSSAPI layer. This section does nothing when interrupting
248  * BackendInitialize(), because pg_GSS_recvauth() makes first use of
249  * "ctx" and "cred".
250  */
251  if (MyProcPort->gss->ctx != GSS_C_NO_CONTEXT)
252  gss_delete_sec_context(&min_s, &MyProcPort->gss->ctx, NULL);
253 
254  if (MyProcPort->gss->cred != GSS_C_NO_CREDENTIAL)
255  gss_release_cred(&min_s, &MyProcPort->gss->cred);
256 #endif /* ENABLE_GSS */
257 
258  /*
259  * GSS and SSPI share the port->gss struct. Since nowhere else does a
260  * postmaster child free this, doing so is safe when interrupting
261  * BackendInitialize().
262  */
263  free(MyProcPort->gss);
264 #endif /* ENABLE_GSS || ENABLE_SSPI */
265 
266  /*
267  * Cleanly shut down SSL layer. Nowhere else does a postmaster child
268  * call this, so this is safe when interrupting BackendInitialize().
269  */
271 
272  /*
273  * Formerly we did an explicit close() here, but it seems better to
274  * leave the socket open until the process dies. This allows clients
275  * to perform a "synchronous close" if they care --- wait till the
276  * transport layer reports connection closure, and you can be sure the
277  * backend has exited.
278  *
279  * We do set sock to PGINVALID_SOCKET to prevent any further I/O,
280  * though.
281  */
283  }
284 }
struct Port * MyProcPort
Definition: globals.c:40
pgsocket sock
Definition: libpq-be.h:120
#define PGINVALID_SOCKET
Definition: port.h:24
#define free(a)
Definition: header.h:60
#define NULL
Definition: c.h:226
void secure_close(Port *port)
Definition: be-secure.c:125
void * gss
Definition: libpq-be.h:177
static void socket_comm_reset ( void  )
static

Definition at line 220 of file pqcomm.c.

References pq_endcopyout, and PqCommBusy.

221 {
222  /* Do not throw away pending data, but do reset the busy flag */
223  PqCommBusy = false;
224  /* We can abort any old-style COPY OUT, too */
225  pq_endcopyout(true);
226 }
static bool PqCommBusy
Definition: pqcomm.c:130
#define pq_endcopyout(errorAbort)
Definition: libpq.h:48
static void socket_endcopyout ( bool  errorAbort)
static

Definition at line 1583 of file pqcomm.c.

References DoingCopyOut, and pq_putbytes().

1584 {
1585  if (!DoingCopyOut)
1586  return;
1587  if (errorAbort)
1588  pq_putbytes("\n\n\\.\n", 5);
1589  /* in non-error case, copy.c will have emitted the terminator line */
1590  DoingCopyOut = false;
1591 }
static bool DoingCopyOut
Definition: pqcomm.c:132
int pq_putbytes(const char *s, size_t len)
Definition: pqcomm.c:1300
static int socket_flush ( void  )
static

Definition at line 1347 of file pqcomm.c.

References internal_flush(), PqCommBusy, and socket_set_nonblocking().

1348 {
1349  int res;
1350 
1351  /* No-op if reentrant call */
1352  if (PqCommBusy)
1353  return 0;
1354  PqCommBusy = true;
1355  socket_set_nonblocking(false);
1356  res = internal_flush();
1357  PqCommBusy = false;
1358  return res;
1359 }
static int internal_flush(void)
Definition: pqcomm.c:1369
static void socket_set_nonblocking(bool nonblocking)
Definition: pqcomm.c:869
static bool PqCommBusy
Definition: pqcomm.c:130
static int socket_flush_if_writable ( void  )
static

Definition at line 1442 of file pqcomm.c.

References internal_flush(), PqCommBusy, PqSendPointer, PqSendStart, and socket_set_nonblocking().

1443 {
1444  int res;
1445 
1446  /* Quick exit if nothing to do */
1447  if (PqSendPointer == PqSendStart)
1448  return 0;
1449 
1450  /* No-op if reentrant call */
1451  if (PqCommBusy)
1452  return 0;
1453 
1454  /* Temporarily put the socket into non-blocking mode */
1455  socket_set_nonblocking(true);
1456 
1457  PqCommBusy = true;
1458  res = internal_flush();
1459  PqCommBusy = false;
1460  return res;
1461 }
static int PqSendStart
Definition: pqcomm.c:121
static int internal_flush(void)
Definition: pqcomm.c:1369
static void socket_set_nonblocking(bool nonblocking)
Definition: pqcomm.c:869
static int PqSendPointer
Definition: pqcomm.c:120
static bool PqCommBusy
Definition: pqcomm.c:130
static bool socket_is_send_pending ( void  )
static

Definition at line 1468 of file pqcomm.c.

References PqSendPointer, and PqSendStart.

1469 {
1470  return (PqSendStart < PqSendPointer);
1471 }
static int PqSendStart
Definition: pqcomm.c:121
static int PqSendPointer
Definition: pqcomm.c:120
static int socket_putmessage ( char  msgtype,
const char *  s,
size_t  len 
)
static

Definition at line 1507 of file pqcomm.c.

References DoingCopyOut, FrontendProtocol, internal_putbytes(), PG_PROTOCOL_MAJOR, and PqCommBusy.

1508 {
1509  if (DoingCopyOut || PqCommBusy)
1510  return 0;
1511  PqCommBusy = true;
1512  if (msgtype)
1513  if (internal_putbytes(&msgtype, 1))
1514  goto fail;
1516  {
1517  uint32 n32;
1518 
1519  n32 = htonl((uint32) (len + 4));
1520  if (internal_putbytes((char *) &n32, 4))
1521  goto fail;
1522  }
1523  if (internal_putbytes(s, len))
1524  goto fail;
1525  PqCommBusy = false;
1526  return 0;
1527 
1528 fail:
1529  PqCommBusy = false;
1530  return EOF;
1531 }
static bool DoingCopyOut
Definition: pqcomm.c:132
#define PG_PROTOCOL_MAJOR(v)
Definition: pqcomm.h:104
unsigned int uint32
Definition: c.h:265
static int internal_putbytes(const char *s, size_t len)
Definition: pqcomm.c:1316
static bool PqCommBusy
Definition: pqcomm.c:130
ProtocolVersion FrontendProtocol
Definition: globals.c:27
static void socket_putmessage_noblock ( char  msgtype,
const char *  s,
size_t  len 
)
static

Definition at line 1540 of file pqcomm.c.

References Assert, PG_USED_FOR_ASSERTS_ONLY, pq_putmessage, PqSendBuffer, PqSendBufferSize, PqSendPointer, and repalloc().

1541 {
1542  int res PG_USED_FOR_ASSERTS_ONLY;
1543  int required;
1544 
1545  /*
1546  * Ensure we have enough space in the output buffer for the message header
1547  * as well as the message itself.
1548  */
1549  required = PqSendPointer + 1 + 4 + len;
1550  if (required > PqSendBufferSize)
1551  {
1552  PqSendBuffer = repalloc(PqSendBuffer, required);
1553  PqSendBufferSize = required;
1554  }
1555  res = pq_putmessage(msgtype, s, len);
1556  Assert(res == 0); /* should not fail when the message fits in
1557  * buffer */
1558 }
static char * PqSendBuffer
Definition: pqcomm.c:118
static int PqSendPointer
Definition: pqcomm.c:120
static int PqSendBufferSize
Definition: pqcomm.c:119
#define Assert(condition)
Definition: c.h:670
void * repalloc(void *pointer, Size size)
Definition: mcxt.c:1021
#define pq_putmessage(msgtype, s, len)
Definition: libpq.h:43
#define PG_USED_FOR_ASSERTS_ONLY
Definition: c.h:985
static void socket_set_nonblocking ( bool  nonblocking)
static

Definition at line 869 of file pqcomm.c.

References ereport, errcode(), errmsg(), ERROR, MyProcPort, Port::noblock, and NULL.

Referenced by internal_putbytes(), pq_getbyte_if_available(), pq_recvbuf(), socket_flush(), and socket_flush_if_writable().

870 {
871  if (MyProcPort == NULL)
872  ereport(ERROR,
873  (errcode(ERRCODE_CONNECTION_DOES_NOT_EXIST),
874  errmsg("there is no client connection")));
875 
876  MyProcPort->noblock = nonblocking;
877 }
struct Port * MyProcPort
Definition: globals.c:40
int errcode(int sqlerrcode)
Definition: elog.c:575
#define ERROR
Definition: elog.h:43
#define ereport(elevel, rest)
Definition: elog.h:122
#define NULL
Definition: c.h:226
bool noblock
Definition: libpq-be.h:121
int errmsg(const char *fmt,...)
Definition: elog.c:797
static void socket_startcopyout ( void  )
static

Definition at line 1567 of file pqcomm.c.

References DoingCopyOut.

1568 {
1569  DoingCopyOut = true;
1570 }
static bool DoingCopyOut
Definition: pqcomm.c:132
void StreamClose ( pgsocket  sock)

Definition at line 791 of file pqcomm.c.

References closesocket.

Referenced by ClosePostmasterPorts(), CloseServerPorts(), ConnCreate(), and ServerLoop().

792 {
793  closesocket(sock);
794 }
#define closesocket
Definition: port.h:328
int StreamConnection ( pgsocket  server_fd,
Port port 
)

Definition at line 663 of file pqcomm.c.

References accept, SockAddr::addr, elog, ereport, errcode_for_socket_access(), errmsg(), IS_AF_UNIX, Port::laddr, LOG, pg_usleep(), PGINVALID_SOCKET, PQ_SEND_BUFFER_SIZE, pq_setkeepalivescount(), pq_setkeepalivesidle(), pq_setkeepalivesinterval(), Port::raddr, SockAddr::salen, Port::sock, STATUS_ERROR, STATUS_OK, tcp_keepalives_count, tcp_keepalives_idle, and tcp_keepalives_interval.

Referenced by ConnCreate().

664 {
665  /* accept connection and fill in the client (remote) address */
666  port->raddr.salen = sizeof(port->raddr.addr);
667  if ((port->sock = accept(server_fd,
668  (struct sockaddr *) & port->raddr.addr,
669  &port->raddr.salen)) == PGINVALID_SOCKET)
670  {
671  ereport(LOG,
673  errmsg("could not accept new connection: %m")));
674 
675  /*
676  * If accept() fails then postmaster.c will still see the server
677  * socket as read-ready, and will immediately try again. To avoid
678  * uselessly sucking lots of CPU, delay a bit before trying again.
679  * (The most likely reason for failure is being out of kernel file
680  * table slots; we can do little except hope some will get freed up.)
681  */
682  pg_usleep(100000L); /* wait 0.1 sec */
683  return STATUS_ERROR;
684  }
685 
686  /* fill in the server (local) address */
687  port->laddr.salen = sizeof(port->laddr.addr);
688  if (getsockname(port->sock,
689  (struct sockaddr *) & port->laddr.addr,
690  &port->laddr.salen) < 0)
691  {
692  elog(LOG, "getsockname() failed: %m");
693  return STATUS_ERROR;
694  }
695 
696  /* select NODELAY and KEEPALIVE options if it's a TCP connection */
697  if (!IS_AF_UNIX(port->laddr.addr.ss_family))
698  {
699  int on;
700 #ifdef WIN32
701  int oldopt;
702  int optlen;
703  int newopt;
704 #endif
705 
706 #ifdef TCP_NODELAY
707  on = 1;
708  if (setsockopt(port->sock, IPPROTO_TCP, TCP_NODELAY,
709  (char *) &on, sizeof(on)) < 0)
710  {
711  elog(LOG, "setsockopt(TCP_NODELAY) failed: %m");
712  return STATUS_ERROR;
713  }
714 #endif
715  on = 1;
716  if (setsockopt(port->sock, SOL_SOCKET, SO_KEEPALIVE,
717  (char *) &on, sizeof(on)) < 0)
718  {
719  elog(LOG, "setsockopt(SO_KEEPALIVE) failed: %m");
720  return STATUS_ERROR;
721  }
722 
723 #ifdef WIN32
724 
725  /*
726  * This is a Win32 socket optimization. The OS send buffer should be
727  * large enough to send the whole Postgres send buffer in one go, or
728  * performance suffers. The Postgres send buffer can be enlarged if a
729  * very large message needs to be sent, but we won't attempt to
730  * enlarge the OS buffer if that happens, so somewhat arbitrarily
731  * ensure that the OS buffer is at least PQ_SEND_BUFFER_SIZE * 4.
732  * (That's 32kB with the current default).
733  *
734  * The default OS buffer size used to be 8kB in earlier Windows
735  * versions, but was raised to 64kB in Windows 2012. So it shouldn't
736  * be necessary to change it in later versions anymore. Changing it
737  * unnecessarily can even reduce performance, because setting
738  * SO_SNDBUF in the application disables the "dynamic send buffering"
739  * feature that was introduced in Windows 7. So before fiddling with
740  * SO_SNDBUF, check if the current buffer size is already large enough
741  * and only increase it if necessary.
742  *
743  * See https://support.microsoft.com/kb/823764/EN-US/ and
744  * https://msdn.microsoft.com/en-us/library/bb736549%28v=vs.85%29.aspx
745  */
746  optlen = sizeof(oldopt);
747  if (getsockopt(port->sock, SOL_SOCKET, SO_SNDBUF, (char *) &oldopt,
748  &optlen) < 0)
749  {
750  elog(LOG, "getsockopt(SO_SNDBUF) failed: %m");
751  return STATUS_ERROR;
752  }
753  newopt = PQ_SEND_BUFFER_SIZE * 4;
754  if (oldopt < newopt)
755  {
756  if (setsockopt(port->sock, SOL_SOCKET, SO_SNDBUF, (char *) &newopt,
757  sizeof(newopt)) < 0)
758  {
759  elog(LOG, "setsockopt(SO_SNDBUF) failed: %m");
760  return STATUS_ERROR;
761  }
762  }
763 #endif
764 
765  /*
766  * Also apply the current keepalive parameters. If we fail to set a
767  * parameter, don't error out, because these aren't universally
768  * supported. (Note: you might think we need to reset the GUC
769  * variables to 0 in such a case, but it's not necessary because the
770  * show hooks for these variables report the truth anyway.)
771  */
775  }
776 
777  return STATUS_OK;
778 }
#define accept(s, addr, addrlen)
Definition: win32.h:382
int pq_setkeepalivesinterval(int interval, Port *port)
Definition: pqcomm.c:1776
struct sockaddr_storage addr
Definition: pqcomm.h:64
#define STATUS_ERROR
Definition: c.h:971
#define LOG
Definition: elog.h:26
pgsocket sock
Definition: libpq-be.h:120
int tcp_keepalives_idle
Definition: guc.c:473
SockAddr raddr
Definition: libpq-be.h:124
void pg_usleep(long microsec)
Definition: signal.c:53
int pq_setkeepalivesidle(int idle, Port *port)
Definition: pqcomm.c:1687
#define IS_AF_UNIX(fam)
Definition: ip.h:24
ACCEPT_TYPE_ARG3 salen
Definition: pqcomm.h:65
#define ereport(elevel, rest)
Definition: elog.h:122
#define STATUS_OK
Definition: c.h:970
int tcp_keepalives_interval
Definition: guc.c:474
int errcode_for_socket_access(void)
Definition: elog.c:669
SockAddr laddr
Definition: libpq-be.h:123
#define PGINVALID_SOCKET
Definition: port.h:24
int pq_setkeepalivescount(int count, Port *port)
Definition: pqcomm.c:1852
int errmsg(const char *fmt,...)
Definition: elog.c:797
int tcp_keepalives_count
Definition: guc.c:475
#define elog
Definition: elog.h:219
#define PQ_SEND_BUFFER_SIZE
Definition: pqcomm.c:115
int StreamServerPort ( int  family,
char *  hostName,
unsigned short  portNumber,
char *  unixSocketDir,
pgsocket  ListenSocket[],
int  MaxListen 
)

Definition at line 311 of file pqcomm.c.

References _, addrinfo::ai_family, addrinfo::ai_flags, addrinfo::ai_next, AI_PASSIVE, addrinfo::ai_socktype, bind, closesocket, ereport, errcode_for_socket_access(), errhint(), errmsg(), fd(), gai_strerror, IS_AF_UNIX, listen, LOG, MaxBackends, MAXPGPATH, MemSet, NULL, pg_freeaddrinfo_all(), pg_getaddrinfo_all(), PG_SOMAXCONN, PGINVALID_SOCKET, snprintf(), socket, STATUS_ERROR, STATUS_OK, UNIXSOCK_PATH, and UNIXSOCK_PATH_BUFLEN.

Referenced by PostmasterMain().

314 {
315  pgsocket fd;
316  int err;
317  int maxconn;
318  int ret;
319  char portNumberStr[32];
320  const char *familyDesc;
321  char familyDescBuf[64];
322  char *service;
323  struct addrinfo *addrs = NULL,
324  *addr;
325  struct addrinfo hint;
326  int listen_index = 0;
327  int added = 0;
328 
329 #ifdef HAVE_UNIX_SOCKETS
330  char unixSocketPath[MAXPGPATH];
331 #endif
332 #if !defined(WIN32) || defined(IPV6_V6ONLY)
333  int one = 1;
334 #endif
335 
336  /* Initialize hint structure */
337  MemSet(&hint, 0, sizeof(hint));
338  hint.ai_family = family;
339  hint.ai_flags = AI_PASSIVE;
340  hint.ai_socktype = SOCK_STREAM;
341 
342 #ifdef HAVE_UNIX_SOCKETS
343  if (family == AF_UNIX)
344  {
345  /*
346  * Create unixSocketPath from portNumber and unixSocketDir and lock
347  * that file path
348  */
349  UNIXSOCK_PATH(unixSocketPath, portNumber, unixSocketDir);
350  if (strlen(unixSocketPath) >= UNIXSOCK_PATH_BUFLEN)
351  {
352  ereport(LOG,
353  (errmsg("Unix-domain socket path \"%s\" is too long (maximum %d bytes)",
354  unixSocketPath,
355  (int) (UNIXSOCK_PATH_BUFLEN - 1))));
356  return STATUS_ERROR;
357  }
358  if (Lock_AF_UNIX(unixSocketDir, unixSocketPath) != STATUS_OK)
359  return STATUS_ERROR;
360  service = unixSocketPath;
361  }
362  else
363 #endif /* HAVE_UNIX_SOCKETS */
364  {
365  snprintf(portNumberStr, sizeof(portNumberStr), "%d", portNumber);
366  service = portNumberStr;
367  }
368 
369  ret = pg_getaddrinfo_all(hostName, service, &hint, &addrs);
370  if (ret || !addrs)
371  {
372  if (hostName)
373  ereport(LOG,
374  (errmsg("could not translate host name \"%s\", service \"%s\" to address: %s",
375  hostName, service, gai_strerror(ret))));
376  else
377  ereport(LOG,
378  (errmsg("could not translate service \"%s\" to address: %s",
379  service, gai_strerror(ret))));
380  if (addrs)
381  pg_freeaddrinfo_all(hint.ai_family, addrs);
382  return STATUS_ERROR;
383  }
384 
385  for (addr = addrs; addr; addr = addr->ai_next)
386  {
387  if (!IS_AF_UNIX(family) && IS_AF_UNIX(addr->ai_family))
388  {
389  /*
390  * Only set up a unix domain socket when they really asked for it.
391  * The service/port is different in that case.
392  */
393  continue;
394  }
395 
396  /* See if there is still room to add 1 more socket. */
397  for (; listen_index < MaxListen; listen_index++)
398  {
399  if (ListenSocket[listen_index] == PGINVALID_SOCKET)
400  break;
401  }
402  if (listen_index >= MaxListen)
403  {
404  ereport(LOG,
405  (errmsg("could not bind to all requested addresses: MAXLISTEN (%d) exceeded",
406  MaxListen)));
407  break;
408  }
409 
410  /* set up family name for possible error messages */
411  switch (addr->ai_family)
412  {
413  case AF_INET:
414  familyDesc = _("IPv4");
415  break;
416 #ifdef HAVE_IPV6
417  case AF_INET6:
418  familyDesc = _("IPv6");
419  break;
420 #endif
421 #ifdef HAVE_UNIX_SOCKETS
422  case AF_UNIX:
423  familyDesc = _("Unix");
424  break;
425 #endif
426  default:
427  snprintf(familyDescBuf, sizeof(familyDescBuf),
428  _("unrecognized address family %d"),
429  addr->ai_family);
430  familyDesc = familyDescBuf;
431  break;
432  }
433 
434  if ((fd = socket(addr->ai_family, SOCK_STREAM, 0)) == PGINVALID_SOCKET)
435  {
436  ereport(LOG,
438  /* translator: %s is IPv4, IPv6, or Unix */
439  errmsg("could not create %s socket: %m",
440  familyDesc)));
441  continue;
442  }
443 
444 #ifndef WIN32
445 
446  /*
447  * Without the SO_REUSEADDR flag, a new postmaster can't be started
448  * right away after a stop or crash, giving "address already in use"
449  * error on TCP ports.
450  *
451  * On win32, however, this behavior only happens if the
452  * SO_EXLUSIVEADDRUSE is set. With SO_REUSEADDR, win32 allows multiple
453  * servers to listen on the same address, resulting in unpredictable
454  * behavior. With no flags at all, win32 behaves as Unix with
455  * SO_REUSEADDR.
456  */
457  if (!IS_AF_UNIX(addr->ai_family))
458  {
459  if ((setsockopt(fd, SOL_SOCKET, SO_REUSEADDR,
460  (char *) &one, sizeof(one))) == -1)
461  {
462  ereport(LOG,
464  errmsg("setsockopt(SO_REUSEADDR) failed: %m")));
465  closesocket(fd);
466  continue;
467  }
468  }
469 #endif
470 
471 #ifdef IPV6_V6ONLY
472  if (addr->ai_family == AF_INET6)
473  {
474  if (setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY,
475  (char *) &one, sizeof(one)) == -1)
476  {
477  ereport(LOG,
479  errmsg("setsockopt(IPV6_V6ONLY) failed: %m")));
480  closesocket(fd);
481  continue;
482  }
483  }
484 #endif
485 
486  /*
487  * Note: This might fail on some OS's, like Linux older than
488  * 2.4.21-pre3, that don't have the IPV6_V6ONLY socket option, and map
489  * ipv4 addresses to ipv6. It will show ::ffff:ipv4 for all ipv4
490  * connections.
491  */
492  err = bind(fd, addr->ai_addr, addr->ai_addrlen);
493  if (err < 0)
494  {
495  ereport(LOG,
497  /* translator: %s is IPv4, IPv6, or Unix */
498  errmsg("could not bind %s socket: %m",
499  familyDesc),
500  (IS_AF_UNIX(addr->ai_family)) ?
501  errhint("Is another postmaster already running on port %d?"
502  " If not, remove socket file \"%s\" and retry.",
503  (int) portNumber, service) :
504  errhint("Is another postmaster already running on port %d?"
505  " If not, wait a few seconds and retry.",
506  (int) portNumber)));
507  closesocket(fd);
508  continue;
509  }
510 
511 #ifdef HAVE_UNIX_SOCKETS
512  if (addr->ai_family == AF_UNIX)
513  {
514  if (Setup_AF_UNIX(service) != STATUS_OK)
515  {
516  closesocket(fd);
517  break;
518  }
519  }
520 #endif
521 
522  /*
523  * Select appropriate accept-queue length limit. PG_SOMAXCONN is only
524  * intended to provide a clamp on the request on platforms where an
525  * overly large request provokes a kernel error (are there any?).
526  */
527  maxconn = MaxBackends * 2;
528  if (maxconn > PG_SOMAXCONN)
529  maxconn = PG_SOMAXCONN;
530 
531  err = listen(fd, maxconn);
532  if (err < 0)
533  {
534  ereport(LOG,
536  /* translator: %s is IPv4, IPv6, or Unix */
537  errmsg("could not listen on %s socket: %m",
538  familyDesc)));
539  closesocket(fd);
540  continue;
541  }
542  ListenSocket[listen_index] = fd;
543  added++;
544  }
545 
546  pg_freeaddrinfo_all(hint.ai_family, addrs);
547 
548  if (!added)
549  return STATUS_ERROR;
550 
551  return STATUS_OK;
552 }
void pg_freeaddrinfo_all(int hint_ai_family, struct addrinfo *ai)
Definition: ip.c:89
#define UNIXSOCK_PATH(path, port, sockdir)
Definition: pqcomm.h:70
int errhint(const char *fmt,...)
Definition: elog.c:987
#define UNIXSOCK_PATH_BUFLEN
Definition: pqcomm.h:86
#define closesocket
Definition: port.h:328
#define socket(af, type, protocol)
Definition: win32.h:379
#define STATUS_ERROR
Definition: c.h:971
#define MemSet(start, val, len)
Definition: c.h:852
int snprintf(char *str, size_t count, const char *fmt,...) pg_attribute_printf(3
#define LOG
Definition: elog.h:26
static int fd(const char *x, int i)
Definition: preproc-init.c:105
#define gai_strerror
Definition: getaddrinfo.h:148
int pg_getaddrinfo_all(const char *hostname, const char *servname, const struct addrinfo *hintp, struct addrinfo **result)
Definition: ip.c:58
#define bind(s, addr, addrlen)
Definition: win32.h:380
#define IS_AF_UNIX(fam)
Definition: ip.h:24
#define MAXPGPATH
int MaxBackends
Definition: globals.c:126
#define AI_PASSIVE
Definition: getaddrinfo.h:64
int pgsocket
Definition: port.h:22
#define ereport(elevel, rest)
Definition: elog.h:122
#define STATUS_OK
Definition: c.h:970
int errcode_for_socket_access(void)
Definition: elog.c:669
static pgsocket ListenSocket[MAXLISTEN]
Definition: postmaster.c:215
#define listen(s, backlog)
Definition: win32.h:381
#define PGINVALID_SOCKET
Definition: port.h:24
#define PG_SOMAXCONN
#define NULL
Definition: c.h:226
struct addrinfo * ai_next
Definition: getaddrinfo.h:109
int errmsg(const char *fmt,...)
Definition: elog.c:797
#define _(x)
Definition: elog.c:84
void TouchSocketFiles ( void  )

Definition at line 806 of file pqcomm.c.

References lfirst, and NULL.

Referenced by ServerLoop().

807 {
808  ListCell *l;
809 
810  /* Loop through all created sockets... */
811  foreach(l, sock_paths)
812  {
813  char *sock_path = (char *) lfirst(l);
814 
815  /*
816  * utime() is POSIX standard, utimes() is a common alternative. If we
817  * have neither, there's no way to affect the mod or access time of
818  * the socket :-(
819  *
820  * In either path, we ignore errors; there's no point in complaining.
821  */
822 #ifdef HAVE_UTIME
823  utime(sock_path, NULL);
824 #else /* !HAVE_UTIME */
825 #ifdef HAVE_UTIMES
826  utimes(sock_path, NULL);
827 #endif /* HAVE_UTIMES */
828 #endif /* HAVE_UTIME */
829  }
830 }
static List * sock_paths
Definition: pqcomm.c:106
#define NULL
Definition: c.h:226
#define lfirst(lc)
Definition: pg_list.h:106

Variable Documentation

bool DoingCopyOut
static
WaitEventSet* FeBeWaitSet

Definition at line 167 of file pqcomm.c.

Referenced by secure_read(), secure_write(), SwitchBackToLocalLatch(), and SwitchToSharedLatch().

bool PqCommBusy
static

Definition at line 165 of file pqcomm.c.

Referenced by pq_redirect_to_shm_mq(), and pq_set_parallel_master().

PQcommMethods PqCommSocketMethods
static
Initial value:
= {
}
static int socket_flush_if_writable(void)
Definition: pqcomm.c:1442
static bool socket_is_send_pending(void)
Definition: pqcomm.c:1468
static void socket_comm_reset(void)
Definition: pqcomm.c:220
static void socket_startcopyout(void)
Definition: pqcomm.c:1567
static void socket_endcopyout(bool errorAbort)
Definition: pqcomm.c:1583
static int socket_putmessage(char msgtype, const char *s, size_t len)
Definition: pqcomm.c:1507
static int socket_flush(void)
Definition: pqcomm.c:1347
static void socket_putmessage_noblock(char msgtype, const char *s, size_t len)
Definition: pqcomm.c:1540

Definition at line 154 of file pqcomm.c.

char PqRecvBuffer[PQ_RECV_BUFFER_SIZE]
static
int PqRecvLength
static
int PqRecvPointer
static
char* PqSendBuffer
static

Definition at line 118 of file pqcomm.c.

Referenced by internal_flush(), internal_putbytes(), pq_init(), and socket_putmessage_noblock().

int PqSendBufferSize
static

Definition at line 119 of file pqcomm.c.

Referenced by internal_putbytes(), pq_init(), and socket_putmessage_noblock().

int PqSendStart
static
List* sock_paths = NIL
static

Definition at line 106 of file pqcomm.c.

char* Unix_socket_group

Definition at line 103 of file pqcomm.c.

int Unix_socket_permissions

Definition at line 102 of file pqcomm.c.

Referenced by show_unix_socket_permissions().