Initialize the register dumping machinery while testing assembler code.

[catacomb] / symm / rijndael.c

/* -*-c-*-
 *
 * The Rijndael block cipher
 *
 * (c) 2000 Straylight/Edgeware
 */

/*----- Licensing notice --------------------------------------------------*
 *
 * This file is part of Catacomb.
 *
 * Catacomb is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Library General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version.
 *
 * Catacomb 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 Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with Catacomb; if not, write to the Free
 * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
 * MA 02111-1307, USA.
 */

/*----- Header files ------------------------------------------------------*/

#include "config.h"

#include <assert.h>
#include <stdio.h>

#include <mLib/bits.h>

#if defined(TEST_RIG) && defined(ENABLE_ASM_DEBUG)
#  include "regdump.h"
#  define BLKC_TESTHOOK do { regdump_init(); } while (0)
#endif

#include "blkc.h"
#include "dispatch.h"
#include "gcipher.h"
#include "rijndael.h"
#include "rijndael-base.h"

/*----- Main code ---------------------------------------------------------*/

/* --- @rijndael_init@ --- *
 *
 * Arguments:	@rijndael_ctx *k@ = pointer to context to initialize
 *		@const void *buf@ = pointer to buffer of key material
 *		@size_t sz@ = size of the key material
 *
 * Returns:	---
 *
 * Use:		Initializes a Rijndael context with a particular key.  This
 *		implementation of Rijndael doesn't impose any particular
 *		limits on the key size except that it must be multiple of 4
 *		bytes long.  256 bits seems sensible, though.
 */

void rijndael_init(rijndael_ctx *k, const void *buf, size_t sz)
{
  rijndael_setup(k, RIJNDAEL_BLKSZ / 4, buf, sz);
}

/* --- @rijndael_eblk@, @rijndael_dblk@ --- *
 *
 * Arguments:	@const rijndael_ctx *k@ = pointer to Rijndael context
 *		@const uint32 s[4]@ = pointer to source block
 *		@uint32 d[4]@ = pointer to destination block
 *
 * Returns:	---
 *
 * Use:		Low-level block encryption and decryption.
 */

CPU_DISPATCH(EMPTY, EMPTY, void, rijndael_eblk,
	     (const rijndael_ctx *k, const uint32 s[4], uint32 d[4]),
	     (k, s, d), pick_eblk, simple_eblk)

CPU_DISPATCH(EMPTY, EMPTY, void, rijndael_dblk,
	     (const rijndael_ctx *k, const uint32 s[4], uint32 d[4]),
	     (k, s, d), pick_dblk, simple_dblk)

#if CPUFAM_X86 || CPUFAM_AMD64
extern rijndael_eblk__functype rijndael_eblk_x86ish_aesni;
extern rijndael_dblk__functype rijndael_dblk_x86ish_aesni;
extern rijndael_eblk__functype rijndael_eblk_x86ish_aesni_avx;
extern rijndael_dblk__functype rijndael_dblk_x86ish_aesni_avx;
#endif
#if CPUFAM_ARMEL && HAVE_AS_ARMV8_CRYPTO
extern rijndael_eblk__functype rijndael_eblk_arm_crypto;
extern rijndael_dblk__functype rijndael_dblk_arm_crypto;
#endif
#if CPUFAM_ARM64
extern rijndael_eblk__functype rijndael_eblk_arm64_crypto;
extern rijndael_dblk__functype rijndael_dblk_arm64_crypto;
#endif

static rijndael_eblk__functype *pick_eblk(void)
{
#if CPUFAM_X86 || CPUFAM_AMD64
  DISPATCH_PICK_COND(rijndael_eblk, rijndael_eblk_x86ish_aesni_avx,
		     cpu_feature_p(CPUFEAT_X86_AVX) &&
		     cpu_feature_p(CPUFEAT_X86_AESNI));
  DISPATCH_PICK_COND(rijndael_eblk, rijndael_eblk_x86ish_aesni,
		     cpu_feature_p(CPUFEAT_X86_AESNI));
#endif
#if CPUFAM_ARMEL && HAVE_AS_ARMV8_CRYPTO
  DISPATCH_PICK_COND(rijndael_eblk, rijndael_eblk_arm_crypto,
		     cpu_feature_p(CPUFEAT_ARM_AES));
#endif
#if CPUFAM_ARM64
  DISPATCH_PICK_COND(rijndael_eblk, rijndael_eblk_arm64_crypto,
		     cpu_feature_p(CPUFEAT_ARM_AES));
#endif
  DISPATCH_PICK_FALLBACK(rijndael_eblk, simple_eblk);
}

