source: trunk/minix/drivers/random/aes/rijndael_api.c@ 9

/*      rijndael-api.c - Rijndael encryption programming interface.
 *                                                      Author: Kees J. Bot
 *                                                              3 Nov 2000
 * Heavily based on the original API code by Antoon Bosselaers,
 * Vincent Rijmen, and Paulo Barreto, but with a different interface.
 *
 * Read this code top to bottom, not all comments are repeated.
 */

#include <stdlib.h>
#include <string.h>
#include <sys/types.h>

#include "rijndael-alg.h"
#include "rijndael-api.h"

/* Map a byte (?) address to a word address or vv. */
#define W(a)    ((word32 *) (a))
#define B(a)    ((word8 *) (a))

#if STRICT_ALIGN
/* This machine checks alignment religiously.  (The code is not proper with
 * respect to alignment.  We need a compiler that doesn't muck about with byte
 * arrays that follow words in structs, and that places automatic variables
 * at word boundaries if not odd-sized.  Most compilers are this nice.)
 */

#define aligned(a)              (((unsigned) (a) & 3) == 0)
#define aligned2(a1, a2)        aligned((unsigned) (a1) | (unsigned) (a2))

static void blockcpy(void *dst, const void *src)
{
    int i= 0;

    do {
        B(dst)[i+0] = B(src)[i+0];
        B(dst)[i+1] = B(src)[i+1];
        B(dst)[i+2] = B(src)[i+2];
        B(dst)[i+3] = B(src)[i+3];
    } while ((i += 4) < 16);
}

#else /* !STRICT_ALIGN */
/* This machine doesn't mind misaligned accesses much. */

#define aligned(a)              ((void) (a), 1)
#define aligned2(a1, a2)        ((void) (a1), (void) (a2), 1)

#if __GNUC__
__inline
#endif
static void blockcpy(void *dst, const void *src)
{
    W(dst)[0] = W(src)[0];
    W(dst)[1] = W(src)[1];
    W(dst)[2] = W(src)[2];
    W(dst)[3] = W(src)[3];
}

#endif /* !STRICT_ALIGN */

#define between(a, c, z)        ((unsigned) (c) - (a) <= (unsigned) (z) - (a))

int rijndael_makekey(rd_keyinstance *key,
        size_t keylen, const void *keymaterial)
{
    word8 k[MAXKC][4];

    /* Initialize key schedule: */
    if (keylen == RD_KEY_HEX) {
        const word8 *kp;
        int c, b;

        kp= keymaterial;
        keylen= 0;

        for (;;) {
            c= *kp++;
            if (between('0', c, '9')) b= (c - '0' + 0x0) << 4;
            else
            if (between('a', c, 'f')) b= (c - 'a' + 0xa) << 4;
            else
            if (between('A', c, 'F')) b= (c - 'A' + 0xA) << 4;
            else break;

            c= *kp++;
            if (between('0', c, '9')) b |= (c - '0' + 0x0);
            else
            if (between('a', c, 'f')) b |= (c - 'a' + 0xa);
            else
            if (between('A', c, 'F')) b |= (c - 'A' + 0xA);
            else break;

            if (keylen >= 256/8) return RD_BAD_KEY_MAT;
            B(k)[keylen++] = b;
        }
        if (c != 0) return RD_BAD_KEY_MAT;

        if (keylen != 128/8 && keylen != 192/8 && keylen != 256/8) {
            return RD_BAD_KEY_MAT;
        }
    } else {
        if (keylen != 128/8 && keylen != 192/8 && keylen != 256/8) {
            return RD_BAD_KEY_MAT;
        }
        memcpy(k, keymaterial, keylen);
    }

    key->rounds= keylen * 8 / 32 + 6;

    rijndael_KeySched(k, key->encsched, key->rounds);
    memcpy(key->decsched, key->encsched, sizeof(key->decsched));
    rijndael_KeyEncToDec(key->decsched, key->rounds);

    return 0;
}

