Materialize/misc/bazel/c_deps/openssl/test.cc

// Copyright Materialize, Inc. and contributors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License in the LICENSE file at the
// root of this repository, or online at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// Small C test that ensures our openssl is built correctly.

#include <openssl/sha.h>

#include <cassert>
#include <iomanip>
#include <sstream>
#include <string>

// Use (void) to silent unused warnings.
#define assertm(exp, msg) assert(((void)msg, exp))

// From https://stackoverflow.com/a/2262447/7768383
bool simpleSHA256(const void* input, unsigned long length, unsigned char* md)
{
    SHA256_CTX context;
    if (!SHA256_Init(&context))
        return false;

    if (!SHA256_Update(&context, (unsigned char*)input, length))
        return false;

    if (!SHA256_Final(md, &context))
        return false;

    return true;
}

// Convert an byte array into a string
std::string fromByteArray(const unsigned char* data, unsigned long length)
{
    std::stringstream shastr;
    shastr << std::hex << std::setfill('0');
    for (unsigned long i = 0; i < length; ++i)
    {
        shastr << std::setw(2) << static_cast<int>(data[i]);
    }

    return shastr.str();
}

std::string MESSAGE = "hello world";
std::string MESSAGE_HASH = "b94d27b9934d3e08a52e52d7da7dabfac484efe37a5380ee9088f7ace2efcde9";

int main(int argc, char* argv[])
{
    unsigned char md[SHA256_DIGEST_LENGTH] = {};

    assertm(simpleSHA256(static_cast<const void*>(MESSAGE.data()), MESSAGE.size(), md), "Failed to generate hash");
    std::string hash = fromByteArray(md, SHA256_DIGEST_LENGTH);

    assertm(hash == MESSAGE_HASH, "Unexpected message hash");

    return 0;
}