Support TCP_SYNCNT in libpq

Add a tcp_syn_count parameter to libpq.

This will allow the user to define the maximum amount of retransmission
that TCP will send before aborting the attempt to establish a
connection. This can be used only on systems where TCP_SYNCNT is
available.

Author: fcanovai

doc/src/sgml/libpq.sgml

@@ -1576,6 +1576,21 @@ postgresql://%2Fvar%2Flib%2Fpostgresql/dbname
       </listitem>
      </varlistentry>
 
+     <varlistentry id="libpq-tcp-syn-count" xreflabel="tcp_syn_count">
+      <term><literal>tcp_syn_count</literal></term>
+      <listitem>
+       <para>
+        Controls the maximum number of SYN retransmissions that TCP
+        will send before aborting the attempt to establish a connection.
+        A higher value increases the number of retries before giving up,
+        while a lower value results in faster failure detection.
+        This parameter is only effective on systems where
+        <symbol>TCP_SYNCNT</symbol> is available; on other systems,
+        it has no effect.
+       </para>
+      </listitem>
+     </varlistentry>
+
      <varlistentry id="libpq-connect-replication" xreflabel="replication">
       <term><literal>replication</literal></term>
       <listitem>

src/interfaces/libpq/fe-connect.c

@@ -279,6 +279,10 @@ static const internalPQconninfoOption PQconninfoOptions[] = {
 		"TCP-User-Timeout", "", 10, /* strlen(INT32_MAX) == 10 */
 	offsetof(struct pg_conn, pgtcp_user_timeout)},
 
+	{"tcp_syn_count", NULL, NULL, NULL,
+		"TCP-SYN-Count", "", 10, /* strlen(INT32_MAX) == 10 */
+	offsetof(struct pg_conn, tcp_syn_count)},
+
 	/*
 	 * ssl options are allowed even without client SSL support because the
 	 * client can still handle SSL modes "disable" and "allow". Other
@@ -2713,6 +2717,41 @@ setTCPUserTimeout(PGconn *conn)
 	return 1;
 }
 
+/*
+* Set the maximum number of SYN retransmissions.
+*/
+static int
+setTCPSynCnt(PGconn *conn)
+{
+	int			count;
+
+	if (conn->tcp_syn_count == NULL)
+		return 1;
+
+	if (!pqParseIntParam(conn->tcp_syn_count, &count, conn,
+						 "tcp_syn_count"))
+		return 0;
+
+	if (count < 0)
+		count = 0;
+
+#ifdef TCP_SYNCNT
+	if (setsockopt(conn->sock, IPPROTO_TCP, TCP_SYNCNT,
+				   (char *) &count, sizeof(count)) < 0)
+	{
+		char		sebuf[PG_STRERROR_R_BUFLEN];
+
+		libpq_append_conn_error(conn, "%s(%s) failed: %s",
+								"setsockopt",
+								"TCP_SYNCNT",
+								SOCK_STRERROR(SOCK_ERRNO, sebuf, sizeof(sebuf)));
+		return 0;
+	}
+#endif
+
+	return 1;
+}
+
 /* ----------
  * pqConnectDBStart -
  *		Begin the process of making a connection to the backend.
@@ -3454,6 +3493,15 @@ PQconnectPoll(PGconn *conn)
 						}
 					}
 
+					if (addr_cur->family != AF_UNIX)
+					{
+						if (!setTCPSynCnt(conn))
+						{
+							conn->try_next_addr = true;
+							goto keep_going;
+						}
+					}
+
 					/*----------
 					 * We have three methods of blocking SIGPIPE during
 					 * send() calls to this socket:
@@ -5130,6 +5178,7 @@ freePGconn(PGconn *conn)
 	free(conn->keepalives_idle);
 	free(conn->keepalives_interval);
 	free(conn->keepalives_count);
+	free(conn->tcp_syn_count);
 	free(conn->sslmode);
 	free(conn->sslnegotiation);
 	free(conn->sslcompression);

src/interfaces/libpq/libpq-int.h

@@ -405,6 +405,7 @@ struct pg_conn
 										 * retransmits */
 	char	   *keepalives_count;	/* maximum number of TCP keepalive
 									 * retransmits */
+	char	   *tcp_syn_count;		/* maximum number of TCP SYN retransmits */
 	char	   *sslmode;		/* SSL mode (require,prefer,allow,disable) */
 	char	   *sslnegotiation; /* SSL initiation style (postgres,direct) */
 	char	   *sslcompression; /* SSL compression (0 or 1) */