59 lines
No EOL
1.8 KiB
Rust
59 lines
No EOL
1.8 KiB
Rust
use sslrelay::{self, TLSConfig, TCPDataType, RelayConfig, HandlerCallbacks, CallbackRet};
|
|
|
|
// Handler object
|
|
#[derive(Clone)] // Must have Clone trait implemented.
|
|
struct Handler;
|
|
|
|
/*
|
|
Callback traits that can be used to read or inject/modify data
|
|
into upstream or downstream data.
|
|
*/
|
|
impl HandlerCallbacks for Handler {
|
|
|
|
// DownStream non blocking callback
|
|
fn ds_nb_callback(&self, _in_data: Vec<u8>) {
|
|
println!("[+] Data before complete rewrite:\n{:#04X?}", _in_data);
|
|
}
|
|
|
|
// DownStream blocking callback
|
|
fn ds_b_callback(&mut self, _in_data: Vec<u8>) -> CallbackRet {
|
|
_in_data.reverse();
|
|
println!("[+] Data rewritten to:\n{:#04X?}", _in_data);
|
|
CallbackRet::Relay(_in_data)
|
|
}
|
|
|
|
// UpStream non blocking callback
|
|
fn us_nb_callback(&self, _in_data: Vec<u8>) {
|
|
println!("[+] Data before complete rewrite:\n{:#04X?}", _in_data);
|
|
}
|
|
|
|
// UpStream blocking callback
|
|
fn us_b_callback(&mut self, _in_data: Vec<u8>) -> CallbackRet {
|
|
_in_data.reverse();
|
|
println!("[+] Data rewritten to:\n{:#04X?}", _in_data);
|
|
CallbackRet::Relay(_in_data)
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
|
|
// Create new SSLRelay object
|
|
let mut relay = sslrelay::SSLRelay::new(
|
|
Handler,
|
|
RelayConfig {
|
|
downstream_data_type: TCPDataType::TLS,
|
|
upstream_data_type: TCPDataType::TLS,
|
|
bind_host: "0.0.0.0".to_string(),
|
|
bind_port: "443".to_string(),
|
|
remote_host: "remote.com".to_string(),
|
|
remote_port: "443".to_string(),
|
|
tls_config: TLSConfig::FILE{
|
|
certificate_path: "./tls.crt".to_string(),
|
|
private_key_path: "./tls.key".to_string(),
|
|
},
|
|
}
|
|
);
|
|
|
|
// Start listening
|
|
relay.start();
|
|
} |