1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Cryptographic API.
4  */
5 
6 #include <crypto/internal/scompress.h>
7 #include <linux/init.h>
8 #include <linux/lzo.h>
9 #include <linux/module.h>
10 #include <linux/slab.h>
11 
12 struct lzo_ctx {
13 	void *lzo_comp_mem;
14 };
15 
lzo_alloc_ctx(void)16 static void *lzo_alloc_ctx(void)
17 {
18 	void *ctx;
19 
20 	ctx = kvmalloc(LZO1X_MEM_COMPRESS, GFP_KERNEL);
21 	if (!ctx)
22 		return ERR_PTR(-ENOMEM);
23 
24 	return ctx;
25 }
26 
lzo_free_ctx(void * ctx)27 static void lzo_free_ctx(void *ctx)
28 {
29 	kvfree(ctx);
30 }
31 
__lzo_compress(const u8 * src,unsigned int slen,u8 * dst,unsigned int * dlen,void * ctx)32 static int __lzo_compress(const u8 *src, unsigned int slen,
33 			  u8 *dst, unsigned int *dlen, void *ctx)
34 {
35 	size_t tmp_len = *dlen; /* size_t(ulong) <-> uint on 64 bit */
36 	int err;
37 
38 	err = lzo1x_1_compress_safe(src, slen, dst, &tmp_len, ctx);
39 
40 	if (err != LZO_E_OK)
41 		return -EINVAL;
42 
43 	*dlen = tmp_len;
44 	return 0;
45 }
46 
lzo_scompress(struct crypto_scomp * tfm,const u8 * src,unsigned int slen,u8 * dst,unsigned int * dlen,void * ctx)47 static int lzo_scompress(struct crypto_scomp *tfm, const u8 *src,
48 			 unsigned int slen, u8 *dst, unsigned int *dlen,
49 			 void *ctx)
50 {
51 	return __lzo_compress(src, slen, dst, dlen, ctx);
52 }
53 
__lzo_decompress(const u8 * src,unsigned int slen,u8 * dst,unsigned int * dlen)54 static int __lzo_decompress(const u8 *src, unsigned int slen,
55 			    u8 *dst, unsigned int *dlen)
56 {
57 	int err;
58 	size_t tmp_len = *dlen; /* size_t(ulong) <-> uint on 64 bit */
59 
60 	err = lzo1x_decompress_safe(src, slen, dst, &tmp_len);
61 
62 	if (err != LZO_E_OK)
63 		return -EINVAL;
64 
65 	*dlen = tmp_len;
66 	return 0;
67 }
68 
lzo_sdecompress(struct crypto_scomp * tfm,const u8 * src,unsigned int slen,u8 * dst,unsigned int * dlen,void * ctx)69 static int lzo_sdecompress(struct crypto_scomp *tfm, const u8 *src,
70 			   unsigned int slen, u8 *dst, unsigned int *dlen,
71 			   void *ctx)
72 {
73 	return __lzo_decompress(src, slen, dst, dlen);
74 }
75 
76 static struct scomp_alg scomp = {
77 	.alloc_ctx		= lzo_alloc_ctx,
78 	.free_ctx		= lzo_free_ctx,
79 	.compress		= lzo_scompress,
80 	.decompress		= lzo_sdecompress,
81 	.base			= {
82 		.cra_name	= "lzo",
83 		.cra_driver_name = "lzo-scomp",
84 		.cra_module	 = THIS_MODULE,
85 	}
86 };
87 
lzo_mod_init(void)88 static int __init lzo_mod_init(void)
89 {
90 	return crypto_register_scomp(&scomp);
91 }
92 
lzo_mod_fini(void)93 static void __exit lzo_mod_fini(void)
94 {
95 	crypto_unregister_scomp(&scomp);
96 }
97 
98 subsys_initcall(lzo_mod_init);
99 module_exit(lzo_mod_fini);
100 
101 MODULE_LICENSE("GPL");
102 MODULE_DESCRIPTION("LZO Compression Algorithm");
103 MODULE_ALIAS_CRYPTO("lzo");
104