/*
* Arithmetic for point groups of elliptic curves over GF(p)
*
* (C) 2007 Martin Doering, Christoph Ludwig, Falko Strenzke
*     2008-2010 Jack Lloyd
*
* Distributed under the terms of the Botan license
*/

#include <botan/point_gfp.h>
#include <botan/gfp_element.h>
#include <botan/numthry.h>

namespace Botan {

namespace {

BigInt decompress_point(bool yMod2,
                        const BigInt& x,
                        const CurveGFp& curve)
   {
   BigInt xpow3 = x * x * x;

   BigInt g = curve.get_a() * x;
   g += xpow3;
   g += curve.get_b();
   g = g % curve.get_p();

   BigInt z = ressol(g, curve.get_p());

   if(z < 0)
      throw Illegal_Point("error during decompression");

   if(z.get_bit(0) != yMod2)
      z = curve.get_p() - z;

   return z;
   }

}

// arithmetic operators
PointGFp& PointGFp::operator+=(const PointGFp& rhs)
   {
   if(rhs.is_zero())
      return *this;

   if(is_zero())
      {
      *this = rhs;
      return *this;
      }

   GFpElement point_x(curve.get_p(), coord_x);
   GFpElement point_y(curve.get_p(), coord_y);
   GFpElement point_z(curve.get_p(), coord_z);

   GFpElement rhs_point_x(curve.get_p(), rhs.coord_x);
   GFpElement rhs_point_y(curve.get_p(), rhs.coord_y);
   GFpElement rhs_point_z(curve.get_p(), rhs.coord_z);

   GFpElement U1 = point_x;
   GFpElement S1 = point_y;

   GFpElement rhs_z2 = rhs_point_z * rhs_point_z;
   U1 *= rhs_z2;
   S1 *= rhs_z2 * rhs_point_z;

   GFpElement U2 = rhs_point_x;
   GFpElement S2 = rhs_point_y;

   GFpElement lhs_z2 = point_z * point_z;
   U2 *= lhs_z2;
   S2 *= lhs_z2 * point_z;

   GFpElement H(U2 - U1);
   GFpElement r(S2 - S1);

   if(H.is_zero())
      {
      if(r.is_zero())
         {
         mult2_in_place();
         return *this;
         }

      *this = PointGFp(curve); // setting myself to zero
      return *this;
      }

   U2 = H * H;

   S2 = U2 * H;

   U2 *= U1;

   GFpElement x(r*r - S2 - (U2+U2));

   GFpElement z(S1 * S2);

   GFpElement y(r * (U2-x) - z);

   z = (point_z * rhs_point_z) * H;

   coord_x = x.get_value();
   coord_y = y.get_value();
   coord_z = z.get_value();

   return *this;
   }

PointGFp& PointGFp::operator-=(const PointGFp& rhs)
   {
   PointGFp minus_rhs = PointGFp(rhs).negate();

   if(is_zero())
      *this = minus_rhs;
   else
      *this += minus_rhs;

   return *this;
   }

PointGFp& PointGFp::operator*=(const BigInt& scalar)
   {
   if(scalar == 0)
      {
      *this = PointGFp(curve);
      return *this;
      }
   else if(scalar == 1)
      return *this;
   else if(scalar == -1)
      {
      this->negate();
      return *this;
      }

   PointGFp H(this->curve); // create as zero
   PointGFp P(*this);

   if(scalar.is_negative())
      P.negate();

   for(int i = scalar.bits() - 1; i >= 0; --i)
      {
      H.mult2_in_place();
      if(scalar.get_bit(i))
         H += P;
      }

   if(!H.is_zero()) // cannot convert if H == O
      {
      /**
      * Convert H to an equivalent point with z == 1, thus x and y
      * correspond to their affine coordinates
      */
      if(H.coord_z != 1)
         {
         GFpElement point_x(curve.get_p(), H.coord_x);
         GFpElement point_y(curve.get_p(), H.coord_y);
         GFpElement point_z(curve.get_p(), H.coord_z);

         // Converts to affine coordinates
         GFpElement z = inverse(point_z);
         GFpElement z2 = z * z;
         z *= z2;
         GFpElement x = point_x * z2;
         GFpElement y = point_y * z;

         H.coord_x = x.get_value();
         H.coord_y = y.get_value();
         H.coord_z = 1;
         }
      }

   *this = H;
   return *this;
   }

PointGFp& PointGFp::negate()
   {
   if(!is_zero())
      coord_y = curve.get_p() - coord_y;

   return *this;
   }

// *this *= 2
PointGFp& PointGFp::mult2_in_place()
   {
   if(is_zero())
      return *this;
   else if(coord_y.is_zero())
      {
      *this = PointGFp(curve); // setting myself to zero
      return *this;
      }

   Modular_Reducer mod_p(curve.get_p());

   BigInt y_2 = mod_p.square(coord_y);

   BigInt S = mod_p.multiply(4, mod_p.multiply(coord_x, y_2));

   BigInt a_z4 = mod_p.multiply(curve.get_a(),
                                mod_p.square(mod_p.square(coord_z)));

   BigInt M = mod_p.reduce(a_z4 + 3 * mod_p.square(coord_x));

   BigInt x = mod_p.reduce(mod_p.square(M) - mod_p.multiply(2, S));

   BigInt y = mod_p.square(y_2);

   BigInt z = mod_p.multiply(2, mod_p.reduce(y + y));

   BigInt U = mod_p.reduce(z + z);

   y = mod_p.reduce(mod_p.multiply(M, S - x) - U);

   z = mod_p.multiply(2, mod_p.multiply(coord_y, coord_z));

   coord_x = x;
   coord_y = y;
   coord_z = z;

   return *this;
   }

BigInt PointGFp::get_affine_x() const
   {
   if(is_zero())
      throw Illegal_Transformation("cannot convert to affine");

   GFpElement point_x(curve.get_p(), coord_x);
   GFpElement point_z(curve.get_p(), coord_z);

   GFpElement z2 = point_z * point_z;
   z2.inverse_in_place();
   z2 *= point_x;

   return z2.get_value();
   }

BigInt PointGFp::get_affine_y() const
   {
   if(is_zero())
      throw Illegal_Transformation("cannot convert to affine");

   GFpElement point_y(curve.get_p(), coord_y);
   GFpElement point_z(curve.get_p(), coord_z);

   GFpElement z3 = point_z * point_z * point_z;
   z3.inverse_in_place();
   z3 *= point_y;

   return z3.get_value();
   }

// Is this the point at infinity?
bool PointGFp::is_zero() const
   {
   return(coord_x.is_zero() && coord_z.is_zero());
   }

void PointGFp::check_invariants() const
   {
   /*
   Is the point still on the curve?? (If everything is correct, the
   point is always on its curve; then the function will return
   silently. If Oskar managed to corrupt this object's state, then it
   will throw an exception.)
   */

   if(is_zero())
      return;

   GFpElement point_x(curve.get_p(), coord_x);
   GFpElement point_y(curve.get_p(), coord_y);
   GFpElement point_z(curve.get_p(), coord_z);

   const GFpElement y2 = point_y * point_y;
   const GFpElement x3 = point_x * point_x * point_x;

   if(coord_z == BigInt(1))
      {
      GFpElement ax(curve.get_p(), curve.get_a());
      ax *= point_x;

      GFpElement b(curve.get_p(), curve.get_b());

      if(y2 != (x3 + ax + b))
         throw Illegal_Point();
      }

   GFpElement Zpow2 = point_z * point_z;
   GFpElement Zpow3 = Zpow2 * point_z;
   GFpElement AZpow4 = Zpow3 * point_z * GFpElement(curve.get_p(), curve.get_a());
   const GFpElement aXZ4 = AZpow4 * point_x;
   const GFpElement bZ6 = GFpElement(curve.get_p(), curve.get_b()) * Zpow3 * Zpow3;

   if(y2 != (x3 + aXZ4 + bZ6))
      throw Illegal_Point();
   }

// swaps the states of *this and other, does not throw!
void PointGFp::swap(PointGFp& other)
   {
   curve.swap(other.curve);
   coord_x.swap(other.coord_x);
   coord_y.swap(other.coord_y);
   coord_z.swap(other.coord_z);
   }

bool PointGFp::operator==(const PointGFp& other) const
   {
   return (coord_x == other.coord_x &&
           coord_y == other.coord_y &&
           coord_z == other.coord_z &&
           get_curve() == other.get_curve());
   }

// arithmetic operators
PointGFp operator+(const PointGFp& lhs, PointGFp const& rhs)
   {
   PointGFp tmp(lhs);
   return tmp += rhs;
   }

PointGFp operator-(const PointGFp& lhs, PointGFp const& rhs)
   {
   PointGFp tmp(lhs);
   return tmp -= rhs;
   }

PointGFp operator-(const PointGFp& lhs)
   {
   return PointGFp(lhs).negate();
   }

PointGFp operator*(const BigInt& scalar, const PointGFp& point)
   {
   PointGFp result(point);
   return result *= scalar;
   }

PointGFp operator*(const PointGFp& point, const BigInt& scalar)
   {
   PointGFp result(point);
   return result *= scalar;
   }

// encoding and decoding
SecureVector<byte> EC2OSP(const PointGFp& point, byte format)
   {
   if(point.is_zero())
      return SecureVector<byte>(1); // single 0 byte

   const u32bit p_bytes = point.get_curve().get_p().bytes();

   BigInt x = point.get_affine_x();
   BigInt y = point.get_affine_y();

   SecureVector<byte> bX = BigInt::encode_1363(x, p_bytes);
   SecureVector<byte> bY = BigInt::encode_1363(y, p_bytes);

   if(format == PointGFp::UNCOMPRESSED)
      {
      SecureVector<byte> result(2*p_bytes+1);
      result[0] = 4;

      result.copy(1, bX.begin(), p_bytes);
      result.copy(p_bytes+1, bY.begin(), p_bytes);
      return result;
      }
   else if(format == PointGFp::COMPRESSED)
      {
      SecureVector<byte> result(p_bytes+1);
      result[0] = 2;

      result.copy(1, bX.begin(), bX.size());

      if(y.get_bit(0))
         result[0] |= 1;

      return result;
      }
   else if(format == PointGFp::HYBRID)
      {
      SecureVector<byte> result(2*p_bytes+1);
      result[0] = 6;

      result.copy(1, bX.begin(), bX.size());
      result.copy(p_bytes+1, bY.begin(), bY.size());

      if(y.get_bit(0))
         result[0] |= 1;

      return result;
      }
   else
      throw Invalid_Argument("illegal point encoding format specification");
   }

PointGFp OS2ECP(const MemoryRegion<byte>& os, const CurveGFp& curve)
   {
   if(os.size() == 1 && os[0] == 0)
      return PointGFp(curve); // return zero

   const byte pc = os[0];

   BigInt x, y;

   if(pc == 2 || pc == 3)
      {
      //compressed form
      x = BigInt::decode(&os[1], os.size() - 1);

      bool yMod2 = ((pc & 0x01) == 1);
      y = decompress_point(yMod2, x, curve);
      }
   else if(pc == 4)
      {
      // uncompressed form
      u32bit l = (os.size() - 1) / 2;

      x = BigInt::decode(&os[1], l);
      y = BigInt::decode(&os[l+1], l);
      }
   else if(pc == 6 || pc == 7)
      {
      // hybrid form
      u32bit l = (os.size() - 1) / 2;

      x = BigInt::decode(&os[1], l);
      y = BigInt::decode(&os[l+1], l);

      bool yMod2 = ((pc & 0x01) == 1);

      if(decompress_point(yMod2, x, curve) != y)
         throw Illegal_Point("OS2ECP: Decoding error in hybrid format");
      }
   else
      throw Invalid_Argument("OS2ECP: Unknown format type");

   PointGFp result(curve, x, y);
   result.check_invariants();
   return result;
   }

PointGFp create_random_point(RandomNumberGenerator& rng,
                             const CurveGFp& curve)
   {
   const BigInt& p = curve.get_p();

   while(true)
      {
      BigInt r(rng, p.bits());

      GFpElement x = GFpElement(p, r);
      GFpElement x3 = x * x * x;

      GFpElement ax(curve.get_p(), curve.get_a());
      ax *= x;

      GFpElement bx3(curve.get_p(), curve.get_b());
      bx3 *= x3;

      GFpElement y = ax + bx3;

      if(ressol(y.get_value(), p) > 0)
         return PointGFp(curve, x.get_value(), y.get_value());
      }
   }

}