ssize_t rijndael_ecb_encrypt(rd_keyinstance *key,
        const void *input, void *output, size_t length, void *dummyIV)
{
    /* Encrypt blocks of data in Electronic Codebook mode. */
    const word8 *inp= input;
    word8 *outp= output;
    size_t i, nr_blocks, extra;
    word32 in[4], out[4];
    word8 t;

    /* Compute the number of whole blocks, and the extra bytes beyond the
     * last block.  Those extra bytes, if any, are encrypted by stealing
     * enough bytes from the previous encrypted block to make a whole block.
     * This is done by encrypting the last block, exchanging the first few
     * encrypted bytes with the extra bytes, and encrypting the last whole
     * block again.
     */
    nr_blocks= length / 16;
    if ((extra= (length % 16)) > 0) {
        if (nr_blocks == 0) return RD_BAD_BLOCK_LENGTH;
        nr_blocks--;
    }

    /* Encrypt a number of blocks. */
    if (aligned2(inp, outp)) {
        for (i= 0; i < nr_blocks; i++) {
            rijndael_Encrypt(inp, outp, key->encsched, key->rounds);
            inp += 16;
            outp += 16;
        }
    } else {
        for (i= 0; i < nr_blocks; i++) {
            blockcpy(in, inp);
            rijndael_Encrypt(in, out, key->encsched, key->rounds);
            blockcpy(outp, out);
            inp += 16;
            outp += 16;
        }
    }

    /* Encrypt extra bytes by stealing from the last full block. */
    if (extra > 0) {
        blockcpy(in, inp);
        rijndael_Encrypt(in, out, key->encsched, key->rounds);
        for (i= 0; i < extra; i++) {
            t= B(out)[i];
            B(out)[i] = inp[16 + i];
            outp[16 + i] = t;
        }
        rijndael_Encrypt(out, out, key->encsched, key->rounds);
        blockcpy(outp, out);
    }
    return length;
}

ssize_t rijndael_ecb_decrypt(rd_keyinstance *key,
        const void *input, void *output, size_t length, void *dummyIV)
{
    /* Decrypt blocks of data in Electronic Codebook mode. */
    const word8 *inp= input;
    word8 *outp= output;
    size_t i, nr_blocks, extra;
    word32 in[4], out[4];
    word8 t;

    nr_blocks= length / 16;
    if ((extra= (length % 16)) > 0) {
        if (nr_blocks == 0) return RD_BAD_BLOCK_LENGTH;
        nr_blocks--;
    }

    /* Decrypt a number of blocks. */
    if (aligned2(inp, outp)) {
        for (i= 0; i < nr_blocks; i++) {
            rijndael_Decrypt(inp, outp, key->decsched, key->rounds);
            inp += 16;
            outp += 16;
        }
    } else {
        for (i= 0; i < nr_blocks; i++) {
            blockcpy(in, inp);
            rijndael_Decrypt(in, out, key->decsched, key->rounds);
            blockcpy(outp, out);
            inp += 16;
            outp += 16;
        }
    }

    /* Decrypt extra bytes that stole from the last full block. */
    if (extra > 0) {
        blockcpy(in, inp);
        rijndael_Decrypt(in, out, key->decsched, key->rounds);
        for (i= 0; i < extra; i++) {
            t= B(out)[i];
            B(out)[i] = inp[16 + i];
            outp[16 + i] = t;
        }
        rijndael_Decrypt(out, out, key->decsched, key->rounds);
        blockcpy(outp, out);
    }
    return length;
}

ssize_t rijndael_cbc_encrypt(rd_keyinstance *key,
        const void *input, void *output, size_t length, void *IV)
{
    /* Encrypt blocks of data in Cypher Block Chaining mode. */
    const word8 *inp= input;
    word8 *outp= output;
    size_t i, nr_blocks, extra;
    word32 in[4], out[4], iv[4], *ivp;
    word8 t;

    nr_blocks= length / 16;
    if ((extra= (length % 16)) > 0) {
        if (nr_blocks == 0) return RD_BAD_BLOCK_LENGTH;
        nr_blocks--;
    }

    /* Each input block is first XORed with the previous encryption result.
     * The "Initialization Vector" is used to XOR the first block with.
     * When done the last crypted block is stored back as the new IV to be
     * used for another call to this function.
     */
    ivp= aligned(IV) ? IV : (blockcpy(iv, IV), iv);

    if (aligned2(inp, outp)) {
        for (i= 0; i < nr_blocks; i++) {
            in[0] = W(inp)[0] ^ ivp[0];
            in[1] = W(inp)[1] ^ ivp[1];
            in[2] = W(inp)[2] ^ ivp[2];
            in[3] = W(inp)[3] ^ ivp[3];
            rijndael_Encrypt(in, outp, key->encsched, key->rounds);
            ivp= W(outp);
            inp += 16;
            outp += 16;
        }
    } else {
        for (i= 0; i < nr_blocks; i++) {
            blockcpy(in, inp);
            in[0] ^= ivp[0];
            in[1] ^= ivp[1];
            in[2] ^= ivp[2];
            in[3] ^= ivp[3];
            rijndael_Encrypt(in, out, key->encsched, key->rounds);
            blockcpy(outp, out);
            ivp= out;
            inp += 16;
            outp += 16;
        }
    }
    if (extra > 0) {
        blockcpy(in, inp);
        in[0] ^= ivp[0];
        in[1] ^= ivp[1];
        in[2] ^= ivp[2];
        in[3] ^= ivp[3];
        rijndael_Encrypt(in, out, key->encsched, key->rounds);
        for (i= 0; i < extra; i++) {
            t= B(out)[i];
            B(out)[i] ^= inp[16 + i];
            outp[16 + i] = t;
        }
        rijndael_Encrypt(out, out, key->encsched, key->rounds);
        blockcpy(outp, out);
        ivp= out;
    }
    blockcpy(IV, ivp);          /* Store last IV back. */
    return length;
}

