internal-isap-pk.h

/*
 * Copyright (C) 2021 Southern Storm Software, Pty Ltd.
 *
 * 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.
 */

/* We expect a number of macros to be defined before this file
 * is included to configure the underlying ISAP variant.
 *
 * ISAP_ALG_NAME        Name of the ISAP algorithm; e.g. isap_keccak_128
 * ISAP_KEY_STATE       Type for the pre-computed key state
 * ISAP_RATE            Number of bytes in the rate for hashing and encryption.
 * ISAP_sH              Number of rounds for hashing.
 * ISAP_sE              Number of rounds for encryption.
 * ISAP_sB              Number of rounds for key bit absorption.
 * ISAP_sK              Number of rounds for keying.
 * ISAP_STATE           Type for the permuation state; e.g. ascon_state_t
 * ISAP_PERMUTE(s,r)    Permutes the state "s" with number of rounds "r".
 * ISAP_PERMUTE_SLICED(s,r) Defined if using the sliced version of ASCON.
 */
#if defined(ISAP_ALG_NAME)

#if !defined(ISAP_KEY_SIZE)
#define ISAP_KEY_SIZE 16
#define ISAP_NONCE_SIZE 16
#define ISAP_TAG_SIZE 16
#endif

#define ISAP_CONCAT_INNER(name,suffix) name##suffix
#define ISAP_CONCAT(name,suffix) ISAP_CONCAT_INNER(name,suffix)

/* IV string for initialising the associated data */
static unsigned char const ISAP_CONCAT(ISAP_ALG_NAME,_pk_IV_A)
        [sizeof(ISAP_STATE) - ISAP_NONCE_SIZE] = {
    0x01, ISAP_KEY_SIZE * 8, ISAP_RATE * 8, 1,
    ISAP_sH, ISAP_sB, ISAP_sE, ISAP_sK
};

/* IV string for authenticating associated data */
static unsigned char const ISAP_CONCAT(ISAP_ALG_NAME,_pk_IV_KA)
        [sizeof(ISAP_STATE) - ISAP_KEY_SIZE] = {
    0x02, ISAP_KEY_SIZE * 8, ISAP_RATE * 8, 1,
    ISAP_sH, ISAP_sB, ISAP_sE, ISAP_sK
};

/* IV string for encrypting payload data */
static unsigned char const ISAP_CONCAT(ISAP_ALG_NAME,_pk_IV_KE)
        [sizeof(ISAP_STATE) - ISAP_KEY_SIZE] = {
    0x03, ISAP_KEY_SIZE * 8, ISAP_RATE * 8, 1,
    ISAP_sH, ISAP_sB, ISAP_sE, ISAP_sK
};

static void ISAP_CONCAT(ISAP_ALG_NAME,_pk_rekey)
    (ISAP_STATE *state, const unsigned char *pk,
     const unsigned char *data, unsigned data_len)
{
#if defined(ISAP_PERMUTE_SLICED)
    unsigned bit, num_bits;

    /* Initialize the state with the key and IV (already pre-sliced) */
    memcpy(state->B, pk, sizeof(state->B));

    /* Absorb all of the bits of the data buffer one by one */
    num_bits = data_len * 8 - 1;
    for (bit = 0; bit < num_bits; ++bit) {
        state->W[1] ^=
            (((uint32_t)(data[bit / 8])) << (24 + bit % 8)) & 0x80000000U;
        ISAP_PERMUTE_SLICED(state, ISAP_sB);
    }
    state->W[1] ^=
        (((uint32_t)(data[bit / 8])) << (24 + bit % 8)) & 0x80000000U;
    ISAP_PERMUTE_SLICED(state, ISAP_sK);
#else
    unsigned bit, num_bits;

    /* Initialize the state with the key and IV */
    memcpy(state->B, pk, sizeof(state->B));

    /* Absorb all of the bits of the data buffer one by one */
    num_bits = data_len * 8 - 1;
    for (bit = 0; bit < num_bits; ++bit) {
        state->B[0] ^= (data[bit / 8] << (bit % 8)) & 0x80;
        ISAP_PERMUTE(state, ISAP_sB);
    }
    state->B[0] ^= (data[bit / 8] << (bit % 8)) & 0x80;
    ISAP_PERMUTE(state, ISAP_sK);
#endif
}

