Merge branch 'akpm' (patches from Andrew)
[deliverable/linux.git] / crypto / rsa-pkcs1pad.c
index d9baefb7d5d1ad58d19e7ba9ed3be057af7a6304..877019a6d3ea81c3e130deb882e5ceccb1fe62ae 100644 (file)
@@ -101,50 +101,57 @@ struct pkcs1pad_inst_ctx {
 };
 
 struct pkcs1pad_request {
-       struct akcipher_request child_req;
-
        struct scatterlist in_sg[2], out_sg[1];
        uint8_t *in_buf, *out_buf;
+       struct akcipher_request child_req;
 };
 
 static int pkcs1pad_set_pub_key(struct crypto_akcipher *tfm, const void *key,
                unsigned int keylen)
 {
        struct pkcs1pad_ctx *ctx = akcipher_tfm_ctx(tfm);
-       int err, size;
+       int err;
+
+       ctx->key_size = 0;
 
        err = crypto_akcipher_set_pub_key(ctx->child, key, keylen);
+       if (err)
+               return err;
 
-       if (!err) {
-               /* Find out new modulus size from rsa implementation */
-               size = crypto_akcipher_maxsize(ctx->child);
+       /* Find out new modulus size from rsa implementation */
+       err = crypto_akcipher_maxsize(ctx->child);
+       if (err < 0)
+               return err;
 
-               ctx->key_size = size > 0 ? size : 0;
-               if (size <= 0)
-                       err = size;
-       }
+       if (err > PAGE_SIZE)
+               return -ENOTSUPP;
 
-       return err;
+       ctx->key_size = err;
+       return 0;
 }
 
 static int pkcs1pad_set_priv_key(struct crypto_akcipher *tfm, const void *key,
                unsigned int keylen)
 {
        struct pkcs1pad_ctx *ctx = akcipher_tfm_ctx(tfm);
-       int err, size;
+       int err;
+
+       ctx->key_size = 0;
 
        err = crypto_akcipher_set_priv_key(ctx->child, key, keylen);
+       if (err)
+               return err;
 
-       if (!err) {
-               /* Find out new modulus size from rsa implementation */
-               size = crypto_akcipher_maxsize(ctx->child);
+       /* Find out new modulus size from rsa implementation */
+       err = crypto_akcipher_maxsize(ctx->child);
+       if (err < 0)
+               return err;
 
-               ctx->key_size = size > 0 ? size : 0;
-               if (size <= 0)
-                       err = size;
-       }
+       if (err > PAGE_SIZE)
+               return -ENOTSUPP;
 
-       return err;
+       ctx->key_size = err;
+       return 0;
 }
 
 static int pkcs1pad_get_max_size(struct crypto_akcipher *tfm)
@@ -177,37 +184,36 @@ static int pkcs1pad_encrypt_sign_complete(struct akcipher_request *req, int err)
        struct crypto_akcipher *tfm = crypto_akcipher_reqtfm(req);
        struct pkcs1pad_ctx *ctx = akcipher_tfm_ctx(tfm);
        struct pkcs1pad_request *req_ctx = akcipher_request_ctx(req);
-       size_t pad_len = ctx->key_size - req_ctx->child_req.dst_len;
-       size_t chunk_len, pad_left;
-       struct sg_mapping_iter miter;
-
-       if (!err) {
-               if (pad_len) {
-                       sg_miter_start(&miter, req->dst,
-                                       sg_nents_for_len(req->dst, pad_len),
-                                       SG_MITER_ATOMIC | SG_MITER_TO_SG);
-
-                       pad_left = pad_len;
-                       while (pad_left) {
-                               sg_miter_next(&miter);
-
-                               chunk_len = min(miter.length, pad_left);
-                               memset(miter.addr, 0, chunk_len);
-                               pad_left -= chunk_len;
-                       }
-
-                       sg_miter_stop(&miter);
-               }
-
-               sg_pcopy_from_buffer(req->dst,
-                               sg_nents_for_len(req->dst, ctx->key_size),
-                               req_ctx->out_buf, req_ctx->child_req.dst_len,
-                               pad_len);
-       }
+       unsigned int pad_len;
+       unsigned int len;
+       u8 *out_buf;
+
+       if (err)
+               goto out;
+
+       len = req_ctx->child_req.dst_len;
+       pad_len = ctx->key_size - len;
+
+       /* Four billion to one */
+       if (likely(!pad_len))
+               goto out;
+
+       out_buf = kzalloc(ctx->key_size, GFP_ATOMIC);
+       err = -ENOMEM;
+       if (!out_buf)
+               goto out;
+
+       sg_copy_to_buffer(req->dst, sg_nents_for_len(req->dst, len),
+                         out_buf + pad_len, len);
+       sg_copy_from_buffer(req->dst,
+                           sg_nents_for_len(req->dst, ctx->key_size),
+                           out_buf, ctx->key_size);
+       kzfree(out_buf);
+
+out:
        req->dst_len = ctx->key_size;
 
        kfree(req_ctx->in_buf);
-       kzfree(req_ctx->out_buf);
 
        return err;
 }
