1 files changed, 6 insertions, 4 deletions
diff --git a/src/utils/cookie.rs b/src/utils/cookie.rs
index 826efa9..c621621 100644
--- a/src/utils/cookie.rs
+++ b/src/utils/cookie.rs
@@ -1,5 +1,5 @@
 use hex;
-use rand_core::{OsRng, RngCore};
+use rand_core::{OsRng, TryRngCore};
 
 use crate::crypto::signed;
 use crate::model::config::Config;
@@ -22,10 +22,12 @@ pub fn extract_token(config: &Config, cookie: &str) -> Result<String, String> {
     signed::verify(&config.auth_secret, signed_cookie)
 }
 
-pub fn generate_token() -> String {
+pub fn generate_token() -> Result<String, String> {
     let mut token = [0u8; TOKEN_BYTES];
-    OsRng.fill_bytes(&mut token);
-    hex::encode(token)
+    OsRng
+        .try_fill_bytes(&mut token)
+        .map_err(|_| "Error generating random token")?;
+    Ok(hex::encode(token))
 }
 
 fn cookie(config: &Config, token: &str, max_age_seconds: i32) -> String {