/*
 * MaliciousShamirMC.cpp
 *
 */

#include "MaliciousShamirMC.h"
#include "Machines/ShamirMachine.h"

template<class T>
MaliciousShamirMC<T>::MaliciousShamirMC()
{
    this->threshold = 2 * ShamirMachine::s().threshold;
}

template<class T>
void MaliciousShamirMC<T>::init_open(const Player& P, int n)
{
    int threshold = ShamirMachine::s().threshold;
    if (reconstructions.empty())
    {
        reconstructions.resize(2 * threshold + 2);
        for (int i = threshold + 1; i <= 2 * threshold + 1; i++)
        {
            reconstructions[i] = ShamirMC<T>::get_reconstruction(P, i);
        }
    }

    ShamirMC<T>::init_open(P, n);
}

template<class T>
typename T::open_type MaliciousShamirMC<T>::finalize_raw()
{
    int threshold = ShamirMachine::s().threshold;
    shares.resize(2 * threshold + 1);
    for (size_t j = 0; j < shares.size(); j++)
        shares[j].unpack((*this->os)[this->player->get_player(j)]);
    return reconstruct(shares);
}

template<class T>
typename T::open_type MaliciousShamirMC<T>::reconstruct(
        const vector<open_type>& shares)
{
    int threshold = ShamirMachine::s().threshold;
    typename T::open_type value = 0;
    for (int j = 0; j < threshold + 1; j++)
        value += shares[j] * reconstructions[threshold + 1][j];
    for (size_t j = threshold + 2; j <= shares.size(); j++)
    {
        typename T::open_type check = 0;
        for (size_t k = 0; k < j; k++)
            check += shares[k] * reconstructions[j][k];
        if (check != value)
            throw mac_fail("inconsistent Shamir secret sharing");
    }
    return value;
}