Rust高性能HTTP代理库pingora-openssl的使用指南

概述

pingora-openssl是一个基于Rust的高性能HTTP代理库，专门支持OpenSSL加密与安全通信功能。该库结合了Rust的内存安全特性和OpenSSL的加密能力，为开发者提供安全可靠的HTTP代理解决方案。

主要特性

• 高性能异步处理
• 完整的TLS/SSL加密支持
• HTTP/1.1和HTTP/2协议支持
• 连接池和负载均衡
• 可定制的中间件系统

安装方法

在Cargo.toml中添加依赖：

[dependencies]
pingora-openssl = "0.1"
openssl = "0.10"

基本使用方法

1. 创建简单的HTTP代理服务器

use pingora_openssl::{proxy::HttpProxy, ssl::OpensslAcceptor};
use openssl::ssl::{SslAcceptor, SslFiletype, SslMethod};

#[tokio::main]
async fn main() {
    // 配置SSL
    let mut acceptor = SslAcceptor::mozilla_intermediate(SslMethod::tls()).unwrap();
    acceptor
        .set_private_key_file("key.pem", SslFiletype::PEM)
        .unwrap();
    acceptor.set_certificate_chain_file("cert.pem").unwrap();
    
    let ssl_acceptor = OpensslAcceptor::new(acceptor.build());
    
    // 创建代理实例
    let proxy = HttpProxy::new(
        "0.0.0.0:8080",
        ssl_acceptor,
    );
    
    println!("代理服务器启动在: https://0.0.0.0:8080");
    proxy.run().await.unwrap();
}

2. 添加自定义中间件

use pingora_openssl::prelude::*;
use std::sync::Arc;

struct LoggingMiddleware;

#[async_trait]
impl Middleware for LoggingMiddleware {
    async fn request(&self, ctx: &mut Context) -> Result<()> {
        println!("收到请求: {} {}", ctx.req.method(), ctx.req.uri());
        Ok(())
    }
    
    async fn response(&self, ctx: &mut Context) -> Result<()> {
        println!("返回响应: {}", ctx.res.status());
        Ok(())
    }
}

#[tokio::main]
async fn main() {
    let mut proxy = HttpProxy::new(
        "0.0.0.0:8080",
        OpensslAcceptor::from_files("cert.pem", "key.pem").unwrap(),
    );
    
    proxy.add_middleware(Arc::new(LoggingMiddleware));
    proxy.run().await.unwrap();
}

3. 配置上游服务器和负载均衡

use pingora_openssl::{proxy::HttpProxy, upstreams::service::Service};

#[tokio::main]
async fn main() {
    let mut proxy = HttpProxy::new(
        "0.0.0.0:8080",
        OpensslAcceptor::from_files("cert.pem", "key.pem").unwrap(),
    );
    
    // 添加上游服务器
    let backend1 = Service::new("https://backend1.example.com:443");
    let backend2 = Service::new("https://backend2.example.com:443");
    
    proxy.add_upstream(backend1);
    proxy.add_upstream(backend2);
    
    // 设置负载均衡策略
    proxy.set_load_balancer(pingora_openssl::lb::RoundRobin::new());
    
    proxy.run().await.unwrap();
}

高级配置示例

自定义SSL配置

use pingora_openssl::ssl::OpensslAcceptor;
use openssl::ssl::{SslAcceptor, SslMethod, SslOptions};

fn create_custom_ssl_acceptor() -> OpensslAcceptor {
    let mut builder = SslAcceptor::mozilla_intermediate(SslMethod::tls()).unwrap();
    
    // 自定义SSL选项
    builder.set_options(SslOptions::NO_SSLV2 | SslOptions::NO_SSLV3);
    
    // 设置密码套件
    builder.set_cipher_list("ECDHE-RSA-AES128-GCM-SHA256:ECDHE-ECDSA-AES128-GCM-SHA256").unwrap();
    
    OpensslAcceptor::new(builder.build())
}

运行和测试

1. 生成测试用的SSL证书：

