rustfs-uring

Cancel-safe async io_uring read backend for RustFS storage.

This crate is the io_uring integration that RustFS's read path is built on. It started as the Spike 0 cancel-safety prototype (rustfs/backlog#894) and was hardened per the #1048/#1051 audit. It lives in its own repository so it can be verified in isolation — with a real io_uring CI leg that the main rustfs/rustfs workspace cannot run — before being wired into the storage layer.

Status: read path only, Linux only. The read path is wired into rustfs/rustfs behind a runtime probe and is off by default (RUSTFS_IO_URING_READ_ENABLE). See CHANGELOG.md for what has landed and the design notes for the invariants.

[target.'cfg(target_os = "linux")'.dependencies]
rustfs-uring = "0.1.0"

The ownership model it enforces

When a caller drops the future of an in-flight read (EC quorum reached, timeout, disconnect), the kernel may still write into the read buffer at any point until the CQE. Freeing the buffer at future-drop is a use-after-free. This crate proves and enforces the invariants any production io_uring integration must follow:

• The buffer and the file handle are owned by the driver's pending (orphan) table from SQE submission until the CQE — never by the caller's future.
• Dropping the future abandons only the result; reclamation always happens at the CQE (optionally accelerated by IORING_OP_ASYNC_CANCEL).
• Shutdown drains in-flight ops to zero (with a bounded escape hatch for hung disks) before the ring is unmapped.
• A driver-thread panic aborts before freeing in-flight buffers (leak over UAF); backpressure permits are released at the CQE, not at future drop; short reads on positioned reads are resubmitted to satisfy the whole-range contract; the probe file is opened via O_TMPFILE.

Each invariant holds per shard (see below), because a shard is an independent instance of the same driver.

The full invariant list, the corrected fd-reuse mechanism, and the design constraints are in the design notes.

Usage

use std::fs::File;
use std::sync::Arc;
use rustfs_uring::UringDriver;

# async fn demo() -> std::io::Result<()> {
    // Probe a real IORING_OP_READ before accepting work. A restricted environment
    // (seccomp/gVisor/old kernel) returns a ProbeFailure whose
    // `is_expected_restriction()` tells you to degrade to the std backend quietly.
    let driver = UringDriver::probe_and_start(64).expect("io_uring available");

    let file = Arc::new(File::open("/data/object")?);

    // Positioned read (pread semantics, whole-range: short reads are resubmitted).
    let bytes = driver.read_at(Arc::clone(&file), 0, 65536).await?;

    // Dropping the returned future before it completes is safe — the driver owns
    // the buffer until the CQE.

    let snapshot = driver.shutdown();
    assert_eq!(snapshot.delivered + snapshot.orphan_reclaimed, snapshot.submitted);
    # Ok(())
    #
}

Sharded rings

A buffered read that hits the page cache completes inline inside io_uring_enter, so the thread driving a ring performs that read's memcpy. One ring is therefore capped at a single core's memory bandwidth (~5 GB/s measured). Give a disk several rings when its reads hit the cache:

# use rustfs_uring::UringDriver;
// Four independent rings, each with `entries` SQ slots and its own driver thread.
// In-flight is capped per shard, so the driver admits up to `shards * entries` reads.
let driver = UringDriver::probe_and_start_sharded(64, 4)?;
# Ok::<_, rustfs_uring::ProbeFailure>(())

probe_and_start(entries) is exactly probe_and_start_sharded(entries, 1), so nobody grows threads by upgrading. Rings stay per-disk: a stalled disk cannot starve another disk's rings.

O_DIRECT

Open the fd with O_DIRECT, pass the device's logical block size, and let the driver do the alignment. offset and len need not be aligned — it reads a block-aligned superset into a block-aligned buffer and hands back exactly the range you asked for. Padding, the bytes before the range, and the block-aligned tail never escape.

# use rustfs_uring::UringDriver;
# use std::{fs::File, sync::Arc};
# async fn demo(driver: &UringDriver, file: Arc<File>) -> std::io::Result<()> {
// `file` was opened with O_DIRECT; 4096 is the probed logical block size.
let bytes = driver.read_at_direct(file, 8_191, 100, 4096).await?;
assert_eq!(bytes.len(), 100);
# Ok(())
# }

When this crate helps — and when it does not

These numbers come from the harnesses in this repository and from end-to-end profiling of RustFS (rustfs/backlog#1159). They are reported as measured, including the cases where io_uring loses.

Two traps this crate's own benchmarking fell into, documented so others do not repeat them:

• A 76× apparent speedup turned out to be a behaviour regression, not a win: the io_uring path had silently stopped honouring RustFS's fadvise(DONTNEED) page-cache reclaim policy, so one leg served everything from cache while the other read the device. Always check disk_read and page-cache deltas, not just throughput.
• Microbenchmarks of the read path measured a page-cache-hit regime that production deliberately avoids for large reads. Isolated-path gains do not transfer end-to-end for free.

Testing

This is a Linux-only crate; on a non-Linux host cargo check only builds the empty stub.

# Native, on a Linux host with io_uring available:
cargo test -- --nocapture --test-threads=1

# Two legs in Docker (also runs on macOS via Docker Desktop / OrbStack):
#   leg 1 — io_uring blocked by an explicit seccomp profile → the suite MUST
#           degrade to a graceful skip (reproduces the #4313 restricted env);
#   leg 2 — seccomp=unconfined → real io_uring, and NO test may skip.
./run-docker.sh

The harness fails on either a non-degrading leg 1 or a vacuous-pass leg 2, so a skipped suite can never masquerade as real coverage. The 15 acceptance tests are the cancel-safety contract: buffer conservation under a mixed drop/keep stress across shards, an orphaned op reclaimed only at its CQE, bounded shutdown drain, O_DIRECT returning exact unaligned ranges, and backpressure deferring rather than blocking a runtime worker.

Benchmarks

Both harnesses refuse to overwrite or follow a symlink at a caller-supplied path (the sweeps run as root), fill their files with an offset-addressable pattern, and check every delivered byte against it under BENCH_VERIFY=1 — throughput alone cannot tell a correct strategy from one reading the wrong offsets.

# Sequential whole-file read: buffered vs O_DIRECT vs pipelined io_uring, warm and cold cache.
./bench-streaming.sh

# Many concurrent positioned reads on one disk — the shape shard reads actually serve.
# Isolates the cost of the per-read open and of the spawn_blocking hop.
./bench-concurrent-pread.sh

Roadmap

• Write path. Untouched today; PUT still goes through the blocking pool, and profiling suggests the win there may exceed the read path's.
• register_files. Would remove the per-op fd lookup. Lower value now that the consumer caches descriptors.
• SQPOLL. Eliminates io_uring_enter under sustained load, at the cost of a kernel polling thread per ring — which multiplies by shards and by disks. Only for high-end deployments.

Closed by measurement, not built — see CHANGELOG.md: streaming reads through io_uring (NO-GO), AsyncFd reaping without a driver thread (would break the public API), a process-wide singleton ring (conflicts with per-disk isolation), and registered buffers (conflicts with the Vec<u8> ownership model, and the bottleneck is elsewhere).

License

Apache-2.0. See LICENSE.