From 3e4350d2cfad570d691d329abb63eb0189c0731a Mon Sep 17 00:00:00 2001
From: Nadja Reitzenstein <me@dequbed.space>
Date: Fri, 11 Mar 2022 23:00:02 +0100
Subject: [PATCH] Import auditlog from v0.3.2

---
 bffhd/audit.rs | 43 +++++++++++++++++++++++++++++++++++++++++++
 bffhd/lib.rs   |  1 +
 2 files changed, 44 insertions(+)
 create mode 100644 bffhd/audit.rs

diff --git a/bffhd/audit.rs b/bffhd/audit.rs
new file mode 100644
index 0000000..2fee201
--- /dev/null
+++ b/bffhd/audit.rs
@@ -0,0 +1,43 @@
+use std::fs::{File, OpenOptions};
+use std::io;
+use std::io::{LineWriter, Write};
+use std::sync::Mutex;
+
+use crate::Config;
+use serde::{Serialize, Deserialize};
+use serde_json::Serializer;
+use time::OffsetDateTime;
+use crate::db::machine::{MachineIdentifier, MachineState};
+
+#[derive(Debug)]
+pub struct AuditLog {
+    writer: Mutex<LineWriter<File>>,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct AuditLogLine {
+    timestamp: i64,
+    machine: MachineIdentifier,
+    state: MachineState,
+}
+
+impl AuditLog {
+    pub fn new(config: &Config) -> io::Result<Self> {
+        let fd = OpenOptions::new().create(true).append(true).open(&config.auditlog_path)?;
+        let writer = Mutex::new(LineWriter::new(fd));
+        Ok(Self { writer })
+    }
+
+    pub fn log(&self, machine: &MachineIdentifier, state: &MachineState) -> io::Result<()> {
+        let timestamp = OffsetDateTime::now_utc().unix_timestamp();
+        let line = AuditLogLine { timestamp, machine: machine.clone(), state: state.clone() };
+
+        let mut guard = self.writer.lock().unwrap();
+        let mut writer: &mut LineWriter<File> = &mut *guard;
+
+        let mut ser = Serializer::new(&mut writer);
+        line.serialize(&mut ser).expect("failed to serialize audit log line");
+        writer.write("\n".as_bytes())?;
+        Ok(())
+    }
+}
diff --git a/bffhd/lib.rs b/bffhd/lib.rs
index 398c8e5..b1633ef 100644
--- a/bffhd/lib.rs
+++ b/bffhd/lib.rs
@@ -36,6 +36,7 @@ pub mod utils;
 mod tls;
 mod keylog;
 mod logging;
+mod audit;
 
 use std::fs::File;
 use std::io::BufReader;