#include <cryptofuzz/util.h>

#include <cryptofuzz/repository.h>

#include <fuzzing/datasource/id.hpp>

#include <type_traits>

#include "bn_ops.h"

#define GET_WHICH(max) uint8_t which = 0; try { which = ds.Get<uint8_t>(); which %= ((max)+1); } catch ( ... ) { }

#define GET_OPTIONAL_BN() (ds.Get<bool>() ? bn.GetDestPtr(3) : nullptr)

namespace cryptofuzz {

namespace module {

namespace wolfCrypt_detail {

    WC_RNG* GetRNG(void);

    WC_RNG* GetSystemRNG(void);

}

namespace wolfCrypt_bignum {

namespace wolfCrypt_bignum_detail {

    template <class ReturnType>

    static ReturnType assertRet(const ReturnType ret) {

        static_assert(std::is_same<ReturnType, int>());

        if ( ret > 0 ) {

            CF_ASSERT(0, "Result of mp_* function is not negative or zero");

        }

        return ret;

    }

#define MP_CHECK_EQ(expr, res) CF_CHECK_EQ(::cryptofuzz::module::wolfCrypt_bignum::wolfCrypt_bignum_detail::assertRet(expr), res);

    static int compare(Bignum& A, Bignum& B, Datasource& ds) {

        util::HintBignumOpt(A.ToDecString());

        util::HintBignumOpt(B.ToDecString());

        bool swap = false;

        try {

            swap = ds.Get<bool>();

        } catch ( ... ) { }

        if ( swap == false ) {

            return mp_cmp(A.GetPtr(), B.GetPtr());

        } else {

            const auto ret = mp_cmp(B.GetPtr(), A.GetPtr());

            /* Because the operands were swapped, invert the result */

            if ( ret == MP_LT ) {

                return MP_GT;

            } else if ( ret == MP_GT ) {

                return MP_LT;

            } else {

                return ret;

            }

        }

    }

#if !defined(WOLFSSL_SP_MATH) && (!defined(USE_FAST_MATH) || defined(WOLFSSL_SP_MATH_ALL))

    static std::optional<int> isPowerOf2(Bignum& A, Datasource& ds) {

        std::optional<int> ret = std::nullopt;

        wolfCrypt_bignum::Bignum tmp(ds);

        auto numBits = mp_count_bits(A.GetPtr());

        CF_CHECK_GTE(numBits, 1);

        numBits--;

        MP_CHECK_EQ(mp_copy(A.GetPtr(), tmp.GetPtr()), MP_OKAY);

#if defined(USE_FAST_MATH) || defined(USE_INTEGER_HEAP_MATH)

        CF_NORET(mp_rshb(tmp.GetPtr(), numBits));

#else

        MP_CHECK_EQ(mp_rshb(tmp.GetPtr(), numBits), MP_OKAY);

#endif

        MP_CHECK_EQ(mp_mul_2d(tmp.GetPtr(), numBits, tmp.GetPtr()), MP_OKAY);

        {

            /* Use the nearest exponent of 2 as a hint for the mutator */

            const auto s = tmp.ToDecString();

            util::HintBignumOpt(s);

        }

        CF_CHECK_EQ(compare(A, tmp, ds), MP_EQ);

        ret = numBits;

end:

        return ret;

    }

#endif

    bool montgomery_reduce(

            fuzzing::datasource::Datasource& ds,

            mp_int* res,

            mp_int* a,

            const mp_digit b) {

        bool ret = false;

        bool which = false;

        try {

            which = ds.Get<bool>();

        } catch ( ... ) { }

        if ( which == false ) {

            MP_CHECK_EQ(mp_montgomery_reduce(res, a, b), MP_OKAY);

        } else {

            MP_CHECK_EQ(mp_montgomery_reduce_ct(res, a, b), MP_OKAY);

        }

        ret = true;

end:

        return ret;

    }

}

bool Add::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    bool ret = false;

    GET_WHICH(1);