ssize_t rijndael_cbc_decrypt(rd_keyinstance *key,
        const void *input, void *output, size_t length, void *IV)
{
    /* Decrypt blocks of data in Cypher Block Chaining mode. */
    const word8 *inp= input;
    word8 *outp= output;
    size_t i, nr_blocks, extra;
    word32 in[4], out[4], iv[4];
    word8 t;

    nr_blocks= length / 16;
    if ((extra= (length % 16)) > 0) {
        if (nr_blocks == 0) return RD_BAD_BLOCK_LENGTH;
        nr_blocks--;
    }

    blockcpy(iv, IV);

    if (aligned2(inp, outp)) {
        for (i= 0; i < nr_blocks; i++) {
            rijndael_Decrypt(inp, out, key->decsched, key->rounds);
            out[0] ^= iv[0];
            out[1] ^= iv[1];
            out[2] ^= iv[2];
            out[3] ^= iv[3];
            iv[0] = W(inp)[0];
            iv[1] = W(inp)[1];
            iv[2] = W(inp)[2];
            iv[3] = W(inp)[3];
            W(outp)[0] = out[0];
            W(outp)[1] = out[1];
            W(outp)[2] = out[2];
            W(outp)[3] = out[3];
            inp += 16;
            outp += 16;
        }
    } else {
        for (i= 0; i < nr_blocks; i++) {
            blockcpy(in, inp);
            rijndael_Decrypt(in, out, key->decsched, key->rounds);
            out[0] ^= iv[0];
            out[1] ^= iv[1];
            out[2] ^= iv[2];
            out[3] ^= iv[3];
            iv[0] = in[0];
            iv[1] = in[1];
            iv[2] = in[2];
            iv[3] = in[3];
            blockcpy(outp, out);
            inp += 16;
            outp += 16;
        }
    }
    if (extra > 0) {
        blockcpy(in, inp);
        blockcpy(IV, in);
        rijndael_Decrypt(in, out, key->decsched, key->rounds);
        for (i= 0; i < extra; i++) {
            t= B(out)[i] ^ inp[16 + i];
            B(out)[i] = inp[16 + i];
            outp[16 + i] = t;
        }
        rijndael_Decrypt(out, out, key->decsched, key->rounds);
        out[0] ^= iv[0];
        out[1] ^= iv[1];
        out[2] ^= iv[2];
        out[3] ^= iv[3];
        blockcpy(outp, out);
    } else {
        blockcpy(IV, iv);
    }
    return length;
}

