/* magma.c - GOST R 34.12-2015 (Magma) cipher implementation

 *

 * Copyright: 2017 Dmitry Eremin-Solenikov <dbaryshkov@gmail.com>

 *

 * 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.

 */

#if HAVE_CONFIG_H

# include "config.h"

#endif

#ifndef HAVE_NETTLE_MAGMA_SET_KEY

# include <assert.h>

# include <nettle/macros.h>

# include "nettle-write.h"

# include "magma.h"

# ifndef HAVE_NETTLE_GOST28147_SET_KEY

#  include "gost28147.h"

# else

#  include <nettle/gost28147.h>

# endif

void magma_set_key(struct magma_ctx *ctx, const uint8_t * key)

{

  unsigned i;

  for (i = 0; i < 8; i++, key += 4)

    ctx->key[i] = READ_UINT32(key);

}

void

magma_encrypt(const struct magma_ctx *ctx,

        size_t length, uint8_t * dst, const uint8_t * src)

{

  uint32_t block[2];

  assert(!(length % MAGMA_BLOCK_SIZE));

  while (length) {

    block[1] = READ_UINT32(src);

    src += 4;

    block[0] = READ_UINT32(src);

    src += 4;

    gost28147_encrypt_simple(ctx->key, gost28147_param_TC26_Z.sbox,

           block, block);

    WRITE_UINT32(dst, block[1]);

    dst += 4;

    WRITE_UINT32(dst, block[0]);

    dst += 4;

    length -= MAGMA_BLOCK_SIZE;

  }

}

void

magma_decrypt(const struct magma_ctx *ctx,

        size_t length, uint8_t * dst, const uint8_t * src)

{

  uint32_t block[2];

  assert(!(length % MAGMA_BLOCK_SIZE));

  while (length) {

    block[1] = READ_UINT32(src);

    src += 4;

    block[0] = READ_UINT32(src);

    src += 4;

    gost28147_decrypt_simple(ctx->key, gost28147_param_TC26_Z.sbox,

           block, block);

    WRITE_UINT32(dst, block[1]);

    dst += 4;

    WRITE_UINT32(dst, block[0]);

    dst += 4;

    length -= MAGMA_BLOCK_SIZE;

  }

}

#endif        /* HAVE_NETTLE_MAGMA_SET_KEY */