Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
test(kyberlib): ✅ tests covering the main functions in the `symmetric…
….rs`
- Loading branch information
1 parent
5d3317b
commit 524278e
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
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
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
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 |
|---|---|---|
| @@ -0,0 +1,142 @@ | ||
| // Copyright © 2023 kyberlib. All rights reserved. | ||
| // SPDX-License-Identifier: Apache-2.0 OR MIT | ||
|
|
||
| #[cfg(test)] | ||
| mod tests { | ||
| // Import necessary items | ||
| use kyberlib::{ | ||
| symmetric::{hash_g, hash_h, kdf, prf}, | ||
| KYBER_SHARED_SECRET_BYTES, | ||
| }; | ||
|
|
||
| // Test the hash_h function | ||
| #[test] | ||
| fn test_hash_h() { | ||
| let input = b"test input"; | ||
| let inlen = input.len(); | ||
| let mut out = [0u8; 32]; | ||
|
|
||
| // Call the hash_h function | ||
| hash_h(&mut out, input, inlen); | ||
|
|
||
| // Assert that the output is not all zeros | ||
| assert_ne!(out, [0u8; 32]); | ||
| } | ||
|
|
||
| // Test the hash_g function | ||
| #[test] | ||
| fn test_hash_g() { | ||
| let input = b"test input"; | ||
| let inlen = input.len(); | ||
| let mut out = [0u8; 64]; | ||
|
|
||
| // Call the hash_g function | ||
| hash_g(&mut out, input, inlen); | ||
|
|
||
| // Assert that the output is not all zeros | ||
| assert_ne!(out, [0u8; 64]); | ||
| } | ||
|
|
||
| // Test the xof_absorb and xof_squeezeblocks functions | ||
| #[test] | ||
| fn test_xof_absorb_squeeze() { | ||
| use kyberlib::params::KYBER_SYM_BYTES; | ||
| use kyberlib::symmetric::KeccakState; | ||
| use kyberlib::symmetric::{kyber_shake128_absorb, kyber_shake128_squeezeblocks}; | ||
|
|
||
| const SHAKE128_RATE: usize = 168; | ||
|
|
||
| let mut state = KeccakState::new(); | ||
| let input = [1u8; KYBER_SYM_BYTES]; | ||
| let x = 1; | ||
| let y = 2; | ||
| let mut out = [0u8; SHAKE128_RATE]; | ||
|
|
||
| // Absorb input data into the Kyber-specific SHAKE128 state | ||
| kyber_shake128_absorb(&mut state, &input, x, y); | ||
| let nblocks = out.len() / SHAKE128_RATE; | ||
| kyber_shake128_squeezeblocks(&mut out, nblocks, &mut state); | ||
| let outlen = out.len(); | ||
| let mut idx = 0; | ||
| for i in 0..outlen { | ||
| assert_ne!(out[i], 0); | ||
| idx += 1; | ||
| } | ||
| assert_eq!(idx, outlen); | ||
| } | ||
|
|
||
| // Test the prf function | ||
| #[test] | ||
| fn test_prf() { | ||
| use kyberlib::params::KYBER_SYM_BYTES; | ||
| let mut key = [0u8; KYBER_SYM_BYTES]; | ||
| key[..8].copy_from_slice(b"test key"); | ||
| let nonce = 42; | ||
| let mut out = [0u8; 64]; | ||
| let outbytes = out.len(); | ||
|
|
||
| // Call the prf function | ||
| prf(&mut out, outbytes, &key, nonce); | ||
|
|
||
| // Assert that the output is not all zeros | ||
| assert_ne!(out, [0u8; 64]); | ||
| } | ||
|
|
||
| // Test the kdf function | ||
| #[test] | ||
| fn test_kdf() { | ||
| let input = b"test input"; | ||
| let inlen = input.len(); | ||
| let mut out = [0u8; KYBER_SHARED_SECRET_BYTES]; | ||
|
|
||
| // Call the kdf function | ||
| kdf(&mut out, input, inlen); | ||
|
|
||
| // Assert that the output is not all zeros | ||
| assert_ne!(out, [0u8; KYBER_SHARED_SECRET_BYTES]); | ||
| } | ||
|
|
||
| #[cfg(not(feature = "90s"))] | ||
| #[test] | ||
| fn test_kyber_shake128_absorb_squeeze() { | ||
| use kyberlib::params::KYBER_SYM_BYTES; | ||
| use kyberlib::symmetric::KeccakState; | ||
| use kyberlib::symmetric::{kyber_shake128_absorb, kyber_shake128_squeezeblocks}; | ||
|
|
||
| const SHAKE128_RATE: usize = 168; | ||
|
|
||
| let mut state = KeccakState::new(); | ||
| let input = [1u8; KYBER_SYM_BYTES]; | ||
| let x = 1; | ||
| let y = 2; | ||
| let mut out = [0u8; SHAKE128_RATE]; | ||
|
|
||
| // Absorb input data into the Kyber-specific SHAKE128 state | ||
| kyber_shake128_absorb(&mut state, &input, x, y); | ||
|
|
||
| // Squeeze Kyber-specific SHAKE128 data into output | ||
| kyber_shake128_squeezeblocks(&mut out, 1, &mut state); | ||
|
|
||
| // Assert that the output is not all zeros | ||
| assert_ne!(out, [0u8; SHAKE128_RATE]); | ||
| } | ||
|
|
||
| #[cfg(not(feature = "90s"))] | ||
| #[test] | ||
| fn test_shake256_prf() { | ||
| use kyberlib::params::KYBER_SYM_BYTES; | ||
| use kyberlib::symmetric::shake256_prf; | ||
|
|
||
| let mut key = [0u8; KYBER_SYM_BYTES]; | ||
| key.copy_from_slice(&[0u8; KYBER_SYM_BYTES]); | ||
| let nonce = 42; | ||
| let mut out = [0u8; 32]; // Output length of SHAKE256 is 32 bytes | ||
| let outlen = out.len(); | ||
|
|
||
| // Call the SHAKE256 PRF function | ||
| shake256_prf(&mut out, outlen, &key, nonce); | ||
|
|
||
| // Assert that the output is not all zeros | ||
| assert_ne!(out, [0u8; 32]); | ||
| } | ||
| } | ||