ssize_t rijndael_cfb1_encrypt(rd_keyinstance *key,
        const void *input, void *output, size_t length, void *IV)
{
    /* Encrypt blocks of data in Cypher Feedback mode, 1 bit at a time. */
    const word8 *inp= input;
    word8 *outp= output;
    word8 t;
    size_t i;
    int b;
    word32 iv[4], civ[4];

    blockcpy(iv, IV);

    for (i= 0; i < length; i++) {
        t= *inp++;
        for (b= 0; b < 8; b++) {
            rijndael_Encrypt(iv, civ, key->encsched, key->rounds);
            t ^= (B(civ)[0] & 0x80) >> b;
            B(iv)[ 0] = (B(iv)[ 0] << 1) | (B(iv)[ 1] >> 7);
            B(iv)[ 1] = (B(iv)[ 1] << 1) | (B(iv)[ 2] >> 7);
            B(iv)[ 2] = (B(iv)[ 2] << 1) | (B(iv)[ 3] >> 7);
            B(iv)[ 3] = (B(iv)[ 3] << 1) | (B(iv)[ 4] >> 7);
            B(iv)[ 4] = (B(iv)[ 4] << 1) | (B(iv)[ 5] >> 7);
            B(iv)[ 5] = (B(iv)[ 5] << 1) | (B(iv)[ 6] >> 7);
            B(iv)[ 6] = (B(iv)[ 6] << 1) | (B(iv)[ 7] >> 7);
            B(iv)[ 7] = (B(iv)[ 7] << 1) | (B(iv)[ 8] >> 7);
            B(iv)[ 8] = (B(iv)[ 8] << 1) | (B(iv)[ 9] >> 7);
            B(iv)[ 9] = (B(iv)[ 9] << 1) | (B(iv)[10] >> 7);
            B(iv)[10] = (B(iv)[10] << 1) | (B(iv)[11] >> 7);
            B(iv)[11] = (B(iv)[11] << 1) | (B(iv)[12] >> 7);
            B(iv)[12] = (B(iv)[12] << 1) | (B(iv)[13] >> 7);
            B(iv)[13] = (B(iv)[13] << 1) | (B(iv)[14] >> 7);
            B(iv)[14] = (B(iv)[14] << 1) | (B(iv)[15] >> 7);
            B(iv)[15] = (B(iv)[15] << 1) | ((t >> (7-b)) & 1);
        }
        *outp++ = t;
    }
    blockcpy(IV, iv);
    return length;
}

ssize_t rijndael_cfb1_decrypt(rd_keyinstance *key,
        const void *input, void *output, size_t length, void *IV)
{
    /* Decrypt blocks of data in Cypher Feedback mode, 1 bit at a time. */
    const word8 *inp= input;
    word8 *outp= output;
    word8 t;
    size_t i;
    int b;
    word32 iv[4], civ[4];

    blockcpy(iv, IV);

    for (i= 0; i < length; i++) {
        t= *inp++;
        for (b= 0; b < 8; b++) {
            rijndael_Encrypt(iv, civ, key->encsched, key->rounds);
            B(iv)[ 0] = (B(iv)[ 0] << 1) | (B(iv)[ 1] >> 7);
            B(iv)[ 1] = (B(iv)[ 1] << 1) | (B(iv)[ 2] >> 7);
            B(iv)[ 2] = (B(iv)[ 2] << 1) | (B(iv)[ 3] >> 7);
            B(iv)[ 3] = (B(iv)[ 3] << 1) | (B(iv)[ 4] >> 7);
            B(iv)[ 4] = (B(iv)[ 4] << 1) | (B(iv)[ 5] >> 7);
            B(iv)[ 5] = (B(iv)[ 5] << 1) | (B(iv)[ 6] >> 7);
            B(iv)[ 6] = (B(iv)[ 6] << 1) | (B(iv)[ 7] >> 7);
            B(iv)[ 7] = (B(iv)[ 7] << 1) | (B(iv)[ 8] >> 7);
            B(iv)[ 8] = (B(iv)[ 8] << 1) | (B(iv)[ 9] >> 7);
            B(iv)[ 9] = (B(iv)[ 9] << 1) | (B(iv)[10] >> 7);
            B(iv)[10] = (B(iv)[10] << 1) | (B(iv)[11] >> 7);
            B(iv)[11] = (B(iv)[11] << 1) | (B(iv)[12] >> 7);
            B(iv)[12] = (B(iv)[12] << 1) | (B(iv)[13] >> 7);
            B(iv)[13] = (B(iv)[13] << 1) | (B(iv)[14] >> 7);
            B(iv)[14] = (B(iv)[14] << 1) | (B(iv)[15] >> 7);
            B(iv)[15] = (B(iv)[15] << 1) | ((t >> (7-b)) & 1);
            t ^= (B(civ)[0] & 0x80) >> b;
        }
        *outp++ = t;
    }
    blockcpy(IV, iv);
    return length;
}

