rshs 0.9.2

A lightweight HTTP + WebDAV file server
Documentation 
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174

mod common;

use axum::body::Body;
use axum::http::Method;
use base64::{Engine as _, engine::general_purpose};
use common::{make_test_router, temp_dir_with_files};
use tower::ServiceExt;

fn basic_auth_header(username: &str, password: &str) -> String {
    let creds = format!("{username}:{password}");
    format!(
        "Basic {}",
        general_purpose::STANDARD.encode(creds.as_bytes())
    )
}

#[tokio::test]
async fn test_no_auth_passes_through() {
    let dir = temp_dir_with_files();
    let app = make_test_router(dir.path(), rshs::AuthState::new());

    let req = axum::http::Request::builder()
        .method(Method::GET)
        .uri("/hello.txt")
        .body(Body::empty())
        .unwrap();
    let resp = app.oneshot(req).await.unwrap();
    assert!(resp.status().is_success(), "no auth should pass through");
}

#[tokio::test]
async fn test_auth_returns_401_without_credentials() {
    let dir = temp_dir_with_files();
    let mut auth = rshs::AuthState::new();
    auth.add_user("admin", "secret");
    let app = make_test_router(dir.path(), auth);

    let req = axum::http::Request::builder()
        .method(Method::GET)
        .uri("/")
        .body(Body::empty())
        .unwrap();
    let resp = app.oneshot(req).await.unwrap();
    assert_eq!(resp.status().as_u16(), 401);
    assert!(
        resp.headers()
            .get("www-authenticate")
            .unwrap()
            .to_str()
            .unwrap()
            .contains(r#"Basic realm="rshs""#)
    );
}

#[tokio::test]
async fn test_auth_success_with_valid_credentials() {
    let dir = temp_dir_with_files();
    let mut auth = rshs::AuthState::new();
    auth.add_user("admin", "secret");
    let app = make_test_router(dir.path(), auth);

    let req = axum::http::Request::builder()
        .method(Method::GET)
        .uri("/hello.txt")
        .header("authorization", basic_auth_header("admin", "secret"))
        .body(Body::empty())
        .unwrap();
    let resp = app.oneshot(req).await.unwrap();
    assert_eq!(resp.status().as_u16(), 200);

    let body = axum::body::to_bytes(resp.into_body(), usize::MAX)
        .await
        .unwrap();
    assert_eq!(body.as_ref(), b"Hello, World!");
}

#[tokio::test]
async fn test_auth_wrong_password_returns_401() {
    let dir = temp_dir_with_files();
    let mut auth = rshs::AuthState::new();
    auth.add_user("admin", "secret");
    let app = make_test_router(dir.path(), auth);

    let req = axum::http::Request::builder()
        .method(Method::GET)
        .uri("/")
        .header("authorization", basic_auth_header("admin", "wrong"))
        .body(Body::empty())
        .unwrap();
    let resp = app.oneshot(req).await.unwrap();
    assert_eq!(resp.status().as_u16(), 401);
}

#[tokio::test]
async fn test_auth_unknown_user_returns_401() {
    let dir = temp_dir_with_files();
    let mut auth = rshs::AuthState::new();
    auth.add_user("admin", "secret");
    let app = make_test_router(dir.path(), auth);

    let req = axum::http::Request::builder()
        .method(Method::GET)
        .uri("/")
        .header("authorization", basic_auth_header("nobody", "pass"))
        .body(Body::empty())
        .unwrap();
    let resp = app.oneshot(req).await.unwrap();
    assert_eq!(resp.status().as_u16(), 401);
}

#[tokio::test]
async fn test_health_check_bypasses_auth() {
    let dir = temp_dir_with_files();
    let mut auth = rshs::AuthState::new();
    auth.add_user("admin", "secret");
    let app = make_test_router(dir.path(), auth);

    let req = axum::http::Request::builder()
        .method(Method::GET)
        .uri("/")
        .header("x-health-check", "true")
        .body(Body::empty())
        .unwrap();
    let resp = app.oneshot(req).await.unwrap();
    assert_eq!(resp.status().as_u16(), 200);

    let body = axum::body::to_bytes(resp.into_body(), usize::MAX)
        .await
        .unwrap();
    assert_eq!(body.as_ref(), b"OK");
}

#[tokio::test]
async fn test_health_check_with_wrong_value_passes_through() {
    let dir = temp_dir_with_files();
    let mut auth = rshs::AuthState::new();
    auth.add_user("admin", "secret");
    let app = make_test_router(dir.path(), auth);

    let req = axum::http::Request::builder()
        .method(Method::GET)
        .uri("/")
        .header("x-health-check", "false")
        .body(Body::empty())
        .unwrap();
    let resp = app.oneshot(req).await.unwrap();
    assert_eq!(resp.status().as_u16(), 401, "should be blocked by auth");
}

#[tokio::test]
async fn test_auth_sha512_cached_request() {
    use sha_crypt::PasswordHasher;

    let dir = temp_dir_with_files();
    let hash = sha_crypt::ShaCrypt::default()
        .hash_password("mypassword".as_bytes())
        .unwrap()
        .to_string();
    let mut auth = rshs::AuthState::new();
    auth.users
        .insert("admin".into(), rshs::auth::Credential::Sha512Crypt(hash));
    let app = make_test_router(dir.path(), auth);

    for _ in 0..3 {
        let req = axum::http::Request::builder()
            .method(Method::GET)
            .uri("/hello.txt")
            .header("authorization", basic_auth_header("admin", "mypassword"))
            .body(Body::empty())
            .unwrap();
        let resp = app.clone().oneshot(req).await.unwrap();
        assert_eq!(resp.status().as_u16(), 200);
    }
}