static void ISAP_CONCAT(ISAP_ALG_NAME,_pk_encrypt)
    (ISAP_STATE *state, const ISAP_KEY_STATE *pk, const unsigned char *npub,
     unsigned char *c, const unsigned char *m, size_t mlen)
{
#if defined(ISAP_PERMUTE_SLICED)
    unsigned char block[ISAP_RATE];

    /* Set up the re-keyed encryption key and nonce in the state */
    ISAP_CONCAT(ISAP_ALG_NAME,_pk_rekey)(state, pk->ke, npub, ISAP_NONCE_SIZE);
    ascon_set_sliced(state, npub, 3);
    ascon_set_sliced(state, npub + 8, 4);

    /* Encrypt the plaintext to produce the ciphertext */
    while (mlen >= ISAP_RATE) {
        ISAP_PERMUTE_SLICED(state, ISAP_sE);
        ascon_squeeze_sliced(state, block, 0);
        lw_xor_block_2_src(c, block, m, ISAP_RATE);
        c += ISAP_RATE;
        m += ISAP_RATE;
        mlen -= ISAP_RATE;
    }
    if (mlen > 0) {
        ISAP_PERMUTE_SLICED(state, ISAP_sE);
        ascon_squeeze_sliced(state, block, 0);
        lw_xor_block_2_src(c, block, m, (unsigned)mlen);
    }
#else
    /* Set up the re-keyed encryption key and nonce in the state */
    ISAP_CONCAT(ISAP_ALG_NAME,_pk_rekey)(state, pk->ke, npub, ISAP_NONCE_SIZE);
    memcpy(state->B + sizeof(ISAP_STATE) - ISAP_NONCE_SIZE,
           npub, ISAP_NONCE_SIZE);

    /* Encrypt the plaintext to produce the ciphertext */
    while (mlen >= ISAP_RATE) {
        ISAP_PERMUTE(state, ISAP_sE);
        lw_xor_block_2_src(c, state->B, m, ISAP_RATE);
        c += ISAP_RATE;
        m += ISAP_RATE;
        mlen -= ISAP_RATE;
    }
    if (mlen > 0) {
        ISAP_PERMUTE(state, ISAP_sE);
        lw_xor_block_2_src(c, state->B, m, (unsigned)mlen);
    }
#endif
}

