Expose libpq's OAuth hooks

Async is not supported yet,

Author: jeltz

ext/extconf.rb

@@ -316,6 +316,7 @@ module PG
   src + " int con(){ return PGRES_PIPELINE_SYNC; }"
 end
 have_func 'PQsetChunkedRowsMode', 'libpq-fe.h' # since PostgreSQL-17
+have_func 'PQsetAuthDataHook', 'libpq-fe.h' # since PostgreSQL-18
 have_func 'timegm'
 have_func 'rb_io_wait' # since ruby-3.0
 have_func 'rb_io_descriptor' # since ruby-3.1

ext/gvl_wrappers.h

@@ -281,17 +281,34 @@ FOR_EACH_BLOCKING_FUNCTION( DEFINE_GVL_STUB_DECL );
  * Definitions of callback functions and their parameters
  */
 
+#define FOR_EACH_PARAM_OF_auth_data_hook_proxy(param) \
+	param(PGauthData, type) \
+	param(PGconn *, conn)
+
 #define FOR_EACH_PARAM_OF_notice_processor_proxy(param) \
 	param(void *, arg)
 
 #define FOR_EACH_PARAM_OF_notice_receiver_proxy(param) \
 	param(void *, arg)
 
+#ifdef HAVE_PQSETAUTHDATAHOOK
+
 /* function( name, void_or_nonvoid, returntype, lastparamtype, lastparamname ) */
 #define FOR_EACH_CALLBACK_FUNCTION(function) \
+	function(auth_data_hook_proxy, GVL_TYPE_NONVOID, int, void *, data) \
 	function(notice_processor_proxy, GVL_TYPE_VOID, void, const char *, message) \
 	function(notice_receiver_proxy, GVL_TYPE_VOID, void, const PGresult *, result) \
 
+#else
+
+/* function( name, void_or_nonvoid, returntype, lastparamtype, lastparamname ) */
+#define FOR_EACH_CALLBACK_FUNCTION(function) \
+	function(notice_processor_proxy, GVL_TYPE_VOID, void, const char *, message) \
+	function(notice_receiver_proxy, GVL_TYPE_VOID, void, const PGresult *, result) \
+
+#endif
+
 FOR_EACH_CALLBACK_FUNCTION( DEFINE_GVL_STUB_DECL );
 
+
 #endif /* end __gvl_wrappers_h */

ext/pg.c

@@ -682,6 +682,7 @@ Init_pg_ext(void)
 
 	/* Initialize the main extension classes */
 	init_pg_connection();
+	init_pg_auth_hooks();
 	init_pg_result();
 	init_pg_errors();
 	init_pg_type_map();

ext/pg.h

@@ -289,6 +289,7 @@ extern VALUE pg_typemap_all_strings;
 void Init_pg_ext                                       _(( void ));
 
 void init_pg_connection                                _(( void ));
+void init_pg_auth_hooks                                _(( void ));
 void init_pg_result                                    _(( void ));
 void init_pg_errors                                    _(( void ));
 void init_pg_type_map                                  _(( void ));
@@ -375,6 +376,9 @@ rb_encoding * pg_get_pg_encname_as_rb_encoding         _(( const char * ));
 const char * pg_get_rb_encoding_as_pg_encoding         _(( rb_encoding * ));
 rb_encoding *pg_conn_enc_get                           _(( PGconn * ));
 
+#ifdef HAVE_PQSETAUTHDATAHOOK
+int auth_data_hook_proxy(PGauthData type, PGconn *conn, void *data);
+#endif
 void notice_receiver_proxy(void *arg, const PGresult *result);
 void notice_processor_proxy(void *arg, const char *message);