通过powershell远程访问postgresql数据库所在的服务器，临时开通用户访问数据库的权限，并在约定时间内关闭

最新推荐文章于 2024-03-28 16:37:48 发布

原创

最新推荐文章于 2024-03-28 16:37:48 发布 · 479 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据库 #服务器 #postgresql

该PowerShell脚本（pgallow.py）用于允许指定用户在给定的IP地址和持续时间内访问PostgreSQL数据库。它通过SSH连接到远程服务器，修改pg_hba.conf配置文件，加载新配置，并在设定时间后恢复原状。同时，脚本会终止所有非当前进程的数据库连接。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

以下是powershell的脚本内容
将脚本保存为pgallow.py

param (
    [Parameter(Mandatory = $true)][string]$user,
    [Parameter(Mandatory = $true)][string]$ip,
    [Parameter(Mandatory = $true)][string]$duration
)

# 设置数据库服务器信息
$server_ip = "192.168.2.53"
$server_user = "root"
$server_password = "123"
$server_ssh_port = 22

# 定义跨平台的 SSH 命令
$ssh_command = @"
echo '$server_password' | C:\tools\plink.exe -ssh $server_user@$server_ip -P $server_ssh_port -pw '$server_password' -hostkey 'ssh-ed25519 255 SHA256:sroxSvZ77nHCuAUKwZqSiwzMlC7U5ZmHAdOUuz5THc0' -batch -t "echo 'host all $user $ip/32 md5' >> /home/postgres/pgdata/pg_hba.conf; psql -U postgres -p 5544 -q -c 'select pg_reload_conf();'; sleep $duration; sed -i '/host all $user $ip\/32 md5/d' /home/postgres/pgdata/pg_hba.conf; psql -U postgres -p 5544 -q -c 'select pg_reload_conf(); select pg_terminate_backend(pid) from pg_stat_activity where pid <> pg_backend_pid() and client_addr = `'$ip`';'"
"@

# 运行 SSH 命令
Invoke-Expr