static void ISAP_CONCAT(ISAP_ALG_NAME,_pk_mac)
    (ISAP_STATE *state, const ISAP_KEY_STATE *pk,
     const unsigned char *npub,
     const unsigned char *ad, size_t adlen,
     const unsigned char *c, size_t clen,
     unsigned char *tag)
{
#if defined(ISAP_PERMUTE_SLICED)
    unsigned char preserve[sizeof(ISAP_STATE) - ISAP_TAG_SIZE];
    unsigned char padded[ISAP_RATE];
    unsigned temp;

    /* Absorb the associated data */
    memcpy(state->B, npub, ISAP_NONCE_SIZE);
    memcpy(state->B + ISAP_NONCE_SIZE, ISAP_CONCAT(ISAP_ALG_NAME,_pk_IV_A),
           sizeof(state->B) - ISAP_NONCE_SIZE);
    ascon_to_sliced(state);
    ISAP_PERMUTE_SLICED(state, ISAP_sH);
    while (adlen >= ISAP_RATE) {
        ascon_absorb_sliced(state, ad, 0);
        ISAP_PERMUTE_SLICED(state, ISAP_sH);
        ad += ISAP_RATE;
        adlen -= ISAP_RATE;
    }
    temp = (unsigned)adlen;
    memcpy(padded, ad, temp);
    padded[temp] = 0x80; /* padding */
    memset(padded + temp + 1, 0, sizeof(padded) - (temp + 1));
    ascon_absorb_sliced(state, padded, 0);
    ISAP_PERMUTE_SLICED(state, ISAP_sH);
    state->W[8] ^= 0x01; /* domain separation */

    /* Absorb the ciphertext */
    while (clen >= ISAP_RATE) {
        ascon_absorb_sliced(state, c, 0);
        ISAP_PERMUTE_SLICED(state, ISAP_sH);
        c += ISAP_RATE;
        clen -= ISAP_RATE;
    }
    temp = (unsigned)clen;
    memcpy(padded, c, temp);
    padded[temp] = 0x80; /* padding */
    memset(padded + temp + 1, 0, sizeof(padded) - (temp + 1));
    ascon_absorb_sliced(state, padded, 0);
    ISAP_PERMUTE_SLICED(state, ISAP_sH);

    /* Re-key the state and generate the authentication tag */
    ascon_from_sliced(state);
    memcpy(tag, state->B, ISAP_TAG_SIZE);
    memcpy(preserve, state->B + ISAP_TAG_SIZE, sizeof(preserve));
    ISAP_CONCAT(ISAP_ALG_NAME,_pk_rekey)(state, pk->ka, tag, ISAP_TAG_SIZE);
    ascon_from_sliced(state);
    memcpy(state->B + ISAP_TAG_SIZE, preserve, sizeof(preserve));
    ascon_to_sliced(state);
    ISAP_PERMUTE_SLICED(state, ISAP_sH);
    ascon_squeeze_sliced(state, tag, 0);
    ascon_squeeze_sliced(state, tag + 8, 1);
#else
    unsigned char preserve[sizeof(ISAP_STATE) - ISAP_TAG_SIZE];
    unsigned temp;

    /* Absorb the associated data */
    memcpy(state->B, npub, ISAP_NONCE_SIZE);
    memcpy(state->B + ISAP_NONCE_SIZE, ISAP_CONCAT(ISAP_ALG_NAME,_pk_IV_A),
           sizeof(state->B) - ISAP_NONCE_SIZE);
    ISAP_PERMUTE(state, ISAP_sH);
    while (adlen >= ISAP_RATE) {
        lw_xor_block(state->B, ad, ISAP_RATE);
        ISAP_PERMUTE(state, ISAP_sH);
        ad += ISAP_RATE;
        adlen -= ISAP_RATE;
    }
    temp = (unsigned)adlen;
    lw_xor_block(state->B, ad, temp);
    state->B[temp] ^= 0x80; /* padding */
    ISAP_PERMUTE(state, ISAP_sH);
    state->B[sizeof(state->B) - 1] ^= 0x01; /* domain separation */

    /* Absorb the ciphertext */
    while (clen >= ISAP_RATE) {
        lw_xor_block(state->B, c, ISAP_RATE);
        ISAP_PERMUTE(state, ISAP_sH);
        c += ISAP_RATE;
        clen -= ISAP_RATE;
    }
    temp = (unsigned)clen;
    lw_xor_block(state->B, c, temp);
    state->B[temp] ^= 0x80; /* padding */
    ISAP_PERMUTE(state, ISAP_sH);

    /* Re-key the state and generate the authentication tag */
    memcpy(tag, state->B, ISAP_TAG_SIZE);
    memcpy(preserve, state->B + ISAP_TAG_SIZE, sizeof(preserve));
    ISAP_CONCAT(ISAP_ALG_NAME,_pk_rekey)(state, pk->ka, tag, ISAP_TAG_SIZE);
    memcpy(state->B + ISAP_TAG_SIZE, preserve, sizeof(preserve));
    ISAP_PERMUTE(state, ISAP_sH);
    memcpy(tag, state->B, ISAP_TAG_SIZE);
#endif
}

