Function TEST

tests/unit/c_api/test_pve_batch.cpp:73–130

}  // namespace

TEST(CApiPveBatch, EncVerDec_DefBasePke_RsaBlob) {
  const cbmpc_curve_id_t curve = CBMPC_CURVE_SECP256K1;
  const cmem_t label = {reinterpret_cast<uint8_t*>(const_cast<char*>("label")), 5};

  constexpr int n = 4;
  std::array<uint8_t, static_cast<size_t>(n) * 32> xs_flat{};
  std::array<int, n> xs_sizes{};
  for (int i = 0; i < n; i++) {
    xs_sizes[static_cast<size_t>(i)] = 32;
    for (int j = 0; j < 32; j++) xs_flat[static_cast<size_t>(i * 32 + j)] = static_cast<uint8_t>(i + j);
  }
  const cmems_t xs_in = {n, xs_flat.data(), xs_sizes.data()};

  cmem_t ek_blob{nullptr, 0};
  cmem_t dk_blob{nullptr, 0};
  ASSERT_EQ(cbmpc_pve_generate_base_pke_rsa_keypair(&ek_blob, &dk_blob), CBMPC_SUCCESS);

  cmem_t ct{nullptr, 0};
  ASSERT_EQ(cbmpc_pve_batch_encrypt(/*base_pke=*/nullptr, curve, ek_blob, label, xs_in, &ct), CBMPC_SUCCESS);

  int batch_count = 0;
  ASSERT_EQ(cbmpc_pve_batch_get_count(ct, &batch_count), CBMPC_SUCCESS);
  ASSERT_EQ(batch_count, n);

  cmem_t L_ct{nullptr, 0};
  ASSERT_EQ(cbmpc_pve_batch_get_Label(ct, &L_ct), CBMPC_SUCCESS);
  expect_eq(L_ct, label);
  cbmpc_cmem_free(L_ct);

  std::array<uint8_t, static_cast<size_t>(n) * 33> Qs_flat{};
  std::array<int, n> Qs_sizes{};
  for (int i = 0; i < n; i++) {
    const mem_t xi(xs_flat.data() + i * 32, 32);
    const buf_t qi = expected_Q(curve, xi);
    ASSERT_EQ(qi.size(), 33);
    Qs_sizes[static_cast<size_t>(i)] = qi.size();
    std::memmove(Qs_flat.data() + i * 33, qi.data(), static_cast<size_t>(qi.size()));
  }
  const cmems_t Qs_expected = {n, Qs_flat.data(), Qs_sizes.data()};

  ASSERT_EQ(cbmpc_pve_batch_verify(/*base_pke=*/nullptr, curve, ek_blob, ct, Qs_expected, label), CBMPC_SUCCESS);

  cmems_t xs_out{0, nullptr, nullptr};
  ASSERT_EQ(cbmpc_pve_batch_decrypt(/*base_pke=*/nullptr, curve, dk_blob, ek_blob, ct, label, &xs_out), CBMPC_SUCCESS);
  ASSERT_EQ(xs_out.count, n);

  int off = 0;
  for (int i = 0; i < n; i++) {
    ASSERT_EQ(xs_out.sizes[i], 32);
    ASSERT_EQ(std::memcmp(xs_out.data + off, xs_flat.data() + i * 32, 32), 0);
    off += xs_out.sizes[i];
  }

  cbmpc_cmems_free(xs_out);
  cbmpc_cmem_free(ct);
  cbmpc_cmem_free(dk_blob);
  cbmpc_cmem_free(ek_blob);
}