Compare commits

...
23 changed files with 277 additions and 109 deletions

16
Cargo.lock generated
View file

@ -39,6 +39,17 @@ dependencies = [
"serde", "serde",
] ]
[[package]]
name = "async-trait"
version = "0.1.92"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "82f6aeea286b8eb4dd3431a1be1b59d290ace00f5bfd8e2a159bc2a05e2c1667"
dependencies = [
"proc-macro2",
"quote",
"syn 3.0.4",
]
[[package]] [[package]]
name = "atomic-waker" name = "atomic-waker"
version = "1.1.2" version = "1.1.2"
@ -169,6 +180,9 @@ name = "camino"
version = "1.2.5" version = "1.2.5"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "bb1307f12aa967b5a58416e87b3653360e0fd614a016b6e970db08fecbb1b80d" checksum = "bb1307f12aa967b5a58416e87b3653360e0fd614a016b6e970db08fecbb1b80d"
dependencies = [
"serde_core",
]
[[package]] [[package]]
name = "cc" name = "cc"
@ -520,6 +534,7 @@ name = "llldap"
version = "0.1.0" version = "0.1.0"
dependencies = [ dependencies = [
"argh", "argh",
"async-trait",
"axum", "axum",
"axum-extra", "axum-extra",
"camino", "camino",
@ -530,6 +545,7 @@ dependencies = [
"minijinja", "minijinja",
"minijinja-embed", "minijinja-embed",
"serde", "serde",
"serde_json",
"static-serve", "static-serve",
"tokio", "tokio",
"tokio-util", "tokio-util",

View file

@ -5,9 +5,10 @@ edition = "2024"
[dependencies] [dependencies]
argh = "0.1.19" argh = "0.1.19"
async-trait = "0.1.92"
axum = { version = "0.8.9", optional = true, features = ["macros"] } axum = { version = "0.8.9", optional = true, features = ["macros"] }
axum-extra = { version = "0.12.6", features = ["cookie"], optional = true } axum-extra = { version = "0.12.6", features = ["cookie"], optional = true }
camino = "1.2.5" camino = { version = "1.2.5", features = ["serde1"] }
dn_escape = { path = "vendor/dn_escape" } dn_escape = { path = "vendor/dn_escape" }
futures-util = { version = "0.3.34", features = ["sink"] } futures-util = { version = "0.3.34", features = ["sink"] }
http = { version = "1.5.0", optional = true } http = { version = "1.5.0", optional = true }
@ -15,6 +16,7 @@ ldap3_proto = "0.8.1"
minijinja = { version = "2.24.0", optional = true } minijinja = { version = "2.24.0", optional = true }
minijinja-embed = { version = "2.24.0", optional = true } minijinja-embed = { version = "2.24.0", optional = true }
serde = { version = "1.0.229", features = ["derive"] } serde = { version = "1.0.229", features = ["derive"] }
serde_json = "1.0.151"
static-serve = { version = "0.6.3", optional = true } static-serve = { version = "0.6.3", optional = true }
tokio = { version = "1.53.1", features = ["macros", "net", "rt", "time", "sync"] } tokio = { version = "1.53.1", features = ["macros", "net", "rt", "time", "sync"] }
tokio-util = { version = "0.7.19", features = ["codec"] } tokio-util = { version = "0.7.19", features = ["codec"] }

View file

@ -1,8 +1,13 @@
use argh::FromArgs; use argh::FromArgs;
use camino::Utf8PathBuf;
/// Run the llldap server /// Run the llldap server
#[derive(FromArgs)] #[derive(FromArgs)]
pub struct CliArgs { pub struct CliArgs {
/// path to the JSON file for the database
#[argh(option)]
pub db: Option<Utf8PathBuf>,
/// address or socket to listen on for LDAP connections /// address or socket to listen on for LDAP connections
#[argh(option, default = "String::from(\"127.0.0.1:3389\")")] #[argh(option, default = "String::from(\"127.0.0.1:3389\")")]
pub listen_ldap: String, pub listen_ldap: String,

View file

@ -6,19 +6,19 @@ use crate::db::error::BoxedError;
use crate::db::{DatabaseInterface, Domain, User, UserRef}; use crate::db::{DatabaseInterface, Domain, User, UserRef};
#[derive(Clone, Debug)] #[derive(Clone, Debug)]
pub struct Database<D: DatabaseInterface> { pub struct Database {
pub inner: Arc<RwLock<D>>, pub inner: Arc<RwLock<Box<dyn DatabaseInterface>>>,
} }
impl<D: DatabaseInterface> Database<D> { impl Database {
pub fn new(db: D) -> Self { pub fn new(db: impl DatabaseInterface) -> Self {
Self { Self {
inner: Arc::new(RwLock::new(db)), inner: Arc::new(RwLock::new(Box::new(db))),
} }
} }
} }
impl<D: DatabaseInterface> Database<D> { impl Database {
/// Return false if the user doesn't exist, or the password is wrong. /// Return false if the user doesn't exist, or the password is wrong.
/// ///
/// TODO: should we return something else when the account doesn't exist? /// TODO: should we return something else when the account doesn't exist?

View file

@ -1,6 +1,6 @@
use serde::Serialize; use serde::{Deserialize, Serialize};
#[derive(Clone, Debug, Default, Serialize)] #[derive(Clone, Debug, Default, Deserialize, Serialize)]
pub struct Domain { pub struct Domain {
pub name: String, pub name: String,
} }

156
src/db/filesystem.rs Normal file
View file

@ -0,0 +1,156 @@
use camino::{Utf8Path, Utf8PathBuf};
use serde::{Deserialize, Serialize};
use serde_json::Error as JsonError;
use std::fmt;
use std::io::Error as IOError;
use crate::db::error::{BoxedError, DomainCreationError, UserCreationError};
use crate::db::{Database, DatabaseInterface, Domain, MemoryDatabase, User, UserRef};
#[derive(Debug)]
pub enum FilesystemDatabaseError {
ReadFileIO(Utf8PathBuf, IOError),
ReadFileJson(Utf8PathBuf, JsonError),
WriteFileIO(Utf8PathBuf, IOError),
WriteFileJson(Utf8PathBuf, JsonError),
}
impl fmt::Display for FilesystemDatabaseError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(
f,
"{}",
match self {
Self::ReadFileIO(path, e) => format!("Failed to read database file {path}: {e}"),
Self::ReadFileJson(path, e) =>
format!("Failed to parse database JSON file {path}: {e}"),
Self::WriteFileIO(path, e) => format!("Failed to write database file {path}: {e}"),
Self::WriteFileJson(path, e) =>
format!("Failed to convert database to JSON file {path}: {e}"),
}
)
}
}
impl std::error::Error for FilesystemDatabaseError {}
#[derive(Clone, Debug, Default, Deserialize, Serialize)]
pub struct FilesystemDatabase {
// TODO: Once we have common validation steps in place across DB backends, we can reduce cloning.
// For now, we clone the DB on every write operation, and update it when saving to disk
// succeeds.
pub inner: MemoryDatabase,
pub path: Utf8PathBuf,
}
impl FilesystemDatabase {
pub async fn from_path(path: impl AsRef<Utf8Path>) -> Result<Database, BoxedError> {
let path = path.as_ref();
if !tokio::fs::try_exists(path)
.await
.map_err(|e| Box::new(FilesystemDatabaseError::ReadFileIO(path.to_path_buf(), e)))?
{
// The database doesn't exist yet, we create an empty one and try to write it
// to make sure the permissions are correct and the destination folder exists.
let mut db = Self {
inner: MemoryDatabase::default(),
path: path.to_path_buf(),
};
db.save().await?;
return Ok(Database::new(db));
}
let s = tokio::fs::read(path)
.await
.map_err(|e| Box::new(FilesystemDatabaseError::ReadFileIO(path.to_path_buf(), e)))?;
let inner: MemoryDatabase = serde_json::from_slice(&s)
.map_err(|e| Box::new(FilesystemDatabaseError::ReadFileJson(path.to_path_buf(), e)))?;
Ok(Database::new(Self {
inner,
path: path.to_path_buf(),
}))
}
pub async fn save(&mut self) -> Result<(), BoxedError> {
let s = serde_json::to_string(&self)
.map_err(|e| Box::new(FilesystemDatabaseError::WriteFileJson(self.path.clone(), e)))?;
Ok(tokio::fs::write(&self.path, &s)
.await
.map_err(|e| Box::new(FilesystemDatabaseError::WriteFileIO(self.path.clone(), e)))?)
}
}
#[async_trait::async_trait]
impl DatabaseInterface for FilesystemDatabase {
async fn create_domain(
&mut self,
domain: &str,
) -> Result<Result<(), DomainCreationError>, BoxedError> {
let mut new_db = self.inner.clone();
if let Err(e) = new_db.create_domain(domain).await? {
return Ok(Err(e));
}
self.save().await?;
self.inner = new_db;
Ok(Ok(()))
}
async fn get_domain(&self, domain: &str) -> Result<Option<Domain>, BoxedError> {
self.inner.get_domain(domain).await
}
async fn get_user(&self, req_user: &UserRef) -> Result<Option<User>, BoxedError> {
self.inner.get_user(req_user).await
}
async fn create_user(
&mut self,
user: User,
) -> Result<Result<(), UserCreationError>, BoxedError> {
let mut new_db = self.inner.clone();
if let Err(e) = new_db.create_user(user).await? {
return Ok(Err(e));
}
self.save().await?;
self.inner = new_db;
Ok(Ok(()))
}
async fn try_create_user(
&mut self,
new_user: User,
current_user: &User,
) -> Result<Result<(), UserCreationError>, BoxedError> {
let mut new_db = self.inner.clone();
if let Err(e) = new_db.try_create_user(new_user, current_user).await? {
return Ok(Err(e));
}
self.save().await?;
self.inner = new_db;
Ok(Ok(()))
}
async fn list_all_domains(&self) -> Result<Vec<Domain>, BoxedError> {
self.inner.list_all_domains().await
}
async fn list_all_users(&self) -> Result<Vec<User>, BoxedError> {
self.inner.list_all_users().await
}
async fn list_domain_users(&self, domain: Option<String>) -> Result<Vec<User>, BoxedError> {
self.inner.list_domain_users(domain).await
}
}

View file

@ -1,7 +1,7 @@
#[derive(Clone, Debug)] use serde::{Deserialize, Serialize};
#[derive(Clone, Debug, Deserialize, Serialize)]
pub struct Group { pub struct Group {
#[expect(unused)]
name: String, name: String,
#[expect(unused)]
domain: String, domain: String,
} }

View file

@ -1,7 +1,8 @@
use crate::db::error::{BoxedError, DomainCreationError, UserCreationError}; use crate::db::error::{BoxedError, DomainCreationError, UserCreationError};
use crate::db::{Database, Domain, User, UserRef}; use crate::db::{Database, Domain, User, UserRef};
impl<D: DatabaseInterface> DatabaseInterface for Database<D> { #[async_trait::async_trait]
impl DatabaseInterface for Database {
async fn create_domain( async fn create_domain(
&mut self, &mut self,
domain: &str, domain: &str,
@ -49,39 +50,31 @@ impl<D: DatabaseInterface> DatabaseInterface for Database<D> {
} }
} }
pub trait DatabaseInterface: Clone + Send + Sync + 'static { #[async_trait::async_trait]
fn create_domain( pub trait DatabaseInterface: std::fmt::Debug + Send + Sync + 'static {
async fn create_domain(
&mut self, &mut self,
domain: &str, domain: &str,
) -> impl Future<Output = Result<Result<(), DomainCreationError>, BoxedError>> + Send; ) -> Result<Result<(), DomainCreationError>, BoxedError>;
fn get_domain( async fn get_domain(&self, domain: &str) -> Result<Option<Domain>, BoxedError>;
&self,
domain: &str,
) -> impl Future<Output = Result<Option<Domain>, BoxedError>> + Send;
fn get_user( async fn get_user(&self, user: &UserRef) -> Result<Option<User>, BoxedError>;
&self,
user: &UserRef,
) -> impl std::future::Future<Output = Result<Option<User>, BoxedError>> + Send;
async fn create_user( async fn create_user(
&mut self, &mut self,
user: User, user: User,
) -> Result<Result<(), UserCreationError>, BoxedError>; ) -> Result<Result<(), UserCreationError>, BoxedError>;
fn try_create_user( async fn try_create_user(
&mut self, &mut self,
new_user: User, new_user: User,
current_user: &User, current_user: &User,
) -> impl Future<Output = Result<Result<(), UserCreationError>, BoxedError>> + Send; ) -> Result<Result<(), UserCreationError>, BoxedError>;
fn list_all_domains(&self) -> impl Future<Output = Result<Vec<Domain>, BoxedError>> + Send; async fn list_all_domains(&self) -> Result<Vec<Domain>, BoxedError>;
#[expect(unused)] #[expect(unused)]
fn list_all_users(&self) -> impl Future<Output = Result<Vec<User>, BoxedError>> + Send; async fn list_all_users(&self) -> Result<Vec<User>, BoxedError>;
/// List users on a specific domain. /// List users on a specific domain.
/// ///
/// A `None` domain requested lists global service users. /// A `None` domain requested lists global service users.
fn list_domain_users( async fn list_domain_users(&self, domain: Option<String>) -> Result<Vec<User>, BoxedError>;
&self,
domain: Option<String>,
) -> impl std::future::Future<Output = Result<Vec<User>, BoxedError>> + Send;
} }

View file

@ -1,64 +1,60 @@
use std::future::{Future, ready}; use serde::{Deserialize, Serialize};
use crate::db::error::{BoxedError, DomainCreationError, UserCreationError}; use crate::db::error::{BoxedError, DomainCreationError, UserCreationError};
use crate::db::{Database, DatabaseInterface, Domain, Group, Operation, User, UserRef}; use crate::db::{Database, DatabaseInterface, Domain, Group, Operation, User, UserRef};
#[derive(Clone, Debug, Default)] #[derive(Clone, Debug, Default, Deserialize, Serialize)]
pub struct MemoryDatabase { pub struct MemoryDatabase {
// We store data in tables like in SQL // We store data in tables like in SQL
pub domains: Vec<Domain>, pub domains: Vec<Domain>,
#[expect(unused)]
pub groups: Vec<Group>, pub groups: Vec<Group>,
pub users: Vec<User>, pub users: Vec<User>,
} }
impl MemoryDatabase { impl MemoryDatabase {
pub fn new() -> Database<Self> { #[allow(clippy::new_ret_no_self)]
pub fn new() -> Database {
Database::new(Self::default()) Database::new(Self::default())
} }
} }
#[async_trait::async_trait]
impl DatabaseInterface for MemoryDatabase { impl DatabaseInterface for MemoryDatabase {
fn create_domain( async fn create_domain(
&mut self, &mut self,
domain: &str, domain: &str,
) -> impl Future<Output = Result<Result<(), DomainCreationError>, BoxedError>> { ) -> Result<Result<(), DomainCreationError>, BoxedError> {
if domain.is_empty() { if domain.is_empty() {
return ready(Ok(Err(DomainCreationError::InvalidDomain( return Ok(Err(DomainCreationError::InvalidDomain(domain.to_string())));
domain.to_string(),
))));
} }
if self.domains.iter().find(|d| d.name == domain).is_some() { if self.domains.iter().find(|d| d.name == domain).is_some() {
return ready(Ok(Err(DomainCreationError::DomainAlreadyExists( return Ok(Err(DomainCreationError::DomainAlreadyExists(
domain.to_string(), domain.to_string(),
)))); )));
} }
self.domains.push(Domain { self.domains.push(Domain {
name: domain.to_string(), name: domain.to_string(),
}); });
ready(Ok(Ok(()))) Ok(Ok(()))
} }
fn get_domain(&self, domain: &str) -> impl Future<Output = Result<Option<Domain>, BoxedError>> { async fn get_domain(&self, domain: &str) -> Result<Option<Domain>, BoxedError> {
let Some(domain) = self.domains.iter().find(|d| d.name == domain) else { let Some(domain) = self.domains.iter().find(|d| d.name == domain) else {
return ready(Ok(None)); return Ok(None);
}; };
ready(Ok(Some(domain.clone()))) Ok(Some(domain.clone()))
} }
fn get_user( async fn get_user(&self, req_user: &UserRef) -> Result<Option<User>, BoxedError> {
&self, Ok(self
req_user: &UserRef,
) -> impl Future<Output = Result<Option<User>, BoxedError>> {
ready(Ok(self
.users .users
.iter() .iter()
.find(|u| u.username == req_user.username && u.domain == req_user.domain) .find(|u| u.username == req_user.username && u.domain == req_user.domain)
.cloned())) .cloned())
} }
async fn create_user( async fn create_user(
@ -103,23 +99,20 @@ impl DatabaseInterface for MemoryDatabase {
self.create_user(new_user).await self.create_user(new_user).await
} }
fn list_all_domains(&self) -> impl Future<Output = Result<Vec<Domain>, BoxedError>> { async fn list_all_domains(&self) -> Result<Vec<Domain>, BoxedError> {
ready(Ok(self.domains.clone())) Ok(self.domains.clone())
} }
fn list_all_users(&self) -> impl Future<Output = Result<Vec<User>, BoxedError>> { async fn list_all_users(&self) -> Result<Vec<User>, BoxedError> {
ready(Ok(self.users.clone())) Ok(self.users.clone())
} }
fn list_domain_users( async fn list_domain_users(&self, domain: Option<String>) -> Result<Vec<User>, BoxedError> {
&self, Ok(self
domain: Option<String>,
) -> impl Future<Output = Result<Vec<User>, BoxedError>> {
ready(Ok(self
.users .users
.iter() .iter()
.filter(|u| u.domain == domain) .filter(|u| u.domain == domain)
.cloned() .cloned()
.collect())) .collect())
} }
} }

View file

@ -3,6 +3,8 @@ pub use common::Database;
mod domain; mod domain;
pub use domain::Domain; pub use domain::Domain;
pub mod error; pub mod error;
mod filesystem;
pub use filesystem::FilesystemDatabase;
mod group; mod group;
pub use group::Group; pub use group::Group;
mod interface; mod interface;

View file

@ -1,4 +1,4 @@
use serde::Serialize; use serde::{Deserialize, Serialize};
#[derive(Clone, Debug)] #[derive(Clone, Debug)]
pub enum Operation { pub enum Operation {
@ -7,12 +7,11 @@ pub enum Operation {
ListUsers(Option<String>), ListUsers(Option<String>),
} }
#[derive(Clone, Debug, PartialEq, Serialize)] #[derive(Clone, Debug, PartialEq, Deserialize, Serialize)]
pub enum Role { pub enum Role {
/// Can do anything /// Can do anything
Admin, Admin,
/// Can only read data across all vhosts /// Can only read data across all vhosts
#[expect(unused)]
ReadonlyAdmin, ReadonlyAdmin,
/// Can do anything on a domain, except removing /// Can do anything on a domain, except removing
/// oneself as a domain admin. /// oneself as a domain admin.
@ -20,7 +19,6 @@ pub enum Role {
/// Can not give away roles other than DomainModerator/User /// Can not give away roles other than DomainModerator/User
DomainAdmin(String), DomainAdmin(String),
/// Can create users and reset passwords on a domain /// Can create users and reset passwords on a domain
#[expect(unused)]
DomainModerator(String), DomainModerator(String),
/// Can only edit own profile /// Can only edit own profile
User, User,

View file

@ -1,4 +1,4 @@
use serde::Serialize; use serde::{Deserialize, Serialize};
use std::fmt; use std::fmt;
@ -54,7 +54,7 @@ impl fmt::Display for UserRef {
} }
} }
#[derive(Clone, Debug, Serialize)] #[derive(Clone, Debug, Deserialize, Serialize)]
pub struct User { pub struct User {
/// Username, without the domain part. Once set, cannot be edited. /// Username, without the domain part. Once set, cannot be edited.
pub username: String, pub username: String,

View file

@ -7,8 +7,8 @@ use serde::Deserialize;
use crate::db::{DatabaseInterface, Operation}; use crate::db::{DatabaseInterface, Operation};
use crate::http::{HttpSession, HttpState}; use crate::http::{HttpSession, HttpState};
pub async fn get_domain<D: DatabaseInterface>( pub async fn get_domain(
State(state): State<HttpState<D>>, State(state): State<HttpState>,
session: HttpSession, session: HttpSession,
Path(domain): Path<String>, Path(domain): Path<String>,
) -> Response { ) -> Response {
@ -53,8 +53,8 @@ pub struct DomainCreationForm {
domainname: String, domainname: String,
} }
pub async fn create_domain<D: DatabaseInterface>( pub async fn create_domain(
State(mut state): State<HttpState<D>>, State(mut state): State<HttpState>,
session: HttpSession, session: HttpSession,
Form(form): Form<DomainCreationForm>, Form(form): Form<DomainCreationForm>,
) -> Response { ) -> Response {

View file

@ -6,8 +6,8 @@ use minijinja::context;
use crate::db::{DatabaseInterface, Operation}; use crate::db::{DatabaseInterface, Operation};
use crate::http::{HttpSession, HttpState}; use crate::http::{HttpSession, HttpState};
pub async fn home<D: DatabaseInterface>( pub async fn home(
State(state): State<HttpState<D>>, State(state): State<HttpState>,
// Only logged in users are allowed here // Only logged in users are allowed here
session: HttpSession, session: HttpSession,
) -> Response { ) -> Response {

View file

@ -20,8 +20,8 @@ pub enum LoginError {
SessionInvalidated, SessionInvalidated,
} }
pub async fn login_page<D: DatabaseInterface>( pub async fn login_page(
State(state): State<HttpState<D>>, State(state): State<HttpState>,
login_error: Option<LoginError>, login_error: Option<LoginError>,
) -> Response { ) -> Response {
let page = state let page = state
@ -33,8 +33,8 @@ pub async fn login_page<D: DatabaseInterface>(
(StatusCode::OK, Html(page)).into_response() (StatusCode::OK, Html(page)).into_response()
} }
pub async fn get_login<D: DatabaseInterface>( pub async fn get_login(
State(state): State<HttpState<D>>, State(state): State<HttpState>,
maybe_session: Option<OptionalHttpSession>, maybe_session: Option<OptionalHttpSession>,
) -> Response { ) -> Response {
if maybe_session.is_some() { if maybe_session.is_some() {
@ -44,8 +44,8 @@ pub async fn get_login<D: DatabaseInterface>(
login_page(State(state), None).await login_page(State(state), None).await
} }
pub async fn post_login<D: DatabaseInterface>( pub async fn post_login(
State(state): State<HttpState<D>>, State(state): State<HttpState>,
session: Option<OptionalHttpSession>, session: Option<OptionalHttpSession>,
cookies: CookieJar, cookies: CookieJar,
Form(form): Form<LoginForm>, Form(form): Form<LoginForm>,

View file

@ -2,12 +2,11 @@ use axum::extract::State;
use axum::response::{IntoResponse, Redirect, Response}; use axum::response::{IntoResponse, Redirect, Response};
use axum_extra::extract::cookie::CookieJar; use axum_extra::extract::cookie::CookieJar;
use crate::db::DatabaseInterface;
use crate::http::login::{LoginError, login_page}; use crate::http::login::{LoginError, login_page};
use crate::http::{HttpState, OptionalHttpSession}; use crate::http::{HttpState, OptionalHttpSession};
pub async fn logout<D: DatabaseInterface>( pub async fn logout(
State(state): State<HttpState<D>>, State(state): State<HttpState>,
session: Option<OptionalHttpSession>, session: Option<OptionalHttpSession>,
cookies: CookieJar, cookies: CookieJar,
) -> Response { ) -> Response {

View file

@ -9,7 +9,7 @@ use static_serve::embed_assets;
#[cfg(not(feature = "embed"))] #[cfg(not(feature = "embed"))]
use tower_http::services::ServeDir; use tower_http::services::ServeDir;
use crate::db::{Database, DatabaseInterface}; use crate::db::Database;
use crate::listener::{Listener, ListenerKind}; use crate::listener::{Listener, ListenerKind};
use crate::stream::{AbstractSocketAddr, AbstractStreamKind}; use crate::stream::{AbstractSocketAddr, AbstractStreamKind};
@ -62,14 +62,14 @@ impl AxumListener for Listener {
} }
#[derive(Clone)] #[derive(Clone)]
pub struct HttpState<D: DatabaseInterface> { pub struct HttpState {
pub db: Database<D>, pub db: Database,
pub sessions: HttpSessionManager, pub sessions: HttpSessionManager,
pub templates: Environment<'static>, pub templates: Environment<'static>,
} }
impl<D: DatabaseInterface> HttpState<D> { impl HttpState {
pub fn new(db: Database<D>) -> Self { pub fn new(db: Database) -> Self {
let mut templates = Environment::new(); let mut templates = Environment::new();
#[cfg(feature = "embed")] #[cfg(feature = "embed")]
minijinja_embed::load_templates!(&mut templates); minijinja_embed::load_templates!(&mut templates);
@ -83,7 +83,7 @@ impl<D: DatabaseInterface> HttpState<D> {
} }
} }
pub async fn http_listen<D: DatabaseInterface>(listener: Listener, db: Database<D>) { pub async fn http_listen(listener: Listener, db: Database) {
#[cfg(all(feature = "embed", feature = "noembed"))] #[cfg(all(feature = "embed", feature = "noembed"))]
compile_error!("You cannot have `embed` and `noembed` features enabled at the same time."); compile_error!("You cannot have `embed` and `noembed` features enabled at the same time.");
#[cfg(not(any(feature = "embed", feature = "noembed")))] #[cfg(not(any(feature = "embed", feature = "noembed")))]

View file

@ -6,7 +6,7 @@ use uuid::Uuid;
use std::sync::{Arc, RwLock}; use std::sync::{Arc, RwLock};
use crate::db::{DatabaseInterface, User}; use crate::db::User;
use crate::http::HttpState; use crate::http::HttpState;
pub const COOKIE_NAME: &str = "lldap_session"; pub const COOKIE_NAME: &str = "lldap_session";
@ -83,12 +83,12 @@ impl PartialEq<HttpSession> for HttpSession {
} }
} }
impl<D: DatabaseInterface> FromRequestParts<HttpState<D>> for HttpSession { impl FromRequestParts<HttpState> for HttpSession {
type Rejection = Redirect; type Rejection = Redirect;
async fn from_request_parts( async fn from_request_parts(
parts: &mut Parts, parts: &mut Parts,
state: &HttpState<D>, state: &HttpState,
) -> Result<Self, Self::Rejection> { ) -> Result<Self, Self::Rejection> {
let cookies = CookieJar::from_request_parts(parts, state).await.unwrap(); let cookies = CookieJar::from_request_parts(parts, state).await.unwrap();
state state
@ -108,12 +108,12 @@ impl std::ops::Deref for OptionalHttpSession {
} }
} }
impl<D: DatabaseInterface> OptionalFromRequestParts<HttpState<D>> for OptionalHttpSession { impl OptionalFromRequestParts<HttpState> for OptionalHttpSession {
type Rejection = Redirect; type Rejection = Redirect;
async fn from_request_parts( async fn from_request_parts(
parts: &mut Parts, parts: &mut Parts,
state: &HttpState<D>, state: &HttpState,
) -> Result<Option<Self>, Self::Rejection> { ) -> Result<Option<Self>, Self::Rejection> {
let maybe_session = HttpSession::from_request_parts(parts, state) let maybe_session = HttpSession::from_request_parts(parts, state)
.await .await

View file

@ -12,8 +12,8 @@ pub struct UserCreationForm {
pub password: String, pub password: String,
} }
pub async fn create_user<D: DatabaseInterface>( pub async fn create_user(
State(mut state): State<HttpState<D>>, State(mut state): State<HttpState>,
session: HttpSession, session: HttpSession,
Form(form): Form<UserCreationForm>, Form(form): Form<UserCreationForm>,
) -> Response { ) -> Response {

View file

@ -1,13 +1,13 @@
use ldap3_proto::LdapMsg; use ldap3_proto::LdapMsg;
use ldap3_proto::proto::LdapOp; use ldap3_proto::proto::LdapOp;
use crate::db::{Database, DatabaseInterface}; use crate::db::Database;
use crate::ldap::{ use crate::ldap::{
LdapClientState, LdapStream, LdapStreamError, op_bind, op_ext, search_by_mail_filter, LdapClientState, LdapStream, LdapStreamError, op_bind, op_ext, search_by_mail_filter,
}; };
#[tracing::instrument(name = "ldap", skip(stream, db), fields(session = %stream.session))] #[tracing::instrument(name = "ldap", skip(stream, db), fields(session = %stream.session))]
pub async fn ldap_handler<D: DatabaseInterface>(mut stream: LdapStream, mut db: Database<D>) { pub async fn ldap_handler(mut stream: LdapStream, mut db: Database) {
tracing::info! { tracing::info! {
remote_addr = ?stream.remote_addr, remote_addr = ?stream.remote_addr,
"New client connection" "New client connection"
@ -45,11 +45,11 @@ pub async fn ldap_handler<D: DatabaseInterface>(mut stream: LdapStream, mut db:
/// Return true to keep the connection going, false to close it. /// Return true to keep the connection going, false to close it.
#[tracing::instrument(name = "ldap-handler", skip(client_state, db, stream))] #[tracing::instrument(name = "ldap-handler", skip(client_state, db, stream))]
pub async fn ldap_handler_inner<D: DatabaseInterface>( pub async fn ldap_handler_inner(
stream: &mut LdapStream, stream: &mut LdapStream,
msg: LdapMsg, msg: LdapMsg,
client_state: &mut LdapClientState, client_state: &mut LdapClientState,
db: &mut Database<D>, db: &mut Database,
) -> Result<bool, LdapStreamError> { ) -> Result<bool, LdapStreamError> {
tracing::debug!(msg = ?msg, "Received LDAP message"); tracing::debug!(msg = ?msg, "Received LDAP message");
match msg { match msg {

View file

@ -2,7 +2,7 @@ use ldap3_proto::proto::{LdapBindCred, LdapBindRequest, LdapBindResponse, LdapOp
use ldap3_proto::{LdapMsg, LdapResultCode}; use ldap3_proto::{LdapMsg, LdapResultCode};
use crate::db::error::BoxedError; use crate::db::error::BoxedError;
use crate::db::{Database, DatabaseInterface, UserRef}; use crate::db::{Database, UserRef};
use crate::ldap::{Dn, LdapReturnError, LdapStream, LdapStreamError, MalformedDn}; use crate::ldap::{Dn, LdapReturnError, LdapStream, LdapStreamError, MalformedDn};
#[derive(Debug)] #[derive(Debug)]
@ -165,9 +165,9 @@ pub async fn bind_success(stream: &mut LdapStream, msgid: i32) -> Result<(), Lda
/// ///
/// On success, returns `Ok(Some(bound_dn))`. `Ok(None)` means credentials failed, /// On success, returns `Ok(Some(bound_dn))`. `Ok(None)` means credentials failed,
/// either because the account does not exist, or the password is wrong. /// either because the account does not exist, or the password is wrong.
pub async fn op_bind<D: DatabaseInterface>( pub async fn op_bind(
stream: &mut LdapStream, stream: &mut LdapStream,
db: &Database<D>, db: &Database,
req: LdapBindRequest, req: LdapBindRequest,
msgid: i32, msgid: i32,
) -> Result<Option<BindDn>, LdapStreamError> { ) -> Result<Option<BindDn>, LdapStreamError> {

View file

@ -174,9 +174,9 @@ fn search_entry_from_user(user: &User, req_attrs: &[String]) -> LdapSearchResult
} }
} }
pub async fn search_by_mail_filter<D: DatabaseInterface>( pub async fn search_by_mail_filter(
stream: &mut LdapStream, stream: &mut LdapStream,
db: &Database<D>, db: &Database,
sr: LdapSearchRequest, sr: LdapSearchRequest,
msgid: i32, msgid: i32,
) -> Result<(), LdapStreamError> { ) -> Result<(), LdapStreamError> {

View file

@ -12,12 +12,12 @@ mod stream;
#[cfg(feature = "http")] #[cfg(feature = "http")]
use crate::http::http_listen; use crate::http::http_listen;
use cli::CliArgs; use cli::CliArgs;
use db::{Database, DatabaseInterface, MemoryDatabase, Role, User}; use db::{Database, DatabaseInterface, FilesystemDatabase, MemoryDatabase, Role, User};
use error::GlobalError; use error::GlobalError;
use ldap::ldap_handler; use ldap::ldap_handler;
use listener::ListenerPath; use listener::ListenerPath;
async fn create_dummy_users<D: DatabaseInterface>(db: &mut Database<D>) { async fn create_dummy_users(db: &mut Database) {
db.create_user(User { db.create_user(User {
username: "admin".to_string(), username: "admin".to_string(),
domain: None, domain: None,
@ -69,7 +69,11 @@ async fn main() -> Result<(), GlobalError> {
let ldap_listener = ListenerPath::new(&cli.listen_ldap)?.listener().await?; let ldap_listener = ListenerPath::new(&cli.listen_ldap)?.listener().await?;
let mut db = MemoryDatabase::new(); let mut db = if let Some(db_path) = &cli.db {
FilesystemDatabase::from_path(db_path).await.unwrap()
} else {
MemoryDatabase::new()
};
create_dummy_users(&mut db).await; create_dummy_users(&mut db).await;
#[cfg(feature = "http")] #[cfg(feature = "http")]