bls12_381_g2.cpp

Go to the documentation of this file.

#include <libff/algebra/curves/bls12_381/bls12_381_g2.hpp>
 
namespace libff
{
 
#ifdef PROFILE_OP_COUNTS
long long bls12_381_G2::add_cnt = 0;
long long bls12_381_G2::dbl_cnt = 0;
#endif
 
std::vector<size_t> bls12_381_G2::wnaf_window_table;
std::vector<size_t> bls12_381_G2::fixed_base_exp_window_table;
bls12_381_G2 bls12_381_G2::G2_zero;
bls12_381_G2 bls12_381_G2::G2_one;
bls12_381_Fq2 bls12_381_G2::coeff_a;
bls12_381_Fq2 bls12_381_G2::coeff_b;
bigint<bls12_381_G2::h_limbs> bls12_381_G2::h;
 
bls12_381_G2::bls12_381_G2()
{
    this->X = G2_zero.X;
    this->Y = G2_zero.Y;
    this->Z = G2_zero.Z;
}
 
bls12_381_Fq2 bls12_381_G2::mul_by_b(const bls12_381_Fq2 &elt)
{
    return bls12_381_Fq2(
        bls12_381_twist_mul_by_b_c0 * elt.coeffs[0],
        bls12_381_twist_mul_by_b_c1 * elt.coeffs[1]);
}
 
void bls12_381_G2::print() const
{
    if (this->is_zero()) {
        printf("O\n");
    } else {
        bls12_381_G2 copy(*this);
        copy.to_affine_coordinates();
        gmp_printf(
            "(%Nd*z + %Nd , %Nd*z + %Nd)\n",
            copy.X.coeffs[1].as_bigint().data,
            bls12_381_Fq::num_limbs,
            copy.X.coeffs[0].as_bigint().data,
            bls12_381_Fq::num_limbs,
            copy.Y.coeffs[1].as_bigint().data,
            bls12_381_Fq::num_limbs,
            copy.Y.coeffs[0].as_bigint().data,
            bls12_381_Fq::num_limbs);
    }
}
 
void bls12_381_G2::print_coordinates() const
{
    if (this->is_zero()) {
        printf("O\n");
    } else {
        gmp_printf(
            "(%Nd*z + %Nd : %Nd*z + %Nd : %Nd*z + %Nd)\n",
            this->X.coeffs[1].as_bigint().data,
            bls12_381_Fq::num_limbs,
            this->X.coeffs[0].as_bigint().data,
            bls12_381_Fq::num_limbs,
            this->Y.coeffs[1].as_bigint().data,
            bls12_381_Fq::num_limbs,
            this->Y.coeffs[0].as_bigint().data,
            bls12_381_Fq::num_limbs,
            this->Z.coeffs[1].as_bigint().data,
            bls12_381_Fq::num_limbs,
            this->Z.coeffs[0].as_bigint().data,
            bls12_381_Fq::num_limbs);
    }
}
 
void bls12_381_G2::to_affine_coordinates()
{
    if (this->is_zero()) {
        this->X = bls12_381_Fq2::zero();
        this->Y = bls12_381_Fq2::one();
        this->Z = bls12_381_Fq2::zero();
    } else {
        bls12_381_Fq2 Z_inv = Z.inverse();
        bls12_381_Fq2 Z2_inv = Z_inv.squared();
        bls12_381_Fq2 Z3_inv = Z2_inv * Z_inv;
        this->X = this->X * Z2_inv;
        this->Y = this->Y * Z3_inv;
        this->Z = bls12_381_Fq2::one();
    }
}
 
void bls12_381_G2::to_special() { this->to_affine_coordinates(); }
 
bool bls12_381_G2::is_special() const
{
    return (this->is_zero() || this->Z == bls12_381_Fq2::one());
}
 
bool bls12_381_G2::is_zero() const { return (this->Z.is_zero()); }
 
bool bls12_381_G2::operator==(const bls12_381_G2 &other) const
{
    if (this->is_zero()) {
        return other.is_zero();
    }
 
    if (other.is_zero()) {
        return false;
    }
 
    /* now neither is O */
 
    // using Jacobian coordinates so:
    // (X1:Y1:Z1) = (X2:Y2:Z2)
    // iff
    // X1/Z1^2 == X2/Z2^2 and Y1/Z1^3 == Y2/Z2^3
    // iff
    // X1 * Z2^2 == X2 * Z1^2 and Y1 * Z2^3 == Y2 * Z1^3
 
    bls12_381_Fq2 Z1_squared = (this->Z).squared();
    bls12_381_Fq2 Z2_squared = (other.Z).squared();
    bls12_381_Fq2 Z1_cubed = (this->Z) * Z1_squared;
    bls12_381_Fq2 Z2_cubed = (other.Z) * Z2_squared;
    if (((this->X * Z2_squared) != (other.X * Z1_squared)) ||
        ((this->Y * Z2_cubed) != (other.Y * Z1_cubed))) {
        return false;
    }
 
    return true;
}
 
bool bls12_381_G2::operator!=(const bls12_381_G2 &other) const
{
    return !(operator==(other));
}
 
bls12_381_G2 bls12_381_G2::operator+(const bls12_381_G2 &other) const
{
    // handle special cases having to do with O
    if (this->is_zero()) {
        return other;
    }
 
    if (other.is_zero()) {
        return *this;
    }
 
    // https://www.hyperelliptic.org/EFD/g1p/data/shortw/jacobian-0/addition/add-2007-bl
    // no need to handle points of order 2,4
    // (they cannot exist in a prime-order subgroup)
 
    // check for doubling case
 
    // using Jacobian coordinates so:
    // (X1:Y1:Z1) = (X2:Y2:Z2)
    // iff
    // X1/Z1^2 == X2/Z2^2 and Y1/Z1^3 == Y2/Z2^3
    // iff
    // X1 * Z2^2 == X2 * Z1^2 and Y1 * Z2^3 == Y2 * Z1^3
 
    bls12_381_Fq2 Z1Z1 = (this->Z).squared();
    bls12_381_Fq2 Z2Z2 = (other.Z).squared();
 
    bls12_381_Fq2 U1 = this->X * Z2Z2;
    bls12_381_Fq2 U2 = other.X * Z1Z1;
 
    // S1 = Y1*Z2*Z2Z2
    bls12_381_Fq2 S1 = (this->Y) * ((other.Z) * Z2Z2);
    // S2 = Y2*Z1*Z1Z1
    bls12_381_Fq2 S2 = (other.Y) * ((this->Z) * Z1Z1);
 
    if (U1 == U2 && S1 == S2) {
        // dbl case; nothing of above can be reused
        return this->dbl();
    }
 
    // rest of add case
    // H = U2-U1
    bls12_381_Fq2 H = U2 - U1;
    // I = (2 * H)^2
    bls12_381_Fq2 I = (H + H).squared();
    // J = H * I
    bls12_381_Fq2 J = H * I;
    // r = 2 * (S2-S1)
    bls12_381_Fq2 S2_minus_S1 = S2 - S1;
    bls12_381_Fq2 r = S2_minus_S1 + S2_minus_S1;
    // V = U1 * I
    bls12_381_Fq2 V = U1 * I;
    // X3 = r^2 - J - 2 * V
    bls12_381_Fq2 X3 = r.squared() - J - (V + V);
    bls12_381_Fq2 S1_J = S1 * J;
    // Y3 = r * (V-X3)-2 S1 J
    bls12_381_Fq2 Y3 = r * (V - X3) - (S1_J + S1_J);
    // Z3 = ((Z1+Z2)^2-Z1Z1-Z2Z2) * H
    bls12_381_Fq2 Z3 = ((this->Z + other.Z).squared() - Z1Z1 - Z2Z2) * H;
 
    return bls12_381_G2(X3, Y3, Z3);
}
 
bls12_381_G2 bls12_381_G2::operator-() const
{
    return bls12_381_G2(this->X, -(this->Y), this->Z);
}
 
bls12_381_G2 bls12_381_G2::operator-(const bls12_381_G2 &other) const
{
    return (*this) + (-other);
}
 
bls12_381_G2 bls12_381_G2::add(const bls12_381_G2 &other) const
{
    return (*this) + other;
}
 
bls12_381_G2 bls12_381_G2::mixed_add(const bls12_381_G2 &other) const
{
#ifdef DEBUG
    assert(other.is_special());
#endif
 
    // handle special cases having to do with O
    if (this->is_zero()) {
        return other;
    }
 
    if (other.is_zero()) {
        return *this;
    }
 
    // no need to handle points of order 2,4
    // (they cannot exist in a prime-order subgroup)
 
    // check for doubling case
 
    // using Jacobian coordinates so:
    // (X1:Y1:Z1) = (X2:Y2:Z2)
    // iff
    // X1/Z1^2 == X2/Z2^2 and Y1/Z1^3 == Y2/Z2^3
    // iff
    // X1 * Z2^2 == X2 * Z1^2 and Y1 * Z2^3 == Y2 * Z1^3
 
    // we know that Z2 = 1
 
    const bls12_381_Fq2 Z1Z1 = (this->Z).squared();
    const bls12_381_Fq2 U2 = other.X * Z1Z1;
    const bls12_381_Fq2 S2 = (other.Y) * ((this->Z) * Z1Z1);
 
    if (this->X == U2 && this->Y == S2) {
        // dbl case; nothing of above can be reused
        return this->dbl();
    }
 
#ifdef PROFILE_OP_COUNTS
    this->add_cnt++;
#endif
 
    // NOTE: does not handle O and pts of order 2,4
    // http://www.hyperelliptic.org/EFD/g1p/auto-shortw-jacobian-0.html#addition-madd-2007-bl
    // H = U2-X1
    bls12_381_Fq2 H = U2 - (this->X);
    // HH = H&2
    bls12_381_Fq2 HH = H.squared();
    // I = 4*HH
    bls12_381_Fq2 I = HH + HH;
    I = I + I;
    // J = H*I
    bls12_381_Fq2 J = H * I;
    // r = 2*(S2-Y1)
    bls12_381_Fq2 r = S2 - (this->Y);
    r = r + r;
    // V = X1*I
    bls12_381_Fq2 V = (this->X) * I;
    // X3 = r^2-J-2*V
    bls12_381_Fq2 X3 = r.squared() - J - V - V;
    // Y3 = r*(V-X3)-2*Y1*J
    bls12_381_Fq2 Y3 = (this->Y) * J;
    Y3 = r * (V - X3) - Y3 - Y3;
    // Z3 = (Z1+H)^2-Z1Z1-HH
    bls12_381_Fq2 Z3 = ((this->Z) + H).squared() - Z1Z1 - HH;
 
    return bls12_381_G2(X3, Y3, Z3);
}
 
bls12_381_G2 bls12_381_G2::dbl() const
{
#ifdef PROFILE_OP_COUNTS
    this->dbl_cnt++;
#endif
    // handle point at infinity
    if (this->is_zero()) {
        return (*this);
    }
 
    // NOTE: does not handle O and pts of order 2,4
    // https://www.hyperelliptic.org/EFD/g1p/data/shortw/jacobian-0/doubling/dbl-2009-l
 
    // A = X1^2
    bls12_381_Fq2 A = (this->X).squared();
    // B = Y1^2
    bls12_381_Fq2 B = (this->Y).squared();
    // C = B^2
    bls12_381_Fq2 C = B.squared();
    bls12_381_Fq2 D = (this->X + B).squared() - A - C;
    // D = 2 * ((X1 + B)^2 - A - C)
    D = D + D;
    // E = 3 * A
    bls12_381_Fq2 E = A + A + A;
    // F = E^2
    bls12_381_Fq2 F = E.squared();
    // X3 = F - 2 D
    bls12_381_Fq2 X3 = F - (D + D);
    bls12_381_Fq2 eightC = C + C;
    eightC = eightC + eightC;
    eightC = eightC + eightC;
    // Y3 = E * (D - X3) - 8 * C
    bls12_381_Fq2 Y3 = E * (D - X3) - eightC;
    bls12_381_Fq2 Y1Z1 = (this->Y) * (this->Z);
    // Z3 = 2 * Y1 * Z1
    bls12_381_Fq2 Z3 = Y1Z1 + Y1Z1;
 
    return bls12_381_G2(X3, Y3, Z3);
}
 
bls12_381_G2 bls12_381_G2::mul_by_q() const
{
    return bls12_381_G2(
        bls12_381_twist_mul_by_q_X * (this->X).Frobenius_map(1),
        bls12_381_twist_mul_by_q_Y * (this->Y).Frobenius_map(1),
        (this->Z).Frobenius_map(1));
}
 
bls12_381_G2 bls12_381_G2::mul_by_cofactor() const
{
    return bls12_381_G2::h * (*this);
}
 
bool bls12_381_G2::is_well_formed() const
{
    if (this->is_zero()) {
        return true;
    }
    /*
        y^2 = x^3 + b
 
        We are using Jacobian coordinates, so equation we need to check is
       actually
 
        (y/z^3)^2 = (x/z^2)^3 + b
        y^2 / z^6 = x^3 / z^6 + b
        y^2 = x^3 + b z^6
    */
    bls12_381_Fq2 X2 = this->X.squared();
    bls12_381_Fq2 Y2 = this->Y.squared();
    bls12_381_Fq2 Z2 = this->Z.squared();
 
    bls12_381_Fq2 X3 = this->X * X2;
    bls12_381_Fq2 Z3 = this->Z * Z2;
    bls12_381_Fq2 Z6 = Z3.squared();
 
    return (Y2 == X3 + bls12_381_twist_coeff_b * Z6);
}
 
bool bls12_381_G2::is_in_safe_subgroup() const
{
    return zero() == scalar_field::mod * (*this);
}
 
const bls12_381_G2 &bls12_381_G2::zero() { return G2_zero; }
 
const bls12_381_G2 &bls12_381_G2::one() { return G2_one; }
 
bls12_381_G2 bls12_381_G2::random_element()
{
    return (bls12_381_Fr::random_element().as_bigint()) * G2_one;
}
 
void bls12_381_G2::write_uncompressed(std::ostream &out) const
{
    bls12_381_G2 copy(*this);
    copy.to_affine_coordinates();
    out << (copy.is_zero() ? 1 : 0) << OUTPUT_SEPARATOR;
    out << copy.X << OUTPUT_SEPARATOR << copy.Y;
}
 
void bls12_381_G2::write_compressed(std::ostream &out) const
{
    bls12_381_G2 copy(*this);
    copy.to_affine_coordinates();
    out << (copy.is_zero() ? 1 : 0) << OUTPUT_SEPARATOR;
    /* storing LSB of Y */
    out << copy.X << OUTPUT_SEPARATOR
        << (copy.Y.coeffs[0].as_bigint().data[0] & 1);
}
 
void bls12_381_G2::read_uncompressed(std::istream &in, bls12_381_G2 &g)
{
    char is_zero;
    bls12_381_Fq2 tX, tY;
    in >> is_zero >> tX >> tY;
    is_zero -= '0';
 
    if (!is_zero) {
        g.X = tX;
        g.Y = tY;
        g.Z = bls12_381_Fq2::one();
    } else {
        g = bls12_381_G2::zero();
    }
}
 
void bls12_381_G2::read_compressed(std::istream &in, bls12_381_G2 &g)
{
    char is_zero;
    bls12_381_Fq2 tX, tY;
    // this reads is_zero;
    in.read((char *)&is_zero, 1);
    is_zero -= '0';
    consume_OUTPUT_SEPARATOR(in);
 
    unsigned char Y_lsb;
    in >> tX;
    consume_OUTPUT_SEPARATOR(in);
    in.read((char *)&Y_lsb, 1);
    Y_lsb -= '0';
 
    // y = +/- sqrt(x^3 + b)
    if (!is_zero) {
        bls12_381_Fq2 tX2 = tX.squared();
        bls12_381_Fq2 tY2 = tX2 * tX + bls12_381_twist_coeff_b;
        tY = tY2.sqrt();
 
        if ((tY.coeffs[0].as_bigint().data[0] & 1) != Y_lsb) {
            tY = -tY;
        }
 
        g.X = tX;
        g.Y = tY;
        g.Z = bls12_381_Fq2::one();
    } else {
        g = bls12_381_G2::zero();
    }
}
 
std::ostream &operator<<(std::ostream &out, const bls12_381_G2 &g)
{
#ifdef NO_PT_COMPRESSION
    g.write_uncompressed(out);
#else
    g.write_compressed(out);
#endif
    return out;
}
 
std::istream &operator>>(std::istream &in, bls12_381_G2 &g)
{
#ifdef NO_PT_COMPRESSION
    bls12_381_G2::read_uncompressed(in, g);
#else
    bls12_381_G2::read_compressed(in, g);
#endif
    return in;
}
 
void bls12_381_G2::batch_to_special_all_non_zeros(
    std::vector<bls12_381_G2> &vec)
{
    std::vector<bls12_381_Fq2> Z_vec;
    Z_vec.reserve(vec.size());
 
    for (auto &el : vec) {
        Z_vec.emplace_back(el.Z);
    }
    batch_invert<bls12_381_Fq2>(Z_vec);
 
    const bls12_381_Fq2 one = bls12_381_Fq2::one();
 
    for (size_t i = 0; i < vec.size(); ++i) {
        bls12_381_Fq2 Z2 = Z_vec[i].squared();
        bls12_381_Fq2 Z3 = Z_vec[i] * Z2;
 
        vec[i].X = vec[i].X * Z2;
        vec[i].Y = vec[i].Y * Z3;
        vec[i].Z = one;
    }
}
 
} // namespace libff