mirror of
https://github.com/sigoden/dufs.git
synced 2026-06-07 23:16:54 +03:00
Compare commits
56 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| b69946df23 | |||
| 82a8865b9f | |||
| 8e90ffa9c8 | |||
| 4f2dee3916 | |||
| b87f87646a | |||
| 30b2979d0a | |||
| 53ea692dd1 | |||
| 1af66d6744 | |||
| 19dc2c205a | |||
| 43c778182b | |||
| 0ccc2cf1e7 | |||
| a88a4ee630 | |||
| a118c1348e | |||
| db7a0530a2 | |||
| bc27c8c479 | |||
| 2b2c7bd5f7 | |||
| ca18df1a36 | |||
| 7cfb97dfdf | |||
| 23619033ae | |||
| db75ba4357 | |||
| 4016715187 | |||
| f8a7873582 | |||
| 7f8269881d | |||
| b2f244a4cf | |||
| 6cc8a18a3d | |||
| a387d727b2 | |||
| 19d65a5aa4 | |||
| d37762d2b9 | |||
| 9c9fca75d3 | |||
| 089d30c5a5 | |||
| 459a4d4f4a | |||
| f8b69f4df8 | |||
| 53f064c73b | |||
| 8a92a0cf1a | |||
| 59685da06e | |||
| 09200860b4 | |||
| 4fbdec2878 | |||
| d0453b7591 | |||
| eda9769b2a | |||
| d255f1376a | |||
| 669c4f8811 | |||
| e576ddcbea | |||
| af95ea1cd7 | |||
| cbc620481d | |||
| f1c9776962 | |||
| ac15ae4e8e | |||
| ab4ef06cb8 | |||
| bc6c573acb | |||
| f27f9e997f | |||
| 835438fc2a | |||
| d445b78f96 | |||
| 881a67e1a4 | |||
| bb5a5564b4 | |||
| 2cf6d39032 | |||
| c500ce7acc | |||
| f87c52fda2 |
@@ -29,7 +29,7 @@ jobs:
|
|||||||
RUSTFLAGS: --deny warnings
|
RUSTFLAGS: --deny warnings
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v6
|
||||||
|
|
||||||
- name: Install Rust Toolchain Components
|
- name: Install Rust Toolchain Components
|
||||||
uses: dtolnay/rust-toolchain@stable
|
uses: dtolnay/rust-toolchain@stable
|
||||||
|
|||||||
@@ -38,14 +38,6 @@ jobs:
|
|||||||
os: ubuntu-latest
|
os: ubuntu-latest
|
||||||
use-cross: true
|
use-cross: true
|
||||||
cargo-flags: ""
|
cargo-flags: ""
|
||||||
- target: i686-unknown-linux-musl
|
|
||||||
os: ubuntu-latest
|
|
||||||
use-cross: true
|
|
||||||
cargo-flags: ""
|
|
||||||
- target: i686-pc-windows-msvc
|
|
||||||
os: windows-latest
|
|
||||||
use-cross: true
|
|
||||||
cargo-flags: ""
|
|
||||||
- target: armv7-unknown-linux-musleabihf
|
- target: armv7-unknown-linux-musleabihf
|
||||||
os: ubuntu-latest
|
os: ubuntu-latest
|
||||||
use-cross: true
|
use-cross: true
|
||||||
@@ -60,7 +52,7 @@ jobs:
|
|||||||
BUILD_CMD: cargo
|
BUILD_CMD: cargo
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v6
|
||||||
|
|
||||||
- name: Check Tag
|
- name: Check Tag
|
||||||
id: check-tag
|
id: check-tag
|
||||||
@@ -170,7 +162,6 @@ jobs:
|
|||||||
platforms: |
|
platforms: |
|
||||||
linux/amd64
|
linux/amd64
|
||||||
linux/arm64
|
linux/arm64
|
||||||
linux/386
|
|
||||||
linux/arm/v7
|
linux/arm/v7
|
||||||
push: ${{ needs.release.outputs.rc == 'false' }}
|
push: ${{ needs.release.outputs.rc == 'false' }}
|
||||||
tags: ${{ github.repository }}:latest, ${{ github.repository }}:${{ github.ref_name }}
|
tags: ${{ github.repository }}:latest, ${{ github.repository }}:${{ github.ref_name }}
|
||||||
@@ -181,7 +172,7 @@ jobs:
|
|||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
needs: release
|
needs: release
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v6
|
||||||
|
|
||||||
- uses: dtolnay/rust-toolchain@stable
|
- uses: dtolnay/rust-toolchain@stable
|
||||||
|
|
||||||
|
|||||||
@@ -2,6 +2,92 @@
|
|||||||
|
|
||||||
All notable changes to this project will be documented in this file.
|
All notable changes to this project will be documented in this file.
|
||||||
|
|
||||||
|
## [0.46.0] - 2026-05-07
|
||||||
|
|
||||||
|
### Features
|
||||||
|
|
||||||
|
- Add option --allow-hash to allow/disallow file hashing ([#657](https://github.com/sigoden/dufs/issues/657))
|
||||||
|
- Support `?json` on file path ([#686](https://github.com/sigoden/dufs/issues/686))
|
||||||
|
- Support customizable 404 page ([#688](https://github.com/sigoden/dufs/issues/688))
|
||||||
|
- Enhance log format ([#692](https://github.com/sigoden/dufs/issues/692))
|
||||||
|
- Webui confirm on exit while uploading ([#693](https://github.com/sigoden/dufs/issues/693))
|
||||||
|
- Skip directory walking in HEAD requests ([#701](https://github.com/sigoden/dufs/issues/701))
|
||||||
|
|
||||||
|
### Bug Fixes
|
||||||
|
|
||||||
|
- Some search results missing due to broken symlinks ([#665](https://github.com/sigoden/dufs/issues/665))
|
||||||
|
- Escape filename in ?simple output ([#669](https://github.com/sigoden/dufs/issues/669))
|
||||||
|
- Ensure symlink inside serve root ([#670](https://github.com/sigoden/dufs/issues/670))
|
||||||
|
- Tweak auth logic ([#689](https://github.com/sigoden/dufs/issues/689))
|
||||||
|
- Http range underflow ([#690](https://github.com/sigoden/dufs/issues/690))
|
||||||
|
- Escape control chars in logged URI and headers ([#691](https://github.com/sigoden/dufs/issues/691))
|
||||||
|
- Webui safari bug uploadspeed ([#695](https://github.com/sigoden/dufs/issues/695))
|
||||||
|
|
||||||
|
### Refactor
|
||||||
|
|
||||||
|
- Update deps ([#655](https://github.com/sigoden/dufs/issues/655))
|
||||||
|
- Improve UI button titles ([#656](https://github.com/sigoden/dufs/issues/656))
|
||||||
|
- Webui file size format ([#698](https://github.com/sigoden/dufs/issues/698))
|
||||||
|
|
||||||
|
## [0.45.0] - 2025-09-03
|
||||||
|
|
||||||
|
### Bug Fixes
|
||||||
|
|
||||||
|
- Perms on `dufs -A -a @/:ro` ([#619](https://github.com/sigoden/dufs/issues/619))
|
||||||
|
- Login btn does not work for readonly anonymous ([#620](https://github.com/sigoden/dufs/issues/620))
|
||||||
|
- Verify token length ([#627](https://github.com/sigoden/dufs/issues/627))
|
||||||
|
|
||||||
|
### Features
|
||||||
|
|
||||||
|
- Make dir urls inherit `?noscript` params ([#614](https://github.com/sigoden/dufs/issues/614))
|
||||||
|
- Log decoded uri ([#615](https://github.com/sigoden/dufs/issues/615))
|
||||||
|
|
||||||
|
## [0.44.0] - 2025-08-02
|
||||||
|
|
||||||
|
### Bug Fixes
|
||||||
|
|
||||||
|
- No authentication check if no auth users ([#497](https://github.com/sigoden/dufs/issues/497))
|
||||||
|
- Webui can't handle hash property of URL well ([#515](https://github.com/sigoden/dufs/issues/515))
|
||||||
|
- Incorrect dir size due to hidden files ([#529](https://github.com/sigoden/dufs/issues/529))
|
||||||
|
- Webui formatDirSize ([#568](https://github.com/sigoden/dufs/issues/568))
|
||||||
|
- Follow symlinks when searching/archiving ([#572](https://github.com/sigoden/dufs/issues/572))
|
||||||
|
- Incorrect separator for zip archives under windows ([#577](https://github.com/sigoden/dufs/issues/577))
|
||||||
|
- Unexpected public auth asking for login info ([#583](https://github.com/sigoden/dufs/issues/583))
|
||||||
|
|
||||||
|
### Features
|
||||||
|
|
||||||
|
- Higher perm auth path shadows lower one ([#521](https://github.com/sigoden/dufs/issues/521))
|
||||||
|
- Add cache-control:no-cache while sending file and index ([#528](https://github.com/sigoden/dufs/issues/528))
|
||||||
|
- Support multipart ranges ([#535](https://github.com/sigoden/dufs/issues/535))
|
||||||
|
- Limit sub directory item counting ([#556](https://github.com/sigoden/dufs/issues/556))
|
||||||
|
- Tolerate the absence of mtime ([#559](https://github.com/sigoden/dufs/issues/559))
|
||||||
|
- Support noscript fallback ([#602](https://github.com/sigoden/dufs/issues/602))
|
||||||
|
- Support downloading via token auth ([#603](https://github.com/sigoden/dufs/issues/603))
|
||||||
|
|
||||||
|
### Refactor
|
||||||
|
|
||||||
|
- Change description for `--allow-archive` ([#511](https://github.com/sigoden/dufs/issues/511))
|
||||||
|
- Removes clippy warnings ([#601](https://github.com/sigoden/dufs/issues/601))
|
||||||
|
- Update deps ([#604](https://github.com/sigoden/dufs/issues/604))
|
||||||
|
- Fix typos ([#605](https://github.com/sigoden/dufs/issues/605))
|
||||||
|
|
||||||
|
## [0.43.0] - 2024-11-04
|
||||||
|
|
||||||
|
### Bug Fixes
|
||||||
|
|
||||||
|
- Auth failed if password contains `:` ([#449](https://github.com/sigoden/dufs/issues/449))
|
||||||
|
- Resolve speed bottleneck in 10G network ([#451](https://github.com/sigoden/dufs/issues/451))
|
||||||
|
|
||||||
|
### Features
|
||||||
|
|
||||||
|
- Webui displays subdirectory items ([#457](https://github.com/sigoden/dufs/issues/457))
|
||||||
|
- Support binding abstract unix socket ([#468](https://github.com/sigoden/dufs/issues/468))
|
||||||
|
- Provide healthcheck API ([#474](https://github.com/sigoden/dufs/issues/474))
|
||||||
|
|
||||||
|
### Refactor
|
||||||
|
|
||||||
|
- Do not show size for Dir ([#447](https://github.com/sigoden/dufs/issues/447))
|
||||||
|
|
||||||
## [0.42.0] - 2024-09-01
|
## [0.42.0] - 2024-09-01
|
||||||
|
|
||||||
### Bug Fixes
|
### Bug Fixes
|
||||||
|
|||||||
Generated
+1591
-673
File diff suppressed because it is too large
Load Diff
+17
-16
@@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "dufs"
|
name = "dufs"
|
||||||
version = "0.42.0"
|
version = "0.46.0"
|
||||||
edition = "2021"
|
edition = "2021"
|
||||||
authors = ["sigoden <sigoden@gmail.com>"]
|
authors = ["sigoden <sigoden@gmail.com>"]
|
||||||
description = "Dufs is a distinctive utility file server"
|
description = "Dufs is a distinctive utility file server"
|
||||||
@@ -14,37 +14,36 @@ keywords = ["static", "file", "server", "webdav", "cli"]
|
|||||||
clap = { version = "4.5", features = ["wrap_help", "env"] }
|
clap = { version = "4.5", features = ["wrap_help", "env"] }
|
||||||
clap_complete = "4.5"
|
clap_complete = "4.5"
|
||||||
chrono = { version = "0.4", default-features = false, features = ["clock"] }
|
chrono = { version = "0.4", default-features = false, features = ["clock"] }
|
||||||
tokio = { version = "1", features = ["rt-multi-thread", "macros", "fs", "io-util", "signal"]}
|
tokio = { version = "1", features = ["rt-multi-thread", "macros", "fs", "io-util", "signal", "net"]}
|
||||||
tokio-util = { version = "0.7", features = ["io-util", "compat"] }
|
tokio-util = { version = "0.7", features = ["io-util", "compat"] }
|
||||||
hyper = { version = "1", features = ["http1", "server"] }
|
hyper = { version = "1", features = ["http1", "server"] }
|
||||||
percent-encoding = "2.3"
|
percent-encoding = "2.3"
|
||||||
serde = { version = "1", features = ["derive"] }
|
serde = { version = "1", features = ["derive"] }
|
||||||
serde_json = "1"
|
serde_json = "1"
|
||||||
futures-util = { version = "0.3", default-features = false, features = ["alloc"] }
|
futures-util = { version = "0.3", default-features = false, features = ["alloc"] }
|
||||||
async_zip = { version = "0.0.17", default-features = false, features = ["deflate", "bzip2", "xz", "chrono", "tokio"] }
|
async_zip = { version = "0.0.18", default-features = false, features = ["deflate", "bzip2", "xz", "chrono", "tokio"] }
|
||||||
headers = "0.4"
|
headers = "0.4"
|
||||||
mime_guess = "2.0"
|
mime_guess = "2.0"
|
||||||
if-addrs = "0.13"
|
if-addrs = "0.15"
|
||||||
rustls-pemfile = { version = "2.0", optional = true }
|
tokio-rustls = { version = "0.26", optional = true }
|
||||||
tokio-rustls = { version = "0.26", optional = true, default-features = false, features = ["ring", "tls12"]}
|
md5 = "0.8"
|
||||||
md5 = "0.7"
|
|
||||||
lazy_static = "1.4"
|
lazy_static = "1.4"
|
||||||
uuid = { version = "1.7", features = ["v4", "fast-rng"] }
|
uuid = { version = "1.7", features = ["v4", "fast-rng"] }
|
||||||
urlencoding = "2.1"
|
urlencoding = "2.1"
|
||||||
xml-rs = "0.8"
|
xml-rs = "1.0.0"
|
||||||
log = { version = "0.4", features = ["std"] }
|
log = { version = "0.4", features = ["std"] }
|
||||||
socket2 = "0.5"
|
socket2 = "0.6"
|
||||||
async-stream = "0.3"
|
async-stream = "0.3"
|
||||||
walkdir = "2.3"
|
walkdir = "2.3"
|
||||||
form_urlencoded = "1.2"
|
form_urlencoded = "1.2"
|
||||||
alphanumeric-sort = "1.4"
|
alphanumeric-sort = "1.4"
|
||||||
content_inspector = "0.2"
|
content_inspector = "0.2"
|
||||||
anyhow = "1.0"
|
anyhow = "1.0"
|
||||||
chardetng = "0.1"
|
chardetng = "1.0"
|
||||||
glob = "0.3"
|
glob = "0.3"
|
||||||
indexmap = "2.2"
|
indexmap = "2.2"
|
||||||
serde_yaml = "0.9"
|
serde_yaml = "0.9"
|
||||||
sha-crypt = "0.5"
|
sha-crypt = "0.6"
|
||||||
base64 = "0.22"
|
base64 = "0.22"
|
||||||
smart-default = "0.7"
|
smart-default = "0.7"
|
||||||
rustls-pki-types = "1.2"
|
rustls-pki-types = "1.2"
|
||||||
@@ -52,18 +51,20 @@ hyper-util = { version = "0.1", features = ["server-auto", "tokio"] }
|
|||||||
http-body-util = "0.1"
|
http-body-util = "0.1"
|
||||||
bytes = "1.5"
|
bytes = "1.5"
|
||||||
pin-project-lite = "0.2"
|
pin-project-lite = "0.2"
|
||||||
sha2 = "0.10.8"
|
sha2 = "0.11.0"
|
||||||
|
ed25519-dalek = "2.2.0"
|
||||||
|
hex = "0.4.3"
|
||||||
|
|
||||||
[features]
|
[features]
|
||||||
default = ["tls"]
|
default = ["tls"]
|
||||||
tls = ["rustls-pemfile", "tokio-rustls"]
|
tls = ["tokio-rustls"]
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
assert_cmd = "2"
|
assert_cmd = "2"
|
||||||
reqwest = { version = "0.12", features = ["blocking", "multipart", "rustls-tls"], default-features = false }
|
reqwest = { version = "0.13", features = ["blocking", "multipart", "rustls"], default-features = false }
|
||||||
assert_fs = "1"
|
assert_fs = "1"
|
||||||
port_check = "0.2"
|
port_check = "0.3"
|
||||||
rstest = "0.22"
|
rstest = "0.26.1"
|
||||||
regex = "1"
|
regex = "1"
|
||||||
url = "2"
|
url = "2"
|
||||||
predicates = "3"
|
predicates = "3"
|
||||||
|
|||||||
@@ -4,8 +4,6 @@ RUN if [ "$TARGETPLATFORM" = "linux/amd64" ]; then \
|
|||||||
TARGET="x86_64-unknown-linux-musl"; \
|
TARGET="x86_64-unknown-linux-musl"; \
|
||||||
elif [ "$TARGETPLATFORM" = "linux/arm64" ]; then \
|
elif [ "$TARGETPLATFORM" = "linux/arm64" ]; then \
|
||||||
TARGET="aarch64-unknown-linux-musl"; \
|
TARGET="aarch64-unknown-linux-musl"; \
|
||||||
elif [ "$TARGETPLATFORM" = "linux/386" ]; then \
|
|
||||||
TARGET="i686-unknown-linux-musl"; \
|
|
||||||
elif [ "$TARGETPLATFORM" = "linux/arm/v7" ]; then \
|
elif [ "$TARGETPLATFORM" = "linux/arm/v7" ]; then \
|
||||||
TARGET="armv7-unknown-linux-musleabihf"; \
|
TARGET="armv7-unknown-linux-musleabihf"; \
|
||||||
fi && \
|
fi && \
|
||||||
|
|||||||
@@ -66,7 +66,8 @@ Options:
|
|||||||
--allow-delete Allow delete files/folders
|
--allow-delete Allow delete files/folders
|
||||||
--allow-search Allow search files/folders
|
--allow-search Allow search files/folders
|
||||||
--allow-symlink Allow symlink to files/folders outside root directory
|
--allow-symlink Allow symlink to files/folders outside root directory
|
||||||
--allow-archive Allow zip archive generation
|
--allow-archive Allow download folders as archive file
|
||||||
|
--allow-hash Allow ?hash query to get file sha256 hash
|
||||||
--enable-cors Enable CORS, sets `Access-Control-Allow-Origin: *`
|
--enable-cors Enable CORS, sets `Access-Control-Allow-Origin: *`
|
||||||
--render-index Serve index.html when requesting a directory, returns 404 if not found index.html
|
--render-index Serve index.html when requesting a directory, returns 404 if not found index.html
|
||||||
--render-try-index Serve index.html when requesting a directory, returns directory listing if not found index.html
|
--render-try-index Serve index.html when requesting a directory, returns directory listing if not found index.html
|
||||||
@@ -216,8 +217,14 @@ dd skip=$upload_offset if=file status=none ibs=1 | \
|
|||||||
curl -X PATCH -H "X-Update-Range: append" --data-binary @- http://127.0.0.1:5000/file
|
curl -X PATCH -H "X-Update-Range: append" --data-binary @- http://127.0.0.1:5000/file
|
||||||
```
|
```
|
||||||
|
|
||||||
|
Health checks
|
||||||
|
|
||||||
|
```sh
|
||||||
|
curl http://127.0.0.1:5000/__dufs__/health
|
||||||
|
```
|
||||||
|
|
||||||
<details>
|
<details>
|
||||||
<summary><h2>Advanced topics</h2></summary>
|
<summary><h2>Advanced Topics</h2></summary>
|
||||||
|
|
||||||
### Access Control
|
### Access Control
|
||||||
|
|
||||||
@@ -238,24 +245,25 @@ dufs -a user:pass@/:rw,/dir1 -a @/
|
|||||||
- `-a user:pass@/:rw,/dir1`: `user` has read-write permissions for `/*`, has read-only permissions for `/dir1/*`.
|
- `-a user:pass@/:rw,/dir1`: `user` has read-write permissions for `/*`, has read-only permissions for `/dir1/*`.
|
||||||
- `-a @/`: All paths is publicly accessible, everyone can view/download it.
|
- `-a @/`: All paths is publicly accessible, everyone can view/download it.
|
||||||
|
|
||||||
> There are no restrictions on using ':' and '@' characters in a password. For example, `user:pa:ss@1@/:rw` is valid, the password is `pa:ss@1`.
|
**Auth permissions are restricted by dufs global permissions.** If dufs does not enable upload permissions via `--allow-upload`, then the account will not have upload permissions even if it is granted `read-write`(`:rw`) permissions.
|
||||||
|
|
||||||
#### Hashed Password
|
#### Hashed Password
|
||||||
|
|
||||||
DUFS supports the use of sha-512 hashed password.
|
DUFS supports the use of sha-512 hashed password.
|
||||||
|
|
||||||
Create hashed password
|
Create hashed password:
|
||||||
|
|
||||||
```
|
```sh
|
||||||
$ mkpasswd -m sha-512 -s
|
$ openssl passwd -6 123456 # or `mkpasswd -m sha-512 123456`
|
||||||
Password: 123456
|
|
||||||
$6$tWMB51u6Kb2ui3wd$5gVHP92V9kZcMwQeKTjyTRgySsYJu471Jb1I6iHQ8iZ6s07GgCIO69KcPBRuwPE5tDq05xMAzye0NxVKuJdYs/
|
$6$tWMB51u6Kb2ui3wd$5gVHP92V9kZcMwQeKTjyTRgySsYJu471Jb1I6iHQ8iZ6s07GgCIO69KcPBRuwPE5tDq05xMAzye0NxVKuJdYs/
|
||||||
```
|
```
|
||||||
|
|
||||||
Use hashed password
|
Use hashed password:
|
||||||
```
|
|
||||||
|
```sh
|
||||||
dufs -a 'admin:$6$tWMB51u6Kb2ui3wd$5gVHP92V9kZcMwQeKTjyTRgySsYJu471Jb1I6iHQ8iZ6s07GgCIO69KcPBRuwPE5tDq05xMAzye0NxVKuJdYs/@/:rw'
|
dufs -a 'admin:$6$tWMB51u6Kb2ui3wd$5gVHP92V9kZcMwQeKTjyTRgySsYJu471Jb1I6iHQ8iZ6s07GgCIO69KcPBRuwPE5tDq05xMAzye0NxVKuJdYs/@/:rw'
|
||||||
```
|
```
|
||||||
|
> The hashed password contains `$6`, which can expand to a variable in some shells, so you have to use **single quotes** to wrap it.
|
||||||
|
|
||||||
Two important things for hashed passwords:
|
Two important things for hashed passwords:
|
||||||
|
|
||||||
@@ -295,11 +303,18 @@ The log format can use following variables.
|
|||||||
| $http_ | arbitrary request header field. examples: $http_user_agent, $http_referer |
|
| $http_ | arbitrary request header field. examples: $http_user_agent, $http_referer |
|
||||||
|
|
||||||
|
|
||||||
The default log format is `'$remote_addr "$request" $status'`.
|
The default log format is `'$time_iso8601 $log_level - $remote_addr "$request" $status`.
|
||||||
```
|
```
|
||||||
2022-08-06T06:59:31+08:00 INFO - 127.0.0.1 "GET /" 200
|
2022-08-06T06:59:31+08:00 INFO - 127.0.0.1 "GET /" 200
|
||||||
```
|
```
|
||||||
|
|
||||||
|
A json log format is also supported.
|
||||||
|
```
|
||||||
|
dufs --log-format '{"time":"$time_local","addr":"$remote_addr","uri":"$request_uri", "method":"$request_method","status":$status}'
|
||||||
|
|
||||||
|
{"time":"2022-08-06T06:59:31+08:00","addr":"127.0.0.1","uri":"/", "method":"GET","status":200}
|
||||||
|
```
|
||||||
|
|
||||||
Disable http log
|
Disable http log
|
||||||
```
|
```
|
||||||
dufs --log-format=''
|
dufs --log-format=''
|
||||||
@@ -339,6 +354,7 @@ All options can be set using environment variables prefixed with `DUFS_`.
|
|||||||
--allow-search DUFS_ALLOW_SEARCH=true
|
--allow-search DUFS_ALLOW_SEARCH=true
|
||||||
--allow-symlink DUFS_ALLOW_SYMLINK=true
|
--allow-symlink DUFS_ALLOW_SYMLINK=true
|
||||||
--allow-archive DUFS_ALLOW_ARCHIVE=true
|
--allow-archive DUFS_ALLOW_ARCHIVE=true
|
||||||
|
--allow-hash DUFS_ALLOW_HASH=true
|
||||||
--enable-cors DUFS_ENABLE_CORS=true
|
--enable-cors DUFS_ENABLE_CORS=true
|
||||||
--render-index DUFS_RENDER_INDEX=true
|
--render-index DUFS_RENDER_INDEX=true
|
||||||
--render-try-index DUFS_RENDER_TRY_INDEX=true
|
--render-try-index DUFS_RENDER_TRY_INDEX=true
|
||||||
@@ -376,6 +392,7 @@ allow-delete: true
|
|||||||
allow-search: true
|
allow-search: true
|
||||||
allow-symlink: true
|
allow-symlink: true
|
||||||
allow-archive: true
|
allow-archive: true
|
||||||
|
allow-hash: true
|
||||||
enable-cors: true
|
enable-cors: true
|
||||||
render-index: true
|
render-index: true
|
||||||
render-try-index: true
|
render-try-index: true
|
||||||
@@ -396,6 +413,8 @@ Dufs allows users to customize the UI with your own assets.
|
|||||||
dufs --assets my-assets-dir/
|
dufs --assets my-assets-dir/
|
||||||
```
|
```
|
||||||
|
|
||||||
|
> If you only need to make slight adjustments to the current UI, you copy dufs's [assets](https://github.com/sigoden/dufs/tree/main/assets) directory and modify it accordingly. The current UI doesn't use any frameworks, just plain HTML/JS/CSS. As long as you have some basic knowledge of web development, it shouldn't be difficult to modify.
|
||||||
|
|
||||||
Your assets folder must contains a `index.html` file.
|
Your assets folder must contains a `index.html` file.
|
||||||
|
|
||||||
`index.html` can use the following placeholder variables to retrieve internal data.
|
`index.html` can use the following placeholder variables to retrieve internal data.
|
||||||
@@ -403,6 +422,8 @@ Your assets folder must contains a `index.html` file.
|
|||||||
- `__INDEX_DATA__`: directory listing data
|
- `__INDEX_DATA__`: directory listing data
|
||||||
- `__ASSETS_PREFIX__`: assets url prefix
|
- `__ASSETS_PREFIX__`: assets url prefix
|
||||||
|
|
||||||
|
> A customized 404.html page is also supported.
|
||||||
|
|
||||||
</details>
|
</details>
|
||||||
|
|
||||||
## License
|
## License
|
||||||
|
|||||||
+6
-3
@@ -4,6 +4,9 @@
|
|||||||
<head>
|
<head>
|
||||||
<meta charset="utf-8" />
|
<meta charset="utf-8" />
|
||||||
<meta name="viewport" content="width=device-width" />
|
<meta name="viewport" content="width=device-width" />
|
||||||
|
<noscript>
|
||||||
|
<meta http-equiv="refresh" content="0; url=?noscript">
|
||||||
|
</noscript>
|
||||||
<link rel="icon" type="image/x-icon" href="__ASSETS_PREFIX__favicon.ico">
|
<link rel="icon" type="image/x-icon" href="__ASSETS_PREFIX__favicon.ico">
|
||||||
<link rel="stylesheet" href="__ASSETS_PREFIX__index.css">
|
<link rel="stylesheet" href="__ASSETS_PREFIX__index.css">
|
||||||
</head>
|
</head>
|
||||||
@@ -20,7 +23,7 @@
|
|||||||
d="M7.646 11.854a.5.5 0 0 0 .708 0l3-3a.5.5 0 0 0-.708-.708L8.5 10.293V1.5a.5.5 0 0 0-1 0v8.793L5.354 8.146a.5.5 0 1 0-.708.708l3 3z" />
|
d="M7.646 11.854a.5.5 0 0 0 .708 0l3-3a.5.5 0 0 0-.708-.708L8.5 10.293V1.5a.5.5 0 0 0-1 0v8.793L5.354 8.146a.5.5 0 1 0-.708.708l3 3z" />
|
||||||
</svg>
|
</svg>
|
||||||
</a>
|
</a>
|
||||||
<div class="control move-file hidden" title="Move to new path">
|
<div class="control move-file hidden" title="Move & Rename">
|
||||||
<svg class="icon-move" width="16" height="16" viewBox="0 0 16 16">
|
<svg class="icon-move" width="16" height="16" viewBox="0 0 16 16">
|
||||||
<path fill-rule="evenodd"
|
<path fill-rule="evenodd"
|
||||||
d="M1.5 1.5A.5.5 0 0 0 1 2v4.8a2.5 2.5 0 0 0 2.5 2.5h9.793l-3.347 3.346a.5.5 0 0 0 .708.708l4.2-4.2a.5.5 0 0 0 0-.708l-4-4a.5.5 0 0 0-.708.708L13.293 8.3H3.5A1.5 1.5 0 0 1 2 6.8V2a.5.5 0 0 0-.5-.5z">
|
d="M1.5 1.5A.5.5 0 0 0 1 2v4.8a2.5 2.5 0 0 0 2.5 2.5h9.793l-3.347 3.346a.5.5 0 0 0 .708.708l4.2-4.2a.5.5 0 0 0 0-.708l-4-4a.5.5 0 0 0-.708.708L13.293 8.3H3.5A1.5 1.5 0 0 1 2 6.8V2a.5.5 0 0 0-.5-.5z">
|
||||||
@@ -35,7 +38,7 @@
|
|||||||
d="M14 14V4.5L9.5 0H4a2 2 0 0 0-2 2v12a2 2 0 0 0 2 2h8a2 2 0 0 0 2-2zM9.5 3A1.5 1.5 0 0 0 11 4.5h2V14a1 1 0 0 1-1 1H4a1 1 0 0 1-1-1V2a1 1 0 0 1 1-1h5.5v2z" />
|
d="M14 14V4.5L9.5 0H4a2 2 0 0 0-2 2v12a2 2 0 0 0 2 2h8a2 2 0 0 0 2-2zM9.5 3A1.5 1.5 0 0 0 11 4.5h2V14a1 1 0 0 1-1 1H4a1 1 0 0 1-1-1V2a1 1 0 0 1 1-1h5.5v2z" />
|
||||||
</svg>
|
</svg>
|
||||||
</div>
|
</div>
|
||||||
<div class="control upload-file hidden" title="Upload files">
|
<div class="control upload-file hidden" title="Upload files/folders">
|
||||||
<label for="file">
|
<label for="file">
|
||||||
<svg width="16" height="16" viewBox="0 0 16 16">
|
<svg width="16" height="16" viewBox="0 0 16 16">
|
||||||
<path
|
<path
|
||||||
@@ -44,7 +47,7 @@
|
|||||||
d="M7.646 1.146a.5.5 0 0 1 .708 0l3 3a.5.5 0 0 1-.708.708L8.5 2.707V11.5a.5.5 0 0 1-1 0V2.707L5.354 4.854a.5.5 0 1 1-.708-.708l3-3z" />
|
d="M7.646 1.146a.5.5 0 0 1 .708 0l3 3a.5.5 0 0 1-.708.708L8.5 2.707V11.5a.5.5 0 0 1-1 0V2.707L5.354 4.854a.5.5 0 1 1-.708-.708l3-3z" />
|
||||||
</svg>
|
</svg>
|
||||||
</label>
|
</label>
|
||||||
<input type="file" id="file" title="Upload files" name="file" multiple>
|
<input type="file" id="file" title="Upload files/folders" name="file" multiple>
|
||||||
</div>
|
</div>
|
||||||
<div class="control new-folder hidden" title="New folder">
|
<div class="control new-folder hidden" title="New folder">
|
||||||
<svg width="16" height="16" viewBox="0 0 16 16">
|
<svg width="16" height="16" viewBox="0 0 16 16">
|
||||||
|
|||||||
+77
-18
@@ -50,6 +50,8 @@ const IFRAME_FORMATS = [
|
|||||||
".mp3", ".ogg", ".wav", ".m4a",
|
".mp3", ".ogg", ".wav", ".m4a",
|
||||||
];
|
];
|
||||||
|
|
||||||
|
const MAX_SUBPATHS_COUNT = 1000;
|
||||||
|
|
||||||
const ICONS = {
|
const ICONS = {
|
||||||
dir: `<svg height="16" viewBox="0 0 14 16" width="14"><path fill-rule="evenodd" d="M13 4H7V3c0-.66-.31-1-1-1H1c-.55 0-1 .45-1 1v10c0 .55.45 1 1 1h12c.55 0 1-.45 1-1V5c0-.55-.45-1-1-1zM6 4H1V3h5v1z"></path></svg>`,
|
dir: `<svg height="16" viewBox="0 0 14 16" width="14"><path fill-rule="evenodd" d="M13 4H7V3c0-.66-.31-1-1-1H1c-.55 0-1 .45-1 1v10c0 .55.45 1 1 1h12c.55 0 1-.45 1-1V5c0-.55-.45-1-1-1zM6 4H1V3h5v1z"></path></svg>`,
|
||||||
symlinkFile: `<svg height="16" viewBox="0 0 12 16" width="12"><path fill-rule="evenodd" d="M8.5 1H1c-.55 0-1 .45-1 1v12c0 .55.45 1 1 1h10c.55 0 1-.45 1-1V4.5L8.5 1zM11 14H1V2h7l3 3v9zM6 4.5l4 3-4 3v-2c-.98-.02-1.84.22-2.55.7-.71.48-1.19 1.25-1.45 2.3.02-1.64.39-2.88 1.13-3.73.73-.84 1.69-1.27 2.88-1.27v-2H6z"></path></svg>`,
|
symlinkFile: `<svg height="16" viewBox="0 0 12 16" width="12"><path fill-rule="evenodd" d="M8.5 1H1c-.55 0-1 .45-1 1v12c0 .55.45 1 1 1h10c.55 0 1-.45 1-1V4.5L8.5 1zM11 14H1V2h7l3 3v9zM6 4.5l4 3-4 3v-2c-.98-.02-1.84.22-2.55.7-.71.48-1.19 1.25-1.45 2.3.02-1.64.39-2.88 1.13-3.73.73-.84 1.69-1.27 2.88-1.27v-2H6z"></path></svg>`,
|
||||||
@@ -104,6 +106,15 @@ let $logoutBtn;
|
|||||||
*/
|
*/
|
||||||
let $userName;
|
let $userName;
|
||||||
|
|
||||||
|
// manage unload event to prevent leaving with uploads in progress
|
||||||
|
const beforeUnloadHandler = (event) => {
|
||||||
|
if (Uploader.queues.length > 0 || Uploader.runnings > 0) {
|
||||||
|
event.preventDefault();
|
||||||
|
event.returnValue = '';
|
||||||
|
return ''; // for some browsers
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
// Produce table when window loads
|
// Produce table when window loads
|
||||||
window.addEventListener("DOMContentLoaded", async () => {
|
window.addEventListener("DOMContentLoaded", async () => {
|
||||||
const $indexData = document.getElementById('index-data');
|
const $indexData = document.getElementById('index-data');
|
||||||
@@ -129,6 +140,8 @@ async function ready() {
|
|||||||
$logoutBtn = document.querySelector(".logout-btn");
|
$logoutBtn = document.querySelector(".logout-btn");
|
||||||
$userName = document.querySelector(".user-name");
|
$userName = document.querySelector(".user-name");
|
||||||
|
|
||||||
|
window.addEventListener('beforeunload', beforeUnloadHandler);
|
||||||
|
|
||||||
addBreadcrumb(DATA.href, DATA.uri_prefix);
|
addBreadcrumb(DATA.href, DATA.uri_prefix);
|
||||||
|
|
||||||
if (DATA.kind === "Index") {
|
if (DATA.kind === "Index") {
|
||||||
@@ -247,12 +260,14 @@ class Uploader {
|
|||||||
|
|
||||||
progress(event) {
|
progress(event) {
|
||||||
const now = Date.now();
|
const now = Date.now();
|
||||||
const speed = (event.loaded - this.uploaded) / (now - this.lastUptime) * 1000;
|
const elapsed = now - this.lastUptime;
|
||||||
const [speedValue, speedUnit] = formatSize(speed);
|
if (elapsed < 300) return; // throttle update for safari
|
||||||
|
const speed = (event.loaded - this.uploaded) / elapsed * 1000;
|
||||||
|
const [speedValue, speedUnit] = formatFileSize(speed);
|
||||||
const speedText = `${speedValue} ${speedUnit}/s`;
|
const speedText = `${speedValue} ${speedUnit}/s`;
|
||||||
const progress = formatPercent(((event.loaded + this.uploadOffset) / this.file.size) * 100);
|
const progress = formatPercent(((event.loaded + this.uploadOffset) / this.file.size) * 100);
|
||||||
const duration = formatDuration((event.total - event.loaded) / speed);
|
const duration = formatDuration((event.total - event.loaded) / speed);
|
||||||
this.$uploadStatus.innerHTML = `<span style="width: 80px;">${speedText}</span><span>${progress} ${duration}</span>`;
|
this.$uploadStatus.innerHTML = `<span style="width: 80px;">${speedText}</span><span style="margin-left: 5px;">${progress} ${duration}</span>`;
|
||||||
this.uploaded = event.loaded;
|
this.uploaded = event.loaded;
|
||||||
this.lastUptime = now;
|
this.lastUptime = now;
|
||||||
}
|
}
|
||||||
@@ -345,6 +360,7 @@ async function setupIndexPage() {
|
|||||||
const $download = document.querySelector(".download");
|
const $download = document.querySelector(".download");
|
||||||
$download.href = baseUrl() + "?zip";
|
$download.href = baseUrl() + "?zip";
|
||||||
$download.title = "Download folder as a .zip file";
|
$download.title = "Download folder as a .zip file";
|
||||||
|
$download.classList.add("dlwt");
|
||||||
$download.classList.remove("hidden");
|
$download.classList.remove("hidden");
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -365,6 +381,10 @@ async function setupIndexPage() {
|
|||||||
|
|
||||||
renderPathsTableHead();
|
renderPathsTableHead();
|
||||||
renderPathsTableBody();
|
renderPathsTableBody();
|
||||||
|
|
||||||
|
if (DATA.user) {
|
||||||
|
setupDownloadWithToken();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -447,18 +467,18 @@ function addPath(file, index) {
|
|||||||
if (DATA.allow_archive) {
|
if (DATA.allow_archive) {
|
||||||
actionDownload = `
|
actionDownload = `
|
||||||
<div class="action-btn">
|
<div class="action-btn">
|
||||||
<a href="${url}?zip" title="Download folder as a .zip file">${ICONS.download}</a>
|
<a class="dlwt" href="${url}?zip" title="Download folder as a .zip file" download>${ICONS.download}</a>
|
||||||
</div>`;
|
</div>`;
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
actionDownload = `
|
actionDownload = `
|
||||||
<div class="action-btn" >
|
<div class="action-btn" >
|
||||||
<a href="${url}" title="Download file" download>${ICONS.download}</a>
|
<a class="dlwt" href="${url}" title="Download file" download>${ICONS.download}</a>
|
||||||
</div>`;
|
</div>`;
|
||||||
}
|
}
|
||||||
if (DATA.allow_delete) {
|
if (DATA.allow_delete) {
|
||||||
if (DATA.allow_upload) {
|
if (DATA.allow_upload) {
|
||||||
actionMove = `<div onclick="movePath(${index})" class="action-btn" id="moveBtn${index}" title="Move to new path">${ICONS.move}</div>`;
|
actionMove = `<div onclick="movePath(${index})" class="action-btn" id="moveBtn${index}" title="Move & Rename">${ICONS.move}</div>`;
|
||||||
if (!isDir) {
|
if (!isDir) {
|
||||||
actionEdit = `<a class="action-btn" title="Edit file" target="_blank" href="${url}?edit">${ICONS.edit}</a>`;
|
actionEdit = `<a class="action-btn" title="Edit file" target="_blank" href="${url}?edit">${ICONS.edit}</a>`;
|
||||||
}
|
}
|
||||||
@@ -478,6 +498,8 @@ function addPath(file, index) {
|
|||||||
${actionEdit}
|
${actionEdit}
|
||||||
</td>`;
|
</td>`;
|
||||||
|
|
||||||
|
let sizeDisplay = isDir ? formatDirSize(file.size) : formatFileSize(file.size).join(" ");
|
||||||
|
|
||||||
$pathsTableBody.insertAdjacentHTML("beforeend", `
|
$pathsTableBody.insertAdjacentHTML("beforeend", `
|
||||||
<tr id="addPath${index}">
|
<tr id="addPath${index}">
|
||||||
<td class="path cell-icon">
|
<td class="path cell-icon">
|
||||||
@@ -487,7 +509,7 @@ function addPath(file, index) {
|
|||||||
<a href="${url}" ${isDir ? "" : `target="_blank"`}>${encodedName}</a>
|
<a href="${url}" ${isDir ? "" : `target="_blank"`}>${encodedName}</a>
|
||||||
</td>
|
</td>
|
||||||
<td class="cell-mtime">${formatMtime(file.mtime)}</td>
|
<td class="cell-mtime">${formatMtime(file.mtime)}</td>
|
||||||
<td class="cell-size">${formatSize(file.size).join(" ")}</td>
|
<td class="cell-size">${sizeDisplay}</td>
|
||||||
${actionCell}
|
${actionCell}
|
||||||
</tr>`);
|
</tr>`);
|
||||||
}
|
}
|
||||||
@@ -525,13 +547,40 @@ async function setupAuth() {
|
|||||||
$loginBtn.classList.remove("hidden");
|
$loginBtn.classList.remove("hidden");
|
||||||
$loginBtn.addEventListener("click", async () => {
|
$loginBtn.addEventListener("click", async () => {
|
||||||
try {
|
try {
|
||||||
await checkAuth();
|
await checkAuth("login");
|
||||||
} catch {}
|
} catch { }
|
||||||
location.reload();
|
location.reload();
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function setupDownloadWithToken() {
|
||||||
|
document.querySelectorAll("a.dlwt").forEach(link => {
|
||||||
|
link.addEventListener("click", async e => {
|
||||||
|
e.preventDefault();
|
||||||
|
try {
|
||||||
|
const link = e.currentTarget || e.target;
|
||||||
|
const originalHref = link.getAttribute("href");
|
||||||
|
const tokengenUrl = new URL(originalHref);
|
||||||
|
tokengenUrl.searchParams.set("tokengen", "");
|
||||||
|
const res = await fetch(tokengenUrl);
|
||||||
|
if (!res.ok) throw new Error("Failed to fetch token");
|
||||||
|
const token = await res.text();
|
||||||
|
const downloadUrl = new URL(originalHref);
|
||||||
|
downloadUrl.searchParams.set("token", token);
|
||||||
|
const tempA = document.createElement("a");
|
||||||
|
tempA.href = downloadUrl.toString();
|
||||||
|
tempA.download = "";
|
||||||
|
document.body.appendChild(tempA);
|
||||||
|
tempA.click();
|
||||||
|
document.body.removeChild(tempA);
|
||||||
|
} catch (err) {
|
||||||
|
alert(`Failed to download, ${err.message}`);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
function setupSearch() {
|
function setupSearch() {
|
||||||
const $searchbar = document.querySelector(".searchbar");
|
const $searchbar = document.querySelector(".searchbar");
|
||||||
$searchbar.classList.remove("hidden");
|
$searchbar.classList.remove("hidden");
|
||||||
@@ -642,7 +691,7 @@ async function setupEditorPage() {
|
|||||||
$editor.value = decoder.decode(dataView);
|
$editor.value = decoder.decode(dataView);
|
||||||
}
|
}
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
alert(`Failed get file, ${err.message}`);
|
alert(`Failed to get file, ${err.message}`);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -746,9 +795,10 @@ async function saveChange() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async function checkAuth() {
|
async function checkAuth(variant) {
|
||||||
if (!DATA.auth) return;
|
if (!DATA.auth) return;
|
||||||
const res = await fetch(baseUrl(), {
|
const qs = variant ? `?${variant}` : "";
|
||||||
|
const res = await fetch(baseUrl() + qs, {
|
||||||
method: "CHECKAUTH",
|
method: "CHECKAUTH",
|
||||||
});
|
});
|
||||||
await assertResOK(res);
|
await assertResOK(res);
|
||||||
@@ -831,7 +881,7 @@ function newUrl(name) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
function baseUrl() {
|
function baseUrl() {
|
||||||
return location.href.split('?')[0];
|
return location.href.split(/[?#]/)[0];
|
||||||
}
|
}
|
||||||
|
|
||||||
function baseName(url) {
|
function baseName(url) {
|
||||||
@@ -876,16 +926,25 @@ function padZero(value, size) {
|
|||||||
return ("0".repeat(size) + value).slice(-1 * size);
|
return ("0".repeat(size) + value).slice(-1 * size);
|
||||||
}
|
}
|
||||||
|
|
||||||
function formatSize(size) {
|
function formatDirSize(size) {
|
||||||
|
const unit = size === 1 ? "item" : "items";
|
||||||
|
const num = size >= MAX_SUBPATHS_COUNT ? `>${MAX_SUBPATHS_COUNT - 1}` : `${size}`;
|
||||||
|
return ` ${num} ${unit}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
function formatFileSize(size) {
|
||||||
if (size == null) return [0, "B"];
|
if (size == null) return [0, "B"];
|
||||||
const sizes = ['B', 'KB', 'MB', 'GB', 'TB'];
|
const sizes = ['B', 'KB', 'MB', 'GB', 'TB'];
|
||||||
if (size == 0) return [0, "B"];
|
if (size == 0) return [0, "B"];
|
||||||
const i = parseInt(Math.floor(Math.log(size) / Math.log(1024)));
|
const i = parseInt(Math.floor(Math.log(size) / Math.log(1024)));
|
||||||
let ratio = 1;
|
const raw = size / Math.pow(1024, i);
|
||||||
if (i >= 3) {
|
let value;
|
||||||
ratio = 100;
|
if (i > 0 && raw < 999.95) {
|
||||||
|
value = Math.round(raw * 10) / 10;
|
||||||
|
} else {
|
||||||
|
value = Math.round(raw);
|
||||||
}
|
}
|
||||||
return [Math.round(size * ratio / Math.pow(1024, i), 2) / ratio, sizes[i]];
|
return [value, sizes[i]];
|
||||||
}
|
}
|
||||||
|
|
||||||
function formatDuration(seconds) {
|
function formatDuration(seconds) {
|
||||||
|
|||||||
+35
-18
@@ -146,7 +146,15 @@ pub fn build_cli() -> Command {
|
|||||||
.hide_env(true)
|
.hide_env(true)
|
||||||
.long("allow-archive")
|
.long("allow-archive")
|
||||||
.action(ArgAction::SetTrue)
|
.action(ArgAction::SetTrue)
|
||||||
.help("Allow zip archive generation"),
|
.help("Allow download folders as archive file"),
|
||||||
|
)
|
||||||
|
.arg(
|
||||||
|
Arg::new("allow-hash")
|
||||||
|
.env("DUFS_ALLOW_HASH")
|
||||||
|
.hide_env(true)
|
||||||
|
.long("allow-hash")
|
||||||
|
.action(ArgAction::SetTrue)
|
||||||
|
.help("Allow ?hash query to get file sha256 hash"),
|
||||||
)
|
)
|
||||||
.arg(
|
.arg(
|
||||||
Arg::new("enable-cors")
|
Arg::new("enable-cors")
|
||||||
@@ -281,11 +289,13 @@ pub struct Args {
|
|||||||
pub allow_search: bool,
|
pub allow_search: bool,
|
||||||
pub allow_symlink: bool,
|
pub allow_symlink: bool,
|
||||||
pub allow_archive: bool,
|
pub allow_archive: bool,
|
||||||
|
pub allow_hash: bool,
|
||||||
pub render_index: bool,
|
pub render_index: bool,
|
||||||
pub render_spa: bool,
|
pub render_spa: bool,
|
||||||
pub render_try_index: bool,
|
pub render_try_index: bool,
|
||||||
pub enable_cors: bool,
|
pub enable_cors: bool,
|
||||||
pub assets: Option<PathBuf>,
|
pub assets: Option<PathBuf>,
|
||||||
|
pub error_page: Option<PathBuf>,
|
||||||
#[serde(deserialize_with = "deserialize_log_http")]
|
#[serde(deserialize_with = "deserialize_log_http")]
|
||||||
#[serde(rename = "log-format")]
|
#[serde(rename = "log-format")]
|
||||||
pub http_logger: HttpLogger,
|
pub http_logger: HttpLogger,
|
||||||
@@ -375,6 +385,9 @@ impl Args {
|
|||||||
if !args.allow_symlink {
|
if !args.allow_symlink {
|
||||||
args.allow_symlink = allow_all || matches.get_flag("allow-symlink");
|
args.allow_symlink = allow_all || matches.get_flag("allow-symlink");
|
||||||
}
|
}
|
||||||
|
if !args.allow_hash {
|
||||||
|
args.allow_hash = allow_all || matches.get_flag("allow-hash");
|
||||||
|
}
|
||||||
if !args.allow_archive {
|
if !args.allow_archive {
|
||||||
args.allow_archive = allow_all || matches.get_flag("allow-archive");
|
args.allow_archive = allow_all || matches.get_flag("allow-archive");
|
||||||
}
|
}
|
||||||
@@ -398,6 +411,13 @@ impl Args {
|
|||||||
args.assets = Some(Args::sanitize_assets_path(assets_path)?);
|
args.assets = Some(Args::sanitize_assets_path(assets_path)?);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if let Some(assets_path) = &args.assets {
|
||||||
|
let p = assets_path.join("404.html");
|
||||||
|
if p.exists() {
|
||||||
|
args.error_page = Some(p);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if let Some(log_format) = matches.get_one::<String>("log-format") {
|
if let Some(log_format) = matches.get_one::<String>("log-format") {
|
||||||
args.http_logger = log_format.parse()?;
|
args.http_logger = log_format.parse()?;
|
||||||
}
|
}
|
||||||
@@ -461,28 +481,30 @@ impl Args {
|
|||||||
|
|
||||||
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord)]
|
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord)]
|
||||||
pub enum BindAddr {
|
pub enum BindAddr {
|
||||||
Address(IpAddr),
|
IpAddr(IpAddr),
|
||||||
Path(PathBuf),
|
#[cfg(unix)]
|
||||||
|
SocketPath(String),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl BindAddr {
|
impl BindAddr {
|
||||||
fn parse_addrs(addrs: &[&str]) -> Result<Vec<Self>> {
|
fn parse_addrs(addrs: &[&str]) -> Result<Vec<Self>> {
|
||||||
let mut bind_addrs = vec![];
|
let mut bind_addrs = vec![];
|
||||||
|
#[cfg(not(unix))]
|
||||||
let mut invalid_addrs = vec![];
|
let mut invalid_addrs = vec![];
|
||||||
for addr in addrs {
|
for addr in addrs {
|
||||||
match addr.parse::<IpAddr>() {
|
match addr.parse::<IpAddr>() {
|
||||||
Ok(v) => {
|
Ok(v) => {
|
||||||
bind_addrs.push(BindAddr::Address(v));
|
bind_addrs.push(BindAddr::IpAddr(v));
|
||||||
}
|
}
|
||||||
Err(_) => {
|
Err(_) => {
|
||||||
if cfg!(unix) {
|
#[cfg(unix)]
|
||||||
bind_addrs.push(BindAddr::Path(PathBuf::from(addr)));
|
bind_addrs.push(BindAddr::SocketPath(addr.to_string()));
|
||||||
} else {
|
#[cfg(not(unix))]
|
||||||
invalid_addrs.push(*addr);
|
invalid_addrs.push(*addr);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
#[cfg(not(unix))]
|
||||||
if !invalid_addrs.is_empty() {
|
if !invalid_addrs.is_empty() {
|
||||||
bail!("Invalid bind address `{}`", invalid_addrs.join(","));
|
bail!("Invalid bind address `{}`", invalid_addrs.join(","));
|
||||||
}
|
}
|
||||||
@@ -490,21 +512,16 @@ impl BindAddr {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Copy, PartialEq, Deserialize)]
|
#[derive(Debug, Clone, Copy, PartialEq, Deserialize, Default)]
|
||||||
#[serde(rename_all = "lowercase")]
|
#[serde(rename_all = "lowercase")]
|
||||||
pub enum Compress {
|
pub enum Compress {
|
||||||
None,
|
None,
|
||||||
|
#[default]
|
||||||
Low,
|
Low,
|
||||||
Medium,
|
Medium,
|
||||||
High,
|
High,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for Compress {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self::Low
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl ValueEnum for Compress {
|
impl ValueEnum for Compress {
|
||||||
fn value_variants<'a>() -> &'a [Self] {
|
fn value_variants<'a>() -> &'a [Self] {
|
||||||
&[Self::None, Self::Low, Self::Medium, Self::High]
|
&[Self::None, Self::Low, Self::Medium, Self::High]
|
||||||
@@ -710,7 +727,7 @@ hidden: tmp,*.log,*.lock
|
|||||||
assert_eq!(args.serve_path, Args::sanitize_path(&tmpdir).unwrap());
|
assert_eq!(args.serve_path, Args::sanitize_path(&tmpdir).unwrap());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
args.addrs,
|
args.addrs,
|
||||||
vec![BindAddr::Address("0.0.0.0".parse().unwrap())]
|
vec![BindAddr::IpAddr("0.0.0.0".parse().unwrap())]
|
||||||
);
|
);
|
||||||
assert_eq!(args.hidden, ["tmp", "*.log", "*.lock"]);
|
assert_eq!(args.hidden, ["tmp", "*.log", "*.lock"]);
|
||||||
assert_eq!(args.port, 3000);
|
assert_eq!(args.port, 3000);
|
||||||
@@ -740,8 +757,8 @@ hidden:
|
|||||||
assert_eq!(
|
assert_eq!(
|
||||||
args.addrs,
|
args.addrs,
|
||||||
vec![
|
vec![
|
||||||
BindAddr::Address("127.0.0.1".parse().unwrap()),
|
BindAddr::IpAddr("127.0.0.1".parse().unwrap()),
|
||||||
BindAddr::Address("192.168.8.10".parse().unwrap())
|
BindAddr::IpAddr("192.168.8.10".parse().unwrap())
|
||||||
]
|
]
|
||||||
);
|
);
|
||||||
assert_eq!(args.hidden, ["tmp", "*.log", "*.lock"]);
|
assert_eq!(args.hidden, ["tmp", "*.log", "*.lock"]);
|
||||||
|
|||||||
+183
-48
@@ -2,11 +2,14 @@ use crate::{args::Args, server::Response, utils::unix_now};
|
|||||||
|
|
||||||
use anyhow::{anyhow, bail, Result};
|
use anyhow::{anyhow, bail, Result};
|
||||||
use base64::{engine::general_purpose::STANDARD, Engine as _};
|
use base64::{engine::general_purpose::STANDARD, Engine as _};
|
||||||
|
use ed25519_dalek::{ed25519::signature::SignerMut, Signature, SigningKey};
|
||||||
use headers::HeaderValue;
|
use headers::HeaderValue;
|
||||||
use hyper::{header::WWW_AUTHENTICATE, Method};
|
use hyper::{header::WWW_AUTHENTICATE, Method};
|
||||||
use indexmap::IndexMap;
|
use indexmap::IndexMap;
|
||||||
use lazy_static::lazy_static;
|
use lazy_static::lazy_static;
|
||||||
use md5::Context;
|
use md5::Context;
|
||||||
|
use sha2::{Digest, Sha256};
|
||||||
|
use sha_crypt::PasswordVerifier;
|
||||||
use std::{
|
use std::{
|
||||||
collections::HashMap,
|
collections::HashMap,
|
||||||
path::{Path, PathBuf},
|
path::{Path, PathBuf},
|
||||||
@@ -14,7 +17,8 @@ use std::{
|
|||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
const REALM: &str = "DUFS";
|
const REALM: &str = "DUFS";
|
||||||
const DIGEST_AUTH_TIMEOUT: u32 = 604800; // 7 days
|
const DIGEST_AUTH_TIMEOUT: u32 = 60 * 60 * 24 * 7; // 7 days
|
||||||
|
const TOKEN_EXPIRATION: u64 = 1000 * 60 * 60 * 24 * 3; // 3 days
|
||||||
|
|
||||||
lazy_static! {
|
lazy_static! {
|
||||||
static ref NONCESTARTHASH: Context = {
|
static ref NONCESTARTHASH: Context = {
|
||||||
@@ -27,6 +31,7 @@ lazy_static! {
|
|||||||
|
|
||||||
#[derive(Debug, Clone, PartialEq)]
|
#[derive(Debug, Clone, PartialEq)]
|
||||||
pub struct AccessControl {
|
pub struct AccessControl {
|
||||||
|
empty: bool,
|
||||||
use_hashed_password: bool,
|
use_hashed_password: bool,
|
||||||
users: IndexMap<String, (String, AccessPaths)>,
|
users: IndexMap<String, (String, AccessPaths)>,
|
||||||
anonymous: Option<AccessPaths>,
|
anonymous: Option<AccessPaths>,
|
||||||
@@ -35,6 +40,7 @@ pub struct AccessControl {
|
|||||||
impl Default for AccessControl {
|
impl Default for AccessControl {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
AccessControl {
|
AccessControl {
|
||||||
|
empty: true,
|
||||||
use_hashed_password: false,
|
use_hashed_password: false,
|
||||||
users: IndexMap::new(),
|
users: IndexMap::new(),
|
||||||
anonymous: Some(AccessPaths::new(AccessPerm::ReadWrite)),
|
anonymous: Some(AccessPaths::new(AccessPerm::ReadWrite)),
|
||||||
@@ -45,7 +51,7 @@ impl Default for AccessControl {
|
|||||||
impl AccessControl {
|
impl AccessControl {
|
||||||
pub fn new(raw_rules: &[&str]) -> Result<Self> {
|
pub fn new(raw_rules: &[&str]) -> Result<Self> {
|
||||||
if raw_rules.is_empty() {
|
if raw_rules.is_empty() {
|
||||||
return Ok(Default::default());
|
return Ok(Self::default());
|
||||||
}
|
}
|
||||||
let new_raw_rules = split_rules(raw_rules);
|
let new_raw_rules = split_rules(raw_rules);
|
||||||
let mut use_hashed_password = false;
|
let mut use_hashed_password = false;
|
||||||
@@ -69,15 +75,26 @@ impl AccessControl {
|
|||||||
let mut anonymous = None;
|
let mut anonymous = None;
|
||||||
if let Some(paths) = annoy_paths {
|
if let Some(paths) = annoy_paths {
|
||||||
let mut access_paths = AccessPaths::default();
|
let mut access_paths = AccessPaths::default();
|
||||||
access_paths.merge(paths);
|
access_paths
|
||||||
|
.merge(paths)
|
||||||
|
.ok_or_else(|| anyhow!("Invalid auth value `@{paths}"))?;
|
||||||
anonymous = Some(access_paths);
|
anonymous = Some(access_paths);
|
||||||
}
|
}
|
||||||
let mut users = IndexMap::new();
|
let mut users = IndexMap::new();
|
||||||
for (user, pass, paths) in account_paths_pairs.into_iter() {
|
for (user, pass, paths) in account_paths_pairs.into_iter() {
|
||||||
let mut access_paths = anonymous.clone().unwrap_or_default();
|
let mut access_paths = AccessPaths::default();
|
||||||
access_paths
|
access_paths
|
||||||
.merge(paths)
|
.merge(paths)
|
||||||
.ok_or_else(|| anyhow!("Invalid auth `{user}:{pass}@{paths}"))?;
|
.ok_or_else(|| anyhow!("Invalid auth value `{user}:{pass}@{paths}"))?;
|
||||||
|
if let Some(anon_ap) = &anonymous {
|
||||||
|
let orig_user = access_paths.clone();
|
||||||
|
access_paths.absorb_anon(
|
||||||
|
anon_ap,
|
||||||
|
&orig_user,
|
||||||
|
AccessPerm::IndexOnly,
|
||||||
|
AccessPerm::IndexOnly,
|
||||||
|
);
|
||||||
|
}
|
||||||
if pass.starts_with("$6$") {
|
if pass.starts_with("$6$") {
|
||||||
use_hashed_password = true;
|
use_hashed_password = true;
|
||||||
}
|
}
|
||||||
@@ -85,13 +102,14 @@ impl AccessControl {
|
|||||||
}
|
}
|
||||||
|
|
||||||
Ok(Self {
|
Ok(Self {
|
||||||
|
empty: false,
|
||||||
use_hashed_password,
|
use_hashed_password,
|
||||||
users,
|
users,
|
||||||
anonymous,
|
anonymous,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn exist(&self) -> bool {
|
pub fn has_users(&self) -> bool {
|
||||||
!self.users.is_empty()
|
!self.users.is_empty()
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -100,16 +118,29 @@ impl AccessControl {
|
|||||||
path: &str,
|
path: &str,
|
||||||
method: &Method,
|
method: &Method,
|
||||||
authorization: Option<&HeaderValue>,
|
authorization: Option<&HeaderValue>,
|
||||||
|
token: Option<&String>,
|
||||||
guard_options: bool,
|
guard_options: bool,
|
||||||
) -> (Option<String>, Option<AccessPaths>) {
|
) -> (Option<String>, Option<AccessPaths>) {
|
||||||
|
if self.empty {
|
||||||
|
return (None, Some(AccessPaths::new(AccessPerm::ReadWrite)));
|
||||||
|
}
|
||||||
|
|
||||||
|
if method == Method::GET {
|
||||||
|
if let Some(token) = token {
|
||||||
|
if let Ok((user, ap)) = self.verify_token(token, path) {
|
||||||
|
return (Some(user), ap.guard(path, method));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if let Some(authorization) = authorization {
|
if let Some(authorization) = authorization {
|
||||||
if let Some(user) = get_auth_user(authorization) {
|
if let Some(user) = get_auth_user(authorization) {
|
||||||
if let Some((pass, paths)) = self.users.get(&user) {
|
if let Some((pass, ap)) = self.users.get(&user) {
|
||||||
if method == Method::OPTIONS {
|
if method == Method::OPTIONS {
|
||||||
return (Some(user), Some(AccessPaths::new(AccessPerm::ReadOnly)));
|
return (Some(user), Some(AccessPaths::new(AccessPerm::ReadOnly)));
|
||||||
}
|
}
|
||||||
if check_auth(authorization, method.as_str(), &user, pass).is_some() {
|
if check_auth(authorization, method.as_str(), &user, pass).is_some() {
|
||||||
return (Some(user), paths.find(path, !is_readonly_method(method)));
|
return (Some(user), ap.guard(path, method));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -121,12 +152,59 @@ impl AccessControl {
|
|||||||
return (None, Some(AccessPaths::new(AccessPerm::ReadOnly)));
|
return (None, Some(AccessPaths::new(AccessPerm::ReadOnly)));
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some(paths) = self.anonymous.as_ref() {
|
if let Some(ap) = self.anonymous.as_ref() {
|
||||||
return (None, paths.find(path, !is_readonly_method(method)));
|
return (None, ap.guard(path, method));
|
||||||
}
|
}
|
||||||
|
|
||||||
(None, None)
|
(None, None)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn generate_token(&self, path: &str, user: &str) -> Result<String> {
|
||||||
|
let (pass, _) = self
|
||||||
|
.users
|
||||||
|
.get(user)
|
||||||
|
.ok_or_else(|| anyhow!("Not found user '{user}'"))?;
|
||||||
|
let exp = unix_now().as_millis() as u64 + TOKEN_EXPIRATION;
|
||||||
|
let message = format!("{path}:{exp}");
|
||||||
|
let mut signing_key = derive_secret_key(user, pass);
|
||||||
|
let sig = signing_key.sign(message.as_bytes()).to_bytes();
|
||||||
|
|
||||||
|
let mut raw = Vec::with_capacity(64 + 8 + user.len());
|
||||||
|
raw.extend_from_slice(&sig);
|
||||||
|
raw.extend_from_slice(&exp.to_be_bytes());
|
||||||
|
raw.extend_from_slice(user.as_bytes());
|
||||||
|
|
||||||
|
Ok(hex::encode(raw))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn verify_token<'a>(&'a self, token: &str, path: &str) -> Result<(String, &'a AccessPaths)> {
|
||||||
|
let raw = hex::decode(token)?;
|
||||||
|
|
||||||
|
if raw.len() < 72 {
|
||||||
|
bail!("Invalid token");
|
||||||
|
}
|
||||||
|
|
||||||
|
let sig_bytes = &raw[..64];
|
||||||
|
let exp_bytes = &raw[64..72];
|
||||||
|
let user_bytes = &raw[72..];
|
||||||
|
|
||||||
|
let exp = u64::from_be_bytes(exp_bytes.try_into()?);
|
||||||
|
if unix_now().as_millis() as u64 > exp {
|
||||||
|
bail!("Token expired");
|
||||||
|
}
|
||||||
|
|
||||||
|
let user = std::str::from_utf8(user_bytes)?;
|
||||||
|
let (pass, ap) = self
|
||||||
|
.users
|
||||||
|
.get(user)
|
||||||
|
.ok_or_else(|| anyhow!("Not found user '{user}'"))?;
|
||||||
|
|
||||||
|
let sig = Signature::from_bytes(&<[u8; 64]>::try_from(sig_bytes)?);
|
||||||
|
|
||||||
|
let message = format!("{path}:{exp}");
|
||||||
|
derive_secret_key(user, pass).verify(message.as_bytes(), &sig)?;
|
||||||
|
Ok((user.to_string(), ap))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Default, Clone, PartialEq, Eq)]
|
#[derive(Debug, Default, Clone, PartialEq, Eq)]
|
||||||
@@ -166,6 +244,14 @@ impl AccessPaths {
|
|||||||
Some(())
|
Some(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn guard(&self, path: &str, method: &Method) -> Option<Self> {
|
||||||
|
let target = self.find(path)?;
|
||||||
|
if !is_readonly_method(method) && !target.perm().readwrite() {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
Some(target)
|
||||||
|
}
|
||||||
|
|
||||||
fn add(&mut self, path: &str, perm: AccessPerm) {
|
fn add(&mut self, path: &str, perm: AccessPerm) {
|
||||||
let path = path.trim_matches('/');
|
let path = path.trim_matches('/');
|
||||||
if path.is_empty() {
|
if path.is_empty() {
|
||||||
@@ -177,26 +263,55 @@ impl AccessPaths {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn add_impl(&mut self, parts: &[&str], perm: AccessPerm) {
|
fn add_impl(&mut self, parts: &[&str], perm: AccessPerm) {
|
||||||
let parts_len = parts.len();
|
if parts.is_empty() {
|
||||||
if parts_len == 0 {
|
self.perm = perm;
|
||||||
self.set_perm(perm);
|
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
let child = self.children.entry(parts[0].to_string()).or_default();
|
let child = self.children.entry(parts[0].to_string()).or_default();
|
||||||
child.add_impl(&parts[1..], perm)
|
child.add_impl(&parts[1..], perm)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn find(&self, path: &str, writable: bool) -> Option<AccessPaths> {
|
/// Merge anonymous `AccessPaths` into `self` (a user's paths) with "higher perm wins" semantics.
|
||||||
|
/// `orig_user` is a snapshot of `self` before any anonymous merging begins, used so that
|
||||||
|
/// the user's own effective perm is measured against the pre-merge state.
|
||||||
|
fn absorb_anon(
|
||||||
|
&mut self,
|
||||||
|
anon: &AccessPaths,
|
||||||
|
orig_user: &AccessPaths,
|
||||||
|
user_inherited: AccessPerm,
|
||||||
|
anon_inherited: AccessPerm,
|
||||||
|
) {
|
||||||
|
let anon_eff = if !anon.perm.indexonly() {
|
||||||
|
anon.perm
|
||||||
|
} else {
|
||||||
|
anon_inherited
|
||||||
|
};
|
||||||
|
let orig_user_eff = if !orig_user.perm.indexonly() {
|
||||||
|
orig_user.perm
|
||||||
|
} else {
|
||||||
|
user_inherited
|
||||||
|
};
|
||||||
|
|
||||||
|
let combined = std::cmp::max(anon_eff, orig_user_eff);
|
||||||
|
if !combined.indexonly() && combined > self.perm {
|
||||||
|
self.perm = combined;
|
||||||
|
}
|
||||||
|
|
||||||
|
let default_ap = AccessPaths::default();
|
||||||
|
for (name, anon_child) in &anon.children {
|
||||||
|
let orig_user_child = orig_user.children.get(name).unwrap_or(&default_ap);
|
||||||
|
let user_child = self.children.entry(name.clone()).or_default();
|
||||||
|
user_child.absorb_anon(anon_child, orig_user_child, orig_user_eff, anon_eff);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn find(&self, path: &str) -> Option<AccessPaths> {
|
||||||
let parts: Vec<&str> = path
|
let parts: Vec<&str> = path
|
||||||
.trim_matches('/')
|
.trim_matches('/')
|
||||||
.split('/')
|
.split('/')
|
||||||
.filter(|v| !v.is_empty())
|
.filter(|v| !v.is_empty())
|
||||||
.collect();
|
.collect();
|
||||||
let target = self.find_impl(&parts, self.perm)?;
|
self.find_impl(&parts, self.perm)
|
||||||
if writable && !target.perm().readwrite() {
|
|
||||||
return None;
|
|
||||||
}
|
|
||||||
Some(target)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn find_impl(&self, parts: &[&str], perm: AccessPerm) -> Option<AccessPaths> {
|
fn find_impl(&self, parts: &[&str], perm: AccessPerm) -> Option<AccessPaths> {
|
||||||
@@ -229,20 +344,20 @@ impl AccessPaths {
|
|||||||
self.children.keys().collect()
|
self.children.keys().collect()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn child_paths(&self, base: &Path) -> Vec<PathBuf> {
|
pub fn entry_paths(&self, base: &Path) -> Vec<PathBuf> {
|
||||||
if !self.perm().indexonly() {
|
if !self.perm().indexonly() {
|
||||||
return vec![base.to_path_buf()];
|
return vec![base.to_path_buf()];
|
||||||
}
|
}
|
||||||
let mut output = vec![];
|
let mut output = vec![];
|
||||||
self.child_paths_impl(&mut output, base);
|
self.entry_paths_impl(&mut output, base);
|
||||||
output
|
output
|
||||||
}
|
}
|
||||||
|
|
||||||
fn child_paths_impl(&self, output: &mut Vec<PathBuf>, base: &Path) {
|
fn entry_paths_impl(&self, output: &mut Vec<PathBuf>, base: &Path) {
|
||||||
for (name, child) in self.children.iter() {
|
for (name, child) in self.children.iter() {
|
||||||
let base = base.join(name);
|
let base = base.join(name);
|
||||||
if child.perm().indexonly() {
|
if child.perm().indexonly() {
|
||||||
child.child_paths_impl(output, &base);
|
child.entry_paths_impl(output, &base);
|
||||||
} else {
|
} else {
|
||||||
output.push(base)
|
output.push(base)
|
||||||
}
|
}
|
||||||
@@ -270,15 +385,14 @@ impl AccessPerm {
|
|||||||
|
|
||||||
pub fn www_authenticate(res: &mut Response, args: &Args) -> Result<()> {
|
pub fn www_authenticate(res: &mut Response, args: &Args) -> Result<()> {
|
||||||
if args.auth.use_hashed_password {
|
if args.auth.use_hashed_password {
|
||||||
let basic = HeaderValue::from_str(&format!("Basic realm=\"{}\"", REALM))?;
|
let basic = HeaderValue::from_str(&format!("Basic realm=\"{REALM}\""))?;
|
||||||
res.headers_mut().insert(WWW_AUTHENTICATE, basic);
|
res.headers_mut().insert(WWW_AUTHENTICATE, basic);
|
||||||
} else {
|
} else {
|
||||||
let nonce = create_nonce()?;
|
let nonce = create_nonce()?;
|
||||||
let digest = HeaderValue::from_str(&format!(
|
let digest = HeaderValue::from_str(&format!(
|
||||||
"Digest realm=\"{}\", nonce=\"{}\", qop=\"auth\"",
|
"Digest realm=\"{REALM}\", nonce=\"{nonce}\", qop=\"auth\""
|
||||||
REALM, nonce
|
|
||||||
))?;
|
))?;
|
||||||
let basic = HeaderValue::from_str(&format!("Basic realm=\"{}\"", REALM))?;
|
let basic = HeaderValue::from_str(&format!("Basic realm=\"{REALM}\""))?;
|
||||||
res.headers_mut().append(WWW_AUTHENTICATE, digest);
|
res.headers_mut().append(WWW_AUTHENTICATE, digest);
|
||||||
res.headers_mut().append(WWW_AUTHENTICATE, basic);
|
res.headers_mut().append(WWW_AUTHENTICATE, basic);
|
||||||
}
|
}
|
||||||
@@ -307,17 +421,20 @@ pub fn check_auth(
|
|||||||
) -> Option<()> {
|
) -> Option<()> {
|
||||||
if let Some(value) = strip_prefix(authorization.as_bytes(), b"Basic ") {
|
if let Some(value) = strip_prefix(authorization.as_bytes(), b"Basic ") {
|
||||||
let value: Vec<u8> = STANDARD.decode(value).ok()?;
|
let value: Vec<u8> = STANDARD.decode(value).ok()?;
|
||||||
let parts: Vec<&str> = std::str::from_utf8(&value).ok()?.split(':').collect();
|
let (user, pass) = std::str::from_utf8(&value).ok()?.split_once(':')?;
|
||||||
|
|
||||||
if parts[0] != auth_user {
|
if user != auth_user {
|
||||||
return None;
|
return None;
|
||||||
}
|
}
|
||||||
|
|
||||||
if auth_pass.starts_with("$6$") {
|
if auth_pass.starts_with("$6$") {
|
||||||
if let Ok(()) = sha_crypt::sha512_check(parts[1], auth_pass) {
|
if sha_crypt::ShaCrypt::SHA512
|
||||||
|
.verify_password(pass.as_bytes(), auth_pass)
|
||||||
|
.is_ok()
|
||||||
|
{
|
||||||
return Some(());
|
return Some(());
|
||||||
}
|
}
|
||||||
} else if parts[1] == auth_pass {
|
} else if pass == auth_pass {
|
||||||
return Some(());
|
return Some(());
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -340,8 +457,8 @@ pub fn check_auth(
|
|||||||
}
|
}
|
||||||
|
|
||||||
let mut h = Context::new();
|
let mut h = Context::new();
|
||||||
h.consume(format!("{}:{}:{}", auth_user, REALM, auth_pass).as_bytes());
|
h.consume(format!("{auth_user}:{REALM}:{auth_pass}").as_bytes());
|
||||||
let auth_pass = format!("{:x}", h.compute());
|
let auth_pass = format!("{:x}", h.finalize());
|
||||||
|
|
||||||
let mut ha = Context::new();
|
let mut ha = Context::new();
|
||||||
ha.consume(method);
|
ha.consume(method);
|
||||||
@@ -349,7 +466,7 @@ pub fn check_auth(
|
|||||||
if let Some(uri) = digest_map.get(b"uri".as_ref()) {
|
if let Some(uri) = digest_map.get(b"uri".as_ref()) {
|
||||||
ha.consume(uri);
|
ha.consume(uri);
|
||||||
}
|
}
|
||||||
let ha = format!("{:x}", ha.compute());
|
let ha = format!("{:x}", ha.finalize());
|
||||||
let mut correct_response = None;
|
let mut correct_response = None;
|
||||||
if let Some(qop) = digest_map.get(b"qop".as_ref()) {
|
if let Some(qop) = digest_map.get(b"qop".as_ref()) {
|
||||||
if qop == &b"auth".as_ref() || qop == &b"auth-int".as_ref() {
|
if qop == &b"auth".as_ref() || qop == &b"auth-int".as_ref() {
|
||||||
@@ -370,7 +487,7 @@ pub fn check_auth(
|
|||||||
c.consume(qop);
|
c.consume(qop);
|
||||||
c.consume(b":");
|
c.consume(b":");
|
||||||
c.consume(&*ha);
|
c.consume(&*ha);
|
||||||
format!("{:x}", c.compute())
|
format!("{:x}", c.finalize())
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -383,7 +500,7 @@ pub fn check_auth(
|
|||||||
c.consume(nonce);
|
c.consume(nonce);
|
||||||
c.consume(b":");
|
c.consume(b":");
|
||||||
c.consume(&*ha);
|
c.consume(&*ha);
|
||||||
format!("{:x}", c.compute())
|
format!("{:x}", c.finalize())
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
if correct_response.as_bytes() == *user_response {
|
if correct_response.as_bytes() == *user_response {
|
||||||
@@ -396,6 +513,13 @@ pub fn check_auth(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn derive_secret_key(user: &str, pass: &str) -> SigningKey {
|
||||||
|
let mut hasher = Sha256::new();
|
||||||
|
hasher.update(format!("{user}:{pass}").as_bytes());
|
||||||
|
let hash = hasher.finalize();
|
||||||
|
SigningKey::from_bytes(&hash.into())
|
||||||
|
}
|
||||||
|
|
||||||
/// Check if a nonce is still valid.
|
/// Check if a nonce is still valid.
|
||||||
/// Return an error if it was never valid
|
/// Return an error if it was never valid
|
||||||
fn validate_nonce(nonce: &[u8]) -> Result<bool> {
|
fn validate_nonce(nonce: &[u8]) -> Result<bool> {
|
||||||
@@ -407,14 +531,14 @@ fn validate_nonce(nonce: &[u8]) -> Result<bool> {
|
|||||||
//get time
|
//get time
|
||||||
if let Ok(secs_nonce) = u32::from_str_radix(&n[..8], 16) {
|
if let Ok(secs_nonce) = u32::from_str_radix(&n[..8], 16) {
|
||||||
//check time
|
//check time
|
||||||
let now = unix_now()?;
|
let now = unix_now();
|
||||||
let secs_now = now.as_secs() as u32;
|
let secs_now = now.as_secs() as u32;
|
||||||
|
|
||||||
if let Some(dur) = secs_now.checked_sub(secs_nonce) {
|
if let Some(dur) = secs_now.checked_sub(secs_nonce) {
|
||||||
//check hash
|
//check hash
|
||||||
let mut h = NONCESTARTHASH.clone();
|
let mut h = NONCESTARTHASH.clone();
|
||||||
h.consume(secs_nonce.to_be_bytes());
|
h.consume(secs_nonce.to_be_bytes());
|
||||||
let h = format!("{:x}", h.compute());
|
let h = format!("{:x}", h.finalize());
|
||||||
if h[..26] == n[8..34] {
|
if h[..26] == n[8..34] {
|
||||||
return Ok(dur < DIGEST_AUTH_TIMEOUT);
|
return Ok(dur < DIGEST_AUTH_TIMEOUT);
|
||||||
}
|
}
|
||||||
@@ -487,12 +611,12 @@ fn to_headermap(header: &[u8]) -> Result<HashMap<&[u8], &[u8]>, ()> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn create_nonce() -> Result<String> {
|
fn create_nonce() -> Result<String> {
|
||||||
let now = unix_now()?;
|
let now = unix_now();
|
||||||
let secs = now.as_secs() as u32;
|
let secs = now.as_secs() as u32;
|
||||||
let mut h = NONCESTARTHASH.clone();
|
let mut h = NONCESTARTHASH.clone();
|
||||||
h.consume(secs.to_be_bytes());
|
h.consume(secs.to_be_bytes());
|
||||||
|
|
||||||
let n = format!("{:08x}{:032x}", secs, h.compute());
|
let n = format!("{:08x}{:032x}", secs, h.finalize());
|
||||||
Ok(n[..34].to_string())
|
Ok(n[..34].to_string())
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -574,7 +698,7 @@ mod tests {
|
|||||||
paths.add("/dir2/dir22/dir221", AccessPerm::ReadWrite);
|
paths.add("/dir2/dir22/dir221", AccessPerm::ReadWrite);
|
||||||
paths.add("/dir2/dir23/dir231", AccessPerm::ReadWrite);
|
paths.add("/dir2/dir23/dir231", AccessPerm::ReadWrite);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
paths.child_paths(Path::new("/tmp")),
|
paths.entry_paths(Path::new("/tmp")),
|
||||||
[
|
[
|
||||||
"/tmp/dir1",
|
"/tmp/dir1",
|
||||||
"/tmp/dir2/dir21",
|
"/tmp/dir2/dir21",
|
||||||
@@ -587,8 +711,8 @@ mod tests {
|
|||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
paths
|
paths
|
||||||
.find("dir2", false)
|
.find("dir2")
|
||||||
.map(|v| v.child_paths(Path::new("/tmp/dir2"))),
|
.map(|v| v.entry_paths(Path::new("/tmp/dir2"))),
|
||||||
Some(
|
Some(
|
||||||
[
|
[
|
||||||
"/tmp/dir2/dir21",
|
"/tmp/dir2/dir21",
|
||||||
@@ -600,19 +724,30 @@ mod tests {
|
|||||||
.collect::<Vec<_>>()
|
.collect::<Vec<_>>()
|
||||||
)
|
)
|
||||||
);
|
);
|
||||||
assert_eq!(paths.find("dir2", true), None);
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
paths.find("dir1/file", true),
|
paths.find("dir1/file"),
|
||||||
Some(AccessPaths::new(AccessPerm::ReadWrite))
|
Some(AccessPaths::new(AccessPerm::ReadWrite))
|
||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
paths.find("dir2/dir21/file", true),
|
paths.find("dir2/dir21/file"),
|
||||||
Some(AccessPaths::new(AccessPerm::ReadWrite))
|
Some(AccessPaths::new(AccessPerm::ReadWrite))
|
||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
paths.find("dir2/dir21/dir211/file", false),
|
paths.find("dir2/dir21/dir211/file"),
|
||||||
Some(AccessPaths::new(AccessPerm::ReadOnly))
|
Some(AccessPaths::new(AccessPerm::ReadOnly))
|
||||||
);
|
);
|
||||||
assert_eq!(paths.find("dir2/dir21/dir211/file", true), None);
|
assert_eq!(
|
||||||
|
paths.find("dir2/dir22/file"),
|
||||||
|
Some(AccessPaths::new(AccessPerm::ReadOnly))
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
paths.find("dir2/dir22/dir221/file"),
|
||||||
|
Some(AccessPaths::new(AccessPerm::ReadWrite))
|
||||||
|
);
|
||||||
|
assert_eq!(paths.find("dir2/dir23/file"), None);
|
||||||
|
assert_eq!(
|
||||||
|
paths.find("dir2/dir23//dir231/file"),
|
||||||
|
Some(AccessPaths::new(AccessPerm::ReadWrite))
|
||||||
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+78
-10
@@ -1,8 +1,15 @@
|
|||||||
use std::{collections::HashMap, str::FromStr};
|
use std::{
|
||||||
|
collections::HashMap,
|
||||||
|
str::FromStr,
|
||||||
|
time::{SystemTime, UNIX_EPOCH},
|
||||||
|
};
|
||||||
|
|
||||||
use crate::{auth::get_auth_user, server::Request};
|
use chrono::{Local, SecondsFormat};
|
||||||
|
|
||||||
pub const DEFAULT_LOG_FORMAT: &str = r#"$remote_addr "$request" $status"#;
|
use crate::{auth::get_auth_user, server::Request, utils::decode_uri};
|
||||||
|
|
||||||
|
pub const DEFAULT_LOG_FORMAT: &str =
|
||||||
|
r#"$time_iso8601 $log_level - $remote_addr "$request" $status"#;
|
||||||
|
|
||||||
#[derive(Debug, Clone, PartialEq)]
|
#[derive(Debug, Clone, PartialEq)]
|
||||||
pub struct HttpLogger {
|
pub struct HttpLogger {
|
||||||
@@ -28,8 +35,17 @@ impl HttpLogger {
|
|||||||
for element in self.elements.iter() {
|
for element in self.elements.iter() {
|
||||||
match element {
|
match element {
|
||||||
LogElement::Variable(name) => match name.as_str() {
|
LogElement::Variable(name) => match name.as_str() {
|
||||||
"request" => {
|
"request" | "request_method" | "request_uri" => {
|
||||||
data.insert(name.to_string(), format!("{} {}", req.method(), req.uri()));
|
let uri = req.uri().to_string();
|
||||||
|
let decoded_uri = decode_uri(&uri)
|
||||||
|
.map(|s| sanitize_log_value(&s))
|
||||||
|
.unwrap_or_else(|| uri.clone());
|
||||||
|
data.entry("request".to_string())
|
||||||
|
.or_insert_with(|| format!("{} {decoded_uri}", req.method()));
|
||||||
|
data.entry("request_method".to_string())
|
||||||
|
.or_insert_with(|| req.method().to_string());
|
||||||
|
data.entry("request_uri".to_string())
|
||||||
|
.or_insert_with(|| decoded_uri);
|
||||||
}
|
}
|
||||||
"remote_user" => {
|
"remote_user" => {
|
||||||
if let Some(user) =
|
if let Some(user) =
|
||||||
@@ -42,7 +58,7 @@ impl HttpLogger {
|
|||||||
},
|
},
|
||||||
LogElement::Header(name) => {
|
LogElement::Header(name) => {
|
||||||
if let Some(value) = req.headers().get(name).and_then(|v| v.to_str().ok()) {
|
if let Some(value) = req.headers().get(name).and_then(|v| v.to_str().ok()) {
|
||||||
data.insert(name.to_string(), value.to_string());
|
data.insert(name.to_string(), sanitize_log_value(value));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
LogElement::Literal(_) => {}
|
LogElement::Literal(_) => {}
|
||||||
@@ -50,26 +66,67 @@ impl HttpLogger {
|
|||||||
}
|
}
|
||||||
data
|
data
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn log(&self, data: &HashMap<String, String>, err: Option<String>) {
|
pub fn log(&self, data: &HashMap<String, String>, err: Option<String>) {
|
||||||
if self.elements.is_empty() {
|
if self.elements.is_empty() {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
let is_error = err.is_some();
|
||||||
|
let now = Local::now();
|
||||||
|
let time_local = now.to_rfc3339_opts(SecondsFormat::Secs, false);
|
||||||
|
let time_iso8601 = now.to_rfc3339_opts(SecondsFormat::Secs, true);
|
||||||
|
let msec = SystemTime::now()
|
||||||
|
.duration_since(UNIX_EPOCH)
|
||||||
|
.map(|d| format!("{:.3}", d.as_secs_f64()))
|
||||||
|
.unwrap_or_default();
|
||||||
|
let log_level = if is_error { "ERROR" } else { "INFO" };
|
||||||
|
|
||||||
let mut output = String::new();
|
let mut output = String::new();
|
||||||
for element in self.elements.iter() {
|
for element in self.elements.iter() {
|
||||||
match element {
|
match element {
|
||||||
LogElement::Literal(value) => output.push_str(value.as_str()),
|
LogElement::Literal(value) => output.push_str(value.as_str()),
|
||||||
LogElement::Header(name) | LogElement::Variable(name) => {
|
LogElement::Variable(name) => {
|
||||||
output.push_str(data.get(name).map(|v| v.as_str()).unwrap_or("-"))
|
let resolved = match name.as_str() {
|
||||||
|
"time_local" => Some(time_local.as_str()),
|
||||||
|
"time_iso8601" => Some(time_iso8601.as_str()),
|
||||||
|
"msec" => Some(msec.as_str()),
|
||||||
|
"log_level" => Some(log_level),
|
||||||
|
_ => None,
|
||||||
|
};
|
||||||
|
let val = resolved
|
||||||
|
.or_else(|| data.get(name.as_str()).map(|v| v.as_str()))
|
||||||
|
.unwrap_or("-");
|
||||||
|
output.push_str(val);
|
||||||
|
}
|
||||||
|
LogElement::Header(name) => {
|
||||||
|
output.push_str(data.get(name.as_str()).map(|v| v.as_str()).unwrap_or("-"))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
match err {
|
match err {
|
||||||
Some(err) => error!("{} {}", output, err),
|
Some(err) => emit_http_access(&format!("{output} {err}"), true),
|
||||||
None => info!("{}", output),
|
None => emit_http_access(&output, false),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Emit via the `log` crate with target `http_access` so the system logger
|
||||||
|
/// prints the line verbatim (no extra timestamp/level prefix).
|
||||||
|
fn emit_http_access(msg: &str, is_error: bool) {
|
||||||
|
let level = if is_error {
|
||||||
|
log::Level::Error
|
||||||
|
} else {
|
||||||
|
log::Level::Info
|
||||||
|
};
|
||||||
|
log::logger().log(
|
||||||
|
&log::Record::builder()
|
||||||
|
.args(format_args!("{}", msg))
|
||||||
|
.level(level)
|
||||||
|
.target("http_access")
|
||||||
|
.build(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
impl FromStr for HttpLogger {
|
impl FromStr for HttpLogger {
|
||||||
type Err = anyhow::Error;
|
type Err = anyhow::Error;
|
||||||
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
||||||
@@ -101,3 +158,14 @@ impl FromStr for HttpLogger {
|
|||||||
Ok(Self { elements })
|
Ok(Self { elements })
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn sanitize_log_value(s: &str) -> String {
|
||||||
|
s.chars()
|
||||||
|
.flat_map(|c| match c {
|
||||||
|
'\\' => vec!['\\', '\\'],
|
||||||
|
'"' => vec!['\\', '"'],
|
||||||
|
c if c.is_control() => format!("\\x{:02x}", c as u32).chars().collect::<Vec<_>>(),
|
||||||
|
c => vec![c],
|
||||||
|
})
|
||||||
|
.collect()
|
||||||
|
}
|
||||||
|
|||||||
+1
-3
@@ -1,5 +1,4 @@
|
|||||||
use anyhow::{Context, Result};
|
use anyhow::{Context, Result};
|
||||||
use chrono::{Local, SecondsFormat};
|
|
||||||
use log::{Level, LevelFilter, Metadata, Record};
|
use log::{Level, LevelFilter, Metadata, Record};
|
||||||
use std::fs::{File, OpenOptions};
|
use std::fs::{File, OpenOptions};
|
||||||
use std::io::Write;
|
use std::io::Write;
|
||||||
@@ -17,8 +16,7 @@ impl log::Log for SimpleLogger {
|
|||||||
|
|
||||||
fn log(&self, record: &Record) {
|
fn log(&self, record: &Record) {
|
||||||
if self.enabled(record.metadata()) {
|
if self.enabled(record.metadata()) {
|
||||||
let timestamp = Local::now().to_rfc3339_opts(SecondsFormat::Secs, true);
|
let text = record.args().to_string();
|
||||||
let text = format!("{} {} - {}", timestamp, record.level(), record.args());
|
|
||||||
match &self.file {
|
match &self.file {
|
||||||
Some(file) => {
|
Some(file) => {
|
||||||
if let Ok(mut file) = file.lock() {
|
if let Ok(mut file) = file.lock() {
|
||||||
|
|||||||
+23
-14
@@ -3,6 +3,7 @@ mod auth;
|
|||||||
mod http_logger;
|
mod http_logger;
|
||||||
mod http_utils;
|
mod http_utils;
|
||||||
mod logger;
|
mod logger;
|
||||||
|
mod noscript;
|
||||||
mod server;
|
mod server;
|
||||||
mod utils;
|
mod utils;
|
||||||
|
|
||||||
@@ -57,7 +58,7 @@ async fn main() -> Result<()> {
|
|||||||
ret = join_all(handles) => {
|
ret = join_all(handles) => {
|
||||||
for r in ret {
|
for r in ret {
|
||||||
if let Err(e) = r {
|
if let Err(e) = r {
|
||||||
error!("{}", e);
|
error!("{e}");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Ok(())
|
Ok(())
|
||||||
@@ -78,7 +79,7 @@ fn serve(args: Args, running: Arc<AtomicBool>) -> Result<Vec<JoinHandle<()>>> {
|
|||||||
for bind_addr in addrs.iter() {
|
for bind_addr in addrs.iter() {
|
||||||
let server_handle = server_handle.clone();
|
let server_handle = server_handle.clone();
|
||||||
match bind_addr {
|
match bind_addr {
|
||||||
BindAddr::Address(ip) => {
|
BindAddr::IpAddr(ip) => {
|
||||||
let listener = create_listener(SocketAddr::new(*ip, port))
|
let listener = create_listener(SocketAddr::new(*ip, port))
|
||||||
.with_context(|| format!("Failed to bind `{ip}:{port}`"))?;
|
.with_context(|| format!("Failed to bind `{ip}:{port}`"))?;
|
||||||
|
|
||||||
@@ -140,14 +141,21 @@ fn serve(args: Args, running: Arc<AtomicBool>) -> Result<Vec<JoinHandle<()>>> {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
BindAddr::Path(path) => {
|
|
||||||
if path.exists() {
|
|
||||||
std::fs::remove_file(path)?;
|
|
||||||
}
|
|
||||||
#[cfg(unix)]
|
#[cfg(unix)]
|
||||||
|
BindAddr::SocketPath(path) => {
|
||||||
|
let socket_path = if path.starts_with("@")
|
||||||
|
&& cfg!(any(target_os = "linux", target_os = "android"))
|
||||||
{
|
{
|
||||||
let listener = tokio::net::UnixListener::bind(path)
|
let mut path_buf = path.as_bytes().to_vec();
|
||||||
.with_context(|| format!("Failed to bind `{}`", path.display()))?;
|
path_buf[0] = b'\0';
|
||||||
|
unsafe { std::ffi::OsStr::from_encoded_bytes_unchecked(&path_buf) }
|
||||||
|
.to_os_string()
|
||||||
|
} else {
|
||||||
|
let _ = std::fs::remove_file(path);
|
||||||
|
path.into()
|
||||||
|
};
|
||||||
|
let listener = tokio::net::UnixListener::bind(socket_path)
|
||||||
|
.with_context(|| format!("Failed to bind `{path}`"))?;
|
||||||
let handle = tokio::spawn(async move {
|
let handle = tokio::spawn(async move {
|
||||||
loop {
|
loop {
|
||||||
let Ok((stream, _addr)) = listener.accept().await else {
|
let Ok((stream, _addr)) = listener.accept().await else {
|
||||||
@@ -162,7 +170,6 @@ fn serve(args: Args, running: Arc<AtomicBool>) -> Result<Vec<JoinHandle<()>>> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
Ok(handles)
|
Ok(handles)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -207,7 +214,7 @@ fn check_addrs(args: &Args) -> Result<(Vec<BindAddr>, Vec<BindAddr>)> {
|
|||||||
let (ipv4_addrs, ipv6_addrs) = interface_addrs()?;
|
let (ipv4_addrs, ipv6_addrs) = interface_addrs()?;
|
||||||
for bind_addr in args.addrs.iter() {
|
for bind_addr in args.addrs.iter() {
|
||||||
match bind_addr {
|
match bind_addr {
|
||||||
BindAddr::Address(ip) => match &ip {
|
BindAddr::IpAddr(ip) => match &ip {
|
||||||
IpAddr::V4(_) => {
|
IpAddr::V4(_) => {
|
||||||
if !ipv4_addrs.is_empty() {
|
if !ipv4_addrs.is_empty() {
|
||||||
new_addrs.push(bind_addr.clone());
|
new_addrs.push(bind_addr.clone());
|
||||||
@@ -229,6 +236,7 @@ fn check_addrs(args: &Args) -> Result<(Vec<BindAddr>, Vec<BindAddr>)> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
#[cfg(unix)]
|
||||||
_ => {
|
_ => {
|
||||||
new_addrs.push(bind_addr.clone());
|
new_addrs.push(bind_addr.clone());
|
||||||
print_addrs.push(bind_addr.clone())
|
print_addrs.push(bind_addr.clone())
|
||||||
@@ -246,10 +254,10 @@ fn interface_addrs() -> Result<(Vec<BindAddr>, Vec<BindAddr>)> {
|
|||||||
for iface in ifaces.into_iter() {
|
for iface in ifaces.into_iter() {
|
||||||
let ip = iface.ip();
|
let ip = iface.ip();
|
||||||
if ip.is_ipv4() {
|
if ip.is_ipv4() {
|
||||||
ipv4_addrs.push(BindAddr::Address(ip))
|
ipv4_addrs.push(BindAddr::IpAddr(ip))
|
||||||
}
|
}
|
||||||
if ip.is_ipv6() {
|
if ip.is_ipv6() {
|
||||||
ipv6_addrs.push(BindAddr::Address(ip))
|
ipv6_addrs.push(BindAddr::IpAddr(ip))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Ok((ipv4_addrs, ipv6_addrs))
|
Ok((ipv4_addrs, ipv6_addrs))
|
||||||
@@ -260,7 +268,7 @@ fn print_listening(args: &Args, print_addrs: &[BindAddr]) -> Result<String> {
|
|||||||
let urls = print_addrs
|
let urls = print_addrs
|
||||||
.iter()
|
.iter()
|
||||||
.map(|bind_addr| match bind_addr {
|
.map(|bind_addr| match bind_addr {
|
||||||
BindAddr::Address(addr) => {
|
BindAddr::IpAddr(addr) => {
|
||||||
let addr = match addr {
|
let addr = match addr {
|
||||||
IpAddr::V4(_) => format!("{}:{}", addr, args.port),
|
IpAddr::V4(_) => format!("{}:{}", addr, args.port),
|
||||||
IpAddr::V6(_) => format!("[{}]:{}", addr, args.port),
|
IpAddr::V6(_) => format!("[{}]:{}", addr, args.port),
|
||||||
@@ -272,7 +280,8 @@ fn print_listening(args: &Args, print_addrs: &[BindAddr]) -> Result<String> {
|
|||||||
};
|
};
|
||||||
format!("{}://{}{}", protocol, addr, args.uri_prefix)
|
format!("{}://{}{}", protocol, addr, args.uri_prefix)
|
||||||
}
|
}
|
||||||
BindAddr::Path(path) => path.display().to_string(),
|
#[cfg(unix)]
|
||||||
|
BindAddr::SocketPath(path) => path.to_string(),
|
||||||
})
|
})
|
||||||
.collect::<Vec<_>>();
|
.collect::<Vec<_>>();
|
||||||
|
|
||||||
|
|||||||
+103
@@ -0,0 +1,103 @@
|
|||||||
|
use crate::{
|
||||||
|
server::{IndexData, PathItem, PathType, MAX_SUBPATHS_COUNT},
|
||||||
|
utils::encode_uri,
|
||||||
|
};
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use chrono::{DateTime, Utc};
|
||||||
|
use xml::escape::escape_str_pcdata;
|
||||||
|
|
||||||
|
pub fn detect_noscript(user_agent: &str) -> bool {
|
||||||
|
[
|
||||||
|
"lynx/", "w3m/", "links ", "elinks/", "curl/", "wget/", "httpie/", "aria2/",
|
||||||
|
]
|
||||||
|
.iter()
|
||||||
|
.any(|v| user_agent.starts_with(v))
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn generate_noscript_html(data: &IndexData) -> Result<String> {
|
||||||
|
let mut html = String::new();
|
||||||
|
|
||||||
|
let title = format!("Index of {}", escape_str_pcdata(&data.href));
|
||||||
|
|
||||||
|
html.push_str("<html>\n");
|
||||||
|
html.push_str("<head>\n");
|
||||||
|
html.push_str(&format!("<title>{title}</title>\n"));
|
||||||
|
html.push_str(
|
||||||
|
r#"<style>
|
||||||
|
td {
|
||||||
|
padding: 0.2rem;
|
||||||
|
text-align: left;
|
||||||
|
}
|
||||||
|
td:nth-child(3) {
|
||||||
|
text-align: right;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
"#,
|
||||||
|
);
|
||||||
|
html.push_str("</head>\n");
|
||||||
|
html.push_str("<body>\n");
|
||||||
|
html.push_str(&format!("<h1>{title}</h1>\n"));
|
||||||
|
html.push_str("<table>\n");
|
||||||
|
html.push_str(" <tbody>\n");
|
||||||
|
html.push_str(&format!(" {}\n", render_parent()));
|
||||||
|
|
||||||
|
for path in &data.paths {
|
||||||
|
html.push_str(&format!(" {}\n", render_path_item(path)));
|
||||||
|
}
|
||||||
|
|
||||||
|
html.push_str(" </tbody>\n");
|
||||||
|
html.push_str("</table>\n");
|
||||||
|
html.push_str("</body>\n");
|
||||||
|
|
||||||
|
Ok(html)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn render_parent() -> String {
|
||||||
|
let value = "../";
|
||||||
|
format!("<tr><td><a href=\"{value}?noscript\">{value}</a></td><td></td><td></td></tr>")
|
||||||
|
}
|
||||||
|
|
||||||
|
fn render_path_item(path: &PathItem) -> String {
|
||||||
|
let mut href = encode_uri(&path.name);
|
||||||
|
let mut name = escape_str_pcdata(&path.name).to_string();
|
||||||
|
if path.path_type.is_dir() {
|
||||||
|
href.push_str("/?noscript");
|
||||||
|
name.push('/');
|
||||||
|
};
|
||||||
|
let mtime = format_mtime(path.mtime).unwrap_or_default();
|
||||||
|
let size = format_size(path.size, path.path_type);
|
||||||
|
|
||||||
|
format!("<tr><td><a href=\"{href}\">{name}</a></td><td>{mtime}</td><td>{size}</td></tr>")
|
||||||
|
}
|
||||||
|
|
||||||
|
fn format_mtime(mtime: u64) -> Option<String> {
|
||||||
|
let datetime = DateTime::<Utc>::from_timestamp_millis(mtime as _)?;
|
||||||
|
Some(datetime.format("%Y-%m-%dT%H:%M:%S.%3fZ").to_string())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn format_size(size: u64, path_type: PathType) -> String {
|
||||||
|
if path_type.is_dir() {
|
||||||
|
let unit = if size == 1 { "item" } else { "items" };
|
||||||
|
let num = match size >= MAX_SUBPATHS_COUNT {
|
||||||
|
true => format!(">{}", MAX_SUBPATHS_COUNT - 1),
|
||||||
|
false => size.to_string(),
|
||||||
|
};
|
||||||
|
format!("{num} {unit}")
|
||||||
|
} else {
|
||||||
|
if size == 0 {
|
||||||
|
return "0 B".to_string();
|
||||||
|
}
|
||||||
|
const UNITS: [&str; 5] = ["B", "KB", "MB", "GB", "TB"];
|
||||||
|
let i = (size as f64).log2() / 10.0;
|
||||||
|
let i = i.floor() as usize;
|
||||||
|
|
||||||
|
if i >= UNITS.len() {
|
||||||
|
// Handle extremely large numbers beyond Terabytes
|
||||||
|
return format!("{:.2} PB", size as f64 / 1024.0f64.powi(5));
|
||||||
|
}
|
||||||
|
|
||||||
|
let size = size as f64 / 1024.0f64.powi(i as i32);
|
||||||
|
format!("{:.2} {}", size, UNITS[i])
|
||||||
|
}
|
||||||
|
}
|
||||||
+402
-208
File diff suppressed because it is too large
Load Diff
+56
-46
@@ -1,17 +1,17 @@
|
|||||||
use anyhow::{anyhow, Context, Result};
|
use anyhow::{anyhow, Context, Result};
|
||||||
use chrono::{DateTime, Utc};
|
use chrono::{DateTime, Utc};
|
||||||
#[cfg(feature = "tls")]
|
#[cfg(feature = "tls")]
|
||||||
use rustls_pki_types::{CertificateDer, PrivateKeyDer};
|
use rustls_pki_types::{pem::PemObject, CertificateDer, PrivateKeyDer};
|
||||||
use std::{
|
use std::{
|
||||||
borrow::Cow,
|
borrow::Cow,
|
||||||
path::Path,
|
path::Path,
|
||||||
time::{Duration, SystemTime, UNIX_EPOCH},
|
time::{Duration, SystemTime, UNIX_EPOCH},
|
||||||
};
|
};
|
||||||
|
|
||||||
pub fn unix_now() -> Result<Duration> {
|
pub fn unix_now() -> Duration {
|
||||||
SystemTime::now()
|
SystemTime::now()
|
||||||
.duration_since(UNIX_EPOCH)
|
.duration_since(UNIX_EPOCH)
|
||||||
.with_context(|| "Invalid system time")
|
.expect("Unable to get unix epoch time")
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn encode_uri(v: &str) -> String {
|
pub fn encode_uri(v: &str) -> String {
|
||||||
@@ -19,7 +19,7 @@ pub fn encode_uri(v: &str) -> String {
|
|||||||
parts.join("/")
|
parts.join("/")
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn decode_uri(v: &str) -> Option<Cow<str>> {
|
pub fn decode_uri(v: &str) -> Option<Cow<'_, str>> {
|
||||||
percent_encoding::percent_decode(v.as_bytes())
|
percent_encoding::percent_decode(v.as_bytes())
|
||||||
.decode_utf8()
|
.decode_utf8()
|
||||||
.ok()
|
.ok()
|
||||||
@@ -62,74 +62,78 @@ pub fn glob(pattern: &str, target: &str) -> bool {
|
|||||||
|
|
||||||
// Load public certificate from file.
|
// Load public certificate from file.
|
||||||
#[cfg(feature = "tls")]
|
#[cfg(feature = "tls")]
|
||||||
pub fn load_certs<T: AsRef<Path>>(filename: T) -> Result<Vec<CertificateDer<'static>>> {
|
pub fn load_certs<T: AsRef<Path>>(file_name: T) -> Result<Vec<CertificateDer<'static>>> {
|
||||||
// Open certificate file.
|
|
||||||
let cert_file = std::fs::File::open(filename.as_ref())
|
|
||||||
.with_context(|| format!("Failed to access `{}`", filename.as_ref().display()))?;
|
|
||||||
let mut reader = std::io::BufReader::new(cert_file);
|
|
||||||
|
|
||||||
// Load and return certificate.
|
|
||||||
let mut certs = vec![];
|
let mut certs = vec![];
|
||||||
for cert in rustls_pemfile::certs(&mut reader) {
|
for cert in CertificateDer::pem_file_iter(file_name.as_ref()).with_context(|| {
|
||||||
let cert = cert.with_context(|| "Failed to load certificate")?;
|
format!(
|
||||||
|
"Failed to load cert file at `{}`",
|
||||||
|
file_name.as_ref().display()
|
||||||
|
)
|
||||||
|
})? {
|
||||||
|
let cert = cert.with_context(|| {
|
||||||
|
format!(
|
||||||
|
"Invalid certificate data in file `{}`",
|
||||||
|
file_name.as_ref().display()
|
||||||
|
)
|
||||||
|
})?;
|
||||||
certs.push(cert)
|
certs.push(cert)
|
||||||
}
|
}
|
||||||
if certs.is_empty() {
|
if certs.is_empty() {
|
||||||
anyhow::bail!("No supported certificate in file");
|
anyhow::bail!(
|
||||||
|
"No supported certificate in file `{}`",
|
||||||
|
file_name.as_ref().display()
|
||||||
|
);
|
||||||
}
|
}
|
||||||
Ok(certs)
|
Ok(certs)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Load private key from file.
|
// Load private key from file.
|
||||||
#[cfg(feature = "tls")]
|
#[cfg(feature = "tls")]
|
||||||
pub fn load_private_key<T: AsRef<Path>>(filename: T) -> Result<PrivateKeyDer<'static>> {
|
pub fn load_private_key<T: AsRef<Path>>(file_name: T) -> Result<PrivateKeyDer<'static>> {
|
||||||
let key_file = std::fs::File::open(filename.as_ref())
|
PrivateKeyDer::from_pem_file(file_name.as_ref()).with_context(|| {
|
||||||
.with_context(|| format!("Failed to access `{}`", filename.as_ref().display()))?;
|
format!(
|
||||||
let mut reader = std::io::BufReader::new(key_file);
|
"Failed to load key file at `{}`",
|
||||||
|
file_name.as_ref().display()
|
||||||
// Load and return a single private key.
|
)
|
||||||
for key in rustls_pemfile::read_all(&mut reader) {
|
})
|
||||||
let key = key.with_context(|| "There was a problem with reading private key")?;
|
|
||||||
match key {
|
|
||||||
rustls_pemfile::Item::Pkcs1Key(key) => return Ok(PrivateKeyDer::Pkcs1(key)),
|
|
||||||
rustls_pemfile::Item::Pkcs8Key(key) => return Ok(PrivateKeyDer::Pkcs8(key)),
|
|
||||||
rustls_pemfile::Item::Sec1Key(key) => return Ok(PrivateKeyDer::Sec1(key)),
|
|
||||||
_ => {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
anyhow::bail!("No supported private key in file");
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn parse_range(range: &str, size: u64) -> Option<(u64, u64)> {
|
pub fn parse_range(range: &str, size: u64) -> Option<Vec<(u64, u64)>> {
|
||||||
let (unit, range) = range.split_once('=')?;
|
let (unit, ranges) = range.split_once('=')?;
|
||||||
if unit != "bytes" || range.contains(',') {
|
if unit != "bytes" {
|
||||||
return None;
|
return None;
|
||||||
}
|
}
|
||||||
let (start, end) = range.split_once('-')?;
|
|
||||||
|
let mut result = Vec::new();
|
||||||
|
for range in ranges.split(',') {
|
||||||
|
let (start, end) = range.trim().split_once('-')?;
|
||||||
if start.is_empty() {
|
if start.is_empty() {
|
||||||
let offset = end.parse::<u64>().ok()?;
|
let offset = end.parse::<u64>().ok()?;
|
||||||
if offset <= size {
|
if offset <= size {
|
||||||
Some((size - offset, size - 1))
|
result.push((size - offset, size - 1));
|
||||||
} else {
|
} else {
|
||||||
None
|
return None;
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
let start = start.parse::<u64>().ok()?;
|
let start = start.parse::<u64>().ok()?;
|
||||||
if start < size {
|
if start < size {
|
||||||
if end.is_empty() {
|
if end.is_empty() {
|
||||||
Some((start, size - 1))
|
result.push((start, size - 1));
|
||||||
} else {
|
} else {
|
||||||
let end = end.parse::<u64>().ok()?;
|
let end = end.parse::<u64>().ok()?;
|
||||||
if end < size {
|
if end < size && start <= end {
|
||||||
Some((start, end))
|
result.push((start, end));
|
||||||
} else {
|
} else {
|
||||||
None
|
return None;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
None
|
return None;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Some(result)
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
@@ -162,13 +166,19 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_parse_range() {
|
fn test_parse_range() {
|
||||||
assert_eq!(parse_range("bytes=0-499", 500), Some((0, 499)));
|
assert_eq!(parse_range("bytes=0-499", 500), Some(vec![(0, 499)]));
|
||||||
assert_eq!(parse_range("bytes=0-", 500), Some((0, 499)));
|
assert_eq!(parse_range("bytes=0-", 500), Some(vec![(0, 499)]));
|
||||||
assert_eq!(parse_range("bytes=299-", 500), Some((299, 499)));
|
assert_eq!(parse_range("bytes=299-", 500), Some(vec![(299, 499)]));
|
||||||
assert_eq!(parse_range("bytes=-500", 500), Some((0, 499)));
|
assert_eq!(parse_range("bytes=-500", 500), Some(vec![(0, 499)]));
|
||||||
assert_eq!(parse_range("bytes=-300", 500), Some((200, 499)));
|
assert_eq!(parse_range("bytes=-300", 500), Some(vec![(200, 499)]));
|
||||||
|
assert_eq!(
|
||||||
|
parse_range("bytes=0-199, 100-399, 400-, -200", 500),
|
||||||
|
Some(vec![(0, 199), (100, 399), (400, 499), (300, 499)])
|
||||||
|
);
|
||||||
assert_eq!(parse_range("bytes=500-", 500), None);
|
assert_eq!(parse_range("bytes=500-", 500), None);
|
||||||
assert_eq!(parse_range("bytes=-501", 500), None);
|
assert_eq!(parse_range("bytes=-501", 500), None);
|
||||||
assert_eq!(parse_range("bytes=0-500", 500), None);
|
assert_eq!(parse_range("bytes=0-500", 500), None);
|
||||||
|
assert_eq!(parse_range("bytes=0-199,", 500), None);
|
||||||
|
assert_eq!(parse_range("bytes=0-199, 500-", 500), None);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+34
-2
@@ -1,7 +1,6 @@
|
|||||||
mod fixtures;
|
mod fixtures;
|
||||||
mod utils;
|
mod utils;
|
||||||
|
|
||||||
use assert_cmd::prelude::*;
|
|
||||||
use assert_fs::fixture::TempDir;
|
use assert_fs::fixture::TempDir;
|
||||||
use fixtures::{port, server, tmpdir, wait_for_port, Error, TestServer, DIR_ASSETS};
|
use fixtures::{port, server, tmpdir, wait_for_port, Error, TestServer, DIR_ASSETS};
|
||||||
use rstest::rstest;
|
use rstest::rstest;
|
||||||
@@ -101,7 +100,7 @@ fn asset_js_with_prefix(
|
|||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn assets_override(tmpdir: TempDir, port: u16) -> Result<(), Error> {
|
fn assets_override(tmpdir: TempDir, port: u16) -> Result<(), Error> {
|
||||||
let mut child = Command::cargo_bin("dufs")?
|
let mut child = Command::new(assert_cmd::cargo::cargo_bin!())
|
||||||
.arg(tmpdir.path())
|
.arg(tmpdir.path())
|
||||||
.arg("-p")
|
.arg("-p")
|
||||||
.arg(port.to_string())
|
.arg(port.to_string())
|
||||||
@@ -124,3 +123,36 @@ fn assets_override(tmpdir: TempDir, port: u16) -> Result<(), Error> {
|
|||||||
child.kill()?;
|
child.kill()?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn assets_override_not_found_page(tmpdir: TempDir, port: u16) -> Result<(), Error> {
|
||||||
|
let not_found_html = "<html><body>custom 404 page</body></html>";
|
||||||
|
std::fs::write(
|
||||||
|
tmpdir.join(format!("{}404.html", DIR_ASSETS)),
|
||||||
|
not_found_html,
|
||||||
|
)?;
|
||||||
|
|
||||||
|
let mut child = Command::new(assert_cmd::cargo::cargo_bin!())
|
||||||
|
.arg(tmpdir.path())
|
||||||
|
.arg("-p")
|
||||||
|
.arg(port.to_string())
|
||||||
|
.arg("--assets")
|
||||||
|
.arg(tmpdir.join(DIR_ASSETS))
|
||||||
|
.stdout(Stdio::piped())
|
||||||
|
.spawn()?;
|
||||||
|
|
||||||
|
wait_for_port(port);
|
||||||
|
|
||||||
|
let url = format!("http://localhost:{port}/missing-path");
|
||||||
|
let resp = reqwest::blocking::get(&url)?;
|
||||||
|
assert_eq!(resp.status(), 404);
|
||||||
|
assert_eq!(resp.text()?, not_found_html);
|
||||||
|
|
||||||
|
let url = format!("http://localhost:{port}/missing-path?noscript");
|
||||||
|
let resp = reqwest::blocking::get(&url)?;
|
||||||
|
assert_eq!(resp.status(), 404);
|
||||||
|
assert_eq!(resp.text()?, "Not Found");
|
||||||
|
|
||||||
|
child.kill()?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|||||||
+76
-9
@@ -57,17 +57,18 @@ fn invalid_auth(
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
const HASHED_PASSWORD_AUTH: &str = "user:$6$gQxZwKyWn/ZmWEA2$4uV7KKMnSUnET2BtWTj/9T5.Jq3h/MdkOlnIl5hdlTxDZ4MZKmJ.kl6C.NL9xnNPqC4lVHC1vuI0E5cLpTJX81@/:rw"; // user:pass
|
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
|
#[case(server(&["--auth", "user:$6$gQxZwKyWn/ZmWEA2$4uV7KKMnSUnET2BtWTj/9T5.Jq3h/MdkOlnIl5hdlTxDZ4MZKmJ.kl6C.NL9xnNPqC4lVHC1vuI0E5cLpTJX81@/:rw", "-A"]), "user", "pass")]
|
||||||
|
#[case(server(&["--auth", "user:$6$YV1J6OHZAAgbzCbS$V55ZEgvJ6JFdz1nLO4AD696PRHAJYhfQf.Gy2HafrCz5itnbgNTtTgfUSqZrt4BJ7FcpRfSt/QZzAan68pido0@/:rw", "-A"]), "user", "pa:ss@1")]
|
||||||
fn auth_hashed_password(
|
fn auth_hashed_password(
|
||||||
#[with(&["--auth", HASHED_PASSWORD_AUTH, "-A"])] server: TestServer,
|
#[case] server: TestServer,
|
||||||
|
#[case] user: &str,
|
||||||
|
#[case] pass: &str,
|
||||||
) -> Result<(), Error> {
|
) -> Result<(), Error> {
|
||||||
let url = format!("{}file1", server.url());
|
let url = format!("{}file1", server.url());
|
||||||
let resp = fetch!(b"PUT", &url).body(b"abc".to_vec()).send()?;
|
let resp = fetch!(b"PUT", &url).body(b"abc".to_vec()).send()?;
|
||||||
assert_eq!(resp.status(), 401);
|
assert_eq!(resp.status(), 401);
|
||||||
if let Err(err) =
|
if let Err(err) = send_with_digest_auth(fetch!(b"PUT", &url).body(b"abc".to_vec()), user, pass)
|
||||||
send_with_digest_auth(fetch!(b"PUT", &url).body(b"abc".to_vec()), "user", "pass")
|
|
||||||
{
|
{
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
err.to_string(),
|
err.to_string(),
|
||||||
@@ -76,7 +77,7 @@ fn auth_hashed_password(
|
|||||||
}
|
}
|
||||||
let resp = fetch!(b"PUT", &url)
|
let resp = fetch!(b"PUT", &url)
|
||||||
.body(b"abc".to_vec())
|
.body(b"abc".to_vec())
|
||||||
.basic_auth("user", Some("pass"))
|
.basic_auth(user, Some(pass))
|
||||||
.send()?;
|
.send()?;
|
||||||
assert_eq!(resp.status(), 201);
|
assert_eq!(resp.status(), 201);
|
||||||
Ok(())
|
Ok(())
|
||||||
@@ -114,11 +115,39 @@ fn auth_skip_on_options_method(
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn auth_skip_if_no_auth_user(server: TestServer) -> Result<(), Error> {
|
||||||
|
let url = format!("{}index.html", server.url());
|
||||||
|
let resp = fetch!(b"GET", &url)
|
||||||
|
.basic_auth("user", Some("pass"))
|
||||||
|
.send()?;
|
||||||
|
assert_eq!(resp.status(), 200);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn auth_no_skip_if_anonymous(
|
||||||
|
#[with(&["--auth", "@/:ro"])] server: TestServer,
|
||||||
|
) -> Result<(), Error> {
|
||||||
|
let url = format!("{}index.html", server.url());
|
||||||
|
let resp = fetch!(b"GET", &url)
|
||||||
|
.basic_auth("user", Some("pass"))
|
||||||
|
.send()?;
|
||||||
|
assert_eq!(resp.status(), 401);
|
||||||
|
let resp = fetch!(b"GET", &url).send()?;
|
||||||
|
assert_eq!(resp.status(), 200);
|
||||||
|
let resp = fetch!(b"DELETE", &url)
|
||||||
|
.basic_auth("user", Some("pass"))
|
||||||
|
.send()?;
|
||||||
|
assert_eq!(resp.status(), 401);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn auth_check(
|
fn auth_check(
|
||||||
#[with(&["--auth", "user:pass@/:rw", "--auth", "user2:pass2@/", "-A"])] server: TestServer,
|
#[with(&["--auth", "user:pass@/:rw", "--auth", "user2:pass2@/", "-A"])] server: TestServer,
|
||||||
) -> Result<(), Error> {
|
) -> Result<(), Error> {
|
||||||
let url = format!("{}index.html", server.url());
|
let url = format!("{}", server.url());
|
||||||
let resp = fetch!(b"CHECKAUTH", &url).send()?;
|
let resp = fetch!(b"CHECKAUTH", &url).send()?;
|
||||||
assert_eq!(resp.status(), 401);
|
assert_eq!(resp.status(), 401);
|
||||||
let resp = send_with_digest_auth(fetch!(b"CHECKAUTH", &url), "user", "pass")?;
|
let resp = send_with_digest_auth(fetch!(b"CHECKAUTH", &url), "user", "pass")?;
|
||||||
@@ -132,7 +161,7 @@ fn auth_check(
|
|||||||
fn auth_check2(
|
fn auth_check2(
|
||||||
#[with(&["--auth", "user:pass@/:rw|user2:pass2@/", "-A"])] server: TestServer,
|
#[with(&["--auth", "user:pass@/:rw|user2:pass2@/", "-A"])] server: TestServer,
|
||||||
) -> Result<(), Error> {
|
) -> Result<(), Error> {
|
||||||
let url = format!("{}index.html", server.url());
|
let url = format!("{}", server.url());
|
||||||
let resp = fetch!(b"CHECKAUTH", &url).send()?;
|
let resp = fetch!(b"CHECKAUTH", &url).send()?;
|
||||||
assert_eq!(resp.status(), 401);
|
assert_eq!(resp.status(), 401);
|
||||||
let resp = send_with_digest_auth(fetch!(b"CHECKAUTH", &url), "user", "pass")?;
|
let resp = send_with_digest_auth(fetch!(b"CHECKAUTH", &url), "user", "pass")?;
|
||||||
@@ -142,6 +171,18 @@ fn auth_check2(
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn auth_check3(
|
||||||
|
#[with(&["--auth", "user:pass@/:rw", "--auth", "@/dir1:rw", "-A"])] server: TestServer,
|
||||||
|
) -> Result<(), Error> {
|
||||||
|
let url = format!("{}dir1/", server.url());
|
||||||
|
let resp = fetch!(b"CHECKAUTH", &url).send()?;
|
||||||
|
assert_eq!(resp.status(), 200);
|
||||||
|
let resp = fetch!(b"CHECKAUTH", format!("{url}?login")).send()?;
|
||||||
|
assert_eq!(resp.status(), 401);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn auth_logout(
|
fn auth_logout(
|
||||||
#[with(&["--auth", "user:pass@/:rw", "-A"])] server: TestServer,
|
#[with(&["--auth", "user:pass@/:rw", "-A"])] server: TestServer,
|
||||||
@@ -332,9 +373,35 @@ fn auth_precedence(
|
|||||||
let resp = send_with_digest_auth(fetch!(b"PUT", &url).body(b"abc".to_vec()), "user", "pass")?;
|
let resp = send_with_digest_auth(fetch!(b"PUT", &url).body(b"abc".to_vec()), "user", "pass")?;
|
||||||
assert_eq!(resp.status(), 403);
|
assert_eq!(resp.status(), 403);
|
||||||
|
|
||||||
let url = format!("{}dir1/file1", server.url());
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn auth_anonymous_no_precedence(
|
||||||
|
#[with(&["--auth", "user:pass@/:rw", "-a", "@/dir1", "-A"])] server: TestServer,
|
||||||
|
) -> Result<(), Error> {
|
||||||
|
let url = format!("{}dir1/test.txt", server.url());
|
||||||
|
let resp = fetch!(b"PUT", &url).body(b"abc".to_vec()).send()?;
|
||||||
|
assert_eq!(resp.status(), 401);
|
||||||
|
|
||||||
let resp = send_with_digest_auth(fetch!(b"PUT", &url).body(b"abc".to_vec()), "user", "pass")?;
|
let resp = send_with_digest_auth(fetch!(b"PUT", &url).body(b"abc".to_vec()), "user", "pass")?;
|
||||||
assert_eq!(resp.status(), 201);
|
assert_eq!(resp.status(), 201);
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn token_auth(#[with(&["-a", "user:pass@/"])] server: TestServer) -> Result<(), Error> {
|
||||||
|
let url = format!("{}index.html", server.url());
|
||||||
|
let resp = fetch!(b"GET", &url).send()?;
|
||||||
|
assert_eq!(resp.status(), 401);
|
||||||
|
let url = format!("{}index.html?tokengen", server.url());
|
||||||
|
let resp = fetch!(b"GET", &url)
|
||||||
|
.basic_auth("user", Some("pass"))
|
||||||
|
.send()?;
|
||||||
|
let token = resp.text()?;
|
||||||
|
let url = format!("{}index.html?token={token}", server.url());
|
||||||
|
let resp = fetch!(b"GET", &url).send()?;
|
||||||
|
assert_eq!(resp.status(), 200);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|||||||
+2
-2
@@ -12,7 +12,7 @@ use std::process::{Command, Stdio};
|
|||||||
#[rstest]
|
#[rstest]
|
||||||
#[case(&["-b", "20.205.243.166"])]
|
#[case(&["-b", "20.205.243.166"])]
|
||||||
fn bind_fails(tmpdir: TempDir, port: u16, #[case] args: &[&str]) -> Result<(), Error> {
|
fn bind_fails(tmpdir: TempDir, port: u16, #[case] args: &[&str]) -> Result<(), Error> {
|
||||||
Command::cargo_bin("dufs")?
|
Command::new(assert_cmd::cargo::cargo_bin!())
|
||||||
.arg(tmpdir.path())
|
.arg(tmpdir.path())
|
||||||
.arg("-p")
|
.arg("-p")
|
||||||
.arg(port.to_string())
|
.arg(port.to_string())
|
||||||
@@ -49,7 +49,7 @@ fn bind_ipv4_ipv6(
|
|||||||
#[case(&[] as &[&str])]
|
#[case(&[] as &[&str])]
|
||||||
#[case(&["--path-prefix", "/prefix"])]
|
#[case(&["--path-prefix", "/prefix"])]
|
||||||
fn validate_printed_urls(tmpdir: TempDir, port: u16, #[case] args: &[&str]) -> Result<(), Error> {
|
fn validate_printed_urls(tmpdir: TempDir, port: u16, #[case] args: &[&str]) -> Result<(), Error> {
|
||||||
let mut child = Command::cargo_bin("dufs")?
|
let mut child = Command::new(assert_cmd::cargo::cargo_bin!())
|
||||||
.arg(tmpdir.path())
|
.arg(tmpdir.path())
|
||||||
.arg("-p")
|
.arg("-p")
|
||||||
.arg(port.to_string())
|
.arg(port.to_string())
|
||||||
|
|||||||
+1
-1
@@ -49,7 +49,7 @@ fn same_etag(etag: &str) -> String {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn different_etag(etag: &str) -> String {
|
fn different_etag(etag: &str) -> String {
|
||||||
format!("{}1234", etag)
|
format!("{etag}1234")
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
|
|||||||
+5
-2
@@ -11,7 +11,10 @@ use std::process::Command;
|
|||||||
#[test]
|
#[test]
|
||||||
/// Show help and exit.
|
/// Show help and exit.
|
||||||
fn help_shows() -> Result<(), Error> {
|
fn help_shows() -> Result<(), Error> {
|
||||||
Command::cargo_bin("dufs")?.arg("-h").assert().success();
|
Command::new(assert_cmd::cargo::cargo_bin!())
|
||||||
|
.arg("-h")
|
||||||
|
.assert()
|
||||||
|
.success();
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
@@ -21,7 +24,7 @@ fn help_shows() -> Result<(), Error> {
|
|||||||
fn print_completions() -> Result<(), Error> {
|
fn print_completions() -> Result<(), Error> {
|
||||||
// let shell_enums = EnumValueParser::<Shell>::new();
|
// let shell_enums = EnumValueParser::<Shell>::new();
|
||||||
for shell in Shell::value_variants() {
|
for shell in Shell::value_variants() {
|
||||||
Command::cargo_bin("dufs")?
|
Command::new(assert_cmd::cargo::cargo_bin!())
|
||||||
.arg("--completions")
|
.arg("--completions")
|
||||||
.arg(shell.to_string())
|
.arg(shell.to_string())
|
||||||
.assert()
|
.assert()
|
||||||
|
|||||||
+1
-2
@@ -2,7 +2,6 @@ mod digest_auth_util;
|
|||||||
mod fixtures;
|
mod fixtures;
|
||||||
mod utils;
|
mod utils;
|
||||||
|
|
||||||
use assert_cmd::prelude::*;
|
|
||||||
use assert_fs::TempDir;
|
use assert_fs::TempDir;
|
||||||
use digest_auth_util::send_with_digest_auth;
|
use digest_auth_util::send_with_digest_auth;
|
||||||
use fixtures::{port, tmpdir, wait_for_port, Error};
|
use fixtures::{port, tmpdir, wait_for_port, Error};
|
||||||
@@ -13,7 +12,7 @@ use std::process::{Command, Stdio};
|
|||||||
#[rstest]
|
#[rstest]
|
||||||
fn use_config_file(tmpdir: TempDir, port: u16) -> Result<(), Error> {
|
fn use_config_file(tmpdir: TempDir, port: u16) -> Result<(), Error> {
|
||||||
let config_path = get_config_path().display().to_string();
|
let config_path = get_config_path().display().to_string();
|
||||||
let mut child = Command::cargo_bin("dufs")?
|
let mut child = Command::new(assert_cmd::cargo::cargo_bin!())
|
||||||
.arg(tmpdir.path())
|
.arg(tmpdir.path())
|
||||||
.arg("-p")
|
.arg("-p")
|
||||||
.arg(port.to_string())
|
.arg(port.to_string())
|
||||||
|
|||||||
+4
-6
@@ -1,4 +1,3 @@
|
|||||||
use assert_cmd::prelude::*;
|
|
||||||
use assert_fs::fixture::TempDir;
|
use assert_fs::fixture::TempDir;
|
||||||
use assert_fs::prelude::*;
|
use assert_fs::prelude::*;
|
||||||
use port_check::free_local_port;
|
use port_check::free_local_port;
|
||||||
@@ -129,8 +128,7 @@ where
|
|||||||
{
|
{
|
||||||
let port = port();
|
let port = port();
|
||||||
let tmpdir = tmpdir();
|
let tmpdir = tmpdir();
|
||||||
let child = Command::cargo_bin("dufs")
|
let child = Command::new(assert_cmd::cargo::cargo_bin!())
|
||||||
.expect("Couldn't find test binary")
|
|
||||||
.arg(tmpdir.path())
|
.arg(tmpdir.path())
|
||||||
.arg("-p")
|
.arg("-p")
|
||||||
.arg(port.to_string())
|
.arg(port.to_string())
|
||||||
@@ -146,14 +144,14 @@ where
|
|||||||
TestServer::new(port, tmpdir, child, is_tls)
|
TestServer::new(port, tmpdir, child, is_tls)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Wait a max of 1s for the port to become available.
|
/// Wait a max of 2s for the port to become available.
|
||||||
pub fn wait_for_port(port: u16) {
|
pub fn wait_for_port(port: u16) {
|
||||||
let start_wait = Instant::now();
|
let start_wait = Instant::now();
|
||||||
|
|
||||||
while !port_check::is_port_reachable(format!("localhost:{port}")) {
|
while !port_check::is_port_reachable(format!("localhost:{port}")) {
|
||||||
sleep(Duration::from_millis(100));
|
sleep(Duration::from_millis(250));
|
||||||
|
|
||||||
if start_wait.elapsed().as_secs() > 1 {
|
if start_wait.elapsed().as_secs() > 2 {
|
||||||
panic!("timeout waiting for port {port}");
|
panic!("timeout waiting for port {port}");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,31 @@
|
|||||||
|
mod fixtures;
|
||||||
|
mod utils;
|
||||||
|
|
||||||
|
use fixtures::{server, Error, TestServer};
|
||||||
|
use rstest::rstest;
|
||||||
|
|
||||||
|
const HEALTH_CHECK_PATH: &str = "__dufs__/health";
|
||||||
|
const HEALTH_CHECK_RESPONSE: &str = r#"{"status":"OK"}"#;
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn normal_health(server: TestServer) -> Result<(), Error> {
|
||||||
|
let resp = reqwest::blocking::get(format!("{}{HEALTH_CHECK_PATH}", server.url()))?;
|
||||||
|
assert_eq!(resp.text()?, HEALTH_CHECK_RESPONSE);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn auth_health(
|
||||||
|
#[with(&["--auth", "user:pass@/:rw", "-A"])] server: TestServer,
|
||||||
|
) -> Result<(), Error> {
|
||||||
|
let resp = reqwest::blocking::get(format!("{}{HEALTH_CHECK_PATH}", server.url()))?;
|
||||||
|
assert_eq!(resp.text()?, HEALTH_CHECK_RESPONSE);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn path_prefix_health(#[with(&["--path-prefix", "xyz"])] server: TestServer) -> Result<(), Error> {
|
||||||
|
let resp = reqwest::blocking::get(format!("{}xyz/{HEALTH_CHECK_PATH}", server.url()))?;
|
||||||
|
assert_eq!(resp.text()?, HEALTH_CHECK_RESPONSE);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
+37
-1
@@ -82,6 +82,19 @@ fn get_dir_simple(#[with(&["-A"])] server: TestServer) -> Result<(), Error> {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn get_dir_noscript(#[with(&["-A"])] server: TestServer) -> Result<(), Error> {
|
||||||
|
let resp = reqwest::blocking::get(format!("{}?noscript", server.url()))?;
|
||||||
|
assert_eq!(resp.status(), 200);
|
||||||
|
assert_eq!(
|
||||||
|
resp.headers().get("content-type").unwrap(),
|
||||||
|
"text/html; charset=utf-8"
|
||||||
|
);
|
||||||
|
let text = resp.text().unwrap();
|
||||||
|
assert!(text.contains(r#"<td><a href="index.html">index.html</a></td>"#));
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn head_dir_zip(#[with(&["-A"])] server: TestServer) -> Result<(), Error> {
|
fn head_dir_zip(#[with(&["-A"])] server: TestServer) -> Result<(), Error> {
|
||||||
let resp = fetch!(b"HEAD", format!("{}?zip", server.url())).send()?;
|
let resp = fetch!(b"HEAD", format!("{}?zip", server.url())).send()?;
|
||||||
@@ -172,6 +185,22 @@ fn get_file(server: TestServer) -> Result<(), Error> {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn get_file_json(server: TestServer) -> Result<(), Error> {
|
||||||
|
let resp = reqwest::blocking::get(format!("{}index.html?json", server.url()))?;
|
||||||
|
assert_eq!(resp.status(), 200);
|
||||||
|
assert_eq!(
|
||||||
|
resp.headers().get("content-type").unwrap(),
|
||||||
|
"application/json"
|
||||||
|
);
|
||||||
|
let json: Value = serde_json::from_str(&resp.text()?).unwrap();
|
||||||
|
assert_eq!(json["name"], "index.html");
|
||||||
|
assert_eq!(json["path_type"], "File");
|
||||||
|
assert!(json["size"].as_u64().is_some());
|
||||||
|
assert!(json["mtime"].as_u64().is_some());
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn head_file(server: TestServer) -> Result<(), Error> {
|
fn head_file(server: TestServer) -> Result<(), Error> {
|
||||||
let resp = fetch!(b"HEAD", format!("{}index.html", server.url())).send()?;
|
let resp = fetch!(b"HEAD", format!("{}index.html", server.url())).send()?;
|
||||||
@@ -190,7 +219,7 @@ fn head_file(server: TestServer) -> Result<(), Error> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn hash_file(server: TestServer) -> Result<(), Error> {
|
fn hash_file(#[with(&["--allow-hash"])] server: TestServer) -> Result<(), Error> {
|
||||||
let resp = reqwest::blocking::get(format!("{}index.html?hash", server.url()))?;
|
let resp = reqwest::blocking::get(format!("{}index.html?hash", server.url()))?;
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
resp.headers().get("content-type").unwrap(),
|
resp.headers().get("content-type").unwrap(),
|
||||||
@@ -204,6 +233,13 @@ fn hash_file(server: TestServer) -> Result<(), Error> {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn no_hash_file(server: TestServer) -> Result<(), Error> {
|
||||||
|
let resp = reqwest::blocking::get(format!("{}index.html?hash", server.url()))?;
|
||||||
|
assert_eq!(resp.status(), 403);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn get_file_404(server: TestServer) -> Result<(), Error> {
|
fn get_file_404(server: TestServer) -> Result<(), Error> {
|
||||||
let resp = reqwest::blocking::get(format!("{}404", server.url()))?;
|
let resp = reqwest::blocking::get(format!("{}404", server.url()))?;
|
||||||
|
|||||||
@@ -5,7 +5,6 @@ mod utils;
|
|||||||
use digest_auth_util::send_with_digest_auth;
|
use digest_auth_util::send_with_digest_auth;
|
||||||
use fixtures::{port, tmpdir, wait_for_port, Error};
|
use fixtures::{port, tmpdir, wait_for_port, Error};
|
||||||
|
|
||||||
use assert_cmd::prelude::*;
|
|
||||||
use assert_fs::fixture::TempDir;
|
use assert_fs::fixture::TempDir;
|
||||||
use rstest::rstest;
|
use rstest::rstest;
|
||||||
use std::io::Read;
|
use std::io::Read;
|
||||||
@@ -20,7 +19,7 @@ fn log_remote_user(
|
|||||||
#[case] args: &[&str],
|
#[case] args: &[&str],
|
||||||
#[case] is_basic: bool,
|
#[case] is_basic: bool,
|
||||||
) -> Result<(), Error> {
|
) -> Result<(), Error> {
|
||||||
let mut child = Command::cargo_bin("dufs")?
|
let mut child = Command::new(assert_cmd::cargo::cargo_bin!())
|
||||||
.arg(tmpdir.path())
|
.arg(tmpdir.path())
|
||||||
.arg("-p")
|
.arg("-p")
|
||||||
.arg(port.to_string())
|
.arg(port.to_string())
|
||||||
@@ -55,7 +54,7 @@ fn log_remote_user(
|
|||||||
#[rstest]
|
#[rstest]
|
||||||
#[case(&["--log-format", ""])]
|
#[case(&["--log-format", ""])]
|
||||||
fn no_log(tmpdir: TempDir, port: u16, #[case] args: &[&str]) -> Result<(), Error> {
|
fn no_log(tmpdir: TempDir, port: u16, #[case] args: &[&str]) -> Result<(), Error> {
|
||||||
let mut child = Command::cargo_bin("dufs")?
|
let mut child = Command::new(assert_cmd::cargo::cargo_bin!())
|
||||||
.arg(tmpdir.path())
|
.arg(tmpdir.path())
|
||||||
.arg("-p")
|
.arg("-p")
|
||||||
.arg(port.to_string())
|
.arg(port.to_string())
|
||||||
|
|||||||
+88
-1
@@ -2,7 +2,7 @@ mod fixtures;
|
|||||||
mod utils;
|
mod utils;
|
||||||
|
|
||||||
use fixtures::{server, Error, TestServer};
|
use fixtures::{server, Error, TestServer};
|
||||||
use reqwest::header::HeaderValue;
|
use reqwest::header::{HeaderMap, HeaderName, HeaderValue};
|
||||||
use rstest::rstest;
|
use rstest::rstest;
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
@@ -39,3 +39,90 @@ fn get_file_range_invalid(server: TestServer) -> Result<(), Error> {
|
|||||||
assert_eq!(resp.headers().get("content-range").unwrap(), "bytes */18");
|
assert_eq!(resp.headers().get("content-range").unwrap(), "bytes */18");
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn parse_multipart_body<'a>(body: &'a str, boundary: &str) -> Vec<(HeaderMap, &'a str)> {
|
||||||
|
body.split(&format!("--{boundary}"))
|
||||||
|
.filter(|part| !part.is_empty() && *part != "--\r\n")
|
||||||
|
.map(|part| {
|
||||||
|
let (head, body) = part.trim_ascii().split_once("\r\n\r\n").unwrap();
|
||||||
|
let headers = head
|
||||||
|
.split("\r\n")
|
||||||
|
.fold(HeaderMap::new(), |mut headers, header| {
|
||||||
|
let (key, value) = header.split_once(":").unwrap();
|
||||||
|
let key = HeaderName::from_bytes(key.as_bytes()).unwrap();
|
||||||
|
let value = HeaderValue::from_str(value.trim_ascii_start()).unwrap();
|
||||||
|
headers.insert(key, value);
|
||||||
|
headers
|
||||||
|
});
|
||||||
|
(headers, body)
|
||||||
|
})
|
||||||
|
.collect()
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn get_file_multipart_range(server: TestServer) -> Result<(), Error> {
|
||||||
|
let resp = fetch!(b"GET", format!("{}index.html", server.url()))
|
||||||
|
.header("range", HeaderValue::from_static("bytes=0-11, 6-17"))
|
||||||
|
.send()?;
|
||||||
|
assert_eq!(resp.status(), 206);
|
||||||
|
assert_eq!(resp.headers().get("accept-ranges").unwrap(), "bytes");
|
||||||
|
|
||||||
|
let content_type = resp
|
||||||
|
.headers()
|
||||||
|
.get("content-type")
|
||||||
|
.unwrap()
|
||||||
|
.to_str()?
|
||||||
|
.to_string();
|
||||||
|
assert!(content_type.starts_with("multipart/byteranges; boundary="));
|
||||||
|
|
||||||
|
let boundary = content_type.split_once('=').unwrap().1.trim_ascii_start();
|
||||||
|
assert!(!boundary.is_empty());
|
||||||
|
|
||||||
|
let body = resp.text()?;
|
||||||
|
let parts = parse_multipart_body(&body, boundary);
|
||||||
|
assert_eq!(parts.len(), 2);
|
||||||
|
|
||||||
|
let (headers, body) = &parts[0];
|
||||||
|
assert_eq!(headers.get("content-range").unwrap(), "bytes 0-11/18");
|
||||||
|
assert_eq!(*body, "This is inde");
|
||||||
|
|
||||||
|
let (headers, body) = &parts[1];
|
||||||
|
assert_eq!(headers.get("content-range").unwrap(), "bytes 6-17/18");
|
||||||
|
assert_eq!(*body, "s index.html");
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn get_file_multipart_range_invalid(server: TestServer) -> Result<(), Error> {
|
||||||
|
let resp = fetch!(b"GET", format!("{}index.html", server.url()))
|
||||||
|
.header("range", HeaderValue::from_static("bytes=0-6, 20-30"))
|
||||||
|
.send()?;
|
||||||
|
assert_eq!(resp.status(), 416);
|
||||||
|
assert_eq!(resp.headers().get("content-range").unwrap(), "bytes */18");
|
||||||
|
assert_eq!(resp.headers().get("accept-ranges").unwrap(), "bytes");
|
||||||
|
assert_eq!(resp.headers().get("content-length").unwrap(), "0");
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn get_file_range_reversed(server: TestServer) -> Result<(), Error> {
|
||||||
|
let resp = fetch!(b"GET", format!("{}index.html", server.url()))
|
||||||
|
.header("range", HeaderValue::from_static("bytes=10-1"))
|
||||||
|
.send()?;
|
||||||
|
assert_eq!(resp.status(), 416);
|
||||||
|
assert_eq!(resp.headers().get("content-range").unwrap(), "bytes */18");
|
||||||
|
assert_eq!(resp.headers().get("accept-ranges").unwrap(), "bytes");
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn get_file_multipart_range_reversed(server: TestServer) -> Result<(), Error> {
|
||||||
|
let resp = fetch!(b"GET", format!("{}index.html", server.url()))
|
||||||
|
.header("range", HeaderValue::from_static("bytes=10-1,20-2"))
|
||||||
|
.send()?;
|
||||||
|
assert_eq!(resp.status(), 416);
|
||||||
|
assert_eq!(resp.headers().get("content-range").unwrap(), "bytes */18");
|
||||||
|
assert_eq!(resp.headers().get("accept-ranges").unwrap(), "bytes");
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|||||||
@@ -3,7 +3,6 @@
|
|||||||
mod fixtures;
|
mod fixtures;
|
||||||
mod utils;
|
mod utils;
|
||||||
|
|
||||||
use assert_cmd::prelude::*;
|
|
||||||
use assert_fs::fixture::TempDir;
|
use assert_fs::fixture::TempDir;
|
||||||
use fixtures::{port, tmpdir, wait_for_port, Error};
|
use fixtures::{port, tmpdir, wait_for_port, Error};
|
||||||
use rstest::rstest;
|
use rstest::rstest;
|
||||||
@@ -12,7 +11,7 @@ use std::process::{Command, Stdio};
|
|||||||
#[rstest]
|
#[rstest]
|
||||||
#[case("index.html")]
|
#[case("index.html")]
|
||||||
fn single_file(tmpdir: TempDir, port: u16, #[case] file: &str) -> Result<(), Error> {
|
fn single_file(tmpdir: TempDir, port: u16, #[case] file: &str) -> Result<(), Error> {
|
||||||
let mut child = Command::cargo_bin("dufs")?
|
let mut child = Command::new(assert_cmd::cargo::cargo_bin!())
|
||||||
.arg(tmpdir.path().join(file))
|
.arg(tmpdir.path().join(file))
|
||||||
.arg("-p")
|
.arg("-p")
|
||||||
.arg(port.to_string())
|
.arg(port.to_string())
|
||||||
@@ -35,7 +34,7 @@ fn single_file(tmpdir: TempDir, port: u16, #[case] file: &str) -> Result<(), Err
|
|||||||
#[rstest]
|
#[rstest]
|
||||||
#[case("index.html")]
|
#[case("index.html")]
|
||||||
fn path_prefix_single_file(tmpdir: TempDir, port: u16, #[case] file: &str) -> Result<(), Error> {
|
fn path_prefix_single_file(tmpdir: TempDir, port: u16, #[case] file: &str) -> Result<(), Error> {
|
||||||
let mut child = Command::cargo_bin("dufs")?
|
let mut child = Command::new(assert_cmd::cargo::cargo_bin!())
|
||||||
.arg(tmpdir.path().join(file))
|
.arg(tmpdir.path().join(file))
|
||||||
.arg("-p")
|
.arg("-p")
|
||||||
.arg(port.to_string())
|
.arg(port.to_string())
|
||||||
|
|||||||
+5
-6
@@ -1,7 +1,6 @@
|
|||||||
mod fixtures;
|
mod fixtures;
|
||||||
mod utils;
|
mod utils;
|
||||||
|
|
||||||
use assert_cmd::Command;
|
|
||||||
use fixtures::{server, Error, TestServer};
|
use fixtures::{server, Error, TestServer};
|
||||||
use predicates::str::contains;
|
use predicates::str::contains;
|
||||||
use reqwest::blocking::ClientBuilder;
|
use reqwest::blocking::ClientBuilder;
|
||||||
@@ -25,7 +24,7 @@ use crate::fixtures::port;
|
|||||||
]))]
|
]))]
|
||||||
fn tls_works(#[case] server: TestServer) -> Result<(), Error> {
|
fn tls_works(#[case] server: TestServer) -> Result<(), Error> {
|
||||||
let client = ClientBuilder::new()
|
let client = ClientBuilder::new()
|
||||||
.danger_accept_invalid_certs(true)
|
.tls_danger_accept_invalid_certs(true)
|
||||||
.build()?;
|
.build()?;
|
||||||
let resp = client.get(server.url()).send()?.error_for_status()?;
|
let resp = client.get(server.url()).send()?.error_for_status()?;
|
||||||
assert_resp_paths!(resp);
|
assert_resp_paths!(resp);
|
||||||
@@ -36,7 +35,7 @@ fn tls_works(#[case] server: TestServer) -> Result<(), Error> {
|
|||||||
#[rstest]
|
#[rstest]
|
||||||
fn wrong_path_cert() -> Result<(), Error> {
|
fn wrong_path_cert() -> Result<(), Error> {
|
||||||
let port = port().to_string();
|
let port = port().to_string();
|
||||||
Command::cargo_bin("dufs")?
|
assert_cmd::cargo::cargo_bin_cmd!()
|
||||||
.args([
|
.args([
|
||||||
"--tls-cert",
|
"--tls-cert",
|
||||||
"wrong",
|
"wrong",
|
||||||
@@ -47,7 +46,7 @@ fn wrong_path_cert() -> Result<(), Error> {
|
|||||||
])
|
])
|
||||||
.assert()
|
.assert()
|
||||||
.failure()
|
.failure()
|
||||||
.stderr(contains("Failed to access `wrong`"));
|
.stderr(contains("Failed to load cert file at `wrong`"));
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
@@ -56,7 +55,7 @@ fn wrong_path_cert() -> Result<(), Error> {
|
|||||||
#[rstest]
|
#[rstest]
|
||||||
fn wrong_path_key() -> Result<(), Error> {
|
fn wrong_path_key() -> Result<(), Error> {
|
||||||
let port = port().to_string();
|
let port = port().to_string();
|
||||||
Command::cargo_bin("dufs")?
|
assert_cmd::cargo::cargo_bin_cmd!()
|
||||||
.args([
|
.args([
|
||||||
"--tls-cert",
|
"--tls-cert",
|
||||||
"tests/data/cert.pem",
|
"tests/data/cert.pem",
|
||||||
@@ -67,7 +66,7 @@ fn wrong_path_key() -> Result<(), Error> {
|
|||||||
])
|
])
|
||||||
.assert()
|
.assert()
|
||||||
.failure()
|
.failure()
|
||||||
.stderr(contains("Failed to access `wrong`"));
|
.stderr(contains("Failed to load key file at `wrong`"));
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user