-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
c5eb19c
commit e352d50
Showing
1 changed file
with
193 additions
and
10 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,15 +1,198 @@ | ||
# Test corpus for tfhe-rs backward compatibility | ||
This repo holds various messages from tfhe-rs that have been versioned and serialized. | ||
The goal is to detect in tfhe-rs Ci when the version of a type should be upgraded because a breaking change has been added. | ||
# tfhe-rs backwards compatibility test corpus | ||
This repo contains various messages from [TFHE-rs](https://github.com/zama-ai/tfhe-rs) that have been versioned and serialized. | ||
The goal is to detect in TFHE-rs CI when the version of a type should be updated because a breaking change has been added. | ||
|
||
The messages are serialized using cbor and bincode because they both support large arrays and are vulnerable to different sets of breaking changes. Each message is stored with a set of metadata to verify that the values are loaded correctly. | ||
|
||
# Usage | ||
In TFHE-rs main repo, run the following command | ||
``` | ||
make test_backward_compatibility | ||
``` | ||
This will clone this repo and check if all messages are handled correctly. By default, this will use the `v0.1` branch (see below). To use a different branch, use this command instead: | ||
``` | ||
BACKWARD_COMPAT_DATA_BRANCH=my_branch_name make test_backward_compatibility | ||
``` | ||
|
||
# Versioning this repo | ||
The tests in this repo are by definition forward compatible (they should run on any future TFHE-rs release). They are also backward compatible (allowing tests to be run on past TFHE-rs versions, for example to bisect a bug), mostly because the TFHE-rs test driver will simply ignore any unknown test types and only load tests for versions inferior to its own. | ||
|
||
However, this does not allow changes to be made to the test metadata scheme itself. In such a case, a new version branch should be created (e.g. `v0.2`), and TFHE-rs should use that branch instead. | ||
|
||
Any commits to `main` should be backported to the latest version branch. | ||
|
||
The messages are serialized using cbor and bincode because they both support large arrays and are vulnerable to different sets of breaking changes. | ||
|
||
# Data generation | ||
To re-generate the data, run the binary target for this project: `cargo run --release`. The prng is seeded using a fixed seed so the data should be identical. | ||
To re-generate the data, run the binary target for this project: `cargo run --release`. The prng is seeded with a fixed seed, so the data should be identical. | ||
|
||
# Adding a test for an existing type | ||
To add a new test for a type that is already tested, you need to create a const global variable with the metadata for that test. The type of metadata depends on the type being tested (for example, the metadata for a test of the `ClientKey' from the `high_level_api' is `HlClientKEy'). Then go to the `data_vvv.rs` file (where "vvv" is the TFHE-rs version of the tested data) and update the `gen_xxx_data` method (where "xxx" is the API layer of your test (hl, shortint, integer,...)). In this method, create the object you want to test and serialize it using the `store_versioned_test` method. Add the metadata of your test to the vector returned by this method. | ||
|
||
The test will be automatically selected when you run TFHE-rs `make test_backward_compatibility`. | ||
|
||
## Example | ||
```rust | ||
// 1. Define the metadata associated with the test | ||
const HL_CT1_TEST: HlCiphertextTest = HlCiphertextTest { | ||
test_filename: Cow::Borrowed("ct1"), | ||
key_filename: Cow::Borrowed("client_key.cbor"), | ||
compressed: false, | ||
compact: false, | ||
clear_value: 0, | ||
}; | ||
|
||
impl TfhersVersion for V0_6 { | ||
// ... | ||
// Impl of trait | ||
// ... | ||
|
||
fn gen_hl_data() -> Vec<TestMetadata> { | ||
// ... | ||
// Init code and generation of other tests | ||
// ... | ||
|
||
// 2. Create the type | ||
let ct1 = fheint8::encrypt(HL_CT1_TEST.clear_value, &hl_client_key); | ||
|
||
// 3. Store it | ||
store_versioned_test(&ct1, &dir, &HL_CT1_TEST.test_filename); | ||
|
||
// 4. Return the metadata | ||
vec![ | ||
TestMetadata::HlCiphertext(HL_CT1_TEST), | ||
// ... | ||
// Metadata for other tests | ||
// ... | ||
] | ||
|
||
} | ||
|
||
``` | ||
|
||
# Adding tests for a new type | ||
|
||
## In this repo | ||
To add a test for a type that has not yet been tested, you should create a new type that implements the `TestType` trait. The type should also store the metadata needed for the test, and be serializable. By convention, its name should start with the API layer being tested. The metadata can be anything that can be used to check that the correct value is retrieved after deserialization. However, it should not use a TFHE-rs internal type. | ||
|
||
Once the type is created, it should be added to the `TestMetadata` enum. You can then add a new testcase using the procedure in the previous paragraph. | ||
|
||
## Example | ||
```rust | ||
// We use `Cow` for strings so that we can define them statically in this crate and load them | ||
// dynamically in the test driver. | ||
// Note that this type do not use anything from TFHE-rs | ||
#derive(Serialize, Deserialize, Clone, Debug)] | ||
pub struct HlCiphertextTest { | ||
pub test_filename: Cow<'static, str>, | ||
pub key_filename: Cow<'static, str>, | ||
pub compressed: bool, | ||
pub compact: bool, | ||
pub clear_value: u64, | ||
} | ||
|
||
impl TestType for HlCiphertextTest { | ||
fn module(&self) -> String { | ||
HL_MODULE_NAME.to_string() | ||
} | ||
|
||
fn target_type(&self) -> String { | ||
"FheUint".to_string() | ||
} | ||
|
||
fn test_filename(&self) -> String { | ||
self.test_filename.to_string() | ||
} | ||
} | ||
|
||
#[derive(Serialize, Deserialize, Clone, Debug, Display)] | ||
pub enum TestMetadata { | ||
// Hl | ||
HlCiphertext(HlCiphertextTest), | ||
// ... | ||
// All other supported types | ||
// ... | ||
} | ||
``` | ||
|
||
## In TFHE-rs | ||
In TFHE-rs, you should update the test driver to handle your new test type. To do this, create a function that loads and unversionizes the message, and then checks its value against the metadata provided: | ||
|
||
## Example | ||
```rust | ||
/// Test HL ciphertext: loads the ciphertext and compares the decrypted value with the one in the | ||
/// metadata. | ||
pub fn test_hl_ciphertext( | ||
dir: &Path, | ||
test: &HlCiphertextTest, | ||
format: DataFormat, | ||
) -> Result<TestSuccess, TestFailure> { | ||
let key_file = dir.join(&*test.key_filename); | ||
let key = ClientKey::unversionize( | ||
load_versioned_auxiliary(key_file).map_err(|e| test.failure(e, format))?, | ||
) | ||
.map_err(|e| test.failure(e, format))?; | ||
|
||
let server_key = key.generate_server_key(); | ||
set_server_key(server_key); | ||
|
||
let ct = if test.compressed { | ||
let compressed: CompressedFheUint8 = load_and_unversionize(dir, test, format)?; | ||
compressed.decompress() | ||
} else if test.compact { | ||
let compact: CompactFheUint8 = load_and_unversionize(dir, test, format)?; | ||
compact.expand().unwrap() | ||
} else { | ||
load_and_unversionize(dir, test, format)? | ||
}; | ||
|
||
let clear: u8 = ct.decrypt(&key); | ||
|
||
if clear != (test.clear_value as u8) { | ||
Err(test.failure( | ||
format!( | ||
"Invalid {} decrypted cleartext:\n Expected :\n{:?}\nGot:\n{:?}", | ||
format, clear, test.clear_value | ||
), | ||
format, | ||
)) | ||
} else { | ||
Ok(test.success(format)) | ||
} | ||
} | ||
|
||
// ... | ||
// Other tests | ||
// ... | ||
|
||
impl TestedModule for Hl { | ||
const METADATA_FILE: &'static str = "high_level_api.ron"; | ||
|
||
fn run_test<P: AsRef<Path>>( | ||
test_dir: P, | ||
testcase: &Testcase, | ||
format: DataFormat, | ||
) -> TestResult { | ||
#[allow(unreachable_patterns)] | ||
match &testcase.metadata { | ||
TestMetadata::HlCiphertext(test) => { | ||
test_hl_ciphertext(test_dir.as_ref(), test, format).into() | ||
} | ||
// ... | ||
// Match other tests | ||
// ... | ||
_ => { | ||
println!("WARNING: missing test: {:?}", testcase.metadata) | ||
TestResult::Skipped(testcase.skip()) | ||
} | ||
} | ||
} | ||
} | ||
``` | ||
|
||
# Adding a new tfhe-rs release | ||
To add data for a new released version of tfhe-rs, you should first add a dependency to that version in the `Cargo.toml` of this project. This dependency should only be enabled with the `generate` feature to avoid conflicts during testing. | ||
|
||
# Adding a new tfhe-rs version | ||
To add data for a new releaseed version of tfhe-rs, you should first add a dependency to this version in the `Cargo.toml` of this project. This dependency should only be activated with the `generate` feature to avoid conflicts in the testing phase. | ||
You should then implement the `TfhersVersion` trait for this version. You may use the code in `data_0_6.rs` as an example. | ||
You should then implement the `TfhersVersion` trait for this version. You can use the code in `data_0_6.rs` as an example. | ||
|
||
# Using the data generated in tests | ||
The data are stored using git-lfs, so first be sure to clone this project with lfs. To be able to parse the metadata and check that the loaded data are valid, your should add this crate as a dependency with the `load` feature activated. | ||
# Using the test data | ||
The data is stored using git-lfs, so be sure to clone this project with lfs first. To be able to parse the metadata and check if the loaded data is valid, you should add this crate as a dependency with the `load` feature enabled. |