/* LibTomCrypt, modular cryptographic library -- Tom St Denis */
/* SPDX-License-Identifier: Unlicense */

#include "tomcrypt_private.h"

/**
  @file ltc_ecc_projective_add_point.c
  ECC Crypto, Tom St Denis
*/

#if defined(LTC_MECC) && (!defined(LTC_MECC_ACCEL) || defined(LTM_DESC))

/**
   Add two ECC points
   @param P        The point to add
   @param Q        The point to add
   @param R        [out] The destination of the double
   @param ma       ECC curve parameter a in montgomery form
   @param modulus  The modulus of the field the ECC curve is in
   @param mp       The "b" value from montgomery_setup()
   @return CRYPT_OK on success
*/
int ltc_ecc_projective_add_point(const ecc_point *P, const ecc_point *Q, ecc_point *R,
                                 const void *ma, const void *modulus, void *mp)
{
   void  *t1, *t2, *x, *y, *z;
   int    err, inf;

   LTC_ARGCHK(P       != NULL);
   LTC_ARGCHK(Q       != NULL);
   LTC_ARGCHK(R       != NULL);
   LTC_ARGCHK(modulus != NULL);
   LTC_ARGCHK(mp      != NULL);

   if ((err = ltc_mp_init_multi(&t1, &t2, &x, &y, &z, LTC_NULL)) != CRYPT_OK) {
      return err;
   }

   /* treat internal Z == 0 states as infinity (for elliptic-curve addition: infinity = identity element) */
   if (P->z != NULL && ltc_mp_iszero(P->z)) {
      if (Q->z != NULL && ltc_mp_iszero(Q->z)) {
         /* inf + inf = inf */
         err = ltc_ecc_set_point_xyz(1, 1, 0, R);
      }
      else {
         /* inf + Q = Q */
         err = ltc_ecc_copy_point(Q, R);
      }
      goto done;
   }
   if (Q->z != NULL && ltc_mp_iszero(Q->z)) {
      /* P + inf = P */
      err = ltc_ecc_copy_point(P, R);
      goto done;
   }

   if ((err = ltc_ecc_is_point_at_infinity(P, modulus, &inf)) != CRYPT_OK) return err;
   if (inf) {
      /* P is point at infinity >> Result = Q */
      err = ltc_ecc_copy_point(Q, R);
      goto done;
   }

   if ((err = ltc_ecc_is_point_at_infinity(Q, modulus, &inf)) != CRYPT_OK) return err;
   if (inf) {
      /* Q is point at infinity >> Result = P */
      err = ltc_ecc_copy_point(P, R);
      goto done;
   }

   if ((ltc_mp_cmp(P->x, Q->x) == LTC_MP_EQ) && (ltc_mp_cmp(P->z, Q->z) == LTC_MP_EQ)) {
      if (ltc_mp_cmp(P->y, Q->y) == LTC_MP_EQ) {
dbl_point:
         /* here P = Q >> Result = 2 * P (use doubling) */
         ltc_mp_deinit_multi(t1, t2, x, y, z, LTC_NULL);
         return ltc_ecc_projective_dbl_point(P, R, ma, modulus, mp);
      }
      if ((err = ltc_mp_sub(modulus, Q->y, t1)) != CRYPT_OK)                       { goto done; }
      if (ltc_mp_cmp(P->y, t1) == LTC_MP_EQ) {
         /* here Q = -P >>> Result = the point at infinity */
         err = ltc_ecc_set_point_xyz(1, 1, 0, R);
         goto done;
      }
   }

   if ((err = ltc_mp_copy(P->x, x)) != CRYPT_OK)                                   { goto done; }
   if ((err = ltc_mp_copy(P->y, y)) != CRYPT_OK)                                   { goto done; }
   if ((err = ltc_mp_copy(P->z, z)) != CRYPT_OK)                                   { goto done; }

   /* if Z is one then these are no-operations */
   if (Q->z != NULL) {
      /* T1 = Z' * Z' */
      if ((err = ltc_mp_sqr(Q->z, t1)) != CRYPT_OK)                                { goto done; }
      if ((err = ltc_mp_montgomery_reduce(t1, modulus, mp)) != CRYPT_OK)           { goto done; }
      /* X = X * T1 */
      if ((err = ltc_mp_mul(t1, x, x)) != CRYPT_OK)                                { goto done; }
      if ((err = ltc_mp_montgomery_reduce(x, modulus, mp)) != CRYPT_OK)            { goto done; }
      /* T1 = Z' * T1 */
      if ((err = ltc_mp_mul(Q->z, t1, t1)) != CRYPT_OK)                            { goto done; }
      if ((err = ltc_mp_montgomery_reduce(t1, modulus, mp)) != CRYPT_OK)           { goto done; }
      /* Y = Y * T1 */
      if ((err = ltc_mp_mul(t1, y, y)) != CRYPT_OK)                                { goto done; }
      if ((err = ltc_mp_montgomery_reduce(y, modulus, mp)) != CRYPT_OK)            { goto done; }
   }

   /* T1 = Z*Z */
   if ((err = ltc_mp_sqr(z, t1)) != CRYPT_OK)                                      { goto done; }
   if ((err = ltc_mp_montgomery_reduce(t1, modulus, mp)) != CRYPT_OK)              { goto done; }
   /* T2 = X' * T1 */
   if ((err = ltc_mp_mul(Q->x, t1, t2)) != CRYPT_OK)                               { goto done; }
   if ((err = ltc_mp_montgomery_reduce(t2, modulus, mp)) != CRYPT_OK)              { goto done; }
   /* T1 = Z * T1 */
   if ((err = ltc_mp_mul(z, t1, t1)) != CRYPT_OK)                                  { goto done; }
   if ((err = ltc_mp_montgomery_reduce(t1, modulus, mp)) != CRYPT_OK)              { goto done; }
   /* T1 = Y' * T1 */
   if ((err = ltc_mp_mul(Q->y, t1, t1)) != CRYPT_OK)                               { goto done; }
   if ((err = ltc_mp_montgomery_reduce(t1, modulus, mp)) != CRYPT_OK)              { goto done; }

   /* Same-affine dispatch: A==B means P,Q share affine x; dispatch to dbl or O before the formulas */
   if (ltc_mp_cmp(x, t2) == LTC_MP_EQ) {
      if (ltc_mp_cmp(y, t1) == LTC_MP_EQ) goto dbl_point;
      err = ltc_ecc_set_point_xyz(1, 1, 0, R);
      goto done;
   }

   /* Y = Y - T1 */
   if ((err = ltc_mp_sub(y, t1, y)) != CRYPT_OK)                                   { goto done; }
   if (ltc_mp_cmp_d(y, 0) == LTC_MP_LT) {
      if ((err = ltc_mp_add(y, modulus, y)) != CRYPT_OK)                           { goto done; }
   }
   /* T1 = 2T1 */
   if ((err = ltc_mp_add(t1, t1, t1)) != CRYPT_OK)                                 { goto done; }
   if (ltc_mp_cmp(t1, modulus) != LTC_MP_LT) {
      if ((err = ltc_mp_sub(t1, modulus, t1)) != CRYPT_OK)                         { goto done; }
   }
   /* T1 = Y + T1 */
   if ((err = ltc_mp_add(t1, y, t1)) != CRYPT_OK)                                  { goto done; }
   if (ltc_mp_cmp(t1, modulus) != LTC_MP_LT) {
      if ((err = ltc_mp_sub(t1, modulus, t1)) != CRYPT_OK)                         { goto done; }
   }
   /* X = X - T2 */
   if ((err = ltc_mp_sub(x, t2, x)) != CRYPT_OK)                                   { goto done; }
   if (ltc_mp_cmp_d(x, 0) == LTC_MP_LT) {
      if ((err = ltc_mp_add(x, modulus, x)) != CRYPT_OK)                           { goto done; }
   }
   /* T2 = 2T2 */
   if ((err = ltc_mp_add(t2, t2, t2)) != CRYPT_OK)                                 { goto done; }
   if (ltc_mp_cmp(t2, modulus) != LTC_MP_LT) {
      if ((err = ltc_mp_sub(t2, modulus, t2)) != CRYPT_OK)                         { goto done; }
   }
   /* T2 = X + T2 */
   if ((err = ltc_mp_add(t2, x, t2)) != CRYPT_OK)                                  { goto done; }
   if (ltc_mp_cmp(t2, modulus) != LTC_MP_LT) {
      if ((err = ltc_mp_sub(t2, modulus, t2)) != CRYPT_OK)                         { goto done; }
   }

   /* if Z' != 1 */
   if (Q->z != NULL) {
      /* Z = Z * Z' */
      if ((err = ltc_mp_mul(z, Q->z, z)) != CRYPT_OK)                              { goto done; }
      if ((err = ltc_mp_montgomery_reduce(z, modulus, mp)) != CRYPT_OK)            { goto done; }
   }

   /* Z = Z * X */
   if ((err = ltc_mp_mul(z, x, z)) != CRYPT_OK)                                    { goto done; }
   if ((err = ltc_mp_montgomery_reduce(z, modulus, mp)) != CRYPT_OK)               { goto done; }

   /* T1 = T1 * X  */
   if ((err = ltc_mp_mul(t1, x, t1)) != CRYPT_OK)                                  { goto done; }
   if ((err = ltc_mp_montgomery_reduce(t1, modulus, mp)) != CRYPT_OK)              { goto done; }
   /* X = X * X */
   if ((err = ltc_mp_sqr(x, x)) != CRYPT_OK)                                       { goto done; }
   if ((err = ltc_mp_montgomery_reduce(x, modulus, mp)) != CRYPT_OK)               { goto done; }
   /* T2 = T2 * x */
   if ((err = ltc_mp_mul(t2, x, t2)) != CRYPT_OK)                                  { goto done; }
   if ((err = ltc_mp_montgomery_reduce(t2, modulus, mp)) != CRYPT_OK)              { goto done; }
   /* T1 = T1 * X  */
   if ((err = ltc_mp_mul(t1, x, t1)) != CRYPT_OK)                                  { goto done; }
   if ((err = ltc_mp_montgomery_reduce(t1, modulus, mp)) != CRYPT_OK)              { goto done; }

   /* X = Y*Y */
   if ((err = ltc_mp_sqr(y, x)) != CRYPT_OK)                                       { goto done; }
   if ((err = ltc_mp_montgomery_reduce(x, modulus, mp)) != CRYPT_OK)               { goto done; }
   /* X = X - T2 */
   if ((err = ltc_mp_sub(x, t2, x)) != CRYPT_OK)                                   { goto done; }
   if (ltc_mp_cmp_d(x, 0) == LTC_MP_LT) {
      if ((err = ltc_mp_add(x, modulus, x)) != CRYPT_OK)                           { goto done; }
   }

   /* T2 = T2 - X */
   if ((err = ltc_mp_sub(t2, x, t2)) != CRYPT_OK)                                  { goto done; }
   if (ltc_mp_cmp_d(t2, 0) == LTC_MP_LT) {
      if ((err = ltc_mp_add(t2, modulus, t2)) != CRYPT_OK)                         { goto done; }
   }
   /* T2 = T2 - X */
   if ((err = ltc_mp_sub(t2, x, t2)) != CRYPT_OK)                                  { goto done; }
   if (ltc_mp_cmp_d(t2, 0) == LTC_MP_LT) {
      if ((err = ltc_mp_add(t2, modulus, t2)) != CRYPT_OK)                         { goto done; }
   }
   /* T2 = T2 * Y */
   if ((err = ltc_mp_mul(t2, y, t2)) != CRYPT_OK)                                  { goto done; }
   if ((err = ltc_mp_montgomery_reduce(t2, modulus, mp)) != CRYPT_OK)              { goto done; }
   /* Y = T2 - T1 */
   if ((err = ltc_mp_sub(t2, t1, y)) != CRYPT_OK)                                  { goto done; }
   if (ltc_mp_cmp_d(y, 0) == LTC_MP_LT) {
      if ((err = ltc_mp_add(y, modulus, y)) != CRYPT_OK)                           { goto done; }
   }
   /* Y = Y/2 */
   if (ltc_mp_isodd(y)) {
      if ((err = ltc_mp_add(y, modulus, y)) != CRYPT_OK)                           { goto done; }
   }
   if ((err = ltc_mp_div_2(y, y)) != CRYPT_OK)                                     { goto done; }

   if ((err = ltc_mp_copy(x, R->x)) != CRYPT_OK)                                   { goto done; }
   if ((err = ltc_mp_copy(y, R->y)) != CRYPT_OK)                                   { goto done; }
   if ((err = ltc_mp_copy(z, R->z)) != CRYPT_OK)                                   { goto done; }

   err = CRYPT_OK;
done:
   ltc_mp_deinit_multi(t1, t2, x, y, z, LTC_NULL);
   return err;
}

#endif
