1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
use crate::{interface::config::WebDavConfig, servers::webdav::adapter::XHFSAdapter};
use dav_server::{DavHandler, body::Body, memls::MemLs};
use eyre::Context;
use headers::{Authorization, HeaderMapExt, authorization::Basic};
use hyper::{Response, StatusCode, server::conn::http1, service::service_fn};
use hyper_util::rt::TokioIo;
use std::{convert::Infallible, net::SocketAddr, sync::Arc};
use tokio::net::TcpListener;
use xhfs_core::xhfs::XHFS;
mod adapter;
pub async fn webdav_main(
addr: String,
port: u16,
xhfs: Arc<XHFS>,
read_only: bool,
auth: Option<WebDavConfig>,
) -> eyre::Result<()> {
println!("-------");
println!("{}", xhfs.format_headers_report().await?);
println!("-------");
let addr = format!("{addr}:{port}").parse::<SocketAddr>()?;
let xhfs_adapter = Box::new(XHFSAdapter {
xhfs,
read_only,
chunk_size: 64 * 1024, // standard 64KB disk read windows
});
let dav_server = DavHandler::builder()
.filesystem(xhfs_adapter)
.locksystem(MemLs::new())
.build_handler();
let listener = TcpListener::bind(addr)
.await
.with_context(|| format!("Binding {addr}:{port}"))?;
println!(
"Listening for WebDAV requests at {addr}{}",
match &auth {
Some(cfg) => format!(" (user: {})", cfg.username),
None => " (no auth)".to_string(),
}
);
loop {
let (stream, _) = listener.accept().await.unwrap();
let dav_server = dav_server.clone();
let auth = auth.clone();
let io = TokioIo::new(stream);
tokio::task::spawn(async move {
if let Err(err) = http1::Builder::new()
.serve_connection(
io,
service_fn({
move |req| {
let dav_server = dav_server.clone();
let auth = auth.clone();
async move {
match &auth {
Some(cfg) => {
if let Some(a) =
req.headers().typed_get::<Authorization<Basic>>()
&& a.username() == cfg.username
&& a.password() == cfg.password
{
let creds = adapter::Creds {
username: a.username().to_string(),
};
return Ok::<_, Infallible>(
dav_server
.handle_guarded(
req,
creds.username.clone(),
creds,
)
.await,
);
}
Ok::<_, Infallible>(
Response::builder()
.status(StatusCode::UNAUTHORIZED)
.header("WWW-Authenticate", r#"Basic realm="xhfs""#)
.body(Body::empty())
.unwrap(),
)
}
None => Ok::<_, Infallible>(
dav_server
.handle_guarded(
req,
"anonymous".to_string(),
adapter::Creds {
username: "anonymous".to_string(),
},
)
.await,
),
}
}
}
}),
)
.await
{
eprintln!("Error executing HTTP session processing context: {err:#}");
}
});
}
}