TOFU host trust: SPKI fingerprint pinning, tests, forget-host

1 files changed, 5 insertions, 1 deletions

diff --git a/src/bin/server/handler.rs b/src/bin/server/handler.rs
index 867910e..48e9d3a 100644
--- a/src/bin/server/handler.rs
+++ b/src/bin/server/handler.rs
@@ -8,7 +8,11 @@ use std::io::{BufRead, BufReader, Write};
 use std::net::TcpStream;
 use std::sync::Mutex;
 
-static DB_CONN: Lazy<Mutex<Connection>> = Lazy::new(|| Mutex::new(database::init_db()));
+static DB_CONN: Lazy<Mutex<Connection>> = Lazy::new(|| {
+    let conn = Connection::open(std::env::var("SERVER_DB").unwrap_or_else(|_| "server.db".to_string())).unwrap();
+    database::init_db(&conn);
+    Mutex::new(conn)
+});
 
 #[allow(dead_code)]
 pub fn handle_client(stream: TcpStream) {