/***********************************************************************************
* FourQlib: a high-performance crypto library based on the elliptic curve FourQ
*
*    Copyright (c) Microsoft Corporation. All rights reserved.
*
* Abstract: crypto utility functions
************************************************************************************/ 

#include "FourQ_internal.h"
#include "FourQ_params.h"
#include <string.h>

static digit_t mask4000 = (digit_t)1 << (sizeof(digit_t)*8 - 2);
static digit_t mask7fff = (digit_t)(-1) >> 1;


bool is_zero_ct(digit_t* a, unsigned int nwords)
{ // Check if multiprecision element is zero
    digit_t x;
    unsigned int i;

    x = a[0];
    for (i = 1; i < nwords; i++) {
        x |= a[i];
    }

    return (bool)(1 ^ ((x | (0-x)) >> (RADIX-1)));
}


void encode(point_t P, unsigned char* Pencoded)
{ // Encode point P
  // SECURITY NOTE: this function does not run in constant time.
    digit_t temp1 = (P->x[1][NWORDS_FIELD-1] & mask4000) << 1;
    digit_t temp2 = (P->x[0][NWORDS_FIELD-1] & mask4000) << 1;

    memmove(Pencoded, P->y, 32);
    if (is_zero_ct((digit_t*)P->x, NWORDS_FIELD) == true) {
        ((digit_t*)Pencoded)[2*NWORDS_FIELD-1] |= temp1;
    } else {
        ((digit_t*)Pencoded)[2*NWORDS_FIELD-1] |= temp2;
    }
}


ECCRYPTO_STATUS decode(const unsigned char* Pencoded, point_t P)
{ // Decode point P
  // SECURITY NOTE: this function does not run in constant time.
    velm_t r, t, t0, t1, t2, t3, t4, VPx0, VPx1;
    v2elm_t u, v, u2, v2, one = {0};
    digit_t sign_dec;
    vpoint_extproj_t R;
    vpoint_t VP;
    unsigned int i, sign;

    one[0] = 1;
    memmove((unsigned char*)P->y, Pencoded, 32);   // Decoding y-coordinate and sign
    sign = (unsigned int)(Pencoded[31] >> 7);
    P->y[1][NWORDS_FIELD-1] &= mask7fff;
    from_std_to_ext(P->y, VP->y);

    v2sqr1271(VP->y, u2);
    v2mul1271(u2, (uint32_t*)&PARAMETER_d, v2);
    v2sub1271(u2, one, u2);
    v2add1271(v2, one, v2);
    from_v2_to_v(u2, &u[0], &u[VWORDS_FIELD]);
    from_v2_to_v(v2, &v[0], &v[VWORDS_FIELD]);

    vsqr1271(&v[0], t0);                              // t0 = v0^2
    vsqr1271(&v[VWORDS_FIELD], t1);                   // t1 = v1^2
    vadd1271(t0, t1, t0);                             // t0 = t0+t1 
    vmul1271(&u[0], &v[0], t1);                       // t1 = u0*v0
    vmul1271(&u[VWORDS_FIELD], &v[VWORDS_FIELD], t2); // t2 = u1*v1 
    vadd1271(t1, t2, t1);                             // t1 = t1+t2 
    vmul1271(&u[VWORDS_FIELD], &v[0], t2);            // t2 = u1*v0
    vmul1271(&u[0], &v[VWORDS_FIELD], t3);            // t3 = u0*v1
    vsub1271(t2, t3, t2);                             // t2 = t2-t3
    vsqr1271(t1, t3);                                 // t3 = t1^2    
    vsqr1271(t2, t4);                                 // t4 = t2^2                   
    vadd1271(t3, t4, t3);                             // t3 = t3+t4
    for (i = 0; i < 125; i++) {                       // t3 = t3^(2^125)
        vsqr1271(t3, t3);
    }

    vadd1271(t1, t3, t);                           // t = t1+t3
    vmod1271(t, t);
    if (is_zero_ct(t, VWORDS_FIELD) == true) {
        vsub1271(t1, t3, t);                       // t = t1-t3
    }
    vadd1271(t, t, t);                             // t = 2*t 
    vsqr1271(t0, t3);                              // t3 = t0^2      
    vmul1271(t0, t3, t3);                          // t3 = t3*t0   
    vmul1271(t, t3, t3);                           // t3 = t3*t
    vexp1251(t3, r);                               // r = t3^(2^125-1)  
    vmul1271(t0, r, t3);                           // t3 = t0*r          
    vmul1271(t, t3, VPx0);                         // x0 = t*t3 
    vsqr1271(VPx0, t1);
    vmul1271(t0, t1, t1);                          // t1 = t0*x0^2 
    vdiv1271(VPx0);                                // x0 = x0/2
    vmul1271(t2, t3, VPx1);                        // x1 = t3*t2  

    vsub1271(t, t1, t);
    vmod1271(t, t);
    if (is_zero_ct(t, VWORDS_FIELD) == false) {        // If t != t1 then swap x0 and x1       
        memmove((unsigned char*)t0, (unsigned char*)VPx0, 20);
        memmove((unsigned char*)VPx0, (unsigned char*)VPx1, 20);
        memmove((unsigned char*)VPx1, t0, 20);
    }

    from_v_to_v2(VPx0, VPx1, VP->x); 
    v2mod1271(VP->x, VP->x);
    if (is_zero_ct(VP->x, VWORDS_FIELD) == true) {
        sign_dec = VP->x[2*VWORDS_FIELD-1] >> 22;
    } else {
        sign_dec = VP->x[2*VWORDS_FIELD-2] >> 22;
    }

    if (sign != (unsigned int)sign_dec) {           // If sign of x-coordinate decoded != input sign bit, then negate x-coordinate
        v2neg1271(VP->x);
    }

    v2mod1271(VP->x, R->x);
    v2mod1271(VP->y, R->y);
    if (ecc_point_validate(R) == false) {
        v2neg1271_felm(R->x);
        if (ecc_point_validate(R) == false) {       // Final point validation
            return ECCRYPTO_ERROR;
        }
    }

    v2mod1271(R->x, R->x);
    from_ext_to_std(R->x, P->x);
    from_ext_to_std(R->y, P->y);

    return ECCRYPTO_SUCCESS;
}


