PostgreSQL外掛hook機制

yzs87發表於2018-10-31
internal_load_library postgresql->
	PG_init = (PG_init_t) pg_dlsym(file_scanner->handle, "_PG_init");
	if (PG_init)
			(*PG_init) ();
internal_unload_library(const char *libname)->
	PG_fini = (PG_fini_t) pg_dlsym(file_scanner->handle, "_PG_fini");
			if (PG_fini)
				(*PG_fini) ();
		
		
以ClientAuthentication_hook_type為例
auth.h:
//宣告外掛使用的函式
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;
auth.c:
//全域性變數初始化為NULL,在_PG_init函式中進行初始化賦值,如果該外掛載入,則ClientAuthentication_hook為
ClientAuthentication_hook_type ClientAuthentication_hook = NULL;
//如果ClientAuthentication_hook被賦值則執行植入的程式碼
InitPostgres->PerformAuthentication->ClientAuthentication->
	if (ClientAuthentication_hook)
		(*ClientAuthentication_hook) (port, status);
auth_delay.c:
static ClientAuthentication_hook_type original_client_auth_hook = NULL;
/*
 * Module Load Callback
 */
void _PG_init(void)
{
	/* Define custom GUC variables */
	DefineCustomIntVariable("auth_delay.milliseconds",
							"Milliseconds to delay before reporting authentication failure",
							NULL,
							&auth_delay_milliseconds,
							0,
							0, INT_MAX / 1000,
							PGC_SIGHUP,
							GUC_UNIT_MS,
							NULL,
							NULL,
							NULL);
	/* Install Hooks */
	original_client_auth_hook = ClientAuthentication_hook;
	ClientAuthentication_hook = auth_delay_checks;
}
/*
如果解除安裝則呼叫該函式,實際上是將ClientAuthentication_hook賦回原值
*/
void
_PG_fini(void)
{
	ClientAuthentication_hook=original_client_auth_hook;
}
/*
*/
static void auth_delay_checks(Port *port, int status)
{
	if (original_client_auth_hook)
		original_client_auth_hook(port, status);
	if (status != STATUS_OK){
		pg_usleep(1000L * auth_delay_milliseconds);
	}
}


來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/31493717/viewspace-2218294/,如需轉載,請註明出處,否則將追究法律責任。

相關文章