Ryan Henry
0d22946a0e
Ship the TLS mesh, composer, Beaver/Yao/leaf MPC, prep/online paths, apps, and docs so the tree is pushable before elevating share_expr, security_mode, and prep resume. Co-authored-by: Cursor <cursoragent@cursor.com>
260 lines
9.1 KiB
C++
260 lines
9.1 KiB
C++
#include <gtest/gtest.h>
|
|
|
|
#include "dpf.hpp"
|
|
|
|
#include <type_traits>
|
|
|
|
namespace
|
|
{
|
|
|
|
template <typename A, typename B>
|
|
auto recon(const A & a, const B & b)
|
|
{
|
|
if constexpr (dpf::is_secret_share_v<std::decay_t<A>>
|
|
&& dpf::is_secret_share_v<std::decay_t<B>>)
|
|
return dpf::reconstruct(a, b);
|
|
else
|
|
{
|
|
using T = std::common_type_t<std::decay_t<A>, std::decay_t<B>>;
|
|
if constexpr (std::is_integral_v<T> && std::is_unsigned_v<T>)
|
|
return static_cast<T>(a - b);
|
|
else
|
|
return a - b;
|
|
}
|
|
}
|
|
|
|
} // namespace
|
|
|
|
#include "helpers/eval_common_multi_data.hpp"
|
|
|
|
template <typename T>
|
|
struct EvalPointMultiTest : public testing::Test
|
|
{
|
|
public:
|
|
using input_type = typename std::tuple_element_t<0, T>;
|
|
using output_type0 = typename std::tuple_element_t<1, T>;
|
|
using output_type1 = typename std::tuple_element_t<2, T>;
|
|
using output_type2 = typename std::tuple_element_t<3, T>;
|
|
using output_type3 = typename std::tuple_element_t<4, T>;
|
|
using integral_type = dpf::utils::integral_type_from_bitlength_t<dpf::utils::bitlength_of_v<input_type>>;
|
|
using dpf_type = dpf::utils::dpf_type_t<dpf::prg::aes128, dpf::prg::aes128, input_type, output_type0, output_type1, output_type2, output_type3>;
|
|
|
|
protected:
|
|
EvalPointMultiTest()
|
|
: params{std::get<std::vector<T>>(allParams)},
|
|
range{(std::size_t(1) << std::min(dpf::utils::bitlength_of_v<input_type>, std::size_t(10))-1)-1},
|
|
zero_output0{from_integral_type_output0(0)},
|
|
zero_output1{from_integral_type_output1(0)},
|
|
zero_output2{from_integral_type_output2(0)},
|
|
zero_output3{from_integral_type_output3(0)}
|
|
{ }
|
|
|
|
void SetUp() override
|
|
{ }
|
|
|
|
void TearDown() override
|
|
{ }
|
|
|
|
input_type get_start(const input_type & x)
|
|
{
|
|
integral_type x_int = to_integral_type(x),
|
|
min_int = to_integral_type(std::numeric_limits<input_type>::min()),
|
|
max_int = to_integral_type(std::numeric_limits<input_type>::max()),
|
|
start_int;
|
|
// set start_int so that the tested range is centered around x_int if possible
|
|
// use start_int = min_int if x_int smaller than min_int+range
|
|
// or start_int = max_int-2*range if x_int larger than max_int-range
|
|
// range is selected to be at most 1 less than half the maximum range for input_type
|
|
// this ensures there are no overflow issues
|
|
// for signed integral types, since the MSB is internally flipped,
|
|
// needed additional check that x_int was in the correct range for the given
|
|
// conditionals (note that these added checks are always try for unsigned types)
|
|
if (x_int < min_int + range && x_int >= min_int)
|
|
{
|
|
start_int = min_int;
|
|
}
|
|
else if (x_int > max_int - range && x_int <= max_int)
|
|
{
|
|
start_int = max_int - (range << 1);
|
|
}
|
|
else
|
|
{
|
|
start_int = x_int - range;
|
|
}
|
|
return from_integral_type(start_int);
|
|
}
|
|
|
|
template <typename UnaryFunction0, typename UnaryFunction1>
|
|
void assert_wrapper(const input_type & x, const output_type0 & y0,
|
|
const output_type1 & y1, const output_type2 & y2, const output_type3 & y3,
|
|
UnaryFunction0 f0, UnaryFunction1 f1)
|
|
{
|
|
input_type cur = get_start(x);
|
|
for (std::size_t i = 0; i <= range<<1; ++i, cur = next_domain_point(cur))
|
|
{
|
|
auto out0 = f0(cur),
|
|
out1 = f1(cur);
|
|
if (cur == x)
|
|
{
|
|
ASSERT_EQ(recon(std::get<0>(out0), std::get<0>(out1)), y0);
|
|
ASSERT_EQ(recon(std::get<1>(out0), std::get<1>(out1)), y1);
|
|
ASSERT_EQ(recon(std::get<2>(out0), std::get<2>(out1)), y2);
|
|
ASSERT_EQ(recon(std::get<3>(out0), std::get<3>(out1)), y3);
|
|
}
|
|
else
|
|
{
|
|
ASSERT_EQ(recon(std::get<0>(out0), std::get<0>(out1)), zero_output0);
|
|
ASSERT_EQ(recon(std::get<1>(out0), std::get<1>(out1)), zero_output1);
|
|
ASSERT_EQ(recon(std::get<2>(out0), std::get<2>(out1)), zero_output2);
|
|
ASSERT_EQ(recon(std::get<3>(out0), std::get<3>(out1)), zero_output3);
|
|
}
|
|
}
|
|
}
|
|
|
|
static constexpr auto to_integral_type = dpf::utils::to_integral_type<input_type>{};
|
|
static constexpr auto from_integral_type = dpf::utils::make_from_integral_value<input_type>{};
|
|
static constexpr auto from_integral_type_output0 = dpf::utils::make_from_integral_value<output_type0>{};
|
|
static constexpr auto from_integral_type_output1 = dpf::utils::make_from_integral_value<output_type1>{};
|
|
static constexpr auto from_integral_type_output2 = dpf::utils::make_from_integral_value<output_type2>{};
|
|
static constexpr auto from_integral_type_output3 = dpf::utils::make_from_integral_value<output_type3>{};
|
|
|
|
std::vector<T> params;
|
|
std::size_t range;
|
|
output_type0 zero_output0;
|
|
output_type1 zero_output1;
|
|
output_type2 zero_output2;
|
|
output_type3 zero_output3;
|
|
};
|
|
|
|
TYPED_TEST_SUITE_P(EvalPointMultiTest);
|
|
|
|
TYPED_TEST_P(EvalPointMultiTest, Basic)
|
|
{
|
|
using input_type = typename TestFixture::input_type;
|
|
|
|
for (auto [x, y0, y1, y2, y3] : this->params)
|
|
{
|
|
auto _keys = dpf::make_dpf(x, y0, y1, y2, y3);
|
|
|
|
auto & dpf0 = std::get<0>(_keys);
|
|
|
|
auto & dpf1 = std::get<1>(_keys);
|
|
|
|
this->assert_wrapper(x, y0, y1, y2, y3,
|
|
[&dpf0](input_type cur)
|
|
{
|
|
return dpf::eval_point<0, 1, 2, 3>(dpf0, cur);
|
|
},
|
|
[&dpf1](input_type cur)
|
|
{
|
|
return dpf::eval_point<0, 1, 2, 3>(dpf1, cur);
|
|
}
|
|
);
|
|
}
|
|
}
|
|
|
|
TYPED_TEST_P(EvalPointMultiTest, BasicPathMemoizer)
|
|
{
|
|
using input_type = typename TestFixture::input_type;
|
|
using dpf_type = typename TestFixture::dpf_type;
|
|
auto memo0 = dpf::make_basic_path_memoizer<dpf_type>(),
|
|
memo1 = dpf::make_basic_path_memoizer<dpf_type>();
|
|
|
|
for (auto [x, y0, y1, y2, y3] : this->params)
|
|
{
|
|
auto _keys = dpf::make_dpf(x, y0, y1, y2, y3);
|
|
|
|
auto & dpf0 = std::get<0>(_keys);
|
|
|
|
auto & dpf1 = std::get<1>(_keys);
|
|
|
|
this->assert_wrapper(x, y0, y1, y2, y3,
|
|
[&dpf0, &memo0](input_type cur)
|
|
{
|
|
return dpf::eval_point<0, 1, 2, 3>(dpf0, cur, memo0);
|
|
},
|
|
[&dpf1, &memo1](input_type cur)
|
|
{
|
|
return dpf::eval_point<0, 1, 2, 3>(dpf1, cur, memo1);
|
|
}
|
|
);
|
|
}
|
|
}
|
|
|
|
TYPED_TEST_P(EvalPointMultiTest, NonmemoizingPathMemoizer)
|
|
{
|
|
using input_type = typename TestFixture::input_type;
|
|
using dpf_type = typename TestFixture::dpf_type;
|
|
auto memo0 = dpf::make_nonmemoizing_path_memoizer<dpf_type>(),
|
|
memo1 = dpf::make_nonmemoizing_path_memoizer<dpf_type>();
|
|
|
|
for (auto [x, y0, y1, y2, y3] : this->params)
|
|
{
|
|
auto _keys = dpf::make_dpf(x, y0, y1, y2, y3);
|
|
|
|
auto & dpf0 = std::get<0>(_keys);
|
|
|
|
auto & dpf1 = std::get<1>(_keys);
|
|
|
|
this->assert_wrapper(x, y0, y1, y2, y3,
|
|
[&dpf0, &memo0](input_type cur)
|
|
{
|
|
return dpf::eval_point<0, 1, 2, 3>(dpf0, cur, memo0);
|
|
},
|
|
[&dpf1, &memo1](input_type cur)
|
|
{
|
|
return dpf::eval_point<0, 1, 2, 3>(dpf1, cur, memo1);
|
|
}
|
|
);
|
|
}
|
|
}
|
|
|
|
REGISTER_TYPED_TEST_SUITE_P(EvalPointMultiTest,
|
|
Basic,
|
|
BasicPathMemoizer,
|
|
NonmemoizingPathMemoizer);
|
|
using Types = testing::Types
|
|
<
|
|
// base test
|
|
test_type<uint16_t, uint64_t>,
|
|
|
|
// test input types
|
|
test_type<int16_t, uint64_t>,
|
|
test_type<uint8_t, uint64_t>,
|
|
test_type<uint64_t, uint64_t>,
|
|
test_type<simde_int128, uint64_t>,
|
|
test_type<simde_uint128, uint64_t>,
|
|
test_type<dpf::bitstring<10>, uint64_t>,
|
|
test_type<dpf::keyword<3, dpf::alphabets::hex>, uint64_t>,
|
|
test_type<dpf::modint<10>, uint64_t>,
|
|
test_type<dpf::xor_wrapper<int16_t>, uint64_t>,
|
|
test_type<dpf::xor_wrapper<uint16_t>, uint64_t>,
|
|
|
|
// test output types
|
|
test_type<uint16_t, int64_t>,
|
|
test_type<uint16_t, uint8_t>,
|
|
test_type<uint16_t, simde_int128>,
|
|
test_type<uint16_t, simde_uint128>,
|
|
test_type<uint16_t, dpf::bit>,
|
|
test_type<uint16_t, dpf::bitstring<20, uint8_t>>,
|
|
test_type<uint16_t, dpf::bitstring<150>>,
|
|
test_type<uint16_t, dpf::xor_wrapper<int64_t>>,
|
|
test_type<uint16_t, dpf::xor_wrapper<uint64_t>>,
|
|
test_type<uint16_t, dpf::gf2>,
|
|
test_type<uint16_t, dpf::gf22>,
|
|
test_type<uint16_t, dpf::gf24>,
|
|
test_type<uint16_t, dpf::gf28>,
|
|
test_type<uint16_t, dpf::gf216>,
|
|
test_type<uint16_t, dpf::gf232>,
|
|
test_type<uint16_t, dpf::gf264>,
|
|
|
|
// custom types
|
|
test_type<custom_input_type, uint64_t>,
|
|
test_type<uint16_t, custom_output_type_small>,
|
|
test_type<uint16_t, custom_output_type_large_plus_minus>,
|
|
test_type<uint16_t, custom_output_type_large_xor>,
|
|
|
|
// distinct output types
|
|
multi_test_type<uint16_t, uint32_t, dpf::xor_wrapper<uint32_t>, dpf::bitstring<20, uint8_t>, dpf::bitstring<32>>
|
|
>;
|
|
INSTANTIATE_TYPED_TEST_SUITE_P(EvalPointMultiTestInstantiation, EvalPointMultiTest, Types);
|