// Copyright 2024 Christian Mazakas.
// Distributed under the Boost Software License, Version 1.0.
// https://www.boost.org/LICENSE_1_0.txt

#define _CRT_SECURE_NO_WARNINGS

#ifdef _MSC_VER
# pragma warning(disable: 4309) // truncation of constant value
#endif

#include <boost/hash2/ripemd.hpp>
#include <boost/core/lightweight_test.hpp>
#include <string>
#include <cstddef>
#include <cstdio>

template<std::size_t N> std::string to_string( std::array<unsigned char, N> const & v )
{
    std::string r;

    for( std::size_t i = 0; i < N; ++i )
    {
        char buffer[ 8 ];

        std::snprintf( buffer, sizeof( buffer ), "%02x", static_cast<int>( v[ i ] ) );

        r += buffer;
    }

    return r;
}

template<class H> std::string digest( std::string const k, std::string const & s )
{
    H h( reinterpret_cast<unsigned char const*>( k.data() ), k.size() );

    h.update( s.data(), s.size() );

    return to_string( h.result() );
}

std::string from_hex( char const* str )
{
    auto f = []( char c ) { return ( c >= 'a' ? c - 'a' + 10 : c - '0' ); };

    std::string s;
    while( *str != '\0' )
    {
        s.push_back( static_cast<char>( ( f( str[ 0 ] ) << 4 ) + f( str[ 1 ] ) ) );
        str += 2;
    }
    return s;
}

using boost::hash2::hmac_ripemd_160;

int main()
{
    // Test vectors from https://datatracker.ietf.org/doc/rfc2286/

    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b" ), "Hi There" ), std::string( "24cb4bd67d20fc1a5d2ed7732dcc39377f0a5668" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( "Jefe", "what do ya want for nothing?" ), std::string( "dda6c0213a485a9e24f4742064a7f033b43c4069" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa" ), from_hex( "dddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd" ) ), std::string( "b0b105360de759960ab4f35298e116e295d8e7c1" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "0102030405060708090a0b0c0d0e0f10111213141516171819" ), from_hex( "cdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcd" ) ), std::string( "d5ca862f4d21d5e610e18b4cf1beb97a4365ecf4" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c" ), "Test With Truncation" ), std::string( "7619693978f91d90539ae786500ff3d8e0518e39" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa" ), "Test Using Larger Than Block-Size Key - Hash Key First" ), std::string( "6466ca07ac5eac29e1bd523e5ada7605b791fd8b" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa" ), "Test Using Larger Than Block-Size Key and Larger Than One Block-Size Data" ), std::string( "69ea60798d71616cce5fd0871e23754cd75d5a0a" ) );

    // Test vectors from https://homes.esat.kuleuven.be/%7Ebosselae/ripemd160.html

    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "00112233445566778899aabbccddeeff01234567" ), "" ), std::string( "cf387677bfda8483e63b57e06c3b5ecd8b7fc055" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "0123456789abcdeffedcba987654321000112233" ), "" ), std::string( "fe69a66c7423eea9c8fa2eff8d9dafb4f17a62f5" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "00112233445566778899aabbccddeeff01234567" ), "a" ), std::string( "0d351d71b78e36dbb7391c810a0d2b6240ddbafc" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "0123456789abcdeffedcba987654321000112233" ), "a" ), std::string( "85743e899bc82dbfa36faaa7a25b7cfd372432cd" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "00112233445566778899aabbccddeeff01234567" ), "abc" ), std::string( "f7ef288cb1bbcc6160d76507e0a3bbf712fb67d6" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "0123456789abcdeffedcba987654321000112233" ), "abc" ), std::string( "6e4afd501fa6b4a1823ca3b10bd9aa0ba97ba182" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "00112233445566778899aabbccddeeff01234567" ), "message digest"  ), std::string( "f83662cc8d339c227e600fcd636c57d2571b1c34" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "0123456789abcdeffedcba987654321000112233" ), "message digest"  ), std::string( "2e066e624badb76a184c8f90fba053330e650e92" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "00112233445566778899aabbccddeeff01234567" ), "abcdefghijklmnopqrstuvwxyz" ), std::string( "843d1c4eb880ac8ac0c9c95696507957d0155ddb" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "0123456789abcdeffedcba987654321000112233" ), "abcdefghijklmnopqrstuvwxyz" ), std::string( "07e942aa4e3cd7c04dedc1d46e2e8cc4c741b3d9" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "00112233445566778899aabbccddeeff01234567" ), "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq" ), std::string( "60f5ef198a2dd5745545c1f0c47aa3fb5776f881" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "0123456789abcdeffedcba987654321000112233" ), "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq" ), std::string( "b6582318ddcfb67a53a67d676b8ad869aded629a" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "00112233445566778899aabbccddeeff01234567" ), "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789" ), std::string( "e49c136a9e5627e0681b808a3b97e6a6e661ae79" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "0123456789abcdeffedcba987654321000112233" ), "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789" ), std::string( "f1be3ee877703140d34f97ea1ab3a07c141333e2" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "00112233445566778899aabbccddeeff01234567" ), "12345678901234567890123456789012345678901234567890123456789012345678901234567890" ), std::string( "31be3cc98cee37b79b0619e3e1c2be4f1aa56e6c" ) );
    BOOST_TEST_EQ( digest<hmac_ripemd_160>( from_hex( "0123456789abcdeffedcba987654321000112233" ), "12345678901234567890123456789012345678901234567890123456789012345678901234567890" ), std::string( "85f164703e61a63131be7e45958e0794123904f9" ) );

    return boost::report_errors();
}