crypto/lzo-rle.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * Cryptographic API.
   4  */
   5
   6 #include <linux/init.h>
   7 #include <linux/module.h>
   8 #include <linux/crypto.h>
   9 #include <linux/vmalloc.h>
  10 #include <linux/mm.h>
  11 #include <linux/lzo.h>
  12 #include <crypto/internal/scompress.h>
  13
  14 struct lzorle_ctx {
  15         void *lzorle_comp_mem;
  16 };
  17
  18 static void *lzorle_alloc_ctx(struct crypto_scomp *tfm)
  19 {
  20         void *ctx;
  21
  22         ctx = kvmalloc(LZO1X_MEM_COMPRESS, GFP_KERNEL);
  23         if (!ctx)
  24                 return ERR_PTR(-ENOMEM);
  25
  26         return ctx;
  27 }
  28
  29 static int lzorle_init(struct crypto_tfm *tfm)
  30 {
  31         struct lzorle_ctx *ctx = crypto_tfm_ctx(tfm);
  32
  33         ctx->lzorle_comp_mem = lzorle_alloc_ctx(NULL);
  34         if (IS_ERR(ctx->lzorle_comp_mem))
  35                 return -ENOMEM;
  36
  37         return 0;
  38 }
  39
  40 static void lzorle_free_ctx(struct crypto_scomp *tfm, void *ctx)
  41 {
  42         kvfree(ctx);
  43 }
  44
  45 static void lzorle_exit(struct crypto_tfm *tfm)
  46 {
  47         struct lzorle_ctx *ctx = crypto_tfm_ctx(tfm);
  48
  49         lzorle_free_ctx(NULL, ctx->lzorle_comp_mem);
  50 }
  51
  52 static int __lzorle_compress(const u8 *src, unsigned int slen,
  53                           u8 *dst, unsigned int *dlen, void *ctx)
  54 {
  55         size_t tmp_len = *dlen; /* size_t(ulong) <-> uint on 64 bit */
  56         int err;
  57
  58         err = lzorle1x_1_compress(src, slen, dst, &tmp_len, ctx);
  59
  60         if (err != LZO_E_OK)
  61                 return -EINVAL;
  62
  63         *dlen = tmp_len;
  64         return 0;
  65 }
  66
  67 static int lzorle_compress(struct crypto_tfm *tfm, const u8 *src,
  68                         unsigned int slen, u8 *dst, unsigned int *dlen)
  69 {
  70         struct lzorle_ctx *ctx = crypto_tfm_ctx(tfm);
  71
  72         return __lzorle_compress(src, slen, dst, dlen, ctx->lzorle_comp_mem);
  73 }
  74
  75 static int lzorle_scompress(struct crypto_scomp *tfm, const u8 *src,
  76                          unsigned int slen, u8 *dst, unsigned int *dlen,
  77                          void *ctx)
  78 {
  79         return __lzorle_compress(src, slen, dst, dlen, ctx);
  80 }
  81
  82 static int __lzorle_decompress(const u8 *src, unsigned int slen,
  83                             u8 *dst, unsigned int *dlen)
  84 {
  85         int err;
  86         size_t tmp_len = *dlen; /* size_t(ulong) <-> uint on 64 bit */
  87
  88         err = lzo1x_decompress_safe(src, slen, dst, &tmp_len);
  89
  90         if (err != LZO_E_OK)
  91                 return -EINVAL;
  92
  93         *dlen = tmp_len;
  94         return 0;
  95 }
  96
  97 static int lzorle_decompress(struct crypto_tfm *tfm, const u8 *src,
  98                           unsigned int slen, u8 *dst, unsigned int *dlen)
  99 {
 100         return __lzorle_decompress(src, slen, dst, dlen);
 101 }
 102
 103 static int lzorle_sdecompress(struct crypto_scomp *tfm, const u8 *src,
 104                            unsigned int slen, u8 *dst, unsigned int *dlen,
 105                            void *ctx)
 106 {
 107         return __lzorle_decompress(src, slen, dst, dlen);
 108 }
 109
 110 static struct crypto_alg alg = {
 111         .cra_name               = "lzo-rle",
 112         .cra_flags              = CRYPTO_ALG_TYPE_COMPRESS,
 113         .cra_ctxsize            = sizeof(struct lzorle_ctx),
 114         .cra_module             = THIS_MODULE,
 115         .cra_init               = lzorle_init,
 116         .cra_exit               = lzorle_exit,
 117         .cra_u                  = { .compress = {
 118         .coa_compress           = lzorle_compress,
 119         .coa_decompress         = lzorle_decompress } }
 120 };
 121
 122 static struct scomp_alg scomp = {
 123         .alloc_ctx              = lzorle_alloc_ctx,
 124         .free_ctx               = lzorle_free_ctx,
 125         .compress               = lzorle_scompress,
 126         .decompress             = lzorle_sdecompress,
 127         .base                   = {
 128                 .cra_name       = "lzo-rle",
 129                 .cra_driver_name = "lzo-rle-scomp",
 130                 .cra_module      = THIS_MODULE,
 131         }
 132 };
 133
 134 static int __init lzorle_mod_init(void)
 135 {
 136         int ret;
 137
 138         ret = crypto_register_alg(&alg);
 139         if (ret)
 140                 return ret;
 141
 142         ret = crypto_register_scomp(&scomp);
 143         if (ret) {
 144                 crypto_unregister_alg(&alg);
 145                 return ret;
 146         }
 147
 148         return ret;
 149 }
 150
 151 static void __exit lzorle_mod_fini(void)
 152 {
 153         crypto_unregister_alg(&alg);
 154         crypto_unregister_scomp(&scomp);
 155 }
 156
 157 subsys_initcall(lzorle_mod_init);
 158 module_exit(lzorle_mod_fini);
 159
 160 MODULE_LICENSE("GPL");
 161 MODULE_DESCRIPTION("LZO-RLE Compression Algorithm");
 162 MODULE_ALIAS_CRYPTO("lzo-rle");