fedora core 6 1.2949 + vserver 2.2.0

[linux-2.6.git] / crypto / deflate.c

diff --git a/crypto/deflate.c b/crypto/deflate.c
index e2726b6..6588bbf 100644 (file)
--- a/crypto/deflate.c
+++ b/crypto/deflate.c
@@ -39,44 +39,16 @@
 #define DEFLATE_DEF_MEMLEVEL           MAX_MEM_LEVEL
 
 struct deflate_ctx {
-       int comp_initialized;
-       int decomp_initialized;
        struct z_stream_s comp_stream;
        struct z_stream_s decomp_stream;
 };
 
-static inline int deflate_gfp(void)
-{
-       return in_softirq() ? GFP_ATOMIC : GFP_KERNEL;
-}
-
-static int deflate_init(void *ctx)
-{
-       return 0;
-}
-
-static void deflate_exit(void *ctx)
-{
-       struct deflate_ctx *dctx = ctx;
-
-       if (dctx->comp_initialized)
-               vfree(dctx->comp_stream.workspace);
-       if (dctx->decomp_initialized)
-               kfree(dctx->decomp_stream.workspace);
-}
-
-/*
- * Lazy initialization to make interface simple without allocating
- * un-needed workspaces.  Thus can be called in softirq context.
- */
 static int deflate_comp_init(struct deflate_ctx *ctx)
 {
        int ret = 0;
        struct z_stream_s *stream = &ctx->comp_stream;
 
-       stream->workspace = __vmalloc(zlib_deflate_workspacesize(),
-                                     deflate_gfp()|__GFP_HIGHMEM,
-                                     PAGE_KERNEL);
+       stream->workspace = vmalloc(zlib_deflate_workspacesize());
        if (!stream->workspace ) {
                ret = -ENOMEM;
                goto out;
@@ -89,7 +61,6 @@ static int deflate_comp_init(struct deflate_ctx *ctx)
                ret = -EINVAL;
                goto out_free;
        }
-       ctx->comp_initialized = 1;
 out:   
        return ret;
 out_free:
@@ -102,19 +73,16 @@ static int deflate_decomp_init(struct deflate_ctx *ctx)
        int ret = 0;
        struct z_stream_s *stream = &ctx->decomp_stream;
 
-       stream->workspace = kmalloc(zlib_inflate_workspacesize(),
-                                   deflate_gfp());
+       stream->workspace = kzalloc(zlib_inflate_workspacesize(), GFP_KERNEL);
        if (!stream->workspace ) {
                ret = -ENOMEM;
                goto out;
        }
-       memset(stream->workspace, 0, zlib_inflate_workspacesize());
        ret = zlib_inflateInit2(stream, -DEFLATE_DEF_WINBITS);
        if (ret != Z_OK) {
                ret = -EINVAL;
                goto out_free;
        }
-       ctx->decomp_initialized = 1;
 out:
        return ret;
 out_free:
@@ -122,19 +90,48 @@ out_free:
        goto out;
 }
 
-static int deflate_compress(void *ctx, const u8 *src, unsigned int slen,
-                           u8 *dst, unsigned int *dlen)
+static void deflate_comp_exit(struct deflate_ctx *ctx)
+{
+       zlib_deflateEnd(&ctx->comp_stream);
+       vfree(ctx->comp_stream.workspace);
+}
+
+static void deflate_decomp_exit(struct deflate_ctx *ctx)
+{
+       zlib_inflateEnd(&ctx->decomp_stream);
+       kfree(ctx->decomp_stream.workspace);
+}
+
+static int deflate_init(struct crypto_tfm *tfm)
+{
+       struct deflate_ctx *ctx = crypto_tfm_ctx(tfm);
+       int ret;
+       
+       ret = deflate_comp_init(ctx);
+       if (ret)
+               goto out;
+       ret = deflate_decomp_init(ctx);
+       if (ret)
+               deflate_comp_exit(ctx);
+out:
+       return ret;
+}
+
+static void deflate_exit(struct crypto_tfm *tfm)
+{
+       struct deflate_ctx *ctx = crypto_tfm_ctx(tfm);
+
+       deflate_comp_exit(ctx);
+       deflate_decomp_exit(ctx);
+}
+
+static int deflate_compress(struct crypto_tfm *tfm, const u8 *src,
+                           unsigned int slen, u8 *dst, unsigned int *dlen)
 {
        int ret = 0;
-       struct deflate_ctx *dctx = ctx;
+       struct deflate_ctx *dctx = crypto_tfm_ctx(tfm);
        struct z_stream_s *stream = &dctx->comp_stream;
 
-       if (!dctx->comp_initialized) {
-               ret = deflate_comp_init(dctx);
-               if (ret)
-                       goto out;
-       }
-
        ret = zlib_deflateReset(stream);
        if (ret != Z_OK) {
                ret = -EINVAL;
@@ -157,20 +154,14 @@ out:
        return ret;
 }
  
-static int deflate_decompress(void *ctx, const u8 *src, unsigned int slen,
-                              u8 *dst, unsigned int *dlen)
+static int deflate_decompress(struct crypto_tfm *tfm, const u8 *src,
+                             unsigned int slen, u8 *dst, unsigned int *dlen)
 {
        
        int ret = 0;
-       struct deflate_ctx *dctx = ctx;
+       struct deflate_ctx *dctx = crypto_tfm_ctx(tfm);
        struct z_stream_s *stream = &dctx->decomp_stream;
 
-       if (!dctx->decomp_initialized) {
-               ret = deflate_decomp_init(dctx);
-               if (ret)
-                       goto out;
-       }
-
        ret = zlib_inflateReset(stream);
        if (ret != Z_OK) {
                ret = -EINVAL;
@@ -210,9 +201,9 @@ static struct crypto_alg alg = {
        .cra_ctxsize            = sizeof(struct deflate_ctx),
        .cra_module             = THIS_MODULE,
        .cra_list               = LIST_HEAD_INIT(alg.cra_list),
+       .cra_init               = deflate_init,
+       .cra_exit               = deflate_exit,
        .cra_u                  = { .compress = {
-       .coa_init               = deflate_init,
-       .coa_exit               = deflate_exit,
        .coa_compress           = deflate_compress,
        .coa_decompress         = deflate_decompress } }
 };