Add a client authentication hook.

robertmhaas · robertmhaas · commit 20709f813601 · 2010-10-26T21:20:38.000-04:00
KaiGai Kohei, with minor cleanup of the comments by me.
diff --git a/src/backend/libpq/auth.c b/src/backend/libpq/auth.c
@@ -216,6 +216,12 @@ static int	CheckRADIUSAuth(Port *port);
  *----------------------------------------------------------------
  */
 
+/*
+ * This hook allows plugins to get control following client authentication,
+ * but before the user has been informed about the results.  It could be used
+ * to record login events, insert a delay after failed authentication, etc.
+ */
+ClientAuthentication_hook_type ClientAuthentication_hook = NULL;
 
 /*
  * Tell the user the authentication failed, but not (much about) why.
@@ -577,6 +583,9 @@ ClientAuthentication(Port *port)
 			break;
 	}
 
+	if (ClientAuthentication_hook)
+		(*ClientAuthentication_hook)(port, status);
+
 	if (status == STATUS_OK)
 		sendAuthRequest(port, AUTH_REQ_OK);
 	else
diff --git a/src/include/libpq/auth.h b/src/include/libpq/auth.h
@@ -24,4 +24,8 @@ extern char *pg_krb_realm;
 
 extern void ClientAuthentication(Port *port);
 
+/* Hook for plugins to get control in ClientAuthentication() */
+typedef void (*ClientAuthentication_hook_type)(Port *, int);
+extern PGDLLIMPORT ClientAuthentication_hook_type ClientAuthentication_hook;
+
 #endif   /* AUTH_H */
