#include "defs.h"
#include "sha3.h"
#include "ge448.h"

/* temporarily use getrandom() */
#include <sys/random.h>

/* Ed448, no prehash or context */
const static unsigned char ed448_dom4[10] = {'S', 'i', 'g', 'E', 'd', '4', '4', '8', 0, 0};

/* pk is 57 bytes, sk is 114 bytes, sk is followed by pk */
int crypto_sign_ed448_keypair(
    unsigned char *pk,
    unsigned char *sk)
{
  int i;

  (void)getrandom(sk, 57, 0);
  crypto_sign_ed448_derive_pk(pk, sk);
  for (i = 0; i < 57; i++)
    sk[57 + i] = pk[i];
  return 0;
}

/* detached method, sign is 114 bytes */
int crypto_sign_ed448_detached(
    unsigned char *sign, unsigned long long *smlen,
    const unsigned char *m, unsigned long long mlen,
    const unsigned char *sk)
{
  sc448 scsk, scr, sck;
  ge448 ger;
  unsigned char extsk[114];
  unsigned char hashr[114];
  sha3_ctx_t ctx;

  if (smlen != NULL)
  {
    *smlen = 114;
  }
  shake256_init(&ctx);
  shake_update(&ctx, sk, 57);
  shake_xof(&ctx);
  shake_out(&ctx, extsk, 114);
  extsk[56] = 0;
  extsk[55] |= 0x80;
  extsk[0] &= 0xFC;
  shake256_init(&ctx);
  shake_update(&ctx, ed448_dom4, 10);
  shake_update(&ctx, extsk + 57, 57);
  shake_update(&ctx, m, mlen);
  shake_xof(&ctx);
  shake_out(&ctx, hashr, 114);
  sc448_from114bytes(&scr, hashr);
  ge448_scalarmult_base(&ger, &scr);
  ge448_pack(sign, &ger); /* part1: R=r[B] */
  shake256_init(&ctx);
  shake_update(&ctx, ed448_dom4, 10);
  shake_update(&ctx, sign, 57);
  shake_update(&ctx, sk + 57, 57);
  shake_update(&ctx, m, mlen);
  shake_xof(&ctx);
  shake_out(&ctx, hashr, 114);
  sc448_from114bytes(&sck, hashr);
  sc448_from57bytes(&scsk, extsk);
  sc448_mul(&scsk, &scsk, &sck);
  sc448_add(&scsk, &scsk, &scr);
  sc448_to57bytes(sign + 57, &scsk); /* part2: S=r+s*k */
  return 0;
}

/* return 0 on success, -1 on fail */
int crypto_sign_ed448_open_detached(
    const unsigned char *m, unsigned long long mlen,
    const unsigned char *sign, const unsigned char *pk)
{
  ge448 get1, get2;
  sc448 scs, sck;
  unsigned char hashk[114];
  unsigned char t2[57];
  sha3_ctx_t ctx;

  if (ge448_unpackneg_vartime(&get1, pk) != 0)
    return -1;
  shake256_init(&ctx);
  shake_update(&ctx, ed448_dom4, 10);
  shake_update(&ctx, sign, 57);
  shake_update(&ctx, pk, 57);
  shake_update(&ctx, m, mlen);
  shake_xof(&ctx);
  shake_out(&ctx, hashk, 114);
  sc448_from114bytes(&sck, hashk);
  sc448_from57bytes(&scs, sign + 57);
  ge448_double_scalarmult_vartime(&get2, &get1, &sck, &ge448_base, &scs);
  ge448_pack(t2, &get2);
  return crypto_verify_57(t2, sign);
}

/* constant time verify */
int crypto_verify_57(const unsigned char *x, const unsigned char *y)
{
  int i;
  unsigned int differentbits = 0;

  for (i = 0; i < 57; i++)
  {
    differentbits |= (x[i] ^ y[i]);
  }
  return (1 & ((differentbits - 1) >> 8)) - 1;
}

/* derive publick key from secret key, not copy */
int crypto_sign_ed448_derive_pk(unsigned char *pk, const unsigned char *sk)
{
  ge448 gepk;
  sc448 scsk;
  unsigned char extsk[114];
  sha3_ctx_t ctx;

  shake256_init(&ctx);
  shake_update(&ctx, sk, 57);
  shake_xof(&ctx);
  shake_out(&ctx, extsk, 114);
  extsk[56] = 0;
  extsk[55] |= 0x80;
  extsk[0] &= 0xFC;
  sc448_from57bytes(&scsk, extsk);
  ge448_scalarmult_base(&gepk, &scsk);
  ge448_pack(pk, &gepk);
  return 0;
}