數(shù)字金融
網(wǎng)絡營銷推廣
電商服務
文章主要介紹了Postgresql限制用戶登錄錯誤次數(shù)的實例代碼,本文給大家介紹的非常詳細,對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下。
在oracle中我們可以通過設置FAILED_LOGIN_ATTEMPTS來限制用戶密碼登錄錯誤的次數(shù),但是在postgresql中是不支持這個功能的。盡管PostgreSQL支持event trigger,可是event局限于DDL,對于登錄登出事件是沒辦法使用event trigger的。
不過像登錄新建會話觸發(fā)某個事件這個需求可以通過hook實現(xiàn),不過該方法比較復雜,需要修改內核代碼,在客戶端認證中添加邏輯,判斷輸入密碼次數(shù)統(tǒng)計。這里推薦一種比較簡單的方法實現(xiàn)類似的功能。
這里我們要使用到session_exec這個插件,使用該插件會在登錄時執(zhí)行一個指定的function。
下載地址:
https://github.com/okbob/session_exec
下載解壓之后需要進行以下配置:
set session_preload_libraries to session_execset
session_exec.login_name to name of your login function
該插件有以下特點:
如果函數(shù)不存在則會進行警告;
函數(shù)執(zhí)行失敗則不允許連接。
利用該插件我們可以寫一個簡單的函數(shù)來實現(xiàn)限制用戶登錄錯誤次數(shù)的功能。
例子:
1、建立外部表記錄數(shù)據(jù)庫日志信息。
CREATE SERVER pglog FOREIGN DATA WRAPPER file_fdw;
CREATE FOREIGN TABLE pglog (
log_time timestamp(3) with time zone,
user_name text,
database_name text,
process_id integer,
connection_from text,
session_id text,
session_line_num bigint,
command_tag text,
session_start_time timestamp with time zone,
virtual_transaction_id text,
transaction_id bigint,
error_severity text,
sql_state_code text,
message text,
detail text,
hint text,
internal_query text,
internal_query_pos integer,
context text,
query text,
query_pos integer,
location text,
application_name text,
backend_type text
) SERVER pglog
OPTIONS ( program 'find $PGDATA/log -type f -name "*.csv" -mtime -1 -exec cat {} \;', format 'csv' );
2、創(chuàng)建表t_login提取數(shù)據(jù)庫日志中的登錄信息。
create table t_login
(
login_time timestamp(3) with time zone --插入時間,
user_name text,
flag int --標志位,0代表過期數(shù)據(jù)
);
插入登錄信息:
bill=# insert into t_login select log_time,user_name from pglog where command_tag='authentication' and error_severity= 'FATAL'
bill-# ;
INSERT 0 4
3、創(chuàng)建登錄執(zhí)行的function
create or replace function lock_user() returns void as $$
declare
res text;
c1 timestamp(3) with time zone;
begin
select login_time from t_login where flag = 0 order by login_time desc limit 1 into c1; --獲取當前日志中最新時間
insert into t_login select log_time,user_name from pglog where command_tag='authentication' and error_severity= 'FATAL' and log_time > c1; --將最新的數(shù)據(jù)插入t_login表
update t_login set flag = 1 where login_time > c1;
for res in select user_name from t_login where flag = 1 group by user_name having count(*) >=3 --檢查登錄失敗次數(shù)是否大于3,若大于3則鎖定用戶
loop
EXECUTE format('alter user %I nologin',res); --鎖定用戶
EXECUTE 'select pg_terminate_backend(pid) from pg_stat_activity where usename=$1' using res; --斷開當前被鎖定用戶會話
raise notice 'Account % is locked!',res;
end loop;
end;
$$ language plpgsql strict;
4、編輯postgresql.conf文件,配置登錄函數(shù)
session_preload_libraries='session_exec'
session_exec.login_name='lock_user'
5、測試
模擬test1用戶登錄錯誤超過3次:
bill=# select * from t_login;
login_time | user_name | flag
----------------------------+-----------+------
2020-08-26 07:26:45.42+08 | test1 | 1
2020-08-26 07:26:50.179+08 | test1 | 1
2020-08-26 07:26:52.487+08 | test1 | 1
2020-08-26 07:26:54.537+08 | test1 | 1
(4 rows)
當我們在使用test1用戶登錄時則無法連接
pg13@cnndr4pptliot-> psql bill test1
Password for user test1:
NOTICE: c1 = <NULL>
psql: error: could not connect to server: FATAL: terminating connection due to administrator command
CONTEXT: SQL statement "select pg_terminate_backend(pid) from pg_stat_activity where usename=$1"
PL/pgSQL function lock_user() line 13 at EXECUTE
再次登錄可以看到提示該用戶被鎖定:
pg13@cnndr4pptliot-> psql bill test1
Password for user test1:
psql: error: could not connect to server: FATAL: role "test1" is not permitted to log in
6、解鎖用戶
此時想要解鎖該用戶則需要執(zhí)行:
bill=# alter user test1 login;
ALTER ROLE
然后需要注意還要將t_login中過期的數(shù)據(jù)修改。
bill=# update t_login set flag = 0;
UPDATE 4