mirror of
https://github.com/data61/MP-SPDZ.git
synced 2026-04-20 03:01:31 -04:00
77 lines
1.8 KiB
C++
77 lines
1.8 KiB
C++
|
|
#include "DistDecrypt.h"
|
|
#include "FHE/P2Data.h"
|
|
#include "Protocols/MAC_Check.h"
|
|
|
|
template<class FD>
|
|
DistDecrypt<FD>::DistDecrypt(const Player& P, const FHE_SK& share,
|
|
const FHE_PK& pk, const FD& FTD) :
|
|
P(P), share(share), pk(pk), mf(FTD), f(FTD)
|
|
{
|
|
vv.resize(pk.get_params().phi_m());
|
|
vv1.resize(pk.get_params().phi_m());
|
|
// extra limb for operations
|
|
bigint limit = pk.get_params().Q() << 64;
|
|
vv.allocate_slots(limit);
|
|
vv1.allocate_slots(limit);
|
|
mf.allocate_slots(pk.p() << 64);
|
|
}
|
|
|
|
template<class FD>
|
|
Plaintext_<FD>& DistDecrypt<FD>::run(const Ciphertext& ctx, bool NewCiphertext)
|
|
{
|
|
const FHE_Params& params=ctx.get_params();
|
|
|
|
share.dist_decrypt_1(vv, ctx,P.my_num(),P.num_players());
|
|
|
|
if (not NewCiphertext)
|
|
intermediate_step();
|
|
|
|
if ((int)vv.size() != params.phi_m())
|
|
throw length_error("wrong length of ring element");
|
|
|
|
if (OnlineOptions::singleton.direct)
|
|
{
|
|
// Now pack into an octetStream for broadcasting
|
|
vector<octetStream> os(P.num_players());
|
|
|
|
for (int i=0; i<params.phi_m(); i++)
|
|
{ (os[P.my_num()]).store(vv[i]); }
|
|
|
|
// Broadcast and Receive the values
|
|
P.Broadcast_Receive(os);
|
|
|
|
// Reconstruct the value mod p0 from all shares
|
|
vv1.resize(params.phi_m());
|
|
for (int i = 0; i < P.num_players(); i++)
|
|
{
|
|
if (i != P.my_num())
|
|
{
|
|
for (int j = 0; j < params.phi_m(); j++)
|
|
{
|
|
os[i].get(vv1[j]);
|
|
}
|
|
share.dist_decrypt_2(vv, vv1);
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
TreeSum<bigint>().run(vv, P);
|
|
bigint mod=params.p0();
|
|
for (auto& v : vv)
|
|
v %= mod;
|
|
}
|
|
|
|
// Now get the final message
|
|
bigint mod=params.p0();
|
|
mf.set_poly_mod(vv,mod);
|
|
return mf;
|
|
}
|
|
|
|
|
|
template class DistDecrypt<FFT_Data>;
|
|
template class DistDecrypt<P2Data>;
|
|
|
|
|