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 网络编程
16KB
138 行
FortifyNet Proxy:安全高效的 Rust 代理服务器
FortifyNet Proxy 是一个轻量级的 Rust 代理服务器,旨在提供具有基本认证和资源缓存功能的 HTTP 请求的安全和高效处理。
- Github 克隆 :
git clone https://github.com/JeninSutradhar/fortifynet_proxy
特性
- 代理认证:在允许访问资源之前,安全地验证用户。
- HTTP 请求转发:将传入的 HTTP 请求转发到目标服务器,并将响应回传给客户端。
- 活动记录:记录代理服务器活动,以便监控和故障排除。
- 缓存:缓存响应,以减少对目标服务器的负载并提高响应时间。
- 优雅关闭:优雅地关闭代理服务器,以确保数据完整性和用户体验。
安装
要在 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);
}