Added Callback return types.

This commit is contained in:
PinkP4nther 2021-09-16 02:11:49 -07:00
commit 953ff1fda5
4 changed files with 82 additions and 20 deletions

View file

@ -1,4 +1,4 @@
use sslrelay::{self, ConfigType, RelayConfig, HandlerCallbacks};
use sslrelay::{self, ConfigType, RelayConfig, HandlerCallbacks, CallbackRet};
// Handler object
#[derive(Clone)] // Must have Clone trait implemented.
@ -16,8 +16,9 @@ impl HandlerCallbacks for Handler {
}
// DownStream blocking callback
fn ds_b_callback(&self, _in_data: &mut Vec<u8>) {
fn ds_b_callback(&self, _in_data: Vec<u8>) {
println!("[CALLBACK] Down Stream Blocking CallBack!");
CallbackRet::Relay(_in_data)
}
// UpStream non blocking callback
@ -26,8 +27,9 @@ impl HandlerCallbacks for Handler {
}
// UpStream blocking callback
fn us_b_callback(&self, _in_data: &mut Vec<u8>) {
fn us_b_callback(&self, _in_data: Vec<u8>) {
println!("[CALLBACK] Up Stream Blocking CallBack!");
CallbackRet::Relay(_in_data)
}
}

View file

@ -16,9 +16,10 @@ impl HandlerCallbacks for Handler {
}
// DownStream blocking callback
fn ds_b_callback(&self, _in_data: &mut Vec<u8>) {
fn ds_b_callback(&self, _in_data: Vec<u8>) {
_in_data.reverse();
println!("[+] Data rewritten to:\n{:#04X?}", _in_data);
CallbackRet::Relay(_in_data)
}
// UpStream non blocking callback
@ -27,9 +28,10 @@ impl HandlerCallbacks for Handler {
}
// UpStream blocking callback
fn us_b_callback(&self, _in_data: &mut Vec<u8>) {
fn us_b_callback(&self, _in_data: Vec<u8>) {
_in_data.reverse();
println!("[+] Data rewritten to:\n{:#04X?}", _in_data);
CallbackRet::Relay(_in_data)
}
}