public
/
blahaj
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Switch match to Clone to add zeroize(drop)

This commit is contained in:
Garrett Thornburg 2021-03-13 14:40:48 -08:00
parent d7a262c61d
commit a1b0dab5b2
3 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/field.rs
View File

@ -64,9 +64,10 @@ const EXP_TABLE: [u8; 512] = [
0x58, 0xb0, 0x7d, 0xfa, 0xe9, 0xcf, 0x83, 0x1b, 0x36, 0x6c, 0xd8, 0xad, 0x47, 0x8e, 0x01, 0x02, 0x58, 0xb0, 0x7d, 0xfa, 0xe9, 0xcf, 0x83, 0x1b, 0x36, 0x6c, 0xd8, 0xad, 0x47, 0x8e, 0x01, 0x02,
]; ];
#[derive(Debug, PartialEq, Copy, Clone)] #[derive(Debug, PartialEq, Clone)]
#[cfg_attr(feature = "fuzzing", derive(Arbitrary))] #[cfg_attr(feature = "fuzzing", derive(Arbitrary))]
#[cfg_attr(feature = "zeroize_memory", derive(Zeroize))] #[cfg_attr(feature = "zeroize_memory", derive(Zeroize))]
#[cfg_attr(feature = "zeroize_memory", zeroize(drop))]
pub struct GF256(pub u8); pub struct GF256(pub u8);
#[allow(clippy::suspicious_arithmetic_impl)] #[allow(clippy::suspicious_arithmetic_impl)]

2
src/lib.rs
View File

@ -201,7 +201,7 @@ mod tests {
let sharks = Sharks(255); let sharks = Sharks(255);
let mut shares: Vec<Share> = sharks.make_shares(&[1]).take(255).collect(); let mut shares: Vec<Share> = sharks.make_shares(&[1]).take(255).collect();
shares[1] = Share { shares[1] = Share {
x: shares[0].x, x: shares[0].x.clone(),
y: shares[0].y.clone(), y: shares[0].y.clone(),
}; };
let secret = sharks.recover(&shares); let secret = sharks.recover(&shares);

11
src/math.rs
View File

@ -20,9 +20,9 @@ pub fn interpolate(shares: &[Share]) -> Vec<u8> {
shares shares
.iter() .iter()
.filter(|s_j| s_j.x != s_i.x) .filter(|s_j| s_j.x != s_i.x)
.map(|s_j| s_j.x / (s_j.x - s_i.x)) .map(|s_j| s_j.x.clone() / (s_j.x.clone() - s_i.x.clone()))
.product::<GF256>() .product::<GF256>()
* s_i.y[s] * s_i.y[s].clone()
}) })
.sum::<GF256>() .sum::<GF256>()
.0 .0
@ -51,10 +51,13 @@ pub fn random_polynomial<R: rand::Rng>(s: GF256, k: u8, rng: &mut R) -> Vec<GF25
// The iterator will start at `x = 1` and end at `x = 255`. // The iterator will start at `x = 1` and end at `x = 255`.
pub fn get_evaluator(polys: Vec<Vec<GF256>>) -> impl Iterator<Item = Share> { pub fn get_evaluator(polys: Vec<Vec<GF256>>) -> impl Iterator<Item = Share> {
(1..=u8::max_value()).map(GF256).map(move |x| Share { (1..=u8::max_value()).map(GF256).map(move |x| Share {
x, x: x.clone(),
y: polys y: polys
.iter() .iter()
.map(|p| p.iter().fold(GF256(0), |acc, c| acc * x + *c)) .map(|p| {
p.iter()
.fold(GF256(0), |acc, c| acc * x.clone() + c.clone())
})
.collect(), .collect(),
}) })
} }