Fix Windows test errors

MythicAgents · Feb 25, 2024 · 68560da · 68560da
1 parent 2870ffb
commit 68560da
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/Payload_Type/thanatos/agent/thanatos/src/os/windows/mod.rs b/Payload_Type/thanatos/agent/thanatos/src/os/windows/mod.rs
@@ -1,12 +1,23 @@
 use errors::ThanatosError;
-use ffiwrappers::windows::processthreadsapi::token::CurrentToken;
-pub use ffiwrappers::windows::{domain, hostname, sysinfoapi, username};
+use ffiwrappers::windows::{processthreadsapi::token::CurrentToken, sysinfoapi};
 
 mod platform;
 pub use platform::{build_number, product};
 
 use crate::proto::checkin::Architecture;
 
+pub fn username() -> Result<String, ThanatosError> {
+    ffiwrappers::windows::username().map_err(ThanatosError::FFIError)
+}
+
+pub fn hostname() -> Result<String, ThanatosError> {
+    ffiwrappers::windows::hostname().map_err(ThanatosError::FFIError)
+}
+
+pub fn domain() -> Result<String, ThanatosError> {
+    ffiwrappers::windows::domain().map_err(ThanatosError::FFIError)
+}
+
 pub fn process_name() -> Result<String, ThanatosError> {
     ffiwrappers::windows::process_name().map_err(ThanatosError::FFIError)
 }