// SPDX-License-Identifier: (GPL-2.0+ OR BSD-3-Clause)
/*
 * Shared descriptors for ahash algorithms
 *
 * Copyright 2017-2019 NXP
 */

#include "compat.h"
#include "desc_constr.h"
#include "caamhash_desc.h"

/**
 * cnstr_shdsc_ahash - ahash shared descriptor
 * @desc: pointer to buffer used for descriptor construction
 * @adata: pointer to authentication transform definitions.
 *         A split key is required for SEC Era < 6; the size of the split key
 *         is specified in this case.
 *         Valid algorithm values - one of OP_ALG_ALGSEL_{MD5, SHA1, SHA224,
 *         SHA256, SHA384, SHA512}.
 * @state: algorithm state OP_ALG_AS_{INIT, FINALIZE, INITFINALIZE, UPDATE}
 * @digestsize: algorithm's digest size
 * @ctx_len: size of Context Register
 * @import_ctx: true if previous Context Register needs to be restored
 *              must be true for ahash update and final
 *              must be false for ahash first and digest
 * @era: SEC Era
 */
void cnstr_shdsc_ahash(u32 * const desc, struct alginfo *adata, u32 state,
         int digestsize, int ctx_len, bool import_ctx, int era)
{
 u32 op = adata->algtype;

 init_sh_desc(desc, HDR_SHARE_SERIAL);

 /* Append key if it has been set; ahash update excluded */
 if (state != OP_ALG_AS_UPDATE && adata->keylen) {
  u32 *skip_key_load;

  /* Skip key loading if already shared */
  skip_key_load = append_jump(desc, JUMP_JSL | JUMP_TEST_ALL |
         JUMP_COND_SHRD);

  if (era < 6)
   append_key_as_imm(desc, adata->key_virt,
       adata->keylen_pad,
       adata->keylen, CLASS_2 |
       KEY_DEST_MDHA_SPLIT | KEY_ENC);
  else
   append_proto_dkp(desc, adata);

  set_jump_tgt_here(desc, skip_key_load);

  op |= OP_ALG_AAI_HMAC_PRECOMP;
 }

 /* If needed, import context from software */
 if (import_ctx)
  append_seq_load(desc, ctx_len, LDST_CLASS_2_CCB |
    LDST_SRCDST_BYTE_CONTEXT);

 /* Class 2 operation */
 append_operation(desc, op | state | OP_ALG_ENCRYPT);

 /*
  * Load from buf and/or src and write to req->result or state->context
  * Calculate remaining bytes to read
 */
 append_math_add(desc, VARSEQINLEN, SEQINLEN, REG0, CAAM_CMD_SZ);
 /* Read remaining bytes */
 append_seq_fifo_load(desc, 0, FIFOLD_CLASS_CLASS2 | FIFOLD_TYPE_LAST2 |
        FIFOLD_TYPE_MSG | KEY_VLF);
 /* Store class2 context bytes */
 append_seq_store(desc, digestsize, LDST_CLASS_2_CCB |
    LDST_SRCDST_BYTE_CONTEXT);
}
EXPORT_SYMBOL(cnstr_shdsc_ahash);

/**
 * cnstr_shdsc_sk_hash - shared descriptor for symmetric key cipher-based
 *                       hash algorithms
 * @desc: pointer to buffer used for descriptor construction
 * @adata: pointer to authentication transform definitions.
 * @state: algorithm state OP_ALG_AS_{INIT, FINALIZE, INITFINALIZE, UPDATE}
 * @digestsize: algorithm's digest size
 * @ctx_len: size of Context Register
 */
void cnstr_shdsc_sk_hash(u32 * const desc, struct alginfo *adata, u32 state,
    int digestsize, int ctx_len)
{
 u32 *skip_key_load;

 init_sh_desc(desc, HDR_SHARE_SERIAL | HDR_SAVECTX);

 /* Skip loading of key, context if already shared */
 skip_key_load = append_jump(desc, JUMP_TEST_ALL | JUMP_COND_SHRD);

 if (state == OP_ALG_AS_INIT || state == OP_ALG_AS_INITFINAL) {
  append_key_as_imm(desc, adata->key_virt, adata->keylen,
      adata->keylen, CLASS_1 | KEY_DEST_CLASS_REG);
 } else { /* UPDATE, FINALIZE */
  if (is_xcbc_aes(adata->algtype))
   /* Load K1 */
   append_key(desc, adata->key_dma, adata->keylen,
       CLASS_1 | KEY_DEST_CLASS_REG | KEY_ENC);
  else /* CMAC */
   append_key_as_imm(desc, adata->key_virt, adata->keylen,
       adata->keylen, CLASS_1 |
       KEY_DEST_CLASS_REG);
  /* Restore context */
  append_seq_load(desc, ctx_len, LDST_CLASS_1_CCB |
    LDST_SRCDST_BYTE_CONTEXT);
 }

 set_jump_tgt_here(desc, skip_key_load);

 /* Class 1 operation */
 append_operation(desc, adata->algtype | state | OP_ALG_ENCRYPT);

 /*
  * Load from buf and/or src and write to req->result or state->context
  * Calculate remaining bytes to read
 */
 append_math_add(desc, VARSEQINLEN, SEQINLEN, REG0, CAAM_CMD_SZ);

 /* Read remaining bytes */
 append_seq_fifo_load(desc, 0, FIFOLD_CLASS_CLASS1 | FIFOLD_TYPE_LAST1 |
        FIFOLD_TYPE_MSG | FIFOLDST_VLF);

 /*
  * Save context:
  * - xcbc: partial hash, keys K2 and K3
  * - cmac: partial hash, constant L = E(K,0)
 */
 append_seq_store(desc, digestsize, LDST_CLASS_1_CCB |
    LDST_SRCDST_BYTE_CONTEXT);
 if (is_xcbc_aes(adata->algtype) && state == OP_ALG_AS_INIT)
  /* Save K1 */
  append_fifo_store(desc, adata->key_dma, adata->keylen,
      LDST_CLASS_1_CCB | FIFOST_TYPE_KEY_KEK);
}
EXPORT_SYMBOL(cnstr_shdsc_sk_hash);

MODULE_LICENSE("Dual BSD/GPL");
MODULE_DESCRIPTION("FSL CAAM ahash descriptors support");
MODULE_AUTHOR("NXP Semiconductors");