@@ -247,21 +253,8 @@ static int pkcs1pad_encrypt(struct akcipher_request *req)
                return -EOVERFLOW;
        }
 
-       if (ctx->key_size > PAGE_SIZE)
-               return -ENOTSUPP;
-
-       /*
-        * Replace both input and output to add the padding in the input and
-        * the potential missing leading zeros in the output.
-        */
-       req_ctx->child_req.src = req_ctx->in_sg;
-       req_ctx->child_req.src_len = ctx->key_size - 1;
-       req_ctx->child_req.dst = req_ctx->out_sg;
-       req_ctx->child_req.dst_len = ctx->key_size;
-
        req_ctx->in_buf = kmalloc(ctx->key_size - 1 - req->src_len,
-                       (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
-                       GFP_KERNEL : GFP_ATOMIC);
+                                 GFP_KERNEL);
        if (!req_ctx->in_buf)
                return -ENOMEM;
 
@@ -274,9 +267,7 @@ static int pkcs1pad_encrypt(struct akcipher_request *req)
        pkcs1pad_sg_set_buf(req_ctx->in_sg, req_ctx->in_buf,
                        ctx->key_size - 1 - req->src_len, req->src);
 
-       req_ctx->out_buf = kmalloc(ctx->key_size,
-                       (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
-                       GFP_KERNEL : GFP_ATOMIC);
+       req_ctx->out_buf = kmalloc(ctx->key_size, GFP_KERNEL);
        if (!req_ctx->out_buf) {
                kfree(req_ctx->in_buf);
                return -ENOMEM;
@@ -289,6 +280,10 @@ static int pkcs1pad_encrypt(struct akcipher_request *req)
        akcipher_request_set_callback(&req_ctx->child_req, req->base.flags,
                        pkcs1pad_encrypt_sign_complete_cb, req);
 
+       /* Reuse output buffer */
+       akcipher_request_set_crypt(&req_ctx->child_req, req_ctx->in_sg,
+                                  req->dst, ctx->key_size - 1, req->dst_len);
+
        err = crypto_akcipher_encrypt(&req_ctx->child_req);
        if (err != -EINPROGRESS &&
                        (err != -EBUSY ||
@@ -370,18 +365,7 @@ static int pkcs1pad_decrypt(struct akcipher_request *req)
        if (!ctx->key_size || req->src_len != ctx->key_size)
                return -EINVAL;
 
-       if (ctx->key_size > PAGE_SIZE)
-               return -ENOTSUPP;
-
-       /* Reuse input buffer, output to a new buffer */
-       req_ctx->child_req.src = req->src;
-       req_ctx->child_req.src_len = req->src_len;
-       req_ctx->child_req.dst = req_ctx->out_sg;
-       req_ctx->child_req.dst_len = ctx->key_size ;
-
-       req_ctx->out_buf = kmalloc(ctx->key_size,
-                       (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
-                       GFP_KERNEL : GFP_ATOMIC);
+       req_ctx->out_buf = kmalloc(ctx->key_size, GFP_KERNEL);
        if (!req_ctx->out_buf)
                return -ENOMEM;
 
@@ -392,6 +376,11 @@ static int pkcs1pad_decrypt(struct akcipher_request *req)
        akcipher_request_set_callback(&req_ctx->child_req, req->base.flags,
                        pkcs1pad_decrypt_complete_cb, req);
 
+       /* Reuse input buffer, output to a new buffer */
+       akcipher_request_set_crypt(&req_ctx->child_req, req->src,
+                                  req_ctx->out_sg, req->src_len,
+                                  ctx->key_size);
+
        err = crypto_akcipher_decrypt(&req_ctx->child_req);
        if (err != -EINPROGRESS &&
                        (err != -EBUSY ||
@@ -425,21 +414,8 @@ static int pkcs1pad_sign(struct akcipher_request *req)
                return -EOVERFLOW;
        }
 
-       if (ctx->key_size > PAGE_SIZE)
-               return -ENOTSUPP;
-
-       /*
-        * Replace both input and output to add the padding in the input and
-        * the potential missing leading zeros in the output.
-        */
-       req_ctx->child_req.src = req_ctx->in_sg;
-       req_ctx->child_req.src_len = ctx->key_size - 1;
-       req_ctx->child_req.dst = req_ctx->out_sg;
-       req_ctx->child_req.dst_len = ctx->key_size;
-
        req_ctx->in_buf = kmalloc(ctx->key_size - 1 - req->src_len,
-                       (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
-                       GFP_KERNEL : GFP_ATOMIC);
+                                 GFP_KERNEL);
        if (!req_ctx->in_buf)
                return -ENOMEM;
 
@@ -454,21 +430,14 @@ static int pkcs1pad_sign(struct akcipher_request *req)
        pkcs1pad_sg_set_buf(req_ctx->in_sg, req_ctx->in_buf,
                        ctx->key_size - 1 - req->src_len, req->src);
 
-       req_ctx->out_buf = kmalloc(ctx->key_size,
-                       (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
-                       GFP_KERNEL : GFP_ATOMIC);
-       if (!req_ctx->out_buf) {
-               kfree(req_ctx->in_buf);
-               return -ENOMEM;
-       }
-
-       pkcs1pad_sg_set_buf(req_ctx->out_sg, req_ctx->out_buf,
-                       ctx->key_size, NULL);
-
        akcipher_request_set_tfm(&req_ctx->child_req, ctx->child);
        akcipher_request_set_callback(&req_ctx->child_req, req->base.flags,
                        pkcs1pad_encrypt_sign_complete_cb, req);
 
+       /* Reuse output buffer */
+       akcipher_request_set_crypt(&req_ctx->child_req, req_ctx->in_sg,
+                                  req->dst, ctx->key_size - 1, req->dst_len);
+
        err = crypto_akcipher_sign(&req_ctx->child_req);
        if (err != -EINPROGRESS &&
                        (err != -EBUSY ||
@@ -486,49 +455,55 @@ static int pkcs1pad_verify_complete(struct akcipher_request *req, int err)
        struct akcipher_instance *inst = akcipher_alg_instance(tfm);
        struct pkcs1pad_inst_ctx *ictx = akcipher_instance_ctx(inst);
        const struct rsa_asn1_template *digest_info = ictx->digest_info;
+       unsigned int dst_len;
        unsigned int pos;
-
-       if (err == -EOVERFLOW)
-               /* Decrypted value had no leading 0 byte */
-               err = -EINVAL;
+       u8 *out_buf;
 
        if (err)
                goto done;
 
-       if (req_ctx->child_req.dst_len != ctx->key_size - 1) {
-               err = -EINVAL;
+       err = -EINVAL;
+       dst_len = req_ctx->child_req.dst_len;
+       if (dst_len < ctx->key_size - 1)
                goto done;
+
+       out_buf = req_ctx->out_buf;
+       if (dst_len == ctx->key_size) {
+               if (out_buf[0] != 0x00)
+                       /* Decrypted value had no leading 0 byte */
+                       goto done;
+
+               dst_len--;
+               out_buf++;
        }
 
        err = -EBADMSG;
-       if (req_ctx->out_buf[0] != 0x01)
+       if (out_buf[0] != 0x01)
                goto done;
 
-       for (pos = 1; pos < req_ctx->child_req.dst_len; pos++)
-               if (req_ctx->out_buf[pos] != 0xff)
+       for (pos = 1; pos < dst_len; pos++)
+               if (out_buf[pos] != 0xff)
                        break;
 
-       if (pos < 9 || pos == req_ctx->child_req.dst_len ||
-           req_ctx->out_buf[pos] != 0x00)
+       if (pos < 9 || pos == dst_len || out_buf[pos] != 0x00)
                goto done;
        pos++;
 
-       if (memcmp(req_ctx->out_buf + pos, digest_info->data,
-                  digest_info->size))
+       if (memcmp(out_buf + pos, digest_info->data, digest_info->size))
                goto done;
 
        pos += digest_info->size;
 
        err = 0;
 
-       if (req->dst_len < req_ctx->child_req.dst_len - pos)
+       if (req->dst_len < dst_len - pos)
                err = -EOVERFLOW;
-       req->dst_len = req_ctx->child_req.dst_len - pos;
+       req->dst_len = dst_len - pos;
 
        if (!err)
                sg_copy_from_buffer(req->dst,
                                sg_nents_for_len(req->dst, req->dst_len),
-                               req_ctx->out_buf + pos, req->dst_len);
+                               out_buf + pos, req->dst_len);
 done:
        kzfree(req_ctx->out_buf);
 
@@ -568,18 +543,7 @@ static int pkcs1pad_verify(struct akcipher_request *req)
        if (!ctx->key_size || req->src_len < ctx->key_size)
                return -EINVAL;
 
-       if (ctx->key_size > PAGE_SIZE)
-               return -ENOTSUPP;
-
-       /* Reuse input buffer, output to a new buffer */
-       req_ctx->child_req.src = req->src;
-       req_ctx->child_req.src_len = req->src_len;
-       req_ctx->child_req.dst = req_ctx->out_sg;
-       req_ctx->child_req.dst_len = ctx->key_size;
-
-       req_ctx->out_buf = kmalloc(ctx->key_size,
-                       (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
-                       GFP_KERNEL : GFP_ATOMIC);
+       req_ctx->out_buf = kmalloc(ctx->key_size, GFP_KERNEL);
        if (!req_ctx->out_buf)
                return -ENOMEM;
 
@@ -590,6 +554,11 @@ static int pkcs1pad_verify(struct akcipher_request *req)
        akcipher_request_set_callback(&req_ctx->child_req, req->base.flags,
                        pkcs1pad_verify_complete_cb, req);
 
+       /* Reuse input buffer, output to a new buffer */
+       akcipher_request_set_crypt(&req_ctx->child_req, req->src,
+                                  req_ctx->out_sg, req->src_len,
+                                  ctx->key_size);
+
        err = crypto_akcipher_verify(&req_ctx->child_req);
        if (err != -EINPROGRESS &&
                        (err != -EBUSY ||
This page took 0.029187 seconds and 5 git commands to generate.