cksum: Fix -a blake3/shake*

Author: oech3

src/uu/cksum/src/cksum.rs

@@ -103,6 +103,14 @@ fn maybe_sanitize_length(
         // For BLAKE2b, if a length is provided, validate it.
         (Some(AlgoKind::Blake2b), Some(len)) => calculate_blake2b_length_str(len),
 
+         // Bug: length is not applied for SHAKE.
+        (Some(AlgoKind::Shake128 | AlgoKind::Shake256), Some(s_len)) => {
+           let len = s_len
+               .parse::<usize>()
+              .map_err(|_| ChecksumError::InvalidLength(s_len.to_string()))?;
+           Ok(Some(len))
+        }
+
         // For any other provided algorithm, check if length is 0.
         // Otherwise, this is an error.
         (_, Some(len)) if len.parse::<u32>() == Ok(0_u32) => Ok(None),

src/uucore/src/lib/features/checksum/mod.rs

@@ -114,6 +114,12 @@ impl AlgoKind {
             ALGORITHM_OPTIONS_SHA256 => Sha256,
             ALGORITHM_OPTIONS_SHA384 => Sha384,
             ALGORITHM_OPTIONS_SHA512 => Sha512,
+
+            // Extension
+            ALGORITHM_OPTIONS_BLAKE3 => Blake3,
+            ALGORITHM_OPTIONS_SHAKE128 => Shake128,
+            ALGORITHM_OPTIONS_SHAKE256 => Shake256,
+
             _ => return Err(ChecksumError::UnknownAlgorithm(algo.as_ref().to_string()).into()),
         })
     }

tests/by-util/test_cksum.rs

@@ -3071,6 +3071,36 @@ mod debug_flag {
     }
 }
 
+#[test]
+#[cfg(target_os = "linux")]
+fn test_dev_null_b3() {
+    new_ucmd!()
+        .arg("-a")
+        .arg("blake3")
+        .arg("/dev/null")
+        .succeeds()
+        .stdout_contains(
+            "BLAKE3 (/dev/null) = af1349b9f5f9a1a6a0404dea36dcc9499bcb25c9adc112b7cc9a93cae41f3262",
+        );
+}
+
+#[test]
+fn test_shake() {
+    let (at, mut ucmd) = at_and_ucmd!();
+
+    at.touch("a");
+
+    ucmd.arg("a")
+        .arg("-a")
+        .arg("shake128")
+        .arg("-l")
+        .arg("128")
+        .succeeds()
+        .stdout_contains(
+            "SHAKE128 (a) = 7f9c2ba4e88f827d616045507605853ed73b8093f6efbc88eb1a6eacfa66ef26",
+        );
+}
+
 #[test]
 #[cfg(all(target_os = "linux", not(target_env = "musl")))]
 fn test_check_file_with_io_error() {