[CRYPTO] ctr: Add countersize

This patch adds countersize to CTR mode.
The template is now ctr(algo,noncesize,ivsize,countersize).

For example, ctr(aes,4,8,4) indicates the counterblock
will be composed of a salt/nonce that is 4 bytes, an iv
that is 8 bytes and the counter is 4 bytes.

When noncesize + ivsize < blocksize, CTR initializes the
last block - ivsize - noncesize portion of the block to
zero.  Otherwise the counter block is composed of the IV
(and nonce if necessary).

If noncesize + ivsize == blocksize, then this indicates that
user is passing in entire counterblock. Thus countersize
indicates the amount of bytes in counterblock to use as
the counter for incrementing. CTR will increment counter
portion by 1, and begin encryption with that value.

Note that CTR assumes the counter portion of the block that
will be incremented is stored in big endian.

Signed-off-by: Joy Latten <latten@austin.ibm.com>
Signed-off-by: Herbert Xu <herbert@gondor.apana.org.au>

diff --git a/crypto/ctr.c b/crypto/ctr.c
index 810d5ec2d5d2438065a8021a5302cd7d254809e8..b974a9f9b8797e1329b78c57b4e1b42e3e017cb8 100644 (file)
--- a/crypto/ctr.c
+++ b/crypto/ctr.c
@@ -23,6 +23,7 @@ struct ctr_instance_ctx {
        struct crypto_spawn alg;
        unsigned int noncesize;
        unsigned int ivsize;
+       unsigned int countersize;
 };
 
 struct crypto_ctr_ctx {
@@ -186,7 +187,6 @@ static int crypto_ctr_crypt(struct blkcipher_desc *desc,
        unsigned long alignmask = crypto_cipher_alignmask(child);
        u8 cblk[bsize + alignmask];
        u8 *counterblk = (u8 *)ALIGN((unsigned long)cblk, alignmask + 1);
-       unsigned int countersize;
        int err;
 
        blkcipher_walk_init(&walk, dst, src, nbytes);
@@ -198,18 +198,18 @@ static int crypto_ctr_crypt(struct blkcipher_desc *desc,
        memcpy(counterblk + ictx->noncesize, walk.iv, ictx->ivsize);
 
        /* initialize counter portion of counter block */
-       countersize = bsize - ictx->noncesize - ictx->ivsize;
-       ctr_inc_quad(counterblk + (bsize - countersize), countersize);
+       ctr_inc_quad(counterblk + (bsize - ictx->countersize),
+                    ictx->countersize);
 
        while (walk.nbytes) {
                if (walk.src.virt.addr == walk.dst.virt.addr)
                        nbytes = crypto_ctr_crypt_inplace(&walk, child,
                                                          counterblk,
-                                                         countersize);
+                                                         ictx->countersize);
                else
                        nbytes = crypto_ctr_crypt_segment(&walk, child,
                                                          counterblk,
-                                                         countersize);
+                                                         ictx->countersize);
 
                err = blkcipher_walk_done(desc, &walk, nbytes);
        }
@@ -251,6 +251,7 @@ static struct crypto_instance *crypto_ctr_alloc(struct rtattr **tb)
        struct ctr_instance_ctx *ictx;
        unsigned int noncesize;
        unsigned int ivsize;
+       unsigned int countersize;
        int err;
 
        err = crypto_check_attr_type(tb, CRYPTO_ALG_TYPE_BLKCIPHER);
@@ -270,9 +271,17 @@ static struct crypto_instance *crypto_ctr_alloc(struct rtattr **tb)
        if (err)
                goto out_put_alg;
 
-       /* verify size of nonce + iv + counter */
+       err = crypto_attr_u32(tb[4], &countersize);
+       if (err)
+               goto out_put_alg;
+
+       /* verify size of nonce + iv + counter
+        * counter must be >= 4 bytes.
+        */
        err = -EINVAL;
-       if ((noncesize + ivsize) >= alg->cra_blocksize)
+       if (((noncesize + ivsize + countersize) < alg->cra_blocksize) ||
+           ((noncesize + ivsize) > alg->cra_blocksize) ||
+           (countersize > alg->cra_blocksize) || (countersize < 4))
                goto out_put_alg;
 
        inst = kzalloc(sizeof(*inst) + sizeof(*ictx), GFP_KERNEL);
@@ -282,20 +291,21 @@ static struct crypto_instance *crypto_ctr_alloc(struct rtattr **tb)
 
        err = -ENAMETOOLONG;
        if (snprintf(inst->alg.cra_name, CRYPTO_MAX_ALG_NAME,
-                    "ctr(%s,%u,%u)", alg->cra_name, noncesize,
-                    ivsize) >= CRYPTO_MAX_ALG_NAME) {
+                    "ctr(%s,%u,%u,%u)", alg->cra_name, noncesize,
+                    ivsize, countersize) >= CRYPTO_MAX_ALG_NAME) {
                goto err_free_inst;
        }
 
        if (snprintf(inst->alg.cra_driver_name, CRYPTO_MAX_ALG_NAME,
-                    "ctr(%s,%u,%u)", alg->cra_driver_name, noncesize,
-                    ivsize) >= CRYPTO_MAX_ALG_NAME) {
+                    "ctr(%s,%u,%u,%u)", alg->cra_driver_name, noncesize,
+                    ivsize, countersize) >= CRYPTO_MAX_ALG_NAME) {
                goto err_free_inst;
        }
 
        ictx = crypto_instance_ctx(inst);
        ictx->noncesize = noncesize;
        ictx->ivsize = ivsize;