void to_Montgomery(const digit_t* ma, digit_t* c)
{ // Converting to Montgomery representation

    Montgomery_multiply_mod_order(ma, (digit_t*)&Montgomery_Rprime, c);
}


void from_Montgomery(const digit_t* a, digit_t* mc)
{ // Converting from Montgomery to standard representation
    digit_t one[NWORDS_ORDER] = {0};
    one[0] = 1;

    Montgomery_multiply_mod_order(a, one, mc);
}


const char* FourQ_get_error_message(ECCRYPTO_STATUS Status)
{ // Output error/success message for a given ECCRYPTO_STATUS
    struct error_mapping {
        unsigned int index;
        char*        string;
    } mapping[ECCRYPTO_STATUS_TYPE_SIZE] = {
        {ECCRYPTO_ERROR, ECCRYPTO_MSG_ERROR},
        {ECCRYPTO_SUCCESS, ECCRYPTO_MSG_SUCCESS},
        {ECCRYPTO_ERROR_DURING_TEST, ECCRYPTO_MSG_ERROR_DURING_TEST},
        {ECCRYPTO_ERROR_UNKNOWN, ECCRYPTO_MSG_ERROR_UNKNOWN},
        {ECCRYPTO_ERROR_NOT_IMPLEMENTED, ECCRYPTO_MSG_ERROR_NOT_IMPLEMENTED},
        {ECCRYPTO_ERROR_NO_MEMORY, ECCRYPTO_MSG_ERROR_NO_MEMORY},
        {ECCRYPTO_ERROR_INVALID_PARAMETER, ECCRYPTO_MSG_ERROR_INVALID_PARAMETER},
        {ECCRYPTO_ERROR_SHARED_KEY, ECCRYPTO_MSG_ERROR_SHARED_KEY},
        {ECCRYPTO_ERROR_SIGNATURE_VERIFICATION, ECCRYPTO_MSG_ERROR_SIGNATURE_VERIFICATION},
    };

    if (Status >= ECCRYPTO_STATUS_TYPE_SIZE || mapping[Status].string == NULL) {
        return "Unrecognized ECCRYPTO_STATUS";
    } else {
        return mapping[Status].string;
    }
};