#proxy-server #http-request #http-proxy #basic-authentication #cache #graceful-shutdown #requests

bin+lib fortifynet_proxy

FortifyNet Proxy 是一个轻量级的 Rust 代理服务器,旨在提供具有基本认证和资源缓存功能的 HTTP 请求的安全和高效处理。

7 个版本 (稳定版)

1.1.9 2024年5月25日
1.1.8 2024年4月14日
1.1.7 2024年3月8日
1.1.6 2024年2月29日
0.1.5 2024年2月28日

#539 in 网络编程

MIT 许可证

16KB
138

FortifyNet Proxy:安全高效的 Rust 代理服务器

FortifyNet Proxy 是一个轻量级的 Rust 代理服务器,旨在提供具有基本认证和资源缓存功能的 HTTP 请求的安全和高效处理。

Crates.io Docs.rs

  • Github 克隆 : git clone https://github.com/JeninSutradhar/fortifynet_proxy

特性

  1. 代理认证:在允许访问资源之前,安全地验证用户。
  2. HTTP 请求转发:将传入的 HTTP 请求转发到目标服务器,并将响应回传给客户端。
  3. 活动记录:记录代理服务器活动,以便监控和故障排除。
  4. 缓存:缓存响应,以减少对目标服务器的负载并提高响应时间。
  5. 优雅关闭:优雅地关闭代理服务器,以确保数据完整性和用户体验。

安装

要在 Rust 项目中使用 FortifyNet Proxy,请将以下行添加到您的 Cargo.toml 文件中

[dependencies]
fortifynet_proxy = "1.1.9"

使用方法

基本使用方法

要使用 FortifyNet Proxy 服务器,请按照以下简单步骤操作

  • 定义一个 ProxyConfig 结构体来指定服务器配置参数。
  • 使用提供的配置使用 start_proxy_server 函数启动代理服务器。
use fortifynet_proxy::{start_proxy_server, ProxyConfig};

fn main() {
    
    // Create a proxy configuration with default values
    let config = ProxyConfig {
        ip_address: "127.0.0.1".to_string(),
        port: 8080,
        authentication: false,
        username: "admin".to_string(),
        password: "password".to_string(),
        cache_enabled: true, // Disable for Faster Execution
    };

    // Start the proxy server with the provided configuration
    start_proxy_server(&config);
}

自定义

FortifyNet Proxy 提供了丰富的配置选项

  • IP 地址和端口:指定服务器所需的 IP 地址和端口。
  • 认证:启用使用自定义用户名和密码的用户认证。
  • 资源缓存:实现缓存策略以存储频繁访问的资源并提高性能。

高级使用方法

自定义认证

配置自定义认证设置以强制执行用户访问控制。

use fortifynet_proxy::{ProxyConfig, start_proxy_server};

fn main() {
    // Configure custom authentication
    let config = ProxyConfig {
        ip_address: "127.0.0.1".to_string(),
        port: 8080,
        authentication: true,
        username: "admin".to_string(),
        password: "password123".to_string(),
        cache_enabled: true,
    };

    // Start the proxy server with custom authentication
    start_proxy_server(config);
}

资源缓存策略

实现资源缓存策略以优化网络性能。

use fortifynet_proxy::{ProxyConfig, start_proxy_server};

fn main() {
    // Configure resource caching
    let config = ProxyConfig {
        ip_address: "127.0.0.1".to_string(),
        port: 8080,
        authentication: false,
        username: "".to_string(),
        password: "".to_string(),
        cache_enabled: true,
    };

    // Start the proxy server with resource caching enabled
    start_proxy_server(config);
}

日志实现

修改 handle_http_request 函数以添加日志记录

pub fn handle_http_request(stream: &mut TcpStream, config: &ProxyConfig, cache: Arc<Mutex<HashMap<String, Vec<u8>>>>) -> std::io::Result<()> {
    let mut buffer = [0; 1024];
    stream.read(&mut buffer)?;

    let request_str = String::from_utf8_lossy(&buffer);
    log_activity(&format!("Incoming request: {}", request_str));

    let request_lines: Vec<&str> = request_str.lines().collect();
    if request_lines.is_empty() {
        return Err(std::io::Error::new(std::io::ErrorKind::InvalidInput, "Invalid HTTP request"));
    }

    let first_line = request_lines[0];
    let request_parts: Vec<&str> = first_line.split_whitespace().collect();
    if request_parts.len() != 3 {
        return Err(std::io::Error::new(std::io::ErrorKind::InvalidInput, "Invalid HTTP request line"));
    }

    let method = request_parts[0];
    let url = request_parts[1];
    let version = request_parts[2];

    let host = if let Some(host_line) = request_lines.iter().find(|&&line| line.starts_with("Host:")) {
        host_line.split_whitespace().nth(1).unwrap_or("")
    } else {
        return Err(std::io::Error::new(std::io::ErrorKind::InvalidInput, "No Host header found"));
    };

    if config.cache_enabled {
        let cache = cache.lock().unwrap();
        if let Some(response) = cache.get(url) {
            log_activity(&format!("Serving from cache: {}", url));
            stream.write_all(response)?;
            stream.flush()?;
            return Ok(());
        }
    }

    let target_address = format!("{}:80", host);
    let mut target_stream = TcpStream::connect(target_address)?;

    target_stream.write_all(buffer)?;
    target_stream.flush()?;

    let mut response_buffer = Vec::new();
    target_stream.read_to_end(&mut response_buffer)?;

    if config.cache_enabled {
        let mut cache = cache.lock().unwrap();
        cache.insert(url.to_string(), response_buffer.clone());
    }

    log_activity(&format!("Outgoing response: {:?}", response_buffer));
    stream.write_all(&response_buffer)?;
    stream.flush()?;
    Ok(())
}

pub fn log_activity(activity: &str) {
    println!("{}", activity);
}

其他资源

无运行时依赖