Skip to content

Add CRC64NVME to smithy-checksums - #868

Draft
jonathan343 wants to merge 9 commits into
developfrom
feat/smithy-checksums-crc64nvme
Draft

jonathan343 wants to merge 9 commits into
developfrom
feat/smithy-checksums-crc64nvme

Conversation

@jonathan343

@jonathan343 jonathan343 commented Oct 25, 2026 •

Copy link
Copy Markdown
Contributor

Note

Stacked on #867, targeting feat/smithy-checksums so this PR contains only CRC64NVME.

Summary

S3 uses CRC64NVME for full-object checksums, and it's the algorithm S3 computes when an upload doesn't specify one. This PR adds it to smithy-checksums as CRC64NVME, also available through new("crc64nvme"). It reuses the native extension and the pure-Python CRC implementation from #867. The native side uses crc-fast's Crc64Nvme. The pure-Python side reuses the table-driven CRC that CRC32C uses, with the CRC-64/NVME polynomial.

Example

import base64

import smithy_checksums

checksum = smithy_checksums.new("CRC64NVME")
checksum.update(b"Hello world")
print(checksum.name, checksum.digest_size, base64.b64encode(checksum.digest()))

The native and pure-Python builds print the same thing:

crc64nvme 8 b'OOJZ0D8xKts='

Testing

The known answers are the CRC-64/NVME check value for "123456789" and the four 64-bit CRC test cases from the NVMe NVM Command Set specification. Both implementations are also compared against the CRT and against each other:

uv run pytest packages/smithy-checksums -k "crc64nvme and (known_answers or matches_crt or native_matches_pure)" --no-cov -q
6 passed, 108 deselected in 0.46s

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

Adds the smithy-checksums package, which provides the checksum algorithms
Smithy clients need for flexible checksums. CRC32 and CRC32C are backed by a
native Rust extension (PyO3, crc-fast) with a pure-Python fallback, and
SHA-1, SHA-256, SHA-512, and MD5 come from hashlib through new().

Wheels are built by an in-tree PEP 517 backend: maturin when a Rust
toolchain is available, hatchling (pure Python) otherwise. Source
distributions are always built with hatchling. SMITHY_PYTHON_NATIVE_BUILD
(auto, always, never) overrides the choice. If Rust is present but too old,
or the native build fails, the build stops with instructions instead of
silently falling back.

Native wheels target abi3 for 3.12+, cp314t, and abi3t for 3.15t+. Native
and pure-Python checksum objects are safe to share between threads.

Repository setup:
- CI builds the native extension in the main matrix, checks Rust
  formatting, lints, dependencies (cargo-deny), and third-party license
  attribution (cargo-about), builds and tests native wheels on Linux,
  macOS, and Windows, and tests the pure-Python wheel and an sdist install
  without Rust.
- Adds make targets check-rust, lint-rust, and third-party-licenses.
- License header checks cover .rs and .pyi files.
- Dependabot updates Cargo dependencies.
Pins the toolchain used for development and CI in rust-toolchain.toml, so
a new stable release can't introduce clippy lints that fail CI without a
change here. Builds from a source distribution are unaffected and use the
installed toolchain.

CI now caches Rust builds with Swatinem/rust-cache, saving caches only on
develop, and adds a job that checks each native extension builds with the
minimum Rust version its Cargo.toml declares. The new jobs also get names
and version comments on their pinned actions.
The byte-at-a-time table loop runs about a dozen bytecodes per byte, which
caps it at roughly 15 MB/s however it's arranged; slicing-by-8 and 16-bit
tables only reach 20-35 MB/s.

A CRC without its initial value and final XOR is linear over GF(2), so each
bit of the CRC of a fixed-size chunk is the parity of the chunk's bits under
a fixed mask. Input is now buffered into 16 KiB chunks, each read as one
integer, and every CRC bit is computed with an AND and bit_count(), which do
the per-byte work in C. The masks are built lazily by doubling, in a few
milliseconds. Remainders under 1 KiB still use the table.

CRC32C goes from about 16 MB/s to about 210 MB/s, whatever the update size.
@jonathan343
jonathan343 force-pushed the feat/smithy-checksums-crc64nvme branch from 82838e3 to fec00fb Compare October 27, 2026 19:15
maturin packages everything under python-source, so a native wheel built
from a source tree that's been imported from picked up its __pycache__
directories, including bytecode for modules that no longer exist. The
hatchling targets already exclude build artifacts; this gives maturin the
same treatment.
The native extension read read-only buffers in place through an unsafe
slice, on the grounds that nothing could change them. The read-only flag
only stops this reader from writing, though: a read-only memoryview of a
bytearray still changes when another thread writes to the bytearray, which
can happen while the interpreter is detached. Only bytes is immutable, so
it's the only input that's still hashed in place; every other buffer is
copied. Copying runs at about 40 GB/s, against about 100 GB/s in place.

Also:
- State::digest returns a fixed-size array, given by an associated Digest
  type, instead of a Vec.
- Import pyo3 items explicitly instead of through the prelude.
Adds CRC64NVME, the algorithm S3 uses for full-object checksums, backed by
the native extension with a pure-Python fallback, and available through
new("crc64nvme").
CRC64NVME shares the chunked big-integer path with CRC32C, which takes it
from about 15 MB/s to about 135 MB/s. Add it to the test that checks that
path against a bit-at-a-time CRC around the chunk and tail boundaries.
@jonathan343
jonathan343 force-pushed the feat/smithy-checksums-crc64nvme branch from fec00fb to eef1834 Compare October 27, 2026 21:04
Base automatically changed from feat/smithy-checksums to develop October 31, 2026 18:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant