Avoid unwrap (#177)

aws · Jul 7, 2023 · a866c04 · a866c04
1 parent b4128a3
commit a866c04
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/aws-lc-rs/src/agreement.rs b/aws-lc-rs/src/agreement.rs
@@ -526,12 +526,8 @@ where
         KeyInner::ECDH_P256(ec_key) | KeyInner::ECDH_P384(ec_key) | KeyInner::ECDH_P521(ec_key) => {
             let pub_key_bytes = peer_public_key.bytes.as_ref();
             unsafe {
-                let result =
-                    ec_key_ecdh(&mut buffer, ec_key.as_const(), pub_key_bytes, expected_nid);
-                if result.is_err() {
-                    return Err(error_value);
-                }
-                result.unwrap()
+                ec_key_ecdh(&mut buffer, ec_key.as_const(), pub_key_bytes, expected_nid)
+                    .or(Err(error_value))?
             }
         }
     };