sha256.cuh

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#pragma once
#include <fss/hash.cuh>
#include <cuda_runtime.h>
#include <cuda/std/array>
#include <cuda/std/span>
#include <cuda/std/tuple>
#include <cassert>
#include <openssl/evp.h>
#include <fss/util.cuh>
 
namespace fss::hash {
 
class Sha256 {
private:
    int4 key_;
 
public:
    explicit Sha256(int4 key) : key_(key) {}
 
    __host__ __device__ cuda::std::array<int4, 2> Hash(cuda::std::span<const int4, 4> msg) {
        cuda::std::array<int4, 2> out{};
 
#ifdef __CUDA_ARCH__
        assert(false && "Sha256 is not supported on device side");
        __trap();
#else
        int4 buf[5] = {key_, msg[0], msg[1], msg[2], msg[3]};
        auto buf_ptr = reinterpret_cast<const unsigned char *>(buf);
        auto out_ptr = reinterpret_cast<unsigned char *>(out.data());
        int ret = EVP_Digest(buf_ptr, 80, out_ptr, NULL, EVP_sha256(), NULL);
        assert(ret == 1);
#endif
 
        return out;
    }
 
    __host__ __device__ cuda::std::array<int4, 4> Hash(cuda::std::tuple<int4, const int4> msg) {
        cuda::std::array<int4, 4> out{};
 
#ifdef __CUDA_ARCH__
        assert(false && "Sha256 is not supported on device side");
        __trap();
#else
        auto [a, b] = msg;
        int4 buf[3] = {key_, fss::util::SetLsb(a, false), b};
        auto buf_ptr = reinterpret_cast<const unsigned char *>(buf);
        auto out_ptr = reinterpret_cast<unsigned char *>(out.data());
        int ret = EVP_Digest(buf_ptr, 48, out_ptr, NULL, EVP_sha256(), NULL);
        assert(ret == 1);
 
        buf[1] = fss::util::SetLsb(a, true);
        ret = EVP_Digest(buf_ptr, 48, out_ptr + 32, NULL, EVP_sha256(), NULL);
        assert(ret == 1);
#endif
 
        return out;
    }
};
static_assert(Hashable<Sha256> && XorHashable<Sha256>);
 
}  // namespace fss::hash