diff options
Diffstat (limited to 'src/session')
| -rw-r--r-- | src/session/manager.rs | 3 | ||||
| -rw-r--r-- | src/session/session.rs | 20 | ||||
| -rw-r--r-- | src/session/tuple.rs | 2 |
3 files changed, 11 insertions, 14 deletions
diff --git a/src/session/manager.rs b/src/session/manager.rs index 462949f..36bb19e 100644 --- a/src/session/manager.rs +++ b/src/session/manager.rs @@ -11,7 +11,6 @@ use std::rc::Rc; * Struct ******************************************************************************/ -#[derive(Debug)] pub struct SessionManager { sessions: HashMap<String, Rc<RefCell<Session>>>, } @@ -154,7 +153,5 @@ mod tests { // Research Session assert_eq!(session_mgr.get_session(&session_id).is_none(), true); assert_eq!(session_mgr.get_session(&reversed_id).is_none(), true); - - dbg!(session_mgr); } } diff --git a/src/session/session.rs b/src/session/session.rs index e8b2be4..1db4bfd 100644 --- a/src/session/session.rs +++ b/src/session/session.rs @@ -7,19 +7,19 @@ use std::collections::HashMap; const MAX_SESSION_EXPIRE_TIME: i64 = 60; -#[derive(Clone, Copy, Debug, PartialEq, Eq)] +#[derive(Clone, Copy, Debug, PartialEq)] pub enum SessionDirection { C2S, S2C, } -#[derive(Clone, Copy, Debug, PartialEq, Eq)] +#[derive(Clone, Copy, Debug, PartialEq)] pub enum SessionProto { TCP, UDP, } -#[derive(Clone, Copy, Debug, PartialEq, Eq)] +#[derive(Clone, Copy, Debug, PartialEq)] pub enum SessionState { New, Active, @@ -27,7 +27,7 @@ pub enum SessionState { Expired, } -#[derive(Clone, Copy, Debug, PartialEq, Eq)] +#[derive(Debug)] pub struct SessionMetrics { pub send_pkts: u64, pub send_bytes: u64, @@ -35,7 +35,7 @@ pub struct SessionMetrics { pub recv_bytes: u64, } -#[derive(Clone, Copy, Debug, PartialEq, Eq)] +#[derive(Debug)] struct SessionTimeStamp { ts_start: i64, ts_end: i64, @@ -49,7 +49,7 @@ struct SessionTimeStamp { * - if current packet is iterm of session, require packet lifetime > session lifetime ******************************************************************************/ -#[derive(Clone, Debug, PartialEq, Eq)] +#[derive(Debug)] pub struct Session { session_id: String, session_proto: SessionProto, @@ -132,8 +132,8 @@ impl Session { self.session_metrics_c2s.recv_bytes += recv_bytes; } - pub fn get_session_c2s_metrics(&self) -> SessionMetrics { - self.session_metrics_c2s + pub fn get_session_c2s_metrics(&self) -> &SessionMetrics { + &self.session_metrics_c2s } pub fn inc_session_s2c_metrics( @@ -149,8 +149,8 @@ impl Session { self.session_metrics_s2c.recv_bytes += recv_bytes; } - pub fn get_session_s2c_metrics(&self) -> SessionMetrics { - self.session_metrics_s2c + pub fn get_session_s2c_metrics(&self) -> &SessionMetrics { + &self.session_metrics_s2c } pub fn update_session_expire_ts(&mut self) { diff --git a/src/session/tuple.rs b/src/session/tuple.rs index 9a4dacf..16616cb 100644 --- a/src/session/tuple.rs +++ b/src/session/tuple.rs @@ -2,7 +2,7 @@ * Struct ******************************************************************************/ -#[derive(Clone, Debug, PartialEq, Eq)] +#[derive(Debug, PartialEq)] pub struct FiveTuple { src_ip: String, src_port: u16, |
