/* MIT License
 *
 * Copyright (c) 2016-2020 INRIA, CMU and Microsoft Corporation
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */


#include "Hacl_NaCl.h"

static void Hacl_Impl_SecretBox_secretbox_init(uint8_t *xkeys, uint8_t *k, uint8_t *n1)
{
  uint8_t *subkey = xkeys;
  uint8_t *aekey = xkeys + (uint32_t)32U;
  uint8_t *n0 = n1;
  uint8_t *n11 = n1 + (uint32_t)16U;
  Hacl_Salsa20_hsalsa20(subkey, k, n0);
  Hacl_Salsa20_salsa20_key_block0(aekey, subkey, n11);
}

static void
Hacl_Impl_SecretBox_secretbox_detached(
  uint32_t mlen,
  uint8_t *c,
  uint8_t *tag,
  uint8_t *k,
  uint8_t *n1,
  uint8_t *m
)
{
  uint8_t xkeys[96U] = { 0U };
  uint8_t *mkey;
  Hacl_Impl_SecretBox_secretbox_init(xkeys, k, n1);
  mkey = xkeys + (uint32_t)32U;
  {
    uint8_t *n11 = n1 + (uint32_t)16U;
    uint8_t *subkey = xkeys;
    uint8_t *ekey0 = xkeys + (uint32_t)64U;
    uint32_t mlen0;
    if (mlen <= (uint32_t)32U)
    {
      mlen0 = mlen;
    }
    else
    {
      mlen0 = (uint32_t)32U;
    }
    {
      uint32_t mlen1 = mlen - mlen0;
      uint8_t *m0 = m;
      uint8_t *m1 = m + mlen0;
      uint8_t block0[32U] = { 0U };
      uint8_t *c0;
      uint8_t *c1;
      memcpy(block0, m0, mlen0 * sizeof m0[0U]);
      {
        uint32_t i;
        for (i = (uint32_t)0U; i < (uint32_t)32U; i = i + (uint32_t)1U)
        {
          uint8_t *os = block0;
          uint8_t x = block0[i] ^ ekey0[i];
          os[i] = x;
        }
      }
      c0 = c;
      c1 = c + mlen0;
      memcpy(c0, block0, mlen0 * sizeof block0[0U]);
      Hacl_Salsa20_salsa20_encrypt(mlen1, c1, m1, subkey, n11, (uint32_t)1U);
      Hacl_Poly1305_32_poly1305_mac(tag, mlen, c, mkey);
    }
  }
}

static uint32_t
Hacl_Impl_SecretBox_secretbox_open_detached(
  uint32_t mlen,
  uint8_t *m,
  uint8_t *k,
  uint8_t *n1,
  uint8_t *c,
  uint8_t *tag
)
{
  uint8_t xkeys[96U] = { 0U };
  uint8_t *mkey;
  Hacl_Impl_SecretBox_secretbox_init(xkeys, k, n1);
  mkey = xkeys + (uint32_t)32U;
  {
    uint8_t tag_[16U] = { 0U };
    Hacl_Poly1305_32_poly1305_mac(tag_, mlen, c, mkey);
    {
      uint8_t res0 = (uint8_t)255U;
      uint8_t z;
      uint32_t res;
      {
        uint32_t i;
        for (i = (uint32_t)0U; i < (uint32_t)16U; i = i + (uint32_t)1U)
        {
          uint8_t uu____0 = FStar_UInt8_eq_mask(tag[i], tag_[i]);
          res0 = uu____0 & res0;
        }
      }
      z = res0;
      if (z == (uint8_t)255U)
      {
        uint8_t *subkey = xkeys;
        uint8_t *ekey0 = xkeys + (uint32_t)64U;
        uint8_t *n11 = n1 + (uint32_t)16U;
        uint32_t mlen0;
        if (mlen <= (uint32_t)32U)
        {
          mlen0 = mlen;
        }
        else
        {
          mlen0 = (uint32_t)32U;
        }
        {
          uint32_t mlen1 = mlen - mlen0;
          uint8_t *c0 = c;
          uint8_t *c1 = c + mlen0;
          uint8_t block0[32U] = { 0U };
          memcpy(block0, c0, mlen0 * sizeof c0[0U]);
          {
            uint32_t i;
            for (i = (uint32_t)0U; i < (uint32_t)32U; i = i + (uint32_t)1U)
            {
              uint8_t *os = block0;
              uint8_t x = block0[i] ^ ekey0[i];
              os[i] = x;
            }
          }
          {
            uint8_t *m0 = m;
            uint8_t *m1 = m + mlen0;
            memcpy(m0, block0, mlen0 * sizeof block0[0U]);
            Hacl_Salsa20_salsa20_decrypt(mlen1, m1, c1, subkey, n11, (uint32_t)1U);
            res = (uint32_t)0U;
          }
        }
      }
      else
      {
        res = (uint32_t)0xffffffffU;
      }
      return res;
    }
  }
}

