1 /***********************license start************************************
2 * Copyright (c) 2003-2017 Cavium, Inc.
3 * All rights reserved.
4 *
5 * License: one of 'Cavium License' or 'GNU General Public License Version 2'
6 *
7 * This file is provided under the terms of the Cavium License (see below)
8 * or under the terms of GNU General Public License, Version 2, as
9 * published by the Free Software Foundation. When using or redistributing
10 * this file, you may do so under either license.
11 *
12 * Cavium License: Redistribution and use in source and binary forms, with
13 * or without modification, are permitted provided that the following
14 * conditions are met:
15 *
16 * * Redistributions of source code must retain the above copyright
17 * notice, this list of conditions and the following disclaimer.
18 *
19 * * Redistributions in binary form must reproduce the above
20 * copyright notice, this list of conditions and the following
21 * disclaimer in the documentation and/or other materials provided
22 * with the distribution.
23 *
24 * * Neither the name of Cavium Inc. nor the names of its contributors may be
25 * used to endorse or promote products derived from this software without
26 * specific prior written permission.
27 *
28 * This Software, including technical data, may be subject to U.S. export
29 * control laws, including the U.S. Export Administration Act and its
30 * associated regulations, and may be subject to export or import
31 * regulations in other countries.
32 *
33 * TO THE MAXIMUM EXTENT PERMITTED BY LAW, THE SOFTWARE IS PROVIDED "AS IS"
34 * AND WITH ALL FAULTS AND CAVIUM INC. MAKES NO PROMISES, REPRESENTATIONS
35 * OR WARRANTIES, EITHER EXPRESS, IMPLIED, STATUTORY, OR OTHERWISE, WITH
36 * RESPECT TO THE SOFTWARE, INCLUDING ITS CONDITION, ITS CONFORMITY TO ANY
37 * REPRESENTATION OR DESCRIPTION, OR THE EXISTENCE OF ANY LATENT OR PATENT
38 * DEFECTS, AND CAVIUM SPECIFICALLY DISCLAIMS ALL IMPLIED (IF ANY)
39 * WARRANTIES OF TITLE, MERCHANTABILITY, NONINFRINGEMENT, FITNESS FOR A
40 * PARTICULAR PURPOSE, LACK OF VIRUSES, ACCURACY OR COMPLETENESS, QUIET
41 * ENJOYMENT, QUIET POSSESSION OR CORRESPONDENCE TO DESCRIPTION. THE
42 * ENTIRE RISK ARISING OUT OF USE OR PERFORMANCE OF THE SOFTWARE LIES
43 * WITH YOU.
44 ***********************license end**************************************/
45
46 #include "zip_crypto.h"
47
zip_static_init_zip_ops(struct zip_operation * zip_ops,int lzs_flag)48 static void zip_static_init_zip_ops(struct zip_operation *zip_ops,
49 int lzs_flag)
50 {
51 zip_ops->flush = ZIP_FLUSH_FINISH;
52
53 /* equivalent to level 6 of opensource zlib */
54 zip_ops->speed = 1;
55
56 if (!lzs_flag) {
57 zip_ops->ccode = 0; /* Auto Huffman */
58 zip_ops->lzs_flag = 0;
59 zip_ops->format = ZLIB_FORMAT;
60 } else {
61 zip_ops->ccode = 3; /* LZS Encoding */
62 zip_ops->lzs_flag = 1;
63 zip_ops->format = LZS_FORMAT;
64 }
65 zip_ops->begin_file = 1;
66 zip_ops->history_len = 0;
67 zip_ops->end_file = 1;
68 zip_ops->compcode = 0;
69 zip_ops->csum = 1; /* Adler checksum desired */
70 }
71
zip_ctx_init(struct zip_kernel_ctx * zip_ctx,int lzs_flag)72 static int zip_ctx_init(struct zip_kernel_ctx *zip_ctx, int lzs_flag)
73 {
74 struct zip_operation *comp_ctx = &zip_ctx->zip_comp;
75 struct zip_operation *decomp_ctx = &zip_ctx->zip_decomp;
76
77 zip_static_init_zip_ops(comp_ctx, lzs_flag);
78 zip_static_init_zip_ops(decomp_ctx, lzs_flag);
79
80 comp_ctx->input = zip_data_buf_alloc(MAX_INPUT_BUFFER_SIZE);
81 if (!comp_ctx->input)
82 return -ENOMEM;
83
84 comp_ctx->output = zip_data_buf_alloc(MAX_OUTPUT_BUFFER_SIZE);
85 if (!comp_ctx->output)
86 goto err_comp_input;
87
88 decomp_ctx->input = zip_data_buf_alloc(MAX_INPUT_BUFFER_SIZE);
89 if (!decomp_ctx->input)
90 goto err_comp_output;
91
92 decomp_ctx->output = zip_data_buf_alloc(MAX_OUTPUT_BUFFER_SIZE);
93 if (!decomp_ctx->output)
94 goto err_decomp_input;
95
96 return 0;
97
98 err_decomp_input:
99 zip_data_buf_free(decomp_ctx->input, MAX_INPUT_BUFFER_SIZE);
100
101 err_comp_output:
102 zip_data_buf_free(comp_ctx->output, MAX_OUTPUT_BUFFER_SIZE);
103
104 err_comp_input:
105 zip_data_buf_free(comp_ctx->input, MAX_INPUT_BUFFER_SIZE);
106
107 return -ENOMEM;
108 }
109
zip_ctx_exit(struct zip_kernel_ctx * zip_ctx)110 static void zip_ctx_exit(struct zip_kernel_ctx *zip_ctx)
111 {
112 struct zip_operation *comp_ctx = &zip_ctx->zip_comp;
113 struct zip_operation *dec_ctx = &zip_ctx->zip_decomp;
114
115 zip_data_buf_free(comp_ctx->input, MAX_INPUT_BUFFER_SIZE);
116 zip_data_buf_free(comp_ctx->output, MAX_OUTPUT_BUFFER_SIZE);
117
118 zip_data_buf_free(dec_ctx->input, MAX_INPUT_BUFFER_SIZE);
119 zip_data_buf_free(dec_ctx->output, MAX_OUTPUT_BUFFER_SIZE);
120 }
121
zip_compress(const u8 * src,unsigned int slen,u8 * dst,unsigned int * dlen,struct zip_kernel_ctx * zip_ctx)122 static int zip_compress(const u8 *src, unsigned int slen,
123 u8 *dst, unsigned int *dlen,
124 struct zip_kernel_ctx *zip_ctx)
125 {
126 struct zip_operation *zip_ops = NULL;
127 struct zip_state *zip_state;
128 struct zip_device *zip = NULL;
129 int ret;
130
131 if (!zip_ctx || !src || !dst || !dlen)
132 return -ENOMEM;
133
134 zip = zip_get_device(zip_get_node_id());
135 if (!zip)
136 return -ENODEV;
137
138 zip_state = kzalloc(sizeof(*zip_state), GFP_ATOMIC);
139 if (!zip_state)
140 return -ENOMEM;
141
142 zip_ops = &zip_ctx->zip_comp;
143
144 zip_ops->input_len = slen;
145 zip_ops->output_len = *dlen;
146 memcpy(zip_ops->input, src, slen);
147
148 ret = zip_deflate(zip_ops, zip_state, zip);
149
150 if (!ret) {
151 *dlen = zip_ops->output_len;
152 memcpy(dst, zip_ops->output, *dlen);
153 }
154 kfree(zip_state);
155 return ret;
156 }
157
zip_decompress(const u8 * src,unsigned int slen,u8 * dst,unsigned int * dlen,struct zip_kernel_ctx * zip_ctx)158 static int zip_decompress(const u8 *src, unsigned int slen,
159 u8 *dst, unsigned int *dlen,
160 struct zip_kernel_ctx *zip_ctx)
161 {
162 struct zip_operation *zip_ops = NULL;
163 struct zip_state *zip_state;
164 struct zip_device *zip = NULL;
165 int ret;
166
167 if (!zip_ctx || !src || !dst || !dlen)
168 return -ENOMEM;
169
170 zip = zip_get_device(zip_get_node_id());
171 if (!zip)
172 return -ENODEV;
173
174 zip_state = kzalloc(sizeof(*zip_state), GFP_ATOMIC);
175 if (!zip_state)
176 return -ENOMEM;
177
178 zip_ops = &zip_ctx->zip_decomp;
179 memcpy(zip_ops->input, src, slen);
180
181 /* Work around for a bug in zlib which needs an extra bytes sometimes */
182 if (zip_ops->ccode != 3) /* Not LZS Encoding */
183 zip_ops->input[slen++] = 0;
184
185 zip_ops->input_len = slen;
186 zip_ops->output_len = *dlen;
187
188 ret = zip_inflate(zip_ops, zip_state, zip);
189
190 if (!ret) {
191 *dlen = zip_ops->output_len;
192 memcpy(dst, zip_ops->output, *dlen);
193 }
194 kfree(zip_state);
195 return ret;
196 }
197
198 /* Legacy Compress framework start */
zip_alloc_comp_ctx_deflate(struct crypto_tfm * tfm)199 int zip_alloc_comp_ctx_deflate(struct crypto_tfm *tfm)
200 {
201 int ret;
202 struct zip_kernel_ctx *zip_ctx = crypto_tfm_ctx(tfm);
203
204 ret = zip_ctx_init(zip_ctx, 0);
205
206 return ret;
207 }
208
zip_alloc_comp_ctx_lzs(struct crypto_tfm * tfm)209 int zip_alloc_comp_ctx_lzs(struct crypto_tfm *tfm)
210 {
211 int ret;
212 struct zip_kernel_ctx *zip_ctx = crypto_tfm_ctx(tfm);
213
214 ret = zip_ctx_init(zip_ctx, 1);
215
216 return ret;
217 }
218
zip_free_comp_ctx(struct crypto_tfm * tfm)219 void zip_free_comp_ctx(struct crypto_tfm *tfm)
220 {
221 struct zip_kernel_ctx *zip_ctx = crypto_tfm_ctx(tfm);
222
223 zip_ctx_exit(zip_ctx);
224 }
225
zip_comp_compress(struct crypto_tfm * tfm,const u8 * src,unsigned int slen,u8 * dst,unsigned int * dlen)226 int zip_comp_compress(struct crypto_tfm *tfm,
227 const u8 *src, unsigned int slen,
228 u8 *dst, unsigned int *dlen)
229 {
230 int ret;
231 struct zip_kernel_ctx *zip_ctx = crypto_tfm_ctx(tfm);
232
233 ret = zip_compress(src, slen, dst, dlen, zip_ctx);
234
235 return ret;
236 }
237
zip_comp_decompress(struct crypto_tfm * tfm,const u8 * src,unsigned int slen,u8 * dst,unsigned int * dlen)238 int zip_comp_decompress(struct crypto_tfm *tfm,
239 const u8 *src, unsigned int slen,
240 u8 *dst, unsigned int *dlen)
241 {
242 int ret;
243 struct zip_kernel_ctx *zip_ctx = crypto_tfm_ctx(tfm);
244
245 ret = zip_decompress(src, slen, dst, dlen, zip_ctx);
246
247 return ret;
248 } /* Legacy compress framework end */
249
250 /* SCOMP framework start */
zip_alloc_scomp_ctx_deflate(struct crypto_scomp * tfm)251 void *zip_alloc_scomp_ctx_deflate(struct crypto_scomp *tfm)
252 {
253 int ret;
254 struct zip_kernel_ctx *zip_ctx;
255
256 zip_ctx = kzalloc(sizeof(*zip_ctx), GFP_KERNEL);
257 if (!zip_ctx)
258 return ERR_PTR(-ENOMEM);
259
260 ret = zip_ctx_init(zip_ctx, 0);
261
262 if (ret) {
263 kfree_sensitive(zip_ctx);
264 return ERR_PTR(ret);
265 }
266
267 return zip_ctx;
268 }
269
zip_alloc_scomp_ctx_lzs(struct crypto_scomp * tfm)270 void *zip_alloc_scomp_ctx_lzs(struct crypto_scomp *tfm)
271 {
272 int ret;
273 struct zip_kernel_ctx *zip_ctx;
274
275 zip_ctx = kzalloc(sizeof(*zip_ctx), GFP_KERNEL);
276 if (!zip_ctx)
277 return ERR_PTR(-ENOMEM);
278
279 ret = zip_ctx_init(zip_ctx, 1);
280
281 if (ret) {
282 kfree_sensitive(zip_ctx);
283 return ERR_PTR(ret);
284 }
285
286 return zip_ctx;
287 }
288
zip_free_scomp_ctx(struct crypto_scomp * tfm,void * ctx)289 void zip_free_scomp_ctx(struct crypto_scomp *tfm, void *ctx)
290 {
291 struct zip_kernel_ctx *zip_ctx = ctx;
292
293 zip_ctx_exit(zip_ctx);
294 kfree_sensitive(zip_ctx);
295 }
296
zip_scomp_compress(struct crypto_scomp * tfm,const u8 * src,unsigned int slen,u8 * dst,unsigned int * dlen,void * ctx)297 int zip_scomp_compress(struct crypto_scomp *tfm,
298 const u8 *src, unsigned int slen,
299 u8 *dst, unsigned int *dlen, void *ctx)
300 {
301 int ret;
302 struct zip_kernel_ctx *zip_ctx = ctx;
303
304 ret = zip_compress(src, slen, dst, dlen, zip_ctx);
305
306 return ret;
307 }
308
zip_scomp_decompress(struct crypto_scomp * tfm,const u8 * src,unsigned int slen,u8 * dst,unsigned int * dlen,void * ctx)309 int zip_scomp_decompress(struct crypto_scomp *tfm,
310 const u8 *src, unsigned int slen,
311 u8 *dst, unsigned int *dlen, void *ctx)
312 {
313 int ret;
314 struct zip_kernel_ctx *zip_ctx = ctx;
315
316 ret = zip_decompress(src, slen, dst, dlen, zip_ctx);
317
318 return ret;
319 } /* SCOMP framework end */
320