BGN2/bgn2/src/Scalar.cpp

#include "Scalar.hpp"
#include <iostream>

extern const scalar_t bn_n;
mpz_class Scalar::mpz_bn_p = 0;
mpz_class Scalar::mpz_bn_n = 0;

Scalar::Scalar()
{
    element = 0;
}

Scalar::Scalar(const scalar_t& input)
{
    set(input);
}

Scalar::Scalar(mpz_class input)
{
    set(input);
}

void Scalar::init()
{
    mpz_bn_p = mpz_class("8FB501E34AA387F9AA6FECB86184DC21EE5B88D120B5B59E185CAC6C5E089667", 16);
    mpz_bn_n = mpz_class("8FB501E34AA387F9AA6FECB86184DC212E8D8E12F82B39241A2EF45B57AC7261", 16);
}

void Scalar::set(const scalar_t& input)
{
    std::stringstream bufferstream;
    std::string buffer;
    mpz_class temp;

    bufferstream << std::hex << input[3] << input[2] << input[1] << input[0];
    bufferstream >> buffer;
    
    temp.set_str(buffer, 16);

    mpz_mod(temp.get_mpz_t(), temp.get_mpz_t(), mpz_bn_p.get_mpz_t());

    element = temp;
}

void Scalar::set(mpz_class input)
{
    mpz_class temp = input;

    mpz_mod(temp.get_mpz_t(), temp.get_mpz_t(), mpz_bn_p.get_mpz_t());

    element = temp;
}

void Scalar::set_random()
{
    scalar_t temp;
    
    /* When we ask for a random number,
     * we really mean a seed to find a random element of a group
     * and the order of the curve either is bn_n or is divided by it
     * (not bn_p) */
    scalar_setrandom(temp, bn_n);

    set(temp);
}

mpz_class Scalar::toInt() const
{
    return element;
}

Scalar Scalar::operator+(const Scalar& b) const
{
    mpz_class temp = element + b.element;

    mpz_mod(temp.get_mpz_t(), temp.get_mpz_t(), mpz_bn_p.get_mpz_t());

    return Scalar(temp);
}

Scalar Scalar::operator-(const Scalar& b) const
{
    mpz_class temp = element - b.element;

    mpz_mod(temp.get_mpz_t(), temp.get_mpz_t(), mpz_bn_p.get_mpz_t());

    return Scalar(temp);
}

Scalar Scalar::operator*(const Scalar& b) const
{
    mpz_class temp = element * b.element;

    mpz_mod(temp.get_mpz_t(), temp.get_mpz_t(), mpz_bn_p.get_mpz_t());

    return Scalar(temp);
}

Scalar Scalar::operator/(const Scalar& b) const
{
    mpz_class temp;
    mpz_invert(temp.get_mpz_t(), b.element.get_mpz_t(), mpz_bn_p.get_mpz_t());

    temp *= element;
    mpz_mod(temp.get_mpz_t(), temp.get_mpz_t(), mpz_bn_p.get_mpz_t());

    return Scalar(temp);
}

Scalar& Scalar::operator++()
{
    element++;
    mpz_mod(element.get_mpz_t(), element.get_mpz_t(), mpz_bn_p.get_mpz_t());

    return *this;
}

Scalar Scalar::operator++(int)
{
    Scalar retval = *this;
    
    element++;
    mpz_mod(element.get_mpz_t(), element.get_mpz_t(), mpz_bn_p.get_mpz_t());

    return retval;
}

Scalar& Scalar::operator--()
{
    element--;
    mpz_mod(element.get_mpz_t(), element.get_mpz_t(), mpz_bn_p.get_mpz_t());

    return *this;
}

Scalar Scalar::operator--(int)
{
    Scalar retval = *this;
    
    element--;
    mpz_mod(element.get_mpz_t(), element.get_mpz_t(), mpz_bn_p.get_mpz_t());

    return retval;
}

Scalar Scalar::curveAdd(const Scalar& b) const
{
    mpz_class temp = element + b.element;

    mpz_mod(temp.get_mpz_t(), temp.get_mpz_t(), mpz_bn_n.get_mpz_t());

    return Scalar(temp);
}

Scalar Scalar::curveSub(const Scalar& b) const
{
    mpz_class temp = element - b.element;

    mpz_mod(temp.get_mpz_t(), temp.get_mpz_t(), mpz_bn_n.get_mpz_t());

    return Scalar(temp);
}

Scalar Scalar::curveMult(const Scalar& b) const
{
    mpz_class temp = element * b.element;

    mpz_mod(temp.get_mpz_t(), temp.get_mpz_t(), mpz_bn_n.get_mpz_t());

    return Scalar(temp);
}

Scalar Scalar::curveInverse() const
{
    mpz_class temp;
    mpz_invert(temp.get_mpz_t(), element.get_mpz_t(), mpz_bn_n.get_mpz_t());

    return Scalar(temp);
}

void Scalar::mult(curvepoint_fp_t rop, const curvepoint_fp_t& op1) const
{
    SecretScalar secret_element = to_scalar_t();

    curvepoint_fp_scalarmult_vartime(rop, op1, secret_element.expose());
}

void Scalar::mult(twistpoint_fp2_t rop, const twistpoint_fp2_t& op1) const
{
    SecretScalar secret_element = to_scalar_t();

    twistpoint_fp2_scalarmult_vartime(rop, op1, secret_element.expose());
}

void Scalar::mult(fp12e_t rop, const fp12e_t& op1) const
{
    SecretScalar secret_element = to_scalar_t();

    fp12e_pow_vartime(rop, op1, secret_element.expose());
}

bool Scalar::operator==(const Scalar& b) const
{
    return element == b.element;
}

bool Scalar::operator<(const Scalar& b) const
{
    return element < b.element;
}

bool Scalar::operator<=(const Scalar& b) const
{
    return element <= b.element;
}

bool Scalar::operator>(const Scalar& b) const
{
    return element > b.element;
}

bool Scalar::operator>=(const Scalar& b) const
{
    return element >= b.element;
}

bool Scalar::operator!=(const Scalar& b) const
{
    return element != b.element;
}

Scalar::SecretScalar::SecretScalar()
{ }

Scalar::SecretScalar::SecretScalar(const Scalar& input)
{
    set(input.element);
}

Scalar::SecretScalar::SecretScalar(mpz_class input)
{
    set(input);
}

const scalar_t& Scalar::SecretScalar::expose() const
{
    return element;
}

void Scalar::SecretScalar::set(mpz_class input)
{
    std::stringstream buffer;
    char temp[17];
    buffer << std::setfill('0') << std::setw(64) << input.get_str(16);

    for (int i = 3; i >= 0; i--)
    {
        buffer.get(temp, 17);
        element[i] = strtoull(temp, NULL, 16);
    }
}
    
Scalar::SecretScalar Scalar::to_scalar_t() const
{
    return SecretScalar(element);
}

std::ostream& operator<<(std::ostream& os, const Scalar& output)
{
    os << output.element;
    return os;
}

std::istream& operator>>(std::istream& is, Scalar& input)
{
    is >> input.element;
    return is;
}