test and doctest

sm2/src/pke.rs

@@ -18,20 +18,23 @@
 //!
 //! // Encrypting
 //! let secret_key = SecretKey::random(&mut OsRng); // serialize with `::to_bytes()`
-//! let encrypting_key = EncryptingKey::new_with_mode(secret_key, Mode::C1C2C3);
+//! let public_key = secret_key.public_key();
+//! let encrypting_key = EncryptingKey::new_with_mode(public_key, Mode::C1C2C3);
 //! let plaintext = b"plaintext";
 //! let ciphertext = encrypting_key.encrypt(plaintext)?;
 //!
 //! use sm2::pke::DecryptingKey;
 //! // Decrypting
-//! let decrypting_key = DecryptingKey::new_with_mode(secret_key, Mode::C1C2C3);
+//! let decrypting_key = DecryptingKey::new_with_mode(secret_key.to_nonzero_scalar(), Mode::C1C2C3);
 //! assert_eq!(decrypting_key.decrypt(&ciphertext)?, plaintext);
 //!
 //! // Encrypting asn.1
 //! let ciphertext = encrypting_key.encrypt_asna1(plaintext)?;
 //!
 //! // Decrypting asn.1
 //! assert_eq!(decrypting_key.decrypt_asna1(&ciphertext)?, plaintext);
+//! 
+//! Ok(())
 //! # }
 //!  ```
 //!
@@ -173,3 +176,4 @@ pub(crate) fn zero_byte_slice<const N: usize>(
     output[num_zeroes..].copy_from_slice(bytes);
     Ok(output)
 }
+

sm2/src/pke/encrypting.rs

@@ -188,11 +188,12 @@ fn encrypt(
     })
 }
 
-fn next_k(bit_length: u32) -> Uint<4> {
+fn next_k(bit_length: u32) -> U256 {
     loop {
         let k = U256::random_bits(&mut rand_core::OsRng, bit_length);
         if k.is_zero().unwrap_u8() == 0 && k <= Sm2::ORDER {
             return k;
         }
     }
 }
+