ssize_t rijndael_cfb8_encrypt(rd_keyinstance *key,
        const void *input, void *output, size_t length, void *IV)
{
    /* Encrypt blocks of data in Cypher Feedback mode, 8 bits at a time. */
    const word8 *inp= input;
    word8 *outp= output;
    word8 t;
    size_t i;
    word32 iv[4], civ[4];

    blockcpy(iv, IV);

    for (i= 0; i < length; i++) {
        t= *inp++;
        rijndael_Encrypt(iv, civ, key->encsched, key->rounds);
        t ^= B(civ)[0];
        B(iv)[ 0] = B(iv)[ 1];
        B(iv)[ 1] = B(iv)[ 2];
        B(iv)[ 2] = B(iv)[ 3];
        B(iv)[ 3] = B(iv)[ 4];
        B(iv)[ 4] = B(iv)[ 5];
        B(iv)[ 5] = B(iv)[ 6];
        B(iv)[ 6] = B(iv)[ 7];
        B(iv)[ 7] = B(iv)[ 8];
        B(iv)[ 8] = B(iv)[ 9];
        B(iv)[ 9] = B(iv)[10];
        B(iv)[10] = B(iv)[11];
        B(iv)[11] = B(iv)[12];
        B(iv)[12] = B(iv)[13];
        B(iv)[13] = B(iv)[14];
        B(iv)[14] = B(iv)[15];
        B(iv)[15] = t;
        *outp++ = t;
    }
    blockcpy(IV, iv);
    return length;
}

ssize_t rijndael_cfb8_decrypt(rd_keyinstance *key,
        const void *input, void *output, size_t length, void *IV)
{
    /* Decrypt blocks of data in Cypher Feedback mode, 1 byte at a time. */
    const word8 *inp= input;
    word8 *outp= output;
    word8 t;
    size_t i;
    word32 iv[4], civ[4];

    blockcpy(iv, IV);

    for (i= 0; i < length; i++) {
        t= *inp++;
        rijndael_Encrypt(iv, civ, key->encsched, key->rounds);
        B(iv)[ 0] = B(iv)[ 1];
        B(iv)[ 1] = B(iv)[ 2];
        B(iv)[ 2] = B(iv)[ 3];
        B(iv)[ 3] = B(iv)[ 4];
        B(iv)[ 4] = B(iv)[ 5];
        B(iv)[ 5] = B(iv)[ 6];
        B(iv)[ 6] = B(iv)[ 7];
        B(iv)[ 7] = B(iv)[ 8];
        B(iv)[ 8] = B(iv)[ 9];
        B(iv)[ 9] = B(iv)[10];
        B(iv)[10] = B(iv)[11];
        B(iv)[11] = B(iv)[12];
        B(iv)[12] = B(iv)[13];
        B(iv)[13] = B(iv)[14];
        B(iv)[14] = B(iv)[15];
        B(iv)[15] = t;
        t ^= B(civ)[0];
        *outp++ = t;
    }
    blockcpy(IV, iv);
    return length;
}

ssize_t rijndael_pad(void *input, size_t length)
{
    /* Adds at most one block of RFC-2040 style padding to the input to make
     * it a whole number of blocks for easier encryption.  To be used if the
     * input may be less then one block in size, otherwise let the encryption
     * routines use cypher stealing.  The input buffer should allow enough
     * space for the padding.  The new length of the input is returned.
     */
    word8 *inp= input;
    size_t padlen;

    /* Add padding up until the next block boundary. */
    padlen= 16 - (length % 16);
    memset(inp + length, padlen, padlen);
    return length + padlen;
}

ssize_t rijndael_unpad(const void *input, size_t length)
{
    /* Remove RFC-2040 style padding after decryption.  The true length of
     * the input is returned, or the usual errors if the padding is incorrect.
     */
    const word8 *inp= input;
    size_t i, padlen;

    if (length == 0 || (length % 16) != 0) return RD_BAD_BLOCK_LENGTH;
    padlen = inp[length-1];
    if (padlen <= 0 || padlen > 16) return RD_BAD_DATA;
    for (i= 2; i <= padlen; i++) {
        if (inp[length-i] != padlen) return RD_BAD_DATA;
    }
    return length - padlen;
}

#ifdef INTERMEDIATE_VALUE_KAT

void cipherEncryptUpdateRounds(rd_keyinstance *key,
        const void *input, void *output, int rounds)
{
    /* Encrypt a block only a specified number of rounds. */
    word8 block[4][4];

    blockcpy(block, input);

    rijndaelEncryptRound(block, key->encsched, key->rounds, rounds);

    blockcpy(output, block);
}

void cipherDecryptUpdateRounds(rd_keyinstance *key,
        const void *input, void *output, int rounds)
{
    /* Decrypt a block only a specified number of rounds. */
    word8 block[4][4];

    blockcpy(block, input);

    rijndaelDecryptRound(block, key->decsched, key->rounds, rounds);

    blockcpy(output, block);
}
#endif /* INTERMEDIATE_VALUE_KAT */

/*
 * $PchId: rijndael_api.c,v 1.2 2001/01/10 22:01:20 philip Exp $
 */