Add oneshot extended functions

cppalliance · Jan 15, 2025 · 0dd0572 · 0dd0572
1 parent cbfdac6
commit 0dd0572
Show file tree

Hide file tree

Showing 2 changed files with 58 additions and 0 deletions.
diff --git a/include/boost/crypt2/hash/shake128.hpp b/include/boost/crypt2/hash/shake128.hpp
@@ -64,6 +64,35 @@ auto shake128(SizedRange&& data, OutputRange&& out) noexcept -> state
     return hasher.get_digest(out);
 }
 
+[[nodiscard]] BOOST_CRYPT_EXPORT BOOST_CRYPT_GPU_ENABLED_CONSTEXPR
+auto shake128(compat::span<const compat::byte> data, compat::span<compat::byte> out, compat::size_t amount) noexcept -> state
+{
+    shake128_hasher hasher;
+    hasher.process_bytes(data);
+    hasher.finalize();
+    return hasher.get_digest(out, amount);
+}
+
+template <concepts::writable_output_range OutputRange>
+[[nodiscard]] BOOST_CRYPT_EXPORT BOOST_CRYPT_GPU_ENABLED
+auto shake128(compat::span<const compat::byte> data, OutputRange&& out, compat::size_t amount) noexcept -> state
+{
+    shake128_hasher hasher;
+    hasher.process_bytes(data);
+    hasher.finalize();
+    return hasher.get_digest(out, amount);
+}
+
+template <compat::sized_range SizedRange, concepts::writable_output_range OutputRange>
+[[nodiscard]] BOOST_CRYPT_EXPORT BOOST_CRYPT_GPU_ENABLED
+auto shake128(SizedRange&& data, OutputRange&& out, compat::size_t amount) noexcept -> state
+{
+    shake128_hasher hasher;
+    hasher.process_bytes(data);
+    hasher.finalize();
+    return hasher.get_digest(out, amount);
+}
+
 #if !BOOST_CRYPT_HAS_CUDA
 
 template <concepts::file_system_path T>

diff --git a/include/boost/crypt2/hash/shake256.hpp b/include/boost/crypt2/hash/shake256.hpp
@@ -64,6 +64,35 @@ auto shake256(SizedRange&& data, OutputRange&& out) noexcept -> state
     return hasher.get_digest(out);
 }
 
+[[nodiscard]] BOOST_CRYPT_EXPORT BOOST_CRYPT_GPU_ENABLED_CONSTEXPR
+auto shake256(compat::span<const compat::byte> data, compat::span<compat::byte> out, std::size_t amount) noexcept -> state
+{
+    shake256_hasher hasher;
+    hasher.process_bytes(data);
+    hasher.finalize();
+    return hasher.get_digest(out, amount);
+}
+
+template <concepts::writable_output_range OutputRange>
+[[nodiscard]] BOOST_CRYPT_EXPORT BOOST_CRYPT_GPU_ENABLED
+auto shake256(compat::span<const compat::byte> data, OutputRange&& out, std::size_t amount) noexcept -> state
+{
+    shake256_hasher hasher;
+    hasher.process_bytes(data);
+    hasher.finalize();
+    return hasher.get_digest(out, amount);
+}
+
+template <compat::sized_range SizedRange, concepts::writable_output_range OutputRange>
+[[nodiscard]] BOOST_CRYPT_EXPORT BOOST_CRYPT_GPU_ENABLED
+auto shake256(SizedRange&& data, OutputRange&& out, std::size_t amount) noexcept -> state
+{
+    shake256_hasher hasher;
+    hasher.process_bytes(data);
+    hasher.finalize();
+    return hasher.get_digest(out, amount);
+}
+
 #if !BOOST_CRYPT_HAS_CUDA
 
 template <concepts::file_system_path T>