Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Refactor digest computations (Arecibo backport) #229

Merged
merged 1 commit into from
Sep 26, 2023

Conversation

huitseeker
Copy link
Contributor

@huitseeker huitseeker commented Sep 25, 2023

Rationale

The multi-accumulator of SuperNova (introduced in #204, incubated in Arecibo) requires introducing digests in a more systematic approach.

What's in this PR?

This PR introduces a systematic way of creating a Digest for a particular data structure. The pattern is documented in the tests of src/digest.rs. Here are the highlights:

  • the digest is computed lazily,
  • the digest is cached and never needs to be computed more than once,
  • structures equipped with digest computation do not require checking if the digest has been correctly populated, or if the deserialized digest matches the structure's bytes,
  • there is no need for a convention on hashing the structure with the digest bytes initialized to "zero",

How do I implement a Digest-equipped data structure ?

Here's the checklist:

  • implement the Digestible trait, which describes which bytes you'd want to pass to the hasher,
  • if the target object is serializable, consider implementing the SimpleDigestible marker trait, which will stream the bincode serialization of the object,
  • add a digest field to the target object wrapped in a OnceCell, use the provided serde attribute (#[serde(skip, default = "OnceCell::new")]) to ensure it's ignored during (de)serialization,
  • implement a getter for the digest using the DigestComputer generic structure and OnceCell's get_or_try_init,

This backports the following PRs:

* Add DigestBuilder.

* Make digest and claims private.

* refactor: Refactor DigestBuilder

- Refactored `src/digest.rs` to replace `Vec<u8>` storage with dedicated Write I/O.
- Removed optional `hasher` and introduced dedicated factory method.
- Reworked digest computation and mapping into separate functions.
- Merged build and digest computation to enhance coherence.
- Improved type safety with Result error propagation.

* Propagate DigestBuilder changes.

* Fix tests.

* Correct assertion for OutputSize scale.

* Remove commented.

* Remove dbg!.

* Fixup rebase.

---------

Co-authored-by: porcuquine <[email protected]>
Co-authored-by: François Garillot <[email protected]>

feat: add a digest to R1CSShape (microsoft#49)

* refactor: Refactor Digestible trait

- Removed `to_bytes` method from the `Digestible` trait in `src/digest.rs` file.

* fix: Make bincode serialization in digest.rs more rigorous

- Updated `bincode::serialize_into(byte_sink, self)` with a configurable version to enable "little endian" and "fixint encoding" options.
- Added a comment in `src/digest.rs` about `bincode`'s recursive length-prefixing during serialization.

* refactor: Refactor digest computation using `OnceCell` and `DigestComputer`

This gives up on a generic builder and instead uses an idempotent `OnceCell`
+ a generic digest computer to populate the digest of a structure.

- this shows how to set up digest computation so it doesn't depend on the digest field,
- the digest can't be set twice,
- an erroneous digest can't be inferred from the serialized data.

In Details:

- Overhauled digest functionality in multiple files by replacing `DigestBuilder` with `DigestComputer`, significantly altering the handling of hashes.
- Incorporated `once_cell::sync::OnceCell` and `ff::PrimeField` dependencies to improve performance and simplify code.
- Modified `VerifierKey` and `RunningClaims` structures to include a `OnceCell` for digest, leading to a change in function calls and procedures.
- Simplified `setup_running_claims` by removing error handling and directly returning `RunningClaims` type.
- Adapted test functions according to the changes including the removal of unnecessary unwrapping in certain scenarios.
- Updated Cargo.toml with the new dependency `once_cell` version `1.18.0`.

* refactor: rename pp digest in VerifierKey to pp_digest

* feat: add a digest to R1CSShape

* fix: Small issues

- Introduced a new assertion within the `write_bytes` method of `src/supernova/mod.rs` for validating whether the `claims` are empty
- Improved code comment clarity regarding the creation of a running claim in `src/supernova/mod.rs`.
@srinathsetty srinathsetty merged commit 284c985 into microsoft:main Sep 26, 2023
2 checks passed
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.

3 participants