Skip to content

Commit

Permalink
Rename EventCategory::Unknown to Unspecified
Browse files Browse the repository at this point in the history
  • Loading branch information
syncpark committed Oct 21, 2024
1 parent 76a307b commit 591fe1e
Show file tree
Hide file tree
Showing 4 changed files with 11 additions and 7 deletions.
1 change: 1 addition & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ Versioning](https://semver.org/spec/v2.0.0.html).
### Changed

- `FromKeyValue` is sealed.
- Renamed `EventCategory::Unknown` to `Unspecified`.

### Removed

Expand Down
13 changes: 8 additions & 5 deletions src/event.rs
Original file line number Diff line number Diff line change
Expand Up @@ -4817,7 +4817,7 @@ mod tests {
issuer_org_name: "org".to_string(),
issuer_org_unit_name: "unit".to_string(),
issuer_common_name: "common".to_string(),
category: EventCategory::Unknown,
category: EventCategory::Unspecified,
last_alert: 1,
}
}
Expand All @@ -4836,7 +4836,7 @@ mod tests {
let syslog_message = message.to_string();
assert_eq!(
&syslog_message,
r#"time="1970-01-01T01:01:01+00:00" event_kind="SuspiciousTlsTraffic" category="Unknown" source="collector1" src_addr="127.0.0.1" src_port="10000" dst_addr="127.0.0.2" dst_port="443" proto="6" last_time="100" server_name="server" alpn_protocol="alpn" ja3="ja3" version="version" client_cipher_suites="1,2,3" client_extensions="4,5,6" cipher="1" extensions="7,8,9" ja3s="ja3s" serial="serial" subject_country="country" subject_org_name="org" subject_common_name="common" validity_not_before="100" validity_not_after="200" subject_alt_name="alt" issuer_country="country" issuer_org_name="org" issuer_org_unit_name="unit" issuer_common_name="common" last_alert="1""#
r#"time="1970-01-01T01:01:01+00:00" event_kind="SuspiciousTlsTraffic" category="Unspecified" source="collector1" src_addr="127.0.0.1" src_port="10000" dst_addr="127.0.0.2" dst_port="443" proto="6" last_time="100" server_name="server" alpn_protocol="alpn" ja3="ja3" version="version" client_cipher_suites="1,2,3" client_extensions="4,5,6" cipher="1" extensions="7,8,9" ja3s="ja3s" serial="serial" subject_country="country" subject_org_name="org" subject_common_name="common" validity_not_before="100" validity_not_after="200" subject_alt_name="alt" issuer_country="country" issuer_org_name="org" issuer_org_unit_name="unit" issuer_common_name="common" last_alert="1""#
);

let suspicious_tls_traffic =
Expand All @@ -4849,7 +4849,10 @@ mod tests {
suspicious_tls_traffic.dst_addr(),
IpAddr::V4(Ipv4Addr::new(127, 0, 0, 2))
);
assert_eq!(suspicious_tls_traffic.category(), EventCategory::Unknown);
assert_eq!(
suspicious_tls_traffic.category(),
EventCategory::Unspecified
);
assert_eq!(suspicious_tls_traffic.src_port(), 10000);
assert_eq!(suspicious_tls_traffic.dst_port(), 443);
assert_eq!(suspicious_tls_traffic.proto(), 6);
Expand All @@ -4858,7 +4861,7 @@ mod tests {

assert_eq!(
&block_list_tls,
r#"time="1970-01-01T01:01:01+00:00" event_kind="SuspiciousTlsTraffic" category="Unknown" source="collector1" src_addr="127.0.0.1" src_port="10000" dst_addr="127.0.0.2" dst_port="443" proto="6" last_time="100" server_name="server" alpn_protocol="alpn" ja3="ja3" version="version" client_cipher_suites="1,2,3" client_extensions="4,5,6" cipher="1" extensions="7,8,9" ja3s="ja3s" serial="serial" subject_country="country" subject_org_name="org" subject_common_name="common" validity_not_before="100" validity_not_after="200" subject_alt_name="alt" issuer_country="country" issuer_org_name="org" issuer_org_unit_name="unit" issuer_common_name="common" last_alert="1" triage_scores="""#
r#"time="1970-01-01T01:01:01+00:00" event_kind="SuspiciousTlsTraffic" category="Unspecified" source="collector1" src_addr="127.0.0.1" src_port="10000" dst_addr="127.0.0.2" dst_port="443" proto="6" last_time="100" server_name="server" alpn_protocol="alpn" ja3="ja3" version="version" client_cipher_suites="1,2,3" client_extensions="4,5,6" cipher="1" extensions="7,8,9" ja3s="ja3s" serial="serial" subject_country="country" subject_org_name="org" subject_common_name="common" validity_not_before="100" validity_not_after="200" subject_alt_name="alt" issuer_country="country" issuer_org_name="org" issuer_org_unit_name="unit" issuer_common_name="common" last_alert="1" triage_scores="""#
);
}

Expand Down Expand Up @@ -4918,7 +4921,7 @@ mod tests {

let mut counter = HashMap::new();
event.count_category(&mut counter, None, &filter).unwrap();
assert_eq!(counter.get(&EventCategory::Unknown), Some(&1));
assert_eq!(counter.get(&EventCategory::Unspecified), Some(&1));

let mut counter = HashMap::new();
event
Expand Down
2 changes: 1 addition & 1 deletion src/migration.rs
Original file line number Diff line number Diff line change
Expand Up @@ -278,7 +278,7 @@ fn migrate_0_30_tidb(store: &super::Store) -> Result<()> {
"HttpUriThreat" => EventCategory::Reconnaissance,
"ProcessCreate" => EventCategory::Impact,
"spamhaus drop ip" => EventCategory::InitialAccess,
_ => EventCategory::Unknown,
_ => EventCategory::Unspecified,

Check warning on line 281 in src/migration.rs

View check run for this annotation

Codecov / codecov/patch

src/migration.rs#L281

Added line #L281 was not covered by tests
};
let new_tidb = Tidb::try_from((old_tidb, category))?;

Expand Down
2 changes: 1 addition & 1 deletion src/types.rs
Original file line number Diff line number Diff line change
Expand Up @@ -76,7 +76,7 @@ pub struct PretrainedModel(pub Vec<u8>);
)]
#[repr(u8)]
pub enum EventCategory {
Unknown = 0,
Unspecified = 0,
Reconnaissance = 1,
InitialAccess,
Execution,
Expand Down

0 comments on commit 591fe1e

Please sign in to comment.