+       ictx->countersize = countersize;
 
        err = crypto_init_spawn(&ictx->alg, alg, inst,
                CRYPTO_ALG_TYPE_MASK | CRYPTO_ALG_ASYNC);

diff --git a/crypto/tcrypt.c b/crypto/tcrypt.c
index 640cbcad32a1f08b287964664bc52d6d201553ce..aa84bc4f23134902e2061e513285b76f147ad154 100644 (file)
--- a/crypto/tcrypt.c
+++ b/crypto/tcrypt.c
@@ -969,9 +969,9 @@ static void do_test(void)
                            AES_XTS_ENC_TEST_VECTORS);
                test_cipher("xts(aes)", DECRYPT, aes_xts_dec_tv_template,
                            AES_XTS_DEC_TEST_VECTORS);
-               test_cipher("ctr(aes,4,8)", ENCRYPT, aes_ctr_enc_tv_template,
+               test_cipher("ctr(aes,4,8,4)", ENCRYPT, aes_ctr_enc_tv_template,
                            AES_CTR_ENC_TEST_VECTORS);
-               test_cipher("ctr(aes,4,8)", DECRYPT, aes_ctr_dec_tv_template,
+               test_cipher("ctr(aes,4,8,4)", DECRYPT, aes_ctr_dec_tv_template,
                            AES_CTR_DEC_TEST_VECTORS);
 
                //CAST5
@@ -1160,9 +1160,9 @@ static void do_test(void)
                            AES_XTS_ENC_TEST_VECTORS);
                test_cipher("xts(aes)", DECRYPT, aes_xts_dec_tv_template,
                            AES_XTS_DEC_TEST_VECTORS);
-               test_cipher("ctr(aes,4,8)", ENCRYPT, aes_ctr_enc_tv_template,
+               test_cipher("ctr(aes,4,8,4)", ENCRYPT, aes_ctr_enc_tv_template,
                            AES_CTR_ENC_TEST_VECTORS);
-               test_cipher("ctr(aes,4,8)", DECRYPT, aes_ctr_dec_tv_template,
+               test_cipher("ctr(aes,4,8,4)", DECRYPT, aes_ctr_dec_tv_template,
                            AES_CTR_DEC_TEST_VECTORS);
                break;