/* pi-cipher.h */
/*
    This file is part of the AVR-Crypto-Lib.
    Copyright (C) 2015 Daniel Otte (daniel.otte@rub.de)

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef PI_CIPHER_H_
#define PI_CIPHER_H_

/* please define PI_SIZE acoording to the primitive to implement (pi16cipher, pi32cipher or pi64cipher) */
/* # define PI_SIZE 16 */

#ifdef SUPERCOP
#include "ecrypt-portable.h"
typedef u8 uint8_t;
typedef u16 uint16_t;
typedef u32 uint32_t;
typedef u64 uint64_t;
#else
#include <inttypes.h>
#endif

#include <stddef.h>

#if PI_SIZE == 16
#include "pi16_parameter.h"
#elif PI_SIZE == 32
#include "pi32_parameter.h"
#elif PI_SIZE == 64
#include "pi64_parameter.h"
#else
#error please define PI_SIZE
#endif

#define PI_N 4

#define PI_IS_BITS (4 * PI_N * PI_WORD_SIZE)
#define PI_IS_BYTES PI_IS_BITS / 8

#define PI_RATE_BITS (PI_IS_BITS / 2)
#define PI_CAPACITY_BITS PI_BITS - PI_RATE_BITS

#define PI_RATE_BYTES (PI_RATE_BITS / 8)
#define PI_CAPACITY_BYTES (PI_CAPACITY_BITS / 8)

#define PI_SMN_LENGTH_BITS  PI_RATE_BITS
#define PI_SMN_LENGTH_BYTES (PI_RATE_BITS / 8)

#define PI_AD_BLOCK_LENGTH_BITS PI_RATE_BITS
#define PI_AD_BLOCK_LENGTH_BYTES (PI_AD_BLOCK_LENGTH_BITS / 8)

#define PI_PT_BLOCK_LENGTH_BITS PI_RATE_BITS
#define PI_PT_BLOCK_LENGTH_BYTES (PI_PT_BLOCK_LENGTH_BITS / 8)

#define PI_CT_BLOCK_LENGTH_BITS PI_RATE_BITS
#define PI_CT_BLOCK_LENGTH_BYTES (PI_CT_BLOCK_LENGTH_BITS / 8)

#define PI_ROUNDS 3

#define PI_MU_V_CONST { 3, 2, 1, 0 }
#define PI_NY_V_CONST { 1, 0, 3, 2 }


#define XSTR(x) STR(x)
#define STR(x) #x

#define CTX_NAME(x)                     pi ## x ## _ctx_t
#define INIT_NAME(x)                    pi ## x ## _init
#define PROCESS_AD_BLOCK_NAME(x)        pi ## x ## _process_ad_block
#define PROCESS_AD_LAST_BLOCK_NAME(x)   pi ## x ## _process_ad_last_block
#define ENCRYPT_SMN_NAME(x)             pi ## x ## _encrypt_smn
#define DECRYPT_SMN_NAME(x)             pi ## x ## _decrypt_smn
#define ENCRYPT_BLOCK_NAME(x)           pi ## x ## _encrypt_block
#define ENCRYPT_LAST_BLOCK_NAME(x)      pi ## x ## _encrypt_last_block
#define EXTRACT_TAG_NAME(x)             pi ## x ## _extract_tag
#define DECRYPT_BLOCK_NAME(x)           pi ## x ## _decrypt_block
#define DECRYPT_LAST_BLOCK_NAME(x)      pi ## x ## _decrypt_last_block
#define ENCRYPT_SIMPLE_NAME(x)          pi ## x ## _encrypt_simple
#define DECRYPT_SIMPLE_NAME(x)          pi ## x ## _decrypt_simple
#define CIPHER_NAME(x)                  pi ## x ## cipher
#define CIPHER_NAME_X(x)                pi ## x ## _cipher_name

#define NAME(f,x) f(x)