static void
Hacl_Impl_SecretBox_secretbox_easy(
  uint32_t mlen,
  uint8_t *c,
  uint8_t *k,
  uint8_t *n1,
  uint8_t *m
)
{
  uint8_t *tag = c;
  uint8_t *cip = c + (uint32_t)16U;
  Hacl_Impl_SecretBox_secretbox_detached(mlen, cip, tag, k, n1, m);
}

static uint32_t
Hacl_Impl_SecretBox_secretbox_open_easy(
  uint32_t mlen,
  uint8_t *m,
  uint8_t *k,
  uint8_t *n1,
  uint8_t *c
)
{
  uint8_t *tag = c;
  uint8_t *cip = c + (uint32_t)16U;
  return Hacl_Impl_SecretBox_secretbox_open_detached(mlen, m, k, n1, cip, tag);
}

inline static uint32_t Hacl_Impl_Box_box_beforenm(uint8_t *k, uint8_t *pk, uint8_t *sk)
{
  uint8_t n0[16U] = { 0U };
  bool r = Hacl_Curve25519_51_ecdh(k, sk, pk);
  if (r)
  {
    Hacl_Salsa20_hsalsa20(k, k, n0);
    return (uint32_t)0U;
  }
  return (uint32_t)0xffffffffU;
}

inline static uint32_t
Hacl_Impl_Box_box_detached_afternm(
  uint32_t mlen,
  uint8_t *c,
  uint8_t *tag,
  uint8_t *k,
  uint8_t *n1,
  uint8_t *m
)
{
  Hacl_Impl_SecretBox_secretbox_detached(mlen, c, tag, k, n1, m);
  return (uint32_t)0U;
}

inline static uint32_t
Hacl_Impl_Box_box_detached(
  uint32_t mlen,
  uint8_t *c,
  uint8_t *tag,
  uint8_t *sk,
  uint8_t *pk,
  uint8_t *n1,
  uint8_t *m
)
{
  uint8_t k[32U] = { 0U };
  uint32_t r = Hacl_Impl_Box_box_beforenm(k, pk, sk);
  if (r == (uint32_t)0U)
  {
    return Hacl_Impl_Box_box_detached_afternm(mlen, c, tag, k, n1, m);
  }
  return (uint32_t)0xffffffffU;
}

inline static uint32_t
Hacl_Impl_Box_box_open_detached_afternm(
  uint32_t mlen,
  uint8_t *m,
  uint8_t *k,
  uint8_t *n1,
  uint8_t *c,
  uint8_t *tag
)
{
  return Hacl_Impl_SecretBox_secretbox_open_detached(mlen, m, k, n1, c, tag);
}

inline static uint32_t
Hacl_Impl_Box_box_open_detached(
  uint32_t mlen,
  uint8_t *m,
  uint8_t *pk,
  uint8_t *sk,
  uint8_t *n1,
  uint8_t *c,
  uint8_t *tag
)
{
  uint8_t k[32U] = { 0U };
  uint32_t r = Hacl_Impl_Box_box_beforenm(k, pk, sk);
  if (r == (uint32_t)0U)
  {
    return Hacl_Impl_Box_box_open_detached_afternm(mlen, m, k, n1, c, tag);
  }
  return (uint32_t)0xffffffffU;
}

inline static uint32_t
Hacl_Impl_Box_box_easy_afternm(uint32_t mlen, uint8_t *c, uint8_t *k, uint8_t *n1, uint8_t *m)
{
  uint8_t *tag = c;
  uint8_t *cip = c + (uint32_t)16U;
  uint32_t res = Hacl_Impl_Box_box_detached_afternm(mlen, cip, tag, k, n1, m);
  return res;
}

inline static uint32_t
Hacl_Impl_Box_box_easy(
  uint32_t mlen,
  uint8_t *c,
  uint8_t *sk,
  uint8_t *pk,
  uint8_t *n1,
  uint8_t *m
)
{
  uint8_t *tag = c;
  uint8_t *cip = c + (uint32_t)16U;
  uint32_t res = Hacl_Impl_Box_box_detached(mlen, cip, tag, sk, pk, n1, m);
  return res;
}

inline static uint32_t
Hacl_Impl_Box_box_open_easy_afternm(
  uint32_t mlen,
  uint8_t *m,
  uint8_t *k,
  uint8_t *n1,
  uint8_t *c
)
{
  uint8_t *tag = c;
  uint8_t *cip = c + (uint32_t)16U;
  return Hacl_Impl_Box_box_open_detached_afternm(mlen, m, k, n1, cip, tag);
}

