Rust-Postgres 连接 PostgreSQL 16 的 TLS 配置指南

在使用 Rust 生态中的 tokio-postgres 库连接 PostgreSQL 数据库时，随着 PostgreSQL 16 的发布，许多开发者遇到了 TLS 连接问题。本文将深入分析问题原因并提供完整的解决方案。

问题背景

在 PostgreSQL 14 及更早版本中，开发者可以简单地使用 NoTls 模式建立连接。但随着 PostgreSQL 16 安全策略的增强，默认配置要求所有连接必须使用 TLS 加密，否则会收到"no pg_hba.conf entry"错误。

错误分析

当尝试无 TLS 连接时，PostgreSQL 16 会返回明确的错误信息：

no pg_hba.conf entry for host, user "postgres", database "postgres", no encryption

这表示数据库服务器已配置为拒绝未加密的连接请求。

解决方案

1. 使用 Rustls 实现 TLS 连接

推荐使用 rustls 作为 TLS 实现，以下是完整的连接代码示例：

use tokio_postgres_rustls::MakeRustlsConnect;
use rustls::{RootCertStore, ClientConfig};
use std::io::BufReader;

pub async fn get_pg_client_with_tls(
    pg_uri: String,
    pg_username: String,
    pg_password: String,
    pg_db_name: String,
) -> Result<Client, String> {
    let pg_connection_str = format!(
        "host={} user={} password={} dbname={} sslmode=require",
        pg_uri, pg_username, pg_password, pg_db_name
    );

    // 加载CA证书
    let bytes = PEM.as_bytes();
    let mut buf = BufReader::new(bytes);
    let mut root_store = RootCertStore::empty();
    root_store.add_parsable_certificates(rustls_pemfile::certs(&mut buf).unwrap();

    // 配置TLS客户端
    let config = ClientConfig::builder()
        .with_root_certificates(root_store)
        .with_no_client_auth();
    let tls = MakeRustlsConnect::new(config);

    // 建立连接
    let (client, connection) = tokio_postgres::connect(&pg_connection_str, tls)
        .await
        .unwrap();

    // 处理连接任务
    tokio::spawn(async move {
        if let Err(e) = connection.await {
            error!("connection error: {}", e);
            panic!("PostgreSQL connection error");
        }
    });
    
    Ok(client)
}

2. 关键点说明

1. 证书处理：必须正确加载数据库服务器的CA证书链。AWS RDS用户可以从官方文档获取最新的CA证书。

2. 连接字符串：务必在连接字符串中添加sslmode=require参数，明确要求使用TLS加密。

3. 错误处理：特别注意证书验证失败的情况，常见错误包括：

   • 证书链不完整
   • 自签名证书问题
   • 证书过期

进阶配置

对于生产环境，建议考虑以下增强措施：

1. 证书固定：实现证书固定(Pinning)以增强安全性。

2. 连接池：结合deadpool-postgres等连接池库管理数据库连接。

3. 配置管理：将证书和连接参数外部化配置，便于不同环境部署。

总结

PostgreSQL 16加强了安全策略，要求默认使用TLS加密连接。通过本文提供的rustls实现方案，开发者可以顺利建立安全连接。理解TLS配置原理不仅能解决当前问题，也为处理其他加密通信场景奠定了基础。