crypto/modes/ecb.c

	ecb.c revision 23c57df72989c916b3e98084eb88d48777999691
/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#pragma ident   "%Z%%M% %I% %E% SMI"

#ifndef _KERNEL
#include <strings.h>
#include <limits.h>
#include <assert.h>
#include <security/cryptoki.h>
#endif

#include <sys/types.h>
#include <modes/modes.h>
#include <sys/crypto/common.h>
#include <sys/crypto/impl.h>

/*
 * Algorithm independent ECB functions.
 */
int
ecb_cipher_contiguous_blocks(ecb_ctx_t *ctx, char *data, size_t length,
    crypto_data_t *out, size_t block_size,
    int (*cipher)(const void *ks, const uint8_t *pt, uint8_t *ct))
{
    size_t remainder = length;
    size_t need;
    uint8_t *datap = (uint8_t *)data;
    uint8_t *blockp;
    uint8_t *lastp;
    void *iov_or_mp;
    offset_t offset;
    uint8_t *out_data_1;
    uint8_t *out_data_2;
    size_t out_data_1_len;

    if (length + ctx->cc_remainder_len < block_size) {
        /* accumulate bytes here and return */
        bcopy(datap,
            (uint8_t *)ctx->cc_remainder + ctx->cc_remainder_len,
            length);
        ctx->cc_remainder_len += length;
        ctx->cc_copy_to = datap;
        return (CRYPTO_SUCCESS);
    }

    lastp = (uint8_t *)ctx->cc_iv;
    if (out != NULL)
        crypto_init_ptrs(out, &iov_or_mp, &offset);

    do {
        /* Unprocessed data from last call. */
        if (ctx->cc_remainder_len > 0) {
            need = block_size - ctx->cc_remainder_len;

            if (need > remainder)
                return (CRYPTO_DATA_LEN_RANGE);

            bcopy(datap, &((uint8_t *)ctx->cc_remainder)
                [ctx->cc_remainder_len], need);

            blockp = (uint8_t *)ctx->cc_remainder;
        } else {
            blockp = datap;
        }

        if (out == NULL) {
            cipher(ctx->cc_keysched, blockp, blockp);

            ctx->cc_lastp = blockp;
            lastp = blockp;

            if (ctx->cc_remainder_len > 0) {
                bcopy(blockp, ctx->cc_copy_to,
                    ctx->cc_remainder_len);
                bcopy(blockp + ctx->cc_remainder_len, datap,
                    need);
            }
        } else {
            cipher(ctx->cc_keysched, blockp, lastp);
            crypto_get_ptrs(out, &iov_or_mp, &offset, &out_data_1,
                &out_data_1_len, &out_data_2, block_size);

            /* copy block to where it belongs */
            bcopy(lastp, out_data_1, out_data_1_len);
            if (out_data_2 != NULL) {
                bcopy(lastp + out_data_1_len, out_data_2,
                    block_size - out_data_1_len);
            }
            /* update offset */
            out->cd_offset += block_size;
        }

        /* Update pointer to next block of data to be processed. */
        if (ctx->cc_remainder_len != 0) {
            datap += need;
            ctx->cc_remainder_len = 0;
        } else {
            datap += block_size;
        }

        remainder = (size_t)&data[length] - (size_t)datap;

        /* Incomplete last block. */
        if (remainder > 0 && remainder < block_size) {
            bcopy(datap, ctx->cc_remainder, remainder);
            ctx->cc_remainder_len = remainder;
            ctx->cc_copy_to = datap;
            goto out;
        }
        ctx->cc_copy_to = NULL;

    } while (remainder > 0);

out:
    return (CRYPTO_SUCCESS);
}

/* ARGSUSED */
void *
ecb_alloc_ctx(int kmflag)
{
    ecb_ctx_t *ecb_ctx;

#ifdef _KERNEL
    if ((ecb_ctx = kmem_zalloc(sizeof (ecb_ctx_t), kmflag)) == NULL)
#else
    if ((ecb_ctx = calloc(1, sizeof (ecb_ctx_t))) == NULL)
#endif
        return (NULL);

    ecb_ctx->cc_flags = ECB_MODE;
    return (ecb_ctx);
}