Skip to content

Commit 9b41486

Browse files
committed
converted round 1 to use more errors
1 parent 916db28 commit 9b41486

File tree

1 file changed

+7
-8
lines changed

1 file changed

+7
-8
lines changed

src/dkg/round1.rs

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -146,16 +146,16 @@ pub fn export_secret_package<R: RngCore + CryptoRng>(
146146
pkg: &SecretPackage,
147147
identity: &Identity,
148148
csrng: R,
149-
) -> io::Result<Vec<u8>> {
149+
) -> Result<Vec<u8>, IronfishFrostError> {
150150
let serializable = <&SerializableSecretPackage>::from(pkg);
151151
if serializable.identifier != identity.to_frost_identifier() {
152-
return Err(io::Error::other("identity mismatch"));
152+
return Err(IronfishFrostError::InvalidInput(
153+
"identity mismatch".to_string(),
154+
));
153155
}
154156

155157
let mut serialized = Vec::new();
156-
serializable
157-
.serialize_into(&mut serialized)
158-
.expect("serialization failed");
158+
serializable.serialize_into(&mut serialized)?;
159159
Ok(multienc::encrypt(&serialized, [identity], csrng))
160160
}
161161

@@ -267,7 +267,7 @@ impl PublicPackage {
267267
}
268268

269269
pub fn deserialize_from<R: io::Read>(mut reader: R) -> Result<Self, IronfishFrostError> {
270-
let identity = Identity::deserialize_from(&mut reader).expect("reading identity failed");
270+
let identity = Identity::deserialize_from(&mut reader)?;
271271

272272
let frost_package = read_variable_length_bytes(&mut reader)?;
273273
let frost_package = Package::deserialize(&frost_package)?;
@@ -320,8 +320,7 @@ where
320320
)?;
321321

322322
let encrypted_secret_package =
323-
export_secret_package(&secret_package, self_identity, &mut csrng)
324-
.map_err(IronfishFrostError::EncryptionError)?;
323+
export_secret_package(&secret_package, self_identity, &mut csrng)?;
325324

326325
let group_secret_key_shard = GroupSecretKeyShard::random(&mut csrng);
327326

0 commit comments

Comments
 (0)