inline static uint32_t
Hacl_Impl_Box_box_open_easy(
  uint32_t mlen,
  uint8_t *m,
  uint8_t *pk,
  uint8_t *sk,
  uint8_t *n1,
  uint8_t *c
)
{
  uint8_t *tag = c;
  uint8_t *cip = c + (uint32_t)16U;
  return Hacl_Impl_Box_box_open_detached(mlen, m, pk, sk, n1, cip, tag);
}

uint32_t
Hacl_NaCl_crypto_secretbox_detached(
  uint8_t *c,
  uint8_t *tag,
  uint8_t *m,
  uint32_t mlen,
  uint8_t *n1,
  uint8_t *k
)
{
  Hacl_Impl_SecretBox_secretbox_detached(mlen, c, tag, k, n1, m);
  return (uint32_t)0U;
}

uint32_t
Hacl_NaCl_crypto_secretbox_open_detached(
  uint8_t *m,
  uint8_t *c,
  uint8_t *tag,
  uint32_t mlen,
  uint8_t *n1,
  uint8_t *k
)
{
  return Hacl_Impl_SecretBox_secretbox_open_detached(mlen, m, k, n1, c, tag);
}

uint32_t
Hacl_NaCl_crypto_secretbox_easy(uint8_t *c, uint8_t *m, uint32_t mlen, uint8_t *n1, uint8_t *k)
{
  Hacl_Impl_SecretBox_secretbox_easy(mlen, c, k, n1, m);
  return (uint32_t)0U;
}

uint32_t
Hacl_NaCl_crypto_secretbox_open_easy(
  uint8_t *m,
  uint8_t *c,
  uint32_t clen,
  uint8_t *n1,
  uint8_t *k
)
{
  return Hacl_Impl_SecretBox_secretbox_open_easy(clen - (uint32_t)16U, m, k, n1, c);
}

uint32_t Hacl_NaCl_crypto_box_beforenm(uint8_t *k, uint8_t *pk, uint8_t *sk)
{
  return Hacl_Impl_Box_box_beforenm(k, pk, sk);
}

uint32_t
Hacl_NaCl_crypto_box_detached_afternm(
  uint8_t *c,
  uint8_t *tag,
  uint8_t *m,
  uint32_t mlen,
  uint8_t *n1,
  uint8_t *k
)
{
  return Hacl_Impl_Box_box_detached_afternm(mlen, c, tag, k, n1, m);
}

uint32_t
Hacl_NaCl_crypto_box_detached(
  uint8_t *c,
  uint8_t *tag,
  uint8_t *m,
  uint32_t mlen,
  uint8_t *n1,
  uint8_t *pk,
  uint8_t *sk
)
{
  return Hacl_Impl_Box_box_detached(mlen, c, tag, sk, pk, n1, m);
}

uint32_t
Hacl_NaCl_crypto_box_open_detached_afternm(
  uint8_t *m,
  uint8_t *c,
  uint8_t *tag,
  uint32_t mlen,
  uint8_t *n1,
  uint8_t *k
)
{
  return Hacl_Impl_Box_box_open_detached_afternm(mlen, m, k, n1, c, tag);
}

uint32_t
Hacl_NaCl_crypto_box_open_detached(
  uint8_t *m,
  uint8_t *c,
  uint8_t *tag,
  uint32_t mlen,
  uint8_t *n1,
  uint8_t *pk,
  uint8_t *sk
)
{
  return Hacl_Impl_Box_box_open_detached(mlen, m, pk, sk, n1, c, tag);
}

uint32_t
Hacl_NaCl_crypto_box_easy_afternm(
  uint8_t *c,
  uint8_t *m,
  uint32_t mlen,
  uint8_t *n1,
  uint8_t *k
)
{
  return Hacl_Impl_Box_box_easy_afternm(mlen, c, k, n1, m);
}

uint32_t
Hacl_NaCl_crypto_box_easy(
  uint8_t *c,
  uint8_t *m,
  uint32_t mlen,
  uint8_t *n1,
  uint8_t *pk,
  uint8_t *sk
)
{
  return Hacl_Impl_Box_box_easy(mlen, c, sk, pk, n1, m);
}

uint32_t
Hacl_NaCl_crypto_box_open_easy_afternm(
  uint8_t *m,
  uint8_t *c,
  uint32_t clen,
  uint8_t *n1,
  uint8_t *k
)
{
  return Hacl_Impl_Box_box_open_easy_afternm(clen - (uint32_t)16U, m, k, n1, c);
}

uint32_t
Hacl_NaCl_crypto_box_open_easy(
  uint8_t *m,
  uint8_t *c,
  uint32_t clen,
  uint8_t *n1,
  uint8_t *pk,
  uint8_t *sk
)
{
  return Hacl_Impl_Box_box_open_easy(clen - (uint32_t)16U, m, pk, sk, n1, c);
}