    switch ( which ) {

        case    0:

            MP_CHECK_EQ(mp_add(bn[0].GetPtr(), bn[1].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            ret = true;

            break;

        case    1:

            {

                const auto op = bn[1].AsUnsigned<mp_digit>();

                CF_CHECK_NE(op, std::nullopt);

                MP_CHECK_EQ(mp_add_d(bn[0].GetPtr(), *op, bn.GetResPtr()), MP_OKAY);

                CF_CHECK_TRUE(bn.CopyResult(res));

                ret = true;

            }

            break;

    }

end:

    return ret;

}

bool Sub::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    bool ret = false;

#if defined(WOLFSSL_SP_MATH) || defined(WOLFSSL_SP_MATH_ALL)

    /* SP math cannot represent negative numbers, so ensure the result

     * of the subtracton is always >= 0.

     *

     * Still run the subtraction operation to see if this can cause

     * memory errors, but don't return the result.

     */

    bool negative = false;

    if ( wolfCrypt_bignum_detail::compare(bn[0], bn[1], ds) == MP_LT) {

        negative = true;

    }

#endif

    GET_WHICH(1);

    switch ( which ) {

        case    0:

            MP_CHECK_EQ(mp_sub(bn[0].GetPtr(), bn[1].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            ret = true;

            break;

        case    1:

            {

                const auto op = bn[1].AsUnsigned<mp_digit>();

                CF_CHECK_NE(op, std::nullopt);

                MP_CHECK_EQ(mp_sub_d(bn[0].GetPtr(), *op, bn.GetResPtr()), MP_OKAY);

                CF_CHECK_TRUE(bn.CopyResult(res));

                ret = true;

            }

            break;

    }

end:

#if defined(WOLFSSL_SP_MATH) || defined(WOLFSSL_SP_MATH_ALL)

    if ( negative == true ) {

        return false;

    }

#endif

    return ret;

}

bool Mul::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    bool ret = false;

    GET_WHICH(3);

    switch ( which ) {

        case    0:

            MP_CHECK_EQ(mp_mul(bn[0].GetPtr(), bn[1].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            ret = true;

            break;

        case    1:

            {

                const auto op = bn[1].AsUnsigned<mp_digit>();

                CF_CHECK_NE(op, std::nullopt);

                MP_CHECK_EQ(mp_mul_d(bn[0].GetPtr(), *op, bn.GetResPtr()), MP_OKAY);

                CF_CHECK_TRUE(bn.CopyResult(res));

                ret = true;

            }

            break;

#if !defined(USE_FAST_MATH) && !defined(WOLFSSL_SP_MATH)

        case    2:

            util::HintBignum("2");

            CF_CHECK_EQ(mp_cmp_d(bn[1].GetPtr(), 2), MP_EQ);

            MP_CHECK_EQ(mp_mul_2(bn[0].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            ret = true;

            break;

#endif

        case    3:

            {

                const auto numBits = mp_cnt_lsb(bn[1].GetPtr());

                CF_CHECK_GTE(numBits, DIGIT_BIT);

                CF_CHECK_EQ(numBits % DIGIT_BIT, 0);

                wolfCrypt_bignum::Bignum multiplier(ds);

                MP_CHECK_EQ(mp_2expt(multiplier.GetPtr(), numBits), MP_OKAY);

                CF_CHECK_EQ(wolfCrypt_bignum_detail::compare(bn[1], multiplier, ds), MP_EQ);

                MP_CHECK_EQ(mp_lshd(bn.GetDestPtr(0), numBits / DIGIT_BIT), MP_OKAY);

                MP_CHECK_EQ(mp_copy(bn[0].GetPtr(), res.GetPtr()), MP_OKAY);

                ret = true;

            }

            break;

    }

end:

    return ret;

}

bool Div::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    bool ret = false;

    GET_WHICH(3);

    switch ( which ) {

        case    0:

            {

                auto r = bn.GetResPtr();

                auto optional_bn = GET_OPTIONAL_BN();

                if ( optional_bn == r ) {

                    /* Prevent that the two result pointers are equal */

                    optional_bn = nullptr;

                }

                MP_CHECK_EQ(mp_div(bn[0].GetPtr(), bn[1].GetPtr(), r, optional_bn), MP_OKAY);

                CF_CHECK_TRUE(bn.CopyResult(res));

            }

            break;

        case    1:

#if !defined(WOLFSSL_SP_MATH)

            util::HintBignum("2");

            CF_CHECK_EQ(mp_cmp_d(bn[1].GetPtr(), 2), MP_EQ);

            MP_CHECK_EQ(mp_div_2(bn[0].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

#endif

#if !defined(USE_FAST_MATH)

        case    2:

            {

                mp_digit remainder;

                util::HintBignum("3");

                CF_CHECK_EQ(mp_cmp_d(bn[1].GetPtr(), 3), MP_EQ);

                MP_CHECK_EQ(mp_div_3(bn[0].GetPtr(), bn.GetResPtr(), ds.Get<bool>() ? &remainder : nullptr), MP_OKAY);

                CF_CHECK_TRUE(bn.CopyResult(res));

            }

            break;

#endif

#if defined(WOLFSSL_SP_MATH_ALL)

        case    3:

            {

                const auto divisor = bn[1].AsUnsigned<mp_digit>();

                mp_digit remainder;

                CF_CHECK_NE(divisor, std::nullopt);

                MP_CHECK_EQ(mp_div_d(bn[0].GetPtr(), *divisor, bn.GetResPtr(), ds.Get<bool>() ? &remainder : nullptr), MP_OKAY);

                CF_CHECK_TRUE(bn.CopyResult(res));

            }

#endif

        default:

            goto end;

    }

    /* wolfCrypt uses different rounding logic with negative divisior.

     * The result is still computed, but don't return it

     */

    CF_CHECK_EQ(mp_isneg(bn[0].GetPtr()), 0);

    ret = true;

end:

    return ret;

}

bool ExpMod::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    bool ret = false;

    try {

        switch ( ds.Get<uint8_t>() % 5 ) {

            case    0:

            {

                auto data = ds.GetData(0, 1024 / 8, 1024 / 8);

                /* Odd */

                data[1024 / 8 - 1] |= 1;

                util::HintBignum(util::BinToDec(data));

            }

            break;

            case    1:

            {

                auto data = ds.GetData(0, 1536 / 8, 1536 / 8);

                data[1536 / 8 - 1] |= 1;

                util::HintBignum(util::BinToDec(data));

            }

            break;

            case    2:

            {

                auto data = ds.GetData(0, 2048 / 8, 2048 / 8);

                data[2048 / 8 - 1] |= 1;

                util::HintBignum(util::BinToDec(data));

            }

            break;

            case    3:

            {

                auto data = ds.GetData(0, 3072 / 8, 3072 / 8);

                data[3072 / 8 - 1] |= 1;

                util::HintBignum(util::BinToDec(data));

            }

            break;

            case    4:

            {

                auto data = ds.GetData(0, 4096 / 8, 4096 / 8);

                data[4096 / 8 - 1] |= 1;

                util::HintBignum(util::BinToDec(data));

            }

            break;

        }

    } catch ( fuzzing::datasource::Datasource::OutOfData ) {

    }

    bool return_result = false;

#if defined(WOLFSSL_SP_MATH)

    return_result = true;

#else

    if (

            !mp_iszero(bn[1].GetPtr()) &&

            !mp_isneg(bn[0].GetPtr()) &&

            !mp_isneg(bn[1].GetPtr()) &&

            !mp_isneg(bn[2].GetPtr()) ) {

            return_result = true;

    }

#endif

    GET_WHICH(9);

    switch ( which ) {

        case    0:

            MP_CHECK_EQ(mp_exptmod(bn[0].GetPtr(), bn[1].GetPtr(), bn[2].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

#if defined(WOLFSSL_SP_MATH_ALL) || defined(USE_FAST_MATH)

        case    1:

            MP_CHECK_EQ(mp_exptmod_nct(bn[0].GetPtr(), bn[1].GetPtr(), bn[2].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

#endif

        case    2:

            MP_CHECK_EQ(mp_exptmod_ex(bn[0].GetPtr(), bn[1].GetPtr(), bn[1].GetPtr()->used, bn[2].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

#if defined(WOLFSSL_SP_MATH)

            /* ZD 15548 */

#if !defined(__aarch64__)

        case    3:

            MP_CHECK_EQ(sp_ModExp_1024(bn[0].GetPtr(), bn[1].GetPtr(), bn[2].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

#endif

        case    4:

            MP_CHECK_EQ(sp_ModExp_1536(bn[0].GetPtr(), bn[1].GetPtr(), bn[2].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

        case    5:

            MP_CHECK_EQ(sp_ModExp_2048(bn[0].GetPtr(), bn[1].GetPtr(), bn[2].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

        case    6:

            MP_CHECK_EQ(sp_ModExp_3072(bn[0].GetPtr(), bn[1].GetPtr(), bn[2].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

#if !defined(__i386__)

        case    7:

            MP_CHECK_EQ(sp_ModExp_4096(bn[0].GetPtr(), bn[1].GetPtr(), bn[2].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

#endif

#endif

#if !defined(WOLFSSL_SP_MATH) && !defined(WOLFSSL_SP_MATH_ALL) && !defined(USE_FAST_MATH)

        case    8:

            MP_CHECK_EQ(mp_exptmod_fast(bn[0].GetPtr(), bn[1].GetPtr(), bn[2].GetPtr(), bn.GetResPtr(), 0), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

#endif

#if !defined(WOLFSSL_SP_MATH) && !defined(WOLFSSL_SP_MATH_ALL) && !defined(USE_FAST_MATH)

        case    9:

            {

                util::HintBignum("2");

                CF_CHECK_EQ(mp_cmp_d(bn[0].GetPtr(), 2), MP_EQ);

                MP_CHECK_EQ(mp_exptmod_base_2(bn[1].GetPtr(), bn[2].GetPtr(), bn.GetResPtr()), MP_OKAY);

                CF_CHECK_TRUE(bn.CopyResult(res));

            }

            break;

#endif

        default:

            goto end;

    }

    CF_CHECK_TRUE(return_result);

    ret = true;

end:

    return ret;

}

bool Sqr::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    bool ret = false;

    MP_CHECK_EQ(mp_sqr(bn[0].GetPtr(), bn.GetResPtr()), MP_OKAY);

    CF_CHECK_TRUE(bn.CopyResult(res));

    ret = true;

end:

    return ret;

}

bool GCD::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    (void)res;

    (void)bn;

    bool ret = false;

    /* mp_gcd does not support negative numbers */

    CF_CHECK_NE(mp_cmp_d(bn[0].GetPtr(), 0), MP_LT);

    CF_CHECK_NE(mp_cmp_d(bn[1].GetPtr(), 0), MP_LT);

    MP_CHECK_EQ(mp_gcd(bn[0].GetPtr(), bn[1].GetPtr(), bn.GetResPtr()), MP_OKAY);

    CF_CHECK_TRUE(bn.CopyResult(res));

    ret = true;

end:

    return ret;

}

bool InvMod::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    bool ret = false;

    GET_WHICH(1);

    switch ( which ) {

        case    0:

            MP_CHECK_EQ(mp_invmod(bn[0].GetPtr(), bn[1].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

#define HAVE_MP_INVMOD_MONT_CT 1

#if defined(USE_INTEGER_HEAP_MATH)

 /* Heapmath does not have mp_invmod_mont_ct */

 #undef HAVE_MP_INVMOD_MONT_CT

#elif defined(WOLFSSL_SP_MATH) && !defined(WOLFSSL_SP_INVMOD_MONT_CT)

 /* SP math has mp_invmod_mont_ct only if WOLFSSL_SP_INVMOD_MONT_CT is defined */

 #undef HAVE_MP_INVMOD_MONT_CT

#endif

#if defined(HAVE_MP_INVMOD_MONT_CT)

        case    1:

            {

                mp_digit mp;

                wolfCrypt_bignum::Bignum tmp1(ds);

                wolfCrypt_bignum::Bignum tmp2(ds);

                int is_prime;

                /* Modulus must be > 2 and prime */

                CF_CHECK_EQ(mp_cmp_d(bn[1].GetPtr(), 2), MP_GT);

                MP_CHECK_EQ(mp_prime_is_prime_ex(bn[1].GetPtr(), 30, &is_prime, wolfCrypt_detail::GetRNG()), MP_OKAY);

                CF_CHECK_EQ(is_prime, 1);

                MP_CHECK_EQ(mp_montgomery_setup(bn[1].GetPtr(), &mp), MP_OKAY);

                MP_CHECK_EQ(mp_montgomery_calc_normalization(tmp1.GetPtr(), bn[1].GetPtr()), MP_OKAY);

                MP_CHECK_EQ(mp_mulmod(bn[0].GetPtr(), tmp1.GetPtr(), bn[1].GetPtr(), tmp2.GetPtr()), MP_OKAY);

                MP_CHECK_EQ(mp_invmod_mont_ct(tmp2.GetPtr(), bn[1].GetPtr(), res.GetPtr(), mp), MP_OKAY);

                CF_CHECK_TRUE(wolfCrypt_bignum_detail::montgomery_reduce(ds, res.GetPtr(), bn[1].GetPtr(), mp));

            }

            break;

#undef HAVE_MP_INVMOD_MONT_CT

#endif

#if !defined(USE_FAST_MATH) && !defined(WOLFSSL_SP_MATH)

#if 0

        case    2:

            MP_CHECK_EQ(mp_invmod_slow(bn[0].GetPtr(), bn[1].GetPtr(), res.GetPtr()), MP_OKAY);

            break;

#endif

#endif

        default:

            goto end;

    }

    ret = true;

end:

    return ret;

}

bool Cmp::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    bool ret = false;

    int cmpRes = 0;

    GET_WHICH(1);

    switch ( which ) {

        case    0:

            cmpRes = wolfCrypt_bignum_detail::compare(bn[0], bn[1], ds);

            break;

        case    1:

            {

                const auto op = bn[1].AsUnsigned<mp_digit>();

                CF_CHECK_NE(op, std::nullopt);

                cmpRes = mp_cmp_d(bn[0].GetPtr(), *op);

            }

            break;

        default:

            goto end;

    }

    switch ( cmpRes ) {

        case    MP_GT:

            CF_CHECK_EQ( res.Set("1"), true);

            break;

        case    MP_LT:

            CF_CHECK_EQ( res.Set("-1"), true);

            break;

        case    MP_EQ:

            CF_CHECK_EQ( res.Set("0"), true);

            break;

        default:

            CF_ASSERT(0, "Compare result is not one of (MP_GT, MP_LT, MP_EQ)");

    }

    ret = true;

end:

    return ret;

}

bool Abs::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    bool ret = false;

    MP_CHECK_EQ(mp_abs(bn[0].GetPtr(), bn.GetResPtr()), MP_OKAY);

    CF_CHECK_TRUE(bn.CopyResult(res));

    ret = true;

end:

    return ret;

}

bool Neg::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    bool ret = false;

#if defined(WOLFSSL_SP_MATH) || defined(WOLFSSL_SP_MATH_ALL)

    (void)res;

    (void)bn;

#else

    CF_CHECK_EQ(res.Set("0"), true);

    MP_CHECK_EQ(mp_sub(res.GetPtr(), bn[0].GetPtr(), res.GetPtr()), MP_OKAY);

    ret = true;

end:

#endif

    return ret;

}

bool RShift::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    bool ret = false;

    std::optional<uint64_t> _numBits;

    int numBits;

    GET_WHICH(2);

    CF_CHECK_NE(_numBits = bn[1].AsUint64(), std::nullopt);

    CF_CHECK_LTE(_numBits, 2147483647);

    numBits = *_numBits;

    switch ( which ) {

        case    0:

            MP_CHECK_EQ(mp_copy(bn[0].GetPtr(), res.GetPtr()), MP_OKAY);

#if defined(USE_FAST_MATH) || defined(USE_INTEGER_HEAP_MATH)

            CF_NORET(mp_rshb(res.GetPtr(), numBits));

#else

            MP_CHECK_EQ(mp_rshb(res.GetPtr(), numBits), MP_OKAY);

#endif

            ret = true;

            break;

#if !defined(WOLFSSL_SP_MATH)

        case    1:

            MP_CHECK_EQ(mp_div_2d(bn[0].GetPtr(), numBits, res.GetPtr(), GET_OPTIONAL_BN()), MP_OKAY);

            ret = true;

            break;

#endif

#if !defined(WOLFSSL_SP_MATH)

        case    2:

            {

                /* Check if number of bits to shift is a multiple of a full digit */

                CF_CHECK_EQ(numBits % (sizeof(mp_digit) * 8), 0);

                MP_CHECK_EQ(mp_copy(bn[0].GetPtr(), res.GetPtr()), MP_OKAY);

                const auto numDigits = numBits / (sizeof(mp_digit) * 8);

                CF_NORET(mp_rshd(res.GetPtr(), numDigits));

                ret = true;

            }

            break;

#endif

    }

end:

    return ret;

}

bool LShift1::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    bool ret = false;

#if defined(WOLFSSL_SP_MATH)

    (void)res;

    (void)bn;

#else

    MP_CHECK_EQ(mp_mul_2d(bn[0].GetPtr(), 1, bn.GetResPtr()), MP_OKAY);

    CF_CHECK_TRUE(bn.CopyResult(res));

    ret = true;

end:

#endif

    return ret;

}

bool IsNeg::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    bool ret = false;

#if defined(WOLFSSL_SP_MATH) || defined(WOLFSSL_SP_MATH_ALL)

    (void)res;

    (void)bn;

#else

    CF_CHECK_EQ( res.Set( std::to_string(mp_isneg(bn[0].GetPtr()) ? 1 : 0) ), true);

    ret = true;

end:

#endif

    return ret;

}

bool IsEq::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    bool ret = false;

    const bool isEq = wolfCrypt_bignum_detail::compare(bn[0], bn[1], ds) == MP_EQ;

    CF_CHECK_EQ( res.Set( std::to_string(isEq ? 1 : 0) ), true);

    ret = true;

end:

    return ret;

}

bool IsZero::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    bool ret = false;

    CF_CHECK_EQ( res.Set( std::to_string(mp_iszero(bn[0].GetPtr()) ? 1 : 0) ), true);

    ret = true;

end:

    return ret;

}

bool IsOne::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    bool ret = false;

    CF_CHECK_EQ( res.Set( std::to_string(mp_isone(bn[0].GetPtr()) ? 1 : 0) ), true);

    ret = true;

end:

    return ret;

}

bool MulMod::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    bool ret = false;

    MP_CHECK_EQ(mp_mulmod(bn[0].GetPtr(), bn[1].GetPtr(), bn[2].GetPtr(), bn.GetResPtr()), MP_OKAY);

    CF_CHECK_TRUE(bn.CopyResult(res));

    ret = true;

end:

    return ret;

}

bool AddMod::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    bool ret = false;

#if defined(WOLFSSL_SP_MATH)

    (void)ds;

    (void)res;

    (void)bn;

#else

    GET_WHICH(1);

    switch ( which ) {

        case    0:

            MP_CHECK_EQ(mp_addmod(bn[0].GetPtr(), bn[1].GetPtr(), bn[2].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

        case    1:

            /* mp_addmod_ct does not support negative numbers */

            CF_CHECK_NE(mp_cmp_d(bn[0].GetPtr(), 0), MP_LT);

            CF_CHECK_NE(mp_cmp_d(bn[1].GetPtr(), 0), MP_LT);

            CF_CHECK_NE(mp_cmp_d(bn[2].GetPtr(), 0), MP_LT);

            CF_CHECK_EQ(wolfCrypt_bignum_detail::compare(bn[0], bn[1], ds), MP_LT)

            CF_CHECK_EQ(wolfCrypt_bignum_detail::compare(bn[1], bn[2], ds), MP_LT)

            MP_CHECK_EQ(mp_addmod_ct(bn[0].GetPtr(), bn[1].GetPtr(), bn[2].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

        default:

            goto end;

    }

    ret = true;

end:

#endif

    return ret;

}

bool SubMod::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    bool ret = false;

#if defined(WOLFSSL_SP_MATH)

    (void)ds;

    (void)res;

    (void)bn;

#else

    GET_WHICH(1);

    switch ( which ) {

        case    0:

            MP_CHECK_EQ(mp_submod(bn[0].GetPtr(), bn[1].GetPtr(), bn[2].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

        case    1:

            /* mp_submod_ct does not support negative numbers */

            CF_CHECK_NE(mp_cmp_d(bn[0].GetPtr(), 0), MP_LT);

            CF_CHECK_NE(mp_cmp_d(bn[1].GetPtr(), 0), MP_LT);

            CF_CHECK_NE(mp_cmp_d(bn[2].GetPtr(), 0), MP_LT);

            /* mp_submod_ct documentation states that:

             *

             * A < modulo

             * B < modulo

             */

            CF_CHECK_EQ(wolfCrypt_bignum_detail::compare(bn[0], bn[2], ds), MP_LT)

            CF_CHECK_EQ(wolfCrypt_bignum_detail::compare(bn[1], bn[2], ds), MP_LT)

            MP_CHECK_EQ(mp_submod_ct(bn[0].GetPtr(), bn[1].GetPtr(), bn[2].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

        default:

            goto end;

    }

    ret = true;

end:

#endif

    return ret;

}

bool SqrMod::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    bool ret = false;

    MP_CHECK_EQ(mp_sqrmod(bn[0].GetPtr(), bn[1].GetPtr(), bn.GetResPtr()), MP_OKAY);

    CF_CHECK_TRUE(bn.CopyResult(res));

    ret = true;

end:

    return ret;

}

bool Bit::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    bool ret = false;

#if defined(WOLFSSL_SP_MATH)

    (void)res;

    (void)bn;

#else

    std::optional<uint64_t> _bitPos;

#if defined(WOLFSSL_SP_MATH_ALL)

    unsigned int bitPos;

#else

    mp_digit bitPos;

#endif

    int isBitSet;

    CF_CHECK_NE(_bitPos = bn[1].AsUint64(), std::nullopt);

    bitPos = *_bitPos;

    /* Ensure no truncation has occurred */

    CF_CHECK_EQ(bitPos, *_bitPos);

    CF_CHECK_GTE(isBitSet = mp_is_bit_set(bn[0].GetPtr(), bitPos), 0);

    CF_CHECK_EQ( res.Set(isBitSet ? "1" : "0"), true);

    ret = true;

end:

#endif

    return ret;

}

bool CmpAbs::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    (void)res;

    (void)bn;

    /* TODO */

    bool ret = false;

    return ret;

}

bool SetBit::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    bool ret = false;

    std::optional<uint64_t> _bitPos;

    mp_digit bitPos;

    CF_CHECK_NE(_bitPos = bn[1].AsUint64(), std::nullopt);

    bitPos = *_bitPos;

    /* Ensure no truncation has occurred */

    CF_CHECK_EQ(bitPos, *_bitPos);

    MP_CHECK_EQ(mp_copy(bn[0].GetPtr(), res.GetPtr()), MP_OKAY);

    MP_CHECK_EQ(mp_set_bit(res.GetPtr(), bitPos), MP_OKAY);

    ret = true;

end:

    return ret;

}

bool LCM::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

#if \

    defined(WOLFSSL_SP_MATH) || \

    (defined(WOLFSSL_SP_MATH_ALL) && \

        !(!defined(NO_RSA) && defined(WOLFSSL_KEY_GEN) && !defined(WC_RSA_BLINDING)) \

    )

    (void)ds;

    (void)res;

    (void)bn;

    return false;

#else

    (void)ds;

    bool ret = false;

    /* mp_lcm does not support negative numbers */

    CF_CHECK_NE(mp_cmp_d(bn[0].GetPtr(), 0), MP_LT);

    CF_CHECK_NE(mp_cmp_d(bn[1].GetPtr(), 0), MP_LT);

    MP_CHECK_EQ(mp_lcm(bn[0].GetPtr(), bn[1].GetPtr(), bn.GetResPtr()), MP_OKAY);

    CF_CHECK_TRUE(bn.CopyResult(res));

    ret = true;

end:

    return ret;

#endif

}

bool Mod::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    bool ret = false;

    GET_WHICH(4);

    switch ( which ) {

        case    0:

            MP_CHECK_EQ(mp_mod(bn[0].GetPtr(), bn[1].GetPtr(), bn.GetResPtr()), MP_OKAY);

            CF_CHECK_TRUE(bn.CopyResult(res));

            break;

#if !defined(WOLFSSL_SP_MATH)

        case    1:

            {

                /* Input must not be negative */

                CF_CHECK_NE(mp_cmp_d(bn[0].GetPtr(), 0), MP_LT);

                const auto op = bn[1].AsUnsigned<mp_digit>();

                CF_CHECK_NE(op, std::nullopt);

                mp_digit modResult;

                MP_CHECK_EQ(mp_mod_d(bn[0].GetPtr(), *op, &modResult), MP_OKAY);

                CF_CHECK_EQ(res.Set(std::to_string(modResult)), true);

            }

            break;

        case    2:

            {

                /* mp_div_2_mod_ct does not support negative numbers */

                CF_CHECK_NE(mp_cmp_d(bn[0].GetPtr(), 0), MP_LT);

                CF_CHECK_NE(mp_cmp_d(bn[0].GetPtr(), 0), MP_LT);

                /* bn[0] *= 2 */

                MP_CHECK_EQ(mp_mul_d(bn[0].GetPtr(), 2, bn.GetDestPtr(0)), MP_OKAY);

                CF_CHECK_EQ(wolfCrypt_bignum_detail::compare(bn[0], bn[1], ds), MP_LT)

                MP_CHECK_EQ(mp_div_2_mod_ct(bn[0].GetPtr(), bn[1].GetPtr(), bn.GetResPtr()), MP_OKAY);

                CF_CHECK_TRUE(bn.CopyResult(res));

            }

            break;

        case    3:

            {

                mp_digit mp;

                wolfCrypt_bignum::Bignum tmp(ds);

                MP_CHECK_EQ(mp_montgomery_setup(bn[1].GetPtr(), &mp), MP_OKAY);

                MP_CHECK_EQ(mp_montgomery_calc_normalization(tmp.GetPtr(), bn[1].GetPtr()), MP_OKAY);

                MP_CHECK_EQ(mp_mulmod(bn[0].GetPtr(), tmp.GetPtr(), bn[1].GetPtr(), res.GetPtr()), MP_OKAY);

                CF_CHECK_TRUE(wolfCrypt_bignum_detail::montgomery_reduce(ds, res.GetPtr(), bn[1].GetPtr(), mp));

            }

            break;

#endif

#if !defined(WOLFSSL_SP_MATH) && (!defined(USE_FAST_MATH) || defined(WOLFSSL_SP_MATH_ALL))

        case    4:

            {

                std::optional<int> exponent = std::nullopt;

                /* Negative modulo not supported */

                CF_CHECK_NE(mp_cmp_d(bn[1].GetPtr(), 0), MP_LT);

                CF_CHECK_NE(exponent = wolfCrypt_bignum_detail::isPowerOf2(bn[1], ds), std::nullopt);

                MP_CHECK_EQ(mp_mod_2d(bn[0].GetPtr(), *exponent, bn.GetResPtr()), MP_OKAY);

                CF_CHECK_TRUE(bn.CopyResult(res));

            }

            break;

#endif

        default:

            goto end;

    }

    ret = true;

end:

    return ret;

}

bool IsEven::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    bool ret = false;

    CF_CHECK_EQ(mp_iszero(bn[0].GetPtr()), false);

    CF_CHECK_EQ( res.Set( std::to_string(mp_iseven(bn[0].GetPtr()) ? 1 : 0) ), true);

    ret = true;

end:

    return ret;

}

bool IsOdd::Run(Datasource& ds, Bignum& res, BignumCluster& bn) const {

    (void)ds;

    bool ret = false;