summaryrefslogtreecommitdiff
path: root/src/session
diff options
context:
space:
mode:
Diffstat (limited to 'src/session')
-rw-r--r--src/session/manager.rs3
-rw-r--r--src/session/session.rs20
-rw-r--r--src/session/tuple.rs2
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,