static rijndael_dblk__functype *pick_dblk(void)
{
#if CPUFAM_X86 || CPUFAM_AMD64
  DISPATCH_PICK_COND(rijndael_dblk, rijndael_dblk_x86ish_aesni_avx,
		     cpu_feature_p(CPUFEAT_X86_AVX) &&
		     cpu_feature_p(CPUFEAT_X86_AESNI));
  DISPATCH_PICK_COND(rijndael_dblk, rijndael_dblk_x86ish_aesni,
		     cpu_feature_p(CPUFEAT_X86_AESNI));
#endif
#if CPUFAM_ARMEL && HAVE_AS_ARMV8_CRYPTO
  DISPATCH_PICK_COND(rijndael_dblk, rijndael_dblk_arm_crypto,
		     cpu_feature_p(CPUFEAT_ARM_AES));
#endif
#if CPUFAM_ARM64
  DISPATCH_PICK_COND(rijndael_dblk, rijndael_dblk_arm64_crypto,
		     cpu_feature_p(CPUFEAT_ARM_AES));
#endif
  DISPATCH_PICK_FALLBACK(rijndael_dblk, simple_dblk);
}

#define DO(what, t, aa, bb, cc, dd, a, b, c, d, w) do {			\
  aa = what(t, a, b, c, d) ^ *w++;					\
  bb = what(t, b, c, d, a) ^ *w++;					\
  cc = what(t, c, d, a, b) ^ *w++;					\
  dd = what(t, d, a, b, c) ^ *w++;					\
} while (0)

#define UNDO(what, t, aa, bb, cc, dd, a, b, c, d, w) do {		\
  aa = what(t, a, d, c, b) ^ *w++;					\
  bb = what(t, b, a, d, c) ^ *w++;					\
  cc = what(t, c, b, a, d) ^ *w++;					\
  dd = what(t, d, c, b, a) ^ *w++;					\
} while (0)

static void simple_eblk(const rijndael_ctx *k, const uint32 *s, uint32 *dst)
{
  uint32 a = s[0], b = s[1], c = s[2], d = s[3];
  uint32 aa, bb, cc, dd;
  const uint32 *w = k->w;

  a ^= *w++; b ^= *w++; c ^= *w++; d ^= *w++;
  aa = a; bb = b; cc = c; dd = d;

  switch (k->nr) {
    case 14:
      DO(MIX, T, aa, bb, cc, dd, a, b, c, d, w);
    case 13:
      DO(MIX, T, a, b, c, d, aa, bb, cc, dd, w);
    case 12:
      DO(MIX, T, aa, bb, cc, dd, a, b, c, d, w);
    case 11:
      DO(MIX, T, a, b, c, d, aa, bb, cc, dd, w);
    case 10:
    default:
      DO(MIX, T, aa, bb, cc, dd, a, b, c, d, w);
      DO(MIX, T, a, b, c, d, aa, bb, cc, dd, w);
      DO(MIX, T, aa, bb, cc, dd, a, b, c, d, w);
      DO(MIX, T, a, b, c, d, aa, bb, cc, dd, w);
      DO(MIX, T, aa, bb, cc, dd, a, b, c, d, w);
      DO(MIX, T, a, b, c, d, aa, bb, cc, dd, w);
      DO(MIX, T, aa, bb, cc, dd, a, b, c, d, w);
      DO(MIX, T, a, b, c, d, aa, bb, cc, dd, w);
      DO(MIX, T, aa, bb, cc, dd, a, b, c, d, w);
  }
  DO(SUB, S, a, b, c, d, aa, bb, cc, dd, w);

  dst[0] = a; dst[1] = b; dst[2] = c; dst[3] = d;
}

static void simple_dblk(const rijndael_ctx *k, const uint32 *s, uint32 *dst)
{
  uint32 a = s[0], b = s[1], c = s[2], d = s[3];
  uint32 aa, bb, cc, dd;
  const uint32 *w = k->wi;

  a ^= *w++; b ^= *w++; c ^= *w++; d ^= *w++;
  aa = a; bb = b; cc = c; dd = d;

  switch (k->nr) {
    case 14:
      UNDO(MIX, TI, aa, bb, cc, dd, a, b, c, d, w);
    case 13:
      UNDO(MIX, TI, a, b, c, d, aa, bb, cc, dd, w);
    case 12:
      UNDO(MIX, TI, aa, bb, cc, dd, a, b, c, d, w);
    case 11:
      UNDO(MIX, TI, a, b, c, d, aa, bb, cc, dd, w);
    case 10:
    default:
      UNDO(MIX, TI, aa, bb, cc, dd, a, b, c, d, w);
      UNDO(MIX, TI, a, b, c, d, aa, bb, cc, dd, w);
      UNDO(MIX, TI, aa, bb, cc, dd, a, b, c, d, w);
      UNDO(MIX, TI, a, b, c, d, aa, bb, cc, dd, w);
      UNDO(MIX, TI, aa, bb, cc, dd, a, b, c, d, w);
      UNDO(MIX, TI, a, b, c, d, aa, bb, cc, dd, w);
      UNDO(MIX, TI, aa, bb, cc, dd, a, b, c, d, w);
      UNDO(MIX, TI, a, b, c, d, aa, bb, cc, dd, w);
      UNDO(MIX, TI, aa, bb, cc, dd, a, b, c, d, w);
  }
  UNDO(SUB, SI, a, b, c, d, aa, bb, cc, dd, w);

  dst[0] = a; dst[1] = b; dst[2] = c; dst[3] = d;
}

BLKC_TEST(RIJNDAEL, rijndael)

/*----- That's all, folks -------------------------------------------------*/