int ISAP_CONCAT(ISAP_ALG_NAME,_aead_pk_init)
    (ISAP_KEY_STATE *pk, const unsigned char *k)
{
#if defined(ISAP_PERMUTE_SLICED)
    ISAP_STATE state;

    /* Expand the encryption key */
    memcpy(state.B, k, ISAP_KEY_SIZE);
    memcpy(state.B + ISAP_KEY_SIZE, ISAP_CONCAT(ISAP_ALG_NAME,_pk_IV_KE),
           sizeof(state.B) - ISAP_KEY_SIZE);
    ascon_to_sliced(&state);
    ISAP_PERMUTE_SLICED(&state, ISAP_sK);
    memcpy(pk->ke, state.B, sizeof(state.B));

    /* Expand the authentication key */
    memcpy(state.B, k, ISAP_KEY_SIZE);
    memcpy(state.B + ISAP_KEY_SIZE, ISAP_CONCAT(ISAP_ALG_NAME,_pk_IV_KA),
           sizeof(state.B) - ISAP_KEY_SIZE);
    ascon_to_sliced(&state);
    ISAP_PERMUTE_SLICED(&state, ISAP_sK);
    memcpy(pk->ka, state.B, sizeof(state.B));

    return 0;
#else
    ISAP_STATE state;

    /* Expand the encryption key */
    memcpy(state.B, k, ISAP_KEY_SIZE);
    memcpy(state.B + ISAP_KEY_SIZE, ISAP_CONCAT(ISAP_ALG_NAME,_pk_IV_KE),
           sizeof(state.B) - ISAP_KEY_SIZE);
    ISAP_PERMUTE(&state, ISAP_sK);
    memcpy(pk->ke, state.B, sizeof(state.B));

    /* Expand the authentication key */
    memcpy(state.B, k, ISAP_KEY_SIZE);
    memcpy(state.B + ISAP_KEY_SIZE, ISAP_CONCAT(ISAP_ALG_NAME,_pk_IV_KA),
           sizeof(state.B) - ISAP_KEY_SIZE);
    ISAP_PERMUTE(&state, ISAP_sK);
    memcpy(pk->ka, state.B, sizeof(state.B));

    return 0;
#endif
}

int ISAP_CONCAT(ISAP_ALG_NAME,_aead_pk_encrypt)
    (unsigned char *c, size_t *clen,
     const unsigned char *m, size_t mlen,
     const unsigned char *ad, size_t adlen,
     const unsigned char *npub,
     const ISAP_KEY_STATE *pk)
{
    ISAP_STATE state;

    /* Set the length of the returned ciphertext */
    *clen = mlen + ISAP_TAG_SIZE;

    /* Encrypt the plaintext to produce the ciphertext */
    ISAP_CONCAT(ISAP_ALG_NAME,_pk_encrypt)(&state, pk, npub, c, m, mlen);

    /* Authenticate the associated data and ciphertext to generate the tag */
    ISAP_CONCAT(ISAP_ALG_NAME,_pk_mac)
        (&state, pk, npub, ad, adlen, c, mlen, c + mlen);
    return 0;
}

int ISAP_CONCAT(ISAP_ALG_NAME,_aead_pk_decrypt)
    (unsigned char *m, size_t *mlen,
     const unsigned char *c, size_t clen,
     const unsigned char *ad, size_t adlen,
     const unsigned char *npub,
     const ISAP_KEY_STATE *pk)
{
    ISAP_STATE state;
    unsigned char tag[ISAP_TAG_SIZE];

    /* Validate the ciphertext length and set the return "mlen" value */
    if (clen < ISAP_TAG_SIZE)
        return -1;
    *mlen = clen - ISAP_TAG_SIZE;

    /* Authenticate the associated data and ciphertext to generate the tag */
    ISAP_CONCAT(ISAP_ALG_NAME,_pk_mac)
        (&state, pk, npub, ad, adlen, c, *mlen, tag);

    /* Decrypt the ciphertext to produce the plaintext */
    ISAP_CONCAT(ISAP_ALG_NAME,_pk_encrypt)(&state, pk, npub, m, c, *mlen);

    /* Check the authentication tag */
    return aead_check_tag(m, *mlen, tag, c + *mlen, ISAP_TAG_SIZE);
}

#endif /* ISAP_ALG_NAME */

/* Now undefine everything so that we can include this file again for
 * another variant on the ISAP algorithm */
#undef ISAP_ALG_NAME
#undef ISAP_KEY_STATE
#undef ISAP_RATE
#undef ISAP_sH
#undef ISAP_sE
#undef ISAP_sB
#undef ISAP_sK
#undef ISAP_STATE
#undef ISAP_PERMUTE
#undef ISAP_PERMUTE_SLICED
#undef ISAP_CONCAT_INNER
#undef ISAP_CONCAT