mii/RustySecrets
1
0
Fork 0
mirror of https://github.com/mii443/RustySecrets.git synced 2025-08-23 00:35:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

error handling for share duplicates

Browse Source
This commit is contained in:
Sebastian Gesemann
2015-01-29 17:20:15 +01:00
parent 3ab77d5b31
commit d7da24cfdd
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/main.rs
View File

@ -58,7 +58,9 @@ fn lagrange_interpolate(src: &[(u8, u8)], raw_x: u8) -> u8 {
for (j, &(raw_xj, _)) in src.iter().enumerate() { for (j, &(raw_xj, _)) in src.iter().enumerate() {
if i != j { if i != j {
let xj = Gf256::from_byte(raw_xj); let xj = Gf256::from_byte(raw_xj);
lix = lix * (x - xj) / (xi - xj); let delta = xi - xj;
assert!(delta.poly !=0, "Duplicate shares");
lix = lix * (x - xj) / delta;
} }
} }
sum = sum + lix * yi; sum = sum + lix * yi;
@ -135,7 +137,7 @@ fn read_shares() -> IoResult<(u8, Vec<(u8,Vec<u8>)>)> {
let mut stdin = BufferedReader::new(stdio::stdin()); let mut stdin = BufferedReader::new(stdio::stdin());
let mut opt_k_l: Option<(u8, usize)> = None; let mut opt_k_l: Option<(u8, usize)> = None;
let mut counter = 0u8; let mut counter = 0u8;
let mut shares = Vec::new(); let mut shares: Vec<(u8,Vec<u8>)> = Vec::new();
for line in stdin.lines() { for line in stdin.lines() {
let line = try!(line); let line = try!(line);
let parts: Vec<_> = line.split('-').collect(); let parts: Vec<_> = line.split('-').collect();
@ -167,13 +169,15 @@ fn read_shares() -> IoResult<(u8, Vec<(u8,Vec<u8>)>)> {
} else { } else {
opt_k_l = Some((k,raw.len())); opt_k_l = Some((k,raw.len()));
} }
if shares.iter().all(|s| s.0 != n) {
shares.push((n, raw)); shares.push((n, raw));
counter += 1; counter += 1;
if counter == k { if counter == k {
return Ok((k, shares)); return Ok((k, shares));
} }
} }
Err(other_io_err("No shares")) }
Err(other_io_err("Not enough shares provided!"))
} }
fn perform_decode() -> IoResult<()> { fn perform_decode() -> IoResult<()> {
@ -213,6 +217,7 @@ fn main() {
"The program secretshare is an implementation of Shamir's secret sharing scheme.\n\ "The program secretshare is an implementation of Shamir's secret sharing scheme.\n\
It is applied byte-wise within a finite field for arbitraty long secrets.\n"); It is applied byte-wise within a finite field for arbitraty long secrets.\n");
println!("{}", opts.usage("Usage: secretshare [options]")); println!("{}", opts.usage("Usage: secretshare [options]"));
println!("Input is read from STDIN and output is written to STDOUT.");
return; return;
} }