#define PI_CTX                          NAME(CTX_NAME, PI_WORD_SIZE)
#define PI_INIT                         NAME(INIT_NAME, PI_WORD_SIZE)
#define PI_PROCESS_AD_BLOCK             NAME(PROCESS_AD_BLOCK_NAME, PI_WORD_SIZE)
#define PI_PROCESS_AD_LAST_BLOCK        NAME(PROCESS_AD_LAST_BLOCK_NAME, PI_WORD_SIZE)
#define PI_ENCRYPT_SMN                  NAME(ENCRYPT_SMN_NAME, PI_WORD_SIZE)
#define PI_DECRYPT_SMN                  NAME(DECRYPT_SMN_NAME, PI_WORD_SIZE)
#define PI_ENCRYPT_BLOCK                NAME(ENCRYPT_BLOCK_NAME, PI_WORD_SIZE)
#define PI_ENCRYPT_LAST_BLOCK           NAME(ENCRYPT_LAST_BLOCK_NAME, PI_WORD_SIZE)
#define PI_EXTRACT_TAG                  NAME(EXTRACT_TAG_NAME, PI_WORD_SIZE)
#define PI_DECRYPT_BLOCK                NAME(DECRYPT_BLOCK_NAME, PI_WORD_SIZE)
#define PI_DECRYPT_LAST_BLOCK           NAME(DECRYPT_LAST_BLOCK_NAME, PI_WORD_SIZE)
#define PI_ENCRYPT_SIMPLE               NAME(ENCRYPT_SIMPLE_NAME, PI_WORD_SIZE)
#define PI_DECRYPT_SIMPLE               NAME(DECRYPT_SIMPLE_NAME, PI_WORD_SIZE)
#define PI_CIPHER_NAME                  NAME(CIPHER_NAME, PI_WORD_SIZE)
#define PI_CIPHER_NAME_X                NAME(CIPHER_NAME_X, PI_WORD_SIZE)



extern const char* PI_CIPHER_NAME_X;

typedef struct {
    word_t cis[4][4];
    word_t tag[8];
    uint64_t ctr;
} PI_CTX;

int PI_INIT(
        PI_CTX *ctx,
        const void *key,
        size_t key_length_b,
        const void *pmn,
        size_t pmn_length_b);

void PI_PROCESS_AD_BLOCK(
        PI_CTX *ctx,
        const void *ad,
        unsigned long ad_num );

void PI_PROCESS_AD_LAST_BLOCK(
        PI_CTX *ctx,
        const void *ad,
        size_t ad_length_b,
        unsigned long ad_num );

void PI_ENCRYPT_SMN(
        PI_CTX *ctx,
        void *c0,
        const void *smn);

void PI_DECRYPT_SMN(
        PI_CTX *ctx,
        void *smn,
        const void *c0);

void PI_ENCRYPT_BLOCK(
        PI_CTX *ctx,
        void *dest,
        const void *src,
		unsigned long  num );

void PI_ENCRYPT_LAST_BLOCK(
        PI_CTX *ctx,
        void *dest,
        const void *src,
        size_t length_b,
		unsigned long  num );

void PI_EXTRACT_TAG(
        PI_CTX *ctx,
        void *dest );

void PI_DECRYPT_BLOCK(
        PI_CTX *ctx,
        void *dest,
        const void *src,
		unsigned long  num );

void PI_DECRYPT_LAST_BLOCK(
        PI_CTX *ctx,
        void *dest,
        const void *src,
        size_t length_b,
		unsigned long  num );

void PI_ENCRYPT_SIMPLE(
        void *cipher,
        size_t *cipher_len_B,
        const void *msg,
        size_t msg_len_B,
        const void *ad,
        size_t ad_len_B,
        const void *nonce_secret,
        const void *nonce_public,
        size_t nonce_public_len_B,
        const void *key,
        size_t key_len_B
        );

int PI_DECRYPT_SIMPLE(
        void *msg,
        size_t *msg_len_B,
		void *nonce_secret,
		const void *cipher,
        size_t cipher_len_B,
        const void *ad,
        size_t ad_len_B,
        const void *nonce_public,
        size_t nonce_public_len_B,
        const void *key,
        size_t key_len_B
        );

#endif /* PI_CIPHER_H_ */