openssl req -x509 -newkey rsa:4096 -nodes -keyout key.pem -out cert.pem -days 365

1. 运行代理服务器：

cargo run

1. 测试代理功能：

curl -x https://localhost:8080 -k https://example.com

注意事项

• 在生产环境中使用前，请确保使用有效的SSL证书
• 根据需要调整连接超时和缓冲区大小
• 监控代理服务器的性能和资源使用情况
• 定期更新OpenSSL库以获取安全补丁

这个库提供了强大的功能来构建安全、高性能的HTTP代理服务，适合需要加密通信的各种应用场景。

完整示例代码

use pingora_openssl::{
    proxy::HttpProxy, 
    ssl::OpensslAcceptor,
    prelude::*,
    upstreams::service::Service,
    lb::RoundRobin
};
use openssl::ssl::{SslAcceptor, SslFiletype, SslMethod, SslOptions};
use std::sync::Arc;

// 自定义日志中间件
struct LoggingMiddleware;

#[async_trait]
impl Middleware for LoggingMiddleware {
    async fn request(&self, ctx: &mut Context) -> Result<()> {
        println!("收到请求: {} {}", ctx.req.method(), ctx.req.uri());
        Ok(())
    }
    
    async fn response(&self, ctx: &mut Context) -> Result<()> {
        println!("返回响应: {}", ctx.res.status());
        Ok(())
    }
}

// 自定义SSL配置函数
fn create_ssl_acceptor() -> OpensslAcceptor {
    let mut builder = SslAcceptor::mozilla_intermediate(SslMethod::tls()).unwrap();
    
    // 设置私钥和证书文件
    builder
        .set_private_key_file("key.pem", SslFiletype::PEM)
        .unwrap();
    builder.set_certificate_chain_file("cert.pem").unwrap();
    
    // 自定义SSL安全选项
    builder.set_options(SslOptions::NO_SSLV2 | SslOptions::NO_SSLV3);
    
    // 设置密码套件
    builder.set_cipher_list("ECDHE-RSA-AES128-GCM-SHA256:ECDHE-ECDSA-AES128-GCM-SHA256").unwrap();
    
    OpensslAcceptor::new(builder.build())
}

#[tokio::main]
async fn main() {
    // 创建SSL接收器
    let ssl_acceptor = create_ssl_acceptor();
    
    // 创建HTTP代理实例
    let mut proxy = HttpProxy::new("0.0.0.0:8443", ssl_acceptor);
    
    // 添加日志中间件
    proxy.add_middleware(Arc::new(LoggingMiddleware));
    
    // 配置上游服务器
    let backend1 = Service::new("https://backend1.example.com:443");
    let backend2 = Service::new("https://backend2.example.com:443");
    let backend3 = Service::new("https://backend3.example.com:443");
    
    proxy.add_upstream(backend1);
    proxy.add_upstream(backend2);
    proxy.add_upstream(backend3);
    
    // 设置负载均衡策略为轮询
    proxy.set_load_balancer(RoundRobin::new());
    
    println!("高性能HTTP代理服务器启动在: https://0.0.0.0:8443");
    println!("已配置3个上游服务器并使用轮询负载均衡");
    
    // 运行代理服务器
    if let Err(e) = proxy.run().await {
        eprintln!("代理服务器运行错误: {}", e);
    }
}

配置文件示例

创建 Cargo.toml 文件：

[package]
name = "pingora-proxy-example"
version = "0.1.0"
edition = "2021"

[dependencies]
pingora-openssl = "0.1"
openssl = "0.10"
tokio = { version = "1.0", features = ["full"] }
async-trait = "0.1"

运行说明

1. 首先生成SSL证书：

openssl req -x509 -newkey rsa:4096 -nodes -keyout key.pem -out cert.pem -days 365 -subj "/CN=localhost"

1. 运行代理服务器：

cargo run

1. 测试代理功能：

curl -x https://localhost:8443 -k https://httpbin.org/get

这个完整示例展示了如何创建一个功能完整的高性能HTTP代理服务器，包含SSL加密、中间件日志、多上游服务器和负载均衡功能。