PG14:auth_delay 插件源码分析-Toy模板网

这篇具有很好参考价值的文章主要介绍了PG14:auth_delay 插件源码分析。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

auth_delay 让服务器在报告身份验证失败前短暂暂停，以增加对数据库密码进行暴力破解的难度。需要注意的是，这对阻止拒绝服务攻击毫无帮助，甚至可能加剧攻击，因为在报告身份验证失败前等待的进程仍会占用连接。

要使用这个模块必须要在 postgresql.conf 中配置参数

shared_preload_libraries = 'auth_delay'
auth_delay.milliseconds = '500'

这个代码比较简单，一共分为三个部分。

hook 函数

在 libpq 中定义了一个 ClientAuthentication_hook 函数指针，代码如下：

typedef void (*ClientAuthentication_hook_type) (Port *, int);

/*
 * 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;

_PG_init 函数

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;
}

这个 _PG_init 函数会在模块调用之前执行，这个函数主要实现的功能是定义了一个 GUC 参数 auth_delay.milliseconds 这个参数我们需要在 postgresql.conf 中进行配置，然后它将原来的钩子函数保存在了 original_client_auth_hook 变量中，然后将我们自己的 auth_delay_checks 函数赋给了 ClientAuthentication_hook 变量，这样我们就可以让数据库调用我们的函数了。

auth_delay_checks 函数

static void
auth_delay_checks(Port *port, int status)
{
	/*
	 * Any other plugins which use ClientAuthentication_hook.
	 */
	if (original_client_auth_hook)
		original_client_auth_hook(port, status);

	/*
	 * Inject a short delay if authentication failed.
	 */
	if (status != STATUS_OK)
	{
		pg_usleep(1000L * auth_delay_milliseconds);
	}
}

这个函数的意思是如果原来存在有钩子函数就先运行原来的钩子函数，然后我们看一下验证的状态 status 是不是正确，如果不正确，就延时 auth_delay_milliseconds 个时间。文章来源地址https://www.toymoban.com/news/detail-839879.html

到了这里，关于PG14:auth_delay 插件源码分析的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！